From f2affb3b3a9117393c2a49e6e0ebce1b8b9ea5ea Mon Sep 17 00:00:00 2001
From: Remi Gacogne <remi.gacogne@powerdns.com>
Date: Wed, 9 Oct 2019 15:41:50 +0200
Subject: [PATCH] dnsdist: Count the number of concurrent connections for DoH
 as well

---
 pdns/dnsdistdist/doh.cc | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/pdns/dnsdistdist/doh.cc b/pdns/dnsdistdist/doh.cc
index 2f3cd457e..d0ea42dcf 100644
--- a/pdns/dnsdistdist/doh.cc
+++ b/pdns/dnsdistdist/doh.cc
@@ -82,7 +82,15 @@ public:
     }
   }
 
+  void decrementConcurrentConnections()
+  {
+    if (d_cs != nullptr) {
+      --d_cs->tcpCurrentConnections;
+    }
+  }
+
   std::map<int, std::string> d_ocspResponses;
+  ClientState* d_cs{nullptr};
 
 private:
   h2o_accept_ctx_t d_h2o_accept_ctx;
@@ -146,6 +154,7 @@ void handleDOHTimeout(DOHUnit* oldDU)
 static void on_socketclose(void *data)
 {
   DOHAcceptContext* ctx = reinterpret_cast<DOHAcceptContext*>(data);
+  ctx->decrementConcurrentConnections();
   ctx->release();
 }
 
@@ -883,6 +892,7 @@ static void on_accept(h2o_socket_t *listener, const char *err)
   auto accept_ctx = dsc->accept_ctx->get();
   sock->on_close.data = dsc->accept_ctx;
   ++dsc->df->d_httpconnects;
+  ++dsc->cs->tcpCurrentConnections;
   h2o_accept(accept_ctx, sock);
 }
 
@@ -1034,7 +1044,7 @@ static void setupAcceptContext(DOHAcceptContext& ctx, DOHServerConfig& dsc, bool
 
     nativeCtx->ssl_ctx = tlsCtx.release();
   }
-
+  ctx.d_cs = dsc.cs;
   ctx.release();
 }
 
-- 
2.40.0