-/*

- Derived from source code of TrueCrypt 7.1a, which is

- Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed

- by the TrueCrypt License 3.0.

-

- Modifications and additions to the original source code (contained in this file)

- and all other portions of this file are Copyright (c) 2013-2016 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.

-*/

-

-#define TC_MAIN_COM_VERSION_MAJOR 2

-#define TC_MAIN_COM_VERSION_MINOR 7

-

-#define TC_FORMAT_COM_VERSION_MAJOR 2

-#define TC_FORMAT_COM_VERSION_MINOR 5

-

-#include <atlbase.h>

-#include <comdef.h>

-#include <statreg.h>

-#include <windows.h>

-#include "ComSetup.h"

-#include "Dlgcode.h"

-#include "Resource.h"

-#include "../Mount/MainCom_i.c"

-#include "../Format/FormatCom_i.c"

-

-

-extern "C" BOOL RegisterComServers (wchar_t *modulePath)

-{

- BOOL ret = TRUE;

- wchar_t mainModule[1024], formatModule[1024];

- CComPtr<ITypeLib> tl, tl2;

-

- wsprintfW (mainModule, L"%sVeraCrypt.exe", modulePath);

- wsprintfW (formatModule, L"%sVeraCrypt Format.exe", modulePath);

-

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR, 0, SYS_WIN32);

- // unregister older versions that may still exist

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-3, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-2, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-1, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR-1, 0, SYS_WIN32);

-

- wchar_t setupModule[MAX_PATH];

- GetModuleFileNameW (NULL, setupModule, sizeof (setupModule) / sizeof (setupModule[0]));

-

- CRegObject ro;

- HRESULT r;

-

- if (!SUCCEEDED (r = ro.FinalConstruct ())

- || !SUCCEEDED (r = ro.AddReplacement (L"MAIN_MODULE", mainModule))

- || !SUCCEEDED (r = ro.AddReplacement (L"FORMAT_MODULE", formatModule))

- || !SUCCEEDED (r = ro.ResourceRegister (setupModule, IDR_COMREG, L"REGISTRY"))

- || !SUCCEEDED (r = LoadTypeLib (mainModule, &tl))

- || !SUCCEEDED (r = RegisterTypeLib (tl, mainModule, 0))

- || !SUCCEEDED (r = LoadTypeLib (formatModule, &tl2))

- || !SUCCEEDED (r = RegisterTypeLib (tl2, formatModule, 0)))

- {

- MessageBox (MainDlg, _com_error (r).ErrorMessage(), _T(TC_APP_NAME), MB_ICONERROR);

- ret = FALSE;

- }

-

- ro.FinalRelease ();

- return ret;

-}

-

-

-extern "C" BOOL UnregisterComServers (wchar_t *modulePath)

-{

- BOOL ret;

-

- if (UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR, 0, SYS_WIN32) != S_OK)

- return FALSE;

- if (UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR, 0, SYS_WIN32) != S_OK)

- return FALSE;

-

- // unregister older versions that may still exist

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-3, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-2, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptMainCom, TC_MAIN_COM_VERSION_MAJOR, TC_MAIN_COM_VERSION_MINOR-1, 0, SYS_WIN32);

- UnRegisterTypeLib (LIBID_TrueCryptFormatCom, TC_FORMAT_COM_VERSION_MAJOR, TC_FORMAT_COM_VERSION_MINOR-1, 0, SYS_WIN32);

-

- wchar_t module[1024];

- CRegObject ro;

- ro.FinalConstruct ();

-

- wsprintfW (module, L"%sVeraCrypt.exe", modulePath);

- ro.AddReplacement (L"MAIN_MODULE", module);

-

- wsprintfW (module, L"%sVeraCrypt Format.exe", modulePath);

- ro.AddReplacement (L"FORMAT_MODULE", module);

-

- wchar_t setupModule[MAX_PATH];

- GetModuleFileNameW (NULL, setupModule, sizeof (setupModule) / sizeof (setupModule[0]));

-

- ret = ro.ResourceUnregister (setupModule, IDR_COMREG, L"REGISTRY") == S_OK;

-

- ro.FinalRelease ();

- return ret;

-}

- Modifications and additions to the original source code (contained in this file)

-*/

-

-#ifdef __cplusplus

-extern "C" {

-#endif

-

-BOOL RegisterComServers (wchar_t *modulePath);

-BOOL UnregisterComServers (wchar_t *modulePath);

-

-#ifdef __cplusplus

-}

-#endif

-HKCR

