Merge pull request #740 from hintjens/master

Fix for issue 574

src/signaler.cpp

@@ -80,13 +80,10 @@
 zmq::signaler_t::signaler_t ()
 {
     //  Create the socketpair for signaling.
-    int rc = make_fdpair (&r, &w);
-    errno_assert (rc == 0);
-
-    //  Set both fds to non-blocking mode.
+    if (make_fdpair (&r, &w) == 0) {
         unblock_socket (w);
         unblock_socket (r);
-
+    }
 #ifdef HAVE_FORK
     pid = getpid();
 #endif
@@ -286,19 +283,20 @@ void zmq::signaler_t::forked()
 }
 #endif
 
-
-
-
+//  Returns -1 if we could not make the socket pair successfully
 int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
 {
 #if defined ZMQ_HAVE_EVENTFD
-
-    // Create eventfd object.
     fd_t fd = eventfd (0, 0);
-    errno_assert (fd != -1);
-    *w_ = fd;
-    *r_ = fd;
+    if (fd == -1) {
+        errno_assert (errno == ENFILE || errno == EMFILE);
+        *w_ = *r_ = -1;
+        return -1;
+    }
+    else {
+        *w_ = *r_ = fd;
         return 0;
+    }
 
 #elif defined ZMQ_HAVE_WINDOWS
 #   if !defined _WIN32_WCE
@@ -328,7 +326,8 @@ int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
     HANDLE sync = CreateEvent (NULL, FALSE, TRUE, TEXT ("Global\\zmq-signaler-port-sync"));
 #   endif
     if (sync == NULL && GetLastError () == ERROR_ACCESS_DENIED)
-      sync = OpenEvent (SYNCHRONIZE | EVENT_MODIFY_STATE, FALSE, TEXT ("Global\\zmq-signaler-port-sync"));
+        sync = OpenEvent (SYNCHRONIZE | EVENT_MODIFY_STATE,
+                          FALSE, TEXT ("Global\\zmq-signaler-port-sync"));
 
     win_assert (sync != NULL);
 
@@ -391,17 +390,14 @@ int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
 
     //  Save errno if connection fails
     int conn_errno = 0;
-    if (rc == SOCKET_ERROR) {
+    if (rc == SOCKET_ERROR)
         conn_errno = WSAGetLastError ();
-    } else {
+    else {
         //  Accept connection from writer.
         *r_ = accept (listener, NULL, NULL);
-
-        if (*r_ == INVALID_SOCKET) {
+        if (*r_ == INVALID_SOCKET)
             conn_errno = WSAGetLastError ();
     }
-    }
-
     //  We don't need the listening socket anymore. Close it.
     rc = closesocket (listener);
     wsa_assert (rc != SOCKET_ERROR);
@@ -421,7 +417,8 @@ int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
         win_assert (brc);
 #   endif
         return 0;
-    } else {
+    }
+    else {
         //  Cleanup writer if connection failed
         rc = closesocket (*w_);
         wsa_assert (rc != SOCKET_ERROR);
@@ -435,7 +432,6 @@ int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
         //  Unfortunately, it uses errno_assert() which gives "Unknown error"
         //  We might as well assert here and print the actual error message
         wsa_assert_no (conn_errno);
-
         return -1;
     }
 
@@ -493,15 +489,20 @@ int zmq::signaler_t::make_fdpair (fd_t *r_, fd_t *w_)
 
     return 0;
 
-#else // All other implementations support socketpair()
-
+#else
+    // All other implementations support socketpair()
     int sv [2];
     int rc = socketpair (AF_UNIX, SOCK_STREAM, 0, sv);
-    errno_assert (rc == 0);
+    if (rc == -1) {
+        errno_assert (errno == ENFILE || errno == EMFILE);
+        sv [0] = sv [1] = -1;
+        return -1;
+    }
+    else {
         *w_ = sv [0];
         *r_ = sv [1];
         return 0;
-
+    }
 #endif
 }
 

src/signaler.hpp

@@ -58,7 +58,8 @@ namespace zmq
         //  to pass the signals.
         static int make_fdpair (fd_t *r_, fd_t *w_);
 
-        //  Underlying write & read file descriptor.
+        //  Underlying write & read file descriptor
+        //  Will be -1 if we exceeded number of available handles
         fd_t w;
         fd_t r;
 
@@ -74,7 +75,6 @@ namespace zmq
         void close_internal();
 #endif
     };
-
 }
 
 #endif

src/socket_base.cpp

@@ -81,7 +81,6 @@ zmq::socket_base_t *zmq::socket_base_t::create (int type_, class ctx_t *parent_,
 {
     socket_base_t *s = NULL;
     switch (type_) {
-
         case ZMQ_PAIR:
             s = new (std::nothrow) pair_t (parent_, tid_, sid_);
             break;
@@ -122,6 +121,9 @@ zmq::socket_base_t *zmq::socket_base_t::create (int type_, class ctx_t *parent_,
             errno = EINVAL;
             return NULL;
     }
+    if (s->mailbox.get_fd () == -1)
+        return NULL;
+
     alloc_assert (s);
     return s;
 }