From 2a5b8f53fe7cb7a4261b765020f9342005046b63 Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Sun, 31 Oct 2021 12:47:50 +0000
Subject: [PATCH] Add `memmove` and `memset` support for `wasm`

---
 core/runtime/procs.odin  | 41 +++++++++++++++++++++++++++++++++-------
 src/build_settings.cpp   |  5 +++--
 src/llvm_backend.cpp     | 12 +++++++++---
 src/llvm_backend_opt.cpp |  5 +++++
 4 files changed, 51 insertions(+), 12 deletions(-)

diff --git a/core/runtime/procs.odin b/core/runtime/procs.odin
index d97a3fadf..09ad25379 100644
--- a/core/runtime/procs.odin
+++ b/core/runtime/procs.odin
@@ -1,12 +1,39 @@
 package runtime
 
-memset :: proc "c" (ptr: rawptr, val: i32, len: int) -> rawptr {
-	if ptr != nil && len != 0 {
-		b := byte(val)
-		p := ([^]byte)(ptr)
-		for i in 0..<len {
-			p[i] = b
+when ODIN_ARCH == "wasm32" || ODIN_ARCH == "wasm64" {
+	@(link_name="memset")
+	memset :: proc "c" (ptr: rawptr, val: i32, len: int) -> rawptr {
+		if ptr != nil && len != 0 {
+			b := byte(val)
+			p := ([^]byte)(ptr)
+			for i in 0..<len {
+				p[i] = b
+			}
 		}
+		return ptr
+	}
+	
+	@(link_name="memmove")
+	memmove :: proc "c" (dst, src: rawptr, len: int) -> rawptr {
+		if dst != src {
+			d, s := ([^]byte)(dst), ([^]byte)(src)
+			d_end, s_end := d[len:], s[len:]
+			for i := len-1; i >= 0; i -= 1 {
+				d[i] = s[i]
+			}
+		}
+		return dst
+		
+	}
+} else {
+	memset :: proc "c" (ptr: rawptr, val: i32, len: int) -> rawptr {
+		if ptr != nil && len != 0 {
+			b := byte(val)
+			p := ([^]byte)(ptr)
+			for i in 0..<len {
+				p[i] = b
+			}
+		}
+		return ptr
 	}
-	return ptr
 }
\ No newline at end of file
diff --git a/src/build_settings.cpp b/src/build_settings.cpp
index d817c87f1..b33fabfda 100644
--- a/src/build_settings.cpp
+++ b/src/build_settings.cpp
@@ -909,8 +909,9 @@ void init_build_context(TargetMetrics *cross_target) {
 			break;
 		}
 	} else if (is_arch_wasm()) {
-		gbString link_flags = gb_string_make(heap_allocator(), "--export-all  ");
-		link_flags = gb_string_appendc(link_flags, "--export-table ");
+		gbString link_flags = gb_string_make(heap_allocator(), " ");
+		// link_flags = gb_string_appendc(link_flags, "--export-all ");
+		// link_flags = gb_string_appendc(link_flags, "--export-table ");
 		link_flags = gb_string_appendc(link_flags, "--allow-undefined ");
 		if (bc->metrics.arch == TargetArch_wasm64) {
 			link_flags = gb_string_appendc(link_flags, "-mwas64 ");
diff --git a/src/llvm_backend.cpp b/src/llvm_backend.cpp
index 27914efb2..d21ff8e5a 100644
--- a/src/llvm_backend.cpp
+++ b/src/llvm_backend.cpp
@@ -771,6 +771,8 @@ lbProcedure *lb_create_main_procedure(lbModule *m, lbProcedure *startup_runtime)
 	Type *results = alloc_type_tuple();
 
 	Type *t_ptr_cstring = alloc_type_pointer(t_cstring);
+	
+	bool call_cleanup = true;
 
 	bool has_args = false;
 	bool is_dll_main = false;
@@ -782,10 +784,12 @@ lbProcedure *lb_create_main_procedure(lbModule *m, lbProcedure *startup_runtime)
 		params->Tuple.variables[0] = alloc_entity_param(nullptr, make_token_ident("hinstDLL"),   t_rawptr, false, true);
 		params->Tuple.variables[1] = alloc_entity_param(nullptr, make_token_ident("fdwReason"),  t_u32,    false, true);
 		params->Tuple.variables[2] = alloc_entity_param(nullptr, make_token_ident("lpReserved"), t_rawptr, false, true);
+		call_cleanup = false;
 	} else if (build_context.metrics.os == TargetOs_windows && build_context.metrics.arch == TargetArch_386) {
 		name = str_lit("mainCRTStartup");
 	} else if (build_context.metrics.os == TargetOs_wasi) {
 		name = str_lit("_start");
+		call_cleanup = false;
 	} else {
 		has_args = true;
 		slice_init(&params->Tuple.variables, permanent_allocator(), 2);
@@ -876,8 +880,10 @@ lbProcedure *lb_create_main_procedure(lbModule *m, lbProcedure *startup_runtime)
 	}
 
 	
-	lbValue cleanup_runtime_value = lb_find_runtime_value(m, str_lit("_cleanup_runtime"));
-	lb_emit_call(p, cleanup_runtime_value, {}, ProcInlining_none, false);
+	if (call_cleanup) {
+		lbValue cleanup_runtime_value = lb_find_runtime_value(m, str_lit("_cleanup_runtime"));
+		lb_emit_call(p, cleanup_runtime_value, {}, ProcInlining_none, false);
+	}
 	
 
 	if (is_dll_main) {
@@ -890,7 +896,7 @@ lbProcedure *lb_create_main_procedure(lbModule *m, lbProcedure *startup_runtime)
 	
 	
 	if (build_context.metrics.os == TargetOs_wasi) {
-		LLVMSetLinkage(p->value, LLVMDLLExportLinkage);	
+		LLVMSetLinkage(p->value, LLVMDLLExportLinkage);
 	} else {
 		LLVMSetLinkage(p->value, LLVMExternalLinkage);
 	}
diff --git a/src/llvm_backend_opt.cpp b/src/llvm_backend_opt.cpp
index 25e290d70..8ddd3360d 100644
--- a/src/llvm_backend_opt.cpp
+++ b/src/llvm_backend_opt.cpp
@@ -383,6 +383,11 @@ void lb_run_remove_unused_function_pass(LLVMModuleRef mod) {
 				continue;
 			}
 			
+			if (name == "memset" ||
+			    name == "memmove" ||
+			    name == "memcpy") {
+				continue;
+			}
 			
 			LLVMLinkage linkage = LLVMGetLinkage(curr_func);