mirror of
				https://github.com/espressif/esp-idf.git
				synced 2025-10-31 04:59:55 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			438 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			438 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // Copyright 2015-2017 Espressif Systems (Shanghai) PTE LTD
 | |
| //
 | |
| // Licensed under the Apache License, Version 2.0 (the "License");
 | |
| // you may not use this file except in compliance with the License.
 | |
| // You may obtain a copy of the License at
 | |
| 
 | |
| //     http://www.apache.org/licenses/LICENSE-2.0
 | |
| //
 | |
| // Unless required by applicable law or agreed to in writing, software
 | |
| // distributed under the License is distributed on an "AS IS" BASIS,
 | |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| // See the License for the specific language governing permissions and
 | |
| // limitations under the License.
 | |
| #ifndef MDNS_PRIVATE_H_
 | |
| #define MDNS_PRIVATE_H_
 | |
| 
 | |
| #include "esp_event_base.h"
 | |
| #include "esp_task.h"
 | |
| #include "esp_timer.h"
 | |
| 
 | |
| //#define MDNS_ENABLE_DEBUG
 | |
| 
 | |
| #ifdef MDNS_ENABLE_DEBUG
 | |
| #define _mdns_dbg_printf(...) printf(__VA_ARGS__)
 | |
| #endif
 | |
| 
 | |
| /** mDNS strict mode: Set this to 1 for the mDNS library to strictly follow the RFC6762:
 | |
|  * Strict features:
 | |
|  *   - to do not set original questions in response packets per RFC6762, sec 6
 | |
|  *
 | |
|  * The actual configuration is 0, i.e. non-strict mode, since some implementations,
 | |
|  * such as lwIP mdns resolver (used by standard POSIX API like getaddrinfo, gethostbyname)
 | |
|  * could not correctly resolve advertised names.
 | |
|  */
 | |
| #define MDNS_STRICT_MODE 0
 | |
| 
 | |
| #if !MDNS_STRICT_MODE
 | |
| /* mDNS responders sometimes repeat queries in responses
 | |
|  * but according to RFC6762, sec 6: Responses MUST NOT contain
 | |
|  * any item in question field */
 | |
| #define  MDNS_REPEAT_QUERY_IN_RESPONSE 1
 | |
| #endif
 | |
| /** The maximum number of services */
 | |
| #define MDNS_MAX_SERVICES           CONFIG_MDNS_MAX_SERVICES
 | |
| 
 | |
| #define MDNS_ANSWER_PTR_TTL         4500
 | |
| #define MDNS_ANSWER_TXT_TTL         4500
 | |
| #define MDNS_ANSWER_SRV_TTL         120
 | |
| #define MDNS_ANSWER_A_TTL           120
 | |
| #define MDNS_ANSWER_AAAA_TTL        120
 | |
| 
 | |
| #define MDNS_FLAGS_AUTHORITATIVE    0x8400
 | |
| #define MDNS_FLAGS_DISTRIBUTED      0x0200
 | |
| 
 | |
| #define MDNS_NAME_REF               0xC000
 | |
| 
 | |
| //custom type! only used by this implementation
 | |
| //to help manage service discovery handling
 | |
| #define MDNS_TYPE_SDPTR             0x0032
 | |
| 
 | |
| #define MDNS_CLASS_IN               0x0001
 | |
| #define MDNS_CLASS_ANY              0x00FF
 | |
| #define MDNS_CLASS_IN_FLUSH_CACHE   0x8001
 | |
| 
 | |
| #define MDNS_ANSWER_ALL             0x3F
 | |
| #define MDNS_ANSWER_PTR             0x08
 | |
| #define MDNS_ANSWER_TXT             0x04
 | |
| #define MDNS_ANSWER_SRV             0x02
 | |
| #define MDNS_ANSWER_A               0x01
 | |
| #define MDNS_ANSWER_AAAA            0x10
 | |
| #define MDNS_ANSWER_NSEC            0x20
 | |
| #define MDNS_ANSWER_SDPTR           0x80
 | |
| #define MDNS_ANSWER_AAAA_SIZE       16
 | |
| 
 | |
| #define MDNS_SERVICE_PORT           5353                    // UDP port that the server runs on
 | |
