Added support for Linux hosts

Author: AlexAltea

.gitignore

@@ -18,3 +18,13 @@ Win8.1Release/
 # macOS
 /darwin/hax_driver/com_intel_hax/build/
 .DS_Store
+
+# Linux
+*.o
+*.cmd
+*.ko
+*.mod.c
+.tmp_versions
+.cache.mk
+modules.order
+Module.symvers

core/cpu.c

@@ -572,7 +572,7 @@ uint32_t load_vmcs(struct vcpu_t *vcpu, preempt_flag *flags)
     /* when wake up from sleep, we need the barrier, as vm operation
      * are not serialized instructions.
      */
-    smp_mb();
+    hax_smp_mb();
 
     cpu_data = current_cpu_data();
 

core/ept.c

@@ -328,7 +328,7 @@ static void invept_smpfunc(struct invept_bundle *bundle)
 {
     struct per_cpu_data *cpu_data;
 
-    smp_mb();
+    hax_smp_mb();
     cpu_data = current_cpu_data();
     cpu_data->invept_res = VMX_SUCCEED;
 
@@ -373,7 +373,7 @@ void invept(hax_vm_t *hax_vm, uint type)
 
     bundle.type = type;
     bundle.desc = &desc;
-    smp_call_function(&cpu_online_map, (void (*)(void *))invept_smpfunc,
+    hax_smp_call_function(&cpu_online_map, (void (*)(void *))invept_smpfunc,
                       &bundle);
 
     /*

core/hax.c

@@ -62,12 +62,12 @@ struct hax_t *hax;
 extern hax_atomic_t vmx_cpu_num, vmx_enabled_num;
 static void hax_enable_vmx(void)
 {
-    smp_call_function(&cpu_online_map, cpu_init_vmx, NULL);
+    hax_smp_call_function(&cpu_online_map, cpu_init_vmx, NULL);
 }
 
 static void hax_disable_vmx(void)
 {
-    smp_call_function(&cpu_online_map, cpu_exit_vmx, NULL);
+    hax_smp_call_function(&cpu_online_map, cpu_exit_vmx, NULL);
 }
 
 static void free_cpu_vmxon_region(void)
@@ -415,7 +415,7 @@ static void hax_pmu_init(void)
     int ref_cpu_id = -1;
 
     // Execute cpu_pmu_init() on each logical processor of the host CPU
-    smp_call_function(&cpu_online_map, cpu_pmu_init, NULL);
+    hax_smp_call_function(&cpu_online_map, cpu_pmu_init, NULL);
 
     // Find the common APM version supported by all host logical processors
     // TODO: Theoretically we should do the same for other APM parameters

core/ia32_ops.asm

@@ -161,7 +161,7 @@ function __nmi, 0
     int 2h
     ret
 
-function __fls, 1
+function asm_fls, 1
     xor reg_ret_32, reg_ret_32
     bsr reg_ret_32, reg_arg1_32
     ret

core/include/cpu.h

@@ -107,12 +107,12 @@ struct per_cpu_data {
 
     /*
      * These fields are used to record the result of certain VMX instructions
-     * when they are used in a function wrapped by smp_call_function(). This is
+     * when they are used in a function wrapped by hax_smp_call_function(). This is
      * because it is not safe to call hax_error(), etc. (whose underlying
      * implementation may use a lock) from the wrapped function to log a
      * failure; doing so may cause a deadlock and thus a host reboot, especially
      * on macOS, where mp_rendezvous_no_intrs() (the legacy Darwin API used by
-     * HAXM to implement smp_call_function()) is known to be prone to deadlocks:
+     * HAXM to implement hax_smp_call_function()) is known to be prone to deadlocks:
      * https://lists.apple.com/archives/darwin-kernel/2006/Dec/msg00006.html
      */
     vmx_result_t    vmxon_res;

core/include/emulate_ops.h

@@ -39,7 +39,7 @@
                     (0 * FASTOP_ALIGN))
 
 /* Instruction handlers */
-typedef void(__cdecl em_handler_t)(void);
+typedef void(ASMCALL em_handler_t)(void);
 em_handler_t em_not;
 em_handler_t em_neg;
 em_handler_t em_inc;
@@ -72,7 +72,7 @@ em_handler_t em_bextr;
 em_handler_t em_andn;
 
 /* Dispatch handlers */
-void __cdecl fastop_dispatch(void *handler, uint64_t *dst,
+void ASMCALL fastop_dispatch(void *handler, uint64_t *dst,
                              uint64_t *src1, uint64_t *src2, uint64_t *flags);
 
 #endif /* HAX_CORE_EMULATE_OPS_H_ */

core/include/ia32.h

@@ -78,7 +78,7 @@ void ASMCALL asm_fxrstor(mword *addr);
 void ASMCALL asm_cpuid(union cpuid_args_t *state);
 
 void ASMCALL __nmi(void);
-uint32_t ASMCALL __fls(uint32_t bit32);
+uint32_t ASMCALL asm_fls(uint32_t bit32);
 
 uint64_t ia32_rdmsr(uint32_t reg);
 void ia32_wrmsr(uint32_t reg, uint64_t val);

core/include/vmx.h

@@ -681,6 +681,17 @@ void vmx_vmwrite(struct vcpu_t *vcpu, const char *name,
             vmwrite(vcpu, GUEST_##seg##_AR, tmp_ar);               \
         }
 
+#elif defined(__linux__)
+#define VMWRITE_SEG(vcpu, seg, val) ({                             \
+            uint32_t tmp_ar = val.ar;                              \
+            if (tmp_ar == 0)                                       \
+                tmp_ar = 0x10000;                                  \
+            vmwrite(vcpu, GUEST_##seg##_SELECTOR, (val).selector); \
+            vmwrite(vcpu, GUEST_##seg##_BASE, (val).base);         \
+            vmwrite(vcpu, GUEST_##seg##_LIMIT, (val).limit);       \
+            vmwrite(vcpu, GUEST_##seg##_AR, tmp_ar);               \
+        })
+
 #elif defined(__MACH__)
 #define VMWRITE_SEG(vcpu, seg, val) ({                             \
             uint32_t tmp_ar = val.ar;                              \

core/intr_exc.c

@@ -48,7 +48,7 @@ uint32_t vcpu_get_pending_intrs(struct vcpu_t *vcpu)
 
     for (i = 7; i >= 0; i--) {
         if (intr_pending[i]) {
-            offset = __fls(intr_pending[i]);
+            offset = asm_fls(intr_pending[i]);
             break;
         }
     }