Fix panic when size 0 passed to malloc

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/runtime_wasi/malloc_test.go b/tests/runtime_wasi/malloc_test.go
index 06b197f13..eb655b424 100644
--- a/tests/runtime_wasi/malloc_test.go
+++ b/tests/runtime_wasi/malloc_test.go
@@ -125,3 +125,32 @@ func TestMallocFree(t *testing.T) {
 		})
 	}
 }
+
+func TestMallocEmpty(t *testing.T) {
+	ptr := libc_malloc(0)
+	if ptr != nil {
+		t.Errorf("expected nil pointer, got %p", ptr)
+	}
+}
+
+func TestCallocEmpty(t *testing.T) {
+	ptr := libc_calloc(0, 1)
+	if ptr != nil {
+		t.Errorf("expected nil pointer, got %p", ptr)
+	}
+	ptr = libc_calloc(1, 0)
+	if ptr != nil {
+		t.Errorf("expected nil pointer, got %p", ptr)
+	}
+}
+
+func TestReallocEmpty(t *testing.T) {
+	ptr := libc_malloc(1)
+	if ptr == nil {
+		t.Error("expected pointer but was nil")
+	}
+	ptr = libc_realloc(ptr, 0)
+	if ptr != nil {
+		t.Errorf("expected nil pointer, got %p", ptr)
+	}
+}