Ping的原代码 网络安全者必看

　　#pragma pack(4)

　　#define WIN32_LEAN_AND_MEAN

　　#include

　　#include

　　#include

　　

　　#define ICMP_ECHO 8

　　#define ICMP_ECHOREPLY 0

　　

　　#define ICMP_MIN 8 // minimum 8 byte icmp packet (just header)

　　

　　/* The IP header */

　　typedef struct iphdr {

　　unsigned int h_len:4; // length of the header

　　unsigned int version:4; // Version of IP

　　unsigned char tos; // Type of service

　　unsigned short total_len; // total length of the packet

　　unsigned short ident; // unique identifier

　　unsigned short frag_and_flags; // flags

　　unsigned char ttl;

　　unsigned char proto; // protocol (TCP, UDP etc)

　　unsigned short checksum; // IP checksum

　　

　　unsigned int sourceIP;

　　unsigned int destIP;

　　

　　}IpHeader;

　　

　　//

　　// ICMP header

　　//

　　typedef struct _ihdr {

　　BYTE i_type;

　　BYTE i_code; /* type sub code */

　　USHORT i_cksum;

　　USHORT i_id;

　　USHORT i_seq;

　　/* This is not the std header, but we reserve space for time */

　　ULONG timestamp;

　　}IcmpHeader;

　　

　　#define STATUS_FAILED 0xFFFF

　　#define DEF_PACKET_SIZE 32

　　#define MAX_PACKET 1024

　　

　　#define xmalloc(s) HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(s))

　　#define xfree(p) HeapFree (GetProcessHeap(),0,(p))

　　

　　void fill_icmp_data(char *, int);

　　USHORT checksum(USHORT *, int);

　　void decode_resp(char *,int ,struct sockaddr_in *);

　　

　　void Usage(char *progname){

　　

　　fprintf(stderr,"Usage:\n");

　　fprintf(stderr,"%s [data_size]\n",progname);

　　fprintf(stderr,"datasize can be up to 1Kb\n");

　　ExitProcess(STATUS_FAILED);

　　

　　}

