ESP32 Forum

Code: Select all

#include "my_keep_alive.h"  // my include


esp_err_t httpd_ws_get_frame_type(httpd_req_t *req)
{
    esp_err_t ret = httpd_ws_check_req(req);
    if (ret != ESP_OK) {
        return ret;
    }

    struct httpd_req_aux *aux = req->aux;
    if (aux == NULL) {
        ESP_LOGW(TAG, LOG_FMT("Invalid Aux pointer"));
        return ESP_ERR_INVALID_ARG;
    }

    /* Read the first byte from the frame to get the FIN flag and Opcode */
    /* Please refer to RFC6455 Section 5.2 for more details */
    uint8_t first_byte = 0;
    if (httpd_recv_with_opt(req, (char *)&first_byte, sizeof(first_byte), false) <= 0) {
        /* If the recv() return code is <= 0, then this socket FD is invalid (i.e. a broken connection) */
        /* Here we mark it as a Close message and close it later. */
        ESP_LOGW(TAG, LOG_FMT("Failed to read header byte (socket FD invalid), closing socket now"));
        aux->ws_final = true;
        aux->ws_type = HTTPD_WS_TYPE_CLOSE;
        return ESP_OK;
    }

    ESP_LOGD(TAG, LOG_FMT("First byte received: 0x%02X"), first_byte);

    /* Decode the FIN flag and Opcode from the byte */
    aux->ws_final = (first_byte & HTTPD_WS_FIN_BIT) != 0;
    aux->ws_type = (first_byte & HTTPD_WS_OPCODE_BITS);

    /* Reply to PING. For PONG and CLOSE, it will be handled elsewhere. */
    if(aux->ws_type == HTTPD_WS_TYPE_PING) {
        ESP_LOGD(TAG, LOG_FMT("Got a WS PING frame, Replying PONG..."));

        /* Read the rest of the PING frame, for PONG to reply back. */
        /* Please refer to RFC6455 Section 5.5.2 for more details */
        httpd_ws_frame_t frame;
        uint8_t frame_buf[128] = { 0 };
        memset(&frame, 0, sizeof(httpd_ws_frame_t));
        frame.payload = frame_buf;

        if(httpd_ws_recv_frame(req, &frame, 126) != ESP_OK) {
            ESP_LOGD(TAG, LOG_FMT("Cannot receive the full PING frame"));
            return ESP_ERR_INVALID_STATE;
        }




// my code
        int sockfd = httpd_req_to_sockfd(req);         
        clear_Keep_Alive_timeout_counter( sockfd );
//




        /* Now turn the frame to PONG */
        frame.type = HTTPD_WS_TYPE_PONG;
        return httpd_ws_send_frame(req, &frame);
    }

    return ESP_OK;
}

Code: Select all

idf_component_register( SRCS "src/httpd_main.c"
                            	     "src/httpd_parse.c"
                            	     "src/httpd_sess.c"
                                     "src/httpd_txrx.c"
                                     "src/httpd_uri.c"
                                     "src/httpd_ws.c"
                                     "src/util/ctrl_sock.c"					
                                     INCLUDE_DIRS "include"
                                     PRIV_INCLUDE_DIRS "src/port/esp32" "src/util"
                                     REQUIRES nghttp # for http_parser.h
                                     PRIV_REQUIRES lwip mbedtls esp_timer

SRCS "C:/esp32Projects/WSS_SERVER_AND_UI_INTEGRATION/components/communication/my_wss_server/src/my_keep_alive.c"
INCLUDE_DIRS "C:/esp32Projects/WSS_SERVER_AND_UI_INTEGRATION/components/communication/my_wss_server/src/include"					 
)