1 /*************************************************************************/ 2 /* net_socket_posix.h */ 3 /*************************************************************************/ 4 /* This file is part of: */ 5 /* GODOT ENGINE */ 6 /* https://godotengine.org */ 7 /*************************************************************************/ 8 /* Copyright (c) 2007-2020 Juan Linietsky, Ariel Manzur. */ 9 /* Copyright (c) 2014-2020 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 31 #ifndef NET_SOCKET_UNIX_H 32 #define NET_SOCKET_UNIX_H 33 34 #include "core/io/net_socket.h" 35 36 #if defined(WINDOWS_ENABLED) 37 #include <winsock2.h> 38 #include <ws2tcpip.h> 39 #define SOCKET_TYPE SOCKET 40 41 #else 42 #include <sys/socket.h> 43 #define SOCKET_TYPE int 44 45 #endif 46 47 class NetSocketPosix : public NetSocket { 48 49 private: 50 SOCKET_TYPE _sock; 51 IP::Type _ip_type; 52 bool _is_stream; 53 54 enum NetError { 55 ERR_NET_WOULD_BLOCK, 56 ERR_NET_IS_CONNECTED, 57 ERR_NET_IN_PROGRESS, 58 ERR_NET_OTHER 59 }; 60 61 NetError _get_socket_error() const; 62 void _set_socket(SOCKET_TYPE p_sock, IP::Type p_ip_type, bool p_is_stream); 63 _FORCE_INLINE_ Error _change_multicast_group(IP_Address p_ip, String p_if_name, bool p_add); 64 _FORCE_INLINE_ void _set_close_exec_enabled(bool p_enabled); 65 66 protected: 67 static NetSocket *_create_func(); 68 69 bool _can_use_ip(const IP_Address &p_ip, const bool p_for_bind) const; 70 71 public: 72 static void make_default(); 73 static void cleanup(); 74 static void _set_ip_port(struct sockaddr_storage *p_addr, IP_Address &r_ip, uint16_t &r_port); 75 static size_t _set_addr_storage(struct sockaddr_storage *p_addr, const IP_Address &p_ip, uint16_t p_port, IP::Type p_ip_type); 76 77 virtual Error open(Type p_sock_type, IP::Type &ip_type); 78 virtual void close(); 79 virtual Error bind(IP_Address p_addr, uint16_t p_port); 80 virtual Error listen(int p_max_pending); 81 virtual Error connect_to_host(IP_Address p_host, uint16_t p_port); 82 virtual Error poll(PollType p_type, int timeout) const; 83 virtual Error recv(uint8_t *p_buffer, int p_len, int &r_read); 84 virtual Error recvfrom(uint8_t *p_buffer, int p_len, int &r_read, IP_Address &r_ip, uint16_t &r_port, bool p_peek = false); 85 virtual Error send(const uint8_t *p_buffer, int p_len, int &r_sent); 86 virtual Error sendto(const uint8_t *p_buffer, int p_len, int &r_sent, IP_Address p_ip, uint16_t p_port); 87 virtual Ref<NetSocket> accept(IP_Address &r_ip, uint16_t &r_port); 88 89 virtual bool is_open() const; 90 virtual int get_available_bytes() const; 91 92 virtual Error set_broadcasting_enabled(bool p_enabled); 93 virtual void set_blocking_enabled(bool p_enabled); 94 virtual void set_ipv6_only_enabled(bool p_enabled); 95 virtual void set_tcp_no_delay_enabled(bool p_enabled); 96 virtual void set_reuse_address_enabled(bool p_enabled); 97 virtual void set_reuse_port_enabled(bool p_enabled); 98 virtual Error join_multicast_group(const IP_Address &p_multi_address, String p_if_name); 99 virtual Error leave_multicast_group(const IP_Address &p_multi_address, String p_if_name); 100 101 NetSocketPosix(); 102 ~NetSocketPosix(); 103 }; 104 105 #endif 106