1 files changed, 20 insertions, 93 deletions

diff --git a/arch/powerpc/boot/main.c b/arch/powerpc/boot/main.c
index 6f6b50d238b6..404620a9e733 100644
--- a/arch/powerpc/boot/main.c
+++ b/arch/powerpc/boot/main.c
@@ -14,8 +14,8 @@
 #include "page.h"
 #include "string.h"
 #include "stdio.h"
-#include "zlib.h"
 #include "ops.h"
+#include "gunzip_util.h"
 #include "flatdevtree.h"
 
 extern void flush_cache(void *, unsigned long);
@@ -30,6 +30,8 @@ extern char _initrd_end[];
 extern char _dtb_start[];
 extern char _dtb_end[];
 
+static struct gunzip_state gzstate;
+
 struct addr_range {
         unsigned long addr;
         unsigned long size;
@@ -42,71 +44,12 @@ static struct addr_range initrd;
 static unsigned long elfoffset;
 static int is_64bit;
 
-/* scratch space for gunzip; 46912 is from zlib_inflate_workspacesize() */
-static char scratch[46912];
 static char elfheader[256];
 
 typedef void (*kernel_entry_t)(unsigned long, unsigned long, void *);
 
 #undef DEBUG
 
-#define HEAD_CRC        2
-#define EXTRA_FIELD     4
-#define ORIG_NAME       8
-#define COMMENT         0x10
-#define RESERVED        0xe0
-
-static void gunzip(void *dst, int dstlen, unsigned char *src, int *lenp)
-{
-        z_stream s;
-        int r, i, flags;
-
-        /* skip header */
-        i = 10;
-        flags = src[3];
-        if (src[2] != Z_DEFLATED || (flags & RESERVED) != 0) {
-                printf("bad gzipped data\n\r");
-                exit();
-        }
-        if ((flags & EXTRA_FIELD) != 0)
-                i = 12 + src[10] + (src[11] << 8);
-        if ((flags & ORIG_NAME) != 0)
-                while (src[i++] != 0)
-                        ;
-        if ((flags & COMMENT) != 0)
-                while (src[i++] != 0)
-                        ;
-        if ((flags & HEAD_CRC) != 0)
-                i += 2;
-        if (i >= *lenp) {
-                printf("gunzip: ran out of data in header\n\r");
-                exit();
-        }
-
-        if (zlib_inflate_workspacesize() > sizeof(scratch)) {
-                printf("gunzip needs more mem\n");
-                exit();
-        }
-        memset(&s, 0, sizeof(s));
-        s.workspace = scratch;
-        r = zlib_inflateInit2(&s, -MAX_WBITS);
-        if (r != Z_OK) {
-                printf("inflateInit2 returned %d\n\r", r);
-                exit();
-        }
-        s.next_in = src + i;
-        s.avail_in = *lenp - i;
-        s.next_out = dst;
-        s.avail_out = dstlen;
-        r = zlib_inflate(&s, Z_FULL_FLUSH);
-        if (r != Z_OK && r != Z_STREAM_END) {
-                printf("inflate returned %d msg: %s\n\r", r, s.msg);
-                exit();
-        }
-        *lenp = s.next_out - (unsigned char *) dst;
-        zlib_inflateEnd(&s);
-}
-
 static int is_elf64(void *hdr)
 {
         Elf64_Ehdr *elf64 = hdr;
@@ -132,8 +75,8 @@ static int is_elf64(void *hdr)
                 return 0;
 
         elfoffset = (unsigned long)elf64ph->p_offset;
-        vmlinux.size = (unsigned long)elf64ph->p_filesz + elfoffset;
-        vmlinux.memsize = (unsigned long)elf64ph->p_memsz + elfoffset;
+        vmlinux.size = (unsigned long)elf64ph->p_filesz;
+        vmlinux.memsize = (unsigned long)elf64ph->p_memsz;
 
         is_64bit = 1;
         return 1;
@@ -164,8 +107,8 @@ static int is_elf32(void *hdr)
                 return 0;
 
         elfoffset = elf32ph->p_offset;
-        vmlinux.size = elf32ph->p_filesz + elf32ph->p_offset;
-        vmlinux.memsize = elf32ph->p_memsz + elf32ph->p_offset;
+        vmlinux.size = elf32ph->p_filesz;
+        vmlinux.memsize = elf32ph->p_memsz;
         return 1;
 }
 
@@ -177,13 +120,8 @@ static void prep_kernel(unsigned long a1, unsigned long a2)
         vmlinuz.size = (unsigned long)(_vmlinux_end - _vmlinux_start);
 
         /* gunzip the ELF header of the kernel */
-        if (*(unsigned short *)vmlinuz.addr == 0x1f8b) {
-                len = vmlinuz.size;
-                gunzip(elfheader, sizeof(elfheader),
-                                (unsigned char *)vmlinuz.addr, &len);
-        } else
-                memcpy(elfheader, (const void *)vmlinuz.addr,
-                       sizeof(elfheader));
+        gunzip_start(&gzstate, (void *)vmlinuz.addr, vmlinuz.size);
+        gunzip_exactly(&gzstate, elfheader, sizeof(elfheader));
 
         if (!is_elf64(elfheader) && !is_elf32(elfheader)) {
                 printf("Error: not a valid PPC32 or PPC64 ELF file!\n\r");
@@ -192,10 +130,10 @@ static void prep_kernel(unsigned long a1, unsigned long a2)
         if (platform_ops.image_hdr)
                 platform_ops.image_hdr(elfheader);
 
-        /* We need to alloc the memsize plus the file offset since gzip
-         * will expand the header (file offset), then the kernel, then
-         * possible rubbish we don't care about. But the kernel bss must
-         * be claimed (it will be zero'd by the kernel itself)
+        /* We need to alloc the memsize: gzip will expand the kernel
+         * text/data, then possible rubbish we don't care about. But
+         * the kernel bss must be claimed (it will be zero'd by the
+         * kernel itself)
          */
         printf("Allocating 0x%lx bytes for kernel ...\n\r", vmlinux.memsize);
         vmlinux.addr = (unsigned long)malloc(vmlinux.memsize);
@@ -237,24 +175,13 @@ static void prep_kernel(unsigned long a1, unsigned long a2)
         }
 
         /* Eventually gunzip the kernel */
-        if (*(unsigned short *)vmlinuz.addr == 0x1f8b) {
-                printf("gunzipping (0x%lx <- 0x%lx:0x%0lx)...",
-                       vmlinux.addr, vmlinuz.addr, vmlinuz.addr+vmlinuz.size);
-                len = vmlinuz.size;
-                gunzip((void *)vmlinux.addr, vmlinux.memsize,
-                        (unsigned char *)vmlinuz.addr, &len);
-                printf("done 0x%lx bytes\n\r", len);
-        } else {
-                memmove((void *)vmlinux.addr,(void *)vmlinuz.addr,
-                        vmlinuz.size);
-        }
-
-        /* Skip over the ELF header */
-#ifdef DEBUG
-        printf("... skipping 0x%lx bytes of ELF header\n\r",
-                        elfoffset);
-#endif
-        vmlinux.addr += elfoffset;
+        printf("gunzipping (0x%lx <- 0x%lx:0x%0lx)...",
+               vmlinux.addr, vmlinuz.addr, vmlinuz.addr+vmlinuz.size);
+        /* discard up to the actual load data */
+        gunzip_discard(&gzstate, elfoffset - sizeof(elfheader));
+        len = gunzip_finish(&gzstate, (void *)vmlinux.addr,
+                            vmlinux.memsize);
+        printf("done 0x%lx bytes\n\r", len);
 
         flush_cache((void *)vmlinux.addr, vmlinux.size);
 }