1 /* $NetBSD: t_pktinfo.c,v 1.2 2013/10/19 17:45:01 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2013 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Christos Zoulas. 9 * 10 * Redistribution and use in source and binary forms, with or without 11 * modification, are permitted provided that the following conditions 12 * are met: 13 * 1. Redistributions of source code must retain the above copyright 14 * notice, this list of conditions and the following disclaimer. 15 * 2. Redistributions in binary form must reproduce the above copyright 16 * notice, this list of conditions and the following disclaimer in the 17 * documentation and/or other materials provided with the distribution. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 21 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 22 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 23 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 24 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 25 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 26 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 27 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 28 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 29 * POSSIBILITY OF SUCH DAMAGE. 30 */ 31 #include <sys/cdefs.h> 32 __RCSID("$NetBSD: t_pktinfo.c,v 1.2 2013/10/19 17:45:01 christos Exp $"); 33 34 #include <sys/types.h> 35 #include <sys/socket.h> 36 #include <netinet/in.h> 37 #include <arpa/inet.h> 38 #include <stdlib.h> 39 #include <unistd.h> 40 #include <stdio.h> 41 #include <string.h> 42 #include <errno.h> 43 44 static const char traffic[] = "foo"; 45 46 #ifdef TEST 47 #include <err.h> 48 #define ERR(msg) err(EXIT_FAILURE, msg) 49 #define ERRX(msg, a) errx(EXIT_FAILURE, msg, a) 50 #define ERRX2(msg, a1, a2) errx(EXIT_FAILURE, msg, a1, a2) 51 #else 52 #include <atf-c.h> 53 #define ERR(msg) ATF_REQUIRE_MSG(0, "%s: %s", msg, strerror(errno)) 54 #define ERRX(msg, a) ATF_REQUIRE_MSG(0, msg, a) 55 #define ERRX2(msg, a1, a2) ATF_REQUIRE_MSG(0, msg, a1, a2) 56 #endif 57 58 static int 59 server(struct sockaddr_in *sin) { 60 int s, one; 61 socklen_t len = sizeof(*sin); 62 63 if ((s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1) 64 ERR("socket"); 65 66 memset(sin, 0, len); 67 sin->sin_family = AF_INET; 68 sin->sin_len = len; 69 sin->sin_port = 0; 70 sin->sin_addr.s_addr = htonl(INADDR_LOOPBACK); 71 72 if (bind(s, (const struct sockaddr *)sin, len) == -1) 73 ERR("bind"); 74 75 if (getsockname(s, (struct sockaddr *)sin, &len) == -1) 76 ERR("getsockname"); 77 78 one = 1; 79 if (setsockopt(s, IPPROTO_IP, IP_PKTINFO, &one, sizeof(one)) == -1) 80 ERR("setsockopt"); 81 if (setsockopt(s, IPPROTO_IP, IP_RECVPKTINFO, &one, sizeof(one)) == -1) 82 ERR("setsockopt"); 83 84 return s; 85 } 86 87 static int 88 client(struct sockaddr_in *sin) { 89 int s; 90 91 if ((s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1) 92 ERR("socket"); 93 if (sendto(s, traffic, sizeof(traffic), 0, 94 (const struct sockaddr *)sin, sizeof(*sin)) == -1) 95 ERR("sendto"); 96 return s; 97 } 98 99 static void 100 receive(int s) { 101 struct msghdr msg; 102 struct cmsghdr *cmsg; 103 struct iovec iov; 104 char buf[sizeof(traffic)]; 105 struct in_pktinfo *ipi; 106 char control[CMSG_SPACE(sizeof(*ipi)) * 2]; 107 108 memset(&msg, 0, sizeof(msg)); 109 msg.msg_name = NULL; 110 msg.msg_namelen = 0; 111 msg.msg_iov = &iov; 112 memset(&iov, 0, sizeof(iov)); 113 iov.iov_base = buf; 114 iov.iov_len = sizeof(buf); 115 msg.msg_iovlen = 1; 116 msg.msg_control = control; 117 msg.msg_controllen = sizeof(control); 118 msg.msg_flags = 0; 119 120 if (recvmsg(s, &msg, 0) == -1) 121 ERR("recvmsg"); 122 123 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg != NULL; 124 cmsg = CMSG_NXTHDR(&msg, cmsg)) { 125 if (cmsg->cmsg_level != IPPROTO_IP) 126 ERRX("bad level %d", cmsg->cmsg_level); 127 const char *m; 128 switch (cmsg->cmsg_type) { 129 case IP_PKTINFO: 130 m = "pktinfo"; 131 break; 132 case IP_RECVPKTINFO: 133 m = "recvpktinfo"; 134 break; 135 default: 136 m = NULL; 137 ERRX("bad type %d", cmsg->cmsg_type); 138 } 139 ipi = (void *)CMSG_DATA(cmsg); 140 #ifdef TEST 141 printf("%s message received on address %s at interface %d\n", 142 m, inet_ntoa(ipi->ipi_addr), ipi->ipi_ifindex); 143 #else 144 __USE(m); 145 ATF_REQUIRE_MSG(ipi->ipi_addr.s_addr == htonl(INADDR_LOOPBACK), 146 "address 0x%x != 0x%x", ipi->ipi_addr.s_addr, 147 htonl(INADDR_LOOPBACK)); 148 #endif 149 } 150 151 if (strcmp(traffic, buf) != 0) 152 ERRX2("Bad message '%s' != '%s'", buf, traffic); 153 } 154 155 static void 156 doit(void) 157 { 158 struct sockaddr_in sin; 159 int s, c; 160 s = server(&sin); 161 c = client(&sin); 162 receive(s); 163 close(s); 164 close(c); 165 } 166 167 #ifndef TEST 168 ATF_TC(pktinfo); 169 ATF_TC_HEAD(pktinfo, tc) 170 { 171 172 atf_tc_set_md_var(tc, "descr", "Check that IP_PKTINFO and " 173 "IP_RECVPKTINFO work"); 174 } 175 176 ATF_TC_BODY(pktinfo, tc) 177 { 178 doit(); 179 } 180 181 ATF_TP_ADD_TCS(tp) 182 { 183 184 ATF_TP_ADD_TC(tp, pktinfo); 185 return atf_no_error(); 186 } 187 #else 188 189 int 190 main(int argc, char *argv[]) { 191 doit(); 192 return 0; 193 } 194 #endif 195