devpoll.cpp 5.63 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 20 21

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

#include "platform.hpp"

Martin Sustrik's avatar
Martin Sustrik committed
22
#if defined ZMQ_HAVE_SOLARIS || defined ZMQ_HAVE_HPUX
Martin Sustrik's avatar
Martin Sustrik committed
23 24 25 26 27 28 29 30 31 32

#include <sys/devpoll.h>
#include <sys/time.h>
#include <sys/resource.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <unistd.h>
#include <limits.h>
malosek's avatar
malosek committed
33
#include <algorithm>
Martin Sustrik's avatar
Martin Sustrik committed
34 35 36 37

#include "devpoll.hpp"
#include "err.hpp"
#include "config.hpp"
malosek's avatar
malosek committed
38
#include "i_poll_events.hpp"
Martin Sustrik's avatar
Martin Sustrik committed
39

40 41
zmq::devpoll_t::devpoll_t () :
    stopping (false)
Martin Sustrik's avatar
Martin Sustrik committed
42 43 44 45 46 47 48 49 50 51 52 53 54 55
{
    //  Get limit on open files
    struct rlimit rl;
    int rc = getrlimit (RLIMIT_NOFILE, &rl);
    errno_assert (rc != -1);
    fd_table.resize (rl.rlim_cur);

    for (rlim_t i = 0; i < rl.rlim_cur; i ++)
        fd_table [i].valid = false;

    devpoll_fd = open ("/dev/poll", O_RDWR);
    errno_assert (devpoll_fd != -1);
}

Martin Sustrik's avatar
Martin Sustrik committed
56
zmq::devpoll_t::~devpoll_t ()
Martin Sustrik's avatar
Martin Sustrik committed
57
{
58 59
    worker.stop ();

60 61 62
    //  Make sure there are no fds registered on shutdown.
    zmq_assert (load.get () == 0);

Martin Sustrik's avatar
Martin Sustrik committed
63 64 65
    close (devpoll_fd);
}

Martin Sustrik's avatar
Martin Sustrik committed
66
void zmq::devpoll_t::devpoll_ctl (fd_t fd_, short events_)
Martin Sustrik's avatar
Martin Sustrik committed
67 68 69
{
    struct pollfd pfd = {fd_, events_, 0};
    ssize_t rc = write (devpoll_fd, &pfd, sizeof pfd);
Martin Sustrik's avatar
Martin Sustrik committed
70
    zmq_assert (rc == sizeof pfd);
Martin Sustrik's avatar
Martin Sustrik committed
71 72
}

73 74
zmq::devpoll_t::handle_t zmq::devpoll_t::add_fd (fd_t fd_,
    i_poll_events *reactor_)
Martin Sustrik's avatar
Martin Sustrik committed
75 76 77 78 79 80 81 82 83 84 85 86 87 88
{
    assert (!fd_table [fd_].valid);

    fd_table [fd_].events = 0;
    fd_table [fd_].reactor = reactor_;
    fd_table [fd_].valid = true;
    fd_table [fd_].accepted = false;

    devpoll_ctl (fd_, 0);
    pending_list.push_back (fd_);

    //  Increase the load metric of the thread.
    load.add (1);

89
    return fd_;
Martin Sustrik's avatar
Martin Sustrik committed
90 91
}

Martin Sustrik's avatar
Martin Sustrik committed
92
void zmq::devpoll_t::rm_fd (handle_t handle_)
Martin Sustrik's avatar
Martin Sustrik committed
93
{
94
    assert (fd_table [handle_].valid);
Martin Sustrik's avatar
Martin Sustrik committed
95

96 97
    devpoll_ctl (handle_, POLLREMOVE);
    fd_table [handle_].valid = false;
Martin Sustrik's avatar
Martin Sustrik committed
98 99 100 101 102

    //  Decrease the load metric of the thread.
    load.sub (1);
}

Martin Sustrik's avatar
Martin Sustrik committed
103
void zmq::devpoll_t::set_pollin (handle_t handle_)
Martin Sustrik's avatar
Martin Sustrik committed
104
{
105 106 107
    devpoll_ctl (handle_, POLLREMOVE);
    fd_table [handle_].events |= POLLIN;
    devpoll_ctl (handle_, fd_table [handle_].events);
Martin Sustrik's avatar
Martin Sustrik committed
108 109
}

Martin Sustrik's avatar
Martin Sustrik committed
110
void zmq::devpoll_t::reset_pollin (handle_t handle_)
Martin Sustrik's avatar
Martin Sustrik committed
111
{
112 113 114
    devpoll_ctl (handle_, POLLREMOVE);
    fd_table [handle_].events &= ~((short) POLLIN);
    devpoll_ctl (handle_, fd_table [handle_].events);
Martin Sustrik's avatar
Martin Sustrik committed
115 116
}

