mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
2.6 KiB
C++
127 lines
2.6 KiB
C++
3 years ago
|
#pragma once
|
||
|
|
||
|
#include <array>
|
||
|
#include <cassert>
|
||
|
#include <cstring>
|
||
|
#include <string>
|
||
|
#include <iosfwd>
|
||
|
|
||
|
#include <ngtcp2/ngtcp2.h>
|
||
|
|
||
|
extern "C"
|
||
|
{
|
||
|
#include <netinet/in.h>
|
||
|
#include <sys/socket.h>
|
||
|
}
|
||
|
|
||
|
// FIXME: replace use of this with a llarp::SockAddr
|
||
|
|
||
|
namespace llarp::quic
|
||
|
{
|
||
|
union sockaddr_any
|
||
|
{
|
||
|
sockaddr_storage storage;
|
||
|
sockaddr sa;
|
||
|
sockaddr_in6 in6;
|
||
|
sockaddr_in in;
|
||
|
};
|
||
|
|
||
|
class Address
|
||
|
{
|
||
|
sockaddr_any s{};
|
||
|
ngtcp2_addr a{0, &s.sa, nullptr};
|
||
|
|
||
|
public:
|
||
|
Address() = default;
|
||
|
Address(std::array<uint8_t, 4> ip, uint16_t port);
|
||
|
Address(const sockaddr_any* addr, size_t addrlen);
|
||
|
Address(const Address& addr)
|
||
|
{
|
||
|
*this = addr;
|
||
|
}
|
||
|
Address&
|
||
|
operator=(const Address& addr);
|
||
|
|
||
|
// Implicit conversion to sockaddr* and ngtcp2_addr& so that an Address can be passed wherever
|
||
|
// one of those is expected.
|
||
|
operator sockaddr*()
|
||
|
{
|
||
|
return a.addr;
|
||
|
}
|
||
|
operator const sockaddr*() const
|
||
|
{
|
||
|
return a.addr;
|
||
|
}
|
||
|
constexpr socklen_t
|
||
|
sockaddr_size() const
|
||
|
{
|
||
|
return a.addrlen;
|
||
|
}
|
||
|
operator ngtcp2_addr&()
|
||
|
{
|
||
|
return a;
|
||
|
}
|
||
|
operator const ngtcp2_addr&() const
|
||
|
{
|
||
|
return a;
|
||
|
}
|
||
|
|
||
|
std::string
|
||
|
to_string() const;
|
||
|
};
|
||
|
|
||
|
// Wraps an ngtcp2_path (which is basically just and address pair) with remote/local components.
|
||
|
// Implicitly convertable to a ngtcp2_path* so that this can be passed wherever a ngtcp2_path* is
|
||
|
// taken in the ngtcp2 API.
|
||
|
struct Path
|
||
|
{
|
||
|
private:
|
||
|
Address local_{}, remote_{};
|
||
|
|
||
|
public:
|
||
|
ngtcp2_path path{
|
||
|
{local_.sockaddr_size(), local_, nullptr}, {remote_.sockaddr_size(), remote_, nullptr}};
|
||
|
|
||
|
// Public accessors are const:
|
||
|
const Address& local = local_;
|
||
|
const Address& remote = remote_;
|
||
|
|
||
|
Path() = default;
|
||
|
Path(const Address& local, const Address& remote) : local_{local}, remote_{remote}
|
||
|
{}
|
||
|
Path(const Address& local, const sockaddr_any* remote_addr, size_t remote_len)
|
||
|
: local_{local}, remote_{remote_addr, remote_len}
|
||
|
{}
|
||
|
Path(const Path& p) : local_{p.local_}, remote_{p.remote_}
|
||
|
{}
|
||
|
|
||
|
Path&
|
||
|
operator=(const Path& p)
|
||
|
{
|
||
|
local_ = p.local_;
|
||
|
remote_ = p.remote_;
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
// Equivalent to `&obj.path`, but slightly more convenient for passing into ngtcp2 functions
|
||
|
// taking a ngtcp2_path pointer.
|
||
|
operator ngtcp2_path*()
|
||
|
{
|
||
|
return &path;
|
||
|
}
|
||
|
operator const ngtcp2_path*() const
|
||
|
{
|
||
|
return &path;
|
||
|
}
|
||
|
|
||
|
std::string
|
||
|
to_string() const;
|
||
|
};
|
||
|
|
||
|
std::ostream&
|
||
|
operator<<(std::ostream& o, const Address& a);
|
||
|
std::ostream&
|
||
|
operator<<(std::ostream& o, const Path& p);
|
||
|
|
||
|
} // namespace llarp::quic
|