Merge pull request #868 from andreagilardoni/mbedclient-stop-issue

Mbedclient stop issue
arduino · Apr 19, 2024 · 8186f2a · 8186f2a
2 parents 79d9a8d + abaef3f
commit 8186f2a
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/libraries/SocketWrapper/src/MbedClient.cpp b/libraries/SocketWrapper/src/MbedClient.cpp
@@ -13,20 +13,24 @@ uint8_t arduino::MbedClient::status() {
   return _status;
 }
 
+
 void arduino::MbedClient::readSocket() {
-  while (1) {
+  uint8_t data[SOCKET_BUFFER_SIZE];
+
+  while (sock != nullptr) {
     event->wait_any(0xFF, 100);
-    uint8_t data[SOCKET_BUFFER_SIZE];
     int ret = NSAPI_ERROR_WOULD_BLOCK;
     do {
+      mutex->lock();
+      if (sock == nullptr) {
+        goto cleanup;
+      }
+      mutex->unlock();
       if (rxBuffer.availableForStore() == 0) {
         yield();
         continue;
       }
       mutex->lock();
-      if (sock == nullptr) {
-        goto cleanup;
-      }
       ret = sock->recv(data, rxBuffer.availableForStore());
       if (ret < 0 && ret != NSAPI_ERROR_WOULD_BLOCK) {
         goto cleanup;
@@ -61,7 +65,7 @@ void arduino::MbedClient::configureSocket(Socket *_s) {
   _s->set_timeout(_timeout);
   _s->set_blocking(false);
   _s->getpeername(&address);
-  
+
   if (event == nullptr) {
     event = new rtos::EventFlags;
   }
@@ -295,6 +299,7 @@ void arduino::MbedClient::stop() {
     mutex = nullptr;
   }
   _status = false;
+  rxBuffer.clear();
 }
 
 uint8_t arduino::MbedClient::connected() {