aboutsummaryrefslogtreecommitdiff
path: root/src/protocol
diff options
context:
space:
mode:
Diffstat (limited to 'src/protocol')
-rw-r--r--src/protocol/bus/bus.c17
-rw-r--r--src/protocol/pair/pair.c8
-rw-r--r--src/protocol/pipeline/pull.c4
-rw-r--r--src/protocol/pipeline/push.c8
-rw-r--r--src/protocol/pubsub/pub.c14
-rw-r--r--src/protocol/pubsub/sub.c8
-rw-r--r--src/protocol/reqrep/rep.c19
-rw-r--r--src/protocol/reqrep/req.c15
-rw-r--r--src/protocol/survey/respond.c19
-rw-r--r--src/protocol/survey/survey.c18
10 files changed, 78 insertions, 52 deletions
diff --git a/src/protocol/bus/bus.c b/src/protocol/bus/bus.c
index b88337d1..e8428e42 100644
--- a/src/protocol/bus/bus.c
+++ b/src/protocol/bus/bus.c
@@ -165,7 +165,7 @@ nni_bus_pipe_fini(void *arg)
static int
-nni_bus_pipe_add(void *arg)
+nni_bus_pipe_start(void *arg)
{
nni_bus_pipe *ppipe = arg;
nni_bus_sock *psock = ppipe->psock;
@@ -182,15 +182,18 @@ nni_bus_pipe_add(void *arg)
static void
-nni_bus_pipe_rem(void *arg)
+nni_bus_pipe_stop(void *arg)
{
nni_bus_pipe *ppipe = arg;
nni_bus_sock *psock = ppipe->psock;
+ nni_sock *nsock = psock->nsock;
- nni_list_remove(&psock->pipes, ppipe);
+ if (nni_list_active(&psock->pipes, ppipe)) {
+ nni_list_remove(&psock->pipes, ppipe);
- nni_msgq_close(ppipe->sendq);
- nni_msgq_aio_cancel(nni_sock_recvq(psock->nsock), &ppipe->aio_putq);
+ nni_msgq_close(ppipe->sendq);
+ nni_msgq_aio_cancel(nni_sock_recvq(nsock), &ppipe->aio_putq);
+ }
}
@@ -390,8 +393,8 @@ nni_bus_sock_getopt(void *arg, int opt, void *buf, size_t *szp)
static nni_proto_pipe_ops nni_bus_pipe_ops = {
.pipe_init = nni_bus_pipe_init,
.pipe_fini = nni_bus_pipe_fini,
- .pipe_add = nni_bus_pipe_add,
- .pipe_rem = nni_bus_pipe_rem,
+ .pipe_start = nni_bus_pipe_start,
+ .pipe_stop = nni_bus_pipe_stop,
};
static nni_proto_sock_ops nni_bus_sock_ops = {
diff --git a/src/protocol/pair/pair.c b/src/protocol/pair/pair.c
index 65eabd87..ed5cee9f 100644
--- a/src/protocol/pair/pair.c
+++ b/src/protocol/pair/pair.c
@@ -130,7 +130,7 @@ nni_pair_pipe_fini(void *arg)
static int
-nni_pair_pipe_add(void *arg)
+nni_pair_pipe_start(void *arg)
{
nni_pair_pipe *ppipe = arg;
nni_pair_sock *psock = ppipe->psock;
@@ -152,7 +152,7 @@ nni_pair_pipe_add(void *arg)
static void
-nni_pair_pipe_rem(void *arg)
+nni_pair_pipe_stop(void *arg)
{
nni_pair_pipe *ppipe = arg;
nni_pair_sock *psock = ppipe->psock;
@@ -277,8 +277,8 @@ nni_pair_sock_getopt(void *arg, int opt, void *buf, size_t *szp)
static nni_proto_pipe_ops nni_pair_pipe_ops = {
.pipe_init = nni_pair_pipe_init,
.pipe_fini = nni_pair_pipe_fini,
- .pipe_add = nni_pair_pipe_add,
- .pipe_rem = nni_pair_pipe_rem,
+ .pipe_start = nni_pair_pipe_start,
+ .pipe_stop = nni_pair_pipe_stop,
};
static nni_proto_sock_ops nni_pair_sock_ops = {
diff --git a/src/protocol/pipeline/pull.c b/src/protocol/pipeline/pull.c
index 622bd958..b569253d 100644
--- a/src/protocol/pipeline/pull.c
+++ b/src/protocol/pipeline/pull.c
@@ -229,8 +229,8 @@ nni_pull_sock_getopt(void *arg, int opt, void *buf, size_t *szp)
static nni_proto_pipe_ops nni_pull_pipe_ops = {
.pipe_init = nni_pull_pipe_init,
.pipe_fini = nni_pull_pipe_fini,
- .pipe_add = nni_pull_pipe_start,
- .pipe_rem = nni_pull_pipe_stop,
+ .pipe_start = nni_pull_pipe_start,
+ .pipe_stop = nni_pull_pipe_stop,
};
static nni_proto_sock_ops nni_pull_sock_ops = {
diff --git a/src/protocol/pipeline/push.c b/src/protocol/pipeline/push.c
index d2d8363a..9691d8d8 100644
--- a/src/protocol/pipeline/push.c
+++ b/src/protocol/pipeline/push.c
@@ -116,7 +116,7 @@ nni_push_pipe_fini(void *arg)
static int
-nni_push_pipe_add(void *arg)
+nni_push_pipe_start(void *arg)
{
nni_push_pipe *pp = arg;
nni_push_sock *push = pp->push;
@@ -139,7 +139,7 @@ nni_push_pipe_add(void *arg)
static void
-nni_push_pipe_rem(void *arg)
+nni_push_pipe_stop(void *arg)
{
nni_push_pipe *pp = arg;
nni_push_sock *push = pp->push;
@@ -243,8 +243,8 @@ nni_push_sock_getopt(void *arg, int opt, void *buf, size_t *szp)
static nni_proto_pipe_ops nni_push_pipe_ops = {
.pipe_init = nni_push_pipe_init,
.pipe_fini = nni_push_pipe_fini,
- .pipe_add = nni_push_pipe_add,
- .pipe_rem = nni_push_pipe_rem,
+ .pipe_start = nni_push_pipe_start,
+ .pipe_stop = nni_push_pipe_stop,
};
static nni_proto_sock_ops nni_push_sock_ops = {
diff --git a/src/protocol/pubsub/pub.c b/src/protocol/pubsub/pub.c
index ad0b9580..a37fe991 100644
--- a/src/protocol/pubsub/pub.c
+++ b/src/protocol/pubsub/pub.c
@@ -147,7 +147,7 @@ nni_pub_pipe_fini(void *arg)
static int
-nni_pub_pipe_add(void *arg)
+nni_pub_pipe_start(void *arg)
{
nni_pub_pipe *pp = arg;
nni_pub_sock *pub = pp->pub;
@@ -168,13 +168,15 @@ nni_pub_pipe_add(void *arg)
static void
-nni_pub_pipe_rem(void *arg)
+nni_pub_pipe_stop(void *arg)
{
nni_pub_pipe *pp = arg;
nni_pub_sock *pub = pp->pub;
- nni_list_remove(&pub->pipes, pp);
- nni_msgq_close(pp->sendq);
+ if (nni_list_active(&pub->pipes, pp)) {
+ nni_list_remove(&pub->pipes, pp);
+ nni_msgq_close(pp->sendq);
+ }
}
@@ -314,8 +316,8 @@ nni_pub_sock_getopt(void *arg, int opt, void *buf, size_t *szp)
static nni_proto_pipe_ops nni_pub_pipe_ops = {
.pipe_init = nni_pub_pipe_init,
.pipe_fini = nni_pub_pipe_fini,
- .pipe_add = nni_pub_pipe_add,
- .pipe_rem = nni_pub_pipe_rem,
+ .pipe_start = nni_pub_pipe_start,
+ .pipe_stop = nni_pub_pipe_stop,
};
nni_proto_sock_ops nni_pub_sock_ops = {
diff --git a/src/protocol/pubsub/sub.c b/src/protocol/pubsub/sub.c
index 6fae1962..ecfbf540 100644
--- a/src/protocol/pubsub/sub.c
+++ b/src/protocol/pubsub/sub.c
@@ -115,7 +115,7 @@ nni_sub_pipe_fini(void *arg)
static int
-nni_sub_pipe_add(void *arg)
+nni_sub_pipe_start(void *arg)
{
nni_sub_pipe *sp = arg;
@@ -126,7 +126,7 @@ nni_sub_pipe_add(void *arg)
static void
-nni_sub_pipe_close(void *arg)
+nni_sub_pipe_stop(void *arg)
{
nni_sub_pipe *sp = arg;
@@ -340,8 +340,8 @@ nni_sub_sock_rfilter(void *arg, nni_msg *msg)
static nni_proto_pipe_ops nni_sub_pipe_ops = {
.pipe_init = nni_sub_pipe_init,
.pipe_fini = nni_sub_pipe_fini,
- .pipe_add = nni_sub_pipe_add,
- .pipe_rem = nni_sub_pipe_close,
+ .pipe_start = nni_sub_pipe_start,
+ .pipe_stop = nni_sub_pipe_stop,
};
static nni_proto_sock_ops nni_sub_sock_ops = {
diff --git a/src/protocol/reqrep/rep.c b/src/protocol/reqrep/rep.c
index 55a6855c..ec5b77cf 100644
--- a/src/protocol/reqrep/rep.c
+++ b/src/protocol/reqrep/rep.c
@@ -49,6 +49,7 @@ struct nni_rep_pipe {
nni_aio aio_send;
nni_aio aio_recv;
nni_aio aio_putq;
+ int running;
};
static int
@@ -168,7 +169,7 @@ nni_rep_pipe_fini(void *arg)
static int
-nni_rep_pipe_add(void *arg)
+nni_rep_pipe_start(void *arg)
{
nni_rep_pipe *rp = arg;
nni_rep_sock *rep = rp->rep;
@@ -183,19 +184,23 @@ nni_rep_pipe_add(void *arg)
nni_msgq_aio_get(rp->sendq, &rp->aio_getq);
nni_pipe_incref(rp->pipe);
nni_pipe_aio_recv(rp->pipe, &rp->aio_recv);
+ rp->running = 1;
return (0);
}
static void
-nni_rep_pipe_rem(void *arg)
+nni_rep_pipe_stop(void *arg)
{
nni_rep_pipe *rp = arg;
nni_rep_sock *rep = rp->rep;
- nni_msgq_close(rp->sendq);
- nni_msgq_aio_cancel(rep->urq, &rp->aio_putq);
- nni_idhash_remove(&rep->pipes, nni_pipe_id(rp->pipe));
+ if (rp->running) {
+ rp->running = 0;
+ nni_msgq_close(rp->sendq);
+ nni_msgq_aio_cancel(rep->urq, &rp->aio_putq);
+ nni_idhash_remove(&rep->pipes, nni_pipe_id(rp->pipe));
+ }
}
@@ -488,8 +493,8 @@ nni_rep_sock_rfilter(void *arg, nni_msg *msg)
static nni_proto_pipe_ops nni_rep_pipe_ops = {
.pipe_init = nni_rep_pipe_init,
.pipe_fini = nni_rep_pipe_fini,
- .pipe_add = nni_rep_pipe_add,
- .pipe_rem = nni_rep_pipe_rem,
+ .pipe_start = nni_rep_pipe_start,
+ .pipe_stop = nni_rep_pipe_stop,
};
static nni_proto_sock_ops nni_rep_sock_ops = {
diff --git a/src/protocol/reqrep/req.c b/src/protocol/reqrep/req.c
index f37913e6..af4dbd42 100644
--- a/src/protocol/reqrep/req.c
+++ b/src/protocol/reqrep/req.c
@@ -56,6 +56,7 @@ struct nni_req_pipe {
nni_aio aio_sendcooked; // cooked mode only
nni_aio aio_recv;
nni_aio aio_putq;
+ int running;
};
static void nni_req_resender(void *);
@@ -173,7 +174,7 @@ nni_req_pipe_fini(void *arg)
static int
-nni_req_pipe_add(void *arg)
+nni_req_pipe_start(void *arg)
{
nni_req_pipe *rp = arg;
nni_req_sock *req = rp->req;
@@ -190,16 +191,22 @@ nni_req_pipe_add(void *arg)
nni_msgq_aio_get(req->uwq, &rp->aio_getq);
nni_pipe_incref(rp->pipe);
nni_pipe_aio_recv(rp->pipe, &rp->aio_recv);
+ rp->running = 1;
return (0);
}
static void
-nni_req_pipe_rem(void *arg)
+nni_req_pipe_stop(void *arg)
{
nni_req_pipe *rp = arg;
nni_req_sock *req = rp->req;
+ if (!rp->running) {
+ return;
+ }
+ rp->running = 0;
+
// This removes the node from either busypipes or readypipes.
// It doesn't much matter which.
nni_list_remove(&req->readypipes, rp);
@@ -565,8 +572,8 @@ nni_req_sock_rfilter(void *arg, nni_msg *msg)
static nni_proto_pipe_ops nni_req_pipe_ops = {
.pipe_init = nni_req_pipe_init,
.pipe_fini = nni_req_pipe_fini,
- .pipe_add = nni_req_pipe_add,
- .pipe_rem = nni_req_pipe_rem,
+ .pipe_start = nni_req_pipe_start,
+ .pipe_stop = nni_req_pipe_stop,
};
static nni_proto_sock_ops nni_req_sock_ops = {
diff --git a/src/protocol/survey/respond.c b/src/protocol/survey/respond.c
index a70b08e6..8fe12e19 100644
--- a/src/protocol/survey/respond.c
+++ b/src/protocol/survey/respond.c
@@ -48,6 +48,7 @@ struct nni_resp_pipe {
nni_aio aio_putq;
nni_aio aio_send;
nni_aio aio_recv;
+ int running;
};
static int
@@ -173,7 +174,7 @@ nni_resp_pipe_fini(void *arg)
static int
-nni_resp_pipe_add(void *arg)
+nni_resp_pipe_start(void *arg)
{
nni_resp_pipe *ppipe = arg;
nni_resp_sock *psock = ppipe->psock;
@@ -186,20 +187,24 @@ nni_resp_pipe_add(void *arg)
nni_pipe_incref(ppipe->npipe);
nni_msgq_aio_get(ppipe->sendq, &ppipe->aio_getq);
+ ppipe->running = 1;
return (rv);
}
static void
-nni_resp_pipe_rem(void *arg)
+nni_resp_pipe_stop(void *arg)
{
nni_resp_pipe *ppipe = arg;
nni_resp_sock *psock = ppipe->psock;
- nni_idhash_remove(&psock->pipes, nni_pipe_id(ppipe->npipe));
+ if (ppipe->running) {
+ ppipe->running = 0;
+ nni_idhash_remove(&psock->pipes, nni_pipe_id(ppipe->npipe));
- nni_msgq_close(ppipe->sendq);
- nni_msgq_aio_cancel(psock->urq, &ppipe->aio_putq);
+ nni_msgq_close(ppipe->sendq);
+ nni_msgq_aio_cancel(psock->urq, &ppipe->aio_putq);
+ }
}
@@ -488,8 +493,8 @@ nni_resp_sock_rfilter(void *arg, nni_msg *msg)
static nni_proto_pipe_ops nni_resp_pipe_ops = {
.pipe_init = nni_resp_pipe_init,
.pipe_fini = nni_resp_pipe_fini,
- .pipe_add = nni_resp_pipe_add,
- .pipe_rem = nni_resp_pipe_rem,
+ .pipe_start = nni_resp_pipe_start,
+ .pipe_stop = nni_resp_pipe_stop,
};
static nni_proto_sock_ops nni_resp_sock_ops = {
diff --git a/src/protocol/survey/survey.c b/src/protocol/survey/survey.c
index 4fa89af3..0edcfef2 100644
--- a/src/protocol/survey/survey.c
+++ b/src/protocol/survey/survey.c
@@ -51,6 +51,7 @@ struct nni_surv_pipe {
nni_aio aio_putq;
nni_aio aio_send;
nni_aio aio_recv;
+ int running;
};
static int
@@ -168,7 +169,7 @@ failed:
static int
-nni_surv_pipe_add(void *arg)
+nni_surv_pipe_start(void *arg)
{
nni_surv_pipe *ppipe = arg;
nni_surv_sock *psock = ppipe->psock;
@@ -180,19 +181,22 @@ nni_surv_pipe_add(void *arg)
nni_pipe_incref(ppipe->npipe);
nni_pipe_aio_recv(ppipe->npipe, &ppipe->aio_recv);
+ ppipe->running = 1;
return (0);
}
static void
-nni_surv_pipe_rem(void *arg)
+nni_surv_pipe_stop(void *arg)
{
nni_surv_pipe *ppipe = arg;
nni_surv_sock *psock = ppipe->psock;
- nni_list_remove(&psock->pipes, ppipe);
- nni_msgq_close(ppipe->sendq);
- nni_msgq_aio_cancel(psock->urq, &ppipe->aio_putq);
+ if (ppipe->running) {
+ nni_list_remove(&psock->pipes, ppipe);
+ nni_msgq_close(ppipe->sendq);
+ nni_msgq_aio_cancel(psock->urq, &ppipe->aio_putq);
+ }
}
@@ -460,8 +464,8 @@ nni_surv_sock_rfilter(void *arg, nni_msg *msg)
static nni_proto_pipe_ops nni_surv_pipe_ops = {
.pipe_init = nni_surv_pipe_init,
.pipe_fini = nni_surv_pipe_fini,
- .pipe_add = nni_surv_pipe_add,
- .pipe_rem = nni_surv_pipe_rem,
+ .pipe_start = nni_surv_pipe_start,
+ .pipe_stop = nni_surv_pipe_stop,
};
static nni_proto_sock_ops nni_surv_sock_ops = {
generated by cgit v1.2.3-70-g09d2 (git 2.46.0) at 2026-01-18 20:11:37 +0000