arch/arm/boot/compressed/misc.c

   1 /*
   2  * misc.c
   3  *
   4  * This is a collection of several routines from gzip-1.0.3
   5  * adapted for Linux.
   6  *
   7  * malloc by Hannu Savolainen 1993 and Matthias Urlichs 1994
   8  *
   9  * Modified for ARM Linux by Russell King
  10  *
  11  * Nicolas Pitre <nico@visuaide.com>  1999/04/14 :
  12  *  For this code to run directly from Flash, all constant variables must
  13  *  be marked with 'const' and all other variables initialized at run-time
  14  *  only.  This way all non constant variables will end up in the bss segment,
  15  *  which should point to addresses in RAM and cleared to 0 on start.
  16  *  This allows for a much quicker boot time.
  17  */
  18
  19 unsigned int __machine_arch_type;
  20
  21 #include <linux/string.h>
  22
  23 #ifdef STANDALONE_DEBUG
  24 #define putstr printf
  25 #else
  26
  27 static void putstr(const char *ptr);
  28
  29 #include <linux/compiler.h>
  30 #include <asm/arch/uncompress.h>
  31
  32 #ifdef CONFIG_DEBUG_ICEDCC
  33
  34 #ifdef CONFIG_CPU_V6
  35
  36 static void icedcc_putc(int ch)
  37 {
  38         int status, i = 0x4000000;
  39
  40         do {
  41                 if (--i < 0)
  42                         return;
  43
  44                 asm volatile ("mrc p14, 0, %0, c0, c1, 0" : "=r" (status));
  45         } while (status & (1 << 29));
  46
  47         asm("mcr p14, 0, %0, c0, c5, 0" : : "r" (ch));
  48 }
  49
  50 #else
  51
  52 static void icedcc_putc(int ch)
  53 {
  54         int status, i = 0x4000000;
  55
  56         do {
  57                 if (--i < 0)
  58                         return;
  59
  60                 asm volatile ("mrc p14, 0, %0, c0, c0, 0" : "=r" (status));
  61         } while (status & 2);
  62
  63         asm("mcr p14, 0, %0, c1, c0, 0" : : "r" (ch));
  64 }
  65
  66 #endif
  67
  68 #define putc(ch)        icedcc_putc(ch)
  69 #define flush() do { } while (0)
  70 #endif
  71
  72 static void putstr(const char *ptr)
  73 {
  74         char c;
  75
  76         while ((c = *ptr++) != '\0') {
  77                 if (c == '\n')
  78                         putc('\r');
  79                 putc(c);
  80         }
  81
  82         flush();
  83 }
  84
  85 #endif
  86
  87 #define __ptr_t void *
  88
  89 /*
  90  * Optimised C version of memzero for the ARM.
  91  */
  92 void __memzero (__ptr_t s, size_t n)
  93 {
  94         union { void *vp; unsigned long *ulp; unsigned char *ucp; } u;
  95         int i;
  96
  97         u.vp = s;
  98
  99         for (i = n >> 5; i > 0; i--) {
 100                 *u.ulp++ = 0;
 101                 *u.ulp++ = 0;
 102                 *u.ulp++ = 0;
 103                 *u.ulp++ = 0;
 104                 *u.ulp++ = 0;
 105                 *u.ulp++ = 0;
 106                 *u.ulp++ = 0;
 107                 *u.ulp++ = 0;
 108         }
 109
 110         if (n & 1 << 4) {
 111                 *u.ulp++ = 0;
 112                 *u.ulp++ = 0;
 113                 *u.ulp++ = 0;
 114                 *u.ulp++ = 0;
 115         }
 116
 117         if (n & 1 << 3) {
 118                 *u.ulp++ = 0;
 119                 *u.ulp++ = 0;
 120         }
 121
 122         if (n & 1 << 2)
 123                 *u.ulp++ = 0;
 124
 125         if (n & 1 << 1) {
 126                 *u.ucp++ = 0;
 127                 *u.ucp++ = 0;
 128         }
 129
 130         if (n & 1)
 131                 *u.ucp++ = 0;
 132 }
 133
 134 static inline __ptr_t memcpy(__ptr_t __dest, __const __ptr_t __src,
 135                             size_t __n)
 136 {
 137         int i = 0;
 138         unsigned char *d = (unsigned char *)__dest, *s = (unsigned char *)__src;
 139
 140         for (i = __n >> 3; i > 0; i--) {
 141                 *d++ = *s++;
 142                 *d++ = *s++;
 143                 *d++ = *s++;
 144                 *d++ = *s++;
 145                 *d++ = *s++;
 146                 *d++ = *s++;
 147                 *d++ = *s++;
 148                 *d++ = *s++;
 149         }
 150
 151         if (__n & 1 << 2) {
 152                 *d++ = *s++;
 153                 *d++ = *s++;
 154                 *d++ = *s++;
 155                 *d++ = *s++;
 156         }
 157
 158         if (__n & 1 << 1) {
 159                 *d++ = *s++;
 160                 *d++ = *s++;
 161         }
 162
 163         if (__n & 1)
 164                 *d++ = *s++;
 165
 166         return __dest;
 167 }
 168
 169 /*
 170  * gzip delarations
 171  */
 172 #define OF(args)  args
 173 #define STATIC static
 174
 175 typedef unsigned char  uch;
 176 typedef unsigned short ush;
 177 typedef unsigned long  ulg;
 178
 179 #define WSIZE 0x8000            /* Window size must be at least 32k, */
 180                                 /* and a power of two */
 181
 182 static uch *inbuf;              /* input buffer */
 183 static uch window[WSIZE];       /* Sliding window buffer */
 184
 185 static unsigned insize;         /* valid bytes in inbuf */
 186 static unsigned inptr;          /* index of next byte to be processed in inbuf */
 187 static unsigned outcnt;         /* bytes in output buffer */
 188
 189 /* gzip flag byte */
 190 #define ASCII_FLAG   0x01 /* bit 0 set: file probably ascii text */
 191 #define CONTINUATION 0x02 /* bit 1 set: continuation of multi-part gzip file */
 192 #define EXTRA_FIELD  0x04 /* bit 2 set: extra field present */
 193 #define ORIG_NAME    0x08 /* bit 3 set: original file name present */
 194 #define COMMENT      0x10 /* bit 4 set: file comment present */
 195 #define ENCRYPTED    0x20 /* bit 5 set: file is encrypted */
 196 #define RESERVED     0xC0 /* bit 6,7:   reserved */
 197
 198 #define get_byte()  (inptr < insize ? inbuf[inptr++] : fill_inbuf())
 199
 200 /* Diagnostic functions */
 201 #ifdef DEBUG
 202 #  define Assert(cond,msg) {if(!(cond)) error(msg);}
 203 #  define Trace(x) fprintf x
 204 #  define Tracev(x) {if (verbose) fprintf x ;}
 205 #  define Tracevv(x) {if (verbose>1) fprintf x ;}
 206 #  define Tracec(c,x) {if (verbose && (c)) fprintf x ;}
 207 #  define Tracecv(c,x) {if (verbose>1 && (c)) fprintf x ;}
 208 #else
 209 #  define Assert(cond,msg)
 210 #  define Trace(x)
 211 #  define Tracev(x)
 212 #  define Tracevv(x)
 213 #  define Tracec(c,x)
 214 #  define Tracecv(c,x)
 215 #endif
 216
 217 static int  fill_inbuf(void);
 218 static void flush_window(void);
 219 static void error(char *m);
 220 static void gzip_mark(void **);
 221 static void gzip_release(void **);
 222
 223 extern char input_data[];
 224 extern char input_data_end[];
 225
 226 static uch *output_data;
 227 static ulg output_ptr;
 228 static ulg bytes_out;
 229
 230 static void *malloc(int size);
 231 static void free(void *where);
 232 static void error(char *m);
 233 static void gzip_mark(void **);
 234 static void gzip_release(void **);
 235
 236 static void putstr(const char *);
 237
 238 extern int end;
 239 static ulg free_mem_ptr;
 240 static ulg free_mem_ptr_end;
 241
 242 #define HEAP_SIZE 0x3000
 243
 244 #include "../../../../lib/inflate.c"
 245
 246 #ifndef STANDALONE_DEBUG
 247 static void *malloc(int size)
 248 {
 249         void *p;
 250
 251         if (size <0) error("Malloc error");
 252         if (free_mem_ptr <= 0) error("Memory error");
 253
 254         free_mem_ptr = (free_mem_ptr + 3) & ~3; /* Align */
 255
 256         p = (void *)free_mem_ptr;
 257         free_mem_ptr += size;
 258
 259         if (free_mem_ptr >= free_mem_ptr_end)
 260                 error("Out of memory");
 261         return p;
 262 }
 263
 264 static void free(void *where)
 265 { /* gzip_mark & gzip_release do the free */
 266 }
 267
 268 static void gzip_mark(void **ptr)
 269 {
 270         arch_decomp_wdog();
 271         *ptr = (void *) free_mem_ptr;
 272 }
 273
 274 static void gzip_release(void **ptr)
 275 {
 276         arch_decomp_wdog();
 277         free_mem_ptr = (long) *ptr;
 278 }
 279 #else
 280 static void gzip_mark(void **ptr)
 281 {
 282 }
 283
 284 static void gzip_release(void **ptr)
 285 {
 286 }
 287 #endif
 288
 289 /* ===========================================================================
 290  * Fill the input buffer. This is called only when the buffer is empty
 291  * and at least one byte is really needed.
 292  */
 293 int fill_inbuf(void)
 294 {
 295         if (insize != 0)
 296                 error("ran out of input data");
 297
 298         inbuf = input_data;
 299         insize = &input_data_end[0] - &input_data[0];
 300
 301         inptr = 1;
 302         return inbuf[0];
 303 }
 304
 305 /* ===========================================================================
 306  * Write the output window window[0..outcnt-1] and update crc and bytes_out.
 307  * (Used for the decompressed data only.)
 308  */
 309 void flush_window(void)
 310 {
 311         ulg c = crc;
 312         unsigned n;
 313         uch *in, *out, ch;
 314
 315         in = window;
 316         out = &output_data[output_ptr];
 317         for (n = 0; n < outcnt; n++) {
 318                 ch = *out++ = *in++;
 319                 c = crc_32_tab[((int)c ^ ch) & 0xff] ^ (c >> 8);
 320         }
 321         crc = c;
 322         bytes_out += (ulg)outcnt;
 323         output_ptr += (ulg)outcnt;
 324         outcnt = 0;
 325         putstr(".");
 326 }
 327
 328 #ifndef arch_error
 329 #define arch_error(x)
 330 #endif
 331
 332 static void error(char *x)
 333 {
 334         arch_error(x);
 335
 336         putstr("\n\n");
 337         putstr(x);
 338         putstr("\n\n -- System halted");
 339
 340         while(1);       /* Halt */
 341 }
 342
 343 #ifndef STANDALONE_DEBUG
 344
 345 ulg
 346 decompress_kernel(ulg output_start, ulg free_mem_ptr_p, ulg free_mem_ptr_end_p,
 347                   int arch_id)
 348 {
 349         output_data             = (uch *)output_start;  /* Points to kernel start */
 350         free_mem_ptr            = free_mem_ptr_p;
 351         free_mem_ptr_end        = free_mem_ptr_end_p;
 352         __machine_arch_type     = arch_id;
 353
 354         arch_decomp_setup();
 355
 356         makecrc();
 357         putstr("Uncompressing Linux...");
 358         gunzip();
 359         putstr(" done, booting the kernel.\n");
 360         return output_ptr;
 361 }
 362 #else
 363
 364 char output_buffer[1500*1024];
 365
 366 int main()
 367 {
 368         output_data = output_buffer;
 369
 370         makecrc();
 371         putstr("Uncompressing Linux...");
 372         gunzip();
 373         putstr("done.\n");
 374         return 0;
 375 }
 376 #endif
 377