1. ==================ping的实现部分==================  
  2. int doit(int argc, char* argv[])  
  3. {//[bugfree] 建议将这个argc和argv的处理拿到main函数中  
  4. // 检查命令行参数  
  5. if (argc < 2) {  
  6. cerr << "usage: ping <host>" << endl;  
  7. return 1;  
  8. }  
  9.  
  10. // 装载ICMP.DLL连接库  
  11. HINSTANCE hIcmp = LoadLibrary("ICMP.DLL");  
  12. if (hIcmp == 0) {  
  13. cerr << "Unable to locate ICMP.DLL!" << endl;  
  14. return 2;  
  15. }  
  16.  
  17. // 查找给定机器的IP地址信息  
  18. struct hostent* phe;  
  19. if ((phe = gethostbyname(argv[1])) == 0) {  
  20. cerr << "Could not find IP address for " << argv[1] << endl;  
  21. return 3;  
  22. }  
  23.  
  24. // 定义函数三个指针类型  
  25. typedef HANDLE (WINAPI* pfnHV)(VOID);  
  26. typedef BOOL (WINAPI* pfnBH)(HANDLE);  
  27. typedef DWORD (WINAPI* pfnDHDPWPipPDD)(HANDLE, DWORD, LPVOID, WORD,  
  28. PIP_OPTION_INFORMATION, LPVOID, DWORD, DWORD); // evil, no?  
  29. //定义三个指针函数  
  30. pfnHV pIcmpCreateFile;  
  31. pfnBH pIcmpCloseHandle;  
  32. pfnDHDPWPipPDD pIcmpSendEcho;  
  33.  
  34. //从ICMP.DLL中得到函数入口地址  
  35. pIcmpCreateFile = (pfnHV)GetProcAddress(hIcmp, "IcmpCreateFile");  
  36. pIcmpCloseHandle = (pfnBH)GetProcAddress(hIcmp, "IcmpCloseHandle");  
  37. pIcmpSendEcho = (pfnDHDPWPipPDD)GetProcAddress(hIcmp, "IcmpSendEcho");  
  38. if ((pIcmpCreateFile == 0) || (pIcmpCloseHandle == 0) ||   
  39. (pIcmpSendEcho == 0)) {  
  40. cerr << "Failed to get proc addr for function." << endl;  
  41. return 4;  
  42. }  
  43.  
  44. // 打开ping服务  
  45. HANDLE hIP = pIcmpCreateFile();  
  46. if (hIP == INVALID_HANDLE_VALUE) {  
  47. cerr << "Unable to open ping service." << endl;  
  48. return 5;  
  49. }  
  50.  
  51. // 构造ping数据包  
  52. char acPingBuffer[64];  
  53. memset(acPingBuffer, '\xAA', sizeof(acPingBuffer));  
  54. PIP_ECHO_REPLY pIpe = (PIP_ECHO_REPLY)GlobalAlloc( GMEM_FIXED | GMEM_ZEROINIT,  
  55. sizeof(IP_ECHO_REPLY) sizeof(acPingBuffer));   
  56. if (pIpe == 0) {  
  57. cerr << "Failed to allocate global ping packet buffer." << endl;  
  58. return 6;  
  59. }  
  60. pIpe->Data = acPingBuffer;  
  61. pIpe->DataSize = sizeof(acPingBuffer);   
  62.  
  63. // 发送ping数据包  
  64. DWORD dwStatus = pIcmpSendEcho(hIP, *((DWORD*)phe->h_addr_list[0]),   
  65. acPingBuffer, sizeof(acPingBuffer), NULL, pIpe,   
  66. sizeof(IP_ECHO_REPLY) sizeof(acPingBuffer), 5000);  
  67. if (dwStatus != 0) {  
  68. cout << "Addr: " << 
  69. int(LOBYTE(LOWORD(pIpe->Address))) << "." << 
  70. int(HIBYTE(LOWORD(pIpe->Address))) << "." << 
  71. int(LOBYTE(HIWORD(pIpe->Address))) << "." << 
  72. int(HIBYTE(HIWORD(pIpe->Address))) << ", " << 
  73. "RTT: " << int(pIpe->RoundTripTime) << "ms, " << 
  74. "TTL: " << int(pIpe->Options.Ttl) << endl;  
  75. }  
  76. else {  
  77. cerr << "Error obtaining info from ping packet." << endl;  
  78. }  
  79.  
  80. // 关闭,回收资源  
  81. GlobalFree(pIpe);  
  82. FreeLibrary(hIcmp);  
  83. return 0;  
  84. }  


共3页: 上一页123下一页

相关内容