Category

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

Hook api

Development Platform:

Visual C++

ApiInfo.cpp：Code Content
  1. // APIInfo.cpp: implementation of the CAPIInfo class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #include <windows.h>
  6. #include "APIInfo.h"
  7. #include "util.h"
  9. //////////////////////////////////////////////////////////////////////
  10. // Construction/Destruction
  11. //////////////////////////////////////////////////////////////////////
  13. #ifdef WIN95
  14. APIINFO g_apiinfo_buf[1000];
  15. #endif
  17. CAPIInfo::CAPIInfo()
  18. {
  19. m_count =0;
  20. m_pInfo =NULL;
  21. }
  23. CAPIInfo::~CAPIInfo()
  24. {
  25. DeleteAll();
  26. }
  28. APIINFO *CAPIInfo::Add(char *module_name, char *api_name, char *my_api_name,
  29. int param_count, APIFUNC old_api, APIFUNC my_api, char *my_friend_api_name, int start_pos)
  30. {
  31. #ifdef WIN95
  32. APIINFO *pinfo =(APIINFO *)&g_apiinfo_buf[m_count];
  33. #else
  34. APIINFO *pinfo =(APIINFO *)malloc(sizeof(APIINFO));
  35. #endif
  37. strcpy(pinfo->module_name, module_name);
  38. strcpy(pinfo->api_name, api_name);
  39. strcpy(pinfo->my_api_name, my_api_name);
  40. pinfo->param_count =param_count;
  42. pinfo->old_api =old_api;
  43. pinfo->my_api =my_api;
  45. if(my_friend_api_name)
  46. strcpy(pinfo->my_friend_api_name, my_friend_api_name);
  47. else pinfo->my_friend_api_name[0] =0;
  49. pinfo->start_pos =start_pos;
  50. pinfo->f_hooked =false;
  52. #ifdef WIN95
  53. pinfo->hMutex =CreateMutex(NULL, FALSE, api_name);
  54. #endif
  56. #ifdef WINNT
  57. InitializeCriticalSection(&pinfo->cs);
  58. #endif
  60. pinfo->pnext =NULL;
  61. APIINFO *pinfo2 =m_pInfo;
  62. while(pinfo2 !=NULL && pinfo2->pnext !=NULL)
  63. pinfo2 =pinfo2->pnext;
  64. if(pinfo2 ==NULL) m_pInfo =pinfo;             // this is head of m_pInfo
  65. else pinfo2->pnext =pinfo;
  67. m_count++;
  69. return pinfo;
  70. }
  72. int CAPIInfo::DeleteAll()
  73. {
  74. APIINFO *pinfo =m_pInfo, *pinfo2;
  75. while(pinfo)
  76. {
  77. pinfo2 =pinfo;
  78. pinfo =pinfo->pnext;
  79. #ifdef WIN95
  80. if (pinfo2->hMutex)
  81. CloseHandle(pinfo2->hMutex);
  82. #else
  83. DeleteCriticalSection(&pinfo2->cs);
  84. free(pinfo2);
  85. #endif
  86. }
  88. m_pInfo =NULL;
  89. m_count =0;
  91. return 0;
  92. }
  94. APIINFO *CAPIInfo::FindByMyAPI(APIFUNC my_api)
  95. {
  96. APIINFO *pinfo =m_pInfo;
  98. while(pinfo)
  99. {
  100. if(pinfo->my_api ==my_api)
  101. break;
  102. pinfo =pinfo->pnext;
  103. }
  105. return pinfo;
  106. }
  108. APIINFO *CAPIInfo::FindByOldAPI(APIFUNC old_api)
  109. {
  110. APIINFO *pinfo =m_pInfo;
  112. while(pinfo)
  113. {
  114. if((BYTE *)pinfo->old_api+pinfo->start_pos ==(BYTE *)old_api)
  115. break;
  116. pinfo =pinfo->pnext;
  117. }
  119. return pinfo;
  120. }
  122. APIINFO *CAPIInfo::FindByAPIName(char *api_name)
  123. {
  124. APIINFO *pinfo =m_pInfo;
  126. while(pinfo)
  127. {
  128. if(!strcmp(pinfo->api_name, api_name))
  129. break;
  130. pinfo =pinfo->pnext;
  131. }
  133. return pinfo;
  134. }
  136. APIINFO *CAPIInfo::FindByMyAPIName(char *my_api_name)
  137. {
  138. APIINFO *pinfo =m_pInfo;
  140. while(pinfo)
  141. {
  142. if(!strcmp(pinfo->my_api_name, my_api_name))
  143. break;
  144. pinfo =pinfo->pnext;
  145. }
  147. return pinfo;
  148. }
  150. int CAPIInfo::lock(APIINFO *pinfo)
  151. {
  152. #ifdef WIN95
  153. if((pinfo->m_hMutex =OpenMutex(MUTEX_ALL_ACCESS, FALSE, pinfo->api_name)) !=NULL)
  154. WaitForSingleObject(pinfo->m_hMutex, INFINITE);
  155. #else
  156. EnterCriticalSection(&pinfo->cs);
  157. #endif
  159. return 0;
  160. }
  162. int CAPIInfo::unlock(APIINFO *pinfo)
  163. {
  164. #ifdef WIN95
  165. if(pinfo->m_hMutex)
  166. {
  167. ReleaseMutex(pinfo->m_hMutex);
  168. CloseHandle(pinfo->m_hMutex);
  169. pinfo->m_hMutex =NULL;
  170. }
  171. #else
  172. LeaveCriticalSection(&pinfo->cs);
  173. #endif
  175. return 0;
  176. }