-
Notifications
You must be signed in to change notification settings - Fork 0
/
control.py
108 lines (90 loc) · 3 KB
/
control.py
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
#!/usr/bin/python
import struct
import binascii
import os
from io import BytesIO
def int_to_string(number,size):
size = size * 2
b = "%.*X" % (size, number)
b = binascii.unhexlify(b)
b = b[::-1]
return b
class pac_file(object):
def __init__(self, filename, buffer=2048):
if hasattr(filename, 'read') and hasattr(filename, 'seek'):
self.the_file = filename
self.the_file.seek(0)
self.size = len(self.the_file.read())
self.the_file.seek(0)
self.filename = ''
else:
self.the_file = open(filename, 'rb')
self.size = os.path.getsize(filename)
self.filename = filename
self.buffer = buffer
def close(self):
self.the_file.close()
def tell(self):
return self.the_file.tell()
def rewind(self):
self.the_file.rewind()
def seek(self, offset, whence=0):
self.the_file.seek(offset, whence)
#def write(self,str):
# self.the_file.write(str)
def read_int(self, size):
string = self.the_file.read(size)
string = string[::-1]
hex_data = binascii.hexlify(string)
#if sys.byteorder == 'little':
return int(hex_data, 16)
def read_string(self, size):
no_chunk = size // self.buffer
remain_s = size - (no_chunk * self.buffer)
buf = BytesIO()
for i in range(0, no_chunk):
data = self.the_file.read(self.buffer)
buf.write(data)
data = self.the_file.read(remain_s)
buf.write(data)
#if sys.byteorder == 'little'
return buf.getvalue()
def int_to_string(self, number,size):
size = size * 2
b = "%.*X" % (size, number)
b = binascii.unhexlify(b)
b = b[::-1]
return b
class memory_file(object):
def __init__(self, fileobj_list=()):
self.file_dict = {}
for fileobj in fileobj_list:
offset = fileobj.tell()
fileobj.seek(0)
data = fileobj.read()
fileobj.seek(offset)
self.file_dict[fileobj.name] = BytesIO(data)
return
def add_file(self, filename, data):
self.file_dict[filename] = BytesIO(data)
return
def remove_file(self, filename):
del self.file_dict[filename]
return
def getvalue(self, filename):
offset = self.file_dict[filename].tell()
self.file_dict[filename].seek(0)
data = self.file_dict[filename].read()
self.file_dict[filename].seek(offset)
return data
def read(self, filename):
return self.file_dict[filename].getvalue()
def get_fileobj(self, filename):
return self.file_dict[filename]
def namelist(self):
list = []
for x in self.file_dict:
list.append(x)
return list
def close(self, filename):
self.file_dict[filename].close()