|
|
|
#include <service/address.hpp>
|
|
|
|
|
|
|
|
#include <algorithm>
|
|
|
|
|
|
|
|
namespace llarp
|
|
|
|
{
|
|
|
|
namespace service
|
|
|
|
{
|
|
|
|
const std::vector< std::string > Address::AllowedTLDs = {".loki", ".snode"};
|
|
|
|
|
|
|
|
bool
|
|
|
|
Address::PermitTLD(const char* tld)
|
|
|
|
{
|
|
|
|
std::string gtld(tld);
|
|
|
|
std::transform(gtld.begin(), gtld.end(), gtld.begin(), ::tolower);
|
|
|
|
for(const auto& allowed : AllowedTLDs)
|
|
|
|
if(allowed == tld)
|
|
|
|
return true;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string
|
|
|
|
Address::ToString(const char* tld) const
|
|
|
|
{
|
|
|
|
if(!PermitTLD(tld))
|
|
|
|
return "";
|
|
|
|
char tmp[(1 + 32) * 2] = {0};
|
|
|
|
std::string str = Base32Encode(*this, tmp);
|
|
|
|
if(subdomain.size())
|
|
|
|
str = subdomain + "." + str;
|
|
|
|
return str + tld;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
Address::FromString(const std::string& str, const char* tld)
|
|
|
|
{
|
|
|
|
if(!PermitTLD(tld))
|
|
|
|
return false;
|
|
|
|
static auto lowercase = [](const std::string s,
|
|
|
|
bool stripDots) -> std::string {
|
|
|
|
std::string ret(s.size(), ' ');
|
|
|
|
std::transform(s.begin(), s.end(), ret.begin(),
|
|
|
|
[stripDots](const char& ch) -> char {
|
|
|
|
if(ch == '.' && stripDots)
|
|
|
|
return 0;
|
|
|
|
return ::tolower(ch);
|
|
|
|
});
|
|
|
|
return ret.substr(0, ret.find_last_of(' '));
|
|
|
|
};
|
|
|
|
const auto pos = str.find_last_of('.');
|
|
|
|
if(pos == std::string::npos)
|
|
|
|
return false;
|
|
|
|
if(str.substr(pos) != tld)
|
|
|
|
return false;
|
|
|
|
auto sub = str.substr(0, pos);
|
|
|
|
// set subdomains if they are there
|
|
|
|
const auto idx = sub.find_last_of('.');
|
|
|
|
if(idx != std::string::npos)
|
|
|
|
{
|
|
|
|
subdomain = lowercase(sub.substr(0, idx), false);
|
|
|
|
sub = sub.substr(idx + 1);
|
|
|
|
}
|
|
|
|
// make sure it's lowercase
|
|
|
|
return Base32Decode(lowercase(sub, true), *this);
|
|
|
|
}
|
|
|
|
} // namespace service
|
|
|
|
} // namespace llarp
|