mirror of https://github.com/oxen-io/lokinet
internal refactor
- created various structs for each record type - changed DNSd to return most of the recieved packet (allowing for multiple answers) - favor vector<byte_t> over c-style buffers - support NS records a little better - fixed output on cname, ptr, txtpull/58/head
Ryan Tharp
6 years ago
parent
373a14b44b
commit
969865c499
@ -0,0 +1,114 @@
|
||||
#ifndef LIBLLARP_DNS_REC_TYPES_HPP
|
||||
#define LIBLLARP_DNS_REC_TYPES_HPP
|
||||
|
||||
#include <vector>
|
||||
#include <llarp/buffer.h> // for byte_t
|
||||
#include <llarp/net.hpp> // for llarp::Addr , llarp::huint32_t
|
||||
|
||||
namespace llarp
|
||||
{
|
||||
namespace dns
|
||||
{
|
||||
struct record
|
||||
{
|
||||
virtual ~record(){};
|
||||
record()
|
||||
{
|
||||
}
|
||||
|
||||
virtual bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
virtual std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_1a : public record
|
||||
{
|
||||
huint32_t ipaddr;
|
||||
|
||||
virtual ~type_1a(){};
|
||||
type_1a();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_2ns : public record
|
||||
{
|
||||
std::string ns;
|
||||
|
||||
virtual ~type_2ns(){};
|
||||
type_2ns();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_5cname : public record
|
||||
{
|
||||
std::string cname;
|
||||
|
||||
virtual ~type_5cname(){};
|
||||
type_5cname();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_12ptr : public record
|
||||
{
|
||||
std::string revname;
|
||||
|
||||
virtual ~type_12ptr(){};
|
||||
type_12ptr();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_15mx : public record
|
||||
{
|
||||
std::string mx;
|
||||
uint16_t priority;
|
||||
|
||||
virtual ~type_15mx(){};
|
||||
type_15mx();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
struct type_16txt : public record
|
||||
{
|
||||
std::string txt;
|
||||
|
||||
virtual ~type_16txt(){};
|
||||
type_16txt();
|
||||
|
||||
bool
|
||||
parse(std::vector< byte_t > bytes);
|
||||
|
||||
std::vector< byte_t >
|
||||
to_bytes();
|
||||
};
|
||||
|
||||
} // namespace dns
|
||||
} // namespace llarp
|
||||
|
||||
#endif
|
||||
@ -0,0 +1,175 @@
|
||||
#include <llarp/dns_rectypes.hpp>
|
||||
#include <llarp/dns.hpp> // for vput16bits()
|
||||
|
||||
namespace llarp
|
||||
{
|
||||
namespace dns
|
||||
{
|
||||
bool
|
||||
record::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
return bytes.size() ? true : false;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
record::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_1a::type_1a() : record()
|
||||
{
|
||||
this->ipaddr.h = 0;
|
||||
}
|
||||
|
||||
bool
|
||||
type_1a::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
if(bytes.size() < 4)
|
||||
{
|
||||
LogWarn("Less than 4 bytes passed in");
|
||||
return false;
|
||||
}
|
||||
// endian problems? no, it should come in, in network order
|
||||
/*
|
||||
LogDebug("Read ", std::to_string(bytes[0]), ".",
|
||||
std::to_string(bytes[1]), ".",
|
||||
std::to_string(bytes[2]), ".",
|
||||
std::to_string(bytes[3]));
|
||||
*/
|
||||
this->ipaddr = ipaddr_ipv4_bits(bytes[3], bytes[2], bytes[1], bytes[0]);
|
||||
// LogDebug("Test ", this->ipaddr);
|
||||
return bytes.size() ? true : false;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_1a::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
vput16bits(retval, 4); // rdLength
|
||||
vput32bits(retval, this->ipaddr.h); // write IP
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_2ns::type_2ns() : record()
|
||||
{
|
||||
this->ns = "";
|
||||
}
|
||||
bool
|
||||
type_2ns::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
// trim last 2 bytes... probably the size
|
||||
this->ns = std::string((char *)bytes.data(), bytes.size() - 2);
|
||||
return true;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_2ns::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
vput16bits(retval, 2 + this->ns.length()); // rdLength
|
||||
vcode_domain(retval, this->ns);
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_5cname::type_5cname() : record()
|
||||
{
|
||||
this->cname = "";
|
||||
}
|
||||
|
||||
bool
|
||||
type_5cname::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
// trim last 2 bytes... probably the size
|
||||
this->cname = std::string((char *)bytes.data(), bytes.size());
|
||||
// LogDebug("type5 parsed ", this->cname);
|
||||
return true;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_5cname::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
vput16bits(retval, 2 + this->cname.length()); // rdLength
|
||||
vcode_domain(retval, this->cname);
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_12ptr::type_12ptr() : record()
|
||||
{
|
||||
this->revname = "";
|
||||
}
|
||||
|
||||
bool
|
||||
type_12ptr::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
// trim last 2 bytes... probably the size
|
||||
this->revname = std::string((char *)bytes.data(), bytes.size() - 2);
|
||||
return bytes.size() ? true : false;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_12ptr::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
// revname has 2 extra bytes at the end we don't want or need
|
||||
vput16bits(retval, 2 + this->revname.length()); // rdLength
|
||||
vcode_domain(retval, this->revname);
|
||||
// vput16bits(retval, this->revname.length()); // rdLength
|
||||
// vcode_domain(retval, this->revname.substr(0, this->revname.size() -
|
||||
// 2));
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_15mx::type_15mx() : record()
|
||||
{
|
||||
this->priority = 99;
|
||||
this->mx = "";
|
||||
}
|
||||
|
||||
bool
|
||||
type_15mx::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
this->mx = std::string((char *)bytes.data(), bytes.size());
|
||||
// LogInfo("parsed ", this->mx);
|
||||
return true;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_15mx::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
vput16bits(retval, 2 + (2 + this->mx.length())); // rdLength
|
||||
vput16bits(retval, this->priority); // priority
|
||||
vcode_domain(retval, this->mx);
|
||||
return retval;
|
||||
};
|
||||
|
||||
type_16txt::type_16txt() : record()
|
||||
{
|
||||
this->txt = "";
|
||||
}
|
||||
|
||||
bool
|
||||
type_16txt::parse(std::vector< byte_t > bytes)
|
||||
{
|
||||
this->txt = std::string((char *)bytes.data(), bytes.size() - 1);
|
||||
return true;
|
||||
};
|
||||
|
||||
std::vector< byte_t >
|
||||
type_16txt::to_bytes()
|
||||
{
|
||||
std::vector< byte_t > retval;
|
||||
vput16bits(retval, 1 + this->txt.length()); // rdLength
|
||||
retval.push_back(this->txt.length()); // length
|
||||
for(auto it : this->txt)
|
||||
{
|
||||
retval.push_back(it);
|
||||
}
|
||||
return retval;
|
||||
};
|
||||
|
||||
} // namespace dns
|
||||
} // namespace llarp
|
||||
Loading…
Reference in New Issue