From 20db1e02ab5ab9025bf10d0ef6fb125afef572ed Mon Sep 17 00:00:00 2001
From: Anton Lydike <me@antonlydike.de>
Date: Sun, 18 Apr 2021 00:25:53 +0200
Subject: [PATCH] implemented instructions sll, slli, srl, srli, sra, srai

---
 riscemu/CPU.py | 54 ++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 48 insertions(+), 6 deletions(-)

diff --git a/riscemu/CPU.py b/riscemu/CPU.py
index 6486d3e..022f43c 100644
--- a/riscemu/CPU.py
+++ b/riscemu/CPU.py
@@ -111,22 +111,64 @@ class CPU:
         self.mmu.write(addr, 4, int_to_bytes(self.regs.get(rd), 4))
 
     def instruction_sll(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,
+            to_signed(to_unsigned(self.regs.get(src1)) << (self.regs.get(src2) & 0b11111))
+        )
 
     def instruction_slli(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        imm = ins.get_imm(2)
+        self.regs.set(
+            dst,
+            to_signed(to_unsigned(self.regs.get(src1)) << (imm & 0b11111))
+        )
 
     def instruction_srl(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,
+            to_signed(to_unsigned(self.regs.get(src1)) >> (self.regs.get(src2) & 0b11111))
+        )
 
     def instruction_srli(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        imm = ins.get_imm(2)
+        self.regs.set(
+            dst,
+            to_signed(to_unsigned(self.regs.get(src1)) >> (imm & 0b11111))
+        )
 
     def instruction_sra(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) & 0b11111)
+        )
 
     def instruction_srai(self, ins: 'LoadedInstruction'):
-        INS_NOT_IMPLEMENTED(ins)
+        ASSERT_LEN(ins.args, 3)
+        dst = ins.get_reg(0)
+        src1 = ins.get_reg(1)
+        imm = ins.get_imm(2)
+        self.regs.set(
+            dst,
+            self.regs.get(src1) >> (imm & 0b11111)
+        )
 
     def instruction_add(self, ins: 'LoadedInstruction'):
         ASSERT_LEN(ins.args, 3)