Split out blocklist retrieving from setup.c to

grub-core/osdep/blocklist.c and add windows implementation since generic version doesn't work on NTFS on Windows due to aggressive unflushable cache.
2013-10-15 17:02:26 +02:00 · 2013-10-15 17:02:26 +02:00 · 5a869940de
commit 5a869940de
parent c06c75201c
13 changed files with 602 additions and 288 deletions
--- a/grub-core/osdep/windows/blocklist.c
+++ b/grub-core/osdep/windows/blocklist.c
@ -0,0 +1,119 @@
+/*
+ *  GRUB  --  GRand Unified Bootloader
+ *  Copyright (C) 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 <http://www.gnu.org/licenses/>.
+ */
+
+#include <config.h>
+
+#include <grub/disk.h>
+#include <grub/partition.h>
+#include <grub/fs.h>
+#include <grub/ntfs.h>
+#include <grub/fat.h>
+#include <grub/exfat.h>
+#include <grub/udf.h>
+#include <grub/util/misc.h>
+#include <grub/util/install.h>
+#include <grub/emu/getroot.h>
+#include <grub/emu/hostfile.h>
+
+#include <windows.h>
+#include <winioctl.h>
+
+void
+grub_install_get_blocklist (grub_device_t root_dev,
+			    const char *core_path,
+			    const char *core_img __attribute__ ((unused)),
+			    size_t core_size,
+			    void (*callback) (grub_disk_addr_t sector,
+					      unsigned offset,
+					      unsigned length,
+					      void *data),
+			    void *hook_data)
+{
+  grub_disk_addr_t first_lcn = 0;
+  HANDLE filehd;
+  DWORD rets;
+  RETRIEVAL_POINTERS_BUFFER *extbuf;
+  size_t extbuf_size;
+  DWORD i, j, k;
+  grub_uint64_t sec_per_lcn;
+  grub_uint64_t curvcn = 0;
+  STARTING_VCN_INPUT_BUFFER start_vcn;
+  grub_fs_t fs;
+  grub_err_t err;
+
+  fs = grub_fs_probe (root_dev);
+  if (!fs)
+    grub_util_error ("%s", grub_errmsg);
+
+  /* This is ugly but windows doesn't give all needed data. Or does anyone
+     have a pointer how to retrieve it?
+   */
+  if (grub_strcmp (fs->name, "ntfs") == 0)
+    {
+      struct grub_ntfs_bpb bpb;
+      err = grub_disk_read (root_dev->disk, 0, 0, sizeof (bpb), &bpb);
+      if (err)
+	grub_util_error ("%s", grub_errmsg);
+      sec_per_lcn = ((grub_uint32_t) bpb.sectors_per_cluster
+		     * (grub_uint32_t) grub_le_to_cpu16 (bpb.bytes_per_sector))
+	>> 9;
+      first_lcn = 0;
+    }
+  else if (grub_strcmp (fs->name, "exfat") == 0)
+    first_lcn = grub_exfat_get_cluster_sector (root_dev->disk, &sec_per_lcn);
+  else if (grub_strcmp (fs->name, "fat") == 0)
+    first_lcn = grub_fat_get_cluster_sector (root_dev->disk, &sec_per_lcn);
+  else if (grub_strcmp (fs->name, "udf") == 0)
+    first_lcn = grub_udf_get_cluster_sector (root_dev->disk, &sec_per_lcn);
+  else
+    grub_util_error ("unsupported fs for blocklist under windows: %s",
+		     fs->name);
+
+  grub_util_info ("sec_per_lcn = %lld, first_lcn=%lld", sec_per_lcn, first_lcn);
+
+  first_lcn += grub_partition_get_start (root_dev->disk->partition);
+
+  start_vcn.StartingVcn.QuadPart = 0;
+
+  filehd = grub_util_fd_open (core_path, GRUB_UTIL_FD_O_RDONLY);
+  if (!GRUB_UTIL_FD_IS_VALID (filehd))
+    grub_util_error (_("cannot open `%s': %s"), core_path,
+		     grub_util_fd_strerror ());
+
+  extbuf_size = sizeof (*extbuf) + sizeof (extbuf->Extents[0])
+    * ((core_size + 511) / 512);
+  extbuf = xmalloc (extbuf_size);
+
+  if (!DeviceIoControl(filehd, FSCTL_GET_RETRIEVAL_POINTERS,
+		       &start_vcn, sizeof (start_vcn),
+		       extbuf, extbuf_size, &rets, NULL))
+    grub_util_error ("FSCTL_GET_RETRIEVAL_POINTERS fails: %s",
+		     grub_util_fd_strerror ());
+
+  CloseHandle (filehd);
+
+  for (i = 0; i < extbuf->ExtentCount; i++)
+    {
+      for (j = 0; j < extbuf->Extents[i].NextVcn.QuadPart - curvcn; j++)
+	for (k = 0; k < sec_per_lcn; k++)
+	  callback ((extbuf->Extents[i].Lcn.QuadPart + j)
+		    * sec_per_lcn + first_lcn
+		    + k, 0, 512, hook_data);
+    }
+  free (extbuf);
+}