/*
 *
 * (c) 2004-2005 Laurent Vivier <LaurentVivier@wanadoo.fr>
 *
 */

#define __NO_INLINE__

#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include <unistd.h>

#include <macos/types.h>
#include <macos/gestalt.h>

#include "bank.h"
#include "memory.h"
#if defined(ARCH_M68K)
#if defined(__LINUX__)
#include "bootinfo.h"
#elif defined(__NETBSD__)
#include "bootenv.h"
#endif
#endif
#ifdef ARCH_PPC
#include "bootx.h"
#endif
#include "arch.h"
#include "misc.h"
#include "load.h"
#include "console.h"
#include "vga.h"
#include "driver.h"
#if defined(USE_CLI) && defined(__LINUX__)
#include "cli.h"
#endif

#include "enter_kernel.h"

#ifdef ARCH_M68K
#define KERNEL_ALIGN	(256L * 1024L)	// Kernel alignment, on 256K boundary
#if defined(__LINUX__)
#define BI_ALLOC_SIZE	(4096L)		// Allocate 4K for bootinfo
#elif defined(__NETBSD__)
#define BI_ALLOC_SIZE	(4096L)
#endif
#endif

static void banner(void)
{
	printf("Early Macintosh Image LoadEr"
#if defined(__LINUX__) && defined(__NETBSD__)
#error Cannot support LINUX AND NETBSD
#elif defined(__LINUX__)
		" for Linux"
#elif defined(__NETBSD__)
		" for NetBSD"
#endif
#if defined(ARCH_M68K) && defined(ARCH_PPC)
		" (mixed mode)\n");
#elif defined(ARCH_M68K)
#ifdef USE_MMU
		" on Motorola 680x0\n");
#else
		" on Motorola 68000\n");
#endif /* USE_MMU */
#elif defined(ARCH_PPC)
		" on PowerPC\n");
#else
		" (unknown processor)\n");
#endif
	printf("EMILE v"VERSION" (c) 2004,2005 Laurent Vivier\n");
	printf("This is free software, redistribute it under GPL\n");
}

