1 /* 2 * linux/compr_mm.h 3 * 4 * Memory management for pre-boot and ramdisk uncompressors 5 * 6 * Authors: Alain Knaff <[email protected]> 7 * 8 */ 9 10 #ifndef DECOMPR_MM_H 11 #define DECOMPR_MM_H 12 13 #ifdef STATIC 14 15 /* Code active when included from pre-boot environment: */ 16 17 /* A trivial malloc implementation, adapted from 18 * malloc by Hannu Savolainen 1993 and Matthias Urlichs 1994 19 */ 20 static unsigned long malloc_ptr; 21 static int malloc_count; 22 23 static void *malloc(int size) 24 { 25 void *p; 26 27 if (size < 0) 28 error("Malloc error"); 29 if (!malloc_ptr) 30 malloc_ptr = free_mem_ptr; 31 32 malloc_ptr = (malloc_ptr + 3) & ~3; /* Align */ 33 34 p = (void *)malloc_ptr; 35 malloc_ptr += size; 36 37 if (free_mem_end_ptr && malloc_ptr >= free_mem_end_ptr) 38 error("Out of memory"); 39 40 malloc_count++; 41 return p; 42 } 43 44 static void free(void *where) 45 { 46 malloc_count--; 47 if (!malloc_count) 48 malloc_ptr = free_mem_ptr; 49 } 50 51 #define large_malloc(a) malloc(a) 52 #define large_free(a) free(a) 53 54 #define set_error_fn(x) 55 56 #define INIT 57 58 #else /* STATIC */ 59 60 /* Code active when compiled standalone for use when loading ramdisk: */ 61 62 #include <linux/kernel.h> 63 #include <linux/fs.h> 64 #include <linux/string.h> 65 #include <linux/vmalloc.h> 66 67 /* Use defines rather than static inline in order to avoid spurious 68 * warnings when not needed (indeed large_malloc / large_free are not 69 * needed by inflate */ 70 71 #define malloc(a) kmalloc(a, GFP_KERNEL) 72 #define free(a) kfree(a) 73 74 #define large_malloc(a) vmalloc(a) 75 #define large_free(a) vfree(a) 76 77 static void(*error)(char *m); 78 #define set_error_fn(x) error = x; 79 80 #define INIT __init 81 #define STATIC 82 83 #include <linux/init.h> 84 85 #endif /* STATIC */ 86 87 #endif /* DECOMPR_MM_H */ 88