1 files changed, 9 insertions, 5 deletions
diff --git a/feature/portmapper/portmapper.go b/feature/portmapper/portmapper.go
index d1b903cb6..9c9d6acfc 100644
--- a/feature/portmapper/portmapper.go
+++ b/feature/portmapper/portmapper.go
@@ -6,6 +6,8 @@
 package portmapper
 
 import (
+	"context"
+
 	"tailscale.com/feature"
 	"tailscale.com/net/netmon"
 	"tailscale.com/net/portmapper"
@@ -20,16 +22,18 @@ func init() {
 }
 
 func newPortMapper(
+	ctx context.Context,
 	logf logger.Logf,
 	bus *eventbus.Bus,
 	netMon *netmon.Monitor,
 	disableUPnPOrNil func() bool,
-	onlyTCP443OrNil func() bool) portmappertype.Client {
-
+	onlyTCP443OrNil func() bool,
+) portmappertype.Client {
 	pm := portmapper.NewClient(portmapper.Config{
-		EventBus: bus,
-		Logf:     logf,
-		NetMon:   netMon,
+		ShutdownCtx: ctx,
+		EventBus:    bus,
+		Logf:        logf,
+		NetMon:      netMon,
 		DebugKnobs: &portmapper.DebugKnobs{
 			DisableAll:      onlyTCP443OrNil,
 			DisableUPnPFunc: disableUPnPOrNil,