nread c语言,Ping 源代码-C语言

#include "winsock2.h" #include "ws2tcpip.h" #include "stdio.h" #include "stdlib.h" #include "windows.h"

#define IP_RECORD_ROUTE  0x7 #define WIN32_LEAN_AND_MEAN

#pragma comment (lib,"ws2_32.lib")

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;            // Time to live  unsigned char  proto;          // Protocol (TCP, UDP etc)  unsigned short checksum;       // IP checksum

unsigned int   sourceIP;  unsigned int   destIP; } IpHeader;

#define ICMP_ECHO        8 #define ICMP_ECHOREPLY   0 #define ICMP_MIN         8 // Minimum 8-byte ICMP packet (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 standard header, but we reserve space for time  ULONG  timestamp; } IcmpHeader;

typedef struct _ipoptionhdr {  unsigned char        code;        // Option type  unsigned char        len;         // Length of option hdr  unsigned char        ptr;         // Offset into options  unsigned long        addr[9];     // List of IP addrs } IpOptionHeader;

#define DEF_PACKET_SIZE  32        // Default packet size #define MAX_PACKET       1024      // Max ICMP packet size #define MAX_IP_HDR_SIZE  60        // Max IP header size w/options

BOOL  bRecordRoute; int   datasize; char *lpdest;

void usage(char *progname)   //使用方法 {  printf("usage: ping -r [data size]\n");  printf("       -r           record route\n");  printf("        host        remote machine to ping\n");  printf("        datasize    can be up to 1KB\n");  ExitProcess(-1); }

void FillICMPData(char *icmp_data, int datasize)  //填充数据 {  IcmpHeader *icmp_hdr = NULL;  char       *datapart = NULL;

icmp_hdr = (IcmpHeader*)icmp_data;  icmp_hdr->i_type = ICMP_ECHO;        // Request an 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)); }

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

void DecodeIPOptions(char *buf, int bytes)  //IP头选项解码 {  IpOptionHeader *ipopt = NULL;  IN_ADDR         inaddr;  int             i;  HOSTENT        *host = NULL;

ipopt = (IpOptionHeader *)(buf + 20);

printf("RR:   ");  for(i = 0; i < (ipopt->ptr / 4) - 1; i++)  {   inaddr.S_un.S_addr = ipopt->addr[i];   if (i != 0)    printf("      ");   host = gethostbyaddr((char *)&inaddr.S_un.S_addr,    sizeof(inaddr.S_un.S_addr), AF_INET);   if (host)    printf("(%-15s) %s\n", inet_ntoa(inaddr), host->h_name);   else    printf("(%-15s)\n", inet_ntoa(inaddr));      }    return; }

void DecodeICMPHeader(char *buf, int bytes, struct sockaddr_in *from) //ICMP解码函数 {  IpHeader       *iphdr = NULL;  IcmpHeader     *icmphdr = NULL;  unsigned short  iphdrlen;  DWORD           tick;  static   int    icmpcount = 0;

iphdr = (IpHeader *)buf;  // Number of 32-bit words * 4 = bytes  iphdrlen = iphdr->h_len * 4;  tick = GetTickCount();

if ((iphdrlen == MAX_IP_HDR_SIZE) && (!icmpcount))   DecodeIPOptions(buf, 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)   {   printf("nonecho type %d recvd\n", icmphdr->i_type);   return;  }  // Make sure this is an ICMP reply to something we sent!  //  if (icmphdr->i_id != (USHORT)GetCurrentProcessId())   {   printf("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", tick - icmphdr->timestamp);  printf("\n");

icmpcount++;  return; }

void ValidateArgs(int argc, char **argv)   //处理输入参数 {  int                i;

bRecordRoute = FALSE;  lpdest = NULL;  datasize = DEF_PACKET_SIZE;