| #define MDNS_SERVICE_STACK_DEPTH    CONFIG_MDNS_TASK_STACK_SIZE
 | |
| #define MDNS_TASK_PRIORITY          CONFIG_MDNS_TASK_PRIORITY
 | |
| #if (MDNS_TASK_PRIORITY > ESP_TASK_PRIO_MAX)
 | |
| #error "mDNS task priority is higher than ESP_TASK_PRIO_MAX"
 | |
| #elif (MDNS_TASK_PRIORITY > ESP_TASKD_EVENT_PRIO)
 | |
| #warning "mDNS task priority is higher than ESP_TASKD_EVENT_PRIO, mDNS library might not work correctly"
 | |
| #endif
 | |
| #define MDNS_TASK_AFFINITY          CONFIG_MDNS_TASK_AFFINITY
 | |
| #define MDNS_SERVICE_ADD_TIMEOUT_MS CONFIG_MDNS_SERVICE_ADD_TIMEOUT_MS
 | |
| 
 | |
| #define MDNS_PACKET_QUEUE_LEN       16                      // Maximum packets that can be queued for parsing
 | |
| #define MDNS_ACTION_QUEUE_LEN       16                      // Maximum actions pending to the server
 | |
| #define MDNS_TXT_MAX_LEN            1024                    // Maximum string length of text data in TXT record
 | |
| #define MDNS_NAME_MAX_LEN           64                      // Maximum string length of hostname, instance, service and proto
 | |
| #define MDNS_NAME_BUF_LEN           (MDNS_NAME_MAX_LEN+1)   // Maximum char buffer size to hold hostname, instance, service or proto
 | |
| #define MDNS_MAX_PACKET_SIZE        1460                    // Maximum size of mDNS  outgoing packet
 | |
| 
 | |
| #define MDNS_HEAD_LEN               12
 | |
| #define MDNS_HEAD_ID_OFFSET         0
 | |
| #define MDNS_HEAD_FLAGS_OFFSET      2
 | |
| #define MDNS_HEAD_QUESTIONS_OFFSET  4
 | |
| #define MDNS_HEAD_ANSWERS_OFFSET    6
 | |
| #define MDNS_HEAD_SERVERS_OFFSET    8
 | |
| #define MDNS_HEAD_ADDITIONAL_OFFSET 10
 | |
| 
 | |
| #define MDNS_TYPE_OFFSET            0
 | |
| #define MDNS_CLASS_OFFSET           2
 | |
| #define MDNS_TTL_OFFSET             4
 | |
| #define MDNS_LEN_OFFSET             8
 | |
| #define MDNS_DATA_OFFSET            10
 | |
| 
 | |
| #define MDNS_SRV_PRIORITY_OFFSET    0
 | |
| #define MDNS_SRV_WEIGHT_OFFSET      2
 | |
| #define MDNS_SRV_PORT_OFFSET        4
 | |
| #define MDNS_SRV_FQDN_OFFSET        6
 | |
| 
 | |
| #define MDNS_TIMER_PERIOD_US        (CONFIG_MDNS_TIMER_PERIOD_MS*1000)
 | |
| 
 | |
| #define MDNS_SERVICE_LOCK()     xSemaphoreTake(_mdns_service_semaphore, portMAX_DELAY)
 | |
| #define MDNS_SERVICE_UNLOCK()   xSemaphoreGive(_mdns_service_semaphore)
 | |
| 
 | |
| #define queueToEnd(type, queue, item)       \
 | |
|     if (!queue) {                           \
 | |
|         queue = item;                       \
 | |
|     } else {                                \
 | |
|         type * _q = queue;                  \
 | |
|         while (_q->next) { _q = _q->next; } \
 | |
|         _q->next = item;                    \
 | |
|     }
 | |
| 
 | |
| #define queueDetach(type, queue, item)              \
 | |
|     if (queue) {                                    \
 | |
|         if (queue == item) {                        \
 | |
|             queue = queue->next;                    \
 | |
|         } else {                                    \
 | |
|             type * _q = queue;                      \
 | |
|             while (_q->next && _q->next != item) {  \
 | |
|                 _q = _q->next;                      \
 | |
|             }                                       \
 | |
|             if (_q->next == item) {                 \
 | |
|                 _q->next = item->next;              \
 | |
|                 item->next = NULL;                  \
 | |
|             }                                       \
 | |
|         }                                           \
 | |
|     }
 | |
