X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/cd09c4b222e4ddf45a42522563ae2471a829d1c8..5a8d4ebecaa50af25bd53c3b2236dca77a14eb81:/lunaix-os/includes/hal/cpu.h

diff --git a/lunaix-os/includes/hal/cpu.h b/lunaix-os/includes/hal/cpu.h
index 2a06f29..af56120 100644
--- a/lunaix-os/includes/hal/cpu.h
+++ b/lunaix-os/includes/hal/cpu.h
@@ -1,36 +1,126 @@
 #ifndef __LUNAIX_CPU_H
 #define __LUNAIX_CPU_H
 
-typedef unsigned long reg32;
+#include <stdint.h>
+
+#define SEL_RPL(selector) ((selector)&0x3)
+
+typedef unsigned int reg32;
 typedef unsigned short reg16;
 
-typedef struct {
+typedef struct
+{
+    reg32 eax;
+    reg32 ebx;
+    reg32 ecx;
+    reg32 edx;
     reg32 edi;
-    reg32 esi;
     reg32 ebp;
+    reg32 esi;
     reg32 esp;
-    reg32 ebx;
-    reg32 edx;
-    reg32 ecx;
-    reg32 eax;
-} __attribute__((packed)) registers;
+} __attribute__((packed)) gp_regs;
+
+typedef struct
+{
+    reg16 ds;
+    reg16 es;
+    reg16 fs;
+    reg16 gs;
+} __attribute__((packed)) sg_reg;
+
+void
+cpu_get_brand(char* brand_out);
+
+int
+cpu_has_apic();
+
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wreturn-type"
+static inline reg32
+cpu_rcr0()
+{
+    uintptr_t val;
+    asm volatile("movl %%cr0,%0" : "=r"(val));
+    return val;
+}
+
+static inline reg32
+cpu_rcr2()
+{
+    uintptr_t val;
+    asm volatile("movl %%cr2,%0" : "=r"(val));
+    return val;
+}
+
+static inline reg32
+cpu_rcr3()
+{
+    uintptr_t val;
+    asm volatile("movl %%cr3,%0" : "=r"(val));
+    return val;
+}
+
+static inline reg32
+cpu_reflags()
+{
+    uintptr_t val;
+    asm volatile("pushf\n"
+                 "popl %0\n"
+                 : "=r"(val)::);
+    return val;
+}
+#pragma GCC diagnostic pop
 
-reg32 cpu_r_cr0();
+static inline void
+cpu_lcr0(reg32 v)
+{
+    asm("mov %0, %%cr0" ::"r"(v));
+}
 
-reg32 cpu_r_cr2();
+static inline void
+cpu_lcr2(reg32 v)
+{
+    asm("mov %0, %%cr2" ::"r"(v));
+}
 
-reg32 cpu_r_cr3();
+static inline void
+cpu_lcr3(reg32 v)
+{
+    asm("mov %0, %%cr3" ::"r"(v));
+}
 
-void cpu_w_cr0(reg32 v);
+static inline void
+cpu_invplg(void* va)
+{
+    asm volatile("invlpg (%0)" ::"r"((uintptr_t)va) : "memory");
+}
 
-void cpu_w_cr2(reg32 v);
+static inline void
+cpu_enable_interrupt()
+{
+    asm volatile("sti");
+}
 
-void cpu_w_cr3(reg32 v);
+static inline void
+cpu_disable_interrupt()
+{
+    asm volatile("cli");
+}
 
-void cpu_get_model(char* model_out);
+static inline void
+cpu_invtlb()
+{
+    reg32 interm;
+    asm("movl %%cr3, %0\n"
+        "movl %0, %%cr3"
+        : "=r"(interm)
+        : "r"(interm));
+}
 
-int cpu_brand_string_supported();
+void
+cpu_rdmsr(uint32_t msr_idx, uint32_t* reg_high, uint32_t* reg_low);
 
-void cpu_get_brand(char* brand_out);
+void
+cpu_wrmsr(uint32_t msr_idx, uint32_t reg_high, uint32_t reg_low);
 
-#endif /* __LUNAIX_CPU_H */
+#endif
\ No newline at end of file