mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-13 01:10:24 +00:00
106 lines
2.4 KiB
C++
106 lines
2.4 KiB
C++
#ifndef _WIN32
|
|
#include <arpa/inet.h>
|
|
#endif
|
|
|
|
#include <net/exit_info.hpp>
|
|
#include <util/bencode.h>
|
|
#include <util/bits.hpp>
|
|
#include <util/mem.h>
|
|
|
|
#include <list>
|
|
#include <string.h>
|
|
|
|
namespace llarp
|
|
{
|
|
ExitInfo::~ExitInfo()
|
|
{
|
|
}
|
|
|
|
ExitInfo&
|
|
ExitInfo::operator=(const ExitInfo& other)
|
|
{
|
|
memcpy(address.s6_addr, other.address.s6_addr, 16);
|
|
memcpy(netmask.s6_addr, other.netmask.s6_addr, 16);
|
|
pubkey = other.pubkey;
|
|
version = other.version;
|
|
return *this;
|
|
}
|
|
|
|
bool
|
|
ExitInfo::BEncode(llarp_buffer_t* buf) const
|
|
{
|
|
char tmp[128] = {0};
|
|
if(!bencode_start_dict(buf))
|
|
return false;
|
|
|
|
if(!inet_ntop(AF_INET6, (void*)&address, tmp, sizeof(tmp)))
|
|
return false;
|
|
if(!BEncodeWriteDictString("a", std::string(tmp), buf))
|
|
return false;
|
|
|
|
if(!inet_ntop(AF_INET6, (void*)&netmask, tmp, sizeof(tmp)))
|
|
return false;
|
|
if(!BEncodeWriteDictString("b", std::string(tmp), buf))
|
|
return false;
|
|
|
|
if(!BEncodeWriteDictEntry("k", pubkey, buf))
|
|
return false;
|
|
|
|
if(!BEncodeWriteDictInt("v", version, buf))
|
|
return false;
|
|
|
|
return bencode_end(buf);
|
|
}
|
|
|
|
static bool
|
|
bdecode_ip_string(llarp_buffer_t* buf, in6_addr& ip)
|
|
{
|
|
char tmp[128] = {0};
|
|
llarp_buffer_t strbuf;
|
|
if(!bencode_read_string(buf, &strbuf))
|
|
return false;
|
|
|
|
if(strbuf.sz >= sizeof(tmp))
|
|
return false;
|
|
|
|
memcpy(tmp, strbuf.base, strbuf.sz);
|
|
tmp[strbuf.sz] = 0;
|
|
return inet_pton(AF_INET6, tmp, &ip.s6_addr[0]) == 1;
|
|
}
|
|
|
|
bool
|
|
ExitInfo::DecodeKey(const llarp_buffer_t& k, llarp_buffer_t* buf)
|
|
{
|
|
bool read = false;
|
|
if(!BEncodeMaybeReadDictEntry("k", pubkey, read, k, buf))
|
|
return false;
|
|
if(!BEncodeMaybeReadDictInt("v", version, read, k, buf))
|
|
return false;
|
|
if(llarp_buffer_eq(k, "a"))
|
|
return bdecode_ip_string(buf, address);
|
|
if(llarp_buffer_eq(k, "b"))
|
|
return bdecode_ip_string(buf, netmask);
|
|
return read;
|
|
}
|
|
|
|
std::ostream&
|
|
operator<<(std::ostream& out, const ExitInfo& xi)
|
|
{
|
|
char tmp[128] = {0};
|
|
if(inet_ntop(AF_INET6, (void*)&xi.address, tmp, sizeof(tmp)))
|
|
out << std::string(tmp);
|
|
else
|
|
return out;
|
|
out << std::string("/");
|
|
#if defined(ANDROID) || defined(RPI)
|
|
snprintf(tmp, sizeof(tmp), "%zu",
|
|
llarp::bits::count_array_bits(xi.netmask.s6_addr));
|
|
return out << tmp;
|
|
#else
|
|
return out << std::to_string(
|
|
llarp::bits::count_array_bits(xi.netmask.s6_addr));
|
|
#endif
|
|
}
|
|
|
|
} // namespace llarp
|