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

set activity flag on noise / secio (#330)

This commit is contained in:
Jacek Sieka 2020-08-15 07:36:15 +02:00 committed by GitHub
parent d1f1e1b31e
commit 9c7e055310
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libp2p/protocols/secure/noise.nim
View File

@ -422,6 +422,7 @@ method write*(sconn: NoiseConnection, message: seq[byte]): Future[void] {.async.
outbuf &= besize outbuf &= besize
outbuf &= cipher outbuf &= cipher
await sconn.stream.write(outbuf) await sconn.stream.write(outbuf)
sconn.activity = true
method handshake*(p: Noise, conn: Connection, initiator: bool): Future[SecureConn] {.async.} = method handshake*(p: Noise, conn: Connection, initiator: bool): Future[SecureConn] {.async.} =
trace "Starting Noise handshake", initiator, peer = $conn trace "Starting Noise handshake", initiator, peer = $conn

1
libp2p/protocols/secure/secio.nim
View File

@ -236,6 +236,7 @@ method write*(sconn: SecioConn, message: seq[byte]) {.async.} =
trace "Writing message", message = msg.shortLog, left, offset trace "Writing message", message = msg.shortLog, left, offset
await sconn.stream.write(msg) await sconn.stream.write(msg)
sconn.activity = true
proc newSecioConn(conn: Connection, proc newSecioConn(conn: Connection,
hash: string, hash: string,

1
libp2p/protocols/secure/secure.nim
View File

@ -103,6 +103,7 @@ method readOnce*(s: SecureConn,
if s.buf.data().len() == 0: if s.buf.data().len() == 0:
let buf = await s.readMessage() let buf = await s.readMessage()
s.activity = true
if buf.len == 0: if buf.len == 0:
raise newLPStreamIncompleteError() raise newLPStreamIncompleteError()
s.buf.add(buf) s.buf.add(buf)