b52799132a2952e3ea1299fa75d0f183750ea3c4
[exim.git] / src / src / store.c
1 /*************************************************
2 *     Exim - an Internet mail transport agent    *
3 *************************************************/
4
5 /* Copyright (c) University of Cambridge 1995 - 2018 */
6 /* See the file NOTICE for conditions of use and distribution. */
7
8 /* Exim gets and frees all its store through these functions. In the original
9 implementation there was a lot of mallocing and freeing of small bits of store.
10 The philosophy has now changed to a scheme which includes the concept of
11 "stacking pools" of store. For the short-lived processes, there isn't any real
12 need to do any garbage collection, but the stack concept allows quick resetting
13 in places where this seems sensible.
14
15 Obviously the long-running processes (the daemon, the queue runner, and eximon)
16 must take care not to eat store.
17
18 The following different types of store are recognized:
19
20 . Long-lived, large blocks: This is implemented by retaining the original
21   malloc/free functions, and it used for permanent working buffers and for
22   getting blocks to cut up for the other types.
23
24 . Long-lived, small blocks: This is used for blocks that have to survive until
25   the process exits. It is implemented as a stacking pool (POOL_PERM). This is
26   functionally the same as store_malloc(), except that the store can't be
27   freed, but I expect it to be more efficient for handling small blocks.
28
29 . Short-lived, short blocks: Most of the dynamic store falls into this
30   category. It is implemented as a stacking pool (POOL_MAIN) which is reset
31   after accepting a message when multiple messages are received by a single
32   process. Resetting happens at some other times as well, usually fairly
33   locally after some specific processing that needs working store.
34
35 . There is a separate pool (POOL_SEARCH) that is used only for lookup storage.
36   This means it can be freed when search_tidyup() is called to close down all
37   the lookup caching.
38 */
39
40
41 #include "exim.h"
42 /* keep config.h before memcheck.h, for NVALGRIND */
43 #include "config.h"
44
45 #include "memcheck.h"
46
47
48 /* We need to know how to align blocks of data for general use. I'm not sure
49 how to get an alignment factor in general. In the current world, a value of 8
50 is probably right, and this is sizeof(double) on some systems and sizeof(void
51 *) on others, so take the larger of those. Since everything in this expression
52 is a constant, the compiler should optimize it to a simple constant wherever it
53 appears (I checked that gcc does do this). */
54
55 #define alignment \
56   ((sizeof(void *) > sizeof(double))? sizeof(void *) : sizeof(double))
57
58 /* Size of block to get from malloc to carve up into smaller ones. This
59 must be a multiple of the alignment. We assume that 8192 is going to be
60 suitably aligned. */
61
62 #define STORE_BLOCK_SIZE 8192
63
64 /* store_reset() will not free the following block if the last used block has
65 less than this much left in it. */
66
67 #define STOREPOOL_MIN_SIZE 256
68
69 /* Structure describing the beginning of each big block. */
70
71 typedef struct storeblock {
72   struct storeblock *next;
73   size_t length;
74 } storeblock;
75
76 /* Just in case we find ourselves on a system where the structure above has a
77 length that is not a multiple of the alignment, set up a macro for the padded
78 length. */
79
80 #define ALIGNED_SIZEOF_STOREBLOCK \
81   (((sizeof(storeblock) + alignment - 1) / alignment) * alignment)
82
83 /* Variables holding data for the local pools of store. The current pool number
84 is held in store_pool, which is global so that it can be changed from outside.
85 Setting the initial length values to -1 forces a malloc for the first call,
86 even if the length is zero (which is used for getting a point to reset to). */
87
88 int store_pool = POOL_PERM;
89
90 static storeblock *chainbase[3] = { NULL, NULL, NULL };
91 static storeblock *current_block[3] = { NULL, NULL, NULL };
92 static void *next_yield[3] = { NULL, NULL, NULL };
93 static int yield_length[3] = { -1, -1, -1 };
94
95 /* pool_malloc holds the amount of memory used by the store pools; this goes up
96 and down as store is reset or released. nonpool_malloc is the total got by
97 malloc from other calls; this doesn't go down because it is just freed by
98 pointer. */
99
100 static int pool_malloc = 0;
101 static int nonpool_malloc = 0;
102
103 /* This variable is set by store_get() to its yield, and by store_reset() to
104 NULL. This enables string_cat() to optimize its store handling for very long
105 strings. That's why the variable is global. */
106
107 void *store_last_get[3] = { NULL, NULL, NULL };
108
109
110
111 /*************************************************
112 *       Get a block from the current pool        *
113 *************************************************/
114
115 /* Running out of store is a total disaster. This function is called via the
116 macro store_get(). It passes back a block of store within the current big
117 block, getting a new one if necessary. The address is saved in
118 store_last_was_get.
119
120 Arguments:
121   size        amount wanted
122   filename    source file from which called
123   linenumber  line number in source file.
124
125 Returns:      pointer to store (panic on malloc failure)
126 */
127
128 void *
129 store_get_3(int size, const char *filename, int linenumber)
130 {
131 /* Round up the size to a multiple of the alignment. Although this looks a
132 messy statement, because "alignment" is a constant expression, the compiler can
133 do a reasonable job of optimizing, especially if the value of "alignment" is a
134 power of two. I checked this with -O2, and gcc did very well, compiling it to 4
135 instructions on a Sparc (alignment = 8). */
136
137 if (size % alignment != 0) size += alignment - (size % alignment);
138
139 /* If there isn't room in the current block, get a new one. The minimum
140 size is STORE_BLOCK_SIZE, and we would expect this to be the norm, since
141 these functions are mostly called for small amounts of store. */
142
143 if (size > yield_length[store_pool])
144   {
145   int length = (size <= STORE_BLOCK_SIZE)? STORE_BLOCK_SIZE : size;
146   int mlength = length + ALIGNED_SIZEOF_STOREBLOCK;
147   storeblock * newblock = NULL;
148
149   /* Sometimes store_reset() may leave a block for us; check if we can use it */
150
151   if (  (newblock = current_block[store_pool])
152      && (newblock = newblock->next)
153      && newblock->length < length
154      )
155     {
156     /* Give up on this block, because it's too small */
157     store_free(newblock);
158     newblock = NULL;
159     }
160
161   /* If there was no free block, get a new one */
162
163   if (!newblock)
164     {
165     pool_malloc += mlength;           /* Used in pools */
166     nonpool_malloc -= mlength;        /* Exclude from overall total */
167     newblock = store_malloc(mlength);
168     newblock->next = NULL;
169     newblock->length = length;
170     if (!chainbase[store_pool])
171       chainbase[store_pool] = newblock;
172     else
173       current_block[store_pool]->next = newblock;
174     }
175
176   current_block[store_pool] = newblock;
177   yield_length[store_pool] = newblock->length;
178   next_yield[store_pool] =
179     (void *)(CS current_block[store_pool] + ALIGNED_SIZEOF_STOREBLOCK);
180   (void) VALGRIND_MAKE_MEM_NOACCESS(next_yield[store_pool], yield_length[store_pool]);
181   }
182
183 /* There's (now) enough room in the current block; the yield is the next
184 pointer. */
185
186 store_last_get[store_pool] = next_yield[store_pool];
187
188 /* Cut out the debugging stuff for utilities, but stop picky compilers from
189 giving warnings. */
190
191 #ifdef COMPILE_UTILITY
192 filename = filename;
193 linenumber = linenumber;
194 #else
195 DEBUG(D_memory)
196   {
197   if (f.running_in_test_harness)
198     debug_printf("---%d Get %5d\n", store_pool, size);
199   else
200     debug_printf("---%d Get %6p %5d %-14s %4d\n", store_pool,
201       store_last_get[store_pool], size, filename, linenumber);
202   }
203 #endif  /* COMPILE_UTILITY */
204
205 (void) VALGRIND_MAKE_MEM_UNDEFINED(store_last_get[store_pool], size);
206 /* Update next pointer and number of bytes left in the current block. */
207
208 next_yield[store_pool] = (void *)(CS next_yield[store_pool] + size);
209 yield_length[store_pool] -= size;
210
211 return store_last_get[store_pool];
212 }
213
214
215
216 /*************************************************
217 *       Get a block from the PERM pool           *
218 *************************************************/
219
220 /* This is just a convenience function, useful when just a single block is to
221 be obtained.
222
223 Arguments:
224   size        amount wanted
225   filename    source file from which called
226   linenumber  line number in source file.
227
228 Returns:      pointer to store (panic on malloc failure)
229 */
230
231 void *
232 store_get_perm_3(int size, const char *filename, int linenumber)
233 {
234 void *yield;
235 int old_pool = store_pool;
236 store_pool = POOL_PERM;
237 yield = store_get_3(size, filename, linenumber);
238 store_pool = old_pool;
239 return yield;
240 }
241
242
243
244 /*************************************************
245 *      Extend a block if it is at the top        *
246 *************************************************/
247
248 /* While reading strings of unknown length, it is often the case that the
249 string is being read into the block at the top of the stack. If it needs to be
250 extended, it is more efficient just to extend the top block rather than
251 allocate a new block and then have to copy the data. This function is provided
252 for the use of string_cat(), but of course can be used elsewhere too.
253
254 Arguments:
255   ptr        pointer to store block
256   oldsize    current size of the block, as requested by user
257   newsize    new size required
258   filename   source file from which called
259   linenumber line number in source file
260
261 Returns:     TRUE if the block is at the top of the stack and has been
262              extended; FALSE if it isn't at the top of the stack, or cannot
263              be extended
264 */
265
266 BOOL
267 store_extend_3(void *ptr, int oldsize, int newsize, const char *filename,
268   int linenumber)
269 {
270 int inc = newsize - oldsize;
271 int rounded_oldsize = oldsize;
272
273 if (rounded_oldsize % alignment != 0)
274   rounded_oldsize += alignment - (rounded_oldsize % alignment);
275
276 if (CS ptr + rounded_oldsize != CS (next_yield[store_pool]) ||
277     inc > yield_length[store_pool] + rounded_oldsize - oldsize)
278   return FALSE;
279
280 /* Cut out the debugging stuff for utilities, but stop picky compilers from
281 giving warnings. */
282
283 #ifdef COMPILE_UTILITY
284 filename = filename;
285 linenumber = linenumber;
286 #else
287 DEBUG(D_memory)
288   {
289   if (f.running_in_test_harness)
290     debug_printf("---%d Ext %5d\n", store_pool, newsize);
291   else
292     debug_printf("---%d Ext %6p %5d %-14s %4d\n", store_pool, ptr, newsize,
293       filename, linenumber);
294   }
295 #endif  /* COMPILE_UTILITY */
296
297 if (newsize % alignment != 0) newsize += alignment - (newsize % alignment);
298 next_yield[store_pool] = CS ptr + newsize;
299 yield_length[store_pool] -= newsize - rounded_oldsize;
300 (void) VALGRIND_MAKE_MEM_UNDEFINED(ptr + oldsize, inc);
301 return TRUE;
302 }
303
304
305
306
307 /*************************************************
308 *    Back up to a previous point on the stack    *
309 *************************************************/
310
311 /* This function resets the next pointer, freeing any subsequent whole blocks
312 that are now unused. Normally it is given a pointer that was the yield of a
313 call to store_get, and is therefore aligned, but it may be given an offset
314 after such a pointer in order to release the end of a block and anything that
315 follows.
316
317 Arguments:
318   ptr         place to back up to
319   filename    source file from which called
320   linenumber  line number in source file
321
322 Returns:      nothing
323 */
324
325 void
326 store_reset_3(void *ptr, const char *filename, int linenumber)
327 {
328 storeblock * bb;
329 storeblock * b = current_block[store_pool];
330 char * bc = CS b + ALIGNED_SIZEOF_STOREBLOCK;
331 int newlength;
332
333 /* Last store operation was not a get */
334
335 store_last_get[store_pool] = NULL;
336
337 /* See if the place is in the current block - as it often will be. Otherwise,
338 search for the block in which it lies. */
339
340 if (CS ptr < bc || CS ptr > bc + b->length)
341   {
342   for (b = chainbase[store_pool]; b; b = b->next)
343     {
344     bc = CS b + ALIGNED_SIZEOF_STOREBLOCK;
345     if (CS ptr >= bc && CS ptr <= bc + b->length) break;
346     }
347   if (!b)
348     log_write(0, LOG_MAIN|LOG_PANIC_DIE, "internal error: store_reset(%p) "
349       "failed: pool=%d %-14s %4d", ptr, store_pool, filename, linenumber);
350   }
351
352 /* Back up, rounding to the alignment if necessary. When testing, flatten
353 the released memory. */
354
355 newlength = bc + b->length - CS ptr;
356 #ifndef COMPILE_UTILITY
357 if (debug_store)
358   {
359   assert_no_variables(ptr, newlength, filename, linenumber);
360   if (f.running_in_test_harness)
361     {
362     (void) VALGRIND_MAKE_MEM_DEFINED(ptr, newlength);
363     memset(ptr, 0xF0, newlength);
364     }
365   }
366 #endif
367 (void) VALGRIND_MAKE_MEM_NOACCESS(ptr, newlength);
368 yield_length[store_pool] = newlength - (newlength % alignment);
369 next_yield[store_pool] = CS ptr + (newlength % alignment);
370 current_block[store_pool] = b;
371
372 /* Free any subsequent block. Do NOT free the first successor, if our
373 current block has less than 256 bytes left. This should prevent us from
374 flapping memory. However, keep this block only when it has the default size. */
375
376 if (yield_length[store_pool] < STOREPOOL_MIN_SIZE &&
377     b->next &&
378     b->next->length == STORE_BLOCK_SIZE)
379   {
380   b = b->next;
381 #ifndef COMPILE_UTILITY
382   if (debug_store)
383     assert_no_variables(b, b->length + ALIGNED_SIZEOF_STOREBLOCK,
384                         filename, linenumber);
385 #endif
386   (void) VALGRIND_MAKE_MEM_NOACCESS(CS b + ALIGNED_SIZEOF_STOREBLOCK,
387                 b->length - ALIGNED_SIZEOF_STOREBLOCK);
388   }
389
390 bb = b->next;
391 b->next = NULL;
392
393 while ((b = bb))
394   {
395 #ifndef COMPILE_UTILITY
396   if (debug_store)
397     assert_no_variables(b, b->length + ALIGNED_SIZEOF_STOREBLOCK,
398                         filename, linenumber);
399 #endif
400   bb = bb->next;
401   pool_malloc -= b->length + ALIGNED_SIZEOF_STOREBLOCK;
402   store_free_3(b, filename, linenumber);
403   }
404
405 /* Cut out the debugging stuff for utilities, but stop picky compilers from
406 giving warnings. */
407
408 #ifdef COMPILE_UTILITY
409 filename = filename;
410 linenumber = linenumber;
411 #else
412 DEBUG(D_memory)
413   {
414   if (f.running_in_test_harness)
415     debug_printf("---%d Rst    ** %d\n", store_pool, pool_malloc);
416   else
417     debug_printf("---%d Rst %6p    ** %-14s %4d %d\n", store_pool, ptr,
418       filename, linenumber, pool_malloc);
419   }
420 #endif  /* COMPILE_UTILITY */
421 }
422
423
424
425
426
427 /************************************************
428 *             Release store                     *
429 ************************************************/
430
431 /* This function checks that the pointer it is given is the first thing in a
432 block, and if so, releases that block.
433
434 Arguments:
435   block       block of store to consider
436   filename    source file from which called
437   linenumber  line number in source file
438
439 Returns:      nothing
440 */
441
442 static void
443 store_release_3(void * block, const char * filename, int linenumber)
444 {
445 storeblock * b;
446
447 /* It will never be the first block, so no need to check that. */
448
449 for (b = chainbase[store_pool]; b; b = b->next)
450   {
451   storeblock * bb = b->next;
452   if (bb && CS block == CS bb + ALIGNED_SIZEOF_STOREBLOCK)
453     {
454     b->next = bb->next;
455     pool_malloc -= bb->length + ALIGNED_SIZEOF_STOREBLOCK;
456
457     /* Cut out the debugging stuff for utilities, but stop picky compilers
458     from giving warnings. */
459
460 #ifdef COMPILE_UTILITY
461     filename = filename;
462     linenumber = linenumber;
463 #else
464     DEBUG(D_memory)
465       if (f.running_in_test_harness)
466         debug_printf("-Release       %d\n", pool_malloc);
467       else
468         debug_printf("-Release %6p %-20s %4d %d\n", (void *)bb, filename,
469           linenumber, pool_malloc);
470
471     if (f.running_in_test_harness)
472       memset(bb, 0xF0, bb->length+ALIGNED_SIZEOF_STOREBLOCK);
473 #endif  /* COMPILE_UTILITY */
474
475     free(bb);
476     return;
477     }
478   }
479 }
480
481
482 /************************************************
483 *             Move store                        *
484 ************************************************/
485
486 /* Allocate a new block big enough to expend to the given size and
487 copy the current data into it.  Free the old one if possible.
488
489 This function is specifically provided for use when reading very
490 long strings, e.g. header lines. When the string gets longer than a
491 complete block, it gets copied to a new block. It is helpful to free
492 the old block iff the previous copy of the string is at its start,
493 and therefore the only thing in it. Otherwise, for very long strings,
494 dead store can pile up somewhat disastrously. This function checks that
495 the pointer it is given is the first thing in a block, and that nothing
496 has been allocated since. If so, releases that block.
497
498 Arguments:
499   block
500   newsize
501   len
502
503 Returns:        new location of data
504 */
505
506 void *
507 store_newblock_3(void * block, int newsize, int len,
508   const char * filename, int linenumber)
509 {
510 BOOL release_ok = store_last_get[store_pool] == block;
511 uschar * newtext = store_get(newsize);
512
513 memcpy(newtext, block, len);
514 if (release_ok) store_release_3(block, filename, linenumber);
515 return (void *)newtext;
516 }
517
518
519
520
521 /*************************************************
522 *                Malloc store                    *
523 *************************************************/
524
525 /* Running out of store is a total disaster for exim. Some malloc functions
526 do not run happily on very small sizes, nor do they document this fact. This
527 function is called via the macro store_malloc().
528
529 Arguments:
530   size        amount of store wanted
531   filename    source file from which called
532   linenumber  line number in source file
533
534 Returns:      pointer to gotten store (panic on failure)
535 */
536
537 void *
538 store_malloc_3(int size, const char *filename, int linenumber)
539 {
540 void *yield;
541
542 if (size < 16) size = 16;
543
544 if (!(yield = malloc((size_t)size)))
545   log_write(0, LOG_MAIN|LOG_PANIC_DIE, "failed to malloc %d bytes of memory: "
546     "called from line %d of %s", size, linenumber, filename);
547
548 nonpool_malloc += size;
549
550 /* Cut out the debugging stuff for utilities, but stop picky compilers from
551 giving warnings. */
552
553 #ifdef COMPILE_UTILITY
554 filename = filename;
555 linenumber = linenumber;
556 #else
557
558 /* If running in test harness, spend time making sure all the new store
559 is not filled with zeros so as to catch problems. */
560
561 if (f.running_in_test_harness)
562   {
563   memset(yield, 0xF0, (size_t)size);
564   DEBUG(D_memory) debug_printf("--Malloc %5d %d %d\n", size, pool_malloc,
565     nonpool_malloc);
566   }
567 else
568   {
569   DEBUG(D_memory) debug_printf("--Malloc %6p %5d %-14s %4d %d %d\n", yield,
570     size, filename, linenumber, pool_malloc, nonpool_malloc);
571   }
572 #endif  /* COMPILE_UTILITY */
573
574 return yield;
575 }
576
577
578 /************************************************
579 *             Free store                        *
580 ************************************************/
581
582 /* This function is called by the macro store_free().
583
584 Arguments:
585   block       block of store to free
586   filename    source file from which called
587   linenumber  line number in source file
588
589 Returns:      nothing
590 */
591
592 void
593 store_free_3(void *block, const char *filename, int linenumber)
594 {
595 #ifdef COMPILE_UTILITY
596 filename = filename;
597 linenumber = linenumber;
598 #else
599 DEBUG(D_memory)
600   {
601   if (f.running_in_test_harness)
602     debug_printf("----Free\n");
603   else
604     debug_printf("----Free %6p %-20s %4d\n", block, filename, linenumber);
605   }
606 #endif  /* COMPILE_UTILITY */
607 free(block);
608 }
609
610 /* End of store.c */