1 /* 2 * Copyright (c) 1988, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. Neither the name of the University nor the names of its contributors 14 * may be used to endorse or promote products derived from this software 15 * without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27 * SUCH DAMAGE. 28 * 29 * @(#) Copyright (c) 1988, 1993 The Regents of the University of California. All rights reserved. 30 * @(#)hostname.c 8.1 (Berkeley) 5/31/93 31 * $FreeBSD: src/bin/hostname/hostname.c,v 1.10.2.1 2001/08/01 02:40:23 obrien Exp $ 32 */ 33 34 #include <sys/param.h> 35 #include <sys/ioctl.h> 36 #include <sys/socket.h> 37 #include <sys/sysctl.h> 38 #include <sys/module.h> 39 #include <sys/linker.h> 40 41 #include <net/ethernet.h> 42 #include <net/if.h> 43 #include <net/if_var.h> 44 #include <net/if_dl.h> 45 #include <net/if_types.h> 46 #include <net/route.h> 47 #include <netinet/in.h> 48 49 #include <err.h> 50 #include <stdio.h> 51 #include <stdlib.h> 52 #include <string.h> 53 #include <unistd.h> 54 55 #include <netdb.h> 56 #include <sys/types.h> 57 #include <arpa/inet.h> 58 59 #include <errno.h> 60 61 #define HST_IF (1 << 0) 62 #define HST_IF_V6 (1 << 1) 63 #define HST_IF_V4 (1 << 2) 64 65 /* 66 * Expand the compacted form of addresses as returned via the 67 * configuration read via sysctl(). 68 * Lifted from getifaddrs(3) 69 */ 70 71 static void rt_xaddrs(caddr_t, caddr_t, struct rt_addrinfo *); 72 static void usage (void); 73 74 #ifndef RT_ROUNDUP 75 #define RT_ROUNDUP(a) \ 76 ((a) > 0 ? (1 + (((a) - 1) | (sizeof(long) - 1))) : sizeof(long)) 77 #endif 78 #ifndef RT_ADVANCE 79 #define RT_ADVANCE(x, n) (x += RT_ROUNDUP((n)->sa_len)) 80 #endif 81 82 static 83 void 84 rt_xaddrs(caddr_t cp, caddr_t cplim, struct rt_addrinfo *rtinfo) 85 { 86 struct sockaddr *sa; 87 int i; 88 89 memset(rtinfo->rti_info, 0, sizeof(rtinfo->rti_info)); 90 for (i = 0; (i < RTAX_MAX) && (cp < cplim); i++) { 91 if ((rtinfo->rti_addrs & (1 << i)) == 0) 92 continue; 93 rtinfo->rti_info[i] = sa = (struct sockaddr *)cp; 94 RT_ADVANCE(cp, sa); 95 } 96 } 97 98 int 99 main(int argc, char **argv) 100 { 101 int ch, sflag, rflag, ret, flag6, iflag; 102 int silen = 0; 103 char hostname[MAXHOSTNAMELEN]; 104 char *srflag, *siflag; 105 struct hostent *hst; 106 struct in_addr ia; 107 struct in6_addr ia6; 108 109 int mib[6]; 110 size_t needed; 111 char *buf, *next, *p; 112 int idx; 113 struct sockaddr_dl *sdl; 114 struct rt_msghdr *rtm; 115 struct if_msghdr *ifm; 116 struct ifa_msghdr *ifam; 117 struct rt_addrinfo info; 118 struct sockaddr_in *sai; 119 struct sockaddr_in6 *sai6; 120 121 srflag = NULL; 122 siflag = NULL; 123 iflag = sflag = rflag = 0; 124 flag6 = 0; 125 hst = NULL; 126 127 while ((ch = getopt(argc, argv, "46fi:r:s")) != -1) { 128 switch (ch) { 129 case '4': 130 iflag |= HST_IF_V4; 131 break; 132 case '6': 133 iflag |= HST_IF_V6; 134 break; 135 case 'f': 136 /* Printing the FQDN is default. */ 137 break; 138 case 'i': 139 siflag = optarg; 140 silen = strlen(siflag); 141 iflag |= HST_IF; 142 break; 143 case 'r': 144 srflag = optarg; 145 rflag = 1; 146 break; 147 case 's': 148 sflag = 1; 149 break; 150 default: 151 usage(); 152 } 153 } 154 argc -= optind; 155 argv += optind; 156 157 if (argc > 1) 158 usage(); 159 160 if (iflag && *argv) 161 usage(); 162 163 if (rflag && *argv) 164 usage(); 165 166 if (rflag && (iflag & HST_IF)) 167 usage(); 168 169 if ((iflag & HST_IF_V6) && (iflag & HST_IF_V4)) 170 usage(); 171 172 if (!(iflag & HST_IF) && ((iflag & HST_IF_V6)||iflag & HST_IF_V4)) 173 usage(); 174 175 if (iflag & HST_IF) { 176 mib[0] = CTL_NET; 177 mib[1] = PF_ROUTE; 178 mib[2] = 0; 179 mib[3] = 0; 180 mib[4] = NET_RT_IFLIST; 181 mib[5] = 0; 182 183 idx = 0; 184 needed = 1; 185 186 if (sysctl(mib, 6, NULL, &needed, NULL, 0) < 0) 187 err(1, "sysctl: iflist-sysctl-estimate"); 188 if ((buf = malloc(needed)) == NULL) 189 err(1, "malloc failed"); 190 if (sysctl(mib, 6, buf, &needed, NULL, 0) < 0) 191 err(1, "sysctl: retrieval of interface table"); 192 193 for (next = buf; next < buf + needed; next += rtm->rtm_msglen) { 194 rtm = (struct rt_msghdr *)(void *)next; 195 if (rtm->rtm_version != RTM_VERSION) 196 continue; 197 switch (rtm->rtm_type) { 198 case RTM_IFINFO: 199 ifm = (struct if_msghdr *)(void *)rtm; 200 201 if ((ifm->ifm_addrs & RTA_IFP) == 0) 202 break; 203 sdl = (struct sockaddr_dl *)(ifm + 1); 204 if (silen != sdl->sdl_nlen) 205 break; 206 if (!strncmp(siflag, sdl->sdl_data, silen)) { 207 idx = ifm->ifm_index; 208 } 209 break; 210 case RTM_NEWADDR: 211 ifam = (struct ifa_msghdr *)(void *)rtm; 212 213 if (ifam->ifam_index == idx) { 214 info.rti_addrs = ifam->ifam_addrs; 215 rt_xaddrs((char *)(ifam + 1), 216 ifam->ifam_msglen + (char *)ifam, &info); 217 sai = (struct sockaddr_in *)info.rti_info[RTAX_IFA]; 218 219 if (iflag & HST_IF_V6) { 220 if (sai->sin_family == AF_INET6) { 221 sai6 = (struct sockaddr_in6 *)info.rti_info[RTAX_IFA]; 222 hst = gethostbyaddr(&sai6->sin6_addr, 223 sizeof(sai6->sin6_addr),AF_INET6); 224 225 if (h_errno == NETDB_SUCCESS) { 226 next = buf + needed; 227 continue; 228 } 229 } 230 } else { 231 if ((sai->sin_family == AF_INET)) { 232 233 hst = gethostbyaddr(&sai->sin_addr, 234 sizeof(sai->sin_addr),AF_INET); 235 236 if (h_errno == NETDB_SUCCESS) { 237 next = buf + needed; 238 continue; 239 } 240 } 241 } 242 } 243 break; 244 } /* switch */ 245 } /* loop */ 246 247 free(buf); 248 249 if (idx == 0) 250 errx(1,"interface not found"); 251 if (hst == NULL) 252 errx(1, "ip not found on interface"); 253 254 if (h_errno == NETDB_SUCCESS) { 255 if (sethostname(hst->h_name, (int)strlen(hst->h_name))) 256 err(1, "sethostname"); 257 } else if (h_errno == HOST_NOT_FOUND) { 258 errx(1,"hostname not found"); 259 } else { 260 herror("gethostbyaddr"); 261 exit(1); 262 } 263 } else if (rflag) { 264 ret = inet_pton(AF_INET, srflag, &ia); 265 if (ret != 1) { 266 /* check IPV6 */ 267 ret = inet_pton(AF_INET6, srflag, &ia6); 268 269 if (ret != 1) { 270 errx(1, "invalid ip address"); 271 } 272 273 flag6 = 1; 274 } 275 276 if (flag6 == 1) 277 hst = gethostbyaddr(&ia6, sizeof(ia6), AF_INET6); 278 else 279 hst = gethostbyaddr(&ia, sizeof(ia), AF_INET); 280 if (!hst) { 281 if (h_errno == HOST_NOT_FOUND) 282 errx(1,"host not found\n"); 283 } 284 285 if (sethostname(hst->h_name, (int)strlen(hst->h_name))) 286 err(1, "sethostname"); 287 } else if (*argv) { 288 if (sethostname(*argv, (int)strlen(*argv))) 289 err(1, "sethostname"); 290 } else { 291 if (gethostname(hostname, (int)sizeof(hostname))) 292 err(1, "gethostname"); 293 if (sflag && (p = strchr(hostname, '.'))) 294 *p = '\0'; 295 printf("%s\n", hostname); 296 } 297 exit(0); 298 } 299 300 static void 301 usage(void) 302 { 303 fprintf(stderr, "usage: hostname [-s] [name-of-host |" 304 " -r ip-address | -i interface [-4 | -6]]\n"); 305 exit(1); 306 } 307 308