src/xbt/mmalloc/mrealloc.c

   1 /* Change the size of a block allocated by `mmalloc'.
   2    Copyright 1990, 1991 Free Software Foundation
   3                   Written May 1989 by Mike Haertel. */
   4
   5 /* Copyright (c) 2010. The SimGrid Team.
   6  * All rights reserved.                                                     */
   7
   8 /* This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the license (GNU LGPL) which comes with this package. */
  10
  11 #include <string.h>             /* Prototypes for memcpy, memmove, memset, etc */
  12
  13 #include "mmprivate.h"
  14
  15 /* Resize the given region to the new size, returning a pointer
  16    to the (possibly moved) region.  This is optimized for speed;
  17    some benchmarks seem to indicate that greater compactness is
  18    achieved by unconditionally allocating and copying to a
  19    new region.  This module has incestuous knowledge of the
  20    internals of both mfree and mmalloc. */
  21
  22 void *mrealloc(void *md, void *ptr, size_t size)
  23 {
  24   struct mdesc *mdp;
  25   void *result;
  26   int type;
  27   size_t block, blocks, oldlimit;
  28
  29
  30   if (size == 0) {
  31     mfree(md, ptr);
  32     return (mmalloc(md, 0));
  33   } else if (ptr == NULL) {
  34     return (mmalloc(md, size));
  35   }
  36
  37   mdp = MD_TO_MDP(md);
  38
  39   //printf("(%s)realloc %p to %d...",xbt_thread_self_name(),ptr,(int)size);
  40
  41   if ((char *) ptr < (char *) mdp->heapbase || BLOCK(ptr) > mdp->heapsize) {
  42     printf
  43         ("FIXME. Ouch, this pointer is not mine. I will malloc it instead of reallocing it. (please report this bug)\n");
  44     result = mmalloc(md, size);
  45     abort();
  46     return result;
  47   }
  48
  49   LOCK(mdp);
  50   if (mdp->mrealloc_hook != NULL) {
  51     UNLOCK(mdp);
  52     return ((*mdp->mrealloc_hook) (md, ptr, size));
  53   }
  54
  55   block = BLOCK(ptr);
  56
  57   type = mdp->heapinfo[block].busy.type;
  58   switch (type) {
  59   case 0:
  60     /* Maybe reallocate a large block to a small fragment.  */
  61     if (size <= BLOCKSIZE / 2) {
  62       UNLOCK(mdp);
  63       //printf("(%s) alloc large block...",xbt_thread_self_name());
  64       result = mmalloc(md, size);
  65       if (result != NULL) {
  66         memcpy(result, ptr, size);
  67         mfree(md, ptr);
  68         return (result);
  69       }
  70     }
  71
  72     /* The new size is a large allocation as well;
  73        see if we can hold it in place. */
  74     LOCK(mdp);
  75     blocks = BLOCKIFY(size);
  76     if (blocks < mdp->heapinfo[block].busy.info.size) {
  77       /* The new size is smaller; return excess memory to the free list. */
  78       //printf("(%s) return excess memory...",xbt_thread_self_name());
  79       mdp->heapinfo[block + blocks].busy.type = 0;
  80       mdp->heapinfo[block + blocks].busy.info.size
  81           = mdp->heapinfo[block].busy.info.size - blocks;
  82       mdp->heapinfo[block].busy.info.size = blocks;
  83       mfree(md, ADDRESS(block + blocks));
  84       result = ptr;
  85     } else if (blocks == mdp->heapinfo[block].busy.info.size) {
  86       /* No size change necessary.  */
  87       result = ptr;
  88     } else {
  89       /* Won't fit, so allocate a new region that will.
  90          Free the old region first in case there is sufficient
  91          adjacent free space to grow without moving. */
  92       blocks = mdp->heapinfo[block].busy.info.size;
  93       /* Prevent free from actually returning memory to the system.  */
  94       oldlimit = mdp->heaplimit;
  95       mdp->heaplimit = 0;
  96       mfree(md, ptr);
  97       mdp->heaplimit = oldlimit;
  98       UNLOCK(mdp);
  99       result = mmalloc(md, size);
 100       if (result == NULL) {
 101         mmalloc(md, blocks * BLOCKSIZE);
 102         return (NULL);
 103       }
 104       if (ptr != result)
 105         memmove(result, ptr, blocks * BLOCKSIZE);
 106       LOCK(mdp);
 107     }
 108     break;
 109
 110   default:
 111     /* Old size is a fragment; type is logarithm
 112        to base two of the fragment size.  */
 113     if (size > (size_t) (1 << (type - 1)) && size <= (size_t) (1 << type)) {
 114       /* The new size is the same kind of fragment.  */
 115       //printf("(%s) new size is same kind of fragment...",xbt_thread_self_name());
 116       result = ptr;
 117     } else {
 118       /* The new size is different; allocate a new space,
 119          and copy the lesser of the new size and the old. */
 120       //printf("(%s) new size is different...",xbt_thread_self_name());
 121
 122       UNLOCK(mdp);
 123       result = mmalloc(md, size);
 124       if (result == NULL)
 125         return (NULL);
 126
 127       memcpy(result, ptr, MIN(size, (size_t) 1 << type));
 128       mfree(md, ptr);
 129     }
 130     break;
 131   }
 132   //printf("(%s) Done reallocing: %p\n",xbt_thread_self_name(),result);fflush(stdout);
 133   return (result);
 134 }