Martin Sustrik's avatar
Martin Sustrik committed
117
void zmq::devpoll_t::set_pollout (handle_t handle_)
Martin Sustrik's avatar
Martin Sustrik committed
118
{
119 120 121
    devpoll_ctl (handle_, POLLREMOVE);
    fd_table [handle_].events |= POLLOUT;
    devpoll_ctl (handle_, fd_table [handle_].events);
Martin Sustrik's avatar
Martin Sustrik committed
122 123
}

Martin Sustrik's avatar
Martin Sustrik committed
124
void zmq::devpoll_t::reset_pollout (handle_t handle_)
Martin Sustrik's avatar
Martin Sustrik committed
125
{
126 127 128
    devpoll_ctl (handle_, POLLREMOVE);
    fd_table [handle_].events &= ~((short) POLLOUT);
    devpoll_ctl (handle_, fd_table [handle_].events);
Martin Sustrik's avatar
Martin Sustrik committed
129 130
}

Martin Sustrik's avatar
Martin Sustrik committed
131
void zmq::devpoll_t::add_timer (i_poll_events *events_)
Martin Sustrik's avatar
Martin Sustrik committed
132 133 134 135
{
     timers.push_back (events_);
}

Martin Sustrik's avatar
Martin Sustrik committed
136
void zmq::devpoll_t::cancel_timer (i_poll_events *events_)
Martin Sustrik's avatar
Martin Sustrik committed
137 138 139 140 141 142
{
    timers_t::iterator it = std::find (timers.begin (), timers.end (), events_);
    if (it != timers.end ())
        timers.erase (it);
}

Martin Sustrik's avatar
Martin Sustrik committed
143
int zmq::devpoll_t::get_load ()
Martin Sustrik's avatar
Martin Sustrik committed
144 145 146 147
{
    return load.get ();
}

Martin Sustrik's avatar
Martin Sustrik committed
148
void zmq::devpoll_t::start ()
Martin Sustrik's avatar
Martin Sustrik committed
149 150 151 152
{
    worker.start (worker_routine, this);
}

Martin Sustrik's avatar
Martin Sustrik committed
153
void zmq::devpoll_t::stop ()
Martin Sustrik's avatar
Martin Sustrik committed
154 155 156 157
{
    stopping = true;
}

malosek's avatar
malosek committed
158
void zmq::devpoll_t::loop ()
Martin Sustrik's avatar
Martin Sustrik committed
159 160 161
{
    //  According to the poll(7d) man page, we can retrieve
    //  no more then (OPEN_MAX - 1) events.
malosek's avatar
malosek committed
162
    int nfds = std::min ((int) max_io_events, OPEN_MAX - 1);
Martin Sustrik's avatar
Martin Sustrik committed
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216

    while (!stopping) {

        struct pollfd ev_buf [max_io_events];
        struct dvpoll poll_req;

        for (pending_list_t::size_type i = 0; i < pending_list.size (); i ++)
            fd_table [pending_list [i]].accepted = true;
        pending_list.clear ();

        poll_req.dp_fds = &ev_buf [0];
        poll_req.dp_nfds = nfds;
        poll_req.dp_timeout = timers.empty () ? -1 : max_timer_period;

        //  Wait for events.
        int n = ioctl (devpoll_fd, DP_POLL, &poll_req);
        if (n == -1 && errno == EINTR)
            continue;
        errno_assert (n != -1);

        //  Handle timer.
        if (!n) {

            //  Use local list of timers as timer handlers may fill new timers
            //  into the original array.
            timers_t t;
            std::swap (timers, t);

            //  Trigger all the timers.
            for (timers_t::iterator it = t.begin (); it != t.end (); it ++)
                (*it)->timer_event ();

            continue;
        }

        for (int i = 0; i < n; i ++) {

            fd_entry_t *fd_ptr = &fd_table [ev_buf [i].fd];
            if (!fd_ptr->valid || !fd_ptr->accepted)
                continue;
            if (ev_buf [i].revents & (POLLERR | POLLHUP))
                fd_ptr->reactor->in_event ();
            if (!fd_ptr->valid || !fd_ptr->accepted)
                continue;
            if (ev_buf [i].revents & POLLOUT)
                fd_ptr->reactor->out_event ();
            if (!fd_ptr->valid || !fd_ptr->accepted)
                continue;
            if (ev_buf [i].revents & POLLIN)
                fd_ptr->reactor->in_event ();
        }
    }
}

Martin Sustrik's avatar
Martin Sustrik committed
217
void zmq::devpoll_t::worker_routine (void *arg_)
Martin Sustrik's avatar
Martin Sustrik committed
218 219 220 221 222
{
    ((devpoll_t*) arg_)->loop ();
}

#endif