summaryrefslogtreecommitdiffstats
path: root/src/vppinfra/test_valloc.c
blob: ea786db92d6d630d3356b90508f05fa1de9fc4a4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
@media only all and (prefers-color-scheme: dark) {
.highlight .hll { background-color: #49483e }
.highlight .c { color: #75715e } /* Comment */
.highlight .err { color: #960050; background-color: #1e0010 } /* Error */
.highlight .k { color: #66d9ef } /* Keyword */
.highlight .l { color: #ae81ff } /* Literal */
.highlight .n { color: #f8f8f2 } /* Name */
.highlight .o { color: #f92672 } /* Operator */
.highlight .p { color: #f8f8f2 } /* Punctuation */
.highlight .ch { color: #75715e } /* Comment.Hashbang */
.highlight .cm { color: #75715e } /* Comment.Multiline */
.highlight .cp { color: #75715e } /* Comment.Preproc */
.highlight .cpf { color: #75715e } /* Comment.PreprocFile */
.highlight .c1 { color: #75715e } /* Comment.Single */
.highlight .cs { color: #75715e } /* Comment.Special */
.highlight .gd { color: #f92672 } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .gi { color: #a6e22e } /* Generic.Inserted */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #75715e } /* Generic.Subheading */
.highlight .kc { color: #66d9ef } /* Keyword.Constant */
.highlight .kd { color: #66d9ef } /* Keyword.Declaration */
.highlight .kn { color: #f92672 } /* Keyword.Namespace */
.highlight .kp { color: #66d9ef } /* Keyword.Pseudo */
.highlight .kr { color: #66d9ef } /* Keyword.Reserved */
.highlight .kt { color: #66d9ef } /* Keyword.Type */
.highlight .ld { color: #e6db74 } /* Literal.Date */
.highlight .m { color: #ae81ff } /* Literal.Number */
.highlight .s { color: #e6db74 } /* Literal.String */
.highlight .na { color: #a6e22e } /* Name.Attribute */
.highlight .nb { color: #f8f8f2 } /* Name.Builtin */
.highlight .nc { color: #a6e22e } /* Name.Class */
.highlight .no { color: #66d9ef } /* Name.Constant */
.highlight .nd { color: #a6e22e } /* Name.Decorator */
.highlight .ni { color: #f8f8f2 } /* Name.Entity */
.highlight .ne { color: #a6e22e } /* Name.Exception */
.highlight .nf { color: #a6e22e } /* Name.Function */
.highlight .nl { color: #f8f8f2 } /* Name.Label */
.highlight .nn { color: #f8f8f2 } /* Name.Namespace */
.highlight .nx { color: #a6e22e } /* Name.Other */
.highlight .py { color: #f8f8f2 } /* Name.Property */
.highlight .nt { color: #f92672 } /* Name.Tag */
.highlight .nv { color: #f8f8f2 } /* Name.Variable */
.highlight .ow { color: #f92672 } /* Operator.Word */
.highlight .w { color: #f8f8f2 } /* Text.Whitespace */
.highlight .mb { color: #ae81ff } /* Literal.Number.Bin */
.highlight .mf { color: #ae81ff } /* Literal.Number.Float */
.highlight .mh { color: #ae81ff } /* Literal.Number.Hex */
.highlight .mi { color: #ae81ff } /* Literal.Number.Integer */
.highlight .mo { color: #ae81ff } /* Literal.Number.Oct */
.highlight .sa { color: #e6db74 } /* Literal.String.Affix */
.highlight .sb { color: #e6db74 } /* Literal.String.Backtick */
.highlight .sc { color: #e6db74 } /* Literal.String.Char */
.highlight .dl { color: #e6db74 } /* Literal.String.Delimiter */
.highlight .sd { color: #e6db74 } /* Literal.String.Doc */
.highlight .s2 { color: #e6db74 } /* Literal.String.Double */
.highlight .se { color: #ae81ff } /* Literal.String.Escape */
.highlight .sh { color: #e6db74 } /* Literal.String.Heredoc */
.highlight .si { color: #e6db74 } /* Literal.String.Interpol */
.highlight .sx { color: #e6db74 } /* Literal.String.Other */
.highlight .sr { color: #e6db74 } /* Literal.String.Regex */
.highlight .s1 { color: #e6db74 } /* Literal.String.Single */
.highlight .ss { color: #e6db74 } /* Literal.String.Symbol */
.highlight .bp { color: #f8f8f2 } /* Name.Builtin.Pseudo */
.highlight .fm { color: #a6e22e } /* Name.Function.Magic */
.highlight .vc { color: #f8f8f2 } /* Name.Variable.Class */
.highlight .vg { color: #f8f8f2 } /* Name.Variable.Global */
.highlight .vi { color: #f8f8f2 } /* Name.Variable.Instance */
.highlight .vm { color: #f8f8f2 } /* Name.Variable.Magic */
.highlight .il { color: #ae81ff } /* Literal.Number.Integer.Long */
}
@media (prefers-color-scheme: light) {
.highlight .hll { background-color: #ffffcc }
.highlight .c { color: #888888 } /* Comment */
.highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */
.highlight .k { color: #008800; font-weight: bold } /* Keyword */
.highlight .ch { color: #888888 } /* Comment.Hashbang */
.highlight .cm { color: #888888 } /* Comment.Multiline */
.highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */
.highlight .cpf { color: #888888 } /* Comment.PreprocFile */
.highlight .c1 { color: #888888 } /* Comment.Single */
.highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */
.highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .gr { color: #aa0000 } /* Generic.Error */
.highlight .gh { color: #333333 } /* Generic.Heading */
.highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */
.highlight .go { color: #888888 } /* Generic.Output */
.highlight .gp { color: #555555 } /* Generic.Prompt */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #666666 } /* Generic.Subheading */
.highlight .gt { color: #aa0000 } /* Generic.Traceback */
.highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */
.highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */
.highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */
.highlight .kp { color: #008800 } /* Keyword.Pseudo */
.highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */
.highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */
.highlight .m { color: #0000DD; font-weight: bold } /* Literal.Nu
/*
 * Copyright (c) 2018 Cisco and/or its affiliates.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <vppinfra/valloc.h>

u32
vl (void *p)
{
  return vec_len (p);
}

/*
 * GDB callable function: pe - call pool_elts - number of elements in a pool
 */
uword
pe (void *v)
{
  return (pool_elts (v));
}

typedef struct
{
  u32 seed;
  uword baseva;
  uword size;
  uword *basevas;
  u8 *item_in_table;
  u32 nitems;
  u32 niter;
  u32 item_size;
  int check_every_add_del;
  clib_valloc_main_t valloc_main;
  int verbose;
} test_main_t;

test_main_t test_main;

clib_error_t *
test_valloc (test_main_t * tm)
{
  clib_valloc_chunk_t _ip, *ip = &_ip;
  uword baseva;
  uword *p;
  int i, j, index;
  u32 currently_in_table;
  u32 found;

  ip->baseva = 0x20000000;
  ip->size = 1024;

  clib_valloc_init (&tm->valloc_main, ip, 1 /* lock */ );

  ip->baseva = 0x20000000 + 1024;
  ip->size = 1024 * 1024 * 1024 - 1024;
  clib_valloc_add_chunk (&tm->valloc_main, ip);

  fformat (stdout, "Allocate %d items...\n", tm->nitems);
  for (i = 0; i < tm->nitems; i++)
    {
      baseva = clib_valloc_alloc (&tm->valloc_main, 1024,
				  1 /* fail:os_out_of_memory */ );
      vec_add1 (tm->basevas, baseva);
      vec_add1 (tm->item_in_table, 1);
    }

  fformat (stdout, "Perform %d random add/delete operations...\n", tm->niter);

  for (i = 0; i < tm->niter; i++)
    {
      index = random_u32 (&tm->seed) % tm->nitems;
      /* Swap state of random entry */
      if (tm->item_in_table[index])
	{
	  if (0)
	    fformat (stdout, "free [%d] %llx\n", index, tm->basevas[index]);
	  clib_valloc_free (&tm->valloc_main, tm->basevas[index]);
	  tm->item_in_table[index] = 0;
	  tm->basevas[index] = ~0;
	}
      else
	{
	  baseva = clib_valloc_alloc (&tm->valloc_main, 1024,
				      1 /* fail:os_out_of_memory */ );
	  tm->basevas[index] = baseva;
	  tm->item_in_table[index] = 1;
	  if (0)
	    fformat (stdout, "alloc [%d] %llx\n", index, tm->basevas[index]);
	}

      /* Check our work... */
      if (tm->check_every_add_del)
	{
	  for (j = 0; j < tm->nitems; j++)
	    {
	      if (tm->item_in_table[j])
		{
		  p = hash_get ((&tm->valloc_main)->chunk_index_by_baseva,
				tm->basevas[j]);
		  if (p)
		    {
		      ip =
			pool_elt_at_index ((&tm->valloc_main)->chunks, p[0]);
		      ASSERT (ip->baseva == tm->basevas[j]);
		      ASSERT (ip->flags & CLIB_VALLOC_BUSY);
		    }
		}
	      else
		{
		  p = hash_get ((&tm->valloc_main)->chunk_index_by_baseva,
				tm->basevas[j]);
		  /* Have to check, it's OK for the block to have been fused */
		  if (p)
		    {
		      ip =
			pool_elt_at_index ((&tm->valloc_main)->chunks, p[0]);
		      if ((ip->flags & CLIB_VALLOC_BUSY))
			{
			  fformat (stdout, "BUG: baseva %llx chunk %d busy\n",
				   tm->basevas[j], p[0]);
			  fformat (stdout, "%U\n", format_valloc,
				   &tm->valloc_main, 1 /* verbose */ );
			  ASSERT ((ip->flags & CLIB_VALLOC_BUSY) == 0);
			}
		    }
		}
	    }
	}
    }

  currently_in_table = 0;

  for (i = 0; i < tm->nitems; i++)
    {
      currently_in_table += tm->item_in_table[i];
    }

  fformat (stdout, "Check that %d items in table can be found...\n",
	   currently_in_table);

  found = 0;

  for (i = 0; i < tm->nitems; i++)
    {
      if (tm->item_in_table[i])
	{
	  p = hash_get ((&tm->valloc_main)->chunk_index_by_baseva,
			tm->basevas[i]);
	  if (p)
	    {
	      ip = pool_elt_at_index ((&tm->valloc_main)->chunks, p[0]);
	      ASSERT (ip->baseva == tm->basevas[i]);
	      ASSERT (ip->flags & CLIB_VALLOC_BUSY);
	    }
	  found++;
	}
      else
	{
	  p = hash_get ((&tm->valloc_main)->chunk_index_by_baseva,
			tm->basevas[i]);
	  /* Have to check, it's OK for the block to have been fused */
	  if (p)
	    {
	      ip = pool_elt_at_index ((&tm->valloc_main)->chunks, p[0]);
	      if ((ip->flags & CLIB_VALLOC_BUSY))
		{
		  fformat (stdout, "BUG: baseva %llx chunk %d busy\n",
			   tm->basevas[i], p[0]);
		  fformat (stdout, "%U\n", format_valloc,
			   &tm->valloc_main, 1 /* verbose */ );
		  ASSERT ((ip->flags & CLIB_VALLOC_BUSY) == 0);
		}
	    }
	}
    }

  fformat (stdout, "Found %d items in table...\n", found);

  for (i = 0; i < tm->nitems; i++)
    {
      if (tm->item_in_table[i])
	clib_valloc_free (&tm->valloc_main, tm->basevas[i]);
    }

  fformat (stdout, "%U", format_valloc, &tm->valloc_main, 1 /* verbose */ );

  return 0;
}

clib_error_t *
test_valloc_main (unformat_input_t * i)
{
  test_main_t *tm = &test_main;
  clib_error_t *error;

  tm->seed = 0xdeaddabe;
  tm->nitems = 5;
  tm->niter = 100;
  tm->item_size = 1024;

  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
    {
      if (unformat (i, "seed %u", &tm->seed))
	;
      else if (unformat (i, "nitems %u", &tm->nitems))
	;
      else if (unformat (i, "niter %u", &tm->niter))
	;
      else if (unformat (i, "item-size %u", &tm->item_size))
	;
      else if (unformat (i, "check-every-add-del"))
	tm->check_every_add_del = 1;
      else if (unformat (i, "verbose %d", &tm->verbose))
	;
      else if (unformat (i, "verbose"))
	tm->verbose = 1;
      else
	return clib_error_return (0, "unknown input '%U'",
				  format_unformat_error, i);
    }

  error = test_valloc (tm);

  return error;
}

#ifdef CLIB_UNIX
int
main (int argc, char *argv[])
{
  unformat_input_t i;
  int rv = 0;
  clib_error_t *error;

  clib_mem_init (0, 3ULL << 30);

  unformat_init_command_line (&i, argv);
  error = test_valloc_main (&i);
  if (error)
    {
      clib_error_report (error);
      rv = 1;
    }
  unformat_free (&i);

  return rv;
}
#endif /* CLIB_UNIX */

/*
 * fd.io coding-style-patch-verification: ON
 *
 * Local Variables:
 * eval: (c-set-style "gnu")
 * End:
 */