From 2278ffdab2c5a3ff47c56021892926d7b47a811c Mon Sep 17 00:00:00 2001
From: Pieter Lexis <pieter.lexis@powerdns.com>
Date: Thu, 22 Mar 2018 17:53:00 +0100
Subject: [PATCH] Expose rpz stats in the web-interface

---
 pdns/rpzloader.cc   | 10 ----------
 pdns/rpzloader.hh   | 12 ++++++++++++
 pdns/ws-recursor.cc | 30 ++++++++++++++++++++++++++++++
 3 files changed, 42 insertions(+), 10 deletions(-)

diff --git a/pdns/rpzloader.cc b/pdns/rpzloader.cc
index bc437c594..c29d60089 100644
--- a/pdns/rpzloader.cc
+++ b/pdns/rpzloader.cc
@@ -250,16 +250,6 @@ void loadRPZFromFile(const std::string& fname, std::shared_ptr<DNSFilterEngine::
   }
 }
 
-struct rpzStats
-{
-  std::atomic<uint64_t> d_failedTransfers;
-  std::atomic<uint64_t> d_successfulTransfers;
-  std::atomic<uint64_t> d_fullTransfers;
-  std::atomic<uint64_t> d_numberOfRecords;
-  std::atomic<time_t> d_lastUpdate;
-  std::atomic<uint32_t> d_serial;
-};
-
 static std::unordered_map<std::string, rpzStats> s_rpzStats;
 static std::mutex s_rpzStatsMutex;
 
diff --git a/pdns/rpzloader.hh b/pdns/rpzloader.hh
index aec9fd23a..b0d73d296 100644
--- a/pdns/rpzloader.hh
+++ b/pdns/rpzloader.hh
@@ -30,3 +30,15 @@ void loadRPZFromFile(const std::string& fname, std::shared_ptr<DNSFilterEngine::
 std::shared_ptr<SOARecordContent> loadRPZFromServer(const ComboAddress& master, const DNSName& zoneName, std::shared_ptr<DNSFilterEngine::Zone> zone, boost::optional<DNSFilterEngine::Policy> defpol, uint32_t maxTTL, const TSIGTriplet& tt, size_t maxReceivedBytes, const ComboAddress& localAddress, const uint16_t axfrTimeout);
 void RPZRecordToPolicy(const DNSRecord& dr, std::shared_ptr<DNSFilterEngine::Zone> zone, bool addOrRemove, boost::optional<DNSFilterEngine::Policy> defpol, uint32_t maxTTL);
 void RPZIXFRTracker(const ComboAddress& master, boost::optional<DNSFilterEngine::Policy> defpol, uint32_t maxTTL, size_t polZone, const TSIGTriplet &tt, size_t maxReceivedBytes, const ComboAddress& localAddress, std::shared_ptr<DNSFilterEngine::Zone> zone, uint16_t axfrTimeout);
+
+struct rpzStats
+{
+  std::atomic<uint64_t> d_failedTransfers;
+  std::atomic<uint64_t> d_successfulTransfers;
+  std::atomic<uint64_t> d_fullTransfers;
+  std::atomic<uint64_t> d_numberOfRecords;
+  std::atomic<time_t> d_lastUpdate;
+  std::atomic<uint32_t> d_serial;
+};
+
+rpzStats& getRPZZoneStats(const std::string& zone);
diff --git a/pdns/ws-recursor.cc b/pdns/ws-recursor.cc
index 318f1cfb6..a6bfaa6f2 100644
--- a/pdns/ws-recursor.cc
+++ b/pdns/ws-recursor.cc
@@ -39,6 +39,8 @@
 #include "ws-api.hh"
 #include "logger.hh"
 #include "ext/incbin/incbin.h"
+#include "rec-lua-conf.hh"
+#include "rpzloader.hh"
 
 extern thread_local FDMultiplexer* t_fdm;
 
@@ -381,6 +383,34 @@ static void apiServerCacheFlush(HttpRequest* req, HttpResponse* resp) {
   });
 }
 
+static void apiServerRPZ(HttpRequest* req, HttpResponse* resp) {
+  if(req->method != "GET")
+    throw HttpMethodNotAllowedException();
+
+  auto luaconf = g_luaconfs.getLocal();
+  auto numZones = luaconf->dfe.size();
+
+  Json::object ret;
+
+  for (size_t i=0; i < numZones; i++) {
+    auto zone = luaconf->dfe.getZone(i);
+    if (zone == nullptr)
+      continue;
+    auto name = zone->getName();
+    auto& stats = getRPZZoneStats(*name);
+    Json::object zoneInfo = {
+      {"transfers_failed", (double)stats.d_failedTransfers},
+      {"transfers_success", (double)stats.d_successfulTransfers},
+      {"transfers_full", (double)stats.d_fullTransfers},
+      {"records", (double)stats.d_numberOfRecords},
+      {"last_update", (double)stats.d_lastUpdate},
+      {"serial", (double)stats.d_serial},
+    };
+    ret[*name] = zoneInfo;
+  }
+  resp->setBody(ret);
+}
+
 #include "htmlfiles.h"
 
 static void serveStuff(HttpRequest* req, HttpResponse* resp)
-- 
2.40.0