gunzip.c 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120
  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 *gzalloc(void *x, unsigned items, unsigned size)
  37. {
  38. void *p;
  39. size *= items;
  40. size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1);
  41. p = malloc (size);
  42. return (p);
  43. }
  44. void gzfree(void *x, void *addr, unsigned nb)
  45. {
  46. free (addr);
  47. }
  48. int gunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp)
  49. {
  50. int i, flags;
  51. /* skip header */
  52. i = 10;
  53. flags = src[3];
  54. if (src[2] != DEFLATED || (flags & RESERVED) != 0) {
  55. puts ("Error: Bad gzipped data\n");
  56. return (-1);
  57. }
  58. if ((flags & EXTRA_FIELD) != 0)
  59. i = 12 + src[10] + (src[11] << 8);
  60. if ((flags & ORIG_NAME) != 0)
  61. while (src[i++] != 0)
  62. ;
  63. if ((flags & COMMENT) != 0)
  64. while (src[i++] != 0)
  65. ;
  66. if ((flags & HEAD_CRC) != 0)
  67. i += 2;
  68. if (i >= *lenp) {
  69. puts ("Error: gunzip out of data in header\n");
  70. return (-1);
  71. }
  72. return zunzip(dst, dstlen, src, lenp, 1, i);
  73. }
  74. /*
  75. * Uncompress blocks compressed with zlib without headers
  76. */
  77. int zunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp,
  78. int stoponerr, int offset)
  79. {
  80. z_stream s;
  81. int r;
  82. s.zalloc = gzalloc;
  83. s.zfree = gzfree;
  84. r = inflateInit2(&s, -MAX_WBITS);
  85. if (r != Z_OK) {
  86. printf ("Error: inflateInit2() returned %d\n", r);
  87. return -1;
  88. }
  89. s.next_in = src + offset;
  90. s.avail_in = *lenp - offset;
  91. s.next_out = dst;
  92. s.avail_out = dstlen;
  93. do {
  94. r = inflate(&s, Z_FINISH);
  95. if (r != Z_STREAM_END && r != Z_BUF_ERROR && stoponerr == 1) {
  96. printf("Error: inflate() returned %d\n", r);
  97. inflateEnd(&s);
  98. return -1;
  99. }
  100. s.avail_in = *lenp - offset - (int)(s.next_out - (unsigned char*)dst);
  101. s.avail_out = dstlen;
  102. } while (r == Z_BUF_ERROR);
  103. *lenp = s.next_out - (unsigned char *) dst;
  104. inflateEnd(&s);
  105. return 0;
  106. }