authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2025-08-25 18:47:33-07:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2025-08-26 21:00:58-07:00
log8523cbef0e0145a1f4de84f49044b8c9a8657108
tree7758461ad457121a94bc7e240855edf3afdd4630
parent58e60697e2930f4311ae9e744ae1c2877e0b69ed

std.compress.lzma: tests passing


2 files changed, 22 insertions(+), 4 deletions(-)

lib/std/compress/lzma.zig+12-1
......@@ -696,6 +696,7 @@ pub const Decompress = struct {
696696 .vtable = &.{
697697 .readVec = readVec,
698698 .stream = stream,
699 .discard = discard,
699700 },
700701 .seek = 0,
701702 .end = 0,
......@@ -746,12 +747,22 @@ pub const Decompress = struct {
746747 return readIndirect(r);
747748 }
748749
750 fn discard(r: *Reader, limit: std.Io.Limit) Reader.Error!usize {
751 const d: *Decompress = @alignCast(@fieldParentPtr("reader", r));
752 _ = d;
753 _ = limit;
754 @panic("TODO");
755 }
756
749757 fn readIndirect(r: *Reader) Reader.Error!usize {
750758 const d: *Decompress = @alignCast(@fieldParentPtr("reader", r));
751759 const gpa = d.gpa;
752760 var allocating = Writer.Allocating.initOwnedSlice(gpa, r.buffer);
753761 allocating.writer.end = r.end;
754 defer r.end = allocating.writer.end;
762 defer {
763 r.buffer = allocating.writer.buffer;
764 r.end = allocating.writer.end;
765 }
755766 if (d.decode.state == math.maxInt(usize)) return error.EndOfStream;
756767 d.decode.process(d.input, &allocating, &d.buffer, &d.range_decoder) catch |err| switch (err) {
757768 error.WriteFailed => {
lib/std/compress/lzma/test.zig+10-3
......@@ -18,7 +18,14 @@ fn testDecompressEqual(expected: []const u8, compressed: []const u8) !void {
1818}
1919
2020fn testDecompressError(expected: anyerror, compressed: []const u8) !void {
21 return std.testing.expectError(expected, testDecompress(compressed));
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);
2229}
2330
2431test "decompress empty world" {
......@@ -76,14 +83,14 @@ test "known size with end of payload marker" {
7683
7784test "too big uncompressed size in header" {
7885 try testDecompressError(
79 error.CorruptInput,
86 error.DecompressedSizeMismatch,
8087 @embedFile("testdata/bad-too_big_size-with_eopm.lzma"),
8188 );
8289}
8390
8491test "too small uncompressed size in header" {
8592 try testDecompressError(
86 error.CorruptInput,
93 error.DecompressedSizeMismatch,
8794 @embedFile("testdata/bad-too_small_size-without_eopm-3.lzma"),
8895 );
8996}