[原创]命令行下可以解读TCP数据报内容的Sniffer

#include "stdafx.h"
#include "stdio.h"
#include "string.h"
#include "mstcpip.h"
#include "Winsock2.h"
#pragma comment(lib,"WS2_32.lib")
#define STATUS_FAILED 0xFFFF //定义异常出错代码
#define MAX_PACK_LEN 65535 //接收的最大
IP 报文
#define MAX_ADDR_LEN 16 //点分十进制
地址的最大长度
#define MAX_HOSTNAME_LAN 255 //最大主机名长度
typedef struct _iphdr //定义IP首部
{
unsigned char h_lenver; //4
位首部长度+4 位IP 版本号
unsigned char tos;
//8 位服务类型TOS
unsigned short total_len; //16
位总长度（字节）
unsigned short ident; //16
位标识
unsigned short frag_and_flags; //3 位标志位
unsigned char ttl;
//8 位生存时间TTL
unsigned char proto; //8
位协议(TCP 或其他)
unsigned short checksum; //16
位IP 首部校验和
unsigned int sourceIP; //32
位源IP 地址
unsigned int destIP; //32
位目的IP 地址
}IP_HEADER;

typedef struct _tcphdr //定义TCP 首
部
{
USHORT th_sport;
//16 位源端口
USHORT th_dport;
//16 位目的端口
unsigned int th_seq;
unsigned int th_ack;
unsigned char th_lenres; //4
位首部长度/6 位保留字
unsigned char th_flag; //6
位标志位
USHORT th_win;
//16 位窗口大小
USHORT th_sum;
//16 位校验和
USHORT th_urp;
//16 位紧急数据偏移量
}TCP_HEADER;

SOCKET SockRaw;
char* TcpFlag[6]={ //定
义TCP 标志位
"FIN ",
//FIN: 表示发送端已经没有数据要求传输了，希望释放连接。
"SYN ",
//SYN: 标志位用来建立连接，让连接双方同步序列号。如果SYN＝1而ACK=0，则表示该数据包为连接请求，如果SYN=1而ACK=1则表示
接受连接。
"RST ",
//RST: 用来复位一个连接。RST标志置位的数据包称为复位包。一般情况下，如果TCP收到的一个分段明显不是属于该主机上的任何一
个连接，则向远端发送一个复位包。
"PSH ",
//PSH: 如果置位，接收端应尽快把数据传送给应用层。
"ACK ",
//ACK: 为确认标志位。如果为1，表示包中的确认号时有效的。否则，包中的确认号无效。
"URG "
//URG: 为紧急数据标志。如果它为1，表示本数据包中包含紧急数据。此时紧急数据指针有效。
};
bool ParamTcp =true; // -t 关注TCP
报文
char *strFromIpFilter=NULL; // 源IP 地址
过滤
char *strDestIpFilter=NULL; // 目的地址过
滤
//Referrence
int DecodeIpPack(char *,int);
int DecodeTcpPack(char *);
void CheckSockError(int,char*);
void usage(void);
bool GetCmdLine(int, char **);
//
void main(int argc, char ** argv)
{
int iErrorCode;
char RecvBuf[MAX_PACK_LEN] = { 0 };
usage();
if(GetCmdLine(argc, argv)==true)
exit(0);

//初始化SOCKET
WSADATA wsaData;
iErrorCode = WSAStartup(MAKEWORD(2,1),&wsaData);
CheckSockError(iErrorCode, "WSAStartup");
SockRaw = socket(AF_INET , SOCK_RAW , IPPROTO_IP);
CheckSockError(SockRaw, "socket");

//获取本机IP 地址
char FAR name[MAX_HOSTNAME_LAN];
iErrorCode = gethostname(name, MAX_HOSTNAME_LAN);
CheckSockError(iErrorCode, "gethostname");
struct hostent FAR * pHostent;
pHostent = (struct hostent * )malloc(sizeof(struct hostent));
pHostent = gethostbyname(name);
SOCKADDR_IN sa;
sa.sin_family = AF_INET;
sa.sin_port = htons(6000);
memcpy(&sa.sin_addr.S_un.S_addr, pHostent->h_addr_list[0], pHostent->h_length);
iErrorCode = bind(SockRaw, (PSOCKADDR)&sa, sizeof(sa));
CheckSockError(iErrorCode, "bind");

//设置SOCK_RAW 为SIO_RCVALL，以便接收所有的IP 包
DWORD dwBufferLen[10] ;
DWORD dwBufferInLen = 1 ;
DWORD dwBytesReturned = 0 ;
iErrorCode=WSAIoctl(SockRaw,
SIO_RCVALL,
&dwBufferInLen,
sizeof(dwBufferInLen),
&dwBufferLen,
sizeof(dwBufferLen),
&dwBytesReturned,
NULL,
NULL );
CheckSockError(iErrorCode, "Ioctl");

//侦听IP 报文
while(true)
{
memset(RecvBuf, 0, sizeof(RecvBuf));
iErrorCode = recv(SockRaw, RecvBuf, sizeof(RecvBuf), 0);
CheckSockError(iErrorCode, "recv");
iErrorCode = DecodeIpPack(RecvBuf, iErrorCode);
CheckSockError(iErrorCode, "Decode");
}
}

