You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
265 lines
11 KiB
265 lines
11 KiB
//
|
|
// machinfo.cpp - SHGetMachineInfo and related functions
|
|
//
|
|
//
|
|
|
|
#include "priv.h"
|
|
#include <dbt.h>
|
|
#include <cfgmgr32.h>
|
|
|
|
#include <batclass.h>
|
|
|
|
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;
|
|
}
|
|
|