editor: fix potential memory leak

Signed-off-by: Stephen Gutekanst <stephen@hexops.com>
This commit is contained in:
Stephen Gutekanst 2023-07-22 18:50:16 -07:00
parent 835748a8c5
commit 095c7795ba
3 changed files with 20 additions and 15 deletions

View file

@ -24,7 +24,7 @@ listen_port: u16 = 1717,
subscribers: std.ArrayListUnmanaged(std.net.Stream) = .{}, subscribers: std.ArrayListUnmanaged(std.net.Stream) = .{},
watch_paths: ?[][]const u8 = null, watch_paths: ?[][]const u8 = null,
mtimes: std.AutoHashMapUnmanaged(std.fs.File.INode, i128) = .{}, mtimes: std.AutoHashMapUnmanaged(std.fs.File.INode, i128) = .{},
@"formated-www/index.html": []const u8 = undefined, www_index_html: []const u8 = undefined,
const Status = union(enum) { const Status = union(enum) {
building, building,
@ -33,7 +33,15 @@ const Status = union(enum) {
compile_error: []const u8, compile_error: []const u8,
}; };
fn deinit(self: *Builder) void {
allocator.free(self.steps);
allocator.free(self.zig_build_args);
if (self.watch_paths) |wp| allocator.free(wp);
}
pub fn run(self: *Builder) !void { pub fn run(self: *Builder) !void {
defer self.deinit();
var child = try self.runZigBuild(.Inherit); var child = try self.runZigBuild(.Inherit);
switch (try child.wait()) { switch (try child.wait()) {
.Exited => |code| { .Exited => |code| {
@ -62,7 +70,7 @@ pub fn run(self: *Builder) !void {
std.os.exit(1); std.os.exit(1);
} }
self.@"formated-www/index.html" = try std.fmt.allocPrint( self.www_index_html = try std.fmt.allocPrint(
allocator, allocator,
@"www/index.html", @"www/index.html",
.{ .{
@ -94,23 +102,20 @@ pub fn run(self: *Builder) !void {
} }
fn runZigBuild(self: Builder, stderr_behavior: std.process.Child.StdIo) !std.process.Child { fn runZigBuild(self: Builder, stderr_behavior: std.process.Child.StdIo) !std.process.Child {
var args_arena = std.heap.ArenaAllocator.init(allocator); var arena = std.heap.ArenaAllocator.init(allocator);
defer args_arena.deinit(); defer arena.deinit();
const arena_allocator = arena.allocator();
const args = try self.buildArgs(args_arena.allocator()); const args = try self.buildArgs(arena_allocator);
var child = std.process.Child.init(args, arena_allocator);
var child = std.process.Child.init(args, allocator);
child.stderr_behavior = stderr_behavior; child.stderr_behavior = stderr_behavior;
try child.spawn(); try child.spawn();
return child; return child;
} }
fn buildArgs(self: Builder, arena: std.mem.Allocator) ![]const []const u8 { fn buildArgs(self: Builder, arena: std.mem.Allocator) ![]const []const u8 {
var argv = std.ArrayList([]const u8).init(arena); var argv = std.ArrayList([]const u8).init(arena);
try argv.ensureTotalCapacity(self.steps.len + self.zig_build_args.len + 6); try argv.ensureTotalCapacity(self.steps.len + self.zig_build_args.len + 6);
argv.appendAssumeCapacity(try arena.dupe(u8, self.zig_path)); argv.appendAssumeCapacity(try arena.dupe(u8, self.zig_path));
argv.appendAssumeCapacity("build"); argv.appendAssumeCapacity("build");
@ -122,7 +127,7 @@ fn buildArgs(self: Builder, arena: std.mem.Allocator) ![]const []const u8 {
argv.appendAssumeCapacity("on"); argv.appendAssumeCapacity("on");
argv.appendAssumeCapacity(try std.fmt.allocPrint(arena, "-Doptimize={s}", .{@tagName(self.optimize)})); argv.appendAssumeCapacity(try std.fmt.allocPrint(arena, "-Doptimize={s}", .{@tagName(self.optimize)}));
if (self.target) |target| { if (self.target) |target| {
argv.appendAssumeCapacity(try std.fmt.allocPrint(arena, "-Dtarget={s}", .{try target.toZigTriple()})); argv.appendAssumeCapacity(try std.fmt.allocPrint(arena, "-Dtarget={s}", .{try target.toZigTriple(arena)}));
} }
for (self.zig_build_args) |arg| { for (self.zig_build_args) |arg| {
@ -140,7 +145,7 @@ fn watch(self: *Builder) void {
const path_handlers = std.ComptimeStringMap(*const fn (*Builder, std.net.Stream) void, .{ const path_handlers = std.ComptimeStringMap(*const fn (*Builder, std.net.Stream) void, .{
.{ "/", struct { .{ "/", struct {
fn h(self: *Builder, stream: std.net.Stream) void { fn h(self: *Builder, stream: std.net.Stream) void {
sendData(stream, .ok, .close, mime_map.get(".html").?, self.@"formated-www/index.html"); sendData(stream, .ok, .close, mime_map.get(".html").?, self.www_index_html);
} }
}.h }, }.h },
.{ "/notify", struct { .{ "/notify", struct {

View file

@ -45,6 +45,7 @@ pub fn main() !void {
if (std.mem.eql(u8, args[arg_i], "build")) { if (std.mem.eql(u8, args[arg_i], "build")) {
arg_i += 1; arg_i += 1;
var builder = Builder{}; var builder = Builder{};
var steps = std.ArrayList([]const u8).init(allocator); var steps = std.ArrayList([]const u8).init(allocator);
var build_args = std.ArrayList([]const u8).init(allocator); var build_args = std.ArrayList([]const u8).init(allocator);
@ -102,7 +103,6 @@ pub fn main() !void {
builder.steps = try steps.toOwnedSlice(); builder.steps = try steps.toOwnedSlice();
builder.zig_build_args = try build_args.toOwnedSlice(); builder.zig_build_args = try build_args.toOwnedSlice();
return builder.run(); return builder.run();
} else if (std.mem.eql(u8, args[arg_i], "help") or std.mem.eql(u8, args[arg_i], "--help") or std.mem.eql(u8, args[arg_i], "-h")) { } else if (std.mem.eql(u8, args[arg_i], "help") or std.mem.eql(u8, args[arg_i], "--help") or std.mem.eql(u8, args[arg_i], "-h")) {
arg_i += 1; arg_i += 1;

View file

@ -27,7 +27,7 @@ pub const Target = enum {
std.meta.stringToEnum(Target, str) orelse return null; std.meta.stringToEnum(Target, str) orelse return null;
} }
pub fn toZigTriple(self: Target) ![]const u8 { pub fn toZigTriple(self: Target, allocator2: std.mem.Allocator) ![]const u8 {
const zig_target = std.zig.CrossTarget{ const zig_target = std.zig.CrossTarget{
.cpu_arch = switch (self) { .cpu_arch = switch (self) {
.@"linux-x86_64", .@"linux-x86_64",
@ -47,6 +47,6 @@ pub const Target = enum {
.wasm32 => .freestanding, .wasm32 => .freestanding,
}, },
}; };
return zig_target.zigTriple(allocator); return zig_target.zigTriple(allocator2);
} }
}; };