winsock 自己写个ping 程序

最新推荐文章于 2022-04-21 11:47:52 发布

heroyuchao

最新推荐文章于 2022-04-21 11:47:52 发布

阅读量1.9k

点赞数 1

分类专栏： C/C++代码篇文章标签： socket include struct

本文链接：https://blog.csdn.net/heroyuchao/article/details/1930928

版权

C/C++代码篇专栏收录该内容

9 篇文章 0 订阅

订阅专栏

#include "stdafx.h"
#include <Winsock2.h>
#include <Ws2tcpip.h>

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

void InitSocket()
{
WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD( 2, 2 );
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) {
  return;
}
if ( LOBYTE( wsaData.wVersion ) != 2 ||
   HIBYTE( wsaData.wVersion ) != 2 ) {
  WSACleanup( );
  return;
}
}

//icmp包结构
typedef struct _ICMPPACK
{
unsigned char icmp_type; //消息类型
unsigned char icmp_code; //清息代码
unsigned short icmp_checksum; //16位效验

unsigned short icmp_id;//用来唯一标识些请求的ID号
unsigned short icmp_sequence; //序列号
unsigned long icmp_timestamp; //时间戳

}ICMPPACK, *PICMPPACK;

BOOL SetTimeOut(SOCKET s,int nTime,BOOL bRecv)
{
int nRet = setsockopt(s,SOL_SOCKET,bRecv?SO_RCVTIMEO:SO_SNDTIMEO,(char *)&nTime,sizeof(nTime));
if(SOCKET_ERROR == nRet)
{
return false;
}
return true;
}

int GetTTL(SOCKET s)
{
int TTLCounts;
int nLen=sizeof(TTLCounts);
int nRet=getsockopt(s,IPPROTO_IP,IP_TTL,(char *)&TTLCounts,&nLen);
return TTLCounts;
}

unsigned short checksum(unsigned short *buff,int size)
{
unsigned long cksum=0;
while(size>1)
{
  cksum+=*buff++;
  size-=sizeof(unsigned short);
}
if(size)
{
  cksum+=*(unsigned char*)buff;
}
cksum=(cksum>>16)+(cksum&0xffff);
cksum+=cksum>>16;
return (unsigned short)(~cksum);
}

int _tmain(int argc, _TCHAR* argv[])
{
InitSocket();
SOCKET sRaw=socket(AF_INET,SOCK_RAW,IPPROTO_ICMP);
if(INVALID_SOCKET == sRaw)
{
  printf("failed to create raw socket/n");
  return 0;
}
if(!SetTimeOut(sRaw,1000,true))
{
  printf("failed to set recv time out/n");
  closesocket(sRaw);
  return 0;
}
//设置目标地址
sockaddr_in desAddr;
desAddr.sin_addr.S_un.S_addr=inet_addr("127.0.0.1");
desAddr.sin_port=htons(0);
desAddr.sin_family=AF_INET;

    char buff[sizeof(ICMPPACK)+32]; //创建ICMP包
PICMPPACK pICMP=(PICMPPACK)buff;

pICMP->icmp_type=8; //请求回显
pICMP->icmp_code=0;
pICMP->icmp_checksum=0;
pICMP->icmp_id=(unsigned short)GetCurrentProcessId();
pICMP->icmp_sequence=0;
pICMP->icmp_timestamp=0;

memset(&buff[sizeof(ICMPPACK)],'E',32); //填充用户数据

unsigned short nSeq=0;
char recvBuf[1024];
sockaddr_in from;
int nLen=sizeof(sockaddr_in);
int count=0;
int nRet;
while(true)
{
  if(count++==4)
  {
   break;
  }
  pICMP->icmp_checksum=0;
  pICMP->icmp_sequence=nSeq++;
  pICMP->icmp_timestamp=::GetTickCount();
  pICMP->icmp_checksum=::checksum((unsigned short *)buff,sizeof(ICMPPACK)+32);
     nRet=sendto(sRaw,buff,sizeof(ICMPPACK)+32,0,(sockaddr *)&desAddr,sizeof(desAddr));
  if(nRet==SOCKET_ERROR)
  {
   printf("sendto() failed /n");
   closesocket(sRaw);
   return 0;
  }
  nRet=recvfrom(sRaw,recvBuf,1024,0,(sockaddr *)&from,&nLen);
  if(nRet == SOCKET_ERROR)
  {
   if(WSAGetLastError()==WSAETIMEDOUT)
   {
    printf("time out /n");
    continue;
   }
   else
   {
    printf("recvfrom() failed /n");
    closesocket(sRaw);
    return 0;
   }
  }
  int nTick=GetTickCount();

  if(nRet<20+sizeof(ICMPPACK))
  {
   printf("too few bytes from %s /n",::inet_ntoa(from.sin_addr));
   continue;
  }
  ICMPPACK *pTemp=(ICMPPACK *)(recvBuf+20);
  if(pTemp->icmp_type!=0)
  {
   printf(" nonecho type recvd /n");
   closesocket(sRaw);
   return 0;
  }
  if(pTemp->icmp_id!=(unsigned short)GetCurrentProcessId())
  {
   printf("some one else packet");
   closesocket(sRaw);
   return 0;
  }
  printf("%d Replay from %s: bytes=%d time<%dms TTL=%d/n",pTemp->icmp_sequence,inet_ntoa(from.sin_addr),nRet,nTick-pTemp->icmp_timestamp,GetTTL(sRaw));
  ::Sleep(2000);
}
return 0;
}