Skip to content

Commit a005589

Browse files
committed
use more efficient bytearray operations in Defragmenter
1 parent 92c98ad commit a005589

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tlslite/defragmenter.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def size_handler(data):
8989
else:
9090
parser = Parser(data)
9191
# skip the header
92-
parser.getFixBytes(size_offset)
92+
parser.skip_bytes(size_offset)
9393

9494
payload_length = parser.get(size_of_size)
9595
if parser.getRemainingLength() < payload_length:
@@ -110,14 +110,15 @@ def add_data(self, msg_type, data):
110110
def get_message(self):
111111
"""Extract the highest priority complete message from buffer"""
112112
for msg_type in self.priorities:
113-
length = self.decoders[msg_type](self.buffers[msg_type])
113+
buf = self.buffers[msg_type]
114+
length = self.decoders[msg_type](buf)
114115
if length is None:
115116
continue
116117

117118
# extract message
118-
data = self.buffers[msg_type][:length]
119+
data = buf[:length]
119120
# remove it from buffer
120-
self.buffers[msg_type] = self.buffers[msg_type][length:]
121+
del buf[:length]
121122
return (msg_type, data)
122123
return None
123124

0 commit comments

Comments
 (0)