소스 검색

TCP connect always opens correct socket type

TCP client connections does not need to rely on ipv6 dual stack sockets

(cherry picked from commit 55b4f3686d8f51958132a1a1745cc4e128fd118d)
Fabio Alessandrelli 8 년 전
부모
커밋
ccf37c4ca2

+ 8 - 3
drivers/unix/stream_peer_tcp_posix.cpp

@@ -98,7 +98,7 @@ Error StreamPeerTCPPosix::_poll_connection(bool p_block) const {
 	};
 	};
 
 
 	struct sockaddr_storage their_addr;
 	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) == -1) {
 	if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == -1) {
 
 
@@ -125,6 +125,7 @@ Error StreamPeerTCPPosix::_poll_connection(bool p_block) const {
 void StreamPeerTCPPosix::set_socket(int p_sockfd, IP_Address p_host, int p_port, IP::Type p_ip_type) {
 void StreamPeerTCPPosix::set_socket(int p_sockfd, IP_Address p_host, int p_port, IP::Type p_ip_type) {
 
 
 	ip_type = p_ip_type;
 	ip_type = p_ip_type;
+	sock_type = p_ip_type;
 	sockfd = p_sockfd;
 	sockfd = p_sockfd;
 #ifndef NO_FCNTL
 #ifndef NO_FCNTL
 	fcntl(sockfd, F_SETFL, O_NONBLOCK);
 	fcntl(sockfd, F_SETFL, O_NONBLOCK);
@@ -143,7 +144,8 @@ Error StreamPeerTCPPosix::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 == IP_Address(), 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 == -1) {
 	if (sockfd == -1) {
 		ERR_PRINT("Socket creation failed!");
 		ERR_PRINT("Socket creation failed!");
 		disconnect();
 		disconnect();
@@ -159,7 +161,7 @@ Error StreamPeerTCPPosix::connect(const IP_Address &p_host, uint16_t p_port) {
 #endif
 #endif
 
 
 	struct sockaddr_storage their_addr;
 	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);
 
 
 	errno = 0;
 	errno = 0;
 	if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == -1 && errno != EINPROGRESS) {
 	if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == -1 && errno != EINPROGRESS) {
@@ -339,6 +341,8 @@ void StreamPeerTCPPosix::disconnect() {
 
 
 	if (sockfd != -1)
 	if (sockfd != -1)
 		close(sockfd);
 		close(sockfd);
+
+	sock_type = IP::TYPE_NONE;
 	sockfd = -1;
 	sockfd = -1;
 
 
 	status = STATUS_NONE;
 	status = STATUS_NONE;
@@ -387,6 +391,7 @@ uint16_t StreamPeerTCPPosix::get_connected_port() const {
 
 
 StreamPeerTCPPosix::StreamPeerTCPPosix() {
 StreamPeerTCPPosix::StreamPeerTCPPosix() {
 
 
+	sock_type = IP::TYPE_NONE;
 	sockfd = -1;
 	sockfd = -1;
 	status = STATUS_NONE;
 	status = STATUS_NONE;
 	peer_port = 0;
 	peer_port = 0;

+ 1 - 2
drivers/unix/stream_peer_tcp_posix.h

@@ -35,13 +35,12 @@
 #include "core/io/stream_peer_tcp.h"
 #include "core/io/stream_peer_tcp.h"
 #include "error_list.h"
 #include "error_list.h"
 
 
-#include "core/io/ip_address.h"
-
 class StreamPeerTCPPosix : public StreamPeerTCP {
 class StreamPeerTCPPosix : public StreamPeerTCP {
 
 
 protected:
 protected:
 	mutable Status status;
 	mutable Status status;
 
 
+	IP::Type sock_type;
 	int sockfd;
 	int sockfd;
 
 
 	Error _block(int p_sockfd, bool p_read, bool p_write) const;
 	Error _block(int p_sockfd, bool p_read, bool p_write) const;

+ 7 - 3
platform/windows/stream_peer_winsock.cpp

@@ -87,7 +87,7 @@ Error StreamPeerWinsock::_poll_connection(bool p_block) const {
 	};
 	};
 
 
 	struct sockaddr_storage their_addr;
 	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) {
 	if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == SOCKET_ERROR) {
 
 
@@ -277,6 +277,7 @@ void StreamPeerWinsock::disconnect() {
 	if (sockfd != INVALID_SOCKET)
 	if (sockfd != INVALID_SOCKET)
 		closesocket(sockfd);
 		closesocket(sockfd);
 	sockfd = INVALID_SOCKET;
 	sockfd = INVALID_SOCKET;
+	sock_type = IP::TYPE_NONE;
 
 
 	status = STATUS_NONE;
 	status = STATUS_NONE;
 
 
@@ -288,6 +289,7 @@ void StreamPeerWinsock::set_socket(int p_sockfd, IP_Address p_host, int p_port,
 
 
 	ip_type = p_ip_type;
 	ip_type = p_ip_type;
 	sockfd = p_sockfd;
 	sockfd = p_sockfd;
+	sock_type = p_ip_type;
 	status = STATUS_CONNECTING;
 	status = STATUS_CONNECTING;
 	peer_host = p_host;
 	peer_host = p_host;
 	peer_port = p_port;
 	peer_port = p_port;
@@ -297,7 +299,8 @@ 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 == IP_Address(), 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) {
 	if (sockfd == INVALID_SOCKET) {
 		ERR_PRINT("Socket creation failed!");
 		ERR_PRINT("Socket creation failed!");
 		disconnect();
 		disconnect();
@@ -313,7 +316,7 @@ Error StreamPeerWinsock::connect(const IP_Address &p_host, uint16_t p_port) {
 	};
 	};
 
 
 	struct sockaddr_storage their_addr;
 	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) {
 	if (::connect(sockfd, (struct sockaddr *)&their_addr, addr_size) == SOCKET_ERROR) {
 
 
@@ -359,6 +362,7 @@ uint16_t StreamPeerWinsock::get_connected_port() const {
 
 
 StreamPeerWinsock::StreamPeerWinsock() {
 StreamPeerWinsock::StreamPeerWinsock() {
 
 
+	sock_type = IP::TYPE_NONE;
 	sockfd = INVALID_SOCKET;
 	sockfd = INVALID_SOCKET;
 	status = STATUS_NONE;
 	status = STATUS_NONE;
 	peer_port = 0;
 	peer_port = 0;

+ 1 - 0
platform/windows/stream_peer_winsock.h

@@ -40,6 +40,7 @@ class StreamPeerWinsock : public StreamPeerTCP {
 
 
 protected:
 protected:
 	mutable Status status;
 	mutable Status status;
+	IP::Type sock_type;
 
 
 	int sockfd;
 	int sockfd;