| 
 | |
| #define queueFree(type, queue)  while (queue) { type * _q = queue; queue = queue->next; free(_q); }
 | |
| 
 | |
| #define PCB_STATE_IS_PROBING(s) (s->state > PCB_OFF && s->state < PCB_ANNOUNCE_1)
 | |
| #define PCB_STATE_IS_ANNOUNCING(s) (s->state > PCB_PROBE_3 && s->state < PCB_RUNNING)
 | |
| #define PCB_STATE_IS_RUNNING(s) (s->state == PCB_RUNNING)
 | |
| 
 | |
| #ifndef HOOK_MALLOC_FAILED
 | |
| #define HOOK_MALLOC_FAILED  ESP_LOGE(TAG, "Cannot allocate memory (line: %d, free heap: %d bytes)", __LINE__, esp_get_free_heap_size());
 | |
| #endif
 | |
| 
 | |
| typedef enum {
 | |
|     PCB_OFF, PCB_DUP, PCB_INIT,
 | |
|     PCB_PROBE_1, PCB_PROBE_2, PCB_PROBE_3,
 | |
|     PCB_ANNOUNCE_1, PCB_ANNOUNCE_2, PCB_ANNOUNCE_3,
 | |
|     PCB_RUNNING
 | |
| } mdns_pcb_state_t;
 | |
| 
 | |
| typedef enum {
 | |
|     MDNS_ANSWER, MDNS_NS, MDNS_EXTRA
 | |
| } mdns_parsed_record_type_t;
 | |
| 
 | |
| typedef enum {
 | |
|     ACTION_SYSTEM_EVENT,
 | |
|     ACTION_HOSTNAME_SET,
 | |
|     ACTION_INSTANCE_SET,
 | |
|     ACTION_SERVICE_ADD,
 | |
|     ACTION_SERVICE_DEL,
 | |
|     ACTION_SERVICE_INSTANCE_SET,
 | |
|     ACTION_SERVICE_PORT_SET,
 | |
|     ACTION_SERVICE_TXT_REPLACE,
 | |
|     ACTION_SERVICE_TXT_SET,
 | |
|     ACTION_SERVICE_TXT_DEL,
 | |
|     ACTION_SERVICES_CLEAR,
 | |
|     ACTION_SEARCH_ADD,
 | |
|     ACTION_SEARCH_SEND,
 | |
|     ACTION_SEARCH_END,
 | |
|     ACTION_TX_HANDLE,
 | |
|     ACTION_RX_HANDLE,
 | |
|     ACTION_TASK_STOP,
 | |
|     ACTION_MAX
 | |
| } mdns_action_type_t;
 | |
| 
 | |
| 
 | |
| typedef struct {
 | |
|     uint16_t id;
 | |
|     union {
 | |
|         struct {
 | |
|             uint16_t qr :1;
 | |
|             uint16_t opCode :4;
 | |
|             uint16_t aa :1;
 | |
|             uint16_t tc :1;
 | |
|             uint16_t rd :1;
 | |
|             uint16_t ra :1;
 | |
|             uint16_t z :1;
 | |
|             uint16_t ad :1;
 | |
|             uint16_t cd :1;
 | |
|             uint16_t rCode :4;//response/error code
 | |
|         };
 | |
|         uint16_t value;
 | |
|     } flags;
 | |
|     uint16_t questions; //QDCOUNT
 | |
|     uint16_t answers;   //ANCOUNT
 | |
|     uint16_t servers;   //NSCOUNT
 | |
|     uint16_t additional;//ARCOUNT
 | |
| } mdns_header_t;
 | |
| 
 | |
| typedef struct {
 | |
|     char host[MDNS_NAME_BUF_LEN];
 | |
|     char service[MDNS_NAME_BUF_LEN];
 | |
|     char proto[MDNS_NAME_BUF_LEN];
 | |
|     char domain[MDNS_NAME_BUF_LEN];
 | |
|     uint8_t parts;
 | |
|     uint8_t sub;
 | |
|     bool    invalid;
 | |
| } mdns_name_t;
 | |
| 
 | |
