#include "Tcdefs.h"

#include <windowsx.h>

#include <dbghelp.h>

#include <dbt.h>

#include <Setupapi.h>

DWORD bytesWritten;

BOOL res = TRUE;

DWORD dwLastError = 0;

dst = CreateFile (destinationFile,

GENERIC_WRITE,

FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, bAppend ? OPEN_EXISTING : CREATE_ALWAYS, 0, NULL);

dwLastError = GetLastError();

{

wchar_t renamedPath[TC_MAX_PATH + 1];

StringCbCopyW (renamedPath, sizeof(renamedPath), destinationFile);

StringCbCatW (renamedPath, sizeof(renamedPath), VC_FILENAME_RENAMED_SUFFIX);

/* rename the locked file in order to be able to create a new one */

if (MoveFileEx (destinationFile, renamedPath, MOVEFILE_REPLACE_EXISTING))

{

}

else

{

/* delete the renamed file when the machine reboots */

MoveFileEx (renamedPath, NULL, MOVEFILE_DELAY_UNTIL_REBOOT);

}

}

}

if (dst == INVALID_HANDLE_VALUE)

WINTRUST_DATA WVTData = {0};

wchar_t filePath [TC_MAX_PATH + 1024];

// Strip quotation marks (if any)

if (path [0] == L'"')

{

bRet = TRUE;

}

#ifdef SETUP_DLL

// we get real version from Kernel32.dll version since MSI always sets current version to 6.0

// https://stackoverflow.com/questions/49335885/windows-10-not-detecting-on-installshield/49343826#49343826

// Version

SendMessage (GetDlgItem (hwndDlg, IDT_ABOUT_VERSION), WM_SETFONT, (WPARAM) hUserBoldFont, 0);

#if (defined(_DEBUG) || defined(DEBUG))

StringCbCatW (szTmp, sizeof(szTmp), L" (debug)");

#endif

StringCbCopyW (newCmdLine, sizeof(newCmdLine), L"/q UAC ");

StringCbCatW (newCmdLine, sizeof (newCmdLine), args);

