Greg Kroah-Hartman | b244131 | 2017-11-01 15:07:57 +0100 | [diff] [blame] | 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 2 | /* |
| 3 | * linux/compr_mm.h |
| 4 | * |
| 5 | * Memory management for pre-boot and ramdisk uncompressors |
| 6 | * |
| 7 | * Authors: Alain Knaff <alain@knaff.lu> |
| 8 | * |
| 9 | */ |
| 10 | |
| 11 | #ifndef DECOMPR_MM_H |
| 12 | #define DECOMPR_MM_H |
| 13 | |
| 14 | #ifdef STATIC |
| 15 | |
| 16 | /* Code active when included from pre-boot environment: */ |
| 17 | |
Russell King | 5ceaa2f | 2010-03-10 15:23:53 -0800 | [diff] [blame] | 18 | /* |
| 19 | * Some architectures want to ensure there is no local data in their |
Lucas De Marchi | 25985ed | 2011-03-30 22:57:33 -0300 | [diff] [blame] | 20 | * pre-boot environment, so that data can arbitrarily relocated (via |
Russell King | 5ceaa2f | 2010-03-10 15:23:53 -0800 | [diff] [blame] | 21 | * GOT references). This is achieved by defining STATIC_RW_DATA to |
| 22 | * be null. |
| 23 | */ |
| 24 | #ifndef STATIC_RW_DATA |
| 25 | #define STATIC_RW_DATA static |
| 26 | #endif |
| 27 | |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 28 | /* A trivial malloc implementation, adapted from |
| 29 | * malloc by Hannu Savolainen 1993 and Matthias Urlichs 1994 |
| 30 | */ |
Russell King | 5ceaa2f | 2010-03-10 15:23:53 -0800 | [diff] [blame] | 31 | STATIC_RW_DATA unsigned long malloc_ptr; |
| 32 | STATIC_RW_DATA int malloc_count; |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 33 | |
| 34 | static void *malloc(int size) |
| 35 | { |
| 36 | void *p; |
| 37 | |
| 38 | if (size < 0) |
Phillip Lougher | c1e7c3ae | 2009-12-14 21:45:19 +0000 | [diff] [blame] | 39 | return NULL; |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 40 | if (!malloc_ptr) |
| 41 | malloc_ptr = free_mem_ptr; |
| 42 | |
| 43 | malloc_ptr = (malloc_ptr + 3) & ~3; /* Align */ |
| 44 | |
| 45 | p = (void *)malloc_ptr; |
| 46 | malloc_ptr += size; |
| 47 | |
| 48 | if (free_mem_end_ptr && malloc_ptr >= free_mem_end_ptr) |
Phillip Lougher | c1e7c3ae | 2009-12-14 21:45:19 +0000 | [diff] [blame] | 49 | return NULL; |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 50 | |
| 51 | malloc_count++; |
| 52 | return p; |
| 53 | } |
| 54 | |
| 55 | static void free(void *where) |
| 56 | { |
| 57 | malloc_count--; |
| 58 | if (!malloc_count) |
| 59 | malloc_ptr = free_mem_ptr; |
| 60 | } |
| 61 | |
| 62 | #define large_malloc(a) malloc(a) |
| 63 | #define large_free(a) free(a) |
| 64 | |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 65 | #define INIT |
| 66 | |
| 67 | #else /* STATIC */ |
| 68 | |
| 69 | /* Code active when compiled standalone for use when loading ramdisk: */ |
| 70 | |
| 71 | #include <linux/kernel.h> |
| 72 | #include <linux/fs.h> |
| 73 | #include <linux/string.h> |
Lasse Collin | 2b6b5ca | 2011-01-12 17:01:15 -0800 | [diff] [blame] | 74 | #include <linux/slab.h> |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 75 | #include <linux/vmalloc.h> |
| 76 | |
| 77 | /* Use defines rather than static inline in order to avoid spurious |
| 78 | * warnings when not needed (indeed large_malloc / large_free are not |
| 79 | * needed by inflate */ |
| 80 | |
| 81 | #define malloc(a) kmalloc(a, GFP_KERNEL) |
| 82 | #define free(a) kfree(a) |
| 83 | |
| 84 | #define large_malloc(a) vmalloc(a) |
| 85 | #define large_free(a) vfree(a) |
| 86 | |
Alain Knaff | bc22c17 | 2009-01-04 22:46:16 +0100 | [diff] [blame] | 87 | #define INIT __init |
| 88 | #define STATIC |
| 89 | |
| 90 | #include <linux/init.h> |
| 91 | |
| 92 | #endif /* STATIC */ |
| 93 | |
| 94 | #endif /* DECOMPR_MM_H */ |