计算机网络课程设计代码程序.doc

六、程序代码 文件scanhost.h#pragma pack(4)#pragma comment (lib,”Ws2_32.lib”)#define WIN32_LEAN_AND_MEAN#include #include #include #include #include #include #include /* The IP header */Typedef struct iphdr unsigned int headlen:4; /IP头长度unsigned int version:4; /IP版本号unsigned char tos; /服务类型unsigned short totallen; /IP包总长度unsigned short id; /ID号unsigned short flag; /标记unsigned char ttl; /生存时间unsigned char prot; /协议（UDP TCP）unsigned short checksum; /校验和unsigned int sourceIP; /源IP unsigned int destIP; /目的IPIpHeader;/ICMP headerTypedef struct icmphdr BYTE type; /ICMP类型码，回送请求的类型码为8BYTE code; /子类型码，保存与特定ICMP报文类型相关的细节信息USHORT checksum; /校验和USHORT id; /ICMP报文ID号USHORT seq; /ICMP数据报的序列号IcmpHeader;#define ICMP_ECHO 8 /请求回送#define ICMP_ECHO_REPLY 0 /请求回应#define ICMP_MIN 8 /ICMP包头长度（最小ICMP包长度）#define STATUS_FAILED 0xFFFF /错误码#define DEF_PACKET_SIZE 32 /缺省数据报长度#define MAX_PACKET 1024 /最大数据块长度/最大接收数据包长度#define MAX_PING_PACKET_SIZE(MAX_PACKET+sizeof(IpHeader) void fill_icmp_data(char *,int); /填充ICMP包USHORT checksum(USHORT *,int); /校验和函数void decode_resp(char *,int,struct sockaddr_in *); /找到此数据报IP地址DWORD WINAPI FindIP(LPVOID pIPAddrTemp); /线程调用子函数文件：scanhost.cpp#include “scanhost.h”WSADATA wsaData; /原始套接字SOCKET sockRaw; /dest:搜索目的IP,from:接收ICMP包的源IPStruct sockaddr_in dest,from,end; /end:搜索终止IP,即输入第三个参数 int fromlen = sizeof(from); /接收ICMP包长度char *recvbuf=new charMAX_PING_PACKET_SIZE; /接受ICMP包缓冲区unsigned int addr=0; /IP地址long ThreadNumCounter=0,ThreadNumLimit=20; /线程个数及最大允许线程数long *aa=&ThreadNumCounter;void main(int argc,char *argv)If(argc!=3) /判断输入格式是否正确 cout”输入格式错误: scanhost start_ip end_ip”endl; return;if (WSAStartup(MAKEWORD(2,1),&wsaData) !=0) /开始使用Ws2_32.dll cout”WSAStartup failed:”GetLastError()endl; ExitProcess(STATUS_FAILED);/创建原始套接字sockRaw=WSASocket(AF_INET,SOCK_RSW,IPPROTO_ICMP,NULL,0,wsa_FLAG_OVERLAPPED);if (sockRaw = INVALID_SOCKET) cout”WSASocket() failed: “WSAGetLastError()endl; ExitProcess(STATUS_FAILED);/设置读取延时int timeout = 1000;int bread=setsockopt(sockRaw,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout,sizeof (timeout);if(bread = SOCKET_ERROR) cout”failed to set recv timeout: “WSAGetLastError()endl; ExitProcess(STATUS_FAILED);timeout = 1000; bread=setsockopt(sockRaw,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout,sizeof(timeout);if(bread = SOCKET_ERROR) cout”failed to set send timeout: “WSAGetLastError()endl; ExitProcess(STATUS_FAILED);memset(&dest,0,sizeof(dest); /初始化dest结构unsigned long startIP,endIP;dest.sin_family =AF_INET;dest.sin_addr.s_addr=inet_addr(argv1); /填入开始搜索 IP地址startIP=inet_addr(argv1);end.sin_family=AF_INET;end.sin_addr.s_addr=inet_addr(argv2); /填入结束IP地址endIP=inet_addr(argv2);HANDLE hThread;while(htonl(startupIP)ThreadNumLimit) /线程数目太多？ Sleep(5000); /主线程休眠一段时间，不创建新线程 Continue; DWORD ThreadID; sockaddr_in * pIPAddrTemp=new(sockaddr_in); if(!pIPAddrTemp) cout”memory alloc failed”type = ICMP_ECHO; /设置类型信息 icmp_hdr-id = (USHORT)GetCurrentThreadId(); /设置其ID号为当前线程的ID号 datapart = icmp_data + sizeof(IcmpHeader); /计算出ICMP数据报的数据部分memset(datapart,A,datasize sizeof (IcmpHeader); /填入数据void decode_resp(char *buf , int bytes,struct sockaddr_in *from)IpHeader *iphdr;IcmpHeader *icmphdr;unsigned short iphdrlen;iphdr = (IpHeader *)buf;iphdrlen = iphdr-headlen * 4; /IP报头的长度icmphdr=(IcmpHeader *)(buf+iphdrlen); /跳过IP报头/数据包太短，丢弃if(bytestype != ICMP_ECHO_REPLY) return;/ID号不相符，丢弃if(icmphdr-id!=(USHORT)GetCurrentThread() return;/输出正在使用的IP地址cout”活动主机:”sin.addr)1) cksum+=*buffer+; size -=sizeof(USHORT); if(size) cksum += *(UCHAR*)buffer; cksum = (cksum 16) + (cksum &.0xffff); cksum += (cksum 16); return (USHORT)(cksum);DWORD WINAPI FindIP(LPVOID pIPAddrTemp) InterlockedIncrement(aa); /线程数目+1 char icmp_dataMAX_PACKET; /MAX_PACKET是数据报最大可能的长度 memset(icmp_data,0,MAX_PACKET); /将数据报清空初始化 int datasize=DEF_PACKET_SIZE; /数据报报文体的缺省长度 datasize+=sizeof(IcmpHeader); /加上报头的长度 fill_icmp_data(icmp_data,datasize); /填充ICMP数据报 (IcmpHeader*)icmp_data)-checksum = 0; /先将校验和置0 (IcmpHeader*)icmp_data)-seq = 0; /序列号为0 /计算校验和为0 (IcmpHeader*)icmp_data)-checksum=checksum(USHORT*)icmp_data,datasize); /发送数据报 int bwrote=sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)pIPAddrTemp,sizeof(dest); int n=0; if (bwrote = SOCKET_ERROR) if (WSAGetLastError() = WSAETIMEDOUT) cout”timed out”endl; cout”sendto failed: “WSAGetLastError()endl; ExitProcess(STATUS_FAILED); n=1; if (WSAGetLastError() = WSAETIMEDOUT) cout”timed out”endl; ExitProcess(STATUS_FAILED); n=1; if (bwrote datasize ) cout”Wrote “bwrote” bytes”endl; ExitProcess(STATUS_FAILED); n=1; /接收数据报 int bread = recvfrom(sockRaw,recvbuf,MAX_PING_PACKET_SIZE,0, ( struct sockaddr*) &from,&fromlen); if (bread = SOCKET_ERROR) if (WS