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_unpack_varint():
buffer = Buffer()
for value, data in varint_vectors:
buffer.add(data)
assert buffer.unpack_varint() == value
assert len(buffer) == 0
