1 /* $NetBSD: t_connect.c,v 1.2 2015/04/05 23:17:41 rtr Exp $ */ 2 /* 3 * Copyright (c) 2007, 2008 The NetBSD Foundation, Inc. 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND 16 * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, 17 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF 18 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19 * IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY 20 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE 22 * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 24 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR 25 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN 26 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #include <err.h> 30 #include <errno.h> 31 #include <string.h> 32 #include <unistd.h> 33 34 #include <arpa/inet.h> 35 #include <netinet/in.h> 36 37 #include <atf-c.h> 38 39 #ifdef __FreeBSD__ 40 #include <sys/socket.h> 41 #endif 42 43 ATF_TC(connect_low_port); 44 ATF_TC_HEAD(connect_low_port, tc) 45 { 46 atf_tc_set_md_var(tc, "descr", "Checks that low-port allocation " 47 "works"); 48 atf_tc_set_md_var(tc, "require.user", "root"); 49 } 50 ATF_TC_BODY(connect_low_port, tc) 51 { 52 struct sockaddr_in sin, sinlist; 53 int sd, val, slist; 54 socklen_t slen; 55 56 slist = socket(AF_INET, SOCK_STREAM, 0); 57 sd = socket(AF_INET, SOCK_STREAM, 0); 58 59 #ifdef __FreeBSD__ 60 ATF_REQUIRE(sd > 0); 61 ATF_REQUIRE(slist > 0); 62 #endif 63 64 /* bind listening socket */ 65 memset(&sinlist, 0, sizeof(sinlist)); 66 sinlist.sin_family = AF_INET; 67 sinlist.sin_port = htons(31522); 68 sinlist.sin_addr.s_addr = inet_addr("127.0.0.1"); 69 70 ATF_REQUIRE_EQ(bind(slist, 71 (struct sockaddr *)&sinlist, sizeof(sinlist)), 0); 72 ATF_REQUIRE_EQ(listen(slist, 1), 0); 73 74 val = IP_PORTRANGE_LOW; 75 if (setsockopt(sd, IPPROTO_IP, IP_PORTRANGE, &val, 76 sizeof(val)) == -1) 77 atf_tc_fail("setsockopt failed: %s", strerror(errno)); 78 79 memset(&sin, 0, sizeof(sin)); 80 81 sin.sin_port = htons(31522); 82 sin.sin_addr.s_addr = inet_addr("127.0.0.1"); 83 sin.sin_family = AF_INET; 84 85 if (connect(sd, (struct sockaddr *)&sin, sizeof(sin)) == -1) { 86 int serrno = errno; 87 atf_tc_fail("connect failed: %s%s", 88 strerror(serrno), 89 serrno != EACCES ? "" : 90 " (see http://mail-index.netbsd.org/" 91 "source-changes/2007/12/16/0011.html)"); 92 } 93 94 slen = sizeof(sin); 95 ATF_REQUIRE_EQ(getsockname(sd, (struct sockaddr *)&sin, &slen), 0); 96 ATF_REQUIRE_EQ(slen, sizeof(sin)); 97 ATF_REQUIRE(ntohs(sin.sin_port) <= IPPORT_RESERVEDMAX); 98 99 close(sd); 100 #ifdef __FreeBSD__ 101 close(slist); 102 #endif 103 } 104 105 ATF_TC(connect_foreign_family); 106 ATF_TC_HEAD(connect_foreign_family, tc) 107 { 108 atf_tc_set_md_var(tc, "descr", "Checks that connecting a socket " 109 "with a different address family fails"); 110 } 111 ATF_TC_BODY(connect_foreign_family, tc) 112 { 113 struct sockaddr_in addr; 114 115 /* addr.sin_family = AF_UNSPEC = 0 */ 116 memset(&addr, 0, sizeof(addr)); 117 118 /* 119 * it is not necessary to initialize sin_{addr,port} since 120 * those structure members shall not be accessed if connect 121 * fails correctly. 122 */ 123 124 int sock = socket(AF_LOCAL, SOCK_STREAM, 0); 125 ATF_REQUIRE(sock != -1); 126 127 ATF_REQUIRE(-1 == connect(sock, (struct sockaddr *)&addr, sizeof(addr))); 128 ATF_REQUIRE(EAFNOSUPPORT == errno); 129 130 close(sock); 131 } 132 133 ATF_TP_ADD_TCS(tp) 134 { 135 136 ATF_TP_ADD_TC(tp, connect_low_port); 137 ATF_TP_ADD_TC(tp, connect_foreign_family); 138 139 return atf_no_error(); 140 } 141