Numa-aware, growable physical memory allocator (pmalloc)

Change-Id: Ic4c46bc733afae8bf0d8146623ed15633928de30
Signed-off-by: Damjan Marion <damarion@cisco.com>

1 files changed, 169 insertions, 0 deletions

diff --git a/src/vppinfra/test_pmalloc.c b/src/vppinfra/test_pmalloc.c
new file mode 100644
index 00000000000..c9ae01fc936
--- /dev/null
+++ b/src/vppinfra/test_pmalloc.c
@@ -0,0 +1,169 @@
+/*
+ * 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/format.h>
+#include <vppinfra/pmalloc.h>
+#include <vppinfra/random.h>
+
+typedef struct
+{
+  uword baseva;
+  uword size;
+  uword *vas;
+  u32 nitems;
+  u32 item_size;
+  u32 align;
+  int max_numa;
+  u32 arena_pages;
+  u32 arena_numa;
+  u32 arena_items;
+  int verbose;
+  clib_pmalloc_main_t pmalloc_main;
+} test_main_t;
+
+test_main_t test_main;
+
+clib_error_t *
+test_palloc (test_main_t * tm)
+{
+  clib_pmalloc_main_t *pm = &tm->pmalloc_main;
+  void *arena;
+  int i;
+  uword *va;
+
+  if (clib_pmalloc_init (pm, 0) != 0)
+    return clib_error_return (0, "pmalloc init failure");
+
+  fformat (stdout, "Allocate %d items...\n", tm->nitems);
+
+  for (i = 0; i < tm->nitems; i++)
+    {
+      u32 size = tm->item_size ? tm->item_size : 64 + 64 * (i % 8);
+      u32 align = tm->align ? tm->align : 64 << (i % 5);
+      u32 numa = i % (tm->max_numa + 1);
+      va = clib_pmalloc_alloc_aligned_on_numa (pm, size, align, numa);
+
+      if (va == 0)
+	clib_error ("Failed to alloc %u byte chunk with align %u on numa %u,"
+		    "\nerror: %U", size, align, numa, format_clib_error,
+		    clib_pmalloc_last_error (pm));
+
+      if ((pointer_to_uword (va) & (align - 1)) != 0)
+	clib_error (0, "Alignment error: %p not aligned with %u", va, align);
+
+      vec_add1 (tm->vas, pointer_to_uword (va));
+    }
+  fformat (stdout, "%U\n", format_pmalloc, pm, tm->verbose);
+
+  /* alloc from arena */
+  if (tm->arena_items)
+    {
+      fformat (stdout, "Allocate %d items from arena ...\n", tm->arena_items);
+      arena = clib_pmalloc_create_shared_arena (pm, "test arena",
+						tm->arena_pages << 21,
+						tm->arena_numa);
+      if (arena == 0)
+	clib_error ("Failed to alloc shared arena: %U", format_clib_error,
+		    clib_pmalloc_last_error (pm));
+
+      for (i = 0; i < tm->arena_items; i++)
+	{
+	  u32 size = tm->item_size ? tm->item_size : 64 + 64 * (i % 8);
+	  u32 align = tm->align ? tm->align : 64 << (i % 5);
+	  va = clib_pmalloc_alloc_from_arena (pm, arena, size, align);
+	  vec_add1 (tm->vas, pointer_to_uword (va));
+	}
+      fformat (stdout, "\n%U\n", format_pmalloc, pm, tm->verbose);
+    }
+
+
+  fformat (stdout, "Freeing %d items ...\n", vec_len (tm->vas));
+  for (i = 0; i < vec_len (tm->vas); i++)
+    clib_pmalloc_free (pm, (void *) tm->vas[i]);
+
+  fformat (stdout, "\n%U\n", format_pmalloc, pm, tm->verbose);
+  return 0;
+}
+
+clib_error_t *
+test_palloc_main (unformat_input_t * i)
+{
+  test_main_t *tm = &test_main;
+  clib_error_t *error;
+
+  tm->nitems = 5;
+  tm->arena_pages = 2;
+  tm->arena_numa = CLIB_PMALLOC_NUMA_LOCAL;
+
+  while (unformat_check_input (i) != UNFORMAT_END_OF_INPUT)
+    {
+      if (unformat (i, "nitems %u", &tm->nitems))
+	;
+      else if (unformat (i, "max-numa %u", &tm->max_numa))
+	;
+      else if (unformat (i, "item-size %u", &tm->item_size))
+	;
+      else if (unformat (i, "align %u", &tm->align))
+	;
+      else if (unformat (i, "verbose %d", &tm->verbose))
+	;
+      else if (unformat (i, "arena-pages %u", &tm->arena_pages))
+	;
+      else if (unformat (i, "arena-numa %u", &tm->arena_numa))
+	;
+      else if (unformat (i, "arena-items %u", &tm->arena_items))
+	;
+      else if (unformat (i, "verbose"))
+	tm->verbose = 1;
+      else
+	return clib_error_return (0, "unknown input '%U'",
+				  format_unformat_error, i);
+    }
+
+  error = test_palloc (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_palloc_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:
+ */