diff options
Diffstat (limited to 'tools/regression/sockets/accept_fd_leak/accept_fd_leak.c')
-rw-r--r-- | tools/regression/sockets/accept_fd_leak/accept_fd_leak.c | 68 |
1 files changed, 24 insertions, 44 deletions
diff --git a/tools/regression/sockets/accept_fd_leak/accept_fd_leak.c b/tools/regression/sockets/accept_fd_leak/accept_fd_leak.c index 6eae8a3..c81230f 100644 --- a/tools/regression/sockets/accept_fd_leak/accept_fd_leak.c +++ b/tools/regression/sockets/accept_fd_leak/accept_fd_leak.c @@ -31,6 +31,7 @@ #include <netinet/in.h> +#include <err.h> #include <errno.h> #include <fcntl.h> #include <stdio.h> @@ -59,16 +60,12 @@ main(int argc, char *argv[]) */ fd1 = dup(STDIN_FILENO); fd2 = dup(STDIN_FILENO); - if (fd2 != fd1 + 1) { - fprintf(stderr, "Non-sequential fd allocation!\n"); - exit(-1); - } + if (fd2 != fd1 + 1) + errx(-1, "Non-sequential fd allocation\n"); s = socket(PF_INET, SOCK_STREAM, 0); - if (s == -1) { - perror("socket"); - exit(-1); - } + if (s == -1) + errx(-1, "socket: %s", strerror(errno)); bzero(&sin, sizeof(sin)); sin.sin_len = sizeof(sin); @@ -76,46 +73,30 @@ main(int argc, char *argv[]) sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK); sin.sin_port = htons(8080); - if (bind(s, (struct sockaddr *) &sin, sizeof(sin)) != 0) { - perror("bind"); - exit(-1); - } + if (bind(s, (struct sockaddr *) &sin, sizeof(sin)) != 0) + errx(-1, "bind: %s", strerror(errno)); - if (listen(s, -1) != 0) { - perror("listen"); - exit(-1); - } + if (listen(s, -1) != 0) + errx(-1, "listen: %s", strerror(errno)); i = fcntl(s, F_GETFL); - if (i == -1) { - perror("F_GETFL"); - exit(-1); - } + if (i == -1) + errx(-1, "ioctl(F_GETFL): %s", strerror(errno)); i |= O_NONBLOCK; - if (fcntl(s, F_SETFL, i) != 0) { - perror("F_SETFL"); - exit(-1); - } + if (fcntl(s, F_SETFL, i) != 0) + errx(-1, "ioctl(F_SETFL): %s", strerror(errno)); i = fcntl(s, F_GETFL); - if (i == -1) { - perror("F_GETFL"); - exit(-1); - } - if ((i & O_NONBLOCK) != O_NONBLOCK) { - fprintf(stderr, "Failed to set O_NONBLOCK (i=%d)\n", i); - exit(-1); - } + if (i == -1) + errx(-1, "ioctl(F_GETFL): %s", strerror(errno)); + if ((i & O_NONBLOCK) != O_NONBLOCK) + errx(-1, "Failed to set O_NONBLOCK (i=0x%x)\n", i); for (i = 0; i < LOOPS; i++) { size = sizeof(sin); - if (accept(s, (struct sockaddr *)&sin, &size) != -1) { - fprintf(stderr, "accept succeeded!\n"); - exit(-1); - } - if (errno != EAGAIN) { - perror("accept"); - exit(-1); - } + if (accept(s, (struct sockaddr *)&sin, &size) != -1) + errx(-1, "accept succeeded\n"); + if (errno != EAGAIN) + errx(-1, "accept: %s", strerror(errno)); } /* @@ -123,10 +104,9 @@ main(int argc, char *argv[]) * we allocate an fd for the socket. */ fd3 = dup(STDIN_FILENO); - if (fd3 != fd2 + 2) { - fprintf(stderr, "FAIL (%d, %d, %d)\n", fd1, fd2, fd3); - exit(-1); - } else + if (fd3 != fd2 + 2) + errx(-1, "FAIL (%d, %d, %d)\n", fd1, fd2, fd3); + else fprintf(stderr, "PASS\n"); return (0); |