SystemInfo.cpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2006-2021 Istituto Italiano di Tecnologia (IIT)
 * SPDX-FileCopyrightText: 2006-2010 RobotCub Consortium
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#include <yarp/os/SystemInfo.h>
 
#include <yarp/os/Os.h>
#include <yarp/os/impl/LogComponent.h>
 
#include <cstdio>
#include <cstdlib>
#include <cstring>
 
using namespace yarp::os;
 
#if defined(__linux__)
#    include <arpa/inet.h>
#    include <ifaddrs.h>
#    include <linux/if.h>
#    include <netinet/in.h>
#    include <pwd.h>
#    include <sys/ioctl.h>
#    include <sys/socket.h>
#    include <sys/statvfs.h>
#    include <sys/types.h>
#    include <sys/utsname.h>
#    include <unistd.h>
 
extern char** environ;
 
#elif defined(__APPLE__)
#    include <mach/mach.h>
#    include <pwd.h>
#    include <sstream>
#    include <sys/mount.h>
#    include <sys/param.h>
#    include <sys/sysctl.h>
#    include <sys/types.h>
#    include <unistd.h>
 
namespace {
 
YARP_OS_LOG_COMPONENT(SYSTEMINFO, "yarp.os.SystemInfo")
 
} // namespace
 
#endif
 
#if defined(_WIN32)
#    include <Lmcons.h>
#    include <comdef.h> // for using bstr_t class
#    include <psapi.h>
#    include <shlobj.h>
#    include <windows.h>
//#define _WIN32_DCOM
#    include <wbemidl.h>
#    if (defined(WINVER)) && (WINVER >= 0x0502)
#        include <pdh.h>
#        include <pdhmsg.h>
#        pragma comment(lib, "pdh.lib")
#        pragma comment(lib, "psapi.lib")    // for get proocess name for pid
#        pragma comment(lib, "wbemuuid.lib") // for get process arguments from pid
#    endif
 
#    include <yarp/os/PeriodicThread.h>
#    include <yarp/os/Semaphore.h>
 
#    include <vector>
 
 
static void enableCpuLoadCollector();
static void disableCpuLoadCollector();
 
#endif
 
#if defined(__linux__)
SystemInfo::capacity_t getMemEntry(const char* tag, const char* bufptr)
{
    char* tail;
    SystemInfo::capacity_t retval;
    size_t len = strlen(tag);
    while (bufptr != nullptr) {
        if (*bufptr == '\n') {
            bufptr++;
        }
        if (strncmp(tag, bufptr, len) == 0) {
            retval = strtol(bufptr + len, &tail, 10);
            if (tail == bufptr + len) {
                return -1;
            }
            return retval;
        }
        bufptr = strchr(bufptr, '\n');
    }
    return -1;
}
 
 
bool getCpuEntry(const char* tag, const char* buff, std::string& value)
{
    if (strlen(buff) <= strlen(tag)) {
        return false;
    }
 
    if (strncmp(buff, tag, strlen(tag)) != 0) {
        return false;
    }
 
    const char* pos1 = strchr(buff, ':');
    if (pos1 == nullptr) {
        return false;
    }
 
    while ((*pos1 != '\0') && ((*pos1 == ' ') || (*pos1 == ':') || (*pos1 == '\t'))) {
        pos1++;
    }
    const char* pos2 = buff + strlen(buff) - 1;
    while ((*pos2 != ':') && ((*pos2 == ' ') || (*pos2 == '\n'))) {
        pos2--;
    }
    if (pos2 < pos1) {
        return false;
    }
    value = std::string(pos1, pos2 - pos1 + 1);
    return true;
}
#endif
 
 
#if defined(_WIN32)
class CpuLoadCollector : public yarp::os::PeriodicThread
{
public:
    CpuLoadCollector() :
            PeriodicThread(5.000, ShouldUseSystemClock::Yes)
    {
        firstRun = true;
        load.cpuLoad1 = 0.0;
        load.cpuLoad5 = 0.0;
        load.cpuLoad15 = 0.0;
        load.cpuLoadInstant = (int)0;
    }
 
    ~CpuLoadCollector()
    {
    }
 