-{

- ForceRemove VeraCrypt.1 = s 'VeraCrypt class'

- {

- CLSID = s '{FE8B3B95-C80C-41f7-830F-FBA271C26F7E}'

- }

-

- ForceRemove VeraCrypt = s 'VeraCrypt class'

- {

- CLSID = s '{FE8B3B95-C80C-41f7-830F-FBA271C26F7E}'

- CurVer = s 'VeraCrypt.1'

- }

-

- NoRemove CLSID

- {

- ForceRemove {FE8B3B95-C80C-41f7-830F-FBA271C26F7E} = s 'VeraCrypt class'

- {

- ProgID = s 'VeraCrypt.1'

- VersionIndependentProgID = s 'VeraCrypt'

- LocalServer32 = s '"%MAIN_MODULE%"'

-

- TypeLib = s '{9ACF6176-5FC4-4690-A025-B3306A50EB6A}'

-

- Elevation

- {

- val Enabled = d 1

- val IconReference = s '@%MAIN_MODULE%,-501'

- }

-

- val AppId = s '{FE8B3B95-C80C-41f7-830F-FBA271C26F7E}'

- val LocalizedString = s '@%MAIN_MODULE%,-110'

- }

- }

-

- NoRemove AppId

- {

- ForceRemove {FE8B3B95-C80C-41f7-830F-FBA271C26F7E} = s 'VeraCrypt class'

- {

- val AccessPermission = b 010004803000000040000000000000001400000002001c000100000000001400070000000101000000000005040000000102000000000005200000002002000001020000000000052000000020020000

- }

-

- ForceRemove VeraCrypt.exe

- {

- val AppId = s '{FE8B3B95-C80C-41f7-830F-FBA271C26F7E}'

- }

- }

-

- ForceRemove VeraCryptFormat.1 = s 'VeraCryptFormat class'

- {

- CLSID = s '{A96D3797-9F31-49f4-A0CE-9657392CF789}'

- }

-

- ForceRemove VeraCryptFormat = s 'VeraCryptFormat class'

- {

- CLSID = s '{A96D3797-9F31-49f4-A0CE-9657392CF789}'

- CurVer = s 'VeraCryptFormat.1'

- }

-

- NoRemove CLSID

- {

- ForceRemove {A96D3797-9F31-49f4-A0CE-9657392CF789} = s 'VeraCryptFormat class'

- {

- ProgID = s 'VeraCryptFormat.1'

- VersionIndependentProgID = s 'VeraCryptFormat'

- LocalServer32 = s '"%FORMAT_MODULE%"'

-

- TypeLib = s '{56327DDA-F1A7-4e13-B128-520D129BDEF6}'

-

- Elevation

- {

- val Enabled = d 1

- val IconReference = s '@%FORMAT_MODULE%,-501'

- }

-

- val AppId = s '{A96D3797-9F31-49f4-A0CE-9657392CF789}'

- val LocalizedString = s '@%FORMAT_MODULE%,-112'

- }

- }

-

- NoRemove AppId

- {

- ForceRemove {A96D3797-9F31-49f4-A0CE-9657392CF789} = s 'VeraCryptFormat class'

- {

- val AccessPermission = b 010004803000000040000000000000001400000002001c000100000000001400070000000101000000000005040000000102000000000005200000002002000001020000000000052000000020020000

- }

-

- ForceRemove 'VeraCrypt Format.exe'

- {

- val AppId = s '{A96D3797-9F31-49f4-A0CE-9657392CF789}'

- }

- }

-/*

- 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-2016 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. */

-

-#include "Tcdefs.h"

-

-#include <sys/types.h>

-#include <sys/stat.h>

-#include <direct.h>

-#include <string.h>

-#include <stdlib.h>

-#include <errno.h>

-#include <Strsafe.h>

-

-#include "Dir.h"

-

-/* create full directory tree. returns 0 for success, -1 if failure */

-int

-mkfulldir (wchar_t *oriPath, BOOL bCheckonly)

-{

- struct _stat st;

- wchar_t *uniq_file;

- wchar_t path [TC_MAX_PATH];

-

- StringCbCopyW (path, TC_MAX_PATH, oriPath);

-

- if (wcslen (path) == 3 && path[1] == L':')

- goto is_root; /* keep final slash in root if present */

-

- /* strip final forward or backslash if we have one! */

- uniq_file = wcsrchr (path, L'\\');

- if (uniq_file && uniq_file[1] == L'\0')

- uniq_file[0] = L'\0';

- else

- {

- uniq_file = wcsrchr (path, L'/');

- if (uniq_file && uniq_file[1] == L'\0')

- uniq_file[0] = L'\0';

- }

-

- is_root:

- if (bCheckonly)

- return _wstat (path, &st);

-

- if (_wstat (path, &st))

- return mkfulldir_internal (path);

- else

- return 0;

-}

-

-

-int

-mkfulldir_internal (wchar_t *path)

-{

- wchar_t *token;

- struct _stat st;

- static wchar_t tokpath[_MAX_PATH];

- static wchar_t trail[_MAX_PATH];

-

- StringCbCopyW (tokpath, _MAX_PATH, path);

- trail[0] = L'\0';

-

- token = wcstok (tokpath, L"\\/");

-

- if (tokpath[0] == L'\\' && tokpath[1] == L'\\')

- { /* unc */

- trail[0] = tokpath[0];

- trail[1] = tokpath[1];

- trail[2] = L'\0';

- if (token)

- {

- StringCbCatW (trail, _MAX_PATH, token);

- StringCbCatW (trail, _MAX_PATH, L"\\");

- token = wcstok (NULL, L"\\/");

- if (token)

- { /* get share name */

- StringCbCatW (trail, _MAX_PATH, token);

- StringCbCatW (trail, _MAX_PATH, L"\\");

- }

- token = wcstok (NULL, L"\\/");

- }

- }

-

- if (tokpath[1] == L':')

- { /* drive letter */

- StringCbCatW (trail, _MAX_PATH, tokpath);

- StringCbCatW (trail, _MAX_PATH, L"\\");

- token = wcstok (NULL, L"\\/");

- }

-

- while (token != NULL)

- {

- int x;

- StringCbCatW (trail, _MAX_PATH, token);

- x = _wmkdir (trail);

- StringCbCatW (trail, _MAX_PATH, L"\\");

- token = wcstok (NULL, L"\\/");

- }

-

- return _wstat (path, &st);

-}

-/*

- 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-2016 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. */

-

-#ifdef __cplusplus

-extern "C" {

-#endif

-

-int mkfulldir ( wchar_t *path , BOOL bCheckonly );

-int mkfulldir_internal ( wchar_t *path );

-

-#ifdef __cplusplus

-}

-#endif

-# Modifications and additions to the original source code (contained in this file)

-

-

-

-VeraCrypt is multi-licensed under Apache License 2.0 and

-the TrueCrypt License version 3.0, a verbatim copy of both

-This license does not grant you rights to use any

-contributors' name, logo, or trademarks, including IDRIX,

-For example, the following names are not allowed: VeraCrypt,

-VeraCrypt+, VeraCrypt Professional, iVeraCrypt, etc. Nor any

-other names confusingly similar to the name VeraCrypt (e.g.,

-modifications, including but not limited to software source

-purposes of this definition, "submitted" means any form

-3. Grant of Patent License. Subject to the terms and conditions

-irrevocable(except as stated in this section) patent license

-to make, have made, use, offer to sell, sell, import, and

-otherwise transfer the Work, where such license applies only

-necessarily infringed by their Contribution(s) alone or by

-litigation against any entity (including a cross-claim or

-counterclaim in a lawsuit) alleging that the Work or a

-and conditions for use, reproduction, or distribution of

-conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY,

-out of the use or inability to use the Work (including but not

-warranty, indemnity, or other liability obligations and/or

- VeraCrypt is a free disk encryption software brought to you by IDRIX

-

-

-

-

-

-

-

-

-Please download the latest OSXFuse version from :

-//{{NO_DEPENDENCIES}}

-// Microsoft Visual C++ generated include file.

-// Used by Setup.rc

-//

-#define IDR_COMREG 10

-#define IDD_INSTALL 101

-#define IDD_INSTALL_OPTIONS_PAGE_DLG 102

-#define IDD_UNINSTALL 103

-#define IDI_SETUP 104

-#define IDR_SETUP_RSRC_HEADER 105

-#define IDD_EXTRACTION_OPTIONS_PAGE_DLG 106

-#define IDB_SETUP_WIZARD 107

-#define IDD_INTRO_PAGE_DLG 108

-#define IDB_SETUP_WIZARD_BKG 109

-#define IDD_INFO_PAGE_DLG 110

-#define IDD_INSTL_DLG 111

-#define IDD_WIZARD_MODE_PAGE_DLG 112

-#define IDD_PROGRESS_PAGE_DLG 113

-#define IDD_DONATIONS_PAGE_DLG 114

-#define IDC_DESTINATION 1000

-#define IDC_BOX_TITLE 1001

-#define IDC_BROWSE 1002

-#define IDC_BOX_INFO 1003

-#define IDC_LICENSE 1004

-#define IDC_BOX_HELP 1005

-#define IDC_LICENSE_TEXT 1006

-#define IDC_BOX_HELP2 1007

-#define IDC_FILE_TYPE 1008

-#define IDT_UNINSTALL_DIR 1009

-#define IDC_PROG_GROUP 1010

-#define IDC_SYSTEM_RESTORE 1011

-#define IDC_DESKTOP_ICON 1012

-#define IDC_ALL_USERS 1013

-#define IDT_INSTALL_DESTINATION 1014

-#define IDC_UNINSTALL 1015

-#define IDC_PROGRESS_BAR 1016

-#define IDC_LOG_WINDOW 1017

-#define IDC_SETUP_WIZARD_BKG 1018

-#define IDC_SETUP_WIZARD_GFX_AREA 1019

-#define IDC_HR 1020

-#define IDC_OPEN_CONTAINING_FOLDER 1021

-#define IDC_AGREE 1022

-#define IDC_HR_BOTTOM 1023

-#define IDC_WIZARD_MODE_INSTALL 1024

-#define IDC_WIZARD_MODE_EXTRACT_ONLY 1025

-#define IDC_NEXT 1026

-#define IDC_PREV 1027

-#define IDT_EXTRACT_DESTINATION 1028

-#define IDC_POS_BOX 1029

-#define IDC_BITMAP_SETUP_WIZARD 1030

-#define IDC_MAIN_CONTENT_CANVAS 1031

-#define IDC_DONATE 1032

-

-// Next default values for new objects

-//

-#ifdef APSTUDIO_INVOKED

-#ifndef APSTUDIO_READONLY_SYMBOLS

-#define _APS_NO_MFC 1

-#define _APS_NEXT_RESOURCE_VALUE 115

-#define _APS_NEXT_COMMAND_VALUE 40001

-#define _APS_NEXT_CONTROL_VALUE 1033

-#define _APS_NEXT_SYMED_VALUE 101

-#endif

-#endif

-/*

- Derived from source code of TrueCrypt 7.1a, which is

- Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed

- by the TrueCrypt License 3.0.

-

- Modifications and additions to the original source code (contained in this file)

- and all other portions of this file are Copyright (c) 2013-2016 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.

-*/

-

-#include "Tcdefs.h"

-

-#include "Inflate.h"

-#include "SelfExtract.h"

-#include "Wizard.h"

-#include "Setup.h"

-#include "Crc.h"

-#include "Endian.h"

-#include "Dlgcode.h"

-#include "Dir.h"

-#include "Language.h"

-#include "Resource.h"

-#include <tchar.h>

-#include <Strsafe.h>

-

-#ifndef SRC_POS

-#define SRC_POS (__FUNCTION__ ":" TC_TO_STRING(__LINE__))

-#endif

-

-#define OutputPackageFile L"VeraCrypt Setup " _T(VERSION_STRING) L".exe"

-

-#define MAG_START_MARKER "TCINSTRT"

-#define MAG_END_MARKER_OBFUSCATED "T/C/I/N/S/C/R/C"

-#define PIPE_BUFFER_LEN (4 * BYTES_PER_KB)

-

-unsigned char MagEndMarker [sizeof (MAG_END_MARKER_OBFUSCATED)];

-wchar_t DestExtractPath [TC_MAX_PATH];

-DECOMPRESSED_FILE Decompressed_Files [NBR_COMPRESSED_FILES];

-

-volatile char *PipeWriteBuf = NULL;

-volatile HANDLE hChildStdinWrite = INVALID_HANDLE_VALUE;

-unsigned char *DecompressedData = NULL;

-

-

-

-void SelfExtractStartupInit (void)

-{

- DeobfuscateMagEndMarker ();

-}

-

-

-// The end marker must be included in the self-extracting exe only once, not twice (used e.g.

-// by IsSelfExtractingPackage()) and that's why MAG_END_MARKER_OBFUSCATED is obfuscated and

-// needs to be deobfuscated using this function at startup.

-static void DeobfuscateMagEndMarker (void)

-{

- int i;

-

- for (i = 0; i < sizeof (MAG_END_MARKER_OBFUSCATED); i += 2)

- MagEndMarker [i/2] = MAG_END_MARKER_OBFUSCATED [i];

-

- MagEndMarker [i/2] = 0;

-}

-

-

-static void PkgError (wchar_t *msg)

-{

- MessageBox (NULL, msg, L"VeraCrypt", MB_ICONERROR | MB_SETFOREGROUND | MB_TOPMOST);

-}

-

-

-static void PkgWarning (wchar_t *msg)

-{

- MessageBox (NULL, msg, L"VeraCrypt", MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);

-}

-

-

-static void PkgInfo (wchar_t *msg)

-{

- MessageBox (NULL, msg, L"VeraCrypt", MB_ICONINFORMATION | MB_SETFOREGROUND | MB_TOPMOST);

-}

-

-

-// Returns 0 if decompression fails or, if successful, returns the size of the decompressed data

-static int DecompressBuffer (char *out, char *in, int len)

-{

- return (DecompressDeflatedData (out, in, len)); // Inflate

-}

-

-

-static void __cdecl PipeWriteThread (void *len)

-{

- int sendBufSize = PIPE_BUFFER_LEN, bytesSent = 0;

- int bytesToSend = *((int *) len), bytesSentTotal = 0;

-

- if (PipeWriteBuf == NULL || (HANDLE) hChildStdinWrite == INVALID_HANDLE_VALUE)

- {

- PkgError (L"Failed sending data to the STDIN pipe");

- return;

- }

-

- while (bytesToSend > 0)

- {

- if (bytesToSend < PIPE_BUFFER_LEN)

- sendBufSize = bytesToSend;

-

- if (!WriteFile ((HANDLE) hChildStdinWrite, (char *) PipeWriteBuf + bytesSentTotal, sendBufSize, &bytesSent, NULL)

- || bytesSent == 0

- || bytesSent != sendBufSize)

- {

- PkgError (L"Failed sending data to the STDIN pipe");

- return;

- }

-

- bytesToSend -= bytesSent;

- bytesSentTotal += bytesSent;

- }

-

- // Closing the pipe causes the child process to stop reading from it

-

- if (!CloseHandle (hChildStdinWrite))

- {

- PkgError (L"Cannot close pipe");

- return;

- }

-}

-

-

-// Returns 0 if compression fails or, if successful, the size of the compressed data

-static int CompressBuffer (char *out, char *in, int len)

-{

- SECURITY_ATTRIBUTES securityAttrib;

- DWORD bytesReceived = 0;

- HANDLE hChildStdoutWrite = INVALID_HANDLE_VALUE;

- HANDLE hChildStdoutRead = INVALID_HANDLE_VALUE;

- HANDLE hChildStdinRead = INVALID_HANDLE_VALUE;

- STARTUPINFO startupInfo;

- PROCESS_INFORMATION procInfo;

- char pipeBuffer [PIPE_BUFFER_LEN];

- int res_len = 0;

- BOOL bGzipHeaderRead = FALSE;

- wchar_t szGzipCmd[64];

-

- ZeroMemory (&startupInfo, sizeof (startupInfo));

- ZeroMemory (&procInfo, sizeof (procInfo));

-

- // Pipe handle inheritance

- securityAttrib.bInheritHandle = TRUE;

- securityAttrib.nLength = sizeof (securityAttrib);

- securityAttrib.lpSecurityDescriptor = NULL;

-

- if (!CreatePipe (&hChildStdoutRead, &hChildStdoutWrite, &securityAttrib, 0))

- {

- PkgError (L"Cannot create STDOUT pipe.");

- return 0;

- }

- SetHandleInformation (hChildStdoutRead, HANDLE_FLAG_INHERIT, 0);

-

- if (!CreatePipe (&hChildStdinRead, &((HANDLE) hChildStdinWrite), &securityAttrib, 0))

- {

- PkgError (L"Cannot create STDIN pipe.");

- CloseHandle(hChildStdoutWrite);

- CloseHandle(hChildStdoutRead);

- return 0;

- }

- SetHandleInformation (hChildStdinWrite, HANDLE_FLAG_INHERIT, 0);

-

- // Create a child process that will compress the data

-

- startupInfo.wShowWindow = SW_HIDE;

- startupInfo.hStdInput = hChildStdinRead;

- startupInfo.hStdOutput = hChildStdoutWrite;

- startupInfo.cb = sizeof (startupInfo);

- startupInfo.hStdError = hChildStdoutWrite;

- startupInfo.dwFlags |= STARTF_USESTDHANDLES | STARTF_USESHOWWINDOW;

-

- StringCchCopyW (szGzipCmd, ARRAYSIZE (szGzipCmd), L"gzip --best");

- if (!CreateProcess (NULL, szGzipCmd, NULL, NULL, TRUE, 0, NULL, NULL, &startupInfo, &procInfo))

- {

- PkgError (L"Error: Cannot run gzip.\n\nBefore you can create a self-extracting VeraCrypt package, you need to have the open-source 'gzip' compression tool placed in any directory in the search path for executable files (for example, in 'C:\\Windows\\').\n\nNote: gzip can be freely downloaded e.g. from www.gzip.org");

- CloseHandle(hChildStdoutWrite);

- CloseHandle(hChildStdoutRead);

- CloseHandle(hChildStdinRead);

- CloseHandle(hChildStdinWrite);

- return 0;

- }

-

- CloseHandle (procInfo.hProcess);

- CloseHandle (procInfo.hThread);

-

- // Start sending the uncompressed data to the pipe (STDIN)

- PipeWriteBuf = in;

- _beginthread (PipeWriteThread, PIPE_BUFFER_LEN * 2, (void *) &len);

-

- if (!CloseHandle (hChildStdoutWrite))

- {

- PkgError (L"Cannot close STDOUT write");

- CloseHandle(hChildStdoutRead);

- CloseHandle(hChildStdinRead);

- return 0;

- }

-

- bGzipHeaderRead = FALSE;

-

- // Read the compressed data from the pipe (sent by the child process to STDOUT)

- while (TRUE)

- {

- if (!ReadFile (hChildStdoutRead, pipeBuffer, bGzipHeaderRead ? PIPE_BUFFER_LEN : 10, &bytesReceived, NULL))

- break;

-

- if (bGzipHeaderRead)

- {

- memcpy (out + res_len, pipeBuffer, bytesReceived);

- res_len += bytesReceived;

- }

- else

- bGzipHeaderRead = TRUE; // Skip the 10-byte gzip header

- }

-

- CloseHandle(hChildStdoutRead);

- CloseHandle(hChildStdinRead);

- return res_len - 8; // A gzip stream ends with a CRC-32 hash and a 32-bit size (those 8 bytes need to be chopped off)

-}

-

-

-// Clears all bytes that change when an exe file is digitally signed, except the data that are appended.

-// If those bytes weren't cleared, CRC-32 checks would fail after signing.

-static void WipeSignatureAreas (char *buffer)

-{

- // Clear bytes 0x130-0x1ff

- memset (buffer + 0x130, 0, 0x200 - 0x130);

-}

-

-

-BOOL MakeSelfExtractingPackage (HWND hwndDlg, wchar_t *szDestDir)

-{

- int i, x;

- wchar_t inputFile [TC_MAX_PATH];

- wchar_t outputFile [TC_MAX_PATH];

- wchar_t szTmpFilePath [TC_MAX_PATH];

- unsigned char szTmp32bit [4] = {0};

- unsigned char *szTmp32bitPtr = szTmp32bit;

- unsigned char *buffer = NULL, *compressedBuffer = NULL;

- unsigned char *bufIndex = NULL;

- wchar_t tmpStr [2048];

- int bufLen = 0, compressedDataLen = 0, uncompressedDataLen = 0;

-

- x = wcslen (szDestDir);

- if (x < 2)

- goto err;

-

- if (szDestDir[x - 1] != L'\\')

- StringCbCatW (szDestDir, MAX_PATH, L"\\");

-

- GetModuleFileName (NULL, inputFile, ARRAYSIZE (inputFile));

-

- StringCchCopyW (outputFile, ARRAYSIZE(outputFile), szDestDir);

- StringCchCatW (outputFile, ARRAYSIZE(outputFile), OutputPackageFile);

-

- // Clone 'VeraCrypt Setup.exe' to create the base of the new self-extracting archive

-

- if (!TCCopyFile (inputFile, outputFile))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- PkgError (L"Cannot copy 'VeraCrypt Setup.exe' to the package");

- goto err;

- }

-

- // Determine the buffer size needed for all the files and meta data and check if all required files exist

-

- bufLen = 0;

-

- for (i = 0; i < sizeof (szCompressedFiles) / sizeof (szCompressedFiles[0]); i++)

- {

- StringCbPrintfW (szTmpFilePath, sizeof(szTmpFilePath), L"%s%s", szDestDir, szCompressedFiles[i]);

-

- if (!FileExists (szTmpFilePath))

- {

- wchar_t tmpstr [1000];

-

- StringCbPrintfW (tmpstr, sizeof(tmpstr), L"File not found:\n\n'%s'", szTmpFilePath);

- if (_wremove (outputFile))

- StringCbCatW (tmpstr, sizeof(tmpstr), L"\nFailed also to delete package file");

- PkgError (tmpstr);

- goto err;

- }

-

- bufLen += (int) GetFileSize64 (szTmpFilePath);

-

- bufLen += 2; // 16-bit filename length

- bufLen += (wcslen(szCompressedFiles[i]) * sizeof (wchar_t)); // Filename

- bufLen += 4; // CRC-32

- bufLen += 4; // 32-bit file length

- }

-

- buffer = malloc (bufLen + 524288); // + 512K reserve

- if (buffer == NULL)

- {

- PkgError (L"Cannot allocate memory for uncompressed data");

- if (_wremove (outputFile))

- PkgError (L"Cannot allocate memory for uncompressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot allocate memory for uncompressed data");

- goto err;

- }

-

-

- // Write the start marker

- if (!SaveBufferToFile (MAG_START_MARKER, outputFile, strlen (MAG_START_MARKER), TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write the start marker\nFailed also to delete package file");

- else

- PkgError (L"Cannot write the start marker");

- goto err;

- }

-

-

- bufIndex = buffer;

-

- // Copy all required files and their meta data to the buffer

- for (i = 0; i < sizeof (szCompressedFiles) / sizeof (szCompressedFiles[0]); i++)

- {

- DWORD tmpFileSize;

- unsigned char *tmpBuffer;

-

- StringCbPrintfW (szTmpFilePath, sizeof(szTmpFilePath), L"%s%s", szDestDir, szCompressedFiles[i]);

-

- tmpBuffer = LoadFile (szTmpFilePath, &tmpFileSize);

-

- if (tmpBuffer == NULL)

- {

- wchar_t tmpstr [1000];

-

- StringCbPrintfW (tmpstr, sizeof(tmpstr), L"Cannot load file \n'%s'", szTmpFilePath);

- if (_wremove (outputFile))

- StringCbCatW (tmpstr, sizeof(tmpstr), L"\nFailed also to delete package file");

- PkgError (tmpstr);

- goto err;

- }

-

- // Copy the filename length to the main buffer

- mputWord (bufIndex, (WORD) wcslen(szCompressedFiles[i]));

-

- // Copy the filename to the main buffer

- wmemcpy ((wchar_t*)bufIndex, szCompressedFiles[i], wcslen(szCompressedFiles[i]));

- bufIndex += (wcslen(szCompressedFiles[i]) * sizeof (wchar_t));

-

- // Compute CRC-32 hash of the uncompressed file and copy it to the main buffer

- mputLong (bufIndex, GetCrc32 (tmpBuffer, tmpFileSize));

-

- // Copy the file length to the main buffer

- mputLong (bufIndex, (unsigned __int32) tmpFileSize);

-

- // Copy the file contents to the main buffer

- memcpy (bufIndex, tmpBuffer, tmpFileSize);

- bufIndex += tmpFileSize;

-

- free (tmpBuffer);

- }

-

- // Calculate the total size of the uncompressed data

- uncompressedDataLen = (int) (bufIndex - buffer);

-

- // Write total size of the uncompressed data

- szTmp32bitPtr = szTmp32bit;

- mputLong (szTmp32bitPtr, (unsigned __int32) uncompressedDataLen);

- if (!SaveBufferToFile (szTmp32bit, outputFile, sizeof (szTmp32bit), TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write the total size of the uncompressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot write the total size of the uncompressed data");

- goto err;

- }

-

- // Compress all the files and meta data in the buffer to create a solid archive

-

- // Test to make Coverity happy. It will always be false

- if (uncompressedDataLen >= (INT_MAX - 524288))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot allocate memory for compressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot allocate memory for compressed data");

- goto err;

- }

-

- compressedBuffer = malloc (uncompressedDataLen + 524288); // + 512K reserve

- if (compressedBuffer == NULL)

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot allocate memory for compressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot allocate memory for compressed data");

- goto err;

- }

-

- compressedDataLen = CompressBuffer (compressedBuffer, buffer, uncompressedDataLen);

- if (compressedDataLen <= 0)

- {

- if (_wremove (outputFile))

- PkgError (L"Failed to compress the data.\nFailed also to delete package file");

- else

- PkgError (L"Failed to compress the data");

- goto err;

- }

-

- free (buffer);

- buffer = NULL;

-

- // Write the total size of the compressed data

- szTmp32bitPtr = szTmp32bit;

- mputLong (szTmp32bitPtr, (unsigned __int32) compressedDataLen);

- if (!SaveBufferToFile (szTmp32bit, outputFile, sizeof (szTmp32bit), TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write the total size of the compressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot write the total size of the compressed data");

- goto err;

- }

-

- // Write the compressed data

- if (!SaveBufferToFile (compressedBuffer, outputFile, compressedDataLen, TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write compressed data to the package.\nFailed also to delete package file");

- else

- PkgError (L"Cannot write compressed data to the package");

- goto err;

- }

-

- // Write the end marker

- if (!SaveBufferToFile (MagEndMarker, outputFile, strlen (MagEndMarker), TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write the end marker.\nFailed also to delete package file");

- else

- PkgError (L"Cannot write the end marker");

- goto err;

- }

-

- free (compressedBuffer);

- compressedBuffer = NULL;

-

- // Compute and write CRC-32 hash of the entire package

- {

- DWORD tmpFileSize;

- char *tmpBuffer;

-

- tmpBuffer = LoadFile (outputFile, &tmpFileSize);

-

- if (tmpBuffer == NULL)

- {

- handleWin32Error (hwndDlg, SRC_POS);

- if (_wremove (outputFile))

- PkgError (L"Cannot load the package to compute CRC.\nFailed also to delete package file");

- else

- PkgError (L"Cannot load the package to compute CRC");

- goto err;

- }

-

- // Zero all bytes that change when the exe is digitally signed (except appended blocks).

- WipeSignatureAreas (tmpBuffer);

-

- szTmp32bitPtr = szTmp32bit;

- mputLong (szTmp32bitPtr, GetCrc32 (tmpBuffer, tmpFileSize));

- free (tmpBuffer);

-

- if (!SaveBufferToFile (szTmp32bit, outputFile, sizeof (szTmp32bit), TRUE, FALSE))

- {

- if (_wremove (outputFile))

- PkgError (L"Cannot write the total size of the compressed data.\nFailed also to delete package file");

- else

- PkgError (L"Cannot write the total size of the compressed data");

- goto err;

- }

- }

-

- StringCbPrintfW (tmpStr, sizeof(tmpStr), L"Self-extracting package successfully created (%s)", outputFile);

- PkgInfo (tmpStr);

- return TRUE;

-

-err:

- if (buffer)

- free (buffer);

- if (compressedBuffer)

- free (compressedBuffer);

-

- return FALSE;

-}

-

-

-// Verifies the CRC-32 of the whole self-extracting package (except the digital signature areas, if present)

-BOOL VerifyPackageIntegrity (void)

-{

- int fileDataEndPos = 0;

- int fileDataStartPos = 0;

- unsigned __int32 crc = 0;

- unsigned char *tmpBuffer;

- int tmpFileSize;

- wchar_t path [TC_MAX_PATH];

-

- GetModuleFileName (NULL, path, ARRAYSIZE (path));

-

- fileDataEndPos = (int) FindStringInFile (path, MagEndMarker, strlen (MagEndMarker));

- if (fileDataEndPos < 0)

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- return FALSE;

- }

- fileDataEndPos--;

-

- fileDataStartPos = (int) FindStringInFile (path, MAG_START_MARKER, strlen (MAG_START_MARKER));

- if (fileDataStartPos < 0)

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- return FALSE;

- }

- fileDataStartPos += strlen (MAG_START_MARKER);

-

-

- if (!LoadInt32 (path, &crc, fileDataEndPos + strlen (MagEndMarker) + 1))

- {

- Error ("CANT_VERIFY_PACKAGE_INTEGRITY", NULL);

- return FALSE;

- }

-

- // Compute the CRC-32 hash of the whole file (except the digital signature area, if present)

- tmpBuffer = LoadFile (path, &tmpFileSize);

-

- if (tmpBuffer == NULL)

- {

- Error ("CANT_VERIFY_PACKAGE_INTEGRITY", NULL);

- return FALSE;

- }

-

- // Zero all bytes that change when an exe is digitally signed (except appended blocks).

- WipeSignatureAreas (tmpBuffer);

-

- if (crc != GetCrc32 (tmpBuffer, fileDataEndPos + 1 + strlen (MagEndMarker)))

- {

- free (tmpBuffer);

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- return FALSE;

- }

-

- free (tmpBuffer);

-

- return TRUE;

-}

-

-

-// Determines whether we are a self-extracting package

-BOOL IsSelfExtractingPackage (void)

-{

- wchar_t path [TC_MAX_PATH];

-

- GetModuleFileName (NULL, path, ARRAYSIZE (path));

-

- return (FindStringInFile (path, MagEndMarker, strlen (MagEndMarker)) != -1);

-}

-

-

-static void FreeAllFileBuffers (void)

-{

- int fileNo;

-

- if (DecompressedData != NULL)

- {

- free (DecompressedData);

- DecompressedData = NULL;

- }

-

- for (fileNo = 0; fileNo < NBR_COMPRESSED_FILES; fileNo++)

- {

- Decompressed_Files[fileNo].fileName = NULL;

- Decompressed_Files[fileNo].fileContent = NULL;

- Decompressed_Files[fileNo].fileNameLength = 0;

- Decompressed_Files[fileNo].fileLength = 0;

- Decompressed_Files[fileNo].crc = 0;

- }

-}

-

-

-// Assumes that VerifyPackageIntegrity() has been used. Returns TRUE, if successful (otherwise FALSE).

-// Creates a table of pointers to buffers containing the following objects for each file:

-// filename size, filename (not null-terminated!), file size, file CRC-32, uncompressed file contents.

-// For details, see the definition of the DECOMPRESSED_FILE structure.

-BOOL SelfExtractInMemory (wchar_t *path)

-{

- int filePos = 0, fileNo = 0;

- int fileDataEndPos = 0;

- int fileDataStartPos = 0;

- int uncompressedLen = 0;

- int compressedLen = 0;

- unsigned char *compressedData = NULL;

- unsigned char *bufPos = NULL, *bufEndPos = NULL;

-

- FreeAllFileBuffers();

-

- fileDataEndPos = (int) FindStringInFile (path, MagEndMarker, strlen (MagEndMarker));

- if (fileDataEndPos < 0)

- {

- Error ("CANNOT_READ_FROM_PACKAGE", NULL);

- return FALSE;

- }

-

- fileDataEndPos--;

-

- fileDataStartPos = (int) FindStringInFile (path, MAG_START_MARKER, strlen (MAG_START_MARKER));

- if (fileDataStartPos < 0)

- {

- Error ("CANNOT_READ_FROM_PACKAGE", NULL);

- return FALSE;

- }

-

- fileDataStartPos += strlen (MAG_START_MARKER);

-

- filePos = fileDataStartPos;

-

- // Read the stored total size of the uncompressed data

- if (!LoadInt32 (path, &uncompressedLen, filePos))

- {

- Error ("CANNOT_READ_FROM_PACKAGE", NULL);

- return FALSE;

- }

-

- filePos += 4;

-

- // Read the stored total size of the compressed data

- if (!LoadInt32 (path, &compressedLen, filePos))

- {

- Error ("CANNOT_READ_FROM_PACKAGE", NULL);

- return FALSE;

- }

-

- filePos += 4;

-

- if (compressedLen != fileDataEndPos - fileDataStartPos - 8 + 1)

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- }

-

- DecompressedData = malloc (uncompressedLen + 524288); // + 512K reserve

- if (DecompressedData == NULL)

- {

- Error ("ERR_MEM_ALLOC", NULL);

- return FALSE;

- }

-

- bufPos = DecompressedData;

- bufEndPos = bufPos + uncompressedLen - 1;

-

- compressedData = LoadFileBlock (path, filePos, compressedLen);

-

- if (compressedData == NULL)

- {

- free (DecompressedData);

- DecompressedData = NULL;

-

- Error ("CANNOT_READ_FROM_PACKAGE", NULL);

- return FALSE;

- }

-

- // Decompress the data

- if (DecompressBuffer (DecompressedData, compressedData, compressedLen) != uncompressedLen)

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- goto sem_end;

- }

-

- while (bufPos <= bufEndPos && fileNo < NBR_COMPRESSED_FILES)

- {

- // Filename length

- Decompressed_Files[fileNo].fileNameLength = mgetWord (bufPos);

-

- // Filename

- Decompressed_Files[fileNo].fileName = (wchar_t*) bufPos;

- bufPos += (Decompressed_Files[fileNo].fileNameLength * sizeof (wchar_t));

-

- // CRC-32 of the file

- Decompressed_Files[fileNo].crc = mgetLong (bufPos);

-

- // File length

- Decompressed_Files[fileNo].fileLength = mgetLong (bufPos);

-

- // File content

- Decompressed_Files[fileNo].fileContent = bufPos;

- bufPos += Decompressed_Files[fileNo].fileLength;

-

- // Verify CRC-32 of the file (to verify that it didn't get corrupted while creating the solid archive).

- if (Decompressed_Files[fileNo].crc

- != GetCrc32 (Decompressed_Files[fileNo].fileContent, Decompressed_Files[fileNo].fileLength))

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- goto sem_end;

- }

-

- fileNo++;

- }

-

- if (fileNo < NBR_COMPRESSED_FILES)

- {

- Error ("DIST_PACKAGE_CORRUPTED", NULL);

- goto sem_end;

- }

-

- free (compressedData);

- return TRUE;

-

-sem_end:

- FreeAllFileBuffers();

- free (compressedData);

- return FALSE;

-}

