mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-07 15:20:31 +00:00
101 lines
3.5 KiB
C++
101 lines
3.5 KiB
C++
#include <gtest/gtest.h>
|
|
#include <llarp.h> // for llarp_main_init
|
|
#include <llarp/logic.h> // for threadpool/llarp_logic
|
|
#include "llarp/net.hpp" // for llarp::Addr
|
|
#include "llarp/dnsd.hpp"
|
|
|
|
unsigned int g_length = 0;
|
|
std::string g_result = "";
|
|
|
|
ssize_t
|
|
test_sendto_dns_hook(__attribute__((unused)) void *sock,
|
|
__attribute__((unused)) const struct sockaddr *from,
|
|
const void *buffer, size_t length)
|
|
{
|
|
char *hex_buffer = new char[length * 3 + 1];
|
|
hex_buffer[length * 3] = 0;
|
|
for(unsigned int j = 0; j < length; j++)
|
|
sprintf(&hex_buffer[3 * j], "%02X ", ((const char *)buffer)[j]);
|
|
// printf("Got [%zu] bytes: [%s]\n", length, hex_buffer);
|
|
g_result = hex_buffer;
|
|
g_length = length;
|
|
delete[] hex_buffer;
|
|
return length;
|
|
}
|
|
|
|
struct llarpDNSdTest : public ::testing::Test
|
|
{
|
|
dnsd_question_request test_request;
|
|
|
|
llarpDNSdTest()
|
|
{
|
|
}
|
|
void
|
|
SetUp()
|
|
{
|
|
test_request.id = 0;
|
|
test_request.llarp = true; // we don't care about raw atm
|
|
test_request.from = nullptr;
|
|
test_request.context = nullptr;
|
|
test_request.sendto_hook = &test_sendto_dns_hook;
|
|
test_request.question.name = "loki.network";
|
|
test_request.question.type = 1;
|
|
test_request.question.qClass = 1;
|
|
g_result = ""; // reset test global
|
|
g_length = 0;
|
|
llarp::SetLogLevel(
|
|
llarp::eLogNone); // turn off logging to keep gtest output pretty
|
|
}
|
|
};
|
|
|
|
TEST_F(llarpDNSdTest, TestNxDomain)
|
|
{
|
|
write404_dnss_response(nullptr, &test_request);
|
|
ASSERT_TRUE(g_length == 55);
|
|
std::string expected_output =
|
|
"00 00 FFF03 00 01 00 01 00 00 00 00 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 "
|
|
"6B 00 00 01 00 01 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 6B 00 00 01 00 01 "
|
|
"00 00 00 01 00 01 00 ";
|
|
ASSERT_TRUE(expected_output == g_result);
|
|
}
|
|
|
|
TEST_F(llarpDNSdTest, TestAResponse)
|
|
{
|
|
llarp::huint32_t hostRes;
|
|
llarp::Zero(&hostRes.h, sizeof(uint32_t));
|
|
// sockaddr hostRes;
|
|
// llarp::Zero(&hostRes, sizeof(sockaddr));
|
|
writesend_dnss_response(&hostRes, nullptr, &test_request);
|
|
ASSERT_TRUE(g_length == 58);
|
|
std::string expected_output =
|
|
"00 00 FFF00 00 01 00 01 00 00 00 00 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 "
|
|
"6B 00 00 01 00 01 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 6B 00 00 01 00 01 "
|
|
"00 00 00 01 00 04 00 00 00 00 ";
|
|
ASSERT_TRUE(expected_output == g_result);
|
|
}
|
|
|
|
TEST_F(llarpDNSdTest, TestPTRResponse)
|
|
{
|
|
writesend_dnss_revresponse("loki.network", nullptr, &test_request);
|
|
ASSERT_TRUE(g_length == 68);
|
|
std::string expected_output =
|
|
"00 00 FFF00 00 01 00 01 00 00 00 00 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 "
|
|
"6B 00 00 01 00 01 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 6B 00 00 01 00 01 "
|
|
"00 00 00 01 00 0E 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 6B 00 ";
|
|
ASSERT_TRUE(expected_output == g_result);
|
|
}
|
|
|
|
TEST_F(llarpDNSdTest, TestCname)
|
|
{
|
|
writecname_dnss_response("test.cname", nullptr, &test_request);
|
|
ASSERT_TRUE(g_length == 122);
|
|
std::string expected_output =
|
|
"00 00 FFF00 00 01 00 01 00 01 00 01 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 "
|
|
"6B 00 00 01 00 01 04 6C 6F 6B 69 07 6E 65 74 77 6F 72 6B 00 00 05 00 01 "
|
|
"00 00 00 01 00 0C 04 74 65 73 74 05 63 6E 61 6D 65 00 04 74 65 73 74 05 "
|
|
"63 6E 61 6D 65 00 00 02 00 01 00 00 00 01 00 0A 03 6E 73 31 04 6C 6F 6B "
|
|
"69 00 03 6E 73 31 04 6C 6F 6B 69 00 00 01 00 01 00 00 00 01 00 04 7F 00 "
|
|
"00 01 ";
|
|
ASSERT_TRUE(expected_output == g_result);
|
|
}
|