// // machinfo.cpp - SHGetMachineInfo and related functions // // #include "priv.h" #include #include #include const GUID GUID_DEVICE_BATTERY = { 0x72631e54L, 0x78A4, 0x11d0, { 0xbc, 0xf7, 0x00, 0xaa, 0x00, 0xb7, 0xb3, 0x2a } }; /***************************************************************************** * * DOCK STATE - Win95, Win98, and WinNT all do this differently (yuck) * *****************************************************************************/ C_ASSERT(GMID_NOTDOCKABLE == CM_HWPI_NOT_DOCKABLE); C_ASSERT(GMID_UNDOCKED == CM_HWPI_UNDOCKED); C_ASSERT(GMID_DOCKED == CM_HWPI_DOCKED); DWORD GetDockedStateNT() { HW_PROFILE_INFO hpi; DWORD Result = GMID_NOTDOCKABLE; // assume the worst if (GetCurrentHwProfile(&hpi)) { Result = hpi.dwDockInfo & (DOCKINFO_UNDOCKED | DOCKINFO_DOCKED); // Wackiness: If the machine does not support docking, then // NT returns >both< flags set. Go figure. if (Result == (DOCKINFO_UNDOCKED | DOCKINFO_DOCKED)) { Result = GMID_NOTDOCKABLE; } } else { TraceMsg(DM_WARNING, "GetDockedStateNT: GetCurrentHwProfile failed"); } return Result; } // // Platforms that do not support Win95/Win98 can just call the NT version. // #define GetDockedState() GetDockedStateNT() /***************************************************************************** * * BATTERY STATE - Once again, Win95 and Win98 and NT all do it differently * *****************************************************************************/ // // Values for SYSTEM_POWER_STATUS.ACLineStatus // #define SPSAC_OFFLINE 0 #define SPSAC_ONLINE 1 // // Values for SYSTEM_POWER_STATUS.BatteryFlag // #define SPSBF_NOBATTERY 128 // // So many ways to detect batteries, so little time... // DWORD GetBatteryState() { // // Since GMIB_HASBATTERY is cumulative (any battery turns it on) // and GMIB_ONBATTERY is subtractive (any AC turns it off), the // state you have to start in before you find a battery is // GMIB_HASBATTERY off and GMIB_ONBATTERY on. // // dwResult & GMIB_ONBATTERY means we have yet to find AC power. // dwResult & GMIB_HASBATTERY means we have found a non-UPS battery. // DWORD dwResult = GMIB_ONBATTERY; //------------------------------------------------------------------ // // First try - IOCTL_BATTERY_QUERY_INFORMATION // //------------------------------------------------------------------ // // Windows 98 and Windows 2000 support IOCTL_BATTERY_QUERY_INFORMATION, // which lets us enumerate the batteries and ask each one for information. // Except that on Windows 98, we can enumerate only ACPI batteries. // We still have to use VPOWERD to enumerate APM batteries. // FEATURE -- deal with Win98 APM batteries HDEVINFO hdev = SetupDiGetClassDevs(&GUID_DEVICE_BATTERY, 0, 0, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE); if (hdev != INVALID_HANDLE_VALUE) { SP_DEVICE_INTERFACE_DATA did; did.cbSize = sizeof(did); // Stop at 100 batteries so we don't go haywire for (int idev = 0; idev < 100; idev++) { // Pre-set the error code because our DLLLOAD wrapper doesn't // and Windows NT 4 supports SetupDiGetClassDevs but not // SetupDiEnumDeviceInterfaces (go figure). SetLastError(ERROR_NO_MORE_ITEMS); if (SetupDiEnumDeviceInterfaces(hdev, 0, &GUID_DEVICE_BATTERY, idev, &did)) { DWORD cbRequired = 0; /* * Ask for the required size then allocate it then fill it. * * Sigh. Windows NT and Windows 98 implement * SetupDiGetDeviceInterfaceDetail differently if you are * querying for the buffer size. * * Windows 98 returns FALSE, and GetLastError() returns * ERROR_INSUFFICIENT_BUFFER. * * Windows NT returns TRUE. * * So we allow the cases either where the call succeeds or * the call fails with ERROR_INSUFFICIENT_BUFFER. */ if (SetupDiGetDeviceInterfaceDetail(hdev, &did, 0, 0, &cbRequired, 0) || GetLastError() == ERROR_INSUFFICIENT_BUFFER) { PSP_DEVICE_INTERFACE_DETAIL_DATA pdidd; pdidd = (PSP_DEVICE_INTERFACE_DETAIL_DATA)LocalAlloc(LPTR, cbRequired); if (pdidd) { pdidd->cbSize = sizeof(*pdidd); if (SetupDiGetDeviceInterfaceDetail(hdev, &did, pdidd, cbRequired, &cbRequired, 0)) { /* * Finally enumerated a battery. Ask it for information. */ HANDLE hBattery = CreateFile(pdidd->DevicePath, GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); if (hBattery != INVALID_HANDLE_VALUE) { /* * Now you have to ask the battery for its tag. */ BATTERY_QUERY_INFORMATION bqi; DWORD dwWait = 0; DWORD dwOut; bqi.BatteryTag = 0; if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_TAG, &dwWait, sizeof(dwWait), &bqi.BatteryTag, sizeof(bqi.BatteryTag), &dwOut, NULL) && bqi.BatteryTag) { /* * With the tag, you can query the battery info. */ BATTERY_INFORMATION bi; bqi.InformationLevel = BatteryInformation; bqi.AtRate = 0; if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_INFORMATION, &bqi, sizeof(bqi), &bi, sizeof(bi), &dwOut, NULL)) { // Only system batteries count if (bi.Capabilities & BATTERY_SYSTEM_BATTERY) { if (!(bi.Capabilities & BATTERY_IS_SHORT_TERM)) { dwResult |= GMIB_HASBATTERY; } /* * And then query the battery status. */ BATTERY_WAIT_STATUS bws; BATTERY_STATUS bs; ZeroMemory(&bws, sizeof(bws)); bws.BatteryTag = bqi.BatteryTag; if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_STATUS, &bws, sizeof(bws), &bs, sizeof(bs), &dwOut, NULL)) { if (bs.PowerState & BATTERY_POWER_ON_LINE) { dwResult &= ~GMIB_ONBATTERY; } } } } } CloseHandle(hBattery); } } LocalFree(pdidd); } } } else { // Enumeration failed - perhaps we're out of items if (GetLastError() == ERROR_NO_MORE_ITEMS) break; } } SetupDiDestroyDeviceInfoList(hdev); } // // Final cleanup: If we didn't find a battery, then presume that we // are on AC power. // if (!(dwResult & GMIB_HASBATTERY)) dwResult &= ~GMIB_ONBATTERY; return dwResult; } /***************************************************************************** * * TERMINAL SERVER CLIENT * * This is particularly gruesome because Terminal Server for NT4 SP3 goes * to extraordinary lengths to prevent you from detecting it. Even the * semi-documented NtCurrentPeb()->SessionId trick doesn't work on NT4 SP3. * So we have to go to the totally undocumented winsta.dll to find out. * *****************************************************************************/ BOOL IsTSClient(void) { // NT5 has a new system metric to detect this return GetSystemMetrics(SM_REMOTESESSION); } /***************************************************************************** * * SHGetMachineInfo * *****************************************************************************/ // // SHGetMachineInfo // // Given an index, returns some info about that index. See shlwapi.w // for documentation on the flags available. // STDAPI_(DWORD_PTR) SHGetMachineInfo(UINT gmi) { switch (gmi) { case GMI_DOCKSTATE: return GetDockedState(); case GMI_BATTERYSTATE: return GetBatteryState(); // // It smell like a laptop if it has a battery or if it can be docked. // case GMI_LAPTOP: return (GetBatteryState() & GMIB_HASBATTERY) || (GetDockedState() != GMID_NOTDOCKABLE); case GMI_TSCLIENT: return IsTSClient(); } TraceMsg(DM_WARNING, "SHGetMachineInfo: Unknown info query %d", gmi); return 0; }