fix(ws): Fix handling of allocation failures

H/t @mathieucarbou

Author: willmmiles

src/AsyncWebSocket.cpp

@@ -448,6 +448,7 @@ void AsyncWebSocketClient::close(uint16_t code, const char *message) {
       if (c) {
         c->abort();
       }
+      return;
     }
   }
   _queueControl(WS_DISCONNECT);
@@ -593,7 +594,9 @@ void AsyncWebSocketClient::_onData(void *pbuf, size_t plen) {
           "[%s][%" PRIu32 "] DATA processing next fragment of %s frame %" PRIu32 ", index: %" PRIu64 ", len: %" PRIu32 "", _server->url(), _clientId,
           (_pinfo.message_opcode == WS_TEXT) ? "text" : "binary", _pinfo.num, _pinfo.index, (uint32_t)datalen
         );
-        _handleDataEvent(data, datalen, datalen == plen);  // datalen == plen means that we are processing the last part of the current TCP packet
+        if (!_handleDataEvent(data, datalen, datalen == plen)) {  // datalen == plen means that we are processing the last part of the current TCP packet
+          return;                                                 // stop processing on failure
+        }
       }
 
       // track index for next fragment
@@ -642,7 +645,9 @@ void AsyncWebSocketClient::_onData(void *pbuf, size_t plen) {
           (_pinfo.message_opcode == WS_TEXT) ? "text" : "binary", _pinfo.num, _pinfo.index, (uint32_t)datalen
         );
 
-        _handleDataEvent(data, datalen, datalen == plen);  // datalen == plen means that we are processing the last part of the current TCP packet
+        if (!_handleDataEvent(data, datalen, datalen == plen)) {  // datalen == plen means that we are processing the last part of the current TCP packet
+          return;                                                 // stop processing on failure
+        }
 
         if (_pinfo.final) {
           _pinfo.num = 0;
@@ -672,7 +677,7 @@ void AsyncWebSocketClient::_onData(void *pbuf, size_t plen) {
   }
 }
 
-void AsyncWebSocketClient::_handleDataEvent(uint8_t *data, size_t len, bool endOfPaquet) {
+bool AsyncWebSocketClient::_handleDataEvent(uint8_t *data, size_t len, bool endOfPaquet) {
   // ------------------------------------------------------------
   // Issue 384: https://github.com/ESP32Async/ESPAsyncWebServer/issues/384
   // Discussion: https://github.com/ESP32Async/ESPAsyncWebServer/pull/383#discussion_r2760425739
@@ -703,9 +708,11 @@ void AsyncWebSocketClient::_handleDataEvent(uint8_t *data, size_t len, bool endO
         _server->_handleEvent(this, WS_EVT_DATA, (void *)&_pinfo, copy.get(), len);
       } else {
         async_ws_log_e("Failed to allocate");
-        if (_client) {
-          _client->abort();
+        AsyncClient *c = _client;
+        if (c) {
+          c->abort();
         }
+        return false;  // failure!
       }
     } else {
       uint8_t backup = data[len];
@@ -716,6 +723,7 @@ void AsyncWebSocketClient::_handleDataEvent(uint8_t *data, size_t len, bool endO
   } else {
     _server->_handleEvent(this, WS_EVT_DATA, (void *)&_pinfo, data, len);
   }
+  return true;
 }
 
 size_t AsyncWebSocketClient::printf(const char *format, ...) {

src/AsyncWebSocket.h

@@ -190,7 +190,8 @@ class AsyncWebSocketClient {
   void _runQueue(asyncsrv::unique_lock_type &lock);
 
   // this function is called when a text message is received, in order to copy the buffer and place a null terminator at the end of the buffer for easier handling of text messages.
-  void _handleDataEvent(uint8_t *data, size_t len, bool endOfPaquet);
+  // Returns true on success, false on failure (e.g. memory allocation failure)
+  bool _handleDataEvent(uint8_t *data, size_t len, bool endOfPaquet);
 
 public:
   void *_tempObject;