compact implementation; runs on msp430

1 files changed, 17 insertions, 10 deletions

diff --git a/state.cpp b/state.cpp
index c371101..1510352 100644
--- a/state.cpp
+++ b/state.cpp
@@ -31,19 +31,26 @@ void State::compiling(bool yes)
     dict.write(Dictionary::Compiling, yes);
 }
 
-void State::execute(Addr addr)
+State::Error State::execute(Addr addr)
 {
-    if (addr < CoreWords::WordCount) {
-        CoreWords::run(addr, *this);
-    } else {
-        pushr(0);
-        ip = addr - sizeof(Cell);
+    auto stat = setjmp(jmpbuf);
+    if (!stat) {
+        if (addr < CoreWords::WordCount) {
+            CoreWords::run(addr, *this);
+        } else {
+            pushr(0);
+            ip = addr - sizeof(Cell);
 
-        do {
-            ip += sizeof(Cell);
-            CoreWords::run(dict.read(ip), *this);
-        } while (ip);
+            do {
+                ip += sizeof(Cell);
+                CoreWords::run(dict.read(ip), *this);
+            } while (ip);
+        }
+    } else {
+        return static_cast<State::Error>(stat);
     }
+
+    return State::Error::none;
 }
 
 std::size_t State::size() const noexcept