Be explicit in order of evaluation in the reiserfs code
[yaboot.git] / lib / malloc.c
1 /*  malloc.c - Dumb memory allocation routines
2  *
3  *  Copyright (C) 1997 Paul Mackerras
4  *                1996 Maurizio Plaza
5  *                1996 Jakub Jelinek
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 2 of the License, or
10  *  (at your option) any later version.
11  *
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, write to the Free Software
19  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
20  */
21
22 #include "types.h"
23 #include "stddef.h"
24 #include "string.h"
25
26 /* Copied from asm-generic/errno-base.h */
27 #define ENOMEM          12      /* Out of memory */
28 #define EINVAL          22      /* Invalid argument */
29
30 /* Imported functions */
31 extern void prom_printf (char *fmt, ...);
32
33 static char *malloc_ptr = 0;
34 static char *malloc_top = 0;
35 static char *last_alloc = 0;
36
37 void malloc_init(void *bottom, unsigned long size)
38 {
39         malloc_ptr = bottom;
40         malloc_top = bottom + size;
41 }
42
43 void malloc_dispose(void)
44 {
45         malloc_ptr = 0;
46         last_alloc = 0;
47 }
48
49 void *malloc (unsigned int size)
50 {
51     char *caddr;
52
53     if (!malloc_ptr)
54         return NULL;
55     if ((malloc_ptr + size + sizeof(int)) > malloc_top) {
56         prom_printf("malloc failed\n");
57         return NULL;
58     }
59     *(int *)malloc_ptr = size;
60     caddr = malloc_ptr + sizeof(int);
61     malloc_ptr += size + sizeof(int);
62     last_alloc = caddr;
63     malloc_ptr = (char *) ((((unsigned int) malloc_ptr) + 3) & (~3));
64     return caddr;
65 }
66
67 /* Do not fall back to the malloc above as posix_memalign is needed by
68  * external libraries not yaboot */
69 int posix_memalign(void **memptr, size_t alignment, size_t size)
70 {
71     char *caddr;
72     /* size of allocation including the alignment */
73     size_t alloc_size;
74
75     if (!malloc_ptr)
76         return EINVAL;
77
78     /* Minimal aligment is sizeof(void *) */
79     if (alignment < sizeof(void*))
80         alignment = sizeof(void*);
81
82     /* Check for valid alignment and power of 2 */
83     if ((alignment % sizeof(void*) != 0) || ((alignment-1)&alignment))
84         return EINVAL;
85
86     if (size == 0) {
87         *memptr=NULL;
88         return 0;
89     }
90
91     caddr = (char*)(
92              (size_t)((malloc_ptr + sizeof(int))+(alignment-1)) &
93              (~(alignment-1))
94             );
95
96     alloc_size = size + (caddr - (malloc_ptr+sizeof(int)));
97
98     if ((malloc_ptr + alloc_size + sizeof(int)) > malloc_top)
99         return ENOMEM;
100
101     *(int *)(caddr - sizeof(int)) = size;
102     malloc_ptr += alloc_size + sizeof(int);
103     last_alloc = caddr;
104     malloc_ptr = (char *) ((((unsigned int) malloc_ptr) + 3) & (~3));
105     *memptr=(void*)caddr;
106
107     return 0;
108 }
109
110 void *realloc(void *ptr, unsigned int size)
111 {
112     char *caddr, *oaddr = ptr;
113
114     if (!malloc_ptr)
115         return NULL;
116     if (oaddr == last_alloc) {
117         if (oaddr + size > malloc_top) {
118                 prom_printf("realloc failed\n");
119                 return NULL;
120         }
121         *(int *)(oaddr - sizeof(int)) = size;
122         malloc_ptr = oaddr + size;
123         return oaddr;
124     }
125     caddr = malloc(size);
126     if (caddr != 0 && oaddr != 0)
127         memcpy(caddr, oaddr, *(int *)(oaddr - sizeof(int)));
128     return caddr;
129 }
130
131 void free (void *m)
132 {
133     if (!malloc_ptr)
134         return;
135     if (m == last_alloc)
136         malloc_ptr = (char *) last_alloc - sizeof(int);
137 }
138
139 void mark (void **ptr)
140 {
141     if (!malloc_ptr)
142         return;
143     *ptr = (void *) malloc_ptr;
144 }
145
146 void release (void *ptr)
147 {
148     if (!malloc_ptr)
149         return;
150     malloc_ptr = (char *) ptr;
151 }
152
153 char *strdup(char const *str)
154 {
155     char *p = malloc(strlen(str) + 1);
156     if (p)
157          strcpy(p, str);
158     return p;
159 }