1*83d290c5STom Rini // SPDX-License-Identifier: GPL-2.0+
221a14facSMarek Behún /*
321a14facSMarek Behún * BTRFS filesystem implementation for U-Boot
421a14facSMarek Behún *
521a14facSMarek Behún * 2017 Marek Behun, CZ.NIC, marek.behun@nic.cz
621a14facSMarek Behún */
721a14facSMarek Behún
821a14facSMarek Behún #include "btrfs.h"
921a14facSMarek Behún #include <linux/lzo.h>
1021a14facSMarek Behún #include <u-boot/zlib.h>
112021f083SAlberto Sánchez Molero #include <asm/unaligned.h>
1221a14facSMarek Behún
decompress_lzo(const u8 * cbuf,u32 clen,u8 * dbuf,u32 dlen)1321a14facSMarek Behún static u32 decompress_lzo(const u8 *cbuf, u32 clen, u8 *dbuf, u32 dlen)
1421a14facSMarek Behún {
1521a14facSMarek Behún u32 tot_len, in_len, res;
1621a14facSMarek Behún size_t out_len;
1721a14facSMarek Behún int ret;
1821a14facSMarek Behún
1921a14facSMarek Behún if (clen < 4)
2021a14facSMarek Behún return -1;
2121a14facSMarek Behún
222021f083SAlberto Sánchez Molero tot_len = le32_to_cpu(get_unaligned((u32 *)cbuf));
2321a14facSMarek Behún cbuf += 4;
2421a14facSMarek Behún clen -= 4;
2521a14facSMarek Behún tot_len -= 4;
2621a14facSMarek Behún
2721a14facSMarek Behún if (tot_len == 0 && dlen)
2821a14facSMarek Behún return -1;
2921a14facSMarek Behún if (tot_len < 4)
3021a14facSMarek Behún return -1;
3121a14facSMarek Behún
3221a14facSMarek Behún res = 0;
3321a14facSMarek Behún
3421a14facSMarek Behún while (tot_len > 4) {
352021f083SAlberto Sánchez Molero in_len = le32_to_cpu(get_unaligned((u32 *)cbuf));
3621a14facSMarek Behún cbuf += 4;
3721a14facSMarek Behún clen -= 4;
3821a14facSMarek Behún
3921a14facSMarek Behún if (in_len > clen || tot_len < 4 + in_len)
4021a14facSMarek Behún return -1;
4121a14facSMarek Behún
4221a14facSMarek Behún tot_len -= 4 + in_len;
4321a14facSMarek Behún
4421a14facSMarek Behún out_len = dlen;
4521a14facSMarek Behún ret = lzo1x_decompress_safe(cbuf, in_len, dbuf, &out_len);
4621a14facSMarek Behún if (ret != LZO_E_OK)
4721a14facSMarek Behún return -1;
4821a14facSMarek Behún
4921a14facSMarek Behún cbuf += in_len;
5021a14facSMarek Behún clen -= in_len;
5121a14facSMarek Behún dbuf += out_len;
5221a14facSMarek Behún dlen -= out_len;
5321a14facSMarek Behún
5421a14facSMarek Behún res += out_len;
5521a14facSMarek Behún }
5621a14facSMarek Behún
5721a14facSMarek Behún return res;
5821a14facSMarek Behún }
5921a14facSMarek Behún
6021a14facSMarek Behún /* from zutil.h */
6121a14facSMarek Behún #define PRESET_DICT 0x20
6221a14facSMarek Behún
decompress_zlib(const u8 * _cbuf,u32 clen,u8 * dbuf,u32 dlen)6321a14facSMarek Behún static u32 decompress_zlib(const u8 *_cbuf, u32 clen, u8 *dbuf, u32 dlen)
6421a14facSMarek Behún {
6521a14facSMarek Behún int wbits = MAX_WBITS, ret = -1;
6621a14facSMarek Behún z_stream stream;
6721a14facSMarek Behún u8 *cbuf;
6821a14facSMarek Behún u32 res;
6921a14facSMarek Behún
7021a14facSMarek Behún memset(&stream, 0, sizeof(stream));
7121a14facSMarek Behún
7221a14facSMarek Behún cbuf = (u8 *) _cbuf;
7321a14facSMarek Behún
7421a14facSMarek Behún stream.total_in = 0;
7521a14facSMarek Behún
7621a14facSMarek Behún stream.next_out = dbuf;
7721a14facSMarek Behún stream.avail_out = dlen;
7821a14facSMarek Behún stream.total_out = 0;
7921a14facSMarek Behún
8021a14facSMarek Behún /* skip adler32 check if deflate and no dictionary */
8121a14facSMarek Behún if (clen > 2 && !(cbuf[1] & PRESET_DICT) &&
8221a14facSMarek Behún ((cbuf[0] & 0x0f) == Z_DEFLATED) &&
8321a14facSMarek Behún !(((cbuf[0] << 8) + cbuf[1]) % 31)) {
8421a14facSMarek Behún wbits = -((cbuf[0] >> 4) + 8);
8521a14facSMarek Behún cbuf += 2;
8621a14facSMarek Behún clen -= 2;
8721a14facSMarek Behún }
8821a14facSMarek Behún
8921a14facSMarek Behún if (Z_OK != inflateInit2(&stream, wbits))
9021a14facSMarek Behún return -1;
9121a14facSMarek Behún
9221a14facSMarek Behún while (stream.total_in < clen) {
9321a14facSMarek Behún stream.next_in = cbuf + stream.total_in;
9421a14facSMarek Behún stream.avail_in = min((u32) (clen - stream.total_in),
9521a14facSMarek Behún (u32) btrfs_info.sb.sectorsize);
9621a14facSMarek Behún
9721a14facSMarek Behún ret = inflate(&stream, Z_NO_FLUSH);
9821a14facSMarek Behún if (ret != Z_OK)
9921a14facSMarek Behún break;
10021a14facSMarek Behún }
10121a14facSMarek Behún
10221a14facSMarek Behún res = stream.total_out;
10321a14facSMarek Behún inflateEnd(&stream);
10421a14facSMarek Behún
10521a14facSMarek Behún if (ret != Z_STREAM_END)
10621a14facSMarek Behún return -1;
10721a14facSMarek Behún
10821a14facSMarek Behún return res;
10921a14facSMarek Behún }
11021a14facSMarek Behún
btrfs_decompress(u8 type,const char * c,u32 clen,char * d,u32 dlen)11121a14facSMarek Behún u32 btrfs_decompress(u8 type, const char *c, u32 clen, char *d, u32 dlen)
11221a14facSMarek Behún {
11321a14facSMarek Behún u32 res;
11421a14facSMarek Behún const u8 *cbuf;
11521a14facSMarek Behún u8 *dbuf;
11621a14facSMarek Behún
11721a14facSMarek Behún cbuf = (const u8 *) c;
11821a14facSMarek Behún dbuf = (u8 *) d;
11921a14facSMarek Behún
12021a14facSMarek Behún switch (type) {
12121a14facSMarek Behún case BTRFS_COMPRESS_NONE:
12221a14facSMarek Behún res = dlen < clen ? dlen : clen;
12321a14facSMarek Behún memcpy(dbuf, cbuf, res);
12421a14facSMarek Behún return res;
12521a14facSMarek Behún case BTRFS_COMPRESS_ZLIB:
12621a14facSMarek Behún return decompress_zlib(cbuf, clen, dbuf, dlen);
12721a14facSMarek Behún case BTRFS_COMPRESS_LZO:
12821a14facSMarek Behún return decompress_lzo(cbuf, clen, dbuf, dlen);
12921a14facSMarek Behún default:
13021a14facSMarek Behún printf("%s: Unsupported compression in extent: %i\n", __func__,
13121a14facSMarek Behún type);
13221a14facSMarek Behún return -1;
13321a14facSMarek Behún }
13421a14facSMarek Behún }
135