cpuid.cpp

取计算机硬件信息的算法、包括CPU、BIOS、HARDID、MAC

// CPUID.cpp: implementation of the CCPUID class.
//
//////////////////////////////////////////////////////////////////////

#include "StdAfx.h"
#include "CPUID.h"

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CCPUID::CCPUID()
{
	
}

CCPUID::~CCPUID()
{
	
}

BOOL CCPUID::GetFirstCPUID(LPSTR szCPUID)
{
	m_hCurrentProcessHandle = GetCurrentProcess();
	GetProcessAffinityMask(m_hCurrentProcessHandle, &m_dwProcessAffinity,&m_dwSystemAffinity);
	
	// Check if available process affinity mask is equal to the
	// available system affinity mask
	if(m_dwProcessAffinity!=m_dwSystemAffinity)
		return FALSE;

	m_dwAffinityMask=1;

	return GetNextCPUID(szCPUID);
}

BOOL CCPUID::GetNextCPUID(LPSTR szCPUID)
{
	if(m_dwAffinityMask!=0 && m_dwAffinityMask<=m_dwProcessAffinity)
	{
		// Check if this CPU is available
		if(m_dwAffinityMask & m_dwProcessAffinity)
		{
			if(SetProcessAffinityMask(m_hCurrentProcessHandle,m_dwAffinityMask))
			{
				Sleep(0); // Give OS time to switch CPU
			}
		}

		GetCPUID(szCPUID);
		
		m_dwAffinityMask=m_dwAffinityMask<<1;
		return TRUE;
	}
	else
	{
		// Reset the processor affinity
		SetProcessAffinityMask(m_hCurrentProcessHandle,m_dwProcessAffinity);
		return FALSE;
	}
}

void CCPUID::GetCPUID(LPSTR szCUPID)
{
	DWORD x[]={0,0,0};
	_asm{
		mov eax,1
		cpuid
		mov dword ptr x,eax
		mov eax,3
		cpuid
		mov x+4,edx
		mov x+8,ecx
	}

	sprintf(szCUPID,"%08X-%08X-%08X",x[0],x[1],x[2]);
}