nihilist/VeraCrypt
1
0
Fork 0
mirror of https://github.com/veracrypt/VeraCrypt synced 2024-11-28 05:53:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Windows: Don't offer creating a restore point if it is disabled in Windows.

Browse Source
This commit is contained in:
Mounir IDRASSI 2015-09-10 01:32:55 +02:00
parent 957ff20f37
commit fe6ea62363
1 changed files with 43 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/Setup/Setup.c
View File

@ -490,6 +490,39 @@ HRESULT CreateLink (char *lpszPathObj, char *lpszArguments,
return hres; return hres;
} }
BOOL IsSystemRestoreEnabled ()
{
BOOL bEnabled = FALSE;
HKEY hKey;
DWORD dwValue = 0, cbValue = sizeof (DWORD);
if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\SystemRestore", 0, KEY_READ | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS)
{
if (IsOSAtLeast (WIN_VISTA))
{
if ( (ERROR_SUCCESS == RegQueryValueEx (hKey, "RPSessionInterval", NULL, NULL, (LPBYTE) &dwValue, &cbValue))
&& (dwValue == 1)
)
{
bEnabled = TRUE;
}
}
else
{
if ( (ERROR_SUCCESS == RegQueryValueEx (hKey, "DisableSR", NULL, NULL, (LPBYTE) &dwValue, &cbValue))
&& (dwValue == 0)
)
{
bEnabled = TRUE;
}
}
RegCloseKey (hKey);
}
return bEnabled;
}
void GetProgramPath (HWND hwndDlg, char *path) void GetProgramPath (HWND hwndDlg, char *path)
{ {
ITEMIDLIST *i; ITEMIDLIST *i;
@ -2554,14 +2587,19 @@ int WINAPI WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, char *lpszComm
} }
// System Restore // System Restore
char dllPath[MAX_PATH]; if (IsSystemRestoreEnabled ())
if (GetSystemDirectory (dllPath, MAX_PATH))
{ {
StringCbCatA(dllPath, sizeof(dllPath), "\\srclient.dll"); char dllPath[MAX_PATH];
if (GetSystemDirectory (dllPath, MAX_PATH))
{
StringCbCatA(dllPath, sizeof(dllPath), "\\srclient.dll");
}
else
StringCbCopyA(dllPath, sizeof(dllPath), "C:\\Windows\\System32\\srclient.dll");
SystemRestoreDll = LoadLibrary (dllPath);
} }
else else
StringCbCopyA(dllPath, sizeof(dllPath), "C:\\Windows\\System32\\srclient.dll"); SystemRestoreDll = 0;
SystemRestoreDll = LoadLibrary (dllPath);
if (!bUninstall) if (!bUninstall)
{ {