blob: 028f5c4cc49dba8174220db6cea1cff7dc8626fe (
plain)
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
|
import struct
class ByteReader:
def __init__(self, data):
self.data = data
self.o = 0
def readbyte(self):
if self.o+1 > len(self.data): raise Exception("EOF")
self.o += 1
return struct.unpack_from("<B", self.data, self.o-1)[0]
def readword(self):
if self.o+2 > len(self.data): raise Exception("EOF")
self.o += 2
return struct.unpack_from("<H", self.data, self.o-2)[0]
def readdword(self):
if self.o+4 > len(self.data): raise Exception("EOF")
self.o += 4
return struct.unpack_from("<L", self.data, self.o-4)[0]
def readbytes(self, n=None):
if n is None:
n = len(self.data) - self.o
if self.o+n > len(self.data): raise Exception("EOF")
self.o += n
return self.data[self.o-n:self.o]
def readname(self):
namelen = self.readbyte()
return self.readbytes(namelen).decode('cp1251')
def eof(self):
return self.o >= len(self.data)
|
