From ca28f79c5ffc6ed26cf732d993b9a296fa8cf108 Mon Sep 17 00:00:00 2001
From: Luke Kenneth Casson Leighton <lkcl@lkcl.net>
Date: Fri, 30 Dec 2022 13:26:45 +0000
Subject: [PATCH] add misaligned mem test

---
 src/openpower/decoder/isa/mem.py      | 22 +++++++++++++++++++---
 src/openpower/decoder/isa/test_mem.py |  2 +-
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/src/openpower/decoder/isa/mem.py b/src/openpower/decoder/isa/mem.py
index 483f74e4..23e2c07e 100644
--- a/src/openpower/decoder/isa/mem.py
+++ b/src/openpower/decoder/isa/mem.py
@@ -51,12 +51,13 @@ def process_mem(initial_mem, row_bytes=8):
 
 class Mem:
 
-    def __init__(self, row_bytes=8, initial_mem=None):
+    def __init__(self, row_bytes=8, initial_mem=None, misaligned_ok=False):
         self.mem = {}
         self.bytes_per_word = row_bytes
         self.word_log2 = math.ceil(math.log2(row_bytes))
         self.last_ld_addr = None
         self.last_st_addr = None
+        self.misaligned_ok = misaligned_ok
         log("Sim-Mem", initial_mem, self.bytes_per_word, self.word_log2)
         if not initial_mem:
             return
@@ -108,9 +109,8 @@ class Mem:
         log("Read 0x%x from addr 0x%x" % (val, ldaddr))
         return val
 
-    def st(self, addr, v, width=8, swap=True):
+    def _st(self, addr, v, width=8, swap=True):
         staddr = addr
-        self.last_st_addr = addr # record last store
         remainder = addr & (self.bytes_per_word - 1)
         addr = addr >> self.word_log2
         log("Writing 0x%x to ST 0x%x memaddr 0x%x/%x swap %s" % \
@@ -136,6 +136,22 @@ class Mem:
             self.mem[addr] = v
         log("mem @ 0x%x: 0x%x" % (staddr, self.mem[addr]))
 
+    def st(self, addr, v, width=8, swap=True):
+        staddr = addr
+        self.last_st_addr = addr # record last store
+        # misaligned not allowed: pass straight to Mem._st
+        if not self.misaligned_ok:
+            return self._st(addr, v, width, swap)
+        remainder = addr & (self.bytes_per_word - 1)
+        addr = addr >> self.word_log2
+        if swap:
+            v = swap_order(v, width)
+        # not misaligned: pass through to Mem._st but we've swapped already
+        if remainder & (width - 1) == 0:
+            return self._st(addr, v, width, swap=False)
+        shifter, mask = self._get_shifter_mask(width, remainder)
+        print ("mask", hex(shifter), hex(mask))
+
     def __call__(self, addr, sz):
         val = self.ld(addr.value, sz, swap=False)
         log("memread", addr, sz, val)
diff --git a/src/openpower/decoder/isa/test_mem.py b/src/openpower/decoder/isa/test_mem.py
index bd847c26..b7734449 100644
--- a/src/openpower/decoder/isa/test_mem.py
+++ b/src/openpower/decoder/isa/test_mem.py
@@ -17,7 +17,7 @@ class TestMem(unittest.TestCase):
         self.assertEqual(d, [(0, 0x1234567800000000)])
 
     def test_mem_misalign_st(self):
-        m = Mem(row_bytes=8, initial_mem={})
+        m = Mem(row_bytes=8, initial_mem={}, misaligned_ok=True)
         m.st(4, 0x912345678, width=8, swap=False)
         d = m.dump()
         log ("dict", d)
-- 
2.30.2