mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-14 02:41:34 +01:00
0e95f16cdd
If a peer connection switches from Relayed to ICE P2P, the Relayed proxy still consumes the data the other peer sends. Because the proxy is operating, the WireGuard switches back to the Relayed proxy automatically, thanks to the roaming feature. Extend the Proxy implementation with pause/resume functions. Before switching to the p2p connection, pause the WireGuard proxy operation to prevent unnecessary package sources. Consider waiting some milliseconds after the pause to be sure the WireGuard engine already processed all UDP msg in from the pipe.
129 lines
2.3 KiB
Go
129 lines
2.3 KiB
Go
//go:build linux
|
|
|
|
package wgproxy
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/wgproxy/ebpf"
|
|
"github.com/netbirdio/netbird/client/internal/wgproxy/usp"
|
|
"github.com/netbirdio/netbird/util"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
_ = util.InitLog("trace", "console")
|
|
code := m.Run()
|
|
os.Exit(code)
|
|
}
|
|
|
|
type mocConn struct {
|
|
closeChan chan struct{}
|
|
closed bool
|
|
}
|
|
|
|
func newMockConn() *mocConn {
|
|
return &mocConn{
|
|
closeChan: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
func (m *mocConn) Read(b []byte) (n int, err error) {
|
|
<-m.closeChan
|
|
return 0, io.EOF
|
|
}
|
|
|
|
func (m *mocConn) Write(b []byte) (n int, err error) {
|
|
<-m.closeChan
|
|
return 0, io.EOF
|
|
}
|
|
|
|
func (m *mocConn) Close() error {
|
|
if m.closed == true {
|
|
return nil
|
|
}
|
|
|
|
m.closed = true
|
|
close(m.closeChan)
|
|
return nil
|
|
}
|
|
|
|
func (m *mocConn) LocalAddr() net.Addr {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (m *mocConn) RemoteAddr() net.Addr {
|
|
return &net.UDPAddr{
|
|
IP: net.ParseIP("172.16.254.1"),
|
|
}
|
|
}
|
|
|
|
func (m *mocConn) SetDeadline(t time.Time) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (m *mocConn) SetReadDeadline(t time.Time) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func (m *mocConn) SetWriteDeadline(t time.Time) error {
|
|
panic("implement me")
|
|
}
|
|
|
|
func TestProxyCloseByRemoteConn(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
tests := []struct {
|
|
name string
|
|
proxy Proxy
|
|
}{
|
|
{
|
|
name: "userspace proxy",
|
|
proxy: usp.NewWGUserSpaceProxy(51830),
|
|
},
|
|
}
|
|
|
|
if runtime.GOOS == "linux" && os.Getenv("GITHUB_ACTIONS") != "true" {
|
|
ebpfProxy := ebpf.NewWGEBPFProxy(51831)
|
|
if err := ebpfProxy.Listen(); err != nil {
|
|
t.Fatalf("failed to initialize ebpf proxy: %s", err)
|
|
}
|
|
defer func() {
|
|
if err := ebpfProxy.Free(); err != nil {
|
|
t.Errorf("failed to free ebpf proxy: %s", err)
|
|
}
|
|
}()
|
|
proxyWrapper := &ebpf.ProxyWrapper{
|
|
WgeBPFProxy: ebpfProxy,
|
|
}
|
|
|
|
tests = append(tests, struct {
|
|
name string
|
|
proxy Proxy
|
|
}{
|
|
name: "ebpf proxy",
|
|
proxy: proxyWrapper,
|
|
})
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
relayedConn := newMockConn()
|
|
err := tt.proxy.AddTurnConn(ctx, relayedConn)
|
|
if err != nil {
|
|
t.Errorf("error: %v", err)
|
|
}
|
|
|
|
_ = relayedConn.Close()
|
|
if err := tt.proxy.CloseConn(); err != nil {
|
|
t.Errorf("error: %v", err)
|
|
}
|
|
})
|
|
}
|
|
}
|