src/xbt/mmalloc/mrealloc.c

   1 /* Change the size of a block allocated by `mmalloc'. */
   2
   3 /* Copyright (c) 2010-2014. The SimGrid Team.
   4  * All rights reserved.                                                     */
   5
   6 /* This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the license (GNU LGPL) which comes with this package. */
   8
   9 /* Copyright 1990, 1991 Free Software Foundation
  10    Written May 1989 by Mike Haertel. */
  11
  12 #include <string.h>             /* Prototypes for memcpy, memmove, memset, etc */
  13 #include <stdlib.h> /* abort */
  14
  15 #include "mmprivate.h"
  16
  17 /* Resize the given region to the new size, returning a pointer
  18    to the (possibly moved) region.  This is optimized for speed;
  19    some benchmarks seem to indicate that greater compactness is
  20    achieved by unconditionally allocating and copying to a
  21    new region.  This module has incestuous knowledge of the
  22    internals of both mfree and mmalloc. */
  23
  24 void *mrealloc(xbt_mheap_t mdp, void *ptr, size_t size)
  25 {
  26   void *result;
  27   int type;
  28   size_t block, blocks, oldlimit;
  29
  30   /* Only keep real realloc, and reroute hidden malloc and free to the relevant functions */
  31   if (size == 0) {
  32     mfree(mdp, ptr);
  33     return mmalloc(mdp, 0);
  34   } else if (ptr == NULL) {
  35     return mmalloc(mdp, size);
  36   }
  37
  38   //printf("(%s)realloc %p to %d...",xbt_thread_self_name(),ptr,(int)size);
  39
  40   if ((char *) ptr < (char *) mdp->heapbase || BLOCK(ptr) > mdp->heapsize) {
  41     printf
  42       ("FIXME. Ouch, this pointer is not mine, refusing to proceed (another solution would be to malloc it instead of reallocing it, see source code)\n");
  43     result = mmalloc(mdp, size);
  44     abort();
  45     return result;
  46   }
  47
  48   size_t requested_size = size; // The amount of memory requested by user, for real
  49
  50   /* Work even if the user was stupid enough to ask a ridicullously small block (even 0-length),
  51    *    ie return a valid block that can be realloced and freed.
  52    * glibc malloc does not use this trick but return a constant pointer, but we need to enlist the free fragments later on.
  53    */
  54   if (size < SMALLEST_POSSIBLE_MALLOC)
  55     size = SMALLEST_POSSIBLE_MALLOC;
  56
  57   block = BLOCK(ptr);
  58
  59   type = mdp->heapinfo[block].type;
  60
  61   switch (type) {
  62   case MMALLOC_TYPE_HEAPINFO:
  63     fprintf(stderr, "Asked realloc a fragment coming from a heapinfo block. I'm confused.\n");
  64     abort();
  65     break;
  66
  67   case MMALLOC_TYPE_FREE:
  68     fprintf(stderr, "Asked realloc a fragment coming from a *free* block. I'm puzzled.\n");
  69     abort();
  70     break;
  71
  72   case MMALLOC_TYPE_UNFRAGMENTED:
  73     /* Maybe reallocate a large block to a small fragment.  */
  74
  75     if (size <= BLOCKSIZE / 2) { // Full block -> Fragment; no need to optimize for time
  76
  77       result = mmalloc(mdp, size);
  78       if (result != NULL) { // useless (mmalloc never returns NULL), but harmless
  79         memcpy(result, ptr, requested_size);
  80         mfree(mdp, ptr);
  81         return (result);
  82       }
  83     }
  84
  85     /* Full blocks -> Full blocks; see if we can hold it in place. */
  86     blocks = BLOCKIFY(size);
  87     if (blocks < mdp->heapinfo[block].busy_block.size) {
  88       int it;
  89       /* The new size is smaller; return excess memory to the free list. */
  90       //printf("(%s) return excess memory...",xbt_thread_self_name());
  91       for (it= block+blocks; it< mdp->heapinfo[block].busy_block.size ; it++){
  92         mdp->heapinfo[it].type = MMALLOC_TYPE_UNFRAGMENTED; // FIXME that should be useless, type should already be 0 here
  93         mdp->heapinfo[it].busy_block.ignore = 0;
  94         mdp->heapinfo[it].busy_block.size = 0;
  95         mdp->heapinfo[it].busy_block.busy_size = 0;
  96       }
  97
  98       mdp->heapinfo[block + blocks].busy_block.size
  99         = mdp->heapinfo[block].busy_block.size - blocks;
 100       mfree(mdp, ADDRESS(block + blocks));
 101
 102       mdp->heapinfo[block].busy_block.size = blocks;
 103       mdp->heapinfo[block].busy_block.busy_size = requested_size;
 104       mdp->heapinfo[block].busy_block.ignore = 0;
 105
 106       result = ptr;
 107     } else if (blocks == mdp->heapinfo[block].busy_block.size) {
 108
 109       /* No block size change necessary; only update the requested size  */
 110       result = ptr;
 111       mdp->heapinfo[block].busy_block.busy_size = requested_size;
 112       mdp->heapinfo[block].busy_block.ignore = 0;
 113
 114     } else {
 115       /* Won't fit, so allocate a new region that will.
 116          Free the old region first in case there is sufficient adjacent free space to grow without moving.
 117          This trick mandates using a specific version of mmalloc that does not memset the memory to 0 after
 118            action for obvious reasons. */
 119       blocks = mdp->heapinfo[block].busy_block.size;
 120       /* Prevent free from actually returning memory to the system.  */
 121       oldlimit = mdp->heaplimit;
 122       mdp->heaplimit = 0;
 123       mfree(mdp, ptr);
 124       mdp->heaplimit = oldlimit;
 125
 126       result = mmalloc_no_memset(mdp, requested_size);
 127       //fprintf(stderr,"remalloc(%zu)~>%p\n",requested_size,result);
 128
 129       if (ptr != result)
 130         memmove(result, ptr, blocks * BLOCKSIZE);
 131       /* FIXME: we should memset the end of the recently area */
 132     }
 133     break;
 134
 135   default: /* Fragment -> ??; type=logarithm to base two of the fragment size.  */
 136
 137     if (type < 0) {
 138       fprintf(stderr, "Unkown mmalloc block type.\n");
 139       abort();
 140     }
 141
 142     if (size > (size_t) (1 << (type - 1)) && size <= (size_t) (1 << type)) {
 143       /* The new size is the same kind of fragment.  */
 144       //printf("(%s) new size is same kind of fragment...",xbt_thread_self_name());
 145
 146       result = ptr;
 147       int frag_nb = RESIDUAL(result, BLOCKSIZE) >> type;
 148       mdp->heapinfo[block].busy_frag.frag_size[frag_nb] = requested_size;
 149       mdp->heapinfo[block].busy_frag.ignore[frag_nb] = 0;
 150
 151     } else { /* fragment -> Either other fragment, or block */
 152       /* The new size is different; allocate a new space,
 153          and copy the lesser of the new size and the old. */
 154       //printf("(%s) new size is different...",xbt_thread_self_name());
 155
 156       result = mmalloc(mdp, requested_size);
 157
 158       memcpy(result, ptr, MIN(requested_size, (size_t) 1 << type));
 159       mfree(mdp, ptr);
 160     }
 161     break;
 162   }
 163   //printf("(%s) Done reallocing: %p\n",xbt_thread_self_name(),result);fflush(stdout);
 164   return (result);
 165 }