diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/socket_base.cpp | 27 | ||||
| -rw-r--r-- | src/socket_base.hpp | 3 |
2 files changed, 25 insertions, 5 deletions
diff --git a/src/socket_base.cpp b/src/socket_base.cpp index ddf2470..3b74359 100644 --- a/src/socket_base.cpp +++ b/src/socket_base.cpp @@ -42,6 +42,7 @@ zmq::socket_base_t::socket_base_t (app_thread_t *parent_) : object_t (parent_), pending_term_acks (0), ticks (0), + rcvmore (false), app_thread (parent_), shutting_down (false), sent_seqnum (0), @@ -70,7 +71,16 @@ int zmq::socket_base_t::setsockopt (int option_, const void *optval_, int zmq::socket_base_t::getsockopt (int option_, void *optval_, size_t *optvallen_) { - // At the moment there are no socket-type-specific overloads of getsockopt. + if (option_ == ZMQ_RCVMORE) { + if (*optvallen_ < sizeof (int64_t)) { + errno = EINVAL; + return -1; + } + *((int64_t*) optval_) = rcvmore ? 1 : 0; + *optvallen_ = sizeof (int64_t); + return 0; + } + return options.getsockopt (option_, optval_, optvallen_); } @@ -318,9 +328,8 @@ int zmq::socket_base_t::connect (const char *addr_) int zmq::socket_base_t::send (::zmq_msg_t *msg_, int flags_) { - // ZMQ_MORE is actually a message flag, not a real send-flag - // such as ZMQ_NOBLOCK. At this point we impose it on the message. - if (flags_ & ZMQ_MORE) + // At this point we impose the MORE flag on the message. + if (flags_ & ZMQ_SNDMORE) msg_->flags |= ZMQ_MSG_MORE; // Process pending commands, if any. @@ -367,8 +376,12 @@ int zmq::socket_base_t::recv (::zmq_msg_t *msg_, int flags_) } // If we have the message, return immediately. - if (rc == 0) + if (rc == 0) { + rcvmore = msg_->flags & ZMQ_MSG_MORE; + if (rcvmore) + msg_->flags &= ~ZMQ_MSG_MORE; return 0; + } // If we don't have the message, restore the original cause of the problem. errno = err; @@ -393,6 +406,10 @@ int zmq::socket_base_t::recv (::zmq_msg_t *msg_, int flags_) rc = xrecv (msg_, flags_); ticks = 0; } + + rcvmore = msg_->flags & ZMQ_MSG_MORE; + if (rcvmore) + msg_->flags &= ~ZMQ_MSG_MORE; return 0; } diff --git a/src/socket_base.hpp b/src/socket_base.hpp index 1b70299..3d95cec 100644 --- a/src/socket_base.hpp +++ b/src/socket_base.hpp @@ -141,6 +141,9 @@ namespace zmq // Number of messages received since last command processing. int ticks; + // If true there's a half-read message in the socket. + bool rcvmore; + // Application thread the socket lives in. class app_thread_t *app_thread; |