{

if (hwndDlg)

handleWin32Error (hwndDlg, SRC_POS);

InitOSVersionInfo();

{

}

SetDefaultDllDirectoriesFn = (SetDefaultDllDirectoriesPtr) GetProcAddress (GetModuleHandle(L"kernel32.dll"), "SetDefaultDllDirectories");

InitHelpFileName ();

#ifndef SETUP

EnableRamEncryption ((ReadDriverConfigurationFlags() & VC_DRIVER_CONFIG_ENABLE_RAM_ENCRYPTION) ? TRUE : FALSE);

if (IsRamEncryptionEnabled())

{

if (!InitializeSecurityParameters(GetAppRandomSeed))

AbortProcess("OUTOFMEMORY");

}

if (!EncryptionThreadPoolStart (ReadEncryptionThreadPoolFreeCpuCountLimit()))

{

handleWin32Error (NULL, SRC_POS);

BOOL GetDriveLabel (int driveNo, wchar_t *label, int labelSize)

{

DWORD fileSystemFlags;

return GetVolumeInformationW (root, label, labelSize / 2, NULL, NULL, &fileSystemFlags, NULL, 0);

}

}

// Find extra boot partition

{

if (drive.ContainsSystem)

{

{

if (sysDrivePartition.Bootable)

{

else

*tmp = 0;

file = FindFirstFile (driverPath, &find);

return status;

}

{

IFileDialog *pfd = NULL;

HRESULT hr;

}

// Set the title.

{

}

IShellItem *psi;

BOOL UpdateDriveCustomLabel (int driveNo, wchar_t* effectiveLabel, BOOL bSetValue)

{

wchar_t wszRegPath[MAX_PATH];

HKEY hKey;

LSTATUS lStatus;

DWORD cbLabelLen = (DWORD) ((wcslen (effectiveLabel) + 1) * sizeof (wchar_t));

*/

{

int thid, i;

char *tmp_salt = {"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xAA\xBB\xCC\xDD\xEE\xFF\x01\x23\x45\x67\x89\xAB\xCD\xEF\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xAA\xBB\xCC\xDD\xEE\xFF\x01\x23\x45\x67\x89\xAB\xCD\xEF\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xAA\xBB\xCC\xDD\xEE\xFF"};

for (thid = FIRST_PRF_ID; thid <= LAST_PRF_ID; thid++)

case SHA512:

/* PKCS-5 test with HMAC-SHA-512 used as the PRF */

break;

case SHA256:

/* PKCS-5 test with HMAC-SHA-256 used as the PRF */

break;

#ifndef WOLFCRYPT_BACKEND

case BLAKE2S:

/* PKCS-5 test with HMAC-BLAKE2s used as the PRF */

break;

case WHIRLPOOL:

/* PKCS-5 test with HMAC-Whirlpool used as the PRF */

break;

case STREEBOG:

/* PKCS-5 test with HMAC-STREEBOG used as the PRF */

break;

}

#endif

{

if (thid == SHA256)

{

benchmarkTable[benchmarkTotalItems].meanBytesPerSec = (benchmarkTable[benchmarkTotalItems].meanBytesPerSec * 26);

}

else

{

benchmarkTable[benchmarkTotalItems].meanBytesPerSec = (benchmarkTable[benchmarkTotalItems].meanBytesPerSec * 21) / 5;

}

}

}

if (EAInitMode (ci, ci->k2))

{

int i;

if (IsRamEncryptionEnabled ())

VcProtectKeys (ci, VcGetEncryptionID (ci));

for (i = 0; i < 10; i++)

{

if (!EAInitMode (ci, ci->k2))

goto counter_error;

if (IsRamEncryptionEnabled ())

VcProtectKeys (ci, VcGetEncryptionID (ci));

if (QueryPerformanceCounter (&performanceCountStart) == 0)

goto counter_error;

else

{

if (bEncrypt)

{

{

if (driveMap & (1 << i))

{

SHChangeNotify (eventId, SHCNF_PATH, root, NULL);

wstring drivePath = L"\\\\.\\X:";

HANDLE dev = INVALID_HANDLE_VALUE;

VOLUME_DISK_EXTENTS extents = {0};

drivePath[4] = root[0];

if ((dev = CreateFile (drivePath.c_str(),0, 0, NULL, OPEN_EXISTING, 0, NULL)) != INVALID_HANDLE_VALUE)

{

{

if (extents.NumberOfDiskExtents > 0)

{

if (bDevice && mount.bProtectHiddenVolume)

{

{

OPEN_TEST_STRUCT openTestStruct;

memset (&openTestStruct, 0, sizeof (openTestStruct));

openTestStruct.bDetectTCBootLoader = TRUE;

WarningDirect ((GetWrongPasswordErrorMessage (hwndDlg) + L"\n\n" + GetString ("HIDDEN_VOL_PROT_PASSWORD_US_KEYB_LAYOUT")).c_str(), hwndDlg);

else

handleError (hwndDlg, mount.nReturnCode, SRC_POS);

if (mount.FilesystemDirty)

{

wchar_t msg[1024];

StringCbPrintfW (msg, sizeof(msg), GetString ("MOUNTED_VOLUME_DIRTY"), mountPoint);

if (AskWarnYesNoStringTopmost (msg, hwndDlg) == IDYES)

&& !IsFileOnReadOnlyFilesystem (volumePath))

{

wchar_t msg[1024];

StringCbPrintfW (msg, sizeof(msg), GetString ("MOUNTED_CONTAINER_FORCED_READ_ONLY"), mountPoint);

WarningDirect (msg, hwndDlg);

&& bDevice)

{

wchar_t msg[1024];

StringCbPrintfW (msg, sizeof(msg), GetString ("MOUNTED_DEVICE_FORCED_READ_ONLY"), mountPoint);

WarningDirect (msg, hwndDlg);

&& wcsstr (volumePath, L"\\Device\\Harddisk") == volumePath)

{

wchar_t msg[1024];

StringCbPrintfW (msg, sizeof(msg), GetString ("MOUNTED_DEVICE_FORCED_READ_ONLY_WRITE_PROTECTION"), mountPoint);

WarningDirect (msg, hwndDlg);

&& bDevice)

{

wchar_t msg[1024];

StringCbPrintfW (msg, sizeof(msg), GetString ("PARTIAL_SYSENC_MOUNT_READONLY"), mountPoint);

WarningDirect (msg, hwndDlg);

}

// Undo SHCNE_DRIVEREMOVED

SHChangeNotify (SHCNE_DRIVEADD, SHCNF_PATH, root, NULL);

return FALSE;

for (i = 0; i < 26; i++)

{

StringCchCopyW (link, MAX_PATH, L"\\DosDevices\\");

StringCchCatW (link, MAX_PATH, drive);

{

wchar_t sysPath[TC_MAX_PATH];

{

typedef UINT (WINAPI *GetSystemWow64Directory_t) (LPWSTR lpBuffer, UINT uSize);

}

else

{

if (r == ERROR_FILE_NOT_FOUND)

{

// Try the secondary help file

if (r == ERROR_FILE_NOT_FOUND)

{

BOOL IsServerOS ()

{

}

// Returns TRUE, if the currently running operating system is installed in a hidden volume. If it's not, or if

// there's an error, returns FALSE.

BOOL IsHiddenOSRunning (void)

{

wstring osname = L"win";

osVer.dwOSVersionInfoSize = sizeof (OSVERSIONINFOEXW);

{

}

}

#ifdef SETUP

wchar_t page[TC_MAX_PATH] = {0};

wchar_t installDir[TC_MAX_PATH] = {0};

BOOL buildUrl = TRUE;

ArrowWaitCursor ();

}

else

{

if (((r == ERROR_FILE_NOT_FOUND) || (r == ERROR_PATH_NOT_FOUND)) && buildUrl)

{

int volumeType;

wchar_t szDiskFile[TC_MAX_PATH], szCFDevice[TC_MAX_PATH];

wchar_t szDosDevice[TC_MAX_PATH];

LARGE_INTEGER headerOffset;

DWORD dwResult;

DISK_GEOMETRY_EX deviceGeometry;

}

{

CRYPTO_INFO *newCryptoInfo = NULL;

void CheckFilesystem (HWND hwndDlg, int driveNo, BOOL fixErrors)

{

wchar_t msg[1024], param[1024], cmdPath[MAX_PATH];

if (fixErrors && AskWarnYesNo ("FILESYS_REPAIR_CONFIRM_BACKUP", hwndDlg) == IDNO)

return;

BOOL LaunchWindowsIsoBurner (HWND hwnd, const wchar_t *isoPath)

{

wchar_t path[MAX_PATH*2] = { 0 };

if (SUCCEEDED(SHGetFolderPath (NULL, CSIDL_SYSTEM, NULL, 0, path)))

StringCbCatW (path, MAX_PATH*2, L"\\" ISO_BURNER_TOOL);

else

StringCbCopyW (path, MAX_PATH*2, L"C:\\Windows\\System32\\" ISO_BURNER_TOOL);

if (r <= 32)

{

handleWin32Error (hwnd, SRC_POS);

return FALSE;

SHGetSpecialFolderLocation (hwndDlg, CSIDL_PROGRAM_FILES, &itemList);

SHGetPathFromIDList (itemList, path);

{

}

StringCbCatW (path, sizeof(path), L"\\VeraCrypt\\");

}

}

#define RtlGenRandom SystemFunction036

extern "C" BOOLEAN NTAPI RtlGenRandom(PVOID RandomBuffer, ULONG RandomBufferLength);

DWORD PasswordEditDropTarget::GotEnter(void)

{

TCHAR szClassName[64];

int maxLen;

HWND hChild = WindowFromPoint (m_DropPoint);

// check that we are on password edit control (we use maximum length to correctly identify password fields since they don't always have ES_PASSWORD style (if the the user checked show password)

if(m_Data)

{

TCHAR szClassName[64];

int maxLen;

HWND hChild = WindowFromPoint (m_DropPoint);

if (hChild && GetClassName (hChild, szClassName, ARRAYSIZE (szClassName)) && (0 == _tcsicmp (szClassName, _T("EDIT")))

return dwRet;

}

#endif // VC_COMREG