gzip.c 2.6 KB

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