feat(sandboxing): -Dmemory_limit

build.zig

@@ -42,6 +42,7 @@ const BuzzGCOptions = struct {
     initial_gc: usize,
     next_gc_ratio: usize,
     next_full_gc_ratio: usize,
+    memory_limit: ?usize,
 
     pub fn step(self: BuzzGCOptions, options: *Build.Step.Options) void {
         options.addOption(@TypeOf(self.debug), "gc_debug", self.debug);
@@ -51,6 +52,7 @@ const BuzzGCOptions = struct {
         options.addOption(@TypeOf(self.initial_gc), "initial_gc", self.initial_gc);
         options.addOption(@TypeOf(self.next_gc_ratio), "next_gc_ratio", self.next_gc_ratio);
         options.addOption(@TypeOf(self.next_full_gc_ratio), "next_full_gc_ratio", self.next_full_gc_ratio);
+        options.addOption(@TypeOf(self.memory_limit), "memory_limit", self.memory_limit);
     }
 };
 
@@ -211,6 +213,11 @@ pub fn build(b: *Build) !void {
                 "next_full_gc_ratio",
                 "Ratio applied to get the next full GC threshold",
             ) orelse 4,
+            .memory_limit = b.option(
+                usize,
+                "memory_limit",
+                "Memory limit",
+            ) orelse null,
         },
         .jit = .{
             .debug = b.option(

src/Codegen.zig

@@ -21,6 +21,7 @@ pub const Error = error{
     CantCompile,
     UnwrappedNull,
     OutOfBound,
+    ReachedMaximumMemoryUsage,
 } || std.mem.Allocator.Error || std.fmt.BufPrintError;
 
 pub const Frame = struct {

src/Jit.zig

@@ -11,7 +11,12 @@ const ZigType = @import("zigtypes.zig").Type;
 const ExternApi = @import("jit_extern_api.zig").ExternApi;
 const api = @import("lib/buzz_api.zig");
 
-pub const Error = error{ CantCompile, UnwrappedNull, OutOfBound } || std.mem.Allocator.Error || std.fmt.BufPrintError;
+pub const Error = error{
+    CantCompile,
+    UnwrappedNull,
+    OutOfBound,
+    ReachedMaximumMemoryUsage,
+} || std.mem.Allocator.Error || std.fmt.BufPrintError;
 
 const OptJump = struct {
     current_insn: std.ArrayList(m.MIR_insn_t),

src/Parser.zig

@@ -193,6 +193,7 @@ pub const Error = error{
     CantCompile,
     UnwrappedNull,
     OutOfBound,
+    ReachedMaximumMemoryUsage,
 } || std.mem.Allocator.Error || std.fmt.BufPrintError || CompileError;
 
 const ParseFn = *const fn (*Self, bool) Error!Ast.Node.Index;
@@ -708,6 +709,10 @@ pub fn parse(self: *Self, source: []const u8, file_name: []const u8) !?Ast {
             }
         } else {
             if (self.declaration() catch |err| {
+                if (err == error.ReachedMaximumMemoryUsage) {
+                    return err;
+                }
+
                 if (BuildOptions.debug) {
                     std.debug.print("Parsing failed with error {}\n", .{err});
                 }

src/main.zig

@@ -264,9 +264,10 @@ pub fn main() !void {
 
     if (flavor == .Repl) {
         repl(allocator) catch |err| {
-            if (BuildOptions.debug) {
-                std.debug.print("Failed with error {}\n", .{err});
-            }
+            std.io.getStdErr().writer().print(
+                "Failed with error {s}\n",
+                .{@errorName(err)},
+            ) catch unreachable;
 
             std.os.exit(1);
         };
@@ -277,9 +278,10 @@ pub fn main() !void {
             positionals.items[1..],
             flavor,
         ) catch |err| {
-            if (BuildOptions.debug) {
-                std.debug.print("Failed with error {}\n", .{err});
-            }
+            std.io.getStdErr().writer().print(
+                "Failed with error {s}\n",
+                .{@errorName(err)},
+            ) catch unreachable;
 
             std.os.exit(1);
         };

src/memory.zig

@@ -215,6 +215,10 @@ pub const GarbageCollector = struct {
             try self.collectGarbage();
         }
 
+        if (BuildOptions.memory_limit != null and self.bytes_allocated > BuildOptions.memory_limit.?) {
+            return error.ReachedMaximumMemoryUsage;
+        }
+
         const allocated = try self.allocator.create(T);
 
         if (BuildOptions.gc_debug) {

src/obj.zig

@@ -297,7 +297,7 @@ pub const Obj = struct {
         }
     }
 
-    pub fn typeOf(self: *Self, gc: *GarbageCollector) error{ OutOfMemory, NoSpaceLeft }!*ObjTypeDef {
+    pub fn typeOf(self: *Self, gc: *GarbageCollector) error{ OutOfMemory, NoSpaceLeft, ReachedMaximumMemoryUsage }!*ObjTypeDef {
         return switch (self.obj_type) {
             .String => try gc.type_registry.getTypeDef(.{ .def_type = .String }),
             .Pattern => try gc.type_registry.getTypeDef(.{ .def_type = .Pattern }),

src/vm.zig

@@ -329,6 +329,7 @@ pub const VM = struct {
         NotInFiber,
         FiberOver,
         BadNumber,
+        ReachedMaximumMemoryUsage,
         Custom, // TODO: remove when user can use this set directly in buzz code
     } || Allocator.Error || std.fmt.BufPrintError;
 

tests/bench/001-btree.buzz

@@ -57,7 +57,7 @@ fun btree(int N) > void {
 }
 
 fun main([str] args) > void !> any {
-    int N = if (args.len() > 0) parseInt(args[0]) ?? 0 else 0;
+    int N = if (args.len() > 0) parseInt(args[0]) ?? 3 else 3;
 
     btree(N);
 }