netrate/accept_connect: Switch to err(3)
authorSepherosa Ziehau <sephe@dragonflybsd.org>
Tue, 9 Sep 2014 14:40:39 +0000 (22:40 +0800)
committerSepherosa Ziehau <sephe@dragonflybsd.org>
Wed, 10 Sep 2014 06:44:11 +0000 (14:44 +0800)
tools/tools/netrate/accept_connect/accept_server/accept_server.c
tools/tools/netrate/accept_connect/connect_client/connect_client.c
tools/tools/netrate/accept_connect/kq_accept_server/kq_accept_server.c
tools/tools/netrate/accept_connect/kq_connect_client/kq_connect_client.c

index b4dd446..4d73260 100644 (file)
@@ -5,7 +5,7 @@
 #include <arpa/inet.h>
 #include <netinet/in.h>
 
-#include <errno.h>
+#include <err.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -26,37 +26,25 @@ create_socket(const struct sockaddr_in *in, int reuseport)
        int serv_s, on;
 
        serv_s = socket(AF_INET, SOCK_STREAM, 0);
-       if (serv_s < 0) {
-               fprintf(stderr, "socket failed: %d\n", errno);
-               exit(1);
-       }
+       if (serv_s < 0)
+               err(1, "socket failed");
 
        on = 1;
        if (!reuseport) {
                if (setsockopt(serv_s, SOL_SOCKET, SO_REUSEADDR,
-                   &on, sizeof(on)) < 0) {
-                       fprintf(stderr, "setsockopt(REUSEADDR) failed: %d\n",
-                           errno);
-                       exit(1);
-               }
+                   &on, sizeof(on)) < 0)
+                       err(1, "setsockopt(REUSEADDR) failed");
        } else {
                if (setsockopt(serv_s, SOL_SOCKET, SO_REUSEPORT,
-                   &on, sizeof(on)) < 0) {
-                       fprintf(stderr, "setsockopt(REUSEPORT) failed: %d\n",
-                           errno);
-                       exit(1);
-               }
+                   &on, sizeof(on)) < 0)
+                       err(1, "setsockopt(REUSEPORT) failed");
        }
 
-       if (bind(serv_s, (const struct sockaddr *)in, sizeof(*in)) < 0) {
-               fprintf(stderr, "bind failed: %d\n", errno);
-               exit(1);
-       }
+       if (bind(serv_s, (const struct sockaddr *)in, sizeof(*in)) < 0)
+               err(1, "bind failed");
 
-       if (listen(serv_s, -1) < 0) {
-               fprintf(stderr, "listen failed: %d\n", errno);
-               exit(1);
-       }
+       if (listen(serv_s, -1) < 0)
+               err(1, "listen failed");
        return serv_s;
 }
 
@@ -68,10 +56,8 @@ main(int argc, char *argv[])
        size_t prm_len;
 
        prm_len = sizeof(ninst);
-       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0) {
-               fprintf(stderr, "sysctl hw.ncpu failed: %d\n", errno);
-               exit(2);
-       }
+       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0)
+               err(2, "sysctl hw.ncpu failed");
 
        memset(&in, 0, sizeof(in));
        in.sin_family = AF_INET;
@@ -113,7 +99,7 @@ main(int argc, char *argv[])
                        mainloop(serv_s, &in);
                        exit(0);
                } else if (pid < 0) {
-                       fprintf(stderr, "fork failed: %d\n", errno);
+                       err(1, "fork failed");
                }
        }
 
index 9209ae4..c596070 100644 (file)
@@ -8,7 +8,7 @@
 #include <arpa/inet.h>
 #include <netinet/in.h>
 
-#include <errno.h>
+#include <err.h>
 #include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
@@ -37,10 +37,8 @@ main(int argc, char *argv[])
        size_t prm_len;
 
        prm_len = sizeof(ninst);
-       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0) {
-               fprintf(stderr, "sysctl hw.ncpu failed: %d\n", errno);
-               exit(2);
-       }
+       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0)
+               err(2, "sysctl hw.ncpu failed");
 
        memset(&in, 0, sizeof(in));
        in.sin_family = AF_INET;
@@ -80,10 +78,8 @@ main(int argc, char *argv[])
 
        result = mmap(NULL, ninst * sizeof(u_long), PROT_READ | PROT_WRITE,
            MAP_ANON | MAP_SHARED, -1, 0);
