summaryrefslogtreecommitdiff
path: root/libc/stdlib/malloc/realloc.c
blob: 4372751d0706b8a000226e4b3c61b5b234e29122 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
/*
 * libc/stdlib/malloc/realloc.c -- realloc function
 *
 *  Copyright (C) 2002  NEC Corporation
 *  Copyright (C) 2002  Miles Bader <miles@gnu.org>
 *
 * This file is subject to the terms and conditions of the GNU Lesser
 * General Public License.  See the file COPYING.LIB in the main
 * directory of this archive for more details.
 * 
 * Written by Miles Bader <miles@gnu.org>
 */

#include <stdlib.h>
#include <string.h>

#include "malloc.h"
#include "heap.h"


void *
realloc (void *mem, size_t new_size)
{
  if (! mem)
    return malloc (new_size);
  else
    {
      char *base_mem = MALLOC_BASE (mem);
      size_t size = MALLOC_SIZE (mem);

      /* Include extra space to record the size of the allocated block.
	 Also make sure that we're dealing in a multiple of the heap
	 allocation unit (SIZE is already guaranteed to be so).*/
      new_size = HEAP_ADJUST_SIZE (new_size + MALLOC_HEADER_SIZE);

      MALLOC_DEBUG ("realloc: 0x%lx, %d (base = 0x%lx, total_size = %d)\n",
		    (long)mem, new_size, (long)base_mem, size);

      if (new_size > size)
	/* Grow the block.  */
	{
	  size_t extra = new_size - size;

	  __malloc_lock ();
	  extra = __heap_alloc_at (&__malloc_heap, base_mem + size, extra);
	  __malloc_unlock ();

	  if (extra)
	    /* Record the changed size.  */
	    MALLOC_SET_SIZE (mem, new_size);
	  else
	    /* Our attempts to extend MEM in place failed, just
	       allocate-and-copy.  */
	    {
	      void *new_mem = malloc (new_size);
	      if (new_mem)
		{
		  memcpy (new_mem, mem, size);
		  free (mem);
		}
	      mem = new_mem;
	    }
	}
      else if (new_size + HEAP_MIN_FREE_AREA_SIZE <= size)
	/* Shrink the block.  */
	{
	  __malloc_lock ();
	  __heap_free (&__malloc_heap, base_mem + new_size, size - new_size);
	  __malloc_unlock ();
	  MALLOC_SET_SIZE (mem, new_size);
	}

      if (mem)
	MALLOC_DEBUG ("  realloc: returning 0x%lx"
		      " (base:0x%lx, total_size:%d)\n",
		      (long)new_mem, (long)new_mem - sizeof(size_t), size);

      return mem;
    }
}