From d041ebacba1a263b22b57d603546b2904fac44bf Mon Sep 17 00:00:00 2001 From: kedixa <1204837541@qq.com> Date: Fri, 3 Dec 2021 18:59:13 +0800 Subject: [PATCH] fix dns message bug when name has trailing dot --- src/protocol/DnsMessage.h | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/protocol/DnsMessage.h b/src/protocol/DnsMessage.h index 1c1de6eb..3b7c2a4c 100644 --- a/src/protocol/DnsMessage.h +++ b/src/protocol/DnsMessage.h @@ -235,9 +235,11 @@ public: this->request_id = id; } - void set_request_name(const std::string& name) + void set_request_name(std::string name) { - this->request_name = name; + while (name.size() > 1 && name.back() == '.') + name.pop_back(); + this->request_name = std::move(name); } protected: