]> Sergey Matveev's repositories - btrtrc.git/blob - portfwd.go
sortimports
[btrtrc.git] / portfwd.go
1 package torrent
2
3 import (
4         "log"
5         "time"
6
7         flog "github.com/anacrolix/log"
8         "github.com/syncthing/syncthing/lib/nat"
9         "github.com/syncthing/syncthing/lib/upnp"
10 )
11
12 func addPortMapping(d nat.Device, proto nat.Protocol, internalPort int, debug bool) {
13         externalPort, err := d.AddPortMapping(proto, internalPort, internalPort, "anacrolix/torrent", 0)
14         if err != nil {
15                 log.Printf("error adding %s port mapping: %s", proto, err)
16         } else if externalPort != internalPort {
17                 log.Printf("external port %d does not match internal port %d in port mapping", externalPort, internalPort)
18         } else if debug {
19                 log.Printf("forwarded external %s port %d", proto, externalPort)
20         }
21 }
22
23 func (cl *Client) forwardPort() {
24         cl.mu.Lock()
25         defer cl.mu.Unlock()
26         if cl.config.NoDefaultPortForwarding {
27                 return
28         }
29         cl.mu.Unlock()
30         ds := upnp.Discover(0, 2*time.Second)
31         cl.mu.Lock()
32         flog.Default.Handle(flog.Fmsg("discovered %d upnp devices", len(ds)))
33         port := cl.incomingPeerPort()
34         cl.mu.Unlock()
35         for _, d := range ds {
36                 go addPortMapping(d, nat.TCP, port, cl.config.Debug)
37                 go addPortMapping(d, nat.UDP, port, cl.config.Debug)
38         }
39         cl.mu.Lock()
40 }