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

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

#include "lowmem.h"
#include "bank.h"
#include "memory.h"
#include "uncompress.h"
#ifdef ARCH_M68K
#include "bootinfo.h"
#endif
#ifdef ARCH_PPC
#include "bootx.h"
#endif
#include "arch.h"
#include "misc.h"
#include "glue.h"
#include "load.h"
#include "console.h"
#include "vga.h"

#ifdef ARCH_M68K

typedef void (*entry_t) (unsigned long , unsigned long , unsigned long );
typedef void (*disable_cache_t) (void);

extern void enter_kernelnoMMU(unsigned long addr, unsigned long size, unsigned long dest);
extern char end_enter_kernelnoMMU;
extern void noMMU_disable_cache(void);
extern void enter_kernel030(unsigned long addr, unsigned long size, unsigned long dest);
extern char end_enter_kernel030;
extern void MMU030_disable_cache(void);
extern void enter_kernel040(unsigned long addr, unsigned long size, unsigned long dest);
extern char end_enter_kernel040;
extern void MMU040_disable_cache(void);

#define PAGE_SHIFT	12
#define PAGE_SIZE	(1UL << PAGE_SHIFT)
#define PAGE_MASK	(~(PAGE_SIZE-1))

#define KERNEL_ALIGN	(256L * 1024L)	// Kernel alignment, on 256K boundary
#define BI_ALLOC_SIZE	(4096L)		// Allocate 4K for bootinfo

#endif
#ifdef ARCH_PPC
#include "enter_kernelPPC.h"
#endif

int start(emile_l2_header_t* info)
{
	char * kernel;
#ifdef ARCH_M68K
	unsigned long physical;
	entry_t entry;
	unsigned long physImage;
	disable_cache_t disable_cache;
	unsigned long aligned_size;
	unsigned long aligned_addr;
	unsigned long enter_kernel;
	unsigned long end_enter_kernel;
	unsigned long start_mem;
#endif
#ifdef ARCH_PPC
	PPCRegisterList regs;
#endif
	int ret;
	unsigned long kernel_image_start;
	unsigned long ramdisk_start;
	int uncompressed_size;
	int bootstrap_size;

	printf("Early Macintosh Image LoadEr"
#if defined(ARCH_M68K) && defined(ARCH_PPC)
		" (mixed mode)\n");
#elif defined(ARCH_M68K)
		" for Motorola 680x0\n");
#elif defined(ARCH_PPC)
		" for 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");

	if (!EMILE_COMPAT(EMILE_03_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 ? */
	
#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
	{
		if (mmu_type == gestalt68040MMU)
		{
			printf("Using 68040 MMU\n");
			enter_kernel = (unsigned long)enter_kernel040;
			end_enter_kernel = (unsigned long)&end_enter_kernel040;
			disable_cache = MMU040_disable_cache;
		}
		else if (mmu_type == gestalt68030MMU)
		{
			printf("Using 68030 MMU\n");
			enter_kernel = (unsigned long)enter_kernel030;
			end_enter_kernel = (unsigned long)&end_enter_kernel030;
			disable_cache = MMU030_disable_cache;
		}
		else if (mmu_type == gestalt68851)
		{
			error("MMU 68851 is not supported");
		}
		else if (mmu_type == gestaltNoMMU)
		{
			printf("No MMU detected\n");
			enter_kernel = (unsigned long)enter_kernelnoMMU;
			end_enter_kernel = (unsigned long)&end_enter_kernelnoMMU;
			disable_cache = noMMU_disable_cache;
		}
		else
			error("Unknown MMU");

		/* and BI_ALLOC_SIZE for bootinfo */

		bootstrap_size = BI_ALLOC_SIZE + 
				 end_enter_kernel - enter_kernel;
	}
	else
#ifndef ARCH_PPC
		error("EMILE doesn't support your architecture");
#endif
#endif
#ifdef ARCH_PPC
	if (arch_type == gestaltPowerPC)
	{
		enter_kernel = NULL;
		end_enter_kernel = NULL;
		disable_cache = NULL;
		bootstrap_size = 0;
	}
	else
		error("EMILE doesn't support your architecture");
#endif

	/* load kernel */

	printf("vmlinux %s\n", info->command_line);
#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 */

	if (info->kernel_size == 0)	/* means uncompressed */
		kernel_image_start = (unsigned long)malloc_contiguous(
					info->kernel_image_size + 4 +
					bootstrap_size
					);
	else
		kernel_image_start = (unsigned long)malloc(
						info->kernel_image_size + 4);

	kernel_image_start = (kernel_image_start + 3) & 0xFFFFFFFC;
	printf("Kernel image base at 0x%lx\n", kernel_image_start);
	if (kernel_image_start == 0)
		error("Cannot allocate memory\n");

	/* load kernel */

	printf("Loading kernel...\n");
	ret = load_image((unsigned long)info->kernel_image_offset, 
			 info->kernel_image_size, (char*)kernel_image_start);
	if (ret == -1)
		error("Cannot load kernel image\n");

	/* uncompress kernel if needed */

	if (info->kernel_image_size == 0)
		error("Kernel is missing !!!!\n");
	else if (info->kernel_size == 0)
	{
		/* kernel is not compressed, execute in place */

		kernel = (char*)kernel_image_start;
		uncompressed_size = info->kernel_image_size;
	}
	else
	{
		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);
		uncompressed_size = uncompress(kernel, (char*)kernel_image_start);
		printf("\n");

		/* free kernel image */

		free((void*)kernel_image_start);
	}

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

#ifdef ARCH_M68K
	if (arch_type == gestalt68k)
	{
		/* copy enter_kernel at end of kernel */

		memcpy((char*)kernel + uncompressed_size + BI_ALLOC_SIZE,
	       		(char*)enter_kernel, end_enter_kernel - enter_kernel);

		end_enter_kernel = (unsigned long)kernel + uncompressed_size 
			   + BI_ALLOC_SIZE + (end_enter_kernel - enter_kernel);
		enter_kernel = (unsigned long)kernel + BI_ALLOC_SIZE 
		       					+ uncompressed_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_contiguous(
							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)
	{
		ret = logical2physical((unsigned long)kernel, &physImage);

		/* disable and flush cache */

		disable_cache();

		/* initialize bootinfo structure */

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

		/* compute final address of kernel */

		/* 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 + uncompressed_size);

		start_mem = boot_info.memory[0].addr + PAGE_SIZE;

		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);
		}
	}
	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

	/* kick off */

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

	error("Kernel startup failed");

	return 0;
}