From 69234aac2d29882e221f734deaf0950c896fb96b Mon Sep 17 00:00:00 2001
From: ElectronixTM <etdiary1@gmail.com>
Date: Wed, 2 Apr 2025 00:15:21 +0300
Subject: [PATCH] =?UTF-8?q?feat:=20=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8?=
 =?UTF-8?q?=D0=BB=20=D1=81=D1=82=D0=B0=D1=82=D1=83=D1=81=D1=8B=20=D0=B2?=
 =?UTF-8?q?=D0=B8=D1=80=D1=82=D1=83=D0=B0=D0=BB=D1=8C=D0=BD=D0=BE=D0=B9=20?=
 =?UTF-8?q?=D0=BC=D0=B0=D1=88=D0=B8=D0=BD=D0=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/vm.py | 29 ++++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/src/vm.py b/src/vm.py
index f46a060..40f43dd 100644
--- a/src/vm.py
+++ b/src/vm.py
@@ -25,6 +25,11 @@ class VMCC(IntFlag):
     NEGATIVE = 1 << 1
     ZERO     = 1 << 0
 
+class VMStatus(Enum):
+    RUNNING = auto()
+    INITED = auto()
+    FINISHED = auto()
+
 @dataclass
 class Breakpoint(Exception):
     address: int
@@ -48,15 +53,12 @@ class VM:
     registers: list[c_int32]
     breakpoints: set[int]
     _vm_flags: VMFlags
+    status: VMStatus
 
     def __init__(self, mem: bytearray):
-        self.mem: bytearray = mem
-        self.cc: VMCC = VMCC(0)
-        self.pc: c_uint32 = c_uint32(0)
-        self.registers: list[c_int32] = [c_int32(0) for _ in range(256)]
-        self.breakpoints: set[int] = set()
-        self._vm_flags: VMFlags = VMFlags(0)
+        self._initial_mem = mem.copy()
         self.__init_callbacks__()
+        self.reset()
 
     def __init_callbacks__(self):
         VM.instr_callbacks = {
@@ -119,16 +121,27 @@ class VM:
                     self._branch_indexed_callback
                 }
 
+    def reset(self):
+        self.mem: bytearray = self._initial_mem.copy()
+        self.cc: VMCC = VMCC(0)
+        self.pc: c_uint32 = c_uint32(0)
+        self.registers: list[c_int32] = [c_int32(0) for _ in range(256)]
+        self.breakpoints: set[int] = set()
+        self._vm_flags: VMFlags = VMFlags(0)
+        self.status = VMStatus.INITED
+
     def step(self) -> None:
         """
         Make one step (only step into)
         """
         if self._to_raw_bytes_offset(self.pc) > len(self.mem) - WORD_SIZE:
+            self.status = VMStatus.FINISHED
             raise VMException(
                     VMExceptionType.END_OF_MEM,
                     self.pc.value,
                     "couldn't perform step because end of memory occured"
                     )
+        self.status = VMStatus.RUNNING
         opcode, *_ = instr = self._fetch_instr()
         opdesc = self._get_opcode_desc(opcode)
         args: tuple[int, ...] = self._parse_instr_fields(bytes(instr))
@@ -148,14 +161,16 @@ class VM:
             if self.pc.value in self.breakpoints:
                 raise Breakpoint(self.pc.value)
             self.step()
+        self.status = VMStatus.FINISHED
 
     def run(self) -> None:
         """
         Run from very beginning
         """
-        self.pc = c_uint32(0)
+        self.reset()
         while (self._to_raw_bytes_offset(self.pc.value) < len(self.mem)):
             self.continue_()
+        self.status = VMStatus.FINISHED
 
     def _fetch_instr(self) -> bytearray:
         """