/**
 * @file llprocessor.cpp
 * @brief Code to figure out the processor. Originally by Benjamin Jurke.
 *
 * $LicenseInfo:firstyear=2002&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#include "linden_common.h"
#include "llprocessor.h"
#include "llstring.h"
#include "stringize.h"
#include "llerror.h"

#include <iomanip>
//#include <memory>

#if LL_WINDOWS
#   include "llwin32headers.h"
#   define _interlockedbittestandset _renamed_interlockedbittestandset
#   define _interlockedbittestandreset _renamed_interlockedbittestandreset
#   include <intrin.h>
#   undef _interlockedbittestandset
#   undef _interlockedbittestandreset
#endif

#include "llsd.h"

class LLProcessorInfoImpl; // foward declaration for the mImpl;

namespace
{
    enum cpu_info
    {
        eBrandName = 0,
        eFrequency,
        eVendor,
        eStepping,
        eFamily,
        eExtendedFamily,
        eModel,
        eExtendedModel,
        eType,
        eBrandID,
        eFamilyName
    };


    const char* cpu_info_names[] =
    {
        "Processor Name",
        "Frequency",
        "Vendor",
        "Stepping",
        "Family",
        "Extended Family",
        "Model",
        "Extended Model",
        "Type",
        "Brand ID",
        "Family Name"
    };

    enum cpu_config
    {
        eMaxID,
        eMaxExtID,
        eCLFLUSHCacheLineSize,
        eAPICPhysicalID,
        eCacheLineSize,
        eL2Associativity,
        eCacheSizeK,
        eFeatureBits,
        eExtFeatureBits
    };

    const char* cpu_config_names[] =
    {
        "Max Supported CPUID level",
        "Max Supported Ext. CPUID level",
        "CLFLUSH cache line size",
        "APIC Physical ID",
        "Cache Line Size",
        "L2 Associativity",
        "Cache Size",
        "Feature Bits",
        "Ext. Feature Bits"
    };



    // *NOTE:Mani - this contains the elements we reference directly and extensions beyond the first 32.
    // The rest of the names are referenced by bit maks returned from cpuid.
    enum cpu_features
    {
        eSSE_Ext=25,
        eSSE2_Ext=26,

        eSSE3_Features=32,
        eMONTIOR_MWAIT=33,
        eCPLDebugStore=34,
        eThermalMonitor2=35,
        eAltivec=36,
        eSSE3S_Features = 37,
        eSSE4_1_Features = 38,
        eSSE4_2_Features = 39,
        eSSE4a_Features = 40,
    };

    const char* cpu_feature_names[] =
    {
        "x87 FPU On Chip",
        "Virtual-8086 Mode Enhancement",
        "Debugging Extensions",
        "Page Size Extensions",
        "Time Stamp Counter",
        "RDMSR and WRMSR Support",
        "Physical Address Extensions",
        "Machine Check Exception",
        "CMPXCHG8B Instruction",
        "APIC On Chip",
        "Unknown1",
        "SYSENTER and SYSEXIT",
        "Memory Type Range Registers",
        "PTE Global Bit",
        "Machine Check Architecture",
        "Conditional Move/Compare Instruction",
        "Page Attribute Table",
        "Page Size Extension",
        "Processor Serial Number",
        "CFLUSH Extension",
        "Unknown2",
        "Debug Store",
        "Thermal Monitor and Clock Ctrl",
        "MMX Technology",
        "FXSAVE/FXRSTOR",
        "SSE Extensions",
        "SSE2 Extensions",
        "Self Snoop",
        "Hyper-threading Technology",
        "Thermal Monitor",
        "Unknown4",
        "Pend. Brk. EN.", // 31 End of FeatureInfo bits

        "SSE3 New Instructions", // 32
        "MONITOR/MWAIT",
        "CPL Qualified Debug Store",
        "Thermal Monitor 2",

        "Altivec",
        "SSE3S Instructions",
        "SSE4.1 Instructions",
        "SSE4.2 Instructions",
        "SSE4a Instructions",
    };

    std::string intel_CPUFamilyName(int composed_family)
    {
        switch(composed_family)
        {
        case 3: return "Intel i386";
        case 4: return "Intel i486";
        case 5: return "Intel Pentium";
        case 6: return "Intel Pentium Pro/2/3, Core";
        case 7: return "Intel Itanium (IA-64)";
        case 0xF: return "Intel Pentium 4";
        case 0x10: return "Intel Itanium 2 (IA-64)";
        }
        return STRINGIZE("Intel <unknown 0x" << std::hex << composed_family << ">");
    }

    std::string amd_CPUFamilyName(int composed_family)
    {
        // https://en.wikipedia.org/wiki/List_of_AMD_CPU_microarchitectures
        // https://developer.amd.com/resources/developer-guides-manuals/
        switch(composed_family)
        {
        case 4: return "AMD 80486/5x86";
        case 5: return "AMD K5/K6";
        case 6: return "AMD K7";
        case 0xF: return "AMD K8";
        case 0x10: return "AMD K8L";
        case 0x12: return "AMD K10";
        case 0x14: return "AMD Bobcat";
        case 0x15: return "AMD Bulldozer";
        case 0x16: return "AMD Jaguar";
        case 0x17: return "AMD Zen/Zen+/Zen2";
        case 0x18: return "AMD Hygon Dhyana";
        case 0x19: return "AMD Zen 3";
        }
        return STRINGIZE("AMD <unknown 0x" << std::hex << composed_family << ">");
    }

    std::string compute_CPUFamilyName(const char* cpu_vendor, int family, int ext_family)
    {
        const char* intel_string = "GenuineIntel";
        const char* amd_string = "AuthenticAMD";
        if (LLStringUtil::startsWith(cpu_vendor, intel_string))
        {
            U32 composed_family = family + ext_family;
            return intel_CPUFamilyName(composed_family);
        }
        else if (LLStringUtil::startsWith(cpu_vendor, amd_string))
        {
            U32 composed_family = (family == 0xF)
                ? family + ext_family
                : family;
            return amd_CPUFamilyName(composed_family);
        }
        return STRINGIZE("Unrecognized CPU vendor <" << cpu_vendor << ">");
    }

} // end unnamed namespace

// The base class for implementations.
// Each platform should override this class.
class LLProcessorInfoImpl
{
public:
    LLProcessorInfoImpl()
    {
        mProcessorInfo["info"] = LLSD::emptyMap();
        mProcessorInfo["config"] = LLSD::emptyMap();
        mProcessorInfo["extension"] = LLSD::emptyMap();
    }
    virtual ~LLProcessorInfoImpl() {}

    F64 getCPUFrequency() const
    {
        return getInfo(eFrequency, 0).asReal();
    }

    bool hasSSE() const
    {
        return hasExtension(cpu_feature_names[eSSE_Ext]);
    }

    bool hasSSE2() const
    {
        return hasExtension(cpu_feature_names[eSSE2_Ext]);
    }

    bool hasSSE3() const
    {
        return hasExtension(cpu_feature_names[eSSE3_Features]);
    }

    bool hasSSE3S() const
    {
        return hasExtension(cpu_feature_names[eSSE3S_Features]);
    }

    bool hasSSE41() const
    {
        return hasExtension(cpu_feature_names[eSSE4_1_Features]);
    }

    bool hasSSE42() const
    {
        return hasExtension(cpu_feature_names[eSSE4_2_Features]);
    }

    bool hasSSE4a() const
    {
        return hasExtension(cpu_feature_names[eSSE4a_Features]);
    }

    bool hasAltivec() const
    {
        return hasExtension("Altivec");
    }

    std::string getCPUFamilyName() const { return getInfo(eFamilyName, "Unset family").asString(); }
    std::string getCPUBrandName() const { return getInfo(eBrandName, "Unset brand").asString(); }

    // This is virtual to support a different linux format.
    // *NOTE:Mani - I didn't want to screw up server use of this data...
    virtual std::string getCPUFeatureDescription() const
    {
        std::ostringstream out;
        out << std::endl << std::endl;
        out << "// CPU General Information" << std::endl;
        out << "//////////////////////////" << std::endl;
        out << "Processor Name:   " << getCPUBrandName() << std::endl;
        out << "Frequency:        " << getCPUFrequency() << " MHz" << std::endl;
        out << "Vendor:           " << getInfo(eVendor, "Unset vendor").asString() << std::endl;
        out << "Family:           " << getCPUFamilyName() << " (" << getInfo(eFamily, 0) << ")" << std::endl;
        out << "Extended family:  " << getInfo(eExtendedFamily, 0) << std::endl;
        out << "Model:            " << getInfo(eModel, 0) << std::endl;
        out << "Extended model:   " << getInfo(eExtendedModel, 0) << std::endl;
        out << "Type:             " << getInfo(eType, 0) << std::endl;
        out << "Brand ID:         " << getInfo(eBrandID, 0) << std::endl;
        out << std::endl;
        out << "// CPU Configuration" << std::endl;
        out << "//////////////////////////" << std::endl;

        // Iterate through the dictionary of configuration options.
        LLSD configs = mProcessorInfo["config"];
        for(LLSD::map_const_iterator cfgItr = configs.beginMap(); cfgItr != configs.endMap(); ++cfgItr)
        {
            out << cfgItr->first << " = " << cfgItr->second << std::endl;
        }
        out << std::endl;

        out << "// CPU Extensions" << std::endl;
        out << "//////////////////////////" << std::endl;

        for(LLSD::map_const_iterator itr = mProcessorInfo["extension"].beginMap(); itr != mProcessorInfo["extension"].endMap(); ++itr)
        {
            out << "  " << itr->first << std::endl;
        }
        return out.str();
    }

protected:
    void setInfo(cpu_info info_type, const LLSD& value)
    {
        setInfo(cpu_info_names[info_type], value);
    }
    LLSD getInfo(cpu_info info_type, const LLSD& defaultVal) const
    {
        return getInfo(cpu_info_names[info_type], defaultVal);
    }

    void setConfig(cpu_config config_type, const LLSD& value)
    {
        setConfig(cpu_config_names[config_type], value);
    }
    LLSD getConfig(cpu_config config_type, const LLSD& defaultVal) const
    {
        return getConfig(cpu_config_names[config_type], defaultVal);
    }

    void setExtension(const std::string& name) { mProcessorInfo["extension"][name] = "true"; }
    bool hasExtension(const std::string& name) const
    {
        return mProcessorInfo["extension"].has(name);
    }

private:
    void setInfo(const std::string& name, const LLSD& value) { mProcessorInfo["info"][name]=value; }
    LLSD getInfo(const std::string& name, const LLSD& defaultVal) const
    {
        if(mProcessorInfo["info"].has(name))
        {
            return mProcessorInfo["info"][name];
        }
        return defaultVal;
    }
    void setConfig(const std::string& name, const LLSD& value) { mProcessorInfo["config"][name]=value; }
    LLSD getConfig(const std::string& name, const LLSD& defaultVal) const
    {
        LLSD r = mProcessorInfo["config"].get(name);
        return r.isDefined() ? r : defaultVal;
    }

private:

    LLSD mProcessorInfo;
};


#ifdef LL_MSVC
// LL_MSVC and not LLWINDOWS because some of the following code
// uses the MSVC compiler intrinsics __cpuid() and __rdtsc().

// Delays for the specified amount of milliseconds
static void _Delay(unsigned int ms)
{
    LARGE_INTEGER freq, c1, c2;
    __int64 x;

    // Get High-Res Timer frequency
    if (!QueryPerformanceFrequency(&freq))
        return;

    // Convert ms to High-Res Timer value
    x = freq.QuadPart/1000*ms;

    // Get first snapshot of High-Res Timer value
    QueryPerformanceCounter(&c1);
    do
    {
        // Get second snapshot
        QueryPerformanceCounter(&c2);
    }while(c2.QuadPart-c1.QuadPart < x);
    // Loop while (second-first < x)
}

static F64 calculate_cpu_frequency(U32 measure_msecs)
{
    if(measure_msecs == 0)
    {
        return 0;
    }

    // After that we declare some vars and check the frequency of the high
    // resolution timer for the measure process.
    // If there"s no high-res timer, we exit.
    unsigned __int64 starttime, endtime, timedif, freq, start, end, dif;
    if (!QueryPerformanceFrequency((LARGE_INTEGER *) &freq))
    {
        return 0;
    }

    // Now we can init the measure process. We set the process and thread priority
    // to the highest available level (Realtime priority). Also we focus the
    // first processor in the multiprocessor system.
    HANDLE hProcess = GetCurrentProcess();
    HANDLE hThread = GetCurrentThread();
    unsigned long dwCurPriorityClass = GetPriorityClass(hProcess);
    int iCurThreadPriority = GetThreadPriority(hThread);
    DWORD_PTR dwProcessMask, dwSystemMask, dwNewMask = 1;
    GetProcessAffinityMask(hProcess, &dwProcessMask, &dwSystemMask);

    SetPriorityClass(hProcess, REALTIME_PRIORITY_CLASS);
    SetThreadPriority(hThread, THREAD_PRIORITY_TIME_CRITICAL);
    SetProcessAffinityMask(hProcess, dwNewMask);

    //// Now we call a CPUID to ensure, that all other prior called functions are
    //// completed now (serialization)
    //__asm cpuid
    int cpu_info[4] = {-1};
    __cpuid(cpu_info, 0);

    // We ask the high-res timer for the start time
    QueryPerformanceCounter((LARGE_INTEGER *) &starttime);

    // Then we get the current cpu clock and store it
    start = __rdtsc();

    // Now we wart for some msecs
    _Delay(measure_msecs);
    //  Sleep(uiMeasureMSecs);

    // We ask for the end time
    QueryPerformanceCounter((LARGE_INTEGER *) &endtime);

    // And also for the end cpu clock
    end = __rdtsc();

    // Now we can restore the default process and thread priorities
    SetProcessAffinityMask(hProcess, dwProcessMask);
    SetThreadPriority(hThread, iCurThreadPriority);
    SetPriorityClass(hProcess, dwCurPriorityClass);

    // Then we calculate the time and clock differences
    dif = end - start;
    timedif = endtime - starttime;

    // And finally the frequency is the clock difference divided by the time
    // difference.
    F64 frequency = (F64)dif / (((F64)timedif) / freq);

    // At last we just return the frequency that is also stored in the call
    // member var uqwFrequency - converted to MHz
    return frequency  / (F64)1000000;
}

// Windows implementation
class LLProcessorInfoWindowsImpl : public LLProcessorInfoImpl
{
public:
    LLProcessorInfoWindowsImpl()
    {
        getCPUIDInfo();
        setInfo(eFrequency, calculate_cpu_frequency(50));
    }

private:
    void getCPUIDInfo()
    {
        // http://msdn.microsoft.com/en-us/library/hskdteyh(VS.80).aspx

        // __cpuid with an InfoType argument of 0 returns the number of
        // valid Ids in cpu_info[0] and the CPU identification string in
        // the other three array elements. The CPU identification string is
        // not in linear order. The code below arranges the information
        // in a human readable form.
        int cpu_info[4] = {-1};
        __cpuid(cpu_info, 0);
        unsigned int ids = (unsigned int)cpu_info[0];
        setConfig(eMaxID, (S32)ids);

        char cpu_vendor[0x20];
        memset(cpu_vendor, 0, sizeof(cpu_vendor));
        *((int*)cpu_vendor) = cpu_info[1];
        *((int*)(cpu_vendor+4)) = cpu_info[3];
        *((int*)(cpu_vendor+8)) = cpu_info[2];
        setInfo(eVendor, cpu_vendor);
        std::string cmp_vendor(cpu_vendor);
        bool is_amd = false;
        if (cmp_vendor == "AuthenticAMD")
        {
            is_amd = true;
        }

        // Get the information associated with each valid Id
        for(unsigned int i=0; i<=ids; ++i)
        {
            __cpuid(cpu_info, i);

            // Interpret CPU feature information.
            if  (i == 1)
            {
                setInfo(eStepping, cpu_info[0] & 0xf);
                setInfo(eModel, (cpu_info[0] >> 4) & 0xf);
                int family = (cpu_info[0] >> 8) & 0xf;
                setInfo(eFamily, family);
                setInfo(eType, (cpu_info[0] >> 12) & 0x3);
                setInfo(eExtendedModel, (cpu_info[0] >> 16) & 0xf);
                int ext_family = (cpu_info[0] >> 20) & 0xff;
                setInfo(eExtendedFamily, ext_family);
                setInfo(eBrandID, cpu_info[1] & 0xff);

                setInfo(eFamilyName, compute_CPUFamilyName(cpu_vendor, family, ext_family));

                setConfig(eCLFLUSHCacheLineSize, ((cpu_info[1] >> 8) & 0xff) * 8);
                setConfig(eAPICPhysicalID, (cpu_info[1] >> 24) & 0xff);

                if(cpu_info[2] & 0x1)
                {
                    setExtension(cpu_feature_names[eSSE3_Features]);
                }

                if(cpu_info[2] & 0x8)
                {
                    // intel specific SSE3 suplements
                    setExtension(cpu_feature_names[eMONTIOR_MWAIT]);
                }

                if(cpu_info[2] & 0x10)
                {
                    setExtension(cpu_feature_names[eCPLDebugStore]);
                }

                if(cpu_info[2] & 0x100)
                {
                    setExtension(cpu_feature_names[eThermalMonitor2]);
                }

                if (cpu_info[2] & 0x200)
                {
                    setExtension(cpu_feature_names[eSSE3S_Features]);
                }

                if (cpu_info[2] & 0x80000)
                {
                    setExtension(cpu_feature_names[eSSE4_1_Features]);
                }

                if (cpu_info[2] & 0x100000)
                {
                    setExtension(cpu_feature_names[eSSE4_2_Features]);
                }

                unsigned int feature_info = (unsigned int) cpu_info[3];
                for(unsigned int index = 0, bit = 1; index < eSSE3_Features; ++index, bit <<= 1)
                {
                    if(feature_info & bit)
                    {
                        setExtension(cpu_feature_names[index]);
                    }
                }
            }
        }

        // Calling __cpuid with 0x80000000 as the InfoType argument
        // gets the number of valid extended IDs.
        __cpuid(cpu_info, 0x80000000);
        unsigned int ext_ids = cpu_info[0];
        setConfig(eMaxExtID, 0);

        char cpu_brand_string[0x40];
        memset(cpu_brand_string, 0, sizeof(cpu_brand_string));

        // Get the information associated with each extended ID.
        for(unsigned int i=0x80000000; i<=ext_ids; ++i)
        {
            __cpuid(cpu_info, i);

            // Interpret CPU brand string and cache information.
            if (i == 0x80000001)
            {
                if (is_amd)
                {
                    setExtension(cpu_feature_names[eSSE4a_Features]);
                }
            }
            else if (i == 0x80000002)
            {
                memcpy(cpu_brand_string, cpu_info, sizeof(cpu_info));
            }
            else if  (i == 0x80000003)
                memcpy(cpu_brand_string + 16, cpu_info, sizeof(cpu_info));
            else if  (i == 0x80000004)
            {
                memcpy(cpu_brand_string + 32, cpu_info, sizeof(cpu_info));
                setInfo(eBrandName, cpu_brand_string);
            }
            else if  (i == 0x80000006)
            {
                setConfig(eCacheLineSize, cpu_info[2] & 0xff);
                setConfig(eL2Associativity, (cpu_info[2] >> 12) & 0xf);
                setConfig(eCacheSizeK, (cpu_info[2] >> 16) & 0xffff);
            }
        }
    }
};

#elif LL_DARWIN

#include <mach/machine.h>
#include <sys/sysctl.h>

class LLProcessorInfoDarwinImpl : public LLProcessorInfoImpl
{
public:
    LLProcessorInfoDarwinImpl()
    {
        getCPUIDInfo();
        uint64_t frequency = getSysctlInt64("hw.cpufrequency");
        if (!frequency) {
            auto tbfrequency = getSysctlInt64("hw.tbfrequency");
            struct clockinfo clockrate;
            auto clockrate_len = sizeof(clockrate);
            if (!sysctlbyname("kern.clockrate", &clockrate, &clockrate_len, NULL, 0))
                frequency = tbfrequency * clockrate.hz;
        }
        setInfo(eFrequency, (F64)frequency  / (F64)1000000);
    }

    virtual ~LLProcessorInfoDarwinImpl() {}

private:
    int getSysctlInt(const char* name)
    {
        int result = 0;
        size_t len = sizeof(int);
        int error = sysctlbyname(name, (void*)&result, &len, NULL, 0);
        return error == -1 ? 0 : result;
    }

    uint64_t getSysctlInt64(const char* name)
    {
        uint64_t value = 0;
        size_t size = sizeof(value);
        int result = sysctlbyname(name, (void*)&value, &size, NULL, 0);
        if ( result == 0 )
        {
            if ( size == sizeof( uint64_t ) )
                ;
            else if ( size == sizeof( uint32_t ) )
                value = (uint64_t)(( uint32_t *)&value);
            else if ( size == sizeof( uint16_t ) )
                value =  (uint64_t)(( uint16_t *)&value);
            else if ( size == sizeof( uint8_t ) )
                value =  (uint64_t)(( uint8_t *)&value);
            else
            {
                LL_WARNS() << "Unknown type returned from sysctl" << LL_ENDL;
            }
        }

        return result == -1 ? 0 : value;
    }

    void getCPUIDInfo()
    {
        size_t len = 0;

        char cpu_brand_string[0x40];
        len = sizeof(cpu_brand_string);
        memset(cpu_brand_string, 0, len);
        sysctlbyname("machdep.cpu.brand_string", (void*)cpu_brand_string, &len, NULL, 0);
        cpu_brand_string[0x3f] = 0;
        setInfo(eBrandName, cpu_brand_string);

        char cpu_vendor[0x20];
        len = sizeof(cpu_vendor);
        memset(cpu_vendor, 0, len);
        sysctlbyname("machdep.cpu.vendor", (void*)cpu_vendor, &len, NULL, 0);
        cpu_vendor[0x1f] = 0;
        // M series CPUs don't provide this field so if empty, just fall back to Apple.
        setInfo(eVendor, (cpu_vendor[0] != '\0') ? cpu_vendor : "Apple");

        setInfo(eStepping, getSysctlInt("machdep.cpu.stepping"));
        setInfo(eModel, getSysctlInt("machdep.cpu.model"));
        int family = getSysctlInt("machdep.cpu.family");
        int ext_family = getSysctlInt("machdep.cpu.extfamily");
        setInfo(eFamily, family);
        setInfo(eExtendedFamily, ext_family);
        setInfo(eFamilyName, compute_CPUFamilyName(cpu_vendor, family, ext_family));
        setInfo(eExtendedModel, getSysctlInt("machdep.cpu.extmodel"));
        setInfo(eBrandID, getSysctlInt("machdep.cpu.brand"));
        setInfo(eType, 0); // ? where to find this?

        //setConfig(eCLFLUSHCacheLineSize, ((cpu_info[1] >> 8) & 0xff) * 8);
        //setConfig(eAPICPhysicalID, (cpu_info[1] >> 24) & 0xff);
        setConfig(eCacheLineSize, getSysctlInt("machdep.cpu.cache.linesize"));
        setConfig(eL2Associativity, getSysctlInt("machdep.cpu.cache.L2_associativity"));
        setConfig(eCacheSizeK, getSysctlInt("machdep.cpu.cache.size"));

        uint64_t feature_info = getSysctlInt64("machdep.cpu.feature_bits");
        S32 *feature_infos = (S32*)(&feature_info);

        setConfig(eFeatureBits, feature_infos[0]);

        for(unsigned int index = 0, bit = 1; index < eSSE3_Features; ++index, bit <<= 1)
        {
            if(feature_info & bit)
            {
                setExtension(cpu_feature_names[index]);
            }
        }

        // *NOTE:Mani - I didn't find any docs that assure me that machdep.cpu.feature_bits will always be
        // The feature bits I think it is. Here's a test:
#ifndef LL_RELEASE_FOR_DOWNLOAD
    #if defined(__i386__) && defined(__PIC__)
            /* %ebx may be the PIC register.  */
        #define __cpuid(level, a, b, c, d)          \
        __asm__ ("xchgl\t%%ebx, %1\n\t"         \
                "cpuid\n\t"                 \
                "xchgl\t%%ebx, %1\n\t"          \
                : "=a" (a), "=r" (b), "=c" (c), "=d" (d)    \
                : "0" (level))
    #else
        #define __cpuid(level, a, b, c, d)          \
        __asm__ ("cpuid\n\t"                    \
                 : "=a" (a), "=b" (b), "=c" (c), "=d" (d)   \
                 : "0" (level))
    #endif