    void run()
    {
        sem.wait();
        load.cpuLoadInstant = (int)phdCpuLoad();
        samples.push_back(load.cpuLoadInstant);
        if (samples.size() > 180)
            samples.erase(samples.begin());
 
        std::vector<int>::reverse_iterator rti;
        int sum = 0;
        int n = 0;
        for (rti = samples.rbegin(); rti < samples.rend(); ++rti) {
            sum += (*rti);
            n++;
            // every 1min
            if (n < 12)
                load.cpuLoad1 = (double)(sum / n) / 100.0;
            // every 5min
            if (n < 60)
                load.cpuLoad5 = (double)(sum / n) / 100.0;
            // every 15min
            load.cpuLoad15 = (double)(sum / n) / 100.0;
        }
        sem.post();
    }
 
    SystemInfo::LoadInfo getCpuLoad()
    {
        sem.wait();
        SystemInfo::LoadInfo ld = load;
        sem.post();
        return ld;
    }
 
    //bool threadInit()
    //void threadRelease()
private:
#    if (defined(WINVER)) && (WINVER >= 0x0502)
    double phdCpuLoad()
    {
        DWORD ret;
        if (firstRun) {
            phdStatus = PdhOpenQuery(nullptr, 0, &hPhdQuery);
            if (phdStatus != ERROR_SUCCESS)
                return 0;
 
            PdhAddCounter(hPhdQuery, TEXT("\\Processor(_Total)\\% Processor Time"), 0, &phdCounter);
            PdhCollectQueryData(hPhdQuery);
            firstRun = false;
            return 0;
        }
 
        phdStatus = PdhCollectQueryData(hPhdQuery);
        if (phdStatus != ERROR_SUCCESS)
            return 0;
        phdStatus = PdhGetFormattedCounterValue(phdCounter,
                                                PDH_FMT_DOUBLE | PDH_FMT_NOCAP100,
                                                &ret,
                                                &phdFmtValue);
        if (phdStatus != ERROR_SUCCESS)
            return 0;
        return phdFmtValue.doubleValue;
    }
#    else
    double phdCpuLoad()
    {
        return 0.0;
    }
#    endif
 
private:
#    if (defined(WINVER)) && (WINVER >= 0x0502)
    PDH_STATUS phdStatus;
    HQUERY hPhdQuery;
    PDH_FMT_COUNTERVALUE phdFmtValue;
    HCOUNTER phdCounter;
#    endif
    bool firstRun;
    SystemInfo::LoadInfo load;
    std::vector<int> samples;
    yarp::os::Semaphore sem;
};
 
static CpuLoadCollector* globalLoadCollector = nullptr;
 
void enableCpuLoadCollector()
{
    if (globalLoadCollector == nullptr) {
        globalLoadCollector = new CpuLoadCollector();
        globalLoadCollector->start();
    }
}
 
void disableCpuLoadCollector()
{
    if (globalLoadCollector) {
        globalLoadCollector->stop();
        delete globalLoadCollector;
        globalLoadCollector = nullptr;
    }
}
 
