tiny.tracy.TracedAllocator

1pub const TracedAllocator = struct {2    backing: std.mem.Allocator,3    name: []const u8,45    pub fn init(backing: std.mem.Allocator, options: Options) TracedAllocator {6        return .{7            .backing = backing,8            .name = options.name,9        };10    }1112    pub fn allocator(self: *TracedAllocator) std.mem.Allocator {13        return .{ .ptr = self, .vtable = &vtable };14    }1516    const vtable: std.mem.Allocator.VTable = .{17        .alloc = rawAlloc,18        .resize = rawResize,19        .remap = rawRemap,20        .free = rawFree,21    };2223    fn rawAlloc(ctx: *anyopaque, len: usize, alignment: std.mem.Alignment, ret_addr: usize) ?[*]u8 {24        const self: *TracedAllocator = @ptrCast(@alignCast(ctx));25        const ptr = self.backing.rawAlloc(len, alignment, ret_addr) orelse return null;26        runtime.allocAddress(addressOf(ptr), len, self.name);27        return ptr;28    }2930    fn rawResize(ctx: *anyopaque, memory: []u8, alignment: std.mem.Alignment, new_len: usize, ret_addr: usize) bool {31        const self: *TracedAllocator = @ptrCast(@alignCast(ctx));32        const resized = self.backing.rawResize(memory, alignment, new_len, ret_addr);33        if (resized) self.recordResize(memory.ptr, memory.len, new_len);34        return resized;35    }3637    fn rawRemap(ctx: *anyopaque, memory: []u8, alignment: std.mem.Alignment, new_len: usize, ret_addr: usize) ?[*]u8 {38        const self: *TracedAllocator = @ptrCast(@alignCast(ctx));39        const ptr = self.backing.rawRemap(memory, alignment, new_len, ret_addr) orelse return null;40        runtime.freeAddress(addressOf(memory.ptr), memory.len, self.name);41        if (new_len != 0) runtime.allocAddress(addressOf(ptr), new_len, self.name);42        return ptr;43    }4445    fn rawFree(ctx: *anyopaque, memory: []u8, alignment: std.mem.Alignment, ret_addr: usize) void {46        const self: *TracedAllocator = @ptrCast(@alignCast(ctx));47        runtime.freeAddress(addressOf(memory.ptr), memory.len, self.name);48        self.backing.rawFree(memory, alignment, ret_addr);49    }5051    fn recordResize(self: *TracedAllocator, ptr: [*]u8, old_len: usize, new_len: usize) void {52        if (old_len == new_len) return;53        runtime.freeAddress(addressOf(ptr), old_len, self.name);54        if (new_len != 0) runtime.allocAddress(addressOf(ptr), new_len, self.name);55    }56};

1pub const TracedAllocator = allocator_mod.TracedAllocator;