sbin/dhclient/dispatch.c

   1 /*      $OpenBSD: src/sbin/dhclient/dispatch.c,v 1.54 2012/08/18 00:20:01 krw Exp $     */
   2
   3 /*
   4  * Copyright 2004 Henning Brauer <henning@openbsd.org>
   5  * Copyright (c) 1995, 1996, 1997, 1998, 1999
   6  * The Internet Software Consortium.   All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  *
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  * 3. Neither the name of The Internet Software Consortium nor the names
  18  *    of its contributors may be used to endorse or promote products derived
  19  *    from this software without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE INTERNET SOFTWARE CONSORTIUM AND
  22  * CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
  23  * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  24  * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  25  * DISCLAIMED.  IN NO EVENT SHALL THE INTERNET SOFTWARE CONSORTIUM OR
  26  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  27  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  28  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  29  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  30  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  31  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  32  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  *
  35  * This software has been written for the Internet Software Consortium
  36  * by Ted Lemon <mellon@fugue.com> in cooperation with Vixie
  37  * Enterprises.  To learn more about the Internet Software Consortium,
  38  * see ``http://www.vix.com/isc''.  To learn more about Vixie
  39  * Enterprises, see ``http://www.vix.com''.
  40  */
  41
  42 #include <sys/ioctl.h>
  43
  44 #include <net/if_media.h>
  45
  46 #include <ifaddrs.h>
  47 #include <poll.h>
  48
  49 #include "dhcpd.h"
  50
  51 struct timeout timeout;
  52 static int interfaces_invalidated;
  53
  54 /*
  55  * Use getifaddrs() to get a list of all the attached interfaces.  For
  56  * each interface that's of type INET and not the loopback interface,
  57  * register that interface with the network I/O software, figure out
  58  * what subnet it's on, and add it to the list of interfaces.
  59  */
  60 void
  61 discover_interface(void)
  62 {
  63         struct ifaddrs *ifap, *ifa;
  64         struct ifreq *tif;
  65         int len = IFNAMSIZ + sizeof(struct sockaddr_storage);
  66
  67         if (getifaddrs(&ifap) != 0)
  68                 error("getifaddrs failed");
  69
  70         for (ifa = ifap; ifa != NULL; ifa = ifa->ifa_next) {
  71                 if ((ifa->ifa_flags & IFF_LOOPBACK) ||
  72                     (ifa->ifa_flags & IFF_POINTOPOINT) ||
  73                     (!(ifa->ifa_flags & IFF_UP)))
  74                         continue;
  75
  76                 if (strcmp(ifi->name, ifa->ifa_name))
  77                         continue;
  78
  79                 /*
  80                  * If we have the capability, extract link information
  81                  * and record it in a linked list.
  82                  */
  83                 if (ifa->ifa_addr->sa_family == AF_LINK) {
  84                         struct sockaddr_dl *foo =
  85                             (struct sockaddr_dl *)ifa->ifa_addr;
  86
  87                         ifi->index = foo->sdl_index;
  88                         ifi->hw_address.hlen = foo->sdl_alen;
  89                         ifi->hw_address.htype = HTYPE_ETHER; /* XXX */
  90                         memcpy(ifi->hw_address.haddr,
  91                             LLADDR(foo), foo->sdl_alen);
  92                 }
  93                 if (!ifi->ifp) {
  94                         if ((tif = malloc(len)) == NULL)
  95                                 error("no space to remember ifp");
  96                         strlcpy(tif->ifr_name, ifa->ifa_name, IFNAMSIZ);
  97                         ifi->ifp = tif;
  98                 }
  99         }
 100
 101         if (!ifi->ifp)
 102                 error("%s: not found", ifi->name);
 103
 104         /* Register the interface... */
 105         if_register_receive();
 106         if_register_send();
 107         freeifaddrs(ifap);
 108 }
 109
 110 void
 111 reinitialize_interface(void)
 112 {
 113         interfaces_invalidated = 1;
 114 }
 115
 116 /*
 117  * Wait for packets to come in using poll().  When a packet comes in, call
 118  * receive_packet to receive the packet and possibly strip hardware addressing
 119  * information from it, and then call do_packet to try to do something with it.
 120  */
 121 void
 122 dispatch(void)
 123 {
 124         int count, to_msec;
 125         struct pollfd fds[2];
 126         time_t howlong;
 127         void (*func)(void);
 128
 129         do {
 130                 /*
 131                  * Call expired timeout, and then if there's still
 132                  * a timeout registered, time out the select call then.
 133                  */
 134 another:
 135                 if (!ifi)
 136                         error("No interfaces available");
 137
 138                 if (!ifi->linkstat)
 139                         interfaces_invalidated = 0;
 140
 141                 if (timeout.func) {
 142                         time(&cur_time);
 143                         if (timeout.when <= cur_time) {
 144                                 func = timeout.func;
 145                                 cancel_timeout();
 146                                 (*(func))();
 147                                 goto another;
 148                         }
 149                         /*
 150                          * Figure timeout in milliseconds, and check for
 151                          * potential overflow, so we can cram into an
 152                          * int for poll, while not polling with a
 153                          * negative timeout and blocking indefinitely.
 154                          */
 155                         howlong = timeout.when - cur_time;
 156                         if (howlong > INT_MAX / 1000)
 157                                 howlong = INT_MAX / 1000;
 158                         to_msec = howlong * 1000;
 159                 } else
 160                         to_msec = -1;
 161
 162                 /* Set up the descriptors to be polled. */
 163                 if (!ifi || ifi->rfdesc == -1)
 164                         error("No live interface to poll on");
 165
 166                 fds[0].fd = ifi->rfdesc;
 167                 fds[1].fd = routefd; /* Could be -1, which will be ignored. */
 168                 fds[0].events = fds[1].events = POLLIN;
 169
 170                 /* Wait for a packet or a timeout... XXX */
 171                 count = poll(fds, 2, to_msec);
 172
 173                 /* Time may have moved on while we polled! */
 174                 time(&cur_time);
 175
 176                 /* Not likely to be transitory... */
 177                 if (count == -1) {
 178                         if (errno == EAGAIN || errno == EINTR) {
 179                                 continue;
 180                         } else
 181                                 error("poll: %m");
 182                 }
 183
 184                 if ((fds[0].revents & (POLLIN | POLLHUP))) {
 185                         if (ifi && ifi->linkstat && ifi->rfdesc != -1)
 186                                 got_one();
 187                 }
 188                 if ((fds[1].revents & (POLLIN | POLLHUP))) {
 189                         if (ifi && !interfaces_invalidated)
 190                                 routehandler();
 191                 }
 192
 193                 interfaces_invalidated = 0;
 194         } while (1);
 195 }
 196
 197 void
 198 got_one(void)
 199 {
 200         struct sockaddr_in from;
 201         struct hardware hfrom;
 202         struct iaddr ifrom;
 203         ssize_t result;
 204
 205         if ((result = receive_packet(&from, &hfrom)) == -1) {
 206                 warning("receive_packet failed on %s: %s", ifi->name,
 207                     strerror(errno));
 208                 ifi->errors++;
 209                 if ((!interface_status(ifi->name)) ||
 210                     (ifi->noifmedia && ifi->errors > 20)) {
 211                         /* our interface has gone away. */
 212                         error("Interface %s no longer appears valid.",
 213                             ifi->name);
 214                 }
 215                 return;
 216         }
 217         if (result == 0)
 218                 return;
 219
 220         ifrom.len = 4;
 221         memcpy(ifrom.iabuf, &from.sin_addr, ifrom.len);
 222
 223         do_packet(result, from.sin_port, ifrom, &hfrom);
 224 }
 225
 226 int
 227 interface_link_forceup(char *ifname)
 228 {
 229         struct ifreq ifr;
 230         int sock;
 231
 232         if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
 233                 error("Can't create socket");
 234
 235         memset(&ifr, 0, sizeof(ifr));
 236         strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
 237         if (ioctl(sock, SIOCGIFFLAGS, (caddr_t)&ifr) == -1) {
 238                 close(sock);
 239                 return (-1);
 240         }
 241
 242         if ((ifr.ifr_flags & (IFF_UP|IFF_RUNNING)) != (IFF_UP|IFF_RUNNING)) {
 243                 ifr.ifr_flags |= IFF_UP;
 244                 if (ioctl(sock, SIOCSIFFLAGS, (caddr_t)&ifr) == -1) {
 245                         close(sock);
 246                         return (-1);
 247                 }
 248                 close(sock);
 249                 return (0);
 250         }
 251         close(sock);
 252         return (1);
 253 }
 254
 255 int
 256 interface_status(char *ifname)
 257 {
 258         struct ifreq ifr;
 259         struct ifmediareq ifmr;
 260         int sock;
 261
 262         if ((sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
 263                 error("Can't create socket");
 264
 265         /* get interface flags */
 266         memset(&ifr, 0, sizeof(ifr));
 267         strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
 268         if (ioctl(sock, SIOCGIFFLAGS, &ifr) == -1) {
 269                 error("ioctl(SIOCGIFFLAGS) on %s: %m", ifname);
 270         }
 271
 272         /*
 273          * if one of UP and RUNNING flags is dropped,
 274          * the interface is not active.
 275          */
 276         if ((ifr.ifr_flags & (IFF_UP|IFF_RUNNING)) != (IFF_UP|IFF_RUNNING))
 277                 goto inactive;
 278
 279         /* Next, check carrier on the interface, if possible */
 280         if (ifi->noifmedia)
 281                 goto active;
 282         memset(&ifmr, 0, sizeof(ifmr));
 283         strlcpy(ifmr.ifm_name, ifname, sizeof(ifmr.ifm_name));
 284         if (ioctl(sock, SIOCGIFMEDIA, (caddr_t)&ifmr) == -1) {
 285                 /*
 286                  * EINVAL or ENOTTY simply means that the interface
 287                  * does not support the SIOCGIFMEDIA ioctl. We regard it alive.
 288                  */
 289 #ifdef DEBUG
 290                 if (errno != EINVAL && errno != ENOTTY)
 291                         debug("ioctl(SIOCGIFMEDIA) on %s: %m", ifname);
 292 #endif
 293                 ifi->noifmedia = 1;
 294                 goto active;
 295         }
 296         if (ifmr.ifm_status & IFM_AVALID) {
 297                 if (ifmr.ifm_status & IFM_ACTIVE)
 298                         goto active;
 299                 else
 300                         goto inactive;
 301         }
 302
 303         /* Assume 'active' if IFM_AVALID is not set. */
 304
 305 active:
 306         close(sock);
 307         return (1);
 308 inactive:
 309         close(sock);
 310         return (0);
 311 }
 312
 313 void
 314 set_timeout(time_t when, void (*where)(void))
 315 {
 316         timeout.when = when;
 317         timeout.func = where;
 318 }
 319
 320 void
 321 cancel_timeout(void)
 322 {
 323         timeout.when = 0;
 324         timeout.func = NULL;
 325 }