forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathResourceReader.py
More file actions
51 lines (47 loc) · 1.45 KB
/
Copy pathResourceReader.py
File metadata and controls
51 lines (47 loc) · 1.45 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
import ntpath
from ..streams import EndianBinaryReader
from ..files import File
def get_resource_data(*args):
"""
Input:
Option 1:
0 - path - file path
1 - assets_file - SerializedFile
2 - offset -
3 - size -
Option 2:
0 - reader - EndianBinaryReader
1 - offset -
2 - size -
-> -2 = offset, -1 = size
"""
if len(args) == 4:
res_path, assets_file, offset, size = args
basename = ntpath.basename(res_path)
name, ext = ntpath.splitext(basename)
possible_names = [
basename,
f"{name}.resource",
f"{name}.assets.resS",
f"{name}.resS",
]
environment = assets_file.environment
reader = None
for possible_name in possible_names:
reader = environment.get_cab(possible_name)
if reader:
break
if not reader:
assets_file.load_dependencies(possible_names)
for possible_name in possible_names:
reader = environment.get_cab(possible_name)
if reader:
break
if not reader:
raise FileNotFoundError(f"Resource file {basename} not found")
elif len(args) == 3:
reader, offset, size = args
else:
raise TypeError(f"3 or 4 arguments required, but only {len(args)} given")
reader.Position = offset
return reader.read_bytes(size)