| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591 |
- /*************************************************************************
- *
- * Copyright (C) 2018-2025 Ruilin Peng (Nick) <[email protected]>.
- *
- * smartdns is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * smartdns is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
- #include "client.h"
- #include "smartdns/dns.h"
- #include "include/utils.h"
- #include "server.h"
- #include "gtest/gtest.h"
- class Server : public ::testing::Test
- {
- protected:
- virtual void SetUp() {}
- virtual void TearDown() {}
- };
- TEST_F(Server, all_unreach)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- EXPECT_EQ(request->domain, "e.com");
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server tls://255.255.255.255
- server https://255.255.255.255
- server tcp://255.255.255.255
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- EXPECT_EQ(client.GetStatus(), "SERVFAIL");
- EXPECT_EQ(client.GetAnswerNum(), 0);
- /* server should not crash */
- ASSERT_TRUE(client.Query("a.com +tcp", 60053));
- std::cout << client.GetResult() << std::endl;
- EXPECT_EQ(client.GetStatus(), "SERVFAIL");
- EXPECT_EQ(client.GetAnswerNum(), 0);
- }
- TEST_F(Server, one_nxdomain)
- {
- smartdns::MockServer server_upstream;
- smartdns::MockServer server_upstream1;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- usleep(50000);
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream1.Start("udp://0.0.0.0:62053",
- [](struct smartdns::ServerRequestContext *request) { return smartdns::SERVER_REQUEST_SOA; });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- server 127.0.0.1:62053
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, retry_no_result_with_NOERROR)
- {
- smartdns::MockServer server_upstream;
- smartdns::MockServer server_upstream1;
- smartdns::Server server;
- int count = 0;
- server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- if (count++ < 2) {
- dns_add_domain(request->response_packet, request->domain.c_str(), request->qtype, request->qclass);
- request->response_packet->head.tc = 1;
- return smartdns::SERVER_REQUEST_OK;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- dualstack-ip-selection no
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, retry_no_response)
- {
- smartdns::MockServer server_upstream;
- smartdns::MockServer server_upstream1;
- smartdns::Server server;
- int count = 0;
- server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
- count++;
- return smartdns::SERVER_REQUEST_NO_RESPONSE;
- });
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 0);
- EXPECT_EQ(client.GetStatus(), "SERVFAIL");
- EXPECT_GE(client.GetQueryTime(), 1500);
- EXPECT_GE(count, 4);
- }
- TEST_F(Server, max_queries)
- {
- smartdns::MockServer server_upstream;
- smartdns::MockServer server_upstream1;
- smartdns::Server server;
- int count = 0;
- server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- sleep(1);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- dualstack-ip-selection no
- max-query-limit 2
- )""");
- std::vector<std::thread> threads;
- int success_num = 0;
- int refused_num = 0;
- for (int i = 0; i < 5; i++) {
- auto t = std::thread([&]() {
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- if (client.GetStatus() == "NOERROR") {
- success_num++;
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- } else if (client.GetStatus() == "REFUSED") {
- refused_num++;
- } else {
- FAIL();
- }
- });
- threads.push_back(std::move(t));
- }
- for (auto &t : threads) {
- t.join();
- }
- EXPECT_EQ(success_num, 2);
- EXPECT_EQ(refused_num, 3);
- for (int i = 0; i < 5; i++) {
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- }
- TEST_F(Server, interface)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053 -interface lo
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, refused)
- {
- smartdns::MockServer server_upstream;
- smartdns::MockServer server_upstream1;
- smartdns::Server server;
- int count = 0;
- server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
- request->response_packet->head.rcode = DNS_RC_REFUSED;
- dns_add_domain(request->response_packet, request->domain.c_str(), request->qtype, request->qclass);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- dualstack-ip-selection no
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 0);
- EXPECT_EQ(client.GetStatus(), "REFUSED");
- EXPECT_LT(client.GetQueryTime(), 100);
- }
- TEST_F(Server, fallback)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053 -fallback
- server 127.0.0.1:61054
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_GE(client.GetQueryTime(), 1000);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, fallback_group)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053 -e -group fallback
- server 127.0.0.1:61054
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_GE(client.GetQueryTime(), 1000);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, groups)
- {
- smartdns::MockServer server_upstream1;
- smartdns::MockServer server_upstream2;
- smartdns::MockServer server_upstream3;
- smartdns::Server server;
- server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.5", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- server 127.0.0.1:61054 -e -group a -group b
- server 127.0.0.1:61055 -e -group c -group d
- nameserver /a.com/a
- nameserver /b.com/b
- nameserver /c.com/c
- nameserver /d.com/d
- nameserver /e.com/unknown
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.5");
- ASSERT_TRUE(client.Query("b.com", 60053));
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "b.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.5");
- ASSERT_TRUE(client.Query("c.com", 60053));
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "c.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
- ASSERT_TRUE(client.Query("d.com", 60053));
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "d.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
- ASSERT_TRUE(client.Query("e.com", 60053));
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "e.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- ASSERT_TRUE(client.Query("f.com", 60053));
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "f.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, repeat_group)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- static int count = 0;
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- count++;
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- if (count > 1) {
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.5", 611);
- }
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053 -e -group a -group a
- nameserver /a.com/a
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- usleep(100000);
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
- TEST_F(Server, bad_block_ip)
- {
- smartdns::MockServer server_upstream1;
- smartdns::MockServer server_upstream2;
- smartdns::MockServer server_upstream3;
- smartdns::Server server;
- server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "127.0.0.1", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "::", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- usleep(100000);
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- server 127.0.0.1:61054
- server 127.0.0.1:61055
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
- }
- TEST_F(Server, bad_block_ip_no_check_speed)
- {
- smartdns::MockServer server_upstream1;
- smartdns::MockServer server_upstream2;
- smartdns::MockServer server_upstream3;
- smartdns::Server server;
- server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "127.0.0.1", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "::", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- usleep(100000);
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
- server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- server 127.0.0.1:61054
- server 127.0.0.1:61055
- speed-check-mode none
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
- }
- TEST_F(Server, case_insensitive)
- {
- smartdns::MockServer server_upstream;
- smartdns::Server server;
- server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
- if (request->qtype != DNS_T_A) {
- return smartdns::SERVER_REQUEST_SOA;
- }
- usleep(100000);
- smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
- return smartdns::SERVER_REQUEST_OK;
- });
- server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 1000);
- server.Start(R"""(bind [::]:60053
- bind-tcp [::]:60053
- server 127.0.0.1:61053
- )""");
- smartdns::Client client;
- ASSERT_TRUE(client.Query("a.com", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- ASSERT_TRUE(client.Query("A.cOm", 60053));
- std::cout << client.GetResult() << std::endl;
- ASSERT_EQ(client.GetAnswerNum(), 1);
- EXPECT_EQ(client.GetStatus(), "NOERROR");
- EXPECT_LE(client.GetQueryTime(), 5);
- EXPECT_EQ(client.GetAnswer()[0].GetName(), "A.cOm");
- EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
- }
|