110 lines
3.2 KiB
C++
110 lines
3.2 KiB
C++
|
#include <Windows.h>
|
||
|
|
#include <QApplication>
|
||
|
|
#include <QDebug>
|
||
|
|
#include <QTextCodec>
|
||
|
|
#include <QMessageBox>
|
||
|
|
#include <QDir>
|
||
|
|
#include <QFileInfo>
|
||
|
|
#include <QSaveFile>
|
||
|
|
#include "MainWindow.h"
|
||
|
|
#include "../Common/ConfigHelper.h"
|
||
|
|
#include "../Common/PolicyHelper.h"
|
||
|
|
#include "../Common/LocalStateHelper.h"
|
||
|
|
|
||
|
|
int main(int argc, char* argv[])
|
||
|
|
{
|
||
|
|
SetConsoleOutputCP(936);
|
||
|
|
QTextCodec::setCodecForLocale(QTextCodec::codecForName("GBK"));
|
||
|
|
|
||
|
|
QApplication a(argc, argv);
|
||
|
|
|
||
|
|
qDebug() << Qt::endl << "entered main app" << Qt::endl;
|
||
|
|
|
||
|
|
const QString validToken = ConfigHelper::instance().getValue("App", "launch_token");
|
||
|
|
bool pass = false;
|
||
|
|
QString healthFilePath;
|
||
|
|
|
||
|
|
for (int i = 0; i < argc; ++i)
|
||
|
|
{
|
||
|
|
QString arg(argv[i]);
|
||
|
|
if (arg.startsWith("--launcher-token="))
|
||
|
|
{
|
||
|
|
QString tk = arg.split("=").last();
|
||
|
|
if (tk == validToken)
|
||
|
|
{
|
||
|
|
pass = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (arg.startsWith("--health-file="))
|
||
|
|
{
|
||
|
|
healthFilePath = arg.mid(QString("--health-file=").size());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
if (!pass)
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Startup Restriction", "Direct double-clicking MainApp.exe is prohibited. Please use Launcher.exe to open the software!");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
QString appDir = QApplication::applicationDirPath();
|
||
|
|
PolicyHelper policy(appDir);
|
||
|
|
if (!policy.loadPolicy("config/version_policy.dat") || !policy.isValid())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Policy Error", QString("Local policy invalid: %1").arg(policy.errorString()));
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (policy.isExpired())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Policy Error", "Policy expired, cannot start the application.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
LocalStateHelper state(appDir);
|
||
|
|
if (!state.loadState())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "State Error", QString("Cannot load local state: %1").arg(state.errorString()));
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state.isPolicySeqRolledBack(policy.policySeq()))
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Policy Error", "Detected policy sequence rollback, startup blocked.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (state.isSystemTimeRewound())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Policy Error", "System time appears to be rewound, startup blocked.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
MainWindow w;
|
||
|
|
w.show();
|
||
|
|
|
||
|
|
state.updateAfterSuccessfulRun(ConfigHelper::instance().getValue("App", "current_version"), policy.policySeq());
|
||
|
|
if (!state.saveState())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "State Error", QString("Cannot save local state: %1").arg(state.errorString()));
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!healthFilePath.isEmpty())
|
||
|
|
{
|
||
|
|
QDir().mkpath(QFileInfo(healthFilePath).path());
|
||
|
|
QSaveFile healthFile(healthFilePath);
|
||
|
|
const QByteArray healthy("ok\n");
|
||
|
|
if (!healthFile.open(QIODevice::WriteOnly)
|
||
|
|
|| healthFile.write(healthy) != healthy.size()
|
||
|
|
|| !healthFile.commit())
|
||
|
|
{
|
||
|
|
QMessageBox::critical(nullptr, "Startup Error", "Cannot write update health confirmation file.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
qDebug() << "Main program MainApp is running normally";
|
||
|
|
return a.exec();
|
||
|
|
}
|