socket/socketpair: Add SOCK_{NONBLOCK,CLOEXEC} support.
[dragonfly.git] / test / sockext / socket / nonblock / socket_nblock.c
1 #include <sys/types.h>
2 #include <sys/socket.h>
3 #include <sys/time.h>
4
5 #include <arpa/inet.h>
6 #include <netinet/in.h>
7
8 #include <err.h>
9 #include <errno.h>
10 #include <signal.h>
11 #include <stdio.h>
12 #include <stdlib.h>
13 #include <string.h>
14 #include <unistd.h>
15
16 #define READ_BLOCK_TIME         5       /* unit: sec */
17
18 static void
19 sig_alarm(int sig __unused)
20 {
21 #define PANIC_STRING    "read blocks\n"
22
23         write(2, PANIC_STRING, strlen(PANIC_STRING));
24         abort();
25 }
26
27 static void
28 usage(const char *cmd)
29 {
30         fprintf(stderr, "%s -p port\n", cmd);
31         exit(1);
32 }
33
34 int
35 main(int argc, char *argv[])
36 {
37         struct sockaddr_in in;
38         struct itimerval it;
39         int s, n, error, port, opt;
40         char buf;
41
42         port = 0;
43         while ((opt = getopt(argc, argv, "p:")) != -1) {
44                 char *endptr;
45
46                 switch (opt) {
47                 case 'p':
48                         port = strtol(optarg, &endptr, 0);
49                         if (*endptr != '\0')
50                                 fprintf(stderr, "invalid -p argument\n");
51                         break;
52
53                 default:
54                         usage(argv[0]);
55                 }
56         }
57         if (port <= 0)
58                 usage(argv[0]);
59
60         s = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, 0);
61         if (s < 0)
62                 err(1, "socket failed");
63
64         memset(&in, 0, sizeof(in));
65         in.sin_family = AF_INET;
66         in.sin_port = htons(port);
67         if (bind(s, (const struct sockaddr *)&in, sizeof(in)) < 0)
68                 err(1, "bind %d failed", port);
69
70         memset(&it, 0, sizeof(it));
71         it.it_value.tv_sec = READ_BLOCK_TIME;
72         if (signal(SIGALRM, sig_alarm) == SIG_ERR)
73                 err(1, "signal failed");
74         if (setitimer(ITIMER_REAL, &it, NULL) < 0)
75                 err(1, "setitimer failed");
76
77         n = read(s, &buf, 1);
78         if (n < 0) {
79                 error = errno;
80                 if (error != EAGAIN) {
81                         fprintf(stderr, "invalid errno %d\n", error);
82                         abort();
83                 }
84         } else {
85                 fprintf(stderr, "read works\n");
86                 abort();
87         }
88
89         fprintf(stderr, "passed\n");
90         exit(0);
91 }