• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // SPDX-License-Identifier: GPL-2.0
2 #include <errno.h>
3 #include <lzma.h>
4 #include <stdio.h>
5 #include <linux/compiler.h>
6 #include "compress.h"
7 #include "util.h"
8 #include "debug.h"
9 
10 #define BUFSIZE 8192
11 
lzma_strerror(lzma_ret ret)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 
lzma_decompress_to_file(const char * input,int output_fd)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;
35 	int err = -1;
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);
52 		goto err_fclose;
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));
67 				goto err_fclose;
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));
81 				goto err_fclose;
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)
90 				break;
91 
92 			pr_err("lzma: failed %s\n", lzma_strerror(ret));
93 			goto err_fclose;
94 		}
95 	}
96 
97 	err = 0;
98 err_fclose:
99 	fclose(infile);
100 	return err;
101 }
102