mirror of https://github.com/vacp2p/nim-libp2p.git
100 lines
3.2 KiB
Nim
100 lines
3.2 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 tables, sequtils
|
|
import chronos
|
|
import connection, transport,
|
|
stream/lpstream,
|
|
multistream, protocol,
|
|
peerinfo, multiaddress,
|
|
identify, muxers/muxer
|
|
|
|
type
|
|
UnableToSecureError = object of CatchableError
|
|
UnableToIdentifyError = object of CatchableError
|
|
|
|
Switch* = ref object of RootObj
|
|
peerInfo*: PeerInfo
|
|
connections*: TableRef[string, Connection]
|
|
transports*: seq[Transport]
|
|
protocols*: seq[LPProtocol]
|
|
muxers*: seq[MuxerProvider]
|
|
ms*: MultisteamSelect
|
|
identity*: Identify
|
|
|
|
proc newSwitch*(peerInfo: PeerInfo,
|
|
transports: seq[Transport],
|
|
identity: Identify,
|
|
muxers: seq[MuxerProvider]): Switch =
|
|
new result
|
|
result.peerInfo = peerInfo
|
|
result.ms = newMultistream()
|
|
result.transports = transports
|
|
result.connections = newTable[string, Connection]()
|
|
result.identity = identity
|
|
result.muxers = muxers
|
|
|
|
result.ms.addHandler(IdentifyCodec, identity)
|
|
|
|
proc secure(s: Switch, conn: Connection) {.async, gcsafe.} =
|
|
## secure the incoming connection
|
|
discard
|
|
|
|
proc identify(s: Switch, conn: Connection, peerInfo: PeerInfo) {.async, gcsafe.} =
|
|
## identify the connection
|
|
s.peerInfo.protocols = s.ms.list() # update protos before engagin in identify
|
|
await s.identity.identify(conn)
|
|
|
|
proc mux(s: Switch, conn: Connection): Future[bool] {.async, gcsafe.} =
|
|
## mux incoming connection
|
|
result = true
|
|
|
|
proc handleConn(s: Switch, conn: Connection) {.async, gcsafe.} =
|
|
## perform upgrade flow
|
|
try:
|
|
result = s.ms.handle(conn) # handler incoming connection
|
|
await s.secure(conn)
|
|
if await s.mux(conn):
|
|
await s.identify(conn)
|
|
finally:
|
|
await conn.close()
|
|
|
|
proc dial*(s: Switch, peer: PeerInfo, proto: string = ""): Future[Connection] {.async.} =
|
|
for t in s.transports: # for each transport
|
|
for a in peer.addrs: # for each address
|
|
if t.handles(a): # check if it can dial it
|
|
result = await t.dial(a)
|
|
await s.secure(result)
|
|
if not await s.ms.select(result, proto):
|
|
raise newException(CatchableError,
|
|
"Unable to select protocol: " & proto)
|
|
|
|
proc mount*[T: LPProtocol](s: Switch, proto: T) =
|
|
if isNil(proto.handler):
|
|
raise newException(CatchableError,
|
|
"Protocol has to define a handle method or proc")
|
|
|
|
if len(proto.codec) <= 0:
|
|
raise newException(CatchableError,
|
|
"Protocol has to define a codec string")
|
|
|
|
s.ms.addHandler(proto.codec, proto)
|
|
|
|
proc start*(s: Switch) {.async.} =
|
|
proc handle(conn: Connection): Future[void] {.async, closure, gcsafe.} =
|
|
await s.handleConn(conn)
|
|
|
|
for t in s.transports: # for each transport
|
|
for a in s.peerInfo.addrs:
|
|
if t.handles(a): # check if it handles the multiaddr
|
|
await t.listen(a, handle) # listen for incoming connections
|
|
|
|
proc stop*(s: Switch) {.async.} =
|
|
await allFutures(s.transports.mapIt(it.close()))
|