poll.hpp 3.03 KB
Newer Older
Martin Sustrik's avatar
Martin Sustrik committed
1
/*
2
    Copyright (c) 2007-2010 iMatix Corporation
Martin Sustrik's avatar
Martin Sustrik committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19

    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/>.
*/

Martin Sustrik's avatar
Martin Sustrik committed
20 21
#ifndef __ZMQ_POLL_HPP_INCLUDED__
#define __ZMQ_POLL_HPP_INCLUDED__
Martin Sustrik's avatar
Martin Sustrik committed
22 23 24

#include "platform.hpp"

Martin Sustrik's avatar
Martin Sustrik committed
25 26 27
#if defined ZMQ_HAVE_LINUX || defined ZMQ_HAVE_FREEBSD ||\
    defined ZMQ_HAVE_OPENBSD || defined ZMQ_HAVE_SOLARIS ||\
    defined ZMQ_HAVE_OSX || defined ZMQ_HAVE_QNXNTO ||\
Martin Lucina's avatar
Martin Lucina committed
28 29
    defined ZMQ_HAVE_HPUX || defined ZMQ_HAVE_AIX ||\
    defined ZMQ_HAVE_NETBSD
Martin Sustrik's avatar
Martin Sustrik committed
30 31 32 33 34 35 36 37 38

#include <poll.h>
#include <stddef.h>
#include <vector>

#include "fd.hpp"
#include "thread.hpp"
#include "atomic_counter.hpp"

Martin Sustrik's avatar
Martin Sustrik committed
39
namespace zmq
Martin Sustrik's avatar
Martin Sustrik committed
40 41 42 43 44
{

    //  Implements socket polling mechanism using the POSIX.1-2001
    //  poll() system call.

45
    class poll_t
Martin Sustrik's avatar
Martin Sustrik committed
46 47 48
    {
    public:

49 50
        typedef fd_t handle_t;

Martin Sustrik's avatar
Martin Sustrik committed
51
        poll_t ();
52
        ~poll_t ();
Martin Sustrik's avatar
Martin Sustrik committed
53

54 55
        //  "poller" concept.
        handle_t add_fd (fd_t fd_, struct i_poll_events *events_);
Martin Sustrik's avatar
Martin Sustrik committed
56 57 58 59 60
        void rm_fd (handle_t handle_);
        void set_pollin (handle_t handle_);
        void reset_pollin (handle_t handle_);
        void set_pollout (handle_t handle_);
        void reset_pollout (handle_t handle_);
61 62
        void add_timer (struct i_poll_events *events_);
        void cancel_timer (struct i_poll_events *events_);
Martin Sustrik's avatar
Martin Sustrik committed
63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
        int get_load ();
        void start ();
        void stop ();

    private:

        //  Main worker thread routine.
        static void worker_routine (void *arg_);

        //  Main event loop.
        void loop ();

        struct fd_entry_t
        {
            fd_t index;
            struct i_poll_events *events;
        };

        //  This table stores data for registered descriptors.
        std::vector <fd_entry_t> fd_table;

        //  Pollset to pass to the poll function.
        typedef std::vector <pollfd> pollset_t;
        pollset_t pollset;

        //  If true, there's at least one retired event source.
        bool retired;

        //  List of all the engines waiting for the timer event.
        typedef std::vector <struct i_poll_events*> timers_t;
        timers_t timers;

        //  If true, thread is in the process of shutting down.
        bool stopping;

        //  Handle of the physical thread doing the I/O work.
        thread_t worker;

        //  Load of the poller. Currently number of file descriptors
        //  registered with the poller.
        atomic_counter_t load;

        poll_t (const poll_t&);
        void operator = (const poll_t&);
    };

}

#endif

#endif