This commit is contained in:
Viktor Liu
2025-07-24 13:35:30 +02:00
parent b2dcad828c
commit da623e6f04
2 changed files with 7 additions and 7 deletions

View File

@@ -7,7 +7,7 @@ import (
) )
func TestWGEBPFProxy_connStore(t *testing.T) { func TestWGEBPFProxy_connStore(t *testing.T) {
wgProxy := NewWGEBPFProxy(1) wgProxy := NewWGEBPFProxy(1, 1280)
p, _ := wgProxy.storeTurnConn(nil) p, _ := wgProxy.storeTurnConn(nil)
if p != 1 { if p != 1 {
@@ -27,7 +27,7 @@ func TestWGEBPFProxy_connStore(t *testing.T) {
} }
func TestWGEBPFProxy_portCalculation_overflow(t *testing.T) { func TestWGEBPFProxy_portCalculation_overflow(t *testing.T) {
wgProxy := NewWGEBPFProxy(1) wgProxy := NewWGEBPFProxy(1, 1280)
_, _ = wgProxy.storeTurnConn(nil) _, _ = wgProxy.storeTurnConn(nil)
wgProxy.lastUsedPort = 65535 wgProxy.lastUsedPort = 65535
@@ -43,7 +43,7 @@ func TestWGEBPFProxy_portCalculation_overflow(t *testing.T) {
} }
func TestWGEBPFProxy_portCalculation_maxConn(t *testing.T) { func TestWGEBPFProxy_portCalculation_maxConn(t *testing.T) {
wgProxy := NewWGEBPFProxy(1) wgProxy := NewWGEBPFProxy(1, 1280)
for i := 0; i < 65535; i++ { for i := 0; i < 65535; i++ {
_, _ = wgProxy.storeTurnConn(nil) _, _ = wgProxy.storeTurnConn(nil)

View File

@@ -21,7 +21,7 @@ func TestShouldReadSTUNOnReadFrom(t *testing.T) {
// create raw socket on a port // create raw socket on a port
testingPort := 51821 testingPort := 51821
rawSock, err := Listen(testingPort, NewIncomingSTUNFilter()) rawSock, err := Listen(testingPort, NewIncomingSTUNFilter(), 1280)
require.NoError(t, err, "received an error while creating STUN listener, error: %s", err) require.NoError(t, err, "received an error while creating STUN listener, error: %s", err)
err = rawSock.SetReadDeadline(time.Now().Add(3 * time.Second)) err = rawSock.SetReadDeadline(time.Now().Add(3 * time.Second))
require.NoError(t, err, "unable to set deadline, error: %s", err) require.NoError(t, err, "unable to set deadline, error: %s", err)
@@ -76,7 +76,7 @@ func TestShouldReadSTUNOnReadFrom(t *testing.T) {
func TestShouldNotReadNonSTUNPackets(t *testing.T) { func TestShouldNotReadNonSTUNPackets(t *testing.T) {
testingPort := 39439 testingPort := 39439
rawSock, err := Listen(testingPort, NewIncomingSTUNFilter()) rawSock, err := Listen(testingPort, NewIncomingSTUNFilter(), 1280)
require.NoError(t, err, "received an error while creating STUN listener, error: %s", err) require.NoError(t, err, "received an error while creating STUN listener, error: %s", err)
defer rawSock.Close() defer rawSock.Close()
@@ -110,7 +110,7 @@ func TestWriteTo(t *testing.T) {
defer udpListener.Close() defer udpListener.Close()
testingPort := 39440 testingPort := 39440
rawSock, err := Listen(testingPort, NewIncomingSTUNFilter()) rawSock, err := Listen(testingPort, NewIncomingSTUNFilter(), 1280)
require.NoError(t, err, "received an error while creating STUN listener, error: %s", err) require.NoError(t, err, "received an error while creating STUN listener, error: %s", err)
defer rawSock.Close() defer rawSock.Close()
@@ -144,7 +144,7 @@ func TestWriteTo(t *testing.T) {
} }
func TestSharedSocket_Close(t *testing.T) { func TestSharedSocket_Close(t *testing.T) {
rawSock, err := Listen(39440, NewIncomingSTUNFilter()) rawSock, err := Listen(39440, NewIncomingSTUNFilter(), 1280)
require.NoError(t, err, "received an error while creating STUN listener, error: %s", err) require.NoError(t, err, "received an error while creating STUN listener, error: %s", err)
errGrp := errgroup.Group{} errGrp := errgroup.Group{}