-/*

- 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 "Platform/Finally.h"

-#include "Platform/ForEach.h"

-#include "BootEncryption.h"

-#include "Dlgcode.h"

-#include "Language.h"

-#include "Mount.h"

-#include "Common/Resource.h"

-#include "Resource.h"

-#include "Xml.h"

-#include "Favorites.h"

-

-using namespace std;

-

-namespace VeraCrypt

-{

- vector <FavoriteVolume> FavoriteVolumes;

- vector <FavoriteVolume> SystemFavoriteVolumes;

- list <FavoriteVolume> FavoritesOnArrivalMountRequired;

- list <FavoriteVolume> FavoritesMountedOnArrivalStillConnected;

- HMENU FavoriteVolumesMenu;

-

-

- BOOL AddMountedVolumeToFavorites (HWND hwndDlg, int driveNo, bool systemFavorites)

- {

- VOLUME_PROPERTIES_STRUCT prop;

- DWORD bytesReturned;

-

- memset (&prop, 0, sizeof (prop));

- prop.driveNo = driveNo;

-

- if (!DeviceIoControl (hDriver, TC_IOCTL_GET_VOLUME_PROPERTIES, &prop, sizeof (prop), &prop, sizeof (prop), &bytesReturned, NULL))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- return FALSE;

- }

-

- FavoriteVolume favorite;

- favorite.MountPoint = L"X:\\";

- favorite.MountPoint[0] = (wchar_t) (prop.driveNo + L'A');

-

- favorite.Path = prop.wszVolume;

- if (favorite.Path.find (L"\\??\\") == 0)

- favorite.Path = favorite.Path.substr (4);

-

- if (wcslen (prop.wszLabel))

- {

- favorite.Label = prop.wszLabel;

- favorite.UseLabelInExplorer = true;

- }

-

- if (IsVolumeDeviceHosted (favorite.Path.c_str()))

- {

- // Get GUID path

- wstring volumeDevPath = favorite.Path;

-

- wchar_t resolvedVolumeDevPath[TC_MAX_PATH];

- if (ResolveSymbolicLink (volumeDevPath.c_str(), resolvedVolumeDevPath, sizeof(resolvedVolumeDevPath)))

- volumeDevPath = resolvedVolumeDevPath;

-

- wchar_t volumeName[TC_MAX_PATH];

- HANDLE find = FindFirstVolume (volumeName, ARRAYSIZE (volumeName));

-

- if (find != INVALID_HANDLE_VALUE)

- {

- do

- {

- wchar_t findVolumeDevPath[TC_MAX_PATH];

- wstring vn = volumeName;

-

- if (QueryDosDevice (vn.substr (4, vn.size() - 5).c_str(), findVolumeDevPath, ARRAYSIZE (findVolumeDevPath)) != 0

- && volumeDevPath == findVolumeDevPath)

- {

- favorite.VolumePathId = volumeName;

- break;

- }

-

- } while (FindNextVolume (find, volumeName, ARRAYSIZE (volumeName)));

-

- FindVolumeClose (find);

- }

- }

-

- favorite.ReadOnly = prop.readOnly ? true : false;

- favorite.Removable = prop.removable ? true : false;

- favorite.SystemEncryption = prop.partitionInInactiveSysEncScope ? true : false;

- favorite.OpenExplorerWindow = (bExplore == TRUE);

- favorite.Pim = prop.volumePim;

- memcpy (favorite.VolumeID, prop.volumeID, VOLUME_ID_SIZE);

-

- if (favorite.VolumePathId.empty()

- && IsVolumeDeviceHosted (favorite.Path.c_str())

- && favorite.Path.find (L"\\\\?\\Volume{") != 0)

- {

- Warning (favorite.Path.find (L"\\Partition0") == wstring::npos ? "FAVORITE_ADD_PARTITION_TYPE_WARNING" : "FAVORITE_ADD_DRIVE_DEV_WARNING", hwndDlg);

- }

-

- return OrganizeFavoriteVolumes (hwndDlg, systemFavorites, favorite);

- }

-

-

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

- {

- /* This dialog is used both for System Favorites and non-system Favorites.

-

- The following options have different meaning in System Favorites mode:

-

- IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT => MOUNT_SYSTEM_FAVORITES_ON_BOOT

- IDC_FAVORITE_DISABLE_HOTKEY => DISABLE_NONADMIN_SYS_FAVORITES_ACCESS

-

- */

-

- WORD lw = LOWORD (wParam);

- WORD hw = HIWORD (wParam);

- static bool SystemFavoritesMode;

- static vector <FavoriteVolume> Favorites;

- static int SelectedItem;

- static HWND FavoriteListControl;

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- try

- {

- FavoriteListControl = GetDlgItem (hwndDlg, IDC_FAVORITE_VOLUMES_LIST);

-

- FavoriteVolumesDlgProcArguments *args = (FavoriteVolumesDlgProcArguments *) lParam;

- SystemFavoritesMode = args->SystemFavorites;

-

- LocalizeDialog (hwndDlg, SystemFavoritesMode ? "SYSTEM_FAVORITES_DLG_TITLE" : "IDD_FAVORITE_VOLUMES");

-

- if (SystemFavoritesMode)

- {

- RECT rec;

-

- BootEncryptionStatus bootEncStatus = BootEncryption (hwndDlg).GetStatus();

-

- if (!bootEncStatus.DriveMounted)

- throw ErrorException ("SYS_FAVORITES_REQUIRE_PBA", SRC_POS);

-

- ShowWindow (GetDlgItem(hwndDlg, IDC_FAVORITE_MOUNT_ON_LOGON), SW_HIDE);

- ShowWindow (GetDlgItem(hwndDlg, IDC_FAVORITE_MOUNT_ON_ARRIVAL), SW_HIDE);

-

- // MOUNT_SYSTEM_FAVORITES_ON_BOOT

-

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT), GetString ("MOUNT_SYSTEM_FAVORITES_ON_BOOT"));

-

- // DISABLE_NONADMIN_SYS_FAVORITES_ACCESS

-

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY), GetString ("DISABLE_NONADMIN_SYS_FAVORITES_ACCESS"));

-

- // Group box

-

- GetClientRect (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), &rec);

-

- SetWindowPos (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), 0, 0, 0,

- rec.right,

- rec.bottom - CompensateYDPI (95),

- SWP_NOMOVE | SWP_NOZORDER);

-

- InvalidateRect (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), NULL, TRUE);

- }

- else

- {

- ShowWindow (GetDlgItem(hwndDlg, IDC_FAV_VOL_OPTIONS_GLOBAL_SETTINGS_BOX), SW_HIDE);

- }

-

- Favorites.clear();

-

- LVCOLUMNW column;

- SendMessageW (FavoriteListControl, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, LVS_EX_FULLROWSELECT);

-

- memset (&column, 0, sizeof (column));

- column.mask = LVCF_TEXT|LVCF_WIDTH|LVCF_SUBITEM|LVCF_FMT;

- column.pszText = GetString ("DRIVE");

- column.cx = CompensateXDPI (38);

- column.fmt = LVCFMT_CENTER;

- SendMessageW (FavoriteListControl, LVM_INSERTCOLUMNW, 1, (LPARAM) &column);

-

- ++column.iSubItem;

- column.fmt = LVCFMT_LEFT;

- column.pszText = GetString ("LABEL");

- column.cx = CompensateXDPI (160);

- SendMessageW (FavoriteListControl, LVM_INSERTCOLUMNW, 2, (LPARAM) &column);

-

- ++column.iSubItem;

- column.fmt = LVCFMT_LEFT;

- column.pszText = GetString ("VOLUME");

- column.cx = CompensateXDPI (330);

- SendMessageW (FavoriteListControl, LVM_INSERTCOLUMNW, 3, (LPARAM) &column);

-

- SetControls (hwndDlg, FavoriteVolume(), SystemFavoritesMode, false);

-

- if (SystemFavoritesMode)

- LoadFavoriteVolumes (Favorites, true);

- else

- Favorites = FavoriteVolumes;

-

- if (args->AddFavoriteVolume)

- Favorites.push_back (args->NewFavoriteVolume);

-

- FillListControl (FavoriteListControl, Favorites);

-

- SelectedItem = -1;

-

- if (args->AddFavoriteVolume)

- {

- ListView_SetItemState (FavoriteListControl, Favorites.size() - 1, LVIS_SELECTED, LVIS_SELECTED);

- ListView_EnsureVisible (FavoriteListControl, Favorites.size() - 1, FALSE);

- }

-

- if (SystemFavoritesMode)

- SetDlgItemTextW (hwndDlg, IDC_FAVORITES_HELP_LINK, GetString ("SYS_FAVORITES_HELP_LINK"));

-

- ToHyperlink (hwndDlg, IDC_FAVORITES_HELP_LINK);

- }

- catch (Exception &e)

- {

- e.Show (hwndDlg);

- EndDialog (hwndDlg, IDCLOSE);

- }

- }

- return 1;

-

- case WM_COMMAND:

-

- switch (lw)

- {

- case IDOK:

-

- /* Global System Favorites settings */

-

- if (SystemFavoritesMode)

- {

- BootEncryption BootEncObj (NULL);

-

- if (BootEncObj.GetStatus().DriveMounted)

- {

- try

- {

- uint32 reqConfig = IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT) ? TC_DRIVER_CONFIG_CACHE_BOOT_PASSWORD_FOR_SYS_FAVORITES : 0;

- if (reqConfig != (ReadDriverConfigurationFlags() & TC_DRIVER_CONFIG_CACHE_BOOT_PASSWORD_FOR_SYS_FAVORITES))

- BootEncObj.RegisterSystemFavoritesService (reqConfig ? TRUE : FALSE);

-

- SetDriverConfigurationFlag (TC_DRIVER_CONFIG_DISABLE_NONADMIN_SYS_FAVORITES_ACCESS, IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY));

- }

- catch (Exception &e)

- {

- e.Show (hwndDlg);

- }

- }

- }

-

- /* (System) Favorites list */

-

- if (SelectedItem != -1 && !Favorites.empty())

- SetFavoriteVolume (hwndDlg, Favorites[SelectedItem], SystemFavoritesMode);

-

- if (SaveFavoriteVolumes (hwndDlg, Favorites, SystemFavoritesMode))

- {

- if (!SystemFavoritesMode)

- {

- bMountFavoritesOnLogon = FALSE;

-

- foreach (const FavoriteVolume &favorite, Favorites)

- {

- if (favorite.MountOnLogOn)

- {

- bMountFavoritesOnLogon = TRUE;

- break;

- }

- }

-

- if (!bEnableBkgTask || bCloseBkgTaskWhenNoVolumes || IsNonInstallMode())

- {

- foreach (const FavoriteVolume favorite, Favorites)

- {

- if (favorite.MountOnArrival)

- {

- Warning ("FAVORITE_ARRIVAL_MOUNT_BACKGROUND_TASK_ERR", hwndDlg);

- break;

- }

- }

- }

-

- FavoriteVolumes = Favorites;

-

- ManageStartupSeq();

- SaveSettings (hwndDlg);

- }

- else

- SystemFavoriteVolumes = Favorites;

-

- OnFavoriteVolumesUpdated();

- LoadDriveLetters (hwndDlg, GetDlgItem (MainDlg, IDC_DRIVELIST), 0);

-

- EndDialog (hwndDlg, IDOK);

- }

-

- return 1;

-

- case IDCANCEL:

- EndDialog (hwndDlg, IDCLOSE);

- return 1;

-

- case IDC_FAVORITE_MOVE_DOWN:

- if (SelectedItem != -1 && Favorites.size() > (size_t) SelectedItem + 1)

- {

- swap (Favorites[SelectedItem], Favorites[SelectedItem + 1]);

-

- FillListControl (FavoriteListControl, Favorites);

- ++SelectedItem;

- ListView_SetItemState (FavoriteListControl, SelectedItem, LVIS_SELECTED, LVIS_SELECTED);

- ListView_EnsureVisible (FavoriteListControl, SelectedItem, FALSE);

- }

- return 1;

-

- case IDC_FAVORITE_MOVE_UP:

- if (SelectedItem > 0)

- {

- swap (Favorites[SelectedItem], Favorites[SelectedItem - 1]);

-

- FillListControl (FavoriteListControl, Favorites);

- --SelectedItem;

- ListView_SetItemState (FavoriteListControl, SelectedItem, LVIS_SELECTED, LVIS_SELECTED);

- ListView_EnsureVisible (FavoriteListControl, SelectedItem, FALSE);

- }

- return 1;

-

- case IDC_FAVORITE_REMOVE:

- if (SelectedItem != -1)

- {

- Favorites.erase (Favorites.begin() + SelectedItem);

- FillListControl (GetDlgItem (hwndDlg, IDC_FAVORITE_VOLUMES_LIST), Favorites);

- SetControls (hwndDlg, FavoriteVolume(), SystemFavoritesMode, false);

- SelectedItem = -1;

- }

- return 1;

-

-

- case IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT: // Note that this option means "MOUNT_SYSTEM_FAVORITES_ON_BOOT" when SystemFavoritesMode is true

- if (SystemFavoritesMode)

- {

- // MOUNT_SYSTEM_FAVORITES_ON_BOOT

-

- if (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT))

- {

- WarningDirect ((wstring (GetString ("SYS_FAVORITES_KEYBOARD_WARNING")) + L"\n\n" + GetString ("BOOT_PASSWORD_CACHE_KEYBOARD_WARNING")).c_str(), hwndDlg);

-

- if (!IsServerOS() && !IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY))

- Info ("SYS_FAVORITES_ADMIN_ONLY_INFO", hwndDlg);

- }

- }

- return 1;

-

- case IDC_FAVORITE_DISABLE_HOTKEY: // Note that this option means "DISABLE_NONADMIN_SYS_FAVORITES_ACCESS" when SystemFavoritesMode is true

- if (SystemFavoritesMode)

- {

- // DISABLE_NONADMIN_SYS_FAVORITES_ACCESS

-

- if (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY))

- WarningDirect ((wstring (GetString ("SYS_FAVORITES_ADMIN_ONLY_WARNING")) + L"\n\n" + GetString ("SETTING_REQUIRES_REBOOT")).c_str(), hwndDlg);

- else

- Warning ("SETTING_REQUIRES_REBOOT", hwndDlg);

- }

- return 1;

-

- case IDC_FAVORITES_HELP_LINK:

- Applink (SystemFavoritesMode ? "sysfavorites" : "favorites", TRUE, "");

- return 1;

- case IDC_SHOW_PIM:

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PIM, IDC_PIM, 0);

- return 1;

-

- case IDC_PIM:

- if (hw == EN_CHANGE)

- {

- int pim = GetPim (hwndDlg, IDC_PIM);

- if (pim > (SystemFavoritesMode? MAX_BOOT_PIM_VALUE: MAX_PIM_VALUE))

- {

- SetDlgItemText (hwndDlg, IDC_PIM, L"");

- SetFocus (GetDlgItem(hwndDlg, IDC_PIM));

- Warning (SystemFavoritesMode? "PIM_SYSENC_TOO_BIG": "PIM_TOO_BIG", hwndDlg);

- return 1;

- }

- }

- break;

- }

-

- return 0;

-

- case WM_NOTIFY:

- if (((LPNMHDR) lParam)->code == LVN_ITEMCHANGED)

- {

- static bool reentry = false;

- if (reentry)

- break;

-

- reentry = true;

-

- if (SelectedItem != -1)

- {

- SetFavoriteVolume (hwndDlg, Favorites[SelectedItem], SystemFavoritesMode);

- FillListControlSubItems (FavoriteListControl, SelectedItem, Favorites[SelectedItem]);

- }

-

- SelectedItem = ListView_GetNextItem (GetDlgItem (hwndDlg, IDC_FAVORITE_VOLUMES_LIST), -1, LVIS_SELECTED);

-

- if (SelectedItem != -1)

- SetControls (hwndDlg, Favorites[SelectedItem], SystemFavoritesMode);

- else

- SetControls (hwndDlg, FavoriteVolume(), SystemFavoritesMode, false);

-

- reentry = false;

- return 1;

- }

- break;

-

- case WM_CLOSE:

- EndDialog (hwndDlg, IDCLOSE);

- return 1;

- case WM_CTLCOLORSTATIC:

- {

- HDC hdc = (HDC) wParam;

- HWND hw = (HWND) lParam;

- if (hw == GetDlgItem(hwndDlg, IDC_FAVORITE_VOLUME_ID))

- {

- // This the favorite ID field. Make its background like normal edit

- HBRUSH hbr = GetSysColorBrush (COLOR_WINDOW);

- ::SelectObject(hdc, hbr);

- return (BOOL) hbr;

- }

- }

- break;

- }

-

- return 0;

- }

-

-

- static void FillFavoriteVolumesMenu ()

- {

- while (DeleteMenu (FavoriteVolumesMenu, 7, MF_BYPOSITION)) { }

-

- if (FavoriteVolumes.empty())

- return;

-

- AppendMenu (FavoriteVolumesMenu, MF_SEPARATOR, 0, L"");

-

- int i = 0;

- foreach (const FavoriteVolume &favorite, FavoriteVolumes)

- {

- UINT flags = MF_STRING;

-

- if (favorite.DisconnectedDevice)

- flags |= MF_GRAYED;

-

- wstring menuText = favorite.Path;

- if (favorite.DisconnectedDevice)

- menuText = favorite.Label.empty() ? wstring (L"(") + GetString ("FAVORITE_DISCONNECTED_DEV") + L")" : L"";

-

- if (!favorite.Label.empty())

- {

- if (favorite.DisconnectedDevice)

- menuText = favorite.Label + L" " + menuText;

- else

- menuText = favorite.Label;

- }

-

- AppendMenuW (FavoriteVolumesMenu, flags, TC_FAVORITE_MENU_CMD_ID_OFFSET + i++,

- (menuText + L"\t" + favorite.MountPoint.substr (0, 2)).c_str());

- }

- }

-

-

- static void FillListControl (HWND favoriteListControl, vector <FavoriteVolume> &favorites)

- {

- SendMessage (favoriteListControl, LVM_DELETEALLITEMS, 0, 0);

-

- int line = 0;

- foreach (const FavoriteVolume favorite, favorites)

- {

- ListItemAdd (favoriteListControl, line, (wchar_t *) favorite.MountPoint.substr (0, 2).c_str());

- FillListControlSubItems (favoriteListControl, line++, favorite);

- }

- }

-

-

- static void FillListControlSubItems (HWND FavoriteListControl, int line, const FavoriteVolume &favorite)

- {

- ListSubItemSet (FavoriteListControl, line, 1, (wchar_t *) favorite.Label.c_str());

-

- if (favorite.DisconnectedDevice)

- ListSubItemSet (FavoriteListControl, line, 2, (wchar_t *) (wstring (L"(") + GetString ("FAVORITE_DISCONNECTED_DEV") + L")").c_str());

- else

- ListSubItemSet (FavoriteListControl, line, 2, (wchar_t *) favorite.Path.c_str());

- }

-

-

- wstring GetFavoriteVolumeLabel (const wstring &volumePath, bool& useInExplorer)

- {

- foreach (const FavoriteVolume &favorite, FavoriteVolumes)

- {

- if (favorite.Path == volumePath)

- {

- useInExplorer = favorite.UseLabelInExplorer;

- return favorite.Label;

- }

- }

-

- foreach (const FavoriteVolume &favorite, SystemFavoriteVolumes)

- {

- if (favorite.Path == volumePath)

- {

- useInExplorer = favorite.UseLabelInExplorer;

- return favorite.Label;

- }

- }

-

- useInExplorer = false;

- return wstring();

- }

-

-

- void LoadFavoriteVolumes ()

- {

- LoadFavoriteVolumes (FavoriteVolumes, false);

-

- try

- {

- LoadFavoriteVolumes (SystemFavoriteVolumes, true, true);

- }

- catch (...) { } // Ignore errors as SystemFavoriteVolumes list is used only for resolving volume paths to labels

-

- OnFavoriteVolumesUpdated();

- }

-

-

- void LoadFavoriteVolumes (vector <FavoriteVolume> &favorites, bool systemFavorites, bool noUacElevation)

- {

- favorites.clear();

- wstring favoritesFilePath = systemFavorites ? GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false) : GetConfigPath (TC_APPD_FILENAME_FAVORITE_VOLUMES);

-

- if (systemFavorites && !IsAdmin() && !noUacElevation)

- {

- favoritesFilePath = GetConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES);

-

- try

- {

- BootEncryption bootEnc (MainDlg);

- bootEnc.CopyFileAdmin (GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false).c_str(), favoritesFilePath.c_str());

- }

- catch (SystemException &e)

- {

- if (e.ErrorCode == ERROR_FILE_NOT_FOUND)

- return;

-

- throw;

- }

- }

-

- DWORD size;

- char *favoritesXml = LoadFile (favoritesFilePath.c_str(), &size);

-

- if (systemFavorites && !IsAdmin() && !noUacElevation)

- DeleteFile (GetConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES));

-

- char *xml = favoritesXml;

- char mountPoint[MAX_PATH], volume[MAX_PATH];

-

- if (xml == NULL)

- return;

-

- while (xml = XmlFindElement (xml, "volume"))

- {

- FavoriteVolume favorite;

-

- XmlGetAttributeText (xml, "mountpoint", mountPoint, sizeof (mountPoint));

- favorite.MountPoint = Utf8StringToWide (mountPoint);

-

- XmlGetNodeText (xml, volume, sizeof (volume));

- favorite.Path = Utf8StringToWide (volume);

-

- char label[1024];

-

- XmlGetAttributeText (xml, "ID", label, sizeof (label));

- if (strlen (label) == (2*VOLUME_ID_SIZE))

- {

- std::vector<byte> arr;

- if (HexWideStringToArray (Utf8StringToWide (label).c_str(), arr) && arr.size() == VOLUME_ID_SIZE)

- {

- memcpy (favorite.VolumeID, &arr[0], VOLUME_ID_SIZE);

- }

- }

-

- XmlGetAttributeText (xml, "label", label, sizeof (label));

- favorite.Label = Utf8StringToWide (label);

-

- XmlGetAttributeText (xml, "pim", label, sizeof (label));

- if (strlen(label) == 0)

- {

- /* support old attribute name before it was changed to PIM*/

- XmlGetAttributeText (xml, "pin", label, sizeof (label));

- }

- favorite.Pim = strtol (label, NULL, 10);

- if (favorite.Pim < 0 || favorite.Pim > (systemFavorites? MAX_BOOT_PIM_VALUE : MAX_PIM_VALUE))

- favorite.Pim = 0;

-

- char boolVal[2];

- XmlGetAttributeText (xml, "readonly", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.ReadOnly = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "removable", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.Removable = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "system", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.SystemEncryption = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "noHotKeyMount", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.DisableHotkeyMount = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "openExplorerWindow", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.OpenExplorerWindow = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "mountOnArrival", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.MountOnArrival = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "mountOnLogOn", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.MountOnLogOn = (boolVal[0] == '1');

-

- XmlGetAttributeText (xml, "useLabelInExplorer", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.UseLabelInExplorer = (boolVal[0] == '1') && !favorite.ReadOnly;

-

- XmlGetAttributeText (xml, "useVolumeID", boolVal, sizeof (boolVal));

- if (boolVal[0])

- favorite.UseVolumeID = (boolVal[0] == '1') && !IsRepeatedByteArray (0, favorite.VolumeID, sizeof (favorite.VolumeID));

-

- if (favorite.Path.find (L"\\\\?\\Volume{") == 0 && favorite.Path.rfind (L"}\\") == favorite.Path.size() - 2)

- {

- wstring resolvedPath = VolumeGuidPathToDevicePath (favorite.Path);

- if (!resolvedPath.empty())

- {

- favorite.DisconnectedDevice = false;

- favorite.VolumePathId = favorite.Path;

- favorite.Path = resolvedPath;

- }

- else

- favorite.DisconnectedDevice = true;

- }

-

- favorites.push_back (favorite);

- xml++;

- }

-

- free (favoritesXml);

- }

-

-

- static void OnFavoriteVolumesUpdated ()

- {

- FillFavoriteVolumesMenu();

-

- FavoritesOnArrivalMountRequired.clear();

-

- foreach (const FavoriteVolume favorite, FavoriteVolumes)

- {

- if (favorite.MountOnArrival)

- {

- FavoritesOnArrivalMountRequired.push_back (favorite);

-

- if (IsMountedVolume (favorite.Path.c_str()))

- {

- bool present = false;

-

- foreach (const FavoriteVolume favoriteConnected, FavoritesMountedOnArrivalStillConnected)

- {

- if (favorite.Path == favoriteConnected.Path)

- {

- present = true;

- break;

- }

- }

-

- if (!present)

- FavoritesMountedOnArrivalStillConnected.push_back (favorite);

- }

- }

- }

- }

-

-

- BOOL OrganizeFavoriteVolumes (HWND hwndDlg, bool systemFavorites, const FavoriteVolume &newFavorite)

- {

- FavoriteVolumesDlgProcArguments args;

- args.SystemFavorites = systemFavorites;

-

- if (!newFavorite.Path.empty())

- {

- args.AddFavoriteVolume = true;

- args.NewFavoriteVolume = newFavorite;

- }

- else

- args.AddFavoriteVolume = false;

-

- return DialogBoxParamW (hInst, MAKEINTRESOURCEW (IDD_FAVORITE_VOLUMES), hwndDlg, (DLGPROC) FavoriteVolumesDlgProc, (LPARAM) &args) == IDOK;

- }

-

-

- bool SaveFavoriteVolumes (HWND hwndDlg, const vector <FavoriteVolume> &favorites, bool systemFavorites)

- {

- FILE *f;

- int cnt = 0;

-

- f = _wfopen (GetConfigPath (systemFavorites ? TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES : TC_APPD_FILENAME_FAVORITE_VOLUMES), L"w,ccs=UTF-8");

- if (f == NULL)

- {

- handleWin32Error (MainDlg, SRC_POS);

- return false;

- }

-

- XmlWriteHeader (f);

- fputws (L"\n\t<favorites>", f);

-

- foreach (const FavoriteVolume &favorite, favorites)

- {

- wchar_t tq[2048];

-

- if (systemFavorites && favorite.Path.find (L"\\\\") == 0 && favorite.Path.find (L"Volume{") == wstring::npos)

- Warning ("SYSTEM_FAVORITE_NETWORK_PATH_ERR", hwndDlg);

-

- XmlQuoteTextW (!favorite.VolumePathId.empty() ? favorite.VolumePathId.c_str() : favorite.Path.c_str(), tq, ARRAYSIZE (tq));

-

- wstring s = L"\n\t\t<volume mountpoint=\"" + favorite.MountPoint + L"\"";

-

- if (!IsRepeatedByteArray (0, favorite.VolumeID, sizeof (favorite.VolumeID)))

- s += L" ID=\"" + ArrayToHexWideString (favorite.VolumeID, sizeof (favorite.VolumeID)) + L"\"";

-

- if (!favorite.Label.empty())

- s += L" label=\"" + favorite.Label + L"\"";

-

- if (favorite.Pim > 0)

- s += L" pim=\"" + IntToWideString(favorite.Pim) + L"\"";

-

- if (favorite.ReadOnly)

- s += L" readonly=\"1\"";

-

- if (favorite.Removable)

- s += L" removable=\"1\"";

-

- if (favorite.SystemEncryption)

- s += L" system=\"1\"";

-

- if (favorite.MountOnArrival)

- s += L" mountOnArrival=\"1\"";

-

- if (favorite.MountOnLogOn)

- s += L" mountOnLogOn=\"1\"";

-

- if (favorite.DisableHotkeyMount)

- s += L" noHotKeyMount=\"1\"";

-

- if (favorite.OpenExplorerWindow)

- s += L" openExplorerWindow=\"1\"";

-

- if (favorite.UseLabelInExplorer && !favorite.ReadOnly)

- s += L" useLabelInExplorer=\"1\"";

-

- if (favorite.UseVolumeID && !IsRepeatedByteArray (0, favorite.VolumeID, sizeof (favorite.VolumeID)))

- s += L" useVolumeID=\"1\"";

-

- s += L">" + wstring (tq) + L"</volume>";

-

- fwprintf (f, L"%ws", s.c_str());

- cnt++;

- }

-

- fputws (L"\n\t</favorites>", f);

- XmlWriteFooter (f);

-

- if (!CheckFileStreamWriteErrors (hwndDlg, f, systemFavorites ? TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES : TC_APPD_FILENAME_FAVORITE_VOLUMES))

- {

- fclose (f);

- return false;

- }

-

- fclose (f);

-

- BootEncryption bootEnc (MainDlg);

-

- if (systemFavorites)

- {

- finally_do ({ _wremove (GetConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES)); });

-

- try

- {

- bootEnc.DeleteFileAdmin (GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false).c_str());

- }

- catch (UserAbort&) { return false; }

- catch (...) { }

-

- try

- {

- if (cnt != 0)

- {

- bootEnc.CopyFileAdmin (GetConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES), GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false).c_str());

-

- if (!(ReadDriverConfigurationFlags() & TC_DRIVER_CONFIG_CACHE_BOOT_PASSWORD_FOR_SYS_FAVORITES))

- Info ("SYS_FAVORITE_VOLUMES_SAVED", hwndDlg);

- }

