// EnumDevices.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <SetupAPI.h>
#pragma comment(lib, "setupapi.lib")

// SPDRP_DEVICEDESC
// SPDRP_CLASS
// SPDRP_FRIENDLYNAME

typedef BOOL (*TfEnumDevicesCallback)(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, PVOID lpContext);

VOID EnumDevices(TCHAR *DeviceClassName, TfEnumDevicesCallback Callback, PVOID CallbackContext)
{
	GUID Guid;
	DWORD cbOpt;

	if (SetupDiClassGuidsFromName(DeviceClassName, &Guid, 1, &cbOpt))
	{
		HDEVINFO hDev;

		hDev = SetupDiGetClassDevs(&Guid, NULL, HWND_DESKTOP, DIGCF_PRESENT);
		if (hDev != INVALID_HANDLE_VALUE)
		{
			DWORD MemberIndex;
			SP_DEVINFO_DATA sdd;

			MemberIndex = 0;
			sdd.cbSize = sizeof(sdd);
			while (SetupDiEnumDeviceInfo(hDev, MemberIndex, &sdd))
			{
				if (!Callback(hDev, &sdd, CallbackContext))
				{
					break;
				}
				MemberIndex++;
			}
			SetupDiDestroyDeviceInfoList(hDev);
		}
	}
	return;
}

BOOL GetDevicePropertyValue(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, DWORD Property, PVOID *lppResult, DWORD *lpbResult)
{
	PBYTE lpBuffer;
	DWORD cbBuffer;

	lpBuffer = NULL;
	cbBuffer = 0;
	if (!SetupDiGetDeviceRegistryProperty(DeviceInfoSet, DeviceInfoData, Property, NULL, lpBuffer, cbBuffer, &cbBuffer)
		&& (cbBuffer > 0))
	{
		if (lpBuffer = (PBYTE)GlobalAlloc(GPTR, cbBuffer))
		{
			if (SetupDiGetDeviceRegistryProperty(DeviceInfoSet, DeviceInfoData, Property, NULL, lpBuffer, cbBuffer, &cbBuffer))
			{
				*lppResult = lpBuffer;
				*lpbResult = cbBuffer;
				return TRUE;
			}
			GlobalFree(lpBuffer);
		}
	}
	return FALSE;
}

VOID FreeGetDevicePropertyValue(PVOID lpBuffer)
{
	GlobalFree(lpBuffer);
}

BOOL EnumNetCallback(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, PVOID CallbackContext)
{
	PVOID lpBusNumber;
	DWORD cbBusNumber;

	if (GetDevicePropertyValue(DeviceInfoSet, DeviceInfoData, SPDRP_BUSNUMBER, &lpBusNumber, &cbBusNumber))
	{
		if (*(DWORD*)lpBusNumber)
		{
			PVOID lpFriendlyName;
			DWORD cbFriendlyName;

			if (GetDevicePropertyValue(DeviceInfoSet, DeviceInfoData, SPDRP_FRIENDLYNAME, &lpFriendlyName, &cbFriendlyName))
			{
				_tprintf(TEXT("%s\n"), lpFriendlyName);
				FreeGetDevicePropertyValue(lpFriendlyName);
			}
		}
		FreeGetDevicePropertyValue(lpBusNumber);
	}
	return TRUE;
}

BOOL EnumCpuCallback(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, PVOID CallbackContext)
{
	PVOID lpFriendlyName;
	DWORD cbFriendlyName;

	if (GetDevicePropertyValue(DeviceInfoSet, DeviceInfoData, SPDRP_FRIENDLYNAME, &lpFriendlyName, &cbFriendlyName))
	{
		_tprintf(TEXT("%s\n"), lpFriendlyName);
		FreeGetDevicePropertyValue(lpFriendlyName);
	}
	return TRUE;
}

BOOL EnumDiskCallback(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, PVOID CallbackContext)
{
	PVOID lpFriendlyName;
	DWORD cbFriendlyName;

	if (GetDevicePropertyValue(DeviceInfoSet, DeviceInfoData, SPDRP_FRIENDLYNAME, &lpFriendlyName, &cbFriendlyName))
	{
		_tprintf(TEXT("%s\n"), lpFriendlyName);
		FreeGetDevicePropertyValue(lpFriendlyName);
	}
	return TRUE;
}

BOOL EnumDisplayCallback(HDEVINFO DeviceInfoSet, PSP_DEVINFO_DATA DeviceInfoData, PVOID CallbackContext)
{
	PVOID lpFriendlyName;
	DWORD cbFriendlyName;

	if (GetDevicePropertyValue(DeviceInfoSet, DeviceInfoData, SPDRP_DEVICEDESC, &lpFriendlyName, &cbFriendlyName))
	{
		_tprintf(TEXT("%s\n"), lpFriendlyName);
		FreeGetDevicePropertyValue(lpFriendlyName);
	}
	return TRUE;
}

int _tmain(int argc, _TCHAR* argv[])
{
	UCHAR Mac[8];

	_tsetlocale(LC_ALL, TEXT("chs"));

	_tprintf(TEXT("%s\n"), TEXT("Net"));
	EnumDevices(TEXT("Net"), EnumNetCallback, NULL);
	_tprintf(TEXT("\n"));
	_tprintf(TEXT("%s\n"), TEXT("Processor"));
	EnumDevices(TEXT("Processor"), EnumCpuCallback, NULL);
	_tprintf(TEXT("\n"));
	_tprintf(TEXT("%s\n"), TEXT("Display"));
	EnumDevices(TEXT("Display"), EnumDisplayCallback, NULL);
	return getchar();
}