103 lines
2.2 KiB
C++
103 lines
2.2 KiB
C++
#include "helpers.h"
|
|
|
|
#include "util/externalcommand.h"
|
|
#include "backend/corebackend.h"
|
|
#include "backend/corebackendmanager.h"
|
|
#include "core/smartstatus.h"
|
|
#include "core/smartparser.h"
|
|
|
|
#include <QCoreApplication>
|
|
#include <QDebug>
|
|
|
|
static QString getDefaultDevicePath();
|
|
static bool testSmartStatus();
|
|
static bool testSmartParser();
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
QCoreApplication app(argc, argv);
|
|
|
|
KPMCoreInitializer i;
|
|
|
|
if (argc == 2)
|
|
i = KPMCoreInitializer(argv[1]);
|
|
|
|
if (!i.isValid())
|
|
return 1;
|
|
|
|
CoreBackend *backend = CoreBackendManager::self()->backend();
|
|
|
|
if (!backend)
|
|
{
|
|
qWarning() << "Couldn't get backend.";
|
|
return 1;
|
|
}
|
|
|
|
if (!testSmartStatus() || !testSmartParser())
|
|
return 1;
|
|
|
|
return app.exec();
|
|
}
|
|
|
|
static QString getDefaultDevicePath()
|
|
{
|
|
// Getting default home partition using 'df -P /home | awk 'END{print $1}'' command
|
|
ExternalCommand command(QStringLiteral("df"), { QStringLiteral("-P"), QStringLiteral("/home"), QStringLiteral("|"),
|
|
QStringLiteral("awk"), QStringLiteral("\'END{print $1}\'") });
|
|
|
|
if (command.run() && command.exitCode() == 0) {
|
|
QString output = command.output();
|
|
return output;
|
|
}
|
|
|
|
return QString();
|
|
}
|
|
|
|
static bool testSmartStatus()
|
|
{
|
|
const QString devicePath = getDefaultDevicePath();
|
|
|
|
SmartStatus smart(devicePath);
|
|
|
|
if (smart.devicePath() != devicePath)
|
|
return false;
|
|
|
|
if (!smart.status())
|
|
return false;
|
|
|
|
if (smart.modelName() == QString())
|
|
return false;
|
|
|
|
if (smart.firmware() == QString())
|
|
return false;
|
|
|
|
if (smart.serial() == QString())
|
|
return false;
|
|
|
|
if (smart.selfTestStatus() != SmartStatus::SelfTestStatus::Success)
|
|
return false;
|
|
|
|
if (!smart.isValid())
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool testSmartParser()
|
|
{
|
|
const QString devicePath = getDefaultDevicePath();
|
|
|
|
SmartParser parser(devicePath);
|
|
|
|
if (!parser.init())
|
|
return false;
|
|
|
|
if (parser.devicePath() != devicePath)
|
|
return false;
|
|
|
|
if (!parser.diskInformation())
|
|
return false;
|
|
|
|
return true;
|
|
}
|