1 /* $NetBSD: util.c,v 1.8 2008/07/02 07:44:15 dyoung Exp $ */ 2 3 /*- 4 * Copyright (c) 2008 David Young. 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 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 28 #include <sys/cdefs.h> 29 #ifndef lint 30 __RCSID("$NetBSD: util.c,v 1.8 2008/07/02 07:44:15 dyoung Exp $"); 31 #endif /* not lint */ 32 33 #include <ctype.h> 34 #include <err.h> 35 #include <errno.h> 36 #include <stddef.h> 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <string.h> 40 #include <unistd.h> 41 #include <util.h> 42 43 #include <sys/socket.h> 44 #include <sys/ioctl.h> 45 #include <net/if.h> 46 #include <netinet/in.h> /* XXX */ 47 48 #include "env.h" 49 #include "util.h" 50 51 int 52 getsock(int naf) 53 { 54 static int oaf = -1, s; 55 56 if (oaf == naf || (oaf != -1 && naf == AF_UNSPEC)) 57 return s; 58 59 if (oaf != -1) 60 close(s); 61 62 if (naf == AF_UNSPEC) 63 naf = AF_INET; 64 65 s = socket(naf, SOCK_DGRAM, 0); 66 if (s == -1) 67 oaf = -1; 68 else 69 oaf = naf; 70 return s; 71 } 72 73 const char * 74 get_string(const char *val, const char *sep, u_int8_t *buf, int *lenp) 75 { 76 int len; 77 bool hexstr; 78 u_int8_t *p; 79 80 len = *lenp; 81 p = buf; 82 hexstr = (val[0] == '0' && tolower((u_char)val[1]) == 'x'); 83 if (hexstr) 84 val += 2; 85 for (;;) { 86 if (*val == '\0') 87 break; 88 if (sep != NULL && strchr(sep, *val) != NULL) { 89 val++; 90 break; 91 } 92 if (hexstr) { 93 if (!isxdigit((u_char)val[0]) || 94 !isxdigit((u_char)val[1])) { 95 warnx("bad hexadecimal digits"); 96 return NULL; 97 } 98 } 99 if (p > buf + len) { 100 if (hexstr) 101 warnx("hexadecimal digits too long"); 102 else 103 warnx("strings too long"); 104 return NULL; 105 } 106 if (hexstr) { 107 #define tohex(x) (isdigit(x) ? (x) - '0' : tolower(x) - 'a' + 10) 108 *p++ = (tohex((u_char)val[0]) << 4) | 109 tohex((u_char)val[1]); 110 #undef tohex 111 val += 2; 112 } else 113 *p++ = *val++; 114 } 115 len = p - buf; 116 if (len < *lenp) 117 memset(p, 0, *lenp - len); 118 *lenp = len; 119 return val; 120 } 121 122 void 123 print_string(const u_int8_t *buf, int len) 124 { 125 int i; 126 bool hasspc; 127 128 i = 0; 129 hasspc = false; 130 if (len < 2 || buf[0] != '0' || tolower(buf[1]) != 'x') { 131 for (; i < len; i++) { 132 if (!isprint(buf[i])) 133 break; 134 if (isspace(buf[i])) 135 hasspc = true; 136 } 137 } 138 if (i == len) { 139 if (hasspc || len == 0) 140 printf("\"%.*s\"", len, buf); 141 else 142 printf("%.*s", len, buf); 143 } else { 144 printf("0x"); 145 for (i = 0; i < len; i++) 146 printf("%02x", buf[i]); 147 } 148 } 149 150 struct paddr_prefix * 151 prefixlen_to_mask(int af, int plen) 152 { 153 union { 154 struct sockaddr sa; 155 struct sockaddr_in sin; 156 struct sockaddr_in6 sin6; 157 } u; 158 struct paddr_prefix *pfx; 159 size_t addrlen; 160 uint8_t *addr; 161 int nbit; 162 163 memset(&u, 0, sizeof(u)); 164 165 switch (af) { 166 case AF_INET: 167 addrlen = sizeof(u.sin.sin_addr); 168 addr = (uint8_t *)&u.sin.sin_addr; 169 u.sa.sa_len = sizeof(u.sin); 170 break; 171 case AF_INET6: 172 addrlen = sizeof(u.sin6.sin6_addr); 173 addr = (uint8_t *)&u.sin6.sin6_addr; 174 u.sa.sa_len = sizeof(u.sin6); 175 break; 176 default: 177 errno = EINVAL; 178 return NULL; 179 } 180 u.sa.sa_family = af; 181 182 if (plen < 0 || plen > addrlen * NBBY) { 183 errno = EINVAL; 184 return NULL; 185 } 186 187 if (plen == 0) 188 plen = addrlen * NBBY; 189 190 memset(addr, 0xff, (plen + NBBY - 1) / NBBY); 191 192 nbit = plen % NBBY; 193 if (nbit != 0) 194 addr[plen / NBBY] &= ~((uint8_t)0xff >> nbit); 195 pfx = malloc(offsetof(struct paddr_prefix, pfx_addr) + u.sa.sa_len); 196 if (pfx == NULL) 197 return NULL; 198 pfx->pfx_len = plen; 199 memcpy(&pfx->pfx_addr, &u.sa, u.sa.sa_len); 200 201 return pfx; 202 } 203 204 int 205 direct_ioctl(prop_dictionary_t env, unsigned long cmd, void *data) 206 { 207 const char *ifname; 208 int s; 209 210 if ((s = getsock(AF_UNSPEC)) == -1) 211 err(EXIT_FAILURE, "getsock"); 212 213 if ((ifname = getifname(env)) == NULL) 214 err(EXIT_FAILURE, "getifname"); 215 216 estrlcpy(data, ifname, IFNAMSIZ); 217 218 return ioctl(s, cmd, data); 219 } 220 221 int 222 indirect_ioctl(prop_dictionary_t env, unsigned long cmd, void *data) 223 { 224 struct ifreq ifr; 225 226 memset(&ifr, 0, sizeof(ifr)); 227 228 ifr.ifr_data = data; 229 230 return direct_ioctl(env, cmd, &ifr); 231 } 232 233 #ifdef INET6 234 /* KAME idiosyncrasy */ 235 void 236 in6_fillscopeid(struct sockaddr_in6 *sin6) 237 { 238 if (IN6_IS_ADDR_LINKLOCAL(&sin6->sin6_addr)) { 239 sin6->sin6_scope_id = 240 ntohs(*(u_int16_t *)&sin6->sin6_addr.s6_addr[2]); 241 sin6->sin6_addr.s6_addr[2] = sin6->sin6_addr.s6_addr[3] = 0; 242 } 243 } 244 #endif /* INET6 */ 245