-       if (result == MAP_FAILED) {
-               fprintf(stderr, "mmap failed: %d\n", errno);
-               exit(1);
-       }
+       if (result == MAP_FAILED)
+               err(1, "mmap failed");
        memset(result, 0, ninst * sizeof(u_long));
 
        for (i = 0; i < ninst; ++i) {
@@ -94,7 +90,7 @@ main(int argc, char *argv[])
                        mainloop(&in, dur, &result[i]);
                        exit(0);
                } else if (pid < 0) {
-                       fprintf(stderr, "fork failed: %d\n", errno);
+                       err(1, "fork failed");
                }
        }
 
@@ -102,10 +98,8 @@ main(int argc, char *argv[])
                pid_t pid;
 
                pid = waitpid(-1, NULL, 0);
-               if (pid < 0) {
-                       fprintf(stderr, "waitpid failed: %d\n", errno);
-                       exit(1);
-               }
+               if (pid < 0)
+                       err(1, "waitpid failed");
        }
 
        sum = 0;
@@ -122,27 +116,22 @@ signal_handler(int signum __unused)
        global_stopped = 1;
 }
 
-
 static void
 mainloop(const struct sockaddr_in *in, long dur, u_long *res)
 {
        struct itimerval it;
        u_long count = 0;
 
-       if (signal(SIGALRM, signal_handler) == SIG_ERR) {
-               fprintf(stderr, "signal failed: %d\n", errno);
-               return;
-       }
+       if (signal(SIGALRM, signal_handler) == SIG_ERR)
+               err(1, "signal failed");
 
        it.it_interval.tv_sec = 0;
        it.it_interval.tv_usec = 0;
        it.it_value.tv_sec = dur;
        it.it_value.tv_usec = 0;
 
-       if (setitimer(ITIMER_REAL, &it, NULL) < 0) {
-               fprintf(stderr, "setitimer failed: %d\n", errno);
-               return;
-       }
+       if (setitimer(ITIMER_REAL, &it, NULL) < 0)
+               err(1, "setitimer failed");
 
        while (!global_stopped) {
                int s;
index d86d834..e4d17d7 100644 (file)
@@ -8,7 +8,7 @@
 #include <arpa/inet.h>
 #include <netinet/in.h>
 
-#include <errno.h>
+#include <err.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -31,43 +31,29 @@ create_socket(const struct sockaddr_in *in, int reuseport)
        int serv_s, on;
 
        serv_s = socket(AF_INET, SOCK_STREAM, 0);
-       if (serv_s < 0) {
-               fprintf(stderr, "socket failed: %d\n", errno);
-               exit(1);
-       }
+       if (serv_s < 0)
+               err(1, "socket failed");
 
        on = 1;
        if (!reuseport) {
                if (setsockopt(serv_s, SOL_SOCKET, SO_REUSEADDR,
-                   &on, sizeof(on)) < 0) {
-                       fprintf(stderr, "setsockopt(REUSEADDR) failed: %d\n",
-                           errno);
-                       exit(1);
-               }
+                   &on, sizeof(on)) < 0)
+                       err(1, "setsockopt(REUSEADDR) failed");
        } else {
                if (setsockopt(serv_s, SOL_SOCKET, SO_REUSEPORT,
-                   &on, sizeof(on)) < 0) {
-                       fprintf(stderr, "setsockopt(REUSEPORT) failed: %d\n",
-                           errno);
-                       exit(1);
-               }
+                   &on, sizeof(on)) < 0)
+                       err(1, "setsockopt(REUSEPORT) failed");
        }
 
        on = 1;
-       if (ioctl(serv_s, FIONBIO, &on, sizeof(on)) < 0) {
-               fprintf(stderr, "ioctl(FIONBIO) failed: %d\n", errno);
-               exit(1);
-       }
+       if (ioctl(serv_s, FIONBIO, &on, sizeof(on)) < 0)
+               err(1, "ioctl(FIONBIO) failed");
 
-       if (bind(serv_s, (const struct sockaddr *)in, sizeof(*in)) < 0) {
-               fprintf(stderr, "bind failed: %d\n", errno);
-               exit(1);
-       }
+       if (bind(serv_s, (const struct sockaddr *)in, sizeof(*in)) < 0)
+               err(1, "bind failed");
 