//IP 解包程序
int DecodeIpPack(char *buf, int iBufSize)
{
IP_HEADER *pIpheader;
int iProtocol, iTTL;
char szProtocol[4];
char szSourceIP[MAX_ADDR_LEN], szDestIP[MAX_ADDR_LEN];
SOCKADDR_IN saSource, saDest;
pIpheader = (IP_HEADER *)buf;
//Check Proto
iProtocol = pIpheader->proto;
strncpy(szProtocol, "TCP", 4);
if(iProtocol==IPPROTO_TCP)
{
//Check Source IP
saSource.sin_addr.s_addr = pIpheader->sourceIP;
strncpy(szSourceIP, inet_ntoa(saSource.sin_addr), MAX_ADDR_LEN);
if (strFromIpFilter)
if (strcmp(strFromIpFilter,szSourceIP))
return true;
//Check Dest IP
saDest.sin_addr.s_addr = pIpheader->destIP;
strncpy(szDestIP, inet_ntoa(saDest.sin_addr), MAX_ADDR_LEN);
if (strDestIpFilter)
if (strcmp(strDestIpFilter,szDestIP))
return true;
//TTL
iTTL = pIpheader->ttl;
//Output
printf("%s ", szProtocol);
printf("%s->%s ", szSourceIP, szDestIP);
printf("bytes=%d TTL=%d ",iBufSize,iTTL);
//Calculate IP Header Length
int iIphLen = sizeof(unsigned long) * (pIpheader->h_lenver & 0xf);
//Decode Sub Protocol:TCP
DecodeTcpPack(buf+iIphLen);
}
return true;
}

//SOCK 错误处理程序
void CheckSockError(int iErrorCode, char *pErrorMsg)
{
if(iErrorCode==SOCKET_ERROR)
{
printf("%s Error:%d/n", pErrorMsg, GetLastError());
closesocket(SockRaw);
exit(0);
}
}

//TCP 解包程序
int DecodeTcpPack(char * TcpBuf)
{
TCP_HEADER * pTcpHeader;
pTcpHeader = (TCP_HEADER * )TcpBuf;
printf("Port:%d->%d ", ntohs(pTcpHeader->th_sport),ntohs(pTcpHeader->th_dport));
unsigned char FlagMask = 1;
int HdrLen = (pTcpHeader->th_lenres)>>2;
printf("/n%s",((BYTE *)pTcpHeader)+HdrLen);
for( int i=0; i<6; i++ )
{
if((pTcpHeader->th_flag) & FlagMask)
printf("%s",TcpFlag[i]);
else
printf(" ");
FlagMask=FlagMask<<1;
}
printf("/n");
return true;
}

//命令行参数处理
bool GetCmdLine(int argc, char ** argv)
{
for(int i=1;i<argc;i++)
{
if(argv[i][0]!='/')
return true;
else
{
switch (argv[i][1])
{
case 'f':
case 'F':
{
strFromIpFilter=(char*)malloc(16*sizeof(char));
memset(strFromIpFilter,0,16*sizeof(char));
strcpy(strFromIpFilter,argv[i]+3);
break;
}
case 'd':
case 'D':
{
strDestIpFilter=(char*)malloc(16*sizeof(char));
memset(strDestIpFilter,0,16*sizeof(char));
strcpy(strDestIpFilter,argv[i]+3);
break;
}
case '?':
return true;
break;
default:
break;
}
}
}
printf("/nWill Sniffer");
printf(" TCP");
if(strFromIpFilter)
printf(" FromIp:%s",strFromIpFilter);
if(strDestIpFilter)
printf(" DestIp:%s",strDestIpFilter);
printf("/n/tCTRL+C to quit/nStart:/n");
return (false);
}

void usage(void)
{
printf("/tSinffer for Win2K Console/n");
printf("USAGE:/n");
printf("/t/f: fromIP Output Packets FromIp=fromIP (default ALL)/n");
printf("/t/d: destIP Output Packets DestIp=destIP (default ALL)");
printf("/nExample:/n");
printf("/tsniffer.exe /d:192.168.15.233/n");
printf("/tsniffer.exe /f:192.168.15.231/n");
}