mirror of https://github.com/waku-org/nwaku.git
77 lines
2.7 KiB
Nim
77 lines
2.7 KiB
Nim
# compile time options here
|
|
const
|
|
libp2p_secure {.strdefine.} = ""
|
|
libp2p_pubsub_sign {.booldefine.} = true
|
|
libp2p_pubsub_verify {.booldefine.} = true
|
|
|
|
import
|
|
options, tables, chronicles, chronos,
|
|
libp2p/[switch, peerinfo, multiaddress, crypto/crypto],
|
|
libp2p/stream/connection,
|
|
libp2p/transports/[transport, tcptransport],
|
|
libp2p/muxers/[muxer, mplex/mplex, mplex/types],
|
|
libp2p/protocols/[identify, secure/secure],
|
|
libp2p/protocols/pubsub/[pubsub, gossipsub],
|
|
../../protocol/v2/waku_protocol2
|
|
|
|
import
|
|
libp2p/protocols/secure/noise,
|
|
libp2p/protocols/secure/secio
|
|
|
|
export
|
|
switch, peerinfo, connection, multiaddress, crypto
|
|
|
|
type
|
|
SecureProtocol* {.pure.} = enum
|
|
Noise,
|
|
Secio
|
|
|
|
proc newStandardSwitch*(privKey = none(PrivateKey),
|
|
address = MultiAddress.init("/ip4/127.0.0.1/tcp/0").tryGet(),
|
|
triggerSelf = false,
|
|
gossip = false,
|
|
secureManagers: openarray[SecureProtocol] = [
|
|
# NOTE below relates to Eth2
|
|
# TODO investigate why we're getting fewer peers on public testnets with noise
|
|
SecureProtocol.Secio,
|
|
SecureProtocol.Noise, # array cos order matters
|
|
],
|
|
verifySignature = libp2p_pubsub_verify,
|
|
sign = libp2p_pubsub_sign,
|
|
transportFlags: set[ServerFlags] = {},
|
|
rng = newRng(),
|
|
inTimeout: Duration = 1.minutes,
|
|
outTimeout: Duration = 1.minutes): Switch =
|
|
info "newStandardSwitch"
|
|
proc createMplex(conn: Connection): Muxer =
|
|
Mplex.init(
|
|
conn,
|
|
inTimeout = inTimeout,
|
|
outTimeout = outTimeout)
|
|
|
|
let
|
|
seckey = privKey.get(otherwise = PrivateKey.random(ECDSA, rng[]).tryGet())
|
|
peerInfo = PeerInfo.init(seckey, [address])
|
|
mplexProvider = newMuxerProvider(createMplex, MplexCodec)
|
|
transports = @[Transport(TcpTransport.init(transportFlags))]
|
|
muxers = {MplexCodec: mplexProvider}.toTable
|
|
identify = newIdentify(peerInfo)
|
|
|
|
var
|
|
secureManagerInstances: seq[Secure]
|
|
for sec in secureManagers:
|
|
case sec
|
|
of SecureProtocol.Noise:
|
|
secureManagerInstances &= newNoise(rng, seckey).Secure
|
|
of SecureProtocol.Secio:
|
|
secureManagerInstances &= newSecio(rng, seckey).Secure
|
|
|
|
let pubSub = PubSub newPubSub(WakuSub, peerInfo, triggerSelf)
|
|
|
|
result = newSwitch(peerInfo,
|
|
transports,
|
|
identify,
|
|
muxers,
|
|
secureManagers = secureManagerInstances,
|
|
pubSub = some(pubSub))
|