netctl-gui/sources/helper/src/main.cpp

/***************************************************************************
 *   This file is part of netctl-gui                                       *
 *                                                                         *
 *   netctl-gui is free software: you can redistribute it and/or           *
 *   modify it under the terms of the GNU General Public License as        *
 *   published by the Free Software Foundation, either version 3 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   netctl-gui is distributed in the hope that it will be useful,         *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with netctl-gui. If not, see http://www.gnu.org/licenses/       *
 ***************************************************************************/


#include <QCoreApplication>
#include <QDBusConnection>
#include <QDBusMessage>
#include <QDir>
#include <QLibraryInfo>
#include <QProcessEnvironment>
#include <QTranslator>
#include <iostream>
#include <unistd.h>

#include <language/language.h>

#include "messages.h"
#include "netctlhelper.h"
#include "version.h"

#if QT_VERSION >= 0x050000
#include <pdebug/pdebug-time.h>
#endif /* QT_VERSION >= 0x050000 */


using namespace std;


bool existingSessionOperation(const QString operation)
{
    QDBusConnection bus = QDBusConnection::systemBus();
    QDBusMessage request = QDBusMessage::createMethodCall(DBUS_HELPER_SERVICE, DBUS_CTRL_PATH,
                                                          DBUS_HELPER_INTERFACE, operation);
    QDBusMessage response = bus.call(request);
    QList<QVariant> arguments = response.arguments();

    return (!arguments.isEmpty() && arguments[0].toBool());
}


int main(int argc, char *argv[])
{
    QVariantMap args = getArgs();
    // reading
    for (int i=1; i<argc; i++) {
        if ((QString(argv[i]) == QString("-c")) || (QString(argv[i]) == QString("--config"))) {
            // config path
            args[QString("config")] = QDir().absoluteFilePath(argv[i+1]);
            i++;
        } else if ((QString(argv[i]) == QString("-d")) || (QString(argv[i]) == QString("--debug"))) {
            // debug
            args[QString("debug")] = true;
        } else if (QString(argv[i]) == QString("--nodaemon")) {
            // daemonized
            args[QString("nodaemon")] = true;
        } else if (QString(argv[i]) == QString("--replace")) {
            // replace
            args[QString("state")] = (int) 1;
        } else if (QString(argv[i]) == QString("--restore")) {
            // restore
            args[QString("state")] = (int) 2;
        } else if (QString(argv[i]) == QString("--session")) {
            // session
            args[QString("session")] = true;
        } else if (QString(argv[i]) == QString("--system")) {
            // system
            args[QString("system")] = true;
        } else if ((QString(argv[i]) == QString("-h")) || (QString(argv[i]) == QString("--help"))) {
            // help message
            args[QString("help")] = true;
        } else if ((QString(argv[i]) == QString("-i")) || (QString(argv[i]) == QString("--info"))) {
            // info message
            args[QString("info")] = true;
        } else if ((QString(argv[i]) == QString("-v")) || (QString(argv[i]) == QString("--version"))) {
            // version message
            args[QString("version")] = true;
        } else {
            args[QString("error")] = true;
        }
    }
    QProcessEnvironment environment = QProcessEnvironment::systemEnvironment();
    QString debugEnv = environment.value(QString("DEBUG"), QString("no"));
    if (!args[QString("debug")].toBool()) args[QString("debug")] = (debugEnv == QString("yes"));
    if ((args[QString("debug")].toBool()) ||
            (args[QString("help")].toBool()) ||
            (args[QString("info")].toBool()) ||
            (args[QString("version")].toBool()) ||
            (args[QString("error")].toBool()))
        args[QString("nodaemon")] = true;
    // check euid
    if (geteuid() != 0) {
        cout << QCoreApplication::translate("NetctlHelper", "The helper is running with EUID %1. Some functions may not be available.")
                .arg(QString::number(geteuid())).toUtf8().data() << endl;
        cout << QCoreApplication::translate("NetctlHelper", "See security notes for more details.")
                .toUtf8().data() << endl;
        args[QString("session")] = true;
    }

    // check
    QString robotEnv = environment.value(QString("SKYNET"), QString(""));
    if (!robotEnv.isEmpty()) {
        cout << "Sorry, this software is only for humans" << endl;
        return 2;
    }

#if QT_VERSION >= 0x050000
    QCoreApplication::setSetuidAllowed(true);
    qInstallMessageHandler(debugString);
#endif
    QCoreApplication a(argc, argv);
    // reread translations according to flags
    QString language = Language::defineLanguage(args[QString("config")].toString(),
            args[QString("options")].toString());
    QTranslator qtTranslator;
    qtTranslator.load(QString("qt_%1").arg(language), QLibraryInfo::location(QLibraryInfo::TranslationsPath));
    a.installTranslator(&qtTranslator);
    QTranslator translator;
    translator.load(QString(":/translations-helper/%1").arg(language));
    a.installTranslator(&translator);

    // running
    if (args[QString("error")].toBool()) {
        cout << errorMessage().toUtf8().data() << endl;
        cout << helpMessage().toUtf8().data();
        return 11;
    } else if (args[QString("help")].toBool()) {
        cout << helpMessage().toUtf8().data();
        return 0;
    } else if (args[QString("info")].toBool()) {
        cout << versionMessage().toUtf8().data() << endl;
        cout << infoMessage().toUtf8().data();
        return 0;
    } else if (args[QString("version")].toBool()) {
        cout << versionMessage().toUtf8().data();
        return 0;
    }

    // check if exists
    if (existingSessionOperation(QString("Active"))) {
        if (args[QString("state")].toInt() == 1) {
            // replace session
            cout << QCoreApplication::translate("NetctlHelper", "Replace existing session.")
                    .toUtf8().data() << endl;
            existingSessionOperation(QString("Close"));
        } else if (args[QString("state")].toInt() == 2) {
            // restore session
            cout << QCoreApplication::translate("NetctlHelper", "Restore existing session.")
                    .toUtf8().data() << endl;
            return 0;
        }
        else if (geteuid() == 0) {
            // replace if running as root
            cout << QCoreApplication::translate("NetctlHelper", "Replace existing session.")
                    .toUtf8().data() << endl;
            existingSessionOperation(QString("Close"));
        }
        else {
            // restore if running as non-root
            cout << QCoreApplication::translate("NetctlHelper", "Restore existing session.")
                    .toUtf8().data() << endl;
            return 0;
        }
    }
    NetctlHelper w(0, args);
    return a.exec();
}