194b212c2SPaul Mackerras /* 294b212c2SPaul Mackerras * Copyright (C) Paul Mackerras 1997. 394b212c2SPaul Mackerras * 494b212c2SPaul Mackerras * Updates for PPC64 by Todd Inglett, Dave Engebretsen & Peter Bergner. 594b212c2SPaul Mackerras * 694b212c2SPaul Mackerras * This program is free software; you can redistribute it and/or 794b212c2SPaul Mackerras * modify it under the terms of the GNU General Public License 894b212c2SPaul Mackerras * as published by the Free Software Foundation; either version 994b212c2SPaul Mackerras * 2 of the License, or (at your option) any later version. 1094b212c2SPaul Mackerras */ 1194b212c2SPaul Mackerras #include <stdarg.h> 1294b212c2SPaul Mackerras #include <stddef.h> 1394b212c2SPaul Mackerras #include "elf.h" 1494b212c2SPaul Mackerras #include "page.h" 1594b212c2SPaul Mackerras #include "string.h" 1694b212c2SPaul Mackerras #include "stdio.h" 17b2c5f619SMark A. Greer #include "ops.h" 18ad9d2716SDavid Gibson #include "gunzip_util.h" 19b2c5f619SMark A. Greer #include "flatdevtree.h" 2094b212c2SPaul Mackerras 2194b212c2SPaul Mackerras extern void flush_cache(void *, unsigned long); 2294b212c2SPaul Mackerras 2394b212c2SPaul Mackerras extern char _start[]; 2494b212c2SPaul Mackerras extern char __bss_start[]; 2594b212c2SPaul Mackerras extern char _end[]; 2694b212c2SPaul Mackerras extern char _vmlinux_start[]; 2794b212c2SPaul Mackerras extern char _vmlinux_end[]; 2894b212c2SPaul Mackerras extern char _initrd_start[]; 2994b212c2SPaul Mackerras extern char _initrd_end[]; 30c888554bSMark A. Greer extern char _dtb_start[]; 31c888554bSMark A. Greer extern char _dtb_end[]; 3294b212c2SPaul Mackerras 33ad9d2716SDavid Gibson static struct gunzip_state gzstate; 34ad9d2716SDavid Gibson 3594b212c2SPaul Mackerras struct addr_range { 36*79c85419SDavid Gibson void *addr; 3794b212c2SPaul Mackerras unsigned long size; 3894b212c2SPaul Mackerras }; 3994b212c2SPaul Mackerras 40*79c85419SDavid Gibson struct elf_info { 41*79c85419SDavid Gibson unsigned long loadsize; 42*79c85419SDavid Gibson unsigned long memsize; 43*79c85419SDavid Gibson unsigned long elfoffset; 44*79c85419SDavid Gibson }; 4594b212c2SPaul Mackerras 46b2c5f619SMark A. Greer typedef void (*kernel_entry_t)(unsigned long, unsigned long, void *); 4794b212c2SPaul Mackerras 4894b212c2SPaul Mackerras #undef DEBUG 4994b212c2SPaul Mackerras 50*79c85419SDavid Gibson static int parse_elf64(void *hdr, struct elf_info *info) 5194b212c2SPaul Mackerras { 5294b212c2SPaul Mackerras Elf64_Ehdr *elf64 = hdr; 5394b212c2SPaul Mackerras Elf64_Phdr *elf64ph; 5494b212c2SPaul Mackerras unsigned int i; 5594b212c2SPaul Mackerras 5694b212c2SPaul Mackerras if (!(elf64->e_ident[EI_MAG0] == ELFMAG0 && 5794b212c2SPaul Mackerras elf64->e_ident[EI_MAG1] == ELFMAG1 && 5894b212c2SPaul Mackerras elf64->e_ident[EI_MAG2] == ELFMAG2 && 5994b212c2SPaul Mackerras elf64->e_ident[EI_MAG3] == ELFMAG3 && 6094b212c2SPaul Mackerras elf64->e_ident[EI_CLASS] == ELFCLASS64 && 6194b212c2SPaul Mackerras elf64->e_ident[EI_DATA] == ELFDATA2MSB && 6294b212c2SPaul Mackerras elf64->e_type == ET_EXEC && 6394b212c2SPaul Mackerras elf64->e_machine == EM_PPC64)) 6494b212c2SPaul Mackerras return 0; 6594b212c2SPaul Mackerras 6694b212c2SPaul Mackerras elf64ph = (Elf64_Phdr *)((unsigned long)elf64 + 6794b212c2SPaul Mackerras (unsigned long)elf64->e_phoff); 6894b212c2SPaul Mackerras for (i = 0; i < (unsigned int)elf64->e_phnum; i++, elf64ph++) 69158daa4cSOlaf Hering if (elf64ph->p_type == PT_LOAD) 7094b212c2SPaul Mackerras break; 7194b212c2SPaul Mackerras if (i >= (unsigned int)elf64->e_phnum) 7294b212c2SPaul Mackerras return 0; 7394b212c2SPaul Mackerras 74*79c85419SDavid Gibson info->loadsize = (unsigned long)elf64ph->p_filesz; 75*79c85419SDavid Gibson info->memsize = (unsigned long)elf64ph->p_memsz; 76*79c85419SDavid Gibson info->elfoffset = (unsigned long)elf64ph->p_offset; 7766a45dd3SPaul Mackerras 7894b212c2SPaul Mackerras return 1; 7994b212c2SPaul Mackerras } 8094b212c2SPaul Mackerras 81*79c85419SDavid Gibson static int parse_elf32(void *hdr, struct elf_info *info) 8294b212c2SPaul Mackerras { 8394b212c2SPaul Mackerras Elf32_Ehdr *elf32 = hdr; 8494b212c2SPaul Mackerras Elf32_Phdr *elf32ph; 8594b212c2SPaul Mackerras unsigned int i; 8694b212c2SPaul Mackerras 8794b212c2SPaul Mackerras if (!(elf32->e_ident[EI_MAG0] == ELFMAG0 && 8894b212c2SPaul Mackerras elf32->e_ident[EI_MAG1] == ELFMAG1 && 8994b212c2SPaul Mackerras elf32->e_ident[EI_MAG2] == ELFMAG2 && 9094b212c2SPaul Mackerras elf32->e_ident[EI_MAG3] == ELFMAG3 && 9194b212c2SPaul Mackerras elf32->e_ident[EI_CLASS] == ELFCLASS32 && 9294b212c2SPaul Mackerras elf32->e_ident[EI_DATA] == ELFDATA2MSB && 9394b212c2SPaul Mackerras elf32->e_type == ET_EXEC && 9494b212c2SPaul Mackerras elf32->e_machine == EM_PPC)) 9594b212c2SPaul Mackerras return 0; 9694b212c2SPaul Mackerras 9794b212c2SPaul Mackerras elf32ph = (Elf32_Phdr *) ((unsigned long)elf32 + elf32->e_phoff); 9894b212c2SPaul Mackerras for (i = 0; i < elf32->e_phnum; i++, elf32ph++) 99158daa4cSOlaf Hering if (elf32ph->p_type == PT_LOAD) 10094b212c2SPaul Mackerras break; 10194b212c2SPaul Mackerras if (i >= elf32->e_phnum) 10294b212c2SPaul Mackerras return 0; 10394b212c2SPaul Mackerras 104*79c85419SDavid Gibson info->loadsize = elf32ph->p_filesz; 105*79c85419SDavid Gibson info->memsize = elf32ph->p_memsz; 106*79c85419SDavid Gibson info->elfoffset = elf32ph->p_offset; 10794b212c2SPaul Mackerras return 1; 10894b212c2SPaul Mackerras } 10994b212c2SPaul Mackerras 110*79c85419SDavid Gibson static struct addr_range prep_kernel(void) 11130d8caf7Smostrows@watson.ibm.com { 112*79c85419SDavid Gibson char elfheader[256]; 113*79c85419SDavid Gibson void *vmlinuz_addr = _vmlinux_start; 114*79c85419SDavid Gibson unsigned long vmlinuz_size = _vmlinux_end - _vmlinux_start; 115*79c85419SDavid Gibson void *addr = 0; 116*79c85419SDavid Gibson struct elf_info ei; 11730d8caf7Smostrows@watson.ibm.com int len; 11866a45dd3SPaul Mackerras 11994b212c2SPaul Mackerras /* gunzip the ELF header of the kernel */ 120*79c85419SDavid Gibson gunzip_start(&gzstate, vmlinuz_addr, vmlinuz_size); 121ad9d2716SDavid Gibson gunzip_exactly(&gzstate, elfheader, sizeof(elfheader)); 12294b212c2SPaul Mackerras 123*79c85419SDavid Gibson if (!parse_elf64(elfheader, &ei) && !parse_elf32(elfheader, &ei)) { 12494b212c2SPaul Mackerras printf("Error: not a valid PPC32 or PPC64 ELF file!\n\r"); 12594b212c2SPaul Mackerras exit(); 12694b212c2SPaul Mackerras } 127b2c5f619SMark A. Greer if (platform_ops.image_hdr) 128b2c5f619SMark A. Greer platform_ops.image_hdr(elfheader); 12994b212c2SPaul Mackerras 130ad9d2716SDavid Gibson /* We need to alloc the memsize: gzip will expand the kernel 131ad9d2716SDavid Gibson * text/data, then possible rubbish we don't care about. But 132ad9d2716SDavid Gibson * the kernel bss must be claimed (it will be zero'd by the 133ad9d2716SDavid Gibson * kernel itself) 13494b212c2SPaul Mackerras */ 135*79c85419SDavid Gibson printf("Allocating 0x%lx bytes for kernel ...\n\r", ei.memsize); 136*79c85419SDavid Gibson 137*79c85419SDavid Gibson if (platform_ops.vmlinux_alloc) { 138*79c85419SDavid Gibson addr = platform_ops.vmlinux_alloc(ei.memsize); 139*79c85419SDavid Gibson } else { 140*79c85419SDavid Gibson if ((unsigned long)_start < ei.memsize) { 141*79c85419SDavid Gibson printf("Insufficient memory for kernel at address 0!" 142*79c85419SDavid Gibson " (_start=%lx)\n\r", _start); 14394b212c2SPaul Mackerras exit(); 14494b212c2SPaul Mackerras } 145*79c85419SDavid Gibson } 146*79c85419SDavid Gibson 147*79c85419SDavid Gibson /* Finally, gunzip the kernel */ 148*79c85419SDavid Gibson printf("gunzipping (0x%p <- 0x%p:0x%p)...", addr, 149*79c85419SDavid Gibson vmlinuz_addr, vmlinuz_addr+vmlinuz_size); 150*79c85419SDavid Gibson /* discard up to the actual load data */ 151*79c85419SDavid Gibson gunzip_discard(&gzstate, ei.elfoffset - sizeof(elfheader)); 152*79c85419SDavid Gibson len = gunzip_finish(&gzstate, addr, ei.memsize); 153*79c85419SDavid Gibson printf("done 0x%lx bytes\n\r", len); 154*79c85419SDavid Gibson 155*79c85419SDavid Gibson flush_cache(addr, ei.loadsize); 156*79c85419SDavid Gibson 157*79c85419SDavid Gibson return (struct addr_range){addr, ei.memsize}; 158*79c85419SDavid Gibson } 159*79c85419SDavid Gibson 160*79c85419SDavid Gibson static struct addr_range prep_initrd(struct addr_range vmlinux, 161*79c85419SDavid Gibson unsigned long initrd_addr, 162*79c85419SDavid Gibson unsigned long initrd_size) 163*79c85419SDavid Gibson { 164*79c85419SDavid Gibson void *devp; 165*79c85419SDavid Gibson u32 initrd_start, initrd_end; 166*79c85419SDavid Gibson 167*79c85419SDavid Gibson /* If we have an image attached to us, it overrides anything 168*79c85419SDavid Gibson * supplied by the loader. */ 169*79c85419SDavid Gibson if (_initrd_end > _initrd_start) { 170*79c85419SDavid Gibson printf("Attached initrd image at 0x%p-0x%p\n\r", 171*79c85419SDavid Gibson _initrd_start, _initrd_end); 172*79c85419SDavid Gibson initrd_addr = (unsigned long)_initrd_start; 173*79c85419SDavid Gibson initrd_size = _initrd_end - _initrd_start; 174*79c85419SDavid Gibson } else if (initrd_size > 0) { 175*79c85419SDavid Gibson printf("Using loader supplied ramdisk at 0x%lx-0x%lx\n\r", 176*79c85419SDavid Gibson initrd_addr, initrd_addr + initrd_size); 177*79c85419SDavid Gibson } 178*79c85419SDavid Gibson 179*79c85419SDavid Gibson /* If there's no initrd at all, we're done */ 180*79c85419SDavid Gibson if (! initrd_size) 181*79c85419SDavid Gibson return (struct addr_range){0, 0}; 18294b212c2SPaul Mackerras 18394b212c2SPaul Mackerras /* 184*79c85419SDavid Gibson * If the initrd is too low it will be clobbered when the 185*79c85419SDavid Gibson * kernel relocates to its final location. In this case, 186*79c85419SDavid Gibson * allocate a safer place and move it. 18794b212c2SPaul Mackerras */ 188*79c85419SDavid Gibson if (initrd_addr < vmlinux.size) { 189*79c85419SDavid Gibson void *old_addr = (void *)initrd_addr; 190*79c85419SDavid Gibson 191b2c5f619SMark A. Greer printf("Allocating 0x%lx bytes for initrd ...\n\r", 192*79c85419SDavid Gibson initrd_size); 193*79c85419SDavid Gibson initrd_addr = (unsigned long)malloc(initrd_size); 194*79c85419SDavid Gibson if (! initrd_addr) { 195b2c5f619SMark A. Greer printf("Can't allocate memory for initial " 196b2c5f619SMark A. Greer "ramdisk !\n\r"); 19794b212c2SPaul Mackerras exit(); 19894b212c2SPaul Mackerras } 199*79c85419SDavid Gibson printf("Relocating initrd 0x%p <- 0x%p (0x%lx bytes)\n\r", 200*79c85419SDavid Gibson initrd_addr, old_addr, initrd_size); 201*79c85419SDavid Gibson memmove((void *)initrd_addr, old_addr, initrd_size); 20294b212c2SPaul Mackerras } 20394b212c2SPaul Mackerras 204*79c85419SDavid Gibson printf("initrd head: 0x%lx\n\r", *((unsigned long *)initrd_addr)); 20594b212c2SPaul Mackerras 206*79c85419SDavid Gibson /* Tell the kernel initrd address via device tree */ 207*79c85419SDavid Gibson devp = finddevice("/chosen"); 208*79c85419SDavid Gibson if (! devp) { 209*79c85419SDavid Gibson printf("Device tree has no chosen node!\n\r"); 210*79c85419SDavid Gibson exit(); 211*79c85419SDavid Gibson } 212*79c85419SDavid Gibson 213*79c85419SDavid Gibson initrd_start = (u32)initrd_addr; 214*79c85419SDavid Gibson initrd_end = (u32)initrd_addr + initrd_size; 215*79c85419SDavid Gibson 216*79c85419SDavid Gibson setprop(devp, "linux,initrd-start", &initrd_start, 217*79c85419SDavid Gibson sizeof(initrd_start)); 218*79c85419SDavid Gibson setprop(devp, "linux,initrd-end", &initrd_end, sizeof(initrd_end)); 219*79c85419SDavid Gibson 220*79c85419SDavid Gibson return (struct addr_range){(void *)initrd_addr, initrd_size}; 22194b212c2SPaul Mackerras } 22294b212c2SPaul Mackerras 223b2c5f619SMark A. Greer /* A buffer that may be edited by tools operating on a zImage binary so as to 224b2c5f619SMark A. Greer * edit the command line passed to vmlinux (by setting /chosen/bootargs). 225b2c5f619SMark A. Greer * The buffer is put in it's own section so that tools may locate it easier. 226b2c5f619SMark A. Greer */ 227b2c5f619SMark A. Greer static char builtin_cmdline[COMMAND_LINE_SIZE] 228b2c5f619SMark A. Greer __attribute__((__section__("__builtin_cmdline"))); 229b2c5f619SMark A. Greer 230b2c5f619SMark A. Greer static void get_cmdline(char *buf, int size) 231b2c5f619SMark A. Greer { 232b2c5f619SMark A. Greer void *devp; 233b2c5f619SMark A. Greer int len = strlen(builtin_cmdline); 234b2c5f619SMark A. Greer 235b2c5f619SMark A. Greer buf[0] = '\0'; 236b2c5f619SMark A. Greer 237b2c5f619SMark A. Greer if (len > 0) { /* builtin_cmdline overrides dt's /chosen/bootargs */ 238b2c5f619SMark A. Greer len = min(len, size-1); 239b2c5f619SMark A. Greer strncpy(buf, builtin_cmdline, len); 240b2c5f619SMark A. Greer buf[len] = '\0'; 241b2c5f619SMark A. Greer } 242b2c5f619SMark A. Greer else if ((devp = finddevice("/chosen"))) 243b2c5f619SMark A. Greer getprop(devp, "bootargs", buf, size); 244b2c5f619SMark A. Greer } 245b2c5f619SMark A. Greer 246b2c5f619SMark A. Greer static void set_cmdline(char *buf) 247b2c5f619SMark A. Greer { 248b2c5f619SMark A. Greer void *devp; 249b2c5f619SMark A. Greer 250b2c5f619SMark A. Greer if ((devp = finddevice("/chosen"))) 251b2c5f619SMark A. Greer setprop(devp, "bootargs", buf, strlen(buf) + 1); 252b2c5f619SMark A. Greer } 253b2c5f619SMark A. Greer 254b2c5f619SMark A. Greer struct platform_ops platform_ops; 255b2c5f619SMark A. Greer struct dt_ops dt_ops; 256b2c5f619SMark A. Greer struct console_ops console_ops; 257b2c5f619SMark A. Greer 258b2c5f619SMark A. Greer void start(unsigned long a1, unsigned long a2, void *promptr, void *sp) 259b2c5f619SMark A. Greer { 260*79c85419SDavid Gibson struct addr_range vmlinux, initrd; 261b2c5f619SMark A. Greer kernel_entry_t kentry; 262b2c5f619SMark A. Greer char cmdline[COMMAND_LINE_SIZE]; 26335af89ebSDavid Gibson unsigned long ft_addr = 0; 264b2c5f619SMark A. Greer 265b2c5f619SMark A. Greer memset(__bss_start, 0, _end - __bss_start); 266b2c5f619SMark A. Greer memset(&platform_ops, 0, sizeof(platform_ops)); 267b2c5f619SMark A. Greer memset(&dt_ops, 0, sizeof(dt_ops)); 268b2c5f619SMark A. Greer memset(&console_ops, 0, sizeof(console_ops)); 269b2c5f619SMark A. Greer 270c888554bSMark A. Greer if (platform_init(promptr, _dtb_start, _dtb_end)) 271b2c5f619SMark A. Greer exit(); 272b2c5f619SMark A. Greer if (console_ops.open && (console_ops.open() < 0)) 273b2c5f619SMark A. Greer exit(); 274b2c5f619SMark A. Greer if (platform_ops.fixups) 275b2c5f619SMark A. Greer platform_ops.fixups(); 276b2c5f619SMark A. Greer 277b2c5f619SMark A. Greer printf("\n\rzImage starting: loaded at 0x%p (sp: 0x%p)\n\r", 278b2c5f619SMark A. Greer _start, sp); 279b2c5f619SMark A. Greer 280*79c85419SDavid Gibson vmlinux = prep_kernel(); 281*79c85419SDavid Gibson initrd = prep_initrd(vmlinux, a1, a2); 282b2c5f619SMark A. Greer 283b2c5f619SMark A. Greer /* If cmdline came from zimage wrapper or if we can edit the one 284b2c5f619SMark A. Greer * in the dt, print it out and edit it, if possible. 285b2c5f619SMark A. Greer */ 286b2c5f619SMark A. Greer if ((strlen(builtin_cmdline) > 0) || console_ops.edit_cmdline) { 287b2c5f619SMark A. Greer get_cmdline(cmdline, COMMAND_LINE_SIZE); 288b2c5f619SMark A. Greer printf("\n\rLinux/PowerPC load: %s", cmdline); 289b2c5f619SMark A. Greer if (console_ops.edit_cmdline) 290b2c5f619SMark A. Greer console_ops.edit_cmdline(cmdline, COMMAND_LINE_SIZE); 291b2c5f619SMark A. Greer printf("\n\r"); 292b2c5f619SMark A. Greer set_cmdline(cmdline); 293b2c5f619SMark A. Greer } 294b2c5f619SMark A. Greer 29535af89ebSDavid Gibson printf("Finalizing device tree..."); 29635af89ebSDavid Gibson if (dt_ops.finalize) 29735af89ebSDavid Gibson ft_addr = dt_ops.finalize(); 29835af89ebSDavid Gibson if (ft_addr) 29935af89ebSDavid Gibson printf(" flat tree at 0x%lx\n\r", ft_addr); 30035af89ebSDavid Gibson else 30135af89ebSDavid Gibson printf(" using OF tree (promptr=%p)\n\r", promptr); 30235af89ebSDavid Gibson 303b2c5f619SMark A. Greer if (console_ops.close) 304b2c5f619SMark A. Greer console_ops.close(); 305b2c5f619SMark A. Greer 306b2c5f619SMark A. Greer kentry = (kernel_entry_t) vmlinux.addr; 30735af89ebSDavid Gibson if (ft_addr) 30835af89ebSDavid Gibson kentry(ft_addr, 0, NULL); 309b2c5f619SMark A. Greer else 310*79c85419SDavid Gibson kentry((unsigned long)initrd.addr, initrd.size, promptr); 311b2c5f619SMark A. Greer 312b2c5f619SMark A. Greer /* console closed so printf below may not work */ 313b2c5f619SMark A. Greer printf("Error: Linux kernel returned to zImage boot wrapper!\n\r"); 314b2c5f619SMark A. Greer exit(); 315b2c5f619SMark A. Greer } 316