From 9d4eccf66e21b7d0314bd723fee726abd99c80f4 Mon Sep 17 00:00:00 2001
From: Jennifer Taylor <dragonminded@dragonminded.com>
Date: Fri, 3 Sep 2021 05:07:15 +0000
Subject: [PATCH] Add in a few additional 64-bit operations I was missing.

---
 bemani/common/pe.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/bemani/common/pe.py b/bemani/common/pe.py
index 9373a98..14e56d3 100644
--- a/bemani/common/pe.py
+++ b/bemani/common/pe.py
@@ -501,6 +501,8 @@ def assign(registers: Registers, memory: Memory, size: int, loc: str, value: int
             data = struct.pack("<H", value)
         elif size == 4:
             data = struct.pack("<I", value)
+        elif size == 8:
+            data = struct.pack("<Q", value)
         else:
             raise Exception(f"Unsupported size {size} for memory assign!")
         memory.store(address, data)
@@ -668,6 +670,8 @@ def fetch(registers: Registers, memory: Memory, size: int, loc: str) -> int:
             return struct.unpack("<H", memory.load(address, size))[0]
         elif size == 4:
             return struct.unpack("<I", memory.load(address, size))[0]
+        elif size == 8:
+            return struct.unpack("<Q", memory.load(address, size))[0]
         else:
             raise Exception(f"Unsupported size {size} for memory fetch!")
 
@@ -679,6 +683,8 @@ def fetch(registers: Registers, memory: Memory, size: int, loc: str) -> int:
             return immediate & 0xFFFF
         if size == 4:
             return immediate & 0xFFFFFFFF
+        if size == 8:
+            return immediate
         raise Exception(f"Unsupported size {size} for immediate fetch!")
 
     if loc == "rax":