@libp2p/upnp-nat 2.0.11 → 2.0.12-339b7df88

package/src/upnp-port-mapper.ts

@@ -0,0 +1,223 @@
+import { isIPv4, isIPv6 } from '@chainsafe/is-ip'
+import { InvalidParametersError, start, stop } from '@libp2p/interface'
+import { isLoopback } from '@libp2p/utils/multiaddr/is-loopback'
+import { isPrivate } from '@libp2p/utils/multiaddr/is-private'
+import { isPrivateIp } from '@libp2p/utils/private-ip'
+import { QUICV1, TCP, WebSockets, WebSocketsSecure, WebTransport } from '@multiformats/multiaddr-matcher'
+import { dynamicExternalAddress } from './check-external-address.js'
+import { DoubleNATError, InvalidIPAddressError } from './errors.js'
+import type { ExternalAddress } from './check-external-address.js'
+import type { Gateway } from '@achingbrain/nat-port-mapper'
+import type { ComponentLogger, Logger } from '@libp2p/interface'
+import type { AddressManager } from '@libp2p/interface-internal'
+import type { Multiaddr } from '@multiformats/multiaddr'
+
+export interface UPnPPortMapperInit {
+  gateway: Gateway
+  externalAddressCheckInterval?: number
+  externalAddressCheckTimeout?: number
+}
+
+export interface UPnPPortMapperComponents {
+  logger: ComponentLogger
+  addressManager: AddressManager
+}
+
+interface PortMapping {
+  externalHost: string
+  externalPort: number
+}
+
+export class UPnPPortMapper {
+  private readonly gateway: Gateway
+  private readonly externalAddress: ExternalAddress
+  private readonly addressManager: AddressManager
+  private readonly log: Logger
+  private readonly mappedPorts: Map<string, PortMapping>
+  private started: boolean
+
+  constructor (components: UPnPPortMapperComponents, init: UPnPPortMapperInit) {
+    this.log = components.logger.forComponent(`libp2p:upnp-nat:gateway:${init.gateway.id}`)
+    this.addressManager = components.addressManager
+    this.gateway = init.gateway
+    this.externalAddress = dynamicExternalAddress({
+      gateway: this.gateway,
+      addressManager: this.addressManager,
+      logger: components.logger
+    }, {
+      interval: init.externalAddressCheckInterval,
+      timeout: init.externalAddressCheckTimeout,
+      onExternalAddressChange: this.remapPorts.bind(this)
+    })
+    this.gateway = init.gateway
+    this.mappedPorts = new Map()
+    this.started = false
+  }
+
+  async start (): Promise<void> {
+    if (this.started) {
+      return
+    }
+
+    await start(this.externalAddress)
+    this.started = true
+  }
+
+  async stop (): Promise<void> {
+    try {
+      const shutdownTimeout = AbortSignal.timeout(1000)
+
+      await this.gateway.stop({
+        signal: shutdownTimeout
+      })
+    } catch (err: any) {
+      this.log.error('error closing gateway - %e', err)
+    }
+
+    await stop(this.externalAddress)
+    this.started = false
+  }
+
+  /**
+   * Update the local address mappings when the gateway's external interface
+   * address changes
+   */
+  private remapPorts (newExternalHost: string): void {
+    for (const [key, { externalHost, externalPort }] of this.mappedPorts.entries()) {
+      const [
+        host,
+        port,
+        transport
+      ] = key.split('-')
+
+      this.addressManager.removePublicAddressMapping(host, parseInt(port), externalHost, externalPort, transport === 'tcp' ? 'tcp' : 'udp')
+      this.addressManager.addPublicAddressMapping(host, parseInt(port), newExternalHost, externalPort, transport === 'tcp' ? 'tcp' : 'udp')
+    }
+  }
+
+  /**
+   * Return any eligible multiaddrs that are not mapped on the detected gateway
+   */
+  private getUnmappedAddresses (multiaddrs: Multiaddr[], ipType: 4 | 6): Multiaddr[] {
+    const output: Multiaddr[] = []
+
+    for (const ma of multiaddrs) {
+      // ignore public addresses
+      if (!isPrivate(ma)) {
+        continue
+      }
+
+      // ignore loopback
+      if (isLoopback(ma)) {
+        continue
+      }
+
+      // only IP based addresses
+      if (!(
+        TCP.exactMatch(ma) ||
+        WebSockets.exactMatch(ma) ||
+        WebSocketsSecure.exactMatch(ma) ||
+        QUICV1.exactMatch(ma) ||
+        WebTransport.exactMatch(ma)
+      )) {
+        continue
+      }
+
+      const { port, host, family, transport } = ma.toOptions()
+
+      if (family !== ipType) {
+        continue
+      }
+
+      if (this.mappedPorts.has(`${host}-${port}-${transport}`)) {
+        continue
+      }
+
+      output.push(ma)
+    }
+
+    return output
+  }
+
+  async mapIpAddresses (): Promise<void> {
+    try {
+      const externalHost = await this.externalAddress.getPublicIp()
+
+      let ipType: 4 | 6 = 4
+
+      if (isIPv4(externalHost)) {
+        ipType = 4
+      } else if (isIPv6(externalHost)) {
+        ipType = 6
+      } else {
+        throw new InvalidIPAddressError(`Public address ${externalHost} was not an IPv4 address`)
+      }
+
+      // filter addresses to get private, non-relay, IP based addresses that we
+      // haven't mapped yet
+      const addresses = this.getUnmappedAddresses(this.addressManager.getAddresses(), ipType)
+
+      if (addresses.length === 0) {
+        this.log('no private, non-relay, unmapped, IP based addresses found')
+        return
+      }
+
+      this.log('%s public IP %s', this.externalAddress != null ? 'using configured' : 'discovered', externalHost)
+
+      this.assertNotBehindDoubleNAT(externalHost)
+
+      for (const addr of addresses) {
+        // try to open uPnP ports for each thin waist address
+        const { family, host, port, transport } = addr.toOptions()
+
+        if (family === 6) {
+          // only support IPv4 addresses
+          continue
+        }
+
+        if (this.mappedPorts.has(`${host}-${port}-${transport}`)) {
+          // already mapped this port
+          continue
+        }
+
+        try {
+          const key = `${host}-${port}-${transport}`
+          this.log('creating mapping of key %s', key)
+
+          const externalPort = await this.gateway.map(port, {
+            localAddress: host,
+            protocol: transport === 'tcp' ? 'tcp' : 'udp'
+          })
+
+          this.mappedPorts.set(key, {
+            externalHost,
+            externalPort
+          })
+
+          this.log('created mapping of %s:%s to %s:%s', externalHost, externalPort, host, port)
+
+          this.addressManager.addPublicAddressMapping(host, port, externalHost, externalPort, transport === 'tcp' ? 'tcp' : 'udp')
+        } catch (err) {
+          this.log.error('failed to create mapping of %s:%s - %e', host, port, err)
+        }
+      }
+    } catch (err: any) {
+      this.log.error('error finding gateways - %e', err)
+    }
+  }
+
+  /**
+   * Some ISPs have double-NATs, there's not much we can do with them
+   */
+  private assertNotBehindDoubleNAT (publicIp: string): void {
+    const isPrivate = isPrivateIp(publicIp)
+
+    if (isPrivate === true) {
+      throw new DoubleNATError(`${publicIp} is private - please init uPnPNAT with 'externalAddress' set to an externally routable IP or ensure you are not behind a double NAT`)
+    }
+
+    if (isPrivate == null) {
+      throw new InvalidParametersError(`${publicIp} is not an IP address`)
+    }
+  }
+}

package/LICENSE

@@ -1,4 +0,0 @@
-This project is dual licensed under MIT and Apache-2.0.
-
-MIT: https://www.opensource.org/licenses/mit
-Apache-2.0: https://www.apache.org/licenses/license-2.0

package/dist/typedoc-urls.json

@@ -1,12 +0,0 @@
-{
-  "PMPOptions": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.PMPOptions.html",
-  ".:PMPOptions": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.PMPOptions.html",
-  "UPnPNAT": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNAT.html",
-  ".:UPnPNAT": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNAT.html",
-  "UPnPNATComponents": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNATComponents.html",
-  ".:UPnPNATComponents": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNATComponents.html",
-  "UPnPNATInit": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNATInit.html",
-  ".:UPnPNATInit": "https://libp2p.github.io/js-libp2p/interfaces/_libp2p_upnp_nat.UPnPNATInit.html",
-  "uPnPNAT": "https://libp2p.github.io/js-libp2p/functions/_libp2p_upnp_nat.uPnPNAT-1.html",
-  ".:uPnPNAT": "https://libp2p.github.io/js-libp2p/functions/_libp2p_upnp_nat.uPnPNAT-1.html"
-}