test-server.cc 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591
  1. /*************************************************************************
  2. *
  3. * Copyright (C) 2018-2025 Ruilin Peng (Nick) <[email protected]>.
  4. *
  5. * smartdns is free software: you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation, either version 3 of the License, or
  8. * (at your option) any later version.
  9. *
  10. * smartdns is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. * GNU General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  17. */
  18. #include "client.h"
  19. #include "smartdns/dns.h"
  20. #include "include/utils.h"
  21. #include "server.h"
  22. #include "gtest/gtest.h"
  23. class Server : public ::testing::Test
  24. {
  25. protected:
  26. virtual void SetUp() {}
  27. virtual void TearDown() {}
  28. };
  29. TEST_F(Server, all_unreach)
  30. {
  31. smartdns::MockServer server_upstream;
  32. smartdns::Server server;
  33. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  34. if (request->qtype != DNS_T_A) {
  35. return smartdns::SERVER_REQUEST_SOA;
  36. }
  37. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  38. EXPECT_EQ(request->domain, "e.com");
  39. return smartdns::SERVER_REQUEST_OK;
  40. });
  41. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  42. server.Start(R"""(bind [::]:60053
  43. bind-tcp [::]:60053
  44. server tls://255.255.255.255
  45. server https://255.255.255.255
  46. server tcp://255.255.255.255
  47. )""");
  48. smartdns::Client client;
  49. ASSERT_TRUE(client.Query("a.com", 60053));
  50. std::cout << client.GetResult() << std::endl;
  51. EXPECT_EQ(client.GetStatus(), "SERVFAIL");
  52. EXPECT_EQ(client.GetAnswerNum(), 0);
  53. /* server should not crash */
  54. ASSERT_TRUE(client.Query("a.com +tcp", 60053));
  55. std::cout << client.GetResult() << std::endl;
  56. EXPECT_EQ(client.GetStatus(), "SERVFAIL");
  57. EXPECT_EQ(client.GetAnswerNum(), 0);
  58. }
  59. TEST_F(Server, one_nxdomain)
  60. {
  61. smartdns::MockServer server_upstream;
  62. smartdns::MockServer server_upstream1;
  63. smartdns::Server server;
  64. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  65. if (request->qtype != DNS_T_A) {
  66. return smartdns::SERVER_REQUEST_SOA;
  67. }
  68. usleep(50000);
  69. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  70. return smartdns::SERVER_REQUEST_OK;
  71. });
  72. server_upstream1.Start("udp://0.0.0.0:62053",
  73. [](struct smartdns::ServerRequestContext *request) { return smartdns::SERVER_REQUEST_SOA; });
  74. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  75. server.Start(R"""(bind [::]:60053
  76. bind-tcp [::]:60053
  77. server 127.0.0.1:61053
  78. server 127.0.0.1:62053
  79. )""");
  80. smartdns::Client client;
  81. ASSERT_TRUE(client.Query("a.com", 60053));
  82. std::cout << client.GetResult() << std::endl;
  83. ASSERT_EQ(client.GetAnswerNum(), 1);
  84. EXPECT_EQ(client.GetStatus(), "NOERROR");
  85. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  86. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  87. }
  88. TEST_F(Server, retry_no_result_with_NOERROR)
  89. {
  90. smartdns::MockServer server_upstream;
  91. smartdns::MockServer server_upstream1;
  92. smartdns::Server server;
  93. int count = 0;
  94. server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
  95. if (request->qtype != DNS_T_A) {
  96. return smartdns::SERVER_REQUEST_SOA;
  97. }
  98. if (count++ < 2) {
  99. dns_add_domain(request->response_packet, request->domain.c_str(), request->qtype, request->qclass);
  100. request->response_packet->head.tc = 1;
  101. return smartdns::SERVER_REQUEST_OK;
  102. }
  103. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  104. return smartdns::SERVER_REQUEST_OK;
  105. });
  106. server.Start(R"""(bind [::]:60053
  107. bind-tcp [::]:60053
  108. server 127.0.0.1:61053
  109. dualstack-ip-selection no
  110. )""");
  111. smartdns::Client client;
  112. ASSERT_TRUE(client.Query("a.com", 60053));
  113. std::cout << client.GetResult() << std::endl;
  114. ASSERT_EQ(client.GetAnswerNum(), 1);
  115. EXPECT_EQ(client.GetStatus(), "NOERROR");
  116. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  117. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  118. }
  119. TEST_F(Server, retry_no_response)
  120. {
  121. smartdns::MockServer server_upstream;
  122. smartdns::MockServer server_upstream1;
  123. smartdns::Server server;
  124. int count = 0;
  125. server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
  126. count++;
  127. return smartdns::SERVER_REQUEST_NO_RESPONSE;
  128. });
  129. server.Start(R"""(bind [::]:60053
  130. bind-tcp [::]:60053
  131. server 127.0.0.1:61053
  132. )""");
  133. smartdns::Client client;
  134. ASSERT_TRUE(client.Query("a.com", 60053));
  135. std::cout << client.GetResult() << std::endl;
  136. ASSERT_EQ(client.GetAnswerNum(), 0);
  137. EXPECT_EQ(client.GetStatus(), "SERVFAIL");
  138. EXPECT_GE(client.GetQueryTime(), 1500);
  139. EXPECT_GE(count, 4);
  140. }
  141. TEST_F(Server, max_queries)
  142. {
  143. smartdns::MockServer server_upstream;
  144. smartdns::MockServer server_upstream1;
  145. smartdns::Server server;
  146. int count = 0;
  147. server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
  148. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  149. sleep(1);
  150. return smartdns::SERVER_REQUEST_OK;
  151. });
  152. server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
  153. server.Start(R"""(bind [::]:60053
  154. bind-tcp [::]:60053
  155. server 127.0.0.1:61053
  156. dualstack-ip-selection no
  157. max-query-limit 2
  158. )""");
  159. std::vector<std::thread> threads;
  160. int success_num = 0;
  161. int refused_num = 0;
  162. for (int i = 0; i < 5; i++) {
  163. auto t = std::thread([&]() {
  164. smartdns::Client client;
  165. ASSERT_TRUE(client.Query("a.com", 60053));
  166. if (client.GetStatus() == "NOERROR") {
  167. success_num++;
  168. EXPECT_EQ(client.GetStatus(), "NOERROR");
  169. ASSERT_EQ(client.GetAnswerNum(), 1);
  170. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  171. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  172. } else if (client.GetStatus() == "REFUSED") {
  173. refused_num++;
  174. } else {
  175. FAIL();
  176. }
  177. });
  178. threads.push_back(std::move(t));
  179. }
  180. for (auto &t : threads) {
  181. t.join();
  182. }
  183. EXPECT_EQ(success_num, 2);
  184. EXPECT_EQ(refused_num, 3);
  185. for (int i = 0; i < 5; i++) {
  186. smartdns::Client client;
  187. ASSERT_TRUE(client.Query("a.com", 60053));
  188. EXPECT_EQ(client.GetStatus(), "NOERROR");
  189. ASSERT_EQ(client.GetAnswerNum(), 1);
  190. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  191. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  192. }
  193. }
  194. TEST_F(Server, interface)
  195. {
  196. smartdns::MockServer server_upstream;
  197. smartdns::Server server;
  198. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  199. if (request->qtype != DNS_T_A) {
  200. return smartdns::SERVER_REQUEST_SOA;
  201. }
  202. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  203. return smartdns::SERVER_REQUEST_OK;
  204. });
  205. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  206. server.Start(R"""(bind [::]:60053
  207. bind-tcp [::]:60053
  208. server 127.0.0.1:61053 -interface lo
  209. )""");
  210. smartdns::Client client;
  211. ASSERT_TRUE(client.Query("a.com", 60053));
  212. std::cout << client.GetResult() << std::endl;
  213. ASSERT_EQ(client.GetAnswerNum(), 1);
  214. EXPECT_EQ(client.GetStatus(), "NOERROR");
  215. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  216. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  217. }
  218. TEST_F(Server, refused)
  219. {
  220. smartdns::MockServer server_upstream;
  221. smartdns::MockServer server_upstream1;
  222. smartdns::Server server;
  223. int count = 0;
  224. server_upstream.Start("udp://0.0.0.0:61053", [&](struct smartdns::ServerRequestContext *request) {
  225. request->response_packet->head.rcode = DNS_RC_REFUSED;
  226. dns_add_domain(request->response_packet, request->domain.c_str(), request->qtype, request->qclass);
  227. return smartdns::SERVER_REQUEST_OK;
  228. });
  229. server.Start(R"""(bind [::]:60053
  230. bind-tcp [::]:60053
  231. server 127.0.0.1:61053
  232. dualstack-ip-selection no
  233. )""");
  234. smartdns::Client client;
  235. ASSERT_TRUE(client.Query("a.com", 60053));
  236. std::cout << client.GetResult() << std::endl;
  237. ASSERT_EQ(client.GetAnswerNum(), 0);
  238. EXPECT_EQ(client.GetStatus(), "REFUSED");
  239. EXPECT_LT(client.GetQueryTime(), 100);
  240. }
  241. TEST_F(Server, fallback)
  242. {
  243. smartdns::MockServer server_upstream;
  244. smartdns::Server server;
  245. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  246. if (request->qtype != DNS_T_A) {
  247. return smartdns::SERVER_REQUEST_SOA;
  248. }
  249. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  250. return smartdns::SERVER_REQUEST_OK;
  251. });
  252. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  253. server.Start(R"""(bind [::]:60053
  254. bind-tcp [::]:60053
  255. server 127.0.0.1:61053 -fallback
  256. server 127.0.0.1:61054
  257. )""");
  258. smartdns::Client client;
  259. ASSERT_TRUE(client.Query("a.com", 60053));
  260. std::cout << client.GetResult() << std::endl;
  261. ASSERT_EQ(client.GetAnswerNum(), 1);
  262. EXPECT_GE(client.GetQueryTime(), 1000);
  263. EXPECT_EQ(client.GetStatus(), "NOERROR");
  264. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  265. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  266. }
  267. TEST_F(Server, fallback_group)
  268. {
  269. smartdns::MockServer server_upstream;
  270. smartdns::Server server;
  271. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  272. if (request->qtype != DNS_T_A) {
  273. return smartdns::SERVER_REQUEST_SOA;
  274. }
  275. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  276. return smartdns::SERVER_REQUEST_OK;
  277. });
  278. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  279. server.Start(R"""(bind [::]:60053
  280. bind-tcp [::]:60053
  281. server 127.0.0.1:61053 -e -group fallback
  282. server 127.0.0.1:61054
  283. )""");
  284. smartdns::Client client;
  285. ASSERT_TRUE(client.Query("a.com", 60053));
  286. std::cout << client.GetResult() << std::endl;
  287. ASSERT_EQ(client.GetAnswerNum(), 1);
  288. EXPECT_GE(client.GetQueryTime(), 1000);
  289. EXPECT_EQ(client.GetStatus(), "NOERROR");
  290. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  291. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  292. }
  293. TEST_F(Server, groups)
  294. {
  295. smartdns::MockServer server_upstream1;
  296. smartdns::MockServer server_upstream2;
  297. smartdns::MockServer server_upstream3;
  298. smartdns::Server server;
  299. server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  300. if (request->qtype != DNS_T_A) {
  301. return smartdns::SERVER_REQUEST_SOA;
  302. }
  303. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  304. return smartdns::SERVER_REQUEST_OK;
  305. });
  306. server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
  307. if (request->qtype != DNS_T_A) {
  308. return smartdns::SERVER_REQUEST_SOA;
  309. }
  310. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.5", 611);
  311. return smartdns::SERVER_REQUEST_OK;
  312. });
  313. server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
  314. if (request->qtype != DNS_T_A) {
  315. return smartdns::SERVER_REQUEST_SOA;
  316. }
  317. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
  318. return smartdns::SERVER_REQUEST_OK;
  319. });
  320. server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
  321. server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
  322. server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
  323. server.Start(R"""(bind [::]:60053
  324. bind-tcp [::]:60053
  325. server 127.0.0.1:61053
  326. server 127.0.0.1:61054 -e -group a -group b
  327. server 127.0.0.1:61055 -e -group c -group d
  328. nameserver /a.com/a
  329. nameserver /b.com/b
  330. nameserver /c.com/c
  331. nameserver /d.com/d
  332. nameserver /e.com/unknown
  333. )""");
  334. smartdns::Client client;
  335. ASSERT_TRUE(client.Query("a.com", 60053));
  336. std::cout << client.GetResult() << std::endl;
  337. ASSERT_EQ(client.GetAnswerNum(), 1);
  338. EXPECT_EQ(client.GetStatus(), "NOERROR");
  339. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  340. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.5");
  341. ASSERT_TRUE(client.Query("b.com", 60053));
  342. ASSERT_EQ(client.GetAnswerNum(), 1);
  343. EXPECT_EQ(client.GetAnswer()[0].GetName(), "b.com");
  344. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.5");
  345. ASSERT_TRUE(client.Query("c.com", 60053));
  346. ASSERT_EQ(client.GetAnswerNum(), 1);
  347. EXPECT_EQ(client.GetAnswer()[0].GetName(), "c.com");
  348. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
  349. ASSERT_TRUE(client.Query("d.com", 60053));
  350. ASSERT_EQ(client.GetAnswerNum(), 1);
  351. EXPECT_EQ(client.GetAnswer()[0].GetName(), "d.com");
  352. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
  353. ASSERT_TRUE(client.Query("e.com", 60053));
  354. ASSERT_EQ(client.GetAnswerNum(), 1);
  355. EXPECT_EQ(client.GetAnswer()[0].GetName(), "e.com");
  356. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  357. ASSERT_TRUE(client.Query("f.com", 60053));
  358. ASSERT_EQ(client.GetAnswerNum(), 1);
  359. EXPECT_EQ(client.GetAnswer()[0].GetName(), "f.com");
  360. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  361. }
  362. TEST_F(Server, repeat_group)
  363. {
  364. smartdns::MockServer server_upstream;
  365. smartdns::Server server;
  366. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  367. static int count = 0;
  368. if (request->qtype != DNS_T_A) {
  369. return smartdns::SERVER_REQUEST_SOA;
  370. }
  371. count++;
  372. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  373. if (count > 1) {
  374. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.5", 611);
  375. }
  376. return smartdns::SERVER_REQUEST_OK;
  377. });
  378. server.MockPing(PING_TYPE_ICMP, "2001::", 128, 10000);
  379. server.Start(R"""(bind [::]:60053
  380. bind-tcp [::]:60053
  381. server 127.0.0.1:61053 -e -group a -group a
  382. nameserver /a.com/a
  383. )""");
  384. smartdns::Client client;
  385. ASSERT_TRUE(client.Query("a.com", 60053));
  386. std::cout << client.GetResult() << std::endl;
  387. ASSERT_EQ(client.GetAnswerNum(), 1);
  388. EXPECT_EQ(client.GetStatus(), "NOERROR");
  389. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  390. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  391. usleep(100000);
  392. ASSERT_TRUE(client.Query("a.com", 60053));
  393. std::cout << client.GetResult() << std::endl;
  394. ASSERT_EQ(client.GetAnswerNum(), 1);
  395. EXPECT_EQ(client.GetStatus(), "NOERROR");
  396. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  397. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  398. }
  399. TEST_F(Server, bad_block_ip)
  400. {
  401. smartdns::MockServer server_upstream1;
  402. smartdns::MockServer server_upstream2;
  403. smartdns::MockServer server_upstream3;
  404. smartdns::Server server;
  405. server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  406. if (request->qtype != DNS_T_A) {
  407. return smartdns::SERVER_REQUEST_SOA;
  408. }
  409. smartdns::MockServer::AddIP(request, request->domain.c_str(), "127.0.0.1", 611);
  410. return smartdns::SERVER_REQUEST_OK;
  411. });
  412. server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
  413. if (request->qtype != DNS_T_A) {
  414. return smartdns::SERVER_REQUEST_SOA;
  415. }
  416. smartdns::MockServer::AddIP(request, request->domain.c_str(), "::", 611);
  417. return smartdns::SERVER_REQUEST_OK;
  418. });
  419. server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
  420. if (request->qtype != DNS_T_A) {
  421. return smartdns::SERVER_REQUEST_SOA;
  422. }
  423. usleep(100000);
  424. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
  425. return smartdns::SERVER_REQUEST_OK;
  426. });
  427. server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
  428. server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
  429. server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
  430. server.Start(R"""(bind [::]:60053
  431. bind-tcp [::]:60053
  432. server 127.0.0.1:61053
  433. server 127.0.0.1:61054
  434. server 127.0.0.1:61055
  435. )""");
  436. smartdns::Client client;
  437. ASSERT_TRUE(client.Query("a.com", 60053));
  438. std::cout << client.GetResult() << std::endl;
  439. ASSERT_EQ(client.GetAnswerNum(), 1);
  440. EXPECT_EQ(client.GetStatus(), "NOERROR");
  441. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  442. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
  443. }
  444. TEST_F(Server, bad_block_ip_no_check_speed)
  445. {
  446. smartdns::MockServer server_upstream1;
  447. smartdns::MockServer server_upstream2;
  448. smartdns::MockServer server_upstream3;
  449. smartdns::Server server;
  450. server_upstream1.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  451. if (request->qtype != DNS_T_A) {
  452. return smartdns::SERVER_REQUEST_SOA;
  453. }
  454. smartdns::MockServer::AddIP(request, request->domain.c_str(), "127.0.0.1", 611);
  455. return smartdns::SERVER_REQUEST_OK;
  456. });
  457. server_upstream2.Start("udp://0.0.0.0:61054", [](struct smartdns::ServerRequestContext *request) {
  458. if (request->qtype != DNS_T_A) {
  459. return smartdns::SERVER_REQUEST_SOA;
  460. }
  461. smartdns::MockServer::AddIP(request, request->domain.c_str(), "::", 611);
  462. return smartdns::SERVER_REQUEST_OK;
  463. });
  464. server_upstream3.Start("udp://0.0.0.0:61055", [](struct smartdns::ServerRequestContext *request) {
  465. if (request->qtype != DNS_T_A) {
  466. return smartdns::SERVER_REQUEST_SOA;
  467. }
  468. usleep(100000);
  469. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.6", 611);
  470. return smartdns::SERVER_REQUEST_OK;
  471. });
  472. server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 10);
  473. server.MockPing(PING_TYPE_ICMP, "1.2.3.5", 128, 10);
  474. server.MockPing(PING_TYPE_ICMP, "1.2.3.6", 128, 10);
  475. server.Start(R"""(bind [::]:60053
  476. bind-tcp [::]:60053
  477. server 127.0.0.1:61053
  478. server 127.0.0.1:61054
  479. server 127.0.0.1:61055
  480. speed-check-mode none
  481. )""");
  482. smartdns::Client client;
  483. ASSERT_TRUE(client.Query("a.com", 60053));
  484. std::cout << client.GetResult() << std::endl;
  485. ASSERT_EQ(client.GetAnswerNum(), 1);
  486. EXPECT_EQ(client.GetStatus(), "NOERROR");
  487. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  488. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.6");
  489. }
  490. TEST_F(Server, case_insensitive)
  491. {
  492. smartdns::MockServer server_upstream;
  493. smartdns::Server server;
  494. server_upstream.Start("udp://0.0.0.0:61053", [](struct smartdns::ServerRequestContext *request) {
  495. if (request->qtype != DNS_T_A) {
  496. return smartdns::SERVER_REQUEST_SOA;
  497. }
  498. usleep(100000);
  499. smartdns::MockServer::AddIP(request, request->domain.c_str(), "1.2.3.4", 611);
  500. return smartdns::SERVER_REQUEST_OK;
  501. });
  502. server.MockPing(PING_TYPE_ICMP, "1.2.3.4", 128, 1000);
  503. server.Start(R"""(bind [::]:60053
  504. bind-tcp [::]:60053
  505. server 127.0.0.1:61053
  506. )""");
  507. smartdns::Client client;
  508. ASSERT_TRUE(client.Query("a.com", 60053));
  509. std::cout << client.GetResult() << std::endl;
  510. ASSERT_EQ(client.GetAnswerNum(), 1);
  511. EXPECT_EQ(client.GetStatus(), "NOERROR");
  512. EXPECT_EQ(client.GetAnswer()[0].GetName(), "a.com");
  513. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  514. ASSERT_TRUE(client.Query("A.cOm", 60053));
  515. std::cout << client.GetResult() << std::endl;
  516. ASSERT_EQ(client.GetAnswerNum(), 1);
  517. EXPECT_EQ(client.GetStatus(), "NOERROR");
  518. EXPECT_LE(client.GetQueryTime(), 5);
  519. EXPECT_EQ(client.GetAnswer()[0].GetName(), "A.cOm");
  520. EXPECT_EQ(client.GetAnswer()[0].GetData(), "1.2.3.4");
  521. }