asgardius/virtualx-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #22020 from Faless/tcp_fix

Browse source 
Fix TCP connecting state not set correctly
This commit is contained in:
Rémi Verschelde 2018-09-13 11:04:01 +02:00 committed by GitHub
commit 3d0638b8e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
core/io/stream_peer_tcp.cpp
View file

@ -44,6 +44,7 @@ Error StreamPeerTCP::_poll_connection() {
return OK;
}
disconnect_from_host();
status = STATUS_ERROR;
return ERR_CONNECTION_ERROR;
}
@ -75,17 +76,16 @@ Error StreamPeerTCP::connect_to_host(const IP_Address &p_host, uint16_t p_port)
err = _sock->connect_to_host(p_host, p_port);
if (err != OK) {
if (err == ERR_BUSY) {
status = STATUS_CONNECTING;
} else {
ERR_PRINT("Connection to remote host failed!");
disconnect_from_host();
return FAILED;
}
if (err == OK) {
status = STATUS_CONNECTED;
} else if (err == ERR_BUSY) {
status = STATUS_CONNECTING;
} else {
ERR_PRINT("Connection to remote host failed!");
disconnect_from_host();
return FAILED;
}
status = STATUS_CONNECTED;
peer_host = p_host;
peer_port = p_port;
@ -163,20 +163,20 @@ Error StreamPeerTCP::read(uint8_t *p_buffer, int p_bytes, int &r_received, bool
if (!is_connected_to_host()) {
return FAILED;
};
}
if (status == STATUS_CONNECTING) {
if (_poll_connection() != OK) {
return FAILED;
};
}
if (status != STATUS_CONNECTED) {
r_received = 0;
return OK;
};
};
}
}
Error err;
int to_read = p_bytes;
@ -209,10 +209,7 @@ Error StreamPeerTCP::read(uint8_t *p_buffer, int p_bytes, int &r_received, bool
} else if (read == 0) {
_sock->close();
status = STATUS_NONE;
peer_port = 0;
peer_host = IP_Address();
disconnect_from_host();
r_received = total_read;
return ERR_FILE_EOF;