Rooba
/
boinc
mirror of https://github.com/BOINC/boinc.git
1
0
Fork 0
Code Issues 1 Packages Projects Releases Wiki Activity
boinc/client/hostinfo_win.cpp

1154 lines
41 KiB
C++
Raw Blame History

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC 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,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC 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 BOINC. If not, see <http://www.gnu.org/licenses/>.
#if defined(_WIN32) && !defined(__STDWX_H__) && !defined(_BOINC_WIN_) && !defined(_AFX_STDAFX_H_)
#include "boinc_win.h"
#endif
#ifdef _MSC_VER
#define snprintf _snprintf
#endif
#ifndef __CYGWIN__
#include <intrin.h>
#endif
#include "client_types.h"
#include "filesys.h"
#include "str_util.h"
#include "str_replace.h"
#include "client_msgs.h"
#include "error_numbers.h"
#include "hostinfo_network.h"
#include "hostinfo.h"
#include "idlemon.h"
// Newer system metrics values than what is currently defined in
// Visual Studio 2005
#ifndef SM_TABLETPC
#define SM_TABLETPC 86
#endif
#ifndef SM_MEDIACENTER
#define SM_MEDIACENTER 87
#endif
#ifndef SM_STARTER
#define SM_STARTER 88
#endif
#ifndef SM_SERVERR2
#define SM_SERVERR2 89
#endif
// Newer product types than what is currently defined in
// Visual Studio 2005
#ifndef PRODUCT_ULTIMATE
#define PRODUCT_ULTIMATE 0x00000001
#endif
#ifndef PRODUCT_HOME_BASIC
#define PRODUCT_HOME_BASIC 0x00000002
#endif
#ifndef PRODUCT_HOME_PREMIUM
#define PRODUCT_HOME_PREMIUM 0x00000003
#endif
#ifndef PRODUCT_ENTERPRISE
#define PRODUCT_ENTERPRISE 0x00000004
#endif
#ifndef PRODUCT_HOME_BASIC_N
#define PRODUCT_HOME_BASIC_N 0x00000005
#endif
#ifndef PRODUCT_BUSINESS
#define PRODUCT_BUSINESS 0x00000006
#endif
#ifndef PRODUCT_STANDARD_SERVER
#define PRODUCT_STANDARD_SERVER 0x00000007
#endif
#ifndef PRODUCT_DATACENTER_SERVER
#define PRODUCT_DATACENTER_SERVER 0x00000008
#endif
#ifndef PRODUCT_SMALLBUSINESS_SERVER
#define PRODUCT_SMALLBUSINESS_SERVER 0x00000009
#endif
#ifndef PRODUCT_ENTERPRISE_SERVER
#define PRODUCT_ENTERPRISE_SERVER 0x0000000A
#endif
#ifndef PRODUCT_STARTER
#define PRODUCT_STARTER 0x0000000B
#endif
#ifndef PRODUCT_DATACENTER_SERVER_CORE
#define PRODUCT_DATACENTER_SERVER_CORE 0x0000000C
#endif
#ifndef PRODUCT_STANDARD_SERVER_CORE
#define PRODUCT_STANDARD_SERVER_CORE 0x0000000D
#endif
#ifndef PRODUCT_ENTERPRISE_SERVER_CORE
#define PRODUCT_ENTERPRISE_SERVER_CORE 0x0000000E
#endif
#ifndef PRODUCT_ENTERPRISE_SERVER_IA64
#define PRODUCT_ENTERPRISE_SERVER_IA64 0x0000000F
#endif
#ifndef PRODUCT_BUSINESS_N
#define PRODUCT_BUSINESS_N 0x00000010
#endif
#ifndef PRODUCT_WEB_SERVER
#define PRODUCT_WEB_SERVER 0x00000011
#endif
#ifndef PRODUCT_CLUSTER_SERVER
#define PRODUCT_CLUSTER_SERVER 0x00000012
#endif
#ifndef PRODUCT_HOME_SERVER
#define PRODUCT_HOME_SERVER 0x00000013
#endif
#ifndef PRODUCT_STORAGE_EXPRESS_SERVER
#define PRODUCT_STORAGE_EXPRESS_SERVER 0x00000014
#endif
#ifndef PRODUCT_STORAGE_STANDARD_SERVER
#define PRODUCT_STORAGE_STANDARD_SERVER 0x00000015
#endif
#ifndef PRODUCT_STORAGE_WORKGROUP_SERVER
#define PRODUCT_STORAGE_WORKGROUP_SERVER 0x00000016
#endif
#ifndef PRODUCT_STORAGE_ENTERPRISE_SERVER
#define PRODUCT_STORAGE_ENTERPRISE_SERVER 0x00000017
#endif
#ifndef PRODUCT_SERVER_FOR_SMALLBUSINESS
#define PRODUCT_SERVER_FOR_SMALLBUSINESS 0x00000018
#endif
#ifndef PRODUCT_SMALLBUSINESS_SERVER_PREMIUM
#define PRODUCT_SMALLBUSINESS_SERVER_PREMIUM 0x00000019
#endif
#ifndef PRODUCT_HOME_PREMIUM_N
#define PRODUCT_HOME_PREMIUM_N 0x0000001A
#endif
#ifndef PRODUCT_ENTERPRISE_N
#define PRODUCT_ENTERPRISE_N 0x0000001B
#endif
#ifndef PRODUCT_ULTIMATE_N
#define PRODUCT_ULTIMATE_N 0x0000001C
#endif
#ifndef PRODUCT_WEB_SERVER_CORE
#define PRODUCT_WEB_SERVER_CORE 0x0000001D
#endif
#ifndef PRODUCT_MEDIUMBUSINESS_SERVER_MANAGEMENT
#define PRODUCT_MEDIUMBUSINESS_SERVER_MANAGEMENT 0x0000001E
#endif
#ifndef PRODUCT_MEDIUMBUSINESS_SERVER_SECURITY
#define PRODUCT_MEDIUMBUSINESS_SERVER_SECURITY 0x0000001F
#endif
#ifndef PRODUCT_MEDIUMBUSINESS_SERVER_MESSAGING
#define PRODUCT_MEDIUMBUSINESS_SERVER_MESSAGING 0x00000020
#endif
#ifndef PRODUCT_SMALLBUSINESS_SERVER_PRIME
#define PRODUCT_SMALLBUSINESS_SERVER_PRIME 0x00000021
#endif
#ifndef PRODUCT_HOME_PREMIUM_SERVER
#define PRODUCT_HOME_PREMIUM_SERVER 0x00000022
#endif
#ifndef PRODUCT_SERVER_FOR_SMALLBUSINESS_V
#define PRODUCT_SERVER_FOR_SMALLBUSINESS_V 0x00000023
#endif
#ifndef PRODUCT_STANDARD_SERVER_V
#define PRODUCT_STANDARD_SERVER_V 0x00000024
#endif
#ifndef PRODUCT_DATACENTER_SERVER_V
#define PRODUCT_DATACENTER_SERVER_V 0x00000025
#endif
#ifndef PRODUCT_ENTERPRISE_SERVER_V
#define PRODUCT_ENTERPRISE_SERVER_V 0x00000026
#endif
#ifndef PRODUCT_DATACENTER_SERVER_CORE_V
#define PRODUCT_DATACENTER_SERVER_CORE_V 0x00000027
#endif
#ifndef PRODUCT_STANDARD_SERVER_CORE_V
#define PRODUCT_STANDARD_SERVER_CORE_V 0x00000028
#endif
#ifndef PRODUCT_ENTERPRISE_SERVER_CORE_V
#define PRODUCT_ENTERPRISE_SERVER_CORE_V 0x00000029
#endif
#ifndef PRODUCT_HYPERV
#define PRODUCT_HYPERV 0x0000002A
#endif
#ifndef PRODUCT_STORAGE_EXPRESS_SERVER_CORE
#define PRODUCT_STORAGE_EXPRESS_SERVER_CORE 0x0000002B
#endif
#ifndef PRODUCT_STORAGE_STANDARD_SERVER_CORE
#define PRODUCT_STORAGE_STANDARD_SERVER_CORE 0x0000002C
#endif
#ifndef PRODUCT_STORAGE_WORKGROUP_SERVER_CORE
#define PRODUCT_STORAGE_WORKGROUP_SERVER_CORE 0x0000002D
#endif
#ifndef PRODUCT_STORAGE_ENTERPRISE_SERVER_CORE
#define PRODUCT_STORAGE_ENTERPRISE_SERVER_CORE 0x0000002E
#endif
#ifndef PRODUCT_PROFESSIONAL
#define PRODUCT_PROFESSIONAL 0x00000030
#endif
#ifndef PRODUCT_PROFESSIONAL_N
#define PRODUCT_PROFESSIONAL_N 0x00000031
#endif
#ifndef PRODUCT_SB_SOLUTION_SERVER
#define PRODUCT_SB_SOLUTION_SERVER 0x00000032
#endif
#ifndef PRODUCT_SERVER_FOR_SB_SOLUTIONS
#define PRODUCT_SERVER_FOR_SB_SOLUTIONS 0x00000033
#endif
#ifndef PRODUCT_STANDARD_SERVER_SOLUTIONS
#define PRODUCT_STANDARD_SERVER_SOLUTIONS 0x00000034
#endif
#ifndef PRODUCT_STANDARD_SERVER_SOLUTIONS_CORE
#define PRODUCT_STANDARD_SERVER_SOLUTIONS_CORE 0x00000035
#endif
#ifndef PRODUCT_SB_SOLUTION_SERVER_EM
#define PRODUCT_SB_SOLUTION_SERVER_EM 0x00000036
#endif
#ifndef PRODUCT_SERVER_FOR_SB_SOLUTIONS_EM
#define PRODUCT_SERVER_FOR_SB_SOLUTIONS_EM 0x00000037
#endif
#ifndef PRODUCT_SOLUTION_EMBEDDEDSERVER
#define PRODUCT_SOLUTION_EMBEDDEDSERVER 0x00000038
#endif
#ifndef PRODUCT_SOLUTION_EMBEDDEDSERVER_CORE
#define PRODUCT_SOLUTION_EMBEDDEDSERVER_CORE 0x00000039
#endif
#ifndef PRODUCT_SMALLBUSINESS_SERVER_PREMIUM_CORE
#define PRODUCT_SMALLBUSINESS_SERVER_PREMIUM_CORE 0x0000003F
#endif
#ifndef PRODUCT_ESSENTIALBUSINESS_SERVER_MGMT
#define PRODUCT_ESSENTIALBUSINESS_SERVER_MGMT 0x0000003B
#endif
#ifndef PRODUCT_ESSENTIALBUSINESS_SERVER_ADDL
#define PRODUCT_ESSENTIALBUSINESS_SERVER_ADDL 0x0000003C
#endif
#ifndef PRODUCT_ESSENTIALBUSINESS_SERVER_MGMTSVC
#define PRODUCT_ESSENTIALBUSINESS_SERVER_MGMTSVC 0x0000003D
#endif
#ifndef PRODUCT_ESSENTIALBUSINESS_SERVER_ADDLSVC
#define PRODUCT_ESSENTIALBUSINESS_SERVER_ADDLSVC 0x0000003E
#endif
#ifndef PRODUCT_CLUSTER_SERVER_V
#define PRODUCT_CLUSTER_SERVER_V 0x00000040
#endif
#ifndef PRODUCT_EMBEDDED
#define PRODUCT_EMBEDDED 0x00000041
#endif
#ifndef PRODUCT_STARTER_E
#define PRODUCT_STARTER_E 0x00000042
#endif
#ifndef PRODUCT_HOME_BASIC_E
#define PRODUCT_HOME_BASIC_E 0x00000043
#endif
#ifndef PRODUCT_HOME_PREMIUM_E
#define PRODUCT_HOME_PREMIUM_E 0x00000044
#endif
#ifndef PRODUCT_PROFESSIONAL_E
#define PRODUCT_PROFESSIONAL_E 0x00000045
#endif
#ifndef PRODUCT_ENTERPRISE_E
#define PRODUCT_ENTERPRISE_E 0x00000046
#endif
#ifndef PRODUCT_ULTIMATE_E
#define PRODUCT_ULTIMATE_E 0x00000047
#endif
// Newer suite types than what is currently defined in
// Visual Studio 2005
#ifndef VER_SUITE_WH_SERVER
#define VER_SUITE_WH_SERVER 0x00008000
#endif
// Returns the number of seconds difference from UTC
//
int get_timezone(int& timezone) {
TIME_ZONE_INFORMATION tzi;
memset(&tzi, 0, sizeof(TIME_ZONE_INFORMATION));
DWORD result = GetTimeZoneInformation(&tzi);
if (result == TIME_ZONE_ID_DAYLIGHT) {
timezone = -(tzi.Bias + tzi.DaylightBias) * 60;
} else {
timezone = -(tzi.Bias + tzi.StandardBias) * 60;
}
return 0;
}
// Returns the memory information
//
int get_memory_info(double& bytes, double& swap) {
MEMORYSTATUSEX mStatusEx;
ZeroMemory(&mStatusEx, sizeof(MEMORYSTATUSEX));
mStatusEx.dwLength = sizeof(MEMORYSTATUSEX);
GlobalMemoryStatusEx(&mStatusEx);
bytes = (double)mStatusEx.ullTotalPhys;
swap = (double)mStatusEx.ullTotalPageFile;
return 0;
}
// Returns the OS name and version
//
typedef void (WINAPI *PGNSI)(LPSYSTEM_INFO);
typedef BOOL (WINAPI *PGPI)(DWORD, DWORD, DWORD, DWORD, PDWORD);
int get_os_information(
char* os_name, int /*os_name_size*/, char* os_version, int os_version_size
)
{
// This code snip-it was copied straight out of the MSDN Platform SDK
// Getting the System Version example and modified to dump the output
// into os_name.
char szVersion[128];
char szSKU[128];
char szServicePack[128];
OSVERSIONINFOEX osvi;
SYSTEM_INFO si;
PGNSI pGNSI;
PGPI pGPI;
BOOL bOsVersionInfoEx;
DWORD dwType = 0;
ZeroMemory(szVersion, sizeof(szVersion));
ZeroMemory(szSKU, sizeof(szSKU));
ZeroMemory(szServicePack, sizeof(szServicePack));
ZeroMemory(&si, sizeof(SYSTEM_INFO));
ZeroMemory(&osvi, sizeof(OSVERSIONINFOEX));
osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
// Try calling GetVersionEx using the OSVERSIONINFOEX structure.
// If that fails, try using the OSVERSIONINFO structure.
bOsVersionInfoEx = GetVersionEx ((OSVERSIONINFO *) &osvi);
if(!bOsVersionInfoEx) {
osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFO);
GetVersionEx ( (OSVERSIONINFO *) &osvi );
}
// Call GetNativeSystemInfo if supported or GetSystemInfo otherwise.
pGNSI = (PGNSI) GetProcAddress(GetModuleHandle(_T("kernel32.dll")), "GetNativeSystemInfo");
if(NULL != pGNSI) {
pGNSI(&si);
} else {
GetSystemInfo(&si);
}
pGPI = (PGPI) GetProcAddress(GetModuleHandle(_T("kernel32.dll")), "GetProductInfo");
// Windows is a Microsoft OS
strcpy(os_name, "Microsoft ");
switch (osvi.dwPlatformId)
{
case VER_PLATFORM_WIN32_NT:
if ( osvi.dwMajorVersion == 6 && osvi.dwMinorVersion == 1 )
{
if( osvi.wProductType == VER_NT_WORKSTATION ) {
strcat(os_name, "Windows 7");
} else {
strcat(os_name, "Windows Server 2008 \"R2\"");
}
pGPI( 6, 1, 0, 0, &dwType);
}
if ( osvi.dwMajorVersion == 6 && osvi.dwMinorVersion == 0 )
{
if( osvi.wProductType == VER_NT_WORKSTATION ) {
strcat(os_name, "Windows Vista");
} else {
strcat(os_name, "Windows Server 2008");
}
pGPI( 6, 0, 0, 0, &dwType);
}
if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 2 )
{
if( osvi.wProductType == VER_NT_WORKSTATION) {
strcat(os_name, "Windows XP");
} else {
if( GetSystemMetrics(SM_SERVERR2) ) {
strcat(os_name, "Windows Server 2003 \"R2\"");
} else {
strcat(os_name, "Windows Server 2003");
}
}
}
if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 1 )
{
strcat(os_name, "Windows XP");
}
if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0 )
{
strcat(os_name, "Windows 2000");
}
if ( osvi.dwMajorVersion <= 4 )
{
strcat(os_name, "Windows NT");
}
break;
case VER_PLATFORM_WIN32_WINDOWS:
if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 0)
{
strcat(os_name, "Windows 95");
}
if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 10)
{
strcat( os_name, "Windows 98");
}
if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 90)
{
strcat( os_name, "Windows Millennium");
}
break;
case VER_PLATFORM_WIN32s:
strcat( os_name, "Win32s");
break;
}
snprintf( szVersion, sizeof(szVersion), ", (%.2u.%.2u.%.4u.%.2u)",
osvi.dwMajorVersion, osvi.dwMinorVersion, (osvi.dwBuildNumber & 0xFFFF), 0 );
switch (osvi.dwPlatformId)
{
// Test for the Windows NT product family.
case VER_PLATFORM_WIN32_NT:
// Test for specific product on Windows NT 4.0 SP6 and later.
if( bOsVersionInfoEx ) {
// Test for the workstation type.
if ( osvi.wProductType == VER_NT_WORKSTATION ) {
if( (osvi.dwMajorVersion == 6) ) {
switch(dwType) {
case PRODUCT_ULTIMATE:
strcat(szSKU, "Ultimate ");
break;
case PRODUCT_PROFESSIONAL:
strcat(szSKU, "Professional ");
break;
case PRODUCT_HOME_PREMIUM:
strcat(szSKU, "Home Premium ");
break;
case PRODUCT_HOME_BASIC:
strcat(szSKU, "Home Basic ");
break;
case PRODUCT_ENTERPRISE:
strcat(szSKU, "Enterprise ");
break;
case PRODUCT_BUSINESS:
strcat(szSKU, "Business ");
break;
case PRODUCT_STARTER:
strcat(szSKU, "Starter ");
break;
case PRODUCT_PROFESSIONAL_N:
strcat(szSKU, "Professional N ");
break;
case PRODUCT_HOME_PREMIUM_N:
strcat(szSKU, "Home Premium N ");
break;
case PRODUCT_HOME_BASIC_N:
strcat(szSKU, "Home Basic N ");
break;
case PRODUCT_ULTIMATE_N:
strcat(szSKU, "Ultimate N ");
break;
case PRODUCT_ENTERPRISE_N:
strcat(szSKU, "Enterprise N ");
break;
case PRODUCT_BUSINESS_N:
strcat(szSKU, "Business N ");
break;
case PRODUCT_PROFESSIONAL_E:
strcat(szSKU, "Professional E ");
break;
case PRODUCT_HOME_PREMIUM_E:
strcat(szSKU, "Home Premium E ");
break;
case PRODUCT_HOME_BASIC_E:
strcat(szSKU, "Home Basic E ");
break;
case PRODUCT_ULTIMATE_E:
strcat(szSKU, "Ultimate E ");
break;
case PRODUCT_ENTERPRISE_E:
strcat(szSKU, "Enterprise E ");
break;
}
} else if( (osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 2) ) {
if( osvi.wSuiteMask & VER_SUITE_PERSONAL ) {
strcat(szSKU, "Home ");
} else {
strcat(szSKU, "Professional ");
}
} else if( (osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 1) ) {
if( osvi.wSuiteMask & VER_SUITE_PERSONAL ) {
strcat(szSKU, "Home ");
} else if( GetSystemMetrics(SM_TABLETPC) ) {
strcat(szSKU, "Tablet PC ");
} else if( GetSystemMetrics(SM_MEDIACENTER) ) {
strcat(szSKU, "Media Center ");
} else if( GetSystemMetrics(SM_STARTER) ) {
strcat(szSKU, "Starter ");
} else {
strcat(szSKU, "Professional ");
}
} else if( (osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0) ) {
strcat(szSKU, "Professional ");
} else if( (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 0) ) {
strcat(szSKU, "Workstation ");
}
}
// Test for the server type.
else if ( (osvi.wProductType == VER_NT_SERVER) || (osvi.wProductType == VER_NT_DOMAIN_CONTROLLER) ) {
if( (osvi.dwMajorVersion == 6) ) {
switch(dwType) {
case PRODUCT_CLUSTER_SERVER:
strcat( szSKU, "Cluster Server ");
break;
case PRODUCT_DATACENTER_SERVER:
strcat( szSKU, "Datacenter ");
break;
case PRODUCT_DATACENTER_SERVER_CORE:
strcat( szSKU, "Datacenter (core installation) ");
break;
case PRODUCT_ENTERPRISE_SERVER:
strcat( szSKU, "Enterprise ");
break;
case PRODUCT_ENTERPRISE_SERVER_CORE:
strcat( szSKU, "Enterprise (core installation) ");
break;
case PRODUCT_ENTERPRISE_SERVER_IA64:
strcat( szSKU, "Enterprise ");
break;
case PRODUCT_SMALLBUSINESS_SERVER:
strcat( szSKU, "Small Business Server");
break;
case PRODUCT_SMALLBUSINESS_SERVER_PREMIUM:
strcat( szSKU, "Small Business Server Premium ");
break;
case PRODUCT_STANDARD_SERVER:
strcat( szSKU, "Standard ");
break;
case PRODUCT_STANDARD_SERVER_CORE:
strcat( szSKU, "Standard (core installation) ");
break;
case PRODUCT_WEB_SERVER:
strcat( szSKU, "Web Server ");
break;
case PRODUCT_WEB_SERVER_CORE:
strcat( szSKU, "Web Server (core installtion) ");
break;
case PRODUCT_HOME_SERVER:
strcat( szSKU, "Home Server ");
break;
case PRODUCT_HOME_PREMIUM_SERVER:
strcat( szSKU, "Home Premium Server ");
break;
case PRODUCT_STORAGE_EXPRESS_SERVER:
strcat( szSKU, "Storage Server Express ");
break;
case PRODUCT_STORAGE_STANDARD_SERVER:
strcat( szSKU, "Storage Server Standard ");
break;
case PRODUCT_STORAGE_WORKGROUP_SERVER:
strcat( szSKU, "Storage Server Workgroup ");
break;
case PRODUCT_STORAGE_ENTERPRISE_SERVER:
strcat( szSKU, "Storage Server Enterprise ");
break;
case PRODUCT_SERVER_FOR_SMALLBUSINESS:
strcat( szSKU, "Server For Small Business ");
break;
}
} else if( (osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 2) ) {
if( osvi.wSuiteMask & VER_SUITE_DATACENTER ) {
strcat( szSKU, "Datacenter Server " );
} else if( osvi.wSuiteMask & VER_SUITE_ENTERPRISE ) {
strcat( szSKU, "Enterprise Server " );
} else if ( osvi.wSuiteMask & VER_SUITE_BLADE ) {
strcat( szSKU, "Web Server " );
} else if ( osvi.wSuiteMask & VER_SUITE_WH_SERVER ) {
strcat( szSKU, "Home Server " );
} else {
strcat( szSKU, "Standard Server " );
}
} else if( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0 ) {
if( osvi.wSuiteMask & VER_SUITE_DATACENTER ) {
strcat( szSKU, "Datacenter Server " );
} else if( osvi.wSuiteMask & VER_SUITE_ENTERPRISE ) {
strcat( szSKU, "Advanced Server " );
} else {
strcat( szSKU, "Standard Server " );
}
} else { // Windows NT 4.0
if( osvi.wSuiteMask & VER_SUITE_ENTERPRISE ) {
strcat( szSKU, "Enterprise Server " );
} else {
strcat( szSKU, "Server " );
}
}
}
switch (si.wProcessorArchitecture)
{
case PROCESSOR_ARCHITECTURE_INTEL:
strcat(szSKU, "x86 ");
break;
case PROCESSOR_ARCHITECTURE_MIPS:
strcat(szSKU, "MIPS ");
break;
case PROCESSOR_ARCHITECTURE_ALPHA:
strcat(szSKU, "Alpha ");
break;
case PROCESSOR_ARCHITECTURE_PPC:
strcat(szSKU, "PowerPC ");
break;
case PROCESSOR_ARCHITECTURE_IA64:
strcat(szSKU, "Itanium ");
break;
case PROCESSOR_ARCHITECTURE_ALPHA64:
strcat(szSKU, "Alpha 64-bit ");
break;
case PROCESSOR_ARCHITECTURE_AMD64:
strcat(szSKU, "x64 ");
break;
case PROCESSOR_ARCHITECTURE_UNKNOWN:
strcat(szSKU, "Unknown ");
break;
}
strcat(szSKU, "Edition");
} else { // Test for specific product on Windows NT 4.0 SP5 and earlier
HKEY hKey;
char szProductType[80];
DWORD dwBufLen=sizeof(szProductType);
LONG lRet;
lRet = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
"SYSTEM\\CurrentControlSet\\Control\\ProductOptions",
0, KEY_QUERY_VALUE, &hKey );
if( lRet != ERROR_SUCCESS )
return FALSE;
lRet = RegQueryValueEx( hKey, "ProductType", NULL, NULL,
(LPBYTE) szProductType, &dwBufLen);
if( (lRet != ERROR_SUCCESS) || (dwBufLen > 80) )
return FALSE;
RegCloseKey( hKey );
if ( lstrcmpi( "WINNT", szProductType) == 0 ) {
strcpy( szSKU, "Workstation Edition" );
} if ( lstrcmpi( "LANMANNT", szProductType) == 0 ) {
strcpy( szSKU, "Server Edition" );
} if ( lstrcmpi( "SERVERNT", szProductType) == 0 ) {
strcpy( szSKU, "Advanced Server Edition" );
}
}
// Display service pack (if any) and build number.
if( osvi.dwMajorVersion == 4 && lstrcmpi( osvi.szCSDVersion, "Service Pack 6" ) == 0 )
{
HKEY hKey;
LONG lRet;
// Test for SP6 versus SP6a.
lRet = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Hotfix\\Q246009",
0, KEY_QUERY_VALUE, &hKey );
if( lRet == ERROR_SUCCESS ) {
strcpy( szServicePack, ", " );
strcat( szServicePack, "Service Pack 6a" );
} else {// Windows NT 4.0 prior to SP6a
if ( strlen(osvi.szCSDVersion) > 0 ) {
strcpy( szServicePack, ", " );
strcat( szServicePack, osvi.szCSDVersion );
}
}
RegCloseKey( hKey );
} else { // Windows NT 3.51 and earlier or Windows 2000 and later
if ( strlen(osvi.szCSDVersion) > 0 ) {
strcpy( szServicePack, ", " );
strcat( szServicePack, osvi.szCSDVersion );
}
}
break;
// Test for the Windows 95 product family.
case VER_PLATFORM_WIN32_WINDOWS:
if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 0) {
if ( osvi.szCSDVersion[1] == 'C' || osvi.szCSDVersion[1] == 'B' )
strcpy( szServicePack, "OSR2" );
}
if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 10) {
if ( osvi.szCSDVersion[1] == 'A' )
strcpy( szServicePack, "SE" );
}
break;
}
snprintf( os_version, os_version_size, "%s%s%s", szSKU, szServicePack, szVersion );
return 0;
}
// Handle the cpuid instruction on supported compilers
// NOTE: This only handles structured exceptions with Microsoft compilers.
//
int get_cpuid(unsigned int info_type, unsigned int& a, unsigned int& b, unsigned int& c, unsigned int& d) {
#ifdef _MSC_VER
// Microsoft compiler - use intrinsic
int retval = 1;
int CPUInfo[4] = {0,0,0,0};
__try {
__cpuid(CPUInfo, info_type);
a = CPUInfo[0];
b = CPUInfo[1];
c = CPUInfo[2];
d = CPUInfo[3];
retval = 0;
}
__except (EXCEPTION_EXECUTE_HANDLER) {}
return retval;
#elif defined(__GNUC__)
// GCC compiler
__asm__ __volatile__ ("cpuid": "=a" (ax), "=b" (bx), "=c" (cx), "=d" (dx) : "a" (info_type));
return 0;
#else
return 1;
#endif
}
// Returns the processor vendor.
// see: http://msdn.microsoft.com/en-us/library/hskdteyh.aspx
// see: http://www.intel.com/Assets/PDF/appnote/241618.pdf
// see: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/25481.pdf
int get_processor_vendor(char* name, int name_size) {
unsigned int eax = 0, ebx = 0, ecx = 0, edx = 0;
if (!name) return ERR_INVALID_PARAM;
if (name_size < 13) return ERR_WRONG_SIZE;
memset(name, 0, sizeof(name_size));
if (!get_cpuid(0x00000000, eax, ebx, ecx, edx)) {
*((int*)(name + 0)) = ebx;
*((int*)(name + 4)) = edx;
*((int*)(name + 8)) = ecx;
*((int*)(name + 12)) = '\0';
}
return 0;
}
// Returns the processor family, model, stepping.
// see: http://msdn.microsoft.com/en-us/library/hskdteyh.aspx
// see: http://www.intel.com/Assets/PDF/appnote/241618.pdf
// see: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/25481.pdf
int get_processor_version(int& family, int& model, int& stepping) {
unsigned int eax = 0, ebx = 0, ecx = 0, edx = 0;
get_cpuid(0x00000000, eax, ebx, ecx, edx);
if (!(eax == 0x00000001)) return ERR_NOT_IMPLEMENTED;
if (!get_cpuid(0x00000001, eax, ebx, ecx, edx)) {
family = (((eax >> 8) + (eax >> 20)) & 0xff);
model = (((((eax >> 16) & 0xf) << 4) + ((eax >> 4) & 0xf)) & 0xff);
stepping = (eax & 0xf);
}
return 0;
}
// Returns the processor name.
// see: http://msdn.microsoft.com/en-us/library/hskdteyh.aspx
// see: http://www.intel.com/Assets/PDF/appnote/241618.pdf
// see: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/25481.pdf
int get_processor_name(char* name, int name_size) {
unsigned int eax = 0, ebx = 0, ecx = 0, edx = 0;
size_t i = 0;
if (!name) return ERR_INVALID_PARAM;
if (name_size < 48) return ERR_WRONG_SIZE;
memset(name, 0, sizeof(name_size));
get_cpuid(0x80000000, eax, ebx, ecx, edx);
if (!(eax >= 0x80000004)) return ERR_NOT_IMPLEMENTED;
if (!get_cpuid(0x80000002, eax, ebx, ecx, edx)) {
*((int*)(name + 0)) = eax;
*((int*)(name + 4)) = ebx;
*((int*)(name + 8)) = ecx;
*((int*)(name + 12)) = edx;
}
if (!get_cpuid(0x80000003, eax, ebx, ecx, edx)) {
*((int*)(name + 16)) = eax;
*((int*)(name + 20)) = ebx;
*((int*)(name + 24)) = ecx;
*((int*)(name + 28)) = edx;
}
if (!get_cpuid(0x80000004, eax, ebx, ecx, edx)) {
*((int*)(name + 32)) = eax;
*((int*)(name + 36)) = ebx;
*((int*)(name + 40)) = ecx;
*((int*)(name + 44)) = edx;
}
// Old processors contain junk in the brand string, Intel's documentation
// doesn't mention this at all. So before returning, change all non-ascii
// characters to spaces.
for (i = 0; i < strlen(name); i++) {
if (!isprint(*(name + i))) {
*(name + i) = ' ';
}
}
return 0;
}
// Returns the processor cache.
// see: http://msdn.microsoft.com/en-us/library/hskdteyh.aspx
// see: http://www.intel.com/Assets/PDF/appnote/241618.pdf
// see: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/25481.pdf
int get_processor_cache(int& cache) {
unsigned int eax = 0, ebx = 0, ecx = 0, edx = 0;
get_cpuid(0x80000000, eax, ebx, ecx, edx);
if (!(eax >= 0x80000006)) return ERR_NOT_IMPLEMENTED;
if (!get_cpuid(0x80000006, eax, ebx, ecx, edx)) {
cache = ((ecx >> 16) & 0xffff) * 1024;
}
return 0;
}
// Returns the features supported by the processor, use the
// Linux CPU processor feature mnemonics.
// see: http://msdn.microsoft.com/en-us/library/hskdteyh.aspx
// see: http://www.intel.com/Assets/PDF/appnote/241618.pdf
// see: http://www.amd.com/us-en/assets/content_type/white_papers_and_tech_docs/25481.pdf
#define FEATURE_TEST(feature_set_supported, test, feature_name) \
if (feature_set_supported && test) strncat(features, feature_name, features_size - strlen(features))
int get_processor_features(char* vendor, char* features, int features_size) {
unsigned int std_eax = 0, std_ebx = 0, std_ecx = 0, std_edx = 0;
unsigned int ext_eax = 0, ext_ebx = 0, ext_ecx = 0, ext_edx = 0;
unsigned int std_supported = 0, ext_supported = 0, intel_supported = 0, amd_supported = 0;
if (!vendor) return ERR_INVALID_PARAM;
if (!features) return ERR_INVALID_PARAM;
if (features_size < 250) return ERR_WRONG_SIZE;
memset(features, 0, sizeof(features_size));
if (strcmp(vendor, "GenuineIntel") == 0) {
intel_supported = 1;
}
if (strcmp(vendor, "AuthenticAMD") == 0) {
amd_supported = 1;
}
get_cpuid(0x00000000, std_eax, std_ebx, std_ecx, std_edx);
if (std_eax >= 0x00000001) {
std_supported = 1;
get_cpuid(0x00000001, std_eax, std_ebx, std_ecx, std_edx);
}
get_cpuid(0x80000000, ext_eax, ext_ebx, ext_ecx, ext_edx);
if (ext_eax >= 0x80000001) {
ext_supported = 1;
get_cpuid(0x80000001, ext_eax, ext_ebx, ext_ecx, ext_edx);
}
FEATURE_TEST(std_supported, (std_edx & (1 << 0)), "fpu ");
FEATURE_TEST(std_supported, (std_edx & (1 << 1)), "vme ");
FEATURE_TEST(std_supported, (std_edx & (1 << 2)), "de ");
FEATURE_TEST(std_supported, (std_edx & (1 << 3)), "pse ");
FEATURE_TEST(std_supported, (std_edx & (1 << 4)), "tsc ");
FEATURE_TEST(std_supported, (std_edx & (1 << 5)), "msr ");
FEATURE_TEST(std_supported, (std_edx & (1 << 6)), "pae ");
FEATURE_TEST(std_supported, (std_edx & (1 << 7)), "mce ");
FEATURE_TEST(std_supported, (std_edx & (1 << 8)), "cx8 ");
FEATURE_TEST(std_supported, (std_edx & (1 << 9)), "apic ");
FEATURE_TEST(std_supported, (std_edx & (1 << 11)), "sep ");
FEATURE_TEST(std_supported, (std_edx & (1 << 12)), "mtrr ");
FEATURE_TEST(std_supported, (std_edx & (1 << 13)), "pge ");
FEATURE_TEST(std_supported, (std_edx & (1 << 14)), "mca ");
FEATURE_TEST(std_supported, (std_edx & (1 << 15)), "cmov ");
FEATURE_TEST(std_supported, (std_edx & (1 << 16)), "pat ");
FEATURE_TEST(std_supported, (std_edx & (1 << 17)), "pse36 ");
FEATURE_TEST(std_supported, (std_edx & (1 << 18)), "psn ");
FEATURE_TEST(std_supported, (std_edx & (1 << 19)), "clflush ");
FEATURE_TEST(std_supported, (std_edx & (1 << 21)), "dts ");
FEATURE_TEST(std_supported, (std_edx & (1 << 22)), "acpi ");
FEATURE_TEST(std_supported, (std_edx & (1 << 23)), "mmx ");
FEATURE_TEST(std_supported, (std_edx & (1 << 24)), "fxsr ");
FEATURE_TEST(std_supported, (std_edx & (1 << 25)), "sse ");
FEATURE_TEST(std_supported, (std_edx & (1 << 26)), "sse2 ");
FEATURE_TEST(std_supported, (std_edx & (1 << 27)), "ss ");
FEATURE_TEST(std_supported, (std_edx & (1 << 28)), "htt ");
FEATURE_TEST(std_supported, (std_edx & (1 << 29)), "tm ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 0)), "pni ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 9)), "ssse3 ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 13)), "cx16 ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 19)), "sse4_1 ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 20)), "sse4_2 ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 11)), "syscall ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 20)), "nx ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 29)), "lm ");
if (intel_supported) {
// Intel only features
FEATURE_TEST(std_supported, (std_ecx & (1 << 5)), "vmx ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 6)), "smx ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 8)), "tm2 ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 12)), "fma ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 18)), "dca ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 22)), "movebe ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 23)), "popcnt ");
FEATURE_TEST(std_supported, (std_ecx & (1 << 25)), "aes ");
FEATURE_TEST(std_supported, (std_edx & (1 << 31)), "pbe ");
}
if (amd_supported) {
// AMD only features
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 2)), "svm ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 6)), "sse4a ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 9)), "osvw ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 10)), "ibs ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 11)), "xop ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 12)), "skinit ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 13)), "wdt ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 15)), "lwp ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 16)), "fma4 ");
FEATURE_TEST(ext_supported, (ext_ecx & (1 << 18)), "cvt16 ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 26)), "page1gb ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 27)), "rdtscp ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 30)), "3dnowext ");
FEATURE_TEST(ext_supported, (ext_edx & (1 << 31)), "3dnow ");
}
strip_whitespace(features);
return 0;
}
// Returns the CPU count
//
int get_processor_count(int& processor_count) {
SYSTEM_INFO SystemInfo;
memset( &SystemInfo, NULL, sizeof( SystemInfo ) );
::GetSystemInfo( &SystemInfo );
processor_count = SystemInfo.dwNumberOfProcessors;
return 0;
}
// Returns the processor make, model, and additional cpu flags supported by
// the processor, use the Linux CPU processor feature descriptions.
//
int get_processor_info(
char* p_vendor, int p_vendor_size, char* p_model, int p_model_size,
char* p_features, int p_features_size, double& p_cache, int& p_ncpus
)
{
int family = 0, model = 0, stepping = 0, cache = 0;
char vendor_name[256], processor_name[256], features[256];
get_processor_vendor(vendor_name, sizeof(vendor_name));
get_processor_version(family, model, stepping);
get_processor_name(processor_name, sizeof(processor_name));
get_processor_cache(cache);
get_processor_features(vendor_name, features, sizeof(features));
get_processor_count(p_ncpus);
snprintf(p_vendor, p_vendor_size,
"%s",
vendor_name
);
snprintf(p_model, p_model_size,
"%s [Family %d Model %d Stepping %d]",
processor_name, family, model, stepping
);
snprintf(p_features, p_features_size,
"%s",
features
);
p_cache = (double)cache;
return 0;
}
// detect the network usage totals for the host.
//
int get_network_usage_totals(unsigned int& total_received, unsigned int& total_sent) {
// Declare and initialize variables.
int i;
int iRetVal = 0;
DWORD dwSize = 0;
MIB_IFTABLE* pIfTable;
MIB_IFROW* pIfRow;
// Allocate memory for our pointers.
pIfTable = (MIB_IFTABLE*)malloc(sizeof(MIB_IFTABLE));
if (pIfTable == NULL) {
return ERR_MALLOC;
}
// Make an initial call to GetIfTable to get the
// necessary size into dwSize
dwSize = sizeof(MIB_IFTABLE);
if (GetIfTable(pIfTable, &dwSize, FALSE) == ERROR_INSUFFICIENT_BUFFER) {
free(pIfTable);
pIfTable = (MIB_IFTABLE*)malloc(dwSize);
if (pIfTable == NULL) {
return ERR_MALLOC;
}
}
// Make a second call to GetIfTable to get the actual
// data we want.
iRetVal = (int)GetIfTable(pIfTable, &dwSize, FALSE);
if (iRetVal == NO_ERROR) {
for (i = 0; i < (int)pIfTable->dwNumEntries; i++) {
pIfRow = (MIB_IFROW *) & pIfTable->table[i];
if (IF_TYPE_SOFTWARE_LOOPBACK != pIfRow->dwType) {
total_received += pIfRow->dwInOctets;
total_sent += pIfRow->dwOutOctets;
}
}
}
if (pIfTable != NULL) {
free(pIfTable);
pIfTable = NULL;
}
return iRetVal;
}
// Gets host information; called on startup and before each sched RPC
//
int HOST_INFO::get_host_info() {
get_timezone(timezone);
get_filesystem_info(d_total, d_free);
get_memory_info(m_nbytes, m_swap);
get_os_information(
os_name, sizeof(os_name), os_version, sizeof(os_version)
);
get_processor_info(
p_vendor, sizeof(p_vendor),
p_model, sizeof(p_model),
p_features, sizeof(p_features),
m_cache,
p_ncpus
);
get_local_network_info();
if (!strlen(host_cpid)) {
generate_host_cpid();
}
return 0;
}
bool HOST_INFO::host_is_running_on_batteries() {
SYSTEM_POWER_STATUS pStatus;
ZeroMemory(&pStatus, sizeof(SYSTEM_POWER_STATUS));
if (!GetSystemPowerStatus(&pStatus)) {
return false;
}
// Sometimes the system reports the ACLineStatus as an
// undocumented value, so lets check to see if the
// battery is charging or missing and make that part
// of the decision.
bool bIsOnBatteryPower = (pStatus.ACLineStatus != 1);
bool bIsBatteryCharging = ((pStatus.BatteryFlag & 8) == 8);
bool bIsBatteryMissing = ((pStatus.BatteryFlag & 128) == 128);
return (bIsOnBatteryPower && !bIsBatteryCharging && !bIsBatteryMissing);
}
bool HOST_INFO::users_idle(bool /*check_all_logins*/, double idle_time_to_run) {
double seconds_idle = get_idle_tick_count() / 1000;
double seconds_time_to_run = 60 * idle_time_to_run;
return seconds_idle > seconds_time_to_run;
}
Reference in New Issue View Git Blame Copy Permalink