home *** CD-ROM | disk | FTP | other *** search
/ Tricks of the Windows Gam…ming Gurus (2nd Edition) / Disc2.iso / msdn_vcb / samples / vc98 / sdk / netds / winsock / ping / ping.c < prev    next >
C/C++ Source or Header  |  1997-10-08  |  7KB  |  282 lines
  1. /******************************************************************************\
  2. * ping.c - Simple ping utility using SOCK_RAW
  3. * 
  4. *       This is a part of the Microsoft Source Code Samples.
  5. *       Copyright 1996-1997 Microsoft Corporation.
  6. *       All rights reserved.
  7. *       This source code is only intended as a supplement to
  8. *       Microsoft Development Tools and/or WinHelp documentation.
  9. *       See these sources for detailed information regarding the
  10. *       Microsoft samples programs.
  11. \******************************************************************************/
  12.  
  13. #pragma pack(4)
  14.  
  15. #define WIN32_LEAN_AND_MEAN
  16. #include <winsock2.h>
  17. #include <stdio.h>
  18. #include <stdlib.h>
  19.  
  20. #define ICMP_ECHO 8
  21. #define ICMP_ECHOREPLY 0
  22.  
  23. #define ICMP_MIN 8 // minimum 8 byte icmp packet (just header)
  24.  
  25. /* The IP header */
  26. typedef struct iphdr {
  27.     unsigned int h_len:4;          // length of the header
  28.     unsigned int version:4;        // Version of IP
  29.     unsigned char tos;             // Type of service
  30.     unsigned short total_len;      // total length of the packet
  31.     unsigned short ident;          // unique identifier
  32.     unsigned short frag_and_flags; // flags
  33.     unsigned char  ttl; 
  34.     unsigned char proto;           // protocol (TCP, UDP etc)
  35.     unsigned short checksum;       // IP checksum
  36.  
  37.     unsigned int sourceIP;
  38.     unsigned int destIP;
  39.  
  40. }IpHeader;
  41.  
  42. //
  43. // ICMP header
  44. //
  45. typedef struct _ihdr {
  46.   BYTE i_type;
  47.   BYTE i_code; /* type sub code */
  48.   USHORT i_cksum;
  49.   USHORT i_id;
  50.   USHORT i_seq;
  51.   /* This is not the std header, but we reserve space for time */
  52.   ULONG timestamp;
  53. }IcmpHeader;
  54.  
  55. #define STATUS_FAILED 0xFFFF
  56. #define DEF_PACKET_SIZE 32
  57. #define MAX_PACKET 1024
  58.  
  59. #define xmalloc(s) HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,(s))
  60. #define xfree(p)   HeapFree (GetProcessHeap(),0,(p))
  61.  
  62. void fill_icmp_data(char *, int);
  63. USHORT checksum(USHORT *, int);
  64. void decode_resp(char *,int ,struct sockaddr_in *);
  65.  
  66. void Usage(char *progname){
  67.   
  68.   fprintf(stderr,"Usage:\n");
  69.   fprintf(stderr,"%s <host> [data_size]\n",progname);
  70.   fprintf(stderr,"datasize can be up to 1Kb\n");
  71.   ExitProcess(STATUS_FAILED);
  72.  
  73. }
  74. int main(int argc, char **argv){
  75.  
  76.   WSADATA wsaData;
  77.   SOCKET sockRaw;
  78.   struct sockaddr_in dest,from;
  79.   struct hostent * hp;
  80.   int bread,datasize;
  81.   int fromlen = sizeof(from);
  82.   int timeout = 1000;
  83.   char *dest_ip;
  84.   char *icmp_data;
  85.   char *recvbuf;
  86.   unsigned int addr=0;
  87.   USHORT seq_no = 0;
  88.  
  89.   if (WSAStartup(MAKEWORD(2,1),&wsaData) != 0){
  90.     fprintf(stderr,"WSAStartup failed: %d\n",GetLastError());
  91.     ExitProcess(STATUS_FAILED);
  92.   }
  93.  
  94.   if (argc <2 ) {
  95.     Usage(argv[0]);
  96.   }
  97.   sockRaw = WSASocket (AF_INET,
  98.                        SOCK_RAW,
  99.                        IPPROTO_ICMP,
  100.                        NULL, 0,0);
  101.   
  102.   if (sockRaw == INVALID_SOCKET) {
  103.     fprintf(stderr,"WSASocket() failed: %d\n",WSAGetLastError());
  104.     ExitProcess(STATUS_FAILED);
  105.   }
  106.   bread = setsockopt(sockRaw,SOL_SOCKET,SO_RCVTIMEO,(char*)&timeout,
  107.                       sizeof(timeout));
  108.   if(bread == SOCKET_ERROR) {
  109.       fprintf(stderr,"failed to set recv timeout: %d\n",WSAGetLastError());
  110.     ExitProcess(STATUS_FAILED);
  111.   }
  112.   timeout = 1000;
  113.   bread = setsockopt(sockRaw,SOL_SOCKET,SO_SNDTIMEO,(char*)&timeout,
  114.                       sizeof(timeout));
  115.   if(bread == SOCKET_ERROR) {
  116.       fprintf(stderr,"failed to set send timeout: %d\n",WSAGetLastError());
  117.     ExitProcess(STATUS_FAILED);
  118.   }
  119.   memset(&dest,0,sizeof(dest));
  120.  
  121.   hp = gethostbyname(argv[1]);
  122.  
  123.   if (!hp){
  124.     addr = inet_addr(argv[1]);
  125.   }
  126.   if ((!hp)  && (addr == INADDR_NONE) ) {
  127.     fprintf(stderr,"Unable to resolve %s\n",argv[1]);
  128.     ExitProcess(STATUS_FAILED);
  129.   }
  130.  
  131.   if (hp != NULL)
  132.       memcpy(&(dest.sin_addr),hp->h_addr,hp->h_length);
  133.   else
  134.       dest.sin_addr.s_addr = addr;
  135.  
  136.   if (hp)
  137.       dest.sin_family = hp->h_addrtype;
  138.   else
  139.       dest.sin_family = AF_INET;
  140.  
  141.   dest_ip = inet_ntoa(dest.sin_addr);
  142.  
  143.   if (argc >2) {
  144.     datasize = atoi(argv[2]);
  145.     if (datasize == 0)
  146.       datasize = DEF_PACKET_SIZE;
  147.  
  148.   }
  149.   else
  150.       datasize = DEF_PACKET_SIZE;
  151.     
  152.   datasize += sizeof(IcmpHeader);  
  153.  
  154.   icmp_data = xmalloc(MAX_PACKET);
  155.   recvbuf = xmalloc(MAX_PACKET);
  156.  
  157.   if (!icmp_data) {
  158.     fprintf(stderr,"HeapAlloc failed %d\n",GetLastError());
  159.     ExitProcess(STATUS_FAILED);
  160.   }
  161.   
  162.  
  163.   memset(icmp_data,0,MAX_PACKET);
  164.   fill_icmp_data(icmp_data,datasize);
  165.  
  166.   while(1) {
  167.     int bwrote;
  168.     
  169.     ((IcmpHeader*)icmp_data)->i_cksum = 0;
  170.     ((IcmpHeader*)icmp_data)->timestamp = GetTickCount();
  171.  
  172.     ((IcmpHeader*)icmp_data)->i_seq = seq_no++;
  173.     ((IcmpHeader*)icmp_data)->i_cksum = checksum((USHORT*)icmp_data, 
  174.                                             datasize);
  175.  
  176.     bwrote = sendto(sockRaw,icmp_data,datasize,0,(struct sockaddr*)&dest,
  177.                     sizeof(dest));
  178.     if (bwrote == SOCKET_ERROR){
  179.       if (WSAGetLastError() == WSAETIMEDOUT) {
  180.           printf("timed out\n");
  181.         continue;
  182.       }
  183.       fprintf(stderr,"sendto failed: %d\n",WSAGetLastError());
  184.       ExitProcess(STATUS_FAILED);
  185.     }
  186.     if (bwrote < datasize ) {
  187.       fprintf(stdout,"Wrote %d bytes\n",bwrote);
  188.     }
  189.     bread = recvfrom(sockRaw,recvbuf,MAX_PACKET,0,(struct sockaddr*)&from,
  190.                  &fromlen);
  191.     if (bread == SOCKET_ERROR){
  192.       if (WSAGetLastError() == WSAETIMEDOUT) {
  193.           printf("timed out\n");
  194.         continue;
  195.       }
  196.       fprintf(stderr,"recvfrom failed: %d\n",WSAGetLastError());
  197.       ExitProcess(STATUS_FAILED);
  198.     }
  199.     decode_resp(recvbuf,bread,&from);
  200.     Sleep(1000);
  201.  
  202.   }
  203.   return 0;
  204.  
  205. }
  206. /* 
  207.     The response is an IP packet. We must decode the IP header to locate 
  208.     the ICMP data 
  209. */
  210. void decode_resp(char *buf, int bytes,struct sockaddr_in *from) {
  211.  
  212.     IpHeader *iphdr;
  213.     IcmpHeader *icmphdr;
  214.     unsigned short iphdrlen;
  215.  
  216.     iphdr = (IpHeader *)buf;
  217.  
  218.     iphdrlen = iphdr->h_len * 4 ; // number of 32-bit words *4 = bytes
  219.  
  220.     if (bytes  < iphdrlen + ICMP_MIN) {
  221.         printf("Too few bytes from %s\n",inet_ntoa(from->sin_addr));
  222.     }
  223.  
  224.     icmphdr = (IcmpHeader*)(buf + iphdrlen);
  225.  
  226.     if (icmphdr->i_type != ICMP_ECHOREPLY) {
  227.         fprintf(stderr,"non-echo type %d recvd\n",icmphdr->i_type);
  228.         return;
  229.     }
  230.     if (icmphdr->i_id != (USHORT)GetCurrentProcessId()) {
  231.         fprintf(stderr,"someone else's packet!\n");
  232.         return ;
  233.     }
  234.     printf("%d bytes from %s:",bytes, inet_ntoa(from->sin_addr));
  235.     printf(" icmp_seq = %d. ",icmphdr->i_seq);
  236.     printf(" time: %d ms ",GetTickCount()-icmphdr->timestamp);
  237.     printf("\n");
  238.         
  239. }
  240.  
  241.  
  242. USHORT checksum(USHORT *buffer, int size) {
  243.  
  244.   unsigned long cksum=0;
  245.  
  246.   while(size >1) {
  247.     cksum+=*buffer++;
  248.     size -=sizeof(USHORT);
  249.   }
  250.   
  251.   if(size ) {
  252.     cksum += *(UCHAR*)buffer;
  253.   }
  254.  
  255.   cksum = (cksum >> 16) + (cksum & 0xffff);
  256.   cksum += (cksum >>16);
  257.   return (USHORT)(~cksum);
  258. }
  259. /* 
  260.     Helper function to fill in various stuff in our ICMP request.
  261. */
  262. void fill_icmp_data(char * icmp_data, int datasize){
  263.  
  264.   IcmpHeader *icmp_hdr;
  265.   char *datapart;
  266.  
  267.   icmp_hdr = (IcmpHeader*)icmp_data;
  268.  
  269.   icmp_hdr->i_type = ICMP_ECHO;
  270.   icmp_hdr->i_code = 0;
  271.   icmp_hdr->i_id = (USHORT)GetCurrentProcessId();
  272.   icmp_hdr->i_cksum = 0;
  273.   icmp_hdr->i_seq = 0;
  274.   
  275.   datapart = icmp_data + sizeof(IcmpHeader);
  276.   //
  277.   // Place some junk in the buffer.
  278.   //
  279.   memset(datapart,'E', datasize - sizeof(IcmpHeader));
  280.  
  281. }
  282.