From 9732b636e4df2c53a99c3cf9810d4f7540a88d87 Mon Sep 17 00:00:00 2001
From: Kirill Volinsky <mataes2007@gmail.com>
Date: Thu, 25 Oct 2012 18:59:20 +0000
Subject: added checking rights for miranda folder

git-svn-id: http://svn.miranda-ng.org/main/trunk@2077 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c
---
 plugins/PluginUpdater/src/Notifications.cpp | 89 ++++++++++++++++++++---------
 1 file changed, 61 insertions(+), 28 deletions(-)

diff --git a/plugins/PluginUpdater/src/Notifications.cpp b/plugins/PluginUpdater/src/Notifications.cpp
index b265090530..e3cb3d6fb8 100644
--- a/plugins/PluginUpdater/src/Notifications.cpp
+++ b/plugins/PluginUpdater/src/Notifications.cpp
@@ -483,8 +483,23 @@ INT_PTR CALLBACK DlgUpdate(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam
 		{
 			OSVERSIONINFO osver = { sizeof(osver) };
 			if (GetVersionEx(&osver) && osver.dwMajorVersion >= 6)
-				// Running Windows Vista or later (major version >= 6).
-				Button_SetElevationRequiredState(GetDlgItem(hDlg, IDOK), !IsProcessElevated());
+			{
+				wchar_t szPath[MAX_PATH];
+				GetModuleFileName(NULL, szPath, SIZEOF(szPath));
+				TCHAR *ext = _tcsrchr(szPath, '.');
+				if (ext != NULL)
+					*ext = '\0';
+				_tcscat(szPath, _T(".test"));
+				HANDLE hFile = CreateFile(szPath, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+				if (hFile == INVALID_HANDLE_VALUE)
+					// Running Windows Vista or later (major version >= 6).
+					Button_SetElevationRequiredState(GetDlgItem(hDlg, IDOK), !IsProcessElevated());
+				else
+				{
+					CloseHandle(hFile);
+					DeleteFile(szPath);
+				}
+			}
 			RECT r;
 			GetClientRect(hwndList, &r);
 
@@ -537,7 +552,7 @@ INT_PTR CALLBACK DlgUpdate(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam
 
 		// do this after filling list - enables 'ITEMCHANGED' below
 		SetWindowLongPtr(hDlg, GWLP_USERDATA, lParam);
-		Utils_RestoreWindowPositionNoSize(hDlg,0,MODNAME,"ConfirmWindow");
+		Utils_RestoreWindowPositionNoSize(hDlg, 0, MODNAME, "ConfirmWindow");
 		return TRUE;
 
 	case WM_NOTIFY:
@@ -577,43 +592,61 @@ INT_PTR CALLBACK DlgUpdate(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam
 		if (HIWORD( wParam ) == BN_CLICKED) {
 			switch(LOWORD(wParam)) {
 			case IDOK:
+			{
 				EnableWindow( GetDlgItem(hDlg, IDOK), FALSE);
 				EnableWindow( GetDlgItem(hDlg, IDC_SELALL), FALSE);
 				EnableWindow( GetDlgItem(hDlg, IDC_SELNONE), FALSE);
 
-				// Check the current process's "run as administrator" status.
-				// Elevate the process if it is not run as administrator.
-				if (!IsRunAsAdmin())
+				wchar_t szPath[MAX_PATH];
+				GetModuleFileName(NULL, szPath, SIZEOF(szPath));
+				TCHAR *ext = _tcsrchr(szPath, '.');
+				if (ext != NULL)
+					*ext = '\0';
+				_tcscat(szPath, _T(".test"));
+				HANDLE hFile = CreateFile(szPath, GENERIC_WRITE, FILE_SHARE_READ, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+				if (hFile == INVALID_HANDLE_VALUE)
 				{
-					wchar_t szPath[MAX_PATH], cmdLine[100];
-					GetModuleFileName(NULL, szPath, ARRAYSIZE(szPath));
-					TCHAR *profilename = Utils_ReplaceVarsT(_T("%miranda_profilename%"));
-					mir_sntprintf(cmdLine, SIZEOF(cmdLine), _T(" /restart:%d /profile=%s"), GetCurrentProcessId(), profilename);
-					// Launch itself as administrator.
-					SHELLEXECUTEINFO sei = { sizeof(sei) };
-					sei.lpVerb = L"runas";
-					sei.lpFile = szPath;
-					sei.lpParameters = cmdLine;
-					sei.hwnd = hDlg;
-					sei.nShow = SW_NORMAL;
-
-					if (!ShellExecuteEx(&sei))
+					// Check the current process's "run as administrator" status.
+					// Elevate the process if it is not run as administrator.
+					if (!IsRunAsAdmin())
 					{
-						DWORD dwError = GetLastError();
-						if (dwError == ERROR_CANCELLED)
+						wchar_t cmdLine[100];
+						GetModuleFileName(NULL, szPath, ARRAYSIZE(szPath));
+						TCHAR *profilename = Utils_ReplaceVarsT(_T("%miranda_profilename%"));
+						mir_sntprintf(cmdLine, SIZEOF(cmdLine), _T(" /restart:%d /profile=%s"), GetCurrentProcessId(), profilename);
+						// Launch itself as administrator.
+						SHELLEXECUTEINFO sei = { sizeof(sei) };
+						sei.lpVerb = L"runas";
+						sei.lpFile = szPath;
+						sei.lpParameters = cmdLine;
+						sei.hwnd = hDlg;
+						sei.nShow = SW_NORMAL;
+
+						if (!ShellExecuteEx(&sei))
 						{
-							// The user refused to allow privileges elevation.
-							// Do nothing ...
+							DWORD dwError = GetLastError();
+							if (dwError == ERROR_CANCELLED)
+							{
+								// The user refused to allow privileges elevation.
+								// Do nothing ...
+							}
+						}
+						else
+						{
+							DestroyWindow(hDlg);  // Quit itself
+							CallService("CloseAction", 0, 0);
+							break;
 						}
-					}
-					else
-					{
-						DestroyWindow(hDlg);  // Quit itself
-						CallService("CloseAction", 0, 0);
 					}
 				}
+				else
+				{
+					CloseHandle(hFile);
+					DeleteFile(szPath);
+				}
 				mir_forkthread(ApplyUpdates, hDlg);
 				return TRUE;
+			}
 
 			case IDC_DETAILS:
 				bShowDetails = !bShowDetails;
-- 
cgit v1.2.3