From 4fc2f55f761d71aae1f145d5aa94ba929cc39676 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 28 Apr 2024 11:34:30 +0200 Subject: Adding upstream version 1.7.3. Signed-off-by: Daniel Baumann --- dnsdist-nghttp2.hh | 75 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 75 insertions(+) create mode 100644 dnsdist-nghttp2.hh (limited to 'dnsdist-nghttp2.hh') diff --git a/dnsdist-nghttp2.hh b/dnsdist-nghttp2.hh new file mode 100644 index 0000000..6e38f28 --- /dev/null +++ b/dnsdist-nghttp2.hh @@ -0,0 +1,75 @@ +/* + * 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. + */ +#pragma once + +#include +#include +#include + +#include "dnsdist-tcp.hh" +#include "stat_t.hh" + +struct CrossProtocolQuery; + +class DoHClientCollection +{ +public: + DoHClientCollection(size_t numberOfThreads); + + uint64_t getThreadsCount() const + { + return d_numberOfThreads; + } + + bool passCrossProtocolQueryToThread(std::unique_ptr&& cpq); + void addThread(); + +private: + struct DoHWorkerThread; + + std::mutex d_mutex; + /* we only alter that vector at configuration time, and then + it is never modified at runtime, so we don't take a lock + after the configuration phase */ + std::vector d_clientThreads; + pdns::stat_t d_pos{0}; + uint64_t d_numberOfThreads{0}; +}; + +extern std::unique_ptr g_dohClientThreads; +extern std::atomic g_dohStatesDumpRequested; +extern std::optional g_outgoingDoHWorkerThreads; + +class TLSCtx; + +bool initDoHWorkers(); +bool setupDoHClientProtocolNegotiation(std::shared_ptr& ctx); + +/* opens a new HTTP/2 connection to the supplied backend (attached to the supplied multiplexer), sends the query, + waits for the response to come back or an error to occur then notifies the sender, closing the connection. */ +bool sendH2Query(const std::shared_ptr& ds, std::unique_ptr& mplexer, std::shared_ptr& sender, InternalQuery&& query, bool healthCheck); +size_t handleH2Timeouts(FDMultiplexer& mplexer, const struct timeval& now); +size_t clearH2Connections(); + +void setDoHDownstreamCleanupInterval(uint16_t max); +void setDoHDownstreamMaxIdleTime(uint16_t max); +void setDoHDownstreamMaxIdleConnectionsPerBackend(size_t max); -- cgit v1.2.3