ping小程序

#include <WinSock2.h>
#include <stdio.h>
#include <string.h>
#pragma comment(lib, "ws2_32")

typedef struct tagIPINFO
{
 u_char Ttl;    // Time To Live
 u_char Tos;    // Type Of Service
 u_char IPFlags;   // IP flags
 u_char OptSize;   // Size of options data
 u_char FAR *Options; // Options data buffer
}IPINFO, *PIPINFO;

typedef struct tagICMPECHO
{
 u_long Source;   // Source address
 u_long Status;   // IP status
 u_long RTTime;   // Round trip time in milliseconds
 u_short DataSize;  // Reply data size
 u_short Reserved;  // Unknown
 void FAR *pData;  // Reply data buffer
 IPINFO ipInfo;   // Reply options
}ICMPECHO, *PICMPECHO;

// ICMP.DLL Export Function Pointers
HANDLE (WINAPI *pIcmpCreateFile)(VOID);
BOOL (WINAPI *pIcmpCloseHandle)(HANDLE);
DWORD (WINAPI *pIcmpSendEcho)
(HANDLE,DWORD,LPVOID,WORD,PIPINFO,LPVOID,DWORD,DWORD);

//
//
void main(int argc, char **argv)
{
 WSADATA wsaData;   // WSADATA
 ICMPECHO icmpEcho;   // ICMP Echo reply buffer
 HMODULE hndlIcmp;   // LoadLibrary() handle to ICMP.DLL
 HANDLE hndlFile;   // Handle for IcmpCreateFile()
 LPHOSTENT pHost;   // Pointer to host entry structure
 struct in_addr iaDest;  // Internet address structure
 DWORD *dwAddress;   // IP Address
 IPINFO ipInfo;    // IP Options structure
 int nRet;     // General use return code
 DWORD dwRet;    // DWORD return code
 int x;

 // Check arguments
 if (argc != 2)
 {
  fprintf(stderr,"/nSyntax: pingi HostNameOrIPAddress/n");
  return;
 }

 // Dynamically load the ICMP.DLL
 hndlIcmp = LoadLibrary("ICMP.DLL");
 if (hndlIcmp == NULL)
 {
  fprintf(stderr,"/nCould not load ICMP.DLL/n");
  return;
 }
 // Retrieve ICMP function pointers
 pIcmpCreateFile = (HANDLE (WINAPI *)(void))
  GetProcAddress(hndlIcmp,"IcmpCreateFile");
 pIcmpCloseHandle = (BOOL (WINAPI *)(HANDLE))
  GetProcAddress(hndlIcmp,"IcmpCloseHandle");
 pIcmpSendEcho = (DWORD (WINAPI *)
  (HANDLE,DWORD,LPVOID,WORD,PIPINFO,LPVOID,DWORD,DWORD))
  GetProcAddress(hndlIcmp,"IcmpSendEcho");
 // Check all the function pointers
 if (pIcmpCreateFile == NULL  ||
  pIcmpCloseHandle == NULL ||
  pIcmpSendEcho == NULL)
 {
  fprintf(stderr,"/nError getting ICMP proc address/n");
  FreeLibrary(hndlIcmp);
  return;
 }

 // Init WinSock
 nRet = WSAStartup(0x0101, &wsaData );
 if (nRet)
 {
  fprintf(stderr,"/nWSAStartup() error: %d/n", nRet);
  WSACleanup();
  FreeLibrary(hndlIcmp);
  return;
 }
 // Check WinSock version
 if (0x0101 != wsaData.wVersion)
 {
  fprintf(stderr,"/nWinSock version 1.1 not supported/n");
  WSACleanup();
  FreeLibrary(hndlIcmp);
  return;
 }

 // Lookup destination
 // Use inet_addr() to determine if we're dealing with a name
 // or an address
 iaDest.s_addr = inet_addr(argv[1]);
 if (iaDest.s_addr == INADDR_NONE)
  pHost = gethostbyname(argv[1]);
 else
  pHost = gethostbyaddr((const char *)&iaDest,
  sizeof(struct in_addr), AF_INET);
 if (pHost == NULL)
 {
  fprintf(stderr, "/n%s not found/n", argv[1]);
  WSACleanup();
  FreeLibrary(hndlIcmp);
  return;
 }

 // Tell the user what we're doing
 printf("/nPinging %s [%s]", pHost->h_name,
  inet_ntoa((*(LPIN_ADDR)pHost->h_addr_list[0])));

 // Copy the IP address
 dwAddress = (DWORD *)(*pHost->h_addr_list);

 // Get an ICMP echo request handle       
 hndlFile = pIcmpCreateFile();
 for (x = 0; x < 4; x++)
 {
  // Set some reasonable default values
  ipInfo.Ttl = 255;
  ipInfo.Tos = 0;
  ipInfo.IPFlags = 0;
  ipInfo.OptSize = 0;
  ipInfo.Options = NULL;
  //icmpEcho.ipInfo.Ttl = 256;
  // Reqest an ICMP echo
  dwRet = pIcmpSendEcho(
   hndlFile,  // Handle from IcmpCreateFile()
   *dwAddress,  // Destination IP address
   NULL,   // Pointer to buffer to send
   0,    // Size of buffer in bytes
   &ipInfo,  // Request options
   &icmpEcho,  // Reply buffer
   sizeof(struct tagICMPECHO),
   5000);   // Time to wait in milliseconds
  // Print the results
  iaDest.s_addr = icmpEcho.Source;
  printf("/nReply from %s  Time=%ldms  TTL=%d",
   inet_ntoa(iaDest),
   icmpEcho.RTTime,
   icmpEcho.ipInfo.Ttl);
  if (icmpEcho.Status)
  {
   printf("/nError: icmpEcho.Status=%ld",
    icmpEcho.Status);
   break;
  }
 }
 printf("/n");
 // Close the echo request file handle
 pIcmpCloseHandle(hndlFile);
 FreeLibrary(hndlIcmp);
 WSACleanup();
}

