From b64b3a8351cae78c4581a27ce8e9cf0a015cd61f Mon Sep 17 00:00:00 2001
From: Alex Shelkovnykov <a@ashelkov.com>
Date: Tue, 3 Oct 2023 17:56:39 -0600
Subject: [PATCH 1/2] Merge w/ #100

---
 rust/ares/Cargo.toml | 2 ++
 rust/ares/src/mem.rs | 9 +++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/rust/ares/Cargo.toml b/rust/ares/Cargo.toml
index 74dd12e0..6f851e5d 100644
--- a/rust/ares/Cargo.toml
+++ b/rust/ares/Cargo.toml
@@ -23,6 +23,8 @@ criterion = "0.4"
 static_assertions = "1.1.0"
 ibig = { path = "../ibig-rs" }
 assert_no_alloc = "1.1.2"
+# use this when debugging requires allocation (e.g. eprintln)
+# assert_no_alloc = {version="1.1.2", features=["warn_debug"]}
 
 [build-dependencies]
 cc = "1.0.79"
diff --git a/rust/ares/src/mem.rs b/rust/ares/src/mem.rs
index c8d078d6..3919c667 100644
--- a/rust/ares/src/mem.rs
+++ b/rust/ares/src/mem.rs
@@ -70,9 +70,9 @@ impl NockStack {
         let stack_pointer = frame_pointer;
         let alloc_pointer = unsafe { start.add(size) } as *mut u64;
         unsafe {
-            *frame_pointer = ptr::null::<u64>() as u64; // "frame pointer" from "previous" frame
+            *frame_pointer.sub(FRAME) = ptr::null::<u64>() as u64; // "frame pointer" from "previous" frame
             *frame_pointer.sub(STACK) = ptr::null::<u64>() as u64; // "stack pointer" from "previous" frame
-            *frame_pointer.sub(ALLOC) = start as u64; // "alloc pointer" from "previous" frame
+            *frame_pointer.sub(ALLOC) = ptr::null::<u64>() as u64; // "alloc pointer" from "previous" frame
         };
         NockStack {
             start,
@@ -805,6 +805,11 @@ unsafe fn senior_pointer_first(
     };
 
     loop {
+        if low_pointer.is_null() || high_pointer.is_null() {
+            // we found the top of the stack
+            break (a, b); // both are in the bottom frame, pick arbitrarily
+        }
+
         match (
             a < high_pointer && a >= low_pointer,
             b < high_pointer && b >= low_pointer,

From 85afbb49eb1ebf549474db3cf5abda279703190f Mon Sep 17 00:00:00 2001
From: Alex Shelkovnykov <a@ashelkov.com>
Date: Wed, 4 Oct 2023 12:03:39 -0600
Subject: [PATCH 2/2] Fix NockStack init; check outside stack for ptr seniority

---
 rust/ares/src/mem.rs | 34 ++++++++++++++++------------------
 1 file changed, 16 insertions(+), 18 deletions(-)

diff --git a/rust/ares/src/mem.rs b/rust/ares/src/mem.rs
index 258bf53e..1ae78024 100644
--- a/rust/ares/src/mem.rs
+++ b/rust/ares/src/mem.rs
@@ -811,8 +811,9 @@ unsafe fn senior_pointer_first(
 
     loop {
         if low_pointer.is_null() || high_pointer.is_null() {
-            // we found the top of the stack
-            break (a, b); // both are in the bottom frame, pick arbitrarily
+            // we found the bottom of the stack; check entirety of the stack
+            low_pointer = stack.start;
+            high_pointer = stack.start.add(stack.size);
         }
 
         match (
@@ -821,11 +822,7 @@ unsafe fn senior_pointer_first(
         ) {
             (true, true) => {
                 // both pointers are in the same frame, pick arbitrarily (lower in mem)
-                if a < b {
-                    break (a, b);
-                } else {
-                    break (b, a);
-                };
+                break lower_pointer_first(a, b);
             }
             (true, false) => break (b, a), // a is in the frame, b is not, so b is senior
             (false, true) => break (a, b), // b is in the frame, a is not, so a is senior
@@ -838,30 +835,23 @@ unsafe fn senior_pointer_first(
                     alloc_pointer = *(frame_pointer.sub(ALLOC + 1)) as *const u64;
                     frame_pointer = *(frame_pointer.sub(FRAME + 1)) as *const u64;
 
+                    // both pointers are in the PMA, pick arbitrarily (lower in mem)
                     if frame_pointer.is_null() {
-                        if a < b {
-                            break (a, b);
-                        } else {
-                            break (b, a);
-                        }
+                        break lower_pointer_first(a, b);
                     };
 
                     // previous allocation pointer
                     high_pointer = alloc_pointer;
                     // "previous previous" stack pointer. this is the other boundary of the previous allocation arena
                     low_pointer = *(frame_pointer.add(STACK)) as *const u64;
-                    continue;
                 } else if stack_pointer > alloc_pointer {
                     stack_pointer = *(frame_pointer.add(STACK)) as *const u64;
                     alloc_pointer = *(frame_pointer.add(ALLOC)) as *const u64;
                     frame_pointer = *(frame_pointer.add(FRAME)) as *const u64;
 
+                    // both pointers are in the PMA, pick arbitrarily (lower in mem)
                     if frame_pointer.is_null() {
-                        if a < b {
-                            break (a, b);
-                        } else {
-                            break (b, a);
-                        }
+                        break lower_pointer_first(a, b);
                     };
 
                     // previous allocation pointer
@@ -876,6 +866,14 @@ unsafe fn senior_pointer_first(
     }
 }
 
+fn lower_pointer_first(a: *const u64, b: *const u64) -> (*const u64, *const u64) {
+    if a < b {
+        (a, b)
+    } else {
+        (b, a)
+    }
+}
+
 impl NounAllocator for NockStack {
     unsafe fn alloc_indirect(&mut self, words: usize) -> *mut u64 {
         self.indirect_alloc(words)