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_lat.c  | 51 ++++++++++++++++++++++++++++++++++----------
 perf/c/local_thr.c  | 61 +++++++++++++++++++++++++++++++++++++++++------------
 perf/c/remote_lat.c | 51 ++++++++++++++++++++++++++++++++++----------
 perf/c/remote_thr.c | 41 +++++++++++++++++++++++++++--------
 4 files changed, 160 insertions(+), 44 deletions(-)

(limited to 'perf')

diff --git a/perf/c/local_lat.c b/perf/c/local_lat.c
index c53a684..bd71f82 100644
--- a/perf/c/local_lat.c
+++ b/perf/c/local_lat.c
@@ -20,7 +20,6 @@
 #include <zmq.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <assert.h>
 
 int main (int argc, char *argv [])
 {
@@ -43,35 +42,65 @@ int main (int argc, char *argv [])
     roundtrip_count = atoi (argv [3]);
 
     ctx = zmq_init (1, 1, 0);
-    assert (ctx);
+    if (!ctx) {
+        printf ("error in zmq_init: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     s = zmq_socket (ctx, ZMQ_REP);
-    assert (s);
+    if (!s) {
+        printf ("error in zmq_socket: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     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;
+    }
 
     for (i = 0; i != roundtrip_count; 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;
+        }
         rc = zmq_send (s, &msg, 0);
-        assert (rc == 0);
+        if (rc != 0) {
+            printf ("error in zmq_send: %s\n", zmq_strerror (errno));
+            return -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;
+    }
 
     zmq_sleep (1);
 
     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;
 }
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;
 }
diff --git a/perf/c/remote_lat.c b/perf/c/remote_lat.c
index 0c0a9f1..901afd5 100644
--- a/perf/c/remote_lat.c
+++ b/perf/c/remote_lat.c
@@ -21,7 +21,6 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
-#include <assert.h>
 
 int main (int argc, char *argv [])
 {
@@ -47,32 +46,56 @@ int main (int argc, char *argv [])
     roundtrip_count = atoi (argv [3]);
 
     ctx = zmq_init (1, 1, 0);
-    assert (ctx);
+    if (!ctx) {
+        printf ("error in zmq_init: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     s = zmq_socket (ctx, ZMQ_REQ);
-    assert (s);
+    if (!s) {
+        printf ("error in zmq_socket: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_connect (s, connect_to);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_connect: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     rc = zmq_msg_init_size (&msg, message_size);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_msg_init_size: %s\n", zmq_strerror (errno));
+        return -1;
+    }
     memset (zmq_msg_data (&msg), 0, message_size);
 
     watch = zmq_stopwatch_start ();
 
     for (i = 0; i != roundtrip_count; i++) {
         rc = zmq_send (s, &msg, 0);
-        assert (rc == 0);
+        if (rc != 0) {
+            printf ("error in zmq_send: %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;
+        }
     }
 
     elapsed = zmq_stopwatch_stop (watch);
 
     rc = zmq_msg_close (&msg);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_msg_close: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     latency = (double) elapsed / (roundtrip_count * 2);
 
@@ -81,10 +104,16 @@ int main (int argc, char *argv [])
     printf ("average latency: %.3f [us]\n", (double) latency);
 
     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;
 }
diff --git a/perf/c/remote_thr.c b/perf/c/remote_thr.c
index ab4f060..43956e6 100644
--- a/perf/c/remote_thr.c
+++ b/perf/c/remote_thr.c
@@ -20,7 +20,6 @@
 #include <zmq.h>
 #include <stdio.h>
 #include <stdlib.h>
-#include <assert.h>
 
 int main (int argc, char *argv [])
 {
@@ -43,33 +42,57 @@ 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_recv: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     s = zmq_socket (ctx, ZMQ_PUB);
-    assert (s);
+    if (!s) {
+        printf ("error in zmq_socket: %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_connect (s, connect_to);
-    assert (rc == 0);
+    if (rc != 0) {
+        printf ("error in zmq_connect: %s\n", zmq_strerror (errno));
+        return -1;
+    }
 
     for (i = 0; i != message_count; i++) {
         rc = zmq_msg_init_size (&msg, message_size);
-        assert (rc == 0);
+        if (rc != 0) {
+            printf ("error in zmq_msg_init_size: %s\n", zmq_strerror (errno));
+            return -1;
+        }
         rc = zmq_send (s, &msg, 0);
-        assert (rc == 0);
+        if (rc != 0) {
+            printf ("error in zmq_send: %s\n", zmq_strerror (errno));
+            return -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;
+        }
     }
 
     zmq_sleep (10);
 
     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