Make the GTK menu dynamic like the QT port ... has display issues though.
[dhcpcd-ui] / src / dhcpcd-gtk / main.c
1 /*
2  * dhcpcd-gtk
3  * Copyright 2009-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 <locale.h>
29 #include <stdlib.h>
30 #include <string.h>
31
32 #ifdef NOTIFY
33 #  include <libnotify/notify.h>
34 #ifndef NOTIFY_CHECK_VERSION
35 #  define NOTIFY_CHECK_VERSION(a,b,c) 0
36 #endif
37 static NotifyNotification *nn;
38 #endif
39
40 #include "config.h"
41 #include "dhcpcd.h"
42 #include "dhcpcd-gtk.h"
43
44 static GtkStatusIcon *status_icon;
45 static guint ani_timer;
46 static int ani_counter;
47 static bool online;
48 static bool carrier;
49
50 struct watch {
51         gpointer ref;
52         int fd;
53         guint eventid;
54         GIOChannel *gio;
55         struct watch *next;
56 };
57 static struct watch *watches;
58
59 WI_SCAN *wi_scans;
60
61 static gboolean dhcpcd_try_open(gpointer data);
62 static gboolean dhcpcd_wpa_try_open(gpointer data);
63
64 WI_SCAN *
65 wi_scan_find(DHCPCD_WI_SCAN *scan)
66 {
67         WI_SCAN *w;
68         DHCPCD_WI_SCAN *dw;
69
70         for (w = wi_scans; w; w = w->next) {
71                 for (dw = w->scans; dw; dw = dw->next)
72                         if (dw == scan)
73                                 break;
74                 if (dw)
75                         return w;
76         }
77         return NULL;
78 }
79
80 static gboolean
81 animate_carrier(_unused gpointer data)
82 {
83         const char *icon;
84
85         if (ani_timer == 0)
86                 return false;
87
88         switch(ani_counter++) {
89         case 0:
90                 icon = "network-transmit";
91                 break;
92         case 1:
93                 icon = "network-receive";
94                 break;
95         default:
96                 icon = "network-idle";
97                 ani_counter = 0;
98                 break;
99         }
100         gtk_status_icon_set_from_icon_name(status_icon, icon);
101         return true;
102 }
103
104 static gboolean
105 animate_online(_unused gpointer data)
106 {
107         const char *icon;
108
109         if (ani_timer == 0)
110                 return false;
111
112         if (ani_counter++ > 6) {
113                 ani_timer = 0;
114                 ani_counter = 0;
115                 return false;
116         }
117
118         if (ani_counter % 2 == 0)
119                 icon = "network-idle";
120         else
121                 icon = "network-transmit-receive";
122         gtk_status_icon_set_from_icon_name(status_icon, icon);
123         return true;
124 }
125
126 static void
127 update_online(DHCPCD_CONNECTION *con, bool showif)
128 {
129         bool ison, iscarrier;
130         char *msg, *msgs, *tmp;
131         DHCPCD_IF *ifs, *i;
132
133         ison = iscarrier = false;
134         msgs = NULL;
135         ifs = dhcpcd_interfaces(con);
136         for (i = ifs; i; i = i->next) {
137                 if (g_strcmp0(i->type, "link") == 0) {
138                         if (i->up)
139                                 iscarrier = true;
140                 } else {
141                         if (i->up)
142                                 ison = true;
143                 }
144                 msg = dhcpcd_if_message(i, NULL);
145                 if (msg) {
146                         if (showif)
147                                 g_message("%s", msg);
148                         if (msgs) {
149                                 tmp = g_strconcat(msgs, "\n", msg, NULL);
150                                 g_free(msgs);
151                                 g_free(msg);
152                                 msgs = tmp;
153                         } else
154                                 msgs = msg;
155                 } else if (showif)
156                         g_message("%s: %s", i->ifname, i->reason);
157         }
158
159         if (online != ison || carrier != iscarrier) {
160                 online = ison;
161                 carrier = iscarrier;
162                 if (ani_timer != 0) {
163                         g_source_remove(ani_timer);
164                         ani_timer = 0;
165                         ani_counter = 0;
166                 }
167                 if (ison) {
168                         animate_online(NULL);
169                         ani_timer = g_timeout_add(300, animate_online, NULL);
170                 } else if (iscarrier) {
171                         animate_carrier(NULL);
172                         ani_timer = g_timeout_add(500, animate_carrier, NULL);
173                 } else {
174                         gtk_status_icon_set_from_icon_name(status_icon,
175                             "network-offline");
176                 }
177         }
178         gtk_status_icon_set_tooltip_text(status_icon, msgs);
179         g_free(msgs);
180 }
181
182 void
183 notify_close(void)
184 {
185 #ifdef NOTIFY
186         if (nn != NULL)
187                 notify_notification_close(nn, NULL);
188 #endif
189 }
190
191 #ifdef NOTIFY
192 static char *notify_last_msg;
193
194 static void
195 notify_closed(void)
196 {
197         nn = NULL;
198 }
199
200 static void
201 notify(const char *title, const char *msg, const char *icon)
202 {
203
204         if (msg == NULL)
205                 return;
206         /* Don't spam the same message */
207         if (notify_last_msg) {
208                 if (notify_last_msg && strcmp(msg, notify_last_msg) == 0)
209                         return;
210                 g_free(notify_last_msg);
211         }
212         notify_last_msg = g_strdup(msg);
213
214         if (nn != NULL)
215                 notify_notification_close(nn, NULL);
216
217 #if NOTIFY_CHECK_VERSION(0,7,0)
218         nn = notify_notification_new(title, msg, icon);
219         notify_notification_set_hint(nn, "transient",
220             g_variant_new_boolean(TRUE));
221 #else
222         if (gtk_status_icon_get_visible(status_icon))
223                 nn = notify_notification_new_with_status_icon(title,
224                     msg, icon, status_icon);
225         else
226                 nn = notify_notification_new(title, msg, icon, NULL);
227 #endif
228
229         notify_notification_set_timeout(nn, 5000);
230         g_signal_connect(nn, "closed", G_CALLBACK(notify_closed), NULL);
231         notify_notification_show(nn, NULL);
232 }
233 #else
234 #  define notify(a, b, c)
235 #endif
236
237 static struct watch *
238 dhcpcd_findwatch(int fd, gpointer data, struct watch **last)
239 {
240         struct watch *w;
241
242         if (last)
243                 *last = NULL;
244         for (w = watches; w; w = w->next) {
245                 if (w->fd == fd || w->ref == data)
246                         return w;
247                 if (last)
248                         *last = w;
249         }
250         return NULL;
251 }
252
253 static void
254 dhcpcd_unwatch(int fd, gpointer data)
255 {
256         struct watch *w, *l;
257
258         if ((w = dhcpcd_findwatch(fd, data, &l))) {
259                 if (l)
260                         l->next = w->next;
261                 else
262                         watches = w->next;
263                 g_source_remove(w->eventid);
264                 g_io_channel_unref(w->gio);
265                 g_free(w);
266         }
267 }
268
269 static gboolean
270 dhcpcd_watch(int fd,
271     gboolean (*cb)(GIOChannel *, GIOCondition, gpointer),
272     gpointer data)
273 {
274         struct watch *w, *l;
275         GIOChannel *gio;
276         GIOCondition flags;
277         guint eventid;
278
279         /* Sanity */
280         if ((w = dhcpcd_findwatch(fd, data, &l))) {
281                 if (w->fd == fd)
282                         return TRUE;
283                 if (l)
284                         l->next = w->next;
285                 else
286                         watches = w->next;
287                 g_source_remove(w->eventid);
288                 g_io_channel_unref(w->gio);
289                 g_free(w);
290         }
291
292         gio = g_io_channel_unix_new(fd);
293         if (gio == NULL) {
294                 g_warning(_("Error creating new GIO Channel\n"));
295                 return FALSE;
296         }
297         flags = G_IO_IN | G_IO_ERR | G_IO_HUP;
298         if ((eventid = g_io_add_watch(gio, flags, cb, data)) == 0) {
299                 g_warning(_("Error creating watch\n"));
300                 g_io_channel_unref(gio);
301                 return FALSE;
302         }
303
304         w = g_try_malloc(sizeof(*w));
305         if (w == NULL) {
306                 g_warning(_("g_try_malloc\n"));
307                 g_source_remove(eventid);
308                 g_io_channel_unref(gio);
309                 return FALSE;
310         }
311
312         w->ref = data;
313         w->fd = fd;
314         w->eventid = eventid;
315         w->gio = gio;
316         w->next = watches;
317         watches = w;
318
319         return TRUE;
320 }
321
322 static void
323 dhcpcd_status_cb(DHCPCD_CONNECTION *con, const char *status,
324     _unused void *data)
325 {
326         static char *last = NULL;
327         const char *msg;
328         bool refresh;
329         WI_SCAN *w;
330
331         g_message("Status changed to %s", status);
332         if (g_strcmp0(status, "down") == 0) {
333                 msg = N_(last ?
334                     "Connection to dhcpcd lost" : "dhcpcd not running");
335                 if (ani_timer != 0) {
336                         g_source_remove(ani_timer);
337                         ani_timer = 0;
338                         ani_counter = 0;
339                 }
340                 online = carrier = false;
341                 gtk_status_icon_set_from_icon_name(status_icon,
342                     "network-offline");
343                 gtk_status_icon_set_tooltip_text(status_icon, msg);
344                 dhcpcd_prefs_abort();
345                 dhcpcd_menu_abort();
346                 while (wi_scans) {
347                         w = wi_scans->next;
348                         dhcpcd_wi_scans_free(wi_scans->scans);
349                         g_free(wi_scans);
350                         wi_scans = w;
351                 }
352                 dhcpcd_unwatch(-1, con);
353                 g_timeout_add(DHCPCD_RETRYOPEN, dhcpcd_try_open, con);
354         } else {
355                 if ((last == NULL || g_strcmp0(last, "down") == 0)) {
356                         g_message(_("Connected to %s-%s"), "dhcpcd",
357                             dhcpcd_version(con));
358                         refresh = true;
359                 } else
360                         refresh = g_strcmp0(last, "opened") ? false : true;
361                 update_online(con, refresh);
362         }
363
364         g_free(last);
365         last = g_strdup(status);
366 }
367
368 static gboolean
369 dhcpcd_cb(_unused GIOChannel *gio, _unused GIOCondition c, gpointer data)
370 {
371         DHCPCD_CONNECTION *con;
372
373         con = (DHCPCD_CONNECTION *)data;
374         if (dhcpcd_get_fd(con) == -1) {
375                 g_warning(_("dhcpcd connection lost"));
376                 dhcpcd_unwatch(-1, con);
377                 g_timeout_add(DHCPCD_RETRYOPEN, dhcpcd_try_open, con);
378                 return FALSE;
379         }
380
381         dhcpcd_dispatch(con);
382         return TRUE;
383 }
384
385 static gboolean
386 dhcpcd_try_open(gpointer data)
387 {
388         DHCPCD_CONNECTION *con;
389         int fd;
390         static int last_error;
391
392         con = (DHCPCD_CONNECTION *)data;
393         fd = dhcpcd_open(con, true);
394         if (fd == -1) {
395                 if (errno == EACCES || errno == EPERM) {
396                         if ((fd = dhcpcd_open(con, false)) != -1)
397                                 goto unprived;
398                 }
399                 if (errno != last_error) {
400                         g_critical("dhcpcd_open: %s", strerror(errno));
401                         last_error = errno;
402                 }
403                 return TRUE;
404         }
405
406 unprived:
407         if (!dhcpcd_watch(fd, dhcpcd_cb, con)) {
408                 dhcpcd_close(con);
409                 return TRUE;
410         }
411
412         /* Start listening to WPA events */
413         dhcpcd_wpa_start(con);
414
415         return FALSE;
416 }
417
418 static void
419 dhcpcd_if_cb(DHCPCD_IF *i, _unused void *data)
420 {
421         DHCPCD_CONNECTION *con;
422         char *msg;
423         const char *icon;
424         bool new_msg;
425
426         /* We should ignore renew and stop so we don't annoy the user */
427         if (g_strcmp0(i->reason, "RENEW") &&
428             g_strcmp0(i->reason, "STOP") &&
429             g_strcmp0(i->reason, "STOPPED"))
430         {
431                 msg = dhcpcd_if_message(i, &new_msg);
432                 if (msg) {
433                         g_message("%s", msg);
434                         if (new_msg) {
435                                 if (i->up)
436                                         icon = "network-transmit-receive";
437                                 //else
438                                 //      icon = "network-transmit";
439                                 if (!i->up)
440                                         icon = "network-offline";
441                                 notify(_("Network event"), msg, icon);
442                         }
443                         g_free(msg);
444                 }
445         }
446
447         /* Update the tooltip with connection information */
448         con = dhcpcd_if_connection(i);
449         update_online(con, false);
450 }
451
452 static gboolean
453 dhcpcd_wpa_cb(_unused GIOChannel *gio, _unused GIOCondition c,
454     gpointer data)
455 {
456         DHCPCD_WPA *wpa;
457         DHCPCD_IF *i;
458
459         wpa = (DHCPCD_WPA *)data;
460         if (dhcpcd_wpa_get_fd(wpa) == -1) {
461                 dhcpcd_unwatch(-1, wpa);
462
463                 /* If the interface hasn't left, try re-opening */
464                 i = dhcpcd_wpa_if(wpa);
465                 if (i == NULL ||
466                     g_strcmp0(i->reason, "DEPARTED") == 0 ||
467                     g_strcmp0(i->reason, "STOPPED") == 0)
468                         return TRUE;
469                 g_warning(_("dhcpcd WPA connection lost: %s"), i->ifname);
470                 g_timeout_add(DHCPCD_RETRYOPEN, dhcpcd_wpa_try_open, wpa);
471                 return FALSE;
472         }
473
474         dhcpcd_wpa_dispatch(wpa);
475         return TRUE;
476 }
477
478 static gboolean
479 dhcpcd_wpa_try_open(gpointer data)
480 {
481         DHCPCD_WPA *wpa;
482         int fd;
483         static int last_error;
484
485         wpa = (DHCPCD_WPA *)data;
486         fd = dhcpcd_wpa_open(wpa);
487         if (fd == -1) {
488                 if (errno != last_error)
489                         g_critical("dhcpcd_wpa_open: %s", strerror(errno));
490                 last_error = errno;
491                 return TRUE;
492         }
493
494         if (!dhcpcd_watch(fd, dhcpcd_wpa_cb, wpa)) {
495                 dhcpcd_wpa_close(wpa);
496                 return TRUE;
497         }
498
499         return FALSE;
500 }
501
502 static void
503 dhcpcd_wpa_scan_cb(DHCPCD_WPA *wpa, _unused void *data)
504 {
505         DHCPCD_IF *i;
506         WI_SCAN *w;
507         DHCPCD_WI_SCAN *scans, *s1, *s2;
508         char *txt, *t;
509         int lerrno, fd;
510         const char *msg;
511
512         /* This could be a new WPA so watch it */
513         fd = dhcpcd_wpa_get_fd(wpa);
514         if (fd == -1) {
515                 g_critical("No fd for WPA %p", wpa);
516                 dhcpcd_unwatch(-1, wpa);
517                 return;
518         }
519         dhcpcd_watch(fd, dhcpcd_wpa_cb, wpa);
520
521         i = dhcpcd_wpa_if(wpa);
522         if (i == NULL) {
523                 g_critical("No interface for WPA %p", wpa);
524                 return;
525         }
526         g_message(_("%s: Received scan results"), i->ifname);
527         lerrno = errno;
528         errno = 0;
529         scans = dhcpcd_wi_scans(i);
530         if (scans == NULL && errno)
531                 g_warning("%s: %s", i->ifname, strerror(errno));
532         errno = lerrno;
533         for (w = wi_scans; w; w = w->next)
534                 if (w->interface == i)
535                         break;
536         if (w == NULL) {
537                 w = g_malloc(sizeof(*w));
538                 w->interface = i;
539                 w->menu = NULL;
540                 w->next = wi_scans;
541                 wi_scans = w;
542         } else {
543                 txt = NULL;
544                 msg = N_("New Access Point");
545                 for (s1 = scans; s1; s1 = s1->next) {
546                         for (s2 = w->scans; s2; s2 = s2->next)
547                                 if (g_strcmp0(s1->ssid, s2->ssid) == 0)
548                                         break;
549                         if (s2 == NULL) {
550                                 if (txt == NULL)
551                                         txt = g_strdup(s1->ssid);
552                                 else {
553                                         msg = N_("New Access Points");
554                                         t = g_strconcat(txt, "\n",
555                                             s1->ssid, NULL);
556                                         g_free(txt);
557                                         txt = t;
558                                 }
559                         }
560                 }
561                 if (txt) {
562                         notify(msg, txt, "network-wireless");
563                         g_free(txt);
564                 }
565                 menu_update_scans(w, scans);
566                 dhcpcd_wi_scans_free(w->scans);
567         }
568         w->scans = scans;
569 }
570
571 static void
572 dhcpcd_wpa_status_cb(DHCPCD_WPA *wpa, const char *status, _unused void *data)
573 {
574         DHCPCD_IF *i;
575
576         i = dhcpcd_wpa_if(wpa);
577         g_message("%s: WPA status %s", i->ifname, status);
578         if (g_strcmp0(status, "down") == 0)
579                 dhcpcd_unwatch(-1, wpa);
580 }
581
582 int
583 main(int argc, char *argv[])
584 {
585         DHCPCD_CONNECTION *con;
586
587         setlocale(LC_ALL, "");
588         bindtextdomain(PACKAGE, NULL);
589         bind_textdomain_codeset(PACKAGE, "UTF-8");
590         textdomain(PACKAGE);
591
592         gtk_init(&argc, &argv);
593         g_set_application_name("Network Configurator");
594         gtk_icon_theme_append_search_path(gtk_icon_theme_get_default(),
595             ICONDIR);
596         status_icon = gtk_status_icon_new_from_icon_name("network-offline");
597
598         gtk_status_icon_set_tooltip_text(status_icon,
599             _("Connecting to dhcpcd ..."));
600         gtk_status_icon_set_visible(status_icon, true);
601         online = false;
602 #ifdef NOTIFY
603         notify_init(PACKAGE);
604 #endif
605
606         g_message(_("Connecting ..."));
607         con = dhcpcd_new();
608         if (con ==  NULL) {
609                 g_critical("libdhcpcd: %s", strerror(errno));
610                 exit(EXIT_FAILURE);
611         }
612         dhcpcd_set_progname(con, "dhcpcd-gtk");
613         dhcpcd_set_status_callback(con, dhcpcd_status_cb, NULL);
614         dhcpcd_set_if_callback(con, dhcpcd_if_cb, NULL);
615         dhcpcd_wpa_set_scan_callback(con, dhcpcd_wpa_scan_cb, NULL);
616         dhcpcd_wpa_set_status_callback(con, dhcpcd_wpa_status_cb, NULL);
617         if (dhcpcd_try_open(con))
618                 g_timeout_add(DHCPCD_RETRYOPEN, dhcpcd_try_open, con);
619
620         menu_init(status_icon, con);
621
622         gtk_main();
623         dhcpcd_close(con);
624         dhcpcd_free(con);
625         return 0;
626 }