X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Usermode%2FLibraries%2Flibc.so_src%2Fstub.c;h=9f629ba409502479ab23ecdd2d1421ba3fe9d63f;hb=5c46f86c5a8ceaa63a1a9919cf1f4d2889c6c233;hp=acb36539ce957e621a7acda3963d5165878bf8b4;hpb=91dd38c34820c03311738439125675d59bf9e3f1;p=tpg%2Facess2.git

diff --git a/Usermode/Libraries/libc.so_src/stub.c b/Usermode/Libraries/libc.so_src/stub.c
index acb36539..9f629ba4 100644
--- a/Usermode/Libraries/libc.so_src/stub.c
+++ b/Usermode/Libraries/libc.so_src/stub.c
@@ -2,10 +2,27 @@
  * AcessOS Basic C Library
  */
 #include "stdio_int.h"
+#include "lib.h"
 
+#define USE_CPUID	0
+
+// === PROTOTYPES ===
+#if USE_CPUID
+static void	cpuid(uint32_t Num, uint32_t *EAX, uint32_t *EBX, uint32_t *EDX, uint32_t *ECX);
+#endif
+
+// === GLOBALS ===
 extern char **_envp;
 extern struct sFILE	_iob[];
+extern struct sFILE	*stdin;
+extern struct sFILE	*stdout;
+extern struct sFILE	*stderr;
+// --- CPU Features ---
+#if USE_CPUID
+tCPUID	gCPU_Features;
+#endif
 
+// === CODE ===
 /**
  * \fn int SoMain()
  * \brief Stub Entrypoint
@@ -20,9 +37,43 @@ int SoMain(unsigned int BaseAddress, int argc, char **argv, char **envp)
 	_envp = envp;
 	
 	// Init FileIO Pointers
-	_iob[0].FD = 0;	_iob[0].Flags = FILE_FLAG_MODE_READ;
-	_iob[1].FD = 1;	_iob[1].Flags = FILE_FLAG_MODE_WRITE;
-	_iob[2].FD = 2;	_iob[2].Flags = FILE_FLAG_MODE_WRITE;
+	stdin = &_iob[0];
+	stdin->FD = 0;	stdin->Flags = FILE_FLAG_MODE_READ;
+	stdout = &_iob[1];
+	stdout->FD = 1;	stdout->Flags = FILE_FLAG_MODE_WRITE;
+	stderr = &_iob[2];
+	stderr->FD = 2;	stderr->Flags = FILE_FLAG_MODE_WRITE;
+	
+	#if USE_CPUID
+	{
+		uint32_t	ecx, edx;
+		cpuid(1, NULL, NULL, &edx, &ecx);
+		gCPU_Features.SSE  = edx & (1 << 25);	// SSE
+		gCPU_Features.SSE2 = edx & (1 << 26);	// SSE2
+	}
+	#endif
 	
 	return 1;
 }
+
+
+#if USE_CPUID
+/**
+ * \brief Call the CPUID opcode
+ */
+static void cpuid(uint32_t Num, uint32_t *EAX, uint32_t *EBX, uint32_t *EDX, uint32_t *ECX)
+{
+	uint32_t	eax, ebx, edx, ecx;
+	
+	__asm__ __volatile__ (
+		"cpuid"
+		: "=a"(eax), "=c"(ecx), "=d"(edx)
+		: "a"(Num)
+		);
+	
+	if(EAX)	*EAX = eax;
+	if(EBX)	*EBX = ebx;
+	if(EDX)	*EDX = edx;
+	if(ECX)	*ECX = ecx;
+}
+#endif