blob: a59ff5a8bbf4e845cdaea8d3ed498a6159d5c5ad [file] [log] [blame]
Marek BehĂșn21a14fa2017-09-03 17:00:28 +02001/*
2 * BTRFS filesystem implementation for U-Boot
3 *
4 * 2017 Marek Behun, CZ.NIC, marek.behun@nic.cz
5 *
6 * SPDX-License-Identifier: GPL-2.0+
7 */
8
9#include "btrfs.h"
10#include <linux/lzo.h>
11#include <u-boot/zlib.h>
12
13static u32 decompress_lzo(const u8 *cbuf, u32 clen, u8 *dbuf, u32 dlen)
14{
15 u32 tot_len, in_len, res;
16 size_t out_len;
17 int ret;
18
19 if (clen < 4)
20 return -1;
21
22 tot_len = le32_to_cpu(*(u32 *) cbuf);
23 cbuf += 4;
24 clen -= 4;
25 tot_len -= 4;
26
27 if (tot_len == 0 && dlen)
28 return -1;
29 if (tot_len < 4)
30 return -1;
31
32 res = 0;
33
34 while (tot_len > 4) {
35 in_len = le32_to_cpu(*(u32 *) cbuf);
36 cbuf += 4;
37 clen -= 4;
38
39 if (in_len > clen || tot_len < 4 + in_len)
40 return -1;
41
42 tot_len -= 4 + in_len;
43
44 out_len = dlen;
45 ret = lzo1x_decompress_safe(cbuf, in_len, dbuf, &out_len);
46 if (ret != LZO_E_OK)
47 return -1;
48
49 cbuf += in_len;
50 clen -= in_len;
51 dbuf += out_len;
52 dlen -= out_len;
53
54 res += out_len;
55 }
56
57 return res;
58}
59
60/* from zutil.h */
61#define PRESET_DICT 0x20
62
63static u32 decompress_zlib(const u8 *_cbuf, u32 clen, u8 *dbuf, u32 dlen)
64{
65 int wbits = MAX_WBITS, ret = -1;
66 z_stream stream;
67 u8 *cbuf;
68 u32 res;
69
70 memset(&stream, 0, sizeof(stream));
71
72 cbuf = (u8 *) _cbuf;
73
74 stream.total_in = 0;
75
76 stream.next_out = dbuf;
77 stream.avail_out = dlen;
78 stream.total_out = 0;
79
80 /* skip adler32 check if deflate and no dictionary */
81 if (clen > 2 && !(cbuf[1] & PRESET_DICT) &&
82 ((cbuf[0] & 0x0f) == Z_DEFLATED) &&
83 !(((cbuf[0] << 8) + cbuf[1]) % 31)) {
84 wbits = -((cbuf[0] >> 4) + 8);
85 cbuf += 2;
86 clen -= 2;
87 }
88
89 if (Z_OK != inflateInit2(&stream, wbits))
90 return -1;
91
92 while (stream.total_in < clen) {
93 stream.next_in = cbuf + stream.total_in;
94 stream.avail_in = min((u32) (clen - stream.total_in),
95 (u32) btrfs_info.sb.sectorsize);
96
97 ret = inflate(&stream, Z_NO_FLUSH);
98 if (ret != Z_OK)
99 break;
100 }
101
102 res = stream.total_out;
103 inflateEnd(&stream);
104
105 if (ret != Z_STREAM_END)
106 return -1;
107
108 return res;
109}
110
111u32 btrfs_decompress(u8 type, const char *c, u32 clen, char *d, u32 dlen)
112{
113 u32 res;
114 const u8 *cbuf;
115 u8 *dbuf;
116
117 cbuf = (const u8 *) c;
118 dbuf = (u8 *) d;
119
120 switch (type) {
121 case BTRFS_COMPRESS_NONE:
122 res = dlen < clen ? dlen : clen;
123 memcpy(dbuf, cbuf, res);
124 return res;
125 case BTRFS_COMPRESS_ZLIB:
126 return decompress_zlib(cbuf, clen, dbuf, dlen);
127 case BTRFS_COMPRESS_LZO:
128 return decompress_lzo(cbuf, clen, dbuf, dlen);
129 default:
130 printf("%s: Unsupported compression in extent: %i\n", __func__,
131 type);
132 return -1;
133 }
134}