1 files changed, 142 insertions, 0 deletions
diff --git a/js/src/jit-test/tests/wasm/bigint/stubs.js b/js/src/jit-test/tests/wasm/bigint/stubs.js
new file mode 100644
index 0000000000..89f7d78413
--- /dev/null
+++ b/js/src/jit-test/tests/wasm/bigint/stubs.js
@@ -0,0 +1,142 @@
+// This is similar to the corresponding JIT exit stub test in the
+// wasm/import-export.js file, but it also tests BigInt cases.
+(function testImportJitExit() {
+  let options = getJitCompilerOptions();
+  if (!options["baseline.enable"]) return;
+
+  let baselineTrigger = options["baseline.warmup.trigger"];
+
+  let valueToConvert = 0;
+  function ffi(n) {
+    if (n == 1337n) {
+      return BigInt(valueToConvert);
+    }
+    return 42n;
+  }
+
+  // This case is intended to test that the I64 to BigInt argument
+  // filling for the JIT exit stub does not interfere with the other
+  // argument registers. It's important that the BigInt argument comes
+  // first (to test that it doesn't clobber the subsequent ones).
+  function ffi2(n, x1, x2, x3, x4, x5, x6) {
+    return (
+      42n +
+      BigInt(x1) +
+      BigInt(x2) +
+      BigInt(x3) +
+      BigInt(x4) +
+      BigInt(x5) +
+      BigInt(x6)
+    );
+  }
+
+  // This case is for testing issues with potential GC.
+  function ffi3(n1, n2, n3, n4) {
+    return n3;
+  }
+
+  // Baseline compile ffis.
+  for (let i = baselineTrigger + 1; i-- > 0; ) {
+    ffi(BigInt(i));
+    ffi2(BigInt(i), i, i, i, i, i, i);
+    ffi3(BigInt(i), BigInt(i), BigInt(i), BigInt(i));
+  }
+
+  let imports = {
+    a: {
+      ffi,
+      ffi2,
+      ffi3,
+    },
+  };
+
+  ex = wasmEvalText(
+    `(module
+        (import "a" "ffi" (func $ffi (param i64) (result i64)))
+        (import "a" "ffi2" (func $ffi2
+          (param i64 i32 i32 i32 i32 i32 i32)
+          (result i64)))
+        (import "a" "ffi3" (func $ffi3 (param i64 i64 i64 i64) (result i64)))
+
+        (func (export "callffi") (param i64) (result i64)
+         local.get 0
+         call $ffi
+        )
+
+        (func (export "callffi2") (param i32 i64) (result i64)
+         local.get 1
+         local.get 0
+         local.get 0
+         local.get 0
+         local.get 0
+         local.get 0
+         local.get 0
+         call $ffi2
+        )
+
+        (func (export "callffi3") (param i64 i64 i64 i64) (result i64)
+          local.get 0
+          local.get 1
+          local.get 2
+          local.get 3
+          call $ffi3
+        )
+    )`,
+    imports
+  ).exports;
+
+  // Enable the jit exit for each JS callee.
+  assertEq(ex.callffi(0n), 42n);
+  assertEq(ex.callffi2(2, 0n), 54n);
+  assertEq(ex.callffi3(0n, 1n, 2n, 3n), 2n);
+
+  // Test the jit exit under normal conditions.
+  assertEq(ex.callffi(0n), 42n);
+  assertEq(ex.callffi(1337n), 0n);
+  assertEq(ex.callffi2(2, 0n), 54n);
+  assertEq(ex.callffi3(0n, 1n, 2n, 3n), 2n);
+
+  // Test the jit exit with GC stress in order to ensure that
+  // any trigger of GC in the stub do not cause errors.
+  if (this.gczeal) {
+    this.gczeal(2, 1); // Collect on every allocation
+  }
+  for (let i = 0; i < 1000; i++) {
+    assertEq(ex.callffi3(0n, 1n, 2n, 3n), 2n);
+  }
+})();
+
+// Test JIT entry stub
+(function testJitEntry() {
+  let options = getJitCompilerOptions();
+  if (!options["baseline.enable"]) return;
+
+  let baselineTrigger = options["baseline.warmup.trigger"];
+
+  i = wasmEvalText(
+    `(module
+      (func (export "foo") (param i64) (result i64)
+       local.get 0
+       i64.const 1
+       i64.add
+       return
+      )
+  )`,
+    {}
+  ).exports;
+
+  function caller(n) {
+    return i.foo(42n);
+  }
+
+  // Baseline compile ffis.
+  for (let i = baselineTrigger + 1; i-- > 0; ) {
+    caller(i);
+  }
+
+  // Enable the jit entry.
+  assertEq(caller(0), 43n);
+
+  // Test the jit exit under normal conditions.
+  assertEq(caller(0), 43n);
+})();