Fix the tgz image build
[openwrt.git] / target / linux / adm5120-2.6 / image / lzma-loader / src / decompress.c
index d361024..9e2a04c 100644 (file)
@@ -1,7 +1,10 @@
 /*
- * LZMA compressed kernel decompressor for bcm947xx boards
+ * $Id$
+ *
+ * LZMA compressed kernel decompressor for ADM5120 boards
  *
  * Copyright (C) 2005 by Oleg I. Vdovikin <oleg@cs.msu.su>
+ * Copyright (C) 2007 OpenWrt.org
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  * 24-Mar-2007 Gabor Juhos
  *   pass original values of the a0,a1,a2,a3 registers to the kernel
  *
+ * 19-May-2007 Gabor Juhos
+ *   endiannes related cleanups
+ *   add support for decompressing an embedded kernel
+ *
  */
 
+#include <stddef.h>
+
+#include "config.h"
+#include "printf.h"
 #include "LzmaDecode.h"
 
-#define BCM4710_FLASH          0x1fc00000      /* Flash */
+#define ADM5120_FLASH_START    0x1fc00000      /* Flash start */
+#define ADM5120_FLASH_END      0x1fe00000      /* Flash end */
 
 #define KSEG0                  0x80000000
 #define KSEG1                  0xa0000000
                "cache %1, (%0);\n"             \
                ".set mips0;\n"                 \
                ".set reorder\n"                \
-               :                                               \
+               :                               \
                : "r" (base),                   \
                  "i" (op));
 
+#ifdef LZMA_DEBUG
+#  define DBG(f, a...) printf(f, ## a)
+#else
+#  define DBG(f, a...) do {} while (0)
+#endif
+
 static __inline__ void blast_icache(unsigned long size, unsigned long lsize)
 {
        unsigned long start = KSEG0;
@@ -81,6 +99,7 @@ static __inline__ void blast_dcache(unsigned long size, unsigned long lsize)
 }
 
 #define TRX_MAGIC       0x30524448      /* "HDR0" */