/////////////////////////////////////////////////////////////////////使用ICMP协议//////////////////////////////////////////////////////////////////////////////////////////////////////////

#pragma pack(4)

#include <WinSock2.h>
#include <iostream>

#pragma comment(lib, "Ws2_32")

using namespace std;

#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 icmphdr {
 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 DEF_PACKET_NUMBER  4    /* 发送数据报的个数 */
#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);
int decode_resp(char *,int ,struct sockaddr_in *);

void Usage(char *progname){

 fprintf(stderr,"Usage:/n");
 fprintf(stderr,"%s [number of packets] [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,times;
 int fromlen = sizeof(from);
 int timeout = 1000;
 int statistic = 0;  /* 用于统计结果 */ 
 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,WSA_FLAG_OVERLAPPED);

 //
 //注：为了使用发送接收超时设置(即设置SO_RCVTIMEO, SO_SNDTIMEO)，
 //    必须将标志位设为WSA_FLAG_OVERLAPPED !
 //

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

 //
 //  atoi函数原型是: int atoi( const char *string );
 //  The return value is 0 if the input cannot be converted to an integer !
 //
 if(argc>2)
 {
  times=atoi(argv[2]);
  if(times == 0)
   times=DEF_PACKET_NUMBER;
 }
 else
  times=DEF_PACKET_NUMBER;

 if (argc >3)
 {
  datasize = atoi(argv[3]);
  if (datasize == 0)
   datasize = DEF_PACKET_SIZE;
  if (datasize >1024)   /* 用户给出的数据包大小太大 */
  {
   fprintf(stderr,"WARNING : data_size is too large !/n");
   datasize = DEF_PACKET_SIZE;
  }
 }
 else
  datasize = DEF_PACKET_SIZE;

 datasize += sizeof(IcmpHeader);

 icmp_data = (char*)xmalloc(MAX_PACKET);
 recvbuf = (char*)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);

 //
 //显示提示信息
 //
 fprintf(stdout,"/nPinging %s ..../n/n",dest_ip);

 for(int i=0;i < 4; i++)
 {
  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("Request timed out./n");
    continue;
   }
   fprintf(stderr,"sendto failed: %d/n",WSAGetLastError());
   ExitProcess(STATUS_FAILED);
  }
  if (bwrote < datasize ) {
   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("Request timed out./n");
    continue;
   }
   fprintf(stderr,"recvfrom failed: %d/n",WSAGetLastError());
   ExitProcess(STATUS_FAILED);
  }
  if(!decode_resp(recvbuf,bread,&from))
   statistic++; /* 成功接收的数目++ */
  Sleep(1000);

 }

 /*
 Display the statistic result
 */
 fprintf(stdout,"/nPing statistics for %s /n",dest_ip);
 fprintf(stdout,"    Packets: Sent = %d,Received = %d, Lost = %d (%2.0f%% loss)/n",times,
  statistic,(times-statistic),(float)(times-statistic)/times*100);

 WSACleanup();
 return 0;

}
/*
The response is an IP packet. We must decode the IP header to locate
the ICMP data
*/
int 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 < iphdrlen + ICMP_MIN) {
  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 1;
 }
 if (icmphdr->i_id != (USHORT)GetCurrentProcessId()) {
  fprintf(stderr,"someone else's packet!/n");
  return 1;
 }
 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");
 return 0;

}

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

}