#endif
 
 
SystemInfo::MemoryInfo SystemInfo::getMemoryInfo()
{
    MemoryInfo memory;
    memory.totalSpace = 0;
    memory.freeSpace = 0;
 
#if defined(_WIN32)
    MEMORYSTATUSEX statex;
    statex.dwLength = sizeof(statex);
    if (GlobalMemoryStatusEx(&statex)) {
        memory.totalSpace = (capacity_t)(statex.ullTotalPhys / 1048576); //in Mb
        memory.freeSpace = (capacity_t)(statex.ullAvailPhys / 1048576);  //in Mb
    }
#endif
 
#if defined(__linux__)
    char buffer[128];
    FILE* procmem = fopen("/proc/meminfo", "r");
    if (procmem != nullptr) {
        while (fgets(buffer, 128, procmem) != nullptr) {
            capacity_t ret;
            if ((ret = getMemEntry("MemTotal:", buffer)) > 0) {
                memory.totalSpace = ret / 1024;
            }
 
            if ((ret = getMemEntry("MemFree:", buffer)) > 0) {
                memory.freeSpace = ret / 1024;
            }
        }
        fclose(procmem);
    }
#elif defined(__APPLE__)
 
    vm_size_t page_size;
    mach_port_t mach_port;
    mach_msg_type_number_t count;
    vm_statistics64_data_t vm_stats;
 
    mach_port = mach_host_self();
    count = HOST_VM_INFO64_COUNT;
    if (KERN_SUCCESS == host_page_size(mach_port, &page_size) && KERN_SUCCESS == host_statistics64(mach_port, HOST_VM_INFO, (host_info64_t)&vm_stats, &count)) {
        //These seem to return the # of pages
        natural_t activePages = vm_stats.active_count + vm_stats.wire_count;
        natural_t inactivePages = vm_stats.inactive_count + vm_stats.free_count;
        natural_t totalPages = activePages + inactivePages;
 
        int64_t total = totalPages * page_size;
        int64_t freeSpace = inactivePages * page_size;
 
        memory.totalSpace = total / 1024;
        memory.freeSpace = freeSpace / 1024;
    }
 
#endif
 
 
    return memory;
}
 
 
SystemInfo::StorageInfo SystemInfo::getStorageInfo()
{
    StorageInfo storage;
    storage.totalSpace = 0;
    storage.freeSpace = 0;
 
#if defined(_WIN32)
 
    DWORD dwSectorsPerCluster = 0, dwBytesPerSector = 0;
    DWORD dwFreeClusters = 0, dwTotalClusters = 0;
    std::string strHome = getUserInfo().homeDir;
    if (strHome.empty())
        strHome = "C:\\";
    if (GetDiskFreeSpaceA(strHome.c_str(), &dwSectorsPerCluster, &dwBytesPerSector, &dwFreeClusters, &dwTotalClusters)) {
        storage.freeSpace = (capacity_t)((dwFreeClusters / 1048576.0) * dwSectorsPerCluster * dwBytesPerSector);
        storage.totalSpace = (capacity_t)((dwTotalClusters) / 1048576.0 * dwSectorsPerCluster * dwBytesPerSector);
    }
#endif
 
#if defined(__linux__)
    std::string strHome = getUserInfo().homeDir;
    if (strHome.empty()) {
        strHome = "/home";
    }
 
    struct statvfs vfs;
    if (statvfs(strHome.c_str(), &vfs) == 0) {
        storage.totalSpace = (int)(vfs.f_blocks * vfs.f_bsize / (1048576)); // in MB
        storage.freeSpace = (int)(vfs.f_bavail * vfs.f_bsize / (1048576));  // in MB
    }
 
#endif
 
#if defined(__APPLE__)
    std::string strHome = getUserInfo().homeDir;
    if (strHome.empty())
        strHome = "/";
 
    struct statfs vfs;
    if (statfs(strHome.c_str(), &vfs) == 0) {
        storage.totalSpace = (int)(vfs.f_blocks * vfs.f_bsize / (1048576)); // in MB
        storage.freeSpace = (int)(vfs.f_bavail * vfs.f_bsize / (1048576));  // in MB
    }
#endif
 
    return storage;
}
 
