| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933 |
- // SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project & 2024 suyu Emulator Project
- // SPDX-License-Identifier: GPL-2.0-or-later
- #include <algorithm>
- #include <cstring>
- #include <limits>
- #include <utility>
- #include <vector>
- #include "common/error.h"
- #ifdef _WIN32
- #include <winsock2.h>
- #include <ws2tcpip.h>
- #elif SUYU_UNIX
- #include <arpa/inet.h>
- #include <errno.h>
- #include <fcntl.h>
- #include <netdb.h>
- #include <netinet/in.h>
- #include <poll.h>
- #include <sys/socket.h>
- #include <unistd.h>
- #else
- #error "Unimplemented platform"
- #endif
- #include "common/assert.h"
- #include "common/common_types.h"
- #include "common/expected.h"
- #include "common/logging/log.h"
- #include "common/settings.h"
- #include "core/internal_network/network.h"
- #include "core/internal_network/network_interface.h"
- #include "core/internal_network/sockets.h"
- #include "network/network.h"
- namespace Network {
- namespace {
- enum class CallType {
- Send,
- Other,
- };
- #ifdef _WIN32
- using socklen_t = int;
- SOCKET interrupt_socket = static_cast<SOCKET>(-1);
- void InterruptSocketOperations() {
- closesocket(interrupt_socket);
- }
- void AcknowledgeInterrupt() {
- interrupt_socket = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
- }
- void Initialize() {
- WSADATA wsa_data;
- (void)WSAStartup(MAKEWORD(2, 2), &wsa_data);
- AcknowledgeInterrupt();
- }
- void Finalize() {
- InterruptSocketOperations();
- WSACleanup();
- }
- SOCKET GetInterruptSocket() {
- return interrupt_socket;
- }
- sockaddr TranslateFromSockAddrIn(SockAddrIn input) {
- sockaddr_in result;
- #if SUYU_UNIX
- result.sin_len = sizeof(result);
- #endif
- switch (static_cast<Domain>(input.family)) {
- case Domain::INET:
- result.sin_family = AF_INET;
- break;
- default:
- UNIMPLEMENTED_MSG("Unhandled sockaddr family={}", input.family);
- result.sin_family = AF_INET;
- break;
- }
- result.sin_port = htons(input.portno);
- auto& ip = result.sin_addr.S_un.S_un_b;
- ip.s_b1 = input.ip[0];
- ip.s_b2 = input.ip[1];
- ip.s_b3 = input.ip[2];
- ip.s_b4 = input.ip[3];
- sockaddr addr;
- std::memcpy(&addr, &result, sizeof(addr));
- return addr;
- }
- LINGER MakeLinger(bool enable, u32 linger_value) {
- ASSERT(linger_value <= std::numeric_limits<u_short>::max());
- LINGER value;
- value.l_onoff = enable ? 1 : 0;
- value.l_linger = static_cast<u_short>(linger_value);
- return value;
- }
- bool EnableNonBlock(SOCKET fd, bool enable) {
- u_long value = enable ? 1 : 0;
- return ioctlsocket(fd, FIONBIO, &value) != SOCKET_ERROR;
- }
- Errno TranslateNativeError(int e, CallType call_type = CallType::Other) {
- switch (e) {
- case 0:
- return Errno::SUCCESS;
- case WSAEBADF:
- return Errno::BADF;
- case WSAEINVAL:
- return Errno::INVAL;
- case WSAEMFILE:
- return Errno::MFILE;
- case WSAENOTCONN:
- return Errno::NOTCONN;
- case WSAEWOULDBLOCK:
- return Errno::AGAIN;
- case WSAECONNREFUSED:
- return Errno::CONNREFUSED;
- case WSAECONNABORTED:
- if (call_type == CallType::Send) {
- // Winsock yields WSAECONNABORTED from `send` in situations where Unix
- // systems, and actual Switches, yield EPIPE.
- return Errno::PIPE;
- } else {
- return Errno::CONNABORTED;
- }
- case WSAECONNRESET:
- return Errno::CONNRESET;
- case WSAEHOSTUNREACH:
- return Errno::HOSTUNREACH;
- case WSAENETDOWN:
- return Errno::NETDOWN;
- case WSAENETUNREACH:
- return Errno::NETUNREACH;
- case WSAEMSGSIZE:
- return Errno::MSGSIZE;
- case WSAETIMEDOUT:
- return Errno::TIMEDOUT;
- case WSAEINPROGRESS:
- return Errno::INPROGRESS;
- default:
- UNIMPLEMENTED_MSG("Unimplemented errno={}", e);
- return Errno::OTHER;
- }
- }
- #elif SUYU_UNIX // ^ _WIN32 v SUYU_UNIX
- using SOCKET = int;
- using WSAPOLLFD = pollfd;
- using ULONG = u64;
- constexpr SOCKET SOCKET_ERROR = -1;
- constexpr int SD_RECEIVE = SHUT_RD;
- constexpr int SD_SEND = SHUT_WR;
- constexpr int SD_BOTH = SHUT_RDWR;
- int interrupt_pipe_fd[2] = {-1, -1};
- void Initialize() {
- if (pipe(interrupt_pipe_fd) != 0) {
- LOG_ERROR(Network, "Failed to create interrupt pipe!");
- }
- int flags = fcntl(interrupt_pipe_fd[0], F_GETFL);
- ASSERT_MSG(fcntl(interrupt_pipe_fd[0], F_SETFL, flags | O_NONBLOCK) == 0,
- "Failed to set nonblocking state for interrupt pipe");
- }
- void Finalize() {
- if (interrupt_pipe_fd[0] >= 0) {
- close(interrupt_pipe_fd[0]);
- }
- if (interrupt_pipe_fd[1] >= 0) {
- close(interrupt_pipe_fd[1]);
- }
- }
- void InterruptSocketOperations() {
- u8 value = 0;
- ASSERT(write(interrupt_pipe_fd[1], &value, sizeof(value)) == 1);
- }
- void AcknowledgeInterrupt() {
- u8 value = 0;
- ssize_t ret = read(interrupt_pipe_fd[0], &value, sizeof(value));
- if (ret != 1 && errno != EAGAIN && errno != EWOULDBLOCK) {
- LOG_ERROR(Network, "Failed to acknowledge interrupt on shutdown");
- }
- }
- SOCKET GetInterruptSocket() {
- return interrupt_pipe_fd[0];
- }
- sockaddr TranslateFromSockAddrIn(SockAddrIn input) {
- sockaddr_in result;
- switch (static_cast<Domain>(input.family)) {
- case Domain::INET:
- result.sin_family = AF_INET;
- break;
- default:
- UNIMPLEMENTED_MSG("Unhandled sockaddr family={}", input.family);
- result.sin_family = AF_INET;
- break;
- }
- result.sin_port = htons(input.portno);
- result.sin_addr.s_addr = input.ip[0] | input.ip[1] << 8 | input.ip[2] << 16 | input.ip[3] << 24;
- sockaddr addr;
- std::memcpy(&addr, &result, sizeof(addr));
- return addr;
- }
- int WSAPoll(WSAPOLLFD* fds, ULONG nfds, int timeout) {
- return poll(fds, static_cast<nfds_t>(nfds), timeout);
- }
- int closesocket(SOCKET fd) {
- return close(fd);
- }
- linger MakeLinger(bool enable, u32 linger_value) {
- linger value;
- value.l_onoff = enable ? 1 : 0;
- value.l_linger = linger_value;
- return value;
- }
- bool EnableNonBlock(int fd, bool enable) {
- int flags = fcntl(fd, F_GETFL);
- if (flags == -1) {
- return false;
- }
- if (enable) {
- flags |= O_NONBLOCK;
- } else {
- flags &= ~O_NONBLOCK;
- }
- return fcntl(fd, F_SETFL, flags) == 0;
- }
- Errno TranslateNativeError(int e, CallType call_type = CallType::Other) {
- switch (e) {
- case 0:
- return Errno::SUCCESS;
- case EBADF:
- return Errno::BADF;
- case EINVAL:
- return Errno::INVAL;
- case EMFILE:
- return Errno::MFILE;
- case EPIPE:
- return Errno::PIPE;
- case ECONNABORTED:
- return Errno::CONNABORTED;
- case ENOTCONN:
- return Errno::NOTCONN;
- case EAGAIN:
- return Errno::AGAIN;
- case ECONNREFUSED:
- return Errno::CONNREFUSED;
- case ECONNRESET:
- return Errno::CONNRESET;
- case EHOSTUNREACH:
- return Errno::HOSTUNREACH;
- case ENETDOWN:
- return Errno::NETDOWN;
- case ENETUNREACH:
- return Errno::NETUNREACH;
- case EMSGSIZE:
- return Errno::MSGSIZE;
- case ETIMEDOUT:
- return Errno::TIMEDOUT;
- case EINPROGRESS:
- return Errno::INPROGRESS;
- default:
- UNIMPLEMENTED_MSG("Unimplemented errno={} ({})", e, strerror(e));
- return Errno::OTHER;
- }
- }
- #endif
- Errno GetAndLogLastError(CallType call_type = CallType::Other) {
- #ifdef _WIN32
- int e = WSAGetLastError();
- #else
- int e = errno;
- #endif
- const Errno err = TranslateNativeError(e, call_type);
- if (err == Errno::AGAIN || err == Errno::TIMEDOUT || err == Errno::INPROGRESS) {
- // These happen during normal operation, so only log them at debug level.
- LOG_DEBUG(Network, "Socket operation error: {}", Common::NativeErrorToString(e));
- return err;
- }
- LOG_ERROR(Network, "Socket operation error: {}", Common::NativeErrorToString(e));
- return err;
- }
- GetAddrInfoError TranslateGetAddrInfoErrorFromNative(int gai_err) {
- switch (gai_err) {
- case 0:
- return GetAddrInfoError::SUCCESS;
- #ifdef EAI_ADDRFAMILY
- case EAI_ADDRFAMILY:
- return GetAddrInfoError::ADDRFAMILY;
- #endif
- case EAI_AGAIN:
- return GetAddrInfoError::AGAIN;
- case EAI_BADFLAGS:
- return GetAddrInfoError::BADFLAGS;
- case EAI_FAIL:
- return GetAddrInfoError::FAIL;
- case EAI_FAMILY:
- return GetAddrInfoError::FAMILY;
- case EAI_MEMORY:
- return GetAddrInfoError::MEMORY;
- case EAI_NONAME:
- return GetAddrInfoError::NONAME;
- case EAI_SERVICE:
- return GetAddrInfoError::SERVICE;
- case EAI_SOCKTYPE:
- return GetAddrInfoError::SOCKTYPE;
- // These codes may not be defined on all systems:
- #ifdef EAI_SYSTEM
- case EAI_SYSTEM:
- return GetAddrInfoError::SYSTEM;
- #endif
- #ifdef EAI_BADHINTS
- case EAI_BADHINTS:
- return GetAddrInfoError::BADHINTS;
- #endif
- #ifdef EAI_PROTOCOL
- case EAI_PROTOCOL:
- return GetAddrInfoError::PROTOCOL;
- #endif
- #ifdef EAI_OVERFLOW
- case EAI_OVERFLOW:
- return GetAddrInfoError::OVERFLOW_;
- #endif
- default:
- #ifdef EAI_NODATA
- // This can't be a case statement because it would create a duplicate
- // case on Windows where EAI_NODATA is an alias for EAI_NONAME.
- if (gai_err == EAI_NODATA) {
- return GetAddrInfoError::NODATA;
- }
- #endif
- return GetAddrInfoError::OTHER;
- }
- }
- Domain TranslateDomainFromNative(int domain) {
- switch (domain) {
- case 0:
- return Domain::Unspecified;
- case AF_INET:
- return Domain::INET;
- default:
- UNIMPLEMENTED_MSG("Unhandled domain={}", domain);
- return Domain::INET;
- }
- }
- int TranslateDomainToNative(Domain domain) {
- switch (domain) {
- case Domain::Unspecified:
- return 0;
- case Domain::INET:
- return AF_INET;
- default:
- UNIMPLEMENTED_MSG("Unimplemented domain={}", domain);
- return 0;
- }
- }
- Type TranslateTypeFromNative(int type) {
- switch (type) {
- case 0:
- return Type::Unspecified;
- case SOCK_STREAM:
- return Type::STREAM;
- case SOCK_DGRAM:
- return Type::DGRAM;
- case SOCK_RAW:
- return Type::RAW;
- case SOCK_SEQPACKET:
- return Type::SEQPACKET;
- default:
- UNIMPLEMENTED_MSG("Unimplemented type={}", type);
- return Type::STREAM;
- }
- }
- int TranslateTypeToNative(Type type) {
- switch (type) {
- case Type::Unspecified:
- return 0;
- case Type::STREAM:
- return SOCK_STREAM;
- case Type::DGRAM:
- return SOCK_DGRAM;
- case Type::RAW:
- return SOCK_RAW;
- default:
- UNIMPLEMENTED_MSG("Unimplemented type={}", type);
- return 0;
- }
- }
- Protocol TranslateProtocolFromNative(int protocol) {
- switch (protocol) {
- case 0:
- return Protocol::Unspecified;
- case IPPROTO_TCP:
- return Protocol::TCP;
- case IPPROTO_UDP:
- return Protocol::UDP;
- default:
- UNIMPLEMENTED_MSG("Unimplemented protocol={}", protocol);
- return Protocol::Unspecified;
- }
- }
- int TranslateProtocolToNative(Protocol protocol) {
- switch (protocol) {
- case Protocol::Unspecified:
- return 0;
- case Protocol::TCP:
- return IPPROTO_TCP;
- case Protocol::UDP:
- return IPPROTO_UDP;
- default:
- UNIMPLEMENTED_MSG("Unimplemented protocol={}", protocol);
- return 0;
- }
- }
- SockAddrIn TranslateToSockAddrIn(sockaddr_in input, size_t input_len) {
- SockAddrIn result;
- result.family = TranslateDomainFromNative(input.sin_family);
- result.portno = ntohs(input.sin_port);
- result.ip = TranslateIPv4(input.sin_addr);
- return result;
- }
- short TranslatePollEvents(PollEvents events) {
- short result = 0;
- const auto translate = [&result, &events](PollEvents guest, short host) {
- if (True(events & guest)) {
- events &= ~guest;
- result |= host;
- }
- };
- translate(PollEvents::In, POLLIN);
- translate(PollEvents::Pri, POLLPRI);
- translate(PollEvents::Out, POLLOUT);
- translate(PollEvents::Err, POLLERR);
- translate(PollEvents::Hup, POLLHUP);
- translate(PollEvents::Nval, POLLNVAL);
- translate(PollEvents::RdNorm, POLLRDNORM);
- translate(PollEvents::RdBand, POLLRDBAND);
- translate(PollEvents::WrBand, POLLWRBAND);
- #ifdef _WIN32
- short allowed_events = POLLRDBAND | POLLRDNORM | POLLWRNORM;
- // Unlike poll on other OSes, WSAPoll will complain if any other flags are set on input.
- if (result & ~allowed_events) {
- LOG_DEBUG(Network,
- "Removing WSAPoll input events 0x{:x} because Windows doesn't support them",
- result & ~allowed_events);
- }
- result &= allowed_events;
- #endif
- UNIMPLEMENTED_IF_MSG((u16)events != 0, "Unhandled guest events=0x{:x}", (u16)events);
- return result;
- }
- PollEvents TranslatePollRevents(short revents) {
- PollEvents result{};
- const auto translate = [&result, &revents](short host, PollEvents guest) {
- if ((revents & host) != 0) {
- revents &= static_cast<short>(~host);
- result |= guest;
- }
- };
- translate(POLLIN, PollEvents::In);
- translate(POLLPRI, PollEvents::Pri);
- translate(POLLOUT, PollEvents::Out);
- translate(POLLERR, PollEvents::Err);
- translate(POLLHUP, PollEvents::Hup);
- translate(POLLNVAL, PollEvents::Nval);
- translate(POLLRDNORM, PollEvents::RdNorm);
- translate(POLLRDBAND, PollEvents::RdBand);
- translate(POLLWRBAND, PollEvents::WrBand);
- UNIMPLEMENTED_IF_MSG(revents != 0, "Unhandled host revents=0x{:x}", revents);
- return result;
- }
- } // Anonymous namespace
- NetworkInstance::NetworkInstance() {
- Initialize();
- }
- NetworkInstance::~NetworkInstance() {
- Finalize();
- }
- void CancelPendingSocketOperations() {
- InterruptSocketOperations();
- }
- void RestartSocketOperations() {
- AcknowledgeInterrupt();
- }
- std::optional<IPv4Address> GetHostIPv4Address() {
- const auto network_interface = Network::GetSelectedNetworkInterface();
- if (!network_interface.has_value()) {
- // Only print the error once to avoid log spam
- static bool print_error = true;
- if (print_error) {
- LOG_ERROR(Network, "GetSelectedNetworkInterface returned no interface");
- print_error = false;
- }
- return {};
- }
- return TranslateIPv4(network_interface->ip_address);
- }
- std::string IPv4AddressToString(IPv4Address ip_addr) {
- std::array<char, INET_ADDRSTRLEN> buf = {};
- ASSERT(inet_ntop(AF_INET, &ip_addr, buf.data(), sizeof(buf)) == buf.data());
- return std::string(buf.data());
- }
- u32 IPv4AddressToInteger(IPv4Address ip_addr) {
- return static_cast<u32>(ip_addr[0]) << 24 | static_cast<u32>(ip_addr[1]) << 16 |
- static_cast<u32>(ip_addr[2]) << 8 | static_cast<u32>(ip_addr[3]);
- }
- Common::Expected<std::vector<AddrInfo>, GetAddrInfoError> GetAddressInfo(
- const std::string& host, const std::optional<std::string>& service) {
- addrinfo hints{};
- hints.ai_family = AF_INET; // Switch only supports IPv4.
- addrinfo* addrinfo;
- s32 gai_err = getaddrinfo(host.c_str(), service.has_value() ? service->c_str() : nullptr,
- &hints, &addrinfo);
- if (gai_err != 0) {
- return Common::Unexpected(TranslateGetAddrInfoErrorFromNative(gai_err));
- }
- std::vector<AddrInfo> ret;
- for (auto* current = addrinfo; current; current = current->ai_next) {
- // We should only get AF_INET results due to the hints value.
- ASSERT_OR_EXECUTE(addrinfo->ai_family == AF_INET &&
- addrinfo->ai_addrlen == sizeof(sockaddr_in),
- continue;);
- AddrInfo& out = ret.emplace_back();
- out.family = TranslateDomainFromNative(current->ai_family);
- out.socket_type = TranslateTypeFromNative(current->ai_socktype);
- out.protocol = TranslateProtocolFromNative(current->ai_protocol);
- out.addr = TranslateToSockAddrIn(*reinterpret_cast<sockaddr_in*>(current->ai_addr),
- current->ai_addrlen);
- if (current->ai_canonname != nullptr) {
- out.canon_name = current->ai_canonname;
- }
- }
- freeaddrinfo(addrinfo);
- return ret;
- }
- std::pair<s32, Errno> Poll(std::vector<PollFD>& pollfds, s32 timeout) {
- const size_t num = pollfds.size();
- std::vector<WSAPOLLFD> host_pollfds(pollfds.size());
- std::transform(pollfds.begin(), pollfds.end(), host_pollfds.begin(), [](PollFD fd) {
- WSAPOLLFD result;
- result.fd = fd.socket->GetFD();
- result.events = TranslatePollEvents(fd.events);
- result.revents = 0;
- return result;
- });
- host_pollfds.push_back(WSAPOLLFD{
- .fd = GetInterruptSocket(),
- .events = POLLIN,
- .revents = 0,
- });
- const int result =
- WSAPoll(host_pollfds.data(), static_cast<ULONG>(host_pollfds.size()), timeout);
- if (result == 0) {
- ASSERT(std::all_of(host_pollfds.begin(), host_pollfds.end(),
- [](WSAPOLLFD fd) { return fd.revents == 0; }));
- return {0, Errno::SUCCESS};
- }
- for (size_t i = 0; i < num; ++i) {
- pollfds[i].revents = TranslatePollRevents(host_pollfds[i].revents);
- }
- if (result > 0) {
- return {result, Errno::SUCCESS};
- }
- ASSERT(result == SOCKET_ERROR);
- return {-1, GetAndLogLastError()};
- }
- Socket::~Socket() {
- if (fd == INVALID_SOCKET) {
- return;
- }
- (void)closesocket(fd);
- fd = INVALID_SOCKET;
- }
- Socket::Socket(Socket&& rhs) noexcept {
- fd = std::exchange(rhs.fd, INVALID_SOCKET);
- }
- template <typename T>
- std::pair<T, Errno> Socket::GetSockOpt(SOCKET fd_so, int option) {
- T value{};
- socklen_t len = sizeof(value);
- const int result = getsockopt(fd_so, SOL_SOCKET, option, reinterpret_cast<char*>(&value), &len);
- if (result != SOCKET_ERROR) {
- ASSERT(len == sizeof(value));
- return {value, Errno::SUCCESS};
- }
- return {value, GetAndLogLastError()};
- }
- template <typename T>
- Errno Socket::SetSockOpt(SOCKET fd_so, int option, T value) {
- const int result =
- setsockopt(fd_so, SOL_SOCKET, option, reinterpret_cast<const char*>(&value), sizeof(value));
- if (result != SOCKET_ERROR) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- Errno Socket::Initialize(Domain domain, Type type, Protocol protocol) {
- fd = socket(TranslateDomainToNative(domain), TranslateTypeToNative(type),
- TranslateProtocolToNative(protocol));
- if (fd != INVALID_SOCKET) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- std::pair<SocketBase::AcceptResult, Errno> Socket::Accept() {
- sockaddr_in addr;
- socklen_t addrlen = sizeof(addr);
- const bool wait_for_accept = !is_non_blocking;
- if (wait_for_accept) {
- std::vector<WSAPOLLFD> host_pollfds{
- WSAPOLLFD{fd, POLLIN, 0},
- WSAPOLLFD{GetInterruptSocket(), POLLIN, 0},
- };
- while (true) {
- const int pollres =
- WSAPoll(host_pollfds.data(), static_cast<ULONG>(host_pollfds.size()), -1);
- if (host_pollfds[1].revents != 0) {
- // Interrupt signaled before a client could be accepted, break
- return {AcceptResult{}, Errno::AGAIN};
- }
- if (pollres > 0) {
- break;
- }
- }
- }
- const SOCKET new_socket = accept(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen);
- if (new_socket == INVALID_SOCKET) {
- return {AcceptResult{}, GetAndLogLastError()};
- }
- AcceptResult result{
- .socket = std::make_unique<Socket>(new_socket),
- .sockaddr_in = TranslateToSockAddrIn(addr, addrlen),
- };
- return {std::move(result), Errno::SUCCESS};
- }
- Errno Socket::Connect(SockAddrIn addr_in) {
- const sockaddr host_addr_in = TranslateFromSockAddrIn(addr_in);
- if (connect(fd, &host_addr_in, sizeof(host_addr_in)) != SOCKET_ERROR) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- std::pair<SockAddrIn, Errno> Socket::GetPeerName() {
- sockaddr_in addr;
- socklen_t addrlen = sizeof(addr);
- if (getpeername(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen) == SOCKET_ERROR) {
- return {SockAddrIn{}, GetAndLogLastError()};
- }
- return {TranslateToSockAddrIn(addr, addrlen), Errno::SUCCESS};
- }
- std::pair<SockAddrIn, Errno> Socket::GetSockName() {
- sockaddr_in addr;
- socklen_t addrlen = sizeof(addr);
- if (getsockname(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen) == SOCKET_ERROR) {
- return {SockAddrIn{}, GetAndLogLastError()};
- }
- return {TranslateToSockAddrIn(addr, addrlen), Errno::SUCCESS};
- }
- Errno Socket::Bind(SockAddrIn addr) {
- const sockaddr addr_in = TranslateFromSockAddrIn(addr);
- if (bind(fd, &addr_in, sizeof(addr_in)) != SOCKET_ERROR) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- Errno Socket::Listen(s32 backlog) {
- if (listen(fd, backlog) != SOCKET_ERROR) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- Errno Socket::Shutdown(ShutdownHow how) {
- int host_how = 0;
- switch (how) {
- case ShutdownHow::RD:
- host_how = SD_RECEIVE;
- break;
- case ShutdownHow::WR:
- host_how = SD_SEND;
- break;
- case ShutdownHow::RDWR:
- host_how = SD_BOTH;
- break;
- default:
- UNIMPLEMENTED_MSG("Unimplemented flag how={}", how);
- return Errno::SUCCESS;
- }
- if (shutdown(fd, host_how) != SOCKET_ERROR) {
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- std::pair<s32, Errno> Socket::Recv(int flags, std::span<u8> message) {
- ASSERT(flags == 0);
- ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
- const auto result =
- recv(fd, reinterpret_cast<char*>(message.data()), static_cast<int>(message.size()), 0);
- if (result != SOCKET_ERROR) {
- return {static_cast<s32>(result), Errno::SUCCESS};
- }
- return {-1, GetAndLogLastError()};
- }
- std::pair<s32, Errno> Socket::RecvFrom(int flags, std::span<u8> message, SockAddrIn* addr) {
- ASSERT(flags == 0);
- ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
- sockaddr_in addr_in{};
- socklen_t addrlen = sizeof(addr_in);
- socklen_t* const p_addrlen = addr ? &addrlen : nullptr;
- sockaddr* const p_addr_in = addr ? reinterpret_cast<sockaddr*>(&addr_in) : nullptr;
- const auto result = recvfrom(fd, reinterpret_cast<char*>(message.data()),
- static_cast<int>(message.size()), 0, p_addr_in, p_addrlen);
- if (result != SOCKET_ERROR) {
- if (addr) {
- *addr = TranslateToSockAddrIn(addr_in, addrlen);
- }
- return {static_cast<s32>(result), Errno::SUCCESS};
- }
- return {-1, GetAndLogLastError()};
- }
- std::pair<s32, Errno> Socket::Send(std::span<const u8> message, int flags) {
- ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
- ASSERT(flags == 0);
- int native_flags = 0;
- #if SUYU_UNIX
- native_flags |= MSG_NOSIGNAL; // do not send us SIGPIPE
- #endif
- const auto result = send(fd, reinterpret_cast<const char*>(message.data()),
- static_cast<int>(message.size()), native_flags);
- if (result != SOCKET_ERROR) {
- return {static_cast<s32>(result), Errno::SUCCESS};
- }
- return {-1, GetAndLogLastError(CallType::Send)};
- }
- std::pair<s32, Errno> Socket::SendTo(u32 flags, std::span<const u8> message,
- const SockAddrIn* addr) {
- ASSERT(flags == 0);
- const sockaddr* to = nullptr;
- const int to_len = addr ? sizeof(sockaddr) : 0;
- sockaddr host_addr_in;
- if (addr) {
- host_addr_in = TranslateFromSockAddrIn(*addr);
- to = &host_addr_in;
- }
- const auto result = sendto(fd, reinterpret_cast<const char*>(message.data()),
- static_cast<int>(message.size()), 0, to, to_len);
- if (result != SOCKET_ERROR) {
- return {static_cast<s32>(result), Errno::SUCCESS};
- }
- return {-1, GetAndLogLastError(CallType::Send)};
- }
- Errno Socket::Close() {
- [[maybe_unused]] const int result = closesocket(fd);
- ASSERT(result == 0);
- fd = INVALID_SOCKET;
- return Errno::SUCCESS;
- }
- std::pair<Errno, Errno> Socket::GetPendingError() {
- auto [pending_err, getsockopt_err] = GetSockOpt<int>(fd, SO_ERROR);
- return {TranslateNativeError(pending_err), getsockopt_err};
- }
- Errno Socket::SetLinger(bool enable, u32 linger) {
- return SetSockOpt(fd, SO_LINGER, MakeLinger(enable, linger));
- }
- Errno Socket::SetReuseAddr(bool enable) {
- return SetSockOpt<u32>(fd, SO_REUSEADDR, enable ? 1 : 0);
- }
- Errno Socket::SetKeepAlive(bool enable) {
- return SetSockOpt<u32>(fd, SO_KEEPALIVE, enable ? 1 : 0);
- }
- Errno Socket::SetBroadcast(bool enable) {
- return SetSockOpt<u32>(fd, SO_BROADCAST, enable ? 1 : 0);
- }
- Errno Socket::SetSndBuf(u32 value) {
- return SetSockOpt(fd, SO_SNDBUF, value);
- }
- Errno Socket::SetRcvBuf(u32 value) {
- return SetSockOpt(fd, SO_RCVBUF, value);
- }
- Errno Socket::SetSndTimeo(u32 value) {
- return SetSockOpt(fd, SO_SNDTIMEO, value);
- }
- Errno Socket::SetRcvTimeo(u32 value) {
- return SetSockOpt(fd, SO_RCVTIMEO, value);
- }
- Errno Socket::SetNonBlock(bool enable) {
- if (EnableNonBlock(fd, enable)) {
- is_non_blocking = enable;
- return Errno::SUCCESS;
- }
- return GetAndLogLastError();
- }
- bool Socket::IsOpened() const {
- return fd != INVALID_SOCKET;
- }
- void Socket::HandleProxyPacket(const ProxyPacket& packet) {
- LOG_WARNING(Network, "ProxyPacket received, but not in Proxy mode!");
- }
- } // namespace Network
|