操作系统  办公  实用知识  设计  开发  WEB开发  移动开发  数据库  软件工程  网管  安全  管理  信息化  答疑  渠道 

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

2008-1-16 网友评论 0 条 点击进入论坛

  #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: ");

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

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

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

  continue;

  }

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

  ExitProcess(STATUS_FAILED);

  }

  if (bwrote < datasize ) {

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

  }

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

  &fromlen);

  if (bread == SOCKET_ERROR){

  if (WSAGetLastError() == WSAETIMEDOUT) {

  printf("timed out ");

  continue;

  }

  fprintf(stderr,"recvfrom failed: %d ",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 ",inet_ntoa(from->sin_addr));

  }

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

  if (icmphdr->i_type != ICMP_ECHOREPLY) {

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

  return;

  }

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

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

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

  }

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

  }

已有 0 位对此文章感兴趣的网友发布了看法    
我来评两句 用户名: 密码:
  匿名发表
今日推荐
技术文库(共有 45972 篇文章)
操作系统
办公软件
实用知识
网络管理
软件开发
WEB开发
软件工程
数据库
设计在线
信息安全
行业信息化
管理信息化
移动开发
重点推荐
电子杂志订阅
点击电子杂志名称查看样刊
输入E-mail地址即可订阅
E-mail