/*
SystemInfo::NetworkInfo SystemInfo::getNetworkInfo()
{
    NetworkInfo network;
 
#if defined(__linux__)
 
    struct ifaddrs * ifAddrStruct=nullptr;
    struct ifaddrs * ifa=nullptr;
    void * tmpAddrPtr=nullptr;
 
    getifaddrs(&ifAddrStruct);
    for (ifa = ifAddrStruct; ifa != nullptr; ifa = ifa->ifa_next)
    {
        if (ifa ->ifa_addr->sa_family == AF_INET)
        {
            // is a valid IP4 Address
            tmpAddrPtr = &((struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
            char addressBuffer[INET_ADDRSTRLEN];
            const char* ret = inet_ntop(AF_INET, tmpAddrPtr, addressBuffer, INET_ADDRSTRLEN);
            if (ret && (strcmp(ifa->ifa_name, "eth0") == 0))
                network.ip4 = addressBuffer;
        }
        else if (ifa->ifa_addr->sa_family == AF_INET6)
        {
            // is a valid IP6 Address
            tmpAddrPtr=&((struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
            char addressBuffer[INET6_ADDRSTRLEN];
            const char* ret = inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
            if (ret && (strcmp(ifa->ifa_name, "eth0") == 0))
                network.ip6 = addressBuffer;
        }
    }
 
    if (ifAddrStruct)
        freeifaddrs(ifAddrStruct);
 
    // getting mac addrress
    struct ifreq ifr;
    struct ifreq *IFR;
    struct ifconf ifc;
    char buf[1024];
    int s;
    bool found = false;
 
    if ( (s = socket(AF_INET, SOCK_DGRAM, 0)) == -1 )
        return network;
 
    ifc.ifc_len = sizeof(buf);
    ifc.ifc_buf = buf;
    ioctl(s, SIOCGIFCONF, &ifc);
    IFR = ifc.ifc_req;
    for (int i = ifc.ifc_len/sizeof(struct ifreq); --i >= 0; IFR++)
    {
        strcpy(ifr.ifr_name, IFR->ifr_name);
        if (ioctl(s, SIOCGIFFLAGS, &ifr) == 0)
        {
            if (!(ifr.ifr_flags & IFF_LOOPBACK))
            {
                if (ioctl(s, SIOCGIFHWADDR, &ifr) == 0)
                {
                    found = true;
                    break;
                }
            }
        }
    }
    close(s);
 
    if (found)
    {
        unsigned char addr[6];
        char mac[32];
        bcopy(ifr.ifr_hwaddr.sa_data, addr, 6);
        sprintf(mac, "%2.2x:%2.2x:%2.2x:%2.2x:%2.2x:%2.2x",
            addr[0], addr[1], addr[2], addr[3], addr[4], addr[5]);
        network.mac = mac;
    }
 
#endif
    return network;
}
*/
 
 
SystemInfo::ProcessorInfo SystemInfo::getProcessorInfo()
{
    ProcessorInfo processor;
    processor.cores = 0;
    processor.frequency = 0.0;
    processor.family = 0;
    processor.modelNumber = 0;
    processor.siblings = 0;
 
#if defined(_WIN32)
    SYSTEM_INFO sysinf;
    GetSystemInfo(&sysinf);
    switch (sysinf.wProcessorArchitecture) {
    case PROCESSOR_ARCHITECTURE_AMD64: {
        processor.architecture = "X64";
        break;
    }
    case PROCESSOR_ARCHITECTURE_IA64: {
        processor.architecture = "Intel Itanium-based";
        break;
    }
    case PROCESSOR_ARCHITECTURE_INTEL: {
        processor.architecture = "X86";
        break;
    }
    default:
        processor.architecture = "Unknown";
    };
    processor.siblings = sysinf.dwNumberOfProcessors;
    processor.modelNumber = sysinf.dwProcessorType;
    switch (sysinf.dwProcessorType) {
    case PROCESSOR_INTEL_386: {
        processor.model = "PROCESSOR_INTEL_386";
        break;
    }
    case PROCESSOR_INTEL_486: {
        processor.model = "PROCESSOR_INTEL_486";
        break;
    }
    case PROCESSOR_INTEL_PENTIUM: {
        processor.model = "PROCESSOR_INTEL_PENTIUM";
        break;
    }
    case PROCESSOR_INTEL_IA64: {
        processor.model = "PROCESSOR_INTEL_IA64";
        break;
    }
    case PROCESSOR_AMD_X8664: {
        processor.model = "PROCESSOR_AMD_X8664";
        break;
    }
    };
 
    DWORD dwMHz;
    DWORD BufSize = sizeof(DWORD);
    HKEY hKey;
    if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                     "HARDWARE\\DESCRIPTION\\System\\CentralProcessor\\0",
                     0,
                     KEY_READ,
                     &hKey)
        == ERROR_SUCCESS) {
        RegQueryValueEx(hKey, "~MHz", nullptr, nullptr, (LPBYTE)&dwMHz, &BufSize);
        processor.frequency = (double)dwMHz;
        RegCloseKey(hKey);
    }
 
    // TODO: this should be fixed to obtain correct number of physical cores
    processor.cores = 1;
 
#endif
 
