mirror of
https://github.com/netbirdio/ice.git
synced 2026-05-22 17:10:58 -07:00
dinvladandSean DuBois
77cc354d7f
Currently, when using UDPMuxDefault with unspecified address, the loopback address is included by default, but agentConfig.IncludeLoopback is not respected when gathering local candidates. The same holds true when UDPMuxDefault is configured with a loopback address, but agentConfig.IncludeLoopback is not explicitly set to true. This commit adds an extra check to gatherCandidatesLocalUDPMux() for respecting that setting in both cases.
101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
// SPDX-FileCopyrightText: 2023 The Pion community <https://pion.ly>
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
//go:build !js
|
|
// +build !js
|
|
|
|
package ice
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pion/logging"
|
|
"github.com/pion/transport/v3/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestMuxAgent is an end to end test over UDP mux, ensuring two agents could connect over mux
|
|
func TestMuxAgent(t *testing.T) {
|
|
report := test.CheckRoutines(t)
|
|
defer report()
|
|
|
|
lim := test.TimeOut(time.Second * 30)
|
|
defer lim.Stop()
|
|
|
|
const muxPort = 7686
|
|
|
|
caseAddrs := map[string]*net.UDPAddr{
|
|
"unspecified": {Port: muxPort},
|
|
"ipv4Loopback": {IP: net.IPv4(127, 0, 0, 1), Port: muxPort},
|
|
}
|
|
|
|
for subTest, addr := range caseAddrs {
|
|
muxAddr := addr
|
|
t.Run(subTest, func(t *testing.T) {
|
|
c, err := net.ListenUDP("udp", muxAddr)
|
|
require.NoError(t, err)
|
|
|
|
loggerFactory := logging.NewDefaultLoggerFactory()
|
|
udpMux := NewUDPMuxDefault(UDPMuxParams{
|
|
Logger: loggerFactory.NewLogger("ice"),
|
|
UDPConn: c,
|
|
})
|
|
|
|
muxedA, err := NewAgent(&AgentConfig{
|
|
UDPMux: udpMux,
|
|
CandidateTypes: []CandidateType{CandidateTypeHost},
|
|
NetworkTypes: []NetworkType{
|
|
NetworkTypeUDP4,
|
|
},
|
|
IncludeLoopback: addr.IP.IsLoopback(),
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
a, err := NewAgent(&AgentConfig{
|
|
CandidateTypes: []CandidateType{CandidateTypeHost},
|
|
NetworkTypes: supportedNetworkTypes(),
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
conn, muxedConn := connect(a, muxedA)
|
|
|
|
pair := muxedA.getSelectedPair()
|
|
require.NotNil(t, pair)
|
|
require.Equal(t, muxPort, pair.Local.Port())
|
|
|
|
// Send a packet to Mux
|
|
data := []byte("hello world")
|
|
_, err = conn.Write(data)
|
|
require.NoError(t, err)
|
|
|
|
buf := make([]byte, 1024)
|
|
n, err := muxedConn.Read(buf)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, buf[:n])
|
|
|
|
// Send a packet from Mux
|
|
_, err = muxedConn.Write(data)
|
|
require.NoError(t, err)
|
|
|
|
n, err = conn.Read(buf)
|
|
require.NoError(t, err)
|
|
require.Equal(t, data, buf[:n])
|
|
|
|
// Close it down
|
|
require.NoError(t, conn.Close())
|
|
require.NoError(t, muxedConn.Close())
|
|
require.NoError(t, udpMux.Close())
|
|
|
|
// Expect error when reading from closed mux
|
|
_, err = muxedConn.Read(data)
|
|
require.Error(t, err)
|
|
|
|
// Expect error when writing to closed mux
|
|
_, err = muxedConn.Write(data)
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
}
|