diff options
author | Nguyễn Thái Ngọc Duy <pclouds@gmail.com> | 2010-09-14 12:32:22 +1000 |
---|---|---|
committer | Nguyễn Thái Ngọc Duy <pclouds@gmail.com> | 2010-09-14 12:32:22 +1000 |
commit | 035857cb1c5ea2956b5bc721992169e2571fc03b (patch) | |
tree | 0979c8c1cad4537232b6be352a4c8da6d96072fb | |
parent | 29ab9e37d0dc05dc6e302437106800bf62748b05 (diff) | |
parent | 8837c5dec402fd6782589c0a676bc7f90dea4061 (diff) | |
download | busybox-w32-035857cb1c5ea2956b5bc721992169e2571fc03b.tar.gz busybox-w32-035857cb1c5ea2956b5bc721992169e2571fc03b.tar.bz2 busybox-w32-035857cb1c5ea2956b5bc721992169e2571fc03b.zip |
Merge branch 'origin/master' (early part)
44 files changed, 4343 insertions, 577 deletions
diff --git a/archival/Config.in b/archival/Config.in index deacc2822..4f762e860 100644 --- a/archival/Config.in +++ b/archival/Config.in | |||
@@ -5,6 +5,12 @@ | |||
5 | 5 | ||
6 | menu "Archival Utilities" | 6 | menu "Archival Utilities" |
7 | 7 | ||
8 | config FEATURE_SEAMLESS_XZ | ||
9 | bool "Make tar, rpm, modprobe etc understand .xz data" | ||
10 | default n | ||
11 | help | ||
12 | Make tar, rpm, modprobe etc understand .xz data. | ||
13 | |||
8 | config FEATURE_SEAMLESS_LZMA | 14 | config FEATURE_SEAMLESS_LZMA |
9 | bool "Make tar, rpm, modprobe etc understand .lzma data" | 15 | bool "Make tar, rpm, modprobe etc understand .lzma data" |
10 | default n | 16 | default n |
@@ -225,7 +231,7 @@ config FEATURE_TAR_CREATE | |||
225 | config FEATURE_TAR_AUTODETECT | 231 | config FEATURE_TAR_AUTODETECT |
226 | bool "Autodetect compressed tarballs" | 232 | bool "Autodetect compressed tarballs" |
227 | default n | 233 | default n |
228 | depends on TAR && (FEATURE_SEAMLESS_Z || FEATURE_SEAMLESS_GZ || FEATURE_SEAMLESS_BZ2 || FEATURE_SEAMLESS_LZMA) | 234 | depends on TAR && (FEATURE_SEAMLESS_Z || FEATURE_SEAMLESS_GZ || FEATURE_SEAMLESS_BZ2 || FEATURE_SEAMLESS_LZMA || FEATURE_SEAMLESS_XZ) |
229 | help | 235 | help |
230 | With this option tar can automatically detect compressed | 236 | With this option tar can automatically detect compressed |
231 | tarballs. Currently it works only on files (not pipes etc). | 237 | tarballs. Currently it works only on files (not pipes etc). |
@@ -327,6 +333,28 @@ config FEATURE_LZMA_FAST | |||
327 | This option reduces decompression time by about 25% at the cost of | 333 | This option reduces decompression time by about 25% at the cost of |
328 | a 1K bigger binary. | 334 | a 1K bigger binary. |
329 | 335 | ||
336 | config LZMA | ||
337 | bool "Provide lzma alias which supports only unpacking" | ||
338 | default n | ||
339 | depends on UNLZMA | ||
340 | help | ||
341 | Enable this option if you want commands like "lzma -d" to work. | ||
342 | IOW: you'll get lzma applet, but it will always require -d option. | ||
343 | |||
344 | config UNXZ | ||
345 | bool "unxz" | ||
346 | default n | ||
347 | help | ||
348 | unxz is a unlzma successor. | ||
349 | |||
350 | config XZ | ||
351 | bool "Provide xz alias which supports only unpacking" | ||
352 | default n | ||
353 | depends on UNXZ | ||
354 | help | ||
355 | Enable this option if you want commands like "xz -d" to work. | ||
356 | IOW: you'll get xz applet, but it will always require -d option. | ||
357 | |||
330 | config UNZIP | 358 | config UNZIP |
331 | bool "unzip" | 359 | bool "unzip" |
332 | default n | 360 | default n |
diff --git a/archival/Kbuild b/archival/Kbuild index 53bd7e21e..3300ea90f 100644 --- a/archival/Kbuild +++ b/archival/Kbuild | |||
@@ -8,18 +8,21 @@ libs-y += libunarchive/ | |||
8 | 8 | ||
9 | lib-y:= | 9 | lib-y:= |
10 | lib-$(CONFIG_AR) += ar.o | 10 | lib-$(CONFIG_AR) += ar.o |
11 | lib-$(CONFIG_BUNZIP2) += bbunzip.o | ||
12 | lib-$(CONFIG_BZIP2) += bzip2.o bbunzip.o | ||
13 | lib-$(CONFIG_UNLZMA) += bbunzip.o | ||
14 | lib-$(CONFIG_CPIO) += cpio.o | 11 | lib-$(CONFIG_CPIO) += cpio.o |
15 | lib-$(CONFIG_DPKG) += dpkg.o | 12 | lib-$(CONFIG_DPKG) += dpkg.o |
16 | lib-$(CONFIG_DPKG_DEB) += dpkg_deb.o | 13 | lib-$(CONFIG_DPKG_DEB) += dpkg_deb.o |
17 | lib-$(CONFIG_GUNZIP) += bbunzip.o | ||
18 | lib-$(CONFIG_GZIP) += gzip.o bbunzip.o | ||
19 | lib-$(CONFIG_LZOP) += lzop.o lzo1x_1.o lzo1x_1o.o lzo1x_d.o bbunzip.o | ||
20 | lib-$(CONFIG_LZOP_COMPR_HIGH) += lzo1x_9x.o | ||
21 | lib-$(CONFIG_RPM2CPIO) += rpm2cpio.o | 14 | lib-$(CONFIG_RPM2CPIO) += rpm2cpio.o |
22 | lib-$(CONFIG_RPM) += rpm.o | 15 | lib-$(CONFIG_RPM) += rpm.o |
23 | lib-$(CONFIG_TAR) += tar.o | 16 | lib-$(CONFIG_TAR) += tar.o |
24 | lib-$(CONFIG_UNCOMPRESS) += bbunzip.o | ||
25 | lib-$(CONFIG_UNZIP) += unzip.o | 17 | lib-$(CONFIG_UNZIP) += unzip.o |
18 | |||
19 | lib-$(CONFIG_LZOP) += lzop.o lzo1x_1.o lzo1x_1o.o lzo1x_d.o bbunzip.o | ||
20 | lib-$(CONFIG_LZOP_COMPR_HIGH) += lzo1x_9x.o | ||
21 | lib-$(CONFIG_GZIP) += gzip.o bbunzip.o | ||
22 | lib-$(CONFIG_BZIP2) += bzip2.o bbunzip.o | ||
23 | |||
24 | lib-$(CONFIG_UNXZ) += bbunzip.o | ||
25 | lib-$(CONFIG_UNLZMA) += bbunzip.o | ||
26 | lib-$(CONFIG_BUNZIP2) += bbunzip.o | ||
27 | lib-$(CONFIG_GUNZIP) += bbunzip.o | ||
28 | lib-$(CONFIG_UNCOMPRESS) += bbunzip.o | ||
diff --git a/archival/bbunzip.c b/archival/bbunzip.c index df674bc6c..08db2752c 100644 --- a/archival/bbunzip.c +++ b/archival/bbunzip.c | |||
@@ -4,17 +4,16 @@ | |||
4 | * | 4 | * |
5 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | 5 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
6 | */ | 6 | */ |
7 | |||
8 | #include "libbb.h" | 7 | #include "libbb.h" |
9 | #include "unarchive.h" | 8 | #include "unarchive.h" |
10 | 9 | ||
11 | enum { | 10 | enum { |
12 | OPT_STDOUT = 0x1, | 11 | OPT_STDOUT = 1 << 0, |
13 | OPT_FORCE = 0x2, | 12 | OPT_FORCE = 1 << 1, |
14 | /* gunzip and bunzip2 only: */ | 13 | /* only some decompressors: */ |
15 | OPT_VERBOSE = 0x4, | 14 | OPT_VERBOSE = 1 << 2, |
16 | OPT_DECOMPRESS = 0x8, | 15 | OPT_DECOMPRESS = 1 << 3, |
17 | OPT_TEST = 0x10, | 16 | OPT_TEST = 1 << 4, |
18 | }; | 17 | }; |
19 | 18 | ||
20 | static | 19 | static |
@@ -28,9 +27,15 @@ int open_to_or_warn(int to_fd, const char *filename, int flags, int mode) | |||
28 | return 0; | 27 | return 0; |
29 | } | 28 | } |
30 | 29 | ||
30 | char* FAST_FUNC append_ext(char *filename, const char *expected_ext) | ||
31 | { | ||
32 | return xasprintf("%s.%s", filename, expected_ext); | ||
33 | } | ||
34 | |||
31 | int FAST_FUNC bbunpack(char **argv, | 35 | int FAST_FUNC bbunpack(char **argv, |
32 | char* (*make_new_name)(char *filename), | 36 | IF_DESKTOP(long long) int FAST_FUNC (*unpacker)(unpack_info_t *info), |
33 | IF_DESKTOP(long long) int (*unpacker)(unpack_info_t *info) | 37 | char* FAST_FUNC (*make_new_name)(char *filename, const char *expected_ext), |
38 | const char *expected_ext | ||
34 | ) | 39 | ) |
35 | { | 40 | { |
36 | struct stat stat_buf; | 41 | struct stat stat_buf; |
@@ -69,7 +74,7 @@ int FAST_FUNC bbunpack(char **argv, | |||
69 | 74 | ||
70 | /* Open dst if we are going to unpack to file */ | 75 | /* Open dst if we are going to unpack to file */ |
71 | if (filename) { | 76 | if (filename) { |
72 | new_name = make_new_name(filename); | 77 | new_name = make_new_name(filename, expected_ext); |
73 | if (!new_name) { | 78 | if (!new_name) { |
74 | bb_error_msg("%s: unknown suffix - ignored", filename); | 79 | bb_error_msg("%s: unknown suffix - ignored", filename); |
75 | goto err; | 80 | goto err; |
@@ -141,10 +146,9 @@ int FAST_FUNC bbunpack(char **argv, | |||
141 | return exitcode; | 146 | return exitcode; |
142 | } | 147 | } |
143 | 148 | ||
144 | #if ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNCOMPRESS | 149 | #if ENABLE_UNCOMPRESS || ENABLE_BUNZIP2 || ENABLE_UNLZMA || ENABLE_UNXZ |
145 | |||
146 | static | 150 | static |
147 | char* make_new_name_generic(char *filename, const char *expected_ext) | 151 | char* FAST_FUNC make_new_name_generic(char *filename, const char *expected_ext) |
148 | { | 152 | { |
149 | char *extension = strrchr(filename, '.'); | 153 | char *extension = strrchr(filename, '.'); |
150 | if (!extension || strcmp(extension + 1, expected_ext) != 0) { | 154 | if (!extension || strcmp(extension + 1, expected_ext) != 0) { |
@@ -155,42 +159,35 @@ char* make_new_name_generic(char *filename, const char *expected_ext) | |||
155 | *extension = '\0'; | 159 | *extension = '\0'; |
156 | return filename; | 160 | return filename; |
157 | } | 161 | } |
158 | |||
159 | #endif | 162 | #endif |
160 | 163 | ||
161 | 164 | ||
162 | /* | 165 | /* |
163 | * Modified for busybox by Glenn McGrath | 166 | * Uncompress applet for busybox (c) 2002 Glenn McGrath |
164 | * Added support output to stdout by Thomas Lundquist <thomasez@zelow.no> | ||
165 | * | 167 | * |
166 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | 168 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
167 | */ | 169 | */ |
168 | 170 | #if ENABLE_UNCOMPRESS | |
169 | #if ENABLE_BUNZIP2 | ||
170 | |||
171 | static | 171 | static |
172 | char* make_new_name_bunzip2(char *filename) | 172 | IF_DESKTOP(long long) int FAST_FUNC unpack_uncompress(unpack_info_t *info UNUSED_PARAM) |
173 | { | 173 | { |
174 | return make_new_name_generic(filename, "bz2"); | 174 | IF_DESKTOP(long long) int status = -1; |
175 | } | ||
176 | 175 | ||
177 | static | 176 | if ((xread_char(STDIN_FILENO) != 0x1f) || (xread_char(STDIN_FILENO) != 0x9d)) { |
178 | IF_DESKTOP(long long) int unpack_bunzip2(unpack_info_t *info UNUSED_PARAM) | 177 | bb_error_msg("invalid magic"); |
179 | { | 178 | } else { |
180 | return unpack_bz2_stream_prime(STDIN_FILENO, STDOUT_FILENO); | 179 | status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO); |
180 | } | ||
181 | return status; | ||
181 | } | 182 | } |
182 | 183 | int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | |
183 | int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | 184 | int uncompress_main(int argc UNUSED_PARAM, char **argv) |
184 | int bunzip2_main(int argc UNUSED_PARAM, char **argv) | ||
185 | { | 185 | { |
186 | getopt32(argv, "cfvdt"); | 186 | getopt32(argv, "cf"); |
187 | argv += optind; | 187 | argv += optind; |
188 | if (applet_name[2] == 'c') | ||
189 | option_mask32 |= OPT_STDOUT; | ||
190 | 188 | ||
191 | return bbunpack(argv, make_new_name_bunzip2, unpack_bunzip2); | 189 | return bbunpack(argv, unpack_uncompress, make_new_name_generic, "Z"); |
192 | } | 190 | } |
193 | |||
194 | #endif | 191 | #endif |
195 | 192 | ||
196 | 193 | ||
@@ -221,11 +218,9 @@ int bunzip2_main(int argc UNUSED_PARAM, char **argv) | |||
221 | * See the license_msg below and the file COPYING for the software license. | 218 | * See the license_msg below and the file COPYING for the software license. |
222 | * See the file algorithm.doc for the compression algorithms and file formats. | 219 | * See the file algorithm.doc for the compression algorithms and file formats. |
223 | */ | 220 | */ |
224 | |||
225 | #if ENABLE_GUNZIP | 221 | #if ENABLE_GUNZIP |
226 | |||
227 | static | 222 | static |
228 | char* make_new_name_gunzip(char *filename) | 223 | char* FAST_FUNC make_new_name_gunzip(char *filename, const char *expected_ext UNUSED_PARAM) |
229 | { | 224 | { |
230 | char *extension = strrchr(filename, '.'); | 225 | char *extension = strrchr(filename, '.'); |
231 | 226 | ||
@@ -249,9 +244,8 @@ char* make_new_name_gunzip(char *filename) | |||
249 | } | 244 | } |
250 | return filename; | 245 | return filename; |
251 | } | 246 | } |
252 | |||
253 | static | 247 | static |
254 | IF_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info) | 248 | IF_DESKTOP(long long) int FAST_FUNC unpack_gunzip(unpack_info_t *info) |
255 | { | 249 | { |
256 | IF_DESKTOP(long long) int status = -1; | 250 | IF_DESKTOP(long long) int status = -1; |
257 | 251 | ||
@@ -277,7 +271,6 @@ IF_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info) | |||
277 | } | 271 | } |
278 | return status; | 272 | return status; |
279 | } | 273 | } |
280 | |||
281 | /* | 274 | /* |
282 | * Linux kernel build uses gzip -d -n. We accept and ignore it. | 275 | * Linux kernel build uses gzip -d -n. We accept and ignore it. |
283 | * Man page says: | 276 | * Man page says: |
@@ -291,7 +284,6 @@ IF_DESKTOP(long long) int unpack_gunzip(unpack_info_t *info) | |||
291 | * gzip: always save the original file name and time stamp (this is the default) | 284 | * gzip: always save the original file name and time stamp (this is the default) |
292 | * gunzip: restore the original file name and time stamp if present. | 285 | * gunzip: restore the original file name and time stamp if present. |
293 | */ | 286 | */ |
294 | |||
295 | int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | 287 | int gunzip_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
296 | int gunzip_main(int argc UNUSED_PARAM, char **argv) | 288 | int gunzip_main(int argc UNUSED_PARAM, char **argv) |
297 | { | 289 | { |
@@ -301,9 +293,33 @@ int gunzip_main(int argc UNUSED_PARAM, char **argv) | |||
301 | if (applet_name[1] == 'c') | 293 | if (applet_name[1] == 'c') |
302 | option_mask32 |= OPT_STDOUT; | 294 | option_mask32 |= OPT_STDOUT; |
303 | 295 | ||
304 | return bbunpack(argv, make_new_name_gunzip, unpack_gunzip); | 296 | return bbunpack(argv, unpack_gunzip, make_new_name_gunzip, /*unused:*/ NULL); |
305 | } | 297 | } |
298 | #endif | ||
306 | 299 | ||
300 | |||
301 | /* | ||
302 | * Modified for busybox by Glenn McGrath | ||
303 | * Added support output to stdout by Thomas Lundquist <thomasez@zelow.no> | ||
304 | * | ||
305 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
306 | */ | ||
307 | #if ENABLE_BUNZIP2 | ||
308 | static | ||
309 | IF_DESKTOP(long long) int FAST_FUNC unpack_bunzip2(unpack_info_t *info UNUSED_PARAM) | ||
310 | { | ||
311 | return unpack_bz2_stream_prime(STDIN_FILENO, STDOUT_FILENO); | ||
312 | } | ||
313 | int bunzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
314 | int bunzip2_main(int argc UNUSED_PARAM, char **argv) | ||
315 | { | ||
316 | getopt32(argv, "cfvdt"); | ||
317 | argv += optind; | ||
318 | if (applet_name[2] == 'c') /* bzcat */ | ||
319 | option_mask32 |= OPT_STDOUT; | ||
320 | |||
321 | return bbunpack(argv, unpack_bunzip2, make_new_name_generic, "bz2"); | ||
322 | } | ||
307 | #endif | 323 | #endif |
308 | 324 | ||
309 | 325 | ||
@@ -315,70 +331,51 @@ int gunzip_main(int argc UNUSED_PARAM, char **argv) | |||
315 | * | 331 | * |
316 | * Licensed under GPL v2, see file LICENSE in this tarball for details. | 332 | * Licensed under GPL v2, see file LICENSE in this tarball for details. |
317 | */ | 333 | */ |
318 | |||
319 | #if ENABLE_UNLZMA | 334 | #if ENABLE_UNLZMA |
320 | |||
321 | static | ||
322 | char* make_new_name_unlzma(char *filename) | ||
323 | { | ||
324 | return make_new_name_generic(filename, "lzma"); | ||
325 | } | ||
326 | |||
327 | static | 335 | static |
328 | IF_DESKTOP(long long) int unpack_unlzma(unpack_info_t *info UNUSED_PARAM) | 336 | IF_DESKTOP(long long) int FAST_FUNC unpack_unlzma(unpack_info_t *info UNUSED_PARAM) |
329 | { | 337 | { |
330 | return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO); | 338 | return unpack_lzma_stream(STDIN_FILENO, STDOUT_FILENO); |
331 | } | 339 | } |
332 | |||
333 | int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | 340 | int unlzma_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
334 | int unlzma_main(int argc UNUSED_PARAM, char **argv) | 341 | int unlzma_main(int argc UNUSED_PARAM, char **argv) |
335 | { | 342 | { |
336 | getopt32(argv, "cf"); | 343 | IF_LZMA(int opts =) getopt32(argv, "cfvdt"); |
337 | argv += optind; | 344 | # if ENABLE_LZMA |
338 | /* lzmacat? */ | 345 | /* lzma without -d or -t? */ |
339 | if (applet_name[4] == 'c') | 346 | if (applet_name[2] == 'm' && !(opts & (OPT_DECOMPRESS|OPT_TEST))) |
347 | bb_show_usage(); | ||
348 | # endif | ||
349 | /* lzcat? */ | ||
350 | if (applet_name[2] == 'c') | ||
340 | option_mask32 |= OPT_STDOUT; | 351 | option_mask32 |= OPT_STDOUT; |
341 | 352 | ||
342 | return bbunpack(argv, make_new_name_unlzma, unpack_unlzma); | 353 | argv += optind; |
354 | return bbunpack(argv, unpack_unlzma, make_new_name_generic, "lzma"); | ||
343 | } | 355 | } |
344 | |||
345 | #endif | 356 | #endif |
346 | 357 | ||
347 | 358 | ||
348 | /* | 359 | #if ENABLE_UNXZ |
349 | * Uncompress applet for busybox (c) 2002 Glenn McGrath | ||
350 | * | ||
351 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
352 | */ | ||
353 | |||
354 | #if ENABLE_UNCOMPRESS | ||
355 | |||
356 | static | 360 | static |
357 | char* make_new_name_uncompress(char *filename) | 361 | IF_DESKTOP(long long) int FAST_FUNC unpack_unxz(unpack_info_t *info UNUSED_PARAM) |
358 | { | 362 | { |
359 | return make_new_name_generic(filename, "Z"); | 363 | return unpack_xz_stream(STDIN_FILENO, STDOUT_FILENO); |
360 | } | 364 | } |
361 | 365 | int unxz_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | |
362 | static | 366 | int unxz_main(int argc UNUSED_PARAM, char **argv) |
363 | IF_DESKTOP(long long) int unpack_uncompress(unpack_info_t *info UNUSED_PARAM) | ||
364 | { | 367 | { |
365 | IF_DESKTOP(long long) int status = -1; | 368 | int opts = getopt32(argv, "cfvdt"); |
366 | 369 | # if ENABLE_XZ | |
367 | if ((xread_char(STDIN_FILENO) != 0x1f) || (xread_char(STDIN_FILENO) != 0x9d)) { | 370 | /* xz without -d or -t? */ |
368 | bb_error_msg("invalid magic"); | 371 | if (applet_name[2] == '\0' && !(opts & (OPT_DECOMPRESS|OPT_TEST))) |
369 | } else { | 372 | bb_show_usage(); |
370 | status = unpack_Z_stream(STDIN_FILENO, STDOUT_FILENO); | 373 | # endif |
371 | } | 374 | /* xzcat? */ |
372 | return status; | 375 | if (applet_name[2] == 'c') |
373 | } | 376 | option_mask32 |= OPT_STDOUT; |
374 | 377 | ||
375 | int uncompress_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
376 | int uncompress_main(int argc UNUSED_PARAM, char **argv) | ||
377 | { | ||
378 | getopt32(argv, "cf"); | ||
379 | argv += optind; | 378 | argv += optind; |
380 | 379 | return bbunpack(argv, unpack_unxz, make_new_name_generic, "xz"); | |
381 | return bbunpack(argv, make_new_name_uncompress, unpack_uncompress); | ||
382 | } | 380 | } |
383 | |||
384 | #endif | 381 | #endif |
diff --git a/archival/bzip2.c b/archival/bzip2.c index bbaf56669..f1c84d681 100644 --- a/archival/bzip2.c +++ b/archival/bzip2.c | |||
@@ -102,7 +102,7 @@ IF_DESKTOP(long long) int bz_write(bz_stream *strm, void* rbuf, ssize_t rlen, vo | |||
102 | } | 102 | } |
103 | 103 | ||
104 | static | 104 | static |
105 | IF_DESKTOP(long long) int compressStream(unpack_info_t *info UNUSED_PARAM) | 105 | IF_DESKTOP(long long) int FAST_FUNC compressStream(unpack_info_t *info UNUSED_PARAM) |
106 | { | 106 | { |
107 | IF_DESKTOP(long long) int total; | 107 | IF_DESKTOP(long long) int total; |
108 | ssize_t count; | 108 | ssize_t count; |
@@ -135,12 +135,6 @@ IF_DESKTOP(long long) int compressStream(unpack_info_t *info UNUSED_PARAM) | |||
135 | return total; | 135 | return total; |
136 | } | 136 | } |
137 | 137 | ||
138 | static | ||
139 | char* make_new_name_bzip2(char *filename) | ||
140 | { | ||
141 | return xasprintf("%s.bz2", filename); | ||
142 | } | ||
143 | |||
144 | int bzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | 138 | int bzip2_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
145 | int bzip2_main(int argc UNUSED_PARAM, char **argv) | 139 | int bzip2_main(int argc UNUSED_PARAM, char **argv) |
146 | { | 140 | { |
@@ -181,5 +175,5 @@ int bzip2_main(int argc UNUSED_PARAM, char **argv) | |||
181 | 175 | ||
182 | argv += optind; | 176 | argv += optind; |
183 | option_mask32 &= 0x7; /* ignore all except -cfv */ | 177 | option_mask32 &= 0x7; /* ignore all except -cfv */ |
184 | return bbunpack(argv, make_new_name_bzip2, compressStream); | 178 | return bbunpack(argv, compressStream, append_ext, "bz2"); |
185 | } | 179 | } |
diff --git a/archival/gzip.c b/archival/gzip.c index 958336303..9e915519d 100644 --- a/archival/gzip.c +++ b/archival/gzip.c | |||
@@ -1998,13 +1998,7 @@ static void zip(ulg time_stamp) | |||
1998 | 1998 | ||
1999 | /* ======================================================================== */ | 1999 | /* ======================================================================== */ |
2000 | static | 2000 | static |
2001 | char* make_new_name_gzip(char *filename) | 2001 | IF_DESKTOP(long long) int FAST_FUNC pack_gzip(unpack_info_t *info UNUSED_PARAM) |
2002 | { | ||
2003 | return xasprintf("%s.gz", filename); | ||
2004 | } | ||
2005 | |||
2006 | static | ||
2007 | IF_DESKTOP(long long) int pack_gzip(unpack_info_t *info UNUSED_PARAM) | ||
2008 | { | 2002 | { |
2009 | struct stat s; | 2003 | struct stat s; |
2010 | 2004 | ||
@@ -2063,7 +2057,7 @@ static const char gzip_longopts[] ALIGN1 = | |||
2063 | #endif | 2057 | #endif |
2064 | 2058 | ||
2065 | /* | 2059 | /* |
2066 | * Linux kernel build uses gzip -d -n. We accept and ignore it. | 2060 | * Linux kernel build uses gzip -d -n. We accept and ignore -n. |
2067 | * Man page says: | 2061 | * Man page says: |
2068 | * -n --no-name | 2062 | * -n --no-name |
2069 | * gzip: do not save the original file name and time stamp. | 2063 | * gzip: do not save the original file name and time stamp. |
@@ -2113,5 +2107,5 @@ int gzip_main(int argc UNUSED_PARAM, char **argv) | |||
2113 | /* Initialise the CRC32 table */ | 2107 | /* Initialise the CRC32 table */ |
2114 | G1.crc_32_tab = crc32_filltable(NULL, 0); | 2108 | G1.crc_32_tab = crc32_filltable(NULL, 0); |
2115 | 2109 | ||
2116 | return bbunpack(argv, make_new_name_gzip, pack_gzip); | 2110 | return bbunpack(argv, pack_gzip, append_ext, "gz"); |
2117 | } | 2111 | } |
diff --git a/archival/libunarchive/Kbuild b/archival/libunarchive/Kbuild index 11d23b25f..ed8e85793 100644 --- a/archival/libunarchive/Kbuild +++ b/archival/libunarchive/Kbuild | |||
@@ -49,6 +49,7 @@ lib-$(CONFIG_FEATURE_SEAMLESS_Z) += open_transformer.o decompress_uncompr | |||
49 | lib-$(CONFIG_FEATURE_SEAMLESS_GZ) += open_transformer.o decompress_unzip.o get_header_tar_gz.o | 49 | lib-$(CONFIG_FEATURE_SEAMLESS_GZ) += open_transformer.o decompress_unzip.o get_header_tar_gz.o |
50 | lib-$(CONFIG_FEATURE_SEAMLESS_BZ2) += open_transformer.o decompress_bunzip2.o get_header_tar_bz2.o | 50 | lib-$(CONFIG_FEATURE_SEAMLESS_BZ2) += open_transformer.o decompress_bunzip2.o get_header_tar_bz2.o |
51 | lib-$(CONFIG_FEATURE_SEAMLESS_LZMA) += open_transformer.o decompress_unlzma.o get_header_tar_lzma.o | 51 | lib-$(CONFIG_FEATURE_SEAMLESS_LZMA) += open_transformer.o decompress_unlzma.o get_header_tar_lzma.o |
52 | lib-$(CONFIG_FEATURE_SEAMLESS_XZ) += open_transformer.o decompress_unxz.o | ||
52 | lib-$(CONFIG_FEATURE_COMPRESS_USAGE) += decompress_bunzip2.o | 53 | lib-$(CONFIG_FEATURE_COMPRESS_USAGE) += decompress_bunzip2.o |
53 | 54 | ||
54 | ifneq ($(lib-y),) | 55 | ifneq ($(lib-y),) |
diff --git a/archival/libunarchive/decompress_uncompress.c b/archival/libunarchive/decompress_uncompress.c index 2877c8981..1ff89ce3c 100644 --- a/archival/libunarchive/decompress_uncompress.c +++ b/archival/libunarchive/decompress_uncompress.c | |||
@@ -229,7 +229,7 @@ unpack_Z_stream(int fd_in, int fd_out) | |||
229 | ("insize:%d posbits:%d inbuf:%02X %02X %02X %02X %02X (%d)", | 229 | ("insize:%d posbits:%d inbuf:%02X %02X %02X %02X %02X (%d)", |
230 | insize, posbits, p[-1], p[0], p[1], p[2], p[3], | 230 | insize, posbits, p[-1], p[0], p[1], p[2], p[3], |
231 | (posbits & 07)); | 231 | (posbits & 07)); |
232 | bb_error_msg("uncompress: corrupt input"); | 232 | bb_error_msg("corrupted data"); |
233 | goto err; | 233 | goto err; |
234 | } | 234 | } |
235 | 235 | ||
diff --git a/archival/libunarchive/decompress_unxz.c b/archival/libunarchive/decompress_unxz.c new file mode 100644 index 000000000..3f9392984 --- /dev/null +++ b/archival/libunarchive/decompress_unxz.c | |||
@@ -0,0 +1,122 @@ | |||
1 | /* | ||
2 | * This file uses XZ Embedded library code which is written | ||
3 | * by Lasse Collin <lasse.collin@tukaani.org> | ||
4 | * and Igor Pavlov <http://7-zip.org/> | ||
5 | * | ||
6 | * See README file in unxz/ directory for more information. | ||
7 | * | ||
8 | * This file is: | ||
9 | * Copyright (C) 2010 Denys Vlasenko <vda.linux@googlemail.com> | ||
10 | * Licensed under GPLv2, see file LICENSE in this tarball for details. | ||
11 | */ | ||
12 | #include "libbb.h" | ||
13 | #include "unarchive.h" | ||
14 | |||
15 | #define XZ_REALLOC_DICT_BUF(ptr, size) xrealloc(ptr, size) | ||
16 | #define XZ_FUNC FAST_FUNC | ||
17 | #define XZ_EXTERN static | ||
18 | |||
19 | /* Skip check (rather than fail) of unsupported hash functions */ | ||
20 | #define XZ_DEC_ANY_CHECK 1 | ||
21 | |||
22 | /* We use our own crc32 function */ | ||
23 | #define XZ_INTERNAL_CRC32 0 | ||
24 | static uint32_t *crc32_table; | ||
25 | static uint32_t xz_crc32(const uint8_t *buf, size_t size, uint32_t crc) | ||
26 | { | ||
27 | crc = ~crc; | ||
28 | |||
29 | while (size != 0) { | ||
30 | crc = crc32_table[*buf++ ^ (crc & 0xFF)] ^ (crc >> 8); | ||
31 | --size; | ||
32 | } | ||
33 | |||
34 | return ~crc; | ||
35 | } | ||
36 | |||
37 | /* We use arch-optimized unaligned accessors */ | ||
38 | #define get_unaligned_le32(buf) ({ uint32_t v; move_from_unaligned32(v, buf); SWAP_LE32(v); }) | ||
39 | #define get_unaligned_be32(buf) ({ uint32_t v; move_from_unaligned32(v, buf); SWAP_BE32(v); }) | ||
40 | #define put_unaligned_le32(val, buf) move_to_unaligned16(buf, SWAP_LE32(val)) | ||
41 | #define put_unaligned_be32(val, buf) move_to_unaligned16(buf, SWAP_BE32(val)) | ||
42 | |||
43 | #include "unxz/xz.h" | ||
44 | #include "unxz/xz_config.h" | ||
45 | |||
46 | #include "unxz/xz_dec_bcj.c" | ||
47 | #include "unxz/xz_dec_lzma2.c" | ||
48 | #include "unxz/xz_dec_stream.c" | ||
49 | #include "unxz/xz_lzma2.h" | ||
50 | #include "unxz/xz_private.h" | ||
51 | #include "unxz/xz_stream.h" | ||
52 | |||
53 | IF_DESKTOP(long long) int FAST_FUNC | ||
54 | unpack_xz_stream(int src_fd, int dst_fd) | ||
55 | { | ||
56 | struct xz_buf iobuf; | ||
57 | struct xz_dec *state; | ||
58 | unsigned char *membuf; | ||
59 | IF_DESKTOP(long long) int total = 0; | ||
60 | enum { | ||
61 | IN_SIZE = 4 * 1024, | ||
62 | OUT_SIZE = 60 * 1024, | ||
63 | }; | ||
64 | |||
65 | if (!crc32_table) | ||
66 | crc32_table = crc32_filltable(NULL, /*endian:*/ 0); | ||
67 | |||
68 | membuf = xmalloc(IN_SIZE + OUT_SIZE); | ||
69 | memset(&iobuf, 0, sizeof(iobuf)); | ||
70 | iobuf.in = membuf; | ||
71 | iobuf.out = membuf + IN_SIZE; | ||
72 | iobuf.out_size = OUT_SIZE; | ||
73 | |||
74 | state = xz_dec_init(64*1024); /* initial dict of 64k */ | ||
75 | |||
76 | while (1) { | ||
77 | enum xz_ret r; | ||
78 | int insz, rd, outpos; | ||
79 | |||
80 | iobuf.in_size -= iobuf.in_pos; | ||
81 | insz = iobuf.in_size; | ||
82 | if (insz) | ||
83 | memmove(membuf, membuf + iobuf.in_pos, insz); | ||
84 | iobuf.in_pos = 0; | ||
85 | rd = IN_SIZE - insz; | ||
86 | if (rd) { | ||
87 | rd = safe_read(src_fd, membuf + insz, rd); | ||
88 | if (rd < 0) { | ||
89 | bb_error_msg("read error"); | ||
90 | total = -1; | ||
91 | break; | ||
92 | } | ||
93 | iobuf.in_size = insz + rd; | ||
94 | } | ||
95 | // bb_error_msg(">in pos:%d size:%d out pos:%d size:%d", | ||
96 | // iobuf.in_pos, iobuf.in_size, iobuf.out_pos, iobuf.out_size); | ||
97 | r = xz_dec_run(state, &iobuf); | ||
98 | // bb_error_msg("<in pos:%d size:%d out pos:%d size:%d r:%d", | ||
99 | // iobuf.in_pos, iobuf.in_size, iobuf.out_pos, iobuf.out_size, r); | ||
100 | outpos = iobuf.out_pos; | ||
101 | if (outpos) { | ||
102 | xwrite(dst_fd, iobuf.out, outpos); | ||
103 | IF_DESKTOP(total += outpos;) | ||
104 | } | ||
105 | if (r == XZ_STREAM_END | ||
106 | /* this happens even with well-formed files: */ | ||
107 | || (r == XZ_BUF_ERROR && insz == 0 && outpos == 0) | ||
108 | ) { | ||
109 | break; | ||
110 | } | ||
111 | if (r != XZ_OK && r != XZ_UNSUPPORTED_CHECK) { | ||
112 | bb_error_msg("corrupted data"); | ||
113 | total = -1; | ||
114 | break; | ||
115 | } | ||
116 | iobuf.out_pos = 0; | ||
117 | } | ||
118 | xz_dec_end(state); | ||
119 | free(membuf); | ||
120 | |||
121 | return total; | ||
122 | } | ||
diff --git a/archival/libunarchive/unxz/README b/archival/libunarchive/unxz/README new file mode 100644 index 000000000..f79b0a404 --- /dev/null +++ b/archival/libunarchive/unxz/README | |||
@@ -0,0 +1,136 @@ | |||
1 | |||
2 | XZ Embedded | ||
3 | =========== | ||
4 | |||
5 | XZ Embedded is a relatively small, limited implementation of the .xz | ||
6 | file format. Currently only decoding is implemented. | ||
7 | |||
8 | XZ Embedded was written for use in the Linux kernel, but the code can | ||
9 | be easily used in other environments too, including regular userspace | ||
10 | applications. | ||
11 | |||
12 | This README contains information that is useful only when the copy | ||
13 | of XZ Embedded isn't part of the Linux kernel tree. You should also | ||
14 | read linux/Documentation/xz.txt even if you aren't using XZ Embedded | ||
15 | as part of Linux; information in that file is not repeated in this | ||
16 | README. | ||
17 | |||
18 | Compiling the Linux kernel module | ||
19 | |||
20 | The xz_dec module depends on crc32 module, so make sure that you have | ||
21 | it enabled (CONFIG_CRC32). | ||
22 | |||
23 | Building the xz_dec and xz_dec_test modules without support for BCJ | ||
24 | filters: | ||
25 | |||
26 | cd linux/lib/xz | ||
27 | make -C /path/to/kernel/source \ | ||
28 | KCPPFLAGS=-I"$(pwd)/../../include" M="$(pwd)" \ | ||
29 | CONFIG_XZ_DEC=m CONFIG_XZ_DEC_TEST=m | ||
30 | |||
31 | Building the xz_dec and xz_dec_test modules with support for BCJ | ||
32 | filters: | ||
33 | |||
34 | cd linux/lib/xz | ||
35 | make -C /path/to/kernel/source \ | ||
36 | KCPPFLAGS=-I"$(pwd)/../../include" M="$(pwd)" \ | ||
37 | CONFIG_XZ_DEC=m CONFIG_XZ_DEC_TEST=m CONFIG_XZ_DEC_BCJ=y \ | ||
38 | CONFIG_XZ_DEC_X86=y CONFIG_XZ_DEC_POWERPC=y \ | ||
39 | CONFIG_XZ_DEC_IA64=y CONFIG_XZ_DEC_ARM=y \ | ||
40 | CONFIG_XZ_DEC_ARMTHUMB=y CONFIG_XZ_DEC_SPARC=y | ||
41 | |||
42 | If you want only one or a few of the BCJ filters, omit the appropriate | ||
43 | variables. CONFIG_XZ_DEC_BCJ=y is always required to build the support | ||
44 | code shared between all BCJ filters. | ||
45 | |||
46 | Most people don't need the xz_dec_test module. You can skip building | ||
47 | it by omitting CONFIG_XZ_DEC_TEST=m from the make command line. | ||
48 | |||
49 | Compiler requirements | ||
50 | |||
51 | XZ Embedded should compile as either GNU-C89 (used in the Linux | ||
52 | kernel) or with any C99 compiler. Getting the code to compile with | ||
53 | non-GNU C89 compiler or a C++ compiler should be quite easy as | ||
54 | long as there is a data type for unsigned 64-bit integer (or the | ||
55 | code is modified not to support large files, which needs some more | ||
56 | care than just using 32-bit integer instead of 64-bit). | ||
57 | |||
58 | If you use GCC, try to use a recent version. For example, on x86, | ||
59 | xz_dec_lzma2.c compiled with GCC 3.3.6 is 15-25 % slower than when | ||
60 | compiled with GCC 4.3.3. | ||
61 | |||
62 | Embedding into userspace applications | ||
63 | |||
64 | To embed the XZ decoder, copy the following files into a single | ||
65 | directory in your source code tree: | ||
66 | |||
67 | linux/include/linux/xz.h | ||
68 | linux/lib/xz/xz_crc32.c | ||
69 | linux/lib/xz/xz_dec_lzma2.c | ||
70 | linux/lib/xz/xz_dec_stream.c | ||
71 | linux/lib/xz/xz_lzma2.h | ||
72 | linux/lib/xz/xz_private.h | ||
73 | linux/lib/xz/xz_stream.h | ||
74 | userspace/xz_config.h | ||
75 | |||
76 | Alternatively, xz.h may be placed into a different directory but then | ||
77 | that directory must be in the compiler include path when compiling | ||
78 | the .c files. | ||
79 | |||
80 | Your code should use only the functions declared in xz.h. The rest of | ||
81 | the .h files are meant only for internal use in XZ Embedded. | ||
82 | |||
83 | You may want to modify xz_config.h to be more suitable for your build | ||
84 | environment. Probably you should at least skim through it even if the | ||
85 | default file works as is. | ||
86 | |||
87 | BCJ filter support | ||
88 | |||
89 | If you want support for one or more BCJ filters, you need to copy also | ||
90 | linux/lib/xz/xz_dec_bcj.c into your application, and use appropriate | ||
91 | #defines in xz_config.h or in compiler flags. You don't need these | ||
92 | #defines in the code that just uses XZ Embedded via xz.h, but having | ||
93 | them always #defined doesn't hurt either. | ||
94 | |||
95 | #define Instruction set BCJ filter endianness | ||
96 | XZ_DEC_X86 x86 or x86-64 Little endian only | ||
97 | XZ_DEC_POWERPC PowerPC Big endian only | ||
98 | XZ_DEC_IA64 Itanium (IA-64) Big or little endian | ||
99 | XZ_DEC_ARM ARM Little endian only | ||
100 | XZ_DEC_ARMTHUMB ARM-Thumb Little endian only | ||
101 | XZ_DEC_SPARC SPARC Big or little endian | ||
102 | |||
103 | While some architectures are (partially) bi-endian, the endianness | ||
104 | setting doesn't change the endianness of the instructions on all | ||
105 | architectures. That's why Itanium and SPARC filters work for both big | ||
106 | and little endian executables (Itanium has little endian instructions | ||
107 | and SPARC has big endian instructions). | ||
108 | |||
109 | There currently is no filter for little endian PowerPC or big endian | ||
110 | ARM or ARM-Thumb. Implementing filters for them can be considered if | ||
111 | there is a need for such filters in real-world applications. | ||
112 | |||
113 | Notes about shared libraries | ||
114 | |||
115 | If you are including XZ Embedded into a shared library, you very | ||
116 | probably should rename the xz_* functions to prevent symbol | ||
117 | conflicts in case your library is linked against some other library | ||
118 | or application that also has XZ Embedded in it (which may even be | ||
119 | a different version of XZ Embedded). TODO: Provide an easy way | ||
120 | to do this. | ||
121 | |||
122 | Please don't create a shared library of XZ Embedded itself unless | ||
123 | it is fine to rebuild everything depending on that shared library | ||
124 | everytime you upgrade to a newer version of XZ Embedded. There are | ||
125 | no API or ABI stability guarantees between different versions of | ||
126 | XZ Embedded. | ||
127 | |||
128 | Specifying the calling convention | ||
129 | |||
130 | XZ_FUNC macro was included to support declaring functions with __init | ||
131 | in Linux. Outside Linux, it can be used to specify the calling | ||
132 | convention on systems that support multiple calling conventions. | ||
133 | For example, on Windows, you may make all functions use the stdcall | ||
134 | calling convention by defining XZ_FUNC=__stdcall when building and | ||
135 | using the functions from XZ Embedded. | ||
136 | |||
diff --git a/archival/libunarchive/unxz/xz.h b/archival/libunarchive/unxz/xz.h new file mode 100644 index 000000000..eb82706b9 --- /dev/null +++ b/archival/libunarchive/unxz/xz.h | |||
@@ -0,0 +1,222 @@ | |||
1 | /* | ||
2 | * XZ decompressor | ||
3 | * | ||
4 | * Authors: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * Igor Pavlov <http://7-zip.org/> | ||
6 | * | ||
7 | * This file has been put into the public domain. | ||
8 | * You can do whatever you want with this file. | ||
9 | */ | ||
10 | |||
11 | #ifndef XZ_H | ||
12 | #define XZ_H | ||
13 | |||
14 | #ifdef __KERNEL__ | ||
15 | # include <linux/stddef.h> | ||
16 | # include <linux/types.h> | ||
17 | #else | ||
18 | # include <stddef.h> | ||
19 | # include <stdint.h> | ||
20 | #endif | ||
21 | |||
22 | /* In Linux, this is used to make extern functions static when needed. */ | ||
23 | #ifndef XZ_EXTERN | ||
24 | # define XZ_EXTERN extern | ||
25 | #endif | ||
26 | |||
27 | /* In Linux, this is used to mark the functions with __init when needed. */ | ||
28 | #ifndef XZ_FUNC | ||
29 | # define XZ_FUNC | ||
30 | #endif | ||
31 | |||
32 | /** | ||
33 | * enum xz_ret - Return codes | ||
34 | * @XZ_OK: Everything is OK so far. More input or more | ||
35 | * output space is required to continue. | ||
36 | * @XZ_STREAM_END: Operation finished successfully. | ||
37 | * @XZ_UNSUPPORTED_CHECK: Integrity check type is not supported. Decoding | ||
38 | * is still possible in multi-call mode by simply | ||
39 | * calling xz_dec_run() again. | ||
40 | * NOTE: This return value is used only if | ||
41 | * XZ_DEC_ANY_CHECK was defined at build time, | ||
42 | * which is not used in the kernel. Unsupported | ||
43 | * check types return XZ_OPTIONS_ERROR if | ||
44 | * XZ_DEC_ANY_CHECK was not defined at build time. | ||
45 | * @XZ_MEMLIMIT_ERROR: Not enough memory was preallocated at decoder | ||
46 | * initialization time. | ||
47 | * @XZ_FORMAT_ERROR: File format was not recognized (wrong magic | ||
48 | * bytes). | ||
49 | * @XZ_OPTIONS_ERROR: This implementation doesn't support the requested | ||
50 | * compression options. In the decoder this means | ||
51 | * that the header CRC32 matches, but the header | ||
52 | * itself specifies something that we don't support. | ||
53 | * @XZ_DATA_ERROR: Compressed data is corrupt. | ||
54 | * @XZ_BUF_ERROR: Cannot make any progress. Details are slightly | ||
55 | * different between multi-call and single-call | ||
56 | * mode; more information below. | ||
57 | * | ||
58 | * In multi-call mode, XZ_BUF_ERROR is returned when two consecutive calls | ||
59 | * to XZ code cannot consume any input and cannot produce any new output. | ||
60 | * This happens when there is no new input available, or the output buffer | ||
61 | * is full while at least one output byte is still pending. Assuming your | ||
62 | * code is not buggy, you can get this error only when decoding a compressed | ||
63 | * stream that is truncated or otherwise corrupt. | ||
64 | * | ||
65 | * In single-call mode, XZ_BUF_ERROR is returned only when the output buffer | ||
66 | * is too small, or the compressed input is corrupt in a way that makes the | ||
67 | * decoder produce more output than the caller expected. When it is | ||
68 | * (relatively) clear that the compressed input is truncated, XZ_DATA_ERROR | ||
69 | * is used instead of XZ_BUF_ERROR. | ||
70 | */ | ||
71 | enum xz_ret { | ||
72 | XZ_OK, | ||
73 | XZ_STREAM_END, | ||
74 | XZ_UNSUPPORTED_CHECK, | ||
75 | XZ_MEMLIMIT_ERROR, | ||
76 | XZ_FORMAT_ERROR, | ||
77 | XZ_OPTIONS_ERROR, | ||
78 | XZ_DATA_ERROR, | ||
79 | XZ_BUF_ERROR | ||
80 | }; | ||
81 | |||
82 | /** | ||
83 | * struct xz_buf - Passing input and output buffers to XZ code | ||
84 | * @in: Beginning of the input buffer. This may be NULL if and only | ||
85 | * if in_pos is equal to in_size. | ||
86 | * @in_pos: Current position in the input buffer. This must not exceed | ||
87 | * in_size. | ||
88 | * @in_size: Size of the input buffer | ||
89 | * @out: Beginning of the output buffer. This may be NULL if and only | ||
90 | * if out_pos is equal to out_size. | ||
91 | * @out_pos: Current position in the output buffer. This must not exceed | ||
92 | * out_size. | ||
93 | * @out_size: Size of the output buffer | ||
94 | * | ||
95 | * Only the contents of the output buffer from out[out_pos] onward, and | ||
96 | * the variables in_pos and out_pos are modified by the XZ code. | ||
97 | */ | ||
98 | struct xz_buf { | ||
99 | const uint8_t *in; | ||
100 | size_t in_pos; | ||
101 | size_t in_size; | ||
102 | |||
103 | uint8_t *out; | ||
104 | size_t out_pos; | ||
105 | size_t out_size; | ||
106 | }; | ||
107 | |||
108 | /** | ||
109 | * struct xz_dec - Opaque type to hold the XZ decoder state | ||
110 | */ | ||
111 | struct xz_dec; | ||
112 | |||
113 | /** | ||
114 | * xz_dec_init() - Allocate and initialize a XZ decoder state | ||
115 | * @dict_max: Maximum size of the LZMA2 dictionary (history buffer) for | ||
116 | * multi-call decoding, or special value of zero to indicate | ||
117 | * single-call decoding mode. | ||
118 | * | ||
119 | * If dict_max > 0, the decoder is initialized to work in multi-call mode. | ||
120 | * dict_max number of bytes of memory is preallocated for the LZMA2 | ||
121 | * dictionary. This way there is no risk that xz_dec_run() could run out | ||
122 | * of memory, since xz_dec_run() will never allocate any memory. Instead, | ||
123 | * if the preallocated dictionary is too small for decoding the given input | ||
124 | * stream, xz_dec_run() will return XZ_MEMLIMIT_ERROR. Thus, it is important | ||
125 | * to know what kind of data will be decoded to avoid allocating excessive | ||
126 | * amount of memory for the dictionary. | ||
127 | * | ||
128 | * LZMA2 dictionary is always 2^n bytes or 2^n + 2^(n-1) bytes (the latter | ||
129 | * sizes are less common in practice). In the kernel, dictionary sizes of | ||
130 | * 64 KiB, 128 KiB, 256 KiB, 512 KiB, and 1 MiB are probably the only | ||
131 | * reasonable values. | ||
132 | * | ||
133 | * If dict_max == 0, the decoder is initialized to work in single-call mode. | ||
134 | * In single-call mode, xz_dec_run() decodes the whole stream at once. The | ||
135 | * caller must provide enough output space or the decoding will fail. The | ||
136 | * output space is used as the dictionary buffer, which is why there is | ||
137 | * no need to allocate the dictionary as part of the decoder's internal | ||
138 | * state. | ||
139 | * | ||
140 | * Because the output buffer is used as the workspace, streams encoded using | ||
141 | * a big dictionary are not a problem in single-call. It is enough that the | ||
142 | * output buffer is big enough to hold the actual uncompressed data; it | ||
143 | * can be smaller than the dictionary size stored in the stream headers. | ||
144 | * | ||
145 | * On success, xz_dec_init() returns a pointer to struct xz_dec, which is | ||
146 | * ready to be used with xz_dec_run(). On error, xz_dec_init() returns NULL. | ||
147 | */ | ||
148 | XZ_EXTERN struct xz_dec * XZ_FUNC xz_dec_init(uint32_t dict_max); | ||
149 | |||
150 | /** | ||
151 | * xz_dec_run() - Run the XZ decoder | ||
152 | * @s: Decoder state allocated using xz_dec_init() | ||
153 | * @b: Input and output buffers | ||
154 | * | ||
155 | * In multi-call mode, this function may return any of the values listed in | ||
156 | * enum xz_ret. | ||
157 | * | ||
158 | * In single-call mode, this function never returns XZ_OK. If an error occurs | ||
159 | * in single-call mode (return value is not XZ_STREAM_END), b->in_pos and | ||
160 | * b->out_pos are not modified, and the contents of the output buffer from | ||
161 | * b->out[b->out_pos] onward are undefined. | ||
162 | * | ||
163 | * NOTE: In single-call mode, the contents of the output buffer are undefined | ||
164 | * also after XZ_BUF_ERROR. This is because with some filter chains, there | ||
165 | * may be a second pass over the output buffer, and this pass cannot be | ||
166 | * properly done if the output buffer is truncated. Thus, you cannot give | ||
167 | * the single-call decoder a too small buffer and then expect to get that | ||
168 | * amount valid data from the beginning of the stream. You must use the | ||
169 | * multi-call decoder if you don't want to uncompress the whole stream. | ||
170 | */ | ||
171 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_run(struct xz_dec *s, struct xz_buf *b); | ||
172 | |||
173 | /** | ||
174 | * xz_dec_reset() - Reset an already allocated decoder state | ||
175 | * @s: Decoder state allocated using xz_dec_init() | ||
176 | * | ||
177 | * This function can be used to reset the multi-call decoder state without | ||
178 | * freeing and reallocating memory with xz_dec_end() and xz_dec_init(). | ||
179 | * | ||
180 | * In single-call mode, xz_dec_reset() is always called in the beginning of | ||
181 | * xz_dec_run(). Thus, explicit call to xz_dec_reset() is useful only in | ||
182 | * multi-call mode. | ||
183 | */ | ||
184 | XZ_EXTERN void XZ_FUNC xz_dec_reset(struct xz_dec *s); | ||
185 | |||
186 | /** | ||
187 | * xz_dec_end() - Free the memory allocated for the decoder state | ||
188 | * @s: Decoder state allocated using xz_dec_init(). If s is NULL, | ||
189 | * this function does nothing. | ||
190 | */ | ||
191 | XZ_EXTERN void XZ_FUNC xz_dec_end(struct xz_dec *s); | ||
192 | |||
193 | /* | ||
194 | * Standalone build (userspace build or in-kernel build for boot time use) | ||
195 | * needs a CRC32 implementation. For normal in-kernel use, kernel's own | ||
196 | * CRC32 module is used instead, and users of this module don't need to | ||
197 | * care about the functions below. | ||
198 | */ | ||
199 | #ifndef XZ_INTERNAL_CRC32 | ||
200 | # ifdef __KERNEL__ | ||
201 | # define XZ_INTERNAL_CRC32 0 | ||
202 | # else | ||
203 | # define XZ_INTERNAL_CRC32 1 | ||
204 | # endif | ||
205 | #endif | ||
206 | |||
207 | #if XZ_INTERNAL_CRC32 | ||
208 | /* | ||
209 | * This must be called before any other xz_* function to initialize | ||
210 | * the CRC32 lookup table. | ||
211 | */ | ||
212 | XZ_EXTERN void XZ_FUNC xz_crc32_init(void); | ||
213 | |||
214 | /* | ||
215 | * Update CRC32 value using the polynomial from IEEE-802.3. To start a new | ||
216 | * calculation, the third argument must be zero. To continue the calculation, | ||
217 | * the previously returned value is passed as the third argument. | ||
218 | */ | ||
219 | XZ_EXTERN uint32_t XZ_FUNC xz_crc32( | ||
220 | const uint8_t *buf, size_t size, uint32_t crc); | ||
221 | #endif | ||
222 | #endif | ||
diff --git a/archival/libunarchive/unxz/xz_config.h b/archival/libunarchive/unxz/xz_config.h new file mode 100644 index 000000000..ff90eff26 --- /dev/null +++ b/archival/libunarchive/unxz/xz_config.h | |||
@@ -0,0 +1,121 @@ | |||
1 | /* | ||
2 | * Private includes and definitions for userspace use of XZ Embedded | ||
3 | * | ||
4 | * Author: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * | ||
6 | * This file has been put into the public domain. | ||
7 | * You can do whatever you want with this file. | ||
8 | */ | ||
9 | |||
10 | #ifndef XZ_CONFIG_H | ||
11 | #define XZ_CONFIG_H | ||
12 | |||
13 | /* Uncomment as needed to enable BCJ filter decoders. */ | ||
14 | /* #define XZ_DEC_X86 */ | ||
15 | /* #define XZ_DEC_POWERPC */ | ||
16 | /* #define XZ_DEC_IA64 */ | ||
17 | /* #define XZ_DEC_ARM */ | ||
18 | /* #define XZ_DEC_ARMTHUMB */ | ||
19 | /* #define XZ_DEC_SPARC */ | ||
20 | |||
21 | #include <stdbool.h> | ||
22 | #include <stdlib.h> | ||
23 | #include <string.h> | ||
24 | |||
25 | #include "xz.h" | ||
26 | |||
27 | #define kmalloc(size, flags) malloc(size) | ||
28 | #define kfree(ptr) free(ptr) | ||
29 | #define vmalloc(size) malloc(size) | ||
30 | #define vfree(ptr) free(ptr) | ||
31 | |||
32 | #define memeq(a, b, size) (memcmp(a, b, size) == 0) | ||
33 | #define memzero(buf, size) memset(buf, 0, size) | ||
34 | |||
35 | #define min(x, y) ((x) < (y) ? (x) : (y)) | ||
36 | #define min_t(type, x, y) min(x, y) | ||
37 | |||
38 | /* | ||
39 | * Some functions have been marked with __always_inline to keep the | ||
40 | * performance reasonable even when the compiler is optimizing for | ||
41 | * small code size. You may be able to save a few bytes by #defining | ||
42 | * __always_inline to plain inline, but don't complain if the code | ||
43 | * becomes slow. | ||
44 | * | ||
45 | * NOTE: System headers on GNU/Linux may #define this macro already, | ||
46 | * so if you want to change it, you need to #undef it first. | ||
47 | */ | ||
48 | #ifndef __always_inline | ||
49 | # ifdef __GNUC__ | ||
50 | # define __always_inline \ | ||
51 | inline __attribute__((__always_inline__)) | ||
52 | # else | ||
53 | # define __always_inline inline | ||
54 | # endif | ||
55 | #endif | ||
56 | |||
57 | /* | ||
58 | * Some functions are marked to never be inlined to reduce stack usage. | ||
59 | * If you don't care about stack usage, you may want to modify this so | ||
60 | * that noinline_for_stack is #defined to be empty even when using GCC. | ||
61 | * Doing so may save a few bytes in binary size. | ||
62 | */ | ||
63 | #ifndef noinline_for_stack | ||
64 | # ifdef __GNUC__ | ||
65 | # define noinline_for_stack __attribute__((__noinline__)) | ||
66 | # else | ||
67 | # define noinline_for_stack | ||
68 | # endif | ||
69 | #endif | ||
70 | |||
71 | /* Inline functions to access unaligned unsigned 32-bit integers */ | ||
72 | #ifndef get_unaligned_le32 | ||
73 | static inline uint32_t XZ_FUNC get_unaligned_le32(const uint8_t *buf) | ||
74 | { | ||
75 | return (uint32_t)buf[0] | ||
76 | | ((uint32_t)buf[1] << 8) | ||
77 | | ((uint32_t)buf[2] << 16) | ||
78 | | ((uint32_t)buf[3] << 24); | ||
79 | } | ||
80 | #endif | ||
81 | |||
82 | #ifndef get_unaligned_be32 | ||
83 | static inline uint32_t XZ_FUNC get_unaligned_be32(const uint8_t *buf) | ||
84 | { | ||
85 | return (uint32_t)(buf[0] << 24) | ||
86 | | ((uint32_t)buf[1] << 16) | ||
87 | | ((uint32_t)buf[2] << 8) | ||
88 | | (uint32_t)buf[3]; | ||
89 | } | ||
90 | #endif | ||
91 | |||
92 | #ifndef put_unaligned_le32 | ||
93 | static inline void XZ_FUNC put_unaligned_le32(uint32_t val, uint8_t *buf) | ||
94 | { | ||
95 | buf[0] = (uint8_t)val; | ||
96 | buf[1] = (uint8_t)(val >> 8); | ||
97 | buf[2] = (uint8_t)(val >> 16); | ||
98 | buf[3] = (uint8_t)(val >> 24); | ||
99 | } | ||
100 | #endif | ||
101 | |||
102 | #ifndef put_unaligned_be32 | ||
103 | static inline void XZ_FUNC put_unaligned_be32(uint32_t val, uint8_t *buf) | ||
104 | { | ||
105 | buf[0] = (uint8_t)(val >> 24); | ||
106 | buf[1] = (uint8_t)(val >> 16); | ||
107 | buf[2] = (uint8_t)(val >> 8); | ||
108 | buf[3] = (uint8_t)val; | ||
109 | } | ||
110 | #endif | ||
111 | |||
112 | /* | ||
113 | * Use get_unaligned_le32() also for aligned access for simplicity. On | ||
114 | * little endian systems, #define get_le32(ptr) (*(const uint32_t *)(ptr)) | ||
115 | * could save a few bytes in code size. | ||
116 | */ | ||
117 | #ifndef get_le32 | ||
118 | # define get_le32 get_unaligned_le32 | ||
119 | #endif | ||
120 | |||
121 | #endif | ||
diff --git a/archival/libunarchive/unxz/xz_dec_bcj.c b/archival/libunarchive/unxz/xz_dec_bcj.c new file mode 100644 index 000000000..09162b51f --- /dev/null +++ b/archival/libunarchive/unxz/xz_dec_bcj.c | |||
@@ -0,0 +1,564 @@ | |||
1 | /* | ||
2 | * Branch/Call/Jump (BCJ) filter decoders | ||
3 | * | ||
4 | * Authors: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * Igor Pavlov <http://7-zip.org/> | ||
6 | * | ||
7 | * This file has been put into the public domain. | ||
8 | * You can do whatever you want with this file. | ||
9 | */ | ||
10 | |||
11 | #include "xz_private.h" | ||
12 | |||
13 | /* | ||
14 | * The rest of the file is inside this ifdef. It makes things a little more | ||
15 | * convenient when building without support for any BCJ filters. | ||
16 | */ | ||
17 | #ifdef XZ_DEC_BCJ | ||
18 | |||
19 | struct xz_dec_bcj { | ||
20 | /* Type of the BCJ filter being used */ | ||
21 | enum { | ||
22 | BCJ_X86 = 4, /* x86 or x86-64 */ | ||
23 | BCJ_POWERPC = 5, /* Big endian only */ | ||
24 | BCJ_IA64 = 6, /* Big or little endian */ | ||
25 | BCJ_ARM = 7, /* Little endian only */ | ||
26 | BCJ_ARMTHUMB = 8, /* Little endian only */ | ||
27 | BCJ_SPARC = 9 /* Big or little endian */ | ||
28 | } type; | ||
29 | |||
30 | /* | ||
31 | * Return value of the next filter in the chain. We need to preserve | ||
32 | * this information across calls, because we must not call the next | ||
33 | * filter anymore once it has returned XZ_STREAM_END. | ||
34 | */ | ||
35 | enum xz_ret ret; | ||
36 | |||
37 | /* True if we are operating in single-call mode. */ | ||
38 | bool single_call; | ||
39 | |||
40 | /* | ||
41 | * Absolute position relative to the beginning of the uncompressed | ||
42 | * data (in a single .xz Block). We care only about the lowest 32 | ||
43 | * bits so this doesn't need to be uint64_t even with big files. | ||
44 | */ | ||
45 | uint32_t pos; | ||
46 | |||
47 | /* x86 filter state */ | ||
48 | uint32_t x86_prev_mask; | ||
49 | |||
50 | /* Temporary space to hold the variables from struct xz_buf */ | ||
51 | uint8_t *out; | ||
52 | size_t out_pos; | ||
53 | size_t out_size; | ||
54 | |||
55 | struct { | ||
56 | /* Amount of already filtered data in the beginning of buf */ | ||
57 | size_t filtered; | ||
58 | |||
59 | /* Total amount of data currently stored in buf */ | ||
60 | size_t size; | ||
61 | |||
62 | /* | ||
63 | * Buffer to hold a mix of filtered and unfiltered data. This | ||
64 | * needs to be big enough to hold Alignment + 2 * Look-ahead: | ||
65 | * | ||
66 | * Type Alignment Look-ahead | ||
67 | * x86 1 4 | ||
68 | * PowerPC 4 0 | ||
69 | * IA-64 16 0 | ||
70 | * ARM 4 0 | ||
71 | * ARM-Thumb 2 2 | ||
72 | * SPARC 4 0 | ||
73 | */ | ||
74 | uint8_t buf[16]; | ||
75 | } temp; | ||
76 | }; | ||
77 | |||
78 | #ifdef XZ_DEC_X86 | ||
79 | /* | ||
80 | * This is macro used to test the most significant byte of a memory address | ||
81 | * in an x86 instruction. | ||
82 | */ | ||
83 | #define bcj_x86_test_msbyte(b) ((b) == 0x00 || (b) == 0xFF) | ||
84 | |||
85 | static noinline_for_stack size_t XZ_FUNC bcj_x86( | ||
86 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
87 | { | ||
88 | static const bool mask_to_allowed_status[8] | ||
89 | = { true, true, true, false, true, false, false, false }; | ||
90 | |||
91 | static const uint8_t mask_to_bit_num[8] = { 0, 1, 2, 2, 3, 3, 3, 3 }; | ||
92 | |||
93 | size_t i; | ||
94 | size_t prev_pos = (size_t)-1; | ||
95 | uint32_t prev_mask = s->x86_prev_mask; | ||
96 | uint32_t src; | ||
97 | uint32_t dest; | ||
98 | uint32_t j; | ||
99 | uint8_t b; | ||
100 | |||
101 | if (size <= 4) | ||
102 | return 0; | ||
103 | |||
104 | size -= 4; | ||
105 | for (i = 0; i < size; ++i) { | ||
106 | if ((buf[i] & 0xFE) != 0xE8) | ||
107 | continue; | ||
108 | |||
109 | prev_pos = i - prev_pos; | ||
110 | if (prev_pos > 3) { | ||
111 | prev_mask = 0; | ||
112 | } else { | ||
113 | prev_mask = (prev_mask << (prev_pos - 1)) & 7; | ||
114 | if (prev_mask != 0) { | ||
115 | b = buf[i + 4 - mask_to_bit_num[prev_mask]]; | ||
116 | if (!mask_to_allowed_status[prev_mask] | ||
117 | || bcj_x86_test_msbyte(b)) { | ||
118 | prev_pos = i; | ||
119 | prev_mask = (prev_mask << 1) | 1; | ||
120 | continue; | ||
121 | } | ||
122 | } | ||
123 | } | ||
124 | |||
125 | prev_pos = i; | ||
126 | |||
127 | if (bcj_x86_test_msbyte(buf[i + 4])) { | ||
128 | src = get_unaligned_le32(buf + i + 1); | ||
129 | while (true) { | ||
130 | dest = src - (s->pos + (uint32_t)i + 5); | ||
131 | if (prev_mask == 0) | ||
132 | break; | ||
133 | |||
134 | j = mask_to_bit_num[prev_mask] * 8; | ||
135 | b = (uint8_t)(dest >> (24 - j)); | ||
136 | if (!bcj_x86_test_msbyte(b)) | ||
137 | break; | ||
138 | |||
139 | src = dest ^ (((uint32_t)1 << (32 - j)) - 1); | ||
140 | } | ||
141 | |||
142 | dest &= 0x01FFFFFF; | ||
143 | dest |= (uint32_t)0 - (dest & 0x01000000); | ||
144 | put_unaligned_le32(dest, buf + i + 1); | ||
145 | i += 4; | ||
146 | } else { | ||
147 | prev_mask = (prev_mask << 1) | 1; | ||
148 | } | ||
149 | } | ||
150 | |||
151 | prev_pos = i - prev_pos; | ||
152 | s->x86_prev_mask = prev_pos > 3 ? 0 : prev_mask << (prev_pos - 1); | ||
153 | return i; | ||
154 | } | ||
155 | #endif | ||
156 | |||
157 | #ifdef XZ_DEC_POWERPC | ||
158 | static noinline_for_stack size_t XZ_FUNC bcj_powerpc( | ||
159 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
160 | { | ||
161 | size_t i; | ||
162 | uint32_t instr; | ||
163 | |||
164 | for (i = 0; i + 4 <= size; i += 4) { | ||
165 | instr = get_unaligned_be32(buf + i); | ||
166 | if ((instr & 0xFC000003) == 0x48000001) { | ||
167 | instr &= 0x03FFFFFC; | ||
168 | instr -= s->pos + (uint32_t)i; | ||
169 | instr &= 0x03FFFFFC; | ||
170 | instr |= 0x48000001; | ||
171 | put_unaligned_be32(instr, buf + i); | ||
172 | } | ||
173 | } | ||
174 | |||
175 | return i; | ||
176 | } | ||
177 | #endif | ||
178 | |||
179 | #ifdef XZ_DEC_IA64 | ||
180 | static noinline_for_stack size_t XZ_FUNC bcj_ia64( | ||
181 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
182 | { | ||
183 | static const uint8_t branch_table[32] = { | ||
184 | 0, 0, 0, 0, 0, 0, 0, 0, | ||
185 | 0, 0, 0, 0, 0, 0, 0, 0, | ||
186 | 4, 4, 6, 6, 0, 0, 7, 7, | ||
187 | 4, 4, 0, 0, 4, 4, 0, 0 | ||
188 | }; | ||
189 | |||
190 | /* | ||
191 | * The local variables take a little bit stack space, but it's less | ||
192 | * than what LZMA2 decoder takes, so it doesn't make sense to reduce | ||
193 | * stack usage here without doing that for the LZMA2 decoder too. | ||
194 | */ | ||
195 | |||
196 | /* Loop counters */ | ||
197 | size_t i; | ||
198 | size_t j; | ||
199 | |||
200 | /* Instruction slot (0, 1, or 2) in the 128-bit instruction word */ | ||
201 | uint32_t slot; | ||
202 | |||
203 | /* Bitwise offset of the instruction indicated by slot */ | ||
204 | uint32_t bit_pos; | ||
205 | |||
206 | /* bit_pos split into byte and bit parts */ | ||
207 | uint32_t byte_pos; | ||
208 | uint32_t bit_res; | ||
209 | |||
210 | /* Address part of an instruction */ | ||
211 | uint32_t addr; | ||
212 | |||
213 | /* Mask used to detect which instructions to convert */ | ||
214 | uint32_t mask; | ||
215 | |||
216 | /* 41-bit instruction stored somewhere in the lowest 48 bits */ | ||
217 | uint64_t instr; | ||
218 | |||
219 | /* Instruction normalized with bit_res for easier manipulation */ | ||
220 | uint64_t norm; | ||
221 | |||
222 | for (i = 0; i + 16 <= size; i += 16) { | ||
223 | mask = branch_table[buf[i] & 0x1F]; | ||
224 | for (slot = 0, bit_pos = 5; slot < 3; ++slot, bit_pos += 41) { | ||
225 | if (((mask >> slot) & 1) == 0) | ||
226 | continue; | ||
227 | |||
228 | byte_pos = bit_pos >> 3; | ||
229 | bit_res = bit_pos & 7; | ||
230 | instr = 0; | ||
231 | for (j = 0; j < 6; ++j) | ||
232 | instr |= (uint64_t)(buf[i + j + byte_pos]) | ||
233 | << (8 * j); | ||
234 | |||
235 | norm = instr >> bit_res; | ||
236 | |||
237 | if (((norm >> 37) & 0x0F) == 0x05 | ||
238 | && ((norm >> 9) & 0x07) == 0) { | ||
239 | addr = (norm >> 13) & 0x0FFFFF; | ||
240 | addr |= ((uint32_t)(norm >> 36) & 1) << 20; | ||
241 | addr <<= 4; | ||
242 | addr -= s->pos + (uint32_t)i; | ||
243 | addr >>= 4; | ||
244 | |||
245 | norm &= ~((uint64_t)0x8FFFFF << 13); | ||
246 | norm |= (uint64_t)(addr & 0x0FFFFF) << 13; | ||
247 | norm |= (uint64_t)(addr & 0x100000) | ||
248 | << (36 - 20); | ||
249 | |||
250 | instr &= (1 << bit_res) - 1; | ||
251 | instr |= norm << bit_res; | ||
252 | |||
253 | for (j = 0; j < 6; j++) | ||
254 | buf[i + j + byte_pos] | ||
255 | = (uint8_t)(instr >> (8 * j)); | ||
256 | } | ||
257 | } | ||
258 | } | ||
259 | |||
260 | return i; | ||
261 | } | ||
262 | #endif | ||
263 | |||
264 | #ifdef XZ_DEC_ARM | ||
265 | static noinline_for_stack size_t XZ_FUNC bcj_arm( | ||
266 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
267 | { | ||
268 | size_t i; | ||
269 | uint32_t addr; | ||
270 | |||
271 | for (i = 0; i + 4 <= size; i += 4) { | ||
272 | if (buf[i + 3] == 0xEB) { | ||
273 | addr = (uint32_t)buf[i] | ((uint32_t)buf[i + 1] << 8) | ||
274 | | ((uint32_t)buf[i + 2] << 16); | ||
275 | addr <<= 2; | ||
276 | addr -= s->pos + (uint32_t)i + 8; | ||
277 | addr >>= 2; | ||
278 | buf[i] = (uint8_t)addr; | ||
279 | buf[i + 1] = (uint8_t)(addr >> 8); | ||
280 | buf[i + 2] = (uint8_t)(addr >> 16); | ||
281 | } | ||
282 | } | ||
283 | |||
284 | return i; | ||
285 | } | ||
286 | #endif | ||
287 | |||
288 | #ifdef XZ_DEC_ARMTHUMB | ||
289 | static noinline_for_stack size_t XZ_FUNC bcj_armthumb( | ||
290 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
291 | { | ||
292 | size_t i; | ||
293 | uint32_t addr; | ||
294 | |||
295 | for (i = 0; i + 4 <= size; i += 2) { | ||
296 | if ((buf[i + 1] & 0xF8) == 0xF0 | ||
297 | && (buf[i + 3] & 0xF8) == 0xF8) { | ||
298 | addr = (((uint32_t)buf[i + 1] & 0x07) << 19) | ||
299 | | ((uint32_t)buf[i] << 11) | ||
300 | | (((uint32_t)buf[i + 3] & 0x07) << 8) | ||
301 | | (uint32_t)buf[i + 2]; | ||
302 | addr <<= 1; | ||
303 | addr -= s->pos + (uint32_t)i + 4; | ||
304 | addr >>= 1; | ||
305 | buf[i + 1] = (uint8_t)(0xF0 | ((addr >> 19) & 0x07)); | ||
306 | buf[i] = (uint8_t)(addr >> 11); | ||
307 | buf[i + 3] = (uint8_t)(0xF8 | ((addr >> 8) & 0x07)); | ||
308 | buf[i + 2] = (uint8_t)addr; | ||
309 | i += 2; | ||
310 | } | ||
311 | } | ||
312 | |||
313 | return i; | ||
314 | } | ||
315 | #endif | ||
316 | |||
317 | #ifdef XZ_DEC_SPARC | ||
318 | static noinline_for_stack size_t XZ_FUNC bcj_sparc( | ||
319 | struct xz_dec_bcj *s, uint8_t *buf, size_t size) | ||
320 | { | ||
321 | size_t i; | ||
322 | uint32_t instr; | ||
323 | |||
324 | for (i = 0; i + 4 <= size; i += 4) { | ||
325 | instr = get_unaligned_be32(buf + i); | ||
326 | if ((instr >> 22) == 0x100 || (instr >> 22) == 0x1FF) { | ||
327 | instr <<= 2; | ||
328 | instr -= s->pos + (uint32_t)i; | ||
329 | instr >>= 2; | ||
330 | instr = ((uint32_t)0x40000000 - (instr & 0x400000)) | ||
331 | | 0x40000000 | (instr & 0x3FFFFF); | ||
332 | put_unaligned_be32(instr, buf + i); | ||
333 | } | ||
334 | } | ||
335 | |||
336 | return i; | ||
337 | } | ||
338 | #endif | ||
339 | |||
340 | /* | ||
341 | * Apply the selected BCJ filter. Update *pos and s->pos to match the amount | ||
342 | * of data that got filtered. | ||
343 | * | ||
344 | * NOTE: This is implemented as a switch statement to avoid using function | ||
345 | * pointers, which could be problematic in the kernel boot code, which must | ||
346 | * avoid pointers to static data (at least on x86). | ||
347 | */ | ||
348 | static void XZ_FUNC bcj_apply(struct xz_dec_bcj *s, | ||
349 | uint8_t *buf, size_t *pos, size_t size) | ||
350 | { | ||
351 | size_t filtered; | ||
352 | |||
353 | buf += *pos; | ||
354 | size -= *pos; | ||
355 | |||
356 | switch (s->type) { | ||
357 | #ifdef XZ_DEC_X86 | ||
358 | case BCJ_X86: | ||
359 | filtered = bcj_x86(s, buf, size); | ||
360 | break; | ||
361 | #endif | ||
362 | #ifdef XZ_DEC_POWERPC | ||
363 | case BCJ_POWERPC: | ||
364 | filtered = bcj_powerpc(s, buf, size); | ||
365 | break; | ||
366 | #endif | ||
367 | #ifdef XZ_DEC_IA64 | ||
368 | case BCJ_IA64: | ||
369 | filtered = bcj_ia64(s, buf, size); | ||
370 | break; | ||
371 | #endif | ||
372 | #ifdef XZ_DEC_ARM | ||
373 | case BCJ_ARM: | ||
374 | filtered = bcj_arm(s, buf, size); | ||
375 | break; | ||
376 | #endif | ||
377 | #ifdef XZ_DEC_ARMTHUMB | ||
378 | case BCJ_ARMTHUMB: | ||
379 | filtered = bcj_armthumb(s, buf, size); | ||
380 | break; | ||
381 | #endif | ||
382 | #ifdef XZ_DEC_SPARC | ||
383 | case BCJ_SPARC: | ||
384 | filtered = bcj_sparc(s, buf, size); | ||
385 | break; | ||
386 | #endif | ||
387 | default: | ||
388 | /* Never reached but silence compiler warnings. */ | ||
389 | filtered = 0; | ||
390 | break; | ||
391 | } | ||
392 | |||
393 | *pos += filtered; | ||
394 | s->pos += filtered; | ||
395 | } | ||
396 | |||
397 | /* | ||
398 | * Flush pending filtered data from temp to the output buffer. | ||
399 | * Move the remaining mixture of possibly filtered and unfiltered | ||
400 | * data to the beginning of temp. | ||
401 | */ | ||
402 | static void XZ_FUNC bcj_flush(struct xz_dec_bcj *s, struct xz_buf *b) | ||
403 | { | ||
404 | size_t copy_size; | ||
405 | |||
406 | copy_size = min_t(size_t, s->temp.filtered, b->out_size - b->out_pos); | ||
407 | memcpy(b->out + b->out_pos, s->temp.buf, copy_size); | ||
408 | b->out_pos += copy_size; | ||
409 | |||
410 | s->temp.filtered -= copy_size; | ||
411 | s->temp.size -= copy_size; | ||
412 | memmove(s->temp.buf, s->temp.buf + copy_size, s->temp.size); | ||
413 | } | ||
414 | |||
415 | /* | ||
416 | * The BCJ filter functions are primitive in sense that they process the | ||
417 | * data in chunks of 1-16 bytes. To hide this issue, this function does | ||
418 | * some buffering. | ||
419 | */ | ||
420 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_bcj_run(struct xz_dec_bcj *s, | ||
421 | struct xz_dec_lzma2 *lzma2, struct xz_buf *b) | ||
422 | { | ||
423 | size_t out_start; | ||
424 | |||
425 | /* | ||
426 | * Flush pending already filtered data to the output buffer. Return | ||
427 | * immediatelly if we couldn't flush everything, or if the next | ||
428 | * filter in the chain had already returned XZ_STREAM_END. | ||
429 | */ | ||
430 | if (s->temp.filtered > 0) { | ||
431 | bcj_flush(s, b); | ||
432 | if (s->temp.filtered > 0) | ||
433 | return XZ_OK; | ||
434 | |||
435 | if (s->ret == XZ_STREAM_END) | ||
436 | return XZ_STREAM_END; | ||
437 | } | ||
438 | |||
439 | /* | ||
440 | * If we have more output space than what is currently pending in | ||
441 | * temp, copy the unfiltered data from temp to the output buffer | ||
442 | * and try to fill the output buffer by decoding more data from the | ||
443 | * next filter in the chain. Apply the BCJ filter on the new data | ||
444 | * in the output buffer. If everything cannot be filtered, copy it | ||
445 | * to temp and rewind the output buffer position accordingly. | ||
446 | */ | ||
447 | if (s->temp.size < b->out_size - b->out_pos) { | ||
448 | out_start = b->out_pos; | ||
449 | memcpy(b->out + b->out_pos, s->temp.buf, s->temp.size); | ||
450 | b->out_pos += s->temp.size; | ||
451 | |||
452 | s->ret = xz_dec_lzma2_run(lzma2, b); | ||
453 | if (s->ret != XZ_STREAM_END | ||
454 | && (s->ret != XZ_OK || s->single_call)) | ||
455 | return s->ret; | ||
456 | |||
457 | bcj_apply(s, b->out, &out_start, b->out_pos); | ||
458 | |||
459 | /* | ||
460 | * As an exception, if the next filter returned XZ_STREAM_END, | ||
461 | * we can do that too, since the last few bytes that remain | ||
462 | * unfiltered are meant to remain unfiltered. | ||
463 | */ | ||
464 | if (s->ret == XZ_STREAM_END) | ||
465 | return XZ_STREAM_END; | ||
466 | |||
467 | s->temp.size = b->out_pos - out_start; | ||
468 | b->out_pos -= s->temp.size; | ||
469 | memcpy(s->temp.buf, b->out + b->out_pos, s->temp.size); | ||
470 | } | ||
471 | |||
472 | /* | ||
473 | * If we have unfiltered data in temp, try to fill by decoding more | ||
474 | * data from the next filter. Apply the BCJ filter on temp. Then we | ||
475 | * hopefully can fill the actual output buffer by copying filtered | ||
476 | * data from temp. A mix of filtered and unfiltered data may be left | ||
477 | * in temp; it will be taken care on the next call to this function. | ||
478 | */ | ||
479 | if (s->temp.size > 0) { | ||
480 | /* Make b->out{,_pos,_size} temporarily point to s->temp. */ | ||
481 | s->out = b->out; | ||
482 | s->out_pos = b->out_pos; | ||
483 | s->out_size = b->out_size; | ||
484 | b->out = s->temp.buf; | ||
485 | b->out_pos = s->temp.size; | ||
486 | b->out_size = sizeof(s->temp.buf); | ||
487 | |||
488 | s->ret = xz_dec_lzma2_run(lzma2, b); | ||
489 | |||
490 | s->temp.size = b->out_pos; | ||
491 | b->out = s->out; | ||
492 | b->out_pos = s->out_pos; | ||
493 | b->out_size = s->out_size; | ||
494 | |||
495 | if (s->ret != XZ_OK && s->ret != XZ_STREAM_END) | ||
496 | return s->ret; | ||
497 | |||
498 | bcj_apply(s, s->temp.buf, &s->temp.filtered, s->temp.size); | ||
499 | |||
500 | /* | ||
501 | * If the next filter returned XZ_STREAM_END, we mark that | ||
502 | * everything is filtered, since the last unfiltered bytes | ||
503 | * of the stream are meant to be left as is. | ||
504 | */ | ||
505 | if (s->ret == XZ_STREAM_END) | ||
506 | s->temp.filtered = s->temp.size; | ||
507 | |||
508 | bcj_flush(s, b); | ||
509 | if (s->temp.filtered > 0) | ||
510 | return XZ_OK; | ||
511 | } | ||
512 | |||
513 | return s->ret; | ||
514 | } | ||
515 | |||
516 | XZ_EXTERN struct xz_dec_bcj * XZ_FUNC xz_dec_bcj_create(bool single_call) | ||
517 | { | ||
518 | struct xz_dec_bcj *s = kmalloc(sizeof(*s), GFP_KERNEL); | ||
519 | if (s != NULL) | ||
520 | s->single_call = single_call; | ||
521 | |||
522 | return s; | ||
523 | } | ||
524 | |||
525 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_bcj_reset( | ||
526 | struct xz_dec_bcj *s, uint8_t id) | ||
527 | { | ||
528 | switch (id) { | ||
529 | #ifdef XZ_DEC_X86 | ||
530 | case BCJ_X86: | ||
531 | #endif | ||
532 | #ifdef XZ_DEC_POWERPC | ||
533 | case BCJ_POWERPC: | ||
534 | #endif | ||
535 | #ifdef XZ_DEC_IA64 | ||
536 | case BCJ_IA64: | ||
537 | #endif | ||
538 | #ifdef XZ_DEC_ARM | ||
539 | case BCJ_ARM: | ||
540 | #endif | ||
541 | #ifdef XZ_DEC_ARMTHUMB | ||
542 | case BCJ_ARMTHUMB: | ||
543 | #endif | ||
544 | #ifdef XZ_DEC_SPARC | ||
545 | case BCJ_SPARC: | ||
546 | #endif | ||
547 | break; | ||
548 | |||
549 | default: | ||
550 | /* Unsupported Filter ID */ | ||
551 | return XZ_OPTIONS_ERROR; | ||
552 | } | ||
553 | |||
554 | s->type = id; | ||
555 | s->ret = XZ_OK; | ||
556 | s->pos = 0; | ||
557 | s->x86_prev_mask = 0; | ||
558 | s->temp.filtered = 0; | ||
559 | s->temp.size = 0; | ||
560 | |||
561 | return XZ_OK; | ||
562 | } | ||
563 | |||
564 | #endif | ||
diff --git a/archival/libunarchive/unxz/xz_dec_lzma2.c b/archival/libunarchive/unxz/xz_dec_lzma2.c new file mode 100644 index 000000000..37de6fc32 --- /dev/null +++ b/archival/libunarchive/unxz/xz_dec_lzma2.c | |||
@@ -0,0 +1,1155 @@ | |||
1 | /* | ||
2 | * LZMA2 decoder | ||
3 | * | ||
4 | * Authors: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * Igor Pavlov <http://7-zip.org/> | ||
6 | * | ||
7 | * This file has been put into the public domain. | ||
8 | * You can do whatever you want with this file. | ||
9 | */ | ||
10 | |||
11 | #include "xz_private.h" | ||
12 | #include "xz_lzma2.h" | ||
13 | |||
14 | /* | ||
15 | * Range decoder initialization eats the first five bytes of each LZMA chunk. | ||
16 | */ | ||
17 | #define RC_INIT_BYTES 5 | ||
18 | |||
19 | /* | ||
20 | * Minimum number of usable input buffer to safely decode one LZMA symbol. | ||
21 | * The worst case is that we decode 22 bits using probabilities and 26 | ||
22 | * direct bits. This may decode at maximum of 20 bytes of input. However, | ||
23 | * lzma_main() does an extra normalization before returning, thus we | ||
24 | * need to put 21 here. | ||
25 | */ | ||
26 | #define LZMA_IN_REQUIRED 21 | ||
27 | |||
28 | /* | ||
29 | * Dictionary (history buffer) | ||
30 | * | ||
31 | * These are always true: | ||
32 | * start <= pos <= full <= end | ||
33 | * pos <= limit <= end | ||
34 | * | ||
35 | * In multi-call mode, also these are true: | ||
36 | * end == size | ||
37 | * size <= allocated | ||
38 | * | ||
39 | * Most of these variables are size_t to support single-call mode, | ||
40 | * in which the dictionary variables address the actual output | ||
41 | * buffer directly. | ||
42 | */ | ||
43 | struct dictionary { | ||
44 | /* Beginning of the history buffer */ | ||
45 | uint8_t *buf; | ||
46 | |||
47 | /* Old position in buf (before decoding more data) */ | ||
48 | size_t start; | ||
49 | |||
50 | /* Position in buf */ | ||
51 | size_t pos; | ||
52 | |||
53 | /* | ||
54 | * How full dictionary is. This is used to detect corrupt input that | ||
55 | * would read beyond the beginning of the uncompressed stream. | ||
56 | */ | ||
57 | size_t full; | ||
58 | |||
59 | /* Write limit; we don't write to buf[limit] or later bytes. */ | ||
60 | size_t limit; | ||
61 | |||
62 | /* | ||
63 | * End of the dictionary buffer. In multi-call mode, this is | ||
64 | * the same as the dictionary size. In single-call mode, this | ||
65 | * indicates the size of the output buffer. | ||
66 | */ | ||
67 | size_t end; | ||
68 | |||
69 | /* | ||
70 | * Size of the dictionary as specified in Block Header. This is used | ||
71 | * together with "full" to detect corrupt input that would make us | ||
72 | * read beyond the beginning of the uncompressed stream. | ||
73 | */ | ||
74 | uint32_t size; | ||
75 | |||
76 | /* | ||
77 | * Amount of memory allocated for the dictionary. A special | ||
78 | * value of zero indicates that we are in single-call mode, | ||
79 | * where the output buffer works as the dictionary. | ||
80 | */ | ||
81 | uint32_t allocated; | ||
82 | }; | ||
83 | |||
84 | /* Range decoder */ | ||
85 | struct rc_dec { | ||
86 | uint32_t range; | ||
87 | uint32_t code; | ||
88 | |||
89 | /* | ||
90 | * Number of initializing bytes remaining to be read | ||
91 | * by rc_read_init(). | ||
92 | */ | ||
93 | uint32_t init_bytes_left; | ||
94 | |||
95 | /* | ||
96 | * Buffer from which we read our input. It can be either | ||
97 | * temp.buf or the caller-provided input buffer. | ||
98 | */ | ||
99 | const uint8_t *in; | ||
100 | size_t in_pos; | ||
101 | size_t in_limit; | ||
102 | }; | ||
103 | |||
104 | /* Probabilities for a length decoder. */ | ||
105 | struct lzma_len_dec { | ||
106 | /* Probability of match length being at least 10 */ | ||
107 | uint16_t choice; | ||
108 | |||
109 | /* Probability of match length being at least 18 */ | ||
110 | uint16_t choice2; | ||
111 | |||
112 | /* Probabilities for match lengths 2-9 */ | ||
113 | uint16_t low[POS_STATES_MAX][LEN_LOW_SYMBOLS]; | ||
114 | |||
115 | /* Probabilities for match lengths 10-17 */ | ||
116 | uint16_t mid[POS_STATES_MAX][LEN_MID_SYMBOLS]; | ||
117 | |||
118 | /* Probabilities for match lengths 18-273 */ | ||
119 | uint16_t high[LEN_HIGH_SYMBOLS]; | ||
120 | }; | ||
121 | |||
122 | struct lzma_dec { | ||
123 | /* | ||
124 | * LZMA properties or related bit masks (number of literal | ||
125 | * context bits, a mask dervied from the number of literal | ||
126 | * position bits, and a mask dervied from the number | ||
127 | * position bits) | ||
128 | */ | ||
129 | uint32_t lc; | ||
130 | uint32_t literal_pos_mask; /* (1 << lp) - 1 */ | ||
131 | uint32_t pos_mask; /* (1 << pb) - 1 */ | ||
132 | |||
133 | /* Types of the most recently seen LZMA symbols */ | ||
134 | enum lzma_state state; | ||
135 | |||
136 | /* Distances of latest four matches */ | ||
137 | uint32_t rep0; | ||
138 | uint32_t rep1; | ||
139 | uint32_t rep2; | ||
140 | uint32_t rep3; | ||
141 | |||
142 | /* | ||
143 | * Length of a match. This is updated so that dict_repeat can | ||
144 | * be called again to finish repeating the whole match. | ||
145 | */ | ||
146 | uint32_t len; | ||
147 | |||
148 | /* If 1, it's a match. Otherwise it's a single 8-bit literal. */ | ||
149 | uint16_t is_match[STATES][POS_STATES_MAX]; | ||
150 | |||
151 | /* If 1, it's a repeated match. The distance is one of rep0 .. rep3. */ | ||
152 | uint16_t is_rep[STATES]; | ||
153 | |||
154 | /* | ||
155 | * If 0, distance of a repeated match is rep0. | ||
156 | * Otherwise check is_rep1. | ||
157 | */ | ||
158 | uint16_t is_rep0[STATES]; | ||
159 | |||
160 | /* | ||
161 | * If 0, distance of a repeated match is rep1. | ||
162 | * Otherwise check is_rep2. | ||
163 | */ | ||
164 | uint16_t is_rep1[STATES]; | ||
165 | |||
166 | /* If 0, distance of a repeated match is rep2. Otherwise it is rep3. */ | ||
167 | uint16_t is_rep2[STATES]; | ||
168 | |||
169 | /* | ||
170 | * If 1, the repeated match has length of one byte. Otherwise | ||
171 | * the length is decoded from rep_len_decoder. | ||
172 | */ | ||
173 | uint16_t is_rep0_long[STATES][POS_STATES_MAX]; | ||
174 | |||
175 | /* | ||
176 | * Probability tree for the highest two bits of the match | ||
177 | * distance. There is a separate probability tree for match | ||
178 | * lengths of 2 (i.e. MATCH_LEN_MIN), 3, 4, and [5, 273]. | ||
179 | */ | ||
180 | uint16_t dist_slot[DIST_STATES][DIST_SLOTS]; | ||
181 | |||
182 | /* | ||
183 | * Probility trees for additional bits for match distance | ||
184 | * when the distance is in the range [4, 127]. | ||
185 | */ | ||
186 | uint16_t dist_special[FULL_DISTANCES - DIST_MODEL_END]; | ||
187 | |||
188 | /* | ||
189 | * Probability tree for the lowest four bits of a match | ||
190 | * distance that is equal to or greater than 128. | ||
191 | */ | ||
192 | uint16_t dist_align[ALIGN_SIZE]; | ||
193 | |||
194 | /* Length of a normal match */ | ||
195 | struct lzma_len_dec match_len_dec; | ||
196 | |||
197 | /* Length of a repeated match */ | ||
198 | struct lzma_len_dec rep_len_dec; | ||
199 | |||
200 | /* Probabilities of literals */ | ||
201 | uint16_t literal[LITERAL_CODERS_MAX][LITERAL_CODER_SIZE]; | ||
202 | }; | ||
203 | |||
204 | struct xz_dec_lzma2 { | ||
205 | /* LZMA2 */ | ||
206 | struct { | ||
207 | /* Position in xz_dec_lzma2_run(). */ | ||
208 | enum lzma2_seq { | ||
209 | SEQ_CONTROL, | ||
210 | SEQ_UNCOMPRESSED_1, | ||
211 | SEQ_UNCOMPRESSED_2, | ||
212 | SEQ_COMPRESSED_0, | ||
213 | SEQ_COMPRESSED_1, | ||
214 | SEQ_PROPERTIES, | ||
215 | SEQ_LZMA_PREPARE, | ||
216 | SEQ_LZMA_RUN, | ||
217 | SEQ_COPY | ||
218 | } sequence; | ||
219 | |||
220 | /* | ||
221 | * Next position after decoding the compressed size of | ||
222 | * the chunk. | ||
223 | */ | ||
224 | enum lzma2_seq next_sequence; | ||
225 | |||
226 | /* Uncompressed size of LZMA chunk (2 MiB at maximum) */ | ||
227 | uint32_t uncompressed; | ||
228 | |||
229 | /* | ||
230 | * Compressed size of LZMA chunk or compressed/uncompressed | ||
231 | * size of uncompressed chunk (64 KiB at maximum) | ||
232 | */ | ||
233 | uint32_t compressed; | ||
234 | |||
235 | /* | ||
236 | * True if dictionary reset is needed. This is false before | ||
237 | * the first chunk (LZMA or uncompressed). | ||
238 | */ | ||
239 | bool need_dict_reset; | ||
240 | |||
241 | /* | ||
242 | * True if new LZMA properties are needed. This is false | ||
243 | * before the first LZMA chunk. | ||
244 | */ | ||
245 | bool need_props; | ||
246 | } lzma2; | ||
247 | |||
248 | /* | ||
249 | * Temporary buffer which holds small number of input bytes between | ||
250 | * decoder calls. See lzma2_lzma() for details. | ||
251 | */ | ||
252 | struct { | ||
253 | uint32_t size; | ||
254 | uint8_t buf[3 * LZMA_IN_REQUIRED]; | ||
255 | } temp; | ||
256 | |||
257 | struct dictionary dict; | ||
258 | struct rc_dec rc; | ||
259 | struct lzma_dec lzma; | ||
260 | }; | ||
261 | |||
262 | /************** | ||
263 | * Dictionary * | ||
264 | **************/ | ||
265 | |||
266 | /* | ||
267 | * Reset the dictionary state. When in single-call mode, set up the beginning | ||
268 | * of the dictionary to point to the actual output buffer. | ||
269 | */ | ||
270 | static void XZ_FUNC dict_reset(struct dictionary *dict, struct xz_buf *b) | ||
271 | { | ||
272 | if (dict->allocated == 0) { | ||
273 | dict->buf = b->out + b->out_pos; | ||
274 | dict->end = b->out_size - b->out_pos; | ||
275 | } | ||
276 | |||
277 | dict->start = 0; | ||
278 | dict->pos = 0; | ||
279 | dict->limit = 0; | ||
280 | dict->full = 0; | ||
281 | } | ||
282 | |||
283 | /* Set dictionary write limit */ | ||
284 | static void XZ_FUNC dict_limit(struct dictionary *dict, size_t out_max) | ||
285 | { | ||
286 | if (dict->end - dict->pos <= out_max) | ||
287 | dict->limit = dict->end; | ||
288 | else | ||
289 | dict->limit = dict->pos + out_max; | ||
290 | } | ||
291 | |||
292 | /* Return true if at least one byte can be written into the dictionary. */ | ||
293 | static __always_inline bool XZ_FUNC dict_has_space(const struct dictionary *dict) | ||
294 | { | ||
295 | return dict->pos < dict->limit; | ||
296 | } | ||
297 | |||
298 | /* | ||
299 | * Get a byte from the dictionary at the given distance. The distance is | ||
300 | * assumed to valid, or as a special case, zero when the dictionary is | ||
301 | * still empty. This special case is needed for single-call decoding to | ||
302 | * avoid writing a '\0' to the end of the destination buffer. | ||
303 | */ | ||
304 | static __always_inline uint32_t XZ_FUNC dict_get( | ||
305 | const struct dictionary *dict, uint32_t dist) | ||
306 | { | ||
307 | size_t offset = dict->pos - dist - 1; | ||
308 | |||
309 | if (dist >= dict->pos) | ||
310 | offset += dict->end; | ||
311 | |||
312 | return dict->full > 0 ? dict->buf[offset] : 0; | ||
313 | } | ||
314 | |||
315 | /* | ||
316 | * Put one byte into the dictionary. It is assumed that there is space for it. | ||
317 | */ | ||
318 | static inline void XZ_FUNC dict_put(struct dictionary *dict, uint8_t byte) | ||
319 | { | ||
320 | dict->buf[dict->pos++] = byte; | ||
321 | |||
322 | if (dict->full < dict->pos) | ||
323 | dict->full = dict->pos; | ||
324 | } | ||
325 | |||
326 | /* | ||
327 | * Repeat given number of bytes from the given distance. If the distance is | ||
328 | * invalid, false is returned. On success, true is returned and *len is | ||
329 | * updated to indicate how many bytes were left to be repeated. | ||
330 | */ | ||
331 | static bool XZ_FUNC dict_repeat( | ||
332 | struct dictionary *dict, uint32_t *len, uint32_t dist) | ||
333 | { | ||
334 | size_t back; | ||
335 | uint32_t left; | ||
336 | |||
337 | if (dist >= dict->full || dist >= dict->size) | ||
338 | return false; | ||
339 | |||
340 | left = min_t(size_t, dict->limit - dict->pos, *len); | ||
341 | *len -= left; | ||
342 | |||
343 | back = dict->pos - dist - 1; | ||
344 | if (dist >= dict->pos) | ||
345 | back += dict->end; | ||
346 | |||
347 | do { | ||
348 | dict->buf[dict->pos++] = dict->buf[back++]; | ||
349 | if (back == dict->end) | ||
350 | back = 0; | ||
351 | } while (--left > 0); | ||
352 | |||
353 | if (dict->full < dict->pos) | ||
354 | dict->full = dict->pos; | ||
355 | |||
356 | return true; | ||
357 | } | ||
358 | |||
359 | /* Copy uncompressed data as is from input to dictionary and output buffers. */ | ||
360 | static void XZ_FUNC dict_uncompressed( | ||
361 | struct dictionary *dict, struct xz_buf *b, uint32_t *left) | ||
362 | { | ||
363 | size_t copy_size; | ||
364 | |||
365 | while (*left > 0 && b->in_pos < b->in_size | ||
366 | && b->out_pos < b->out_size) { | ||
367 | copy_size = min(b->in_size - b->in_pos, | ||
368 | b->out_size - b->out_pos); | ||
369 | if (copy_size > dict->end - dict->pos) | ||
370 | copy_size = dict->end - dict->pos; | ||
371 | if (copy_size > *left) | ||
372 | copy_size = *left; | ||
373 | |||
374 | *left -= copy_size; | ||
375 | |||
376 | memcpy(dict->buf + dict->pos, b->in + b->in_pos, copy_size); | ||
377 | dict->pos += copy_size; | ||
378 | |||
379 | if (dict->full < dict->pos) | ||
380 | dict->full = dict->pos; | ||
381 | |||
382 | if (dict->allocated != 0) { | ||
383 | if (dict->pos == dict->end) | ||
384 | dict->pos = 0; | ||
385 | |||
386 | memcpy(b->out + b->out_pos, b->in + b->in_pos, | ||
387 | copy_size); | ||
388 | } | ||
389 | |||
390 | dict->start = dict->pos; | ||
391 | |||
392 | b->out_pos += copy_size; | ||
393 | b->in_pos += copy_size; | ||
394 | |||
395 | } | ||
396 | } | ||
397 | |||
398 | /* | ||
399 | * Flush pending data from dictionary to b->out. It is assumed that there is | ||
400 | * enough space in b->out. This is guaranteed because caller uses dict_limit() | ||
401 | * before decoding data into the dictionary. | ||
402 | */ | ||
403 | static uint32_t XZ_FUNC dict_flush(struct dictionary *dict, struct xz_buf *b) | ||
404 | { | ||
405 | size_t copy_size = dict->pos - dict->start; | ||
406 | |||
407 | if (dict->allocated != 0) { | ||
408 | if (dict->pos == dict->end) | ||
409 | dict->pos = 0; | ||
410 | |||
411 | memcpy(b->out + b->out_pos, dict->buf + dict->start, | ||
412 | copy_size); | ||
413 | } | ||
414 | |||
415 | dict->start = dict->pos; | ||
416 | b->out_pos += copy_size; | ||
417 | return copy_size; | ||
418 | } | ||
419 | |||
420 | /***************** | ||
421 | * Range decoder * | ||
422 | *****************/ | ||
423 | |||
424 | /* Reset the range decoder. */ | ||
425 | static __always_inline void XZ_FUNC rc_reset(struct rc_dec *rc) | ||
426 | { | ||
427 | rc->range = (uint32_t)-1; | ||
428 | rc->code = 0; | ||
429 | rc->init_bytes_left = RC_INIT_BYTES; | ||
430 | } | ||
431 | |||
432 | /* | ||
433 | * Read the first five initial bytes into rc->code if they haven't been | ||
434 | * read already. (Yes, the first byte gets completely ignored.) | ||
435 | */ | ||
436 | static bool XZ_FUNC rc_read_init(struct rc_dec *rc, struct xz_buf *b) | ||
437 | { | ||
438 | while (rc->init_bytes_left > 0) { | ||
439 | if (b->in_pos == b->in_size) | ||
440 | return false; | ||
441 | |||
442 | rc->code = (rc->code << 8) + b->in[b->in_pos++]; | ||
443 | --rc->init_bytes_left; | ||
444 | } | ||
445 | |||
446 | return true; | ||
447 | } | ||
448 | |||
449 | /* Return true if there may not be enough input for the next decoding loop. */ | ||
450 | static inline bool XZ_FUNC rc_limit_exceeded(const struct rc_dec *rc) | ||
451 | { | ||
452 | return rc->in_pos > rc->in_limit; | ||
453 | } | ||
454 | |||
455 | /* | ||
456 | * Return true if it is possible (from point of view of range decoder) that | ||
457 | * we have reached the end of the LZMA chunk. | ||
458 | */ | ||
459 | static inline bool XZ_FUNC rc_is_finished(const struct rc_dec *rc) | ||
460 | { | ||
461 | return rc->code == 0; | ||
462 | } | ||
463 | |||
464 | /* Read the next input byte if needed. */ | ||
465 | static __always_inline void XZ_FUNC rc_normalize(struct rc_dec *rc) | ||
466 | { | ||
467 | if (rc->range < RC_TOP_VALUE) { | ||
468 | rc->range <<= RC_SHIFT_BITS; | ||
469 | rc->code = (rc->code << RC_SHIFT_BITS) + rc->in[rc->in_pos++]; | ||
470 | } | ||
471 | } | ||
472 | |||
473 | /* | ||
474 | * Decode one bit. In some versions, this function has been splitted in three | ||
475 | * functions so that the compiler is supposed to be able to more easily avoid | ||
476 | * an extra branch. In this particular version of the LZMA decoder, this | ||
477 | * doesn't seem to be a good idea (tested with GCC 3.3.6, 3.4.6, and 4.3.3 | ||
478 | * on x86). Using a non-splitted version results in nicer looking code too. | ||
479 | * | ||
480 | * NOTE: This must return an int. Do not make it return a bool or the speed | ||
481 | * of the code generated by GCC 3.x decreases 10-15 %. (GCC 4.3 doesn't care, | ||
482 | * and it generates 10-20 % faster code than GCC 3.x from this file anyway.) | ||
483 | */ | ||
484 | static __always_inline int XZ_FUNC rc_bit(struct rc_dec *rc, uint16_t *prob) | ||
485 | { | ||
486 | uint32_t bound; | ||
487 | int bit; | ||
488 | |||
489 | rc_normalize(rc); | ||
490 | bound = (rc->range >> RC_BIT_MODEL_TOTAL_BITS) * *prob; | ||
491 | if (rc->code < bound) { | ||
492 | rc->range = bound; | ||
493 | *prob += (RC_BIT_MODEL_TOTAL - *prob) >> RC_MOVE_BITS; | ||
494 | bit = 0; | ||
495 | } else { | ||
496 | rc->range -= bound; | ||
497 | rc->code -= bound; | ||
498 | *prob -= *prob >> RC_MOVE_BITS; | ||
499 | bit = 1; | ||
500 | } | ||
501 | |||
502 | return bit; | ||
503 | } | ||
504 | |||
505 | /* Decode a bittree starting from the most significant bit. */ | ||
506 | static __always_inline uint32_t XZ_FUNC rc_bittree( | ||
507 | struct rc_dec *rc, uint16_t *probs, uint32_t limit) | ||
508 | { | ||
509 | uint32_t symbol = 1; | ||
510 | |||
511 | do { | ||
512 | if (rc_bit(rc, &probs[symbol])) | ||
513 | symbol = (symbol << 1) + 1; | ||
514 | else | ||
515 | symbol <<= 1; | ||
516 | } while (symbol < limit); | ||
517 | |||
518 | return symbol; | ||
519 | } | ||
520 | |||
521 | /* Decode a bittree starting from the least significant bit. */ | ||
522 | static __always_inline void XZ_FUNC rc_bittree_reverse(struct rc_dec *rc, | ||
523 | uint16_t *probs, uint32_t *dest, uint32_t limit) | ||
524 | { | ||
525 | uint32_t symbol = 1; | ||
526 | uint32_t i = 0; | ||
527 | |||
528 | do { | ||
529 | if (rc_bit(rc, &probs[symbol])) { | ||
530 | symbol = (symbol << 1) + 1; | ||
531 | *dest += 1 << i; | ||
532 | } else { | ||
533 | symbol <<= 1; | ||
534 | } | ||
535 | } while (++i < limit); | ||
536 | } | ||
537 | |||
538 | /* Decode direct bits (fixed fifty-fifty probability) */ | ||
539 | static inline void XZ_FUNC rc_direct( | ||
540 | struct rc_dec *rc, uint32_t *dest, uint32_t limit) | ||
541 | { | ||
542 | uint32_t mask; | ||
543 | |||
544 | do { | ||
545 | rc_normalize(rc); | ||
546 | rc->range >>= 1; | ||
547 | rc->code -= rc->range; | ||
548 | mask = (uint32_t)0 - (rc->code >> 31); | ||
549 | rc->code += rc->range & mask; | ||
550 | *dest = (*dest << 1) + (mask + 1); | ||
551 | } while (--limit > 0); | ||
552 | } | ||
553 | |||
554 | /******** | ||
555 | * LZMA * | ||
556 | ********/ | ||
557 | |||
558 | /* Get pointer to literal coder probability array. */ | ||
559 | static uint16_t * XZ_FUNC lzma_literal_probs(struct xz_dec_lzma2 *s) | ||
560 | { | ||
561 | uint32_t prev_byte = dict_get(&s->dict, 0); | ||
562 | uint32_t low = prev_byte >> (8 - s->lzma.lc); | ||
563 | uint32_t high = (s->dict.pos & s->lzma.literal_pos_mask) << s->lzma.lc; | ||
564 | return s->lzma.literal[low + high]; | ||
565 | } | ||
566 | |||
567 | /* Decode a literal (one 8-bit byte) */ | ||
568 | static void XZ_FUNC lzma_literal(struct xz_dec_lzma2 *s) | ||
569 | { | ||
570 | uint16_t *probs; | ||
571 | uint32_t symbol; | ||
572 | uint32_t match_byte; | ||
573 | uint32_t match_bit; | ||
574 | uint32_t offset; | ||
575 | uint32_t i; | ||
576 | |||
577 | probs = lzma_literal_probs(s); | ||
578 | |||
579 | if (lzma_state_is_literal(s->lzma.state)) { | ||
580 | symbol = rc_bittree(&s->rc, probs, 0x100); | ||
581 | } else { | ||
582 | symbol = 1; | ||
583 | match_byte = dict_get(&s->dict, s->lzma.rep0) << 1; | ||
584 | offset = 0x100; | ||
585 | |||
586 | do { | ||
587 | match_bit = match_byte & offset; | ||
588 | match_byte <<= 1; | ||
589 | i = offset + match_bit + symbol; | ||
590 | |||
591 | if (rc_bit(&s->rc, &probs[i])) { | ||
592 | symbol = (symbol << 1) + 1; | ||
593 | offset &= match_bit; | ||
594 | } else { | ||
595 | symbol <<= 1; | ||
596 | offset &= ~match_bit; | ||
597 | } | ||
598 | } while (symbol < 0x100); | ||
599 | } | ||
600 | |||
601 | dict_put(&s->dict, (uint8_t)symbol); | ||
602 | lzma_state_literal(&s->lzma.state); | ||
603 | } | ||
604 | |||
605 | /* Decode the length of the match into s->lzma.len. */ | ||
606 | static void XZ_FUNC lzma_len(struct xz_dec_lzma2 *s, struct lzma_len_dec *l, | ||
607 | uint32_t pos_state) | ||
608 | { | ||
609 | uint16_t *probs; | ||
610 | uint32_t limit; | ||
611 | |||
612 | if (!rc_bit(&s->rc, &l->choice)) { | ||
613 | probs = l->low[pos_state]; | ||
614 | limit = LEN_LOW_SYMBOLS; | ||
615 | s->lzma.len = MATCH_LEN_MIN; | ||
616 | } else { | ||
617 | if (!rc_bit(&s->rc, &l->choice2)) { | ||
618 | probs = l->mid[pos_state]; | ||
619 | limit = LEN_MID_SYMBOLS; | ||
620 | s->lzma.len = MATCH_LEN_MIN + LEN_LOW_SYMBOLS; | ||
621 | } else { | ||
622 | probs = l->high; | ||
623 | limit = LEN_HIGH_SYMBOLS; | ||
624 | s->lzma.len = MATCH_LEN_MIN + LEN_LOW_SYMBOLS | ||
625 | + LEN_MID_SYMBOLS; | ||
626 | } | ||
627 | } | ||
628 | |||
629 | s->lzma.len += rc_bittree(&s->rc, probs, limit) - limit; | ||
630 | } | ||
631 | |||
632 | /* Decode a match. The distance will be stored in s->lzma.rep0. */ | ||
633 | static void XZ_FUNC lzma_match(struct xz_dec_lzma2 *s, uint32_t pos_state) | ||
634 | { | ||
635 | uint16_t *probs; | ||
636 | uint32_t dist_slot; | ||
637 | uint32_t limit; | ||
638 | |||
639 | lzma_state_match(&s->lzma.state); | ||
640 | |||
641 | s->lzma.rep3 = s->lzma.rep2; | ||
642 | s->lzma.rep2 = s->lzma.rep1; | ||
643 | s->lzma.rep1 = s->lzma.rep0; | ||
644 | |||
645 | lzma_len(s, &s->lzma.match_len_dec, pos_state); | ||
646 | |||
647 | probs = s->lzma.dist_slot[lzma_get_dist_state(s->lzma.len)]; | ||
648 | dist_slot = rc_bittree(&s->rc, probs, DIST_SLOTS) - DIST_SLOTS; | ||
649 | |||
650 | if (dist_slot < DIST_MODEL_START) { | ||
651 | s->lzma.rep0 = dist_slot; | ||
652 | } else { | ||
653 | limit = (dist_slot >> 1) - 1; | ||
654 | s->lzma.rep0 = 2 + (dist_slot & 1); | ||
655 | |||
656 | if (dist_slot < DIST_MODEL_END) { | ||
657 | s->lzma.rep0 <<= limit; | ||
658 | probs = s->lzma.dist_special + s->lzma.rep0 | ||
659 | - dist_slot - 1; | ||
660 | rc_bittree_reverse(&s->rc, probs, | ||
661 | &s->lzma.rep0, limit); | ||
662 | } else { | ||
663 | rc_direct(&s->rc, &s->lzma.rep0, limit - ALIGN_BITS); | ||
664 | s->lzma.rep0 <<= ALIGN_BITS; | ||
665 | rc_bittree_reverse(&s->rc, s->lzma.dist_align, | ||
666 | &s->lzma.rep0, ALIGN_BITS); | ||
667 | } | ||
668 | } | ||
669 | } | ||
670 | |||
671 | /* | ||
672 | * Decode a repeated match. The distance is one of the four most recently | ||
673 | * seen matches. The distance will be stored in s->lzma.rep0. | ||
674 | */ | ||
675 | static void XZ_FUNC lzma_rep_match(struct xz_dec_lzma2 *s, uint32_t pos_state) | ||
676 | { | ||
677 | uint32_t tmp; | ||
678 | |||
679 | if (!rc_bit(&s->rc, &s->lzma.is_rep0[s->lzma.state])) { | ||
680 | if (!rc_bit(&s->rc, &s->lzma.is_rep0_long[ | ||
681 | s->lzma.state][pos_state])) { | ||
682 | lzma_state_short_rep(&s->lzma.state); | ||
683 | s->lzma.len = 1; | ||
684 | return; | ||
685 | } | ||
686 | } else { | ||
687 | if (!rc_bit(&s->rc, &s->lzma.is_rep1[s->lzma.state])) { | ||
688 | tmp = s->lzma.rep1; | ||
689 | } else { | ||
690 | if (!rc_bit(&s->rc, &s->lzma.is_rep2[s->lzma.state])) { | ||
691 | tmp = s->lzma.rep2; | ||
692 | } else { | ||
693 | tmp = s->lzma.rep3; | ||
694 | s->lzma.rep3 = s->lzma.rep2; | ||
695 | } | ||
696 | |||
697 | s->lzma.rep2 = s->lzma.rep1; | ||
698 | } | ||
699 | |||
700 | s->lzma.rep1 = s->lzma.rep0; | ||
701 | s->lzma.rep0 = tmp; | ||
702 | } | ||
703 | |||
704 | lzma_state_long_rep(&s->lzma.state); | ||
705 | lzma_len(s, &s->lzma.rep_len_dec, pos_state); | ||
706 | } | ||
707 | |||
708 | /* LZMA decoder core */ | ||
709 | static bool XZ_FUNC lzma_main(struct xz_dec_lzma2 *s) | ||
710 | { | ||
711 | uint32_t pos_state; | ||
712 | |||
713 | /* | ||
714 | * If the dictionary was reached during the previous call, try to | ||
715 | * finish the possibly pending repeat in the dictionary. | ||
716 | */ | ||
717 | if (dict_has_space(&s->dict) && s->lzma.len > 0) | ||
718 | dict_repeat(&s->dict, &s->lzma.len, s->lzma.rep0); | ||
719 | |||
720 | /* | ||
721 | * Decode more LZMA symbols. One iteration may consume up to | ||
722 | * LZMA_IN_REQUIRED - 1 bytes. | ||
723 | */ | ||
724 | while (dict_has_space(&s->dict) && !rc_limit_exceeded(&s->rc)) { | ||
725 | pos_state = s->dict.pos & s->lzma.pos_mask; | ||
726 | |||
727 | if (!rc_bit(&s->rc, &s->lzma.is_match[ | ||
728 | s->lzma.state][pos_state])) { | ||
729 | lzma_literal(s); | ||
730 | } else { | ||
731 | if (rc_bit(&s->rc, &s->lzma.is_rep[s->lzma.state])) | ||
732 | lzma_rep_match(s, pos_state); | ||
733 | else | ||
734 | lzma_match(s, pos_state); | ||
735 | |||
736 | if (!dict_repeat(&s->dict, &s->lzma.len, s->lzma.rep0)) | ||
737 | return false; | ||
738 | } | ||
739 | } | ||
740 | |||
741 | /* | ||
742 | * Having the range decoder always normalized when we are outside | ||
743 | * this function makes it easier to correctly handle end of the chunk. | ||
744 | */ | ||
745 | rc_normalize(&s->rc); | ||
746 | |||
747 | return true; | ||
748 | } | ||
749 | |||
750 | /* | ||
751 | * Reset the LZMA decoder and range decoder state. Dictionary is nore reset | ||
752 | * here, because LZMA state may be reset without resetting the dictionary. | ||
753 | */ | ||
754 | static void XZ_FUNC lzma_reset(struct xz_dec_lzma2 *s) | ||
755 | { | ||
756 | uint16_t *probs; | ||
757 | size_t i; | ||
758 | |||
759 | s->lzma.state = STATE_LIT_LIT; | ||
760 | s->lzma.rep0 = 0; | ||
761 | s->lzma.rep1 = 0; | ||
762 | s->lzma.rep2 = 0; | ||
763 | s->lzma.rep3 = 0; | ||
764 | |||
765 | /* | ||
766 | * All probabilities are initialized to the same value. This hack | ||
767 | * makes the code smaller by avoiding a separate loop for each | ||
768 | * probability array. | ||
769 | * | ||
770 | * This could be optimized so that only that part of literal | ||
771 | * probabilities that are actually required. In the common case | ||
772 | * we would write 12 KiB less. | ||
773 | */ | ||
774 | probs = s->lzma.is_match[0]; | ||
775 | for (i = 0; i < PROBS_TOTAL; ++i) | ||
776 | probs[i] = RC_BIT_MODEL_TOTAL / 2; | ||
777 | |||
778 | rc_reset(&s->rc); | ||
779 | } | ||
780 | |||
781 | /* | ||
782 | * Decode and validate LZMA properties (lc/lp/pb) and calculate the bit masks | ||
783 | * from the decoded lp and pb values. On success, the LZMA decoder state is | ||
784 | * reset and true is returned. | ||
785 | */ | ||
786 | static bool XZ_FUNC lzma_props(struct xz_dec_lzma2 *s, uint8_t props) | ||
787 | { | ||
788 | if (props > (4 * 5 + 4) * 9 + 8) | ||
789 | return false; | ||
790 | |||
791 | s->lzma.pos_mask = 0; | ||
792 | while (props >= 9 * 5) { | ||
793 | props -= 9 * 5; | ||
794 | ++s->lzma.pos_mask; | ||
795 | } | ||
796 | |||
797 | s->lzma.pos_mask = (1 << s->lzma.pos_mask) - 1; | ||
798 | |||
799 | s->lzma.literal_pos_mask = 0; | ||
800 | while (props >= 9) { | ||
801 | props -= 9; | ||
802 | ++s->lzma.literal_pos_mask; | ||
803 | } | ||
804 | |||
805 | s->lzma.lc = props; | ||
806 | |||
807 | if (s->lzma.lc + s->lzma.literal_pos_mask > 4) | ||
808 | return false; | ||
809 | |||
810 | s->lzma.literal_pos_mask = (1 << s->lzma.literal_pos_mask) - 1; | ||
811 | |||
812 | lzma_reset(s); | ||
813 | |||
814 | return true; | ||
815 | } | ||
816 | |||
817 | /********* | ||
818 | * LZMA2 * | ||
819 | *********/ | ||
820 | |||
821 | /* | ||
822 | * The LZMA decoder assumes that if the input limit (s->rc.in_limit) hasn't | ||
823 | * been exceeded, it is safe to read up to LZMA_IN_REQUIRED bytes. This | ||
824 | * wrapper function takes care of making the LZMA decoder's assumption safe. | ||
825 | * | ||
826 | * As long as there is plenty of input left to be decoded in the current LZMA | ||
827 | * chunk, we decode directly from the caller-supplied input buffer until | ||
828 | * there's LZMA_IN_REQUIRED bytes left. Those remaining bytes are copied into | ||
829 | * s->temp.buf, which (hopefully) gets filled on the next call to this | ||
830 | * function. We decode a few bytes from the temporary buffer so that we can | ||
831 | * continue decoding from the caller-supplied input buffer again. | ||
832 | */ | ||
833 | static bool XZ_FUNC lzma2_lzma(struct xz_dec_lzma2 *s, struct xz_buf *b) | ||
834 | { | ||
835 | size_t in_avail; | ||
836 | uint32_t tmp; | ||
837 | |||
838 | in_avail = b->in_size - b->in_pos; | ||
839 | if (s->temp.size > 0 || s->lzma2.compressed == 0) { | ||
840 | tmp = 2 * LZMA_IN_REQUIRED - s->temp.size; | ||
841 | if (tmp > s->lzma2.compressed - s->temp.size) | ||
842 | tmp = s->lzma2.compressed - s->temp.size; | ||
843 | if (tmp > in_avail) | ||
844 | tmp = in_avail; | ||
845 | |||
846 | memcpy(s->temp.buf + s->temp.size, b->in + b->in_pos, tmp); | ||
847 | |||
848 | if (s->temp.size + tmp == s->lzma2.compressed) { | ||
849 | memzero(s->temp.buf + s->temp.size + tmp, | ||
850 | sizeof(s->temp.buf) | ||
851 | - s->temp.size - tmp); | ||
852 | s->rc.in_limit = s->temp.size + tmp; | ||
853 | } else if (s->temp.size + tmp < LZMA_IN_REQUIRED) { | ||
854 | s->temp.size += tmp; | ||
855 | b->in_pos += tmp; | ||
856 | return true; | ||
857 | } else { | ||
858 | s->rc.in_limit = s->temp.size + tmp - LZMA_IN_REQUIRED; | ||
859 | } | ||
860 | |||
861 | s->rc.in = s->temp.buf; | ||
862 | s->rc.in_pos = 0; | ||
863 | |||
864 | if (!lzma_main(s) || s->rc.in_pos > s->temp.size + tmp) | ||
865 | return false; | ||
866 | |||
867 | s->lzma2.compressed -= s->rc.in_pos; | ||
868 | |||
869 | if (s->rc.in_pos < s->temp.size) { | ||
870 | s->temp.size -= s->rc.in_pos; | ||
871 | memmove(s->temp.buf, s->temp.buf + s->rc.in_pos, | ||
872 | s->temp.size); | ||
873 | return true; | ||
874 | } | ||
875 | |||
876 | b->in_pos += s->rc.in_pos - s->temp.size; | ||
877 | s->temp.size = 0; | ||
878 | } | ||
879 | |||
880 | in_avail = b->in_size - b->in_pos; | ||
881 | if (in_avail >= LZMA_IN_REQUIRED) { | ||
882 | s->rc.in = b->in; | ||
883 | s->rc.in_pos = b->in_pos; | ||
884 | |||
885 | if (in_avail >= s->lzma2.compressed + LZMA_IN_REQUIRED) | ||
886 | s->rc.in_limit = b->in_pos + s->lzma2.compressed; | ||
887 | else | ||
888 | s->rc.in_limit = b->in_size - LZMA_IN_REQUIRED; | ||
889 | |||
890 | if (!lzma_main(s)) | ||
891 | return false; | ||
892 | |||
893 | in_avail = s->rc.in_pos - b->in_pos; | ||
894 | if (in_avail > s->lzma2.compressed) | ||
895 | return false; | ||
896 | |||
897 | s->lzma2.compressed -= in_avail; | ||
898 | b->in_pos = s->rc.in_pos; | ||
899 | } | ||
900 | |||
901 | in_avail = b->in_size - b->in_pos; | ||
902 | if (in_avail < LZMA_IN_REQUIRED) { | ||
903 | if (in_avail > s->lzma2.compressed) | ||
904 | in_avail = s->lzma2.compressed; | ||
905 | |||
906 | memcpy(s->temp.buf, b->in + b->in_pos, in_avail); | ||
907 | s->temp.size = in_avail; | ||
908 | b->in_pos += in_avail; | ||
909 | } | ||
910 | |||
911 | return true; | ||
912 | } | ||
913 | |||
914 | /* | ||
915 | * Take care of the LZMA2 control layer, and forward the job of actual LZMA | ||
916 | * decoding or copying of uncompressed chunks to other functions. | ||
917 | */ | ||
918 | XZ_EXTERN NOINLINE enum xz_ret XZ_FUNC xz_dec_lzma2_run( | ||
919 | struct xz_dec_lzma2 *s, struct xz_buf *b) | ||
920 | { | ||
921 | uint32_t tmp; | ||
922 | |||
923 | while (b->in_pos < b->in_size || s->lzma2.sequence == SEQ_LZMA_RUN) { | ||
924 | switch (s->lzma2.sequence) { | ||
925 | case SEQ_CONTROL: | ||
926 | /* | ||
927 | * LZMA2 control byte | ||
928 | * | ||
929 | * Exact values: | ||
930 | * 0x00 End marker | ||
931 | * 0x01 Dictionary reset followed by | ||
932 | * an uncompressed chunk | ||
933 | * 0x02 Uncompressed chunk (no dictionary reset) | ||
934 | * | ||
935 | * Highest three bits (s->control & 0xE0): | ||
936 | * 0xE0 Dictionary reset, new properties and state | ||
937 | * reset, followed by LZMA compressed chunk | ||
938 | * 0xC0 New properties and state reset, followed | ||
939 | * by LZMA compressed chunk (no dictionary | ||
940 | * reset) | ||
941 | * 0xA0 State reset using old properties, | ||
942 | * followed by LZMA compressed chunk (no | ||
943 | * dictionary reset) | ||
944 | * 0x80 LZMA chunk (no dictionary or state reset) | ||
945 | * | ||
946 | * For LZMA compressed chunks, the lowest five bits | ||
947 | * (s->control & 1F) are the highest bits of the | ||
948 | * uncompressed size (bits 16-20). | ||
949 | * | ||
950 | * A new LZMA2 stream must begin with a dictionary | ||
951 | * reset. The first LZMA chunk must set new | ||
952 | * properties and reset the LZMA state. | ||
953 | * | ||
954 | * Values that don't match anything described above | ||
955 | * are invalid and we return XZ_DATA_ERROR. | ||
956 | */ | ||
957 | tmp = b->in[b->in_pos++]; | ||
958 | |||
959 | if (tmp >= 0xE0 || tmp == 0x01) { | ||
960 | s->lzma2.need_props = true; | ||
961 | s->lzma2.need_dict_reset = false; | ||
962 | dict_reset(&s->dict, b); | ||
963 | } else if (s->lzma2.need_dict_reset) { | ||
964 | return XZ_DATA_ERROR; | ||
965 | } | ||
966 | |||
967 | if (tmp >= 0x80) { | ||
968 | s->lzma2.uncompressed = (tmp & 0x1F) << 16; | ||
969 | s->lzma2.sequence = SEQ_UNCOMPRESSED_1; | ||
970 | |||
971 | if (tmp >= 0xC0) { | ||
972 | /* | ||
973 | * When there are new properties, | ||
974 | * state reset is done at | ||
975 | * SEQ_PROPERTIES. | ||
976 | */ | ||
977 | s->lzma2.need_props = false; | ||
978 | s->lzma2.next_sequence | ||
979 | = SEQ_PROPERTIES; | ||
980 | |||
981 | } else if (s->lzma2.need_props) { | ||
982 | return XZ_DATA_ERROR; | ||
983 | |||
984 | } else { | ||
985 | s->lzma2.next_sequence | ||
986 | = SEQ_LZMA_PREPARE; | ||
987 | if (tmp >= 0xA0) | ||
988 | lzma_reset(s); | ||
989 | } | ||
990 | } else { | ||
991 | if (tmp == 0x00) | ||
992 | return XZ_STREAM_END; | ||
993 | |||
994 | if (tmp > 0x02) | ||
995 | return XZ_DATA_ERROR; | ||
996 | |||
997 | s->lzma2.sequence = SEQ_COMPRESSED_0; | ||
998 | s->lzma2.next_sequence = SEQ_COPY; | ||
999 | } | ||
1000 | |||
1001 | break; | ||
1002 | |||
1003 | case SEQ_UNCOMPRESSED_1: | ||
1004 | s->lzma2.uncompressed | ||
1005 | += (uint32_t)b->in[b->in_pos++] << 8; | ||
1006 | s->lzma2.sequence = SEQ_UNCOMPRESSED_2; | ||
1007 | break; | ||
1008 | |||
1009 | case SEQ_UNCOMPRESSED_2: | ||
1010 | s->lzma2.uncompressed | ||
1011 | += (uint32_t)b->in[b->in_pos++] + 1; | ||
1012 | s->lzma2.sequence = SEQ_COMPRESSED_0; | ||
1013 | break; | ||
1014 | |||
1015 | case SEQ_COMPRESSED_0: | ||
1016 | s->lzma2.compressed | ||
1017 | = (uint32_t)b->in[b->in_pos++] << 8; | ||
1018 | s->lzma2.sequence = SEQ_COMPRESSED_1; | ||
1019 | break; | ||
1020 | |||
1021 | case SEQ_COMPRESSED_1: | ||
1022 | s->lzma2.compressed | ||
1023 | += (uint32_t)b->in[b->in_pos++] + 1; | ||
1024 | s->lzma2.sequence = s->lzma2.next_sequence; | ||
1025 | break; | ||
1026 | |||
1027 | case SEQ_PROPERTIES: | ||
1028 | if (!lzma_props(s, b->in[b->in_pos++])) | ||
1029 | return XZ_DATA_ERROR; | ||
1030 | |||
1031 | s->lzma2.sequence = SEQ_LZMA_PREPARE; | ||
1032 | |||
1033 | case SEQ_LZMA_PREPARE: | ||
1034 | if (s->lzma2.compressed < RC_INIT_BYTES) | ||
1035 | return XZ_DATA_ERROR; | ||
1036 | |||
1037 | if (!rc_read_init(&s->rc, b)) | ||
1038 | return XZ_OK; | ||
1039 | |||
1040 | s->lzma2.compressed -= RC_INIT_BYTES; | ||
1041 | s->lzma2.sequence = SEQ_LZMA_RUN; | ||
1042 | |||
1043 | case SEQ_LZMA_RUN: | ||
1044 | /* | ||
1045 | * Set dictionary limit to indicate how much we want | ||
1046 | * to be encoded at maximum. Decode new data into the | ||
1047 | * dictionary. Flush the new data from dictionary to | ||
1048 | * b->out. Check if we finished decoding this chunk. | ||
1049 | * In case the dictionary got full but we didn't fill | ||
1050 | * the output buffer yet, we may run this loop | ||
1051 | * multiple times without changing s->lzma2.sequence. | ||
1052 | */ | ||
1053 | dict_limit(&s->dict, min_t(size_t, | ||
1054 | b->out_size - b->out_pos, | ||
1055 | s->lzma2.uncompressed)); | ||
1056 | if (!lzma2_lzma(s, b)) | ||
1057 | return XZ_DATA_ERROR; | ||
1058 | |||
1059 | s->lzma2.uncompressed -= dict_flush(&s->dict, b); | ||
1060 | |||
1061 | if (s->lzma2.uncompressed == 0) { | ||
1062 | if (s->lzma2.compressed > 0 || s->lzma.len > 0 | ||
1063 | || !rc_is_finished(&s->rc)) | ||
1064 | return XZ_DATA_ERROR; | ||
1065 | |||
1066 | rc_reset(&s->rc); | ||
1067 | s->lzma2.sequence = SEQ_CONTROL; | ||
1068 | |||
1069 | } else if (b->out_pos == b->out_size | ||
1070 | || (b->in_pos == b->in_size | ||
1071 | && s->temp.size | ||
1072 | < s->lzma2.compressed)) { | ||
1073 | return XZ_OK; | ||
1074 | } | ||
1075 | |||
1076 | break; | ||
1077 | |||
1078 | case SEQ_COPY: | ||
1079 | dict_uncompressed(&s->dict, b, &s->lzma2.compressed); | ||
1080 | if (s->lzma2.compressed > 0) | ||
1081 | return XZ_OK; | ||
1082 | |||
1083 | s->lzma2.sequence = SEQ_CONTROL; | ||
1084 | break; | ||
1085 | } | ||
1086 | } | ||
1087 | |||
1088 | return XZ_OK; | ||
1089 | } | ||
1090 | |||
1091 | XZ_EXTERN struct xz_dec_lzma2 * XZ_FUNC xz_dec_lzma2_create(uint32_t dict_max) | ||
1092 | { | ||
1093 | struct xz_dec_lzma2 *s; | ||
1094 | |||
1095 | /* Maximum supported dictionary by this implementation is 3 GiB. */ | ||
1096 | if (dict_max > ((uint32_t)3 << 30)) | ||
1097 | return NULL; | ||
1098 | |||
1099 | s = kmalloc(sizeof(*s), GFP_KERNEL); | ||
1100 | if (s == NULL) | ||
1101 | return NULL; | ||
1102 | |||
1103 | if (dict_max > 0) { | ||
1104 | s->dict.buf = vmalloc(dict_max); | ||
1105 | if (s->dict.buf == NULL) { | ||
1106 | kfree(s); | ||
1107 | return NULL; | ||
1108 | } | ||
1109 | } | ||
1110 | |||
1111 | s->dict.allocated = dict_max; | ||
1112 | |||
1113 | return s; | ||
1114 | } | ||
1115 | |||
1116 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_lzma2_reset( | ||
1117 | struct xz_dec_lzma2 *s, uint8_t props) | ||
1118 | { | ||
1119 | /* This limits dictionary size to 3 GiB to keep parsing simpler. */ | ||
1120 | if (props > 39) | ||
1121 | return XZ_OPTIONS_ERROR; | ||
1122 | |||
1123 | s->dict.size = 2 + (props & 1); | ||
1124 | s->dict.size <<= (props >> 1) + 11; | ||
1125 | |||
1126 | if (s->dict.allocated > 0 && s->dict.allocated < s->dict.size) { | ||
1127 | #ifdef XZ_REALLOC_DICT_BUF | ||
1128 | s->dict.buf = XZ_REALLOC_DICT_BUF(s->dict.buf, s->dict.size); | ||
1129 | if (!s->dict.buf) | ||
1130 | return XZ_MEMLIMIT_ERROR; | ||
1131 | s->dict.allocated = s->dict.size; | ||
1132 | #else | ||
1133 | return XZ_MEMLIMIT_ERROR; | ||
1134 | #endif | ||
1135 | } | ||
1136 | |||
1137 | s->dict.end = s->dict.size; | ||
1138 | |||
1139 | s->lzma.len = 0; | ||
1140 | |||
1141 | s->lzma2.sequence = SEQ_CONTROL; | ||
1142 | s->lzma2.need_dict_reset = true; | ||
1143 | |||
1144 | s->temp.size = 0; | ||
1145 | |||
1146 | return XZ_OK; | ||
1147 | } | ||
1148 | |||
1149 | XZ_EXTERN void XZ_FUNC xz_dec_lzma2_end(struct xz_dec_lzma2 *s) | ||
1150 | { | ||
1151 | if (s->dict.allocated > 0) | ||
1152 | vfree(s->dict.buf); | ||
1153 | |||
1154 | kfree(s); | ||
1155 | } | ||
diff --git a/archival/libunarchive/unxz/xz_dec_stream.c b/archival/libunarchive/unxz/xz_dec_stream.c new file mode 100644 index 000000000..21db283fb --- /dev/null +++ b/archival/libunarchive/unxz/xz_dec_stream.c | |||
@@ -0,0 +1,821 @@ | |||
1 | /* | ||
2 | * .xz Stream decoder | ||
3 | * | ||
4 | * Author: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * | ||
6 | * This file has been put into the public domain. | ||
7 | * You can do whatever you want with this file. | ||
8 | */ | ||
9 | |||
10 | #include "xz_private.h" | ||
11 | #include "xz_stream.h" | ||
12 | |||
13 | /* Hash used to validate the Index field */ | ||
14 | struct xz_dec_hash { | ||
15 | vli_type unpadded; | ||
16 | vli_type uncompressed; | ||
17 | uint32_t crc32; | ||
18 | }; | ||
19 | |||
20 | struct xz_dec { | ||
21 | /* Position in dec_main() */ | ||
22 | enum { | ||
23 | SEQ_STREAM_HEADER, | ||
24 | SEQ_BLOCK_START, | ||
25 | SEQ_BLOCK_HEADER, | ||
26 | SEQ_BLOCK_UNCOMPRESS, | ||
27 | SEQ_BLOCK_PADDING, | ||
28 | SEQ_BLOCK_CHECK, | ||
29 | SEQ_INDEX, | ||
30 | SEQ_INDEX_PADDING, | ||
31 | SEQ_INDEX_CRC32, | ||
32 | SEQ_STREAM_FOOTER | ||
33 | } sequence; | ||
34 | |||
35 | /* Position in variable-length integers and Check fields */ | ||
36 | uint32_t pos; | ||
37 | |||
38 | /* Variable-length integer decoded by dec_vli() */ | ||
39 | vli_type vli; | ||
40 | |||
41 | /* Saved in_pos and out_pos */ | ||
42 | size_t in_start; | ||
43 | size_t out_start; | ||
44 | |||
45 | /* CRC32 value in Block or Index */ | ||
46 | uint32_t crc32; | ||
47 | |||
48 | /* Type of the integrity check calculated from uncompressed data */ | ||
49 | enum xz_check check_type; | ||
50 | |||
51 | /* True if we are operating in single-call mode. */ | ||
52 | bool single_call; | ||
53 | |||
54 | /* | ||
55 | * True if the next call to xz_dec_run() is allowed to return | ||
56 | * XZ_BUF_ERROR. | ||
57 | */ | ||
58 | bool allow_buf_error; | ||
59 | |||
60 | /* Information stored in Block Header */ | ||
61 | struct { | ||
62 | /* | ||
63 | * Value stored in the Compressed Size field, or | ||
64 | * VLI_UNKNOWN if Compressed Size is not present. | ||
65 | */ | ||
66 | vli_type compressed; | ||
67 | |||
68 | /* | ||
69 | * Value stored in the Uncompressed Size field, or | ||
70 | * VLI_UNKNOWN if Uncompressed Size is not present. | ||
71 | */ | ||
72 | vli_type uncompressed; | ||
73 | |||
74 | /* Size of the Block Header field */ | ||
75 | uint32_t size; | ||
76 | } block_header; | ||
77 | |||
78 | /* Information collected when decoding Blocks */ | ||
79 | struct { | ||
80 | /* Observed compressed size of the current Block */ | ||
81 | vli_type compressed; | ||
82 | |||
83 | /* Observed uncompressed size of the current Block */ | ||
84 | vli_type uncompressed; | ||
85 | |||
86 | /* Number of Blocks decoded so far */ | ||
87 | vli_type count; | ||
88 | |||
89 | /* | ||
90 | * Hash calculated from the Block sizes. This is used to | ||
91 | * validate the Index field. | ||
92 | */ | ||
93 | struct xz_dec_hash hash; | ||
94 | } block; | ||
95 | |||
96 | /* Variables needed when verifying the Index field */ | ||
97 | struct { | ||
98 | /* Position in dec_index() */ | ||
99 | enum { | ||
100 | SEQ_INDEX_COUNT, | ||
101 | SEQ_INDEX_UNPADDED, | ||
102 | SEQ_INDEX_UNCOMPRESSED | ||
103 | } sequence; | ||
104 | |||
105 | /* Size of the Index in bytes */ | ||
106 | vli_type size; | ||
107 | |||
108 | /* Number of Records (matches block.count in valid files) */ | ||
109 | vli_type count; | ||
110 | |||
111 | /* | ||
112 | * Hash calculated from the Records (matches block.hash in | ||
113 | * valid files). | ||
114 | */ | ||
115 | struct xz_dec_hash hash; | ||
116 | } index; | ||
117 | |||
118 | /* | ||
119 | * Temporary buffer needed to hold Stream Header, Block Header, | ||
120 | * and Stream Footer. The Block Header is the biggest (1 KiB) | ||
121 | * so we reserve space according to that. buf[] has to be aligned | ||
122 | * to a multiple of four bytes; the size_t variables before it | ||
123 | * should guarantee this. | ||
124 | */ | ||
125 | struct { | ||
126 | size_t pos; | ||
127 | size_t size; | ||
128 | uint8_t buf[1024]; | ||
129 | } temp; | ||
130 | |||
131 | struct xz_dec_lzma2 *lzma2; | ||
132 | |||
133 | #ifdef XZ_DEC_BCJ | ||
134 | struct xz_dec_bcj *bcj; | ||
135 | bool bcj_active; | ||
136 | #endif | ||
137 | }; | ||
138 | |||
139 | #ifdef XZ_DEC_ANY_CHECK | ||
140 | /* Sizes of the Check field with different Check IDs */ | ||
141 | static const uint8_t check_sizes[16] = { | ||
142 | 0, | ||
143 | 4, 4, 4, | ||
144 | 8, 8, 8, | ||
145 | 16, 16, 16, | ||
146 | 32, 32, 32, | ||
147 | 64, 64, 64 | ||
148 | }; | ||
149 | #endif | ||
150 | |||
151 | /* | ||
152 | * Fill s->temp by copying data starting from b->in[b->in_pos]. Caller | ||
153 | * must have set s->temp.pos to indicate how much data we are supposed | ||
154 | * to copy into s->temp.buf. Return true once s->temp.pos has reached | ||
155 | * s->temp.size. | ||
156 | */ | ||
157 | static bool XZ_FUNC fill_temp(struct xz_dec *s, struct xz_buf *b) | ||
158 | { | ||
159 | size_t copy_size = min_t(size_t, | ||
160 | b->in_size - b->in_pos, s->temp.size - s->temp.pos); | ||
161 | |||
162 | memcpy(s->temp.buf + s->temp.pos, b->in + b->in_pos, copy_size); | ||
163 | b->in_pos += copy_size; | ||
164 | s->temp.pos += copy_size; | ||
165 | |||
166 | if (s->temp.pos == s->temp.size) { | ||
167 | s->temp.pos = 0; | ||
168 | return true; | ||
169 | } | ||
170 | |||
171 | return false; | ||
172 | } | ||
173 | |||
174 | /* Decode a variable-length integer (little-endian base-128 encoding) */ | ||
175 | static enum xz_ret XZ_FUNC dec_vli(struct xz_dec *s, | ||
176 | const uint8_t *in, size_t *in_pos, size_t in_size) | ||
177 | { | ||
178 | uint8_t byte; | ||
179 | |||
180 | if (s->pos == 0) | ||
181 | s->vli = 0; | ||
182 | |||
183 | while (*in_pos < in_size) { | ||
184 | byte = in[*in_pos]; | ||
185 | ++*in_pos; | ||
186 | |||
187 | s->vli |= (vli_type)(byte & 0x7F) << s->pos; | ||
188 | |||
189 | if ((byte & 0x80) == 0) { | ||
190 | /* Don't allow non-minimal encodings. */ | ||
191 | if (byte == 0 && s->pos != 0) | ||
192 | return XZ_DATA_ERROR; | ||
193 | |||
194 | s->pos = 0; | ||
195 | return XZ_STREAM_END; | ||
196 | } | ||
197 | |||
198 | s->pos += 7; | ||
199 | if (s->pos == 7 * VLI_BYTES_MAX) | ||
200 | return XZ_DATA_ERROR; | ||
201 | } | ||
202 | |||
203 | return XZ_OK; | ||
204 | } | ||
205 | |||
206 | /* | ||
207 | * Decode the Compressed Data field from a Block. Update and validate | ||
208 | * the observed compressed and uncompressed sizes of the Block so that | ||
209 | * they don't exceed the values possibly stored in the Block Header | ||
210 | * (validation assumes that no integer overflow occurs, since vli_type | ||
211 | * is normally uint64_t). Update the CRC32 if presence of the CRC32 | ||
212 | * field was indicated in Stream Header. | ||
213 | * | ||
214 | * Once the decoding is finished, validate that the observed sizes match | ||
215 | * the sizes possibly stored in the Block Header. Update the hash and | ||
216 | * Block count, which are later used to validate the Index field. | ||
217 | */ | ||
218 | static enum xz_ret XZ_FUNC dec_block(struct xz_dec *s, struct xz_buf *b) | ||
219 | { | ||
220 | enum xz_ret ret; | ||
221 | |||
222 | s->in_start = b->in_pos; | ||
223 | s->out_start = b->out_pos; | ||
224 | |||
225 | #ifdef XZ_DEC_BCJ | ||
226 | if (s->bcj_active) | ||
227 | ret = xz_dec_bcj_run(s->bcj, s->lzma2, b); | ||
228 | else | ||
229 | #endif | ||
230 | ret = xz_dec_lzma2_run(s->lzma2, b); | ||
231 | |||
232 | s->block.compressed += b->in_pos - s->in_start; | ||
233 | s->block.uncompressed += b->out_pos - s->out_start; | ||
234 | |||
235 | /* | ||
236 | * There is no need to separately check for VLI_UNKNOWN, since | ||
237 | * the observed sizes are always smaller than VLI_UNKNOWN. | ||
238 | */ | ||
239 | if (s->block.compressed > s->block_header.compressed | ||
240 | || s->block.uncompressed | ||
241 | > s->block_header.uncompressed) | ||
242 | return XZ_DATA_ERROR; | ||
243 | |||
244 | if (s->check_type == XZ_CHECK_CRC32) | ||
245 | s->crc32 = xz_crc32(b->out + s->out_start, | ||
246 | b->out_pos - s->out_start, s->crc32); | ||
247 | |||
248 | if (ret == XZ_STREAM_END) { | ||
249 | if (s->block_header.compressed != VLI_UNKNOWN | ||
250 | && s->block_header.compressed | ||
251 | != s->block.compressed) | ||
252 | return XZ_DATA_ERROR; | ||
253 | |||
254 | if (s->block_header.uncompressed != VLI_UNKNOWN | ||
255 | && s->block_header.uncompressed | ||
256 | != s->block.uncompressed) | ||
257 | return XZ_DATA_ERROR; | ||
258 | |||
259 | s->block.hash.unpadded += s->block_header.size | ||
260 | + s->block.compressed; | ||
261 | |||
262 | #ifdef XZ_DEC_ANY_CHECK | ||
263 | s->block.hash.unpadded += check_sizes[s->check_type]; | ||
264 | #else | ||
265 | if (s->check_type == XZ_CHECK_CRC32) | ||
266 | s->block.hash.unpadded += 4; | ||
267 | #endif | ||
268 | |||
269 | s->block.hash.uncompressed += s->block.uncompressed; | ||
270 | s->block.hash.crc32 = xz_crc32( | ||
271 | (const uint8_t *)&s->block.hash, | ||
272 | sizeof(s->block.hash), s->block.hash.crc32); | ||
273 | |||
274 | ++s->block.count; | ||
275 | } | ||
276 | |||
277 | return ret; | ||
278 | } | ||
279 | |||
280 | /* Update the Index size and the CRC32 value. */ | ||
281 | static void XZ_FUNC index_update(struct xz_dec *s, const struct xz_buf *b) | ||
282 | { | ||
283 | size_t in_used = b->in_pos - s->in_start; | ||
284 | s->index.size += in_used; | ||
285 | s->crc32 = xz_crc32(b->in + s->in_start, in_used, s->crc32); | ||
286 | } | ||
287 | |||
288 | /* | ||
289 | * Decode the Number of Records, Unpadded Size, and Uncompressed Size | ||
290 | * fields from the Index field. That is, Index Padding and CRC32 are not | ||
291 | * decoded by this function. | ||
292 | * | ||
293 | * This can return XZ_OK (more input needed), XZ_STREAM_END (everything | ||
294 | * successfully decoded), or XZ_DATA_ERROR (input is corrupt). | ||
295 | */ | ||
296 | static enum xz_ret XZ_FUNC dec_index(struct xz_dec *s, struct xz_buf *b) | ||
297 | { | ||
298 | enum xz_ret ret; | ||
299 | |||
300 | do { | ||
301 | ret = dec_vli(s, b->in, &b->in_pos, b->in_size); | ||
302 | if (ret != XZ_STREAM_END) { | ||
303 | index_update(s, b); | ||
304 | return ret; | ||
305 | } | ||
306 | |||
307 | switch (s->index.sequence) { | ||
308 | case SEQ_INDEX_COUNT: | ||
309 | s->index.count = s->vli; | ||
310 | |||
311 | /* | ||
312 | * Validate that the Number of Records field | ||
313 | * indicates the same number of Records as | ||
314 | * there were Blocks in the Stream. | ||
315 | */ | ||
316 | if (s->index.count != s->block.count) | ||
317 | return XZ_DATA_ERROR; | ||
318 | |||
319 | s->index.sequence = SEQ_INDEX_UNPADDED; | ||
320 | break; | ||
321 | |||
322 | case SEQ_INDEX_UNPADDED: | ||
323 | s->index.hash.unpadded += s->vli; | ||
324 | s->index.sequence = SEQ_INDEX_UNCOMPRESSED; | ||
325 | break; | ||
326 | |||
327 | case SEQ_INDEX_UNCOMPRESSED: | ||
328 | s->index.hash.uncompressed += s->vli; | ||
329 | s->index.hash.crc32 = xz_crc32( | ||
330 | (const uint8_t *)&s->index.hash, | ||
331 | sizeof(s->index.hash), | ||
332 | s->index.hash.crc32); | ||
333 | --s->index.count; | ||
334 | s->index.sequence = SEQ_INDEX_UNPADDED; | ||
335 | break; | ||
336 | } | ||
337 | } while (s->index.count > 0); | ||
338 | |||
339 | return XZ_STREAM_END; | ||
340 | } | ||
341 | |||
342 | /* | ||
343 | * Validate that the next four input bytes match the value of s->crc32. | ||
344 | * s->pos must be zero when starting to validate the first byte. | ||
345 | */ | ||
346 | static enum xz_ret XZ_FUNC crc32_validate(struct xz_dec *s, struct xz_buf *b) | ||
347 | { | ||
348 | do { | ||
349 | if (b->in_pos == b->in_size) | ||
350 | return XZ_OK; | ||
351 | |||
352 | if (((s->crc32 >> s->pos) & 0xFF) != b->in[b->in_pos++]) | ||
353 | return XZ_DATA_ERROR; | ||
354 | |||
355 | s->pos += 8; | ||
356 | |||
357 | } while (s->pos < 32); | ||
358 | |||
359 | s->crc32 = 0; | ||
360 | s->pos = 0; | ||
361 | |||
362 | return XZ_STREAM_END; | ||
363 | } | ||
364 | |||
365 | #ifdef XZ_DEC_ANY_CHECK | ||
366 | /* | ||
367 | * Skip over the Check field when the Check ID is not supported. | ||
368 | * Returns true once the whole Check field has been skipped over. | ||
369 | */ | ||
370 | static bool XZ_FUNC check_skip(struct xz_dec *s, struct xz_buf *b) | ||
371 | { | ||
372 | while (s->pos < check_sizes[s->check_type]) { | ||
373 | if (b->in_pos == b->in_size) | ||
374 | return false; | ||
375 | |||
376 | ++b->in_pos; | ||
377 | ++s->pos; | ||
378 | } | ||
379 | |||
380 | s->pos = 0; | ||
381 | |||
382 | return true; | ||
383 | } | ||
384 | #endif | ||
385 | |||
386 | /* Decode the Stream Header field (the first 12 bytes of the .xz Stream). */ | ||
387 | static enum xz_ret XZ_FUNC dec_stream_header(struct xz_dec *s) | ||
388 | { | ||
389 | if (!memeq(s->temp.buf, HEADER_MAGIC, HEADER_MAGIC_SIZE)) | ||
390 | return XZ_FORMAT_ERROR; | ||
391 | |||
392 | if (xz_crc32(s->temp.buf + HEADER_MAGIC_SIZE, 2, 0) | ||
393 | != get_le32(s->temp.buf + HEADER_MAGIC_SIZE + 2)) | ||
394 | return XZ_DATA_ERROR; | ||
395 | |||
396 | if (s->temp.buf[HEADER_MAGIC_SIZE] != 0) | ||
397 | return XZ_OPTIONS_ERROR; | ||
398 | |||
399 | /* | ||
400 | * Of integrity checks, we support only none (Check ID = 0) and | ||
401 | * CRC32 (Check ID = 1). However, if XZ_DEC_ANY_CHECK is defined, | ||
402 | * we will accept other check types too, but then the check won't | ||
403 | * be verified and a warning (XZ_UNSUPPORTED_CHECK) will be given. | ||
404 | */ | ||
405 | s->check_type = s->temp.buf[HEADER_MAGIC_SIZE + 1]; | ||
406 | |||
407 | #ifdef XZ_DEC_ANY_CHECK | ||
408 | if (s->check_type > XZ_CHECK_MAX) | ||
409 | return XZ_OPTIONS_ERROR; | ||
410 | |||
411 | if (s->check_type > XZ_CHECK_CRC32) | ||
412 | return XZ_UNSUPPORTED_CHECK; | ||
413 | #else | ||
414 | if (s->check_type > XZ_CHECK_CRC32) | ||
415 | return XZ_OPTIONS_ERROR; | ||
416 | #endif | ||
417 | |||
418 | return XZ_OK; | ||
419 | } | ||
420 | |||
421 | /* Decode the Stream Footer field (the last 12 bytes of the .xz Stream) */ | ||
422 | static enum xz_ret XZ_FUNC dec_stream_footer(struct xz_dec *s) | ||
423 | { | ||
424 | if (!memeq(s->temp.buf + 10, FOOTER_MAGIC, FOOTER_MAGIC_SIZE)) | ||
425 | return XZ_DATA_ERROR; | ||
426 | |||
427 | if (xz_crc32(s->temp.buf + 4, 6, 0) != get_le32(s->temp.buf)) | ||
428 | return XZ_DATA_ERROR; | ||
429 | |||
430 | /* | ||
431 | * Validate Backward Size. Note that we never added the size of the | ||
432 | * Index CRC32 field to s->index.size, thus we use s->index.size / 4 | ||
433 | * instead of s->index.size / 4 - 1. | ||
434 | */ | ||
435 | if ((s->index.size >> 2) != get_le32(s->temp.buf + 4)) | ||
436 | return XZ_DATA_ERROR; | ||
437 | |||
438 | if (s->temp.buf[8] != 0 || s->temp.buf[9] != s->check_type) | ||
439 | return XZ_DATA_ERROR; | ||
440 | |||
441 | /* | ||
442 | * Use XZ_STREAM_END instead of XZ_OK to be more convenient | ||
443 | * for the caller. | ||
444 | */ | ||
445 | return XZ_STREAM_END; | ||
446 | } | ||
447 | |||
448 | /* Decode the Block Header and initialize the filter chain. */ | ||
449 | static enum xz_ret XZ_FUNC dec_block_header(struct xz_dec *s) | ||
450 | { | ||
451 | enum xz_ret ret; | ||
452 | |||
453 | /* | ||
454 | * Validate the CRC32. We know that the temp buffer is at least | ||
455 | * eight bytes so this is safe. | ||
456 | */ | ||
457 | s->temp.size -= 4; | ||
458 | if (xz_crc32(s->temp.buf, s->temp.size, 0) | ||
459 | != get_le32(s->temp.buf + s->temp.size)) | ||
460 | return XZ_DATA_ERROR; | ||
461 | |||
462 | s->temp.pos = 2; | ||
463 | |||
464 | /* | ||
465 | * Catch unsupported Block Flags. We support only one or two filters | ||
466 | * in the chain, so we catch that with the same test. | ||
467 | */ | ||
468 | #ifdef XZ_DEC_BCJ | ||
469 | if (s->temp.buf[1] & 0x3E) | ||
470 | #else | ||
471 | if (s->temp.buf[1] & 0x3F) | ||
472 | #endif | ||
473 | return XZ_OPTIONS_ERROR; | ||
474 | |||
475 | /* Compressed Size */ | ||
476 | if (s->temp.buf[1] & 0x40) { | ||
477 | if (dec_vli(s, s->temp.buf, &s->temp.pos, s->temp.size) | ||
478 | != XZ_STREAM_END) | ||
479 | return XZ_DATA_ERROR; | ||
480 | |||
481 | s->block_header.compressed = s->vli; | ||
482 | } else { | ||
483 | s->block_header.compressed = VLI_UNKNOWN; | ||
484 | } | ||
485 | |||
486 | /* Uncompressed Size */ | ||
487 | if (s->temp.buf[1] & 0x80) { | ||
488 | if (dec_vli(s, s->temp.buf, &s->temp.pos, s->temp.size) | ||
489 | != XZ_STREAM_END) | ||
490 | return XZ_DATA_ERROR; | ||
491 | |||
492 | s->block_header.uncompressed = s->vli; | ||
493 | } else { | ||
494 | s->block_header.uncompressed = VLI_UNKNOWN; | ||
495 | } | ||
496 | |||
497 | #ifdef XZ_DEC_BCJ | ||
498 | /* If there are two filters, the first one must be a BCJ filter. */ | ||
499 | s->bcj_active = s->temp.buf[1] & 0x01; | ||
500 | if (s->bcj_active) { | ||
501 | if (s->temp.size - s->temp.pos < 2) | ||
502 | return XZ_OPTIONS_ERROR; | ||
503 | |||
504 | ret = xz_dec_bcj_reset(s->bcj, s->temp.buf[s->temp.pos++]); | ||
505 | if (ret != XZ_OK) | ||
506 | return ret; | ||
507 | |||
508 | /* | ||
509 | * We don't support custom start offset, | ||
510 | * so Size of Properties must be zero. | ||
511 | */ | ||
512 | if (s->temp.buf[s->temp.pos++] != 0x00) | ||
513 | return XZ_OPTIONS_ERROR; | ||
514 | } | ||
515 | #endif | ||
516 | |||
517 | /* Valid Filter Flags always take at least two bytes. */ | ||
518 | if (s->temp.size - s->temp.pos < 2) | ||
519 | return XZ_DATA_ERROR; | ||
520 | |||
521 | /* Filter ID = LZMA2 */ | ||
522 | if (s->temp.buf[s->temp.pos++] != 0x21) | ||
523 | return XZ_OPTIONS_ERROR; | ||
524 | |||
525 | /* Size of Properties = 1-byte Filter Properties */ | ||
526 | if (s->temp.buf[s->temp.pos++] != 0x01) | ||
527 | return XZ_OPTIONS_ERROR; | ||
528 | |||
529 | /* Filter Properties contains LZMA2 dictionary size. */ | ||
530 | if (s->temp.size - s->temp.pos < 1) | ||
531 | return XZ_DATA_ERROR; | ||
532 | |||
533 | ret = xz_dec_lzma2_reset(s->lzma2, s->temp.buf[s->temp.pos++]); | ||
534 | if (ret != XZ_OK) | ||
535 | return ret; | ||
536 | |||
537 | /* The rest must be Header Padding. */ | ||
538 | while (s->temp.pos < s->temp.size) | ||
539 | if (s->temp.buf[s->temp.pos++] != 0x00) | ||
540 | return XZ_OPTIONS_ERROR; | ||
541 | |||
542 | s->temp.pos = 0; | ||
543 | s->block.compressed = 0; | ||
544 | s->block.uncompressed = 0; | ||
545 | |||
546 | return XZ_OK; | ||
547 | } | ||
548 | |||
549 | static enum xz_ret XZ_FUNC dec_main(struct xz_dec *s, struct xz_buf *b) | ||
550 | { | ||
551 | enum xz_ret ret; | ||
552 | |||
553 | /* | ||
554 | * Store the start position for the case when we are in the middle | ||
555 | * of the Index field. | ||
556 | */ | ||
557 | s->in_start = b->in_pos; | ||
558 | |||
559 | while (true) { | ||
560 | switch (s->sequence) { | ||
561 | case SEQ_STREAM_HEADER: | ||
562 | /* | ||
563 | * Stream Header is copied to s->temp, and then | ||
564 | * decoded from there. This way if the caller | ||
565 | * gives us only little input at a time, we can | ||
566 | * still keep the Stream Header decoding code | ||
567 | * simple. Similar approach is used in many places | ||
568 | * in this file. | ||
569 | */ | ||
570 | if (!fill_temp(s, b)) | ||
571 | return XZ_OK; | ||
572 | |||
573 | /* | ||
574 | * If dec_stream_header() returns | ||
575 | * XZ_UNSUPPORTED_CHECK, it is still possible | ||
576 | * to continue decoding if working in multi-call | ||
577 | * mode. Thus, update s->sequence before calling | ||
578 | * dec_stream_header(). | ||
579 | */ | ||
580 | s->sequence = SEQ_BLOCK_START; | ||
581 | |||
582 | ret = dec_stream_header(s); | ||
583 | if (ret != XZ_OK) | ||
584 | return ret; | ||
585 | |||
586 | case SEQ_BLOCK_START: | ||
587 | /* We need one byte of input to continue. */ | ||
588 | if (b->in_pos == b->in_size) | ||
589 | return XZ_OK; | ||
590 | |||
591 | /* See if this is the beginning of the Index field. */ | ||
592 | if (b->in[b->in_pos] == 0) { | ||
593 | s->in_start = b->in_pos++; | ||
594 | s->sequence = SEQ_INDEX; | ||
595 | break; | ||
596 | } | ||
597 | |||
598 | /* | ||
599 | * Calculate the size of the Block Header and | ||
600 | * prepare to decode it. | ||
601 | */ | ||
602 | s->block_header.size | ||
603 | = ((uint32_t)b->in[b->in_pos] + 1) * 4; | ||
604 | |||
605 | s->temp.size = s->block_header.size; | ||
606 | s->temp.pos = 0; | ||
607 | s->sequence = SEQ_BLOCK_HEADER; | ||
608 | |||
609 | case SEQ_BLOCK_HEADER: | ||
610 | if (!fill_temp(s, b)) | ||
611 | return XZ_OK; | ||
612 | |||
613 | ret = dec_block_header(s); | ||
614 | if (ret != XZ_OK) | ||
615 | return ret; | ||
616 | |||
617 | s->sequence = SEQ_BLOCK_UNCOMPRESS; | ||
618 | |||
619 | case SEQ_BLOCK_UNCOMPRESS: | ||
620 | ret = dec_block(s, b); | ||
621 | if (ret != XZ_STREAM_END) | ||
622 | return ret; | ||
623 | |||
624 | s->sequence = SEQ_BLOCK_PADDING; | ||
625 | |||
626 | case SEQ_BLOCK_PADDING: | ||
627 | /* | ||
628 | * Size of Compressed Data + Block Padding | ||
629 | * must be a multiple of four. We don't need | ||
630 | * s->block.compressed for anything else | ||
631 | * anymore, so we use it here to test the size | ||
632 | * of the Block Padding field. | ||
633 | */ | ||
634 | while (s->block.compressed & 3) { | ||
635 | if (b->in_pos == b->in_size) | ||
636 | return XZ_OK; | ||
637 | |||
638 | if (b->in[b->in_pos++] != 0) | ||
639 | return XZ_DATA_ERROR; | ||
640 | |||
641 | ++s->block.compressed; | ||
642 | } | ||
643 | |||
644 | s->sequence = SEQ_BLOCK_CHECK; | ||
645 | |||
646 | case SEQ_BLOCK_CHECK: | ||
647 | if (s->check_type == XZ_CHECK_CRC32) { | ||
648 | ret = crc32_validate(s, b); | ||
649 | if (ret != XZ_STREAM_END) | ||
650 | return ret; | ||
651 | } | ||
652 | #ifdef XZ_DEC_ANY_CHECK | ||
653 | else if (!check_skip(s, b)) { | ||
654 | return XZ_OK; | ||
655 | } | ||
656 | #endif | ||
657 | |||
658 | s->sequence = SEQ_BLOCK_START; | ||
659 | break; | ||
660 | |||
661 | case SEQ_INDEX: | ||
662 | ret = dec_index(s, b); | ||
663 | if (ret != XZ_STREAM_END) | ||
664 | return ret; | ||
665 | |||
666 | s->sequence = SEQ_INDEX_PADDING; | ||
667 | |||
668 | case SEQ_INDEX_PADDING: | ||
669 | while ((s->index.size + (b->in_pos - s->in_start)) | ||
670 | & 3) { | ||
671 | if (b->in_pos == b->in_size) { | ||
672 | index_update(s, b); | ||
673 | return XZ_OK; | ||
674 | } | ||
675 | |||
676 | if (b->in[b->in_pos++] != 0) | ||
677 | return XZ_DATA_ERROR; | ||
678 | } | ||
679 | |||
680 | /* Finish the CRC32 value and Index size. */ | ||
681 | index_update(s, b); | ||
682 | |||
683 | /* Compare the hashes to validate the Index field. */ | ||
684 | if (!memeq(&s->block.hash, &s->index.hash, | ||
685 | sizeof(s->block.hash))) | ||
686 | return XZ_DATA_ERROR; | ||
687 | |||
688 | s->sequence = SEQ_INDEX_CRC32; | ||
689 | |||
690 | case SEQ_INDEX_CRC32: | ||
691 | ret = crc32_validate(s, b); | ||
692 | if (ret != XZ_STREAM_END) | ||
693 | return ret; | ||
694 | |||
695 | s->temp.size = STREAM_HEADER_SIZE; | ||
696 | s->sequence = SEQ_STREAM_FOOTER; | ||
697 | |||
698 | case SEQ_STREAM_FOOTER: | ||
699 | if (!fill_temp(s, b)) | ||
700 | return XZ_OK; | ||
701 | |||
702 | return dec_stream_footer(s); | ||
703 | } | ||
704 | } | ||
705 | |||
706 | /* Never reached */ | ||
707 | } | ||
708 | |||
709 | /* | ||
710 | * xz_dec_run() is a wrapper for dec_main() to handle some special cases in | ||
711 | * multi-call and single-call decoding. | ||
712 | * | ||
713 | * In multi-call mode, we must return XZ_BUF_ERROR when it seems clear that we | ||
714 | * are not going to make any progress anymore. This is to prevent the caller | ||
715 | * from calling us infinitely when the input file is truncated or otherwise | ||
716 | * corrupt. Since zlib-style API allows that the caller fills the input buffer | ||
717 | * only when the decoder doesn't produce any new output, we have to be careful | ||
718 | * to avoid returning XZ_BUF_ERROR too easily: XZ_BUF_ERROR is returned only | ||
719 | * after the second consecutive call to xz_dec_run() that makes no progress. | ||
720 | * | ||
721 | * In single-call mode, if we couldn't decode everything and no error | ||
722 | * occurred, either the input is truncated or the output buffer is too small. | ||
723 | * Since we know that the last input byte never produces any output, we know | ||
724 | * that if all the input was consumed and decoding wasn't finished, the file | ||
725 | * must be corrupt. Otherwise the output buffer has to be too small or the | ||
726 | * file is corrupt in a way that decoding it produces too big output. | ||
727 | * | ||
728 | * If single-call decoding fails, we reset b->in_pos and b->out_pos back to | ||
729 | * their original values. This is because with some filter chains there won't | ||
730 | * be any valid uncompressed data in the output buffer unless the decoding | ||
731 | * actually succeeds (that's the price to pay of using the output buffer as | ||
732 | * the workspace). | ||
733 | */ | ||
734 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_run(struct xz_dec *s, struct xz_buf *b) | ||
735 | { | ||
736 | size_t in_start; | ||
737 | size_t out_start; | ||
738 | enum xz_ret ret; | ||
739 | |||
740 | if (s->single_call) | ||
741 | xz_dec_reset(s); | ||
742 | |||
743 | in_start = b->in_pos; | ||
744 | out_start = b->out_pos; | ||
745 | ret = dec_main(s, b); | ||
746 | |||
747 | if (s->single_call) { | ||
748 | if (ret == XZ_OK) | ||
749 | ret = b->in_pos == b->in_size | ||
750 | ? XZ_DATA_ERROR : XZ_BUF_ERROR; | ||
751 | |||
752 | if (ret != XZ_STREAM_END) { | ||
753 | b->in_pos = in_start; | ||
754 | b->out_pos = out_start; | ||
755 | } | ||
756 | |||
757 | } else if (ret == XZ_OK && in_start == b->in_pos | ||
758 | && out_start == b->out_pos) { | ||
759 | if (s->allow_buf_error) | ||
760 | ret = XZ_BUF_ERROR; | ||
761 | |||
762 | s->allow_buf_error = true; | ||
763 | } else { | ||
764 | s->allow_buf_error = false; | ||
765 | } | ||
766 | |||
767 | return ret; | ||
768 | } | ||
769 | |||
770 | XZ_EXTERN struct xz_dec * XZ_FUNC xz_dec_init(uint32_t dict_max) | ||
771 | { | ||
772 | struct xz_dec *s = kmalloc(sizeof(*s), GFP_KERNEL); | ||
773 | if (s == NULL) | ||
774 | return NULL; | ||
775 | |||
776 | s->single_call = dict_max == 0; | ||
777 | |||
778 | #ifdef XZ_DEC_BCJ | ||
779 | s->bcj = xz_dec_bcj_create(s->single_call); | ||
780 | if (s->bcj == NULL) | ||
781 | goto error_bcj; | ||
782 | #endif | ||
783 | |||
784 | s->lzma2 = xz_dec_lzma2_create(dict_max); | ||
785 | if (s->lzma2 == NULL) | ||
786 | goto error_lzma2; | ||
787 | |||
788 | xz_dec_reset(s); | ||
789 | return s; | ||
790 | |||
791 | error_lzma2: | ||
792 | #ifdef XZ_DEC_BCJ | ||
793 | xz_dec_bcj_end(s->bcj); | ||
794 | error_bcj: | ||
795 | #endif | ||
796 | kfree(s); | ||
797 | return NULL; | ||
798 | } | ||
799 | |||
800 | XZ_EXTERN void XZ_FUNC xz_dec_reset(struct xz_dec *s) | ||
801 | { | ||
802 | s->sequence = SEQ_STREAM_HEADER; | ||
803 | s->allow_buf_error = false; | ||
804 | s->pos = 0; | ||
805 | s->crc32 = 0; | ||
806 | memzero(&s->block, sizeof(s->block)); | ||
807 | memzero(&s->index, sizeof(s->index)); | ||
808 | s->temp.pos = 0; | ||
809 | s->temp.size = STREAM_HEADER_SIZE; | ||
810 | } | ||
811 | |||
812 | XZ_EXTERN void XZ_FUNC xz_dec_end(struct xz_dec *s) | ||
813 | { | ||
814 | if (s != NULL) { | ||
815 | xz_dec_lzma2_end(s->lzma2); | ||
816 | #ifdef XZ_DEC_BCJ | ||
817 | xz_dec_bcj_end(s->bcj); | ||
818 | #endif | ||
819 | kfree(s); | ||
820 | } | ||
821 | } | ||
diff --git a/archival/libunarchive/unxz/xz_lzma2.h b/archival/libunarchive/unxz/xz_lzma2.h new file mode 100644 index 000000000..47f21afbc --- /dev/null +++ b/archival/libunarchive/unxz/xz_lzma2.h | |||
@@ -0,0 +1,204 @@ | |||
1 | /* | ||
2 | * LZMA2 definitions | ||
3 | * | ||
4 | * Authors: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * Igor Pavlov <http://7-zip.org/> | ||
6 | * | ||
7 | * This file has been put into the public domain. | ||
8 | * You can do whatever you want with this file. | ||
9 | */ | ||
10 | |||
11 | #ifndef XZ_LZMA2_H | ||
12 | #define XZ_LZMA2_H | ||
13 | |||
14 | /* Range coder constants */ | ||
15 | #define RC_SHIFT_BITS 8 | ||
16 | #define RC_TOP_BITS 24 | ||
17 | #define RC_TOP_VALUE (1 << RC_TOP_BITS) | ||
18 | #define RC_BIT_MODEL_TOTAL_BITS 11 | ||
19 | #define RC_BIT_MODEL_TOTAL (1 << RC_BIT_MODEL_TOTAL_BITS) | ||
20 | #define RC_MOVE_BITS 5 | ||
21 | |||
22 | /* | ||
23 | * Maximum number of position states. A position state is the lowest pb | ||
24 | * number of bits of the current uncompressed offset. In some places there | ||
25 | * are different sets of probabilities for different position states. | ||
26 | */ | ||
27 | #define POS_STATES_MAX (1 << 4) | ||
28 | |||
29 | /* | ||
30 | * This enum is used to track which LZMA symbols have occurred most recently | ||
31 | * and in which order. This information is used to predict the next symbol. | ||
32 | * | ||
33 | * Symbols: | ||
34 | * - Literal: One 8-bit byte | ||
35 | * - Match: Repeat a chunk of data at some distance | ||
36 | * - Long repeat: Multi-byte match at a recently seen distance | ||
37 | * - Short repeat: One-byte repeat at a recently seen distance | ||
38 | * | ||
39 | * The symbol names are in from STATE_oldest_older_previous. REP means | ||
40 | * either short or long repeated match, and NONLIT means any non-literal. | ||
41 | */ | ||
42 | enum lzma_state { | ||
43 | STATE_LIT_LIT, | ||
44 | STATE_MATCH_LIT_LIT, | ||
45 | STATE_REP_LIT_LIT, | ||
46 | STATE_SHORTREP_LIT_LIT, | ||
47 | STATE_MATCH_LIT, | ||
48 | STATE_REP_LIT, | ||
49 | STATE_SHORTREP_LIT, | ||
50 | STATE_LIT_MATCH, | ||
51 | STATE_LIT_LONGREP, | ||
52 | STATE_LIT_SHORTREP, | ||
53 | STATE_NONLIT_MATCH, | ||
54 | STATE_NONLIT_REP | ||
55 | }; | ||
56 | |||
57 | /* Total number of states */ | ||
58 | #define STATES 12 | ||
59 | |||
60 | /* The lowest 7 states indicate that the previous state was a literal. */ | ||
61 | #define LIT_STATES 7 | ||
62 | |||
63 | /* Indicate that the latest symbol was a literal. */ | ||
64 | static inline void XZ_FUNC lzma_state_literal(enum lzma_state *state) | ||
65 | { | ||
66 | if (*state <= STATE_SHORTREP_LIT_LIT) | ||
67 | *state = STATE_LIT_LIT; | ||
68 | else if (*state <= STATE_LIT_SHORTREP) | ||
69 | *state -= 3; | ||
70 | else | ||
71 | *state -= 6; | ||
72 | } | ||
73 | |||
74 | /* Indicate that the latest symbol was a match. */ | ||
75 | static inline void XZ_FUNC lzma_state_match(enum lzma_state *state) | ||
76 | { | ||
77 | *state = *state < LIT_STATES ? STATE_LIT_MATCH : STATE_NONLIT_MATCH; | ||
78 | } | ||
79 | |||
80 | /* Indicate that the latest state was a long repeated match. */ | ||
81 | static inline void XZ_FUNC lzma_state_long_rep(enum lzma_state *state) | ||
82 | { | ||
83 | *state = *state < LIT_STATES ? STATE_LIT_LONGREP : STATE_NONLIT_REP; | ||
84 | } | ||
85 | |||
86 | /* Indicate that the latest symbol was a short match. */ | ||
87 | static inline void XZ_FUNC lzma_state_short_rep(enum lzma_state *state) | ||
88 | { | ||
89 | *state = *state < LIT_STATES ? STATE_LIT_SHORTREP : STATE_NONLIT_REP; | ||
90 | } | ||
91 | |||
92 | /* Test if the previous symbol was a literal. */ | ||
93 | static inline bool XZ_FUNC lzma_state_is_literal(enum lzma_state state) | ||
94 | { | ||
95 | return state < LIT_STATES; | ||
96 | } | ||
97 | |||
98 | /* Each literal coder is divided in three sections: | ||
99 | * - 0x001-0x0FF: Without match byte | ||
100 | * - 0x101-0x1FF: With match byte; match bit is 0 | ||
101 | * - 0x201-0x2FF: With match byte; match bit is 1 | ||
102 | * | ||
103 | * Match byte is used when the previous LZMA symbol was something else than | ||
104 | * a literal (that is, it was some kind of match). | ||
105 | */ | ||
106 | #define LITERAL_CODER_SIZE 0x300 | ||
107 | |||
108 | /* Maximum number of literal coders */ | ||
109 | #define LITERAL_CODERS_MAX (1 << 4) | ||
110 | |||
111 | /* Minimum length of a match is two bytes. */ | ||
112 | #define MATCH_LEN_MIN 2 | ||
113 | |||
114 | /* Match length is encoded with 4, 5, or 10 bits. | ||
115 | * | ||
116 | * Length Bits | ||
117 | * 2-9 4 = Choice=0 + 3 bits | ||
118 | * 10-17 5 = Choice=1 + Choice2=0 + 3 bits | ||
119 | * 18-273 10 = Choice=1 + Choice2=1 + 8 bits | ||
120 | */ | ||
121 | #define LEN_LOW_BITS 3 | ||
122 | #define LEN_LOW_SYMBOLS (1 << LEN_LOW_BITS) | ||
123 | #define LEN_MID_BITS 3 | ||
124 | #define LEN_MID_SYMBOLS (1 << LEN_MID_BITS) | ||
125 | #define LEN_HIGH_BITS 8 | ||
126 | #define LEN_HIGH_SYMBOLS (1 << LEN_HIGH_BITS) | ||
127 | #define LEN_SYMBOLS (LEN_LOW_SYMBOLS + LEN_MID_SYMBOLS + LEN_HIGH_SYMBOLS) | ||
128 | |||
129 | /* | ||
130 | * Maximum length of a match is 273 which is a result of the encoding | ||
131 | * described above. | ||
132 | */ | ||
133 | #define MATCH_LEN_MAX (MATCH_LEN_MIN + LEN_SYMBOLS - 1) | ||
134 | |||
135 | /* | ||
136 | * Different sets of probabilities are used for match distances that have | ||
137 | * very short match length: Lengths of 2, 3, and 4 bytes have a separate | ||
138 | * set of probabilities for each length. The matches with longer length | ||
139 | * use a shared set of probabilities. | ||
140 | */ | ||
141 | #define DIST_STATES 4 | ||
142 | |||
143 | /* | ||
144 | * Get the index of the appropriate probability array for decoding | ||
145 | * the distance slot. | ||
146 | */ | ||
147 | static inline uint32_t XZ_FUNC lzma_get_dist_state(uint32_t len) | ||
148 | { | ||
149 | return len < DIST_STATES + MATCH_LEN_MIN | ||
150 | ? len - MATCH_LEN_MIN : DIST_STATES - 1; | ||
151 | } | ||
152 | |||
153 | /* | ||
154 | * The highest two bits of a 32-bit match distance are encoded using six bits. | ||
155 | * This six-bit value is called a distance slot. This way encoding a 32-bit | ||
156 | * value takes 6-36 bits, larger values taking more bits. | ||
157 | */ | ||
158 | #define DIST_SLOT_BITS 6 | ||
159 | #define DIST_SLOTS (1 << DIST_SLOT_BITS) | ||
160 | |||
161 | /* Match distances up to 127 are fully encoded using probabilities. Since | ||
162 | * the highest two bits (distance slot) are always encoded using six bits, | ||
163 | * the distances 0-3 don't need any additional bits to encode, since the | ||
164 | * distance slot itself is the same as the actual distance. DIST_MODEL_START | ||
165 | * indicates the first distance slot where at least one additional bit is | ||
166 | * needed. | ||
167 | */ | ||
168 | #define DIST_MODEL_START 4 | ||
169 | |||
170 | /* | ||
171 | * Match distances greater than 127 are encoded in three pieces: | ||
172 | * - distance slot: the highest two bits | ||
173 | * - direct bits: 2-26 bits below the highest two bits | ||
174 | * - alignment bits: four lowest bits | ||
175 | * | ||
176 | * Direct bits don't use any probabilities. | ||
177 | * | ||
178 | * The distance slot value of 14 is for distances 128-191. | ||
179 | */ | ||
180 | #define DIST_MODEL_END 14 | ||
181 | |||
182 | /* Distance slots that indicate a distance <= 127. */ | ||
183 | #define FULL_DISTANCES_BITS (DIST_MODEL_END / 2) | ||
184 | #define FULL_DISTANCES (1 << FULL_DISTANCES_BITS) | ||
185 | |||
186 | /* | ||
187 | * For match distances greater than 127, only the highest two bits and the | ||
188 | * lowest four bits (alignment) is encoded using probabilities. | ||
189 | */ | ||
190 | #define ALIGN_BITS 4 | ||
191 | #define ALIGN_SIZE (1 << ALIGN_BITS) | ||
192 | #define ALIGN_MASK (ALIGN_SIZE - 1) | ||
193 | |||
194 | /* Total number of all probability variables */ | ||
195 | #define PROBS_TOTAL (1846 + LITERAL_CODERS_MAX * LITERAL_CODER_SIZE) | ||
196 | |||
197 | /* | ||
198 | * LZMA remembers the four most recent match distances. Reusing these | ||
199 | * distances tends to take less space than re-encoding the actual | ||
200 | * distance value. | ||
201 | */ | ||
202 | #define REPS 4 | ||
203 | |||
204 | #endif | ||
diff --git a/archival/libunarchive/unxz/xz_private.h b/archival/libunarchive/unxz/xz_private.h new file mode 100644 index 000000000..f4e0b4010 --- /dev/null +++ b/archival/libunarchive/unxz/xz_private.h | |||
@@ -0,0 +1,120 @@ | |||
1 | /* | ||
2 | * Private includes and definitions | ||
3 | * | ||
4 | * Author: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * | ||
6 | * This file has been put into the public domain. | ||
7 | * You can do whatever you want with this file. | ||
8 | */ | ||
9 | |||
10 | #ifndef XZ_PRIVATE_H | ||
11 | #define XZ_PRIVATE_H | ||
12 | |||
13 | #ifdef __KERNEL__ | ||
14 | /* XZ_PREBOOT may be defined only via decompress_unxz.c. */ | ||
15 | # ifndef XZ_PREBOOT | ||
16 | # include <linux/slab.h> | ||
17 | # include <linux/vmalloc.h> | ||
18 | # include <linux/string.h> | ||
19 | # define memeq(a, b, size) (memcmp(a, b, size) == 0) | ||
20 | # define memzero(buf, size) memset(buf, 0, size) | ||
21 | # endif | ||
22 | # include <asm/byteorder.h> | ||
23 | # include <asm/unaligned.h> | ||
24 | # define get_le32(p) le32_to_cpup((const uint32_t *)(p)) | ||
25 | /* XZ_IGNORE_KCONFIG may be defined only via decompress_unxz.c. */ | ||
26 | # ifndef XZ_IGNORE_KCONFIG | ||
27 | # ifdef CONFIG_XZ_DEC_X86 | ||
28 | # define XZ_DEC_X86 | ||
29 | # endif | ||
30 | # ifdef CONFIG_XZ_DEC_POWERPC | ||
31 | # define XZ_DEC_POWERPC | ||
32 | # endif | ||
33 | # ifdef CONFIG_XZ_DEC_IA64 | ||
34 | # define XZ_DEC_IA64 | ||
35 | # endif | ||
36 | # ifdef CONFIG_XZ_DEC_ARM | ||
37 | # define XZ_DEC_ARM | ||
38 | # endif | ||
39 | # ifdef CONFIG_XZ_DEC_ARMTHUMB | ||
40 | # define XZ_DEC_ARMTHUMB | ||
41 | # endif | ||
42 | # ifdef CONFIG_XZ_DEC_SPARC | ||
43 | # define XZ_DEC_SPARC | ||
44 | # endif | ||
45 | # endif | ||
46 | # include <linux/xz.h> | ||
47 | #else | ||
48 | /* | ||
49 | * For userspace builds, use a separate header to define the required | ||
50 | * macros and functions. This makes it easier to adapt the code into | ||
51 | * different environments and avoids clutter in the Linux kernel tree. | ||
52 | */ | ||
53 | # include "xz_config.h" | ||
54 | #endif | ||
55 | |||
56 | /* | ||
57 | * If any of the BCJ filter decoders are wanted, define XZ_DEC_BCJ. | ||
58 | * XZ_DEC_BCJ is used to enable generic support for BCJ decoders. | ||
59 | */ | ||
60 | #ifndef XZ_DEC_BCJ | ||
61 | # if defined(XZ_DEC_X86) || defined(XZ_DEC_POWERPC) \ | ||
62 | || defined(XZ_DEC_IA64) || defined(XZ_DEC_ARM) \ | ||
63 | || defined(XZ_DEC_ARM) || defined(XZ_DEC_ARMTHUMB) \ | ||
64 | || defined(XZ_DEC_SPARC) | ||
65 | # define XZ_DEC_BCJ | ||
66 | # endif | ||
67 | #endif | ||
68 | |||
69 | /* | ||
70 | * Allocate memory for LZMA2 decoder. xz_dec_lzma2_reset() must be used | ||
71 | * before calling xz_dec_lzma2_run(). | ||
72 | */ | ||
73 | XZ_EXTERN struct xz_dec_lzma2 * XZ_FUNC xz_dec_lzma2_create( | ||
74 | uint32_t dict_max); | ||
75 | |||
76 | /* | ||
77 | * Decode the LZMA2 properties (one byte) and reset the decoder. Return | ||
78 | * XZ_OK on success, XZ_MEMLIMIT_ERROR if the preallocated dictionary is not | ||
79 | * big enough, and XZ_OPTIONS_ERROR if props indicates something that this | ||
80 | * decoder doesn't support. | ||
81 | */ | ||
82 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_lzma2_reset( | ||
83 | struct xz_dec_lzma2 *s, uint8_t props); | ||
84 | |||
85 | /* Decode raw LZMA2 stream from b->in to b->out. */ | ||
86 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_lzma2_run( | ||
87 | struct xz_dec_lzma2 *s, struct xz_buf *b); | ||
88 | |||
89 | /* Free the memory allocated for the LZMA2 decoder. */ | ||
90 | XZ_EXTERN void XZ_FUNC xz_dec_lzma2_end(struct xz_dec_lzma2 *s); | ||
91 | |||
92 | #ifdef XZ_DEC_BCJ | ||
93 | /* | ||
94 | * Allocate memory for BCJ decoders. xz_dec_bcj_reset() must be used before | ||
95 | * calling xz_dec_bcj_run(). | ||
96 | */ | ||
97 | XZ_EXTERN struct xz_dec_bcj * XZ_FUNC xz_dec_bcj_create(bool single_call); | ||
98 | |||
99 | /* | ||
100 | * Decode the Filter ID of a BCJ filter. This implementation doesn't | ||
101 | * support custom start offsets, so no decoding of Filter Properties | ||
102 | * is needed. Returns XZ_OK if the given Filter ID is supported. | ||
103 | * Otherwise XZ_OPTIONS_ERROR is returned. | ||
104 | */ | ||
105 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_bcj_reset( | ||
106 | struct xz_dec_bcj *s, uint8_t id); | ||
107 | |||
108 | /* | ||
109 | * Decode raw BCJ + LZMA2 stream. This must be used only if there actually is | ||
110 | * a BCJ filter in the chain. If the chain has only LZMA2, xz_dec_lzma2_run() | ||
111 | * must be called directly. | ||
112 | */ | ||
113 | XZ_EXTERN enum xz_ret XZ_FUNC xz_dec_bcj_run(struct xz_dec_bcj *s, | ||
114 | struct xz_dec_lzma2 *lzma2, struct xz_buf *b); | ||
115 | |||
116 | /* Free the memory allocated for the BCJ filters. */ | ||
117 | #define xz_dec_bcj_end(s) kfree(s) | ||
118 | #endif | ||
119 | |||
120 | #endif | ||
diff --git a/archival/libunarchive/unxz/xz_stream.h b/archival/libunarchive/unxz/xz_stream.h new file mode 100644 index 000000000..36f2a7cbf --- /dev/null +++ b/archival/libunarchive/unxz/xz_stream.h | |||
@@ -0,0 +1,57 @@ | |||
1 | /* | ||
2 | * Definitions for handling the .xz file format | ||
3 | * | ||
4 | * Author: Lasse Collin <lasse.collin@tukaani.org> | ||
5 | * | ||
6 | * This file has been put into the public domain. | ||
7 | * You can do whatever you want with this file. | ||
8 | */ | ||
9 | |||
10 | #ifndef XZ_STREAM_H | ||
11 | #define XZ_STREAM_H | ||
12 | |||
13 | #if defined(__KERNEL__) && !XZ_INTERNAL_CRC32 | ||
14 | # include <linux/crc32.h> | ||
15 | # undef crc32 | ||
16 | # define xz_crc32(buf, size, crc) \ | ||
17 | (~crc32_le(~(uint32_t)(crc), buf, size)) | ||
18 | #endif | ||
19 | |||
20 | /* | ||
21 | * See the .xz file format specification at | ||
22 | * http://tukaani.org/xz/xz-file-format.txt | ||
23 | * to understand the container format. | ||
24 | */ | ||
25 | |||
26 | #define STREAM_HEADER_SIZE 12 | ||
27 | |||
28 | #define HEADER_MAGIC "\3757zXZ\0" | ||
29 | #define HEADER_MAGIC_SIZE 6 | ||
30 | |||
31 | #define FOOTER_MAGIC "YZ" | ||
32 | #define FOOTER_MAGIC_SIZE 2 | ||
33 | |||
34 | /* | ||
35 | * Variable-length integer can hold a 63-bit unsigned integer, or a special | ||
36 | * value to indicate that the value is unknown. | ||
37 | */ | ||
38 | typedef uint64_t vli_type; | ||
39 | |||
40 | #define VLI_MAX ((vli_type)-1 / 2) | ||
41 | #define VLI_UNKNOWN ((vli_type)-1) | ||
42 | |||
43 | /* Maximum encoded size of a VLI */ | ||
44 | #define VLI_BYTES_MAX (sizeof(vli_type) * 8 / 7) | ||
45 | |||
46 | /* Integrity Check types */ | ||
47 | enum xz_check { | ||
48 | XZ_CHECK_NONE = 0, | ||
49 | XZ_CHECK_CRC32 = 1, | ||
50 | XZ_CHECK_CRC64 = 4, | ||
51 | XZ_CHECK_SHA256 = 10 | ||
52 | }; | ||
53 | |||
54 | /* Maximum possible Check ID */ | ||
55 | #define XZ_CHECK_MAX 15 | ||
56 | |||
57 | #endif | ||
diff --git a/archival/lzop.c b/archival/lzop.c index 0a15c51aa..ab4d34c88 100644 --- a/archival/lzop.c +++ b/archival/lzop.c | |||
@@ -475,7 +475,7 @@ lzo_crc32(uint32_t c, const uint8_t* buf, unsigned len) | |||
475 | 475 | ||
476 | crc = ~c; | 476 | crc = ~c; |
477 | if (len != 0) do { | 477 | if (len != 0) do { |
478 | crc = G.lzo_crc32_table[((int)crc ^ *buf) & 0xff] ^ (crc >> 8); | 478 | crc = G.lzo_crc32_table[(uint8_t)((int)crc ^ *buf)] ^ (crc >> 8); |
479 | buf += 1; | 479 | buf += 1; |
480 | len -= 1; | 480 | len -= 1; |
481 | } while (len > 0); | 481 | } while (len > 0); |
@@ -738,12 +738,12 @@ static NOINLINE smallint lzo_decompress(const header_t *h) | |||
738 | bb_error_msg_and_die("this file is a split lzop file"); | 738 | bb_error_msg_and_die("this file is a split lzop file"); |
739 | 739 | ||
740 | if (dst_len > MAX_BLOCK_SIZE) | 740 | if (dst_len > MAX_BLOCK_SIZE) |
741 | bb_error_msg_and_die("lzop file corrupted"); | 741 | bb_error_msg_and_die("corrupted data"); |
742 | 742 | ||
743 | /* read compressed block size */ | 743 | /* read compressed block size */ |
744 | src_len = read32(); | 744 | src_len = read32(); |
745 | if (src_len <= 0 || src_len > dst_len) | 745 | if (src_len <= 0 || src_len > dst_len) |
746 | bb_error_msg_and_die("lzop file corrupted"); | 746 | bb_error_msg_and_die("corrupted data"); |
747 | 747 | ||
748 | if (dst_len > block_size) { | 748 | if (dst_len > block_size) { |
749 | if (b2) { | 749 | if (b2) { |
@@ -797,7 +797,7 @@ static NOINLINE smallint lzo_decompress(const header_t *h) | |||
797 | r = lzo1x_decompress_safe(b1, src_len, b2, &d, NULL); | 797 | r = lzo1x_decompress_safe(b1, src_len, b2, &d, NULL); |
798 | 798 | ||
799 | if (r != 0 /*LZO_E_OK*/ || dst_len != d) { | 799 | if (r != 0 /*LZO_E_OK*/ || dst_len != d) { |
800 | bb_error_msg_and_die("corrupted compressed data"); | 800 | bb_error_msg_and_die("corrupted data"); |
801 | } | 801 | } |
802 | dst = b2; | 802 | dst = b2; |
803 | } else { | 803 | } else { |
@@ -1042,7 +1042,7 @@ static smallint do_lzo_decompress(void) | |||
1042 | return lzo_decompress(&header); | 1042 | return lzo_decompress(&header); |
1043 | } | 1043 | } |
1044 | 1044 | ||
1045 | static char* make_new_name_lzop(char *filename) | 1045 | static char* FAST_FUNC make_new_name_lzop(char *filename, const char *expected_ext UNUSED_PARAM) |
1046 | { | 1046 | { |
1047 | if (option_mask32 & OPT_DECOMPRESS) { | 1047 | if (option_mask32 & OPT_DECOMPRESS) { |
1048 | char *extension = strrchr(filename, '.'); | 1048 | char *extension = strrchr(filename, '.'); |
@@ -1054,7 +1054,7 @@ static char* make_new_name_lzop(char *filename) | |||
1054 | return xasprintf("%s.lzo", filename); | 1054 | return xasprintf("%s.lzo", filename); |
1055 | } | 1055 | } |
1056 | 1056 | ||
1057 | static IF_DESKTOP(long long) int pack_lzop(unpack_info_t *info UNUSED_PARAM) | 1057 | static IF_DESKTOP(long long) int FAST_FUNC pack_lzop(unpack_info_t *info UNUSED_PARAM) |
1058 | { | 1058 | { |
1059 | if (option_mask32 & OPT_DECOMPRESS) | 1059 | if (option_mask32 & OPT_DECOMPRESS) |
1060 | return do_lzo_decompress(); | 1060 | return do_lzo_decompress(); |
@@ -1074,5 +1074,5 @@ int lzop_main(int argc UNUSED_PARAM, char **argv) | |||
1074 | option_mask32 |= OPT_DECOMPRESS; | 1074 | option_mask32 |= OPT_DECOMPRESS; |
1075 | 1075 | ||
1076 | G.lzo_crc32_table = crc32_filltable(NULL, 0); | 1076 | G.lzo_crc32_table = crc32_filltable(NULL, 0); |
1077 | return bbunpack(argv, make_new_name_lzop, pack_lzop); | 1077 | return bbunpack(argv, pack_lzop, make_new_name_lzop, /*unused:*/ NULL); |
1078 | } | 1078 | } |
diff --git a/archival/rpm2cpio.c b/archival/rpm2cpio.c index 4ed5b023b..598ec8670 100644 --- a/archival/rpm2cpio.c +++ b/archival/rpm2cpio.c | |||
@@ -68,22 +68,35 @@ int rpm2cpio_main(int argc UNUSED_PARAM, char **argv) | |||
68 | #else | 68 | #else |
69 | /* BLOAT */ | 69 | /* BLOAT */ |
70 | { | 70 | { |
71 | unsigned char magic[2]; | 71 | unsigned char magic[8]; |
72 | IF_DESKTOP(long long) int FAST_FUNC (*unpack)(int src_fd, int dst_fd); | 72 | IF_DESKTOP(long long) int FAST_FUNC (*unpack)(int src_fd, int dst_fd); |
73 | 73 | ||
74 | xread(rpm_fd, &magic, 2); | 74 | xread(rpm_fd, &magic, 2); |
75 | unpack = unpack_gz_stream; | 75 | if (magic[0] == 0x1f && magic[1] == 0x8b) { |
76 | if (magic[0] != 0x1f || magic[1] != 0x8b) { | 76 | unpack = unpack_gz_stream; |
77 | if (!ENABLE_FEATURE_SEAMLESS_BZ2 | 77 | } else |
78 | || magic[0] != 'B' || magic[1] != 'Z' | 78 | if (ENABLE_FEATURE_SEAMLESS_BZ2 |
79 | ) { | 79 | && magic[0] == 'B' && magic[1] == 'Z' |
80 | bb_error_msg_and_die("invalid gzip" | 80 | ) { |
81 | IF_FEATURE_SEAMLESS_BZ2("/bzip2") | ||
82 | " magic"); | ||
83 | } | ||
84 | unpack = unpack_bz2_stream; | 81 | unpack = unpack_bz2_stream; |
82 | } else | ||
83 | if (ENABLE_FEATURE_SEAMLESS_XZ | ||
84 | && magic[0] == 0xfd && magic[1] == '7' | ||
85 | ) { | ||
86 | /* .xz signature: 0xfd, '7', 'z', 'X', 'Z', 0x00 */ | ||
87 | /* More info at: http://tukaani.org/xz/xz-file-format.txt */ | ||
88 | xread(rpm_fd, magic + 2, 4); | ||
89 | if (strcmp((char*)magic + 2, "zXZ") != 0) | ||
90 | goto no_magic; | ||
91 | xlseek(rpm_fd, -6, SEEK_CUR); | ||
92 | unpack = unpack_xz_stream; | ||
93 | } else { | ||
94 | no_magic: | ||
95 | bb_error_msg_and_die("no gzip" | ||
96 | IF_FEATURE_SEAMLESS_BZ2("/bzip2") | ||
97 | IF_FEATURE_SEAMLESS_XZ("/xz") | ||
98 | " magic"); | ||
85 | } | 99 | } |
86 | |||
87 | if (unpack(rpm_fd, STDOUT_FILENO) < 0) | 100 | if (unpack(rpm_fd, STDOUT_FILENO) < 0) |
88 | bb_error_msg_and_die("error unpacking"); | 101 | bb_error_msg_and_die("error unpacking"); |
89 | } | 102 | } |
diff --git a/archival/unzip.c b/archival/unzip.c index afab3280d..1d3291ab8 100644 --- a/archival/unzip.c +++ b/archival/unzip.c | |||
@@ -25,12 +25,12 @@ | |||
25 | enum { | 25 | enum { |
26 | #if BB_BIG_ENDIAN | 26 | #if BB_BIG_ENDIAN |
27 | ZIP_FILEHEADER_MAGIC = 0x504b0304, | 27 | ZIP_FILEHEADER_MAGIC = 0x504b0304, |
28 | ZIP_CDS_MAGIC = 0x504b0102, | 28 | ZIP_CDF_MAGIC = 0x504b0102, /* central directory's file header */ |
29 | ZIP_CDE_MAGIC = 0x504b0506, | 29 | ZIP_CDE_MAGIC = 0x504b0506, /* "end of central directory" record */ |
30 | ZIP_DD_MAGIC = 0x504b0708, | 30 | ZIP_DD_MAGIC = 0x504b0708, |
31 | #else | 31 | #else |
32 | ZIP_FILEHEADER_MAGIC = 0x04034b50, | 32 | ZIP_FILEHEADER_MAGIC = 0x04034b50, |
33 | ZIP_CDS_MAGIC = 0x02014b50, | 33 | ZIP_CDF_MAGIC = 0x02014b50, |
34 | ZIP_CDE_MAGIC = 0x06054b50, | 34 | ZIP_CDE_MAGIC = 0x06054b50, |
35 | ZIP_DD_MAGIC = 0x08074b50, | 35 | ZIP_DD_MAGIC = 0x08074b50, |
36 | #endif | 36 | #endif |
@@ -42,7 +42,7 @@ typedef union { | |||
42 | uint8_t raw[ZIP_HEADER_LEN]; | 42 | uint8_t raw[ZIP_HEADER_LEN]; |
43 | struct { | 43 | struct { |
44 | uint16_t version; /* 0-1 */ | 44 | uint16_t version; /* 0-1 */ |
45 | uint16_t flags; /* 2-3 */ | 45 | uint16_t zip_flags; /* 2-3 */ |
46 | uint16_t method; /* 4-5 */ | 46 | uint16_t method; /* 4-5 */ |
47 | uint16_t modtime; /* 6-7 */ | 47 | uint16_t modtime; /* 6-7 */ |
48 | uint16_t moddate; /* 8-9 */ | 48 | uint16_t moddate; /* 8-9 */ |
@@ -66,7 +66,6 @@ struct BUG_zip_header_must_be_26_bytes { | |||
66 | 66 | ||
67 | #define FIX_ENDIANNESS_ZIP(zip_header) do { \ | 67 | #define FIX_ENDIANNESS_ZIP(zip_header) do { \ |
68 | (zip_header).formatted.version = SWAP_LE16((zip_header).formatted.version ); \ | 68 | (zip_header).formatted.version = SWAP_LE16((zip_header).formatted.version ); \ |
69 | (zip_header).formatted.flags = SWAP_LE16((zip_header).formatted.flags ); \ | ||
70 | (zip_header).formatted.method = SWAP_LE16((zip_header).formatted.method ); \ | 69 | (zip_header).formatted.method = SWAP_LE16((zip_header).formatted.method ); \ |
71 | (zip_header).formatted.modtime = SWAP_LE16((zip_header).formatted.modtime ); \ | 70 | (zip_header).formatted.modtime = SWAP_LE16((zip_header).formatted.modtime ); \ |
72 | (zip_header).formatted.moddate = SWAP_LE16((zip_header).formatted.moddate ); \ | 71 | (zip_header).formatted.moddate = SWAP_LE16((zip_header).formatted.moddate ); \ |
@@ -77,15 +76,15 @@ struct BUG_zip_header_must_be_26_bytes { | |||
77 | (zip_header).formatted.extra_len = SWAP_LE16((zip_header).formatted.extra_len ); \ | 76 | (zip_header).formatted.extra_len = SWAP_LE16((zip_header).formatted.extra_len ); \ |
78 | } while (0) | 77 | } while (0) |
79 | 78 | ||
80 | #define CDS_HEADER_LEN 42 | 79 | #define CDF_HEADER_LEN 42 |
81 | 80 | ||
82 | typedef union { | 81 | typedef union { |
83 | uint8_t raw[CDS_HEADER_LEN]; | 82 | uint8_t raw[CDF_HEADER_LEN]; |
84 | struct { | 83 | struct { |
85 | /* uint32_t signature; 50 4b 01 02 */ | 84 | /* uint32_t signature; 50 4b 01 02 */ |
86 | uint16_t version_made_by; /* 0-1 */ | 85 | uint16_t version_made_by; /* 0-1 */ |
87 | uint16_t version_needed; /* 2-3 */ | 86 | uint16_t version_needed; /* 2-3 */ |
88 | uint16_t cds_flags; /* 4-5 */ | 87 | uint16_t cdf_flags; /* 4-5 */ |
89 | uint16_t method; /* 6-7 */ | 88 | uint16_t method; /* 6-7 */ |
90 | uint16_t mtime; /* 8-9 */ | 89 | uint16_t mtime; /* 8-9 */ |
91 | uint16_t mdate; /* 10-11 */ | 90 | uint16_t mdate; /* 10-11 */ |
@@ -100,21 +99,25 @@ typedef union { | |||
100 | uint32_t external_file_attributes PACKED; /* 34-37 */ | 99 | uint32_t external_file_attributes PACKED; /* 34-37 */ |
101 | uint32_t relative_offset_of_local_header PACKED; /* 38-41 */ | 100 | uint32_t relative_offset_of_local_header PACKED; /* 38-41 */ |
102 | } formatted PACKED; | 101 | } formatted PACKED; |
103 | } cds_header_t; | 102 | } cdf_header_t; |
104 | 103 | ||
105 | struct BUG_cds_header_must_be_42_bytes { | 104 | struct BUG_cdf_header_must_be_42_bytes { |
106 | char BUG_cds_header_must_be_42_bytes[ | 105 | char BUG_cdf_header_must_be_42_bytes[ |
107 | offsetof(cds_header_t, formatted.relative_offset_of_local_header) + 4 | 106 | offsetof(cdf_header_t, formatted.relative_offset_of_local_header) + 4 |
108 | == CDS_HEADER_LEN ? 1 : -1]; | 107 | == CDF_HEADER_LEN ? 1 : -1]; |
109 | }; | 108 | }; |
110 | 109 | ||
111 | #define FIX_ENDIANNESS_CDS(cds_header) do { \ | 110 | #define FIX_ENDIANNESS_CDF(cdf_header) do { \ |
112 | (cds_header).formatted.crc32 = SWAP_LE32((cds_header).formatted.crc32 ); \ | 111 | (cdf_header).formatted.crc32 = SWAP_LE32((cdf_header).formatted.crc32 ); \ |
113 | (cds_header).formatted.cmpsize = SWAP_LE32((cds_header).formatted.cmpsize ); \ | 112 | (cdf_header).formatted.cmpsize = SWAP_LE32((cdf_header).formatted.cmpsize ); \ |
114 | (cds_header).formatted.ucmpsize = SWAP_LE32((cds_header).formatted.ucmpsize ); \ | 113 | (cdf_header).formatted.ucmpsize = SWAP_LE32((cdf_header).formatted.ucmpsize ); \ |
115 | (cds_header).formatted.file_name_length = SWAP_LE16((cds_header).formatted.file_name_length); \ | 114 | (cdf_header).formatted.file_name_length = SWAP_LE16((cdf_header).formatted.file_name_length); \ |
116 | (cds_header).formatted.extra_field_length = SWAP_LE16((cds_header).formatted.extra_field_length); \ | 115 | (cdf_header).formatted.extra_field_length = SWAP_LE16((cdf_header).formatted.extra_field_length); \ |
117 | (cds_header).formatted.file_comment_length = SWAP_LE16((cds_header).formatted.file_comment_length); \ | 116 | (cdf_header).formatted.file_comment_length = SWAP_LE16((cdf_header).formatted.file_comment_length); \ |
117 | IF_DESKTOP( \ | ||
118 | (cdf_header).formatted.version_made_by = SWAP_LE16((cdf_header).formatted.version_made_by); \ | ||
119 | (cdf_header).formatted.external_file_attributes = SWAP_LE32((cdf_header).formatted.external_file_attributes); \ | ||
120 | ) \ | ||
118 | } while (0) | 121 | } while (0) |
119 | 122 | ||
120 | #define CDE_HEADER_LEN 16 | 123 | #define CDE_HEADER_LEN 16 |
@@ -124,11 +127,11 @@ typedef union { | |||
124 | struct { | 127 | struct { |
125 | /* uint32_t signature; 50 4b 05 06 */ | 128 | /* uint32_t signature; 50 4b 05 06 */ |
126 | uint16_t this_disk_no; | 129 | uint16_t this_disk_no; |
127 | uint16_t disk_with_cds_no; | 130 | uint16_t disk_with_cdf_no; |
128 | uint16_t cds_entries_on_this_disk; | 131 | uint16_t cdf_entries_on_this_disk; |
129 | uint16_t cds_entries_total; | 132 | uint16_t cdf_entries_total; |
130 | uint32_t cds_size; | 133 | uint32_t cdf_size; |
131 | uint32_t cds_offset; | 134 | uint32_t cdf_offset; |
132 | /* uint16_t file_comment_length; */ | 135 | /* uint16_t file_comment_length; */ |
133 | /* .ZIP file comment (variable size) */ | 136 | /* .ZIP file comment (variable size) */ |
134 | } formatted PACKED; | 137 | } formatted PACKED; |
@@ -140,7 +143,7 @@ struct BUG_cde_header_must_be_16_bytes { | |||
140 | }; | 143 | }; |
141 | 144 | ||
142 | #define FIX_ENDIANNESS_CDE(cde_header) do { \ | 145 | #define FIX_ENDIANNESS_CDE(cde_header) do { \ |
143 | (cde_header).formatted.cds_offset = SWAP_LE32((cde_header).formatted.cds_offset); \ | 146 | (cde_header).formatted.cdf_offset = SWAP_LE32((cde_header).formatted.cdf_offset); \ |
144 | } while (0) | 147 | } while (0) |
145 | 148 | ||
146 | enum { zip_fd = 3 }; | 149 | enum { zip_fd = 3 }; |
@@ -148,7 +151,7 @@ enum { zip_fd = 3 }; | |||
148 | 151 | ||
149 | #if ENABLE_DESKTOP | 152 | #if ENABLE_DESKTOP |
150 | /* NB: does not preserve file position! */ | 153 | /* NB: does not preserve file position! */ |
151 | static uint32_t find_cds_offset(void) | 154 | static uint32_t find_cdf_offset(void) |
152 | { | 155 | { |
153 | unsigned char buf[1024]; | 156 | unsigned char buf[1024]; |
154 | cde_header_t cde_header; | 157 | cde_header_t cde_header; |
@@ -156,9 +159,9 @@ static uint32_t find_cds_offset(void) | |||
156 | off_t end; | 159 | off_t end; |
157 | 160 | ||
158 | end = xlseek(zip_fd, 0, SEEK_END); | 161 | end = xlseek(zip_fd, 0, SEEK_END); |
159 | if (end < 1024) | ||
160 | end = 1024; | ||
161 | end -= 1024; | 162 | end -= 1024; |
163 | if (end < 0) | ||
164 | end = 0; | ||
162 | xlseek(zip_fd, end, SEEK_SET); | 165 | xlseek(zip_fd, end, SEEK_SET); |
163 | full_read(zip_fd, buf, 1024); | 166 | full_read(zip_fd, buf, 1024); |
164 | 167 | ||
@@ -177,32 +180,30 @@ static uint32_t find_cds_offset(void) | |||
177 | /* we found CDE! */ | 180 | /* we found CDE! */ |
178 | memcpy(cde_header.raw, p + 1, CDE_HEADER_LEN); | 181 | memcpy(cde_header.raw, p + 1, CDE_HEADER_LEN); |
179 | FIX_ENDIANNESS_CDE(cde_header); | 182 | FIX_ENDIANNESS_CDE(cde_header); |
180 | return cde_header.formatted.cds_offset; | 183 | return cde_header.formatted.cdf_offset; |
181 | } | 184 | } |
182 | bb_error_msg_and_die("can't find file table"); | 185 | bb_error_msg_and_die("can't find file table"); |
183 | }; | 186 | }; |
184 | 187 | ||
185 | static uint32_t read_next_cds(int count_m1, uint32_t cds_offset, cds_header_t *cds_ptr) | 188 | static uint32_t read_next_cdf(uint32_t cdf_offset, cdf_header_t *cdf_ptr) |
186 | { | 189 | { |
187 | off_t org; | 190 | off_t org; |
188 | 191 | ||
189 | org = xlseek(zip_fd, 0, SEEK_CUR); | 192 | org = xlseek(zip_fd, 0, SEEK_CUR); |
190 | 193 | ||
191 | if (!cds_offset) | 194 | if (!cdf_offset) |
192 | cds_offset = find_cds_offset(); | 195 | cdf_offset = find_cdf_offset(); |
193 | 196 | ||
194 | while (count_m1-- >= 0) { | 197 | xlseek(zip_fd, cdf_offset + 4, SEEK_SET); |
195 | xlseek(zip_fd, cds_offset + 4, SEEK_SET); | 198 | xread(zip_fd, cdf_ptr->raw, CDF_HEADER_LEN); |
196 | xread(zip_fd, cds_ptr->raw, CDS_HEADER_LEN); | 199 | FIX_ENDIANNESS_CDF(*cdf_ptr); |
197 | FIX_ENDIANNESS_CDS(*cds_ptr); | 200 | cdf_offset += 4 + CDF_HEADER_LEN |
198 | cds_offset += 4 + CDS_HEADER_LEN | 201 | + cdf_ptr->formatted.file_name_length |
199 | + cds_ptr->formatted.file_name_length | 202 | + cdf_ptr->formatted.extra_field_length |
200 | + cds_ptr->formatted.extra_field_length | 203 | + cdf_ptr->formatted.file_comment_length; |
201 | + cds_ptr->formatted.file_comment_length; | ||
202 | } | ||
203 | 204 | ||
204 | xlseek(zip_fd, org, SEEK_SET); | 205 | xlseek(zip_fd, org, SEEK_SET); |
205 | return cds_offset; | 206 | return cdf_offset; |
206 | }; | 207 | }; |
207 | #endif | 208 | #endif |
208 | 209 | ||
@@ -258,8 +259,7 @@ int unzip_main(int argc, char **argv) | |||
258 | smallint listing = 0; | 259 | smallint listing = 0; |
259 | smallint overwrite = O_PROMPT; | 260 | smallint overwrite = O_PROMPT; |
260 | #if ENABLE_DESKTOP | 261 | #if ENABLE_DESKTOP |
261 | uint32_t cds_offset; | 262 | uint32_t cdf_offset; |
262 | unsigned cds_entries; | ||
263 | #endif | 263 | #endif |
264 | unsigned long total_usize; | 264 | unsigned long total_usize; |
265 | unsigned long total_size; | 265 | unsigned long total_size; |
@@ -435,20 +435,42 @@ int unzip_main(int argc, char **argv) | |||
435 | } | 435 | } |
436 | } | 436 | } |
437 | 437 | ||
438 | /* Example of an archive with one 0-byte long file named 'z' | ||
439 | * created by Zip 2.31 on Unix: | ||
440 | * 0000 [50 4b]03 04 0a 00 00 00 00 00 42 1a b8 3c 00 00 |PK........B..<..| | ||
441 | * sig........ vneed flags compr mtime mdate crc32> | ||
442 | * 0010 00 00 00 00 00 00 00 00 00 00 01 00 15 00 7a 55 |..............zU| | ||
443 | * >..... csize...... usize...... fnlen exlen fn ex> | ||
444 | * 0020 54 09 00 03 cc d3 f9 4b cc d3 f9 4b 55 78 04 00 |T......K...KUx..| | ||
445 | * >tra_field...................................... | ||
446 | * 0030 00 00 00 00[50 4b]01 02 17 03 0a 00 00 00 00 00 |....PK..........| | ||
447 | * ........... sig........ vmade vneed flags compr | ||
448 | * 0040 42 1a b8 3c 00 00 00 00 00 00 00 00 00 00 00 00 |B..<............| | ||
449 | * mtime mdate crc32...... csize...... usize...... | ||
450 | * 0050 01 00 0d 00 00 00 00 00 00 00 00 00 a4 81 00 00 |................| | ||
451 | * fnlen exlen clen. dnum. iattr eattr...... relofs> (eattr = rw-r--r--) | ||
452 | * 0060 00 00 7a 55 54 05 00 03 cc d3 f9 4b 55 78 00 00 |..zUT......KUx..| | ||
453 | * >..... fn extra_field........................... | ||
454 | * 0070 [50 4b]05 06 00 00 00 00 01 00 01 00 3c 00 00 00 |PK..........<...| | ||
455 | * 0080 34 00 00 00 00 00 |4.....| | ||
456 | */ | ||
438 | total_usize = 0; | 457 | total_usize = 0; |
439 | total_size = 0; | 458 | total_size = 0; |
440 | total_entries = 0; | 459 | total_entries = 0; |
441 | #if ENABLE_DESKTOP | 460 | #if ENABLE_DESKTOP |
442 | cds_entries = 0; | 461 | cdf_offset = 0; |
443 | cds_offset = 0; | ||
444 | #endif | 462 | #endif |
445 | while (1) { | 463 | while (1) { |
446 | uint32_t magic; | 464 | uint32_t magic; |
465 | mode_t dir_mode = 0777; | ||
466 | #if ENABLE_DESKTOP | ||
467 | mode_t file_mode = 0666; | ||
468 | #endif | ||
447 | 469 | ||
448 | /* Check magic number */ | 470 | /* Check magic number */ |
449 | xread(zip_fd, &magic, 4); | 471 | xread(zip_fd, &magic, 4); |
450 | /* Central directory? It's at the end, so exit */ | 472 | /* Central directory? It's at the end, so exit */ |
451 | if (magic == ZIP_CDS_MAGIC) | 473 | if (magic == ZIP_CDF_MAGIC) |
452 | break; | 474 | break; |
453 | #if ENABLE_DESKTOP | 475 | #if ENABLE_DESKTOP |
454 | /* Data descriptor? It was a streaming file, go on */ | 476 | /* Data descriptor? It was a streaming file, go on */ |
@@ -468,23 +490,29 @@ int unzip_main(int argc, char **argv) | |||
468 | bb_error_msg_and_die("unsupported method %d", zip_header.formatted.method); | 490 | bb_error_msg_and_die("unsupported method %d", zip_header.formatted.method); |
469 | } | 491 | } |
470 | #if !ENABLE_DESKTOP | 492 | #if !ENABLE_DESKTOP |
471 | if (zip_header.formatted.flags & 0x0009) { | 493 | if (zip_header.formatted.zip_flags & SWAP_LE16(0x0009)) { |
472 | bb_error_msg_and_die("zip flags 1 and 8 are not supported"); | 494 | bb_error_msg_and_die("zip flags 1 and 8 are not supported"); |
473 | } | 495 | } |
474 | #else | 496 | #else |
475 | if (zip_header.formatted.flags & 0x0001) { | 497 | if (zip_header.formatted.zip_flags & SWAP_LE16(0x0001)) { |
476 | /* 0x0001 - encrypted */ | 498 | /* 0x0001 - encrypted */ |
477 | bb_error_msg_and_die("zip flag 1 (encryption) is not supported"); | 499 | bb_error_msg_and_die("zip flag 1 (encryption) is not supported"); |
478 | } | 500 | } |
479 | if (zip_header.formatted.flags & 0x0008) { | 501 | |
480 | cds_header_t cds_header; | 502 | { |
481 | /* 0x0008 - streaming. [u]cmpsize can be reliably gotten | 503 | cdf_header_t cdf_header; |
482 | * only from Central Directory. See unzip_doc.txt */ | 504 | cdf_offset = read_next_cdf(cdf_offset, &cdf_header); |
483 | cds_offset = read_next_cds(total_entries - cds_entries, cds_offset, &cds_header); | 505 | if (zip_header.formatted.zip_flags & SWAP_LE16(0x0008)) { |
484 | cds_entries = total_entries + 1; | 506 | /* 0x0008 - streaming. [u]cmpsize can be reliably gotten |
485 | zip_header.formatted.crc32 = cds_header.formatted.crc32; | 507 | * only from Central Directory. See unzip_doc.txt */ |
486 | zip_header.formatted.cmpsize = cds_header.formatted.cmpsize; | 508 | zip_header.formatted.crc32 = cdf_header.formatted.crc32; |
487 | zip_header.formatted.ucmpsize = cds_header.formatted.ucmpsize; | 509 | zip_header.formatted.cmpsize = cdf_header.formatted.cmpsize; |
510 | zip_header.formatted.ucmpsize = cdf_header.formatted.ucmpsize; | ||
511 | } | ||
512 | if ((cdf_header.formatted.version_made_by >> 8) == 3) { | ||
513 | /* this archive is created on Unix */ | ||
514 | dir_mode = file_mode = (cdf_header.formatted.external_file_attributes >> 16); | ||
515 | } | ||
488 | } | 516 | } |
489 | #endif | 517 | #endif |
490 | 518 | ||
@@ -550,7 +578,7 @@ int unzip_main(int argc, char **argv) | |||
550 | printf(" creating: %s\n", dst_fn); | 578 | printf(" creating: %s\n", dst_fn); |
551 | } | 579 | } |
552 | unzip_create_leading_dirs(dst_fn); | 580 | unzip_create_leading_dirs(dst_fn); |
553 | if (bb_make_directory(dst_fn, 0777, 0)) { | 581 | if (bb_make_directory(dst_fn, dir_mode, 0)) { |
554 | bb_error_msg_and_die("exiting"); | 582 | bb_error_msg_and_die("exiting"); |
555 | } | 583 | } |
556 | } else { | 584 | } else { |
@@ -592,7 +620,11 @@ int unzip_main(int argc, char **argv) | |||
592 | overwrite = O_ALWAYS; | 620 | overwrite = O_ALWAYS; |
593 | case 'y': /* Open file and fall into unzip */ | 621 | case 'y': /* Open file and fall into unzip */ |
594 | unzip_create_leading_dirs(dst_fn); | 622 | unzip_create_leading_dirs(dst_fn); |
623 | #if ENABLE_DESKTOP | ||
624 | dst_fd = xopen3(dst_fn, O_WRONLY | O_CREAT | O_TRUNC, file_mode); | ||
625 | #else | ||
595 | dst_fd = xopen(dst_fn, O_WRONLY | O_CREAT | O_TRUNC); | 626 | dst_fd = xopen(dst_fn, O_WRONLY | O_CREAT | O_TRUNC); |
627 | #endif | ||
596 | case -1: /* Unzip */ | 628 | case -1: /* Unzip */ |
597 | if (!quiet) { | 629 | if (!quiet) { |
598 | printf(" inflating: %s\n", dst_fn); | 630 | printf(" inflating: %s\n", dst_fn); |
diff --git a/console-tools/Config.in b/console-tools/Config.in index 195685b97..a7e995936 100644 --- a/console-tools/Config.in +++ b/console-tools/Config.in | |||
@@ -12,6 +12,12 @@ config CHVT | |||
12 | This program is used to change to another terminal. | 12 | This program is used to change to another terminal. |
13 | Example: chvt 4 (change to terminal /dev/tty4) | 13 | Example: chvt 4 (change to terminal /dev/tty4) |
14 | 14 | ||
15 | config FGCONSOLE | ||
16 | bool "fgconsole" | ||
17 | default n | ||
18 | help | ||
19 | This program prints active (foreground) console number. | ||
20 | |||
15 | config CLEAR | 21 | config CLEAR |
16 | bool "clear" | 22 | bool "clear" |
17 | default n | 23 | default n |
diff --git a/console-tools/Kbuild b/console-tools/Kbuild index df5ffdb96..ad8b8ce77 100644 --- a/console-tools/Kbuild +++ b/console-tools/Kbuild | |||
@@ -6,6 +6,7 @@ | |||
6 | 6 | ||
7 | lib-y:= | 7 | lib-y:= |
8 | lib-$(CONFIG_CHVT) += chvt.o | 8 | lib-$(CONFIG_CHVT) += chvt.o |
9 | lib-$(CONFIG_FGCONSOLE) += fgconsole.o | ||
9 | lib-$(CONFIG_CLEAR) += clear.o | 10 | lib-$(CONFIG_CLEAR) += clear.o |
10 | lib-$(CONFIG_DEALLOCVT) += deallocvt.o | 11 | lib-$(CONFIG_DEALLOCVT) += deallocvt.o |
11 | lib-$(CONFIG_DUMPKMAP) += dumpkmap.o | 12 | lib-$(CONFIG_DUMPKMAP) += dumpkmap.o |
diff --git a/console-tools/fgconsole.c b/console-tools/fgconsole.c new file mode 100644 index 000000000..75fd98fd5 --- /dev/null +++ b/console-tools/fgconsole.c | |||
@@ -0,0 +1,30 @@ | |||
1 | /* vi: set sw=4 ts=4: */ | ||
2 | /* | ||
3 | * Mini fgconsole implementation for busybox | ||
4 | * | ||
5 | * Copyright (C) 2010 by Grigory Batalov <bga@altlinux.org> | ||
6 | * | ||
7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. | ||
8 | */ | ||
9 | |||
10 | #include "libbb.h" | ||
11 | |||
12 | /* From <linux/vt.h> */ | ||
13 | struct vt_stat { | ||
14 | unsigned short v_active; /* active vt */ | ||
15 | unsigned short v_signal; /* signal to send */ | ||
16 | unsigned short v_state; /* vt bitmask */ | ||
17 | }; | ||
18 | enum { VT_GETSTATE = 0x5603 }; /* get global vt state info */ | ||
19 | |||
20 | int fgconsole_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; | ||
21 | int fgconsole_main(int argc UNUSED_PARAM, char **argv UNUSED_PARAM) | ||
22 | { | ||
23 | struct vt_stat vtstat; | ||
24 | |||
25 | vtstat.v_active = 0; | ||
26 | xioctl(get_console_fd_or_die(), VT_GETSTATE, &vtstat); | ||
27 | printf("%d\n", vtstat.v_active); | ||
28 | |||
29 | return EXIT_SUCCESS; | ||
30 | } | ||
diff --git a/coreutils/od_bloaty.c b/coreutils/od_bloaty.c index 7b1ab8981..cb97ff8c5 100644 --- a/coreutils/od_bloaty.c +++ b/coreutils/od_bloaty.c | |||
@@ -528,7 +528,7 @@ check_and_close(void) | |||
528 | S_ORIG is solely for reporting errors. It should be the full format | 528 | S_ORIG is solely for reporting errors. It should be the full format |
529 | string argument. */ | 529 | string argument. */ |
530 | 530 | ||
531 | static const char * | 531 | static NOINLINE const char * |
532 | decode_one_format(const char *s_orig, const char *s, struct tspec *tspec) | 532 | decode_one_format(const char *s_orig, const char *s, struct tspec *tspec) |
533 | { | 533 | { |
534 | enum size_spec size_spec; | 534 | enum size_spec size_spec; |
diff --git a/include/applets.h b/include/applets.h index ff8799c63..d8a706b44 100644 --- a/include/applets.h +++ b/include/applets.h | |||
@@ -157,6 +157,7 @@ IF_FBSPLASH(APPLET(fbsplash, _BB_DIR_SBIN, _BB_SUID_DROP)) | |||
157 | IF_FDFLUSH(APPLET_ODDNAME(fdflush, freeramdisk, _BB_DIR_BIN, _BB_SUID_DROP, fdflush)) | 157 | IF_FDFLUSH(APPLET_ODDNAME(fdflush, freeramdisk, _BB_DIR_BIN, _BB_SUID_DROP, fdflush)) |
158 | IF_FDFORMAT(APPLET(fdformat, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 158 | IF_FDFORMAT(APPLET(fdformat, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
159 | IF_FDISK(APPLET(fdisk, _BB_DIR_SBIN, _BB_SUID_DROP)) | 159 | IF_FDISK(APPLET(fdisk, _BB_DIR_SBIN, _BB_SUID_DROP)) |
160 | IF_FGCONSOLE(APPLET(fgconsole, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | ||
160 | IF_FEATURE_GREP_FGREP_ALIAS(APPLET_ODDNAME(fgrep, grep, _BB_DIR_BIN, _BB_SUID_DROP, fgrep)) | 161 | IF_FEATURE_GREP_FGREP_ALIAS(APPLET_ODDNAME(fgrep, grep, _BB_DIR_BIN, _BB_SUID_DROP, fgrep)) |
161 | IF_FIND(APPLET_NOEXEC(find, find, _BB_DIR_USR_BIN, _BB_SUID_DROP, find)) | 162 | IF_FIND(APPLET_NOEXEC(find, find, _BB_DIR_USR_BIN, _BB_SUID_DROP, find)) |
162 | IF_FINDFS(APPLET(findfs, _BB_DIR_SBIN, _BB_SUID_MAYBE)) | 163 | IF_FINDFS(APPLET(findfs, _BB_DIR_SBIN, _BB_SUID_MAYBE)) |
@@ -252,7 +253,8 @@ IF_LSMOD(APPLET(lsmod, _BB_DIR_SBIN, _BB_SUID_DROP)) | |||
252 | IF_MODPROBE_SMALL(APPLET_ODDNAME(lsmod, modprobe, _BB_DIR_SBIN, _BB_SUID_DROP, modprobe)) | 253 | IF_MODPROBE_SMALL(APPLET_ODDNAME(lsmod, modprobe, _BB_DIR_SBIN, _BB_SUID_DROP, modprobe)) |
253 | IF_LSPCI(APPLET(lspci, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 254 | IF_LSPCI(APPLET(lspci, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
254 | IF_LSUSB(APPLET(lsusb, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 255 | IF_LSUSB(APPLET(lsusb, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
255 | IF_UNLZMA(APPLET_ODDNAME(lzmacat, unlzma, _BB_DIR_USR_BIN, _BB_SUID_DROP, lzmacat)) | 256 | IF_UNLZMA(APPLET_ODDNAME(lzcat, unlzma, _BB_DIR_USR_BIN, _BB_SUID_DROP, lzcat)) |
257 | IF_LZMA(APPLET_ODDNAME(lzma, unlzma, _BB_DIR_USR_BIN, _BB_SUID_DROP, lzma)) | ||
256 | IF_LZOP(APPLET(lzop, _BB_DIR_BIN, _BB_SUID_DROP)) | 258 | IF_LZOP(APPLET(lzop, _BB_DIR_BIN, _BB_SUID_DROP)) |
257 | IF_LZOP(APPLET_ODDNAME(lzopcat, lzop, _BB_DIR_USR_BIN, _BB_SUID_DROP, lzopcat)) | 259 | IF_LZOP(APPLET_ODDNAME(lzopcat, lzop, _BB_DIR_USR_BIN, _BB_SUID_DROP, lzopcat)) |
258 | IF_MAKEDEVS(APPLET(makedevs, _BB_DIR_SBIN, _BB_SUID_DROP)) | 260 | IF_MAKEDEVS(APPLET(makedevs, _BB_DIR_SBIN, _BB_SUID_DROP)) |
@@ -417,6 +419,7 @@ IF_UNCOMPRESS(APPLET(uncompress, _BB_DIR_BIN, _BB_SUID_DROP)) | |||
417 | IF_UNEXPAND(APPLET_ODDNAME(unexpand, expand, _BB_DIR_USR_BIN, _BB_SUID_DROP, unexpand)) | 419 | IF_UNEXPAND(APPLET_ODDNAME(unexpand, expand, _BB_DIR_USR_BIN, _BB_SUID_DROP, unexpand)) |
418 | IF_UNIQ(APPLET(uniq, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 420 | IF_UNIQ(APPLET(uniq, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
419 | IF_UNIX2DOS(APPLET_ODDNAME(unix2dos, dos2unix, _BB_DIR_USR_BIN, _BB_SUID_DROP, unix2dos)) | 421 | IF_UNIX2DOS(APPLET_ODDNAME(unix2dos, dos2unix, _BB_DIR_USR_BIN, _BB_SUID_DROP, unix2dos)) |
422 | IF_UNXZ(APPLET(unxz, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | ||
420 | IF_UNLZMA(APPLET(unlzma, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 423 | IF_UNLZMA(APPLET(unlzma, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
421 | IF_LZOP(APPLET_ODDNAME(unlzop, lzop, _BB_DIR_USR_BIN, _BB_SUID_DROP, unlzop)) | 424 | IF_LZOP(APPLET_ODDNAME(unlzop, lzop, _BB_DIR_USR_BIN, _BB_SUID_DROP, unlzop)) |
422 | IF_UNZIP(APPLET(unzip, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 425 | IF_UNZIP(APPLET(unzip, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
@@ -437,6 +440,8 @@ IF_WHICH(APPLET(which, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | |||
437 | IF_WHO(APPLET(who, _BB_DIR_USR_BIN, _BB_SUID_DROP)) | 440 | IF_WHO(APPLET(who, _BB_DIR_USR_BIN, _BB_SUID_DROP)) |
438 | IF_WHOAMI(APPLET_NOFORK(whoami, whoami, _BB_DIR_USR_BIN, _BB_SUID_DROP, whoami)) | 441 | IF_WHOAMI(APPLET_NOFORK(whoami, whoami, _BB_DIR_USR_BIN, _BB_SUID_DROP, whoami)) |
439 | IF_XARGS(APPLET_NOEXEC(xargs, xargs, _BB_DIR_USR_BIN, _BB_SUID_DROP, xargs)) | 442 | IF_XARGS(APPLET_NOEXEC(xargs, xargs, _BB_DIR_USR_BIN, _BB_SUID_DROP, xargs)) |
443 | IF_UNXZ(APPLET_ODDNAME(xzcat, unxz, _BB_DIR_USR_BIN, _BB_SUID_DROP, xzcat)) | ||
444 | IF_XZ(APPLET_ODDNAME(xz, unxz, _BB_DIR_USR_BIN, _BB_SUID_DROP, xz)) | ||
440 | IF_YES(APPLET_NOFORK(yes, yes, _BB_DIR_USR_BIN, _BB_SUID_DROP, yes)) | 445 | IF_YES(APPLET_NOFORK(yes, yes, _BB_DIR_USR_BIN, _BB_SUID_DROP, yes)) |
441 | IF_GUNZIP(APPLET_ODDNAME(zcat, gunzip, _BB_DIR_BIN, _BB_SUID_DROP, zcat)) | 446 | IF_GUNZIP(APPLET_ODDNAME(zcat, gunzip, _BB_DIR_BIN, _BB_SUID_DROP, zcat)) |
442 | IF_ZCIP(APPLET(zcip, _BB_DIR_SBIN, _BB_SUID_DROP)) | 447 | IF_ZCIP(APPLET(zcip, _BB_DIR_SBIN, _BB_SUID_DROP)) |
diff --git a/include/libbb.h b/include/libbb.h index 58a5968c5..c9b3ffaa0 100644 --- a/include/libbb.h +++ b/include/libbb.h | |||
@@ -669,7 +669,7 @@ extern void *xmalloc_open_read_close(const char *filename, size_t *maxsz_p) FAST | |||
669 | || ENABLE_FEATURE_SEAMLESS_BZ2 \ | 669 | || ENABLE_FEATURE_SEAMLESS_BZ2 \ |
670 | || ENABLE_FEATURE_SEAMLESS_GZ \ | 670 | || ENABLE_FEATURE_SEAMLESS_GZ \ |
671 | /* || ENABLE_FEATURE_SEAMLESS_Z */ | 671 | /* || ENABLE_FEATURE_SEAMLESS_Z */ |
672 | extern int setup_unzip_on_fd(int fd /*, int fail_if_not_detected*/) FAST_FUNC; | 672 | extern void setup_unzip_on_fd(int fd /*, int fail_if_not_detected*/) FAST_FUNC; |
673 | #else | 673 | #else |
674 | # define setup_unzip_on_fd(...) ((void)0) | 674 | # define setup_unzip_on_fd(...) ((void)0) |
675 | #endif | 675 | #endif |
diff --git a/include/unarchive.h b/include/unarchive.h index a834816ba..aa7ecec55 100644 --- a/include/unarchive.h +++ b/include/unarchive.h | |||
@@ -143,19 +143,24 @@ typedef struct inflate_unzip_result { | |||
143 | } inflate_unzip_result; | 143 | } inflate_unzip_result; |
144 | 144 | ||
145 | IF_DESKTOP(long long) int inflate_unzip(inflate_unzip_result *res, off_t compr_size, int src_fd, int dst_fd) FAST_FUNC; | 145 | IF_DESKTOP(long long) int inflate_unzip(inflate_unzip_result *res, off_t compr_size, int src_fd, int dst_fd) FAST_FUNC; |
146 | /* xz unpacker takes .xz stream from offset 0 */ | ||
147 | IF_DESKTOP(long long) int unpack_xz_stream(int src_fd, int dst_fd) FAST_FUNC; | ||
146 | /* lzma unpacker takes .lzma stream from offset 0 */ | 148 | /* lzma unpacker takes .lzma stream from offset 0 */ |
147 | IF_DESKTOP(long long) int unpack_lzma_stream(int src_fd, int dst_fd) FAST_FUNC; | 149 | IF_DESKTOP(long long) int unpack_lzma_stream(int src_fd, int dst_fd) FAST_FUNC; |
148 | /* the rest wants 2 first bytes already skipped by the caller */ | 150 | /* the rest wants 2 first bytes already skipped by the caller */ |
149 | IF_DESKTOP(long long) int unpack_bz2_stream(int src_fd, int dst_fd) FAST_FUNC; | 151 | IF_DESKTOP(long long) int unpack_bz2_stream(int src_fd, int dst_fd) FAST_FUNC; |
150 | IF_DESKTOP(long long) int unpack_gz_stream(int src_fd, int dst_fd) FAST_FUNC; | 152 | IF_DESKTOP(long long) int unpack_gz_stream(int src_fd, int dst_fd) FAST_FUNC; |
151 | IF_DESKTOP(long long) int unpack_gz_stream_with_info(int src_fd, int dst_fd, unpack_info_t *info) FAST_FUNC; | 153 | IF_DESKTOP(long long) int unpack_gz_stream_with_info(int src_fd, int dst_fd, unpack_info_t *info) FAST_FUNC; |
152 | IF_DESKTOP(long long) int unpack_Z_stream(int fd_in, int fd_out) FAST_FUNC; | 154 | IF_DESKTOP(long long) int unpack_Z_stream(int src_fd, int dst_fd) FAST_FUNC; |
153 | /* wrapper which checks first two bytes to be "BZ" */ | 155 | /* wrapper which checks first two bytes to be "BZ" */ |
154 | IF_DESKTOP(long long) int unpack_bz2_stream_prime(int src_fd, int dst_fd) FAST_FUNC; | 156 | IF_DESKTOP(long long) int unpack_bz2_stream_prime(int src_fd, int dst_fd) FAST_FUNC; |
155 | 157 | ||
158 | char* append_ext(char *filename, const char *expected_ext) FAST_FUNC; | ||
156 | int bbunpack(char **argv, | 159 | int bbunpack(char **argv, |
157 | char* (*make_new_name)(char *filename), | 160 | IF_DESKTOP(long long) int FAST_FUNC (*unpacker)(unpack_info_t *info), |
158 | IF_DESKTOP(long long) int (*unpacker)(unpack_info_t *info)) FAST_FUNC; | 161 | char* FAST_FUNC (*make_new_name)(char *filename, const char *expected_ext), |
162 | const char *expected_ext | ||
163 | ) FAST_FUNC; | ||
159 | 164 | ||
160 | #if BB_MMU | 165 | #if BB_MMU |
161 | void open_transformer(int fd, | 166 | void open_transformer(int fd, |
diff --git a/include/unicode.h b/include/unicode.h index eaf67c833..e9e2bd14a 100644 --- a/include/unicode.h +++ b/include/unicode.h | |||
@@ -23,7 +23,8 @@ enum { | |||
23 | 23 | ||
24 | #if !ENABLE_UNICODE_SUPPORT | 24 | #if !ENABLE_UNICODE_SUPPORT |
25 | 25 | ||
26 | # define unicode_strlen(string) strlen(string) | 26 | # define unicode_strlen(string) strlen(string) |
27 | # define unicode_strwidth(string) strlen(string) | ||
27 | # define unicode_status UNICODE_OFF | 28 | # define unicode_status UNICODE_OFF |
28 | # define init_unicode() ((void)0) | 29 | # define init_unicode() ((void)0) |
29 | 30 | ||
@@ -49,7 +50,10 @@ enum { | |||
49 | # define ENABLE_UNICODE_BIDI_SUPPORT 0 | 50 | # define ENABLE_UNICODE_BIDI_SUPPORT 0 |
50 | # endif | 51 | # endif |
51 | 52 | ||
53 | /* Number of unicode chars. Falls back to strlen() on invalid unicode */ | ||
52 | size_t FAST_FUNC unicode_strlen(const char *string); | 54 | size_t FAST_FUNC unicode_strlen(const char *string); |
55 | /* Width on terminal */ | ||
56 | size_t FAST_FUNC unicode_strwidth(const char *string); | ||
53 | enum { | 57 | enum { |
54 | UNI_FLAG_PAD = (1 << 0), | 58 | UNI_FLAG_PAD = (1 << 0), |
55 | }; | 59 | }; |
diff --git a/include/usage.h b/include/usage.h index d53b86731..a9c4c4294 100644 --- a/include/usage.h +++ b/include/usage.h | |||
@@ -156,6 +156,11 @@ | |||
156 | "\n -r Repetitions" \ | 156 | "\n -r Repetitions" \ |
157 | "\n -n Start new tone" \ | 157 | "\n -n Start new tone" \ |
158 | 158 | ||
159 | #define blkid_trivial_usage \ | ||
160 | "" | ||
161 | #define blkid_full_usage "\n\n" \ | ||
162 | "Print UUIDs of all filesystems" | ||
163 | |||
159 | #define bootchartd_trivial_usage \ | 164 | #define bootchartd_trivial_usage \ |
160 | "start [PROG ARGS]|stop|init" | 165 | "start [PROG ARGS]|stop|init" |
161 | #define bootchartd_full_usage "\n\n" \ | 166 | #define bootchartd_full_usage "\n\n" \ |
@@ -189,23 +194,15 @@ | |||
189 | "\n stp BRIDGE [1/yes/on|0/no/off] STP on/off" \ | 194 | "\n stp BRIDGE [1/yes/on|0/no/off] STP on/off" \ |
190 | ) \ | 195 | ) \ |
191 | 196 | ||
192 | #define bunzip2_trivial_usage \ | ||
193 | "[OPTIONS] [FILE]..." | ||
194 | #define bunzip2_full_usage "\n\n" \ | ||
195 | "Uncompress FILEs (or stdin)\n" \ | ||
196 | "\nOptions:" \ | ||
197 | "\n -c Write to stdout" \ | ||
198 | "\n -f Force" \ | ||
199 | |||
200 | #define bzip2_trivial_usage \ | 197 | #define bzip2_trivial_usage \ |
201 | "[OPTIONS] [FILE]..." | 198 | "[OPTIONS] [FILE]..." |
202 | #define bzip2_full_usage "\n\n" \ | 199 | #define bzip2_full_usage "\n\n" \ |
203 | "Compress FILEs (or stdin) with bzip2 algorithm\n" \ | 200 | "Compress FILEs (or stdin) with bzip2 algorithm\n" \ |
204 | "\nOptions:" \ | 201 | "\nOptions:" \ |
205 | "\n -c Write to stdout" \ | 202 | "\n -1..9 Compression level" \ |
206 | "\n -d Decompress" \ | 203 | "\n -d Decompress" \ |
204 | "\n -c Write to stdout" \ | ||
207 | "\n -f Force" \ | 205 | "\n -f Force" \ |
208 | "\n -1..-9 Compression level" \ | ||
209 | 206 | ||
210 | #define busybox_notes_usage \ | 207 | #define busybox_notes_usage \ |
211 | "Hello world!\n" | 208 | "Hello world!\n" |
@@ -213,13 +210,14 @@ | |||
213 | #define lzop_trivial_usage \ | 210 | #define lzop_trivial_usage \ |
214 | "[-cfvd123456789CF] [FILE]..." | 211 | "[-cfvd123456789CF] [FILE]..." |
215 | #define lzop_full_usage "\n\n" \ | 212 | #define lzop_full_usage "\n\n" \ |
216 | " -c Write to stdout" \ | 213 | "Options:" \ |
214 | "\n -1..9 Compression level" \ | ||
215 | "\n -d Decompress" \ | ||
216 | "\n -c Write to stdout" \ | ||
217 | "\n -f Force" \ | 217 | "\n -f Force" \ |
218 | "\n -v Verbose" \ | 218 | "\n -v Verbose" \ |
219 | "\n -d Decompress" \ | ||
220 | "\n -F Don't store or verify checksum" \ | 219 | "\n -F Don't store or verify checksum" \ |
221 | "\n -C Also write checksum of compressed block" \ | 220 | "\n -C Also write checksum of compressed block" \ |
222 | "\n -1..9 Compression level" \ | ||
223 | 221 | ||
224 | #define lzopcat_trivial_usage \ | 222 | #define lzopcat_trivial_usage \ |
225 | "[-vCF] [FILE]..." | 223 | "[-vCF] [FILE]..." |
@@ -230,28 +228,68 @@ | |||
230 | #define unlzop_trivial_usage \ | 228 | #define unlzop_trivial_usage \ |
231 | "[-cfvCF] [FILE]..." | 229 | "[-cfvCF] [FILE]..." |
232 | #define unlzop_full_usage "\n\n" \ | 230 | #define unlzop_full_usage "\n\n" \ |
233 | " -c Write to stdout" \ | 231 | "Options:" \ |
232 | "\n -c Write to stdout" \ | ||
234 | "\n -f Force" \ | 233 | "\n -f Force" \ |
235 | "\n -v Verbose" \ | 234 | "\n -v Verbose" \ |
236 | "\n -F Don't store or verify checksum" \ | 235 | "\n -F Don't store or verify checksum" \ |
237 | 236 | ||
237 | #define bunzip2_trivial_usage \ | ||
238 | "[OPTIONS] [FILE]..." | ||
239 | #define bunzip2_full_usage "\n\n" \ | ||
240 | "Decompress FILEs (or stdin)\n" \ | ||
241 | "\nOptions:" \ | ||
242 | "\n -c Write to stdout" \ | ||
243 | "\n -f Force" \ | ||
244 | |||
238 | #define bzcat_trivial_usage \ | 245 | #define bzcat_trivial_usage \ |
239 | "FILE" | 246 | "FILE" |
240 | #define bzcat_full_usage "\n\n" \ | 247 | #define bzcat_full_usage "\n\n" \ |
241 | "Uncompress to stdout" | 248 | "Decompress to stdout" |
242 | 249 | ||
243 | #define unlzma_trivial_usage \ | 250 | #define unlzma_trivial_usage \ |
244 | "[OPTIONS] [FILE]..." | 251 | "[OPTIONS] [FILE]..." |
245 | #define unlzma_full_usage "\n\n" \ | 252 | #define unlzma_full_usage "\n\n" \ |
246 | "Uncompress FILE (or stdin)\n" \ | 253 | "Decompress FILE (or stdin)\n" \ |
254 | "\nOptions:" \ | ||
255 | "\n -c Write to stdout" \ | ||
256 | "\n -f Force" \ | ||
257 | |||
258 | #define lzma_trivial_usage \ | ||
259 | "-d [OPTIONS] [FILE]..." | ||
260 | #define lzma_full_usage "\n\n" \ | ||
261 | "Decompress FILE (or stdin)\n" \ | ||
262 | "\nOptions:" \ | ||
263 | "\n -d Decompress" \ | ||
264 | "\n -c Write to stdout" \ | ||
265 | "\n -f Force" \ | ||
266 | |||
267 | #define lzcat_trivial_usage \ | ||
268 | "FILE" | ||
269 | #define lzcat_full_usage "\n\n" \ | ||
270 | "Decompress to stdout" | ||
271 | |||
272 | #define unxz_trivial_usage \ | ||
273 | "[OPTIONS] [FILE]..." | ||
274 | #define unxz_full_usage "\n\n" \ | ||
275 | "Decompress FILE (or stdin)\n" \ | ||
247 | "\nOptions:" \ | 276 | "\nOptions:" \ |
248 | "\n -c Write to stdout" \ | 277 | "\n -c Write to stdout" \ |
249 | "\n -f Force" \ | 278 | "\n -f Force" \ |
250 | 279 | ||
251 | #define lzmacat_trivial_usage \ | 280 | #define xz_trivial_usage \ |
281 | "-d [OPTIONS] [FILE]..." | ||
282 | #define xz_full_usage "\n\n" \ | ||
283 | "Decompress FILE (or stdin)\n" \ | ||
284 | "\nOptions:" \ | ||
285 | "\n -d Decompress" \ | ||
286 | "\n -c Write to stdout" \ | ||
287 | "\n -f Force" \ | ||
288 | |||
289 | #define xzcat_trivial_usage \ | ||
252 | "FILE" | 290 | "FILE" |
253 | #define lzmacat_full_usage "\n\n" \ | 291 | #define xzcat_full_usage "\n\n" \ |
254 | "Uncompress to stdout" | 292 | "Decompress to stdout" |
255 | 293 | ||
256 | #define cal_trivial_usage \ | 294 | #define cal_trivial_usage \ |
257 | "[-jy] [[MONTH] YEAR]" | 295 | "[-jy] [[MONTH] YEAR]" |
@@ -262,11 +300,9 @@ | |||
262 | "\n -y Display the entire year" \ | 300 | "\n -y Display the entire year" \ |
263 | 301 | ||
264 | #define cat_trivial_usage \ | 302 | #define cat_trivial_usage \ |
265 | "[-u] [FILE]..." | 303 | "[FILE]..." |
266 | #define cat_full_usage "\n\n" \ | 304 | #define cat_full_usage "\n\n" \ |
267 | "Concatenate FILEs and print them to stdout\n" \ | 305 | "Concatenate FILEs and print them to stdout" \ |
268 | "\nOptions:" \ | ||
269 | "\n -u Use unbuffered i/o (ignored)" \ | ||
270 | 306 | ||
271 | #define cat_example_usage \ | 307 | #define cat_example_usage \ |
272 | "$ cat /proc/uptime\n" \ | 308 | "$ cat /proc/uptime\n" \ |
@@ -555,7 +591,7 @@ | |||
555 | "\n -L Follow all symlinks" \ | 591 | "\n -L Follow all symlinks" \ |
556 | "\n -H Follow symlinks on command line" \ | 592 | "\n -H Follow symlinks on command line" \ |
557 | "\n -p Preserve file attributes if possible" \ | 593 | "\n -p Preserve file attributes if possible" \ |
558 | "\n -f Force overwrite" \ | 594 | "\n -f Overwrite" \ |
559 | "\n -i Prompt before overwrite" \ | 595 | "\n -i Prompt before overwrite" \ |
560 | "\n -l,-s Create (sym)links" \ | 596 | "\n -l,-s Create (sym)links" \ |
561 | 597 | ||
@@ -1243,10 +1279,10 @@ | |||
1243 | "\n -H HEADS" \ | 1279 | "\n -H HEADS" \ |
1244 | "\n -S SECTORS" \ | 1280 | "\n -S SECTORS" \ |
1245 | 1281 | ||
1246 | #define blkid_trivial_usage \ | 1282 | #define fgconsole_trivial_usage \ |
1247 | "" | 1283 | "" |
1248 | #define blkid_full_usage "\n\n" \ | 1284 | #define fgconsole_full_usage "\n\n" \ |
1249 | "Print UUIDs of all filesystems" | 1285 | "Get active console" |
1250 | 1286 | ||
1251 | #define findfs_trivial_usage \ | 1287 | #define findfs_trivial_usage \ |
1252 | "LABEL=label or UUID=uuid" | 1288 | "LABEL=label or UUID=uuid" |
@@ -1605,7 +1641,7 @@ | |||
1605 | #define gunzip_trivial_usage \ | 1641 | #define gunzip_trivial_usage \ |
1606 | "[OPTIONS] [FILE]..." | 1642 | "[OPTIONS] [FILE]..." |
1607 | #define gunzip_full_usage "\n\n" \ | 1643 | #define gunzip_full_usage "\n\n" \ |
1608 | "Uncompress FILEs (or stdin)\n" \ | 1644 | "Decompress FILEs (or stdin)\n" \ |
1609 | "\nOptions:" \ | 1645 | "\nOptions:" \ |
1610 | "\n -c Write to stdout" \ | 1646 | "\n -c Write to stdout" \ |
1611 | "\n -f Force" \ | 1647 | "\n -f Force" \ |
@@ -1623,8 +1659,8 @@ | |||
1623 | #define gzip_full_usage "\n\n" \ | 1659 | #define gzip_full_usage "\n\n" \ |
1624 | "Compress FILEs (or stdin)\n" \ | 1660 | "Compress FILEs (or stdin)\n" \ |
1625 | "\nOptions:" \ | 1661 | "\nOptions:" \ |
1626 | "\n -c Write to stdout" \ | ||
1627 | "\n -d Decompress" \ | 1662 | "\n -d Decompress" \ |
1663 | "\n -c Write to stdout" \ | ||
1628 | "\n -f Force" \ | 1664 | "\n -f Force" \ |
1629 | 1665 | ||
1630 | #define gzip_example_usage \ | 1666 | #define gzip_example_usage \ |
@@ -4972,10 +5008,10 @@ | |||
4972 | #define uncompress_trivial_usage \ | 5008 | #define uncompress_trivial_usage \ |
4973 | "[-cf] [FILE]..." | 5009 | "[-cf] [FILE]..." |
4974 | #define uncompress_full_usage "\n\n" \ | 5010 | #define uncompress_full_usage "\n\n" \ |
4975 | "Uncompress .Z file[s]\n" \ | 5011 | "Decompress .Z file[s]\n" \ |
4976 | "\nOptions:" \ | 5012 | "\nOptions:" \ |
4977 | "\n -c Extract to stdout" \ | 5013 | "\n -c Write to stdout" \ |
4978 | "\n -f Overwrite an existing file" \ | 5014 | "\n -f Overwrite" \ |
4979 | 5015 | ||
4980 | #define unexpand_trivial_usage \ | 5016 | #define unexpand_trivial_usage \ |
4981 | "[-fa][-t N] [FILE]..." | 5017 | "[-fa][-t N] [FILE]..." |
@@ -5017,8 +5053,8 @@ | |||
5017 | "Extract files from ZIP archives\n" \ | 5053 | "Extract files from ZIP archives\n" \ |
5018 | "\nOptions:" \ | 5054 | "\nOptions:" \ |
5019 | "\n -l List archive contents (with -q for short form)" \ | 5055 | "\n -l List archive contents (with -q for short form)" \ |
5020 | "\n -n Never overwrite existing files (default)" \ | 5056 | "\n -n Never overwrite files (default)" \ |
5021 | "\n -o Overwrite files without prompting" \ | 5057 | "\n -o Overwrite" \ |
5022 | "\n -p Send output to stdout" \ | 5058 | "\n -p Send output to stdout" \ |
5023 | "\n -q Quiet" \ | 5059 | "\n -q Quiet" \ |
5024 | "\n -x XLST Exclude these files" \ | 5060 | "\n -x XLST Exclude these files" \ |
@@ -5219,7 +5255,7 @@ | |||
5219 | #define zcat_trivial_usage \ | 5255 | #define zcat_trivial_usage \ |
5220 | "FILE" | 5256 | "FILE" |
5221 | #define zcat_full_usage "\n\n" \ | 5257 | #define zcat_full_usage "\n\n" \ |
5222 | "Uncompress to stdout" | 5258 | "Decompress to stdout" |
5223 | 5259 | ||
5224 | #define zcip_trivial_usage \ | 5260 | #define zcip_trivial_usage \ |
5225 | "[OPTIONS] IFACE SCRIPT" | 5261 | "[OPTIONS] IFACE SCRIPT" |
diff --git a/libbb/lineedit.c b/libbb/lineedit.c index f7faf4639..ab418c0ab 100644 --- a/libbb/lineedit.c +++ b/libbb/lineedit.c | |||
@@ -1000,7 +1000,7 @@ static void showfiles(void) | |||
1000 | 1000 | ||
1001 | /* find the longest file name - use that as the column width */ | 1001 | /* find the longest file name - use that as the column width */ |
1002 | for (row = 0; row < nrows; row++) { | 1002 | for (row = 0; row < nrows; row++) { |
1003 | l = unicode_strlen(matches[row]); | 1003 | l = unicode_strwidth(matches[row]); |
1004 | if (column_width < l) | 1004 | if (column_width < l) |
1005 | column_width = l; | 1005 | column_width = l; |
1006 | } | 1006 | } |
@@ -1020,10 +1020,13 @@ static void showfiles(void) | |||
1020 | 1020 | ||
1021 | for (nc = 1; nc < ncols && n+nrows < nfiles; n += nrows, nc++) { | 1021 | for (nc = 1; nc < ncols && n+nrows < nfiles; n += nrows, nc++) { |
1022 | printf("%s%-*s", matches[n], | 1022 | printf("%s%-*s", matches[n], |
1023 | (int)(column_width - unicode_strlen(matches[n])), "" | 1023 | (int)(column_width - unicode_strwidth(matches[n])), "" |
1024 | ); | 1024 | ); |
1025 | } | 1025 | } |
1026 | puts(matches[n]); | 1026 | if (ENABLE_UNICODE_SUPPORT) |
1027 | puts(printable_string(NULL, matches[n])); | ||
1028 | else | ||
1029 | puts(matches[n]); | ||
1027 | } | 1030 | } |
1028 | } | 1031 | } |
1029 | 1032 | ||
diff --git a/libbb/read.c b/libbb/read.c index f3af144f0..cd6bbeb13 100644 --- a/libbb/read.c +++ b/libbb/read.c | |||
@@ -305,22 +305,26 @@ void* FAST_FUNC xmalloc_xopen_read_close(const char *filename, size_t *maxsz_p) | |||
305 | return buf; | 305 | return buf; |
306 | } | 306 | } |
307 | 307 | ||
308 | /* Used by e.g. rpm which gives us a fd without filename, | ||
309 | * thus we can't guess the format from filename's extension. | ||
310 | */ | ||
308 | #if ZIPPED | 311 | #if ZIPPED |
309 | int FAST_FUNC setup_unzip_on_fd(int fd /*, int fail_if_not_detected*/) | 312 | void FAST_FUNC setup_unzip_on_fd(int fd /*, int fail_if_not_detected*/) |
310 | { | 313 | { |
311 | const int fail_if_not_detected = 1; | 314 | const int fail_if_not_detected = 1; |
312 | unsigned char magic[2]; | 315 | unsigned char magic[8]; |
313 | #if BB_MMU | 316 | int offset = -2; |
317 | # if BB_MMU | ||
314 | IF_DESKTOP(long long) int FAST_FUNC (*xformer)(int src_fd, int dst_fd); | 318 | IF_DESKTOP(long long) int FAST_FUNC (*xformer)(int src_fd, int dst_fd); |
315 | enum { xformer_prog = 0 }; | 319 | enum { xformer_prog = 0 }; |
316 | #else | 320 | # else |
317 | enum { xformer = 0 }; | 321 | enum { xformer = 0 }; |
318 | const char *xformer_prog; | 322 | const char *xformer_prog; |
319 | #endif | 323 | # endif |
320 | 324 | ||
321 | /* .gz and .bz2 both have 2-byte signature, and their | 325 | /* .gz and .bz2 both have 2-byte signature, and their |
322 | * unpack_XXX_stream wants this header skipped. */ | 326 | * unpack_XXX_stream wants this header skipped. */ |
323 | xread(fd, &magic, 2); | 327 | xread(fd, magic, 2); |
324 | if (ENABLE_FEATURE_SEAMLESS_GZ | 328 | if (ENABLE_FEATURE_SEAMLESS_GZ |
325 | && magic[0] == 0x1f && magic[1] == 0x8b | 329 | && magic[0] == 0x1f && magic[1] == 0x8b |
326 | ) { | 330 | ) { |
@@ -341,28 +345,41 @@ int FAST_FUNC setup_unzip_on_fd(int fd /*, int fail_if_not_detected*/) | |||
341 | # endif | 345 | # endif |
342 | goto found_magic; | 346 | goto found_magic; |
343 | } | 347 | } |
344 | // TODO: xz format support. rpm adopted it, "rpm -i FILE.rpm" badly needs this. | 348 | if (ENABLE_FEATURE_SEAMLESS_XZ |
345 | // Signature: 0xFD, '7', 'z', 'X', 'Z', 0x00 | 349 | && magic[0] == 0xfd && magic[1] == '7' |
346 | // More info at: http://tukaani.org/xz/xz-file-format.txt | 350 | ) { |
351 | /* .xz signature: 0xfd, '7', 'z', 'X', 'Z', 0x00 */ | ||
352 | /* More info at: http://tukaani.org/xz/xz-file-format.txt */ | ||
353 | offset = -6; | ||
354 | xread(fd, magic + 2, 4); | ||
355 | if (strcmp((char*)magic + 2, "zXZ") == 0) { | ||
356 | # if BB_MMU | ||
357 | xformer = unpack_xz_stream; | ||
358 | # else | ||
359 | xformer_prog = "unxz"; | ||
360 | # endif | ||
361 | xlseek(fd, offset, SEEK_CUR); | ||
362 | goto found_magic; | ||
363 | } | ||
364 | } | ||
347 | 365 | ||
348 | /* No known magic seen */ | 366 | /* No known magic seen */ |
349 | if (fail_if_not_detected) | 367 | if (fail_if_not_detected) |
350 | bb_error_msg_and_die("no gzip" | 368 | bb_error_msg_and_die("no gzip" |
351 | IF_FEATURE_SEAMLESS_BZ2("/bzip2") | 369 | IF_FEATURE_SEAMLESS_BZ2("/bzip2") |
370 | IF_FEATURE_SEAMLESS_XZ("/xz") | ||
352 | " magic"); | 371 | " magic"); |
353 | xlseek(fd, -2, SEEK_CUR); | 372 | xlseek(fd, offset, SEEK_CUR); |
354 | return fd; | 373 | return; |
355 | 374 | ||
356 | found_magic: | 375 | found_magic: |
357 | # if !BB_MMU | 376 | # if !BB_MMU |
358 | /* NOMMU version of open_transformer execs | 377 | /* NOMMU version of open_transformer execs |
359 | * an external unzipper that wants | 378 | * an external unzipper that wants |
360 | * file position at the start of the file */ | 379 | * file position at the start of the file */ |
361 | xlseek(fd, -2, SEEK_CUR); | 380 | xlseek(fd, offset, SEEK_CUR); |
362 | # endif | 381 | # endif |
363 | open_transformer(fd, xformer, xformer_prog); | 382 | open_transformer(fd, xformer, xformer_prog); |
364 | |||
365 | return fd; | ||
366 | } | 383 | } |
367 | #endif /* ZIPPED */ | 384 | #endif /* ZIPPED */ |
368 | 385 | ||
@@ -380,12 +397,14 @@ int FAST_FUNC open_zipped(const char *fname) | |||
380 | 397 | ||
381 | sfx = strrchr(fname, '.'); | 398 | sfx = strrchr(fname, '.'); |
382 | if (sfx) { | 399 | if (sfx) { |
383 | if (ENABLE_FEATURE_SEAMLESS_LZMA && strcmp(sfx, ".lzma") == 0) | 400 | sfx++; |
401 | if (ENABLE_FEATURE_SEAMLESS_LZMA && strcmp(sfx, "lzma") == 0) | ||
384 | /* .lzma has no header/signature, just trust it */ | 402 | /* .lzma has no header/signature, just trust it */ |
385 | open_transformer(fd, unpack_lzma_stream, "unlzma"); | 403 | open_transformer(fd, unpack_lzma_stream, "unlzma"); |
386 | else | 404 | else |
387 | if ((ENABLE_FEATURE_SEAMLESS_GZ && strcmp(sfx, ".gz") == 0) | 405 | if ((ENABLE_FEATURE_SEAMLESS_GZ && strcmp(sfx, "gz") == 0) |
388 | || (ENABLE_FEATURE_SEAMLESS_BZ2 && strcmp(sfx, ".bz2") == 0) | 406 | || (ENABLE_FEATURE_SEAMLESS_BZ2 && strcmp(sfx, "bz2") == 0) |
407 | || (ENABLE_FEATURE_SEAMLESS_XZ && strcmp(sfx, "xz") == 0) | ||
389 | ) { | 408 | ) { |
390 | setup_unzip_on_fd(fd /*, fail_if_not_detected: 1*/); | 409 | setup_unzip_on_fd(fd /*, fail_if_not_detected: 1*/); |
391 | } | 410 | } |
diff --git a/libbb/unicode.c b/libbb/unicode.c index b2c28239b..d6fcf7a43 100644 --- a/libbb/unicode.c +++ b/libbb/unicode.c | |||
@@ -25,13 +25,15 @@ uint8_t unicode_status; | |||
25 | 25 | ||
26 | void FAST_FUNC init_unicode(void) | 26 | void FAST_FUNC init_unicode(void) |
27 | { | 27 | { |
28 | /* In unicode, this is a one character string */ | ||
29 | static const char unicode_0x394[] = { 0xce, 0x94, 0 }; | 28 | static const char unicode_0x394[] = { 0xce, 0x94, 0 }; |
29 | size_t width; | ||
30 | 30 | ||
31 | if (unicode_status != UNICODE_UNKNOWN) | 31 | if (unicode_status != UNICODE_UNKNOWN) |
32 | return; | 32 | return; |
33 | 33 | /* In unicode, this is a one character string */ | |
34 | unicode_status = unicode_strlen(unicode_0x394) == 1 ? UNICODE_ON : UNICODE_OFF; | 34 | // can use unicode_strlen(string) too, but otherwise unicode_strlen() is unused |
35 | width = mbstowcs(NULL, unicode_0x394, INT_MAX); | ||
36 | unicode_status = (width == 1 ? UNICODE_ON : UNICODE_OFF); | ||
35 | } | 37 | } |
36 | 38 | ||
37 | #else | 39 | #else |
@@ -954,6 +956,7 @@ int FAST_FUNC unicode_bidi_is_neutral_wchar(wint_t wc) | |||
954 | 956 | ||
955 | /* The rest is mostly same for libc and for "homegrown" support */ | 957 | /* The rest is mostly same for libc and for "homegrown" support */ |
956 | 958 | ||
959 | #if 0 // UNUSED | ||
957 | size_t FAST_FUNC unicode_strlen(const char *string) | 960 | size_t FAST_FUNC unicode_strlen(const char *string) |
958 | { | 961 | { |
959 | size_t width = mbstowcs(NULL, string, INT_MAX); | 962 | size_t width = mbstowcs(NULL, string, INT_MAX); |
@@ -961,6 +964,14 @@ size_t FAST_FUNC unicode_strlen(const char *string) | |||
961 | return strlen(string); | 964 | return strlen(string); |
962 | return width; | 965 | return width; |
963 | } | 966 | } |
967 | #endif | ||
968 | |||
969 | size_t FAST_FUNC unicode_strwidth(const char *string) | ||
970 | { | ||
971 | uni_stat_t uni_stat; | ||
972 | printable_string(&uni_stat, string); | ||
973 | return uni_stat.unicode_width; | ||
974 | } | ||
964 | 975 | ||
965 | static char* FAST_FUNC unicode_conv_to_printable2(uni_stat_t *stats, const char *src, unsigned width, int flags) | 976 | static char* FAST_FUNC unicode_conv_to_printable2(uni_stat_t *stats, const char *src, unsigned width, int flags) |
966 | { | 977 | { |
diff --git a/networking/interface.c b/networking/interface.c index a59f310a6..659ac36ea 100644 --- a/networking/interface.c +++ b/networking/interface.c | |||
@@ -1002,7 +1002,7 @@ static void ife_print(struct interface *ptr) | |||
1002 | if (hw == NULL) | 1002 | if (hw == NULL) |
1003 | hw = get_hwntype(-1); | 1003 | hw = get_hwntype(-1); |
1004 | 1004 | ||
1005 | printf("%-9.9s Link encap:%s ", ptr->name, hw->title); | 1005 | printf("%-9s Link encap:%s ", ptr->name, hw->title); |
1006 | /* For some hardware types (eg Ash, ATM) we don't print the | 1006 | /* For some hardware types (eg Ash, ATM) we don't print the |
1007 | hardware address if it's null. */ | 1007 | hardware address if it's null. */ |
1008 | if (hw->print != NULL | 1008 | if (hw->print != NULL |
@@ -1178,7 +1178,6 @@ static int for_all_interfaces(int (*doit) (struct interface *, void *), | |||
1178 | return -1; | 1178 | return -1; |
1179 | for (ife = int_list; ife; ife = ife->next) { | 1179 | for (ife = int_list; ife; ife = ife->next) { |
1180 | int err = doit(ife, cookie); | 1180 | int err = doit(ife, cookie); |
1181 | |||
1182 | if (err) | 1181 | if (err) |
1183 | return err; | 1182 | return err; |
1184 | } | 1183 | } |
diff --git a/networking/libiproute/ipaddress.c b/networking/libiproute/ipaddress.c index daea9dd03..a603053e1 100644 --- a/networking/libiproute/ipaddress.c +++ b/networking/libiproute/ipaddress.c | |||
@@ -54,6 +54,8 @@ static void print_link_flags(unsigned flags, unsigned mdown) | |||
54 | "MULTICAST\0""NOARP\0""UP\0""LOWER_UP\0"; | 54 | "MULTICAST\0""NOARP\0""UP\0""LOWER_UP\0"; |
55 | 55 | ||
56 | bb_putchar('<'); | 56 | bb_putchar('<'); |
57 | if (flags & IFF_UP && !(flags & IFF_RUNNING)) | ||
58 | printf("NO-CARRIER,"); | ||
57 | flags &= ~IFF_RUNNING; | 59 | flags &= ~IFF_RUNNING; |
58 | #if 0 | 60 | #if 0 |
59 | _PF(ALLMULTI); | 61 | _PF(ALLMULTI); |
@@ -162,6 +164,13 @@ static NOINLINE int print_linkinfo(const struct nlmsghdr *n) | |||
162 | printf("master %s ", ll_idx_n2a(*(int*)RTA_DATA(tb[IFLA_MASTER]), b1)); | 164 | printf("master %s ", ll_idx_n2a(*(int*)RTA_DATA(tb[IFLA_MASTER]), b1)); |
163 | } | 165 | } |
164 | #endif | 166 | #endif |
167 | if (tb[IFLA_OPERSTATE]) { | ||
168 | static const char operstate_labels[] ALIGN1 = | ||
169 | "UNKNOWN\0""NOTPRESENT\0""DOWN\0""LOWERLAYERDOWN\0" | ||
170 | "TESTING\0""DORMANT\0""UP\0"; | ||
171 | printf("state %s ", nth_string(operstate_labels, | ||
172 | *(uint8_t *)RTA_DATA(tb[IFLA_OPERSTATE]))); | ||
173 | } | ||
165 | if (G_filter.showqueue) | 174 | if (G_filter.showqueue) |
166 | print_queuelen((char*)RTA_DATA(tb[IFLA_IFNAME])); | 175 | print_queuelen((char*)RTA_DATA(tb[IFLA_IFNAME])); |
167 | 176 | ||
@@ -433,16 +442,14 @@ int ipaddr_list_or_flush(char **argv, int flush) | |||
433 | } | 442 | } |
434 | 443 | ||
435 | while (*argv) { | 444 | while (*argv) { |
436 | const int option_num = index_in_strings(option, *argv); | 445 | const smalluint key = index_in_strings(option, *argv); |
437 | switch (option_num) { | 446 | if (key == 0) { /* to */ |
438 | case 0: /* to */ | ||
439 | NEXT_ARG(); | 447 | NEXT_ARG(); |
440 | get_prefix(&G_filter.pfx, *argv, G_filter.family); | 448 | get_prefix(&G_filter.pfx, *argv, G_filter.family); |
441 | if (G_filter.family == AF_UNSPEC) { | 449 | if (G_filter.family == AF_UNSPEC) { |
442 | G_filter.family = G_filter.pfx.family; | 450 | G_filter.family = G_filter.pfx.family; |
443 | } | 451 | } |
444 | break; | 452 | } else if (key == 1) { /* scope */ |
445 | case 1: { /* scope */ | ||
446 | uint32_t scope = 0; | 453 | uint32_t scope = 0; |
447 | NEXT_ARG(); | 454 | NEXT_ARG(); |
448 | G_filter.scopemask = -1; | 455 | G_filter.scopemask = -1; |
@@ -454,22 +461,17 @@ int ipaddr_list_or_flush(char **argv, int flush) | |||
454 | G_filter.scopemask = 0; | 461 | G_filter.scopemask = 0; |
455 | } | 462 | } |
456 | G_filter.scope = scope; | 463 | G_filter.scope = scope; |
457 | break; | 464 | } else if (key == 2) { /* up */ |
458 | } | ||
459 | case 2: /* up */ | ||
460 | G_filter.up = 1; | 465 | G_filter.up = 1; |
461 | break; | 466 | } else if (key == 3) { /* label */ |
462 | case 3: /* label */ | ||
463 | NEXT_ARG(); | 467 | NEXT_ARG(); |
464 | G_filter.label = *argv; | 468 | G_filter.label = *argv; |
465 | break; | 469 | } else { |
466 | case 4: /* dev */ | 470 | if (key == 4) /* dev */ |
467 | NEXT_ARG(); | 471 | NEXT_ARG(); |
468 | default: | 472 | if (filter_dev) |
469 | if (filter_dev) { | 473 | duparg2("dev", *argv); |
470 | duparg2("dev", *argv); | 474 | filter_dev = *argv; |
471 | } | ||
472 | filter_dev = *argv; | ||
473 | } | 475 | } |
474 | argv++; | 476 | argv++; |
475 | } | 477 | } |
@@ -622,99 +624,85 @@ static int ipaddr_modify(int cmd, char **argv) | |||
622 | req.ifa.ifa_family = preferred_family; | 624 | req.ifa.ifa_family = preferred_family; |
623 | 625 | ||
624 | while (*argv) { | 626 | while (*argv) { |
625 | const int option_num = index_in_strings(option, *argv); | 627 | const smalluint arg = index_in_strings(option, *argv); |
626 | switch (option_num) { | 628 | if (arg <= 1) { /* peer, remote */ |
627 | case 0: /* peer */ | 629 | NEXT_ARG(); |
628 | case 1: /* remote */ | ||
629 | NEXT_ARG(); | ||
630 | 630 | ||
631 | if (peer_len) { | 631 | if (peer_len) { |
632 | duparg("peer", *argv); | 632 | duparg("peer", *argv); |
633 | } | ||
634 | get_prefix(&peer, *argv, req.ifa.ifa_family); | ||
635 | peer_len = peer.bytelen; | ||
636 | if (req.ifa.ifa_family == AF_UNSPEC) { | ||
637 | req.ifa.ifa_family = peer.family; | ||
638 | } | ||
639 | addattr_l(&req.n, sizeof(req), IFA_ADDRESS, &peer.data, peer.bytelen); | ||
640 | req.ifa.ifa_prefixlen = peer.bitlen; | ||
641 | break; | ||
642 | case 2: /* broadcast */ | ||
643 | case 3: /* brd */ | ||
644 | { | ||
645 | inet_prefix addr; | ||
646 | NEXT_ARG(); | ||
647 | if (brd_len) { | ||
648 | duparg("broadcast", *argv); | ||
649 | } | ||
650 | if (LONE_CHAR(*argv, '+')) { | ||
651 | brd_len = -1; | ||
652 | } else if (LONE_DASH(*argv)) { | ||
653 | brd_len = -2; | ||
654 | } else { | ||
655 | get_addr(&addr, *argv, req.ifa.ifa_family); | ||
656 | if (req.ifa.ifa_family == AF_UNSPEC) | ||
657 | req.ifa.ifa_family = addr.family; | ||
658 | addattr_l(&req.n, sizeof(req), IFA_BROADCAST, &addr.data, addr.bytelen); | ||
659 | brd_len = addr.bytelen; | ||
660 | } | ||
661 | break; | ||
662 | } | 633 | } |
663 | case 4: /* anycast */ | 634 | get_prefix(&peer, *argv, req.ifa.ifa_family); |
664 | { | 635 | peer_len = peer.bytelen; |
665 | inet_prefix addr; | 636 | if (req.ifa.ifa_family == AF_UNSPEC) { |
666 | NEXT_ARG(); | 637 | req.ifa.ifa_family = peer.family; |
667 | if (any_len) { | 638 | } |
668 | duparg("anycast", *argv); | 639 | addattr_l(&req.n, sizeof(req), IFA_ADDRESS, &peer.data, peer.bytelen); |
669 | } | 640 | req.ifa.ifa_prefixlen = peer.bitlen; |
641 | } else if (arg <= 3) { /* broadcast, brd */ | ||
642 | inet_prefix addr; | ||
643 | NEXT_ARG(); | ||
644 | if (brd_len) { | ||
645 | duparg("broadcast", *argv); | ||
646 | } | ||
647 | if (LONE_CHAR(*argv, '+')) { | ||
648 | brd_len = -1; | ||
649 | } else if (LONE_DASH(*argv)) { | ||
650 | brd_len = -2; | ||
651 | } else { | ||
670 | get_addr(&addr, *argv, req.ifa.ifa_family); | 652 | get_addr(&addr, *argv, req.ifa.ifa_family); |
671 | if (req.ifa.ifa_family == AF_UNSPEC) { | 653 | if (req.ifa.ifa_family == AF_UNSPEC) |
672 | req.ifa.ifa_family = addr.family; | 654 | req.ifa.ifa_family = addr.family; |
673 | } | 655 | addattr_l(&req.n, sizeof(req), IFA_BROADCAST, &addr.data, addr.bytelen); |
674 | addattr_l(&req.n, sizeof(req), IFA_ANYCAST, &addr.data, addr.bytelen); | 656 | brd_len = addr.bytelen; |
675 | any_len = addr.bytelen; | ||
676 | break; | ||
677 | } | 657 | } |
678 | case 5: /* scope */ | 658 | } else if (arg == 4) { /* anycast */ |
679 | { | 659 | inet_prefix addr; |
680 | uint32_t scope = 0; | 660 | NEXT_ARG(); |
681 | NEXT_ARG(); | 661 | if (any_len) { |
682 | if (rtnl_rtscope_a2n(&scope, *argv)) { | 662 | duparg("anycast", *argv); |
683 | invarg(*argv, "scope"); | ||
684 | } | ||
685 | req.ifa.ifa_scope = scope; | ||
686 | scoped = 1; | ||
687 | break; | ||
688 | } | 663 | } |
689 | case 6: /* dev */ | 664 | get_addr(&addr, *argv, req.ifa.ifa_family); |
690 | NEXT_ARG(); | 665 | if (req.ifa.ifa_family == AF_UNSPEC) { |
691 | d = *argv; | 666 | req.ifa.ifa_family = addr.family; |
692 | break; | 667 | } |
693 | case 7: /* label */ | 668 | addattr_l(&req.n, sizeof(req), IFA_ANYCAST, &addr.data, addr.bytelen); |
694 | NEXT_ARG(); | 669 | any_len = addr.bytelen; |
695 | l = *argv; | 670 | } else if (arg == 5) { /* scope */ |
696 | addattr_l(&req.n, sizeof(req), IFA_LABEL, l, strlen(l)+1); | 671 | uint32_t scope = 0; |
697 | break; | 672 | NEXT_ARG(); |
698 | case 8: /* local */ | 673 | if (rtnl_rtscope_a2n(&scope, *argv)) { |
674 | invarg(*argv, "scope"); | ||
675 | } | ||
676 | req.ifa.ifa_scope = scope; | ||
677 | scoped = 1; | ||
678 | } else if (arg == 6) { /* dev */ | ||
679 | NEXT_ARG(); | ||
680 | d = *argv; | ||
681 | } else if (arg == 7) { /* label */ | ||
682 | NEXT_ARG(); | ||
683 | l = *argv; | ||
684 | addattr_l(&req.n, sizeof(req), IFA_LABEL, l, strlen(l)+1); | ||
685 | } else { | ||
686 | if (arg == 8) /* local */ | ||
699 | NEXT_ARG(); | 687 | NEXT_ARG(); |
700 | default: | 688 | if (local_len) { |
701 | if (local_len) { | 689 | duparg2("local", *argv); |
702 | duparg2("local", *argv); | 690 | } |
703 | } | 691 | get_prefix(&lcl, *argv, req.ifa.ifa_family); |
704 | get_prefix(&lcl, *argv, req.ifa.ifa_family); | 692 | if (req.ifa.ifa_family == AF_UNSPEC) { |
705 | if (req.ifa.ifa_family == AF_UNSPEC) { | 693 | req.ifa.ifa_family = lcl.family; |
706 | req.ifa.ifa_family = lcl.family; | 694 | } |
707 | } | 695 | addattr_l(&req.n, sizeof(req), IFA_LOCAL, &lcl.data, lcl.bytelen); |
708 | addattr_l(&req.n, sizeof(req), IFA_LOCAL, &lcl.data, lcl.bytelen); | 696 | local_len = lcl.bytelen; |
709 | local_len = lcl.bytelen; | ||
710 | } | 697 | } |
711 | argv++; | 698 | argv++; |
712 | } | 699 | } |
713 | 700 | ||
714 | if (d == NULL) { | 701 | // d cannot be null here, NEXT_ARG() of "dev" ensures that |
715 | bb_error_msg(bb_msg_requires_arg, "\"dev\""); | 702 | //if (d == NULL) { |
716 | return -1; | 703 | // bb_error_msg(bb_msg_requires_arg, "\"dev\""); |
717 | } | 704 | // return -1; |
705 | //} | ||
718 | if (l && strncmp(d, l, strlen(d)) != 0) { | 706 | if (l && strncmp(d, l, strlen(d)) != 0) { |
719 | bb_error_msg_and_die("\"dev\" (%s) must match \"label\" (%s)", d, l); | 707 | bb_error_msg_and_die("\"dev\" (%s) must match \"label\" (%s)", d, l); |
720 | } | 708 | } |
@@ -764,21 +752,15 @@ int do_ipaddr(char **argv) | |||
764 | { | 752 | { |
765 | static const char commands[] ALIGN1 = | 753 | static const char commands[] ALIGN1 = |
766 | "add\0""delete\0""list\0""show\0""lst\0""flush\0"; | 754 | "add\0""delete\0""list\0""show\0""lst\0""flush\0"; |
767 | 755 | smalluint cmd = 2; | |
768 | int command_num = 2; /* default command is list */ | ||
769 | |||
770 | if (*argv) { | 756 | if (*argv) { |
771 | command_num = index_in_substrings(commands, *argv); | 757 | cmd = index_in_substrings(commands, *argv); |
772 | if (command_num < 0 || command_num > 5) | 758 | if (cmd > 5) |
773 | bb_error_msg_and_die("unknown command %s", *argv); | 759 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); |
774 | argv++; | 760 | argv++; |
761 | if (cmd <= 1) | ||
762 | return ipaddr_modify((cmd == 0) ? RTM_NEWADDR : RTM_DELADDR, argv); | ||
775 | } | 763 | } |
776 | if (command_num == 0) /* add */ | ||
777 | return ipaddr_modify(RTM_NEWADDR, argv); | ||
778 | if (command_num == 1) /* delete */ | ||
779 | return ipaddr_modify(RTM_DELADDR, argv); | ||
780 | if (command_num == 5) /* flush */ | ||
781 | return ipaddr_list_or_flush(argv, 1); | ||
782 | /* 2 == list, 3 == show, 4 == lst */ | 764 | /* 2 == list, 3 == show, 4 == lst */ |
783 | return ipaddr_list_or_flush(argv, 0); | 765 | return ipaddr_list_or_flush(argv, cmd == 5); |
784 | } | 766 | } |
diff --git a/networking/libiproute/iplink.c b/networking/libiproute/iplink.c index df8a354a1..8bf892797 100644 --- a/networking/libiproute/iplink.c +++ b/networking/libiproute/iplink.c | |||
@@ -171,9 +171,9 @@ static int do_set(char **argv) | |||
171 | char *newname = NULL; | 171 | char *newname = NULL; |
172 | int htype, halen; | 172 | int htype, halen; |
173 | static const char keywords[] ALIGN1 = | 173 | static const char keywords[] ALIGN1 = |
174 | "up\0""down\0""name\0""mtu\0""multicast\0" | 174 | "up\0""down\0""name\0""mtu\0""qlen\0""multicast\0" |
175 | "arp\0""address\0""dev\0"; | 175 | "arp\0""address\0""dev\0"; |
176 | enum { ARG_up = 0, ARG_down, ARG_name, ARG_mtu, ARG_multicast, | 176 | enum { ARG_up = 0, ARG_down, ARG_name, ARG_mtu, ARG_qlen, ARG_multicast, |
177 | ARG_arp, ARG_addr, ARG_dev }; | 177 | ARG_arp, ARG_addr, ARG_dev }; |
178 | static const char str_on_off[] ALIGN1 = "on\0""off\0"; | 178 | static const char str_on_off[] ALIGN1 = "on\0""off\0"; |
179 | enum { PARM_on = 0, PARM_off }; | 179 | enum { PARM_on = 0, PARM_off }; |
@@ -186,56 +186,53 @@ static int do_set(char **argv) | |||
186 | if (key == ARG_up) { | 186 | if (key == ARG_up) { |
187 | mask |= IFF_UP; | 187 | mask |= IFF_UP; |
188 | flags |= IFF_UP; | 188 | flags |= IFF_UP; |
189 | } | 189 | } else if (key == ARG_down) { |
190 | if (key == ARG_down) { | ||
191 | mask |= IFF_UP; | 190 | mask |= IFF_UP; |
192 | flags &= ~IFF_UP; | 191 | flags &= ~IFF_UP; |
193 | } | 192 | } else if (key == ARG_name) { |
194 | if (key == ARG_name) { | ||
195 | NEXT_ARG(); | 193 | NEXT_ARG(); |
196 | newname = *argv; | 194 | newname = *argv; |
197 | } | 195 | } else if (key == ARG_mtu) { |
198 | if (key == ARG_mtu) { | ||
199 | NEXT_ARG(); | 196 | NEXT_ARG(); |
200 | if (mtu != -1) | 197 | if (mtu != -1) |
201 | duparg("mtu", *argv); | 198 | duparg("mtu", *argv); |
202 | mtu = get_unsigned(*argv, "mtu"); | 199 | mtu = get_unsigned(*argv, "mtu"); |
203 | } | 200 | } else if (key == ARG_qlen) { |
204 | if (key == ARG_multicast) { | ||
205 | int param; | ||
206 | NEXT_ARG(); | 201 | NEXT_ARG(); |
207 | mask |= IFF_MULTICAST; | 202 | if (qlen != -1) |
208 | param = index_in_strings(str_on_off, *argv); | 203 | duparg("qlen", *argv); |
209 | if (param < 0) | 204 | qlen = get_unsigned(*argv, "qlen"); |
210 | die_must_be_on_off("multicast"); | 205 | } else if (key == ARG_addr) { |
211 | if (param == PARM_on) | ||
212 | flags |= IFF_MULTICAST; | ||
213 | else | ||
214 | flags &= ~IFF_MULTICAST; | ||
215 | } | ||
216 | if (key == ARG_arp) { | ||
217 | int param; | ||
218 | NEXT_ARG(); | ||
219 | mask |= IFF_NOARP; | ||
220 | param = index_in_strings(str_on_off, *argv); | ||
221 | if (param < 0) | ||
222 | die_must_be_on_off("arp"); | ||
223 | if (param == PARM_on) | ||
224 | flags &= ~IFF_NOARP; | ||
225 | else | ||
226 | flags |= IFF_NOARP; | ||
227 | } | ||
228 | if (key == ARG_addr) { | ||
229 | NEXT_ARG(); | 206 | NEXT_ARG(); |
230 | newaddr = *argv; | 207 | newaddr = *argv; |
231 | } | 208 | } else if (key >= ARG_dev) { |
232 | if (key >= ARG_dev) { | ||
233 | if (key == ARG_dev) { | 209 | if (key == ARG_dev) { |
234 | NEXT_ARG(); | 210 | NEXT_ARG(); |
235 | } | 211 | } |
236 | if (dev) | 212 | if (dev) |
237 | duparg2("dev", *argv); | 213 | duparg2("dev", *argv); |
238 | dev = *argv; | 214 | dev = *argv; |
215 | } else { | ||
216 | int param; | ||
217 | NEXT_ARG(); | ||
218 | param = index_in_strings(str_on_off, *argv); | ||
219 | if (key == ARG_multicast) { | ||
220 | if (param < 0) | ||
221 | die_must_be_on_off("multicast"); | ||
222 | mask |= IFF_MULTICAST; | ||
223 | if (param == PARM_on) | ||
224 | flags |= IFF_MULTICAST; | ||
225 | else | ||
226 | flags &= ~IFF_MULTICAST; | ||
227 | } else if (key == ARG_arp) { | ||
228 | if (param < 0) | ||
229 | die_must_be_on_off("arp"); | ||
230 | mask |= IFF_NOARP; | ||
231 | if (param == PARM_on) | ||
232 | flags &= ~IFF_NOARP; | ||
233 | else | ||
234 | flags |= IFF_NOARP; | ||
235 | } | ||
239 | } | 236 | } |
240 | argv++; | 237 | argv++; |
241 | } | 238 | } |
@@ -248,9 +245,11 @@ static int do_set(char **argv) | |||
248 | halen = get_address(dev, &htype); | 245 | halen = get_address(dev, &htype); |
249 | if (newaddr) { | 246 | if (newaddr) { |
250 | parse_address(dev, htype, halen, newaddr, &ifr0); | 247 | parse_address(dev, htype, halen, newaddr, &ifr0); |
248 | set_address(&ifr0, 0); | ||
251 | } | 249 | } |
252 | if (newbrd) { | 250 | if (newbrd) { |
253 | parse_address(dev, htype, halen, newbrd, &ifr1); | 251 | parse_address(dev, htype, halen, newbrd, &ifr1); |
252 | set_address(&ifr1, 1); | ||
254 | } | 253 | } |
255 | } | 254 | } |
256 | 255 | ||
@@ -264,14 +263,6 @@ static int do_set(char **argv) | |||
264 | if (mtu != -1) { | 263 | if (mtu != -1) { |
265 | set_mtu(dev, mtu); | 264 | set_mtu(dev, mtu); |
266 | } | 265 | } |
267 | if (newaddr || newbrd) { | ||
268 | if (newbrd) { | ||
269 | set_address(&ifr1, 1); | ||
270 | } | ||
271 | if (newaddr) { | ||
272 | set_address(&ifr0, 0); | ||
273 | } | ||
274 | } | ||
275 | if (mask) | 266 | if (mask) |
276 | do_chflags(dev, flags, mask); | 267 | do_chflags(dev, flags, mask); |
277 | return 0; | 268 | return 0; |
@@ -283,20 +274,108 @@ static int ipaddr_list_link(char **argv) | |||
283 | return ipaddr_list_or_flush(argv, 0); | 274 | return ipaddr_list_or_flush(argv, 0); |
284 | } | 275 | } |
285 | 276 | ||
277 | #ifndef NLMSG_TAIL | ||
278 | #define NLMSG_TAIL(nmsg) \ | ||
279 | ((struct rtattr *) (((void *) (nmsg)) + NLMSG_ALIGN((nmsg)->nlmsg_len))) | ||
280 | #endif | ||
281 | /* Return value becomes exitcode. It's okay to not return at all */ | ||
282 | static int do_change(char **argv, const unsigned rtm) | ||
283 | { | ||
284 | static const char keywords[] ALIGN1 = | ||
285 | "link\0""name\0""type\0""dev\0"; | ||
286 | enum { | ||
287 | ARG_link, | ||
288 | ARG_name, | ||
289 | ARG_type, | ||
290 | ARG_dev, | ||
291 | }; | ||
292 | struct rtnl_handle rth; | ||
293 | struct { | ||
294 | struct nlmsghdr n; | ||
295 | struct ifinfomsg i; | ||
296 | char buf[1024]; | ||
297 | } req; | ||
298 | smalluint arg; | ||
299 | char *name_str = NULL, *link_str = NULL, *type_str = NULL, *dev_str = NULL; | ||
300 | |||
301 | memset(&req, 0, sizeof(req)); | ||
302 | |||
303 | req.n.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg)); | ||
304 | req.n.nlmsg_flags = NLM_F_REQUEST; | ||
305 | req.n.nlmsg_type = rtm; | ||
306 | req.i.ifi_family = preferred_family; | ||
307 | if (rtm == RTM_NEWLINK) | ||
308 | req.n.nlmsg_flags |= NLM_F_CREATE|NLM_F_EXCL; | ||
309 | |||
310 | while (*argv) { | ||
311 | arg = index_in_substrings(keywords, *argv); | ||
312 | if (arg == ARG_link) { | ||
313 | NEXT_ARG(); | ||
314 | link_str = *argv; | ||
315 | } else if (arg == ARG_name) { | ||
316 | NEXT_ARG(); | ||
317 | name_str = *argv; | ||
318 | } else if (arg == ARG_type) { | ||
319 | NEXT_ARG(); | ||
320 | type_str = *argv; | ||
321 | } else { | ||
322 | if (arg == ARG_dev) { | ||
323 | if (dev_str) | ||
324 | duparg(*argv, "dev"); | ||
325 | NEXT_ARG(); | ||
326 | } | ||
327 | dev_str = *argv; | ||
328 | } | ||
329 | argv++; | ||
330 | } | ||
331 | xrtnl_open(&rth); | ||
332 | ll_init_map(&rth); | ||
333 | if (type_str) { | ||
334 | struct rtattr *linkinfo = NLMSG_TAIL(&req.n); | ||
335 | |||
336 | addattr_l(&req.n, sizeof(req), IFLA_LINKINFO, NULL, 0); | ||
337 | addattr_l(&req.n, sizeof(req), IFLA_INFO_KIND, type_str, | ||
338 | strlen(type_str)); | ||
339 | linkinfo->rta_len = (void *)NLMSG_TAIL(&req.n) - (void *)linkinfo; | ||
340 | } | ||
341 | if (rtm != RTM_NEWLINK) { | ||
342 | if (!dev_str) | ||
343 | return 1; /* Need a device to delete */ | ||
344 | req.i.ifi_index = xll_name_to_index(dev_str); | ||
345 | } else { | ||
346 | if (!name_str) | ||
347 | name_str = dev_str; | ||
348 | if (link_str) { | ||
349 | int idx = xll_name_to_index(link_str); | ||
350 | addattr_l(&req.n, sizeof(req), IFLA_LINK, &idx, 4); | ||
351 | } | ||
352 | } | ||
353 | if (name_str) { | ||
354 | const size_t name_len = strlen(name_str) + 1; | ||
355 | if (name_len < 2 || name_len > IFNAMSIZ) | ||
356 | invarg(name_str, "name"); | ||
357 | addattr_l(&req.n, sizeof(req), IFLA_IFNAME, name_str, name_len); | ||
358 | } | ||
359 | if (rtnl_talk(&rth, &req.n, 0, 0, NULL, NULL, NULL) < 0) | ||
360 | return 2; | ||
361 | return 0; | ||
362 | } | ||
363 | |||
286 | /* Return value becomes exitcode. It's okay to not return at all */ | 364 | /* Return value becomes exitcode. It's okay to not return at all */ |
287 | int do_iplink(char **argv) | 365 | int do_iplink(char **argv) |
288 | { | 366 | { |
289 | static const char keywords[] ALIGN1 = | 367 | static const char keywords[] ALIGN1 = |
290 | "set\0""show\0""lst\0""list\0"; | 368 | "add\0""delete\0""set\0""show\0""lst\0""list\0"; |
291 | int key; | 369 | if (*argv) { |
292 | if (!*argv) | 370 | smalluint key = index_in_substrings(keywords, *argv); |
293 | return ipaddr_list_link(argv); | 371 | if (key > 5) /* invalid argument */ |
294 | key = index_in_substrings(keywords, *argv); | 372 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); |
295 | if (key < 0) | 373 | argv++; |
296 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); | 374 | if (key <= 1) /* add/delete */ |
297 | argv++; | 375 | return do_change(argv, key ? RTM_DELLINK : RTM_NEWLINK); |
298 | if (key == 0) /* set */ | 376 | else if (key == 2) /* set */ |
299 | return do_set(argv); | 377 | return do_set(argv); |
378 | } | ||
300 | /* show, lst, list */ | 379 | /* show, lst, list */ |
301 | return ipaddr_list_link(argv); | 380 | return ipaddr_list_link(argv); |
302 | } | 381 | } |
diff --git a/networking/libiproute/iprule.c b/networking/libiproute/iprule.c index 535019d87..835529ec4 100644 --- a/networking/libiproute/iprule.c +++ b/networking/libiproute/iprule.c | |||
@@ -69,11 +69,9 @@ static int FAST_FUNC print_rule(const struct sockaddr_nl *who UNUSED_PARAM, | |||
69 | else if (r->rtm_family == AF_IPX) | 69 | else if (r->rtm_family == AF_IPX) |
70 | host_len = 80; | 70 | host_len = 80; |
71 | */ | 71 | */ |
72 | if (tb[RTA_PRIORITY]) | 72 | printf("%u:\t", tb[RTA_PRIORITY] ? |
73 | printf("%u:\t", *(unsigned*)RTA_DATA(tb[RTA_PRIORITY])); | 73 | *(unsigned*)RTA_DATA(tb[RTA_PRIORITY]) |
74 | else | 74 | : 0); |
75 | printf("0:\t"); | ||
76 | |||
77 | printf("from "); | 75 | printf("from "); |
78 | if (tb[RTA_SRC]) { | 76 | if (tb[RTA_SRC]) { |
79 | if (r->rtm_src_len != host_len) { | 77 | if (r->rtm_src_len != host_len) { |
@@ -310,25 +308,13 @@ int do_iprule(char **argv) | |||
310 | { | 308 | { |
311 | static const char ip_rule_commands[] ALIGN1 = | 309 | static const char ip_rule_commands[] ALIGN1 = |
312 | "add\0""delete\0""list\0""show\0"; | 310 | "add\0""delete\0""list\0""show\0"; |
313 | int cmd = 2; /* list */ | 311 | if (*argv) { |
314 | 312 | smalluint cmd = index_in_substrings(ip_rule_commands, *argv); | |
315 | if (!*argv) | 313 | if (cmd > 3) |
316 | return iprule_list(argv); | 314 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); |
317 | 315 | argv++; | |
318 | cmd = index_in_substrings(ip_rule_commands, *argv); | 316 | if (cmd < 2) |
319 | switch (cmd) { | 317 | return iprule_modify((cmd == 0) ? RTM_NEWRULE : RTM_DELRULE, argv); |
320 | case 0: /* add */ | ||
321 | cmd = RTM_NEWRULE; | ||
322 | break; | ||
323 | case 1: /* delete */ | ||
324 | cmd = RTM_DELRULE; | ||
325 | break; | ||
326 | case 2: /* list */ | ||
327 | case 3: /* show */ | ||
328 | return iprule_list(argv+1); | ||
329 | break; | ||
330 | default: | ||
331 | bb_error_msg_and_die("unknown command %s", *argv); | ||
332 | } | 318 | } |
333 | return iprule_modify(cmd, argv+1); | 319 | return iprule_list(argv); |
334 | } | 320 | } |
diff --git a/networking/libiproute/iptunnel.c b/networking/libiproute/iptunnel.c index c36c7136f..257343826 100644 --- a/networking/libiproute/iptunnel.c +++ b/networking/libiproute/iptunnel.c | |||
@@ -561,11 +561,10 @@ int do_iptunnel(char **argv) | |||
561 | static const char keywords[] ALIGN1 = | 561 | static const char keywords[] ALIGN1 = |
562 | "add\0""change\0""delete\0""show\0""list\0""lst\0"; | 562 | "add\0""change\0""delete\0""show\0""list\0""lst\0"; |
563 | enum { ARG_add = 0, ARG_change, ARG_del, ARG_show, ARG_list, ARG_lst }; | 563 | enum { ARG_add = 0, ARG_change, ARG_del, ARG_show, ARG_list, ARG_lst }; |
564 | int key; | ||
565 | 564 | ||
566 | if (*argv) { | 565 | if (*argv) { |
567 | key = index_in_substrings(keywords, *argv); | 566 | smalluint key = index_in_substrings(keywords, *argv); |
568 | if (key < 0) | 567 | if (key > 5) |
569 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); | 568 | bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name); |
570 | argv++; | 569 | argv++; |
571 | if (key == ARG_add) | 570 | if (key == ARG_add) |
diff --git a/networking/udhcp/common.c b/networking/udhcp/common.c index 7cda34a18..90a07ed09 100644 --- a/networking/udhcp/common.c +++ b/networking/udhcp/common.c | |||
@@ -176,7 +176,7 @@ unsigned FAST_FUNC udhcp_option_idx(const char *name) | |||
176 | } | 176 | } |
177 | } | 177 | } |
178 | 178 | ||
179 | /* get an option with bounds checking (warning, result is not aligned). */ | 179 | /* Get an option with bounds checking (warning, result is not aligned) */ |
180 | uint8_t* FAST_FUNC udhcp_get_option(struct dhcp_packet *packet, int code) | 180 | uint8_t* FAST_FUNC udhcp_get_option(struct dhcp_packet *packet, int code) |
181 | { | 181 | { |
182 | uint8_t *optionptr; | 182 | uint8_t *optionptr; |
@@ -240,7 +240,7 @@ uint8_t* FAST_FUNC udhcp_get_option(struct dhcp_packet *packet, int code) | |||
240 | return NULL; | 240 | return NULL; |
241 | } | 241 | } |
242 | 242 | ||
243 | /* return the position of the 'end' option (no bounds checking) */ | 243 | /* Return the position of the 'end' option (no bounds checking) */ |
244 | int FAST_FUNC udhcp_end_option(uint8_t *optionptr) | 244 | int FAST_FUNC udhcp_end_option(uint8_t *optionptr) |
245 | { | 245 | { |
246 | int i = 0; | 246 | int i = 0; |
diff --git a/networking/udhcp/common.h b/networking/udhcp/common.h index c9dd0bb25..ce81d1807 100644 --- a/networking/udhcp/common.h +++ b/networking/udhcp/common.h | |||
@@ -57,10 +57,20 @@ struct ip_udp_dhcp_packet { | |||
57 | struct dhcp_packet data; | 57 | struct dhcp_packet data; |
58 | } PACKED; | 58 | } PACKED; |
59 | 59 | ||
60 | struct udp_dhcp_packet { | ||
61 | struct udphdr udp; | ||
62 | struct dhcp_packet data; | ||
63 | } PACKED; | ||
64 | |||
65 | enum { | ||
66 | IP_UPD_DHCP_SIZE = sizeof(struct ip_udp_dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS, | ||
67 | UPD_DHCP_SIZE = sizeof(struct udp_dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS, | ||
68 | DHCP_SIZE = sizeof(struct dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS, | ||
69 | }; | ||
70 | |||
60 | /* Let's see whether compiler understood us right */ | 71 | /* Let's see whether compiler understood us right */ |
61 | struct BUG_bad_sizeof_struct_ip_udp_dhcp_packet { | 72 | struct BUG_bad_sizeof_struct_ip_udp_dhcp_packet { |
62 | char BUG_bad_sizeof_struct_ip_udp_dhcp_packet | 73 | char c[IP_UPD_DHCP_SIZE == 576 ? 1 : -1]; |
63 | [(sizeof(struct ip_udp_dhcp_packet) != 576 + CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS) ? -1 : 1]; | ||
64 | }; | 74 | }; |
65 | 75 | ||
66 | 76 | ||
@@ -270,13 +280,13 @@ void udhcp_init_header(struct dhcp_packet *packet, char type) FAST_FUNC; | |||
270 | int udhcp_recv_kernel_packet(struct dhcp_packet *packet, int fd) FAST_FUNC; | 280 | int udhcp_recv_kernel_packet(struct dhcp_packet *packet, int fd) FAST_FUNC; |
271 | 281 | ||
272 | int udhcp_send_raw_packet(struct dhcp_packet *dhcp_pkt, | 282 | int udhcp_send_raw_packet(struct dhcp_packet *dhcp_pkt, |
273 | uint32_t source_ip, int source_port, | 283 | uint32_t source_nip, int source_port, |
274 | uint32_t dest_ip, int dest_port, const uint8_t *dest_arp, | 284 | uint32_t dest_nip, int dest_port, const uint8_t *dest_arp, |
275 | int ifindex) FAST_FUNC; | 285 | int ifindex) FAST_FUNC; |
276 | 286 | ||
277 | int udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, | 287 | int udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, |
278 | uint32_t source_ip, int source_port, | 288 | uint32_t source_nip, int source_port, |
279 | uint32_t dest_ip, int dest_port) FAST_FUNC; | 289 | uint32_t dest_nip, int dest_port) FAST_FUNC; |
280 | 290 | ||
281 | void udhcp_sp_setup(void) FAST_FUNC; | 291 | void udhcp_sp_setup(void) FAST_FUNC; |
282 | int udhcp_sp_fd_set(fd_set *rfds, int extra_fd) FAST_FUNC; | 292 | int udhcp_sp_fd_set(fd_set *rfds, int extra_fd) FAST_FUNC; |
diff --git a/networking/udhcp/dhcpc.c b/networking/udhcp/dhcpc.c index 24ff82d2f..c2b21c695 100644 --- a/networking/udhcp/dhcpc.c +++ b/networking/udhcp/dhcpc.c | |||
@@ -229,37 +229,41 @@ static NOINLINE char *xmalloc_optname_optval(uint8_t *option, const struct dhcp_ | |||
229 | /* put all the parameters into the environment */ | 229 | /* put all the parameters into the environment */ |
230 | static char **fill_envp(struct dhcp_packet *packet) | 230 | static char **fill_envp(struct dhcp_packet *packet) |
231 | { | 231 | { |
232 | int num_options = 0; | 232 | int envc; |
233 | int i; | 233 | int i; |
234 | char **envp, **curr; | 234 | char **envp, **curr; |
235 | const char *opt_name; | 235 | const char *opt_name; |
236 | uint8_t *temp; | 236 | uint8_t *temp; |
237 | uint8_t over = 0; | 237 | uint8_t overload = 0; |
238 | 238 | ||
239 | /* We need 6 elements for: | ||
240 | * "interface=IFACE" | ||
241 | * "ip=N.N.N.N" from packet->yiaddr | ||
242 | * "siaddr=IP" from packet->siaddr_nip (unless 0) | ||
243 | * "boot_file=FILE" from packet->file (unless overloaded) | ||
244 | * "sname=SERVER_HOSTNAME" from packet->sname (unless overloaded) | ||
245 | * terminating NULL | ||
246 | */ | ||
247 | envc = 6; | ||
248 | /* +1 element for each option, +2 for subnet option: */ | ||
239 | if (packet) { | 249 | if (packet) { |
240 | for (i = 0; dhcp_optflags[i].code; i++) { | 250 | for (i = 0; dhcp_optflags[i].code; i++) { |
241 | if (udhcp_get_option(packet, dhcp_optflags[i].code)) { | 251 | if (udhcp_get_option(packet, dhcp_optflags[i].code)) { |
242 | num_options++; | ||
243 | if (dhcp_optflags[i].code == DHCP_SUBNET) | 252 | if (dhcp_optflags[i].code == DHCP_SUBNET) |
244 | num_options++; /* for mton */ | 253 | envc++; /* for mton */ |
254 | envc++; | ||
245 | } | 255 | } |
246 | } | 256 | } |
247 | if (packet->siaddr_nip) | ||
248 | num_options++; | ||
249 | temp = udhcp_get_option(packet, DHCP_OPTION_OVERLOAD); | 257 | temp = udhcp_get_option(packet, DHCP_OPTION_OVERLOAD); |
250 | if (temp) | 258 | if (temp) |
251 | over = *temp; | 259 | overload = *temp; |
252 | if (!(over & FILE_FIELD) && packet->file[0]) | ||
253 | num_options++; | ||
254 | if (!(over & SNAME_FIELD) && packet->sname[0]) | ||
255 | num_options++; | ||
256 | } | 260 | } |
261 | curr = envp = xzalloc(sizeof(char *) * envc); | ||
257 | 262 | ||
258 | curr = envp = xzalloc(sizeof(char *) * (num_options + 3)); | ||
259 | *curr = xasprintf("interface=%s", client_config.interface); | 263 | *curr = xasprintf("interface=%s", client_config.interface); |
260 | putenv(*curr++); | 264 | putenv(*curr++); |
261 | 265 | ||
262 | if (packet == NULL) | 266 | if (!packet) |
263 | return envp; | 267 | return envp; |
264 | 268 | ||
265 | *curr = xmalloc(sizeof("ip=255.255.255.255")); | 269 | *curr = xmalloc(sizeof("ip=255.255.255.255")); |
@@ -274,9 +278,8 @@ static char **fill_envp(struct dhcp_packet *packet) | |||
274 | goto next; | 278 | goto next; |
275 | *curr = xmalloc_optname_optval(temp, &dhcp_optflags[i], opt_name); | 279 | *curr = xmalloc_optname_optval(temp, &dhcp_optflags[i], opt_name); |
276 | putenv(*curr++); | 280 | putenv(*curr++); |
277 | |||
278 | /* Fill in a subnet bits option for things like /24 */ | ||
279 | if (dhcp_optflags[i].code == DHCP_SUBNET) { | 281 | if (dhcp_optflags[i].code == DHCP_SUBNET) { |
282 | /* Subnet option: make things like "$ip/$mask" possible */ | ||
280 | uint32_t subnet; | 283 | uint32_t subnet; |
281 | move_from_unaligned32(subnet, temp); | 284 | move_from_unaligned32(subnet, temp); |
282 | *curr = xasprintf("mask=%d", mton(subnet)); | 285 | *curr = xasprintf("mask=%d", mton(subnet)); |
@@ -291,12 +294,12 @@ static char **fill_envp(struct dhcp_packet *packet) | |||
291 | sprint_nip(*curr, "siaddr=", (uint8_t *) &packet->siaddr_nip); | 294 | sprint_nip(*curr, "siaddr=", (uint8_t *) &packet->siaddr_nip); |
292 | putenv(*curr++); | 295 | putenv(*curr++); |
293 | } | 296 | } |
294 | if (!(over & FILE_FIELD) && packet->file[0]) { | 297 | if (!(overload & FILE_FIELD) && packet->file[0]) { |
295 | /* watch out for invalid packets */ | 298 | /* watch out for invalid packets */ |
296 | *curr = xasprintf("boot_file=%."DHCP_PKT_FILE_LEN_STR"s", packet->file); | 299 | *curr = xasprintf("boot_file=%."DHCP_PKT_FILE_LEN_STR"s", packet->file); |
297 | putenv(*curr++); | 300 | putenv(*curr++); |
298 | } | 301 | } |
299 | if (!(over & SNAME_FIELD) && packet->sname[0]) { | 302 | if (!(overload & SNAME_FIELD) && packet->sname[0]) { |
300 | /* watch out for invalid packets */ | 303 | /* watch out for invalid packets */ |
301 | *curr = xasprintf("sname=%."DHCP_PKT_SNAME_LEN_STR"s", packet->sname); | 304 | *curr = xasprintf("sname=%."DHCP_PKT_SNAME_LEN_STR"s", packet->sname); |
302 | putenv(*curr++); | 305 | putenv(*curr++); |
diff --git a/networking/udhcp/packet.c b/networking/udhcp/packet.c index 1bfe12041..4badc9cbc 100644 --- a/networking/udhcp/packet.c +++ b/networking/udhcp/packet.c | |||
@@ -167,69 +167,72 @@ uint16_t FAST_FUNC udhcp_checksum(void *addr, int count) | |||
167 | 167 | ||
168 | /* Construct a ip/udp header for a packet, send packet */ | 168 | /* Construct a ip/udp header for a packet, send packet */ |
169 | int FAST_FUNC udhcp_send_raw_packet(struct dhcp_packet *dhcp_pkt, | 169 | int FAST_FUNC udhcp_send_raw_packet(struct dhcp_packet *dhcp_pkt, |
170 | uint32_t source_ip, int source_port, | 170 | uint32_t source_nip, int source_port, |
171 | uint32_t dest_ip, int dest_port, const uint8_t *dest_arp, | 171 | uint32_t dest_nip, int dest_port, const uint8_t *dest_arp, |
172 | int ifindex) | 172 | int ifindex) |
173 | { | 173 | { |
174 | struct sockaddr_ll dest; | 174 | struct sockaddr_ll dest_sll; |
175 | struct ip_udp_dhcp_packet packet; | 175 | struct ip_udp_dhcp_packet packet; |
176 | unsigned padding; | ||
176 | int fd; | 177 | int fd; |
177 | int result = -1; | 178 | int result = -1; |
178 | const char *msg; | 179 | const char *msg; |
179 | 180 | ||
180 | enum { | ||
181 | IP_UPD_DHCP_SIZE = sizeof(struct ip_udp_dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS, | ||
182 | UPD_DHCP_SIZE = IP_UPD_DHCP_SIZE - offsetof(struct ip_udp_dhcp_packet, udp), | ||
183 | }; | ||
184 | |||
185 | fd = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_IP)); | 181 | fd = socket(PF_PACKET, SOCK_DGRAM, htons(ETH_P_IP)); |
186 | if (fd < 0) { | 182 | if (fd < 0) { |
187 | msg = "socket(%s)"; | 183 | msg = "socket(%s)"; |
188 | goto ret_msg; | 184 | goto ret_msg; |
189 | } | 185 | } |
190 | 186 | ||
191 | memset(&dest, 0, sizeof(dest)); | 187 | memset(&dest_sll, 0, sizeof(dest_sll)); |
192 | memset(&packet, 0, sizeof(packet)); | 188 | memset(&packet, 0, offsetof(struct ip_udp_dhcp_packet, data)); |
193 | packet.data = *dhcp_pkt; /* struct copy */ | 189 | packet.data = *dhcp_pkt; /* struct copy */ |
194 | 190 | ||
195 | dest.sll_family = AF_PACKET; | 191 | dest_sll.sll_family = AF_PACKET; |
196 | dest.sll_protocol = htons(ETH_P_IP); | 192 | dest_sll.sll_protocol = htons(ETH_P_IP); |
197 | dest.sll_ifindex = ifindex; | 193 | dest_sll.sll_ifindex = ifindex; |
198 | dest.sll_halen = 6; | 194 | dest_sll.sll_halen = 6; |
199 | memcpy(dest.sll_addr, dest_arp, 6); | 195 | memcpy(dest_sll.sll_addr, dest_arp, 6); |
200 | if (bind(fd, (struct sockaddr *)&dest, sizeof(dest)) < 0) { | 196 | |
197 | if (bind(fd, (struct sockaddr *)&dest_sll, sizeof(dest_sll)) < 0) { | ||
201 | msg = "bind(%s)"; | 198 | msg = "bind(%s)"; |
202 | goto ret_close; | 199 | goto ret_close; |
203 | } | 200 | } |
204 | 201 | ||
202 | /* We were sending full-sized DHCP packets (zero padded), | ||
203 | * but some badly configured servers were seen dropping them. | ||
204 | * Apparently they drop all DHCP packets >576 *ethernet* octets big, | ||
205 | * whereas they may only drop packets >576 *IP* octets big | ||
206 | * (which for typical Ethernet II means 590 octets: 6+6+2 + 576). | ||
207 | * | ||
208 | * In order to work with those buggy servers, | ||
209 | * we truncate packets after end option byte. | ||
210 | */ | ||
211 | padding = DHCP_OPTIONS_BUFSIZE - 1 - udhcp_end_option(packet.data.options); | ||
212 | |||
205 | packet.ip.protocol = IPPROTO_UDP; | 213 | packet.ip.protocol = IPPROTO_UDP; |
206 | packet.ip.saddr = source_ip; | 214 | packet.ip.saddr = source_nip; |
207 | packet.ip.daddr = dest_ip; | 215 | packet.ip.daddr = dest_nip; |
208 | packet.udp.source = htons(source_port); | 216 | packet.udp.source = htons(source_port); |
209 | packet.udp.dest = htons(dest_port); | 217 | packet.udp.dest = htons(dest_port); |
210 | /* size, excluding IP header: */ | 218 | /* size, excluding IP header: */ |
211 | packet.udp.len = htons(UPD_DHCP_SIZE); | 219 | packet.udp.len = htons(UPD_DHCP_SIZE - padding); |
212 | /* for UDP checksumming, ip.len is set to UDP packet len */ | 220 | /* for UDP checksumming, ip.len is set to UDP packet len */ |
213 | packet.ip.tot_len = packet.udp.len; | 221 | packet.ip.tot_len = packet.udp.len; |
214 | packet.udp.check = udhcp_checksum(&packet, IP_UPD_DHCP_SIZE); | 222 | packet.udp.check = udhcp_checksum(&packet, IP_UPD_DHCP_SIZE - padding); |
215 | /* but for sending, it is set to IP packet len */ | 223 | /* but for sending, it is set to IP packet len */ |
216 | packet.ip.tot_len = htons(IP_UPD_DHCP_SIZE); | 224 | packet.ip.tot_len = htons(IP_UPD_DHCP_SIZE - padding); |
217 | packet.ip.ihl = sizeof(packet.ip) >> 2; | 225 | packet.ip.ihl = sizeof(packet.ip) >> 2; |
218 | packet.ip.version = IPVERSION; | 226 | packet.ip.version = IPVERSION; |
219 | packet.ip.ttl = IPDEFTTL; | 227 | packet.ip.ttl = IPDEFTTL; |
220 | packet.ip.check = udhcp_checksum(&packet.ip, sizeof(packet.ip)); | 228 | packet.ip.check = udhcp_checksum(&packet.ip, sizeof(packet.ip)); |
221 | 229 | ||
222 | /* Currently we send full-sized DHCP packets (zero padded). | ||
223 | * If you need to change this: last byte of the packet is | ||
224 | * packet.data.options[udhcp_end_option(packet.data.options)] | ||
225 | */ | ||
226 | udhcp_dump_packet(dhcp_pkt); | 230 | udhcp_dump_packet(dhcp_pkt); |
227 | result = sendto(fd, &packet, IP_UPD_DHCP_SIZE, 0, | 231 | result = sendto(fd, &packet, IP_UPD_DHCP_SIZE - padding, /*flags:*/ 0, |
228 | (struct sockaddr *) &dest, sizeof(dest)); | 232 | (struct sockaddr *) &dest_sll, sizeof(dest_sll)); |
229 | msg = "sendto"; | 233 | msg = "sendto"; |
230 | ret_close: | 234 | ret_close: |
231 | close(fd); | 235 | close(fd); |
232 | /* FIXME: and if result >= 0 but != IP_UPD_DHCP_SIZE? */ | ||
233 | if (result < 0) { | 236 | if (result < 0) { |
234 | ret_msg: | 237 | ret_msg: |
235 | bb_perror_msg(msg, "PACKET"); | 238 | bb_perror_msg(msg, "PACKET"); |
@@ -239,18 +242,15 @@ int FAST_FUNC udhcp_send_raw_packet(struct dhcp_packet *dhcp_pkt, | |||
239 | 242 | ||
240 | /* Let the kernel do all the work for packet generation */ | 243 | /* Let the kernel do all the work for packet generation */ |
241 | int FAST_FUNC udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, | 244 | int FAST_FUNC udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, |
242 | uint32_t source_ip, int source_port, | 245 | uint32_t source_nip, int source_port, |
243 | uint32_t dest_ip, int dest_port) | 246 | uint32_t dest_nip, int dest_port) |
244 | { | 247 | { |
245 | struct sockaddr_in client; | 248 | struct sockaddr_in client; |
249 | unsigned padding; | ||
246 | int fd; | 250 | int fd; |
247 | int result = -1; | 251 | int result = -1; |
248 | const char *msg; | 252 | const char *msg; |
249 | 253 | ||
250 | enum { | ||
251 | DHCP_SIZE = sizeof(struct dhcp_packet) - CONFIG_UDHCPC_SLACK_FOR_BUGGY_SERVERS, | ||
252 | }; | ||
253 | |||
254 | fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP); | 254 | fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP); |
255 | if (fd < 0) { | 255 | if (fd < 0) { |
256 | msg = "socket(%s)"; | 256 | msg = "socket(%s)"; |
@@ -261,7 +261,7 @@ int FAST_FUNC udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, | |||
261 | memset(&client, 0, sizeof(client)); | 261 | memset(&client, 0, sizeof(client)); |
262 | client.sin_family = AF_INET; | 262 | client.sin_family = AF_INET; |
263 | client.sin_port = htons(source_port); | 263 | client.sin_port = htons(source_port); |
264 | client.sin_addr.s_addr = source_ip; | 264 | client.sin_addr.s_addr = source_nip; |
265 | if (bind(fd, (struct sockaddr *)&client, sizeof(client)) == -1) { | 265 | if (bind(fd, (struct sockaddr *)&client, sizeof(client)) == -1) { |
266 | msg = "bind(%s)"; | 266 | msg = "bind(%s)"; |
267 | goto ret_close; | 267 | goto ret_close; |
@@ -270,19 +270,19 @@ int FAST_FUNC udhcp_send_kernel_packet(struct dhcp_packet *dhcp_pkt, | |||
270 | memset(&client, 0, sizeof(client)); | 270 | memset(&client, 0, sizeof(client)); |
271 | client.sin_family = AF_INET; | 271 | client.sin_family = AF_INET; |
272 | client.sin_port = htons(dest_port); | 272 | client.sin_port = htons(dest_port); |
273 | client.sin_addr.s_addr = dest_ip; | 273 | client.sin_addr.s_addr = dest_nip; |
274 | if (connect(fd, (struct sockaddr *)&client, sizeof(client)) == -1) { | 274 | if (connect(fd, (struct sockaddr *)&client, sizeof(client)) == -1) { |
275 | msg = "connect"; | 275 | msg = "connect"; |
276 | goto ret_close; | 276 | goto ret_close; |
277 | } | 277 | } |
278 | 278 | ||
279 | /* Currently we send full-sized DHCP packets (see above) */ | ||
280 | udhcp_dump_packet(dhcp_pkt); | 279 | udhcp_dump_packet(dhcp_pkt); |
281 | result = safe_write(fd, dhcp_pkt, DHCP_SIZE); | 280 | |
281 | padding = DHCP_OPTIONS_BUFSIZE - 1 - udhcp_end_option(dhcp_pkt->options); | ||
282 | result = safe_write(fd, dhcp_pkt, DHCP_SIZE - padding); | ||
282 | msg = "write"; | 283 | msg = "write"; |
283 | ret_close: | 284 | ret_close: |
284 | close(fd); | 285 | close(fd); |
285 | /* FIXME: and if result >= 0 but != DHCP_SIZE? */ | ||
286 | if (result < 0) { | 286 | if (result < 0) { |
287 | ret_msg: | 287 | ret_msg: |
288 | bb_perror_msg(msg, "UDP"); | 288 | bb_perror_msg(msg, "UDP"); |
diff --git a/scripts/defconfig b/scripts/defconfig index 0a748febc..896571bd9 100644 --- a/scripts/defconfig +++ b/scripts/defconfig | |||
@@ -287,6 +287,7 @@ CONFIG_FEATURE_MD5_SHA1_SUM_CHECK=y | |||
287 | # Console Utilities | 287 | # Console Utilities |
288 | # | 288 | # |
289 | CONFIG_CHVT=y | 289 | CONFIG_CHVT=y |
290 | CONFIG_FGCONSOLE=y | ||
290 | CONFIG_CLEAR=y | 291 | CONFIG_CLEAR=y |
291 | CONFIG_DEALLOCVT=y | 292 | CONFIG_DEALLOCVT=y |
292 | CONFIG_DUMPKMAP=y | 293 | CONFIG_DUMPKMAP=y |
diff --git a/scripts/gen_build_files.sh b/scripts/gen_build_files.sh index d2db907f3..b3aa132a3 100755 --- a/scripts/gen_build_files.sh +++ b/scripts/gen_build_files.sh | |||
@@ -8,10 +8,11 @@ cd -- "$2" || { echo "Syntax: $0 SRCTREE OBJTREE"; exit 1; } | |||
8 | srctree="$1" | 8 | srctree="$1" |
9 | 9 | ||
10 | find -type d | while read -r d; do | 10 | find -type d | while read -r d; do |
11 | d="${d#./}" | ||
11 | src="$srctree/$d/Kbuild.src" | 12 | src="$srctree/$d/Kbuild.src" |
12 | dst="$d/Kbuild" | 13 | dst="$d/Kbuild" |
13 | if test -f "$src"; then | 14 | if test -f "$src"; then |
14 | echo " CHK $dst" | 15 | #echo " CHK $dst" |
15 | 16 | ||
16 | s=`sed -n 's@^//kbuild:@@p' -- "$srctree/$d"/*.c` | 17 | s=`sed -n 's@^//kbuild:@@p' -- "$srctree/$d"/*.c` |
17 | echo "# DO NOT EDIT. This file is generated from Kbuild.src" >"$dst.$$.tmp" | 18 | echo "# DO NOT EDIT. This file is generated from Kbuild.src" >"$dst.$$.tmp" |
@@ -36,7 +37,7 @@ find -type d | while read -r d; do | |||
36 | src="$srctree/$d/Config.src" | 37 | src="$srctree/$d/Config.src" |
37 | dst="$d/Config.in" | 38 | dst="$d/Config.in" |
38 | if test -f "$src"; then | 39 | if test -f "$src"; then |
39 | echo " CHK $dst" | 40 | #echo " CHK $dst" |
40 | 41 | ||
41 | s=`sed -n 's@^//config:@@p' -- "$srctree/$d"/*.c` | 42 | s=`sed -n 's@^//config:@@p' -- "$srctree/$d"/*.c` |
42 | echo "# DO NOT EDIT. This file is generated from Config.src" >"$dst.$$.tmp" | 43 | echo "# DO NOT EDIT. This file is generated from Config.src" >"$dst.$$.tmp" |
diff --git a/shell/ash.c b/shell/ash.c index d6838f82c..13072cd04 100644 --- a/shell/ash.c +++ b/shell/ash.c | |||
@@ -328,6 +328,14 @@ static int isdigit_str9(const char *str) | |||
328 | return (*str == '\0'); | 328 | return (*str == '\0'); |
329 | } | 329 | } |
330 | 330 | ||
331 | static const char *var_end(const char *var) | ||
332 | { | ||
333 | while (*var) | ||
334 | if (*var++ == '=') | ||
335 | break; | ||
336 | return var; | ||
337 | } | ||
338 | |||
331 | 339 | ||
332 | /* ============ Interrupts / exceptions */ | 340 | /* ============ Interrupts / exceptions */ |
333 | /* | 341 | /* |
@@ -1761,8 +1769,8 @@ static void FAST_FUNC getoptsreset(const char *value); | |||
1761 | struct var { | 1769 | struct var { |
1762 | struct var *next; /* next entry in hash list */ | 1770 | struct var *next; /* next entry in hash list */ |
1763 | int flags; /* flags are defined above */ | 1771 | int flags; /* flags are defined above */ |
1764 | const char *text; /* name=value */ | 1772 | const char *var_text; /* name=value */ |
1765 | void (*func)(const char *) FAST_FUNC; /* function to be called when */ | 1773 | void (*var_func)(const char *) FAST_FUNC; /* function to be called when */ |
1766 | /* the variable gets set/unset */ | 1774 | /* the variable gets set/unset */ |
1767 | }; | 1775 | }; |
1768 | 1776 | ||
@@ -1816,13 +1824,13 @@ static void change_random(const char *) FAST_FUNC; | |||
1816 | 1824 | ||
1817 | static const struct { | 1825 | static const struct { |
1818 | int flags; | 1826 | int flags; |
1819 | const char *text; | 1827 | const char *var_text; |
1820 | void (*func)(const char *) FAST_FUNC; | 1828 | void (*var_func)(const char *) FAST_FUNC; |
1821 | } varinit_data[] = { | 1829 | } varinit_data[] = { |
1822 | { VSTRFIXED|VTEXTFIXED , defifsvar , NULL }, | 1830 | { VSTRFIXED|VTEXTFIXED , defifsvar , NULL }, |
1823 | #if ENABLE_ASH_MAIL | 1831 | #if ENABLE_ASH_MAIL |
1824 | { VSTRFIXED|VTEXTFIXED|VUNSET, "MAIL\0" , changemail }, | 1832 | { VSTRFIXED|VTEXTFIXED|VUNSET, "MAIL" , changemail }, |
1825 | { VSTRFIXED|VTEXTFIXED|VUNSET, "MAILPATH\0", changemail }, | 1833 | { VSTRFIXED|VTEXTFIXED|VUNSET, "MAILPATH" , changemail }, |
1826 | #endif | 1834 | #endif |
1827 | { VSTRFIXED|VTEXTFIXED , bb_PATH_root_path, changepath }, | 1835 | { VSTRFIXED|VTEXTFIXED , bb_PATH_root_path, changepath }, |
1828 | { VSTRFIXED|VTEXTFIXED , "PS1=$ " , NULL }, | 1836 | { VSTRFIXED|VTEXTFIXED , "PS1=$ " , NULL }, |
@@ -1832,14 +1840,14 @@ static const struct { | |||
1832 | { VSTRFIXED|VTEXTFIXED , "OPTIND=1" , getoptsreset }, | 1840 | { VSTRFIXED|VTEXTFIXED , "OPTIND=1" , getoptsreset }, |
1833 | #endif | 1841 | #endif |
1834 | #if ENABLE_ASH_RANDOM_SUPPORT | 1842 | #if ENABLE_ASH_RANDOM_SUPPORT |
1835 | { VSTRFIXED|VTEXTFIXED|VUNSET|VDYNAMIC, "RANDOM\0", change_random }, | 1843 | { VSTRFIXED|VTEXTFIXED|VUNSET|VDYNAMIC, "RANDOM", change_random }, |
1836 | #endif | 1844 | #endif |
1837 | #if ENABLE_LOCALE_SUPPORT | 1845 | #if ENABLE_LOCALE_SUPPORT |
1838 | { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_ALL\0" , change_lc_all }, | 1846 | { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_ALL" , change_lc_all }, |
1839 | { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_CTYPE\0", change_lc_ctype }, | 1847 | { VSTRFIXED|VTEXTFIXED|VUNSET, "LC_CTYPE" , change_lc_ctype }, |
1840 | #endif | 1848 | #endif |
1841 | #if ENABLE_FEATURE_EDITING_SAVEHISTORY | 1849 | #if ENABLE_FEATURE_EDITING_SAVEHISTORY |
1842 | { VSTRFIXED|VTEXTFIXED|VUNSET, "HISTFILE\0", NULL }, | 1850 | { VSTRFIXED|VTEXTFIXED|VUNSET, "HISTFILE" , NULL }, |
1843 | #endif | 1851 | #endif |
1844 | }; | 1852 | }; |
1845 | 1853 | ||
@@ -1866,9 +1874,9 @@ extern struct globals_var *const ash_ptr_to_globals_var; | |||
1866 | (*(struct globals_var**)&ash_ptr_to_globals_var) = xzalloc(sizeof(G_var)); \ | 1874 | (*(struct globals_var**)&ash_ptr_to_globals_var) = xzalloc(sizeof(G_var)); \ |
1867 | barrier(); \ | 1875 | barrier(); \ |
1868 | for (i = 0; i < ARRAY_SIZE(varinit_data); i++) { \ | 1876 | for (i = 0; i < ARRAY_SIZE(varinit_data); i++) { \ |
1869 | varinit[i].flags = varinit_data[i].flags; \ | 1877 | varinit[i].flags = varinit_data[i].flags; \ |
1870 | varinit[i].text = varinit_data[i].text; \ | 1878 | varinit[i].var_text = varinit_data[i].var_text; \ |
1871 | varinit[i].func = varinit_data[i].func; \ | 1879 | varinit[i].var_func = varinit_data[i].var_func; \ |
1872 | } \ | 1880 | } \ |
1873 | } while (0) | 1881 | } while (0) |
1874 | 1882 | ||
@@ -1899,19 +1907,19 @@ extern struct globals_var *const ash_ptr_to_globals_var; | |||
1899 | * They have to skip over the name. They return the null string | 1907 | * They have to skip over the name. They return the null string |
1900 | * for unset variables. | 1908 | * for unset variables. |
1901 | */ | 1909 | */ |
1902 | #define ifsval() (vifs.text + 4) | 1910 | #define ifsval() (vifs.var_text + 4) |
1903 | #define ifsset() ((vifs.flags & VUNSET) == 0) | 1911 | #define ifsset() ((vifs.flags & VUNSET) == 0) |
1904 | #if ENABLE_ASH_MAIL | 1912 | #if ENABLE_ASH_MAIL |
1905 | # define mailval() (vmail.text + 5) | 1913 | # define mailval() (vmail.var_text + 5) |
1906 | # define mpathval() (vmpath.text + 9) | 1914 | # define mpathval() (vmpath.var_text + 9) |
1907 | # define mpathset() ((vmpath.flags & VUNSET) == 0) | 1915 | # define mpathset() ((vmpath.flags & VUNSET) == 0) |
1908 | #endif | 1916 | #endif |
1909 | #define pathval() (vpath.text + 5) | 1917 | #define pathval() (vpath.var_text + 5) |
1910 | #define ps1val() (vps1.text + 4) | 1918 | #define ps1val() (vps1.var_text + 4) |
1911 | #define ps2val() (vps2.text + 4) | 1919 | #define ps2val() (vps2.var_text + 4) |
1912 | #define ps4val() (vps4.text + 4) | 1920 | #define ps4val() (vps4.var_text + 4) |
1913 | #if ENABLE_ASH_GETOPTS | 1921 | #if ENABLE_ASH_GETOPTS |
1914 | # define optindval() (voptind.text + 7) | 1922 | # define optindval() (voptind.var_text + 7) |
1915 | #endif | 1923 | #endif |
1916 | 1924 | ||
1917 | 1925 | ||
@@ -1970,12 +1978,6 @@ varcmp(const char *p, const char *q) | |||
1970 | return c - d; | 1978 | return c - d; |
1971 | } | 1979 | } |
1972 | 1980 | ||
1973 | static int | ||
1974 | varequal(const char *a, const char *b) | ||
1975 | { | ||
1976 | return !varcmp(a, b); | ||
1977 | } | ||
1978 | |||
1979 | /* | 1981 | /* |
1980 | * Find the appropriate entry in the hash table from the name. | 1982 | * Find the appropriate entry in the hash table from the name. |
1981 | */ | 1983 | */ |
@@ -2010,15 +2012,15 @@ initvar(void) | |||
2010 | * PS1 depends on uid | 2012 | * PS1 depends on uid |
2011 | */ | 2013 | */ |
2012 | #if ENABLE_FEATURE_EDITING && ENABLE_FEATURE_EDITING_FANCY_PROMPT | 2014 | #if ENABLE_FEATURE_EDITING && ENABLE_FEATURE_EDITING_FANCY_PROMPT |
2013 | vps1.text = "PS1=\\w \\$ "; | 2015 | vps1.var_text = "PS1=\\w \\$ "; |
2014 | #else | 2016 | #else |
2015 | if (!geteuid()) | 2017 | if (!geteuid()) |
2016 | vps1.text = "PS1=# "; | 2018 | vps1.var_text = "PS1=# "; |
2017 | #endif | 2019 | #endif |
2018 | vp = varinit; | 2020 | vp = varinit; |
2019 | end = vp + ARRAY_SIZE(varinit); | 2021 | end = vp + ARRAY_SIZE(varinit); |
2020 | do { | 2022 | do { |
2021 | vpp = hashvar(vp->text); | 2023 | vpp = hashvar(vp->var_text); |
2022 | vp->next = *vpp; | 2024 | vp->next = *vpp; |
2023 | *vpp = vp; | 2025 | *vpp = vp; |
2024 | } while (++vp < end); | 2026 | } while (++vp < end); |
@@ -2028,7 +2030,7 @@ static struct var ** | |||
2028 | findvar(struct var **vpp, const char *name) | 2030 | findvar(struct var **vpp, const char *name) |
2029 | { | 2031 | { |
2030 | for (; *vpp; vpp = &(*vpp)->next) { | 2032 | for (; *vpp; vpp = &(*vpp)->next) { |
2031 | if (varequal((*vpp)->text, name)) { | 2033 | if (varcmp((*vpp)->var_text, name) == 0) { |
2032 | break; | 2034 | break; |
2033 | } | 2035 | } |
2034 | } | 2036 | } |
@@ -2052,11 +2054,11 @@ lookupvar(const char *name) | |||
2052 | * As soon as they're unset, they're no longer dynamic, and dynamic | 2054 | * As soon as they're unset, they're no longer dynamic, and dynamic |
2053 | * lookup will no longer happen at that point. -- PFM. | 2055 | * lookup will no longer happen at that point. -- PFM. |
2054 | */ | 2056 | */ |
2055 | if ((v->flags & VDYNAMIC)) | 2057 | if (v->flags & VDYNAMIC) |
2056 | (*v->func)(NULL); | 2058 | v->var_func(NULL); |
2057 | #endif | 2059 | #endif |
2058 | if (!(v->flags & VUNSET)) | 2060 | if (!(v->flags & VUNSET)) |
2059 | return strchrnul(v->text, '=') + 1; | 2061 | return var_end(v->var_text); |
2060 | } | 2062 | } |
2061 | return NULL; | 2063 | return NULL; |
2062 | } | 2064 | } |
@@ -2070,8 +2072,8 @@ bltinlookup(const char *name) | |||
2070 | struct strlist *sp; | 2072 | struct strlist *sp; |
2071 | 2073 | ||
2072 | for (sp = cmdenviron; sp; sp = sp->next) { | 2074 | for (sp = cmdenviron; sp; sp = sp->next) { |
2073 | if (varequal(sp->text, name)) | 2075 | if (varcmp(sp->text, name) == 0) |
2074 | return strchrnul(sp->text, '=') + 1; | 2076 | return var_end(sp->text); |
2075 | } | 2077 | } |
2076 | return lookupvar(name); | 2078 | return lookupvar(name); |
2077 | } | 2079 | } |
@@ -2097,24 +2099,24 @@ setvareq(char *s, int flags) | |||
2097 | 2099 | ||
2098 | if (flags & VNOSAVE) | 2100 | if (flags & VNOSAVE) |
2099 | free(s); | 2101 | free(s); |
2100 | n = vp->text; | 2102 | n = vp->var_text; |
2101 | ash_msg_and_raise_error("%.*s: is read only", strchrnul(n, '=') - n, n); | 2103 | ash_msg_and_raise_error("%.*s: is read only", strchrnul(n, '=') - n, n); |
2102 | } | 2104 | } |
2103 | 2105 | ||
2104 | if (flags & VNOSET) | 2106 | if (flags & VNOSET) |
2105 | return; | 2107 | return; |
2106 | 2108 | ||
2107 | if (vp->func && (flags & VNOFUNC) == 0) | 2109 | if (vp->var_func && !(flags & VNOFUNC)) |
2108 | (*vp->func)(strchrnul(s, '=') + 1); | 2110 | vp->var_func(var_end(s)); |
2109 | 2111 | ||
2110 | if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0) | 2112 | if (!(vp->flags & (VTEXTFIXED|VSTACK))) |
2111 | free((char*)vp->text); | 2113 | free((char*)vp->var_text); |
2112 | 2114 | ||
2113 | flags |= vp->flags & ~(VTEXTFIXED|VSTACK|VNOSAVE|VUNSET); | 2115 | flags |= vp->flags & ~(VTEXTFIXED|VSTACK|VNOSAVE|VUNSET); |
2114 | } else { | 2116 | } else { |
2117 | /* variable s is not found */ | ||
2115 | if (flags & VNOSET) | 2118 | if (flags & VNOSET) |
2116 | return; | 2119 | return; |
2117 | /* not found */ | ||
2118 | vp = ckzalloc(sizeof(*vp)); | 2120 | vp = ckzalloc(sizeof(*vp)); |
2119 | vp->next = *vpp; | 2121 | vp->next = *vpp; |
2120 | /*vp->func = NULL; - ckzalloc did it */ | 2122 | /*vp->func = NULL; - ckzalloc did it */ |
@@ -2122,7 +2124,7 @@ setvareq(char *s, int flags) | |||
2122 | } | 2124 | } |
2123 | if (!(flags & (VTEXTFIXED|VSTACK|VNOSAVE))) | 2125 | if (!(flags & (VTEXTFIXED|VSTACK|VNOSAVE))) |
2124 | s = ckstrdup(s); | 2126 | s = ckstrdup(s); |
2125 | vp->text = s; | 2127 | vp->var_text = s; |
2126 | vp->flags = flags; | 2128 | vp->flags = flags; |
2127 | } | 2129 | } |
2128 | 2130 | ||
@@ -2220,7 +2222,7 @@ unsetvar(const char *s) | |||
2220 | if ((flags & VSTRFIXED) == 0) { | 2222 | if ((flags & VSTRFIXED) == 0) { |
2221 | INT_OFF; | 2223 | INT_OFF; |
2222 | if ((flags & (VTEXTFIXED|VSTACK)) == 0) | 2224 | if ((flags & (VTEXTFIXED|VSTACK)) == 0) |
2223 | free((char*)vp->text); | 2225 | free((char*)vp->var_text); |
2224 | *vpp = vp->next; | 2226 | *vpp = vp->next; |
2225 | free(vp); | 2227 | free(vp); |
2226 | INT_ON; | 2228 | INT_ON; |
@@ -2272,7 +2274,7 @@ listvars(int on, int off, char ***end) | |||
2272 | if ((vp->flags & mask) == on) { | 2274 | if ((vp->flags & mask) == on) { |
2273 | if (ep == stackstrend()) | 2275 | if (ep == stackstrend()) |
2274 | ep = growstackstr(); | 2276 | ep = growstackstr(); |
2275 | *ep++ = (char *) vp->text; | 2277 | *ep++ = (char*)vp->var_text; |
2276 | } | 2278 | } |
2277 | } | 2279 | } |
2278 | } while (++vpp < vartab + VTABSIZE); | 2280 | } while (++vpp < vartab + VTABSIZE); |
@@ -6912,7 +6914,7 @@ evalvar(char *p, int flags, struct strlist *var_str_list) | |||
6912 | var = p; | 6914 | var = p; |
6913 | easy = (!quoted || (*var == '@' && shellparam.nparam)); | 6915 | easy = (!quoted || (*var == '@' && shellparam.nparam)); |
6914 | startloc = expdest - (char *)stackblock(); | 6916 | startloc = expdest - (char *)stackblock(); |
6915 | p = strchr(p, '=') + 1; | 6917 | p = strchr(p, '=') + 1; //TODO: use var_end(p)? |
6916 | 6918 | ||
6917 | again: | 6919 | again: |
6918 | varlen = varvalue(var, varflags, flags, var_str_list); | 6920 | varlen = varvalue(var, varflags, flags, var_str_list); |
@@ -9067,14 +9069,14 @@ poplocalvars(void) | |||
9067 | free((char*)lvp->text); | 9069 | free((char*)lvp->text); |
9068 | optschanged(); | 9070 | optschanged(); |
9069 | } else if ((lvp->flags & (VUNSET|VSTRFIXED)) == VUNSET) { | 9071 | } else if ((lvp->flags & (VUNSET|VSTRFIXED)) == VUNSET) { |
9070 | unsetvar(vp->text); | 9072 | unsetvar(vp->var_text); |
9071 | } else { | 9073 | } else { |
9072 | if (vp->func) | 9074 | if (vp->var_func) |
9073 | (*vp->func)(strchrnul(lvp->text, '=') + 1); | 9075 | vp->var_func(var_end(lvp->text)); |
9074 | if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0) | 9076 | if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0) |
9075 | free((char*)vp->text); | 9077 | free((char*)vp->var_text); |
9076 | vp->flags = lvp->flags; | 9078 | vp->flags = lvp->flags; |
9077 | vp->text = lvp->text; | 9079 | vp->var_text = lvp->text; |
9078 | } | 9080 | } |
9079 | free(lvp); | 9081 | free(lvp); |
9080 | } | 9082 | } |
@@ -9193,7 +9195,7 @@ mklocal(char *name) | |||
9193 | vp = *vpp; /* the new variable */ | 9195 | vp = *vpp; /* the new variable */ |
9194 | lvp->flags = VUNSET; | 9196 | lvp->flags = VUNSET; |
9195 | } else { | 9197 | } else { |
9196 | lvp->text = vp->text; | 9198 | lvp->text = vp->var_text; |
9197 | lvp->flags = vp->flags; | 9199 | lvp->flags = vp->flags; |
9198 | vp->flags |= VSTRFIXED|VTEXTFIXED; | 9200 | vp->flags |= VSTRFIXED|VTEXTFIXED; |
9199 | if (eq) | 9201 | if (eq) |
@@ -9517,7 +9519,7 @@ evalcommand(union node *cmd, int flags) | |||
9517 | expredir(cmd->ncmd.redirect); | 9519 | expredir(cmd->ncmd.redirect); |
9518 | status = redirectsafe(cmd->ncmd.redirect, REDIR_PUSH | REDIR_SAVEFD2); | 9520 | status = redirectsafe(cmd->ncmd.redirect, REDIR_PUSH | REDIR_SAVEFD2); |
9519 | 9521 | ||
9520 | path = vpath.text; | 9522 | path = vpath.var_text; |
9521 | for (argp = cmd->ncmd.assign; argp; argp = argp->narg.next) { | 9523 | for (argp = cmd->ncmd.assign; argp; argp = argp->narg.next) { |
9522 | struct strlist **spp; | 9524 | struct strlist **spp; |
9523 | char *p; | 9525 | char *p; |
@@ -9530,7 +9532,7 @@ evalcommand(union node *cmd, int flags) | |||
9530 | * is present | 9532 | * is present |
9531 | */ | 9533 | */ |
9532 | p = (*spp)->text; | 9534 | p = (*spp)->text; |
9533 | if (varequal(p, path)) | 9535 | if (varcmp(p, path) == 0) |
9534 | path = p; | 9536 | path = p; |
9535 | } | 9537 | } |
9536 | 9538 | ||
@@ -10578,7 +10580,7 @@ change_random(const char *value) | |||
10578 | /* "get", generate */ | 10580 | /* "get", generate */ |
10579 | t = next_random(&random_gen); | 10581 | t = next_random(&random_gen); |
10580 | /* set without recursion */ | 10582 | /* set without recursion */ |
10581 | setvar(vrandom.text, utoa(t), VNOFUNC); | 10583 | setvar(vrandom.var_text, utoa(t), VNOFUNC); |
10582 | vrandom.flags &= ~VNOFUNC; | 10584 | vrandom.flags &= ~VNOFUNC; |
10583 | } else { | 10585 | } else { |
10584 | /* set/reset */ | 10586 | /* set/reset */ |