mirror of
https://github.com/oxen-io/lokinet
synced 2023-12-14 06:53:00 +01:00
baf8019fe5
This commit refactors functionality from the Router class into separate, dedicated classes. There are a few behavior changes that came as a result of discussion on what the correct behavior should be. In addition, many things Router was previously doing can now be provided callback functions to alert the calling point when the asynchronous action completes, successfully or otherwise.
416 lines
9.9 KiB
C++
416 lines
9.9 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/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 * >(Version::LLARP_NET_ID));
|
|
return defaultID;
|
|
}
|
|
|
|
bool RouterContact::IgnoreBogons = false;
|
|
|
|
#ifdef TESTNET
|
|
// 1 minute for testnet
|
|
llarp_time_t RouterContact::Lifetime = 60 * 1000;
|
|
#else
|
|
/// 1 day for real network
|
|
llarp_time_t RouterContact::Lifetime = 24 * 60 * 60 * 1000;
|
|
#endif
|
|
/// update RCs every 5 minutes
|
|
llarp_time_t RouterContact::UpdateInterval = 5 * 60 * 1000;
|
|
/// an RC inserted long enough ago (30 min) is considered stale and is removed
|
|
llarp_time_t RouterContact::StaleInsertionAge = 30 * 60 * 1000;
|
|
|
|
NetID::NetID(const byte_t *val) : AlignedBuffer< 8 >()
|
|
{
|
|
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.size())
|
|
{
|
|
/* 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 last updated */
|
|
if(!bencode_write_bytestring(buf, "u", 1))
|
|
return false;
|
|
if(!bencode_write_uint64(buf, last_updated))
|
|
return false;
|
|
|
|
/* write version */
|
|
if(!bencode_write_version_entry(buf))
|
|
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();
|
|
last_updated = 0;
|
|
}
|
|
|
|
util::StatusObject
|
|
RouterContact::ExtractStatus() const
|
|
{
|
|
util::StatusObject obj{{"lastUpdated", last_updated},
|
|
{"exit", IsExit()},
|
|
{"publicRouter", IsPublicRouter()},
|
|
{"identity", pubkey.ToHex()}};
|
|
if(HasNick())
|
|
obj.Put("nickname", Nick());
|
|
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 == "n")
|
|
{
|
|
llarp_buffer_t strbuf;
|
|
if(!bencode_read_string(buf, &strbuf))
|
|
return false;
|
|
if(strbuf.sz > nickname.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
|
|
{
|
|
return addrs.size() > 0;
|
|
}
|
|
|
|
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
|
|
{
|
|
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 : 0;
|
|
}
|
|
|
|
llarp_time_t
|
|
RouterContact::Age(llarp_time_t now) const
|
|
{
|
|
return now > last_updated ? now - last_updated : 0;
|
|
}
|
|
|
|
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) && !IgnoreBogons)
|
|
{
|
|
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");
|
|
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);
|
|
}
|
|
|
|
RouterContact &
|
|
RouterContact::operator=(const RouterContact &other)
|
|
{
|
|
addrs = other.addrs;
|
|
exits = other.exits;
|
|
signature = other.signature;
|
|
last_updated = other.last_updated;
|
|
enckey = other.enckey;
|
|
pubkey = other.pubkey;
|
|
nickname = other.nickname;
|
|
version = other.version;
|
|
netID = other.netID;
|
|
return *this;
|
|
}
|
|
|
|
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);
|
|
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
|