TraceRoute(tracert)源码（基于ICMP.DLL实现）

  TraceRoute(tracert)源码（基于ICMP.DLL实现） 佟强 2008.11.4

TraceRoute的实现原理 http://blog.csdn.net/microtong/archive/2008/11/04/3220450.aspx

    本程序直接使用ICMP.DLL动态链接库中的函数来发送ICMP回显请求报文，用以探测路由信息，输出每一跳路由器的IP和往返时间，实现tracert的功能。

1. //TraceRoute3.cpp
2. #include <stdio.h>
3. #include <winsock2.h>
4. #include <windows.h>
5. #include <IPHlpApi.h>
7. //增加静态链接库ws2_32.lib
8. #pragma comment(lib,"ws2_32.lib")
10. //声明3个函数类型的指针
11. typedef HANDLE (WINAPI *lpIcmpCreateFile)(VOID);
12. typedef BOOL (WINAPI *lpIcmpCloseHandle)(HANDLE  IcmpHandle);
13. typedef DWORD (WINAPI *lpIcmpSendEcho)(
14.     HANDLE                   IcmpHandle,
15.     IPAddr                   DestinationAddress,
16.     LPVOID                   RequestData,
17.     WORD                     RequestSize,
18.     PIP_OPTION_INFORMATION   RequestOptions,
19.     LPVOID                   ReplyBuffer,
20.     DWORD                    ReplySize,
21.     DWORD                    Timeout
22.     );
24. int main(int argc,char* argv[]){
26.  if(argc!=2){
27.   printf("Usage: %s destIP/n",argv[0]);
28.   exit(-1);
29.  }
30.  WSADATA wsa;
31.  if(WSAStartup(MAKEWORD(2,2),&wsa)!=0){
32.   printf("WSAStartup failed./n");
33.   exit(-1);
34.  }
35.  //转换IP地址到整数
36.  unsigned long ip = inet_addr(argv[1]);
37.  if(ip==INADDR_NONE){
38.   //用户可能输入的是域名
39.   hostent* pHost = gethostbyname(argv[1]);
40.   //如果域名无法解析
41.   if(pHost==NULL){
42.    printf("Invalid IP or domain name: %s/n", argv[1]);
43.    exit(-1);
44.   }
45.   //取域名的第一个IP地址
46.   ip = *(unsigned long*)pHost->h_addr_list[0];
47.   printf("trace route to %s(%s)/n/n",argv[1],inet_ntoa(*(in_addr*)&ip));
48.  }else{
49.   printf("trace route to %s/n/n",argv[1]);
50.  }
52.  //载入ICMP.DLL动态库
53.  HMODULE hIcmpDll = LoadLibrary("icmp.dll");
54.  if(hIcmpDll==NULL){
55.   printf("fail to load icmp.dll/n");
56.   exit(-1);
57.  }
58.  //定义3个函数指针
59.  lpIcmpCreateFile IcmpCreateFile;
60.  lpIcmpCloseHandle IcmpCloseHandle;
61.  lpIcmpSendEcho IcmpSendEcho;
63.  //从ICMP.DLL中获取所需的函数入口地址
64.  IcmpCreateFile = (lpIcmpCreateFile)GetProcAddress(hIcmpDll,"IcmpCreateFile");
65.  IcmpCloseHandle = (lpIcmpCloseHandle)GetProcAddress(hIcmpDll,"IcmpCloseHandle");
66.  IcmpSendEcho = (lpIcmpSendEcho)GetProcAddress(hIcmpDll,"IcmpSendEcho");
68.  //打开ICMP句柄
69.  HANDLE hIcmp;
70.  if ((hIcmp = IcmpCreateFile()) == INVALID_HANDLE_VALUE){
71.   printf("/tUnable to open ICMP file./n");
72.   exit(-1);
73.  }
75.  //设置IP报头的TTL值
76.  IP_OPTION_INFORMATION IpOption;
77.  ZeroMemory(&IpOption,sizeof(IP_OPTION_INFORMATION));
78.  IpOption.Ttl = 1;
80.  //设置要发送的数据
81.  char SendData[32];
82.  memset(SendData,'0',sizeof(SendData));
84.  //设置接收缓冲区
85.  char ReplyBuffer[sizeof(ICMP_ECHO_REPLY)+32];
86.  PICMP_ECHO_REPLY pEchoReply = (PICMP_ECHO_REPLY)ReplyBuffer;
88.  BOOL bLoop = TRUE;
89.  int iMaxHop = 30;
90.  while(bLoop && iMaxHop--){
91.   printf("%2d: ",IpOption.Ttl);
92.   //发送ICMP回显请求
93.   if(IcmpSendEcho(hIcmp,(IPAddr)ip, SendData, sizeof(SendData), &IpOption, 
94.    ReplyBuffer, sizeof(ReplyBuffer), 3000)!=0){
95.     if(pEchoReply->RoundTripTime==0){
96.      printf("/t<1ms");
97.     }else{
98.      printf("/t%dms",pEchoReply->RoundTripTime);
99.     }
100.     printf("/t%s/n",inet_ntoa(*(in_addr*)&(pEchoReply->Address)));
101.     //判断是否完成路由路径探测
102.     if((unsigned long)pEchoReply->Address==ip){
103.      printf("/nTrace complete./n");
104.      bLoop = FALSE;
105.     }
106.   }else{
107.    printf("/t*/tRequest time out./n");
108.   }
109.   IpOption.Ttl++;
110.  }
112.  IcmpCloseHandle(hIcmp);
114.  FreeLibrary(hIcmpDll);
116.  WSACleanup();
117.  return 0;
118. }