aboutsummaryrefslogtreecommitdiff
path: root/src/transport/ipc
diff options
context:
space:
mode:
authorGarrett D'Amore <garrett@damore.org>2018-08-06 19:01:12 +0300
committerGarrett D'Amore <garrett@damore.org>2018-08-06 19:01:12 +0300
commitae944a8de32c107eea9427104e153c25e4a681f1 (patch)
tree7029f7668fe3e1a9899da57bf6c1e60e0394bacb /src/transport/ipc
parentd7f7c896c0ede24249ef63b1e45b1878bf4bd473 (diff)
downloadnng-ae944a8de32c107eea9427104e153c25e4a681f1.tar.gz
nng-ae944a8de32c107eea9427104e153c25e4a681f1.tar.bz2
nng-ae944a8de32c107eea9427104e153c25e4a681f1.zip
Revert "fixes #599 nng_dial sync should not return until added to socket"
This changeset needs work. We are seeing errors described by This reverts commit d7f7c896c0ede24249ef63b1e45b1878bf4bd473.
Diffstat (limited to 'src/transport/ipc')
-rw-r--r--src/transport/ipc/ipc.c567
1 files changed, 312 insertions, 255 deletions
diff --git a/src/transport/ipc/ipc.c b/src/transport/ipc/ipc.c
index 3d798d4c..ee72d6d8 100644
--- a/src/transport/ipc/ipc.c
+++ b/src/transport/ipc/ipc.c
@@ -23,20 +23,14 @@
typedef struct ipctran_pipe ipctran_pipe;
typedef struct ipctran_dialer ipctran_dialer;
typedef struct ipctran_listener ipctran_listener;
-typedef struct ipctran_ep ipctran_ep;
// ipc_pipe is one end of an IPC connection.
struct ipctran_pipe {
- nni_ipc_conn * conn;
- uint16_t peer;
- uint16_t proto;
- size_t rcvmax;
- nni_atomic_flag reaped;
- nni_sockaddr sa;
- nni_list_node node;
- ipctran_ep * ep;
- nni_reap_item reap;
- bool closed;
+ nni_ipc_conn *conn;
+ uint16_t peer;
+ uint16_t proto;
+ size_t rcvmax;
+ nni_sockaddr sa;
uint8_t txhead[1 + sizeof(uint64_t)];
uint8_t rxhead[1 + sizeof(uint64_t)];
@@ -47,34 +41,32 @@ struct ipctran_pipe {
nni_list recvq;
nni_list sendq;
- nni_aio *useraio;
+ nni_aio *user_negaio;
nni_aio *txaio;
nni_aio *rxaio;
- nni_aio *negoaio;
- nni_aio *connaio;
+ nni_aio *negaio;
nni_msg *rxmsg;
nni_mtx mtx;
};
-struct ipctran_ep {
- nni_mtx mtx;
- nni_list pipes;
- bool fini;
- size_t rcvmax;
- uint16_t proto;
- nni_cb dtor;
- nni_sockaddr sa;
- nni_reap_item reap;
-};
-
struct ipctran_dialer {
- ipctran_ep ep;
+ nni_sockaddr sa;
nni_ipc_dialer *dialer;
+ uint16_t proto;
+ size_t rcvmax;
+ nni_aio * aio;
+ nni_aio * user_aio;
+ nni_mtx mtx;
};
struct ipctran_listener {
- ipctran_ep ep;
+ nni_sockaddr sa;
nni_ipc_listener *listener;
+ uint16_t proto;
+ size_t rcvmax;
+ nni_aio * aio;
+ nni_aio * user_aio;
+ nni_mtx mtx;
};
static void ipctran_pipe_send_start(ipctran_pipe *);
@@ -82,8 +74,8 @@ static void ipctran_pipe_recv_start(ipctran_pipe *);
static void ipctran_pipe_send_cb(void *);
static void ipctran_pipe_recv_cb(void *);
static void ipctran_pipe_nego_cb(void *);
-static void ipctran_pipe_conn_cb(void *);
-static void ipctran_pipe_reap(ipctran_pipe *);
+static void ipctran_dialer_cb(void *);
+static void ipctran_listener_cb(void *);
static int
ipctran_init(void)
@@ -103,8 +95,7 @@ ipctran_pipe_close(void *arg)
nni_aio_close(p->rxaio);
nni_aio_close(p->txaio);
- nni_aio_close(p->negoaio);
- nni_aio_close(p->connaio);
+ nni_aio_close(p->negaio);
nni_ipc_conn_close(p->conn);
}
@@ -114,39 +105,19 @@ ipctran_pipe_stop(void *arg)
{
ipctran_pipe *p = arg;
- nni_mtx_lock(&p->mtx);
- p->closed = true;
- nni_mtx_unlock(&p->mtx);
-
nni_aio_stop(p->rxaio);
nni_aio_stop(p->txaio);
- nni_aio_stop(p->negoaio);
- nni_aio_stop(p->connaio);
+ nni_aio_stop(p->negaio);
}
static void
ipctran_pipe_fini(void *arg)
{
ipctran_pipe *p = arg;
- ipctran_ep * ep;
-
- if (p == NULL) {
- return;
- }
- ipctran_pipe_stop(p);
- if ((ep = p->ep) != NULL) {
- nni_mtx_lock(&ep->mtx);
- nni_list_remove(&ep->pipes, p);
- if (ep->fini && nni_list_empty(&ep->pipes)) {
- nni_reap(&ep->reap, ep->dtor, ep);
- }
- nni_mtx_unlock(&ep->mtx);
- }
nni_aio_fini(p->rxaio);
nni_aio_fini(p->txaio);
- nni_aio_fini(p->negoaio);
- nni_aio_fini(p->connaio);
+ nni_aio_fini(p->negaio);
if (p->conn != NULL) {
nni_ipc_conn_fini(p->conn);
}
@@ -157,16 +128,8 @@ ipctran_pipe_fini(void *arg)
NNI_FREE_STRUCT(p);
}
-static void
-ipctran_pipe_reap(ipctran_pipe *p)
-{
- if (!nni_atomic_flag_test_and_set(&p->reaped)) {
- nni_reap(&p->reap, ipctran_pipe_fini, p);
- }
-}
-
static int
-ipctran_pipe_init(ipctran_pipe **pipep, ipctran_ep *ep)
+ipctran_pipe_init(ipctran_pipe **pipep, void *conn)
{
ipctran_pipe *p;
int rv;
@@ -175,75 +138,51 @@ ipctran_pipe_init(ipctran_pipe **pipep, ipctran_ep *ep)
return (NNG_ENOMEM);
}
nni_mtx_init(&p->mtx);
- NNI_LIST_NODE_INIT(&p->node);
if (((rv = nni_aio_init(&p->txaio, ipctran_pipe_send_cb, p)) != 0) ||
((rv = nni_aio_init(&p->rxaio, ipctran_pipe_recv_cb, p)) != 0) ||
- ((rv = nni_aio_init(&p->negoaio, ipctran_pipe_nego_cb, p)) != 0) ||
- ((rv = nni_aio_init(&p->connaio, ipctran_pipe_conn_cb, p)) != 0)) {
- ipctran_pipe_reap(p);
+ ((rv = nni_aio_init(&p->negaio, ipctran_pipe_nego_cb, p)) != 0)) {
+ ipctran_pipe_fini(p);
return (rv);
}
nni_aio_list_init(&p->sendq);
nni_aio_list_init(&p->recvq);
- nni_list_append(&ep->pipes, p);
- p->ep = ep;
- p->proto = ep->proto;
- p->rcvmax = ep->rcvmax;
- p->sa = ep->sa;
-
+ p->conn = conn;
+#if 0
+ p->proto = ep->proto;
+ p->rcvmax = ep->rcvmax;
+ p->sa.s_ipc.sa_family = NNG_AF_IPC;
+ p->sa = ep->sa;
+#endif
*pipep = p;
return (0);
}
static void
-ipctran_pipe_conn_cb(void *arg)
+ipctran_pipe_nego_cancel(nni_aio *aio, int rv)
{
- ipctran_pipe *p = arg;
- nni_aio * aio = p->connaio;
- nni_iov iov;
- int rv;
+ ipctran_pipe *p = nni_aio_get_prov_data(aio);
- nni_mtx_lock(&p->ep->mtx);
- if ((rv = nni_aio_result(aio)) != 0) {
- nni_aio *uaio;
- if ((uaio = p->useraio) != NULL) {
- p->useraio = NULL;
- nni_aio_finish_error(uaio, rv);
- }
- nni_mtx_unlock(&p->ep->mtx);
- ipctran_pipe_reap(p);
+ nni_mtx_lock(&p->mtx);
+ if (p->user_negaio != aio) {
+ nni_mtx_unlock(&p->mtx);
return;
}
+ p->user_negaio = NULL;
+ nni_mtx_unlock(&p->mtx);
- p->conn = nni_aio_get_output(aio, 0);
- p->txhead[0] = 0;
- p->txhead[1] = 'S';
- p->txhead[2] = 'P';
- p->txhead[3] = 0;
- NNI_PUT16(&p->txhead[4], p->proto);
- NNI_PUT16(&p->txhead[6], 0);
-
- p->gotrxhead = 0;
- p->gottxhead = 0;
- p->wantrxhead = 8;
- p->wanttxhead = 8;
- iov.iov_len = 8;
- iov.iov_buf = &p->txhead[0];
- nni_aio_set_iov(p->negoaio, 1, &iov);
- nni_ipc_conn_send(p->conn, p->negoaio);
- nni_mtx_unlock(&p->ep->mtx);
+ nni_aio_abort(p->negaio, rv);
+ nni_aio_finish_error(aio, rv);
}
static void
ipctran_pipe_nego_cb(void *arg)
{
ipctran_pipe *p = arg;
- nni_aio * aio = p->negoaio;
- nni_aio * uaio;
+ nni_aio * aio = p->negaio;
int rv;
- nni_mtx_lock(&p->ep->mtx);
+ nni_mtx_lock(&p->mtx);
if ((rv = nni_aio_result(aio)) != 0) {
goto done;
}
@@ -262,7 +201,7 @@ ipctran_pipe_nego_cb(void *arg)
nni_aio_set_iov(aio, 1, &iov);
// send it down...
nni_ipc_conn_send(p->conn, aio);
- nni_mtx_unlock(&p->ep->mtx);
+ nni_mtx_unlock(&p->mtx);
return;
}
if (p->gotrxhead < p->wantrxhead) {
@@ -271,7 +210,7 @@ ipctran_pipe_nego_cb(void *arg)
iov.iov_buf = &p->rxhead[p->gotrxhead];
nni_aio_set_iov(aio, 1, &iov);
nni_ipc_conn_recv(p->conn, aio);
- nni_mtx_unlock(&p->ep->mtx);
+ nni_mtx_unlock(&p->mtx);
return;
}
// We have both sent and received the headers. Lets check the
@@ -286,19 +225,11 @@ ipctran_pipe_nego_cb(void *arg)
NNI_GET16(&p->rxhead[4], p->peer);
done:
-
- if ((uaio = p->useraio) != NULL) {
- p->useraio = NULL;
- if (rv == 0) {
- nni_aio_set_output(uaio, 0, p);
- nni_aio_finish(uaio, 0, 0);
- nni_mtx_unlock(&p->ep->mtx);
- return;
- }
- nni_aio_finish_error(uaio, rv);
+ if ((aio = p->user_negaio) != NULL) {
+ p->user_negaio = NULL;
+ nni_aio_finish(aio, rv, 0);
}
- nni_mtx_unlock(&p->ep->mtx);
- ipctran_pipe_reap(p);
+ nni_mtx_unlock(&p->mtx);
}
static void
@@ -312,20 +243,17 @@ ipctran_pipe_send_cb(void *arg)
nni_aio * txaio = p->txaio;
nni_mtx_lock(&p->mtx);
+ aio = nni_list_first(&p->sendq);
- rv = p->closed ? NNG_ECLOSED : nni_aio_result(txaio);
-
- if (rv != 0) {
+ if ((rv = nni_aio_result(txaio)) != 0) {
// Intentionally we do not queue up another transfer.
// There's an excellent chance that the pipe is no longer
// usable, with a partial transfer.
// The protocol should see this error, and close the
// pipe itself, we hope.
- while ((aio = nni_list_first(&p->sendq)) != NULL) {
- nni_aio_list_remove(aio);
- nni_aio_finish_error(aio, rv);
- }
+ nni_aio_list_remove(aio);
nni_mtx_unlock(&p->mtx);
+ nni_aio_finish_error(aio, rv);
return;
}
@@ -337,7 +265,6 @@ ipctran_pipe_send_cb(void *arg)
return;
}
- aio = nni_list_first(&p->sendq);
nni_aio_list_remove(aio);
ipctran_pipe_send_start(p);
@@ -361,6 +288,7 @@ ipctran_pipe_recv_cb(void *arg)
nni_aio * rxaio = p->rxaio;
nni_mtx_lock(&p->mtx);
+ aio = nni_list_first(&p->recvq);
if ((rv = nni_aio_result(rxaio)) != 0) {
// Error on receive. This has to cause an error back
@@ -368,10 +296,6 @@ ipctran_pipe_recv_cb(void *arg)
// toss it.
goto recv_error;
}
- if (p->closed) {
- rv = NNG_ECLOSED;
- goto recv_error;
- }
n = nni_aio_count(rxaio);
nni_aio_iov_advance(rxaio, n);
@@ -430,7 +354,6 @@ ipctran_pipe_recv_cb(void *arg)
// Otherwise we got a message read completely. Let the user know the
// good news.
- aio = nni_list_first(&p->recvq);
nni_aio_list_remove(aio);
msg = p->rxmsg;
p->rxmsg = NULL;
@@ -444,10 +367,7 @@ ipctran_pipe_recv_cb(void *arg)
return;
recv_error:
- while ((aio = nni_list_first(&p->recvq)) != NULL) {
- nni_aio_list_remove(aio);
- nni_aio_finish_error(aio, rv);
- }
+ nni_aio_list_remove(aio);
msg = p->rxmsg;
p->rxmsg = NULL;
// Intentionally, we do not queue up another receive.
@@ -455,6 +375,7 @@ recv_error:
nni_mtx_unlock(&p->mtx);
nni_msg_free(msg);
+ nni_aio_finish_error(aio, rv);
}
static void
@@ -531,11 +452,6 @@ ipctran_pipe_send(void *arg, nni_aio *aio)
return;
}
nni_mtx_lock(&p->mtx);
- if (p->closed) {
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- return;
- }
if ((rv = nni_aio_schedule(aio, ipctran_pipe_send_cancel, p)) != 0) {
nni_mtx_unlock(&p->mtx);
nni_aio_finish_error(aio, rv);
@@ -598,11 +514,6 @@ ipctran_pipe_recv(void *arg, nni_aio *aio)
}
nni_mtx_lock(&p->mtx);
- if (p->closed) {
- nni_mtx_unlock(&p->mtx);
- nni_aio_finish_error(aio, NNG_ECLOSED);
- return;
- }
if ((rv = nni_aio_schedule(aio, ipctran_pipe_recv_cancel, p)) != 0) {
nni_mtx_unlock(&p->mtx);
nni_aio_finish_error(aio, rv);
@@ -616,6 +527,43 @@ ipctran_pipe_recv(void *arg, nni_aio *aio)
nni_mtx_unlock(&p->mtx);
}
+static void
+ipctran_pipe_start(void *arg, nni_aio *aio)
+{
+ ipctran_pipe *p = arg;
+ nni_aio * negaio;
+ nni_iov iov;
+ int rv;
+
+ if (nni_aio_begin(aio) != 0) {
+ return;
+ }
+ nni_mtx_lock(&p->mtx);
+ if ((rv = nni_aio_schedule(aio, ipctran_pipe_nego_cancel, p)) != 0) {
+ nni_mtx_unlock(&p->mtx);
+ nni_aio_finish_error(aio, rv);
+ return;
+ }
+ p->txhead[0] = 0;
+ p->txhead[1] = 'S';
+ p->txhead[2] = 'P';
+ p->txhead[3] = 0;
+ NNI_PUT16(&p->txhead[4], p->proto);
+ NNI_PUT16(&p->txhead[6], 0);
+
+ p->user_negaio = aio;
+ p->gotrxhead = 0;
+ p->gottxhead = 0;
+ p->wantrxhead = 8;
+ p->wanttxhead = 8;
+ negaio = p->negaio;
+ iov.iov_len = 8;
+ iov.iov_buf = &p->txhead[0];
+ nni_aio_set_iov(negaio, 1, &iov);
+ nni_ipc_conn_send(p->conn, negaio);
+ nni_mtx_unlock(&p->mtx);
+}
+
static uint16_t
ipctran_pipe_peer(void *arg)
{
@@ -684,42 +632,21 @@ ipctran_dialer_fini(void *arg)
{
ipctran_dialer *d = arg;
- nni_mtx_lock(&d->ep.mtx);
+ nni_aio_stop(d->aio);
if (d->dialer != NULL) {
nni_ipc_dialer_fini(d->dialer);
- d->dialer = NULL;
- }
- d->ep.fini = true;
- if (!nni_list_empty(&d->ep.pipes)) {
- nni_mtx_unlock(&d->ep.mtx);
- return;
}
- nni_mtx_unlock(&d->ep.mtx);
-
- nni_mtx_fini(&d->ep.mtx);
+ nni_aio_fini(d->aio);
+ nni_mtx_fini(&d->mtx);
NNI_FREE_STRUCT(d);
}
static void
-ipctran_ep_close(void *arg)
-{
- ipctran_ep * ep = arg;
- ipctran_pipe *p;
-
- nni_mtx_lock(&ep->mtx);
- NNI_LIST_FOREACH (&ep->pipes, p) {
- nni_aio_close(p->negoaio);
- nni_aio_close(p->connaio);
- }
- nni_mtx_unlock(&ep->mtx);
-}
-
-static void
ipctran_dialer_close(void *arg)
{
ipctran_dialer *d = arg;
- ipctran_ep_close(&d->ep);
+ nni_aio_close(d->aio);
nni_ipc_dialer_close(d->dialer);
}
@@ -733,46 +660,90 @@ ipctran_dialer_init(void **dp, nni_url *url, nni_sock *sock)
if ((d = NNI_ALLOC_STRUCT(d)) == NULL) {
return (NNG_ENOMEM);
}
- nni_mtx_init(&d->ep.mtx);
- NNI_LIST_INIT(&d->ep.pipes, ipctran_pipe, node);
+ nni_mtx_init(&d->mtx);
- sz = sizeof(d->ep.sa.s_ipc.sa_path);
- d->ep.sa.s_ipc.sa_family = NNG_AF_IPC;
- d->ep.proto = nni_sock_proto_id(sock);
- d->ep.dtor = ipctran_dialer_fini;
+ sz = sizeof(d->sa.s_ipc.sa_path);
+ d->sa.s_ipc.sa_family = NNG_AF_IPC;
- if (nni_strlcpy(d->ep.sa.s_ipc.sa_path, url->u_path, sz) >= sz) {
+ if (nni_strlcpy(d->sa.s_ipc.sa_path, url->u_path, sz) >= sz) {
ipctran_dialer_fini(d);
return (NNG_EADDRINVAL);
}
- if ((rv = nni_ipc_dialer_init(&d->dialer)) != 0) {
+ if (((rv = nni_ipc_dialer_init(&d->dialer)) != 0) ||
+ ((rv = nni_aio_init(&d->aio, ipctran_dialer_cb, d)) != 0)) {
ipctran_dialer_fini(d);
return (rv);
}
+ d->proto = nni_sock_proto_id(sock);
+
*dp = d;
return (0);
}
static void
-ipctran_pipe_conn_cancel(nni_aio *aio, int rv)
+ipctran_dialer_cb(void *arg)
{
- ipctran_pipe *p = nni_aio_get_prov_data(aio);
+ ipctran_dialer *d = arg;
+ ipctran_pipe * p;
+ nni_ipc_conn * conn;
+ nni_aio * aio;
+ int rv;
+
+ nni_mtx_lock(&d->mtx);
+ aio = d->user_aio;
+ rv = nni_aio_result(d->aio);
+
+ if (aio == NULL) {
+ nni_mtx_unlock(&d->mtx);
+ if (rv == 0) {
+ conn = nni_aio_get_output(d->aio, 0);
+ nni_ipc_conn_fini(conn);
+ }
+ return;
+ }
+
+ if (rv != 0) {
+ d->user_aio = NULL;
+ nni_mtx_unlock(&d->mtx);
+ nni_aio_finish_error(aio, rv);
+ return;
+ }
- nni_mtx_lock(&p->ep->mtx);
- if (p->useraio != aio) {
- nni_mtx_unlock(&p->ep->mtx);
+ d->user_aio = NULL;
+ conn = nni_aio_get_output(d->aio, 0);
+ NNI_ASSERT(conn != NULL);
+ if ((rv = ipctran_pipe_init(&p, conn)) != 0) {
+ nni_mtx_unlock(&d->mtx);
+ nni_ipc_conn_fini(conn);
+ nni_aio_finish_error(aio, rv);
return;
}
- // Close the underlying AIOs. This will abort the operation.
- // The pipe is removed from pending list at completion callback.
- p->useraio = NULL;
- nni_aio_close(p->negoaio);
- nni_aio_close(p->connaio);
- nni_mtx_unlock(&p->ep->mtx);
+ p->proto = d->proto;
+ p->rcvmax = d->rcvmax;
+ p->sa = d->sa;
+ nni_mtx_unlock(&d->mtx);
+
+ nni_aio_set_output(aio, 0, p);
+ nni_aio_finish(aio, 0, 0);
+}
+
+static void
+ipctran_dialer_cancel(nni_aio *aio, int rv)
+{
+ ipctran_dialer *d = nni_aio_get_prov_data(aio);
+
+ nni_mtx_lock(&d->mtx);
+ if (d->user_aio != aio) {
+ nni_mtx_unlock(&d->mtx);
+ return;
+ }
+ d->user_aio = NULL;
+ nni_mtx_unlock(&d->mtx);
+ nni_aio_abort(d->aio, rv);
nni_aio_finish_error(aio, rv);
}
@@ -780,48 +751,47 @@ static void
ipctran_dialer_connect(void *arg, nni_aio *aio)
{
ipctran_dialer *d = arg;
- ipctran_pipe * p = NULL;
int rv;
if (nni_aio_begin(aio) != 0) {
return;
}
+ nni_mtx_lock(&d->mtx);
+ NNI_ASSERT(d->user_aio == NULL);
- nni_mtx_lock(&d->ep.mtx);
- if (((rv = ipctran_pipe_init(&p, &d->ep)) != 0) ||
- ((rv = nni_aio_schedule(aio, ipctran_pipe_conn_cancel, p)) != 0)) {
- nni_mtx_unlock(&d->ep.mtx);
+ if ((rv = nni_aio_schedule(aio, ipctran_dialer_cancel, d)) != 0) {
+ nni_mtx_unlock(&d->mtx);
nni_aio_finish_error(aio, rv);
- ipctran_pipe_reap(p);
return;
}
- p->useraio = aio;
+ d->user_aio = aio;
- nni_ipc_dialer_dial(d->dialer, &p->sa, p->connaio);
- nni_mtx_unlock(&d->ep.mtx);
+ nni_ipc_dialer_dial(d->dialer, &d->sa, d->aio);
+ nni_mtx_unlock(&d->mtx);
}
static int
-ipctran_ep_get_recvmaxsz(void *arg, void *v, size_t *szp, nni_opt_type t)
+ipctran_dialer_get_recvmaxsz(void *arg, void *v, size_t *szp, nni_opt_type t)
{
- ipctran_ep *ep = arg;
- int rv;
- nni_mtx_lock(&ep->mtx);
- rv = nni_copyout_size(ep->rcvmax, v, szp, t);
- nni_mtx_unlock(&ep->mtx);
+ ipctran_dialer *d = arg;
+ int rv;
+ nni_mtx_lock(&d->mtx);
+ rv = nni_copyout_size(d->rcvmax, v, szp, t);
+ nni_mtx_unlock(&d->mtx);
return (rv);
}
static int
-ipctran_ep_set_recvmaxsz(void *arg, const void *v, size_t sz, nni_opt_type t)
+ipctran_dialer_set_recvmaxsz(
+ void *arg, const void *v, size_t sz, nni_opt_type t)
{
- ipctran_ep *ep = arg;
- size_t val;
- int rv;
+ ipctran_dialer *d = arg;
+ size_t val;
+ int rv;
if ((rv = nni_copyin_size(&val, v, sz, 0, NNI_MAXSZ, t)) == 0) {
- nni_mtx_lock(&ep->mtx);
- ep->rcvmax = val;
- nni_mtx_unlock(&ep->mtx);
+ nni_mtx_lock(&d->mtx);
+ d->rcvmax = val;
+ nni_mtx_unlock(&d->mtx);
}
return (rv);
}
@@ -831,19 +801,12 @@ ipctran_listener_fini(void *arg)
{
ipctran_listener *l = arg;
- nni_mtx_lock(&l->ep.mtx);
+ nni_aio_stop(l->aio);
if (l->listener != NULL) {
nni_ipc_listener_fini(l->listener);
- l->listener = NULL;
- }
- l->ep.fini = true;
- if (!nni_list_empty(&l->ep.pipes)) {
- nni_mtx_unlock(&l->ep.mtx);
- return;
}
- nni_mtx_unlock(&l->ep.mtx);
-
- nni_mtx_fini(&l->ep.mtx);
+ nni_aio_fini(l->aio);
+ nni_mtx_fini(&l->mtx);
NNI_FREE_STRUCT(l);
}
@@ -857,24 +820,24 @@ ipctran_listener_init(void **lp, nni_url *url, nni_sock *sock)
if ((l = NNI_ALLOC_STRUCT(l)) == NULL) {
return (NNG_ENOMEM);
}
- nni_mtx_init(&l->ep.mtx);
- NNI_LIST_INIT(&l->ep.pipes, ipctran_pipe, node);
+ nni_mtx_init(&l->mtx);
- sz = sizeof(l->ep.sa.s_ipc.sa_path);
- l->ep.sa.s_ipc.sa_family = NNG_AF_IPC;
- l->ep.proto = nni_sock_proto_id(sock);
- l->ep.dtor = ipctran_listener_fini;
+ sz = sizeof(l->sa.s_ipc.sa_path);
+ l->sa.s_ipc.sa_family = NNG_AF_IPC;
- if (nni_strlcpy(l->ep.sa.s_ipc.sa_path, url->u_path, sz) >= sz) {
+ if (nni_strlcpy(l->sa.s_ipc.sa_path, url->u_path, sz) >= sz) {
ipctran_listener_fini(l);
return (NNG_EADDRINVAL);
}
- if ((rv = nni_ipc_listener_init(&l->listener)) != 0) {
+ if (((rv = nni_ipc_listener_init(&l->listener)) != 0) ||
+ ((rv = nni_aio_init(&l->aio, ipctran_listener_cb, l)) != 0)) {
ipctran_listener_fini(l);
return (rv);
}
+ l->proto = nni_sock_proto_id(sock);
+
*lp = l;
return (0);
}
@@ -884,7 +847,7 @@ ipctran_listener_close(void *arg)
{
ipctran_listener *l = arg;
- ipctran_ep_close(&l->ep);
+ nni_aio_close(l->aio);
nni_ipc_listener_close(l->listener);
}
@@ -894,35 +857,128 @@ ipctran_listener_bind(void *arg)
ipctran_listener *l = arg;
int rv;
- nni_mtx_lock(&l->ep.mtx);
- rv = nni_ipc_listener_listen(l->listener, &l->ep.sa);
- nni_mtx_unlock(&l->ep.mtx);
+ nni_mtx_lock(&l->mtx);
+ rv = nni_ipc_listener_listen(l->listener, &l->sa);
+ nni_mtx_unlock(&l->mtx);
return (rv);
}
static void
+ipctran_listener_cb(void *arg)
+{
+ ipctran_listener *l = arg;
+ nni_aio * aio;
+ int rv;
+ ipctran_pipe * p = NULL;
+ nni_ipc_conn * conn;
+
+ nni_mtx_lock(&l->mtx);
+ rv = nni_aio_result(l->aio);
+ aio = l->user_aio;
+ l->user_aio = NULL;
+
+ if (aio == NULL) {
+ nni_mtx_unlock(&l->mtx);
+ if (rv == 0) {
+ conn = nni_aio_get_output(l->aio, 0);
+ nni_ipc_conn_fini(conn);
+ }
+ return;
+ }
+
+ if (rv != 0) {
+ nni_mtx_unlock(&l->mtx);
+ nni_aio_finish_error(aio, rv);
+ return;
+ }
+
+ conn = nni_aio_get_output(l->aio, 0);
+ NNI_ASSERT(conn != NULL);
+
+ // Attempt to allocate the parent pipe. If this fails we'll
+ // drop the connection (ENOMEM probably).
+ if ((rv = ipctran_pipe_init(&p, conn)) != 0) {
+ nni_mtx_unlock(&l->mtx);
+ nni_ipc_conn_fini(conn);
+ nni_aio_finish_error(aio, rv);
+ return;
+ }
+ p->proto = l->proto;
+ p->rcvmax = l->rcvmax;
+ p->sa = l->sa;
+ nni_mtx_unlock(&l->mtx);
+
+ nni_aio_set_output(aio, 0, p);
+ nni_aio_finish(aio, 0, 0);
+}
+
+static void
+ipctran_listener_cancel(nni_aio *aio, int rv)
+{
+ ipctran_listener *l = nni_aio_get_prov_data(aio);
+
+ NNI_ASSERT(rv != 0);
+ nni_mtx_lock(&l->mtx);
+ if (l->user_aio != aio) {
+ nni_mtx_unlock(&l->mtx);
+ return;
+ }
+ l->user_aio = NULL;
+ nni_mtx_unlock(&l->mtx);
+
+ nni_aio_abort(l->aio, rv);
+ nni_aio_finish_error(aio, rv);
+}
+
+static void
ipctran_listener_accept(void *arg, nni_aio *aio)
{
ipctran_listener *l = arg;
- ipctran_pipe * p;
int rv;
if (nni_aio_begin(aio) != 0) {
return;
}
+ nni_mtx_lock(&l->mtx);
+ NNI_ASSERT(l->user_aio == NULL);
- nni_mtx_lock(&l->ep.mtx);
- if (((rv = ipctran_pipe_init(&p, &l->ep)) != 0) ||
- ((rv = nni_aio_schedule(aio, ipctran_pipe_conn_cancel, p)) != 0)) {
- nni_mtx_unlock(&l->ep.mtx);
+ if ((rv = nni_aio_schedule(aio, ipctran_listener_cancel, l)) != 0) {
+ nni_mtx_unlock(&l->mtx);
nni_aio_finish_error(aio, rv);
- ipctran_pipe_reap(p);
return;
}
- p->useraio = aio;
+ l->user_aio = aio;
+
+ nni_ipc_listener_accept(l->listener, l->aio);
+ nni_mtx_unlock(&l->mtx);
+}
+
+static int
+ipctran_listener_set_recvmaxsz(
+ void *arg, const void *data, size_t sz, nni_opt_type t)
+{
+ ipctran_listener *l = arg;
+ size_t val;
+ int rv;
+
+ if ((rv = nni_copyin_size(&val, data, sz, 0, NNI_MAXSZ, t)) == 0) {
+ nni_mtx_lock(&l->mtx);
+ l->rcvmax = val;
+ nni_mtx_unlock(&l->mtx);
+ }
+ return (rv);
+}
- nni_ipc_listener_accept(l->listener, p->connaio);
- nni_mtx_unlock(&l->ep.mtx);
+static int
+ipctran_listener_get_recvmaxsz(
+ void *arg, void *data, size_t *szp, nni_opt_type t)
+{
+ ipctran_listener *l = arg;
+ int rv;
+ nni_mtx_lock(&l->mtx);
+ rv = nni_copyout_size(l->rcvmax, data, szp, t);
+ nni_mtx_unlock(&l->mtx);
+ return (rv);
}
static int
@@ -931,9 +987,9 @@ ipctran_listener_get_locaddr(void *arg, void *buf, size_t *szp, nni_opt_type t)
ipctran_listener *l = arg;
int rv;
- nni_mtx_lock(&l->ep.mtx);
- rv = nni_copyout_sockaddr(&l->ep.sa, buf, szp, t);
- nni_mtx_unlock(&l->ep.mtx);
+ nni_mtx_lock(&l->mtx);
+ rv = nni_copyout_sockaddr(&l->sa, buf, szp, t);
+ nni_mtx_unlock(&l->mtx);
return (rv);
}
@@ -954,9 +1010,9 @@ ipctran_listener_set_perms(
// Probably we could further limit this -- most systems don't have
// meaningful chmod beyond the lower 9 bits.
if ((rv = nni_copyin_int(&val, data, sz, 0, 0x7FFFFFFF, t)) == 0) {
- nni_mtx_lock(&l->ep.mtx);
+ nni_mtx_lock(&l->mtx);
rv = nni_ipc_listener_set_permissions(l->listener, val);
- nni_mtx_unlock(&l->ep.mtx);
+ nni_mtx_unlock(&l->mtx);
}
return (rv);
}
@@ -976,10 +1032,10 @@ ipctran_listener_set_sec_desc(
int rv;
if ((rv = nni_copyin_ptr(&ptr, data, sz, t)) == 0) {
- nni_mtx_lock(&l->ep.mtx);
+ nni_mtx_lock(&l->mtx);
rv =
nni_ipc_listener_set_security_descriptor(l->listener, ptr);
- nni_mtx_unlock(&l->ep.mtx);
+ nni_mtx_unlock(&l->mtx);
}
return (rv);
}
@@ -1029,6 +1085,7 @@ static nni_tran_option ipctran_pipe_options[] = {
static nni_tran_pipe_ops ipctran_pipe_ops = {
.p_fini = ipctran_pipe_fini,
+ .p_start = ipctran_pipe_start,
.p_stop = ipctran_pipe_stop,
.p_send = ipctran_pipe_send,
.p_recv = ipctran_pipe_recv,
@@ -1041,8 +1098,8 @@ static nni_tran_option ipctran_dialer_options[] = {
{
.o_name = NNG_OPT_RECVMAXSZ,
.o_type = NNI_TYPE_SIZE,
- .o_get = ipctran_ep_get_recvmaxsz,
- .o_set = ipctran_ep_set_recvmaxsz,
+ .o_get = ipctran_dialer_get_recvmaxsz,
+ .o_set = ipctran_dialer_set_recvmaxsz,
.o_chk = ipctran_check_recvmaxsz,
},
// terminate list
@@ -1055,8 +1112,8 @@ static nni_tran_option ipctran_listener_options[] = {
{
.o_name = NNG_OPT_RECVMAXSZ,
.o_type = NNI_TYPE_SIZE,
- .o_get = ipctran_ep_get_recvmaxsz,
- .o_set = ipctran_ep_set_recvmaxsz,
+ .o_get = ipctran_listener_get_recvmaxsz,
+ .o_set = ipctran_listener_set_recvmaxsz,
.o_chk = ipctran_check_recvmaxsz,
},
{