mirror of
https://github.com/pion/ice.git
synced 2025-09-27 03:45:54 +08:00
0194bd6ce8
This change adapts pion/ice to use a new interface for most network related operations. The interface was formerly a simple struct vnet.Net which was originally intended to facilicate testing. By replacing it with an interface we have greater flexibility and allow users to hook into the networking stack by providing their own implementation of the interface.
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
package ice
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/pion/logging"
|
|
"github.com/pion/stun"
|
|
"github.com/pion/transport/v2/test"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var (
|
|
_ TCPMux = &TCPMuxDefault{}
|
|
_ TCPMux = &invalidTCPMux{}
|
|
)
|
|
|
|
func TestTCPMux_Recv(t *testing.T) {
|
|
for name, bufSize := range map[string]int{
|
|
"no buffer": 0,
|
|
"buffered 4MB": 4 * 1024 * 1024,
|
|
} {
|
|
bufSize := bufSize
|
|
t.Run(name, func(t *testing.T) {
|
|
report := test.CheckRoutines(t)
|
|
defer report()
|
|
|
|
loggerFactory := logging.NewDefaultLoggerFactory()
|
|
|
|
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
|
IP: net.IP{127, 0, 0, 1},
|
|
Port: 0,
|
|
})
|
|
require.NoError(t, err, "error starting listener")
|
|
defer func() {
|
|
_ = listener.Close()
|
|
}()
|
|
|
|
tcpMux := NewTCPMuxDefault(TCPMuxParams{
|
|
Listener: listener,
|
|
Logger: loggerFactory.NewLogger("ice"),
|
|
ReadBufferSize: 20,
|
|
WriteBufferSize: bufSize,
|
|
})
|
|
|
|
defer func() {
|
|
_ = tcpMux.Close()
|
|
}()
|
|
|
|
require.NotNil(t, tcpMux.LocalAddr(), "tcpMux.LocalAddr() is nil")
|
|
|
|
conn, err := net.DialTCP("tcp", nil, tcpMux.LocalAddr().(*net.TCPAddr))
|
|
require.NoError(t, err, "error dialing test tcp connection")
|
|
|
|
msg := stun.New()
|
|
msg.Type = stun.MessageType{Method: stun.MethodBinding, Class: stun.ClassRequest}
|
|
msg.Add(stun.AttrUsername, []byte("myufrag:otherufrag"))
|
|
msg.Encode()
|
|
|
|
n, err := writeStreamingPacket(conn, msg.Raw)
|
|
require.NoError(t, err, "error writing tcp stun packet")
|
|
|
|
pktConn, err := tcpMux.GetConnByUfrag("myufrag", false, listener.Addr().(*net.TCPAddr).IP)
|
|
require.NoError(t, err, "error retrieving muxed connection for ufrag")
|
|
defer func() {
|
|
_ = pktConn.Close()
|
|
}()
|
|
|
|
recv := make([]byte, n)
|
|
n2, rAddr, err := pktConn.ReadFrom(recv)
|
|
require.NoError(t, err, "error receiving data")
|
|
assert.Equal(t, conn.LocalAddr(), rAddr, "remote tcp address mismatch")
|
|
assert.Equal(t, n, n2, "received byte size mismatch")
|
|
assert.Equal(t, msg.Raw, recv, "received bytes mismatch")
|
|
|
|
// check echo response
|
|
n, err = pktConn.WriteTo(recv, conn.LocalAddr())
|
|
require.NoError(t, err, "error writing echo stun packet")
|
|
recvEcho := make([]byte, n)
|
|
n3, err := readStreamingPacket(conn, recvEcho)
|
|
require.NoError(t, err, "error receiving echo data")
|
|
assert.Equal(t, n2, n3, "received byte size mismatch")
|
|
assert.Equal(t, msg.Raw, recvEcho, "received bytes mismatch")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTCPMux_NoDeadlockWhenClosingUnusedPacketConn(t *testing.T) {
|
|
report := test.CheckRoutines(t)
|
|
defer report()
|
|
|
|
loggerFactory := logging.NewDefaultLoggerFactory()
|
|
|
|
listener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
|
IP: net.IP{127, 0, 0, 1},
|
|
Port: 0,
|
|
})
|
|
require.NoError(t, err, "error starting listener")
|
|
defer func() {
|
|
_ = listener.Close()
|
|
}()
|
|
|
|
tcpMux := NewTCPMuxDefault(TCPMuxParams{
|
|
Listener: listener,
|
|
Logger: loggerFactory.NewLogger("ice"),
|
|
ReadBufferSize: 20,
|
|
})
|
|
|
|
_, err = tcpMux.GetConnByUfrag("test", false, listener.Addr().(*net.TCPAddr).IP)
|
|
require.NoError(t, err, "error getting conn by ufrag")
|
|
|
|
require.NoError(t, tcpMux.Close(), "error closing tcpMux")
|
|
|
|
conn, err := tcpMux.GetConnByUfrag("test", false, listener.Addr().(*net.TCPAddr).IP)
|
|
assert.Nil(t, conn, "should receive nil because mux is closed")
|
|
assert.Equal(t, io.ErrClosedPipe, err, "should receive error because mux is closed")
|
|
}
|