portfwd.go

   1 package torrent
   2
   3 import (
   4         "time"
   5
   6         "github.com/anacrolix/log"
   7         "github.com/elgatito/upnp"
   8 )
   9
  10 func addPortMapping(d upnp.Device, proto upnp.Protocol, internalPort int, debug bool) {
  11         externalPort, err := d.AddPortMapping(proto, internalPort, internalPort, "anacrolix/torrent", 0)
  12         if err != nil {
  13                 log.Printf("error adding %s port mapping: %s", proto, err)
  14         } else if externalPort != internalPort {
  15                 log.Printf("external port %d does not match internal port %d in port mapping", externalPort, internalPort)
  16         } else if debug {
  17                 log.Printf("forwarded external %s port %d", proto, externalPort)
  18         }
  19 }
  20
  21 func (cl *Client) forwardPort() {
  22         cl.lock()
  23         defer cl.unlock()
  24         if cl.config.NoDefaultPortForwarding {
  25                 return
  26         }
  27         cl.unlock()
  28         ds := upnp.Discover(0, 2*time.Second)
  29         cl.lock()
  30         cl.logger.Printf("discovered %d upnp devices", len(ds))
  31         port := cl.incomingPeerPort()
  32         cl.unlock()
  33         for _, d := range ds {
  34                 go addPortMapping(d, upnp.TCP, port, cl.config.Debug)
  35                 go addPortMapping(d, upnp.UDP, port, cl.config.Debug)
  36         }
  37         cl.lock()
  38 }