]>
Commit | Line | Data |
---|---|---|
82a91ddf CH |
1 | /* |
2 | * This file is part of PowerDNS or dnsdist. | |
3 | * Copyright -- PowerDNS.COM B.V. and its contributors | |
4 | * | |
5 | * This program is free software; you can redistribute it and/or modify | |
6 | * it under the terms of version 2 of the GNU General Public License as | |
7 | * published by the Free Software Foundation. | |
8 | * | |
9 | * In addition, for the avoidance of any doubt, permission is granted to | |
10 | * link this program with OpenSSL and to (re)distribute the binaries | |
11 | * produced as the result of such linking. | |
12 | * | |
13 | * This program is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
16 | * GNU General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU General Public License | |
19 | * along with this program; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
21 | */ | |
22 | #pragma once | |
23 | #include "config.h" | |
24 | #include "remote_logger.hh" | |
25 | ||
26 | #ifdef HAVE_FSTRM | |
27 | ||
573f4ff0 | 28 | #include <unordered_map> |
82a91ddf CH |
29 | #include <fstrm.h> |
30 | #include <fstrm/iothr.h> | |
31 | #include <fstrm/unix_writer.h> | |
32 | #ifdef HAVE_FSTRM_TCP_WRITER_INIT | |
33 | #include <fstrm/tcp_writer.h> | |
34 | #endif | |
35 | ||
36 | class FrameStreamLogger : public RemoteLoggerInterface, boost::noncopyable | |
37 | { | |
38 | public: | |
573f4ff0 | 39 | FrameStreamLogger(int family, const std::string& address, bool connect, const std::unordered_map<string,unsigned>& options = std::unordered_map<string,unsigned>()); |
414a03e1 | 40 | ~FrameStreamLogger(); |
4d7db3d7 | 41 | [[nodiscard]] RemoteLoggerInterface::Result queueData(const std::string& data) override; |
9472436f | 42 | |
7c846157 | 43 | [[nodiscard]] std::string address() const override |
9472436f OM |
44 | { |
45 | return d_address; | |
46 | } | |
47 | ||
7c846157 | 48 | [[nodiscard]] std::string name() const override |
4d7db3d7 | 49 | { |
2b7a8236 | 50 | return "dnstap"; |
4d7db3d7 | 51 | } |
7c846157 OM |
52 | |
53 | [[nodiscard]] std::string toString() override | |
82a91ddf | 54 | { |
414a03e1 | 55 | return "FrameStreamLogger to " + d_address + " (" + std::to_string(d_framesSent) + " frames sent, " + std::to_string(d_queueFullDrops) + " dropped, " + std::to_string(d_permanentFailures) + " permanent failures)"; |
82a91ddf | 56 | } |
573f4ff0 | 57 | |
7c846157 | 58 | [[nodiscard]] RemoteLoggerInterface::Stats getStats() override |
2b7a8236 OM |
59 | { |
60 | return Stats{.d_queued = d_framesSent, | |
61 | .d_pipeFull = d_queueFullDrops, | |
b98b553a OM |
62 | .d_tooLarge = 0, |
63 | .d_otherError = d_permanentFailures | |
2b7a8236 OM |
64 | }; |
65 | } | |
66 | ||
82a91ddf | 67 | private: |
573f4ff0 | 68 | |
82a91ddf CH |
69 | const int d_family; |
70 | const std::string d_address; | |
71 | struct fstrm_iothr_queue *d_ioqueue{nullptr}; | |
72 | struct fstrm_writer_options *d_fwopt{nullptr}; | |
73 | struct fstrm_unix_writer_options *d_uwopt{nullptr}; | |
74 | #ifdef HAVE_FSTRM_TCP_WRITER_INIT | |
75 | struct fstrm_tcp_writer_options *d_twopt{nullptr}; | |
76 | #endif | |
77 | struct fstrm_writer *d_writer{nullptr}; | |
78 | struct fstrm_iothr_options *d_iothropt{nullptr}; | |
79 | struct fstrm_iothr *d_iothr{nullptr}; | |
414a03e1 RG |
80 | std::atomic<uint64_t> d_framesSent{0}; |
81 | std::atomic<uint64_t> d_queueFullDrops{0}; | |
82 | std::atomic<uint64_t> d_permanentFailures{0}; | |
82a91ddf CH |
83 | |
84 | void cleanup(); | |
85 | }; | |
86 | ||
10ba6d01 OM |
87 | #else |
88 | class FrameStreamLogger : public RemoteLoggerInterface, boost::noncopyable {}; | |
82a91ddf | 89 | #endif /* HAVE_FSTRM */ |