mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
7caa87862e
All #ifndef guards on headers have been removed, I think, in favor of #pragma once Headers are now included as `#include "filename"` if the included file resides in the same directory as the file including it, or any subdirectory therein. Otherwise they are included as `#include <project/top/dir/relative/path/filename>` The above does not include system/os headers.
118 lines
2.7 KiB
C++
118 lines
2.7 KiB
C++
#include "question.hpp"
|
|
|
|
#include <llarp/util/logging/logger.hpp>
|
|
#include <llarp/util/printer.hpp>
|
|
#include <llarp/util/str.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace dns
|
|
{
|
|
Question::Question(Question&& other)
|
|
: qname(std::move(other.qname))
|
|
, qtype(std::move(other.qtype))
|
|
, qclass(std::move(other.qclass))
|
|
{}
|
|
Question::Question(const Question& other)
|
|
: qname(other.qname), qtype(other.qtype), qclass(other.qclass)
|
|
{}
|
|
|
|
bool
|
|
Question::Encode(llarp_buffer_t* buf) const
|
|
{
|
|
if (!EncodeName(buf, qname))
|
|
return false;
|
|
if (!buf->put_uint16(qtype))
|
|
return false;
|
|
return buf->put_uint16(qclass);
|
|
}
|
|
|
|
bool
|
|
Question::Decode(llarp_buffer_t* buf)
|
|
{
|
|
if (!DecodeName(buf, qname))
|
|
{
|
|
llarp::LogError("failed to decode name");
|
|
return false;
|
|
}
|
|
if (!buf->read_uint16(qtype))
|
|
{
|
|
llarp::LogError("failed to decode type");
|
|
return false;
|
|
}
|
|
if (!buf->read_uint16(qclass))
|
|
{
|
|
llarp::LogError("failed to decode class");
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
Question::IsName(const std::string& other) const
|
|
{
|
|
// does other have a . at the end?
|
|
if (other.find_last_of('.') == (other.size() - 1))
|
|
return other == qname;
|
|
// no, add it and retry
|
|
return IsName(other + ".");
|
|
}
|
|
|
|
bool
|
|
Question::IsLocalhost() const
|
|
{
|
|
return (qname == "localhost.loki." or llarp::ends_with(qname, ".localhost.loki."));
|
|
}
|
|
|
|
bool
|
|
Question::HasSubdomains() const
|
|
{
|
|
const auto parts = split(qname, ".", true);
|
|
return parts.size() >= 3;
|
|
}
|
|
|
|
std::string
|
|
Question::Subdomains() const
|
|
{
|
|
if (qname.size() < 2)
|
|
return "";
|
|
|
|
size_t pos;
|
|
|
|
pos = qname.rfind('.', qname.size() - 2);
|
|
if (pos == std::string::npos or pos == 0)
|
|
return "";
|
|
|
|
pos = qname.rfind('.', pos - 1);
|
|
if (pos == std::string::npos or pos == 0)
|
|
return "";
|
|
|
|
return qname.substr(0, pos);
|
|
}
|
|
|
|
std::string
|
|
Question::Name() const
|
|
{
|
|
return qname.substr(0, qname.find_last_of('.'));
|
|
}
|
|
|
|
bool
|
|
Question::HasTLD(const std::string& tld) const
|
|
{
|
|
return qname.find(tld) != std::string::npos
|
|
&& qname.rfind(tld) == (qname.size() - tld.size()) - 1;
|
|
}
|
|
|
|
std::ostream&
|
|
Question::print(std::ostream& stream, int level, int spaces) const
|
|
{
|
|
Printer printer(stream, level, spaces);
|
|
printer.printAttribute("qname", qname);
|
|
printer.printAttributeAsHex("qtype", qtype);
|
|
printer.printAttributeAsHex("qclass", qclass);
|
|
|
|
return stream;
|
|
}
|
|
} // namespace dns
|
|
} // namespace llarp
|