| /* |
| Copyright (c) 2012-2013 Martin Sustrik All rights reserved. |
| Copyright 2016 Garrett D'Amore <garrett@damore.org> |
| |
| Permission is hereby granted, free of charge, to any person obtaining a copy |
| of this software and associated documentation files (the "Software"), |
| to deal in the Software without restriction, including without limitation |
| the rights to use, copy, modify, merge, publish, distribute, sublicense, |
| and/or sell copies of the Software, and to permit persons to whom |
| the Software is furnished to do so, subject to the following conditions: |
| |
| The above copyright notice and this permission notice shall be included |
| in all copies or substantial portions of the Software. |
| |
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL |
| THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
| FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
| IN THE SOFTWARE. |
| */ |
| |
| #include "../transport.h" |
| #include "../protocol.h" |
| |
| #include "sock.h" |
| #include "ep.h" |
| |
| #include "../utils/err.h" |
| #include "../utils/fast.h" |
| |
| /* Internal pipe states. */ |
| #define NN_PIPEBASE_STATE_IDLE 1 |
| #define NN_PIPEBASE_STATE_ACTIVE 2 |
| #define NN_PIPEBASE_STATE_FAILED 3 |
| |
| #define NN_PIPEBASE_INSTATE_DEACTIVATED 0 |
| #define NN_PIPEBASE_INSTATE_IDLE 1 |
| #define NN_PIPEBASE_INSTATE_RECEIVING 2 |
| #define NN_PIPEBASE_INSTATE_RECEIVED 3 |
| #define NN_PIPEBASE_INSTATE_ASYNC 4 |
| |
| #define NN_PIPEBASE_OUTSTATE_DEACTIVATED 0 |
| #define NN_PIPEBASE_OUTSTATE_IDLE 1 |
| #define NN_PIPEBASE_OUTSTATE_SENDING 2 |
| #define NN_PIPEBASE_OUTSTATE_SENT 3 |
| #define NN_PIPEBASE_OUTSTATE_ASYNC 4 |
| |
| void nn_pipebase_init (struct nn_pipebase *self, |
| const struct nn_pipebase_vfptr *vfptr, struct nn_ep *ep) |
| { |
| nn_assert (ep->sock); |
| |
| nn_fsm_init (&self->fsm, NULL, NULL, 0, self, &ep->sock->fsm); |
| self->vfptr = vfptr; |
| self->state = NN_PIPEBASE_STATE_IDLE; |
| self->instate = NN_PIPEBASE_INSTATE_DEACTIVATED; |
| self->outstate = NN_PIPEBASE_OUTSTATE_DEACTIVATED; |
| self->sock = ep->sock; |
| memcpy (&self->options, &ep->options, sizeof (struct nn_ep_options)); |
| nn_fsm_event_init (&self->in); |
| nn_fsm_event_init (&self->out); |
| } |
| |
| void nn_pipebase_term (struct nn_pipebase *self) |
| { |
| nn_assert_state (self, NN_PIPEBASE_STATE_IDLE); |
| |
| nn_fsm_event_term (&self->out); |
| nn_fsm_event_term (&self->in); |
| nn_fsm_term (&self->fsm); |
| } |
| |
| int nn_pipebase_start (struct nn_pipebase *self) |
| { |
| int rc; |
| |
| nn_assert_state (self, NN_PIPEBASE_STATE_IDLE); |
| |
| self->state = NN_PIPEBASE_STATE_ACTIVE; |
| self->instate = NN_PIPEBASE_INSTATE_ASYNC; |
| self->outstate = NN_PIPEBASE_OUTSTATE_IDLE; |
| rc = nn_sock_add (self->sock, (struct nn_pipe*) self); |
| if (nn_slow (rc < 0)) { |
| self->state = NN_PIPEBASE_STATE_FAILED; |
| return rc; |
| } |
| nn_fsm_raise (&self->fsm, &self->out, NN_PIPE_OUT); |
| |
| return 0; |
| } |
| |
| void nn_pipebase_stop (struct nn_pipebase *self) |
| { |
| if (self->state == NN_PIPEBASE_STATE_ACTIVE) |
| nn_sock_rm (self->sock, (struct nn_pipe*) self); |
| self->state = NN_PIPEBASE_STATE_IDLE; |
| } |
| |
| void nn_pipebase_received (struct nn_pipebase *self) |
| { |
| if (nn_fast (self->instate == NN_PIPEBASE_INSTATE_RECEIVING)) { |
| self->instate = NN_PIPEBASE_INSTATE_RECEIVED; |
| return; |
| } |
| nn_assert (self->instate == NN_PIPEBASE_INSTATE_ASYNC); |
| self->instate = NN_PIPEBASE_INSTATE_IDLE; |
| nn_fsm_raise (&self->fsm, &self->in, NN_PIPE_IN); |
| } |
| |
| void nn_pipebase_sent (struct nn_pipebase *self) |
| { |
| if (nn_fast (self->outstate == NN_PIPEBASE_OUTSTATE_SENDING)) { |
| self->outstate = NN_PIPEBASE_OUTSTATE_SENT; |
| return; |
| } |
| nn_assert (self->outstate == NN_PIPEBASE_OUTSTATE_ASYNC); |
| self->outstate = NN_PIPEBASE_OUTSTATE_IDLE; |
| nn_fsm_raise (&self->fsm, &self->out, NN_PIPE_OUT); |
| } |
| |
| void nn_pipebase_getopt (struct nn_pipebase *self, int level, int option, |
| void *optval, size_t *optvallen) |
| { |
| int rc; |
| int intval; |
| |
| if (level == NN_SOL_SOCKET) { |
| switch (option) { |
| |
| /* Endpoint options */ |
| case NN_SNDPRIO: |
| intval = self->options.sndprio; |
| break; |
| case NN_RCVPRIO: |
| intval = self->options.rcvprio; |
| break; |
| case NN_IPV4ONLY: |
| intval = self->options.ipv4only; |
| break; |
| |
| /* Fallback to socket options */ |
| default: |
| rc = nn_sock_getopt_inner (self->sock, level, |
| option, optval, optvallen); |
| errnum_assert (rc == 0, -rc); |
| return; |
| } |
| |
| memcpy (optval, &intval, |
| *optvallen < sizeof (int) ? *optvallen : sizeof (int)); |
| *optvallen = sizeof (int); |
| |
| return; |
| } |
| |
| rc = nn_sock_getopt_inner (self->sock, level, option, optval, optvallen); |
| errnum_assert (rc == 0, -rc); |
| } |
| |
| int nn_pipebase_ispeer (struct nn_pipebase *self, int socktype) |
| { |
| return nn_sock_ispeer (self->sock, socktype); |
| } |
| |
| void nn_pipe_setdata (struct nn_pipe *self, void *data) |
| { |
| ((struct nn_pipebase*) self)->data = data; |
| } |
| |
| void *nn_pipe_getdata (struct nn_pipe *self) |
| { |
| return ((struct nn_pipebase*) self)->data; |
| } |
| |
| int nn_pipe_send (struct nn_pipe *self, struct nn_msg *msg) |
| { |
| int rc; |
| struct nn_pipebase *pipebase; |
| |
| pipebase = (struct nn_pipebase*) self; |
| nn_assert (pipebase->outstate == NN_PIPEBASE_OUTSTATE_IDLE); |
| pipebase->outstate = NN_PIPEBASE_OUTSTATE_SENDING; |
| rc = pipebase->vfptr->send (pipebase, msg); |
| errnum_assert (rc >= 0, -rc); |
| if (nn_fast (pipebase->outstate == NN_PIPEBASE_OUTSTATE_SENT)) { |
| pipebase->outstate = NN_PIPEBASE_OUTSTATE_IDLE; |
| return rc; |
| } |
| nn_assert (pipebase->outstate == NN_PIPEBASE_OUTSTATE_SENDING); |
| pipebase->outstate = NN_PIPEBASE_OUTSTATE_ASYNC; |
| return rc | NN_PIPEBASE_RELEASE; |
| } |
| |
| int nn_pipe_recv (struct nn_pipe *self, struct nn_msg *msg) |
| { |
| int rc; |
| struct nn_pipebase *pipebase; |
| |
| pipebase = (struct nn_pipebase*) self; |
| nn_assert (pipebase->instate == NN_PIPEBASE_INSTATE_IDLE); |
| pipebase->instate = NN_PIPEBASE_INSTATE_RECEIVING; |
| rc = pipebase->vfptr->recv (pipebase, msg); |
| errnum_assert (rc >= 0, -rc); |
| |
| if (nn_fast (pipebase->instate == NN_PIPEBASE_INSTATE_RECEIVED)) { |
| pipebase->instate = NN_PIPEBASE_INSTATE_IDLE; |
| return rc; |
| } |
| nn_assert (pipebase->instate == NN_PIPEBASE_INSTATE_RECEIVING); |
| pipebase->instate = NN_PIPEBASE_INSTATE_ASYNC; |
| return rc | NN_PIPEBASE_RELEASE; |
| } |
| |
| void nn_pipe_getopt (struct nn_pipe *self, int level, int option, |
| void *optval, size_t *optvallen) |
| { |
| |
| struct nn_pipebase *pipebase; |
| |
| pipebase = (struct nn_pipebase*) self; |
| nn_pipebase_getopt (pipebase, level, option, optval, optvallen); |
| } |