389 lines
10 KiB
C
389 lines
10 KiB
C
|
#include <stdio.h>
|
||
|
|
#include <errno.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <string.h>
|
||
|
|
#define _GNU_SOURCE
|
||
|
|
#include <getopt.h>
|
||
|
|
#include "elf.h"
|
||
|
|
#include "elf_boot.h"
|
||
|
|
#include "convert.h"
|
||
|
|
#include "x86-linux.h"
|
||
|
|
#include "mkelfImage.h"
|
||
|
|
|
||
|
|
static unsigned char payload[] = {
|
||
|
|
#include "convert.bin.c"
|
||
|
|
};
|
||
|
|
|
||
|
|
struct kernel_info;
|
||
|
|
static void (*parse_kernel_type)(struct kernel_info *info, char *kernel_buf, size_t kernel_size);
|
||
|
|
static void parse_bzImage_kernel(struct kernel_info *info, char *kernel_buf, size_t kernel_size);
|
||
|
|
static void parse_elf32_kernel(struct kernel_info *info, char *kernel_buf, size_t kernel_size);
|
||
|
|
|
||
|
|
char *vmlinux_i386_probe(char *kernel_buf, off_t kernel_size)
|
||
|
|
{
|
||
|
|
Elf32_Ehdr *ehdr;
|
||
|
|
Elf32_Phdr *phdr;
|
||
|
|
int i;
|
||
|
|
int hdr1, hdr2;
|
||
|
|
ehdr = (Elf32_Ehdr *)kernel_buf;
|
||
|
|
if (
|
||
|
|
(ehdr->e_ident[EI_MAG0] != ELFMAG0) ||
|
||
|
|
(ehdr->e_ident[EI_MAG1] != ELFMAG1) ||
|
||
|
|
(ehdr->e_ident[EI_MAG2] != ELFMAG2) ||
|
||
|
|
(ehdr->e_ident[EI_MAG3] != ELFMAG3)) {
|
||
|
|
return "No ELF signature found on kernel\n";
|
||
|
|
}
|
||
|
|
if (ehdr->e_ident[EI_CLASS] != ELFCLASS32) {
|
||
|
|
return "Not a 32bit ELF kernel\n";
|
||
|
|
}
|
||
|
|
if (ehdr->e_ident[EI_DATA] != ELFDATA2LSB) {
|
||
|
|
return "Not a little endian ELF kernel\n";
|
||
|
|
}
|
||
|
|
if (le16_to_cpu(ehdr->e_type) != ET_EXEC) {
|
||
|
|
return "Not an executable kernel\n";
|
||
|
|
}
|
||
|
|
if (le16_to_cpu(ehdr->e_machine) != EM_386) {
|
||
|
|
return "Not an i386 kernel\n";
|
||
|
|
}
|
||
|
|
if ( (ehdr->e_ident[EI_VERSION] != EV_CURRENT) ||
|
||
|
|
(le32_to_cpu(ehdr->e_version) != EV_CURRENT)) {
|
||
|
|
return "Kernel not using ELF version 1.\n";
|
||
|
|
}
|
||
|
|
if (le16_to_cpu(ehdr->e_phentsize) != sizeof(*phdr)) {
|
||
|
|
return "Kernel uses bad program header size.\n";
|
||
|
|
}
|
||
|
|
phdr = (Elf32_Phdr *)(kernel_buf + le32_to_cpu(ehdr->e_phoff));
|
||
|
|
hdr1 = hdr2 = -1;
|
||
|
|
for(i = 0; i < le32_to_cpu(ehdr->e_phnum); i++) {
|
||
|
|
if (le32_to_cpu(phdr[i].p_type) != PT_LOAD)
|
||
|
|
continue;
|
||
|
|
if (((hdr1 != -1) &&
|
||
|
|
((le32_to_cpu(phdr[hdr1].p_paddr) & 0xfffffff) != 0x100000)) ||
|
||
|
|
(hdr2 != -1)) {
|
||
|
|
return "Too many PT_LOAD segments to be a linux kernel\n";
|
||
|
|
}
|
||
|
|
if (hdr1 == -1) {
|
||
|
|
hdr1 = i;
|
||
|
|
} else {
|
||
|
|
hdr2 = i;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (hdr1 == -1) {
|
||
|
|
return "No PT_LOAD segments!\n";
|
||
|
|
}
|
||
|
|
parse_kernel_type = parse_elf32_kernel;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char *bzImage_i386_probe(char *kernel_buf, off_t kernel_size)
|
||
|
|
{
|
||
|
|
struct x86_linux_header *hdr;
|
||
|
|
unsigned long offset;
|
||
|
|
int setup_sects;
|
||
|
|
hdr = (struct x86_linux_header *)kernel_buf;
|
||
|
|
|
||
|
|
if (le16_to_cpu(hdr->boot_sector_magic) != 0xaa55) {
|
||
|
|
return "No bootsector magic";
|
||
|
|
}
|
||
|
|
if (memcmp(hdr->header_magic, "HdrS", 4) != 0) {
|
||
|
|
return "Not a linux kernel";
|
||
|
|
}
|
||
|
|
|
||
|
|
if (le16_to_cpu(hdr->protocol_version) < 0x202) {
|
||
|
|
return "Kernel protcols version before 2.02 not supported";
|
||
|
|
}
|
||
|
|
|
||
|
|
setup_sects = hdr->setup_sects;
|
||
|
|
if (setup_sects == 0) {
|
||
|
|
setup_sects = 4;
|
||
|
|
}
|
||
|
|
offset = 512 + (512 *setup_sects);
|
||
|
|
if (offset > kernel_size) {
|
||
|
|
return "Not enough bytes";
|
||
|
|
}
|
||
|
|
parse_kernel_type = parse_bzImage_kernel;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
char *linux_i386_probe(char *kernel_buf, off_t kernel_size)
|
||
|
|
{
|
||
|
|
char *result;
|
||
|
|
result = "";
|
||
|
|
if (result) result = bzImage_i386_probe(kernel_buf, kernel_size);
|
||
|
|
if (result) result = vmlinux_i386_probe(kernel_buf, kernel_size);
|
||
|
|
if (result) result = bzImage_i386_probe(kernel_buf, kernel_size);
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
|
||
|
|
struct kernel_info
|
||
|
|
{
|
||
|
|
void *kernel;
|
||
|
|
size_t filesz;
|
||
|
|
size_t memsz;
|
||
|
|
size_t paddr;
|
||
|
|
size_t vaddr;
|
||
|
|
void *kernel2;
|
||
|
|
size_t filesz2;
|
||
|
|
size_t memsz2;
|
||
|
|
size_t paddr2;
|
||
|
|
size_t vaddr2;
|
||
|
|
size_t entry;
|
||
|
|
char *version;
|
||
|
|
};
|
||
|
|
|
||
|
|
static void parse_elf32_kernel(struct kernel_info *info, char *kernel_buf, size_t kernel_size)
|
||
|
|
{
|
||
|
|
Elf32_Ehdr *ehdr;
|
||
|
|
Elf32_Phdr *phdr;
|
||
|
|
int i;
|
||
|
|
int hdr1, hdr2;
|
||
|
|
ehdr = (Elf32_Ehdr *)kernel_buf;
|
||
|
|
phdr = (Elf32_Phdr *)(kernel_buf + ehdr->e_phoff);
|
||
|
|
hdr1 = hdr2 = -1;
|
||
|
|
for(i = 0; i < le16_to_cpu(ehdr->e_phnum); i++) {
|
||
|
|
if (le32_to_cpu(phdr[i].p_type) != PT_LOAD)
|
||
|
|
continue;
|
||
|
|
if (hdr2 != -1) {
|
||
|
|
die("Too many PT_LOAD segments to be a linux kernel\n");
|
||
|
|
}
|
||
|
|
if (hdr1 == -1) {
|
||
|
|
hdr1 = i;
|
||
|
|
} else {
|
||
|
|
hdr2 = i;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (hdr1 == -1) {
|
||
|
|
die("No PT_LOAD segments!\n");
|
||
|
|
}
|
||
|
|
info->kernel = kernel_buf + le32_to_cpu(phdr[hdr1].p_offset);
|
||
|
|
info->filesz = le32_to_cpu(phdr[hdr1].p_filesz);
|
||
|
|
info->memsz = le32_to_cpu(phdr[hdr1].p_memsz);
|
||
|
|
info->paddr = le32_to_cpu(phdr[hdr1].p_paddr) & 0xfffffff;
|
||
|
|
info->vaddr = le32_to_cpu(phdr[hdr1].p_vaddr);
|
||
|
|
|
||
|
|
if (hdr2 != -1) {
|
||
|
|
info->kernel2 = kernel_buf + le32_to_cpu(phdr[hdr2].p_offset);
|
||
|
|
info->filesz2 = le32_to_cpu(phdr[hdr2].p_filesz);
|
||
|
|
info->memsz2 = le32_to_cpu(phdr[hdr2].p_memsz);
|
||
|
|
info->paddr2 = le32_to_cpu(phdr[hdr2].p_paddr) & 0xfffffff;
|
||
|
|
info->vaddr2 = le32_to_cpu(phdr[hdr2].p_vaddr);
|
||
|
|
}
|
||
|
|
|
||
|
|
info->entry = 0x100000;
|
||
|
|
info->version = "unknown";
|
||
|
|
}
|
||
|
|
|
||
|
|
static void parse_bzImage_kernel(struct kernel_info *info, char *kernel_buf, size_t kernel_size)
|
||
|
|
{
|
||
|
|
struct x86_linux_header *hdr;
|
||
|
|
unsigned long offset;
|
||
|
|
int setup_sects;
|
||
|
|
hdr = (struct x86_linux_header *)kernel_buf;
|
||
|
|
setup_sects = hdr->setup_sects;
|
||
|
|
if (setup_sects == 0) {
|
||
|
|
setup_sects = 4;
|
||
|
|
}
|
||
|
|
offset = 512 + (512 *setup_sects);
|
||
|
|
|
||
|
|
info->kernel = kernel_buf + offset;
|
||
|
|
info->filesz = kernel_size - offset;
|
||
|
|
info->memsz = 0x700000;
|
||
|
|
info->paddr = 0x100000;
|
||
|
|
info->vaddr = 0x100000;
|
||
|
|
info->entry = info->paddr;
|
||
|
|
info->version = kernel_buf + 512 + le16_to_cpu(hdr->kver_addr);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void parse_kernel(struct kernel_info *info, char *kernel_buf, size_t kernel_size)
|
||
|
|
{
|
||
|
|
memset(info, 0, sizeof(*info));
|
||
|
|
if (parse_kernel_type) {
|
||
|
|
parse_kernel_type(info, kernel_buf, kernel_size);
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
die("Unknown kernel format");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void linux_i386_usage(void)
|
||
|
|
{
|
||
|
|
printf(
|
||
|
|
" --command-line=<string> Set the command line to <string>\n"
|
||
|
|
" --append=<string> Set the command line to <string>\n"
|
||
|
|
" --initrd=<filename> Set the initrd to <filename>\n"
|
||
|
|
" --ramdisk=<filename> Set the initrd to <filename>\n"
|
||
|
|
" --ramdisk-base=<addr> Set the initrd load address to <addr>\n"
|
||
|
|
);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#define OPT_CMDLINE OPT_MAX+0
|
||
|
|
#define OPT_RAMDISK OPT_MAX+1
|
||
|
|
#define OPT_RAMDISK_BASE OPT_MAX+2
|
||
|
|
|
||
|
|
#define DEFAULT_RAMDISK_BASE (8*1024*1024)
|
||
|
|
|
||
|
|
int linux_i386_mkelf(int argc, char **argv,
|
||
|
|
struct memelfheader *ehdr, char *kernel_buf, off_t kernel_size)
|
||
|
|
{
|
||
|
|
const char *ramdisk, *cmdline;
|
||
|
|
unsigned long ramdisk_base;
|
||
|
|
char *payload_buf, *ramdisk_buf;
|
||
|
|
off_t payload_size, ramdisk_size;
|
||
|
|
struct memelfphdr *phdr;
|
||
|
|
struct memelfnote *note;
|
||
|
|
struct kernel_info kinfo;
|
||
|
|
struct image_parameters *params;
|
||
|
|
int index;
|
||
|
|
|
||
|
|
int opt;
|
||
|
|
static const struct option options[] = {
|
||
|
|
MKELF_OPTIONS
|
||
|
|
{ "command-line", 1, 0, OPT_CMDLINE },
|
||
|
|
{ "append", 1, 0, OPT_CMDLINE },
|
||
|
|
{ "initrd", 1, 0, OPT_RAMDISK },
|
||
|
|
{ "ramdisk", 1, 0, OPT_RAMDISK },
|
||
|
|
{ "ramdisk-base", 1, 0, OPT_RAMDISK_BASE },
|
||
|
|
{ 0 , 0, 0, 0 },
|
||
|
|
};
|
||
|
|
static const char short_options[] = MKELF_OPT_STR;
|
||
|
|
|
||
|
|
ramdisk_base = DEFAULT_RAMDISK_BASE;
|
||
|
|
ramdisk = 0;
|
||
|
|
cmdline="";
|
||
|
|
|
||
|
|
while((opt = getopt_long(argc, argv, short_options, options, 0)) != -1) {
|
||
|
|
switch(opt) {
|
||
|
|
case '?':
|
||
|
|
error("Unknown option %s\n", argv[optind]);
|
||
|
|
break;
|
||
|
|
case OPT_RAMDISK_BASE:
|
||
|
|
{
|
||
|
|
char *end;
|
||
|
|
unsigned long base;
|
||
|
|
base = strtoul(optarg, &end, 0);
|
||
|
|
if ((end == optarg) || (*end != '\0')) {
|
||
|
|
error("Invalid ramdisk base\n");
|
||
|
|
}
|
||
|
|
ramdisk_base = base;
|
||
|
|
}
|
||
|
|
case OPT_RAMDISK:
|
||
|
|
ramdisk = optarg;
|
||
|
|
break;
|
||
|
|
case OPT_CMDLINE:
|
||
|
|
cmdline = optarg;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
ehdr->ei_class = ELFCLASS32;
|
||
|
|
ehdr->ei_data = ELFDATA2LSB;
|
||
|
|
ehdr->e_type = ET_EXEC;
|
||
|
|
ehdr->e_machine = EM_386;
|
||
|
|
|
||
|
|
/* locate the payload buffer */
|
||
|
|
payload_buf = payload;
|
||
|
|
payload_size = sizeof(payload);
|
||
|
|
|
||
|
|
/* slurp the input files */
|
||
|
|
ramdisk_buf = slurp_file(ramdisk, &ramdisk_size);
|
||
|
|
|
||
|
|
/* parse the kernel */
|
||
|
|
parse_kernel(&kinfo, kernel_buf, kernel_size);
|
||
|
|
|
||
|
|
/* Find the parameters */
|
||
|
|
params = (void *)(payload_buf + (payload_size - sizeof(*params)));
|
||
|
|
|
||
|
|
/* A sanity check against bad versions of binutils */
|
||
|
|
if (params->convert_magic != CONVERT_MAGIC) {
|
||
|
|
die("Internal error convert_magic %08x != %08x\n",
|
||
|
|
params->convert_magic, CONVERT_MAGIC);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Copy the command line */
|
||
|
|
strncpy(params->cmdline, cmdline, sizeof(params->cmdline));
|
||
|
|
params->cmdline[sizeof(params->cmdline)-1]= '\0';
|
||
|
|
|
||
|
|
|
||
|
|
/* Add a program header for the note section */
|
||
|
|
index = 4;
|
||
|
|
index += kinfo.kernel2 ? 1:0;
|
||
|
|
index += ramdisk_size ? 1:0;
|
||
|
|
phdr = add_program_headers(ehdr, index);
|
||
|
|
|
||
|
|
/* Fill in the program headers*/
|
||
|
|
phdr[0].p_type = PT_NOTE;
|
||
|
|
|
||
|
|
/* Fill in the converter program headers */
|
||
|
|
phdr[1].p_paddr = CONVERTLOC;
|
||
|
|
phdr[1].p_vaddr = CONVERTLOC;
|
||
|
|
phdr[1].p_filesz = payload_size;
|
||
|
|
phdr[1].p_memsz = payload_size + params->bss_size;
|
||
|
|
phdr[1].p_data = payload;
|
||
|
|
|
||
|
|
/* Reserve space for the REAL MODE DATA segment AND the GDT segment */
|
||
|
|
phdr[2].p_paddr = REAL_MODE_DATA_LOC;
|
||
|
|
phdr[2].p_vaddr = REAL_MODE_DATA_LOC;
|
||
|
|
phdr[2].p_filesz = 0;
|
||
|
|
phdr[2].p_memsz = (GDTLOC - REAL_MODE_DATA_LOC) + params->gdt_size;
|
||
|
|
phdr[2].p_data = 0;
|
||
|
|
|
||
|
|
phdr[3].p_paddr = kinfo.paddr;
|
||
|
|
phdr[3].p_vaddr = kinfo.vaddr;
|
||
|
|
phdr[3].p_filesz = kinfo.filesz;
|
||
|
|
phdr[3].p_memsz = kinfo.memsz;
|
||
|
|
phdr[3].p_data = kinfo.kernel;
|
||
|
|
|
||
|
|
index = 4;
|
||
|
|
/* Put the second kernel frament if present */
|
||
|
|
if (kinfo.kernel2) {
|
||
|
|
phdr[index].p_paddr = kinfo.paddr2;
|
||
|
|
phdr[index].p_vaddr = kinfo.vaddr2;
|
||
|
|
phdr[index].p_filesz = kinfo.filesz2;
|
||
|
|
phdr[index].p_memsz = kinfo.memsz2;
|
||
|
|
phdr[index].p_data = kinfo.kernel2;
|
||
|
|
index++;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Put the ramdisk at ramdisk base.
|
||
|
|
*/
|
||
|
|
params->initrd_start = params->initrd_size = 0;
|
||
|
|
if (ramdisk_size) {
|
||
|
|
phdr[index].p_paddr = ramdisk_base;
|
||
|
|
phdr[index].p_vaddr = ramdisk_base;
|
||
|
|
phdr[index].p_filesz = ramdisk_size;
|
||
|
|
phdr[index].p_memsz = ramdisk_size;
|
||
|
|
phdr[index].p_data = ramdisk_buf;
|
||
|
|
params->initrd_start = phdr[index].p_paddr;
|
||
|
|
params->initrd_size = phdr[index].p_filesz;
|
||
|
|
index++;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Set the start location */
|
||
|
|
params->entry = kinfo.entry;
|
||
|
|
ehdr->e_entry = phdr[1].p_paddr;
|
||
|
|
|
||
|
|
/* Setup the elf notes */
|
||
|
|
note = add_notes(ehdr, 3);
|
||
|
|
note[0].n_type = EIN_PROGRAM_NAME;
|
||
|
|
note[0].n_name = "ELFBoot";
|
||
|
|
note[0].n_desc = "Linux";
|
||
|
|
note[0].n_descsz = strlen(note[0].n_desc)+1;
|
||
|
|
|
||
|
|
note[1].n_type = EIN_PROGRAM_VERSION;
|
||
|
|
note[1].n_name = "ELFBoot";
|
||
|
|
note[1].n_desc = kinfo.version;
|
||
|
|
note[1].n_descsz = strlen(note[1].n_desc);
|
||
|
|
|
||
|
|
note[2].n_type = EIN_PROGRAM_CHECKSUM;
|
||
|
|
note[2].n_name = "ELFBoot";
|
||
|
|
note[2].n_desc = 0;
|
||
|
|
note[2].n_descsz = 2;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|