VeraCrypt
aboutsummaryrefslogtreecommitdiff
path: root/src/Setup/Setup.h
blob: 07655060c912053fb70cd5f2df68e7e97671fd33 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/*
 Legal Notice: Some portions of the source code contained in this file were
 derived from the source code of TrueCrypt 7.1a, which is
 Copyright (c) 2003-2012 TrueCrypt Developers Association and which is
 governed by the TrueCrypt License 3.0, also from the source code of
 Encryption for the Masses 2.02a, which is Copyright (c) 1998-2000 Paul Le Roux
 and which is governed by the 'License Agreement for Encryption for the Masses'
 Modifications and additions to the original source code (contained in this file)
 and all other portions of this file are Copyright (c) 2013-2017 IDRIX
 and are governed by the Apache License 2.0 the full text of which is
 contained in the file License.txt included in VeraCrypt binary and source
 code distribution packages. */

#ifndef SETUP_H
#define SETUP_H

#ifdef __cplusplus
extern "C" {
#endif

// Specifies what files to install, where (determined by the prefix), and in what order
static wchar_t *szFiles[]=
{
	L"ALicense.txt",
	L"ALICENSE",
	L"ANOTICE",
	L"AVeraCrypt.exe",
	L"AVeraCryptExpander.exe",
	L"AVeraCrypt Format.exe",
	L"AVeraCrypt-x86.exe",
	L"AVeraCryptExpander-x86.exe",
	L"AVeraCrypt Format-x86.exe",
	L"AVeraCrypt-x64.exe",
	L"AVeraCryptExpander-x64.exe",
	L"AVeraCrypt Format-x64.exe",
	L"Averacrypt.sys",
	L"Averacrypt-x64.sys",
	L"Dveracrypt.sys",
	L"AVeraCrypt Setup.exe",
	L"XLanguages.zip",
	L"Xdocs.zip",
};

// Specifies what files are included in self-extracting packages (no other files will be packaged or extracted).
static wchar_t *szCompressedFiles[]=
{
	L"License.txt",
	L"LICENSE",
	L"NOTICE",
	L"VeraCrypt.exe",
	L"VeraCryptExpander.exe",
	L"VeraCrypt Format.exe",
	L"VeraCrypt-x64.exe",
	L"VeraCryptExpander-x64.exe",
	L"VeraCrypt Format-x64.exe",
	L"veracrypt.sys",
	L"veracrypt-x64.sys",
	L"Languages.zip",
	L"docs.zip"
};

#define FILENAME_64BIT_DRIVER	L"veracrypt-x64.sys"
#define NBR_COMPRESSED_FILES (sizeof(szCompressedFiles) / sizeof(szCompressedFiles[0]))

void localcleanup (void);
BOOL StatDeleteFile ( wchar_t *lpszFile, BOOL bCheckForOldFile );
BOOL StatRemoveDirectory ( wchar_t *lpszDir );
HRESULT CreateLink ( wchar_t *lpszPathObj , wchar_t *lpszArguments , wchar_t *lpszPathLink );
void GetProgramPath ( HWND hwndDlg , wchar_t *path );
void StatusMessage (HWND hwndDlg, char *stringId);
void StatusMessageParam (HWND hwndDlg, char *stringId, const wchar_t *param);
void ClearLogWindow (HWND hwndDlg);
void RegMessage ( HWND hwndDlg , const wchar_t *txt );
void RegRemoveMessage (HWND hwndDlg, const wchar_t *txt);
void _cdecl CopyMessage ( HWND hwndDlg , const wchar_t *txt );
void RemoveMessage ( HWND hwndDlg , const wchar_t *txt );
void IconMessage ( HWND hwndDlg , const wchar_t *txt );
static int CALLBACK BrowseCallbackProc ( HWND hwnd , UINT uMsg , LPARAM lp , LPARAM pData );
void LoadLicense ( HWND hwndDlg );
void DetermineUpgradeDowngradeStatus (BOOL bCloseDriverHandle, LONG *driverVersionPtr);
BOOL DoFilesInstall ( HWND hwndDlg , wchar_t *szDestDir );
BOOL DoRegInstall ( HWND hwndDlg , wchar_t *szDestDir , BOOL bInstallType );
BOOL DoRegUninstall (HWND hwndDlg, BOOL bRemoveDeprecated);
BOOL DoServiceUninstall ( HWND hwndDlg , wchar_t *lpszService );
BOOL DoDriverUnload ( HWND hwndDlg );
BOOL DoShortcutsInstall ( HWND hwndDlg , wchar_t *szDestDir , BOOL bProgGroup, BOOL bDesktopIcon );
BOOL DoShortcutsUninstall (HWND hwndDlg, wchar_t *szDestDir);
void OutcomePrompt ( HWND hwndDlg , BOOL bOK );
void DoUninstall ( void *hwndDlg );
void DoInstall ( void *hwndDlg );
void SetInstallationPath (HWND hwndDlg);
BOOL UpgradeBootLoader (HWND hwndDlg);
BOOL CALLBACK InstallDlgProc ( HWND hwndDlg , UINT msg , WPARAM wParam , LPARAM lParam );

extern BOOL bDevm;
extern BOOL Rollback;
extern BOOL bUpgrade;
extern BOOL bPossiblyFirstTimeInstall;
extern BOOL bRepairMode;
extern BOOL bReinstallMode;
extern BOOL bSystemRestore;
extern BOOL bDisableSwapFiles;
extern BOOL bForAllUsers;
extern BOOL bRegisterFileExt;
extern BOOL bAddToStartMenu;
extern BOOL bDesktopIcon;
extern BOOL bDesktopIconStatusDetermined;
extern BOOL SystemEncryptionUpdate;
extern BOOL bRestartRequired;
extern HMODULE volatile SystemRestoreDll;
extern wchar_t InstallationPath[TC_MAX_PATH];
extern wchar_t SetupFilesDir[TC_MAX_PATH];

#ifdef __cplusplus
}
#endif

