Add homomorphic host redirection abstraction for vm->host calls.

wasmerio · Sep 17, 2019 · 9ab3da3 · 9ab3da3
1 parent 574e4c4
commit 9ab3da3
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/lib/singlepass-backend/src/codegen_x64.rs b/lib/singlepass-backend/src/codegen_x64.rs
@@ -644,7 +644,7 @@ impl ModuleCodeGenerator<X64FunctionCode, X64ExecutionContext, CodegenError>
             ),
             Location::GPR(GPR::RAX),
         );
-        a.emit_jmp_location(Location::GPR(GPR::RAX));
+        a.emit_homomorphic_host_redirection(GPR::RAX);
 
         self.func_import_count += 1;
 

diff --git a/lib/singlepass-backend/src/emitter_x64.rs b/lib/singlepass-backend/src/emitter_x64.rs
@@ -168,6 +168,8 @@ pub trait Emitter {
     fn emit_call_location(&mut self, loc: Location);
 
     fn emit_bkpt(&mut self);
+
+    fn emit_homomorphic_host_redirection(&mut self, target: GPR);
 }
 
 fn _dummy(a: &mut Assembler) {
@@ -940,4 +942,8 @@ impl Emitter for Assembler {
     fn emit_bkpt(&mut self) {
         dynasm!(self ; int 0x3);
     }
+
+    fn emit_homomorphic_host_redirection(&mut self, target: GPR) {
+        self.emit_jmp_location(Location::GPR(target));
+    }
 }
diff --git a/lib/singlepass-backend/src/translator_aarch64.rs b/lib/singlepass-backend/src/translator_aarch64.rs
@@ -977,4 +977,56 @@ impl Emitter for Assembler {
     fn emit_bkpt(&mut self) {
         dynasm!(self ; brk 1)
     }
+
+    fn emit_homomorphic_host_redirection(&mut self, target: GPR) {
+        let target = map_gpr(target);
+        dynasm!(
+            self
+            ; sub sp, sp, 96
+            ; str x19, [sp, 0]
+            ; str x20, [sp, 8]
+            ; str x21, [sp, 16]
+            ; str x22, [sp, 24]
+            ; str x23, [sp, 32]
+            ; str x24, [sp, 40]
+            ; str x25, [sp, 48]
+            ; str x26, [sp, 56]
+            ; str x27, [sp, 64]
+            ; str x28, [sp, 72] // x_rsp
+            ; str x30, [sp, 80] // LR
+            ; adr x30, >after
+
+            // Put parameters in correct order
+            ; sub sp, sp, 64
+            ; str X(map_gpr(GPR::RDI).x()), [sp, 0]
+            ; str X(map_gpr(GPR::RSI).x()), [sp, 8]
+            ; str X(map_gpr(GPR::RDX).x()), [sp, 16]
+            ; str X(map_gpr(GPR::RCX).x()), [sp, 24]
+            ; str X(map_gpr(GPR::R8).x()), [sp, 32]
+            ; str X(map_gpr(GPR::R9).x()), [sp, 40]
+            ; ldr x0, [sp, 0]
+            ; ldr x1, [sp, 8]
+            ; ldr x2, [sp, 16]
+            ; ldr x3, [sp, 24]
+            ; ldr x4, [sp, 32]
+            ; ldr x5, [sp, 40]
+            ; add sp, sp, 64
+
+            ; br X(target.x())
+            ; after:
+            ; ldr x19, [sp, 0]
+            ; ldr x20, [sp, 8]
+            ; ldr x21, [sp, 16]
+            ; ldr x22, [sp, 24]
+            ; ldr x23, [sp, 32]
+            ; ldr x24, [sp, 40]
+            ; ldr x25, [sp, 48]
+            ; ldr x26, [sp, 56]
+            ; ldr x27, [sp, 64]
+            ; ldr x28, [sp, 72] // x_rsp
+            ; ldr x30, [sp, 80] // LR
+            ; add sp, sp, 96
+            ; br x30
+        );
+    }
 }