Fix UDP wait() not returning after first received packet

(cherry picked from commit 5c6715a291)
2017-05-05 17:41:11 +02:00 · 2017-05-05 17:41:11 +02:00 · e9a7eeaf9c
parent bd26eacc59
commit e9a7eeaf9c
2 changed files with 4 additions and 0 deletions
--- a/drivers/unix/packet_peer_udp_posix.cpp
+++ b/drivers/unix/packet_peer_udp_posix.cpp
@ -216,6 +216,8 @@ Error PacketPeerUDPPosix::_poll(bool p_wait) {

 		len = sizeof(struct sockaddr_storage);
 		++queue_count;
+		if (p_wait)
+			break;
 	};

 	// TODO: Should ECONNRESET be handled here?
--- a/platform/windows/packet_peer_udp_winsock.cpp
+++ b/platform/windows/packet_peer_udp_winsock.cpp
@ -203,6 +203,8 @@ Error PacketPeerUDPWinsock::_poll(bool p_wait) {

 		len = sizeof(struct sockaddr_storage);
 		++queue_count;
+		if (p_wait)
+			break;
 	};

 	if (ret == SOCKET_ERROR) {