portfwd.go

   1 package torrent
   2
   3 import (
   4         "fmt"
   5         "time"
   6
   7         "github.com/anacrolix/log"
   8         "github.com/anacrolix/upnp"
   9 )
  10
  11 const UpnpDiscoverLogTag = "upnp-discover"
  12
  13 func (cl *Client) addPortMapping(d upnp.Device, proto upnp.Protocol, internalPort int, upnpID string) {
  14         logger := cl.logger.WithContextText(fmt.Sprintf("UPnP device at %v: mapping internal %v port %v", d.GetLocalIPAddress(), proto, internalPort))
  15         externalPort, err := d.AddPortMapping(proto, internalPort, internalPort, upnpID, 0)
  16         if err != nil {
  17                 logger.WithDefaultLevel(log.Warning).Printf("error: %v", err)
  18                 return
  19         }
  20         level := log.Info
  21         if externalPort != internalPort {
  22                 level = log.Warning
  23         }
  24         logger.WithDefaultLevel(level).Printf("success: external port %v", externalPort)
  25 }
  26
  27 func (cl *Client) forwardPort() {
  28         cl.lock()
  29         defer cl.unlock()
  30         if cl.config.NoDefaultPortForwarding {
  31                 return
  32         }
  33         cl.unlock()
  34         ds := upnp.Discover(0, 2*time.Second, cl.logger.WithValues(UpnpDiscoverLogTag))
  35         cl.lock()
  36         cl.logger.WithDefaultLevel(log.Debug).Printf("discovered %d upnp devices", len(ds))
  37         port := cl.incomingPeerPort()
  38         id := cl.config.UpnpID
  39         cl.unlock()
  40         for _, d := range ds {
  41                 go cl.addPortMapping(d, upnp.TCP, port, id)
  42                 go cl.addPortMapping(d, upnp.UDP, port, id)
  43         }
  44         cl.lock()
  45 }