aboutsummaryrefslogtreecommitdiff
path: root/shared/afs/io/block.zig
blob: 7cbade8f4c9e41a022f0884fad332a460793b0fe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
//! AFS Block I/O Interface

const errors = @import("shared").afs.errors;

const BlockError = errors.block.BlockError;

pub const BlockReader = struct {
    Context: *anyopaque,
    ReadFn: *const fn (context: *anyopaque, cell: u64, buffer: []u8) BlockError!void,
    CellSize: u32,
    TotalCells: u64,

    pub fn readCell(self: *const BlockReader, cell: u64, buffer: []u8) BlockError!void {
        if (cell >= self.TotalCells) {
            return BlockError.OutOfBounds;
        }
        if (buffer.len < self.CellSize) {
            return BlockError.InvalidCell;
        }
        return self.ReadFn(self.Context, cell, buffer);
    }

    pub fn readCells(self: *const BlockReader, start_cell: u64, buffer: []u8) BlockError!void {
        const cells_to_read = buffer.len / self.CellSize;
        var offset: usize = 0;
        var cell = start_cell;

        while (offset < buffer.len and cell < start_cell + cells_to_read) : ({
            offset += self.CellSize;
            cell += 1;
        }) {
            try self.readCell(cell, buffer[offset..][0..self.CellSize]);
        }
    }
};

pub const BlockWriter = struct {
    Context: *anyopaque,
    WriteFn: *const fn (context: *anyopaque, cell: u64, data: []const u8) BlockError!void,
    CellSize: u32,
    TotalCells: u64,

    pub fn writeCell(self: *const BlockWriter, cell: u64, data: []const u8) BlockError!void {
        if (cell >= self.TotalCells) {
            return BlockError.OutOfBounds;
        }
        if (data.len < self.CellSize) {
            return BlockError.InvalidCell;
        }
        return self.WriteFn(self.Context, cell, data);
    }

    pub fn writeCells(self: *const BlockWriter, start_cell: u64, data: []const u8) BlockError!void {
        const cells_to_write = data.len / self.CellSize;
        var offset: usize = 0;
        var cell = start_cell;

        while (offset < data.len and cell < start_cell + cells_to_write) : ({
            offset += self.CellSize;
            cell += 1;
        }) {
            try self.writeCell(cell, data[offset..][0..self.CellSize]);
        }
    }
};

pub const BlockDevice = struct {
    Reader: BlockReader,
    Writer: BlockWriter,

    pub fn readCell(self: *const BlockDevice, cell: u64, buffer: []u8) BlockError!void {
        return self.Reader.readCell(cell, buffer);
    }

    pub fn writeCell(self: *const BlockDevice, cell: u64, data: []const u8) BlockError!void {
        return self.Writer.writeCell(cell, data);
    }
};