Correct the logic for incorrect passwords. Looks like spaghetti. 🍝

pull/2/head
Simon Quigley 11 months ago
parent c6b13dd919
commit a757d52244

@ -12,7 +12,6 @@
#include <QScreen>
#include <QMessageBox>
#include <QUuid>
#include <QLineEdit>
#include <QDBusPendingReply>
#include "installerprompt.h"
#include "./ui_installerprompt.h"
@ -22,6 +21,9 @@ InstallerPrompt::InstallerPrompt(QWidget *parent)
, ui(new Ui::InstallerPrompt) {
ui->setupUi(this);
// Hide the Incorrect Password text
ui->incorrectPassword->setVisible(false);
// Set the background image and scale it
QPixmap bg(":/background");
if (bg.isNull()) {
@ -78,6 +80,8 @@ void InstallerPrompt::updateConnectionStatus() {
switch (status) {
case NetworkManager::Status::Disconnected:
case NetworkManager::ConnectedLinkLocal:
case NetworkManager::Asleep:
statusText = tr("Not Connected");
statusIndicator = "<span style=\"color: red;\">❌</span> " + statusText;
break;
@ -95,6 +99,7 @@ void InstallerPrompt::updateConnectionStatus() {
statusIndicator = "<span style=\"color: yellow;\">🟡</span> " + statusText;
break;
default:
qDebug() << "Unknown status:" << status;
statusText = tr("Unknown Status");
statusIndicator = "<span style=\"color: grey;\">⚪</span> " + statusText;
}
@ -120,162 +125,154 @@ void InstallerPrompt::updateConnectionStatus() {
void InstallerPrompt::handleWiFiConnectionChange(NetworkManager::Device::State newstate, NetworkManager::Device::State oldstate, NetworkManager::Device::StateChangeReason reason)
{
if (reason == NetworkManager::Device::NoSecretsReason) {
QMutexLocker locker(&wifiChangeMutex);
if (reason == NetworkManager::Device::NoSecretsReason && !wifiWrongHandling) {
wifiWrongHandling = true;
qDebug() << wifiSSID;
foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
if (wirelessSetting && wirelessSetting->ssid() == ui->networkComboBox->currentText()) {
qDebug() << "Wiping connection with wrong password: " << ui->networkComboBox->currentText();
if (wirelessSetting && wirelessSetting->ssid() == wifiSSID) {
qDebug() << "Wiping connection with wrong password: " << wifiSSID;
// Show the Incorrect Password text
ui->incorrectPassword->setVisible(true);
QDBusPendingReply removeReply = connection->remove();
removeReply.waitForFinished();
handleWifiConnection(ui->networkComboBox->currentText(), true);
}
}
}
wifiWrongHandling = false;
}
}
void InstallerPrompt::handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword) {
ui->networkComboBox->setEnabled(false);
qDebug() << "Attempting to find connection for SSID:" << ssid;
if (!recoverFromWrongPassword) {
foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
if (wirelessSetting && wirelessSetting->ssid() == ssid) {
qDebug() << "Attempting to use existing connection:" << ssid;
NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
qDebug() << "Successfully connected:" << ssid;
ui->networkComboBox->setEnabled(true);
return;
}
NetworkManager::Connection::Ptr InstallerPrompt::findConnectionBySsid(const QString &ssid) {
foreach (const NetworkManager::Connection::Ptr &connection, NetworkManager::listConnections()) {
if (connection->settings()->connectionType() == NetworkManager::ConnectionSettings::Wireless) {
auto wirelessSetting = connection->settings()->setting(NetworkManager::Setting::Wireless).dynamicCast<NetworkManager::WirelessSetting>();
if (wirelessSetting && wirelessSetting->ssid() == ssid) {
return connection;
}
}
}
return NetworkManager::Connection::Ptr(); // Return null pointer if not found
}
QString InstallerPrompt::promptForWifiPassword(const QString &ssid, bool isWrongPassword) {
QDialog passwordDialog(this);
passwordDialog.setModal(true);
passwordDialog.setWindowTitle(tr("Wi-Fi Password Required"));
passwordDialog.setWindowIcon(QIcon::fromTheme("network-wireless"));
passwordDialog.setStyleSheet("QLabel { color: black; } ");
passwordDialog.setMinimumWidth(250);
passwordDialog.setMinimumHeight(120);
passwordDialog.setMaximumWidth(5000);
passwordDialog.setMaximumHeight(500);
QVBoxLayout layout;
QLabel passwordLabel(tr("Enter password for \"%1\":").arg(ssid), &passwordDialog);
QLineEdit passwordLineEdit(&passwordDialog);
QPushButton passwordButton(tr("Connect"), &passwordDialog);
passwordLineEdit.setEchoMode(QLineEdit::Password);
layout.addWidget(&passwordLabel);
layout.addWidget(&passwordLineEdit);
layout.addWidget(&passwordButton);
passwordDialog.setLayout(&layout);
// Connect with a lambda function for inline validation
connect(&passwordLineEdit, &QLineEdit::textChanged, this, [&passwordLineEdit](const QString &text) {
int minLength = 8;
int maxLength = 64;
bool isValid = text.length() >= minLength && text.length() <= maxLength;
passwordLineEdit.setStyleSheet(isValid ? "" : "border: 1px solid red;");
});
QMap<QString, QVariant> fullSettings = createSettingsBySSID(ssid);
NMVariantMapMap nmMap;
connect(&passwordButton, &QPushButton::clicked, &passwordDialog, &QDialog::accept);
for (const auto &key : fullSettings.keys()) {
nmMap[key] = fullSettings[key].toMap();
if (passwordDialog.exec() == QDialog::Accepted) {
return passwordLineEdit.text();
}
NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
newConnectionSettings->fromMap(nmMap);
QVariantMap wirelessSecurity = fullSettings.value("802-11-wireless-security").toMap();
return QString();
}
//bool isOpenNetwork = wirelessSecurity.isEmpty() || wirelessSecurity.value("key-mgmt").toString().isEmpty();
//if (isOpenNetwork && wifiDevice && wifiDevice->isValid() && connection) {
// qDebug() << "Attempting to connect to open network: " << ssid;
// NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
// return;
//}
void InstallerPrompt::handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword) {
ui->incorrectPassword->setVisible(false);
ui->networkComboBox->setEnabled(false);
wifiSSID = ssid;
qDebug() << "Attempting to find connection for SSID:" << ssid;
// If the network is secured, display the password dialog
QDialog passwordDialog(this);
QVBoxLayout layout(&passwordDialog);
QLabel label(&passwordDialog);
if (!recoverFromWrongPassword) {
label.setText(tr("Enter Wi-Fi Password for %1:").arg(ssid));
label.setStyleSheet("color: black");
} else {
label.setText(tr("Wrong Wi-Fi password for %1. Try again:").arg(ssid));
label.setStyleSheet("color: red");
// Check for existing connection
NetworkManager::Connection::Ptr connection = findConnectionBySsid(ssid);
if (connection && !recoverFromWrongPassword) {
qDebug() << "Using existing connection for:" << ssid;
NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
ui->networkComboBox->setEnabled(true);
return;
}
QLineEdit lineEdit(&passwordDialog);
QPushButton button(tr("Connect"), &passwordDialog);
lineEdit.setEchoMode(QLineEdit::Password);
layout.addWidget(&label);
layout.addWidget(&lineEdit);
layout.addWidget(&button);
// Prompt for Wi-Fi password
QString password = promptForWifiPassword(ssid);
if (password.isEmpty()) {
ui->networkComboBox->setEnabled(true);
return;
}
connect(&button, &QPushButton::clicked, &passwordDialog, &QDialog::accept);
// Create new Wi-Fi connection
NMVariantMapMap settings = createSettingsBySSID(ssid);
if (settings.isEmpty()) {
QMessageBox::warning(this, tr("Error"), tr("Failed to create Wi-Fi settings."));
ui->networkComboBox->setEnabled(true);
return;
}
for (int attempts = 0; attempts < 3; ++attempts) {
if (passwordDialog.exec() == QDialog::Rejected) {
ui->networkComboBox->setEnabled(true);
return;
}
QString password = lineEdit.text();
if (wifiDevice && wifiDevice->isValid()) {
// Update the wireless security settings in the map
wirelessSecurity["key-mgmt"] = "wpa-psk";
wirelessSecurity["psk"] = password;
fullSettings["802-11-wireless-security"] = wirelessSecurity;
// Convert QMap<QString, QVariant> to NMVariantMapMap
NMVariantMapMap nmMap;
for (const auto &key : fullSettings.keys()) {
nmMap[key] = fullSettings[key].toMap();
}
// Update the wireless security settings
QVariantMap wirelessSecurity;
wirelessSecurity["key-mgmt"] = "wpa-psk";
wirelessSecurity["psk"] = password;
settings["802-11-wireless-security"] = wirelessSecurity;
// Add the new connection
QDBusPendingReply<QDBusObjectPath> reply = NetworkManager::addConnection(settings);
reply.waitForFinished();
if (reply.isError()) {
QMessageBox::warning(this, tr("Error"), tr("Failed to add Wi-Fi connection."));
ui->networkComboBox->setEnabled(true);
return;
}
// Update the connection settings
qDebug() << "Saving the connection...";
QDBusObjectPath path;
NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
newConnectionSettings->fromMap(nmMap);
QDBusPendingReply<QDBusObjectPath> addreply = NetworkManager::addConnection(nmMap);
addreply.waitForFinished();
if (addreply.isError()) {
qDebug() << nmMap;
qDebug() << "Unable to save the connection:" << addreply.error().message();
} else {
path = addreply.value();
qDebug() << "Added connection path:" << path.path();
}
// Activate the new connection
QDBusObjectPath path = reply.value();
NetworkManager::activateConnection(path.path(), wifiDevice->uni(), QString());
ui->networkComboBox->setEnabled(true);
}
NetworkManager::Connection::Ptr connection = NetworkManager::findConnection(path.path());
if (!connection) {
qDebug() << "Unable to retrieve the connection after saving:" << addreply.error().message();
}
NMVariantMapMap InstallerPrompt::createSettingsBySSID(const QString &ssid) {
NMVariantMapMap convertedSettings;
QDBusPendingReply<QDBusObjectPath> reply = NetworkManager::activateConnection(connection->path(), wifiDevice->uni(), QString());
reply.waitForFinished();
if (reply.isError()) {
qDebug() << "Unable to activate the connection:" << addreply.error().message();
QMessageBox::warning(this, tr("Connection Failed"), tr("Unable to connect to the network."));
ui->networkComboBox->setEnabled(true);
return;
} else {
NetworkManager::reloadConnections();
qDebug() << "Successfully connected:" << ssid;
// ui->networkComboBox->setEnabled(true); !!! We don't run this here since we actually *want* the box to remain locked right now
return;
}
}
if (!wifiDevice) {
qWarning() << "Wi-Fi device not found. Unable to set interface name.";
return convertedSettings;
}
}
QMap<QString, QVariant> InstallerPrompt::createSettingsBySSID(const QString &ssid) {
// Create new connection settings
NetworkManager::ConnectionSettings::Ptr newConnectionSettings(new NetworkManager::ConnectionSettings(NetworkManager::ConnectionSettings::Wireless));
newConnectionSettings->setId(ssid);
newConnectionSettings->setUuid(NetworkManager::ConnectionSettings::createNewUuid());
// Set interface name from wifiDevice
if (wifiDevice) {
newConnectionSettings->setInterfaceName(wifiDevice->interfaceName());
} else {
qWarning() << "Wi-Fi device not found. Unable to set interface name.";
return QMap<QString, QVariant>();
}
newConnectionSettings->setInterfaceName(wifiDevice->interfaceName());
// Configure wireless settings
QVariantMap wirelessSetting;
wirelessSetting.insert("ssid", ssid.toUtf8());
convertedSettings.insert("802-11-wireless", wirelessSetting);
// Configure wireless security settings
NetworkManager::WirelessSecuritySetting::Ptr wirelessSecuritySetting = newConnectionSettings->setting(NetworkManager::Setting::WirelessSecurity).staticCast<NetworkManager::WirelessSecuritySetting>();
wirelessSecuritySetting->setKeyMgmt(NetworkManager::WirelessSecuritySetting::WpaPsk);
// Convert settings to QVariantMap
QMap<QString, QVariant> convertedSettings;
// Convert other settings
const auto settingsMap = newConnectionSettings->toMap();
for (const auto &key : settingsMap.keys()) {
convertedSettings[key] = QVariant::fromValue(settingsMap[key]);
QVariant value = settingsMap.value(key);
convertedSettings.insert(key, value.toMap());
}
convertedSettings.insert("802-11-wireless", wirelessSetting);
return convertedSettings;
}
@ -334,7 +331,6 @@ void InstallerPrompt::refreshNetworkList() {
}
// Update the main map and combo box only after the new list is ready
wifiNetworkMap.swap(tempWifiNetworkMap);
ui->networkComboBox->clear();
ui->networkComboBox->addItems(ssidList);

@ -7,16 +7,12 @@
#include <QPushButton>
#include <QLabel>
#include <QDialog>
#include <QMutex>
#include <QLineEdit>
#include <NetworkManagerQt/Device>
#include <NetworkManagerQt/WirelessDevice>
#include <NetworkManagerQt/WirelessNetwork>
namespace NetworkManager {
class Device;
class WirelessDevice;
class WirelessNetwork;
}
namespace Ui { class InstallerPrompt; }
class InstallerPrompt : public QMainWindow {
@ -39,14 +35,19 @@ private:
Ui::InstallerPrompt *ui;
QProcess *process;
NetworkManager::WirelessDevice::Ptr wifiDevice;
QMap<QString, NetworkManager::WirelessNetwork::Ptr> wifiNetworkMap;
QString wifiSSID;
QMutex wifiChangeMutex;
NetworkManager::Connection::Ptr findConnectionBySsid(const QString &ssid);
bool wifiWrongHandling = false;
QLineEdit *passwordLineEdit;
void handleWifiConnection(const QString &ssid, bool recoverFromWrongPassword = false);
QString promptForWifiPassword(const QString &ssid, bool isWrongPassword = false);
void connectToWifi(const QString &ssid, const QString &password, bool recoverFromWrongPassword = false);
void initLanguageComboBox();
QStringList getAvailableLanguages() const;
void showWifiOptions();
NetworkManager::Connection::Ptr findConnectionBySsid(const QString &ssid);
QMap<QString, QVariant> createSettingsBySSID(const QString &ssid);
NMVariantMapMap createSettingsBySSID(const QString &ssid);
};
#endif // INSTALLERPROMPT_H

@ -59,7 +59,10 @@ QLabel#logoLabel {
image: url(:/logo);
background-color: transparent;
}
</string>
QLabel#incorrectPassword {
color: red;
}</string>
</property>
<widget class="QWidget" name="centralwidget">
<property name="sizePolicy">
@ -128,6 +131,7 @@ QLabel#logoLabel {
<property name="font">
<font>
<family>Ubuntu</family>
<weight>50</weight>
<italic>false</italic>
<bold>false</bold>
</font>
@ -200,6 +204,7 @@ QLabel#logoLabel {
<property name="font">
<font>
<pointsize>18</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
@ -258,80 +263,61 @@ QLabel#logoLabel {
</spacer>
</item>
<item>
<layout class="QHBoxLayout" name="WiFiLayout">
<item>
<spacer name="networkSpacer1">
<property name="orientation">
<enum>Qt::Horizontal</enum>
<layout class="QGridLayout" name="WiFiLayout">
<item row="1" column="3">
<widget class="QPushButton" name="connectWiFiButton">
<property name="minimumSize">
<size>
<width>150</width>
<height>65</height>
</size>
</property>
<property name="sizeHint" stdset="0">
<property name="maximumSize">
<size>
<width>40</width>
<height>20</height>
<width>150</width>
<height>65</height>
</size>
</property>
</spacer>
</item>
<item>
<widget class="QLabel" name="WiFiLabel">
<property name="font">
<font>
<pointsize>18</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
<property name="text">
<string>Select a Wi-Fi Network:</string>
<string>Connect</string>
</property>
</widget>
</item>
<item>
<widget class="QComboBox" name="networkComboBox">
<property name="minimumSize">
<size>
<width>352</width>
<height>50</height>
</size>
</property>
<property name="maximumSize">
<size>
<width>400</width>
<height>50</height>
</size>
</property>
<item row="2" column="2">
<widget class="QLabel" name="WiFiInfoLabel">
<property name="font">
<font>
<pointsize>16</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
<property name="text">
<string>(For advanced network configuration, select &quot;Try Lubuntu&quot;)</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="connectWiFiButton">
<property name="minimumSize">
<size>
<width>100</width>
<height>65</height>
</size>
</property>
<property name="maximumSize">
<size>
<width>16777215</width>
<height>65</height>
</size>
</property>
<item row="1" column="1">
<widget class="QLabel" name="WiFiLabel">
<property name="font">
<font>
<pointsize>18</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
<property name="text">
<string>Connect</string>
<string>Select a Wi-Fi Network:</string>
</property>
</widget>
</item>
<item>
<spacer name="networkSpacer2">
<item row="2" column="3">
<spacer name="advancedSpacer_2">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
@ -343,12 +329,8 @@ QLabel#logoLabel {
</property>
</spacer>
</item>
</layout>
</item>
<item>
<layout class="QHBoxLayout" name="WiFiLayout_2">
<item>
<spacer name="advancedSpacer">
<item row="1" column="4">
<spacer name="networkSpacer2">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
@ -360,20 +342,21 @@ QLabel#logoLabel {
</property>
</spacer>
</item>
<item>
<widget class="QLabel" name="WiFiInfoLabel">
<property name="font">
<font>
<bold>true</bold>
</font>
<item row="1" column="0">
<spacer name="networkSpacer1">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
<property name="text">
<string>(For advanced network configuration, select &quot;Try Lubuntu&quot;)</string>
<property name="sizeHint" stdset="0">
<size>
<width>40</width>
<height>20</height>
</size>
</property>
</widget>
</spacer>
</item>
<item>
<spacer name="advancedSpacer_2">
<item row="2" column="1">
<spacer name="advancedSpacer">
<property name="orientation">
<enum>Qt::Horizontal</enum>
</property>
@ -385,6 +368,44 @@ QLabel#logoLabel {
</property>
</spacer>
</item>
<item row="1" column="2">
<widget class="QComboBox" name="networkComboBox">
<property name="minimumSize">
<size>
<width>352</width>
<height>50</height>
</size>
</property>
<property name="maximumSize">
<size>
<width>550</width>
<height>50</height>
</size>
</property>
<property name="font">
<font>
<pointsize>16</pointsize>
</font>
</property>
</widget>
</item>
<item row="0" column="2">
<widget class="QLabel" name="incorrectPassword">
<property name="enabled">
<bool>true</bool>
</property>
<property name="font">
<font>
<pointsize>16</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
<property name="text">
<string>You entered an incorrect password. Please try again.</string>
</property>
</widget>
</item>
</layout>
</item>
<item>
@ -459,6 +480,7 @@ QLabel#logoLabel {
<font>
<family>Ubuntu</family>
<pointsize>24</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
@ -535,6 +557,7 @@ QToolTip {
<font>
<family>Ubuntu</family>
<pointsize>24</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>
@ -677,6 +700,7 @@ QToolTip {
<property name="font">
<font>
<pointsize>24</pointsize>
<weight>75</weight>
<bold>true</bold>
</font>
</property>

Loading…
Cancel
Save