forked from K0lb3/UnityPy
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathEndianBinaryWriter.py
More file actions
169 lines (128 loc) · 5.13 KB
/
Copy pathEndianBinaryWriter.py
File metadata and controls
169 lines (128 loc) · 5.13 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
import io
import struct
from ..math import Color, Matrix4x4, Quaternion, Vector2, Vector3, Vector4, Rectangle
class EndianBinaryWriter:
endian: str
Length: int
Position: int
stream: io.BufferedReader
def __init__(self, input_=b"", endian=">"):
if isinstance(input_, (bytes, bytearray)):
self.stream = io.BytesIO(input_)
self.stream.seek(0, 2)
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()
pass
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(struct.pack(self.endian + "b", value))
def write_u_byte(self, value: int):
self.write(struct.pack(self.endian + "B", value))
def write_bytes(self, value: bytes):
return self.write(value)
def write_short(self, value: int):
self.write(struct.pack(self.endian + "h", value))
def write_int(self, value: int):
self.write(struct.pack(self.endian + "i", value))
def write_long(self, value: int):
self.write(struct.pack(self.endian + "q", value))
def write_u_short(self, value: int):
self.write(struct.pack(self.endian + "H", value))
def write_u_int(self, value: int):
self.write(struct.pack(self.endian + "I", value))
def write_u_long(self, value: int):
self.write(struct.pack(self.endian + "Q", value))
def write_float(self, value: float):
self.write(struct.pack(self.endian + "f", value))
def write_double(self, value: float):
self.write(struct.pack(self.endian + "d", value))
def write_boolean(self, value: bool):
self.write(struct.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=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_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, value: list, write_length: bool = True):
if write_length:
self.write_int(len(value))
for val in value:
command(val)
def write_byte_array(self, value: bytes):
self.write_int(len(value))
self.write(value)
def write_boolean_array(self, value: list):
self.write_array(self.write_boolean, value)
def write_u_short_array(self, value: list):
self.write_array(self.write_u_short, value)
def write_int_array(self, value: list, write_length: bool = False):
return self.write_array(self.write_int, value, write_length)
def write_u_int_array(self, value: list, write_length: bool = False):
return self.write_array(self.write_u_int, value, write_length)
def write_float_array(self, value: list, write_length: bool = False):
return self.write_array(self.write_float, value, write_length)
def write_string_array(self, value: list):
self.write_array(self.write_aligned_string, value)
def write_vector2_array(self, value: list):
self.write_array(self.write_vector2, value)
def write_vector4_array(self, value: list):
self.write_array(self.write_vector4, value)
def write_matrix_array(self, value: list):
self.write_array(self.write_matrix, value)