-

-

-void __cdecl ExtractAllFilesThread (void *hwndDlg)

-{

- int fileNo;

- BOOL bSuccess = FALSE;

- wchar_t packageFile [TC_MAX_PATH];

-

- InvalidateRect (GetDlgItem (GetParent (hwndDlg), IDD_INSTL_DLG), NULL, TRUE);

-

- ClearLogWindow (hwndDlg);

-

- GetModuleFileName (NULL, packageFile, ARRAYSIZE (packageFile));

-

- if (!(bSuccess = SelfExtractInMemory (packageFile)))

- goto eaf_end;

-

- if (mkfulldir (DestExtractPath, TRUE) != 0)

- {

- if (mkfulldir (DestExtractPath, FALSE) != 0)

- {

- wchar_t szTmp[TC_MAX_PATH];

-

- handleWin32Error (hwndDlg, SRC_POS);

- StringCbPrintfW (szTmp, sizeof(szTmp), GetString ("CANT_CREATE_FOLDER"), DestExtractPath);

- MessageBoxW (hwndDlg, szTmp, lpszTitle, MB_ICONHAND);

- bSuccess = FALSE;

- goto eaf_end;

- }

- }

-

- for (fileNo = 0; fileNo < NBR_COMPRESSED_FILES; fileNo++)

- {

- wchar_t fileName [TC_MAX_PATH] = {0};

- wchar_t filePath [TC_MAX_PATH] = {0};

-

- // Filename

- StringCchCopyNW (fileName, ARRAYSIZE(fileName), Decompressed_Files[fileNo].fileName, Decompressed_Files[fileNo].fileNameLength);

- StringCchCopyW (filePath, ARRAYSIZE(filePath), DestExtractPath);

- StringCchCatW (filePath, ARRAYSIZE(filePath), fileName);

-

- StatusMessageParam (hwndDlg, "EXTRACTING_VERB", filePath);

-

- // Write the file

- if (!SaveBufferToFile (

- Decompressed_Files[fileNo].fileContent,

- filePath,

- Decompressed_Files[fileNo].fileLength,

- FALSE, FALSE))

- {

- wchar_t szTmp[512];

-

- StringCbPrintfW (szTmp, sizeof (szTmp), GetString ("CANNOT_WRITE_FILE_X"), filePath);

- MessageBoxW (hwndDlg, szTmp, lpszTitle, MB_ICONERROR | MB_SETFOREGROUND | MB_TOPMOST);

- bSuccess = FALSE;

- goto eaf_end;

- }

- UpdateProgressBarProc ((int) (100 * ((float) fileNo / NBR_COMPRESSED_FILES)));

- }

-

-eaf_end:

- FreeAllFileBuffers();

-

- if (bSuccess)

- PostMessage (MainDlg, TC_APPMSG_EXTRACTION_SUCCESS, 0, 0);

- else

- PostMessage (MainDlg, TC_APPMSG_EXTRACTION_FAILURE, 0, 0);

-}

-

-/*

- Derived from source code of TrueCrypt 7.1a, which is

- Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed

- by the TrueCrypt License 3.0.

-

- Modifications and additions to the original source code (contained in this file)

- and all other portions of this file are Copyright (c) 2013-2016 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.

-*/

-

-#include "Setup.h"

-

-#ifdef __cplusplus

-extern "C" {

-#endif

-

-typedef struct

-{

- // WARNING: file name is NOT null-terminated (use fileNameLength).

- wchar_t *fileName;

- int fileNameLength;

- uint32 crc;

- __int32 fileLength;

- unsigned char *fileContent;

-} DECOMPRESSED_FILE;

-

-extern DECOMPRESSED_FILE Decompressed_Files [NBR_COMPRESSED_FILES];

-

-void SelfExtractStartupInit (void);

-BOOL SelfExtractInMemory (wchar_t *path);

-void __cdecl ExtractAllFilesThread (void *hwndDlg);

-BOOL MakeSelfExtractingPackage (HWND hwndDlg, wchar_t *szDestDir);

-BOOL VerifyPackageIntegrity (void);

-BOOL IsSelfExtractingPackage (void);

-static void DeobfuscateMagEndMarker (void);

-

-extern wchar_t DestExtractPath [TC_MAX_PATH];

-

-#ifdef __cplusplus

-}

-#endif

-/*

- 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-2016 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. */

-

-#include "Tcdefs.h"

-#include <SrRestorePtApi.h>

-#include <io.h>

-#include <propkey.h>

-#include <propvarutil.h>

-#include <sys/types.h>

-#include <sys/stat.h>

-#include <tchar.h>

-

-#include "Apidrvr.h"

-#include "BootEncryption.h"

-#include "Boot/Windows/BootCommon.h"

-#include "Combo.h"

-#include "ComSetup.h"

-#include "Dlgcode.h"

-#include "Language.h"

-#include "Registry.h"

-#include "Resource.h"

-

-#include "Dir.h"

-#include "Setup.h"

-#include "SelfExtract.h"

-#include "Wizard.h"

-

-#include "../Common/Resource.h"

-

-using namespace VeraCrypt;

-

-#pragma warning( disable : 4201 )

-#pragma warning( disable : 4115 )

-

-#include <shlobj.h>

-

-#pragma warning( default : 4201 )

-#pragma warning( default : 4115 )

-

-#include <Strsafe.h>

-

-wchar_t InstallationPath[TC_MAX_PATH];

-wchar_t SetupFilesDir[TC_MAX_PATH];

-wchar_t UninstallBatch[MAX_PATH];

-

-BOOL bUninstall = FALSE;

-BOOL bRestartRequired = FALSE;

-BOOL bMakePackage = FALSE;

-BOOL bDone = FALSE;

-BOOL Rollback = FALSE;

-BOOL bUpgrade = FALSE;

-BOOL bDowngrade = FALSE;

-BOOL SystemEncryptionUpdate = FALSE;

-BOOL PortableMode = FALSE;

-BOOL bRepairMode = FALSE;

-BOOL bReinstallMode = FALSE;

-BOOL bChangeMode = FALSE;

-BOOL bDevm = FALSE;

-BOOL bPossiblyFirstTimeInstall = FALSE;

-BOOL bUninstallInProgress = FALSE;

-BOOL UnloadDriver = TRUE;

-

-BOOL bSystemRestore = TRUE;

-BOOL bDisableSwapFiles = FALSE;

-BOOL bForAllUsers = TRUE;

-BOOL bRegisterFileExt = TRUE;

-BOOL bAddToStartMenu = TRUE;

-BOOL bDesktopIcon = TRUE;

-

-BOOL bDesktopIconStatusDetermined = FALSE;

-

-HMODULE volatile SystemRestoreDll = 0;

-

-void localcleanup (void)

-{

- localcleanupwiz ();

- cleanup ();

-

- CloseAppSetupMutex ();

-}

-

-BOOL ForceDeleteFile (LPCWSTR szFileName)

-{

- if (!DeleteFile (szFileName))

- {

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

- return MoveFileEx (szFileName, NULL, MOVEFILE_DELAY_UNTIL_REBOOT);

- }

- else

- return TRUE;

-}

-

-BOOL StatDeleteFile (wchar_t *lpszFile, BOOL bCheckForOldFile)

-{

- struct __stat64 st;

-

- if (bCheckForOldFile)

- {

- wchar_t szOldPath[MAX_PATH + 1];

- StringCbCopyW (szOldPath, sizeof(szOldPath), lpszFile);

- StringCbCatW (szOldPath, sizeof(szOldPath), VC_FILENAME_RENAMED_SUFFIX);

-

- if (_wstat64 (szOldPath, &st) == 0)

- {

- ForceDeleteFile (szOldPath);

- }

- }

-

- if (_wstat64 (lpszFile, &st) == 0)

- return ForceDeleteFile (lpszFile);

- else

- return TRUE;

-}

-

-BOOL StatRemoveDirectory (wchar_t *lpszDir)

-{

- struct __stat64 st;

-

- if (_wstat64 (lpszDir, &st) == 0)

- {

- BOOL bStatus = RemoveDirectory (lpszDir);

- if (!bStatus)

- {

- /* force removal of the non empty directory */

- wchar_t szOpPath[TC_MAX_PATH + 1] = {0};

- SHFILEOPSTRUCTW op;

-

- StringCchCopyW(szOpPath, ARRAYSIZE(szOpPath)-1, lpszDir);

- ZeroMemory(&op, sizeof(op));

- op.wFunc = FO_DELETE;

- op.pFrom = szOpPath;

- op.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_NOCONFIRMMKDIR;

-

- if ((0 == SHFileOperation(&op)) && (!op.fAnyOperationsAborted))

- bStatus = TRUE;

- }

- return bStatus;

- }

- else

- return TRUE;

-}

-

-

-/* Recursively set the given OWNER security descriptor to the key and its subkeys */

-static void RecursiveSetOwner (HKEY hKey, PSECURITY_DESCRIPTOR pSD)

-{

- LSTATUS status = 0;

- DWORD dwIndex = 0, dwMaxNameLen = 0, dwNameLen = 0, numberSubKeys = 0;

- HKEY hSubKey;

-

- if ( (ERROR_SUCCESS == status) && (ERROR_SUCCESS == RegQueryInfoKey(hKey, NULL, NULL, NULL, &numberSubKeys, &dwMaxNameLen, NULL, NULL, NULL, NULL, NULL, NULL))

- && (numberSubKeys >= 1)

- )

- {

- dwMaxNameLen++;

- wchar_t* szNameValue = new wchar_t[dwMaxNameLen];

- while (true)

- {

- dwNameLen = dwMaxNameLen;

- status = RegEnumKeyExW (hKey, dwIndex++, szNameValue, &dwNameLen, NULL, NULL, NULL, NULL);

- if (status == ERROR_SUCCESS)

- {

- status = RegOpenKeyExW (hKey, szNameValue, 0, WRITE_OWNER | KEY_READ , &hSubKey);

- if (ERROR_SUCCESS == status)

- {

- RecursiveSetOwner (hSubKey, pSD);

- RegCloseKey(hSubKey);

- }

- }

- else

- break;

- }

- delete [] szNameValue;

- }

-

- RegSetKeySecurity (hKey, OWNER_SECURITY_INFORMATION, pSD);

-}

