vbatts/grub
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

2009-04-06 Vladimir Serbinenko <phcoder@gmail.com>

Browse source 
Parttool

	* parttool/pcpart.c: new file
	* commands/parttool.c: likewise
	* conf/common.rmk (pkglib_MODULES): Added parttool.mod and pcpart.mod
	(parttool_mod_SOURCES): new variable
	(parttool_mod_CFLAGS): likewise
	(parttool_mod_LDFLAGS): likewise
	(pcpart_mod_SOURCES): likewise
	(pcpart_mod_CFLAGS): likewise
	(pcpart_mod_LDFLAGS): likewise
	* conf/i386-coreboot.rmk (grub_emu_SOURCES): added commands/parttool.c 
	and parttool/pcpart.c
	* conf/i386-efi.rmk: likewise
	* conf/i386-ieee1275.rmk: likewise
	* conf/i386-pc.rmk: likewise
	* conf/powerpc-ieee1275.rmk: likewise
	* conf/sparc64-ieee1275.rmk: likewise
	* conf/x86_64-ieee1275.rmk: likewise
This commit is contained in:
phcoder 2009-04-06 09:51:01 +00:00
parent 05aaebfb7b
commit ea761d4014
19 changed files with 671 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

152
parttool/pcpart.c Normal file
View file

@ -0,0 +1,152 @@
/* pcpart.c - manipulate fdisk partitions */
/*
* GRUB -- GRand Unified Bootloader
* Copyright (C) 2009 Free Software Foundation, Inc.
*
* This program 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 2 of the License, or
* (at your option) any later version.
*
* This program 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 this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
#include <grub/types.h>
#include <grub/misc.h>
#include <grub/mm.h>
#include <grub/err.h>
#include <grub/pc_partition.h>
#include <grub/device.h>
#include <grub/disk.h>
#include <grub/partition.h>
#include <grub/parttool.h>
static int activate_table_handle = -1;
static int type_table_handle = -1;
static struct grub_parttool_argdesc grub_pcpart_bootargs[] =
{
{"boot", "Make partition active", GRUB_PARTTOOL_ARG_BOOL},
{0, 0, 0}
};
static grub_err_t grub_pcpart_boot (const grub_device_t dev,
const struct grub_parttool_args *args)
{
int i, index;
grub_partition_t part;
struct grub_pc_partition_mbr mbr;
if (dev->disk->partition->offset)
return grub_error (GRUB_ERR_BAD_ARGUMENT, "not a primary partition");
index = dev->disk->partition->index;
part = dev->disk->partition;
dev->disk->partition = 0;
/* Read the MBR. */
if (grub_disk_read (dev->disk, 0, 0, sizeof (mbr), (char *) &mbr))
{
dev->disk->partition = part;
return grub_errno;
}
if (args[0].set && args[0].bool)
{
for (i = 0; i < 4; i++)
mbr.entries[i].flag = 0x0;
mbr.entries[index].flag = 0x80;
}
else
mbr.entries[index].flag = 0x0;
/* Write the MBR. */
grub_disk_write (dev->disk, 0, 0, sizeof (mbr), (char *) &mbr);
dev->disk->partition = part;
return GRUB_ERR_NONE;
}
static struct grub_parttool_argdesc grub_pcpart_typeargs[] =
{
{"type", "Change partition type", GRUB_PARTTOOL_ARG_VAL},
{"hidden", "Make partition hidden", GRUB_PARTTOOL_ARG_BOOL},
{0, 0, 0}
};
static grub_err_t grub_pcpart_type (const grub_device_t dev,
const struct grub_parttool_args *args)
{
int index;
grub_uint8_t type;
grub_partition_t part;
struct grub_pc_partition_mbr mbr;
index = dev->disk->partition->index;
part = dev->disk->partition;
dev->disk->partition = 0;
/* Read the parttable. */
if (grub_disk_read (dev->disk, part->offset, 0,
sizeof (mbr), (char *) &mbr))
{
dev->disk->partition = part;
return grub_errno;
}
if (args[0].set)
type = grub_strtoul (args[0].str, 0, 0);
else
type = mbr.entries[index].type;
if (args[1].set)
{
if (args[1].bool)
type |= GRUB_PC_PARTITION_TYPE_HIDDEN_FLAG;
else
type &= ~GRUB_PC_PARTITION_TYPE_HIDDEN_FLAG;
}
if (grub_pc_partition_is_empty (type)
|| grub_pc_partition_is_extended (type))
{
dev->disk->partition = part;
return grub_error (GRUB_ERR_BAD_ARGUMENT, "invalid type");
}
mbr.entries[index].type = type;
/* Write the parttable. */
grub_disk_write (dev->disk, part->offset, 0,
sizeof (mbr), (char *) &mbr);
dev->disk->partition = part;
return GRUB_ERR_NONE;
}
GRUB_MOD_INIT (pcpart)
{
(void)mod; /* To stop warning. */
activate_table_handle = grub_parttool_register ("pc_partition_map",
grub_pcpart_boot,
grub_pcpart_bootargs);
type_table_handle = grub_parttool_register ("pc_partition_map",
grub_pcpart_type,
grub_pcpart_typeargs);
}
GRUB_MOD_FINI(pcpart)
{
grub_parttool_unregister (activate_table_handle);
grub_parttool_unregister (type_table_handle);
}