mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-03 23:15:52 +00:00
96 lines
2.1 KiB
C++
96 lines
2.1 KiB
C++
#ifndef LLARP_DNS_MESSAGE_HPP
|
|
#define LLARP_DNS_MESSAGE_HPP
|
|
|
|
#include <dns/serialize.hpp>
|
|
#include <dns/rr.hpp>
|
|
#include <dns/question.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace dns
|
|
{
|
|
using MsgID_t = uint16_t;
|
|
using Fields_t = uint16_t;
|
|
using Count_t = uint16_t;
|
|
|
|
struct MessageHeader : public Serialize
|
|
{
|
|
const static size_t Size = 12;
|
|
|
|
MessageHeader() = default;
|
|
|
|
MsgID_t id;
|
|
Fields_t fields;
|
|
Count_t qd_count;
|
|
Count_t an_count;
|
|
Count_t ns_count;
|
|
Count_t ar_count;
|
|
|
|
bool
|
|
Encode(llarp_buffer_t* buf) const override;
|
|
|
|
bool
|
|
Decode(llarp_buffer_t* buf) override;
|
|
|
|
bool
|
|
operator==(const MessageHeader& other) const
|
|
{
|
|
return id == other.id && fields == other.fields
|
|
&& qd_count == other.qd_count && an_count == other.an_count
|
|
&& ns_count == other.ns_count && ar_count == other.ar_count;
|
|
}
|
|
};
|
|
|
|
struct Message : public Serialize
|
|
{
|
|
Message(const MessageHeader& hdr);
|
|
|
|
Message(Message&& other);
|
|
Message(const Message& other);
|
|
|
|
void
|
|
UpdateHeader();
|
|
|
|
void
|
|
AddNXReply(RR_TTL_t ttl = 1);
|
|
|
|
void
|
|
AddMXReply(std::string name, uint16_t priority, RR_TTL_t ttl = 1);
|
|
|
|
void
|
|
AddCNAMEReply(std::string name, RR_TTL_t ttl = 1);
|
|
|
|
void
|
|
AddINReply(llarp::huint32_t addr, RR_TTL_t ttl = 1);
|
|
|
|
void
|
|
AddAReply(std::string name, RR_TTL_t ttl = 1);
|
|
|
|
bool
|
|
Encode(llarp_buffer_t* buf) const override;
|
|
|
|
bool
|
|
Decode(llarp_buffer_t* buf) override;
|
|
|
|
std::ostream&
|
|
print(std::ostream& stream, int level, int spaces) const;
|
|
|
|
MsgID_t hdr_id;
|
|
Fields_t hdr_fields;
|
|
std::vector< Question > questions;
|
|
std::vector< ResourceRecord > answers;
|
|
std::vector< ResourceRecord > authorities;
|
|
std::vector< ResourceRecord > additional;
|
|
};
|
|
|
|
inline std::ostream&
|
|
operator<<(std::ostream& out, const Message& msg)
|
|
{
|
|
msg.print(out, -1, -1);
|
|
return out;
|
|
}
|
|
} // namespace dns
|
|
} // namespace llarp
|
|
|
|
#endif
|