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 #include <stdlib.h> /* abort */
  13
  14 #include "mmprivate.h"
  15
  16 /* Resize the given region to the new size, returning a pointer
  17    to the (possibly moved) region.  This is optimized for speed;
  18    some benchmarks seem to indicate that greater compactness is
  19    achieved by unconditionally allocating and copying to a
  20    new region.  This module has incestuous knowledge of the
  21    internals of both mfree and mmalloc. */
  22
  23 void *mrealloc(xbt_mheap_t mdp, void *ptr, size_t size)
  24 {
  25   void *result;
  26   int type;
  27   size_t block, blocks, oldlimit;
  28
  29   /* Only keep real realloc and hidden malloc and free to the relevant functions */
  30   if (size == 0) {
  31         fprintf(stderr,"free from realloc...");
  32     mfree(mdp, ptr);
  33     fprintf(stderr,"done\n");
  34     return mmalloc(mdp, 0);
  35   } else if (ptr == NULL) {
  36     return mmalloc(mdp, size);
  37   }
  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, refusing to proceed (another solution would be to malloc it instead of reallocing it, see source code)\n");
  44     result = mmalloc(mdp, size);
  45     abort();
  46     return result;
  47   }
  48
  49   block = BLOCK(ptr);
  50
  51   type = mdp->heapinfo[block].type;
  52   if (type<0)
  53       THROWF(arg_error,0,"Asked realloc a fragment comming from a *free* block. I'm puzzled.");
  54
  55   switch (type) {
  56   case 0:
  57     /* Maybe reallocate a large block to a small fragment.  */
  58     if (size <= BLOCKSIZE / 2) {
  59       //printf("(%s) alloc large block...",xbt_thread_self_name());
  60       result = mmalloc(mdp, size);
  61       if (result != NULL) {
  62         memcpy(result, ptr, size);
  63         mfree(mdp, ptr);
  64         return (result);
  65       }
  66     }
  67
  68     /* The new size is a large allocation as well;
  69        see if we can hold it in place. */
  70     blocks = BLOCKIFY(size);
  71     if (blocks < mdp->heapinfo[block].busy_block.size) {
  72         int it;
  73       /* The new size is smaller; return excess memory to the free list. */
  74       //printf("(%s) return excess memory...",xbt_thread_self_name());
  75      for (it= block+blocks; it< mdp->heapinfo[block].busy_block.size ; it++)
  76          mdp->heapinfo[it].type = 0;
  77       mdp->heapinfo[block + blocks].busy_block.size
  78           = mdp->heapinfo[block].busy_block.size - blocks;
  79       mdp->heapinfo[block].busy_block.size = blocks;
  80       mdp->heapinfo[block].busy_block.busy_size = size;
  81
  82       mfree(mdp, ADDRESS(block + blocks));
  83       result = ptr;
  84     } else if (blocks == mdp->heapinfo[block].busy_block.size) {
  85       /* No size change necessary.  */
  86       result = ptr;
  87     } else {
  88       /* Won't fit, so allocate a new region that will.
  89          Free the old region first in case there is sufficient
  90          adjacent free space to grow without moving. */
  91       blocks = mdp->heapinfo[block].busy_block.size;
  92       /* Prevent free from actually returning memory to the system.  */
  93       oldlimit = mdp->heaplimit;
  94       mdp->heaplimit = 0;
  95       mfree(mdp, ptr);
  96       mdp->heaplimit = oldlimit;
  97       result = mmalloc(mdp, size);
  98       if (result == NULL) {
  99         mmalloc(mdp, blocks * BLOCKSIZE);
 100         return (NULL);
 101       }
 102       if (ptr != result)
 103         memmove(result, ptr, blocks * BLOCKSIZE);
 104     }
 105     break;
 106
 107   default:
 108     /* Old size is a fragment; type is logarithm
 109        to base two of the fragment size.  */
 110     if (size > (size_t) (1 << (type - 1)) && size <= (size_t) (1 << type)) {
 111       /* The new size is the same kind of fragment.  */
 112       //printf("(%s) new size is same kind of fragment...",xbt_thread_self_name());
 113       result = ptr;
 114     } else {
 115       /* The new size is different; allocate a new space,
 116          and copy the lesser of the new size and the old. */
 117       //printf("(%s) new size is different...",xbt_thread_self_name());
 118
 119       result = mmalloc(mdp, size);
 120       if (result == NULL)
 121         return (NULL);
 122
 123       memcpy(result, ptr, MIN(size, (size_t) 1 << type));
 124       mfree(mdp, ptr);
 125     }
 126     break;
 127   }
 128   //printf("(%s) Done reallocing: %p\n",xbt_thread_self_name(),result);fflush(stdout);
 129   return (result);
 130 }