1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
278acc472SPeter Tyser /*
378acc472SPeter Tyser * (C) Copyright 2000-2006
478acc472SPeter Tyser * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
578acc472SPeter Tyser */
678acc472SPeter Tyser
778acc472SPeter Tyser #include <common.h>
878acc472SPeter Tyser #include <watchdog.h>
978acc472SPeter Tyser #include <command.h>
1024b852a7SSimon Glass #include <console.h>
1178acc472SPeter Tyser #include <image.h>
1278acc472SPeter Tyser #include <malloc.h>
13d025021eSClemens Gruber #include <memalign.h>
1478acc472SPeter Tyser #include <u-boot/zlib.h>
15918e9ebbSEric Nelson #include <div64.h>
1678acc472SPeter Tyser
17918e9ebbSEric Nelson #define HEADER0 '\x1f'
18918e9ebbSEric Nelson #define HEADER1 '\x8b'
1978acc472SPeter Tyser #define ZALLOC_ALIGNMENT 16
2078acc472SPeter Tyser #define HEAD_CRC 2
2178acc472SPeter Tyser #define EXTRA_FIELD 4
2278acc472SPeter Tyser #define ORIG_NAME 8
2378acc472SPeter Tyser #define COMMENT 0x10
2478acc472SPeter Tyser #define RESERVED 0xe0
2578acc472SPeter Tyser #define DEFLATED 8
2678acc472SPeter Tyser
gzalloc(void * x,unsigned items,unsigned size)27e3ed0575SMike Frysinger void *gzalloc(void *x, unsigned items, unsigned size)
2878acc472SPeter Tyser {
2978acc472SPeter Tyser void *p;
3078acc472SPeter Tyser
3178acc472SPeter Tyser size *= items;
3278acc472SPeter Tyser size = (size + ZALLOC_ALIGNMENT - 1) & ~(ZALLOC_ALIGNMENT - 1);
3378acc472SPeter Tyser
3478acc472SPeter Tyser p = malloc (size);
3578acc472SPeter Tyser
3678acc472SPeter Tyser return (p);
3778acc472SPeter Tyser }
3878acc472SPeter Tyser
gzfree(void * x,void * addr,unsigned nb)39e3ed0575SMike Frysinger void gzfree(void *x, void *addr, unsigned nb)
4078acc472SPeter Tyser {
4178acc472SPeter Tyser free (addr);
4278acc472SPeter Tyser }
4378acc472SPeter Tyser
gzip_parse_header(const unsigned char * src,unsigned long len)44376ddf9dSJean-Jacques Hiblot int gzip_parse_header(const unsigned char *src, unsigned long len)
4578acc472SPeter Tyser {
4678acc472SPeter Tyser int i, flags;
4778acc472SPeter Tyser
4878acc472SPeter Tyser /* skip header */
4978acc472SPeter Tyser i = 10;
5078acc472SPeter Tyser flags = src[3];
5178acc472SPeter Tyser if (src[2] != DEFLATED || (flags & RESERVED) != 0) {
5278acc472SPeter Tyser puts ("Error: Bad gzipped data\n");
5378acc472SPeter Tyser return (-1);
5478acc472SPeter Tyser }
5578acc472SPeter Tyser if ((flags & EXTRA_FIELD) != 0)
5678acc472SPeter Tyser i = 12 + src[10] + (src[11] << 8);
5778acc472SPeter Tyser if ((flags & ORIG_NAME) != 0)
5878acc472SPeter Tyser while (src[i++] != 0)
5978acc472SPeter Tyser ;
6078acc472SPeter Tyser if ((flags & COMMENT) != 0)
6178acc472SPeter Tyser while (src[i++] != 0)
6278acc472SPeter Tyser ;
6378acc472SPeter Tyser if ((flags & HEAD_CRC) != 0)
6478acc472SPeter Tyser i += 2;
65376ddf9dSJean-Jacques Hiblot if (i >= len) {
6678acc472SPeter Tyser puts ("Error: gunzip out of data in header\n");
6778acc472SPeter Tyser return (-1);
6878acc472SPeter Tyser }
69376ddf9dSJean-Jacques Hiblot return i;
70376ddf9dSJean-Jacques Hiblot }
7178acc472SPeter Tyser
gunzip(void * dst,int dstlen,unsigned char * src,unsigned long * lenp)72376ddf9dSJean-Jacques Hiblot int gunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp)
73376ddf9dSJean-Jacques Hiblot {
74376ddf9dSJean-Jacques Hiblot int offset = gzip_parse_header(src, *lenp);
75376ddf9dSJean-Jacques Hiblot
76376ddf9dSJean-Jacques Hiblot if (offset < 0)
77376ddf9dSJean-Jacques Hiblot return offset;
78376ddf9dSJean-Jacques Hiblot
79376ddf9dSJean-Jacques Hiblot return zunzip(dst, dstlen, src, lenp, 1, offset);
8078acc472SPeter Tyser }
8178acc472SPeter Tyser
821811b7d3STom Rini #ifdef CONFIG_CMD_UNZIP
83918e9ebbSEric Nelson __weak
gzwrite_progress_init(u64 expectedsize)84918e9ebbSEric Nelson void gzwrite_progress_init(u64 expectedsize)
85918e9ebbSEric Nelson {
86918e9ebbSEric Nelson putc('\n');
87918e9ebbSEric Nelson }
88918e9ebbSEric Nelson
89918e9ebbSEric Nelson __weak
gzwrite_progress(int iteration,u64 bytes_written,u64 total_bytes)90918e9ebbSEric Nelson void gzwrite_progress(int iteration,
91918e9ebbSEric Nelson u64 bytes_written,
92918e9ebbSEric Nelson u64 total_bytes)
93918e9ebbSEric Nelson {
94918e9ebbSEric Nelson if (0 == (iteration & 3))
95918e9ebbSEric Nelson printf("%llu/%llu\r", bytes_written, total_bytes);
96918e9ebbSEric Nelson }
97918e9ebbSEric Nelson
98918e9ebbSEric Nelson __weak
gzwrite_progress_finish(int returnval,u64 bytes_written,u64 total_bytes,u32 expected_crc,u32 calculated_crc)99918e9ebbSEric Nelson void gzwrite_progress_finish(int returnval,
100918e9ebbSEric Nelson u64 bytes_written,
101918e9ebbSEric Nelson u64 total_bytes,
102918e9ebbSEric Nelson u32 expected_crc,
103918e9ebbSEric Nelson u32 calculated_crc)
104918e9ebbSEric Nelson {
105918e9ebbSEric Nelson if (0 == returnval) {
106918e9ebbSEric Nelson printf("\n\t%llu bytes, crc 0x%08x\n",
107918e9ebbSEric Nelson total_bytes, calculated_crc);
108918e9ebbSEric Nelson } else {
109918e9ebbSEric Nelson printf("\n\tuncompressed %llu of %llu\n"
110918e9ebbSEric Nelson "\tcrcs == 0x%08x/0x%08x\n",
111918e9ebbSEric Nelson bytes_written, total_bytes,
112918e9ebbSEric Nelson expected_crc, calculated_crc);
113918e9ebbSEric Nelson }
114918e9ebbSEric Nelson }
115918e9ebbSEric Nelson
gzwrite(unsigned char * src,int len,struct blk_desc * dev,unsigned long szwritebuf,u64 startoffs,u64 szexpected)116918e9ebbSEric Nelson int gzwrite(unsigned char *src, int len,
1174101f687SSimon Glass struct blk_desc *dev,
118918e9ebbSEric Nelson unsigned long szwritebuf,
119918e9ebbSEric Nelson u64 startoffs,
120918e9ebbSEric Nelson u64 szexpected)
121918e9ebbSEric Nelson {
122918e9ebbSEric Nelson int i, flags;
123918e9ebbSEric Nelson z_stream s;
124918e9ebbSEric Nelson int r = 0;
125918e9ebbSEric Nelson unsigned char *writebuf;
126918e9ebbSEric Nelson unsigned crc = 0;
127918e9ebbSEric Nelson u64 totalfilled = 0;
128918e9ebbSEric Nelson lbaint_t blksperbuf, outblock;
129918e9ebbSEric Nelson u32 expected_crc;
130918e9ebbSEric Nelson u32 payload_size;
131918e9ebbSEric Nelson int iteration = 0;
132918e9ebbSEric Nelson
133918e9ebbSEric Nelson if (!szwritebuf ||
134918e9ebbSEric Nelson (szwritebuf % dev->blksz) ||
135918e9ebbSEric Nelson (szwritebuf < dev->blksz)) {
136918e9ebbSEric Nelson printf("%s: size %lu not a multiple of %lu\n",
137918e9ebbSEric Nelson __func__, szwritebuf, dev->blksz);
138918e9ebbSEric Nelson return -1;
139918e9ebbSEric Nelson }
140918e9ebbSEric Nelson
141918e9ebbSEric Nelson if (startoffs & (dev->blksz-1)) {
142918e9ebbSEric Nelson printf("%s: start offset %llu not a multiple of %lu\n",
143918e9ebbSEric Nelson __func__, startoffs, dev->blksz);
144918e9ebbSEric Nelson return -1;
145918e9ebbSEric Nelson }
146918e9ebbSEric Nelson
147918e9ebbSEric Nelson blksperbuf = szwritebuf / dev->blksz;
148918e9ebbSEric Nelson outblock = lldiv(startoffs, dev->blksz);
149918e9ebbSEric Nelson
150918e9ebbSEric Nelson /* skip header */
151918e9ebbSEric Nelson i = 10;
152918e9ebbSEric Nelson flags = src[3];
153918e9ebbSEric Nelson if (src[2] != DEFLATED || (flags & RESERVED) != 0) {
154918e9ebbSEric Nelson puts("Error: Bad gzipped data\n");
155918e9ebbSEric Nelson return -1;
156918e9ebbSEric Nelson }
157918e9ebbSEric Nelson if ((flags & EXTRA_FIELD) != 0)
158918e9ebbSEric Nelson i = 12 + src[10] + (src[11] << 8);
159918e9ebbSEric Nelson if ((flags & ORIG_NAME) != 0)
160918e9ebbSEric Nelson while (src[i++] != 0)
161918e9ebbSEric Nelson ;
162918e9ebbSEric Nelson if ((flags & COMMENT) != 0)
163918e9ebbSEric Nelson while (src[i++] != 0)
164918e9ebbSEric Nelson ;
165918e9ebbSEric Nelson if ((flags & HEAD_CRC) != 0)
166918e9ebbSEric Nelson i += 2;
167918e9ebbSEric Nelson
168918e9ebbSEric Nelson if (i >= len-8) {
169918e9ebbSEric Nelson puts("Error: gunzip out of data in header");
170918e9ebbSEric Nelson return -1;
171918e9ebbSEric Nelson }
172918e9ebbSEric Nelson
173918e9ebbSEric Nelson payload_size = len - i - 8;
174918e9ebbSEric Nelson
175918e9ebbSEric Nelson memcpy(&expected_crc, src + len - 8, sizeof(expected_crc));
176918e9ebbSEric Nelson expected_crc = le32_to_cpu(expected_crc);
177918e9ebbSEric Nelson u32 szuncompressed;
178918e9ebbSEric Nelson memcpy(&szuncompressed, src + len - 4, sizeof(szuncompressed));
179918e9ebbSEric Nelson if (szexpected == 0) {
180918e9ebbSEric Nelson szexpected = le32_to_cpu(szuncompressed);
181918e9ebbSEric Nelson } else if (szuncompressed != (u32)szexpected) {
182918e9ebbSEric Nelson printf("size of %llx doesn't match trailer low bits %x\n",
183918e9ebbSEric Nelson szexpected, szuncompressed);
184918e9ebbSEric Nelson return -1;
185918e9ebbSEric Nelson }
186918e9ebbSEric Nelson if (lldiv(szexpected, dev->blksz) > (dev->lba - outblock)) {
187918e9ebbSEric Nelson printf("%s: uncompressed size %llu exceeds device size\n",
188918e9ebbSEric Nelson __func__, szexpected);
189918e9ebbSEric Nelson return -1;
190918e9ebbSEric Nelson }
191918e9ebbSEric Nelson
192918e9ebbSEric Nelson gzwrite_progress_init(szexpected);
193918e9ebbSEric Nelson
194918e9ebbSEric Nelson s.zalloc = gzalloc;
195918e9ebbSEric Nelson s.zfree = gzfree;
196918e9ebbSEric Nelson
197918e9ebbSEric Nelson r = inflateInit2(&s, -MAX_WBITS);
198918e9ebbSEric Nelson if (r != Z_OK) {
199918e9ebbSEric Nelson printf("Error: inflateInit2() returned %d\n", r);
200918e9ebbSEric Nelson return -1;
201918e9ebbSEric Nelson }
202918e9ebbSEric Nelson
203918e9ebbSEric Nelson s.next_in = src + i;
204918e9ebbSEric Nelson s.avail_in = payload_size+8;
205d025021eSClemens Gruber writebuf = (unsigned char *)malloc_cache_aligned(szwritebuf);
206918e9ebbSEric Nelson
207918e9ebbSEric Nelson /* decompress until deflate stream ends or end of file */
208918e9ebbSEric Nelson do {
209918e9ebbSEric Nelson if (s.avail_in == 0) {
210918e9ebbSEric Nelson printf("%s: weird termination with result %d\n",
211918e9ebbSEric Nelson __func__, r);
212918e9ebbSEric Nelson break;
213918e9ebbSEric Nelson }
214918e9ebbSEric Nelson
215918e9ebbSEric Nelson /* run inflate() on input until output buffer not full */
216918e9ebbSEric Nelson do {
217918e9ebbSEric Nelson unsigned long blocks_written;
218918e9ebbSEric Nelson int numfilled;
219918e9ebbSEric Nelson lbaint_t writeblocks;
220918e9ebbSEric Nelson
221918e9ebbSEric Nelson s.avail_out = szwritebuf;
222918e9ebbSEric Nelson s.next_out = writebuf;
223918e9ebbSEric Nelson r = inflate(&s, Z_SYNC_FLUSH);
224918e9ebbSEric Nelson if ((r != Z_OK) &&
225918e9ebbSEric Nelson (r != Z_STREAM_END)) {
226918e9ebbSEric Nelson printf("Error: inflate() returned %d\n", r);
227918e9ebbSEric Nelson goto out;
228918e9ebbSEric Nelson }
229918e9ebbSEric Nelson numfilled = szwritebuf - s.avail_out;
230918e9ebbSEric Nelson crc = crc32(crc, writebuf, numfilled);
231918e9ebbSEric Nelson totalfilled += numfilled;
232918e9ebbSEric Nelson if (numfilled < szwritebuf) {
233918e9ebbSEric Nelson writeblocks = (numfilled+dev->blksz-1)
234918e9ebbSEric Nelson / dev->blksz;
235918e9ebbSEric Nelson memset(writebuf+numfilled, 0,
236918e9ebbSEric Nelson dev->blksz-(numfilled%dev->blksz));
237918e9ebbSEric Nelson } else {
238918e9ebbSEric Nelson writeblocks = blksperbuf;
239918e9ebbSEric Nelson }
240918e9ebbSEric Nelson
241918e9ebbSEric Nelson gzwrite_progress(iteration++,
242918e9ebbSEric Nelson totalfilled,
243918e9ebbSEric Nelson szexpected);
2446a3bf3e5SEric Nelson blocks_written = blk_dwrite(dev, outblock,
2456a3bf3e5SEric Nelson writeblocks, writebuf);
246918e9ebbSEric Nelson outblock += blocks_written;
247918e9ebbSEric Nelson if (ctrlc()) {
248918e9ebbSEric Nelson puts("abort\n");
249918e9ebbSEric Nelson goto out;
250918e9ebbSEric Nelson }
251918e9ebbSEric Nelson WATCHDOG_RESET();
252918e9ebbSEric Nelson } while (s.avail_out == 0);
253918e9ebbSEric Nelson /* done when inflate() says it's done */
254918e9ebbSEric Nelson } while (r != Z_STREAM_END);
255918e9ebbSEric Nelson
256918e9ebbSEric Nelson if ((szexpected != totalfilled) ||
257918e9ebbSEric Nelson (crc != expected_crc))
258918e9ebbSEric Nelson r = -1;
259918e9ebbSEric Nelson else
260918e9ebbSEric Nelson r = 0;
261918e9ebbSEric Nelson
262918e9ebbSEric Nelson out:
263918e9ebbSEric Nelson gzwrite_progress_finish(r, totalfilled, szexpected,
264918e9ebbSEric Nelson expected_crc, crc);
265918e9ebbSEric Nelson free(writebuf);
266918e9ebbSEric Nelson inflateEnd(&s);
267918e9ebbSEric Nelson
268918e9ebbSEric Nelson return r;
269918e9ebbSEric Nelson }
2701811b7d3STom Rini #endif
271918e9ebbSEric Nelson
27278acc472SPeter Tyser /*
27378acc472SPeter Tyser * Uncompress blocks compressed with zlib without headers
27478acc472SPeter Tyser */
zunzip(void * dst,int dstlen,unsigned char * src,unsigned long * lenp,int stoponerr,int offset)27578acc472SPeter Tyser int zunzip(void *dst, int dstlen, unsigned char *src, unsigned long *lenp,
27678acc472SPeter Tyser int stoponerr, int offset)
27778acc472SPeter Tyser {
27878acc472SPeter Tyser z_stream s;
2799c55c54fSSimon Glass int err = 0;
28078acc472SPeter Tyser int r;
28178acc472SPeter Tyser
282e3ed0575SMike Frysinger s.zalloc = gzalloc;
283e3ed0575SMike Frysinger s.zfree = gzfree;
28478acc472SPeter Tyser
28578acc472SPeter Tyser r = inflateInit2(&s, -MAX_WBITS);
28678acc472SPeter Tyser if (r != Z_OK) {
28778acc472SPeter Tyser printf("Error: inflateInit2() returned %d\n", r);
28878acc472SPeter Tyser return -1;
28978acc472SPeter Tyser }
29078acc472SPeter Tyser s.next_in = src + offset;
29178acc472SPeter Tyser s.avail_in = *lenp - offset;
29278acc472SPeter Tyser s.next_out = dst;
29378acc472SPeter Tyser s.avail_out = dstlen;
294f039ada5SCatalin Radu do {
29578acc472SPeter Tyser r = inflate(&s, Z_FINISH);
296b75650d8SKees Cook if (stoponerr == 1 && r != Z_STREAM_END &&
29719346657SStephen Warren (s.avail_in == 0 || s.avail_out == 0 || r != Z_BUF_ERROR)) {
29878acc472SPeter Tyser printf("Error: inflate() returned %d\n", r);
2999c55c54fSSimon Glass err = -1;
3009c55c54fSSimon Glass break;
30178acc472SPeter Tyser }
302f039ada5SCatalin Radu } while (r == Z_BUF_ERROR);
30378acc472SPeter Tyser *lenp = s.next_out - (unsigned char *) dst;
30478acc472SPeter Tyser inflateEnd(&s);
30578acc472SPeter Tyser
3069c55c54fSSimon Glass return err;
30778acc472SPeter Tyser }
308