--- /dev/null
+#include <stdlib.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+
+
+#include "test.h"
+#include "vmm_xed.h"
+#include "vmm_decoder.h"
+#include "vm_guest.h"
+
+
+/* Disgusting mask hack...
+ I can't think right now, so we'll do it this way...
+*/
+static const ullong_t mask_1 = 0x00000000000000ffLL;
+static const ullong_t mask_2 = 0x000000000000ffffLL;
+static const ullong_t mask_4 = 0x00000000ffffffffLL;
+static const ullong_t mask_8 = 0xffffffffffffffffLL;
+
+
+#define MASK(val, length) ({ \
+ ullong_t mask = 0x0LL; \
+ switch (length) { \
+ case 1: \
+ mask = mask_1; \
+ case 2: \
+ mask = mask_2; \
+ case 4: \
+ mask = mask_4; \
+ case 8: \
+ mask = mask_8; \
+ } \
+ val & mask;}) \
+
+static void init_guest_info(struct guest_info * info) {
+ memset(info, 0, sizeof(struct guest_info));
+ info->cpu_mode = PROTECTED;
+
+ info->vm_regs.rax = 0x01010101;
+ info->vm_regs.rbx = 0x02020202;
+ info->vm_regs.rcx = 0x03030303;
+ info->vm_regs.rdx = 0x04040404;
+
+ info->vm_regs.rdi = 0x05050505;
+ info->vm_regs.rsi = 0x06060606;
+ info->vm_regs.rsp = 0x07070707;
+ info->vm_regs.rbp = 0x08080808;
+
+ info->vm_regs.rdi = 0x05050505;
+ info->vm_regs.rsi = 0x06060606;
+ info->vm_regs.rsp = 0x07070707;
+ info->vm_regs.rbp = 0x08080808;
+
+
+ info->segments.ds.base = 0xf0f0f0f0;
+ info->segments.es.base = 0xe0e0e0e0;
+
+
+
+
+}
+static const char * mem = "MEMORY";
+static const char * reg = "REGISTER";
+static const char * imm = "IMMEDIATE";
+static const char * invalid = "INVALID";
+
+static const char * get_op_type_str(operand_type_t type) {
+ if (type == MEM_OPERAND) {
+ return mem;
+ } else if (type == REG_OPERAND) {
+ return reg;
+ } else if (type == IMM_OPERAND) {
+ return imm;
+ } else {
+ return invalid;
+ }
+}
+
+static int print_op(struct x86_operand *op) {
+ printf("\ttype=%s\n", get_op_type_str(op->type));
+
+ switch (op->type) {
+ case REG_OPERAND:
+ printf("\tsize=%d\n", op->size);
+ printf("\taddr=0x%x (val=%x)\n", op->operand, MASK(*(uint_t*)(op->operand), op->size));
+ return 0;
+ case MEM_OPERAND:
+ printf("\tsize=%d\n", op->size);
+ printf("\taddr=0x%x\n", op->operand);
+ return 0;
+
+ case IMM_OPERAND:
+ printf("\tsize=%d\n", op->size);
+ printf("\tval=0x%x\n", op->operand);
+ return 0;
+
+ default:
+ return -1;
+ }
+}
+
+int main(int argc, char ** argv) {
+ char * filename;
+ int fd;
+ struct stat file_state;
+ int ret;
+ char * file_buf;
+ int buf_offset = 0;
+ int file_size = 0;
+ char * instr_ptr = 0;
+
+ struct guest_info * info = (struct guest_info *)malloc(sizeof(struct guest_info ));;
+
+ init_decoder();
+ init_guest_info(info);
+
+ if (argc == 1) {
+ printf("Error: Must give a binary file\n");
+ exit(-1);
+ }
+
+ filename = argv[1];
+
+ ret = stat(filename, &file_state);
+
+ if (ret == -1) {
+ printf("Could not stat file\n");
+ return -1;
+ }
+ file_size = file_state.st_size;
+
+ file_buf = malloc(file_size);
+
+ fd = open(filename, NULL);
+
+ if (fd == -1) {
+ printf("Could not open file\n");
+ return -1;
+ } else {
+ int total_read = 0;
+ int num_read = 0;
+
+ while (total_read < file_size) {
+ num_read = read(fd, file_buf + total_read, file_size - total_read);
+
+ if (num_read == 0) {
+ printf("end of file\n");
+ break;
+ }
+ if (num_read == -1) {
+ printf("Read error\n");
+ exit(-1);
+ }
+
+ total_read += num_read;
+ }
+
+ }
+
+
+ instr_ptr = file_buf;
+
+
+ PrintV3CtrlRegs(info);
+ PrintV3GPRs(info);
+ PrintV3Segments(info);
+
+
+ while (buf_offset < file_size) {
+ struct x86_instr instr;
+
+ if (v3_decode(info, (addr_t)instr_ptr + buf_offset, &instr) == -1) {
+ printf("Unhandled instruction\n");
+ buf_offset += instr.instr_length;
+ continue;
+ }
+ printf("instr_length = %d, noperands=%d\n", instr.instr_length, instr.num_operands);
+
+ printf("Source:\n");
+ print_op(&(instr.src_operand));
+
+ printf("Dest:\n");
+ print_op(&(instr.dst_operand));
+
+
+ printf("\n\n");
+
+ buf_offset += instr.instr_length;
+ }
+
+ return 0;
+}
