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