aboutsummaryrefslogtreecommitdiff
path: root/src/protocol/survey
diff options
context:
space:
mode:
Diffstat (limited to 'src/protocol/survey')
-rw-r--r--src/protocol/survey/respond.c1
-rw-r--r--src/protocol/survey/survey.c1
2 files changed, 2 insertions, 0 deletions
diff --git a/src/protocol/survey/respond.c b/src/protocol/survey/respond.c
index 70dbd704..fcb067b0 100644
--- a/src/protocol/survey/respond.c
+++ b/src/protocol/survey/respond.c
@@ -285,6 +285,7 @@ resp_recv_cb(void *arg)
msg = nni_aio_get_msg(p->aio_recv);
nni_aio_set_msg(p->aio_recv, NULL);
+ nni_msg_set_pipe(msg, p->id);
// Store the pipe id in the header, first thing.
if (nni_msg_header_append_u32(msg, p->id) != 0) {
diff --git a/src/protocol/survey/survey.c b/src/protocol/survey/survey.c
index 361b9d37..de9df7a5 100644
--- a/src/protocol/survey/survey.c
+++ b/src/protocol/survey/survey.c
@@ -244,6 +244,7 @@ surv_recv_cb(void *arg)
msg = nni_aio_get_msg(p->aio_recv);
nni_aio_set_msg(p->aio_recv, NULL);
+ nni_msg_set_pipe(msg, nni_pipe_id(p->npipe));
// We yank 4 bytes of body, and move them to the header.
if (nni_msg_len(msg) < 4) {