ping(1)

 
#pragma pack(4)
#define WIN32_LEAN_AND_MEAN
#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>
#pragma comment(lib,"Ws2_32");
#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 <host> [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 = (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);
­
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 < 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("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 < 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;
}
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));
­
­
}
阅读更多
换一批

没有更多推荐了,返回首页