tests/slice-test.c

   1 /* GLIB sliced memory - fast threaded memory chunk allocator
   2  * Copyright (C) 2005 Tim Janik
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19 #include <glib.h>
  20
  21 #include <stdio.h>
  22 #include <string.h>
  23
  24 #define quick_rand32()  (rand_accu = 1664525 * rand_accu + 1013904223, rand_accu)
  25 static guint    prime_size = 1021; /* 769; 509 */
  26 static gboolean clean_memchunks = FALSE;
  27 static guint    number_of_blocks = 10000;          /* total number of blocks allocated */
  28 static guint    number_of_repetitions = 10000;     /* number of alloc+free repetitions */
  29 static gboolean want_corruption = FALSE;
  30
  31 /* --- old memchunk prototypes (memchunks.c) --- */
  32 GMemChunk*      old_mem_chunk_new       (const gchar  *name,
  33                                          gint          atom_size,
  34                                          gulong        area_size,
  35                                          gint          type);
  36 void            old_mem_chunk_destroy   (GMemChunk *mem_chunk);
  37 gpointer        old_mem_chunk_alloc     (GMemChunk *mem_chunk);
  38 gpointer        old_mem_chunk_alloc0    (GMemChunk *mem_chunk);
  39 void            old_mem_chunk_free      (GMemChunk *mem_chunk,
  40                                          gpointer   mem);
  41 void            old_mem_chunk_clean     (GMemChunk *mem_chunk);
  42 void            old_mem_chunk_reset     (GMemChunk *mem_chunk);
  43 void            old_mem_chunk_print     (GMemChunk *mem_chunk);
  44 void            old_mem_chunk_info      (void);
  45 #ifndef G_ALLOC_AND_FREE
  46 #define G_ALLOC_AND_FREE  2
  47 #endif
  48
  49 /* --- functions --- */
  50 static inline int
  51 corruption (void)
  52 {
  53   if (G_UNLIKELY (want_corruption))
  54     {
  55       /* corruption per call likelyness is about 1:4000000 */
  56       guint32 r = g_random_int() % 8000009;
  57       return r == 277 ? +1 : r == 281 ? -1 : 0;
  58     }
  59   return 0;
  60 }
  61
  62 static inline gpointer
  63 memchunk_alloc (GMemChunk **memchunkp,
  64                 guint       size)
  65 {
  66   size = MAX (size, 1);
  67   if (G_UNLIKELY (!*memchunkp))
  68     *memchunkp = old_mem_chunk_new ("", size, 4096, G_ALLOC_AND_FREE);
  69   return old_mem_chunk_alloc (*memchunkp);
  70 }
  71
  72 static inline void
  73 memchunk_free (GMemChunk *memchunk,
  74                gpointer   chunk)
  75 {
  76   old_mem_chunk_free (memchunk, chunk);
  77   if (clean_memchunks)
  78     old_mem_chunk_clean (memchunk);
  79 }
  80
  81 static gpointer
  82 test_memchunk_thread (gpointer data)
  83 {
  84   GMemChunk **memchunks;
  85   guint i, j;
  86   guint8 **ps;
  87   guint   *ss;
  88   guint32 rand_accu = 2147483563;
  89   /* initialize random numbers */
  90   if (data)
  91     rand_accu = *(guint32*) data;
  92   else
  93     {
  94       GTimeVal rand_tv;
  95       g_get_current_time (&rand_tv);
  96       rand_accu = rand_tv.tv_usec + (rand_tv.tv_sec << 16);
  97     }
  98
  99   /* prepare for memchunk creation */
 100   memchunks = g_alloca (sizeof (memchunks[0]) * prime_size);
 101   memset (memchunks, 0, sizeof (memchunks[0]) * prime_size);
 102
 103   ps = g_new (guint8*, number_of_blocks);
 104   ss = g_new (guint, number_of_blocks);
 105   /* create number_of_blocks random sizes */
 106   for (i = 0; i < number_of_blocks; i++)
 107     ss[i] = quick_rand32() % prime_size;
 108   /* allocate number_of_blocks blocks */
 109   for (i = 0; i < number_of_blocks; i++)
 110     ps[i] = memchunk_alloc (&memchunks[ss[i]], ss[i]);
 111   for (j = 0; j < number_of_repetitions; j++)
 112     {
 113       /* free number_of_blocks/2 blocks */
 114       for (i = 0; i < number_of_blocks; i += 2)
 115         memchunk_free (memchunks[ss[i]], ps[i]);
 116       /* allocate number_of_blocks/2 blocks with new sizes */
 117       for (i = 0; i < number_of_blocks; i += 2)
 118         {
 119           ss[i] = quick_rand32() % prime_size;
 120           ps[i] = memchunk_alloc (&memchunks[ss[i]], ss[i]);
 121         }
 122     }
 123   /* free number_of_blocks blocks */
 124   for (i = 0; i < number_of_blocks; i++)
 125     memchunk_free (memchunks[ss[i]], ps[i]);
 126   /* alloc and free many equally sized chunks in a row */
 127   for (i = 0; i < number_of_repetitions; i++)
 128     {
 129       guint sz = quick_rand32() % prime_size;
 130       guint k = number_of_blocks / 100;
 131       for (j = 0; j < k; j++)
 132         ps[j] = memchunk_alloc (&memchunks[sz], sz);
 133       for (j = 0; j < k; j++)
 134         memchunk_free (memchunks[sz], ps[j]);
 135     }
 136   /* cleanout memchunks */
 137   for (i = 0; i < prime_size; i++)
 138     if (memchunks[i])
 139       old_mem_chunk_destroy (memchunks[i]);
 140   g_free (ps);
 141   g_free (ss);
 142
 143   return NULL;
 144 }
 145
 146 static gpointer
 147 test_sliced_mem_thread (gpointer data)
 148 {
 149   guint32 rand_accu = 2147483563;
 150   guint i, j;
 151   guint8 **ps;
 152   guint   *ss;
 153
 154   /* initialize random numbers */
 155   if (data)
 156     rand_accu = *(guint32*) data;
 157   else
 158     {
 159       GTimeVal rand_tv;
 160       g_get_current_time (&rand_tv);
 161       rand_accu = rand_tv.tv_usec + (rand_tv.tv_sec << 16);
 162     }
 163
 164   ps = g_new (guint8*, number_of_blocks);
 165   ss = g_new (guint, number_of_blocks);
 166   /* create number_of_blocks random sizes */
 167   for (i = 0; i < number_of_blocks; i++)
 168     ss[i] = quick_rand32() % prime_size;
 169   /* allocate number_of_blocks blocks */
 170   for (i = 0; i < number_of_blocks; i++)
 171     ps[i] = g_slice_alloc (ss[i] + corruption());
 172   for (j = 0; j < number_of_repetitions; j++)
 173     {
 174       /* free number_of_blocks/2 blocks */
 175       for (i = 0; i < number_of_blocks; i += 2)
 176         g_slice_free1 (ss[i] + corruption(), ps[i] + corruption());
 177       /* allocate number_of_blocks/2 blocks with new sizes */
 178       for (i = 0; i < number_of_blocks; i += 2)
 179         {
 180           ss[i] = quick_rand32() % prime_size;
 181           ps[i] = g_slice_alloc (ss[i] + corruption());
 182         }
 183     }
 184   /* free number_of_blocks blocks */
 185   for (i = 0; i < number_of_blocks; i++)
 186     g_slice_free1 (ss[i] + corruption(), ps[i] + corruption());
 187   /* alloc and free many equally sized chunks in a row */
 188   for (i = 0; i < number_of_repetitions; i++)
 189     {
 190       guint sz = quick_rand32() % prime_size;
 191       guint k = number_of_blocks / 100;
 192       for (j = 0; j < k; j++)
 193         ps[j] = g_slice_alloc (sz + corruption());
 194       for (j = 0; j < k; j++)
 195         g_slice_free1 (sz + corruption(), ps[j] + corruption());
 196     }
 197   g_free (ps);
 198   g_free (ss);
 199
 200   return NULL;
 201 }
 202
 203 static void
 204 usage (void)
 205 {
 206   g_print ("Usage: slice-test [n_threads] [G|S|M|O][f][c][~] [maxblocksize] [seed]\n");
 207 }
 208
 209 int
 210 main (int   argc,
 211       char *argv[])
 212 {
 213   guint seed32, *seedp = NULL;
 214   gboolean ccounters = FALSE, use_memchunks = FALSE;
 215   guint n_threads = 1;
 216   const gchar *mode = "slab allocator + magazine cache", *emode = " ";
 217   if (argc > 1)
 218     n_threads = g_ascii_strtoull (argv[1], NULL, 10);
 219   if (argc > 2)
 220     {
 221       guint i, l = strlen (argv[2]);
 222       for (i = 0; i < l; i++)
 223         switch (argv[2][i])
 224           {
 225           case 'G': /* GLib mode */
 226             g_slice_set_config (G_SLICE_CONFIG_ALWAYS_MALLOC, FALSE);
 227             g_slice_set_config (G_SLICE_CONFIG_BYPASS_MAGAZINES, FALSE);
 228             mode = "slab allocator + magazine cache";
 229             break;
 230           case 'S': /* slab mode */
 231             g_slice_set_config (G_SLICE_CONFIG_ALWAYS_MALLOC, FALSE);
 232             g_slice_set_config (G_SLICE_CONFIG_BYPASS_MAGAZINES, TRUE);
 233             mode = "slab allocator";
 234             break;
 235           case 'M': /* malloc mode */
 236             g_slice_set_config (G_SLICE_CONFIG_ALWAYS_MALLOC, TRUE);
 237             mode = "system malloc";
 238             break;
 239           case 'O': /* old memchunks */
 240             use_memchunks = TRUE;
 241             mode = "old memchunks";
 242             break;
 243           case 'f': /* eager freeing */
 244             g_slice_set_config (G_SLICE_CONFIG_WORKING_SET_MSECS, 0);
 245             clean_memchunks = TRUE;
 246             emode = " with eager freeing";
 247             break;
 248           case 'c': /* print contention counters */
 249             ccounters = TRUE;
 250             break;
 251           case '~':
 252             want_corruption = TRUE; /* force occasional corruption */
 253             break;
 254           default:
 255             usage();
 256             return 1;
 257           }
 258     }
 259   if (argc > 3)
 260     prime_size = g_ascii_strtoull (argv[3], NULL, 10);
 261   if (argc > 4)
 262     {
 263       seed32 = g_ascii_strtoull (argv[4], NULL, 10);
 264       seedp = &seed32;
 265     }
 266
 267   g_thread_init (NULL);
 268
 269   if (argc <= 1)
 270     usage();
 271
 272   {
 273     gchar strseed[64] = "<random>";
 274     GThread **threads;
 275     guint i;
 276
 277     if (seedp)
 278       g_snprintf (strseed, 64, "%u", *seedp);
 279     g_print ("Starting %d threads allocating random blocks <= %u bytes with seed=%s using %s%s\n", n_threads, prime_size, strseed, mode, emode);
 280
 281     threads = g_alloca (sizeof(GThread*) * n_threads);
 282     if (!use_memchunks)
 283       for (i = 0; i < n_threads; i++)
 284         threads[i] = g_thread_create (test_sliced_mem_thread, seedp, TRUE, NULL);
 285     else
 286       {
 287         for (i = 0; i < n_threads; i++)
 288           threads[i] = g_thread_create (test_memchunk_thread, seedp, TRUE, NULL);
 289       }
 290     for (i = 0; i < n_threads; i++)
 291       g_thread_join (threads[i]);
 292
 293     if (ccounters)
 294       {
 295         guint n, n_chunks = g_slice_get_config (G_SLICE_CONFIG_CHUNK_SIZES);
 296         g_print ("    ChunkSize | MagazineSize | Contention\n");
 297         for (i = 0; i < n_chunks; i++)
 298           {
 299             gint64 *vals = g_slice_get_config_state (G_SLICE_CONFIG_CONTENTION_COUNTER, i, &n);
 300             g_print ("  %9" G_GINT64_FORMAT "   |  %9" G_GINT64_FORMAT "   |  %9" G_GINT64_FORMAT "\n", vals[0], vals[2], vals[1]);
 301             g_free (vals);
 302           }
 303       }
 304     else
 305       g_print ("Done.\n");
 306     return 0;
 307   }
 308 }