ostinato/server/drone_main.cpp
2022-06-27 14:35:22 +05:30

144 lines
3.8 KiB
C++

/*
Copyright (C) 2010 Srivats P.
This file is part of "Ostinato"
This 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.
This program 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 this program. If not, see <http://www.gnu.org/licenses/>
*/
#include "drone.h"
#include "../common/protocolmanager.h"
#include "params.h"
#include "settings.h"
#include "turbo.h"
#include <google/protobuf/stubs/common.h>
#include <QCoreApplication>
#include <QFile>
#include <signal.h>
extern ProtocolManager *OstProtocolManager;
extern char *version;
extern char *revision;
Drone *drone;
QSettings *appSettings;
Params appParams;
void NoMsgHandler(QtMsgType type, const QMessageLogContext &context,
const QString &msg);
void cleanup(int /*signum*/)
{
fprintf(stderr, "\nCleaning up (may take a few seconds) ... ");
fflush(stderr);
QCoreApplication::instance()->exit(-1);
}
int main(int argc, char *argv[])
{
int exitCode = 0;
QCoreApplication app(argc, argv);
app.setApplicationName("Drone");
app.setOrganizationName("Ostinato");
appParams.parseCommandLine(argc, argv);
fprintf(stderr, "Starting (will take a few seconds) ...\n");
fflush(stderr);
#ifdef QT_NO_DEBUG
if (appParams.optLogsDisabled())
qInstallMessageHandler(NoMsgHandler);
#endif
qDebug("Version: %s", version);
qDebug("Revision: %s", revision);
/* (Portable Mode) If we have a .ini file in the same directory as the
executable, we use that instead of the platform specific location
and format for the settings */
QString portableIni = QCoreApplication::applicationDirPath()
+ "/drone.ini";
if (QFile::exists(portableIni))
appSettings = new QSettings(portableIni, QSettings::IniFormat);
else
appSettings = new QSettings(QSettings::IniFormat,
QSettings::UserScope,
app.organizationName(),
app.applicationName().toLower());
qDebug("Settings: %s", qPrintable(appSettings->fileName()));
if (!initTurbo())
{
exitCode = 1;
goto _exit2;
}
drone = new Drone();
OstProtocolManager = new ProtocolManager();
if (!drone->init())
{
exitCode = 1;
goto _exit;
}
qDebug("Version: %s", version);
qDebug("Revision: %s", revision);
#ifdef Q_OS_UNIX
struct sigaction sa;
memset(&sa, 0, sizeof(sa));
sa.sa_handler = cleanup;
if (sigaction(SIGTERM, &sa, NULL))
qDebug("Failed to install SIGTERM handler. Cleanup may not happen!!!");
if (sigaction(SIGINT, &sa, NULL))
qDebug("Failed to install SIGINT handler. Cleanup may not happen!!!");
#elif defined(Q_OS_WIN32)
if (signal(SIGTERM, cleanup) == SIG_ERR)
qDebug("Failed to install SIGTERM handler. Cleanup may not happen!!!");
if (signal(SIGINT, cleanup) == SIG_ERR)
qDebug("Failed to install SIGINT handler. Cleanup may not happen!!!");
#endif
exitCode = app.exec();
_exit:
delete drone;
delete OstProtocolManager;
_exit2:
google::protobuf::ShutdownProtobufLibrary();
fprintf(stderr, "done.\n");
fflush(stderr);
return exitCode;
}
void NoMsgHandler(QtMsgType type, const QMessageLogContext &/*context*/,
const QString &msg)
{
if (type == QtFatalMsg) {
fprintf(stderr, "%s\n", qPrintable(msg));
fflush(stderr);
abort();
}
}