334960ca4ba98b1464cf49972ebdddeb254a9ac9
[dhcpcd-ui] / 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 }