lokinet/test/test_dnsd_unit.cpp

#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(void *sock, const struct sockaddr *from,
                             const void *buffer, size_t length) {
  char hex_buffer[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;
  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)
{
  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);
}