diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 09:34:30 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-28 09:34:30 +0000 |
commit | 4fc2f55f761d71aae1f145d5aa94ba929cc39676 (patch) | |
tree | 5c1e1db3b46dd4edbe11f612d93cb94b96891ce3 /dnsdist-nghttp2.hh | |
parent | Initial commit. (diff) | |
download | dnsdist-4fc2f55f761d71aae1f145d5aa94ba929cc39676.tar.xz dnsdist-4fc2f55f761d71aae1f145d5aa94ba929cc39676.zip |
Adding upstream version 1.7.3.upstream/1.7.3upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'dnsdist-nghttp2.hh')
-rw-r--r-- | dnsdist-nghttp2.hh | 75 |
1 files changed, 75 insertions, 0 deletions
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 <memory> +#include <mutex> +#include <vector> + +#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<CrossProtocolQuery>&& 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<DoHWorkerThread> d_clientThreads; + pdns::stat_t d_pos{0}; + uint64_t d_numberOfThreads{0}; +}; + +extern std::unique_ptr<DoHClientCollection> g_dohClientThreads; +extern std::atomic<uint64_t> g_dohStatesDumpRequested; +extern std::optional<uint16_t> g_outgoingDoHWorkerThreads; + +class TLSCtx; + +bool initDoHWorkers(); +bool setupDoHClientProtocolNegotiation(std::shared_ptr<TLSCtx>& 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<DownstreamState>& ds, std::unique_ptr<FDMultiplexer>& mplexer, std::shared_ptr<TCPQuerySender>& 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); |