fix: add option to use fixed port for BehaviorTest

ccding · Sep 8, 2023 · fa3ef19 · fa3ef19
1 parent 293c9ea
commit fa3ef19
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/stun/client.go b/stun/client.go
@@ -126,7 +126,14 @@ func (c *Client) BehaviorTest() (*NATBehavior, error) {
 	// create a connection and close it at the end.
 	conn := c.conn
 	if conn == nil {
-		conn, err = net.ListenUDP("udp", nil)
+		var laddr *net.UDPAddr
+		if c.localPort != 0 {
+			laddr, err = net.ResolveUDPAddr("udp", fmt.Sprintf(":%d", c.localPort))
+			if err != nil {
+				return nil, err
+			}
+		}
+		conn, err = net.ListenUDP("udp", laddr)
 		if err != nil {
 			return nil, err
 		}