Commit | Line | Data |
---|---|---|
d2912cb1 | 1 | // SPDX-License-Identifier: GPL-2.0-only |
e76e1fdf KL |
2 | /* |
3 | * Wrapper for decompressing LZ4-compressed kernel, initramfs, and initrd | |
4 | * | |
5 | * Copyright (C) 2013, LG Electronics, Kyungsik Lee <kyungsik.lee@lge.com> | |
e76e1fdf KL |
6 | */ |
7 | ||
8 | #ifdef STATIC | |
9 | #define PREBOOT | |
10 | #include "lz4/lz4_decompress.c" | |
11 | #else | |
12 | #include <linux/decompress/unlz4.h> | |
13 | #endif | |
14 | #include <linux/types.h> | |
15 | #include <linux/lz4.h> | |
16 | #include <linux/decompress/mm.h> | |
17 | #include <linux/compiler.h> | |
18 | ||
19 | #include <asm/unaligned.h> | |
20 | ||
21 | /* | |
22 | * Note: Uncompressed chunk size is used in the compressor side | |
23 | * (userspace side for compression). | |
24 | * It is hardcoded because there is not proper way to extract it | |
25 | * from the binary stream which is generated by the preliminary | |
26 | * version of LZ4 tool so far. | |
27 | */ | |
28 | #define LZ4_DEFAULT_UNCOMPRESSED_CHUNK_SIZE (8 << 20) | |
29 | #define ARCHIVE_MAGICNUMBER 0x184C2102 | |
30 | ||
d97b07c5 YL |
31 | STATIC inline int INIT unlz4(u8 *input, long in_len, |
32 | long (*fill)(void *, unsigned long), | |
33 | long (*flush)(void *, unsigned long), | |
34 | u8 *output, long *posp, | |
e76e1fdf KL |
35 | void (*error) (char *x)) |
36 | { | |
37 | int ret = -1; | |
38 | size_t chunksize = 0; | |
39 | size_t uncomp_chunksize = LZ4_DEFAULT_UNCOMPRESSED_CHUNK_SIZE; | |
40 | u8 *inp; | |
41 | u8 *inp_start; | |
42 | u8 *outp; | |
d97b07c5 | 43 | long size = in_len; |
e76e1fdf KL |
44 | #ifdef PREBOOT |
45 | size_t out_len = get_unaligned_le32(input + in_len); | |
46 | #endif | |
47 | size_t dest_len; | |
48 | ||
49 | ||
50 | if (output) { | |
51 | outp = output; | |
52 | } else if (!flush) { | |
53 | error("NULL output pointer and no flush function provided"); | |
54 | goto exit_0; | |
55 | } else { | |
56 | outp = large_malloc(uncomp_chunksize); | |
57 | if (!outp) { | |
58 | error("Could not allocate output buffer"); | |
59 | goto exit_0; | |
60 | } | |
61 | } | |
62 | ||
63 | if (input && fill) { | |
64 | error("Both input pointer and fill function provided,"); | |
65 | goto exit_1; | |
66 | } else if (input) { | |
67 | inp = input; | |
68 | } else if (!fill) { | |
69 | error("NULL input pointer and missing fill function"); | |
70 | goto exit_1; | |
71 | } else { | |
e23d54e4 | 72 | inp = large_malloc(LZ4_compressBound(uncomp_chunksize)); |
e76e1fdf KL |
73 | if (!inp) { |
74 | error("Could not allocate input buffer"); | |
75 | goto exit_1; | |
76 | } | |
77 | } | |
78 | inp_start = inp; | |
79 | ||
80 | if (posp) | |
81 | *posp = 0; | |
82 | ||
4d4b866a YL |
83 | if (fill) { |
84 | size = fill(inp, 4); | |
85 | if (size < 4) { | |
86 | error("data corrupted"); | |
87 | goto exit_2; | |
88 | } | |
89 | } | |
e76e1fdf KL |
90 | |
91 | chunksize = get_unaligned_le32(inp); | |
92 | if (chunksize == ARCHIVE_MAGICNUMBER) { | |
4d4b866a YL |
93 | if (!fill) { |
94 | inp += 4; | |
95 | size -= 4; | |
96 | } | |
e76e1fdf KL |
97 | } else { |
98 | error("invalid header"); | |
99 | goto exit_2; | |
100 | } | |
101 | ||
102 | if (posp) | |
103 | *posp += 4; | |
104 | ||
105 | for (;;) { | |
106 | ||
4d4b866a YL |
107 | if (fill) { |
108 | size = fill(inp, 4); | |
109 | if (size == 0) | |
110 | break; | |
111 | if (size < 4) { | |
112 | error("data corrupted"); | |
113 | goto exit_2; | |
114 | } | |
115 | } | |
e76e1fdf KL |
116 | |
117 | chunksize = get_unaligned_le32(inp); | |
118 | if (chunksize == ARCHIVE_MAGICNUMBER) { | |
4d4b866a YL |
119 | if (!fill) { |
120 | inp += 4; | |
121 | size -= 4; | |
122 | } | |
e76e1fdf KL |
123 | if (posp) |
124 | *posp += 4; | |
125 | continue; | |
126 | } | |
4d4b866a | 127 | |
e76e1fdf KL |
128 | |
129 | if (posp) | |
130 | *posp += 4; | |
131 | ||
4d4b866a YL |
132 | if (!fill) { |
133 | inp += 4; | |
134 | size -= 4; | |
135 | } else { | |
e23d54e4 | 136 | if (chunksize > LZ4_compressBound(uncomp_chunksize)) { |
e76e1fdf KL |
137 | error("chunk length is longer than allocated"); |
138 | goto exit_2; | |
139 | } | |
4d4b866a YL |
140 | size = fill(inp, chunksize); |
141 | if (size < chunksize) { | |
142 | error("data corrupted"); | |
143 | goto exit_2; | |
144 | } | |
e76e1fdf KL |
145 | } |
146 | #ifdef PREBOOT | |
147 | if (out_len >= uncomp_chunksize) { | |
148 | dest_len = uncomp_chunksize; | |
149 | out_len -= dest_len; | |
150 | } else | |
151 | dest_len = out_len; | |
e23d54e4 SS |
152 | |
153 | ret = LZ4_decompress_fast(inp, outp, dest_len); | |
154 | chunksize = ret; | |
e76e1fdf KL |
155 | #else |
156 | dest_len = uncomp_chunksize; | |
e23d54e4 SS |
157 | |
158 | ret = LZ4_decompress_safe(inp, outp, chunksize, dest_len); | |
159 | dest_len = ret; | |
e76e1fdf KL |
160 | #endif |
161 | if (ret < 0) { | |
162 | error("Decoding failed"); | |
163 | goto exit_2; | |
164 | } | |
165 | ||
2a1d689c | 166 | ret = -1; |
e76e1fdf KL |
167 | if (flush && flush(outp, dest_len) != dest_len) |
168 | goto exit_2; | |
169 | if (output) | |
170 | outp += dest_len; | |
171 | if (posp) | |
172 | *posp += chunksize; | |
173 | ||
4d4b866a YL |
174 | if (!fill) { |
175 | size -= chunksize; | |
e76e1fdf | 176 | |
4d4b866a YL |
177 | if (size == 0) |
178 | break; | |
179 | else if (size < 0) { | |
180 | error("data corrupted"); | |
181 | goto exit_2; | |
182 | } | |
183 | inp += chunksize; | |
e76e1fdf | 184 | } |
e76e1fdf KL |
185 | } |
186 | ||
187 | ret = 0; | |
188 | exit_2: | |
189 | if (!input) | |
190 | large_free(inp_start); | |
191 | exit_1: | |
192 | if (!output) | |
193 | large_free(outp); | |
194 | exit_0: | |
195 | return ret; | |
196 | } | |
197 | ||
198 | #ifdef PREBOOT | |
2d3862d2 | 199 | STATIC int INIT __decompress(unsigned char *buf, long in_len, |
d97b07c5 YL |
200 | long (*fill)(void*, unsigned long), |
201 | long (*flush)(void*, unsigned long), | |
2d3862d2 | 202 | unsigned char *output, long out_len, |
d97b07c5 | 203 | long *posp, |
2d3862d2 | 204 | void (*error)(char *x) |
e76e1fdf KL |
205 | ) |
206 | { | |
207 | return unlz4(buf, in_len - 4, fill, flush, output, posp, error); | |
208 | } | |
209 | #endif |