-
背景
在调用gethostbyname函数时会遇到阻塞10秒钟问题,为了控制阻塞时间,在调用之前查询DNS信息,类似nslookup命令。只当查询DNS正确返回时,才调用gethostbyname函数,也即自己控制阻塞时间。 -
根据RFC1035文档,DNS查询采用53端口,可使用UDP/TCP协议。
DNS查询/回复包格式为:
ID(16bits):标识符,一般填入本进程的标识符
QR(1bits):标志位,查询包为0,回复包为1
Opcode(4bits):查询的种类,标准查询为0
QDCOUNT(16bits):DNS查询/回复包数据部分Question字段的个数
ANCOUNT(16bits):DNS查询/回复包数据部分Answer字段的个数DNS查询/回复包数据部分格式为:
Name(不定长):域名
TYPE(16bits):查询类型/回复包RDATA类型,比如TYPE=1表示主机IP地址、TYPE=5表示CNAME,详见RFC1035 3.2.2
CLASS(16bits):类,一般情况下CLASS=1表示Internet,详见RFC1035 3.2.4
TTL(32bits,仅回复包):生存时间
RDLENGTH(16bits,仅回复包):RDATA部分的字节数
RDATA(不定长,仅回复包):资源数据,具体格式取决于TYPE和CLASS,比如TYPE=1、CLASS=1时,RDATA为四个字节的IP地址对域名的解析规则为:标签内容长度(1个字节) + 标签内容,以标签内容长度0作为Name的结束符
例如: www.csdn.com --> 3www4csdn3com (需16进制表示),最后加0结束 -
实现如下
重点关注组装报文的逻辑
- dns_lookup.h 头文件
#ifndef NET_DNS_TOOL_H
#define NET_DNS_TOOL_H
#include <string>
#include <netinet/in.h>
class NetDnsTool
{
public:
NetDnsTool();
~NetDnsTool();
bool Init();
std::string GetDnsServer(void);
bool DnsLookup(const std::string &dnsServer, const std::string &domain, long timeOut = 5);
private:
bool QueryInternal(const std::string &dnsServer, const std::string &domain, long timeOut = 5);
bool SendDnsRequest(const struct sockaddr_in &addr, const std::string &domain);
bool RecvDnsResponse(const struct sockaddr_in &addr, long timeOut);
std::string EncodeDomain(const std::string &domain);
private:
int mSocket;
unsigned short mProcessId;
};
#endif
- dns_lookup.cpp
#include "dns_lookup.h"
#include <fstream>
#include <memory>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <time.h>
constexpr char DNS_CONF_FILE[] = "/etc/resolv.conf";
constexpr int RECV_BUF_LENGTH = 1024;
constexpr int DNS_PORT = 53;
constexpr int MAX_DOMAIN_LENGTH = 255;
constexpr int DNS_TYPE_SIZE = 2;
constexpr int DNS_CLASS_SIZE = 2;
struct DnsHeader
{
unsigned short transId;
unsigned short flags;
unsigned short questionCount;
unsigned short answerCount;
unsigned short authorityCount;
unsigned