Fix eBPF pause handling

netbirdio · Oct 7, 2024 · 48184ec · 48184ec
1 parent f18ae8b
commit 48184ec
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/client/internal/wgproxy/ebpf/wrapper.go b/client/internal/wgproxy/ebpf/wrapper.go
@@ -101,7 +101,16 @@ func (p *ProxyWrapper) proxyToLocal(ctx context.Context) {
  return
  }
 
- if err := p.WgeBPFProxy.sendPkg(buf[:n], p.wgEndpointAddr.Port); err != nil {
+ p.pausedMu.Lock()
+ if p.paused {
+ p.pausedMu.Unlock()
+ continue
+ }
+
+ err := p.WgeBPFProxy.sendPkg(buf[:n], p.wgEndpointAddr.Port)
+ p.pausedMu.Unlock()
+
+ if err != nil {
  if ctx.Err() != nil {
  return
  }