Almost there?

This commit is contained in:
Caleb Gardner
2025-05-17 14:33:16 -05:00
parent f77c2ecf48
commit 9f345e5fdb
5 changed files with 163 additions and 21 deletions
+1
View File
@@ -17,6 +17,7 @@ pub const DecompressType = enum(u16) {
pub fn decompress(self: DecompressType, alloc: std.mem.Allocator, rdr: io.AnyReader) !std.ArrayList(u8) {
var out = std.ArrayList(u8).init(alloc);
errdefer out.deinit();
switch (self) {
.zlib => try compress.zlib.decompress(rdr, out.writer()),
.lzma => {
+60 -3
View File
@@ -1,5 +1,62 @@
pub const DirHeader = packed struct {};
const std = @import("std");
const io = std.io;
pub const RawDirEntry = struct {};
const InodeType = @import("inode/inode.zig").InodeType;
pub const DirEntry = struct {};
const DirHeader = packed struct {
count: u32,
inode_block_start: u32,
inode_num: u32,
};
const RawDirEntryStart = packed struct {
inode_block_offset: u16,
/// Difference from the current DirHeader inode_num
inode_num_difference: i16,
/// Extended inodes will be their basic type.
inode_type: InodeType,
name_size: u16,
};
pub const DirEntry = struct {
block_start: u32,
offset: u16,
inode_num: u32,
name: []const u8,
fn init(alloc: std.mem.Allocator, hdr: DirHeader, rdr: io.AnyReader) !DirEntry {
const raw = try rdr.readStruct(RawDirEntryStart);
const name = try alloc.alloc(u8, raw.name_size + 1);
_ = try rdr.read(name);
return .{
.block_start = hdr.inode_block_start,
.offset = raw.inode_block_offset,
.inode_num = if (raw.inode_num_difference > 0)
hdr.inode_num + @abs(raw.inode_num_difference)
else
hdr.inode_num - @abs(raw.inode_num_difference),
.name = name,
};
}
pub fn deinit(self: DirEntry, alloc: std.mem.Allocator) void {
alloc.free(self.name);
}
};
pub fn readDirectory(alloc: std.mem.Allocator, rdr: io.AnyReader, size: u64) !std.StringHashMap(DirEntry) {
var out: std.StringHashMap(DirEntry) = .init(alloc);
errdefer out.deinit();
var red_size: u64 = 3;
var hdr: DirHeader = try rdr.readStruct(DirHeader);
while (red_size < size) : (hdr = try rdr.readStruct(DirHeader)) {
var i: u32 = 0;
while (i <= hdr.count) : (i += 1) {
var tmp: DirEntry = try .init(alloc, hdr, rdr);
errdefer tmp.deinit(alloc);
try out.put(tmp.name, tmp);
red_size += 8 + tmp.name.len;
}
}
return out;
}
+87 -8
View File
@@ -2,6 +2,7 @@ const std = @import("std");
const io = std.io;
const inode = @import("inode/inode.zig");
const directory = @import("directory.zig");
const Reader = @import("reader.zig").Reader;
const DirEntry = @import("directory.zig").DirEntry;
@@ -11,24 +12,55 @@ const FileError = error{
NotDirectory,
NotNormalFile,
NotSymlink,
NotFound,
};
pub const File = struct {
name: []const u8,
inode: inode.Inode,
dirEntries: std.StringHashMap(DirEntry) = undefined,
hasEntries: bool = false,
// pub fn fromDirEntry(read: Reader, ent: DirEntry) !File {}
pub fn open(self: File, reader: *Reader, path: []const u8) !File {
if (path.len == 0 || std.mem.eql(u8, path, ".")) {
return self;
pub fn deinit(self: *File, alloc: std.mem.Allocator) void {
self.inode.deinit();
alloc.free(self.name);
if (self.hasEntries) {
var iter = self.dirEntries.iterator();
while (iter.next()) |ent| {
ent.value_ptr.deinit(alloc);
}
switch (inode.InodeHeader.inode_type) {
self.dirEntries.deinit();
}
}
pub fn open(self: *File, reader: *Reader, path: []const u8) !File {
return self.realOpen(reader, path, true);
}
fn realOpen(self: *File, reader: *Reader, path: []const u8, first: bool) !File {
const clean_path: []const u8 = std.mem.trimLeft(u8, path, "/");
if (clean_path.len == 0) {
return self.*;
}
if (!first) {
defer reader.alloc.free(path);
defer self.deinit(reader.alloc);
}
switch (self.inode.header.inode_type) {
.dir, .ext_dir => {},
else => return FileError.NotDirectory,
}
_ = reader;
//TODO: read dir entries and find correct inode and file
if (!self.hasEntries) {
try self.readDirEntries(reader);
}
const split_idx = std.mem.indexOf(u8, clean_path, "/") orelse clean_path.len;
const name = clean_path[0..split_idx];
const ent = self.dirEntries.get(name);
if (ent == null) {
return FileError.NotFound;
}
var fil = try fileFromDirEntry(reader, ent.?);
return fil.realOpen(reader, clean_path[split_idx..], false);
}
pub fn symPath(self: File) ![]const u8 {
@@ -38,4 +70,51 @@ pub const File = struct {
else => FileError.NotSymlink,
};
}
fn readDirEntries(self: *File, reader: *Reader) !void {
var block_start: u32 = 0;
var offset: u16 = 0;
var size: u32 = 0;
switch (self.inode.data) {
.dir => |d| {
block_start = d.block_start;
offset = d.offset;
size = d.size;
},
.ext_dir => |d| {
block_start = d.block_start;
offset = d.offset;
size = d.size;
},
else => return FileError.NotDirectory,
}
var offset_rdr = reader.holder.readerAt(reader.super.dir_table_start + block_start);
var meta_rdr: MetadataReader = try .init(
reader.alloc,
offset_rdr.any(),
reader.super.decomp,
);
defer meta_rdr.deinit();
self.dirEntries = try directory.readDirectory(reader.alloc, meta_rdr.any(), size);
self.hasEntries = true;
}
};
fn fileFromDirEntry(read: *Reader, ent: DirEntry) !File {
var offset_rdr = read.holder.readerAt(ent.block_start + read.super.inode_table_start);
var meta_rdr: MetadataReader = try .init(
read.alloc,
offset_rdr.any(),
read.super.decomp,
);
defer meta_rdr.deinit();
try meta_rdr.skip(ent.offset);
return .{
.name = ent.name,
.inode = try .init(
read.alloc,
meta_rdr.any(),
read.super.block_size,
),
};
}
+1 -1
View File
@@ -15,7 +15,7 @@ pub const InodeRef = packed struct {
};
pub const InodeType = enum(u16) {
dir,
dir = 1,
file,
sym,
block,
+14 -9
View File
@@ -6,9 +6,10 @@ const FileHolder = @import("readers/file_holder.zig").FileHolder;
const Superblock = @import("superblock.zig").Superblock;
const File = @import("file.zig").File;
const MetadataReader = @import("readers/metadata.zig").MetadataReader;
const DirEntry = @import("directory.zig").DirEntry;
pub const Reader = struct {
arena: std.heap.ArenaAllocator,
alloc: std.mem.Allocator,
holder: FileHolder,
super: Superblock,
root: File,
@@ -17,9 +18,8 @@ pub const Reader = struct {
var holder: FileHolder = try .init(filepath, offset);
const super: Superblock = try holder.reader().readStruct(Superblock);
try super.validate();
const arena: std.heap.ArenaAllocator = .init(alloc);
var out: Reader = .{
.arena = arena,
.alloc = alloc,
.holder = holder,
.super = super,
.root = undefined,
@@ -28,14 +28,14 @@ pub const Reader = struct {
return out;
}
pub fn deinit(self: *Reader) void {
self.arena.deinit();
self.root.deinit(self.alloc);
self.holder.deinit();
}
fn fileFromRef(self: *Reader, ref: inode.InodeRef, name: []const u8) !File {
var offset_rdr = self.holder.readerAt(ref.block_start + self.super.inode_table_start);
var meta_rdr: MetadataReader = try .init(
self.arena.allocator(),
self.alloc,
offset_rdr.any(),
self.super.decomp,
);
@@ -44,7 +44,7 @@ pub const Reader = struct {
return .{
.name = name,
.inode = try .init(
self.arena.allocator(),
self.alloc,
meta_rdr.any(),
self.super.block_size,
),
@@ -53,8 +53,13 @@ pub const Reader = struct {
};
test "reader" {
const test_file_path = "testing/LinuxPATest.sfs";
var rdr: Reader = try .init(std.testing.allocator, test_file_path, 0);
const test_sfs_path = "testing/LinuxPATest.sfs";
const test_file_path = "PortableApps/PortableApps.com/Data/PortableAppsMenu.ini";
var rdr: Reader = try .init(std.testing.allocator, test_sfs_path, 0);
defer rdr.deinit();
std.debug.print("{}\n", .{rdr.root});
var fil = try rdr.root.open(&rdr, test_file_path);
defer fil.deinit(rdr.alloc);
std.debug.print("{}\n", .{fil});
}