fix for potential leaking connections

nanomsg · Dec 27, 2024 · 5183093 · 5183093
1 parent c64d0e5
commit 5183093
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/src/sp/transport/socket/sockfd.c b/src/sp/transport/socket/sockfd.c
@@ -170,7 +170,10 @@ sfd_tran_pipe_nego_cb(void *arg)
 	int            rv;
 
 	nni_mtx_lock(&ep->mtx);
-
+	if (ep->closed) {
+		rv = NNG_ECLOSED;
+		goto error;
+	}
 	if ((rv = nni_aio_result(aio)) != 0) {
 		goto error;
 	}

diff --git a/src/sp/transport/tcp/tcp.c b/src/sp/transport/tcp/tcp.c
@@ -174,7 +174,10 @@ tcptran_pipe_nego_cb(void *arg)
 	int           rv;
 
 	nni_mtx_lock(&ep->mtx);
-
+	if (ep->closed) {
+		rv = NNG_ECLOSED;
+		goto error;
+	}
 	if ((rv = nni_aio_result(aio)) != 0) {
 		goto error;
 	}

diff --git a/src/sp/transport/tls/tls.c b/src/sp/transport/tls/tls.c
@@ -175,6 +175,10 @@ tlstran_pipe_nego_cb(void *arg)
 	int           rv;
 
 	nni_mtx_lock(&ep->mtx);
+	if (ep->closed) {
+		rv = NNG_ECLOSED;
+		goto error;
+	}
 	if ((rv = nni_aio_result(aio)) != 0) {
 		goto error;
 	}