forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathFile.py
More file actions
111 lines (90 loc) · 3.53 KB
/
Copy pathFile.py
File metadata and controls
111 lines (90 loc) · 3.53 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
from ..enums import FileType
from ..helpers import ImportHelper
from ..streams import EndianBinaryReader, EndianBinaryWriter
from collections import namedtuple
DirectoryInfo = namedtuple("DirectoryInfo", "path offset size")
class File(object):
name: str
files: dict
is_changed: bool
signature: str
packer: str
# parent: File
def __init__(self, parent=None):
self.files = {}
self.is_changed = False
self.parent = parent
def get_objects(self):
if self.isinstance(SerializedFile.SerializedFile):
return self.objects.values()
for f in self.files.values():
if isinstance(f, (BundleFile.BundleFile, WebFile.WebFile)):
for obj in f.objects:
yield obj
elif isinstance(f, SerializedFile.SerializedFile):
for obj in f.objects.values():
yield obj
elif isinstance(f, ObjectReader.ObjectReader):
yield f
def read_files(self, reader: EndianBinaryReader, files: list):
# read file data and convert it
for node in files:
reader.Position = node.offset
name = node.path
f = EndianBinaryReader(reader.read(node.size), offset=(reader.BaseOffset + node.offset))
# f._flag = getattr(node, "flags", None) # required for save
typ, _ = ImportHelper.check_file_type(f)
if typ == FileType.BundleFile:
f = BundleFile.BundleFile(f, self)
elif typ == FileType.WebFile:
f = WebFile.WebFile(f, self)
elif typ == FileType.AssetsFile:
# pre-check if resource file
if not name.endswith((".resS", ".resource", ".config", ".xml", ".dat")):
# try to load the file as serialized file
try:
f = SerializedFile.SerializedFile(f, self)
except ValueError:
pass
# required for BundleFiles
f.flags = getattr(node, "flags", 0)
self.files[name] = f
def get_writeable_cab(self, name : str = "CAB-UnityPy_Mod.resS"):
"""
Creates a new cab file in the bundle that contains the given data.
This is usefull for asset types that use resource files.
"""
if name in self.files:
if isinstance(self.files[name], EndianBinaryWriter):
return self.files[name]
else:
raise ValueError("This cab already exists and isn't an EndianBinaryWriter")
writer = EndianBinaryWriter()
writer.flags = 4
writer.name = name
self.files[name] = writer
return writer
@property
def container(self):
return {
path: obj
for f in self.files.values()
if isinstance(f, File)
for path, obj in f.container.items()
}
def get(self, key, default = None):
return getattr(self, key, default)
def keys(self):
return self.files.keys()
def items(self):
return self.files.items()
def values(self):
return self.files.values()
def __getitem__(self, item):
return self.files[item]
def __repr__(self):
return f"<{self.__class__.__name__}>"
def mark_changed(self):
self.is_changed = True
# recursive import requires the import down here
from . import BundleFile, SerializedFile, WebFile, ObjectReader