-

-/* Recursively set the given DACL security descriptor to the key and its subkeys */

-static void RecursiveSetDACL (HKEY hKey, const wchar_t* SubKeyName, PSECURITY_DESCRIPTOR pSD)

-{

- HKEY hSubKey;

- DWORD dwIndex = 0, dwMaxNameLen = 0, dwNameLen = 0, numberSubKeys = 0;

- LSTATUS status = RegOpenKeyExW(hKey, SubKeyName, 0, WRITE_DAC | KEY_READ /*| ACCESS_SYSTEM_SECURITY*/, &hSubKey);

- if (status == ERROR_SUCCESS)

- {

- status = RegSetKeySecurity (hSubKey, DACL_SECURITY_INFORMATION, pSD);

- if (status == ERROR_SUCCESS)

- {

- RegCloseKey(hSubKey);

- status = RegOpenKeyExW(hKey, SubKeyName, 0, WRITE_DAC | KEY_READ , &hSubKey);

- }

-

- if ( (ERROR_SUCCESS == status)

- && (ERROR_SUCCESS == RegQueryInfoKeyW(hSubKey, NULL, NULL, NULL, &numberSubKeys, &dwMaxNameLen, NULL, NULL, NULL, NULL, NULL, NULL))

- && (numberSubKeys >= 1)

- )

- {

- dwMaxNameLen++;

- wchar_t* szNameValue = new wchar_t[dwMaxNameLen];

- while (true)

- {

- dwNameLen = dwMaxNameLen;

- status = RegEnumKeyExW (hSubKey, dwIndex++, szNameValue, &dwNameLen, NULL, NULL, NULL, NULL);

- if (status == ERROR_SUCCESS)

- {

- RecursiveSetDACL (hSubKey, szNameValue, pSD);

- }

- else

- break;

- }

- delete [] szNameValue;

- }

- }

-}

-

-/* Correct the key permissions to allow its deletion */

-static void AllowKeyAccess(HKEY Key,const wchar_t* SubKeyName)

-{

- LSTATUS RegResult;

- HKEY SvcKey = NULL;

- DWORD dwLength = 0;

- HANDLE Token = NULL;

- PTOKEN_USER pTokenUser = NULL;

- std::string sNewSD;

-

- RegResult = RegOpenKeyExW(Key, SubKeyName, 0, WRITE_OWNER | KEY_READ, &SvcKey);

- if (RegResult==ERROR_SUCCESS)

- {

- if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &Token))

- {

- if (!GetTokenInformation(Token, TokenUser, pTokenUser, 0, &dwLength))

- {

- if (GetLastError() ==ERROR_INSUFFICIENT_BUFFER)

- {

- pTokenUser = (PTOKEN_USER) HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, dwLength);

- if (pTokenUser)

- {

- if (GetTokenInformation(Token, TokenUser, pTokenUser, dwLength, &dwLength))

- {

- SECURITY_DESCRIPTOR SecDesc;

- if ( InitializeSecurityDescriptor(&SecDesc, SECURITY_DESCRIPTOR_REVISION)

- && SetSecurityDescriptorDacl(&SecDesc, TRUE, NULL, FALSE) // NULL DACL: full access to everyone

- && SetSecurityDescriptorOwner(&SecDesc, pTokenUser->User.Sid, FALSE)

- )

- {

- RecursiveSetOwner(SvcKey, &SecDesc);

- }

- }

-

- }

- }

- }

- }

- RegCloseKey(SvcKey);

- }

-

- if (pTokenUser)

- {

- PSID pSid = pTokenUser->User.Sid;

- DWORD dwAclSize = sizeof(ACL) + sizeof(ACCESS_ALLOWED_ACE) + ::GetLengthSid(pSid) - sizeof(DWORD);

- PACL pDacl = (PACL) new BYTE[dwAclSize];

- if (pDacl)

- {

- if (TRUE == ::InitializeAcl(pDacl, dwAclSize, ACL_REVISION))

- {

- if (TRUE == AddAccessAllowedAceEx(pDacl, ACL_REVISION, CONTAINER_INHERIT_ACE | OBJECT_INHERIT_ACE, WRITE_DAC | KEY_ALL_ACCESS, pSid))

- {

- SECURITY_DESCRIPTOR SecDesc;

- if (TRUE == ::InitializeSecurityDescriptor(&SecDesc, SECURITY_DESCRIPTOR_REVISION))

- {

- if (TRUE == ::SetSecurityDescriptorDacl(&SecDesc, TRUE, pDacl, FALSE))

- {

- RecursiveSetDACL (Key, SubKeyName, &SecDesc);

- }

- }

- }

- }

- delete [] pDacl;

- }

- }

-

- if (pTokenUser)

- HeapFree(GetProcessHeap(), 0, pTokenUser);

- if (Token)

- CloseHandle(Token);

-}

-

-void SearchAndDeleteRegistrySubString (HKEY hKey, const wchar_t *subKey, const wchar_t *str, BOOL bEnumSubKeys, const wchar_t* enumMatchSubStr)

-{

- HKEY hSubKey = 0;

- LSTATUS status = 0;

- DWORD dwIndex = 0, dwType, dwValueNameLen, dwDataLen;

- std::list<std::wstring> subKeysList;

- size_t subStringLength = str? wcslen(str) : 0;

-

- if (bEnumSubKeys)

- {

- DWORD dwMaxNameLen = 0;

- if (ERROR_SUCCESS == RegQueryInfoKey(hKey, NULL, NULL, NULL, NULL, &dwMaxNameLen, NULL, NULL, NULL, NULL, NULL, NULL))

- {

- dwMaxNameLen++;

- wchar_t* szNameValue = new wchar_t[dwMaxNameLen];

- dwIndex = 0;

- while (true)

- {

- dwValueNameLen = dwMaxNameLen;

- status = RegEnumKeyExW (hKey, dwIndex++, szNameValue, &dwValueNameLen, NULL, NULL, NULL, NULL);

- if (status == ERROR_SUCCESS)

- {

- if (enumMatchSubStr && !wcsstr(szNameValue, enumMatchSubStr))

- continue;

- std::wstring entryName = szNameValue;

- entryName += L"\\";

- entryName += subKey;

- entryName += L"\\";

- subKeysList.push_back(entryName);

- }

- else

- break;

- }

- delete [] szNameValue;

- }

- }

- else

- {

- subKeysList.push_back(subKey);

- }

-

- for (std::list<std::wstring>::iterator ItSubKey = subKeysList.begin(); ItSubKey != subKeysList.end(); ItSubKey++)

- {

- // if the string to search for is empty, delete the sub key, otherwise, look for matching value and delete them

- if (subStringLength == 0)

- {

- if (ERROR_ACCESS_DENIED == DeleteRegistryKey (hKey, ItSubKey->c_str()))

- {

- // grant permission to delete

- AllowKeyAccess (hKey, ItSubKey->c_str());

-

- // try again

- DeleteRegistryKey (hKey, ItSubKey->c_str());

- }

- }

- else

- {

- if (RegOpenKeyExW (hKey, ItSubKey->c_str(), 0, KEY_ALL_ACCESS, &hSubKey) == ERROR_SUCCESS)

- {

- DWORD dwMaxNameLen = 0, dwMaxDataLen = 0;

- if (ERROR_SUCCESS == RegQueryInfoKey(hSubKey, NULL, NULL, NULL, NULL, NULL, NULL, NULL, &dwMaxNameLen, &dwMaxDataLen, NULL, NULL))

- {

- dwMaxNameLen++;

- wchar_t* szNameValue = new wchar_t[dwMaxNameLen];

- LPBYTE pbData = new BYTE[dwMaxDataLen];

-

- std::list<std::wstring> foundEntries;

- dwIndex = 0;

- do

- {

- dwValueNameLen = dwMaxNameLen;

- dwDataLen = dwMaxDataLen;

- status = RegEnumValueW(hSubKey, dwIndex++, szNameValue, &dwValueNameLen, NULL, &dwType, pbData, &dwDataLen);

- if (status == ERROR_SUCCESS)

- {

- if ( (wcslen(szNameValue) >= subStringLength && wcsstr(szNameValue, str))

- || (dwType == REG_SZ && wcslen((wchar_t*) pbData) >= subStringLength && wcsstr((wchar_t*) pbData, str))

- )

- {

- foundEntries.push_back(szNameValue);

- }

- }

- } while ((status == ERROR_SUCCESS) || (status == ERROR_MORE_DATA)); // we ignore ERROR_MORE_DATA errors since

- // we are sure to use the correct sizes

-

- // delete the entries

- if (!foundEntries.empty())

- {

- for (std::list<std::wstring>::iterator It = foundEntries.begin();

- It != foundEntries.end(); It++)

- {

- RegDeleteValueW (hSubKey, It->c_str());

- }

- }

-

- delete [] szNameValue;

- delete [] pbData;

- }

-

-

- RegCloseKey (hSubKey);

- }

- }

- }

-}

-

-/* Set the given privilege of the current process */

-BOOL SetPrivilege(LPTSTR szPrivilegeName, BOOL bEnable)

-{

- TOKEN_PRIVILEGES tp;

- LUID luid;

- HANDLE hProcessToken;

- BOOL bStatus = FALSE;

-

- if ( OpenProcessToken(GetCurrentProcess(),

- TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY,

- &hProcessToken) )

- {

- if ( LookupPrivilegeValue(

- NULL,

- szPrivilegeName,

- &luid ) )

- {

-

- tp.PrivilegeCount = 1;

- tp.Privileges[0].Luid = luid;

- tp.Privileges[0].Attributes = bEnable? SE_PRIVILEGE_ENABLED : SE_PRIVILEGE_REMOVED;

-

- // Enable the privilege

- bStatus = AdjustTokenPrivileges(

- hProcessToken,

- FALSE,

- &tp,

- sizeof(TOKEN_PRIVILEGES),

- (PTOKEN_PRIVILEGES) NULL,

- (PDWORD) NULL);

- }

-

- CloseHandle(hProcessToken);

- }

-

- return bStatus;

-}

-

-/*

- * Creates a VT_LPWSTR propvariant.

- * we use our own implementation to use SHStrDupW function pointer

- * that we retreive ourselves to avoid dll hijacking attacks

- */

-inline HRESULT VCInitPropVariantFromString(__in PCWSTR psz, __out PROPVARIANT *ppropvar)

-{

- ppropvar->vt = VT_LPWSTR;

- HRESULT hr = VCStrDupW(psz, &ppropvar->pwszVal);

- if (FAILED(hr))

- {

- PropVariantInit(ppropvar);

- }

- return hr;

-}

-

-HRESULT CreateLink (wchar_t *lpszPathObj, wchar_t *lpszArguments,

- wchar_t *lpszPathLink, const wchar_t* iconFile, int iconIndex)

-{

- HRESULT hres;

- IShellLink *psl;

-

- /* Get a pointer to the IShellLink interface. */

- hres = CoCreateInstance (CLSID_ShellLink, NULL,

- CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID *) &psl);

- if (SUCCEEDED (hres))

- {

- IPersistFile *ppf;

-

- /* Set the path to the shortcut target, and add the

- description. */

- psl->SetPath (lpszPathObj);

- psl->SetArguments (lpszArguments);

- if (iconFile)

- {

- psl->SetIconLocation (iconFile, iconIndex);

- }

-

- // Application ID

- if (_tcsstr (lpszPathObj, _T(TC_APP_NAME) _T(".exe")))

- {

- IPropertyStore *propStore;

-

- if (SUCCEEDED (psl->QueryInterface (IID_PPV_ARGS (&propStore))))

- {

- PROPVARIANT propVariant;

- if (SUCCEEDED (VCInitPropVariantFromString (TC_APPLICATION_ID, &propVariant)))

- {

- if (SUCCEEDED (propStore->SetValue (PKEY_AppUserModel_ID, propVariant)))

- propStore->Commit();

-

- PropVariantClear (&propVariant);

- }

-

- propStore->Release();

- }

- }

-

- /* Query IShellLink for the IPersistFile interface for saving

- the shortcut in persistent storage. */

- hres = psl->QueryInterface (IID_IPersistFile,

- (void **) &ppf);

-

- if (SUCCEEDED (hres))

- {

- /* Save the link by calling IPersistFile::Save. */

- hres = ppf->Save (lpszPathLink, TRUE);

- ppf->Release ();

- }

- psl->Release ();

- }

- return hres;

-}

-

-BOOL IsSystemRestoreEnabled ()

-{

- BOOL bEnabled = FALSE;

- HKEY hKey;

- DWORD dwValue = 0, cbValue = sizeof (DWORD);

- wchar_t szRegPath[MAX_PATH];

- GetRestorePointRegKeyName (szRegPath, sizeof (szRegPath));

- if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, szRegPath, 0, KEY_READ | KEY_WOW64_64KEY, &hKey) == ERROR_SUCCESS)

- {

- if (IsOSAtLeast (WIN_VISTA))

- {

- if ( (ERROR_SUCCESS == RegQueryValueEx (hKey, L"RPSessionInterval", NULL, NULL, (LPBYTE) &dwValue, &cbValue))

- && (dwValue == 1)

- )

- {

- bEnabled = TRUE;

- }

- }

- else

- {

- if ( (ERROR_SUCCESS == RegQueryValueEx (hKey, L"DisableSR", NULL, NULL, (LPBYTE) &dwValue, &cbValue))

- && (dwValue == 0)

- )

- {

- bEnabled = TRUE;

- }

- }

-

-

- RegCloseKey (hKey);

- }

-

- return bEnabled;

-}

-

-void GetProgramPath (HWND hwndDlg, wchar_t *path)

-{

- ITEMIDLIST *i;

- HRESULT res;

-

- if (bForAllUsers)

- res = SHGetSpecialFolderLocation (hwndDlg, CSIDL_COMMON_PROGRAMS, &i);

- else

- res = SHGetSpecialFolderLocation (hwndDlg, CSIDL_PROGRAMS, &i);

-

- SHGetPathFromIDList (i, path);

-}

-

-void StatusMessage (HWND hwndDlg, char *stringId)

-{

- if (Rollback)

- return;

-

- SendMessageW (GetDlgItem (hwndDlg, IDC_LOG_WINDOW), LB_ADDSTRING, 0, (LPARAM) GetString (stringId));

-

- SendDlgItemMessage (hwndDlg, IDC_LOG_WINDOW, LB_SETTOPINDEX,

- SendDlgItemMessage (hwndDlg, IDC_LOG_WINDOW, LB_GETCOUNT, 0, 0) - 1, 0);

-}

-

-void StatusMessageParam (HWND hwndDlg, char *stringId, wchar_t *param)

-{

- wchar_t szTmp[1024];

-

- if (Rollback)

- return;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s %s", GetString (stringId), param);

- SendMessageW (GetDlgItem (hwndDlg, IDC_LOG_WINDOW), LB_ADDSTRING, 0, (LPARAM) szTmp);

-

- SendDlgItemMessage (hwndDlg, IDC_LOG_WINDOW, LB_SETTOPINDEX,

- SendDlgItemMessage (hwndDlg, IDC_LOG_WINDOW, LB_GETCOUNT, 0, 0) - 1, 0);

-}

-

-void ClearLogWindow (HWND hwndDlg)

-{

- SendMessage (GetDlgItem (hwndDlg, IDC_LOG_WINDOW), LB_RESETCONTENT, 0, 0);

-}

-

-void RegMessage (HWND hwndDlg, wchar_t *txt)

-{

- StatusMessageParam (hwndDlg, "ADDING_REG", txt);

-}

-

-void CopyMessage (HWND hwndDlg, wchar_t *txt)

-{

- StatusMessageParam (hwndDlg, "INSTALLING", txt);

-}

-

-void RemoveMessage (HWND hwndDlg, wchar_t *txt)

-{

- if (!Rollback)

- StatusMessageParam (hwndDlg, "REMOVING", txt);

-}

-

-void IconMessage (HWND hwndDlg, wchar_t *txt)

-{

- StatusMessageParam (hwndDlg, "ADDING_ICON", txt);

-}

