dnsdist: Add unit tests for the AF_XDP/XSK code

Signed-off-by: Remi Gacogne <remi.gacogne@powerdns.com>

diff --git a/pdns/dnsdistdist/Makefile.am b/pdns/dnsdistdist/Makefile.am
index cef13c50efa518c4d2b8ce1d245f66e233103c9f..72fc709c65b02f9da392d6ef83d0443643576db6 100644 (file)
--- a/pdns/dnsdistdist/Makefile.am
+++ b/pdns/dnsdistdist/Makefile.am
@@ -409,6 +409,7 @@ testrunner_SOURCES = \
        test-dnsdist-ipcrypt2_cc.cc \
        test-dnsdist-lua-ffi.cc \
        test-dnsdist-opentelemetry_cc.cc \
+       test-dnsdist-xsk.cc \
        test-dnsdist_cc.cc \
        test-dnsdistasync.cc \
        test-dnsdistbackend_cc.cc \

diff --git a/pdns/dnsdistdist/meson.build b/pdns/dnsdistdist/meson.build
index 69e42d5b3bf8d3c706b5af98b2bcc2d1087064e7..50936999276577b37990f2a76b97bb824526c999 100644 (file)
--- a/pdns/dnsdistdist/meson.build
+++ b/pdns/dnsdistdist/meson.build
@@ -546,6 +546,7 @@ test_sources += files(
   src_dir / 'test-dnsdistsvc_cc.cc',
   src_dir / 'test-dnsdistserverpool.cc',
   src_dir / 'test-dnsdisttcp_cc.cc',
+  src_dir / 'test-dnsdist-xsk.cc',
   src_dir / 'test-dnsparser_cc.cc',
   src_dir / 'test-iputils_hh.cc',
   src_dir / 'test-luawrapper.cc',

diff --git a/pdns/dnsdistdist/test-dnsdist-xsk.cc b/pdns/dnsdistdist/test-dnsdist-xsk.cc
new file mode 100644 (file)
index 0000000..7fa6230
--- /dev/null
+++ b/pdns/dnsdistdist/test-dnsdist-xsk.cc
@@ -0,0 +1,184 @@
+/*
+ * This file is part of PowerDNS or dnsdist.
+ * Copyright -- PowerDNS.COM B.V. and its contributors
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of version 2 of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * In addition, for the avoidance of any doubt, permission is granted to
+ * link this program with OpenSSL and to (re)distribute the binaries
+ * produced as the result of such linking.
+ *
+ * This program 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+#ifndef BOOST_TEST_DYN_LINK
+#define BOOST_TEST_DYN_LINK
+#endif
+
+#define BOOST_TEST_NO_MAIN
+
+#include <boost/test/unit_test.hpp>
+
+#include "config.h"
+#include "dnsdist-mac-address.hh"
+#include "iputils.hh"
+#include "xsk.hh"
+
+BOOST_AUTO_TEST_SUITE(test_dnsdist_xsk)
+
+#if defined(HAVE_XSK)
+
+BOOST_AUTO_TEST_CASE(test_XskSocket)
+{
+  auto itfs = getListOfNetworkInterfaces();
+  if (itfs.empty()) {
+    /* we won't be able to create a XskSocket without a valid interface name */
+    return;
+  }
+  const auto& itfName = *itfs.begin();
+  const uint32_t queueId = 0;
+  const std::string path = "/tmp/xsk-test";
+
+  {
+    /* not a power of two, should throw */
+    BOOST_CHECK_THROW(XskSocket(1000U, itfName, queueId, path), std::runtime_error);
+  }
+
+  {
+    /* not a valid interface name, should throw */
+    BOOST_CHECK_THROW(XskSocket(1024U, "not-an-interface-name", queueId, path), std::runtime_error);
+  }
+
+  {
+    /* not enough privileges, should throw */
+    const size_t numberOfFrames = 1024U;
+    BOOST_CHECK_THROW(XskSocket(numberOfFrames, itfName, queueId, path), std::runtime_error);
+  }
+}
+
+BOOST_AUTO_TEST_CASE(test_XskPacket)
+{
+  const dnsdist::MacAddress fromMAC{};
+  const dnsdist::MacAddress toMAC{0x01, 0x02, 0x03, 0x04, 0x05, 0x06};
+
+  {
+    /* empty packet with no room! */
+    auto packet = XskPacket(nullptr, 0U, 0U);
+    BOOST_CHECK(!packet.parse(false));
+    BOOST_CHECK(packet.getFrameOffsetFrom(nullptr) == 0);
+  }
+
+  {
+    /* IPv4 */
+    const ComboAddress fromAddr("192.0.2.1:42");
+    const ComboAddress toAddr("192.0.2.2:53");
+
+    /* empty packet but with decent room to grow */
+    PacketBuffer payload(XskSocket::getFrameSize());
+    auto packet = XskPacket(payload.data(), 0U, payload.size());
+    BOOST_CHECK_EQUAL(packet.getFlags(), 0U);
+    BOOST_CHECK(packet.getFrameOffsetFrom(payload.data()) == 0);
+    BOOST_CHECK(!packet.parse(false));
+    BOOST_CHECK_EQUAL(packet.getFrameLen(), 0U);
+    BOOST_CHECK_EQUAL(packet.getFlags(), 0U);
+    packet.setAddr(fromAddr, fromMAC, toAddr, toMAC);
+    BOOST_CHECK_EQUAL(packet.getFlags(), XskPacket::UPDATED);
+    BOOST_CHECK_EQUAL(packet.getFromAddr().toStringWithPort(), fromAddr.toStringWithPort());
+    BOOST_CHECK_EQUAL(packet.getToAddr().toStringWithPort(), toAddr.toStringWithPort());
+    BOOST_CHECK_EQUAL(packet.isIPV6(), fromAddr.isIPv6());
+    packet.setPayload(PacketBuffer());
+    packet.rewrite();
+    BOOST_CHECK_EQUAL(packet.getFlags(), XskPacket::REWRITTEN | XskPacket::UPDATED);
+
+    BOOST_CHECK(packet.parse(false));
+    BOOST_CHECK_EQUAL(packet.isIPV6(), fromAddr.isIPv6());
+    BOOST_CHECK_EQUAL(packet.getFrameLen(), 42U);
+    BOOST_CHECK_EQUAL(packet.getCapacity(), (XskSocket::getFrameSize() - XDP_PACKET_HEADROOM - packet.getFrameLen()));
+    BOOST_CHECK_EQUAL(packet.getDataLen(), 0U);
+    BOOST_CHECK(packet.getPayloadData() == payload.data() + packet.getFrameLen());
+    {
+      auto cloned = packet.clonePacketBuffer();
+      BOOST_CHECK(cloned.empty());
+      auto header = packet.cloneHeaderToPacketBuffer();
+      BOOST_CHECK_EQUAL(header.size(), 42U);
+      auto newPacket = XskPacket(payload.data(), 0U, payload.size());
+      newPacket.setHeader(header);
+      BOOST_CHECK_EQUAL(newPacket.getFrameLen(), 42U);
+    }
+
+    {
+      auto smallPayload = PacketBuffer(packet.getCapacity() - 1U);
+      BOOST_CHECK(packet.setPayload(smallPayload));
+      BOOST_CHECK_EQUAL(packet.getFlags(), XskPacket::REWRITTEN | XskPacket::UPDATED);
+    }
+
+    {
+      auto newPacket = XskPacket(payload.data(), 0U, payload.size());
+      auto bigPayload = PacketBuffer(newPacket.getCapacity() + 1U);
+      BOOST_CHECK_EQUAL(newPacket.getFlags(), 0U);
+      /* try to add a payload that is too big for the frame */
+      BOOST_CHECK(!newPacket.setPayload(bigPayload));
+      BOOST_CHECK_EQUAL(newPacket.getFlags(), 0U);
+    }
+
+    {
+      auto sendTime = packet.getSendTime();
+      BOOST_CHECK_EQUAL(sendTime.tv_sec, 0);
+      BOOST_CHECK_EQUAL(sendTime.tv_nsec, 0);
+      BOOST_CHECK_EQUAL(packet.getFlags() & XskPacket::Flags::DELAY, 0U);
+      /* adding 100 ms */
+      packet.addDelay(100U);
+      auto newSendTime = packet.getSendTime();
+      BOOST_CHECK(sendTime < newSendTime);
+      BOOST_CHECK_EQUAL(packet.getFlags() & XskPacket::Flags::DELAY, XskPacket::Flags::DELAY);
+    }
+  }
+
+  {
+    /* IPv6 */
+    const ComboAddress fromAddr("[2001:db8::1]:42");
+    const ComboAddress toAddr("[2001:db8::2]:53");
+
+    /* empty packet but with decent room to grow */
+    PacketBuffer payload(XskSocket::getFrameSize());
+    auto packet = XskPacket(payload.data(), 0U, payload.size());
+    BOOST_CHECK_EQUAL(packet.getFlags(), 0U);
+    BOOST_CHECK(packet.getFrameOffsetFrom(payload.data()) == 0);
+    BOOST_CHECK(!packet.parse(false));
+    BOOST_CHECK_EQUAL(packet.getFrameLen(), 0U);
+    packet.setAddr(fromAddr, fromMAC, toAddr, toMAC);
+    BOOST_CHECK_EQUAL(packet.getFromAddr().toStringWithPort(), fromAddr.toStringWithPort());
+    BOOST_CHECK_EQUAL(packet.getToAddr().toStringWithPort(), toAddr.toStringWithPort());
+    BOOST_CHECK_EQUAL(packet.isIPV6(), fromAddr.isIPv6());
+    packet.setPayload(PacketBuffer());
+    packet.rewrite();
+
+    BOOST_CHECK(packet.parse(false));
+    BOOST_CHECK_EQUAL(packet.isIPV6(), fromAddr.isIPv6());
+    BOOST_CHECK_EQUAL(packet.getFrameLen(), 62U);
+    BOOST_CHECK_EQUAL(packet.getCapacity(), (XskSocket::getFrameSize() - XDP_PACKET_HEADROOM - packet.getFrameLen()));
+    BOOST_CHECK_EQUAL(packet.getDataLen(), 0U);
+    BOOST_CHECK(packet.getPayloadData() == payload.data() + packet.getFrameLen());
+    {
+      auto cloned = packet.clonePacketBuffer();
+      BOOST_CHECK(cloned.empty());
+      auto header = packet.cloneHeaderToPacketBuffer();
+      BOOST_CHECK_EQUAL(header.size(), 62U);
+      auto newPacket = XskPacket(payload.data(), 0U, payload.size());
+      newPacket.setHeader(header);
+      BOOST_CHECK_EQUAL(newPacket.getFrameLen(), 62U);
+    }
+  }
+}
+
+#endif /* HAVE_XSK */
+
+BOOST_AUTO_TEST_SUITE_END();