Category

Development Platform
Home > SourceCode/Document > Windows Develop > Hook api > View Code
ProcessModule.h
Package: HookAPI　[view]
Upload User: nbcables
Upload Date: 2007-01-11
Package Size: 1243k
Code Size: 2k
Category:

Hook api

Development Platform:

Visual C++

ProcessModule.h：Code Content
  1. #ifndef __PROCESS_MODULE_H__
  2. #define __PROCESS_MODULE_H__
  4. /*#ifdef WINNT
  5. typedef BOOL (WINAPI *ENUMPROCESSES_PROC)(DWORD *lpidProcess, DWORD cb, DWORD *cbNeeded);
  6. typedef BOOL (WINAPI *ENUMPROCESSMODULES_PROC)(HANDLE hProcess, HMODULE *lphModule, DWORD cb, LPDWORD lpcbNeeded);
  7. typedef DWORD (WINAPI *GETMODULEBASENAME_PROC)(HANDLE hProcess, HMODULE hModule, LPTSTR lpBaseName, DWORD nSize);
  8. #else*/
  9. #include <tlhelp32.h>
  10. typedef HANDLE (WINAPI *CREATETOOLHELP32SNAPSHOT_PROC)(DWORD dwFlags, DWORD th32ProcessID);
  11. typedef BOOL (WINAPI *PROCESS32FIRST_PROC)(HANDLE hSnapshot, LPPROCESSENTRY32 lppe);
  12. typedef BOOL (WINAPI *PROCESS32NEXT_PROC)(HANDLE hSnapshot, LPPROCESSENTRY32 lppe);
  13. typedef BOOL (WINAPI *MODULE32FIRST_PROC)(HANDLE hSnapshot, LPMODULEENTRY32 lpme);
  14. typedef BOOL (WINAPI *MODULE32NEXT_PROC)(HANDLE hSnapshot, LPMODULEENTRY32 lpme);
  15. //#endif
  17. #define MAX_PROCESS_COUNT 512
  18. #define MAX_MODULE_COUNT 500
  20. class CProcessModule
  21. {
  22. public:
  23. CProcessModule();
  24. ~CProcessModule();
  26. private:
  27. /*#ifdef WINNT
  28. HINSTANCE m_hLib;  // from LoadLibrary("PSAPI.dll")
  29. ENUMPROCESSES_PROC m_pEnumProcesses;  // api EnumProcesses()
  30. ENUMPROCESSMODULES_PROC m_pEnumProcessModules;  // api EnumProcessModules()
  31. GETMODULEBASENAME_PROC m_pGetModuleBaseName;  // api GetModuleBaseName()
  32. HANDLE m_hProcess;
  33. #else*/
  34. char m_mod_name[MAX_MODULE_COUNT][100];  // max 100 modules in one process
  35. CREATETOOLHELP32SNAPSHOT_PROC pCreateToolhelp32Snapshot;
  36. PROCESS32FIRST_PROC pProcess32First;
  37. PROCESS32NEXT_PROC pProcess32Next;
  38. MODULE32FIRST_PROC pModule32First;
  39. MODULE32NEXT_PROC pModule32Next;
  40. //#endif
  42. public:
  43. DWORD m_dwProcessIDs[MAX_PROCESS_COUNT];  // 
  44. DWORD m_dwProcessCount;
  45. HMODULE m_hMods[MAX_MODULE_COUNT];
  46. int m_nModCount;
  48. int EnumProcess();
  49. int EnumProcessModules(DWORD process_id);
  50. int GetModuleBaseNameA(int num, char *mod_name);
  51. void EndEnumProcessModules();
  53. HANDLE GetProcessModuleHandle(DWORD process_id, char *mod_name);
  54. DWORD GetProcessID(char *exe_name);
  56. #ifdef WINNT
  57. BOOL ObtainSeDebugPrivilege(HANDLE hProcess);
  58. #endif
  59. };
  62. #endif