diff options
| author | Garrett D'Amore <garrett@damore.org> | 2017-08-10 16:33:09 -0700 |
|---|---|---|
| committer | Garrett D'Amore <garrett@damore.org> | 2017-08-10 16:33:09 -0700 |
| commit | 34ceda3c2dd4990d15e0341e86861dd291003f63 (patch) | |
| tree | 278a7bf91e466e00f4c2bea088b73cfeb285ecef /src/protocol/pair/pair.c | |
| parent | ac5f0ef7cf501693a9db2fcbd95b7cde419cbb2a (diff) | |
| download | nng-34ceda3c2dd4990d15e0341e86861dd291003f63.tar.gz nng-34ceda3c2dd4990d15e0341e86861dd291003f63.tar.bz2 nng-34ceda3c2dd4990d15e0341e86861dd291003f63.zip | |
Add new PAIR_V1 protocol.
The PAIR_V1 protocol supports both raw and cooked modes, and has loop
prevention included. It also has a polyamorous mode, wherein it allows
multiple connections to be established. In polyamorous mode (set by
an option), the sender requests a paritcular pipe by setting it on the
message.
We default to PAIR_V1 now.
Diffstat (limited to 'src/protocol/pair/pair.c')
| -rw-r--r-- | src/protocol/pair/pair.c | 307 |
1 files changed, 0 insertions, 307 deletions
diff --git a/src/protocol/pair/pair.c b/src/protocol/pair/pair.c deleted file mode 100644 index 59465293..00000000 --- a/src/protocol/pair/pair.c +++ /dev/null @@ -1,307 +0,0 @@ -// -// Copyright 2017 Garrett D'Amore <garrett@damore.org> -// Copyright 2017 Capitar IT Group BV <info@capitar.com> -// -// This software is supplied under the terms of the MIT License, a -// copy of which should be located in the distribution where this -// file was obtained (LICENSE.txt). A copy of the license may also be -// found online at https://opensource.org/licenses/MIT. -// - -#include <stdlib.h> -#include <string.h> - -#include "core/nng_impl.h" - -// Pair protocol. The PAIR protocol is a simple 1:1 messaging pattern. -// While a peer is connected to the server, all other peer connection -// attempts are discarded. - -typedef struct nni_pair_pipe nni_pair_pipe; -typedef struct nni_pair_sock nni_pair_sock; - -static void nni_pair_send_cb(void *); -static void nni_pair_recv_cb(void *); -static void nni_pair_getq_cb(void *); -static void nni_pair_putq_cb(void *); -static void nni_pair_pipe_fini(void *); - -// An nni_pair_sock is our per-socket protocol private structure. -struct nni_pair_sock { - nni_sock * nsock; - nni_pair_pipe *ppipe; - nni_msgq * uwq; - nni_msgq * urq; - int raw; - nni_mtx mtx; -}; - -// An nni_pair_pipe is our per-pipe protocol private structure. We keep -// one of these even though in theory we'd only have a single underlying -// pipe. The separate data structure is more like other protocols that do -// manage multiple pipes. -struct nni_pair_pipe { - nni_pipe * npipe; - nni_pair_sock *psock; - nni_aio aio_send; - nni_aio aio_recv; - nni_aio aio_getq; - nni_aio aio_putq; -}; - -static int -nni_pair_sock_init(void **sp, nni_sock *nsock) -{ - nni_pair_sock *psock; - int rv; - - if ((psock = NNI_ALLOC_STRUCT(psock)) == NULL) { - return (NNG_ENOMEM); - } - if ((rv = nni_mtx_init(&psock->mtx)) != 0) { - NNI_FREE_STRUCT(psock); - return (rv); - } - psock->nsock = nsock; - psock->ppipe = NULL; - psock->raw = 0; - psock->uwq = nni_sock_sendq(nsock); - psock->urq = nni_sock_recvq(nsock); - *sp = psock; - return (0); -} - -static void -nni_pair_sock_fini(void *arg) -{ - nni_pair_sock *psock = arg; - - if (psock != NULL) { - nni_mtx_fini(&psock->mtx); - - NNI_FREE_STRUCT(psock); - } -} - -static int -nni_pair_pipe_init(void **pp, nni_pipe *npipe, void *psock) -{ - nni_pair_pipe *ppipe; - int rv; - - if ((ppipe = NNI_ALLOC_STRUCT(ppipe)) == NULL) { - return (NNG_ENOMEM); - } - rv = nni_aio_init(&ppipe->aio_send, nni_pair_send_cb, ppipe); - if (rv != 0) { - goto fail; - } - rv = nni_aio_init(&ppipe->aio_recv, nni_pair_recv_cb, ppipe); - if (rv != 0) { - goto fail; - } - rv = nni_aio_init(&ppipe->aio_getq, nni_pair_getq_cb, ppipe); - if (rv != 0) { - goto fail; - } - rv = nni_aio_init(&ppipe->aio_putq, nni_pair_putq_cb, ppipe); - if (rv != 0) { - goto fail; - } - ppipe->npipe = npipe; - ppipe->psock = psock; - *pp = ppipe; - return (0); - -fail: - nni_pair_pipe_fini(ppipe); - return (rv); -} - -static void -nni_pair_pipe_fini(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_aio_fini(&ppipe->aio_send); - nni_aio_fini(&ppipe->aio_recv); - nni_aio_fini(&ppipe->aio_putq); - nni_aio_fini(&ppipe->aio_getq); - NNI_FREE_STRUCT(ppipe); -} - -static int -nni_pair_pipe_start(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_pair_sock *psock = ppipe->psock; - - nni_mtx_lock(&psock->mtx); - if (psock->ppipe != NULL) { - nni_mtx_unlock(&psock->mtx); - return (NNG_EBUSY); // Already have a peer, denied. - } - psock->ppipe = ppipe; - nni_mtx_unlock(&psock->mtx); - - // Schedule a getq on the upper, and a read from the pipe. - // Each of these also sets up another hold on the pipe itself. - nni_msgq_aio_get(psock->uwq, &ppipe->aio_getq); - nni_pipe_recv(ppipe->npipe, &ppipe->aio_recv); - - return (0); -} - -static void -nni_pair_pipe_stop(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_pair_sock *psock = ppipe->psock; - - nni_aio_cancel(&ppipe->aio_send, NNG_ECANCELED); - nni_aio_cancel(&ppipe->aio_recv, NNG_ECANCELED); - nni_aio_cancel(&ppipe->aio_putq, NNG_ECANCELED); - nni_aio_cancel(&ppipe->aio_getq, NNG_ECANCELED); - - nni_mtx_lock(&psock->mtx); - if (psock->ppipe == ppipe) { - psock->ppipe = NULL; - } - nni_mtx_unlock(&psock->mtx); -} - -static void -nni_pair_recv_cb(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_pair_sock *psock = ppipe->psock; - - if (nni_aio_result(&ppipe->aio_recv) != 0) { - nni_pipe_stop(ppipe->npipe); - return; - } - - ppipe->aio_putq.a_msg = ppipe->aio_recv.a_msg; - ppipe->aio_recv.a_msg = NULL; - nni_msgq_aio_put(psock->urq, &ppipe->aio_putq); -} - -static void -nni_pair_putq_cb(void *arg) -{ - nni_pair_pipe *ppipe = arg; - - if (nni_aio_result(&ppipe->aio_putq) != 0) { - nni_msg_free(ppipe->aio_putq.a_msg); - ppipe->aio_putq.a_msg = NULL; - nni_pipe_stop(ppipe->npipe); - return; - } - nni_pipe_recv(ppipe->npipe, &ppipe->aio_recv); -} - -static void -nni_pair_getq_cb(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_pair_sock *psock = ppipe->psock; - - if (nni_aio_result(&ppipe->aio_getq) != 0) { - nni_pipe_stop(ppipe->npipe); - return; - } - - ppipe->aio_send.a_msg = ppipe->aio_getq.a_msg; - ppipe->aio_getq.a_msg = NULL; - nni_pipe_send(ppipe->npipe, &ppipe->aio_send); -} - -static void -nni_pair_send_cb(void *arg) -{ - nni_pair_pipe *ppipe = arg; - nni_pair_sock *psock = ppipe->psock; - - if (nni_aio_result(&ppipe->aio_send) != 0) { - nni_msg_free(ppipe->aio_send.a_msg); - ppipe->aio_send.a_msg = NULL; - nni_pipe_stop(ppipe->npipe); - return; - } - - nni_msgq_aio_get(psock->uwq, &ppipe->aio_getq); -} - -static void -nni_pair_sock_open(void *arg) -{ - NNI_ARG_UNUSED(arg); -} - -static void -nni_pair_sock_close(void *arg) -{ - NNI_ARG_UNUSED(arg); -} - -static int -nni_pair_sock_setopt(void *arg, int opt, const void *buf, size_t sz) -{ - nni_pair_sock *psock = arg; - int rv; - - switch (opt) { - case NNG_OPT_RAW: - rv = nni_setopt_int(&psock->raw, buf, sz, 0, 1); - break; - default: - rv = NNG_ENOTSUP; - } - return (rv); -} - -static int -nni_pair_sock_getopt(void *arg, int opt, void *buf, size_t *szp) -{ - nni_pair_sock *psock = arg; - int rv; - - switch (opt) { - case NNG_OPT_RAW: - rv = nni_getopt_int(&psock->raw, buf, szp); - break; - default: - rv = NNG_ENOTSUP; - } - return (rv); -} - -static nni_proto_pipe_ops nni_pair_pipe_ops = { - .pipe_init = nni_pair_pipe_init, - .pipe_fini = nni_pair_pipe_fini, - .pipe_start = nni_pair_pipe_start, - .pipe_stop = nni_pair_pipe_stop, -}; - -static nni_proto_sock_ops nni_pair_sock_ops = { - .sock_init = nni_pair_sock_init, - .sock_fini = nni_pair_sock_fini, - .sock_open = nni_pair_sock_open, - .sock_close = nni_pair_sock_close, - .sock_setopt = nni_pair_sock_setopt, - .sock_getopt = nni_pair_sock_getopt, -}; - -nni_proto nni_pair_proto = { - .proto_version = NNI_PROTOCOL_VERSION, - .proto_self = { NNG_PROTO_PAIR_V0, "pair" }, - .proto_peer = { NNG_PROTO_PAIR_V0, "pair" }, - .proto_flags = NNI_PROTO_FLAG_SNDRCV, - .proto_sock_ops = &nni_pair_sock_ops, - .proto_pipe_ops = &nni_pair_pipe_ops, -}; - -int -nng_pair0_open(nng_socket *sidp) -{ - return (nni_proto_open(sidp, &nni_pair_proto)); -} |
