feat(websocket): Create specific Event for PONG message

- Separate the PONG message from generic payload, this can make it easier to identify when a timeout occur when receiving a payload

Author: glmfe

components/esp_websocket_client/esp_websocket_client.c

@@ -984,7 +984,11 @@ static esp_err_t esp_websocket_client_recv(esp_websocket_client_handle_t client)
             return ESP_OK;
         }
 
-        esp_websocket_client_dispatch_event(client, WEBSOCKET_EVENT_DATA, client->rx_buffer, rlen);
+        if (client->last_opcode == WS_TRANSPORT_OPCODES_PONG) {
+            esp_websocket_client_dispatch_event(client, WEBSOCKET_EVENT_PONG, client->rx_buffer, rlen);
+        } else {
+            esp_websocket_client_dispatch_event(client, WEBSOCKET_EVENT_DATA, client->rx_buffer, rlen);
+        }
 
         client->payload_offset += rlen;
     } while (client->payload_offset < client->payload_len);

components/esp_websocket_client/examples/linux/main/websocket_linux.c

@@ -30,6 +30,7 @@ static void websocket_event_handler(void *handler_args, esp_event_base_t base, i
         break;
     case WEBSOCKET_EVENT_CONNECTED:
         ESP_LOGI(TAG, "WEBSOCKET_EVENT_CONNECTED");
+        esp_websocket_client_send_with_opcode((esp_websocket_client_handle_t)handler_args, WS_TRANSPORT_OPCODES_PING, NULL, 0, portMAX_DELAY);
         break;
     case WEBSOCKET_EVENT_DISCONNECTED:
         ESP_LOGI(TAG, "WEBSOCKET_EVENT_DISCONNECTED");
@@ -52,6 +53,9 @@ static void websocket_event_handler(void *handler_args, esp_event_base_t base, i
         // If received data contains json structure it succeed to parse
         ESP_LOGW(TAG, "Total payload length=%d, data_len=%d, current payload offset=%d\r\n", data->payload_len, data->data_len, data->payload_offset);
 
+        break;
+    case WEBSOCKET_EVENT_PONG:
+        ESP_LOGI(TAG, "WEBSOCKET_EVENT_PONG (op=%d, len=%d)", data->op_code, data->data_len);
         break;
     case WEBSOCKET_EVENT_ERROR:
         ESP_LOGI(TAG, "WEBSOCKET_EVENT_ERROR");

components/esp_websocket_client/examples/target/main/websocket_example.c

@@ -79,6 +79,7 @@ static void websocket_event_handler(void *handler_args, esp_event_base_t base, i
         break;
     case WEBSOCKET_EVENT_CONNECTED:
         ESP_LOGI(TAG, "WEBSOCKET_EVENT_CONNECTED");
+        esp_websocket_client_send_with_opcode((esp_websocket_client_handle_t)handler_args, WS_TRANSPORT_OPCODES_PING, NULL, 0, portMAX_DELAY);
         break;
     case WEBSOCKET_EVENT_DISCONNECTED:
         ESP_LOGI(TAG, "WEBSOCKET_EVENT_DISCONNECTED");
@@ -114,6 +115,12 @@ static void websocket_event_handler(void *handler_args, esp_event_base_t base, i
 
         ESP_LOGW(TAG, "Total payload length=%d, data_len=%d, current payload offset=%d\r\n", data->payload_len, data->data_len, data->payload_offset);
 
+        xTimerReset(shutdown_signal_timer, portMAX_DELAY);
+        break;
+    case WEBSOCKET_EVENT_PONG:
+        ESP_LOGI(TAG, "WEBSOCKET_EVENT_PONG");
+        ESP_LOGI(TAG, "Received opcode=%d", data->op_code);
+        // Reset shutdown timer on PONG as well
         xTimerReset(shutdown_signal_timer, portMAX_DELAY);
         break;
     case WEBSOCKET_EVENT_ERROR:

components/esp_websocket_client/examples/target/pytest_websocket.py

@@ -172,6 +172,14 @@ def test_recv_fragmented_msg1(dut):
     def test_recv_fragmented_msg2(dut):
         dut.expect('websocket: Total payload length=2000, data_len=976, current payload offset=1024')
 
+    # Test for PONG message reception:
+    # Verifies that the WebSocket client correctly receives a PONG response after sending a PING.
+    # The PING is automatically sent when the connection is established.
+    def test_pong_received(dut):
+        dut.expect('WEBSOCKET_EVENT_PONG')
+        print('\nPONG message received successfully\n')
+        sys.stdout.flush()
+
     # Test for receiving fragmented text messages:
     # Checks if the WebSocket can accurately reconstruct a message sent in several smaller parts.
     def test_fragmented_txt_msg(dut):
@@ -237,6 +245,7 @@ def test_fragmented_binary_msg(dut):
             dut.expect('Please enter uri of websocket endpoint', timeout=30)
             dut.write(uri)
             test_echo(dut)
+            test_pong_received(dut)
             test_recv_long_msg(dut, ws, 2000, 3)
             test_json(dut, ws)
             test_fragmented_txt_msg(dut)
@@ -247,3 +256,4 @@ def test_fragmented_binary_msg(dut):
     else:
         print('DUT connecting to {}'.format(uri))
         test_echo(dut)
+        test_pong_received(dut)

components/esp_websocket_client/include/esp_websocket_client.h

@@ -34,6 +34,7 @@ typedef enum {
     WEBSOCKET_EVENT_CONNECTED,      /*!< Once the Websocket has been connected to the server, no data exchange has been performed */
     WEBSOCKET_EVENT_DISCONNECTED,   /*!< The connection has been disconnected */
     WEBSOCKET_EVENT_DATA,           /*!< When receiving data from the server, possibly multiple portions of the packet */
+    WEBSOCKET_EVENT_PONG,           /*!< When receiving a PONG control frame from the server */
     WEBSOCKET_EVENT_CLOSED,         /*!< The connection has been closed cleanly */
     WEBSOCKET_EVENT_BEFORE_CONNECT, /*!< The event occurs before connecting */
     WEBSOCKET_EVENT_BEGIN,          /*!< The event occurs once after thread creation, before event loop */