net_socket_posix.cpp 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717
  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-2019 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2019 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. #ifndef UNIX_SOCKET_UNAVAILABLE
  32. #if defined(UNIX_ENABLED)
  33. #include <errno.h>
  34. #include <netdb.h>
  35. #include <poll.h>
  36. #include <stdio.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include <sys/ioctl.h>
  40. #include <sys/types.h>
  41. #include <unistd.h>
  42. #ifndef NO_FCNTL
  43. #include <fcntl.h>
  44. #else
  45. #include <sys/ioctl.h>
  46. #endif
  47. #include <netinet/in.h>
  48. #include <sys/socket.h>
  49. #ifdef JAVASCRIPT_ENABLED
  50. #include <arpa/inet.h>
  51. #endif
  52. #include <netinet/tcp.h>
  53. // BSD calls this flag IPV6_JOIN_GROUP
  54. #if !defined(IPV6_ADD_MEMBERSHIP) && defined(IPV6_JOIN_GROUP)
  55. #define IPV6_ADD_MEMBERSHIP IPV6_JOIN_GROUP
  56. #endif
  57. #if !defined(IPV6_DROP_MEMBERSHIP) && defined(IPV6_LEAVE_GROUP)
  58. #define IPV6_DROP_MEMBERSHIP IPV6_LEAVE_GROUP
  59. #endif
  60. // Some custom defines to minimize ifdefs
  61. #define SOCK_EMPTY -1
  62. #define SOCK_BUF(x) x
  63. #define SOCK_CBUF(x) x
  64. #define SOCK_IOCTL ioctl
  65. #define SOCK_CLOSE ::close
  66. #define SOCK_CONNECT(p_sock, p_addr, p_addr_len) ::connect(p_sock, p_addr, p_addr_len)
  67. /* Windows */
  68. #elif defined(WINDOWS_ENABLED)
  69. #include <winsock2.h>
  70. #include <ws2tcpip.h>
  71. #include <mswsock.h>
  72. // Some custom defines to minimize ifdefs
  73. #define SOCK_EMPTY INVALID_SOCKET
  74. #define SOCK_BUF(x) (char *)(x)
  75. #define SOCK_CBUF(x) (const char *)(x)
  76. #define SOCK_IOCTL ioctlsocket
  77. #define SOCK_CLOSE closesocket
  78. // connect is broken on windows under certain conditions, reasons unknown:
  79. // See https://github.com/godotengine/webrtc-native/issues/6
  80. #define SOCK_CONNECT(p_sock, p_addr, p_addr_len) ::WSAConnect(p_sock, p_addr, p_addr_len, NULL, NULL, NULL, NULL)
  81. // Workaround missing flag in MinGW
  82. #if defined(__MINGW32__) && !defined(SIO_UDP_NETRESET)
  83. #define SIO_UDP_NETRESET _WSAIOW(IOC_VENDOR, 15)
  84. #endif
  85. #endif
  86. size_t NetSocketPosix::_set_addr_storage(struct sockaddr_storage *p_addr, const IP_Address &p_ip, uint16_t p_port, IP::Type p_ip_type) {
  87. memset(p_addr, 0, sizeof(struct sockaddr_storage));
  88. if (p_ip_type == IP::TYPE_IPV6 || p_ip_type == IP::TYPE_ANY) { // IPv6 socket
  89. // IPv6 only socket with IPv4 address
  90. ERR_FAIL_COND_V(!p_ip.is_wildcard() && p_ip_type == IP::TYPE_IPV6 && p_ip.is_ipv4(), 0);
  91. struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)p_addr;
  92. addr6->sin6_family = AF_INET6;
  93. addr6->sin6_port = htons(p_port);
  94. if (p_ip.is_valid()) {
  95. copymem(&addr6->sin6_addr.s6_addr, p_ip.get_ipv6(), 16);
  96. } else {
  97. addr6->sin6_addr = in6addr_any;
  98. }
  99. return sizeof(sockaddr_in6);
  100. } else { // IPv4 socket
  101. // IPv4 socket with IPv6 address
  102. ERR_FAIL_COND_V(!p_ip.is_wildcard() && !p_ip.is_ipv4(), 0);
  103. struct sockaddr_in *addr4 = (struct sockaddr_in *)p_addr;
  104. addr4->sin_family = AF_INET;
  105. addr4->sin_port = htons(p_port); // short, network byte order
  106. if (p_ip.is_valid()) {
  107. copymem(&addr4->sin_addr.s_addr, p_ip.get_ipv4(), 4);
  108. } else {
  109. addr4->sin_addr.s_addr = INADDR_ANY;
  110. }
  111. return sizeof(sockaddr_in);
  112. }
  113. }
  114. void NetSocketPosix::_set_ip_port(struct sockaddr_storage *p_addr, IP_Address &r_ip, uint16_t &r_port) {
  115. if (p_addr->ss_family == AF_INET) {
  116. struct sockaddr_in *addr4 = (struct sockaddr_in *)p_addr;
  117. r_ip.set_ipv4((uint8_t *)&(addr4->sin_addr.s_addr));
  118. r_port = ntohs(addr4->sin_port);
  119. } else if (p_addr->ss_family == AF_INET6) {
  120. struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *)p_addr;
  121. r_ip.set_ipv6(addr6->sin6_addr.s6_addr);
  122. r_port = ntohs(addr6->sin6_port);
  123. };
  124. }
  125. NetSocket *NetSocketPosix::_create_func() {
  126. return memnew(NetSocketPosix);
  127. }
  128. void NetSocketPosix::make_default() {
  129. #if defined(WINDOWS_ENABLED)
  130. if (_create == NULL) {
  131. WSADATA data;
  132. WSAStartup(MAKEWORD(2, 2), &data);
  133. }
  134. #endif
  135. _create = _create_func;
  136. }
  137. void NetSocketPosix::cleanup() {
  138. #if defined(WINDOWS_ENABLED)
  139. if (_create != NULL) {
  140. WSACleanup();
  141. }
  142. _create = NULL;
  143. #endif
  144. }
  145. NetSocketPosix::NetSocketPosix() :
  146. _sock(SOCK_EMPTY),
  147. _ip_type(IP::TYPE_NONE),
  148. _is_stream(false) {
  149. }
  150. NetSocketPosix::~NetSocketPosix() {
  151. close();
  152. }
  153. // Silent a warning reported in #27594
  154. #if defined(__GNUC__) && !defined(__clang__)
  155. #pragma GCC diagnostic push
  156. #pragma GCC diagnostic ignored "-Wlogical-op"
  157. #endif
  158. NetSocketPosix::NetError NetSocketPosix::_get_socket_error() {
  159. #if defined(WINDOWS_ENABLED)
  160. int err = WSAGetLastError();
  161. if (err == WSAEISCONN)
  162. return ERR_NET_IS_CONNECTED;
  163. if (err == WSAEINPROGRESS || err == WSAEALREADY)
  164. return ERR_NET_IN_PROGRESS;
  165. if (err == WSAEWOULDBLOCK)
  166. return ERR_NET_WOULD_BLOCK;
  167. ERR_PRINTS("Socket error: " + itos(err));
  168. return ERR_NET_OTHER;
  169. #else
  170. if (errno == EISCONN)
  171. return ERR_NET_IS_CONNECTED;
  172. if (errno == EINPROGRESS || errno == EALREADY)
  173. return ERR_NET_IN_PROGRESS;
  174. if (errno == EAGAIN || errno == EWOULDBLOCK)
  175. return ERR_NET_WOULD_BLOCK;
  176. ERR_PRINTS("Socket error: " + itos(errno));
  177. return ERR_NET_OTHER;
  178. #endif
  179. }
  180. #if defined(__GNUC__) && !defined(__clang__)
  181. #pragma GCC diagnostic pop
  182. #endif
  183. bool NetSocketPosix::_can_use_ip(const IP_Address &p_ip, const bool p_for_bind) const {
  184. if (p_for_bind && !(p_ip.is_valid() || p_ip.is_wildcard())) {
  185. return false;
  186. } else if (!p_for_bind && !p_ip.is_valid()) {
  187. return false;
  188. }
  189. // Check if socket support this IP type.
  190. IP::Type type = p_ip.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6;
  191. return !(_ip_type != IP::TYPE_ANY && !p_ip.is_wildcard() && _ip_type != type);
  192. }
  193. _FORCE_INLINE_ Error NetSocketPosix::_change_multicast_group(IP_Address p_ip, String p_if_name, bool p_add) {
  194. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  195. ERR_FAIL_COND_V(!_can_use_ip(p_ip, false), ERR_INVALID_PARAMETER);
  196. // Need to force level and af_family to IP(v4) when using dual stacking and provided multicast group is IPv4
  197. IP::Type type = _ip_type == IP::TYPE_ANY && p_ip.is_ipv4() ? IP::TYPE_IPV4 : _ip_type;
  198. // This needs to be the proper level for the multicast group, no matter if the socket is dual stacking.
  199. int level = type == IP::TYPE_IPV4 ? IPPROTO_IP : IPPROTO_IPV6;
  200. int ret = -1;
  201. IP_Address if_ip;
  202. uint32_t if_v6id = 0;
  203. Map<String, IP::Interface_Info> if_info;
  204. IP::get_singleton()->get_local_interfaces(&if_info);
  205. for (Map<String, IP::Interface_Info>::Element *E = if_info.front(); E; E = E->next()) {
  206. IP::Interface_Info &c = E->get();
  207. if (c.name != p_if_name)
  208. continue;
  209. if_v6id = (uint32_t)c.index.to_int64();
  210. if (type == IP::TYPE_IPV6)
  211. break; // IPv6 uses index.
  212. for (List<IP_Address>::Element *F = c.ip_addresses.front(); F; F = F->next()) {
  213. if (!F->get().is_ipv4())
  214. continue; // Wrong IP type
  215. if_ip = F->get();
  216. break;
  217. }
  218. break;
  219. }
  220. if (level == IPPROTO_IP) {
  221. ERR_FAIL_COND_V(!if_ip.is_valid(), ERR_INVALID_PARAMETER);
  222. struct ip_mreq greq;
  223. int sock_opt = p_add ? IP_ADD_MEMBERSHIP : IP_DROP_MEMBERSHIP;
  224. copymem(&greq.imr_multiaddr, p_ip.get_ipv4(), 4);
  225. copymem(&greq.imr_interface, if_ip.get_ipv4(), 4);
  226. ret = setsockopt(_sock, level, sock_opt, (const char *)&greq, sizeof(greq));
  227. } else {
  228. struct ipv6_mreq greq;
  229. int sock_opt = p_add ? IPV6_ADD_MEMBERSHIP : IPV6_DROP_MEMBERSHIP;
  230. copymem(&greq.ipv6mr_multiaddr, p_ip.get_ipv6(), 16);
  231. greq.ipv6mr_interface = if_v6id;
  232. ret = setsockopt(_sock, level, sock_opt, (const char *)&greq, sizeof(greq));
  233. }
  234. ERR_FAIL_COND_V(ret != 0, FAILED);
  235. return OK;
  236. }
  237. void NetSocketPosix::_set_socket(SOCKET_TYPE p_sock, IP::Type p_ip_type, bool p_is_stream) {
  238. _sock = p_sock;
  239. _ip_type = p_ip_type;
  240. _is_stream = p_is_stream;
  241. // Disable descriptor sharing with subprocesses.
  242. _set_close_exec_enabled(true);
  243. }
  244. void NetSocketPosix::_set_close_exec_enabled(bool p_enabled) {
  245. #ifndef WINDOWS_ENABLED
  246. // Enable close on exec to avoid sharing with subprocesses. Off by default on Windows.
  247. #if defined(NO_FCNTL)
  248. unsigned long par = p_enabled ? 1 : 0;
  249. SOCK_IOCTL(_sock, FIOCLEX, &par);
  250. #else
  251. int opts = fcntl(_sock, F_GETFD);
  252. fcntl(_sock, F_SETFD, opts | FD_CLOEXEC);
  253. #endif
  254. #endif
  255. }
  256. Error NetSocketPosix::open(Type p_sock_type, IP::Type &ip_type) {
  257. ERR_FAIL_COND_V(is_open(), ERR_ALREADY_IN_USE);
  258. ERR_FAIL_COND_V(ip_type > IP::TYPE_ANY || ip_type < IP::TYPE_NONE, ERR_INVALID_PARAMETER);
  259. #if defined(__OpenBSD__)
  260. // OpenBSD does not support dual stacking, fallback to IPv4 only.
  261. if (ip_type == IP::TYPE_ANY)
  262. ip_type = IP::TYPE_IPV4;
  263. #endif
  264. int family = ip_type == IP::TYPE_IPV4 ? AF_INET : AF_INET6;
  265. int protocol = p_sock_type == TYPE_TCP ? IPPROTO_TCP : IPPROTO_UDP;
  266. int type = p_sock_type == TYPE_TCP ? SOCK_STREAM : SOCK_DGRAM;
  267. _sock = socket(family, type, protocol);
  268. if (_sock == SOCK_EMPTY && ip_type == IP::TYPE_ANY) {
  269. // Careful here, changing the referenced parameter so the caller knows that we are using an IPv4 socket
  270. // in place of a dual stack one, and further calls to _set_sock_addr will work as expected.
  271. ip_type = IP::TYPE_IPV4;
  272. family = AF_INET;
  273. _sock = socket(family, type, protocol);
  274. }
  275. ERR_FAIL_COND_V(_sock == SOCK_EMPTY, FAILED);
  276. _ip_type = ip_type;
  277. if (family == AF_INET6) {
  278. // Select IPv4 over IPv6 mapping
  279. set_ipv6_only_enabled(ip_type != IP::TYPE_ANY);
  280. }
  281. if (protocol == IPPROTO_UDP && ip_type != IP::TYPE_IPV6) {
  282. // Enable broadcasting for UDP sockets if it's not IPv6 only (IPv6 has no broadcast option).
  283. set_broadcasting_enabled(true);
  284. }
  285. _is_stream = p_sock_type == TYPE_TCP;
  286. // Disable descriptor sharing with subprocesses.
  287. _set_close_exec_enabled(true);
  288. #if defined(WINDOWS_ENABLED)
  289. if (!_is_stream) {
  290. // Disable windows feature/bug reporting WSAECONNRESET/WSAENETRESET when
  291. // recv/recvfrom and an ICMP reply was received from a previous send/sendto.
  292. unsigned long disable = 0;
  293. if (ioctlsocket(_sock, SIO_UDP_CONNRESET, &disable) == SOCKET_ERROR) {
  294. print_verbose("Unable to turn off UDP WSAECONNRESET behaviour on Windows");
  295. }
  296. if (ioctlsocket(_sock, SIO_UDP_NETRESET, &disable) == SOCKET_ERROR) {
  297. // This feature seems not to be supported on wine.
  298. print_verbose("Unable to turn off UDP WSAENETRESET behaviour on Windows");
  299. }
  300. }
  301. #endif
  302. #if defined(SO_NOSIGPIPE)
  303. // Disable SIGPIPE (should only be relevant to stream sockets, but seems to affect UDP too on iOS)
  304. int par = 1;
  305. if (setsockopt(_sock, SOL_SOCKET, SO_NOSIGPIPE, SOCK_CBUF(&par), sizeof(int)) != 0) {
  306. print_verbose("Unable to turn off SIGPIPE on socket");
  307. }
  308. #endif
  309. return OK;
  310. }
  311. void NetSocketPosix::close() {
  312. if (_sock != SOCK_EMPTY)
  313. SOCK_CLOSE(_sock);
  314. _sock = SOCK_EMPTY;
  315. _ip_type = IP::TYPE_NONE;
  316. _is_stream = false;
  317. }
  318. Error NetSocketPosix::bind(IP_Address p_addr, uint16_t p_port) {
  319. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  320. ERR_FAIL_COND_V(!_can_use_ip(p_addr, true), ERR_INVALID_PARAMETER);
  321. sockaddr_storage addr;
  322. size_t addr_size = _set_addr_storage(&addr, p_addr, p_port, _ip_type);
  323. if (::bind(_sock, (struct sockaddr *)&addr, addr_size) != 0) {
  324. close();
  325. ERR_FAIL_V(ERR_UNAVAILABLE);
  326. }
  327. return OK;
  328. }
  329. Error NetSocketPosix::listen(int p_max_pending) {
  330. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  331. if (::listen(_sock, p_max_pending) != 0) {
  332. close();
  333. ERR_FAIL_V(FAILED);
  334. };
  335. return OK;
  336. }
  337. Error NetSocketPosix::connect_to_host(IP_Address p_host, uint16_t p_port) {
  338. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  339. ERR_FAIL_COND_V(!_can_use_ip(p_host, false), ERR_INVALID_PARAMETER);
  340. struct sockaddr_storage addr;
  341. size_t addr_size = _set_addr_storage(&addr, p_host, p_port, _ip_type);
  342. if (SOCK_CONNECT(_sock, (struct sockaddr *)&addr, addr_size) != 0) {
  343. NetError err = _get_socket_error();
  344. switch (err) {
  345. // We are already connected
  346. case ERR_NET_IS_CONNECTED:
  347. return OK;
  348. // Still waiting to connect, try again in a while
  349. case ERR_NET_WOULD_BLOCK:
  350. case ERR_NET_IN_PROGRESS:
  351. return ERR_BUSY;
  352. default:
  353. ERR_PRINT("Connection to remote host failed!");
  354. close();
  355. return FAILED;
  356. }
  357. }
  358. return OK;
  359. }
  360. Error NetSocketPosix::poll(PollType p_type, int p_timeout) const {
  361. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  362. #if defined(WINDOWS_ENABLED)
  363. bool ready = false;
  364. fd_set rd, wr, ex;
  365. fd_set *rdp = NULL;
  366. fd_set *wrp = NULL;
  367. FD_ZERO(&rd);
  368. FD_ZERO(&wr);
  369. FD_ZERO(&ex);
  370. FD_SET(_sock, &ex);
  371. struct timeval timeout = { p_timeout, 0 };
  372. // For blocking operation, pass NULL timeout pointer to select.
  373. struct timeval *tp = NULL;
  374. if (p_timeout >= 0) {
  375. // If timeout is non-negative, we want to specify the timeout instead.
  376. tp = &timeout;
  377. }
  378. switch (p_type) {
  379. case POLL_TYPE_IN:
  380. FD_SET(_sock, &rd);
  381. rdp = &rd;
  382. break;
  383. case POLL_TYPE_OUT:
  384. FD_SET(_sock, &wr);
  385. wrp = &wr;
  386. break;
  387. case POLL_TYPE_IN_OUT:
  388. FD_SET(_sock, &rd);
  389. FD_SET(_sock, &wr);
  390. rdp = &rd;
  391. wrp = &wr;
  392. }
  393. int ret = select(1, rdp, wrp, &ex, tp);
  394. ERR_FAIL_COND_V(ret == SOCKET_ERROR, FAILED);
  395. if (ret == 0)
  396. return ERR_BUSY;
  397. ERR_FAIL_COND_V(FD_ISSET(_sock, &ex), FAILED);
  398. if (rdp && FD_ISSET(_sock, rdp))
  399. ready = true;
  400. if (wrp && FD_ISSET(_sock, wrp))
  401. ready = true;
  402. return ready ? OK : ERR_BUSY;
  403. #else
  404. struct pollfd pfd;
  405. pfd.fd = _sock;
  406. pfd.events = POLLIN;
  407. pfd.revents = 0;
  408. switch (p_type) {
  409. case POLL_TYPE_IN:
  410. pfd.events = POLLIN;
  411. break;
  412. case POLL_TYPE_OUT:
  413. pfd.events = POLLOUT;
  414. break;
  415. case POLL_TYPE_IN_OUT:
  416. pfd.events = POLLOUT | POLLIN;
  417. }
  418. int ret = ::poll(&pfd, 1, p_timeout);
  419. ERR_FAIL_COND_V(ret < 0, FAILED);
  420. ERR_FAIL_COND_V(pfd.revents & POLLERR, FAILED);
  421. if (ret == 0)
  422. return ERR_BUSY;
  423. return OK;
  424. #endif
  425. }
  426. Error NetSocketPosix::recv(uint8_t *p_buffer, int p_len, int &r_read) {
  427. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  428. r_read = ::recv(_sock, SOCK_BUF(p_buffer), p_len, 0);
  429. if (r_read < 0) {
  430. NetError err = _get_socket_error();
  431. if (err == ERR_NET_WOULD_BLOCK)
  432. return ERR_BUSY;
  433. return FAILED;
  434. }
  435. return OK;
  436. }
  437. Error NetSocketPosix::recvfrom(uint8_t *p_buffer, int p_len, int &r_read, IP_Address &r_ip, uint16_t &r_port) {
  438. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  439. struct sockaddr_storage from;
  440. socklen_t len = sizeof(struct sockaddr_storage);
  441. memset(&from, 0, len);
  442. r_read = ::recvfrom(_sock, SOCK_BUF(p_buffer), p_len, 0, (struct sockaddr *)&from, &len);
  443. if (r_read < 0) {
  444. NetError err = _get_socket_error();
  445. if (err == ERR_NET_WOULD_BLOCK)
  446. return ERR_BUSY;
  447. return FAILED;
  448. }
  449. if (from.ss_family == AF_INET) {
  450. struct sockaddr_in *sin_from = (struct sockaddr_in *)&from;
  451. r_ip.set_ipv4((uint8_t *)&sin_from->sin_addr);
  452. r_port = ntohs(sin_from->sin_port);
  453. } else if (from.ss_family == AF_INET6) {
  454. struct sockaddr_in6 *s6_from = (struct sockaddr_in6 *)&from;
  455. r_ip.set_ipv6((uint8_t *)&s6_from->sin6_addr);
  456. r_port = ntohs(s6_from->sin6_port);
  457. } else {
  458. // Unsupported socket family, should never happen.
  459. ERR_FAIL_V(FAILED);
  460. }
  461. return OK;
  462. }
  463. Error NetSocketPosix::send(const uint8_t *p_buffer, int p_len, int &r_sent) {
  464. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  465. int flags = 0;
  466. #ifdef MSG_NOSIGNAL
  467. if (_is_stream)
  468. flags = MSG_NOSIGNAL;
  469. #endif
  470. r_sent = ::send(_sock, SOCK_CBUF(p_buffer), p_len, flags);
  471. if (r_sent < 0) {
  472. NetError err = _get_socket_error();
  473. if (err == ERR_NET_WOULD_BLOCK)
  474. return ERR_BUSY;
  475. return FAILED;
  476. }
  477. return OK;
  478. }
  479. Error NetSocketPosix::sendto(const uint8_t *p_buffer, int p_len, int &r_sent, IP_Address p_ip, uint16_t p_port) {
  480. ERR_FAIL_COND_V(!is_open(), ERR_UNCONFIGURED);
  481. struct sockaddr_storage addr;
  482. size_t addr_size = _set_addr_storage(&addr, p_ip, p_port, _ip_type);
  483. r_sent = ::sendto(_sock, SOCK_CBUF(p_buffer), p_len, 0, (struct sockaddr *)&addr, addr_size);
  484. if (r_sent < 0) {
  485. NetError err = _get_socket_error();
  486. if (err == ERR_NET_WOULD_BLOCK)
  487. return ERR_BUSY;
  488. return FAILED;
  489. }
  490. return OK;
  491. }
  492. void NetSocketPosix::set_broadcasting_enabled(bool p_enabled) {
  493. ERR_FAIL_COND(!is_open());
  494. // IPv6 has no broadcast support.
  495. ERR_FAIL_COND(_ip_type == IP::TYPE_IPV6);
  496. int par = p_enabled ? 1 : 0;
  497. if (setsockopt(_sock, SOL_SOCKET, SO_BROADCAST, SOCK_CBUF(&par), sizeof(int)) != 0) {
  498. WARN_PRINT("Unable to change broadcast setting");
  499. }
  500. }
  501. void NetSocketPosix::set_blocking_enabled(bool p_enabled) {
  502. ERR_FAIL_COND(!is_open());
  503. int ret = 0;
  504. #if defined(WINDOWS_ENABLED) || defined(NO_FCNTL)
  505. unsigned long par = p_enabled ? 0 : 1;
  506. ret = SOCK_IOCTL(_sock, FIONBIO, &par);
  507. #else
  508. int opts = fcntl(_sock, F_GETFL);
  509. if (p_enabled)
  510. ret = fcntl(_sock, F_SETFL, opts & ~O_NONBLOCK);
  511. else
  512. ret = fcntl(_sock, F_SETFL, opts | O_NONBLOCK);
  513. #endif
  514. if (ret != 0)
  515. WARN_PRINT("Unable to change non-block mode");
  516. }
  517. void NetSocketPosix::set_ipv6_only_enabled(bool p_enabled) {
  518. ERR_FAIL_COND(!is_open());
  519. // This option is only available in IPv6 sockets.
  520. ERR_FAIL_COND(_ip_type == IP::TYPE_IPV4);
  521. int par = p_enabled ? 1 : 0;
  522. if (setsockopt(_sock, IPPROTO_IPV6, IPV6_V6ONLY, SOCK_CBUF(&par), sizeof(int)) != 0) {
  523. WARN_PRINT("Unable to change IPv4 address mapping over IPv6 option");
  524. }
  525. }
  526. void NetSocketPosix::set_tcp_no_delay_enabled(bool p_enabled) {
  527. ERR_FAIL_COND(!is_open());
  528. ERR_FAIL_COND(!_is_stream); // Not TCP
  529. int par = p_enabled ? 1 : 0;
  530. if (setsockopt(_sock, IPPROTO_TCP, TCP_NODELAY, SOCK_CBUF(&par), sizeof(int)) < 0) {
  531. ERR_PRINT("Unable to set TCP no delay option");
  532. }
  533. }
  534. void NetSocketPosix::set_reuse_address_enabled(bool p_enabled) {
  535. ERR_FAIL_COND(!is_open());
  536. int par = p_enabled ? 1 : 0;
  537. if (setsockopt(_sock, SOL_SOCKET, SO_REUSEADDR, SOCK_CBUF(&par), sizeof(int)) < 0) {
  538. WARN_PRINT("Unable to set socket REUSEADDR option!");
  539. }
  540. }
  541. void NetSocketPosix::set_reuse_port_enabled(bool p_enabled) {
  542. // Windows does not have this option, as it is always ON when setting REUSEADDR.
  543. #ifndef WINDOWS_ENABLED
  544. ERR_FAIL_COND(!is_open());
  545. int par = p_enabled ? 1 : 0;
  546. if (setsockopt(_sock, SOL_SOCKET, SO_REUSEPORT, SOCK_CBUF(&par), sizeof(int)) < 0) {
  547. WARN_PRINT("Unable to set socket REUSEPORT option!");
  548. }
  549. #endif
  550. }
  551. bool NetSocketPosix::is_open() const {
  552. return _sock != SOCK_EMPTY;
  553. }
  554. int NetSocketPosix::get_available_bytes() const {
  555. ERR_FAIL_COND_V(_sock == SOCK_EMPTY, -1);
  556. unsigned long len;
  557. int ret = SOCK_IOCTL(_sock, FIONREAD, &len);
  558. ERR_FAIL_COND_V(ret == -1, 0);
  559. return len;
  560. }
  561. Ref<NetSocket> NetSocketPosix::accept(IP_Address &r_ip, uint16_t &r_port) {
  562. Ref<NetSocket> out;
  563. ERR_FAIL_COND_V(!is_open(), out);
  564. struct sockaddr_storage their_addr;
  565. socklen_t size = sizeof(their_addr);
  566. SOCKET_TYPE fd = ::accept(_sock, (struct sockaddr *)&their_addr, &size);
  567. ERR_FAIL_COND_V(fd == SOCK_EMPTY, out);
  568. _set_ip_port(&their_addr, r_ip, r_port);
  569. NetSocketPosix *ns = memnew(NetSocketPosix);
  570. ns->_set_socket(fd, _ip_type, _is_stream);
  571. ns->set_blocking_enabled(false);
  572. return Ref<NetSocket>(ns);
  573. }
  574. Error NetSocketPosix::join_multicast_group(const IP_Address &p_multi_address, String p_if_name) {
  575. return _change_multicast_group(p_multi_address, p_if_name, true);
  576. }
  577. Error NetSocketPosix::leave_multicast_group(const IP_Address &p_multi_address, String p_if_name) {
  578. return _change_multicast_group(p_multi_address, p_if_name, false);
  579. }
  580. #endif