1 /*
2 * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. The name of the author may not be used to endorse or promote products
14 * derived from this software without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
27
28 #include "tests.h"
29 #include <assert.h>
30 #include <fcntl.h>
31 #include <stdio.h>
32 #include <stdint.h>
33 #include <unistd.h>
34 #include <sys/socket.h>
35 #include <netinet/in.h>
36 #include <arpa/inet.h>
37
38 static void
print_pktinfo(const struct cmsghdr * c)39 print_pktinfo(const struct cmsghdr *c)
40 {
41 printf("IP_PKTINFO, cmsg_data={ipi_ifindex=if_nametoindex(\"lo\")"
42 ", ipi_spec_dst=inet_addr(\"127.0.0.1\")"
43 ", ipi_addr=inet_addr(\"127.0.0.1\")}");
44 }
45
46 static void
print_ttl(const struct cmsghdr * c)47 print_ttl(const struct cmsghdr *c)
48 {
49 const unsigned int *ttl = (const unsigned int *) CMSG_DATA(c);
50
51 printf("IP_TTL, cmsg_data=[%u]", *ttl);
52 }
53
54 static void
print_tos(const struct cmsghdr * c)55 print_tos(const struct cmsghdr *c)
56 {
57 const uint8_t *tos = (const uint8_t *) CMSG_DATA(c);
58
59 printf("IP_TOS, cmsg_data=[%#x]", *tos);
60 }
61
62 static void
print_opts(const char * name,const struct cmsghdr * c)63 print_opts(const char *name, const struct cmsghdr *c)
64 {
65 const unsigned char *opts = (const unsigned char *) CMSG_DATA(c);
66 const size_t len = c->cmsg_len - CMSG_ALIGN(sizeof(*c));
67
68 printf("%s", name);
69 if (len) {
70 printf(", cmsg_data=[");
71 size_t i;
72 for (i = 0; i < len; ++i)
73 printf("%s0x%02x", i ? ", " : "", opts[i]);
74 printf("]");
75 }
76 }
77
78 #ifdef IP_ORIGDSTADDR
79 static void
print_origdstaddr(const struct cmsghdr * c)80 print_origdstaddr(const struct cmsghdr *c)
81 {
82 const struct sockaddr_in *sin =
83 (const struct sockaddr_in *) CMSG_DATA(c);
84
85 printf("IP_ORIGDSTADDR, cmsg_data={sa_family=AF_INET, sin_port=htons(%u)"
86 ", sin_addr=inet_addr(\"127.0.0.1\")}", ntohs(sin->sin_port));
87 }
88 #endif
89
90 int
main(void)91 main(void)
92 {
93 int i;
94 while ((i = open("/dev/null", O_RDWR)) < 3)
95 assert(i >= 0);
96 assert(!close(0));
97 assert(!close(3));
98
99 if (socket(AF_INET, SOCK_DGRAM, 0))
100 perror_msg_and_skip("socket");
101 struct sockaddr_in addr = {
102 .sin_family = AF_INET,
103 .sin_addr.s_addr = htonl(INADDR_LOOPBACK)
104 };
105 socklen_t len = sizeof(addr);
106 if (bind(0, (struct sockaddr *) &addr, len))
107 perror_msg_and_skip("bind");
108 assert(!getsockname(0, (struct sockaddr *) &addr, &len));
109
110 assert(socket(AF_INET, SOCK_DGRAM, 0) == 3);
111 assert(!connect(3, (struct sockaddr *) &addr, len));
112
113 const int opt_1 = htonl(0x01000000);
114 #define SETSOCKOPT(fd, name) assert(!setsockopt(fd, IPPROTO_IP, (name), &opt_1, sizeof(opt_1)))
115 SETSOCKOPT(3, IP_OPTIONS);
116 SETSOCKOPT(0, IP_PKTINFO);
117 SETSOCKOPT(0, IP_RECVTTL);
118 SETSOCKOPT(0, IP_RECVTOS);
119 SETSOCKOPT(0, IP_RECVOPTS);
120 SETSOCKOPT(0, IP_RETOPTS);
121 #ifdef IP_RECVORIGDSTADDR
122 SETSOCKOPT(0, IP_RECVORIGDSTADDR);
123 #endif
124
125 static const char data[] = "data";
126 const size_t size = sizeof(data) - 1;
127 assert(send(3, data, size, 0) == (int) size);
128 assert(!close(3));
129
130 char buf[size];
131 struct iovec iov = {
132 .iov_base = buf,
133 .iov_len = sizeof(buf)
134 };
135 struct cmsghdr control[16];
136 struct msghdr mh = {
137 .msg_name = &addr,
138 .msg_namelen = len,
139 .msg_iov = &iov,
140 .msg_iovlen = 1,
141 .msg_control = control,
142 .msg_controllen = sizeof(control)
143 };
144
145 assert(recvmsg(0, &mh, 0) == (int) size);
146 assert(!close(0));
147
148 printf("recvmsg(0, {msg_name={sa_family=AF_INET, sin_port=htons(%u)"
149 ", sin_addr=inet_addr(\"127.0.0.1\")}, msg_namelen=%u"
150 ", msg_iov=[{iov_base=\"%s\", iov_len=%u}], msg_iovlen=1"
151 ", msg_control=[",
152 ntohs(addr.sin_port), (unsigned) mh.msg_namelen,
153 data, (unsigned) size);
154
155 struct cmsghdr *c;
156 for (c = CMSG_FIRSTHDR(&mh); c; c = CMSG_NXTHDR(&mh, c)) {
157 if (IPPROTO_IP != c->cmsg_level)
158 continue;
159 if (c != control)
160 printf(", ");
161 printf("{cmsg_len=%lu, cmsg_level=SOL_IP, cmsg_type=",
162 (unsigned long) c->cmsg_len);
163 switch (c->cmsg_type) {
164 case IP_PKTINFO:
165 print_pktinfo(c);
166 break;
167 case IP_TTL:
168 print_ttl(c);
169 break;
170 case IP_TOS:
171 print_tos(c);
172 break;
173 case IP_RECVOPTS:
174 print_opts("IP_RECVOPTS", c);
175 break;
176 case IP_RETOPTS:
177 print_opts("IP_RETOPTS", c);
178 break;
179 #ifdef IP_ORIGDSTADDR
180 case IP_ORIGDSTADDR:
181 print_origdstaddr(c);
182 break;
183 #endif
184 default:
185 printf("%d", c->cmsg_type);
186 break;
187 }
188 printf("}");
189 }
190 printf("], msg_controllen=%lu, msg_flags=0}, 0) = %u\n",
191 (unsigned long) mh.msg_controllen, (unsigned) size);
192 puts("+++ exited with 0 +++");
193
194 return 0;
195 }
196