network.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846
  1. // SPDX-FileCopyrightText: Copyright 2020 yuzu Emulator Project
  2. // SPDX-License-Identifier: GPL-2.0-or-later
  3. #include <algorithm>
  4. #include <cstring>
  5. #include <limits>
  6. #include <utility>
  7. #include <vector>
  8. #include "common/error.h"
  9. #ifdef _WIN32
  10. #include <winsock2.h>
  11. #include <ws2tcpip.h>
  12. #elif YUZU_UNIX
  13. #include <arpa/inet.h>
  14. #include <errno.h>
  15. #include <fcntl.h>
  16. #include <netdb.h>
  17. #include <netinet/in.h>
  18. #include <poll.h>
  19. #include <sys/socket.h>
  20. #include <unistd.h>
  21. #else
  22. #error "Unimplemented platform"
  23. #endif
  24. #include "common/assert.h"
  25. #include "common/common_types.h"
  26. #include "common/expected.h"
  27. #include "common/logging/log.h"
  28. #include "common/settings.h"
  29. #include "core/internal_network/network.h"
  30. #include "core/internal_network/network_interface.h"
  31. #include "core/internal_network/sockets.h"
  32. #include "network/network.h"
  33. namespace Network {
  34. namespace {
  35. enum class CallType {
  36. Send,
  37. Other,
  38. };
  39. #ifdef _WIN32
  40. using socklen_t = int;
  41. void Initialize() {
  42. WSADATA wsa_data;
  43. (void)WSAStartup(MAKEWORD(2, 2), &wsa_data);
  44. }
  45. void Finalize() {
  46. WSACleanup();
  47. }
  48. sockaddr TranslateFromSockAddrIn(SockAddrIn input) {
  49. sockaddr_in result;
  50. #if YUZU_UNIX
  51. result.sin_len = sizeof(result);
  52. #endif
  53. switch (static_cast<Domain>(input.family)) {
  54. case Domain::INET:
  55. result.sin_family = AF_INET;
  56. break;
  57. default:
  58. UNIMPLEMENTED_MSG("Unhandled sockaddr family={}", input.family);
  59. result.sin_family = AF_INET;
  60. break;
  61. }
  62. result.sin_port = htons(input.portno);
  63. auto& ip = result.sin_addr.S_un.S_un_b;
  64. ip.s_b1 = input.ip[0];
  65. ip.s_b2 = input.ip[1];
  66. ip.s_b3 = input.ip[2];
  67. ip.s_b4 = input.ip[3];
  68. sockaddr addr;
  69. std::memcpy(&addr, &result, sizeof(addr));
  70. return addr;
  71. }
  72. LINGER MakeLinger(bool enable, u32 linger_value) {
  73. ASSERT(linger_value <= std::numeric_limits<u_short>::max());
  74. LINGER value;
  75. value.l_onoff = enable ? 1 : 0;
  76. value.l_linger = static_cast<u_short>(linger_value);
  77. return value;
  78. }
  79. bool EnableNonBlock(SOCKET fd, bool enable) {
  80. u_long value = enable ? 1 : 0;
  81. return ioctlsocket(fd, FIONBIO, &value) != SOCKET_ERROR;
  82. }
  83. Errno TranslateNativeError(int e, CallType call_type = CallType::Other) {
  84. switch (e) {
  85. case 0:
  86. return Errno::SUCCESS;
  87. case WSAEBADF:
  88. return Errno::BADF;
  89. case WSAEINVAL:
  90. return Errno::INVAL;
  91. case WSAEMFILE:
  92. return Errno::MFILE;
  93. case WSAENOTCONN:
  94. return Errno::NOTCONN;
  95. case WSAEWOULDBLOCK:
  96. return Errno::AGAIN;
  97. case WSAECONNREFUSED:
  98. return Errno::CONNREFUSED;
  99. case WSAECONNABORTED:
  100. if (call_type == CallType::Send) {
  101. // Winsock yields WSAECONNABORTED from `send` in situations where Unix
  102. // systems, and actual Switches, yield EPIPE.
  103. return Errno::PIPE;
  104. } else {
  105. return Errno::CONNABORTED;
  106. }
  107. case WSAECONNRESET:
  108. return Errno::CONNRESET;
  109. case WSAEHOSTUNREACH:
  110. return Errno::HOSTUNREACH;
  111. case WSAENETDOWN:
  112. return Errno::NETDOWN;
  113. case WSAENETUNREACH:
  114. return Errno::NETUNREACH;
  115. case WSAEMSGSIZE:
  116. return Errno::MSGSIZE;
  117. case WSAETIMEDOUT:
  118. return Errno::TIMEDOUT;
  119. case WSAEINPROGRESS:
  120. return Errno::INPROGRESS;
  121. default:
  122. UNIMPLEMENTED_MSG("Unimplemented errno={}", e);
  123. return Errno::OTHER;
  124. }
  125. }
  126. #elif YUZU_UNIX // ^ _WIN32 v YUZU_UNIX
  127. using SOCKET = int;
  128. using WSAPOLLFD = pollfd;
  129. using ULONG = u64;
  130. constexpr SOCKET SOCKET_ERROR = -1;
  131. constexpr int SD_RECEIVE = SHUT_RD;
  132. constexpr int SD_SEND = SHUT_WR;
  133. constexpr int SD_BOTH = SHUT_RDWR;
  134. void Initialize() {}
  135. void Finalize() {}
  136. sockaddr TranslateFromSockAddrIn(SockAddrIn input) {
  137. sockaddr_in result;
  138. switch (static_cast<Domain>(input.family)) {
  139. case Domain::INET:
  140. result.sin_family = AF_INET;
  141. break;
  142. default:
  143. UNIMPLEMENTED_MSG("Unhandled sockaddr family={}", input.family);
  144. result.sin_family = AF_INET;
  145. break;
  146. }
  147. result.sin_port = htons(input.portno);
  148. result.sin_addr.s_addr = input.ip[0] | input.ip[1] << 8 | input.ip[2] << 16 | input.ip[3] << 24;
  149. sockaddr addr;
  150. std::memcpy(&addr, &result, sizeof(addr));
  151. return addr;
  152. }
  153. int WSAPoll(WSAPOLLFD* fds, ULONG nfds, int timeout) {
  154. return poll(fds, static_cast<nfds_t>(nfds), timeout);
  155. }
  156. int closesocket(SOCKET fd) {
  157. return close(fd);
  158. }
  159. linger MakeLinger(bool enable, u32 linger_value) {
  160. linger value;
  161. value.l_onoff = enable ? 1 : 0;
  162. value.l_linger = linger_value;
  163. return value;
  164. }
  165. bool EnableNonBlock(int fd, bool enable) {
  166. int flags = fcntl(fd, F_GETFL);
  167. if (flags == -1) {
  168. return false;
  169. }
  170. if (enable) {
  171. flags |= O_NONBLOCK;
  172. } else {
  173. flags &= ~O_NONBLOCK;
  174. }
  175. return fcntl(fd, F_SETFL, flags) == 0;
  176. }
  177. Errno TranslateNativeError(int e, CallType call_type = CallType::Other) {
  178. switch (e) {
  179. case 0:
  180. return Errno::SUCCESS;
  181. case EBADF:
  182. return Errno::BADF;
  183. case EINVAL:
  184. return Errno::INVAL;
  185. case EMFILE:
  186. return Errno::MFILE;
  187. case EPIPE:
  188. return Errno::PIPE;
  189. case ECONNABORTED:
  190. return Errno::CONNABORTED;
  191. case ENOTCONN:
  192. return Errno::NOTCONN;
  193. case EAGAIN:
  194. return Errno::AGAIN;
  195. case ECONNREFUSED:
  196. return Errno::CONNREFUSED;
  197. case ECONNRESET:
  198. return Errno::CONNRESET;
  199. case EHOSTUNREACH:
  200. return Errno::HOSTUNREACH;
  201. case ENETDOWN:
  202. return Errno::NETDOWN;
  203. case ENETUNREACH:
  204. return Errno::NETUNREACH;
  205. case EMSGSIZE:
  206. return Errno::MSGSIZE;
  207. case ETIMEDOUT:
  208. return Errno::TIMEDOUT;
  209. case EINPROGRESS:
  210. return Errno::INPROGRESS;
  211. default:
  212. UNIMPLEMENTED_MSG("Unimplemented errno={} ({})", e, strerror(e));
  213. return Errno::OTHER;
  214. }
  215. }
  216. #endif
  217. Errno GetAndLogLastError(CallType call_type = CallType::Other) {
  218. #ifdef _WIN32
  219. int e = WSAGetLastError();
  220. #else
  221. int e = errno;
  222. #endif
  223. const Errno err = TranslateNativeError(e, call_type);
  224. if (err == Errno::AGAIN || err == Errno::TIMEDOUT || err == Errno::INPROGRESS) {
  225. // These happen during normal operation, so only log them at debug level.
  226. LOG_DEBUG(Network, "Socket operation error: {}", Common::NativeErrorToString(e));
  227. return err;
  228. }
  229. LOG_ERROR(Network, "Socket operation error: {}", Common::NativeErrorToString(e));
  230. return err;
  231. }
  232. GetAddrInfoError TranslateGetAddrInfoErrorFromNative(int gai_err) {
  233. switch (gai_err) {
  234. case 0:
  235. return GetAddrInfoError::SUCCESS;
  236. #ifdef EAI_ADDRFAMILY
  237. case EAI_ADDRFAMILY:
  238. return GetAddrInfoError::ADDRFAMILY;
  239. #endif
  240. case EAI_AGAIN:
  241. return GetAddrInfoError::AGAIN;
  242. case EAI_BADFLAGS:
  243. return GetAddrInfoError::BADFLAGS;
  244. case EAI_FAIL:
  245. return GetAddrInfoError::FAIL;
  246. case EAI_FAMILY:
  247. return GetAddrInfoError::FAMILY;
  248. case EAI_MEMORY:
  249. return GetAddrInfoError::MEMORY;
  250. case EAI_NONAME:
  251. return GetAddrInfoError::NONAME;
  252. case EAI_SERVICE:
  253. return GetAddrInfoError::SERVICE;
  254. case EAI_SOCKTYPE:
  255. return GetAddrInfoError::SOCKTYPE;
  256. // These codes may not be defined on all systems:
  257. #ifdef EAI_SYSTEM
  258. case EAI_SYSTEM:
  259. return GetAddrInfoError::SYSTEM;
  260. #endif
  261. #ifdef EAI_BADHINTS
  262. case EAI_BADHINTS:
  263. return GetAddrInfoError::BADHINTS;
  264. #endif
  265. #ifdef EAI_PROTOCOL
  266. case EAI_PROTOCOL:
  267. return GetAddrInfoError::PROTOCOL;
  268. #endif
  269. #ifdef EAI_OVERFLOW
  270. case EAI_OVERFLOW:
  271. return GetAddrInfoError::OVERFLOW_;
  272. #endif
  273. default:
  274. #ifdef EAI_NODATA
  275. // This can't be a case statement because it would create a duplicate
  276. // case on Windows where EAI_NODATA is an alias for EAI_NONAME.
  277. if (gai_err == EAI_NODATA) {
  278. return GetAddrInfoError::NODATA;
  279. }
  280. #endif
  281. return GetAddrInfoError::OTHER;
  282. }
  283. }
  284. Domain TranslateDomainFromNative(int domain) {
  285. switch (domain) {
  286. case 0:
  287. return Domain::Unspecified;
  288. case AF_INET:
  289. return Domain::INET;
  290. default:
  291. UNIMPLEMENTED_MSG("Unhandled domain={}", domain);
  292. return Domain::INET;
  293. }
  294. }
  295. int TranslateDomainToNative(Domain domain) {
  296. switch (domain) {
  297. case Domain::Unspecified:
  298. return 0;
  299. case Domain::INET:
  300. return AF_INET;
  301. default:
  302. UNIMPLEMENTED_MSG("Unimplemented domain={}", domain);
  303. return 0;
  304. }
  305. }
  306. Type TranslateTypeFromNative(int type) {
  307. switch (type) {
  308. case 0:
  309. return Type::Unspecified;
  310. case SOCK_STREAM:
  311. return Type::STREAM;
  312. case SOCK_DGRAM:
  313. return Type::DGRAM;
  314. case SOCK_RAW:
  315. return Type::RAW;
  316. case SOCK_SEQPACKET:
  317. return Type::SEQPACKET;
  318. default:
  319. UNIMPLEMENTED_MSG("Unimplemented type={}", type);
  320. return Type::STREAM;
  321. }
  322. }
  323. int TranslateTypeToNative(Type type) {
  324. switch (type) {
  325. case Type::Unspecified:
  326. return 0;
  327. case Type::STREAM:
  328. return SOCK_STREAM;
  329. case Type::DGRAM:
  330. return SOCK_DGRAM;
  331. case Type::RAW:
  332. return SOCK_RAW;
  333. default:
  334. UNIMPLEMENTED_MSG("Unimplemented type={}", type);
  335. return 0;
  336. }
  337. }
  338. Protocol TranslateProtocolFromNative(int protocol) {
  339. switch (protocol) {
  340. case 0:
  341. return Protocol::Unspecified;
  342. case IPPROTO_TCP:
  343. return Protocol::TCP;
  344. case IPPROTO_UDP:
  345. return Protocol::UDP;
  346. default:
  347. UNIMPLEMENTED_MSG("Unimplemented protocol={}", protocol);
  348. return Protocol::Unspecified;
  349. }
  350. }
  351. int TranslateProtocolToNative(Protocol protocol) {
  352. switch (protocol) {
  353. case Protocol::Unspecified:
  354. return 0;
  355. case Protocol::TCP:
  356. return IPPROTO_TCP;
  357. case Protocol::UDP:
  358. return IPPROTO_UDP;
  359. default:
  360. UNIMPLEMENTED_MSG("Unimplemented protocol={}", protocol);
  361. return 0;
  362. }
  363. }
  364. SockAddrIn TranslateToSockAddrIn(sockaddr_in input, size_t input_len) {
  365. SockAddrIn result;
  366. result.family = TranslateDomainFromNative(input.sin_family);
  367. result.portno = ntohs(input.sin_port);
  368. result.ip = TranslateIPv4(input.sin_addr);
  369. return result;
  370. }
  371. short TranslatePollEvents(PollEvents events) {
  372. short result = 0;
  373. const auto translate = [&result, &events](PollEvents guest, short host) {
  374. if (True(events & guest)) {
  375. events &= ~guest;
  376. result |= host;
  377. }
  378. };
  379. translate(PollEvents::In, POLLIN);
  380. translate(PollEvents::Pri, POLLPRI);
  381. translate(PollEvents::Out, POLLOUT);
  382. translate(PollEvents::Err, POLLERR);
  383. translate(PollEvents::Hup, POLLHUP);
  384. translate(PollEvents::Nval, POLLNVAL);
  385. translate(PollEvents::RdNorm, POLLRDNORM);
  386. translate(PollEvents::RdBand, POLLRDBAND);
  387. translate(PollEvents::WrBand, POLLWRBAND);
  388. #ifdef _WIN32
  389. short allowed_events = POLLRDBAND | POLLRDNORM | POLLWRNORM;
  390. // Unlike poll on other OSes, WSAPoll will complain if any other flags are set on input.
  391. if (result & ~allowed_events) {
  392. LOG_DEBUG(Network,
  393. "Removing WSAPoll input events 0x{:x} because Windows doesn't support them",
  394. result & ~allowed_events);
  395. }
  396. result &= allowed_events;
  397. #endif
  398. UNIMPLEMENTED_IF_MSG((u16)events != 0, "Unhandled guest events=0x{:x}", (u16)events);
  399. return result;
  400. }
  401. PollEvents TranslatePollRevents(short revents) {
  402. PollEvents result{};
  403. const auto translate = [&result, &revents](short host, PollEvents guest) {
  404. if ((revents & host) != 0) {
  405. revents &= static_cast<short>(~host);
  406. result |= guest;
  407. }
  408. };
  409. translate(POLLIN, PollEvents::In);
  410. translate(POLLPRI, PollEvents::Pri);
  411. translate(POLLOUT, PollEvents::Out);
  412. translate(POLLERR, PollEvents::Err);
  413. translate(POLLHUP, PollEvents::Hup);
  414. translate(POLLNVAL, PollEvents::Nval);
  415. translate(POLLRDNORM, PollEvents::RdNorm);
  416. translate(POLLRDBAND, PollEvents::RdBand);
  417. translate(POLLWRBAND, PollEvents::WrBand);
  418. UNIMPLEMENTED_IF_MSG(revents != 0, "Unhandled host revents=0x{:x}", revents);
  419. return result;
  420. }
  421. } // Anonymous namespace
  422. NetworkInstance::NetworkInstance() {
  423. Initialize();
  424. }
  425. NetworkInstance::~NetworkInstance() {
  426. Finalize();
  427. }
  428. std::optional<IPv4Address> GetHostIPv4Address() {
  429. const auto network_interface = Network::GetSelectedNetworkInterface();
  430. if (!network_interface.has_value()) {
  431. // Only print the error once to avoid log spam
  432. static bool print_error = true;
  433. if (print_error) {
  434. LOG_ERROR(Network, "GetSelectedNetworkInterface returned no interface");
  435. print_error = false;
  436. }
  437. return {};
  438. }
  439. return TranslateIPv4(network_interface->ip_address);
  440. }
  441. std::string IPv4AddressToString(IPv4Address ip_addr) {
  442. std::array<char, INET_ADDRSTRLEN> buf = {};
  443. ASSERT(inet_ntop(AF_INET, &ip_addr, buf.data(), sizeof(buf)) == buf.data());
  444. return std::string(buf.data());
  445. }
  446. u32 IPv4AddressToInteger(IPv4Address ip_addr) {
  447. return static_cast<u32>(ip_addr[0]) << 24 | static_cast<u32>(ip_addr[1]) << 16 |
  448. static_cast<u32>(ip_addr[2]) << 8 | static_cast<u32>(ip_addr[3]);
  449. }
  450. Common::Expected<std::vector<AddrInfo>, GetAddrInfoError> GetAddressInfo(
  451. const std::string& host, const std::optional<std::string>& service) {
  452. addrinfo hints{};
  453. hints.ai_family = AF_INET; // Switch only supports IPv4.
  454. addrinfo* addrinfo;
  455. s32 gai_err = getaddrinfo(host.c_str(), service.has_value() ? service->c_str() : nullptr,
  456. &hints, &addrinfo);
  457. if (gai_err != 0) {
  458. return Common::Unexpected(TranslateGetAddrInfoErrorFromNative(gai_err));
  459. }
  460. std::vector<AddrInfo> ret;
  461. for (auto* current = addrinfo; current; current = current->ai_next) {
  462. // We should only get AF_INET results due to the hints value.
  463. ASSERT_OR_EXECUTE(addrinfo->ai_family == AF_INET &&
  464. addrinfo->ai_addrlen == sizeof(sockaddr_in),
  465. continue;);
  466. AddrInfo& out = ret.emplace_back();
  467. out.family = TranslateDomainFromNative(current->ai_family);
  468. out.socket_type = TranslateTypeFromNative(current->ai_socktype);
  469. out.protocol = TranslateProtocolFromNative(current->ai_protocol);
  470. out.addr = TranslateToSockAddrIn(*reinterpret_cast<sockaddr_in*>(current->ai_addr),
  471. current->ai_addrlen);
  472. if (current->ai_canonname != nullptr) {
  473. out.canon_name = current->ai_canonname;
  474. }
  475. }
  476. freeaddrinfo(addrinfo);
  477. return ret;
  478. }
  479. std::pair<s32, Errno> Poll(std::vector<PollFD>& pollfds, s32 timeout) {
  480. const size_t num = pollfds.size();
  481. std::vector<WSAPOLLFD> host_pollfds(pollfds.size());
  482. std::transform(pollfds.begin(), pollfds.end(), host_pollfds.begin(), [](PollFD fd) {
  483. WSAPOLLFD result;
  484. result.fd = fd.socket->GetFD();
  485. result.events = TranslatePollEvents(fd.events);
  486. result.revents = 0;
  487. return result;
  488. });
  489. const int result = WSAPoll(host_pollfds.data(), static_cast<ULONG>(num), timeout);
  490. if (result == 0) {
  491. ASSERT(std::all_of(host_pollfds.begin(), host_pollfds.end(),
  492. [](WSAPOLLFD fd) { return fd.revents == 0; }));
  493. return {0, Errno::SUCCESS};
  494. }
  495. for (size_t i = 0; i < num; ++i) {
  496. pollfds[i].revents = TranslatePollRevents(host_pollfds[i].revents);
  497. }
  498. if (result > 0) {
  499. return {result, Errno::SUCCESS};
  500. }
  501. ASSERT(result == SOCKET_ERROR);
  502. return {-1, GetAndLogLastError()};
  503. }
  504. Socket::~Socket() {
  505. if (fd == INVALID_SOCKET) {
  506. return;
  507. }
  508. (void)closesocket(fd);
  509. fd = INVALID_SOCKET;
  510. }
  511. Socket::Socket(Socket&& rhs) noexcept {
  512. fd = std::exchange(rhs.fd, INVALID_SOCKET);
  513. }
  514. template <typename T>
  515. std::pair<T, Errno> Socket::GetSockOpt(SOCKET fd_so, int option) {
  516. T value{};
  517. socklen_t len = sizeof(value);
  518. const int result = getsockopt(fd_so, SOL_SOCKET, option, reinterpret_cast<char*>(&value), &len);
  519. if (result != SOCKET_ERROR) {
  520. ASSERT(len == sizeof(value));
  521. return {value, Errno::SUCCESS};
  522. }
  523. return {value, GetAndLogLastError()};
  524. }
  525. template <typename T>
  526. Errno Socket::SetSockOpt(SOCKET fd_so, int option, T value) {
  527. const int result =
  528. setsockopt(fd_so, SOL_SOCKET, option, reinterpret_cast<const char*>(&value), sizeof(value));
  529. if (result != SOCKET_ERROR) {
  530. return Errno::SUCCESS;
  531. }
  532. return GetAndLogLastError();
  533. }
  534. Errno Socket::Initialize(Domain domain, Type type, Protocol protocol) {
  535. fd = socket(TranslateDomainToNative(domain), TranslateTypeToNative(type),
  536. TranslateProtocolToNative(protocol));
  537. if (fd != INVALID_SOCKET) {
  538. return Errno::SUCCESS;
  539. }
  540. return GetAndLogLastError();
  541. }
  542. std::pair<SocketBase::AcceptResult, Errno> Socket::Accept() {
  543. sockaddr_in addr;
  544. socklen_t addrlen = sizeof(addr);
  545. const SOCKET new_socket = accept(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen);
  546. if (new_socket == INVALID_SOCKET) {
  547. return {AcceptResult{}, GetAndLogLastError()};
  548. }
  549. AcceptResult result{
  550. .socket = std::make_unique<Socket>(new_socket),
  551. .sockaddr_in = TranslateToSockAddrIn(addr, addrlen),
  552. };
  553. return {std::move(result), Errno::SUCCESS};
  554. }
  555. Errno Socket::Connect(SockAddrIn addr_in) {
  556. const sockaddr host_addr_in = TranslateFromSockAddrIn(addr_in);
  557. if (connect(fd, &host_addr_in, sizeof(host_addr_in)) != SOCKET_ERROR) {
  558. return Errno::SUCCESS;
  559. }
  560. return GetAndLogLastError();
  561. }
  562. std::pair<SockAddrIn, Errno> Socket::GetPeerName() {
  563. sockaddr_in addr;
  564. socklen_t addrlen = sizeof(addr);
  565. if (getpeername(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen) == SOCKET_ERROR) {
  566. return {SockAddrIn{}, GetAndLogLastError()};
  567. }
  568. return {TranslateToSockAddrIn(addr, addrlen), Errno::SUCCESS};
  569. }
  570. std::pair<SockAddrIn, Errno> Socket::GetSockName() {
  571. sockaddr_in addr;
  572. socklen_t addrlen = sizeof(addr);
  573. if (getsockname(fd, reinterpret_cast<sockaddr*>(&addr), &addrlen) == SOCKET_ERROR) {
  574. return {SockAddrIn{}, GetAndLogLastError()};
  575. }
  576. return {TranslateToSockAddrIn(addr, addrlen), Errno::SUCCESS};
  577. }
  578. Errno Socket::Bind(SockAddrIn addr) {
  579. const sockaddr addr_in = TranslateFromSockAddrIn(addr);
  580. if (bind(fd, &addr_in, sizeof(addr_in)) != SOCKET_ERROR) {
  581. return Errno::SUCCESS;
  582. }
  583. return GetAndLogLastError();
  584. }
  585. Errno Socket::Listen(s32 backlog) {
  586. if (listen(fd, backlog) != SOCKET_ERROR) {
  587. return Errno::SUCCESS;
  588. }
  589. return GetAndLogLastError();
  590. }
  591. Errno Socket::Shutdown(ShutdownHow how) {
  592. int host_how = 0;
  593. switch (how) {
  594. case ShutdownHow::RD:
  595. host_how = SD_RECEIVE;
  596. break;
  597. case ShutdownHow::WR:
  598. host_how = SD_SEND;
  599. break;
  600. case ShutdownHow::RDWR:
  601. host_how = SD_BOTH;
  602. break;
  603. default:
  604. UNIMPLEMENTED_MSG("Unimplemented flag how={}", how);
  605. return Errno::SUCCESS;
  606. }
  607. if (shutdown(fd, host_how) != SOCKET_ERROR) {
  608. return Errno::SUCCESS;
  609. }
  610. return GetAndLogLastError();
  611. }
  612. std::pair<s32, Errno> Socket::Recv(int flags, std::span<u8> message) {
  613. ASSERT(flags == 0);
  614. ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
  615. const auto result =
  616. recv(fd, reinterpret_cast<char*>(message.data()), static_cast<int>(message.size()), 0);
  617. if (result != SOCKET_ERROR) {
  618. return {static_cast<s32>(result), Errno::SUCCESS};
  619. }
  620. return {-1, GetAndLogLastError()};
  621. }
  622. std::pair<s32, Errno> Socket::RecvFrom(int flags, std::span<u8> message, SockAddrIn* addr) {
  623. ASSERT(flags == 0);
  624. ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
  625. sockaddr_in addr_in{};
  626. socklen_t addrlen = sizeof(addr_in);
  627. socklen_t* const p_addrlen = addr ? &addrlen : nullptr;
  628. sockaddr* const p_addr_in = addr ? reinterpret_cast<sockaddr*>(&addr_in) : nullptr;
  629. const auto result = recvfrom(fd, reinterpret_cast<char*>(message.data()),
  630. static_cast<int>(message.size()), 0, p_addr_in, p_addrlen);
  631. if (result != SOCKET_ERROR) {
  632. if (addr) {
  633. *addr = TranslateToSockAddrIn(addr_in, addrlen);
  634. }
  635. return {static_cast<s32>(result), Errno::SUCCESS};
  636. }
  637. return {-1, GetAndLogLastError()};
  638. }
  639. std::pair<s32, Errno> Socket::Send(std::span<const u8> message, int flags) {
  640. ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max()));
  641. ASSERT(flags == 0);
  642. int native_flags = 0;
  643. #if YUZU_UNIX
  644. native_flags |= MSG_NOSIGNAL; // do not send us SIGPIPE
  645. #endif
  646. const auto result = send(fd, reinterpret_cast<const char*>(message.data()),
  647. static_cast<int>(message.size()), native_flags);
  648. if (result != SOCKET_ERROR) {
  649. return {static_cast<s32>(result), Errno::SUCCESS};
  650. }
  651. return {-1, GetAndLogLastError(CallType::Send)};
  652. }
  653. std::pair<s32, Errno> Socket::SendTo(u32 flags, std::span<const u8> message,
  654. const SockAddrIn* addr) {
  655. ASSERT(flags == 0);
  656. const sockaddr* to = nullptr;
  657. const int to_len = addr ? sizeof(sockaddr) : 0;
  658. sockaddr host_addr_in;
  659. if (addr) {
  660. host_addr_in = TranslateFromSockAddrIn(*addr);
  661. to = &host_addr_in;
  662. }
  663. const auto result = sendto(fd, reinterpret_cast<const char*>(message.data()),
  664. static_cast<int>(message.size()), 0, to, to_len);
  665. if (result != SOCKET_ERROR) {
  666. return {static_cast<s32>(result), Errno::SUCCESS};
  667. }
  668. return {-1, GetAndLogLastError(CallType::Send)};
  669. }
  670. Errno Socket::Close() {
  671. [[maybe_unused]] const int result = closesocket(fd);
  672. ASSERT(result == 0);
  673. fd = INVALID_SOCKET;
  674. return Errno::SUCCESS;
  675. }
  676. std::pair<Errno, Errno> Socket::GetPendingError() {
  677. auto [pending_err, getsockopt_err] = GetSockOpt<int>(fd, SO_ERROR);
  678. return {TranslateNativeError(pending_err), getsockopt_err};
  679. }
  680. Errno Socket::SetLinger(bool enable, u32 linger) {
  681. return SetSockOpt(fd, SO_LINGER, MakeLinger(enable, linger));
  682. }
  683. Errno Socket::SetReuseAddr(bool enable) {
  684. return SetSockOpt<u32>(fd, SO_REUSEADDR, enable ? 1 : 0);
  685. }
  686. Errno Socket::SetKeepAlive(bool enable) {
  687. return SetSockOpt<u32>(fd, SO_KEEPALIVE, enable ? 1 : 0);
  688. }
  689. Errno Socket::SetBroadcast(bool enable) {
  690. return SetSockOpt<u32>(fd, SO_BROADCAST, enable ? 1 : 0);
  691. }
  692. Errno Socket::SetSndBuf(u32 value) {
  693. return SetSockOpt(fd, SO_SNDBUF, value);
  694. }
  695. Errno Socket::SetRcvBuf(u32 value) {
  696. return SetSockOpt(fd, SO_RCVBUF, value);
  697. }
  698. Errno Socket::SetSndTimeo(u32 value) {
  699. return SetSockOpt(fd, SO_SNDTIMEO, value);
  700. }
  701. Errno Socket::SetRcvTimeo(u32 value) {
  702. return SetSockOpt(fd, SO_RCVTIMEO, value);
  703. }
  704. Errno Socket::SetNonBlock(bool enable) {
  705. if (EnableNonBlock(fd, enable)) {
  706. return Errno::SUCCESS;
  707. }
  708. return GetAndLogLastError();
  709. }
  710. bool Socket::IsOpened() const {
  711. return fd != INVALID_SOCKET;
  712. }
  713. void Socket::HandleProxyPacket(const ProxyPacket& packet) {
  714. LOG_WARNING(Network, "ProxyPacket received, but not in Proxy mode!");
  715. }
  716. } // namespace Network