wpa_supplicant/wpa_gui-qt4/peers.cpp

   1 /*
   2  * wpa_gui - Peers class
   3  * Copyright (c) 2009, Atheros Communications
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * Alternatively, this software may be distributed under the terms of BSD
  10  * license.
  11  *
  12  * See README and COPYING for more details.
  13  */
  14
  15 #include <cstdio>
  16 #include <QImageReader>
  17 #include <QMessageBox>
  18
  19 #include "common/wpa_ctrl.h"
  20 #include "wpagui.h"
  21 #include "stringquery.h"
  22 #include "peers.h"
  23
  24
  25 enum {
  26         peer_role_address = Qt::UserRole + 1,
  27         peer_role_type,
  28         peer_role_uuid,
  29         peer_role_details,
  30         peer_role_pri_dev_type,
  31         peer_role_ssid,
  32         peer_role_config_methods,
  33         peer_role_dev_passwd_id,
  34         peer_role_bss_id
  35 };
  36
  37 /*
  38  * TODO:
  39  * - add current AP info (e.g., from WPS) in station mode
  40  */
  41
  42 enum peer_type {
  43         PEER_TYPE_ASSOCIATED_STATION,
  44         PEER_TYPE_AP,
  45         PEER_TYPE_AP_WPS,
  46         PEER_TYPE_WPS_PIN_NEEDED,
  47         PEER_TYPE_WPS_ER_AP,
  48         PEER_TYPE_WPS_ER_AP_UNCONFIGURED,
  49         PEER_TYPE_WPS_ER_ENROLLEE,
  50         PEER_TYPE_WPS_ENROLLEE
  51 };
  52
  53
  54 Peers::Peers(QWidget *parent, const char *, bool, Qt::WFlags)
  55         : QDialog(parent)
  56 {
  57         setupUi(this);
  58
  59         if (QImageReader::supportedImageFormats().contains(QByteArray("svg")))
  60         {
  61                 default_icon = new QIcon(":/icons/wpa_gui.svg");
  62                 ap_icon = new QIcon(":/icons/ap.svg");
  63                 laptop_icon = new QIcon(":/icons/laptop.svg");
  64         } else {
  65                 default_icon = new QIcon(":/icons/wpa_gui.png");
  66                 ap_icon = new QIcon(":/icons/ap.png");
  67                 laptop_icon = new QIcon(":/icons/laptop.png");
  68         }
  69
  70         peers->setModel(&model);
  71         peers->setResizeMode(QListView::Adjust);
  72
  73         peers->setContextMenuPolicy(Qt::CustomContextMenu);
  74         connect(peers, SIGNAL(customContextMenuRequested(const QPoint &)),
  75                 this, SLOT(context_menu(const QPoint &)));
  76
  77         wpagui = NULL;
  78 }
  79
  80
  81 void Peers::setWpaGui(WpaGui *_wpagui)
  82 {
  83         wpagui = _wpagui;
  84         update_peers();
  85 }
  86
  87
  88 Peers::~Peers()
  89 {
  90         delete default_icon;
  91         delete ap_icon;
  92         delete laptop_icon;
  93 }
  94
  95
  96 void Peers::languageChange()
  97 {
  98         retranslateUi(this);
  99 }
 100
 101
 102 QString Peers::ItemType(int type)
 103 {
 104         QString title;
 105         switch (type) {
 106         case PEER_TYPE_ASSOCIATED_STATION:
 107                 title = tr("Associated station");
 108                 break;
 109         case PEER_TYPE_AP:
 110                 title = tr("AP");
 111                 break;
 112         case PEER_TYPE_AP_WPS:
 113                 title = tr("WPS AP");
 114                 break;
 115         case PEER_TYPE_WPS_PIN_NEEDED:
 116                 title = tr("WPS PIN needed");
 117                 break;
 118         case PEER_TYPE_WPS_ER_AP:
 119                 title = tr("ER: WPS AP");
 120                 break;
 121         case PEER_TYPE_WPS_ER_AP_UNCONFIGURED:
 122                 title = tr("ER: WPS AP (Unconfigured)");
 123                 break;
 124         case PEER_TYPE_WPS_ER_ENROLLEE:
 125                 title = tr("ER: WPS Enrollee");
 126                 break;
 127         case PEER_TYPE_WPS_ENROLLEE:
 128                 title = tr("WPS Enrollee");
 129                 break;
 130         }
 131         return title;
 132 }
 133
 134
 135 void Peers::context_menu(const QPoint &pos)
 136 {
 137         QMenu *menu = new QMenu;
 138         if (menu == NULL)
 139                 return;
 140
 141         QModelIndex idx = peers->indexAt(pos);
 142         if (idx.isValid()) {
 143                 ctx_item = model.itemFromIndex(idx);
 144                 int type = ctx_item->data(peer_role_type).toInt();
 145                 menu->addAction(Peers::ItemType(type))->setEnabled(false);
 146                 menu->addSeparator();
 147
 148                 int config_methods = -1;
 149                 QVariant var = ctx_item->data(peer_role_config_methods);
 150                 if (var.isValid())
 151                         config_methods = var.toInt();
 152
 153                 if ((type == PEER_TYPE_ASSOCIATED_STATION ||
 154                      type == PEER_TYPE_AP_WPS ||
 155                      type == PEER_TYPE_WPS_PIN_NEEDED ||
 156                      type == PEER_TYPE_WPS_ER_ENROLLEE ||
 157                      type == PEER_TYPE_WPS_ENROLLEE) &&
 158                     (config_methods == -1 || (config_methods & 0x010c))) {
 159                         menu->addAction(tr("Enter WPS PIN"), this,
 160                                         SLOT(enter_pin()));
 161                 }
 162
 163                 if (type == PEER_TYPE_AP_WPS) {
 164                         menu->addAction(tr("Connect (PBC)"), this,
 165                                         SLOT(connect_pbc()));
 166                 }
 167
 168                 if ((type == PEER_TYPE_ASSOCIATED_STATION ||
 169                      type == PEER_TYPE_WPS_ER_ENROLLEE ||
 170                      type == PEER_TYPE_WPS_ENROLLEE) &&
 171                     config_methods >= 0 && (config_methods & 0x0080)) {
 172                         menu->addAction(tr("Enroll (PBC)"), this,
 173                                         SLOT(connect_pbc()));
 174                 }
 175
 176                 if (type == PEER_TYPE_WPS_ER_AP) {
 177                         menu->addAction(tr("Learn Configuration"), this,
 178                                         SLOT(learn_ap_config()));
 179                 }
 180
 181                 menu->addAction(tr("Properties"), this, SLOT(properties()));
 182         } else {
 183                 ctx_item = NULL;
 184                 menu->addAction(QString(tr("Refresh")), this,
 185                                 SLOT(ctx_refresh()));
 186         }
 187
 188         menu->exec(peers->mapToGlobal(pos));
 189 }
 190
 191
 192 void Peers::enter_pin()
 193 {
 194         if (ctx_item == NULL)
 195                 return;
 196
 197         int peer_type = ctx_item->data(peer_role_type).toInt();
 198         QString uuid;
 199         QString addr;
 200         if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE)
 201                 uuid = ctx_item->data(peer_role_uuid).toString();
 202         else
 203                 addr = ctx_item->data(peer_role_address).toString();
 204
 205         StringQuery input(tr("PIN:"));
 206         input.setWindowTitle(tr("PIN for ") + ctx_item->text());
 207         if (input.exec() != QDialog::Accepted)
 208                 return;
 209
 210         char cmd[100];
 211         char reply[100];
 212         size_t reply_len;
 213
 214         if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE) {
 215                 snprintf(cmd, sizeof(cmd), "WPS_ER_PIN %s %s",
 216                          uuid.toAscii().constData(),
 217                          input.get_string().toAscii().constData());
 218         } else {
 219                 snprintf(cmd, sizeof(cmd), "WPS_PIN %s %s",
 220                          addr.toAscii().constData(),
 221                          input.get_string().toAscii().constData());
 222         }
 223         reply_len = sizeof(reply) - 1;
 224         if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
 225                 QMessageBox msg;
 226                 msg.setIcon(QMessageBox::Warning);
 227                 msg.setText(tr("Failed to set the WPS PIN."));
 228                 msg.exec();
 229         }
 230 }
 231
 232
 233 void Peers::ctx_refresh()
 234 {
 235         update_peers();
 236 }
 237
 238
 239 void Peers::add_station(QString info)
 240 {
 241         QStringList lines = info.split(QRegExp("\\n"));
 242         QString name;
 243
 244         for (QStringList::Iterator it = lines.begin();
 245              it != lines.end(); it++) {
 246                 int pos = (*it).indexOf('=') + 1;
 247                 if (pos < 1)
 248                         continue;
 249
 250                 if ((*it).startsWith("wpsDeviceName="))
 251                         name = (*it).mid(pos);
 252         }
 253
 254         if (name.isEmpty())
 255                 name = lines[0];
 256
 257         QStandardItem *item = new QStandardItem(*laptop_icon, name);
 258         if (item) {
 259                 item->setData(lines[0], peer_role_address);
 260                 item->setData(PEER_TYPE_ASSOCIATED_STATION,
 261                               peer_role_type);
 262                 item->setData(info, peer_role_details);
 263                 item->setToolTip(ItemType(PEER_TYPE_ASSOCIATED_STATION));
 264                 model.appendRow(item);
 265         }
 266 }
 267
 268
 269 void Peers::add_stations()
 270 {
 271         char reply[2048];
 272         size_t reply_len;
 273         char cmd[30];
 274         int res;
 275
 276         reply_len = sizeof(reply) - 1;
 277         if (wpagui->ctrlRequest("STA-FIRST", reply, &reply_len) < 0)
 278                 return;
 279
 280         do {
 281                 reply[reply_len] = '\0';
 282                 QString info(reply);
 283                 char *txt = reply;
 284                 while (*txt != '\0' && *txt != '\n')
 285                         txt++;
 286                 *txt++ = '\0';
 287                 if (strncmp(reply, "FAIL", 4) == 0 ||
 288                     strncmp(reply, "UNKNOWN", 7) == 0)
 289                         break;
 290
 291                 add_station(info);
 292
 293                 reply_len = sizeof(reply) - 1;
 294                 snprintf(cmd, sizeof(cmd), "STA-NEXT %s", reply);
 295                 res = wpagui->ctrlRequest(cmd, reply, &reply_len);
 296         } while (res >= 0);
 297 }
 298
 299
 300 void Peers::add_single_station(const char *addr)
 301 {
 302         char reply[2048];
 303         size_t reply_len;
 304         char cmd[30];
 305
 306         reply_len = sizeof(reply) - 1;
 307         snprintf(cmd, sizeof(cmd), "STA %s", addr);
 308         if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
 309                 return;
 310
 311         reply[reply_len] = '\0';
 312         QString info(reply);
 313         char *txt = reply;
 314         while (*txt != '\0' && *txt != '\n')
 315                 txt++;
 316         *txt++ = '\0';
 317         if (strncmp(reply, "FAIL", 4) == 0 ||
 318             strncmp(reply, "UNKNOWN", 7) == 0)
 319                 return;
 320
 321         add_station(info);
 322 }
 323
 324
 325 void Peers::remove_bss(int id)
 326 {
 327         if (model.rowCount() == 0)
 328                 return;
 329
 330         QModelIndexList lst = model.match(model.index(0, 0), peer_role_bss_id,
 331                                           id);
 332         if (lst.size() == 0)
 333                 return;
 334         model.removeRow(lst[0].row());
 335 }
 336
 337
 338 bool Peers::add_bss(const char *cmd)
 339 {
 340         char reply[2048];
 341         size_t reply_len;
 342
 343         reply_len = sizeof(reply) - 1;
 344         if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0)
 345                 return false;
 346         reply[reply_len] = '\0';
 347
 348         QString bss(reply);
 349         if (bss.isEmpty() || bss.startsWith("FAIL"))
 350                 return false;
 351
 352         QString ssid, bssid, flags, wps_name, pri_dev_type;
 353         int id = -1;
 354
 355         QStringList lines = bss.split(QRegExp("\\n"));
 356         for (QStringList::Iterator it = lines.begin();
 357              it != lines.end(); it++) {
 358                 int pos = (*it).indexOf('=') + 1;
 359                 if (pos < 1)
 360                         continue;
 361
 362                 if ((*it).startsWith("bssid="))
 363                         bssid = (*it).mid(pos);
 364                 else if ((*it).startsWith("id="))
 365                         id = (*it).mid(pos).toInt();
 366                 else if ((*it).startsWith("flags="))
 367                         flags = (*it).mid(pos);
 368                 else if ((*it).startsWith("ssid="))
 369                         ssid = (*it).mid(pos);
 370                 else if ((*it).startsWith("wps_device_name="))
 371                         wps_name = (*it).mid(pos);
 372                 else if ((*it).startsWith("wps_primary_device_type="))
 373                         pri_dev_type = (*it).mid(pos);
 374         }
 375
 376         QString name = wps_name;
 377         if (name.isEmpty())
 378                 name = ssid + "\n" + bssid;
 379
 380         QStandardItem *item = new QStandardItem(*ap_icon, name);
 381         if (item) {
 382                 item->setData(bssid, peer_role_address);
 383                 if (id >= 0)
 384                         item->setData(id, peer_role_bss_id);
 385                 int type;
 386                 if (flags.contains("[WPS"))
 387                         type = PEER_TYPE_AP_WPS;
 388                 else
 389                         type = PEER_TYPE_AP;
 390                 item->setData(type, peer_role_type);
 391
 392                 for (int i = 0; i < lines.size(); i++) {
 393                         if (lines[i].length() > 60) {
 394                                 lines[i].remove(60, lines[i].length());
 395                                 lines[i] += "..";
 396                         }
 397                 }
 398                 item->setToolTip(ItemType(type));
 399                 item->setData(lines.join("\n"), peer_role_details);
 400                 if (!pri_dev_type.isEmpty())
 401                         item->setData(pri_dev_type,
 402                                       peer_role_pri_dev_type);
 403                 if (!ssid.isEmpty())
 404                         item->setData(ssid, peer_role_ssid);
 405                 model.appendRow(item);
 406         }
 407
 408         return true;
 409 }
 410
 411
 412 void Peers::add_scan_results()
 413 {
 414         int index;
 415         char cmd[20];
 416
 417         index = 0;
 418         while (wpagui) {
 419                 snprintf(cmd, sizeof(cmd), "BSS %d", index++);
 420                 if (index > 1000)
 421                         break;
 422
 423                 if (!add_bss(cmd))
 424                         break;
 425         }
 426 }
 427
 428
 429 void Peers::update_peers()
 430 {
 431         model.clear();
 432         if (wpagui == NULL)
 433                 return;
 434
 435         char reply[20];
 436         size_t replylen = sizeof(reply) - 1;
 437         wpagui->ctrlRequest("WPS_ER_START", reply, &replylen);
 438
 439         add_stations();
 440         add_scan_results();
 441 }
 442
 443
 444 QStandardItem * Peers::find_addr(QString addr)
 445 {
 446         if (model.rowCount() == 0)
 447                 return NULL;
 448
 449         QModelIndexList lst = model.match(model.index(0, 0), peer_role_address,
 450                                           addr);
 451         if (lst.size() == 0)
 452                 return NULL;
 453         return model.itemFromIndex(lst[0]);
 454 }
 455
 456
 457 QStandardItem * Peers::find_uuid(QString uuid)
 458 {
 459         if (model.rowCount() == 0)
 460                 return NULL;
 461
 462         QModelIndexList lst = model.match(model.index(0, 0), peer_role_uuid,
 463                                           uuid);
 464         if (lst.size() == 0)
 465                 return NULL;
 466         return model.itemFromIndex(lst[0]);
 467 }
 468
 469
 470 void Peers::event_notify(WpaMsg msg)
 471 {
 472         QString text = msg.getMsg();
 473
 474         if (text.startsWith(WPS_EVENT_PIN_NEEDED)) {
 475                 /*
 476                  * WPS-PIN-NEEDED 5a02a5fa-9199-5e7c-bc46-e183d3cb32f7
 477                  * 02:2a:c4:18:5b:f3
 478                  * [Wireless Client|Company|cmodel|123|12345|1-0050F204-1]
 479                  */
 480                 QStringList items = text.split(' ');
 481                 QString uuid = items[1];
 482                 QString addr = items[2];
 483                 QString name = "";
 484
 485                 QStandardItem *item = find_addr(addr);
 486                 if (item)
 487                         return;
 488
 489                 int pos = text.indexOf('[');
 490                 if (pos >= 0) {
 491                         int pos2 = text.lastIndexOf(']');
 492                         if (pos2 >= pos) {
 493                                 items = text.mid(pos + 1, pos2 - pos - 1).
 494                                         split('|');
 495                                 name = items[0];
 496                                 items.append(addr);
 497                         }
 498                 }
 499
 500                 item = new QStandardItem(*laptop_icon, name);
 501                 if (item) {
 502                         item->setData(addr, peer_role_address);
 503                         item->setData(PEER_TYPE_WPS_PIN_NEEDED,
 504                                       peer_role_type);
 505                         item->setToolTip(ItemType(PEER_TYPE_WPS_PIN_NEEDED));
 506                         item->setData(items.join("\n"), peer_role_details);
 507                         item->setData(items[5], peer_role_pri_dev_type);
 508                         model.appendRow(item);
 509                 }
 510                 return;
 511         }
 512
 513         if (text.startsWith(AP_STA_CONNECTED)) {
 514                 /* AP-STA-CONNECTED 02:2a:c4:18:5b:f3 */
 515                 QStringList items = text.split(' ');
 516                 QString addr = items[1];
 517                 QStandardItem *item = find_addr(addr);
 518                 if (item == NULL || item->data(peer_role_type).toInt() !=
 519                     PEER_TYPE_ASSOCIATED_STATION)
 520                         add_single_station(addr.toAscii().constData());
 521                 return;
 522         }
 523
 524         if (text.startsWith(AP_STA_DISCONNECTED)) {
 525                 /* AP-STA-DISCONNECTED 02:2a:c4:18:5b:f3 */
 526                 QStringList items = text.split(' ');
 527                 QString addr = items[1];
 528
 529                 if (model.rowCount() == 0)
 530                         return;
 531
 532                 QModelIndexList lst = model.match(model.index(0, 0),
 533                                                   peer_role_address, addr);
 534                 for (int i = 0; i < lst.size(); i++) {
 535                         QStandardItem *item = model.itemFromIndex(lst[i]);
 536                         if (item && item->data(peer_role_type).toInt() ==
 537                             PEER_TYPE_ASSOCIATED_STATION)
 538                                 model.removeRow(lst[i].row());
 539                 }
 540                 return;
 541         }
 542
 543         if (text.startsWith(WPS_EVENT_ER_AP_ADD)) {
 544                 /*
 545                  * WPS-ER-AP-ADD 87654321-9abc-def0-1234-56789abc0002
 546                  * 02:11:22:33:44:55 pri_dev_type=6-0050F204-1 wps_state=1
 547                  * |Very friendly name|Company|Long description of the model|
 548                  * WAP|http://w1.fi/|http://w1.fi/hostapd/
 549                  */
 550                 QStringList items = text.split(' ');
 551                 if (items.size() < 5)
 552                         return;
 553                 QString uuid = items[1];
 554                 QString addr = items[2];
 555                 QString pri_dev_type = items[3].mid(13);
 556                 int wps_state = items[4].mid(10).toInt();
 557
 558                 int pos = text.indexOf('|');
 559                 if (pos < 0)
 560                         return;
 561                 items = text.mid(pos + 1).split('|');
 562                 if (items.size() < 1)
 563                         return;
 564
 565                 QStandardItem *item = find_uuid(uuid);
 566                 if (item)
 567                         return;
 568
 569                 item = new QStandardItem(*ap_icon, items[0]);
 570                 if (item) {
 571                         item->setData(uuid, peer_role_uuid);
 572                         item->setData(addr, peer_role_address);
 573                         int type = wps_state == 2 ? PEER_TYPE_WPS_ER_AP:
 574                                 PEER_TYPE_WPS_ER_AP_UNCONFIGURED;
 575                         item->setData(type, peer_role_type);
 576                         item->setToolTip(ItemType(type));
 577                         item->setData(pri_dev_type, peer_role_pri_dev_type);
 578                         item->setData(items.join(QString("\n")),
 579                                       peer_role_details);
 580                         model.appendRow(item);
 581                 }
 582
 583                 return;
 584         }
 585
 586         if (text.startsWith(WPS_EVENT_ER_AP_REMOVE)) {
 587                 /* WPS-ER-AP-REMOVE 87654321-9abc-def0-1234-56789abc0002 */
 588                 QStringList items = text.split(' ');
 589                 if (items.size() < 2)
 590                         return;
 591                 if (model.rowCount() == 0)
 592                         return;
 593
 594                 QModelIndexList lst = model.match(model.index(0, 0),
 595                                                   peer_role_uuid, items[1]);
 596                 for (int i = 0; i < lst.size(); i++) {
 597                         QStandardItem *item = model.itemFromIndex(lst[i]);
 598                         if (item &&
 599                             (item->data(peer_role_type).toInt() ==
 600                              PEER_TYPE_WPS_ER_AP ||
 601                              item->data(peer_role_type).toInt() ==
 602                              PEER_TYPE_WPS_ER_AP_UNCONFIGURED))
 603                                 model.removeRow(lst[i].row());
 604                 }
 605                 return;
 606         }
 607
 608         if (text.startsWith(WPS_EVENT_ER_ENROLLEE_ADD)) {
 609                 /*
 610                  * WPS-ER-ENROLLEE-ADD 2b7093f1-d6fb-5108-adbb-bea66bb87333
 611                  * 02:66:a0:ee:17:27 M1=1 config_methods=0x14d dev_passwd_id=0
 612                  * pri_dev_type=1-0050F204-1
 613                  * |Wireless Client|Company|cmodel|123|12345|
 614                  */
 615                 QStringList items = text.split(' ');
 616                 if (items.size() < 3)
 617                         return;
 618                 QString uuid = items[1];
 619                 QString addr = items[2];
 620                 QString pri_dev_type = items[6].mid(13);
 621                 int config_methods = -1;
 622                 int dev_passwd_id = -1;
 623
 624                 for (int i = 3; i < items.size(); i++) {
 625                         int pos = items[i].indexOf('=') + 1;
 626                         if (pos < 1)
 627                                 continue;
 628                         QString val = items[i].mid(pos);
 629                         if (items[i].startsWith("config_methods=")) {
 630                                 config_methods = val.toInt(0, 0);
 631                         } else if (items[i].startsWith("dev_passwd_id=")) {
 632                                 dev_passwd_id = val.toInt();
 633                         }
 634                 }
 635
 636                 int pos = text.indexOf('|');
 637                 if (pos < 0)
 638                         return;
 639                 items = text.mid(pos + 1).split('|');
 640                 if (items.size() < 1)
 641                         return;
 642                 QString name = items[0];
 643                 if (name.length() == 0)
 644                         name = addr;
 645
 646                 remove_enrollee_uuid(uuid);
 647
 648                 QStandardItem *item;
 649                 item = new QStandardItem(*laptop_icon, name);
 650                 if (item) {
 651                         item->setData(uuid, peer_role_uuid);
 652                         item->setData(addr, peer_role_address);
 653                         item->setData(PEER_TYPE_WPS_ER_ENROLLEE,
 654                                       peer_role_type);
 655                         item->setToolTip(ItemType(PEER_TYPE_WPS_ER_ENROLLEE));
 656                         item->setData(items.join(QString("\n")),
 657                                       peer_role_details);
 658                         item->setData(pri_dev_type, peer_role_pri_dev_type);
 659                         if (config_methods >= 0)
 660                                 item->setData(config_methods,
 661                                               peer_role_config_methods);
 662                         if (dev_passwd_id >= 0)
 663                                 item->setData(dev_passwd_id,
 664                                               peer_role_dev_passwd_id);
 665                         model.appendRow(item);
 666                 }
 667
 668                 return;
 669         }
 670
 671         if (text.startsWith(WPS_EVENT_ER_ENROLLEE_REMOVE)) {
 672                 /*
 673                  * WPS-ER-ENROLLEE-REMOVE 2b7093f1-d6fb-5108-adbb-bea66bb87333
 674                  * 02:66:a0:ee:17:27
 675                  */
 676                 QStringList items = text.split(' ');
 677                 if (items.size() < 2)
 678                         return;
 679                 remove_enrollee_uuid(items[1]);
 680                 return;
 681         }
 682
 683         if (text.startsWith(WPS_EVENT_ENROLLEE_SEEN)) {
 684                 /* TODO: need to time out this somehow or remove on successful
 685                  * WPS run, etc. */
 686                 /*
 687                  * WPS-ENROLLEE-SEEN 02:00:00:00:01:00
 688                  * 572cf82f-c957-5653-9b16-b5cfb298abf1 1-0050F204-1 0x80 4 1
 689                  * [Wireless Client]
 690                  * (MAC addr, UUID-E, pri dev type, config methods,
 691                  * dev passwd id, request type, [dev name])
 692                  */
 693                 QStringList items = text.split(' ');
 694                 if (items.size() < 7)
 695                         return;
 696                 QString addr = items[1];
 697                 QString uuid = items[2];
 698                 QString pri_dev_type = items[3];
 699                 int config_methods = items[4].toInt(0, 0);
 700                 int dev_passwd_id = items[5].toInt();
 701                 QString name;
 702
 703                 int pos = text.indexOf('[');
 704                 if (pos >= 0) {
 705                         int pos2 = text.lastIndexOf(']');
 706                         if (pos2 >= pos) {
 707                                 QStringList items2 =
 708                                         text.mid(pos + 1, pos2 - pos - 1).
 709                                         split('|');
 710                                 name = items2[0];
 711                         }
 712                 }
 713                 if (name.isEmpty())
 714                         name = addr;
 715
 716                 QStandardItem *item;
 717
 718                 item = find_uuid(uuid);
 719                 if (item) {
 720                         QVariant var = item->data(peer_role_config_methods);
 721                         QVariant var2 = item->data(peer_role_dev_passwd_id);
 722                         if ((var.isValid() && config_methods != var.toInt()) ||
 723                             (var2.isValid() && dev_passwd_id != var2.toInt()))
 724                                 remove_enrollee_uuid(uuid);
 725                         else
 726                                 return;
 727                 }
 728
 729                 item = new QStandardItem(*laptop_icon, name);
 730                 if (item) {
 731                         item->setData(uuid, peer_role_uuid);
 732                         item->setData(addr, peer_role_address);
 733                         item->setData(PEER_TYPE_WPS_ENROLLEE,
 734                                       peer_role_type);
 735                         item->setToolTip(ItemType(PEER_TYPE_WPS_ENROLLEE));
 736                         item->setData(items.join(QString("\n")),
 737                                       peer_role_details);
 738                         item->setData(pri_dev_type, peer_role_pri_dev_type);
 739                         item->setData(config_methods,
 740                                       peer_role_config_methods);
 741                         item->setData(dev_passwd_id, peer_role_dev_passwd_id);
 742                         model.appendRow(item);
 743                 }
 744
 745                 return;
 746         }
 747
 748         if (text.startsWith(WPA_EVENT_BSS_ADDED)) {
 749                 /* CTRL-EVENT-BSS-ADDED 34 00:11:22:33:44:55 */
 750                 QStringList items = text.split(' ');
 751                 if (items.size() < 2)
 752                         return;
 753                 char cmd[20];
 754                 snprintf(cmd, sizeof(cmd), "BSS ID-%d", items[1].toInt());
 755                 add_bss(cmd);
 756                 return;
 757         }
 758
 759         if (text.startsWith(WPA_EVENT_BSS_REMOVED)) {
 760                 /* CTRL-EVENT-BSS-REMOVED 34 00:11:22:33:44:55 */
 761                 QStringList items = text.split(' ');
 762                 if (items.size() < 2)
 763                         return;
 764                 remove_bss(items[1].toInt());
 765                 return;
 766         }
 767 }
 768
 769
 770 void Peers::closeEvent(QCloseEvent *)
 771 {
 772         if (wpagui) {
 773                 char reply[20];
 774                 size_t replylen = sizeof(reply) - 1;
 775                 wpagui->ctrlRequest("WPS_ER_STOP", reply, &replylen);
 776         }
 777 }
 778
 779
 780 void Peers::done(int r)
 781 {
 782         QDialog::done(r);
 783         close();
 784 }
 785
 786
 787 void Peers::remove_enrollee_uuid(QString uuid)
 788 {
 789         if (model.rowCount() == 0)
 790                 return;
 791
 792         QModelIndexList lst = model.match(model.index(0, 0),
 793                                           peer_role_uuid, uuid);
 794         for (int i = 0; i < lst.size(); i++) {
 795                 QStandardItem *item = model.itemFromIndex(lst[i]);
 796                 if (item == NULL)
 797                         continue;
 798                 int type = item->data(peer_role_type).toInt();
 799                 if (type == PEER_TYPE_WPS_ER_ENROLLEE ||
 800                     type == PEER_TYPE_WPS_ENROLLEE)
 801                         model.removeRow(lst[i].row());
 802         }
 803 }
 804
 805
 806 void Peers::properties()
 807 {
 808         if (ctx_item == NULL)
 809                 return;
 810
 811         QMessageBox msg(this);
 812         msg.setStandardButtons(QMessageBox::Ok);
 813         msg.setDefaultButton(QMessageBox::Ok);
 814         msg.setEscapeButton(QMessageBox::Ok);
 815         msg.setWindowTitle(tr("Peer Properties"));
 816
 817         int type = ctx_item->data(peer_role_type).toInt();
 818         QString title = Peers::ItemType(type);
 819
 820         msg.setText(title + QString("\n") + tr("Name: ") + ctx_item->text());
 821
 822         QVariant var;
 823         QString info;
 824
 825         var = ctx_item->data(peer_role_address);
 826         if (var.isValid())
 827                 info += tr("Address: ") + var.toString() + QString("\n");
 828
 829         var = ctx_item->data(peer_role_uuid);
 830         if (var.isValid())
 831                 info += tr("UUID: ") + var.toString() + QString("\n");
 832
 833         var = ctx_item->data(peer_role_pri_dev_type);
 834         if (var.isValid())
 835                 info += tr("Primary Device Type: ") + var.toString() +
 836                         QString("\n");
 837
 838         var = ctx_item->data(peer_role_ssid);
 839         if (var.isValid())
 840                 info += tr("SSID: ") + var.toString() + QString("\n");
 841
 842         var = ctx_item->data(peer_role_config_methods);
 843         if (var.isValid()) {
 844                 int methods = var.toInt();
 845                 info += tr("Configuration Methods: ");
 846                 if (methods & 0x0001)
 847                         info += tr("[USBA]");
 848                 if (methods & 0x0002)
 849                         info += tr("[Ethernet]");
 850                 if (methods & 0x0004)
 851                         info += tr("[Label]");
 852                 if (methods & 0x0008)
 853                         info += tr("[Display]");
 854                 if (methods & 0x0010)
 855                         info += tr("[Ext. NFC Token]");
 856                 if (methods & 0x0020)
 857                         info += tr("[Int. NFC Token]");
 858                 if (methods & 0x0040)
 859                         info += tr("[NFC Interface]");
 860                 if (methods & 0x0080)
 861                         info += tr("[Push Button]");
 862                 if (methods & 0x0100)
 863                         info += tr("[Keypad]");
 864                 info += "\n";
 865         }
 866
 867         var = ctx_item->data(peer_role_dev_passwd_id);
 868         if (var.isValid()) {
 869                 info += tr("Device Password ID: ") + var.toString();
 870                 switch (var.toInt()) {
 871                 case 0:
 872                         info += tr(" (Default PIN)");
 873                         break;
 874                 case 1:
 875                         info += tr(" (User-specified PIN)");
 876                         break;
 877                 case 2:
 878                         info += tr(" (Machine-specified PIN)");
 879                         break;
 880                 case 3:
 881                         info += tr(" (Rekey)");
 882                         break;
 883                 case 4:
 884                         info += tr(" (Push Button)");
 885                         break;
 886                 case 5:
 887                         info += tr(" (Registrar-specified)");
 888                         break;
 889                 }
 890                 info += "\n";
 891         }
 892
 893         msg.setInformativeText(info);
 894
 895         var = ctx_item->data(peer_role_details);
 896         if (var.isValid())
 897                 msg.setDetailedText(var.toString());
 898
 899         msg.exec();
 900 }
 901
 902
 903 void Peers::connect_pbc()
 904 {
 905         if (ctx_item == NULL)
 906                 return;
 907
 908         char cmd[100];
 909         char reply[100];
 910         size_t reply_len;
 911
 912         int peer_type = ctx_item->data(peer_role_type).toInt();
 913         if (peer_type == PEER_TYPE_WPS_ER_ENROLLEE) {
 914                 snprintf(cmd, sizeof(cmd), "WPS_ER_PBC %s",
 915                          ctx_item->data(peer_role_uuid).toString().toAscii().
 916                          constData());
 917         } else {
 918                 snprintf(cmd, sizeof(cmd), "WPS_PBC");
 919         }
 920         reply_len = sizeof(reply) - 1;
 921         if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
 922                 QMessageBox msg;
 923                 msg.setIcon(QMessageBox::Warning);
 924                 msg.setText(tr("Failed to start WPS PBC."));
 925                 msg.exec();
 926         }
 927 }
 928
 929
 930 void Peers::learn_ap_config()
 931 {
 932         if (ctx_item == NULL)
 933                 return;
 934
 935         QString uuid = ctx_item->data(peer_role_uuid).toString();
 936
 937         StringQuery input(tr("AP PIN:"));
 938         input.setWindowTitle(tr("AP PIN for ") + ctx_item->text());
 939         if (input.exec() != QDialog::Accepted)
 940                 return;
 941
 942         char cmd[100];
 943         char reply[100];
 944         size_t reply_len;
 945
 946         snprintf(cmd, sizeof(cmd), "WPS_ER_LEARN %s %s",
 947                  uuid.toAscii().constData(),
 948                  input.get_string().toAscii().constData());
 949         reply_len = sizeof(reply) - 1;
 950         if (wpagui->ctrlRequest(cmd, reply, &reply_len) < 0) {
 951                 QMessageBox msg;
 952                 msg.setIcon(QMessageBox::Warning);
 953                 msg.setText(tr("Failed to start learning AP configuration."));
 954                 msg.exec();
 955         }
 956 }