summaryrefslogtreecommitdiff
path: root/as/alloc.c
blob: b4e94fc1df6b88b04a1a0faccb13c32c75faa719 (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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

#include "syshead.h"
#include "const.h"
#include "type.h"
#include "align.h"

PRIVATE char NOMEMEORY[] = "Cannot allocate sufficient memory";

#ifdef USE_FIXED_HEAP
PRIVATE char *heapend;		/* end of free space for symbol list */
PRIVATE char *heapptr;		/* next free space in symbol list */
#endif

#ifndef USE_FIXED_HEAP
PRIVATE char tempbuf[2048];
#endif

void
init_heap()
{
#ifdef USE_FIXED_HEAP
#ifndef USERMEM
#define USERMEM 0xAC00U
#endif

#ifdef __AS386_16__
    int stk;
    heapptr = sbrk(0);
    heapend = ((char*)&stk) - STAKSIZ - 16;
    brk(heapend);
    if(sbrk(0) != heapend)
       as_abort(NOMEMEORY);
#else
#ifdef SOS_EDOS
    heapend = stackreg() - STAKSIZ;
#else
    heapptr = malloc(USERMEM);
    heapend = heapptr + USERMEM;
    if (heapptr == 0)
	as_abort(NOMEMEORY);
#endif
#endif
#endif
}

void * temp_buf()
{
#ifdef USE_FIXED_HEAP
    return heapptr;
#else
    return tempbuf;
#endif
}

void * 
asalloc(size)
unsigned int size;
{
    void * rv;
#ifdef USE_FIXED_HEAP
    align(heapptr);
    if (heapptr+size < heapend)
    {
        rv = heapptr;
        heapptr += size;
    }
    else
       rv = 0;
#else
    rv = malloc(size);
#endif
    if (rv == 0 && size) as_abort(NOMEMEORY);
    return rv;
}


void * 
asrealloc(oldptr, size)
void * oldptr;
unsigned int size;
{
    void * rv;
#ifdef USE_FIXED_HEAP
    if (oldptr == 0) return asalloc(size);

    if ((char*)oldptr+size < heapend)
    {
        heapptr = (char*)oldptr + size;
        rv = oldptr;
    }
    else
        rv = 0;
#else
    rv = realloc(oldptr, size);
#endif

    if (rv == 0) as_abort(NOMEMEORY);
    return rv;
}