-       if (listen(serv_s, -1) < 0) {
-               fprintf(stderr, "listen failed: %d\n", errno);
-               exit(1);
-       }
+       if (listen(serv_s, -1) < 0)
+               err(1, "listen failed");
        return serv_s;
 }
 
@@ -79,10 +65,8 @@ main(int argc, char *argv[])
        size_t prm_len;
 
        prm_len = sizeof(ninst);
-       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0) {
-               fprintf(stderr, "sysctl hw.ncpu failed: %d\n", errno);
-               exit(2);
-       }
+       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0)
+               err(2, "sysctl hw.ncpu failed");
 
        memset(&in, 0, sizeof(in));
        in.sin_family = AF_INET;
@@ -124,7 +108,7 @@ main(int argc, char *argv[])
                        mainloop(serv_s, &in);
                        exit(0);
                } else if (pid < 0) {
-                       fprintf(stderr, "fork failed: %d\n", errno);
+                       err(1, "fork failed");
                }
        }
 
@@ -142,10 +126,8 @@ mainloop(int serv_s, const struct sockaddr_in *in)
                serv_s = create_socket(in, 1);
 
        kq = kqueue();
-       if (kq < 0) {
-               fprintf(stderr, "kqueue failed: %d\n", errno);
-               exit(1);
-       }
+       if (kq < 0)
+               err(1, "kqueue failed");
 
        EV_SET(&change_evt0[0], serv_s, EVFILT_READ, EV_ADD, 0, 0, NULL);
        nchange = 1;
@@ -159,10 +141,8 @@ mainloop(int serv_s, const struct sockaddr_in *in)
                        change_evt = change_evt0;
 
                nevt = kevent(kq, change_evt, nchange, evt, EVENT_MAX, NULL);
-               if (nevt < 0) {
-                       fprintf(stderr, "kevent failed: %d\n", errno);
-                       exit(1);
-               }
+               if (nevt < 0)
+                       err(1, "kevent failed");
                nchange = 0;
 
                for (i = 0; i < nevt; ++i) {
index 8192986..5cf2eb5 100644 (file)
@@ -10,6 +10,7 @@
 #include <arpa/inet.h>
 #include <netinet/in.h>
 
+#include <err.h>
 #include <errno.h>
 #include <signal.h>
 #include <stdio.h>
@@ -39,10 +40,8 @@ main(int argc, char *argv[])
        size_t prm_len;
 
        prm_len = sizeof(ninst);
-       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0) {
-               fprintf(stderr, "sysctl hw.ncpu failed: %d\n", errno);
-               exit(2);
-       }
+       if (sysctlbyname("hw.ncpu", &ninst, &prm_len, NULL, 0) != 0)
+               err(2, "sysctl hw.ncpu failed");
 
        nconn = 8;
        dur = 10;
@@ -51,10 +50,8 @@ main(int argc, char *argv[])
 
        in_max = 8;
        in = calloc(in_max, sizeof(struct sockaddr_in));
-       if (in == NULL) {
-               fprintf(stderr, "calloc failed\n");
-               exit(1);
-       }
+       if (in == NULL)
+               err(1, "calloc failed");
        in_cnt = 0;
 
        while ((opt = getopt(argc, argv, "4:p:i:l:c:u")) != -1) {
@@ -66,10 +63,8 @@ main(int argc, char *argv[])
 
                                in_max <<= 1;
                                in = calloc(in_max, sizeof(struct sockaddr_in));
-                               if (in == NULL) {
-                                       fprintf(stderr, "calloc failed\n");
-                                       exit(1);
-                               }
+                               if (in == NULL)
+                                       err(1, "calloc failed");
 
                                memcpy(in, old_in,
                                    old_in_max * sizeof(struct sockaddr_in));
@@ -121,10 +116,8 @@ main(int argc, char *argv[])
 
        result = mmap(NULL, ninst * sizeof(u_long), PROT_READ | PROT_WRITE,
            MAP_ANON | MAP_SHARED, -1, 0);
-       if (result == MAP_FAILED) {
-               fprintf(stderr, "mmap failed: %d\n", errno);
-               exit(1);
-       }
+       if (result == MAP_FAILED)
+               err(1, "mmap failed");
        memset(result, 0, ninst * sizeof(u_long));
 
        for (i = 0; i < ninst; ++i) {
@@ -135,7 +128,7 @@ main(int argc, char *argv[])
                        mainloop(in, in_cnt, nconn, dur, &result[i], do_udp);
                        exit(0);
                } else if (pid < 0) {
-                       fprintf(stderr, "fork failed: %d\n", errno);
+                       err(1, "fork failed");
                }
        }
 
