New gist

kingsamchen/self_elevation.cpp
Created 3 years ago

Raw

          
              self_elevation.cpp
            
          Permalink

1	int APIENTRY _tWinMain(HINSTANCE hInstance,
2	HINSTANCE hPrevInstance,
3	LPTSTR lpCmdLine,
4	int nCmdShow)
5	{
6	UNREFERENCED_PARAMETER(hPrevInstance);
7	UNREFERENCED_PARAMETER(lpCmdLine);
8
9	return DialogBox(hInstance, MAKEINTRESOURCE(IDD_DLGMAIN), NULL, DlgProc);
10
11	return 0;
12	}
13
14	BOOL IsProcessElevated()
15	{
16	BOOL isElevated = FALSE;
17	HANDLE hToken = NULL;
18	DWORD errCode = 0;
19
20	BOOL ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &hToken);
21	errCode = GetLastError();
22	ON_SCOPE_EXIT([&]()
23	{
24	if (hToken)
25	{
26	CloseHandle(hToken);
27	hToken = NULL;
28	}
29	});
30
31	if (!ret)
32	{
33	throw errCode;
34	}
35
36	TOKEN_ELEVATION elevationInfo;
37	DWORD retSize = 0;
38	if (!GetTokenInformation(hToken, TokenElevation, &elevationInfo,
39	sizeof(elevationInfo), &retSize))
40	{
41	errCode = GetLastError();
42	throw errCode;
43	}
44
45	isElevated = elevationInfo.TokenIsElevated;
46
47	return isElevated;
48	}
49
50	void OnClose(HWND hwnd)
51	{
52	EndDialog(hwnd, FALSE);
53	}
54
55	BOOL OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
56	{
57	try
58	{
59	BOOL isElevated = IsProcessElevated();
60	if (!isElevated)
61	{
62	HWND hBtn = GetDlgItem(hwnd, IDC_BTNELEVATION);
63	Button_SetElevationRequiredState(hBtn, TRUE);
64	}
65	}
66	catch (DWORD errCode)
67	{
68	MessageBox(hwnd, L"Retrieve Process Elevation Information Failed!",
69	L"Warning", MB_OK \| MB_ICONEXCLAMATION);
70	OutputDebugString(L"Error Code: ");
71	OutputDebugString(std::to_wstring(static_cast<UINT64>(errCode)).c_str());
72	}
73
74	return TRUE;
75	}
76
77	void OnCommand(HWND hWnd, int id, HWND hwndCtl, UINT codeNotify)
78	{
79	switch (id)
80	{
81	case IDC_BTNELEVATION:
82	BOOL isElevated = IsProcessElevated();
83	if (isElevated)
84	{
85	MessageBox(hWnd, L"Arealdy run at full administrator token",
86	L"warning", MB_OK \| MB_ICONEXCLAMATION);
87	return;
88	}
89	else
90	{
91	wchar_t path[MAX_PATH];
92	if (GetModuleFileName(NULL, path, MAX_PATH))
93	{
94	SHELLEXECUTEINFO sei = {sizeof(SHELLEXECUTEINFO)};
95	sei.lpVerb = L"runas";
96	sei.lpFile = path;
97	sei.hwnd = hWnd;
98	sei.nShow = SW_NORMAL;
99
100	if (!ShellExecuteEx(&sei))
101	{
102	DWORD errCode = GetLastError();
103	if (errCode == ERROR_CANCELLED)
104	{
105	// cancelled elevation
106	}
107	}
108	else
109	{
110	EndDialog(hWnd, TRUE);
111	}
112	}
113	}
114
115	break;
116	}
117	}
118
119	INT_PTR CALLBACK DlgProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
120	{
121	switch (msg)
122	{
123	HANDLE_MSG(hwnd, WM_INITDIALOG, OnInitDialog);
124	HANDLE_MSG(hwnd, WM_COMMAND, OnCommand);
125	HANDLE_MSG(hwnd, WM_CLOSE, OnClose);
126
127	default:
128	return FALSE;
129	}
130
131	return FALSE;
132	}

Attach files by dragging & dropping, selecting them, or pasting from the clipboard.

Styling with Markdown is supported

Terms
Privacy
Security
Status
Help

kingsamchen/self_elevation.cpp Created 3 years ago

kingsamchen/self_elevation.cpp
Created 3 years ago