Fix GETATTR_LITE() build breakage with 'pseudo-device gzip' in the config.
[dragonfly.git] / test / udp / conn2send / udp_conn2send.c
1 #include <sys/types.h>
2 #include <sys/socket.h>
3
4 #include <arpa/inet.h>
5 #include <netinet/in.h>
6
7 #include <err.h>
8 #include <errno.h>
9 #include <stdio.h>
10 #include <stdint.h>
11 #include <stdlib.h>
12 #include <string.h>
13 #include <unistd.h>
14
15 static void
16 usage(const char *cmd)
17 {
18         fprintf(stderr, "%s -4 ip4 -p port [-r]\n", cmd);
19         exit(1);
20 }
21
22 int
23 main(int argc, char *argv[])
24 {
25         struct sockaddr_in in, in2;
26         int s, opt, n, reuseport;
27         uint8_t buf[18];
28
29         memset(&in, 0, sizeof(in));
30         in.sin_family = AF_INET;
31
32         reuseport = 0;
33
34         while ((opt = getopt(argc, argv, "4:p:r")) != -1) {
35                 switch (opt) {
36                 case '4':
37                         if (inet_pton(AF_INET, optarg, &in.sin_addr) <= 0)
38                                 usage(argv[0]);
39                         break;
40
41                 case 'p':
42                         in.sin_port = strtol(optarg, NULL, 10);
43                         in.sin_port = htons(in.sin_port);
44                         break;
45
46                 case 'r':
47                         reuseport = 1;
48                         break;
49
50                 default:
51                         usage(argv[0]);
52                 }
53         }
54
55         if (in.sin_addr.s_addr == INADDR_ANY || in.sin_port == 0)
56                 usage(argv[0]);
57
58         s = socket(AF_INET, SOCK_DGRAM, 0);
59         if (s < 0)
60                 err(2, "socket failed");
61
62         if (reuseport) {
63                 if (setsockopt(s, SOL_SOCKET, SO_REUSEPORT,
64                     &reuseport, sizeof(reuseport)) < 0)
65                         err(2, "setsockopt SO_REUSEPORT failed");
66         }
67
68         if (connect(s, (const struct sockaddr *)&in, sizeof(in)) < 0)
69                 err(2, "connect failed");
70
71         memset(&in2, 0, sizeof(in2));
72         in2.sin_family = AF_UNSPEC;
73         if (connect(s, (const struct sockaddr *)&in2, sizeof(in2)) < 0) {
74                 int error = errno;
75
76                 if (error != EAFNOSUPPORT)
77                         errc(2, error, "di-connect failed");
78         }
79
80         n = sendto(s, buf, sizeof(buf), 0,
81             (const struct sockaddr *)&in, sizeof(in));
82         if (n < 0)
83                 err(2, "sendto failed");
84         else if (n != (int)sizeof(buf))
85                 errx(2, "sent truncated data %d", n);
86
87         n = read(s, buf, sizeof(buf));
88         if (n < 0)
89                 err(2, "read failed");
90         printf("read %d, done\n", n);
91
92         exit(0);
93 }