From 7094edd6baf0d69c3879da7f16c7222388952931 Mon Sep 17 00:00:00 2001
From: Martin Sustrik <sustrik@fastmq.commkdir>
Date: Mon, 18 Jan 2010 15:57:33 +0100
Subject: error handling in C perf tests improved

---
 perf/c/local_thr.c | 61 ++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 48 insertions(+), 13 deletions(-)

(limited to 'perf/c/local_thr.c')

diff --git a/perf/c/local_thr.c b/perf/c/local_thr.c
index f39d63f..cfebd2d 100644
--- a/perf/c/local_thr.c
+++ b/perf/c/local_thr.c
@@ -20,7 +20,6 @@
 #include <zmq.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <assert.h>
 
 int main (int argc, char *argv [])
 {
@@ -46,33 +45,60 @@ int main (int argc, char *argv [])
     message_count = atoi (argv [3]);
 
     ctx = zmq_init (1, 1, 0);
-    assert (ctx);
+    if (!ctx) {
+        printf ("error in zmq_send: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     s = zmq_socket (ctx, ZMQ_SUB);
-    assert (s);
+    if (!s) {
+        printf ("error in zmq_socket: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_setsockopt (s, ZMQ_SUBSCRIBE , "", 0);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_setsockopt: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     //  Add your socket options here.
     //  For example ZMQ_RATE, ZMQ_RECOVERY_IVL and ZMQ_MCAST_LOOP for PGM.
 
     rc = zmq_bind (s, bind_to);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_bind: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_msg_init (&msg);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_msg_init: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_recv (s, &msg, 0);
-    assert (rc == 0);
-    assert (zmq_msg_size (&msg) == message_size);
+    if (rc != 0) {
+        printf ("error in zmq_recv: %s\n", zmq_strerror (errno));
+        return -1;
+    }
+    if (zmq_msg_size (&msg) != message_size) {
+        printf ("message of incorrect size received\n");
+        return -1;
+    }
 
     watch = zmq_stopwatch_start ();
 
     for (i = 0; i != message_count - 1; i++) {
         rc = zmq_recv (s, &msg, 0);
-        assert (rc == 0);
-        assert (zmq_msg_size (&msg) == message_size);
+        if (rc != 0) {
+            printf ("error in zmq_recv: %s\n", zmq_strerror (errno));
+            return -1;
+        }
+        if (zmq_msg_size (&msg) != message_size) {
+            printf ("message of incorrect size received\n");
+            return -1;
+        }
     }
 
     elapsed = zmq_stopwatch_stop (watch);
@@ -80,7 +106,10 @@ int main (int argc, char *argv [])
         elapsed = 1;
 
     rc = zmq_msg_close (&msg);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_msg_close: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     throughput = (unsigned long)
         ((double) message_count / (double) elapsed * 1000000);
@@ -92,10 +121,16 @@ int main (int argc, char *argv [])
     printf ("mean throughput: %.3f [Mb/s]\n", (double) megabits);
 
     rc = zmq_close (s);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_close: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_term (ctx);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_term: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     return 0;
 }
-- 
cgit v1.2.3