mirror of
https://github.com/obsproject/obs-studio.git
synced 2026-03-29 11:51:59 -04:00
83 lines
2.1 KiB
C
83 lines
2.1 KiB
C
#include <windows.h>
|
|
#include <stdio.h>
|
|
#include "app-helpers.h"
|
|
#include "nt-stuff.h"
|
|
#include "util/base.h"
|
|
|
|
WINADVAPI BOOL WINAPI ConvertSidToStringSidW(PSID sid, LPWSTR *str);
|
|
|
|
bool is_app(HANDLE process)
|
|
{
|
|
DWORD size_ret;
|
|
DWORD ret = 0;
|
|
HANDLE token;
|
|
|
|
if (OpenProcessToken(process, TOKEN_QUERY, &token)) {
|
|
BOOL success = GetTokenInformation(token, TokenIsAppContainer, &ret, sizeof(ret), &size_ret);
|
|
if (!success) {
|
|
const DWORD error = GetLastError();
|
|
blog(LOG_ERROR, "is_app GetTokenInformation failed: 0x%08lX", error);
|
|
}
|
|
|
|
CloseHandle(token);
|
|
}
|
|
return !!ret;
|
|
}
|
|
|
|
wchar_t *get_app_sid(HANDLE process)
|
|
{
|
|
wchar_t *ret = NULL;
|
|
DWORD size_ret;
|
|
BOOL success;
|
|
HANDLE token;
|
|
|
|
if (OpenProcessToken(process, TOKEN_QUERY, &token)) {
|
|
DWORD info_len = GetSidLengthRequired(12) + sizeof(TOKEN_APPCONTAINER_INFORMATION);
|
|
|
|
PTOKEN_APPCONTAINER_INFORMATION info = malloc(info_len);
|
|
|
|
success = GetTokenInformation(token, TokenAppContainerSid, info, info_len, &size_ret);
|
|
if (success)
|
|
ConvertSidToStringSidW(info->TokenAppContainer, &ret);
|
|
|
|
free(info);
|
|
CloseHandle(token);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
static const wchar_t *path_format = L"\\Sessions\\%lu\\AppContainerNamedObjects\\%s\\%s";
|
|
|
|
HANDLE create_app_mutex(const wchar_t *sid, const wchar_t *name)
|
|
{
|
|
wchar_t path[MAX_PATH];
|
|
DWORD session_id = WTSGetActiveConsoleSessionId();
|
|
_snwprintf(path, MAX_PATH, path_format, session_id, sid, name);
|
|
return nt_create_mutex(path);
|
|
}
|
|
|
|
HANDLE open_app_mutex(const wchar_t *sid, const wchar_t *name)
|
|
{
|
|
wchar_t path[MAX_PATH];
|
|
DWORD session_id = WTSGetActiveConsoleSessionId();
|
|
_snwprintf(path, MAX_PATH, path_format, session_id, sid, name);
|
|
return nt_open_mutex(path);
|
|
}
|
|
|
|
HANDLE open_app_event(const wchar_t *sid, const wchar_t *name)
|
|
{
|
|
wchar_t path[MAX_PATH];
|
|
DWORD session_id = WTSGetActiveConsoleSessionId();
|
|
_snwprintf(path, MAX_PATH, path_format, session_id, sid, name);
|
|
return nt_open_event(path);
|
|
}
|
|
|
|
HANDLE open_app_map(const wchar_t *sid, const wchar_t *name)
|
|
{
|
|
wchar_t path[MAX_PATH];
|
|
DWORD session_id = WTSGetActiveConsoleSessionId();
|
|
_snwprintf(path, MAX_PATH, path_format, session_id, sid, name);
|
|
return nt_open_map(path);
|
|
}
|