mirrors/Atmosphere
1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere.git synced 2025-06-04 08:29:44 -04:00
Code Issues Projects Releases Packages Wiki Activity

strat: add windows socket api, linux/macos TODO

Browse source
This commit is contained in:
Michael Scire 2022-03-27 14:36:31 -07:00
parent 1bef1b58d4
commit c0d5140ef0
17 changed files with 2258 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

2
libraries/libstratosphere/source/socket/impl/socket_api.hpp
View file

@ -47,6 +47,8 @@ namespace ams::socket::impl {
s32 Accept(s32 desc, SockAddr *out_address, SockLenT *out_addr_len);
s32 Bind(s32 desc, const SockAddr *address, SockLenT len);
s32 Connect(s32 desc, const SockAddr *address, SockLenT len);
s32 GetSockName(s32 desc, SockAddr *out_address, SockLenT *out_addr_len);
s32 SetSockOpt(s32 desc, Level level, Option option_name, const void *option_value, SockLenT option_size);

22
libraries/libstratosphere/source/socket/impl/socket_api.os.horizon.cpp
View file

@ -505,6 +505,28 @@ namespace ams::socket::impl {
return result;
}
s32 Connect(s32 desc, const SockAddr *address, SockLenT len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Check input. */
if (address == nullptr || len == 0) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
/* Perform the call. */
Errno error = Errno::ESuccess;
int result = ::bsdConnect(desc, ConvertForLibnx(address), len);
TranslateResultToBsdError(error, result);
if (result < 0) {
socket::impl::SetLastError(error);
}
return result;
}
s32 GetSockName(s32 desc, SockAddr *out_address, SockLenT *out_addr_len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());

741
libraries/libstratosphere/source/socket/impl/socket_api.os.windows.cpp Normal file
View file

@ -0,0 +1,741 @@
/*
* Copyright (c) Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stratosphere.hpp>
#include "socket_api.hpp"
#include "socket_allocator.hpp"
#include <ws2tcpip.h>
#include <stratosphere/socket/impl/socket_platform_types_translation.hpp>
namespace ams::socket::impl {
extern PosixWinSockConverter g_posix_winsock_converter;
namespace {
constinit util::Atomic<int> g_init_counter = 0;
ALWAYS_INLINE bool IsInitialized() {
return g_init_counter > 0;
}
class FcntlState {
private:
FcntlFlag m_flags[MaxSocketsPerClient]{};
os::SdkRecursiveMutex m_mutexes[MaxSocketsPerClient]{};
public:
constexpr FcntlState() = default;
public:
void ClearFlag(int fd, FcntlFlag flag) {
std::scoped_lock lk(m_mutexes[fd]);
m_flags[fd] &= ~flag;
}
void ClearFlags(int fd) {
std::scoped_lock lk(m_mutexes[fd]);
m_flags[fd] = FcntlFlag::None;
}
FcntlFlag GetFlags(int fd) {
std::scoped_lock lk(m_mutexes[fd]);
return m_flags[fd];
}
int GetFlagsInt(int fd) {
return static_cast<int>(this->GetFlags(fd));
}
os::SdkRecursiveMutex &GetSocketLock(int fd) {
return m_mutexes[fd];
}
bool IsFlagClear(int fd, FcntlFlag flag) {
return !this->IsFlagSet(fd, flag);
}
bool IsFlagSet(int fd, FcntlFlag flag) {
std::scoped_lock lk(m_mutexes[fd]);
return (m_flags[fd] & flag) != static_cast<FcntlFlag>(0);
}
bool IsSocketBlocking(int fd) {
return !this->IsSocketNonBlocking(fd);
}
bool IsSocketNonBlocking(int fd) {
return this->IsFlagSet(fd, FcntlFlag::O_NonBlock);
}
void SetFlag(int fd, FcntlFlag flag) {
std::scoped_lock lk(m_mutexes[fd]);
m_flags[fd] |= flag;
}
};
constinit FcntlState g_fcntl_state;
void TransmuteWsaError() {
switch (::WSAGetLastError()) {
case WSAEFAULT: ::WSASetLastError(WSAEINVAL); break;
case WSAENOTSOCK: ::WSASetLastError(WSAEBADF); break;
case WSAETIMEDOUT: ::WSASetLastError(WSAEWOULDBLOCK); break;
}
}
template<std::integral T>
void TransmuteWsaError(T res) {
if (static_cast<decltype(SOCKET_ERROR)>(res) == SOCKET_ERROR) {
TransmuteWsaError();
}
}
}
#define AMS_SOCKET_IMPL_SCOPED_MAKE_NON_BLOCKING(_cond, _fd) \
/* If the socket is blocking and we need to make it non-blocking, do so. */ \
int nonblock_##__LINE__ = 1; \
bool set_nonblock_##__LINE__ = false; \
if (_cond && g_fcntl_state.IsSocketBlocking(_fd)) { \
if (const auto res = ::ioctlsocket(handle, FIONBIO, reinterpret_cast<u_long *>(std::addressof( nonblock_##__LINE__ ))); res == SOCKET_ERROR) { \
TransmuteWsaError(); \
return res; \
} \
\
set_nonblock_##__LINE__ = true; \
} \
\
ON_SCOPE_EXIT { \
/* Preserve last error. */ \
const auto last_err = socket::impl::GetLastError(); \
ON_SCOPE_EXIT { socket::impl::SetLastError(last_err); }; \
\
/* Restore non-blocking state. */ \
if (set_nonblock_##__LINE__) { \
nonblock_##__LINE__ = 0; \
\
while (true) { \
const auto restore_res = ::ioctlsocket(handle, FIONBIO, reinterpret_cast<u_long *>(std::addressof( nonblock_##__LINE__ ))); \
TransmuteWsaError(restore_res); \
if (!(restore_res == SOCKET_ERROR && socket::impl::GetLastError() == Errno::EInProgress)) { \
break; \
} \
\
os::SleepThread(TimeSpan::FromMilliSeconds(1)); \
} \
} \
}
#define AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(expr) ({ const auto res = (expr); TransmuteWsaError(res); res; })
void *Alloc(size_t size) {
return ::std::malloc(size);
}
void *Calloc(size_t num, size_t size) {
const size_t total_size = size * num;
void *buf = Alloc(size);
if (buf != nullptr) {
std::memset(buf, 0, total_size);
}
return buf;
}
void Free(void *ptr) {
return ::std::free(ptr);
}
Errno GetLastError() {
if (AMS_LIKELY(IsInitialized())) {
return MapErrnoValue(::WSAGetLastError());
} else {
return Errno::EInval;
}
}
void SetLastError(Errno err) {
if (AMS_LIKELY(IsInitialized())) {
::WSASetLastError(MapErrnoValue(err));
}
}
u32 InetHtonl(u32 host) {
return ::htonl(host);
}
u16 InetHtons(u16 host) {
return ::htons(host);
}
u32 InetNtohl(u32 net) {
return ::ntohl(net);
}
u16 InetNtohs(u16 net) {
return ::ntohs(net);
}
Result Initialize(const Config &config) {
AMS_UNUSED(config);
/* Increment init counter. */
++g_init_counter;
/* Initialize winsock. */
WSADATA wsa_data;
WORD wVersionRequested = MAKEWORD(2, 2);
const auto res = ::WSAStartup(wVersionRequested, std::addressof(wsa_data));
AMS_ABORT_UNLESS(res == 0);
/* Initialize time services. */
R_ABORT_UNLESS(time::Initialize());
R_SUCCEED();
}
Result Finalize() {
/* Check pre-conditions. */
--g_init_counter;
AMS_ABORT_UNLESS(g_init_counter >= 0);
/* Cleanup WSA. */
::WSACleanup();
/* Finalize time services. */
time::Finalize();
/* Release all posix handles. */
g_posix_winsock_converter.ReleaseAllPosixHandles();
R_SUCCEED();
}
ssize_t RecvFromInternal(s32 desc, void *buffer, size_t buffer_size, MsgFlag flags, SockAddr *out_address, SockLenT *out_addr_len) {
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Convert the sockaddr. */
sockaddr sa = {};
socklen_t addr_len = sizeof(sa);
/* Perform the call. */
const auto res = ::recvfrom(handle, static_cast<char *>(buffer), static_cast<int>(buffer_size), MapMsgFlagValue(flags), std::addressof(sa), std::addressof(addr_len));
if (res == SOCKET_ERROR) {
if (::WSAGetLastError() == WSAESHUTDOWN) {
::WSASetLastError(WSAENETDOWN);
} else {
TransmuteWsaError();
}
}
/* Set output. */
if (out_address != nullptr && out_addr_len != nullptr) {
if (addr_len > static_cast<socklen_t>(sizeof(*out_address))) {
addr_len = sizeof(*out_address);
}
if (*out_addr_len != 0) {
if (static_cast<socklen_t>(*out_addr_len) > addr_len) {
*out_addr_len = addr_len;
}
SockAddr sa_pl = {};
CopyFromPlatform(reinterpret_cast<SockAddrIn *>(std::addressof(sa_pl)), reinterpret_cast<const sockaddr_in *>(std::addressof(sa)));
std::memcpy(out_address, std::addressof(sa_pl), *out_addr_len);
}
}
return res;
}
ssize_t RecvFrom(s32 desc, void *buffer, size_t buffer_size, MsgFlag flags, SockAddr *out_address, SockLenT *out_addr_len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* If the flags have DontWait set, clear WaitAll. */
if ((flags & MsgFlag::Msg_DontWait) == MsgFlag::Msg_DontWait) {
flags &= ~MsgFlag::Msg_WaitAll;
}
/* If the flags haev WaitAll set but the socket is non-blocking, clear WaitAll. */
if ((flags & MsgFlag::Msg_WaitAll) == MsgFlag::Msg_WaitAll && g_fcntl_state.IsSocketNonBlocking(desc)) {
flags &= ~MsgFlag::Msg_WaitAll;
}
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
} else if (buffer_size == 0) {
return 0;
} else if (buffer == nullptr) {
socket::impl::SetLastError(Errno::EInval);
return -1;
} else if (buffer_size > std::numeric_limits<u32>::max()) {
socket::impl::SetLastError(Errno::EFault);
return -1;
}
/* Handle blocking vs non-blocking. */
if ((flags & MsgFlag::Msg_DontWait) == MsgFlag::Msg_DontWait) {
return RecvFromInternal(desc, buffer, buffer_size, flags, out_address, out_addr_len);
} else {
/* Lock the socket. */
std::scoped_lock lk(g_fcntl_state.GetSocketLock(desc));
/* Clear don't wait from the flags. */
flags &= MsgFlag::Msg_DontWait;
/* If the socket is blocking, we need to make it non-blocking. */
AMS_SOCKET_IMPL_SCOPED_MAKE_NON_BLOCKING(true, desc);
/* Do the recv from. */
return RecvFromInternal(desc, buffer, buffer_size, flags, out_address, out_addr_len);
}
}
ssize_t Recv(s32 desc, void *buffer, size_t buffer_size, MsgFlag flags) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
} else if (buffer_size == 0) {
return 0;
} else if (buffer == nullptr) {
socket::impl::SetLastError(Errno::EInval);
return -1;
} else if (buffer_size > std::numeric_limits<u32>::max()) {
socket::impl::SetLastError(Errno::EFault);
return -1;
}
/* If the socket is blocking, we need to make it non-blocking. */
AMS_SOCKET_IMPL_SCOPED_MAKE_NON_BLOCKING(((flags & MsgFlag::Msg_DontWait) == MsgFlag::Msg_DontWait), desc);
/* Perform the call. */
return AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(::recv(handle, static_cast<char *>(buffer), static_cast<int>(buffer_size), MapMsgFlagValue(flags & ~MsgFlag::Msg_DontWait)));
}
ssize_t SendTo(s32 desc, const void *buffer, size_t buffer_size, MsgFlag flags, const SockAddr *address, SockLenT len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Clear don't wait from flags. */
flags &= ~MsgFlag::Msg_DontWait;
/* Convert the sockaddr. */
sockaddr sa = {};
socket::impl::CopyToPlatform(reinterpret_cast<sockaddr_in *>(std::addressof(sa)), reinterpret_cast<const SockAddrIn *>(address));
/* Perform the call. */
const auto res = ::sendto(handle, static_cast<const char *>(buffer), static_cast<int>(buffer_size), MapMsgFlagValue(flags), address != nullptr ? std::addressof(sa) : nullptr, static_cast<socklen_t>(len));
if (res == SOCKET_ERROR) {
if (::WSAGetLastError() == WSAESHUTDOWN) {
::WSASetLastError(109);
} else {
TransmuteWsaError();
}
}
return res;
}
ssize_t Send(s32 desc, const void *buffer, size_t buffer_size, MsgFlag flags) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Perform the call. */
return AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(::send(handle, static_cast<const char *>(buffer), static_cast<int>(buffer_size), MapMsgFlagValue(flags)));
}
s32 Shutdown(s32 desc, ShutdownMethod how) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Perform the call. */
const auto res = ::shutdown(handle, MapShutdownMethodValue(how));
g_posix_winsock_converter.SetShutdown(desc, true);
TransmuteWsaError(res);
return res;
}
s32 Socket(Family domain, Type type, Protocol protocol, bool exempt) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
const auto res = ::socket(MapFamilyValue(domain), MapTypeValue(type), MapProtocolValue(protocol));
TransmuteWsaError(res);
s32 posix_socket = -1;
if (res != static_cast<typename std::remove_cv<decltype(res)>::type>(SOCKET_ERROR)) {
if (posix_socket = g_posix_winsock_converter.AcquirePosixHandle(res, exempt); posix_socket < 0) {
/* Preserve last error. */
const auto last_err = socket::impl::GetLastError();
ON_SCOPE_EXIT { socket::impl::SetLastError(last_err); };
/* Close the socket. */
::closesocket(res);
}
}
return posix_socket;
}
s32 Socket(Family domain, Type type, Protocol protocol) {
return Socket(domain, type, protocol, false);
}
s32 SocketExempt(Family domain, Type type, Protocol protocol) {
return Socket(domain, type, protocol, true);
}
s32 Accept(s32 desc, SockAddr *out_address, SockLenT *out_addr_len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Check shutdown. */
bool is_shutdown = false;
if (const auto res = g_posix_winsock_converter.GetShutdown(is_shutdown, desc); res == SOCKET_ERROR || (res == 0 && is_shutdown)) {
socket::impl::SetLastError(Errno::EConnAborted);
return -1;
}
/* Accept. */
sockaddr sa = {};
socklen_t sa_len = sizeof(sa);
const auto res = ::accept(handle, std::addressof(sa), std::addressof(sa_len));
if (res == static_cast<typename std::remove_cv<decltype(res)>::type>(SOCKET_ERROR)) {
if (::WSAGetLastError() == WSAEOPNOTSUPP) {
::WSASetLastError(WSAEINVAL);
} else {
TransmuteWsaError();
}
}
/* Set output. */
if (out_address != nullptr && out_addr_len != nullptr) {
if (sa_len > static_cast<socklen_t>(sizeof(*out_address))) {
sa_len = sizeof(*out_address);
}
if (*out_addr_len != 0) {
if (static_cast<socklen_t>(*out_addr_len) > sa_len) {
*out_addr_len = sa_len;
}
SockAddr sa_pl = {};
CopyFromPlatform(reinterpret_cast<SockAddrIn *>(std::addressof(sa_pl)), reinterpret_cast<const sockaddr_in *>(std::addressof(sa)));
std::memcpy(out_address, std::addressof(sa_pl), *out_addr_len);
}
*out_addr_len = sa_len;
}
if (res == static_cast<typename std::remove_cv<decltype(res)>::type>(SOCKET_ERROR)) {
return res;
}
s32 fd = -1;
bool is_exempt = false;
if (g_posix_winsock_converter.GetSocketExempt(is_exempt, desc) == 0) {
fd = g_posix_winsock_converter.AcquirePosixHandle(res, is_exempt);
}
if (fd < 0) {
/* Preserve last error. */
const auto last_err = socket::impl::GetLastError();
ON_SCOPE_EXIT { socket::impl::SetLastError(last_err); };
::closesocket(res);
return SOCKET_ERROR;
}
return fd;
}
s32 Bind(s32 desc, const SockAddr *address, SockLenT len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
} else if (address == nullptr) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
/* Convert the sockaddr. */
sockaddr sa = {};
socket::impl::CopyToPlatform(reinterpret_cast<sockaddr_in *>(std::addressof(sa)), reinterpret_cast<const SockAddrIn *>(address));
return AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(::bind(handle, std::addressof(sa), static_cast<socklen_t>(len)));
}
s32 Connect(s32 desc, const SockAddr *address, SockLenT len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Convert the sockaddr. */
sockaddr sa = {};
if (address != nullptr) {
if (reinterpret_cast<const SockAddrIn *>(address)->sin_port == 0) {
socket::impl::SetLastError(Errno::EAddrNotAvail);
return -1;
}
socket::impl::CopyToPlatform(reinterpret_cast<sockaddr_in *>(std::addressof(sa)), reinterpret_cast<const SockAddrIn *>(address));
}
const auto res = ::connect(handle, address != nullptr ? std::addressof(sa) : nullptr, len);
if (res == SOCKET_ERROR) {
const auto wsa_err = ::WSAGetLastError();
if (wsa_err == WSAEWOULDBLOCK) {
::WSASetLastError(WSAEINPROGRESS);
} else if (wsa_err != WSAETIMEDOUT) {
TransmuteWsaError();
}
}
return res;
}
s32 GetSockName(s32 desc, SockAddr *out_address, SockLenT *out_addr_len) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* We may end up preserving the last wsa error. */
const auto last_err = ::WSAGetLastError();
/* Do the call. */
sockaddr sa = {};
auto res = ::getsockname(handle, out_address != nullptr ? std::addressof(sa) : nullptr, reinterpret_cast<socklen_t *>(out_addr_len));
if (res == SOCKET_ERROR) {
if (::WSAGetLastError() == WSAEINVAL) {
::WSASetLastError(last_err);
sa = {};
res = 0;
} else {
TransmuteWsaError();
}
}
/* Copy out. */
if (out_address != nullptr) {
CopyFromPlatform(reinterpret_cast<SockAddrIn *>(out_address), reinterpret_cast<const sockaddr_in *>(std::addressof(sa)));
}
return res;
}
s32 SetSockOpt(s32 desc, Level level, Option option_name, const void *option_value, SockLenT option_size) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
union SocketOptionValue {
linger option_linger;
DWORD option_timeout_ms;
DWORD option_exempt;
};
SocketOptionValue sockopt_value = {};
socklen_t option_value_length = option_size;
const char *p_option_value = nullptr;
switch (option_name) {
case Option::So_Linger:
case Option::So_Nn_Linger:
{
if (option_value_length < static_cast<socklen_t>(sizeof(sockopt_value.option_linger))) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
option_value_length = sizeof(sockopt_value.option_linger);
CopyToPlatform(std::addressof(sockopt_value.option_linger), reinterpret_cast<const Linger *>(option_value));
p_option_value = reinterpret_cast<const char *>(std::addressof(sockopt_value.option_linger));
}
break;
case Option::So_SndTimeo:
case Option::So_RcvTimeo:
{
if (option_value_length < static_cast<socklen_t>(sizeof(sockopt_value.option_timeout_ms))) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
option_value_length = sizeof(sockopt_value.option_timeout_ms);
sockopt_value.option_timeout_ms = (reinterpret_cast<const TimeVal *>(option_value)->tv_sec * 1000) + (reinterpret_cast<const TimeVal *>(option_value)->tv_usec / 1000);
p_option_value = reinterpret_cast<const char *>(std::addressof(sockopt_value.option_timeout_ms));
}
break;
case Option::So_Nn_Shutdown_Exempt:
{
if (option_value_length < static_cast<socklen_t>(sizeof(sockopt_value.option_exempt))) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
return g_posix_winsock_converter.SetSocketExempt(desc, *reinterpret_cast<const decltype(sockopt_value.option_exempt) *>(option_value) != 0);
}
break;
default:
p_option_value = reinterpret_cast<const char *>(option_value);
break;
}
return AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(::setsockopt(handle, MapLevelValue(level), MapOptionValue(level, option_name), p_option_value, option_value_length));
}
s32 Listen(s32 desc, s32 backlog) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Convert socket. */
SOCKET handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
/* Check input. */
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
socket::impl::SetLastError(Errno::EBadf);
return -1;
}
/* Check shutdown. */
bool is_shutdown = false;
if (const auto res = g_posix_winsock_converter.GetShutdown(is_shutdown, desc); res == SOCKET_ERROR || (res == 0 && is_shutdown)) {
socket::impl::SetLastError(Errno::EInval);
return -1;
}
return AMS_SOCKET_IMPL_DO_WITH_TRANSMUTE(::listen(handle, backlog));
}
s32 Close(s32 desc) {
/* Check pre-conditions. */
AMS_ABORT_UNLESS(IsInitialized());
/* Check that we can close. */
static constinit os::SdkMutex s_close_lock;
SOCKET handle = static_cast<SOCKET>(socket::InvalidSocket);
{
std::scoped_lock lk(s_close_lock);
handle = g_posix_winsock_converter.PosixToWinsockSocket(desc);
if (handle == static_cast<SOCKET>(socket::InvalidSocket)) {
return SOCKET_ERROR;
}
g_posix_winsock_converter.ReleasePosixHandle(desc);
}
/* Do the close. */
const auto res = ::closesocket(handle);
g_fcntl_state.ClearFlags(desc);
return res;
}
}

