1 files changed, 76 insertions, 0 deletions
diff --git a/inspect_memoryregion.py b/inspect_memoryregion.py
new file mode 100644
index 0000000..7fdccd2
--- /dev/null
+++ b/inspect_memoryregion.py
@@ -0,0 +1,76 @@
+# Part of the Fifteen-Thieves Project
+# Chris Xiong 2020
+# License: Expat (MIT)
+import sys
+import xv.util
+import xv.amap
+
+wavenames = ['None']
+
+def dfs_checkpopulated(node):
+    node.populated = False
+    for ch in node.children_nodes:
+        if node.template.is_aggregate_node:
+            if type(ch) == list:
+                for el in ch:
+                    node.populated |= dfs_checkpopulated(el)
+            else:
+                node.populated |= dfs_checkpopulated(ch)
+        else:
+            for a in node.get_data():
+                if a != 0xff:
+                    node.populated = True
+                    break
+    return node.populated
+
+def dfs_print(node, level=0, instance=None):
+    if not node.populated:
+        return
+    ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(node.base_address))])
+    print(f"{ba_rep}{level*'  '} {node.template.name}{'['+str(instance)+']' if instance is not None else ''}")
+    for child in node.children_nodes:
+        if node.template.is_aggregate_node:
+            if type(child) == list:
+                for i, element in enumerate(child):
+                    dfs_print(element, level + 1, i)
+            else:
+                dfs_print(child, level + 1)
+        else:
+            if type(child) == list:
+                if child[0].name == "name":
+                    print(f"{ba_rep}{(level + 1)*'  '} {child[0].name} {node.data[child[0].offset:child[0].offset + 12].decode()}")
+                else:
+                    for i, element in enumerate(child):
+                        ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(element.offset))])
+                        print(f"{ba_rep}{(level + 1)*'  '} {element.name}[{i}] {xv.amap.get_leaf_data(node.data, element)}")
+            else:
+                ba_rep = ''.join([format(x, "02x") for x in xv.util.padbelist(xv.util.int2b7belist(child.offset))])
+                if child.name.startswith("wave_number"):
+                    try:
+                        print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}: {wavenames[xv.amap.get_leaf_data(node.data, child)]}")
+                    except IndexError:
+                        print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}: unknown")
+                else:
+                    print(f"{ba_rep}{(level + 1)*'  '} {child.name} {xv.amap.get_leaf_data(node.data, child)}")
+
+if __name__ == "__main__":
+    if len(sys.argv) < 2:
+        print(f"Usage: {sys.argv[0]} <memoryregion file>")
+    print("loading address mapping...", flush=True, file=sys.stderr)
+    m = xv.amap.AddrMapTemplate("amaps/sd-80.amap")
+    t = xv.amap.create_addr_map_tree(m)
+    t.data[:] = [0xff for i in range(0, len(t.data))]
+
+    print("loading wave names...", file=sys.stderr)
+    with open("data/sdwavelist","r") as f:
+        for l in f:
+            wavenames += [l.split('\t')[1].strip()]
+
+    print("loading memoryregion...", file=sys.stderr)
+    mr = xv.util.load_memoryregion(sys.argv[1])
+    for addr, data in mr:
+        ra = xv.util.b7belist2int(addr)
+        t.data[ra:ra + len(data)] = data
+
+    dfs_checkpopulated(t)
+    dfs_print(t)