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 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; 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