From 6e7a315eb67cb6c113cf37e1d66c4f11a51a2b3e Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 7 Apr 2024 18:29:51 +0200 Subject: Adding upstream version 2.06. Signed-off-by: Daniel Baumann --- grub-core/osdep/windows/getroot.c | 355 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 355 insertions(+) create mode 100644 grub-core/osdep/windows/getroot.c (limited to 'grub-core/osdep/windows/getroot.c') diff --git a/grub-core/osdep/windows/getroot.c b/grub-core/osdep/windows/getroot.c new file mode 100644 index 0000000..eada663 --- /dev/null +++ b/grub-core/osdep/windows/getroot.c @@ -0,0 +1,355 @@ +/* + * GRUB -- GRand Unified Bootloader + * Copyright (C) 1999,2000,2001,2002,2003,2006,2007,2008,2009,2010,2011,2012,2013 Free Software Foundation, Inc. + * + * GRUB is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * GRUB is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with GRUB. If not, see . + */ + +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#ifdef HAVE_LIMITS_H +#include +#endif + +#include + +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +TCHAR * +grub_get_mount_point (const TCHAR *path) +{ + const TCHAR *ptr; + TCHAR *out; + TCHAR letter = 0; + size_t allocsize; + + for (ptr = path; *ptr; ptr++); + allocsize = (ptr - path + 10) * 2; + out = xcalloc (allocsize, sizeof (out[0])); + + /* When pointing to EFI system partition GetVolumePathName fails + for ESP root and returns abberant information for everything + else. Since GetVolumePathName shouldn't fail for any valid + //?/X: we use it as indicator. */ + if ((path[0] == '/' || path[0] == '\\') + && (path[1] == '/' || path[1] == '\\') + && (path[2] == '?' || path[2] == '.') + && (path[3] == '/' || path[3] == '\\') + && path[4] + && (path[5] == ':')) + letter = path[4]; + if (path[0] && path[1] == ':') + letter = path[0]; + if (letter) + { + TCHAR letterpath[10] = TEXT("\\\\?\\#:"); + letterpath[4] = letter; + if (!GetVolumePathName (letterpath, out, allocsize)) + { + if (path[1] == ':') + { + out[0] = path[0]; + out[1] = ':'; + out[2] = '\0'; + return out; + } + memcpy (out, path, sizeof (out[0]) * 6); + out[6] = '\0'; + return out; + } + } + + if (!GetVolumePathName (path, out, allocsize)) + { + free (out); + return NULL; + } + return out; +} + +char ** +grub_guess_root_devices (const char *dir) +{ + char **os_dev = NULL; + TCHAR *dirwindows, *mntpointwindows; + TCHAR *ptr; + TCHAR volumename[100]; + + dirwindows = grub_util_get_windows_path (dir); + if (!dirwindows) + return 0; + + mntpointwindows = grub_get_mount_point (dirwindows); + + if (!mntpointwindows) + { + free (dirwindows); + grub_util_info ("can't get volume path name: %d", (int) GetLastError ()); + return 0; + } + + if (!mntpointwindows[0]) + { + free (dirwindows); + free (mntpointwindows); + return 0; + } + + for (ptr = mntpointwindows; *ptr; ptr++); + if (*(ptr - 1) != '\\') + { + *ptr = '\\'; + *(ptr + 1) = '\0'; + } + + if (!GetVolumeNameForVolumeMountPoint (mntpointwindows, + volumename, + ARRAY_SIZE (volumename))) + { + TCHAR letter = 0; + if ((mntpointwindows[0] == '/' || mntpointwindows[0] == '\\') + && (mntpointwindows[1] == '/' || mntpointwindows[1] == '\\') + && (mntpointwindows[2] == '?' || mntpointwindows[2] == '.') + && (mntpointwindows[3] == '/' || mntpointwindows[3] == '\\') + && mntpointwindows[4] + && (mntpointwindows[5] == ':')) + letter = mntpointwindows[4]; + if (mntpointwindows[0] && mntpointwindows[1] == ':') + letter = mntpointwindows[0]; + if (!letter) + { + free (dirwindows); + free (mntpointwindows); + return 0; + } + volumename[0] = '\\'; + volumename[1] = '\\'; + volumename[2] = '?'; + volumename[3] = '\\'; + volumename[4] = letter; + volumename[5] = ':'; + volumename[6] = '\0'; + } + os_dev = xmalloc (2 * sizeof (os_dev[0])); + + for (ptr = volumename; *ptr; ptr++); + while (ptr > volumename && *(ptr - 1) == '\\') + *--ptr = '\0'; + + os_dev[0] = grub_util_tchar_to_utf8 (volumename); + free (dirwindows); + free (mntpointwindows); + + if (!os_dev[0]) + { + free (os_dev); + return 0; + } + + os_dev[1] = 0; + + return os_dev; +} + +static int tcharncasecmp (LPCTSTR a, const char *b, size_t sz) +{ + for (; sz; sz--, a++, b++) + { + char ac, bc; + if(*a >= 0x80) + return +1; + if (*b & 0x80) + return -1; + if (*a == '\0' && *b == '\0') + return 0; + ac = *a; + bc = *b; + if (ac >= 'A' && ac <= 'Z') + ac -= 'A' - 'a'; + if (bc >= 'A' && bc <= 'Z') + bc -= 'A' - 'a'; + if (ac > bc) + return +1; + if (ac < bc) + return -1; + } + return 0; +} + +char * +grub_util_part_to_disk (const char *os_dev, + struct stat *st __attribute__ ((unused)), + int *is_part) +{ + HANDLE hd; + LPTSTR name = grub_util_get_windows_path (os_dev); + VOLUME_DISK_EXTENTS exts; + DWORD extsbytes; + char *ret; + + if (((name[0] == '/') || (name[0] == '\\')) && + ((name[1] == '/') || (name[1] == '\\')) && + ((name[2] == '.') || (name[2] == '?')) && + ((name[3] == '/') || (name[3] == '\\')) + && (tcharncasecmp (name + 4, "PhysicalDrive", sizeof ("PhysicalDrive") - 1) == 0 + || tcharncasecmp (name + 4, "Harddisk", sizeof ("Harddisk") - 1) == 0 + || ((name[4] == 'A' || name[4] == 'a' || name[4] == 'B' || name[4] == 'b') + && name[5] == ':' && name[6] == '\0'))) + { + grub_util_info ("Matches full disk pattern"); + ret = grub_util_tchar_to_utf8 (name); + free (name); + return ret; + } + + hd = CreateFile (name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, + 0, OPEN_EXISTING, 0, 0); + if (hd == INVALID_HANDLE_VALUE) + { + grub_util_info ("CreateFile failed"); + ret = grub_util_tchar_to_utf8 (name); + free (name); + return ret; + } + + if (!DeviceIoControl(hd, IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS, + NULL, 0, &exts, sizeof (exts), &extsbytes, NULL)) + { + grub_util_info ("IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS failed"); + ret = grub_util_tchar_to_utf8 (name); + CloseHandle (hd); + free (name); + return ret; + } + + CloseHandle (hd); + + *is_part = 1; + free (name); + return xasprintf ("\\\\?\\PhysicalDrive%lu", (unsigned long) exts.Extents[0].DiskNumber); +} + +enum grub_dev_abstraction_types +grub_util_get_dev_abstraction_os (const char *os_dev __attribute__((unused))) +{ + return GRUB_DEV_ABSTRACTION_NONE; +} + +int +grub_util_pull_device_os (const char *os_dev __attribute__ ((unused)), + enum grub_dev_abstraction_types ab __attribute__ ((unused))) +{ + return 0; +} + +char * +grub_util_get_grub_dev_os (const char *os_dev __attribute__ ((unused))) +{ + return NULL; +} + + +grub_disk_addr_t +grub_util_find_partition_start_os (const char *os_dev) +{ + HANDLE hd; + LPTSTR name = grub_util_get_windows_path (os_dev); + VOLUME_DISK_EXTENTS exts; + DWORD extsbytes; + char *ret; + + if (((name[0] == '/') || (name[0] == '\\')) && + ((name[1] == '/') || (name[1] == '\\')) && + ((name[2] == '.') || (name[2] == '?')) && + ((name[3] == '/') || (name[3] == '\\')) + && (tcharncasecmp (name + 4, "PhysicalDrive", sizeof ("PhysicalDrive") - 1) == 0 + || tcharncasecmp (name + 4, "Harddisk", sizeof ("Harddisk") - 1) == 0 + || ((name[4] == 'A' || name[4] == 'a' || name[4] == 'B' || name[4] == 'b') + && name[5] == ':' && name[6] == '\0'))) + { + ret = grub_util_tchar_to_utf8 (name); + free (name); + return 0; + } + + hd = CreateFile (name, GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, + 0, OPEN_EXISTING, 0, 0); + if (hd == INVALID_HANDLE_VALUE) + { + ret = grub_util_tchar_to_utf8 (name); + free (name); + return 0; + } + + if (!DeviceIoControl(hd, IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS, + NULL, 0, &exts, sizeof (exts), &extsbytes, NULL)) + { + ret = grub_util_tchar_to_utf8 (name); + CloseHandle (hd); + free (name); + return 0; + } + + CloseHandle (hd); + free (name); + return exts.Extents[0].StartingOffset.QuadPart / 512; +} + +int +grub_util_biosdisk_is_floppy (grub_disk_t disk) +{ + int ret; + const char *dname; + LPTSTR name; + + dname = grub_util_biosdisk_get_osdev (disk); + + if (!dname) + return 0; + + name = grub_util_get_windows_path (dname); + + ret = (((name[0] == '/') || (name[0] == '\\')) && + ((name[1] == '/') || (name[1] == '\\')) && + ((name[2] == '.') || (name[2] == '?')) && + ((name[3] == '/') || (name[3] == '\\')) + && (name[4] == 'A' || name[4] == 'a' || name[4] == 'B' || name[4] == 'b') + && name[5] == ':' && name[6] == '\0'); + free (name); + + return ret; +} -- cgit v1.2.3