+#define TRX_ALIGN      0x1000
 
 struct trx_header {
        unsigned int magic;             /* "HDR0" */
@@ -90,44 +109,173 @@ struct trx_header {
        unsigned int offsets[3];        /* Offsets of partitions from start of header */
 };
 
+struct env_var {
+       char    *name;
+       char    *value;
+};
+
 /* beyound the image end, size not known in advance */
 extern unsigned char workspace[];
-
-unsigned int offset;
-unsigned char *data;
+extern void board_init(void);
 
 typedef void (*kernel_entry)(unsigned long reg_a0, unsigned long reg_a1,
        unsigned long reg_a2, unsigned long reg_a3);
 
-/* flash access should be aligned, so wrapper is used */
-/* read byte from the flash, all accesses are 32-bit aligned */
-static int read_byte(void *object, unsigned char **buffer, UInt32 *bufferSize)
+static int decompress_data(unsigned char *buffer, UInt32 bufferSize,
+       int lc, int lp, int pb, unsigned char *outStream, UInt32 outSize,
+       UInt32 *outSizeProcessed);
+
+#ifdef CONFIG_PASS_KARGS
+#define ENVV(n,v)      {.name = (n), .value = (v)}
+struct env_var env_vars[] = {
+       ENVV("board_name",      CONFIG_BOARD_NAME),
+       ENVV(NULL, NULL)
+};
+#endif
+
+static void halt(void)
+{
+       printf("\nSystem halted!\n");
+       for(;;);
+}
+
+#if LZMA_WRAPPER
+extern unsigned char _lzma_data_start[];
+extern unsigned char _lzma_data_end[];
+
+unsigned char *data;
+unsigned long datalen;
+
+static __inline__ unsigned char get_byte(void)
+{
+       datalen--;
+       return *data++;
+}
+
+static void decompress_init(void)
+{
+       data = _lzma_data_start;
+       datalen = _lzma_data_end - _lzma_data_start;
+}
+
+static int decompress_data(unsigned char *buffer, UInt32 bufferSize,
+       int lc, int lp, int pb, unsigned char *outStream, UInt32 outSize,
+       UInt32 *outSizeProcessed)
 {
-       static unsigned int val;
+       return LzmaDecode(buffer, bufferSize, lc, lp, pb, data, datalen,
+               outStream, outSize, outSizeProcessed);
+}
+#endif /* LZMA_WRAPPER */
+
+#if !(LZMA_WRAPPER)
+
+#define FLASH_BANK_SIZE        (2<<20)
 
-       if (((unsigned int)offset % 4) == 0) {
-               val = *(unsigned int *)data;
-               data += 4;
+static unsigned char *flash_base = (unsigned char *) KSEG1ADDR(ADM5120_FLASH_START);
+static unsigned long flash_ofs = 0;
+static unsigned long flash_max = 0;
+static unsigned long flash_ofs_mask = (FLASH_BANK_SIZE-1);
+
+static __inline__ unsigned char get_byte(void)
+{
+       return *(flash_base+flash_ofs++);
+}
+
+static int lzma_read_byte(void *object, unsigned char **buffer, UInt32 *bufferSize)
+{
+       unsigned long len;
+
+       if (flash_ofs >= flash_max)
+               return LZMA_RESULT_DATA_ERROR;
+
+       len = flash_max-flash_ofs;
+
+#if (CONFIG_FLASH_SIZE > FLASH_BANK_SIZE)
+       if (flash_ofs < FLASH_BANK_SIZE) {
+               /* switch to bank 0 */
+               DBG("lzma_read_byte: switch to bank 0\n");
+
+               if (len > FLASH_BANK_SIZE-flash_ofs)
+                       len = FLASH_BANK_SIZE-flash_ofs;
+       } else {
+               /* switch to bank 1 */
+               DBG("lzma_read_byte: switch to bank 1\n");
        }
-       
-       *bufferSize = 1;
-       *buffer = ((unsigned char *)&val) + (offset++ & 3);
-       
+#endif
+       DBG("lzma_read_byte: ofs=%08X, len=%08X\n", flash_ofs, len);
+
+       *buffer = flash_base+(flash_ofs & flash_ofs_mask);
+       *bufferSize = len;
+       flash_ofs += len;
+
        return LZMA_RESULT_OK;
 }
 
-static __inline__ unsigned char get_byte(void)
+static ILzmaInCallback lzma_callback = {
+       .Read   = lzma_read_byte,
+};
+
+static __inline__ unsigned int read_le32(void *buf)
 {
-       unsigned char *buffer;
-       UInt32 fake;
-       
-       return read_byte(0, &buffer, &fake), *buffer;
+       unsigned char *p = buf;
+
+       return ((unsigned int)p[0] + ((unsigned int)p[1] << 8) +
+               ((unsigned int)p[2] << 16) +((unsigned int)p[3] << 24));
 }
 
+static void decompress_init(void)
+{
+       struct trx_header *hdr = NULL;
+       unsigned long kofs,klen;
+
+       printf("Looking for TRX header... ");
+       /* look for trx header, 32-bit data access */
+       for (flash_ofs = 0; flash_ofs < FLASH_BANK_SIZE; flash_ofs += TRX_ALIGN) {
+               if (read_le32(&flash_base[flash_ofs]) == TRX_MAGIC) {
+                       hdr = (struct trx_header *)&flash_base[flash_ofs];
+                       break;
+               }
+       }
+
+       if (hdr == NULL) {
+               printf("not found!\n");
+               /* no compressed kernel found, halting */
+               halt();
+       }
+
+       /* compressed kernel is in the partition 0 or 1 */
+       kofs = read_le32(&hdr->offsets[1]);
+       if (kofs == 0 || kofs > 65536) {
+               klen = kofs-read_le32(&hdr->offsets[0]);
+               kofs = read_le32(&hdr->offsets[0]);
+       } else {
+               klen = read_le32(&hdr->offsets[2]);
+               if (klen > kofs)
+                       klen -= kofs;
+               else
+                       klen = read_le32(&hdr->len)-kofs;
+       }
+
+       printf("found at %08X, kernel:%08X len:%08X\n", flash_ofs,
+               kofs, klen);
+
+       flash_ofs += kofs;
+       flash_max = flash_ofs+klen;
+}
+
+static int decompress_data(unsigned char *buffer, UInt32 bufferSize,
+       int lc, int lp, int pb, unsigned char *outStream, UInt32 outSize,
+       UInt32 *outSizeProcessed)
+{
+       return LzmaDecode(buffer, bufferSize, lc, lp, pb, &lzma_callback,
+               outStream, outSize, outSizeProcessed);
+}
+#endif /* !(LZMA_WRAPPER) */
+
 /* should be the first function */
-void entry(unsigned long reg_a0, unsigned long reg_a1,
+void decompress_entry(unsigned long reg_a0, unsigned long reg_a1,
        unsigned long reg_a2, unsigned long reg_a3,
-       unsigned long icache_size, unsigned long icache_lsize, 
+       unsigned long icache_size, unsigned long icache_lsize,
        unsigned long dcache_size, unsigned long dcache_lsize)
 {
        unsigned int i;  /* temp value */
@@ -135,23 +283,14 @@ void entry(unsigned long reg_a0, unsigned long reg_a1,
        unsigned int lp; /* literal pos state bits */
        unsigned int pb; /* pos state bits */
        unsigned int osize; /* uncompressed size */
+       int res;
 
-       ILzmaInCallback callback;
-       callback.Read = read_byte;
+       board_init();
 
-    uart_write_str("decompress kernel ... ");
-    
-       /* look for trx header, 32-bit data access */
-       for (data = ((unsigned char *) KSEG1ADDR(BCM4710_FLASH));
-               ((struct trx_header *)data)->magic != TRX_MAGIC; data += 65536);
-
-       /* compressed kernel is in the partition 0 or 1 */
-       if (((struct trx_header *)data)->offsets[1] > 65536) 
-               data += ((struct trx_header *)data)->offsets[0];
-       else
-               data += ((struct trx_header *)data)->offsets[1];
+       printf("\n\nLZMA loader for " CONFIG_BOARD_NAME
+                       ", Copyright (C) 2007 OpenWrt.org\n\n");
 
-       offset = 0;
+       decompress_init();
 
        /* lzma args */
        i = get_byte();
@@ -169,71 +308,42 @@ void entry(unsigned long reg_a0, unsigned long reg_a1,
                ((unsigned int)get_byte() << 24);
 
        /* skip rest of the header (upper half of uncompressed size) */
-       for (i = 0; i < 4; i++) 
+       for (i = 0; i < 4; i++)
                get_byte();
 
-       /* decompress kernel */
-       if (LzmaDecode(workspace, ~0, lc, lp, pb, &callback,
-               (unsigned char*)LOADADDR, osize, &i) == LZMA_RESULT_OK)
-       {
-               blast_dcache(dcache_size, dcache_lsize);
-               blast_icache(icache_size, icache_lsize);
-
-               /* Jump to load address */
-        uart_write_str("ok\r\n");
-               ((kernel_entry) LOADADDR)(reg_a0, reg_a1, reg_a2, reg_a3);
-       }
-    uart_write_str("failed\r\n");
-    while (1 );
-}
-
-/*  *********************************************************************
-    *  
-    *  ADM5120 UART driver                     File: dev_adm_uart.c
-    *  
-    *  This is a console device driver for an ADM5120 UART
-    *  
-    *********************************************************************
-    *
-    *  Copyright 2006
-    *  Compex Systems. All rights reserved.
-    *
-    ********************************************************************* */
-
-#define READCSR(r)      *(volatile UInt32 *)(0xB2600000+(r))
-#define WRITECSR(r,v)   *(volatile UInt32 *)(0xB2600000+(r)) = v
-
-#define UART_DR_REG         0x00
-#define UART_FR_REG         0x18
-#define UART_TX_FIFO_FULL   0x20
-
-int uart_write(int val)
-{
-       WRITECSR(UART_DR_REG, val);
-    while ( (READCSR(UART_FR_REG) & UART_TX_FIFO_FULL) );
-    return 0;
-}
+       printf("decompressing kernel... ");
 
-int uart_write_str(char * str)
-{
-    while ( *str != 0 ) {
-        uart_write ( *str++ );
-    }
-    return 0;
-}
-
-int uart_write_hex(int val)
-{
-    int i;
-    int tmp;
-    
-    uart_write_str("0x");
-    for ( i=0 ; i<8 ; i++ ) {
-        tmp = (val >> ((7-i) * 4 )) & 0xf;
-        tmp = tmp < 10 ? (tmp + '0') : (tmp + 'A' - 10);
-        uart_write(tmp);
-    }
-    uart_write_str("\r\n");
-    return 0;
+       /* decompress kernel */
+       res = decompress_data(workspace, ~0, lc, lp, pb,
+               (unsigned char *)LOADADDR, osize, &i);
+
+       if (res != LZMA_RESULT_OK) {
+               printf("failed, ");
+               switch (res) {
+               case LZMA_RESULT_DATA_ERROR:
+                       printf("data error!\n");
+                       break;
+               case LZMA_RESULT_NOT_ENOUGH_MEM:
+                       printf("not enough memory!\n");
+                       break;
+               default:
+                       printf("unknown error %d!\n", res);
+               }
+               halt();
+       } else
+               printf("done!\n");
+
+       blast_dcache(dcache_size, dcache_lsize);
+       blast_icache(icache_size, icache_lsize);
+
+       printf("launching kernel...\n\n");
+
+#ifdef CONFIG_PASS_KARGS
+       reg_a0 = 0;
+       reg_a1 = 0;
+       reg_a2 = (unsigned long)env_vars;
+       reg_a3 = 0;
+#endif
+       /* Jump to load address */
+       ((kernel_entry) LOADADDR)(reg_a0, reg_a1, reg_a2, reg_a3);
 }
-
This page took 0.037613 seconds and 4 git commands to generate.