134 lines
3.7 KiB
C
134 lines
3.7 KiB
C
/* grub-setup.c - make GRUB usable */
|
|
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 1999,2000,2001,2002,2003,2004,2005,2006,2007,2008,2009,2010,2011 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/file.h>
|
|
#include <grub/partition.h>
|
|
#include <grub/util/misc.h>
|
|
#include <grub/util/install.h>
|
|
#include <grub/emu/hostdisk.h>
|
|
|
|
#include <string.h>
|
|
|
|
#define MAX_TRIES 5
|
|
|
|
void
|
|
grub_install_get_blocklist (grub_device_t root_dev,
|
|
const char *core_path, const char *core_img,
|
|
size_t core_size,
|
|
void (*callback) (grub_disk_addr_t sector,
|
|
unsigned offset,
|
|
unsigned length,
|
|
void *data),
|
|
void *hook_data)
|
|
{
|
|
int i;
|
|
char *tmp_img;
|
|
char *core_path_dev;
|
|
|
|
core_path_dev = grub_make_system_path_relative_to_its_root (core_path);
|
|
|
|
/* Make sure that GRUB reads the identical image as the OS. */
|
|
tmp_img = xmalloc (core_size);
|
|
|
|
for (i = 0; i < MAX_TRIES; i++)
|
|
{
|
|
grub_file_t file;
|
|
|
|
grub_util_info ((i == 0) ? _("attempting to read the core image `%s' from GRUB")
|
|
: _("attempting to read the core image `%s' from GRUB again"),
|
|
core_path_dev);
|
|
|
|
grub_disk_cache_invalidate_all ();
|
|
|
|
grub_file_filter_disable_compression ();
|
|
file = grub_file_open (core_path_dev);
|
|
if (file)
|
|
{
|
|
if (grub_file_size (file) != core_size)
|
|
grub_util_info ("succeeded in opening the core image but the size is different (%d != %d)",
|
|
(int) grub_file_size (file), (int) core_size);
|
|
else if (grub_file_read (file, tmp_img, core_size)
|
|
!= (grub_ssize_t) core_size)
|
|
grub_util_info ("succeeded in opening the core image but cannot read %d bytes",
|
|
(int) core_size);
|
|
else if (memcmp (core_img, tmp_img, core_size) != 0)
|
|
{
|
|
#if 0
|
|
FILE *dump;
|
|
FILE *dump2;
|
|
|
|
dump = fopen ("dump.img", "wb");
|
|
if (dump)
|
|
{
|
|
fwrite (tmp_img, 1, core_size, dump);
|
|
fclose (dump);
|
|
}
|
|
|
|
dump2 = fopen ("dump2.img", "wb");
|
|
if (dump2)
|
|
{
|
|
fwrite (core_img, 1, core_size, dump2);
|
|
fclose (dump2);
|
|
}
|
|
|
|
#endif
|
|
grub_util_info ("succeeded in opening the core image but the data is different");
|
|
}
|
|
else
|
|
{
|
|
grub_file_close (file);
|
|
break;
|
|
}
|
|
|
|
grub_file_close (file);
|
|
}
|
|
else
|
|
grub_util_info ("couldn't open the core image");
|
|
|
|
if (grub_errno)
|
|
grub_util_info ("error message = %s", grub_errmsg);
|
|
|
|
grub_errno = GRUB_ERR_NONE;
|
|
grub_util_biosdisk_flush (root_dev->disk);
|
|
sleep (1);
|
|
}
|
|
|
|
if (i == MAX_TRIES)
|
|
grub_util_error (_("cannot read `%s' correctly"), core_path_dev);
|
|
|
|
grub_file_t file;
|
|
/* Now read the core image to determine where the sectors are. */
|
|
grub_file_filter_disable_compression ();
|
|
file = grub_file_open (core_path_dev);
|
|
if (! file)
|
|
grub_util_error ("%s", grub_errmsg);
|
|
|
|
file->read_hook = callback;
|
|
file->read_hook_data = hook_data;
|
|
if (grub_file_read (file, tmp_img, core_size) != (grub_ssize_t) core_size)
|
|
grub_util_error ("%s", _("failed to read the sectors of the core image"));
|
|
|
|
grub_file_close (file);
|
|
free (tmp_img);
|
|
|
|
free (core_path_dev);
|
|
}
|