1const std = @import("../../std.zig");
2const lzma = std.compress.lzma;
3
4fn testDecompress(compressed: []const u8) ![]u8 {
5 const gpa = std.testing.allocator;
6 var stream: std.Io.Reader = .fixed(compressed);
7
8 var decompressor = try lzma.Decompress.initOptions(&stream, gpa, &.{}, .{}, std.math.maxInt(u32));
9 defer decompressor.deinit();
10 return decompressor.reader.allocRemaining(gpa, .unlimited);
11}
12
13fn testDecompressEqual(expected: []const u8, compressed: []const u8) !void {
14 const gpa = std.testing.allocator;
15 const decomp = try testDecompress(compressed);
16 defer gpa.free(decomp);
17 try std.testing.expectEqualSlices(u8, expected, decomp);
18}
19
20fn testDecompressError(expected: anyerror, compressed: []const u8) !void {
21 const gpa = std.testing.allocator;
22 var stream: std.Io.Reader = .fixed(compressed);
23
24 var decompressor = try lzma.Decompress.initOptions(&stream, gpa, &.{}, .{}, std.math.maxInt(u32));
25 defer decompressor.deinit();
26
27 try std.testing.expectError(error.ReadFailed, decompressor.reader.allocRemaining(gpa, .unlimited));
28 try std.testing.expectEqual(expected, decompressor.err orelse return error.TestFailed);
29}
30
31test "decompress empty world" {
32 try testDecompressEqual(
33 "",
34 &[_]u8{
35 0x5d, 0x00, 0x00, 0x80, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x83, 0xff,
36 0xfb, 0xff, 0xff, 0xc0, 0x00, 0x00, 0x00,
37 },
38 );
39}
40
41test "decompress hello world" {
42 try testDecompressEqual(
43 "Hello world\n",
44 &[_]u8{
45 0x5d, 0x00, 0x00, 0x80, 0x00, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x24, 0x19,
46 0x49, 0x98, 0x6f, 0x10, 0x19, 0xc6, 0xd7, 0x31, 0xeb, 0x36, 0x50, 0xb2, 0x98, 0x48, 0xff, 0xfe,
47 0xa5, 0xb0, 0x00,
48 },
49 );
50}
51
52test "decompress huge dict" {
53 try testDecompressEqual(
54 "Hello world\n",
55 &[_]u8{
56 0x5d, 0x7f, 0x7f, 0x7f, 0x7f, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0x00, 0x24, 0x19,
57 0x49, 0x98, 0x6f, 0x10, 0x19, 0xc6, 0xd7, 0x31, 0xeb, 0x36, 0x50, 0xb2, 0x98, 0x48, 0xff, 0xfe,
58 0xa5, 0xb0, 0x00,
59 },
60 );
61}
62
63test "unknown size with end of payload marker" {
64 try testDecompressEqual(
65 "Hello\nWorld!\n",
66 @embedFile("testdata/good-unknown_size-with_eopm.lzma"),
67 );
68}
69
70test "known size without end of payload marker" {
71 try testDecompressEqual(
72 "Hello\nWorld!\n",
73 @embedFile("testdata/good-known_size-without_eopm.lzma"),
74 );
75}
76
77test "known size with end of payload marker" {
78 try testDecompressEqual(
79 "Hello\nWorld!\n",
80 @embedFile("testdata/good-known_size-with_eopm.lzma"),
81 );
82}
83
84test "too big uncompressed size in header" {
85 try testDecompressError(
86 error.DecompressedSizeMismatch,
87 @embedFile("testdata/bad-too_big_size-with_eopm.lzma"),
88 );
89}
90
91test "too small uncompressed size in header" {
92 try testDecompressError(
93 error.DecompressedSizeMismatch,
94 @embedFile("testdata/bad-too_small_size-without_eopm-3.lzma"),
95 );
96}
97
98test "reading one byte" {
99 const gpa = std.testing.allocator;
100 const compressed = @embedFile("testdata/good-known_size-with_eopm.lzma");
101 var stream: std.Io.Reader = .fixed(compressed);
102 var decompressor = try lzma.Decompress.initOptions(&stream, gpa, &.{}, .{}, std.math.maxInt(u32));
103 defer decompressor.deinit();
104
105 var buffer: [1]u8 = undefined;
106 try decompressor.reader.readSliceAll(&buffer);
107 try std.testing.expectEqual(72, buffer[0]);
108}