MbedClient: improving logic of readSocket

arduino · Apr 23, 2024 · b172e12 · b172e12
1 parent 2d27acf
commit b172e12
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/libraries/SocketWrapper/src/MbedClient.cpp b/libraries/SocketWrapper/src/MbedClient.cpp
@@ -22,15 +22,13 @@ void arduino::MbedClient::readSocket() {
     int ret = NSAPI_ERROR_WOULD_BLOCK;
     do {
       mutex->lock();
-      if (sock == nullptr) {
-        goto cleanup;
-      }
-      mutex->unlock();
-      if (rxBuffer.availableForStore() == 0) {
+      if (sock != nullptr && rxBuffer.availableForStore() == 0) {
+        mutex->unlock();
         yield();
         continue;
+      } else if (sock == nullptr) {
+        goto cleanup;
       }
-      mutex->lock();
       ret = sock->recv(data, rxBuffer.availableForStore());
       if (ret < 0 && ret != NSAPI_ERROR_WOULD_BLOCK) {
         goto cleanup;