include/linux/decompress/mm.h

   1 /*
   2  * linux/compr_mm.h
   3  *
   4  * Memory management for pre-boot and ramdisk uncompressors
   5  *
   6  * Authors: Alain Knaff <alain@knaff.lu>
   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 */