Skip to content

Commit

Permalink
Problem: valid_pollset_bytes not reused in socket_poller
Browse files Browse the repository at this point in the history
Solution: use it
  • Loading branch information
sigiesec committed Jun 1, 2018
1 parent a8d62d1 commit 3d39bb9
Showing 1 changed file with 9 additions and 23 deletions.
32 changes: 9 additions & 23 deletions src/socket_poller.cpp
Expand Up @@ -30,6 +30,7 @@
#include "precompiled.hpp"
#include "socket_poller.hpp"
#include "err.hpp"
#include "polling_util.hpp"

#include <limits.h>

Expand Down Expand Up @@ -637,34 +638,18 @@ int zmq::socket_poller_t::wait (zmq::socket_poller_t::event_t *events_,

// Wait for events. Ignore interrupts if there's infinite timeout.
while (true) {
memcpy (&inset, &_pollset_in, valid_pollset_bytes (_pollset_in));
memcpy (&outset, &_pollset_out, valid_pollset_bytes (_pollset_out));
memcpy (&errset, &_pollset_err, valid_pollset_bytes (_pollset_err));
const int rc = select (static_cast<int> (_max_fd + 1), &inset,
&outset, &errset, ptimeout);
#if defined ZMQ_HAVE_WINDOWS
// On Windows we don't need to copy the whole fd_set.
// SOCKETS are continuous from the beginning of fd_array in fd_set.
// We just need to copy fd_count elements of fd_array.
// We gain huge memcpy() improvement if number of used SOCKETs is much lower than FD_SETSIZE.
memcpy (&inset, &_pollset_in,
reinterpret_cast<char *> (_pollset_in.fd_array
+ _pollset_in.fd_count)
- reinterpret_cast<char *> (&_pollset_in));
memcpy (&outset, &_pollset_out,
reinterpret_cast<char *> (_pollset_out.fd_array
+ _pollset_out.fd_count)
- reinterpret_cast<char *> (&_pollset_out));
memcpy (&errset, &_pollset_err,
reinterpret_cast<char *> (_pollset_err.fd_array
+ _pollset_err.fd_count)
- reinterpret_cast<char *> (&_pollset_err));
int rc = select (0, &inset, &outset, &errset, ptimeout);
if (unlikely (rc == SOCKET_ERROR)) {
errno = zmq::wsa_error_to_errno (WSAGetLastError ());
errno = wsa_error_to_errno (WSAGetLastError ());
wsa_assert (errno == ENOTSOCK);
return -1;
}
#else
memcpy (&inset, &_pollset_in, sizeof (fd_set));
memcpy (&outset, &_pollset_out, sizeof (fd_set));
memcpy (&errset, &_pollset_err, sizeof (fd_set));
int rc = select (_max_fd + 1, &inset, &outset, &errset, ptimeout);
if (unlikely (rc == -1)) {
errno_assert (errno == EINTR || errno == EBADF);
return -1;
Expand All @@ -677,7 +662,8 @@ int zmq::socket_poller_t::wait (zmq::socket_poller_t::event_t *events_,
_signaler->recv ();

// Check for the events.
int found = check_events (events_, n_events_, inset, outset, errset);
const int found =
check_events (events_, n_events_, inset, outset, errset);
if (found) {
if (found > 0)
zero_trail_events (events_, n_events_, found);
Expand Down

0 comments on commit 3d39bb9

Please sign in to comment.