From ae409c8f1f97753f6fe373d55428c478ba9b3f6f Mon Sep 17 00:00:00 2001
From: Duncan Ogilvie <mr.exodia.tpodt@gmail.com>
Date: Wed, 25 Dec 2024 00:36:46 +0100
Subject: [PATCH] Redirect reg_write("rip") to write_pc

Closes #13
---
 src/lib.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 7b30c3d..72d281a 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -455,7 +455,13 @@ impl Icicle {
     }
 
     pub fn reg_write(&mut self, name: &str, value: u64) -> PyResult<()> {
-        Ok(self.vm.cpu.write_reg(reg_find(self, name)?.var, value))
+        let var = reg_find(self, name)?.var;
+        if var == self.vm.cpu.arch.reg_pc {
+            self.vm.cpu.write_pc(value);
+        } else {
+            self.vm.cpu.write_reg(var, value);
+        }
+        Ok(())
     }
 
     pub fn reset(&mut self) {