vaseboot/VasEBoot-core/loader/arm/linux.c

/* linux.c - boot Linux */
/*
 *  VAS_EBOOT  --  GRand Unified Bootloader
 *  Copyright (C) 2013  Free Software Foundation, Inc.
 *
 *  VAS_EBOOT 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 3 of the License, or
 *  (at your option) any later version.
 *
 *  VAS_EBOOT 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 VAS_EBOOT.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <VasEBoot/dl.h>
#include <VasEBoot/fdt.h>
#include <VasEBoot/file.h>
#include <VasEBoot/loader.h>
#include <VasEBoot/mm.h>
#include <VasEBoot/misc.h>
#include <VasEBoot/command.h>
#include <VasEBoot/cache.h>
#include <VasEBoot/cpu/linux.h>
#include <VasEBoot/efi/efi.h>
#include <VasEBoot/lib/cmdline.h>
#include <VasEBoot/linux.h>
#include <VasEBoot/verify.h>

VAS_EBOOT_MOD_LICENSE ("GPLv3+");

static VasEBoot_dl_t my_mod;

static VasEBoot_addr_t initrd_start;
static VasEBoot_addr_t initrd_end;

static VasEBoot_addr_t linux_addr;
static VasEBoot_size_t linux_size;

static char *linux_args;

static VasEBoot_uint32_t machine_type;
static const void *current_fdt;

typedef void (*kernel_entry_t) (int, unsigned long, void *);

#define LINUX_PHYS_OFFSET        (0x00008000)
#define LINUX_INITRD_PHYS_OFFSET (LINUX_PHYS_OFFSET + 0x03000000)
#define LINUX_FDT_PHYS_OFFSET    (LINUX_INITRD_PHYS_OFFSET - 0x10000)

static VasEBoot_size_t
get_atag_size (const VasEBoot_uint32_t *atag)
{
  const VasEBoot_uint32_t *atag0 = atag;
  while (atag[0] && atag[1])
    atag += atag[0];
  return atag - atag0;
}

/*
 * linux_prepare_fdt():
 *   Prepares a loaded FDT for being passed to Linux.
 *   Merges in command line parameters and sets up initrd addresses.
 */
static VasEBoot_err_t
linux_prepare_atag (void *target_atag)
{
  const VasEBoot_uint32_t *atag_orig = (const VasEBoot_uint32_t *) current_fdt;
  VasEBoot_uint32_t *tmp_atag, *to;
  const VasEBoot_uint32_t *from;
  VasEBoot_size_t tmp_size;
  VasEBoot_size_t arg_size = VasEBoot_strlen (linux_args);
  char *cmdline_orig = NULL;
  VasEBoot_size_t cmdline_orig_len = 0;

  /* some place for cmdline, initrd and terminator.  */
  tmp_size = get_atag_size (atag_orig) + 20 + (arg_size) / 4;
  tmp_atag = VasEBoot_calloc (tmp_size, sizeof (VasEBoot_uint32_t));
  if (!tmp_atag)
    return VasEBoot_errno;

  for (from = atag_orig, to = tmp_atag; from[0] && from[1];
       from += from[0])
    switch (from[1])
      {
      case 0x54410004:
      case 0x54410005:
      case 0x54420005:
	break;
      case 0x54410009:
	if (*(char *) (from + 2))
	  {
	    cmdline_orig = (char *) (from + 2);
	    cmdline_orig_len = VasEBoot_strlen (cmdline_orig) + 1;
	  }
	break;
      default:
	VasEBoot_memcpy (to, from, sizeof (VasEBoot_uint32_t) * from[0]);
	to += from[0];
	break;
      }

  VasEBoot_dprintf ("linux", "linux inherited args: '%s'\n",
		cmdline_orig ? : "");
  VasEBoot_dprintf ("linux", "linux_args: '%s'\n", linux_args);

  /* Generate and set command line */
  to[0] = 3 + (arg_size + cmdline_orig_len) / 4;
  to[1] = 0x54410009;
  if (cmdline_orig)
    {
      VasEBoot_memcpy ((char *) to + 8, cmdline_orig, cmdline_orig_len - 1);
      *((char *) to + 8 + cmdline_orig_len - 1) = ' ';
    }
  VasEBoot_memcpy ((char *) to + 8 + cmdline_orig_len, linux_args, arg_size);
  VasEBoot_memset ((char *) to + 8 + cmdline_orig_len + arg_size, 0,
	       4 - ((arg_size + cmdline_orig_len) & 3));
  to += to[0];

  if (initrd_start && initrd_end)
    {
      /*
       * We're using physical addresses, so even if we have LPAE, we're
       * restricted to a 32-bit address space.
       */
      VasEBoot_dprintf ("loader", "Initrd @ 0x%08x-0x%08x\n",
		    initrd_start, initrd_end);

      to[0] = 4;
      to[1] = 0x54420005;
      to[2] = initrd_start;
      to[3] = initrd_end - initrd_start;
      to += 4;
    }

  to[0] = 0;
  to[1] = 0;
  to += 2;

  /* Copy updated FDT to its launch location */
  VasEBoot_memcpy (target_atag, tmp_atag, sizeof (VasEBoot_uint32_t) * (to - tmp_atag));
  VasEBoot_free (tmp_atag);

  VasEBoot_dprintf ("loader", "ATAG updated for Linux boot\n");

  return VAS_EBOOT_ERR_NONE;
}

