From d07864939a62b98fb1facd5db9b30d9a4d0bcad5 Mon Sep 17 00:00:00 2001
From: Tamaki Nishino <otamachan@gmail.com>
Date: Tue, 13 Nov 2018 22:09:18 +0900
Subject: [PATCH] Check SSL_DATA_PENDING in mosquitto_loop_read()

Signed-off-by: Tamaki Nishino <otamachan@gmail.com>
---
 lib/loop.c | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/lib/loop.c b/lib/loop.c
index 349ee5dc..b8fd2c57 100644
--- a/lib/loop.c
+++ b/lib/loop.c
@@ -147,12 +147,10 @@ int mosquitto_loop(struct mosquitto *mosq, int timeout, int max_packets)
 	}else{
 		if(mosq->sock != INVALID_SOCKET){
 			if(FD_ISSET(mosq->sock, &readfds)){
-				do{
-					rc = mosquitto_loop_read(mosq, max_packets);
-					if(rc || mosq->sock == INVALID_SOCKET){
-						return rc;
-					}
-				}while(SSL_DATA_PENDING(mosq));
+				rc = mosquitto_loop_read(mosq, max_packets);
+				if(rc || mosq->sock == INVALID_SOCKET){
+					return rc;
+				}
 			}
 			if(mosq->sockpairR != INVALID_SOCKET && FD_ISSET(mosq->sockpairR, &readfds)){
 #ifndef WIN32
@@ -368,7 +366,7 @@ int mosquitto_loop_read(struct mosquitto *mosq, int max_packets)
 	/* Queue len here tells us how many messages are awaiting processing and
 	 * have QoS > 0. We should try to deal with that many in this loop in order
 	 * to keep up. */
-	for(i=0; i<max_packets; i++){
+	for(i=0; i<max_packets || SSL_DATA_PENDING(mosq); i++){
 #ifdef WITH_SOCKS
 		if(mosq->socks5_host){
 			rc = socks5__read(mosq);