rename handle, fix potential deadlock

Signed-off-by: Naman Sood <mail@nsood.in>

wgengine/userspace.go

@@ -120,7 +120,7 @@ type userspaceEngine struct {
 	pingers             map[wgkey.Key]*pinger // legacy pingers for pre-discovery peers
 	linkState           *interfaces.State
 	pendOpen            map[flowtrack.Tuple]*pendingOpenFlow // see pendopen.go
-	networkMapCallbacks map[*networkMapCallbackHandle]NetworkMapCallback
+	networkMapCallbacks map[*someHandle]NetworkMapCallback
 
 	// Lock ordering: magicsock.Conn.mu, wgLock, then mu.
 }
@@ -1282,14 +1282,14 @@ func (e *userspaceEngine) AddNetworkMapCallback(cb NetworkMapCallback) func() {
 	e.mu.Lock()
 	defer e.mu.Unlock()
 	if e.networkMapCallbacks == nil {
-		e.networkMapCallbacks = make(map[*networkMapCallbackHandle]NetworkMapCallback)
+		e.networkMapCallbacks = make(map[*someHandle]NetworkMapCallback)
 	}
-	handle := new(networkMapCallbackHandle)
-	e.networkMapCallbacks[handle] = cb
+	h := new(someHandle)
+	e.networkMapCallbacks[h] = cb
 	return func() {
 		e.mu.Lock()
 		defer e.mu.Unlock()
-		delete(e.networkMapCallbacks, handle)
+		delete(e.networkMapCallbacks, h)
 	}
 }
 
@@ -1312,8 +1312,9 @@ func (e *userspaceEngine) SetDERPMap(dm *tailcfg.DERPMap) {
 func (e *userspaceEngine) SetNetworkMap(nm *controlclient.NetworkMap) {
 	e.magicConn.SetNetworkMap(nm)
 	e.mu.Lock()
-	defer e.mu.Unlock()
-	for _, fn := range e.networkMapCallbacks {
+	callbacks := e.networkMapCallbacks
+	e.mu.Unlock()
+	for _, fn := range callbacks {
 		fn(nm)
 	}
 }

wgengine/wgengine.go

@@ -53,9 +53,10 @@ type NetInfoCallback func(*tailcfg.NetInfo)
 // into network map updates.
 type NetworkMapCallback func(*controlclient.NetworkMap)
 
-// networkMapCallbackHandle is a unique key used to keep track
-// of callback functions so they can be removed in the future.
-type networkMapCallbackHandle struct{ _ byte }
+// someHandle is allocated so its pointer address acts as a unique
+// map key handle. (It needs to have non-zero size for Go to guarantee
+// the pointer is unique.)
+type someHandle struct{ _ byte }
 
 // ErrNoChanges is returned by Engine.Reconfig if no changes were made.
 var ErrNoChanges = errors.New("no changes made to Engine config")