2 files changed, 28 insertions, 7 deletions
diff --git a/src/protocol/pipeline0/pull.c b/src/protocol/pipeline0/pull.c
index c5017d50..81f6c137 100644
--- a/src/protocol/pipeline0/pull.c
+++ b/src/protocol/pipeline0/pull.c
@@ -68,6 +68,15 @@ pull0_sock_fini(void *arg)
 }
 
 static void
+pull0_pipe_stop(void *arg)
+{
+	pull0_pipe *p = arg;
+
+	nni_aio_stop(p->putq_aio);
+	nni_aio_stop(p->recv_aio);
+}
+
+static void
 pull0_pipe_fini(void *arg)
 {
 	pull0_pipe *p = arg;
@@ -110,12 +119,12 @@ pull0_pipe_start(void *arg)
 }
 
 static void
-pull0_pipe_stop(void *arg)
+pull0_pipe_close(void *arg)
 {
 	pull0_pipe *p = arg;
 
-	nni_aio_stop(p->putq_aio);
-	nni_aio_stop(p->recv_aio);
+	nni_aio_close(p->putq_aio);
+	nni_aio_close(p->recv_aio);
 }
 
 static void
@@ -198,6 +207,7 @@ static nni_proto_pipe_ops pull0_pipe_ops = {
 	.pipe_init  = pull0_pipe_init,
 	.pipe_fini  = pull0_pipe_fini,
 	.pipe_start = pull0_pipe_start,
+	.pipe_close = pull0_pipe_close,
 	.pipe_stop  = pull0_pipe_stop,
 };
 
diff --git a/src/protocol/pipeline0/push.c b/src/protocol/pipeline0/push.c
index 2ad657b6..e413cf46 100644
--- a/src/protocol/pipeline0/push.c
+++ b/src/protocol/pipeline0/push.c
@@ -83,6 +83,16 @@ push0_sock_close(void *arg)
 }
 
 static void
+push0_pipe_stop(void *arg)
+{
+	push0_pipe *p = arg;
+
+	nni_aio_stop(p->aio_recv);
+	nni_aio_stop(p->aio_send);
+	nni_aio_stop(p->aio_getq);
+}
+
+static void
 push0_pipe_fini(void *arg)
 {
 	push0_pipe *p = arg;
@@ -136,13 +146,13 @@ push0_pipe_start(void *arg)
 }
 
 static void
-push0_pipe_stop(void *arg)
+push0_pipe_close(void *arg)
 {
 	push0_pipe *p = arg;
 
-	nni_aio_stop(p->aio_recv);
-	nni_aio_stop(p->aio_send);
-	nni_aio_stop(p->aio_getq);
+	nni_aio_close(p->aio_recv);
+	nni_aio_close(p->aio_send);
+	nni_aio_close(p->aio_getq);
 }
 
 static void
@@ -214,6 +224,7 @@ static nni_proto_pipe_ops push0_pipe_ops = {
 	.pipe_init  = push0_pipe_init,
 	.pipe_fini  = push0_pipe_fini,
 	.pipe_start = push0_pipe_start,
+	.pipe_close = push0_pipe_close,
 	.pipe_stop  = push0_pipe_stop,
 };