Skip to content

Commit

Permalink
Merge pull request #51 from funkelab/dev/master
Browse files Browse the repository at this point in the history
Increase max message size assertion in TCPStream class
  • Loading branch information
rhoadesScholar authored Mar 17, 2024
2 parents fd10335 + b688d2f commit 750e8ca
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion daisy/tcp/tcp_stream.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,8 @@ async def _get_message(self):

size = await self.stream.read_bytes(4)
size = struct.unpack("I", size)[0]
assert size < 65535 # TODO: parameterize max message size
assert size < 2**64 # TODO: parameterize max message size
# assert size < 2**16 # TODO: parameterize max message size
pickled_data = await self.stream.read_bytes(size)

except tornado.iostream.StreamClosedError:
Expand Down

0 comments on commit 750e8ca

Please sign in to comment.