2009-09-10 12:00:47 +02:00
|
|
|
/*
|
2010-01-05 08:29:35 +01:00
|
|
|
Copyright (c) 2007-2010 iMatix Corporation
|
2009-09-10 12:00:47 +02:00
|
|
|
|
|
|
|
This file is part of 0MQ.
|
|
|
|
|
|
|
|
0MQ is free software; you can redistribute it and/or modify it under
|
|
|
|
the terms of the Lesser GNU General Public License as published by
|
|
|
|
the Free Software Foundation; either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
0MQ is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
Lesser GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the Lesser GNU General Public License
|
|
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2009-12-28 21:29:31 +01:00
|
|
|
#include <string.h>
|
|
|
|
|
2009-09-16 16:49:09 +02:00
|
|
|
#include "../bindings/c/zmq.h"
|
2009-09-10 12:00:47 +02:00
|
|
|
|
|
|
|
#include "sub.hpp"
|
|
|
|
#include "err.hpp"
|
|
|
|
|
|
|
|
zmq::sub_t::sub_t (class app_thread_t *parent_) :
|
2009-09-21 17:20:13 +02:00
|
|
|
socket_base_t (parent_),
|
2009-09-11 16:23:16 +02:00
|
|
|
all_count (0)
|
2009-09-10 12:00:47 +02:00
|
|
|
{
|
2009-09-21 17:20:13 +02:00
|
|
|
options.requires_in = true;
|
|
|
|
options.requires_out = false;
|
2009-09-10 12:00:47 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
zmq::sub_t::~sub_t ()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2009-09-21 14:39:59 +02:00
|
|
|
void zmq::sub_t::xattach_pipes (class reader_t *inpipe_,
|
|
|
|
class writer_t *outpipe_)
|
|
|
|
{
|
2009-12-15 09:09:19 +01:00
|
|
|
zmq_assert (inpipe_ && !outpipe_);
|
|
|
|
fq.attach (inpipe_);
|
2009-09-21 14:39:59 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void zmq::sub_t::xdetach_inpipe (class reader_t *pipe_)
|
|
|
|
{
|
2009-12-15 09:09:19 +01:00
|
|
|
zmq_assert (pipe_);
|
|
|
|
fq.detach (pipe_);
|
2009-09-21 14:39:59 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void zmq::sub_t::xdetach_outpipe (class writer_t *pipe_)
|
|
|
|
{
|
2009-12-15 09:09:19 +01:00
|
|
|
// SUB socket is read-only thus there should be no outpipes.
|
2009-09-21 14:39:59 +02:00
|
|
|
zmq_assert (false);
|
|
|
|
}
|
|
|
|
|
|
|
|
void zmq::sub_t::xkill (class reader_t *pipe_)
|
|
|
|
{
|
2009-12-15 09:09:19 +01:00
|
|
|
fq.kill (pipe_);
|
2009-09-21 14:39:59 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
void zmq::sub_t::xrevive (class reader_t *pipe_)
|
|
|
|
{
|
2009-12-15 09:09:19 +01:00
|
|
|
fq.revive (pipe_);
|
2009-09-21 14:39:59 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
int zmq::sub_t::xsetsockopt (int option_, const void *optval_,
|
2009-09-10 12:00:47 +02:00
|
|
|
size_t optvallen_)
|
|
|
|
{
|
|
|
|
if (option_ == ZMQ_SUBSCRIBE) {
|
2009-12-28 21:29:31 +01:00
|
|
|
if (!optvallen_)
|
2009-09-11 16:23:16 +02:00
|
|
|
all_count++;
|
2009-12-28 21:29:31 +01:00
|
|
|
else
|
|
|
|
subscriptions.insert (std::string ((const char*) optval_,
|
|
|
|
optvallen_));
|
2009-09-10 12:00:47 +02:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (option_ == ZMQ_UNSUBSCRIBE) {
|
2009-12-28 21:29:31 +01:00
|
|
|
if (!optvallen_) {
|
2009-09-11 16:23:16 +02:00
|
|
|
if (!all_count) {
|
|
|
|
errno = EINVAL;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
all_count--;
|
|
|
|
}
|
|
|
|
else {
|
2009-12-28 21:29:31 +01:00
|
|
|
subscriptions_t::iterator it = subscriptions.find (
|
|
|
|
std::string ((const char*) optval_, optvallen_));
|
|
|
|
if (it == subscriptions.end ()) {
|
2009-09-11 16:23:16 +02:00
|
|
|
errno = EINVAL;
|
|
|
|
return -1;
|
|
|
|
}
|
2009-12-28 21:29:31 +01:00
|
|
|
subscriptions.erase (it);
|
2009-09-10 12:00:47 +02:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2009-09-21 14:39:59 +02:00
|
|
|
errno = EINVAL;
|
|
|
|
return -1;
|
2009-09-10 12:00:47 +02:00
|
|
|
}
|
|
|
|
|
2009-09-23 10:22:54 +02:00
|
|
|
int zmq::sub_t::xsend (zmq_msg_t *msg_, int flags_)
|
2009-09-11 18:16:47 +02:00
|
|
|
{
|
2009-09-22 11:52:35 +02:00
|
|
|
errno = ENOTSUP;
|
2009-09-11 18:16:47 +02:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2009-09-21 14:39:59 +02:00
|
|
|
int zmq::sub_t::xflush ()
|
2009-09-11 18:16:47 +02:00
|
|
|
{
|
2009-09-22 11:52:35 +02:00
|
|
|
errno = ENOTSUP;
|
2009-09-11 18:16:47 +02:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2009-09-23 10:22:54 +02:00
|
|
|
int zmq::sub_t::xrecv (zmq_msg_t *msg_, int flags_)
|
2009-09-10 12:00:47 +02:00
|
|
|
{
|
2009-12-28 21:29:31 +01:00
|
|
|
// Get a message using fair queueing algorithm.
|
|
|
|
int rc = fq.recv (msg_, flags_);
|
2009-09-10 12:00:47 +02:00
|
|
|
|
2009-12-28 21:29:31 +01:00
|
|
|
// If there's no message available, return immediately.
|
|
|
|
if (rc != 0 && errno == EAGAIN)
|
|
|
|
return -1;
|
2009-09-10 12:00:47 +02:00
|
|
|
|
2009-12-28 21:29:31 +01:00
|
|
|
// If there is at least one * subscription, the message matches.
|
|
|
|
if (all_count)
|
|
|
|
return 0;
|
2009-09-11 16:23:16 +02:00
|
|
|
|
2009-12-28 21:29:31 +01:00
|
|
|
// Check whether the message matches at least one prefix subscription.
|
|
|
|
// TODO: Make this efficient - O(log(n)) where n is number of characters in
|
|
|
|
// the longest subscription string.
|
|
|
|
for (subscriptions_t::iterator it = subscriptions.begin ();
|
|
|
|
it != subscriptions.end (); it++) {
|
|
|
|
size_t msg_size = zmq_msg_size (msg_);
|
|
|
|
size_t sub_size = it->size ();
|
|
|
|
if (sub_size <= msg_size &&
|
|
|
|
memcmp (zmq_msg_data (msg_), it->data (), sub_size) == 0)
|
2009-09-11 16:23:16 +02:00
|
|
|
return 0;
|
|
|
|
}
|
2009-12-28 21:29:31 +01:00
|
|
|
|
|
|
|
// The message did not pass the filter. Trim it.
|
|
|
|
// Note that we are returning a different error code so that the caller
|
|
|
|
// knows there are more messages available. We cannot loop here as
|
|
|
|
// a stream of non-matching messages would create a DoS situation.
|
|
|
|
zmq_msg_close (msg_);
|
|
|
|
zmq_msg_init (msg_);
|
|
|
|
errno = EINPROGRESS;
|
|
|
|
return -1;
|
2009-09-10 12:00:47 +02:00
|
|
|
}
|
2009-09-21 14:39:59 +02:00
|
|
|
|
2009-10-01 10:56:17 +02:00
|
|
|
bool zmq::sub_t::xhas_in ()
|
|
|
|
{
|
2009-12-29 07:53:33 +01:00
|
|
|
return fq.has_in ();
|
2009-10-01 10:56:17 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
bool zmq::sub_t::xhas_out ()
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|