codex-storage/nim-libp2p
1
0
Fork 0
mirror of https://github.com/codex-storage/nim-libp2p.git synced 2025-01-12 12:04:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve connection.nim read error clarity when buffer overflow

Browse Source
This commit is contained in:
Giovanni Petrantoni 2020-02-07 15:40:16 +09:00 committed by Dmitriy Ryajov
parent 120ba0d528
commit f9cbdc252f
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libp2p/connection.nim
View File

@ -122,8 +122,10 @@ proc readLp*(s: Connection): Future[seq[byte]] {.async, gcsafe.} =
res = LP.getUVarint(buff.toOpenArray(0, i), length, size) res = LP.getUVarint(buff.toOpenArray(0, i), length, size)
if res == VarintStatus.Success: if res == VarintStatus.Success:
break break
if res != VarintStatus.Success or size > DefaultReadSize: if res != VarintStatus.Success:
raise newInvalidVarintException() raise newInvalidVarintException()
if size > DefaultReadSize:
raise newLPStreamLimitError()
buff.setLen(size) buff.setLen(size)
if size > 0.uint: if size > 0.uint:
trace "reading exact bytes from stream", size = size trace "reading exact bytes from stream", size = size