/* 
 * 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) 2008, Jack Lange <jarusl@cs.northwestern.edu> 
 * Copyright (c) 2008, The V3VEE Project <http://www.v3vee.org> 
 * All rights reserved.
 *
 * Author: Jack Lange <jarusl@cs.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_debug.h>
#include <palacios/vmm.h>
#include <palacios/vmm_host_events.h>
#include <palacios/vm_guest.h>
#include <palacios/vmm_decoder.h>
#include <palacios/vm_guest_mem.h>

#define PRINT_TELEMETRY  1
#define PRINT_CORE_STATE 2
#define PRINT_ARCH_STATE 3
#define PRINT_STACK      4
#define PRINT_BACKTRACE  5


#define PRINT_ALL        100 // Absolutely everything
#define PRINT_STATE      101 // telemetry, core state, arch state




static int core_handler(struct guest_info * core, uint32_t cmd) {


    switch (cmd) {
	case PRINT_TELEMETRY: 
	    v3_print_core_telemetry(core);
	    break;
	
	case PRINT_CORE_STATE:
	    v3_raise_barrier(core->vm_info, NULL);

	    v3_print_guest_state(core);

	    v3_lower_barrier(core->vm_info);
	    break;
	case PRINT_ARCH_STATE:
	    v3_raise_barrier(core->vm_info, NULL);

	    v3_print_arch_state(core);

	    v3_lower_barrier(core->vm_info);
	    break;
	case PRINT_STACK:
	    v3_raise_barrier(core->vm_info, NULL);

	    v3_print_stack(core);

	    v3_lower_barrier(core->vm_info);
	    break;
	case PRINT_BACKTRACE:
	    v3_raise_barrier(core->vm_info, NULL);

	    v3_print_backtrace(core);
	    
	    v3_lower_barrier(core->vm_info);
	    break;

	case PRINT_STATE:
	    v3_raise_barrier(core->vm_info, NULL);

	    v3_print_core_telemetry(core);
	    v3_print_guest_state(core);
	    v3_print_arch_state(core);

	    v3_lower_barrier(core->vm_info);
	    break;

    }

    return 0;
}


static int evt_handler(struct v3_vm_info * vm, struct v3_debug_event * evt, void * priv_data) {

    V3_Print("Debug Event Handler for core %d\n", evt->core_id);

    if (evt->core_id == -1) {
	int i = 0;
	for (i = 0; i < vm->num_cores; i++) {
	    core_handler(&(vm->cores[i]), evt->cmd);
	}
    } else {
	return core_handler(&vm->cores[evt->core_id], evt->cmd);
    }

    
    return 0;
}


int v3_init_vm_debugging(struct v3_vm_info * vm) {
    v3_hook_host_event(vm, HOST_DEBUG_EVT, 
		       V3_HOST_EVENT_HANDLER(evt_handler), 
		       NULL);


    return 0;
}





void v3_print_segments(struct v3_segments * segs) {
    int i = 0;
    struct v3_segment * seg_ptr;

    seg_ptr=(struct v3_segment *)segs;
  
    char *seg_names[] = {"CS", "DS" , "ES", "FS", "GS", "SS" , "LDTR", "GDTR", "IDTR", "TR", NULL};
    V3_Print("Segments\n");

    for (i = 0; seg_names[i] != NULL; i++) {

	V3_Print("\t%s: Sel=%x, base=%p, limit=%x (long_mode=%d, db=%d)\n", seg_names[i], seg_ptr[i].selector, 
		   (void *)(addr_t)seg_ptr[i].base, seg_ptr[i].limit,
		   seg_ptr[i].long_mode, seg_ptr[i].db);

    }
}



void v3_print_ctrl_regs(struct guest_info * core) {
    struct v3_ctrl_regs * regs = &(core->ctrl_regs);
    int i = 0;
    v3_reg_t * reg_ptr;
    char * reg_names[] = {"CR0", "CR2", "CR3", "CR4", "CR8", "FLAGS", "EFER", NULL};
   

    reg_ptr = (v3_reg_t *)regs;

    V3_Print("Ctrl Regs:\n");

    for (i = 0; reg_names[i] != NULL; i++) {
	V3_Print("\t%s=0x%p (at %p)\n", reg_names[i], (void *)(addr_t)reg_ptr[i], &(reg_ptr[i]));  
    }


}

#if 0
static int safe_gva_to_hva(struct guest_info * core, addr_t linear_addr, addr_t * host_addr) {
    /* select the proper translation based on guest mode */
    if (core->mem_mode == PHYSICAL_MEM) {
    	if (v3_gpa_to_hva(core, linear_addr, host_addr) == -1) return -1;
    } else if (core->mem_mode == VIRTUAL_MEM) {
	if (v3_gva_to_hva(core, linear_addr, host_addr) == -1) return -1;
    }
    return 0;
}

