116 lines
3.7 KiB
C
116 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",
|
|
(long long) sec_per_lcn, (long long) 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);
|
|
}
|