#if defined(__linux__)
    char buffer[128];
    FILE* proccpu = fopen("/proc/cpuinfo", "r");
    if (proccpu != nullptr) {
        while (fgets(buffer, 128, proccpu) != nullptr) {
            std::string value;
            if (getCpuEntry("model", buffer, value) && !getCpuEntry("model name", buffer, value)) {
                processor.modelNumber = atoi(value.c_str());
            }
            if (getCpuEntry("model name", buffer, value)) {
                processor.model = value;
            }
            if (getCpuEntry("vendor_id", buffer, value)) {
                processor.vendor = value;
            }
            if (getCpuEntry("cpu family", buffer, value)) {
                processor.family = atoi(value.c_str());
            }
            if (getCpuEntry("cpu cores", buffer, value)) {
                processor.cores = atoi(value.c_str());
            }
            if (getCpuEntry("siblings", buffer, value)) {
                processor.siblings = atoi(value.c_str());
            }
            if (getCpuEntry("cpu MHz", buffer, value)) {
                processor.frequency = atof(value.c_str());
            }
        }
        fclose(proccpu);
    }
 
    struct utsname uts;
    if (uname(&uts) == 0) {
        processor.architecture = uts.machine;
    }
#elif defined(__APPLE__)
    //    std::string architecture;
 
    int mib[] = {CTL_HW, HW_CPU_FREQ};
    int64_t value = 0;
    size_t length = sizeof(value);
 
    if (!sysctl(mib, 2, &value, &length, nullptr, 0)) {
        processor.frequency = value / 1e+6; //this is in Hz. What is the expected frequency?
    }
 
    if (!sysctlbyname("hw.logicalcpu", &value, &length, nullptr, 0)) {
        processor.cores = value; //this is the number of cores
        //or cpus: hw.physicalcpu
    }
 
    char buff[513];
    size_t buffLen = 512;
    if (!sysctlbyname("machdep.cpu.vendor", buff, &buffLen, nullptr, 0)) {
        processor.vendor = buff; //this is the number of cores
        //or cpus: hw.physicalcpu
    }
    buffLen = 512;
    if (!sysctlbyname("machdep.cpu.brand_string", buff, &buffLen, nullptr, 0)) {
        processor.model = buff; //this is the number of cores
        //or cpus: hw.physicalcpu
    }
    if (!sysctlbyname("machdep.cpu.family", &value, &length, nullptr, 0)) {
        processor.family = value; //this is the number of cores
        //or cpus: hw.physicalcpu
    }
    if (!sysctlbyname("machdep.cpu.model", &value, &length, nullptr, 0)) {
        processor.modelNumber = value; //this is the number of cores
        //or cpus: hw.physicalcpu
    }
 
 
#endif
    return processor;
}
 
 
SystemInfo::PlatformInfo SystemInfo::getPlatformInfo()
{
    PlatformInfo platform;
 
#if defined(_WIN32)
    platform.name = "Windows";
    OSVERSIONINFOEX osver;
    osver.dwOSVersionInfoSize = sizeof(osver);
    if (GetVersionEx((LPOSVERSIONINFO)&osver)) {
        char buff[64];
        sprintf(buff, "%d.%d", osver.dwMajorVersion, osver.dwMinorVersion);
        platform.release = buff;
        platform.codename = buff;
        if ((osver.dwMajorVersion == 10) && (osver.dwMinorVersion == 0) && (osver.wProductType == VER_NT_WORKSTATION)) {
            platform.distribution = "10";
        } else if ((osver.dwMajorVersion == 10) && (osver.dwMinorVersion == 0) && (osver.wProductType != VER_NT_WORKSTATION)) {
            platform.distribution = "Server 2016";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 2) && (osver.wProductType == VER_NT_WORKSTATION)) {
            platform.distribution = "8.1";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 3) && (osver.wProductType != VER_NT_WORKSTATION)) {
            platform.distribution = "Server 2012 R2";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 2) && (osver.wProductType == VER_NT_WORKSTATION)) {
            platform.distribution = "8";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 2) && (osver.wProductType != VER_NT_WORKSTATION)) {
            platform.distribution = "Server 2012";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 1) && (osver.wProductType == VER_NT_WORKSTATION)) {
            platform.distribution = "7";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 1) && (osver.wProductType != VER_NT_WORKSTATION)) {
            platform.distribution = "Server 2008 R2";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 0) && (osver.wProductType == VER_NT_WORKSTATION)) {
            platform.distribution = "Vista";
        } else if ((osver.dwMajorVersion == 6) && (osver.dwMinorVersion == 0) && (osver.wProductType != VER_NT_WORKSTATION)) {
            platform.distribution = "Server 2008";
        } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 2)) {
            platform.distribution = "Server 2003";
            //    } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 2) && (osver.wSuiteMask & VER_SUITE_WH_SERVER)) {
            //       platform.distribution = "Home Server";
            //    } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 2) && (GetSystemMetrics(SM_SERVERR2) != 0)) {
            //       platform.distribution = "Server 2003 R2";
        } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 2) && (osver.wProductType == VER_NT_WORKSTATION)) /* &&(osver.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64)*/ {
            platform.distribution = "XP Professional x64 Edition";
        } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 1)) {
            platform.distribution = "XP";
        } else if ((osver.dwMajorVersion == 5) && (osver.dwMinorVersion == 0)) {
            platform.distribution = "2000";
        }
    }
 
    const char* a = GetEnvironmentStrings();
    size_t prev = 0;
    for (size_t i = 0;; i++) {
        if (a[i] == '\0') {
            std::string tmpVariable(a + prev, a + i);
            size_t equalsSign = tmpVariable.find("=");
            if (equalsSign != std::string::npos && equalsSign != 0) // among environment variables there are DOS-related ones that start with a =
            {
                platform.environmentVars.put(tmpVariable.substr(0, equalsSign), tmpVariable.substr(equalsSign + 1));
            }
            prev = i + 1;
            if (a[i + 1] == '\0') {
                break;
            }
        }
    }
 
