codex-storage
/
nim-libp2p
mirror of https://github.com/codex-storage/nim-libp2p.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Handle trying to write empty byte seq (#780)

This commit is contained in:
diegomrsantos 2022-09-29 20:02:10 +02:00 committed by GitHub
parent 0cd3554ce4
commit 4f18dd30e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libp2p/protocols/connectivity/relay/utils.nim
View File

@ -64,15 +64,17 @@ proc bridge*(connSrc: Connection, connDst: Connection) {.async.} =
await futSrc or futDst await futSrc or futDst
if futSrc.finished(): if futSrc.finished():
bufRead = await futSrc bufRead = await futSrc
bytesSendFromSrcToDst.inc(bufRead) if bufRead > 0:
await connDst.write(@bufSrcToDst[0..<bufRead]) bytesSendFromSrcToDst.inc(bufRead)
zeroMem(addr(bufSrcToDst), bufSrcToDst.high + 1) await connDst.write(@bufSrcToDst[0..<bufRead])
zeroMem(addr(bufSrcToDst), bufSrcToDst.high + 1)
futSrc = connSrc.readOnce(addr bufSrcToDst[0], bufSrcToDst.high + 1) futSrc = connSrc.readOnce(addr bufSrcToDst[0], bufSrcToDst.high + 1)
if futDst.finished(): if futDst.finished():
bufRead = await futDst bufRead = await futDst
bytesSendFromDstToSrc += bufRead if bufRead > 0:
await connSrc.write(bufDstToSrc[0..<bufRead]) bytesSendFromDstToSrc += bufRead
zeroMem(addr(bufDstToSrc), bufDstToSrc.high + 1) await connSrc.write(bufDstToSrc[0..<bufRead])
zeroMem(addr(bufDstToSrc), bufDstToSrc.high + 1)
futDst = connDst.readOnce(addr bufDstToSrc[0], bufDstToSrc.high + 1) futDst = connDst.readOnce(addr bufDstToSrc[0], bufDstToSrc.high + 1)
except CancelledError as exc: except CancelledError as exc:
raise exc raise exc

3
libp2p/stream/chronosstream.nim
View File

@ -130,6 +130,9 @@ proc completeWrite(
method write*(s: ChronosStream, msg: seq[byte]): Future[void] = method write*(s: ChronosStream, msg: seq[byte]): Future[void] =
# Avoid a copy of msg being kept in the closure created by `{.async.}` as this # Avoid a copy of msg being kept in the closure created by `{.async.}` as this
# drives up memory usage # drives up memory usage
if msg.len == 0:
trace "Empty byte seq, nothing to write"
return
if s.closed: if s.closed:
let fut = newFuture[void]("chronosstream.write.closed") let fut = newFuture[void]("chronosstream.write.closed")
fut.fail(newLPStreamClosedError()) fut.fail(newLPStreamClosedError())