4 files changed, 64 insertions, 0 deletions
diff --git a/kernel/src/cpu/cpu.c b/kernel/src/cpu/cpu.c
new file mode 100644
index 0000000..f78ef26
--- /dev/null
+++ b/kernel/src/cpu/cpu.c
@@ -0,0 +1,20 @@
+#include "cpu.h"
+
+#include "print.h"
+
+extern int sse_init (void);
+extern int fpu_init (void);
+
+void init_registers (void) {
+    if (!sse_init()) {
+        debugk("Loaded SIMD");
+    } else {
+        errork("SIMD not supported");
+    }
+
+    if (!fpu_init()) {
+        debugk("Loaded FPU");
+    } else {
+        errork("FPU not supported");
+    }
+}
diff --git a/kernel/src/cpu/cpu.h b/kernel/src/cpu/cpu.h
new file mode 100644
index 0000000..3044292
--- /dev/null
+++ b/kernel/src/cpu/cpu.h
@@ -0,0 +1,3 @@
+#pragma once
+
+void init_registers (void);
diff --git a/kernel/src/cpu/fpu.asm b/kernel/src/cpu/fpu.asm
new file mode 100644
index 0000000..e49ab6f
--- /dev/null
+++ b/kernel/src/cpu/fpu.asm
@@ -0,0 +1,19 @@
+global fpu_init
+
+fpu_init:
+    mov edx, cr0
+    and edx, (-1) - ((1 << 3) + (1 << 4))       
+    mov cr0, edx
+    fninit                                  
+    fnstsw [test]                      
+    cmp word [test], 0                 
+    jne no_fpu
+    
+    xor eax, 0
+    ret
+
+no_fpu:
+	mov eax, 1
+	ret
+
+test: dw 0x55AA
diff --git a/kernel/src/cpu/sse.asm b/kernel/src/cpu/sse.asm
new file mode 100644
index 0000000..e7b5a99
--- /dev/null
+++ b/kernel/src/cpu/sse.asm
@@ -0,0 +1,22 @@
+global sse_init
+
+sse_init:
+    mov eax, 0x1
+    cpuid
+    test edx, 1<<25
+    jmp no_sse
+
+    mov eax, cr0
+    and ax, 0xFFFB		
+    or ax, 0x2			
+    mov cr0, eax
+    mov eax, cr4
+    or ax, 3 << 9		
+    mov cr4, eax
+
+    mov eax, 0
+    ret
+
+no_sse:
+	mov eax, 1
+	ret