mirrors/akvirtualcamera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Removed dependency with Restart Manager and using Process Status API instead (Windows).

Browse source
This commit is contained in:
Gonzalo Exequiel Pedone 2020-06-06 19:02:10 -03:00
parent 7e35bf47f5
commit 602829f351
No known key found for this signature in database
GPG key ID: B8B09E63E9B85BAF
2 changed files with 46 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
dshow/VCamIPC/VCamIPC.pro
View file

@ -45,8 +45,7 @@ LIBS = \
-L$${OUT_PWD}/../../VCamUtils/$${BIN_DIR} -lVCamUtils \
-ladvapi32 \
-lkernel32 \
-lpsapi \
-lrstrmgr
-lpsapi
win32-g++: LIBS += -lssp

83
dshow/VCamIPC/src/ipcbridge.cpp
View file

@ -26,7 +26,6 @@
#include <thread>
#include <dshow.h>
#include <psapi.h>
#include <restartmanager.h>
#include "PlatformUtils/src/messageserver.h"
#include "PlatformUtils/src/mutex.h"
@ -571,55 +570,63 @@ std::vector<uint64_t> AkVCam::IpcBridge::clientsPids() const
}
std::vector<uint64_t> pids;
DWORD sessionHnd = 0;
WCHAR sessionKey[CCH_RM_SESSION_KEY + 1];
memset(sessionKey, 0, (CCH_RM_SESSION_KEY + 1) * sizeof(WCHAR));
const DWORD nElements = 4096;
DWORD process[nElements];
memset(process, 0, nElements * sizeof(DWORD));
DWORD needed = 0;
if (!EnumProcesses(process, nElements * sizeof(DWORD), &needed))
return {};
size_t nProcess = needed / sizeof(DWORD);
auto currentPid = GetCurrentProcessId();
if (SUCCEEDED(RmStartSession(&sessionHnd, 0, sessionKey))) {
std::vector<LPCWSTR> resources;
for (size_t i = 0; i < nProcess; i++) {
auto processHnd = OpenProcess(PROCESS_QUERY_INFORMATION |
PROCESS_VM_READ,
FALSE,
process[i]);
if (!processHnd)
continue;
for (auto &plugin: pluginsPaths)
resources.push_back(plugin.c_str());
HMODULE modules[nElements];
memset(modules, 0, nElements * sizeof(HMODULE));
if (SUCCEEDED(RmRegisterResources(sessionHnd,
UINT(resources.size()),
resources.data(),
0,
nullptr,
0,
nullptr))) {
UINT nProcInfoNeeded = 0;
UINT nProcInfo = 0;
DWORD rebootReasons = 0;
if (EnumProcessModules(processHnd,
modules,
nElements * sizeof(HMODULE),
&needed)) {
size_t nModules =
std::min<DWORD>(needed / sizeof(HMODULE), nElements);
if (SUCCEEDED(RmGetList(sessionHnd,
&nProcInfoNeeded,
&nProcInfo,
nullptr,
&rebootReasons))) {
nProcInfo = nProcInfoNeeded;
nProcInfoNeeded = 0;
rebootReasons = 0;
std::vector<RM_PROCESS_INFO> affectedApps(nProcInfo);
for (size_t j = 0; j < nModules; j++) {
WCHAR moduleName[MAX_PATH];
memset(moduleName, 0, MAX_PATH * sizeof(WCHAR));
if (SUCCEEDED(RmGetList(sessionHnd,
&nProcInfoNeeded,
&nProcInfo,
affectedApps.data(),
&rebootReasons))) {
for (UINT i = 0; i < nProcInfo; i++) {
auto pid = affectedApps[i].Process.dwProcessId;
auto it = std::find(pids.begin(), pids.end(), pid);
if (GetModuleFileNameExW(processHnd,
modules[j],
moduleName,
MAX_PATH)) {
auto pluginsIt = std::find(pluginsPaths.begin(),
pluginsPaths.end(),
std::wstring(moduleName));
if (pid > 0 && it == pids.end() && pid != currentPid)
pids.push_back(pid);
if (pluginsIt != pluginsPaths.end()) {
auto pidsIt = std::find(pids.begin(),
pids.end(),
process[i]);
if (process[i] > 0
&& pidsIt == pids.end()
&& process[i] != currentPid)
pids.push_back(process[i]);
}
}
}
}
RmEndSession(sessionHnd);
CloseHandle(processHnd);
}
return pids;