def data_received(self, data):
# Decrypt data
data = self.cipher.decrypt(data)
# Add it to our buffer
self.recv_buff.add(data)
# Read some packets
while not self.closed:
# Save the buffer, in case we read an incomplete packet
self.recv_buff.save()
# Try to read a packet
try:
max_bits = 32 if self.protocol_mode == "play" else 21
packet_length = self.recv_buff.unpack_varint(max_bits=max_bits)
packet_body = self.recv_buff.read(packet_length)
# Incomplete packet read, restore the buffer.
except BufferUnderrun:
self.recv_buff.restore()
break
# Load the packet body into a buffer
packet_buff = self.buff_type()
packet_buff.add(packet_body)
try: # Catch protocol errors
try: # Catch buffer overrun/underrun
if self.compression_enabled:
uncompressed_length = packet_buff.unpack_varint()
if uncompressed_length > 0:
data = zlib.decompress(packet_buff.read())
packet_buff = Buffer()
packet_buff.add(data)
ident = packet_buff.unpack_varint()
key = (self.protocol_version, self.protocol_mode,
self.recv_direction, ident)
try:
name = packets.packet_names[key]
except KeyError:
raise ProtocolError("No name known for packet: %s" %
(key, ))
self.packet_received(packet_buff, name)
except BufferUnderrun:
raise ProtocolError("Packet is too short!")
if len(packet_buff) > 0:
raise ProtocolError("Packet is too long!")
except ProtocolError as e:
self.protocol_error(e)
break
# We've read a complete packet, so reset the inactivity timeout
self.connection_timer.restart()
def test_read():
buffer = Buffer()
buffer.add(b"spam")
assert buffer.read(2) == b"sp"
assert buffer.read(2) == b"am"
buffer.add(b"eggs")
assert buffer.read() == b"eggs"
with pytest.raises(BufferUnderrun):
buffer.read(1)
def test_save_discard_restore():
buffer = Buffer()
buffer.add(b"spam")
buffer.save()
assert len(buffer) == 4
buffer.discard()
assert len(buffer) == 0
buffer.restore()
assert len(buffer) == 4
def test_unpack_chat():
buffer = Buffer()
buffer.add(b'\x11["spam", " eggs"]')
assert buffer.unpack_chat() == "spam eggs"
buffer.add(b'\x22{"text": "spam", "extra": " eggs"}')
assert buffer.unpack_chat() == "spam eggs"
buffer.add(b'\x14{"translate": "foo"}')
assert buffer.unpack_chat() == "foo"
buffer.add(b'\x2E{"translate": "foo", "with": ["spam", "eggs"]}')
assert buffer.unpack_chat() == "foo{spam, eggs}"
def __init__(self,
topleft=(-1000, -1000),
bottomright=(1000, 1000),
cuty=0,
world=1,
norenderblocks=["none"],
multymatblocks=[
35, 43, 44, 95, 97, 98, 125, 126, 139, 155, 159, 160, 171
],
colormats={}):
self.colormats = colormats
self.topleft = topleft
self.bottomright = bottomright
self.cuty = cuty
self.world = world
self.norenderblocks = norenderblocks
self.multymatblocks = multymatblocks
self.chunks = {}
self.materials = {}
self.neightbors = {}
self.worldnum = 0
self.chunkbuffer = Buffer()
def test_unpack_uuid():
buffer = Buffer()
buffer.add(uuid_vector)
assert buffer.unpack_uuid().to_bytes() == uuid_vector
def test_unpack_varint():
buffer = Buffer()
for value, data in varint_vectors:
buffer.add(data)
assert buffer.unpack_varint(signed=True) == value
assert len(buffer) == 0
def test_unpack_json():
buffer = Buffer()
buffer.add(b'\x10{"spam": "eggs"}')
assert buffer.unpack_json() == {"spam": "eggs"}
def test_unpack_string():
buffer = Buffer()
buffer.add(b"\x04spam")
assert buffer.unpack_string() == "spam"
def test_unpack():
buffer = Buffer()
for fmt, data, values in pack_unpack_vectors:
buffer.add(data)
assert buffer.unpack(fmt) == values
def test_add():
buffer = Buffer()
buffer.add(b"spam")
assert len(buffer) == 4
assert buffer.read() == b"spam"
def testVarInt(self):
value = 5533
buff = Buffer()
buff.buff = buff.pack_varint(value)
self.assertTrue(buff.unpack_varint() == value)
def testUuid(self):
uuid = types.UUID.from_offline_player("KLonter")
buff = Buffer()
buff.buff = buff.pack_uuid(uuid)
self.assertTrue(uuid == buff.unpack_uuid())
def testString(self):
text = "more tests, and stuff"
buff = Buffer()
buff.buff = buff.pack_string(text)
self.assertTrue(text == buff.unpack_string())
def testChat(self):
text = "Hi there !"
buff = Buffer()
buff.buff = buff.pack_chat(text)
self.assertTrue(text == buff.unpack_chat())
def testBlockPosition(self):
blockpos = (345, -100, -2005)
buff = Buffer()
buff.buff = buff.pack_blockposition(blockpos)
self.assertTrue(blockpos == buff.unpack_blockposition())
