mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 09:50:47 +01:00
4e918e55ba
Rethink the peer reconnection implementation
50 lines
1019 B
Go
50 lines
1019 B
Go
//go:build linux && !android
|
|
|
|
package wgproxy
|
|
|
|
import (
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/iface/wgproxy/ebpf"
|
|
udpProxy "github.com/netbirdio/netbird/client/iface/wgproxy/udp"
|
|
)
|
|
|
|
type KernelFactory struct {
|
|
wgPort int
|
|
|
|
ebpfProxy *ebpf.WGEBPFProxy
|
|
}
|
|
|
|
func NewKernelFactory(wgPort int) *KernelFactory {
|
|
f := &KernelFactory{
|
|
wgPort: wgPort,
|
|
}
|
|
|
|
ebpfProxy := ebpf.NewWGEBPFProxy(wgPort)
|
|
if err := ebpfProxy.Listen(); err != nil {
|
|
log.Infof("WireGuard Proxy Factory will produce UDP proxy")
|
|
log.Warnf("failed to initialize ebpf proxy, fallback to user space proxy: %s", err)
|
|
return f
|
|
}
|
|
log.Infof("WireGuard Proxy Factory will produce eBPF proxy")
|
|
f.ebpfProxy = ebpfProxy
|
|
return f
|
|
}
|
|
|
|
func (w *KernelFactory) GetProxy() Proxy {
|
|
if w.ebpfProxy == nil {
|
|
return udpProxy.NewWGUDPProxy(w.wgPort)
|
|
}
|
|
|
|
return &ebpf.ProxyWrapper{
|
|
WgeBPFProxy: w.ebpfProxy,
|
|
}
|
|
}
|
|
|
|
func (w *KernelFactory) Free() error {
|
|
if w.ebpfProxy == nil {
|
|
return nil
|
|
}
|
|
return w.ebpfProxy.Free()
|
|
}
|