Merge pull request #1482 from brian-peloton/master

Fix memory leak when creating a socket fails.

src/signaler.cpp

@@ -133,12 +133,16 @@ zmq::signaler_t::signaler_t ()
 #endif
 }
 
+// This might get run after some part of construction failed, leaving one or
+// both of r and w retired_fd.
 zmq::signaler_t::~signaler_t ()
 {
 #if defined ZMQ_HAVE_EVENTFD
+    if (r == retired_fd) return;
     int rc = close_wait_ms (r);
     errno_assert (rc == 0);
 #elif defined ZMQ_HAVE_WINDOWS
+    if (w != retired_fd) {
         const struct linger so_linger = { 1, 0 };
         int rc = setsockopt (w, SOL_SOCKET, SO_LINGER,
             (const char *) &so_linger, sizeof so_linger);
@@ -147,14 +151,19 @@ zmq::signaler_t::~signaler_t ()
             wsa_assert (rc != SOCKET_ERROR);
             rc = closesocket (w);
             wsa_assert (rc != SOCKET_ERROR);
+            if (r == retired_fd) return;
             rc = closesocket (r);
             wsa_assert (rc != SOCKET_ERROR);
         }
 #else
+    if (w != retired_fd) {
         int rc = close_wait_ms (w);
         errno_assert (rc == 0);
+    }
+    if (r != retired_fd) {
         rc = close_wait_ms (r);
         errno_assert (rc == 0);
+    }
 #endif
 }
 

src/socket_base.cpp

@@ -154,8 +154,11 @@ zmq::socket_base_t *zmq::socket_base_t::create (int type_, class ctx_t *parent_,
 
     mailbox_t *mailbox = dynamic_cast<mailbox_t*> (s->mailbox);
 
-    if (mailbox != NULL && mailbox->get_fd () == retired_fd)
+    if (mailbox != NULL && mailbox->get_fd () == retired_fd) {
+        s->destroyed = true;
+        delete s;
         return NULL;
+    }
 
     return s;
 }