1const std = @import("../../std.zig");
2const testing = std.testing;
3const xz = std.compress.xz;
4
5fn decompress(data: []const u8) ![]u8 {
6 const gpa = testing.allocator;
7
8 var in_stream: std.Io.Reader = .fixed(data);
9
10 var xz_stream = try xz.Decompress.init(&in_stream, gpa, &.{});
11 defer xz_stream.deinit();
12
13 return xz_stream.reader.allocRemaining(gpa, .unlimited);
14}
15
16fn testReader(data: []const u8, expected: []const u8) !void {
17 const gpa = testing.allocator;
18
19 const result = try decompress(data);
20 defer gpa.free(result);
21
22 try testing.expectEqualSlices(u8, expected, result);
23}
24
25fn testDecompressError(expected: anyerror, compressed: []const u8) !void {
26 const gpa = std.testing.allocator;
27 var stream: std.Io.Reader = .fixed(compressed);
28
29 var decompressor = try xz.Decompress.init(&stream, gpa, &.{});
30 defer decompressor.deinit();
31
32 try std.testing.expectError(error.ReadFailed, decompressor.reader.allocRemaining(gpa, .unlimited));
33 try std.testing.expectEqual(expected, decompressor.err orelse return error.TestFailed);
34}
35
36test "fixture good-0-empty.xz" {
37 try testReader(@embedFile("testdata/good-0-empty.xz"), "");
38}
39
40const hello_world_text =
41 \\Hello
42 \\World!
43 \\
44;
45
46test "fixture good-1-check-none.xz" {
47 try testReader(@embedFile("testdata/good-1-check-none.xz"), hello_world_text);
48}
49
50test "fixture good-1-check-crc32.xz" {
51 try testReader(@embedFile("testdata/good-1-check-crc32.xz"), hello_world_text);
52}
53
54test "fixture good-1-check-crc64.xz" {
55 try testReader(@embedFile("testdata/good-1-check-crc64.xz"), hello_world_text);
56}
57
58test "fixture good-1-check-sha256.xz" {
59 try testReader(@embedFile("testdata/good-1-check-sha256.xz"), hello_world_text);
60}
61
62test "fixture good-2-lzma2.xz" {
63 try testReader(@embedFile("testdata/good-2-lzma2.xz"), hello_world_text);
64}
65
66test "fixture good-1-block_header-1.xz" {
67 try testReader(@embedFile("testdata/good-1-block_header-1.xz"), hello_world_text);
68}
69
70test "fixture good-1-block_header-2.xz" {
71 try testReader(@embedFile("testdata/good-1-block_header-2.xz"), hello_world_text);
72}
73
74test "fixture good-1-block_header-3.xz" {
75 try testReader(@embedFile("testdata/good-1-block_header-3.xz"), hello_world_text);
76}
77
78const lorem_ipsum_text =
79 \\Lorem ipsum dolor sit amet, consectetur adipisicing
80 \\elit, sed do eiusmod tempor incididunt ut
81 \\labore et dolore magna aliqua. Ut enim
82 \\ad minim veniam, quis nostrud exercitation ullamco
83 \\laboris nisi ut aliquip ex ea commodo
84 \\consequat. Duis aute irure dolor in reprehenderit
85 \\in voluptate velit esse cillum dolore eu
86 \\fugiat nulla pariatur. Excepteur sint occaecat cupidatat
87 \\non proident, sunt in culpa qui officia
88 \\deserunt mollit anim id est laborum.
89 \\
90;
91
92test "fixture good-1-lzma2-1.xz" {
93 try testReader(@embedFile("testdata/good-1-lzma2-1.xz"), lorem_ipsum_text);
94}
95
96test "fixture good-1-lzma2-2.xz" {
97 try testReader(@embedFile("testdata/good-1-lzma2-2.xz"), lorem_ipsum_text);
98}
99
100test "fixture good-1-lzma2-3.xz" {
101 try testReader(@embedFile("testdata/good-1-lzma2-3.xz"), lorem_ipsum_text);
102}
103
104test "fixture good-1-lzma2-4.xz" {
105 try testReader(@embedFile("testdata/good-1-lzma2-4.xz"), lorem_ipsum_text);
106}
107
108test "fixture good-1-lzma2-5.xz" {
109 try testReader(@embedFile("testdata/good-1-lzma2-5.xz"), "");
110}
111
112test "fixture good-1-delta-lzma2.tiff.xz" {
113 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-delta-lzma2.tiff.xz"));
114}
115
116test "fixture good-1-x86-lzma2.xz" {
117 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-x86-lzma2.xz"));
118}
119
120test "fixture good-1-sparc-lzma2.xz" {
121 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-sparc-lzma2.xz"));
122}
123
124test "fixture good-1-arm64-lzma2-1.xz" {
125 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-arm64-lzma2-1.xz"));
126}
127
128test "fixture good-1-arm64-lzma2-2.xz" {
129 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-arm64-lzma2-2.xz"));
130}
131
132test "fixture good-1-3delta-lzma2.xz" {
133 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-3delta-lzma2.xz"));
134}
135
136test "fixture good-1-empty-bcj-lzma2.xz" {
137 try testDecompressError(error.Unsupported, @embedFile("testdata/good-1-empty-bcj-lzma2.xz"));
138}
139
140fn testDontPanic(data: []const u8) !void {
141 const buf = decompress(data) catch |err| switch (err) {
142 error.OutOfMemory => |e| return e,
143 else => return,
144 };
145 defer testing.allocator.free(buf);
146}
147
148test "size fields: integer overflow avoidance" {
149 // These cases were found via fuzz testing and each previously caused
150 // an integer overflow when decoding. We just want to ensure they no longer
151 // cause a panic
152 // TODO this not a sufficient way to test. tests should always check the result,
153 // not merely ensure that the code does not crash.
154 const header_size_overflow = "\xfd7zXZ\x00\x00\x01i\"\xde6z";
155 try testDontPanic(header_size_overflow);
156 const lzma2_chunk_size_overflow = "\xfd7zXZ\x00\x00\x01i\"\xde6\x02\x00!\x01\x08\x00\x00\x00\xd8\x0f#\x13\x01\xff\xff";
157 try testDontPanic(lzma2_chunk_size_overflow);
158 const backward_size_overflow = "\xfd7zXZ\x00\x00\x01i\"\xde6\x00\x00\x00\x00\x1c\xdfD!\x90B\x99\r\x01\x00\x00\xff\xff\x10\x00\x00\x00\x01DD\xff\xff\xff\x01";
159 try testDontPanic(backward_size_overflow);
160}