2 files changed, 15 insertions, 7 deletions

diff --git a/build.zig b/build.zig
index 03c07d7..49ef164 100644
--- a/build.zig
+++ b/build.zig
@@ -8,8 +8,8 @@ pub fn build(b: *std.Build) void {
         .root_source_file = b.path("src/root.zig"),
         .target = target,
         .optimize = optimize,
-        .link_libc = true,
     });
+    if (!target.result.isWasiLibC()) mod.link_libc = true;
     const lib = b.addLibrary(.{
         .name = "typdown",
         .linkage = .dynamic,
@@ -22,9 +22,9 @@ pub fn build(b: *std.Build) void {
     const example = b.addExecutable(.{
         .name = "example",
         .root_module = b.createModule(.{
-            .link_libc = true,
             .target = target,
             .optimize = optimize,
+            .link_libc = true,
         }),
     });
     example.root_module.addCSourceFile(.{
diff --git a/src/root.zig b/src/root.zig
index 82ac3b1..32f48ef 100644
--- a/src/root.zig
+++ b/src/root.zig
@@ -1,4 +1,5 @@
 const std = @import("std");
+const builtin = @import("builtin");
 const parser = @import("parser.zig");
 pub const Error = parser.Error;
 
@@ -28,6 +29,14 @@ export fn typdown_getErrorString(code: u8) [*:0]const u8 {
     };
 }
 
+var default_alloc: std.mem.Allocator =
+    if (builtin.target.isWasiLibC())
+        std.heap.wasm_allocator
+    else if (builtin.is_test)
+        std.testing.allocator
+    else
+        std.heap.c_allocator;
+
 /// Parse the content.
 /// Code is a pointer to an u8 populated with an error code > 0.
 ///
@@ -36,14 +45,13 @@ export fn typdown_getErrorString(code: u8) [*:0]const u8 {
 /// Use typdown_getErrorString to retrieve the string linked with the error code.
 /// Use parse if you are in Zig.
 export fn typdown_parse(content: [*:0]const u8, code: *u8) ?[*:0]const u8 {
-    const alloc = std.heap.c_allocator;
-    const res = parse(alloc, std.mem.span(content)) catch |err| {
+    const res = parse(default_alloc, std.mem.span(content)) catch |err| {
         code.* = getErrorCode(err);
         return null;
     };
-    defer alloc.free(res);
+    defer default_alloc.free(res);
     code.* = 0;
-    return alloc.dupeZ(u8, res) catch |err| {
+    return default_alloc.dupeZ(u8, res) catch |err| {
         code.* = getErrorCode(err);
         return null;
     };
@@ -76,7 +84,7 @@ fn doTest(content: [*:0]const u8, exp: []const u8, comptime exp_code: u8) !void
         return;
     };
     const res = std.mem.span(raw);
-    defer std.heap.c_allocator.free(res);
+    defer std.testing.allocator.free(res);
 
     expect(code == 0) catch |err| {
         std.debug.print("{}\n", .{code});