工具函数（代码块的大小，代码块起始地址，提升进程权限）

一些在编程中经常要用到的功能编写成函数，方便使用.

[cpp] view plain copy

1. #include <windows.h>
2. //系统类型
3. typedef enum SystemType
4. {
5. WINDOWS_2000 = 1, //5.0
6. WINDOWS_XP, //5.1
7. WINDOWS__SERVER_2003, //5.2
8. WINDOWS__SERVER_2003_R2, //5.2
9. WINDOWS_VISTA, //6.0
10. WINDOWS__SERVER_2008, //6.0
11. WINDOWS__SERVER_2008_R2, //6.1
12. WINDOWS_7, //6.1
13. WINDOWS_SERVER_2012,//6.2
14. WINDOWS_8//6.2
15. }ST;
16. /*******获取系统类型*******/
17. ST GetSystemType()
18. {
19. OSVERSIONINFO osvi;
20. ST st;
21. ZeroMemory(&osvi, sizeof(OSVERSIONINFO));
22. osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
23. GetVersionEx(&osvi);
24. if (osvi.dwMajorVersion==5)
25. {
26. if (osvi.dwMinorVersion==0)
27. {
28. st = WINDOWS_2000; //5.0
29. }
30. else if (osvi.dwMinorVersion==1)
31. {
32. st = WINDOWS_XP; //5.1
33. }
34. else if (osvi.dwMinorVersion==2)
35. {
36. st = WINDOWS__SERVER_2003; //5.2
37. }
38. }
39. else if(osvi.dwMajorVersion==6)
40. {
41. if (osvi.dwMinorVersion==0)
42. {
43. st = WINDOWS_VISTA; //6.0
44. }
45. else if (osvi.dwMinorVersion==1)
46. {
47. st = WINDOWS_7; //6.1
48. }
49. else if (osvi.dwMinorVersion==2)
50. {
51. st = WINDOWS_8; //6.2
52. }
53. }
54. return st;
55. }
56. /*找到代码块的大小*/
57. DWORD GetCodeSize(HANDLE hModule)
58. {
59. if (!hModule) return 0;
60. unsigned int nSize = 0;
61. unsigned int v1;
62. DWORD result;
63. DWORD p = (DWORD)hModule;
64. if ( *(WORD *)p == 0x5A4D && (v1 = p + *(DWORD *)(p + 0x3c), *(WORD *)v1 == 0x4550) )
65. result = *(DWORD *)(v1 + 0x1c);
66. else
67. result = 0;
68. return result;
69. }
70. /*找到代码块起始地址*/
71. DWORD GetCodeStart(HANDLE hModule)
72. {
73. if (!hModule) return 0;
74. unsigned int nSize = 0;
75. unsigned int v1;
76. DWORD result;
77. DWORD p = (DWORD)hModule;
78. if ( *(WORD *)p == 0x5A4D && (v1 = p + *(DWORD *)(p + 0x3c), *(WORD *)v1 == 0x4550) )
79. result = *(DWORD *)(v1 + 0x2c);
80. else
81. result = 0;
82. return (DWORD)hModule + result;
83. }
84. /*搜索代码特征，找到挂钩地址*/
85. BYTE* SearchOpcode(BYTE* pOpcodeBytes, unsigned int nOpcodeLen, const char* pLibName)
86. {
87. BYTE * pCode = pOpcodeBytes;
88. int nCodeLen = nOpcodeLen;
89. HANDLE hModule;
90. hModule = GetModuleHandleA(pLibName);
91. DWORD dwCodeSize = GetCodeSize(hModule);
92. DWORD dwCodeStart = GetCodeStart(hModule);
93. for (DWORD i = 0; i<dwCodeSize; i++)
94. {
95. BYTE* p = (BYTE*)dwCodeStart+i;
96. int j;
97. for (j=0; j<nCodeLen; j++)
98. {
99. if (*(p+j)==pCode[j])
100. continue;
101. else
102. break;
103. }
104. if (j==nCodeLen)
105. return p;
106. else
107. continue;
108. }
109. return NULL;
110. }
111. BOOL Is64Bit_OS()
112. {
113. BOOL bRetVal = FALSE;
114. IsWow64Process(NULL, &bRetVal);
115. return bRetVal;
116. }

单字节转多字节：

[cpp] view plain copy

1. BOOL ThreadCheckFileTrust(CHAR* lpfileName)
2. {
3. if (lpFileName == NULL)
4. {
5. return TRUE;
6. }
7. HCATADMIN hCatAdmin = NULL;
8. DWORD dwNum = MultiByteToWideChar(CP_ACP,0,lpFileName,-1,NULL,0);
9. wchar_t* pcwszFile = new wchar_t[dwNum];
10. if (!pcwszFile)
11. {
12. return FALSE;
13. }
14. MultiByteToWideChar(CP_ACP,0,lpFileName,-1,pcwszFile,dwNum);
15. BOOL bRet =  ThreadCheckFileTrustW(pcwszFile);
16. delete pcwszFile;
17. return bRet;
18. }

根据进程名称枚举进程：

[cpp] view plain copy

1. static DWORD GetProcessID(wchar_t *ProcessName)  //获得进程PID
2. {
3. PROCESSENTRY32 pe32;
4. pe32.dwSize=sizeof(pe32);
5. //获得系统内所有进程快照
6. HANDLE hProcessSnap=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);
7. if(hProcessSnap==INVALID_HANDLE_VALUE)
8. {
9. OutputDebugStringA("CreateToolhelp32Snapshot error");
10. return 0;
11. }
12. //枚举列表中的第一个进程
13. BOOL bProcess=Process32First(hProcessSnap,&pe32);
14. while(bProcess)
15. {
16. //比较找到的进程名和我们要查找的进程名，一样则返回进程id
17. if(wcscmp(wcsupr(pe32.szExeFile),wcsupr(ProcessName))==0)
18. return pe32.th32ProcessID;
19. //继续查找
20. bProcess=Process32Next(hProcessSnap,&pe32);
21. }
22. CloseHandle(hProcessSnap);
23. return 0;
24. }

提升进程权限：

[cpp] view plain copy

1. static int EnableDebugPriv(const char * name)  //提升进程为DEBUG权限--"SeDebugPrivilege"
2. {
3. HANDLE hToken;
4. TOKEN_PRIVILEGES tp;
5. LUID luid;
6. if(!OpenProcessToken(GetCurrentProcess(),
7. TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,
8. &hToken) )
9. {
10. printf("OpenProcessToken error\n");
11. return 1;
12. }
13. if(!LookupPrivilegeValueA(NULL,name,&luid))
14. {
15. printf("LookupPrivilege error!\n");
16. }
17. tp.PrivilegeCount = 1;
18. tp.Privileges[0].Attributes =SE_PRIVILEGE_ENABLED;
19. tp.Privileges[0].Luid = luid;
20. if(!AdjustTokenPrivileges(hToken,0,&tp,sizeof(TOKEN_PRIVILEGES),NULL,NULL) )
21. {
22. printf("AdjustTokenPrivileges error!\n");
23. return 1;
24. }
25. return 0;
26. }

http://blog.csdn.net/chence19871/article/details/37881101