- }

- catch (Exception &e)

- {

- e.Show (NULL);

- }

- }

-

- if (cnt == 0)

- {

- if (systemFavorites)

- {

- try

- {

- bootEnc.DeleteFileAdmin (GetServiceConfigPath (TC_APPD_FILENAME_SYSTEM_FAVORITE_VOLUMES, false).c_str());

- }

- catch (...) { }

- }

- else

- _wremove (GetConfigPath (TC_APPD_FILENAME_FAVORITE_VOLUMES));

- }

-

- return true;

- }

-

-

- static void SetControls (HWND hwndDlg, const FavoriteVolume &favorite, bool systemFavoritesMode, bool enable)

- {

- BOOL bIsDevice = favorite.DisconnectedDevice || IsVolumeDeviceHosted (favorite.Path.c_str()) || !enable;

- if (favorite.Pim > 0)

- {

- wchar_t szTmp[MAX_PIM + 1];

- StringCbPrintfW (szTmp, sizeof(szTmp), L"%d", favorite.Pim);

- SetDlgItemText (hwndDlg, IDC_PIM, szTmp);

- }

- else

- SetDlgItemText (hwndDlg, IDC_PIM, L"");

- SetDlgItemTextW (hwndDlg, IDC_FAVORITE_LABEL, favorite.Label.c_str());

- SetCheckBox (hwndDlg, IDC_FAVORITE_USE_LABEL_IN_EXPLORER, favorite.UseLabelInExplorer);

- SetCheckBox (hwndDlg, IDC_FAVORITE_MOUNT_ON_LOGON, favorite.MountOnLogOn);

- SetCheckBox (hwndDlg, IDC_FAVORITE_MOUNT_ON_ARRIVAL, favorite.MountOnArrival);

- SetCheckBox (hwndDlg, IDC_FAVORITE_MOUNT_READONLY, favorite.ReadOnly);

- SetCheckBox (hwndDlg, IDC_FAVORITE_MOUNT_REMOVABLE, favorite.Removable);

- SetCheckBox (hwndDlg, IDC_FAVORITE_USE_VOLUME_ID, favorite.UseVolumeID && bIsDevice);

-

- if (IsRepeatedByteArray (0, favorite.VolumeID, sizeof (favorite.VolumeID)) || !bIsDevice)

- {

- SetDlgItemText (hwndDlg, IDC_FAVORITE_VOLUME_ID, L"");

- }

- else

- SetDlgItemText (hwndDlg, IDC_FAVORITE_VOLUME_ID, ArrayToHexWideString (favorite.VolumeID, sizeof (favorite.VolumeID)).c_str());

-

- if (systemFavoritesMode)

- {

- uint32 driverConfig = ReadDriverConfigurationFlags();

-

- // MOUNT_SYSTEM_FAVORITES_ON_BOOT

- CheckDlgButton (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT, (driverConfig & TC_DRIVER_CONFIG_CACHE_BOOT_PASSWORD_FOR_SYS_FAVORITES) ? BST_CHECKED : BST_UNCHECKED);

-

- // DISABLE_NONADMIN_SYS_FAVORITES_ACCESS

- CheckDlgButton (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY, (driverConfig & TC_DRIVER_CONFIG_DISABLE_NONADMIN_SYS_FAVORITES_ACCESS) ? BST_CHECKED : BST_UNCHECKED);

- }

- else

- {

- SetCheckBox (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT, favorite.OpenExplorerWindow);

- SetCheckBox (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY, favorite.DisableHotkeyMount);

- }

-

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOVE_UP), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOVE_DOWN), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_REMOVE), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDT_PIM), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PIM), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_SHOW_PIM), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PIM_HELP), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDT_FAVORITE_LABEL), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_LABEL), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_USE_LABEL_IN_EXPLORER), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOUNT_ON_LOGON), enable && !systemFavoritesMode);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOUNT_ON_ARRIVAL), enable && !systemFavoritesMode);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOUNT_READONLY), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_MOUNT_REMOVABLE), enable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT), enable || systemFavoritesMode);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY), enable || systemFavoritesMode);

- EnableWindow (GetDlgItem (hwndDlg, IDT_VOLUME_ID), enable && bIsDevice);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_VOLUME_ID), enable && bIsDevice);

- EnableWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_USE_VOLUME_ID), enable && bIsDevice && !IsRepeatedByteArray (0, favorite.VolumeID, sizeof (favorite.VolumeID)));

-

- ShowWindow (GetDlgItem (hwndDlg, IDT_VOLUME_ID), bIsDevice? SW_SHOW : SW_HIDE);

- ShowWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_VOLUME_ID), bIsDevice? SW_SHOW : SW_HIDE);

- ShowWindow (GetDlgItem (hwndDlg, IDC_FAVORITE_USE_VOLUME_ID), bIsDevice? SW_SHOW : SW_HIDE);

-

- // Group box

- RECT boxRect, checkRect, labelRect;

-

- GetWindowRect (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), &boxRect);

- GetWindowRect (GetDlgItem (hwndDlg, IDC_FAVORITE_USE_VOLUME_ID), &checkRect);

- GetWindowRect (GetDlgItem (hwndDlg, IDT_VOLUME_ID), &labelRect);

-

- if (!bIsDevice && (boxRect.top < checkRect.top))

- {

- POINT pt = {boxRect.left, checkRect.bottom};

- ScreenToClient (hwndDlg, &pt);

- SetWindowPos (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), 0, pt.x, pt.y,

- boxRect.right - boxRect.left,

- boxRect.bottom - checkRect.bottom,

- SWP_NOZORDER);

-

- InvalidateRect (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), NULL, TRUE);

- }

-

- if (bIsDevice && (boxRect.top >= checkRect.top))

- {

- POINT pt = {boxRect.left, labelRect.top - CompensateYDPI (10)};

- ScreenToClient (hwndDlg, &pt);

- SetWindowPos (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), 0, pt.x, pt.y,

- boxRect.right - boxRect.left,

- boxRect.bottom - labelRect.top + CompensateYDPI (10),

- SWP_NOZORDER);

-

- InvalidateRect (GetDlgItem (hwndDlg, IDC_FAV_VOL_OPTIONS_GROUP_BOX), NULL, TRUE);

- }

- }

-

-

- static void SetFavoriteVolume (HWND hwndDlg, FavoriteVolume &favorite, bool systemFavoritesMode)

- {

- wchar_t label[1024];

- if (GetDlgItemTextW (hwndDlg, IDC_FAVORITE_LABEL, label, ARRAYSIZE (label)) != 0)

- {

- favorite.Label = label;

-

- for (size_t i = 0; i < favorite.Label.size(); ++i)

- {

- if (favorite.Label[i] == L'"')

- favorite.Label.at (i) = L'\'';

- }

- }

- else

- favorite.Label.clear();

-

- favorite.Pim = GetPim (hwndDlg, IDC_PIM);

- favorite.UseLabelInExplorer = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_USE_LABEL_IN_EXPLORER) != 0);

- favorite.UseVolumeID = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_USE_VOLUME_ID) != 0);

-

- favorite.ReadOnly = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_MOUNT_READONLY) != 0);

- favorite.Removable = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_MOUNT_REMOVABLE) != 0);

-

- if (!systemFavoritesMode)

- {

- favorite.MountOnLogOn = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_MOUNT_ON_LOGON) != 0);

- favorite.MountOnArrival = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_MOUNT_ON_ARRIVAL) != 0);

- favorite.DisableHotkeyMount = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_DISABLE_HOTKEY) != 0);

- favorite.OpenExplorerWindow = (IsDlgButtonChecked (hwndDlg, IDC_FAVORITE_OPEN_EXPLORER_WIN_ON_MOUNT) != 0);

- }

-

- if (favorite.VolumePathId.empty()

- && IsVolumeDeviceHosted (favorite.Path.c_str())

- && favorite.Path.find (L"\\\\?\\Volume{") != 0)

- {

- bool partition = (favorite.Path.find (L"\\Partition0") == wstring::npos);

-

- if (!favorite.Label.empty())

- {

- ErrorDirect ((GetString (partition ? "FAVORITE_LABEL_PARTITION_TYPE_ERR" : "FAVORITE_LABEL_DEVICE_PATH_ERR") + wstring (L"\n\n") + favorite.Path).c_str(), hwndDlg);

- favorite.Label.clear();

- }

-

- if (favorite.MountOnArrival)

- {

- ErrorDirect ((GetString (partition ? "FAVORITE_ARRIVAL_MOUNT_PARTITION_TYPE_ERR" : "FAVORITE_ARRIVAL_MOUNT_DEVICE_PATH_ERR") + wstring (L"\n\n") + favorite.Path).c_str(), hwndDlg);

- favorite.MountOnArrival = false;

- }

- }

-

- if (favorite.MountOnArrival && favorite.Path.find (L"\\\\") == 0 && favorite.Path.find (L"Volume{") == wstring::npos)

- {

- Error ("FAVORITE_ARRIVAL_MOUNT_NETWORK_PATH_ERR", hwndDlg);

- favorite.MountOnArrival = false;

- }

- }

-

-

- void UpdateDeviceHostedFavoriteVolumes ()

- {

- try

- {

- LoadFavoriteVolumes();

- }

- catch (Exception &e)

- {

- e.Show (MainDlg);

- }

- }

-}

-/*

- 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.

-*/

-

-#ifndef TC_HEADER_Mount_FavoriteVolumes

-#define TC_HEADER_Mount_FavoriteVolumes

-

-#include <Tcdefs.h>

-

-namespace VeraCrypt

-{

- struct FavoriteVolume

- {

- FavoriteVolume()

- :

- Pim (0),

- DisableHotkeyMount (false),

- DisconnectedDevice (false),

- MountOnLogOn (false),

- MountOnArrival (false),

- OpenExplorerWindow (false),

- ReadOnly (false),

- Removable (false),

- SystemEncryption (false),

- UseLabelInExplorer (false),

- UseVolumeID (false)

- {

- memset (VolumeID, 0, VOLUME_ID_SIZE);

- }

-

- wstring Path;

- wstring MountPoint;

- wstring VolumePathId;

- wstring Label;

- int Pim;

- BYTE VolumeID[VOLUME_ID_SIZE];

-

- bool DisableHotkeyMount;

- bool DisconnectedDevice;

- bool MountOnLogOn;

- bool MountOnArrival;

- bool OpenExplorerWindow;

- bool ReadOnly;

- bool Removable;

- bool SystemEncryption;

- bool UseLabelInExplorer;

- bool UseVolumeID;

- };

-

- struct FavoriteVolumesDlgProcArguments

- {

- bool SystemFavorites;

- bool AddFavoriteVolume;

- FavoriteVolume NewFavoriteVolume;

- };

-

- extern vector <FavoriteVolume> FavoriteVolumes;

- extern vector <FavoriteVolume> SystemFavoriteVolumes;

- extern list <FavoriteVolume> FavoritesOnArrivalMountRequired;

- extern list <FavoriteVolume> FavoritesMountedOnArrivalStillConnected;

- extern HMENU FavoriteVolumesMenu;

-

- BOOL AddMountedVolumeToFavorites (HWND hwndDlg, int driveNo, bool systemFavorites);

- static BOOL CALLBACK FavoriteVolumesDlgProc (HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam);

- static void FillFavoriteVolumesMenu ();

- static void FillListControl (HWND favoriteListControl, vector <FavoriteVolume> &favorites);

- static void FillListControlSubItems (HWND favoriteListControl, int line, const FavoriteVolume &favorite);

- wstring GetFavoriteVolumeLabel (const wstring &volumePath, bool& useInExplorer);

- void LoadFavoriteVolumes ();

- void LoadFavoriteVolumes (vector <FavoriteVolume> &favorites, bool systemFavorites, bool noUacElevation = false);

- static void OnFavoriteVolumesUpdated ();

- BOOL OrganizeFavoriteVolumes (HWND hwndDlg, bool systemFavorites, const FavoriteVolume &newFavorite = FavoriteVolume());

- bool SaveFavoriteVolumes (HWND hwndDlg, const vector <FavoriteVolume> &favorites, bool systemFavorites);

- static void SetControls (HWND hwndDlg, const FavoriteVolume &favorite, bool systemFavoritesMode, bool enable = true);

- static void SetFavoriteVolume (HWND hwndDlg, FavoriteVolume &favorite, bool systemFavoritesMode);

- void UpdateDeviceHostedFavoriteVolumes ();

-}

-

-#endif // TC_HEADER_Mount_FavoriteVolumes

-/*

- 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 <windows.h>

-#include "Dlgcode.h"

-#include "Hotkeys.h"

-#include "Language.h"

-#include "Mount.h"

-#include "Resource.h"

-

-#include <Strsafe.h>

-

-#ifndef SRC_POS

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

-#endif

-

-#define MAX_KEY_COMB_NAME_LEN 260

-

-TCHOTKEY Hotkeys [NBR_HOTKEYS];

-static TCHOTKEY tmpHotkeys [NBR_HOTKEYS];

-

-static int nSelectedHotkeyId;

-static UINT currentVKeyCode;

-static BYTE vkeysDown[256];

-

-

-static void ScanAndProcessKey (UINT *vKeyCode, wchar_t *keyName)

-{

- UINT vKey;

- *vKeyCode = 0;

-

- for (vKey = 0; vKey <= 0xFF; vKey++)

- {

- if (GetAsyncKeyState (vKey) < 0)

- {

- if (!vkeysDown [vKey])

- {

- vkeysDown [vKey] = 1;

- if (GetKeyName (vKey, keyName)) // If the key is allowed and its name has been resolved

- *vKeyCode = vKey;

- }

- }

- else

- vkeysDown [vKey] = 0;

- }

-}

-

-

-/* Returns TRUE if the key is allowed and its name is resolved. */

-BOOL GetKeyName (UINT vKey, wchar_t *keyName)

-{

- BOOL result = TRUE;

-

- if (vKey >= 0x30 && vKey <= 0x5a)

- {

- // ASCII characters

- StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%hc", (char) vKey);

- }

- else if (vKey >= 0xE9 && vKey <= 0xF5)

- {

- // OEM-specific

- StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM-%d", vKey);

-

- // mapping taken from:

- // http://www.hotkeynet.com/ref/keynames.html

- // https://mojoware.googlecode.com/svn-history/r3/trunk/mojo_engine/cKeyboard.cpp

- // http://www.screenio.com/gui_screenio/gs_htmlhelp_subweb/download/SIMKEYS.cob

- //

- // These values seem to come from Nokia/Ericsson mobile device keys

-

- switch (vKey)

- {

- case 0xE9: // OEMReset = 0xE9

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMReset)");

- break;

- case 0xEA: // OEMJump = 0xEA

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMJump)");

- break;

- case 0xEB: // OEMPA1 = 0xEB

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMPA1)");

- break;

- case 0xEC: // OEMPA2 = 0xEC

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMPA2)");

- break;

- case 0xED: // OEMPA3 = 0xED

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMPA3)");

- break;

- case 0xEE: // OEMWSCtrl = 0xEE

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMWSCtrl)");

- break;

- case 0xEF: // OEMCUSel = 0xEF

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMCUSel)");

- break;

- case 0xF0: // OEMATTN = 0xF0

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMATTN)");

- break;

- case 0xF1: // OEMFinish = 0xF1

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMFinish)");

- break;

- case 0xF2: // OEMCopy = 0xF2

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMCopy)");

- break;

- case 0xF3: // OEMAuto = 0xF3

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMAuto)");

- break;

- case 0xF4: // OEMENLW = 0xF4

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMENLW)");

- break;

- case 0xF5: // OEMBackTab = 0xF5

- StringCbCatW (keyName, MAX_KEY_COMB_NAME_LEN, L" (OEMBackTab)");

- break;

- }

- }

- else if (vKey >= VK_F1 && vKey <= VK_F24)

- {

- // F1-F24

- StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"F%d", vKey - VK_F1 + 1);

- }

- else if (vKey >= VK_NUMPAD0 && vKey <= VK_NUMPAD9)

- {

- // Numpad numbers

- StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s %d", GetString ("VK_NUMPAD"), vKey - VK_NUMPAD0);

- }

- else

- {

- switch (vKey)

- {

- case VK_MULTIPLY: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s *", GetString ("VK_NUMPAD")); break;

- case VK_ADD: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s +", GetString ("VK_NUMPAD")); break;

- case VK_SEPARATOR: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s Separator", GetString ("VK_NUMPAD")); break;

- case VK_SUBTRACT: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s -", GetString ("VK_NUMPAD")); break;

- case VK_DECIMAL: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s .", GetString ("VK_NUMPAD")); break;

- case VK_DIVIDE: StringCbPrintfW (keyName, MAX_KEY_COMB_NAME_LEN, L"%s /", GetString ("VK_NUMPAD")); break;

- case VK_OEM_1: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 1 (';')"); break;

- case VK_OEM_PLUS: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"+"); break;

- case VK_OEM_COMMA: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L","); break;

- case VK_OEM_MINUS: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"-"); break;

- case VK_OEM_PERIOD: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"."); break;

- case VK_OEM_2: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 2 ('/')"); break;