#endif	// #ifndef SETUP_H
">GetEncryptionMode () const { if_debug (ValidateState ()); return EA->GetMode(); } void Volume::Open (const VolumePath &volumePath, bool preserveTimestamps, shared_ptr <VolumePassword> password, int pim, shared_ptr <Pkcs5Kdf> kdf, bool truecryptMode, shared_ptr <KeyfileList> keyfiles, VolumeProtection::Enum protection, shared_ptr <VolumePassword> protectionPassword, int protectionPim, shared_ptr <Pkcs5Kdf> protectionKdf, shared_ptr <KeyfileList> protectionKeyfiles, bool sharedAccessAllowed, VolumeType::Enum volumeType, bool useBackupHeaders, bool partitionInSystemEncryptionScope) { make_shared_auto (File, file); File::FileOpenFlags flags = (preserveTimestamps ? File::PreserveTimestamps : File::FlagsNone); try { if (protection == VolumeProtection::ReadOnly) file->Open (volumePath, File::OpenRead, File::ShareRead, flags); else file->Open (volumePath, File::OpenReadWrite, File::ShareNone, flags); } catch (SystemException &e) { if (e.GetErrorCode() == #ifdef TC_WINDOWS ERROR_SHARING_VIOLATION) #else EAGAIN) #endif { if (!sharedAccessAllowed) throw VolumeHostInUse (SRC_POS); file->Open (volumePath, protection == VolumeProtection::ReadOnly ? File::OpenRead : File::OpenReadWrite, File::ShareReadWriteIgnoreLock, flags); } else throw; } return Open (file, password, pim, kdf, truecryptMode, keyfiles, protection, protectionPassword, protectionPim, protectionKdf,protectionKeyfiles, volumeType, useBackupHeaders, partitionInSystemEncryptionScope); } void Volume::Open (shared_ptr <File> volumeFile, shared_ptr <VolumePassword> password, int pim, shared_ptr <Pkcs5Kdf> kdf, bool truecryptMode, shared_ptr <KeyfileList> keyfiles, VolumeProtection::Enum protection, shared_ptr <VolumePassword> protectionPassword, int protectionPim, shared_ptr <Pkcs5Kdf> protectionKdf,shared_ptr <KeyfileList> protectionKeyfiles, VolumeType::Enum volumeType, bool useBackupHeaders, bool partitionInSystemEncryptionScope) { if (!volumeFile) throw ParameterIncorrect (SRC_POS); // TrueCrypt doesn't support SHA-256 and Streebog if (kdf && truecryptMode && (kdf->GetName() == L"HMAC-SHA-256" || kdf->GetName() == L"HMAC-Streebog")) throw UnsupportedAlgoInTrueCryptMode (SRC_POS); Protection = protection; VolumeFile = volumeFile; SystemEncryption = partitionInSystemEncryptionScope; try { VolumeHostSize = VolumeFile->Length(); shared_ptr <VolumePassword> passwordKey = Keyfile::ApplyListToPassword (keyfiles, password); bool skipLayoutV1Normal = false; // Test volume layouts foreach (shared_ptr <VolumeLayout> layout, VolumeLayout::GetAvailableLayouts (volumeType)) { if (skipLayoutV1Normal && typeid (*layout) == typeid (VolumeLayoutV1Normal)) { // Skip VolumeLayoutV1Normal as it shares header location with VolumeLayoutV2Normal continue; } if (useBackupHeaders && !layout->HasBackupHeader()) continue; SecureBuffer headerBuffer (layout->GetHeaderSize()); if (layout->HasDriveHeader()) { if (!partitionInSystemEncryptionScope) continue; if (!GetPath().IsDevice()) throw PartitionDeviceRequired (SRC_POS); File driveDevice; driveDevice.Open (DevicePath (wstring (GetPath())).ToHostDriveOfPartition()); int headerOffset = layout->GetHeaderOffset(); if (headerOffset >= 0) driveDevice.SeekAt (headerOffset); else driveDevice.SeekEnd (headerOffset); if (driveDevice.Read (headerBuffer) != layout->GetHeaderSize()) continue; } else { if (partitionInSystemEncryptionScope) continue; int headerOffset = useBackupHeaders ? layout->GetBackupHeaderOffset() : layout->GetHeaderOffset(); if (headerOffset >= 0) VolumeFile->SeekAt (headerOffset); else VolumeFile->SeekEnd (headerOffset); if (VolumeFile->Read (headerBuffer) != layout->GetHeaderSize()) continue; } EncryptionAlgorithmList layoutEncryptionAlgorithms = layout->GetSupportedEncryptionAlgorithms(); EncryptionModeList layoutEncryptionModes = layout->GetSupportedEncryptionModes(); if (typeid (*layout) == typeid (VolumeLayoutV2Normal)) { skipLayoutV1Normal = true; // Test all algorithms and modes of VolumeLayoutV1Normal as it shares header location with VolumeLayoutV2Normal layoutEncryptionAlgorithms = EncryptionAlgorithm::GetAvailableAlgorithms(); layoutEncryptionModes = EncryptionMode::GetAvailableModes(); } shared_ptr <VolumeHeader> header = layout->GetHeader(); if (header->Decrypt (headerBuffer, *passwordKey, pim, kdf, truecryptMode, layout->GetSupportedKeyDerivationFunctions(truecryptMode), layoutEncryptionAlgorithms, layoutEncryptionModes)) { // Header decrypted if (!truecryptMode && typeid (*layout) == typeid (VolumeLayoutV2Normal) && header->GetRequiredMinProgramVersion() < 0x10b) { // VolumeLayoutV1Normal has been opened as VolumeLayoutV2Normal layout.reset (new VolumeLayoutV1Normal); header->SetSize (layout->GetHeaderSize()); layout->SetHeader (header); } TrueCryptMode = truecryptMode; Pim = pim; Type = layout->GetType(); SectorSize = header->GetSectorSize(); VolumeDataOffset = layout->GetDataOffset (VolumeHostSize); VolumeDataSize = layout->GetDataSize (VolumeHostSize); EncryptedDataSize = header->GetEncryptedAreaLength(); Header = header; Layout = layout; EA = header->GetEncryptionAlgorithm(); EncryptionMode &mode = *EA->GetMode(); if (layout->HasDriveHeader()) { if (header->GetEncryptedAreaLength() != header->GetVolumeDataSize()) { EncryptionNotCompleted = true; // we avoid writing data to the partition since it is only partially encrypted Protection = VolumeProtection::ReadOnly; } uint64 partitionStartOffset = VolumeFile->GetPartitionDeviceStartOffset(); if (partitionStartOffset < header->GetEncryptedAreaStart() || partitionStartOffset >= header->GetEncryptedAreaStart() + header->GetEncryptedAreaLength()) throw PasswordIncorrect (SRC_POS); EncryptedDataSize -= partitionStartOffset - header->GetEncryptedAreaStart(); mode.SetSectorOffset (partitionStartOffset / ENCRYPTION_DATA_UNIT_SIZE); } // Volume protection if (Protection == VolumeProtection::HiddenVolumeReadOnly) { if (Type == VolumeType::Hidden) throw PasswordIncorrect (SRC_POS); else { try { Volume protectedVolume; protectedVolume.Open (VolumeFile, protectionPassword, protectionPim, protectionKdf, truecryptMode, protectionKeyfiles, VolumeProtection::ReadOnly, shared_ptr <VolumePassword> (), 0, shared_ptr <Pkcs5Kdf> (),shared_ptr <KeyfileList> (), VolumeType::Hidden, useBackupHeaders); if (protectedVolume.GetType() != VolumeType::Hidden) ParameterIncorrect (SRC_POS); ProtectedRangeStart = protectedVolume.VolumeDataOffset; ProtectedRangeEnd = protectedVolume.VolumeDataOffset + protectedVolume.VolumeDataSize; } catch (PasswordException&) { if (protectionKeyfiles && !protectionKeyfiles->empty()) throw ProtectionPasswordKeyfilesIncorrect (SRC_POS); throw ProtectionPasswordIncorrect (SRC_POS); } } } return; } } if (partitionInSystemEncryptionScope) throw PasswordOrKeyboardLayoutIncorrect (SRC_POS); if (!partitionInSystemEncryptionScope && GetPath().IsDevice()) { // Check if the device contains VeraCrypt Boot Loader try { File driveDevice; driveDevice.Open (DevicePath (wstring (GetPath())).ToHostDriveOfPartition()); Buffer mbr (VolumeFile->GetDeviceSectorSize()); driveDevice.ReadAt (mbr, 0); // Search for the string "VeraCrypt" or "TrueCrypt" const char* bootSignature = truecryptMode? "TrueCrypt" : TC_APP_NAME; size_t nameLen = strlen (bootSignature); for (size_t i = 0; i < mbr.Size() - nameLen; ++i) { if (memcmp (mbr.Ptr() + i, bootSignature, nameLen) == 0) throw PasswordOrMountOptionsIncorrect (SRC_POS); } } catch (PasswordOrMountOptionsIncorrect&) { throw; } catch (...) { } } if (keyfiles && !keyfiles->empty()) throw PasswordKeyfilesIncorrect (SRC_POS); throw PasswordIncorrect (SRC_POS); } catch (...) { Close(); throw; } } void Volume::ReadSectors (const BufferPtr &buffer, uint64 byteOffset) { if_debug (ValidateState ()); uint64 length = buffer.Size(); uint64 hostOffset = VolumeDataOffset + byteOffset; size_t bufferOffset = 0; if (length % SectorSize != 0 || byteOffset % SectorSize != 0) throw ParameterIncorrect (SRC_POS); if (VolumeFile->ReadAt (buffer, hostOffset) != length) throw MissingVolumeData (SRC_POS); // first sector can be unencrypted in some cases (e.g. windows repair) // detect this case by looking for NTFS header if (SystemEncryption && (hostOffset == 0) && ((BE64 (*(uint64 *) buffer.Get ())) == 0xEB52904E54465320ULL)) { bufferOffset = (size_t) SectorSize; hostOffset += SectorSize; length -= SectorSize; } if (length) { if (EncryptionNotCompleted) { // if encryption is not complete, we decrypt only the encrypted sectors if (hostOffset < EncryptedDataSize) { uint64 encryptedLength = VC_MIN (length, (EncryptedDataSize - hostOffset)); EA->DecryptSectors (buffer.GetRange (bufferOffset, encryptedLength), hostOffset / SectorSize, encryptedLength / SectorSize, SectorSize); } } else EA->DecryptSectors (buffer.GetRange (bufferOffset, length), hostOffset / SectorSize, length / SectorSize, SectorSize); } TotalDataRead += length; } void Volume::ReEncryptHeader (bool backupHeader, const ConstBufferPtr &newSalt, const ConstBufferPtr &newHeaderKey, shared_ptr <Pkcs5Kdf> newPkcs5Kdf) { if_debug (ValidateState ()); if (Protection == VolumeProtection::ReadOnly) throw VolumeReadOnly (SRC_POS); SecureBuffer newHeaderBuffer (Layout->GetHeaderSize()); Header->EncryptNew (newHeaderBuffer, newSalt, newHeaderKey, newPkcs5Kdf); int headerOffset = backupHeader ? Layout->GetBackupHeaderOffset() : Layout->GetHeaderOffset(); if (headerOffset >= 0) VolumeFile->SeekAt (headerOffset); else VolumeFile->SeekEnd (headerOffset); VolumeFile->Write (newHeaderBuffer); } void Volume::ValidateState () const { if (VolumeFile.get() == nullptr) throw NotInitialized (SRC_POS); } void Volume::WriteSectors (const ConstBufferPtr &buffer, uint64 byteOffset) { if_debug (ValidateState ()); uint64 length = buffer.Size(); uint64 hostOffset = VolumeDataOffset + byteOffset; if (length % SectorSize != 0 || byteOffset % SectorSize != 0 || byteOffset + length > VolumeDataSize) throw ParameterIncorrect (SRC_POS); if (Protection == VolumeProtection::ReadOnly) throw VolumeReadOnly (SRC_POS); if (HiddenVolumeProtectionTriggered) throw VolumeProtected (SRC_POS); if (Protection == VolumeProtection::HiddenVolumeReadOnly) CheckProtectedRange (hostOffset, length); SecureBuffer encBuf (buffer.Size()); encBuf.CopyFrom (buffer); EA->EncryptSectors (encBuf, hostOffset / SectorSize, length / SectorSize, SectorSize); VolumeFile->WriteAt (encBuf, hostOffset); TotalDataWritten += length; uint64 writeEndOffset = byteOffset + buffer.Size(); if (writeEndOffset > TopWriteOffset) TopWriteOffset = writeEndOffset; } }