[major] C++ Migration completed

Author: utcq

src/include/arch/sse.h

@@ -0,0 +1,9 @@
+#ifndef SSE_H
+#define SSE_H
+
+namespace SSE {
+  bool __attribute__((naked)) enable();
+  bool test();
+}
+
+#endif

src/kernel/arch/sse.cxx

@@ -0,0 +1,46 @@
+#include <arch/sse.h>
+
+bool __attribute__((naked)) SSE::enable() {
+__asm__ __volatile__(
+    "mov %cs, %ax\n"
+    "and $3, %ax\n"
+    "cmp $0, %ax\n"
+    "jne fl\n"
+    
+    "mov $1, %eax\n"
+    "cpuid\n"
+    "bt $25, %edx\n"
+    "jnc fl\n"
+
+    "test $1, %ecx\n"
+    "jz fl\n"
+    "mov %ecx, %ebx\n"
+    "test $(1 << 19), %ebx\n"
+    "jz fl\n"
+    "test $(1 << 20), %ebx\n"
+    "jz fl\n"
+    
+    "mov %cr0, %rax\n"
+    "and $~(1<<2), %rax\n"
+    "or $(1<<1), %rax\n"
+    "mov %rax, %cr0\n"
+
+    "mov %cr4, %rax\n"
+    "or $((1<<9) | (1<<10)), %rax\n"
+    "mov %rax, %cr4\n"
+    "xor %eax, %eax\n"
+    "mov $1, %al\n"
+    "ret\n"
+
+
+    "fl:\n"
+    "xor %eax, %eax\n"
+    "ret\n"
+  );
+}
+
+bool SSE::test() {
+  unsigned int eax, ebx, ecx, edx;
+  __asm__ __volatile__("cpuid" : "=a"(eax), "=b"(ebx), "=c"(ecx), "=d"(edx) : "a"(1));
+  return edx & (1 << 25);
+}

src/kernel/arch/sse.s

@@ -18,25 +18,19 @@
 
 #include <drivers/fb_gfx.h>
 
-const char *SSE_EXT_MSG_TABLE[5] = {
-  "Unsupported",
-  "SSE",
-  "SSE3, SSE4.1, SSE4.2",
-  "SSE2",
-  "XOP (SSE5)",
-};
-extern "C" {
-  extern bool __enable_sse_ext();
-  extern bool __test_sse();
+
+namespace SSE {
+  u8 enable();
+  bool test();
 }
 void _kernel_pre_setup() {
   UART::init();
   boot_ctx.gather();
   Mem::Frame::init();
   Mem::Paging::init(); Log::success("Memory mapped"); Log::success("Paging initialized");
   Err::Handler::init_symbols(); Log::success("Kernel symbols mapped");
-  bool sse_ext = __enable_sse_ext();
-  if (sse_ext && __test_sse()) Log::success("SSE extensions enabled up to %s", SSE_EXT_MSG_TABLE[sse_ext]);
+
+  if (SSE::enable() && SSE::test()) Log::success("SSE extensions enabled");
   else Log::critical("Failed to enable SSE");
 }