SECURITY: Refuse negative and large store allocations
[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 /* Copyright (c) The Exim maintainers 2019 - 2020 */
7 /* See the file NOTICE for conditions of use and distribution. */
8
9 /* Exim gets and frees all its store through these functions. In the original
10 implementation there was a lot of mallocing and freeing of small bits of store.
11 The philosophy has now changed to a scheme which includes the concept of
12 "stacking pools" of store. For the short-lived processes, there isn't any real
13 need to do any garbage collection, but the stack concept allows quick resetting
14 in places where this seems sensible.
15
16 Obviously the long-running processes (the daemon, the queue runner, and eximon)
17 must take care not to eat store.
18
19 The following different types of store are recognized:
20
21 . Long-lived, large blocks: This is implemented by retaining the original
22   malloc/free functions, and it used for permanent working buffers and for
23   getting blocks to cut up for the other types.
24
25 . Long-lived, small blocks: This is used for blocks that have to survive until
26   the process exits. It is implemented as a stacking pool (POOL_PERM). This is
27   functionally the same as store_malloc(), except that the store can't be
28   freed, but I expect it to be more efficient for handling small blocks.
29
30 . Short-lived, short blocks: Most of the dynamic store falls into this
31   category. It is implemented as a stacking pool (POOL_MAIN) which is reset
32   after accepting a message when multiple messages are received by a single
33   process. Resetting happens at some other times as well, usually fairly
34   locally after some specific processing that needs working store.
35
36 . There is a separate pool (POOL_SEARCH) that is used only for lookup storage.
37   This means it can be freed when search_tidyup() is called to close down all
38   the lookup caching.
39
40 . Orthogonal to the three pool types, there are two classes of memory: untainted
41   and tainted.  The latter is used for values derived from untrusted input, and
42   the string-expansion mechanism refuses to operate on such values (obviously,
43   it can expand an untainted value to return a tainted result).  The classes
44   are implemented by duplicating the three pool types.  Pool resets are requested
45   against the nontainted sibling and apply to both siblings.
46
47   Only memory blocks requested for tainted use are regarded as tainted; anything
48   else (including stack auto variables) is untainted.  Care is needed when coding
49   to not copy untrusted data into untainted memory, as downstream taint-checks
50   would be avoided.
51
52   Intermediate layers (eg. the string functions) can test for taint, and use this
53   for ensurinng that results have proper state.  For example the
54   string_vformat_trc() routing supporting the string_sprintf() interface will
55   recopy a string being built into a tainted allocation if it meets a %s for a
56   tainted argument.  Any intermediate-layer function that (can) return a new
57   allocation should behave this way; returning a tainted result if any tainted
58   content is used.  Intermediate-layer functions (eg. Ustrncpy) that modify
59   existing allocations fail if tainted data is written into an untainted area.
60   Users of functions that modify existing allocations should check if a tainted
61   source and an untainted destination is used, and fail instead (sprintf() being
62   the classic case).
63 */
64
65
66 #include "exim.h"
67 /* keep config.h before memcheck.h, for NVALGRIND */
68 #include "config.h"
69
70 #include <sys/mman.h>
71 #include "memcheck.h"
72
73
74 /* We need to know how to align blocks of data for general use. I'm not sure
75 how to get an alignment factor in general. In the current world, a value of 8
76 is probably right, and this is sizeof(double) on some systems and sizeof(void
77 *) on others, so take the larger of those. Since everything in this expression
78 is a constant, the compiler should optimize it to a simple constant wherever it
79 appears (I checked that gcc does do this). */
80
81 #define alignment \
82   (sizeof(void *) > sizeof(double) ? sizeof(void *) : sizeof(double))
83
84 /* store_reset() will not free the following block if the last used block has
85 less than this much left in it. */
86
87 #define STOREPOOL_MIN_SIZE 256
88
89 /* Structure describing the beginning of each big block. */
90
91 typedef struct storeblock {
92   struct storeblock *next;
93   size_t length;
94 } storeblock;
95
96 /* Just in case we find ourselves on a system where the structure above has a
97 length that is not a multiple of the alignment, set up a macro for the padded
98 length. */
99
100 #define ALIGNED_SIZEOF_STOREBLOCK \
101   (((sizeof(storeblock) + alignment - 1) / alignment) * alignment)
102
103 /* Size of block to get from malloc to carve up into smaller ones. This
104 must be a multiple of the alignment. We assume that 8192 is going to be
105 suitably aligned. */
106
107 #define STORE_BLOCK_SIZE (8192 - ALIGNED_SIZEOF_STOREBLOCK)
108
109 /* Variables holding data for the local pools of store. The current pool number
110 is held in store_pool, which is global so that it can be changed from outside.
111 Setting the initial length values to -1 forces a malloc for the first call,
112 even if the length is zero (which is used for getting a point to reset to). */
113
114 int store_pool = POOL_MAIN;
115
116 #define NPOOLS 6
117 static storeblock *chainbase[NPOOLS];
118 static storeblock *current_block[NPOOLS];
119 static void *next_yield[NPOOLS];
120 static int yield_length[NPOOLS] = { -1, -1, -1,  -1, -1, -1 };
121
122 /* pool_malloc holds the amount of memory used by the store pools; this goes up
123 and down as store is reset or released. nonpool_malloc is the total got by
124 malloc from other calls; this doesn't go down because it is just freed by
125 pointer. */
126
127 static int pool_malloc;
128 static int nonpool_malloc;
129
130 /* This variable is set by store_get() to its yield, and by store_reset() to
131 NULL. This enables string_cat() to optimize its store handling for very long
132 strings. That's why the variable is global. */
133
134 void *store_last_get[NPOOLS];
135
136 /* These are purely for stats-gathering */
137
138 static int nbytes[NPOOLS];      /* current bytes allocated */
139 static int maxbytes[NPOOLS];    /* max number reached */
140 static int nblocks[NPOOLS];     /* current number of blocks allocated */
141 static int maxblocks[NPOOLS];
142 static int n_nonpool_blocks;    /* current number of direct store_malloc() blocks */
143 static int max_nonpool_blocks;
144 static int max_pool_malloc;     /* max value for pool_malloc */
145 static int max_nonpool_malloc;  /* max value for nonpool_malloc */
146
147
148 #ifndef COMPILE_UTILITY
149 static const uschar * pooluse[NPOOLS] = {
150 [POOL_MAIN] =           US"main",
151 [POOL_PERM] =           US"perm",
152 [POOL_SEARCH] =         US"search",
153 [POOL_TAINT_MAIN] =     US"main",
154 [POOL_TAINT_PERM] =     US"perm",
155 [POOL_TAINT_SEARCH] =   US"search",
156 };
157 static const uschar * poolclass[NPOOLS] = {
158 [POOL_MAIN] =           US"untainted",
159 [POOL_PERM] =           US"untainted",
160 [POOL_SEARCH] =         US"untainted",
161 [POOL_TAINT_MAIN] =     US"tainted",
162 [POOL_TAINT_PERM] =     US"tainted",
163 [POOL_TAINT_SEARCH] =   US"tainted",
164 };
165 #endif
166
167
168 static void * internal_store_malloc(int, const char *, int);
169 static void   internal_store_free(void *, const char *, int linenumber);
170
171 /******************************************************************************/
172
173 /* Test if a pointer refers to tainted memory.
174
175 Slower version check, for use when platform intermixes malloc and mmap area
176 addresses. Test against the current-block of all tainted pools first, then all
177 blocks of all tainted pools.
178
179 Return: TRUE iff tainted
180 */
181
182 BOOL
183 is_tainted_fn(const void * p)
184 {
185 storeblock * b;
186
187 for (int pool = POOL_TAINT_BASE; pool < nelem(chainbase); pool++)
188   if ((b = current_block[pool]))
189     {
190     uschar * bc = US b + ALIGNED_SIZEOF_STOREBLOCK;
191     if (US p >= bc && US p < bc + b->length) return TRUE;
192     }
193
194 for (int pool = POOL_TAINT_BASE; pool < nelem(chainbase); pool++)
195   for (b = chainbase[pool]; b; b = b->next)
196     {
197     uschar * bc = US b + ALIGNED_SIZEOF_STOREBLOCK;
198     if (US p >= bc && US p < bc + b->length) return TRUE;
199     }
200 return FALSE;
201 }
202
203
204 void
205 die_tainted(const uschar * msg, const uschar * func, int line)
206 {
207 log_write(0, LOG_MAIN|LOG_PANIC_DIE, "Taint mismatch, %s: %s %d\n",
208         msg, func, line);
209 }
210
211
212
213 /*************************************************
214 *       Get a block from the current pool        *
215 *************************************************/
216
217 /* Running out of store is a total disaster. This function is called via the
218 macro store_get(). It passes back a block of store within the current big
219 block, getting a new one if necessary. The address is saved in
220 store_last_was_get.
221
222 Arguments:
223   size        amount wanted, bytes
224   tainted     class: set to true for untrusted data (eg. from smtp input)
225   func        function 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_3(int size, BOOL tainted, const char *func, int linenumber)
233 {
234 int pool = tainted ? store_pool + POOL_TAINT_BASE : store_pool;
235
236 /* Ensure we've been asked to allocate memory.
237 A negative size is a sign of a security problem.
238 A zero size might be also suspect, but our internal usage deliberately
239 does this to return a current watermark value for a later release of
240 allocated store. */
241
242 if (size < 0 || size >= INT_MAX/2)
243   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
244             "bad memory allocation requested (%d bytes) at %s %d",
245             size, func, linenumber);
246
247 /* Round up the size to a multiple of the alignment. Although this looks a
248 messy statement, because "alignment" is a constant expression, the compiler can
249 do a reasonable job of optimizing, especially if the value of "alignment" is a
250 power of two. I checked this with -O2, and gcc did very well, compiling it to 4
251 instructions on a Sparc (alignment = 8). */
252
253 if (size % alignment != 0) size += alignment - (size % alignment);
254
255 /* If there isn't room in the current block, get a new one. The minimum
256 size is STORE_BLOCK_SIZE, and we would expect this to be the norm, since
257 these functions are mostly called for small amounts of store. */
258
259 if (size > yield_length[pool])
260   {
261   int length = size <= STORE_BLOCK_SIZE ? STORE_BLOCK_SIZE : size;
262   int mlength = length + ALIGNED_SIZEOF_STOREBLOCK;
263   storeblock * newblock;
264
265   /* Sometimes store_reset() may leave a block for us; check if we can use it */
266
267   if (  (newblock = current_block[pool])
268      && (newblock = newblock->next)
269      && newblock->length < length
270      )
271     {
272     /* Give up on this block, because it's too small */
273     nblocks[pool]--;
274     internal_store_free(newblock, func, linenumber);
275     newblock = NULL;
276     }
277
278   /* If there was no free block, get a new one */
279
280   if (!newblock)
281     {
282     if ((nbytes[pool] += mlength) > maxbytes[pool])
283       maxbytes[pool] = nbytes[pool];
284     if ((pool_malloc += mlength) > max_pool_malloc)     /* Used in pools */
285       max_pool_malloc = pool_malloc;
286     nonpool_malloc -= mlength;                  /* Exclude from overall total */
287     if (++nblocks[pool] > maxblocks[pool])
288       maxblocks[pool] = nblocks[pool];
289
290     newblock = internal_store_malloc(mlength, func, linenumber);
291     newblock->next = NULL;
292     newblock->length = length;
293
294     if (!chainbase[pool])
295       chainbase[pool] = newblock;
296     else
297       current_block[pool]->next = newblock;
298     }
299
300   current_block[pool] = newblock;
301   yield_length[pool] = newblock->length;
302   next_yield[pool] =
303     (void *)(CS current_block[pool] + ALIGNED_SIZEOF_STOREBLOCK);
304   (void) VALGRIND_MAKE_MEM_NOACCESS(next_yield[pool], yield_length[pool]);
305   }
306
307 /* There's (now) enough room in the current block; the yield is the next
308 pointer. */
309
310 store_last_get[pool] = next_yield[pool];
311
312 /* Cut out the debugging stuff for utilities, but stop picky compilers from
313 giving warnings. */
314
315 #ifdef COMPILE_UTILITY
316 func = func;
317 linenumber = linenumber;
318 #else
319 DEBUG(D_memory)
320   debug_printf("---%d Get %6p %5d %-14s %4d\n", pool,
321     store_last_get[pool], size, func, linenumber);
322 #endif  /* COMPILE_UTILITY */
323
324 (void) VALGRIND_MAKE_MEM_UNDEFINED(store_last_get[pool], size);
325 /* Update next pointer and number of bytes left in the current block. */
326
327 next_yield[pool] = (void *)(CS next_yield[pool] + size);
328 yield_length[pool] -= size;
329 return store_last_get[pool];
330 }
331
332
333
334 /*************************************************
335 *       Get a block from the PERM pool           *
336 *************************************************/
337
338 /* This is just a convenience function, useful when just a single block is to
339 be obtained.
340
341 Arguments:
342   size        amount wanted
343   func        function from which called
344   linenumber  line number in source file
345
346 Returns:      pointer to store (panic on malloc failure)
347 */
348
349 void *
350 store_get_perm_3(int size, BOOL tainted, const char *func, int linenumber)
351 {
352 void *yield;
353 int old_pool = store_pool;
354 store_pool = POOL_PERM;
355 yield = store_get_3(size, tainted, func, linenumber);
356 store_pool = old_pool;
357 return yield;
358 }
359
360
361
362 /*************************************************
363 *      Extend a block if it is at the top        *
364 *************************************************/
365
366 /* While reading strings of unknown length, it is often the case that the
367 string is being read into the block at the top of the stack. If it needs to be
368 extended, it is more efficient just to extend within the top block rather than
369 allocate a new block and then have to copy the data. This function is provided
370 for the use of string_cat(), but of course can be used elsewhere too.
371 The block itself is not expanded; only the top allocation from it.
372
373 Arguments:
374   ptr        pointer to store block
375   oldsize    current size of the block, as requested by user
376   newsize    new size required
377   func       function from which called
378   linenumber line number in source file
379
380 Returns:     TRUE if the block is at the top of the stack and has been
381              extended; FALSE if it isn't at the top of the stack, or cannot
382              be extended
383 */
384
385 BOOL
386 store_extend_3(void *ptr, BOOL tainted, int oldsize, int newsize,
387    const char *func, int linenumber)
388 {
389 int pool = tainted ? store_pool + POOL_TAINT_BASE : store_pool;
390 int inc = newsize - oldsize;
391 int rounded_oldsize = oldsize;
392
393 if (oldsize < 0 || newsize < oldsize || newsize >= INT_MAX/2)
394   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
395             "bad memory extension requested (%d -> %d bytes) at %s %d",
396             oldsize, newsize, func, linenumber);
397
398 /* Check that the block being extended was already of the required taint status;
399 refuse to extend if not. */
400
401 if (is_tainted(ptr) != tainted)
402   return FALSE;
403
404 if (rounded_oldsize % alignment != 0)
405   rounded_oldsize += alignment - (rounded_oldsize % alignment);
406
407 if (CS ptr + rounded_oldsize != CS (next_yield[pool]) ||
408     inc > yield_length[pool] + rounded_oldsize - oldsize)
409   return FALSE;
410
411 /* Cut out the debugging stuff for utilities, but stop picky compilers from
412 giving warnings. */
413
414 #ifdef COMPILE_UTILITY
415 func = func;
416 linenumber = linenumber;
417 #else
418 DEBUG(D_memory)
419   debug_printf("---%d Ext %6p %5d %-14s %4d\n", pool, ptr, newsize,
420     func, linenumber);
421 #endif  /* COMPILE_UTILITY */
422
423 if (newsize % alignment != 0) newsize += alignment - (newsize % alignment);
424 next_yield[pool] = CS ptr + newsize;
425 yield_length[pool] -= newsize - rounded_oldsize;
426 (void) VALGRIND_MAKE_MEM_UNDEFINED(ptr + oldsize, inc);
427 return TRUE;
428 }
429
430
431
432
433 /*************************************************
434 *    Back up to a previous point on the stack    *
435 *************************************************/
436
437 /* This function resets the next pointer, freeing any subsequent whole blocks
438 that are now unused. Call with a cookie obtained from store_mark() only; do
439 not call with a pointer returned by store_get().  Both the untainted and tainted
440 pools corresposding to store_pool are reset.
441
442 Arguments:
443   r           place to back up to
444   func        function from which called
445   linenumber  line number in source file
446
447 Returns:      nothing
448 */
449
450 static void
451 internal_store_reset(void * ptr, int pool, const char *func, int linenumber)
452 {
453 storeblock * bb;
454 storeblock * b = current_block[pool];
455 char * bc = CS b + ALIGNED_SIZEOF_STOREBLOCK;
456 int newlength, count;
457 #ifndef COMPILE_UTILITY
458 int oldmalloc = pool_malloc;
459 #endif
460
461 /* Last store operation was not a get */
462
463 store_last_get[pool] = NULL;
464
465 /* See if the place is in the current block - as it often will be. Otherwise,
466 search for the block in which it lies. */
467
468 if (CS ptr < bc || CS ptr > bc + b->length)
469   {
470   for (b = chainbase[pool]; b; b = b->next)
471     {
472     bc = CS b + ALIGNED_SIZEOF_STOREBLOCK;
473     if (CS ptr >= bc && CS ptr <= bc + b->length) break;
474     }
475   if (!b)
476     log_write(0, LOG_MAIN|LOG_PANIC_DIE, "internal error: store_reset(%p) "
477       "failed: pool=%d %-14s %4d", ptr, pool, func, linenumber);
478   }
479
480 /* Back up, rounding to the alignment if necessary. When testing, flatten
481 the released memory. */
482
483 newlength = bc + b->length - CS ptr;
484 #ifndef COMPILE_UTILITY
485 if (debug_store)
486   {
487   assert_no_variables(ptr, newlength, func, linenumber);
488   if (f.running_in_test_harness)
489     {
490     (void) VALGRIND_MAKE_MEM_DEFINED(ptr, newlength);
491     memset(ptr, 0xF0, newlength);
492     }
493   }
494 #endif
495 (void) VALGRIND_MAKE_MEM_NOACCESS(ptr, newlength);
496 next_yield[pool] = CS ptr + (newlength % alignment);
497 count = yield_length[pool];
498 count = (yield_length[pool] = newlength - (newlength % alignment)) - count;
499 current_block[pool] = b;
500
501 /* Free any subsequent block. Do NOT free the first
502 successor, if our current block has less than 256 bytes left. This should
503 prevent us from flapping memory. However, keep this block only when it has
504 the default size. */
505
506 if (  yield_length[pool] < STOREPOOL_MIN_SIZE
507    && b->next
508    && b->next->length == STORE_BLOCK_SIZE)
509   {
510   b = b->next;
511 #ifndef COMPILE_UTILITY
512   if (debug_store)
513     assert_no_variables(b, b->length + ALIGNED_SIZEOF_STOREBLOCK,
514                         func, linenumber);
515 #endif
516   (void) VALGRIND_MAKE_MEM_NOACCESS(CS b + ALIGNED_SIZEOF_STOREBLOCK,
517                 b->length - ALIGNED_SIZEOF_STOREBLOCK);
518   }
519
520 bb = b->next;
521 b->next = NULL;
522
523 while ((b = bb))
524   {
525   int siz = b->length + ALIGNED_SIZEOF_STOREBLOCK;
526 #ifndef COMPILE_UTILITY
527   if (debug_store)
528     assert_no_variables(b, b->length + ALIGNED_SIZEOF_STOREBLOCK,
529                         func, linenumber);
530 #endif
531   bb = bb->next;
532   nbytes[pool] -= siz;
533   pool_malloc -= siz;
534   nblocks[pool]--;
535   internal_store_free(b, func, linenumber);
536   }
537
538 /* Cut out the debugging stuff for utilities, but stop picky compilers from
539 giving warnings. */
540
541 #ifdef COMPILE_UTILITY
542 func = func;
543 linenumber = linenumber;
544 #else
545 DEBUG(D_memory)
546   debug_printf("---%d Rst %6p %5d %-14s %4d %d\n", pool, ptr,
547     count + oldmalloc - pool_malloc,
548     func, linenumber, pool_malloc);
549 #endif  /* COMPILE_UTILITY */
550 }
551
552
553 rmark
554 store_reset_3(rmark r, int pool, const char *func, int linenumber)
555 {
556 void ** ptr = r;
557
558 if (pool >= POOL_TAINT_BASE)
559   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
560     "store_reset called for pool %d: %s %d\n", pool, func, linenumber);
561 if (!r)
562   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
563     "store_reset called with bad mark: %s %d\n", func, linenumber);
564
565 internal_store_reset(*ptr, pool + POOL_TAINT_BASE, func, linenumber);
566 internal_store_reset(ptr,  pool,                   func, linenumber);
567 return NULL;
568 }
569
570
571
572 /* Free tail-end unused allocation.  This lets us allocate a big chunk
573 early, for cases when we only discover later how much was really needed.
574
575 Can be called with a value from store_get(), or an offset after such.  Only
576 the tainted or untainted pool that serviced the store_get() will be affected.
577
578 This is mostly a cut-down version of internal_store_reset().
579 XXX needs rationalising
580 */
581
582 void
583 store_release_above_3(void *ptr, const char *func, int linenumber)
584 {
585 /* Search all pools' "current" blocks.  If it isn't one of those,
586 ignore it (it usually will be). */
587
588 for (int pool = 0; pool < nelem(current_block); pool++)
589   {
590   storeblock * b = current_block[pool];
591   char * bc;
592   int count, newlength;
593
594   if (!b)
595     continue;
596
597   bc = CS b + ALIGNED_SIZEOF_STOREBLOCK;
598   if (CS ptr < bc || CS ptr > bc + b->length)
599     continue;
600
601   /* Last store operation was not a get */
602
603   store_last_get[pool] = NULL;
604
605   /* Back up, rounding to the alignment if necessary. When testing, flatten
606   the released memory. */
607
608   newlength = bc + b->length - CS ptr;
609 #ifndef COMPILE_UTILITY
610   if (debug_store)
611     {
612     assert_no_variables(ptr, newlength, func, linenumber);
613     if (f.running_in_test_harness)
614       {
615       (void) VALGRIND_MAKE_MEM_DEFINED(ptr, newlength);
616       memset(ptr, 0xF0, newlength);
617       }
618     }
619 #endif
620   (void) VALGRIND_MAKE_MEM_NOACCESS(ptr, newlength);
621   next_yield[pool] = CS ptr + (newlength % alignment);
622   count = yield_length[pool];
623   count = (yield_length[pool] = newlength - (newlength % alignment)) - count;
624
625   /* Cut out the debugging stuff for utilities, but stop picky compilers from
626   giving warnings. */
627
628 #ifdef COMPILE_UTILITY
629   func = func;
630   linenumber = linenumber;
631 #else
632   DEBUG(D_memory)
633     debug_printf("---%d Rel %6p %5d %-14s %4d %d\n", pool, ptr, count,
634       func, linenumber, pool_malloc);
635 #endif
636   return;
637   }
638 #ifndef COMPILE_UTILITY
639 DEBUG(D_memory)
640   debug_printf("non-last memory release try: %s %d\n", func, linenumber);
641 #endif
642 }
643
644
645
646 rmark
647 store_mark_3(const char *func, int linenumber)
648 {
649 void ** p;
650
651 if (store_pool >= POOL_TAINT_BASE)
652   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
653     "store_mark called for pool %d: %s %d\n", store_pool, func, linenumber);
654
655 /* Stash a mark for the tainted-twin release, in the untainted twin. Return
656 a cookie (actually the address in the untainted pool) to the caller.
657 Reset uses the cookie to recover the t-mark, winds back the tainted pool with it
658 and winds back the untainted pool with the cookie. */
659
660 p = store_get_3(sizeof(void *), FALSE, func, linenumber);
661 *p = store_get_3(0, TRUE, func, linenumber);
662 return p;
663 }
664
665
666
667
668 /************************************************
669 *             Release store                     *
670 ************************************************/
671
672 /* This function checks that the pointer it is given is the first thing in a
673 block, and if so, releases that block.
674
675 Arguments:
676   block       block of store to consider
677   func        function from which called
678   linenumber  line number in source file
679
680 Returns:      nothing
681 */
682
683 static void
684 store_release_3(void * block, int pool, const char * func, int linenumber)
685 {
686 /* It will never be the first block, so no need to check that. */
687
688 for (storeblock * b = chainbase[pool]; b; b = b->next)
689   {
690   storeblock * bb = b->next;
691   if (bb && CS block == CS bb + ALIGNED_SIZEOF_STOREBLOCK)
692     {
693     int siz = bb->length + ALIGNED_SIZEOF_STOREBLOCK;
694     b->next = bb->next;
695     nbytes[pool] -= siz;
696     pool_malloc -= siz;
697     nblocks[pool]--;
698
699     /* Cut out the debugging stuff for utilities, but stop picky compilers
700     from giving warnings. */
701
702 #ifdef COMPILE_UTILITY
703     func = func;
704     linenumber = linenumber;
705 #else
706     DEBUG(D_memory)
707       debug_printf("-Release %6p %-20s %4d %d\n", (void *)bb, func,
708         linenumber, pool_malloc);
709
710     if (f.running_in_test_harness)
711       memset(bb, 0xF0, bb->length+ALIGNED_SIZEOF_STOREBLOCK);
712 #endif  /* COMPILE_UTILITY */
713
714     free(bb);
715     return;
716     }
717   }
718 }
719
720
721 /************************************************
722 *             Move store                        *
723 ************************************************/
724
725 /* Allocate a new block big enough to expend to the given size and
726 copy the current data into it.  Free the old one if possible.
727
728 This function is specifically provided for use when reading very
729 long strings, e.g. header lines. When the string gets longer than a
730 complete block, it gets copied to a new block. It is helpful to free
731 the old block iff the previous copy of the string is at its start,
732 and therefore the only thing in it. Otherwise, for very long strings,
733 dead store can pile up somewhat disastrously. This function checks that
734 the pointer it is given is the first thing in a block, and that nothing
735 has been allocated since. If so, releases that block.
736
737 Arguments:
738   block
739   newsize
740   len
741
742 Returns:        new location of data
743 */
744
745 void *
746 store_newblock_3(void * block, BOOL tainted, int newsize, int len,
747   const char * func, int linenumber)
748 {
749 int pool = tainted ? store_pool + POOL_TAINT_BASE : store_pool;
750 BOOL release_ok = !tainted && store_last_get[pool] == block;
751 uschar * newtext;
752
753 #if !defined(MACRO_PREDEF) && !defined(COMPILE_UTILITY)
754 if (is_tainted(block) != tainted)
755   die_tainted(US"store_newblock", CUS func, linenumber);
756 #endif
757
758 if (len < 0 || len > newsize)
759   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
760             "bad memory extension requested (%d -> %d bytes) at %s %d",
761             len, newsize, func, linenumber);
762
763 newtext = store_get(newsize, tainted);
764 memcpy(newtext, block, len);
765 if (release_ok) store_release_3(block, pool, func, linenumber);
766 return (void *)newtext;
767 }
768
769
770
771
772 /*************************************************
773 *                Malloc store                    *
774 *************************************************/
775
776 /* Running out of store is a total disaster for exim. Some malloc functions
777 do not run happily on very small sizes, nor do they document this fact. This
778 function is called via the macro store_malloc().
779
780 Arguments:
781   size        amount of store wanted
782   func        function from which called
783   line        line number in source file
784
785 Returns:      pointer to gotten store (panic on failure)
786 */
787
788 static void *
789 internal_store_malloc(int size, const char *func, int line)
790 {
791 void * yield;
792
793 if (size < 0 || size >= INT_MAX/2)
794   log_write(0, LOG_MAIN|LOG_PANIC_DIE,
795             "bad memory allocation requested (%d bytes) at %s %d",
796             size, func, line);
797
798 if (size < 16) size = 16;
799
800 if (!(yield = malloc((size_t)size)))
801   log_write(0, LOG_MAIN|LOG_PANIC_DIE, "failed to malloc %d bytes of memory: "
802     "called from line %d in %s", size, line, func);
803
804 if ((nonpool_malloc += size) > max_nonpool_malloc)
805   max_nonpool_malloc = nonpool_malloc;
806
807 /* Cut out the debugging stuff for utilities, but stop picky compilers from
808 giving warnings. */
809
810 #ifdef COMPILE_UTILITY
811 func = func; line = line;
812 #else
813
814 /* If running in test harness, spend time making sure all the new store
815 is not filled with zeros so as to catch problems. */
816
817 if (f.running_in_test_harness)
818   memset(yield, 0xF0, (size_t)size);
819 DEBUG(D_memory) debug_printf("--Malloc %6p %5d bytes\t%-14s %4d\tpool %5d  nonpool %5d\n",
820   yield, size, func, line, pool_malloc, nonpool_malloc);
821 #endif  /* COMPILE_UTILITY */
822
823 return yield;
824 }
825
826 void *
827 store_malloc_3(int size, const char *func, int linenumber)
828 {
829 if (n_nonpool_blocks++ > max_nonpool_blocks)
830   max_nonpool_blocks = n_nonpool_blocks;
831 return internal_store_malloc(size, func, linenumber);
832 }
833
834
835 /************************************************
836 *             Free store                        *
837 ************************************************/
838
839 /* This function is called by the macro store_free().
840
841 Arguments:
842   block       block of store to free
843   func        function from which called
844   linenumber  line number in source file
845
846 Returns:      nothing
847 */
848
849 static void
850 internal_store_free(void * block, const char * func, int linenumber)
851 {
852 #ifdef COMPILE_UTILITY
853 func = func;
854 linenumber = linenumber;
855 #else
856 DEBUG(D_memory)
857   debug_printf("----Free %6p %-20s %4d\n", block, func, linenumber);
858 #endif  /* COMPILE_UTILITY */
859 free(block);
860 }
861
862 void
863 store_free_3(void * block, const char * func, int linenumber)
864 {
865 n_nonpool_blocks--;
866 internal_store_free(block, func, linenumber);
867 }
868
869 /******************************************************************************/
870 /* Stats output on process exit */
871 void
872 store_exit(void)
873 {
874 #ifndef COMPILE_UTILITY
875 DEBUG(D_memory)
876  {
877  debug_printf("----Exit nonpool max: %3d kB in %d blocks\n",
878   (max_nonpool_malloc+1023)/1024, max_nonpool_blocks);
879  debug_printf("----Exit npools  max: %3d kB\n", max_pool_malloc/1024);
880  for (int i = 0; i < NPOOLS; i++)
881   debug_printf("----Exit  pool %d max: %3d kB in %d blocks\t%s %s\n",
882     i, maxbytes[i]/1024, maxblocks[i], poolclass[i], pooluse[i]);
883  }
884 #endif
885 }
886
887 /* End of store.c */