/* Northwestern University */
/* (c) 2008, Jack Lange <jarusl@cs.northwestern.edu> */

#include <palacios/vm_guest.h>
#include <palacios/vmm_ctrl_regs.h>
#include <palacios/vmm.h>


vm_cpu_mode_t get_cpu_mode(struct guest_info * info) {
  struct cr0_32 * cr0;
  struct cr4_32 * cr4 = (struct cr4_32 *)&(info->ctrl_regs.cr4);
  struct efer_64 * efer = (struct efer_64 *)&(info->ctrl_regs.efer);
  struct v3_segment * cs = &(info->segments.cs);

  if (info->shdw_pg_mode == SHADOW_PAGING) {
    cr0 = (struct cr0_32 *)&(info->shdw_pg_state.guest_cr0);
  } else if (info->shdw_pg_mode == NESTED_PAGING) {
    cr0 = (struct cr0_32 *)&(info->ctrl_regs.cr0);
  } else {
    PrintError("Invalid Paging Mode...\n");
    V3_ASSERT(0);
    return -1;
  }

  if (cr0->pe == 0) {
    return REAL;
  } else if ((cr4->pae == 0) && (efer->lma == 0)) {
    return PROTECTED;
  } else if (efer->lma == 0) {
    return PROTECTED_PAE;
  } else if ((efer->lma == 1) && (cs->long_mode == 1)) {
    return LONG;
  } else {
    return LONG_32_COMPAT;
  }
}

vm_mem_mode_t get_mem_mode(struct guest_info * info) {
  struct cr0_32 * cr0;

  if (info->shdw_pg_mode == SHADOW_PAGING) {
    cr0 = (struct cr0_32 *)&(info->shdw_pg_state.guest_cr0);
  } else if (info->shdw_pg_mode == NESTED_PAGING) {
    cr0 = (struct cr0_32 *)&(info->ctrl_regs.cr0);
  } else {
    PrintError("Invalid Paging Mode...\n");
    V3_ASSERT(0);
    return -1;
  }



  if (cr0->pg == 0) {
    return PHYSICAL_MEM;
  } else {
    return VIRTUAL_MEM;
  }
}


void PrintV3Segments(struct guest_info * info) {
  struct v3_segments * segs = &(info->segments);
  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};
  PrintDebug("Segments\n");

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

    PrintDebug("\t%s: Sel=%x, base=%x, limit=%x\n", seg_names[i], seg_ptr[i].selector, seg_ptr[i].base, seg_ptr[i].limit);

  }

}


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

  reg_ptr= (v3_reg_t *)regs;

  PrintDebug("32 bit Ctrl Regs:\n");

  for (i = 0; reg_names[i] != NULL; i++) {
    PrintDebug("\t%s=0x%x\n", reg_names[i], reg_ptr[i]);  
  }
}


void PrintV3GPRs(struct guest_info * info) {
  struct v3_gprs * regs = &(info->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;

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

  for (i = 0; reg_names[i] != NULL; i++) {
    PrintDebug("\t%s=0x%x\n", reg_names[i], reg_ptr[i]);  
  }
}