diff options
Diffstat (limited to 'utils/jztool/src/ucl_unpack.c')
-rw-r--r-- | utils/jztool/src/ucl_unpack.c | 128 |
1 files changed, 128 insertions, 0 deletions
diff --git a/utils/jztool/src/ucl_unpack.c b/utils/jztool/src/ucl_unpack.c new file mode 100644 index 0000000000..3b199c7008 --- /dev/null +++ b/utils/jztool/src/ucl_unpack.c | |||
@@ -0,0 +1,128 @@ | |||
1 | /*************************************************************************** | ||
2 | * __________ __ ___. | ||
3 | * Open \______ \ ____ ____ | | _\_ |__ _______ ___ | ||
4 | * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / | ||
5 | * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < | ||
6 | * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ | ||
7 | * \/ \/ \/ \/ \/ | ||
8 | * $Id$ | ||
9 | * | ||
10 | * Copyright (C) 2021 Aidan MacDonald | ||
11 | * | ||
12 | * This program is free software; you can redistribute it and/or | ||
13 | * modify it under the terms of the GNU General Public License | ||
14 | * as published by the Free Software Foundation; either version 2 | ||
15 | * of the License, or (at your option) any later version. | ||
16 | * | ||
17 | * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY | ||
18 | * KIND, either express or implied. | ||
19 | * | ||
20 | ****************************************************************************/ | ||
21 | |||
22 | #include "jztool.h" | ||
23 | #include "ucl/ucl.h" | ||
24 | |||
25 | static uint32_t xread32(const uint8_t* d) | ||
26 | { | ||
27 | uint32_t r = 0; | ||
28 | r |= d[0] << 24; | ||
29 | r |= d[1] << 16; | ||
30 | r |= d[2] << 8; | ||
31 | r |= d[3] << 0; | ||
32 | return r; | ||
33 | } | ||
34 | |||
35 | /* adapted from firmware/common/ucl_decompress.c */ | ||
36 | jz_buffer* jz_ucl_unpack(const uint8_t* src, uint32_t src_len, uint32_t* dst_len) | ||
37 | { | ||
38 | static const uint8_t magic[8] = | ||
39 | {0x00, 0xe9, 0x55, 0x43, 0x4c, 0xff, 0x01, 0x1a}; | ||
40 | |||
41 | jz_buffer* buffer = NULL; | ||
42 | |||
43 | /* make sure there are enough bytes for the header */ | ||
44 | if(src_len < 18) | ||
45 | goto error; | ||
46 | |||
47 | /* avoid memcmp for reasons of code size */ | ||
48 | for(size_t i = 0; i < sizeof(magic); ++i) | ||
49 | if(src[i] != magic[i]) | ||
50 | goto error; | ||
51 | |||
52 | /* read the other header fields */ | ||
53 | /* uint32_t flags = xread32(&src[8]); */ | ||
54 | uint8_t method = src[12]; | ||
55 | /* uint8_t level = src[13]; */ | ||
56 | uint32_t block_size = xread32(&src[14]); | ||
57 | |||
58 | /* check supported compression method */ | ||
59 | if(method != 0x2e) | ||
60 | goto error; | ||
61 | |||
62 | /* validate */ | ||
63 | if(block_size < 1024 || block_size > 8*1024*1024) | ||
64 | goto error; | ||
65 | |||
66 | src += 18; | ||
67 | src_len -= 18; | ||
68 | |||
69 | /* Calculate amount of space that we might need & allocate a buffer: | ||
70 | * - subtract 4 to account for end of file marker | ||
71 | * - each block is block_size bytes + 8 bytes of header | ||
72 | * - add one to nr_blocks to account for case where file size < block size | ||
73 | * - total size = max uncompressed size of block * nr_blocks | ||
74 | */ | ||
75 | uint32_t nr_blocks = (src_len - 4) / (8 + block_size) + 1; | ||
76 | uint32_t max_size = nr_blocks * (block_size + block_size/8 + 256); | ||
77 | buffer = jz_buffer_alloc(max_size, NULL); | ||
78 | if(!buffer) | ||
79 | goto error; | ||
80 | |||
81 | /* perform the decompression */ | ||
82 | uint32_t dst_ilen = buffer->size; | ||
83 | uint8_t* dst = buffer->data; | ||
84 | while(1) { | ||
85 | if(src_len < 4) | ||
86 | goto error; | ||
87 | |||
88 | uint32_t out_len = xread32(src); src += 4, src_len -= 4; | ||
89 | if(out_len == 0) | ||
90 | break; | ||
91 | |||
92 | if(src_len < 4) | ||
93 | goto error; | ||
94 | |||
95 | uint32_t in_len = xread32(src); src += 4, src_len -= 4; | ||
96 | if(in_len > block_size || out_len > block_size || | ||
97 | in_len == 0 || in_len > out_len) | ||
98 | goto error; | ||
99 | |||
100 | if(src_len < in_len) | ||
101 | goto error; | ||
102 | |||
103 | if(in_len < out_len) { | ||
104 | uint32_t actual_out_len = dst_ilen; | ||
105 | int rc = ucl_nrv2e_decompress_safe_8(src, in_len, dst, &actual_out_len, NULL); | ||
106 | if(rc != UCL_E_OK) | ||
107 | goto error; | ||
108 | if(actual_out_len != out_len) | ||
109 | goto error; | ||
110 | } else { | ||
111 | for(size_t i = 0; i < in_len; ++i) | ||
112 | dst[i] = src[i]; | ||
113 | } | ||
114 | |||
115 | src += in_len; | ||
116 | src_len -= in_len; | ||
117 | dst += out_len; | ||
118 | dst_ilen -= out_len; | ||
119 | } | ||
120 | |||
121 | /* subtract leftover number of bytes to get size of compressed output */ | ||
122 | *dst_len = buffer->size - dst_ilen; | ||
123 | return buffer; | ||
124 | |||
125 | error: | ||
126 | jz_buffer_free(buffer); | ||
127 | return NULL; | ||
128 | } | ||