| typedef struct mdns_parsed_question_s {
 | |
|     struct mdns_parsed_question_s * next;
 | |
|     uint16_t type;
 | |
|     bool unicast;
 | |
|     char * host;
 | |
|     char * service;
 | |
|     char * proto;
 | |
|     char * domain;
 | |
| } mdns_parsed_question_t;
 | |
| 
 | |
| typedef struct mdns_parsed_record_s {
 | |
|     struct mdns_parsed_record_s * next;
 | |
|     mdns_parsed_record_type_t record_type;
 | |
|     uint16_t type;
 | |
|     uint16_t clas;
 | |
|     uint8_t flush;
 | |
|     uint32_t ttl;
 | |
|     char * host;
 | |
|     char * service;
 | |
|     char * proto;
 | |
|     char * domain;
 | |
|     uint16_t data_len;
 | |
|     uint8_t *data;
 | |
| } mdns_parsed_record_t;
 | |
| 
 | |
| typedef struct {
 | |
|     mdns_if_t tcpip_if;
 | |
|     mdns_ip_protocol_t ip_protocol;
 | |
|     //struct udp_pcb *pcb;
 | |
|     esp_ip_addr_t src;
 | |
|     uint16_t src_port;
 | |
|     uint8_t multicast;
 | |
|     uint8_t authoritative;
 | |
|     uint8_t probe;
 | |
|     uint8_t discovery;
 | |
|     uint8_t distributed;
 | |
|     mdns_parsed_question_t * questions;
 | |
|     mdns_parsed_record_t * records;
 | |
|     uint16_t id;
 | |
| } mdns_parsed_packet_t;
 | |
| 
 | |
| typedef struct {
 | |
|     mdns_if_t tcpip_if;
 | |
|     mdns_ip_protocol_t ip_protocol;
 | |
|     struct pbuf *pb;
 | |
|     esp_ip_addr_t src;
 | |
|     esp_ip_addr_t dest;
 | |
|     uint16_t src_port;
 | |
|     uint8_t multicast;
 | |
| } mdns_rx_packet_t;
 | |
| 
 | |
| typedef struct mdns_txt_linked_item_s {
 | |
|     const char * key;                       /*!< item key name */
 | |
|     const char * value;                     /*!< item value string */
 | |
|     struct mdns_txt_linked_item_s * next;   /*!< next result, or NULL for the last result in the list */
 | |
| } mdns_txt_linked_item_t;
 | |
| 
 | |
| typedef struct {
 | |
|     const char * instance;
 | |
|     const char * service;
 | |
|     const char * proto;
 | |
|     uint16_t priority;
 | |
|     uint16_t weight;
 | |
|     uint16_t port;
 | |
|     mdns_txt_linked_item_t * txt;
 | |
| } mdns_service_t;
 | |
| 
 | |
| typedef struct mdns_srv_item_s {
 | |
|     struct mdns_srv_item_s * next;
 | |
|     mdns_service_t * service;
 | |
| } mdns_srv_item_t;
 | |
| 
 | |
| typedef struct mdns_out_question_s {
 | |
|     struct mdns_out_question_s * next;
 | |
|     uint16_t type;
 | |
|     bool unicast;
 | |
|     const char * host;
 | |
|     const char * service;
 | |
|     const char * proto;
 | |
|     const char * domain;
 | |
| } mdns_out_question_t;
 | |
| 
 | |
| typedef struct mdns_out_answer_s {
 | |
|     struct mdns_out_answer_s * next;
 | |
|     uint16_t type;
 | |
|     uint8_t bye;
 | |
|     uint8_t flush;
 | |
|     mdns_service_t * service;
 | |
|     const char * custom_instance;
 | |
|     const char * custom_service;
 | |
|     const char * custom_proto;
 | |
| } mdns_out_answer_t;
 | |
| 
 | |
| typedef struct mdns_tx_packet_s {
 | |
|     struct mdns_tx_packet_s * next;
 | |
|     uint32_t send_at;
 | |
|     mdns_if_t tcpip_if;
 | |
|     mdns_ip_protocol_t ip_protocol;
 | |
|     esp_ip_addr_t dst;
 | |
|     uint16_t port;
 | |
|     uint16_t flags;
 | |
|     uint8_t distributed;
 | |
|     mdns_out_question_t * questions;
 | |
|     mdns_out_answer_t * answers;
 | |
|     mdns_out_answer_t * servers;
 | |
|     mdns_out_answer_t * additional;
 | |
|     bool queued;
 | |
|     uint16_t id;
 | |
| } mdns_tx_packet_t;
 | |
