62 lines
2.0 KiB
C++
62 lines
2.0 KiB
C++
|
// SPDX-FileCopyrightText: 2023 UnionTech Software Technology Co., Ltd.
|
||
|
|
//
|
||
|
|
// SPDX-License-Identifier: LGPL-3.0-or-later
|
||
|
|
|
||
|
|
#include "cgroupsidentifier.h"
|
||
|
|
#include "global.h"
|
||
|
|
#include <QFile>
|
||
|
|
#include <QString>
|
||
|
|
#include <QDebug>
|
||
|
|
|
||
|
|
IdentifyRet CGroupsIdentifier::Identify(pid_t pid)
|
||
|
|
{
|
||
|
|
auto AppCgroupPath = QString("/proc/%1/cgroup").arg(pid);
|
||
|
|
QFile AppCgroupFile{AppCgroupPath};
|
||
|
|
if (!AppCgroupFile.open(QFile::ExistingOnly | QFile::ReadOnly | QFile::Text)) {
|
||
|
|
qWarning() << "open " << AppCgroupPath << "failed: " << AppCgroupFile.errorString();
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
auto appInstance = parseCGroupsPath(
|
||
|
|
AppCgroupFile.readAll().split(':').last().trimmed()); // FIXME: support CGroup version detection and multi-line parsing
|
||
|
|
auto appInstanceComponent = appInstance.split('@');
|
||
|
|
if (appInstanceComponent.count() != 2) {
|
||
|
|
qWarning() << "Application Instance format is invalid." << appInstanceComponent;
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
return {appInstanceComponent.first(), appInstanceComponent.last()};
|
||
|
|
}
|
||
|
|
|
||
|
|
QString CGroupsIdentifier::parseCGroupsPath(QString CGP) const noexcept
|
||
|
|
{
|
||
|
|
if (CGP.isEmpty()) {
|
||
|
|
qWarning() << "CGroupPath is empty.";
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
auto unescapedCGP = unescapeString(CGP);
|
||
|
|
auto CGPSlices = unescapedCGP.split('/');
|
||
|
|
auto curUID = getCurrentUID();
|
||
|
|
|
||
|
|
if (CGPSlices.first() != "user.slice") {
|
||
|
|
qWarning() << "unrecognized process.";
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
auto processUIDStr = CGPSlices[1].split('.').first().split('-').last();
|
||
|
|
|
||
|
|
if (processUIDStr.isEmpty()) {
|
||
|
|
qWarning() << "no uid found.";
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
|
||
|
|
if (auto processUID = processUIDStr.toInt(); processUID != getCurrentUID()) {
|
||
|
|
qWarning() << "process is not in CGroups of current user, ignore....";
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
|
||
|
|
auto appInstance = CGPSlices.last().split('.').first();
|
||
|
|
if (appInstance.isEmpty()) {
|
||
|
|
qWarning() << "get AppId failed.";
|
||
|
|
return {};
|
||
|
|
}
|
||
|
|
return appInstance;
|
||
|
|
}
|