/*
 * linux_prepare_fdt():
 *   Prepares a loaded FDT for being passed to Linux.
 *   Merges in command line parameters and sets up initrd addresses.
 */
static VasEBoot_err_t
linux_prepare_fdt (void *target_fdt)
{
  int node;
  int retval;
  int tmp_size;
  void *tmp_fdt;

  tmp_size = VasEBoot_fdt_get_totalsize (current_fdt) + 0x100 + VasEBoot_strlen (linux_args);
  tmp_fdt = VasEBoot_malloc (tmp_size);
  if (!tmp_fdt)
    return VasEBoot_errno;

  VasEBoot_memcpy (tmp_fdt, current_fdt, VasEBoot_fdt_get_totalsize (current_fdt));
  VasEBoot_fdt_set_totalsize (tmp_fdt, tmp_size);

  /* Find or create '/chosen' node */
  node = VasEBoot_fdt_find_subnode (tmp_fdt, 0, "chosen");
  if (node < 0)
    {
      VasEBoot_dprintf ("linux", "No 'chosen' node in FDT - creating.\n");
      node = VasEBoot_fdt_add_subnode (tmp_fdt, 0, "chosen");
      if (node < 0)
	goto failure;
    }

  VasEBoot_dprintf ("linux", "linux_args: '%s'\n", linux_args);

  /* Generate and set command line */
  retval = VasEBoot_fdt_set_prop (tmp_fdt, node, "bootargs", linux_args,
			      VasEBoot_strlen (linux_args) + 1);
  if (retval)
    goto failure;

  if (initrd_start && initrd_end)
    {
      /*
       * We're using physical addresses, so even if we have LPAE, we're
       * restricted to a 32-bit address space.
       */
      VasEBoot_dprintf ("loader", "Initrd @ 0x%08x-0x%08x\n",
		    initrd_start, initrd_end);

      retval = VasEBoot_fdt_set_prop32 (tmp_fdt, node, "linux,initrd-start",
				    initrd_start);
      if (retval)
	goto failure;
      retval = VasEBoot_fdt_set_prop32 (tmp_fdt, node, "linux,initrd-end",
				    initrd_end);
      if (retval)
	goto failure;
    }

  /* Copy updated FDT to its launch location */
  VasEBoot_memcpy (target_fdt, tmp_fdt, tmp_size);
  VasEBoot_free (tmp_fdt);

  VasEBoot_dprintf ("loader", "FDT updated for Linux boot\n");

  return VAS_EBOOT_ERR_NONE;

failure:
  VasEBoot_free (tmp_fdt);
  return VasEBoot_error (VAS_EBOOT_ERR_BAD_ARGUMENT, "unable to prepare FDT");
}

