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 #include <string.h> 21 22 #define ALIGN(size, base) ((base) * (gsize) (((size) + (base) - 1) / (base))) 23 24 static gdouble parse_memsize (const gchar *cstring); 25 static void usage (void); 26 27 static void 28 fill_memory (guint **mem, 29 guint n, 30 guint val) 31 { 32 guint j, o = 0; 33 for (j = 0; j < n; j++) 34 mem[j][o] = val; 35 } 36 37 static guint64 38 access_memory3 (guint **mema, 39 guint **memb, 40 guint **memd, 41 guint n, 42 guint64 repeats) 43 { 44 guint64 accu = 0, i, j; 45 const guint o = 0; 46 for (i = 0; i < repeats; i++) 47 { 48 for (j = 1; j < n; j += 2) 49 memd[j][o] = mema[j][o] + memb[j][o]; 50 } 51 for (i = 0; i < repeats; i++) 52 for (j = 0; j < n; j++) 53 accu += memd[j][o]; 54 return accu; 55 } 56 57 static void 58 touch_mem (guint64 block_size, 59 guint64 n_blocks, 60 guint64 repeats) 61 { 62 guint64 j, accu, n = n_blocks; 63 GTimer *timer; 64 guint **memc; 65 guint **memb; 66 guint **mema = g_new (guint*, n); 67 for (j = 0; j < n; j++) 68 mema[j] = g_slice_alloc (block_size); 69 memb = g_new (guint*, n); 70 for (j = 0; j < n; j++) 71 memb[j] = g_slice_alloc (block_size); 72 memc = g_new (guint*, n); 73 for (j = 0; j < n; j++) 74 memc[j] = g_slice_alloc (block_size); 75 76 timer = g_timer_new(); 77 fill_memory (mema, n, 2); 78 fill_memory (memb, n, 3); 79 fill_memory (memc, n, 4); 80 access_memory3 (mema, memb, memc, n, 3); 81 g_timer_start (timer); 82 accu = access_memory3 (mema, memb, memc, n, repeats); 83 g_timer_stop (timer); 84 85 g_print ("Access-time = %fs\n", g_timer_elapsed (timer, NULL)); 86 g_assert (accu / repeats == (2 + 3) * n / 2 + 4 * n / 2); 87 88 for (j = 0; j < n; j++) 89 { 90 g_slice_free1 (block_size, mema[j]); 91 g_slice_free1 (block_size, memb[j]); 92 g_slice_free1 (block_size, memc[j]); 93 } 94 g_timer_destroy (timer); 95 g_free (mema); 96 g_free (memb); 97 g_free (memc); 98 } 99 100 static void 101 usage (void) 102 { 103 g_print ("Usage: slice-color <block-size> [memory-size] [repeats] [colorization]\n"); 104 } 105 106 int 107 main (int argc, 108 char *argv[]) 109 { 110 guint64 block_size = 512, area_size = 1024 * 1024, n_blocks, repeats = 1000000; 111 112 if (argc > 1) 113 block_size = parse_memsize (argv[1]); 114 else 115 { 116 usage(); 117 block_size = 512; 118 } 119 if (argc > 2) 120 area_size = parse_memsize (argv[2]); 121 if (argc > 3) 122 repeats = parse_memsize (argv[3]); 123 if (argc > 4) 124 g_slice_set_config (G_SLICE_CONFIG_COLOR_INCREMENT, parse_memsize (argv[4])); 125 126 /* figure number of blocks from block and area size. 127 * divide area by 3 because touch_mem() allocates 3 areas 128 */ 129 n_blocks = area_size / 3 / ALIGN (block_size, sizeof (gsize) * 2); 130 131 /* basic sanity checks */ 132 if (!block_size || !n_blocks || block_size >= area_size) 133 { 134 g_printerr ("Invalid arguments: block-size=%llu memory-size=%llu\n", block_size, area_size); 135 usage(); 136 return 1; 137 } 138 139 g_printerr ("Will allocate and touch %llu blocks of %llu bytes (= %llu bytes) %llu times with color increment: 0x%08llx\n", 140 n_blocks, block_size, n_blocks * block_size, repeats, g_slice_get_config (G_SLICE_CONFIG_COLOR_INCREMENT)); 141 142 touch_mem (block_size, n_blocks, repeats); 143 144 return 0; 145 } 146 147 static gdouble 148 parse_memsize (const gchar *cstring) 149 { 150 gchar *mem = g_strdup (cstring); 151 gchar *string = g_strstrip (mem); 152 guint l = strlen (string); 153 gdouble f = 0; 154 gchar *derr = NULL; 155 gdouble msize; 156 157 switch (l ? string[l - 1] : 0) 158 { 159 case 'k': f = 1000; break; 160 case 'K': f = 1024; break; 161 case 'm': f = 1000000; break; 162 case 'M': f = 1024 * 1024; break; 163 case 'g': f = 1000000000; break; 164 case 'G': f = 1024 * 1024 * 1024; break; 165 } 166 if (f) 167 string[l - 1] = 0; 168 msize = g_ascii_strtod (string, &derr); 169 g_free (mem); 170 if (derr && *derr) 171 { 172 g_printerr ("failed to parse number at: %s\n", derr); 173 msize = 0; 174 } 175 if (f) 176 msize *= f; 177 return msize; 178 } 179