aboutsummaryrefslogtreecommitdiff
path: root/strip-store-paths/strip-store-paths.zig
blob: 9ded95e4d32500e2b8a273c6057a1af93428c9ea (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
const std = @import("std");

pub const File = struct {
    ptr: []align(std.mem.page_size) u8,
    len: u64,
    allocator: std.mem.Allocator,

    pub fn init(fd: std.os.fd_t, allocator: std.mem.Allocator) !File {
        var stats = try std.os.fstat(fd);
        if (stats.size == 0) {
            return error.ZeroFile;
        }
        var ptr = try std.os.mmap(null, @intCast(usize, stats.size), std.os.PROT.READ | std.os.PROT.WRITE, std.os.MAP.SHARED, fd, 0);
        return File{ .ptr = ptr, .len = @intCast(u64, stats.size), .allocator = allocator };
    }

    pub fn deinit(self: *File) void {
        std.os.munmap(self.ptr);
        self.ptr = undefined;
        self.len = 0;
    }
};

pub fn main() !void {
    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
    defer arena.deinit();
    const allocator = arena.allocator();

    const args = try std.process.argsAlloc(allocator);
    if (args.len != 2) {
        std.debug.print("usage: {s} file\n", .{args[0]});
        std.os.exit(1);
    }
    const path = args[1];

    // mmap input
    var fd = try std.os.open(path, std.os.O.RDWR, 0);
    var input = File.init(fd, allocator) catch |err| if (err == error.ZeroFile) {
        return;
    } else {
        return err;
    };
    defer input.deinit();

    // search for /nix/store
    var i: usize = 0;
    const needle = "/nix/store/";
    while (std.mem.indexOfPos(u8, input.ptr, i, needle)) |idx| {
        i = idx + needle.len;
        var j = i + 32; // pos of - in a true path
        if (j < input.len and input.ptr[j] == '-') {
            std.mem.copy(u8, input.ptr[i..], "00000000000000000000000000000000");
        }
    }
}