1*5d5fbe79SDavid van Moolenbroek /**
2*5d5fbe79SDavid van Moolenbroek * @file
3*5d5fbe79SDavid van Moolenbroek * MDNS responder implementation
4*5d5fbe79SDavid van Moolenbroek *
5*5d5fbe79SDavid van Moolenbroek * @defgroup mdns MDNS
6*5d5fbe79SDavid van Moolenbroek * @ingroup apps
7*5d5fbe79SDavid van Moolenbroek *
8*5d5fbe79SDavid van Moolenbroek * RFC 6762 - Multicast DNS\n
9*5d5fbe79SDavid van Moolenbroek * RFC 6763 - DNS-Based Service Discovery\n
10*5d5fbe79SDavid van Moolenbroek *
11*5d5fbe79SDavid van Moolenbroek * @verbinclude mdns.txt
12*5d5fbe79SDavid van Moolenbroek *
13*5d5fbe79SDavid van Moolenbroek * Things left to implement:
14*5d5fbe79SDavid van Moolenbroek * -------------------------
15*5d5fbe79SDavid van Moolenbroek *
16*5d5fbe79SDavid van Moolenbroek * - Probing/conflict resolution
17*5d5fbe79SDavid van Moolenbroek * - Sending goodbye messages (zero ttl) - shutdown, DHCP lease about to expire, DHCP turned off...
18*5d5fbe79SDavid van Moolenbroek * - Checking that source address of unicast requests are on the same network
19*5d5fbe79SDavid van Moolenbroek * - Limiting multicast responses to 1 per second per resource record
20*5d5fbe79SDavid van Moolenbroek * - Fragmenting replies if required
21*5d5fbe79SDavid van Moolenbroek * - Handling multi-packet known answers
22*5d5fbe79SDavid van Moolenbroek * - Individual known answer detection for all local IPv6 addresses
23*5d5fbe79SDavid van Moolenbroek * - Dynamic size of outgoing packet
24*5d5fbe79SDavid van Moolenbroek */
25*5d5fbe79SDavid van Moolenbroek
26*5d5fbe79SDavid van Moolenbroek /*
27*5d5fbe79SDavid van Moolenbroek * Copyright (c) 2015 Verisure Innovation AB
28*5d5fbe79SDavid van Moolenbroek * All rights reserved.
29*5d5fbe79SDavid van Moolenbroek *
30*5d5fbe79SDavid van Moolenbroek * Redistribution and use in source and binary forms, with or without modification,
31*5d5fbe79SDavid van Moolenbroek * are permitted provided that the following conditions are met:
32*5d5fbe79SDavid van Moolenbroek *
33*5d5fbe79SDavid van Moolenbroek * 1. Redistributions of source code must retain the above copyright notice,
34*5d5fbe79SDavid van Moolenbroek * this list of conditions and the following disclaimer.
35*5d5fbe79SDavid van Moolenbroek * 2. Redistributions in binary form must reproduce the above copyright notice,
36*5d5fbe79SDavid van Moolenbroek * this list of conditions and the following disclaimer in the documentation
37*5d5fbe79SDavid van Moolenbroek * and/or other materials provided with the distribution.
38*5d5fbe79SDavid van Moolenbroek * 3. The name of the author may not be used to endorse or promote products
39*5d5fbe79SDavid van Moolenbroek * derived from this software without specific prior written permission.
40*5d5fbe79SDavid van Moolenbroek *
41*5d5fbe79SDavid van Moolenbroek * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
42*5d5fbe79SDavid van Moolenbroek * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
43*5d5fbe79SDavid van Moolenbroek * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
44*5d5fbe79SDavid van Moolenbroek * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
45*5d5fbe79SDavid van Moolenbroek * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
46*5d5fbe79SDavid van Moolenbroek * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
47*5d5fbe79SDavid van Moolenbroek * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
48*5d5fbe79SDavid van Moolenbroek * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
49*5d5fbe79SDavid van Moolenbroek * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
50*5d5fbe79SDavid van Moolenbroek * OF SUCH DAMAGE.
51*5d5fbe79SDavid van Moolenbroek *
52*5d5fbe79SDavid van Moolenbroek * This file is part of the lwIP TCP/IP stack.
53*5d5fbe79SDavid van Moolenbroek *
54*5d5fbe79SDavid van Moolenbroek * Author: Erik Ekman <erik@kryo.se>
55*5d5fbe79SDavid van Moolenbroek *
56*5d5fbe79SDavid van Moolenbroek */
57*5d5fbe79SDavid van Moolenbroek
58*5d5fbe79SDavid van Moolenbroek #include "lwip/apps/mdns.h"
59*5d5fbe79SDavid van Moolenbroek #include "lwip/apps/mdns_priv.h"
60*5d5fbe79SDavid van Moolenbroek #include "lwip/netif.h"
61*5d5fbe79SDavid van Moolenbroek #include "lwip/udp.h"
62*5d5fbe79SDavid van Moolenbroek #include "lwip/ip_addr.h"
63*5d5fbe79SDavid van Moolenbroek #include "lwip/mem.h"
64*5d5fbe79SDavid van Moolenbroek #include "lwip/prot/dns.h"
65*5d5fbe79SDavid van Moolenbroek
66*5d5fbe79SDavid van Moolenbroek #include <string.h>
67*5d5fbe79SDavid van Moolenbroek
68*5d5fbe79SDavid van Moolenbroek #if LWIP_MDNS_RESPONDER
69*5d5fbe79SDavid van Moolenbroek
70*5d5fbe79SDavid van Moolenbroek #if (LWIP_IPV4 && !LWIP_IGMP)
71*5d5fbe79SDavid van Moolenbroek #error "If you want to use MDNS with IPv4, you have to define LWIP_IGMP=1 in your lwipopts.h"
72*5d5fbe79SDavid van Moolenbroek #endif
73*5d5fbe79SDavid van Moolenbroek #if (LWIP_IPV6 && !LWIP_IPV6_MLD)
74*5d5fbe79SDavid van Moolenbroek #error "If you want to use MDNS with IPv6, you have to define LWIP_IPV6_MLD=1 in your lwipopts.h"
75*5d5fbe79SDavid van Moolenbroek #endif
76*5d5fbe79SDavid van Moolenbroek #if (!LWIP_UDP)
77*5d5fbe79SDavid van Moolenbroek #error "If you want to use MDNS, you have to define LWIP_UDP=1 in your lwipopts.h"
78*5d5fbe79SDavid van Moolenbroek #endif
79*5d5fbe79SDavid van Moolenbroek
80*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
81*5d5fbe79SDavid van Moolenbroek #include "lwip/igmp.h"
82*5d5fbe79SDavid van Moolenbroek /* IPv4 multicast group 224.0.0.251 */
83*5d5fbe79SDavid van Moolenbroek static const ip_addr_t v4group = DNS_MQUERY_IPV4_GROUP_INIT;
84*5d5fbe79SDavid van Moolenbroek #endif
85*5d5fbe79SDavid van Moolenbroek
86*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
87*5d5fbe79SDavid van Moolenbroek #include "lwip/mld6.h"
88*5d5fbe79SDavid van Moolenbroek /* IPv6 multicast group FF02::FB */
89*5d5fbe79SDavid van Moolenbroek static const ip_addr_t v6group = DNS_MQUERY_IPV6_GROUP_INIT;
90*5d5fbe79SDavid van Moolenbroek #endif
91*5d5fbe79SDavid van Moolenbroek
92*5d5fbe79SDavid van Moolenbroek #define MDNS_PORT 5353
93*5d5fbe79SDavid van Moolenbroek #define MDNS_TTL 255
94*5d5fbe79SDavid van Moolenbroek
95*5d5fbe79SDavid van Moolenbroek /* Stored offsets to beginning of domain names
96*5d5fbe79SDavid van Moolenbroek * Used for compression.
97*5d5fbe79SDavid van Moolenbroek */
98*5d5fbe79SDavid van Moolenbroek #define NUM_DOMAIN_OFFSETS 10
99*5d5fbe79SDavid van Moolenbroek #define DOMAIN_JUMP_SIZE 2
100*5d5fbe79SDavid van Moolenbroek #define DOMAIN_JUMP 0xc000
101*5d5fbe79SDavid van Moolenbroek
102*5d5fbe79SDavid van Moolenbroek static u8_t mdns_netif_client_id;
103*5d5fbe79SDavid van Moolenbroek static struct udp_pcb *mdns_pcb;
104*5d5fbe79SDavid van Moolenbroek NETIF_DECLARE_EXT_CALLBACK(netif_callback)
105*5d5fbe79SDavid van Moolenbroek
106*5d5fbe79SDavid van Moolenbroek #define NETIF_TO_HOST(netif) (struct mdns_host*)(netif_get_client_data(netif, mdns_netif_client_id))
107*5d5fbe79SDavid van Moolenbroek
108*5d5fbe79SDavid van Moolenbroek #define TOPDOMAIN_LOCAL "local"
109*5d5fbe79SDavid van Moolenbroek
110*5d5fbe79SDavid van Moolenbroek #define REVERSE_PTR_TOPDOMAIN "arpa"
111*5d5fbe79SDavid van Moolenbroek #define REVERSE_PTR_V4_DOMAIN "in-addr"
112*5d5fbe79SDavid van Moolenbroek #define REVERSE_PTR_V6_DOMAIN "ip6"
113*5d5fbe79SDavid van Moolenbroek
114*5d5fbe79SDavid van Moolenbroek #define SRV_PRIORITY 0
115*5d5fbe79SDavid van Moolenbroek #define SRV_WEIGHT 0
116*5d5fbe79SDavid van Moolenbroek
117*5d5fbe79SDavid van Moolenbroek /* Payload size allocated for each outgoing UDP packet */
118*5d5fbe79SDavid van Moolenbroek #define OUTPACKET_SIZE 500
119*5d5fbe79SDavid van Moolenbroek
120*5d5fbe79SDavid van Moolenbroek /* Lookup from hostname -> IPv4 */
121*5d5fbe79SDavid van Moolenbroek #define REPLY_HOST_A 0x01
122*5d5fbe79SDavid van Moolenbroek /* Lookup from IPv4/v6 -> hostname */
123*5d5fbe79SDavid van Moolenbroek #define REPLY_HOST_PTR_V4 0x02
124*5d5fbe79SDavid van Moolenbroek /* Lookup from hostname -> IPv6 */
125*5d5fbe79SDavid van Moolenbroek #define REPLY_HOST_AAAA 0x04
126*5d5fbe79SDavid van Moolenbroek /* Lookup from hostname -> IPv6 */
127*5d5fbe79SDavid van Moolenbroek #define REPLY_HOST_PTR_V6 0x08
128*5d5fbe79SDavid van Moolenbroek
129*5d5fbe79SDavid van Moolenbroek /* Lookup for service types */
130*5d5fbe79SDavid van Moolenbroek #define REPLY_SERVICE_TYPE_PTR 0x10
131*5d5fbe79SDavid van Moolenbroek /* Lookup for instances of service */
132*5d5fbe79SDavid van Moolenbroek #define REPLY_SERVICE_NAME_PTR 0x20
133*5d5fbe79SDavid van Moolenbroek /* Lookup for location of service instance */
134*5d5fbe79SDavid van Moolenbroek #define REPLY_SERVICE_SRV 0x40
135*5d5fbe79SDavid van Moolenbroek /* Lookup for text info on service instance */
136*5d5fbe79SDavid van Moolenbroek #define REPLY_SERVICE_TXT 0x80
137*5d5fbe79SDavid van Moolenbroek
138*5d5fbe79SDavid van Moolenbroek static const char *dnssd_protos[] = {
139*5d5fbe79SDavid van Moolenbroek "_udp", /* DNSSD_PROTO_UDP */
140*5d5fbe79SDavid van Moolenbroek "_tcp", /* DNSSD_PROTO_TCP */
141*5d5fbe79SDavid van Moolenbroek };
142*5d5fbe79SDavid van Moolenbroek
143*5d5fbe79SDavid van Moolenbroek /** Description of a service */
144*5d5fbe79SDavid van Moolenbroek struct mdns_service {
145*5d5fbe79SDavid van Moolenbroek /** TXT record to answer with */
146*5d5fbe79SDavid van Moolenbroek struct mdns_domain txtdata;
147*5d5fbe79SDavid van Moolenbroek /** Name of service, like 'myweb' */
148*5d5fbe79SDavid van Moolenbroek char name[MDNS_LABEL_MAXLEN + 1];
149*5d5fbe79SDavid van Moolenbroek /** Type of service, like '_http' */
150*5d5fbe79SDavid van Moolenbroek char service[MDNS_LABEL_MAXLEN + 1];
151*5d5fbe79SDavid van Moolenbroek /** Callback function and userdata
152*5d5fbe79SDavid van Moolenbroek * to update txtdata buffer */
153*5d5fbe79SDavid van Moolenbroek service_get_txt_fn_t txt_fn;
154*5d5fbe79SDavid van Moolenbroek void *txt_userdata;
155*5d5fbe79SDavid van Moolenbroek /** TTL in seconds of SRV/TXT replies */
156*5d5fbe79SDavid van Moolenbroek u32_t dns_ttl;
157*5d5fbe79SDavid van Moolenbroek /** Protocol, TCP or UDP */
158*5d5fbe79SDavid van Moolenbroek u16_t proto;
159*5d5fbe79SDavid van Moolenbroek /** Port of the service */
160*5d5fbe79SDavid van Moolenbroek u16_t port;
161*5d5fbe79SDavid van Moolenbroek };
162*5d5fbe79SDavid van Moolenbroek
163*5d5fbe79SDavid van Moolenbroek /** Description of a host/netif */
164*5d5fbe79SDavid van Moolenbroek struct mdns_host {
165*5d5fbe79SDavid van Moolenbroek /** Hostname */
166*5d5fbe79SDavid van Moolenbroek char name[MDNS_LABEL_MAXLEN + 1];
167*5d5fbe79SDavid van Moolenbroek /** Pointer to services */
168*5d5fbe79SDavid van Moolenbroek struct mdns_service *services[MDNS_MAX_SERVICES];
169*5d5fbe79SDavid van Moolenbroek /** TTL in seconds of A/AAAA/PTR replies */
170*5d5fbe79SDavid van Moolenbroek u32_t dns_ttl;
171*5d5fbe79SDavid van Moolenbroek };
172*5d5fbe79SDavid van Moolenbroek
173*5d5fbe79SDavid van Moolenbroek /** Information about received packet */
174*5d5fbe79SDavid van Moolenbroek struct mdns_packet {
175*5d5fbe79SDavid van Moolenbroek /** Sender IP/port */
176*5d5fbe79SDavid van Moolenbroek ip_addr_t source_addr;
177*5d5fbe79SDavid van Moolenbroek u16_t source_port;
178*5d5fbe79SDavid van Moolenbroek /** If packet was received unicast */
179*5d5fbe79SDavid van Moolenbroek u16_t recv_unicast;
180*5d5fbe79SDavid van Moolenbroek /** Netif that received the packet */
181*5d5fbe79SDavid van Moolenbroek struct netif *netif;
182*5d5fbe79SDavid van Moolenbroek /** Packet data */
183*5d5fbe79SDavid van Moolenbroek struct pbuf *pbuf;
184*5d5fbe79SDavid van Moolenbroek /** Current parsing offset in packet */
185*5d5fbe79SDavid van Moolenbroek u16_t parse_offset;
186*5d5fbe79SDavid van Moolenbroek /** Identifier. Used in legacy queries */
187*5d5fbe79SDavid van Moolenbroek u16_t tx_id;
188*5d5fbe79SDavid van Moolenbroek /** Number of questions in packet,
189*5d5fbe79SDavid van Moolenbroek * read from packet header */
190*5d5fbe79SDavid van Moolenbroek u16_t questions;
191*5d5fbe79SDavid van Moolenbroek /** Number of unparsed questions */
192*5d5fbe79SDavid van Moolenbroek u16_t questions_left;
193*5d5fbe79SDavid van Moolenbroek /** Number of answers in packet,
194*5d5fbe79SDavid van Moolenbroek * (sum of normal, authorative and additional answers)
195*5d5fbe79SDavid van Moolenbroek * read from packet header */
196*5d5fbe79SDavid van Moolenbroek u16_t answers;
197*5d5fbe79SDavid van Moolenbroek /** Number of unparsed answers */
198*5d5fbe79SDavid van Moolenbroek u16_t answers_left;
199*5d5fbe79SDavid van Moolenbroek };
200*5d5fbe79SDavid van Moolenbroek
201*5d5fbe79SDavid van Moolenbroek /** Information about outgoing packet */
202*5d5fbe79SDavid van Moolenbroek struct mdns_outpacket {
203*5d5fbe79SDavid van Moolenbroek /** Netif to send the packet on */
204*5d5fbe79SDavid van Moolenbroek struct netif *netif;
205*5d5fbe79SDavid van Moolenbroek /** Packet data */
206*5d5fbe79SDavid van Moolenbroek struct pbuf *pbuf;
207*5d5fbe79SDavid van Moolenbroek /** Current write offset in packet */
208*5d5fbe79SDavid van Moolenbroek u16_t write_offset;
209*5d5fbe79SDavid van Moolenbroek /** Identifier. Used in legacy queries */
210*5d5fbe79SDavid van Moolenbroek u16_t tx_id;
211*5d5fbe79SDavid van Moolenbroek /** Destination IP/port if sent unicast */
212*5d5fbe79SDavid van Moolenbroek ip_addr_t dest_addr;
213*5d5fbe79SDavid van Moolenbroek u16_t dest_port;
214*5d5fbe79SDavid van Moolenbroek /** Number of questions written */
215*5d5fbe79SDavid van Moolenbroek u16_t questions;
216*5d5fbe79SDavid van Moolenbroek /** Number of normal answers written */
217*5d5fbe79SDavid van Moolenbroek u16_t answers;
218*5d5fbe79SDavid van Moolenbroek /** Number of additional answers written */
219*5d5fbe79SDavid van Moolenbroek u16_t additional;
220*5d5fbe79SDavid van Moolenbroek /** Offsets for written domain names in packet.
221*5d5fbe79SDavid van Moolenbroek * Used for compression */
222*5d5fbe79SDavid van Moolenbroek u16_t domain_offsets[NUM_DOMAIN_OFFSETS];
223*5d5fbe79SDavid van Moolenbroek /** If all answers in packet should set cache_flush bit */
224*5d5fbe79SDavid van Moolenbroek u8_t cache_flush;
225*5d5fbe79SDavid van Moolenbroek /** If reply should be sent unicast */
226*5d5fbe79SDavid van Moolenbroek u8_t unicast_reply;
227*5d5fbe79SDavid van Moolenbroek /** If legacy query. (tx_id needed, and write
228*5d5fbe79SDavid van Moolenbroek * question again in reply before answer) */
229*5d5fbe79SDavid van Moolenbroek u8_t legacy_query;
230*5d5fbe79SDavid van Moolenbroek /* Reply bitmask for host information */
231*5d5fbe79SDavid van Moolenbroek u8_t host_replies;
232*5d5fbe79SDavid van Moolenbroek /* Bitmask for which reverse IPv6 hosts to answer */
233*5d5fbe79SDavid van Moolenbroek u8_t host_reverse_v6_replies;
234*5d5fbe79SDavid van Moolenbroek /* Reply bitmask per service */
235*5d5fbe79SDavid van Moolenbroek u8_t serv_replies[MDNS_MAX_SERVICES];
236*5d5fbe79SDavid van Moolenbroek };
237*5d5fbe79SDavid van Moolenbroek
238*5d5fbe79SDavid van Moolenbroek /** Domain, type and class.
239*5d5fbe79SDavid van Moolenbroek * Shared between questions and answers */
240*5d5fbe79SDavid van Moolenbroek struct mdns_rr_info {
241*5d5fbe79SDavid van Moolenbroek struct mdns_domain domain;
242*5d5fbe79SDavid van Moolenbroek u16_t type;
243*5d5fbe79SDavid van Moolenbroek u16_t klass;
244*5d5fbe79SDavid van Moolenbroek };
245*5d5fbe79SDavid van Moolenbroek
246*5d5fbe79SDavid van Moolenbroek struct mdns_question {
247*5d5fbe79SDavid van Moolenbroek struct mdns_rr_info info;
248*5d5fbe79SDavid van Moolenbroek /** unicast reply requested */
249*5d5fbe79SDavid van Moolenbroek u16_t unicast;
250*5d5fbe79SDavid van Moolenbroek };
251*5d5fbe79SDavid van Moolenbroek
252*5d5fbe79SDavid van Moolenbroek struct mdns_answer {
253*5d5fbe79SDavid van Moolenbroek struct mdns_rr_info info;
254*5d5fbe79SDavid van Moolenbroek /** cache flush command bit */
255*5d5fbe79SDavid van Moolenbroek u16_t cache_flush;
256*5d5fbe79SDavid van Moolenbroek /* Validity time in seconds */
257*5d5fbe79SDavid van Moolenbroek u32_t ttl;
258*5d5fbe79SDavid van Moolenbroek /** Length of variable answer */
259*5d5fbe79SDavid van Moolenbroek u16_t rd_length;
260*5d5fbe79SDavid van Moolenbroek /** Offset of start of variable answer in packet */
261*5d5fbe79SDavid van Moolenbroek u16_t rd_offset;
262*5d5fbe79SDavid van Moolenbroek };
263*5d5fbe79SDavid van Moolenbroek
264*5d5fbe79SDavid van Moolenbroek /**
265*5d5fbe79SDavid van Moolenbroek * Add a label part to a domain
266*5d5fbe79SDavid van Moolenbroek * @param domain The domain to add a label to
267*5d5fbe79SDavid van Moolenbroek * @param label The label to add, like <hostname>, 'local', 'com' or ''
268*5d5fbe79SDavid van Moolenbroek * @param len The length of the label
269*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise if label too long
270*5d5fbe79SDavid van Moolenbroek */
271*5d5fbe79SDavid van Moolenbroek err_t
mdns_domain_add_label(struct mdns_domain * domain,const char * label,u8_t len)272*5d5fbe79SDavid van Moolenbroek mdns_domain_add_label(struct mdns_domain *domain, const char *label, u8_t len)
273*5d5fbe79SDavid van Moolenbroek {
274*5d5fbe79SDavid van Moolenbroek if (len > MDNS_LABEL_MAXLEN) {
275*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
276*5d5fbe79SDavid van Moolenbroek }
277*5d5fbe79SDavid van Moolenbroek if (len > 0 && (1 + len + domain->length >= MDNS_DOMAIN_MAXLEN)) {
278*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
279*5d5fbe79SDavid van Moolenbroek }
280*5d5fbe79SDavid van Moolenbroek /* Allow only zero marker on last byte */
281*5d5fbe79SDavid van Moolenbroek if (len == 0 && (1 + domain->length > MDNS_DOMAIN_MAXLEN)) {
282*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
283*5d5fbe79SDavid van Moolenbroek }
284*5d5fbe79SDavid van Moolenbroek domain->name[domain->length] = len;
285*5d5fbe79SDavid van Moolenbroek domain->length++;
286*5d5fbe79SDavid van Moolenbroek if (len) {
287*5d5fbe79SDavid van Moolenbroek MEMCPY(&domain->name[domain->length], label, len);
288*5d5fbe79SDavid van Moolenbroek domain->length += len;
289*5d5fbe79SDavid van Moolenbroek }
290*5d5fbe79SDavid van Moolenbroek return ERR_OK;
291*5d5fbe79SDavid van Moolenbroek }
292*5d5fbe79SDavid van Moolenbroek
293*5d5fbe79SDavid van Moolenbroek /**
294*5d5fbe79SDavid van Moolenbroek * Internal readname function with max 6 levels of recursion following jumps
295*5d5fbe79SDavid van Moolenbroek * while decompressing name
296*5d5fbe79SDavid van Moolenbroek */
297*5d5fbe79SDavid van Moolenbroek static u16_t
mdns_readname_loop(struct pbuf * p,u16_t offset,struct mdns_domain * domain,unsigned depth)298*5d5fbe79SDavid van Moolenbroek mdns_readname_loop(struct pbuf *p, u16_t offset, struct mdns_domain *domain, unsigned depth)
299*5d5fbe79SDavid van Moolenbroek {
300*5d5fbe79SDavid van Moolenbroek u8_t c;
301*5d5fbe79SDavid van Moolenbroek
302*5d5fbe79SDavid van Moolenbroek do {
303*5d5fbe79SDavid van Moolenbroek if (depth > 5) {
304*5d5fbe79SDavid van Moolenbroek /* Too many jumps */
305*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
306*5d5fbe79SDavid van Moolenbroek }
307*5d5fbe79SDavid van Moolenbroek
308*5d5fbe79SDavid van Moolenbroek c = pbuf_get_at(p, offset);
309*5d5fbe79SDavid van Moolenbroek offset++;
310*5d5fbe79SDavid van Moolenbroek
311*5d5fbe79SDavid van Moolenbroek /* is this a compressed label? */
312*5d5fbe79SDavid van Moolenbroek if((c & 0xc0) == 0xc0) {
313*5d5fbe79SDavid van Moolenbroek u16_t jumpaddr;
314*5d5fbe79SDavid van Moolenbroek if (offset >= p->tot_len) {
315*5d5fbe79SDavid van Moolenbroek /* Make sure both jump bytes fit in the packet */
316*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
317*5d5fbe79SDavid van Moolenbroek }
318*5d5fbe79SDavid van Moolenbroek jumpaddr = (((c & 0x3f) << 8) | (pbuf_get_at(p, offset) & 0xff));
319*5d5fbe79SDavid van Moolenbroek offset++;
320*5d5fbe79SDavid van Moolenbroek if (jumpaddr >= SIZEOF_DNS_HDR && jumpaddr < p->tot_len) {
321*5d5fbe79SDavid van Moolenbroek u16_t res;
322*5d5fbe79SDavid van Moolenbroek /* Recursive call, maximum depth will be checked */
323*5d5fbe79SDavid van Moolenbroek res = mdns_readname_loop(p, jumpaddr, domain, depth + 1);
324*5d5fbe79SDavid van Moolenbroek /* Dont return offset since new bytes were not read (jumped to somewhere in packet) */
325*5d5fbe79SDavid van Moolenbroek if (res == MDNS_READNAME_ERROR) {
326*5d5fbe79SDavid van Moolenbroek return res;
327*5d5fbe79SDavid van Moolenbroek }
328*5d5fbe79SDavid van Moolenbroek } else {
329*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
330*5d5fbe79SDavid van Moolenbroek }
331*5d5fbe79SDavid van Moolenbroek break;
332*5d5fbe79SDavid van Moolenbroek }
333*5d5fbe79SDavid van Moolenbroek
334*5d5fbe79SDavid van Moolenbroek /* normal label */
335*5d5fbe79SDavid van Moolenbroek if (c <= MDNS_LABEL_MAXLEN) {
336*5d5fbe79SDavid van Moolenbroek u8_t label[MDNS_LABEL_MAXLEN];
337*5d5fbe79SDavid van Moolenbroek err_t res;
338*5d5fbe79SDavid van Moolenbroek
339*5d5fbe79SDavid van Moolenbroek if (c + domain->length >= MDNS_DOMAIN_MAXLEN) {
340*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
341*5d5fbe79SDavid van Moolenbroek }
342*5d5fbe79SDavid van Moolenbroek if (c != 0) {
343*5d5fbe79SDavid van Moolenbroek if (pbuf_copy_partial(p, label, c, offset) != c) {
344*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
345*5d5fbe79SDavid van Moolenbroek }
346*5d5fbe79SDavid van Moolenbroek offset += c;
347*5d5fbe79SDavid van Moolenbroek }
348*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, (char *) label, c);
349*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
350*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
351*5d5fbe79SDavid van Moolenbroek }
352*5d5fbe79SDavid van Moolenbroek } else {
353*5d5fbe79SDavid van Moolenbroek /* bad length byte */
354*5d5fbe79SDavid van Moolenbroek return MDNS_READNAME_ERROR;
355*5d5fbe79SDavid van Moolenbroek }
356*5d5fbe79SDavid van Moolenbroek } while (c != 0);
357*5d5fbe79SDavid van Moolenbroek
358*5d5fbe79SDavid van Moolenbroek return offset;
359*5d5fbe79SDavid van Moolenbroek }
360*5d5fbe79SDavid van Moolenbroek
361*5d5fbe79SDavid van Moolenbroek /**
362*5d5fbe79SDavid van Moolenbroek * Read possibly compressed domain name from packet buffer
363*5d5fbe79SDavid van Moolenbroek * @param p The packet
364*5d5fbe79SDavid van Moolenbroek * @param offset start position of domain name in packet
365*5d5fbe79SDavid van Moolenbroek * @param domain The domain name destination
366*5d5fbe79SDavid van Moolenbroek * @return The new offset after the domain, or MDNS_READNAME_ERROR
367*5d5fbe79SDavid van Moolenbroek * if reading failed
368*5d5fbe79SDavid van Moolenbroek */
369*5d5fbe79SDavid van Moolenbroek u16_t
mdns_readname(struct pbuf * p,u16_t offset,struct mdns_domain * domain)370*5d5fbe79SDavid van Moolenbroek mdns_readname(struct pbuf *p, u16_t offset, struct mdns_domain *domain)
371*5d5fbe79SDavid van Moolenbroek {
372*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
373*5d5fbe79SDavid van Moolenbroek return mdns_readname_loop(p, offset, domain, 0);
374*5d5fbe79SDavid van Moolenbroek }
375*5d5fbe79SDavid van Moolenbroek
376*5d5fbe79SDavid van Moolenbroek /**
377*5d5fbe79SDavid van Moolenbroek * Print domain name to debug output
378*5d5fbe79SDavid van Moolenbroek * @param domain The domain name
379*5d5fbe79SDavid van Moolenbroek */
380*5d5fbe79SDavid van Moolenbroek static void
mdns_domain_debug_print(struct mdns_domain * domain)381*5d5fbe79SDavid van Moolenbroek mdns_domain_debug_print(struct mdns_domain *domain)
382*5d5fbe79SDavid van Moolenbroek {
383*5d5fbe79SDavid van Moolenbroek u8_t *src = domain->name;
384*5d5fbe79SDavid van Moolenbroek u8_t i;
385*5d5fbe79SDavid van Moolenbroek
386*5d5fbe79SDavid van Moolenbroek while (*src) {
387*5d5fbe79SDavid van Moolenbroek u8_t label_len = *src;
388*5d5fbe79SDavid van Moolenbroek src++;
389*5d5fbe79SDavid van Moolenbroek for (i = 0; i < label_len; i++) {
390*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("%c", src[i]));
391*5d5fbe79SDavid van Moolenbroek }
392*5d5fbe79SDavid van Moolenbroek src += label_len;
393*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("."));
394*5d5fbe79SDavid van Moolenbroek }
395*5d5fbe79SDavid van Moolenbroek }
396*5d5fbe79SDavid van Moolenbroek
397*5d5fbe79SDavid van Moolenbroek /**
398*5d5fbe79SDavid van Moolenbroek * Return 1 if contents of domains match (case-insensitive)
399*5d5fbe79SDavid van Moolenbroek * @param a Domain name to compare 1
400*5d5fbe79SDavid van Moolenbroek * @param b Domain name to compare 2
401*5d5fbe79SDavid van Moolenbroek * @return 1 if domains are equal ignoring case, 0 otherwise
402*5d5fbe79SDavid van Moolenbroek */
403*5d5fbe79SDavid van Moolenbroek int
mdns_domain_eq(struct mdns_domain * a,struct mdns_domain * b)404*5d5fbe79SDavid van Moolenbroek mdns_domain_eq(struct mdns_domain *a, struct mdns_domain *b)
405*5d5fbe79SDavid van Moolenbroek {
406*5d5fbe79SDavid van Moolenbroek u8_t *ptra, *ptrb;
407*5d5fbe79SDavid van Moolenbroek u8_t len;
408*5d5fbe79SDavid van Moolenbroek int res;
409*5d5fbe79SDavid van Moolenbroek
410*5d5fbe79SDavid van Moolenbroek if (a->length != b->length) {
411*5d5fbe79SDavid van Moolenbroek return 0;
412*5d5fbe79SDavid van Moolenbroek }
413*5d5fbe79SDavid van Moolenbroek
414*5d5fbe79SDavid van Moolenbroek ptra = a->name;
415*5d5fbe79SDavid van Moolenbroek ptrb = b->name;
416*5d5fbe79SDavid van Moolenbroek while (*ptra && *ptrb && ptra < &a->name[a->length]) {
417*5d5fbe79SDavid van Moolenbroek if (*ptra != *ptrb) {
418*5d5fbe79SDavid van Moolenbroek return 0;
419*5d5fbe79SDavid van Moolenbroek }
420*5d5fbe79SDavid van Moolenbroek len = *ptra;
421*5d5fbe79SDavid van Moolenbroek ptra++;
422*5d5fbe79SDavid van Moolenbroek ptrb++;
423*5d5fbe79SDavid van Moolenbroek res = lwip_strnicmp((char *) ptra, (char *) ptrb, len);
424*5d5fbe79SDavid van Moolenbroek if (res != 0) {
425*5d5fbe79SDavid van Moolenbroek return 0;
426*5d5fbe79SDavid van Moolenbroek }
427*5d5fbe79SDavid van Moolenbroek ptra += len;
428*5d5fbe79SDavid van Moolenbroek ptrb += len;
429*5d5fbe79SDavid van Moolenbroek }
430*5d5fbe79SDavid van Moolenbroek if (*ptra != *ptrb && ptra < &a->name[a->length]) {
431*5d5fbe79SDavid van Moolenbroek return 0;
432*5d5fbe79SDavid van Moolenbroek }
433*5d5fbe79SDavid van Moolenbroek return 1;
434*5d5fbe79SDavid van Moolenbroek }
435*5d5fbe79SDavid van Moolenbroek
436*5d5fbe79SDavid van Moolenbroek /**
437*5d5fbe79SDavid van Moolenbroek * Call user supplied function to setup TXT data
438*5d5fbe79SDavid van Moolenbroek * @param service The service to build TXT record for
439*5d5fbe79SDavid van Moolenbroek */
440*5d5fbe79SDavid van Moolenbroek static void
mdns_prepare_txtdata(struct mdns_service * service)441*5d5fbe79SDavid van Moolenbroek mdns_prepare_txtdata(struct mdns_service *service)
442*5d5fbe79SDavid van Moolenbroek {
443*5d5fbe79SDavid van Moolenbroek memset(&service->txtdata, 0, sizeof(struct mdns_domain));
444*5d5fbe79SDavid van Moolenbroek if (service->txt_fn) {
445*5d5fbe79SDavid van Moolenbroek service->txt_fn(service, service->txt_userdata);
446*5d5fbe79SDavid van Moolenbroek }
447*5d5fbe79SDavid van Moolenbroek }
448*5d5fbe79SDavid van Moolenbroek
449*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
450*5d5fbe79SDavid van Moolenbroek /**
451*5d5fbe79SDavid van Moolenbroek * Build domain for reverse lookup of IPv4 address
452*5d5fbe79SDavid van Moolenbroek * like 12.0.168.192.in-addr.arpa. for 192.168.0.12
453*5d5fbe79SDavid van Moolenbroek * @param domain Where to write the domain name
454*5d5fbe79SDavid van Moolenbroek * @param addr Pointer to an IPv4 address to encode
455*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if domain was written, an err_t otherwise
456*5d5fbe79SDavid van Moolenbroek */
457*5d5fbe79SDavid van Moolenbroek static err_t
mdns_build_reverse_v4_domain(struct mdns_domain * domain,const ip4_addr_t * addr)458*5d5fbe79SDavid van Moolenbroek mdns_build_reverse_v4_domain(struct mdns_domain *domain, const ip4_addr_t *addr)
459*5d5fbe79SDavid van Moolenbroek {
460*5d5fbe79SDavid van Moolenbroek int i;
461*5d5fbe79SDavid van Moolenbroek err_t res;
462*5d5fbe79SDavid van Moolenbroek const u8_t *ptr;
463*5d5fbe79SDavid van Moolenbroek if (!domain || !addr) {
464*5d5fbe79SDavid van Moolenbroek return ERR_ARG;
465*5d5fbe79SDavid van Moolenbroek }
466*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
467*5d5fbe79SDavid van Moolenbroek ptr = (const u8_t *) addr;
468*5d5fbe79SDavid van Moolenbroek for (i = sizeof(ip4_addr_t) - 1; i >= 0; i--) {
469*5d5fbe79SDavid van Moolenbroek char buf[4];
470*5d5fbe79SDavid van Moolenbroek u8_t val = ptr[i];
471*5d5fbe79SDavid van Moolenbroek
472*5d5fbe79SDavid van Moolenbroek lwip_itoa(buf, sizeof(buf), val);
473*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, buf, (u8_t)strlen(buf));
474*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v4_domain: Failed to add label", (res == ERR_OK), return res);
475*5d5fbe79SDavid van Moolenbroek }
476*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, REVERSE_PTR_V4_DOMAIN, (u8_t)(sizeof(REVERSE_PTR_V4_DOMAIN)-1));
477*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v4_domain: Failed to add label", (res == ERR_OK), return res);
478*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, REVERSE_PTR_TOPDOMAIN, (u8_t)(sizeof(REVERSE_PTR_TOPDOMAIN)-1));
479*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v4_domain: Failed to add label", (res == ERR_OK), return res);
480*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, NULL, 0);
481*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v4_domain: Failed to add label", (res == ERR_OK), return res);
482*5d5fbe79SDavid van Moolenbroek
483*5d5fbe79SDavid van Moolenbroek return ERR_OK;
484*5d5fbe79SDavid van Moolenbroek }
485*5d5fbe79SDavid van Moolenbroek #endif
486*5d5fbe79SDavid van Moolenbroek
487*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
488*5d5fbe79SDavid van Moolenbroek /**
489*5d5fbe79SDavid van Moolenbroek * Build domain for reverse lookup of IP address
490*5d5fbe79SDavid van Moolenbroek * like b.a.9.8.7.6.5.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.b.d.0.1.0.0.2.ip6.arpa. for 2001:db8::567:89ab
491*5d5fbe79SDavid van Moolenbroek * @param domain Where to write the domain name
492*5d5fbe79SDavid van Moolenbroek * @param addr Pointer to an IPv6 address to encode
493*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if domain was written, an err_t otherwise
494*5d5fbe79SDavid van Moolenbroek */
495*5d5fbe79SDavid van Moolenbroek static err_t
mdns_build_reverse_v6_domain(struct mdns_domain * domain,const ip6_addr_t * addr)496*5d5fbe79SDavid van Moolenbroek mdns_build_reverse_v6_domain(struct mdns_domain *domain, const ip6_addr_t *addr)
497*5d5fbe79SDavid van Moolenbroek {
498*5d5fbe79SDavid van Moolenbroek int i;
499*5d5fbe79SDavid van Moolenbroek err_t res;
500*5d5fbe79SDavid van Moolenbroek const u8_t *ptr;
501*5d5fbe79SDavid van Moolenbroek if (!domain || !addr) {
502*5d5fbe79SDavid van Moolenbroek return ERR_ARG;
503*5d5fbe79SDavid van Moolenbroek }
504*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
505*5d5fbe79SDavid van Moolenbroek ptr = (const u8_t *) addr;
506*5d5fbe79SDavid van Moolenbroek for (i = sizeof(ip6_addr_t) - 1; i >= 0; i--) {
507*5d5fbe79SDavid van Moolenbroek char buf;
508*5d5fbe79SDavid van Moolenbroek u8_t byte = ptr[i];
509*5d5fbe79SDavid van Moolenbroek int j;
510*5d5fbe79SDavid van Moolenbroek for (j = 0; j < 2; j++) {
511*5d5fbe79SDavid van Moolenbroek if ((byte & 0x0F) < 0xA) {
512*5d5fbe79SDavid van Moolenbroek buf = '0' + (byte & 0x0F);
513*5d5fbe79SDavid van Moolenbroek } else {
514*5d5fbe79SDavid van Moolenbroek buf = 'a' + (byte & 0x0F) - 0xA;
515*5d5fbe79SDavid van Moolenbroek }
516*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, &buf, sizeof(buf));
517*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v6_domain: Failed to add label", (res == ERR_OK), return res);
518*5d5fbe79SDavid van Moolenbroek byte >>= 4;
519*5d5fbe79SDavid van Moolenbroek }
520*5d5fbe79SDavid van Moolenbroek }
521*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, REVERSE_PTR_V6_DOMAIN, (u8_t)(sizeof(REVERSE_PTR_V6_DOMAIN)-1));
522*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v6_domain: Failed to add label", (res == ERR_OK), return res);
523*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, REVERSE_PTR_TOPDOMAIN, (u8_t)(sizeof(REVERSE_PTR_TOPDOMAIN)-1));
524*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v6_domain: Failed to add label", (res == ERR_OK), return res);
525*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, NULL, 0);
526*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_reverse_v6_domain: Failed to add label", (res == ERR_OK), return res);
527*5d5fbe79SDavid van Moolenbroek
528*5d5fbe79SDavid van Moolenbroek return ERR_OK;
529*5d5fbe79SDavid van Moolenbroek }
530*5d5fbe79SDavid van Moolenbroek #endif
531*5d5fbe79SDavid van Moolenbroek
532*5d5fbe79SDavid van Moolenbroek /* Add .local. to domain */
533*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_dotlocal(struct mdns_domain * domain)534*5d5fbe79SDavid van Moolenbroek mdns_add_dotlocal(struct mdns_domain *domain)
535*5d5fbe79SDavid van Moolenbroek {
536*5d5fbe79SDavid van Moolenbroek err_t res = mdns_domain_add_label(domain, TOPDOMAIN_LOCAL, (u8_t)(sizeof(TOPDOMAIN_LOCAL)-1));
537*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_add_dotlocal: Failed to add label", (res == ERR_OK), return res);
538*5d5fbe79SDavid van Moolenbroek return mdns_domain_add_label(domain, NULL, 0);
539*5d5fbe79SDavid van Moolenbroek }
540*5d5fbe79SDavid van Moolenbroek
541*5d5fbe79SDavid van Moolenbroek /**
542*5d5fbe79SDavid van Moolenbroek * Build the <hostname>.local. domain name
543*5d5fbe79SDavid van Moolenbroek * @param domain Where to write the domain name
544*5d5fbe79SDavid van Moolenbroek * @param mdns TMDNS netif descriptor.
545*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if domain <hostname>.local. was written, an err_t otherwise
546*5d5fbe79SDavid van Moolenbroek */
547*5d5fbe79SDavid van Moolenbroek static err_t
mdns_build_host_domain(struct mdns_domain * domain,struct mdns_host * mdns)548*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(struct mdns_domain *domain, struct mdns_host *mdns)
549*5d5fbe79SDavid van Moolenbroek {
550*5d5fbe79SDavid van Moolenbroek err_t res;
551*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
552*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_host_domain: mdns != NULL", (mdns != NULL), return ERR_VAL);
553*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, mdns->name, (u8_t)strlen(mdns->name));
554*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_host_domain: Failed to add label", (res == ERR_OK), return res);
555*5d5fbe79SDavid van Moolenbroek return mdns_add_dotlocal(domain);
556*5d5fbe79SDavid van Moolenbroek }
557*5d5fbe79SDavid van Moolenbroek
558*5d5fbe79SDavid van Moolenbroek /**
559*5d5fbe79SDavid van Moolenbroek * Build the lookup-all-services special DNS-SD domain name
560*5d5fbe79SDavid van Moolenbroek * @param domain Where to write the domain name
561*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if domain _services._dns-sd._udp.local. was written, an err_t otherwise
562*5d5fbe79SDavid van Moolenbroek */
563*5d5fbe79SDavid van Moolenbroek static err_t
mdns_build_dnssd_domain(struct mdns_domain * domain)564*5d5fbe79SDavid van Moolenbroek mdns_build_dnssd_domain(struct mdns_domain *domain)
565*5d5fbe79SDavid van Moolenbroek {
566*5d5fbe79SDavid van Moolenbroek err_t res;
567*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
568*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, "_services", (u8_t)(sizeof("_services")-1));
569*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_dnssd_domain: Failed to add label", (res == ERR_OK), return res);
570*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, "_dns-sd", (u8_t)(sizeof("_dns-sd")-1));
571*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_dnssd_domain: Failed to add label", (res == ERR_OK), return res);
572*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, dnssd_protos[DNSSD_PROTO_UDP], (u8_t)strlen(dnssd_protos[DNSSD_PROTO_UDP]));
573*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_dnssd_domain: Failed to add label", (res == ERR_OK), return res);
574*5d5fbe79SDavid van Moolenbroek return mdns_add_dotlocal(domain);
575*5d5fbe79SDavid van Moolenbroek }
576*5d5fbe79SDavid van Moolenbroek
577*5d5fbe79SDavid van Moolenbroek /**
578*5d5fbe79SDavid van Moolenbroek * Build domain name for a service
579*5d5fbe79SDavid van Moolenbroek * @param domain Where to write the domain name
580*5d5fbe79SDavid van Moolenbroek * @param service The service struct, containing service name, type and protocol
581*5d5fbe79SDavid van Moolenbroek * @param include_name Whether to include the service name in the domain
582*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if domain was written. If service name is included,
583*5d5fbe79SDavid van Moolenbroek * <name>.<type>.<proto>.local. will be written, otherwise <type>.<proto>.local.
584*5d5fbe79SDavid van Moolenbroek * An err_t is returned on error.
585*5d5fbe79SDavid van Moolenbroek */
586*5d5fbe79SDavid van Moolenbroek static err_t
mdns_build_service_domain(struct mdns_domain * domain,struct mdns_service * service,int include_name)587*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(struct mdns_domain *domain, struct mdns_service *service, int include_name)
588*5d5fbe79SDavid van Moolenbroek {
589*5d5fbe79SDavid van Moolenbroek err_t res;
590*5d5fbe79SDavid van Moolenbroek memset(domain, 0, sizeof(struct mdns_domain));
591*5d5fbe79SDavid van Moolenbroek if (include_name) {
592*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, service->name, (u8_t)strlen(service->name));
593*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_service_domain: Failed to add label", (res == ERR_OK), return res);
594*5d5fbe79SDavid van Moolenbroek }
595*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, service->service, (u8_t)strlen(service->service));
596*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_service_domain: Failed to add label", (res == ERR_OK), return res);
597*5d5fbe79SDavid van Moolenbroek res = mdns_domain_add_label(domain, dnssd_protos[service->proto], (u8_t)strlen(dnssd_protos[service->proto]));
598*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_build_service_domain: Failed to add label", (res == ERR_OK), return res);
599*5d5fbe79SDavid van Moolenbroek return mdns_add_dotlocal(domain);
600*5d5fbe79SDavid van Moolenbroek }
601*5d5fbe79SDavid van Moolenbroek
602*5d5fbe79SDavid van Moolenbroek /**
603*5d5fbe79SDavid van Moolenbroek * Check which replies we should send for a host/netif based on question
604*5d5fbe79SDavid van Moolenbroek * @param netif The network interface that received the question
605*5d5fbe79SDavid van Moolenbroek * @param rr Domain/type/class from a question
606*5d5fbe79SDavid van Moolenbroek * @param reverse_v6_reply Bitmask of which IPv6 addresses to send reverse PTRs for
607*5d5fbe79SDavid van Moolenbroek * if reply bit has REPLY_HOST_PTR_V6 set
608*5d5fbe79SDavid van Moolenbroek * @return Bitmask of which replies to send
609*5d5fbe79SDavid van Moolenbroek */
610*5d5fbe79SDavid van Moolenbroek static int
check_host(struct netif * netif,struct mdns_rr_info * rr,u8_t * reverse_v6_reply)611*5d5fbe79SDavid van Moolenbroek check_host(struct netif *netif, struct mdns_rr_info *rr, u8_t *reverse_v6_reply)
612*5d5fbe79SDavid van Moolenbroek {
613*5d5fbe79SDavid van Moolenbroek err_t res;
614*5d5fbe79SDavid van Moolenbroek int replies = 0;
615*5d5fbe79SDavid van Moolenbroek struct mdns_domain mydomain;
616*5d5fbe79SDavid van Moolenbroek
617*5d5fbe79SDavid van Moolenbroek LWIP_UNUSED_ARG(reverse_v6_reply); /* if ipv6 is disabled */
618*5d5fbe79SDavid van Moolenbroek
619*5d5fbe79SDavid van Moolenbroek if (rr->klass != DNS_RRCLASS_IN && rr->klass != DNS_RRCLASS_ANY) {
620*5d5fbe79SDavid van Moolenbroek /* Invalid class */
621*5d5fbe79SDavid van Moolenbroek return replies;
622*5d5fbe79SDavid van Moolenbroek }
623*5d5fbe79SDavid van Moolenbroek
624*5d5fbe79SDavid van Moolenbroek /* Handle PTR for our addresses */
625*5d5fbe79SDavid van Moolenbroek if (rr->type == DNS_RRTYPE_PTR || rr->type == DNS_RRTYPE_ANY) {
626*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
627*5d5fbe79SDavid van Moolenbroek int i;
628*5d5fbe79SDavid van Moolenbroek for (i = 0; i < LWIP_IPV6_NUM_ADDRESSES; i++) {
629*5d5fbe79SDavid van Moolenbroek if (ip6_addr_isvalid(netif_ip6_addr_state(netif, i))) {
630*5d5fbe79SDavid van Moolenbroek res = mdns_build_reverse_v6_domain(&mydomain, netif_ip6_addr(netif, i));
631*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain)) {
632*5d5fbe79SDavid van Moolenbroek replies |= REPLY_HOST_PTR_V6;
633*5d5fbe79SDavid van Moolenbroek /* Mark which addresses where requested */
634*5d5fbe79SDavid van Moolenbroek if (reverse_v6_reply) {
635*5d5fbe79SDavid van Moolenbroek *reverse_v6_reply |= (1 << i);
636*5d5fbe79SDavid van Moolenbroek }
637*5d5fbe79SDavid van Moolenbroek }
638*5d5fbe79SDavid van Moolenbroek }
639*5d5fbe79SDavid van Moolenbroek }
640*5d5fbe79SDavid van Moolenbroek #endif
641*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
642*5d5fbe79SDavid van Moolenbroek if (!ip4_addr_isany_val(*netif_ip4_addr(netif))) {
643*5d5fbe79SDavid van Moolenbroek res = mdns_build_reverse_v4_domain(&mydomain, netif_ip4_addr(netif));
644*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain)) {
645*5d5fbe79SDavid van Moolenbroek replies |= REPLY_HOST_PTR_V4;
646*5d5fbe79SDavid van Moolenbroek }
647*5d5fbe79SDavid van Moolenbroek }
648*5d5fbe79SDavid van Moolenbroek #endif
649*5d5fbe79SDavid van Moolenbroek }
650*5d5fbe79SDavid van Moolenbroek
651*5d5fbe79SDavid van Moolenbroek res = mdns_build_host_domain(&mydomain, NETIF_TO_HOST(netif));
652*5d5fbe79SDavid van Moolenbroek /* Handle requests for our hostname */
653*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain)) {
654*5d5fbe79SDavid van Moolenbroek /* TODO return NSEC if unsupported protocol requested */
655*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
656*5d5fbe79SDavid van Moolenbroek if (!ip4_addr_isany_val(*netif_ip4_addr(netif))
657*5d5fbe79SDavid van Moolenbroek && (rr->type == DNS_RRTYPE_A || rr->type == DNS_RRTYPE_ANY)) {
658*5d5fbe79SDavid van Moolenbroek replies |= REPLY_HOST_A;
659*5d5fbe79SDavid van Moolenbroek }
660*5d5fbe79SDavid van Moolenbroek #endif
661*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
662*5d5fbe79SDavid van Moolenbroek if (rr->type == DNS_RRTYPE_AAAA || rr->type == DNS_RRTYPE_ANY) {
663*5d5fbe79SDavid van Moolenbroek replies |= REPLY_HOST_AAAA;
664*5d5fbe79SDavid van Moolenbroek }
665*5d5fbe79SDavid van Moolenbroek #endif
666*5d5fbe79SDavid van Moolenbroek }
667*5d5fbe79SDavid van Moolenbroek
668*5d5fbe79SDavid van Moolenbroek return replies;
669*5d5fbe79SDavid van Moolenbroek }
670*5d5fbe79SDavid van Moolenbroek
671*5d5fbe79SDavid van Moolenbroek /**
672*5d5fbe79SDavid van Moolenbroek * Check which replies we should send for a service based on question
673*5d5fbe79SDavid van Moolenbroek * @param service A registered MDNS service
674*5d5fbe79SDavid van Moolenbroek * @param rr Domain/type/class from a question
675*5d5fbe79SDavid van Moolenbroek * @return Bitmask of which replies to send
676*5d5fbe79SDavid van Moolenbroek */
677*5d5fbe79SDavid van Moolenbroek static int
check_service(struct mdns_service * service,struct mdns_rr_info * rr)678*5d5fbe79SDavid van Moolenbroek check_service(struct mdns_service *service, struct mdns_rr_info *rr)
679*5d5fbe79SDavid van Moolenbroek {
680*5d5fbe79SDavid van Moolenbroek err_t res;
681*5d5fbe79SDavid van Moolenbroek int replies = 0;
682*5d5fbe79SDavid van Moolenbroek struct mdns_domain mydomain;
683*5d5fbe79SDavid van Moolenbroek
684*5d5fbe79SDavid van Moolenbroek if (rr->klass != DNS_RRCLASS_IN && rr->klass != DNS_RRCLASS_ANY) {
685*5d5fbe79SDavid van Moolenbroek /* Invalid class */
686*5d5fbe79SDavid van Moolenbroek return 0;
687*5d5fbe79SDavid van Moolenbroek }
688*5d5fbe79SDavid van Moolenbroek
689*5d5fbe79SDavid van Moolenbroek res = mdns_build_dnssd_domain(&mydomain);
690*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain) &&
691*5d5fbe79SDavid van Moolenbroek (rr->type == DNS_RRTYPE_PTR || rr->type == DNS_RRTYPE_ANY)) {
692*5d5fbe79SDavid van Moolenbroek /* Request for all service types */
693*5d5fbe79SDavid van Moolenbroek replies |= REPLY_SERVICE_TYPE_PTR;
694*5d5fbe79SDavid van Moolenbroek }
695*5d5fbe79SDavid van Moolenbroek
696*5d5fbe79SDavid van Moolenbroek res = mdns_build_service_domain(&mydomain, service, 0);
697*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain) &&
698*5d5fbe79SDavid van Moolenbroek (rr->type == DNS_RRTYPE_PTR || rr->type == DNS_RRTYPE_ANY)) {
699*5d5fbe79SDavid van Moolenbroek /* Request for the instance of my service */
700*5d5fbe79SDavid van Moolenbroek replies |= REPLY_SERVICE_NAME_PTR;
701*5d5fbe79SDavid van Moolenbroek }
702*5d5fbe79SDavid van Moolenbroek
703*5d5fbe79SDavid van Moolenbroek res = mdns_build_service_domain(&mydomain, service, 1);
704*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&rr->domain, &mydomain)) {
705*5d5fbe79SDavid van Moolenbroek /* Request for info about my service */
706*5d5fbe79SDavid van Moolenbroek if (rr->type == DNS_RRTYPE_SRV || rr->type == DNS_RRTYPE_ANY) {
707*5d5fbe79SDavid van Moolenbroek replies |= REPLY_SERVICE_SRV;
708*5d5fbe79SDavid van Moolenbroek }
709*5d5fbe79SDavid van Moolenbroek if (rr->type == DNS_RRTYPE_TXT || rr->type == DNS_RRTYPE_ANY) {
710*5d5fbe79SDavid van Moolenbroek replies |= REPLY_SERVICE_TXT;
711*5d5fbe79SDavid van Moolenbroek }
712*5d5fbe79SDavid van Moolenbroek }
713*5d5fbe79SDavid van Moolenbroek
714*5d5fbe79SDavid van Moolenbroek return replies;
715*5d5fbe79SDavid van Moolenbroek }
716*5d5fbe79SDavid van Moolenbroek
717*5d5fbe79SDavid van Moolenbroek /**
718*5d5fbe79SDavid van Moolenbroek * Return bytes needed to write before jump for best result of compressing supplied domain
719*5d5fbe79SDavid van Moolenbroek * against domain in outpacket starting at specified offset.
720*5d5fbe79SDavid van Moolenbroek * If a match is found, offset is updated to where to jump to
721*5d5fbe79SDavid van Moolenbroek * @param pbuf Pointer to pbuf with the partially constructed DNS packet
722*5d5fbe79SDavid van Moolenbroek * @param offset Start position of a domain written earlier. If this location is suitable
723*5d5fbe79SDavid van Moolenbroek * for compression, the pointer is updated to where in the domain to jump to.
724*5d5fbe79SDavid van Moolenbroek * @param domain The domain to write
725*5d5fbe79SDavid van Moolenbroek * @return Number of bytes to write of the new domain before writing a jump to the offset.
726*5d5fbe79SDavid van Moolenbroek * If compression can not be done against this previous domain name, the full new
727*5d5fbe79SDavid van Moolenbroek * domain length is returned.
728*5d5fbe79SDavid van Moolenbroek */
729*5d5fbe79SDavid van Moolenbroek u16_t
mdns_compress_domain(struct pbuf * pbuf,u16_t * offset,struct mdns_domain * domain)730*5d5fbe79SDavid van Moolenbroek mdns_compress_domain(struct pbuf *pbuf, u16_t *offset, struct mdns_domain *domain)
731*5d5fbe79SDavid van Moolenbroek {
732*5d5fbe79SDavid van Moolenbroek struct mdns_domain target;
733*5d5fbe79SDavid van Moolenbroek u16_t target_end;
734*5d5fbe79SDavid van Moolenbroek u8_t target_len;
735*5d5fbe79SDavid van Moolenbroek u8_t writelen = 0;
736*5d5fbe79SDavid van Moolenbroek u8_t *ptr;
737*5d5fbe79SDavid van Moolenbroek if (pbuf == NULL) {
738*5d5fbe79SDavid van Moolenbroek return domain->length;
739*5d5fbe79SDavid van Moolenbroek }
740*5d5fbe79SDavid van Moolenbroek target_end = mdns_readname(pbuf, *offset, &target);
741*5d5fbe79SDavid van Moolenbroek if (target_end == MDNS_READNAME_ERROR) {
742*5d5fbe79SDavid van Moolenbroek return domain->length;
743*5d5fbe79SDavid van Moolenbroek }
744*5d5fbe79SDavid van Moolenbroek target_len = (u8_t)(target_end - *offset);
745*5d5fbe79SDavid van Moolenbroek ptr = domain->name;
746*5d5fbe79SDavid van Moolenbroek while (writelen < domain->length) {
747*5d5fbe79SDavid van Moolenbroek u8_t domainlen = (u8_t)(domain->length - writelen);
748*5d5fbe79SDavid van Moolenbroek u8_t labellen;
749*5d5fbe79SDavid van Moolenbroek if (domainlen <= target.length && domainlen > DOMAIN_JUMP_SIZE) {
750*5d5fbe79SDavid van Moolenbroek /* Compare domains if target is long enough, and we have enough left of the domain */
751*5d5fbe79SDavid van Moolenbroek u8_t targetpos = (u8_t)(target.length - domainlen);
752*5d5fbe79SDavid van Moolenbroek if ((targetpos + DOMAIN_JUMP_SIZE) >= target_len) {
753*5d5fbe79SDavid van Moolenbroek /* We are checking at or beyond a jump in the original, stop looking */
754*5d5fbe79SDavid van Moolenbroek break;
755*5d5fbe79SDavid van Moolenbroek }
756*5d5fbe79SDavid van Moolenbroek if (target.length >= domainlen &&
757*5d5fbe79SDavid van Moolenbroek memcmp(&domain->name[writelen], &target.name[targetpos], domainlen) == 0) {
758*5d5fbe79SDavid van Moolenbroek *offset += targetpos;
759*5d5fbe79SDavid van Moolenbroek return writelen;
760*5d5fbe79SDavid van Moolenbroek }
761*5d5fbe79SDavid van Moolenbroek }
762*5d5fbe79SDavid van Moolenbroek /* Skip to next label in domain */
763*5d5fbe79SDavid van Moolenbroek labellen = *ptr;
764*5d5fbe79SDavid van Moolenbroek writelen += 1 + labellen;
765*5d5fbe79SDavid van Moolenbroek ptr += 1 + labellen;
766*5d5fbe79SDavid van Moolenbroek }
767*5d5fbe79SDavid van Moolenbroek /* Nothing found */
768*5d5fbe79SDavid van Moolenbroek return domain->length;
769*5d5fbe79SDavid van Moolenbroek }
770*5d5fbe79SDavid van Moolenbroek
771*5d5fbe79SDavid van Moolenbroek /**
772*5d5fbe79SDavid van Moolenbroek * Write domain to outpacket. Compression will be attempted,
773*5d5fbe79SDavid van Moolenbroek * unless domain->skip_compression is set.
774*5d5fbe79SDavid van Moolenbroek * @param outpkt The outpacket to write to
775*5d5fbe79SDavid van Moolenbroek * @param domain The domain name to write
776*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
777*5d5fbe79SDavid van Moolenbroek */
778*5d5fbe79SDavid van Moolenbroek static err_t
mdns_write_domain(struct mdns_outpacket * outpkt,struct mdns_domain * domain)779*5d5fbe79SDavid van Moolenbroek mdns_write_domain(struct mdns_outpacket *outpkt, struct mdns_domain *domain)
780*5d5fbe79SDavid van Moolenbroek {
781*5d5fbe79SDavid van Moolenbroek int i;
782*5d5fbe79SDavid van Moolenbroek err_t res;
783*5d5fbe79SDavid van Moolenbroek u16_t writelen = domain->length;
784*5d5fbe79SDavid van Moolenbroek u16_t jump_offset = 0;
785*5d5fbe79SDavid van Moolenbroek u16_t jump;
786*5d5fbe79SDavid van Moolenbroek
787*5d5fbe79SDavid van Moolenbroek if (!domain->skip_compression) {
788*5d5fbe79SDavid van Moolenbroek for (i = 0; i < NUM_DOMAIN_OFFSETS; ++i) {
789*5d5fbe79SDavid van Moolenbroek u16_t offset = outpkt->domain_offsets[i];
790*5d5fbe79SDavid van Moolenbroek if (offset) {
791*5d5fbe79SDavid van Moolenbroek u16_t len = mdns_compress_domain(outpkt->pbuf, &offset, domain);
792*5d5fbe79SDavid van Moolenbroek if (len < writelen) {
793*5d5fbe79SDavid van Moolenbroek writelen = len;
794*5d5fbe79SDavid van Moolenbroek jump_offset = offset;
795*5d5fbe79SDavid van Moolenbroek }
796*5d5fbe79SDavid van Moolenbroek }
797*5d5fbe79SDavid van Moolenbroek }
798*5d5fbe79SDavid van Moolenbroek }
799*5d5fbe79SDavid van Moolenbroek
800*5d5fbe79SDavid van Moolenbroek if (writelen) {
801*5d5fbe79SDavid van Moolenbroek /* Write uncompressed part of name */
802*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(outpkt->pbuf, domain->name, writelen, outpkt->write_offset);
803*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
804*5d5fbe79SDavid van Moolenbroek return res;
805*5d5fbe79SDavid van Moolenbroek }
806*5d5fbe79SDavid van Moolenbroek
807*5d5fbe79SDavid van Moolenbroek /* Store offset of this new domain */
808*5d5fbe79SDavid van Moolenbroek for (i = 0; i < NUM_DOMAIN_OFFSETS; ++i) {
809*5d5fbe79SDavid van Moolenbroek if (outpkt->domain_offsets[i] == 0) {
810*5d5fbe79SDavid van Moolenbroek outpkt->domain_offsets[i] = outpkt->write_offset;
811*5d5fbe79SDavid van Moolenbroek break;
812*5d5fbe79SDavid van Moolenbroek }
813*5d5fbe79SDavid van Moolenbroek }
814*5d5fbe79SDavid van Moolenbroek
815*5d5fbe79SDavid van Moolenbroek outpkt->write_offset += writelen;
816*5d5fbe79SDavid van Moolenbroek }
817*5d5fbe79SDavid van Moolenbroek if (jump_offset) {
818*5d5fbe79SDavid van Moolenbroek /* Write jump */
819*5d5fbe79SDavid van Moolenbroek jump = lwip_htons(DOMAIN_JUMP | jump_offset);
820*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(outpkt->pbuf, &jump, DOMAIN_JUMP_SIZE, outpkt->write_offset);
821*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
822*5d5fbe79SDavid van Moolenbroek return res;
823*5d5fbe79SDavid van Moolenbroek }
824*5d5fbe79SDavid van Moolenbroek outpkt->write_offset += DOMAIN_JUMP_SIZE;
825*5d5fbe79SDavid van Moolenbroek }
826*5d5fbe79SDavid van Moolenbroek return ERR_OK;
827*5d5fbe79SDavid van Moolenbroek }
828*5d5fbe79SDavid van Moolenbroek
829*5d5fbe79SDavid van Moolenbroek /**
830*5d5fbe79SDavid van Moolenbroek * Write a question to an outpacket
831*5d5fbe79SDavid van Moolenbroek * A question contains domain, type and class. Since an answer also starts with these fields this function is also
832*5d5fbe79SDavid van Moolenbroek * called from mdns_add_answer().
833*5d5fbe79SDavid van Moolenbroek * @param outpkt The outpacket to write to
834*5d5fbe79SDavid van Moolenbroek * @param domain The domain name the answer is for
835*5d5fbe79SDavid van Moolenbroek * @param type The DNS type of the answer (like 'AAAA', 'SRV')
836*5d5fbe79SDavid van Moolenbroek * @param klass The DNS type of the answer (like 'IN')
837*5d5fbe79SDavid van Moolenbroek * @param unicast If highest bit in class should be set, to instruct the responder to
838*5d5fbe79SDavid van Moolenbroek * reply with a unicast packet
839*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
840*5d5fbe79SDavid van Moolenbroek */
841*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_question(struct mdns_outpacket * outpkt,struct mdns_domain * domain,u16_t type,u16_t klass,u16_t unicast)842*5d5fbe79SDavid van Moolenbroek mdns_add_question(struct mdns_outpacket *outpkt, struct mdns_domain *domain, u16_t type, u16_t klass, u16_t unicast)
843*5d5fbe79SDavid van Moolenbroek {
844*5d5fbe79SDavid van Moolenbroek u16_t question_len;
845*5d5fbe79SDavid van Moolenbroek u16_t field16;
846*5d5fbe79SDavid van Moolenbroek err_t res;
847*5d5fbe79SDavid van Moolenbroek
848*5d5fbe79SDavid van Moolenbroek if (!outpkt->pbuf) {
849*5d5fbe79SDavid van Moolenbroek /* If no pbuf is active, allocate one */
850*5d5fbe79SDavid van Moolenbroek outpkt->pbuf = pbuf_alloc(PBUF_TRANSPORT, OUTPACKET_SIZE, PBUF_RAM);
851*5d5fbe79SDavid van Moolenbroek if (!outpkt->pbuf) {
852*5d5fbe79SDavid van Moolenbroek return ERR_MEM;
853*5d5fbe79SDavid van Moolenbroek }
854*5d5fbe79SDavid van Moolenbroek outpkt->write_offset = SIZEOF_DNS_HDR;
855*5d5fbe79SDavid van Moolenbroek }
856*5d5fbe79SDavid van Moolenbroek
857*5d5fbe79SDavid van Moolenbroek /* Worst case calculation. Domain string might be compressed */
858*5d5fbe79SDavid van Moolenbroek question_len = domain->length + sizeof(type) + sizeof(klass);
859*5d5fbe79SDavid van Moolenbroek if (outpkt->write_offset + question_len > outpkt->pbuf->tot_len) {
860*5d5fbe79SDavid van Moolenbroek /* No space */
861*5d5fbe79SDavid van Moolenbroek return ERR_MEM;
862*5d5fbe79SDavid van Moolenbroek }
863*5d5fbe79SDavid van Moolenbroek
864*5d5fbe79SDavid van Moolenbroek /* Write name */
865*5d5fbe79SDavid van Moolenbroek res = mdns_write_domain(outpkt, domain);
866*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
867*5d5fbe79SDavid van Moolenbroek return res;
868*5d5fbe79SDavid van Moolenbroek }
869*5d5fbe79SDavid van Moolenbroek
870*5d5fbe79SDavid van Moolenbroek /* Write type */
871*5d5fbe79SDavid van Moolenbroek field16 = lwip_htons(type);
872*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(outpkt->pbuf, &field16, sizeof(field16), outpkt->write_offset);
873*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
874*5d5fbe79SDavid van Moolenbroek return res;
875*5d5fbe79SDavid van Moolenbroek }
876*5d5fbe79SDavid van Moolenbroek outpkt->write_offset += sizeof(field16);
877*5d5fbe79SDavid van Moolenbroek
878*5d5fbe79SDavid van Moolenbroek /* Write class */
879*5d5fbe79SDavid van Moolenbroek if (unicast) {
880*5d5fbe79SDavid van Moolenbroek klass |= 0x8000;
881*5d5fbe79SDavid van Moolenbroek }
882*5d5fbe79SDavid van Moolenbroek field16 = lwip_htons(klass);
883*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(outpkt->pbuf, &field16, sizeof(field16), outpkt->write_offset);
884*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
885*5d5fbe79SDavid van Moolenbroek return res;
886*5d5fbe79SDavid van Moolenbroek }
887*5d5fbe79SDavid van Moolenbroek outpkt->write_offset += sizeof(field16);
888*5d5fbe79SDavid van Moolenbroek
889*5d5fbe79SDavid van Moolenbroek return ERR_OK;
890*5d5fbe79SDavid van Moolenbroek }
891*5d5fbe79SDavid van Moolenbroek
892*5d5fbe79SDavid van Moolenbroek /**
893*5d5fbe79SDavid van Moolenbroek * Write answer to reply packet.
894*5d5fbe79SDavid van Moolenbroek * buf or answer_domain can be null. The rd_length written will be buf_length +
895*5d5fbe79SDavid van Moolenbroek * size of (compressed) domain. Most uses will need either buf or answer_domain,
896*5d5fbe79SDavid van Moolenbroek * special case is SRV that starts with 3 u16 and then a domain name.
897*5d5fbe79SDavid van Moolenbroek * @param reply The outpacket to write to
898*5d5fbe79SDavid van Moolenbroek * @param domain The domain name the answer is for
899*5d5fbe79SDavid van Moolenbroek * @param type The DNS type of the answer (like 'AAAA', 'SRV')
900*5d5fbe79SDavid van Moolenbroek * @param klass The DNS type of the answer (like 'IN')
901*5d5fbe79SDavid van Moolenbroek * @param cache_flush If highest bit in class should be set, to instruct receiver that
902*5d5fbe79SDavid van Moolenbroek * this reply replaces any earlier answer for this domain/type/class
903*5d5fbe79SDavid van Moolenbroek * @param ttl Validity time in seconds to send out for IP address data in DNS replies
904*5d5fbe79SDavid van Moolenbroek * @param buf Pointer to buffer of answer data
905*5d5fbe79SDavid van Moolenbroek * @param buf_length Length of variable data
906*5d5fbe79SDavid van Moolenbroek * @param answer_domain A domain to write after any buffer data as answer
907*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
908*5d5fbe79SDavid van Moolenbroek */
909*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_answer(struct mdns_outpacket * reply,struct mdns_domain * domain,u16_t type,u16_t klass,u16_t cache_flush,u32_t ttl,const u8_t * buf,size_t buf_length,struct mdns_domain * answer_domain)910*5d5fbe79SDavid van Moolenbroek mdns_add_answer(struct mdns_outpacket *reply, struct mdns_domain *domain, u16_t type, u16_t klass, u16_t cache_flush,
911*5d5fbe79SDavid van Moolenbroek u32_t ttl, const u8_t *buf, size_t buf_length, struct mdns_domain *answer_domain)
912*5d5fbe79SDavid van Moolenbroek {
913*5d5fbe79SDavid van Moolenbroek u16_t answer_len;
914*5d5fbe79SDavid van Moolenbroek u16_t field16;
915*5d5fbe79SDavid van Moolenbroek u16_t rdlen_offset;
916*5d5fbe79SDavid van Moolenbroek u16_t answer_offset;
917*5d5fbe79SDavid van Moolenbroek u32_t field32;
918*5d5fbe79SDavid van Moolenbroek err_t res;
919*5d5fbe79SDavid van Moolenbroek
920*5d5fbe79SDavid van Moolenbroek if (!reply->pbuf) {
921*5d5fbe79SDavid van Moolenbroek /* If no pbuf is active, allocate one */
922*5d5fbe79SDavid van Moolenbroek reply->pbuf = pbuf_alloc(PBUF_TRANSPORT, OUTPACKET_SIZE, PBUF_RAM);
923*5d5fbe79SDavid van Moolenbroek if (!reply->pbuf) {
924*5d5fbe79SDavid van Moolenbroek return ERR_MEM;
925*5d5fbe79SDavid van Moolenbroek }
926*5d5fbe79SDavid van Moolenbroek reply->write_offset = SIZEOF_DNS_HDR;
927*5d5fbe79SDavid van Moolenbroek }
928*5d5fbe79SDavid van Moolenbroek
929*5d5fbe79SDavid van Moolenbroek /* Worst case calculation. Domain strings might be compressed */
930*5d5fbe79SDavid van Moolenbroek answer_len = domain->length + sizeof(type) + sizeof(klass) + sizeof(ttl) + sizeof(field16)/*rd_length*/;
931*5d5fbe79SDavid van Moolenbroek if (buf) {
932*5d5fbe79SDavid van Moolenbroek answer_len += (u16_t)buf_length;
933*5d5fbe79SDavid van Moolenbroek }
934*5d5fbe79SDavid van Moolenbroek if (answer_domain) {
935*5d5fbe79SDavid van Moolenbroek answer_len += answer_domain->length;
936*5d5fbe79SDavid van Moolenbroek }
937*5d5fbe79SDavid van Moolenbroek if (reply->write_offset + answer_len > reply->pbuf->tot_len) {
938*5d5fbe79SDavid van Moolenbroek /* No space */
939*5d5fbe79SDavid van Moolenbroek return ERR_MEM;
940*5d5fbe79SDavid van Moolenbroek }
941*5d5fbe79SDavid van Moolenbroek
942*5d5fbe79SDavid van Moolenbroek /* Answer starts with same data as question, then more fields */
943*5d5fbe79SDavid van Moolenbroek mdns_add_question(reply, domain, type, klass, cache_flush);
944*5d5fbe79SDavid van Moolenbroek
945*5d5fbe79SDavid van Moolenbroek /* Write TTL */
946*5d5fbe79SDavid van Moolenbroek field32 = lwip_htonl(ttl);
947*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(reply->pbuf, &field32, sizeof(field32), reply->write_offset);
948*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
949*5d5fbe79SDavid van Moolenbroek return res;
950*5d5fbe79SDavid van Moolenbroek }
951*5d5fbe79SDavid van Moolenbroek reply->write_offset += sizeof(field32);
952*5d5fbe79SDavid van Moolenbroek
953*5d5fbe79SDavid van Moolenbroek /* Store offsets and skip forward to the data */
954*5d5fbe79SDavid van Moolenbroek rdlen_offset = reply->write_offset;
955*5d5fbe79SDavid van Moolenbroek reply->write_offset += sizeof(field16);
956*5d5fbe79SDavid van Moolenbroek answer_offset = reply->write_offset;
957*5d5fbe79SDavid van Moolenbroek
958*5d5fbe79SDavid van Moolenbroek if (buf) {
959*5d5fbe79SDavid van Moolenbroek /* Write static data */
960*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(reply->pbuf, buf, (u16_t)buf_length, reply->write_offset);
961*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
962*5d5fbe79SDavid van Moolenbroek return res;
963*5d5fbe79SDavid van Moolenbroek }
964*5d5fbe79SDavid van Moolenbroek reply->write_offset += (u16_t)buf_length;
965*5d5fbe79SDavid van Moolenbroek }
966*5d5fbe79SDavid van Moolenbroek
967*5d5fbe79SDavid van Moolenbroek if (answer_domain) {
968*5d5fbe79SDavid van Moolenbroek /* Write name answer (compressed if possible) */
969*5d5fbe79SDavid van Moolenbroek res = mdns_write_domain(reply, answer_domain);
970*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
971*5d5fbe79SDavid van Moolenbroek return res;
972*5d5fbe79SDavid van Moolenbroek }
973*5d5fbe79SDavid van Moolenbroek }
974*5d5fbe79SDavid van Moolenbroek
975*5d5fbe79SDavid van Moolenbroek /* Write rd_length after when we know the answer size */
976*5d5fbe79SDavid van Moolenbroek field16 = lwip_htons(reply->write_offset - answer_offset);
977*5d5fbe79SDavid van Moolenbroek res = pbuf_take_at(reply->pbuf, &field16, sizeof(field16), rdlen_offset);
978*5d5fbe79SDavid van Moolenbroek
979*5d5fbe79SDavid van Moolenbroek return res;
980*5d5fbe79SDavid van Moolenbroek }
981*5d5fbe79SDavid van Moolenbroek
982*5d5fbe79SDavid van Moolenbroek /**
983*5d5fbe79SDavid van Moolenbroek * Helper function for mdns_read_question/mdns_read_answer
984*5d5fbe79SDavid van Moolenbroek * Reads a domain, type and class from the packet
985*5d5fbe79SDavid van Moolenbroek * @param pkt The MDNS packet to read from. The parse_offset field will be
986*5d5fbe79SDavid van Moolenbroek * incremented to point to the next unparsed byte.
987*5d5fbe79SDavid van Moolenbroek * @param info The struct to fill with domain, type and class
988*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
989*5d5fbe79SDavid van Moolenbroek */
990*5d5fbe79SDavid van Moolenbroek static err_t
mdns_read_rr_info(struct mdns_packet * pkt,struct mdns_rr_info * info)991*5d5fbe79SDavid van Moolenbroek mdns_read_rr_info(struct mdns_packet *pkt, struct mdns_rr_info *info)
992*5d5fbe79SDavid van Moolenbroek {
993*5d5fbe79SDavid van Moolenbroek u16_t field16, copied;
994*5d5fbe79SDavid van Moolenbroek pkt->parse_offset = mdns_readname(pkt->pbuf, pkt->parse_offset, &info->domain);
995*5d5fbe79SDavid van Moolenbroek if (pkt->parse_offset == MDNS_READNAME_ERROR) {
996*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
997*5d5fbe79SDavid van Moolenbroek }
998*5d5fbe79SDavid van Moolenbroek
999*5d5fbe79SDavid van Moolenbroek copied = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), pkt->parse_offset);
1000*5d5fbe79SDavid van Moolenbroek if (copied != sizeof(field16)) {
1001*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1002*5d5fbe79SDavid van Moolenbroek }
1003*5d5fbe79SDavid van Moolenbroek pkt->parse_offset += copied;
1004*5d5fbe79SDavid van Moolenbroek info->type = lwip_ntohs(field16);
1005*5d5fbe79SDavid van Moolenbroek
1006*5d5fbe79SDavid van Moolenbroek copied = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), pkt->parse_offset);
1007*5d5fbe79SDavid van Moolenbroek if (copied != sizeof(field16)) {
1008*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1009*5d5fbe79SDavid van Moolenbroek }
1010*5d5fbe79SDavid van Moolenbroek pkt->parse_offset += copied;
1011*5d5fbe79SDavid van Moolenbroek info->klass = lwip_ntohs(field16);
1012*5d5fbe79SDavid van Moolenbroek
1013*5d5fbe79SDavid van Moolenbroek return ERR_OK;
1014*5d5fbe79SDavid van Moolenbroek }
1015*5d5fbe79SDavid van Moolenbroek
1016*5d5fbe79SDavid van Moolenbroek /**
1017*5d5fbe79SDavid van Moolenbroek * Read a question from the packet.
1018*5d5fbe79SDavid van Moolenbroek * All questions have to be read before the answers.
1019*5d5fbe79SDavid van Moolenbroek * @param pkt The MDNS packet to read from. The questions_left field will be decremented
1020*5d5fbe79SDavid van Moolenbroek * and the parse_offset will be updated.
1021*5d5fbe79SDavid van Moolenbroek * @param question The struct to fill with question data
1022*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
1023*5d5fbe79SDavid van Moolenbroek */
1024*5d5fbe79SDavid van Moolenbroek static err_t
mdns_read_question(struct mdns_packet * pkt,struct mdns_question * question)1025*5d5fbe79SDavid van Moolenbroek mdns_read_question(struct mdns_packet *pkt, struct mdns_question *question)
1026*5d5fbe79SDavid van Moolenbroek {
1027*5d5fbe79SDavid van Moolenbroek /* Safety check */
1028*5d5fbe79SDavid van Moolenbroek if (pkt->pbuf->tot_len < pkt->parse_offset) {
1029*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1030*5d5fbe79SDavid van Moolenbroek }
1031*5d5fbe79SDavid van Moolenbroek
1032*5d5fbe79SDavid van Moolenbroek if (pkt->questions_left) {
1033*5d5fbe79SDavid van Moolenbroek err_t res;
1034*5d5fbe79SDavid van Moolenbroek pkt->questions_left--;
1035*5d5fbe79SDavid van Moolenbroek
1036*5d5fbe79SDavid van Moolenbroek memset(question, 0, sizeof(struct mdns_question));
1037*5d5fbe79SDavid van Moolenbroek res = mdns_read_rr_info(pkt, &question->info);
1038*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1039*5d5fbe79SDavid van Moolenbroek return res;
1040*5d5fbe79SDavid van Moolenbroek }
1041*5d5fbe79SDavid van Moolenbroek
1042*5d5fbe79SDavid van Moolenbroek /* Extract unicast flag from class field */
1043*5d5fbe79SDavid van Moolenbroek question->unicast = question->info.klass & 0x8000;
1044*5d5fbe79SDavid van Moolenbroek question->info.klass &= 0x7FFF;
1045*5d5fbe79SDavid van Moolenbroek
1046*5d5fbe79SDavid van Moolenbroek return ERR_OK;
1047*5d5fbe79SDavid van Moolenbroek }
1048*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1049*5d5fbe79SDavid van Moolenbroek }
1050*5d5fbe79SDavid van Moolenbroek
1051*5d5fbe79SDavid van Moolenbroek /**
1052*5d5fbe79SDavid van Moolenbroek * Read an answer from the packet
1053*5d5fbe79SDavid van Moolenbroek * The variable length reply is not copied, its pbuf offset and length is stored instead.
1054*5d5fbe79SDavid van Moolenbroek * @param pkt The MDNS packet to read. The answers_left field will be decremented and
1055*5d5fbe79SDavid van Moolenbroek * the parse_offset will be updated.
1056*5d5fbe79SDavid van Moolenbroek * @param answer The struct to fill with answer data
1057*5d5fbe79SDavid van Moolenbroek * @return ERR_OK on success, an err_t otherwise
1058*5d5fbe79SDavid van Moolenbroek */
1059*5d5fbe79SDavid van Moolenbroek static err_t
mdns_read_answer(struct mdns_packet * pkt,struct mdns_answer * answer)1060*5d5fbe79SDavid van Moolenbroek mdns_read_answer(struct mdns_packet *pkt, struct mdns_answer *answer)
1061*5d5fbe79SDavid van Moolenbroek {
1062*5d5fbe79SDavid van Moolenbroek /* Read questions first */
1063*5d5fbe79SDavid van Moolenbroek if (pkt->questions_left) {
1064*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1065*5d5fbe79SDavid van Moolenbroek }
1066*5d5fbe79SDavid van Moolenbroek
1067*5d5fbe79SDavid van Moolenbroek /* Safety check */
1068*5d5fbe79SDavid van Moolenbroek if (pkt->pbuf->tot_len < pkt->parse_offset) {
1069*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1070*5d5fbe79SDavid van Moolenbroek }
1071*5d5fbe79SDavid van Moolenbroek
1072*5d5fbe79SDavid van Moolenbroek if (pkt->answers_left) {
1073*5d5fbe79SDavid van Moolenbroek u16_t copied, field16;
1074*5d5fbe79SDavid van Moolenbroek u32_t ttl;
1075*5d5fbe79SDavid van Moolenbroek err_t res;
1076*5d5fbe79SDavid van Moolenbroek pkt->answers_left--;
1077*5d5fbe79SDavid van Moolenbroek
1078*5d5fbe79SDavid van Moolenbroek memset(answer, 0, sizeof(struct mdns_answer));
1079*5d5fbe79SDavid van Moolenbroek res = mdns_read_rr_info(pkt, &answer->info);
1080*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1081*5d5fbe79SDavid van Moolenbroek return res;
1082*5d5fbe79SDavid van Moolenbroek }
1083*5d5fbe79SDavid van Moolenbroek
1084*5d5fbe79SDavid van Moolenbroek /* Extract cache_flush flag from class field */
1085*5d5fbe79SDavid van Moolenbroek answer->cache_flush = answer->info.klass & 0x8000;
1086*5d5fbe79SDavid van Moolenbroek answer->info.klass &= 0x7FFF;
1087*5d5fbe79SDavid van Moolenbroek
1088*5d5fbe79SDavid van Moolenbroek copied = pbuf_copy_partial(pkt->pbuf, &ttl, sizeof(ttl), pkt->parse_offset);
1089*5d5fbe79SDavid van Moolenbroek if (copied != sizeof(ttl)) {
1090*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1091*5d5fbe79SDavid van Moolenbroek }
1092*5d5fbe79SDavid van Moolenbroek pkt->parse_offset += copied;
1093*5d5fbe79SDavid van Moolenbroek answer->ttl = lwip_ntohl(ttl);
1094*5d5fbe79SDavid van Moolenbroek
1095*5d5fbe79SDavid van Moolenbroek copied = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), pkt->parse_offset);
1096*5d5fbe79SDavid van Moolenbroek if (copied != sizeof(field16)) {
1097*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1098*5d5fbe79SDavid van Moolenbroek }
1099*5d5fbe79SDavid van Moolenbroek pkt->parse_offset += copied;
1100*5d5fbe79SDavid van Moolenbroek answer->rd_length = lwip_ntohs(field16);
1101*5d5fbe79SDavid van Moolenbroek
1102*5d5fbe79SDavid van Moolenbroek answer->rd_offset = pkt->parse_offset;
1103*5d5fbe79SDavid van Moolenbroek pkt->parse_offset += answer->rd_length;
1104*5d5fbe79SDavid van Moolenbroek
1105*5d5fbe79SDavid van Moolenbroek return ERR_OK;
1106*5d5fbe79SDavid van Moolenbroek }
1107*5d5fbe79SDavid van Moolenbroek return ERR_VAL;
1108*5d5fbe79SDavid van Moolenbroek }
1109*5d5fbe79SDavid van Moolenbroek
1110*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1111*5d5fbe79SDavid van Moolenbroek /** Write an IPv4 address (A) RR to outpacket */
1112*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_a_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct netif * netif)1113*5d5fbe79SDavid van Moolenbroek mdns_add_a_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct netif *netif)
1114*5d5fbe79SDavid van Moolenbroek {
1115*5d5fbe79SDavid van Moolenbroek struct mdns_domain host;
1116*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&host, NETIF_TO_HOST(netif));
1117*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with A record\n"));
1118*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &host, DNS_RRTYPE_A, DNS_RRCLASS_IN, cache_flush, (NETIF_TO_HOST(netif))->dns_ttl, (const u8_t *) netif_ip4_addr(netif), sizeof(ip4_addr_t), NULL);
1119*5d5fbe79SDavid van Moolenbroek }
1120*5d5fbe79SDavid van Moolenbroek
1121*5d5fbe79SDavid van Moolenbroek /** Write a 4.3.2.1.in-addr.arpa -> hostname.local PTR RR to outpacket */
1122*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_hostv4_ptr_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct netif * netif)1123*5d5fbe79SDavid van Moolenbroek mdns_add_hostv4_ptr_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct netif *netif)
1124*5d5fbe79SDavid van Moolenbroek {
1125*5d5fbe79SDavid van Moolenbroek struct mdns_domain host, revhost;
1126*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&host, NETIF_TO_HOST(netif));
1127*5d5fbe79SDavid van Moolenbroek mdns_build_reverse_v4_domain(&revhost, netif_ip4_addr(netif));
1128*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with v4 PTR record\n"));
1129*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &revhost, DNS_RRTYPE_PTR, DNS_RRCLASS_IN, cache_flush, (NETIF_TO_HOST(netif))->dns_ttl, NULL, 0, &host);
1130*5d5fbe79SDavid van Moolenbroek }
1131*5d5fbe79SDavid van Moolenbroek #endif
1132*5d5fbe79SDavid van Moolenbroek
1133*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1134*5d5fbe79SDavid van Moolenbroek /** Write an IPv6 address (AAAA) RR to outpacket */
1135*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_aaaa_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct netif * netif,int addrindex)1136*5d5fbe79SDavid van Moolenbroek mdns_add_aaaa_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct netif *netif, int addrindex)
1137*5d5fbe79SDavid van Moolenbroek {
1138*5d5fbe79SDavid van Moolenbroek struct mdns_domain host;
1139*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&host, NETIF_TO_HOST(netif));
1140*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with AAAA record\n"));
1141*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &host, DNS_RRTYPE_AAAA, DNS_RRCLASS_IN, cache_flush, (NETIF_TO_HOST(netif))->dns_ttl, (const u8_t *) netif_ip6_addr(netif, addrindex), sizeof(ip6_addr_t), NULL);
1142*5d5fbe79SDavid van Moolenbroek }
1143*5d5fbe79SDavid van Moolenbroek
1144*5d5fbe79SDavid van Moolenbroek /** Write a x.y.z.ip6.arpa -> hostname.local PTR RR to outpacket */
1145*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_hostv6_ptr_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct netif * netif,int addrindex)1146*5d5fbe79SDavid van Moolenbroek mdns_add_hostv6_ptr_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct netif *netif, int addrindex)
1147*5d5fbe79SDavid van Moolenbroek {
1148*5d5fbe79SDavid van Moolenbroek struct mdns_domain host, revhost;
1149*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&host, NETIF_TO_HOST(netif));
1150*5d5fbe79SDavid van Moolenbroek mdns_build_reverse_v6_domain(&revhost, netif_ip6_addr(netif, addrindex));
1151*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with v6 PTR record\n"));
1152*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &revhost, DNS_RRTYPE_PTR, DNS_RRCLASS_IN, cache_flush, (NETIF_TO_HOST(netif))->dns_ttl, NULL, 0, &host);
1153*5d5fbe79SDavid van Moolenbroek }
1154*5d5fbe79SDavid van Moolenbroek #endif
1155*5d5fbe79SDavid van Moolenbroek
1156*5d5fbe79SDavid van Moolenbroek /** Write an all-services -> servicetype PTR RR to outpacket */
1157*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_servicetype_ptr_answer(struct mdns_outpacket * reply,struct mdns_service * service)1158*5d5fbe79SDavid van Moolenbroek mdns_add_servicetype_ptr_answer(struct mdns_outpacket *reply, struct mdns_service *service)
1159*5d5fbe79SDavid van Moolenbroek {
1160*5d5fbe79SDavid van Moolenbroek struct mdns_domain service_type, service_dnssd;
1161*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(&service_type, service, 0);
1162*5d5fbe79SDavid van Moolenbroek mdns_build_dnssd_domain(&service_dnssd);
1163*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with service type PTR record\n"));
1164*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &service_dnssd, DNS_RRTYPE_PTR, DNS_RRCLASS_IN, 0, service->dns_ttl, NULL, 0, &service_type);
1165*5d5fbe79SDavid van Moolenbroek }
1166*5d5fbe79SDavid van Moolenbroek
1167*5d5fbe79SDavid van Moolenbroek /** Write a servicetype -> servicename PTR RR to outpacket */
1168*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_servicename_ptr_answer(struct mdns_outpacket * reply,struct mdns_service * service)1169*5d5fbe79SDavid van Moolenbroek mdns_add_servicename_ptr_answer(struct mdns_outpacket *reply, struct mdns_service *service)
1170*5d5fbe79SDavid van Moolenbroek {
1171*5d5fbe79SDavid van Moolenbroek struct mdns_domain service_type, service_instance;
1172*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(&service_type, service, 0);
1173*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(&service_instance, service, 1);
1174*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with service name PTR record\n"));
1175*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &service_type, DNS_RRTYPE_PTR, DNS_RRCLASS_IN, 0, service->dns_ttl, NULL, 0, &service_instance);
1176*5d5fbe79SDavid van Moolenbroek }
1177*5d5fbe79SDavid van Moolenbroek
1178*5d5fbe79SDavid van Moolenbroek /** Write a SRV RR to outpacket */
1179*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_srv_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct mdns_host * mdns,struct mdns_service * service)1180*5d5fbe79SDavid van Moolenbroek mdns_add_srv_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct mdns_host *mdns, struct mdns_service *service)
1181*5d5fbe79SDavid van Moolenbroek {
1182*5d5fbe79SDavid van Moolenbroek struct mdns_domain service_instance, srvhost;
1183*5d5fbe79SDavid van Moolenbroek u16_t srvdata[3];
1184*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(&service_instance, service, 1);
1185*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&srvhost, mdns);
1186*5d5fbe79SDavid van Moolenbroek if (reply->legacy_query) {
1187*5d5fbe79SDavid van Moolenbroek /* RFC 6762 section 18.14:
1188*5d5fbe79SDavid van Moolenbroek * In legacy unicast responses generated to answer legacy queries,
1189*5d5fbe79SDavid van Moolenbroek * name compression MUST NOT be performed on SRV records.
1190*5d5fbe79SDavid van Moolenbroek */
1191*5d5fbe79SDavid van Moolenbroek srvhost.skip_compression = 1;
1192*5d5fbe79SDavid van Moolenbroek }
1193*5d5fbe79SDavid van Moolenbroek srvdata[0] = lwip_htons(SRV_PRIORITY);
1194*5d5fbe79SDavid van Moolenbroek srvdata[1] = lwip_htons(SRV_WEIGHT);
1195*5d5fbe79SDavid van Moolenbroek srvdata[2] = lwip_htons(service->port);
1196*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with SRV record\n"));
1197*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &service_instance, DNS_RRTYPE_SRV, DNS_RRCLASS_IN, cache_flush, service->dns_ttl,
1198*5d5fbe79SDavid van Moolenbroek (const u8_t *) &srvdata, sizeof(srvdata), &srvhost);
1199*5d5fbe79SDavid van Moolenbroek }
1200*5d5fbe79SDavid van Moolenbroek
1201*5d5fbe79SDavid van Moolenbroek /** Write a TXT RR to outpacket */
1202*5d5fbe79SDavid van Moolenbroek static err_t
mdns_add_txt_answer(struct mdns_outpacket * reply,u16_t cache_flush,struct mdns_service * service)1203*5d5fbe79SDavid van Moolenbroek mdns_add_txt_answer(struct mdns_outpacket *reply, u16_t cache_flush, struct mdns_service *service)
1204*5d5fbe79SDavid van Moolenbroek {
1205*5d5fbe79SDavid van Moolenbroek struct mdns_domain service_instance;
1206*5d5fbe79SDavid van Moolenbroek mdns_build_service_domain(&service_instance, service, 1);
1207*5d5fbe79SDavid van Moolenbroek mdns_prepare_txtdata(service);
1208*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Responding with TXT record\n"));
1209*5d5fbe79SDavid van Moolenbroek return mdns_add_answer(reply, &service_instance, DNS_RRTYPE_TXT, DNS_RRCLASS_IN, cache_flush, service->dns_ttl,
1210*5d5fbe79SDavid van Moolenbroek (u8_t *) &service->txtdata.name, service->txtdata.length, NULL);
1211*5d5fbe79SDavid van Moolenbroek }
1212*5d5fbe79SDavid van Moolenbroek
1213*5d5fbe79SDavid van Moolenbroek /**
1214*5d5fbe79SDavid van Moolenbroek * Setup outpacket as a reply to the incoming packet
1215*5d5fbe79SDavid van Moolenbroek */
1216*5d5fbe79SDavid van Moolenbroek static void
mdns_init_outpacket(struct mdns_outpacket * out,struct mdns_packet * in)1217*5d5fbe79SDavid van Moolenbroek mdns_init_outpacket(struct mdns_outpacket *out, struct mdns_packet *in)
1218*5d5fbe79SDavid van Moolenbroek {
1219*5d5fbe79SDavid van Moolenbroek memset(out, 0, sizeof(struct mdns_outpacket));
1220*5d5fbe79SDavid van Moolenbroek out->cache_flush = 1;
1221*5d5fbe79SDavid van Moolenbroek out->netif = in->netif;
1222*5d5fbe79SDavid van Moolenbroek
1223*5d5fbe79SDavid van Moolenbroek /* Copy source IP/port to use when responding unicast, or to choose
1224*5d5fbe79SDavid van Moolenbroek * which pcb to use for multicast (IPv4/IPv6)
1225*5d5fbe79SDavid van Moolenbroek */
1226*5d5fbe79SDavid van Moolenbroek SMEMCPY(&out->dest_addr, &in->source_addr, sizeof(ip_addr_t));
1227*5d5fbe79SDavid van Moolenbroek out->dest_port = in->source_port;
1228*5d5fbe79SDavid van Moolenbroek
1229*5d5fbe79SDavid van Moolenbroek if (in->source_port != MDNS_PORT) {
1230*5d5fbe79SDavid van Moolenbroek out->unicast_reply = 1;
1231*5d5fbe79SDavid van Moolenbroek out->cache_flush = 0;
1232*5d5fbe79SDavid van Moolenbroek if (in->questions == 1) {
1233*5d5fbe79SDavid van Moolenbroek out->legacy_query = 1;
1234*5d5fbe79SDavid van Moolenbroek out->tx_id = in->tx_id;
1235*5d5fbe79SDavid van Moolenbroek }
1236*5d5fbe79SDavid van Moolenbroek }
1237*5d5fbe79SDavid van Moolenbroek
1238*5d5fbe79SDavid van Moolenbroek if (in->recv_unicast) {
1239*5d5fbe79SDavid van Moolenbroek out->unicast_reply = 1;
1240*5d5fbe79SDavid van Moolenbroek }
1241*5d5fbe79SDavid van Moolenbroek }
1242*5d5fbe79SDavid van Moolenbroek
1243*5d5fbe79SDavid van Moolenbroek /**
1244*5d5fbe79SDavid van Moolenbroek * Send chosen answers as a reply
1245*5d5fbe79SDavid van Moolenbroek *
1246*5d5fbe79SDavid van Moolenbroek * Add all selected answers (first write will allocate pbuf)
1247*5d5fbe79SDavid van Moolenbroek * Add additional answers based on the selected answers
1248*5d5fbe79SDavid van Moolenbroek * Send the packet
1249*5d5fbe79SDavid van Moolenbroek */
1250*5d5fbe79SDavid van Moolenbroek static void
mdns_send_outpacket(struct mdns_outpacket * outpkt)1251*5d5fbe79SDavid van Moolenbroek mdns_send_outpacket(struct mdns_outpacket *outpkt)
1252*5d5fbe79SDavid van Moolenbroek {
1253*5d5fbe79SDavid van Moolenbroek struct mdns_service *service;
1254*5d5fbe79SDavid van Moolenbroek err_t res;
1255*5d5fbe79SDavid van Moolenbroek int i;
1256*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns = NETIF_TO_HOST(outpkt->netif);
1257*5d5fbe79SDavid van Moolenbroek
1258*5d5fbe79SDavid van Moolenbroek /* Write answers to host questions */
1259*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1260*5d5fbe79SDavid van Moolenbroek if (outpkt->host_replies & REPLY_HOST_A) {
1261*5d5fbe79SDavid van Moolenbroek res = mdns_add_a_answer(outpkt, outpkt->cache_flush, outpkt->netif);
1262*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1263*5d5fbe79SDavid van Moolenbroek goto cleanup;
1264*5d5fbe79SDavid van Moolenbroek }
1265*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1266*5d5fbe79SDavid van Moolenbroek }
1267*5d5fbe79SDavid van Moolenbroek if (outpkt->host_replies & REPLY_HOST_PTR_V4) {
1268*5d5fbe79SDavid van Moolenbroek res = mdns_add_hostv4_ptr_answer(outpkt, outpkt->cache_flush, outpkt->netif);
1269*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1270*5d5fbe79SDavid van Moolenbroek goto cleanup;
1271*5d5fbe79SDavid van Moolenbroek }
1272*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1273*5d5fbe79SDavid van Moolenbroek }
1274*5d5fbe79SDavid van Moolenbroek #endif
1275*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1276*5d5fbe79SDavid van Moolenbroek if (outpkt->host_replies & REPLY_HOST_AAAA) {
1277*5d5fbe79SDavid van Moolenbroek int addrindex;
1278*5d5fbe79SDavid van Moolenbroek for (addrindex = 0; addrindex < LWIP_IPV6_NUM_ADDRESSES; ++addrindex) {
1279*5d5fbe79SDavid van Moolenbroek if (ip6_addr_isvalid(netif_ip6_addr_state(outpkt->netif, addrindex))) {
1280*5d5fbe79SDavid van Moolenbroek res = mdns_add_aaaa_answer(outpkt, outpkt->cache_flush, outpkt->netif, addrindex);
1281*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1282*5d5fbe79SDavid van Moolenbroek goto cleanup;
1283*5d5fbe79SDavid van Moolenbroek }
1284*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1285*5d5fbe79SDavid van Moolenbroek }
1286*5d5fbe79SDavid van Moolenbroek }
1287*5d5fbe79SDavid van Moolenbroek }
1288*5d5fbe79SDavid van Moolenbroek if (outpkt->host_replies & REPLY_HOST_PTR_V6) {
1289*5d5fbe79SDavid van Moolenbroek u8_t rev_addrs = outpkt->host_reverse_v6_replies;
1290*5d5fbe79SDavid van Moolenbroek int addrindex = 0;
1291*5d5fbe79SDavid van Moolenbroek while (rev_addrs) {
1292*5d5fbe79SDavid van Moolenbroek if (rev_addrs & 1) {
1293*5d5fbe79SDavid van Moolenbroek res = mdns_add_hostv6_ptr_answer(outpkt, outpkt->cache_flush, outpkt->netif, addrindex);
1294*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1295*5d5fbe79SDavid van Moolenbroek goto cleanup;
1296*5d5fbe79SDavid van Moolenbroek }
1297*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1298*5d5fbe79SDavid van Moolenbroek }
1299*5d5fbe79SDavid van Moolenbroek addrindex++;
1300*5d5fbe79SDavid van Moolenbroek rev_addrs >>= 1;
1301*5d5fbe79SDavid van Moolenbroek }
1302*5d5fbe79SDavid van Moolenbroek }
1303*5d5fbe79SDavid van Moolenbroek #endif
1304*5d5fbe79SDavid van Moolenbroek
1305*5d5fbe79SDavid van Moolenbroek /* Write answers to service questions */
1306*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; ++i) {
1307*5d5fbe79SDavid van Moolenbroek service = mdns->services[i];
1308*5d5fbe79SDavid van Moolenbroek if (!service) {
1309*5d5fbe79SDavid van Moolenbroek continue;
1310*5d5fbe79SDavid van Moolenbroek }
1311*5d5fbe79SDavid van Moolenbroek
1312*5d5fbe79SDavid van Moolenbroek if (outpkt->serv_replies[i] & REPLY_SERVICE_TYPE_PTR) {
1313*5d5fbe79SDavid van Moolenbroek res = mdns_add_servicetype_ptr_answer(outpkt, service);
1314*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1315*5d5fbe79SDavid van Moolenbroek goto cleanup;
1316*5d5fbe79SDavid van Moolenbroek }
1317*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1318*5d5fbe79SDavid van Moolenbroek }
1319*5d5fbe79SDavid van Moolenbroek
1320*5d5fbe79SDavid van Moolenbroek if (outpkt->serv_replies[i] & REPLY_SERVICE_NAME_PTR) {
1321*5d5fbe79SDavid van Moolenbroek res = mdns_add_servicename_ptr_answer(outpkt, service);
1322*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1323*5d5fbe79SDavid van Moolenbroek goto cleanup;
1324*5d5fbe79SDavid van Moolenbroek }
1325*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1326*5d5fbe79SDavid van Moolenbroek }
1327*5d5fbe79SDavid van Moolenbroek
1328*5d5fbe79SDavid van Moolenbroek if (outpkt->serv_replies[i] & REPLY_SERVICE_SRV) {
1329*5d5fbe79SDavid van Moolenbroek res = mdns_add_srv_answer(outpkt, outpkt->cache_flush, mdns, service);
1330*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1331*5d5fbe79SDavid van Moolenbroek goto cleanup;
1332*5d5fbe79SDavid van Moolenbroek }
1333*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1334*5d5fbe79SDavid van Moolenbroek }
1335*5d5fbe79SDavid van Moolenbroek
1336*5d5fbe79SDavid van Moolenbroek if (outpkt->serv_replies[i] & REPLY_SERVICE_TXT) {
1337*5d5fbe79SDavid van Moolenbroek res = mdns_add_txt_answer(outpkt, outpkt->cache_flush, service);
1338*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1339*5d5fbe79SDavid van Moolenbroek goto cleanup;
1340*5d5fbe79SDavid van Moolenbroek }
1341*5d5fbe79SDavid van Moolenbroek outpkt->answers++;
1342*5d5fbe79SDavid van Moolenbroek }
1343*5d5fbe79SDavid van Moolenbroek }
1344*5d5fbe79SDavid van Moolenbroek
1345*5d5fbe79SDavid van Moolenbroek /* All answers written, add additional RRs */
1346*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; ++i) {
1347*5d5fbe79SDavid van Moolenbroek service = mdns->services[i];
1348*5d5fbe79SDavid van Moolenbroek if (!service) {
1349*5d5fbe79SDavid van Moolenbroek continue;
1350*5d5fbe79SDavid van Moolenbroek }
1351*5d5fbe79SDavid van Moolenbroek
1352*5d5fbe79SDavid van Moolenbroek if (outpkt->serv_replies[i] & REPLY_SERVICE_NAME_PTR) {
1353*5d5fbe79SDavid van Moolenbroek /* Our service instance requested, include SRV & TXT
1354*5d5fbe79SDavid van Moolenbroek * if they are already not requested. */
1355*5d5fbe79SDavid van Moolenbroek if (!(outpkt->serv_replies[i] & REPLY_SERVICE_SRV)) {
1356*5d5fbe79SDavid van Moolenbroek res = mdns_add_srv_answer(outpkt, outpkt->cache_flush, mdns, service);
1357*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1358*5d5fbe79SDavid van Moolenbroek goto cleanup;
1359*5d5fbe79SDavid van Moolenbroek }
1360*5d5fbe79SDavid van Moolenbroek outpkt->additional++;
1361*5d5fbe79SDavid van Moolenbroek }
1362*5d5fbe79SDavid van Moolenbroek
1363*5d5fbe79SDavid van Moolenbroek if (!(outpkt->serv_replies[i] & REPLY_SERVICE_TXT)) {
1364*5d5fbe79SDavid van Moolenbroek res = mdns_add_txt_answer(outpkt, outpkt->cache_flush, service);
1365*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1366*5d5fbe79SDavid van Moolenbroek goto cleanup;
1367*5d5fbe79SDavid van Moolenbroek }
1368*5d5fbe79SDavid van Moolenbroek outpkt->additional++;
1369*5d5fbe79SDavid van Moolenbroek }
1370*5d5fbe79SDavid van Moolenbroek }
1371*5d5fbe79SDavid van Moolenbroek
1372*5d5fbe79SDavid van Moolenbroek /* If service instance, SRV, record or an IP address is requested,
1373*5d5fbe79SDavid van Moolenbroek * supply all addresses for the host
1374*5d5fbe79SDavid van Moolenbroek */
1375*5d5fbe79SDavid van Moolenbroek if ((outpkt->serv_replies[i] & (REPLY_SERVICE_NAME_PTR | REPLY_SERVICE_SRV)) ||
1376*5d5fbe79SDavid van Moolenbroek (outpkt->host_replies & (REPLY_HOST_A | REPLY_HOST_AAAA))) {
1377*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1378*5d5fbe79SDavid van Moolenbroek if (!(outpkt->host_replies & REPLY_HOST_AAAA)) {
1379*5d5fbe79SDavid van Moolenbroek int addrindex;
1380*5d5fbe79SDavid van Moolenbroek for (addrindex = 0; addrindex < LWIP_IPV6_NUM_ADDRESSES; ++addrindex) {
1381*5d5fbe79SDavid van Moolenbroek if (ip6_addr_isvalid(netif_ip6_addr_state(outpkt->netif, addrindex))) {
1382*5d5fbe79SDavid van Moolenbroek res = mdns_add_aaaa_answer(outpkt, outpkt->cache_flush, outpkt->netif, addrindex);
1383*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1384*5d5fbe79SDavid van Moolenbroek goto cleanup;
1385*5d5fbe79SDavid van Moolenbroek }
1386*5d5fbe79SDavid van Moolenbroek outpkt->additional++;
1387*5d5fbe79SDavid van Moolenbroek }
1388*5d5fbe79SDavid van Moolenbroek }
1389*5d5fbe79SDavid van Moolenbroek }
1390*5d5fbe79SDavid van Moolenbroek #endif
1391*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1392*5d5fbe79SDavid van Moolenbroek if (!(outpkt->host_replies & REPLY_HOST_A)) {
1393*5d5fbe79SDavid van Moolenbroek res = mdns_add_a_answer(outpkt, outpkt->cache_flush, outpkt->netif);
1394*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1395*5d5fbe79SDavid van Moolenbroek goto cleanup;
1396*5d5fbe79SDavid van Moolenbroek }
1397*5d5fbe79SDavid van Moolenbroek outpkt->additional++;
1398*5d5fbe79SDavid van Moolenbroek }
1399*5d5fbe79SDavid van Moolenbroek #endif
1400*5d5fbe79SDavid van Moolenbroek }
1401*5d5fbe79SDavid van Moolenbroek }
1402*5d5fbe79SDavid van Moolenbroek
1403*5d5fbe79SDavid van Moolenbroek if (outpkt->pbuf) {
1404*5d5fbe79SDavid van Moolenbroek const ip_addr_t *mcast_destaddr;
1405*5d5fbe79SDavid van Moolenbroek struct dns_hdr hdr;
1406*5d5fbe79SDavid van Moolenbroek
1407*5d5fbe79SDavid van Moolenbroek /* Write header */
1408*5d5fbe79SDavid van Moolenbroek memset(&hdr, 0, sizeof(hdr));
1409*5d5fbe79SDavid van Moolenbroek hdr.flags1 = DNS_FLAG1_RESPONSE | DNS_FLAG1_AUTHORATIVE;
1410*5d5fbe79SDavid van Moolenbroek hdr.numanswers = lwip_htons(outpkt->answers);
1411*5d5fbe79SDavid van Moolenbroek hdr.numextrarr = lwip_htons(outpkt->additional);
1412*5d5fbe79SDavid van Moolenbroek if (outpkt->legacy_query) {
1413*5d5fbe79SDavid van Moolenbroek hdr.numquestions = lwip_htons(1);
1414*5d5fbe79SDavid van Moolenbroek hdr.id = lwip_htons(outpkt->tx_id);
1415*5d5fbe79SDavid van Moolenbroek }
1416*5d5fbe79SDavid van Moolenbroek pbuf_take(outpkt->pbuf, &hdr, sizeof(hdr));
1417*5d5fbe79SDavid van Moolenbroek
1418*5d5fbe79SDavid van Moolenbroek /* Shrink packet */
1419*5d5fbe79SDavid van Moolenbroek pbuf_realloc(outpkt->pbuf, outpkt->write_offset);
1420*5d5fbe79SDavid van Moolenbroek
1421*5d5fbe79SDavid van Moolenbroek if (IP_IS_V6_VAL(outpkt->dest_addr)) {
1422*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1423*5d5fbe79SDavid van Moolenbroek mcast_destaddr = &v6group;
1424*5d5fbe79SDavid van Moolenbroek #endif
1425*5d5fbe79SDavid van Moolenbroek } else {
1426*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1427*5d5fbe79SDavid van Moolenbroek mcast_destaddr = &v4group;
1428*5d5fbe79SDavid van Moolenbroek #endif
1429*5d5fbe79SDavid van Moolenbroek }
1430*5d5fbe79SDavid van Moolenbroek /* Send created packet */
1431*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Sending packet, len=%d, unicast=%d\n", outpkt->write_offset, outpkt->unicast_reply));
1432*5d5fbe79SDavid van Moolenbroek if (outpkt->unicast_reply) {
1433*5d5fbe79SDavid van Moolenbroek udp_sendto_if(mdns_pcb, outpkt->pbuf, &outpkt->dest_addr, outpkt->dest_port, outpkt->netif);
1434*5d5fbe79SDavid van Moolenbroek } else {
1435*5d5fbe79SDavid van Moolenbroek udp_sendto_if(mdns_pcb, outpkt->pbuf, mcast_destaddr, MDNS_PORT, outpkt->netif);
1436*5d5fbe79SDavid van Moolenbroek }
1437*5d5fbe79SDavid van Moolenbroek }
1438*5d5fbe79SDavid van Moolenbroek
1439*5d5fbe79SDavid van Moolenbroek cleanup:
1440*5d5fbe79SDavid van Moolenbroek if (outpkt->pbuf) {
1441*5d5fbe79SDavid van Moolenbroek pbuf_free(outpkt->pbuf);
1442*5d5fbe79SDavid van Moolenbroek outpkt->pbuf = NULL;
1443*5d5fbe79SDavid van Moolenbroek }
1444*5d5fbe79SDavid van Moolenbroek }
1445*5d5fbe79SDavid van Moolenbroek
1446*5d5fbe79SDavid van Moolenbroek /**
1447*5d5fbe79SDavid van Moolenbroek * Send unsolicited answer containing all our known data
1448*5d5fbe79SDavid van Moolenbroek * @param netif The network interface to send on
1449*5d5fbe79SDavid van Moolenbroek * @param destination The target address to send to (usually multicast address)
1450*5d5fbe79SDavid van Moolenbroek */
1451*5d5fbe79SDavid van Moolenbroek static void
mdns_announce(struct netif * netif,const ip_addr_t * destination)1452*5d5fbe79SDavid van Moolenbroek mdns_announce(struct netif *netif, const ip_addr_t *destination)
1453*5d5fbe79SDavid van Moolenbroek {
1454*5d5fbe79SDavid van Moolenbroek struct mdns_outpacket announce;
1455*5d5fbe79SDavid van Moolenbroek int i;
1456*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns = NETIF_TO_HOST(netif);
1457*5d5fbe79SDavid van Moolenbroek
1458*5d5fbe79SDavid van Moolenbroek memset(&announce, 0, sizeof(announce));
1459*5d5fbe79SDavid van Moolenbroek announce.netif = netif;
1460*5d5fbe79SDavid van Moolenbroek announce.cache_flush = 1;
1461*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1462*5d5fbe79SDavid van Moolenbroek if (!ip4_addr_isany_val(*netif_ip4_addr(netif)))
1463*5d5fbe79SDavid van Moolenbroek announce.host_replies = REPLY_HOST_A | REPLY_HOST_PTR_V4;
1464*5d5fbe79SDavid van Moolenbroek #endif
1465*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1466*5d5fbe79SDavid van Moolenbroek for (i = 0; i < LWIP_IPV6_NUM_ADDRESSES; ++i) {
1467*5d5fbe79SDavid van Moolenbroek if (ip6_addr_isvalid(netif_ip6_addr_state(netif, i))) {
1468*5d5fbe79SDavid van Moolenbroek announce.host_replies |= REPLY_HOST_AAAA | REPLY_HOST_PTR_V6;
1469*5d5fbe79SDavid van Moolenbroek announce.host_reverse_v6_replies |= (1 << i);
1470*5d5fbe79SDavid van Moolenbroek }
1471*5d5fbe79SDavid van Moolenbroek }
1472*5d5fbe79SDavid van Moolenbroek #endif
1473*5d5fbe79SDavid van Moolenbroek
1474*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; i++) {
1475*5d5fbe79SDavid van Moolenbroek struct mdns_service *serv = mdns->services[i];
1476*5d5fbe79SDavid van Moolenbroek if (serv) {
1477*5d5fbe79SDavid van Moolenbroek announce.serv_replies[i] = REPLY_SERVICE_TYPE_PTR | REPLY_SERVICE_NAME_PTR |
1478*5d5fbe79SDavid van Moolenbroek REPLY_SERVICE_SRV | REPLY_SERVICE_TXT;
1479*5d5fbe79SDavid van Moolenbroek }
1480*5d5fbe79SDavid van Moolenbroek }
1481*5d5fbe79SDavid van Moolenbroek
1482*5d5fbe79SDavid van Moolenbroek announce.dest_port = MDNS_PORT;
1483*5d5fbe79SDavid van Moolenbroek SMEMCPY(&announce.dest_addr, destination, sizeof(announce.dest_addr));
1484*5d5fbe79SDavid van Moolenbroek mdns_send_outpacket(&announce);
1485*5d5fbe79SDavid van Moolenbroek }
1486*5d5fbe79SDavid van Moolenbroek
1487*5d5fbe79SDavid van Moolenbroek /**
1488*5d5fbe79SDavid van Moolenbroek * Handle question MDNS packet
1489*5d5fbe79SDavid van Moolenbroek * 1. Parse all questions and set bits what answers to send
1490*5d5fbe79SDavid van Moolenbroek * 2. Clear pending answers if known answers are supplied
1491*5d5fbe79SDavid van Moolenbroek * 3. Put chosen answers in new packet and send as reply
1492*5d5fbe79SDavid van Moolenbroek */
1493*5d5fbe79SDavid van Moolenbroek static void
mdns_handle_question(struct mdns_packet * pkt)1494*5d5fbe79SDavid van Moolenbroek mdns_handle_question(struct mdns_packet *pkt)
1495*5d5fbe79SDavid van Moolenbroek {
1496*5d5fbe79SDavid van Moolenbroek struct mdns_service *service;
1497*5d5fbe79SDavid van Moolenbroek struct mdns_outpacket reply;
1498*5d5fbe79SDavid van Moolenbroek int replies = 0;
1499*5d5fbe79SDavid van Moolenbroek int i;
1500*5d5fbe79SDavid van Moolenbroek err_t res;
1501*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns = NETIF_TO_HOST(pkt->netif);
1502*5d5fbe79SDavid van Moolenbroek
1503*5d5fbe79SDavid van Moolenbroek mdns_init_outpacket(&reply, pkt);
1504*5d5fbe79SDavid van Moolenbroek
1505*5d5fbe79SDavid van Moolenbroek while (pkt->questions_left) {
1506*5d5fbe79SDavid van Moolenbroek struct mdns_question q;
1507*5d5fbe79SDavid van Moolenbroek
1508*5d5fbe79SDavid van Moolenbroek res = mdns_read_question(pkt, &q);
1509*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1510*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Failed to parse question, skipping query packet\n"));
1511*5d5fbe79SDavid van Moolenbroek return;
1512*5d5fbe79SDavid van Moolenbroek }
1513*5d5fbe79SDavid van Moolenbroek
1514*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Query for domain "));
1515*5d5fbe79SDavid van Moolenbroek mdns_domain_debug_print(&q.info.domain);
1516*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, (" type %d class %d\n", q.info.type, q.info.klass));
1517*5d5fbe79SDavid van Moolenbroek
1518*5d5fbe79SDavid van Moolenbroek if (q.unicast) {
1519*5d5fbe79SDavid van Moolenbroek /* Reply unicast if any question is unicast */
1520*5d5fbe79SDavid van Moolenbroek reply.unicast_reply = 1;
1521*5d5fbe79SDavid van Moolenbroek }
1522*5d5fbe79SDavid van Moolenbroek
1523*5d5fbe79SDavid van Moolenbroek reply.host_replies |= check_host(pkt->netif, &q.info, &reply.host_reverse_v6_replies);
1524*5d5fbe79SDavid van Moolenbroek replies |= reply.host_replies;
1525*5d5fbe79SDavid van Moolenbroek
1526*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; ++i) {
1527*5d5fbe79SDavid van Moolenbroek service = mdns->services[i];
1528*5d5fbe79SDavid van Moolenbroek if (!service) {
1529*5d5fbe79SDavid van Moolenbroek continue;
1530*5d5fbe79SDavid van Moolenbroek }
1531*5d5fbe79SDavid van Moolenbroek reply.serv_replies[i] |= check_service(service, &q.info);
1532*5d5fbe79SDavid van Moolenbroek replies |= reply.serv_replies[i];
1533*5d5fbe79SDavid van Moolenbroek }
1534*5d5fbe79SDavid van Moolenbroek
1535*5d5fbe79SDavid van Moolenbroek if (replies && reply.legacy_query) {
1536*5d5fbe79SDavid van Moolenbroek /* Add question to reply packet (legacy packet only has 1 question) */
1537*5d5fbe79SDavid van Moolenbroek res = mdns_add_question(&reply, &q.info.domain, q.info.type, q.info.klass, 0);
1538*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1539*5d5fbe79SDavid van Moolenbroek goto cleanup;
1540*5d5fbe79SDavid van Moolenbroek }
1541*5d5fbe79SDavid van Moolenbroek }
1542*5d5fbe79SDavid van Moolenbroek }
1543*5d5fbe79SDavid van Moolenbroek
1544*5d5fbe79SDavid van Moolenbroek /* Handle known answers */
1545*5d5fbe79SDavid van Moolenbroek while (pkt->answers_left) {
1546*5d5fbe79SDavid van Moolenbroek struct mdns_answer ans;
1547*5d5fbe79SDavid van Moolenbroek u8_t rev_v6;
1548*5d5fbe79SDavid van Moolenbroek int match;
1549*5d5fbe79SDavid van Moolenbroek
1550*5d5fbe79SDavid van Moolenbroek res = mdns_read_answer(pkt, &ans);
1551*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1552*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Failed to parse answer, skipping query packet\n"));
1553*5d5fbe79SDavid van Moolenbroek goto cleanup;
1554*5d5fbe79SDavid van Moolenbroek }
1555*5d5fbe79SDavid van Moolenbroek
1556*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Known answer for domain "));
1557*5d5fbe79SDavid van Moolenbroek mdns_domain_debug_print(&ans.info.domain);
1558*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, (" type %d class %d\n", ans.info.type, ans.info.klass));
1559*5d5fbe79SDavid van Moolenbroek
1560*5d5fbe79SDavid van Moolenbroek
1561*5d5fbe79SDavid van Moolenbroek if (ans.info.type == DNS_RRTYPE_ANY || ans.info.klass == DNS_RRCLASS_ANY) {
1562*5d5fbe79SDavid van Moolenbroek /* Skip known answers for ANY type & class */
1563*5d5fbe79SDavid van Moolenbroek continue;
1564*5d5fbe79SDavid van Moolenbroek }
1565*5d5fbe79SDavid van Moolenbroek
1566*5d5fbe79SDavid van Moolenbroek rev_v6 = 0;
1567*5d5fbe79SDavid van Moolenbroek match = reply.host_replies & check_host(pkt->netif, &ans.info, &rev_v6);
1568*5d5fbe79SDavid van Moolenbroek if (match && (ans.ttl > (mdns->dns_ttl / 2))) {
1569*5d5fbe79SDavid van Moolenbroek /* The RR in the known answer matches an RR we are planning to send,
1570*5d5fbe79SDavid van Moolenbroek * and the TTL is less than half gone.
1571*5d5fbe79SDavid van Moolenbroek * If the payload matches we should not send that answer.
1572*5d5fbe79SDavid van Moolenbroek */
1573*5d5fbe79SDavid van Moolenbroek if (ans.info.type == DNS_RRTYPE_PTR) {
1574*5d5fbe79SDavid van Moolenbroek /* Read domain and compare */
1575*5d5fbe79SDavid van Moolenbroek struct mdns_domain known_ans, my_ans;
1576*5d5fbe79SDavid van Moolenbroek u16_t len;
1577*5d5fbe79SDavid van Moolenbroek len = mdns_readname(pkt->pbuf, ans.rd_offset, &known_ans);
1578*5d5fbe79SDavid van Moolenbroek res = mdns_build_host_domain(&my_ans, mdns);
1579*5d5fbe79SDavid van Moolenbroek if (len != MDNS_READNAME_ERROR && res == ERR_OK && mdns_domain_eq(&known_ans, &my_ans)) {
1580*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1581*5d5fbe79SDavid van Moolenbroek if (match & REPLY_HOST_PTR_V4) {
1582*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: v4 PTR\n"));
1583*5d5fbe79SDavid van Moolenbroek reply.host_replies &= ~REPLY_HOST_PTR_V4;
1584*5d5fbe79SDavid van Moolenbroek }
1585*5d5fbe79SDavid van Moolenbroek #endif
1586*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1587*5d5fbe79SDavid van Moolenbroek if (match & REPLY_HOST_PTR_V6) {
1588*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: v6 PTR\n"));
1589*5d5fbe79SDavid van Moolenbroek reply.host_reverse_v6_replies &= ~rev_v6;
1590*5d5fbe79SDavid van Moolenbroek if (reply.host_reverse_v6_replies == 0) {
1591*5d5fbe79SDavid van Moolenbroek reply.host_replies &= ~REPLY_HOST_PTR_V6;
1592*5d5fbe79SDavid van Moolenbroek }
1593*5d5fbe79SDavid van Moolenbroek }
1594*5d5fbe79SDavid van Moolenbroek #endif
1595*5d5fbe79SDavid van Moolenbroek }
1596*5d5fbe79SDavid van Moolenbroek } else if (match & REPLY_HOST_A) {
1597*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1598*5d5fbe79SDavid van Moolenbroek if (ans.rd_length == sizeof(ip4_addr_t) &&
1599*5d5fbe79SDavid van Moolenbroek pbuf_memcmp(pkt->pbuf, ans.rd_offset, netif_ip4_addr(pkt->netif), ans.rd_length) == 0) {
1600*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: A\n"));
1601*5d5fbe79SDavid van Moolenbroek reply.host_replies &= ~REPLY_HOST_A;
1602*5d5fbe79SDavid van Moolenbroek }
1603*5d5fbe79SDavid van Moolenbroek #endif
1604*5d5fbe79SDavid van Moolenbroek } else if (match & REPLY_HOST_AAAA) {
1605*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1606*5d5fbe79SDavid van Moolenbroek if (ans.rd_length == sizeof(ip6_addr_t) &&
1607*5d5fbe79SDavid van Moolenbroek /* TODO this clears all AAAA responses if first addr is set as known */
1608*5d5fbe79SDavid van Moolenbroek pbuf_memcmp(pkt->pbuf, ans.rd_offset, netif_ip6_addr(pkt->netif, 0), ans.rd_length) == 0) {
1609*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: AAAA\n"));
1610*5d5fbe79SDavid van Moolenbroek reply.host_replies &= ~REPLY_HOST_AAAA;
1611*5d5fbe79SDavid van Moolenbroek }
1612*5d5fbe79SDavid van Moolenbroek #endif
1613*5d5fbe79SDavid van Moolenbroek }
1614*5d5fbe79SDavid van Moolenbroek }
1615*5d5fbe79SDavid van Moolenbroek
1616*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; ++i) {
1617*5d5fbe79SDavid van Moolenbroek service = mdns->services[i];
1618*5d5fbe79SDavid van Moolenbroek if (!service) {
1619*5d5fbe79SDavid van Moolenbroek continue;
1620*5d5fbe79SDavid van Moolenbroek }
1621*5d5fbe79SDavid van Moolenbroek match = reply.serv_replies[i] & check_service(service, &ans.info);
1622*5d5fbe79SDavid van Moolenbroek if (match && (ans.ttl > (service->dns_ttl / 2))) {
1623*5d5fbe79SDavid van Moolenbroek /* The RR in the known answer matches an RR we are planning to send,
1624*5d5fbe79SDavid van Moolenbroek * and the TTL is less than half gone.
1625*5d5fbe79SDavid van Moolenbroek * If the payload matches we should not send that answer.
1626*5d5fbe79SDavid van Moolenbroek */
1627*5d5fbe79SDavid van Moolenbroek if (ans.info.type == DNS_RRTYPE_PTR) {
1628*5d5fbe79SDavid van Moolenbroek /* Read domain and compare */
1629*5d5fbe79SDavid van Moolenbroek struct mdns_domain known_ans, my_ans;
1630*5d5fbe79SDavid van Moolenbroek u16_t len;
1631*5d5fbe79SDavid van Moolenbroek len = mdns_readname(pkt->pbuf, ans.rd_offset, &known_ans);
1632*5d5fbe79SDavid van Moolenbroek if (len != MDNS_READNAME_ERROR) {
1633*5d5fbe79SDavid van Moolenbroek if (match & REPLY_SERVICE_TYPE_PTR) {
1634*5d5fbe79SDavid van Moolenbroek res = mdns_build_service_domain(&my_ans, service, 0);
1635*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&known_ans, &my_ans)) {
1636*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: service type PTR\n"));
1637*5d5fbe79SDavid van Moolenbroek reply.serv_replies[i] &= ~REPLY_SERVICE_TYPE_PTR;
1638*5d5fbe79SDavid van Moolenbroek }
1639*5d5fbe79SDavid van Moolenbroek }
1640*5d5fbe79SDavid van Moolenbroek if (match & REPLY_SERVICE_NAME_PTR) {
1641*5d5fbe79SDavid van Moolenbroek res = mdns_build_service_domain(&my_ans, service, 1);
1642*5d5fbe79SDavid van Moolenbroek if (res == ERR_OK && mdns_domain_eq(&known_ans, &my_ans)) {
1643*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: service name PTR\n"));
1644*5d5fbe79SDavid van Moolenbroek reply.serv_replies[i] &= ~REPLY_SERVICE_NAME_PTR;
1645*5d5fbe79SDavid van Moolenbroek }
1646*5d5fbe79SDavid van Moolenbroek }
1647*5d5fbe79SDavid van Moolenbroek }
1648*5d5fbe79SDavid van Moolenbroek } else if (match & REPLY_SERVICE_SRV) {
1649*5d5fbe79SDavid van Moolenbroek /* Read and compare to my SRV record */
1650*5d5fbe79SDavid van Moolenbroek u16_t field16, len, read_pos;
1651*5d5fbe79SDavid van Moolenbroek struct mdns_domain known_ans, my_ans;
1652*5d5fbe79SDavid van Moolenbroek read_pos = ans.rd_offset;
1653*5d5fbe79SDavid van Moolenbroek do {
1654*5d5fbe79SDavid van Moolenbroek /* Check priority field */
1655*5d5fbe79SDavid van Moolenbroek len = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), read_pos);
1656*5d5fbe79SDavid van Moolenbroek if (len != sizeof(field16) || lwip_ntohs(field16) != SRV_PRIORITY) {
1657*5d5fbe79SDavid van Moolenbroek break;
1658*5d5fbe79SDavid van Moolenbroek }
1659*5d5fbe79SDavid van Moolenbroek read_pos += len;
1660*5d5fbe79SDavid van Moolenbroek /* Check weight field */
1661*5d5fbe79SDavid van Moolenbroek len = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), read_pos);
1662*5d5fbe79SDavid van Moolenbroek if (len != sizeof(field16) || lwip_ntohs(field16) != SRV_WEIGHT) {
1663*5d5fbe79SDavid van Moolenbroek break;
1664*5d5fbe79SDavid van Moolenbroek }
1665*5d5fbe79SDavid van Moolenbroek read_pos += len;
1666*5d5fbe79SDavid van Moolenbroek /* Check port field */
1667*5d5fbe79SDavid van Moolenbroek len = pbuf_copy_partial(pkt->pbuf, &field16, sizeof(field16), read_pos);
1668*5d5fbe79SDavid van Moolenbroek if (len != sizeof(field16) || lwip_ntohs(field16) != service->port) {
1669*5d5fbe79SDavid van Moolenbroek break;
1670*5d5fbe79SDavid van Moolenbroek }
1671*5d5fbe79SDavid van Moolenbroek read_pos += len;
1672*5d5fbe79SDavid van Moolenbroek /* Check host field */
1673*5d5fbe79SDavid van Moolenbroek len = mdns_readname(pkt->pbuf, read_pos, &known_ans);
1674*5d5fbe79SDavid van Moolenbroek mdns_build_host_domain(&my_ans, mdns);
1675*5d5fbe79SDavid van Moolenbroek if (len == MDNS_READNAME_ERROR || !mdns_domain_eq(&known_ans, &my_ans)) {
1676*5d5fbe79SDavid van Moolenbroek break;
1677*5d5fbe79SDavid van Moolenbroek }
1678*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: SRV\n"));
1679*5d5fbe79SDavid van Moolenbroek reply.serv_replies[i] &= ~REPLY_SERVICE_SRV;
1680*5d5fbe79SDavid van Moolenbroek } while (0);
1681*5d5fbe79SDavid van Moolenbroek } else if (match & REPLY_SERVICE_TXT) {
1682*5d5fbe79SDavid van Moolenbroek mdns_prepare_txtdata(service);
1683*5d5fbe79SDavid van Moolenbroek if (service->txtdata.length == ans.rd_length &&
1684*5d5fbe79SDavid van Moolenbroek pbuf_memcmp(pkt->pbuf, ans.rd_offset, service->txtdata.name, ans.rd_length) == 0) {
1685*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Skipping known answer: TXT\n"));
1686*5d5fbe79SDavid van Moolenbroek reply.serv_replies[i] &= ~REPLY_SERVICE_TXT;
1687*5d5fbe79SDavid van Moolenbroek }
1688*5d5fbe79SDavid van Moolenbroek }
1689*5d5fbe79SDavid van Moolenbroek }
1690*5d5fbe79SDavid van Moolenbroek }
1691*5d5fbe79SDavid van Moolenbroek }
1692*5d5fbe79SDavid van Moolenbroek
1693*5d5fbe79SDavid van Moolenbroek mdns_send_outpacket(&reply);
1694*5d5fbe79SDavid van Moolenbroek
1695*5d5fbe79SDavid van Moolenbroek cleanup:
1696*5d5fbe79SDavid van Moolenbroek if (reply.pbuf) {
1697*5d5fbe79SDavid van Moolenbroek /* This should only happen if we fail to alloc/write question for legacy query */
1698*5d5fbe79SDavid van Moolenbroek pbuf_free(reply.pbuf);
1699*5d5fbe79SDavid van Moolenbroek reply.pbuf = NULL;
1700*5d5fbe79SDavid van Moolenbroek }
1701*5d5fbe79SDavid van Moolenbroek }
1702*5d5fbe79SDavid van Moolenbroek
1703*5d5fbe79SDavid van Moolenbroek /**
1704*5d5fbe79SDavid van Moolenbroek * Handle response MDNS packet
1705*5d5fbe79SDavid van Moolenbroek * Only prints debug for now. Will need more code to do conflict resolution.
1706*5d5fbe79SDavid van Moolenbroek */
1707*5d5fbe79SDavid van Moolenbroek static void
mdns_handle_response(struct mdns_packet * pkt)1708*5d5fbe79SDavid van Moolenbroek mdns_handle_response(struct mdns_packet *pkt)
1709*5d5fbe79SDavid van Moolenbroek {
1710*5d5fbe79SDavid van Moolenbroek /* Ignore all questions */
1711*5d5fbe79SDavid van Moolenbroek while (pkt->questions_left) {
1712*5d5fbe79SDavid van Moolenbroek struct mdns_question q;
1713*5d5fbe79SDavid van Moolenbroek err_t res;
1714*5d5fbe79SDavid van Moolenbroek
1715*5d5fbe79SDavid van Moolenbroek res = mdns_read_question(pkt, &q);
1716*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1717*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Failed to parse question, skipping response packet\n"));
1718*5d5fbe79SDavid van Moolenbroek return;
1719*5d5fbe79SDavid van Moolenbroek }
1720*5d5fbe79SDavid van Moolenbroek }
1721*5d5fbe79SDavid van Moolenbroek
1722*5d5fbe79SDavid van Moolenbroek while (pkt->answers_left) {
1723*5d5fbe79SDavid van Moolenbroek struct mdns_answer ans;
1724*5d5fbe79SDavid van Moolenbroek err_t res;
1725*5d5fbe79SDavid van Moolenbroek
1726*5d5fbe79SDavid van Moolenbroek res = mdns_read_answer(pkt, &ans);
1727*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1728*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Failed to parse answer, skipping response packet\n"));
1729*5d5fbe79SDavid van Moolenbroek return;
1730*5d5fbe79SDavid van Moolenbroek }
1731*5d5fbe79SDavid van Moolenbroek
1732*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Answer for domain "));
1733*5d5fbe79SDavid van Moolenbroek mdns_domain_debug_print(&ans.info.domain);
1734*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, (" type %d class %d\n", ans.info.type, ans.info.klass));
1735*5d5fbe79SDavid van Moolenbroek }
1736*5d5fbe79SDavid van Moolenbroek }
1737*5d5fbe79SDavid van Moolenbroek
1738*5d5fbe79SDavid van Moolenbroek /**
1739*5d5fbe79SDavid van Moolenbroek * Receive input function for MDNS packets.
1740*5d5fbe79SDavid van Moolenbroek * Handles both IPv4 and IPv6 UDP pcbs.
1741*5d5fbe79SDavid van Moolenbroek */
1742*5d5fbe79SDavid van Moolenbroek static void
mdns_recv(void * arg,struct udp_pcb * pcb,struct pbuf * p,const ip_addr_t * addr,u16_t port)1743*5d5fbe79SDavid van Moolenbroek mdns_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p, const ip_addr_t *addr, u16_t port)
1744*5d5fbe79SDavid van Moolenbroek {
1745*5d5fbe79SDavid van Moolenbroek struct dns_hdr hdr;
1746*5d5fbe79SDavid van Moolenbroek struct mdns_packet packet;
1747*5d5fbe79SDavid van Moolenbroek struct netif *recv_netif = ip_current_input_netif();
1748*5d5fbe79SDavid van Moolenbroek u16_t offset = 0;
1749*5d5fbe79SDavid van Moolenbroek
1750*5d5fbe79SDavid van Moolenbroek LWIP_UNUSED_ARG(arg);
1751*5d5fbe79SDavid van Moolenbroek LWIP_UNUSED_ARG(pcb);
1752*5d5fbe79SDavid van Moolenbroek
1753*5d5fbe79SDavid van Moolenbroek LWIP_DEBUGF(MDNS_DEBUG, ("MDNS: Received IPv%d MDNS packet, len %d\n", IP_IS_V6(addr)? 6 : 4, p->tot_len));
1754*5d5fbe79SDavid van Moolenbroek
1755*5d5fbe79SDavid van Moolenbroek if (NETIF_TO_HOST(recv_netif) == NULL) {
1756*5d5fbe79SDavid van Moolenbroek /* From netif not configured for MDNS */
1757*5d5fbe79SDavid van Moolenbroek goto dealloc;
1758*5d5fbe79SDavid van Moolenbroek }
1759*5d5fbe79SDavid van Moolenbroek
1760*5d5fbe79SDavid van Moolenbroek if (pbuf_copy_partial(p, &hdr, SIZEOF_DNS_HDR, offset) < SIZEOF_DNS_HDR) {
1761*5d5fbe79SDavid van Moolenbroek /* Too small */
1762*5d5fbe79SDavid van Moolenbroek goto dealloc;
1763*5d5fbe79SDavid van Moolenbroek }
1764*5d5fbe79SDavid van Moolenbroek offset += SIZEOF_DNS_HDR;
1765*5d5fbe79SDavid van Moolenbroek
1766*5d5fbe79SDavid van Moolenbroek if (DNS_HDR_GET_OPCODE(&hdr)) {
1767*5d5fbe79SDavid van Moolenbroek /* Ignore non-standard queries in multicast packets (RFC 6762, section 18.3) */
1768*5d5fbe79SDavid van Moolenbroek goto dealloc;
1769*5d5fbe79SDavid van Moolenbroek }
1770*5d5fbe79SDavid van Moolenbroek
1771*5d5fbe79SDavid van Moolenbroek memset(&packet, 0, sizeof(packet));
1772*5d5fbe79SDavid van Moolenbroek SMEMCPY(&packet.source_addr, addr, sizeof(packet.source_addr));
1773*5d5fbe79SDavid van Moolenbroek packet.source_port = port;
1774*5d5fbe79SDavid van Moolenbroek packet.netif = recv_netif;
1775*5d5fbe79SDavid van Moolenbroek packet.pbuf = p;
1776*5d5fbe79SDavid van Moolenbroek packet.parse_offset = offset;
1777*5d5fbe79SDavid van Moolenbroek packet.tx_id = lwip_ntohs(hdr.id);
1778*5d5fbe79SDavid van Moolenbroek packet.questions = packet.questions_left = lwip_ntohs(hdr.numquestions);
1779*5d5fbe79SDavid van Moolenbroek packet.answers = packet.answers_left = lwip_ntohs(hdr.numanswers) + lwip_ntohs(hdr.numauthrr) + lwip_ntohs(hdr.numextrarr);
1780*5d5fbe79SDavid van Moolenbroek
1781*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1782*5d5fbe79SDavid van Moolenbroek if (IP_IS_V6(ip_current_dest_addr())) {
1783*5d5fbe79SDavid van Moolenbroek if (!ip_addr_cmp(ip_current_dest_addr(), &v6group)) {
1784*5d5fbe79SDavid van Moolenbroek packet.recv_unicast = 1;
1785*5d5fbe79SDavid van Moolenbroek }
1786*5d5fbe79SDavid van Moolenbroek }
1787*5d5fbe79SDavid van Moolenbroek #endif
1788*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1789*5d5fbe79SDavid van Moolenbroek if (!IP_IS_V6(ip_current_dest_addr())) {
1790*5d5fbe79SDavid van Moolenbroek if (!ip_addr_cmp(ip_current_dest_addr(), &v4group)) {
1791*5d5fbe79SDavid van Moolenbroek packet.recv_unicast = 1;
1792*5d5fbe79SDavid van Moolenbroek }
1793*5d5fbe79SDavid van Moolenbroek }
1794*5d5fbe79SDavid van Moolenbroek #endif
1795*5d5fbe79SDavid van Moolenbroek
1796*5d5fbe79SDavid van Moolenbroek if (hdr.flags1 & DNS_FLAG1_RESPONSE) {
1797*5d5fbe79SDavid van Moolenbroek mdns_handle_response(&packet);
1798*5d5fbe79SDavid van Moolenbroek } else {
1799*5d5fbe79SDavid van Moolenbroek mdns_handle_question(&packet);
1800*5d5fbe79SDavid van Moolenbroek }
1801*5d5fbe79SDavid van Moolenbroek
1802*5d5fbe79SDavid van Moolenbroek dealloc:
1803*5d5fbe79SDavid van Moolenbroek pbuf_free(p);
1804*5d5fbe79SDavid van Moolenbroek }
1805*5d5fbe79SDavid van Moolenbroek
1806*5d5fbe79SDavid van Moolenbroek /**
1807*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
1808*5d5fbe79SDavid van Moolenbroek * Announce IP settings have changed on netif.
1809*5d5fbe79SDavid van Moolenbroek * Call this in your callback registered by netif_set_status_callback().
1810*5d5fbe79SDavid van Moolenbroek * No need to call this function when LWIP_NETIF_EXT_STATUS_CALLBACK==1,
1811*5d5fbe79SDavid van Moolenbroek * this handled automatically for you.
1812*5d5fbe79SDavid van Moolenbroek * @param netif The network interface where settings have changed.
1813*5d5fbe79SDavid van Moolenbroek */
1814*5d5fbe79SDavid van Moolenbroek void
mdns_resp_netif_settings_changed(struct netif * netif)1815*5d5fbe79SDavid van Moolenbroek mdns_resp_netif_settings_changed(struct netif *netif)
1816*5d5fbe79SDavid van Moolenbroek {
1817*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_netif_ip_changed: netif != NULL", (netif != NULL), return);
1818*5d5fbe79SDavid van Moolenbroek
1819*5d5fbe79SDavid van Moolenbroek if (NETIF_TO_HOST(netif) == NULL) {
1820*5d5fbe79SDavid van Moolenbroek return;
1821*5d5fbe79SDavid van Moolenbroek }
1822*5d5fbe79SDavid van Moolenbroek
1823*5d5fbe79SDavid van Moolenbroek /* Announce on IPv6 and IPv4 */
1824*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1825*5d5fbe79SDavid van Moolenbroek mdns_announce(netif, IP6_ADDR_ANY);
1826*5d5fbe79SDavid van Moolenbroek #endif
1827*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1828*5d5fbe79SDavid van Moolenbroek mdns_announce(netif, IP4_ADDR_ANY);
1829*5d5fbe79SDavid van Moolenbroek #endif
1830*5d5fbe79SDavid van Moolenbroek }
1831*5d5fbe79SDavid van Moolenbroek
1832*5d5fbe79SDavid van Moolenbroek #if LWIP_NETIF_EXT_STATUS_CALLBACK
1833*5d5fbe79SDavid van Moolenbroek static void
mdns_netif_ext_status_callback(struct netif * netif,netif_nsc_reason_t reason,const netif_ext_callback_args_t * args)1834*5d5fbe79SDavid van Moolenbroek mdns_netif_ext_status_callback(struct netif* netif, netif_nsc_reason_t reason, const netif_ext_callback_args_t* args)
1835*5d5fbe79SDavid van Moolenbroek {
1836*5d5fbe79SDavid van Moolenbroek LWIP_UNUSED_ARG(args);
1837*5d5fbe79SDavid van Moolenbroek
1838*5d5fbe79SDavid van Moolenbroek /* MDNS enabled on netif? */
1839*5d5fbe79SDavid van Moolenbroek if (NETIF_TO_HOST(netif) == NULL) {
1840*5d5fbe79SDavid van Moolenbroek return;
1841*5d5fbe79SDavid van Moolenbroek }
1842*5d5fbe79SDavid van Moolenbroek
1843*5d5fbe79SDavid van Moolenbroek switch (reason)
1844*5d5fbe79SDavid van Moolenbroek {
1845*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_STATUS_CHANGED:
1846*5d5fbe79SDavid van Moolenbroek if (args->status_changed.state != 0) {
1847*5d5fbe79SDavid van Moolenbroek mdns_resp_netif_settings_changed(netif);
1848*5d5fbe79SDavid van Moolenbroek }
1849*5d5fbe79SDavid van Moolenbroek /* TODO: send goodbye message */
1850*5d5fbe79SDavid van Moolenbroek break;
1851*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_LINK_CHANGED:
1852*5d5fbe79SDavid van Moolenbroek if (args->link_changed.state != 0) {
1853*5d5fbe79SDavid van Moolenbroek mdns_resp_netif_settings_changed(netif);
1854*5d5fbe79SDavid van Moolenbroek }
1855*5d5fbe79SDavid van Moolenbroek break;
1856*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV4_ADDRESS_CHANGED: /* fall through */
1857*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV4_GATEWAY_CHANGED: /* fall through */
1858*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV4_NETMASK_CHANGED: /* fall through */
1859*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV4_SETTINGS_CHANGED: /* fall through */
1860*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV6_SET: /* fall through */
1861*5d5fbe79SDavid van Moolenbroek case LWIP_NSC_IPV6_ADDR_STATE_CHANGED:
1862*5d5fbe79SDavid van Moolenbroek mdns_resp_netif_settings_changed(netif);
1863*5d5fbe79SDavid van Moolenbroek break;
1864*5d5fbe79SDavid van Moolenbroek default:
1865*5d5fbe79SDavid van Moolenbroek break;
1866*5d5fbe79SDavid van Moolenbroek }
1867*5d5fbe79SDavid van Moolenbroek }
1868*5d5fbe79SDavid van Moolenbroek #endif
1869*5d5fbe79SDavid van Moolenbroek
1870*5d5fbe79SDavid van Moolenbroek /**
1871*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
1872*5d5fbe79SDavid van Moolenbroek * Activate MDNS responder for a network interface and send announce packets.
1873*5d5fbe79SDavid van Moolenbroek * @param netif The network interface to activate.
1874*5d5fbe79SDavid van Moolenbroek * @param hostname Name to use. Queries for <hostname>.local will be answered
1875*5d5fbe79SDavid van Moolenbroek * with the IP addresses of the netif. The hostname will be copied, the
1876*5d5fbe79SDavid van Moolenbroek * given pointer can be on the stack.
1877*5d5fbe79SDavid van Moolenbroek * @param dns_ttl Validity time in seconds to send out for IP address data in DNS replies
1878*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if netif was added, an err_t otherwise
1879*5d5fbe79SDavid van Moolenbroek */
1880*5d5fbe79SDavid van Moolenbroek err_t
mdns_resp_add_netif(struct netif * netif,const char * hostname,u32_t dns_ttl)1881*5d5fbe79SDavid van Moolenbroek mdns_resp_add_netif(struct netif *netif, const char *hostname, u32_t dns_ttl)
1882*5d5fbe79SDavid van Moolenbroek {
1883*5d5fbe79SDavid van Moolenbroek err_t res;
1884*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns;
1885*5d5fbe79SDavid van Moolenbroek
1886*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_netif: netif != NULL", (netif != NULL), return ERR_VAL);
1887*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_netif: Hostname too long", (strlen(hostname) <= MDNS_LABEL_MAXLEN), return ERR_VAL);
1888*5d5fbe79SDavid van Moolenbroek
1889*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("mdns_resp_add_netif: Double add", NETIF_TO_HOST(netif) == NULL);
1890*5d5fbe79SDavid van Moolenbroek mdns = (struct mdns_host *) mem_malloc(sizeof(struct mdns_host));
1891*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_netif: Alloc failed", (mdns != NULL), return ERR_MEM);
1892*5d5fbe79SDavid van Moolenbroek
1893*5d5fbe79SDavid van Moolenbroek netif_set_client_data(netif, mdns_netif_client_id, mdns);
1894*5d5fbe79SDavid van Moolenbroek
1895*5d5fbe79SDavid van Moolenbroek memset(mdns, 0, sizeof(struct mdns_host));
1896*5d5fbe79SDavid van Moolenbroek MEMCPY(&mdns->name, hostname, LWIP_MIN(MDNS_LABEL_MAXLEN, strlen(hostname)));
1897*5d5fbe79SDavid van Moolenbroek mdns->dns_ttl = dns_ttl;
1898*5d5fbe79SDavid van Moolenbroek
1899*5d5fbe79SDavid van Moolenbroek /* Join multicast groups */
1900*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1901*5d5fbe79SDavid van Moolenbroek res = igmp_joingroup_netif(netif, ip_2_ip4(&v4group));
1902*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1903*5d5fbe79SDavid van Moolenbroek goto cleanup;
1904*5d5fbe79SDavid van Moolenbroek }
1905*5d5fbe79SDavid van Moolenbroek #endif
1906*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1907*5d5fbe79SDavid van Moolenbroek res = mld6_joingroup_netif(netif, ip_2_ip6(&v6group));
1908*5d5fbe79SDavid van Moolenbroek if (res != ERR_OK) {
1909*5d5fbe79SDavid van Moolenbroek goto cleanup;
1910*5d5fbe79SDavid van Moolenbroek }
1911*5d5fbe79SDavid van Moolenbroek #endif
1912*5d5fbe79SDavid van Moolenbroek
1913*5d5fbe79SDavid van Moolenbroek mdns_resp_netif_settings_changed(netif);
1914*5d5fbe79SDavid van Moolenbroek return ERR_OK;
1915*5d5fbe79SDavid van Moolenbroek
1916*5d5fbe79SDavid van Moolenbroek cleanup:
1917*5d5fbe79SDavid van Moolenbroek mem_free(mdns);
1918*5d5fbe79SDavid van Moolenbroek netif_set_client_data(netif, mdns_netif_client_id, NULL);
1919*5d5fbe79SDavid van Moolenbroek return res;
1920*5d5fbe79SDavid van Moolenbroek }
1921*5d5fbe79SDavid van Moolenbroek
1922*5d5fbe79SDavid van Moolenbroek /**
1923*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
1924*5d5fbe79SDavid van Moolenbroek * Stop responding to MDNS queries on this interface, leave multicast groups,
1925*5d5fbe79SDavid van Moolenbroek * and free the helper structure and any of its services.
1926*5d5fbe79SDavid van Moolenbroek * @param netif The network interface to remove.
1927*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if netif was removed, an err_t otherwise
1928*5d5fbe79SDavid van Moolenbroek */
1929*5d5fbe79SDavid van Moolenbroek err_t
mdns_resp_remove_netif(struct netif * netif)1930*5d5fbe79SDavid van Moolenbroek mdns_resp_remove_netif(struct netif *netif)
1931*5d5fbe79SDavid van Moolenbroek {
1932*5d5fbe79SDavid van Moolenbroek int i;
1933*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns;
1934*5d5fbe79SDavid van Moolenbroek
1935*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("mdns_resp_remove_netif: Null pointer", netif);
1936*5d5fbe79SDavid van Moolenbroek mdns = NETIF_TO_HOST(netif);
1937*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_remove_netif: Not an active netif", (mdns != NULL), return ERR_VAL);
1938*5d5fbe79SDavid van Moolenbroek
1939*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; i++) {
1940*5d5fbe79SDavid van Moolenbroek struct mdns_service *service = mdns->services[i];
1941*5d5fbe79SDavid van Moolenbroek if (service) {
1942*5d5fbe79SDavid van Moolenbroek mem_free(service);
1943*5d5fbe79SDavid van Moolenbroek }
1944*5d5fbe79SDavid van Moolenbroek }
1945*5d5fbe79SDavid van Moolenbroek
1946*5d5fbe79SDavid van Moolenbroek /* Leave multicast groups */
1947*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
1948*5d5fbe79SDavid van Moolenbroek igmp_leavegroup_netif(netif, ip_2_ip4(&v4group));
1949*5d5fbe79SDavid van Moolenbroek #endif
1950*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
1951*5d5fbe79SDavid van Moolenbroek mld6_leavegroup_netif(netif, ip_2_ip6(&v6group));
1952*5d5fbe79SDavid van Moolenbroek #endif
1953*5d5fbe79SDavid van Moolenbroek
1954*5d5fbe79SDavid van Moolenbroek mem_free(mdns);
1955*5d5fbe79SDavid van Moolenbroek netif_set_client_data(netif, mdns_netif_client_id, NULL);
1956*5d5fbe79SDavid van Moolenbroek return ERR_OK;
1957*5d5fbe79SDavid van Moolenbroek }
1958*5d5fbe79SDavid van Moolenbroek
1959*5d5fbe79SDavid van Moolenbroek /**
1960*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
1961*5d5fbe79SDavid van Moolenbroek * Add a service to the selected network interface.
1962*5d5fbe79SDavid van Moolenbroek * @param netif The network interface to publish this service on
1963*5d5fbe79SDavid van Moolenbroek * @param name The name of the service
1964*5d5fbe79SDavid van Moolenbroek * @param service The service type, like "_http"
1965*5d5fbe79SDavid van Moolenbroek * @param proto The service protocol, DNSSD_PROTO_TCP for TCP ("_tcp") and DNSSD_PROTO_UDP
1966*5d5fbe79SDavid van Moolenbroek * for others ("_udp")
1967*5d5fbe79SDavid van Moolenbroek * @param port The port the service listens to
1968*5d5fbe79SDavid van Moolenbroek * @param dns_ttl Validity time in seconds to send out for service data in DNS replies
1969*5d5fbe79SDavid van Moolenbroek * @param txt_fn Callback to get TXT data. Will be called each time a TXT reply is created to
1970*5d5fbe79SDavid van Moolenbroek * allow dynamic replies.
1971*5d5fbe79SDavid van Moolenbroek * @param txt_data Userdata pointer for txt_fn
1972*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if the service was added to the netif, an err_t otherwise
1973*5d5fbe79SDavid van Moolenbroek */
1974*5d5fbe79SDavid van Moolenbroek err_t
mdns_resp_add_service(struct netif * netif,const char * name,const char * service,enum mdns_sd_proto proto,u16_t port,u32_t dns_ttl,service_get_txt_fn_t txt_fn,void * txt_data)1975*5d5fbe79SDavid van Moolenbroek mdns_resp_add_service(struct netif *netif, const char *name, const char *service, enum mdns_sd_proto proto, u16_t port, u32_t dns_ttl, service_get_txt_fn_t txt_fn, void *txt_data)
1976*5d5fbe79SDavid van Moolenbroek {
1977*5d5fbe79SDavid van Moolenbroek int i;
1978*5d5fbe79SDavid van Moolenbroek int slot = -1;
1979*5d5fbe79SDavid van Moolenbroek struct mdns_service *srv;
1980*5d5fbe79SDavid van Moolenbroek struct mdns_host* mdns;
1981*5d5fbe79SDavid van Moolenbroek
1982*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("mdns_resp_add_service: netif != NULL", netif);
1983*5d5fbe79SDavid van Moolenbroek mdns = NETIF_TO_HOST(netif);
1984*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Not an mdns netif", (mdns != NULL), return ERR_VAL);
1985*5d5fbe79SDavid van Moolenbroek
1986*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Name too long", (strlen(name) <= MDNS_LABEL_MAXLEN), return ERR_VAL);
1987*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Service too long", (strlen(service) <= MDNS_LABEL_MAXLEN), return ERR_VAL);
1988*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Bad proto (need TCP or UDP)", (proto == DNSSD_PROTO_TCP || proto == DNSSD_PROTO_UDP), return ERR_VAL);
1989*5d5fbe79SDavid van Moolenbroek
1990*5d5fbe79SDavid van Moolenbroek for (i = 0; i < MDNS_MAX_SERVICES; i++) {
1991*5d5fbe79SDavid van Moolenbroek if (mdns->services[i] == NULL) {
1992*5d5fbe79SDavid van Moolenbroek slot = i;
1993*5d5fbe79SDavid van Moolenbroek break;
1994*5d5fbe79SDavid van Moolenbroek }
1995*5d5fbe79SDavid van Moolenbroek }
1996*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Service list full (increase MDNS_MAX_SERVICES)", (slot >= 0), return ERR_MEM);
1997*5d5fbe79SDavid van Moolenbroek
1998*5d5fbe79SDavid van Moolenbroek srv = (struct mdns_service*)mem_malloc(sizeof(struct mdns_service));
1999*5d5fbe79SDavid van Moolenbroek LWIP_ERROR("mdns_resp_add_service: Alloc failed", (srv != NULL), return ERR_MEM);
2000*5d5fbe79SDavid van Moolenbroek
2001*5d5fbe79SDavid van Moolenbroek memset(srv, 0, sizeof(struct mdns_service));
2002*5d5fbe79SDavid van Moolenbroek
2003*5d5fbe79SDavid van Moolenbroek MEMCPY(&srv->name, name, LWIP_MIN(MDNS_LABEL_MAXLEN, strlen(name)));
2004*5d5fbe79SDavid van Moolenbroek MEMCPY(&srv->service, service, LWIP_MIN(MDNS_LABEL_MAXLEN, strlen(service)));
2005*5d5fbe79SDavid van Moolenbroek srv->txt_fn = txt_fn;
2006*5d5fbe79SDavid van Moolenbroek srv->txt_userdata = txt_data;
2007*5d5fbe79SDavid van Moolenbroek srv->proto = (u16_t)proto;
2008*5d5fbe79SDavid van Moolenbroek srv->port = port;
2009*5d5fbe79SDavid van Moolenbroek srv->dns_ttl = dns_ttl;
2010*5d5fbe79SDavid van Moolenbroek
2011*5d5fbe79SDavid van Moolenbroek mdns->services[slot] = srv;
2012*5d5fbe79SDavid van Moolenbroek
2013*5d5fbe79SDavid van Moolenbroek /* Announce on IPv6 and IPv4 */
2014*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV6
2015*5d5fbe79SDavid van Moolenbroek mdns_announce(netif, IP6_ADDR_ANY);
2016*5d5fbe79SDavid van Moolenbroek #endif
2017*5d5fbe79SDavid van Moolenbroek #if LWIP_IPV4
2018*5d5fbe79SDavid van Moolenbroek mdns_announce(netif, IP4_ADDR_ANY);
2019*5d5fbe79SDavid van Moolenbroek #endif
2020*5d5fbe79SDavid van Moolenbroek
2021*5d5fbe79SDavid van Moolenbroek return ERR_OK;
2022*5d5fbe79SDavid van Moolenbroek }
2023*5d5fbe79SDavid van Moolenbroek
2024*5d5fbe79SDavid van Moolenbroek /**
2025*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
2026*5d5fbe79SDavid van Moolenbroek * Call this function from inside the service_get_txt_fn_t callback to add text data.
2027*5d5fbe79SDavid van Moolenbroek * Buffer for TXT data is 256 bytes, and each field is prefixed with a length byte.
2028*5d5fbe79SDavid van Moolenbroek * @param service The service provided to the get_txt callback
2029*5d5fbe79SDavid van Moolenbroek * @param txt String to add to the TXT field.
2030*5d5fbe79SDavid van Moolenbroek * @param txt_len Length of string
2031*5d5fbe79SDavid van Moolenbroek * @return ERR_OK if the string was added to the reply, an err_t otherwise
2032*5d5fbe79SDavid van Moolenbroek */
2033*5d5fbe79SDavid van Moolenbroek err_t
mdns_resp_add_service_txtitem(struct mdns_service * service,const char * txt,u8_t txt_len)2034*5d5fbe79SDavid van Moolenbroek mdns_resp_add_service_txtitem(struct mdns_service *service, const char *txt, u8_t txt_len)
2035*5d5fbe79SDavid van Moolenbroek {
2036*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("mdns_resp_add_service_txtitem: service != NULL", service);
2037*5d5fbe79SDavid van Moolenbroek
2038*5d5fbe79SDavid van Moolenbroek /* Use a mdns_domain struct to store txt chunks since it is the same encoding */
2039*5d5fbe79SDavid van Moolenbroek return mdns_domain_add_label(&service->txtdata, txt, txt_len);
2040*5d5fbe79SDavid van Moolenbroek }
2041*5d5fbe79SDavid van Moolenbroek
2042*5d5fbe79SDavid van Moolenbroek /**
2043*5d5fbe79SDavid van Moolenbroek * @ingroup mdns
2044*5d5fbe79SDavid van Moolenbroek * Initiate MDNS responder. Will open UDP sockets on port 5353
2045*5d5fbe79SDavid van Moolenbroek */
2046*5d5fbe79SDavid van Moolenbroek void
mdns_resp_init(void)2047*5d5fbe79SDavid van Moolenbroek mdns_resp_init(void)
2048*5d5fbe79SDavid van Moolenbroek {
2049*5d5fbe79SDavid van Moolenbroek err_t res;
2050*5d5fbe79SDavid van Moolenbroek
2051*5d5fbe79SDavid van Moolenbroek mdns_pcb = udp_new_ip_type(IPADDR_TYPE_ANY);
2052*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("Failed to allocate pcb", mdns_pcb != NULL);
2053*5d5fbe79SDavid van Moolenbroek #if LWIP_MULTICAST_TX_OPTIONS
2054*5d5fbe79SDavid van Moolenbroek udp_set_multicast_ttl(mdns_pcb, MDNS_TTL);
2055*5d5fbe79SDavid van Moolenbroek #else
2056*5d5fbe79SDavid van Moolenbroek mdns_pcb->ttl = MDNS_TTL;
2057*5d5fbe79SDavid van Moolenbroek #endif
2058*5d5fbe79SDavid van Moolenbroek res = udp_bind(mdns_pcb, IP_ANY_TYPE, MDNS_PORT);
2059*5d5fbe79SDavid van Moolenbroek LWIP_UNUSED_ARG(res); /* in case of LWIP_NOASSERT */
2060*5d5fbe79SDavid van Moolenbroek LWIP_ASSERT("Failed to bind pcb", res == ERR_OK);
2061*5d5fbe79SDavid van Moolenbroek udp_recv(mdns_pcb, mdns_recv, NULL);
2062*5d5fbe79SDavid van Moolenbroek
2063*5d5fbe79SDavid van Moolenbroek mdns_netif_client_id = netif_alloc_client_data_id();
2064*5d5fbe79SDavid van Moolenbroek
2065*5d5fbe79SDavid van Moolenbroek /* register for netif events when started on first netif */
2066*5d5fbe79SDavid van Moolenbroek netif_add_ext_callback(&netif_callback, mdns_netif_ext_status_callback);
2067*5d5fbe79SDavid van Moolenbroek }
2068*5d5fbe79SDavid van Moolenbroek
2069*5d5fbe79SDavid van Moolenbroek #endif /* LWIP_MDNS_RESPONDER */
2070