static VasEBoot_err_t
linux_boot (void)
{
  kernel_entry_t linuxmain;
  int fdt_valid, atag_valid;
  void *target_fdt = 0;

  fdt_valid = (current_fdt && VasEBoot_fdt_check_header_nosize (current_fdt) == 0);
  atag_valid = ((((const VasEBoot_uint16_t *) current_fdt)[3] & ~3) == 0x5440
		&& *((const VasEBoot_uint32_t *) current_fdt));
  VasEBoot_dprintf ("loader", "atag: %p, %x, %x, %s, %s\n",
		current_fdt,
		((const VasEBoot_uint16_t *) current_fdt)[3],
		*((const VasEBoot_uint32_t *) current_fdt),
		(const char *) current_fdt,
		(const char *) current_fdt + 1);

  if (!fdt_valid && machine_type == VAS_EBOOT_ARM_MACHINE_TYPE_FDT)
    return VasEBoot_error (VAS_EBOOT_ERR_FILE_NOT_FOUND,
		       N_("device tree must be supplied (see `devicetree' command)"));

  VasEBoot_arch_sync_caches ((void *) linux_addr, linux_size);

  VasEBoot_dprintf ("loader", "Kernel at: 0x%x\n", linux_addr);

  if (fdt_valid || atag_valid)
    {
#ifdef VAS_EBOOT_MACHINE_EFI
      VasEBoot_size_t size;
      if (fdt_valid)
	size = VasEBoot_fdt_get_totalsize (current_fdt);
      else
	size = 4 * get_atag_size (current_fdt);
      size += VasEBoot_strlen (linux_args) + 256;
      target_fdt = VasEBoot_efi_allocate_loader_memory (LINUX_FDT_PHYS_OFFSET, size);
      if (!target_fdt)
	return VasEBoot_error (VAS_EBOOT_ERR_OUT_OF_MEMORY, N_("out of memory"));
#else
      target_fdt = (void *) LINUX_FDT_ADDRESS;
#endif
    }

  if (fdt_valid)
    {
      VasEBoot_err_t err;

      err = linux_prepare_fdt (target_fdt);
      if (err)
	return err;
      VasEBoot_dprintf ("loader", "FDT @ %p\n", target_fdt);
    }
  else if (atag_valid)
    {
      VasEBoot_err_t err;

      err = linux_prepare_atag (target_fdt);
      if (err)
	return err;
      VasEBoot_dprintf ("loader", "ATAG @ %p\n", target_fdt);
    }

  VasEBoot_dprintf ("loader", "Jumping to Linux...\n");

  /* Boot the kernel.
   *   Arguments to kernel:
   *     r0 - 0
   *     r1 - machine type
   *     r2 - address of DTB
   */
  linuxmain = (kernel_entry_t) linux_addr;

  VasEBoot_arm_disable_caches_mmu ();

  linuxmain (0, machine_type, target_fdt);

  return VasEBoot_error (VAS_EBOOT_ERR_BAD_OS, "Linux call returned");
}

/*
 * Only support zImage, so no relocations necessary
 */
static VasEBoot_err_t
linux_load (const char *filename, VasEBoot_file_t file)
{
  struct linux_arch_kernel_header *lh;
  int size;

  size = VasEBoot_file_size (file);

  linux_addr = LINUX_ADDRESS;
  VasEBoot_dprintf ("loader", "Loading Linux to 0x%08x\n",
		(VasEBoot_addr_t) linux_addr);

  if (VasEBoot_file_read (file, (void *) linux_addr, size) != size)
    {
      if (!VasEBoot_errno)
	VasEBoot_error (VAS_EBOOT_ERR_BAD_OS, N_("premature end of file %s"),
		    filename);
      return VasEBoot_errno;
    }

  lh = (void *) linux_addr;

  if ((VasEBoot_size_t) size > sizeof (*lh) &&
      lh->magic == VAS_EBOOT_LINUX_ARM_MAGIC_SIGNATURE)
    ;
  else if (size > 0x8000 && *(VasEBoot_uint32_t *) (linux_addr) == 0xea000006
	   && machine_type == VAS_EBOOT_ARM_MACHINE_TYPE_RASPBERRY_PI)
    VasEBoot_memmove ((void *) linux_addr, (void *) (linux_addr + 0x8000),
		  size - 0x8000);
  else
    return VasEBoot_error (VAS_EBOOT_ERR_BAD_FILE_TYPE, N_("invalid zImage"));

  linux_size = size;

  return VAS_EBOOT_ERR_NONE;
}

static VasEBoot_err_t
linux_unload (void)
{
  VasEBoot_dl_unref (my_mod);

  VasEBoot_free (linux_args);
  linux_args = NULL;

  initrd_start = initrd_end = 0;

  return VAS_EBOOT_ERR_NONE;
}

static VasEBoot_err_t
VasEBoot_cmd_linux (VasEBoot_command_t cmd __attribute__ ((unused)),
		int argc, char *argv[])
{
  int size;
  VasEBoot_err_t err;
  VasEBoot_file_t file;
  VasEBoot_dl_ref (my_mod);

  if (argc == 0)
    return VasEBoot_error (VAS_EBOOT_ERR_BAD_ARGUMENT, N_("filename expected"));

  file = VasEBoot_file_open (argv[0], VAS_EBOOT_FILE_TYPE_LINUX_KERNEL);
  if (!file)
    goto fail;

  err = linux_load (argv[0], file);
  VasEBoot_file_close (file);
  if (err)
    goto fail;

  VasEBoot_loader_set (linux_boot, linux_unload, 0);

  size = VasEBoot_loader_cmdline_size (argc, argv);
  linux_args = VasEBoot_malloc (size + sizeof (LINUX_IMAGE));
  if (!linux_args)
    {
      VasEBoot_loader_unset();
      goto fail;
    }

  /* Create kernel command line.  */
  VasEBoot_memcpy (linux_args, LINUX_IMAGE, sizeof (LINUX_IMAGE));
  err = VasEBoot_create_loader_cmdline (argc, argv,
				    linux_args + sizeof (LINUX_IMAGE) - 1, size,
				    VAS_EBOOT_VERIFY_KERNEL_CMDLINE);
  if (err)
    goto fail;

  return VAS_EBOOT_ERR_NONE;

fail:
  VasEBoot_dl_unref (my_mod);
  return VasEBoot_errno;
}

