forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 5
Expand file tree
/
Copy pathEndianBinaryWriter.py
More file actions
184 lines (140 loc) · 5.64 KB
/
Copy pathEndianBinaryWriter.py
File metadata and controls
184 lines (140 loc) · 5.64 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
178
179
180
181
182
183
184
from struct import pack
from io import BytesIO, IOBase
import builtins
from typing import Callable, Sequence, TypeVar, Union
from ..math import Color, Matrix4x4, Quaternion, Vector2, Vector3, Vector4, Rectangle
T = TypeVar("T")
class EndianBinaryWriter:
endian: str
Position: int
stream: IOBase
def __init__(
self,
input_: Union[bytes, bytearray, IOBase] = b"",
endian: str = ">"
):
if isinstance(input_, (bytes, bytearray)):
self.stream = BytesIO(input_)
self.stream.seek(0, 2)
elif isinstance(input_, IOBase):
self.stream = input_
else:
raise ValueError("Invalid input type - %s." % type(input_))
self.endian = endian
self.Position = self.stream.tell()
@property
def bytes(self):
self.stream.seek(0)
return self.stream.read()
@property
def Length(self) -> int:
pos = self.stream.tell()
self.stream.seek(0, 2)
l = self.stream.tell()
self.stream.seek(pos)
return l
def dispose(self):
self.stream.close()
def write(self, *args):
if self.Position != self.stream.tell():
self.stream.seek(self.Position)
ret = self.stream.write(*args)
self.Position = self.stream.tell()
return ret
def write_byte(self, value: int):
self.write(pack(self.endian + "b", value))
def write_u_byte(self, value: int):
self.write(pack(self.endian + "B", value))
def write_bytes(self, value: builtins.bytes):
return self.write(value)
def write_short(self, value: int):
self.write(pack(self.endian + "h", value))
def write_int(self, value: int):
self.write(pack(self.endian + "i", value))
def write_long(self, value: int):
self.write(pack(self.endian + "q", value))
def write_u_short(self, value: int):
self.write(pack(self.endian + "H", value))
def write_u_int(self, value: int):
self.write(pack(self.endian + "I", value))
def write_u_long(self, value: int):
self.write(pack(self.endian + "Q", value))
def write_float(self, value: float):
self.write(pack(self.endian + "f", value))
def write_double(self, value: float):
self.write(pack(self.endian + "d", value))
def write_boolean(self, value: bool):
self.write(pack(self.endian + "?", value))
def write_string_to_null(self, value: str):
self.write(value.encode("utf8", "surrogateescape"))
self.write(b"\0")
def write_aligned_string(self, value: str):
bstring = value.encode("utf8", "surrogateescape")
self.write_int(len(bstring))
self.write(bstring)
self.align_stream(4)
def align_stream(self, alignment:int = 4):
pos = self.stream.tell()
align = (alignment - pos % alignment) % alignment
self.write(b"\0" * align)
def write_quaternion(self, value: Quaternion):
self.write_float(value.X)
self.write_float(value.Y)
self.write_float(value.Z)
self.write_float(value.W)
def write_vector2(self, value: Vector2):
self.write_float(value.X)
self.write_float(value.Y)
def write_vector3(self, value: Vector3):
self.write_float(value.X)
self.write_float(value.Y)
self.write_float(value.Z)
def write_vector4(self, value: Vector4):
self.write_float(value.X)
self.write_float(value.Y)
self.write_float(value.Z)
self.write_float(value.W)
def write_rectangle_f(self, value: Rectangle):
self.write_float(value.x)
self.write_float(value.y)
self.write_float(value.width)
self.write_float(value.height)
def write_color_uint(self, value: Color):
self.write_u_byte(int(value.R * 255))
self.write_u_byte(int(value.G * 255))
self.write_u_byte(int(value.B * 255))
self.write_u_byte(int(value.A * 255))
def write_color4(self, value: Color):
self.write_float(value.R)
self.write_float(value.G)
self.write_float(value.B)
self.write_float(value.A)
def write_matrix(self, value: Matrix4x4):
for val in value.M:
self.write_float(val)
def write_array(self, command: Callable[[T], None], value: Sequence[T], write_length: bool = True):
if write_length:
self.write_int(len(value))
for val in value:
command(val)
def write_byte_array(self, value: builtins.bytes):
self.write_int(len(value))
self.write(value)
def write_boolean_array(self, value: Sequence[bool]):
self.write_array(self.write_boolean, value)
def write_u_short_array(self, value: Sequence[int]):
self.write_array(self.write_u_short, value)
def write_int_array(self, value: Sequence[int], write_length: bool = False):
return self.write_array(self.write_int, value, write_length)
def write_u_int_array(self, value: Sequence[int], write_length: bool = False):
return self.write_array(self.write_u_int, value, write_length)
def write_float_array(self, value: Sequence[float], write_length: bool = False):
return self.write_array(self.write_float, value, write_length)
def write_string_array(self, value: Sequence[str]):
self.write_array(self.write_aligned_string, value)
def write_vector2_array(self, value: Sequence[Vector2]):
self.write_array(self.write_vector2, value)
def write_vector4_array(self, value: Sequence[Vector4]):
self.write_array(self.write_vector4, value)
def write_matrix_array(self, value: Sequence[Matrix4x4]):
self.write_array(self.write_matrix, value)