improved error handling

1 files changed, 54 insertions, 73 deletions

diff --git a/include/zmq.hpp b/include/zmq.hpp
index 8397464..ffdea7b 100644
--- a/include/zmq.hpp
+++ b/include/zmq.hpp
@@ -24,6 +24,7 @@
 
 #include <assert.h>
 #include <errno.h>
+#include <string.h>
 #include <exception>
 
 namespace zmq
@@ -38,36 +39,21 @@ namespace zmq
         message_delimiter = 1 << ZMQ_DELIMITER
     };
 
-    class no_memory : public std::exception
+    //  The class masquerades POSIX-style errno error as a C++ exception.
+    class error_t : public std::exception
     {
-        virtual const char *what ()
-        {
-            return "Out of memory";
-        }
-    };
+    public:
 
-    class invalid_argument : public std::exception
-    {
-        virtual const char *what ()
-        {
-            return "Invalid argument";
-        }
-    };
+        error_t () : errnum (errno) {}
 
-    class too_many_threads : public std::exception
-    {
-        virtual const char *what ()
+        virtual const char *what () const throw ()
         {
-            return "Too many threads";
+            return strerror (errnum);
         }
-    };
 
-    class address_in_use : public std::exception
-    {
-        virtual const char *what ()
-        {
-            return "Address in use";
-        }
+    private:
+
+        int errnum;
     };
 
     //  A message. Caution: Don't change the body of the message once you've
@@ -84,10 +70,8 @@ namespace zmq
         inline message_t (size_t size_ = 0)
         {
             int rc = zmq_msg_init_size (this, size_);
-            if (rc == -1) {
-                assert (errno == ENOMEM);
-                throw no_memory ();
-            }
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Creates message from the supplied buffer. 0MQ takes care of
@@ -98,14 +82,16 @@ namespace zmq
             free_fn *ffn_)
         {
             int rc = zmq_msg_init_data (this, data_, size_, ffn_);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Destroys the message.
         inline ~message_t ()
         {
             int rc = zmq_msg_close (this);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Destroys old content of the message and allocates buffer for the
@@ -114,12 +100,11 @@ namespace zmq
         inline void rebuild (size_t size_)
         {
             int rc = zmq_msg_close (this);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
             rc = zmq_msg_init_size (this, size_);
-            if (rc == -1) {
-                assert (errno == ENOMEM);
-                throw no_memory ();
-            }
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Same as above, however, the message is rebuilt from the supplied
@@ -128,9 +113,11 @@ namespace zmq
         inline void rebuild (void *data_, size_t size_, free_fn *ffn_)
         {
             int rc = zmq_msg_close (this);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
             rc = zmq_msg_init_data (this, data_, size_, ffn_);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Moves the message content from one message to the another. If the
@@ -140,7 +127,8 @@ namespace zmq
         inline void move_to (message_t *msg_)
         {
             int rc = zmq_msg_move (this, (zmq_msg_t*) msg_);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Copies the message content from one message to the another. If the
@@ -149,7 +137,8 @@ namespace zmq
         inline void copy_to (message_t *msg_)
         {
             int rc = zmq_msg_copy (this, (zmq_msg_t*) msg_);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         //  Returns message type.
@@ -187,10 +176,8 @@ namespace zmq
         inline context_t (int app_threads_, int io_threads_)
         {
             ptr = zmq_init (app_threads_, io_threads_);
-            if (ptr == NULL) {
-                assert (errno == EINVAL);
-                throw invalid_argument ();
-            }
+            if (ptr == NULL)
+                throw error_t ();
         }
 
         inline ~context_t ()
@@ -215,70 +202,64 @@ namespace zmq
         inline socket_t (context_t &context_, int type_ = 0)
         {
             ptr = zmq_socket (context_.ptr, type_);
-            if (ptr == NULL) {
-                assert (errno == EMFILE || errno == EINVAL);
-                if (errno == EMFILE)
-                    throw too_many_threads ();
-                else
-                    throw invalid_argument ();
-            }
+            if (ptr == NULL)
+                throw error_t ();
         }
 
         inline ~socket_t ()
         {
             int rc = zmq_close (ptr);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         inline void setsockopt (int option_, const void *optval_,
             size_t optvallen_)
         {
             int rc = zmq_setsockopt (ptr, option_, optval_, optvallen_);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
         inline void bind (const char *addr_)
         {
             int rc = zmq_bind (ptr, addr_);
-            if (rc == -1) {
-                assert (errno == EINVAL || errno == EADDRINUSE);
-                if (errno == EINVAL)
-                    throw invalid_argument ();
-                else
-                    throw address_in_use ();
-            }
+            if (rc != 0)
+                throw error_t ();
         }
 
         inline void connect (const char *addr_)
         {
             int rc = zmq_connect (ptr, addr_);
-            if (rc == -1) {
-                assert (errno == EINVAL || errno == EADDRINUSE);
-                if (errno == EINVAL)
-                    throw invalid_argument ();
-                else
-                    throw address_in_use ();
-            }
+            if (rc != 0)
+                throw error_t ();
         }
 
-        inline int send (message_t &msg_, int flags_ = 0)
+        inline bool send (message_t &msg_, int flags_ = 0)
         {
             int rc = zmq_send (ptr, &msg_, flags_);
-            assert (rc == 0 || (rc == -1 && errno == EAGAIN));
-            return rc;
+            if (rc == 0)
+                return true;
+            if (rc == -1 && errno == EAGAIN)
+                return false;
+            throw error_t ();
         }
 
         inline void flush ()
         {
             int rc = zmq_flush (ptr);
-            assert (rc == 0);
+            if (rc != 0)
+                throw error_t ();
         }
 
-        inline int recv (message_t *msg_, int flags_ = 0)
+        inline bool recv (message_t *msg_, int flags_ = 0)
         {
             int rc = zmq_recv (ptr, msg_, flags_);
-            assert (rc == 0 || (rc == -1 && errno == EAGAIN));
-            return rc;
+            if (rc == 0)
+                return true;
+            if (rc == -1 && errno == EAGAIN)
+                return false;
+            throw error_t ();
         }
 
     private: