#pragma pack(4)
#pragma comment(lib,"WS2_32.LIB")
#define WIN32_LEAN_AND_MEAN
#include<winsock2.h>
#include<stdio.h>
#include<winsock.h>
#include<iostream.h>
#include<sys/timeb.h>
#include<time.h>
#include<winbase.h>
//頭文件

typedef struct iphdr{                  //IP頭
	unsigned int headlen:4;            //IP頭長(zhǎng)度
	unsigned int version:4;            //IP版本號(hào)
	unsigned char tos;                 //服務(wù)類(lèi)型
	unsigned short id;                 //ID號(hào)
	unsigned short flag;               //標(biāo)記
	unsigned char ttl;                 //生存時(shí)間
	unsigned char prot;                //協(xié)議
	unsigned short checksum;           //效驗(yàn)和
	unsigned int sourceIP;             //源IP
	unsigned int destIP;               //目的IP
}IpHeader;
//IP頭部

typedef struct icmphdr{                //ICMP頭
	BYTE type;                         //ICMP類(lèi)型碼
	BYTE code;                         //子類(lèi)型碼
	USHORT checksum;                   //效驗(yàn)和
    USHORT id;                         //ID號(hào)
	USHORT seq;                        //ICMP數(shù)據(jù)報(bào)的序列號(hào)
}IcmpHeader;
//ICMP包頭部

#define ICMP_ECHO 8                    //請(qǐng)求回送
#define ICMP_ECHO_REPLY 0              //請(qǐng)求回應(yīng)
#define ICMP_MIN 8                     //ICMP包頭長(zhǎng)度（最小ICMP包長(zhǎng)度）
#define STATUS_FAILED 0xFFFF           //錯(cuò)誤碼
#define DEF_PACKET_SIZE 32             //缺省數(shù)據(jù)塊長(zhǎng)度
#define MAX_PACKET 1024                //最大數(shù)據(jù)塊長(zhǎng)度
#define MAX_PING_PACKET_SIZE (MAX_PACKET+sizeof(IpHeader))
//最大接收數(shù)據(jù)包長(zhǎng)度

void fill_icmp_data(char *,int);
USHORT checksum(USHORT *,int);
void decode_resp(char *,int,struct sockaddr_in *);
DWORD WINAPI FindIP(LPVOID pIPAddrTemp);
//函數(shù)的申明

WSADATA wsaData;
SOCKET sockRaw;             //原始套接字
struct sockaddr_in dest,from,end;
//dest：搜索目的IP，
//from：接收ICMP包的源IP
//end：搜索終止IP。

int fromlen=sizeof(from);                      //接收ICMP包長(zhǎng)度
char *recvbuf=new char[MAX_PING_PACKET_SIZE];  //接受ICMP包緩沖區(qū)
unsigned int addr=0;                           //IP地址
long ThreadNumCounter=0,ThreadNumLimit=20;     //線程數(shù)及最大允許線程數(shù)
long *aa=&ThreadNumCounter;
//全局變量的申明

void main(int argc,char *argv[])
{
	if(argc!=3)                               //判斷格式是否正確
	{
		cout<<"輸入格式錯(cuò)誤：scanhost start_ip end_ip"<<endl;
		return;
	}
	if(WSAStartup(MAKEWORD(2,1),&wsaData)!=0)
	{
		cout<<"WSAStartup failed:"<<GetLastError()<<endl;
		ExitProcess(STATUS_FAILED);
	}
//創(chuàng)建原始套接字
	sockRaw=WSASocket(AF_INET,SOCK_RAW,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 timeou:"<<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結(jié)構(gòu)
	unsigned long startIP,endIP;  
	dest.sin_family=AF_INET;
	dest.sin_addr.s_addr=inet_addr(argv[1]);  //填入開(kāi)始搜索IP
	startIP=inet_addr(argv[1]);
	end.sin_family=AF_INET;
	end.sin_addr.s_addr=inet_addr(argv[2]);
	endIP=inet_addr(argv[2]);    //填入結(jié)束搜索IP地址
	HANDLE hThread;
	while(htonl(startIP)<=htonl(endIP))   //起始IP比結(jié)束IP小
	{
		if(ThreadNumCounter>ThreadNumLimit) //判斷線程數(shù)目，如果太多，休眠
		{
			Sleep(5000);
			continue;
		}
		DWORD ThreadID;
		sockaddr_in * pIPAddrTemp=new(sockaddr_in);
		if(!pIPAddrTemp)
		{
			cout<<"memory alloc failed"<<endl;
			return;
		}
		*pIPAddrTemp=dest;
		clock_t start;
		start=clock();
		hThread=CreateThread(NULL,NULL,FindIP,(LPVOID)pIPAddrTemp,NULL,&ThreadID);
		long i=60000000L;
		while(i--);
		TerminateThread(hThread,0);
		InterlockedDecrement(aa);
		memset(&from,0,sizeof(from));
		startIP=htonl(htonl(startIP)+1);
		dest.sin_addr.s_addr=startIP;
	}
	while(ThreadNumCounter!=0)
	{
		Sleep(2000);
		return;
	}
}

void fill_icmp_data(char *icmp_data,int datasize)
{
	IcmpHeader *icmp_hdr;
	char *datapart;
	icmp_hdr=(IcmpHeader*)icmp_data;
	icmp_hdr->type=ICMP_ECHO;//設(shè)置類(lèi)型信息
	icmp_hdr->id=(USHORT)GetCurrentThreadId();//設(shè)置其ID號(hào)為當(dāng)前線程ID號(hào)
	datapart=icmp_data+sizeof(IcmpHeader);//計(jì)算ICMP數(shù)據(jù)報(bào)的數(shù)據(jù)部分
	memset(datapart,'A',datasize-sizeof(IcmpHeader));//填入數(shù)據(jù)
}
//ICMP數(shù)據(jù)包的填充

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;
	icmphdr=(IcmpHeader *)(buf+iphdrlen);

	if(bytes<iphdrlen+ICMP_MIN)return;
	if(icmphdr->type!=ICMP_ECHO_REPLY)return;
	if(icmphdr->id!=(USHORT)GetCurrentThreadId())return;
	cout<<"活動(dòng)主機(jī)："<<inet_ntoa(from->sin_addr)<<endl;
}
//返回包的解析，以及輸出

USHORT checksum(USHORT *buffer,int size)
{
	unsigned long cksum=0;
	while(size>1)
	{
		cksum+=*buffer++;
		size-=sizeof(USHORT);
	}
	if(size)
	{
		cksum+=*(UCHAR*)buffer;
	}
	cksum=(cksum>>16)+(cksum & 0xffff);
	cksum+=(cksum>>16);
	return (USHORT)(~cksum);
}
//效驗(yàn)和的計(jì)算

DWORD WINAPI FindIP(LPVOID pIPAddrTemp)
{
	InterlockedIncrement(aa);//線程數(shù)目+1
	char icmp_data[MAX_PACKET];
	memset(icmp_data,0,MAX_PACKET);//數(shù)據(jù)報(bào)初始化
	int datasize=DEF_PACKET_SIZE;//數(shù)據(jù)報(bào)報(bào)文的缺省長(zhǎng)度
	datasize+=sizeof(IcmpHeader);//加上icmp頭部長(zhǎng)度
	fill_icmp_data(icmp_data,datasize);//填充包
	((IcmpHeader*)icmp_data)->checksum=0;//效驗(yàn)和置零
	((IcmpHeader*)icmp_data)->seq=0;//序列號(hào)置零
	((IcmpHeader*)icmp_data)->checksum=checksum((USHORT*)icmp_data,datasize);
	//計(jì)算效驗(yàn)和后填人
	int bwrote=sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)pIPAddrTemp,sizeof(dest));
	//發(fā)送數(shù)據(jù)報(bào)
	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);
	//數(shù)據(jù)包的接收
	if(bread==SOCKET_ERROR)
	{
		if (WSAGetLastError()==WSAETIMEDOUT)
		{
			cout<<"timed out"<<endl;
		}
		cout<<"recvfrom failed:"<<WSAGetLastError()<<endl;
		ExitProcess(STATUS_FAILED);
		n=1;
	}
	if(n==0)
		decode_resp(recvbuf,bread,&from);
	InterlockedDecrement(aa);
	return 0;
}
//子線程的編寫(xiě)