#include <rand.hpp>
#include <cpu.hpp>

namespace Random
{
    static uint64_t Seed = 0xdeadbeef;

    uint16_t rand16()
    {
        int RDRANDFlag = 0;
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_AMD) == 0)
        {
#if defined(__amd64__)
            CPU::x64::AMD::CPUID0x1 cpuid1amd;
#elif defined(__i386__)
            CPU::x32::AMD::CPUID0x1 cpuid1amd;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1amd.EAX.raw), "=b"(cpuid1amd.EBX.raw), "=c"(cpuid1amd.ECX.raw), "=d"(cpuid1amd.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1amd.ECX.RDRAND;
        }
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_INTEL) == 0)
        {
#if defined(__amd64__)
            CPU::x64::Intel::CPUID0x1 cpuid1intel;
#elif defined(__i386__)
            CPU::x32::Intel::CPUID0x1 cpuid1intel;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1intel.EAX.raw), "=b"(cpuid1intel.EBX.raw), "=c"(cpuid1intel.ECX.raw), "=d"(cpuid1intel.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1intel.ECX.RDRAND;
        }

        if (strcmp(CPU::Hypervisor(), x86_CPUID_VENDOR_TCG) == 0)
            RDRANDFlag = 0;

#if defined(__amd64__) || defined(__i386__)
        if (RDRANDFlag)
        {
            uint16_t RDRANDValue = 0;
            asmv("1: rdrand %0; jnc 1b"
                 : "=r"(RDRANDValue));
            return RDRANDValue;
        }
#endif

        Seed = Seed * 1103515245 + 12345;
        return (uint16_t)(Seed / 65536) % __UINT16_MAX__;
    }

    uint32_t rand32()
    {
        int RDRANDFlag = 0;
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_AMD) == 0)
        {
#if defined(__amd64__)
            CPU::x64::AMD::CPUID0x1 cpuid1amd;
#elif defined(__i386__)
            CPU::x32::AMD::CPUID0x1 cpuid1amd;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1amd.EAX.raw), "=b"(cpuid1amd.EBX.raw), "=c"(cpuid1amd.ECX.raw), "=d"(cpuid1amd.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1amd.ECX.RDRAND;
        }
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_INTEL) == 0)
        {
#if defined(__amd64__)
            CPU::x64::Intel::CPUID0x1 cpuid1intel;
#elif defined(__i386__)
            CPU::x32::Intel::CPUID0x1 cpuid1intel;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1intel.EAX.raw), "=b"(cpuid1intel.EBX.raw), "=c"(cpuid1intel.ECX.raw), "=d"(cpuid1intel.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1intel.ECX.RDRAND;
        }

        if (strcmp(CPU::Hypervisor(), x86_CPUID_VENDOR_TCG) == 0)
            RDRANDFlag = 0;

#if defined(__amd64__) || defined(__i386__)
        if (RDRANDFlag)
        {
            uint32_t RDRANDValue = 0;
            asmv("1: rdrand %0; jnc 1b"
                 : "=r"(RDRANDValue));
            return RDRANDValue;
        }
#endif

        Seed = Seed * 1103515245 + 12345;
        return (uint32_t)(Seed / 65536) % __UINT16_MAX__;
    }

    uint64_t rand64()
    {
        int RDRANDFlag = 0;
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_AMD) == 0)
        {
#if defined(__amd64__)
            CPU::x64::AMD::CPUID0x1 cpuid1amd;
#elif defined(__i386__)
            CPU::x32::AMD::CPUID0x1 cpuid1amd;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1amd.EAX.raw), "=b"(cpuid1amd.EBX.raw), "=c"(cpuid1amd.ECX.raw), "=d"(cpuid1amd.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1amd.ECX.RDRAND;
        }
        if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_INTEL) == 0)
        {
#if defined(__amd64__)
            CPU::x64::Intel::CPUID0x1 cpuid1intel;
#elif defined(__i386__)
            CPU::x32::Intel::CPUID0x1 cpuid1intel;
#endif
#if defined(__amd64__) || defined(__i386__)
            asmv("cpuid"
                 : "=a"(cpuid1intel.EAX.raw), "=b"(cpuid1intel.EBX.raw), "=c"(cpuid1intel.ECX.raw), "=d"(cpuid1intel.EDX.raw)
                 : "a"(0x1));
#endif
            RDRANDFlag = cpuid1intel.ECX.RDRAND;
        }

        if (strcmp(CPU::Hypervisor(), x86_CPUID_VENDOR_TCG) == 0)
            RDRANDFlag = 0;

#if defined(__amd64__) || defined(__i386__)
        if (RDRANDFlag)
        {
            uint64_t RDRANDValue = 0;
            asmv("1: rdrand %0; jnc 1b"
                 : "=r"(RDRANDValue));
            return RDRANDValue;
        }
#endif

        Seed = Seed * 1103515245 + 12345;
        return (uint64_t)(Seed / 65536) % __UINT16_MAX__;
    }

    void ChangeSeed(uint64_t CustomSeed) { Seed = CustomSeed; }
}