mirror of
https://github.com/logos-storage/logos-storage-nim.git
synced 2026-05-17 08:49:31 +00:00
132 lines
4.0 KiB
Nim
132 lines
4.0 KiB
Nim
# Copyright (c) 2019-2023 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.
|
|
|
|
{.push raises: [].}
|
|
|
|
import std/[options, net]
|
|
import results
|
|
|
|
import pkg/chronos
|
|
import pkg/chronicles
|
|
import pkg/libp2p
|
|
import pkg/libp2p/services/autorelayservice
|
|
|
|
import ./utils
|
|
import ./utils/natutils
|
|
import ./utils/addrutils
|
|
import ./discovery
|
|
|
|
logScope:
|
|
topics = "nat"
|
|
|
|
type NatConfig* = object
|
|
case hasExtIp*: bool
|
|
of true: extIp*: IpAddress
|
|
of false: nat*: NatStrategy
|
|
|
|
type NatMapper* = ref object of RootObj
|
|
natConfig*: NatConfig
|
|
tcpPort*: Port
|
|
discoveryPort*: Port
|
|
hasUpnpMapping: bool
|
|
|
|
method mapNatPorts*(m: NatMapper): Option[(Port, Port)] {.base, gcsafe, raises: [].} =
|
|
if m.natConfig.hasExtIp:
|
|
return none((Port, Port))
|
|
|
|
# Devices are recreated on each call: discover() costs ~200ms but only fires
|
|
# when AutoNAT reports NotReachable, which is exactly when we want a fresh scan.
|
|
let upnpRes = UpnpDevice.init()
|
|
if upnpRes.isOk:
|
|
let ports = upnpRes.value.mapPorts(m.tcpPort, m.discoveryPort)
|
|
if ports.isSome:
|
|
m.hasUpnpMapping = true
|
|
return ports
|
|
|
|
let pmpRes = PmpDevice.init()
|
|
if pmpRes.isOk:
|
|
let ports = pmpRes.value.mapPorts(m.tcpPort, m.discoveryPort)
|
|
if ports.isSome:
|
|
return ports
|
|
|
|
none((Port, Port))
|
|
|
|
method handleNatStatus*(
|
|
m: NatMapper,
|
|
networkReachability: NetworkReachability,
|
|
dialBackAddr: Opt[MultiAddress],
|
|
discoveryPort: Port,
|
|
discovery: Discovery,
|
|
switch: Switch,
|
|
autoRelayService: AutoRelayService,
|
|
) {.async: (raises: [CancelledError]), base, gcsafe.} =
|
|
case networkReachability
|
|
of Unknown:
|
|
discard
|
|
of Reachable:
|
|
if dialBackAddr.isNone:
|
|
warn "Got empty dialback address in AutoNat when node is Reachable"
|
|
return
|
|
|
|
if autoRelayService.isRunning:
|
|
if not await autoRelayService.stop(switch):
|
|
debug "AutoRelayService stop method returned false"
|
|
|
|
discovery.updateRecords(@[dialBackAddr.get], discoveryPort)
|
|
# TODO: switch DHT to server mode
|
|
of NotReachable:
|
|
var hasPortMapping = false
|
|
|
|
if dialBackAddr.isNone:
|
|
warn "Got empty dialback address in AutoNat when node is NotReachable"
|
|
else:
|
|
let maybePorts = m.mapNatPorts()
|
|
|
|
if maybePorts.isSome:
|
|
let (tcpPort, udpPort) = maybePorts.get()
|
|
let announceAddress = dialBackAddr.get.remapAddr(port = some(tcpPort))
|
|
|
|
# TODO: Try a dial me to make sure we are reachable
|
|
|
|
if autoRelayService.isRunning:
|
|
if not await autoRelayService.stop(switch):
|
|
debug "AutoRelayService stop method returned false"
|
|
|
|
discovery.updateRecords(@[announceAddress], udpPort)
|
|
hasPortMapping = true
|
|
|
|
if not hasPortMapping and not autoRelayService.isRunning:
|
|
if not await autoRelayService.setup(switch):
|
|
debug "AutoRelayService setup method returned false"
|
|
|
|
proc close*(m: NatMapper, device = UpnpDevice()) =
|
|
# UPnP mappings are permanent (leaseDuration=0) and must be deleted explicitly.
|
|
# NAT-PMP mappings expire automatically after NATPMP_LIFETIME seconds.
|
|
if not m.hasUpnpMapping:
|
|
return
|
|
|
|
# deletePortMapping requires the IGD control URL set during init
|
|
let deviceRes = device.init()
|
|
if deviceRes.isErr:
|
|
warn "UPnP reinit failed during cleanup, port mappings may remain",
|
|
msg = deviceRes.error
|
|
return
|
|
|
|
for (port, proto) in [
|
|
(m.tcpPort, NatIpProtocol.Tcp), (m.discoveryPort, NatIpProtocol.Udp)
|
|
]:
|
|
let res = deviceRes.value.deletePortMapping(port, proto)
|
|
if res.isErr:
|
|
error "UPnP port mapping deletion failed", port, proto, msg = res.error
|
|
|
|
proc findReachableNodes*(bootstrapNodes: seq[SignedPeerRecord]): seq[SignedPeerRecord] =
|
|
## Returns the list of nodes known to be directly reachable.
|
|
## Currently returns bootstrap nodes. In the future, any network participant
|
|
## confirmed reachable by AutoNAT could be included.
|
|
bootstrapNodes
|