Patrick Welche <prlw1@cam.ac.uk>
[netbsd-mini2440.git] / external / apache2 / mDNSResponder / dist / mDNSPosix / mDNSPosix.c
blobe984ed0e0cba66f4651096f68c0388b8d09a41f1
1 /* -*- Mode: C; tab-width: 4 -*-
3 * Copyright (c) 2002-2004 Apple Computer, Inc. All rights reserved.
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
17 * Formatting notes:
18 * This code follows the "Whitesmiths style" C indentation rules. Plenty of discussion
19 * on C indentation can be found on the web, such as <http://www.kafejo.com/komp/1tbs.htm>,
20 * but for the sake of brevity here I will say just this: Curly braces are not syntactially
21 * part of an "if" statement; they are the beginning and ending markers of a compound statement;
22 * therefore common sense dictates that if they are part of a compound statement then they
23 * should be indented to the same level as everything else in that compound statement.
24 * Indenting curly braces at the same level as the "if" implies that curly braces are
25 * part of the "if", which is false. (This is as misleading as people who write "char* x,y;"
26 * thinking that variables x and y are both of type "char*" -- and anyone who doesn't
27 * understand why variable y is not of type "char*" just proves the point that poor code
28 * layout leads people to unfortunate misunderstandings about how the C language really works.)
30 Change History (most recent first):
32 Log: mDNSPosix.c,v $
33 Revision 1.108 2009/01/25 03:16:46 mkrochma
34 Added skeleton definition of mDNSPlatformSetLocalARP
36 Revision 1.107 2009/01/07 08:25:03 mkrochma
37 Added skeleton definition of mDNSPlatformUpdateProxyList
39 Revision 1.106 2008/10/22 17:19:57 cheshire
40 Don't need to define BPF_fd any more (it's now per-interface, not global)
42 Revision 1.105 2008/10/03 23:34:08 cheshire
43 Added skeleton definition of mDNSPlatformSendRawPacket
45 Revision 1.104 2008/09/05 22:16:48 cheshire
46 <rdar://problem/3988320> Should use randomized source ports and transaction IDs to avoid DNS cache poisoning
47 Add "UDPSocket *src" parameter in mDNSPlatformSendUDP
49 Revision 1.103 2007/10/02 19:31:17 cheshire
50 In ParseDNSServers, should use strncasecmp for case-insensitive compare
52 Revision 1.102 2007/09/12 19:23:17 cheshire
53 Get rid of unnecessary mDNSPlatformTCPIsConnected() routine
55 Revision 1.101 2007/07/20 00:54:23 cheshire
56 <rdar://problem/4641118> Need separate SCPreferences for per-user .Mac settings
58 Revision 1.100 2007/07/19 21:45:30 cheshire
59 Fixed code spacing
61 Revision 1.99 2007/07/11 02:56:51 cheshire
62 <rdar://problem/5303807> Register IPv6-only hostname and don't create port mappings for AutoTunnel services
63 Remove unused mDNSPlatformDefaultRegDomainChanged
65 Revision 1.98 2007/06/20 01:10:13 cheshire
66 <rdar://problem/5280520> Sync iPhone changes into main mDNSResponder code
68 Revision 1.97 2007/04/26 00:35:16 cheshire
69 <rdar://problem/5140339> uDNS: Domain discovery not working over VPN
70 Fixes to make sure results update correctly when connectivity changes (e.g. a DNS server
71 inside the firewall may give answers where a public one gives none, and vice versa.)
73 Revision 1.96 2007/04/22 20:29:59 cheshire
74 Fix locking error
76 Revision 1.95 2007/04/22 20:15:46 cheshire
77 Add missing parameters for mDNSPosixEventCallback
79 Revision 1.94 2007/04/17 19:21:29 cheshire
80 <rdar://problem/5140339> Domain discovery not working over VPN
82 Revision 1.93 2007/04/16 20:49:40 cheshire
83 Fix compile errors for mDNSPosix build
85 Revision 1.92 2007/04/05 20:40:37 cheshire
86 Remove unused mDNSPlatformTCPGetFlags()
88 Revision 1.91 2007/03/22 18:31:48 cheshire
89 Put dst parameter first in mDNSPlatformStrCopy/mDNSPlatformMemCopy, like conventional Posix strcpy/memcpy
91 Revision 1.90 2007/03/21 00:31:45 cheshire
92 Remove unnecessary (and unimplemented) platform functions
94 Revision 1.89 2007/03/20 17:07:15 cheshire
95 Rename "struct uDNS_TCPSocket_struct" to "TCPSocket", "struct uDNS_UDPSocket_struct" to "UDPSocket"
97 Revision 1.88 2007/03/07 00:30:18 mkrochma
98 <rdar://problem/5034370> POSIX: kDNSServiceInterfaceIndexAny not correctly handled
99 Thanks goes to Aidan Williams of Audinate who did a lot of work in diagnosing this
101 Revision 1.87 2007/02/08 21:12:28 cheshire
102 <rdar://problem/4386497> Stop reading /etc/mDNSResponder.conf on every sleep/wake
104 Revision 1.86 2007/01/05 08:30:52 cheshire
105 Trim excessive "Log" checkin history from before 2006
106 (checkin history still available via "cvs log ..." of course)
108 Revision 1.85 2007/01/04 23:12:20 cheshire
109 Remove unused mDNSPlatformDefaultBrowseDomainChanged
111 Revision 1.84 2006/12/22 21:07:35 cheshire
112 <rdar://problem/4742742> Read *all* DNS keys from keychain,
113 not just key for the system-wide default registration domain
115 Revision 1.83 2006/12/21 00:09:46 cheshire
116 Use mDNSPlatformMemZero instead of bzero
118 Revision 1.82 2006/12/19 22:43:55 cheshire
119 Fix compiler warnings
121 Revision 1.81 2006/08/14 23:24:46 cheshire
122 Re-licensed mDNSResponder daemon source code under Apache License, Version 2.0
124 Revision 1.80 2006/07/22 03:05:33 cheshire
125 Improve error reporting for socket creation failures
127 Revision 1.79 2006/07/06 00:02:16 cheshire
128 <rdar://problem/4472014> Add Private DNS client functionality to mDNSResponder
130 Revision 1.78 2006/06/28 09:12:22 cheshire
131 Added debugging message
133 Revision 1.77 2006/03/19 02:00:11 cheshire
134 <rdar://problem/4073825> Improve logic for delaying packets after repeated interface transitions
136 Revision 1.76 2006/01/09 19:29:16 cheshire
137 <rdar://problem/4403128> Cap number of "sendto failed" messages we allow mDNSResponder to log
139 Revision 1.75 2006/01/05 22:04:57 cheshire
140 <rdar://problem/4399479> Log error message when send fails with "operation not permitted"
142 Revision 1.74 2006/01/05 21:45:27 cheshire
143 <rdar://problem/4400118> Fix uninitialized structure member in IPv6 code
147 #include "mDNSEmbeddedAPI.h" // Defines the interface provided to the client layer above
148 #include "DNSCommon.h"
149 #include "mDNSPosix.h" // Defines the specific types needed to run mDNS on this platform
150 #include "dns_sd.h"
152 #include <assert.h>
153 #include <stdio.h>
154 #include <stdlib.h>
155 #include <errno.h>
156 #include <string.h>
157 #include <unistd.h>
158 #include <syslog.h>
159 #include <stdarg.h>
160 #include <fcntl.h>
161 #include <sys/types.h>
162 #include <sys/time.h>
163 #include <sys/socket.h>
164 #include <sys/uio.h>
165 #include <sys/select.h>
166 #include <netinet/in.h>
167 #include <arpa/inet.h>
168 #include <time.h> // platform support for UTC time
170 #if USES_NETLINK
171 #include <asm/types.h>
172 #include <linux/netlink.h>
173 #include <linux/rtnetlink.h>
174 #else // USES_NETLINK
175 #include <net/route.h>
176 #include <net/if.h>
177 #endif // USES_NETLINK
179 #include "mDNSUNP.h"
180 #include "GenLinkedList.h"
182 // ***************************************************************************
183 // Structures
185 // We keep a list of client-supplied event sources in PosixEventSource records
186 struct PosixEventSource
188 mDNSPosixEventCallback Callback;
189 void *Context;
190 int fd;
191 struct PosixEventSource *Next;
193 typedef struct PosixEventSource PosixEventSource;
195 // Context record for interface change callback
196 struct IfChangeRec
198 int NotifySD;
199 mDNS *mDNS;
201 typedef struct IfChangeRec IfChangeRec;
203 // Note that static data is initialized to zero in (modern) C.
204 static fd_set gEventFDs;
205 static int gMaxFD; // largest fd in gEventFDs
206 static GenLinkedList gEventSources; // linked list of PosixEventSource's
207 static sigset_t gEventSignalSet; // Signals which event loop listens for
208 static sigset_t gEventSignals; // Signals which were received while inside loop
210 // ***************************************************************************
211 // Globals (for debugging)
213 static int num_registered_interfaces = 0;
214 static int num_pkts_accepted = 0;
215 static int num_pkts_rejected = 0;
217 // ***************************************************************************
218 // Functions
220 int gMDNSPlatformPosixVerboseLevel = 0;
222 #define PosixErrorToStatus(errNum) ((errNum) == 0 ? mStatus_NoError : mStatus_UnknownErr)
224 mDNSlocal void SockAddrTomDNSAddr(const struct sockaddr *const sa, mDNSAddr *ipAddr, mDNSIPPort *ipPort)
226 switch (sa->sa_family)
228 case AF_INET:
230 struct sockaddr_in *sin = (struct sockaddr_in*)sa;
231 ipAddr->type = mDNSAddrType_IPv4;
232 ipAddr->ip.v4.NotAnInteger = sin->sin_addr.s_addr;
233 if (ipPort) ipPort->NotAnInteger = sin->sin_port;
234 break;
237 #if HAVE_IPV6
238 case AF_INET6:
240 struct sockaddr_in6 *sin6 = (struct sockaddr_in6*)sa;
241 #ifndef NOT_HAVE_SA_LEN
242 assert(sin6->sin6_len == sizeof(*sin6));
243 #endif
244 ipAddr->type = mDNSAddrType_IPv6;
245 ipAddr->ip.v6 = *(mDNSv6Addr*)&sin6->sin6_addr;
246 if (ipPort) ipPort->NotAnInteger = sin6->sin6_port;
247 break;
249 #endif
251 default:
252 verbosedebugf("SockAddrTomDNSAddr: Uknown address family %d\n", sa->sa_family);
253 ipAddr->type = mDNSAddrType_None;
254 if (ipPort) ipPort->NotAnInteger = 0;
255 break;
259 #if COMPILER_LIKES_PRAGMA_MARK
260 #pragma mark ***** Send and Receive
261 #endif
263 // mDNS core calls this routine when it needs to send a packet.
264 mDNSexport mStatus mDNSPlatformSendUDP(const mDNS *const m, const void *const msg, const mDNSu8 *const end,
265 mDNSInterfaceID InterfaceID, UDPSocket *src, const mDNSAddr *dst, mDNSIPPort dstPort)
267 int err = 0;
268 struct sockaddr_storage to;
269 PosixNetworkInterface * thisIntf = (PosixNetworkInterface *)(InterfaceID);
270 int sendingsocket = -1;
272 (void)src; // Will need to use this parameter once we implement mDNSPlatformUDPSocket/mDNSPlatformUDPClose
274 assert(m != NULL);
275 assert(msg != NULL);
276 assert(end != NULL);
277 assert((((char *) end) - ((char *) msg)) > 0);
278 assert(dstPort.NotAnInteger != 0);
280 if (dst->type == mDNSAddrType_IPv4)
282 struct sockaddr_in *sin = (struct sockaddr_in*)&to;
283 #ifndef NOT_HAVE_SA_LEN
284 sin->sin_len = sizeof(*sin);
285 #endif
286 sin->sin_family = AF_INET;
287 sin->sin_port = dstPort.NotAnInteger;
288 sin->sin_addr.s_addr = dst->ip.v4.NotAnInteger;
289 sendingsocket = thisIntf ? thisIntf->multicastSocket4 : m->p->unicastSocket4;
292 #if HAVE_IPV6
293 else if (dst->type == mDNSAddrType_IPv6)
295 struct sockaddr_in6 *sin6 = (struct sockaddr_in6*)&to;
296 mDNSPlatformMemZero(sin6, sizeof(*sin6));
297 #ifndef NOT_HAVE_SA_LEN
298 sin6->sin6_len = sizeof(*sin6);
299 #endif
300 sin6->sin6_family = AF_INET6;
301 sin6->sin6_port = dstPort.NotAnInteger;
302 sin6->sin6_addr = *(struct in6_addr*)&dst->ip.v6;
303 sendingsocket = thisIntf ? thisIntf->multicastSocket6 : m->p->unicastSocket6;
305 #endif
307 if (sendingsocket >= 0)
308 err = sendto(sendingsocket, msg, (char*)end - (char*)msg, 0, (struct sockaddr *)&to, GET_SA_LEN(to));
310 if (err > 0) err = 0;
311 else if (err < 0)
313 static int MessageCount = 0;
314 // Don't report EHOSTDOWN (i.e. ARP failure), ENETDOWN, or no route to host for unicast destinations
315 if (!mDNSAddressIsAllDNSLinkGroup(dst))
316 if (errno == EHOSTDOWN || errno == ENETDOWN || errno == EHOSTUNREACH || errno == ENETUNREACH) return(mStatus_TransientErr);
318 if (MessageCount < 1000)
320 MessageCount++;
321 if (thisIntf)
322 LogMsg("mDNSPlatformSendUDP got error %d (%s) sending packet to %#a on interface %#a/%s/%d",
323 errno, strerror(errno), dst, &thisIntf->coreIntf.ip, thisIntf->intfName, thisIntf->index);
324 else
325 LogMsg("mDNSPlatformSendUDP got error %d (%s) sending packet to %#a", errno, strerror(errno), dst);
329 return PosixErrorToStatus(err);
332 // This routine is called when the main loop detects that data is available on a socket.
333 mDNSlocal void SocketDataReady(mDNS *const m, PosixNetworkInterface *intf, int skt)
335 mDNSAddr senderAddr, destAddr;
336 mDNSIPPort senderPort;
337 ssize_t packetLen;
338 DNSMessage packet;
339 struct my_in_pktinfo packetInfo;
340 struct sockaddr_storage from;
341 socklen_t fromLen;
342 int flags;
343 mDNSu8 ttl;
344 mDNSBool reject;
345 const mDNSInterfaceID InterfaceID = intf ? intf->coreIntf.InterfaceID : NULL;
347 assert(m != NULL);
348 assert(skt >= 0);
350 fromLen = sizeof(from);
351 flags = 0;
352 packetLen = recvfrom_flags(skt, &packet, sizeof(packet), &flags, (struct sockaddr *) &from, &fromLen, &packetInfo, &ttl);
354 if (packetLen >= 0)
356 SockAddrTomDNSAddr((struct sockaddr*)&from, &senderAddr, &senderPort);
357 SockAddrTomDNSAddr((struct sockaddr*)&packetInfo.ipi_addr, &destAddr, NULL);
359 // If we have broken IP_RECVDSTADDR functionality (so far
360 // I've only seen this on OpenBSD) then apply a hack to
361 // convince mDNS Core that this isn't a spoof packet.
362 // Basically what we do is check to see whether the
363 // packet arrived as a multicast and, if so, set its
364 // destAddr to the mDNS address.
366 // I must admit that I could just be doing something
367 // wrong on OpenBSD and hence triggering this problem
368 // but I'm at a loss as to how.
370 // If this platform doesn't have IP_PKTINFO or IP_RECVDSTADDR, then we have
371 // no way to tell the destination address or interface this packet arrived on,
372 // so all we can do is just assume it's a multicast
374 #if HAVE_BROKEN_RECVDSTADDR || (!defined(IP_PKTINFO) && !defined(IP_RECVDSTADDR))
375 if ((destAddr.NotAnInteger == 0) && (flags & MSG_MCAST))
377 destAddr.type = senderAddr.type;
378 if (senderAddr.type == mDNSAddrType_IPv4) destAddr.ip.v4 = AllDNSLinkGroup_v4.ip.v4;
379 else if (senderAddr.type == mDNSAddrType_IPv6) destAddr.ip.v6 = AllDNSLinkGroup_v6.ip.v6;
381 #endif
383 // We only accept the packet if the interface on which it came
384 // in matches the interface associated with this socket.
385 // We do this match by name or by index, depending on which
386 // information is available. recvfrom_flags sets the name
387 // to "" if the name isn't available, or the index to -1
388 // if the index is available. This accomodates the various
389 // different capabilities of our target platforms.
391 reject = mDNSfalse;
392 if (!intf)
394 // Ignore multicasts accidentally delivered to our unicast receiving socket
395 if (mDNSAddrIsDNSMulticast(&destAddr)) packetLen = -1;
397 else
399 if (packetInfo.ipi_ifname[0] != 0) reject = (strcmp(packetInfo.ipi_ifname, intf->intfName) != 0);
400 else if (packetInfo.ipi_ifindex != -1) reject = (packetInfo.ipi_ifindex != intf->index);
402 if (reject)
404 verbosedebugf("SocketDataReady ignored a packet from %#a to %#a on interface %s/%d expecting %#a/%s/%d/%d",
405 &senderAddr, &destAddr, packetInfo.ipi_ifname, packetInfo.ipi_ifindex,
406 &intf->coreIntf.ip, intf->intfName, intf->index, skt);
407 packetLen = -1;
408 num_pkts_rejected++;
409 if (num_pkts_rejected > (num_pkts_accepted + 1) * (num_registered_interfaces + 1) * 2)
411 fprintf(stderr,
412 "*** WARNING: Received %d packets; Accepted %d packets; Rejected %d packets because of interface mismatch\n",
413 num_pkts_accepted + num_pkts_rejected, num_pkts_accepted, num_pkts_rejected);
414 num_pkts_accepted = 0;
415 num_pkts_rejected = 0;
418 else
420 verbosedebugf("SocketDataReady got a packet from %#a to %#a on interface %#a/%s/%d/%d",
421 &senderAddr, &destAddr, &intf->coreIntf.ip, intf->intfName, intf->index, skt);
422 num_pkts_accepted++;
427 if (packetLen >= 0)
428 mDNSCoreReceive(m, &packet, (mDNSu8 *)&packet + packetLen,
429 &senderAddr, senderPort, &destAddr, MulticastDNSPort, InterfaceID);
432 mDNSexport TCPSocket *mDNSPlatformTCPSocket(mDNS * const m, TCPSocketFlags flags, mDNSIPPort * port)
434 (void)m; // Unused
435 (void)flags; // Unused
436 (void)port; // Unused
437 return NULL;
440 mDNSexport TCPSocket *mDNSPlatformTCPAccept(TCPSocketFlags flags, int sd)
442 (void)flags; // Unused
443 (void)sd; // Unused
444 return NULL;
447 mDNSexport int mDNSPlatformTCPGetFD(TCPSocket *sock)
449 (void)sock; // Unused
450 return -1;
453 mDNSexport mStatus mDNSPlatformTCPConnect(TCPSocket *sock, const mDNSAddr *dst, mDNSOpaque16 dstport, mDNSInterfaceID InterfaceID,
454 TCPConnectionCallback callback, void *context)
456 (void)sock; // Unused
457 (void)dst; // Unused
458 (void)dstport; // Unused
459 (void)InterfaceID; // Unused
460 (void)callback; // Unused
461 (void)context; // Unused
462 return(mStatus_UnsupportedErr);
465 mDNSexport void mDNSPlatformTCPCloseConnection(TCPSocket *sock)
467 (void)sock; // Unused
470 mDNSexport long mDNSPlatformReadTCP(TCPSocket *sock, void *buf, unsigned long buflen, mDNSBool * closed)
472 (void)sock; // Unused
473 (void)buf; // Unused
474 (void)buflen; // Unused
475 (void)closed; // Unused
476 return 0;
479 mDNSexport long mDNSPlatformWriteTCP(TCPSocket *sock, const char *msg, unsigned long len)
481 (void)sock; // Unused
482 (void)msg; // Unused
483 (void)len; // Unused
484 return 0;
487 mDNSexport UDPSocket *mDNSPlatformUDPSocket(mDNS * const m, mDNSIPPort port)
489 (void)m; // Unused
490 (void)port; // Unused
491 return NULL;
494 mDNSexport void mDNSPlatformUDPClose(UDPSocket *sock)
496 (void)sock; // Unused
499 mDNSexport void mDNSPlatformUpdateProxyList(mDNS *const m, const mDNSInterfaceID InterfaceID)
501 (void)m; // Unused
502 (void)InterfaceID; // Unused
505 mDNSexport void mDNSPlatformSendRawPacket(const void *const msg, const mDNSu8 *const end, mDNSInterfaceID InterfaceID)
507 (void)msg; // Unused
508 (void)end; // Unused
509 (void)InterfaceID; // Unused
512 mDNSexport void mDNSPlatformSetLocalARP(const mDNSv4Addr *const tpa, const mDNSEthAddr *const tha, mDNSInterfaceID InterfaceID)
514 (void)tpa; // Unused
515 (void)tha; // Unused
516 (void)InterfaceID; // Unused
519 mDNSexport mStatus mDNSPlatformTLSSetupCerts(void)
521 return(mStatus_UnsupportedErr);
524 mDNSexport void mDNSPlatformTLSTearDownCerts(void)
528 #if COMPILER_LIKES_PRAGMA_MARK
529 #pragma mark ***** DDNS Config Platform Functions
530 #endif
532 mDNSexport void mDNSPlatformSetDNSConfig(mDNS *const m, mDNSBool setservers, mDNSBool setsearch, domainname *const fqdn, DNameListElem **RegDomains, DNameListElem **BrowseDomains)
534 (void) m;
535 (void) setservers;
536 (void) fqdn;
537 (void) setsearch;
538 (void) RegDomains;
539 (void) BrowseDomains;
542 mDNSexport mStatus mDNSPlatformGetPrimaryInterface(mDNS * const m, mDNSAddr * v4, mDNSAddr * v6, mDNSAddr * router)
544 (void) m;
545 (void) v4;
546 (void) v6;
547 (void) router;
549 return mStatus_UnsupportedErr;
552 mDNSexport void mDNSPlatformDynDNSHostNameStatusChanged(const domainname *const dname, const mStatus status)
554 (void) dname;
555 (void) status;
558 #if COMPILER_LIKES_PRAGMA_MARK
559 #pragma mark ***** Init and Term
560 #endif
562 // This gets the current hostname, truncating it at the first dot if necessary
563 mDNSlocal void GetUserSpecifiedRFC1034ComputerName(domainlabel *const namelabel)
565 int len = 0;
566 gethostname((char *)(&namelabel->c[1]), MAX_DOMAIN_LABEL);
567 while (len < MAX_DOMAIN_LABEL && namelabel->c[len+1] && namelabel->c[len+1] != '.') len++;
568 namelabel->c[0] = len;
571 // On OS X this gets the text of the field labelled "Computer Name" in the Sharing Prefs Control Panel
572 // Other platforms can either get the information from the appropriate place,
573 // or they can alternatively just require all registering services to provide an explicit name
574 mDNSlocal void GetUserSpecifiedFriendlyComputerName(domainlabel *const namelabel)
576 // On Unix we have no better name than the host name, so we just use that.
577 GetUserSpecifiedRFC1034ComputerName(namelabel);
580 mDNSexport int ParseDNSServers(mDNS *m, const char *filePath)
582 char line[256];
583 char nameserver[16];
584 char keyword[11];
585 int numOfServers = 0;
586 FILE *fp = fopen(filePath, "r");
587 if (fp == NULL) return -1;
588 while (fgets(line,sizeof(line),fp))
590 struct in_addr ina;
591 line[255]='\0'; // just to be safe
592 if (sscanf(line,"%10s %15s", keyword, nameserver) != 2) continue; // it will skip whitespaces
593 if (strncasecmp(keyword,"nameserver",10)) continue;
594 if (inet_aton(nameserver, (struct in_addr *)&ina) != 0)
596 mDNSAddr DNSAddr;
597 DNSAddr.type = mDNSAddrType_IPv4;
598 DNSAddr.ip.v4.NotAnInteger = ina.s_addr;
599 mDNS_AddDNSServer(m, NULL, mDNSInterface_Any, &DNSAddr, UnicastDNSPort);
600 numOfServers++;
603 fclose(fp);
604 return (numOfServers > 0) ? 0 : -1;
607 // Searches the interface list looking for the named interface.
608 // Returns a pointer to if it found, or NULL otherwise.
609 mDNSlocal PosixNetworkInterface *SearchForInterfaceByName(mDNS *const m, const char *intfName)
611 PosixNetworkInterface *intf;
613 assert(m != NULL);
614 assert(intfName != NULL);
616 intf = (PosixNetworkInterface*)(m->HostInterfaces);
617 while ((intf != NULL) && (strcmp(intf->intfName, intfName) != 0))
618 intf = (PosixNetworkInterface *)(intf->coreIntf.next);
620 return intf;
623 mDNSexport mDNSInterfaceID mDNSPlatformInterfaceIDfromInterfaceIndex(mDNS *const m, mDNSu32 index)
625 PosixNetworkInterface *intf;
627 assert(m != NULL);
629 if (index == kDNSServiceInterfaceIndexLocalOnly) return(mDNSInterface_LocalOnly);
630 if (index == kDNSServiceInterfaceIndexAny ) return(mDNSInterface_Any);
632 intf = (PosixNetworkInterface*)(m->HostInterfaces);
633 while ((intf != NULL) && (mDNSu32) intf->index != index)
634 intf = (PosixNetworkInterface *)(intf->coreIntf.next);
636 return (mDNSInterfaceID) intf;
639 mDNSexport mDNSu32 mDNSPlatformInterfaceIndexfromInterfaceID(mDNS *const m, mDNSInterfaceID id)
641 PosixNetworkInterface *intf;
643 assert(m != NULL);
645 if (id == mDNSInterface_LocalOnly) return(kDNSServiceInterfaceIndexLocalOnly);
646 if (id == mDNSInterface_Any ) return(kDNSServiceInterfaceIndexAny);
648 intf = (PosixNetworkInterface*)(m->HostInterfaces);
649 while ((intf != NULL) && (mDNSInterfaceID) intf != id)
650 intf = (PosixNetworkInterface *)(intf->coreIntf.next);
652 return intf ? intf->index : 0;
655 // Frees the specified PosixNetworkInterface structure. The underlying
656 // interface must have already been deregistered with the mDNS core.
657 mDNSlocal void FreePosixNetworkInterface(PosixNetworkInterface *intf)
659 assert(intf != NULL);
660 if (intf->intfName != NULL) free((void *)intf->intfName);
661 if (intf->multicastSocket4 != -1) assert(close(intf->multicastSocket4) == 0);
662 #if HAVE_IPV6
663 if (intf->multicastSocket6 != -1) assert(close(intf->multicastSocket6) == 0);
664 #endif
665 free(intf);
668 // Grab the first interface, deregister it, free it, and repeat until done.
669 mDNSlocal void ClearInterfaceList(mDNS *const m)
671 assert(m != NULL);
673 while (m->HostInterfaces)
675 PosixNetworkInterface *intf = (PosixNetworkInterface*)(m->HostInterfaces);
676 mDNS_DeregisterInterface(m, &intf->coreIntf, mDNSfalse);
677 if (gMDNSPlatformPosixVerboseLevel > 0) fprintf(stderr, "Deregistered interface %s\n", intf->intfName);
678 FreePosixNetworkInterface(intf);
680 num_registered_interfaces = 0;
681 num_pkts_accepted = 0;
682 num_pkts_rejected = 0;
685 // Sets up a send/receive socket.
686 // If mDNSIPPort port is non-zero, then it's a multicast socket on the specified interface
687 // If mDNSIPPort port is zero, then it's a randomly assigned port number, used for sending unicast queries
688 mDNSlocal int SetupSocket(struct sockaddr *intfAddr, mDNSIPPort port, int interfaceIndex, int *sktPtr)
690 int err = 0;
691 static const int kOn = 1;
692 static const int kIntTwoFiveFive = 255;
693 static const unsigned char kByteTwoFiveFive = 255;
694 const mDNSBool JoinMulticastGroup = (port.NotAnInteger != 0);
696 (void) interfaceIndex; // This parameter unused on plaforms that don't have IPv6
697 assert(intfAddr != NULL);
698 assert(sktPtr != NULL);
699 assert(*sktPtr == -1);
701 // Open the socket...
702 if (intfAddr->sa_family == AF_INET) *sktPtr = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
703 #if HAVE_IPV6
704 else if (intfAddr->sa_family == AF_INET6) *sktPtr = socket(PF_INET6, SOCK_DGRAM, IPPROTO_UDP);
705 #endif
706 else return EINVAL;
708 if (*sktPtr < 0) { err = errno; perror((intfAddr->sa_family == AF_INET) ? "socket AF_INET" : "socket AF_INET6"); }
710 // ... with a shared UDP port, if it's for multicast receiving
711 if (err == 0 && port.NotAnInteger)
713 #if defined(SO_REUSEPORT)
714 err = setsockopt(*sktPtr, SOL_SOCKET, SO_REUSEPORT, &kOn, sizeof(kOn));
715 #elif defined(SO_REUSEADDR)
716 err = setsockopt(*sktPtr, SOL_SOCKET, SO_REUSEADDR, &kOn, sizeof(kOn));
717 #else
718 #error This platform has no way to avoid address busy errors on multicast.
719 #endif
720 if (err < 0) { err = errno; perror("setsockopt - SO_REUSExxxx"); }
723 // We want to receive destination addresses and interface identifiers.
724 if (intfAddr->sa_family == AF_INET)
726 struct ip_mreq imr;
727 struct sockaddr_in bindAddr;
728 if (err == 0)
730 #if defined(IP_PKTINFO) // Linux
731 err = setsockopt(*sktPtr, IPPROTO_IP, IP_PKTINFO, &kOn, sizeof(kOn));
732 if (err < 0) { err = errno; perror("setsockopt - IP_PKTINFO"); }
733 #elif defined(IP_RECVDSTADDR) || defined(IP_RECVIF) // BSD and Solaris
734 #if defined(IP_RECVDSTADDR)
735 err = setsockopt(*sktPtr, IPPROTO_IP, IP_RECVDSTADDR, &kOn, sizeof(kOn));
736 if (err < 0) { err = errno; perror("setsockopt - IP_RECVDSTADDR"); }
737 #endif
738 #if defined(IP_RECVIF)
739 if (err == 0)
741 err = setsockopt(*sktPtr, IPPROTO_IP, IP_RECVIF, &kOn, sizeof(kOn));
742 if (err < 0) { err = errno; perror("setsockopt - IP_RECVIF"); }
744 #endif
745 #else
746 #warning This platform has no way to get the destination interface information -- will only work for single-homed hosts
747 #endif
749 #if defined(IP_RECVTTL) // Linux
750 if (err == 0)
752 setsockopt(*sktPtr, IPPROTO_IP, IP_RECVTTL, &kOn, sizeof(kOn));
753 // We no longer depend on being able to get the received TTL, so don't worry if the option fails
755 #endif
757 // Add multicast group membership on this interface
758 if (err == 0 && JoinMulticastGroup)
760 imr.imr_multiaddr.s_addr = AllDNSLinkGroup_v4.ip.v4.NotAnInteger;
761 imr.imr_interface = ((struct sockaddr_in*)intfAddr)->sin_addr;
762 err = setsockopt(*sktPtr, IPPROTO_IP, IP_ADD_MEMBERSHIP, &imr, sizeof(imr));
763 if (err < 0) { err = errno; perror("setsockopt - IP_ADD_MEMBERSHIP"); }
766 // Specify outgoing interface too
767 if (err == 0 && JoinMulticastGroup)
769 err = setsockopt(*sktPtr, IPPROTO_IP, IP_MULTICAST_IF, &((struct sockaddr_in*)intfAddr)->sin_addr, sizeof(struct in_addr));
770 if (err < 0) { err = errno; perror("setsockopt - IP_MULTICAST_IF"); }
773 // Per the mDNS spec, send unicast packets with TTL 255
774 if (err == 0)
776 err = setsockopt(*sktPtr, IPPROTO_IP, IP_TTL, &kIntTwoFiveFive, sizeof(kIntTwoFiveFive));
777 if (err < 0) { err = errno; perror("setsockopt - IP_TTL"); }
780 // and multicast packets with TTL 255 too
781 // There's some debate as to whether IP_MULTICAST_TTL is an int or a byte so we just try both.
782 if (err == 0)
784 err = setsockopt(*sktPtr, IPPROTO_IP, IP_MULTICAST_TTL, &kByteTwoFiveFive, sizeof(kByteTwoFiveFive));
785 if (err < 0 && errno == EINVAL)
786 err = setsockopt(*sktPtr, IPPROTO_IP, IP_MULTICAST_TTL, &kIntTwoFiveFive, sizeof(kIntTwoFiveFive));
787 if (err < 0) { err = errno; perror("setsockopt - IP_MULTICAST_TTL"); }
790 // And start listening for packets
791 if (err == 0)
793 bindAddr.sin_family = AF_INET;
794 bindAddr.sin_port = port.NotAnInteger;
795 bindAddr.sin_addr.s_addr = INADDR_ANY; // Want to receive multicasts AND unicasts on this socket
796 err = bind(*sktPtr, (struct sockaddr *) &bindAddr, sizeof(bindAddr));
797 if (err < 0) { err = errno; perror("bind"); fflush(stderr); }
799 } // endif (intfAddr->sa_family == AF_INET)
801 #if HAVE_IPV6
802 else if (intfAddr->sa_family == AF_INET6)
804 struct ipv6_mreq imr6;
805 struct sockaddr_in6 bindAddr6;
806 #if defined(IPV6_RECVPKTINFO)
807 if (err == 0)
809 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_RECVPKTINFO, &kOn, sizeof(kOn));
810 if (err < 0) { err = errno; perror("setsockopt - IPV6_RECVPKTINFO"); }
812 #elif defined(IPV6_PKTINFO)
813 if (err == 0)
815 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_PKTINFO, &kOn, sizeof(kOn));
816 if (err < 0) { err = errno; perror("setsockopt - IPV6_PKTINFO"); }
818 #else
819 #warning This platform has no way to get the destination interface information for IPv6 -- will only work for single-homed hosts
820 #endif
821 #if defined(IPV6_RECVHOPLIMIT)
822 if (err == 0)
824 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_RECVHOPLIMIT, &kOn, sizeof(kOn));
825 if (err < 0) { err = errno; perror("setsockopt - IPV6_RECVHOPLIMIT"); }
827 #elif defined(IPV6_HOPLIMIT)
828 if (err == 0)
830 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_HOPLIMIT, &kOn, sizeof(kOn));
831 if (err < 0) { err = errno; perror("setsockopt - IPV6_HOPLIMIT"); }
833 #endif
835 // Add multicast group membership on this interface
836 if (err == 0 && JoinMulticastGroup)
838 imr6.ipv6mr_multiaddr = *(const struct in6_addr*)&AllDNSLinkGroup_v6.ip.v6;
839 imr6.ipv6mr_interface = interfaceIndex;
840 //LogMsg("Joining %.16a on %d", &imr6.ipv6mr_multiaddr, imr6.ipv6mr_interface);
841 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_JOIN_GROUP, &imr6, sizeof(imr6));
842 if (err < 0)
844 err = errno;
845 verbosedebugf("IPV6_JOIN_GROUP %.16a on %d failed.\n", &imr6.ipv6mr_multiaddr, imr6.ipv6mr_interface);
846 perror("setsockopt - IPV6_JOIN_GROUP");
850 // Specify outgoing interface too
851 if (err == 0 && JoinMulticastGroup)
853 u_int multicast_if = interfaceIndex;
854 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_MULTICAST_IF, &multicast_if, sizeof(multicast_if));
855 if (err < 0) { err = errno; perror("setsockopt - IPV6_MULTICAST_IF"); }
858 // We want to receive only IPv6 packets on this socket.
859 // Without this option, we may get IPv4 addresses as mapped addresses.
860 if (err == 0)
862 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_V6ONLY, &kOn, sizeof(kOn));
863 if (err < 0) { err = errno; perror("setsockopt - IPV6_V6ONLY"); }
866 // Per the mDNS spec, send unicast packets with TTL 255
867 if (err == 0)
869 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &kIntTwoFiveFive, sizeof(kIntTwoFiveFive));
870 if (err < 0) { err = errno; perror("setsockopt - IPV6_UNICAST_HOPS"); }
873 // and multicast packets with TTL 255 too
874 // There's some debate as to whether IPV6_MULTICAST_HOPS is an int or a byte so we just try both.
875 if (err == 0)
877 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &kByteTwoFiveFive, sizeof(kByteTwoFiveFive));
878 if (err < 0 && errno == EINVAL)
879 err = setsockopt(*sktPtr, IPPROTO_IPV6, IPV6_MULTICAST_HOPS, &kIntTwoFiveFive, sizeof(kIntTwoFiveFive));
880 if (err < 0) { err = errno; perror("setsockopt - IPV6_MULTICAST_HOPS"); }
883 // And start listening for packets
884 if (err == 0)
886 mDNSPlatformMemZero(&bindAddr6, sizeof(bindAddr6));
887 #ifndef NOT_HAVE_SA_LEN
888 bindAddr6.sin6_len = sizeof(bindAddr6);
889 #endif
890 bindAddr6.sin6_family = AF_INET6;
891 bindAddr6.sin6_port = port.NotAnInteger;
892 bindAddr6.sin6_flowinfo = 0;
893 bindAddr6.sin6_addr = in6addr_any; // Want to receive multicasts AND unicasts on this socket
894 bindAddr6.sin6_scope_id = 0;
895 err = bind(*sktPtr, (struct sockaddr *) &bindAddr6, sizeof(bindAddr6));
896 if (err < 0) { err = errno; perror("bind"); fflush(stderr); }
898 } // endif (intfAddr->sa_family == AF_INET6)
899 #endif
901 // Set the socket to non-blocking.
902 if (err == 0)
904 err = fcntl(*sktPtr, F_GETFL, 0);
905 if (err < 0) err = errno;
906 else
908 err = fcntl(*sktPtr, F_SETFL, err | O_NONBLOCK);
909 if (err < 0) err = errno;
913 // Clean up
914 if (err != 0 && *sktPtr != -1) { assert(close(*sktPtr) == 0); *sktPtr = -1; }
915 assert((err == 0) == (*sktPtr != -1));
916 return err;
919 // Creates a PosixNetworkInterface for the interface whose IP address is
920 // intfAddr and whose name is intfName and registers it with mDNS core.
921 mDNSlocal int SetupOneInterface(mDNS *const m, struct sockaddr *intfAddr, struct sockaddr *intfMask, const char *intfName, int intfIndex)
923 int err = 0;
924 PosixNetworkInterface *intf;
925 PosixNetworkInterface *alias = NULL;
927 assert(m != NULL);
928 assert(intfAddr != NULL);
929 assert(intfName != NULL);
930 assert(intfMask != NULL);
932 // Allocate the interface structure itself.
933 intf = (PosixNetworkInterface*)malloc(sizeof(*intf));
934 if (intf == NULL) { assert(0); err = ENOMEM; }
936 // And make a copy of the intfName.
937 if (err == 0)
939 intf->intfName = strdup(intfName);
940 if (intf->intfName == NULL) { assert(0); err = ENOMEM; }
943 if (err == 0)
945 // Set up the fields required by the mDNS core.
946 SockAddrTomDNSAddr(intfAddr, &intf->coreIntf.ip, NULL);
947 SockAddrTomDNSAddr(intfMask, &intf->coreIntf.mask, NULL);
948 //LogMsg("SetupOneInterface: %#a %#a", &intf->coreIntf.ip, &intf->coreIntf.mask);
949 strncpy(intf->coreIntf.ifname, intfName, sizeof(intf->coreIntf.ifname));
950 intf->coreIntf.ifname[sizeof(intf->coreIntf.ifname)-1] = 0;
951 intf->coreIntf.Advertise = m->AdvertiseLocalAddresses;
952 intf->coreIntf.McastTxRx = mDNStrue;
954 // Set up the extra fields in PosixNetworkInterface.
955 assert(intf->intfName != NULL); // intf->intfName already set up above
956 intf->index = intfIndex;
957 intf->multicastSocket4 = -1;
958 #if HAVE_IPV6
959 intf->multicastSocket6 = -1;
960 #endif
961 alias = SearchForInterfaceByName(m, intf->intfName);
962 if (alias == NULL) alias = intf;
963 intf->coreIntf.InterfaceID = (mDNSInterfaceID)alias;
965 if (alias != intf)
966 debugf("SetupOneInterface: %s %#a is an alias of %#a", intfName, &intf->coreIntf.ip, &alias->coreIntf.ip);
969 // Set up the multicast socket
970 if (err == 0)
972 if (alias->multicastSocket4 == -1 && intfAddr->sa_family == AF_INET)
973 err = SetupSocket(intfAddr, MulticastDNSPort, intf->index, &alias->multicastSocket4);
974 #if HAVE_IPV6
975 else if (alias->multicastSocket6 == -1 && intfAddr->sa_family == AF_INET6)
976 err = SetupSocket(intfAddr, MulticastDNSPort, intf->index, &alias->multicastSocket6);
977 #endif
980 // The interface is all ready to go, let's register it with the mDNS core.
981 if (err == 0)
982 err = mDNS_RegisterInterface(m, &intf->coreIntf, mDNSfalse);
984 // Clean up.
985 if (err == 0)
987 num_registered_interfaces++;
988 debugf("SetupOneInterface: %s %#a Registered", intf->intfName, &intf->coreIntf.ip);
989 if (gMDNSPlatformPosixVerboseLevel > 0)
990 fprintf(stderr, "Registered interface %s\n", intf->intfName);
992 else
994 // Use intfName instead of intf->intfName in the next line to avoid dereferencing NULL.
995 debugf("SetupOneInterface: %s %#a failed to register %d", intfName, &intf->coreIntf.ip, err);
996 if (intf) { FreePosixNetworkInterface(intf); intf = NULL; }
999 assert((err == 0) == (intf != NULL));
1001 return err;
1004 // Call get_ifi_info() to obtain a list of active interfaces and call SetupOneInterface() on each one.
1005 mDNSlocal int SetupInterfaceList(mDNS *const m)
1007 mDNSBool foundav4 = mDNSfalse;
1008 int err = 0;
1009 struct ifi_info *intfList = get_ifi_info(AF_INET, mDNStrue);
1010 struct ifi_info *firstLoopback = NULL;
1012 assert(m != NULL);
1013 debugf("SetupInterfaceList");
1015 if (intfList == NULL) err = ENOENT;
1017 #if HAVE_IPV6
1018 if (err == 0) /* Link the IPv6 list to the end of the IPv4 list */
1020 struct ifi_info **p = &intfList;
1021 while (*p) p = &(*p)->ifi_next;
1022 *p = get_ifi_info(AF_INET6, mDNStrue);
1024 #endif
1026 if (err == 0)
1028 struct ifi_info *i = intfList;
1029 while (i)
1031 if ( ((i->ifi_addr->sa_family == AF_INET)
1032 #if HAVE_IPV6
1033 || (i->ifi_addr->sa_family == AF_INET6)
1034 #endif
1035 ) && (i->ifi_flags & IFF_UP) && !(i->ifi_flags & IFF_POINTOPOINT))
1037 if (i->ifi_flags & IFF_LOOPBACK)
1039 if (firstLoopback == NULL)
1040 firstLoopback = i;
1042 else
1044 if (SetupOneInterface(m, i->ifi_addr, i->ifi_netmask, i->ifi_name, i->ifi_index) == 0)
1045 if (i->ifi_addr->sa_family == AF_INET)
1046 foundav4 = mDNStrue;
1049 i = i->ifi_next;
1052 // If we found no normal interfaces but we did find a loopback interface, register the
1053 // loopback interface. This allows self-discovery if no interfaces are configured.
1054 // Temporary workaround: Multicast loopback on IPv6 interfaces appears not to work.
1055 // In the interim, we skip loopback interface only if we found at least one v4 interface to use
1056 // if ((m->HostInterfaces == NULL) && (firstLoopback != NULL))
1057 if (!foundav4 && firstLoopback)
1058 (void) SetupOneInterface(m, firstLoopback->ifi_addr, firstLoopback->ifi_netmask, firstLoopback->ifi_name, firstLoopback->ifi_index);
1061 // Clean up.
1062 if (intfList != NULL) free_ifi_info(intfList);
1063 return err;
1066 #if USES_NETLINK
1068 // See <http://www.faqs.org/rfcs/rfc3549.html> for a description of NetLink
1070 // Open a socket that will receive interface change notifications
1071 mDNSlocal mStatus OpenIfNotifySocket(int *pFD)
1073 mStatus err = mStatus_NoError;
1074 struct sockaddr_nl snl;
1075 int sock;
1076 int ret;
1078 sock = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1079 if (sock < 0)
1080 return errno;
1082 // Configure read to be non-blocking because inbound msg size is not known in advance
1083 (void) fcntl(sock, F_SETFL, O_NONBLOCK);
1085 /* Subscribe the socket to Link & IP addr notifications. */
1086 mDNSPlatformMemZero(&snl, sizeof snl);
1087 snl.nl_family = AF_NETLINK;
1088 snl.nl_groups = RTMGRP_LINK | RTMGRP_IPV4_IFADDR;
1089 ret = bind(sock, (struct sockaddr *) &snl, sizeof snl);
1090 if (0 == ret)
1091 *pFD = sock;
1092 else
1093 err = errno;
1095 return err;
1098 #if MDNS_DEBUGMSGS
1099 mDNSlocal void PrintNetLinkMsg(const struct nlmsghdr *pNLMsg)
1101 const char *kNLMsgTypes[] = { "", "NLMSG_NOOP", "NLMSG_ERROR", "NLMSG_DONE", "NLMSG_OVERRUN" };
1102 const char *kNLRtMsgTypes[] = { "RTM_NEWLINK", "RTM_DELLINK", "RTM_GETLINK", "RTM_NEWADDR", "RTM_DELADDR", "RTM_GETADDR" };
1104 printf("nlmsghdr len=%d, type=%s, flags=0x%x\n", pNLMsg->nlmsg_len,
1105 pNLMsg->nlmsg_type < RTM_BASE ? kNLMsgTypes[pNLMsg->nlmsg_type] : kNLRtMsgTypes[pNLMsg->nlmsg_type - RTM_BASE],
1106 pNLMsg->nlmsg_flags);
1108 if (RTM_NEWLINK <= pNLMsg->nlmsg_type && pNLMsg->nlmsg_type <= RTM_GETLINK)
1110 struct ifinfomsg *pIfInfo = (struct ifinfomsg*) NLMSG_DATA(pNLMsg);
1111 printf("ifinfomsg family=%d, type=%d, index=%d, flags=0x%x, change=0x%x\n", pIfInfo->ifi_family,
1112 pIfInfo->ifi_type, pIfInfo->ifi_index, pIfInfo->ifi_flags, pIfInfo->ifi_change);
1115 else if (RTM_NEWADDR <= pNLMsg->nlmsg_type && pNLMsg->nlmsg_type <= RTM_GETADDR)
1117 struct ifaddrmsg *pIfAddr = (struct ifaddrmsg*) NLMSG_DATA(pNLMsg);
1118 printf("ifaddrmsg family=%d, index=%d, flags=0x%x\n", pIfAddr->ifa_family,
1119 pIfAddr->ifa_index, pIfAddr->ifa_flags);
1121 printf("\n");
1123 #endif
1125 mDNSlocal mDNSu32 ProcessRoutingNotification(int sd)
1126 // Read through the messages on sd and if any indicate that any interface records should
1127 // be torn down and rebuilt, return affected indices as a bitmask. Otherwise return 0.
1129 ssize_t readCount;
1130 char buff[4096];
1131 struct nlmsghdr *pNLMsg = (struct nlmsghdr*) buff;
1132 mDNSu32 result = 0;
1134 // The structure here is more complex than it really ought to be because,
1135 // unfortunately, there's no good way to size a buffer in advance large
1136 // enough to hold all pending data and so avoid message fragmentation.
1137 // (Note that FIONREAD is not supported on AF_NETLINK.)
1139 readCount = read(sd, buff, sizeof buff);
1140 while (1)
1142 // Make sure we've got an entire nlmsghdr in the buffer, and payload, too.
1143 // If not, discard already-processed messages in buffer and read more data.
1144 if (((char*) &pNLMsg[1] > (buff + readCount)) || // i.e. *pNLMsg extends off end of buffer
1145 ((char*) pNLMsg + pNLMsg->nlmsg_len > (buff + readCount)))
1147 if (buff < (char*) pNLMsg) // we have space to shuffle
1149 // discard processed data
1150 readCount -= ((char*) pNLMsg - buff);
1151 memmove(buff, pNLMsg, readCount);
1152 pNLMsg = (struct nlmsghdr*) buff;
1154 // read more data
1155 readCount += read(sd, buff + readCount, sizeof buff - readCount);
1156 continue; // spin around and revalidate with new readCount
1158 else
1159 break; // Otherwise message does not fit in buffer
1162 #if MDNS_DEBUGMSGS
1163 PrintNetLinkMsg(pNLMsg);
1164 #endif
1166 // Process the NetLink message
1167 if (pNLMsg->nlmsg_type == RTM_GETLINK || pNLMsg->nlmsg_type == RTM_NEWLINK)
1168 result |= 1 << ((struct ifinfomsg*) NLMSG_DATA(pNLMsg))->ifi_index;
1169 else if (pNLMsg->nlmsg_type == RTM_DELADDR || pNLMsg->nlmsg_type == RTM_NEWADDR)
1170 result |= 1 << ((struct ifaddrmsg*) NLMSG_DATA(pNLMsg))->ifa_index;
1172 // Advance pNLMsg to the next message in the buffer
1173 if ((pNLMsg->nlmsg_flags & NLM_F_MULTI) != 0 && pNLMsg->nlmsg_type != NLMSG_DONE)
1175 ssize_t len = readCount - ((char*)pNLMsg - buff);
1176 pNLMsg = NLMSG_NEXT(pNLMsg, len);
1178 else
1179 break; // all done!
1182 return result;
1185 #else // USES_NETLINK
1187 // Open a socket that will receive interface change notifications
1188 mDNSlocal mStatus OpenIfNotifySocket(int *pFD)
1190 *pFD = socket(AF_ROUTE, SOCK_RAW, 0);
1192 if (*pFD < 0)
1193 return mStatus_UnknownErr;
1195 // Configure read to be non-blocking because inbound msg size is not known in advance
1196 (void) fcntl(*pFD, F_SETFL, O_NONBLOCK);
1198 return mStatus_NoError;
1201 #if MDNS_DEBUGMSGS
1202 mDNSlocal void PrintRoutingSocketMsg(const struct ifa_msghdr *pRSMsg)
1204 const char *kRSMsgTypes[] = { "", "RTM_ADD", "RTM_DELETE", "RTM_CHANGE", "RTM_GET", "RTM_LOSING",
1205 "RTM_REDIRECT", "RTM_MISS", "RTM_LOCK", "RTM_OLDADD", "RTM_OLDDEL", "RTM_RESOLVE",
1206 "RTM_NEWADDR", "RTM_DELADDR", "RTM_IFINFO", "RTM_NEWMADDR", "RTM_DELMADDR" };
1208 int index = pRSMsg->ifam_type == RTM_IFINFO ? ((struct if_msghdr*) pRSMsg)->ifm_index : pRSMsg->ifam_index;
1210 printf("ifa_msghdr len=%d, type=%s, index=%d\n", pRSMsg->ifam_msglen, kRSMsgTypes[pRSMsg->ifam_type], index);
1212 #endif
1214 mDNSlocal mDNSu32 ProcessRoutingNotification(int sd)
1215 // Read through the messages on sd and if any indicate that any interface records should
1216 // be torn down and rebuilt, return affected indices as a bitmask. Otherwise return 0.
1218 ssize_t readCount;
1219 char buff[4096];
1220 struct ifa_msghdr *pRSMsg = (struct ifa_msghdr*) buff;
1221 mDNSu32 result = 0;
1223 readCount = read(sd, buff, sizeof buff);
1224 if (readCount < (ssize_t) sizeof(struct ifa_msghdr))
1225 return mStatus_UnsupportedErr; // cannot decipher message
1227 #if MDNS_DEBUGMSGS
1228 PrintRoutingSocketMsg(pRSMsg);
1229 #endif
1231 // Process the message
1232 if (pRSMsg->ifam_type == RTM_NEWADDR || pRSMsg->ifam_type == RTM_DELADDR ||
1233 pRSMsg->ifam_type == RTM_IFINFO)
1235 if (pRSMsg->ifam_type == RTM_IFINFO)
1236 result |= 1 << ((struct if_msghdr*) pRSMsg)->ifm_index;
1237 else
1238 result |= 1 << pRSMsg->ifam_index;
1241 return result;
1244 #endif // USES_NETLINK
1246 // Called when data appears on interface change notification socket
1247 mDNSlocal void InterfaceChangeCallback(int fd, short filter, void *context)
1249 IfChangeRec *pChgRec = (IfChangeRec*) context;
1250 fd_set readFDs;
1251 mDNSu32 changedInterfaces = 0;
1252 struct timeval zeroTimeout = { 0, 0 };
1254 (void)fd; // Unused
1255 (void)filter; // Unused
1257 FD_ZERO(&readFDs);
1258 FD_SET(pChgRec->NotifySD, &readFDs);
1262 changedInterfaces |= ProcessRoutingNotification(pChgRec->NotifySD);
1264 while (0 < select(pChgRec->NotifySD + 1, &readFDs, (fd_set*) NULL, (fd_set*) NULL, &zeroTimeout));
1266 // Currently we rebuild the entire interface list whenever any interface change is
1267 // detected. If this ever proves to be a performance issue in a multi-homed
1268 // configuration, more care should be paid to changedInterfaces.
1269 if (changedInterfaces)
1270 mDNSPlatformPosixRefreshInterfaceList(pChgRec->mDNS);
1273 // Register with either a Routing Socket or RtNetLink to listen for interface changes.
1274 mDNSlocal mStatus WatchForInterfaceChange(mDNS *const m)
1276 mStatus err;
1277 IfChangeRec *pChgRec;
1279 pChgRec = (IfChangeRec*) mDNSPlatformMemAllocate(sizeof *pChgRec);
1280 if (pChgRec == NULL)
1281 return mStatus_NoMemoryErr;
1283 pChgRec->mDNS = m;
1284 err = OpenIfNotifySocket(&pChgRec->NotifySD);
1285 if (err == 0)
1286 err = mDNSPosixAddFDToEventLoop(pChgRec->NotifySD, InterfaceChangeCallback, pChgRec);
1288 return err;
1291 // Test to see if we're the first client running on UDP port 5353, by trying to bind to 5353 without using SO_REUSEPORT.
1292 // If we fail, someone else got here first. That's not a big problem; we can share the port for multicast responses --
1293 // we just need to be aware that we shouldn't expect to successfully receive unicast UDP responses.
1294 mDNSlocal mDNSBool mDNSPlatformInit_CanReceiveUnicast(void)
1296 int err;
1297 int s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
1298 struct sockaddr_in s5353;
1299 s5353.sin_family = AF_INET;
1300 s5353.sin_port = MulticastDNSPort.NotAnInteger;
1301 s5353.sin_addr.s_addr = 0;
1302 err = bind(s, (struct sockaddr *)&s5353, sizeof(s5353));
1303 close(s);
1304 if (err) debugf("No unicast UDP responses");
1305 else debugf("Unicast UDP responses okay");
1306 return(err == 0);
1309 // mDNS core calls this routine to initialise the platform-specific data.
1310 mDNSexport mStatus mDNSPlatformInit(mDNS *const m)
1312 int err = 0;
1313 struct sockaddr sa;
1314 assert(m != NULL);
1316 if (mDNSPlatformInit_CanReceiveUnicast()) m->CanReceiveUnicastOn5353 = mDNStrue;
1318 // Tell mDNS core the names of this machine.
1320 // Set up the nice label
1321 m->nicelabel.c[0] = 0;
1322 GetUserSpecifiedFriendlyComputerName(&m->nicelabel);
1323 if (m->nicelabel.c[0] == 0) MakeDomainLabelFromLiteralString(&m->nicelabel, "Computer");
1325 // Set up the RFC 1034-compliant label
1326 m->hostlabel.c[0] = 0;
1327 GetUserSpecifiedRFC1034ComputerName(&m->hostlabel);
1328 if (m->hostlabel.c[0] == 0) MakeDomainLabelFromLiteralString(&m->hostlabel, "Computer");
1330 mDNS_SetFQDN(m);
1332 sa.sa_family = AF_INET;
1333 m->p->unicastSocket4 = -1;
1334 if (err == mStatus_NoError) err = SetupSocket(&sa, zeroIPPort, 0, &m->p->unicastSocket4);
1335 #if HAVE_IPV6
1336 sa.sa_family = AF_INET6;
1337 m->p->unicastSocket6 = -1;
1338 if (err == mStatus_NoError) err = SetupSocket(&sa, zeroIPPort, 0, &m->p->unicastSocket6);
1339 #endif
1341 // Tell mDNS core about the network interfaces on this machine.
1342 if (err == mStatus_NoError) err = SetupInterfaceList(m);
1344 // Tell mDNS core about DNS Servers
1345 mDNS_Lock(m);
1346 if (err == mStatus_NoError) ParseDNSServers(m, uDNS_SERVERS_FILE);
1347 mDNS_Unlock(m);
1349 if (err == mStatus_NoError)
1351 err = WatchForInterfaceChange(m);
1352 // Failure to observe interface changes is non-fatal.
1353 if (err != mStatus_NoError)
1355 fprintf(stderr, "mDNS(%d) WARNING: Unable to detect interface changes (%d).\n", getpid(), err);
1356 err = mStatus_NoError;
1360 // We don't do asynchronous initialization on the Posix platform, so by the time
1361 // we get here the setup will already have succeeded or failed. If it succeeded,
1362 // we should just call mDNSCoreInitComplete() immediately.
1363 if (err == mStatus_NoError)
1364 mDNSCoreInitComplete(m, mStatus_NoError);
1366 return PosixErrorToStatus(err);
1369 // mDNS core calls this routine to clean up the platform-specific data.
1370 // In our case all we need to do is to tear down every network interface.
1371 mDNSexport void mDNSPlatformClose(mDNS *const m)
1373 assert(m != NULL);
1374 ClearInterfaceList(m);
1375 if (m->p->unicastSocket4 != -1) assert(close(m->p->unicastSocket4) == 0);
1376 #if HAVE_IPV6
1377 if (m->p->unicastSocket6 != -1) assert(close(m->p->unicastSocket6) == 0);
1378 #endif
1381 mDNSexport mStatus mDNSPlatformPosixRefreshInterfaceList(mDNS *const m)
1383 int err;
1384 ClearInterfaceList(m);
1385 err = SetupInterfaceList(m);
1386 return PosixErrorToStatus(err);
1389 #if COMPILER_LIKES_PRAGMA_MARK
1390 #pragma mark ***** Locking
1391 #endif
1393 // On the Posix platform, locking is a no-op because we only ever enter
1394 // mDNS core on the main thread.
1396 // mDNS core calls this routine when it wants to prevent
1397 // the platform from reentering mDNS core code.
1398 mDNSexport void mDNSPlatformLock (const mDNS *const m)
1400 (void) m; // Unused
1403 // mDNS core calls this routine when it release the lock taken by
1404 // mDNSPlatformLock and allow the platform to reenter mDNS core code.
1405 mDNSexport void mDNSPlatformUnlock (const mDNS *const m)
1407 (void) m; // Unused
1410 #if COMPILER_LIKES_PRAGMA_MARK
1411 #pragma mark ***** Strings
1412 #endif
1414 // mDNS core calls this routine to copy C strings.
1415 // On the Posix platform this maps directly to the ANSI C strcpy.
1416 mDNSexport void mDNSPlatformStrCopy(void *dst, const void *src)
1418 strcpy((char *)dst, (char *)src);
1421 // mDNS core calls this routine to get the length of a C string.
1422 // On the Posix platform this maps directly to the ANSI C strlen.
1423 mDNSexport mDNSu32 mDNSPlatformStrLen (const void *src)
1425 return strlen((char*)src);
1428 // mDNS core calls this routine to copy memory.
1429 // On the Posix platform this maps directly to the ANSI C memcpy.
1430 mDNSexport void mDNSPlatformMemCopy(void *dst, const void *src, mDNSu32 len)
1432 memcpy(dst, src, len);
1435 // mDNS core calls this routine to test whether blocks of memory are byte-for-byte
1436 // identical. On the Posix platform this is a simple wrapper around ANSI C memcmp.
1437 mDNSexport mDNSBool mDNSPlatformMemSame(const void *dst, const void *src, mDNSu32 len)
1439 return memcmp(dst, src, len) == 0;
1442 // mDNS core calls this routine to clear blocks of memory.
1443 // On the Posix platform this is a simple wrapper around ANSI C memset.
1444 mDNSexport void mDNSPlatformMemZero(void *dst, mDNSu32 len)
1446 memset(dst, 0, len);
1449 mDNSexport void * mDNSPlatformMemAllocate(mDNSu32 len) { return(malloc(len)); }
1450 mDNSexport void mDNSPlatformMemFree (void *mem) { free(mem); }
1452 mDNSexport mDNSu32 mDNSPlatformRandomSeed(void)
1454 struct timeval tv;
1455 gettimeofday(&tv, NULL);
1456 return(tv.tv_usec);
1459 mDNSexport mDNSs32 mDNSPlatformOneSecond = 1024;
1461 mDNSexport mStatus mDNSPlatformTimeInit(void)
1463 // No special setup is required on Posix -- we just use gettimeofday();
1464 // This is not really safe, because gettimeofday can go backwards if the user manually changes the date or time
1465 // We should find a better way to do this
1466 return(mStatus_NoError);
1469 mDNSexport mDNSs32 mDNSPlatformRawTime()
1471 struct timeval tv;
1472 gettimeofday(&tv, NULL);
1473 // tv.tv_sec is seconds since 1st January 1970 (GMT, with no adjustment for daylight savings time)
1474 // tv.tv_usec is microseconds since the start of this second (i.e. values 0 to 999999)
1475 // We use the lower 22 bits of tv.tv_sec for the top 22 bits of our result
1476 // and we multiply tv.tv_usec by 16 / 15625 to get a value in the range 0-1023 to go in the bottom 10 bits.
1477 // This gives us a proper modular (cyclic) counter that has a resolution of roughly 1ms (actually 1/1024 second)
1478 // and correctly cycles every 2^22 seconds (4194304 seconds = approx 48 days).
1479 return((tv.tv_sec << 10) | (tv.tv_usec * 16 / 15625));
1482 mDNSexport mDNSs32 mDNSPlatformUTC(void)
1484 return time(NULL);
1487 mDNSlocal void mDNSPosixAddToFDSet(int *nfds, fd_set *readfds, int s)
1489 if (*nfds < s + 1) *nfds = s + 1;
1490 FD_SET(s, readfds);
1493 mDNSexport void mDNSPosixGetFDSet(mDNS *m, int *nfds, fd_set *readfds, struct timeval *timeout)
1495 mDNSs32 ticks;
1496 struct timeval interval;
1498 // 1. Call mDNS_Execute() to let mDNSCore do what it needs to do
1499 mDNSs32 nextevent = mDNS_Execute(m);
1501 // 2. Build our list of active file descriptors
1502 PosixNetworkInterface *info = (PosixNetworkInterface *)(m->HostInterfaces);
1503 if (m->p->unicastSocket4 != -1) mDNSPosixAddToFDSet(nfds, readfds, m->p->unicastSocket4);
1504 #if HAVE_IPV6
1505 if (m->p->unicastSocket6 != -1) mDNSPosixAddToFDSet(nfds, readfds, m->p->unicastSocket6);
1506 #endif
1507 while (info)
1509 if (info->multicastSocket4 != -1) mDNSPosixAddToFDSet(nfds, readfds, info->multicastSocket4);
1510 #if HAVE_IPV6
1511 if (info->multicastSocket6 != -1) mDNSPosixAddToFDSet(nfds, readfds, info->multicastSocket6);
1512 #endif
1513 info = (PosixNetworkInterface *)(info->coreIntf.next);
1516 // 3. Calculate the time remaining to the next scheduled event (in struct timeval format)
1517 ticks = nextevent - mDNS_TimeNow(m);
1518 if (ticks < 1) ticks = 1;
1519 interval.tv_sec = ticks >> 10; // The high 22 bits are seconds
1520 interval.tv_usec = ((ticks & 0x3FF) * 15625) / 16; // The low 10 bits are 1024ths
1522 // 4. If client's proposed timeout is more than what we want, then reduce it
1523 if (timeout->tv_sec > interval.tv_sec ||
1524 (timeout->tv_sec == interval.tv_sec && timeout->tv_usec > interval.tv_usec))
1525 *timeout = interval;
1528 mDNSexport void mDNSPosixProcessFDSet(mDNS *const m, fd_set *readfds)
1530 PosixNetworkInterface *info;
1531 assert(m != NULL);
1532 assert(readfds != NULL);
1533 info = (PosixNetworkInterface *)(m->HostInterfaces);
1535 if (m->p->unicastSocket4 != -1 && FD_ISSET(m->p->unicastSocket4, readfds))
1537 FD_CLR(m->p->unicastSocket4, readfds);
1538 SocketDataReady(m, NULL, m->p->unicastSocket4);
1540 #if HAVE_IPV6
1541 if (m->p->unicastSocket6 != -1 && FD_ISSET(m->p->unicastSocket6, readfds))
1543 FD_CLR(m->p->unicastSocket6, readfds);
1544 SocketDataReady(m, NULL, m->p->unicastSocket6);
1546 #endif
1548 while (info)
1550 if (info->multicastSocket4 != -1 && FD_ISSET(info->multicastSocket4, readfds))
1552 FD_CLR(info->multicastSocket4, readfds);
1553 SocketDataReady(m, info, info->multicastSocket4);
1555 #if HAVE_IPV6
1556 if (info->multicastSocket6 != -1 && FD_ISSET(info->multicastSocket6, readfds))
1558 FD_CLR(info->multicastSocket6, readfds);
1559 SocketDataReady(m, info, info->multicastSocket6);
1561 #endif
1562 info = (PosixNetworkInterface *)(info->coreIntf.next);
1566 // update gMaxFD
1567 mDNSlocal void DetermineMaxEventFD(void)
1569 PosixEventSource *iSource;
1571 gMaxFD = 0;
1572 for (iSource=(PosixEventSource*)gEventSources.Head; iSource; iSource = iSource->Next)
1573 if (gMaxFD < iSource->fd)
1574 gMaxFD = iSource->fd;
1577 // Add a file descriptor to the set that mDNSPosixRunEventLoopOnce() listens to.
1578 mStatus mDNSPosixAddFDToEventLoop(int fd, mDNSPosixEventCallback callback, void *context)
1580 PosixEventSource *newSource;
1582 if (gEventSources.LinkOffset == 0)
1583 InitLinkedList(&gEventSources, offsetof(PosixEventSource, Next));
1585 if (fd >= (int) FD_SETSIZE || fd < 0)
1586 return mStatus_UnsupportedErr;
1587 if (callback == NULL)
1588 return mStatus_BadParamErr;
1590 newSource = (PosixEventSource*) malloc(sizeof *newSource);
1591 if (NULL == newSource)
1592 return mStatus_NoMemoryErr;
1594 newSource->Callback = callback;
1595 newSource->Context = context;
1596 newSource->fd = fd;
1598 AddToTail(&gEventSources, newSource);
1599 FD_SET(fd, &gEventFDs);
1601 DetermineMaxEventFD();
1603 return mStatus_NoError;
1606 // Remove a file descriptor from the set that mDNSPosixRunEventLoopOnce() listens to.
1607 mStatus mDNSPosixRemoveFDFromEventLoop(int fd)
1609 PosixEventSource *iSource;
1611 for (iSource=(PosixEventSource*)gEventSources.Head; iSource; iSource = iSource->Next)
1613 if (fd == iSource->fd)
1615 FD_CLR(fd, &gEventFDs);
1616 RemoveFromList(&gEventSources, iSource);
1617 free(iSource);
1618 DetermineMaxEventFD();
1619 return mStatus_NoError;
1622 return mStatus_NoSuchNameErr;
1625 // Simply note the received signal in gEventSignals.
1626 mDNSlocal void NoteSignal(int signum)
1628 sigaddset(&gEventSignals, signum);
1631 // Tell the event package to listen for signal and report it in mDNSPosixRunEventLoopOnce().
1632 mStatus mDNSPosixListenForSignalInEventLoop(int signum)
1634 struct sigaction action;
1635 mStatus err;
1637 mDNSPlatformMemZero(&action, sizeof action); // more portable than member-wise assignment
1638 action.sa_handler = NoteSignal;
1639 err = sigaction(signum, &action, (struct sigaction*) NULL);
1641 sigaddset(&gEventSignalSet, signum);
1643 return err;
1646 // Tell the event package to stop listening for signal in mDNSPosixRunEventLoopOnce().
1647 mStatus mDNSPosixIgnoreSignalInEventLoop(int signum)
1649 struct sigaction action;
1650 mStatus err;
1652 mDNSPlatformMemZero(&action, sizeof action); // more portable than member-wise assignment
1653 action.sa_handler = SIG_DFL;
1654 err = sigaction(signum, &action, (struct sigaction*) NULL);
1656 sigdelset(&gEventSignalSet, signum);
1658 return err;
1661 // Do a single pass through the attendent event sources and dispatch any found to their callbacks.
1662 // Return as soon as internal timeout expires, or a signal we're listening for is received.
1663 mStatus mDNSPosixRunEventLoopOnce(mDNS *m, const struct timeval *pTimeout,
1664 sigset_t *pSignalsReceived, mDNSBool *pDataDispatched)
1666 fd_set listenFDs = gEventFDs;
1667 int fdMax = 0, numReady;
1668 struct timeval timeout = *pTimeout;
1670 // Include the sockets that are listening to the wire in our select() set
1671 mDNSPosixGetFDSet(m, &fdMax, &listenFDs, &timeout); // timeout may get modified
1672 if (fdMax < gMaxFD)
1673 fdMax = gMaxFD;
1675 numReady = select(fdMax + 1, &listenFDs, (fd_set*) NULL, (fd_set*) NULL, &timeout);
1677 // If any data appeared, invoke its callback
1678 if (numReady > 0)
1680 PosixEventSource *iSource;
1682 (void) mDNSPosixProcessFDSet(m, &listenFDs); // call this first to process wire data for clients
1684 for (iSource=(PosixEventSource*)gEventSources.Head; iSource; iSource = iSource->Next)
1686 if (FD_ISSET(iSource->fd, &listenFDs))
1688 iSource->Callback(iSource->fd, 0, iSource->Context);
1689 break; // in case callback removed elements from gEventSources
1692 *pDataDispatched = mDNStrue;
1694 else
1695 *pDataDispatched = mDNSfalse;
1697 (void) sigprocmask(SIG_BLOCK, &gEventSignalSet, (sigset_t*) NULL);
1698 *pSignalsReceived = gEventSignals;
1699 sigemptyset(&gEventSignals);
1700 (void) sigprocmask(SIG_UNBLOCK, &gEventSignalSet, (sigset_t*) NULL);
1702 return mStatus_NoError;