107 lines
2.8 KiB
C++
107 lines
2.8 KiB
C++
#pragma once
|
|
#include "precomp.hpp"
|
|
#include <format>
|
|
|
|
namespace Chattr {
|
|
|
|
struct Address {
|
|
Address() {
|
|
zeroFill();
|
|
}
|
|
|
|
Address(int type, gsl::czstring presentationAddr, std::uint16_t port) {
|
|
set(type, presentationAddr, port);
|
|
}
|
|
|
|
void zeroFill() {
|
|
memset(&addr_in6, 0, sizeof(addr_in6));
|
|
}
|
|
|
|
void set(int type, gsl::czstring presentationAddr, std::uint16_t port) {
|
|
zeroFill();
|
|
|
|
if (type == AF_INET) {
|
|
addr_in.sin_family = AF_INET;
|
|
::inet_pton(AF_INET, presentationAddr, &addr_in.sin_addr);
|
|
addr_in.sin_port = htons(port);
|
|
length = sizeof(sockaddr_in);
|
|
} else if (type == AF_INET6) {
|
|
addr_in6.sin6_family = AF_INET6;
|
|
::inet_pton(AF_INET6, presentationAddr, &addr_in6.sin6_addr);
|
|
addr_in6.sin6_port = htons(port);
|
|
length = sizeof(sockaddr_in6);
|
|
}
|
|
}
|
|
|
|
void set(int type, in_addr_t addr, std::uint16_t port) {
|
|
zeroFill();
|
|
|
|
if (type == AF_INET) {
|
|
addr_in.sin_family = AF_INET;
|
|
addr_in.sin_addr.s_addr = htonl(addr);
|
|
addr_in.sin_port = htons(port);
|
|
length = sizeof(sockaddr_in);
|
|
}
|
|
}
|
|
|
|
void set(int type, in_addr addr, std::uint16_t port) {
|
|
zeroFill();
|
|
|
|
if (type == AF_INET) {
|
|
addr_in.sin_family = AF_INET;
|
|
addr_in.sin_addr = addr;
|
|
addr_in.sin_port = htons(port);
|
|
length = sizeof(sockaddr_in);
|
|
}
|
|
}
|
|
|
|
void set(int type, in6_addr addr, std::uint16_t port) {
|
|
zeroFill();
|
|
|
|
if (type == AF_INET6) {
|
|
addr_in6.sin6_family = AF_INET6;
|
|
addr_in6.sin6_addr = addr;
|
|
addr_in6.sin6_port = htons(port);
|
|
length = sizeof(sockaddr_in6);
|
|
}
|
|
}
|
|
|
|
operator std::string() {
|
|
std::optional<std::uint16_t> port = getPort();
|
|
|
|
if (!port)
|
|
return std::string();
|
|
|
|
if (length == sizeof(addr_in)) {
|
|
char addrStr[INET_ADDRSTRLEN];
|
|
::inet_ntop(AF_INET, &addr_in.sin_addr, addrStr, sizeof(addrStr));
|
|
|
|
return std::format("{}:{}", addrStr, port.value());
|
|
} else if (length == sizeof(addr_in6)) {
|
|
char addrStr[INET6_ADDRSTRLEN];
|
|
::inet_ntop(AF_INET6, &addr_in6.sin6_addr, addrStr, sizeof(addrStr));
|
|
|
|
return std::format("{}:{}", addrStr, port.value());
|
|
}
|
|
|
|
return std::string();
|
|
}
|
|
|
|
std::optional<std::uint16_t> getPort() {
|
|
if (length == sizeof(addr_in))
|
|
return ntohs(addr_in.sin_port);
|
|
else if (length == sizeof(addr_in6))
|
|
return ntohs(addr_in6.sin6_port);
|
|
else
|
|
return std::nullopt;
|
|
}
|
|
|
|
union {
|
|
struct sockaddr addr;
|
|
struct sockaddr_in addr_in;
|
|
struct sockaddr_in6 addr_in6;
|
|
};
|
|
socklen_t length;
|
|
};
|
|
|
|
} |