| 
 | |
| typedef struct {
 | |
|     mdns_pcb_state_t state;
 | |
|     struct udp_pcb * pcb;
 | |
|     mdns_srv_item_t ** probe_services;
 | |
|     uint8_t probe_services_len;
 | |
|     uint8_t probe_ip;
 | |
|     uint8_t probe_running;
 | |
|     uint16_t failed_probes;
 | |
| } mdns_pcb_t;
 | |
| 
 | |
| typedef enum {
 | |
|     SEARCH_OFF,
 | |
|     SEARCH_INIT,
 | |
|     SEARCH_RUNNING,
 | |
|     SEARCH_MAX
 | |
| } mdns_search_once_state_t;
 | |
| 
 | |
| typedef struct mdns_search_once_s {
 | |
|     struct mdns_search_once_s * next;
 | |
| 
 | |
|     mdns_search_once_state_t state;
 | |
|     uint32_t started_at;
 | |
|     uint32_t sent_at;
 | |
|     uint32_t timeout;
 | |
|     SemaphoreHandle_t done_semaphore;
 | |
|     uint16_t type;
 | |
|     uint8_t max_results;
 | |
|     uint8_t num_results;
 | |
|     char * instance;
 | |
|     char * service;
 | |
|     char * proto;
 | |
|     mdns_result_t * result;
 | |
| } mdns_search_once_t;
 | |
| 
 | |
| typedef struct mdns_server_s {
 | |
|     struct {
 | |
|         mdns_pcb_t pcbs[MDNS_IP_PROTOCOL_MAX];
 | |
|     } interfaces[MDNS_IF_MAX];
 | |
|     const char * hostname;
 | |
|     const char * instance;
 | |
|     mdns_srv_item_t * services;
 | |
|     SemaphoreHandle_t lock;
 | |
|     QueueHandle_t action_queue;
 | |
|     mdns_tx_packet_t * tx_queue_head;
 | |
|     mdns_search_once_t * search_once;
 | |
|     esp_timer_handle_t timer_handle;
 | |
| } mdns_server_t;
 | |
| 
 | |
| typedef struct {
 | |
|     mdns_action_type_t type;
 | |
|     union {
 | |
|         char * hostname;
 | |
|         char * instance;
 | |
|         struct {
 | |
|             esp_event_base_t event_base;
 | |
|             int32_t event_id;
 | |
|             esp_netif_t* interface;
 | |
|         } sys_event;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|         } srv_add;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|         } srv_del;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|             char * instance;
 | |
|         } srv_instance;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|             uint16_t port;
 | |
|         } srv_port;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|             mdns_txt_linked_item_t * txt;
 | |
|         } srv_txt_replace;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|             char * key;
 | |
|             char * value;
 | |
|         } srv_txt_set;
 | |
|         struct {
 | |
|             mdns_srv_item_t * service;
 | |
|             char * key;
 | |
|         } srv_txt_del;
 | |
|         struct {
 | |
|             mdns_search_once_t * search;
 | |
|         } search_add;
 | |
|         struct {
 | |
|             mdns_tx_packet_t * packet;
 | |
|         } tx_handle;
 | |
|         struct {
 | |
|             mdns_rx_packet_t * packet;
 | |
|         } rx_handle;
 | |
|     } data;
 | |
| } mdns_action_t;
 | |
| 
 | |
| /*
 | |
|  * @brief  Convert mnds if to esp-netif handle
 | |
|  *
 | |
|  * @param  tcpip_if     mdns supported interface as internal enum
 | |
|  *
 | |
|  * @return
 | |
|  *     - ptr to esp-netif on success
 | |
|  *     - NULL if no available netif for current interface index
 | |
|  */
 | |
| esp_netif_t *_mdns_get_esp_netif(mdns_if_t tcpip_if);
 | |
| 
 | |
| 
 | |
| #endif /* MDNS_PRIVATE_H_ */
 | 
