mirror of
https://github.com/espressif/esp-idf.git
synced 2026-10-01 18:50:34 +03:00
fix(esp_http_server): prevent silent message drop in httpd_queue_work
Closes https://github.com/espressif/esp-idf/issues/18563
This commit is contained in:
@@ -17,6 +17,8 @@
|
||||
|
||||
#include <esp_http_server.h>
|
||||
#include "osal.h"
|
||||
#include "freertos/semphr.h"
|
||||
#include "sdkconfig.h"
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
@@ -129,9 +131,7 @@ struct httpd_data {
|
||||
httpd_config_t config; /*!< HTTPD server configuration */
|
||||
int listen_fd; /*!< Server listener FD */
|
||||
int ctrl_fd; /*!< Ctrl message receiver FD */
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
SemaphoreHandle_t ctrl_sock_semaphore; /*!< Ctrl socket semaphore */
|
||||
#endif
|
||||
SemaphoreHandle_t ctrl_sock_semaphore; /*!< Ctrl mbox slot reservation (sized to LWIP_UDP_RECVMBOX_SIZE) */
|
||||
int msg_fd; /*!< Ctrl message sender FD */
|
||||
struct thread_data hd_td; /*!< Information for the HTTPD thread */
|
||||
struct sock_db *hd_sd; /*!< The socket database */
|
||||
|
||||
@@ -150,24 +150,27 @@ esp_err_t httpd_queue_work(httpd_handle_t handle, httpd_work_fn_t work, void *ar
|
||||
.hc_work = work,
|
||||
.hc_work_arg = arg,
|
||||
};
|
||||
|
||||
/* Reserve a slot in the control mbox before sending. In blocking mode
|
||||
* the caller waits for a slot; in the default non-blocking mode we
|
||||
* fail fast so the caller knows the work was not queued. */
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
// Semaphore is acquired here and released after work function is executed.
|
||||
if (xSemaphoreTake(hd->ctrl_sock_semaphore, portMAX_DELAY) == pdTRUE) {
|
||||
const TickType_t wait = portMAX_DELAY;
|
||||
#else
|
||||
const TickType_t wait = 0;
|
||||
#endif
|
||||
int ret = cs_send_to_ctrl_sock(hd->msg_fd, hd->config.ctrl_port, &msg, sizeof(msg));
|
||||
if (ret < 0) {
|
||||
ESP_LOGW(TAG, LOG_FMT("failed to queue work"));
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
xSemaphoreGive(hd->ctrl_sock_semaphore);
|
||||
#endif
|
||||
return ESP_FAIL;
|
||||
}
|
||||
return ESP_OK;
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
if (xSemaphoreTake(hd->ctrl_sock_semaphore, wait) != pdTRUE) {
|
||||
ESP_LOGW(TAG, LOG_FMT("ctrl socket queue full, work not queued"));
|
||||
return ESP_FAIL;
|
||||
}
|
||||
ESP_LOGE(TAG, "Unable to acquire semaphore");
|
||||
return ESP_FAIL;
|
||||
#endif
|
||||
|
||||
int ret = cs_send_to_ctrl_sock(hd->msg_fd, hd->config.ctrl_port, &msg, sizeof(msg));
|
||||
if (ret < 0) {
|
||||
ESP_LOGW(TAG, LOG_FMT("failed to queue work"));
|
||||
xSemaphoreGive(hd->ctrl_sock_semaphore);
|
||||
return ESP_FAIL;
|
||||
}
|
||||
return ESP_OK;
|
||||
}
|
||||
|
||||
esp_err_t httpd_get_client_list(httpd_handle_t handle, size_t *fds, int *client_fds)
|
||||
@@ -207,16 +210,16 @@ static void httpd_process_ctrl_msg(struct httpd_data *hd)
|
||||
int ret = recv(hd->ctrl_fd, &msg, sizeof(msg), 0);
|
||||
if (ret <= 0) {
|
||||
ESP_LOGW(TAG, LOG_FMT("error in recv (%d)"), errno);
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
/* Returning the mbox slot to the producer side. xSemaphoreGive is a
|
||||
* no-op once the counting semaphore is at its max, which keeps the
|
||||
* accounting safe under producers that don't take (e.g. async
|
||||
* handler wakeups, shutdown). */
|
||||
xSemaphoreGive(hd->ctrl_sock_semaphore);
|
||||
#endif
|
||||
return;
|
||||
}
|
||||
if (ret != sizeof(msg)) {
|
||||
ESP_LOGW(TAG, LOG_FMT("incomplete msg"));
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
xSemaphoreGive(hd->ctrl_sock_semaphore);
|
||||
#endif
|
||||
return;
|
||||
}
|
||||
|
||||
@@ -234,9 +237,7 @@ static void httpd_process_ctrl_msg(struct httpd_data *hd)
|
||||
default:
|
||||
break;
|
||||
}
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
xSemaphoreGive(hd->ctrl_sock_semaphore);
|
||||
#endif
|
||||
}
|
||||
|
||||
// Called for each session from httpd_server
|
||||
@@ -469,6 +470,10 @@ static void httpd_delete(struct httpd_data *hd)
|
||||
free(hd->err_handler_fns);
|
||||
free(ra->resp_hdrs);
|
||||
free(hd->hd_sd);
|
||||
if (hd->ctrl_sock_semaphore) {
|
||||
vSemaphoreDelete(hd->ctrl_sock_semaphore);
|
||||
hd->ctrl_sock_semaphore = NULL;
|
||||
}
|
||||
|
||||
/* Free registered URI handlers */
|
||||
httpd_unregister_all_uri_handlers(hd);
|
||||
@@ -504,18 +509,18 @@ esp_err_t httpd_start(httpd_handle_t *handle, const httpd_config_t *config)
|
||||
/* Failed to allocate memory */
|
||||
return ESP_ERR_HTTPD_ALLOC_MEM;
|
||||
}
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
/* Using a Counting Semaphore with count equals CONFIG_LWIP_UDP_RECVMBOX_SIZE
|
||||
* as the number of UDP messages which can be stored is equal to UDP mailbox size.
|
||||
* Using this, we can make sure that the work function is always received by the ctrl socket.
|
||||
*/
|
||||
/* Counting semaphore sized to the UDP control-socket recv mbox. Each
|
||||
* httpd_queue_work() take reserves one mbox slot; httpd_process_ctrl_msg()
|
||||
* gives one back per drain. This bounds the producer to the mbox capacity
|
||||
* and prevents silent lwIP-mbox overflow drops that would otherwise leak
|
||||
* the caller's async-send context. Always created so the default
|
||||
* (non-blocking) httpd_queue_work() path can also rely on it. */
|
||||
hd->ctrl_sock_semaphore = xSemaphoreCreateCounting(CONFIG_LWIP_UDP_RECVMBOX_SIZE, CONFIG_LWIP_UDP_RECVMBOX_SIZE);
|
||||
if (hd->ctrl_sock_semaphore == NULL) {
|
||||
ESP_LOGE(TAG, "Failed to create Semaphore");
|
||||
httpd_delete(hd);
|
||||
return ESP_ERR_HTTPD_ALLOC_MEM;
|
||||
}
|
||||
#endif
|
||||
|
||||
if (httpd_server_init(hd) != ESP_OK) {
|
||||
httpd_delete(hd);
|
||||
@@ -529,6 +534,12 @@ esp_err_t httpd_start(httpd_handle_t *handle, const httpd_config_t *config)
|
||||
httpd_thread, hd,
|
||||
hd->config.core_id,
|
||||
hd->config.task_caps) != ESP_OK) {
|
||||
/* Close the open socket */
|
||||
close(hd->listen_fd);
|
||||
/* Close the control socket */
|
||||
cs_free_ctrl_sock(hd->ctrl_fd);
|
||||
/* Close the message socket */
|
||||
close(hd->msg_fd);
|
||||
/* Failed to launch task */
|
||||
httpd_delete(hd);
|
||||
return ESP_ERR_HTTPD_TASK;
|
||||
@@ -582,9 +593,6 @@ esp_err_t httpd_stop(httpd_handle_t handle)
|
||||
}
|
||||
|
||||
ESP_LOGD(TAG, LOG_FMT("server stopped"));
|
||||
#if CONFIG_HTTPD_QUEUE_WORK_BLOCKING
|
||||
vSemaphoreDelete(hd->ctrl_sock_semaphore);
|
||||
#endif
|
||||
httpd_delete(hd);
|
||||
esp_http_server_dispatch_event(HTTP_SERVER_EVENT_STOP, NULL, 0);
|
||||
return ESP_OK;
|
||||
|
||||
Reference in New Issue
Block a user