gnunetutil: add 2d and 3d allocation including tests

2 files changed, 95 insertions, 0 deletions

diff --git a/src/util/common_allocation.c b/src/util/common_allocation.c
index 71dd46bf1..f0299a223 100644
--- a/src/util/common_allocation.c
+++ b/src/util/common_allocation.c
@@ -86,6 +86,71 @@ GNUNET_xmalloc_ (size_t size,
 
 
 /**
+ * Allocate memory for a two dimensional array in one block
+ * and set up pointers. Aborts if no more memory is available.
+ * Don't use GNUNET_xnew_array_2d_ directly. Use the
+ * #GNUNET_new_array_2d macro.
+ * The memory of the elements will be zero'ed out.
+ *
+ * @param n size of the first dimension
+ * @param m size of the second dimension
+ * @param elementSize size of a single element in bytes
+ * @return allocated memory, never NULL
+ */
+void **
+GNUNET_xnew_array_2d_ (size_t n, size_t m, size_t elementSize)
+{
+        /* use char pointer internally to avoid void pointer arithmetic warnings */
+        char **ret = GNUNET_malloc (n * sizeof (void *) +  /* 1. dim header */
+                                    n * m * elementSize);  /* element data */
+
+        for (size_t i = 0; i < n; i++)
+                ret[i] = (char *)ret +          /* base address */
+                         n * sizeof (void *) +  /* skip 1. dim header */
+                         i * m * elementSize;   /* skip to 2. dim row header */
+        return (void **)ret;
+}
+
+
+/**
+ * Allocate memory for a three dimensional array in one block
+ * and set up pointers. Aborts if no more memory is available.
+ * Don't use GNUNET_xnew_array_3d_ directly. Use the
+ * #GNUNET_new_array_3d macro.
+ * The memory of the elements will be zero'ed out.
+ *
+ * @param n size of the first dimension
+ * @param m size of the second dimension
+ * @param o size of the third dimension
+ * @param elementSize size of a single element in bytes
+ * @return allocated memory, never NULL
+ */
+void ***
+GNUNET_xnew_array_3d_ (size_t n, size_t m, size_t o, size_t elementSize)
+{
+        /* use char pointer internally to avoid void pointer arithmetic warnings */
+        char ***ret = GNUNET_malloc (n * sizeof (void **) +     /* 1. dim header */
+                                     n * m * sizeof (void *) +  /* 2. dim header */
+                                     n * m * o * elementSize);  /* element data */
+
+        for (size_t i = 0; i < n; i++)
+        {
+                /* need to cast to (char *) temporarily for byte level acuracy */
+                ret[i] = (char **)((char *)ret +             /* base address */
+                                   n * sizeof (void **) +    /* skip 1. dim header */
+                                   i * m * sizeof (void *)); /* skip to 2. dim header */
+                for (size_t j = 0; j < m; j++)
+                        ret[i][j] = (char *)ret +              /* base address */
+                                    n * sizeof (void **) +     /* skip 1. dim header */
+                                    n * m * sizeof (void *) +  /* skip 2. dim header */
+                                    i * m * o * elementSize +  /* skip to 2. dim part */
+                                        j * o * elementSize;   /* skip to 3. dim row data */
+        }
+        return (void ***)ret;
+}
+
+
+/**
  * Allocate and initialize memory. Checks the return value, aborts if no more
  * memory is available.  Don't use #GNUNET_xmemdup_() directly. Use the
  * GNUNET_memdup() macro.
diff --git a/src/util/test_common_allocation.c b/src/util/test_common_allocation.c
index aa4809f58..4ef98b629 100644
--- a/src/util/test_common_allocation.c
+++ b/src/util/test_common_allocation.c
@@ -30,6 +30,8 @@ check ()
 {
 #define MAX_TESTVAL 1024
   char *ptrs[MAX_TESTVAL];
+  unsigned int **a2;
+  char ***a3;
   int i;
   int j;
   int k;
@@ -93,6 +95,34 @@ check ()
   if (ptrs[0] != NULL)
     return 9;
 
+        /* GNUNET_new_array_2d tests */
+        a2 = GNUNET_new_array_2d (17, 22, unsigned int);
+        for (i = 0; i < 17; i++)
+        {
+                for (j = 0; j < 22; j++)
+                {
+                        if (0 != a2[i][j])
+                                return 10;
+                        a2[i][j] = i * 100 + j;
+                }
+        }
+        free (a2);
+
+        /* GNUNET_new_array_3d tests */
+        a3 = GNUNET_new_array_3d (2, 3, 4, char);
+        for (i = 0; i < 2; i++)
+        {
+                for (j = 0; j < 3; j++)
+                {
+                        for (k = 0; k < 4; k++)
+                        {
+                                if (0 != a3[i][j][k])
+                                        return 11;
+                                a3[i][j][k] = i * 100 + j * 10 + k;
+                        }
+                }
+        }
+        free (a3);
 
   return 0;
 }