]>
Commit | Line | Data |
---|---|---|
b2441318 | 1 | // SPDX-License-Identifier: GPL-2.0 |
a43783ae | 2 | #include <errno.h> |
80a32e5b JO |
3 | #include <lzma.h> |
4 | #include <stdio.h> | |
5 | #include <linux/compiler.h> | |
611f0afe | 6 | #include "compress.h" |
80a32e5b JO |
7 | #include "util.h" |
8 | #include "debug.h" | |
9 | ||
10 | #define BUFSIZE 8192 | |
11 | ||
12 | static const char *lzma_strerror(lzma_ret ret) | |
13 | { | |
14 | switch ((int) ret) { | |
15 | case LZMA_MEM_ERROR: | |
16 | return "Memory allocation failed"; | |
17 | case LZMA_OPTIONS_ERROR: | |
18 | return "Unsupported decompressor flags"; | |
19 | case LZMA_FORMAT_ERROR: | |
20 | return "The input is not in the .xz format"; | |
21 | case LZMA_DATA_ERROR: | |
22 | return "Compressed file is corrupt"; | |
23 | case LZMA_BUF_ERROR: | |
24 | return "Compressed file is truncated or otherwise corrupt"; | |
25 | default: | |
26 | return "Unknown error, possibly a bug"; | |
27 | } | |
28 | } | |
29 | ||
30 | int lzma_decompress_to_file(const char *input, int output_fd) | |
31 | { | |
32 | lzma_action action = LZMA_RUN; | |
33 | lzma_stream strm = LZMA_STREAM_INIT; | |
34 | lzma_ret ret; | |
ffe67c2f | 35 | int err = -1; |
80a32e5b JO |
36 | |
37 | u8 buf_in[BUFSIZE]; | |
38 | u8 buf_out[BUFSIZE]; | |
39 | FILE *infile; | |
40 | ||
41 | infile = fopen(input, "rb"); | |
42 | if (!infile) { | |
43 | pr_err("lzma: fopen failed on %s: '%s'\n", | |
44 | input, strerror(errno)); | |
45 | return -1; | |
46 | } | |
47 | ||
48 | ret = lzma_stream_decoder(&strm, UINT64_MAX, LZMA_CONCATENATED); | |
49 | if (ret != LZMA_OK) { | |
50 | pr_err("lzma: lzma_stream_decoder failed %s (%d)\n", | |
51 | lzma_strerror(ret), ret); | |
ffe67c2f | 52 | goto err_fclose; |
80a32e5b JO |
53 | } |
54 | ||
55 | strm.next_in = NULL; | |
56 | strm.avail_in = 0; | |
57 | strm.next_out = buf_out; | |
58 | strm.avail_out = sizeof(buf_out); | |
59 | ||
60 | while (1) { | |
61 | if (strm.avail_in == 0 && !feof(infile)) { | |
62 | strm.next_in = buf_in; | |
63 | strm.avail_in = fread(buf_in, 1, sizeof(buf_in), infile); | |
64 | ||
65 | if (ferror(infile)) { | |
66 | pr_err("lzma: read error: %s\n", strerror(errno)); | |
ffe67c2f | 67 | goto err_fclose; |
80a32e5b JO |
68 | } |
69 | ||
70 | if (feof(infile)) | |
71 | action = LZMA_FINISH; | |
72 | } | |
73 | ||
74 | ret = lzma_code(&strm, action); | |
75 | ||
76 | if (strm.avail_out == 0 || ret == LZMA_STREAM_END) { | |
77 | ssize_t write_size = sizeof(buf_out) - strm.avail_out; | |
78 | ||
79 | if (writen(output_fd, buf_out, write_size) != write_size) { | |
80 | pr_err("lzma: write error: %s\n", strerror(errno)); | |
ffe67c2f | 81 | goto err_fclose; |
80a32e5b JO |
82 | } |
83 | ||
84 | strm.next_out = buf_out; | |
85 | strm.avail_out = sizeof(buf_out); | |
86 | } | |
87 | ||
88 | if (ret != LZMA_OK) { | |
89 | if (ret == LZMA_STREAM_END) | |
ffe67c2f | 90 | break; |
80a32e5b JO |
91 | |
92 | pr_err("lzma: failed %s\n", lzma_strerror(ret)); | |
ffe67c2f | 93 | goto err_fclose; |
80a32e5b JO |
94 | } |
95 | } | |
96 | ||
ffe67c2f SL |
97 | err = 0; |
98 | err_fclose: | |
80a32e5b | 99 | fclose(infile); |
ffe67c2f | 100 | return err; |
80a32e5b | 101 | } |