//IP数据报头
typedef struct
{
unsigned char hdr_len : 4; // 4位首部长度
unsigned char version : 4; // 4位版本号
unsigned char tos; // 8位服务类型
unsigned short total_len; // 16位总长度
unsigned short identifier; // 16位标识位
unsigned short frag_and_flags; // flags
unsigned char ttl; // 8位生存时间(TTL)
unsigned char protocol; // 8位上层协议
unsigned short checksum; // 16位校验和
unsigned long sourceIP; // 源IP地址
unsigned long destIP; // 目标IP地址
} IP_HEADER;
//ICMP数据报头
typedef struct
{
BYTE type; //8位类型
BYTE code; //8位代码
USHORT cksum; //16位校验和
USHORT id; //16位标识符
USHORT seq; //16位序列号
} ICMP_HEADER;
Code:
TraceRoute.h
#pragma once
#include <windows.h>
#pragma pack(1)
//IP数据报头
typedef struct
{
unsigned char hdr_len : 4; // 4位首部长度
unsigned char version : 4; // 4位版本号
unsigned char tos; // 8位服务类型
unsigned short total_len; // 16位总长度
unsigned short identifier; // 16位标识位
unsigned short frag_and_flags; // flags
unsigned char ttl; // 8位生存时间(TTL)
unsigned char protocol; // 8位上层协议
unsigned short checksum; // 16位校验和
unsigned long sourceIP; // 源IP地址
unsigned long destIP; // 目标IP地址
} IP_HEADER;
//ICMP数据报头
typedef struct
{
BYTE type; //8位类型
BYTE code; //8位代码
USHORT cksum; //16位校验和
USHORT id; //16位标识符
USHORT seq; //16位序列号
} ICMP_HEADER;
//解码结果
typedef struct
{
USHORT usSeqNo; //包序列号
DWORD dwRoundTripTime; //往返时间
in_addr dwIPaddr; //对端IP地址
} DECODE_RESULT;
#pragma pack()
const BYTE ICMP_ECHO_REQUEST = 8; //请求回显
const BYTE ICMP_ECHO_REPLY = 0; //回显应答
const BYTE ICMP_TIMEOUT = 11; //传输超时
const int DEF_MAX_HOP = 30; //最大跳站数
const DWORD DEF_ICMP_TIMEOUT = 3000; //默认超时时间,单位ms
const int DEF_ICMP_DATA_SIZE = 32; //默认ICMP数据部分长度
const int MAX_ICMP_PACKET_SIZE = 1024; //最大ICMP数据报的大小
USHORT GenerateChecksum(USHORT* pBuf, int iSize);
BOOL DecodeIcmpResponse(char* pBuf, int iPacketSize, DECODE_RESULT& stDecodeResult);
TraceRoute.cpp
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#include <iostream>
#include <ws2tcpip.h>
#include <winsock2.h>
#include <iomanip>
#include "TraceRoute.h"
#pragma comment(lib,"ws2_32")
using namespace std;
int main(int argc, char* argv[]) {
//检查命令行参数
if (argc != 2)
{
cerr << "\nUsage: itracert ip_or_hostname\n";
return -1;
}
//初始化winsock2环境
WSADATA wsa;
if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)
{
cerr << "\nFailed to initialize the WinSock2 DLL\n"
<< "error code: " << WSAGetLastError() << endl;
return -1;
}
//将命令行参数转换为IP地址
u_long ulDestIP = inet_addr(argv[1]);
if (ulDestIP == INADDR_NONE){
//转换不成功时按域名解析
hostent* pHostent = gethostbyname(argv[1]);
if (pHostent)
{
ulDestIP = (*(in_addr*)pHostent->h_addr).s_addr;
//输出屏幕信息
cout << "\nTracing route to " << argv[1]
<< " [" << inet_ntoa(*(in_addr*)(&ulDestIP)) << "]"
<< " with a maximum of " << DEF_MAX_HOP << " hops.\n" << endl;
}
else //解析主机名失败
{
cerr << "\nCould not resolve the host name " << argv[1] << '\n'
<< "error code: " << WSAGetLastError() << endl;
WSACleanup();
return -1;
}
}
else{
//输出屏幕信息
cout << "\nTracing route to " << argv[1]
<< " with a maximum of " << DEF_MAX_HOP << " hops.\n" << endl;
}
//填充目的Socket地址
sockaddr_in destSockAddr;
ZeroMemory(&destSockAddr, sizeof(sockaddr_in));
destSockAddr.sin_family = AF_INET;
destSockAddr.sin_addr.s_addr = ulDestIP;
//使用ICMP协议创建Raw Socket
SOCKET sockRaw = WSASocket(AF_INET, SOCK_RAW, IPPROTO_ICMP, NULL, 0, WSA_FLAG_OVERLAPPED);
if (sockRaw == INVALID_SOCKET)
{
cerr << "\nFailed to create a raw socket\n"
<< "error code: " << WSAGetLastError() << endl;
WSACleanup();
return -1;
}
//设置端口属性
int iTimeout = DEF_ICMP_TIMEOUT;
if (setsockopt(sockRaw, SOL_SOCKET, SO_RCVTIMEO, (char*)&iTimeout, sizeof(iTimeout)) == SOCKET_ERROR)
{
cerr << "\nFailed to set recv timeout\n"
<< "error code: " << WSAGetLastError() << endl;
closesocket(sockRaw);
WSACleanup();
return -1;
}
//创建ICMP包发送缓冲区和接收缓冲区
char IcmpSendBuf[sizeof(ICMP_HEADER) + DEF_ICMP_DATA_SIZE];
memset(IcmpSendBuf, 0, sizeof(IcmpSendBuf));
char IcmpRecvBuf[MAX_ICMP_PACKET_SIZE];
memset(IcmpRecvBuf, 0, sizeof(IcmpRecvBuf));
//填充待发送的ICMP包
ICMP_HEADER* pIcmpHeader = (ICMP_HEADER*)IcmpSendBuf;
pIcmpHeader->type = ICMP_ECHO_REQUEST;
pIcmpHeader->code = 0;
pIcmpHeader->id = (USHORT)GetCurrentProcessId();
memset(IcmpSendBuf + sizeof(ICMP_HEADER), 'E', DEF_ICMP_DATA_SIZE);
//开始探测路由
DECODE_RESULT stDecodeResult;
BOOL bReachDestHost = FALSE;
USHORT usSeqNo = 0;
int iTTL = 1;
int iMaxHop = DEF_MAX_HOP;
while (!bReachDestHost && iMaxHop--) {
//设置IP数据报头的ttl字段
setsockopt(sockRaw, IPPROTO_IP, IP_TTL, (char*)&iTTL, sizeof(iTTL));
//输出当前跳站数作为路由信息序号
cout << setw(3) << iTTL << flush;
//填充ICMP数据报剩余字段
((ICMP_HEADER*)IcmpSendBuf)->seq = htons(usSeqNo++);
((ICMP_HEADER*)IcmpSendBuf)->cksum = GenerateChecksum((USHORT*)IcmpSendBuf, sizeof(ICMP_HEADER) + DEF_ICMP_DATA_SIZE);
//记录序列号和当前时间
stDecodeResult.usSeqNo = ((ICMP_HEADER*)IcmpSendBuf)->seq;
stDecodeResult.dwRoundTripTime = GetTickCount();
//发送ICMP的EchoRequest数据报
if (sendto(sockRaw, IcmpSendBuf, sizeof(IcmpSendBuf), 0,
(sockaddr*)&destSockAddr, sizeof(destSockAddr)) == SOCKET_ERROR){
//如果目的主机不可达则直接退出
if (WSAGetLastError() == WSAEHOSTUNREACH)
cout << '\t' << "Destination host unreachable./n"
<< "\nTrace complete.\n" << endl;
closesocket(sockRaw);
WSACleanup();
return 0;
}
//接收ICMP的EchoReply数据报
//因为收到的可能并非程序所期待的数据报,所以需要循环接收直到收到所要数据或超时
sockaddr_in from;
int iFromLen = sizeof(from);
int iReadDataLen;
while (1){
//等待数据到达
iReadDataLen = recvfrom(sockRaw, IcmpRecvBuf, MAX_ICMP_PACKET_SIZE,
0, (sockaddr*)&from, &iFromLen);
if (iReadDataLen != SOCKET_ERROR) //有数据包到达
{
//解码得到的数据包,如果解码正确则跳出接收循环发送下一个EchoRequest包
if (DecodeIcmpResponse(IcmpRecvBuf, iReadDataLen, stDecodeResult))
{
if (stDecodeResult.dwIPaddr.s_addr == destSockAddr.sin_addr.s_addr)
bReachDestHost = TRUE;
cout << '\t' << inet_ntoa(stDecodeResult.dwIPaddr) << endl;
break;
}
}
else if (WSAGetLastError() == WSAETIMEDOUT) //接收超时,打印星号
{
cout << setw(9) << '*' << '\t' << "Request timed out." << endl;
break;
}
else
{
cerr << "\nFailed to call recvfrom\n"
<< "error code: " << WSAGetLastError() << endl;
closesocket(sockRaw);
WSACleanup();
return -1;
}
}
//TTL值加1
iTTL++;
}
//输出屏幕信息
cout << "\nTrace complete.\n" << endl;
closesocket(sockRaw);
WSACleanup();
return 0;
}
//产生网际校验和
USHORT GenerateChecksum(USHORT* pBuf, int iSize)
{
unsigned long cksum = 0;
while (iSize>1)
{
cksum += *pBuf++;
iSize -= sizeof(USHORT);
}
if (iSize)
cksum += *(UCHAR*)pBuf;
cksum = (cksum >> 16) + (cksum & 0xffff);
cksum += (cksum >> 16);
return (USHORT)(~cksum);
}
//解码得到的数据报
BOOL DecodeIcmpResponse(char* pBuf, int iPacketSize, DECODE_RESULT& stDecodeResult)
{
//检查数据报大小的合法性
IP_HEADER* pIpHdr = (IP_HEADER*)pBuf;
int iIpHdrLen = pIpHdr->hdr_len * 4;
if (iPacketSize < (int)(iIpHdrLen + sizeof(ICMP_HEADER)))
return FALSE;
//按照ICMP包类型检查id字段和序列号以确定是否是程序应接收的Icmp包
ICMP_HEADER* pIcmpHdr = (ICMP_HEADER*)(pBuf + iIpHdrLen);
USHORT usID, usSquNo;
if (pIcmpHdr->type == ICMP_ECHO_REPLY){
usID = pIcmpHdr->id;
usSquNo = pIcmpHdr->seq;
}else if (pIcmpHdr->type == ICMP_TIMEOUT){
char* pInnerIpHdr = pBuf + iIpHdrLen + sizeof(ICMP_HEADER); //载荷中的IP头
int iInnerIPHdrLen = ((IP_HEADER*)pInnerIpHdr)->hdr_len * 4;//载荷中的IP头长
ICMP_HEADER* pInnerIcmpHdr = (ICMP_HEADER*)(pInnerIpHdr + iInnerIPHdrLen);//载荷中的ICMP头
usID = pInnerIcmpHdr->id;
usSquNo = pInnerIcmpHdr->seq;
}else
return FALSE;
if (usID != (USHORT)GetCurrentProcessId() || usSquNo != stDecodeResult.usSeqNo)
return FALSE;
//处理正确收到的ICMP数据报
if (pIcmpHdr->type == ICMP_ECHO_REPLY || pIcmpHdr->type == ICMP_TIMEOUT){
//返回解码结果
stDecodeResult.dwIPaddr.s_addr = pIpHdr->sourceIP;
stDecodeResult.dwRoundTripTime = GetTickCount() - stDecodeResult.dwRoundTripTime;
//打印屏幕信息
if (stDecodeResult.dwRoundTripTime)
cout << setw(6) << stDecodeResult.dwRoundTripTime << " ms" << flush;
else
cout << setw(6) << "<1" << " ms" << flush;
return TRUE;
}
return FALSE;
}