Reduce lock contention and simplify logic for endpoint creation.

1 files changed, 24 insertions, 16 deletions

diff --git a/src/core/endpt.c b/src/core/endpt.c
index 58b30b9f..2366a8e8 100644
--- a/src/core/endpt.c
+++ b/src/core/endpt.c
@@ -37,41 +37,49 @@ nni_ep_create(nni_ep **epp, nni_sock *sock, const char *addr)
 	ep->ep_bound = 0;
 	ep->ep_pipe = NULL;
 	ep->ep_tran = tran;
-
-	nni_mtx_lock(nni_idlock);
-	rv = nni_idhash_alloc(nni_endpoints, &ep->ep_id, ep);
-	nni_mtx_unlock(nni_idlock);
-	if (rv != 0) {
-		NNI_FREE_STRUCT(ep);
-		return (rv);
-	}
+	NNI_LIST_NODE_INIT(&ep->ep_node);
+	// Could safely use strcpy here, but this avoids discussion.
+	(void) snprintf(ep->ep_addr, sizeof (ep->ep_addr), "%s", addr);
 
 	// Make a copy of the endpoint operations.  This allows us to
 	// modify them (to override NULLs for example), and avoids an extra
 	// dereference on hot paths.
 	ep->ep_ops = *tran->tran_ep;
-	NNI_LIST_NODE_INIT(&ep->ep_node);
 
-	if ((rv = nni_cv_init(&ep->ep_cv, &ep->ep_sock->s_mx)) != 0) {
-		nni_mtx_lock(nni_idlock);
-		nni_idhash_remove(nni_endpoints, ep->ep_id);
-		nni_mtx_unlock(nni_idlock);
+	if ((rv = nni_cv_init(&ep->ep_cv, &sock->s_mx)) != 0) {
 		NNI_FREE_STRUCT(ep);
-		return (NNG_ENOMEM);
+		return (rv);
 	}
 
-	// Could safely use strcpy here, but this avoids discussion.
-	(void) snprintf(ep->ep_addr, sizeof (ep->ep_addr), "%s", addr);
+	nni_mtx_lock(&sock->s_mx);
+	if (sock->s_closing) {
+		nni_mtx_unlock(&sock->s_mx);
+		nni_cv_fini(&ep->ep_cv);
+		NNI_FREE_STRUCT(ep);
+		return (NNG_ECLOSED);
+	}
+	nni_mtx_lock(nni_idlock);
+	rv = nni_idhash_alloc(nni_endpoints, &ep->ep_id, ep);
+	nni_mtx_unlock(nni_idlock);
+	if (rv != 0) {
+		nni_mtx_unlock(&sock->s_mx);
+		nni_cv_fini(&ep->ep_cv);
+		NNI_FREE_STRUCT(ep);
+		return (rv);
+	}
 
 	rv = ep->ep_ops.ep_init(&ep->ep_data, addr, nni_sock_proto(sock));
 	if (rv != 0) {
 		nni_mtx_lock(nni_idlock);
 		nni_idhash_remove(nni_endpoints, ep->ep_id);
 		nni_mtx_unlock(nni_idlock);
+		nni_mtx_unlock(&sock->s_mx);
 		nni_cv_fini(&ep->ep_cv);
 		NNI_FREE_STRUCT(ep);
 		return (rv);
 	}
+	nni_list_append(&sock->s_eps, ep);
+	nni_mtx_unlock(&sock->s_mx);
 
 	*epp = ep;
 	return (0);