1const std = @import("std");
 2const abi = std.Build.abi.fuzz;
 3const native_endian = @import("builtin").cpu.arch.endian();
 4
 5fn testOne(in: abi.Slice) callconv(.c) void {
 6    std.debug.assertReadable(in.toSlice());
 7}
 8
 9pub fn main() !void {
10    var debug_gpa_ctx: std.heap.DebugAllocator(.{}) = .init;
11    defer _ = debug_gpa_ctx.deinit();
12    const gpa = debug_gpa_ctx.allocator();
13
14    var args = try std.process.argsWithAllocator(gpa);
15    defer args.deinit();
16    _ = args.skip(); // executable name
17
18    var threaded: std.Io.Threaded = .init(gpa);
19    defer threaded.deinit();
20    const io = threaded.io();
21
22    const cache_dir_path = args.next() orelse @panic("expected cache directory path argument");
23    var cache_dir = try std.fs.cwd().openDir(cache_dir_path, .{});
24    defer cache_dir.close();
25
26    abi.fuzzer_init(.fromSlice(cache_dir_path));
27    abi.fuzzer_init_test(testOne, .fromSlice("test"));
28    abi.fuzzer_new_input(.fromSlice(""));
29    abi.fuzzer_new_input(.fromSlice("hello"));
30
31    const pc_digest = abi.fuzzer_coverage().id;
32    const coverage_file_path = "v/" ++ std.fmt.hex(pc_digest);
33    const coverage_file = try cache_dir.openFile(coverage_file_path, .{});
34    defer coverage_file.close();
35
36    var read_buf: [@sizeOf(abi.SeenPcsHeader)]u8 = undefined;
37    var r = coverage_file.reader(io, &read_buf);
38    const pcs_header = r.interface.takeStruct(abi.SeenPcsHeader, native_endian) catch return r.err.?;
39
40    if (pcs_header.pcs_len == 0)
41        return error.ZeroPcs;
42    const expected_len = @sizeOf(abi.SeenPcsHeader) +
43        try std.math.divCeil(usize, pcs_header.pcs_len, @bitSizeOf(usize)) * @sizeOf(usize) +
44        pcs_header.pcs_len * @sizeOf(usize);
45    if (try coverage_file.getEndPos() != expected_len)
46        return error.WrongEnd;
47}