1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#include <boost/uuid/uuid.hpp>
#include "config.h"
#include "gettime.hh"
#include "dnstap.hh"
#ifndef DISABLE_PROTOBUF
#include <protozero/pbf_writer.hpp>
namespace DnstapBaseFields {
enum : protozero::pbf_tag_type { identity = 1, version = 2, extra = 3, message = 14, type = 15 };
}
namespace DnstapMessageTypes {
enum : protozero::pbf_tag_type { message = 1 };
}
namespace DnstapSocketFamilyTypes {
enum : protozero::pbf_tag_type { inet = 1, inet6 = 2 };
}
namespace DnstapMessageFields {
enum : protozero::pbf_tag_type { type = 1, socket_family = 2, socket_protocol = 3, query_address = 4, response_address = 5, query_port = 6, response_port = 7, query_time_sec = 8, query_time_nsec = 9, query_message = 10, query_zone = 11, response_time_sec = 12, response_time_nsec = 13, response_message = 14 };
}
DnstapMessage::DnstapMessage(std::string& buffer, DnstapMessage::MessageType type, const std::string& identity, const ComboAddress* requestor, const ComboAddress* responder, DnstapMessage::ProtocolType protocol, const char* packet, const size_t len, const struct timespec* queryTime, const struct timespec* responseTime, boost::optional<const DNSName&> auth): d_buffer(buffer)
{
protozero::pbf_writer pbf{d_buffer};
pbf.add_bytes(DnstapBaseFields::identity, identity);
pbf.add_bytes(DnstapBaseFields::version, PACKAGE_STRING);
pbf.add_enum(DnstapBaseFields::type, DnstapMessageTypes::message);
protozero::pbf_writer pbf_message{pbf, DnstapBaseFields::message};
pbf_message.add_enum(DnstapMessageFields::type, static_cast<protozero::pbf_tag_type>(type));
pbf_message.add_enum(DnstapMessageFields::socket_protocol, static_cast<protozero::pbf_tag_type>(protocol));
if (requestor != nullptr) {
pbf_message.add_enum(DnstapMessageFields::socket_family, requestor->sin4.sin_family == AF_INET ? DnstapSocketFamilyTypes::inet : DnstapSocketFamilyTypes::inet6);
}
else if (responder != nullptr) {
pbf_message.add_enum(DnstapMessageFields::socket_family, responder->sin4.sin_family == AF_INET ? DnstapSocketFamilyTypes::inet : DnstapSocketFamilyTypes::inet6);
}
if (requestor != nullptr) {
if (requestor->sin4.sin_family == AF_INET) {
pbf_message.add_bytes(DnstapMessageFields::query_address, reinterpret_cast<const char*>(&requestor->sin4.sin_addr.s_addr), sizeof(requestor->sin4.sin_addr.s_addr));
}
else if (requestor->sin4.sin_family == AF_INET6) {
pbf_message.add_bytes(DnstapMessageFields::query_address, reinterpret_cast<const char*>(&requestor->sin6.sin6_addr.s6_addr), sizeof(requestor->sin6.sin6_addr.s6_addr));
}
pbf_message.add_uint32(DnstapMessageFields::query_port, ntohs(requestor->sin4.sin_port));
}
if (responder != nullptr) {
if (responder->sin4.sin_family == AF_INET) {
pbf_message.add_bytes(DnstapMessageFields::response_address, reinterpret_cast<const char*>(&responder->sin4.sin_addr.s_addr), sizeof(responder->sin4.sin_addr.s_addr));
}
else if (responder->sin4.sin_family == AF_INET6) {
pbf_message.add_bytes(DnstapMessageFields::response_address, reinterpret_cast<const char*>(&responder->sin6.sin6_addr.s6_addr), sizeof(responder->sin6.sin6_addr.s6_addr));
}
pbf_message.add_uint32(DnstapMessageFields::response_port, ntohs(responder->sin4.sin_port));
}
if (queryTime != nullptr) {
pbf_message.add_uint64(DnstapMessageFields::query_time_sec, queryTime->tv_sec);
pbf_message.add_fixed32(DnstapMessageFields::query_time_nsec, queryTime->tv_nsec);
}
if (responseTime != nullptr) {
pbf_message.add_uint64(DnstapMessageFields::response_time_sec, responseTime->tv_sec);
pbf_message.add_fixed32(DnstapMessageFields::response_time_nsec, responseTime->tv_nsec);
}
if (packet != nullptr && len >= sizeof(dnsheader)) {
const struct dnsheader* dh = reinterpret_cast<const struct dnsheader*>(packet);
if (!dh->qr) {
pbf_message.add_bytes(DnstapMessageFields::query_message, packet, len);
} else {
pbf_message.add_bytes(DnstapMessageFields::response_message, packet, len);
}
}
if (auth) {
pbf_message.add_bytes(DnstapMessageFields::query_zone, auth->toDNSString());
}
pbf_message.commit();
}
void DnstapMessage::setExtra(const std::string& extra)
{
protozero::pbf_writer pbf{d_buffer};
pbf.add_bytes(DnstapBaseFields::extra, extra);
}
#endif /* DISABLE_PROTOBUF */
|
