zmq_sendmsg and zmq_recvmsg checks for NULL message object

Signed-off-by: Martin Sustrik <sustrik@250bpm.com>

1 files changed, 2 insertions, 2 deletions

diff --git a/src/socket_base.cpp b/src/socket_base.cpp
index 302003f..7293419 100644
--- a/src/socket_base.cpp
+++ b/src/socket_base.cpp
@@ -475,7 +475,7 @@ int zmq::socket_base_t::send (msg_t *msg_, int flags_)
     }
 
     //  Check whether message passed to the function is valid.
-    if (unlikely (!msg_->check ())) {
+    if (unlikely (!msg_ || !msg_->check ())) {
         errno = EFAULT;
         return -1;
     }
@@ -541,7 +541,7 @@ int zmq::socket_base_t::recv (msg_t *msg_, int flags_)
     }
 
     //  Check whether message passed to the function is valid.
-    if (unlikely (!msg_->check ())) {
+    if (unlikely (!msg_ || !msg_->check ())) {
         errno = EFAULT;
         return -1;
     }