83 lines
2.4 KiB
C++
83 lines
2.4 KiB
C++
#pragma once
|
|
#include <memory>
|
|
#include "dns-storage.hh"
|
|
#include "dnsmessages.hh"
|
|
#include "comboaddress.hh"
|
|
|
|
struct AGenerator : RRGenerator
|
|
{
|
|
AGenerator(uint32_t ip) : d_ip(ip) {}
|
|
uint32_t d_ip;
|
|
static std::unique_ptr<RRGenerator> make(const ComboAddress&);
|
|
static std::unique_ptr<RRGenerator> make(const std::string& s)
|
|
{
|
|
return make(ComboAddress(s));
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
};
|
|
|
|
struct AAAAGenerator : RRGenerator
|
|
{
|
|
AAAAGenerator(unsigned char ip[16])
|
|
{
|
|
memcpy(d_ip, ip, 16);
|
|
}
|
|
static std::unique_ptr<RRGenerator> make(const ComboAddress&);
|
|
static std::unique_ptr<RRGenerator> make(const std::string& s)
|
|
{
|
|
return make(ComboAddress(s));
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
unsigned char d_ip[16];
|
|
};
|
|
|
|
struct SOAGenerator : RRGenerator
|
|
{
|
|
SOAGenerator(const dnsname& mname, const dnsname& rname, uint32_t serial, uint32_t minimum=3600, uint32_t refresh=10800, uint32_t retry=3600, uint32_t expire=604800) :
|
|
d_mname(mname), d_rname(rname), d_serial(serial), d_minimum(minimum), d_refresh(refresh), d_retry(retry), d_expire(expire)
|
|
{}
|
|
|
|
template<typename ... Targs>
|
|
static std::unique_ptr<RRGenerator> make(const dnsname& mname, const dnsname& rname, Targs&& ... fargs)
|
|
{
|
|
return std::make_unique<SOAGenerator>(mname, rname, std::forward<Targs>(fargs)...);
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
dnsname d_mname, d_rname;
|
|
uint32_t d_serial, d_minimum, d_refresh, d_retry, d_expire;
|
|
};
|
|
|
|
struct NameGenerator : RRGenerator
|
|
{
|
|
NameGenerator(const dnsname& name) : d_name(name) {}
|
|
static std::unique_ptr<RRGenerator> make(const dnsname& mname)
|
|
{
|
|
return std::make_unique<NameGenerator>(mname);
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
dnsname d_name;
|
|
};
|
|
|
|
struct MXGenerator : RRGenerator
|
|
{
|
|
MXGenerator(uint16_t prio, const dnsname& name) : d_prio(prio), d_name(name) {}
|
|
static std::unique_ptr<RRGenerator> make(uint16_t prio, const dnsname& name)
|
|
{
|
|
return std::make_unique<MXGenerator>(prio, name);
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
uint16_t d_prio;
|
|
dnsname d_name;
|
|
};
|
|
|
|
struct TXTGenerator : RRGenerator
|
|
{
|
|
TXTGenerator(const std::string& txt) : d_txt(txt) {}
|
|
static std::unique_ptr<RRGenerator> make(const std::string& txt)
|
|
{
|
|
return std::make_unique<TXTGenerator>(txt);
|
|
}
|
|
void toMessage(DNSMessageWriter& dpw) override;
|
|
std::string d_txt;
|
|
};
|