static int v3_print_disassembly(struct guest_info * core) {
    int passed_rip = 0;
    addr_t rip, rip_linear, rip_host;

    /* we don't know where the instructions preceding RIP start, so we just take
     * a guess and hope the instruction stream synced up with our disassembly
     * some time before RIP; if it has not we correct RIP at that point
     */

    /* start disassembly 64 bytes before current RIP, continue 32 bytes after */
    rip = (addr_t) core->rip - 64;
    while ((int) (rip - core->rip) < 32) {
	V3_Print("disassembly step\n");

    	/* always print RIP, even if the instructions before were bad */
    	if (!passed_rip && rip >= core->rip) {
    	    if (rip != core->rip) {
    	    	V3_Print("***** bad disassembly up to this point *****\n");
    	    	rip = core->rip;
    	    }
    	    passed_rip = 1;
    	}

    	/* look up host virtual address for this instruction */
    	rip_linear = get_addr_linear(core, rip, &(core->segments.cs));
    	if (safe_gva_to_hva(core, rip_linear, &rip_host) < 0) {
    	    rip++;
    	    continue;
    	}

    	/* print disassembled instrcution (updates rip) */
    	if (v3_disasm(core, (void *) rip_host, &rip, rip == core->rip) < 0) {
    	    rip++;
    	    continue;
    	}

    }

    return 0;
}

#endif

void v3_print_guest_state(struct guest_info * core) {
    addr_t linear_addr = 0; 

    V3_Print("RIP: %p\n", (void *)(addr_t)(core->rip));
    linear_addr = get_addr_linear(core, core->rip, &(core->segments.cs));
    V3_Print("RIP Linear: %p\n", (void *)linear_addr);

    V3_Print("NumExits: %u\n", (uint32_t)core->num_exits);

    V3_Print("IRQ STATE: started=%d, pending=%d\n", 
	     core->intr_core_state.irq_started, 
	     core->intr_core_state.irq_pending);
    V3_Print("EXCP STATE: err_code_valid=%d, err_code=%x\n", 
	     core->excp_state.excp_error_code_valid, 
	     core->excp_state.excp_error_code);


    v3_print_segments(&(core->segments));
    v3_print_ctrl_regs(core);

    if (core->shdw_pg_mode == SHADOW_PAGING) {
	V3_Print("Shadow Paging Guest Registers:\n");
	V3_Print("\tGuest CR0=%p\n", (void *)(addr_t)(core->shdw_pg_state.guest_cr0));
	V3_Print("\tGuest CR3=%p\n", (void *)(addr_t)(core->shdw_pg_state.guest_cr3));
	V3_Print("\tGuest EFER=%p\n", (void *)(addr_t)(core->shdw_pg_state.guest_efer.value));
	// CR4
    }
    v3_print_GPRs(core);

    v3_print_mem_map(core->vm_info);

    v3_print_stack(core);

    //  v3_print_disassembly(core);
}


void v3_print_arch_state(struct guest_info * core) {


}


void v3_print_guest_state_all(struct v3_vm_info * vm) {
    int i = 0;

    V3_Print("VM Core states for %s\n", vm->name);

    for (i = 0; i < 80; i++) {
	V3_Print("-");
    }

    for (i = 0; i < vm->num_cores; i++) {
	v3_print_guest_state(&vm->cores[i]);  
    }
    
    for (i = 0; i < 80; i++) {
	V3_Print("-");
    }

    V3_Print("\n");    
}



