mirrors/fort
1
0
Fork 0
mirror of https://github.com/tnodir/fort synced 2024-11-15 04:15:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

UI: ServiceInfoManager: Check a per-user service's name

Browse Source
This commit is contained in:
Nodir Temirkhodjaev 2023-04-29 12:13:39 +03:00
parent 06a7f9700a
commit 29ff36f7cc
1 changed files with 29 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/ui/manager/serviceinfomanager.cpp
View File

@ -30,20 +30,35 @@ QString getServiceDll(const RegKey &svcReg, bool *expand = nullptr)
return dllPathVar.toString(); return dllPathVar.toString();
} }
bool checkIsSvcHostService(const RegKey &svcReg, bool extendedCheck) QString resolveSvcHostServiceName(const RegKey &servicesReg, const QString &serviceName)
{
const RegKey svcReg(servicesReg, serviceName);
const quint32 serviceType = svcReg.value(serviceTypeKey).toUInt();
// Check a per-user service
if (serviceType == 224) {
const int pos = serviceName.lastIndexOf('_');
if (pos > 0) {
return serviceName.left(pos);
}
}
return serviceName;
}
bool checkIsSvcHostService(const RegKey &svcReg)
{ {
const auto imagePath = svcReg.value(serviceImagePathKey).toString(); const auto imagePath = svcReg.value(serviceImagePathKey).toString();
if (!imagePath.contains(R"(\system32\svchost.exe)", Qt::CaseInsensitive)) if (!imagePath.contains(R"(\system32\svchost.exe)", Qt::CaseInsensitive))
return false; return false;
if (extendedCheck) { if (!svcReg.contains("ServiceSidType") || svcReg.value("SvcHostSplitDisable").toInt() != 0)
if (!svcReg.contains("ServiceSidType") || svcReg.value("SvcHostSplitDisable").toInt() != 0) return false;
return false;
const QString dllPath = getServiceDll(svcReg); const QString dllPath = getServiceDll(svcReg);
if (dllPath.isEmpty()) if (dllPath.isEmpty())
return false; return false;
}
return true; return true;
} }
@ -69,10 +84,14 @@ QVector<ServiceInfo> getServiceInfoList(SC_HANDLE mngr, DWORD serviceType = SERV
for (int infoIndex = infoList.size(); serviceCount > 0; for (int infoIndex = infoList.size(); serviceCount > 0;
--serviceCount, ++service, ++infoIndex) { --serviceCount, ++service, ++infoIndex) {
const auto serviceName = QString::fromUtf16((const char16_t *) service->lpServiceName);
const auto originalServiceName =
QString::fromUtf16((const char16_t *) service->lpServiceName);
const auto serviceName = resolveSvcHostServiceName(servicesReg, originalServiceName);
const RegKey svcReg(servicesReg, serviceName); const RegKey svcReg(servicesReg, serviceName);
if (!checkIsSvcHostService(svcReg, /*extendedCheck=*/displayName)) if (!checkIsSvcHostService(svcReg))
continue; continue;
const quint32 trackFlags = svcReg.value(serviceTrackFlagsKey).toUInt(); const quint32 trackFlags = svcReg.value(serviceTrackFlagsKey).toUInt();