mirror of https://github.com/vacp2p/nim-libp2p.git
85 lines
2.5 KiB
Nim
85 lines
2.5 KiB
Nim
## Nim-LibP2P
|
|
## Copyright (c) 2019 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, tables
|
|
import chronos, chronicles
|
|
import ../connection,
|
|
../multiaddress,
|
|
../multicodec,
|
|
../errors
|
|
|
|
type
|
|
ConnHandler* = proc (conn: Connection): Future[void] {.gcsafe.}
|
|
|
|
TransportFlag* {.pure.} = enum
|
|
ReuseAddr
|
|
|
|
TransportFlags* = set[TransportFlag]
|
|
|
|
Transport* = ref object of RootObj
|
|
ma*: Multiaddress
|
|
connections*: seq[Connection]
|
|
handler*: ConnHandler
|
|
multicodec*: MultiCodec
|
|
flags*: TransportFlags
|
|
|
|
proc transportFlagsToServerFlags*(flags: TransportFlags): set[ServerFlags] {.gcsafe.} =
|
|
let transportFlagToServerFlagMapping = {
|
|
TransportFlag.ReuseAddr: ServerFlags.ReuseAddr,
|
|
}.toTable()
|
|
|
|
for flag in flags:
|
|
result.incl(transportFlagToServerFlagMapping[flag])
|
|
|
|
method init*(t: Transport) {.base, gcsafe.} =
|
|
## perform protocol initialization
|
|
discard
|
|
|
|
proc newTransport*(t: typedesc[Transport], flags: TransportFlags = {}): t {.gcsafe.} =
|
|
new result
|
|
result.flags = flags
|
|
result.init()
|
|
|
|
method close*(t: Transport) {.base, async, gcsafe.} =
|
|
## stop and cleanup the transport
|
|
## including all outstanding connections
|
|
let futs = await allFinished(t.connections.mapIt(it.close()))
|
|
checkFutures(futs)
|
|
|
|
method listen*(t: Transport,
|
|
ma: MultiAddress,
|
|
handler: ConnHandler):
|
|
Future[Future[void]] {.base, async, gcsafe.} =
|
|
## listen for incoming connections
|
|
t.ma = ma
|
|
t.handler = handler
|
|
trace "starting node", address = $ma
|
|
|
|
method dial*(t: Transport,
|
|
address: MultiAddress):
|
|
Future[Connection] {.base, async, gcsafe.} =
|
|
## dial a peer
|
|
discard
|
|
|
|
method upgrade*(t: Transport) {.base, async, gcsafe.} =
|
|
## base upgrade method that the transport uses to perform
|
|
## transport specific upgrades
|
|
discard
|
|
|
|
method handles*(t: Transport, address: MultiAddress): bool {.base, gcsafe.} =
|
|
## check if transport supportes the multiaddress
|
|
|
|
# by default we skip circuit addresses to avoid
|
|
# having to repeat the check in every transport
|
|
address.protocols.filterIt( it == multiCodec("p2p-circuit") ).len == 0
|
|
|
|
method localAddress*(t: Transport): MultiAddress {.base, gcsafe.} =
|
|
## get the local address of the transport in case started with 0.0.0.0:0
|
|
discard
|