-

-void DetermineUpgradeDowngradeStatus (BOOL bCloseDriverHandle, LONG *driverVersionPtr)

-{

- LONG driverVersion = VERSION_NUM;

- int status = 0;

-

- if (hDriver == INVALID_HANDLE_VALUE)

- status = DriverAttach();

-

- if ((status == 0) && (hDriver != INVALID_HANDLE_VALUE))

- {

- DWORD dwResult;

- BOOL bResult = DeviceIoControl (hDriver, TC_IOCTL_GET_DRIVER_VERSION, NULL, 0, &driverVersion, sizeof (driverVersion), &dwResult, NULL);

-

- if (!bResult)

- bResult = DeviceIoControl (hDriver, TC_IOCTL_LEGACY_GET_DRIVER_VERSION, NULL, 0, &driverVersion, sizeof (driverVersion), &dwResult, NULL);

-

-

- bUpgrade = (bResult && driverVersion <= VERSION_NUM);

- bDowngrade = (bResult && driverVersion > VERSION_NUM);

- bReinstallMode = (bResult && driverVersion == VERSION_NUM);

-

- PortableMode = DeviceIoControl (hDriver, TC_IOCTL_GET_PORTABLE_MODE_STATUS, NULL, 0, NULL, 0, &dwResult, NULL);

-

- if (bCloseDriverHandle)

- {

- CloseHandle (hDriver);

- hDriver = INVALID_HANDLE_VALUE;

- }

- }

-

- *driverVersionPtr = driverVersion;

-}

-

-

-static BOOL IsFileInUse (const wstring &filePath)

-{

- HANDLE useTestHandle = CreateFile (filePath.c_str(), GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);

-

- if (useTestHandle != INVALID_HANDLE_VALUE)

- CloseHandle (useTestHandle);

- else if (GetLastError() == ERROR_SHARING_VIOLATION)

- return TRUE;

-

- return FALSE;

-}

-

-

-BOOL DoFilesInstall (HWND hwndDlg, wchar_t *szDestDir)

-{

- /* WARNING: Note that, despite its name, this function is used during UNinstallation as well. */

-

- wchar_t szTmp[TC_MAX_PATH];

- BOOL bOK = TRUE;

- int i, x, fileNo;

- wchar_t curFileName [TC_MAX_PATH] = {0};

-

- if (!bUninstall && !bDevm)

- {

- // Self-extract all files to memory

-

- GetModuleFileName (NULL, szTmp, ARRAYSIZE (szTmp));

-

- if (!SelfExtractInMemory (szTmp))

- return FALSE;

- }

-

- x = wcslen (szDestDir);

- if (x < 2)

- return FALSE;

-

- if (szDestDir[x - 1] != L'\\')

- StringCbCatW (szDestDir, MAX_PATH, L"\\");

-

- for (i = 0; i < sizeof (szFiles) / sizeof (szFiles[0]); i++)

- {

- BOOL bResult, driver64 = FALSE;

- wchar_t szDir[TC_MAX_PATH];

-

- if (wcsstr (szFiles[i], L"VeraCrypt Setup") != 0)

- {

- if (bUninstall)

- continue; // Prevent 'access denied' error

-

- if (bRepairMode)

- continue; // Destination = target

- }

-

- // skip files that don't apply to the current architecture

- if ( (Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCrypt-x64.exe") == 0))

- || (Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCryptExpander-x64.exe") == 0))

- || (Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCrypt Format-x64.exe") == 0))

- || (!Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCrypt-x86.exe") == 0))

- || (!Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCryptExpander-x86.exe") == 0))

- || (!Is64BitOs () && (wcscmp (szFiles[i], L"AVeraCrypt Format-x86.exe") == 0))

- )

- {

- continue;

- }

-

- if (*szFiles[i] == L'A')

- StringCbCopyW (szDir, sizeof(szDir), szDestDir);

- else if (*szFiles[i] == L'D')

- {

- if (Is64BitOs ())

- driver64 = TRUE;

-

- GetSystemDirectory (szDir, ARRAYSIZE (szDir));

-

- x = wcslen (szDir);

- if (szDir[x - 1] != L'\\')

- StringCbCatW (szDir, sizeof(szDir), L"\\");

-

- StringCbCatW (szDir, sizeof(szDir), L"Drivers\\");

- }

- else if (*szFiles[i] == L'W')

- GetWindowsDirectory (szDir, ARRAYSIZE (szDir));

-

- if (*szFiles[i] == L'I')

- continue;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s%s", szDir, szFiles[i] + 1);

-

- if (bUninstall == FALSE)

- CopyMessage (hwndDlg, szTmp);

- else

- RemoveMessage (hwndDlg, szTmp);

-

- if (bUninstall == FALSE)

- {

- SetCurrentDirectory (SetupFilesDir);

-

- if (wcsstr (szFiles[i], L"VeraCrypt Setup") != 0)

- {

- // Copy ourselves (the distribution package) to the destination location as 'VeraCrypt Setup.exe'

-

- wchar_t mp[MAX_PATH];

-

- GetModuleFileName (NULL, mp, ARRAYSIZE (mp));

- bResult = TCCopyFile (mp, szTmp);

- }

- else

- {

- StringCchCopyNW (curFileName, ARRAYSIZE(curFileName), szFiles[i] + 1, wcslen (szFiles[i]) - 1);

- curFileName [wcslen (szFiles[i]) - 1] = 0;

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"Dveracrypt.sys") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), FILENAME_64BIT_DRIVER, sizeof (FILENAME_64BIT_DRIVER));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCrypt.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCrypt-x64.exe", sizeof (L"VeraCrypt-x64.exe"));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCrypt-x86.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCrypt.exe", sizeof (L"VeraCrypt.exe"));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCryptExpander.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCryptExpander-x64.exe", sizeof (L"VeraCryptExpander-x64.exe"));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCryptExpander-x86.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCryptExpander.exe", sizeof (L"VeraCryptExpander.exe"));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCrypt Format.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCrypt Format-x64.exe", sizeof (L"VeraCrypt Format-x64.exe"));

- }

-

- if (Is64BitOs ()

- && wcscmp (szFiles[i], L"AVeraCrypt Format-x86.exe") == 0)

- {

- StringCbCopyNW (curFileName, sizeof(curFileName), L"VeraCrypt Format.exe", sizeof (L"VeraCrypt Format.exe"));

- }

-

- if (!bDevm)

- {

- bResult = FALSE;

-

- // Find the correct decompressed file in memory

- for (fileNo = 0; fileNo < NBR_COMPRESSED_FILES; fileNo++)

- {

- // Write the file (stored in memory) directly to the destination location

- // (there will be no temporary files).

- if (wmemcmp (

- curFileName,

- Decompressed_Files[fileNo].fileName,

- min (wcslen (curFileName), (size_t) Decompressed_Files[fileNo].fileNameLength)) == 0)

- {

- // Dump filter driver cannot be installed to SysWOW64 directory

- if (driver64 && !EnableWow64FsRedirection (FALSE))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- bResult = FALSE;

- goto err;

- }

-

- bResult = SaveBufferToFile (

- (char *) Decompressed_Files[fileNo].fileContent,

- szTmp,

- Decompressed_Files[fileNo].fileLength,

- FALSE,

- TRUE);

-

- if (driver64)

- {

- if (!EnableWow64FsRedirection (TRUE))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- bResult = FALSE;

- goto err;

- }

-

- if (!bResult)

- goto err;

-

- }

-

- break;

- }

- }

- }

- else

- {

- if (driver64)

- EnableWow64FsRedirection (FALSE);

-

- bResult = TCCopyFile (curFileName, szTmp);

-

- if (driver64)

- EnableWow64FsRedirection (TRUE);

- }

-

- if (bResult && wcscmp (szFiles[i], L"AVeraCrypt.exe") == 0)

- {

- if (Is64BitOs ())

- EnableWow64FsRedirection (FALSE);

-

- wstring servicePath = GetServiceConfigPath (_T(TC_APP_NAME) L".exe", false);

- wstring serviceLegacyPath = GetServiceConfigPath (_T(TC_APP_NAME) L".exe", true);

- wstring favoritesFile = GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false);

- wstring favoritesLegacyFile = GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, true);

-

- if ( FileExists (servicePath.c_str())

- || (Is64BitOs () && FileExists (serviceLegacyPath.c_str()))

- )

- {

- CopyMessage (hwndDlg, (wchar_t *) servicePath.c_str());

- bResult = CopyFile (szTmp, servicePath.c_str(), FALSE);

- }

-

- if (bResult && Is64BitOs ()

- && FileExists (favoritesLegacyFile.c_str())

- && !FileExists (favoritesFile.c_str()))

- {

- // copy the favorites XML file to the native system directory

- bResult = CopyFile (favoritesLegacyFile.c_str(), favoritesFile.c_str(), FALSE);

- }

-

- if (bResult && Is64BitOs () && FileExists (favoritesFile.c_str()) && FileExists (servicePath.c_str()))

- {

- // Update the path of the service

- BootEncryption BootEncObj (hwndDlg);

-

- try

- {

- if (BootEncObj.GetDriverServiceStartType() == SERVICE_BOOT_START)

- {

- BootEncObj.UpdateSystemFavoritesService ();

- }

- }

- catch (...) {}

- }

-

- if (Is64BitOs ())

- {

- // delete files from legacy path

- if (FileExists (favoritesLegacyFile.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) favoritesLegacyFile.c_str());

- ForceDeleteFile (favoritesLegacyFile.c_str());

- }

-

- if (FileExists (serviceLegacyPath.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) serviceLegacyPath.c_str());

- ForceDeleteFile (serviceLegacyPath.c_str());

- }

-

- EnableWow64FsRedirection (TRUE);

- }

- }

- }

- }

- else

- {

- if (driver64)

- EnableWow64FsRedirection (FALSE);

- bResult = StatDeleteFile (szTmp, TRUE);

- if (driver64)

- EnableWow64FsRedirection (TRUE);

-

- if (bResult && wcscmp (szFiles[i], L"AVeraCrypt.exe") == 0)

- {

- if (Is64BitOs ())

- EnableWow64FsRedirection (FALSE);

-

- wstring servicePath = GetServiceConfigPath (_T(TC_APP_NAME) L".exe", false);

- wstring serviceLegacyPath = GetServiceConfigPath (_T(TC_APP_NAME) L".exe", true);

- wstring favoritesFile = GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false);

- wstring favoritesLegacyFile = GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, true);

-

- // delete all files related to system favorites service

- if (FileExists (favoritesFile.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) favoritesFile.c_str());

- ForceDeleteFile (favoritesFile.c_str());

- }

-

- if (FileExists (servicePath.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) servicePath.c_str());

- ForceDeleteFile (servicePath.c_str());

- }

-

- if (Is64BitOs ())

- {

- if (FileExists (favoritesLegacyFile.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) favoritesLegacyFile.c_str());

- ForceDeleteFile (favoritesLegacyFile.c_str());

- }

-

- if (FileExists (serviceLegacyPath.c_str()))

- {

- RemoveMessage (hwndDlg, (wchar_t *) serviceLegacyPath.c_str());

- ForceDeleteFile (serviceLegacyPath.c_str());

- }

-

- EnableWow64FsRedirection (TRUE);

- }

- }

- }

-

-err:

- if (bResult == FALSE)

- {

- LPVOID lpMsgBuf;

- DWORD dwError = GetLastError ();

- wchar_t szTmp2[700];

- wchar_t szErrorValue[16];

- wchar_t* pszDesc;

-

- FormatMessage (

- FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,

- NULL,

- dwError,

- MAKELANGID (LANG_NEUTRAL, SUBLANG_DEFAULT), /* Default language */

- (wchar_t *) &lpMsgBuf,

- 0,

- NULL

- );

-

- if (lpMsgBuf)

- pszDesc = (wchar_t*) lpMsgBuf;

- else

- {

- StringCbPrintfW (szErrorValue, sizeof (szErrorValue), L"0x%.8X", dwError);

- pszDesc = szErrorValue;

- }

-

- if (bUninstall == FALSE)

- StringCbPrintfW (szTmp2, sizeof(szTmp2), GetString ("INSTALL_OF_FAILED"), szTmp, pszDesc);

- else

- StringCbPrintfW (szTmp2, sizeof(szTmp2), GetString ("UNINSTALL_OF_FAILED"), szTmp, pszDesc);

-

- if (lpMsgBuf) LocalFree (lpMsgBuf);

-

- if (!Silent && MessageBoxW (hwndDlg, szTmp2, lpszTitle, MB_YESNO | MB_ICONHAND) != IDYES)

- return FALSE;

- }

- }

-

- // Language pack

- if (bUninstall == FALSE)

- {

- WIN32_FIND_DATA f;

- HANDLE h;

-

- SetCurrentDirectory (SetupFilesDir);

- h = FindFirstFile (L"Language.*.xml", &f);

-

- if (h != INVALID_HANDLE_VALUE)

- {

- wchar_t d[MAX_PATH*2];

- StringCbPrintfW (d, sizeof(d), L"%s%s", szDestDir, f.cFileName);

- CopyMessage (hwndDlg, d);

- TCCopyFile (f.cFileName, d);

- FindClose (h);

- }

-

- SetCurrentDirectory (SetupFilesDir);

- SetCurrentDirectory (L"Setup files");

- h = FindFirstFile (L"VeraCrypt User Guide.*.pdf", &f);

- if (h != INVALID_HANDLE_VALUE)

- {

- wchar_t d[MAX_PATH*2];

- StringCbPrintfW (d, sizeof(d), L"%s%s", szDestDir, f.cFileName);

- CopyMessage (hwndDlg, d);

- TCCopyFile (f.cFileName, d);

- FindClose (h);

- }

- SetCurrentDirectory (SetupFilesDir);

- }

-

- return bOK;

-}

-

-BOOL DoRegInstall (HWND hwndDlg, wchar_t *szDestDir, BOOL bInstallType)

-{

- wchar_t szDir[TC_MAX_PATH], *key;

- wchar_t szTmp[TC_MAX_PATH*4];

- HKEY hkey = 0;

- BOOL bSlash, bOK = FALSE;

- DWORD dw;

- int x;

-

- if (SystemEncryptionUpdate)

- {

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\VeraCrypt",

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE | KEY_WOW64_32KEY, NULL, &hkey, &dw) == ERROR_SUCCESS)

- {

- StringCbCopyW (szTmp, sizeof(szTmp), _T(VERSION_STRING));

- RegSetValueEx (hkey, L"DisplayVersion", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t));

-

- StringCbCopyW (szTmp, sizeof(szTmp), _T(TC_HOMEPAGE));

- RegSetValueEx (hkey, L"URLInfoAbout", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t));

-

- RegCloseKey (hkey);

- }

-

- return TRUE;

- }

-

- StringCbCopyW (szDir, sizeof(szDir), szDestDir);

- x = wcslen (szDestDir);

- if (szDestDir[x - 1] == L'\\')

- bSlash = TRUE;

- else

- bSlash = FALSE;

-

- if (bSlash == FALSE)

- StringCbCatW (szDir, sizeof(szDir), L"\\");

-

- if (bInstallType)

- {

-

- key = L"Software\\Classes\\VeraCryptVolume";

- RegMessage (hwndDlg, key);

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,

- key,

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), L"VeraCrypt Volume");

