1 /*
2 * wpa_gui - Peers class
3 * Copyright (c) 2009-2010, Atheros Communications
4 *
5 * This software may be distributed under the terms of the BSD license.
6 * See README for more details.
7 */
8
9 #include <cstdio>
10 #include <QImageReader>
11 #include <QMessageBox>
12
13 #include "common/wpa_ctrl.h"
14 #include "wpagui.h"
15 #include "stringquery.h"
16 #include "peers.h"
17
18
19 enum {
20 peer_role_address = Qt::UserRole + 1,
21 peer_role_type,
22 peer_role_uuid,
23 peer_role_details,
24 peer_role_ifname,
25 peer_role_pri_dev_type,
26 peer_role_ssid,
27 peer_role_config_methods,
28 peer_role_dev_passwd_id,
29 peer_role_bss_id,
30 peer_role_selected_method,
31 peer_role_selected_pin,
32 peer_role_requested_method,
33 peer_role_network_id
34 };
35
36 enum selected_method {
37 SEL_METHOD_NONE,
38 SEL_METHOD_PIN_PEER_DISPLAY,
39 SEL_METHOD_PIN_LOCAL_DISPLAY
40 };
41
42 /*
43 * TODO:
44 * - add current AP info (e.g., from WPS) in station mode
45 */
46
47 enum peer_type {
48 PEER_TYPE_ASSOCIATED_STATION,
49 PEER_TYPE_AP,
50 PEER_TYPE_AP_WPS,
51 PEER_TYPE_WPS_PIN_NEEDED,
52 PEER_TYPE_P2P,
53 PEER_TYPE_P2P_CLIENT,
54 PEER_TYPE_P2P_GROUP,
55 PEER_TYPE_P2P_PERSISTENT_GROUP_GO,
56 PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT,
57 PEER_TYPE_P2P_INVITATION,
58 PEER_TYPE_WPS_ER_AP,
59 PEER_TYPE_WPS_ER_AP_UNCONFIGURED,
60 PEER_TYPE_WPS_ER_ENROLLEE,
61 PEER_TYPE_WPS_ENROLLEE
62 };
63
64
Peers(QWidget * parent,const char *,bool,Qt::WindowFlags)65 Peers::Peers(QWidget *parent, const char *, bool, Qt::WindowFlags)
66 : QDialog(parent)
67 {
68 setupUi(this);
69
70 if (QImageReader::supportedImageFormats().contains(QByteArray("svg")))
71 {
72 default_icon = new QIcon(":/icons/wpa_gui.svg");
73 ap_icon = new QIcon(":/icons/ap.svg");
74 laptop_icon = new QIcon(":/icons/laptop.svg");
75 group_icon = new QIcon(":/icons/group.svg");
76 invitation_icon = new QIcon(":/icons/invitation.svg");
77 } else {
78 default_icon = new QIcon(":/icons/wpa_gui.png");
79 ap_icon = new QIcon(":/icons/ap.png");
80 laptop_icon = new QIcon(":/icons/laptop.png");
81 group_icon = new QIcon(":/icons/group.png");
82 invitation_icon = new QIcon(":/icons/invitation.png");
83 }
84
85 peers->setModel(&model);
86 peers->setResizeMode(QListView::Adjust);
87 peers->setDragEnabled(false);
88 peers->setSelectionMode(QAbstractItemView::NoSelection);
89
90 peers->setContextMenuPolicy(Qt::CustomContextMenu);
91 connect(peers, SIGNAL(customContextMenuRequested(const QPoint &)),
92 this, SLOT(context_menu(const QPoint &)));
93
94 wpagui = NULL;
95 hide_ap = false;
96 }
97
98
setWpaGui(WpaGui * _wpagui)99 void Peers::setWpaGui(WpaGui *_wpagui)
100 {
101 wpagui = _wpagui;
102 update_peers();
103 }
104
105
~Peers()106 Peers::~Peers()
107 {
108 delete default_icon;
109 delete ap_icon;
110 delete laptop_icon;
111 delete group_icon;
112 delete invitation_icon;
113 }
114
115
languageChange()116 void Peers::languageChange()
117 {
118 retranslateUi(this);
119 }
120
121
ItemType(int type)122 QString Peers::ItemType(int type)
123 {
124 QString title;
125 switch (type) {
126 case PEER_TYPE_ASSOCIATED_STATION:
127 title = tr("Associated station");
128 break;
129 case PEER_TYPE_AP:
130 title = tr("AP");
131 break;
132 case PEER_TYPE_AP_WPS:
133 title = tr("WPS AP");
134 break;
135 case PEER_TYPE_WPS_PIN_NEEDED:
136 title = tr("WPS PIN needed");
137 break;
138 case PEER_TYPE_P2P:
139 title = tr("P2P Device");
140 break;
141 case PEER_TYPE_P2P_CLIENT:
142 title = tr("P2P Device (group client)");
143 break;
144 case PEER_TYPE_P2P_GROUP:
145 title = tr("P2P Group");
146 break;
147 case PEER_TYPE_P2P_PERSISTENT_GROUP_GO:
148 title = tr("P2P Persistent Group (GO)");
149 break;
150 case PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT:
151 title = tr("P2P Persistent Group (client)");
152 break;
153 case PEER_TYPE_P2P_INVITATION:
154 title = tr("P2P Invitation");
155 break;
156 case PEER_TYPE_WPS_ER_AP:
157 title = tr("ER: WPS AP");
158 break;
159 case PEER_TYPE_WPS_ER_AP_UNCONFIGURED:
160 title = tr("ER: WPS AP (Unconfigured)");
161 break;
162 case PEER_TYPE_WPS_ER_ENROLLEE:
163 title = tr("ER: WPS Enrollee");
164 break;
165 case PEER_TYPE_WPS_ENROLLEE:
166 title = tr("WPS Enrollee");
167 break;
168 }
169 return title;
170 }
171
172
context_menu(const QPoint & pos)173 void Peers::context_menu(const QPoint &pos)
174 {
175 QMenu *menu = new QMenu;
176 if (menu == NULL)
177 return;
178
179 QModelIndex idx = peers->indexAt(pos);
180 if (idx.isValid()) {
181 ctx_item = model.itemFromIndex(idx);
182 int type = ctx_item->data(peer_role_type).toInt();
183 menu->addAction(Peers::ItemType(type))->setEnabled(false);
184 menu->addSeparator();
185
186 int config_methods = -1;
187 QVariant var = ctx_item->data(peer_role_config_methods);
188 if (var.isValid())
189 config_methods = var.toInt();
190
191 enum selected_method method = SEL_METHOD_NONE;
192 var = ctx_item->data(peer_role_selected_method);
193 if (var.isValid())
194 method = (enum selected_method) var.toInt();
195
196 if ((type == PEER_TYPE_ASSOCIATED_STATION ||
197 type == PEER_TYPE_AP_WPS ||
198 type == PEER_TYPE_WPS_PIN_NEEDED ||
199 type == PEER_TYPE_WPS_ER_ENROLLEE ||
200 type == PEER_TYPE_WPS_ENROLLEE) &&
201 (config_methods == -1 || (config_methods & 0x010c))) {
202 menu->addAction(tr("Enter WPS PIN"), this,
203 SLOT(enter_pin()));
204 }
205
206 if (type == PEER_TYPE_P2P || type == PEER_TYPE_P2P_CLIENT) {
207 menu->addAction(tr("P2P Connect"), this,
208 SLOT(ctx_p2p_connect()));
209 if (method == SEL_METHOD_NONE &&
210 config_methods > -1 &&
211 config_methods & 0x0080 /* PBC */ &&
212 config_methods != 0x0080)
213 menu->addAction(tr("P2P Connect (PBC)"), this,
214 SLOT(connect_pbc()));
215 if (method == SEL_METHOD_NONE) {
216 menu->addAction(tr("P2P Request PIN"), this,
217 SLOT(ctx_p2p_req_pin()));
218 menu->addAction(tr("P2P Show PIN"), this,
219 SLOT(ctx_p2p_show_pin()));
220 }
221
222 if (config_methods > -1 && (config_methods & 0x0100)) {
223 /* Peer has Keypad */
224 menu->addAction(tr("P2P Display PIN"), this,
225 SLOT(ctx_p2p_display_pin()));
226 }
227
228 if (config_methods > -1 && (config_methods & 0x000c)) {
229 /* Peer has Label or Display */
230 menu->addAction(tr("P2P Enter PIN"), this,
231 SLOT(ctx_p2p_enter_pin()));
232 }
233 }
234
235 if (type == PEER_TYPE_P2P_GROUP) {
236 menu->addAction(tr("Show passphrase"), this,
237 SLOT(ctx_p2p_show_passphrase()));
238 menu->addAction(tr("Remove P2P Group"), this,
239 SLOT(ctx_p2p_remove_group()));
240 }
241
242 if (type == PEER_TYPE_P2P_PERSISTENT_GROUP_GO ||
243 type == PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT ||
244 type == PEER_TYPE_P2P_INVITATION) {
245 menu->addAction(tr("Start group"), this,
246 SLOT(ctx_p2p_start_persistent()));
247 }
248
249 if (type == PEER_TYPE_P2P_PERSISTENT_GROUP_GO ||
250 type == PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT) {
251 menu->addAction(tr("Invite"), this,
252 SLOT(ctx_p2p_invite()));
253 }
254
255 if (type == PEER_TYPE_P2P_INVITATION) {
256 menu->addAction(tr("Ignore"), this,
257 SLOT(ctx_p2p_delete()));
258 }
259
260 if (type == PEER_TYPE_AP_WPS) {
261 menu->addAction(tr("Connect (PBC)"), this,
262 SLOT(connect_pbc()));
263 }
264
265 if ((type == PEER_TYPE_ASSOCIATED_STATION ||
266 type == PEER_TYPE_WPS_ER_ENROLLEE ||
267 type == PEER_TYPE_WPS_ENROLLEE) &&
268 config_methods >= 0 && (config_methods & 0x0080)) {
269 menu->addAction(tr("Enroll (PBC)"), this,
270 SLOT(connect_pbc()));
271 }
272
273 if (type == PEER_TYPE_WPS_ER_AP) {
274 menu->addAction(tr("Learn Configuration"), this,
275 SLOT(learn_ap_config()));
276 }
277
278 menu->addAction(tr("Properties"), this, SLOT(properties()));
279 } else {
280 ctx_item = NULL;
281 menu->addAction(QString(tr("Refresh")), this,
282 SLOT(ctx_refresh()));
283 menu->addAction(tr("Start P2P discovery"), this,
284 SLOT(ctx_p2p_start()));
285 menu->addAction(tr("Stop P2P discovery"), this,
286 SLOT(ctx_p2p_stop()));
287 menu->addAction(tr("P2P listen only"), this,
288 SLOT(ctx_p2p_listen()));
289 menu->addAction(tr("Start P2P group"), this,
290 SLOT(ctx_p2p_start_group()));
291 if (hide_ap)
292 menu->addAction(tr("Show AP entries"), this,
293 SLOT(ctx_show_ap()));
294 else
295 menu->addAction(tr("Hide AP entries"), this,
296 SLOT(ctx_hide_ap()));
297 }
298
299 menu->exec(peers->mapToGlobal(pos));
300 }
301
302
enter_pin()303 void Peers::enter_pin()
304 {
305 if (ctx_item == NULL)
306 return;
307
308 int peer_type = ctx_item->data(peer_role_type).toInt();
309 QString uuid;
310 QString addr;
311 addr = ctx_item->data(peer_role_address).toString();
312 if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE)
313 uuid = ctx_item->data(peer_role_uuid).toString();
314
315 StringQuery input(tr("PIN:"));
316 input.setWindowTitle(tr("PIN for ") + ctx_item->text());
317 if (input.exec() != QDialog::Accepted)
318 return;
319
320 char cmd[100];
321 char reply[100];
322 size_t reply_len;
323
324 if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE) {
325 snprintf(cmd, sizeof(cmd), "WPS_ER_PIN %s %s %s",
326 uuid.toLocal8Bit().constData(),
327 input.get_string().toLocal8Bit().constData(),
328 addr.toLocal8Bit().constData());
329 } else {
330 snprintf(cmd, sizeof(cmd), "WPS_PIN %s %s",
331 addr.toLocal8Bit().constData(),
332 input.get_string().toLocal8Bit().constData());
333 }
334 reply_len = sizeof(reply) - 1;
335 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
336 QMessageBox msg;
337 msg.setIcon(QMessageBox::Warning);
338 msg.setText(tr("Failed to set the WPS PIN."));
339 msg.exec();
340 }
341 }
342
343
ctx_refresh()344 void Peers::ctx_refresh()
345 {
346 update_peers();
347 }
348
349
ctx_p2p_start()350 void Peers::ctx_p2p_start()
351 {
352 char reply[20];
353 size_t reply_len;
354 reply_len = sizeof(reply) - 1;
355 if (wpagui->ctrlRequest("P2P_FIND", reply, &reply_len) < 0 ||
356 memcmp(reply, "FAIL", 4) == 0) {
357 QMessageBox msg;
358 msg.setIcon(QMessageBox::Warning);
359 msg.setText("Failed to start P2P discovery.");
360 msg.exec();
361 }
362 }
363
364
ctx_p2p_stop()365 void Peers::ctx_p2p_stop()
366 {
367 char reply[20];
368 size_t reply_len;
369 reply_len = sizeof(reply) - 1;
370 wpagui->ctrlRequest("P2P_STOP_FIND", reply, &reply_len);
371 }
372
373
ctx_p2p_listen()374 void Peers::ctx_p2p_listen()
375 {
376 char reply[20];
377 size_t reply_len;
378 reply_len = sizeof(reply) - 1;
379 if (wpagui->ctrlRequest("P2P_LISTEN 3600", reply, &reply_len) < 0 ||
380 memcmp(reply, "FAIL", 4) == 0) {
381 QMessageBox msg;
382 msg.setIcon(QMessageBox::Warning);
383 msg.setText("Failed to start P2P listen.");
384 msg.exec();
385 }
386 }
387
388
ctx_p2p_start_group()389 void Peers::ctx_p2p_start_group()
390 {
391 char reply[20];
392 size_t reply_len;
393 reply_len = sizeof(reply) - 1;
394 if (wpagui->ctrlRequest("P2P_GROUP_ADD", reply, &reply_len) < 0 ||
395 memcmp(reply, "FAIL", 4) == 0) {
396 QMessageBox msg;
397 msg.setIcon(QMessageBox::Warning);
398 msg.setText("Failed to start P2P group.");
399 msg.exec();
400 }
401 }
402
403
add_station(QString info)404 void Peers::add_station(QString info)
405 {
406 QStringList lines = info.split(QRegExp("\\n"));
407 QString name;
408
409 for (QStringList::Iterator it = lines.begin();
410 it != lines.end(); it++) {
411 int pos = (*it).indexOf('=') + 1;
412 if (pos < 1)
413 continue;
414
415 if ((*it).startsWith("wpsDeviceName="))
416 name = (*it).mid(pos);
417 else if ((*it).startsWith("p2p_device_name="))
418 name = (*it).mid(pos);
419 }
420
421 if (name.isEmpty())
422 name = lines[0];
423
424 QStandardItem *item = new QStandardItem(*laptop_icon, name);
425 if (item) {
426 /* Remove WPS enrollee entry if one is still pending */
427 if (model.rowCount() > 0) {
428 QModelIndexList lst = model.match(model.index(0, 0),
429 peer_role_address,
430 lines[0]);
431 for (int i = 0; i < lst.size(); i++) {
432 QStandardItem *item;
433 item = model.itemFromIndex(lst[i]);
434 if (item == NULL)
435 continue;
436 int type = item->data(peer_role_type).toInt();
437 if (type == PEER_TYPE_WPS_ENROLLEE) {
438 model.removeRow(lst[i].row());
439 break;
440 }
441 }
442 }
443
444 item->setData(lines[0], peer_role_address);
445 item->setData(PEER_TYPE_ASSOCIATED_STATION,
446 peer_role_type);
447 item->setData(info, peer_role_details);
448 item->setToolTip(ItemType(PEER_TYPE_ASSOCIATED_STATION));
449 model.appendRow(item);
450 }
451 }
452
453
add_stations()454 void Peers::add_stations()
455 {
456 char reply[2048];
457 size_t reply_len;
458 char cmd[30];
459 int res;
460
461 reply_len = sizeof(reply) - 1;
462 if (wpagui->ctrlRequest("STA-FIRST", reply, &reply_len) < 0)
463 return;
464
465 do {
466 reply[reply_len] = '\0';
467 QString info(reply);
468 char *txt = reply;
469 while (*txt != '\0' && *txt != '\n')
470 txt++;
471 *txt++ = '\0';
472 if (strncmp(reply, "FAIL", 4) == 0 ||
473 strncmp(reply, "UNKNOWN", 7) == 0)
474 break;
475
476 add_station(info);
477
478 reply_len = sizeof(reply) - 1;
479 res = snprintf(cmd, sizeof(cmd), "STA-NEXT %s", reply);
480 if (res < 0 || (size_t) res >= sizeof(cmd))
481 break;
482 res = wpagui->ctrlRequest(cmd, reply, &reply_len);
483 } while (res >= 0);
484 }
485
486
add_single_station(const char * addr)487 void Peers::add_single_station(const char *addr)
488 {
489 char reply[2048];
490 size_t reply_len;
491 char cmd[30];
492
493 reply_len = sizeof(reply) - 1;
494 snprintf(cmd, sizeof(cmd), "STA %s", addr);
495 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
496 return;
497
498 reply[reply_len] = '\0';
499 QString info(reply);
500 char *txt = reply;
501 while (*txt != '\0' && *txt != '\n')
502 txt++;
503 *txt++ = '\0';
504 if (strncmp(reply, "FAIL", 4) == 0 ||
505 strncmp(reply, "UNKNOWN", 7) == 0)
506 return;
507
508 add_station(info);
509 }
510
511
add_p2p_group_client(QStandardItem *,QString params)512 void Peers::add_p2p_group_client(QStandardItem * /*parent*/, QString params)
513 {
514 /*
515 * dev=02:b5:64:63:30:63 iface=02:b5:64:63:30:63 dev_capab=0x0
516 * dev_type=1-0050f204-1 dev_name='Wireless Client'
517 * config_methods=0x8c
518 */
519
520 QStringList items =
521 params.split(QRegExp(" (?=[^']*('[^']*'[^']*)*$)"));
522 QString addr = "";
523 QString name = "";
524 int config_methods = 0;
525 QString dev_type;
526
527 for (int i = 0; i < items.size(); i++) {
528 QString str = items.at(i);
529 int pos = str.indexOf('=') + 1;
530 if (str.startsWith("dev_name='"))
531 name = str.section('\'', 1, -2);
532 else if (str.startsWith("config_methods="))
533 config_methods =
534 str.section('=', 1).toInt(0, 0);
535 else if (str.startsWith("dev="))
536 addr = str.mid(pos);
537 else if (str.startsWith("dev_type=") && dev_type.isEmpty())
538 dev_type = str.mid(pos);
539 }
540
541 QStandardItem *item = find_addr(addr);
542 if (item)
543 return;
544
545 item = new QStandardItem(*default_icon, name);
546 if (item) {
547 /* TODO: indicate somehow the relationship to the group owner
548 * (parent) */
549 item->setData(addr, peer_role_address);
550 item->setData(config_methods, peer_role_config_methods);
551 item->setData(PEER_TYPE_P2P_CLIENT, peer_role_type);
552 if (!dev_type.isEmpty())
553 item->setData(dev_type, peer_role_pri_dev_type);
554 item->setData(items.join(QString("\n")), peer_role_details);
555 item->setToolTip(ItemType(PEER_TYPE_P2P_CLIENT));
556 model.appendRow(item);
557 }
558 }
559
560
remove_bss(int id)561 void Peers::remove_bss(int id)
562 {
563 if (model.rowCount() == 0)
564 return;
565
566 QModelIndexList lst = model.match(model.index(0, 0), peer_role_bss_id,
567 id);
568 if (lst.size() == 0)
569 return;
570 model.removeRow(lst[0].row());
571 }
572
573
add_bss(const char * cmd)574 bool Peers::add_bss(const char *cmd)
575 {
576 char reply[2048];
577 size_t reply_len;
578
579 if (hide_ap)
580 return false;
581
582 reply_len = sizeof(reply) - 1;
583 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
584 return false;
585 reply[reply_len] = '\0';
586
587 QString bss(reply);
588 if (bss.isEmpty() || bss.startsWith("FAIL"))
589 return false;
590
591 QString ssid, bssid, flags, wps_name, pri_dev_type;
592 int id = -1;
593
594 QStringList lines = bss.split(QRegExp("\\n"));
595 for (QStringList::Iterator it = lines.begin();
596 it != lines.end(); it++) {
597 int pos = (*it).indexOf('=') + 1;
598 if (pos < 1)
599 continue;
600
601 if ((*it).startsWith("bssid="))
602 bssid = (*it).mid(pos);
603 else if ((*it).startsWith("id="))
604 id = (*it).mid(pos).toInt();
605 else if ((*it).startsWith("flags="))
606 flags = (*it).mid(pos);
607 else if ((*it).startsWith("ssid="))
608 ssid = (*it).mid(pos);
609 else if ((*it).startsWith("wps_device_name="))
610 wps_name = (*it).mid(pos);
611 else if ((*it).startsWith("wps_primary_device_type="))
612 pri_dev_type = (*it).mid(pos);
613 }
614
615 QString name = wps_name;
616 if (name.isEmpty())
617 name = ssid + "\n" + bssid;
618
619 QStandardItem *item = new QStandardItem(*ap_icon, name);
620 if (item) {
621 item->setData(bssid, peer_role_address);
622 if (id >= 0)
623 item->setData(id, peer_role_bss_id);
624 int type;
625 if (flags.contains("[WPS"))
626 type = PEER_TYPE_AP_WPS;
627 else
628 type = PEER_TYPE_AP;
629 item->setData(type, peer_role_type);
630
631 for (int i = 0; i < lines.size(); i++) {
632 if (lines[i].length() > 60) {
633 lines[i].remove(60, lines[i].length());
634 lines[i] += "..";
635 }
636 }
637 item->setToolTip(ItemType(type));
638 item->setData(lines.join("\n"), peer_role_details);
639 if (!pri_dev_type.isEmpty())
640 item->setData(pri_dev_type,
641 peer_role_pri_dev_type);
642 if (!ssid.isEmpty())
643 item->setData(ssid, peer_role_ssid);
644 model.appendRow(item);
645
646 lines = bss.split(QRegExp("\\n"));
647 for (QStringList::Iterator it = lines.begin();
648 it != lines.end(); it++) {
649 if ((*it).startsWith("p2p_group_client:"))
650 add_p2p_group_client(item,
651 (*it).mid(18));
652 }
653 }
654
655 return true;
656 }
657
658
add_scan_results()659 void Peers::add_scan_results()
660 {
661 int index;
662 char cmd[20];
663
664 index = 0;
665 while (wpagui) {
666 snprintf(cmd, sizeof(cmd), "BSS %d", index++);
667 if (index > 1000)
668 break;
669
670 if (!add_bss(cmd))
671 break;
672 }
673 }
674
675
add_persistent(int id,const char * ssid,const char * bssid)676 void Peers::add_persistent(int id, const char *ssid, const char *bssid)
677 {
678 char cmd[100];
679 char reply[100];
680 size_t reply_len;
681 int mode;
682
683 snprintf(cmd, sizeof(cmd), "GET_NETWORK %d mode", id);
684 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
685 return;
686 reply[reply_len] = '\0';
687 mode = atoi(reply);
688
689 QString name = ssid;
690 name = '[' + name + ']';
691
692 QStandardItem *item = new QStandardItem(*group_icon, name);
693 if (!item)
694 return;
695
696 int type;
697 if (mode == 3)
698 type = PEER_TYPE_P2P_PERSISTENT_GROUP_GO;
699 else
700 type = PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT;
701 item->setData(type, peer_role_type);
702 item->setToolTip(ItemType(type));
703 item->setData(ssid, peer_role_ssid);
704 if (bssid && strcmp(bssid, "any") == 0)
705 bssid = NULL;
706 if (bssid)
707 item->setData(bssid, peer_role_address);
708 item->setData(id, peer_role_network_id);
709 item->setBackground(Qt::BDiagPattern);
710
711 model.appendRow(item);
712 }
713
714
add_persistent_groups()715 void Peers::add_persistent_groups()
716 {
717 char buf[2048], *start, *end, *id, *ssid, *bssid, *flags;
718 size_t len;
719
720 len = sizeof(buf) - 1;
721 if (wpagui->ctrlRequest("LIST_NETWORKS", buf, &len) < 0)
722 return;
723
724 buf[len] = '\0';
725 start = strchr(buf, '\n');
726 if (start == NULL)
727 return;
728 start++;
729
730 while (*start) {
731 bool last = false;
732 end = strchr(start, '\n');
733 if (end == NULL) {
734 last = true;
735 end = start;
736 while (end[0] && end[1])
737 end++;
738 }
739 *end = '\0';
740
741 id = start;
742 ssid = strchr(id, '\t');
743 if (ssid == NULL)
744 break;
745 *ssid++ = '\0';
746 bssid = strchr(ssid, '\t');
747 if (bssid == NULL)
748 break;
749 *bssid++ = '\0';
750 flags = strchr(bssid, '\t');
751 if (flags == NULL)
752 break;
753 *flags++ = '\0';
754
755 if (strstr(flags, "[DISABLED][P2P-PERSISTENT]"))
756 add_persistent(atoi(id), ssid, bssid);
757
758 if (last)
759 break;
760 start = end + 1;
761 }
762 }
763
764
update_peers()765 void Peers::update_peers()
766 {
767 model.clear();
768 if (wpagui == NULL)
769 return;
770
771 char reply[20];
772 size_t replylen = sizeof(reply) - 1;
773 wpagui->ctrlRequest("WPS_ER_START", reply, &replylen);
774
775 add_stations();
776 add_scan_results();
777 add_persistent_groups();
778 }
779
780
find_addr(QString addr)781 QStandardItem * Peers::find_addr(QString addr)
782 {
783 if (model.rowCount() == 0)
784 return NULL;
785
786 QModelIndexList lst = model.match(model.index(0, 0), peer_role_address,
787 addr);
788 if (lst.size() == 0)
789 return NULL;
790 return model.itemFromIndex(lst[0]);
791 }
792
793
find_addr_type(QString addr,int type)794 QStandardItem * Peers::find_addr_type(QString addr, int type)
795 {
796 if (model.rowCount() == 0)
797 return NULL;
798
799 QModelIndexList lst = model.match(model.index(0, 0), peer_role_address,
800 addr);
801 for (int i = 0; i < lst.size(); i++) {
802 QStandardItem *item = model.itemFromIndex(lst[i]);
803 if (item->data(peer_role_type).toInt() == type)
804 return item;
805 }
806 return NULL;
807 }
808
809
find_uuid(QString uuid)810 QStandardItem * Peers::find_uuid(QString uuid)
811 {
812 if (model.rowCount() == 0)
813 return NULL;
814
815 QModelIndexList lst = model.match(model.index(0, 0), peer_role_uuid,
816 uuid);
817 if (lst.size() == 0)
818 return NULL;
819 return model.itemFromIndex(lst[0]);
820 }
821
822
event_notify(WpaMsg msg)823 void Peers::event_notify(WpaMsg msg)
824 {
825 QString text = msg.getMsg();
826
827 if (text.startsWith(WPS_EVENT_PIN_NEEDED)) {
828 /*
829 * WPS-PIN-NEEDED 5a02a5fa-9199-5e7c-bc46-e183d3cb32f7
830 * 02:2a:c4:18:5b:f3
831 * [Wireless Client|Company|cmodel|123|12345|1-0050F204-1]
832 */
833 QStringList items = text.split(' ');
834 QString uuid = items[1];
835 QString addr = items[2];
836 QString name = "";
837
838 QStandardItem *item = find_addr(addr);
839 if (item)
840 return;
841
842 int pos = text.indexOf('[');
843 if (pos >= 0) {
844 int pos2 = text.lastIndexOf(']');
845 if (pos2 >= pos) {
846 items = text.mid(pos + 1, pos2 - pos - 1).
847 split('|');
848 name = items[0];
849 items.append(addr);
850 }
851 }
852
853 item = new QStandardItem(*laptop_icon, name);
854 if (item) {
855 item->setData(addr, peer_role_address);
856 item->setData(PEER_TYPE_WPS_PIN_NEEDED,
857 peer_role_type);
858 item->setToolTip(ItemType(PEER_TYPE_WPS_PIN_NEEDED));
859 item->setData(items.join("\n"), peer_role_details);
860 item->setData(items[5], peer_role_pri_dev_type);
861 model.appendRow(item);
862 }
863 return;
864 }
865
866 if (text.startsWith(AP_STA_CONNECTED)) {
867 /* AP-STA-CONNECTED 02:2a:c4:18:5b:f3 */
868 QStringList items = text.split(' ');
869 QString addr = items[1];
870 QStandardItem *item = find_addr(addr);
871 if (item == NULL || item->data(peer_role_type).toInt() !=
872 PEER_TYPE_ASSOCIATED_STATION)
873 add_single_station(addr.toLocal8Bit().constData());
874 return;
875 }
876
877 if (text.startsWith(AP_STA_DISCONNECTED)) {
878 /* AP-STA-DISCONNECTED 02:2a:c4:18:5b:f3 */
879 QStringList items = text.split(' ');
880 QString addr = items[1];
881
882 if (model.rowCount() == 0)
883 return;
884
885 QModelIndexList lst = model.match(model.index(0, 0),
886 peer_role_address, addr, -1);
887 for (int i = 0; i < lst.size(); i++) {
888 QStandardItem *item = model.itemFromIndex(lst[i]);
889 if (item && item->data(peer_role_type).toInt() ==
890 PEER_TYPE_ASSOCIATED_STATION) {
891 model.removeRow(lst[i].row());
892 break;
893 }
894 }
895 return;
896 }
897
898 if (text.startsWith(P2P_EVENT_DEVICE_FOUND)) {
899 /*
900 * P2P-DEVICE-FOUND 02:b5:64:63:30:63
901 * p2p_dev_addr=02:b5:64:63:30:63 pri_dev_type=1-0050f204-1
902 * name='Wireless Client' config_methods=0x84 dev_capab=0x21
903 * group_capab=0x0
904 */
905 QStringList items =
906 text.split(QRegExp(" (?=[^']*('[^']*'[^']*)*$)"));
907 QString addr = items[1];
908 QString name = "";
909 QString pri_dev_type;
910 int config_methods = 0;
911 for (int i = 0; i < items.size(); i++) {
912 QString str = items.at(i);
913 if (str.startsWith("name='"))
914 name = str.section('\'', 1, -2);
915 else if (str.startsWith("config_methods="))
916 config_methods =
917 str.section('=', 1).toInt(0, 0);
918 else if (str.startsWith("pri_dev_type="))
919 pri_dev_type = str.section('=', 1);
920 }
921
922 QStandardItem *item = find_addr(addr);
923 if (item) {
924 int type = item->data(peer_role_type).toInt();
925 if (type == PEER_TYPE_P2P)
926 return;
927 }
928
929 item = new QStandardItem(*default_icon, name);
930 if (item) {
931 item->setData(addr, peer_role_address);
932 item->setData(config_methods,
933 peer_role_config_methods);
934 item->setData(PEER_TYPE_P2P, peer_role_type);
935 if (!pri_dev_type.isEmpty())
936 item->setData(pri_dev_type,
937 peer_role_pri_dev_type);
938 item->setData(items.join(QString("\n")),
939 peer_role_details);
940 item->setToolTip(ItemType(PEER_TYPE_P2P));
941 model.appendRow(item);
942 }
943
944 item = find_addr_type(addr,
945 PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT);
946 if (item)
947 item->setBackground(Qt::NoBrush);
948 }
949
950 if (text.startsWith(P2P_EVENT_GROUP_STARTED)) {
951 /* P2P-GROUP-STARTED wlan0-p2p-0 GO ssid="DIRECT-3F"
952 * passphrase="YOyTkxID" go_dev_addr=02:40:61:c2:f3:b7
953 * [PERSISTENT] */
954 QStringList items = text.split(' ');
955 if (items.size() < 4)
956 return;
957
958 int pos = text.indexOf(" ssid=\"");
959 if (pos < 0)
960 return;
961 QString ssid = text.mid(pos + 7);
962 pos = ssid.indexOf(" passphrase=\"");
963 if (pos < 0)
964 pos = ssid.indexOf(" psk=");
965 if (pos >= 0)
966 ssid.truncate(pos);
967 pos = ssid.lastIndexOf('"');
968 if (pos >= 0)
969 ssid.truncate(pos);
970
971 QStandardItem *item = new QStandardItem(*group_icon, ssid);
972 if (item) {
973 item->setData(PEER_TYPE_P2P_GROUP, peer_role_type);
974 item->setData(items[1], peer_role_ifname);
975 QString details;
976 if (items[2] == "GO") {
977 details = tr("P2P GO for interface ") +
978 items[1];
979 } else {
980 details = tr("P2P client for interface ") +
981 items[1];
982 }
983 if (text.contains(" [PERSISTENT]"))
984 details += "\nPersistent group";
985 item->setData(details, peer_role_details);
986 item->setToolTip(ItemType(PEER_TYPE_P2P_GROUP));
987 model.appendRow(item);
988 }
989 }
990
991 if (text.startsWith(P2P_EVENT_GROUP_REMOVED)) {
992 /* P2P-GROUP-REMOVED wlan0-p2p-0 GO */
993 QStringList items = text.split(' ');
994 if (items.size() < 2)
995 return;
996
997 if (model.rowCount() == 0)
998 return;
999
1000 QModelIndexList lst = model.match(model.index(0, 0),
1001 peer_role_ifname, items[1]);
1002 for (int i = 0; i < lst.size(); i++)
1003 model.removeRow(lst[i].row());
1004 return;
1005 }
1006
1007 if (text.startsWith(P2P_EVENT_PROV_DISC_SHOW_PIN)) {
1008 /* P2P-PROV-DISC-SHOW-PIN 02:40:61:c2:f3:b7 12345670 */
1009 QStringList items = text.split(' ');
1010 if (items.size() < 3)
1011 return;
1012 QString addr = items[1];
1013 QString pin = items[2];
1014
1015 QStandardItem *item = find_addr_type(addr, PEER_TYPE_P2P);
1016 if (item == NULL)
1017 return;
1018 item->setData(SEL_METHOD_PIN_LOCAL_DISPLAY,
1019 peer_role_selected_method);
1020 item->setData(pin, peer_role_selected_pin);
1021 QVariant var = item->data(peer_role_requested_method);
1022 if (var.isValid() &&
1023 var.toInt() == SEL_METHOD_PIN_LOCAL_DISPLAY) {
1024 ctx_item = item;
1025 ctx_p2p_display_pin_pd();
1026 }
1027 return;
1028 }
1029
1030 if (text.startsWith(P2P_EVENT_PROV_DISC_ENTER_PIN)) {
1031 /* P2P-PROV-DISC-ENTER-PIN 02:40:61:c2:f3:b7 */
1032 QStringList items = text.split(' ');
1033 if (items.size() < 2)
1034 return;
1035 QString addr = items[1];
1036
1037 QStandardItem *item = find_addr_type(addr, PEER_TYPE_P2P);
1038 if (item == NULL)
1039 return;
1040 item->setData(SEL_METHOD_PIN_PEER_DISPLAY,
1041 peer_role_selected_method);
1042 QVariant var = item->data(peer_role_requested_method);
1043 if (var.isValid() &&
1044 var.toInt() == SEL_METHOD_PIN_PEER_DISPLAY) {
1045 ctx_item = item;
1046 ctx_p2p_connect();
1047 }
1048 return;
1049 }
1050
1051 if (text.startsWith(P2P_EVENT_INVITATION_RECEIVED)) {
1052 /* P2P-INVITATION-RECEIVED sa=02:f0:bc:44:87:62 persistent=4 */
1053 QStringList items = text.split(' ');
1054 if (items.size() < 3)
1055 return;
1056 if (!items[1].startsWith("sa=") ||
1057 !items[2].startsWith("persistent="))
1058 return;
1059 QString addr = items[1].mid(3);
1060 int id = items[2].mid(11).toInt();
1061
1062 char cmd[100];
1063 char reply[100];
1064 size_t reply_len;
1065
1066 snprintf(cmd, sizeof(cmd), "GET_NETWORK %d ssid", id);
1067 reply_len = sizeof(reply) - 1;
1068 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
1069 return;
1070 reply[reply_len] = '\0';
1071 QString name;
1072 char *pos = strrchr(reply, '"');
1073 if (pos && reply[0] == '"') {
1074 *pos = '\0';
1075 name = reply + 1;
1076 } else
1077 name = reply;
1078
1079 QStandardItem *item;
1080 item = find_addr_type(addr, PEER_TYPE_P2P_INVITATION);
1081 if (item)
1082 model.removeRow(item->row());
1083
1084 item = new QStandardItem(*invitation_icon, name);
1085 if (!item)
1086 return;
1087 item->setData(PEER_TYPE_P2P_INVITATION, peer_role_type);
1088 item->setToolTip(ItemType(PEER_TYPE_P2P_INVITATION));
1089 item->setData(addr, peer_role_address);
1090 item->setData(id, peer_role_network_id);
1091
1092 model.appendRow(item);
1093
1094 enable_persistent(id);
1095
1096 return;
1097 }
1098
1099 if (text.startsWith(P2P_EVENT_INVITATION_RESULT)) {
1100 /* P2P-INVITATION-RESULT status=1 */
1101 /* TODO */
1102 return;
1103 }
1104
1105 if (text.startsWith(WPS_EVENT_ER_AP_ADD)) {
1106 /*
1107 * WPS-ER-AP-ADD 87654321-9abc-def0-1234-56789abc0002
1108 * 02:11:22:33:44:55 pri_dev_type=6-0050F204-1 wps_state=1
1109 * |Very friendly name|Company|Long description of the model|
1110 * WAP|http://w1.fi/|http://w1.fi/hostapd/
1111 */
1112 QStringList items = text.split(' ');
1113 if (items.size() < 5)
1114 return;
1115 QString uuid = items[1];
1116 QString addr = items[2];
1117 QString pri_dev_type = items[3].mid(13);
1118 int wps_state = items[4].mid(10).toInt();
1119
1120 int pos = text.indexOf('|');
1121 if (pos < 0)
1122 return;
1123 items = text.mid(pos + 1).split('|');
1124 if (items.size() < 1)
1125 return;
1126
1127 QStandardItem *item = find_uuid(uuid);
1128 if (item)
1129 return;
1130
1131 item = new QStandardItem(*ap_icon, items[0]);
1132 if (item) {
1133 item->setData(uuid, peer_role_uuid);
1134 item->setData(addr, peer_role_address);
1135 int type = wps_state == 2 ? PEER_TYPE_WPS_ER_AP:
1136 PEER_TYPE_WPS_ER_AP_UNCONFIGURED;
1137 item->setData(type, peer_role_type);
1138 item->setToolTip(ItemType(type));
1139 item->setData(pri_dev_type, peer_role_pri_dev_type);
1140 item->setData(items.join(QString("\n")),
1141 peer_role_details);
1142 model.appendRow(item);
1143 }
1144
1145 return;
1146 }
1147
1148 if (text.startsWith(WPS_EVENT_ER_AP_REMOVE)) {
1149 /* WPS-ER-AP-REMOVE 87654321-9abc-def0-1234-56789abc0002 */
1150 QStringList items = text.split(' ');
1151 if (items.size() < 2)
1152 return;
1153 if (model.rowCount() == 0)
1154 return;
1155
1156 QModelIndexList lst = model.match(model.index(0, 0),
1157 peer_role_uuid, items[1]);
1158 for (int i = 0; i < lst.size(); i++) {
1159 QStandardItem *item = model.itemFromIndex(lst[i]);
1160 if (item &&
1161 (item->data(peer_role_type).toInt() ==
1162 PEER_TYPE_WPS_ER_AP ||
1163 item->data(peer_role_type).toInt() ==
1164 PEER_TYPE_WPS_ER_AP_UNCONFIGURED))
1165 model.removeRow(lst[i].row());
1166 }
1167 return;
1168 }
1169
1170 if (text.startsWith(WPS_EVENT_ER_ENROLLEE_ADD)) {
1171 /*
1172 * WPS-ER-ENROLLEE-ADD 2b7093f1-d6fb-5108-adbb-bea66bb87333
1173 * 02:66:a0:ee:17:27 M1=1 config_methods=0x14d dev_passwd_id=0
1174 * pri_dev_type=1-0050F204-1
1175 * |Wireless Client|Company|cmodel|123|12345|
1176 */
1177 QStringList items = text.split(' ');
1178 if (items.size() < 3)
1179 return;
1180 QString uuid = items[1];
1181 QString addr = items[2];
1182 QString pri_dev_type = items[6].mid(13);
1183 int config_methods = -1;
1184 int dev_passwd_id = -1;
1185
1186 for (int i = 3; i < items.size(); i++) {
1187 int pos = items[i].indexOf('=') + 1;
1188 if (pos < 1)
1189 continue;
1190 QString val = items[i].mid(pos);
1191 if (items[i].startsWith("config_methods=")) {
1192 config_methods = val.toInt(0, 0);
1193 } else if (items[i].startsWith("dev_passwd_id=")) {
1194 dev_passwd_id = val.toInt();
1195 }
1196 }
1197
1198 int pos = text.indexOf('|');
1199 if (pos < 0)
1200 return;
1201 items = text.mid(pos + 1).split('|');
1202 if (items.size() < 1)
1203 return;
1204 QString name = items[0];
1205 if (name.length() == 0)
1206 name = addr;
1207
1208 remove_enrollee_uuid(uuid);
1209
1210 QStandardItem *item;
1211 item = new QStandardItem(*laptop_icon, name);
1212 if (item) {
1213 item->setData(uuid, peer_role_uuid);
1214 item->setData(addr, peer_role_address);
1215 item->setData(PEER_TYPE_WPS_ER_ENROLLEE,
1216 peer_role_type);
1217 item->setToolTip(ItemType(PEER_TYPE_WPS_ER_ENROLLEE));
1218 item->setData(items.join(QString("\n")),
1219 peer_role_details);
1220 item->setData(pri_dev_type, peer_role_pri_dev_type);
1221 if (config_methods >= 0)
1222 item->setData(config_methods,
1223 peer_role_config_methods);
1224 if (dev_passwd_id >= 0)
1225 item->setData(dev_passwd_id,
1226 peer_role_dev_passwd_id);
1227 model.appendRow(item);
1228 }
1229
1230 return;
1231 }
1232
1233 if (text.startsWith(WPS_EVENT_ER_ENROLLEE_REMOVE)) {
1234 /*
1235 * WPS-ER-ENROLLEE-REMOVE 2b7093f1-d6fb-5108-adbb-bea66bb87333
1236 * 02:66:a0:ee:17:27
1237 */
1238 QStringList items = text.split(' ');
1239 if (items.size() < 2)
1240 return;
1241 remove_enrollee_uuid(items[1]);
1242 return;
1243 }
1244
1245 if (text.startsWith(WPS_EVENT_ENROLLEE_SEEN)) {
1246 /* TODO: need to time out this somehow or remove on successful
1247 * WPS run, etc. */
1248 /*
1249 * WPS-ENROLLEE-SEEN 02:00:00:00:01:00
1250 * 572cf82f-c957-5653-9b16-b5cfb298abf1 1-0050F204-1 0x80 4 1
1251 * [Wireless Client]
1252 * (MAC addr, UUID-E, pri dev type, config methods,
1253 * dev passwd id, request type, [dev name])
1254 */
1255 QStringList items = text.split(' ');
1256 if (items.size() < 7)
1257 return;
1258 QString addr = items[1];
1259 QString uuid = items[2];
1260 QString pri_dev_type = items[3];
1261 int config_methods = items[4].toInt(0, 0);
1262 int dev_passwd_id = items[5].toInt();
1263 QString name;
1264
1265 QStandardItem *item = find_addr(addr);
1266 if (item) {
1267 int type = item->data(peer_role_type).toInt();
1268 if (type == PEER_TYPE_ASSOCIATED_STATION)
1269 return; /* already associated */
1270 }
1271
1272 int pos = text.indexOf('[');
1273 if (pos >= 0) {
1274 int pos2 = text.lastIndexOf(']');
1275 if (pos2 >= pos) {
1276 QStringList items2 =
1277 text.mid(pos + 1, pos2 - pos - 1).
1278 split('|');
1279 name = items2[0];
1280 }
1281 }
1282 if (name.isEmpty())
1283 name = addr;
1284
1285 item = find_uuid(uuid);
1286 if (item) {
1287 QVariant var = item->data(peer_role_config_methods);
1288 QVariant var2 = item->data(peer_role_dev_passwd_id);
1289 if ((var.isValid() && config_methods != var.toInt()) ||
1290 (var2.isValid() && dev_passwd_id != var2.toInt()))
1291 remove_enrollee_uuid(uuid);
1292 else
1293 return;
1294 }
1295
1296 item = new QStandardItem(*laptop_icon, name);
1297 if (item) {
1298 item->setData(uuid, peer_role_uuid);
1299 item->setData(addr, peer_role_address);
1300 item->setData(PEER_TYPE_WPS_ENROLLEE,
1301 peer_role_type);
1302 item->setToolTip(ItemType(PEER_TYPE_WPS_ENROLLEE));
1303 item->setData(items.join(QString("\n")),
1304 peer_role_details);
1305 item->setData(pri_dev_type, peer_role_pri_dev_type);
1306 item->setData(config_methods,
1307 peer_role_config_methods);
1308 item->setData(dev_passwd_id, peer_role_dev_passwd_id);
1309 model.appendRow(item);
1310 }
1311
1312 return;
1313 }
1314
1315 if (text.startsWith(WPA_EVENT_BSS_ADDED)) {
1316 /* CTRL-EVENT-BSS-ADDED 34 00:11:22:33:44:55 */
1317 QStringList items = text.split(' ');
1318 if (items.size() < 2)
1319 return;
1320 char cmd[20];
1321 snprintf(cmd, sizeof(cmd), "BSS ID-%d", items[1].toInt());
1322 add_bss(cmd);
1323 return;
1324 }
1325
1326 if (text.startsWith(WPA_EVENT_BSS_REMOVED)) {
1327 /* CTRL-EVENT-BSS-REMOVED 34 00:11:22:33:44:55 */
1328 QStringList items = text.split(' ');
1329 if (items.size() < 2)
1330 return;
1331 remove_bss(items[1].toInt());
1332 return;
1333 }
1334 }
1335
1336
ctx_p2p_connect()1337 void Peers::ctx_p2p_connect()
1338 {
1339 if (ctx_item == NULL)
1340 return;
1341 QString addr = ctx_item->data(peer_role_address).toString();
1342 QString arg;
1343 int config_methods =
1344 ctx_item->data(peer_role_config_methods).toInt();
1345 enum selected_method method = SEL_METHOD_NONE;
1346 QVariant var = ctx_item->data(peer_role_selected_method);
1347 if (var.isValid())
1348 method = (enum selected_method) var.toInt();
1349 if (method == SEL_METHOD_PIN_LOCAL_DISPLAY) {
1350 arg = ctx_item->data(peer_role_selected_pin).toString();
1351 char cmd[100];
1352 char reply[100];
1353 size_t reply_len;
1354 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s %s display",
1355 addr.toLocal8Bit().constData(),
1356 arg.toLocal8Bit().constData());
1357 reply_len = sizeof(reply) - 1;
1358 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1359 QMessageBox msg;
1360 msg.setIcon(QMessageBox::Warning);
1361 msg.setText("Failed to initiate P2P connect.");
1362 msg.exec();
1363 return;
1364 }
1365 QMessageBox::information(this,
1366 tr("PIN for ") + ctx_item->text(),
1367 tr("Enter the following PIN on the\n"
1368 "peer device: ") + arg);
1369 } else if (method == SEL_METHOD_PIN_PEER_DISPLAY) {
1370 StringQuery input(tr("PIN from peer display:"));
1371 input.setWindowTitle(tr("PIN for ") + ctx_item->text());
1372 if (input.exec() != QDialog::Accepted)
1373 return;
1374 arg = input.get_string();
1375 } else if (config_methods == 0x0080 /* PBC */) {
1376 arg = "pbc";
1377 } else {
1378 StringQuery input(tr("PIN:"));
1379 input.setWindowTitle(tr("PIN for ") + ctx_item->text());
1380 if (input.exec() != QDialog::Accepted)
1381 return;
1382 arg = input.get_string();
1383 }
1384
1385 char cmd[100];
1386 char reply[100];
1387 size_t reply_len;
1388 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s %s",
1389 addr.toLocal8Bit().constData(),
1390 arg.toLocal8Bit().constData());
1391 reply_len = sizeof(reply) - 1;
1392 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1393 QMessageBox msg;
1394 msg.setIcon(QMessageBox::Warning);
1395 msg.setText("Failed to initiate P2P connect.");
1396 msg.exec();
1397 }
1398 }
1399
1400
ctx_p2p_req_pin()1401 void Peers::ctx_p2p_req_pin()
1402 {
1403 if (ctx_item == NULL)
1404 return;
1405 QString addr = ctx_item->data(peer_role_address).toString();
1406 ctx_item->setData(SEL_METHOD_PIN_PEER_DISPLAY,
1407 peer_role_requested_method);
1408
1409 char cmd[100];
1410 char reply[100];
1411 size_t reply_len;
1412 snprintf(cmd, sizeof(cmd), "P2P_PROV_DISC %s display",
1413 addr.toLocal8Bit().constData());
1414 reply_len = sizeof(reply) - 1;
1415 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1416 QMessageBox msg;
1417 msg.setIcon(QMessageBox::Warning);
1418 msg.setText(tr("Failed to request PIN from peer."));
1419 msg.exec();
1420 }
1421 }
1422
1423
ctx_p2p_show_pin()1424 void Peers::ctx_p2p_show_pin()
1425 {
1426 if (ctx_item == NULL)
1427 return;
1428 QString addr = ctx_item->data(peer_role_address).toString();
1429 ctx_item->setData(SEL_METHOD_PIN_LOCAL_DISPLAY,
1430 peer_role_requested_method);
1431
1432 char cmd[100];
1433 char reply[100];
1434 size_t reply_len;
1435 snprintf(cmd, sizeof(cmd), "P2P_PROV_DISC %s keypad",
1436 addr.toLocal8Bit().constData());
1437 reply_len = sizeof(reply) - 1;
1438 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1439 QMessageBox msg;
1440 msg.setIcon(QMessageBox::Warning);
1441 msg.setText(tr("Failed to request peer to enter PIN."));
1442 msg.exec();
1443 }
1444 }
1445
1446
ctx_p2p_display_pin()1447 void Peers::ctx_p2p_display_pin()
1448 {
1449 if (ctx_item == NULL)
1450 return;
1451 QString addr = ctx_item->data(peer_role_address).toString();
1452
1453 char cmd[100];
1454 char reply[100];
1455 size_t reply_len;
1456 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s pin",
1457 addr.toLocal8Bit().constData());
1458 reply_len = sizeof(reply) - 1;
1459 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1460 QMessageBox msg;
1461 msg.setIcon(QMessageBox::Warning);
1462 msg.setText("Failed to initiate P2P connect.");
1463 msg.exec();
1464 return;
1465 }
1466 reply[reply_len] = '\0';
1467 QMessageBox::information(this,
1468 tr("PIN for ") + ctx_item->text(),
1469 tr("Enter the following PIN on the\n"
1470 "peer device: ") + reply);
1471 }
1472
1473
ctx_p2p_display_pin_pd()1474 void Peers::ctx_p2p_display_pin_pd()
1475 {
1476 if (ctx_item == NULL)
1477 return;
1478 QString addr = ctx_item->data(peer_role_address).toString();
1479 QString arg = ctx_item->data(peer_role_selected_pin).toString();
1480
1481 char cmd[100];
1482 char reply[100];
1483 size_t reply_len;
1484 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s %s display",
1485 addr.toLocal8Bit().constData(),
1486 arg.toLocal8Bit().constData());
1487 reply_len = sizeof(reply) - 1;
1488 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1489 QMessageBox msg;
1490 msg.setIcon(QMessageBox::Warning);
1491 msg.setText("Failed to initiate P2P connect.");
1492 msg.exec();
1493 return;
1494 }
1495 reply[reply_len] = '\0';
1496 QMessageBox::information(this,
1497 tr("PIN for ") + ctx_item->text(),
1498 tr("Enter the following PIN on the\n"
1499 "peer device: ") + arg);
1500 }
1501
1502
ctx_p2p_enter_pin()1503 void Peers::ctx_p2p_enter_pin()
1504 {
1505 if (ctx_item == NULL)
1506 return;
1507 QString addr = ctx_item->data(peer_role_address).toString();
1508 QString arg;
1509
1510 StringQuery input(tr("PIN from peer:"));
1511 input.setWindowTitle(tr("PIN for ") + ctx_item->text());
1512 if (input.exec() != QDialog::Accepted)
1513 return;
1514 arg = input.get_string();
1515
1516 char cmd[100];
1517 char reply[100];
1518 size_t reply_len;
1519 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s %s keypad",
1520 addr.toLocal8Bit().constData(),
1521 arg.toLocal8Bit().constData());
1522 reply_len = sizeof(reply) - 1;
1523 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1524 QMessageBox msg;
1525 msg.setIcon(QMessageBox::Warning);
1526 msg.setText("Failed to initiate P2P connect.");
1527 msg.exec();
1528 }
1529 }
1530
1531
ctx_p2p_remove_group()1532 void Peers::ctx_p2p_remove_group()
1533 {
1534 if (ctx_item == NULL)
1535 return;
1536 char cmd[100];
1537 char reply[100];
1538 size_t reply_len;
1539 snprintf(cmd, sizeof(cmd), "P2P_GROUP_REMOVE %s",
1540 ctx_item->data(peer_role_ifname).toString().toLocal8Bit().
1541 constData());
1542 reply_len = sizeof(reply) - 1;
1543 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1544 QMessageBox msg;
1545 msg.setIcon(QMessageBox::Warning);
1546 msg.setText("Failed to remove P2P Group.");
1547 msg.exec();
1548 }
1549 }
1550
1551
closeEvent(QCloseEvent *)1552 void Peers::closeEvent(QCloseEvent *)
1553 {
1554 if (wpagui) {
1555 char reply[20];
1556 size_t replylen = sizeof(reply) - 1;
1557 wpagui->ctrlRequest("WPS_ER_STOP", reply, &replylen);
1558 }
1559 }
1560
1561
done(int r)1562 void Peers::done(int r)
1563 {
1564 QDialog::done(r);
1565 close();
1566 }
1567
1568
remove_enrollee_uuid(QString uuid)1569 void Peers::remove_enrollee_uuid(QString uuid)
1570 {
1571 if (model.rowCount() == 0)
1572 return;
1573
1574 QModelIndexList lst = model.match(model.index(0, 0),
1575 peer_role_uuid, uuid);
1576 for (int i = 0; i < lst.size(); i++) {
1577 QStandardItem *item = model.itemFromIndex(lst[i]);
1578 if (item == NULL)
1579 continue;
1580 int type = item->data(peer_role_type).toInt();
1581 if (type == PEER_TYPE_WPS_ER_ENROLLEE ||
1582 type == PEER_TYPE_WPS_ENROLLEE)
1583 model.removeRow(lst[i].row());
1584 }
1585 }
1586
1587
properties()1588 void Peers::properties()
1589 {
1590 if (ctx_item == NULL)
1591 return;
1592
1593 QMessageBox msg(this);
1594 msg.setStandardButtons(QMessageBox::Ok);
1595 msg.setDefaultButton(QMessageBox::Ok);
1596 msg.setEscapeButton(QMessageBox::Ok);
1597 msg.setWindowTitle(tr("Peer Properties"));
1598
1599 int type = ctx_item->data(peer_role_type).toInt();
1600 QString title = Peers::ItemType(type);
1601
1602 msg.setText(title + QString("\n") + tr("Name: ") + ctx_item->text());
1603
1604 QVariant var;
1605 QString info;
1606
1607 var = ctx_item->data(peer_role_address);
1608 if (var.isValid())
1609 info += tr("Address: ") + var.toString() + QString("\n");
1610
1611 var = ctx_item->data(peer_role_uuid);
1612 if (var.isValid())
1613 info += tr("UUID: ") + var.toString() + QString("\n");
1614
1615 var = ctx_item->data(peer_role_pri_dev_type);
1616 if (var.isValid())
1617 info += tr("Primary Device Type: ") + var.toString() +
1618 QString("\n");
1619
1620 var = ctx_item->data(peer_role_ssid);
1621 if (var.isValid())
1622 info += tr("SSID: ") + var.toString() + QString("\n");
1623
1624 var = ctx_item->data(peer_role_config_methods);
1625 if (var.isValid()) {
1626 int methods = var.toInt();
1627 info += tr("Configuration Methods: ");
1628 if (methods & 0x0001)
1629 info += tr("[USBA]");
1630 if (methods & 0x0002)
1631 info += tr("[Ethernet]");
1632 if (methods & 0x0004)
1633 info += tr("[Label]");
1634 if (methods & 0x0008)
1635 info += tr("[Display]");
1636 if (methods & 0x0010)
1637 info += tr("[Ext. NFC Token]");
1638 if (methods & 0x0020)
1639 info += tr("[Int. NFC Token]");
1640 if (methods & 0x0040)
1641 info += tr("[NFC Interface]");
1642 if (methods & 0x0080)
1643 info += tr("[Push Button]");
1644 if (methods & 0x0100)
1645 info += tr("[Keypad]");
1646 info += "\n";
1647 }
1648
1649 var = ctx_item->data(peer_role_selected_method);
1650 if (var.isValid()) {
1651 enum selected_method method =
1652 (enum selected_method) var.toInt();
1653 switch (method) {
1654 case SEL_METHOD_NONE:
1655 break;
1656 case SEL_METHOD_PIN_PEER_DISPLAY:
1657 info += tr("Selected Method: PIN on peer display\n");
1658 break;
1659 case SEL_METHOD_PIN_LOCAL_DISPLAY:
1660 info += tr("Selected Method: PIN on local display\n");
1661 break;
1662 }
1663 }
1664
1665 var = ctx_item->data(peer_role_selected_pin);
1666 if (var.isValid()) {
1667 info += tr("PIN to enter on peer: ") + var.toString() + "\n";
1668 }
1669
1670 var = ctx_item->data(peer_role_dev_passwd_id);
1671 if (var.isValid()) {
1672 info += tr("Device Password ID: ") + var.toString();
1673 switch (var.toInt()) {
1674 case 0:
1675 info += tr(" (Default PIN)");
1676 break;
1677 case 1:
1678 info += tr(" (User-specified PIN)");
1679 break;
1680 case 2:
1681 info += tr(" (Machine-specified PIN)");
1682 break;
1683 case 3:
1684 info += tr(" (Rekey)");
1685 break;
1686 case 4:
1687 info += tr(" (Push Button)");
1688 break;
1689 case 5:
1690 info += tr(" (Registrar-specified)");
1691 break;
1692 }
1693 info += "\n";
1694 }
1695
1696 msg.setInformativeText(info);
1697
1698 var = ctx_item->data(peer_role_details);
1699 if (var.isValid())
1700 msg.setDetailedText(var.toString());
1701
1702 msg.exec();
1703 }
1704
1705
connect_pbc()1706 void Peers::connect_pbc()
1707 {
1708 if (ctx_item == NULL)
1709 return;
1710
1711 char cmd[100];
1712 char reply[100];
1713 size_t reply_len;
1714
1715 int peer_type = ctx_item->data(peer_role_type).toInt();
1716 if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE) {
1717 snprintf(cmd, sizeof(cmd), "WPS_ER_PBC %s",
1718 ctx_item->data(peer_role_uuid).toString().toLocal8Bit().
1719 constData());
1720 } else if (peer_type == PEER_TYPE_P2P ||
1721 peer_type == PEER_TYPE_P2P_CLIENT) {
1722 snprintf(cmd, sizeof(cmd), "P2P_CONNECT %s pbc",
1723 ctx_item->data(peer_role_address).toString().
1724 toLocal8Bit().constData());
1725 } else {
1726 snprintf(cmd, sizeof(cmd), "WPS_PBC");
1727 }
1728 reply_len = sizeof(reply) - 1;
1729 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1730 QMessageBox msg;
1731 msg.setIcon(QMessageBox::Warning);
1732 msg.setText(tr("Failed to start WPS PBC."));
1733 msg.exec();
1734 }
1735 }
1736
1737
learn_ap_config()1738 void Peers::learn_ap_config()
1739 {
1740 if (ctx_item == NULL)
1741 return;
1742
1743 QString uuid = ctx_item->data(peer_role_uuid).toString();
1744
1745 StringQuery input(tr("AP PIN:"));
1746 input.setWindowTitle(tr("AP PIN for ") + ctx_item->text());
1747 if (input.exec() != QDialog::Accepted)
1748 return;
1749
1750 char cmd[100];
1751 char reply[100];
1752 size_t reply_len;
1753
1754 snprintf(cmd, sizeof(cmd), "WPS_ER_LEARN %s %s",
1755 uuid.toLocal8Bit().constData(),
1756 input.get_string().toLocal8Bit().constData());
1757 reply_len = sizeof(reply) - 1;
1758 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
1759 QMessageBox msg;
1760 msg.setIcon(QMessageBox::Warning);
1761 msg.setText(tr("Failed to start learning AP configuration."));
1762 msg.exec();
1763 }
1764 }
1765
1766
ctx_hide_ap()1767 void Peers::ctx_hide_ap()
1768 {
1769 hide_ap = true;
1770
1771 if (model.rowCount() == 0)
1772 return;
1773
1774 do {
1775 QModelIndexList lst;
1776 lst = model.match(model.index(0, 0),
1777 peer_role_type, PEER_TYPE_AP);
1778 if (lst.size() == 0) {
1779 lst = model.match(model.index(0, 0),
1780 peer_role_type, PEER_TYPE_AP_WPS);
1781 if (lst.size() == 0)
1782 break;
1783 }
1784
1785 model.removeRow(lst[0].row());
1786 } while (1);
1787 }
1788
1789
ctx_show_ap()1790 void Peers::ctx_show_ap()
1791 {
1792 hide_ap = false;
1793 add_scan_results();
1794 }
1795
1796
ctx_p2p_show_passphrase()1797 void Peers::ctx_p2p_show_passphrase()
1798 {
1799 char reply[64];
1800 size_t reply_len;
1801
1802 reply_len = sizeof(reply) - 1;
1803 if (wpagui->ctrlRequest("P2P_GET_PASSPHRASE", reply, &reply_len) < 0 ||
1804 memcmp(reply, "FAIL", 4) == 0) {
1805 QMessageBox msg;
1806 msg.setIcon(QMessageBox::Warning);
1807 msg.setText("Failed to get P2P group passphrase.");
1808 msg.exec();
1809 } else {
1810 reply[reply_len] = '\0';
1811 QMessageBox::information(this, tr("Passphrase"),
1812 tr("P2P group passphrase:\n") +
1813 reply);
1814 }
1815 }
1816
1817
ctx_p2p_start_persistent()1818 void Peers::ctx_p2p_start_persistent()
1819 {
1820 if (ctx_item == NULL)
1821 return;
1822
1823 char cmd[100];
1824 char reply[100];
1825 size_t reply_len;
1826
1827 snprintf(cmd, sizeof(cmd), "P2P_GROUP_ADD persistent=%d",
1828 ctx_item->data(peer_role_network_id).toInt());
1829 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0 ||
1830 memcmp(reply, "FAIL", 4) == 0) {
1831 QMessageBox msg;
1832 msg.setIcon(QMessageBox::Warning);
1833 msg.setText(tr("Failed to start persistent P2P Group."));
1834 msg.exec();
1835 } else if (ctx_item->data(peer_role_type).toInt() ==
1836 PEER_TYPE_P2P_INVITATION)
1837 model.removeRow(ctx_item->row());
1838 }
1839
1840
ctx_p2p_invite()1841 void Peers::ctx_p2p_invite()
1842 {
1843 if (ctx_item == NULL)
1844 return;
1845
1846 char cmd[100];
1847 char reply[100];
1848 size_t reply_len;
1849
1850 snprintf(cmd, sizeof(cmd), "P2P_INVITE persistent=%d",
1851 ctx_item->data(peer_role_network_id).toInt());
1852 if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0 ||
1853 memcmp(reply, "FAIL", 4) == 0) {
1854 QMessageBox msg;
1855 msg.setIcon(QMessageBox::Warning);
1856 msg.setText(tr("Failed to invite peer to start persistent "
1857 "P2P Group."));
1858 msg.exec();
1859 }
1860 }
1861
1862
ctx_p2p_delete()1863 void Peers::ctx_p2p_delete()
1864 {
1865 if (ctx_item == NULL)
1866 return;
1867 model.removeRow(ctx_item->row());
1868 }
1869
1870
enable_persistent(int id)1871 void Peers::enable_persistent(int id)
1872 {
1873 if (model.rowCount() == 0)
1874 return;
1875
1876 QModelIndexList lst = model.match(model.index(0, 0),
1877 peer_role_network_id, id);
1878 for (int i = 0; i < lst.size(); i++) {
1879 QStandardItem *item = model.itemFromIndex(lst[i]);
1880 int type = item->data(peer_role_type).toInt();
1881 if (type == PEER_TYPE_P2P_PERSISTENT_GROUP_GO ||
1882 type == PEER_TYPE_P2P_PERSISTENT_GROUP_CLIENT)
1883 item->setBackground(Qt::NoBrush);
1884 }
1885 }
1886
