一个简单的c语言版的DNS解析客户端，能提供解析耗时

成功的做了一次搬运工，该部分代码基本来自于http://blog.csdn.net/jxfgh，不过，原作者没有添加测试解析耗时的功能，其实我也不知道到时是不是原作者

作为第一次做与网络有关的东西，还是i挺有纪念意义的，所以发博文留念。有添加缓存功能的欲望，可惜不知从何下手，故没有改动，也希望会的童鞋能帮一下忙，搞定缓存的问题。

/********实现简单的DNS解析耗时测试**********
***************************************/
#include <stdio.h>
#include <Winsock2.h>
#include <windows.h>
#include <time.h>


typedef struct _DNSHEAD{   //dns 头部
  USHORT ID;
  USHORT tag;   // dns 标志(参数)
  USHORT numQ;  // 问题数
  USHORT numA;  // 答案数
  USHORT numA1;  // 权威答案数
  USHORT numA2;  // 附加答案数
}DnsHead;
typedef struct _DNSQUERY //dns 查询结构
{
//    char   name[64];
//   //查询的域名,这是一个大小在0到63之间的字符串；
   USHORT type;
   //查询类型，大约有20个不同的类型
   USHORT classes;
   //查询类,通常是A类既查询IP地址。
}DnsQuery;
#pragma comment(lib,"ws2_32.lib")


bool initWSA();// 初始化操作函数申明


void displayErrWSA(char *str);//显示错误函数申明


SOCKET CreateSocket(int type);//创建套接字函数申明
int MySendto(SOCKET sockTo, const char FAR * buf,int len,char *addr,USHORT port);//UDP sendto函数申明


int MyRecvFrom(SOCKET s, char FAR * buf,int len);//UDP recvfrom函数申明


bool SetDNSHead(char *name,char *buf);//设置DNS 头部函数申明
int main(int arg,char *are[])//主函数
{
 int Result=0;
 char buf[1024]={0};
 char addr[16] = "42.49.134.2";// dns 服务器地址
 char *name = 0; //要查询的域名
 
 if ( !initWSA() )//初始化
 {
  displayErrWSA("initWSA err!");
  return 1;
 }
 SOCKET sockTo ;//创建套接字
 if ( (sockTo = CreateSocket(SOCK_DGRAM)) == 0)
 {
  displayErrWSA("CreatSocket err!");
  return 1;
  }
 while (1)//死循环，不停的查询
 {
  if (arg < 2)
  {
   char temp[1024]={0};//设定要解析域名的字符数组
   printf("请输入要查询的域名:");
   scanf("%s",temp);//输入待解析域名
   if (temp[0] == 'q' ||temp[0] == 'Q')//判断是否输入的是”q“，如果是，退出
   {
    break;
   }
   name =  temp;
  }
  else
  {
   arg = 1;
   name =  are[1];
  }
  clock_t begin=0, end=0;//创建时间变量begin、end，分别用来保存解析开始可解析结束所对应的时间，单位为ms
  double cost;//设定统计变量，用来保存整个解析过程的耗时，由end减去begin得出。
  //设置dns 头部
  SetDNSHead(name,buf);
  //发送出去的请求数据长度
  int len = sizeof(DnsHead)+sizeof(DnsQuery)+strlen(name)+2;
  
  if ( ( Result =MySendto(sockTo,buf,len,addr,53) ) <= 0)//发送DNS 请求
  {
   displayErrWSA("sendto err!");
   continue;
  }
  begin = clock();//开始计时
  if ( (Result =MyRecvFrom(sockTo,buf,1024) ) <=  0)//接收应答
  {
   displayErrWSA("recvfrom err!");
   continue;
  }
  end = clock();//返回了结果，停止计时
  cost = (double)(end - begin); //计算解析时间差，单位：ms
  //简单的取得返回的 IP 地址( 收到的最后4字节 )
  DnsHead *DnsH = (DnsHead *)buf;
  if (DnsH->numA == 0)
  {
   printf("无法解析 %s 的IP 地址。\n",name);
   continue;
  }
  
  char *getIP =(char *)buf +Result - 4;
  printf("%s 的IP地址为: ",name);
  int Result= 0 ;//输出DNS解析结果，格式为A.B.C.D
  printf("%u",(UCHAR )getIP[Result]);
  for (Result= 1 ;Result<4 ;Result++)
  {
   printf(".%u",(UCHAR )getIP[Result]);
  }
   printf("  解析耗时：%.0fms",cost);
  printf("\n");
 }
 closesocket(sockTo);
 return 0;
}