- if (RegSetValueEx (hkey, L"", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%ws", TC_APPLICATION_ID);

- if (RegSetValueEx (hkey, L"AppUserModelID", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- RegCloseKey (hkey);

- hkey = 0;

-

- key = L"Software\\Classes\\VeraCryptVolume\\DefaultIcon";

- RegMessage (hwndDlg, key);

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,

- key,

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%sVeraCrypt.exe,1", szDir);

- if (RegSetValueEx (hkey, L"", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- RegCloseKey (hkey);

- hkey = 0;

-

- key = L"Software\\Classes\\VeraCryptVolume\\Shell\\open\\command";

- RegMessage (hwndDlg, key);

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,

- key,

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"\"%sVeraCrypt.exe\" /v \"%%1\"", szDir );

- if (RegSetValueEx (hkey, L"", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- RegCloseKey (hkey);

- hkey = 0;

-

- key = L"Software\\Classes\\.hc";

- BOOL typeClassChanged = TRUE;

- wchar_t typeClass[256];

- DWORD typeClassSize = sizeof (typeClass);

-

- if (ReadLocalMachineRegistryString (key, L"", typeClass, &typeClassSize) && typeClassSize > 0 && wcscmp (typeClass, L"VeraCryptVolume") == 0)

- typeClassChanged = FALSE;

-

- RegMessage (hwndDlg, key);

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,

- key,

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &hkey, &dw) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), L"VeraCryptVolume");

- if (RegSetValueEx (hkey, L"", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- RegCloseKey (hkey);

- hkey = 0;

-

- if (typeClassChanged)

- SHChangeNotify (SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);

- }

-

- key = L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\VeraCrypt";

- RegMessage (hwndDlg, key);

- if (RegCreateKeyEx (HKEY_LOCAL_MACHINE,

- key,

- 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE | KEY_WOW64_32KEY, NULL, &hkey, &dw) != ERROR_SUCCESS)

- goto error;

-

- /* IMPORTANT: IF YOU CHANGE THIS IN ANY WAY, REVISE AND UPDATE SetInstallationPath() ACCORDINGLY! */

- StringCbPrintfW (szTmp, sizeof(szTmp), L"\"%sVeraCrypt Setup.exe\" /u", szDir);

- if (RegSetValueEx (hkey, L"UninstallString", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"\"%sVeraCrypt Setup.exe\" /c", szDir);

- if (RegSetValueEx (hkey, L"ModifyPath", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"\"%sVeraCrypt Setup.exe\"", szDir);

- if (RegSetValueEx (hkey, L"DisplayIcon", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), _T(VERSION_STRING));

- if (RegSetValueEx (hkey, L"DisplayVersion", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), L"VeraCrypt");

- if (RegSetValueEx (hkey, L"DisplayName", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), L"IDRIX");

- if (RegSetValueEx (hkey, L"Publisher", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- StringCbCopyW (szTmp, sizeof(szTmp), _T(TC_HOMEPAGE));

- if (RegSetValueEx (hkey, L"URLInfoAbout", 0, REG_SZ, (BYTE *) szTmp, (wcslen (szTmp) + 1) * sizeof (wchar_t)) != ERROR_SUCCESS)

- goto error;

-

- bOK = TRUE;

-

-error:

- if (hkey != 0)

- RegCloseKey (hkey);

-

- if (bOK == FALSE)

- {

- handleWin32Error (hwndDlg, SRC_POS);

- Error ("REG_INSTALL_FAILED", hwndDlg);

- }

-

- // Register COM servers for UAC

- if (IsOSAtLeast (WIN_VISTA))

- {

- if (!RegisterComServers (szDir))

- {

- Error ("COM_REG_FAILED", hwndDlg);

- return FALSE;

- }

- }

-

- return bOK;

-}

-

-BOOL DoApplicationDataUninstall (HWND hwndDlg)

-{

- wchar_t path[MAX_PATH];

- wchar_t path2[MAX_PATH];

- BOOL bOK = TRUE;

-

- StatusMessage (hwndDlg, "REMOVING_APPDATA");

-

- SHGetFolderPath (NULL, CSIDL_APPDATA, NULL, 0, path);

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

-

- // Delete favorite volumes file

- StringCbPrintfW (path2, sizeof(path2), L"%s%s", path, TC_APPD_FILENAME_FAVORITE_VOLUMES);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- // Delete keyfile defaults

- StringCbPrintfW (path2, sizeof(path2), L"%s%s", path, TC_APPD_FILENAME_DEFAULT_KEYFILES);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- // Delete history file

- StringCbPrintfW (path2, sizeof(path2), L"%s%s", path, TC_APPD_FILENAME_HISTORY);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- // Delete configuration file

- StringCbPrintfW (path2, sizeof(path2), L"%s%s", path, TC_APPD_FILENAME_CONFIGURATION);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- // Delete system encryption configuration file

- StringCbPrintfW (path2, sizeof(path2), L"%s%s", path, TC_APPD_FILENAME_SYSTEM_ENCRYPTION);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- SHGetFolderPath (NULL, CSIDL_APPDATA, NULL, 0, path);

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

- RemoveMessage (hwndDlg, path);

- if (!StatRemoveDirectory (path))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- bOK = FALSE;

- }

-

- // remove VeraCrypt under common appdata

- if (SUCCEEDED (SHGetFolderPath (NULL, CSIDL_COMMON_APPDATA | CSIDL_FLAG_CREATE, NULL, 0, path)))

- {

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

-

- // Delete original bootloader

- StringCbPrintfW (path2, sizeof(path2), L"%s\\%s", path, TC_SYS_BOOT_LOADER_BACKUP_NAME);

- RemoveMessage (hwndDlg, path2);

- StatDeleteFile (path2, FALSE);

-

- // remove VeraCrypt folder

- RemoveMessage (hwndDlg, path);

- StatRemoveDirectory (path);

- }

-

-

- return bOK;

-}

-

-BOOL DoRegUninstall (HWND hwndDlg, BOOL bRemoveDeprecated)

-{

- wchar_t regk [64];

- typedef LSTATUS (WINAPI *RegDeleteKeyExWFn) (HKEY hKey,LPCWSTR lpSubKey,REGSAM samDesired,WORD Reserved);

- RegDeleteKeyExWFn RegDeleteKeyExWPtr = NULL;

- HMODULE hAdvapiDll = LoadLibrary (L"Advapi32.dll");

- if (hAdvapiDll)

- {

- RegDeleteKeyExWPtr = (RegDeleteKeyExWFn) GetProcAddress(hAdvapiDll, "RegDeleteKeyExW");

- }

-

- // Unregister COM servers

- if (!bRemoveDeprecated && IsOSAtLeast (WIN_VISTA))

- {

- if (!UnregisterComServers (InstallationPath))

- StatusMessage (hwndDlg, "COM_DEREG_FAILED");

- }

-

- if (!bRemoveDeprecated)

- StatusMessage (hwndDlg, "REMOVING_REG");

-

- if (RegDeleteKeyExWPtr)

- {

- RegDeleteKeyExWPtr (HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\VeraCrypt", KEY_WOW64_32KEY, 0);

- RegDeleteKeyExWPtr (HKEY_CURRENT_USER, L"Software\\VeraCrypt", KEY_WOW64_32KEY, 0);

- }

- else

- {

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\VeraCrypt");

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\VeraCrypt");

- }

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\VeraCryptVolume\\Shell\\open\\command");

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\VeraCryptVolume\\Shell\\open");

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\VeraCryptVolume\\Shell");

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\VeraCryptVolume\\DefaultIcon");

- RegDeleteKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\VeraCryptVolume");

-

- if (!bRemoveDeprecated)

- {

- HKEY hKey;

- GetStartupRegKeyName (regk, sizeof(regk));

- DeleteRegistryValue (regk, L"VeraCrypt");

-

- DeleteRegistryKey (HKEY_LOCAL_MACHINE, L"Software\\Classes\\.hc");

-

- // enable the SE_TAKE_OWNERSHIP_NAME privilege for this operation

- SetPrivilege (SE_TAKE_OWNERSHIP_NAME, TRUE);

-

- // clean MuiCache list from VeraCrypt entries

- SearchAndDeleteRegistrySubString (HKEY_CLASSES_ROOT, L"Local Settings\\Software\\Microsoft\\Windows\\Shell\\MuiCache", L"VeraCrypt", FALSE, NULL);

-

- // clean other VeraCrypt entries from all users

- SearchAndDeleteRegistrySubString (HKEY_USERS, L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\.hc", NULL, TRUE, NULL);

- SearchAndDeleteRegistrySubString (HKEY_USERS, L"Software\\Microsoft\\Windows NT\\CurrentVersion\\AppCompatFlags\\Compatibility Assistant\\Persisted", L"VeraCrypt", TRUE, NULL);

- SearchAndDeleteRegistrySubString (HKEY_USERS, L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\StartPage\\NewShortcuts", L"VeraCrypt", TRUE, NULL);

-

- if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS | WRITE_DAC | WRITE_OWNER, &hKey) == ERROR_SUCCESS)

- {

- SearchAndDeleteRegistrySubString (hKey, L"Enum\\Root\\LEGACY_VERACRYPT", NULL, TRUE, L"ControlSet");

- SearchAndDeleteRegistrySubString (hKey, L"services\\veracrypt", NULL, TRUE, L"ControlSet");

- RegCloseKey(hKey);

- }

-

- // disable the SE_TAKE_OWNERSHIP_NAME privilege for this operation

- SetPrivilege (SE_TAKE_OWNERSHIP_NAME, FALSE);

-

- SHChangeNotify (SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);

- }

-

- if (hAdvapiDll)

- FreeLibrary (hAdvapiDll);

-

- return TRUE;

-}

-

-

-BOOL DoServiceUninstall (HWND hwndDlg, wchar_t *lpszService)

-{

- SC_HANDLE hManager, hService = NULL;

- BOOL bOK = FALSE, bRet;

- SERVICE_STATUS status;

- BOOL firstTry = TRUE;

- int x;

-

- memset (&status, 0, sizeof (status)); /* Keep VC6 quiet */

-

-retry:

-

- hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS);

- if (hManager == NULL)

- goto error;

-

- hService = OpenService (hManager, lpszService, SERVICE_ALL_ACCESS);

- if (hService == NULL)

- goto error;

-

- if (wcscmp (L"veracrypt", lpszService) == 0)

- {

- try

- {

- BootEncryption bootEnc (hwndDlg);

- if (bootEnc.GetDriverServiceStartType() == SERVICE_BOOT_START)

- {

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::DriveFilter); } catch (...) { }

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::VolumeFilter); } catch (...) { }

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::DumpFilter); } catch (...) { }

- }

- }

- catch (...) { }

-

- StatusMessage (hwndDlg, "STOPPING_DRIVER");

- }

- else

- StatusMessageParam (hwndDlg, "STOPPING", lpszService);

-

-#define WAIT_PERIOD 3

-

- for (x = 0; x < WAIT_PERIOD; x++)

- {

- bRet = QueryServiceStatus (hService, &status);

- if (bRet != TRUE)

- goto error;

-

- if (status.dwCurrentState != SERVICE_START_PENDING &&

- status.dwCurrentState != SERVICE_STOP_PENDING &&

- status.dwCurrentState != SERVICE_CONTINUE_PENDING)

- break;

-

- Sleep (1000);

- }

-

- if (status.dwCurrentState != SERVICE_STOPPED)

- {

- bRet = ControlService (hService, SERVICE_CONTROL_STOP, &status);

- if (bRet == FALSE)

- goto try_delete;

-

- for (x = 0; x < WAIT_PERIOD; x++)

- {

- bRet = QueryServiceStatus (hService, &status);

- if (bRet != TRUE)

- goto error;

-

- if (status.dwCurrentState != SERVICE_START_PENDING &&

- status.dwCurrentState != SERVICE_STOP_PENDING &&

- status.dwCurrentState != SERVICE_CONTINUE_PENDING)

- break;

-

- Sleep (1000);

- }

-

- if (status.dwCurrentState != SERVICE_STOPPED && status.dwCurrentState != SERVICE_STOP_PENDING)

- goto error;

- }

-

-try_delete:

-

- if (wcscmp (L"veracrypt", lpszService) == 0)

- StatusMessage (hwndDlg, "REMOVING_DRIVER");

- else

- StatusMessageParam (hwndDlg, "REMOVING", lpszService);

-

- if (hService != NULL)

- {

- CloseServiceHandle (hService);

- hService = NULL;

- }

-

- if (hManager != NULL)

- {

- CloseServiceHandle (hManager);

- hManager = NULL;

- }

-

- hManager = OpenSCManager (NULL, NULL, SC_MANAGER_ALL_ACCESS);

- if (hManager == NULL)

- goto error;

-

- hService = OpenService (hManager, lpszService, SERVICE_ALL_ACCESS);

- if (hService == NULL)

- goto error;

-

- bRet = DeleteService (hService);

- if (bRet == FALSE)

- {

- if (firstTry && GetLastError () == ERROR_SERVICE_MARKED_FOR_DELETE)

- {

- // Second try for an eventual no-install driver instance

- CloseServiceHandle (hService);

- CloseServiceHandle (hManager);

- hService = NULL;

- hManager = NULL;

-

- Sleep(1000);

- firstTry = FALSE;

- goto retry;

- }

-

- goto error;

- }

-

- bOK = TRUE;

-

-error:

-

- if (bOK == FALSE && GetLastError ()!= ERROR_SERVICE_DOES_NOT_EXIST)

- {

- handleWin32Error (hwndDlg, SRC_POS);

- MessageBoxW (hwndDlg, GetString ("DRIVER_UINSTALL_FAILED"), lpszTitle, MB_ICONHAND);

- }

- else

- bOK = TRUE;

-

- if (hService != NULL)

- CloseServiceHandle (hService);

-

- if (hManager != NULL)

- CloseServiceHandle (hManager);

-

- return bOK;

-}

-

-

-BOOL DoDriverUnload (HWND hwndDlg)

-{

- BOOL bOK = TRUE;

- int status;

-

- status = DriverAttach ();

- if (status != 0)

- {

- if (status == ERR_OS_ERROR && GetLastError () != ERROR_FILE_NOT_FOUND)

- {

- handleWin32Error (hwndDlg, SRC_POS);

- AbortProcess ("NODRIVER");

- }

-

- if (status != ERR_OS_ERROR)

- {

- handleError (NULL, status, SRC_POS);

- AbortProcess ("NODRIVER");

- }

- }

-

- if (hDriver != INVALID_HANDLE_VALUE)

- {

- MOUNT_LIST_STRUCT driver;

- LONG driverVersion = VERSION_NUM;

- int refCount;

- DWORD dwResult;

- BOOL bResult;

-

- // Try to determine if it's upgrade (and not reinstall, downgrade, or first-time install).

- DetermineUpgradeDowngradeStatus (FALSE, &driverVersion);

-

- // Test for encrypted boot drive

- try

- {

- BootEncryption bootEnc (hwndDlg);

- if (bootEnc.GetDriverServiceStartType() == SERVICE_BOOT_START)

- {

- try

- {

- // Check hidden OS update consistency

- if (IsHiddenOSRunning())

- {

- if (bootEnc.GetInstalledBootLoaderVersion() != VERSION_NUM)

- {

- if (AskWarnNoYes ("UPDATE_TC_IN_DECOY_OS_FIRST", hwndDlg) == IDNO)

- AbortProcessSilent ();

- }

- }

- }

- catch (...) { }

-

- if (bUninstallInProgress && !bootEnc.GetStatus().DriveMounted)

- {

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::DriveFilter); } catch (...) { }

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::VolumeFilter); } catch (...) { }

- try { bootEnc.RegisterFilterDriver (false, BootEncryption::DumpFilter); } catch (...) { }

- bootEnc.SetDriverServiceStartType (SERVICE_SYSTEM_START);

- }

- else if (bUninstallInProgress || bDowngrade)

- {

- Error (bDowngrade ? "SETUP_FAILED_BOOT_DRIVE_ENCRYPTED_DOWNGRADE" : "SETUP_FAILED_BOOT_DRIVE_ENCRYPTED", hwndDlg);

- return FALSE;

- }

- else

- {

- if (CurrentOSMajor == 6 && CurrentOSMinor == 0 && CurrentOSServicePack < 1)

- AbortProcess ("SYS_ENCRYPTION_UPGRADE_UNSUPPORTED_ON_VISTA_SP0");

-

- SystemEncryptionUpdate = TRUE;

- PortableMode = FALSE;

- }

- }

- }

- catch (...) { }

-

- if (!bUninstall

- && (bUpgrade || SystemEncryptionUpdate)

- && (!bDevm || SystemEncryptionUpdate))

- {

- UnloadDriver = FALSE;

- }

-

- if (PortableMode && !SystemEncryptionUpdate)

- UnloadDriver = TRUE;

-

- if (UnloadDriver)

- {

- int volumesMounted = 0;

-

- // Check mounted volumes

- bResult = DeviceIoControl (hDriver, TC_IOCTL_IS_ANY_VOLUME_MOUNTED, NULL, 0, &volumesMounted, sizeof (volumesMounted), &dwResult, NULL);

-

- if (!bResult)

- {

- bResult = DeviceIoControl (hDriver, TC_IOCTL_LEGACY_GET_MOUNTED_VOLUMES, NULL, 0, &driver, sizeof (driver), &dwResult, NULL);

- if (bResult)

- volumesMounted = driver.ulMountedDrives;

- }

-

- if (bResult)

- {

- if (volumesMounted != 0)

- {

- bOK = FALSE;

- MessageBoxW (hwndDlg, GetString ("DISMOUNT_ALL_FIRST"), lpszTitle, MB_ICONHAND);

- }

- }

- else

- {

- bOK = FALSE;

- handleWin32Error (hwndDlg, SRC_POS);

- }

- }

-

- // Try to close all open TC windows

- if (bOK)