for(i = 1; i < argc; i++)  {   if ((argv[i][0] == '-') || (argv[i][0] == '/'))   {    switch (tolower(argv[i][1]))    {    case 'r':        // Record route option     bRecordRoute = TRUE;     break;    default:     usage(argv[0]);     break;    }   }   else if (isdigit(argv[i][0]))    datasize = atoi(argv[i]);   else    lpdest = argv[i];  } }

int main(int argc, char **argv) {  WSADATA            wsaData;  SOCKET             sockRaw = INVALID_SOCKET;  struct sockaddr_in dest,   from;  int                bread,   fromlen = sizeof(from),   timeout = 1000,   ret;  char              *icmp_data = NULL,   *recvbuf = NULL;  unsigned int       addr = 0;  USHORT             seq_no = 0;  struct hostent    *hp = NULL;  IpOptionHeader     ipopt;  if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)  {   printf("WSAStartup() failed: %d\n", GetLastError());   return -1;  }  ValidateArgs(argc, argv);  //处理输入参数

sockRaw = WSASocket (AF_INET, SOCK_RAW, IPPROTO_ICMP, NULL, 0,   WSA_FLAG_OVERLAPPED);    //创建原始套接字  if (sockRaw == INVALID_SOCKET)   {   printf("WSASocket() failed: %d\n", WSAGetLastError());   return -1;  }  if (bRecordRoute)  {   // Setup the IP option header to go out on every ICMP packet   //   ZeroMemory(&ipopt, sizeof(ipopt));   ipopt.code = IP_RECORD_ROUTE; // Record route option   ipopt.ptr  = 4;               // Point to the first addr offset   ipopt.len  = 39;              // Length of option header

ret = setsockopt(sockRaw, IPPROTO_IP, IP_OPTIONS,     (char *)&ipopt, sizeof(ipopt));   if (ret == SOCKET_ERROR)   {    printf("setsockopt(IP_OPTIONS) failed: %d\n",      WSAGetLastError());   }  }  // Set the send/recv timeout values  //  bread = setsockopt(sockRaw, SOL_SOCKET, SO_RCVTIMEO,    (char*)&timeout, sizeof(timeout));  if(bread == SOCKET_ERROR)   {   printf("setsockopt(SO_RCVTIMEO) failed: %d\n",     WSAGetLastError());   return -1;  }  timeout = 1000;  bread = setsockopt(sockRaw, SOL_SOCKET, SO_SNDTIMEO,    (char*)&timeout, sizeof(timeout));  if (bread == SOCKET_ERROR)   {   printf("setsockopt(SO_SNDTIMEO) failed: %d\n",     WSAGetLastError());   return -1;  }  memset(&dest, 0, sizeof(dest));  //  // Resolve the endpoint's name if necessary  //  dest.sin_family = AF_INET;  if ((dest.sin_addr.s_addr = inet_addr(lpdest)) == INADDR_NONE)  {   if ((hp = gethostbyname(lpdest)) != NULL)   {    memcpy(&(dest.sin_addr), hp->h_addr, hp->h_length);    dest.sin_family = hp->h_addrtype;    printf("dest.sin_addr = %s\n", inet_ntoa(dest.sin_addr));   }   else   {    printf("gethostbyname() failed: %d\n",      WSAGetLastError());    return -1;   }  }

//   // Create the ICMP packet  //         datasize += sizeof(IcmpHeader);

icmp_data = (char*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,MAX_PACKET);  recvbuf = (char*)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,MAX_PACKET);

if (!icmp_data)   {   printf("HeapAlloc() failed: %d\n", GetLastError());   return -1;  }  memset(icmp_data,0,MAX_PACKET);  //初始化icmp_data数据缓冲区  FillICMPData(icmp_data,datasize);   //填冲ICMP数据  //  // Start sending/receiving ICMP packets  //  while(1)   {   static int nCount = 0;   int        bwrote;

if (nCount++ == 4)     break;

((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));  //发送ICMP数据包   if (bwrote == SOCKET_ERROR)   {    if (WSAGetLastError() == WSAETIMEDOUT)     {     printf("timed out\n");     continue;    }    printf("sendto() failed: %d\n", WSAGetLastError());    return -1;   }   if (bwrote < datasize)    {    printf("Wrote %d bytes\n", bwrote);   }   bread = recvfrom(sockRaw, recvbuf, MAX_PACKET, 0,     (struct sockaddr*)&from, &fromlen);  //接受ICMP数据包   if (bread == SOCKET_ERROR)   {    if (WSAGetLastError() == WSAETIMEDOUT)     {     printf("timed out\n");     continue;    }    printf("recvfrom() failed: %d\n", WSAGetLastError());    return -1;   }   DecodeICMPHeader(recvbuf, bread, &from);  //ICMP解码

Sleep(1000);  }  // Cleanup  //  if (sockRaw != INVALID_SOCKET)    closesocket(sockRaw);  HeapFree(GetProcessHeap(), 0, recvbuf);  HeapFree(GetProcessHeap(), 0, icmp_data);

WSACleanup();  return 0;

}