bool initWSA()// 初始化操作函数
{
 WORD   wVersionRequested;
 WSADATA   wsaData;
 int Result;
 wVersionRequested = MAKEWORD( 2, 2 );
 Result = WSAStartup( wVersionRequested, &wsaData );
 if(Result   !=   0 ) 
 {
  return false;
 }
 
 if( LOBYTE( wsaData.wVersion) != 2 ||
    HIBYTE(wsaData.wVersion)!= 2 )
 {
  WSACleanup();
  return false;  
 }
 return true;
}


SOCKET CreateSocket(int type)//创建套接字
{
 SOCKET sock=socket(AF_INET,type,0);
 if (sock == INVALID_SOCKET )
 {
  return 0;
 }
 return sock;
}


/*****************         UDP sendto*/
int MySendto(SOCKET sockTo, const char FAR * buf,int len,char *addr,USHORT port)
{
 if (addr[0]<'0'||addr[0]>'9') //addr 中不是IP 地址，域名解析
 { 
  struct   sockaddr_in   addr_in;
  HOSTENT *host = NULL;
  host = gethostbyname(addr);//域名解析函数
  if ( host->h_addr_list[0]!=NULL )
  {
   memcpy  ( &addr_in.sin_addr.s_addr,
      host->h_addr_list[0], host->h_length);  
   addr = inet_ntoa(addr_in.sin_addr); 
  }
 }
 //设置发送数据到的 套接字及地址结构
 SOCKADDR_IN   addrTo;
 addrTo.sin_addr.S_un.S_addr=inet_addr(addr);
 addrTo.sin_family=AF_INET;
 addrTo.sin_port=htons(port);


 return sendto(  sockTo, buf, len, 0,
     (sockaddr*)&addrTo, sizeof(sockaddr)  );


}
//*********************************************/
/*********************** UDP recvform*/
int MyRecvFrom(SOCKET s, char FAR * buf,int len)
{
 //设置发送数据到的 套接字及地址结构
 SOCKADDR_IN   addrFrom;
 int addrlen = sizeof(SOCKADDR_IN);
 return recvfrom( s, buf, len, 0, (SOCKADDR *)&addrFrom, &addrlen);
}
/********************** 设置DNS 头部 */
int  ChName(char *fname,char *tname);//域名转化
bool SetDNSHead(char *name,char *buf)
{
 memset(buf,0,sizeof(DnsHead));
 //设置头部
 DnsHead *DnsH = (DnsHead *)buf;
 DnsH->ID = (USHORT)1;
 DnsH->tag = htons(0x0100);
 DnsH->numQ = htons(1);
 DnsH->numA = 0;
 DnsQuery *DnsQ =(DnsQuery *) ( buf+ sizeof(DnsHead) );
 int NameLen = ChName(name,(char *)DnsQ);
 //设置查询信息
 DnsQ = (DnsQuery *)( (char *)DnsQ + NameLen );
 DnsQ->classes = htons(1);
 DnsQ->type = htons(1);
 return true;
}
//显示错误信息
void displayErrWSA(char *str)
{
 printf("/n%s,err = %d/n",str,WSAGetLastError());
 getchar();
}
//域名转化
int  ChName(char *fname,char *tname)
{
 int j =0;
 int i =strlen(fname)-1;
 tname[i+2] = 0;
 int k = i+1;
 for (; i>=0;i--,k--)
 {
  if (fname[i] == '.')
  {
   tname[k] = j;
   j=0;
  }
  else
  {
   tname[k] = fname[i];
   j++;
  }
 }
 tname[k] = j;
 return strlen(tname)+1;
}