static VasEBoot_err_t
VasEBoot_cmd_initrd (VasEBoot_command_t cmd __attribute__ ((unused)),
		 int argc, char *argv[])
{
  VasEBoot_file_t file;
  VasEBoot_size_t size = 0;
  struct VasEBoot_linux_initrd_context initrd_ctx = { 0, 0, 0 };

  if (argc == 0)
    return VasEBoot_error (VAS_EBOOT_ERR_BAD_ARGUMENT, N_("filename expected"));

  file = VasEBoot_file_open (argv[0], VAS_EBOOT_FILE_TYPE_LINUX_INITRD);
  if (!file)
    return VasEBoot_errno;

  if (VasEBoot_initrd_init (argc, argv, &initrd_ctx))
    goto fail;

  size = VasEBoot_get_initrd_size (&initrd_ctx);

  initrd_start = LINUX_INITRD_ADDRESS;

  VasEBoot_dprintf ("loader", "Loading initrd to 0x%08x\n",
		(VasEBoot_addr_t) initrd_start);

  if (VasEBoot_initrd_load (&initrd_ctx, (void *) initrd_start))
    goto fail;

  initrd_end = initrd_start + size;

  return VAS_EBOOT_ERR_NONE;

fail:
  VasEBoot_file_close (file);

  return VasEBoot_errno;
}

static VasEBoot_err_t
load_dtb (VasEBoot_file_t dtb, int size)
{
  void *new_fdt = VasEBoot_zalloc (size);
  if (!new_fdt)
    return VasEBoot_errno;
  VasEBoot_dprintf ("loader", "Loading device tree to %p\n",
		new_fdt);
  if ((VasEBoot_file_read (dtb, new_fdt, size) != size)
      || (VasEBoot_fdt_check_header (new_fdt, size) != 0))
    {
      VasEBoot_free (new_fdt);
      return VasEBoot_error (VAS_EBOOT_ERR_BAD_OS, N_("invalid device tree"));
    }

  VasEBoot_fdt_set_totalsize (new_fdt, size);
  current_fdt = new_fdt;
  /*
   * We've successfully loaded an FDT, so any machine type passed
   * from firmware is now obsolete.
   */
  machine_type = VAS_EBOOT_ARM_MACHINE_TYPE_FDT;

  return VAS_EBOOT_ERR_NONE;
}

static VasEBoot_err_t
VasEBoot_cmd_devicetree (VasEBoot_command_t cmd __attribute__ ((unused)),
		     int argc, char *argv[])
{
  VasEBoot_file_t dtb;
  int size;

  if (argc != 1)
    return VasEBoot_error (VAS_EBOOT_ERR_BAD_ARGUMENT, N_("filename expected"));

  dtb = VasEBoot_file_open (argv[0], VAS_EBOOT_FILE_TYPE_DEVICE_TREE_IMAGE);
  if (!dtb)
    return VasEBoot_errno;

  size = VasEBoot_file_size (dtb);
  if (size == 0)
    VasEBoot_error (VAS_EBOOT_ERR_BAD_OS, "empty file");
  else
    load_dtb (dtb, size);
  VasEBoot_file_close (dtb);

  return VasEBoot_errno;
}

static VasEBoot_command_t cmd_linux, cmd_initrd, cmd_devicetree;

VAS_EBOOT_MOD_INIT (linux)
{
  cmd_linux = VasEBoot_register_command ("linux", VasEBoot_cmd_linux,
				     0, N_("Load Linux."));
  cmd_initrd = VasEBoot_register_command ("initrd", VasEBoot_cmd_initrd,
				      0, N_("Load initrd."));
  cmd_devicetree = VasEBoot_register_command_lockdown ("devicetree", VasEBoot_cmd_devicetree,
						   /* TRANSLATORS: DTB stands for device tree blob. */
						   0, N_("Load DTB file."));
  my_mod = mod;
  current_fdt = (const void *) VasEBoot_arm_firmware_get_boot_data ();
  machine_type = VasEBoot_arm_firmware_get_machine_type ();
}

VAS_EBOOT_MOD_FINI (linux)
{
  VasEBoot_unregister_command (cmd_linux);
  VasEBoot_unregister_command (cmd_initrd);
  VasEBoot_unregister_command (cmd_devicetree);
}