diff options
Diffstat (limited to 'src/protocol/reqrep0')
| -rw-r--r-- | src/protocol/reqrep0/rep.c | 5 | ||||
| -rw-r--r-- | src/protocol/reqrep0/xrep.c | 5 |
2 files changed, 10 insertions, 0 deletions
diff --git a/src/protocol/reqrep0/rep.c b/src/protocol/reqrep0/rep.c index c483b777..24fc7335 100644 --- a/src/protocol/reqrep0/rep.c +++ b/src/protocol/reqrep0/rep.c @@ -346,6 +346,11 @@ rep0_pipe_start(void *arg) rep0_sock *s = p->rep; int rv; + if (nni_pipe_peer(p->pipe) != NNI_PROTO_REQ_V0) { + // Peer protocol mismatch. + return (NNG_EPROTO); + } + if ((rv = nni_idhash_insert(s->pipes, nni_pipe_id(p->pipe), p)) != 0) { return (rv); } diff --git a/src/protocol/reqrep0/xrep.c b/src/protocol/reqrep0/xrep.c index 6dcfe6be..f6cd29eb 100644 --- a/src/protocol/reqrep0/xrep.c +++ b/src/protocol/reqrep0/xrep.c @@ -178,6 +178,11 @@ xrep0_pipe_start(void *arg) xrep0_sock *s = p->rep; int rv; + if (nni_pipe_peer(p->pipe) != NNI_PROTO_REQ_V0) { + // Peer protocol mismatch. + return (NNG_EPROTO); + } + if ((rv = nni_idhash_insert(s->pipes, nni_pipe_id(p->pipe), p)) != 0) { return (rv); } |