sockfd: check the result of nni_aio_schedule

This fixes a possible failure condition on teardown.
nanomsg · Dec 4, 2024 · 397b500 · 397b500
1 parent d017a6a
commit 397b500
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/core/sockfd.c b/src/core/sockfd.c
@@ -109,6 +109,7 @@ static void
 sfd_listener_accept(void *arg, nng_aio *aio)
 {
 	sfd_listener *l = arg;
+	int           rv;
 
 	if (nni_aio_begin(aio) != 0) {
 		return;
@@ -122,8 +123,9 @@ sfd_listener_accept(void *arg, nng_aio *aio)
 
 	if (l->listen_cnt) {
 		sfd_start_conn(l, aio);
+	} else if ((rv = nni_aio_schedule(aio, sfd_cancel_accept, l)) != 0) {
+		nni_aio_finish_error(aio, rv);
 	} else {
-		nni_aio_schedule(aio, sfd_cancel_accept, l);
 		nni_aio_list_append(&l->accept_q, aio);
 	}
 	nni_mtx_unlock(&l->mtx);