void v3_print_stack(struct guest_info * core) {
    addr_t linear_addr = 0;
    addr_t host_addr = 0;
    int i = 0;
    v3_cpu_mode_t cpu_mode = v3_get_vm_cpu_mode(core);

    linear_addr = get_addr_linear(core, core->vm_regs.rsp, &(core->segments.ss));
 
    V3_Print("Stack at %p:\n", (void *)linear_addr);
   
    if (core->mem_mode == PHYSICAL_MEM) {
	if (v3_gpa_to_hva(core, linear_addr, &host_addr) == -1) {
	    PrintError("Could not translate Stack address\n");
	    return;
	}
    } else if (core->mem_mode == VIRTUAL_MEM) {
	if (v3_gva_to_hva(core, linear_addr, &host_addr) == -1) {
	    PrintError("Could not translate Virtual Stack address\n");
	    return;
	}
    }
    
    V3_Print("Host Address of rsp = 0x%p\n", (void *)host_addr);
 
    // We start i at one because the current stack pointer points to an unused stack element
    for (i = 0; i <= 24; i++) {

	if (cpu_mode == REAL) {
	    V3_Print("\t0x%.4x\n", *((uint16_t *)host_addr + (i * 2)));
	} else if (cpu_mode == LONG) {
	    V3_Print("\t%p\n", (void *)*(addr_t *)(host_addr + (i * 8)));
	} else {
	    // 32 bit stacks...
	    V3_Print("\t0x%.8x\n", *(uint32_t *)(host_addr + (i * 4)));
	}
    }

}    


void v3_print_backtrace(struct guest_info * core) {
    addr_t gla_rbp = 0;
    int i = 0;
    v3_cpu_mode_t cpu_mode = v3_get_vm_cpu_mode(core);

    V3_Print("Performing Backtrace for Core %d\n", core->vcpu_id);
    V3_Print("\tRSP=%p, RBP=%p\n", (void *)core->vm_regs.rsp, (void *)core->vm_regs.rbp);

    gla_rbp = get_addr_linear(core, core->vm_regs.rbp, &(core->segments.ss));

    for (i = 0; i < 10; i++) {
	addr_t hva_rbp = 0; 
	addr_t hva_rip = 0; 

	if (core->mem_mode == PHYSICAL_MEM) {
	    if (v3_gpa_to_hva(core, gla_rbp, &hva_rbp) == -1) {
		PrintError("Could not translate Stack address\n");
		return;
	    }
	} else if (core->mem_mode == VIRTUAL_MEM) {
	    if (v3_gva_to_hva(core, gla_rbp, &hva_rbp) == -1) {
		PrintError("Could not translate Virtual Stack address\n");
		return;
	    }
	}

	hva_rip = hva_rbp + v3_get_addr_width(core);

	
	if (cpu_mode == REAL) {
	    V3_Print("Next RBP=0x%.4x, RIP=0x%.4x\n", *(uint16_t *)hva_rbp,*(uint16_t *)hva_rip);
	    gla_rbp = *(uint16_t *)hva_rbp;
	} else if (cpu_mode == LONG) {
	    V3_Print("Next RBP=%p, RIP=%p\n", (void *)*(uint64_t *)hva_rbp, (void *)*(uint64_t *)hva_rip);
	    gla_rbp = *(uint64_t *)hva_rbp;
	} else {
	    V3_Print("Next RBP=0x%.8x, RIP=0x%.8x\n", *(uint32_t *)hva_rbp, *(uint32_t *)hva_rip);
	    gla_rbp = *(uint32_t *)hva_rbp;
	}
    }
}


#ifdef __V3_32BIT__

void v3_print_GPRs(struct guest_info * core) {
    struct v3_gprs * regs = &(core->vm_regs);
    int i = 0;
    v3_reg_t * reg_ptr;
    char * reg_names[] = { "RDI", "RSI", "RBP", "RSP", "RBX", "RDX", "RCX", "RAX", NULL};

    reg_ptr = (v3_reg_t *)regs;

    V3_Print("32 bit GPRs:\n");

    for (i = 0; reg_names[i] != NULL; i++) {
	V3_Print("\t%s=0x%p (at %p)\n", reg_names[i], (void *)(addr_t)reg_ptr[i], &(reg_ptr[i]));  
    }
}

#elif __V3_64BIT__

void v3_print_GPRs(struct guest_info * core) {
    struct v3_gprs * regs = &(core->vm_regs);
    int i = 0;
    v3_reg_t * reg_ptr;
    char * reg_names[] = { "RDI", "RSI", "RBP", "RSP", "RBX", "RDX", "RCX", "RAX", \
			   "R8", "R9", "R10", "R11", "R12", "R13", "R14", "R15", NULL};

    reg_ptr = (v3_reg_t *)regs;

    V3_Print("64 bit GPRs:\n");

    for (i = 0; reg_names[i] != NULL; i++) {
	V3_Print("\t%s=0x%p (at %p)\n", reg_names[i], (void *)(addr_t)reg_ptr[i], &(reg_ptr[i]));  
    }
}

#endif