21d5ca08fa988537300424252066b1c6bf9f7232
[dhcpcd-ui] / 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 }