rbczmq 1.6.2 → 1.6.4

data/ext/zeromq/src/poller.hpp

@@ -0,0 +1,82 @@
+/*
+    Copyright (c) 2010-2011 250bpm s.r.o.
+    Copyright (c) 2007-2009 iMatix Corporation
+    Copyright (c) 2007-2011 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef __ZMQ_POLLER_HPP_INCLUDED__
+#define __ZMQ_POLLER_HPP_INCLUDED__
+
+#include "platform.hpp"
+
+#if defined ZMQ_FORCE_SELECT
+#define ZMQ_USE_SELECT
+#include "select.hpp"
+#elif defined ZMQ_FORCE_POLL
+#define ZMQ_USE_POLL
+#include "poll.hpp"
+#elif defined ZMQ_FORCE_EPOLL
+#define ZMQ_USE_EPOLL
+#include "epoll.hpp"
+#elif defined ZMQ_FORCE_DEVPOLL
+#define ZMQ_USE_DEVPOLL
+#include "devpoll.hpp"
+#elif defined ZMQ_FORCE_KQUEUE
+#define ZMQ_USE_KQUEUE
+#include "kqueue.hpp"
+#elif defined ZMQ_HAVE_LINUX
+#define ZMQ_USE_EPOLL
+#include "epoll.hpp"
+#elif defined ZMQ_HAVE_WINDOWS
+#define ZMQ_USE_SELECT
+#include "select.hpp"
+#elif defined ZMQ_HAVE_FREEBSD
+#define ZMQ_USE_KQUEUE
+#include "kqueue.hpp"
+#elif defined ZMQ_HAVE_OPENBSD
+#define ZMQ_USE_KQUEUE
+#include "kqueue.hpp"
+#elif defined ZMQ_HAVE_NETBSD
+#define ZMQ_USE_KQUEUE
+#include "kqueue.hpp"
+#elif defined ZMQ_HAVE_SOLARIS
+#define ZMQ_USE_DEVPOLL
+#include "devpoll.hpp"
+#elif defined ZMQ_HAVE_OSX
+#define ZMQ_USE_KQUEUE
+#include "kqueue.hpp"
+#elif defined ZMQ_HAVE_QNXNTO
+#define ZMQ_USE_POLL
+#include "poll.hpp"
+#elif defined ZMQ_HAVE_AIX
+#define ZMQ_USE_POLL
+#include "poll.hpp"
+#elif defined ZMQ_HAVE_HPUX
+#define ZMQ_USE_DEVPOLL
+#include "devpoll.hpp"
+#elif defined ZMQ_HAVE_OPENVMS
+#define ZMQ_USE_SELECT
+#include "select.hpp"
+#elif defined ZMQ_HAVE_CYGWIN
+#define ZMQ_USE_SELECT
+#include "select.hpp"
+#else
+#error Unsupported platform
+#endif
+
+#endif

data/ext/zeromq/src/poller_base.cpp

@@ -0,0 +1,99 @@
+/*
+    Copyright (c) 2010-2011 250bpm s.r.o.
+    Copyright (c) 2010-2011 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "poller_base.hpp"
+#include "i_poll_events.hpp"
+#include "err.hpp"
+
+zmq::poller_base_t::poller_base_t ()
+{
+}
+
+zmq::poller_base_t::~poller_base_t ()
+{
+    //  Make sure there is no more load on the shutdown.
+    zmq_assert (get_load () == 0);
+}
+
+int zmq::poller_base_t::get_load ()
+{
+    return load.get ();
+}
+
+void zmq::poller_base_t::adjust_load (int amount_)
+{
+    if (amount_ > 0)
+        load.add (amount_);
+    else if (amount_ < 0)
+        load.sub (-amount_);
+}
+
+void zmq::poller_base_t::add_timer (int timeout_, i_poll_events *sink_, int id_)
+{
+    uint64_t expiration = clock.now_ms () + timeout_;
+    timer_info_t info = {sink_, id_};
+    timers.insert (timers_t::value_type (expiration, info));
+}
+
+void zmq::poller_base_t::cancel_timer (i_poll_events *sink_, int id_)
+{
+    //  Complexity of this operation is O(n). We assume it is rarely used.
+    for (timers_t::iterator it = timers.begin (); it != timers.end (); ++it)
+        if (it->second.sink == sink_ && it->second.id == id_) {
+            timers.erase (it);
+            return;
+        }
+
+    //  Timer not found.
+    zmq_assert (false);
+}
+
+uint64_t zmq::poller_base_t::execute_timers ()
+{
+    //  Fast track.
+    if (timers.empty ())
+        return 0;
+
+    //  Get the current time.
+    uint64_t current = clock.now_ms ();
+
+    //   Execute the timers that are already due.
+    timers_t::iterator it = timers.begin ();
+    while (it != timers.end ()) {
+
+        //  If we have to wait to execute the item, same will be true about
+        //  all the following items (multimap is sorted). Thus we can stop
+        //  checking the subsequent timers and return the time to wait for
+        //  the next timer (at least 1ms).
+        if (it->first > current)
+            return it->first - current;
+
+        //  Trigger the timer.
+        it->second.sink->timer_event (it->second.id);
+
+        //  Remove it from the list of active timers.
+        timers_t::iterator o = it;
+        ++it;
+        timers.erase (o);
+    }
+
+    //  There are no more timers.
+    return 0;
+}

data/ext/zeromq/src/poller_base.hpp

@@ -0,0 +1,86 @@
+/*
+    Copyright (c) 2010-2011 250bpm s.r.o.
+    Copyright (c) 2010-2011 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef __ZMQ_POLLER_BASE_HPP_INCLUDED__
+#define __ZMQ_POLLER_BASE_HPP_INCLUDED__
+
+#include <map>
+
+#include "clock.hpp"
+#include "atomic_counter.hpp"
+
+namespace zmq
+{
+
+    struct i_poll_events;
+
+    class poller_base_t
+    {
+    public:
+
+        poller_base_t ();
+        virtual ~poller_base_t ();
+
+        //  Returns load of the poller. Note that this function can be
+        //  invoked from a different thread!
+        int get_load ();
+
+        //  Add a timeout to expire in timeout_ milliseconds. After the
+        //  expiration timer_event on sink_ object will be called with
+        //  argument set to id_.
+        void add_timer (int timeout_, zmq::i_poll_events *sink_, int id_);
+
+        //  Cancel the timer created by sink_ object with ID equal to id_.
+        void cancel_timer (zmq::i_poll_events *sink_, int id_);
+
+    protected:
+
+        //  Called by individual poller implementations to manage the load.
+        void adjust_load (int amount_);
+
+        //  Executes any timers that are due. Returns number of milliseconds
+        //  to wait to match the next timer or 0 meaning "no timers".
+        uint64_t execute_timers ();
+
+    private:
+
+        //  Clock instance private to this I/O thread.
+        clock_t clock;
+
+        //  List of active timers.
+        struct timer_info_t
+        {
+            zmq::i_poll_events *sink;
+            int id;
+        };
+        typedef std::multimap <uint64_t, timer_info_t> timers_t;
+        timers_t timers;
+
+        //  Load of the poller. Currently the number of file descriptors
+        //  registered.
+        atomic_counter_t load;
+
+        poller_base_t (const poller_base_t&);
+        const poller_base_t &operator = (const poller_base_t&);
+    };
+
+}
+
+#endif

data/ext/zeromq/src/precompiled.cpp

@@ -0,0 +1,21 @@
+/*
+    Copyright (c) 2011 250bpm s.r.o.
+    Copyright (c) 2011 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "precompiled.hpp"

data/ext/zeromq/src/precompiled.hpp

@@ -0,0 +1,47 @@
+/*
+    Copyright (c) 2011 250bpm s.r.o.
+    Copyright (c) 2011 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#ifndef __ZMQ_PRECOMPILED_HPP_INCLUDED__
+#define __ZMQ_PRECOMPILED_HPP_INCLUDED__
+
+#ifdef _MSC_VER
+
+// Windows headers
+#include "platform.hpp"
+#include "windows.hpp"
+#include <fcntl.h>
+#include <intrin.h>
+#include <io.h>
+#include <rpc.h>
+#include <sys/stat.h>
+
+// standard C++ headers
+#include <algorithm>
+#include <map>
+#include <set>
+#include <string>
+#include <vector>
+
+// 0MQ definitions and exported functions
+#include "../include/zmq.h"
+
+#endif // _MSC_VER
+
+#endif

data/ext/zeromq/src/proxy.cpp

@@ -0,0 +1,150 @@
+/*
+    Copyright (c) 2007-2012 iMatix Corporation
+    Copyright (c) 2007-2012 Other contributors as noted in the AUTHORS file
+
+    This file is part of 0MQ.
+
+    0MQ is free software; you can redistribute it and/or modify it under
+    the terms of the GNU Lesser 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
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <stddef.h>
+#include "platform.hpp"
+#include "proxy.hpp"
+#include "likely.hpp"
+
+#if defined ZMQ_FORCE_SELECT
+#define ZMQ_POLL_BASED_ON_SELECT
+#elif defined ZMQ_FORCE_POLL
+#define ZMQ_POLL_BASED_ON_POLL
+#elif defined ZMQ_HAVE_LINUX || defined ZMQ_HAVE_FREEBSD ||\
+    defined ZMQ_HAVE_OPENBSD || defined ZMQ_HAVE_SOLARIS ||\
+    defined ZMQ_HAVE_OSX || defined ZMQ_HAVE_QNXNTO ||\
+    defined ZMQ_HAVE_HPUX || defined ZMQ_HAVE_AIX ||\
+    defined ZMQ_HAVE_NETBSD
+#define ZMQ_POLL_BASED_ON_POLL
+#elif defined ZMQ_HAVE_WINDOWS || defined ZMQ_HAVE_OPENVMS ||\
+     defined ZMQ_HAVE_CYGWIN
+#define ZMQ_POLL_BASED_ON_SELECT
+#endif
+
+//  On AIX platform, poll.h has to be included first to get consistent
+//  definition of pollfd structure (AIX uses 'reqevents' and 'retnevents'
+//  instead of 'events' and 'revents' and defines macros to map from POSIX-y
+//  names to AIX-specific names).
+#if defined ZMQ_POLL_BASED_ON_POLL
+#include <poll.h>
+#endif
+
+// These headers end up pulling in zmq.h somewhere in their include
+// dependency chain
+#include "socket_base.hpp"
+#include "err.hpp"
+
+// zmq.h must be included *after* poll.h for AIX to build properly
+#include "../include/zmq.h"
+
+
+int zmq::proxy (
+    class socket_base_t *frontend_,
+    class socket_base_t *backend_,
+    class socket_base_t *capture_)
+{
+    msg_t msg;
+    int rc = msg.init ();
+    if (rc != 0)
+        return -1;
+
+    //  The algorithm below assumes ratio of requests and replies processed
+    //  under full load to be 1:1.
+
+    int more;
+    size_t moresz;
+    zmq_pollitem_t items [] = {
+        { frontend_, 0, ZMQ_POLLIN, 0 },
+        { backend_, 0, ZMQ_POLLIN, 0 }
+    };
+    while (true) {
+        //  Wait while there are either requests or replies to process.
+        rc = zmq_poll (&items [0], 2, -1);
+        if (unlikely (rc < 0))
+            return -1;
+
+        //  Process a request
+        if (items [0].revents & ZMQ_POLLIN) {
+            while (true) {
+                rc = frontend_->recv (&msg, 0);
+                if (unlikely (rc < 0))
+                    return -1;
+
+                moresz = sizeof more;
+                rc = frontend_->getsockopt (ZMQ_RCVMORE, &more, &moresz);
+                if (unlikely (rc < 0))
+                    return -1;
+
+                //  Copy message to capture socket if any
+                if (capture_) {
+                    msg_t ctrl;
+                    rc = ctrl.init ();
+                    if (unlikely (rc < 0))
+                        return -1;
+                    rc = ctrl.copy (msg);
+                    if (unlikely (rc < 0))
+                        return -1;
+                    rc = capture_->send (&ctrl, more? ZMQ_SNDMORE: 0);
+                    if (unlikely (rc < 0))
+                        return -1;
+                }
+                rc = backend_->send (&msg, more? ZMQ_SNDMORE: 0);
+                if (unlikely (rc < 0))
+                    return -1;
+                if (more == 0)
+                    break;
+            }
+        }
+        //  Process a reply
+        if (items [1].revents & ZMQ_POLLIN) {
+            while (true) {
+                rc = backend_->recv (&msg, 0);
+                if (unlikely (rc < 0))
+                    return -1;
+
+                moresz = sizeof more;
+                rc = backend_->getsockopt (ZMQ_RCVMORE, &more, &moresz);
+                if (unlikely (rc < 0))
+                    return -1;
+
+                //  Copy message to capture socket if any
+                if (capture_) {
+                    msg_t ctrl;
+                    rc = ctrl.init ();
+                    if (unlikely (rc < 0))
+                        return -1;
+                    rc = ctrl.copy (msg);
+                    if (unlikely (rc < 0))
+                        return -1;
+                    rc = capture_->send (&ctrl, more? ZMQ_SNDMORE: 0);
+                    if (unlikely (rc < 0))
+                        return -1;
+                }
+                rc = frontend_->send (&msg, more? ZMQ_SNDMORE: 0);
+                if (unlikely (rc < 0))
+                    return -1;
+                if (more == 0)
+                    break;
+            }
+        }
+
+    }
+    return 0;
+}