@@ -143,10 +136,8 @@ main(int argc, char *argv[])
                pid_t pid;
 
                pid = waitpid(-1, NULL, 0);
-               if (pid < 0) {
-                       fprintf(stderr, "waitpid failed: %d\n", errno);
-                       exit(1);
-               }
+               if (pid < 0)
+                       err(1, "waitpid failed");
        }
 
        sum = 0;
@@ -165,12 +156,12 @@ udp_send(const struct sockaddr_in *in)
 
        s = socket(AF_INET, SOCK_DGRAM, 0);
        if (s < 0) {
-               fprintf(stderr, "socket DGRAM failed: %d\n", errno);
+               warn("socket DGRAM failed");
                return;
        }
 
        if (connect(s, (const struct sockaddr *)in, sizeof(*in)) < 0) {
-               fprintf(stderr, "connect DGRAM failed: %d\n", errno);
+               warn("connect DGRAM failed");
                goto done;
        }
 
@@ -190,24 +181,18 @@ mainloop(const struct sockaddr_in *in, int in_cnt, int nconn_max,
        int nblock = 1;
 
        kq = kqueue();
-       if (kq < 0) {
-               fprintf(stderr, "kqueue failed: %d\n", errno);
-               return;
-       }
+       if (kq < 0)
+               err(1, "kqueue failed");
 
        nevt_max = nconn_max + 1; /* timer */
 
        evt_change0 = malloc(nevt_max * sizeof(struct kevent));
-       if (evt_change0 == NULL) {
-               fprintf(stderr, "malloc evt_change failed\n");
-               return;
-       }
+       if (evt_change0 == NULL)
+               err(1, "malloc evt_change failed");
 
        evt = malloc(nevt_max * sizeof(struct kevent));
-       if (evt == NULL) {
-               fprintf(stderr, "malloc evt failed\n");
-               return;
-       }
+       if (evt == NULL)
+               err(1, "malloc evt failed");
 
        EV_SET(&evt_change0[0], 0, EVFILT_TIMER, EV_ADD | EV_ONESHOT, 0,
            dur * 1000L, NULL);
@@ -228,15 +213,11 @@ mainloop(const struct sockaddr_in *in, int in_cnt, int nconn_max,
                                udp_send(tmp);
 
                        s = socket(AF_INET, SOCK_STREAM, 0);
-                       if (s < 0) {
-                               fprintf(stderr, "socket failed: %d\n", errno);
-                               return;
-                       }
+                       if (s < 0)
+                               err(1, "socket failed");
 
-                       if (ioctl(s, FIONBIO, &nblock, sizeof(nblock)) < 0) {
-                               fprintf(stderr, "ioctl failed: %d\n", errno);
-                               return;
-                       }
+                       if (ioctl(s, FIONBIO, &nblock, sizeof(nblock)) < 0)
+                               err(1, "ioctl failed");
 
                        n = connect(s, (const struct sockaddr *)tmp,
                            sizeof(*tmp));
@@ -247,11 +228,8 @@ mainloop(const struct sockaddr_in *in, int in_cnt, int nconn_max,
                        } else {
                                int error = errno;
 
-                               if (error != EINPROGRESS) {
-                                       fprintf(stderr, "connect failed: %d\n",
-                                           error);
-                                       return;
-                               }
+                               if (error != EINPROGRESS)
+                                       errc(1, error, "connect failed");
                        }
                        ++nconn;
 
@@ -269,10 +247,8 @@ mainloop(const struct sockaddr_in *in, int in_cnt, int nconn_max,
                        evt_change = evt_change0;
 
                n = kevent(kq, evt_change, nchange, evt, nevt_max, NULL);
-               if (n < 0) {
-                       fprintf(stderr, "kevent failed: %d\n", errno);
-                       return;
-               }
+               if (n < 0)
+                       err(1, "kevent failed");
                nchange = 0;
 
                for (i = 0; i < n; ++i) {