pyCraft/networking/DataUtil.py

120 lines
4.0 KiB
Python

import struct
import types
def readBoolean(FileObject):
return struct.unpack('?', FileObject.read(1))[0]
def readByte(FileObject):
return struct.unpack('!b', FileObject.read(1))[0]
def readUnsignedByte(FileObject):
return struct.unpack('!B', FileObject.read(1))[0]
def readShort(FileObject):
return struct.unpack('!h', FileObject.read(2))[0]
def readUnsignedShort(FileObject):
return struct.unpack('!H', FileObject.read(2))[0]
def readInt(FileObject):
return struct.unpack('!i', FileObject.read(4))[0]
def readFloat(FileObject):
return struct.unpack('!f', FileObject.read(4))[0]
def readLong(FileObject):
return struct.unpack('!q', FileObject.read(8))[0]
def readDouble(FileObject):
return struct.unpack('!d', FileObject.read(8))[0]
def readByteArray(FileObject, length):
return struct.unpack(str(length) + "s", FileObject.read(length))[0]
def readString(FileObject):
length = readShort(FileObject) * 2
return FileObject.read(length).decode("utf-16be")
def sendBoolean(socket, value):
assert type(value) is types.BooleanType, "value is not a boolean: %r" % value
socket.send(struct.pack('?', value))
def sendByte(socket, value):
socket.send(struct.pack('!b', value))
def sendUnsignedByte(socket, value):
socket.send(struct.pack('!B', value))
def sendShort(socket, value):
socket.send(struct.pack('!h', value))
def sendUnsignedShort(socket, value):
socket.send(struct.pack('!H', value))
def sendInt(socket, value):
assert type(value) is types.IntType, "value is not an integer: %r" % value
socket.send(struct.pack('!i', value))
def sendFloat(socket, value):
socket.send(struct.pack('!f', value))
def sendLong(socket, value):
socket.send(struct.pack('!q', value))
def sendDouble(socket, value):
socket.send(struct.pack('!d'), value)
def sendString(socket, value):
if (type(value) is not types.StringType):
value = str(value)
socket.send(struct.pack('!h', value.__len__()))
socket.send(value.encode('utf-16be'))
def readEntityMetadata(FileObject):
metadata = {}
byte = struct.unpack('!B', FileObject.read(1))[0]
while byte != 127:
index = byte & 0x1F # Lower 5 bits
ty = byte >> 5 # Upper 3 bits
if ty == 0: val = struct.unpack('!b', FileObject.read(1))[0]
if ty == 1: val = struct.unpack('!h', FileObject.read(2))[0]
if ty == 2: val = readInt(FileObject)
if ty == 3: val = struct.unpack('!f', FileObject.read(4))[0]
if ty == 4:
length = struct.unpack('!h', FileObject.read(2))[0] * 2
val = FileObject.read(length).decode('utf-16be')
if ty == 5:
val = {}
val["id"] = struct.unpack('!h', FileObject.read(2))[0]
if (val["id"] != -1):
val["count"] = struct.unpack('!b', FileObject.read(1))[0]
val["damage"] = struct.unpack('!h', FileObject.read(2))[0]
if ty == 6:
val = []
for i in range(3):
val.append(readInt(FileObject))
metadata[index] = (ty, val)
byte = struct.unpack('!B', FileObject.read(1))[0]
return metadata
def readSlotData(FileObject):
BlockID = struct.unpack('!h', FileObject.read(2))[0]
if(BlockID != -1):
ItemCount = struct.unpack('!b', FileObject.read(1))[0]
Damage = struct.unpack('!h', FileObject.read(2))[0]
MetadataLength = struct.unpack('!h', FileObject.read(2))[0]
if(MetadataLength != -1):
raw = FileObject.read(MetadataLength)
ByteArray = struct.unpack(str(MetadataLength) + "s", raw)[0]
return {'BlockID' : BlockID,
'ItemCount' : ItemCount,
'Damage' : Damage,
'Data' : ByteArray
}
return {'BlockID' : BlockID,
'ItemCount' : ItemCount,
'Damage' : Damage
}
return {'BlockID' : -1,
'ItemCount' : 0
}