Public/VeraCrypt
1
0
Fork 0
mirror of https://github.com/veracrypt/VeraCrypt.git synced 2025-11-11 11:08:02 -06:00
Code Activity

Windows: enhance ReflectDrivers mechanism by persisting it across major Windows upgrades.

Browse Source
This commit is contained in:
Mounir IDRASSI
2018-04-22 16:14:53 +02:00
parent f4edd72662
commit 3659ddd70c
3 changed files with 42 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/Common/BootEncryption.cpp
View File

@@ -2722,8 +2722,37 @@ namespace VeraCrypt
StringCchCatW (szSetupconfigLocation, ARRAYSIZE (szSetupconfigLocation), L"SetupConfig.ini");
if (bForInstall || FileExists (szSetupconfigLocation))
WritePrivateProfileStringW (L"SetupConfig", L"ReflectDrivers", bForInstall? szInstallPath : NULL, szSetupconfigLocation);
if (bForInstall)
{
wstring szPathParam = L"\"";
szPathParam += szInstallPath;
szPathParam += L"\"";
WritePrivateProfileStringW (L"SetupConfig", L"ReflectDrivers", szPathParam.c_str(), szSetupconfigLocation);
szPathParam = GetProgramConfigPath (L"SetupComplete.cmd");
FILE* scriptFile = _wfopen (szPathParam.c_str(), L"w");
if (scriptFile)
{
fwprintf (scriptFile, L"\"%s\\VeraCrypt.exe\" /PostOOBE\n", szInstallPath);
fclose (scriptFile);
WritePrivateProfileStringW (L"SetupConfig", L"PostOOBE", szPathParam.c_str(), szSetupconfigLocation);
}
}
else
{
if (FileExists (szSetupconfigLocation))
{
WritePrivateProfileStringW (L"SetupConfig", L"ReflectDrivers", NULL, szSetupconfigLocation);
WritePrivateProfileStringW (L"SetupConfig", L"PostOOBE", NULL, szSetupconfigLocation);
}
wstring scriptFilePath = GetProgramConfigPath (L"SetupComplete.cmd");
if (FileExists (scriptFilePath.c_str()))
{
::DeleteFileW (scriptFilePath.c_str());
}
}
}
}
}
@@ -3717,7 +3746,7 @@ namespace VeraCrypt
finally_do ({ EfiBootInst.DismountBootPartition(); });
EfiBootInst.MountBootPartition(0);
EfiBootInst.MountBootPartition(0);
EfiBootInst.GetFileSize(Is64BitOs()? L"\\EFI\\Boot\\bootx64.efi" : L"\\EFI\\Boot\\bootia32.efi", loaderSize);