projects / J-u-boot.git / blame
| Commit | Line | Data |
|---|---|---|
| 321359f2 MB |
1 | /* |
| 2 | * (C) Copyright 2000-2006 | |
| 3 | * Wolfgang Denk, DENX Software Engineering, [email protected]. | |
| 4 | * | |
| 5 | * See file CREDITS for list of people who contributed to this | |
| 6 | * project. | |
| 7 | * | |
| 8 | * This program is free software; you can redistribute it and/or | |
| 9 | * modify it under the terms of the GNU General Public License as | |
| 10 | * published by the Free Software Foundation; either version 2 of | |
| 11 | * the License, or (at your option) any later version. | |
| 12 | * | |
| 13 | * This program is distributed in the hope that it will be useful, | |
| 14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 16 | * GNU General Public License for more details. | |
| 17 | * | |
| 18 | * You should have received a copy of the GNU General Public License | |
| 19 | * along with this program; if not, write to the Free Software | |
| 20 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, | |
| 21 | * MA 02111-1307 USA | |
| 22 | */ | |
| 23 | ||
| 24 | #include <common.h> | |
| 25 | #include <watchdog.h> | |
| 26 | #include <command.h> | |
| 27 | #include <image.h> | |
| 28 | #include <malloc.h> | |
| 29 | #include <zlib.h> | |
| 30 | ||
| 31 | #define ZALLOC_ALIGNMENT 16 | |
| 32 | #define HEAD_CRC 2 | |
| 33 | #define EXTRA_FIELD 4 | |
| 34 | #define ORIG_NAME 8 | |
| 35 | #define COMMENT 0x10 | |
| 36 | #define RESERVED 0xe0 | |
| 37 | #define DEFLATED 8 | |
| 38 | ||
| 39 | int gunzip(void *, int, unsigned char *, unsigned long *); | |
| 40 | void *zalloc(void *, unsigned, unsigned); | |
| 41 | void zfree(void *, void *, unsigned); | |
| 42 | ||
| 43 | void *zalloc(void *x, unsigned items, unsigned size) | |
| 44 | { | |
| 45 | void *p; | |
| 46 | ||
| 47 | size *= items; | |
| 48 | size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1); | |
| 49 | ||
| 50 | p = malloc (size); | |
| 51 | ||
| 52 | return (p); | |
| 53 | } | |
| 54 | ||
| 55 | void zfree(void *x, void *addr, unsigned nb) | |
| 56 | { | |
| 57 | free (addr); | |
| 58 | } | |
| 59 | ||
| 60 | int gunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp) | |
| 61 | { | |
| 62 | z_stream s; | |
| 63 | int r, i, flags; | |
| 64 | ||
| 65 | /* skip header */ | |
| 66 | i = 10; | |
| 67 | flags = src[3]; | |
| 68 | if (src[2] != DEFLATED || (flags & RESERVED) != 0) { | |
| 69 | puts ("Error: Bad gzipped data\n"); | |
| 70 | return (-1); | |
| 71 | } | |
| 72 | if ((flags & EXTRA_FIELD) != 0) | |
| 73 | i = 12 + src[10] + (src[11] << 8); | |
| 74 | if ((flags & ORIG_NAME) != 0) | |
| 75 | while (src[i++] != 0) | |
| 76 | ; | |
| 77 | if ((flags & COMMENT) != 0) | |
| 78 | while (src[i++] != 0) | |
| 79 | ; | |
| 80 | if ((flags & HEAD_CRC) != 0) | |
| 81 | i += 2; | |
| 82 | if (i >= *lenp) { | |
| 83 | puts ("Error: gunzip out of data in header\n"); | |
| 84 | return (-1); | |
| 85 | } | |
| 86 | ||
| 87 | s.zalloc = zalloc; | |
| 88 | s.zfree = zfree; | |
| 89 | #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG) | |
| 90 | s.outcb = (cb_func)WATCHDOG_RESET; | |
| 91 | #else | |
| 92 | s.outcb = Z_NULL; | |
| 93 | #endif /* CONFIG_HW_WATCHDOG */ | |
| 94 | ||
| 95 | r = inflateInit2(&s, -MAX_WBITS); | |
| 96 | if (r != Z_OK) { | |
| 97 | printf ("Error: inflateInit2() returned %d\n", r); | |
| 98 | return (-1); | |
| 99 | } | |
| 100 | s.next_in = src + i; | |
| 101 | s.avail_in = *lenp - i; | |
| 102 | s.next_out = dst; | |
| 103 | s.avail_out = dstlen; | |
| 104 | r = inflate(&s, Z_FINISH); | |
| 105 | if (r != Z_OK && r != Z_STREAM_END) { | |
| 106 | printf ("Error: inflate() returned %d\n", r); | |
| 107 | return (-1); | |
| 108 | } | |
| 109 | *lenp = s.next_out - (unsigned char *) dst; | |
| 110 | inflateEnd(&s); | |
| 111 | ||
| 112 | return (0); | |
| 113 | } |