1 files changed, 11 insertions, 1 deletions

diff --git a/build.zig b/build.zig
index c70f52b2..f0675a70 100644
--- a/build.zig
+++ b/build.zig
@@ -390,7 +390,17 @@ pub fn build(b: *std.Build) void {
                     .root_module = b.createModule(.{ .target = target, .optimize = c_optimize, .link_libc = !is_web }),
                     .linkage = .static,
                 });
-                const ts_cflags = [_][]const u8{"-fno-sanitize=function"};
+                // Every allocation a grammar's external scanner makes goes to the
+                // runtime's allocator, which syntax.zig points at a Zig one:
+                // TREE_SITTER_REUSE_ALLOCATOR for the scanners that use
+                // ts_malloc, and src/tree_sitter_heap.h for the ones that call
+                // malloc themselves.
+                const ts_cflags = [_][]const u8{
+                    "-fno-sanitize=function",
+                    "-DTREE_SITTER_REUSE_ALLOCATOR",
+                    "-include",
+                    b.path("src/tree_sitter_heap.h").getPath(b),
+                };
                 lib.root_module.addCSourceFile(.{ .file = dep.path(g.src ++ "/parser.c"), .flags = &ts_cflags });
                 if (g.scanner) lib.root_module.addCSourceFile(.{ .file = dep.path(g.src ++ "/scanner.c"), .flags = &ts_cflags });
                 lib.root_module.addIncludePath(dep.path(g.src));