roy.marples.name Git - dhcpcd-ui/blob - src/dhcpcd-qt/dhcpcd-wi.cpp

   1 /*
   2  * dhcpcd-qt
   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 <QAction>
  28 #include <QObject>
  29 #include <QInputDialog>
  30 #include <QLineEdit>
  31 #include <QMenu>
  32 #include <QMessageBox>
  33 #include <QSocketNotifier>
  34 #include <QTimer>
  35 #include <QWidgetAction>
  36
  37 #include <cerrno>
  38
  39 #include "config.h"
  40 #include "dhcpcd-wi.h"
  41 #include "dhcpcd-qt.h"
  42 #include "dhcpcd-ifmenu.h"
  43 #include "dhcpcd-ssidmenu.h"
  44
  45 DhcpcdWi::DhcpcdWi(DhcpcdQt *parent, DHCPCD_WPA *wpa)
  46 {
  47
  48         this->dhcpcdQt = parent;
  49         this->wpa = wpa;
  50         menu = NULL;
  51         scans = NULL;
  52
  53         int fd = dhcpcd_wpa_get_fd(wpa);
  54         notifier = new QSocketNotifier(fd, QSocketNotifier::Read);
  55         connect(notifier, SIGNAL(activated(int)), this, SLOT(dispatch()));
  56         retryOpenTimer = NULL;
  57 }
  58
  59 DhcpcdWi::~DhcpcdWi()
  60 {
  61
  62         if (menu) {
  63                 delete menu;
  64                 menu = NULL;
  65         }
  66
  67         if (notifier) {
  68                 delete notifier;
  69                 notifier = NULL;
  70         }
  71
  72         dhcpcd_wi_scans_free(scans);
  73 }
  74
  75 DHCPCD_WPA *DhcpcdWi::getWpa()
  76 {
  77
  78         return wpa;
  79 }
  80
  81 DHCPCD_WI_SCAN *DhcpcdWi::getScans()
  82 {
  83
  84         return scans;
  85 }
  86
  87 bool DhcpcdWi::setScans(DHCPCD_WI_SCAN *scans)
  88 {
  89         int changed = 0;
  90
  91         if (menu) {
  92                 QList<DhcpcdSsidMenu*> lst;
  93                 DHCPCD_WI_SCAN *scan;
  94
  95                 lst = menu->findChildren<DhcpcdSsidMenu*>();
  96                 for (scan = scans; scan; scan = scan->next) {
  97                         bool found = false;
  98
  99                         foreach(DhcpcdSsidMenu *sm, lst) {
 100                                 DHCPCD_WI_SCAN *s = sm->getScan();
 101                                 if (memcmp(scan->bssid, s->bssid,
 102                                     sizeof(scan->bssid)) == 0)
 103                                 {
 104                                         sm->setScan(scan);
 105                                         found = true;
 106                                         break;
 107                                 }
 108                         }
 109
 110                         if (!found) {
 111                                 createMenuItem(menu, scan);
 112                                 changed++;
 113                         }
 114                 }
 115
 116                 foreach(DhcpcdSsidMenu *sm, lst) {
 117                         DHCPCD_WI_SCAN *s = sm->getScan();
 118                         for (scan = scans; scan; scan = scan->next) {
 119                                 if (memcmp(scan->bssid, s->bssid,
 120                                     sizeof(scan->bssid)) == 0)
 121                                         break;
 122                         }
 123                         if (scan == NULL) {
 124                                 menu->removeAction(sm);
 125                                 changed--;
 126                         }
 127                 }
 128         }
 129
 130         dhcpcd_wi_scans_free(this->scans);
 131         this->scans = scans;
 132
 133         return !(changed == 0);
 134 }
 135
 136 void DhcpcdWi::createMenuItem(QMenu *menu, DHCPCD_WI_SCAN *scan)
 137 {
 138         DhcpcdSsidMenu *ssidMenu = new DhcpcdSsidMenu(menu, this, scan);
 139         menu->addAction(ssidMenu);
 140         connect(ssidMenu, SIGNAL(triggered(DHCPCD_WI_SCAN *)),
 141             this, SLOT(connectSsid(DHCPCD_WI_SCAN *)));
 142 }
 143
 144 void DhcpcdWi::createMenu1(QMenu *menu)
 145 {
 146         DHCPCD_WI_SCAN *scan;
 147
 148         for (scan = scans; scan; scan = scan->next)
 149                 createMenuItem(menu, scan);
 150 }
 151
 152 void DhcpcdWi::createMenu(QMenu *menu)
 153 {
 154
 155         this->menu = menu;
 156         createMenu1(menu);
 157 }
 158
 159 QMenu *DhcpcdWi::createIfMenu(QMenu *parent)
 160 {
 161         DHCPCD_IF *ifp;
 162         QIcon icon;
 163
 164         ifp = dhcpcd_wpa_if(wpa);
 165         menu = new DhcpcdIfMenu(ifp, parent);
 166         icon = DhcpcdQt::getIcon("devices", "network-wireless");
 167         menu->setIcon(icon);
 168         createMenu1(menu);
 169         return menu;
 170 }
 171
 172 void DhcpcdWi::wpaOpen()
 173 {
 174         int fd = dhcpcd_wpa_open(wpa);
 175         static int last_error;
 176
 177         if (fd == -1) {
 178                 if (errno != last_error) {
 179                         last_error = errno;
 180                         qCritical("%s: dhcpcd_wpa_open: %s",
 181                             dhcpcd_wpa_if(wpa)->ifname,
 182                             strerror(last_error));
 183                 }
 184                 return;
 185         }
 186
 187         notifier = new QSocketNotifier(fd, QSocketNotifier::Read);
 188         connect(notifier, SIGNAL(activated(int)), this, SLOT(dispatch()));
 189         if (retryOpenTimer) {
 190                 delete retryOpenTimer;
 191                 retryOpenTimer = NULL;
 192         }
 193 }
 194
 195 void DhcpcdWi::dispatch()
 196 {
 197
 198         if (dhcpcd_wpa_get_fd(wpa) == -1) {
 199                 delete notifier;
 200                 notifier = NULL;
 201                 DHCPCD_IF *i = dhcpcd_wpa_if(wpa);
 202                 if (i == NULL ||
 203                     strcmp(i->reason, "DEPARTED") == 0 ||
 204                     strcmp(i->reason, "STOPPED") == 0)
 205                         return;
 206                 qWarning("%s: %s",
 207                     i->ifname,
 208                     qPrintable(tr("dhcpcd WPA connection lost")));
 209                 if (retryOpenTimer == NULL) {
 210                         retryOpenTimer = new QTimer(this);
 211                         connect(retryOpenTimer, SIGNAL(timeout()),
 212                             this, SLOT(wpaOpen()));
 213                         retryOpenTimer->start(DHCPCD_RETRYOPEN);
 214                 }
 215                 return;
 216         }
 217
 218         dhcpcd_wpa_dispatch(wpa);
 219 }
 220
 221 void DhcpcdWi::connectSsid(DHCPCD_WI_SCAN *scan)
 222 {
 223         bool ok;
 224         DHCPCD_WI_SCAN s;
 225
 226         /* Take a copy of scan incase it's destroyed by a scan update */
 227         memcpy(&s, scan, sizeof(s));
 228         s.next = NULL;
 229
 230         QString psk = QInputDialog::getText(dhcpcdQt, s.ssid,
 231             tr("Pre Shared key"), QLineEdit::Normal, NULL, &ok);
 232
 233         if (!ok)
 234                 return;
 235
 236         QString errt;
 237
 238         switch (dhcpcd_wpa_configure_psk(wpa, &s, psk.toAscii())) {
 239         case DHCPCD_WPA_SUCCESS:
 240                 return;
 241         case DHCPCD_WPA_ERR_SET:
 242                 errt = tr("Failed to set key management.");
 243                 break;
 244         case DHCPCD_WPA_ERR_SET_PSK:
 245                 errt = tr("Failed to set password, probably too short.");
 246                 break;
 247         case DHCPCD_WPA_ERR_ENABLE:
 248                 errt = tr("Failed to enable the network.");
 249                 break;
 250         case DHCPCD_WPA_ERR_ASSOC:
 251                 errt = tr("Failed to start association.");
 252                 break;
 253         case DHCPCD_WPA_ERR_WRITE:
 254                 errt = tr("Failed to save wpa_supplicant configuration.\n\nYou should add update_config=1 to /etc/wpa_supplicant.conf.");
 255                 break;
 256         default:
 257                 errt = strerror(errno);
 258                 break;
 259         }
 260
 261         QMessageBox::critical(dhcpcdQt, tr("Error setting wireless properties"),
 262             errt);
 263 }