Util.cpp
上传用户:nbcables
上传日期:2007-01-11
资源大小:1243k
文件大小:5k
源码类别:

钩子与API截获

开发平台:

Visual C++

  1. #include "stdafx.h"
  2. #include <stdio.h>
  3. #include <io.h>
  4. #include "resource.h"
  5. #include "util.h"
  6. extern HINSTANCE g_hInstance;
  7. void GetFileName(char *fname)
  8. {
  9. char temp[200];
  10. GetModuleFileName(NULL, temp, sizeof(temp));
  11. int i =strlen(temp);
  12. while(i >0 && temp[i-1] !='\' && temp[i-1] !=':') i--;
  13. strcpy(fname, &temp[i]);
  14. }
  15. // 如果是win9x,不能使用fopen函数
  16. void WriteLog(char *fmt,...)
  17. {
  18. va_list args;
  19. char modname[200];
  20. char temp[5000];
  21. HANDLE hFile;
  22. GetModuleFileName(NULL, modname, sizeof(modname));
  23. if((hFile =CreateFile("c:\hookapi.log", GENERIC_WRITE, 0, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL)) <0)
  24. {
  25. return;
  26. }
  27. _llseek((HFILE)hFile, 0, SEEK_END);
  28. wsprintf(temp, "mydll.dll:%s:", modname);
  29. DWORD dw;
  30. WriteFile(hFile, temp, strlen(temp), &dw, NULL);
  31. va_start(args,fmt);
  32. vsprintf(temp, fmt, args);
  33. va_end(args);
  34. WriteFile(hFile, temp, strlen(temp), &dw, NULL);
  35. wsprintf(temp, "rn");
  36. WriteFile(hFile, temp, strlen(temp), &dw, NULL);
  37. _lclose((HFILE)hFile);
  38. }
  39. int ipcmp(char *szip1, char *szip2)
  40. {
  41. ULONG ip1 =GetIntIP(szip1);
  42. ULONG ip2 =GetIntIP(szip2);
  43. if(ip1 > ip2) return 1;
  44. else if(ip1 <ip2) return -1;
  45. else return 0;
  46. }
  47. ULONG GetIntIP(char *szip)
  48. {
  49. char *p, *p1;
  50. char ip[16];
  51. int i;
  52. ULONG ii[4];
  53. i =0;
  54. strcpy(ip, szip);
  55. p =ip;
  56. while(*p && i<4)
  57. {
  58. p1 =p;
  59. while(*p && *p !='.') p++;
  60. *p =0;
  61. ii[i] =my_atoi(p1);
  62. p ++;
  63. i++;
  64. }
  65. return ii[0]*256*256*256+ii[1]*256*256+ii[2]*256+ii[3];
  66. }
  67. ULONG my_atoi(char *p)
  68. {
  69. ULONG i;
  70. i =0;
  71. while(*p)
  72. {
  73. i =(i+*p-'0')*10;
  74. p++;
  75. }
  76. i =i/10;
  77. return i;
  78. }
  79. BOOL CALLBACK PasswordDlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
  80. {
  81. static PASSWORD_DATA *pdata;
  82. char password[9];
  83. switch(msg)
  84. {
  85. case WM_INITDIALOG:
  86. pdata =(PASSWORD_DATA *)lParam;
  87. SetDlgItemText(hDlg, IDC_APP_NAME, pdata->app_name);
  88. SetDlgItemText(hDlg, IDC_IP, pdata->ip);
  89. SetWindowPos(hDlg, NULL, 300, 300, 0, 0, SWP_NOSIZE);
  90. SendDlgItemMessage(hDlg, IDE_PASSWORD, EM_SETLIMITTEXT, 8, 0L);
  91. break;
  92. case WM_COMMAND:
  93. switch(LOWORD(wParam))
  94. {
  95. case IDOK:
  96. GetDlgItemText(hDlg, IDE_PASSWORD, password, sizeof(password));
  97. if(strcmp(password, pdata->password) !=0)
  98. {
  99. MessageBox(hDlg, "密码不对", pdata->app_name, MB_OK);
  100. return TRUE;
  101. }
  102. EndDialog(hDlg, IDOK);
  103. break;
  104. case IDCANCEL:
  105. EndDialog(hDlg, IDCANCEL);
  106. }
  107. break;
  108. }
  109. return FALSE;
  110. }
  111. int CheckPassword(char *app, char *ip, char *password)
  112. {
  113. PASSWORD_DATA data;
  114. strcpy(data.app_name, app);
  115. strcpy(data.ip, ip);
  116. strcpy(data.password, password);
  117. if(DialogBoxParam(g_hInstance, MAKEINTRESOURCE(IDD_PASSWORD), NULL, PasswordDlgProc, (LONG)&data) ==IDCANCEL)
  118. return -1;
  119. return 0;
  120. }
  121. int EncURL(unsigned char *url, int len, char *new_url)
  122. {
  123. int k =0;
  124. for(int i =0; i<len; i++)
  125. {
  126. if(url[i] >127)
  127. {
  128. wsprintf(&new_url[k], "%%%2X", url[i]);
  129. k +=3;
  130. }
  131. else new_url[k++] =url[i];
  132. }
  133. new_url[k] =0;
  134. return k;
  135. }
  136. int FindData(unsigned char *data1, int len1, unsigned char *data2, int len2, int enc_url)
  137. {
  138. int i, len_url;
  139. char url[1024];
  140. //unsigned char *data =new BYTE[len2];
  141. int count =0;
  142. int pos[256], len[256], pos3[256], len3[256];
  143. int k =0;
  144. pos [0] =0;
  145. len [0] =0;
  146. char *p =(char *)data2;
  147. for(i =0; i<len2; i++)
  148. {
  149. if(p[i] !=',') len[k] ++;
  150. else
  151. {
  152. k++;
  153. pos[k] =i+1;
  154. len[k] =0;
  155. }
  156. }
  157. count =k+1;
  158. if(enc_url)
  159. {
  160. len_url =EncURL(data2, len2, url);
  161. k =0;
  162. pos3[0] =0;
  163. len3[0] =0;
  164. char *p =url;
  165. for(i =0; i<len_url; i++)
  166. {
  167. if(p[i] !=',') len3[k] ++;
  168. else
  169. {
  170. k++;
  171. pos3[k] =i+1;
  172. len3[k] =0;
  173. }
  174. }
  175. }
  176. else len_url =len2;
  177. //WriteLog("url =%s,len_url=%dndata1=%sn####len1=%d", url,len_url, data1,len1);
  178. i =0;
  179. while(i+len_url<=len1)
  180. {
  181. for(k =0; k<count; k++)
  182. {
  183. if(len[k] && memcmp(data1+i, data2+pos[k], len[k]) ==0) return i;
  184. }
  185. if(enc_url)
  186. {
  187. for(k =0; k<count; k++)
  188. {
  189. if(len3[k] && memcmp(data1+i, url+pos3[k], len3[k]) ==0) return i;
  190. }
  191. }
  192. i++;
  193. }
  194. //WriteLog("not found!!!");
  195. //delete data;
  196. return -1;
  197. }
  198. //加密例子:使用异或,输出长度不变。
  199. int EncryptData(int algrithm, char *password, unsigned char *inbuf, int inbuf_len, unsigned char *outbuf, int *outbuf_len)
  200. {
  201. int i;
  202. for(i =0; i<inbuf_len-1; i++)
  203. {
  204. outbuf[i] =inbuf[i]^password[i%8];
  205. }
  206. *outbuf_len =inbuf_len;
  207. return *outbuf_len;
  208. }
  209. int DecryptData(int algrithm, char *password, unsigned char *inbuf, int inbuf_len, unsigned char *outbuf, int *outbuf_len)
  210. {
  211. int i;
  212. for(i =0; i<inbuf_len; i++)
  213. {
  214. outbuf[i] =inbuf[i]^password[i%8];
  215. }
  216. *outbuf_len =inbuf_len;
  217. return *outbuf_len;
  218. }