/*
 *  VAS_EBOOT  --  GRand Unified Bootloader
 *  Copyright (C) 2009,2010  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/mm.h>
#include <VasEBoot/misc.h>

#include <VasEBoot/types.h>
#include <VasEBoot/types.h>
#include <VasEBoot/err.h>
#include <VasEBoot/cache.h>

#include <VasEBoot/powerpc/relocator.h>
#include <VasEBoot/relocator_private.h>

extern VasEBoot_uint8_t VasEBoot_relocator_forward_start;
extern VasEBoot_uint8_t VasEBoot_relocator_forward_end;
extern VasEBoot_uint8_t VasEBoot_relocator_backward_start;
extern VasEBoot_uint8_t VasEBoot_relocator_backward_end;

#define REGW_SIZEOF (2 * sizeof (VasEBoot_uint32_t))
#define JUMP_SIZEOF (sizeof (VasEBoot_uint32_t))

#define RELOCATOR_SRC_SIZEOF(x) (&VasEBoot_relocator_##x##_end \
				 - &VasEBoot_relocator_##x##_start)
#define RELOCATOR_SIZEOF(x)	(RELOCATOR_SRC_SIZEOF(x) \
				 + REGW_SIZEOF * 3)
VasEBoot_size_t VasEBoot_relocator_align = sizeof (VasEBoot_uint32_t);
VasEBoot_size_t VasEBoot_relocator_forward_size;
VasEBoot_size_t VasEBoot_relocator_backward_size;
VasEBoot_size_t VasEBoot_relocator_jumper_size = JUMP_SIZEOF + REGW_SIZEOF;
VasEBoot_size_t VasEBoot_relocator_preamble_size = 0;

void
VasEBoot_cpu_relocator_init (void)
{
  VasEBoot_relocator_forward_size = RELOCATOR_SIZEOF(forward);
  VasEBoot_relocator_backward_size = RELOCATOR_SIZEOF(backward);
}

void
VasEBoot_cpu_relocator_preamble (void *rels __attribute__((unused)))
{
}

static void
write_reg (int regn, VasEBoot_uint32_t val, void **target)
{
  /* lis r, val >> 16  */
  *(VasEBoot_uint32_t *) *target =
    ((0x3c00 | (regn << 5)) << 16) | (val >> 16);
  *target = ((VasEBoot_uint32_t *) *target) + 1;
  /* ori r, r, val & 0xffff. */
  *(VasEBoot_uint32_t *) *target = (((0x6000 | regn << 5 | regn) << 16)
				| (val & 0xffff));
  *target = ((VasEBoot_uint32_t *) *target) + 1;
}

static void
write_jump (void **target)
{
  /* blr.  */
  *(VasEBoot_uint32_t *) *target = 0x4e800020;
  *target = ((VasEBoot_uint32_t *) *target) + 1;
}

void
VasEBoot_cpu_relocator_jumper (void *rels, VasEBoot_addr_t addr)
{
  write_reg (VAS_EBOOT_PPC_JUMP_REGISTER, addr, &rels);
  write_jump (&rels);
}

void
VasEBoot_cpu_relocator_backward (void *ptr0, void *src, void *dest,
			     VasEBoot_size_t size)
{
  void *ptr = ptr0;
  write_reg (8, (VasEBoot_uint32_t) src, &ptr);
  write_reg (9, (VasEBoot_uint32_t) dest, &ptr);
  write_reg (10, (VasEBoot_uint32_t) size, &ptr);
  VasEBoot_memcpy (ptr, &VasEBoot_relocator_backward_start,
	       RELOCATOR_SRC_SIZEOF (backward));
}

void
VasEBoot_cpu_relocator_forward (void *ptr0, void *src, void *dest,
			     VasEBoot_size_t size)
{
  void *ptr = ptr0;
  write_reg (8, (VasEBoot_uint32_t) src, &ptr);
  write_reg (9, (VasEBoot_uint32_t) dest, &ptr);
  write_reg (10, (VasEBoot_uint32_t) size, &ptr);
  VasEBoot_memcpy (ptr, &VasEBoot_relocator_forward_start,
	       RELOCATOR_SRC_SIZEOF (forward));
}

VasEBoot_err_t
VasEBoot_relocator32_boot (struct VasEBoot_relocator *rel,
		       struct VasEBoot_relocator32_state state)
{
  void *ptr;
  VasEBoot_err_t err;
  void *relst;
  VasEBoot_size_t relsize;
  VasEBoot_size_t stateset_size = 32 * REGW_SIZEOF + JUMP_SIZEOF;
  unsigned i;
  VasEBoot_relocator_chunk_t ch;

  err = VasEBoot_relocator_alloc_chunk_align (rel, &ch, 0, UP_TO_TOP32 (stateset_size),
					  stateset_size, sizeof (VasEBoot_uint32_t),
					  VAS_EBOOT_RELOCATOR_PREFERENCE_NONE, 0);
  if (err)
    return err;

  ptr = get_virtual_current_address (ch);
  for (i = 0; i < 32; i++)
    write_reg (i, state.gpr[i], &ptr);
  write_jump (&ptr);

  err = VasEBoot_relocator_prepare_relocs (rel, get_physical_target_address (ch),
				       &relst, &relsize);
  if (err)
    return err;

  VasEBoot_arch_sync_caches ((void *) relst, relsize);

  ((void (*) (void)) relst) ();

  /* Not reached.  */
  return VAS_EBOOT_ERR_NONE;
}