mirror of
https://github.com/MrUnknownDE/VRCX.git
synced 2026-04-12 03:13:50 +02:00
70249ea790
* feat: Add option for worlds to get data from other worlds * docs: Add docs to new methods * refactor: Add more error handling for potential edge cases * fix: Catch exception for http listener start * fix: Maybe stop throwing js exceptions at start from process check * fix: Stop VRCX from dying if monitored processes are elevated * fix: If auto close is turned off, update process states properly * refactor: Limit db key length to 255, limit /getall to 10000 entries
68 lines
3.4 KiB
C#
68 lines
3.4 KiB
C#
// Copyright(c) 2019-2022 pypy, Natsumi and individual contributors.
|
|
// All rights reserved.
|
|
//
|
|
// This work is licensed under the terms of the MIT license.
|
|
// For a copy, see <https://opensource.org/licenses/MIT>.
|
|
|
|
using System;
|
|
using System.Runtime.InteropServices;
|
|
|
|
namespace VRCX
|
|
{
|
|
public static class WinApi
|
|
{
|
|
[DllImport("kernel32.dll", SetLastError = false)]
|
|
public static extern void CopyMemory(IntPtr destination, IntPtr source, uint length);
|
|
|
|
[DllImport("user32.dll", SetLastError = true)]
|
|
public static extern IntPtr FindWindow(string lpClassName, string lpWindowName);
|
|
|
|
[DllImport("user32.dll", SetLastError = true)]
|
|
public static extern uint GetWindowThreadProcessId(IntPtr hWnd, out uint lpdwProcessId);
|
|
[DllImport("shell32.dll", CharSet = CharSet.Unicode)]
|
|
public static extern int SHParseDisplayName([MarshalAs(UnmanagedType.LPWStr)] string pszName, IntPtr pbc, out IntPtr ppidl, uint sfgaoIn, out uint psfgaoOut);
|
|
[DllImport("shell32.dll", CharSet = CharSet.Auto)]
|
|
public static extern IntPtr SHOpenFolderAndSelectItems(IntPtr pidlFolder, uint cidl, IntPtr[] apidl, uint dwFlags);
|
|
[DllImport("kernel32.dll", SetLastError = true)]
|
|
public static extern IntPtr OpenProcess(int dwDesiredAccess, bool bInheritHandle, int dwProcessId);
|
|
|
|
[DllImport("kernel32.dll", SetLastError = true)]
|
|
public static extern bool GetExitCodeProcess(IntPtr hProcess, out uint lpExitCode);
|
|
|
|
/// <summary>
|
|
/// Flag that specifies the access rights to query limited information about a process.
|
|
/// This won't throw an exception when we try to access info about an elevated process
|
|
/// </summary>
|
|
private const int PROCESS_QUERY_LIMITED_INFORMATION = 0x1000;
|
|
|
|
/// <summary>
|
|
/// Determines whether the specified process has exited using WinAPI's GetExitCodeProcess running with PROCESS_QUERY_LIMITED_INFORMATION.
|
|
/// We do this because Process.HasExited in .net framework opens a handle with PROCESS_QUERY_INFORMATION, which will throw an exception if the process is elevated.
|
|
/// GetExitCodeProcess works with PROCESS_QUERY_LIMITED_INFORMATION, which will not throw an exception if the process is elevated.
|
|
/// https://learn.microsoft.com/en-us/windows/win32/api/processthreadsapi/nf-processthreadsapi-getexitcodeprocess
|
|
/// </summary>
|
|
/// <param name="process">The process to check.</param>
|
|
/// <returns>true if the process has exited; otherwise, false.</returns>
|
|
internal static bool HasProcessExited(int processId)
|
|
{
|
|
IntPtr hProcess = WinApi.OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, false, processId);
|
|
if (hProcess == IntPtr.Zero)
|
|
{
|
|
// this is probably fine
|
|
return true;
|
|
//throw new System.ComponentModel.Win32Exception(Marshal.GetLastWin32Error());
|
|
}
|
|
|
|
bool exited;
|
|
if (!WinApi.GetExitCodeProcess(hProcess, out uint exitCode))
|
|
{
|
|
throw new System.ComponentModel.Win32Exception(Marshal.GetLastWin32Error());
|
|
}
|
|
|
|
// Fun fact, If a program uses STILL_ACTIVE (259) as an exit code, GetExitCodeProcess will return 259, since it returns... the exit code. This would break this function.
|
|
exited = exitCode != 259;
|
|
return exited;
|
|
}
|
|
}
|
|
}
|