roy.marples.name Git - dhcpcd-ui/blob - 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 }