#endif
 
#if defined(__linux__)
    struct utsname uts;
    if (uname(&uts) == 0) {
        platform.name = uts.sysname;
        platform.kernel = uts.release;
    } else {
        platform.name = "Linux";
    }
 
    char buffer[128];
    FILE* release = popen("lsb_release -ric", "r");
    if (release != nullptr) {
        while (fgets(buffer, 128, release) != nullptr) {
            std::string value;
            if (getCpuEntry("Distributor ID", buffer, value)) {
                platform.distribution = value;
            }
            if (getCpuEntry("Release", buffer, value)) {
                platform.release = value;
            }
            if (getCpuEntry("Codename", buffer, value)) {
                platform.codename = value;
            }
        }
        pclose(release);
    }
 
    char* varChar = *environ;
 
    for (int i = 0; varChar != nullptr; i++) {
        std::string tmpVariable(varChar);
        size_t equalsSign = tmpVariable.find('=');
        if (equalsSign != std::string::npos) {
            platform.environmentVars.put(tmpVariable.substr(0, equalsSign), tmpVariable.substr(equalsSign + 1));
        }
        varChar = *(environ + i);
    }
#endif
 
#if defined(__APPLE__)
 
    char buff[513];
    size_t buffLen = 512;
    if (!sysctlbyname("kern.ostype", buff, &buffLen, nullptr, 0)) {
        platform.name = buff;
    }
 
    if (!sysctlbyname("kern.osrelease", buff, &buffLen, nullptr, 0)) {
        platform.release = buff;
    }
 
