1 /* SPDX-License-Identifier: GPL-2.0 */ 1 /* SPDX-License-Identifier: GPL-2.0 */
2 /* 2 /*
3 * linux/compr_mm.h 3 * linux/compr_mm.h
4 * 4 *
5 * Memory management for pre-boot and ramdisk 5 * Memory management for pre-boot and ramdisk uncompressors
6 * 6 *
7 * Authors: Alain Knaff <alain@knaff.lu> 7 * Authors: Alain Knaff <alain@knaff.lu>
8 * 8 *
9 */ 9 */
10 10
11 #ifndef DECOMPR_MM_H 11 #ifndef DECOMPR_MM_H
12 #define DECOMPR_MM_H 12 #define DECOMPR_MM_H
13 13
14 #ifdef STATIC 14 #ifdef STATIC
15 15
16 /* Code active when included from pre-boot env 16 /* Code active when included from pre-boot environment: */
17 17
18 /* 18 /*
19 * Some architectures want to ensure there is 19 * Some architectures want to ensure there is no local data in their
20 * pre-boot environment, so that data can arbi 20 * pre-boot environment, so that data can arbitrarily relocated (via
21 * GOT references). This is achieved by defin 21 * GOT references). This is achieved by defining STATIC_RW_DATA to
22 * be null. 22 * be null.
23 */ 23 */
24 #ifndef STATIC_RW_DATA 24 #ifndef STATIC_RW_DATA
25 #define STATIC_RW_DATA static 25 #define STATIC_RW_DATA static
26 #endif 26 #endif
27 27
28 /* 28 /*
29 * When an architecture needs to share the mal 29 * When an architecture needs to share the malloc()/free() implementation
30 * between compilation units, it needs to have 30 * between compilation units, it needs to have non-local visibility.
31 */ 31 */
32 #ifndef MALLOC_VISIBLE 32 #ifndef MALLOC_VISIBLE
33 #define MALLOC_VISIBLE static 33 #define MALLOC_VISIBLE static
34 #endif 34 #endif
35 35
36 /* A trivial malloc implementation, adapted fr 36 /* A trivial malloc implementation, adapted from
37 * malloc by Hannu Savolainen 1993 and Matthi 37 * malloc by Hannu Savolainen 1993 and Matthias Urlichs 1994
38 */ 38 */
39 STATIC_RW_DATA unsigned long malloc_ptr; 39 STATIC_RW_DATA unsigned long malloc_ptr;
40 STATIC_RW_DATA int malloc_count; 40 STATIC_RW_DATA int malloc_count;
41 41
42 MALLOC_VISIBLE void *malloc(int size) 42 MALLOC_VISIBLE void *malloc(int size)
43 { 43 {
44 void *p; 44 void *p;
45 45
46 if (size < 0) 46 if (size < 0)
47 return NULL; 47 return NULL;
48 if (!malloc_ptr) 48 if (!malloc_ptr)
49 malloc_ptr = free_mem_ptr; 49 malloc_ptr = free_mem_ptr;
50 50
51 malloc_ptr = (malloc_ptr + 7) & ~7; 51 malloc_ptr = (malloc_ptr + 7) & ~7; /* Align */
52 52
53 p = (void *)malloc_ptr; 53 p = (void *)malloc_ptr;
54 malloc_ptr += size; 54 malloc_ptr += size;
55 55
56 if (free_mem_end_ptr && malloc_ptr >= 56 if (free_mem_end_ptr && malloc_ptr >= free_mem_end_ptr)
57 return NULL; 57 return NULL;
58 58
59 malloc_count++; 59 malloc_count++;
60 return p; 60 return p;
61 } 61 }
62 62
63 MALLOC_VISIBLE void free(void *where) 63 MALLOC_VISIBLE void free(void *where)
64 { 64 {
65 malloc_count--; 65 malloc_count--;
66 if (!malloc_count) 66 if (!malloc_count)
67 malloc_ptr = free_mem_ptr; 67 malloc_ptr = free_mem_ptr;
68 } 68 }
69 69
70 #define large_malloc(a) malloc(a) 70 #define large_malloc(a) malloc(a)
71 #define large_free(a) free(a) 71 #define large_free(a) free(a)
72 72
73 #define INIT 73 #define INIT
74 74
75 #else /* STATIC */ 75 #else /* STATIC */
76 76
77 /* Code active when compiled standalone for us 77 /* Code active when compiled standalone for use when loading ramdisk: */
78 78
79 #include <linux/kernel.h> 79 #include <linux/kernel.h>
80 #include <linux/fs.h> 80 #include <linux/fs.h>
81 #include <linux/string.h> 81 #include <linux/string.h>
82 #include <linux/slab.h> 82 #include <linux/slab.h>
83 #include <linux/vmalloc.h> 83 #include <linux/vmalloc.h>
84 84
85 /* Use defines rather than static inline in or 85 /* Use defines rather than static inline in order to avoid spurious
86 * warnings when not needed (indeed large_mall 86 * warnings when not needed (indeed large_malloc / large_free are not
87 * needed by inflate */ 87 * needed by inflate */
88 88
89 #define malloc(a) kmalloc(a, GFP_KERNEL) 89 #define malloc(a) kmalloc(a, GFP_KERNEL)
90 #define free(a) kfree(a) 90 #define free(a) kfree(a)
91 91
92 #define large_malloc(a) vmalloc(a) 92 #define large_malloc(a) vmalloc(a)
93 #define large_free(a) vfree(a) 93 #define large_free(a) vfree(a)
94 94
95 #define INIT __init 95 #define INIT __init
96 #define STATIC 96 #define STATIC
97 97
98 #include <linux/init.h> 98 #include <linux/init.h>
99 99
100 #endif /* STATIC */ 100 #endif /* STATIC */
101 101
102 #endif /* DECOMPR_MM_H */ 102 #endif /* DECOMPR_MM_H */
103 103
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.