]> Sergey Matveev's repositories - btrtrc.git/blob - portfwd.go
Switch Go CI to go1.20
[btrtrc.git] / 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 }