2019-01-14 21:46:07 +00:00
|
|
|
#include <net/address_info.hpp>
|
|
|
|
|
2018-07-30 04:38:14 +00:00
|
|
|
#ifndef _WIN32
|
2018-05-22 15:54:19 +00:00
|
|
|
#include <arpa/inet.h>
|
2018-07-30 04:38:14 +00:00
|
|
|
#endif
|
2019-01-11 01:42:02 +00:00
|
|
|
#include <net/net.hpp>
|
2019-01-10 19:41:51 +00:00
|
|
|
#include <util/bencode.h>
|
|
|
|
#include <util/mem.h>
|
|
|
|
|
2018-12-12 02:52:51 +00:00
|
|
|
#include <string.h>
|
2018-05-22 15:54:19 +00:00
|
|
|
|
2018-08-31 12:46:54 +00:00
|
|
|
namespace llarp
|
2018-05-22 15:54:19 +00:00
|
|
|
{
|
2018-08-31 12:46:54 +00:00
|
|
|
AddressInfo::~AddressInfo()
|
2018-05-22 15:54:19 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2018-08-31 13:51:24 +00:00
|
|
|
AddressInfo &
|
|
|
|
AddressInfo::operator=(const AddressInfo &other)
|
|
|
|
{
|
|
|
|
rank = other.rank;
|
|
|
|
dialect = other.dialect;
|
|
|
|
pubkey = other.pubkey;
|
|
|
|
memcpy(ip.s6_addr, other.ip.s6_addr, 16);
|
2019-01-17 15:11:17 +00:00
|
|
|
port = other.port;
|
|
|
|
version = other.version;
|
2018-08-31 13:51:24 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2018-09-04 12:41:25 +00:00
|
|
|
bool
|
2018-12-13 16:14:44 +00:00
|
|
|
operator==(const AddressInfo &lhs, const AddressInfo &rhs)
|
2018-09-04 12:41:25 +00:00
|
|
|
{
|
|
|
|
// we don't care about rank
|
2018-12-12 00:26:37 +00:00
|
|
|
return lhs.pubkey == rhs.pubkey && lhs.port == rhs.port
|
|
|
|
&& lhs.dialect == rhs.dialect && lhs.ip == rhs.ip;
|
2018-09-04 12:41:25 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
2018-12-13 16:14:44 +00:00
|
|
|
operator<(const AddressInfo &lhs, const AddressInfo &rhs)
|
2018-09-04 12:41:25 +00:00
|
|
|
{
|
2018-12-12 00:26:37 +00:00
|
|
|
return lhs.rank < rhs.rank || lhs.ip < rhs.ip || lhs.port < rhs.port;
|
2018-09-04 12:41:25 +00:00
|
|
|
}
|
|
|
|
|
2018-08-31 12:46:54 +00:00
|
|
|
bool
|
|
|
|
AddressInfo::DecodeKey(llarp_buffer_t key, llarp_buffer_t *buf)
|
2018-05-22 15:54:19 +00:00
|
|
|
{
|
2018-08-31 12:46:54 +00:00
|
|
|
uint64_t i;
|
|
|
|
char tmp[128] = {0};
|
|
|
|
|
|
|
|
llarp_buffer_t strbuf;
|
|
|
|
|
|
|
|
// rank
|
|
|
|
if(llarp_buffer_eq(key, "c"))
|
|
|
|
{
|
|
|
|
if(!bencode_read_integer(buf, &i))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if(i > 65536 || i <= 0)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
rank = i;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// dialect
|
|
|
|
if(llarp_buffer_eq(key, "d"))
|
|
|
|
{
|
|
|
|
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;
|
|
|
|
dialect = std::string(tmp);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// encryption public key
|
|
|
|
if(llarp_buffer_eq(key, "e"))
|
|
|
|
{
|
2018-09-09 11:23:21 +00:00
|
|
|
return pubkey.BDecode(buf);
|
2018-08-31 12:46:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// ip address
|
|
|
|
if(llarp_buffer_eq(key, "i"))
|
|
|
|
{
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
// port
|
|
|
|
if(llarp_buffer_eq(key, "p"))
|
|
|
|
{
|
|
|
|
if(!bencode_read_integer(buf, &i))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if(i > 65536 || i <= 0)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
port = i;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// version
|
|
|
|
if(llarp_buffer_eq(key, "v"))
|
|
|
|
{
|
|
|
|
if(!bencode_read_integer(buf, &i))
|
|
|
|
return false;
|
|
|
|
return i == LLARP_PROTO_VERSION;
|
|
|
|
}
|
|
|
|
|
|
|
|
// bad key
|
|
|
|
return false;
|
2018-05-22 15:54:19 +00:00
|
|
|
}
|
|
|
|
|
2018-08-31 12:46:54 +00:00
|
|
|
bool
|
|
|
|
AddressInfo::BEncode(llarp_buffer_t *buff) const
|
2018-05-22 15:54:19 +00:00
|
|
|
{
|
2018-08-31 12:46:54 +00:00
|
|
|
char ipbuff[128] = {0};
|
|
|
|
const char *ipstr;
|
|
|
|
if(!bencode_start_dict(buff))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-08-31 12:46:54 +00:00
|
|
|
/* rank */
|
|
|
|
if(!bencode_write_bytestring(buff, "c", 1))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-08-31 12:46:54 +00:00
|
|
|
if(!bencode_write_uint64(buff, rank))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-08-31 12:46:54 +00:00
|
|
|
/* dialect */
|
|
|
|
if(!bencode_write_bytestring(buff, "d", 1))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-08-31 12:46:54 +00:00
|
|
|
if(!bencode_write_bytestring(buff, dialect.c_str(), dialect.size()))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-08-31 12:46:54 +00:00
|
|
|
/* encryption key */
|
|
|
|
if(!bencode_write_bytestring(buff, "e", 1))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2019-01-02 01:04:04 +00:00
|
|
|
if(!bencode_write_bytestring(buff, pubkey.data(), PUBKEYSIZE))
|
2018-08-31 12:46:54 +00:00
|
|
|
return false;
|
|
|
|
/** ip */
|
2018-11-09 14:48:43 +00:00
|
|
|
ipstr = inet_ntop(AF_INET6, (void *)&ip, ipbuff, sizeof(ipbuff));
|
2018-08-31 12:46:54 +00:00
|
|
|
if(!ipstr)
|
|
|
|
return false;
|
|
|
|
if(!bencode_write_bytestring(buff, "i", 1))
|
|
|
|
return false;
|
|
|
|
if(!bencode_write_bytestring(buff, ipstr, strnlen(ipstr, sizeof(ipbuff))))
|
|
|
|
return false;
|
|
|
|
/** port */
|
|
|
|
if(!bencode_write_bytestring(buff, "p", 1))
|
|
|
|
return false;
|
|
|
|
if(!bencode_write_uint64(buff, port))
|
2018-05-22 15:54:19 +00:00
|
|
|
return false;
|
2018-05-27 13:42:55 +00:00
|
|
|
|
2018-08-31 12:46:54 +00:00
|
|
|
/** version */
|
|
|
|
if(!bencode_write_version_entry(buff))
|
|
|
|
return false;
|
|
|
|
/** end */
|
|
|
|
return bencode_end(buff);
|
2018-05-22 15:54:19 +00:00
|
|
|
}
|
2018-08-31 12:46:54 +00:00
|
|
|
} // namespace llarp
|