1 /*- 2 * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 3 * 4 * Copyright (c) 2019 Alexander V. Chernikov 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 AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 * 27 * $FreeBSD$ 28 */ 29 30 #ifndef _NET_ROUTING_RTSOCK_CONFIG_H_ 31 #define _NET_ROUTING_RTSOCK_CONFIG_H_ 32 33 #include "params.h" 34 35 struct rtsock_config_options { 36 int num_interfaces; /* number of interfaces to create */ 37 }; 38 39 struct rtsock_test_config { 40 int ifindex; 41 char net4_str[INET_ADDRSTRLEN]; 42 char addr4_str[INET_ADDRSTRLEN]; 43 char net6_str[INET6_ADDRSTRLEN]; 44 char addr6_str[INET6_ADDRSTRLEN]; 45 struct sockaddr_in net4; 46 struct sockaddr_in mask4; 47 struct sockaddr_in addr4; 48 struct sockaddr_in6 net6; 49 struct sockaddr_in6 mask6; 50 struct sockaddr_in6 addr6; 51 int plen4; 52 int plen6; 53 char *remote_lladdr; 54 char *ifname; 55 char **ifnames; 56 bool autocreated_interface; 57 int rtsock_fd; 58 int num_interfaces; 59 }; 60 61 struct rtsock_test_config * 62 config_setup(const atf_tc_t *tc, struct rtsock_config_options *co) 63 { 64 struct rtsock_config_options default_co; 65 struct rtsock_test_config *c; 66 char buf[64], *s; 67 const char *key; 68 int mask; 69 70 if (co == NULL) { 71 bzero(&default_co, sizeof(default_co)); 72 co = &default_co; 73 co->num_interfaces = 1; 74 } 75 76 c = calloc(1, sizeof(struct rtsock_test_config)); 77 c->rtsock_fd = -1; 78 79 key = atf_tc_get_config_var_wd(tc, "rtsock.v4prefix", "192.0.2.0/24"); 80 strlcpy(buf, key, sizeof(buf)); 81 if ((s = strchr(buf, '/')) == NULL) 82 return (NULL); 83 *s++ = '\0'; 84 mask = strtol(s, NULL, 10); 85 if (mask < 0 || mask > 32) 86 return (NULL); 87 c->plen4 = mask; 88 inet_pton(AF_INET, buf, &c->net4.sin_addr); 89 90 c->net4.sin_len = sizeof(struct sockaddr_in); 91 c->net4.sin_family = AF_INET; 92 c->addr4.sin_len = sizeof(struct sockaddr_in); 93 c->addr4.sin_family = AF_INET; 94 95 sa_fill_mask4(&c->mask4, c->plen4); 96 97 /* Fill in interface IPv4 address. Assume the first address in net */ 98 c->addr4.sin_addr.s_addr = htonl(ntohl(c->net4.sin_addr.s_addr) + 1); 99 inet_ntop(AF_INET, &c->net4.sin_addr, c->net4_str, INET_ADDRSTRLEN); 100 inet_ntop(AF_INET, &c->addr4.sin_addr, c->addr4_str, INET_ADDRSTRLEN); 101 102 key = atf_tc_get_config_var_wd(tc, "rtsock.v6prefix", "2001:DB8::/32"); 103 strlcpy(buf, key, sizeof(buf)); 104 if ((s = strchr(buf, '/')) == NULL) 105 return (NULL); 106 *s++ = '\0'; 107 mask = strtol(s, NULL, 10); 108 if (mask < 0 || mask > 128) 109 return (NULL); 110 c->plen6 = mask; 111 112 inet_pton(AF_INET6, buf, &c->net6.sin6_addr); 113 114 c->net6.sin6_len = sizeof(struct sockaddr_in6); 115 c->net6.sin6_family = AF_INET6; 116 c->addr6.sin6_len = sizeof(struct sockaddr_in6); 117 c->addr6.sin6_family = AF_INET6; 118 119 sa_fill_mask6(&c->mask6, c->plen6); 120 121 /* Fill in interface IPv6 address. Assume the first address in net */ 122 memcpy(&c->addr6.sin6_addr, &c->net6.sin6_addr, sizeof(struct in6_addr)); 123 #define _s6_addr32 __u6_addr.__u6_addr32 124 c->addr6.sin6_addr._s6_addr32[3] = htonl(ntohl(c->net6.sin6_addr._s6_addr32[3]) + 1); 125 #undef _s6_addr32 126 inet_ntop(AF_INET6, &c->net6.sin6_addr, c->net6_str, INET6_ADDRSTRLEN); 127 inet_ntop(AF_INET6, &c->addr6.sin6_addr, c->addr6_str, INET6_ADDRSTRLEN); 128 129 if (co->num_interfaces > 0) { 130 kldload("if_epair"); 131 ATF_REQUIRE_KERNEL_MODULE("if_epair"); 132 133 c->ifnames = calloc(co->num_interfaces, sizeof(char *)); 134 for (int i = 0; i < co->num_interfaces; i++) 135 c->ifnames[i] = iface_create("epair"); 136 137 c->ifname = c->ifnames[0]; 138 c->ifindex = if_nametoindex(c->ifname); 139 ATF_REQUIRE_MSG(c->ifindex != 0, "interface %s not found", 140 c->ifname); 141 } 142 c->num_interfaces = co->num_interfaces; 143 144 c->remote_lladdr = strdup(atf_tc_get_config_var_wd(tc, 145 "rtsock.remote_lladdr", "00:00:5E:00:53:42")); 146 147 return (c); 148 } 149 150 void 151 config_generic_cleanup(const atf_tc_t *tc) 152 { 153 const char *srcdir = atf_tc_get_config_var(tc, "srcdir"); 154 char cmd[512]; 155 int ret; 156 157 /* XXX: sleep 100ms to avoid epair qflush panic */ 158 usleep(1000 * 100); 159 snprintf(cmd, sizeof(cmd), "%s/generic_cleanup.sh", srcdir); 160 ret = system(cmd); 161 if (ret != 0) 162 RLOG("'%s' failed, error %d", cmd, ret); 163 } 164 165 void 166 config_describe_root_test(atf_tc_t *tc, char *test_descr) 167 { 168 169 atf_tc_set_md_var(tc, "descr", test_descr); 170 // Adding/deleting prefix requires root privileges 171 atf_tc_set_md_var(tc, "require.user", "root"); 172 } 173 174 #endif 175