espressif/esp-protocols
1
0
Fork 1
mirror of https://github.com/espressif/esp-protocols.git synced 2025-07-21 22:42:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

mdns: fix memory free issue when repeating the query in reply

Browse Source 
The repeated query will be copied in the next event loop while the
memory is freed instantly. Delay the free to fix this issue.


* Original commit: espressif/esp-idf@5f244c86f2
This commit is contained in:
Jiacheng Guo
2021-04-16 15:19:34 +08:00
committed by suren-gabrielyan-espressif
parent 4d8aec1ad3
commit b62b4b3e25
2 changed files with 47 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
components/mdns/mdns.c
View File

@ -1057,7 +1057,26 @@ static void _mdns_free_tx_packet(mdns_tx_packet_t * packet)
if (!packet) { if (!packet) {
return; return;
} }
queueFree(mdns_out_question_t, packet->questions); mdns_out_question_t *q = packet->questions;
while (q) {
mdns_out_question_t *next = q->next;
if (q->own_dynamic_memory) {
if (q->host) {
free((char *)q->host);
}
if (q->service) {
free((char *)q->service);
}
if (q->proto) {
free((char *)q->proto);
}
if (q->domain) {
free((char *)q->domain);
}
}
free(q);
q = next;
}
queueFree(mdns_out_answer_t, packet->answers); queueFree(mdns_out_answer_t, packet->answers);
queueFree(mdns_out_answer_t, packet->servers); queueFree(mdns_out_answer_t, packet->servers);
queueFree(mdns_out_answer_t, packet->additional); queueFree(mdns_out_answer_t, packet->additional);
@ -1341,8 +1360,18 @@ static void _mdns_create_answer_from_parsed_packet(mdns_parsed_packet_t * parsed
_mdns_free_tx_packet(packet); _mdns_free_tx_packet(packet);
return; return;
} }
memcpy(out_question, q, sizeof(mdns_out_question_t)); out_question->type = q->type;
out_question->unicast = q->unicast;
out_question->host = q->host;
q->host = NULL;
out_question->service = q->service;
q->service = NULL;
out_question->proto = q->proto;
q->proto = NULL;
out_question->domain = q->domain;
q->domain = NULL;
out_question->next = NULL; out_question->next = NULL;
out_question->own_dynamic_memory = true;
queueToEnd(mdns_out_question_t, packet->questions, out_question); queueToEnd(mdns_out_question_t, packet->questions, out_question);
#endif // MDNS_REPEAT_QUERY_IN_RESPONSE #endif // MDNS_REPEAT_QUERY_IN_RESPONSE
} else if (!_mdns_alloc_answer(&packet->answers, q->type, NULL, send_flush, false)) { } else if (!_mdns_alloc_answer(&packet->answers, q->type, NULL, send_flush, false)) {
@ -1409,6 +1438,7 @@ static mdns_tx_packet_t * _mdns_create_probe_packet(mdns_if_t tcpip_if, mdns_ip_
q->service = services[i]->service->service; q->service = services[i]->service->service;
q->proto = services[i]->service->proto; q->proto = services[i]->service->proto;
q->domain = MDNS_DEFAULT_DOMAIN; q->domain = MDNS_DEFAULT_DOMAIN;
q->own_dynamic_memory = false;
if (!q->host || _mdns_question_exists(q, packet->questions)) { if (!q->host || _mdns_question_exists(q, packet->questions)) {
free(q); free(q);
continue; continue;
@ -1436,6 +1466,7 @@ static mdns_tx_packet_t * _mdns_create_probe_packet(mdns_if_t tcpip_if, mdns_ip_
q->service = NULL; q->service = NULL;
q->proto = NULL; q->proto = NULL;
q->domain = MDNS_DEFAULT_DOMAIN; q->domain = MDNS_DEFAULT_DOMAIN;
q->own_dynamic_memory = false;
if (_mdns_question_exists(q, packet->questions)) { if (_mdns_question_exists(q, packet->questions)) {
free(q); free(q);
} else { } else {
@ -3117,10 +3148,18 @@ clear_rx_packet:
while (parsed_packet->questions) { while (parsed_packet->questions) {
mdns_parsed_question_t * question = parsed_packet->questions; mdns_parsed_question_t * question = parsed_packet->questions;
parsed_packet->questions = parsed_packet->questions->next; parsed_packet->questions = parsed_packet->questions->next;
free(question->host); if (question->host) {
free(question->service); free(question->host);
free(question->proto); }
free(question->domain); if (question->service) {
free(question->service);
}
if (question->proto) {
free(question->proto);
}
if (question->domain) {
free(question->domain);
}
free(question); free(question);
} }
free(parsed_packet); free(parsed_packet);
@ -3650,6 +3689,7 @@ static mdns_tx_packet_t * _mdns_create_search_packet(mdns_search_once_t * search
q->service = search->service; q->service = search->service;
q->proto = search->proto; q->proto = search->proto;
q->domain = MDNS_DEFAULT_DOMAIN; q->domain = MDNS_DEFAULT_DOMAIN;
q->own_dynamic_memory = false;
queueToEnd(mdns_out_question_t, packet->questions, q); queueToEnd(mdns_out_question_t, packet->questions, q);
if (search->type == MDNS_TYPE_PTR) { if (search->type == MDNS_TYPE_PTR) {

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

@ -299,6 +299,7 @@ typedef struct mdns_out_question_s {
const char * service; const char * service;
const char * proto; const char * proto;
const char * domain; const char * domain;
bool own_dynamic_memory;
} mdns_out_question_t; } mdns_out_question_t;
typedef struct mdns_out_answer_s { typedef struct mdns_out_answer_s {