　　int main(int argc, char **argv){

　　

　　WSADATA wsaData;

　　SOCKET sockRaw;

　　struct sockaddr_in dest,from;

　　struct hostent * hp;

　　int bread,datasize;

　　int fromlen = sizeof(from);

　　int timeout = 1000;

　　char *dest_ip;

　　char *icmp_data;

　　char *recvbuf;

　　unsigned int addr=0;

　　USHORT seq_no = 0;

　　

　　if (WSAStartup(MAKEWORD(2,1),&wsaData) != 0){

　　fprintf(stderr,"WSAStartup failed: %d\n",GetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　

　　if (argc <2 ) {

　　Usage(argv[0]);

　　}

　　sockRaw = WSASocket (AF_INET,

　　SOCK_RAW,

　　IPPROTO_ICMP,

　　NULL, 0,0);

　　

　　if (sockRaw == INVALID_SOCKET) {

　　fprintf(stderr,"WSASocket() failed: %d\n",WSAGetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　bread = setsockopt(sockRaw,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout,

　　sizeof(timeout));

　　if(bread == SOCKET_ERROR) {

　　fprintf(stderr,"failed to set recv timeout: %d\n",WSAGetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　timeout = 1000;

　　bread = setsockopt(sockRaw,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout,

　　sizeof(timeout));

　　if(bread == SOCKET_ERROR) {

　　fprintf(stderr,"failed to set send timeout: %d\n",WSAGetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　memset(&dest,0,sizeof(dest));

　　

　　hp = gethostbyname(argv[1]);

　　

　　if (!hp){

　　addr = inet_addr(argv[1]);

　　}

　　if ((!hp) &&(addr == INADDR_NONE) ) {

　　fprintf(stderr,"Unable to resolve %s\n",argv[1]);

　　ExitProcess(STATUS_FAILED);

　　}

　　

　　if (hp != NULL)

　　memcpy(&(dest.sin_addr),hp->h_addr,hp->h_length);

　　else

　　dest.sin_addr.s_addr = addr;

　　

　　if (hp)

　　dest.sin_family = hp->h_addrtype;

　　else

　　dest.sin_family = AF_INET;

　　

　　dest_ip = inet_ntoa(dest.sin_addr);

　　

　　if (argc >2) {

　　datasize = atoi(argv[2]);

　　if (datasize == 0)

　　datasize = DEF_PACKET_SIZE;

　　

　　}

　　else

　　datasize = DEF_PACKET_SIZE;

　　

　　datasize += sizeof(IcmpHeader);

　　

　　icmp_data = xmalloc(MAX_PACKET);

　　recvbuf = xmalloc(MAX_PACKET);

　　

　　if (!icmp_data) {

　　fprintf(stderr,"HeapAlloc failed %d\n",GetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　

　　

　　memset(icmp_data,0,MAX_PACKET);

　　fill_icmp_data(icmp_data,datasize);

　　

　　while(1) {

　　int bwrote;

　　

　　((IcmpHeader*)icmp_data)->i_cksum = 0;

　　((IcmpHeader*)icmp_data)->timestamp = GetTickCount();

　　

　　((IcmpHeader*)icmp_data)->i_seq = seq_no++;

　　((IcmpHeader*)icmp_data)->i_cksum = checksum((USHORT*)icmp_data,

　　datasize);

　　

　　bwrote = sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)&dest,

　　sizeof(dest));

　　if (bwrote == SOCKET_ERROR){

　　if (WSAGetLastError() == WSAETIMEDOUT) {

　　printf("timed out\n");

　　continue;

　　}

　　fprintf(stderr,"sendto failed: %d\n",WSAGetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　if (bwrote

　　fprintf(stdout,"Wrote %d bytes\n",bwrote);

　　}

　　bread = recvfrom(sockRaw,recvbuf,MAX_PACKET,0,(struct sockaddr*)&from,

　　&fromlen);

　　if (bread == SOCKET_ERROR){

　　if (WSAGetLastError() == WSAETIMEDOUT) {

　　printf("timed out\n");

　　continue;

　　}

　　fprintf(stderr,"recvfrom failed: %d\n",WSAGetLastError());

　　ExitProcess(STATUS_FAILED);

　　}

　　decode_resp(recvbuf,bread,&from);

　　Sleep(1000);

　　

　　}

　　return 0;

　　

　　}

　　/*

　　The response is an IP packet. We must decode the IP header to locate

　　the ICMP data

　　*/

　　void decode_resp(char *buf, int bytes,struct sockaddr_in *from) {

　　

　　IpHeader *iphdr;

　　IcmpHeader *icmphdr;

　　unsigned short iphdrlen;

　　

　　iphdr = (IpHeader *)buf;

　　

　　iphdrlen = iphdr->h_len * 4 ;// number of 32-bit words *4 = bytes

　　

　　if (bytes

　　printf("Too few bytes from %s\n",inet_ntoa(from->sin_addr));

　　}

　　

　　icmphdr = (IcmpHeader*)(buf + iphdrlen);

　　

　　if (icmphdr->i_type != ICMP_ECHOREPLY) {

　　fprintf(stderr,"non-echo type %d recvd\n",icmphdr->i_type);

　　return;

　　}

　　if (icmphdr->i_id != (USHORT)GetCurrentProcessId()) {

　　fprintf(stderr,"someone else's packet!\n");

　　return ;

　　}

　　printf("%d bytes from %s:",bytes, inet_ntoa(from->sin_addr));

　　printf(" icmp_seq = %d. ",icmphdr->i_seq);

　　printf(" time: %d ms ",GetTickCount()-icmphdr->timestamp);

　　printf("\n");

　　

　　}

　　

　　

　　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);

　　}

　　/*

　　Helper function to fill in various stuff in our ICMP request.

　　*/

　　void fill_icmp_data(char * icmp_data, int datasize){

　　

　　IcmpHeader *icmp_hdr;

　　char *datapart;

　　

　　icmp_hdr = (IcmpHeader*)icmp_data;

　　

　　icmp_hdr->i_type = ICMP_ECHO;

　　icmp_hdr->i_code = 0;

　　icmp_hdr->i_id = (USHORT)GetCurrentProcessId();

　　icmp_hdr->i_cksum = 0;

　　icmp_hdr->i_seq = 0;

　　

　　datapart = icmp_data + sizeof(IcmpHeader);

　　//

　　// Place some junk in the buffer.

　　//

　　memset(datapart,'E', datasize - sizeof(IcmpHeader));

　　

　　}