gunzip.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. /*
  2. * (C) Copyright 2000-2006
  3. * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
  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. #include <common.h>
  24. #include <watchdog.h>
  25. #include <command.h>
  26. #include <image.h>
  27. #include <malloc.h>
  28. #include <u-boot/zlib.h>
  29. #define ZALLOC_ALIGNMENT 16
  30. #define HEAD_CRC 2
  31. #define EXTRA_FIELD 4
  32. #define ORIG_NAME 8
  33. #define COMMENT 0x10
  34. #define RESERVED 0xe0
  35. #define DEFLATED 8
  36. void *zalloc(void *, unsigned, unsigned);
  37. void zfree(void *, void *, unsigned);
  38. void *zalloc(void *x, unsigned items, unsigned size)
  39. {
  40. void *p;
  41. size *= items;
  42. size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1);
  43. p = malloc (size);
  44. return (p);
  45. }
  46. void zfree(void *x, void *addr, unsigned nb)
  47. {
  48. free (addr);
  49. }
  50. int gunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp)
  51. {
  52. int i, flags;
  53. /* skip header */
  54. i = 10;
  55. flags = src[3];
  56. if (src[2] != DEFLATED || (flags & RESERVED) != 0) {
  57. puts ("Error: Bad gzipped data\n");
  58. return (-1);
  59. }
  60. if ((flags & EXTRA_FIELD) != 0)
  61. i = 12 + src[10] + (src[11] << 8);
  62. if ((flags & ORIG_NAME) != 0)
  63. while (src[i++] != 0)
  64. ;
  65. if ((flags & COMMENT) != 0)
  66. while (src[i++] != 0)
  67. ;
  68. if ((flags & HEAD_CRC) != 0)
  69. i += 2;
  70. if (i >= *lenp) {
  71. puts ("Error: gunzip out of data in header\n");
  72. return (-1);
  73. }
  74. return zunzip(dst, dstlen, src, lenp, 1, i);
  75. }
  76. /*
  77. * Uncompress blocks compressed with zlib without headers
  78. */
  79. int zunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp,
  80. int stoponerr, int offset)
  81. {
  82. z_stream s;
  83. int r;
  84. s.zalloc = zalloc;
  85. s.zfree = zfree;
  86. #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
  87. s.outcb = (cb_func)WATCHDOG_RESET;
  88. #else
  89. s.outcb = Z_NULL;
  90. #endif /* CONFIG_HW_WATCHDOG */
  91. r = inflateInit2(&s, -MAX_WBITS);
  92. if (r != Z_OK) {
  93. printf ("Error: inflateInit2() returned %d\n", r);
  94. return -1;
  95. }
  96. s.next_in = src + offset;
  97. s.avail_in = *lenp - offset;
  98. s.next_out = dst;
  99. s.avail_out = dstlen;
  100. r = inflate(&s, Z_FINISH);
  101. if ((r != Z_STREAM_END) && (stoponerr==1)) {
  102. printf ("Error: inflate() returned %d\n", r);
  103. inflateEnd(&s);
  104. return (-1);
  105. }
  106. *lenp = s.next_out - (unsigned char *) dst;
  107. inflateEnd(&s);
  108. return 0;
  109. }