#endif
    return platform;
}
 
 
SystemInfo::UserInfo SystemInfo::getUserInfo()
{
    UserInfo user;
    user.userID = 0;
 
#if defined(_WIN32)
    char path[MAX_PATH + 1];
    if (SHGetFolderPathA(nullptr, CSIDL_PROFILE, nullptr, 0, path) == S_OK)
        user.homeDir = path;
 
    char username[UNLEN + 1];
    DWORD nsize = UNLEN + 1;
    if (GetUserName(username, &nsize)) {
        user.userName = username;
        user.realName = username;
    }
#endif
 
#if defined(__linux__) || defined(__APPLE__)
    struct passwd* pwd = getpwuid(getuid());
    user.userID = getuid();
    if (pwd != nullptr) {
        user.userName = pwd->pw_name;
        user.realName = pwd->pw_gecos;
        user.homeDir = pwd->pw_dir;
    }
#endif
    return user;
}
 
 
SystemInfo::LoadInfo SystemInfo::getLoadInfo()
{
    LoadInfo load;
    load.cpuLoad1 = 0.0;
    load.cpuLoad5 = 0.0;
    load.cpuLoad15 = 0.0;
    load.cpuLoadInstant = 0;
 
#if defined(_WIN32)
    if (globalLoadCollector) {
        load = globalLoadCollector->getCpuLoad();
        int siblings = getProcessorInfo().siblings;
        if (siblings > 1) {
            load.cpuLoad1 *= siblings;
            load.cpuLoad5 *= siblings;
            load.cpuLoad15 *= siblings;
        }
    }
#endif
 
#if defined(__linux__)
    FILE* procload = fopen("/proc/loadavg", "r");
    if (procload != nullptr) {
        char buff[128];
        int ret = fscanf(procload, "%lf %lf %lf %127s", &(load.cpuLoad1), &(load.cpuLoad5), &(load.cpuLoad15), buff);
        if (ret > 0) {
            char* tail = strchr(buff, '/');
            if ((tail != nullptr) && (tail != buff)) {
                load.cpuLoadInstant = (int)(strtol(buff, &tail, 0) - 1);
            }
        }
        fclose(procload);
    }
#endif
 
#if defined(__APPLE__)
 
    mach_msg_type_number_t count = HOST_CPU_LOAD_INFO_COUNT;
    host_cpu_load_info_data_t cpu_load;
 
    if (KERN_SUCCESS == host_statistics(mach_host_self(), HOST_CPU_LOAD_INFO, (host_info64_t)&cpu_load, &count)) {
        //How to map this information into yarp structure?
        natural_t total = 0;
        for (int i = 0; i < CPU_STATE_MAX; ++i) {
            total += cpu_load.cpu_ticks[i];
        }
 
        load.cpuLoad1 = 100.0 * cpu_load.cpu_ticks[CPU_STATE_USER] / total;
    }
#endif
    return load;
}
 
 
SystemInfo::ProcessInfo SystemInfo::getProcessInfo(int pid)
{
    // If not specified, get information for current process
    if (pid == 0) {
        pid = yarp::os::getpid();
    }
 
    SystemInfo::ProcessInfo info;
    info.pid = -1; // invalid
    info.schedPolicy = -1;
    info.schedPriority = -1;
#if defined(__linux__)
    FILE* file;
    char cmdline[256] = {0};
    char filename[256];
    sprintf(filename, "/proc/%d/cmdline", pid);
    file = fopen(filename, "r");
    if (file != nullptr) {
        char* p = fgets(cmdline, sizeof(cmdline) / sizeof(*cmdline), file);
        fclose(file);
        if (p != nullptr) {
            while (*p != 0) {
                p += strlen(p);
                if (*(p + 1) != 0) {
                    *p = ' ';
                }
                p++;
            }
            info.pid = pid;
            // split the cmdline to find the arguments
            info.name = cmdline;
            size_t index = info.name.find(' ');
            if (index != std::string::npos) {
                info.name = info.name.substr(0, index);
                info.arguments = cmdline;
                info.arguments = info.arguments.substr(index + 1);
            }
        }
    }
 
    // scheduling params
    struct sched_param param;
    if (sched_getparam(pid, &param) == 0) {
        info.schedPriority = param.__sched_priority;
    }
    info.schedPolicy = sched_getscheduler(pid);
 
#elif defined(_WIN32)
    HANDLE hnd = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, pid);
    if (hnd) {
        TCHAR filename[MAX_PATH];
        if (GetModuleBaseName(hnd, 0, filename, MAX_PATH)) {
            info.name = filename;
            info.pid = pid;
        }
        CloseHandle(hnd);
    }
    // reterieving arguments
    HRESULT hr = 0;
    IWbemLocator* WbemLocator = nullptr;
    IWbemServices* WbemServices = nullptr;
    IEnumWbemClassObject* EnumWbem = nullptr;
 
    //initializate the Windows security
    hr = CoInitializeEx(0, COINIT_MULTITHREADED);
    hr = CoInitializeSecurity(nullptr, -1, nullptr, nullptr, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, nullptr, EOAC_NONE, nullptr);
    hr = CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID*)&WbemLocator);
    if (WbemLocator != nullptr) {
        //connect to the WMI
        BSTR val1 = SysAllocString(L"ROOT\\CimV2");
        hr = WbemLocator->ConnectServer(val1, nullptr, nullptr, 0, 0, 0, 0, &WbemServices);
        SysFreeString(val1);
        if (WbemServices != nullptr) {
            //Run the WQL Query
            BSTR val2 = SysAllocString(L"WQL");
            BSTR val3 = SysAllocString(L"SELECT ProcessId, CommandLine FROM Win32_Process");
            hr = WbemServices->ExecQuery(val2, val3, WBEM_FLAG_FORWARD_ONLY, nullptr, &EnumWbem);
            SysFreeString(val2);
            SysFreeString(val3);
            // Iterate over the enumerator
            if (EnumWbem != nullptr) {
                IWbemClassObject* result = nullptr;
                ULONG returnedCount = 0;
 
                while ((hr = EnumWbem->Next(WBEM_INFINITE, 1, &result, &returnedCount)) == S_OK) {
                    VARIANT ProcessId;
                    VARIANT CommandLine;
 
                    // access the properties
                    hr = result->Get(L"ProcessId", 0, &ProcessId, 0, 0);
                    hr = result->Get(L"CommandLine", 0, &CommandLine, 0, 0);
                    if (!(CommandLine.vt == VT_NULL) && ProcessId.uintVal == (unsigned int)pid) {
                        // covert BSTR to std::string
                        int res = WideCharToMultiByte(CP_UTF8, 0, CommandLine.bstrVal, -1, nullptr, 0, nullptr, nullptr);
                        info.arguments.resize(res);
                        WideCharToMultiByte(CP_UTF8, 0, CommandLine.bstrVal, -1, &info.arguments[0], res, nullptr, nullptr);
                        size_t idx = info.arguments.find(' ');
                        if (idx == info.arguments.npos) {
                            info.arguments.clear();
                        } else {
                            info.arguments = info.arguments.substr(idx + 2); // it seems windows adds two spaces after the program name
                        }
                        info.pid = pid;
                        VariantClear(&ProcessId);
                        VariantClear(&CommandLine);
                        break;
                    }
                    result->Release();
                } // end while
 
                EnumWbem->Release();
            } // end if EnumWbem
 
            WbemServices->Release();
        } // end if WbemServices
 
        WbemLocator->Release();
    } // end if WbemLocator
 
    CoUninitialize();
