345 lines
9.7 KiB
C
345 lines
9.7 KiB
C
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 2004,2005,2006 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 <stdio.h>
|
|
#include <stdint.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <getopt.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <grub/elf.h>
|
|
#include <grub/util/misc.h>
|
|
#include <grub/util/resolve.h>
|
|
#include <grub/kernel.h>
|
|
#include <grub/machine/kernel.h>
|
|
|
|
#define ALIGN_UP(addr, align) ((long)((char *)addr + align - 1) & ~(align - 1))
|
|
|
|
#define GRUB_IEEE1275_NOTE_NAME "PowerPC"
|
|
#define GRUB_IEEE1275_NOTE_TYPE 0x1275
|
|
|
|
/* These structures are defined according to the CHRP binding to IEEE1275,
|
|
"Client Program Format" section. */
|
|
|
|
struct grub_ieee1275_note_hdr
|
|
{
|
|
grub_uint32_t namesz;
|
|
grub_uint32_t descsz;
|
|
grub_uint32_t type;
|
|
char name[sizeof (GRUB_IEEE1275_NOTE_NAME)];
|
|
};
|
|
|
|
struct grub_ieee1275_note_desc
|
|
{
|
|
grub_uint32_t real_mode;
|
|
grub_uint32_t real_base;
|
|
grub_uint32_t real_size;
|
|
grub_uint32_t virt_base;
|
|
grub_uint32_t virt_size;
|
|
grub_uint32_t load_base;
|
|
};
|
|
|
|
struct grub_ieee1275_note
|
|
{
|
|
struct grub_ieee1275_note_hdr header;
|
|
struct grub_ieee1275_note_desc descriptor;
|
|
};
|
|
|
|
void
|
|
load_note (Elf32_Phdr *phdr, FILE *out)
|
|
{
|
|
struct grub_ieee1275_note note;
|
|
int note_size = sizeof (struct grub_ieee1275_note);
|
|
|
|
grub_util_info ("adding CHRP NOTE segment");
|
|
|
|
note.header.namesz = grub_cpu_to_be32 (sizeof (GRUB_IEEE1275_NOTE_NAME));
|
|
note.header.descsz = grub_cpu_to_be32 (note_size);
|
|
note.header.type = grub_cpu_to_be32 (GRUB_IEEE1275_NOTE_TYPE);
|
|
strcpy (note.header.name, GRUB_IEEE1275_NOTE_NAME);
|
|
note.descriptor.real_mode = grub_cpu_to_be32 (0xffffffff);
|
|
note.descriptor.real_base = grub_cpu_to_be32 (0x00c00000);
|
|
note.descriptor.real_size = grub_cpu_to_be32 (0xffffffff);
|
|
note.descriptor.virt_base = grub_cpu_to_be32 (0xffffffff);
|
|
note.descriptor.virt_size = grub_cpu_to_be32 (0xffffffff);
|
|
note.descriptor.load_base = grub_cpu_to_be32 (0x00004000);
|
|
|
|
/* Write the note data to the new segment. */
|
|
grub_util_write_image_at (¬e, note_size,
|
|
grub_be_to_cpu32 (phdr->p_offset), out);
|
|
|
|
/* Fill in the rest of the segment header. */
|
|
phdr->p_type = grub_cpu_to_be32 (PT_NOTE);
|
|
phdr->p_flags = grub_cpu_to_be32 (PF_R);
|
|
phdr->p_align = grub_cpu_to_be32 (sizeof (long));
|
|
phdr->p_vaddr = 0;
|
|
phdr->p_paddr = 0;
|
|
phdr->p_filesz = grub_cpu_to_be32 (note_size);
|
|
phdr->p_memsz = 0;
|
|
}
|
|
|
|
void
|
|
load_modules (Elf32_Phdr *phdr, const char *dir, char *mods[], FILE *out)
|
|
{
|
|
char *module_img;
|
|
struct grub_util_path_list *path_list;
|
|
struct grub_util_path_list *p;
|
|
struct grub_module_info *modinfo;
|
|
size_t offset;
|
|
size_t total_module_size;
|
|
|
|
path_list = grub_util_resolve_dependencies (dir, "moddep.lst", mods);
|
|
|
|
offset = sizeof (struct grub_module_info);
|
|
total_module_size = sizeof (struct grub_module_info);
|
|
for (p = path_list; p; p = p->next)
|
|
{
|
|
total_module_size += (grub_util_get_image_size (p->name)
|
|
+ sizeof (struct grub_module_header));
|
|
}
|
|
|
|
grub_util_info ("the total module size is 0x%x", total_module_size);
|
|
|
|
module_img = xmalloc (total_module_size);
|
|
modinfo = (struct grub_module_info *) module_img;
|
|
modinfo->magic = grub_cpu_to_be32 (GRUB_MODULE_MAGIC);
|
|
modinfo->offset = grub_cpu_to_be32 (sizeof (struct grub_module_info));
|
|
modinfo->size = grub_cpu_to_be32 (total_module_size);
|
|
|
|
/* Load all the modules, with headers, into module_img. */
|
|
for (p = path_list; p; p = p->next)
|
|
{
|
|
struct grub_module_header *header;
|
|
size_t mod_size;
|
|
|
|
grub_util_info ("adding module %s", p->name);
|
|
|
|
mod_size = grub_util_get_image_size (p->name);
|
|
|
|
header = (struct grub_module_header *) (module_img + offset);
|
|
header->offset = grub_cpu_to_be32 (sizeof (*header));
|
|
header->size = grub_cpu_to_be32 (mod_size + sizeof (*header));
|
|
|
|
grub_util_load_image (p->name, module_img + offset + sizeof (*header));
|
|
|
|
offset += sizeof (*header) + mod_size;
|
|
}
|
|
|
|
/* Write the module data to the new segment. */
|
|
grub_util_write_image_at (module_img, total_module_size,
|
|
grub_cpu_to_be32 (phdr->p_offset), out);
|
|
|
|
/* Fill in the rest of the segment header. */
|
|
phdr->p_type = grub_cpu_to_be32 (PT_LOAD);
|
|
phdr->p_flags = grub_cpu_to_be32 (PF_R | PF_W | PF_X);
|
|
phdr->p_align = grub_cpu_to_be32 (sizeof (long));
|
|
phdr->p_vaddr = grub_cpu_to_be32 (GRUB_IEEE1275_MODULE_BASE);
|
|
phdr->p_paddr = grub_cpu_to_be32 (GRUB_IEEE1275_MODULE_BASE);
|
|
phdr->p_filesz = grub_cpu_to_be32 (total_module_size);
|
|
phdr->p_memsz = grub_cpu_to_be32 (total_module_size);
|
|
}
|
|
|
|
void
|
|
add_segments (char *dir, FILE *out, int chrp, char *mods[])
|
|
{
|
|
Elf32_Ehdr ehdr;
|
|
Elf32_Phdr *phdrs = NULL;
|
|
Elf32_Phdr *phdr;
|
|
FILE *in;
|
|
char *kernel_path;
|
|
off_t phdroff;
|
|
int i;
|
|
|
|
/* Read ELF header. */
|
|
kernel_path = grub_util_get_path (dir, "grubof");
|
|
in = fopen (kernel_path, "rb");
|
|
if (! in)
|
|
grub_util_error ("cannot open %s", kernel_path);
|
|
|
|
grub_util_read_at (&ehdr, sizeof (ehdr), 0, in);
|
|
|
|
phdrs = xmalloc (grub_be_to_cpu16 (ehdr.e_phentsize)
|
|
* (grub_be_to_cpu16 (ehdr.e_phnum) + 2));
|
|
/* Copy all existing segments. */
|
|
for (i = 0; i < grub_be_to_cpu16 (ehdr.e_phnum); i++)
|
|
{
|
|
char *segment_img;
|
|
|
|
phdr = phdrs + i;
|
|
|
|
/* Read segment header. */
|
|
grub_util_read_at (phdr, sizeof (Elf32_Phdr),
|
|
(grub_be_to_cpu32 (ehdr.e_phoff)
|
|
+ (i * grub_be_to_cpu16 (ehdr.e_phentsize))),
|
|
in);
|
|
grub_util_info ("copying segment %d, type %d", i,
|
|
grub_be_to_cpu32 (phdr->p_type));
|
|
|
|
/* Read segment data and write it to new file. */
|
|
segment_img = xmalloc (grub_be_to_cpu32 (phdr->p_filesz));
|
|
|
|
grub_util_read_at (segment_img, grub_be_to_cpu32 (phdr->p_filesz),
|
|
grub_be_to_cpu32 (phdr->p_offset), in);
|
|
grub_util_write_image_at (segment_img, grub_be_to_cpu32 (phdr->p_filesz),
|
|
grub_be_to_cpu32 (phdr->p_offset), out);
|
|
|
|
free (segment_img);
|
|
}
|
|
|
|
if (mods[0] != NULL)
|
|
{
|
|
/* Construct new segment header for modules. */
|
|
phdr = phdrs + grub_be_to_cpu16 (ehdr.e_phnum);
|
|
ehdr.e_phnum = grub_cpu_to_be16 (grub_be_to_cpu16 (ehdr.e_phnum) + 1);
|
|
|
|
/* Fill in p_offset so the callees know where to write. */
|
|
phdr->p_offset = grub_cpu_to_be32 (ALIGN_UP (grub_util_get_fp_size (out),
|
|
sizeof (long)));
|
|
|
|
load_modules (phdr, dir, mods, out);
|
|
}
|
|
|
|
if (chrp)
|
|
{
|
|
/* Construct new segment header for the CHRP note. */
|
|
phdr = phdrs + grub_be_to_cpu16 (ehdr.e_phnum);
|
|
ehdr.e_phnum = grub_cpu_to_be16 (grub_be_to_cpu16 (ehdr.e_phnum) + 1);
|
|
|
|
/* Fill in p_offset so the callees know where to write. */
|
|
phdr->p_offset = grub_cpu_to_be32 (ALIGN_UP (grub_util_get_fp_size (out),
|
|
sizeof (long)));
|
|
|
|
load_note (phdr, out);
|
|
}
|
|
|
|
/* Don't bother preserving the section headers. */
|
|
ehdr.e_shoff = 0;
|
|
ehdr.e_shnum = 0;
|
|
ehdr.e_shstrndx = 0;
|
|
|
|
/* Append entire segment table to the file. */
|
|
phdroff = ALIGN_UP (grub_util_get_fp_size (out), sizeof (long));
|
|
grub_util_write_image_at (phdrs, grub_be_to_cpu16 (ehdr.e_phentsize)
|
|
* grub_be_to_cpu16 (ehdr.e_phnum), phdroff,
|
|
out);
|
|
|
|
/* Write ELF header. */
|
|
ehdr.e_phoff = grub_cpu_to_be32 (phdroff);
|
|
grub_util_write_image_at (&ehdr, sizeof (ehdr), 0, out);
|
|
|
|
free (phdrs);
|
|
free (kernel_path);
|
|
}
|
|
|
|
static struct option options[] =
|
|
{
|
|
{"directory", required_argument, 0, 'd'},
|
|
{"output", required_argument, 0, 'o'},
|
|
{"help", no_argument, 0, 'h'},
|
|
{"note", no_argument, 0, 'n'},
|
|
{"version", no_argument, 0, 'V'},
|
|
{"verbose", no_argument, 0, 'v'},
|
|
{ 0, 0, 0, 0 },
|
|
};
|
|
|
|
static void
|
|
usage (int status)
|
|
{
|
|
if (status)
|
|
fprintf (stderr, "Try ``grub-mkimage --help'' for more information.\n");
|
|
else
|
|
printf ("\
|
|
Usage: grub-mkimage -o FILE [OPTION]... [MODULES]\n\
|
|
\n\
|
|
Make a bootable image of GRUB.\n\
|
|
\n\
|
|
-d, --directory=DIR use images and modules under DIR [default=%s]\n\
|
|
-o, --output=FILE output a generated image to FILE\n\
|
|
-h, --help display this message and exit\n\
|
|
-n, --note add NOTE segment for CHRP Open Firmware\n\
|
|
-V, --version print version information and exit\n\
|
|
-v, --verbose print verbose messages\n\
|
|
\n\
|
|
Report bugs to <%s>.\n\
|
|
", GRUB_LIBDIR, PACKAGE_BUGREPORT);
|
|
|
|
exit (status);
|
|
}
|
|
|
|
int
|
|
main (int argc, char *argv[])
|
|
{
|
|
FILE *fp;
|
|
char *output = NULL;
|
|
char *dir = NULL;
|
|
int chrp = 0;
|
|
|
|
progname = "grub-mkimage";
|
|
|
|
while (1)
|
|
{
|
|
int c = getopt_long (argc, argv, "d:o:hVvn", options, 0);
|
|
if (c == -1)
|
|
break;
|
|
|
|
switch (c)
|
|
{
|
|
case 'd':
|
|
if (dir)
|
|
free (dir);
|
|
dir = xstrdup (optarg);
|
|
break;
|
|
case 'h':
|
|
usage (0);
|
|
break;
|
|
case 'n':
|
|
chrp = 1;
|
|
break;
|
|
case 'o':
|
|
if (output)
|
|
free (output);
|
|
output = xstrdup (optarg);
|
|
break;
|
|
case 'V':
|
|
printf ("grub-mkimage (%s) %s\n", PACKAGE_NAME, PACKAGE_VERSION);
|
|
return 0;
|
|
case 'v':
|
|
verbosity++;
|
|
break;
|
|
default:
|
|
usage (1);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!output)
|
|
usage (1);
|
|
|
|
fp = fopen (output, "wb");
|
|
if (! fp)
|
|
grub_util_error ("cannot open %s", output);
|
|
|
|
add_segments (dir ? : GRUB_LIBDIR, fp, chrp, argv + optind);
|
|
|
|
fclose (fp);
|
|
|
|
return 0;
|
|
}
|