1 files changed, 16 insertions, 0 deletions
diff --git a/sys/include/arch/amd64/cpu.h b/sys/include/arch/amd64/cpu.h
index dd753b7..46e5df7 100644
--- a/sys/include/arch/amd64/cpu.h
+++ b/sys/include/arch/amd64/cpu.h
@@ -33,18 +33,27 @@
 #include <sys/types.h>
 #include <sys/cdefs.h>
 #include <sys/proc.h>
+#include <sys/spinlock.h>
 #include <machine/tss.h>
 
 #define CPU_IRQ(IRQ_N) (BIT((IRQ_N)) & 0xFF)
 
+/* Feature bits */
+#define CPU_FEAT_SMAP  BIT(0)
+#define CPU_FEAT_SMEP  BIT(1)
+
 struct cpu_info {
     uint32_t apicid;
+    uint32_t feat;
     uint8_t has_x2apic : 1;
+    uint8_t tlb_shootdown : 1;
     uint8_t ipl;
     size_t lapic_tmr_freq;
     uint8_t irq_mask;
+    vaddr_t shootdown_va;
     struct tss_entry *tss;
     struct proc *curtd;
+    struct spinlock lock;
     struct cpu_info *self;
 };
 
@@ -52,6 +61,13 @@ __dead void cpu_halt_all(void);
 void cpu_halt_others(void);
 void cpu_startup(struct cpu_info *ci);
 
+void cpu_enable_smep(void);
+void cpu_disable_smep(void);
+
+struct cpu_info *cpu_get(uint32_t index);
+uint32_t cpu_count(void);
+void cpu_shootdown_tlb(vaddr_t va);
+
 struct cpu_info *this_cpu(void);
 void mp_bootstrap_aps(struct cpu_info *ci);