roy.marples.name Git - dhcpcd-ui/blob - src/dhcpcd-online/dhcpcd-online.c

   1 /*
   2  * dhcpcd-online
   3  * Copyright 2014 Roy Marples <roy@marples.name>
   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  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include <errno.h>
  28 #include <limits.h>
  29 #include <poll.h>
  30 #include <stdbool.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <syslog.h>
  35 #include <time.h>
  36 #include <unistd.h>
  37
  38 #include "dhcpcd.h"
  39
  40 #if __GNUC__ > 2 || defined(__INTEL_COMPILER)
  41 # ifndef __dead
  42 #  define __dead __attribute__((__noreturn__))
  43 # endif
  44 # ifndef __unused
  45 #  define __unused   __attribute__((__unused__))
  46 # endif
  47 #else
  48 # ifndef __dead
  49 #  define __dead
  50 # endif
  51 # ifndef __unused
  52 #  define __unused
  53 # endif
  54 #endif
  55
  56 static void __dead
  57 do_exit(DHCPCD_CONNECTION *con, int code)
  58 {
  59
  60         dhcpcd_close(con);
  61         dhcpcd_free(con);
  62         exit(code);
  63 }
  64
  65 static void
  66 do_status_cb(DHCPCD_CONNECTION *con, const char *status, void __unused *arg)
  67 {
  68
  69         syslog(LOG_INFO, "%s", status);
  70         if (strcmp(status, "connected") == 0)
  71                 do_exit(con, EXIT_SUCCESS);
  72 }
  73
  74 int
  75 main(int argc, char **argv)
  76 {
  77         DHCPCD_CONNECTION *con;
  78         bool xflag;
  79         struct timespec now, end, t;
  80         struct pollfd pfd;
  81         int timeout, n, lerrno;
  82         long lnum;
  83         char *lend;
  84
  85         /* Defaults */
  86         timeout = 30;
  87
  88         xflag = false;
  89
  90         openlog("dhcpcd-online", LOG_PERROR, 0);
  91         setlogmask(LOG_UPTO(LOG_INFO));
  92
  93         while ((n = getopt(argc, argv, "qt:x")) != -1) {
  94                 switch (n) {
  95                 case 'q':
  96                         closelog();
  97                         openlog("dhcpcd-online", 0, 0);
  98                         break;
  99                 case 't':
 100                         lnum = strtol(optarg, &lend, 0);
 101                         if (lend == NULL || *lend != '\0' ||
 102                             lnum < 0 || lnum > INT_MAX)
 103                         {
 104                                 syslog(LOG_ERR, "-t %s: invalid timeout",
 105                                     optarg);
 106                                 exit(EXIT_FAILURE);
 107                         }
 108                         timeout = (int)lnum;
 109                         break;
 110                 case 'x':
 111                         xflag = true;
 112                         break;
 113                 case '?':
 114                         fprintf(stderr, "usage: dhcpcd-online "
 115                             "[-q] [-t timeout]\n");
 116                         exit(EXIT_FAILURE);
 117                 }
 118         }
 119
 120         if ((con = dhcpcd_new()) == NULL) {
 121                 syslog(LOG_ERR, "dhcpcd_new: %m");
 122                 return EXIT_FAILURE;
 123         }
 124
 125         dhcpcd_set_status_callback(con, do_status_cb, NULL);
 126
 127         if ((pfd.fd = dhcpcd_open(con)) == -1) {
 128                 lerrno = errno;
 129                 syslog(LOG_WARNING, "dhcpcd_open: %m");
 130                 if (xflag)
 131                         do_exit(con, EXIT_FAILURE);
 132         } else
 133                 lerrno = 0;
 134         pfd.events = POLLIN;
 135         pfd.revents = 0;
 136
 137         /* Work out our timeout time */
 138         if (clock_gettime(CLOCK_MONOTONIC, &end) == -1) {
 139                 syslog(LOG_ERR, "clock_gettime: %m");
 140                 do_exit(con, EXIT_FAILURE);
 141         }
 142         end.tv_sec += timeout;
 143
 144         for (;;) {
 145                 if (clock_gettime(CLOCK_MONOTONIC, &now) == -1) {
 146                         syslog(LOG_ERR, "clock_gettime: %m");
 147                         do_exit(con, EXIT_FAILURE);
 148                 }
 149                 if (timespeccmp(&now, &end, >)) {
 150                         syslog(LOG_ERR, "timed out");
 151                         do_exit(con, EXIT_FAILURE);
 152                 }
 153                 if (pfd.fd == -1) {
 154                         n = poll(NULL, 0, DHCPCD_RETRYOPEN);
 155                 } else {
 156                         /* poll(2) should really take a timespec */
 157                         timespecsub(&end, &now, &t);
 158                         if (t.tv_sec > INT_MAX / 1000 ||
 159                             (t.tv_sec == INT_MAX / 1000 &&
 160                             (t.tv_nsec + 999999) / 1000000 > INT_MAX % 1000000))
 161                                 timeout = INT_MAX;
 162                         else
 163                                 timeout = (int)t.tv_sec * 1000 +
 164                                     (t.tv_nsec + 999999) / 1000000;
 165                         n = poll(&pfd, 1, timeout);
 166                 }
 167                 if (n == -1) {
 168                         syslog(LOG_ERR, "poll: %m");
 169                         do_exit(con, EXIT_FAILURE);
 170                 }
 171                 if (pfd.fd == -1) {
 172                         if ((pfd.fd = dhcpcd_open(con)) == -1) {
 173                                 if (lerrno != errno) {
 174                                         lerrno = errno;
 175                                         syslog(LOG_WARNING, "dhcpcd_open: %m");
 176                                 }
 177                         }
 178                 } else {
 179                         if (n > 0 && pfd.revents)
 180                                 dhcpcd_dispatch(con);
 181                 }
 182         }
 183
 184         /* Impossible to reach here */
 185         do_exit(con, EXIT_FAILURE);
 186 }