Skip to content
This repository has been archived by the owner on Feb 12, 2023. It is now read-only.

Commit

Permalink
feat(proxy): provide commandline tools for proxy settings
Browse files Browse the repository at this point in the history
  • Loading branch information
Kribylet authored and anthonybilinski committed Sep 3, 2019
1 parent 2ea5030 commit 31fec74
Show file tree
Hide file tree
Showing 7 changed files with 278 additions and 17 deletions.
113 changes: 111 additions & 2 deletions src/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,96 @@ static QList<QByteArray>* logBuffer =
QMutex* logBufferMutex = new QMutex();
#endif

/**
* Verifies that commandline proxy settings are at least reasonable. Does not verify provided IP
* or hostname addresses are valid. Code duplication with Settings::applyCommandLineOptions, which
* also verifies arguments, should be removed in a future refactor.
* @param parser QCommandLineParser instance
*/
bool verifyProxySettings(QCommandLineParser& parser)
{
QString IPv6Setting = parser.value("I");
QString LANSetting = parser.value("L");
QString UDPSetting = parser.value("U");
QString proxySettingString = parser.value("proxy");
QStringList proxySettings = proxySettingString.split(":");
// Check for incompatible settings
bool activeProxyType = false;

if (parser.isSet("P")) {
activeProxyType = proxySettings[0].compare(QString("SOCKS5"), Qt::CaseInsensitive) == 0
|| proxySettings[0].compare(QString("HTTP"), Qt::CaseInsensitive) == 0;
}


if (activeProxyType && (UDPSetting.compare(QString("on"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Cannot set UDP on with proxy.";
return false;
}

if (activeProxyType && (LANSetting.compare(QString("on"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Cannot set LAN discovery on with proxy.";
return false;
}

if ((LANSetting.compare(QString("on"), Qt::CaseInsensitive) == 0)
&& (UDPSetting.compare(QString("off"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Incompatible UDP/LAN settings.";
return false;
}

if (parser.isSet("I")) {
if (!(IPv6Setting.compare(QString("on"), Qt::CaseInsensitive) == 0
|| IPv6Setting.compare(QString("off"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Unable to parse IPv6 setting.";
return false;
}
}

if (parser.isSet("U")) {
if (!(UDPSetting.compare(QString("on"), Qt::CaseInsensitive) == 0
|| UDPSetting.compare(QString("off"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Unable to parse UDP setting.";
return false;
}
}

if (parser.isSet("L")) {
if (!(LANSetting.compare(QString("on"), Qt::CaseInsensitive) == 0
|| LANSetting.compare(QString("off"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Unable to parse LAN setting.";
return false;
}
}

if (parser.isSet("P")) {
if (proxySettings[0].compare(QString("NONE"), Qt::CaseInsensitive) == 0) {
return true;
// slightly lazy check here, accepting 'NONE[:.*]' is fine since no other
// arguments will be investigated when proxy settings are applied.
}
// Since the first argument isn't 'none', verify format of remaining arguments
if (proxySettings.size() != 3) {
qCritical() << "Invalid number of proxy arguments.";
return false;
}

if (!(proxySettings[0].compare(QString("SOCKS5"), Qt::CaseInsensitive) == 0
|| proxySettings[0].compare(QString("HTTP"), Qt::CaseInsensitive) == 0)) {
qCritical() << "Unable to parse proxy type.";
return false;
}

// Kriby: Sanity checking IPv4+IPv6/hostnames sure is a lot of work!

int portNumber = proxySettings[2].toInt();
if (!(portNumber > 0 && portNumber < 65536)) {
qCritical() << "Invalid port number range.";
}
}
return true;
}

void cleanup()
{
// force save early even though destruction saves, because Windows OS will
Expand Down Expand Up @@ -215,6 +305,21 @@ int main(int argc, char* argv[])
QCommandLineOption(QStringList() << "l"
<< "login",
QObject::tr("Starts new instance and opens the login screen.")));
parser.addOption(QCommandLineOption(QStringList() << "I"
<< "IPv6",
QObject::tr("Sets IPv6 <on>/<off>"), QObject::tr("on/off")));
parser.addOption(QCommandLineOption(QStringList() << "U"
<< "UDP",
QObject::tr("Sets UDP <on>/<off>"), QObject::tr("on/off")));
parser.addOption(
QCommandLineOption(QStringList() << "L"
<< "LAN",
QObject::tr("Sets LAN discovery <on>/<off>. UDP off overrides."),
QObject::tr("on/off")));
parser.addOption(QCommandLineOption(QStringList() << "P"
<< "proxy",
QObject::tr("Sets proxy settings."),
QObject::tr("(SOCKS5/HTTP/NONE):(ADDRESS):(PORT)")));
parser.process(*a);

uint32_t profileId = settings.getCurrentProfileId();
Expand Down Expand Up @@ -330,6 +435,10 @@ int main(int argc, char* argv[])
}
}

if (!verifyProxySettings(parser)) {
return -1;
}

// TODO(sudden6): remove once we get rid of Nexus
Nexus& nexus = Nexus::getInstance();
// TODO(kriby): Consider moving application initializing variables into a globalSettings object
Expand All @@ -342,16 +451,16 @@ int main(int argc, char* argv[])
// Further: generate view instances separately (loginScreen, mainGUI, audio)
Profile* profile = nullptr;
if (autoLogin && Profile::exists(profileName) && !Profile::isEncrypted(profileName)) {
profile = Profile::loadProfile(profileName);
profile = Profile::loadProfile(profileName, &parser);
if (!profile) {
QMessageBox::information(nullptr, QObject::tr("Error"),
QObject::tr("Failed to load profile automatically."));
}
}
if (profile) {
settings.updateProfileData(profile);
nexus.bootstrapWithProfile(profile);
} else {
nexus.setParser(&parser);
int returnval = nexus.showLogin(profileName);
if (returnval != 0) {
return returnval;
Expand Down
18 changes: 11 additions & 7 deletions src/nexus.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
#include "widget/gui.h"
#include "widget/loginscreen.h"
#include <QApplication>
#include <QCommandLineParser>
#include <QDebug>
#include <QDesktopWidget>
#include <QThread>
Expand Down Expand Up @@ -183,14 +184,10 @@ void Nexus::bootstrapWithProfile(Profile* p)
void Nexus::setSettings(Settings* settings)
{
if (this->settings) {
QObject::disconnect(this, &Nexus::currentProfileChanged, this->settings,
&Settings::updateProfileData);
QObject::disconnect(this, &Nexus::saveGlobal, this->settings, &Settings::saveGlobal);
}
this->settings = settings;
if (this->settings) {
QObject::connect(this, &Nexus::currentProfileChanged, this->settings,
&Settings::updateProfileData);
QObject::connect(this, &Nexus::saveGlobal, this->settings, &Settings::saveGlobal);
}
}
Expand Down Expand Up @@ -292,15 +289,17 @@ Profile* Nexus::getProfile()
*/
void Nexus::onCreateNewProfile(const QString& name, const QString& pass)
{
setProfile(Profile::createProfile(name, pass));
setProfile(Profile::createProfile(name, parser, pass));
parser = nullptr; // only apply cmdline proxy settings once
}

/**
* Loads an existing profile and replaces the current one.
*/
void Nexus::onLoadProfile(const QString& name, const QString& pass)
{
setProfile(Profile::loadProfile(name, pass));
setProfile(Profile::loadProfile(name, parser, pass));
parser = nullptr; // only apply cmdline proxy settings once
}
/**
* Changes the loaded profile and notifies listeners.
Expand All @@ -319,6 +318,11 @@ void Nexus::setProfile(Profile* p)
emit currentProfileChanged(p);
}

void Nexus::setParser(QCommandLineParser* parser)
{
this->parser = parser;
}

/**
* @brief Get desktop GUI widget.
* @return nullptr if not started, desktop widget otherwise.
Expand Down Expand Up @@ -389,7 +393,7 @@ void Nexus::updateWindowsArg(QWindow* closedWindow)
QAction* action = windowActions->addAction(windowList[i]->title());
action->setCheckable(true);
action->setChecked(windowList[i] == activeWindow);
connect(action, &QAction::triggered, [=] { onOpenWindow(windowList[i]);});
connect(action, &QAction::triggered, [=] { onOpenWindow(windowList[i]); });
windowMenu->addAction(action);
dockMenu->insertAction(dockLast, action);
}
Expand Down
5 changes: 4 additions & 1 deletion src/nexus.h
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ class Profile;
class Settings;
class LoginScreen;
class Core;
class QCommandLineParser;

#ifdef Q_OS_MAC
class QMenuBar;
Expand All @@ -47,6 +48,7 @@ class Nexus : public QObject
void start();
void showMainGUI();
void setSettings(Settings* settings);
void setParser(QCommandLineParser* parser);
static Nexus& getInstance();
static void destroyInstance();
static Core* getCore();
Expand Down Expand Up @@ -89,7 +91,7 @@ public slots:
void onCreateNewProfile(const QString& name, const QString& pass);
void onLoadProfile(const QString& name, const QString& pass);
int showLogin(const QString& profileName = QString());
void bootstrapWithProfile(Profile *p);
void bootstrapWithProfile(Profile* p);

private:
explicit Nexus(QObject* parent = nullptr);
Expand All @@ -102,6 +104,7 @@ public slots:
Settings* settings;
Widget* widget;
std::unique_ptr<IAudioControl> audioControl;
QCommandLineParser* parser = nullptr;
};

#endif // NEXUS_H
9 changes: 6 additions & 3 deletions src/persistence/profile.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ Profile::Profile(QString name, const QString& password, bool isNewProfile,
*
* @example If the profile is already in use return nullptr.
*/
Profile* Profile::loadProfile(QString name, const QString& password)
Profile* Profile::loadProfile(QString name, const QCommandLineParser* parser, const QString& password)
{
if (ProfileLocker::hasLock()) {
qCritical() << "Tried to load profile " << name << ", but another profile is already locked!";
Expand All @@ -140,7 +140,8 @@ Profile* Profile::loadProfile(QString name, const QString& password)
Profile* p = nullptr;
qint64 fileSize = 0;

QString path = Settings::getInstance().getSettingsDirPath() + name + ".tox";
Settings& s = Settings::getInstance();
QString path = s.getSettingsDirPath() + name + ".tox";
QFile saveFile(path);
qDebug() << "Loading tox save " << path;

Expand Down Expand Up @@ -186,6 +187,8 @@ Profile* Profile::loadProfile(QString name, const QString& password)

saveFile.close();
p = new Profile(name, password, false, data, std::move(tmpKey));

s.updateProfileData(p, parser);
return p;

// cleanup in case of error
Expand All @@ -203,7 +206,7 @@ Profile* Profile::loadProfile(QString name, const QString& password)
*
* @note If the profile is already in use return nullptr.
*/
Profile* Profile::createProfile(QString name, QString password)
Profile* Profile::createProfile(QString name, const QCommandLineParser* parser, QString password)
{
std::unique_ptr<ToxEncrypt> tmpKey;
if (!password.isEmpty()) {
Expand Down
7 changes: 5 additions & 2 deletions src/persistence/profile.h
Original file line number Diff line number Diff line change
Expand Up @@ -34,13 +34,16 @@
#include <QVector>
#include <memory>

class Settings;
class QCommandLineParser;

class Profile : public QObject
{
Q_OBJECT

public:
static Profile* loadProfile(QString name, const QString& password = QString());
static Profile* createProfile(QString name, QString password);
static Profile* loadProfile(QString name, const QCommandLineParser* parser, const QString& password = QString());
static Profile* createProfile(QString name, const QCommandLineParser* parser, QString password);
~Profile();

Core* getCore();
Expand Down
Loading

0 comments on commit 31fec74

Please sign in to comment.