Logo AND Algorithmique Numérique Distribuée

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