1 files changed, 99 insertions, 0 deletions
diff --git a/js/src/jit-test/tests/atomics/nursery-non-shared-moved-gc.js b/js/src/jit-test/tests/atomics/nursery-non-shared-moved-gc.js
new file mode 100644
index 0000000000..472076ce63
--- /dev/null
+++ b/js/src/jit-test/tests/atomics/nursery-non-shared-moved-gc.js
@@ -0,0 +1,99 @@
+// Small (non-shared) TypedArrays can use inline storage and can also be nursery
+// allocated. Ensure Atomics functions use correct rooting and don't store the
+// TypedArray's data pointer on the stack across a GC.
+const INLINE_MEMORY_LENGTH = 1;
+
+function gcValue(value) {
+  return {
+    valueOf() {
+      minorgc();
+      return value;
+    }
+  };
+};
+
+// Atomics.store
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+
+  Atomics.store(ta, 0, gcValue(1));
+
+  assertEq(ta[0], 1);
+}
+
+// Atomics.compareExchange
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 1;
+
+  let val = Atomics.compareExchange(ta, 0, 1, gcValue(2));
+
+  assertEq(val, 1);
+  assertEq(ta[0], 2);
+}
+
+// Atomics.exchange
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 1;
+
+  let val = Atomics.exchange(ta, 0, gcValue(2));
+
+  assertEq(val, 1);
+  assertEq(ta[0], 2);
+}
+
+// Atomics.add
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 1;
+
+  let val = Atomics.add(ta, 0, gcValue(2));
+
+  assertEq(val, 1);
+  assertEq(ta[0], 3);
+}
+
+// Atomics.sub
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 3;
+
+  let val = Atomics.sub(ta, 0, gcValue(2));
+
+  assertEq(val, 3);
+  assertEq(ta[0], 1);
+}
+
+// Atomics.and
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 3;
+
+  let val = Atomics.and(ta, 0, gcValue(1));
+
+  assertEq(val, 3);
+  assertEq(ta[0], 1);
+}
+
+// Atomics.or
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 2;
+
+  let val = Atomics.or(ta, 0, gcValue(1));
+
+  assertEq(val, 2);
+  assertEq(ta[0], 3);
+}
+
+// Atomics.xor
+{
+  let ta = new Int32Array(INLINE_MEMORY_LENGTH);
+  ta[0] = 3;
+
+  let val = Atomics.xor(ta, 0, gcValue(1));
+
+  assertEq(val, 3);
+  assertEq(ta[0], 2);
+}