2
0

net_socket_posix.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778
  1. /*************************************************************************/
  2. /* net_socket_posix.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
  10. /* */
  11. /* Permission is hereby granted, free of charge, to any person obtaining */
  12. /* a copy of this software and associated documentation files (the */
  13. /* "Software"), to deal in the Software without restriction, including */
  14. /* without limitation the rights to use, copy, modify, merge, publish, */
  15. /* distribute, sublicense, and/or sell copies of the Software, and to */
  16. /* permit persons to whom the Software is furnished to do so, subject to */
  17. /* the following conditions: */
  18. /* */
  19. /* The above copyright notice and this permission notice shall be */
  20. /* included in all copies or substantial portions of the Software. */
  21. /* */
  22. /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
  23. /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
  24. /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
  25. /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
  26. /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
  27. /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
  28. /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
  29. /*************************************************************************/
  30. #include "net_socket_posix.h"
  31. // Some proprietary Unix-derived platforms don't expose Unix sockets
  32. // so this allows skipping this file to reimplement this API differently.
  33. #ifndef UNIX_SOCKET_UNAVAILABLE
  34. #if defined(UNIX_ENABLED)
  35. #include <errno.h>
  36. #include <fcntl.h>
  37. #include <netdb.h>
  38. #include <netinet/in.h>
  39. #include <netinet/tcp.h>
  40. #include <poll.h>
  41. #include <stdio.h>
  42. #include <stdlib.h>
  43. #include <string.h>
  44. #include <sys/ioctl.h>
  45. #include <sys/socket.h>
  46. #include <sys/types.h>
  47. #include <unistd.h>
  48. #ifdef WEB_ENABLED
  49. #include <arpa/inet.h>
  50. #endif
  51. // BSD calls this flag IPV6_JOIN_GROUP
  52. #if !defined(IPV6_ADD_MEMBERSHIP) && defined(IPV6_JOIN_GROUP)
  53. #define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP
  54. #endif
  55. #if !defined(IPV6_DROP_MEMBERSHIP) && defined(IPV6_LEAVE_GROUP)
  56. #define IPV6_DROP_MEMBERSHIP IPV6_LEAVE_GROUP
  57. #endif
  58. // Some custom defines to minimize ifdefs
  59. #define SOCK_EMPTY -1
  60. #define SOCK_BUF(x) x
  61. #define SOCK_CBUF(x) x
  62. #define SOCK_IOCTL ioctl
  63. #define SOCK_CLOSE ::close
  64. #define SOCK_CONNECT(p_sock, p_addr, p_addr_len) ::connect(p_sock, p_addr, p_addr_len)
  65. /* Windows */
  66. #elif defined(WINDOWS_ENABLED)
  67. #include <winsock2.h>
  68. #include <ws2tcpip.h>
  69. #include <mswsock.h>
  70. // Some custom defines to minimize ifdefs
  71. #define SOCK_EMPTY INVALID_SOCKET
  72. #define SOCK_BUF(x) (char *)(x)
  73. #define SOCK_CBUF(x) (const char *)(x)
  74. #define SOCK_IOCTL ioctlsocket
  75. #define SOCK_CLOSE closesocket
  76. // connect is broken on windows under certain conditions, reasons unknown:
  77. // See https://github.com/godotengine/webrtc-native/issues/6
  78. #define SOCK_CONNECT(p_sock, p_addr, p_addr_len) ::WSAConnect(p_sock, p_addr, p_addr_len, nullptr, nullptr, nullptr, nullptr)
  79. // Workaround missing flag in MinGW
  80. #if defined(__MINGW32__) && !defined(SIO_UDP_NETRESET)
  81. #define SIO_UDP_NETRESET _WSAIOW(IOC_VENDOR, 15)
  82. #endif
  83. #endif // UNIX_ENABLED
  84. size_t NetSocketPosix::_set_addr_storage(struct sockaddr_storage *p_addr, const IPAddress &p_ip, uint16_t p_port, IP::Type p_ip_type) {
  85. memset(p_addr, 0, sizeof(struct sockaddr_storage));
  86. if (p_ip_type == IP::TYPE_IPV6 || p_ip_type == IP::TYPE_ANY) { // IPv6 socket
  87. // IPv6 only socket with IPv4 address
  88. ERR_FAIL_COND_V(!p_ip.is_wildcard() && p_ip_type == IP::TYPE_IPV6 && p_ip.is_ipv4(), 0);
  89. struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)p_addr;
  90. addr6->sin6_family = AF_INET6;
  91. addr6->sin6_port = htons(p_port);
  92. if (p_ip.is_valid()) {
  93. memcpy(&addr6->sin6_addr.s6_addr, p_ip.get_ipv6(), 16);
  94. } else {
  95. addr6->sin6_addr = in6addr_any;
  96. }
  97. return sizeof(sockaddr_in6);
  98. } else { // IPv4 socket
  99. // IPv4 socket with IPv6 address
  100. ERR_FAIL_COND_V(!p_ip.is_wildcard() && !p_ip.is_ipv4(), 0);
  101. struct sockaddr_in *addr4 = (struct sockaddr_in *)p_addr;
  102. addr4->sin_family = AF_INET;
  103. addr4->sin_port = htons(p_port); // short, network byte order
  104. if (p_ip.is_valid()) {
  105. memcpy(&addr4->sin_addr.s_addr, p_ip.get_ipv4(), 4);
  106. } else {
  107. addr4->sin_addr.s_addr = INADDR_ANY;
  108. }
  109. return sizeof(sockaddr_in);
  110. }
  111. }
  112. void NetSocketPosix::_set_ip_port(struct sockaddr_storage *p_addr, IPAddress *r_ip, uint16_t *r_port) {
  113. if (p_addr->ss_family == AF_INET) {
  114. struct sockaddr_in *addr4 = (struct sockaddr_in *)p_addr;
  115. if (r_ip) {
  116. r_ip->set_ipv4((uint8_t *)&(addr4->sin_addr.s_addr));
  117. }
  118. if (r_port) {
  119. *r_port = ntohs(addr4->sin_port);
  120. }
  121. } else if (p_addr->ss_family == AF_INET6) {
  122. struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)p_addr;
  123. if (r_ip) {
  124. r_ip->set_ipv6(addr6->sin6_addr.s6_addr);
  125. }
  126. if (r_port) {
  127. *r_port = ntohs(addr6->sin6_port);
  128. }
  129. }
  130. }
  131. NetSocket *NetSocketPosix::_create_func() {
  132. return memnew(NetSocketPosix);
  133. }
  134. void NetSocketPosix::make_default() {
  135. #if defined(WINDOWS_ENABLED)
  136. if (_create == nullptr) {
  137. WSADATA data;
  138. WSAStartup(MAKEWORD(2, 2), &data);
  139. }
  140. #endif
  141. _create = _create_func;
  142. }
  143. void NetSocketPosix::cleanup() {
  144. #if defined(WINDOWS_ENABLED)
  145. if (_create != nullptr) {
  146. WSACleanup();
  147. }
  148. _create = nullptr;
  149. #endif
  150. }
  151. NetSocketPosix::NetSocketPosix() :
  152. _sock(SOCK_EMPTY) {
  153. }
  154. NetSocketPosix::~NetSocketPosix() {
  155. close();
  156. }
  157. // Silence a warning reported in GH-27594.
  158. // EAGAIN and EWOULDBLOCK have the same value on most platforms, but it's not guaranteed.
  159. #if defined(__GNUC__) && !defined(__clang__)
  160. #pragma GCC diagnostic push
  161. #pragma GCC diagnostic ignored "-Wlogical-op"
  162. #endif
  163. NetSocketPosix::NetError NetSocketPosix::_get_socket_error() const {
  164. #if defined(WINDOWS_ENABLED)
  165. int err = WSAGetLastError();
  166. if (err == WSAEISCONN) {
  167. return ERR_NET_IS_CONNECTED;
  168. }
  169. if (err == WSAEINPROGRESS || err == WSAEALREADY) {
  170. return ERR_NET_IN_PROGRESS;
  171. }
  172. if (err == WSAEWOULDBLOCK) {
  173. return ERR_NET_WOULD_BLOCK;
  174. }
  175. if (err == WSAEADDRINUSE || err == WSAEADDRNOTAVAIL) {
  176. return ERR_NET_ADDRESS_INVALID_OR_UNAVAILABLE;
  177. }
  178. if (err == WSAEACCES) {
  179. return ERR_NET_UNAUTHORIZED;
  180. }
  181. print_verbose("Socket error: " + itos(err));
  182. return ERR_NET_OTHER;
  183. #else
  184. if (errno == EISCONN) {
  185. return ERR_NET_IS_CONNECTED;
  186. }
  187. if (errno == EINPROGRESS || errno == EALREADY) {
  188. return ERR_NET_IN_PROGRESS;
  189. }
  190. if (errno == EAGAIN || errno == EWOULDBLOCK) {
  191. return ERR_NET_WOULD_BLOCK;
  192. }
  193. if (errno == EADDRINUSE || errno == EINVAL || errno == EADDRNOTAVAIL) {
  194. return ERR_NET_ADDRESS_INVALID_OR_UNAVAILABLE;
  195. }
  196. if (errno == EACCES) {
  197. return ERR_NET_UNAUTHORIZED;
  198. }
  199. print_verbose("Socket error: " + itos(errno));
  200. return ERR_NET_OTHER;
  201. #endif
  202. }
  203. #if defined(__GNUC__) && !defined(__clang__)
  204. #pragma GCC diagnostic pop
  205. #endif
  206. bool NetSocketPosix::_can_use_ip(const IPAddress &p_ip, const bool p_for_bind) const {
  207. if (p_for_bind && !(p_ip.is_valid() || p_ip.is_wildcard())) {
  208. return false;
  209. } else if (!p_for_bind && !p_ip.is_valid()) {
  210. return false;
  211. }
  212. // Check if socket support this IP type.
  213. IP::Type type = p_ip.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6;
  214. return !(_ip_type != IP::TYPE_ANY && !p_ip.is_wildcard() && _ip_type != type);
  215. }
  216. _FORCE_INLINE_ Error NetSocketPosix::_change_multicast_group(IPAddress p_ip, String p_if_name, bool p_add) {
  217. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  218. ERR_FAIL_COND_V(!_can_use_ip(p_ip, false), ERR_INVALID_PARAMETER);
  219. // Need to force level and af_family to IP(v4) when using dual stacking and provided multicast group is IPv4
  220. IP::Type type = _ip_type == IP::TYPE_ANY && p_ip.is_ipv4() ? IP::TYPE_IPV4 : _ip_type;
  221. // This needs to be the proper level for the multicast group, no matter if the socket is dual stacking.
  222. int level = type == IP::TYPE_IPV4 ? IPPROTO_IP : IPPROTO_IPV6;
  223. int ret = -1;
  224. IPAddress if_ip;
  225. uint32_t if_v6id = 0;
  226. HashMap<String, IP::Interface_Info> if_info;
  227. IP::get_singleton()->get_local_interfaces(&if_info);
  228. for (KeyValue<String, IP::Interface_Info> &E : if_info) {
  229. IP::Interface_Info &c = E.value;
  230. if (c.name != p_if_name) {
  231. continue;
  232. }
  233. if_v6id = (uint32_t)c.index.to_int();
  234. if (type == IP::TYPE_IPV6) {
  235. break; // IPv6 uses index.
  236. }
  237. for (const IPAddress &F : c.ip_addresses) {
  238. if (!F.is_ipv4()) {
  239. continue; // Wrong IP type
  240. }
  241. if_ip = F;
  242. break;
  243. }
  244. break;
  245. }
  246. if (level == IPPROTO_IP) {
  247. ERR_FAIL_COND_V(!if_ip.is_valid(), ERR_INVALID_PARAMETER);
  248. struct ip_mreq greq;
  249. int sock_opt = p_add ? IP_ADD_MEMBERSHIP : IP_DROP_MEMBERSHIP;
  250. memcpy(&greq.imr_multiaddr, p_ip.get_ipv4(), 4);
  251. memcpy(&greq.imr_interface, if_ip.get_ipv4(), 4);
  252. ret = setsockopt(_sock, level, sock_opt, (const char *)&greq, sizeof(greq));
  253. } else {
  254. struct ipv6_mreq greq;
  255. int sock_opt = p_add ? IPV6_ADD_MEMBERSHIP : IPV6_DROP_MEMBERSHIP;
  256. memcpy(&greq.ipv6mr_multiaddr, p_ip.get_ipv6(), 16);
  257. greq.ipv6mr_interface = if_v6id;
  258. ret = setsockopt(_sock, level, sock_opt, (const char *)&greq, sizeof(greq));
  259. }
  260. ERR_FAIL_COND_V(ret != 0, FAILED);
  261. return OK;
  262. }
  263. void NetSocketPosix::_set_socket(SOCKET_TYPE p_sock, IP::Type p_ip_type, bool p_is_stream) {
  264. _sock = p_sock;
  265. _ip_type = p_ip_type;
  266. _is_stream = p_is_stream;
  267. // Disable descriptor sharing with subprocesses.
  268. _set_close_exec_enabled(true);
  269. }
  270. void NetSocketPosix::_set_close_exec_enabled(bool p_enabled) {
  271. #ifndef WINDOWS_ENABLED
  272. // Enable close on exec to avoid sharing with subprocesses. Off by default on Windows.
  273. int opts = fcntl(_sock, F_GETFD);
  274. fcntl(_sock, F_SETFD, opts | FD_CLOEXEC);
  275. #endif
  276. }
  277. Error NetSocketPosix::open(Type p_sock_type, IP::Type &ip_type) {
  278. ERR_FAIL_COND_V(is_open(), ERR_ALREADY_IN_USE);
  279. ERR_FAIL_COND_V(ip_type > IP::TYPE_ANY || ip_type < IP::TYPE_NONE, ERR_INVALID_PARAMETER);
  280. #if defined(__OpenBSD__)
  281. // OpenBSD does not support dual stacking, fallback to IPv4 only.
  282. if (ip_type == IP::TYPE_ANY) {
  283. ip_type = IP::TYPE_IPV4;
  284. }
  285. #endif
  286. int family = ip_type == IP::TYPE_IPV4 ? AF_INET : AF_INET6;
  287. int protocol = p_sock_type == TYPE_TCP ? IPPROTO_TCP : IPPROTO_UDP;
  288. int type = p_sock_type == TYPE_TCP ? SOCK_STREAM : SOCK_DGRAM;
  289. _sock = socket(family, type, protocol);
  290. if (_sock == SOCK_EMPTY && ip_type == IP::TYPE_ANY) {
  291. // Careful here, changing the referenced parameter so the caller knows that we are using an IPv4 socket
  292. // in place of a dual stack one, and further calls to _set_sock_addr will work as expected.
  293. ip_type = IP::TYPE_IPV4;
  294. family = AF_INET;
  295. _sock = socket(family, type, protocol);
  296. }
  297. ERR_FAIL_COND_V(_sock == SOCK_EMPTY, FAILED);
  298. _ip_type = ip_type;
  299. if (family == AF_INET6) {
  300. // Select IPv4 over IPv6 mapping
  301. set_ipv6_only_enabled(ip_type != IP::TYPE_ANY);
  302. }
  303. if (protocol == IPPROTO_UDP) {
  304. // Make sure to disable broadcasting for UDP sockets.
  305. // Depending on the OS, this option might or might not be enabled by default. Let's normalize it.
  306. set_broadcasting_enabled(false);
  307. }
  308. _is_stream = p_sock_type == TYPE_TCP;
  309. // Disable descriptor sharing with subprocesses.
  310. _set_close_exec_enabled(true);
  311. #if defined(WINDOWS_ENABLED)
  312. if (!_is_stream) {
  313. // Disable windows feature/bug reporting WSAECONNRESET/WSAENETRESET when
  314. // recv/recvfrom and an ICMP reply was received from a previous send/sendto.
  315. unsigned long disable = 0;
  316. if (ioctlsocket(_sock, SIO_UDP_CONNRESET, &disable) == SOCKET_ERROR) {
  317. print_verbose("Unable to turn off UDP WSAECONNRESET behavior on Windows");
  318. }
  319. if (ioctlsocket(_sock, SIO_UDP_NETRESET, &disable) == SOCKET_ERROR) {
  320. // This feature seems not to be supported on wine.
  321. print_verbose("Unable to turn off UDP WSAENETRESET behavior on Windows");
  322. }
  323. }
  324. #endif
  325. #if defined(SO_NOSIGPIPE)
  326. // Disable SIGPIPE (should only be relevant to stream sockets, but seems to affect UDP too on iOS)
  327. int par = 1;
  328. if (setsockopt(_sock, SOL_SOCKET, SO_NOSIGPIPE, SOCK_CBUF(&par), sizeof(int)) != 0) {
  329. print_verbose("Unable to turn off SIGPIPE on socket");
  330. }
  331. #endif
  332. return OK;
  333. }
  334. void NetSocketPosix::close() {
  335. if (_sock != SOCK_EMPTY) {
  336. SOCK_CLOSE(_sock);
  337. }
  338. _sock = SOCK_EMPTY;
  339. _ip_type = IP::TYPE_NONE;
  340. _is_stream = false;
  341. }
  342. Error NetSocketPosix::bind(IPAddress p_addr, uint16_t p_port) {
  343. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  344. ERR_FAIL_COND_V(!_can_use_ip(p_addr, true), ERR_INVALID_PARAMETER);
  345. sockaddr_storage addr;
  346. size_t addr_size = _set_addr_storage(&addr, p_addr, p_port, _ip_type);
  347. if (::bind(_sock, (struct sockaddr *)&addr, addr_size) != 0) {
  348. NetError err = _get_socket_error();
  349. print_verbose("Failed to bind socket. Error: " + itos(err));
  350. close();
  351. return ERR_UNAVAILABLE;
  352. }
  353. return OK;
  354. }
  355. Error NetSocketPosix::listen(int p_max_pending) {
  356. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  357. if (::listen(_sock, p_max_pending) != 0) {
  358. _get_socket_error();
  359. print_verbose("Failed to listen from socket.");
  360. close();
  361. return FAILED;
  362. }
  363. return OK;
  364. }
  365. Error NetSocketPosix::connect_to_host(IPAddress p_host, uint16_t p_port) {
  366. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  367. ERR_FAIL_COND_V(!_can_use_ip(p_host, false), ERR_INVALID_PARAMETER);
  368. struct sockaddr_storage addr;
  369. size_t addr_size = _set_addr_storage(&addr, p_host, p_port, _ip_type);
  370. if (SOCK_CONNECT(_sock, (struct sockaddr *)&addr, addr_size) != 0) {
  371. NetError err = _get_socket_error();
  372. switch (err) {
  373. // We are already connected
  374. case ERR_NET_IS_CONNECTED:
  375. return OK;
  376. // Still waiting to connect, try again in a while
  377. case ERR_NET_WOULD_BLOCK:
  378. case ERR_NET_IN_PROGRESS:
  379. return ERR_BUSY;
  380. default:
  381. print_verbose("Connection to remote host failed!");
  382. close();
  383. return FAILED;
  384. }
  385. }
  386. return OK;
  387. }
  388. Error NetSocketPosix::poll(PollType p_type, int p_timeout) const {
  389. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  390. #if defined(WINDOWS_ENABLED)
  391. bool ready = false;
  392. fd_set rd, wr, ex;
  393. fd_set *rdp = nullptr;
  394. fd_set *wrp = nullptr;
  395. FD_ZERO(&rd);
  396. FD_ZERO(&wr);
  397. FD_ZERO(&ex);
  398. FD_SET(_sock, &ex);
  399. struct timeval timeout = { p_timeout / 1000, (p_timeout % 1000) * 1000 };
  400. // For blocking operation, pass nullptr timeout pointer to select.
  401. struct timeval *tp = nullptr;
  402. if (p_timeout >= 0) {
  403. // If timeout is non-negative, we want to specify the timeout instead.
  404. tp = &timeout;
  405. }
  406. switch (p_type) {
  407. case POLL_TYPE_IN:
  408. FD_SET(_sock, &rd);
  409. rdp = &rd;
  410. break;
  411. case POLL_TYPE_OUT:
  412. FD_SET(_sock, &wr);
  413. wrp = &wr;
  414. break;
  415. case POLL_TYPE_IN_OUT:
  416. FD_SET(_sock, &rd);
  417. FD_SET(_sock, &wr);
  418. rdp = &rd;
  419. wrp = &wr;
  420. }
  421. int ret = select(1, rdp, wrp, &ex, tp);
  422. if (ret == SOCKET_ERROR) {
  423. return FAILED;
  424. }
  425. if (ret == 0) {
  426. return ERR_BUSY;
  427. }
  428. if (FD_ISSET(_sock, &ex)) {
  429. _get_socket_error();
  430. print_verbose("Exception when polling socket.");
  431. return FAILED;
  432. }
  433. if (rdp && FD_ISSET(_sock, rdp)) {
  434. ready = true;
  435. }
  436. if (wrp && FD_ISSET(_sock, wrp)) {
  437. ready = true;
  438. }
  439. return ready ? OK : ERR_BUSY;
  440. #else
  441. struct pollfd pfd;
  442. pfd.fd = _sock;
  443. pfd.events = POLLIN;
  444. pfd.revents = 0;
  445. switch (p_type) {
  446. case POLL_TYPE_IN:
  447. pfd.events = POLLIN;
  448. break;
  449. case POLL_TYPE_OUT:
  450. pfd.events = POLLOUT;
  451. break;
  452. case POLL_TYPE_IN_OUT:
  453. pfd.events = POLLOUT | POLLIN;
  454. }
  455. int ret = ::poll(&pfd, 1, p_timeout);
  456. if (ret < 0 || pfd.revents & POLLERR) {
  457. _get_socket_error();
  458. print_verbose("Error when polling socket.");
  459. return FAILED;
  460. }
  461. if (ret == 0) {
  462. return ERR_BUSY;
  463. }
  464. return OK;
  465. #endif
  466. }
  467. Error NetSocketPosix::recv(uint8_t *p_buffer, int p_len, int &r_read) {
  468. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  469. r_read = ::recv(_sock, SOCK_BUF(p_buffer), p_len, 0);
  470. if (r_read < 0) {
  471. NetError err = _get_socket_error();
  472. if (err == ERR_NET_WOULD_BLOCK) {
  473. return ERR_BUSY;
  474. }
  475. return FAILED;
  476. }
  477. return OK;
  478. }
  479. Error NetSocketPosix::recvfrom(uint8_t *p_buffer, int p_len, int &r_read, IPAddress &r_ip, uint16_t &r_port, bool p_peek) {
  480. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  481. struct sockaddr_storage from;
  482. socklen_t len = sizeof(struct sockaddr_storage);
  483. memset(&from, 0, len);
  484. r_read = ::recvfrom(_sock, SOCK_BUF(p_buffer), p_len, p_peek ? MSG_PEEK : 0, (struct sockaddr *)&from, &len);
  485. if (r_read < 0) {
  486. NetError err = _get_socket_error();
  487. if (err == ERR_NET_WOULD_BLOCK) {
  488. return ERR_BUSY;
  489. }
  490. return FAILED;
  491. }
  492. if (from.ss_family == AF_INET) {
  493. struct sockaddr_in *sin_from = (struct sockaddr_in *)&from;
  494. r_ip.set_ipv4((uint8_t *)&sin_from->sin_addr);
  495. r_port = ntohs(sin_from->sin_port);
  496. } else if (from.ss_family == AF_INET6) {
  497. struct sockaddr_in6 *s6_from = (struct sockaddr_in6 *)&from;
  498. r_ip.set_ipv6((uint8_t *)&s6_from->sin6_addr);
  499. r_port = ntohs(s6_from->sin6_port);
  500. } else {
  501. // Unsupported socket family, should never happen.
  502. ERR_FAIL_V(FAILED);
  503. }
  504. return OK;
  505. }
  506. Error NetSocketPosix::send(const uint8_t *p_buffer, int p_len, int &r_sent) {
  507. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  508. int flags = 0;
  509. #ifdef MSG_NOSIGNAL
  510. if (_is_stream) {
  511. flags = MSG_NOSIGNAL;
  512. }
  513. #endif
  514. r_sent = ::send(_sock, SOCK_CBUF(p_buffer), p_len, flags);
  515. if (r_sent < 0) {
  516. NetError err = _get_socket_error();
  517. if (err == ERR_NET_WOULD_BLOCK) {
  518. return ERR_BUSY;
  519. }
  520. return FAILED;
  521. }
  522. return OK;
  523. }
  524. Error NetSocketPosix::sendto(const uint8_t *p_buffer, int p_len, int &r_sent, IPAddress p_ip, uint16_t p_port) {
  525. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  526. struct sockaddr_storage addr;
  527. size_t addr_size = _set_addr_storage(&addr, p_ip, p_port, _ip_type);
  528. r_sent = ::sendto(_sock, SOCK_CBUF(p_buffer), p_len, 0, (struct sockaddr *)&addr, addr_size);
  529. if (r_sent < 0) {
  530. NetError err = _get_socket_error();
  531. if (err == ERR_NET_WOULD_BLOCK) {
  532. return ERR_BUSY;
  533. }
  534. return FAILED;
  535. }
  536. return OK;
  537. }
  538. Error NetSocketPosix::set_broadcasting_enabled(bool p_enabled) {
  539. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  540. // IPv6 has no broadcast support.
  541. if (_ip_type == IP::TYPE_IPV6) {
  542. return ERR_UNAVAILABLE;
  543. }
  544. int par = p_enabled ? 1 : 0;
  545. if (setsockopt(_sock, SOL_SOCKET, SO_BROADCAST, SOCK_CBUF(&par), sizeof(int)) != 0) {
  546. WARN_PRINT("Unable to change broadcast setting");
  547. return FAILED;
  548. }
  549. return OK;
  550. }
  551. void NetSocketPosix::set_blocking_enabled(bool p_enabled) {
  552. ERR_FAIL_COND(!is_open());
  553. int ret = 0;
  554. #if defined(WINDOWS_ENABLED)
  555. unsigned long par = p_enabled ? 0 : 1;
  556. ret = SOCK_IOCTL(_sock, FIONBIO, &par);
  557. #else
  558. int opts = fcntl(_sock, F_GETFL);
  559. if (p_enabled) {
  560. ret = fcntl(_sock, F_SETFL, opts & ~O_NONBLOCK);
  561. } else {
  562. ret = fcntl(_sock, F_SETFL, opts | O_NONBLOCK);
  563. }
  564. #endif
  565. if (ret != 0) {
  566. WARN_PRINT("Unable to change non-block mode");
  567. }
  568. }
  569. void NetSocketPosix::set_ipv6_only_enabled(bool p_enabled) {
  570. ERR_FAIL_COND(!is_open());
  571. // This option is only available in IPv6 sockets.
  572. ERR_FAIL_COND(_ip_type == IP::TYPE_IPV4);
  573. int par = p_enabled ? 1 : 0;
  574. if (setsockopt(_sock, IPPROTO_IPV6, IPV6_V6ONLY, SOCK_CBUF(&par), sizeof(int)) != 0) {
  575. WARN_PRINT("Unable to change IPv4 address mapping over IPv6 option");
  576. }
  577. }
  578. void NetSocketPosix::set_tcp_no_delay_enabled(bool p_enabled) {
  579. ERR_FAIL_COND(!is_open());
  580. ERR_FAIL_COND(!_is_stream); // Not TCP
  581. int par = p_enabled ? 1 : 0;
  582. if (setsockopt(_sock, IPPROTO_TCP, TCP_NODELAY, SOCK_CBUF(&par), sizeof(int)) < 0) {
  583. ERR_PRINT("Unable to set TCP no delay option");
  584. }
  585. }
  586. void NetSocketPosix::set_reuse_address_enabled(bool p_enabled) {
  587. ERR_FAIL_COND(!is_open());
  588. // On Windows, enabling SO_REUSEADDR actually would also enable reuse port, very bad on TCP. Denying...
  589. // Windows does not have this option, SO_REUSEADDR in this magical world means SO_REUSEPORT
  590. #ifndef WINDOWS_ENABLED
  591. int par = p_enabled ? 1 : 0;
  592. if (setsockopt(_sock, SOL_SOCKET, SO_REUSEADDR, SOCK_CBUF(&par), sizeof(int)) < 0) {
  593. WARN_PRINT("Unable to set socket REUSEADDR option!");
  594. }
  595. #endif
  596. }
  597. void NetSocketPosix::set_reuse_port_enabled(bool p_enabled) {
  598. ERR_FAIL_COND(!is_open());
  599. // See comment above...
  600. #ifdef WINDOWS_ENABLED
  601. #define SO_REUSEPORT SO_REUSEADDR
  602. #endif
  603. int par = p_enabled ? 1 : 0;
  604. if (setsockopt(_sock, SOL_SOCKET, SO_REUSEPORT, SOCK_CBUF(&par), sizeof(int)) < 0) {
  605. WARN_PRINT("Unable to set socket REUSEPORT option!");
  606. }
  607. }
  608. bool NetSocketPosix::is_open() const {
  609. return _sock != SOCK_EMPTY;
  610. }
  611. int NetSocketPosix::get_available_bytes() const {
  612. ERR_FAIL_COND_V(!is_open(), -1);
  613. unsigned long len;
  614. int ret = SOCK_IOCTL(_sock, FIONREAD, &len);
  615. if (ret == -1) {
  616. _get_socket_error();
  617. print_verbose("Error when checking available bytes on socket.");
  618. return -1;
  619. }
  620. return len;
  621. }
  622. Error NetSocketPosix::get_socket_address(IPAddress *r_ip, uint16_t *r_port) const {
  623. ERR_FAIL_COND_V(!is_open(), FAILED);
  624. struct sockaddr_storage saddr;
  625. socklen_t len = sizeof(saddr);
  626. if (getsockname(_sock, (struct sockaddr *)&saddr, &len) != 0) {
  627. _get_socket_error();
  628. print_verbose("Error when reading local socket address.");
  629. return FAILED;
  630. }
  631. _set_ip_port(&saddr, r_ip, r_port);
  632. return OK;
  633. }
  634. Ref<NetSocket> NetSocketPosix::accept(IPAddress &r_ip, uint16_t &r_port) {
  635. Ref<NetSocket> out;
  636. ERR_FAIL_COND_V(!is_open(), out);
  637. struct sockaddr_storage their_addr;
  638. socklen_t size = sizeof(their_addr);
  639. SOCKET_TYPE fd = ::accept(_sock, (struct sockaddr *)&their_addr, &size);
  640. if (fd == SOCK_EMPTY) {
  641. _get_socket_error();
  642. print_verbose("Error when accepting socket connection.");
  643. return out;
  644. }
  645. _set_ip_port(&their_addr, &r_ip, &r_port);
  646. NetSocketPosix *ns = memnew(NetSocketPosix);
  647. ns->_set_socket(fd, _ip_type, _is_stream);
  648. ns->set_blocking_enabled(false);
  649. return Ref<NetSocket>(ns);
  650. }
  651. Error NetSocketPosix::join_multicast_group(const IPAddress &p_multi_address, String p_if_name) {
  652. return _change_multicast_group(p_multi_address, p_if_name, true);
  653. }
  654. Error NetSocketPosix::leave_multicast_group(const IPAddress &p_multi_address, String p_if_name) {
  655. return _change_multicast_group(p_multi_address, p_if_name, false);
  656. }
  657. #endif // UNIX_SOCKET_UNAVAILABLE