X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=portfwd.go;h=0136578223df0303f6c9a00725c53cfcc69fdace;hb=3a92268f10184f5ed4602fa43d476d0ada96dc8e;hp=b8d70482aa80f21aa07c71df8af58a541796ef35;hpb=5a2c1c9b7654f9f0f27590e0cd3bcfa520096964;p=btrtrc.git

diff --git a/portfwd.go b/portfwd.go
index b8d70482..01365782 100644
--- a/portfwd.go
+++ b/portfwd.go
@@ -1,21 +1,27 @@
 package torrent
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/anacrolix/log"
-	"github.com/elgatito/upnp"
+	"github.com/anacrolix/upnp"
 )
 
-func addPortMapping(d upnp.Device, proto upnp.Protocol, internalPort int, debug bool) {
-	externalPort, err := d.AddPortMapping(proto, internalPort, internalPort, "anacrolix/torrent", 0)
+const UpnpDiscoverLogTag = "upnp-discover"
+
+func (cl *Client) addPortMapping(d upnp.Device, proto upnp.Protocol, internalPort int, upnpID string) {
+	logger := cl.logger.WithContextText(fmt.Sprintf("UPnP device at %v: mapping internal %v port %v", d.GetLocalIPAddress(), proto, internalPort))
+	externalPort, err := d.AddPortMapping(proto, internalPort, internalPort, upnpID, 0)
 	if err != nil {
-		log.Printf("error adding %s port mapping: %s", proto, err)
-	} else if externalPort != internalPort {
-		log.Printf("external port %d does not match internal port %d in port mapping", externalPort, internalPort)
-	} else if debug {
-		log.Printf("forwarded external %s port %d", proto, externalPort)
+		logger.WithDefaultLevel(log.Warning).Printf("error: %v", err)
+		return
+	}
+	level := log.Info
+	if externalPort != internalPort {
+		level = log.Warning
 	}
+	logger.WithDefaultLevel(level).Printf("success: external port %v", externalPort)
 }
 
 func (cl *Client) forwardPort() {
@@ -25,14 +31,15 @@ func (cl *Client) forwardPort() {
 		return
 	}
 	cl.unlock()
-	ds := upnp.Discover(0, 2*time.Second)
+	ds := upnp.Discover(0, 2*time.Second, cl.logger.WithValues(UpnpDiscoverLogTag))
 	cl.lock()
-	cl.logger.Printf("discovered %d upnp devices", len(ds))
+	cl.logger.WithDefaultLevel(log.Debug).Printf("discovered %d upnp devices", len(ds))
 	port := cl.incomingPeerPort()
+	id := cl.config.UpnpID
 	cl.unlock()
 	for _, d := range ds {
-		go addPortMapping(d, upnp.TCP, port, cl.config.Debug)
-		go addPortMapping(d, upnp.UDP, port, cl.config.Debug)
+		go cl.addPortMapping(d, upnp.TCP, port, id)
+		go cl.addPortMapping(d, upnp.UDP, port, id)
 	}
 	cl.lock()
 }