From 1f25bb075169a7e8ff9d9f021e53ceaee0a487f4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Zolt=C3=A1n=20Papp?= <zoltan.pmail@gmail.com>
Date: Mon, 7 Oct 2024 12:58:45 +0200
Subject: [PATCH] Reducate cognitive complexity

---
 client/internal/wgproxy/ebpf/wrapper.go | 34 ++++++++++++++++---------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/client/internal/wgproxy/ebpf/wrapper.go b/client/internal/wgproxy/ebpf/wrapper.go
index ed9fa1312..a220fa92e 100644
--- a/client/internal/wgproxy/ebpf/wrapper.go
+++ b/client/internal/wgproxy/ebpf/wrapper.go
@@ -84,20 +84,10 @@ func (e *ProxyWrapper) CloseConn() error {
 func (p *ProxyWrapper) proxyToLocal(ctx context.Context) {
 	defer p.WgeBPFProxy.removeTurnConn(uint16(p.wgEndpointAddr.Port))
 
-	var (
-		err error
-		n   int
-	)
 	buf := make([]byte, 1500)
 	for ctx.Err() == nil {
-		n, err = p.remoteConn.Read(buf)
+		n, err := p.readFromRemote(ctx, buf)
 		if err != nil {
-			if ctx.Err() != nil {
-				return
-			}
-			if err != io.EOF {
-				log.Errorf("failed to read from turn conn (endpoint: :%d): %s", p.wgEndpointAddr.Port, err)
-			}
 			return
 		}
 
@@ -107,7 +97,7 @@ func (p *ProxyWrapper) proxyToLocal(ctx context.Context) {
 			continue
 		}
 
-		err := p.WgeBPFProxy.sendPkg(buf[:n], p.wgEndpointAddr.Port)
+		err = p.WgeBPFProxy.sendPkg(buf[:n], p.wgEndpointAddr.Port)
 		p.pausedMu.Unlock()
 
 		if err != nil {
@@ -118,3 +108,23 @@ func (p *ProxyWrapper) proxyToLocal(ctx context.Context) {
 		}
 	}
 }
+
+func (p *ProxyWrapper) readFromRemote(ctx context.Context, buf []byte) (int, error) {
+	n, err := p.remoteConn.Read(buf)
+	if err != nil {
+		if ctx.Err() != nil {
+			return 0, ctx.Err()
+		}
+		if err != io.EOF {
+			log.Errorf("failed to read from turn conn (endpoint: :%d): %s", p.wgEndpointAddr.Port, err)
+		}
+		return 0, err
+	}
+	return n, nil
+}
+
+func (p *ProxyWrapper) isPaused() bool {
+	p.pausedMu.Lock()
+	defer p.pausedMu.Unlock()
+	return p.paused
+}