Differential D20007 Diff 56237 src/lib/util/kprocesslist_win.cpp

Changeset View

Standalone View

View Options

src/lib/util/kprocesslist_win.cpp

  • This file was added.
1​/**************************************************************************
2​**
hallasAuthorUnsubmitted
Not Done

I do not have access to a Windows PC so I haven't built the Windows version

hallas: I do not have access to a Windows PC so I haven't built the Windows version
3​** This file is part of the KDE Frameworks
4​**
5​** Copyright (c) 2011 Nokia Corporation and/or its subsidiary(-ies).
6​** Copyright (c) 2019 David Hallas <david@davidhallas.dk>
7​**
8​** GNU Lesser General Public License Usage
9​**
10​** This file may be used under the terms of the GNU Lesser General Public
11​** License version 2.1 as published by the Free Software Foundation and
12​** appearing in the file LICENSE.LGPL included in the packaging of this file.
13​** Please review the following information to ensure the GNU Lesser General
14​** Public License version 2.1 requirements will be met:
15​** http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
16​**
17​** In addition, as a special exception, Nokia gives you certain additional
18​** rights. These rights are described in the Nokia Qt LGPL Exception
19​** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
20​**
21​** Other Usage
22​**
23​** Alternatively, this file may be used in accordance with the terms and
24​** conditions contained in a signed written agreement between you and Nokia.
25​**
26​** If you have questions regarding the use of this file, please contact
27​** Nokia at info@qt.nokia.com.
28​**
29​**************************************************************************/
30​
31​#include "kprocesslist.h"
32​
33​#include <QLibrary>
34​
35​// Enable Win API of XP SP1 and later
36​#ifdef Q_OS_WIN
37​# if !defined(_WIN32_WINNT)
38​# define _WIN32_WINNT 0x0502
39​# endif
40​# include <qt_windows.h>
41​# if !defined(PROCESS_SUSPEND_RESUME) // Check flag for MinGW
42​# define PROCESS_SUSPEND_RESUME (0x0800)
43​# endif // PROCESS_SUSPEND_RESUME
44​#endif // Q_OS_WIN
45​
46​#include <tlhelp32.h>
47​#include <psapi.h>
48​
49​// Resolve QueryFullProcessImageNameW out of kernel32.dll due
50​// to incomplete MinGW import libs and it not being present
51​// on Windows XP.
52​static inline BOOL queryFullProcessImageName(HANDLE h, DWORD flags, LPWSTR buffer, DWORD *size)
53​{
54​ // Resolve required symbols from the kernel32.dll
55​ typedef BOOL (WINAPI *QueryFullProcessImageNameWProtoType)(HANDLE, DWORD, LPWSTR, PDWORD);
56​ static QueryFullProcessImageNameWProtoType queryFullProcessImageNameW = 0;
57​ if (!queryFullProcessImageNameW) {
58​ QLibrary kernel32Lib(QLatin1String("kernel32.dll"), 0);
59​ if (kernel32Lib.isLoaded() || kernel32Lib.load()) {
60​ queryFullProcessImageNameW
61​ = (QueryFullProcessImageNameWProtoType)kernel32Lib.resolve(
62​ "QueryFullProcessImageNameW");
63​ }
64​ }
65​ if (!queryFullProcessImageNameW)
66​ return FALSE;
67​ // Read out process
68​ return (*queryFullProcessImageNameW)(h, flags, buffer, size);
69​}
70​
71​struct ProcessInfo {
72​ QString processOwner;
73​};
74​
75​static inline ProcessInfo processInfo(DWORD processId)
76​{
77​ ProcessInfo pi;
78​ HANDLE handle = OpenProcess(PROCESS_QUERY_INFORMATION, TOKEN_READ, processId);
79​ if (handle == INVALID_HANDLE_VALUE)
80​ return pi;
81​ HANDLE processTokenHandle = NULL;
82​ if (!OpenProcessToken(handle, TOKEN_READ, &processTokenHandle) || !processTokenHandle)
83​ return pi;
84​
85​ DWORD size = 0;
86​ GetTokenInformation(processTokenHandle, TokenUser, NULL, 0, &size);
87​
88​ if (GetLastError() == ERROR_INSUFFICIENT_BUFFER) {
89​ QByteArray buf;
90​ buf.resize(size);
91​ PTOKEN_USER userToken = reinterpret_cast<PTOKEN_USER>(buf.data());
92​ if (userToken
93​ && GetTokenInformation(processTokenHandle, TokenUser, userToken, size, &size)) {
94​ SID_NAME_USE sidNameUse;
95​ TCHAR user[MAX_PATH] = { 0 };
96​ DWORD userNameLength = MAX_PATH;
97​ TCHAR domain[MAX_PATH] = { 0 };
98​ DWORD domainNameLength = MAX_PATH;
99​
100​ if (LookupAccountSid(NULL,
101​ userToken->User.Sid,
102​ user,
103​ &userNameLength,
104​ domain,
105​ &domainNameLength,
106​ &sidNameUse))
107​ pi.processOwner = QString::fromUtf16(reinterpret_cast<const ushort *>(user));
108​ }
109​ }
110​
111​ CloseHandle(processTokenHandle);
112​ CloseHandle(handle);
113​ return pi;
114​}
115​
116​KProcessList GetProcessList()
117​{
118​ KProcessList rc;
119​
120​ PROCESSENTRY32 pe;
121​ pe.dwSize = sizeof(PROCESSENTRY32);
122​ HANDLE snapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
123​ if (snapshot == INVALID_HANDLE_VALUE)
124​ return rc;
125​
126​ for (bool hasNext = Process32First(snapshot, &pe); hasNext; hasNext = Process32Next(snapshot, &pe)) {
127​ const ProcessInfo processInf = processInfo(pe.th32ProcessID);
128​ rc.push_back(KProcessInfo(pe.th32ProcessID, QString::fromUtf16(reinterpret_cast<ushort *>(pe.szExeFile)), processInf.processOwner));
129​ }
130​ CloseHandle(snapshot);
131​ return rc;
132​}