grub/grub-core/osdep/windows/blocklist.c
Vladimir Serbinenko cb72aa1809 Rewrite blocklist functions in order to get progress when
reading large extents and decrease amount of blocklist hook calls.
2013-11-01 23:28:03 +01:00

115 lines
3.7 KiB
C

/*
* 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;
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++)
callback (extbuf->Extents[i].Lcn.QuadPart
* sec_per_lcn + first_lcn,
0, 512 * sec_per_lcn * (extbuf->Extents[i].NextVcn.QuadPart - curvcn), hook_data);
free (extbuf);
}