/* 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); } } }