/*
 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);
		}
	}
}
</span> <span class="p">(</span><span class="k">const</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">Print</span> <span class="p">(</span><span class="n">chs</span><span class="p">.</span><span class="n">Cylinder</span><span class="p">);</span>
	<span class="n">PrintChar</span> <span class="p">(</span><span class="sc">&#39;/&#39;</span><span class="p">);</span>
	<span class="n">Print</span> <span class="p">(</span><span class="n">chs</span><span class="p">.</span><span class="n">Head</span><span class="p">);</span>
	<span class="n">PrintChar</span> <span class="p">(</span><span class="sc">&#39;/&#39;</span><span class="p">);</span>
	<span class="n">Print</span> <span class="p">(</span><span class="n">chs</span><span class="p">.</span><span class="n">Sector</span><span class="p">);</span>
<span class="p">}</span>


<span class="kt">void</span> <span class="nf">PrintSectorCountInMB</span> <span class="p">(</span><span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sectorCount</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">Print</span> <span class="p">(</span><span class="n">sectorCount</span> <span class="o">&gt;&gt;</span> <span class="p">(</span><span class="n">TC_LB_SIZE_BIT_SHIFT_DIVISOR</span> <span class="o">+</span> <span class="mi">2</span><span class="p">));</span> <span class="n">Print</span> <span class="p">(</span><span class="s">&quot; MiB &quot;</span><span class="p">);</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">ReadWriteSectors</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">bufferSegment</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">bufferOffset</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">,</span> <span class="n">byte</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">CheckStack</span><span class="p">();</span>

	<span class="n">byte</span> <span class="n">cylinderLow</span> <span class="o">=</span> <span class="p">(</span><span class="n">byte</span><span class="p">)</span> <span class="n">chs</span><span class="p">.</span><span class="n">Cylinder</span><span class="p">;</span>
	<span class="n">byte</span> <span class="n">sector</span> <span class="o">=</span> <span class="n">chs</span><span class="p">.</span><span class="n">Sector</span><span class="p">;</span>
	<span class="n">sector</span> <span class="o">|=</span> <span class="n">byte</span> <span class="p">(</span><span class="n">chs</span><span class="p">.</span><span class="n">Cylinder</span> <span class="o">&gt;&gt;</span> <span class="mi">2</span><span class="p">)</span> <span class="o">&amp;</span> <span class="mh">0xc0</span><span class="p">;</span>
	<span class="n">byte</span> <span class="n">function</span> <span class="o">=</span> <span class="n">write</span> <span class="o">?</span> <span class="mh">0x03</span> <span class="o">:</span> <span class="mh">0x02</span><span class="p">;</span>

	<span class="n">BiosResult</span> <span class="n">result</span><span class="p">;</span>
	<span class="n">byte</span> <span class="n">tryCount</span> <span class="o">=</span> <span class="n">TC_MAX_BIOS_DISK_IO_RETRIES</span><span class="p">;</span>

	<span class="k">do</span>
	<span class="p">{</span>
		<span class="n">result</span> <span class="o">=</span> <span class="n">BiosResultSuccess</span><span class="p">;</span>

		<span class="kr">__asm</span>
		<span class="p">{</span>
			<span class="n">push</span> <span class="n">es</span>
			<span class="n">mov</span> <span class="n">ax</span><span class="p">,</span> <span class="n">bufferSegment</span>
			<span class="n">mov</span>	<span class="n">es</span><span class="p">,</span> <span class="n">ax</span>
			<span class="n">mov</span>	<span class="n">bx</span><span class="p">,</span> <span class="n">bufferOffset</span>
			<span class="n">mov</span> <span class="n">dl</span><span class="p">,</span> <span class="n">drive</span>
			<span class="n">mov</span> <span class="n">ch</span><span class="p">,</span> <span class="n">cylinderLow</span>
			<span class="n">mov</span> <span class="n">si</span><span class="p">,</span> <span class="n">chs</span>
			<span class="n">mov</span> <span class="n">dh</span><span class="p">,</span> <span class="p">[</span><span class="n">si</span><span class="p">].</span><span class="n">Head</span>
			<span class="n">mov</span> <span class="n">cl</span><span class="p">,</span> <span class="n">sector</span>
			<span class="n">mov</span>	<span class="n">al</span><span class="p">,</span> <span class="n">sectorCount</span>
			<span class="n">mov</span>	<span class="n">ah</span><span class="p">,</span> <span class="n">function</span>
			<span class="kt">int</span>	<span class="mh">0x13</span>
			<span class="n">jnc</span> <span class="n">ok</span>				<span class="c1">// If CF=0, ignore AH to prevent issues caused by potential bugs in BIOSes</span>
			<span class="n">mov</span>	<span class="n">result</span><span class="p">,</span> <span class="n">ah</span>
		<span class="nl">ok</span><span class="p">:</span>
			<span class="n">pop</span> <span class="n">es</span>
		<span class="p">}</span>

		<span class="k">if</span> <span class="p">(</span><span class="n">result</span> <span class="o">==</span> <span class="n">BiosResultEccCorrected</span><span class="p">)</span>
			<span class="n">result</span> <span class="o">=</span> <span class="n">BiosResultSuccess</span><span class="p">;</span>

	<span class="c1">// Some BIOSes report I/O errors prematurely in some cases</span>
	<span class="p">}</span> <span class="k">while</span> <span class="p">(</span><span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span> <span class="o">&amp;&amp;</span> <span class="o">--</span><span class="n">tryCount</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">);</span>

	<span class="k">if</span> <span class="p">(</span><span class="o">!</span><span class="n">silent</span> <span class="o">&amp;&amp;</span> <span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span><span class="p">)</span>
		<span class="n">PrintDiskError</span> <span class="p">(</span><span class="n">result</span><span class="p">,</span> <span class="n">write</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="k">nullptr</span><span class="p">,</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">);</span>

	<span class="k">return</span> <span class="n">result</span><span class="p">;</span>
<span class="p">}</span>

<span class="cp">#ifdef TC_WINDOWS_BOOT_RESCUE_DISK_MODE</span>

<span class="n">BiosResult</span> <span class="nf">ReadWriteSectors</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">byte</span> <span class="o">*</span><span class="n">buffer</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">,</span> <span class="n">byte</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">uint16</span> <span class="n">codeSeg</span><span class="p">;</span>
	<span class="kr">__asm</span> <span class="n">mov</span> <span class="n">codeSeg</span><span class="p">,</span> <span class="n">cs</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="n">write</span><span class="p">,</span> <span class="n">codeSeg</span><span class="p">,</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">chs</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>

<span class="n">BiosResult</span> <span class="nf">ReadSectors</span> <span class="p">(</span><span class="n">byte</span> <span class="o">*</span><span class="n">buffer</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">,</span> <span class="n">byte</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="nb">false</span><span class="p">,</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">chs</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>

<span class="cp">#if 0</span><span class="c"></span>
<span class="c">BiosResult WriteSectors (byte *buffer, byte drive, const ChsAddress &amp;chs, byte sectorCount, bool silent)</span>
<span class="c">{</span>
<span class="c">	return ReadWriteSectors (true, buffer, drive, chs, sectorCount, silent);</span>
<span class="c">}</span>
<span class="cp">#endif</span>

<span class="cp">#endif</span>

<span class="k">static</span> <span class="n">BiosResult</span> <span class="nf">ReadWriteSectors</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">BiosLbaPacket</span> <span class="o">&amp;</span><span class="n">dapPacket</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">CheckStack</span><span class="p">();</span>

	<span class="k">if</span> <span class="p">(</span><span class="o">!</span><span class="n">IsLbaSupported</span> <span class="p">(</span><span class="n">drive</span><span class="p">))</span>
	<span class="p">{</span>
		<span class="n">DriveGeometry</span> <span class="n">geometry</span><span class="p">;</span>

		<span class="n">BiosResult</span> <span class="n">result</span> <span class="o">=</span> <span class="n">GetDriveGeometry</span> <span class="p">(</span><span class="n">drive</span><span class="p">,</span> <span class="n">geometry</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
		<span class="k">if</span> <span class="p">(</span><span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span><span class="p">)</span>
			<span class="k">return</span> <span class="n">result</span><span class="p">;</span>

		<span class="n">ChsAddress</span> <span class="n">chs</span><span class="p">;</span>
		<span class="n">LbaToChs</span> <span class="p">(</span><span class="n">geometry</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">chs</span><span class="p">);</span>
		<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="n">write</span><span class="p">,</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="p">(</span><span class="n">dapPacket</span><span class="p">.</span><span class="n">Buffer</span> <span class="o">&gt;&gt;</span> <span class="mi">16</span><span class="p">),</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="n">dapPacket</span><span class="p">.</span><span class="n">Buffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">chs</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
	<span class="p">}</span>

	<span class="n">dapPacket</span><span class="p">.</span><span class="n">Size</span> <span class="o">=</span> <span class="k">sizeof</span> <span class="p">(</span><span class="n">dapPacket</span><span class="p">);</span>
	<span class="n">dapPacket</span><span class="p">.</span><span class="n">Reserved</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">dapPacket</span><span class="p">.</span><span class="n">SectorCount</span> <span class="o">=</span> <span class="n">sectorCount</span><span class="p">;</span>
	<span class="n">dapPacket</span><span class="p">.</span><span class="n">Sector</span> <span class="o">=</span> <span class="n">sector</span><span class="p">;</span>

	<span class="n">byte</span> <span class="n">function</span> <span class="o">=</span> <span class="n">write</span> <span class="o">?</span> <span class="mh">0x43</span> <span class="o">:</span> <span class="mh">0x42</span><span class="p">;</span>

	<span class="n">BiosResult</span> <span class="n">result</span><span class="p">;</span>
	<span class="n">byte</span> <span class="n">tryCount</span> <span class="o">=</span> <span class="n">TC_MAX_BIOS_DISK_IO_RETRIES</span><span class="p">;</span>

	<span class="k">do</span>
	<span class="p">{</span>
		<span class="n">result</span> <span class="o">=</span> <span class="n">BiosResultSuccess</span><span class="p">;</span>

		<span class="kr">__asm</span>
		<span class="p">{</span>
			<span class="n">mov</span>	<span class="n">bx</span><span class="p">,</span> <span class="mh">0x55aa</span>
			<span class="n">mov</span>	<span class="n">dl</span><span class="p">,</span> <span class="n">drive</span>
			<span class="n">mov</span> <span class="n">si</span><span class="p">,</span> <span class="p">[</span><span class="n">dapPacket</span><span class="p">]</span>
			<span class="n">mov</span>	<span class="n">ah</span><span class="p">,</span> <span class="n">function</span>
			<span class="n">xor</span> <span class="n">al</span><span class="p">,</span> <span class="n">al</span>
			<span class="kt">int</span>	<span class="mh">0x13</span>
			<span class="n">jnc</span> <span class="n">ok</span>				<span class="c1">// If CF=0, ignore AH to prevent issues caused by potential bugs in BIOSes</span>
			<span class="n">mov</span>	<span class="n">result</span><span class="p">,</span> <span class="n">ah</span>
		<span class="nl">ok</span><span class="p">:</span>
		<span class="p">}</span>

		<span class="k">if</span> <span class="p">(</span><span class="n">result</span> <span class="o">==</span> <span class="n">BiosResultEccCorrected</span><span class="p">)</span>
			<span class="n">result</span> <span class="o">=</span> <span class="n">BiosResultSuccess</span><span class="p">;</span>

	<span class="c1">// Some BIOSes report I/O errors prematurely in some cases</span>
	<span class="p">}</span> <span class="k">while</span> <span class="p">(</span><span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span> <span class="o">&amp;&amp;</span> <span class="o">--</span><span class="n">tryCount</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">);</span>

	<span class="k">if</span> <span class="p">(</span><span class="o">!</span><span class="n">silent</span> <span class="o">&amp;&amp;</span> <span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span><span class="p">)</span>
		<span class="n">PrintDiskError</span> <span class="p">(</span><span class="n">result</span><span class="p">,</span> <span class="n">write</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">);</span>

	<span class="k">return</span> <span class="n">result</span><span class="p">;</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">ReadWriteSectors</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">byte</span> <span class="o">*</span><span class="n">buffer</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">BiosLbaPacket</span> <span class="n">dapPacket</span><span class="p">;</span>
	<span class="n">dapPacket</span><span class="p">.</span><span class="n">Buffer</span> <span class="o">=</span> <span class="p">(</span><span class="n">uint32</span><span class="p">)</span> <span class="n">buffer</span><span class="p">;</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="n">write</span><span class="p">,</span> <span class="n">dapPacket</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">ReadWriteSectors</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">bufferSegment</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">bufferOffset</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">BiosLbaPacket</span> <span class="n">dapPacket</span><span class="p">;</span>
	<span class="n">dapPacket</span><span class="p">.</span><span class="n">Buffer</span> <span class="o">=</span> <span class="p">((</span><span class="n">uint32</span><span class="p">)</span> <span class="n">bufferSegment</span> <span class="o">&lt;&lt;</span> <span class="mi">16</span><span class="p">)</span> <span class="o">|</span> <span class="n">bufferOffset</span><span class="p">;</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="n">write</span><span class="p">,</span> <span class="n">dapPacket</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>

<span class="n">BiosResult</span> <span class="nf">ReadSectors</span> <span class="p">(</span><span class="n">uint16</span> <span class="n">bufferSegment</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">bufferOffset</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="nb">false</span><span class="p">,</span> <span class="n">bufferSegment</span><span class="p">,</span> <span class="n">bufferOffset</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">ReadSectors</span> <span class="p">(</span><span class="n">byte</span> <span class="o">*</span><span class="n">buffer</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">BiosResult</span> <span class="n">result</span><span class="p">;</span>
	<span class="n">uint16</span> <span class="n">codeSeg</span><span class="p">;</span>
	<span class="kr">__asm</span> <span class="n">mov</span> <span class="n">codeSeg</span><span class="p">,</span> <span class="n">cs</span>

	<span class="n">result</span> <span class="o">=</span> <span class="n">ReadSectors</span> <span class="p">(</span><span class="n">BootStarted</span> <span class="o">?</span> <span class="nl">codeSeg</span> <span class="p">:</span> <span class="n">TC_BOOT_LOADER_ALT_SEGMENT</span><span class="p">,</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>

	<span class="c1">// Alternative segment is used to prevent memory corruption caused by buggy BIOSes</span>
	<span class="k">if</span> <span class="p">(</span><span class="o">!</span><span class="n">BootStarted</span><span class="p">)</span>
		<span class="n">CopyMemory</span> <span class="p">(</span><span class="n">TC_BOOT_LOADER_ALT_SEGMENT</span><span class="p">,</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">sectorCount</span> <span class="o">*</span> <span class="n">TC_LB_SIZE</span><span class="p">);</span>

	<span class="k">return</span> <span class="n">result</span><span class="p">;</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">WriteSectors</span> <span class="p">(</span><span class="n">byte</span> <span class="o">*</span><span class="n">buffer</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">sector</span><span class="p">,</span> <span class="n">uint16</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="k">return</span> <span class="n">ReadWriteSectors</span> <span class="p">(</span><span class="nb">true</span><span class="p">,</span> <span class="n">buffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">sector</span><span class="p">,</span> <span class="n">sectorCount</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">GetDriveGeometry</span> <span class="p">(</span><span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="n">DriveGeometry</span> <span class="o">&amp;</span><span class="n">geometry</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">CheckStack</span><span class="p">();</span>

	<span class="n">byte</span> <span class="n">maxCylinderLow</span><span class="p">,</span> <span class="n">maxHead</span><span class="p">,</span> <span class="n">maxSector</span><span class="p">;</span>
	<span class="n">BiosResult</span> <span class="n">result</span><span class="p">;</span>
	<span class="kr">__asm</span>
	<span class="p">{</span>
		<span class="n">push</span> <span class="n">es</span>
		<span class="n">mov</span> <span class="n">dl</span><span class="p">,</span> <span class="n">drive</span>
		<span class="n">mov</span> <span class="n">ah</span><span class="p">,</span> <span class="mh">0x08</span>
		<span class="kt">int</span>	<span class="mh">0x13</span>

		<span class="n">mov</span>	<span class="n">result</span><span class="p">,</span> <span class="n">ah</span>
		<span class="n">mov</span> <span class="n">maxCylinderLow</span><span class="p">,</span> <span class="n">ch</span>
		<span class="n">mov</span> <span class="n">maxSector</span><span class="p">,</span> <span class="n">cl</span>
		<span class="n">mov</span> <span class="n">maxHead</span><span class="p">,</span> <span class="n">dh</span>
		<span class="n">pop</span> <span class="n">es</span>
	<span class="p">}</span>

	<span class="k">if</span> <span class="p">(</span><span class="n">result</span> <span class="o">==</span> <span class="n">BiosResultSuccess</span><span class="p">)</span>
	<span class="p">{</span>
		<span class="n">geometry</span><span class="p">.</span><span class="n">Cylinders</span> <span class="o">=</span> <span class="p">(</span><span class="n">maxCylinderLow</span> <span class="o">|</span> <span class="p">(</span><span class="n">uint16</span> <span class="p">(</span><span class="n">maxSector</span> <span class="o">&amp;</span> <span class="mh">0xc0</span><span class="p">)</span> <span class="o">&lt;&lt;</span> <span class="mi">2</span><span class="p">))</span> <span class="o">+</span> <span class="mi">1</span><span class="p">;</span>
		<span class="n">geometry</span><span class="p">.</span><span class="n">Heads</span> <span class="o">=</span> <span class="n">maxHead</span> <span class="o">+</span> <span class="mi">1</span><span class="p">;</span>
		<span class="n">geometry</span><span class="p">.</span><span class="n">Sectors</span> <span class="o">=</span> <span class="n">maxSector</span> <span class="o">&amp;</span> <span class="o">~</span><span class="mh">0xc0</span><span class="p">;</span>
	<span class="p">}</span>
	<span class="k">else</span> <span class="k">if</span> <span class="p">(</span><span class="o">!</span><span class="n">silent</span><span class="p">)</span>
	<span class="p">{</span>
		<span class="n">Print</span> <span class="p">(</span><span class="s">&quot;Drive &quot;</span><span class="p">);</span>
		<span class="n">Print</span> <span class="p">(</span><span class="n">drive</span> <span class="o">^</span> <span class="mh">0x80</span><span class="p">);</span>
		<span class="n">Print</span> <span class="p">(</span><span class="s">&quot; not found: &quot;</span><span class="p">);</span>
		<span class="n">PrintErrorNoEndl</span> <span class="p">(</span><span class="s">&quot;&quot;</span><span class="p">);</span>
		<span class="n">Print</span> <span class="p">(</span><span class="n">result</span><span class="p">);</span>
		<span class="n">PrintEndl</span><span class="p">();</span>
	<span class="p">}</span>

	<span class="k">return</span> <span class="n">result</span><span class="p">;</span>
<span class="p">}</span>


<span class="kt">void</span> <span class="nf">ChsToLba</span> <span class="p">(</span><span class="k">const</span> <span class="n">DriveGeometry</span> <span class="o">&amp;</span><span class="n">geometry</span><span class="p">,</span> <span class="k">const</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">,</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">lba</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">lba</span><span class="p">.</span><span class="n">HighPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">lba</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="p">(</span><span class="n">uint32</span> <span class="p">(</span><span class="n">chs</span><span class="p">.</span><span class="n">Cylinder</span><span class="p">)</span> <span class="o">*</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Heads</span> <span class="o">+</span> <span class="n">chs</span><span class="p">.</span><span class="n">Head</span><span class="p">)</span> <span class="o">*</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Sectors</span> <span class="o">+</span> <span class="n">chs</span><span class="p">.</span><span class="n">Sector</span> <span class="o">-</span> <span class="mi">1</span><span class="p">;</span>
<span class="p">}</span>


<span class="kt">void</span> <span class="nf">LbaToChs</span> <span class="p">(</span><span class="k">const</span> <span class="n">DriveGeometry</span> <span class="o">&amp;</span><span class="n">geometry</span><span class="p">,</span> <span class="k">const</span> <span class="n">uint64</span> <span class="o">&amp;</span><span class="n">lba</span><span class="p">,</span> <span class="n">ChsAddress</span> <span class="o">&amp;</span><span class="n">chs</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">chs</span><span class="p">.</span><span class="n">Sector</span> <span class="o">=</span> <span class="p">(</span><span class="n">byte</span><span class="p">)</span> <span class="p">((</span><span class="n">lba</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">%</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Sectors</span><span class="p">)</span> <span class="o">+</span> <span class="mi">1</span><span class="p">);</span>
	<span class="n">uint32</span> <span class="n">ch</span> <span class="o">=</span> <span class="n">lba</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">/</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Sectors</span><span class="p">;</span>
	<span class="n">chs</span><span class="p">.</span><span class="n">Head</span> <span class="o">=</span> <span class="p">(</span><span class="n">byte</span><span class="p">)</span> <span class="p">(</span><span class="n">ch</span> <span class="o">%</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Heads</span><span class="p">);</span>
	<span class="n">chs</span><span class="p">.</span><span class="n">Cylinder</span> <span class="o">=</span> <span class="p">(</span><span class="n">uint16</span><span class="p">)</span> <span class="p">(</span><span class="n">ch</span> <span class="o">/</span> <span class="n">geometry</span><span class="p">.</span><span class="n">Heads</span><span class="p">);</span>
<span class="p">}</span>


<span class="kt">void</span> <span class="nf">PartitionEntryMBRToPartition</span> <span class="p">(</span><span class="k">const</span> <span class="n">PartitionEntryMBR</span> <span class="o">&amp;</span><span class="n">partEntry</span><span class="p">,</span> <span class="n">Partition</span> <span class="o">&amp;</span><span class="n">partition</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">Active</span> <span class="o">=</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">BootIndicator</span> <span class="o">==</span> <span class="mh">0x80</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">EndSector</span><span class="p">.</span><span class="n">HighPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">EndSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">StartLBA</span> <span class="o">+</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">SectorCountLBA</span> <span class="o">-</span> <span class="mi">1</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">SectorCount</span><span class="p">.</span><span class="n">HighPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">SectorCount</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">SectorCountLBA</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">HighPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">StartLBA</span><span class="p">;</span>
	<span class="n">partition</span><span class="p">.</span><span class="n">Type</span> <span class="o">=</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">Type</span><span class="p">;</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">ReadWriteMBR</span> <span class="p">(</span><span class="kt">bool</span> <span class="n">write</span><span class="p">,</span> <span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">uint64</span> <span class="n">mbrSector</span><span class="p">;</span>
	<span class="n">mbrSector</span><span class="p">.</span><span class="n">HighPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
	<span class="n">mbrSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>

	<span class="k">if</span> <span class="p">(</span><span class="n">write</span><span class="p">)</span>
		<span class="k">return</span> <span class="n">WriteSectors</span> <span class="p">(</span><span class="n">SectorBuffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">mbrSector</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>

	<span class="k">return</span> <span class="n">ReadSectors</span> <span class="p">(</span><span class="n">SectorBuffer</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">mbrSector</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>		<span class="c1">// Uses alternative segment</span>
<span class="p">}</span>


<span class="n">BiosResult</span> <span class="nf">GetDrivePartitions</span> <span class="p">(</span><span class="n">byte</span> <span class="n">drive</span><span class="p">,</span> <span class="n">Partition</span> <span class="o">*</span><span class="n">partitionArray</span><span class="p">,</span> <span class="kt">size_t</span> <span class="n">partitionArrayCapacity</span><span class="p">,</span> <span class="kt">size_t</span> <span class="o">&amp;</span><span class="n">partitionCount</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">activeOnly</span><span class="p">,</span> <span class="n">Partition</span> <span class="o">*</span><span class="n">findPartitionFollowingThis</span><span class="p">,</span> <span class="kt">bool</span> <span class="n">silent</span><span class="p">)</span>
<span class="p">{</span>
	<span class="n">Partition</span> <span class="o">*</span><span class="n">followingPartition</span><span class="p">;</span>
	<span class="n">Partition</span> <span class="n">tmpPartition</span><span class="p">;</span>

	<span class="k">if</span> <span class="p">(</span><span class="n">findPartitionFollowingThis</span><span class="p">)</span>
	<span class="p">{</span>
		<span class="n">assert</span> <span class="p">(</span><span class="n">partitionArrayCapacity</span> <span class="o">==</span> <span class="mi">1</span><span class="p">);</span>
		<span class="n">partitionArrayCapacity</span> <span class="o">=</span> <span class="mh">0xff</span><span class="p">;</span>
		<span class="n">followingPartition</span> <span class="o">=</span> <span class="n">partitionArray</span><span class="p">;</span>
		<span class="n">partitionArray</span> <span class="o">=</span> <span class="o">&amp;</span><span class="n">tmpPartition</span><span class="p">;</span>

		<span class="n">followingPartition</span><span class="o">-&gt;</span><span class="n">Drive</span> <span class="o">=</span> <span class="n">TC_INVALID_BIOS_DRIVE</span><span class="p">;</span>
		<span class="n">followingPartition</span><span class="o">-&gt;</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="mh">0xFFFFffffUL</span><span class="p">;</span>
	<span class="p">}</span>

	<span class="n">AcquireSectorBuffer</span><span class="p">();</span>
	<span class="n">BiosResult</span> <span class="n">result</span> <span class="o">=</span> <span class="n">ReadWriteMBR</span> <span class="p">(</span><span class="nb">false</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">silent</span><span class="p">);</span>
	<span class="n">ReleaseSectorBuffer</span><span class="p">();</span>

	<span class="n">partitionCount</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>

	<span class="n">MBR</span> <span class="o">*</span><span class="n">mbr</span> <span class="o">=</span> <span class="p">(</span><span class="n">MBR</span> <span class="o">*</span><span class="p">)</span> <span class="n">SectorBuffer</span><span class="p">;</span>
	<span class="k">if</span> <span class="p">(</span><span class="n">result</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span> <span class="o">||</span> <span class="n">mbr</span><span class="o">-&gt;</span><span class="n">Signature</span> <span class="o">!=</span> <span class="mh">0xaa55</span><span class="p">)</span>
		<span class="k">return</span> <span class="n">result</span><span class="p">;</span>

	<span class="n">PartitionEntryMBR</span> <span class="n">mbrPartitions</span><span class="p">[</span><span class="mi">4</span><span class="p">];</span>
	<span class="n">memcpy</span> <span class="p">(</span><span class="n">mbrPartitions</span><span class="p">,</span> <span class="n">mbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">,</span> <span class="k">sizeof</span> <span class="p">(</span><span class="n">mbrPartitions</span><span class="p">));</span>
	<span class="kt">size_t</span> <span class="n">partitionArrayPos</span> <span class="o">=</span> <span class="mi">0</span><span class="p">,</span> <span class="n">partitionNumber</span><span class="p">;</span>

	<span class="k">for</span> <span class="p">(</span><span class="n">partitionNumber</span> <span class="o">=</span> <span class="mi">0</span><span class="p">;</span>
		<span class="n">partitionNumber</span> <span class="o">&lt;</span> <span class="n">array_capacity</span> <span class="p">(</span><span class="n">mbrPartitions</span><span class="p">)</span> <span class="o">&amp;&amp;</span> <span class="n">partitionArrayPos</span> <span class="o">&lt;</span> <span class="n">partitionArrayCapacity</span><span class="p">;</span>
		<span class="o">++</span><span class="n">partitionNumber</span><span class="p">)</span>
	<span class="p">{</span>
		<span class="k">const</span> <span class="n">PartitionEntryMBR</span> <span class="o">&amp;</span><span class="n">partEntry</span> <span class="o">=</span> <span class="n">mbrPartitions</span><span class="p">[</span><span class="n">partitionNumber</span><span class="p">];</span>

		<span class="k">if</span> <span class="p">(</span><span class="n">partEntry</span><span class="p">.</span><span class="n">SectorCountLBA</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">)</span>
		<span class="p">{</span>
			<span class="n">Partition</span> <span class="o">&amp;</span><span class="n">partition</span> <span class="o">=</span> <span class="n">partitionArray</span><span class="p">[</span><span class="n">partitionArrayPos</span><span class="p">];</span>
			<span class="n">PartitionEntryMBRToPartition</span> <span class="p">(</span><span class="n">partEntry</span><span class="p">,</span> <span class="n">partition</span><span class="p">);</span>

			<span class="k">if</span> <span class="p">(</span><span class="n">activeOnly</span> <span class="o">&amp;&amp;</span> <span class="o">!</span><span class="n">partition</span><span class="p">.</span><span class="n">Active</span><span class="p">)</span>
				<span class="k">continue</span><span class="p">;</span>

			<span class="n">partition</span><span class="p">.</span><span class="n">Drive</span> <span class="o">=</span> <span class="n">drive</span><span class="p">;</span>
			<span class="n">partition</span><span class="p">.</span><span class="n">Number</span> <span class="o">=</span> <span class="n">partitionArrayPos</span><span class="p">;</span>

			<span class="k">if</span> <span class="p">(</span><span class="n">partEntry</span><span class="p">.</span><span class="n">Type</span> <span class="o">==</span> <span class="mh">0x5</span> <span class="o">||</span> <span class="n">partEntry</span><span class="p">.</span><span class="n">Type</span> <span class="o">==</span> <span class="mh">0xf</span><span class="p">)</span> <span class="c1">// Extended partition</span>
			<span class="p">{</span>
				<span class="k">if</span> <span class="p">(</span><span class="n">IsLbaSupported</span> <span class="p">(</span><span class="n">drive</span><span class="p">))</span>
				<span class="p">{</span>
					<span class="c1">// Find all extended partitions</span>
					<span class="n">uint64</span> <span class="n">firstExtStartLBA</span> <span class="o">=</span> <span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">;</span>
					<span class="n">uint64</span> <span class="n">extStartLBA</span> <span class="o">=</span> <span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">;</span>
					<span class="n">MBR</span> <span class="o">*</span><span class="n">extMbr</span> <span class="o">=</span> <span class="p">(</span><span class="n">MBR</span> <span class="o">*</span><span class="p">)</span> <span class="n">SectorBuffer</span><span class="p">;</span>

					<span class="k">while</span> <span class="p">(</span><span class="n">partitionArrayPos</span> <span class="o">&lt;</span> <span class="n">partitionArrayCapacity</span> <span class="o">&amp;&amp;</span>
						<span class="p">(</span><span class="n">result</span> <span class="o">=</span> <span class="n">ReadSectors</span> <span class="p">((</span><span class="n">byte</span> <span class="o">*</span><span class="p">)</span> <span class="n">extMbr</span><span class="p">,</span> <span class="n">drive</span><span class="p">,</span> <span class="n">extStartLBA</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">silent</span><span class="p">))</span> <span class="o">==</span> <span class="n">BiosResultSuccess</span>
						<span class="o">&amp;&amp;</span> <span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Signature</span> <span class="o">==</span> <span class="mh">0xaa55</span><span class="p">)</span>
					<span class="p">{</span>
						<span class="k">if</span> <span class="p">(</span><span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">0</span><span class="p">].</span><span class="n">SectorCountLBA</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">)</span>
						<span class="p">{</span>
							<span class="n">Partition</span> <span class="o">&amp;</span><span class="n">logPart</span> <span class="o">=</span> <span class="n">partitionArray</span><span class="p">[</span><span class="n">partitionArrayPos</span><span class="p">];</span>
							<span class="n">PartitionEntryMBRToPartition</span> <span class="p">(</span><span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">logPart</span><span class="p">);</span>
							<span class="n">logPart</span><span class="p">.</span><span class="n">Drive</span> <span class="o">=</span> <span class="n">drive</span><span class="p">;</span>

							<span class="n">logPart</span><span class="p">.</span><span class="n">Number</span> <span class="o">=</span> <span class="n">partitionArrayPos</span><span class="p">;</span>
							<span class="n">logPart</span><span class="p">.</span><span class="n">Primary</span> <span class="o">=</span> <span class="nb">false</span><span class="p">;</span>

							<span class="n">logPart</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">+=</span> <span class="n">extStartLBA</span><span class="p">.</span><span class="n">LowPart</span><span class="p">;</span>
							<span class="n">logPart</span><span class="p">.</span><span class="n">EndSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">+=</span> <span class="n">extStartLBA</span><span class="p">.</span><span class="n">LowPart</span><span class="p">;</span>

							<span class="k">if</span> <span class="p">(</span><span class="n">findPartitionFollowingThis</span><span class="p">)</span>
							<span class="p">{</span>
								<span class="k">if</span> <span class="p">(</span><span class="n">logPart</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">&gt;</span> <span class="n">findPartitionFollowingThis</span><span class="o">-&gt;</span><span class="n">EndSector</span><span class="p">.</span><span class="n">LowPart</span>
									<span class="o">&amp;&amp;</span> <span class="n">logPart</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">&lt;</span> <span class="n">followingPartition</span><span class="o">-&gt;</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span><span class="p">)</span>
								<span class="p">{</span>
									<span class="o">*</span><span class="n">followingPartition</span> <span class="o">=</span> <span class="n">logPart</span><span class="p">;</span>
								<span class="p">}</span>
							<span class="p">}</span>
							<span class="k">else</span>
								<span class="o">++</span><span class="n">partitionArrayPos</span><span class="p">;</span>
						<span class="p">}</span>

						<span class="c1">// Secondary extended</span>
						<span class="k">if</span> <span class="p">(</span><span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">1</span><span class="p">].</span><span class="n">Type</span> <span class="o">!=</span> <span class="mh">0x5</span> <span class="o">&amp;&amp;</span> <span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">1</span><span class="p">].</span><span class="n">Type</span> <span class="o">==</span> <span class="mh">0xf</span>
							<span class="o">||</span> <span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">1</span><span class="p">].</span><span class="n">SectorCountLBA</span> <span class="o">==</span> <span class="mi">0</span><span class="p">)</span>
							<span class="k">break</span><span class="p">;</span>

						<span class="n">extStartLBA</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">=</span> <span class="n">extMbr</span><span class="o">-&gt;</span><span class="n">Partitions</span><span class="p">[</span><span class="mi">1</span><span class="p">].</span><span class="n">StartLBA</span> <span class="o">+</span> <span class="n">firstExtStartLBA</span><span class="p">.</span><span class="n">LowPart</span><span class="p">;</span>
					<span class="p">}</span>
				<span class="p">}</span>
			<span class="p">}</span>
			<span class="k">else</span>
			<span class="p">{</span>
				<span class="n">partition</span><span class="p">.</span><span class="n">Primary</span> <span class="o">=</span> <span class="nb">true</span><span class="p">;</span>

				<span class="k">if</span> <span class="p">(</span><span class="n">findPartitionFollowingThis</span><span class="p">)</span>
				<span class="p">{</span>
					<span class="k">if</span> <span class="p">(</span><span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">&gt;</span> <span class="n">findPartitionFollowingThis</span><span class="o">-&gt;</span><span class="n">EndSector</span><span class="p">.</span><span class="n">LowPart</span>
						<span class="o">&amp;&amp;</span> <span class="n">partition</span><span class="p">.</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span> <span class="o">&lt;</span> <span class="n">followingPartition</span><span class="o">-&gt;</span><span class="n">StartSector</span><span class="p">.</span><span class="n">LowPart</span><span class="p">)</span>
					<span class="p">{</span>
						<span class="o">*</span><span class="n">followingPartition</span> <span class="o">=</span> <span class="n">partition</span><span class="p">;</span>
					<span class="p">}</span>
				<span class="p">}</span>
				<span class="k">else</span>
					<span class="o">++</span><span class="n">partitionArrayPos</span><span class="p">;</span>
			<span class="p">}</span>
		<span class="p">}</span>
	<span class="p">}</span>

	<span class="n">partitionCount</span> <span class="o">=</span> <span class="n">partitionArrayPos</span><span class="p">;</span>
	<span class="k">return</span> <span class="n">result</span><span class="p">;</span>
<span class="p">}</span>


<span class="kt">bool</span> <span class="nf">GetActivePartition</span> <span class="p">(</span><span class="n">byte</span> <span class="n">drive</span><span class="p">)</span>
<span class="p">{</span>
	<span class="kt">size_t</span> <span class="n">partCount</span><span class="p">;</span>

	<span class="k">if</span> <span class="p">(</span><span class="n">GetDrivePartitions</span> <span class="p">(</span><span class="n">drive</span><span class="p">,</span> <span class="o">&amp;</span><span class="n">ActivePartition</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">partCount</span><span class="p">,</span> <span class="nb">true</span><span class="p">)</span> <span class="o">!=</span> <span class="n">BiosResultSuccess</span> <span class="o">||</span> <span class="n">partCount</span> <span class="o">&lt;</span> <span class="mi">1</span><span class="p">)</span>
	<span class="p">{</span>
		<span class="n">ActivePartition</span><span class="p">.</span><span class="n">Drive</span> <span class="o">=</span> <span class="n">TC_INVALID_BIOS_DRIVE</span><span class="p">;</span>
		<span class="n">PrintError</span> <span class="p">(</span><span class="n">TC_BOOT_STR_NO_BOOT_PARTITION</span><span class="p">);</span>
		<span class="k">return</span> <span class="nb">false</span><span class="p">;</span>
	<span class="p">}</span>

	<span class="k">return</span> <span class="nb">true</span><span class="p">;</span>
<span class="p">}</span>
</pre></div>