1 /* $NetBSD: t_rfc6056.c,v 1.3 2012/06/22 14:54:35 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2011 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_rfc6056.c,v 1.3 2012/06/22 14:54:35 christos Exp $"); 33 34 #include <sys/types.h> 35 #include <sys/socket.h> 36 #include <netinet/in.h> 37 #include <netinet/udp.h> 38 #include <arpa/inet.h> 39 #include <string.h> 40 #include <strings.h> 41 #include <stdio.h> 42 #include <unistd.h> 43 #include <errno.h> 44 #include <stdlib.h> 45 #include <netdb.h> 46 #include <err.h> 47 48 #include <atf-c.h> 49 50 static void 51 test(const char *hostname, const char *service, int family, int al) 52 { 53 static const char hello[] = "hello\n"; 54 int s, error, proto, option; 55 struct sockaddr_storage ss; 56 struct addrinfo hints, *res; 57 58 memset(&hints, 0, sizeof(hints)); 59 hints.ai_family = family; 60 hints.ai_socktype = SOCK_DGRAM; 61 62 switch (family) { 63 case AF_INET: 64 proto = IPPROTO_IP; 65 option = IP_PORTALGO; 66 break; 67 case AF_INET6: 68 proto = IPPROTO_IPV6; 69 option = IPV6_PORTALGO; 70 break; 71 default: 72 abort(); 73 } 74 75 error = getaddrinfo(hostname, service, &hints, &res); 76 if (error) 77 errx(EXIT_FAILURE, "Cannot get address for %s (%s)", 78 hostname, gai_strerror(error)); 79 80 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol); 81 if (s == -1) 82 err(EXIT_FAILURE, "socket"); 83 84 if (setsockopt(s, proto, option, &al, sizeof(al)) == -1) 85 err(EXIT_FAILURE, "setsockopt"); 86 87 memset(&ss, 0, sizeof(ss)); 88 ss.ss_len = res->ai_addrlen; 89 ss.ss_family = res->ai_family; 90 91 if (bind(s, (struct sockaddr *)&ss, ss.ss_len) == -1) 92 err(EXIT_FAILURE, "bind"); 93 94 if (sendto(s, hello, sizeof(hello) - 1, 0, 95 res->ai_addr, res->ai_addrlen) == -1) 96 err(EXIT_FAILURE, "sendto"); 97 98 if (close(s) == -1) 99 err(EXIT_FAILURE, "close"); 100 101 s = socket(res->ai_family, res->ai_socktype, res->ai_protocol); 102 if (s == -1) 103 err(EXIT_FAILURE, "socket"); 104 105 if (setsockopt(s, proto, option, &al, sizeof(al)) == -1) 106 err(EXIT_FAILURE, "setsockopt"); 107 108 if (connect(s, res->ai_addr, res->ai_addrlen) == -1) 109 err(EXIT_FAILURE, "connect"); 110 111 if (send(s, hello, sizeof(hello) - 1, 0) == -1) 112 err(EXIT_FAILURE, "send"); 113 114 if (close(s) == -1) 115 err(EXIT_FAILURE, "close"); 116 117 freeaddrinfo(res); 118 } 119 120 ATF_TC(inet4); 121 ATF_TC_HEAD(inet4, tc) 122 { 123 atf_tc_set_md_var(tc, "descr", "Checks random port allocation " 124 "for ipv4"); 125 } 126 127 ATF_TC_BODY(inet4, tc) 128 { 129 for (int i = 0; i < 6; i++) 130 test("localhost", "http", AF_INET, i); 131 } 132 133 ATF_TC(inet6); 134 ATF_TC_HEAD(inet6, tc) 135 { 136 atf_tc_set_md_var(tc, "descr", "Checks random port allocation " 137 "for ipv6"); 138 } 139 140 ATF_TC_BODY(inet6, tc) 141 { 142 for (int i = 0; i < 6; i++) 143 test("localhost", "http", AF_INET6, i); 144 } 145 146 ATF_TP_ADD_TCS(tp) 147 { 148 ATF_TP_ADD_TC(tp, inet4); 149 ATF_TP_ADD_TC(tp, inet6); 150 151 return atf_no_error(); 152 } 153