Mercurial > libanimone

annotate src/fd/win32.cc @ 25:60ded877339b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
dep/animone: fix tons of issues for example, the window ID stuff was just... completely wrong. since we're supporting multiple different window systems, it *has* to be a union rather than just a single integer type. HWND is also not a DWORD, it's a pointer(!), so now it's stored as a std::uintptr_t. (this probably breaks things)
author Paper <paper@paper.us.eu.org>
date Thu, 20 Jun 2024 03:03:05 -0400
parents 1cfa7d57c022
children 77a5ea5e996c
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
19
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
1 /*
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
2 * fd/win32.cc: support for windows
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
3 *
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
4 * this file is noticably more complex than *nix, and that's because
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
5 * there is no "simple" way to get the paths of a file. In fact, this thing requires
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
6 * you to use *internal functions* that can't even be linked to, hence why we have to
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
7 * use GetProcAddress and such. what a mess.
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
8 *
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
9 * Speaking of which, because this file uses internal functions of the OS, it is not
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
10 * even guaranteed to work far into the future. however, just like with macOS, these
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
11 * things have stayed the same since Vista so if Microsoft *really* wants compatibility
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
12 * then they're pretty much forced to keeping this the same anyway.
6596be6917a1 dep/animone: prepare for v1.0 release; it should be ready by now
Paper <paper@paper.us.eu.org>
parents: 14
diff changeset
13 */
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
14 #include "animone/fd/win32.h"
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
15 #include "animone/util.h"
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
16 #include "animone.h"
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
17 #include "animone/util/win32.h"
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
18
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
19 #include <stdexcept>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
20 #include <string>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
21 #include <unordered_map>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
22 #include <vector>
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
23 #include <optional>
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
24
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
25 #include <fileapi.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
26 #include <handleapi.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
27 #include <libloaderapi.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
28 #include <ntdef.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
29 #include <psapi.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
30 #include <shlobj.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
31 #include <stringapiset.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
32 #include <tlhelp32.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
33 #include <windows.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
34 #include <winternl.h>
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
35
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
36 /* SystemExtendedHandleInformation is only available in NT 5.1+ (XP and higher) and provides information for
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
37 * 32-bit PIDs, unlike SystemHandleInformation */
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
38 static constexpr SYSTEM_INFORMATION_CLASS SystemExtendedHandleInformation = static_cast<SYSTEM_INFORMATION_CLASS>(0x40);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
39 static constexpr NTSTATUS STATUS_INFO_LENGTH_MISMATCH = 0xC0000004UL;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
40
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
41 struct SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
42 PVOID Object;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
43 ULONG_PTR UniqueProcessId;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
44 HANDLE HandleValue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
45 ACCESS_MASK GrantedAccess;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
46 USHORT CreatorBackTraceIndex;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
47 USHORT ObjectTypeIndex;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
48 ULONG HandleAttributes;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
49 ULONG Reserved;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
50 };
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
51
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
52 struct SYSTEM_HANDLE_INFORMATION_EX {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
53 ULONG_PTR NumberOfHandles;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
54 ULONG_PTR Reserved;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
55 SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX Handles[1];
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
56 };
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
57
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
58 namespace animone::internal::win32 {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
59
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
60 class Ntdll {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
61 public:
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
62 Ntdll() {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
63 ntdll = ::GetModuleHandleW(L"ntdll.dll");
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
64 nt_query_system_information = reinterpret_cast<decltype(::NtQuerySystemInformation)*>(
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
65 ::GetProcAddress(ntdll, "NtQuerySystemInformation"));
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
66 nt_query_object = reinterpret_cast<decltype(::NtQueryObject)*>(::GetProcAddress(ntdll, "NtQueryObject"));
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
67 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
68
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
69 NTSTATUS QuerySystemInformation(SYSTEM_INFORMATION_CLASS cls, PVOID sysinfo, ULONG len,
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
70 PULONG retlen){
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
71 return nt_query_system_information(cls, sysinfo, len, retlen);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
72 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
73
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
74 NTSTATUS QueryObject(HANDLE handle, OBJECT_INFORMATION_CLASS cls, PVOID objinf, ULONG objinflen, PULONG retlen) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
75 return nt_query_object(handle, cls, objinf, objinflen, retlen);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
76 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
77
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
78 private:
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
79 HMODULE ntdll;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
80 decltype(::NtQuerySystemInformation)* nt_query_system_information;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
81 decltype(::NtQueryObject)* nt_query_object;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
82 };
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
83
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
84 Ntdll ntdll;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
85
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
86 static HANDLE DuplicateHandle(HANDLE process_handle, HANDLE handle) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
87 HANDLE dup_handle = nullptr;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
88 const bool result =
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
89 ::DuplicateHandle(process_handle, handle, ::GetCurrentProcess(), &dup_handle, 0, false, DUPLICATE_SAME_ACCESS);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
90 return result ? dup_handle : nullptr;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
91 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
92
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
93 static std::vector<SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX> GetSystemHandleInformation() {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
94 /* we should really put a cap on this */
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
95 ULONG cb = 1 << 19;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
96 NTSTATUS status = STATUS_NO_MEMORY;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
97 std::unique_ptr<SYSTEM_HANDLE_INFORMATION_EX> info;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
98
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
99 do {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
100 info.reset(reinterpret_cast<SYSTEM_HANDLE_INFORMATION_EX*>(malloc(cb *= 2)));
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
101 if (!info)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
102 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
103
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
104 status = ntdll.QuerySystemInformation(SystemExtendedHandleInformation, info.get(), cb, &cb);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
105 } while (status == STATUS_INFO_LENGTH_MISMATCH);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
106
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
107 if (!NT_SUCCESS(status))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
108 return {};
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
109
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
110 std::vector<SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX> res;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
111
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
112 ULONG_PTR handles = info->NumberOfHandles;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
113 if (!handles)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
114 return {};
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
115
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
116 res.reserve(handles);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
117
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
118 SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX* entry = info->Handles;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
119 do {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
120 if (entry)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
121 res.push_back(*(entry++));
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
122 } while (--handles);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
123
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
124 return res;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
125 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
126
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
127 static std::wstring GetHandleType(HANDLE handle) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
128 OBJECT_TYPE_INFORMATION info = {0};
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
129 ntdll.QueryObject(handle, ObjectTypeInformation, &info, sizeof(info), NULL);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
130 return std::wstring(info.TypeName.Buffer, info.TypeName.Length);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
131 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
132
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
133 static std::wstring GetFinalPathNameByHandle(HANDLE handle) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
134 std::wstring buffer;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
135
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
136 DWORD size = ::GetFinalPathNameByHandleW(handle, nullptr, 0, FILE_NAME_NORMALIZED | VOLUME_NAME_DOS);
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
137 buffer.resize(size);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
138 ::GetFinalPathNameByHandleW(handle, &buffer.front(), buffer.size(), FILE_NAME_NORMALIZED | VOLUME_NAME_DOS);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
139
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
140 return buffer;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
141 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
142
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
143 /* ------------------------------------------------------------------- */
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
144
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
145 static bool GetSystemDirectory(std::wstring& str) {
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
146 str.assign(MAX_PATH, '\0');
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
147
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
148 return SUCCEEDED(::SHGetFolderPathW(NULL, CSIDL_WINDOWS, NULL, SHGFP_TYPE_CURRENT, &str.front()));
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
149 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
150
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
151 static bool IsSystemDirectory(std::wstring path) {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
152 std::wstring windir;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
153 if (!GetSystemDirectory(windir))
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
154 return false;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
155
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
156 ::CharUpperBuffW(&path.front(), path.length());
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
157 ::CharUpperBuffW(&windir.front(), windir.length());
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
158
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
159 // XXX wtf is 4?
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
160 return path.find(windir) == 4;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
161 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
162
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
163 static bool IsSystemDirectory(const std::string& path) {
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
164 return IsSystemDirectory(ToWstring(path));
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
165 }
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
166
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
167 static bool IsFileHandle(HANDLE handle, unsigned short object_type_index) {
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
168 /* this is filled in at runtime because it's not guaranteed to be (and isn't)
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
169 * constant between different versions of Windows */
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
170 static std::optional<unsigned short> file_type_index;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
171
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
172 if (file_type_index.has_value()) {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
173 return object_type_index == file_type_index.value();
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
174 } else if (!handle) {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
175 /* XXX what? */
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
176 return true;
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
177 } else if (GetHandleType(handle) == L"File") {
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
178 file_type_index = object_type_index;
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
179 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
180 }
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
181
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
182 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
183 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
184
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
185 static bool IsFileMaskOk(ACCESS_MASK access_mask) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
186 if (!(access_mask & FILE_READ_DATA))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
187 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
188
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
189 if ((access_mask & FILE_APPEND_DATA) || (access_mask & FILE_WRITE_EA) || (access_mask & FILE_WRITE_ATTRIBUTES))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
190 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
191
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
192 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
193 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
194
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
195 static bool IsFilePathOk(const std::wstring& path) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
196 if (path.empty())
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
197 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
198
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
199 if (IsSystemDirectory(path))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
200 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
201
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
202 const auto file_attributes = GetFileAttributesW(path.c_str());
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
203 if ((file_attributes == INVALID_FILE_ATTRIBUTES) || (file_attributes & FILE_ATTRIBUTE_DIRECTORY))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
204 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
205
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
206 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
207 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
208
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
209 /* ------------------------------------------------------------------- */
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
210
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
211 static bool GetProcessPath(DWORD process_id, std::string& path) {
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
212 // If we try to open a SYSTEM process, this function fails and the last error
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
213 // code is ERROR_ACCESS_DENIED.
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
214 //
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
215 // Note that if we requested PROCESS_QUERY_INFORMATION access right instead
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
216 // of PROCESS_QUERY_LIMITED_INFORMATION, this function would fail when used
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
217 // to open an elevated process.
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
218 Handle process_handle(::OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, process_id));
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
219
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
220 if (!process_handle)
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
221 return false;
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
222
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
223 std::wstring buffer(MAX_PATH, L'\0');
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
224 DWORD buf_size = buffer.length();
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
225
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
226 // Note that this function requires Windows Vista or above. You may use
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
227 // GetProcessImageFileName or GetModuleFileNameEx on earlier versions.
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
228 if (!::QueryFullProcessImageNameW(process_handle.get(), 0, &buffer.front(), &buf_size))
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
229 return false;
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
230
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
231 buffer.resize(buf_size);
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
232 path = ToUtf8String(buffer);
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
233 return true;
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
234 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
235
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
236 static std::string GetFilenameFromPath(const std::string& path) {
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
237 const auto pos = path.find_last_of("/\\");
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
238 return pos != std::wstring::npos ? path.substr(pos + 1) : path;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
239 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
240
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
241 static bool VerifyProcessPath(const std::string& path) {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
242 return !path.empty() && !IsSystemDirectory(path);
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
243 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
244
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
245 static bool VerifyProcessFilename(const std::string& name) {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
246 static const std::set<std::string> invalid_names = {
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
247 // System files
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
248 "explorer.exe", // Windows Explorer
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
249 "taskeng.exe", // Task Scheduler Engine
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
250 "taskhost.exe", // Host Process for Windows Tasks
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
251 "taskhostex.exe", // Host Process for Windows Tasks
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
252 "taskmgr.exe", // Task Manager
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
253 "services.exe", // Service Control Manager
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
254 };
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
255
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
256 if (name.empty())
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
257 return false;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
258
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
259 for (const auto& invalid_name : invalid_names)
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
260 if (util::EqualStrings(name, invalid_name))
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
261 return false;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
262
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
263 return true;
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
264 }
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
265
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
266 /* ------------------------------------------------------------------- */
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
267 /* extern functions */
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
268
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
269 bool GetProcessName(pid_t pid, std::string& name) {
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
270 if (!GetProcessPath(pid, name) || !VerifyProcessPath(name))
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
271 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
272
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
273 name = GetFilenameFromPath(name);
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
274 if (!VerifyProcessFilename(name))
21
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
275 return false;
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
276
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
277 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
278 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
279
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
280 bool EnumerateOpenProcesses(process_proc_t process_proc) {
21
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
281 Handle process_snap(::CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0));
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
282 if (process_snap.get() == INVALID_HANDLE_VALUE)
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
283 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
284
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
285 PROCESSENTRY32 pe32;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
286 pe32.dwSize = sizeof(PROCESSENTRY32);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
287
21
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
288 if (!::Process32First(process_snap.get(), &pe32))
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
289 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
290
21
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
291 do {
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
292 std::string name;
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
293 if (!GetProcessName(pe32.th32ProcessID, name))
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
294 continue;
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
295
25
60ded877339b dep/animone: fix tons of issues
Paper <paper@paper.us.eu.org>
parents: 23
diff changeset
296 if (!process_proc({.pid = pe32.th32ProcessID, .platform = ExecutablePlatform::Win32, .comm = name}))
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
297 return false;
21
973734ebd2be dep/animone: REFACTOR ALL THE THINGS
Paper <paper@paper.us.eu.org>
parents: 19
diff changeset
298 } while (::Process32Next(process_snap.get(), &pe32));
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
299
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
300 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
301 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
302
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
303 bool EnumerateOpenFiles(const std::set<pid_t>& pids, open_file_proc_t open_file_proc) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
304 if (!open_file_proc)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
305 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
306
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
307 std::unordered_map<pid_t, Handle> proc_handles;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
308
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
309 for (const pid_t& pid : pids) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
310 const HANDLE handle = ::OpenProcess(PROCESS_DUP_HANDLE, false, pid);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
311 if (handle != INVALID_HANDLE_VALUE)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
312 proc_handles[pid] = Handle(handle);
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
313 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
314
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
315 if (proc_handles.empty())
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
316 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
317
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
318 std::vector<SYSTEM_HANDLE_TABLE_ENTRY_INFO_EX> info = GetSystemHandleInformation();
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
319
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
320 for (const auto& h : info) {
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
321 const pid_t pid = h.UniqueProcessId;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
322 if (!pids.count(pid))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
323 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
324
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
325 if (!IsFileHandle(nullptr, h.ObjectTypeIndex))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
326 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
327
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
328 if (!IsFileMaskOk(h.GrantedAccess))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
329 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
330
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
331 Handle handle(DuplicateHandle(proc_handles[pid].get(), h.HandleValue));
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
332 if (handle.get() == INVALID_HANDLE_VALUE)
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
333 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
334
23
1cfa7d57c022 dep/animone: add experimental accessibility strategy
Paper <paper@paper.us.eu.org>
parents: 21
diff changeset
335 if (::GetFileType(handle.get()) != FILE_TYPE_DISK)
14
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
336 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
337
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
338 const std::wstring path = GetFinalPathNameByHandle(handle.get());
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
339 if (!IsFilePathOk(path))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
340 continue;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
341
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
342 if (!open_file_proc({pid, ToUtf8String(path)}))
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
343 return false;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
344 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
345
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
346 return true;
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
347 }
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
348
27b988a1048c *: convert all files CRLF -> LF
Paper <paper@paper.us.eu.org>
parents: 12
diff changeset
349 } // namespace animone::internal::win32