1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
/*
* 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 on windows: %s",
fs->name);
grub_util_info ("sec_per_lcn = %" GRUB_HOST_PRIuLONG_LONG
", first_lcn=%" GRUB_HOST_PRIuLONG_LONG,
(unsigned long long) sec_per_lcn,
(unsigned 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);
}
|
