m1n1.adt: Add create_node() method

Signed-off-by: Martin Povišer <[email protected]>

Author: povik

proxyclient/m1n1/adt.py

@@ -308,12 +308,9 @@ def parse_prop(node, path, node_name, name, v, is_template=False):
                 break
             n = n._parent
         else:
-            ac, sc = node._parent.address_cells, node._parent.size_cells
-            at = Hex(Int64ul) if ac == 2 else Array(ac, Hex(Int32ul))
-            st = Hex(Int64ul) if sc == 2 else Array(sc, Hex(Int32ul))
-            t = SafeGreedyRange(Struct("addr" / at, "size" / st))
-            if len(v) % ((ac + sc) * 4):
-                t = None
+            rs = node._reg_struct
+            if len(v) % rs.sizeof() == 0:
+                t = SafeGreedyRange(rs)
 
     elif name == "ranges":
         try:
@@ -571,6 +568,14 @@ def __repr__(self):
     def __iter__(self):
         return iter(self._children)
 
+    @property
+    def _reg_struct(self):
+        ac, sc = self._parent.address_cells, self._parent.size_cells
+        return Struct(
+            "addr" / Hex(Int64ul) if ac == 2 else Array(ac, Hex(Int32ul)),
+            "size" / Hex(Int64ul) if sc == 2 else Array(sc, Hex(Int32ul))
+        )
+
     def get_reg(self, idx):
         reg = self.reg[idx]
         addr = reg.addr
@@ -641,6 +646,19 @@ def build_addr_lookup(self):
 
         return lookup
 
+    def create_node(self, name):
+        while name.startswith("/"):
+            name = name[1:]
+        if "/" in name:
+            a, b = name.split("/", 1)
+            return self[a].create_node(b)
+
+        node = ADTNode(path=self._path + "/", parent=self)
+        node.name = name
+        node._types["reg"] = (SafeGreedyRange(node._reg_struct), False)
+        self[name] = node
+        return node
+
 def load_adt(data):
     return ADTNode(ADTNodeStruct.parse(data))