/*
 * This file is part of the Palacios Virtual Machine Monitor developed
 * by the V3VEE Project with funding from the United States National 
 * Science Foundation and the Department of Energy.  
 *
 * The V3VEE Project is a joint project between Northwestern University
 * and the University of New Mexico.  You can find out more at 
 * http://www.v3vee.org
 *
 * Copyright (c) 2015, The V3VEE Project <http://www.v3vee.org> 
 * All rights reserved.
 *
 * Author: Peter Dinda <pdinda@northwestern.edu>
 *
 * This is free software.  You are permitted to use,
 * redistribute, and modify it as specified in the file "V3VEE_LICENSE".
 */

#include <palacios/vmm_bios.h>
#include <palacios/vm_guest_mem.h>
#include <palacios/vmm_xml.h>



int v3_setup_bioses(struct v3_vm_info * vm, v3_cfg_tree_t * cfg) 
{

    v3_cfg_tree_t * bios_tree = NULL;
    v3_cfg_tree_t * bios_list = NULL;
    
    bios_tree = v3_cfg_subtree(cfg, "bioses");
    
    
    if (!v3_cfg_val(bios_tree, "disable_vgabios")) {
	
#if defined(V3_CONFIG_SEABIOS) || defined(V3_CONFIG_BOCHSBIOS)

#define VGABIOS_START 0x000c0000
        /* layout vgabios */
        {
            extern uint8_t v3_vgabios_start[];
            extern uint8_t v3_vgabios_end[];

            V3_Print(vm,VCORE_NONE,"Mapping VGA BIOS of %llu bytes at gpa %p\n", (uint64_t)(v3_vgabios_end-v3_vgabios_start), (void*)VGABIOS_START);

            if (v3_write_gpa_memory(&(vm->cores[0]), VGABIOS_START, v3_vgabios_end - v3_vgabios_start, v3_vgabios_start) != (v3_vgabios_end - v3_vgabios_start)) { 
		PrintError(vm, VCORE_NONE, "Could not write VGA BIOS\n");
		return -1;
	    }
        }

#endif

    }


    if (!v3_cfg_val(bios_tree, "disable_rombios")) {

        /* layout rombios */
        {
            extern uint8_t v3_rombios_start[];
            extern uint8_t v3_rombios_end[];

            V3_Print(vm,VCORE_NONE,"Mapping BIOS of %llu bytes at gpa %p\n", (uint64_t)(v3_rombios_end-v3_rombios_start), (void*)V3_CONFIG_BIOS_START);

            if (v3_write_gpa_memory(&(vm->cores[0]), V3_CONFIG_BIOS_START, v3_rombios_end - v3_rombios_start, v3_rombios_start) != (v3_rombios_end - v3_rombios_start)) { 
		PrintError(vm, VCORE_NONE, "Could not write ROM BIOS\n");
		return -1;
	    }

#ifdef V3_CONFIG_SEABIOS
	    
#define SEABIOS_HIGH_START 0xfffe0000ULL
#define SEABIOS_HIGH_END   0x100000000ULL

	    V3_Print(vm,VCORE_NONE,"Additionally mapping SEABIOS of %llu bytes at gpa %p\n", (uint64_t)(v3_rombios_end-v3_rombios_start), (void*)SEABIOS_HIGH_START);

	    if (v3_get_mem_region(vm,V3_MEM_CORE_ANY, SEABIOS_HIGH_START)) {
		// it is already mapped, we are done
		// note it's mapped to the same place as the low memory copy
		// so it's now pointing to the fresh copy
		V3_Print(vm,VCORE_NONE,"BIOS is already mapped\n");
	    } else {
		extern uint64_t v3_mem_block_size;
		void *rombios_dst;

		if (v3_gpa_to_hva(&(vm->cores[0]), V3_CONFIG_BIOS_START, (addr_t *)&rombios_dst) == -1) { 
		    PrintError(vm, VCORE_NONE, "Could not find ROMBIOS destination address\n");
		    return -1;
		}

		if (v3_add_shadow_mem(vm, V3_MEM_CORE_ANY, 
				      SEABIOS_HIGH_START, SEABIOS_HIGH_END,
				      (addr_t)V3_PAddr(rombios_dst)) == -1) {
		    PrintError(vm, VCORE_NONE, "Error mapping SEABIOS to end of memory\n");
		    return -1;
		}

		if ((V3_CONFIG_BIOS_START / v3_mem_block_size) != 
		    ((V3_CONFIG_BIOS_START+(SEABIOS_HIGH_END-SEABIOS_HIGH_START-1)) / v3_mem_block_size)) { 
		    PrintError(vm,VCORE_NONE, "ALERT: MAPPING OF SEABIOS SPANS MEMORY BLOCKS: %llx %llx %llx\n",
			       (uint64_t) V3_CONFIG_BIOS_START,(uint64_t)(V3_CONFIG_BIOS_START+(SEABIOS_HIGH_END-SEABIOS_HIGH_START-1)), v3_mem_block_size);
		}

		V3_Print(vm,VCORE_NONE,"BIOS  mapped\n");

            }
	    
#endif

        }

	{

// traditional BIOS data area (and IVT, etc).  Technically only 0x400-0x4ff is the BDA
#define BDA_START   0x0
#define BDA_END     0xfff

// Extended BDA and EBDA2.  Technically not standardized.  Usually at the
// end of the 640K chunk.  We're just using that last page to play it save
#define EBDA_START 0x9f000
#define EBDA_END   0x9ffff

	    V3_Print(vm,VCORE_NONE,"Clearing BDA %p through %p\n",(void*)BDA_START,(void*)BDA_END);

	    if (v3_set_gpa_memory(&vm->cores[0],BDA_START,BDA_END-BDA_START,0)!=BDA_END-BDA_START) {
		PrintError(vm, VCORE_NONE, "Could not zero BDA\n");
		return -1;
	    }

	    V3_Print(vm,VCORE_NONE,"Clearing EBDA %p through %p\n",(void*)EBDA_START,(void*)EBDA_END);

	    if (v3_set_gpa_memory(&vm->cores[0],EBDA_START,EBDA_END-EBDA_START,0)!=EBDA_END-EBDA_START) {
		PrintError(vm, VCORE_NONE, "Could not zero eBDA\n");
		return -1;
	    }


	}

    }

    

    bios_list = v3_cfg_subtree(bios_tree, "bios");

    while (bios_list) {
        char * id = v3_cfg_val(bios_list, "file");
        char * addr = v3_cfg_val(bios_list, "address");
        uint64_t file_ptr = 0;
        struct v3_cfg_file * file = NULL;

        if (!id) {
            PrintError(vm, VCORE_NONE, "Could not find bios file\n");
            continue;
        }

        if (!addr) {
            PrintError(vm, VCORE_NONE, "Could not find bios address\n");
            continue;
        }

        file = v3_cfg_get_file(vm, id);
        if (!file) {
            PrintError(vm, VCORE_NONE, "Invalid BIOS file: %s\n", id);
            continue;
        }

        file_ptr = atox(addr);

        V3_Print(vm, VCORE_NONE, "Copying BIOS ROM (%s) to %p (size=%lld)\n", 
                 id,
                 (void*)file_ptr,
                 file->size);

	if (v3_write_gpa_memory(&vm->cores[0],(addr_t)file_ptr,file->size,file->data)!=file->size) {
	    PrintError(vm, VCORE_NONE, "Could not copy BIOS (%s)\n",id);
	    return -1;
	}

        V3_Print(vm, VCORE_NONE, "Moving on to next BIOS file\n");
        bios_list = v3_cfg_next_branch(bios_list);
    }

    // NOW BLOW AWAY BDA and EBDA here
    // possibly also reset CMOS


    return 0;
}