status-im/nim-websock
2
0
Fork 0
mirror of https://github.com/status-im/nim-websock.git synced 2025-02-04 13:44:32 +00:00
Code Issues Projects Releases Wiki Activity

avoid spin/live-lock on 0 bytes read

Browse Source
This commit is contained in:
Dmitriy Ryajov 2022-01-25 10:05:17 -06:00
parent 73edde4417
commit 5448b623bd
No known key found for this signature in database
GPG Key ID: DA8C680CE7C657A4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
websock/session.nim
View File

@ -399,6 +399,9 @@ proc recvMsg*(
var buf = newSeq[byte](min(size, ws.frameSize)) var buf = newSeq[byte](min(size, ws.frameSize))
let read = await ws.recv(addr buf[0], buf.len) let read = await ws.recv(addr buf[0], buf.len)
if read <= 0:
return res
buf.setLen(read) buf.setLen(read)
if res.len + buf.len > size: if res.len + buf.len > size:
raise newException(WSMaxMessageSizeError, "Max message size exceeded") raise newException(WSMaxMessageSizeError, "Max message size exceeded")