mirror of
https://github.com/zeromq/libzmq.git
synced 2025-03-09 07:16:04 +00:00
create signaler::recv_failable()
In real world usage, there have been reported signaler failures where the eventfd read() or socket recv() system call in signaler::recv() fails, despite having made a prior successful signaler::wait() call. this patch creates a signaler::recv_failable() method that allows unreadable eventfd / socket to return an error without asserting.
This commit is contained in:
parent
52ee724144
commit
596d6e5b1c
@ -77,14 +77,18 @@ int zmq::mailbox_t::recv (command_t *cmd_, int timeout_)
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Wait for signal from the command sender.
|
// Wait for signal from the command sender.
|
||||||
const int rc = signaler.wait (timeout_);
|
int rc = signaler.wait (timeout_);
|
||||||
if (rc == -1) {
|
if (rc == -1) {
|
||||||
errno_assert (errno == EAGAIN || errno == EINTR);
|
errno_assert (errno == EAGAIN || errno == EINTR);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Receive the signal.
|
// Receive the signal.
|
||||||
signaler.recv ();
|
rc = signaler.recv_failable ();
|
||||||
|
if (rc == -1) {
|
||||||
|
errno_assert (errno == EAGAIN);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
// Switch into active state.
|
// Switch into active state.
|
||||||
active = true;
|
active = true;
|
||||||
|
@ -289,23 +289,18 @@ void zmq::signaler_t::recv ()
|
|||||||
#if defined ZMQ_HAVE_EVENTFD
|
#if defined ZMQ_HAVE_EVENTFD
|
||||||
uint64_t dummy;
|
uint64_t dummy;
|
||||||
ssize_t sz = read (r, &dummy, sizeof (dummy));
|
ssize_t sz = read (r, &dummy, sizeof (dummy));
|
||||||
if (sz == -1) {
|
errno_assert (sz == sizeof (dummy));
|
||||||
errno_assert (errno == EAGAIN);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
errno_assert (sz == sizeof (dummy));
|
|
||||||
|
|
||||||
// If we accidentally grabbed the next signal(s) along with the current
|
// If we accidentally grabbed the next signal(s) along with the current
|
||||||
// one, return it back to the eventfd object.
|
// one, return it back to the eventfd object.
|
||||||
if (unlikely (dummy > 1)) {
|
if (unlikely (dummy > 1)) {
|
||||||
const uint64_t inc = dummy - 1;
|
const uint64_t inc = dummy - 1;
|
||||||
ssize_t sz2 = write (w, &inc, sizeof (inc));
|
ssize_t sz2 = write (w, &inc, sizeof (inc));
|
||||||
errno_assert (sz2 == sizeof (inc));
|
errno_assert (sz2 == sizeof (inc));
|
||||||
return;
|
return;
|
||||||
}
|
|
||||||
|
|
||||||
zmq_assert (dummy == 1);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
zmq_assert (dummy == 1);
|
||||||
#else
|
#else
|
||||||
unsigned char dummy;
|
unsigned char dummy;
|
||||||
#if defined ZMQ_HAVE_WINDOWS
|
#if defined ZMQ_HAVE_WINDOWS
|
||||||
@ -320,6 +315,58 @@ void zmq::signaler_t::recv ()
|
|||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int zmq::signaler_t::recv_failable ()
|
||||||
|
{
|
||||||
|
// Attempt to read a signal.
|
||||||
|
#if defined ZMQ_HAVE_EVENTFD
|
||||||
|
uint64_t dummy;
|
||||||
|
ssize_t sz = read (r, &dummy, sizeof (dummy));
|
||||||
|
if (sz == -1) {
|
||||||
|
errno_assert (errno == EAGAIN);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
errno_assert (sz == sizeof (dummy));
|
||||||
|
|
||||||
|
// If we accidentally grabbed the next signal(s) along with the current
|
||||||
|
// one, return it back to the eventfd object.
|
||||||
|
if (unlikely (dummy > 1)) {
|
||||||
|
const uint64_t inc = dummy - 1;
|
||||||
|
ssize_t sz2 = write (w, &inc, sizeof (inc));
|
||||||
|
errno_assert (sz2 == sizeof (inc));
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
zmq_assert (dummy == 1);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
unsigned char dummy;
|
||||||
|
#if defined ZMQ_HAVE_WINDOWS
|
||||||
|
int nbytes = ::recv (r, (char*) &dummy, sizeof (dummy), 0);
|
||||||
|
if (nbytes == SOCKET_ERROR) {
|
||||||
|
const int last_error = WSAGetLastError();
|
||||||
|
if (last_error == WSAEWOULDBLOCK) {
|
||||||
|
errno = EAGAIN;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
wsa_assert (last_error == WSAEWOULDBLOCK);
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
ssize_t nbytes = ::recv (r, &dummy, sizeof (dummy), 0);
|
||||||
|
if (nbytes == -1) {
|
||||||
|
if (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR) {
|
||||||
|
errno = EAGAIN;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
errno_assert (errno == EAGAIN || errno == EWOULDBLOCK || errno == EINTR);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
zmq_assert (nbytes == sizeof (dummy));
|
||||||
|
zmq_assert (dummy == 0);
|
||||||
|
#endif
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef HAVE_FORK
|
#ifdef HAVE_FORK
|
||||||
void zmq::signaler_t::forked ()
|
void zmq::signaler_t::forked ()
|
||||||
{
|
{
|
||||||
|
@ -55,6 +55,7 @@ namespace zmq
|
|||||||
void send ();
|
void send ();
|
||||||
int wait (int timeout_);
|
int wait (int timeout_);
|
||||||
void recv ();
|
void recv ();
|
||||||
|
int recv_failable ();
|
||||||
|
|
||||||
#ifdef HAVE_FORK
|
#ifdef HAVE_FORK
|
||||||
// close the file descriptors in a forked child process so that they
|
// close the file descriptors in a forked child process so that they
|
||||||
|
Loading…
x
Reference in New Issue
Block a user