From dd79c11b3b92069edcc2c13e5659c225f4c238d7 Mon Sep 17 00:00:00 2001
From: Anton Lydike <me@antonlydike.de>
Date: Sat, 17 Apr 2021 23:53:47 +0200
Subject: [PATCH] implemented instructions and, or, slt, sltu

---
 riscemu/CPU.py        | 36 ++++++++++++++++++++++++++++++++----
 riscemu/arithmetic.py |  0
 2 files changed, 32 insertions(+), 4 deletions(-)
 create mode 100644 riscemu/arithmetic.py

diff --git a/riscemu/CPU.py b/riscemu/CPU.py
index d0ab25e..311cd91 100644
--- a/riscemu/CPU.py
+++ b/riscemu/CPU.py
@@ -172,25 +172,53 @@ class CPU:
         INS_NOT_IMPLEMENTED(ins)
 
     def instruction_or(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        src2 = ins.get_reg(2)
+        self.regs.set(
+            dst,
+            self.regs.get(src1) | self.regs.get(src2)
+        )
 
     def instruction_ori(self, ins: 'LoadedInstruction'):
         INS_NOT_IMPLEMENTED(ins)
 
     def instruction_and(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        src2 = ins.get_reg(2)
+        self.regs.set(
+            dst,
+            self.regs.get(src1) & self.regs.get(src2)
+        )
 
     def instruction_andi(self, ins: 'LoadedInstruction'):
         INS_NOT_IMPLEMENTED(ins)
 
     def instruction_slt(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        src2 = ins.get_reg(2)
+        self.regs.set(
+            dst,
+            int(self.regs.get(src1) < self.regs.get(src2))
+        )
 
     def instruction_slti(self, ins: 'LoadedInstruction'):
         INS_NOT_IMPLEMENTED(ins)
 
     def instruction_sltu(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        src2 = ins.get_reg(2)
+        self.regs.set(
+            dst,
+            int(to_unsigned(self.regs.get(src1)) < to_unsigned(self.regs.get(src2)))
+        )
 
     def instruction_sltiu(self, ins: 'LoadedInstruction'):
         INS_NOT_IMPLEMENTED(ins)
diff --git a/riscemu/arithmetic.py b/riscemu/arithmetic.py
new file mode 100644
index 0000000..e69de29