Finished up some errors
Kinda finished extraction
This commit is contained in:
+16
-7
@@ -19,7 +19,7 @@ alloc: std.mem.Allocator,
|
||||
|
||||
block_size: u32,
|
||||
buffers: std.ArrayList(Buffer),
|
||||
buffer_queue: std.SinglyLinkedList,
|
||||
buffer_queue: std.SinglyLinkedList = .{},
|
||||
|
||||
pub fn init(alloc: std.mem.Allocator, block_size: u32) !Self {
|
||||
return .{
|
||||
@@ -37,22 +37,31 @@ pub fn deinit(self: Self) void {
|
||||
|
||||
fn decomp(d: ?*const Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
|
||||
if (d == null) {
|
||||
const buf = try alloc.alloc(u8, in.len * 2);
|
||||
var buf = try alloc.alloc(u8, in.len * 2);
|
||||
defer alloc.free(buf);
|
||||
return lzmaDecomp(buf, in, out);
|
||||
return lzmaDecomp(alloc, &buf, in, out) catch |err| return switch (err) {
|
||||
error.OutOfMemory => Error.OutOfMemory,
|
||||
else => Error.ReadFailed,
|
||||
};
|
||||
}
|
||||
var self: Self = @fieldParentPtr("interface", d.?);
|
||||
var self: *Self = @fieldParentPtr("interface", @constCast(d.?));
|
||||
const buf_node = self.buffer_queue.popFirst();
|
||||
var buf: *Buffer = undefined;
|
||||
if (buf_node == null) {
|
||||
const new_buf = try self.buffers.addOne(self.alloc);
|
||||
new_buf.* = .{ .{}, try self.alloc.alloc(u8, self.block_size + lzma.block_size_max) };
|
||||
new_buf.* = .{ .node = .{}, .buf = try self.alloc.alloc(u8, self.block_size) };
|
||||
buf = new_buf;
|
||||
} else {
|
||||
buf = @fieldParentPtr("node", buf_node);
|
||||
buf = @fieldParentPtr("node", buf_node.?);
|
||||
}
|
||||
defer self.buffer_queue.prepend(&buf.node);
|
||||
return lzmaDecomp(self.alloc, &buf.buf, in, out);
|
||||
return lzmaDecomp(self.alloc, &buf.buf, in, out) catch |err| {
|
||||
// self.err = err;
|
||||
return switch (err) {
|
||||
error.OutOfMemory => Error.OutOfMemory,
|
||||
else => Error.ReadFailed,
|
||||
};
|
||||
};
|
||||
}
|
||||
|
||||
inline fn lzmaDecomp(alloc: std.mem.Allocator, buffer: *[]u8, in: []u8, out: []u8) !usize {
|
||||
|
||||
+12
-9
@@ -19,7 +19,7 @@ alloc: std.mem.Allocator,
|
||||
|
||||
block_size: u32,
|
||||
buffers: std.ArrayList(Buffer),
|
||||
buffer_queue: std.SinglyLinkedList,
|
||||
buffer_queue: std.SinglyLinkedList = .{},
|
||||
|
||||
pub fn init(alloc: std.mem.Allocator, block_size: u32) !Self {
|
||||
return .{
|
||||
@@ -37,25 +37,28 @@ pub fn deinit(self: Self) void {
|
||||
|
||||
fn decomp(d: ?*const Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
|
||||
if (d == null) {
|
||||
const buf = try alloc.alloc(u8, in.len * 2);
|
||||
var buf = try alloc.alloc(u8, in.len * 2);
|
||||
defer alloc.free(buf);
|
||||
return lzmaDecomp(buf, in, out);
|
||||
return xzDecomp(alloc, &buf, in, out) catch return Error.ReadFailed;
|
||||
}
|
||||
var self: Self = @fieldParentPtr("interface", d.?);
|
||||
var self: *Self = @fieldParentPtr("interface", @constCast(d.?));
|
||||
const buf_node = self.buffer_queue.popFirst();
|
||||
var buf: *Buffer = undefined;
|
||||
if (buf_node == null) {
|
||||
const new_buf = try self.buffers.addOne(self.alloc);
|
||||
new_buf.* = .{ .{}, try self.alloc.alloc(u8, self.block_size + xz.block_size_max) };
|
||||
new_buf.* = .{ .node = .{}, .buf = try self.alloc.alloc(u8, self.block_size) };
|
||||
buf = new_buf;
|
||||
} else {
|
||||
buf = @fieldParentPtr("node", buf_node);
|
||||
buf = @fieldParentPtr("node", buf_node.?);
|
||||
}
|
||||
defer self.buffer_queue.prepend(&buf.node);
|
||||
return lzmaDecomp(self.alloc, &buf.buf, in, out);
|
||||
return xzDecomp(self.alloc, &buf.buf, in, out) catch {
|
||||
// self.err = err;
|
||||
return Error.ReadFailed;
|
||||
};
|
||||
}
|
||||
|
||||
inline fn lzmaDecomp(alloc: std.mem.Allocator, buffer: *[]u8, in: []u8, out: []u8) !usize {
|
||||
inline fn xzDecomp(alloc: std.mem.Allocator, buffer: *[]u8, in: []u8, out: []u8) !usize {
|
||||
var rdr: Reader = .fixed(in);
|
||||
var d = try xz.Decompress.init(&rdr, alloc, buffer.*);
|
||||
defer {
|
||||
@@ -73,5 +76,5 @@ pub const stateless_decompressor: Decompressor = .{ .decomp_fn = statelessDecomp
|
||||
fn statelessDecomp(_: ?*const Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8) Error!usize {
|
||||
var buf = try alloc.alloc(u8, in.len);
|
||||
defer alloc.free(buf);
|
||||
return lzmaDecomp(alloc, &buf, in, out) catch return Error.ReadFailed;
|
||||
return xzDecomp(alloc, &buf, in, out) catch return Error.ReadFailed;
|
||||
}
|
||||
|
||||
+4
-4
@@ -19,7 +19,7 @@ alloc: std.mem.Allocator,
|
||||
|
||||
block_size: u32,
|
||||
buffers: std.ArrayList(Buffer),
|
||||
buffer_queue: std.SinglyLinkedList,
|
||||
buffer_queue: std.SinglyLinkedList = .{},
|
||||
|
||||
pub fn init(alloc: std.mem.Allocator, block_size: u32) !Self {
|
||||
return .{
|
||||
@@ -41,15 +41,15 @@ fn decomp(d: ?*const Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8
|
||||
defer alloc.free(buf);
|
||||
return zlibDecomp(buf, in, out);
|
||||
}
|
||||
var self: Self = @fieldParentPtr("interface", d.?);
|
||||
var self: *Self = @fieldParentPtr("interface", @constCast(d.?));
|
||||
const buf_node = self.buffer_queue.popFirst();
|
||||
var buf: *Buffer = undefined;
|
||||
if (buf_node == null) {
|
||||
const new_buf = try self.buffers.addOne(self.alloc);
|
||||
new_buf.* = .{ .{}, try self.alloc.alloc(u8, self.block_size) };
|
||||
new_buf.* = .{ .node = .{}, .buf = try self.alloc.alloc(u8, self.block_size) };
|
||||
buf = new_buf;
|
||||
} else {
|
||||
buf = @fieldParentPtr("node", buf_node);
|
||||
buf = @fieldParentPtr("node", buf_node.?);
|
||||
}
|
||||
defer self.buffer_queue.prepend(&buf.node);
|
||||
return zlibDecomp(buf.buf, in, out);
|
||||
|
||||
+4
-4
@@ -19,7 +19,7 @@ alloc: std.mem.Allocator,
|
||||
|
||||
block_size: u32,
|
||||
buffers: std.ArrayList(Buffer),
|
||||
buffer_queue: std.SinglyLinkedList,
|
||||
buffer_queue: std.SinglyLinkedList = .{},
|
||||
|
||||
pub fn init(alloc: std.mem.Allocator, block_size: u32) !Self {
|
||||
return .{
|
||||
@@ -41,15 +41,15 @@ fn decomp(d: ?*const Decompressor, alloc: std.mem.Allocator, in: []u8, out: []u8
|
||||
defer alloc.free(buf);
|
||||
return zstdDecomp(buf, in, out);
|
||||
}
|
||||
var self: Self = @fieldParentPtr("interface", d.?);
|
||||
var self: *Self = @fieldParentPtr("interface", @constCast(d.?));
|
||||
const buf_node = self.buffer_queue.popFirst();
|
||||
var buf: *Buffer = undefined;
|
||||
if (buf_node == null) {
|
||||
const new_buf = try self.buffers.addOne(self.alloc);
|
||||
new_buf.* = .{ .{}, try self.alloc.alloc(u8, self.block_size + zstd.block_size_max) };
|
||||
new_buf.* = .{ .node = .{}, .buf = try self.alloc.alloc(u8, self.block_size + zstd.block_size_max) };
|
||||
buf = new_buf;
|
||||
} else {
|
||||
buf = @fieldParentPtr("node", buf_node);
|
||||
buf = @fieldParentPtr("node", buf_node.?);
|
||||
}
|
||||
defer self.buffer_queue.prepend(&buf.node);
|
||||
return zstdDecomp(buf.buf, in, out);
|
||||
|
||||
Reference in New Issue
Block a user