soc_u.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762
  1. // Copyright 2014 Citra Emulator Project
  2. // Licensed under GPLv2 or any later version
  3. // Refer to the license.txt file included.
  4. #include <algorithm>
  5. #include <cstring>
  6. #include <unordered_map>
  7. #include "common/assert.h"
  8. #include "common/bit_field.h"
  9. #include "common/common_types.h"
  10. #include "common/logging/log.h"
  11. #include "common/scope_exit.h"
  12. #include "core/hle/kernel/session.h"
  13. #include "core/hle/result.h"
  14. #include "core/hle/service/soc_u.h"
  15. #include "core/memory.h"
  16. #ifdef _WIN32
  17. #include <winsock2.h>
  18. #include <ws2tcpip.h>
  19. // MinGW does not define several errno constants
  20. #ifndef _MSC_VER
  21. #define EBADMSG 104
  22. #define ENODATA 120
  23. #define ENOMSG 122
  24. #define ENOSR 124
  25. #define ENOSTR 125
  26. #define ETIME 137
  27. #define EIDRM 2001
  28. #define ENOLINK 2002
  29. #endif // _MSC_VER
  30. #else
  31. #include <cerrno>
  32. #include <fcntl.h>
  33. #include <netinet/in.h>
  34. #include <netdb.h>
  35. #include <poll.h>
  36. #include <sys/socket.h>
  37. #include <unistd.h>
  38. #endif
  39. #ifdef _WIN32
  40. # define WSAEAGAIN WSAEWOULDBLOCK
  41. # define WSAEMULTIHOP -1 // Invalid dummy value
  42. # define ERRNO(x) WSA##x
  43. # define GET_ERRNO WSAGetLastError()
  44. # define poll(x, y, z) WSAPoll(x, y, z);
  45. #else
  46. # define ERRNO(x) x
  47. # define GET_ERRNO errno
  48. # define closesocket(x) close(x)
  49. #endif
  50. static const s32 SOCKET_ERROR_VALUE = -1;
  51. ////////////////////////////////////////////////////////////////////////////////////////////////////
  52. // Namespace SOC_U
  53. namespace SOC_U {
  54. /// Holds the translation from system network errors to 3DS network errors
  55. static const std::unordered_map<int, int> error_map = { {
  56. { E2BIG, 1 },
  57. { ERRNO(EACCES), 2 },
  58. { ERRNO(EADDRINUSE), 3 },
  59. { ERRNO(EADDRNOTAVAIL), 4 },
  60. { ERRNO(EAFNOSUPPORT), 5 },
  61. { ERRNO(EAGAIN), 6 },
  62. { ERRNO(EALREADY), 7 },
  63. { ERRNO(EBADF), 8 },
  64. { EBADMSG, 9 },
  65. { EBUSY, 10 },
  66. { ECANCELED, 11 },
  67. { ECHILD, 12 },
  68. { ERRNO(ECONNABORTED), 13 },
  69. { ERRNO(ECONNREFUSED), 14 },
  70. { ERRNO(ECONNRESET), 15 },
  71. { EDEADLK, 16 },
  72. { ERRNO(EDESTADDRREQ), 17 },
  73. { EDOM, 18 },
  74. { ERRNO(EDQUOT), 19 },
  75. { EEXIST, 20 },
  76. { ERRNO(EFAULT), 21 },
  77. { EFBIG, 22 },
  78. { ERRNO(EHOSTUNREACH), 23 },
  79. { EIDRM, 24 },
  80. { EILSEQ, 25 },
  81. { ERRNO(EINPROGRESS), 26 },
  82. { ERRNO(EINTR), 27 },
  83. { ERRNO(EINVAL), 28 },
  84. { EIO, 29 },
  85. { ERRNO(EISCONN), 30 },
  86. { EISDIR, 31 },
  87. { ERRNO(ELOOP), 32 },
  88. { ERRNO(EMFILE), 33 },
  89. { EMLINK, 34 },
  90. { ERRNO(EMSGSIZE), 35 },
  91. { ERRNO(EMULTIHOP), 36 },
  92. { ERRNO(ENAMETOOLONG), 37 },
  93. { ERRNO(ENETDOWN), 38 },
  94. { ERRNO(ENETRESET), 39 },
  95. { ERRNO(ENETUNREACH), 40 },
  96. { ENFILE, 41 },
  97. { ERRNO(ENOBUFS), 42 },
  98. { ENODATA, 43 },
  99. { ENODEV, 44 },
  100. { ENOENT, 45 },
  101. { ENOEXEC, 46 },
  102. { ENOLCK, 47 },
  103. { ENOLINK, 48 },
  104. { ENOMEM, 49 },
  105. { ENOMSG, 50 },
  106. { ERRNO(ENOPROTOOPT), 51 },
  107. { ENOSPC, 52 },
  108. { ENOSR, 53 },
  109. { ENOSTR, 54 },
  110. { ENOSYS, 55 },
  111. { ERRNO(ENOTCONN), 56 },
  112. { ENOTDIR, 57 },
  113. { ERRNO(ENOTEMPTY), 58 },
  114. { ERRNO(ENOTSOCK), 59 },
  115. { ENOTSUP, 60 },
  116. { ENOTTY, 61 },
  117. { ENXIO, 62 },
  118. { ERRNO(EOPNOTSUPP), 63 },
  119. { EOVERFLOW, 64 },
  120. { EPERM, 65 },
  121. { EPIPE, 66 },
  122. { EPROTO, 67 },
  123. { ERRNO(EPROTONOSUPPORT), 68 },
  124. { ERRNO(EPROTOTYPE), 69 },
  125. { ERANGE, 70 },
  126. { EROFS, 71 },
  127. { ESPIPE, 72 },
  128. { ESRCH, 73 },
  129. { ERRNO(ESTALE), 74 },
  130. { ETIME, 75 },
  131. { ERRNO(ETIMEDOUT), 76 }
  132. }};
  133. /// Converts a network error from platform-specific to 3ds-specific
  134. static int TranslateError(int error) {
  135. auto found = error_map.find(error);
  136. if (found != error_map.end())
  137. return -found->second;
  138. return error;
  139. }
  140. /// Holds information about a particular socket
  141. struct SocketHolder {
  142. u32 socket_fd; ///< The socket descriptor
  143. bool blocking; ///< Whether the socket is blocking or not, it is only read on Windows.
  144. };
  145. /// Structure to represent the 3ds' pollfd structure, which is different than most implementations
  146. struct CTRPollFD {
  147. u32 fd; ///< Socket handle
  148. union Events {
  149. u32 hex; ///< The complete value formed by the flags
  150. BitField<0, 1, u32> pollin;
  151. BitField<1, 1, u32> pollpri;
  152. BitField<2, 1, u32> pollhup;
  153. BitField<3, 1, u32> pollerr;
  154. BitField<4, 1, u32> pollout;
  155. BitField<5, 1, u32> pollnval;
  156. Events& operator=(const Events& other) {
  157. hex = other.hex;
  158. return *this;
  159. }
  160. /// Translates the resulting events of a Poll operation from platform-specific to 3ds specific
  161. static Events TranslateTo3DS(u32 input_event) {
  162. Events ev = {};
  163. if (input_event & POLLIN)
  164. ev.pollin = 1;
  165. if (input_event & POLLPRI)
  166. ev.pollpri = 1;
  167. if (input_event & POLLHUP)
  168. ev.pollhup = 1;
  169. if (input_event & POLLERR)
  170. ev.pollerr = 1;
  171. if (input_event & POLLOUT)
  172. ev.pollout = 1;
  173. if (input_event & POLLNVAL)
  174. ev.pollnval = 1;
  175. return ev;
  176. }
  177. /// Translates the resulting events of a Poll operation from 3ds specific to platform specific
  178. static u32 TranslateToPlatform(Events input_event) {
  179. u32 ret = 0;
  180. if (input_event.pollin)
  181. ret |= POLLIN;
  182. if (input_event.pollpri)
  183. ret |= POLLPRI;
  184. if (input_event.pollhup)
  185. ret |= POLLHUP;
  186. if (input_event.pollerr)
  187. ret |= POLLERR;
  188. if (input_event.pollout)
  189. ret |= POLLOUT;
  190. if (input_event.pollnval)
  191. ret |= POLLNVAL;
  192. return ret;
  193. }
  194. };
  195. Events events; ///< Events to poll for (input)
  196. Events revents; ///< Events received (output)
  197. /// Converts a platform-specific pollfd to a 3ds specific structure
  198. static CTRPollFD FromPlatform(pollfd const& fd) {
  199. CTRPollFD result;
  200. result.events.hex = Events::TranslateTo3DS(fd.events).hex;
  201. result.revents.hex = Events::TranslateTo3DS(fd.revents).hex;
  202. result.fd = static_cast<u32>(fd.fd);
  203. return result;
  204. }
  205. /// Converts a 3ds specific pollfd to a platform-specific structure
  206. static pollfd ToPlatform(CTRPollFD const& fd) {
  207. pollfd result;
  208. result.events = Events::TranslateToPlatform(fd.events);
  209. result.revents = Events::TranslateToPlatform(fd.revents);
  210. result.fd = fd.fd;
  211. return result;
  212. }
  213. };
  214. /// Union to represent the 3ds' sockaddr structure
  215. union CTRSockAddr {
  216. /// Structure to represent a raw sockaddr
  217. struct {
  218. u8 len; ///< The length of the entire structure, only the set fields count
  219. u8 sa_family; ///< The address family of the sockaddr
  220. u8 sa_data[0x1A]; ///< The extra data, this varies, depending on the address family
  221. } raw;
  222. /// Structure to represent the 3ds' sockaddr_in structure
  223. struct CTRSockAddrIn {
  224. u8 len; ///< The length of the entire structure
  225. u8 sin_family; ///< The address family of the sockaddr_in
  226. u16 sin_port; ///< The port associated with this sockaddr_in
  227. u32 sin_addr; ///< The actual address of the sockaddr_in
  228. } in;
  229. /// Convert a 3DS CTRSockAddr to a platform-specific sockaddr
  230. static sockaddr ToPlatform(CTRSockAddr const& ctr_addr) {
  231. sockaddr result;
  232. result.sa_family = ctr_addr.raw.sa_family;
  233. memset(result.sa_data, 0, sizeof(result.sa_data));
  234. // We can not guarantee ABI compatibility between platforms so we copy the fields manually
  235. switch (result.sa_family) {
  236. case AF_INET:
  237. {
  238. sockaddr_in* result_in = reinterpret_cast<sockaddr_in*>(&result);
  239. result_in->sin_port = ctr_addr.in.sin_port;
  240. result_in->sin_addr.s_addr = ctr_addr.in.sin_addr;
  241. memset(result_in->sin_zero, 0, sizeof(result_in->sin_zero));
  242. break;
  243. }
  244. default:
  245. ASSERT_MSG(false, "Unhandled address family (sa_family) in CTRSockAddr::ToPlatform");
  246. break;
  247. }
  248. return result;
  249. }
  250. /// Convert a platform-specific sockaddr to a 3DS CTRSockAddr
  251. static CTRSockAddr FromPlatform(sockaddr const& addr) {
  252. CTRSockAddr result;
  253. result.raw.sa_family = static_cast<u8>(addr.sa_family);
  254. // We can not guarantee ABI compatibility between platforms so we copy the fields manually
  255. switch (result.raw.sa_family) {
  256. case AF_INET:
  257. {
  258. sockaddr_in const* addr_in = reinterpret_cast<sockaddr_in const*>(&addr);
  259. result.raw.len = sizeof(CTRSockAddrIn);
  260. result.in.sin_port = addr_in->sin_port;
  261. result.in.sin_addr = addr_in->sin_addr.s_addr;
  262. break;
  263. }
  264. default:
  265. ASSERT_MSG(false, "Unhandled address family (sa_family) in CTRSockAddr::ToPlatform");
  266. break;
  267. }
  268. return result;
  269. }
  270. };
  271. /// Holds info about the currently open sockets
  272. static std::unordered_map<u32, SocketHolder> open_sockets;
  273. /// Close all open sockets
  274. static void CleanupSockets() {
  275. for (auto sock : open_sockets)
  276. closesocket(sock.second.socket_fd);
  277. open_sockets.clear();
  278. }
  279. static void Socket(Service::Interface* self) {
  280. u32* cmd_buffer = Kernel::GetCommandBuffer();
  281. u32 domain = cmd_buffer[1]; // Address family
  282. u32 type = cmd_buffer[2];
  283. u32 protocol = cmd_buffer[3];
  284. // Only 0 is allowed according to 3dbrew, using 0 will let the OS decide which protocol to use
  285. if (protocol != 0) {
  286. cmd_buffer[1] = UnimplementedFunction(ErrorModule::SOC).raw; // TODO(Subv): Correct error code
  287. return;
  288. }
  289. if (domain != AF_INET) {
  290. cmd_buffer[1] = UnimplementedFunction(ErrorModule::SOC).raw; // TODO(Subv): Correct error code
  291. return;
  292. }
  293. if (type != SOCK_DGRAM && type != SOCK_STREAM) {
  294. cmd_buffer[1] = UnimplementedFunction(ErrorModule::SOC).raw; // TODO(Subv): Correct error code
  295. return;
  296. }
  297. u32 socket_handle = static_cast<u32>(::socket(domain, type, protocol));
  298. if ((s32)socket_handle != SOCKET_ERROR_VALUE)
  299. open_sockets[socket_handle] = { socket_handle, true };
  300. int result = 0;
  301. if ((s32)socket_handle == SOCKET_ERROR_VALUE)
  302. result = TranslateError(GET_ERRNO);
  303. cmd_buffer[0] = IPC::MakeHeader(2, 2, 0);
  304. cmd_buffer[1] = result;
  305. cmd_buffer[2] = socket_handle;
  306. }
  307. static void Bind(Service::Interface* self) {
  308. u32* cmd_buffer = Kernel::GetCommandBuffer();
  309. u32 socket_handle = cmd_buffer[1];
  310. u32 len = cmd_buffer[2];
  311. CTRSockAddr* ctr_sock_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[6]));
  312. if (ctr_sock_addr == nullptr) {
  313. cmd_buffer[1] = -1; // TODO(Subv): Correct code
  314. return;
  315. }
  316. sockaddr sock_addr = CTRSockAddr::ToPlatform(*ctr_sock_addr);
  317. int res = ::bind(socket_handle, &sock_addr, std::max<u32>(sizeof(sock_addr), len));
  318. int result = 0;
  319. if (res != 0)
  320. result = TranslateError(GET_ERRNO);
  321. cmd_buffer[0] = IPC::MakeHeader(5, 2, 0);
  322. cmd_buffer[1] = result;
  323. cmd_buffer[2] = res;
  324. }
  325. static void Fcntl(Service::Interface* self) {
  326. u32* cmd_buffer = Kernel::GetCommandBuffer();
  327. u32 socket_handle = cmd_buffer[1];
  328. u32 ctr_cmd = cmd_buffer[2];
  329. u32 ctr_arg = cmd_buffer[3];
  330. int result = 0;
  331. u32 posix_ret = 0; // TODO: Check what hardware returns for F_SETFL (unspecified by POSIX)
  332. SCOPE_EXIT({
  333. cmd_buffer[1] = result;
  334. cmd_buffer[2] = posix_ret;
  335. });
  336. if (ctr_cmd == 3) { // F_GETFL
  337. #ifdef _WIN32
  338. posix_ret = 0;
  339. auto iter = open_sockets.find(socket_handle);
  340. if (iter != open_sockets.end() && iter->second.blocking == false)
  341. posix_ret |= 4; // O_NONBLOCK
  342. #else
  343. int ret = ::fcntl(socket_handle, F_GETFL, 0);
  344. if (ret == SOCKET_ERROR_VALUE) {
  345. result = TranslateError(GET_ERRNO);
  346. posix_ret = -1;
  347. return;
  348. }
  349. posix_ret = 0;
  350. if (ret & O_NONBLOCK)
  351. posix_ret |= 4; // O_NONBLOCK
  352. #endif
  353. } else if (ctr_cmd == 4) { // F_SETFL
  354. #ifdef _WIN32
  355. unsigned long tmp = (ctr_arg & 4 /* O_NONBLOCK */) ? 1 : 0;
  356. int ret = ioctlsocket(socket_handle, FIONBIO, &tmp);
  357. if (ret == SOCKET_ERROR_VALUE) {
  358. result = TranslateError(GET_ERRNO);
  359. posix_ret = -1;
  360. return;
  361. }
  362. auto iter = open_sockets.find(socket_handle);
  363. if (iter != open_sockets.end())
  364. iter->second.blocking = (tmp == 0);
  365. #else
  366. int flags = ::fcntl(socket_handle, F_GETFL, 0);
  367. if (flags == SOCKET_ERROR_VALUE) {
  368. result = TranslateError(GET_ERRNO);
  369. posix_ret = -1;
  370. return;
  371. }
  372. flags &= ~O_NONBLOCK;
  373. if (ctr_arg & 4) // O_NONBLOCK
  374. flags |= O_NONBLOCK;
  375. int ret = ::fcntl(socket_handle, F_SETFL, flags);
  376. if (ret == SOCKET_ERROR_VALUE) {
  377. result = TranslateError(GET_ERRNO);
  378. posix_ret = -1;
  379. return;
  380. }
  381. #endif
  382. } else {
  383. LOG_ERROR(Service_SOC, "Unsupported command (%d) in fcntl call", ctr_cmd);
  384. result = TranslateError(EINVAL); // TODO: Find the correct error
  385. posix_ret = -1;
  386. return;
  387. }
  388. }
  389. static void Listen(Service::Interface* self) {
  390. u32* cmd_buffer = Kernel::GetCommandBuffer();
  391. u32 socket_handle = cmd_buffer[1];
  392. u32 backlog = cmd_buffer[2];
  393. int ret = ::listen(socket_handle, backlog);
  394. int result = 0;
  395. if (ret != 0)
  396. result = TranslateError(GET_ERRNO);
  397. cmd_buffer[0] = IPC::MakeHeader(3, 2, 0);
  398. cmd_buffer[1] = result;
  399. cmd_buffer[2] = ret;
  400. }
  401. static void Accept(Service::Interface* self) {
  402. // TODO(Subv): Calling this function on a blocking socket will block the emu thread,
  403. // preventing graceful shutdown when closing the emulator, this can be fixed by always
  404. // performing nonblocking operations and spinlock until the data is available
  405. u32* cmd_buffer = Kernel::GetCommandBuffer();
  406. u32 socket_handle = cmd_buffer[1];
  407. socklen_t max_addr_len = static_cast<socklen_t>(cmd_buffer[2]);
  408. sockaddr addr;
  409. socklen_t addr_len = sizeof(addr);
  410. u32 ret = static_cast<u32>(::accept(socket_handle, &addr, &addr_len));
  411. if ((s32)ret != SOCKET_ERROR_VALUE)
  412. open_sockets[ret] = { ret, true };
  413. int result = 0;
  414. if ((s32)ret == SOCKET_ERROR_VALUE) {
  415. result = TranslateError(GET_ERRNO);
  416. } else {
  417. CTRSockAddr ctr_addr = CTRSockAddr::FromPlatform(addr);
  418. Memory::WriteBlock(cmd_buffer[0x104 >> 2], (const u8*)&ctr_addr, max_addr_len);
  419. }
  420. cmd_buffer[0] = IPC::MakeHeader(4, 2, 2);
  421. cmd_buffer[1] = result;
  422. cmd_buffer[2] = ret;
  423. cmd_buffer[3] = IPC::StaticBufferDesc(static_cast<u32>(max_addr_len), 0);
  424. }
  425. static void GetHostId(Service::Interface* self) {
  426. u32* cmd_buffer = Kernel::GetCommandBuffer();
  427. char name[128];
  428. gethostname(name, sizeof(name));
  429. hostent* host = gethostbyname(name);
  430. in_addr* addr = reinterpret_cast<in_addr*>(host->h_addr);
  431. cmd_buffer[2] = addr->s_addr;
  432. cmd_buffer[1] = 0;
  433. }
  434. static void Close(Service::Interface* self) {
  435. u32* cmd_buffer = Kernel::GetCommandBuffer();
  436. u32 socket_handle = cmd_buffer[1];
  437. int ret = 0;
  438. open_sockets.erase(socket_handle);
  439. ret = closesocket(socket_handle);
  440. int result = 0;
  441. if (ret != 0)
  442. result = TranslateError(GET_ERRNO);
  443. cmd_buffer[2] = ret;
  444. cmd_buffer[1] = result;
  445. }
  446. static void SendTo(Service::Interface* self) {
  447. u32* cmd_buffer = Kernel::GetCommandBuffer();
  448. u32 socket_handle = cmd_buffer[1];
  449. u32 len = cmd_buffer[2];
  450. u32 flags = cmd_buffer[3];
  451. u32 addr_len = cmd_buffer[4];
  452. u8* input_buff = Memory::GetPointer(cmd_buffer[8]);
  453. CTRSockAddr* ctr_dest_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[10]));
  454. if (ctr_dest_addr == nullptr) {
  455. cmd_buffer[1] = -1; // TODO(Subv): Find the right error code
  456. return;
  457. }
  458. int ret = -1;
  459. if (addr_len > 0) {
  460. sockaddr dest_addr = CTRSockAddr::ToPlatform(*ctr_dest_addr);
  461. ret = ::sendto(socket_handle, (const char*)input_buff, len, flags, &dest_addr, sizeof(dest_addr));
  462. } else {
  463. ret = ::sendto(socket_handle, (const char*)input_buff, len, flags, nullptr, 0);
  464. }
  465. int result = 0;
  466. if (ret == SOCKET_ERROR_VALUE)
  467. result = TranslateError(GET_ERRNO);
  468. cmd_buffer[2] = ret;
  469. cmd_buffer[1] = result;
  470. }
  471. static void RecvFrom(Service::Interface* self) {
  472. // TODO(Subv): Calling this function on a blocking socket will block the emu thread,
  473. // preventing graceful shutdown when closing the emulator, this can be fixed by always
  474. // performing nonblocking operations and spinlock until the data is available
  475. u32* cmd_buffer = Kernel::GetCommandBuffer();
  476. u32 socket_handle = cmd_buffer[1];
  477. u32 len = cmd_buffer[2];
  478. u32 flags = cmd_buffer[3];
  479. socklen_t addr_len = static_cast<socklen_t>(cmd_buffer[4]);
  480. u8* output_buff = Memory::GetPointer(cmd_buffer[0x104 >> 2]);
  481. sockaddr src_addr;
  482. socklen_t src_addr_len = sizeof(src_addr);
  483. int ret = ::recvfrom(socket_handle, (char*)output_buff, len, flags, &src_addr, &src_addr_len);
  484. if (cmd_buffer[0x1A0 >> 2] != 0) {
  485. CTRSockAddr* ctr_src_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[0x1A0 >> 2]));
  486. *ctr_src_addr = CTRSockAddr::FromPlatform(src_addr);
  487. }
  488. int result = 0;
  489. int total_received = ret;
  490. if (ret == SOCKET_ERROR_VALUE) {
  491. result = TranslateError(GET_ERRNO);
  492. total_received = 0;
  493. }
  494. cmd_buffer[1] = result;
  495. cmd_buffer[2] = ret;
  496. cmd_buffer[3] = total_received;
  497. }
  498. static void Poll(Service::Interface* self) {
  499. u32* cmd_buffer = Kernel::GetCommandBuffer();
  500. u32 nfds = cmd_buffer[1];
  501. int timeout = cmd_buffer[2];
  502. CTRPollFD* input_fds = reinterpret_cast<CTRPollFD*>(Memory::GetPointer(cmd_buffer[6]));
  503. CTRPollFD* output_fds = reinterpret_cast<CTRPollFD*>(Memory::GetPointer(cmd_buffer[0x104 >> 2]));
  504. // The 3ds_pollfd and the pollfd structures may be different (Windows/Linux have different sizes)
  505. // so we have to copy the data
  506. pollfd* platform_pollfd = new pollfd[nfds];
  507. for (unsigned current_fds = 0; current_fds < nfds; ++current_fds)
  508. platform_pollfd[current_fds] = CTRPollFD::ToPlatform(input_fds[current_fds]);
  509. int ret = ::poll(platform_pollfd, nfds, timeout);
  510. // Now update the output pollfd structure
  511. for (unsigned current_fds = 0; current_fds < nfds; ++current_fds)
  512. output_fds[current_fds] = CTRPollFD::FromPlatform(platform_pollfd[current_fds]);
  513. delete[] platform_pollfd;
  514. int result = 0;
  515. if (ret == SOCKET_ERROR_VALUE)
  516. result = TranslateError(GET_ERRNO);
  517. cmd_buffer[1] = result;
  518. cmd_buffer[2] = ret;
  519. }
  520. static void GetSockName(Service::Interface* self) {
  521. u32* cmd_buffer = Kernel::GetCommandBuffer();
  522. u32 socket_handle = cmd_buffer[1];
  523. socklen_t ctr_len = cmd_buffer[2];
  524. CTRSockAddr* ctr_dest_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[0x104 >> 2]));
  525. sockaddr dest_addr;
  526. socklen_t dest_addr_len = sizeof(dest_addr);
  527. int ret = ::getsockname(socket_handle, &dest_addr, &dest_addr_len);
  528. if (ctr_dest_addr != nullptr) {
  529. *ctr_dest_addr = CTRSockAddr::FromPlatform(dest_addr);
  530. } else {
  531. cmd_buffer[1] = -1; // TODO(Subv): Verify error
  532. return;
  533. }
  534. int result = 0;
  535. if (ret != 0)
  536. result = TranslateError(GET_ERRNO);
  537. cmd_buffer[2] = ret;
  538. cmd_buffer[1] = result;
  539. }
  540. static void Shutdown(Service::Interface* self) {
  541. u32* cmd_buffer = Kernel::GetCommandBuffer();
  542. u32 socket_handle = cmd_buffer[1];
  543. int how = cmd_buffer[2];
  544. int ret = ::shutdown(socket_handle, how);
  545. int result = 0;
  546. if (ret != 0)
  547. result = TranslateError(GET_ERRNO);
  548. cmd_buffer[2] = ret;
  549. cmd_buffer[1] = result;
  550. }
  551. static void GetPeerName(Service::Interface* self) {
  552. u32* cmd_buffer = Kernel::GetCommandBuffer();
  553. u32 socket_handle = cmd_buffer[1];
  554. socklen_t len = cmd_buffer[2];
  555. CTRSockAddr* ctr_dest_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[0x104 >> 2]));
  556. sockaddr dest_addr;
  557. socklen_t dest_addr_len = sizeof(dest_addr);
  558. int ret = ::getpeername(socket_handle, &dest_addr, &dest_addr_len);
  559. if (ctr_dest_addr != nullptr) {
  560. *ctr_dest_addr = CTRSockAddr::FromPlatform(dest_addr);
  561. } else {
  562. cmd_buffer[1] = -1;
  563. return;
  564. }
  565. int result = 0;
  566. if (ret != 0)
  567. result = TranslateError(GET_ERRNO);
  568. cmd_buffer[2] = ret;
  569. cmd_buffer[1] = result;
  570. }
  571. static void Connect(Service::Interface* self) {
  572. // TODO(Subv): Calling this function on a blocking socket will block the emu thread,
  573. // preventing graceful shutdown when closing the emulator, this can be fixed by always
  574. // performing nonblocking operations and spinlock until the data is available
  575. u32* cmd_buffer = Kernel::GetCommandBuffer();
  576. u32 socket_handle = cmd_buffer[1];
  577. socklen_t len = cmd_buffer[2];
  578. CTRSockAddr* ctr_input_addr = reinterpret_cast<CTRSockAddr*>(Memory::GetPointer(cmd_buffer[6]));
  579. if (ctr_input_addr == nullptr) {
  580. cmd_buffer[1] = -1; // TODO(Subv): Verify error
  581. return;
  582. }
  583. sockaddr input_addr = CTRSockAddr::ToPlatform(*ctr_input_addr);
  584. int ret = ::connect(socket_handle, &input_addr, sizeof(input_addr));
  585. int result = 0;
  586. if (ret != 0)
  587. result = TranslateError(GET_ERRNO);
  588. cmd_buffer[0] = IPC::MakeHeader(6, 2, 0);
  589. cmd_buffer[1] = result;
  590. cmd_buffer[2] = ret;
  591. }
  592. static void InitializeSockets(Service::Interface* self) {
  593. // TODO(Subv): Implement
  594. #ifdef _WIN32
  595. WSADATA data;
  596. WSAStartup(MAKEWORD(2, 2), &data);
  597. #endif
  598. u32* cmd_buffer = Kernel::GetCommandBuffer();
  599. cmd_buffer[0] = IPC::MakeHeader(1, 1, 0);
  600. cmd_buffer[1] = RESULT_SUCCESS.raw;
  601. }
  602. static void ShutdownSockets(Service::Interface* self) {
  603. // TODO(Subv): Implement
  604. CleanupSockets();
  605. #ifdef _WIN32
  606. WSACleanup();
  607. #endif
  608. u32* cmd_buffer = Kernel::GetCommandBuffer();
  609. cmd_buffer[1] = 0;
  610. }
  611. const Interface::FunctionInfo FunctionTable[] = {
  612. {0x00010044, InitializeSockets, "InitializeSockets"},
  613. {0x000200C2, Socket, "Socket"},
  614. {0x00030082, Listen, "Listen"},
  615. {0x00040082, Accept, "Accept"},
  616. {0x00050084, Bind, "Bind"},
  617. {0x00060084, Connect, "Connect"},
  618. {0x00070104, nullptr, "recvfrom_other"},
  619. {0x00080102, RecvFrom, "RecvFrom"},
  620. {0x00090106, nullptr, "sendto_other"},
  621. {0x000A0106, SendTo, "SendTo"},
  622. {0x000B0042, Close, "Close"},
  623. {0x000C0082, Shutdown, "Shutdown"},
  624. {0x000D0082, nullptr, "GetHostByName"},
  625. {0x000E00C2, nullptr, "GetHostByAddr"},
  626. {0x000F0106, nullptr, "unknown_resolve_ip"},
  627. {0x00110102, nullptr, "GetSockOpt"},
  628. {0x00120104, nullptr, "SetSockOpt"},
  629. {0x001300C2, Fcntl, "Fcntl"},
  630. {0x00140084, Poll, "Poll"},
  631. {0x00150042, nullptr, "SockAtMark"},
  632. {0x00160000, GetHostId, "GetHostId"},
  633. {0x00170082, GetSockName, "GetSockName"},
  634. {0x00180082, GetPeerName, "GetPeerName"},
  635. {0x00190000, ShutdownSockets, "ShutdownSockets"},
  636. {0x001A00C0, nullptr, "GetNetworkOpt"},
  637. {0x001B0040, nullptr, "ICMPSocket"},
  638. {0x001C0104, nullptr, "ICMPPing"},
  639. {0x001D0040, nullptr, "ICMPCancel"},
  640. {0x001E0040, nullptr, "ICMPClose"},
  641. {0x001F0040, nullptr, "GetResolverInfo"},
  642. {0x00210002, nullptr, "CloseSockets"},
  643. };
  644. ////////////////////////////////////////////////////////////////////////////////////////////////////
  645. // Interface class
  646. Interface::Interface() {
  647. Register(FunctionTable);
  648. }
  649. Interface::~Interface() {
  650. CleanupSockets();
  651. #ifdef _WIN32
  652. WSACleanup();
  653. #endif
  654. }
  655. } // namespace