C++实现DNS域名解析

  1. 背景
    在调用gethostbyname函数时会遇到阻塞10秒钟问题,为了控制阻塞时间,在调用之前查询DNS信息,类似nslookup命令。只当查询DNS正确返回时,才调用gethostbyname函数,也即自己控制阻塞时间。

  2. 根据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结束

  3. 实现如下
    重点关注组装报文的逻辑

  • 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 
  • 0
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

togolife

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值