Use netns only if EADDRINUSE occurred

daeuniverse · Jan 5, 2024 · 282c5f9 · 282c5f9
1 parent 4df5451
commit 282c5f9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 14 deletions.
diff --git a/control/anyfrom_pool.go b/control/anyfrom_pool.go
@@ -190,11 +190,7 @@ func (p *AnyfromPool) GetOrCreate(lAddr string, ttl time.Duration) (conn *Anyfro
 			},
 			KeepAlive: 0,
 		}
-		var pc net.PacketConn
-		err := WithIndieNetns(func() (err error) {
-			pc, err = d.ListenPacket(context.Background(), "udp", lAddr)
-			return err
-		})
+		pc, err := d.ListenPacket(context.Background(), "udp", lAddr)
 		if err != nil {
 			return nil, true, err
 		}

diff --git a/control/udp.go b/control/udp.go
@@ -90,17 +90,16 @@ func sendPkt(data []byte, from netip.AddrPort, realTo, to netip.AddrPort, lConn
 	}
 
 	uConn, _, err := DefaultAnyfromPool.GetOrCreate(from.String(), AnyfromTimeout)
-	if err != nil {
-		if errors.Is(err, syscall.EADDRINUSE) {
-			// Port collision, use traditional method.
-			return sendPktWithHdrWithFlag(data, from, lConn, to, lanWanFlag)
-		}
-		return err
+	if err != nil && errors.Is(err, syscall.EADDRINUSE) {
+		err = WithIndieNetns(func() (err error) {
+			uConn, _, err = DefaultAnyfromPool.GetOrCreate(from.String(), AnyfromTimeout)
+			return err
+		})
 	}
-	err = WithIndieNetns(func() (err error) {
-		_, err = uConn.WriteToUDPAddrPort(data, realTo)
+	if err != nil {
 		return
-	})
+	}
+	_, err = uConn.WriteToUDPAddrPort(data, realTo)
 	return err
 }