1 files changed, 86 insertions, 0 deletions
diff --git a/dnsdist-protocols.cc b/dnsdist-protocols.cc
new file mode 100644
index 0000000..aee63f2
--- /dev/null
+++ b/dnsdist-protocols.cc
@@ -0,0 +1,86 @@
+/*
+ * 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.
+ */
+
+#include <algorithm>
+#include <stdexcept>
+
+#include "dnsdist-protocols.hh"
+
+namespace dnsdist
+{
+const std::array<std::string, Protocol::s_numberOfProtocols> Protocol::s_names = {
+  "DoUDP",
+  "DoTCP",
+  "DNSCryptUDP",
+  "DNSCryptTCP",
+  "DoT",
+  "DoH"};
+
+const std::array<std::string, Protocol::s_numberOfProtocols> Protocol::s_prettyNames = {
+  "Do53 UDP",
+  "Do53 TCP",
+  "DNSCrypt UDP",
+  "DNSCrypt TCP",
+  "DNS over TLS",
+  "DNS over HTTPS"};
+
+Protocol::Protocol(const std::string& s)
+{
+  const auto& it = std::find(s_names.begin(), s_names.end(), s);
+  if (it == s_names.end()) {
+    throw std::runtime_error("Unknown protocol name: '" + s + "'");
+  }
+
+  auto index = std::distance(s_names.begin(), it);
+  d_protocol = static_cast<Protocol::typeenum>(index);
+}
+
+bool Protocol::operator==(Protocol::typeenum type) const
+{
+  return d_protocol == type;
+}
+
+bool Protocol::operator!=(Protocol::typeenum type) const
+{
+  return d_protocol != type;
+}
+
+const std::string& Protocol::toString() const
+{
+  return s_names.at(static_cast<uint8_t>(d_protocol));
+}
+
+const std::string& Protocol::toPrettyString() const
+{
+  return s_prettyNames.at(static_cast<uint8_t>(d_protocol));
+}
+
+bool Protocol::isUDP() const
+{
+  return d_protocol == DoUDP || d_protocol == DNSCryptUDP;
+}
+
+uint8_t Protocol::toNumber() const
+{
+  return static_cast<uint8_t>(d_protocol);
+}
+}