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/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

@@ -527,6 +527,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;
         }
     }

core/memory.c

@@ -287,8 +287,8 @@ int hax_vm_set_ram(struct vm_t *vm, struct hax_set_ram_info *info)
     return handle_set_ram(vm, info->pa_start, info->size, info->va,
                           info->flags);
 #else  // !CONFIG_HAX_EPT2
-    int num = info->size >> page_shift;
-    uint64_t gpfn = info->pa_start >> page_shift;
+    int num = info->size >> PAGE_SHIFT;
+    uint64_t gpfn = info->pa_start >> PAGE_SHIFT;
     uint64_t cur_va = info->va;
     bool is_unmap = info->flags & HAX_RAM_INFO_INVALID;
     bool is_readonly = info->flags & HAX_RAM_INFO_ROM;
@@ -342,13 +342,13 @@ int hax_vm_set_ram(struct vm_t *vm, struct hax_set_ram_info *info)
             hva = 0;
 #endif
 #endif
-            cur_va += page_size;
+            cur_va += PAGE_SIZE;
         }
 
         if (!hax_core_set_p2m(vm, gpfn, hpfn, hva, info->flags)) {
             return -ENOMEM;
         }
-        if (!ept_set_pte(vm, gpfn << page_shift, hpfn << page_shift, emt, perm,
+        if (!ept_set_pte(vm, gpfn << PAGE_SHIFT, hpfn << PAGE_SHIFT, emt, perm,
                          &epte_modified)) {
             hax_error("ept_set_pte() failed at gpfn 0x%llx hpfn 0x%llx\n", gpfn,
                       hpfn);