139 lines
3.7 KiB
Nim
139 lines
3.7 KiB
Nim
## Nim-LibP2P
|
|
## Copyright (c) 2018 Status Research & Development GmbH
|
|
## Licensed under either of
|
|
## * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
## * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
## at your option.
|
|
## This file may not be copied, modified, or distributed except according to
|
|
## those terms.
|
|
|
|
import sequtils
|
|
import chronos, nimcrypto/sysrand
|
|
import ../../peerinfo,
|
|
../../peer,
|
|
../../crypto/crypto,
|
|
../../protobuf/minprotobuf
|
|
|
|
const SignPrefix = "libp2p-pubsub:"
|
|
|
|
type
|
|
SubOpts* = object
|
|
subscribe*: bool
|
|
topic*: string
|
|
|
|
Message* = object
|
|
fromPeer*: seq[byte]
|
|
data*: seq[byte]
|
|
seqno*: seq[byte]
|
|
topicIDs*: seq[string]
|
|
signature*: seq[byte]
|
|
key*: seq[byte]
|
|
|
|
RPCMsg* = object
|
|
subscriptions*: seq[SubOpts]
|
|
messages*: seq[Message]
|
|
|
|
proc encodeMessage(msg: Message, buff: var ProtoBuffer) {.gcsafe.} =
|
|
buff.write(initProtoField(1, msg.fromPeer))
|
|
buff.write(initProtoField(2, msg.data))
|
|
buff.write(initProtoField(3, msg.seqno))
|
|
|
|
for t in msg.topicIDs:
|
|
buff.write(initProtoField(4, t))
|
|
|
|
buff.write(initProtoField(5, msg.signature))
|
|
buff.write(initProtoField(6, msg.key))
|
|
|
|
buff.finish()
|
|
|
|
proc encodeSubs(subs: SubOpts, buff: var ProtoBuffer) {.gcsafe.} =
|
|
buff.write(initProtoField(1, ord(subs.subscribe)))
|
|
buff.write(initProtoField(2, subs.topic))
|
|
|
|
proc encodeRpcMsg*(msg: RPCMsg): ProtoBuffer {.gcsafe.} =
|
|
result = initProtoBuffer({WithVarintLength})
|
|
|
|
var subs = initProtoBuffer()
|
|
for s in msg.subscriptions:
|
|
encodeSubs(s, subs)
|
|
|
|
subs.finish()
|
|
result.write(initProtoField(1, subs))
|
|
|
|
var messages = initProtoBuffer()
|
|
for m in msg.messages:
|
|
encodeMessage(m, messages)
|
|
|
|
messages.finish()
|
|
result.write(initProtoField(2, messages))
|
|
|
|
result.finish()
|
|
|
|
proc decodeRpcMsg*(msg: seq[byte]): RPCMsg {.gcsafe.} =
|
|
var pb = initProtoBuffer(msg)
|
|
|
|
result.subscriptions = newSeq[SubOpts]()
|
|
var subscr = newSeq[byte](1)
|
|
|
|
# decode SubOpts array
|
|
if pb.enterSubMessage() > 0:
|
|
while true:
|
|
var subOpt: SubOpts
|
|
if pb.getBytes(1, subscr) < 0:
|
|
break
|
|
subOpt.subscribe = cast[bool](subscr)
|
|
|
|
if pb.getString(2, subOpt.topic) < 0:
|
|
break
|
|
result.subscriptions.add(subOpt)
|
|
|
|
result.messages = newSeq[Message]()
|
|
# TODO: which of this fields are really optional?
|
|
# Decode Messages array
|
|
if pb.enterSubMessage() > 0:
|
|
while true:
|
|
var msg: Message
|
|
if pb.getBytes(1, msg.fromPeer) < 0:
|
|
break
|
|
|
|
if pb.getBytes(2, msg.data) < 0:
|
|
break
|
|
|
|
if pb.getBytes(3, msg.seqno) < 0:
|
|
break
|
|
|
|
var topic: string
|
|
while true:
|
|
if pb.getString(4, topic) < 0:
|
|
break
|
|
topic.add(topic)
|
|
topic = ""
|
|
|
|
if pb.getBytes(5, msg.signature) < 0:
|
|
break
|
|
|
|
if pb.getBytes(6, msg.key) < 0:
|
|
break
|
|
|
|
proc sign*(peerId: PeerID, msg: Message): Message =
|
|
var buff = initProtoBuffer()
|
|
var prefix = cast[seq[byte]](toSeq(SignPrefix.items)) # TODO: can we cache this?
|
|
encodeMessage(msg, buff)
|
|
if buff.buffer.len > 0:
|
|
result = msg
|
|
result.signature = peerId.
|
|
privateKey.
|
|
sign(prefix & buff.buffer).
|
|
getBytes()
|
|
|
|
proc makeMessage*(peerId: PeerID, data: seq[byte], name: string): Message =
|
|
var seqno: seq[byte] = newSeq[byte](20)
|
|
if randomBytes(addr seqno[0], 20) > 0:
|
|
result = Message(fromPeer: peerId.getBytes(),
|
|
data: data,
|
|
seqno: seqno,
|
|
topicIDs: @[name],
|
|
signature: @[],
|
|
key: peerId.publicKey.getRawBytes())
|
|
result = sign(peerId, result)
|