From cb1b6d0ab7d4a6edeea3d3b7190cbfacea351881 Mon Sep 17 00:00:00 2001
From: denis-tingaikin <denis.tingajkin@xored.com>
Date: Fri, 24 May 2024 13:44:55 +0300
Subject: [PATCH] fix race cond

Signed-off-by: denis-tingaikin <denis.tingajkin@xored.com>
---
 connwrap.go | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/connwrap.go b/connwrap.go
index 9a0bdf4..e18be2f 100644
--- a/connwrap.go
+++ b/connwrap.go
@@ -14,6 +14,7 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+//go:build !windows
 // +build !windows
 
 package grpcfd
@@ -72,6 +73,7 @@ type connWrap struct {
 	sendFDs      []int
 	errChs       []chan error
 	sendExecutor serialize.Executor
+	closed       bool
 
 	recvFDChans  map[inodeKey][]chan uintptr
 	recvedFDs    map[inodeKey]uintptr
@@ -126,6 +128,7 @@ func (w *connWrap) close() error {
 			w.sendFDs = nil
 			w.errChs = nil
 		})
+		w.closed = true
 	})
 	return err
 }
@@ -180,6 +183,10 @@ func (w *connWrap) SendFD(fd uintptr) <-chan error {
 		return errCh
 	}
 	w.sendExecutor.AsyncExec(func() {
+		if w.closed {
+			close(errCh)
+			return
+		}
 		w.sendFDs = append(w.sendFDs, int(fd))
 		w.errChs = append(w.errChs, errCh)
 	})
@@ -231,6 +238,10 @@ func (w *connWrap) String() string {
 func (w *connWrap) RecvFD(dev, ino uint64) <-chan uintptr {
 	fdCh := make(chan uintptr, 1)
 	w.recvExecutor.AsyncExec(func() {
+		if w.closed {
+			close(fdCh)
+			return
+		}
 		key := inodeKey{
 			dev: dev,
 			ino: ino,
@@ -352,14 +363,16 @@ func (w *connWrap) Read(b []byte) (n int, err error) {
 }
 
 // FromPeer - return grpcfd.FDTransceiver from peer.Peer
-//            ok is true of successful, false otherwise
+//
+//	ok is true of successful, false otherwise
 func FromPeer(p *peer.Peer) (transceiver FDTransceiver, ok bool) {
 	transceiver, ok = p.Addr.(FDTransceiver)
 	return transceiver, ok
 }
 
 // FromContext - return grpcfd.FDTransceiver from context.Context
-//               ok is true of successful, false otherwise
+//
+//	ok is true of successful, false otherwise
 func FromContext(ctx context.Context) (transceiver FDTransceiver, ok bool) {
 	p, ok := peer.FromContext(ctx)
 	if !ok {