gzip.c 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * (C) Copyright 2012
  4. * Lei Wen <leiwen@marvell.com>, Marvell Inc.
  5. */
  6. #include <common.h>
  7. #include <watchdog.h>
  8. #include <command.h>
  9. #include <gzip.h>
  10. #include <image.h>
  11. #include <malloc.h>
  12. #include <memalign.h>
  13. #include <u-boot/zlib.h>
  14. #include "zlib/zutil.h"
  15. #ifndef CONFIG_GZIP_COMPRESS_DEF_SZ
  16. #define CONFIG_GZIP_COMPRESS_DEF_SZ 0x200
  17. #endif
  18. #define ZALLOC_ALIGNMENT 16
  19. static void *zalloc(void *x, unsigned items, unsigned size)
  20. {
  21. void *p;
  22. size *= items;
  23. size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1);
  24. p = malloc_cache_aligned(size);
  25. return (p);
  26. }
  27. static void zfree(void *x, void *addr, unsigned nb)
  28. {
  29. free (addr);
  30. }
  31. int gzip(void *dst, unsigned long *lenp,
  32. unsigned char *src, unsigned long srclen)
  33. {
  34. return zzip(dst, lenp, src, srclen, 1, NULL);
  35. }
  36. /*
  37. * Compress blocks with zlib
  38. */
  39. int zzip(void *dst, unsigned long *lenp, unsigned char *src,
  40. unsigned long srclen, int stoponerr,
  41. int (*func)(unsigned long, unsigned long))
  42. {
  43. z_stream s;
  44. int r, flush, orig, window;
  45. unsigned long comp_len, left_len;
  46. if (!srclen)
  47. return 0;
  48. #ifndef CONFIG_GZIP
  49. window = MAX_WBITS;
  50. #else
  51. window = 2 * MAX_WBITS;
  52. #endif
  53. orig = *lenp;
  54. s.zalloc = zalloc;
  55. s.zfree = zfree;
  56. s.opaque = Z_NULL;
  57. r = deflateInit2_(&s, Z_BEST_SPEED, Z_DEFLATED, window,
  58. DEF_MEM_LEVEL, Z_DEFAULT_STRATEGY,
  59. ZLIB_VERSION, sizeof(z_stream));
  60. if (r != Z_OK) {
  61. printf ("Error: deflateInit2_() returned %d\n", r);
  62. return -1;
  63. }
  64. while (srclen > 0) {
  65. comp_len = (srclen > CONFIG_GZIP_COMPRESS_DEF_SZ) ?
  66. CONFIG_GZIP_COMPRESS_DEF_SZ : srclen;
  67. s.next_in = src;
  68. s.avail_in = comp_len;
  69. flush = (srclen > CONFIG_GZIP_COMPRESS_DEF_SZ)?
  70. Z_NO_FLUSH : Z_FINISH;
  71. do {
  72. left_len = (*lenp > CONFIG_GZIP_COMPRESS_DEF_SZ) ?
  73. CONFIG_GZIP_COMPRESS_DEF_SZ : *lenp;
  74. s.next_out = dst;
  75. s.avail_out = left_len;
  76. r = deflate(&s, flush);
  77. if (r == Z_STREAM_ERROR && stoponerr == 1) {
  78. printf("Error: deflate() returned %d\n", r);
  79. r = -1;
  80. goto bail;
  81. }
  82. if (!func) {
  83. dst += (left_len - s.avail_out);
  84. *lenp -= (left_len - s.avail_out);
  85. } else if (left_len - s.avail_out > 0) {
  86. r = func((unsigned long)dst,
  87. left_len - s.avail_out);
  88. if (r < 0)
  89. goto bail;
  90. }
  91. } while (s.avail_out == 0 && (*lenp > 0));
  92. if (s.avail_in) {
  93. printf("Deflate failed to consume %u bytes", s.avail_in);
  94. r = -1;
  95. goto bail;
  96. }
  97. if (*lenp == 0) {
  98. printf("Deflate need more space to compress "
  99. "left %lu bytes\n", srclen);
  100. r = -1;
  101. goto bail;
  102. }
  103. srclen -= comp_len;
  104. src += comp_len;
  105. }
  106. r = 0;
  107. bail:
  108. deflateEnd(&s);
  109. *lenp = orig - *lenp;
  110. return r;
  111. }