dnsdist: Fix dnsdist_ffi_ring_entry_get_protocol

diff --git a/pdns/dnsdist-protocols.cc b/pdns/dnsdist-protocols.cc
index 4fc02eeb63b74fedddabbfa8f8e6195239596e93..adafcf759bfd239e25f208b04d796eab79f551a0 100644 (file)
--- a/pdns/dnsdist-protocols.cc
+++ b/pdns/dnsdist-protocols.cc
@@ -74,4 +74,8 @@ const std::string& Protocol::toPrettyString() const
   return s_prettyNames.at(static_cast<uint8_t>(d_protocol));
 }
 
+uint8_t Protocol::toNumber() const
+{
+  return static_cast<uint8_t>(d_protocol);
+}
 }

diff --git a/pdns/dnsdist-protocols.hh b/pdns/dnsdist-protocols.hh
index fbd51683b815096aecf6b350f0eae4412ec6962a..36284c37f04efaae0b4b1c628274da516dd692a1 100644 (file)
--- a/pdns/dnsdist-protocols.hh
+++ b/pdns/dnsdist-protocols.hh
@@ -31,7 +31,7 @@ class Protocol
 public:
   enum typeenum : uint8_t
   {
-    DoUDP,
+    DoUDP = 0,
     DoTCP,
     DNSCryptUDP,
     DNSCryptTCP,
@@ -54,6 +54,7 @@ public:
 
   const std::string& toString() const;
   const std::string& toPrettyString() const;
+  uint8_t toNumber() const;
 
 private:
   typeenum d_protocol;

diff --git a/pdns/dnsdistdist/dnsdist-lua-ffi.cc b/pdns/dnsdistdist/dnsdist-lua-ffi.cc
index 38f01ce9f0c7221c5c96100219e14a879740f367..8100c625167850321a80824413ff0b5bec696bf4 100644 (file)
--- a/pdns/dnsdistdist/dnsdist-lua-ffi.cc
+++ b/pdns/dnsdistdist/dnsdist-lua-ffi.cc
@@ -924,9 +924,7 @@ uint8_t dnsdist_ffi_ring_entry_get_protocol(const dnsdist_ffi_ring_entry_list_t*
     return 0;
   }
 
-  //return static_cast<uint8_t>(list->d_entries.at(idx).protocol);
-  #warning FIXME
-  return 0;
+  return list->d_entries.at(idx).protocol.toNumber();
 }
 
 uint16_t dnsdist_ffi_ring_entry_get_size(const dnsdist_ffi_ring_entry_list_t* list, size_t idx)