#include <system.h>

static char fxsave_region[512] __attribute__((aligned(16)));

static inline void cpuid(uint32_t leaf, uint32_t *eax, uint32_t *ebx,
                         uint32_t *ecx, uint32_t *edx) {
	asm volatile("cpuid"
	             : "=a"(*eax), "=b"(*ebx), "=c"(*ecx), "=d"(*edx)
	             : "a"(leaf));
}

module_info_t init(env_t *env) {
	init_env(env);

	uint32_t eax, ebx, ecx, edx;
	cpuid(1, &eax, &ebx, &ecx, &edx);

	if ((edx >> 23) & 1) { env->fb_printf("MMX поддерживается!\n"); }

	if ((edx >> 25) & 1) {
		env->fb_printf("SSE2 поддерживается!\n");
		env->fb_printf("Адрес региона fxsave 0x%x\n", &fxsave_region);
		asm volatile(" fxsave %0 " ::"m"(fxsave_region));
		uint32_t sse_version = (ecx >> 25) & 0x7;
		env->fb_printf("SSE%u включен\n", sse_version);
	}

	if ((ecx >> 28) & 1) { env->fb_printf("AVX поддерживается!\n"); }

	return (module_info_t){
		.name = (char *)"SIMD",
		.message = (char *)"SIMD инструкции",
		.err_code = 0,
	};
}