Skip to content

Commit 3c3e628

Browse files
jakobhellermannK0lb3
authored andcommitted
type tree generator: prepend monobehaviour base fields
instead of relying on generator to do it
1 parent 909460d commit 3c3e628

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

UnityPy/files/ObjectReader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ def _try_monobehaviour_node(self, base_node: TypeTreeNode) -> TypeTreeNode:
310310
else:
311311
fullname = script.m_ClassName
312312

313-
node = generator.get_nodes_up(script.m_AssemblyName, fullname)
313+
node = generator.get_nodes_up(base_node, script.m_AssemblyName, fullname)
314314
if node:
315315
return node
316316
else:

UnityPy/helpers/TypeTreeGenerator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def load_local_dll_folder(self, dll_dir: str):
4545
data = f.read()
4646
self.load_dll(data)
4747

48-
def get_nodes_up(self, assembly: str, fullname: str) -> TypeTreeNode:
48+
def get_nodes_up(self, base_node: TypeTreeNode, assembly: str, fullname: str) -> TypeTreeNode:
4949
root = self.cache.get((assembly, fullname))
5050
if root is not None:
5151
return root
@@ -62,6 +62,7 @@ def get_nodes_up(self, assembly: str, fullname: str) -> TypeTreeNode:
6262
0,
6363
0,
6464
m_MetaFlag=base_root.m_MetaFlag,
65+
m_Children=base_node.m_Children[:],
6566
)
6667
stack: List[TypeTreeNode] = []
6768
parent = root

0 commit comments

Comments
 (0)