blackbeard420/esp-idf
1
0
Fork 0
mirror of https://github.com/espressif/esp-idf synced 2025-03-12 10:39:11 -04:00
Code Issues Projects Releases Wiki Activity

Merge branch 'bugfix/mdns_tcpip_threadsafe' into 'master'

Browse Source 
mdns: Use LwIP IPC for low-level API calls

See merge request !925
This commit is contained in:
Angus Gratton 2017-08-30 13:28:51 +08:00
commit ef60d73fe4
1 changed files with 182 additions and 116 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

298
components/mdns/mdns.c
View File

@ -23,6 +23,8 @@
#include "lwip/pbuf.h" #include "lwip/pbuf.h"
#include "lwip/igmp.h" #include "lwip/igmp.h"
#include "lwip/udp.h" #include "lwip/udp.h"
#include "lwip/tcpip.h"
#include "lwip/priv/tcpip_priv.h"
#include "esp_wifi.h" #include "esp_wifi.h"
#endif #endif
@ -154,6 +156,14 @@ struct mdns_server_s {
} search; } search;
}; };
typedef struct {
struct tcpip_api_call call;
mdns_server_t *server;
uint8_t *data;
size_t len;
esp_err_t err;
} mdns_api_call_t;
#define MDNS_MUTEX_LOCK() xSemaphoreTake(server->lock, portMAX_DELAY) #define MDNS_MUTEX_LOCK() xSemaphoreTake(server->lock, portMAX_DELAY)
#define MDNS_MUTEX_UNLOCK() xSemaphoreGive(server->lock) #define MDNS_MUTEX_UNLOCK() xSemaphoreGive(server->lock)
@ -163,7 +173,7 @@ struct mdns_server_s {
static const char * MDNS_DEFAULT_DOMAIN = "local"; static const char * MDNS_DEFAULT_DOMAIN = "local";
static const char * MDNS_SUB_STR = "_sub"; static const char * MDNS_SUB_STR = "_sub";
static mdns_server_t * _mdns_servers[TCPIP_ADAPTER_IF_MAX] = {0,0,0}; static mdns_server_t * _mdns_servers[TCPIP_ADAPTER_IF_MAX] = {0, 0, 0};
#ifndef MDNS_TEST_MODE #ifndef MDNS_TEST_MODE
@ -184,7 +194,7 @@ static xSemaphoreHandle _mdns_service_semaphore = NULL;
*/ */
static void _mdns_server_recv(void *arg, struct udp_pcb *upcb, struct pbuf *pb, const ip_addr_t *addr, uint16_t port) static void _mdns_server_recv(void *arg, struct udp_pcb *upcb, struct pbuf *pb, const ip_addr_t *addr, uint16_t port)
{ {
while(pb != NULL) { while (pb != NULL) {
struct pbuf * this_pb = pb; struct pbuf * this_pb = pb;
pb = pb->next; pb = pb->next;
this_pb->next = NULL; this_pb->next = NULL;
@ -195,6 +205,56 @@ static void _mdns_server_recv(void *arg, struct udp_pcb *upcb, struct pbuf *pb,
} }
} }
/**
* @brief init the network of MDNS server (called in tcpip thread context)
*/
static err_t _mdns_server_init_api(struct tcpip_api_call *api_call_msg)
{
mdns_api_call_t *msg = (mdns_api_call_t*)api_call_msg;
mdns_server_t *server = msg->server;
esp_err_t err = ESP_OK;
tcpip_adapter_ip_info_t if_ip_info;
err = tcpip_adapter_get_ip_info(server->tcpip_if, &if_ip_info);
if (err) {
msg->err = err;
return ERR_OK;
}
ip_addr_t laddr;
IP_ADDR4(&laddr, 224, 0, 0, 251);
ip_addr_t multicast_if_addr = IPADDR4_INIT(if_ip_info.ip.addr);
if (igmp_joingroup((const struct ip4_addr *)&multicast_if_addr.u_addr.ip4, (const struct ip4_addr *)&laddr.u_addr.ip4)) {
msg->err = ESP_ERR_INVALID_STATE;
return ERR_OK;
}
struct udp_pcb * pcb = udp_new();
if (!pcb) {
msg->err = ESP_ERR_NO_MEM;
return ERR_OK;
}
pcb->remote_port = MDNS_SERVICE_PORT;
if (udp_bind(pcb, &multicast_if_addr, pcb->remote_port) != 0) {
udp_remove(pcb);
msg->err = ESP_ERR_INVALID_STATE;
return ERR_OK;
}
pcb->mcast_ttl = 1;
ip_addr_copy(pcb->multicast_ip, multicast_if_addr);
ip_addr_copy(pcb->remote_ip, laddr);
server->pcb = pcb;
udp_recv(pcb, &_mdns_server_recv, server);
msg->err = err;
return ERR_OK;
}
/** /**
* @brief init the network of MDNS server * @brief init the network of MDNS server
* *
@ -207,42 +267,29 @@ static void _mdns_server_recv(void *arg, struct udp_pcb *upcb, struct pbuf *pb,
*/ */
esp_err_t _mdns_server_init(mdns_server_t * server) esp_err_t _mdns_server_init(mdns_server_t * server)
{ {
esp_err_t err = ESP_OK; mdns_api_call_t msg = {
.server = server,
};
tcpip_api_call(_mdns_server_init_api, (struct tcpip_api_call*)&msg);
return msg.err;
}
tcpip_adapter_ip_info_t if_ip_info; /**
err = tcpip_adapter_get_ip_info(server->tcpip_if, &if_ip_info); * @brief stop the network of MDNS server (called in tcpip thread context)
if (err) { */
return err; static err_t _mdns_server_deinit_api(struct tcpip_api_call *api_call_msg)
{
mdns_api_call_t *msg = (mdns_api_call_t*)api_call_msg;
mdns_server_t *server = msg->server;
if (server->pcb) {
udp_recv(server->pcb, NULL, NULL);
udp_disconnect(server->pcb);
udp_remove(server->pcb);
server->pcb = NULL;
} }
msg->err = ESP_OK;
ip_addr_t laddr; return ERR_OK;
IP_ADDR4(&laddr, 224, 0, 0, 251);
ip_addr_t multicast_if_addr = IPADDR4_INIT(if_ip_info.ip.addr);
if (igmp_joingroup((const struct ip4_addr *)&multicast_if_addr.u_addr.ip4, (const struct ip4_addr *)&laddr.u_addr.ip4)) {
return ESP_ERR_INVALID_STATE;
}
struct udp_pcb * pcb = udp_new();
if (!pcb) {
return ESP_ERR_NO_MEM;
}
pcb->remote_port = MDNS_SERVICE_PORT;
if (udp_bind(pcb, &multicast_if_addr, pcb->remote_port) != 0) {
udp_remove(pcb);
return ESP_ERR_INVALID_STATE;
}
pcb->mcast_ttl = 1;
ip_addr_copy(pcb->multicast_ip, multicast_if_addr);
ip_addr_copy(pcb->remote_ip, laddr);
server->pcb = pcb;
udp_recv(pcb, &_mdns_server_recv, server);
return err;
} }
/** /**
@ -254,16 +301,39 @@ esp_err_t _mdns_server_init(mdns_server_t * server)
*/ */
esp_err_t _mdns_server_deinit(mdns_server_t * server) esp_err_t _mdns_server_deinit(mdns_server_t * server)
{ {
if (server->pcb) { mdns_api_call_t msg = {
udp_recv(server->pcb, NULL, NULL); .server = server,
udp_disconnect(server->pcb); };
udp_remove(server->pcb); tcpip_api_call(_mdns_server_deinit_api, (struct tcpip_api_call*)&msg);
server->pcb = NULL; return msg.err;
}
return ESP_OK;
} }
#endif #endif
/**
* @brief send packet over UDP (called in tcpip thread context)
*/
static err_t _mdns_server_write_api(struct tcpip_api_call *api_call_msg)
{
mdns_api_call_t *msg = (mdns_api_call_t*)api_call_msg;
mdns_server_t *server = msg->server;
struct pbuf* pbt = pbuf_alloc(PBUF_TRANSPORT, msg->len, PBUF_RAM);
if (pbt == NULL) {
msg->len = 0;
return ERR_OK;
}
uint8_t* dst = (uint8_t *)pbt->payload;
memcpy(dst, msg->data, msg->len);
err_t err = udp_sendto(server->pcb, pbt, &(server->pcb->remote_ip), server->pcb->remote_port);
pbuf_free(pbt);
if (err) {
msg->len = 0;
return ERR_OK;
}
return ERR_OK;
}
/** /**
* @brief send packet over UDP * @brief send packet over UDP
* *
@ -273,22 +343,18 @@ esp_err_t _mdns_server_deinit(mdns_server_t * server)
* *
* @return length of sent packet or 0 on error * @return length of sent packet or 0 on error
*/ */
static size_t _mdns_server_write(mdns_server_t * server, uint8_t * data, size_t len) static size_t _mdns_server_write(mdns_server_t * server, uint8_t * data, size_t len) {
{
#ifndef MDNS_TEST_MODE #ifndef MDNS_TEST_MODE
struct pbuf* pbt = pbuf_alloc(PBUF_TRANSPORT, len, PBUF_RAM); mdns_api_call_t msg = {
if (pbt == NULL) { .server = server,
return 0; .data = data,
} .len = len,
uint8_t* dst = (uint8_t *)pbt->payload; };
memcpy(dst, data, len); tcpip_api_call(_mdns_server_write_api, (struct tcpip_api_call*)&msg);
err_t err = udp_sendto(server->pcb, pbt, &(server->pcb->remote_ip), server->pcb->remote_port); return msg.len;
pbuf_free(pbt); #else
if (err) {
return 0;
}
#endif
return len; return len;
#endif
} }
/* /*
@ -307,10 +373,10 @@ static void _mdns_service_task(void *pvParameters)
struct pbuf * pb; struct pbuf * pb;
QueueSetMemberHandle_t queue; QueueSetMemberHandle_t queue;
for(;;) { for (;;) {
queue = xQueueSelectFromSet(_mdns_queue_set, portMAX_DELAY); queue = xQueueSelectFromSet(_mdns_queue_set, portMAX_DELAY);
if (queue && xQueueReceive(queue, &pb, 0) == pdTRUE) { if (queue && xQueueReceive(queue, &pb, 0) == pdTRUE) {
for(i=0; i<TCPIP_ADAPTER_IF_MAX; i++) { for (i = 0; i < TCPIP_ADAPTER_IF_MAX; i++) {
mdns_server_t * server = _mdns_servers[i]; mdns_server_t * server = _mdns_servers[i];
if (server && server->queue == queue) { if (server && server->queue == queue) {
MDNS_MUTEX_LOCK(); MDNS_MUTEX_LOCK();
@ -412,7 +478,7 @@ static esp_err_t _mdns_server_remove(mdns_server_t * server)
} }
uint8_t i; uint8_t i;
for(i=0; i<TCPIP_ADAPTER_IF_MAX; i++) { for (i = 0; i < TCPIP_ADAPTER_IF_MAX; i++) {
if (_mdns_servers[i]) { if (_mdns_servers[i]) {
break; break;
} }
@ -450,7 +516,7 @@ static mdns_answer_item_t * _mdns_add_answer(mdns_answer_item_t * answers, mdns_
{ {
//see if we already have the service queued //see if we already have the service queued
mdns_answer_item_t * a = answers; mdns_answer_item_t * a = answers;
while(a) { while (a) {
if (a->service == service) { if (a->service == service) {
//just add the new answer type to it //just add the new answer type to it
a->answer |= type; a->answer |= type;
@ -484,7 +550,7 @@ static mdns_answer_item_t * _mdns_add_answer(mdns_answer_item_t * answers, mdns_
static const uint8_t * _mdns_read_fqdn(const uint8_t * packet, const uint8_t * start, mdns_name_t * name, char * buf) static const uint8_t * _mdns_read_fqdn(const uint8_t * packet, const uint8_t * start, mdns_name_t * name, char * buf)
{ {
size_t index = 0; size_t index = 0;
while(start[index]) { while (start[index]) {
if (name->parts == 4) { if (name->parts == 4) {
return NULL; return NULL;
} }
@ -495,7 +561,7 @@ static const uint8_t * _mdns_read_fqdn(const uint8_t * packet, const uint8_t * s
return NULL; return NULL;
} }
uint8_t i; uint8_t i;
for(i=0; i<len; i++) { for (i = 0; i < len; i++) {
buf[i] = start[index++]; buf[i] = start[index++];
} }
buf[len] = '\0'; buf[len] = '\0';
@ -507,7 +573,7 @@ static const uint8_t * _mdns_read_fqdn(const uint8_t * packet, const uint8_t * s
} else if (strcmp(buf, MDNS_SUB_STR) == 0) { } else if (strcmp(buf, MDNS_SUB_STR) == 0) {
name->sub = 1; name->sub = 1;
} else { } else {
memcpy((uint8_t*)name + (name->parts++ * (MDNS_NAME_BUF_LEN)), buf, len+1); memcpy((uint8_t*)name + (name->parts++ * (MDNS_NAME_BUF_LEN)), buf, len + 1);
} }
} else { } else {
size_t address = (((uint16_t)len & 0x3F) << 8) | start[index++]; size_t address = (((uint16_t)len & 0x3F) << 8) | start[index++];
@ -549,7 +615,7 @@ static const uint8_t * _mdns_parse_fqdn(const uint8_t * packet, const uint8_t *
return 0; return 0;
} }
if (name->parts == 3) { if (name->parts == 3) {
memmove((uint8_t*)name + (MDNS_NAME_BUF_LEN), (uint8_t*)name, 3*(MDNS_NAME_BUF_LEN)); memmove((uint8_t*)name + (MDNS_NAME_BUF_LEN), (uint8_t*)name, 3 * (MDNS_NAME_BUF_LEN));
name->host[0] = 0; name->host[0] = 0;
} else if (name->parts == 2) { } else if (name->parts == 2) {
memmove((uint8_t*)(name->domain), (uint8_t*)(name->service), (MDNS_NAME_BUF_LEN)); memmove((uint8_t*)(name->domain), (uint8_t*)(name->service), (MDNS_NAME_BUF_LEN));
@ -579,7 +645,7 @@ static inline void _mdns_set_u16(uint8_t * packet, uint16_t index, uint16_t valu
return; return;
} }
packet[index] = (value >> 8) & 0xFF; packet[index] = (value >> 8) & 0xFF;
packet[index+1] = value & 0xFF; packet[index + 1] = value & 0xFF;
} }
/** /**
@ -719,10 +785,10 @@ static uint16_t _mdns_append_fqdn(uint8_t * packet, uint16_t * index, const char
static char buf[MDNS_NAME_BUF_LEN]; static char buf[MDNS_NAME_BUF_LEN];
uint8_t len = strlen(strings[0]); uint8_t len = strlen(strings[0]);
uint8_t * len_location = (uint8_t *)memchr(packet, (char)len, *index); uint8_t * len_location = (uint8_t *)memchr(packet, (char)len, *index);
while(len_location) { while (len_location) {
if (memcmp(len_location+1, strings[0], len)) { //not continuing with our string if (memcmp(len_location + 1, strings[0], len)) { //not continuing with our string
search_next: search_next:
len_location = (uint8_t *)memchr(len_location+1, (char)len, *index - (len_location+1 - packet)); len_location = (uint8_t *)memchr(len_location + 1, (char)len, *index - (len_location + 1 - packet));
continue; continue;
} }
//read string into name and compare //read string into name and compare
@ -738,7 +804,7 @@ search_next:
} }
if (name.parts == count) { if (name.parts == count) {
uint8_t i; uint8_t i;
for(i=0; i<count; i++) { for (i = 0; i < count; i++) {
if (strcmp(strings[i], (const char *)&name + (i * (MDNS_NAME_BUF_LEN)))) { if (strcmp(strings[i], (const char *)&name + (i * (MDNS_NAME_BUF_LEN)))) {
//not our string //not our string
goto search_next; goto search_next;
@ -779,9 +845,9 @@ static uint16_t _mdns_append_ptr_record(uint8_t * packet, uint16_t * index, mdns
uint16_t record_length = 0; uint16_t record_length = 0;
uint8_t part_length; uint8_t part_length;
str[0] = (service->instance)?service->instance str[0] = (service->instance) ? service->instance
:(server->instance)?server->instance : (server->instance) ? server->instance
:server->hostname; : server->hostname;
str[1] = service->service; str[1] = service->service;
str[2] = service->proto; str[2] = service->proto;
str[3] = MDNS_DEFAULT_DOMAIN; str[3] = MDNS_DEFAULT_DOMAIN;
@ -824,7 +890,7 @@ static uint16_t _mdns_append_sdptr_record(uint8_t * packet, uint16_t * index, md
const char * sd_str[4]; const char * sd_str[4];
uint16_t record_length = 0; uint16_t record_length = 0;
uint8_t part_length; uint8_t part_length;
sd_str[0] = (char*)"_services"; sd_str[0] = (char*)"_services";
sd_str[1] = (char*)"_dns-sd"; sd_str[1] = (char*)"_dns-sd";
sd_str[2] = (char*)"_udp"; sd_str[2] = (char*)"_udp";
@ -835,7 +901,7 @@ static uint16_t _mdns_append_sdptr_record(uint8_t * packet, uint16_t * index, md
str[2] = MDNS_DEFAULT_DOMAIN; str[2] = MDNS_DEFAULT_DOMAIN;
part_length = _mdns_append_fqdn(packet, index, sd_str, 4); part_length = _mdns_append_fqdn(packet, index, sd_str, 4);
record_length += part_length; record_length += part_length;
part_length = _mdns_append_type(packet, index, MDNS_ANSWER_PTR, MDNS_ANSWER_PTR_TTL); part_length = _mdns_append_type(packet, index, MDNS_ANSWER_PTR, MDNS_ANSWER_PTR_TTL);
@ -870,9 +936,9 @@ static uint16_t _mdns_append_txt_record(uint8_t * packet, uint16_t * index, mdns
uint16_t record_length = 0; uint16_t record_length = 0;
uint8_t part_length; uint8_t part_length;
str[0] = (service->instance)?service->instance str[0] = (service->instance) ? service->instance
:(server->instance)?server->instance : (server->instance) ? server->instance
:server->hostname; : server->hostname;
str[1] = service->service; str[1] = service->service;
str[2] = service->proto; str[2] = service->proto;
str[3] = MDNS_DEFAULT_DOMAIN; str[3] = MDNS_DEFAULT_DOMAIN;
@ -895,10 +961,10 @@ static uint16_t _mdns_append_txt_record(uint8_t * packet, uint16_t * index, mdns
uint8_t len = service->txt_num_items; uint8_t len = service->txt_num_items;
const char ** txt = service->txt; const char ** txt = service->txt;
uint8_t i, l; uint8_t i, l;
for(i=0; i<len; i++) { for (i = 0; i < len; i++) {
l = _mdns_append_string(packet, index, txt[i]); l = _mdns_append_string(packet, index, txt[i]);
if (!l) { if (!l) {
return 0; return 0;
} }
data_len += l; data_len += l;
} }
@ -924,9 +990,9 @@ static uint16_t _mdns_append_srv_record(uint8_t * packet, uint16_t * index, mdns
uint16_t record_length = 0; uint16_t record_length = 0;
uint8_t part_length; uint8_t part_length;
str[0] = (service->instance)?service->instance str[0] = (service->instance) ? service->instance
:(server->instance)?server->instance : (server->instance) ? server->instance
:server->hostname; : server->hostname;
str[1] = service->service; str[1] = service->service;
str[2] = service->proto; str[2] = service->proto;
str[3] = MDNS_DEFAULT_DOMAIN; str[3] = MDNS_DEFAULT_DOMAIN;
@ -1047,7 +1113,7 @@ static uint16_t _mdns_append_aaaa_record(uint8_t * packet, uint16_t * index, mdn
if ((*index + 15) >= MDNS_MAX_PACKET_SIZE) { if ((*index + 15) >= MDNS_MAX_PACKET_SIZE) {
return 0; return 0;
} }
part_length = sizeof(ip6_addr_t); part_length = sizeof(ip6_addr_t);
memcpy(packet + *index, ipv6, part_length); memcpy(packet + *index, ipv6, part_length);
*index += part_length; *index += part_length;
@ -1073,7 +1139,7 @@ static void _mdns_send_answers(mdns_server_t * server, mdns_answer_item_t * answ
_mdns_set_u16(packet, MDNS_HEAD_FLAGS_OFFSET, MDNS_FLAGS_AUTHORITATIVE); _mdns_set_u16(packet, MDNS_HEAD_FLAGS_OFFSET, MDNS_FLAGS_AUTHORITATIVE);
while(answers) { while (answers) {
if (answers->answer & MDNS_ANSWER_A) { if (answers->answer & MDNS_ANSWER_A) {
answers->answer &= ~MDNS_ANSWER_A; answers->answer &= ~MDNS_ANSWER_A;
send_ip = true; send_ip = true;
@ -1100,7 +1166,7 @@ static void _mdns_send_answers(mdns_server_t * server, mdns_answer_item_t * answ
} }
answer_count += 1; answer_count += 1;
} }
if (answers->answer & MDNS_ANSWER_SDPTR) { if (answers->answer & MDNS_ANSWER_SDPTR) {
if (!_mdns_append_sdptr_record(packet, &index, server, answers->service)) { if (!_mdns_append_sdptr_record(packet, &index, server, answers->service)) {
return; return;
@ -1127,12 +1193,12 @@ static void _mdns_send_answers(mdns_server_t * server, mdns_answer_item_t * answ
uint8_t * v6addr = (uint8_t*)if_ip6.addr; uint8_t * v6addr = (uint8_t*)if_ip6.addr;
//check if not 0 //check if not 0
int i; int i;
for(i=0;i<sizeof(ip6_addr_t);i++) { for (i = 0; i < sizeof(ip6_addr_t); i++) {
if (v6addr[i]) { if (v6addr[i]) {
break; break;
} }
} }
if (i<sizeof(ip6_addr_t)) { if (i < sizeof(ip6_addr_t)) {
if (!_mdns_append_aaaa_record(packet, &index, server, v6addr)) { if (!_mdns_append_aaaa_record(packet, &index, server, v6addr)) {
return; return;
} }
@ -1216,7 +1282,7 @@ static void _mdns_add_result(mdns_server_t * server, mdns_result_temp_t * r)
static mdns_srv_item_t * _mdns_get_service_item(mdns_server_t * server, const char * service, const char * proto) static mdns_srv_item_t * _mdns_get_service_item(mdns_server_t * server, const char * service, const char * proto)
{ {
mdns_srv_item_t * s = server->services; mdns_srv_item_t * s = server->services;
while(s) { while (s) {
if (!strcmp(s->service->service, service) && !strcmp(s->service->proto, proto)) { if (!strcmp(s->service->service, service) && !strcmp(s->service->proto, proto)) {
return s; return s;
} }
@ -1278,7 +1344,7 @@ static void _mdns_free_service(mdns_service_t * service)
free((char *)service->proto); free((char *)service->proto);
if (service->txt_num_items) { if (service->txt_num_items) {
uint8_t i; uint8_t i;
for(i=0; i<service->txt_num_items; i++) { for (i = 0; i < service->txt_num_items; i++) {
free((char *)service->txt[i]); free((char *)service->txt[i]);
} }
} }
@ -1295,7 +1361,7 @@ static void _mdns_free_service(mdns_service_t * service)
*/ */
static inline uint16_t _mdns_read_u16(const uint8_t * packet, uint16_t index) static inline uint16_t _mdns_read_u16(const uint8_t * packet, uint16_t index)
{ {
return (uint16_t)(packet[index]) << 8 | packet[index+1]; return (uint16_t)(packet[index]) << 8 | packet[index + 1];
} }
/** /**
@ -1322,7 +1388,7 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
uint8_t qs = questions; uint8_t qs = questions;
mdns_answer_item_t * answer_items = NULL; mdns_answer_item_t * answer_items = NULL;
while(qs--) { while (qs--) {
content = _mdns_parse_fqdn(data, content, name); content = _mdns_parse_fqdn(data, content, name);
if (!content) { if (!content) {
answers = 0; answers = 0;
@ -1341,13 +1407,13 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
} }
continue; continue;
} }
//is this a dns-sd service discovery meta query? //is this a dns-sd service discovery meta query?
if (!strcmp(name->host, "_services") && !strcmp(name->service, "_dns-sd") && !strcmp(name->proto, "_udp") && !strcmp(name->domain, MDNS_DEFAULT_DOMAIN) && type == MDNS_TYPE_PTR) if (!strcmp(name->host, "_services") && !strcmp(name->service, "_dns-sd") && !strcmp(name->proto, "_udp") && !strcmp(name->domain, MDNS_DEFAULT_DOMAIN) && type == MDNS_TYPE_PTR)
{ {
//add answers for all services //add answers for all services
mdns_srv_item_t * s = server->services; mdns_srv_item_t * s = server->services;
while(s) { while (s) {
if (s->service->service && s->service->proto) { if (s->service->service && s->service->proto) {
answer_items = _mdns_add_answer(answer_items, s->service, MDNS_ANSWER_SDPTR); answer_items = _mdns_add_answer(answer_items, s->service, MDNS_ANSWER_SDPTR);
} }
@ -1370,18 +1436,18 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
answer_items = _mdns_add_answer(answer_items, si->service, MDNS_ANSWER_ALL); answer_items = _mdns_add_answer(answer_items, si->service, MDNS_ANSWER_ALL);
} else if (type == MDNS_TYPE_TXT) { } else if (type == MDNS_TYPE_TXT) {
//match instance/host //match instance/host
const char * host = (si->service->instance)?si->service->instance const char * host = (si->service->instance) ? si->service->instance
:(server->instance)?server->instance : (server->instance) ? server->instance
:server->hostname; : server->hostname;
if (!host || !host[0] || !name->host[0] || strcmp(name->host, host)) { if (!host || !host[0] || !name->host[0] || strcmp(name->host, host)) {
continue; continue;
} }
answer_items = _mdns_add_answer(answer_items, si->service, MDNS_ANSWER_TXT); answer_items = _mdns_add_answer(answer_items, si->service, MDNS_ANSWER_TXT);
} else if (type == MDNS_TYPE_SRV) { } else if (type == MDNS_TYPE_SRV) {
//match instance/host //match instance/host
const char * host = (si->service->instance)?si->service->instance const char * host = (si->service->instance) ? si->service->instance
:(server->instance)?server->instance : (server->instance) ? server->instance
:server->hostname; : server->hostname;
if (!host || !host[0] || !name->host[0] || strcmp(name->host, host)) { if (!host || !host[0] || !name->host[0] || strcmp(name->host, host)) {
continue; continue;
} }
@ -1402,7 +1468,7 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
mdns_result_temp_t * answer = &a; mdns_result_temp_t * answer = &a;
memset(answer, 0, sizeof(mdns_result_temp_t)); memset(answer, 0, sizeof(mdns_result_temp_t));
while(content < (data + len)) { while (content < (data + len)) {
content = _mdns_parse_fqdn(data, content, name); content = _mdns_parse_fqdn(data, content, name);
if (!content) { if (!content) {
return;//error return;//error
@ -1412,7 +1478,7 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
const uint8_t * data_ptr = content + MDNS_DATA_OFFSET; const uint8_t * data_ptr = content + MDNS_DATA_OFFSET;
content = data_ptr + data_len; content = data_ptr + data_len;
if(content > (data + len)){ if (content > (data + len)) {
return; return;
} }
@ -1461,21 +1527,21 @@ void mdns_parse_packet(mdns_server_t * server, const uint8_t * data, size_t len)
strlcpy(answer->host, name->host, MDNS_NAME_BUF_LEN); strlcpy(answer->host, name->host, MDNS_NAME_BUF_LEN);
} }
} else if (type == MDNS_TYPE_TXT) { } else if (type == MDNS_TYPE_TXT) {
uint16_t i=0,b=0, y; uint16_t i = 0, b = 0, y;
while(i < data_len) { while (i < data_len) {
uint8_t partLen = data_ptr[i++]; uint8_t partLen = data_ptr[i++];
if((i+partLen) > data_len){ if ((i + partLen) > data_len) {
break;//error break;//error
} }
//check if partLen will fit in the buffer //check if partLen will fit in the buffer
if (partLen > (MDNS_TXT_MAX_LEN - b - 1)) { if (partLen > (MDNS_TXT_MAX_LEN - b - 1)) {
break; break;
} }
for(y=0; y<partLen; y++) { for (y = 0; y < partLen; y++) {
char d = data_ptr[i++]; char d = data_ptr[i++];
answer->txt[b++] = d; answer->txt[b++] = d;
} }
if (i<data_len) { if (i < data_len) {
answer->txt[b++] = '&'; answer->txt[b++] = '&';
} }
} }
@ -1613,7 +1679,7 @@ void mdns_free(mdns_server_t * server)
free((char*)server->instance); free((char*)server->instance);
if (server->queue) { if (server->queue) {
struct pbuf * c; struct pbuf * c;
while(xQueueReceive(server->queue, &c, 0) == pdTRUE) { while (xQueueReceive(server->queue, &c, 0) == pdTRUE) {
pbuf_free(c); pbuf_free(c);
} }
vQueueDelete(server->queue); vQueueDelete(server->queue);
@ -1725,7 +1791,7 @@ esp_err_t mdns_service_txt_set(mdns_server_t * server, const char * service, con
MDNS_MUTEX_LOCK(); MDNS_MUTEX_LOCK();
if (s->service->txt_num_items) { if (s->service->txt_num_items) {
uint8_t i; uint8_t i;
for(i=0; i<s->service->txt_num_items; i++) { for (i = 0; i < s->service->txt_num_items; i++) {
free((char *)s->service->txt[i]); free((char *)s->service->txt[i]);
} }
} }
@ -1738,7 +1804,7 @@ esp_err_t mdns_service_txt_set(mdns_server_t * server, const char * service, con
} }
uint8_t i; uint8_t i;
s->service->txt_num_items = num_items; s->service->txt_num_items = num_items;
for(i=0; i<num_items; i++) { for (i = 0; i < num_items; i++) {
s->service->txt[i] = strdup(txt[i]); s->service->txt[i] = strdup(txt[i]);
if (!s->service->txt[i]) { if (!s->service->txt[i]) {
s->service->txt_num_items = i; s->service->txt_num_items = i;
@ -1796,7 +1862,7 @@ esp_err_t mdns_service_remove(mdns_server_t * server, const char * service, cons
} }
//not first item //not first item
mdns_srv_item_t * a = server->services; mdns_srv_item_t * a = server->services;
while(a->next && a->next != s) { while (a->next && a->next != s) {
a = a->next; a = a->next;
} }
//next item of the current item is our item //next item of the current item is our item
@ -1823,7 +1889,7 @@ esp_err_t mdns_service_remove_all(mdns_server_t * server)
MDNS_MUTEX_LOCK(); MDNS_MUTEX_LOCK();
mdns_srv_item_t * a = server->services; mdns_srv_item_t * a = server->services;
server->services = NULL; server->services = NULL;
while(a) { while (a) {
mdns_srv_item_t * s = a; mdns_srv_item_t * s = a;
a = a->next; a = a->next;
_mdns_free_service(s->service); _mdns_free_service(s->service);
@ -1899,7 +1965,7 @@ size_t mdns_query(mdns_server_t * server, const char * service, const char * pro
server->search.running = true; server->search.running = true;
if (timeout) { if (timeout) {
uint32_t startAt = xTaskGetTickCount() * portTICK_PERIOD_MS; uint32_t startAt = xTaskGetTickCount() * portTICK_PERIOD_MS;
while(server->search.running && ((xTaskGetTickCount() * portTICK_PERIOD_MS) - startAt) < timeout) { while (server->search.running && ((xTaskGetTickCount() * portTICK_PERIOD_MS) - startAt) < timeout) {
vTaskDelay(1 / portTICK_PERIOD_MS); vTaskDelay(1 / portTICK_PERIOD_MS);
} }
server->search.running = false; server->search.running = false;
@ -1924,7 +1990,7 @@ esp_err_t mdns_result_free(mdns_server_t * server)
if (!server || server->search.running || !server->search.results) { if (!server || server->search.running || !server->search.results) {
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
} }
while(server->search.results) { while (server->search.results) {
const mdns_result_t * r = server->search.results; const mdns_result_t * r = server->search.results;
server->search.results = (mdns_result_t *)r->next; server->search.results = (mdns_result_t *)r->next;
free((char *)r->host); free((char *)r->host);
@ -1943,7 +2009,7 @@ size_t mdns_result_get_count(mdns_server_t * server)
} }
size_t len = 0; size_t len = 0;
const mdns_result_t * r = server->search.results; const mdns_result_t * r = server->search.results;
while(r) { while (r) {
len++; len++;
r = r->next; r = r->next;
} }
@ -1957,7 +2023,7 @@ const mdns_result_t * mdns_result_get(mdns_server_t * server, size_t num)
} }
size_t len = 0; size_t len = 0;
const mdns_result_t * r = server->search.results; const mdns_result_t * r = server->search.results;
while(r) { while (r) {
if (len++ == num) { if (len++ == num) {
return r; return r;
} }