Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-07-06 04:35:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

[REGEDIT]

Browse source 
- Acquire the necessary privileges when loading/unloading hives. Patch by Hermes Belusca.
See issue #7159 for more details.

svn path=/trunk/; revision=56825
This commit is contained in:
Thomas Faber 2012-07-04 18:28:49 +00:00
parent feae1a7ab7
commit 7b2fb517d6
1 changed files with 56 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
reactos/base/applications/regedit/framewnd.c
View file

@ -312,6 +312,38 @@ static INT_PTR CALLBACK LoadHive_KeyNameInHookProc(HWND hWndDlg, UINT uMsg, WPAR
return FALSE;
}
static BOOL EnablePrivilege(LPCTSTR lpszPrivilegeName, LPCTSTR lpszSystemName, BOOL bEnablePrivilege)
{
BOOL bRet = FALSE;
HANDLE hToken = NULL;
if (OpenProcessToken(GetCurrentProcess(),
TOKEN_ADJUST_PRIVILEGES,
&hToken))
{
TOKEN_PRIVILEGES tp;
tp.PrivilegeCount = 1;
tp.Privileges[0].Attributes = (bEnablePrivilege ? SE_PRIVILEGE_ENABLED : 0);
if (LookupPrivilegeValue(lpszSystemName,
lpszPrivilegeName,
&tp.Privileges[0].Luid))
{
bRet = AdjustTokenPrivileges(hToken,
FALSE,
&tp,
sizeof(tp),
NULL,
NULL);
}
CloseHandle(hToken);
}
return bRet;
}
static BOOL LoadHive(HWND hWnd)
{
OPENFILENAME ofn;
@ -340,7 +372,19 @@ static BOOL LoadHive(HWND hWnd)
{
if(DialogBoxParam(hInst, MAKEINTRESOURCE(IDD_LOADHIVE), hWnd, &LoadHive_KeyNameInHookProc, (LPARAM)xPath))
{
LONG regLoadResult = RegLoadKey(hRootKey, xPath, ofn.lpstrFile);
LONG regLoadResult;
/* Enable the required privileges */
EnablePrivilege(SE_BACKUP_NAME, NULL, TRUE);
EnablePrivilege(SE_RESTORE_NAME, NULL, TRUE);
/* Load the hive */
regLoadResult = RegLoadKey(hRootKey, xPath, ofn.lpstrFile);
/* Disable the privileges */
EnablePrivilege(SE_RESTORE_NAME, NULL, FALSE);
EnablePrivilege(SE_BACKUP_NAME, NULL, FALSE);
if(regLoadResult == ERROR_SUCCESS)
{
/* refresh tree and list views */
@ -373,8 +417,18 @@ static BOOL UnloadHive(HWND hWnd)
pszKeyPath = GetItemPath(g_pChildWnd->hTreeWnd, 0, &hRootKey);
/* load and set the caption and flags for dialog */
LoadString(hInst, IDS_UNLOAD_HIVE, Caption, COUNT_OF(Caption));
/* now unload the hive */
/* Enable the required privileges */
EnablePrivilege(SE_BACKUP_NAME, NULL, TRUE);
EnablePrivilege(SE_RESTORE_NAME, NULL, TRUE);
/* Unload the hive */
regUnloadResult = RegUnLoadKey(hRootKey, pszKeyPath);
/* Disable the privileges */
EnablePrivilege(SE_RESTORE_NAME, NULL, FALSE);
EnablePrivilege(SE_BACKUP_NAME, NULL, FALSE);
if(regUnloadResult == ERROR_SUCCESS)
{
/* refresh tree and list views */