#if __i386__ || __x86_64__
        unsigned int eax, ebx, ecx, edx;
        __cpuid(0x1, eax, ebx, ecx, edx);
        if(feature_infos[0] != (S32)edx)
        {
            LL_WARNS() << "machdep.cpu.feature_bits doesn't match expected cpuid result!" << LL_ENDL;
        }
#endif // __i386__ || __x86_64__
#endif // LL_RELEASE_FOR_DOWNLOAD


        uint64_t ext_feature_info = getSysctlInt64("machdep.cpu.extfeature_bits");
        S32 *ext_feature_infos = (S32*)(&ext_feature_info);
        setConfig(eExtFeatureBits, ext_feature_infos[0]);


        char cpu_features[1024];
        len = sizeof(cpu_features);
        memset(cpu_features, 0, len);
        sysctlbyname("machdep.cpu.features", (void*)cpu_features, &len, NULL, 0);

        std::string cpu_features_str(cpu_features);
        cpu_features_str = " " + cpu_features_str + " ";

        if (cpu_features_str.find(" SSE3 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE3_Features]);
        }

        if (cpu_features_str.find(" SSSE3 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE3S_Features]);
        }

        if (cpu_features_str.find(" SSE4.1 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE4_1_Features]);
        }

        if (cpu_features_str.find(" SSE4.2 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE4_2_Features]);
        }

        if (cpu_features_str.find(" SSE4A ") != std::string::npos)
        {
            // Not supposed to happen?
            setExtension(cpu_feature_names[eSSE4a_Features]);
        }
    }
};

#elif __FreeBSD__

#include <sys/sysctl.h>
class LLProcessorInfoFreeBSDImpl : public LLProcessorInfoImpl
{
public:
	LLProcessorInfoFreeBSDImpl()
	{
		size_t len = 0;
		using std::string;

		char cpu_brand_string[0x40];
		len = sizeof cpu_brand_string;
		memset(cpu_brand_string, '\0', len);
		sysctlbyname("hw.model", (void *)cpu_brand_string, &len,
				NULL, 0);
		cpu_brand_string[0x3f] = '\0';
		setInfo(eBrandName, cpu_brand_string);

		uint64_t cpu_frequency = 0;
		len = sizeof cpu_frequency;
		sysctlbyname("hw.clockrate", (void *)&cpu_frequency, &len,
				NULL, 0);
		setInfo(eFrequency, (F64)cpu_frequency);

		auto dmesgboot = LLFile::fopen("/var/run/dmesg.boot", "rb");
		std::ostringstream s;
		if (dmesgboot) {
			char line[MAX_STRING];
			memset(line, 0, MAX_STRING);
			while (fgets(line, MAX_STRING, dmesgboot)) {
				line[strlen(line) - 1] = ' ';
				s << line;
				s << std::endl;
			}
			fclose(dmesgboot);
			s << std::endl;
		}

		auto dmesgboot_str = s.str();
		int decimal;

		auto idx1 = dmesgboot_str.find_first_of("\"") + 1;
		auto idx2 = (idx1 != string::npos)
			? dmesgboot_str.find_first_of("\"", idx1)
			: string::npos;
		auto vendor = dmesgboot_str.substr(idx1, idx2 - idx1);
		if (vendor.length() > 0) setInfo(eVendor, vendor.c_str());

		idx1 = dmesgboot_str.find_first_of("=", idx2);
		idx1 = dmesgboot_str.find_first_of("=", idx1 + 1) + 3;
		idx2 = dmesgboot_str.find_first_of(" ", idx1);
		auto family = dmesgboot_str.substr(idx1, idx2 - idx1);
		std::istringstream(family) >> std::hex >> decimal;
		setInfo(eFamily, decimal);
		setInfo(eFamilyName, compute_CPUFamilyName(vendor.c_str(),
					decimal, 0));

		idx1 = dmesgboot_str.find_first_of("=", idx2) + 3;
		idx2 = dmesgboot_str.find_first_of(" ", idx1);
		auto model = dmesgboot_str.substr(idx1, idx2 - idx1);
		std::istringstream(model) >> std::hex >> decimal;
		setInfo(eModel, decimal);

		idx1 = dmesgboot_str.find_first_of("=", idx2) + 1;
		idx2 = dmesgboot_str.find_first_of("\n", idx1);
		auto stepping = dmesgboot_str.substr(idx1, idx2 - idx1);
		setInfo(eStepping, std::stoi(stepping));

		if (dmesgboot_str.find(",SSE,") != string::npos)
			setExtension(cpu_feature_names[eSSE_Ext]);

		if (dmesgboot_str.find(",SSE2,") != string::npos)
			setExtension(cpu_feature_names[eSSE2_Ext]);

		if (dmesgboot_str.find("<SSE3,") != string::npos)
			setExtension(cpu_feature_names[eSSE3_Features]);

		if (dmesgboot_str.find(",SSSE3,") != string::npos)
			setExtension(cpu_feature_names[eSSE3S_Features]);

		if (dmesgboot_str.find(",SSE4.1,") != string::npos)
			setExtension(cpu_feature_names[eSSE4_1_Features]);

		if (dmesgboot_str.find(",SSE4.2,") != string::npos)
			setExtension(cpu_feature_names[eSSE4_2_Features]);

		if (dmesgboot_str.find(",SSE4A,") != string::npos)
			setExtension(cpu_feature_names[eSSE4a_Features]);
	}

	virtual ~LLProcessorInfoFreeBSDImpl() {}
};

#elif LL_LINUX

// *NOTE:Mani - eww, macros! srry.
#define LLPI_SET_INFO_STRING(llpi_id, cpuinfo_id) \
        if (!cpuinfo[cpuinfo_id].empty()) \
        { setInfo(llpi_id, cpuinfo[cpuinfo_id]);}

#define LLPI_SET_INFO_INT(llpi_id, cpuinfo_id) \
        {\
            S32 result; \
            if (!cpuinfo[cpuinfo_id].empty() \
                && LLStringUtil::convertToS32(cpuinfo[cpuinfo_id], result)) \
            { setInfo(llpi_id, result);} \
        }

const char CPUINFO_FILE[] = "/proc/cpuinfo";

class LLProcessorInfoLinuxImpl : public LLProcessorInfoImpl {
public:
    LLProcessorInfoLinuxImpl() {
        get_proc_cpuinfo();
    }

    virtual ~LLProcessorInfoLinuxImpl() {}

private:

    F64 getCPUMaxMHZ()
    {
        // Nicky: We just look into cpu0. In theory we could iterate over all cores
        // "/sys/devices/system/cpu/cpu*/cpufreq/cpuinfo_max_freq"
        // But those should not fluctuate that much?
        std::ifstream fIn { "/sys/devices/system/cpu/cpu0/cpufreq/cpuinfo_max_freq" };

        if( !fIn.is_open() )
            return 0.0;

        std::string strLine;
        fIn >> strLine;
        if( strLine.empty() )
            return 0.0l;

        F64 mhz {};
        if( !LLStringUtil::convertToF64(strLine, mhz ) )
            return 0.0;

        mhz = mhz / 1000.0;
        return mhz;
    }

    void get_proc_cpuinfo()
    {
        std::map< std::string, std::string > cpuinfo;
        LLFILE* cpuinfo_fp = LLFile::fopen(CPUINFO_FILE, "rb");
        if(cpuinfo_fp)
        {
            char line[MAX_STRING];
            memset(line, 0, MAX_STRING);
            while(fgets(line, MAX_STRING, cpuinfo_fp))
            {
                // /proc/cpuinfo on Linux looks like:
                // name\t*: value\n
                char* tabspot = strchr( line, '\t' );
                if (tabspot == NULL)
                    continue;
                char* colspot = strchr( tabspot, ':' );
                if (colspot == NULL)
                    continue;
                char* spacespot = strchr( colspot, ' ' );
                if (spacespot == NULL)
                    continue;
                char* nlspot = strchr( line, '\n' );
                if (nlspot == NULL)
                    nlspot = line + strlen( line ); // Fallback to terminating NUL
                std::string linename( line, tabspot );
                std::string llinename(linename);
                LLStringUtil::toLower(llinename);
                std::string lineval( spacespot + 1, nlspot );
                    cpuinfo[ llinename ] = lineval;
            }
            fclose(cpuinfo_fp);
        }
# if LL_X86

        F64 mhzFromSys = getCPUMaxMHZ();
        F64 mhzFromProc {};
        if( !LLStringUtil::convertToF64(cpuinfo["cpu mhz"], mhzFromProc ) )
            mhzFromProc = 0.0;
        if (mhzFromSys > 1.0 && mhzFromSys > mhzFromProc )
        {
            setInfo( eFrequency, mhzFromSys );
        }
        else if (  200.0 < mhzFromProc && mhzFromProc < 10000.0)
        {
            setInfo(eFrequency,(F64)(mhzFromProc));
        }

        LLPI_SET_INFO_STRING(eBrandName, "model name");
        LLPI_SET_INFO_STRING(eVendor, "vendor_id");

        LLPI_SET_INFO_INT(eStepping, "stepping");
        LLPI_SET_INFO_INT(eModel, "model");


        S32 family{};
        if (!cpuinfo["cpu family"].empty()
            && LLStringUtil::convertToS32(cpuinfo["cpu family"], family))
        {
            setInfo(eFamily, family);
        }

        setInfo(eFamilyName, compute_CPUFamilyName(cpuinfo["vendor_id"].c_str(), family, 0));

        // setInfo(eExtendedModel, getSysctlInt("machdep.cpu.extmodel"));
        // setInfo(eBrandID, getSysctlInt("machdep.cpu.brand"));
        // setInfo(eType, 0); // ? where to find this?

        //setConfig(eCLFLUSHCacheLineSize, ((cpu_info[1] >> 8) & 0xff) * 8);
        //setConfig(eAPICPhysicalID, (cpu_info[1] >> 24) & 0xff);
        //setConfig(eCacheLineSize, getSysctlInt("machdep.cpu.cache.linesize"));
        //setConfig(eL2Associativity, getSysctlInt("machdep.cpu.cache.L2_associativity"));
        //setConfig(eCacheSizeK, getSysctlInt("machdep.cpu.cache.size"));

        // Read extensions
        std::string flags = " " + cpuinfo["flags"] + " ";
        LLStringUtil::toLower(flags);

        if( flags.find( " sse " ) != std::string::npos )
        {
            setExtension(cpu_feature_names[eSSE_Ext]);
        }

        if( flags.find( " sse2 " ) != std::string::npos )
        {
            setExtension(cpu_feature_names[eSSE2_Ext]);
        }

        if (flags.find(" pni ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE3_Features]);
        }

        if (flags.find(" ssse3 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE3S_Features]);
        }

        if (flags.find(" sse4_1 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE4_1_Features]);
        }

        if (flags.find(" sse4_2 ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE4_2_Features]);
        }

        if (flags.find(" sse4a ") != std::string::npos)
        {
            setExtension(cpu_feature_names[eSSE4a_Features]);
        }

# endif // LL_X86
    }

    std::string getCPUFeatureDescription() const
    {
        std::ostringstream s;

        // *NOTE:Mani - This is for linux only.
        LLFILE* cpuinfo = LLFile::fopen(CPUINFO_FILE, "rb");
        if(cpuinfo)
        {
            char line[MAX_STRING];
            memset(line, 0, MAX_STRING);
            while(fgets(line, MAX_STRING, cpuinfo))
            {
                line[strlen(line)-1] = ' ';
                s << line;
                s << std::endl;
            }
            fclose(cpuinfo);
            s << std::endl;
        }
        else
        {
            s << "Unable to collect processor information" << std::endl;
        }
        return s.str();
    }

};


#endif // LL_MSVC elif LL_DARWIN elif LL_LINUX

//////////////////////////////////////////////////////
// Interface definition
LLProcessorInfo::LLProcessorInfo() : mImpl(NULL)
{
    // *NOTE:Mani - not thread safe.
    if(!mImpl)
    {
#ifdef LL_MSVC
        static LLProcessorInfoWindowsImpl the_impl;
        mImpl = &the_impl;
#elif LL_DARWIN
        static LLProcessorInfoDarwinImpl the_impl;
        mImpl = &the_impl;
#elif __FreeBSD__
        static LLProcessorInfoFreeBSDImpl the_impl;
        mImpl = &the_impl;
#else
        static LLProcessorInfoLinuxImpl the_impl;
        mImpl = &the_impl;
#endif // LL_MSVC
    }
}


LLProcessorInfo::~LLProcessorInfo() {}
F64MegahertzImplicit LLProcessorInfo::getCPUFrequency() const { return mImpl->getCPUFrequency(); }
bool LLProcessorInfo::hasSSE() const { return mImpl->hasSSE(); }
bool LLProcessorInfo::hasSSE2() const { return mImpl->hasSSE2(); }
bool LLProcessorInfo::hasSSE3() const { return mImpl->hasSSE3(); }
bool LLProcessorInfo::hasSSE3S() const { return mImpl->hasSSE3S(); }
bool LLProcessorInfo::hasSSE41() const { return mImpl->hasSSE41(); }
bool LLProcessorInfo::hasSSE42() const { return mImpl->hasSSE42(); }
bool LLProcessorInfo::hasSSE4a() const { return mImpl->hasSSE4a(); }
bool LLProcessorInfo::hasAltivec() const { return mImpl->hasAltivec(); }
std::string LLProcessorInfo::getCPUFamilyName() const { return mImpl->getCPUFamilyName(); }
std::string LLProcessorInfo::getCPUBrandName() const { return mImpl->getCPUBrandName(); }
std::string LLProcessorInfo::getCPUFeatureDescription() const { return mImpl->getCPUFeatureDescription(); }