ZMQII-53: Memory leak when using invalid socket type

5d16070f · Martin Sustrik · a1e70f29 · 5d16070f
Commit 5d16070f authored Feb 02, 2010 by Martin Sustrik
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

dispatcher.cpp src/dispatcher.cpp +5 -1

No files found.
--- a/src/dispatcher.cpp
+++ b/src/dispatcher.cpp
@@ -134,11 +134,15 @@ zmq::socket_base_t *zmq::dispatcher_t::create_socket (int type_)
    }
    threads_sync.unlock ();

+    socket_base_t *s = thread->create_socket (type_);
+    if (!s)
+        return NULL;
+
    term_sync.lock ();
    sockets++;
    term_sync.unlock ();

-    return thread->create_socket (type_);
+    return s;
 }

 void zmq::dispatcher_t::destroy_socket ()