Track SSH connections

kylecarbs · kylecarbs · commit 29983eba0edd · 2023-01-09T19:49:59.000-06:00
diff --git a/agent/agent.go b/agent/agent.go
@@ -304,7 +304,9 @@ func (a *agent) createTailnet(ctx context.Context, derpMap *tailcfg.DERPMap) (_
 			if err != nil {
 				return
 			}
-			go a.sshServer.HandleConn(conn)
+			_ = a.trackConnGoroutine(func() {
+				a.sshServer.HandleConn(conn)
+			})
 		}
 	}); err != nil {
 		return nil, err

Original file line number	Diff line number	Diff line change
`@@ -304,7 +304,9 @@ func (a agent) createTailnet(ctx context.Context, derpMap tailcfg.DERPMap) (_`
`304`	`304`	`if err != nil {`
`305`	`305`	`return`
`306`	`306`	`}`
`307`		`- go a.sshServer.HandleConn(conn)`
	`307`	`+ _ = a.trackConnGoroutine(func() {`
	`308`	`+ a.sshServer.HandleConn(conn)`
	`309`	`+ })`
`308`	`310`	`}`
`309`	`311`	`}); err != nil {`
`310`	`312`	`return nil, err`