forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPPtr.py
More file actions
177 lines (147 loc) · 5.31 KB
/
Copy pathPPtr.py
File metadata and controls
177 lines (147 loc) · 5.31 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
from ..streams import EndianBinaryReader, EndianBinaryWriter
class PPtr:
def __init__(self, reader: EndianBinaryReader):
self.index = -2
self.file_id = reader.read_int()
self.path_id = reader.read_int() if reader.version2 < 14 else reader.read_long()
self.assets_file = reader.assets_file
def __getattr__(self, key):
# get manager
manager = None
if self.file_id == 0:
manager = self.assets_file
elif self.file_id > 0 and self.file_id - 1 < len(self.assets_file.externals):
if self.index == -2:
external_name = self.assets_file.externals[self.file_id - 1].name
files = self.assets_file.parent.files
if external_name not in files:
external_name = external_name.upper()
manager = self.assets_file.parent.files[external_name]
if manager and self.path_id in manager.objects:
self = manager.objects[self.path_id]
return getattr(self, key)
if self.index != -2:
raise NotImplementedError("PPtr")
"""
if (!assetsFileIndexCache.TryGetValue(name, out index))
{
index = assetsFileList.FindIndex(x => x.upperFileName == name);
assetsFileIndexCache.Add(name, index);
}
}
if (index >= 0)
{
result = assetsFileList[index];
return true;
}
}
if self.path_id in self.assets_file.objects:
return self.assets_file.objects[self.path_id]
else:
return super(PPtr, self).__new__(self)
"""
def __repr__(self):
return self.__class__.__name__
def __bool__(self):
return False
def save_ptr(obj, writer: EndianBinaryWriter):
if isinstance(obj, PPtr):
writer.write_int(obj.file_id)
writer.write_int(obj.path_id)
else:
writer.write_int(0) # it's usually 0......
writer.write_int(obj.path_id)
"""
def TryGetAssetsFile(self):
result = None
if m_FileID == 0:
return self.assets_file
if m_FileID > 0 and m_FileID - 1 < len(self.assets_file.externals):
var assetsManager = assetsFile.assetsManager
var assetsFileList = assetsManager.assetsFileList
var assetsFileIndexCache = assetsManager.assetsFileIndexCache
if self.index == -2:
external_name = self.assets_file.eternals[self.file_id - 1].filename
if (!assetsFileIndexCache.TryGetValue(name, out index))
{
index = assetsFileList.FindIndex(x => x.upperFileName == name);
assetsFileIndexCache.Add(name, index);
}
}
if (index >= 0)
{
result = assetsFileList[index];
return true;
}
}
return false;
}
public bool TryGet(out T result)
{
if (TryGetAssetsFile(out var sourceFile))
{
if (sourceFile.Objects.TryGetValue(m_PathID, out var obj))
{
if (obj is T variable)
{
result = variable;
return true;
}
}
}
result = null;
return false;
}
public bool TryGet<T2>(out T2 result) where T2 : Object
{
if (TryGetAssetsFile(out var sourceFile))
{
if (sourceFile.Objects.TryGetValue(m_PathID, out var obj))
{
if (obj is T2 variable)
{
result = variable;
return true;
}
}
}
result = null;
return false;
}
public void Set(T m_Object)
{
var name = m_Object.assetsFile.upperFileName;
if (string.Equals(assetsFile.upperFileName, name, StringComparison.Ordinal))
{
m_FileID = 0;
}
else
{
m_FileID = assetsFile.m_Externals.FindIndex(x => string.Equals(x.fileName, name, StringComparison.OrdinalIgnoreCase));
if (m_FileID == -1)
{
assetsFile.m_Externals.Add(new FileIdentifier
{
fileName = m_Object.assetsFile.fileName
});
m_FileID = assetsFile.m_Externals.Count;
}
else
{
m_FileID += 1;
}
}
var assetsManager = assetsFile.assetsManager;
var assetsFileList = assetsManager.assetsFileList;
var assetsFileIndexCache = assetsManager.assetsFileIndexCache;
if (!assetsFileIndexCache.TryGetValue(name, out index))
{
index = assetsFileList.FindIndex(x => x.upperFileName == name);
assetsFileIndexCache.Add(name, index);
}
m_PathID = m_Object.m_PathID;
}
public bool IsNull => m_PathID == 0 || m_FileID < 0;
}
}
"""