174 lines
6.3 KiB
Nim
174 lines
6.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.
|
|
|
|
## This module implements wire network connection procedures.
|
|
import chronos, stew/endians2
|
|
import multiaddress, multicodec
|
|
|
|
when defined(windows):
|
|
import winlean
|
|
else:
|
|
import posix
|
|
|
|
const
|
|
TRANSPMA* = mapOr(
|
|
mapAnd(IP, mapEq("udp")),
|
|
mapAnd(IP, mapEq("tcp")),
|
|
mapAnd(mapEq("unix"))
|
|
)
|
|
|
|
RTRANSPMA* = mapOr(
|
|
mapAnd(IP, mapEq("tcp")),
|
|
mapAnd(mapEq("unix"))
|
|
)
|
|
|
|
proc initTAddress*(ma: MultiAddress): MaResult[TransportAddress] {.
|
|
raises: [Defect, ResultError[string]] .} =
|
|
## Initialize ``TransportAddress`` with MultiAddress ``ma``.
|
|
##
|
|
## MultiAddress must be wire address, e.g. ``{IP4, IP6, UNIX}/{TCP, UDP}``.
|
|
if TRANSPMA.match(ma):
|
|
var pbuf: array[2, byte]
|
|
let code = ma[0].tryGet().protoCode().tryGet()
|
|
if code == multiCodec("unix"):
|
|
var res = TransportAddress(family: AddressFamily.Unix)
|
|
if ma[0].tryGet().protoArgument(res.address_un).tryGet() == 0:
|
|
err("Incorrect Unix domain address")
|
|
else:
|
|
res.port = Port(1)
|
|
ok(res)
|
|
elif code == multiCodec("ip4"):
|
|
var res = TransportAddress(family: AddressFamily.IPv4)
|
|
if ma[0].tryGet().protoArgument(res.address_v4).tryGet() == 0:
|
|
err("Incorrect IPv4 address")
|
|
else:
|
|
if ma[1].tryGet().protoArgument(pbuf).tryGet() == 0:
|
|
err("Incorrect port number")
|
|
else:
|
|
res.port = Port(fromBytesBE(uint16, pbuf))
|
|
ok(res)
|
|
else:
|
|
var res = TransportAddress(family: AddressFamily.IPv6)
|
|
if ma[0].tryGet().protoArgument(res.address_v6).tryGet() == 0:
|
|
err("Incorrect IPv6 address")
|
|
else:
|
|
if ma[1].tryGet().protoArgument(pbuf).tryGet() == 0:
|
|
err("Incorrect port number")
|
|
else:
|
|
res.port = Port(fromBytesBE(uint16, pbuf))
|
|
ok(res)
|
|
else:
|
|
err("MultiAddress must be wire address (tcp, udp or unix)")
|
|
|
|
proc connect*(ma: MultiAddress, bufferSize = DefaultStreamBufferSize,
|
|
child: StreamTransport = nil): Future[StreamTransport] {.async.} =
|
|
## Open new connection to remote peer with address ``ma`` and create
|
|
## new transport object ``StreamTransport`` for established connection.
|
|
## ``bufferSize`` is size of internal buffer for transport.
|
|
if not(RTRANSPMA.match(ma)):
|
|
raise newException(MaInvalidAddress, "Incorrect or unsupported address!")
|
|
|
|
let address = initTAddress(ma).tryGet()
|
|
result = await connect(address, bufferSize, child)
|
|
|
|
proc createStreamServer*[T](ma: MultiAddress,
|
|
cbproc: StreamCallback,
|
|
flags: set[ServerFlags] = {},
|
|
udata: ref T,
|
|
sock: AsyncFD = asyncInvalidSocket,
|
|
backlog: int = 100,
|
|
bufferSize: int = DefaultStreamBufferSize,
|
|
child: StreamServer = nil,
|
|
init: TransportInitCallback = nil): StreamServer =
|
|
## Create new TCP stream server which bounds to ``ma`` address.
|
|
if not(RTRANSPMA.match(ma)):
|
|
raise newException(MaInvalidAddress, "Incorrect or unsupported address!")
|
|
|
|
let address = initTAddress(ma)
|
|
result = createStreamServer(address.tryGet(), cbproc, flags, udata, sock,
|
|
backlog, bufferSize, child, init)
|
|
|
|
proc createStreamServer*[T](ma: MultiAddress,
|
|
flags: set[ServerFlags] = {},
|
|
udata: ref T,
|
|
sock: AsyncFD = asyncInvalidSocket,
|
|
backlog: int = 100,
|
|
bufferSize: int = DefaultStreamBufferSize,
|
|
child: StreamServer = nil,
|
|
init: TransportInitCallback = nil): StreamServer =
|
|
## Create new TCP stream server which bounds to ``ma`` address.
|
|
if not(RTRANSPMA.match(ma)):
|
|
raise newException(MaInvalidAddress, "Incorrect or unsupported address!")
|
|
|
|
let address = initTAddress(ma)
|
|
result = createStreamServer(address.tryGet(), flags, udata, sock, backlog,
|
|
bufferSize, child, init)
|
|
|
|
proc createAsyncSocket*(ma: MultiAddress): AsyncFD =
|
|
## Create new asynchronous socket using MultiAddress' ``ma`` socket type and
|
|
## protocol information.
|
|
##
|
|
## Returns ``asyncInvalidSocket`` on error.
|
|
##
|
|
## Note: This procedure only used in `go-libp2p-daemon` wrapper.
|
|
var
|
|
socktype: SockType = SockType.SOCK_STREAM
|
|
protocol: Protocol = Protocol.IPPROTO_TCP
|
|
|
|
let maddr = initTAddress(ma)
|
|
if maddr.isErr():
|
|
return asyncInvalidSocket
|
|
|
|
let address = maddr.tryGet()
|
|
|
|
if address.family in {AddressFamily.IPv4, AddressFamily.IPv6}:
|
|
if ma[1].tryGet().protoCode().tryGet() == multiCodec("udp"):
|
|
socktype = SockType.SOCK_DGRAM
|
|
protocol = Protocol.IPPROTO_UDP
|
|
elif ma[1].tryGet().protoCode().tryGet() == multiCodec("tcp"):
|
|
socktype = SockType.SOCK_STREAM
|
|
protocol = Protocol.IPPROTO_TCP
|
|
elif address.family in {AddressFamily.Unix}:
|
|
socktype = SockType.SOCK_STREAM
|
|
protocol = cast[Protocol](0)
|
|
else:
|
|
return asyncInvalidSocket
|
|
result = createAsyncSocket(address.getDomain(), socktype, protocol)
|
|
|
|
proc bindAsyncSocket*(sock: AsyncFD, ma: MultiAddress): bool =
|
|
## Bind socket ``sock`` to MultiAddress ``ma``.
|
|
##
|
|
## Note: This procedure only used in `go-libp2p-daemon` wrapper.
|
|
var
|
|
saddr: Sockaddr_storage
|
|
slen: SockLen
|
|
|
|
let maddr = initTAddress(ma)
|
|
if maddr.isErr():
|
|
return false
|
|
|
|
let address = maddr.tryGet()
|
|
toSAddr(address, saddr, slen)
|
|
if bindSocket(SocketHandle(sock), cast[ptr SockAddr](addr saddr),
|
|
slen) == 0:
|
|
result = true
|
|
else:
|
|
result = false
|
|
|
|
proc getLocalAddress*(sock: AsyncFD): TransportAddress =
|
|
## Retrieve local socket ``sock`` address.
|
|
##
|
|
## Note: This procedure only used in `go-libp2p-daemon` wrapper.
|
|
var saddr: Sockaddr_storage
|
|
var slen = SockLen(sizeof(Sockaddr_storage))
|
|
|
|
if getsockname(SocketHandle(sock), cast[ptr SockAddr](addr saddr),
|
|
addr slen) == 0:
|
|
fromSAddr(addr saddr, slen, result)
|