- {

- BOOL TCWindowClosed = FALSE;

-

- EnumWindows (CloseTCWindowsEnum, (LPARAM) &TCWindowClosed);

-

- if (TCWindowClosed)

- Sleep (2000);

- }

-

- // Test for any applications attached to driver

- if (!bUpgrade)

- {

- bResult = DeviceIoControl (hDriver, TC_IOCTL_GET_DEVICE_REFCOUNT, &refCount, sizeof (refCount), &refCount,

- sizeof (refCount), &dwResult, NULL);

-

- if (bOK && bResult && refCount > 1)

- {

- MessageBoxW (hwndDlg, GetString ("CLOSE_TC_FIRST"), lpszTitle, MB_ICONSTOP);

- bOK = FALSE;

- }

- }

-

- if (!bOK || UnloadDriver)

- {

- CloseHandle (hDriver);

- hDriver = INVALID_HANDLE_VALUE;

- }

- }

- else

- {

- // Note that the driver may have already been unloaded during this session (e.g. retry after an error, etc.) so it is not

- // guaranteed that the user is installing VeraCrypt for the first time now (we also cannot know if the user has already

- // installed and used VeraCrypt on another system before).

- bPossiblyFirstTimeInstall = TRUE;

- }

-

- return bOK;

-}

-

-

-BOOL UpgradeBootLoader (HWND hwndDlg)

-{

- if (!SystemEncryptionUpdate)

- return TRUE;

-

- try

- {

- BootEncryption bootEnc (hwndDlg);

- uint64 bootLoaderVersion = bootEnc.GetInstalledBootLoaderVersion();

- if ((bootLoaderVersion < VERSION_NUM) || (bReinstallMode && (bootLoaderVersion == VERSION_NUM)))

- {

- StatusMessage (hwndDlg, "INSTALLER_UPDATING_BOOT_LOADER");

-

- bootEnc.InstallBootLoader (true);

-

- if (bootEnc.GetInstalledBootLoaderVersion() <= TC_RESCUE_DISK_UPGRADE_NOTICE_MAX_VERSION)

- Info (IsHiddenOSRunning() ? "BOOT_LOADER_UPGRADE_OK_HIDDEN_OS" : "BOOT_LOADER_UPGRADE_OK", hwndDlg);

- }

- return TRUE;

- }

- catch (Exception &e)

- {

- e.Show (hwndDlg);

- }

- catch (...) { }

-

- Error ("BOOT_LOADER_UPGRADE_FAILED", hwndDlg);

- return FALSE;

-}

-

-

-BOOL DoShortcutsUninstall (HWND hwndDlg, wchar_t *szDestDir)

-{

- wchar_t szLinkDir[TC_MAX_PATH];

- wchar_t szTmp2[TC_MAX_PATH];

- BOOL bSlash, bOK = FALSE;

- HRESULT hOle;

- int x;

- BOOL allUsers = FALSE;

-

- hOle = OleInitialize (NULL);

-

- // User start menu

- SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_PROGRAMS, 0);

- x = wcslen (szLinkDir);

- if (szLinkDir[x - 1] == L'\\')

- bSlash = TRUE;

- else

- bSlash = FALSE;

-

- if (bSlash == FALSE)

- StringCbCatW (szLinkDir, sizeof(szLinkDir), L"\\");

-

- StringCbCatW (szLinkDir, sizeof(szLinkDir), L"VeraCrypt");

-

- // Global start menu

- {

- struct _stat st;

- wchar_t path[TC_MAX_PATH];

-

- SHGetSpecialFolderPath (hwndDlg, path, CSIDL_COMMON_PROGRAMS, 0);

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

-

- if (_wstat (path, &st) == 0)

- {

- StringCbCopyW (szLinkDir, sizeof(szLinkDir), path);

- allUsers = TRUE;

- }

- }

-

- // Start menu entries

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt.lnk");

- RemoveMessage (hwndDlg, szTmp2);

- if (StatDeleteFile (szTmp2, FALSE) == FALSE)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCryptExpander.lnk");

- RemoveMessage (hwndDlg, szTmp2);

- if (StatDeleteFile (szTmp2, FALSE) == FALSE)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt Website.url");

- RemoveMessage (hwndDlg, szTmp2);

- if (StatDeleteFile (szTmp2, FALSE) == FALSE)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\Uninstall VeraCrypt.lnk");

- RemoveMessage (hwndDlg, szTmp2);

- if (StatDeleteFile (szTmp2, FALSE) == FALSE)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt User's Guide.lnk");

- StatDeleteFile (szTmp2, FALSE);

-

- // Start menu group

- RemoveMessage ((HWND) hwndDlg, szLinkDir);

- if (StatRemoveDirectory (szLinkDir) == FALSE)

- handleWin32Error ((HWND) hwndDlg, SRC_POS);

-

- // Desktop icon

-

- if (allUsers)

- SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_COMMON_DESKTOPDIRECTORY, 0);

- else

- SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_DESKTOPDIRECTORY, 0);

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt.lnk");

-

- RemoveMessage (hwndDlg, szTmp2);

- if (StatDeleteFile (szTmp2, FALSE) == FALSE)

- goto error;

-

- bOK = TRUE;

-

-error:

- OleUninitialize ();

-

- return bOK;

-}

-

-BOOL DoShortcutsInstall (HWND hwndDlg, wchar_t *szDestDir, BOOL bProgGroup, BOOL bDesktopIcon)

-{

- wchar_t szLinkDir[TC_MAX_PATH], szDir[TC_MAX_PATH];

- wchar_t szTmp[TC_MAX_PATH], szTmp2[TC_MAX_PATH], szTmp3[TC_MAX_PATH];

- BOOL bSlash, bOK = FALSE;

- HRESULT hOle;

- int x;

-

- if (bProgGroup == FALSE && bDesktopIcon == FALSE)

- return TRUE;

-

- hOle = OleInitialize (NULL);

-

- GetProgramPath (hwndDlg, szLinkDir);

-

- x = wcslen (szLinkDir);

- if (szLinkDir[x - 1] == L'\\')

- bSlash = TRUE;

- else

- bSlash = FALSE;

-

- if (bSlash == FALSE)

- StringCbCatW (szLinkDir, sizeof(szLinkDir), L"\\");

-

- StringCbCatW (szLinkDir, sizeof(szLinkDir), L"VeraCrypt");

-

- StringCbCopyW (szDir, sizeof(szDir), szDestDir);

- x = wcslen (szDestDir);

- if (szDestDir[x - 1] == L'\\')

- bSlash = TRUE;

- else

- bSlash = FALSE;

-

- if (bSlash == FALSE)

- StringCbCatW (szDir, sizeof(szDir), L"\\");

-

- if (bProgGroup)

- {

- FILE *f;

-

- if (mkfulldir (szLinkDir, TRUE) != 0)

- {

- if (mkfulldir (szLinkDir, FALSE) != 0)

- {

- wchar_t szTmpW[TC_MAX_PATH];

-

- handleWin32Error (hwndDlg, SRC_POS);

- StringCbPrintfW (szTmpW, sizeof(szTmpW), GetString ("CANT_CREATE_FOLDER"), szLinkDir);

- MessageBoxW (hwndDlg, szTmpW, lpszTitle, MB_ICONHAND);

- goto error;

- }

- }

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s%s", szDir, L"VeraCrypt.exe");

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt.lnk");

-

- IconMessage (hwndDlg, szTmp2);

- if (CreateLink (szTmp, L"", szTmp2, NULL, -1) != S_OK)

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s%s", szDir, L"VeraCryptExpander.exe");

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCryptExpander.lnk");

-

- IconMessage (hwndDlg, szTmp2);

- if (CreateLink (szTmp, L"", szTmp2, NULL, -1) != S_OK)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt Website.url");

- IconMessage (hwndDlg, szTmp2);

- f = _wfopen (szTmp2, L"w");

- if (f)

- {

- fprintf (f, "[InternetShortcut]\nURL=%s\n", TC_APPLINK);

-

- CheckFileStreamWriteErrors (hwndDlg, f, szTmp2);

- fclose (f);

- }

- else

- goto error;

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s%s", szDir, L"VeraCrypt Setup.exe");

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\Uninstall VeraCrypt.lnk");

- if (GetSystemDirectory (szTmp3, ARRAYSIZE(szTmp3)))

- {

- StringCbCatW (szTmp3, sizeof(szTmp3), L"\\control.exe");

- }

- else

- StringCbCopyW(szTmp3, sizeof(szTmp3), L"C:\\Windows\\System32\\control.exe");

-

- IconMessage (hwndDlg, szTmp2);

- if (CreateLink (szTmp3, L"appwiz.cpl", szTmp2, szTmp, 0) != S_OK)

- goto error;

-

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt User's Guide.lnk");

- StatDeleteFile (szTmp2, FALSE);

- }

-

- if (bDesktopIcon)

- {

- StringCbCopyW (szDir, sizeof(szDir), szDestDir);

- x = wcslen (szDestDir);

- if (szDestDir[x - 1] == L'\\')

- bSlash = TRUE;

- else

- bSlash = FALSE;

-

- if (bSlash == FALSE)

- StringCbCatW (szDir, sizeof(szDir), L"\\");

-

- if (bForAllUsers)

- SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_COMMON_DESKTOPDIRECTORY, 0);

- else

- SHGetSpecialFolderPath (hwndDlg, szLinkDir, CSIDL_DESKTOPDIRECTORY, 0);

-

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%s%s", szDir, L"VeraCrypt.exe");

- StringCbPrintfW (szTmp2, sizeof(szTmp2), L"%s%s", szLinkDir, L"\\VeraCrypt.lnk");

-

- IconMessage (hwndDlg, szTmp2);

-

- if (CreateLink (szTmp, L"", szTmp2, NULL, -1) != S_OK)

- goto error;

- }

-

- bOK = TRUE;

-

-error:

- OleUninitialize ();

-

- return bOK;

-}

-

-

-void OutcomePrompt (HWND hwndDlg, BOOL bOK)

-{

- if (bOK)

- {

- EnableWindow (GetDlgItem ((HWND) hwndDlg, IDCANCEL), FALSE);

-

- bDone = TRUE;

-

- if (bUninstall == FALSE)

- {

- if (bDevm)

- PostMessage (MainDlg, WM_CLOSE, 0, 0);

- else if (bPossiblyFirstTimeInstall || bRepairMode || (!bUpgrade && !bDowngrade))

- Info ("INSTALL_OK", hwndDlg);

- else

- Info ("SETUP_UPDATE_OK", hwndDlg);

- }

- else

- {

- wchar_t str[4096];

-

- StringCbPrintfW (str, sizeof(str), GetString ("UNINSTALL_OK"), InstallationPath);

- MessageBoxW (hwndDlg, str, lpszTitle, MB_ICONASTERISK);

- }

- }

- else

- {

- if (bUninstall == FALSE)

- Error ("INSTALL_FAILED", hwndDlg);

- else

- Error ("UNINSTALL_FAILED", hwndDlg);

- }

-}

-

-static void SetSystemRestorePoint (HWND hwndDlg, BOOL finalize)

-{

- static RESTOREPOINTINFO RestPtInfo;

- static STATEMGRSTATUS SMgrStatus;

- static BOOL failed = FALSE;

- static BOOL (__stdcall *_SRSetRestorePoint)(PRESTOREPOINTINFO, PSTATEMGRSTATUS);

-

- if (!SystemRestoreDll) return;

-

- _SRSetRestorePoint = (BOOL (__stdcall *)(PRESTOREPOINTINFO, PSTATEMGRSTATUS))GetProcAddress (SystemRestoreDll,"SRSetRestorePointW");

- if (_SRSetRestorePoint == 0)

- {

- FreeLibrary (SystemRestoreDll);

- SystemRestoreDll = 0;

- return;

- }

-

- if (!finalize)

- {

- StatusMessage (hwndDlg, "CREATING_SYS_RESTORE");

-

- RestPtInfo.dwEventType = BEGIN_SYSTEM_CHANGE;

- RestPtInfo.dwRestorePtType = bUninstall ? APPLICATION_UNINSTALL : APPLICATION_INSTALL | DEVICE_DRIVER_INSTALL;

- RestPtInfo.llSequenceNumber = 0;

- StringCbCopyW (RestPtInfo.szDescription, sizeof(RestPtInfo.szDescription), bUninstall ? L"VeraCrypt uninstallation" : L"VeraCrypt installation");

-

- if(!_SRSetRestorePoint (&RestPtInfo, &SMgrStatus))

- {

- StatusMessage (hwndDlg, "FAILED_SYS_RESTORE");

- failed = TRUE;

- }

- }

- else if (!failed)

- {

- RestPtInfo.dwEventType = END_SYSTEM_CHANGE;

- RestPtInfo.llSequenceNumber = SMgrStatus.llSequenceNumber;

-

- if(!_SRSetRestorePoint(&RestPtInfo, &SMgrStatus))

- {

- StatusMessage (hwndDlg, "FAILED_SYS_RESTORE");

- }

- }

-}

-

-void DoUninstall (void *arg)

-{

- HWND hwndDlg = (HWND) arg;

- BOOL bOK = TRUE;

- BOOL bTempSkipSysRestore = FALSE;

-

- if (!Rollback)

- EnableWindow (GetDlgItem ((HWND) hwndDlg, IDC_UNINSTALL), FALSE);

-

- WaitCursor ();

-

- if (!Rollback)

- {

- ClearLogWindow (hwndDlg);

- }

-

- if (DoDriverUnload (hwndDlg) == FALSE)

- {

- bOK = FALSE;

- bTempSkipSysRestore = TRUE; // Volumes are possibly mounted; defer System Restore point creation for this uninstall attempt.

- }

- else

- {

- if (!Rollback && bSystemRestore && !bTempSkipSysRestore)

- SetSystemRestorePoint (hwndDlg, FALSE);

-

- if (DoServiceUninstall (hwndDlg, L"veracrypt") == FALSE)

- {

- bOK = FALSE;

- }

- else if (DoRegUninstall ((HWND) hwndDlg, FALSE) == FALSE)

- {

- bOK = FALSE;

- }

- else if (DoFilesInstall ((HWND) hwndDlg, InstallationPath) == FALSE)

- {

- bOK = FALSE;

- }

- else if (DoShortcutsUninstall (hwndDlg, InstallationPath) == FALSE)

- {

- bOK = FALSE;

- }

- else if (!DoApplicationDataUninstall (hwndDlg))

- {

- bOK = FALSE;

- }

- else

- {

- wchar_t temp[MAX_PATH];

- FILE *f;

-

- // Deprecated service

- DoServiceUninstall (hwndDlg, L"VeraCryptService");

-

- GetTempPath (ARRAYSIZE (temp), temp);

- StringCbPrintfW (UninstallBatch, sizeof (UninstallBatch), L"%sVeraCrypt-Uninstall.bat", temp);

-

- UninstallBatch [ARRAYSIZE(UninstallBatch)-1] = 0;

-

- // Create uninstall batch

- f = _wfopen (UninstallBatch, L"w");

- if (!f)

- bOK = FALSE;

- else

- {

- fwprintf (f,L":loop\n"

- L"del \"%s%s\"\n"

- L"if exist \"%s%s\" goto loop\n"

- L"rmdir \"%s\"\n"

- L"del \"%s\"",

- InstallationPath, L"VeraCrypt Setup.exe",

- InstallationPath, L"VeraCrypt Setup.exe",

- InstallationPath,

- UninstallBatch

- );

-

- CheckFileStreamWriteErrors (hwndDlg, f, UninstallBatch);

- fclose (f);

- }

- }

- }

-

- NormalCursor ();

-

- if (Rollback)

- return;

-

- if (bSystemRestore && !bTempSkipSysRestore)

- SetSystemRestorePoint (hwndDlg, TRUE);

-

- if (bOK)

- PostMessage (hwndDlg, TC_APPMSG_UNINSTALL_SUCCESS, 0, 0);

- else

- bUninstallInProgress = FALSE;

-

- EnableWindow (GetDlgItem ((HWND) hwndDlg, IDC_UNINSTALL), TRUE);

- OutcomePrompt (hwndDlg, bOK);

-}

-

-void DoInstall (void *arg)

