code clean up

[palacios.git] / palacios / include / palacios / vmcb.h

diff --git a/palacios/include/palacios/vmcb.h b/palacios/include/palacios/vmcb.h
index cfbeb8b..3b97500 100644 (file)
--- a/palacios/include/palacios/vmcb.h
+++ b/palacios/include/palacios/vmcb.h
@@ -1,8 +1,10 @@
 #ifndef __VMCB_H
 #define __VMCB_H
 
-#include <palacios/vmm_types.h>
+#ifdef __V3VEE__
 
+#include <palacios/vmm_types.h>
+#include <palacios/vm_guest.h>
 
 #define VMCB_CTRL_AREA_OFFSET                   0x0
 #define VMCB_STATE_SAVE_AREA_OFFSET             0x400
@@ -357,4 +359,9 @@ typedef struct VMCB_State_Save_Area {
 void PrintDebugVMCB(vmcb_t * vmcb);
 
 
+void set_vmcb_segments(vmcb_t * vmcb, struct v3_segments * segs);
+void get_vmcb_segments(vmcb_t * vmcb, struct v3_segments * segs);
+
+#endif // ! __V3VEE__
+
 #endif