fix: add option to use fixed port for BehaviorTest (#48)

This commit is contained in:
Anya Lin
2023-09-09 05:30:42 +08:00
committed by GitHub
parent 293c9ea4f2
commit 0f417a9a49

View File

@@ -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
}