#elif defined(__APPLE__)
    kinfo_proc procInfo;
    size_t length = sizeof(procInfo);
 
    int mib[4] = {CTL_KERN, KERN_PROC, KERN_PROC_PID, pid};
 
    if (!sysctl(mib, 4, &procInfo, &length, nullptr, 0)) {
        info.name = procInfo.kp_proc.p_comm;
        info.pid = procInfo.kp_proc.p_pid;
 
        //Some info here: https://gist.github.com/nonowarn/770696
        mib[1] = KERN_PROCARGS;
        mib[2] = pid;
        char* proc_argv;
        size_t argv_len;
        //getting length of execute string
        int result = sysctl(mib, 3, nullptr, &argv_len, nullptr, 0);
        if (result != 0) {
            yCError(SYSTEMINFO, "sysctl: %d, %s", errno, strerror(errno));
            return info;
        }
 
        //now getting the string
        proc_argv = (char*)malloc(sizeof(char) * argv_len);
        result = sysctl(mib, 3, proc_argv, &argv_len, nullptr, 0);
        if (result != 0) {
            yCError(SYSTEMINFO, "sysctl: %d, %s", errno, strerror(errno));
            free(proc_argv);
            return info;
        }
 
        //looking for '\0', i.e. NULL char
        //skip first string which is the executable
        size_t index = 0;
        while (index < argv_len && proc_argv[index] != '\0') {
            index++;
        }
        index++;
        //now we have to split the remaining string
        //Note: this is not easy. We don't know the format
        //See: http://lists.apple.com/archives/darwin-kernel/2012/Mar/msg00025.html
        //for example, I get both arguments and environment variables
 
        std::stringstream arguments;
        while (index < argv_len) {
            if (proc_argv[index] == '\0' && index != argv_len - 1) {
                arguments << " ";
            } else {
                arguments << proc_argv[index];
            }
            index++;
        }
 
        free(proc_argv);
        info.arguments = arguments.str();
    }
 
#endif
    return info;
}