代码碎片（1）

// 将进程名字转换为进程ID, 返回非0表示成功，0表示失败
DWORD WINAPI ExeName2PID(LPCTSTR lpszExeName)
{
PROCESSENTRY32 pe32;
pe32.dwSize = sizeof(pe32);
HANDLE hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if(INVALID_HANDLE_VALUE != hSnapShot)
{
BOOL findResult;
findResult = Process32First(hSnapShot, &pe32);
while(findResult)
{
if(_tcscmp(pe32.szExeFile, lpszExeName) == 0)
{
CloseHandle(hSnapShot);
return pe32.th32ProcessID;
}
findResult = Process32Next(hSnapShot, &pe32);
}
}
CloseHandle(hSnapShot);
return 0;
}

SendMessage(FindWindow(0,0), WM_SYSCOMMAND, SC_MONITORPOWER, 1);

int CmdEcho(LPCTSTR lpszCmd, CString &strOutput)
{
SECURITY_ATTRIBUTES sa;
HANDLE hRead,hWrite;

sa.nLength = sizeof(SECURITY_ATTRIBUTES);
sa.lpSecurityDescriptor = NULL; //使用系统默认的安全描述符
sa.bInheritHandle = TRUE; //创建的进程继承句柄
if (!CreatePipe(&hRead,&hWrite,&sa,0)) //创建匿名管道
{
return FALSE;
}

STARTUPINFO si;
PROCESS_INFORMATION pi;

ZeroMemory(&si,sizeof(STARTUPINFO));
si.cb = sizeof(STARTUPINFO);
GetStartupInfo(&si);
si.hStdError = hWrite;
si.hStdOutput = hWrite;    //新创建进程的标准输出连在写管道一端
si.wShowWindow = SW_HIDE; //隐藏窗口
si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;

char cmdline[200];
sprintf(cmdline,"cmd /C %s",lpszCmd);
if (!CreateProcess(NULL,cmdline,NULL,NULL,TRUE,NULL,NULL,NULL,&si,&pi)) //创建子进程
{
return FALSE;
}

CloseHandle(hWrite); //关闭管道句柄

char buffer[4096] = {0};
DWORD bytesRead;

while (true)
{
if (ReadFile(hRead, buffer, 4095, &bytesRead, NULL) == NULL) //读取管道
break;

strOutput += buffer;
Sleep(10);
}
CloseHandle(hRead);
return TRUE;
}

BOOL WINAPI AdjustPrivileges(LPCSTR lpName = SE_DEBUG_NAME, DWORD dwAttrib = SE_PRIVILEGE_ENABLED )
{
HANDLE hToken;
TOKEN_PRIVILEGES tkp;
BOOL bResult=FALSE;
//打开当前进程信令
if(!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&hToken))
return bResult;
//调整权限
LookupPrivilegeValue(NULL,lpName,&tkp.Privileges[0].Luid);
tkp.PrivilegeCount=1;
tkp.Privileges[0].Attributes = dwAttrib;
//通知系统修改权限
AdjustTokenPrivileges(hToken,FALSE,&tkp,0,(PTOKEN_PRIVILEGES)NULL,0);
if(GetLastError() == ERROR_SUCCESS)
bResult=TRUE;
return bResult;
}

char GetIdCardChkSum(char* szIdCard)
{
if (NULL == szIdCard || strlen(szIdCard) < 17)
{
return 'E';
}

char szVerifyMap[] = "10X98765432";  // 加权乘积求和除以11的余数所对应的校验数
char szFactor[] = {7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2, 1};  // 加权因子
long sum = 0l;  //加权乘积求和
int  m = 0;  // 加权乘积求和的模数
char *p = szIdCard;  // 当前位置

while(*p)  // 在 '\0' 之前一直成立
{

sum += (*p - '0') * szFactor[p - szIdCard];  // 加权乘积求和
p++;  // 当前位置增加1
}

m = sum % 11;  // 取模
return  szVerifyMap[m];
}

LPBYTE MapFile(LPCTSTR szFile, LPDWORD lpSize)
{
HANDLE hFile;
hFile = ::CreateFile(szFile,
GENERIC_READ,
0,
NULL,
OPEN_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL
);

if (INVALID_HANDLE_VALUE != hFile)
{
DWORD dwFileSize;
dwFileSize = GetFileSize(hFile, NULL);
if (NULL != lpSize)
{
*lpSize = dwFileSize;
}
if (dwFileSize > 0)
{
HANDLE hFileMapping;
hFileMapping = ::CreateFileMapping(hFile, NULL,	PAGE_READONLY, 0, 0, NULL);
::CloseHandle(hFile);
if (NULL != hFileMapping)
{
LPBYTE pbFile = (PBYTE)::MapViewOfFile(hFileMapping, FILE_MAP_READ, 0, 0, 0);
::CloseHandle(hFileMapping);
if (NULL != hFileMapping)
{
return pbFile;
//::UnmapViewOfFile(pbFile);
}
}
}

}
return NULL;
}

void UnmapFile(LPVOID lpFile)
{
if (NULL != lpFile)
{
::UnmapViewOfFile(lpFile);
}
}

CString GetCPUID()
{
CString strCpuId, strId1, strId2;
ULONG   s1, s2;

__asm
{
mov eax, 01h   //eax=1:取CPU序列号
xor edx, edx
cpuid
mov  s1, edx
mov  s2, eax
}
strId1.Format("%08X%08X", s1, s2);

__asm
{
mov eax, 03h
xor ecx, ecx
xor edx, edx
cpuid
mov  s1, edx
mov  s2, ecx
}
strId2.Format("%08X%08X", s1, s2);

strCpuId = strId1 + strId2;
return strCpuId;
}