mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-10-31 09:20:21 +00:00
449 lines
10 KiB
C++
449 lines
10 KiB
C++
#include <router_contact.hpp>
|
|
|
|
#include <constants/version.hpp>
|
|
#include <crypto/crypto.hpp>
|
|
#include <net/net.hpp>
|
|
#include <util/bencode.hpp>
|
|
#include <util/buffer.hpp>
|
|
#include <util/logging/logger.hpp>
|
|
#include <util/mem.hpp>
|
|
#include <util/printer.hpp>
|
|
#include <util/time.hpp>
|
|
|
|
#include <fstream>
|
|
#include <util/fs.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
NetID &
|
|
NetID::DefaultValue()
|
|
{
|
|
static NetID defaultID(
|
|
reinterpret_cast< const byte_t * >(llarp::DEFAULT_NETID));
|
|
return defaultID;
|
|
}
|
|
|
|
bool RouterContact::BlockBogons = true;
|
|
|
|
#ifdef TESTNET
|
|
// 1 minute for testnet
|
|
llarp_time_t RouterContact::Lifetime = 1min;
|
|
#else
|
|
/// 1 day for real network
|
|
llarp_time_t RouterContact::Lifetime = 24h;
|
|
#endif
|
|
/// an RC inserted long enough ago (4 hrs) is considered stale and is removed
|
|
llarp_time_t RouterContact::StaleInsertionAge = 4h;
|
|
/// update RCs shortly before they are about to expire
|
|
llarp_time_t RouterContact::UpdateInterval =
|
|
RouterContact::StaleInsertionAge - 5min;
|
|
|
|
NetID::NetID(const byte_t *val)
|
|
{
|
|
size_t len = strnlen(reinterpret_cast< const char * >(val), size());
|
|
std::copy(val, val + len, begin());
|
|
}
|
|
|
|
NetID::NetID() : NetID(DefaultValue().data())
|
|
{
|
|
}
|
|
|
|
bool
|
|
NetID::operator==(const NetID &other) const
|
|
{
|
|
return ToString() == other.ToString();
|
|
}
|
|
|
|
std::string
|
|
NetID::ToString() const
|
|
{
|
|
auto term = std::find(begin(), end(), '\0');
|
|
return std::string(begin(), term);
|
|
}
|
|
|
|
bool
|
|
NetID::BDecode(llarp_buffer_t *buf)
|
|
{
|
|
Zero();
|
|
llarp_buffer_t strbuf;
|
|
if(!bencode_read_string(buf, &strbuf))
|
|
return false;
|
|
|
|
if(strbuf.sz > size())
|
|
return false;
|
|
|
|
std::copy(strbuf.base, strbuf.base + strbuf.sz, begin());
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
NetID::BEncode(llarp_buffer_t *buf) const
|
|
{
|
|
auto term = std::find(begin(), end(), '\0');
|
|
return bencode_write_bytestring(buf, data(), std::distance(begin(), term));
|
|
}
|
|
|
|
bool
|
|
RouterContact::BEncode(llarp_buffer_t *buf) const
|
|
{
|
|
/* write dict begin */
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
|
|
/* write ai if they exist */
|
|
if(!bencode_write_bytestring(buf, "a", 1))
|
|
return false;
|
|
if(!BEncodeWriteList(addrs.begin(), addrs.end(), buf))
|
|
return false;
|
|
|
|
/* write netid */
|
|
if(!bencode_write_bytestring(buf, "i", 1))
|
|
return false;
|
|
if(!netID.BEncode(buf))
|
|
return false;
|
|
/* write signing pubkey */
|
|
if(!bencode_write_bytestring(buf, "k", 1))
|
|
return false;
|
|
if(!pubkey.BEncode(buf))
|
|
return false;
|
|
|
|
std::string nick = Nick();
|
|
if(!nick.empty())
|
|
{
|
|
/* write nickname */
|
|
if(!bencode_write_bytestring(buf, "n", 1))
|
|
{
|
|
return false;
|
|
}
|
|
if(!bencode_write_bytestring(buf, nick.c_str(), nick.size()))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/* write encryption pubkey */
|
|
if(!bencode_write_bytestring(buf, "p", 1))
|
|
return false;
|
|
if(!enckey.BEncode(buf))
|
|
return false;
|
|
// write router version if present
|
|
if(routerVersion.has_value())
|
|
{
|
|
if(not BEncodeWriteDictEntry("r", routerVersion.value(), buf))
|
|
return false;
|
|
}
|
|
/* write last updated */
|
|
if(!bencode_write_bytestring(buf, "u", 1))
|
|
return false;
|
|
if(!bencode_write_uint64(buf, last_updated.count()))
|
|
return false;
|
|
|
|
/* write versions */
|
|
if(!bencode_write_uint64_entry(buf, "v", 1, version))
|
|
return false;
|
|
|
|
/* write xi if they exist */
|
|
if(!bencode_write_bytestring(buf, "x", 1))
|
|
return false;
|
|
if(!BEncodeWriteList(exits.begin(), exits.end(), buf))
|
|
return false;
|
|
|
|
/* write signature */
|
|
if(!bencode_write_bytestring(buf, "z", 1))
|
|
return false;
|
|
if(!signature.BEncode(buf))
|
|
return false;
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
void
|
|
RouterContact::Clear()
|
|
{
|
|
addrs.clear();
|
|
exits.clear();
|
|
signature.Zero();
|
|
nickname.Zero();
|
|
enckey.Zero();
|
|
pubkey.Zero();
|
|
routerVersion = nonstd::optional< RouterVersion >{};
|
|
last_updated = 0s;
|
|
}
|
|
|
|
util::StatusObject
|
|
RouterContact::ExtractStatus() const
|
|
{
|
|
util::StatusObject obj{{"lastUpdated", last_updated.count()},
|
|
{"exit", IsExit()},
|
|
{"publicRouter", IsPublicRouter()},
|
|
{"identity", pubkey.ToString()},
|
|
{"addresses", addrs}};
|
|
|
|
if(HasNick())
|
|
{
|
|
obj["nickname"] = Nick();
|
|
}
|
|
if(routerVersion.has_value())
|
|
{
|
|
obj["routerVersion"] = routerVersion->ToString();
|
|
}
|
|
return obj;
|
|
}
|
|
|
|
bool
|
|
RouterContact::DecodeKey(const llarp_buffer_t &key, llarp_buffer_t *buf)
|
|
{
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictList("a", addrs, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictEntry("i", netID, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictEntry("k", pubkey, read, key, buf))
|
|
return false;
|
|
|
|
if(key == "r")
|
|
{
|
|
RouterVersion r;
|
|
if(not r.BDecode(buf))
|
|
return false;
|
|
routerVersion = r;
|
|
return true;
|
|
}
|
|
|
|
if(key == "n")
|
|
{
|
|
llarp_buffer_t strbuf;
|
|
if(!bencode_read_string(buf, &strbuf))
|
|
{
|
|
return false;
|
|
}
|
|
if(strbuf.sz > llarp::AlignedBuffer< (32) >::size())
|
|
{
|
|
return false;
|
|
}
|
|
nickname.Zero();
|
|
std::copy(strbuf.base, strbuf.base + strbuf.sz, nickname.begin());
|
|
return true;
|
|
}
|
|
|
|
if(!BEncodeMaybeReadDictEntry("p", enckey, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictInt("u", last_updated, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictInt("v", version, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictList("x", exits, read, key, buf))
|
|
return false;
|
|
|
|
if(!BEncodeMaybeReadDictEntry("z", signature, read, key, buf))
|
|
return false;
|
|
|
|
return read;
|
|
}
|
|
|
|
bool
|
|
RouterContact::IsPublicRouter() const
|
|
{
|
|
if(not routerVersion.has_value())
|
|
return false;
|
|
return !addrs.empty();
|
|
}
|
|
|
|
bool
|
|
RouterContact::HasNick() const
|
|
{
|
|
return nickname[0] != 0;
|
|
}
|
|
|
|
void
|
|
RouterContact::SetNick(string_view nick)
|
|
{
|
|
nickname.Zero();
|
|
std::copy(nick.begin(),
|
|
nick.begin() + std::min(nick.size(), nickname.size()),
|
|
nickname.begin());
|
|
}
|
|
|
|
bool
|
|
RouterContact::IsExpired(llarp_time_t now) const
|
|
{
|
|
(void)now;
|
|
return false;
|
|
// return Age(now) >= Lifetime;
|
|
}
|
|
|
|
llarp_time_t
|
|
RouterContact::TimeUntilExpires(llarp_time_t now) const
|
|
{
|
|
const auto expiresAt = last_updated + Lifetime;
|
|
return now < expiresAt ? expiresAt - now : 0s;
|
|
}
|
|
|
|
llarp_time_t
|
|
RouterContact::Age(llarp_time_t now) const
|
|
{
|
|
return now > last_updated ? now - last_updated : 0s;
|
|
}
|
|
|
|
bool
|
|
RouterContact::ExpiresSoon(llarp_time_t now, llarp_time_t dlt) const
|
|
{
|
|
return TimeUntilExpires(now) <= dlt;
|
|
}
|
|
|
|
std::string
|
|
RouterContact::Nick() const
|
|
{
|
|
auto term = std::find(nickname.begin(), nickname.end(), '\0');
|
|
return std::string(nickname.begin(), term);
|
|
}
|
|
|
|
bool
|
|
RouterContact::Sign(const SecretKey &secretkey)
|
|
{
|
|
pubkey = llarp::seckey_topublic(secretkey);
|
|
std::array< byte_t, MAX_RC_SIZE > tmp;
|
|
llarp_buffer_t buf(tmp);
|
|
signature.Zero();
|
|
last_updated = time_now_ms();
|
|
if(!BEncode(&buf))
|
|
{
|
|
return false;
|
|
}
|
|
buf.sz = buf.cur - buf.base;
|
|
buf.cur = buf.base;
|
|
return CryptoManager::instance()->sign(signature, secretkey, buf);
|
|
}
|
|
|
|
bool
|
|
RouterContact::Verify(llarp_time_t now, bool allowExpired) const
|
|
{
|
|
if(netID != NetID::DefaultValue())
|
|
{
|
|
llarp::LogError("netid missmatch: '", netID, "' != '",
|
|
NetID::DefaultValue(), "'");
|
|
return false;
|
|
}
|
|
if(IsExpired(now))
|
|
{
|
|
if(!allowExpired)
|
|
{
|
|
llarp::LogError("RC is expired");
|
|
return false;
|
|
}
|
|
llarp::LogWarn("RC is expired");
|
|
}
|
|
for(const auto &a : addrs)
|
|
{
|
|
if(IsBogon(a.ip) && BlockBogons)
|
|
{
|
|
llarp::LogError("invalid address info: ", a);
|
|
return false;
|
|
}
|
|
}
|
|
for(const auto &exit : exits)
|
|
{
|
|
if(IsBogonRange(exit.address, exit.netmask))
|
|
{
|
|
llarp::LogError("bogon exit: ", exit);
|
|
return false;
|
|
}
|
|
}
|
|
if(!VerifySignature())
|
|
{
|
|
llarp::LogError("invalid signature: ", *this);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
RouterContact::VerifySignature() const
|
|
{
|
|
RouterContact copy;
|
|
copy = *this;
|
|
copy.signature.Zero();
|
|
std::array< byte_t, MAX_RC_SIZE > tmp;
|
|
llarp_buffer_t buf(tmp);
|
|
if(!copy.BEncode(&buf))
|
|
{
|
|
llarp::LogError("bencode failed");
|
|
return false;
|
|
}
|
|
buf.sz = buf.cur - buf.base;
|
|
buf.cur = buf.base;
|
|
return CryptoManager::instance()->verify(pubkey, buf, signature);
|
|
}
|
|
|
|
bool
|
|
RouterContact::Write(const char *fname) const
|
|
{
|
|
std::array< byte_t, MAX_RC_SIZE > tmp;
|
|
llarp_buffer_t buf(tmp);
|
|
if(!BEncode(&buf))
|
|
{
|
|
return false;
|
|
}
|
|
buf.sz = buf.cur - buf.base;
|
|
buf.cur = buf.base;
|
|
const fs::path fpath = std::string(fname); /* */
|
|
auto optional_f =
|
|
llarp::util::OpenFileStream< std::ofstream >(fpath, std::ios::binary);
|
|
if(!optional_f)
|
|
{
|
|
return false;
|
|
}
|
|
auto &f = optional_f.value();
|
|
if(!f.is_open())
|
|
{
|
|
return false;
|
|
}
|
|
f.write((char *)buf.base, buf.sz);
|
|
return true;
|
|
}
|
|
|
|
bool
|
|
RouterContact::Read(const char *fname)
|
|
{
|
|
std::array< byte_t, MAX_RC_SIZE > tmp;
|
|
llarp_buffer_t buf(tmp);
|
|
std::ifstream f;
|
|
f.open(fname, std::ios::binary);
|
|
if(!f.is_open())
|
|
{
|
|
llarp::LogError("Failed to open ", fname);
|
|
return false;
|
|
}
|
|
f.seekg(0, std::ios::end);
|
|
auto l = f.tellg();
|
|
if(l > static_cast< std::streamoff >(sizeof tmp))
|
|
{
|
|
return false;
|
|
}
|
|
f.seekg(0, std::ios::beg);
|
|
f.read((char *)tmp.data(), l);
|
|
return BDecode(&buf);
|
|
}
|
|
|
|
std::ostream &
|
|
RouterContact::print(std::ostream &stream, int level, int spaces) const
|
|
{
|
|
Printer printer(stream, level, spaces);
|
|
printer.printAttribute("k", pubkey);
|
|
printer.printAttribute("updated", last_updated.count());
|
|
printer.printAttribute("netid", netID);
|
|
printer.printAttribute("v", version);
|
|
printer.printAttribute("ai", addrs);
|
|
printer.printAttribute("xi", exits);
|
|
printer.printAttribute("e", enckey);
|
|
printer.printAttribute("z", signature);
|
|
|
|
return stream;
|
|
}
|
|
|
|
} // namespace llarp
|