espressif/esp-protocols
1
0
Fork 1
mirror of https://github.com/espressif/esp-protocols.git synced 2025-06-26 09:51:34 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(mdns): Fix API races adding txt item for services

Browse Source 
Fixes **API race issue** (described in 8a690503) for API
mdns_service_txt_item_set_for_host_with_explicit_value_len()
This commit is contained in:
David Cermak
2024-07-23 09:36:50 +02:00
parent a927bf3a8d
commit c62b920bb9
2 changed files with 44 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

124
components/mdns/mdns.c
View File

@ -5111,10 +5111,6 @@ static void _mdns_free_action(mdns_action_t *action)
case ACTION_SERVICE_INSTANCE_SET: case ACTION_SERVICE_INSTANCE_SET:
free(action->data.srv_instance.instance); free(action->data.srv_instance.instance);
break; break;
case ACTION_SERVICE_TXT_SET:
free(action->data.srv_txt_set.key);
free(action->data.srv_txt_set.value);
break;
case ACTION_SERVICE_TXT_DEL: case ACTION_SERVICE_TXT_DEL:
free(action->data.srv_txt_del.key); free(action->data.srv_txt_del.key);
break; break;
@ -5164,7 +5160,6 @@ static void _mdns_execute_action(mdns_action_t *action)
mdns_srv_item_t *a = NULL; mdns_srv_item_t *a = NULL;
mdns_service_t *service; mdns_service_t *service;
char *key; char *key;
char *value;
char *subtype; char *subtype;
mdns_subtype_t *subtype_item; mdns_subtype_t *subtype_item;
mdns_txt_linked_item_t *txt, * t; mdns_txt_linked_item_t *txt, * t;
@ -5197,38 +5192,6 @@ static void _mdns_execute_action(mdns_action_t *action)
action->data.srv_instance.service->service->instance = action->data.srv_instance.instance; action->data.srv_instance.service->service->instance = action->data.srv_instance.instance;
_mdns_probe_all_pcbs(&action->data.srv_instance.service, 1, false, false); _mdns_probe_all_pcbs(&action->data.srv_instance.service, 1, false, false);
break;
case ACTION_SERVICE_TXT_SET:
service = action->data.srv_txt_set.service->service;
key = action->data.srv_txt_set.key;
value = action->data.srv_txt_set.value;
txt = service->txt;
while (txt) {
if (strcmp(txt->key, key) == 0) {
free((char *)txt->value);
free(key);
txt->value = value;
txt->value_len = action->data.srv_txt_set.value_len;
break;
}
txt = txt->next;
}
if (!txt) {
txt = (mdns_txt_linked_item_t *)malloc(sizeof(mdns_txt_linked_item_t));
if (!txt) {
HOOK_MALLOC_FAILED;
_mdns_free_action(action);
return;
}
txt->key = key;
txt->value = value;
txt->value_len = action->data.srv_txt_set.value_len;
txt->next = service->txt;
service->txt = txt;
}
_mdns_announce_all_pcbs(&action->data.srv_txt_set.service, 1, false);
break; break;
case ACTION_SERVICE_TXT_DEL: case ACTION_SERVICE_TXT_DEL:
service = action->data.srv_txt_del.service->service; service = action->data.srv_txt_del.service->service;
@ -5258,7 +5221,7 @@ static void _mdns_execute_action(mdns_action_t *action)
} }
free(key); free(key);
_mdns_announce_all_pcbs(&action->data.srv_txt_set.service, 1, false); _mdns_announce_all_pcbs(&action->data.srv_txt_del.service, 1, false);
break; break;
case ACTION_SERVICE_SUBTYPE_ADD: case ACTION_SERVICE_SUBTYPE_ADD:
@ -6281,51 +6244,53 @@ esp_err_t mdns_service_txt_set(const char *service, const char *proto, mdns_txt_
} }
esp_err_t mdns_service_txt_item_set_for_host_with_explicit_value_len(const char *instance, const char *service, const char *proto, esp_err_t mdns_service_txt_item_set_for_host_with_explicit_value_len(const char *instance, const char *service, const char *proto,
const char *hostname, const char *key, const char *host, const char *key, const char *value_arg, uint8_t value_len)
const char *value, uint8_t value_len)
{ {
MDNS_SERVICE_LOCK(); MDNS_SERVICE_LOCK();
if (!_mdns_server || !_mdns_server->services || _str_null_or_empty(service) || _str_null_or_empty(proto) || esp_err_t ret = ESP_OK;
_str_null_or_empty(key) || (!value && value_len)) { const char *hostname = host ? host : _mdns_server->hostname;
MDNS_SERVICE_UNLOCK(); ESP_GOTO_ON_FALSE(_mdns_server && _mdns_server->services && !_str_null_or_empty(service) && !_str_null_or_empty(proto) && !_str_null_or_empty(key) &&
return ESP_ERR_INVALID_ARG; !((!value_arg && value_len)), ESP_ERR_INVALID_ARG, err, TAG, "Invalid state or arguments");
}
mdns_srv_item_t *s = _mdns_get_service_item_instance(instance, service, proto, hostname); mdns_srv_item_t *s = _mdns_get_service_item_instance(instance, service, proto, hostname);
MDNS_SERVICE_UNLOCK(); ESP_GOTO_ON_FALSE(s, ESP_ERR_NOT_FOUND, err, TAG, "Service doesn't exist");
if (!s) {
return ESP_ERR_NOT_FOUND; mdns_service_t *srv = s->service;
char *value = NULL;
if (value_len > 0) {
value = (char *) malloc(value_len);
ESP_GOTO_ON_FALSE(s, ESP_ERR_NO_MEM, err, TAG, "Out of memory");
memcpy(value, value_arg, value_len);
} else {
value_len = 0;
} }
mdns_action_t *action = (mdns_action_t *)malloc(sizeof(mdns_action_t)); mdns_txt_linked_item_t *txt = srv->txt;
if (!action) { while (txt) {
HOOK_MALLOC_FAILED; if (strcmp(txt->key, key) == 0) {
return ESP_ERR_NO_MEM; free((char *)txt->value);
txt->value = value;
txt->value_len = value_len;
break;
}
txt = txt->next;
}
if (!txt) {
txt = (mdns_txt_linked_item_t *)malloc(sizeof(mdns_txt_linked_item_t));
ESP_GOTO_ON_FALSE(s, ESP_ERR_NO_MEM, err, TAG, "Out of memory");
txt->key = strdup(key);
ESP_GOTO_ON_FALSE(txt->key, ESP_ERR_NO_MEM, err, TAG, "Out of memory");
txt->value = value;
txt->value_len = value_len;
txt->next = srv->txt;
srv->txt = txt;
} }
action->type = ACTION_SERVICE_TXT_SET; _mdns_announce_all_pcbs(&s, 1, false);
action->data.srv_txt_set.service = s;
action->data.srv_txt_set.key = strdup(key); err:
if (!action->data.srv_txt_set.key) { MDNS_SERVICE_UNLOCK();
free(action); if (ret == ESP_ERR_NO_MEM) {
return ESP_ERR_NO_MEM; HOOK_MALLOC_FAILED;
}
if (value_len > 0) {
action->data.srv_txt_set.value = (char *)malloc(value_len);
if (!action->data.srv_txt_set.value) {
free(action->data.srv_txt_set.key);
free(action);
return ESP_ERR_NO_MEM;
}
memcpy(action->data.srv_txt_set.value, value, value_len);
action->data.srv_txt_set.value_len = value_len;
} else {
action->data.srv_txt_set.value = NULL;
action->data.srv_txt_set.value_len = 0;
}
if (xQueueSend(_mdns_server->action_queue, &action, (TickType_t)0) != pdPASS) {
free(action->data.srv_txt_set.key);
free(action->data.srv_txt_set.value);
free(action);
return ESP_ERR_NO_MEM;
} }
return ESP_OK; return ESP_OK;
} }
@ -6343,7 +6308,7 @@ esp_err_t mdns_service_txt_item_set(const char *service, const char *proto, cons
if (!_mdns_server) { if (!_mdns_server) {
return ESP_ERR_INVALID_STATE; return ESP_ERR_INVALID_STATE;
} }
return mdns_service_txt_item_set_for_host_with_explicit_value_len(NULL, service, proto, _mdns_server->hostname, key, return mdns_service_txt_item_set_for_host_with_explicit_value_len(NULL, service, proto, NULL, key,
value, strlen(value)); value, strlen(value));
} }
@ -6353,8 +6318,7 @@ esp_err_t mdns_service_txt_item_set_with_explicit_value_len(const char *service,
if (!_mdns_server) { if (!_mdns_server) {
return ESP_ERR_INVALID_STATE; return ESP_ERR_INVALID_STATE;
} }
return mdns_service_txt_item_set_for_host_with_explicit_value_len(NULL, service, proto, _mdns_server->hostname, key, return mdns_service_txt_item_set_for_host_with_explicit_value_len(NULL, service, proto, NULL, key, value, value_len);
value, value_len);
} }
esp_err_t mdns_service_txt_item_remove_for_host(const char *instance, const char *service, const char *proto, const char *hostname, esp_err_t mdns_service_txt_item_remove_for_host(const char *instance, const char *service, const char *proto, const char *hostname,

7
components/mdns/private_include/mdns_private.h
View File

@ -188,7 +188,6 @@ typedef enum {
ACTION_HOSTNAME_SET, ACTION_HOSTNAME_SET,
ACTION_INSTANCE_SET, ACTION_INSTANCE_SET,
ACTION_SERVICE_INSTANCE_SET, ACTION_SERVICE_INSTANCE_SET,
ACTION_SERVICE_TXT_SET,
ACTION_SERVICE_TXT_DEL, ACTION_SERVICE_TXT_DEL,
ACTION_SERVICE_SUBTYPE_ADD, ACTION_SERVICE_SUBTYPE_ADD,
ACTION_SERVICES_CLEAR, ACTION_SERVICES_CLEAR,
@ -446,12 +445,6 @@ typedef struct {
mdns_srv_item_t *service; mdns_srv_item_t *service;
char *instance; char *instance;
} srv_instance; } srv_instance;
struct {
mdns_srv_item_t *service;
char *key;
char *value;
uint8_t value_len;
} srv_txt_set;
struct { struct {
mdns_srv_item_t *service; mdns_srv_item_t *service;
char *key; char *key;