-{

- HWND hwndDlg = (HWND) arg;

- BOOL bOK = TRUE;

- wchar_t path[MAX_PATH];

-

- BootEncryption bootEnc (hwndDlg);

-

- // Refresh the main GUI (wizard thread)

- InvalidateRect (MainDlg, NULL, TRUE);

-

- ClearLogWindow (hwndDlg);

-

- if (mkfulldir (InstallationPath, TRUE) != 0)

- {

- if (mkfulldir (InstallationPath, FALSE) != 0)

- {

- wchar_t szTmp[TC_MAX_PATH];

-

- handleWin32Error (hwndDlg, SRC_POS);

- StringCbPrintfW (szTmp, sizeof(szTmp), GetString ("CANT_CREATE_FOLDER"), InstallationPath);

- MessageBoxW (hwndDlg, szTmp, lpszTitle, MB_ICONHAND);

- Error ("INSTALL_FAILED", hwndDlg);

- PostMessage (MainDlg, TC_APPMSG_INSTALL_FAILURE, 0, 0);

- return;

- }

- }

-

- UpdateProgressBarProc(2);

-

- if (DoDriverUnload (hwndDlg) == FALSE)

- {

- NormalCursor ();

- PostMessage (MainDlg, TC_APPMSG_INSTALL_FAILURE, 0, 0);

- return;

- }

-

- if (bUpgrade

- && (IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L".exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L"-x86.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L"-x64.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L" Format.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L" Format-x86.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L" Format-x64.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L"Expander.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L"Expander-x86.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L"Expander-x64.exe")

- || IsFileInUse (wstring (InstallationPath) + L'\\' + _T(TC_APP_NAME) L" Setup.exe")

- )

- )

- {

- NormalCursor ();

- Error ("CLOSE_TC_FIRST", hwndDlg);

- PostMessage (MainDlg, TC_APPMSG_INSTALL_FAILURE, 0, 0);

- return;

- }

-

- UpdateProgressBarProc(12);

-

- if (bSystemRestore)

- SetSystemRestorePoint (hwndDlg, FALSE);

-

- UpdateProgressBarProc(48);

-

- if (bDisableSwapFiles

- && IsPagingFileActive (FALSE))

- {

- if (!DisablePagingFile())

- {

- handleWin32Error (hwndDlg, SRC_POS);

- Error ("FAILED_TO_DISABLE_PAGING_FILES", hwndDlg);

- }

- else

- bRestartRequired = TRUE;

- }

-

- UpdateProgressBarProc(50);

-

- // Remove deprecated

- DoServiceUninstall (hwndDlg, L"VeraCryptService");

-

- UpdateProgressBarProc(55);

-

- if (!SystemEncryptionUpdate)

- DoRegUninstall ((HWND) hwndDlg, TRUE);

-

- UpdateProgressBarProc(61);

-

- GetWindowsDirectory (path, ARRAYSIZE (path));

- StringCbCatW (path, sizeof (path), L"\\VeraCrypt Setup.exe");

- StatDeleteFile (path, FALSE);

-

- if (UpdateProgressBarProc(63) && UnloadDriver && DoServiceUninstall (hwndDlg, L"veracrypt") == FALSE)

- {

- bOK = FALSE;

- }

- else if (UpdateProgressBarProc(72) && DoFilesInstall ((HWND) hwndDlg, InstallationPath) == FALSE)

- {

- bOK = FALSE;

- }

- else if (UpdateProgressBarProc(80) && DoRegInstall ((HWND) hwndDlg, InstallationPath, bRegisterFileExt) == FALSE)

- {

- bOK = FALSE;

- }

- else if (UpdateProgressBarProc(85) && UnloadDriver && DoDriverInstall (hwndDlg) == FALSE)

- {

- bOK = FALSE;

- }

- else if (UpdateProgressBarProc(90) && SystemEncryptionUpdate && UpgradeBootLoader (hwndDlg) == FALSE)

- {

- bOK = FALSE;

- }

- else if (UpdateProgressBarProc(93) && DoShortcutsInstall (hwndDlg, InstallationPath, bAddToStartMenu, bDesktopIcon) == FALSE)

- {

- bOK = FALSE;

- }

-

- if (!UnloadDriver)

- bRestartRequired = TRUE;

-

- try

- {

- bootEnc.RenameDeprecatedSystemLoaderBackup();

- }

- catch (...) { }

-

- if (bOK)

- UpdateProgressBarProc(97);

-

- if (bSystemRestore)

- SetSystemRestorePoint (hwndDlg, TRUE);

-

- if (bOK)

- {

- UpdateProgressBarProc(100);

- UninstallBatch[0] = 0;

- StatusMessage (hwndDlg, "INSTALL_COMPLETED");

- }

- else

- {

- UpdateProgressBarProc(0);

-

- if (!SystemEncryptionUpdate)

- {

- bUninstall = TRUE;

- Rollback = TRUE;

- Silent = TRUE;

-

- DoUninstall (hwndDlg);

-

- bUninstall = FALSE;

- Rollback = FALSE;

- Silent = FALSE;

-

- StatusMessage (hwndDlg, "ROLLBACK");

- }

- else

- {

- Warning ("SYS_ENC_UPGRADE_FAILED", hwndDlg);

- }

- }

-

- OutcomePrompt (hwndDlg, bOK);

-

- if (bOK && !bUninstall && !bDowngrade && !bRepairMode && !bDevm)

- {

- if (!IsHiddenOSRunning()) // A hidden OS user should not see the post-install notes twice (on decoy OS and then on hidden OS).

- {

- if (bRestartRequired || SystemEncryptionUpdate)

- {

- // Restart required

-

- if (bUpgrade)

- {

- SavePostInstallTasksSettings (TC_POST_INSTALL_CFG_RELEASE_NOTES);

- }

- else if (bPossiblyFirstTimeInstall)

- {

- SavePostInstallTasksSettings (TC_POST_INSTALL_CFG_TUTORIAL);

- }

- }

- else

- {

- // No restart will be required

-

- if (bUpgrade)

- {

- bPromptReleaseNotes = TRUE;

- }

- else if (bPossiblyFirstTimeInstall)

- {

- bPromptTutorial = TRUE;

- }

- }

- }

- }

-

- PostMessage (MainDlg, bOK ? TC_APPMSG_INSTALL_SUCCESS : TC_APPMSG_INSTALL_FAILURE, 0, 0);

-}

-

-

-void SetInstallationPath (HWND hwndDlg)

-{

- HKEY hkey;

- BOOL bInstallPathDetermined = FALSE;

- wchar_t path[MAX_PATH+20];

- ITEMIDLIST *itemList;

-

- memset (InstallationPath, 0, sizeof (InstallationPath));

-

- // Determine if VeraCrypt is already installed and try to determine its "Program Files" location

- if (RegOpenKeyEx (HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\VeraCrypt", 0, KEY_READ | KEY_WOW64_32KEY, &hkey) == ERROR_SUCCESS)

- {

- /* Default 'UninstallString' registry strings written by VeraCrypt:

- ------------------------------------------------------------------------------------

- 5.0+ "C:\Program Files\VeraCrypt\VeraCrypt Setup.exe" /u

- */

-

- wchar_t rv[MAX_PATH*4];

- DWORD size = sizeof (rv);

- if (RegQueryValueEx (hkey, L"UninstallString", 0, 0, (LPBYTE) &rv, &size) == ERROR_SUCCESS && wcsrchr (rv, L'/'))

- {

- size_t len = 0;

-

- // Cut and paste the location (path) where VeraCrypt is installed to InstallationPath

- if (rv[0] == L'"')

- {

- len = wcsrchr (rv, L'/') - rv - 2;

- StringCchCopyNW (InstallationPath, ARRAYSIZE(InstallationPath), rv + 1, len);

- InstallationPath [len] = 0;

- bInstallPathDetermined = TRUE;

-

- if (InstallationPath [wcslen (InstallationPath) - 1] != L'\\')

- {

- len = wcsrchr (InstallationPath, L'\\') - InstallationPath;

- InstallationPath [len] = 0;

- }

- }

-

- }

- RegCloseKey (hkey);

- }

-

- if (bInstallPathDetermined)

- {

- wchar_t mp[MAX_PATH];

-

- // Determine whether we were launched from the folder where VeraCrypt is installed

- GetModuleFileName (NULL, mp, ARRAYSIZE (mp));

- if (wcsncmp (InstallationPath, mp, min (wcslen(InstallationPath), wcslen(mp))) == 0)

- {

- // We were launched from the folder where VeraCrypt is installed

-

- if (!IsNonInstallMode() && !bDevm)

- bChangeMode = TRUE;

- }

- }

- else

- {

- /* VeraCrypt is not installed or it wasn't possible to determine where it is installed. */

-

- // Default "Program Files" path.

- SHGetSpecialFolderLocation (hwndDlg, CSIDL_PROGRAM_FILES, &itemList);

- SHGetPathFromIDList (itemList, path);

-

- if (Is64BitOs())

- {

- // Use a unified default installation path (registry redirection of %ProgramFiles% does not work if the installation path is user-selectable)

- wstring s = path;

- size_t p = s.find (L" (x86)");

- if (p != wstring::npos)

- {

- s = s.substr (0, p);

- if (_waccess (s.c_str(), 0) != -1)

- StringCbCopyW (path, sizeof (path), s.c_str());

- }

- }

-

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

- StringCbCopyW (InstallationPath, sizeof(InstallationPath), path);

- }

-

- // Make sure the path ends with a backslash

- if (InstallationPath [wcslen (InstallationPath) - 1] != L'\\')

- {

- StringCbCatW (InstallationPath, sizeof(InstallationPath), L"\\");

- }

-}

-

-

-// Handler for uninstall only (install is handled by the wizard)

-BOOL CALLBACK UninstallDlgProc (HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)

-{

- WORD lw = LOWORD (wParam);

-

- switch (msg)

- {

- case WM_INITDIALOG:

-

- MainDlg = hwndDlg;

-

- if (!CreateAppSetupMutex ())

- AbortProcess ("TC_INSTALLER_IS_RUNNING");

-

- InitDialog (hwndDlg);

- LocalizeDialog (hwndDlg, NULL);

-

- SetWindowTextW (hwndDlg, lpszTitle);

-

- // System Restore

- SetCheckBox (hwndDlg, IDC_SYSTEM_RESTORE, bSystemRestore);

- if (SystemRestoreDll == 0)

- {

- SetCheckBox (hwndDlg, IDC_SYSTEM_RESTORE, FALSE);

- EnableWindow (GetDlgItem (hwndDlg, IDC_SYSTEM_RESTORE), FALSE);

- }

-

- SetFocus (GetDlgItem (hwndDlg, IDC_UNINSTALL));

-

- return 1;

-

- case WM_SYSCOMMAND:

- if (lw == IDC_ABOUT)

- {

- DialogBoxW (hInst, MAKEINTRESOURCEW (IDD_ABOUT_DLG), hwndDlg, (DLGPROC) AboutDlgProc);

- return 1;

- }

- return 0;

-

- case WM_COMMAND:

- if (lw == IDC_UNINSTALL)

- {

- if (bDone)

- {

- bUninstallInProgress = FALSE;

- PostMessage (hwndDlg, WM_CLOSE, 0, 0);

- return 1;

- }

-

- bUninstallInProgress = TRUE;

-

- WaitCursor ();

-

- if (bUninstall)

- _beginthread (DoUninstall, 0, (void *) hwndDlg);

-

- return 1;

- }

-

- if (lw == IDC_SYSTEM_RESTORE)

- {

- bSystemRestore = IsButtonChecked (GetDlgItem (hwndDlg, IDC_SYSTEM_RESTORE));

- return 1;

- }

-

- if (lw == IDCANCEL)

- {

- PostMessage (hwndDlg, WM_CLOSE, 0, 0);

- return 1;

- }

-

- return 0;

-

- case TC_APPMSG_UNINSTALL_SUCCESS:

- SetWindowTextW (GetDlgItem ((HWND) hwndDlg, IDC_UNINSTALL), GetString ("FINALIZE"));

- NormalCursor ();

- return 1;

-

- case WM_CLOSE:

- if (bUninstallInProgress)

- {

- NormalCursor();

- if (AskNoYes("CONFIRM_EXIT_UNIVERSAL", hwndDlg) == IDNO)

- {

- return 1;

- }

- WaitCursor ();

- }

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- return 0;

-}

-

-

-int WINAPI wWinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance, wchar_t *lpszCommandLine, int nCmdShow)

-{

- atexit (localcleanup);

-

- SelfExtractStartupInit();

-

- lpszTitle = L"VeraCrypt Setup";

-

- /* Call InitApp to initialize the common code */

- InitApp (hInstance, NULL);

-

- if (IsAdmin () != TRUE)

- if (MessageBoxW (NULL, GetString ("SETUP_ADMIN"), lpszTitle, MB_YESNO | MB_ICONQUESTION) != IDYES)

- {

- FinalizeApp ();

- exit (1);

- }

-

- /* Setup directory */

- {

- wchar_t *s;

- GetModuleFileName (NULL, SetupFilesDir, ARRAYSIZE (SetupFilesDir));

- s = wcsrchr (SetupFilesDir, L'\\');

- if (s)

- s[1] = 0;

- }

-

- /* Parse command line arguments */

-

- if (lpszCommandLine[0] == L'/')

- {

- if (lpszCommandLine[1] == L'u')

- {

- // Uninstall: /u

-

- bUninstall = TRUE;

- }

- else if (lpszCommandLine[1] == L'c')

- {

- // Change: /c

-

- bChangeMode = TRUE;

- }

- else if (lpszCommandLine[1] == L'p')

- {

- // Create self-extracting package: /p

-

- bMakePackage = TRUE;

- }

- else if (lpszCommandLine[1] == L'd')

- {

- // Dev mode: /d

- bDevm = TRUE;

- }

- }

-

- if (bMakePackage)

- {

- /* Create self-extracting package */

-

- MakeSelfExtractingPackage (NULL, SetupFilesDir);

- }

- else

- {

- SetInstallationPath (NULL);

-

- if (!bUninstall)

- {

- if (IsSelfExtractingPackage())

- {

- if (!VerifyPackageIntegrity())

- {

- // Package corrupted

- exit (1);

- }

- bDevm = FALSE;

- }

- else if (!bDevm)

- {

- MessageBox (NULL, L"Error: This installer file does not contain any compressed files.\n\nTo create a self-extracting installation package (with embedded compressed files), run:\n\"VeraCrypt Setup.exe\" /p", L"VeraCrypt", MB_ICONERROR | MB_SETFOREGROUND | MB_TOPMOST);

- FinalizeApp ();

- exit (1);

- }

-

- if (bChangeMode)

- {

- /* VeraCrypt is already installed on this system and we were launched from the Program Files folder */

-

- char *tmpStr[] = {0, "SELECT_AN_ACTION", "REPAIR_REINSTALL", "UNINSTALL", "EXIT", 0};

-

- // Ask the user to select either Repair or Unistallation

- switch (AskMultiChoice ((void **) tmpStr, FALSE, NULL))

- {

- case 1:

- bRepairMode = TRUE;

- break;

- case 2:

- bUninstall = TRUE;

- break;

- default:

- FinalizeApp ();

- exit (1);

- }

- }

- }

-

- // System Restore

- if (IsSystemRestoreEnabled ())

- {

- wchar_t dllPath[MAX_PATH];

- if (GetSystemDirectory (dllPath, MAX_PATH))

- {

- StringCbCatW(dllPath, sizeof(dllPath), L"\\srclient.dll");

- }

- else

- StringCbCopyW(dllPath, sizeof(dllPath), L"C:\\Windows\\System32\\srclient.dll");

- SystemRestoreDll = LoadLibrary (dllPath);

- }

- else

- SystemRestoreDll = 0;

-

- if (!bUninstall)

- {

- /* Create the main dialog for install */

-

- DialogBoxParamW (hInstance, MAKEINTRESOURCEW (IDD_INSTL_DLG), NULL, (DLGPROC) MainDialogProc,

- (LPARAM)lpszCommandLine);

- }

- else

- {

- /* Create the main dialog for uninstall */

-

- DialogBoxW (hInstance, MAKEINTRESOURCEW (IDD_UNINSTALL), NULL, (DLGPROC) UninstallDlgProc);

-

- if (UninstallBatch[0])

- {

- STARTUPINFO si;

- PROCESS_INFORMATION pi;

-

- ZeroMemory (&si, sizeof (si));

- si.cb = sizeof (si);

- si.dwFlags = STARTF_USESHOWWINDOW;

- si.wShowWindow = SW_HIDE;

-

- if (!CreateProcess (UninstallBatch, NULL, NULL, NULL, FALSE, IDLE_PRIORITY_CLASS, NULL, NULL, &si, &pi))

- DeleteFile (UninstallBatch);

- else

- {

- CloseHandle (pi.hProcess);

- CloseHandle (pi.hThread);

- }

- }

- }

- }

- FinalizeApp ();

- return 0;

-}