error handling for closed socket

backend_udp.go

@@ -3,6 +3,7 @@ package netbounce
 import (
 	"bytes"
 	"context"
+	"errors"
 	"fmt"
 	"net"
 	"sync"
@@ -231,7 +232,19 @@ func (b *backendUDP) udpBackend2Client(wg *sync.WaitGroup, rel udpRel) {
 udpBackendLoop:
 	for {
 		if n, _, err = rel.backend.ReadFrom(buf); err != nil {
-			//TODO: error handling
+			select {
+			case <-rel.ctx.Done():
+				break udpBackendLoop
+			default:
+			}
+
+			if errors.Is(err, net.ErrClosed) { //TODO: implement this error handling to all socket read/write operation
+				rel.ctxCancel()
+				break udpBackendLoop
+			}
+
+			log.Error().Err(err).Str(CONNECTION, rel.clientAddr).Str(DIRECTION, BACKEND_TO_CLIENT).Msg("udpBackend2Client: read from udp")
+			continue udpBackendLoop
 		}
 
 		clientAddr = rel.clientAddr