forked from hipstercat/giantsd
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import io
|
|
import struct
|
|
|
|
|
|
class Packet(io.BytesIO):
|
|
def putByte(self, val):
|
|
self.write(struct.pack('<B', val % 256))
|
|
|
|
def getByte(self):
|
|
return struct.unpack('<B', self.read(1))[0]
|
|
|
|
def putBytes(self, bytes):
|
|
for byte in bytes:
|
|
self.write(struct.pack('<B', byte))
|
|
|
|
def getBytes(self, num):
|
|
return bytes(struct.unpack('<'+str(num)+'B', self.read(num)))
|
|
|
|
def putShort(self, val):
|
|
self.write(struct.pack('<H', val))
|
|
|
|
def getShort(self):
|
|
return struct.unpack('<H', self.read(2))[0]
|
|
|
|
def putLong(self, val):
|
|
self.write(struct.pack('<l', val))
|
|
|
|
def putULong(self, val):
|
|
self.write(struct.pack('<L', val))
|
|
|
|
def getLong(self):
|
|
return struct.unpack('<l', self.read(4))[0]
|
|
|
|
def getULong(self):
|
|
return struct.unpack('<L', self.read(4))[0]
|
|
|
|
def putLongLong(self, val):
|
|
self.write(struct.pack('<Q', val))
|
|
|
|
def getLongLong(self):
|
|
return struct.unpack('<Q', self.read(8))[0]
|
|
|
|
def putFloat(self, val):
|
|
self.write(struct.pack('<f', val))
|
|
|
|
def getFloat(self):
|
|
return struct.unpack('<f', self.read(4))[0]
|
|
|
|
def putString(self, val):
|
|
self.write(val + '\x00')
|