|
@@ -77,17 +77,12 @@ Error StreamPeerWinsock::_block(int p_sockfd, bool p_read, bool p_write) const {
|
|
|
return ret < 0 ? FAILED : OK;
|
|
|
};
|
|
|
|
|
|
-Error StreamPeerWinsock::_poll_connection(bool p_block) const {
|
|
|
+Error StreamPeerWinsock::_poll_connection() const {
|
|
|
|
|
|
ERR_FAIL_COND_V(status != STATUS_CONNECTING || sockfd == INVALID_SOCKET, FAILED);
|
|
|
|
|
|
- if (p_block) {
|
|
|
-
|
|
|
- _block(sockfd, false, true);
|
|
|
- };
|
|
|
-
|
|
|
struct sockaddr_storage their_addr;
|
|
|
- size_t addr_size = _set_sockaddr(&their_addr, peer_host, peer_port, ip_type);
|
|
|
+ size_t addr_size = _set_sockaddr(&their_addr, peer_host, peer_port, sock_type);
|
|
|
|
|
|
if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == SOCKET_ERROR) {
|
|
|
|
|
@@ -121,7 +116,7 @@ Error StreamPeerWinsock::write(const uint8_t *p_data, int p_bytes, int &r_sent,
|
|
|
|
|
|
if (status != STATUS_CONNECTED) {
|
|
|
|
|
|
- if (_poll_connection(p_block) != OK) {
|
|
|
+ if (_poll_connection() != OK) {
|
|
|
|
|
|
return FAILED;
|
|
|
};
|
|
@@ -179,7 +174,7 @@ Error StreamPeerWinsock::read(uint8_t *p_buffer, int p_bytes, int &r_received, b
|
|
|
|
|
|
if (status != STATUS_CONNECTED) {
|
|
|
|
|
|
- if (_poll_connection(p_block) != OK) {
|
|
|
+ if (_poll_connection() != OK) {
|
|
|
|
|
|
return FAILED;
|
|
|
};
|
|
@@ -253,7 +248,7 @@ Error StreamPeerWinsock::get_partial_data(uint8_t *p_buffer, int p_bytes, int &r
|
|
|
StreamPeerTCP::Status StreamPeerWinsock::get_status() const {
|
|
|
|
|
|
if (status == STATUS_CONNECTING) {
|
|
|
- _poll_connection(false);
|
|
|
+ _poll_connection();
|
|
|
};
|
|
|
|
|
|
return status;
|
|
@@ -277,6 +272,7 @@ void StreamPeerWinsock::disconnect() {
|
|
|
if (sockfd != INVALID_SOCKET)
|
|
|
closesocket(sockfd);
|
|
|
sockfd = INVALID_SOCKET;
|
|
|
+ sock_type = IP::TYPE_NONE;
|
|
|
|
|
|
status = STATUS_NONE;
|
|
|
|
|
@@ -284,10 +280,10 @@ void StreamPeerWinsock::disconnect() {
|
|
|
peer_port = 0;
|
|
|
};
|
|
|
|
|
|
-void StreamPeerWinsock::set_socket(int p_sockfd, IP_Address p_host, int p_port, IP::Type p_ip_type) {
|
|
|
+void StreamPeerWinsock::set_socket(int p_sockfd, IP_Address p_host, int p_port, IP::Type p_sock_type) {
|
|
|
|
|
|
- ip_type = p_ip_type;
|
|
|
sockfd = p_sockfd;
|
|
|
+ sock_type = p_sock_type;
|
|
|
status = STATUS_CONNECTING;
|
|
|
peer_host = p_host;
|
|
|
peer_port = p_port;
|
|
@@ -295,9 +291,10 @@ void StreamPeerWinsock::set_socket(int p_sockfd, IP_Address p_host, int p_port,
|
|
|
|
|
|
Error StreamPeerWinsock::connect(const IP_Address &p_host, uint16_t p_port) {
|
|
|
|
|
|
- ERR_FAIL_COND_V(p_host == IP_Address(), ERR_INVALID_PARAMETER);
|
|
|
+ ERR_FAIL_COND_V(!p_host.is_valid(), ERR_INVALID_PARAMETER);
|
|
|
|
|
|
- sockfd = _socket_create(ip_type, SOCK_STREAM, IPPROTO_TCP);
|
|
|
+ sock_type = p_host.is_ipv4() ? IP::TYPE_IPV4 : IP::TYPE_IPV6;
|
|
|
+ sockfd = _socket_create(sock_type, SOCK_STREAM, IPPROTO_TCP);
|
|
|
if (sockfd == INVALID_SOCKET) {
|
|
|
ERR_PRINT("Socket creation failed!");
|
|
|
disconnect();
|
|
@@ -313,7 +310,7 @@ Error StreamPeerWinsock::connect(const IP_Address &p_host, uint16_t p_port) {
|
|
|
};
|
|
|
|
|
|
struct sockaddr_storage their_addr;
|
|
|
- size_t addr_size = _set_sockaddr(&their_addr, p_host, p_port, ip_type);
|
|
|
+ size_t addr_size = _set_sockaddr(&their_addr, p_host, p_port, sock_type);
|
|
|
|
|
|
if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == SOCKET_ERROR) {
|
|
|
|
|
@@ -359,10 +356,10 @@ uint16_t StreamPeerWinsock::get_connected_port() const {
|
|
|
|
|
|
StreamPeerWinsock::StreamPeerWinsock() {
|
|
|
|
|
|
+ sock_type = IP::TYPE_NONE;
|
|
|
sockfd = INVALID_SOCKET;
|
|
|
status = STATUS_NONE;
|
|
|
peer_port = 0;
|
|
|
- ip_type = IP::TYPE_ANY;
|
|
|
};
|
|
|
|
|
|
StreamPeerWinsock::~StreamPeerWinsock() {
|