- case VK_OEM_3: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 3 (`)"); break;

- case VK_OEM_4: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 4 ('[')"); break;

- case VK_OEM_5: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 5 ('\\')"); break;

- case VK_OEM_6: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 6 (']')"); break;

- case VK_OEM_7: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 7 (')"); break;

- case VK_OEM_8: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 8"); break;

- case VK_OEM_AX: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM AX"); break;

- case VK_OEM_102: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM 102"); break;

- case VK_ICO_HELP: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"ICO_HELP"); break;

- case VK_ICO_00: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"ICO_00"); break;

- case VK_ICO_CLEAR: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"ICO_CLEAR"); break;

- case VK_ATTN: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"Attn"); break;

- case VK_CRSEL: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"CrSel"); break;

- case VK_EXSEL: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"ExSel"); break;

- case VK_EREOF: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"Erase EOF"); break;

- case VK_PA1: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"PA1"); break;

- case VK_OEM_CLEAR: StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, L"OEM Clear"); break;

-

- case 0:

- case 1:

- case 0xFF:

- result = FALSE;

- break;

-

- default:

- {

- char key[16];

- wchar_t *desc;

- StringCbPrintfA (key, sizeof(key),"VKEY_%02X", vKey);

- desc = GetString (key);

- if (desc == UnknownString)

- result = FALSE;

- else

- StringCbCopyW (keyName, MAX_KEY_COMB_NAME_LEN, desc);

- }

- }

- }

- return result;

-}

-

-

-static BOOL ShortcutInUse (UINT vKeyCode, UINT modifiers, TCHOTKEY hotkeys[])

-{

- int i;

-

- for (i = 0; i < NBR_HOTKEYS; i++)

- {

- if (hotkeys[i].vKeyCode == vKeyCode && hotkeys[i].vKeyModifiers == modifiers)

- return TRUE;

- }

- return FALSE;

-}

-

-

-void UnregisterAllHotkeys (HWND hwndDlg, TCHOTKEY hotkeys[])

-{

- int i;

-

- for (i = 0; i < NBR_HOTKEYS; i++)

- {

- if (hotkeys[i].vKeyCode != 0)

- UnregisterHotKey (hwndDlg, i);

-

- }

-}

-

-

-BOOL RegisterAllHotkeys (HWND hwndDlg, TCHOTKEY hotkeys[])

-{

- BOOL result = TRUE;

- int i;

-

- for (i = 0; i < NBR_HOTKEYS; i++)

- {

- if (hotkeys[i].vKeyCode != 0

- && !RegisterHotKey (hwndDlg, i, hotkeys[i].vKeyModifiers, hotkeys[i].vKeyCode))

- result = FALSE;

- }

-

- return result;

-}

-

-

-static void DisplayHotkeyList (HWND hwndDlg)

-{

- LVITEMW item;

- HWND hList = GetDlgItem (hwndDlg, IDC_HOTKEY_LIST);

- int i;

- wchar_t ShortcutMod [MAX_KEY_COMB_NAME_LEN];

- wchar_t ShortcutFinal [MAX_KEY_COMB_NAME_LEN*2];

- wchar_t Shortcut [MAX_KEY_COMB_NAME_LEN];

-

- SendMessage (hList, LVM_DELETEALLITEMS,0, (LPARAM)&item);

-

- for (i = 0; i < NBR_HOTKEYS; i++)

- {

- memset (&item,0,sizeof(item));

- item.mask = LVIF_TEXT;

- item.iItem = i;

- item.iSubItem = 0;

-

- switch (i)

- {

-

- case HK_AUTOMOUNT_DEVICES:

- item.pszText = GetString ("HK_AUTOMOUNT_DEVICES");

- break;

-

- case HK_DISMOUNT_ALL:

- item.pszText = GetString ("HK_DISMOUNT_ALL");

- break;

-

- case HK_WIPE_CACHE:

- item.pszText = GetString ("HK_WIPE_CACHE");

- break;

-

- case HK_DISMOUNT_ALL_AND_WIPE:

- item.pszText = GetString ("HK_DISMOUNT_ALL_AND_WIPE");

- break;

-

- case HK_FORCE_DISMOUNT_ALL_AND_WIPE:

- item.pszText = GetString ("HK_FORCE_DISMOUNT_ALL_AND_WIPE");

- break;

-

- case HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT:

- item.pszText = GetString ("HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT");

- break;

-

- case HK_MOUNT_FAVORITE_VOLUMES:

- item.pszText = GetString ("HK_MOUNT_FAVORITE_VOLUMES");

- break;

-

- case HK_SHOW_HIDE_MAIN_WINDOW:

- item.pszText = GetString ("HK_SHOW_HIDE_MAIN_WINDOW");

- break;

-

- case HK_CLOSE_SECURITY_TOKEN_SESSIONS:

- item.pszText = GetString ("IDM_CLOSE_ALL_TOKEN_SESSIONS");

- break;

-

- default:

- item.pszText = L"[?]";

- }

-

- SendMessageW (hList,LVM_INSERTITEMW,0,(LPARAM)&item);

-

- item.iSubItem = 1;

- Shortcut[0] = 0;

- ShortcutMod[0] = 0;

-

- if (GetKeyName (tmpHotkeys[i].vKeyCode, Shortcut))

- {

- if (tmpHotkeys[i].vKeyModifiers & MOD_CONTROL)

- {

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),GetString ("VK_CONTROL"));

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),L"+");

- }

-

- if (tmpHotkeys[i].vKeyModifiers & MOD_SHIFT)

- {

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),GetString ("VK_SHIFT"));

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),L"+");

- }

-

- if (tmpHotkeys[i].vKeyModifiers & MOD_ALT)

- {

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),GetString ("VK_ALT"));

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),L"+");

- }

-

- if (tmpHotkeys[i].vKeyModifiers & MOD_WIN)

- {

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),GetString ("VK_WIN"));

- StringCbCatW (ShortcutMod, sizeof(ShortcutMod),L"+");

- }

-

- StringCbPrintfW (ShortcutFinal, sizeof(ShortcutFinal), L"%s%s", ShortcutMod, Shortcut);

- item.pszText = ShortcutFinal;

- }

- else

- item.pszText = L"";

-

- SendMessageW (hList, LVM_SETITEMW, 0, (LPARAM)&item);

- }

-}

-

-

-

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

-{

- WORD lw = LOWORD (wParam);

- WORD hw = HIWORD (wParam);

- static BOOL bKeyScanOn;

- static BOOL bTPlaySoundOnSuccessfulHkDismount;

- static BOOL bTDisplayBalloonOnSuccessfulHkDismount;

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- LVCOLUMNW col;

- HWND hList = GetDlgItem (hwndDlg, IDC_HOTKEY_LIST);

-

- bKeyScanOn = FALSE;

- nSelectedHotkeyId = -1;

- currentVKeyCode = 0;

- memcpy (tmpHotkeys, Hotkeys, sizeof(tmpHotkeys));

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

-

- SendMessageW (hList,LVM_SETEXTENDEDLISTVIEWSTYLE,0,

- LVS_EX_FULLROWSELECT|LVS_EX_HEADERDRAGDROP|LVS_EX_LABELTIP

- );

-

- memset (&col,0,sizeof(col));

- col.mask = LVCF_TEXT|LVCF_WIDTH|LVCF_SUBITEM|LVCF_FMT;

- col.pszText = GetString ("ACTION");

- col.cx = CompensateXDPI (341);

- col.fmt = LVCFMT_LEFT;

- SendMessageW (hList,LVM_INSERTCOLUMNW,0,(LPARAM)&col);

-

- col.pszText = GetString ("SHORTCUT");

- col.cx = CompensateXDPI (190);

- col.fmt = LVCFMT_LEFT;

- SendMessageW (hList,LVM_INSERTCOLUMNW,1,(LPARAM)&col);

-

- LocalizeDialog (hwndDlg, "IDD_HOTKEYS_DLG");

-

- SetCheckBox (hwndDlg, IDC_HK_MOD_CTRL, TRUE);

- SetCheckBox (hwndDlg, IDC_HK_MOD_SHIFT, FALSE);

- SetCheckBox (hwndDlg, IDC_HK_MOD_ALT, TRUE);

- SetCheckBox (hwndDlg, IDC_HK_MOD_WIN, FALSE);

-

- SetCheckBox (hwndDlg, IDC_HK_DISMOUNT_PLAY_SOUND, bPlaySoundOnSuccessfulHkDismount);

- SetCheckBox (hwndDlg, IDC_HK_DISMOUNT_BALLOON_TOOLTIP, bDisplayBalloonOnSuccessfulHkDismount);

-

- bTPlaySoundOnSuccessfulHkDismount = bPlaySoundOnSuccessfulHkDismount;

- bTDisplayBalloonOnSuccessfulHkDismount = bDisplayBalloonOnSuccessfulHkDismount;

-

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

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

-

- DisplayHotkeyList(hwndDlg);

-

- if (SetTimer (hwndDlg, 0xfe, 10, NULL) == 0)

- {

- Error ("CANNOT_SET_TIMER", MainDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

- return 1;

- }

-

- case WM_TIMER:

- {

- if ((nSelectedHotkeyId > -1) && (GetFocus () == GetDlgItem (hwndDlg, IDC_HOTKEY_KEY)))

- {

- wchar_t keyName [MAX_KEY_COMB_NAME_LEN];

- UINT tmpVKeyCode;

-

- keyName[0] = 0;

-

- ScanAndProcessKey (&tmpVKeyCode, &keyName[0]);

-

- if (keyName[0] != 0)

- {

- currentVKeyCode = tmpVKeyCode;

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), keyName);

- EnableWindow (GetDlgItem (hwndDlg, IDC_HOTKEY_ASSIGN), TRUE);

- }

- else if ((currentVKeyCode != 0) && GetKeyName (currentVKeyCode, keyName))

- {

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), keyName);

- }

- }

- return 1;

- }

-

- case WM_NOTIFY:

- if (wParam == IDC_HOTKEY_LIST)

- {

- if (((LPNMHDR) lParam)->code == LVN_ITEMACTIVATE

- || ((LPNMHDR) lParam)->code == LVN_ITEMCHANGED && (((LPNMLISTVIEW) lParam)->uNewState & LVIS_FOCUSED))

- {

- LVITEM item;

- memset(&item,0,sizeof(item));

- nSelectedHotkeyId = ((LPNMLISTVIEW) lParam)->iItem;

- currentVKeyCode = 0;

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

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), GetString ("PRESS_A_KEY_TO_ASSIGN"));

-

- EnableWindow (GetDlgItem (hwndDlg, IDC_HOTKEY_REMOVE), (tmpHotkeys[nSelectedHotkeyId].vKeyCode > 0));

-

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

- bKeyScanOn = TRUE;

- return 1;

- }

- }

-

- return 0;

-

- case WM_COMMAND:

- if (lw == IDC_HOTKEY_KEY && hw == EN_CHANGE)

- {

- if (!bKeyScanOn && nSelectedHotkeyId < 0 && GetWindowTextLengthW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY)))

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), L"");

- }

-

- if (lw == IDC_HOTKEY_ASSIGN)

- {

- BOOL bOwnActiveShortcut = FALSE;

-

- if (nSelectedHotkeyId >= 0 && currentVKeyCode != 0)

- {

- UINT modifiers = 0;

- if (GetCheckBox (hwndDlg, IDC_HK_MOD_CTRL))

- modifiers = MOD_CONTROL;

-

- if (GetCheckBox (hwndDlg, IDC_HK_MOD_ALT))

- modifiers |= MOD_ALT;

-

- if (GetCheckBox (hwndDlg, IDC_HK_MOD_SHIFT))

- modifiers |= MOD_SHIFT;

-

- if (GetCheckBox (hwndDlg, IDC_HK_MOD_WIN))

- modifiers |= MOD_WIN;

-

- // Check if it's not already assigned

- if (ShortcutInUse (currentVKeyCode, modifiers, tmpHotkeys))

- {

- Error ("SHORTCUT_ALREADY_IN_USE", hwndDlg);

- return 1;

- }

-

- // Check for reserved system keys

- switch (currentVKeyCode)

- {

- case VK_F1:

- case VK_F12:

- /* F1 is help and F12 is reserved for use by the debugger at all times */

- if (modifiers == 0)

- {

- Error ("CANNOT_USE_RESERVED_KEY", hwndDlg);

- return 1;

- }

- break;

- }

-

- bOwnActiveShortcut = ShortcutInUse (currentVKeyCode, modifiers, Hotkeys);

-

- // Test if the shortcut can be assigned without errors

- if (!bOwnActiveShortcut

- && !RegisterHotKey (hwndDlg, nSelectedHotkeyId, modifiers, currentVKeyCode))

- {

- handleWin32Error(hwndDlg, SRC_POS);

- return 1;

- }

- else

- {

- if (!bOwnActiveShortcut && !UnregisterHotKey (hwndDlg, nSelectedHotkeyId))

- handleWin32Error(hwndDlg, SRC_POS);

-

- tmpHotkeys[nSelectedHotkeyId].vKeyCode = currentVKeyCode;

- tmpHotkeys[nSelectedHotkeyId].vKeyModifiers = modifiers;

-

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), L"");

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

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

- nSelectedHotkeyId = -1;

- bKeyScanOn = FALSE;

- currentVKeyCode = 0;

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

- }

- }

- DisplayHotkeyList(hwndDlg);

- return 1;

- }

-

- if (lw == IDC_HOTKEY_REMOVE)

- {

- if (nSelectedHotkeyId >= 0)

- {

- tmpHotkeys[nSelectedHotkeyId].vKeyCode = 0;

- tmpHotkeys[nSelectedHotkeyId].vKeyModifiers = 0;

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), L"");

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

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

- nSelectedHotkeyId = -1;

- bKeyScanOn = FALSE;

- currentVKeyCode = 0;

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

- DisplayHotkeyList(hwndDlg);

- }

- return 1;

- }

-

- if (lw == IDC_RESET_HOTKEYS)

- {

- int i;

-

- for (i = 0; i < NBR_HOTKEYS; i++)

- {

- tmpHotkeys[i].vKeyCode = 0;

- tmpHotkeys[i].vKeyModifiers = 0;

- }

- SetWindowTextW (GetDlgItem (hwndDlg, IDC_HOTKEY_KEY), L"");

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

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

- nSelectedHotkeyId = -1;

- bKeyScanOn = FALSE;

- currentVKeyCode = 0;

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

- DisplayHotkeyList(hwndDlg);

- return 1;

- }

-

- if (lw == IDC_HK_DISMOUNT_PLAY_SOUND)

- {

- bTPlaySoundOnSuccessfulHkDismount = GetCheckBox (hwndDlg, IDC_HK_DISMOUNT_PLAY_SOUND);

- }

-

- if (lw == IDC_HK_DISMOUNT_BALLOON_TOOLTIP)

- {

- bTDisplayBalloonOnSuccessfulHkDismount = GetCheckBox (hwndDlg, IDC_HK_DISMOUNT_BALLOON_TOOLTIP);

- }

-

- if (lw == IDCANCEL || lw == IDCLOSE)

- {

- KillTimer (hwndDlg, 0xfe);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- if (lw == IDOK)

- {

- HWND hwndMainDlg = hwndDlg;

-

- while (GetParent (hwndMainDlg) != NULL)

- {

- hwndMainDlg = GetParent (hwndMainDlg);

- }

- UnregisterAllHotkeys (hwndMainDlg, Hotkeys);

- memcpy (Hotkeys, tmpHotkeys, sizeof(Hotkeys));

- RegisterAllHotkeys (hwndMainDlg, Hotkeys);

- KillTimer (hwndDlg, 0xfe);

- bPlaySoundOnSuccessfulHkDismount = bTPlaySoundOnSuccessfulHkDismount;

- bDisplayBalloonOnSuccessfulHkDismount = bTDisplayBalloonOnSuccessfulHkDismount;

-

- SaveSettings (hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

- return 0;

-

- case WM_CLOSE:

-

- KillTimer (hwndDlg, 0xfe);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

- return 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.

-*/

-

-#ifdef __cplusplus

-extern "C" {

-#endif

-

-enum

-{

- /* When adding/removing hot keys, update the following functions in Mount.c:

- DisplayHotkeyList()

- SaveSettings()

- LoadSettings()

- HandleHotKey() */

-

- HK_AUTOMOUNT_DEVICES = 0,

- HK_CLOSE_SECURITY_TOKEN_SESSIONS,

- HK_DISMOUNT_ALL,

- HK_DISMOUNT_ALL_AND_WIPE,

- HK_FORCE_DISMOUNT_ALL_AND_WIPE,

- HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT,

- HK_MOUNT_FAVORITE_VOLUMES,

- HK_SHOW_HIDE_MAIN_WINDOW,

- HK_WIPE_CACHE,

- NBR_HOTKEYS

-};

-

-typedef struct

-{

- UINT vKeyCode;

- UINT vKeyModifiers;

-} TCHOTKEY;

-

-extern TCHOTKEY Hotkeys [NBR_HOTKEYS];

-

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

-BOOL GetKeyName (UINT vKey, wchar_t *keyName);

-void UnregisterAllHotkeys (HWND hwndDlg, TCHOTKEY hotkeys[]);

-BOOL RegisterAllHotkeys (HWND hwndDlg, TCHOTKEY hotkeys[]);

-

-#ifdef __cplusplus

-}

-#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 <atlcomcli.h>

-#include <atlconv.h>

-#include <windows.h>

-#include "BaseCom.h"

-#include "BootEncryption.h"

-#include "Dlgcode.h"

-#include "MainCom.h"

-#include "MainCom_h.h"

-#include "MainCom_i.c"

-#include "Mount.h"

-#include "Password.h"

-

-using namespace VeraCrypt;

-

-static volatile LONG ObjectCount = 0;

-

-class TrueCryptMainCom : public ITrueCryptMainCom

-{

-

-public:

- TrueCryptMainCom (DWORD messageThreadId) : RefCount (0), MessageThreadId (messageThreadId)

- {

- InterlockedIncrement (&ObjectCount);

- }

-

- virtual ~TrueCryptMainCom ()

- {

- if (InterlockedDecrement (&ObjectCount) == 0)

- PostThreadMessage (MessageThreadId, WM_APP, 0, 0);

- }

-

- virtual ULONG STDMETHODCALLTYPE AddRef ()

- {

- return InterlockedIncrement (&RefCount);

- }

-

- virtual ULONG STDMETHODCALLTYPE Release ()

- {

- if (!InterlockedDecrement (&RefCount))

- {

- delete this;

- return 0;

- }

-

- return RefCount;

- }

-

- virtual HRESULT STDMETHODCALLTYPE QueryInterface (REFIID riid, void **ppvObject)

- {

- if (riid == IID_IUnknown || riid == IID_ITrueCryptMainCom)

- *ppvObject = this;

- else

- {

- *ppvObject = NULL;

- return E_NOINTERFACE;

- }

-

- AddRef ();

- return S_OK;

- }

-

- virtual void STDMETHODCALLTYPE AnalyzeKernelMiniDump (LONG_PTR hwndDlg)

- {

- // Do nothing

- MainDlg = (HWND) hwndDlg;

- }

-

- virtual int STDMETHODCALLTYPE BackupVolumeHeader (LONG_PTR hwndDlg, BOOL bRequireConfirmation, BSTR lpszVolume)

- {

- MainDlg = (HWND) hwndDlg;

- return ::BackupVolumeHeader ((HWND) hwndDlg, bRequireConfirmation, lpszVolume);

- }

-

- virtual int STDMETHODCALLTYPE RestoreVolumeHeader (LONG_PTR hwndDlg, BSTR lpszVolume)

- {

- MainDlg = (HWND) hwndDlg;

- return ::RestoreVolumeHeader ((HWND) hwndDlg, lpszVolume);

- }

-

- virtual DWORD STDMETHODCALLTYPE CallDriver (DWORD ioctl, BSTR input, BSTR *output)

- {

- return BaseCom::CallDriver (ioctl, input, output);

- }

-

- virtual int STDMETHODCALLTYPE ChangePassword (BSTR volumePath, Password *oldPassword, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd)

- {

- MainDlg = (HWND) hWnd;

- return ::ChangePwd (volumePath, oldPassword, 0, 0, FALSE, newPassword, pkcs5, 0, wipePassCount, (HWND) hWnd);

- }

-

- virtual DWORD STDMETHODCALLTYPE CopyFile (BSTR sourceFile, BSTR destinationFile)

- {

- return BaseCom::CopyFile (sourceFile, destinationFile);

- }

-

- virtual DWORD STDMETHODCALLTYPE DeleteFile (BSTR file)

- {

- return BaseCom::DeleteFile (file);

- }

-

- virtual BOOL STDMETHODCALLTYPE IsPagingFileActive (BOOL checkNonWindowsPartitionsOnly)

- {

- return BaseCom::IsPagingFileActive (checkNonWindowsPartitionsOnly);

- }

-

- virtual DWORD STDMETHODCALLTYPE ReadWriteFile (BOOL write, BOOL device, BSTR filePath, BSTR *bufferBstr, unsigned __int64 offset, unsigned __int32 size, DWORD *sizeDone)

- {

- return BaseCom::ReadWriteFile (write, device, filePath, bufferBstr, offset, size, sizeDone);

- }

-

- virtual DWORD STDMETHODCALLTYPE RegisterFilterDriver (BOOL registerDriver, int filterType)

- {

- return BaseCom::RegisterFilterDriver (registerDriver, filterType);

- }

-

- virtual DWORD STDMETHODCALLTYPE RegisterSystemFavoritesService (BOOL registerService)

- {

- return BaseCom::RegisterSystemFavoritesService (registerService);

- }

-

- virtual DWORD STDMETHODCALLTYPE SetDriverServiceStartType (DWORD startType)

- {

- return BaseCom::SetDriverServiceStartType (startType);

- }

-

- virtual DWORD STDMETHODCALLTYPE WriteLocalMachineRegistryDwordValue (BSTR keyPath, BSTR valueName, DWORD value)

- {

- return BaseCom::WriteLocalMachineRegistryDwordValue (keyPath, valueName, value);

- }

-

- virtual int STDMETHODCALLTYPE ChangePasswordEx (BSTR volumePath, Password *oldPassword, int old_pkcs5, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd)

- {

- MainDlg = (HWND) hWnd;

- return ::ChangePwd (volumePath, oldPassword, old_pkcs5, 0, FALSE, newPassword, pkcs5, 0, wipePassCount, (HWND) hWnd);

- }

-

- virtual int STDMETHODCALLTYPE ChangePasswordEx2 (BSTR volumePath, Password *oldPassword, int old_pkcs5, BOOL truecryptMode, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd)

- {

- MainDlg = (HWND) hWnd;

- return ::ChangePwd (volumePath, oldPassword, old_pkcs5, 0, truecryptMode, newPassword, pkcs5, 0, wipePassCount, (HWND) hWnd);

- }

-

- virtual int STDMETHODCALLTYPE ChangePasswordEx3 (BSTR volumePath, Password *oldPassword, int old_pkcs5, int old_pim, BOOL truecryptMode, Password *newPassword, int pkcs5, int pim, int wipePassCount, LONG_PTR hWnd)

- {

- MainDlg = (HWND) hWnd;

- return ::ChangePwd (volumePath, oldPassword, old_pkcs5, old_pim, truecryptMode, newPassword, pkcs5, pim, wipePassCount, (HWND) hWnd);

- }

-

-protected:

- DWORD MessageThreadId;

- LONG RefCount;

-};

-

-

-extern "C" BOOL ComServerMain ()

-{

- SetProcessShutdownParameters (0x100, 0);

-

- TrueCryptFactory<TrueCryptMainCom> factory (GetCurrentThreadId ());

- DWORD cookie;

-

- if (IsUacSupported ())

- UacElevated = TRUE;

-

- if (CoRegisterClassObject (CLSID_TrueCryptMainCom, (LPUNKNOWN) &factory,

- CLSCTX_LOCAL_SERVER, REGCLS_SINGLEUSE, &cookie) != S_OK)

- return FALSE;

-

- MSG msg;

- while (int r = GetMessageW (&msg, NULL, 0, 0))

- {

- if (r == -1)

- return FALSE;

-

- TranslateMessage (&msg);

- DispatchMessageW (&msg);

-

- if (msg.message == WM_APP

- && ObjectCount < 1

- && !factory.IsServerLocked ())

- break;

- }

- CoRevokeClassObject (cookie);

-

- return TRUE;

-}

-

-

-static BOOL ComGetInstance (HWND hWnd, ITrueCryptMainCom **tcServer)

-{

- return ComGetInstanceBase (hWnd, CLSID_TrueCryptMainCom, IID_ITrueCryptMainCom, (void **) tcServer);

-}

-

-

-ITrueCryptMainCom *GetElevatedInstance (HWND parent)

-{

- ITrueCryptMainCom *instance;

-

- if (!ComGetInstance (parent, &instance))

- throw UserAbort (SRC_POS);

-

- return instance;

-}

-

-

-extern "C" int UacBackupVolumeHeader (HWND hwndDlg, BOOL bRequireConfirmation, wchar_t *lpszVolume)

-{

- CComPtr<ITrueCryptMainCom> tc;

- int r;

-

- CoInitialize (NULL);

-

- if (ComGetInstance (hwndDlg, &tc))

- {

- CComBSTR volumeBstr;

- BSTR bstr = W2BSTR(lpszVolume);

- if (bstr)

- {

- volumeBstr.Attach (bstr);

- r = tc->BackupVolumeHeader ((LONG_PTR) hwndDlg, bRequireConfirmation, volumeBstr);

- }

- else

- r = ERR_OUTOFMEMORY;

- }

- else

- r = -1;

-

- CoUninitialize ();

-

- return r;

-}

-

-

-extern "C" int UacRestoreVolumeHeader (HWND hwndDlg, wchar_t *lpszVolume)

-{

- CComPtr<ITrueCryptMainCom> tc;

- int r;

-

- CoInitialize (NULL);

-

- if (ComGetInstance (hwndDlg, &tc))

- {

- CComBSTR volumeBstr;

- BSTR bstr = W2BSTR(lpszVolume);

- if (bstr)

- {

- volumeBstr.Attach (bstr);

- r = tc->RestoreVolumeHeader ((LONG_PTR) hwndDlg, volumeBstr);

- }

- else

- r = ERR_OUTOFMEMORY;

- }

- else

- r = -1;

-

- CoUninitialize ();

-

- return r;

-}

-

-

-extern "C" int UacChangePwd (wchar_t *lpszVolume, Password *oldPassword, int old_pkcs5, int old_pim, BOOL truecryptMode, Password *newPassword, int pkcs5, int pim, int wipePassCount, HWND hwndDlg)

-{

- CComPtr<ITrueCryptMainCom> tc;

- int r;

-

- CoInitialize (NULL);

-

- if (ComGetInstance (hwndDlg, &tc))

- {

- CComBSTR bstrVolume (lpszVolume);

- WaitCursor ();

- r = tc->ChangePasswordEx3 (bstrVolume, oldPassword, old_pkcs5, old_pim, truecryptMode, newPassword, pkcs5, pim, wipePassCount, (LONG_PTR) hwndDlg);

- NormalCursor ();

- }

- else

- r = -1;

-

- CoUninitialize ();

-

- return r;

-}

-/*

- 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.

-*/

-

-#ifndef TC_HEADER_MAIN_COM

-#define TC_HEADER_MAIN_COM

-

-#include <windows.h>

-

-#ifdef __cplusplus

-

-#include "MainCom_h.h"

-ITrueCryptMainCom *GetElevatedInstance (HWND parent);

-

-extern "C" {

-#endif

-

-BOOL ComServerMain ();

-int UacBackupVolumeHeader (HWND hwndDlg, BOOL bRequireConfirmation, wchar_t *lpszVolume);

-int UacRestoreVolumeHeader (HWND hwndDlg, wchar_t *lpszVolume);

-int UacChangePwd (wchar_t *lpszVolume, Password *oldPassword, int old_pkcs5, int old_pim, BOOL truecryptMode, Password *newPassword, int pkcs5, int pim, int wipePassCount, HWND hwndDlg);

-

-#ifdef __cplusplus

-}

-#endif

-

-#endif // TC_HEADER_MAIN_COM

-/*

- 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.

-*/

-

-import "wtypes.idl";

-import "..\Common\Password.h";

-

-[

- uuid(9ACF6176-5FC4-4690-A025-B3306A50EB6A),

- helpstring("VeraCrypt Main UAC Support Library"),

- version(2.7) // Update ComSetup.cpp when changing version number

-]

-library TrueCryptMainCom

-{

- [

- uuid(C786E27C-2801-482c-B45D-D4357B270A29),

- object,

- oleautomation,

- helpstring("VeraCrypt Main UAC Support Interface")

- ]

- interface ITrueCryptMainCom : IUnknown

- {

- void AnalyzeKernelMiniDump (LONG_PTR hwndDlg);

- int BackupVolumeHeader (LONG_PTR hwndDlg, BOOL bRequireConfirmation, BSTR lpszVolume);

- DWORD CallDriver (DWORD ioctl, BSTR input, BSTR *output);

- int ChangePassword (BSTR volumePath, Password *oldPassword, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd);

- DWORD CopyFile (BSTR sourceFile, BSTR destinationFile);

- DWORD DeleteFile (BSTR file);

- BOOL IsPagingFileActive (BOOL checkNonWindowsPartitionsOnly);

- DWORD ReadWriteFile (BOOL write, BOOL device, BSTR filePath, BSTR *bufferBstr, unsigned __int64 offset, unsigned __int32 size, DWORD *sizeDone);

- DWORD RegisterFilterDriver (BOOL registerDriver, int filterType);

- DWORD RegisterSystemFavoritesService (BOOL registerService);

- int RestoreVolumeHeader (LONG_PTR hwndDlg, BSTR lpszVolume);

- DWORD SetDriverServiceStartType (DWORD startType);

- DWORD WriteLocalMachineRegistryDwordValue (BSTR keyPath, BSTR valueName, DWORD value);

- int ChangePasswordEx (BSTR volumePath, Password *oldPassword, int old_pkcs5, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd);

- int ChangePasswordEx2 (BSTR volumePath, Password *oldPassword, int old_pkcs5, BOOL truecryptMode, Password *newPassword, int pkcs5, int wipePassCount, LONG_PTR hWnd);

- int ChangePasswordEx3 (BSTR volumePath, Password *oldPassword, int old_pkcs5, int old_pim, BOOL truecryptMode, Password *newPassword, int pkcs5, int pim, int wipePassCount, LONG_PTR hWnd);

- };

-

- [

- uuid(FE8B3B95-C80C-41f7-830F-FBA271C26F7E),

- helpstring("VeraCrypt Main UAC Support Coclass")

- ]

- coclass TrueCryptMainCom

- {

- [default] interface ITrueCryptMainCom;

- }

-}

-/*

- 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 "cpu.h"

-

-#include <time.h>

-#include <math.h>

-#include <dbt.h>

-#include <fcntl.h>

-#include <io.h>

-#include <shlobj.h>

-#include <sys/stat.h>

-#include <windowsx.h>

-

-#include "Apidrvr.h"

-#include "BootEncryption.h"

-#include "Cmdline.h"

-#include "Crypto.h"

-#include "Dlgcode.h"

-#include "Combo.h"

-#include "Favorites.h"

-#include "Hotkeys.h"

-#include "Keyfiles.h"

-#include "Language.h"

-#include "MainCom.h"

-#include "Mount.h"

-#include "Pkcs5.h"

-#include "Random.h"

-#include "Registry.h"

-#include "Resource.h"

-#include "Password.h"

-#include "Xml.h"

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

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

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

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

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

-#include "../Platform/Finally.h"

-#include "../Platform/ForEach.h"

-

-#include <Strsafe.h>

-

-#include <wtsapi32.h>

-

-typedef BOOL (WINAPI *WTSREGISTERSESSIONNOTIFICATION)(HWND, DWORD);

-typedef BOOL (WINAPI *WTSUNREGISTERSESSIONNOTIFICATION)(HWND);

-

-using namespace VeraCrypt;

-

-enum timer_ids

-{

- TIMER_ID_MAIN = 0xff,

- TIMER_ID_KEYB_LAYOUT_GUARD

-};

-

-enum hidden_os_read_only_notif_mode

-{

- TC_HIDDEN_OS_READ_ONLY_NOTIF_MODE_NONE = 0,

- TC_HIDDEN_OS_READ_ONLY_NOTIF_MODE_COMPACT,

- TC_HIDDEN_OS_READ_ONLY_NOTIF_MODE_DISABLED

-};

-

-#define TIMER_INTERVAL_MAIN 500

-#define TIMER_INTERVAL_KEYB_LAYOUT_GUARD 10

-

-BootEncryption *BootEncObj = NULL;

-BootEncryptionStatus BootEncStatus;

-BootEncryptionStatus RecentBootEncStatus;

-

-BOOL bExplore = FALSE; /* Display explorer window after mount */

-BOOL bBeep = FALSE; /* Donot beep after mount */

-wchar_t szFileName[TC_MAX_PATH+1]; /* Volume to mount */

-wchar_t szDriveLetter[3]; /* Drive Letter to mount */

-wchar_t commandLineDrive = 0;

-BOOL bCacheInDriver = FALSE; /* Cache any passwords we see */

-BOOL bCacheInDriverDefault = FALSE;

-BOOL bCacheDuringMultipleMount = FALSE;

-BOOL bCmdCacheDuringMultipleMount = FALSE;

-BOOL bIncludePimInCache = FALSE;

-BOOL bTryEmptyPasswordWhenKeyfileUsed = FALSE;

-BOOL bCmdTryEmptyPasswordWhenKeyfileUsed = FALSE;

-BOOL bCmdTryEmptyPasswordWhenKeyfileUsedValid = FALSE;

-BOOL bHistoryCmdLine = FALSE; /* History control is always disabled */

-BOOL bUseDifferentTrayIconIfVolMounted = TRUE;

-BOOL bCloseDismountedWindows=TRUE; /* Close all open explorer windows of dismounted volume */

-BOOL bWipeCacheOnExit = FALSE; /* Wipe password from chace on exit */

-BOOL bWipeCacheOnAutoDismount = TRUE;

-BOOL bEnableBkgTask = FALSE;

-BOOL bCloseBkgTaskWhenNoVolumes = FALSE;

-BOOL bDismountOnLogOff = TRUE;

-BOOL bDismountOnSessionLocked = TRUE;

-BOOL bDismountOnScreenSaver = TRUE;

-BOOL bDismountOnPowerSaving = FALSE;

-BOOL bForceAutoDismount = TRUE;

-BOOL bForceMount = FALSE; /* Mount volume even if host file/device already in use */

-BOOL bForceUnmount = FALSE; /* Unmount volume even if it cannot be locked */

-BOOL bWipe = FALSE; /* Wipe driver passwords */

-BOOL bAuto = FALSE; /* Do everything without user input */

-BOOL LogOn = FALSE;

-BOOL bAutoMountDevices = FALSE; /* Auto-mount devices */

-BOOL bAutoMountFavorites = FALSE;

-BOOL bPlaySoundOnSuccessfulHkDismount = TRUE;

-BOOL bDisplayBalloonOnSuccessfulHkDismount = TRUE;

-BOOL bHibernationPreventionNotified = FALSE; /* TRUE if the user has been notified that hibernation was prevented (system encryption) during the session. */

-BOOL bHiddenSysLeakProtNotifiedDuringSession = FALSE; /* TRUE if the user has been notified during the session that unencrypted filesystems and non-hidden TrueCrypt volumes are mounted as read-only under hidden OS. */

-BOOL CloseSecurityTokenSessionsAfterMount = FALSE;

-

-BOOL Quit = FALSE; /* Exit after processing command line */

-BOOL ComServerMode = FALSE;

-BOOL ServiceMode = FALSE;

-BOOL UsePreferences = TRUE;

-

-int HiddenSysLeakProtectionNotificationStatus = TC_HIDDEN_OS_READ_ONLY_NOTIF_MODE_NONE;

-int MaxVolumeIdleTime = -120;

-int nCurrentShowType = 0; /* current display mode, mount, unmount etc */

-int nSelectedDriveIndex = -1; /* Item number of selected drive */

-

-int cmdUnmountDrive = -2; /* Volume drive letter to unmount (-1 = all) */

-Password VolumePassword; /* Password used for mounting volumes */

-Password CmdVolumePassword; /* Password passed from command line */

-char CmdTokenPin [SecurityToken::MaxPasswordLength + 1] = {0};

-int VolumePkcs5 = 0;

-int CmdVolumePkcs5 = 0;

-int VolumePim = -1;

-int CmdVolumePim = -1;

-int DefaultVolumePkcs5 = 0;

-BOOL VolumeTrueCryptMode = FALSE;

-BOOL CmdVolumeTrueCryptMode = FALSE;

-BOOL DefaultVolumeTrueCryptMode = FALSE;

-BOOL CmdVolumePasswordValid = FALSE;

-MountOptions CmdMountOptions;

-BOOL CmdMountOptionsValid = FALSE;

-MountOptions mountOptions;

-MountOptions defaultMountOptions;

-KeyFile *FirstCmdKeyFile;

-

-HBITMAP hbmLogoBitmapRescaled = NULL;

-wchar_t OrigKeyboardLayout [8+1] = L"00000409";

-BOOL bKeyboardLayoutChanged = FALSE; /* TRUE if the keyboard layout was changed to the standard US keyboard layout (from any other layout). */

-BOOL bKeybLayoutAltKeyWarningShown = FALSE; /* TRUE if the user has been informed that it is not possible to type characters by pressing keys while the right Alt key is held down. */

-

-static KeyFilesDlgParam hidVolProtKeyFilesParam;

-

-static MOUNT_LIST_STRUCT LastKnownMountList;

-VOLUME_NOTIFICATIONS_LIST VolumeNotificationsList;

-static DWORD LastKnownLogicalDrives;

-

-static HANDLE TaskBarIconMutex = NULL;

-static BOOL MainWindowHidden = FALSE;

-static int pwdChangeDlgMode = PCDM_CHANGE_PASSWORD;

-static int bSysEncPwdChangeDlgMode = FALSE;

-static int bPrebootPasswordDlgMode = FALSE;

-static int NoCmdLineArgs;

-static BOOL CmdLineVolumeSpecified;

-static int LastDriveListVolumeColumnWidth;

-// WTS handling

-static HMODULE hWtsLib = NULL;

-static WTSREGISTERSESSIONNOTIFICATION fnWtsRegisterSessionNotification = NULL;

-static WTSUNREGISTERSESSIONNOTIFICATION fnWtsUnRegisterSessionNotification = NULL;

-

-static void RegisterWtsNotification(HWND hWnd)

-{

- if (!hWtsLib)

- {

- wchar_t dllPath[MAX_PATH];

- if (GetSystemDirectory(dllPath, MAX_PATH))

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

- else

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

-

- hWtsLib = LoadLibrary(dllPath);

- if (hWtsLib)

- {

- fnWtsRegisterSessionNotification = (WTSREGISTERSESSIONNOTIFICATION) GetProcAddress(hWtsLib, "WTSRegisterSessionNotification" );

- fnWtsUnRegisterSessionNotification = (WTSUNREGISTERSESSIONNOTIFICATION) GetProcAddress(hWtsLib, "WTSUnRegisterSessionNotification" );

- if ( !fnWtsRegisterSessionNotification

- || !fnWtsUnRegisterSessionNotification

- || !fnWtsRegisterSessionNotification( hWnd, NOTIFY_FOR_THIS_SESSION )

- )

- {

- fnWtsRegisterSessionNotification = NULL;

- fnWtsUnRegisterSessionNotification = NULL;

- FreeLibrary(hWtsLib);

- hWtsLib = NULL;

- }

- }

- }

-}

-

-static void UnregisterWtsNotification(HWND hWnd)

-{

- if (hWtsLib && fnWtsUnRegisterSessionNotification)

- {

- fnWtsUnRegisterSessionNotification(hWnd);

- FreeLibrary(hWtsLib);

- hWtsLib = NULL;

- fnWtsRegisterSessionNotification = NULL;

- fnWtsUnRegisterSessionNotification = NULL;

- }

-}

-

-static void localcleanup (void)

-{

- // Wipe command line

- char *c = GetCommandLineA ();

- wchar_t *wc = GetCommandLineW ();

- burn(c, strlen (c));

- burn(wc, wcslen (wc) * sizeof (wchar_t));

-

- /* Delete buffered bitmaps (if any) */

- if (hbmLogoBitmapRescaled != NULL)

- {

- DeleteObject ((HGDIOBJ) hbmLogoBitmapRescaled);

- hbmLogoBitmapRescaled = NULL;

- }

-

- /* These items should have already been cleared by the functions that used them, but we're going to

- clear them for extra security. */

- burn (&VolumePassword, sizeof (VolumePassword));

- burn (&CmdVolumePassword, sizeof (CmdVolumePassword));

- burn (&VolumePkcs5, sizeof (VolumePkcs5));

- burn (&CmdVolumePkcs5, sizeof (CmdVolumePkcs5));

- burn (&VolumePim, sizeof (VolumePim));

- burn (&CmdVolumePim, sizeof (CmdVolumePim));

- burn (&VolumeTrueCryptMode, sizeof (VolumeTrueCryptMode));

- burn (&CmdVolumeTrueCryptMode, sizeof (CmdVolumeTrueCryptMode));

- burn (&mountOptions, sizeof (mountOptions));

- burn (&defaultMountOptions, sizeof (defaultMountOptions));

- burn (szFileName, sizeof(szFileName));

- burn (&CmdTokenPin, sizeof (CmdTokenPin));

-

- /* Cleanup common code resources */

- cleanup ();

-

- if (BootEncObj != NULL)

- {

- delete BootEncObj;

- BootEncObj = NULL;

- }

-

- RandStop (TRUE);

-}

-

-void RefreshMainDlg (HWND hwndDlg)

-{

- int drive = (wchar_t) (HIWORD (GetSelectedLong (GetDlgItem (hwndDlg, IDC_DRIVELIST))));

-

- MoveEditToCombo (GetDlgItem (hwndDlg, IDC_VOLUME), bHistory);

- LoadDriveLetters (hwndDlg, GetDlgItem (hwndDlg, IDC_DRIVELIST), drive);

- EnableDisableButtons (hwndDlg);

-}

-

-void EndMainDlg (HWND hwndDlg)

-{

- MoveEditToCombo (GetDlgItem (hwndDlg, IDC_VOLUME), bHistory);

-

- if (UsePreferences)

- SaveSettings (hwndDlg);

-

- if (bWipeCacheOnExit)

- {

- DWORD dwResult;

- DeviceIoControl (hDriver, TC_IOCTL_WIPE_PASSWORD_CACHE, NULL, 0, NULL, 0, &dwResult, NULL);

- }

-

- if (!bHistory)

- {

- SetWindowText (GetDlgItem (hwndDlg, IDC_VOLUME), L"");

- ClearHistory (GetDlgItem (hwndDlg, IDC_VOLUME));

- }

-

- if (TaskBarIconMutex != NULL)

- {

- MainWindowHidden = TRUE;

- ShowWindow (hwndDlg, SW_HIDE);

- }

- else

- {

- KillTimer (hwndDlg, TIMER_ID_MAIN);

- TaskBarIconRemove (hwndDlg);

- UnregisterWtsNotification(hwndDlg);

- EndDialog (hwndDlg, 0);

- }

-}

-

-static void InitMainDialog (HWND hwndDlg)

-{

- MENUITEMINFOW info;

- char *popupTexts[] = {"MENU_VOLUMES", "MENU_SYSTEM_ENCRYPTION", "MENU_FAVORITES", "MENU_TOOLS", "MENU_SETTINGS", "MENU_HELP", "MENU_WEBSITE", 0};

- wchar_t *str;

- int i;

-

- /* Call the common dialog init code */

- InitDialog (hwndDlg);

- LocalizeDialog (hwndDlg, NULL);

-

- SetWindowLongPtrW (hwndDlg, DWLP_USER, (LONG_PTR) (IsAdmin() ? TC_MAIN_WINDOW_FLAG_ADMIN_PRIVILEGES : 0));

-

- DragAcceptFiles (hwndDlg, TRUE);

-

- SendMessageW (GetDlgItem (hwndDlg, IDC_VOLUME), CB_LIMITTEXT, TC_MAX_PATH, 0);

- SetWindowTextW (hwndDlg, (IsAdmin() && !IsBuiltInAdmin() && IsUacSupported() && !IsNonInstallMode()) ? (wstring (lpszTitle) + L" [" + GetString ("ADMINISTRATOR") + L"]").c_str() : lpszTitle);

-

- // Help file name

- InitHelpFileName();

-

- // Localize menu strings

- for (i = 40001; str = (wchar_t *)GetDictionaryValueByInt (i); i++)

- {

- info.cbSize = sizeof (info);

- info.fMask = MIIM_TYPE;

- info.fType = MFT_STRING;

- info.dwTypeData = str;

- info.cch = (UINT) wcslen (str);

-

- SetMenuItemInfoW (GetMenu (hwndDlg), i, FALSE, &info);

- }

-

- for (i = 0; popupTexts[i] != 0; i++)

- {

- str = GetString (popupTexts[i]);

-

- info.cbSize = sizeof (info);

- info.fMask = MIIM_TYPE;

-

- if (strcmp (popupTexts[i], "MENU_WEBSITE") == 0)

- info.fType = MFT_STRING | MFT_RIGHTJUSTIFY;

- else

- info.fType = MFT_STRING;

-

- if (strcmp (popupTexts[i], "MENU_FAVORITES") == 0)

- FavoriteVolumesMenu = GetSubMenu (GetMenu (hwndDlg), i);

-

- info.dwTypeData = str;

- info.cch = (UINT) wcslen (str);

-

- SetMenuItemInfoW (GetMenu (hwndDlg), i, TRUE, &info);

- }

-

- // Disable menu item for changing system header key derivation algorithm until it's implemented

- EnableMenuItem (GetMenu (hwndDlg), IDM_CHANGE_SYS_HEADER_KEY_DERIV_ALGO, MF_GRAYED);

-

- try

- {

- LoadFavoriteVolumes();

- }

- catch (Exception &e)

- {

- e.Show (NULL);

- }

-

- // Resize the logo bitmap if the user has a non-default DPI

- if (ScreenDPI != USER_DEFAULT_SCREEN_DPI

- && hbmLogoBitmapRescaled == NULL) // If not re-called (e.g. after language pack change)

- {

- hbmLogoBitmapRescaled = RenderBitmap (MAKEINTRESOURCE (IDB_LOGO_288DPI),

- GetDlgItem (hwndDlg, IDC_LOGO),

- 0, 0, 0, 0, FALSE, TRUE);

- }

-

- BuildTree (hwndDlg, GetDlgItem (hwndDlg, IDC_DRIVELIST));

-

- if (*szDriveLetter != 0)

- {

- SelectItem (GetDlgItem (hwndDlg, IDC_DRIVELIST), *szDriveLetter);

-

- if(nSelectedDriveIndex > SendMessage (GetDlgItem (hwndDlg, IDC_DRIVELIST), LVM_GETITEMCOUNT, 0, 0)/2)

- SendMessage(GetDlgItem (hwndDlg, IDC_DRIVELIST), LVM_SCROLL, 0, 10000);

- }

- else

- {

- SendMessage(hwndDlg, WM_NEXTDLGCTL, (WPARAM) GetDlgItem (hwndDlg, IDC_DRIVELIST), 1L);

- }

-

- SendMessage (GetDlgItem (hwndDlg, IDC_NO_HISTORY), BM_SETCHECK, bHistory ? BST_UNCHECKED : BST_CHECKED, 0);

- EnableDisableButtons (hwndDlg);

-}

-

-void EnableDisableButtons (HWND hwndDlg)

-{

- HWND hOKButton = GetDlgItem (hwndDlg, IDOK);

- WORD x;

-

- x = LOWORD (GetSelectedLong (GetDlgItem (hwndDlg, IDC_DRIVELIST)));

-

- EnableMenuItem (GetMenu (hwndDlg), IDM_MOUNT_VOLUME, MF_ENABLED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_MOUNT_VOLUME_OPTIONS, MF_ENABLED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_BACKUP_VOL_HEADER, MF_ENABLED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_RESTORE_VOL_HEADER, MF_ENABLED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_CHANGE_PASSWORD, MF_ENABLED);

- EnableWindow (hOKButton, TRUE);

-

- switch (x)

- {

- case TC_MLIST_ITEM_NONSYS_VOL:

- {

- SetWindowTextW (hOKButton, GetString ("UNMOUNT_BUTTON"));

- EnableWindow (hOKButton, TRUE);

- EnableMenuItem (GetMenu (hwndDlg), IDM_UNMOUNT_VOLUME, MF_ENABLED);

-

- EnableWindow (GetDlgItem (hwndDlg, IDC_VOLUME_PROPERTIES), TRUE);

- EnableMenuItem (GetMenu (hwndDlg), IDM_VOLUME_PROPERTIES, MF_ENABLED);

- }

- break;

-

- case TC_MLIST_ITEM_SYS_PARTITION:

- case TC_MLIST_ITEM_SYS_DRIVE:

- EnableWindow (hOKButton, FALSE);

- SetWindowTextW (hOKButton, GetString ("MOUNT_BUTTON"));

- EnableWindow (GetDlgItem (hwndDlg, IDC_VOLUME_PROPERTIES), TRUE);

- EnableMenuItem (GetMenu (hwndDlg), IDM_UNMOUNT_VOLUME, MF_GRAYED);

- break;

-

- case TC_MLIST_ITEM_FREE:

- default:

- SetWindowTextW (hOKButton, GetString ("MOUNT_BUTTON"));

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

- EnableMenuItem (GetMenu (hwndDlg), IDM_VOLUME_PROPERTIES, MF_GRAYED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_UNMOUNT_VOLUME, MF_GRAYED);

- }

-

- EnableWindow (GetDlgItem (hwndDlg, IDC_WIPE_CACHE), !IsPasswordCacheEmpty());

- EnableMenuItem (GetMenu (hwndDlg), IDM_WIPE_CACHE, IsPasswordCacheEmpty() ? MF_GRAYED:MF_ENABLED);

- EnableMenuItem (GetMenu (hwndDlg), IDM_CLEAR_HISTORY, IsComboEmpty (GetDlgItem (hwndDlg, IDC_VOLUME)) ? MF_GRAYED:MF_ENABLED);

-}

-

-BOOL VolumeSelected (HWND hwndDlg)

-{

- return (GetWindowTextLength (GetDlgItem (hwndDlg, IDC_VOLUME)) > 0);

-}

-

-void GetVolumePath (HWND hwndDlg, LPWSTR szPath, int nMaxCount)

-{

- GetWindowText (GetDlgItem (hwndDlg, IDC_VOLUME), szPath, nMaxCount);

- CorrectFileName (szPath);

-}

-

-/* Returns TRUE if the last partition/drive selected via the Select Device dialog box was the system

-partition/drive and if it is encrypted.

- WARNING: This function is very fast but not always reliable (for example, if the user manually types

- a device path before Select Device is invoked during the session; after the Select Device dialog

- has been invoked at least once, the correct system device paths are cached). Therefore, it must NOT

- be used before performing any dangerous operations (such as header backup restore or formatting a

- supposedly non-system device) -- instead use IsSystemDevicePath(path, hwndDlg, TRUE) for such

- purposes. This function can be used only for preliminary GUI checks requiring very fast responses. */

-BOOL ActiveSysEncDeviceSelected (void)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

-

- if (BootEncStatus.DriveEncrypted)

- {

- int retCode = 0;

-

- GetVolumePath (MainDlg, szFileName, ARRAYSIZE (szFileName));

-

- retCode = IsSystemDevicePath (szFileName, MainDlg, FALSE);

-

- return (WholeSysDriveEncryption(FALSE) ? (retCode == 2 || retCode == 1) : (retCode == 1));

- }

- }

- catch (Exception &e)

- {

- e.Show (MainDlg);

- }

-

- return FALSE;

-}

-

-// When a function does not require the affected volume to be dismounted, there may be cases where we have two valid

-// paths selected in the main window and we cannot be sure which of them the user really intends to apply the function to.

-// This function asks the user to explicitly select either the volume path specified in the input field below the main

-// drive list (whether mounted or not), or the path to the volume selected in the main drive list. If, however, both

-// of the GUI elements contain the same volume (or one of them does not contain any path), this function does not

-// ask the user and returns the volume path directly (no selection ambiguity).

-// If driveNoPtr is not NULL, and the volume is mounted, its drive letter is returned in *driveNoPtr (if no valid drive

-// letter is resolved, -1 is stored instead).

-static wstring ResolveAmbiguousSelection (HWND hwndDlg, int *driveNoPtr)

-{

- LPARAM selectedDrive = GetSelectedLong (GetDlgItem (MainDlg, IDC_DRIVELIST));

-

- wchar_t volPathInputField [TC_MAX_PATH];

-

- wchar_t volPathDriveListW [TC_MAX_PATH];

- wstring volPathDriveListStr;

-

- wstring retPath;

-

- VOLUME_PROPERTIES_STRUCT prop;

- DWORD dwResult;

-

- BOOL useInputField = TRUE;

-

- memset (&prop, 0, sizeof(prop));

-

- BOOL ambig = (LOWORD (selectedDrive) != TC_MLIST_ITEM_FREE && LOWORD (selectedDrive) != 0xffff && HIWORD (selectedDrive) != 0xffff

- && VolumeSelected (MainDlg));

-

- if (VolumeSelected (MainDlg))

- {

- // volPathInputField will contain the volume path (if any) from the input field below the drive list

- GetVolumePath (MainDlg, volPathInputField, ARRAYSIZE (volPathInputField));

-

- if (!ambig)

- retPath = (wstring) volPathInputField;

- }

-

- if (LOWORD (selectedDrive) != TC_MLIST_ITEM_FREE && LOWORD (selectedDrive) != 0xffff && HIWORD (selectedDrive) != 0xffff)

- {

- // A volume is selected in the main drive list.

-

- switch (LOWORD (selectedDrive))

- {

- case TC_MLIST_ITEM_NONSYS_VOL:

- prop.driveNo = HIWORD (selectedDrive) - L'A';

-

- if (!DeviceIoControl (hDriver, TC_IOCTL_GET_VOLUME_PROPERTIES, &prop, sizeof (prop), &prop, sizeof (prop), &dwResult, NULL) || dwResult == 0)

- {

- // The driver did not return any path for this drive letter (the volume may have been dismounted).

-

- // Return whatever is in the input field below the drive list (even if empty)

- return ((wstring) volPathInputField);

- }

-

- // volPathDriveListWStr will contain the volume path selected in the main drive list

- volPathDriveListStr = (wstring) prop.wszVolume;

- break;

-

- case TC_MLIST_ITEM_SYS_PARTITION:

-

- GetSysDevicePaths (MainDlg);

-

- if (bCachedSysDevicePathsValid)

- {

- volPathDriveListStr = (wstring) SysPartitionDevicePath;

- }

-

- break;

-

- case TC_MLIST_ITEM_SYS_DRIVE:

-

- GetSysDevicePaths (MainDlg);

-

- if (bCachedSysDevicePathsValid)

- {

- volPathDriveListStr = (wstring) SysDriveDevicePath;

- }

-

- break;

- }

-

- if (!ambig)

- {

- useInputField = FALSE;

- retPath = volPathDriveListStr;

- }

- }

-

- if (ambig)

- {

- /* We have two paths. Compare them and if they don't match, ask the user to select one of them. Otherwise, return the path without asking. */

-

- if (wmemcmp (volPathDriveListStr.c_str (), L"\\??\\", 4) == 0)

- {

- // The volume path starts with "\\??\\" which is used for file-hosted containers. We're going to strip this prefix.

-

- volPathDriveListStr = (wstring) (volPathDriveListStr.c_str () + 4);

- }

-

- StringCbCopyW (volPathDriveListW, sizeof(volPathDriveListW), volPathDriveListStr.c_str ());

-

- if (wcscmp (((wmemcmp (volPathDriveListW, L"\\??\\", 4) == 0) ? volPathDriveListW + 4 : volPathDriveListW), volPathInputField) != 0)

- {

- // The path selected in the input field is different from the path to the volume selected

- // in the drive lettter list. We have to resolve possible ambiguity.

-

- wchar_t *tmp[] = {L"", L"", L"", L"", L"", 0};

- const int maxVolPathLen = 80;

-

- if (volPathDriveListStr.length () > maxVolPathLen)

- {

- // Ellipsis (path too long)

- volPathDriveListStr = wstring (L"...") + volPathDriveListStr.substr (volPathDriveListStr.length () - maxVolPathLen, maxVolPathLen);

- }

-

- wstring volPathInputFieldWStr (volPathInputField);

-

- if (volPathInputFieldWStr.length () > maxVolPathLen)

- {

- // Ellipsis (path too long)

- volPathInputFieldWStr = wstring (L"...") + volPathInputFieldWStr.substr (volPathInputFieldWStr.length () - maxVolPathLen, maxVolPathLen);

- }

-

- tmp[1] = GetString ("AMBIGUOUS_VOL_SELECTION");

- tmp[2] = (wchar_t *) volPathDriveListStr.c_str();

- tmp[3] = (wchar_t *) volPathInputFieldWStr.c_str();

- tmp[4] = GetString ("IDCANCEL");

-

- switch (AskMultiChoice ((void **) tmp, FALSE, hwndDlg))

- {

- case 1:

- retPath = volPathDriveListStr;

- break;

-

- case 2:

- retPath = (wstring) volPathInputField;

- break;

-

- default:

- if (driveNoPtr != NULL)

- *driveNoPtr = -1;

-

- return wstring (L"");

- }

- }

- else

- {

- // Both selected paths are the same

- retPath = (wstring) volPathInputField;

- }

- }

-

- if (driveNoPtr != NULL)

- *driveNoPtr = GetMountedVolumeDriveNo ((wchar_t *) retPath.c_str ());

-

-

- if (wmemcmp (retPath.c_str (), L"\\??\\", 4) == 0)

- {

- // The selected volume path starts with "\\??\\" which is used for file-hosted containers. We're going to strip this prefix.

-

- retPath = (wstring) (retPath.c_str () + 4);

- }

-

- return retPath;

-}

-

-void LoadSettingsAndCheckModified (HWND hwndDlg, BOOL bOnlyCheckModified, BOOL* pbSettingsModified, BOOL* pbHistoryModified)

-{

- char langid[6] = {0};

- if (!bOnlyCheckModified)

- EnableHwEncryption ((ReadDriverConfigurationFlags() & TC_DRIVER_CONFIG_DISABLE_HARDWARE_ENCRYPTION) ? FALSE : TRUE);

-

- WipeAlgorithmId savedWipeAlgorithm = TC_WIPE_NONE;

-

- if (!bOnlyCheckModified)

- LoadSysEncSettings ();

-

- if (!bOnlyCheckModified && LoadNonSysInPlaceEncSettings (&savedWipeAlgorithm) != 0)

- bInPlaceEncNonSysPending = TRUE;

-

- // If the config file has already been loaded during this session

- if (ConfigBuffer != NULL)

- {

- free (ConfigBuffer);

- ConfigBuffer = NULL;

- }

-

- // Options

- ConfigReadCompareInt ("OpenExplorerWindowAfterMount", FALSE, &bExplore, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("UseDifferentTrayIconIfVolumesMounted", TRUE, &bUseDifferentTrayIconIfVolMounted, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("SaveVolumeHistory", FALSE, &bHistory, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("CachePasswords", FALSE, &bCacheInDriverDefault, bOnlyCheckModified, pbSettingsModified);

- if (!bOnlyCheckModified)

- bCacheInDriver = bCacheInDriverDefault;

-

- ConfigReadCompareInt ("CachePasswordDuringMultipleMount", FALSE, &bCacheDuringMultipleMount, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("WipePasswordCacheOnExit", FALSE, &bWipeCacheOnExit, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("WipeCacheOnAutoDismount", TRUE, &bWipeCacheOnAutoDismount, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("IncludePimInCache", FALSE, &bIncludePimInCache, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("TryEmptyPasswordWhenKeyfileUsed",FALSE, &bTryEmptyPasswordWhenKeyfileUsed, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("StartOnLogon", FALSE, &bStartOnLogon, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("MountDevicesOnLogon", FALSE, &bMountDevicesOnLogon, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("MountFavoritesOnLogon", FALSE, &bMountFavoritesOnLogon, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("EnableBackgroundTask", TRUE, &bEnableBkgTask, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("CloseBackgroundTaskOnNoVolumes", FALSE, &bCloseBkgTaskWhenNoVolumes, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("DismountOnLogOff", !(IsServerOS() && IsAdmin()), &bDismountOnLogOff, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("DismountOnSessionLocked", FALSE, &bDismountOnSessionLocked, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("DismountOnPowerSaving", FALSE, &bDismountOnPowerSaving, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("DismountOnScreenSaver", FALSE, &bDismountOnScreenSaver, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("ForceAutoDismount", TRUE, &bForceAutoDismount, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("MaxVolumeIdleTime", -60, &MaxVolumeIdleTime, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("HiddenSectorDetectionStatus", 0, &HiddenSectorDetectionStatus, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("UseKeyfiles", FALSE, &defaultKeyFilesParam.EnableKeyFiles, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("PreserveTimestamps", TRUE, &defaultMountOptions.PreserveTimestamp, bOnlyCheckModified, pbSettingsModified);

- if (!bOnlyCheckModified)

- bPreserveTimestamp = defaultMountOptions.PreserveTimestamp;

-

- ConfigReadCompareInt ("ShowDisconnectedNetworkDrives", FALSE, &bShowDisconnectedNetworkDrives, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("HideWaitingDialog", FALSE, &bHideWaitingDialog, bOnlyCheckModified, pbSettingsModified);

-

- ConfigReadCompareInt ("MountVolumesRemovable", FALSE, &defaultMountOptions.Removable, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("MountVolumesReadOnly", FALSE, &defaultMountOptions.ReadOnly, bOnlyCheckModified, pbSettingsModified);

-

- if (!bOnlyCheckModified)

- {

- defaultMountOptions.ProtectHiddenVolume = FALSE;

- defaultMountOptions.ProtectedHidVolPkcs5Prf = 0;

- defaultMountOptions.ProtectedHidVolPim = 0;

- defaultMountOptions.PartitionInInactiveSysEncScope = FALSE;

- defaultMountOptions.RecoveryMode = FALSE;

- defaultMountOptions.UseBackupHeader = FALSE;

-

- mountOptions = defaultMountOptions;

- }

-

- ConfigReadCompareInt ("CloseSecurityTokenSessionsAfterMount", 0, &CloseSecurityTokenSessionsAfterMount, bOnlyCheckModified, pbSettingsModified);

-

- if (IsHiddenOSRunning())

- ConfigReadCompareInt ("HiddenSystemLeakProtNotifStatus", TC_HIDDEN_OS_READ_ONLY_NOTIF_MODE_NONE, &HiddenSysLeakProtectionNotificationStatus, bOnlyCheckModified, pbSettingsModified);

-

- // Drive letter - command line arg overrides registry

- if (!bOnlyCheckModified && bHistory && szDriveLetter[0] == 0)

- {

- char szTmp[3] = {0};

- ConfigReadString ("LastSelectedDrive", "", szTmp, sizeof (szTmp));

- MultiByteToWideChar (CP_UTF8, 0, szTmp, -1, szDriveLetter, ARRAYSIZE (szDriveLetter));

- }

- if (bHistory && pbSettingsModified)

- {

- // only check for last drive modification if history enabled

- char szTmp[32] = {0};

- LPARAM lLetter;

- lLetter = GetSelectedLong (GetDlgItem (MainDlg, IDC_DRIVELIST));

- if (LOWORD (lLetter) != 0xffff)

- StringCbPrintfA (szTmp, sizeof(szTmp), "%lc:", (wchar_t) HIWORD (lLetter));

-

- ConfigReadCompareString ("LastSelectedDrive", "", szTmp, sizeof (szTmp), bOnlyCheckModified, pbSettingsModified);

- }

-

- {

- char szTmp[MAX_PATH];

- WideCharToMultiByte (CP_UTF8, 0, SecurityTokenLibraryPath, -1, szTmp, MAX_PATH, NULL, NULL);

- ConfigReadCompareString ("SecurityTokenLibrary", "", szTmp, sizeof (szTmp) - 1, bOnlyCheckModified, pbSettingsModified);

- MultiByteToWideChar (CP_UTF8, 0, szTmp, -1, SecurityTokenLibraryPath, ARRAYSIZE (SecurityTokenLibraryPath));

- if (!bOnlyCheckModified && SecurityTokenLibraryPath[0])

- {

- InitSecurityTokenLibrary(hwndDlg);

- }

- }

-

- // Hotkeys

- ConfigReadCompareInt ("PlaySoundOnHotkeyMountDismount", TRUE, &bPlaySoundOnSuccessfulHkDismount, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("DisplayMsgBoxOnHotkeyDismount", TRUE, &bDisplayBalloonOnSuccessfulHkDismount, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModAutoMountDevices", 0, (int*) &Hotkeys [HK_AUTOMOUNT_DEVICES].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeAutoMountDevices", 0, (int*) &Hotkeys [HK_AUTOMOUNT_DEVICES].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModDismountAll", 0, (int*) &Hotkeys [HK_DISMOUNT_ALL].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeDismountAll", 0, (int*) &Hotkeys [HK_DISMOUNT_ALL].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModWipeCache", 0, (int*) &Hotkeys [HK_WIPE_CACHE].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeWipeCache", 0, (int*) &Hotkeys [HK_WIPE_CACHE].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModDismountAllWipe", 0, (int*) &Hotkeys [HK_DISMOUNT_ALL_AND_WIPE].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeDismountAllWipe", 0, (int*) &Hotkeys [HK_DISMOUNT_ALL_AND_WIPE].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModForceDismountAllWipe", 0, (int*) &Hotkeys [HK_FORCE_DISMOUNT_ALL_AND_WIPE].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeForceDismountAllWipe", 0, (int*) &Hotkeys [HK_FORCE_DISMOUNT_ALL_AND_WIPE].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModForceDismountAllWipeExit", 0, (int*) &Hotkeys [HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeForceDismountAllWipeExit", 0, (int*) &Hotkeys [HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModMountFavoriteVolumes", 0, (int*) &Hotkeys [HK_MOUNT_FAVORITE_VOLUMES].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeMountFavoriteVolumes", 0, (int*) &Hotkeys [HK_MOUNT_FAVORITE_VOLUMES].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModShowHideMainWindow", 0, (int*) &Hotkeys [HK_SHOW_HIDE_MAIN_WINDOW].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeShowHideMainWindow", 0, (int*) &Hotkeys [HK_SHOW_HIDE_MAIN_WINDOW].vKeyCode, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyModCloseSecurityTokenSessions", 0, (int*) &Hotkeys [HK_CLOSE_SECURITY_TOKEN_SESSIONS].vKeyModifiers, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("HotkeyCodeCloseSecurityTokenSessions", 0, (int*) &Hotkeys [HK_CLOSE_SECURITY_TOKEN_SESSIONS].vKeyCode, bOnlyCheckModified, pbSettingsModified);

-

- // History

- if (bHistoryCmdLine != TRUE)

- {

- LoadCombo (GetDlgItem (MainDlg, IDC_VOLUME), bHistory, bOnlyCheckModified, pbHistoryModified);

- if (!bOnlyCheckModified && CmdLineVolumeSpecified)

- SetWindowText (GetDlgItem (MainDlg, IDC_VOLUME), szFileName);

- }

-

- // Mount Options

- ConfigReadCompareInt ("DefaultPRF", 0, &DefaultVolumePkcs5, bOnlyCheckModified, pbSettingsModified);

- ConfigReadCompareInt ("DefaultTrueCryptMode", FALSE, &DefaultVolumeTrueCryptMode, bOnlyCheckModified, pbSettingsModified);

-

- if (bOnlyCheckModified)

- {

- if (!IsNonInstallMode ())

- {

- ConfigReadString ("Language", "", langid, sizeof (langid));

- // when installed, if no preferred language set by user, English is set default

- //

- if (langid [0] == 0)

- StringCbCopyA (langid, sizeof(langid), "en");

-

- if (pbSettingsModified && strcmp (langid, GetPreferredLangId ()))

- *pbSettingsModified = TRUE;

- }

- else

- {

- StringCbCopyA (langid, sizeof(langid), GetPreferredLangId ());

- ConfigReadCompareString ("Language", "", langid, sizeof (langid), TRUE, pbSettingsModified);

- }

- }

-

- if (DefaultVolumePkcs5 < 0 || DefaultVolumePkcs5 > LAST_PRF_ID)

- DefaultVolumePkcs5 = 0;

- if (DefaultVolumeTrueCryptMode != TRUE && DefaultVolumeTrueCryptMode != FALSE)

- DefaultVolumeTrueCryptMode = FALSE;

-

-}

-

-void LoadSettings ( HWND hwndDlg )

-{

- LoadSettingsAndCheckModified (hwndDlg, FALSE, NULL, NULL);

-}

-

-void SaveSettings (HWND hwndDlg)

-{

- WaitCursor ();

-

- // Check first if modifications ocurred before writing to the settings and history files

- // This avoids leaking information about VeraCrypt usage when user only mount volumes without changing setttings or history

- BOOL bSettingsChanged = FALSE;

- BOOL bHistoryChanged = FALSE;

-

- LoadSettingsAndCheckModified (hwndDlg, TRUE, &bSettingsChanged, &bHistoryChanged);

-

- if (bSettingsChanged)

- {

- char szTmp[32] = {0};

- LPARAM lLetter;

-

- // Options

- ConfigWriteBegin ();

-

- ConfigWriteInt ("OpenExplorerWindowAfterMount", bExplore);

- ConfigWriteInt ("UseDifferentTrayIconIfVolumesMounted", bUseDifferentTrayIconIfVolMounted);

- ConfigWriteInt ("SaveVolumeHistory", bHistory);

-

- ConfigWriteInt ("CachePasswords", bCacheInDriverDefault);

- ConfigWriteInt ("CachePasswordDuringMultipleMount", bCacheDuringMultipleMount);

- ConfigWriteInt ("WipePasswordCacheOnExit", bWipeCacheOnExit);

- ConfigWriteInt ("WipeCacheOnAutoDismount", bWipeCacheOnAutoDismount);

-

- ConfigWriteInt ("IncludePimInCache", bIncludePimInCache);

-

- ConfigWriteInt ("TryEmptyPasswordWhenKeyfileUsed", bTryEmptyPasswordWhenKeyfileUsed);

-

- ConfigWriteInt ("StartOnLogon", bStartOnLogon);

- ConfigWriteInt ("MountDevicesOnLogon", bMountDevicesOnLogon);

- ConfigWriteInt ("MountFavoritesOnLogon", bMountFavoritesOnLogon);

-

- ConfigWriteInt ("MountVolumesReadOnly", defaultMountOptions.ReadOnly);

- ConfigWriteInt ("MountVolumesRemovable", defaultMountOptions.Removable);

- ConfigWriteInt ("PreserveTimestamps", defaultMountOptions.PreserveTimestamp);

- ConfigWriteInt ("ShowDisconnectedNetworkDrives",bShowDisconnectedNetworkDrives);

- ConfigWriteInt ("HideWaitingDialog", bHideWaitingDialog);

-

- ConfigWriteInt ("EnableBackgroundTask", bEnableBkgTask);

- ConfigWriteInt ("CloseBackgroundTaskOnNoVolumes", bCloseBkgTaskWhenNoVolumes);

-

- ConfigWriteInt ("DismountOnLogOff", bDismountOnLogOff);

- ConfigWriteInt ("DismountOnSessionLocked", bDismountOnSessionLocked);

- ConfigWriteInt ("DismountOnPowerSaving", bDismountOnPowerSaving);

- ConfigWriteInt ("DismountOnScreenSaver", bDismountOnScreenSaver);

- ConfigWriteInt ("ForceAutoDismount", bForceAutoDismount);

- ConfigWriteInt ("MaxVolumeIdleTime", MaxVolumeIdleTime);

-

- ConfigWriteInt ("HiddenSectorDetectionStatus", HiddenSectorDetectionStatus);

-

- ConfigWriteInt ("UseKeyfiles", defaultKeyFilesParam.EnableKeyFiles);

-

- if (IsHiddenOSRunning())

- ConfigWriteInt ("HiddenSystemLeakProtNotifStatus", HiddenSysLeakProtectionNotificationStatus);

-

- // save last selected drive only when history enabled

- if (bHistory)

- {

- // Drive Letter

- lLetter = GetSelectedLong (GetDlgItem (MainDlg, IDC_DRIVELIST));

- if (LOWORD (lLetter) != 0xffff)

- StringCbPrintfA (szTmp, sizeof(szTmp), "%lc:", (wchar_t) HIWORD (lLetter));

- ConfigWriteString ("LastSelectedDrive", szTmp);

- }

-

- ConfigWriteInt ("CloseSecurityTokenSessionsAfterMount", CloseSecurityTokenSessionsAfterMount);

-

- // Hotkeys

- ConfigWriteInt ("HotkeyModAutoMountDevices", Hotkeys[HK_AUTOMOUNT_DEVICES].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeAutoMountDevices", Hotkeys[HK_AUTOMOUNT_DEVICES].vKeyCode);

- ConfigWriteInt ("HotkeyModDismountAll", Hotkeys[HK_DISMOUNT_ALL].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeDismountAll", Hotkeys[HK_DISMOUNT_ALL].vKeyCode);

- ConfigWriteInt ("HotkeyModWipeCache", Hotkeys[HK_WIPE_CACHE].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeWipeCache", Hotkeys[HK_WIPE_CACHE].vKeyCode);

- ConfigWriteInt ("HotkeyModDismountAllWipe", Hotkeys[HK_DISMOUNT_ALL_AND_WIPE].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeDismountAllWipe", Hotkeys[HK_DISMOUNT_ALL_AND_WIPE].vKeyCode);

- ConfigWriteInt ("HotkeyModForceDismountAllWipe", Hotkeys[HK_FORCE_DISMOUNT_ALL_AND_WIPE].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeForceDismountAllWipe", Hotkeys[HK_FORCE_DISMOUNT_ALL_AND_WIPE].vKeyCode);

- ConfigWriteInt ("HotkeyModForceDismountAllWipeExit", Hotkeys[HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeForceDismountAllWipeExit", Hotkeys[HK_FORCE_DISMOUNT_ALL_AND_WIPE_AND_EXIT].vKeyCode);

- ConfigWriteInt ("HotkeyModMountFavoriteVolumes", Hotkeys[HK_MOUNT_FAVORITE_VOLUMES].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeMountFavoriteVolumes", Hotkeys[HK_MOUNT_FAVORITE_VOLUMES].vKeyCode);

- ConfigWriteInt ("HotkeyModShowHideMainWindow", Hotkeys[HK_SHOW_HIDE_MAIN_WINDOW].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeShowHideMainWindow", Hotkeys[HK_SHOW_HIDE_MAIN_WINDOW].vKeyCode);

- ConfigWriteInt ("HotkeyModCloseSecurityTokenSessions", Hotkeys[HK_CLOSE_SECURITY_TOKEN_SESSIONS].vKeyModifiers);

- ConfigWriteInt ("HotkeyCodeCloseSecurityTokenSessions", Hotkeys[HK_CLOSE_SECURITY_TOKEN_SESSIONS].vKeyCode);

- ConfigWriteInt ("PlaySoundOnHotkeyMountDismount", bPlaySoundOnSuccessfulHkDismount);

- ConfigWriteInt ("DisplayMsgBoxOnHotkeyDismount", bDisplayBalloonOnSuccessfulHkDismount);

-

- // Language

- ConfigWriteString ("Language", GetPreferredLangId ());

-

- // PKCS#11 Library Path

- ConfigWriteStringW ("SecurityTokenLibrary", SecurityTokenLibraryPath[0] ? SecurityTokenLibraryPath : L"");

-

- // Mount Options

- ConfigWriteInt ("DefaultPRF", DefaultVolumePkcs5);

- ConfigWriteInt ("DefaultTrueCryptMode", DefaultVolumeTrueCryptMode);

-

- ConfigWriteEnd (hwndDlg);

- }

-

- if (bHistoryChanged)

- {

- // History

- DumpCombo (GetDlgItem (MainDlg, IDC_VOLUME), IsButtonChecked (GetDlgItem (MainDlg, IDC_NO_HISTORY)));

- }

-

- NormalCursor ();

-}

-

-// Returns TRUE if system encryption or decryption had been or is in progress and has not been completed

-static BOOL SysEncryptionOrDecryptionRequired (void)

-{

- /* If you update this function, revise SysEncryptionOrDecryptionRequired() in Tcformat.c as well. */

-

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- e.Show (MainDlg);

- }

-

- return (SystemEncryptionStatus == SYSENC_STATUS_ENCRYPTING

- || SystemEncryptionStatus == SYSENC_STATUS_DECRYPTING

- ||

- (

- BootEncStatus.DriveMounted

- &&

- (

- BootEncStatus.ConfiguredEncryptedAreaStart != BootEncStatus.EncryptedAreaStart

- || BootEncStatus.ConfiguredEncryptedAreaEnd != BootEncStatus.EncryptedAreaEnd

- )

- )

- );

-}

-

-// Returns TRUE if the system partition/drive is completely encrypted

-static BOOL SysDriveOrPartitionFullyEncrypted (BOOL bSilent)

-{

- /* If you update this function, revise SysDriveOrPartitionFullyEncrypted() in Tcformat.c as well. */

-

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- if (!bSilent)

- e.Show (MainDlg);

- }

-

- return (!BootEncStatus.SetupInProgress

- && BootEncStatus.ConfiguredEncryptedAreaEnd != 0

- && BootEncStatus.ConfiguredEncryptedAreaEnd != -1

- && BootEncStatus.ConfiguredEncryptedAreaStart == BootEncStatus.EncryptedAreaStart

- && BootEncStatus.ConfiguredEncryptedAreaEnd == BootEncStatus.EncryptedAreaEnd);

-}

-

-// Returns TRUE if the system partition/drive is being filtered by the TrueCrypt driver and the key data

-// was successfully decrypted (the device is fully ready to be encrypted or decrypted). Note that this

-// function does not examine whether the system device is encrypted or not (or to what extent).

-static BOOL SysEncDeviceActive (BOOL bSilent)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- if (!bSilent)

- e.Show (MainDlg);

-

- return FALSE;

- }

-

- return (BootEncStatus.DriveMounted);

-}

-

-// Returns TRUE if the entire system drive (as opposed to the system partition only) of the currently running OS is (or is to be) encrypted

-BOOL WholeSysDriveEncryption (BOOL bSilent)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

-

- if (BootEncStatus.BootDriveLength.QuadPart < 1) // paranoid check

- return FALSE;

- else

- return (BootEncStatus.ConfiguredEncryptedAreaStart == TC_BOOT_LOADER_AREA_SIZE

- && BootEncStatus.ConfiguredEncryptedAreaEnd >= BootEncStatus.BootDriveLength.QuadPart - 1);

- }

- catch (Exception &e)

- {

- if (!bSilent)

- e.Show (MainDlg);

-

- return FALSE;

- }

-}

-

-// Returns the size of the system drive/partition (if encrypted) in bytes

-unsigned __int64 GetSysEncDeviceSize (BOOL bSilent)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- if (!bSilent)

- e.Show (MainDlg);

- return 1;

- }

-

- if ( BootEncStatus.ConfiguredEncryptedAreaEnd < 0

- || BootEncStatus.ConfiguredEncryptedAreaStart < 0

- || BootEncStatus.ConfiguredEncryptedAreaEnd < BootEncStatus.ConfiguredEncryptedAreaStart

- )

- return 1; // we return 1 to avoid devision by zero

- else

- return ((unsigned __int64)(BootEncStatus.ConfiguredEncryptedAreaEnd - BootEncStatus.ConfiguredEncryptedAreaStart)) + 1;

-}

-

-// Returns the current size of the encrypted area of the system drive/partition in bytes

-unsigned __int64 GetSysEncDeviceEncryptedPartSize (BOOL bSilent)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- if (!bSilent)

- e.Show (MainDlg);

- return 0;

- }

-

- if ( BootEncStatus.EncryptedAreaEnd < 0

- || BootEncStatus.EncryptedAreaStart < 0

- || BootEncStatus.EncryptedAreaEnd < BootEncStatus.EncryptedAreaStart

- )

- return 0;

- else

- return ((unsigned __int64)(BootEncStatus.EncryptedAreaEnd - BootEncStatus.EncryptedAreaStart)) + 1;

-}

-

-

-static void PopulateSysEncContextMenu (HMENU popup, BOOL bToolsOnly)

-{

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- }

- catch (Exception &e)

- {

- e.Show (MainDlg);

- }

-

- if (!bToolsOnly && !IsHiddenOSRunning())

- {

- if (SysEncryptionOrDecryptionRequired ())

- {

- if (!BootEncStatus.SetupInProgress)

- AppendMenuW (popup, MF_STRING, IDM_SYSENC_RESUME, GetString ("IDM_SYSENC_RESUME"));

-

- if (SystemEncryptionStatus != SYSENC_STATUS_DECRYPTING)

- AppendMenuW (popup, MF_STRING, IDM_PERMANENTLY_DECRYPT_SYS, GetString ("PERMANENTLY_DECRYPT"));

-

- AppendMenuW (popup, MF_STRING, IDM_ENCRYPT_SYSTEM_DEVICE, GetString ("ENCRYPT"));

- AppendMenu (popup, MF_SEPARATOR, 0, L"");

- }

- }

-

- AppendMenuW (popup, MF_STRING, IDM_CHANGE_SYS_PASSWORD, GetString ("IDM_CHANGE_SYS_PASSWORD"));

- // AppendMenuW (popup, MF_STRING, IDM_CHANGE_SYS_HEADER_KEY_DERIV_ALGO, GetString ("IDM_CHANGE_SYS_HEADER_KEY_DERIV_ALGO"));

-

- AppendMenu (popup, MF_SEPARATOR, 0, L"");

- AppendMenuW (popup, MF_STRING, IDM_SYS_ENC_SETTINGS, GetString ("IDM_SYS_ENC_SETTINGS"));

-

- if (!IsHiddenOSRunning())

- {

- AppendMenu (popup, MF_SEPARATOR, 0, L"");

- AppendMenuW (popup, MF_STRING, IDM_CREATE_RESCUE_DISK, GetString ("IDM_CREATE_RESCUE_DISK"));

- AppendMenuW (popup, MF_STRING, IDM_VERIFY_RESCUE_DISK, GetString ("IDM_VERIFY_RESCUE_DISK"));

- AppendMenuW (popup, MF_STRING, IDM_VERIFY_RESCUE_DISK_ISO, GetString ("IDM_VERIFY_RESCUE_DISK_ISO"));

- }

-

- if (!bToolsOnly)

- {

- if (SysDriveOrPartitionFullyEncrypted (FALSE) && !IsHiddenOSRunning())

- {

- AppendMenu (popup, MF_SEPARATOR, 0, L"");

- AppendMenuW (popup, MF_STRING, IDM_PERMANENTLY_DECRYPT_SYS, GetString ("PERMANENTLY_DECRYPT"));

- }

- AppendMenu (popup, MF_SEPARATOR, 0, L"");

- AppendMenuW (popup, MF_STRING, IDM_VOLUME_PROPERTIES, GetString ("IDPM_PROPERTIES"));

- }

-}

-

-

-// WARNING: This function may take a long time to complete. To prevent data corruption, it MUST be called before

-// mounting a partition (as a regular volume) that is within key scope of system encryption.

-// Returns TRUE if the partition can be mounted as a partition within key scope of inactive system encryption.

-// If devicePath is empty, the currently selected partition in the GUI is checked.

-BOOL CheckSysEncMountWithoutPBA (HWND hwndDlg, const wchar_t *devicePath, BOOL quiet)

-{

- BOOL tmpbDevice;

- wchar_t szDevicePath [TC_MAX_PATH+1];

- wchar_t szDiskFile [TC_MAX_PATH+1];

-

- if (wcslen (devicePath) < 2)

- {

- GetVolumePath (MainDlg, szDevicePath, ARRAYSIZE (szDevicePath));

- CreateFullVolumePath (szDiskFile, sizeof(szDiskFile), szDevicePath, &tmpbDevice);

-

- if (!tmpbDevice)

- {

- if (!quiet)

- Warning ("NO_SYSENC_PARTITION_SELECTED", hwndDlg);

-

- return FALSE;

- }

-

- if (LOWORD (GetSelectedLong (GetDlgItem (MainDlg, IDC_DRIVELIST))) != TC_MLIST_ITEM_FREE)

- {

- if (!quiet)

- Warning ("SELECT_FREE_DRIVE", hwndDlg);

-

- return FALSE;

- }

- }

- else

- StringCbCopyW (szDevicePath, sizeof(szDevicePath), devicePath);

-

- wchar_t *partionPortion = wcsrchr (szDevicePath, L'\\');

-

- if (!partionPortion

- || !_wcsicmp (partionPortion, L"\\Partition0"))

- {

- // Only partitions are supported (not whole drives)

- if (!quiet)

- Warning ("NO_SYSENC_PARTITION_SELECTED", hwndDlg);

-

- return FALSE;

- }

-

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

-

- if (BootEncStatus.DriveMounted)

- {

- int retCode = 0;

- int driveNo;

- wchar_t parentDrivePath [TC_MAX_PATH+1];

-

- if (swscanf (szDevicePath, L"\\Device\\Harddisk%d\\Partition", &driveNo) != 1)

- {

- if (!quiet)

- Error ("INVALID_PATH", hwndDlg);

-

- return FALSE;

- }

-

- StringCbPrintfW (parentDrivePath,

- sizeof (parentDrivePath),

- L"\\Device\\Harddisk%d\\Partition0",

- driveNo);

-

- WaitCursor ();

-

- // This is critical (re-mounting a mounted system volume as a normal volume could cause data corruption)

- // so we force the slower but reliable method

- retCode = IsSystemDevicePath (parentDrivePath, MainDlg, TRUE);

-

- NormalCursor();

-

- if (retCode != 2)

- return TRUE;

- else

- {

- // The partition is located on active system drive

-

- if (!quiet)

- Warning ("MOUNT_WITHOUT_PBA_VOL_ON_ACTIVE_SYSENC_DRIVE", hwndDlg);

-

- return FALSE;

- }

- }

- else

- return TRUE;

- }

- catch (Exception &e)

- {

- NormalCursor();

- e.Show (hwndDlg);

- }

-

- return FALSE;

-}

-

-

-// Returns TRUE if the host drive of the specified partition contains a portion of the TrueCrypt Boot Loader

-// and if the drive is not within key scope of active system encryption (e.g. the system drive of the running OS).

-// If bPrebootPasswordDlgMode is TRUE, this function returns FALSE (because the check would be redundant).

-BOOL TCBootLoaderOnInactiveSysEncDrive (wchar_t *szDevicePath)

-{

- try

- {

- int driveNo;

- wchar_t parentDrivePath [TC_MAX_PATH+1];

-

- if (bPrebootPasswordDlgMode)

- return FALSE;

-

-

- if (swscanf (szDevicePath, L"\\Device\\Harddisk%d\\Partition", &driveNo) != 1)

- return FALSE;

-

- StringCbPrintfW (parentDrivePath,

- sizeof (parentDrivePath),

- L"\\Device\\Harddisk%d\\Partition0",

- driveNo);

-

- BootEncStatus = BootEncObj->GetStatus();

-

- if (BootEncStatus.DriveMounted

- && IsSystemDevicePath (parentDrivePath, MainDlg, FALSE) == 2)

- {

- // The partition is within key scope of active system encryption

- return FALSE;

- }

-

- return ((BOOL) BootEncObj->IsBootLoaderOnDrive (parentDrivePath));

- }

- catch (...)

- {

- return FALSE;

- }

-

-}

-

-

-BOOL SelectItem (HWND hTree, wchar_t nLetter)

-{

- if (nLetter == 0)

- {

- // The caller specified an invalid drive letter (typically because it is unknown).

- // Find out which drive letter is currently selected in the list and use it.

- nLetter = (wchar_t) (HIWORD (GetSelectedLong (hTree)));

- }

-

- int i;

- LVITEM item;

-

- for (i = 0; i < ListView_GetItemCount(hTree); i++)

- {

- memset(&item, 0, sizeof(LVITEM));

- item.mask = LVIF_PARAM;

- item.iItem = i;

-

- if (ListView_GetItem (hTree, &item) == FALSE)

- return FALSE;

- else

- {

- if (HIWORD (item.lParam) == nLetter)

- {

- memset(&item, 0, sizeof(LVITEM));

- item.state = LVIS_FOCUSED|LVIS_SELECTED;

- item.stateMask = LVIS_FOCUSED|LVIS_SELECTED;

- item.mask = LVIF_STATE;

- item.iItem = i;

- SendMessage(hTree, LVM_SETITEMSTATE, i, (LPARAM) &item);

- return TRUE;

- }

- }

- }

-

- return TRUE;

-}

-

-

-static void LaunchVolCreationWizard (HWND hwndDlg, const wchar_t *arg)

-{

- wchar_t t[TC_MAX_PATH + 1024] = {L'"',0};

- wchar_t *tmp;

-

- GetModuleFileName (NULL, t+1, ARRAYSIZE(t)-1);

-

- tmp = wcsrchr (t, L'\\');

- if (tmp)

- {

- STARTUPINFO si;

- PROCESS_INFORMATION pi;

- wchar_t formatExeName[64];

- wchar_t* suffix = NULL;

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

-

- StringCbCopyW (formatExeName, sizeof (formatExeName), L"\\VeraCrypt Format");

-

- // check if there is a suffix in VeraCrypt file name

- // in order to use the same for "VeraCrypt Format"

- suffix = wcsrchr (tmp + 1, L'-');

- if (suffix)

- {

- StringCbCatW (formatExeName, sizeof (formatExeName), suffix);

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

- }

- else

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

-

- *tmp = 0;

- StringCbCatW (t, sizeof(t), formatExeName);

-

- if (!FileExists(t))

- Error ("VOL_CREATION_WIZARD_NOT_FOUND", hwndDlg); // Display a user-friendly error message and advise what to do

-

- if (wcslen (arg) > 0)

- {

- StringCbCatW (t, sizeof(t), L" ");

- StringCbCatW (t, sizeof(t), arg);

- }

-

- if (!CreateProcess (NULL, (LPWSTR) t, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL, NULL, &si, &pi))

- {

- handleWin32Error (hwndDlg, SRC_POS);

- }

- else

- {

- CloseHandle (pi.hProcess);

- CloseHandle (pi.hThread);

- }

- }

-}

-

-static void LaunchVolExpander (HWND hwndDlg)

-{

- wchar_t t[TC_MAX_PATH + TC_MAX_PATH] = {L'"',0};

- wchar_t *tmp;

-

- GetModuleFileName (NULL, t+1, ARRAYSIZE(t)-1);

-

- tmp = wcsrchr (t, L'\\');

- if (tmp)

- {

- wchar_t expanderExeName[64];

- wchar_t* suffix = NULL;

-

- StringCbCopyW (expanderExeName, sizeof (expanderExeName), L"\\VeraCryptExpander");

-

- // check if there is a suffix in VeraCrypt file name

- // in order to use the same for "VeraCrypt Format"

- suffix = wcsrchr (tmp + 1, L'-');

- if (suffix)

- {

- StringCbCatW (expanderExeName, sizeof (expanderExeName), suffix);

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

- }

- else

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

-

- *tmp = 0;

- StringCbCatW (t, sizeof(t), expanderExeName);

-

- if (!FileExists(t))

- Error ("VOL_EXPANDER_NOT_FOUND", hwndDlg); // Display a user-friendly error message and advise what to do

- else if (((int)ShellExecuteW (NULL, (!IsAdmin() && IsUacSupported()) ? L"runas" : L"open", t, NULL, NULL, SW_SHOW)) <= 32)

- {

- handleWin32Error (hwndDlg, SRC_POS);

- }

- }

-}

-

-

-// Fills drive list

-// drive>0 = update only the corresponding drive subitems

-void LoadDriveLetters (HWND hwndDlg, HWND hTree, int drive)

-{

- // Remember the top-most visible item

- int lastTopMostVisibleItem = ListView_GetTopIndex (hTree);

-

- wchar_t *szDriveLetters[]=

- {L"A:", L"B:", L"C:", L"D:",

- L"E:", L"F:", L"G:", L"H:", L"I:", L"J:", L"K:",

- L"L:", L"M:", L"N:", L"O:", L"P:", L"Q:", L"R:",

- L"S:", L"T:", L"U:", L"V:", L"W:", L"X:", L"Y:",

- L"Z:"};

-

- DWORD dwResult;

- BOOL bResult;

- DWORD dwUsedDrives;

- MOUNT_LIST_STRUCT driver;

- VOLUME_PROPERTIES_STRUCT propSysEnc;

- wchar_t sysDriveLetter = 0;

-

- BOOL bSysEnc = FALSE;

- BOOL bWholeSysDriveEncryption = FALSE;

-

- LVITEM listItem;

- int item = 0;

- char i;

-

- try

- {

- BootEncStatus = BootEncObj->GetStatus();

- if (bSysEnc = BootEncStatus.DriveMounted)

- {

- BootEncObj->GetVolumeProperties (&propSysEnc);

- }

- }

- catch (...)

- {

- bSysEnc = FALSE;

- }

-

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

- bResult = DeviceIoControl (hDriver, TC_IOCTL_GET_MOUNTED_VOLUMES, &driver,

- sizeof (driver), &driver, sizeof (driver), &dwResult,

- NULL);

- memcpy (&LastKnownMountList, &driver, sizeof (driver));

-

- if (bResult == FALSE)

- {

- KillTimer (MainDlg, TIMER_ID_MAIN);

- handleWin32Error (hTree, SRC_POS);

- AbortProcessSilent();

- }

-

- LastKnownLogicalDrives = dwUsedDrives = GetUsedLogicalDrives ();

- if (dwUsedDrives == 0)

- Warning ("DRIVELETTERS", hwndDlg);

-

- if(drive == 0)

- ListView_DeleteAllItems(hTree);

-

- if (bSysEnc)

- {

- bWholeSysDriveEncryption = WholeSysDriveEncryption (TRUE);

-

- sysDriveLetter = GetSystemDriveLetter ();

- }

-

- /* System drive */

-

- if (bWholeSysDriveEncryption)

- {

- int curDrive = 0;

-

- if (drive > 0)

- {

- LVITEM tmp;

- memset(&tmp, 0, sizeof(LVITEM));

- tmp.mask = LVIF_PARAM;

- tmp.iItem = item;

- if (ListView_GetItem (hTree, &tmp))

- curDrive = HIWORD(tmp.lParam);

- }

-

- {

- wchar_t szTmp[1024];

- wchar_t szTmpW[1024];

-

- memset(&listItem, 0, sizeof(listItem));

-

- listItem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;

- listItem.iImage = 2;

- listItem.iItem = item++;

-

- listItem.pszText = szTmp;

- szTmp[0] = L' ';

- szTmp[1] = 0;

-

- listItem.lParam = MAKELONG (TC_MLIST_ITEM_SYS_DRIVE, ENC_SYSDRIVE_PSEUDO_DRIVE_LETTER);

-

- if(drive == 0)

- ListView_InsertItem (hTree, &listItem);

- else

- ListView_SetItem (hTree, &listItem);

-

- listItem.mask=LVIF_TEXT;

-

- // Fully encrypted

- if (SysDriveOrPartitionFullyEncrypted (TRUE))

- {

- StringCbCopyW (szTmpW, sizeof(szTmpW), GetString ("SYSTEM_DRIVE"));

- }

- else

- {

- // Partially encrypted

-

- if (BootEncStatus.SetupInProgress)

- {

- // Currently encrypting/decrypting

-

- if (BootEncStatus.SetupMode != SetupDecryption)

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_DRIVE_ENCRYPTING"),

- (double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0);

- }

- else

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_DRIVE_DECRYPTING"),

- 100.0 - ((double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0));

- }

- }

- else

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_DRIVE_PARTIALLY_ENCRYPTED"),

- (double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0);

- }

- }

-

- ListSubItemSet (hTree, listItem.iItem, 1, szTmpW);

-

- GetSizeString (GetSysEncDeviceSize(TRUE), szTmpW, sizeof(szTmpW));

- ListSubItemSet (hTree, listItem.iItem, 2, szTmpW);

-

- if (propSysEnc.ea >= EAGetFirst() && propSysEnc.ea <= EAGetCount())

- {

- EAGetName (szTmp, propSysEnc.ea, 1);

- }

- else

- {

- szTmp[0] = L'?';

- szTmp[1] = 0;

- }

- listItem.iSubItem = 3;

- ListView_SetItem (hTree, &listItem);

-

- ListSubItemSet (hTree, listItem.iItem, 4, GetString (IsHiddenOSRunning() ? "HIDDEN" : "SYSTEM_VOLUME_TYPE_ADJECTIVE"));

- }

- }

-

- /* Drive letters */

-

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

- {

- int curDrive = 0;

-

- BOOL bSysEncPartition = (bSysEnc && !bWholeSysDriveEncryption && sysDriveLetter == *((wchar_t *) szDriveLetters[i]));

-

- if (drive > 0)

- {

- LVITEM tmp;

- memset(&tmp, 0, sizeof(LVITEM));

- tmp.mask = LVIF_PARAM;

- tmp.iItem = item;

- if (ListView_GetItem (hTree, &tmp))

- curDrive = HIWORD(tmp.lParam);

- }

-

- if (driver.ulMountedDrives & (1 << i)

- || bSysEncPartition)

- {

- wchar_t szTmp[1024];

- wchar_t szTmpW[1024];

- wchar_t *ws;

-

- memset(&listItem, 0, sizeof(listItem));

-

- listItem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;

- listItem.iImage = bSysEncPartition ? 2 : 1;

- listItem.iItem = item++;

-

- if (drive > 0 && drive != curDrive)

- continue;

-

- listItem.lParam = MAKELONG (

- bSysEncPartition ? TC_MLIST_ITEM_SYS_PARTITION : TC_MLIST_ITEM_NONSYS_VOL,

- i + L'A');

-

- listItem.pszText = szDriveLetters[i];

-

- if (drive == 0)

- ListView_InsertItem (hTree, &listItem);

- else

- ListView_SetItem (hTree, &listItem);

-

- listItem.mask=LVIF_TEXT;

- listItem.pszText = szTmp;

-

- if (bSysEncPartition)

- {

- // Fully encrypted

- if (SysDriveOrPartitionFullyEncrypted (TRUE))

- {

- StringCbCopyW (szTmpW, sizeof(szTmpW), GetString (IsHiddenOSRunning() ? "HIDDEN_SYSTEM_PARTITION" : "SYSTEM_PARTITION"));

- }

- else

- {

- // Partially encrypted

-

- if (BootEncStatus.SetupInProgress)

- {

- // Currently encrypting/decrypting

-

- if (BootEncStatus.SetupMode != SetupDecryption)

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_PARTITION_ENCRYPTING"),

- (double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0);

- }

- else

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_PARTITION_DECRYPTING"),

- 100.0 - ((double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0));

- }

- }

- else

- {

- StringCbPrintfW (szTmpW,

- sizeof szTmpW,

- GetString ("SYSTEM_PARTITION_PARTIALLY_ENCRYPTED"),

- (double) GetSysEncDeviceEncryptedPartSize (TRUE) / (double) GetSysEncDeviceSize (TRUE) * 100.0);

- }

- }

-

- ListSubItemSet (hTree, listItem.iItem, 1, szTmpW);

- }

- else

- {

- wchar_t *path = driver.wszVolume[i];

-

- if (wmemcmp (path, L"\\??\\", 4) == 0)

- path += 4;

-

- listItem.iSubItem = 1;

-

- // first check label used for mounting. If empty, look for it in favorites.

- bool useInExplorer = false;

- wstring label = (wchar_t *) driver.wszLabel[i];

- if (label.empty())

- label = GetFavoriteVolumeLabel (path, useInExplorer);

- if (!label.empty())

- ListSubItemSet (hTree, listItem.iItem, 1, (wchar_t *) label.c_str());

- else

- ListSubItemSet (hTree, listItem.iItem, 1, (wchar_t *) FitPathInGfxWidth (hTree, hUserFont, ListView_GetColumnWidth (hTree, 1) - GetTextGfxWidth (hTree, L"___", hUserFont), path).c_str());

- }

-

- GetSizeString (bSysEncPartition ? GetSysEncDeviceSize(TRUE) : driver.diskLength[i], szTmpW, sizeof(szTmpW));

- ListSubItemSet (hTree, listItem.iItem, 2, szTmpW);

-

- EAGetName (szTmp, bSysEncPartition ? propSysEnc.ea : driver.ea[i], 1);

- listItem.iSubItem = 3;

- ListView_SetItem (hTree, &listItem);

-

- if (bSysEncPartition)

- {

- ws = GetString (IsHiddenOSRunning() ? "HIDDEN" : "SYSTEM_VOLUME_TYPE_ADJECTIVE");

- VolumeNotificationsList.bHidVolDamagePrevReported[i] = FALSE;

- ListSubItemSet (hTree, listItem.iItem, 4, ws);

- }

- else

- {

- switch (driver.volumeType[i])

- {

- case PROP_VOL_TYPE_NORMAL:

- ws = GetString ("NORMAL");

- break;

- case PROP_VOL_TYPE_HIDDEN:

- ws = GetString ("HIDDEN");

- break;

- case PROP_VOL_TYPE_OUTER:

- ws = GetString ("OUTER"); // Normal/outer volume (hidden volume protected)

- break;

- case PROP_VOL_TYPE_OUTER_VOL_WRITE_PREVENTED:

- ws = GetString ("OUTER_VOL_WRITE_PREVENTED"); // Normal/outer volume (hidden volume protected AND write denied)

- break;

- default:

- ws = L"?";

- }

-

- if (driver.truecryptMode[i])

- {

- StringCbPrintfW (szTmpW, sizeof(szTmpW), L"TrueCrypt-%s", ws);

- ListSubItemSet (hTree, listItem.iItem, 4, szTmpW);

- }

- else

- ListSubItemSet (hTree, listItem.iItem, 4, ws);

-

- if (driver.volumeType[i] == PROP_VOL_TYPE_OUTER_VOL_WRITE_PREVENTED) // Normal/outer volume (hidden volume protected AND write denied)

- {

- if (!VolumeNotificationsList.bHidVolDamagePrevReported[i])

- {

- wchar_t szTmp[4096];

-

- VolumeNotificationsList.bHidVolDamagePrevReported[i] = TRUE;

- StringCbPrintfW (szTmp, sizeof(szTmp), GetString ("DAMAGE_TO_HIDDEN_VOLUME_PREVENTED"), i+L'A');

- SetForegroundWindow (GetParent(hTree));

- MessageBoxW (GetParent(hTree), szTmp, lpszTitle, MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);

- }

- }

- else

- {

- VolumeNotificationsList.bHidVolDamagePrevReported[i] = FALSE;

- }

- }

- }

- else

- {

- VolumeNotificationsList.bHidVolDamagePrevReported[i] = FALSE;

-

- if (!(dwUsedDrives & 1 << i))

- {

- if(drive > 0 && drive != HIWORD (GetSelectedLong (hTree)))

- {

- item++;

- continue;

- }

-

- memset(&listItem,0,sizeof(listItem));

-

- listItem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;

- listItem.iImage = 0;

- listItem.iItem = item++;

- listItem.pszText = szDriveLetters[i];

- listItem.lParam = MAKELONG (TC_MLIST_ITEM_FREE, i + 'A');

-

- if(drive == 0)

- ListView_InsertItem (hTree, &listItem);

- else

- ListView_SetItem (hTree, &listItem);

-

- listItem.mask=LVIF_TEXT;

- listItem.pszText = L"";

- listItem.iSubItem = 1;

- ListView_SetItem (hTree, &listItem);

- listItem.iSubItem = 2;

- ListView_SetItem (hTree, &listItem);

- listItem.iSubItem = 3;

- ListView_SetItem (hTree, &listItem);

- listItem.iSubItem = 4;

- ListView_SetItem (hTree, &listItem);

-

- }

- }

- }

-

- // Restore the original scroll position (the topmost item that was visible when we were called) and the

- // last selected item.

- SetListScrollHPos (hTree, lastTopMostVisibleItem);

- SelectItem (hTree, 0);

-}

-

-static void PasswordChangeEnable (HWND hwndDlg, int button, int passwordId, BOOL keyFilesEnabled,

- int newPasswordId, int newVerifyId, BOOL newKeyFilesEnabled)

-{

- char password[MAX_PASSWORD + 1];

- char newPassword[MAX_PASSWORD + 1];

- char newVerify[MAX_PASSWORD + 1];

- wchar_t tmp[MAX_PASSWORD + 1];

- BOOL bEnable = TRUE;

- int passwordUtf8Len, newPasswordUtf8Len, newVerifyUtf8Len;

-

- GetWindowText (GetDlgItem (hwndDlg, passwordId), tmp, ARRAYSIZE (tmp));

- passwordUtf8Len = WideCharToMultiByte (CP_UTF8, 0, tmp, -1, password, sizeof (password), NULL, NULL);

-

- if (pwdChangeDlgMode == PCDM_CHANGE_PKCS5_PRF)

- newKeyFilesEnabled = keyFilesEnabled;

-

- switch (pwdChangeDlgMode)

- {

- case PCDM_REMOVE_ALL_KEYFILES_FROM_VOL:

- case PCDM_ADD_REMOVE_VOL_KEYFILES:

- case PCDM_CHANGE_PKCS5_PRF:

- memcpy (newPassword, password, sizeof (newPassword));

- memcpy (newVerify, password, sizeof (newVerify));

- newPasswordUtf8Len = passwordUtf8Len;

- newVerifyUtf8Len = passwordUtf8Len;

- break;

-

- default:

- GetWindowText (GetDlgItem (hwndDlg, newPasswordId), tmp, ARRAYSIZE (tmp));

- newPasswordUtf8Len = WideCharToMultiByte (CP_UTF8, 0, tmp, -1, newPassword, sizeof (newPassword), NULL, NULL);

- GetWindowText (GetDlgItem (hwndDlg, newVerifyId), tmp, ARRAYSIZE (tmp));

- newVerifyUtf8Len = WideCharToMultiByte (CP_UTF8, 0, tmp, -1, newVerify, sizeof (newVerify), NULL, NULL);

-

- }

-

- if (passwordUtf8Len <= 0 || (!keyFilesEnabled && ((passwordUtf8Len - 1) < MIN_PASSWORD)))

- bEnable = FALSE;

- else if (strcmp (newPassword, newVerify) != 0)

- bEnable = FALSE;

- else if ((newPasswordUtf8Len <= 0) || (!newKeyFilesEnabled && ((newPasswordUtf8Len - 1) < MIN_PASSWORD)))

- bEnable = FALSE;

-

- burn (password, sizeof (password));

- burn (newPassword, sizeof (newPassword));

- burn (newVerify, sizeof (newVerify));

- burn (tmp, sizeof (tmp));

-

- EnableWindow (GetDlgItem (hwndDlg, button), bEnable);

-}

-

-// implementation for support of change password operation in wait dialog mechanism

-

-typedef struct

-{

- Password *oldPassword;

- int old_pkcs5;

- int old_pim;

- Password *newPassword;

- int pkcs5;

- int pim;

- int wipePassCount;

- BOOL truecryptMode;

- int* pnStatus;

-} ChangePwdThreadParam;

-

-void CALLBACK ChangePwdWaitThreadProc(void* pArg, HWND hwndDlg)

-{

- ChangePwdThreadParam* pThreadParam = (ChangePwdThreadParam*) pArg;

-

- if (bSysEncPwdChangeDlgMode)

- {

- // System

-

- try

- {

- VOLUME_PROPERTIES_STRUCT properties;

- BootEncObj->GetVolumeProperties(&properties);

- pThreadParam->old_pkcs5 = properties.pkcs5;

- }

- catch(...)

- {}

-

- pThreadParam->pkcs5 = 0; // PKCS-5 PRF unchanged (currently we can't change PRF of system encryption)

-

- try

- {

- *pThreadParam->pnStatus = BootEncObj->ChangePassword (pThreadParam->oldPassword, pThreadParam->old_pkcs5, pThreadParam->old_pim, pThreadParam->newPassword, pThreadParam->pkcs5, pThreadParam->pim, pThreadParam->wipePassCount, hwndDlg);

- }

- catch (Exception &e)

- {

- e.Show (hwndDlg);

- *(pThreadParam->pnStatus) = ERR_OS_ERROR;

- }

- }

- else

- {

- // Non-system

-

- *pThreadParam->pnStatus = ChangePwd (szFileName, pThreadParam->oldPassword, pThreadParam->old_pkcs5, pThreadParam->old_pim, pThreadParam->truecryptMode, pThreadParam->newPassword, pThreadParam->pkcs5, pThreadParam->pim, pThreadParam->wipePassCount, hwndDlg);

-

- if (*pThreadParam->pnStatus == ERR_OS_ERROR

- && GetLastError () == ERROR_ACCESS_DENIED

- && IsUacSupported ()

- && IsVolumeDeviceHosted (szFileName))

- {

- *pThreadParam->pnStatus = UacChangePwd (szFileName, pThreadParam->oldPassword, pThreadParam->old_pkcs5, pThreadParam->old_pim, pThreadParam->truecryptMode, pThreadParam->newPassword, pThreadParam->pkcs5, pThreadParam->pim, pThreadParam->wipePassCount, hwndDlg);

- }

- }

-}

-

-// implementation for support of backup header operation in wait dialog mechanism

-

-typedef struct

-{

- BOOL bRequireConfirmation;

- wchar_t *lpszVolume;

- size_t cchVolume;

- int* iResult;

-} BackupHeaderThreadParam;

-

-void CALLBACK BackupHeaderWaitThreadProc(void* pArg, HWND hwndDlg)

-{

- BackupHeaderThreadParam* pThreadParam = (BackupHeaderThreadParam*) pArg;

-

- if (TranslateVolumeID (hwndDlg, pThreadParam->lpszVolume, pThreadParam->cchVolume))

- {

- if (!IsAdmin () && IsUacSupported () && IsVolumeDeviceHosted (pThreadParam->lpszVolume))

- *(pThreadParam->iResult) = UacBackupVolumeHeader (hwndDlg, pThreadParam->bRequireConfirmation, pThreadParam->lpszVolume);

- else

- *(pThreadParam->iResult) = BackupVolumeHeader (hwndDlg, pThreadParam->bRequireConfirmation, pThreadParam->lpszVolume);

- }

- else

- *(pThreadParam->iResult) = ERR_OS_ERROR;

-}

-

-// implementation for support of restoring header operation in wait dialog mechanism

-

-typedef struct

-{

- wchar_t *lpszVolume;

- size_t cchVolume;

- int* iResult;

-} RestoreHeaderThreadParam;

-

-void CALLBACK RestoreHeaderWaitThreadProc(void* pArg, HWND hwndDlg)

-{

- RestoreHeaderThreadParam* pThreadParam = (RestoreHeaderThreadParam*) pArg;

-

- if (TranslateVolumeID (hwndDlg, pThreadParam->lpszVolume, pThreadParam->cchVolume))

- {

- if (!IsAdmin () && IsUacSupported () && IsVolumeDeviceHosted (pThreadParam->lpszVolume))

- *(pThreadParam->iResult) = UacRestoreVolumeHeader (hwndDlg, pThreadParam->lpszVolume);

- else

- *(pThreadParam->iResult) = RestoreVolumeHeader (hwndDlg, pThreadParam->lpszVolume);

- }

- else

- *(pThreadParam->iResult) = ERR_OS_ERROR;

-}

-

-/* Except in response to the WM_INITDIALOG message, the dialog box procedure

- should return nonzero if it processes the message, and zero if it does

- not. - see DialogProc */

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

-{

- static KeyFilesDlgParam newKeyFilesParam;

- static BOOL PimValueChangedWarning = FALSE;

- static int* NewPimValuePtr = NULL;

-

- WORD lw = LOWORD (wParam);

- WORD hw = HIWORD (wParam);

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- LPARAM nIndex, nSelectedIndex = 0;

- HWND hComboBox = GetDlgItem (hwndDlg, IDC_PKCS5_OLD_PRF_ID);

- int i;

- WipeAlgorithmId headerWipeMode = TC_WIPE_3_DOD_5220;

- int EffectiveVolumePkcs5 = CmdVolumePkcs5;

- BOOL EffectiveVolumeTrueCryptMode = CmdVolumeTrueCryptMode;

- int EffectiveVolumePim = CmdVolumePim;

-

- /* Priority is given to command line parameters

- * Default values used only when nothing specified in command line

- */

- if (EffectiveVolumePkcs5 == 0)

- EffectiveVolumePkcs5 = DefaultVolumePkcs5;

- if (!EffectiveVolumeTrueCryptMode)

- EffectiveVolumeTrueCryptMode = DefaultVolumeTrueCryptMode;

-

- NewPimValuePtr = (int*) lParam;

-

- PimValueChangedWarning = FALSE;

-

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

- if (NewPimValuePtr)

- {

- /* we are in the case of a volume. Store its name to use it in the key file dialog

- * this will help avoid using the current container file as a key file

- */

- StringCbCopyW (newKeyFilesParam.VolumeFileName, sizeof (newKeyFilesParam.VolumeFileName), szFileName);

- }

-

- SetWindowTextW (hwndDlg, GetString ("IDD_PASSWORDCHANGE_DLG"));

- LocalizeDialog (hwndDlg, "IDD_PASSWORDCHANGE_DLG");

-

- SendMessage (GetDlgItem (hwndDlg, IDC_OLD_PASSWORD), EM_LIMITTEXT, MAX_PASSWORD, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_PASSWORD), EM_LIMITTEXT, MAX_PASSWORD, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_VERIFY), EM_LIMITTEXT, MAX_PASSWORD, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_OLD_PIM), EM_LIMITTEXT, MAX_PIM, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_PIM), EM_LIMITTEXT, MAX_PIM, 0);

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

-

- SetCheckBox (hwndDlg, IDC_ENABLE_KEYFILES, KeyFilesEnable);

- EnableWindow (GetDlgItem (hwndDlg, IDC_KEYFILES), TRUE);

- EnableWindow (GetDlgItem (hwndDlg, IDC_NEW_KEYFILES), TRUE);

-

- /* Add PRF algorithm list for current password */

- SendMessage (hComboBox, CB_RESETCONTENT, 0, 0);

-

- nIndex = SendMessageW (hComboBox, CB_ADDSTRING, 0, (LPARAM) GetString ("AUTODETECTION"));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) 0);

-

- for (i = FIRST_PRF_ID; i <= LAST_PRF_ID; i++)

- {

- nIndex = SendMessage (hComboBox, CB_ADDSTRING, 0, (LPARAM) get_pkcs5_prf_name(i));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) i);

- if (i == EffectiveVolumePkcs5)

- {

- nSelectedIndex = nIndex;

- }

- }

-

- SendMessage (hComboBox, CB_SETCURSEL, nSelectedIndex, 0);

-

- /* check TrueCrypt Mode if it was set as default*/

- SetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE, EffectiveVolumeTrueCryptMode);

-

- /* set default PIM if set in the command line*/

- if (EffectiveVolumePim > 0)

- {

- SetCheckBox (hwndDlg, IDC_PIM_ENABLE, TRUE);

- ShowWindow (GetDlgItem (hwndDlg, IDC_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_OLD_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_OLD_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_OLD_PIM_HELP), SW_SHOW);

- SetPim (hwndDlg, IDC_OLD_PIM, EffectiveVolumePim);

- }

-

- /* Add PRF algorithm list for new password */

- hComboBox = GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID);

- SendMessage (hComboBox, CB_RESETCONTENT, 0, 0);

-

- nIndex = SendMessageW (hComboBox, CB_ADDSTRING, 0, (LPARAM) GetString ("UNCHANGED"));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) 0);

-

- for (i = FIRST_PRF_ID; i <= LAST_PRF_ID; i++)

- {

- if (!HashIsDeprecated (i))

- {

- nIndex = SendMessage (hComboBox, CB_ADDSTRING, 0, (LPARAM) get_pkcs5_prf_name(i));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) i);

- }

- }

-

- SendMessage (hComboBox, CB_SETCURSEL, 0, 0);

-

- PopulateWipeModeCombo (GetDlgItem (hwndDlg, IDC_WIPE_MODE), FALSE, FALSE, TRUE);

- SelectAlgo (GetDlgItem (hwndDlg, IDC_WIPE_MODE), (int *) &headerWipeMode);

-

- switch (pwdChangeDlgMode)

- {

- case PCDM_CHANGE_PKCS5_PRF:

- SetWindowTextW (hwndDlg, GetString ("IDD_PCDM_CHANGE_PKCS5_PRF"));

- LocalizeDialog (hwndDlg, "IDD_PCDM_CHANGE_PKCS5_PRF");

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

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

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

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

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

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

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

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

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

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

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

- break;

-

- case PCDM_ADD_REMOVE_VOL_KEYFILES:

- SetWindowTextW (hwndDlg, GetString ("IDD_PCDM_ADD_REMOVE_VOL_KEYFILES"));

- LocalizeDialog (hwndDlg, "IDD_PCDM_ADD_REMOVE_VOL_KEYFILES");

- newKeyFilesParam.EnableKeyFiles = TRUE;

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

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

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

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

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

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

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

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

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

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

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

- break;

-

- case PCDM_REMOVE_ALL_KEYFILES_FROM_VOL:

- newKeyFilesParam.EnableKeyFiles = FALSE;

- SetWindowTextW (hwndDlg, GetString ("IDD_PCDM_REMOVE_ALL_KEYFILES_FROM_VOL"));

- LocalizeDialog (hwndDlg, "IDD_PCDM_REMOVE_ALL_KEYFILES_FROM_VOL");

- KeyFilesEnable = TRUE;

- SetCheckBox (hwndDlg, IDC_ENABLE_KEYFILES, TRUE);

- EnableWindow (GetDlgItem (hwndDlg, IDC_KEYFILES), TRUE);

- EnableWindow (GetDlgItem (hwndDlg, IDC_ENABLE_KEYFILES), TRUE);

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

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

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

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

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

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

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

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

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

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

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

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

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

- break;

-

- case PCDM_CHANGE_PASSWORD:

- default:

- // NOP

- break;

- };

-

- if (bSysEncPwdChangeDlgMode)

- {

- /* No support for changing the password of TrueCrypt system partition */

- SetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE, FALSE);

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

-

- ToBootPwdField (hwndDlg, IDC_PASSWORD);

- ToBootPwdField (hwndDlg, IDC_VERIFY);

- ToBootPwdField (hwndDlg, IDC_OLD_PASSWORD);

-

- if ((DWORD) GetKeyboardLayout (NULL) != 0x00000409 && (DWORD) GetKeyboardLayout (NULL) != 0x04090409)

- {

- DWORD keybLayout = (DWORD) LoadKeyboardLayout (L"00000409", KLF_ACTIVATE);

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- Error ("CANT_CHANGE_KEYB_LAYOUT_FOR_SYS_ENCRYPTION", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 0;

- }

-

- bKeyboardLayoutChanged = TRUE;

- }

-

-

- /* for system encryption, we can't change the PRF */

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

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

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

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

-

- if (SetTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD, TIMER_INTERVAL_KEYB_LAYOUT_GUARD, NULL) == 0)

- {

- Error ("CANNOT_SET_TIMER", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 0;

- }

-

- newKeyFilesParam.EnableKeyFiles = FALSE;

- KeyFilesEnable = FALSE;

- SetCheckBox (hwndDlg, IDC_ENABLE_KEYFILES, FALSE);

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

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

- }

-

- CheckCapsLock (hwndDlg, FALSE);

-

- return 0;

- }

-

- case WM_TIMER:

- switch (wParam)

- {

- case TIMER_ID_KEYB_LAYOUT_GUARD:

- if (bSysEncPwdChangeDlgMode)

- {

- DWORD keybLayout = (DWORD) GetKeyboardLayout (NULL);

-

- /* Watch the keyboard layout */

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- // Keyboard layout is not standard US

-

- // Attempt to wipe passwords stored in the input field buffers

- wchar_t tmp[MAX_PASSWORD+1];

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp [MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_OLD_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_VERIFY), tmp);

-

- SetWindowText (GetDlgItem (hwndDlg, IDC_OLD_PASSWORD), L"");

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), L"");

- SetWindowText (GetDlgItem (hwndDlg, IDC_VERIFY), L"");

-

- keybLayout = (DWORD) LoadKeyboardLayout (L"00000409", KLF_ACTIVATE);

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- KillTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD);

- Error ("CANT_CHANGE_KEYB_LAYOUT_FOR_SYS_ENCRYPTION", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- bKeyboardLayoutChanged = TRUE;

-

- wchar_t szTmp [4096];

- StringCbCopyW (szTmp, sizeof(szTmp), GetString ("KEYB_LAYOUT_CHANGE_PREVENTED"));

- StringCbCatW (szTmp, sizeof(szTmp), L"\n\n");

- StringCbCatW (szTmp, sizeof(szTmp), GetString ("KEYB_LAYOUT_SYS_ENC_EXPLANATION"));

- MessageBoxW (MainDlg, szTmp, lpszTitle, MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);

- }

-

-

- /* Watch the right Alt key (which is used to enter various characters on non-US keyboards) */

-

- if (bKeyboardLayoutChanged && !bKeybLayoutAltKeyWarningShown)

- {

- if (GetAsyncKeyState (VK_RMENU) < 0)

- {

- bKeybLayoutAltKeyWarningShown = TRUE;

-

- wchar_t szTmp [4096];

- StringCbCopyW (szTmp, sizeof(szTmp), GetString ("ALT_KEY_CHARS_NOT_FOR_SYS_ENCRYPTION"));

- StringCbCatW (szTmp, sizeof(szTmp), L"\n\n");

- StringCbCatW (szTmp, sizeof(szTmp), GetString ("KEYB_LAYOUT_SYS_ENC_EXPLANATION"));

- MessageBoxW (MainDlg, szTmp, lpszTitle, MB_ICONINFORMATION | MB_SETFOREGROUND | MB_TOPMOST);

- }

- }

- }

- return 1;

- }

- return 0;

-

- case WM_CTLCOLORSTATIC:

- {

- if (PimValueChangedWarning && ((HWND)lParam == GetDlgItem(hwndDlg, IDC_PIM_HELP)) )

- {

- // we're about to draw the static

- // set the text colour in (HDC)lParam

- SetBkMode((HDC)wParam,TRANSPARENT);

- SetTextColor((HDC)wParam, RGB(255,0,0));

- // NOTE: per documentation as pointed out by selbie, GetSolidBrush would leak a GDI handle.

- return (BOOL)GetSysColorBrush(COLOR_MENU);

- }

- }

- return 0;

-

- case WM_COMMAND:

- if (lw == IDCANCEL)

- {

- // Attempt to wipe passwords stored in the input field buffers

- wchar_t tmp[MAX_PASSWORD+1];

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp[MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_OLD_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_VERIFY), tmp);

- RestoreDefaultKeyFilesParam ();

-

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- if (hw == EN_CHANGE)

- {

- PasswordChangeEnable (hwndDlg, IDOK,

- IDC_OLD_PASSWORD,

- KeyFilesEnable && FirstKeyFile != NULL,

- IDC_PASSWORD, IDC_VERIFY,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

-

- if ((lw == IDC_OLD_PIM) && IsWindowEnabled (GetDlgItem (hwndDlg, IDC_PIM)))

- {

- wchar_t tmp[MAX_PIM+1] = {0};

- GetDlgItemText (hwndDlg, IDC_OLD_PIM, tmp, MAX_PIM + 1);

- SetDlgItemText (hwndDlg, IDC_PIM, tmp);

- }

-

- if (lw == IDC_PIM)

- {

- if(GetPim (hwndDlg, IDC_OLD_PIM) != GetPim (hwndDlg, IDC_PIM))

- {

- PimValueChangedWarning = TRUE;

- SetDlgItemTextW (hwndDlg, IDC_PIM_HELP, GetString (bSysEncPwdChangeDlgMode? "PIM_SYSENC_CHANGE_WARNING" : "PIM_CHANGE_WARNING"));

- }

- else

- {

- PimValueChangedWarning = FALSE;

- SetDlgItemTextW (hwndDlg, IDC_PIM_HELP, (wchar_t *) GetDictionaryValueByInt (IDC_PIM_HELP));

- }

- }

-

- return 1;

- }

-

- if (lw == IDC_PIM_ENABLE)

- {

- ShowWindow (GetDlgItem (hwndDlg, IDC_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_OLD_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_OLD_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_OLD_PIM_HELP), SW_SHOW);

-

- // check also the "Use PIM" for the new password if it is enabled

- if (IsWindowEnabled (GetDlgItem (hwndDlg, IDC_NEW_PIM_ENABLE)))

- {

- SetCheckBox (hwndDlg, IDC_NEW_PIM_ENABLE, TRUE);

-

- ShowWindow (GetDlgItem (hwndDlg, IDC_NEW_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM_HELP), SW_SHOW);

- }

-

- SetFocus (GetDlgItem (hwndDlg, IDC_OLD_PIM));

-

- return 1;

- }

-

- if (lw == IDC_NEW_PIM_ENABLE)

- {

- ShowWindow (GetDlgItem (hwndDlg, IDC_NEW_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM_HELP), SW_SHOW);

-

- SetFocus (GetDlgItem (hwndDlg, IDC_PIM));

-

- return 1;

- }

-

- if (lw == IDC_KEYFILES)

- {

- if (bSysEncPwdChangeDlgMode)

- {

- Warning ("KEYFILES_NOT_SUPPORTED_FOR_SYS_ENCRYPTION", hwndDlg);

- return 1;

- }

-

- KeyFilesDlgParam param;

- param.EnableKeyFiles = KeyFilesEnable;

- param.FirstKeyFile = FirstKeyFile;

-

- if (IDOK == DialogBoxParamW (hInst,

- MAKEINTRESOURCEW (IDD_KEYFILES), hwndDlg,

- (DLGPROC) KeyFilesDlgProc, (LPARAM) &param))

- {

- KeyFilesEnable = param.EnableKeyFiles;

- FirstKeyFile = param.FirstKeyFile;

-

- SetCheckBox (hwndDlg, IDC_ENABLE_KEYFILES, KeyFilesEnable);

- }

-

- PasswordChangeEnable (hwndDlg, IDOK,

- IDC_OLD_PASSWORD,

- KeyFilesEnable && FirstKeyFile != NULL,

- IDC_PASSWORD, IDC_VERIFY,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

-

- return 1;

- }

-

-

- if (lw == IDC_NEW_KEYFILES)

- {

- if (bSysEncPwdChangeDlgMode)

- {

- Warning ("KEYFILES_NOT_SUPPORTED_FOR_SYS_ENCRYPTION", hwndDlg);

- return 1;

- }

-

- if (IDOK == DialogBoxParamW (hInst,

- MAKEINTRESOURCEW (IDD_KEYFILES), hwndDlg,

- (DLGPROC) KeyFilesDlgProc, (LPARAM) &newKeyFilesParam))

- {

- SetCheckBox (hwndDlg, IDC_ENABLE_NEW_KEYFILES, newKeyFilesParam.EnableKeyFiles);

-

- VerifyPasswordAndUpdate (hwndDlg, GetDlgItem (hwndDlg, IDOK), GetDlgItem (hwndDlg, IDC_PASSWORD),

- GetDlgItem (hwndDlg, IDC_VERIFY), NULL, NULL,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

- }

-

- PasswordChangeEnable (hwndDlg, IDOK,

- IDC_OLD_PASSWORD,

- KeyFilesEnable && FirstKeyFile != NULL,

- IDC_PASSWORD, IDC_VERIFY,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

-

- return 1;

- }

-

- if (lw == IDC_ENABLE_KEYFILES)

- {

- KeyFilesEnable = GetCheckBox (hwndDlg, IDC_ENABLE_KEYFILES);

-

- PasswordChangeEnable (hwndDlg, IDOK,

- IDC_OLD_PASSWORD,

- KeyFilesEnable && FirstKeyFile != NULL,

- IDC_PASSWORD, IDC_VERIFY,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

-

- return 1;

- }

-

- if (lw == IDC_ENABLE_NEW_KEYFILES)

- {

- newKeyFilesParam.EnableKeyFiles = GetCheckBox (hwndDlg, IDC_ENABLE_NEW_KEYFILES);

-

- PasswordChangeEnable (hwndDlg, IDOK,

- IDC_OLD_PASSWORD,

- KeyFilesEnable && FirstKeyFile != NULL,

- IDC_PASSWORD, IDC_VERIFY,

- newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL);

-

- return 1;

- }

-

- if (hw == CBN_SELCHANGE)

- {

- switch (lw)

- {

- case IDC_PKCS5_PRF_ID:

- if (bSysEncPwdChangeDlgMode)

- {

- int new_hash_algo_id = (int) SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETITEMDATA,

- SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETCURSEL, 0, 0), 0);

-

- if (new_hash_algo_id != 0 && !HashForSystemEncryption(new_hash_algo_id))

- {

- int new_hash_algo_id = DEFAULT_HASH_ALGORITHM_BOOT;

- Info ("ALGO_NOT_SUPPORTED_FOR_SYS_ENCRYPTION", hwndDlg);

- SelectAlgo (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), &new_hash_algo_id);

- }

- }

- break;

- }

- return 1;

-

- }

-

- if (lw == IDC_TRUECRYPT_MODE)

- {

- BOOL bEnablePim = GetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE) ? FALSE: TRUE;

- EnableWindow (GetDlgItem (hwndDlg, IDT_OLD_PIM), bEnablePim);

- EnableWindow (GetDlgItem (hwndDlg, IDC_OLD_PIM), bEnablePim);

- EnableWindow (GetDlgItem (hwndDlg, IDC_OLD_PIM_HELP), bEnablePim);

- }

-

- if (lw == IDC_SHOW_PASSWORD_CHPWD_ORI)

- {

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PASSWORD_CHPWD_ORI, IDC_OLD_PASSWORD, IDC_OLD_PIM);

- return 1;

- }

-

- if (lw == IDC_SHOW_PASSWORD_CHPWD_NEW)

- {

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PASSWORD_CHPWD_NEW, IDC_PASSWORD, IDC_VERIFY);

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PASSWORD_CHPWD_NEW, IDC_PIM, 0);

- return 1;

- }

-

- if (lw == IDOK)

- {

- HWND hParent = GetParent (hwndDlg);

- Password oldPassword;

- Password newPassword;

- WipeAlgorithmId headerWiperMode = (WipeAlgorithmId) SendMessage (

- GetDlgItem (hwndDlg, IDC_WIPE_MODE),

- CB_GETITEMDATA,

- SendMessage (GetDlgItem (hwndDlg, IDC_WIPE_MODE), CB_GETCURSEL, 0, 0),

- 0);

- int nStatus;

- int old_pkcs5 = (int) SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_OLD_PRF_ID), CB_GETITEMDATA,

- SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_OLD_PRF_ID), CB_GETCURSEL, 0, 0), 0);

- int pkcs5 = (int) SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETITEMDATA,

- SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETCURSEL, 0, 0), 0);

- BOOL truecryptMode = GetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE);

-

- int old_pim = GetPim (hwndDlg, IDC_OLD_PIM);

- int pim = GetPim (hwndDlg, IDC_PIM);

-

- if (truecryptMode && (old_pkcs5 == SHA256))

- {

- Error ("ALGO_NOT_SUPPORTED_FOR_TRUECRYPT_MODE", hwndDlg);

- return 1;

- }

- else if (truecryptMode && (old_pim != 0))

- {

- Error ("PIM_NOT_SUPPORTED_FOR_TRUECRYPT_MODE", hwndDlg);

- return 1;

- }

-

- if (bSysEncPwdChangeDlgMode && !CheckPasswordCharEncoding (GetDlgItem (hwndDlg, IDC_PASSWORD), NULL))

- {

- Error ("UNSUPPORTED_CHARS_IN_PWD", hwndDlg);

- return 1;

- }

-

- if (bSysEncPwdChangeDlgMode && (pim > MAX_BOOT_PIM_VALUE))

- {

- SetFocus (GetDlgItem(hwndDlg, IDC_PIM));

- Error ("PIM_SYSENC_TOO_BIG", hwndDlg);

- return 1;

- }

-

- if (!bSysEncPwdChangeDlgMode && (pim > MAX_PIM_VALUE))

- {

- SetFocus (GetDlgItem(hwndDlg, IDC_PIM));

- Error ("PIM_TOO_BIG", hwndDlg);

- return 1;

- }

-

- if (pwdChangeDlgMode == PCDM_CHANGE_PKCS5_PRF)

- {

- newKeyFilesParam.EnableKeyFiles = KeyFilesEnable;

- }

- else if (!(newKeyFilesParam.EnableKeyFiles && newKeyFilesParam.FirstKeyFile != NULL)

- && pwdChangeDlgMode == PCDM_CHANGE_PASSWORD)

- {

- if (!CheckPasswordLength (hwndDlg, GetWindowTextLength(GetDlgItem (hwndDlg, IDC_PASSWORD)), pim, bSysEncPwdChangeDlgMode, FALSE, FALSE))

- return 1;

- }

-

- GetVolumePath (hParent, szFileName, ARRAYSIZE (szFileName));

-

- if (GetPassword (hwndDlg, IDC_OLD_PASSWORD, (LPSTR) oldPassword.Text, sizeof (oldPassword.Text), TRUE))

- oldPassword.Length = (unsigned __int32) strlen ((char *) oldPassword.Text);

- else

- {

- return 1;

- }

-

- switch (pwdChangeDlgMode)

- {

- case PCDM_REMOVE_ALL_KEYFILES_FROM_VOL:

- case PCDM_ADD_REMOVE_VOL_KEYFILES:

- case PCDM_CHANGE_PKCS5_PRF:

- memcpy (newPassword.Text, oldPassword.Text, sizeof (newPassword.Text));

- newPassword.Length = (unsigned __int32) strlen ((char *) oldPassword.Text);

- pim = old_pim;

- break;

-

- default:

- if (GetPassword (hwndDlg, IDC_PASSWORD, (LPSTR) newPassword.Text, sizeof (newPassword.Text), TRUE))

- newPassword.Length = (unsigned __int32) strlen ((char *) newPassword.Text);

- else

- return 1;

- }

-

- WaitCursor ();

-

- if (KeyFilesEnable)

- KeyFilesApply (hwndDlg, &oldPassword, FirstKeyFile, szFileName);

-

- if (newKeyFilesParam.EnableKeyFiles)

- {

- if (!KeyFilesApply (hwndDlg, &newPassword, pwdChangeDlgMode == PCDM_CHANGE_PKCS5_PRF ? FirstKeyFile : newKeyFilesParam.FirstKeyFile, szFileName))

- {

- nStatus = ERR_DONT_REPORT;

- goto err;

- }

- }

-

- ChangePwdThreadParam changePwdParam;

- changePwdParam.oldPassword = &oldPassword;

- changePwdParam.old_pkcs5 = old_pkcs5;

- changePwdParam.old_pim = old_pim;

- changePwdParam.newPassword = &newPassword;

- changePwdParam.pkcs5 = pkcs5;

- changePwdParam.pim = pim;

- changePwdParam.wipePassCount = GetWipePassCount(headerWiperMode);

- changePwdParam.pnStatus = &nStatus;

- changePwdParam.truecryptMode = truecryptMode;

-

- ShowWaitDialog(hwndDlg, TRUE, ChangePwdWaitThreadProc, &changePwdParam);

-

-err:

- // notify the caller in case the PIM has changed

- if (NewPimValuePtr)

- {

- if (pim != old_pim)

- *NewPimValuePtr = pim;

- else

- *NewPimValuePtr = -1;

- }

-

- burn (&oldPassword, sizeof (oldPassword));

- burn (&newPassword, sizeof (newPassword));

- burn (&old_pim, sizeof(old_pim));

- burn (&pim, sizeof(pim));

-

- NormalCursor ();

-

- if (nStatus == 0)

- {

- // Attempt to wipe passwords stored in the input field buffers

- wchar_t tmp[MAX_PASSWORD+1];

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp[MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_OLD_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_VERIFY), tmp);

-

- KeyFileRemoveAll (&newKeyFilesParam.FirstKeyFile);

- RestoreDefaultKeyFilesParam ();

-

- if (bSysEncPwdChangeDlgMode)

- {

- KillTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD);

- }

-

- EndDialog (hwndDlg, IDOK);

- }

- return 1;

- }

- return 0;

- }

-

- return 0;

-}

-

-static wchar_t PasswordDlgVolume[MAX_PATH + 1];

-static BOOL PasswordDialogDisableMountOptions;

-static char *PasswordDialogTitleStringId;

-

-/* Except in response to the WM_INITDIALOG message, the dialog box procedure

- should return nonzero if it processes the message, and zero if it does

- not. - see DialogProc */

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

-{

- WORD lw = LOWORD (wParam);

- static Password *szXPwd;

- static int *pkcs5;

- static int *pim;

- static BOOL* truecryptMode;

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- int i, nIndex, defaultPrfIndex = 0;

- szXPwd = ((PasswordDlgParam *) lParam) -> password;

- pkcs5 = ((PasswordDlgParam *) lParam) -> pkcs5;

- pim = ((PasswordDlgParam *) lParam) -> pim;

- truecryptMode = ((PasswordDlgParam *) lParam) -> truecryptMode;

- LocalizeDialog (hwndDlg, "IDD_PASSWORD_DLG");

- DragAcceptFiles (hwndDlg, TRUE);

-

- if (PasswordDialogTitleStringId)

- {

- SetWindowTextW (hwndDlg, GetString (PasswordDialogTitleStringId));

- }

- else if (wcslen (PasswordDlgVolume) > 0)

- {

- wchar_t s[1024];

- RECT rect;

- GetWindowRect (hwndDlg, &rect);

-

- bool useInExplorer = false;

- wstring label = GetFavoriteVolumeLabel (PasswordDlgVolume, useInExplorer);

- if (!label.empty())

- {

- StringCbPrintfW (s, sizeof(s), GetString ("ENTER_PASSWORD_FOR_LABEL"), label.c_str());

- if (useInExplorer)

- StringCbCopyW (mountOptions.Label, sizeof (mountOptions.Label), label.c_str());

- }

- else

- {

- StringCbPrintfW (s, sizeof(s), GetString ("ENTER_PASSWORD_FOR"), L"___");

- StringCbPrintfW (s, sizeof(s), GetString ("ENTER_PASSWORD_FOR"), FitPathInGfxWidth (hwndDlg, WindowTitleBarFont, rect.right - rect.left - GetTextGfxWidth (hwndDlg, s, WindowTitleBarFont), PasswordDlgVolume).c_str());

- }

-

- SetWindowTextW (hwndDlg, s);

- }

-

- /* Populate the PRF algorithms list */

- HWND hComboBox = GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID);

- SendMessage (hComboBox, CB_RESETCONTENT, 0, 0);

-

- nIndex = (int) SendMessageW (hComboBox, CB_ADDSTRING, 0, (LPARAM) GetString ("AUTODETECTION"));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) 0);

-

- for (i = FIRST_PRF_ID; i <= LAST_PRF_ID; i++)

- {

- nIndex = (int) SendMessage (hComboBox, CB_ADDSTRING, 0, (LPARAM) get_pkcs5_prf_name(i));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) i);

- if (*pkcs5 && (*pkcs5 == i))

- defaultPrfIndex = nIndex;

- }

-

- /* make autodetection the default unless a specific PRF was specified in the command line */

- SendMessage (hComboBox, CB_SETCURSEL, defaultPrfIndex, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PASSWORD), EM_LIMITTEXT, MAX_PASSWORD, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_CACHE), BM_SETCHECK, bCacheInDriver ? BST_CHECKED:BST_UNCHECKED, 0);

- SendMessage (GetDlgItem (hwndDlg, IDC_PIM), EM_LIMITTEXT, MAX_PIM, 0);

-

- SetPim (hwndDlg, IDC_PIM, *pim);

-

- /* make PIM field visible if a PIM value has been explicitely specified */

- if (*pim > 0)

- {

- SetCheckBox (hwndDlg, IDC_PIM_ENABLE, TRUE);

- ShowWindow (GetDlgItem (hwndDlg, IDC_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM_HELP), SW_SHOW);

- }

-

- SetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE, KeyFilesEnable);

-

- mountOptions.PartitionInInactiveSysEncScope = bPrebootPasswordDlgMode;

-

- if (bPrebootPasswordDlgMode)

- {

- SendMessage (hwndDlg, TC_APPMSG_PREBOOT_PASSWORD_MODE, 0, 0);

- }

-

- if (PasswordDialogDisableMountOptions)

- {

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

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

- /* Disable TrueCrypt mode option in case of backup/restore header operation */

- SetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE, FALSE);

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

- }

- else if (*truecryptMode)

- {

- /* Check TrueCryptMode if it is enabled on the command line */

- SetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE, TRUE);

- }

-

- if (!SetForegroundWindow (hwndDlg) && (FavoriteMountOnArrivalInProgress || LogOn))

- {

- SetWindowPos (hwndDlg, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);

-

- FLASHWINFO flash;

- flash.cbSize = sizeof (flash);

- flash.dwFlags = FLASHW_ALL | FLASHW_TIMERNOFG;

- flash.dwTimeout = 0;

- flash.hwnd = hwndDlg;

- flash.uCount = 0;

-

- FlashWindowEx (&flash);

-

- SetWindowPos (hwndDlg, HWND_NOTOPMOST, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);

- }

- }

- return 0;

-

- case TC_APPMSG_PREBOOT_PASSWORD_MODE:

- {

- /* Repopulate the PRF algorithms list with algorithms that support system encryption */

- HWND hComboBox = GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID);

- SendMessage (hComboBox, CB_RESETCONTENT, 0, 0);

-

- int i, defaultPrfIndex = 0, nIndex = (int) SendMessageW (hComboBox, CB_ADDSTRING, 0, (LPARAM) GetString ("AUTODETECTION"));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) 0);

-

- for (i = FIRST_PRF_ID; i <= LAST_PRF_ID; i++)

- {

- if (HashForSystemEncryption(i))

- {

- nIndex = (int) SendMessage (hComboBox, CB_ADDSTRING, 0, (LPARAM) get_pkcs5_prf_name(i));

- SendMessage (hComboBox, CB_SETITEMDATA, nIndex, (LPARAM) i);

- if (*pkcs5 && (*pkcs5 == i))

- defaultPrfIndex = nIndex;

- }

- }

-

- /* make autodetection the default unless a specific PRF was specified in the command line */

- SendMessage (hComboBox, CB_SETCURSEL, defaultPrfIndex, 0);

-

- ToBootPwdField (hwndDlg, IDC_PASSWORD);

-

- // Attempt to wipe the password stored in the input field buffer

- wchar_t tmp[MAX_PASSWORD+1];

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp [MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), L"");

-

- StringCbPrintfW (OrigKeyboardLayout, sizeof(OrigKeyboardLayout),L"%08X", (DWORD) GetKeyboardLayout (NULL) & 0xFFFF);

-

- DWORD keybLayout = (DWORD) LoadKeyboardLayout (L"00000409", KLF_ACTIVATE);

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- Error ("CANT_CHANGE_KEYB_LAYOUT_FOR_SYS_ENCRYPTION", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- if (SetTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD, TIMER_INTERVAL_KEYB_LAYOUT_GUARD, NULL) == 0)

- {

- Error ("CANNOT_SET_TIMER", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- if (GetCheckBox (hwndDlg, IDC_SHOW_PASSWORD))

- {

- // simulate hiding password

- SetCheckBox (hwndDlg, IDC_SHOW_PASSWORD, FALSE);

-

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PASSWORD, IDC_PASSWORD, IDC_PIM);

- }

-

- SetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE, FALSE);

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

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

-

- SetPim (hwndDlg, IDC_PIM, *pim);

-

- bPrebootPasswordDlgMode = TRUE;

- }

- return 1;

-

- case WM_TIMER:

- switch (wParam)

- {

- case TIMER_ID_KEYB_LAYOUT_GUARD:

- if (bPrebootPasswordDlgMode)

- {

- DWORD keybLayout = (DWORD) GetKeyboardLayout (NULL);

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- // Keyboard layout is not standard US

-

- // Attempt to wipe the password stored in the input field buffer

- wchar_t tmp[MAX_PASSWORD+1];

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp [MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), L"");

-

- keybLayout = (DWORD) LoadKeyboardLayout (L"00000409", KLF_ACTIVATE);

-

- if (keybLayout != 0x00000409 && keybLayout != 0x04090409)

- {

- KillTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD);

- Error ("CANT_CHANGE_KEYB_LAYOUT_FOR_SYS_ENCRYPTION", hwndDlg);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- wchar_t szTmp [4096];

- StringCbCopyW (szTmp, sizeof(szTmp), GetString ("KEYB_LAYOUT_CHANGE_PREVENTED"));

- StringCbCatW (szTmp, sizeof(szTmp), L"\n\n");

- StringCbCatW (szTmp, sizeof(szTmp), GetString ("KEYB_LAYOUT_SYS_ENC_EXPLANATION"));

- MessageBoxW (MainDlg, szTmp, lpszTitle, MB_ICONWARNING | MB_SETFOREGROUND | MB_TOPMOST);

- }

- }

- return 1;

- }

- return 0;

-

- case WM_COMMAND:

-

- if (lw == IDC_MOUNT_OPTIONS)

- {

- /* Use default PRF specified by the user if any */

- if (mountOptions.ProtectedHidVolPkcs5Prf == 0)

- mountOptions.ProtectedHidVolPkcs5Prf = *pkcs5;

- if (mountOptions.ProtectedHidVolPim == 0)

- mountOptions.ProtectedHidVolPim = *pim;

- DialogBoxParamW (hInst,

- MAKEINTRESOURCEW (IDD_MOUNT_OPTIONS), hwndDlg,

- (DLGPROC) MountOptionsDlgProc, (LPARAM) &mountOptions);

-

- if (!bPrebootPasswordDlgMode && mountOptions.PartitionInInactiveSysEncScope)

- SendMessage (hwndDlg, TC_APPMSG_PREBOOT_PASSWORD_MODE, 0, 0);

-

- return 1;

- }

-

- if (lw == IDC_PIM_ENABLE)

- {

- ShowWindow (GetDlgItem (hwndDlg, IDC_PIM_ENABLE), SW_HIDE);

- ShowWindow (GetDlgItem( hwndDlg, IDT_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM), SW_SHOW);

- ShowWindow (GetDlgItem( hwndDlg, IDC_PIM_HELP), SW_SHOW);

-

- SetFocus (GetDlgItem (hwndDlg, IDC_PIM));

- return 1;

- }

-

- if (lw == IDC_SHOW_PASSWORD)

- {

- HandleShowPasswordFieldAction (hwndDlg, IDC_SHOW_PASSWORD, IDC_PASSWORD, IDC_PIM);

- return 1;

- }

-

- if (lw == IDC_TRUECRYPT_MODE)

- {

- BOOL bEnablePim = GetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE) ? FALSE: TRUE;

- EnableWindow (GetDlgItem (hwndDlg, IDT_PIM), bEnablePim);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PIM), bEnablePim);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PIM_HELP), bEnablePim);

- }

-

- if (lw == IDC_KEY_FILES)

- {

- KeyFilesDlgParam param;

- param.EnableKeyFiles = KeyFilesEnable;

- param.FirstKeyFile = FirstKeyFile;

-

- if (IDOK == DialogBoxParamW (hInst,

- MAKEINTRESOURCEW (IDD_KEYFILES), hwndDlg,

- (DLGPROC) KeyFilesDlgProc, (LPARAM) &param))

- {

- KeyFilesEnable = param.EnableKeyFiles;

- FirstKeyFile = param.FirstKeyFile;

-

- SetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE, KeyFilesEnable);

- }

-

- return 1;

- }

-

- if (lw == IDC_KEYFILES_ENABLE)

- {

- KeyFilesEnable = GetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE);

-

- return 1;

- }

-

- if (lw == IDCANCEL || lw == IDOK)

- {

- wchar_t tmp[MAX_PASSWORD+1];

-

- if (lw == IDOK)

- {

- if (mountOptions.ProtectHiddenVolume && hidVolProtKeyFilesParam.EnableKeyFiles)

- KeyFilesApply (hwndDlg, &mountOptions.ProtectedHidVolPassword, hidVolProtKeyFilesParam.FirstKeyFile, wcslen (PasswordDlgVolume) > 0 ? PasswordDlgVolume : NULL);

-

- if (GetPassword (hwndDlg, IDC_PASSWORD, (LPSTR) szXPwd->Text, MAX_PASSWORD + 1, TRUE))

- szXPwd->Length = (unsigned __int32) strlen ((char *) szXPwd->Text);

- else

- return 1;

-

- bCacheInDriver = IsButtonChecked (GetDlgItem (hwndDlg, IDC_CACHE));

- *pkcs5 = (int) SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETITEMDATA, SendMessage (GetDlgItem (hwndDlg, IDC_PKCS5_PRF_ID), CB_GETCURSEL, 0, 0), 0);

- *truecryptMode = GetCheckBox (hwndDlg, IDC_TRUECRYPT_MODE);

-

- *pim = GetPim (hwndDlg, IDC_PIM);

-

- /* SHA-256 is not supported by TrueCrypt */

- if ( (*truecryptMode)

- && ((*pkcs5 == SHA256) || (mountOptions.ProtectHiddenVolume && mountOptions.ProtectedHidVolPkcs5Prf == SHA256))

- )

- {

- Error ("ALGO_NOT_SUPPORTED_FOR_TRUECRYPT_MODE", hwndDlg);

- return 1;

- }

-

- if ( (*truecryptMode)

- && (*pim != 0)

- )

- {

- Error ("PIM_NOT_SUPPORTED_FOR_TRUECRYPT_MODE", hwndDlg);

- return 1;

- }

- }

-

- // Attempt to wipe password stored in the input field buffer

- wmemset (tmp, L'X', MAX_PASSWORD);

- tmp[MAX_PASSWORD] = 0;

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD), tmp);

- SetWindowText (GetDlgItem (hwndDlg, IDC_PASSWORD_PROT_HIDVOL), tmp);

-

- if (hidVolProtKeyFilesParam.FirstKeyFile != NULL)

- {

- KeyFileRemoveAll (&hidVolProtKeyFilesParam.FirstKeyFile);

- hidVolProtKeyFilesParam.EnableKeyFiles = FALSE;

- }

-

- if (bPrebootPasswordDlgMode)

- {

- KillTimer (hwndDlg, TIMER_ID_KEYB_LAYOUT_GUARD);

-

- // Restore the original keyboard layout

- if (LoadKeyboardLayout (OrigKeyboardLayout, KLF_ACTIVATE | KLF_SUBSTITUTE_OK) == NULL)

- Warning ("CANNOT_RESTORE_KEYBOARD_LAYOUT", hwndDlg);

- }

-

- EndDialog (hwndDlg, lw);

- return 1;

- }

- return 0;

-

- case WM_CONTEXTMENU:

- {

- RECT buttonRect;

- GetWindowRect (GetDlgItem (hwndDlg, IDC_KEY_FILES), &buttonRect);

-

- if (LOWORD (lParam) >= buttonRect.left && LOWORD (lParam) <= buttonRect.right

- && HIWORD (lParam) >= buttonRect.top && HIWORD (lParam) <= buttonRect.bottom)

- {

- // The "Keyfiles" button has been right-clicked

-

- KeyFilesDlgParam param;

- param.EnableKeyFiles = KeyFilesEnable;

- param.FirstKeyFile = FirstKeyFile;

-

- POINT popupPos;

- popupPos.x = buttonRect.left + 2;

- popupPos.y = buttonRect.top + 2;

-

- if (KeyfilesPopupMenu (hwndDlg, popupPos, &param))

- {

- KeyFilesEnable = param.EnableKeyFiles;

- FirstKeyFile = param.FirstKeyFile;

- SetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE, KeyFilesEnable);

- }

- }

- }

- break;

-

- case WM_DROPFILES:

- {

- HDROP hdrop = (HDROP) wParam;

- int i = 0, count = DragQueryFile (hdrop, 0xFFFFFFFF, NULL, 0);

-

- while (count-- > 0)

- {

- KeyFile *kf = (KeyFile *) malloc (sizeof (KeyFile));

- if (kf)

- {

- DragQueryFile (hdrop, i++, kf->FileName, ARRAYSIZE (kf->FileName));

- FirstKeyFile = KeyFileAdd (FirstKeyFile, kf);

- KeyFilesEnable = TRUE;

- }

- }

-

- SetCheckBox (hwndDlg, IDC_KEYFILES_ENABLE, KeyFilesEnable);

- DragFinish (hdrop);

- }

- return 1;

- }

-

- return 0;

-}

-

-static void PreferencesDlgEnableButtons (HWND hwndDlg)

-{

- BOOL back = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_BKG_TASK_ENABLE));

- BOOL idle = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE));

- BOOL installed = !IsNonInstallMode();

- BOOL wtsEnabled = (hWtsLib != NULL) ? TRUE : FALSE;

-

- EnableWindow (GetDlgItem (hwndDlg, IDC_CLOSE_BKG_TASK_WHEN_NOVOL), back && installed);

- EnableWindow (GetDlgItem (hwndDlg, IDT_LOGON), installed);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_LOGON_START), back && installed);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_LOGON_MOUNT_DEVICES), installed);

- EnableWindow (GetDlgItem (hwndDlg, IDT_AUTO_DISMOUNT), back);

- EnableWindow (GetDlgItem (hwndDlg, IDT_AUTO_DISMOUNT_ON), back);

- EnableWindow (GetDlgItem (hwndDlg, IDT_MINUTES), back);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_LOGOFF), back);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SESSION_LOCKED), back && wtsEnabled);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_POWERSAVING), back);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SCREENSAVER), back);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE), back);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE_TIME), back && idle);

- EnableWindow (GetDlgItem (hwndDlg, IDC_PREF_FORCE_AUTO_DISMOUNT), back);

-}

-

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

-{

- static BOOL PreferencesDialogActive = FALSE;

- static HWND ActivePreferencesDialogWindow;

-

- WORD lw = LOWORD (wParam);

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- if (PreferencesDialogActive)

- {

- ShowWindow (ActivePreferencesDialogWindow, SW_SHOW);

- SetForegroundWindow (ActivePreferencesDialogWindow);

- EndDialog (hwndDlg, IDCANCEL);

- return 1;

- }

-

- ActivePreferencesDialogWindow = hwndDlg;

- PreferencesDialogActive = TRUE;

-

- LocalizeDialog (hwndDlg, "IDD_PREFERENCES_DLG");

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_OPEN_EXPLORER), BM_SETCHECK,

- bExplore ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_USE_DIFF_TRAY_ICON_IF_VOL_MOUNTED), BM_SETCHECK,

- bUseDifferentTrayIconIfVolMounted ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PRESERVE_TIMESTAMPS), BM_SETCHECK,

- defaultMountOptions.PreserveTimestamp ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_SHOW_DISCONNECTED_NETWORK_DRIVES), BM_SETCHECK,

- bShowDisconnectedNetworkDrives ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_HIDE_WAITING_DIALOG), BM_SETCHECK,

- bHideWaitingDialog ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_TEMP_CACHE_ON_MULTIPLE_MOUNT), BM_SETCHECK,

- bCacheDuringMultipleMount ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_WIPE_CACHE_ON_EXIT), BM_SETCHECK,

- bWipeCacheOnExit ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_WIPE_CACHE_ON_AUTODISMOUNT), BM_SETCHECK,

- bWipeCacheOnAutoDismount ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_CACHE_PASSWORDS), BM_SETCHECK,

- bCacheInDriver ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_CACHE_PIM), BM_SETCHECK,

- bIncludePimInCache? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_MOUNT_READONLY), BM_SETCHECK,

- defaultMountOptions.ReadOnly ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_MOUNT_REMOVABLE), BM_SETCHECK,

- defaultMountOptions.Removable ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_LOGON_START), BM_SETCHECK,

- bStartOnLogon ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_LOGON_MOUNT_DEVICES), BM_SETCHECK,

- bMountDevicesOnLogon ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_BKG_TASK_ENABLE), BM_SETCHECK,

- bEnableBkgTask ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_CLOSE_BKG_TASK_WHEN_NOVOL), BM_SETCHECK,

- bCloseBkgTaskWhenNoVolumes || IsNonInstallMode() ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_LOGOFF), BM_SETCHECK,

- bDismountOnLogOff ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SESSION_LOCKED), BM_SETCHECK,

- bDismountOnSessionLocked ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_POWERSAVING), BM_SETCHECK,

- bDismountOnPowerSaving ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SCREENSAVER), BM_SETCHECK,

- bDismountOnScreenSaver ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_FORCE_AUTO_DISMOUNT), BM_SETCHECK,

- bForceAutoDismount ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SendMessage (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE), BM_SETCHECK,

- MaxVolumeIdleTime > 0 ? BST_CHECKED:BST_UNCHECKED, 0);

-

- SetDlgItemInt (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE_TIME, abs (MaxVolumeIdleTime), FALSE);

-

- PreferencesDlgEnableButtons (hwndDlg);

- }

- return 0;

-

- case WM_COMMAND:

-

- if (lw == IDC_PREF_BKG_TASK_ENABLE && !IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_BKG_TASK_ENABLE)))

- {

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

- SetCheckBox (hwndDlg, IDC_PREF_BKG_TASK_ENABLE, TRUE);

- }

-

- // Forced dismount disabled warning

- if (lw == IDC_PREF_DISMOUNT_INACTIVE

- || lw == IDC_PREF_DISMOUNT_LOGOFF

- || lw == IDC_PREF_DISMOUNT_SESSION_LOCKED

- || lw == IDC_PREF_DISMOUNT_POWERSAVING

- || lw == IDC_PREF_DISMOUNT_SCREENSAVER

- || lw == IDC_PREF_FORCE_AUTO_DISMOUNT)

- {

- BOOL i = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE));

- BOOL l = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_LOGOFF));

- BOOL sl = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SESSION_LOCKED));

- BOOL p = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_POWERSAVING));

- BOOL s = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SCREENSAVER));

- BOOL q = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_FORCE_AUTO_DISMOUNT));

-

- if (!q)

- {

- if (lw == IDC_PREF_FORCE_AUTO_DISMOUNT && (i || l || sl || p || s))

- {

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

- SetCheckBox (hwndDlg, IDC_PREF_FORCE_AUTO_DISMOUNT, TRUE);

- }

- else if ((lw == IDC_PREF_DISMOUNT_INACTIVE && i

- || lw == IDC_PREF_DISMOUNT_LOGOFF && l

- || lw == IDC_PREF_DISMOUNT_SESSION_LOCKED && sl

- || lw == IDC_PREF_DISMOUNT_POWERSAVING && p

- || lw == IDC_PREF_DISMOUNT_SCREENSAVER && s))

- Warning ("WARN_PREF_AUTO_DISMOUNT", hwndDlg);

- }

-

- if (p && lw == IDC_PREF_DISMOUNT_POWERSAVING)

- Warning ("WARN_PREF_AUTO_DISMOUNT_ON_POWER", hwndDlg);

- }

-

- if (lw == IDCANCEL)

- {

- PreferencesDialogActive = FALSE;

- EndDialog (hwndDlg, lw);

- return 1;

- }

-

- if (lw == IDOK)

- {

- WaitCursor ();

-

- bExplore = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_OPEN_EXPLORER));

- bUseDifferentTrayIconIfVolMounted = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_USE_DIFF_TRAY_ICON_IF_VOL_MOUNTED));

- bPreserveTimestamp = defaultMountOptions.PreserveTimestamp = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PRESERVE_TIMESTAMPS));

- bShowDisconnectedNetworkDrives = IsButtonChecked (GetDlgItem (hwndDlg, IDC_SHOW_DISCONNECTED_NETWORK_DRIVES));

- bHideWaitingDialog = IsButtonChecked (GetDlgItem (hwndDlg, IDC_HIDE_WAITING_DIALOG));

- bCacheDuringMultipleMount = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_TEMP_CACHE_ON_MULTIPLE_MOUNT));

- bWipeCacheOnExit = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_WIPE_CACHE_ON_EXIT));

- bWipeCacheOnAutoDismount = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_WIPE_CACHE_ON_AUTODISMOUNT));

- bCacheInDriverDefault = bCacheInDriver = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_CACHE_PASSWORDS));

- bIncludePimInCache = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_CACHE_PIM));

- defaultMountOptions.ReadOnly = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_MOUNT_READONLY));

- defaultMountOptions.Removable = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_MOUNT_REMOVABLE));

- bEnableBkgTask = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_BKG_TASK_ENABLE));

- bCloseBkgTaskWhenNoVolumes = IsNonInstallMode() ? bCloseBkgTaskWhenNoVolumes : IsButtonChecked (GetDlgItem (hwndDlg, IDC_CLOSE_BKG_TASK_WHEN_NOVOL));

- bDismountOnLogOff = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_LOGOFF));

- bDismountOnSessionLocked = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SESSION_LOCKED));

- bDismountOnPowerSaving = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_POWERSAVING));

- bDismountOnScreenSaver = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_SCREENSAVER));

- bForceAutoDismount = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_FORCE_AUTO_DISMOUNT));

- MaxVolumeIdleTime = GetDlgItemInt (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE_TIME, NULL, FALSE)

- * (IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_DISMOUNT_INACTIVE)) ? 1 : -1);

- bStartOnLogon = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_LOGON_START));

- bMountDevicesOnLogon = IsButtonChecked (GetDlgItem (hwndDlg, IDC_PREF_LOGON_MOUNT_DEVICES));

-

- ManageStartupSeq ();

-

- SaveSettings (hwndDlg);

-

- NormalCursor ();

-

- PreferencesDialogActive = FALSE;

- EndDialog (hwndDlg, lw);

- return 1;

- }

-

- if (lw == IDC_MORE_SETTINGS)

- {

- HMENU popup = CreatePopupMenu ();

- if (popup)

- {

- AppendMenuW (popup, MF_STRING, IDM_LANGUAGE, GetString ("IDM_LANGUAGE"));

- AppendMenuW (popup, MF_STRING, IDM_HOTKEY_SETTINGS, GetString ("IDM_HOTKEY_SETTINGS"));

- AppendMenuW (popup, MF_STRING, IDM_PERFORMANCE_SETTINGS, GetString ("IDM_PERFORMANCE_SETTINGS"));

- AppendMenuW (popup, MF_STRING, IDM_SYSENC_SETTINGS, GetString ("IDM_SYSENC_SETTINGS"));

- AppendMenuW (popup, MF_STRING, IDM_SYS_FAVORITES_SETTINGS, GetString ("IDM_SYS_FAVORITES_SETTINGS"));

- AppendMenuW (popup, MF_STRING, IDM_DEFAULT_KEYFILES, GetString ("IDM_DEFAULT_KEYFILES"));

- AppendMenuW (popup, MF_STRING, IDM_DEFAULT_MOUNT_PARAMETERS, GetString ("IDM_DEFAULT_MOUNT_PARAMETERS"));

- AppendMenuW (popup, MF_STRING, IDM_TOKEN_PREFERENCES, GetString ("IDM_TOKEN_PREFERENCES"));

-

- RECT rect;

- GetWindowRect (GetDlgItem (hwndDlg, IDC_MORE_SETTINGS), &rect);

-

- int menuItem = TrackPopupMenu (popup, TPM_RETURNCMD | TPM_LEFTBUTTON, rect.left + 2, rect.top + 2, 0, hwndDlg, NULL);

- DestroyMenu (popup);

-

- SendMessage (MainDlg, WM_COMMAND, menuItem, NULL);

- return 1;

- }

- else

- return 0;

- }

-

- if (HIWORD (wParam) == BN_CLICKED)

- {

- PreferencesDlgEnableButtons (hwndDlg);

- return 1;

- }

-

- return 0;

- }

-

- return 0;

-}

-

-

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

-{

- static MountOptions *mountOptions;

-

- WORD lw = LOWORD (wParam);

-

- switch (msg)

- {

- case WM_INITDIALOG:

- {

- BOOL protect;

-

- mountOptions = (MountOptions *) lParam;

-

- LocalizeDialog (hwndDlg, "IDD_MOUNT_OPTIONS");

-

- SendDlgItemMessage (hwndDlg, IDC_MOUNT_READONLY, BM_SETCHECK,

- mountOptions->ReadOnly ? BST_CHECKED : BST_UNCHECKED, 0);

- SendDlgItemMessage (hwndDlg, IDC_MOUNT_REMOVABLE, BM_SETCHECK,

- mountOptions->Removable ? BST_CHECKED : BST_UNCHECKED, 0);

- SendDlgItemMessage (hwndDlg, IDC_PROTECT_HIDDEN_VOL, BM_SETCHECK,

- mountOptions->ProtectHiddenVolume ? BST_CHECKED : BST_UNCHECKED, 0);

-

- SendDlgItemMessage (hwndDlg, IDC_PROTECT_HIDDEN_VOL, BM_SETCHECK,

- mountOptions->ProtectHiddenVolume ? BST_CHECKED : BST_UNCHECKED, 0);

-

- mountOptions->PartitionInInactiveSysEncScope = bPrebootPasswordDlgMode;

-

- SendDlgItemMessage (hwndDlg, IDC_MOUNT_SYSENC_PART_WITHOUT_PBA, BM_SETCHECK,

- bPrebootPasswordDlgMode ? BST_CHECKED : BST_UNCHECKED, 0);

-