diff options
Diffstat (limited to 'util/mkelfImage/linux-i386/mkelf-linux-i386.c')
| -rw-r--r-- | util/mkelfImage/linux-i386/mkelf-linux-i386.c | 388 |
1 files changed, 388 insertions, 0 deletions
diff --git a/util/mkelfImage/linux-i386/mkelf-linux-i386.c b/util/mkelfImage/linux-i386/mkelf-linux-i386.c new file mode 100644 index 0000000000..200d4a55f9 --- /dev/null +++ b/util/mkelfImage/linux-i386/mkelf-linux-i386.c @@ -0,0 +1,388 @@ +#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; +} + |