int start(emile_l2_header_t* info)
{
	char * kernel;
#ifdef ARCH_M68K
	entry_t entry;
	unsigned long physImage;
#ifdef USE_MMU
	unsigned long physical;
#ifdef __LINUX__
	unsigned long aligned_size;
	unsigned long aligned_addr;
#endif /* __LINUX__ */
#endif /* USE_MMU */
	unsigned long start_mem;
	unsigned long entry_point;
#endif /* ARCH_M68K */
#ifdef ARCH_PPC
	PPCRegisterList regs;
#endif
	int ret;
	unsigned long ramdisk_start;
	int bootstrap_size;

	banner();

	if (!EMILE_COMPAT(EMILE_05_SIGNATURE, info->signature))
		error("Bad header signature !\n");

	arch_init();

	init_memory_map();

#ifdef BANK_DUMP
	bank_dump();
#endif

	printf("Available Memory: %ld kB\n", bank_mem_avail() / 1024);

	if (info->gestaltID != 0) {
		machine_id = info->gestaltID;
		printf("User forces gestalt ID to %ld\n", machine_id);
	}

	/* where is mapped my boot function ? */
	
	enter_kernel_init();

#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
	{
#if defined(__LINUX__)
		/* and BI_ALLOC_SIZE for bootinfo */

		bootstrap_size = BI_ALLOC_SIZE + 
				 end_enter_kernel - enter_kernel;
#elif defined(__NETBSD__)
		bootstrap_size = BI_ALLOC_SIZE + 
				 end_enter_kernel - enter_kernel;
#endif
	}
#ifndef ARCH_PPC
	else
		error("EMILE doesn't support your architecture");
#endif
#endif

	/* load kernel */

	if (info->kernel_image_size == 0)
		error("Kernel is missing !!!!\n");

#if defined(USE_CLI) && defined(__LINUX__)
	printf("Parameters: ");
	console_cursor_save();
	printf("%s", info->command_line);
	console_cursor_on();
	if (console_keypressed(5 * 60))
	{
		console_cursor_restore();
		cli_edit(info->command_line, COMMAND_LINE_LENGTH);
	}
	console_cursor_off();
	putchar('\n');
#else
#ifdef __LINUX__
	printf("%s\n", info->command_line);
#endif
#endif
#ifdef SCSI_SUPPORT
	info->kernel_image_offset = (unsigned long)info->kernel_image_offset + (unsigned long)info;
#endif
	printf("Kernel image size is %d Bytes\n", info->kernel_image_size);

	/* allocate memory for kernel */

	printf("Allocating %d bytes for kernel\n", info->kernel_size);
	kernel = (char*)malloc_contiguous(info->kernel_size + 4 +
					  bootstrap_size
					);
	if (kernel == 0)
	{
		printf("cannot allocate %d bytes\n", info->kernel_size);
		while(1);
	}

	/* align kernel address to a 4 byte word */

	kernel = (unsigned char*)(((unsigned long)kernel + 3) & 0xFFFFFFFC);

	printf("Kernel image base at %p\n", kernel);

	if (!check_full_in_bank((unsigned long)kernel, info->kernel_size))
		error("Kernel between two banks, contact maintainer\n");

	/* load kernel */

	printf("Loading kernel...\n");
	if (info->kernel_size == info->kernel_image_size)	/* means uncompressed */
	{
		/* load kernel */

		ret = load_image((unsigned long)info->kernel_image_offset, 
			 	info->kernel_size, kernel);
		if (ret == -1)
			error("Cannot load kernel image\n");
	}
	else
	{
		/* load and uncompress kernel */

		ret = load_gzip((unsigned long)info->kernel_image_offset, 
			 	info->kernel_image_size, kernel);
		if (ret == -1)
			error("Cannot load and uncompress kernel image\n");
	}

#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
	{
		unsigned long enter_size = end_enter_kernel - enter_kernel;

		/* copy enter_kernel at end of kernel */

		memcpy((char*)kernel +
			info->kernel_size + bootstrap_size - enter_size,
	       		(char*)enter_kernel, enter_size);

		end_enter_kernel = (unsigned long)kernel + 
				   info->kernel_size + bootstrap_size;
		enter_kernel = (unsigned long)kernel + 
			       bootstrap_size - enter_size + info->kernel_size;
	}
#endif

	/* load ramdisk if needed */

	if (info->ramdisk_size != 0)
	{
		printf("RAMDISK size is %d Bytes\n", info->ramdisk_size);
		ramdisk_start = (unsigned long)malloc_top(
							info->ramdisk_size + 4);
		ramdisk_start = (ramdisk_start + 3) & 0xFFFFFFFC;
		printf("RAMDISK base at 0x%lx\n", ramdisk_start);

		printf("Loading RAMDISK...\n");
		ret = load_image((unsigned long)info->ramdisk_offset, 
				 info->ramdisk_size, (char*)ramdisk_start);
		if (ret == -1)
			error("Cannot load ramdisk\n");

		if (!check_full_in_bank(ramdisk_start, info->ramdisk_size))
			error("ramdisk between two banks, contact maintainer\n");
	}
	else
	{
		ramdisk_start = 0;
		printf("no RAMDISK\n");
	}

#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
	{
		/* compute final address of kernel */

		if  (mmu_type == gestaltNoMMU)
		{
			unsigned long size = end_enter_kernel - enter_kernel;

#if defined(__LINUX__)
			/* initialize bootinfo structure */

			bootinfo_init(info->command_line, 
		      		(char*)ramdisk_start, info->ramdisk_size);

			/* set bootinfo at end of kernel image */

			set_kernel_bootinfo(kernel + info->kernel_size);

			physImage = (unsigned long)kernel;
			start_mem = KERNEL_BASEADDR + 0x1000;
			entry_point = start_mem;
#elif defined(__NETBSD__)
			start_mem = 0;
			entry_point = 0x2e00;
#endif
			entry = (entry_t)(start_mem - size);

			printf("\n");
			printf("Physical address of kernel will be 0x%08lx\n", 
				start_mem);
			printf("Ok, booting the kernel.\n");

			memcpy(entry, (char*)enter_kernel, size);
		} else
#ifndef USE_MMU
			error("Unsupported MMU");
#else
		{
			ret = logical2physical((unsigned long)kernel, &physImage);

			/* disable and flush cache */

			disable_cache();

#if defined(__LINUX__)
			/* initialize bootinfo structure */

			bootinfo_init(info->command_line, 
		      		(char*)ramdisk_start, info->ramdisk_size);

			/* add KERNEL_ALIGN if we have to align */
		 
			aligned_size = boot_info.memory[0].addr & (KERNEL_ALIGN - 1);
			if ( aligned_size > 0 )
			{
				aligned_size = KERNEL_ALIGN - aligned_size;
				aligned_addr = boot_info.memory[0].addr + aligned_size;
				aligned_size = boot_info.memory[0].size - aligned_size;
				boot_info.memory[0].addr = aligned_addr;
				boot_info.memory[0].size = aligned_size;
			}

			/* set bootinfo at end of kernel image */

			set_kernel_bootinfo(kernel + info->kernel_size);

			start_mem = boot_info.memory[0].addr + PAGE_SIZE;
			entry_point = start_mem;
#elif defined(__NETBSD__)
			bootenv_init(kernel + info->kernel_size);
			start_mem = 0;
			entry_point = 0x2e00;
#endif

			printf("\n");
			printf("Physical address of kernel will be 0x%08lx\n", 
				start_mem);
			printf("Ok, booting the kernel.\n");

			ret = logical2physical(enter_kernel, &physical);
			entry = (entry_t)physical;

			if ( (ret == 0) && (enter_kernel != (unsigned long)entry) )
			{
				unsigned long logi;
				unsigned long size = end_enter_kernel - enter_kernel;

				logi = vga_get_video();
				ret = logical2physical(logi, &physical);
				entry = (entry_t)physical;
	
				memcpy((char*)logi, (char*)enter_kernel, size);
				memcpy((char*)entry, (char*)enter_kernel, size);
			}
		}
#endif /* USE_MMU */
	}
	else
#ifndef ARCH_PPC
		error("EMILE doesn't support your architecture");
#endif
#endif /* ARCH_M68K */
#ifdef ARCH_PPC
	if (arch_type == gestaltPowerPC)
	{
		bootx_init(info->command_line, 
				(char*)ramdisk_start, info->ramdisk_size);

		regs.PC      = (u_int32_t)kernel;
#define BOOT_KERNEL_STACK_SIZE 65536
		regs.GPR[1]  = (u_int32_t)malloc_contiguous(BOOT_KERNEL_STACK_SIZE)
				+ BOOT_KERNEL_STACK_SIZE - 512;
		regs.GPR[2]  = 0;
		regs.GPR[3]  = 'BooX';
		regs.GPR[4]  = (u_int32_t)&bootx_infos;

		/* Set up the info for BAT mapping on Nubus */

		regs.GPR[5]  = vga_get_videobase() & 0xFF800000UL;
		regs.GPR[11]  = 1;

		printf("\n");
		printf("Physical address of kernel will be 0x%08lx\n", 
			(unsigned long)kernel);
		printf("Ok, booting the kernel.\n");
	}
	else
		error("EMILE doesn't support your architecture");
#endif

	turn_off_interrupts();

	 asm("ori.w #0x0700,%sr");

	/* kick off */

#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
		entry(physImage, info->kernel_size + BI_ALLOC_SIZE, start_mem, entry_point);
#endif
#ifdef ARCH_PPC
	if (arch_type == gestaltPowerPC)
		enter_kernelPPC((unsigned long)kernel, &regs);
#endif

	error("Kernel startup failed");

	return 0;
}