X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=portfwd.go;h=0136578223df0303f6c9a00725c53cfcc69fdace;hb=HEAD;hp=db5bb7df570c6002e4a16a58f438ea1c94d307ef;hpb=82e1c81a9ab271c76c164a5584e2f454534cd54d;p=btrtrc.git diff --git a/portfwd.go b/portfwd.go index db5bb7df..01365782 100644 --- a/portfwd.go +++ b/portfwd.go @@ -1,21 +1,27 @@ package torrent import ( + "fmt" "time" "github.com/anacrolix/log" "github.com/anacrolix/upnp" ) +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 { - cl.logger.WithValues(log.Warning).Printf("error adding %s port mapping: %s", proto, err) - } else if externalPort != internalPort { - cl.logger.WithValues(log.Warning).Printf("external port %d does not match internal port %d in port mapping", externalPort, internalPort) - } else { - cl.logger.WithValues(log.Info).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,9 +31,9 @@ func (cl *Client) forwardPort() { return } cl.unlock() - ds := upnp.Discover(0, 2*time.Second, cl.logger.WithValues("upnp-discover")) + 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()