mirror of https://github.com/vacp2p/nim-libp2p.git
70 lines
2.3 KiB
Nim
70 lines
2.3 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
|
|
|
|
type
|
|
Switch* = ref object of RootObj
|
|
peerInfo*: PeerInfo
|
|
connections*: TableRef[string, Connection]
|
|
transports*: seq[Transport]
|
|
protocols*: seq[LPProtocol]
|
|
ms*: MultisteamSelect
|
|
|
|
proc newSwitch*(peerInfo: PeerInfo, transports: seq[Transport]): Switch =
|
|
new result
|
|
result.peerInfo = peerInfo
|
|
result.ms = newMultistream()
|
|
result.transports = transports
|
|
result.connections = newTable[string, Connection]()
|
|
result.protocols = newSeq[LPProtocol]()
|
|
|
|
proc secure(s: Switch, conn: Connection) {.async, gcsafe.} =
|
|
## secure the incoming connection
|
|
discard
|
|
|
|
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.secure(conn) # secure the connection
|
|
await s.ms.handle(conn) # fire up the ms handler
|
|
|
|
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()))
|