From 57805f8d514ce8666eec17131a679a5d03dd5866 Mon Sep 17 00:00:00 2001
From: jordan <jordan@wolfssl.com>
Date: Wed, 20 Mar 2024 21:20:13 -0500
Subject: [PATCH] Fix double lock of lockRecv.

---
 src/mqtt_client.c | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/mqtt_client.c b/src/mqtt_client.c
index 5aff6e80..0d464d62 100644
--- a/src/mqtt_client.c
+++ b/src/mqtt_client.c
@@ -100,7 +100,6 @@ static int MqttClient_CancelMessage(MqttClient *client, MqttObject* msg);
         s->sem = NULL;
         return 0;
     }
-
     int wm_SemLock(wm_Sem *s) {
         dispatch_semaphore_wait(s->sem, DISPATCH_TIME_FOREVER);
         return 0;
@@ -275,11 +274,14 @@ static int MqttReadStart(MqttClient* client, MqttMsgStat* stat)
 {
     int rc = MQTT_CODE_SUCCESS;
 
-#ifdef WOLFMQTT_DEBUG_CLIENT
+#if defined(WOLFMQTT_DEBUG_CLIENT) || !defined(WOLFMQTT_ALLOW_NODATA_UNLOCK)
+  #ifdef WOLFMQTT_DEBUG_CLIENT
     if (stat->isReadActive) {
         MQTT_TRACE_MSG("Warning, recv already locked!");
         rc = MQTT_CODE_ERROR_SYSTEM;
     }
+  #endif /* WOLFMQTT_DEBUG_CLIENT */
+  #ifndef WOLFMQTT_ALLOW_NODATA_UNLOCK
     /* detect if a read is already in progress */
     #ifdef WOLFMQTT_MULTITHREAD
     if (wm_SemLock(&client->lockClient) == 0)
@@ -293,9 +295,11 @@ static int MqttReadStart(MqttClient* client, MqttMsgStat* stat)
         wm_SemUnlock(&client->lockClient);
     #endif
     }
-    if (rc != 0)
+  #endif /* WOLFMQTT_ALLOW_NODATA_UNLOCK */
+    if (rc != MQTT_CODE_SUCCESS) {
         return rc;
-#endif /* WOLFMQTT_DEBUG_CLIENT */
+    }
+#endif /* WOLFMQTT_DEBUG_CLIENT || !WOLFMQTT_ALLOW_NODATA_UNLOCK */
 
 #ifdef WOLFMQTT_MULTITHREAD
     rc = wm_SemLock(&client->lockRecv);