776
libraries/libstratosphere/source/socket/impl/socket_platform_types_translation.os.windows.cpp Normal file
View file

@ -0,0 +1,776 @@
/*
* Copyright (c) Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stratosphere.hpp>
#include "socket_api.hpp"
#include <ws2tcpip.h>
#include <stratosphere/socket/impl/socket_platform_types_translation.hpp>
namespace ams::socket::impl {
///* TODO: Custom sys/* headers, probably. */
#define AF_LINK 18
#define MSG_TRUNC 0x10
#define MSG_CTRUNC 0x20
#define MSG_DONTWAIT 0x80
#define SHUT_RD 0
#define SHUT_WR 1
#define SHUT_RDWR 2
#define O_NONBLOCK 4
#define TCP_MAXSEG 4
PosixWinSockConverter g_posix_winsock_converter;
s32 PosixWinSockConverter::AcquirePosixHandle(SOCKET winsock, bool exempt) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Get initial index. */
const auto initial_index = GetInitialIndex(winsock);
/* Try to find an open index. */
for (auto posix = initial_index; posix < static_cast<int>(MaxSocketsPerClient); ++posix) {
if (m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
m_data[posix].winsock = winsock;
m_data[posix].exempt = exempt;
return posix;
}
}
for (auto posix = 0; posix < initial_index; ++posix) {
if (m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
m_data[posix].winsock = winsock;
m_data[posix].exempt = exempt;
return posix;
}
}
/* We're out of open handles. */
socket::impl::SetLastError(Errno::EMFile);
return SOCKET_ERROR;
}
s32 PosixWinSockConverter::GetShutdown(bool &shutdown, s32 posix) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Check input. */
if (static_cast<u32>(posix) >= MaxSocketsPerClient || m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
socket::impl::SetLastError(Errno::EBadf);
return SOCKET_ERROR;
}
/* Set the output. */
shutdown = m_data[posix].shutdown;
return 0;
}
s32 PosixWinSockConverter::GetSocketExempt(bool &exempt, s32 posix) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Check input. */
if (static_cast<u32>(posix) >= MaxSocketsPerClient || m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
socket::impl::SetLastError(Errno::EBadf);
return SOCKET_ERROR;
}
/* Set the output. */
exempt = m_data[posix].exempt;
return 0;
}
SOCKET PosixWinSockConverter::PosixToWinsockSocket(s32 posix) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Check input. */
if (static_cast<u32>(posix) >= MaxSocketsPerClient || m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
socket::impl::SetLastError(Errno::EBadf);
return SOCKET_ERROR;
}
return m_data[posix].winsock;
}
void PosixWinSockConverter::ReleaseAllPosixHandles() {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
for (size_t i = 0; i < MaxSocketsPerClient; ++i) {
m_data[i] = SocketData{};
}
}
void PosixWinSockConverter::ReleasePosixHandle(s32 posix) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
AMS_ASSERT(static_cast<u32>(posix) < MaxSocketsPerClient);
m_data[posix] = SocketData{};
}
s32 PosixWinSockConverter::SetShutdown(s32 posix, bool shutdown) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Check input. */
if (static_cast<u32>(posix) >= MaxSocketsPerClient || m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
socket::impl::SetLastError(Errno::EBadf);
return SOCKET_ERROR;
}
/* Set the shutdown. */
m_data[posix].shutdown = shutdown;
return 0;
}
s32 PosixWinSockConverter::SetSocketExempt(s32 posix, bool exempt) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Check input. */
if (static_cast<u32>(posix) >= MaxSocketsPerClient || m_data[posix].winsock == static_cast<SOCKET>(INVALID_SOCKET)) {
socket::impl::SetLastError(Errno::EBadf);
return SOCKET_ERROR;
}
/* Set the exempt. */
m_data[posix].exempt = exempt;
return 0;
}
s32 PosixWinSockConverter::WinsockToPosixSocket(SOCKET winsock) {
/* Acquire exclusive access. */
std::scoped_lock lk(m_mutex);
/* Get initial index. */
const auto initial_index = GetInitialIndex(winsock);
/* Try to find an open index. */
for (auto posix = initial_index; posix < static_cast<int>(MaxSocketsPerClient); ++posix) {
if (m_data[posix].winsock == winsock) {
return posix;
}
}
for (auto posix = 0; posix < initial_index; ++posix) {
if (m_data[posix].winsock == winsock) {
return posix;
}
}
/* We failed to find the posix handle. */
return -1;
}
s32 MapProtocolValue(Protocol protocol) {
s32 mapped = -1;
switch (protocol) {
case Protocol::IpProto_Ip: mapped = IPPROTO_IP; break;
case Protocol::IpProto_Icmp: mapped = IPPROTO_ICMP; break;
case Protocol::IpProto_Tcp: mapped = IPPROTO_TCP; break;
case Protocol::IpProto_Udp: mapped = IPPROTO_UDP; break;
case Protocol::IpProto_None: mapped = IPPROTO_NONE; break;
case Protocol::IpProto_UdpLite: mapped = IPPROTO_UDP; break;
case Protocol::IpProto_Raw: mapped = IPPROTO_RAW; break;
case Protocol::IpProto_Max: mapped = IPPROTO_MAX; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Protocol %d\n", static_cast<int>(protocol));
break;
}
if (mapped == -1) {
AMS_SDK_LOG("WARNING: ams::Socket::Protocol %d is not supported by Win32/Win64.\n", static_cast<int>(protocol));
}
return mapped;
}
Protocol MapProtocolValue(s32 protocol) {
Protocol mapped = Protocol::IpProto_None;
switch (protocol) {
case IPPROTO_IP: mapped = Protocol::IpProto_Ip; break;
case IPPROTO_ICMP: mapped = Protocol::IpProto_Icmp; break;
case IPPROTO_TCP: mapped = Protocol::IpProto_Tcp; break;
case IPPROTO_UDP: mapped = Protocol::IpProto_Udp; break;
case IPPROTO_NONE: mapped = Protocol::IpProto_None; break;
case IPPROTO_RAW: mapped = Protocol::IpProto_Raw; break;
case IPPROTO_MAX: mapped = Protocol::IpProto_Max; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket protocol %d\n", static_cast<int>(protocol));
break;
}
return mapped;
}
s32 MapTypeValue(Type type) {
s32 mapped = -1;
switch (type) {
case Type::Sock_Default: mapped = 0; break;
case Type::Sock_Stream: mapped = SOCK_STREAM; break;
case Type::Sock_Dgram: mapped = SOCK_DGRAM; break;
case Type::Sock_Raw: mapped = SOCK_RAW; break;
case Type::Sock_SeqPacket: mapped = SOCK_SEQPACKET; break;
case Type::Sock_NonBlock: mapped = -1; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Type %d\n", static_cast<int>(type));
break;
}
if (mapped == -1) {
AMS_SDK_LOG("WARNING: ams::Socket::Type %d is not supported by Win32/Win64.\n", static_cast<int>(type));
}
return mapped;
}
Type MapTypeValue(s32 type) {
Type mapped = Type::Sock_Default;
switch (type) {
case 0: mapped = Type::Sock_Default; break;
case SOCK_STREAM: mapped = Type::Sock_Stream; break;
case SOCK_DGRAM: mapped = Type::Sock_Dgram; break;
case SOCK_RAW: mapped = Type::Sock_Raw; break;
case SOCK_SEQPACKET: mapped = Type::Sock_SeqPacket; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket type %d\n", static_cast<int>(type));
break;
}
return mapped;
}
s8 MapFamilyValue(Family family) {
s8 mapped = -1;
switch (family) {
case Family::Af_Unspec: mapped = AF_UNSPEC; break;
case Family::Af_Inet: mapped = AF_INET; break;
case Family::Af_Route: mapped = -1; break;
case Family::Af_Link: mapped = AF_LINK; break;
case Family::Af_Inet6: mapped = AF_INET6; break;
case Family::Af_Max: mapped = AF_MAX; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Family %d\n", static_cast<int>(family));
break;
}
if (mapped == -1) {
AMS_SDK_LOG("WARNING: ams::Socket::Family %d is not supported by Win32/Win64.\n", static_cast<int>(family));
}
return mapped;
}
Family MapFamilyValue(s8 family) {
Family mapped = Family::Af_Unspec;
switch (family) {
case AF_UNSPEC:mapped = Family::Af_Unspec; break;
case AF_INET: mapped = Family::Af_Inet; break;
case AF_LINK: mapped = Family::Af_Link; break;
case AF_INET6: mapped = Family::Af_Inet6; break;
case AF_MAX: mapped = Family::Af_Max; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket family %d\n", static_cast<int>(family));
break;
}
return mapped;
}
s32 MapMsgFlagValue(MsgFlag flag) {
s32 mapped = 0;
if ((flag & MsgFlag::Msg_Oob) != MsgFlag::Msg_None) { mapped |= MSG_OOB; }
if ((flag & MsgFlag::Msg_Peek) != MsgFlag::Msg_None) { mapped |= MSG_PEEK; }
if ((flag & MsgFlag::Msg_DontRoute) != MsgFlag::Msg_None) { mapped |= MSG_DONTROUTE; }
if ((flag & MsgFlag::Msg_Trunc) != MsgFlag::Msg_None) { mapped |= MSG_TRUNC; }
if ((flag & MsgFlag::Msg_CTrunc) != MsgFlag::Msg_None) { mapped |= MSG_CTRUNC; }
if ((flag & MsgFlag::Msg_WaitAll) != MsgFlag::Msg_None) { mapped |= MSG_WAITALL; }
if ((flag & MsgFlag::Msg_DontWait) != MsgFlag::Msg_None) { mapped |= MSG_DONTWAIT; }
return mapped;
}
MsgFlag MapMsgFlagValue(s32 flag) {
MsgFlag mapped = MsgFlag::Msg_None;
if (flag & MSG_OOB) { mapped |= MsgFlag::Msg_Oob; }
if (flag & MSG_PEEK) { mapped |= MsgFlag::Msg_Peek; }
if (flag & MSG_DONTROUTE) { mapped |= MsgFlag::Msg_DontRoute; }
if (flag & MSG_TRUNC) { mapped |= MsgFlag::Msg_Trunc; }
if (flag & MSG_CTRUNC) { mapped |= MsgFlag::Msg_CTrunc; }
if (flag & MSG_WAITALL) { mapped |= MsgFlag::Msg_WaitAll; }
if (flag & MSG_DONTWAIT) { mapped |= MsgFlag::Msg_DontWait; }
return mapped;
}
u32 MapAddrInfoFlagValue(AddrInfoFlag flag) {
u32 mapped = 0;
if ((flag & AddrInfoFlag::Ai_Passive) != AddrInfoFlag::Ai_None) { mapped |= AI_PASSIVE; }
if ((flag & AddrInfoFlag::Ai_CanonName) != AddrInfoFlag::Ai_None) { mapped |= AI_CANONNAME; }
if ((flag & AddrInfoFlag::Ai_NumericHost) != AddrInfoFlag::Ai_None) { mapped |= AI_NUMERICHOST; }
if ((flag & AddrInfoFlag::Ai_NumericServ) != AddrInfoFlag::Ai_None) { mapped |= AI_NUMERICSERV; }
if ((flag & AddrInfoFlag::Ai_AddrConfig) != AddrInfoFlag::Ai_None) { mapped |= AI_ADDRCONFIG; }
return mapped;
}
AddrInfoFlag MapAddrInfoFlagValue(u32 flag) {
AddrInfoFlag mapped = AddrInfoFlag::Ai_None;
if (flag & AI_PASSIVE) { mapped |= AddrInfoFlag::Ai_Passive; }
if (flag & AI_CANONNAME) { mapped |= AddrInfoFlag::Ai_CanonName; }
if (flag & AI_NUMERICHOST) { mapped |= AddrInfoFlag::Ai_NumericHost; }
if (flag & AI_NUMERICSERV) { mapped |= AddrInfoFlag::Ai_NumericServ; }
if (flag & AI_ADDRCONFIG) { mapped |= AddrInfoFlag::Ai_AddrConfig; }
return mapped;
}
u32 MapShutdownMethodValue(ShutdownMethod how) {
u32 mapped = -1;
switch (how) {
case ShutdownMethod::Shut_Rd: mapped = SHUT_RD; break;
case ShutdownMethod::Shut_Wr: mapped = SHUT_WR; break;
case ShutdownMethod::Shut_RdWr: mapped = SHUT_RDWR; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::ShutdownMethod %d\n", static_cast<int>(how));
break;
}
return mapped;
}
ShutdownMethod MapShutdownMethodValue(u32 how) {
ShutdownMethod mapped = static_cast<ShutdownMethod>(-1);
switch (how) {
case SHUT_RD: mapped = ShutdownMethod::Shut_Rd; break;
case SHUT_WR: mapped = ShutdownMethod::Shut_Wr; break;
case SHUT_RDWR: mapped = ShutdownMethod::Shut_RdWr; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket shutdown %d\n", static_cast<int>(how));
break;
}
return mapped;
}
u32 MapFcntlFlagValue(FcntlFlag flag) {
u32 mapped = 0;
switch (flag) {
case FcntlFlag::O_NonBlock: mapped = O_NONBLOCK; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::FcntlFlag %d\n", static_cast<int>(flag));
break;
}
return mapped;
}
FcntlFlag MapFcntlFlagValue(u32 flag) {
FcntlFlag mapped = FcntlFlag::None;
switch (flag) {
case O_NONBLOCK: mapped = FcntlFlag::O_NonBlock; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket fcntl flag %d\n", static_cast<int>(flag));
break;
}
return mapped;
}
s32 MapLevelValue(Level level) {
s32 mapped = -1;
switch (level) {
case Level::Sol_Socket: mapped = SOL_SOCKET; break;
case Level::Sol_Ip: mapped = IPPROTO_IP; break;
case Level::Sol_Icmp: mapped = IPPROTO_ICMP; break;
case Level::Sol_Tcp: mapped = IPPROTO_TCP; break;
case Level::Sol_Udp: mapped = IPPROTO_UDP; break;
case Level::Sol_UdpLite: mapped = IPPROTO_UDP; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Level %d\n", static_cast<int>(level));
break;
}
return mapped;
}
Level MapLevelValue(s32 level) {
Level mapped = static_cast<Level>(0);
switch (level) {
case SOL_SOCKET: mapped = Level::Sol_Socket; break;
case IPPROTO_IP: mapped = Level::Sol_Ip; break;
case IPPROTO_ICMP: mapped = Level::Sol_Icmp; break;
case IPPROTO_TCP: mapped = Level::Sol_Tcp; break;
case IPPROTO_UDP: mapped = Level::Sol_Udp; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket level %d\n", static_cast<int>(level));
break;
}
return mapped;
}
s32 MapOptionValue(Level level, Option option) {
s32 mapped = -1;
switch (level) {
case Level::Sol_Socket:
switch (option) {
case Option::So_Debug: mapped = SO_DEBUG; break;
case Option::So_AcceptConn: mapped = SO_ACCEPTCONN; break;
case Option::So_ReuseAddr: mapped = SO_REUSEADDR; break;
case Option::So_KeepAlive: mapped = SO_KEEPALIVE; break;
case Option::So_DontRoute: mapped = SO_DONTROUTE; break;
case Option::So_Broadcast: mapped = SO_BROADCAST; break;
case Option::So_UseLoopback: mapped = SO_USELOOPBACK; break;
case Option::So_Linger: mapped = SO_LINGER; break;
case Option::So_OobInline: mapped = SO_OOBINLINE; break;
case Option::So_ReusePort: mapped = -1; break;
case Option::So_SndBuf: mapped = SO_SNDBUF; break;
case Option::So_RcvBuf: mapped = SO_RCVBUF; break;
case Option::So_SndLoWat: mapped = SO_SNDLOWAT; break;
case Option::So_RcvLoWat: mapped = SO_RCVLOWAT; break;
case Option::So_SndTimeo: mapped = SO_SNDTIMEO; break;
case Option::So_RcvTimeo: mapped = SO_RCVTIMEO; break;
case Option::So_Error: mapped = SO_ERROR; break;
case Option::So_Type: mapped = SO_TYPE; break;
case Option::So_Label: mapped = -1; break;
case Option::So_PeerLabel: mapped = -1; break;
case Option::So_ListenQLimit: mapped = -1; break;
case Option::So_ListenQLen: mapped = -1; break;
case Option::So_ListenIncQLen: mapped = -1; break;
case Option::So_SetFib: mapped = -1; break;
case Option::So_User_Cookie: mapped = -1; break;
case Option::So_Protocol: mapped = -1; break;
case Option::So_Vendor: mapped = -1; break;
case Option::So_Nn_Linger: mapped = -1; break;
case Option::So_Nn_Shutdown_Exempt: mapped = -1; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Option %d for Level::Sol_Socket\n", static_cast<int>(option));
break;
}
break;
case Level::Sol_Ip:
switch (option) {
case Option::Ip_Options: mapped = IP_OPTIONS; break;
case Option::Ip_HdrIncl: mapped = IP_HDRINCL; break;
case Option::Ip_Tos: mapped = IP_TOS; break;
case Option::Ip_Ttl: mapped = IP_TTL; break;
case Option::Ip_RecvOpts: mapped = -1; break;
case Option::Ip_Multicast_If: mapped = IP_MULTICAST_IF; break;
case Option::Ip_Multicast_Ttl: mapped = IP_MULTICAST_TTL; break;
case Option::Ip_Multicast_Loop: mapped = IP_MULTICAST_LOOP; break;
case Option::Ip_Add_Membership: mapped = IP_ADD_MEMBERSHIP; break;
case Option::Ip_Drop_Membership: mapped = IP_DROP_MEMBERSHIP; break;
case Option::Ip_Multicast_Vif: mapped = -1; break;
case Option::Ip_Rsvp_On: mapped = -1; break;
case Option::Ip_Rsvp_Off: mapped = -1; break;
case Option::Ip_Rsvp_Vif_On: mapped = -1; break;
case Option::Ip_Rsvp_Vif_Off: mapped = -1; break;
case Option::Ip_PortRange: mapped = -1; break;
case Option::Ip_Faith: mapped = -1; break;
case Option::Ip_OnesBcast: mapped = -1; break;
case Option::Ip_BindAny: mapped = -1; break;
case Option::Ip_RecvTtl: mapped = -1; break;
case Option::Ip_MinTtl: mapped = -1; break;
case Option::Ip_DontFrag: mapped = -1; break;
case Option::Ip_RecvTos: mapped = -1; break;
case Option::Ip_Add_Source_Membership: mapped = IP_ADD_SOURCE_MEMBERSHIP; break;
case Option::Ip_Drop_Source_Membership: mapped = IP_DROP_SOURCE_MEMBERSHIP; break;
case Option::Ip_Block_Source: mapped = IP_BLOCK_SOURCE; break;
case Option::Ip_Unblock_Source: mapped = IP_UNBLOCK_SOURCE; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Option %d for Level::Sol_Ip\n", static_cast<int>(option));
break;
}
break;
case Level::Sol_Tcp:
switch (option) {
case Option::Tcp_NoDelay: mapped = TCP_NODELAY; break;
case Option::Tcp_MaxSeg: mapped = TCP_MAXSEG; break;
case Option::Tcp_NoPush: mapped = -1; break;
case Option::Tcp_NoOpt: mapped = -1; break;
case Option::Tcp_Md5Sig: mapped = -1; break;
case Option::Tcp_Info: mapped = -1; break;
case Option::Tcp_Congestion: mapped = -1; break;
case Option::Tcp_KeepInit: mapped = -1; break;
case Option::Tcp_KeepIdle: mapped = -1; break;
case Option::Tcp_KeepIntvl: mapped = -1; break;
case Option::Tcp_KeepCnt: mapped = -1; break;
case Option::Tcp_Vendor: mapped = -1; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Option %d for Level::Sol_Tcp\n", static_cast<int>(option));
break;
}
break;
default:
AMS_SDK_LOG("WARNING: Invalid option level %d\n", static_cast<int>(level));
break;
}
if (mapped == -1) {
AMS_SDK_LOG("WARNING: ams::Socket::Option %d is not supported by Win32/Win64.\n", static_cast<int>(option));
}
return mapped;
}
Option MapOptionValue(s32 level, s32 option) {
Option mapped = static_cast<Option>(0);
switch (level) {
case SOL_SOCKET:
switch (option) {
case SO_DEBUG: mapped = Option::So_Debug; break;
case SO_ACCEPTCONN: mapped = Option::So_AcceptConn; break;
case SO_REUSEADDR: mapped = Option::So_ReuseAddr; break;
case SO_KEEPALIVE: mapped = Option::So_KeepAlive; break;
case SO_DONTROUTE: mapped = Option::So_DontRoute; break;
case SO_BROADCAST: mapped = Option::So_Broadcast; break;
case SO_USELOOPBACK: mapped = Option::So_UseLoopback; break;
case SO_LINGER: mapped = Option::So_Linger; break;
case SO_OOBINLINE: mapped = Option::So_OobInline; break;
case SO_SNDBUF: mapped = Option::So_SndBuf; break;
case SO_RCVBUF: mapped = Option::So_RcvBuf; break;
case SO_SNDLOWAT: mapped = Option::So_SndLoWat; break;
case SO_RCVLOWAT: mapped = Option::So_RcvLoWat; break;
case SO_SNDTIMEO: mapped = Option::So_SndTimeo; break;
case SO_RCVTIMEO: mapped = Option::So_RcvTimeo; break;
case SO_ERROR: mapped = Option::So_Error; break;
case SO_TYPE: mapped = Option::So_Type; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket option %d for SOL_SOCKET\n", static_cast<int>(option));
break;
}
break;
case IPPROTO_IP:
switch (option) {
case IP_OPTIONS: mapped = Option::Ip_Options; break;
case IP_HDRINCL: mapped = Option::Ip_HdrIncl; break;
case IP_TOS: mapped = Option::Ip_Tos; break;
case IP_TTL: mapped = Option::Ip_Ttl; break;
case IP_MULTICAST_IF: mapped = Option::Ip_Multicast_If; break;
case IP_MULTICAST_TTL: mapped = Option::Ip_Multicast_Ttl; break;
case IP_MULTICAST_LOOP: mapped = Option::Ip_Multicast_Loop; break;
case IP_ADD_MEMBERSHIP: mapped = Option::Ip_Add_Membership; break;
case IP_DROP_MEMBERSHIP: mapped = Option::Ip_Drop_Membership; break;
case IP_ADD_SOURCE_MEMBERSHIP: mapped = Option::Ip_Add_Source_Membership; break;
case IP_DROP_SOURCE_MEMBERSHIP: mapped = Option::Ip_Drop_Source_Membership; break;
case IP_BLOCK_SOURCE: mapped = Option::Ip_Block_Source; break;
case IP_UNBLOCK_SOURCE: mapped = Option::Ip_Unblock_Source; break;
default:
AMS_SDK_LOG("WARNING: Invalid socket option %d for SOL_IP\n", static_cast<int>(option));
break;
}
break;
case IPPROTO_TCP:
switch (option) {
case TCP_NODELAY: mapped = Option::Tcp_NoDelay; break;
case TCP_MAXSEG: mapped = Option::Tcp_MaxSeg; break;
default:
AMS_SDK_LOG("WARNING: Invalid ams::Socket::Option %d for SOL_TCP\n", static_cast<int>(option));
break;
}
break;
default:
AMS_SDK_LOG("WARNING: Invalid option level %d\n", static_cast<int>(level));
break;
}
return mapped;
}
s32 MapErrnoValue(Errno error) {
s32 mapped = -1;
switch (error) {
case Errno::EIntr: mapped = WSAEINTR; break;
case Errno::EBadf: mapped = WSAEBADF; break;
case Errno::EWouldBlock: mapped = WSAEWOULDBLOCK; break;
case Errno::EAcces: mapped = WSAEACCES; break;
case Errno::EFault: mapped = WSAEFAULT; break;
case Errno::EInval: mapped = WSAEINVAL; break;
case Errno::EMFile: mapped = WSAEMFILE; break;
case Errno::EPipe: mapped = 109; break;
case Errno::ENameTooLong: mapped = WSAENAMETOOLONG; break;
case Errno::ENotEmpty: mapped = WSAENOTEMPTY; break;
case Errno::ELoop: mapped = WSAELOOP; break;
case Errno::ERemote: mapped = WSAEREMOTE; break;
case Errno::EUsers: mapped = WSAEUSERS; break;
case Errno::ENotSock: mapped = WSAENOTSOCK; break;
case Errno::EDestAddrReq: mapped = WSAEDESTADDRREQ; break;
case Errno::EMsgSize: mapped = WSAEMSGSIZE; break;
case Errno::EPrototype: mapped = WSAEPROTOTYPE; break;
case Errno::ENoProtoOpt: mapped = WSAENOPROTOOPT; break;
case Errno::EProtoNoSupport: mapped = WSAEPROTONOSUPPORT; break;
case Errno::ESocktNoSupport: mapped = WSAESOCKTNOSUPPORT; break;
case Errno::EOpNotSupp: mapped = WSAEOPNOTSUPP; break;
case Errno::EPfNoSupport: mapped = WSAEPFNOSUPPORT; break;
case Errno::EAfNoSupport: mapped = WSAEAFNOSUPPORT; break;
case Errno::EAddrInUse: mapped = WSAEADDRINUSE; break;
case Errno::EAddrNotAvail: mapped = WSAEADDRNOTAVAIL; break;
case Errno::ENetDown: mapped = WSAENETDOWN; break;
case Errno::ENetUnreach: mapped = WSAENETUNREACH; break;
case Errno::ENetReset: mapped = WSAENETRESET; break;
case Errno::EConnAborted: mapped = WSAECONNABORTED; break;
case Errno::EConnReset: mapped = WSAECONNRESET; break;
case Errno::ENoBufs: mapped = WSAENOBUFS; break;
case Errno::EIsConn: mapped = WSAEISCONN; break;
case Errno::ENotConn: mapped = WSAENOTCONN; break;
case Errno::EShutDown: mapped = WSAESHUTDOWN; break;
case Errno::ETooManyRefs: mapped = WSAETOOMANYREFS; break;
case Errno::ETimedOut: mapped = WSAETIMEDOUT; break;
case Errno::EConnRefused: mapped = WSAECONNREFUSED; break;
case Errno::EHostDown: mapped = WSAEHOSTDOWN; break;
case Errno::EHostUnreach: mapped = WSAEHOSTUNREACH; break;
case Errno::EAlready: mapped = WSAEALREADY; break;
case Errno::EInProgress: mapped = WSAEINPROGRESS; break;
case Errno::EStale: mapped = WSAESTALE; break;
case Errno::EDQuot: mapped = WSAEDQUOT; break;
case Errno::ECanceled: mapped = WSAECANCELLED; break;
case Errno::EProcLim: mapped = WSAEPROCLIM; break;
default: mapped = static_cast<s32>(error); break;
}
return mapped;
}
Errno MapErrnoValue(s32 error) {
Errno mapped = Errno::ESuccess;
switch (error) {
case WSAEBADF: mapped = Errno::EBadf; break;
case WSAEACCES: mapped = Errno::EAcces; break;
case WSAEFAULT: mapped = Errno::EFault; break;
case WSAEINVAL: mapped = Errno::EInval; break;
case WSAEMFILE: mapped = Errno::EMFile; break;
case WSAEWOULDBLOCK: mapped = Errno::EWouldBlock; break;
case WSAEINPROGRESS: mapped = Errno::EInProgress; break;
case WSAEALREADY: mapped = Errno::EAlready; break;
case WSAENOTSOCK: mapped = Errno::ENotSock; break;
case WSAEDESTADDRREQ: mapped = Errno::EDestAddrReq; break;
case WSAEMSGSIZE: mapped = Errno::EMsgSize; break;
case WSAEPROTOTYPE: mapped = Errno::EPrototype; break;
case WSAENOPROTOOPT: mapped = Errno::ENoProtoOpt; break;
case WSAEPROTONOSUPPORT: mapped = Errno::EProtoNoSupport; break;
case WSAESOCKTNOSUPPORT: mapped = Errno::ESocktNoSupport; break;
case WSAEOPNOTSUPP: mapped = Errno::EOpNotSupp; break;
case WSAEPFNOSUPPORT: mapped = Errno::EPfNoSupport; break;
case WSAEAFNOSUPPORT: mapped = Errno::EAfNoSupport; break;
case WSAEADDRINUSE: mapped = Errno::EAddrInUse; break;
case WSAEADDRNOTAVAIL: mapped = Errno::EAddrNotAvail; break;
case WSAENETDOWN: mapped = Errno::ENetDown; break;
case WSAENETUNREACH: mapped = Errno::ENetUnreach; break;
case WSAENETRESET: mapped = Errno::ENetReset; break;
case WSAECONNABORTED: mapped = Errno::EConnAborted; break;
case WSAECONNRESET: mapped = Errno::EConnReset; break;
case WSAENOBUFS: mapped = Errno::ENoBufs; break;
case WSAEISCONN: mapped = Errno::EIsConn; break;
case WSAENOTCONN: mapped = Errno::ENotConn; break;
case WSAESHUTDOWN: mapped = Errno::EShutDown; break;
case WSAETOOMANYREFS: mapped = Errno::ETooManyRefs; break;
case WSAETIMEDOUT: mapped = Errno::ETimedOut; break;
case WSAECONNREFUSED: mapped = Errno::EConnRefused; break;
case WSAELOOP: mapped = Errno::ELoop; break;
case WSAENAMETOOLONG: mapped = Errno::ENameTooLong; break;
case WSAEHOSTDOWN: mapped = Errno::EHostDown; break;
case WSAEHOSTUNREACH: mapped = Errno::EHostUnreach; break;
case WSAENOTEMPTY: mapped = Errno::ENotEmpty; break;
case WSAEPROCLIM: mapped = Errno::EProcLim; break;
case WSAEUSERS: mapped = Errno::EUsers; break;
case WSAEDQUOT: mapped = Errno::EDQuot; break;
case WSAESTALE: mapped = Errno::EStale; break;
case WSAEREMOTE: mapped = Errno::ERemote; break;
case WSAECANCELLED: mapped = Errno::ECanceled; break;
case WSAEINTR: mapped = Errno::EIntr; break;
case 109: mapped = Errno::EPipe; break;
default: mapped = static_cast<Errno>(error); break;
}
return mapped;
}
void CopyToPlatform(sockaddr_in *dst, const SockAddrIn *src) {
if (src != nullptr) {
dst->sin_family = MapFamilyValue(src->sin_family);
dst->sin_port = src->sin_port;
std::memcpy(std::addressof(dst->sin_addr), std::addressof(src->sin_addr), sizeof(dst->sin_addr));
std::memcpy(dst->sin_zero, src->sin_zero, sizeof(dst->sin_zero));
}
}
void CopyFromPlatform(SockAddrIn *dst, const sockaddr_in *src) {
if (dst != nullptr) {
dst->sin_family = MapFamilyValue(static_cast<s8>(src->sin_family));
dst->sin_port = src->sin_port;
std::memcpy(std::addressof(dst->sin_addr), std::addressof(src->sin_addr), sizeof(dst->sin_addr));
std::memcpy(dst->sin_zero, src->sin_zero, sizeof(dst->sin_zero));
}
}
void CopyToPlatform(timeval *dst, const TimeVal *src) {
if (src != nullptr) {
dst->tv_sec = src->tv_sec;
dst->tv_usec = src->tv_usec;
}
}
void CopyFromPlatform(TimeVal *dst, const timeval *src) {
if (dst != nullptr) {
dst->tv_sec = src->tv_sec;
dst->tv_usec = src->tv_usec;
}
}
void CopyToPlatform(linger *dst, const Linger *src) {
if (src != nullptr) {
dst->l_onoff = static_cast<u_short>(src->l_onoff);
dst->l_linger = static_cast<u_short>(src->l_linger);
}
}
void CopyFromPlatform(Linger *dst, const linger *src) {
if (dst != nullptr) {
dst->l_onoff = src->l_onoff;
dst->l_linger = src->l_linger;
}
}
}

4
libraries/libstratosphere/source/socket/socket_api.cpp
View file

@ -90,6 +90,10 @@ namespace ams::socket {
return impl::Bind(desc, address, len);
}
s32 Connect(s32 desc, const SockAddr *address, SockLenT len) {
return impl::Connect(desc, address, len);
}
s32 GetSockName(s32 desc, SockAddr *out_address, SockLenT *out_addr_len) {
return impl::GetSockName(desc, out_address, out_addr_len);
}