status-im
/
nim-libp2p
mirror of https://github.com/status-im/nim-libp2p.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

use `readExactly` not `read` to read all bytes

This commit is contained in:
Dmitriy Ryajov 2020-03-11 07:34:35 -06:00
parent 52cb7ced04
commit 07695b2260
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libp2p/muxers/mplex/coder.nim
View File

@ -54,10 +54,11 @@ proc readMsg*(conn: Connection): Future[Msg] {.async, gcsafe.} =
let dataLenVarint = await conn.readMplexVarint() let dataLenVarint = await conn.readMplexVarint()
trace "read data len varint", varint = dataLenVarint trace "read data len varint", varint = dataLenVarint
var data: seq[byte]
var data: seq[byte] = newSeq[byte](dataLenVarint.int)
if dataLenVarint.int > 0: if dataLenVarint.int > 0:
data = await conn.read(dataLenVarint.int) await conn.readExactly(addr data[0], dataLenVarint.int)
trace "read data", data = data trace "read data", len = data.len
let header = headerVarint let header = headerVarint
result = (header shr 3, MessageType(header and 0x7), data) result = (header shr 3, MessageType(header and 0x7), data)
@ -66,6 +67,9 @@ proc writeMsg*(conn: Connection,
id: uint, id: uint,
msgType: MessageType, msgType: MessageType,
data: seq[byte] = @[]) {.async, gcsafe.} = data: seq[byte] = @[]) {.async, gcsafe.} =
trace "seding data over mplex", id,
msgType,
data
## write lenght prefixed ## write lenght prefixed
var buf = initVBuffer() var buf = initVBuffer()
buf.writePBVarint(id shl 3 or ord(msgType).uint) buf.writePBVarint(id shl 3 or ord(msgType).uint)