]> git.saurik.com Git - apple/mdnsresponder.git/blob - mDNSCore/mDNS.c
mDNSResponder-765.50.9.tar.gz
[apple/mdnsresponder.git] / mDNSCore / mDNS.c
1 /* -*- Mode: C; tab-width: 4 -*-
2 *
3 * Copyright (c) 2002-2015 Apple Inc. All rights reserved.
4 *
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
10 *
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.
16 *
17 * This code is completely 100% portable C. It does not depend on any external header files
18 * from outside the mDNS project -- all the types it expects to find are defined right here.
19 *
20 * The previous point is very important: This file does not depend on any external
21 * header files. It should compile on *any* platform that has a C compiler, without
22 * making *any* assumptions about availability of so-called "standard" C functions,
23 * routines, or types (which may or may not be present on any given platform).
24 */
25
26 #include "DNSCommon.h" // Defines general DNS utility routines
27 #include "uDNS.h" // Defines entry points into unicast-specific routines
28 #include "nsec.h"
29 #include "dnssec.h"
30 #include "anonymous.h"
31
32 // Disable certain benign warnings with Microsoft compilers
33 #if (defined(_MSC_VER))
34 // Disable "conditional expression is constant" warning for debug macros.
35 // Otherwise, this generates warnings for the perfectly natural construct "while(1)"
36 // If someone knows a variant way of writing "while(1)" that doesn't generate warning messages, please let us know
37 #pragma warning(disable:4127)
38
39 // Disable "assignment within conditional expression".
40 // Other compilers understand the convention that if you place the assignment expression within an extra pair
41 // of parentheses, this signals to the compiler that you really intended an assignment and no warning is necessary.
42 // The Microsoft compiler doesn't understand this convention, so in the absense of any other way to signal
43 // to the compiler that the assignment is intentional, we have to just turn this warning off completely.
44 #pragma warning(disable:4706)
45 #endif
46
47 #include "dns_sd.h" // for kDNSServiceFlags* definitions
48
49 #if APPLE_OSX_mDNSResponder
50 #include <WebFilterDNS/WebFilterDNS.h>
51
52 // Delay in seconds before disabling multicast after there are no active queries or registrations.
53 #define BONJOUR_DISABLE_DELAY 60
54
55 #if !NO_WCF
56 WCFConnection *WCFConnectionNew(void) __attribute__((weak_import));
57 void WCFConnectionDealloc(WCFConnection* c) __attribute__((weak_import));
58
59 // Do we really need to define a macro for "if"?
60 #define CHECK_WCF_FUNCTION(X) if (X)
61 #endif // ! NO_WCF
62
63 #else
64
65 #define NO_WCF 1
66 #endif // APPLE_OSX_mDNSResponder
67
68 #if TARGET_OS_EMBEDDED
69 #include "Metrics.h"
70 #endif
71
72 // Forward declarations
73 mDNSlocal void BeginSleepProcessing(mDNS *const m);
74 mDNSlocal void RetrySPSRegistrations(mDNS *const m);
75 mDNSlocal void SendWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *EthAddr, mDNSOpaque48 *password, mDNSBool unicastOnly);
76 mDNSlocal mDNSBool CacheRecordRmvEventsForQuestion(mDNS *const m, DNSQuestion *q);
77 mDNSlocal mDNSBool LocalRecordRmvEventsForQuestion(mDNS *const m, DNSQuestion *q);
78 mDNSlocal void mDNS_PurgeForQuestion(mDNS *const m, DNSQuestion *q);
79 mDNSlocal void CheckForDNSSECRecords(mDNS *const m, DNSQuestion *q);
80 mDNSlocal void mDNS_SendKeepalives(mDNS *const m);
81 mDNSlocal void mDNS_ExtractKeepaliveInfo(AuthRecord *ar, mDNSu32 *timeout, mDNSAddr *laddr, mDNSAddr *raddr, mDNSEthAddr *eth,
82 mDNSu32 *seq, mDNSu32 *ack, mDNSIPPort *lport, mDNSIPPort *rport, mDNSu16 *win);
83
84 mDNSlocal void AdvertiseAllInterfaceRecords(mDNS *const m);
85 mDNSlocal void DeadvertiseAllInterfaceRecords(mDNS *const m);
86 mDNSlocal void FreeNSECRecords(mDNS *const m, CacheRecord *NSECRecords);
87 mDNSlocal void mDNSParseNSEC3Records(mDNS *const m, const DNSMessage *const response, const mDNSu8 *end,
88 const mDNSInterfaceID InterfaceID, CacheRecord **NSEC3Records);
89 mDNSlocal mDNSu8 *GetValueForMACAddr(mDNSu8 *ptr, mDNSu8 *limit, mDNSEthAddr *eth);
90
91
92 // ***************************************************************************
93 #if COMPILER_LIKES_PRAGMA_MARK
94 #pragma mark - Program Constants
95 #endif
96
97 // To Turn OFF mDNS_Tracer set MDNS_TRACER to 0 or undef it
98 #define MDNS_TRACER 1
99
100 #define NO_HINFO 1
101
102 // Any records bigger than this are considered 'large' records
103 #define SmallRecordLimit 1024
104
105 #define kMaxUpdateCredits 10
106 #define kUpdateCreditRefreshInterval (mDNSPlatformOneSecond * 6)
107
108 // define special NR_AnswerTo values
109 #define NR_AnswerMulticast (mDNSu8*)~0
110 #define NR_AnswerUnicast (mDNSu8*)~1
111
112 // The code (see SendQueries() and BuildQuestion()) needs to have the
113 // RequestUnicast value set to a value one greater than the number of times you want the query
114 // sent with the "request unicast response" (QU) bit set.
115 #define SET_QU_IN_FIRST_QUERY 2
116 #define kDefaultRequestUnicastCount SET_QU_IN_FIRST_QUERY
117
118 // The time needed to offload records to a sleep proxy after powerd sends the kIOMessageSystemWillSleep notification
119 #define DARK_WAKE_DELAY_SLEEP 5
120 #define kDarkWakeDelaySleep (mDNSPlatformOneSecond * DARK_WAKE_DELAY_SLEEP)
121
122 // The maximum number of times we delay probing to prevent spurious conflicts due to stale packets
123 #define MAX_CONFLICT_PROCESSING_DELAYS 3
124
125 // RFC 6762 defines Passive Observation Of Failures (POOF)
126 //
127 // A host observes the multicast queries issued by the other hosts on
128 // the network. One of the major benefits of also sending responses
129 // using multicast is that it allows all hosts to see the responses
130 // (or lack thereof) to those queries.
131 //
132 // If a host sees queries, for which a record in its cache would be
133 // expected to be given as an answer in a multicast response, but no
134 // such answer is seen, then the host may take this as an indication
135 // that the record may no longer be valid.
136 //
137 // After seeing two or more of these queries, and seeing no multicast
138 // response containing the expected answer within ten seconds, then even
139 // though its TTL may indicate that it is not yet due to expire, that
140 // record SHOULD be flushed from the cache.
141 //
142 // <https://tools.ietf.org/html/rfc6762#section-10.5>
143
144 #define POOF_ENABLED 1
145
146 mDNSexport const char *const mDNS_DomainTypeNames[] =
147 {
148 "b._dns-sd._udp.", // Browse
149 "db._dns-sd._udp.", // Default Browse
150 "lb._dns-sd._udp.", // Automatic Browse
151 "r._dns-sd._udp.", // Registration
152 "dr._dns-sd._udp." // Default Registration
153 };
154
155 #ifdef UNICAST_DISABLED
156 #define uDNS_IsActiveQuery(q, u) mDNSfalse
157 #endif
158
159 // ***************************************************************************
160 #if COMPILER_LIKES_PRAGMA_MARK
161 #pragma mark -
162 #pragma mark - General Utility Functions
163 #endif
164
165 // Returns true if this is a unique, authoritative LocalOnly record that answers questions of type
166 // A, AAAA , CNAME, or PTR. The caller should answer the question with this record and not send out
167 // the question on the wire if LocalOnlyRecordAnswersQuestion() also returns true.
168 // Main use is to handle /etc/hosts records and the LocalOnly PTR records created for localhost.
169 #define UniqueLocalOnlyRecord(rr) ((rr)->ARType == AuthRecordLocalOnly && \
170 (rr)->resrec.RecordType & kDNSRecordTypeUniqueMask && \
171 ((rr)->resrec.rrtype == kDNSType_A || (rr)->resrec.rrtype == kDNSType_AAAA || \
172 (rr)->resrec.rrtype == kDNSType_CNAME || \
173 (rr)->resrec.rrtype == kDNSType_PTR))
174
175 mDNSlocal void SetNextQueryStopTime(mDNS *const m, const DNSQuestion *const q)
176 {
177 mDNS_CheckLock(m);
178
179 if (m->NextScheduledStopTime - q->StopTime > 0)
180 m->NextScheduledStopTime = q->StopTime;
181 }
182
183 mDNSexport void SetNextQueryTime(mDNS *const m, const DNSQuestion *const q)
184 {
185 mDNS_CheckLock(m);
186
187 if (ActiveQuestion(q))
188 {
189 // Depending on whether this is a multicast or unicast question we want to set either:
190 // m->NextScheduledQuery = NextQSendTime(q) or
191 // m->NextuDNSEvent = NextQSendTime(q)
192 mDNSs32 *const timer = mDNSOpaque16IsZero(q->TargetQID) ? &m->NextScheduledQuery : &m->NextuDNSEvent;
193 if (*timer - NextQSendTime(q) > 0)
194 *timer = NextQSendTime(q);
195 }
196 }
197
198 mDNSlocal void ReleaseAuthEntity(AuthHash *r, AuthEntity *e)
199 {
200 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING >= 1
201 unsigned int i;
202 for (i=0; i<sizeof(*e); i++) ((char*)e)[i] = 0xFF;
203 #endif
204 e->next = r->rrauth_free;
205 r->rrauth_free = e;
206 r->rrauth_totalused--;
207 }
208
209 mDNSlocal void ReleaseAuthGroup(AuthHash *r, AuthGroup **cp)
210 {
211 AuthEntity *e = (AuthEntity *)(*cp);
212 LogMsg("ReleaseAuthGroup: Releasing AuthGroup %##s", (*cp)->name->c);
213 if ((*cp)->rrauth_tail != &(*cp)->members)
214 LogMsg("ERROR: (*cp)->members == mDNSNULL but (*cp)->rrauth_tail != &(*cp)->members)");
215 if ((*cp)->name != (domainname*)((*cp)->namestorage)) mDNSPlatformMemFree((*cp)->name);
216 (*cp)->name = mDNSNULL;
217 *cp = (*cp)->next; // Cut record from list
218 ReleaseAuthEntity(r, e);
219 }
220
221 mDNSlocal AuthEntity *GetAuthEntity(AuthHash *r, const AuthGroup *const PreserveAG)
222 {
223 AuthEntity *e = mDNSNULL;
224
225 if (r->rrauth_lock) { LogMsg("GetFreeCacheRR ERROR! Cache already locked!"); return(mDNSNULL); }
226 r->rrauth_lock = 1;
227
228 if (!r->rrauth_free)
229 {
230 // We allocate just one AuthEntity at a time because we need to be able
231 // free them all individually which normally happens when we parse /etc/hosts into
232 // AuthHash where we add the "new" entries and discard (free) the already added
233 // entries. If we allocate as chunks, we can't free them individually.
234 AuthEntity *storage = mDNSPlatformMemAllocate(sizeof(AuthEntity));
235 storage->next = mDNSNULL;
236 r->rrauth_free = storage;
237 }
238
239 // If we still have no free records, recycle all the records we can.
240 // Enumerating the entire auth is moderately expensive, so when we do it, we reclaim all the records we can in one pass.
241 if (!r->rrauth_free)
242 {
243 mDNSu32 oldtotalused = r->rrauth_totalused;
244 mDNSu32 slot;
245 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
246 {
247 AuthGroup **cp = &r->rrauth_hash[slot];
248 while (*cp)
249 {
250 if ((*cp)->members || (*cp)==PreserveAG) cp=&(*cp)->next;
251 else ReleaseAuthGroup(r, cp);
252 }
253 }
254 LogInfo("GetAuthEntity: Recycled %d records to reduce auth cache from %d to %d",
255 oldtotalused - r->rrauth_totalused, oldtotalused, r->rrauth_totalused);
256 }
257
258 if (r->rrauth_free) // If there are records in the free list, take one
259 {
260 e = r->rrauth_free;
261 r->rrauth_free = e->next;
262 if (++r->rrauth_totalused >= r->rrauth_report)
263 {
264 LogInfo("RR Auth now using %ld objects", r->rrauth_totalused);
265 if (r->rrauth_report < 100) r->rrauth_report += 10;
266 else if (r->rrauth_report < 1000) r->rrauth_report += 100;
267 else r->rrauth_report += 1000;
268 }
269 mDNSPlatformMemZero(e, sizeof(*e));
270 }
271
272 r->rrauth_lock = 0;
273
274 return(e);
275 }
276
277 mDNSexport AuthGroup *AuthGroupForName(AuthHash *r, const mDNSu32 slot, const mDNSu32 namehash, const domainname *const name)
278 {
279 AuthGroup *ag;
280 for (ag = r->rrauth_hash[slot]; ag; ag=ag->next)
281 if (ag->namehash == namehash && SameDomainName(ag->name, name))
282 break;
283 return(ag);
284 }
285
286 mDNSexport AuthGroup *AuthGroupForRecord(AuthHash *r, const mDNSu32 slot, const ResourceRecord *const rr)
287 {
288 return(AuthGroupForName(r, slot, rr->namehash, rr->name));
289 }
290
291 mDNSlocal AuthGroup *GetAuthGroup(AuthHash *r, const mDNSu32 slot, const ResourceRecord *const rr)
292 {
293 mDNSu16 namelen = DomainNameLength(rr->name);
294 AuthGroup *ag = (AuthGroup*)GetAuthEntity(r, mDNSNULL);
295 if (!ag) { LogMsg("GetAuthGroup: Failed to allocate memory for %##s", rr->name->c); return(mDNSNULL); }
296 ag->next = r->rrauth_hash[slot];
297 ag->namehash = rr->namehash;
298 ag->members = mDNSNULL;
299 ag->rrauth_tail = &ag->members;
300 ag->NewLocalOnlyRecords = mDNSNULL;
301 if (namelen > sizeof(ag->namestorage))
302 ag->name = mDNSPlatformMemAllocate(namelen);
303 else
304 ag->name = (domainname*)ag->namestorage;
305 if (!ag->name)
306 {
307 LogMsg("GetAuthGroup: Failed to allocate name storage for %##s", rr->name->c);
308 ReleaseAuthEntity(r, (AuthEntity*)ag);
309 return(mDNSNULL);
310 }
311 AssignDomainName(ag->name, rr->name);
312
313 if (AuthGroupForRecord(r, slot, rr)) LogMsg("GetAuthGroup: Already have AuthGroup for %##s", rr->name->c);
314 r->rrauth_hash[slot] = ag;
315 if (AuthGroupForRecord(r, slot, rr) != ag) LogMsg("GetAuthGroup: Not finding AuthGroup for %##s", rr->name->c);
316
317 return(ag);
318 }
319
320 // Returns the AuthGroup in which the AuthRecord was inserted
321 mDNSexport AuthGroup *InsertAuthRecord(mDNS *const m, AuthHash *r, AuthRecord *rr)
322 {
323 (void)m;
324 AuthGroup *ag;
325 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
326 ag = AuthGroupForRecord(r, slot, &rr->resrec);
327 if (!ag) ag = GetAuthGroup(r, slot, &rr->resrec); // If we don't have a AuthGroup for this name, make one now
328 if (ag)
329 {
330 *(ag->rrauth_tail) = rr; // Append this record to tail of cache slot list
331 ag->rrauth_tail = &(rr->next); // Advance tail pointer
332 }
333 return ag;
334 }
335
336 mDNSexport AuthGroup *RemoveAuthRecord(mDNS *const m, AuthHash *r, AuthRecord *rr)
337 {
338 AuthGroup *a;
339 AuthRecord **rp;
340 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
341
342 a = AuthGroupForRecord(r, slot, &rr->resrec);
343 if (!a) { LogMsg("RemoveAuthRecord: ERROR!! AuthGroup not found for %s", ARDisplayString(m, rr)); return mDNSNULL; }
344 rp = &a->members;
345 while (*rp)
346 {
347 if (*rp != rr)
348 rp=&(*rp)->next;
349 else
350 {
351 // We don't break here, so that we can set the tail below without tracking "prev" pointers
352
353 LogInfo("RemoveAuthRecord: removing auth record %s from table", ARDisplayString(m, rr));
354 *rp = (*rp)->next; // Cut record from list
355 }
356 }
357 // TBD: If there are no more members, release authgroup ?
358 a->rrauth_tail = rp;
359 return a;
360 }
361
362 mDNSexport CacheGroup *CacheGroupForName(const mDNS *const m, const mDNSu32 slot, const mDNSu32 namehash, const domainname *const name)
363 {
364 CacheGroup *cg;
365 for (cg = m->rrcache_hash[slot]; cg; cg=cg->next)
366 if (cg->namehash == namehash && SameDomainName(cg->name, name))
367 break;
368 return(cg);
369 }
370
371 mDNSlocal CacheGroup *CacheGroupForRecord(const mDNS *const m, const mDNSu32 slot, const ResourceRecord *const rr)
372 {
373 return(CacheGroupForName(m, slot, rr->namehash, rr->name));
374 }
375
376 mDNSexport mDNSBool mDNS_AddressIsLocalSubnet(mDNS *const m, const mDNSInterfaceID InterfaceID, const mDNSAddr *addr)
377 {
378 NetworkInterfaceInfo *intf;
379
380 if (addr->type == mDNSAddrType_IPv4)
381 {
382 // Normally we resist touching the NotAnInteger fields, but here we're doing tricky bitwise masking so we make an exception
383 if (mDNSv4AddressIsLinkLocal(&addr->ip.v4)) return(mDNStrue);
384 for (intf = m->HostInterfaces; intf; intf = intf->next)
385 if (intf->ip.type == addr->type && intf->InterfaceID == InterfaceID && intf->McastTxRx)
386 if (((intf->ip.ip.v4.NotAnInteger ^ addr->ip.v4.NotAnInteger) & intf->mask.ip.v4.NotAnInteger) == 0)
387 return(mDNStrue);
388 }
389
390 if (addr->type == mDNSAddrType_IPv6)
391 {
392 if (mDNSv6AddressIsLinkLocal(&addr->ip.v6)) return(mDNStrue);
393 for (intf = m->HostInterfaces; intf; intf = intf->next)
394 if (intf->ip.type == addr->type && intf->InterfaceID == InterfaceID && intf->McastTxRx)
395 if ((((intf->ip.ip.v6.l[0] ^ addr->ip.v6.l[0]) & intf->mask.ip.v6.l[0]) == 0) &&
396 (((intf->ip.ip.v6.l[1] ^ addr->ip.v6.l[1]) & intf->mask.ip.v6.l[1]) == 0) &&
397 (((intf->ip.ip.v6.l[2] ^ addr->ip.v6.l[2]) & intf->mask.ip.v6.l[2]) == 0) &&
398 (((intf->ip.ip.v6.l[3] ^ addr->ip.v6.l[3]) & intf->mask.ip.v6.l[3]) == 0))
399 return(mDNStrue);
400 }
401
402 return(mDNSfalse);
403 }
404
405 mDNSlocal NetworkInterfaceInfo *FirstInterfaceForID(mDNS *const m, const mDNSInterfaceID InterfaceID)
406 {
407 NetworkInterfaceInfo *intf = m->HostInterfaces;
408 while (intf && intf->InterfaceID != InterfaceID) intf = intf->next;
409 return(intf);
410 }
411
412 mDNSlocal NetworkInterfaceInfo *FirstIPv4LLInterfaceForID(mDNS *const m, const mDNSInterfaceID InterfaceID)
413 {
414 NetworkInterfaceInfo *intf;
415
416 if (!InterfaceID)
417 return mDNSNULL;
418
419 // Note: We don't check for InterfaceActive, as the active interface could be IPv6 and
420 // we still want to find the first IPv4 Link-Local interface
421 for (intf = m->HostInterfaces; intf; intf = intf->next)
422 {
423 if (intf->InterfaceID == InterfaceID &&
424 intf->ip.type == mDNSAddrType_IPv4 && mDNSv4AddressIsLinkLocal(&intf->ip.ip.v4))
425 {
426 debugf("FirstIPv4LLInterfaceForID: found LL interface with address %.4a", &intf->ip.ip.v4);
427 return intf;
428 }
429 }
430 return (mDNSNULL);
431 }
432
433 mDNSexport char *InterfaceNameForID(mDNS *const m, const mDNSInterfaceID InterfaceID)
434 {
435 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
436 return(intf ? intf->ifname : mDNSNULL);
437 }
438
439 // Caller should hold the lock
440 mDNSlocal void GenerateNegativeResponse(mDNS *const m, mDNSInterfaceID InterfaceID, QC_result qc)
441 {
442 DNSQuestion *q;
443 if (!m->CurrentQuestion) { LogMsg("GenerateNegativeResponse: ERROR!! CurrentQuestion not set"); return; }
444 q = m->CurrentQuestion;
445 LogInfo("GenerateNegativeResponse: Generating negative response for question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
446
447 MakeNegativeCacheRecord(m, &m->rec.r, &q->qname, q->qnamehash, q->qtype, q->qclass, 60, InterfaceID, mDNSNULL);
448
449 // We need to force the response through in the following cases
450 //
451 // a) SuppressUnusable questions that are suppressed
452 // b) Append search domains and retry the question
453 //
454 // The question may not have set Intermediates in which case we don't deliver negative responses. So, to force
455 // through we use "QC_forceresponse".
456 AnswerCurrentQuestionWithResourceRecord(m, &m->rec.r, qc);
457 if (m->CurrentQuestion == q) { q->ThisQInterval = 0; } // Deactivate this question
458 // Don't touch the question after this
459 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
460 }
461
462 mDNSexport void AnswerQuestionByFollowingCNAME(mDNS *const m, DNSQuestion *q, ResourceRecord *rr)
463 {
464 const mDNSBool selfref = SameDomainName(&q->qname, &rr->rdata->u.name);
465 if (q->CNAMEReferrals >= 10 || selfref)
466 {
467 LogMsg("AnswerQuestionByFollowingCNAME: %p %##s (%s) NOT following CNAME referral %d%s for %s",
468 q, q->qname.c, DNSTypeName(q->qtype), q->CNAMEReferrals, selfref ? " (Self-Referential)" : "", RRDisplayString(m, rr));
469 }
470 else
471 {
472 const mDNSu32 c = q->CNAMEReferrals + 1; // Stash a copy of the new q->CNAMEReferrals value
473 UDPSocket *sock = q->LocalSocket;
474 mDNSOpaque16 id = q->TargetQID;
475 #if TARGET_OS_EMBEDDED
476 uDNSMetrics metrics;
477 #endif
478
479 q->LocalSocket = mDNSNULL;
480
481 // The SameDomainName check above is to ignore bogus CNAME records that point right back at
482 // themselves. Without that check we can get into a case where we have two duplicate questions,
483 // A and B, and when we stop question A, UpdateQuestionDuplicates copies the value of CNAMEReferrals
484 // from A to B, and then A is re-appended to the end of the list as a duplicate of B (because
485 // the target name is still the same), and then when we stop question B, UpdateQuestionDuplicates
486 // copies the B's value of CNAMEReferrals back to A, and we end up not incrementing CNAMEReferrals
487 // for either of them. This is not a problem for CNAME loops of two or more records because in
488 // those cases the newly re-appended question A has a different target name and therefore cannot be
489 // a duplicate of any other question ('B') which was itself a duplicate of the previous question A.
490
491 // Right now we just stop and re-use the existing query. If we really wanted to be 100% perfect,
492 // and track CNAMEs coming and going, we should really create a subordinate query here,
493 // which we would subsequently cancel and retract if the CNAME referral record were removed.
494 // In reality this is such a corner case we'll ignore it until someone actually needs it.
495
496 LogInfo("AnswerQuestionByFollowingCNAME: %p %##s (%s) following CNAME referral %d for %s",
497 q, q->qname.c, DNSTypeName(q->qtype), q->CNAMEReferrals, RRDisplayString(m, rr));
498
499 #if TARGET_OS_EMBEDDED
500 if ((q->CNAMEReferrals == 0) && !q->metrics.originalQName)
501 {
502 domainname * qName;
503 mDNSu16 qNameLen;
504
505 qNameLen = DomainNameLength(&q->qname);
506 if ((qNameLen > 0) && (qNameLen <= MAX_DOMAIN_NAME))
507 {
508 qName = mDNSPlatformMemAllocate(qNameLen);
509 if (qName)
510 {
511 mDNSPlatformMemCopy(qName->c, q->qname.c, qNameLen);
512 q->metrics.originalQName = qName;
513 }
514 }
515 }
516 metrics = q->metrics;
517 mDNSPlatformMemZero(&q->metrics, sizeof(q->metrics));
518 #endif
519 mDNS_StopQuery_internal(m, q); // Stop old query
520 AssignDomainName(&q->qname, &rr->rdata->u.name); // Update qname
521 q->qnamehash = DomainNameHashValue(&q->qname); // and namehash
522 // If a unicast query results in a CNAME that points to a .local, we need to re-try
523 // this as unicast. Setting the mDNSInterface_Unicast tells mDNS_StartQuery_internal
524 // to try this as unicast query even though it is a .local name
525 if (!mDNSOpaque16IsZero(q->TargetQID) && IsLocalDomain(&q->qname))
526 {
527 LogInfo("AnswerQuestionByFollowingCNAME: Resolving a .local CNAME %p %##s (%s) Record %s",
528 q, q->qname.c, DNSTypeName(q->qtype), RRDisplayString(m, rr));
529 q->InterfaceID = mDNSInterface_Unicast;
530 }
531 mDNS_StartQuery_internal(m, q); // start new query
532 // Record how many times we've done this. We need to do this *after* mDNS_StartQuery_internal,
533 // because mDNS_StartQuery_internal re-initializes CNAMEReferrals to zero
534 q->CNAMEReferrals = c;
535 #if TARGET_OS_EMBEDDED
536 q->metrics = metrics;
537 #endif
538 if (sock)
539 {
540 // If our new query is a duplicate, then it can't have a socket of its own, so we have to close the one we saved.
541 if (q->DuplicateOf) mDNSPlatformUDPClose(sock);
542 else
543 {
544 // Transplant the old socket into the new question, and copy the query ID across too.
545 // No need to close the old q->LocalSocket value because it won't have been created yet (they're made lazily on-demand).
546 q->LocalSocket = sock;
547 q->TargetQID = id;
548 }
549 }
550 }
551 }
552
553 #ifdef USE_LIBIDN
554
555 #include <unicode/uidna.h>
556
557 // #define DEBUG_PUNYCODE 1
558
559 mDNSlocal mDNSu8 *PunycodeConvert(const mDNSu8 *const src, mDNSu8 *const dst, const mDNSu8 *const end)
560 {
561 UErrorCode errorCode = U_ZERO_ERROR;
562 UIDNAInfo info = UIDNA_INFO_INITIALIZER;
563 UIDNA *uts46 = uidna_openUTS46(UIDNA_USE_STD3_RULES|UIDNA_NONTRANSITIONAL_TO_UNICODE, &errorCode);
564 int32_t len = uidna_nameToASCII_UTF8(uts46, (const char *)src+1, src[0], (char *)dst+1, end-(dst+1), &info, &errorCode);
565 uidna_close(uts46);
566 #if DEBUG_PUNYCODE
567 if (errorCode) LogMsg("uidna_nameToASCII_UTF8(%##s) failed errorCode %d", src, errorCode);
568 if (info.errors) LogMsg("uidna_nameToASCII_UTF8(%##s) failed info.errors 0x%08X", src, info.errors);
569 if (len > MAX_DOMAIN_LABEL) LogMsg("uidna_nameToASCII_UTF8(%##s) result too long %d", src, len);
570 #endif
571 if (errorCode || info.errors || len > MAX_DOMAIN_LABEL) return mDNSNULL;
572 *dst = len;
573 return(dst + 1 + len);
574 }
575
576 mDNSlocal mDNSBool IsHighASCIILabel(const mDNSu8 *d)
577 {
578 int i;
579 for (i=1; i<=d[0]; i++) if (d[i] & 0x80) return mDNStrue;
580 return mDNSfalse;
581 }
582
583 mDNSlocal const mDNSu8 *FindLastHighASCIILabel(const domainname *const d)
584 {
585 const mDNSu8 *ptr = d->c;
586 const mDNSu8 *ans = mDNSNULL;
587 while (ptr[0])
588 {
589 const mDNSu8 *const next = ptr + 1 + ptr[0];
590 if (ptr[0] > MAX_DOMAIN_LABEL || next >= d->c + MAX_DOMAIN_NAME) return mDNSNULL;
591 if (IsHighASCIILabel(ptr)) ans = ptr;
592 ptr = next;
593 }
594 return ans;
595 }
596
597 mDNSlocal mDNSBool PerformNextPunycodeConversion(const DNSQuestion *const q, domainname *const newname)
598 {
599 const mDNSu8 *h = FindLastHighASCIILabel(&q->qname);
600 #if DEBUG_PUNYCODE
601 LogMsg("PerformNextPunycodeConversion: %##s (%s) Last High-ASCII Label %##s", q->qname.c, DNSTypeName(q->qtype), h);
602 #endif
603 if (!h) return mDNSfalse; // There are no high-ascii labels to convert
604
605 mDNSu8 *const dst = PunycodeConvert(h, newname->c + (h - q->qname.c), newname->c + MAX_DOMAIN_NAME);
606 if (!dst)
607 return mDNSfalse; // The label was not convertible to Punycode
608 else
609 {
610 // If Punycode conversion of final eligible label was successful, copy the rest of the domainname
611 const mDNSu8 *const src = h + 1 + h[0];
612 const mDNSu8 remainder = DomainNameLength((domainname*)src);
613 if (dst + remainder > newname->c + MAX_DOMAIN_NAME) return mDNSfalse; // Name too long -- cannot be converted to Punycode
614
615 mDNSPlatformMemCopy(newname->c, q->qname.c, h - q->qname.c); // Fill in the leading part
616 mDNSPlatformMemCopy(dst, src, remainder); // Fill in the trailing part
617 #if DEBUG_PUNYCODE
618 LogMsg("PerformNextPunycodeConversion: %##s converted to %##s", q->qname.c, newname->c);
619 #endif
620 return mDNStrue;
621 }
622 }
623
624 #endif // USE_LIBIDN
625
626 // For a single given DNSQuestion pointed to by CurrentQuestion, deliver an add/remove result for the single given AuthRecord
627 // Note: All the callers should use the m->CurrentQuestion to see if the question is still valid or not
628 mDNSlocal void AnswerLocalQuestionWithLocalAuthRecord(mDNS *const m, AuthRecord *rr, QC_result AddRecord)
629 {
630 DNSQuestion *q = m->CurrentQuestion;
631 mDNSBool followcname;
632
633 if (!q)
634 {
635 LogMsg("AnswerLocalQuestionWithLocalAuthRecord: ERROR!! CurrentQuestion NULL while answering with %s", ARDisplayString(m, rr));
636 return;
637 }
638
639 followcname = FollowCNAME(q, &rr->resrec, AddRecord);
640
641 // We should not be delivering results for record types Unregistered, Deregistering, and (unverified) Unique
642 if (!(rr->resrec.RecordType & kDNSRecordTypeActiveMask))
643 {
644 LogMsg("AnswerLocalQuestionWithLocalAuthRecord: *NOT* delivering %s event for local record type %X %s",
645 AddRecord ? "Add" : "Rmv", rr->resrec.RecordType, ARDisplayString(m, rr));
646 return;
647 }
648
649 // Indicate that we've given at least one positive answer for this record, so we should be prepared to send a goodbye for it
650 if (AddRecord) rr->AnsweredLocalQ = mDNStrue;
651 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
652 if (q->QuestionCallback && !q->NoAnswer)
653 {
654 q->CurrentAnswers += AddRecord ? 1 : -1;
655 if (UniqueLocalOnlyRecord(rr))
656 {
657 if (!followcname || q->ReturnIntermed)
658 {
659 // Don't send this packet on the wire as we answered from /etc/hosts
660 q->ThisQInterval = 0;
661 q->LOAddressAnswers += AddRecord ? 1 : -1;
662 q->QuestionCallback(m, q, &rr->resrec, AddRecord);
663 }
664 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
665 // The callback above could have caused the question to stop. Detect that
666 // using m->CurrentQuestion
667 if (followcname && m->CurrentQuestion == q)
668 AnswerQuestionByFollowingCNAME(m, q, &rr->resrec);
669 return;
670 }
671 else
672 {
673 q->QuestionCallback(m, q, &rr->resrec, AddRecord);
674 }
675 }
676 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
677 }
678
679 mDNSlocal void AnswerInterfaceAnyQuestionsWithLocalAuthRecord(mDNS *const m, AuthRecord *rr, QC_result AddRecord)
680 {
681 if (m->CurrentQuestion)
682 LogMsg("AnswerInterfaceAnyQuestionsWithLocalAuthRecord: ERROR m->CurrentQuestion already set: %##s (%s)",
683 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
684 m->CurrentQuestion = m->Questions;
685 while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
686 {
687 mDNSBool answered;
688 DNSQuestion *q = m->CurrentQuestion;
689 if (RRAny(rr))
690 answered = ResourceRecordAnswersQuestion(&rr->resrec, q);
691 else
692 answered = LocalOnlyRecordAnswersQuestion(rr, q);
693 if (answered)
694 AnswerLocalQuestionWithLocalAuthRecord(m, rr, AddRecord); // MUST NOT dereference q again
695 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
696 m->CurrentQuestion = q->next;
697 }
698 m->CurrentQuestion = mDNSNULL;
699 }
700
701 // When a new local AuthRecord is created or deleted, AnswerAllLocalQuestionsWithLocalAuthRecord()
702 // delivers the appropriate add/remove events to listening questions:
703 // 1. It runs though all our LocalOnlyQuestions delivering answers as appropriate,
704 // stopping if it reaches a NewLocalOnlyQuestion -- brand-new questions are handled by AnswerNewLocalOnlyQuestion().
705 // 2. If the AuthRecord is marked mDNSInterface_LocalOnly or mDNSInterface_P2P, then it also runs though
706 // our main question list, delivering answers to mDNSInterface_Any questions as appropriate,
707 // stopping if it reaches a NewQuestion -- brand-new questions are handled by AnswerNewQuestion().
708 //
709 // AnswerAllLocalQuestionsWithLocalAuthRecord is used by the m->NewLocalRecords loop in mDNS_Execute(),
710 // and by mDNS_Deregister_internal()
711
712 mDNSlocal void AnswerAllLocalQuestionsWithLocalAuthRecord(mDNS *const m, AuthRecord *rr, QC_result AddRecord)
713 {
714 if (m->CurrentQuestion)
715 LogMsg("AnswerAllLocalQuestionsWithLocalAuthRecord ERROR m->CurrentQuestion already set: %##s (%s)",
716 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
717
718 m->CurrentQuestion = m->LocalOnlyQuestions;
719 while (m->CurrentQuestion && m->CurrentQuestion != m->NewLocalOnlyQuestions)
720 {
721 mDNSBool answered;
722 DNSQuestion *q = m->CurrentQuestion;
723 // We are called with both LocalOnly/P2P record or a regular AuthRecord
724 if (RRAny(rr))
725 answered = ResourceRecordAnswersQuestion(&rr->resrec, q);
726 else
727 answered = LocalOnlyRecordAnswersQuestion(rr, q);
728 if (answered)
729 AnswerLocalQuestionWithLocalAuthRecord(m, rr, AddRecord); // MUST NOT dereference q again
730 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
731 m->CurrentQuestion = q->next;
732 }
733
734 m->CurrentQuestion = mDNSNULL;
735
736 // If this AuthRecord is marked LocalOnly or P2P, then we want to deliver it to all local 'mDNSInterface_Any' questions
737 if (rr->ARType == AuthRecordLocalOnly || rr->ARType == AuthRecordP2P)
738 AnswerInterfaceAnyQuestionsWithLocalAuthRecord(m, rr, AddRecord);
739
740 }
741
742 // ***************************************************************************
743 #if COMPILER_LIKES_PRAGMA_MARK
744 #pragma mark -
745 #pragma mark - Resource Record Utility Functions
746 #endif
747
748 #define RRTypeIsAddressType(T) ((T) == kDNSType_A || (T) == kDNSType_AAAA)
749
750 #define ResourceRecordIsValidAnswer(RR) ( ((RR)->resrec.RecordType & kDNSRecordTypeActiveMask) && \
751 ((RR)->Additional1 == mDNSNULL || ((RR)->Additional1->resrec.RecordType & kDNSRecordTypeActiveMask)) && \
752 ((RR)->Additional2 == mDNSNULL || ((RR)->Additional2->resrec.RecordType & kDNSRecordTypeActiveMask)) && \
753 ((RR)->DependentOn == mDNSNULL || ((RR)->DependentOn->resrec.RecordType & kDNSRecordTypeActiveMask)) )
754
755 #define ResourceRecordIsValidInterfaceAnswer(RR, INTID) \
756 (ResourceRecordIsValidAnswer(RR) && \
757 ((RR)->resrec.InterfaceID == mDNSInterface_Any || (RR)->resrec.InterfaceID == (INTID)))
758
759 #define DefaultProbeCountForTypeUnique ((mDNSu8)3)
760 #define DefaultProbeCountForRecordType(X) ((X) == kDNSRecordTypeUnique ? DefaultProbeCountForTypeUnique : (mDNSu8)0)
761
762 // See RFC 6762: "8.3 Announcing"
763 // "The Multicast DNS responder MUST send at least two unsolicited responses, one second apart."
764 // Send 4, which is really 8 since we send on both IPv4 and IPv6.
765 #define InitialAnnounceCount ((mDNSu8)4)
766
767 // For goodbye packets we set the count to 3, and for wakeups we set it to 18
768 // (which will be up to 15 wakeup attempts over the course of 30 seconds,
769 // and then if the machine fails to wake, 3 goodbye packets).
770 #define GoodbyeCount ((mDNSu8)3)
771 #define WakeupCount ((mDNSu8)18)
772 #define MAX_PROBE_RESTARTS ((mDNSu8)20)
773
774 // Number of wakeups we send if WakeOnResolve is set in the question
775 #define InitialWakeOnResolveCount ((mDNSu8)3)
776
777 // Note that the announce intervals use exponential backoff, doubling each time. The probe intervals do not.
778 // This means that because the announce interval is doubled after sending the first packet, the first
779 // observed on-the-wire inter-packet interval between announcements is actually one second.
780 // The half-second value here may be thought of as a conceptual (non-existent) half-second delay *before* the first packet is sent.
781 #define DefaultProbeIntervalForTypeUnique (mDNSPlatformOneSecond/4)
782 #define DefaultAnnounceIntervalForTypeShared (mDNSPlatformOneSecond/2)
783 #define DefaultAnnounceIntervalForTypeUnique (mDNSPlatformOneSecond/2)
784
785 #define DefaultAPIntervalForRecordType(X) ((X) &kDNSRecordTypeActiveSharedMask ? DefaultAnnounceIntervalForTypeShared : \
786 (X) &kDNSRecordTypeUnique ? DefaultProbeIntervalForTypeUnique : \
787 (X) &kDNSRecordTypeActiveUniqueMask ? DefaultAnnounceIntervalForTypeUnique : 0)
788
789 #define TimeToAnnounceThisRecord(RR,time) ((RR)->AnnounceCount && (time) - ((RR)->LastAPTime + (RR)->ThisAPInterval) >= 0)
790 #define TimeToSendThisRecord(RR,time) ((TimeToAnnounceThisRecord(RR,time) || (RR)->ImmedAnswer) && ResourceRecordIsValidAnswer(RR))
791 #define TicksTTL(RR) ((mDNSs32)(RR)->resrec.rroriginalttl * mDNSPlatformOneSecond)
792 #define RRExpireTime(RR) ((RR)->TimeRcvd + TicksTTL(RR))
793
794 // Adjustment factor to avoid race condition (used for unicast cache entries) :
795 // Suppose real record has TTL of 3600, and our local caching server has held it for 3500 seconds, so it returns an aged TTL of 100.
796 // If we do our normal refresh at 80% of the TTL, our local caching server will return 20 seconds, so we'll do another
797 // 80% refresh after 16 seconds, and then the server will return 4 seconds, and so on, in the fashion of Zeno's paradox.
798 // To avoid this, we extend the record's effective TTL to give it a little extra grace period.
799 // We adjust the 100 second TTL to 127. This means that when we do our 80% query at 102 seconds,
800 // the cached copy at our local caching server will already have expired, so the server will be forced
801 // to fetch a fresh copy from the authoritative server, and then return a fresh record with the full TTL of 3600 seconds.
802
803 #define RRAdjustTTL(ttl) ((ttl) + ((ttl)/4) + 2)
804 #define RRUnadjustedTTL(ttl) ((((ttl) - 2) * 4) / 5)
805
806 #define MaxUnansweredQueries 4
807
808 // SameResourceRecordSignature returns true if two resources records have the same name, type, and class, and may be sent
809 // (or were received) on the same interface (i.e. if *both* records specify an interface, then it has to match).
810 // TTL and rdata may differ.
811 // This is used for cache flush management:
812 // When sending a unique record, all other records matching "SameResourceRecordSignature" must also be sent
813 // When receiving a unique record, all old cache records matching "SameResourceRecordSignature" are flushed
814
815 // SameResourceRecordNameClassInterface is functionally the same as SameResourceRecordSignature, except rrtype does not have to match
816
817 #define SameResourceRecordSignature(A,B) (A)->resrec.rrtype == (B)->resrec.rrtype && SameResourceRecordNameClassInterface((A),(B))
818
819 mDNSlocal mDNSBool SameResourceRecordNameClassInterface(const AuthRecord *const r1, const AuthRecord *const r2)
820 {
821 if (!r1) { LogMsg("SameResourceRecordSignature ERROR: r1 is NULL"); return(mDNSfalse); }
822 if (!r2) { LogMsg("SameResourceRecordSignature ERROR: r2 is NULL"); return(mDNSfalse); }
823 if (r1->resrec.InterfaceID &&
824 r2->resrec.InterfaceID &&
825 r1->resrec.InterfaceID != r2->resrec.InterfaceID) return(mDNSfalse);
826 return (mDNSBool)(
827 r1->resrec.rrclass == r2->resrec.rrclass &&
828 r1->resrec.namehash == r2->resrec.namehash &&
829 SameDomainName(r1->resrec.name, r2->resrec.name));
830 }
831
832 // PacketRRMatchesSignature behaves as SameResourceRecordSignature, except that types may differ if our
833 // authoratative record is unique (as opposed to shared). For unique records, we are supposed to have
834 // complete ownership of *all* types for this name, so *any* record type with the same name is a conflict.
835 // In addition, when probing we send our questions with the wildcard type kDNSQType_ANY,
836 // so a response of any type should match, even if it is not actually the type the client plans to use.
837
838 // For now, to make it easier to avoid false conflicts, we treat SPS Proxy records like shared records,
839 // and require the rrtypes to match for the rdata to be considered potentially conflicting
840 mDNSlocal mDNSBool PacketRRMatchesSignature(const CacheRecord *const pktrr, const AuthRecord *const authrr)
841 {
842 if (!pktrr) { LogMsg("PacketRRMatchesSignature ERROR: pktrr is NULL"); return(mDNSfalse); }
843 if (!authrr) { LogMsg("PacketRRMatchesSignature ERROR: authrr is NULL"); return(mDNSfalse); }
844 if (pktrr->resrec.InterfaceID &&
845 authrr->resrec.InterfaceID &&
846 pktrr->resrec.InterfaceID != authrr->resrec.InterfaceID) return(mDNSfalse);
847 if (!(authrr->resrec.RecordType & kDNSRecordTypeUniqueMask) || authrr->WakeUp.HMAC.l[0])
848 if (pktrr->resrec.rrtype != authrr->resrec.rrtype) return(mDNSfalse);
849 if ((authrr->resrec.InterfaceID == mDNSInterface_Any) &&
850 !mDNSPlatformValidRecordForInterface(authrr, pktrr->resrec.InterfaceID)) return(mDNSfalse);
851 return (mDNSBool)(
852 pktrr->resrec.rrclass == authrr->resrec.rrclass &&
853 pktrr->resrec.namehash == authrr->resrec.namehash &&
854 SameDomainName(pktrr->resrec.name, authrr->resrec.name));
855 }
856
857 // CacheRecord *ka is the CacheRecord from the known answer list in the query.
858 // This is the information that the requester believes to be correct.
859 // AuthRecord *rr is the answer we are proposing to give, if not suppressed.
860 // This is the information that we believe to be correct.
861 // We've already determined that we plan to give this answer on this interface
862 // (either the record is non-specific, or it is specific to this interface)
863 // so now we just need to check the name, type, class, rdata and TTL.
864 mDNSlocal mDNSBool ShouldSuppressKnownAnswer(const CacheRecord *const ka, const AuthRecord *const rr)
865 {
866 // If RR signature is different, or data is different, then don't suppress our answer
867 if (!IdenticalResourceRecord(&ka->resrec, &rr->resrec)) return(mDNSfalse);
868
869 // If the requester's indicated TTL is less than half the real TTL,
870 // we need to give our answer before the requester's copy expires.
871 // If the requester's indicated TTL is at least half the real TTL,
872 // then we can suppress our answer this time.
873 // If the requester's indicated TTL is greater than the TTL we believe,
874 // then that's okay, and we don't need to do anything about it.
875 // (If two responders on the network are offering the same information,
876 // that's okay, and if they are offering the information with different TTLs,
877 // the one offering the lower TTL should defer to the one offering the higher TTL.)
878 return (mDNSBool)(ka->resrec.rroriginalttl >= rr->resrec.rroriginalttl / 2);
879 }
880
881 mDNSlocal void SetNextAnnounceProbeTime(mDNS *const m, const AuthRecord *const rr)
882 {
883 if (rr->resrec.RecordType == kDNSRecordTypeUnique)
884 {
885 if ((rr->LastAPTime + rr->ThisAPInterval) - m->timenow > mDNSPlatformOneSecond * 10)
886 {
887 LogMsg("SetNextAnnounceProbeTime: ProbeCount %d Next in %d %s", rr->ProbeCount, (rr->LastAPTime + rr->ThisAPInterval) - m->timenow, ARDisplayString(m, rr));
888 LogMsg("SetNextAnnounceProbeTime: m->SuppressProbes %d m->timenow %d diff %d", m->SuppressProbes, m->timenow, m->SuppressProbes - m->timenow);
889 }
890 if (m->NextScheduledProbe - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
891 m->NextScheduledProbe = (rr->LastAPTime + rr->ThisAPInterval);
892 // Some defensive code:
893 // If (rr->LastAPTime + rr->ThisAPInterval) happens to be far in the past, we don't want to allow
894 // NextScheduledProbe to be set excessively in the past, because that can cause bad things to happen.
895 // See: <rdar://problem/7795434> mDNS: Sometimes advertising stops working and record interval is set to zero
896 if (m->NextScheduledProbe - m->timenow < 0)
897 m->NextScheduledProbe = m->timenow;
898 }
899 else if (rr->AnnounceCount && (ResourceRecordIsValidAnswer(rr) || rr->resrec.RecordType == kDNSRecordTypeDeregistering))
900 {
901 if (m->NextScheduledResponse - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
902 m->NextScheduledResponse = (rr->LastAPTime + rr->ThisAPInterval);
903 }
904 }
905
906 mDNSlocal void InitializeLastAPTime(mDNS *const m, AuthRecord *const rr)
907 {
908 // For reverse-mapping Sleep Proxy PTR records, probe interval is one second
909 rr->ThisAPInterval = rr->AddressProxy.type ? mDNSPlatformOneSecond : DefaultAPIntervalForRecordType(rr->resrec.RecordType);
910
911 // * If this is a record type that's going to probe, then we use the m->SuppressProbes time.
912 // * Otherwise, if it's not going to probe, but m->SuppressProbes is set because we have other
913 // records that are going to probe, then we delay its first announcement so that it will
914 // go out synchronized with the first announcement for the other records that *are* probing.
915 // This is a minor performance tweak that helps keep groups of related records synchronized together.
916 // The addition of "interval / 2" is to make sure that, in the event that any of the probes are
917 // delayed by a few milliseconds, this announcement does not inadvertently go out *before* the probing is complete.
918 // When the probing is complete and those records begin to announce, these records will also be picked up and accelerated,
919 // because they will meet the criterion of being at least half-way to their scheduled announcement time.
920 // * If it's not going to probe and m->SuppressProbes is not already set then we should announce immediately.
921
922 if (rr->ProbeCount)
923 {
924 // If we have no probe suppression time set, or it is in the past, set it now
925 if (m->SuppressProbes == 0 || m->SuppressProbes - m->timenow < 0)
926 {
927 // To allow us to aggregate probes when a group of services are registered together,
928 // the first probe is delayed by a random delay in the range 1/8 to 1/4 second.
929 // This means the common-case behaviour is:
930 // randomized wait; probe
931 // 1/4 second wait; probe
932 // 1/4 second wait; probe
933 // 1/4 second wait; announce (i.e. service is normally announced 7/8 to 1 second after being registered)
934 m->SuppressProbes = NonZeroTime(m->timenow + DefaultProbeIntervalForTypeUnique/2 + mDNSRandom(DefaultProbeIntervalForTypeUnique/2));
935
936 // If we already have a *probe* scheduled to go out sooner, then use that time to get better aggregation
937 if (m->SuppressProbes - m->NextScheduledProbe >= 0)
938 m->SuppressProbes = NonZeroTime(m->NextScheduledProbe);
939 if (m->SuppressProbes - m->timenow < 0) // Make sure we don't set m->SuppressProbes excessively in the past
940 m->SuppressProbes = m->timenow;
941
942 // If we already have a *query* scheduled to go out sooner, then use that time to get better aggregation
943 if (m->SuppressProbes - m->NextScheduledQuery >= 0)
944 m->SuppressProbes = NonZeroTime(m->NextScheduledQuery);
945 if (m->SuppressProbes - m->timenow < 0) // Make sure we don't set m->SuppressProbes excessively in the past
946 m->SuppressProbes = m->timenow;
947
948 // except... don't expect to be able to send before the m->SuppressSending timer fires
949 if (m->SuppressSending && m->SuppressProbes - m->SuppressSending < 0)
950 m->SuppressProbes = NonZeroTime(m->SuppressSending);
951
952 if (m->SuppressProbes - m->timenow > mDNSPlatformOneSecond * 8)
953 {
954 LogMsg("InitializeLastAPTime ERROR m->SuppressProbes %d m->NextScheduledProbe %d m->NextScheduledQuery %d m->SuppressSending %d %d",
955 m->SuppressProbes - m->timenow,
956 m->NextScheduledProbe - m->timenow,
957 m->NextScheduledQuery - m->timenow,
958 m->SuppressSending,
959 m->SuppressSending - m->timenow);
960 m->SuppressProbes = NonZeroTime(m->timenow + DefaultProbeIntervalForTypeUnique/2 + mDNSRandom(DefaultProbeIntervalForTypeUnique/2));
961 }
962 }
963 rr->LastAPTime = m->SuppressProbes - rr->ThisAPInterval;
964 }
965 // Skip kDNSRecordTypeKnownUnique records here and set their LastAPTime in the "else" block below so that they get announced immediately,
966 // otherwise, their announcement would be delayed until all other record probes complete.
967 else if ((rr->resrec.RecordType != kDNSRecordTypeKnownUnique) && m->SuppressProbes && m->SuppressProbes - m->timenow >= 0)
968 rr->LastAPTime = m->SuppressProbes - rr->ThisAPInterval + DefaultProbeIntervalForTypeUnique * DefaultProbeCountForTypeUnique + rr->ThisAPInterval / 2;
969 else
970 rr->LastAPTime = m->timenow - rr->ThisAPInterval;
971
972 // For reverse-mapping Sleep Proxy PTR records we don't want to start probing instantly -- we
973 // wait one second to give the client a chance to go to sleep, and then start our ARP/NDP probing.
974 // After three probes one second apart with no answer, we conclude the client is now sleeping
975 // and we can begin broadcasting our announcements to take over ownership of that IP address.
976 // If we don't wait for the client to go to sleep, then when the client sees our ARP Announcements there's a risk
977 // (depending on the OS and networking stack it's using) that it might interpret it as a conflict and change its IP address.
978 if (rr->AddressProxy.type)
979 rr->LastAPTime = m->timenow;
980
981 // Set LastMCTime to now, to inhibit multicast responses
982 // (no need to send additional multicast responses when we're announcing anyway)
983 rr->LastMCTime = m->timenow;
984 rr->LastMCInterface = mDNSInterfaceMark;
985
986 SetNextAnnounceProbeTime(m, rr);
987 }
988
989 mDNSlocal const domainname *SetUnicastTargetToHostName(mDNS *const m, AuthRecord *rr)
990 {
991 const domainname *target;
992 if (rr->AutoTarget)
993 {
994 // For autotunnel services pointing at our IPv6 ULA we don't need or want a NAT mapping, but for all other
995 // advertised services referencing our uDNS hostname, we want NAT mappings automatically created as appropriate,
996 // with the port number in our advertised SRV record automatically tracking the external mapped port.
997 DomainAuthInfo *AuthInfo = GetAuthInfoForName_internal(m, rr->resrec.name);
998 if (!AuthInfo || !AuthInfo->AutoTunnel) rr->AutoTarget = Target_AutoHostAndNATMAP;
999 }
1000
1001 target = GetServiceTarget(m, rr);
1002 if (!target || target->c[0] == 0)
1003 {
1004 // defer registration until we've got a target
1005 LogInfo("SetUnicastTargetToHostName No target for %s", ARDisplayString(m, rr));
1006 rr->state = regState_NoTarget;
1007 return mDNSNULL;
1008 }
1009 else
1010 {
1011 LogInfo("SetUnicastTargetToHostName target %##s for resource record %s", target->c, ARDisplayString(m,rr));
1012 return target;
1013 }
1014 }
1015
1016 // Right now this only applies to mDNS (.local) services where the target host is always m->MulticastHostname
1017 // Eventually we should unify this with GetServiceTarget() in uDNS.c
1018 mDNSlocal void SetTargetToHostName(mDNS *const m, AuthRecord *const rr)
1019 {
1020 domainname *const target = GetRRDomainNameTarget(&rr->resrec);
1021 const domainname *newname = &m->MulticastHostname;
1022
1023 if (!target) LogInfo("SetTargetToHostName: Don't know how to set the target of rrtype %s", DNSTypeName(rr->resrec.rrtype));
1024
1025 if (!(rr->ForceMCast || rr->ARType == AuthRecordLocalOnly || rr->ARType == AuthRecordP2P || IsLocalDomain(&rr->namestorage)))
1026 {
1027 const domainname *const n = SetUnicastTargetToHostName(m, rr);
1028 if (n) newname = n;
1029 else { if (target) target->c[0] = 0; SetNewRData(&rr->resrec, mDNSNULL, 0); return; }
1030 }
1031
1032 if (target && SameDomainName(target, newname))
1033 debugf("SetTargetToHostName: Target of %##s is already %##s", rr->resrec.name->c, target->c);
1034
1035 if (target && !SameDomainName(target, newname))
1036 {
1037 AssignDomainName(target, newname);
1038 SetNewRData(&rr->resrec, mDNSNULL, 0); // Update rdlength, rdestimate, rdatahash
1039
1040 // If we're in the middle of probing this record, we need to start again,
1041 // because changing its rdata may change the outcome of the tie-breaker.
1042 // (If the record type is kDNSRecordTypeUnique (unconfirmed unique) then DefaultProbeCountForRecordType is non-zero.)
1043 rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
1044
1045 // If we've announced this record, we really should send a goodbye packet for the old rdata before
1046 // changing to the new rdata. However, in practice, we only do SetTargetToHostName for unique records,
1047 // so when we announce them we'll set the kDNSClass_UniqueRRSet and clear any stale data that way.
1048 if (rr->RequireGoodbye && rr->resrec.RecordType == kDNSRecordTypeShared)
1049 debugf("Have announced shared record %##s (%s) at least once: should have sent a goodbye packet before updating",
1050 rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1051
1052 rr->AnnounceCount = InitialAnnounceCount;
1053 rr->RequireGoodbye = mDNSfalse;
1054 rr->ProbeRestartCount = 0;
1055 InitializeLastAPTime(m, rr);
1056 }
1057 }
1058
1059 mDNSlocal void AcknowledgeRecord(mDNS *const m, AuthRecord *const rr)
1060 {
1061 if (rr->RecordCallback)
1062 {
1063 // CAUTION: MUST NOT do anything more with rr after calling rr->Callback(), because the client's callback function
1064 // is allowed to do anything, including starting/stopping queries, registering/deregistering records, etc.
1065 rr->Acknowledged = mDNStrue;
1066 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
1067 rr->RecordCallback(m, rr, mStatus_NoError);
1068 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
1069 }
1070 }
1071
1072 mDNSexport void ActivateUnicastRegistration(mDNS *const m, AuthRecord *const rr)
1073 {
1074 // Make sure that we don't activate the SRV record and associated service records, if it is in
1075 // NoTarget state. First time when a service is being instantiated, SRV record may be in NoTarget state.
1076 // We should not activate any of the other reords (PTR, TXT) that are part of the service. When
1077 // the target becomes available, the records will be reregistered.
1078 if (rr->resrec.rrtype != kDNSType_SRV)
1079 {
1080 AuthRecord *srvRR = mDNSNULL;
1081 if (rr->resrec.rrtype == kDNSType_PTR)
1082 srvRR = rr->Additional1;
1083 else if (rr->resrec.rrtype == kDNSType_TXT)
1084 srvRR = rr->DependentOn;
1085 if (srvRR)
1086 {
1087 if (srvRR->resrec.rrtype != kDNSType_SRV)
1088 {
1089 LogMsg("ActivateUnicastRegistration: ERROR!! Resource record %s wrong, expecting SRV type", ARDisplayString(m, srvRR));
1090 }
1091 else
1092 {
1093 LogInfo("ActivateUnicastRegistration: Found Service Record %s in state %d for %##s (%s)",
1094 ARDisplayString(m, srvRR), srvRR->state, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1095 rr->state = srvRR->state;
1096 }
1097 }
1098 }
1099
1100 if (rr->state == regState_NoTarget)
1101 {
1102 LogInfo("ActivateUnicastRegistration record %s in regState_NoTarget, not activating", ARDisplayString(m, rr));
1103 return;
1104 }
1105 // When we wake up from sleep, we call ActivateUnicastRegistration. It is possible that just before we went to sleep,
1106 // the service/record was being deregistered. In that case, we should not try to register again. For the cases where
1107 // the records are deregistered due to e.g., no target for the SRV record, we would have returned from above if it
1108 // was already in NoTarget state. If it was in the process of deregistration but did not complete fully before we went
1109 // to sleep, then it is okay to start in Pending state as we will go back to NoTarget state if we don't have a target.
1110 if (rr->resrec.RecordType == kDNSRecordTypeDeregistering)
1111 {
1112 LogInfo("ActivateUnicastRegistration: Resource record %s, current state %d, moving to DeregPending", ARDisplayString(m, rr), rr->state);
1113 rr->state = regState_DeregPending;
1114 }
1115 else
1116 {
1117 LogInfo("ActivateUnicastRegistration: Resource record %s, current state %d, moving to Pending", ARDisplayString(m, rr), rr->state);
1118 rr->state = regState_Pending;
1119 }
1120 rr->ProbeCount = 0;
1121 rr->ProbeRestartCount = 0;
1122 rr->AnnounceCount = 0;
1123 rr->ThisAPInterval = INIT_RECORD_REG_INTERVAL;
1124 rr->LastAPTime = m->timenow - rr->ThisAPInterval;
1125 rr->expire = 0; // Forget about all the leases, start fresh
1126 rr->uselease = mDNStrue;
1127 rr->updateid = zeroID;
1128 rr->SRVChanged = mDNSfalse;
1129 rr->updateError = mStatus_NoError;
1130 // RestartRecordGetZoneData calls this function whenever a new interface gets registered with core.
1131 // The records might already be registered with the server and hence could have NAT state.
1132 if (rr->NATinfo.clientContext)
1133 {
1134 mDNS_StopNATOperation_internal(m, &rr->NATinfo);
1135 rr->NATinfo.clientContext = mDNSNULL;
1136 }
1137 if (rr->nta) { CancelGetZoneData(m, rr->nta); rr->nta = mDNSNULL; }
1138 if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
1139 if (m->NextuDNSEvent - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
1140 m->NextuDNSEvent = (rr->LastAPTime + rr->ThisAPInterval);
1141 }
1142
1143 // Two records qualify to be local duplicates if:
1144 // (a) the RecordTypes are the same, or
1145 // (b) one is Unique and the other Verified
1146 // (c) either is in the process of deregistering
1147 #define RecordLDT(A,B) ((A)->resrec.RecordType == (B)->resrec.RecordType || \
1148 ((A)->resrec.RecordType | (B)->resrec.RecordType) == (kDNSRecordTypeUnique | kDNSRecordTypeVerified) || \
1149 ((A)->resrec.RecordType == kDNSRecordTypeDeregistering || (B)->resrec.RecordType == kDNSRecordTypeDeregistering))
1150
1151 #define RecordIsLocalDuplicate(A,B) \
1152 ((A)->resrec.InterfaceID == (B)->resrec.InterfaceID && RecordLDT((A),(B)) && IdenticalResourceRecord(& (A)->resrec, & (B)->resrec))
1153
1154 mDNSlocal AuthRecord *CheckAuthIdenticalRecord(AuthHash *r, AuthRecord *rr)
1155 {
1156 const AuthGroup *a;
1157 AuthRecord *rp;
1158 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
1159
1160 a = AuthGroupForRecord(r, slot, &rr->resrec);
1161 if (!a) return mDNSNULL;
1162 rp = a->members;
1163 while (rp)
1164 {
1165 if (!RecordIsLocalDuplicate(rp, rr))
1166 rp = rp->next;
1167 else
1168 {
1169 if (rp->resrec.RecordType == kDNSRecordTypeDeregistering)
1170 {
1171 rp->AnnounceCount = 0;
1172 rp = rp->next;
1173 }
1174 else return rp;
1175 }
1176 }
1177 return (mDNSNULL);
1178 }
1179
1180 mDNSlocal mDNSBool CheckAuthRecordConflict(AuthHash *r, AuthRecord *rr)
1181 {
1182 const AuthGroup *a;
1183 const AuthRecord *rp;
1184 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
1185
1186 a = AuthGroupForRecord(r, slot, &rr->resrec);
1187 if (!a) return mDNSfalse;
1188 rp = a->members;
1189 while (rp)
1190 {
1191 const AuthRecord *s1 = rr->RRSet ? rr->RRSet : rr;
1192 const AuthRecord *s2 = rp->RRSet ? rp->RRSet : rp;
1193 if (s1 != s2 && SameResourceRecordSignature(rp, rr) && !IdenticalSameNameRecord(&rp->resrec, &rr->resrec))
1194 return mDNStrue;
1195 else
1196 rp = rp->next;
1197 }
1198 return (mDNSfalse);
1199 }
1200
1201 // checks to see if "rr" is already present
1202 mDNSlocal AuthRecord *CheckAuthSameRecord(AuthHash *r, AuthRecord *rr)
1203 {
1204 const AuthGroup *a;
1205 AuthRecord *rp;
1206 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
1207
1208 a = AuthGroupForRecord(r, slot, &rr->resrec);
1209 if (!a) return mDNSNULL;
1210 rp = a->members;
1211 while (rp)
1212 {
1213 if (rp != rr)
1214 rp = rp->next;
1215 else
1216 {
1217 return rp;
1218 }
1219 }
1220 return (mDNSNULL);
1221 }
1222
1223
1224 mDNSlocal void DecrementAutoTargetServices(mDNS *const m, AuthRecord *const rr)
1225 {
1226 if (RRLocalOnly(rr))
1227 {
1228 // A sanity check, this should be prevented in calling code.
1229 LogInfo("DecrementAutoTargetServices: called for RRLocalOnly() record: %s", ARDisplayString(m, rr));
1230 return;
1231 }
1232
1233 if (!AuthRecord_uDNS(rr) && rr->resrec.rrtype == kDNSType_SRV && rr->AutoTarget == Target_AutoHost)
1234 {
1235 // If about to get rid of the last advertised service
1236 if (m->AutoTargetServices == 1)
1237 DeadvertiseAllInterfaceRecords(m);
1238
1239 m->AutoTargetServices--;
1240 LogInfo("DecrementAutoTargetServices: AutoTargetServices %d Record %s", m->AutoTargetServices, ARDisplayString(m, rr));
1241 }
1242
1243 #if BONJOUR_ON_DEMAND
1244 if (!AuthRecord_uDNS(rr))
1245 {
1246 if (m->NumAllInterfaceRecords + m->NumAllInterfaceQuestions == 1)
1247 m->NextBonjourDisableTime = NonZeroTime(m->timenow + (BONJOUR_DISABLE_DELAY * mDNSPlatformOneSecond));
1248 m->NumAllInterfaceRecords--;
1249 LogInfo("DecrementAutoTargetServices: NumAllInterfaceRecords %d NumAllInterfaceQuestions %d %s",
1250 m->NumAllInterfaceRecords, m->NumAllInterfaceQuestions, ARDisplayString(m, rr));
1251 }
1252 #endif // BONJOUR_ON_DEMAND
1253 }
1254
1255 mDNSlocal void IncrementAutoTargetServices(mDNS *const m, AuthRecord *const rr)
1256 {
1257 mDNSBool enablingBonjour = 0;
1258
1259 if (RRLocalOnly(rr))
1260 {
1261 // A sanity check, this should be prevented in calling code.
1262 LogInfo("IncrementAutoTargetServices: called for RRLocalOnly() record: %s", ARDisplayString(m, rr));
1263 return;
1264 }
1265
1266 #if BONJOUR_ON_DEMAND
1267 if (!AuthRecord_uDNS(rr))
1268 {
1269 m->NumAllInterfaceRecords++;
1270 LogInfo("IncrementAutoTargetServices: NumAllInterfaceRecords %d NumAllInterfaceQuestions %d %s",
1271 m->NumAllInterfaceRecords, m->NumAllInterfaceQuestions, ARDisplayString(m, rr));
1272 if (m->NumAllInterfaceRecords + m->NumAllInterfaceQuestions == 1)
1273 {
1274 m->NextBonjourDisableTime = 0;
1275 if (m->BonjourEnabled == 0)
1276 {
1277 // Enable Bonjour immediately by scheduling network changed processing where
1278 // we will join the multicast group on each active interface.
1279 m->BonjourEnabled = 1;
1280 enablingBonjour = 1;
1281 m->NetworkChanged = m->timenow;
1282 }
1283 }
1284 }
1285 #endif // BONJOUR_ON_DEMAND
1286
1287 if (!AuthRecord_uDNS(rr) && rr->resrec.rrtype == kDNSType_SRV && rr->AutoTarget == Target_AutoHost)
1288 {
1289 m->AutoTargetServices++;
1290 LogInfo("IncrementAutoTargetServices: AutoTargetServices %d Record %s", m->AutoTargetServices, ARDisplayString(m, rr));
1291
1292 // If this is the first advertised service and we did not just enable Bonjour above, then
1293 // advertise all the interface records. If we did enable Bonjour above, the interface records will
1294 // be advertised during the network changed processing scheduled above, so no need
1295 // to do it here.
1296 if ((m->AutoTargetServices == 1) && (enablingBonjour == 0))
1297 AdvertiseAllInterfaceRecords(m);
1298 }
1299 }
1300
1301 mDNSlocal void getKeepaliveRaddr(mDNS *const m, AuthRecord *rr, mDNSAddr *raddr)
1302 {
1303 mDNSAddr laddr = zeroAddr;
1304 mDNSEthAddr eth = zeroEthAddr;
1305 mDNSIPPort lport = zeroIPPort;
1306 mDNSIPPort rport = zeroIPPort;
1307 mDNSu32 timeout = 0;
1308 mDNSu32 seq = 0;
1309 mDNSu32 ack = 0;
1310 mDNSu16 win = 0;
1311
1312 if (mDNS_KeepaliveRecord(&rr->resrec))
1313 {
1314 mDNS_ExtractKeepaliveInfo(rr, &timeout, &laddr, raddr, &eth, &seq, &ack, &lport, &rport, &win);
1315 if (!timeout || mDNSAddressIsZero(&laddr) || mDNSAddressIsZero(raddr) || mDNSIPPortIsZero(lport) || mDNSIPPortIsZero(rport))
1316 {
1317 LogMsg("getKeepaliveRaddr: not a valid record %s for keepalive %#a:%d %#a:%d", ARDisplayString(m, rr), &laddr, lport.NotAnInteger, raddr, rport.NotAnInteger);
1318 return;
1319 }
1320 }
1321 }
1322
1323 // Exported so uDNS.c can call this
1324 mDNSexport mStatus mDNS_Register_internal(mDNS *const m, AuthRecord *const rr)
1325 {
1326 domainname *target = GetRRDomainNameTarget(&rr->resrec);
1327 AuthRecord *r;
1328 AuthRecord **p = &m->ResourceRecords;
1329 AuthRecord **d = &m->DuplicateRecords;
1330
1331 if ((mDNSs32)rr->resrec.rroriginalttl <= 0)
1332 { LogMsg("mDNS_Register_internal: TTL %X should be 1 - 0x7FFFFFFF %s", rr->resrec.rroriginalttl, ARDisplayString(m, rr)); return(mStatus_BadParamErr); }
1333
1334 if (!rr->resrec.RecordType)
1335 { LogMsg("mDNS_Register_internal: RecordType must be non-zero %s", ARDisplayString(m, rr)); return(mStatus_BadParamErr); }
1336
1337 if (m->ShutdownTime)
1338 { LogMsg("mDNS_Register_internal: Shutting down, can't register %s", ARDisplayString(m, rr)); return(mStatus_ServiceNotRunning); }
1339
1340 if (m->DivertMulticastAdvertisements && !AuthRecord_uDNS(rr))
1341 {
1342 mDNSInterfaceID previousID = rr->resrec.InterfaceID;
1343 if (rr->resrec.InterfaceID == mDNSInterface_Any || rr->resrec.InterfaceID == mDNSInterface_P2P)
1344 {
1345 rr->resrec.InterfaceID = mDNSInterface_LocalOnly;
1346 rr->ARType = AuthRecordLocalOnly;
1347 }
1348 if (rr->resrec.InterfaceID != mDNSInterface_LocalOnly)
1349 {
1350 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
1351 if (intf && !intf->Advertise) { rr->resrec.InterfaceID = mDNSInterface_LocalOnly; rr->ARType = AuthRecordLocalOnly; }
1352 }
1353 if (rr->resrec.InterfaceID != previousID)
1354 LogInfo("mDNS_Register_internal: Diverting record to local-only %s", ARDisplayString(m, rr));
1355 }
1356
1357 if (RRLocalOnly(rr))
1358 {
1359 if (CheckAuthSameRecord(&m->rrauth, rr))
1360 {
1361 LogMsg("mDNS_Register_internal: ERROR!! Tried to register LocalOnly AuthRecord %p %##s (%s) that's already in the list",
1362 rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1363 return(mStatus_AlreadyRegistered);
1364 }
1365 }
1366 else
1367 {
1368 while (*p && *p != rr) p=&(*p)->next;
1369 if (*p)
1370 {
1371 LogMsg("mDNS_Register_internal: ERROR!! Tried to register AuthRecord %p %##s (%s) that's already in the list",
1372 rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1373 return(mStatus_AlreadyRegistered);
1374 }
1375 }
1376
1377 while (*d && *d != rr) d=&(*d)->next;
1378 if (*d)
1379 {
1380 LogMsg("mDNS_Register_internal: ERROR!! Tried to register AuthRecord %p %##s (%s) that's already in the Duplicate list",
1381 rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1382 return(mStatus_AlreadyRegistered);
1383 }
1384
1385 if (rr->DependentOn)
1386 {
1387 if (rr->resrec.RecordType == kDNSRecordTypeUnique)
1388 rr->resrec.RecordType = kDNSRecordTypeVerified;
1389 else
1390 {
1391 LogMsg("mDNS_Register_internal: ERROR! %##s (%s): rr->DependentOn && RecordType != kDNSRecordTypeUnique",
1392 rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1393 return(mStatus_Invalid);
1394 }
1395 if (!(rr->DependentOn->resrec.RecordType & (kDNSRecordTypeUnique | kDNSRecordTypeVerified | kDNSRecordTypeKnownUnique)))
1396 {
1397 LogMsg("mDNS_Register_internal: ERROR! %##s (%s): rr->DependentOn->RecordType bad type %X",
1398 rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), rr->DependentOn->resrec.RecordType);
1399 return(mStatus_Invalid);
1400 }
1401 }
1402
1403 rr->next = mDNSNULL;
1404
1405 // Field Group 1: The actual information pertaining to this resource record
1406 // Set up by client prior to call
1407
1408 // Field Group 2: Persistent metadata for Authoritative Records
1409 // rr->Additional1 = set to mDNSNULL in mDNS_SetupResourceRecord; may be overridden by client
1410 // rr->Additional2 = set to mDNSNULL in mDNS_SetupResourceRecord; may be overridden by client
1411 // rr->DependentOn = set to mDNSNULL in mDNS_SetupResourceRecord; may be overridden by client
1412 // rr->RRSet = set to mDNSNULL in mDNS_SetupResourceRecord; may be overridden by client
1413 // rr->Callback = already set in mDNS_SetupResourceRecord
1414 // rr->Context = already set in mDNS_SetupResourceRecord
1415 // rr->RecordType = already set in mDNS_SetupResourceRecord
1416 // rr->HostTarget = set to mDNSfalse in mDNS_SetupResourceRecord; may be overridden by client
1417 // rr->AllowRemoteQuery = set to mDNSfalse in mDNS_SetupResourceRecord; may be overridden by client
1418 // Make sure target is not uninitialized data, or we may crash writing debugging log messages
1419 if (rr->AutoTarget && target) target->c[0] = 0;
1420
1421 // Field Group 3: Transient state for Authoritative Records
1422 rr->Acknowledged = mDNSfalse;
1423 rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
1424 rr->ProbeRestartCount = 0;
1425 rr->AnnounceCount = InitialAnnounceCount;
1426 rr->RequireGoodbye = mDNSfalse;
1427 rr->AnsweredLocalQ = mDNSfalse;
1428 rr->IncludeInProbe = mDNSfalse;
1429 rr->ImmedUnicast = mDNSfalse;
1430 rr->SendNSECNow = mDNSNULL;
1431 rr->ImmedAnswer = mDNSNULL;
1432 rr->ImmedAdditional = mDNSNULL;
1433 rr->SendRNow = mDNSNULL;
1434 rr->v4Requester = zerov4Addr;
1435 rr->v6Requester = zerov6Addr;
1436 rr->NextResponse = mDNSNULL;
1437 rr->NR_AnswerTo = mDNSNULL;
1438 rr->NR_AdditionalTo = mDNSNULL;
1439 if (!rr->AutoTarget) InitializeLastAPTime(m, rr);
1440 // rr->LastAPTime = Set for us in InitializeLastAPTime()
1441 // rr->LastMCTime = Set for us in InitializeLastAPTime()
1442 // rr->LastMCInterface = Set for us in InitializeLastAPTime()
1443 rr->NewRData = mDNSNULL;
1444 rr->newrdlength = 0;
1445 rr->UpdateCallback = mDNSNULL;
1446 rr->UpdateCredits = kMaxUpdateCredits;
1447 rr->NextUpdateCredit = 0;
1448 rr->UpdateBlocked = 0;
1449
1450 // For records we're holding as proxy (except reverse-mapping PTR records) two announcements is sufficient
1451 if (rr->WakeUp.HMAC.l[0] && !rr->AddressProxy.type) rr->AnnounceCount = 2;
1452
1453 // Field Group 4: Transient uDNS state for Authoritative Records
1454 rr->state = regState_Zero;
1455 rr->uselease = 0;
1456 rr->expire = 0;
1457 rr->Private = 0;
1458 rr->updateid = zeroID;
1459 rr->updateIntID = zeroOpaque64;
1460 rr->zone = rr->resrec.name;
1461 rr->nta = mDNSNULL;
1462 rr->tcp = mDNSNULL;
1463 rr->OrigRData = 0;
1464 rr->OrigRDLen = 0;
1465 rr->InFlightRData = 0;
1466 rr->InFlightRDLen = 0;
1467 rr->QueuedRData = 0;
1468 rr->QueuedRDLen = 0;
1469 //mDNSPlatformMemZero(&rr->NATinfo, sizeof(rr->NATinfo));
1470 // We should be recording the actual internal port for this service record here. Once we initiate our NAT mapping
1471 // request we'll subsequently overwrite srv.port with the allocated external NAT port -- potentially multiple
1472 // times with different values if the external NAT port changes during the lifetime of the service registration.
1473 //if (rr->resrec.rrtype == kDNSType_SRV) rr->NATinfo.IntPort = rr->resrec.rdata->u.srv.port;
1474
1475 // rr->resrec.interface = already set in mDNS_SetupResourceRecord
1476 // rr->resrec.name->c = MUST be set by client
1477 // rr->resrec.rrtype = already set in mDNS_SetupResourceRecord
1478 // rr->resrec.rrclass = already set in mDNS_SetupResourceRecord
1479 // rr->resrec.rroriginalttl = already set in mDNS_SetupResourceRecord
1480 // rr->resrec.rdata = MUST be set by client, unless record type is CNAME or PTR and rr->HostTarget is set
1481
1482 // BIND named (name daemon) doesn't allow TXT records with zero-length rdata. This is strictly speaking correct,
1483 // since RFC 1035 specifies a TXT record as "One or more <character-string>s", not "Zero or more <character-string>s".
1484 // Since some legacy apps try to create zero-length TXT records, we'll silently correct it here.
1485 if (rr->resrec.rrtype == kDNSType_TXT && rr->resrec.rdlength == 0) { rr->resrec.rdlength = 1; rr->resrec.rdata->u.txt.c[0] = 0; }
1486
1487 if (rr->AutoTarget)
1488 {
1489 SetTargetToHostName(m, rr); // Also sets rdlength and rdestimate for us, and calls InitializeLastAPTime();
1490 #ifndef UNICAST_DISABLED
1491 // If we have no target record yet, SetTargetToHostName will set rr->state == regState_NoTarget
1492 // In this case we leave the record half-formed in the list, and later we'll remove it from the list and re-add it properly.
1493 if (rr->state == regState_NoTarget)
1494 {
1495 // Initialize the target so that we don't crash while logging etc.
1496 domainname *tar = GetRRDomainNameTarget(&rr->resrec);
1497 if (tar) tar->c[0] = 0;
1498 LogInfo("mDNS_Register_internal: record %s in NoTarget state", ARDisplayString(m, rr));
1499 }
1500 #endif
1501 }
1502 else
1503 {
1504 rr->resrec.rdlength = GetRDLength(&rr->resrec, mDNSfalse);
1505 rr->resrec.rdestimate = GetRDLength(&rr->resrec, mDNStrue);
1506 }
1507
1508 if (!ValidateDomainName(rr->resrec.name))
1509 { LogMsg("Attempt to register record with invalid name: %s", ARDisplayString(m, rr)); return(mStatus_Invalid); }
1510
1511 // Don't do this until *after* we've set rr->resrec.rdlength
1512 if (!ValidateRData(rr->resrec.rrtype, rr->resrec.rdlength, rr->resrec.rdata))
1513 { LogMsg("Attempt to register record with invalid rdata: %s", ARDisplayString(m, rr)); return(mStatus_Invalid); }
1514
1515 rr->resrec.namehash = DomainNameHashValue(rr->resrec.name);
1516 rr->resrec.rdatahash = target ? DomainNameHashValue(target) : RDataHashValue(&rr->resrec);
1517
1518 if (RRLocalOnly(rr))
1519 {
1520 // If this is supposed to be unique, make sure we don't have any name conflicts.
1521 // If we found a conflict, we may still want to insert the record in the list but mark it appropriately
1522 // (kDNSRecordTypeDeregistering) so that we deliver RMV events to the application. But this causes more
1523 // complications and not clear whether there are any benefits. See rdar:9304275 for details.
1524 // Hence, just bail out.
1525 // This comment is doesn’t make any sense. -- SC
1526 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
1527 {
1528 if (CheckAuthRecordConflict(&m->rrauth, rr))
1529 {
1530 LogInfo("mDNS_Register_internal: Name conflict %s (%p), InterfaceID %p", ARDisplayString(m, rr), rr, rr->resrec.InterfaceID);
1531 return mStatus_NameConflict;
1532 }
1533 }
1534 }
1535
1536 // For uDNS records, we don't support duplicate checks at this time.
1537 #ifndef UNICAST_DISABLED
1538 if (AuthRecord_uDNS(rr))
1539 {
1540 if (!m->NewLocalRecords) m->NewLocalRecords = rr;
1541 // When we called SetTargetToHostName, it may have caused mDNS_Register_internal to be re-entered, appending new
1542 // records to the list, so we now need to update p to advance to the new end to the list before appending our new record.
1543 // Note that for AutoTunnel this should never happen, but this check makes the code future-proof.
1544 while (*p) p=&(*p)->next;
1545 *p = rr;
1546 if (rr->resrec.RecordType == kDNSRecordTypeUnique) rr->resrec.RecordType = kDNSRecordTypeVerified;
1547 rr->ProbeCount = 0;
1548 rr->ProbeRestartCount = 0;
1549 rr->AnnounceCount = 0;
1550 if (rr->state != regState_NoTarget) ActivateUnicastRegistration(m, rr);
1551 return(mStatus_NoError); // <--- Note: For unicast records, code currently bails out at this point
1552 }
1553 #endif
1554
1555 // Now that we've finished building our new record, make sure it's not identical to one we already have
1556 if (RRLocalOnly(rr))
1557 {
1558 rr->ProbeCount = 0;
1559 rr->ProbeRestartCount = 0;
1560 rr->AnnounceCount = 0;
1561 r = CheckAuthIdenticalRecord(&m->rrauth, rr);
1562 }
1563 else
1564 {
1565 for (r = m->ResourceRecords; r; r=r->next)
1566 if (RecordIsLocalDuplicate(r, rr))
1567 {
1568 if (r->resrec.RecordType == kDNSRecordTypeDeregistering) r->AnnounceCount = 0;
1569 else break;
1570 }
1571 }
1572
1573 if (r)
1574 {
1575 LogInfo("mDNS_Register_internal: Adding to duplicate list %s", ARDisplayString(m,rr));
1576 *d = rr;
1577 // If the previous copy of this record is already verified unique,
1578 // then indicate that we should move this record promptly to kDNSRecordTypeUnique state.
1579 // Setting ProbeCount to zero will cause SendQueries() to advance this record to
1580 // kDNSRecordTypeVerified state and call the client callback at the next appropriate time.
1581 if (rr->resrec.RecordType == kDNSRecordTypeUnique && r->resrec.RecordType == kDNSRecordTypeVerified)
1582 rr->ProbeCount = 0;
1583 }
1584 else
1585 {
1586 LogInfo("mDNS_Register_internal: Adding to active record list %s", ARDisplayString(m,rr));
1587 if (RRLocalOnly(rr))
1588 {
1589 AuthGroup *ag;
1590 ag = InsertAuthRecord(m, &m->rrauth, rr);
1591 if (ag && !ag->NewLocalOnlyRecords)
1592 {
1593 m->NewLocalOnlyRecords = mDNStrue;
1594 ag->NewLocalOnlyRecords = rr;
1595 }
1596 // No probing for LocalOnly records; acknowledge them right away
1597 if (rr->resrec.RecordType == kDNSRecordTypeUnique) rr->resrec.RecordType = kDNSRecordTypeVerified;
1598 AcknowledgeRecord(m, rr);
1599 return(mStatus_NoError);
1600 }
1601 else
1602 {
1603 if (!m->NewLocalRecords) m->NewLocalRecords = rr;
1604 *p = rr;
1605 }
1606 }
1607
1608 if (!AuthRecord_uDNS(rr)) // This check is superfluous, given that for unicast records we (currently) bail out above
1609 {
1610 // We have inserted the record in the list. See if we have to advertise the A/AAAA, HINFO, PTR records.
1611 IncrementAutoTargetServices(m, rr);
1612
1613 // For records that are not going to probe, acknowledge them right away
1614 if (rr->resrec.RecordType != kDNSRecordTypeUnique && rr->resrec.RecordType != kDNSRecordTypeDeregistering)
1615 AcknowledgeRecord(m, rr);
1616
1617 // Adding a record may affect whether or not we should sleep
1618 mDNS_UpdateAllowSleep(m);
1619 }
1620
1621 // If this is a non-sleep proxy keepalive record, fetch the MAC address of the remote host.
1622 // This is used by the in-NIC proxy to send the keepalive packets.
1623 if (!rr->WakeUp.HMAC.l[0] && mDNS_KeepaliveRecord(&rr->resrec))
1624 {
1625 mDNSAddr raddr;
1626 // Set the record type to known unique to prevent probing keep alive records.
1627 // Also make sure we do not announce the keepalive records.
1628 rr->resrec.RecordType = kDNSRecordTypeKnownUnique;
1629 rr->AnnounceCount = 0;
1630 getKeepaliveRaddr(m, rr, &raddr);
1631 // This is an asynchronous call. Once the remote MAC address is available, helper will schedule an
1632 // asynchronous task to update the resource record
1633 mDNSPlatformGetRemoteMacAddr(m, &raddr);
1634 }
1635
1636 return(mStatus_NoError);
1637 }
1638
1639 mDNSlocal void RecordProbeFailure(mDNS *const m, const AuthRecord *const rr)
1640 {
1641 m->ProbeFailTime = m->timenow;
1642 m->NumFailedProbes++;
1643 // If we've had fifteen or more probe failures, rate-limit to one every five seconds.
1644 // If a bunch of hosts have all been configured with the same name, then they'll all
1645 // conflict and run through the same series of names: name-2, name-3, name-4, etc.,
1646 // up to name-10. After that they'll start adding random increments in the range 1-100,
1647 // so they're more likely to branch out in the available namespace and settle on a set of
1648 // unique names quickly. If after five more tries the host is still conflicting, then we
1649 // may have a serious problem, so we start rate-limiting so we don't melt down the network.
1650 if (m->NumFailedProbes >= 15)
1651 {
1652 m->SuppressProbes = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 5);
1653 LogMsg("Excessive name conflicts (%lu) for %##s (%s); rate limiting in effect",
1654 m->NumFailedProbes, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1655 }
1656 }
1657
1658 mDNSlocal void CompleteRDataUpdate(mDNS *const m, AuthRecord *const rr)
1659 {
1660 RData *OldRData = rr->resrec.rdata;
1661 mDNSu16 OldRDLen = rr->resrec.rdlength;
1662 SetNewRData(&rr->resrec, rr->NewRData, rr->newrdlength); // Update our rdata
1663 rr->NewRData = mDNSNULL; // Clear the NewRData pointer ...
1664 if (rr->UpdateCallback)
1665 rr->UpdateCallback(m, rr, OldRData, OldRDLen); // ... and let the client know
1666 }
1667
1668 // Note: mDNS_Deregister_internal can call a user callback, which may change the record list and/or question list.
1669 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
1670 // Exported so uDNS.c can call this
1671 mDNSexport mStatus mDNS_Deregister_internal(mDNS *const m, AuthRecord *const rr, mDNS_Dereg_type drt)
1672 {
1673 AuthRecord *r2;
1674 mDNSu8 RecordType = rr->resrec.RecordType;
1675 AuthRecord **p = &m->ResourceRecords; // Find this record in our list of active records
1676 mDNSBool dupList = mDNSfalse;
1677
1678 if (RRLocalOnly(rr))
1679 {
1680 AuthGroup *a;
1681 AuthRecord **rp;
1682 const mDNSu32 slot = AuthHashSlot(rr->resrec.name);
1683
1684 a = AuthGroupForRecord(&m->rrauth, slot, &rr->resrec);
1685 if (!a) return mDNSfalse;
1686 rp = &a->members;
1687 while (*rp && *rp != rr) rp=&(*rp)->next;
1688 p = rp;
1689 }
1690 else
1691 {
1692 while (*p && *p != rr) p=&(*p)->next;
1693 }
1694
1695 if (*p)
1696 {
1697 // We found our record on the main list. See if there are any duplicates that need special handling.
1698 if (drt == mDNS_Dereg_conflict) // If this was a conflict, see that all duplicates get the same treatment
1699 {
1700 // Scan for duplicates of rr, and mark them for deregistration at the end of this routine, after we've finished
1701 // deregistering rr. We need to do this scan *before* we give the client the chance to free and reuse the rr memory.
1702 for (r2 = m->DuplicateRecords; r2; r2=r2->next) if (RecordIsLocalDuplicate(r2, rr)) r2->ProbeCount = 0xFF;
1703 }
1704 else
1705 {
1706 // Before we delete the record (and potentially send a goodbye packet)
1707 // first see if we have a record on the duplicate list ready to take over from it.
1708 AuthRecord **d = &m->DuplicateRecords;
1709 while (*d && !RecordIsLocalDuplicate(*d, rr)) d=&(*d)->next;
1710 if (*d)
1711 {
1712 AuthRecord *dup = *d;
1713 debugf("mDNS_Register_internal: Duplicate record %p taking over from %p %##s (%s)",
1714 dup, rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1715 *d = dup->next; // Cut replacement record from DuplicateRecords list
1716 if (RRLocalOnly(rr))
1717 {
1718 dup->next = mDNSNULL;
1719 if (!InsertAuthRecord(m, &m->rrauth, dup)) LogMsg("mDNS_Deregister_internal: ERROR!! cannot insert %s", ARDisplayString(m, dup));
1720 }
1721 else
1722 {
1723 dup->next = rr->next; // And then...
1724 rr->next = dup; // ... splice it in right after the record we're about to delete
1725 }
1726 dup->resrec.RecordType = rr->resrec.RecordType;
1727 dup->ProbeCount = rr->ProbeCount;
1728 dup->ProbeRestartCount = rr->ProbeRestartCount;
1729 dup->AnnounceCount = rr->AnnounceCount;
1730 dup->RequireGoodbye = rr->RequireGoodbye;
1731 dup->AnsweredLocalQ = rr->AnsweredLocalQ;
1732 dup->ImmedAnswer = rr->ImmedAnswer;
1733 dup->ImmedUnicast = rr->ImmedUnicast;
1734 dup->ImmedAdditional = rr->ImmedAdditional;
1735 dup->v4Requester = rr->v4Requester;
1736 dup->v6Requester = rr->v6Requester;
1737 dup->ThisAPInterval = rr->ThisAPInterval;
1738 dup->LastAPTime = rr->LastAPTime;
1739 dup->LastMCTime = rr->LastMCTime;
1740 dup->LastMCInterface = rr->LastMCInterface;
1741 dup->Private = rr->Private;
1742 dup->state = rr->state;
1743 rr->RequireGoodbye = mDNSfalse;
1744 rr->AnsweredLocalQ = mDNSfalse;
1745 }
1746 }
1747 }
1748 else
1749 {
1750 // We didn't find our record on the main list; try the DuplicateRecords list instead.
1751 p = &m->DuplicateRecords;
1752 while (*p && *p != rr) p=&(*p)->next;
1753 // If we found our record on the duplicate list, then make sure we don't send a goodbye for it
1754 if (*p)
1755 {
1756 // Duplicate records are not used for sending wakeups or goodbyes. Hence, deregister them
1757 // immediately. When there is a conflict, we deregister all the conflicting duplicate records
1758 // also that have been marked above in this function. In that case, we come here and if we don't
1759 // deregister (unilink from the DuplicateRecords list), we will be recursing infinitely. Hence,
1760 // clear the HMAC which will cause it to deregister. See <rdar://problem/10380988> for
1761 // details.
1762 rr->WakeUp.HMAC = zeroEthAddr;
1763 rr->RequireGoodbye = mDNSfalse;
1764 rr->resrec.RecordType = kDNSRecordTypeDeregistering;
1765 dupList = mDNStrue;
1766 }
1767 if (*p) debugf("mDNS_Deregister_internal: Deleting DuplicateRecord %p %##s (%s)",
1768 rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1769 }
1770
1771 if (!*p)
1772 {
1773 // No need to log an error message if we already know this is a potentially repeated deregistration
1774 if (drt != mDNS_Dereg_repeat)
1775 LogMsg("mDNS_Deregister_internal: Record %p not found in list %s", rr, ARDisplayString(m,rr));
1776 return(mStatus_BadReferenceErr);
1777 }
1778
1779 // If this is a shared record and we've announced it at least once,
1780 // we need to retract that announcement before we delete the record
1781
1782 // If this is a record (including mDNSInterface_LocalOnly records) for which we've given local-only answers then
1783 // it's tempting to just do "AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse)" here, but that would not not be safe.
1784 // The AnswerAllLocalQuestionsWithLocalAuthRecord routine walks the question list invoking client callbacks, using the "m->CurrentQuestion"
1785 // mechanism to cope with the client callback modifying the question list while that's happening.
1786 // However, mDNS_Deregister could have been called from a client callback (e.g. from the domain enumeration callback FoundDomain)
1787 // which means that the "m->CurrentQuestion" mechanism is already in use to protect that list, so we can't use it twice.
1788 // More generally, if we invoke callbacks from within a client callback, then those callbacks could deregister other
1789 // records, thereby invoking yet more callbacks, without limit.
1790 // The solution is to defer delivering the "Remove" events until mDNS_Execute time, just like we do for sending
1791 // actual goodbye packets.
1792
1793 #ifndef UNICAST_DISABLED
1794 if (AuthRecord_uDNS(rr))
1795 {
1796 if (rr->RequireGoodbye)
1797 {
1798 if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
1799 rr->resrec.RecordType = kDNSRecordTypeDeregistering;
1800 m->LocalRemoveEvents = mDNStrue;
1801 uDNS_DeregisterRecord(m, rr);
1802 // At this point unconditionally we bail out
1803 // Either uDNS_DeregisterRecord will have completed synchronously, and called CompleteDeregistration,
1804 // which calls us back here with RequireGoodbye set to false, or it will have initiated the deregistration
1805 // process and will complete asynchronously. Either way we don't need to do anything more here.
1806 return(mStatus_NoError);
1807 }
1808 // Sometimes the records don't complete proper deregistration i.e., don't wait for a response
1809 // from the server. In that case, if the records have been part of a group update, clear the
1810 // state here. Some recors e.g., AutoTunnel gets reused without ever being completely initialized
1811 rr->updateid = zeroID;
1812
1813 // We defer cleaning up NAT state only after sending goodbyes. This is important because
1814 // RecordRegistrationGotZoneData guards against creating NAT state if clientContext is non-NULL.
1815 // This happens today when we turn on/off interface where we get multiple network transitions
1816 // and RestartRecordGetZoneData triggers re-registration of the resource records even though
1817 // they may be in Registered state which causes NAT information to be setup multiple times. Defering
1818 // the cleanup here keeps clientContext non-NULL and hence prevents that. Note that cleaning up
1819 // NAT state here takes care of the case where we did not send goodbyes at all.
1820 if (rr->NATinfo.clientContext)
1821 {
1822 mDNS_StopNATOperation_internal(m, &rr->NATinfo);
1823 rr->NATinfo.clientContext = mDNSNULL;
1824 }
1825 if (rr->nta) { CancelGetZoneData(m, rr->nta); rr->nta = mDNSNULL; }
1826 if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
1827 }
1828 #endif // UNICAST_DISABLED
1829
1830 if (RecordType == kDNSRecordTypeUnregistered)
1831 LogMsg("mDNS_Deregister_internal: %s already marked kDNSRecordTypeUnregistered", ARDisplayString(m, rr));
1832 else if (RecordType == kDNSRecordTypeDeregistering)
1833 {
1834 LogMsg("mDNS_Deregister_internal: %s already marked kDNSRecordTypeDeregistering", ARDisplayString(m, rr));
1835 return(mStatus_BadReferenceErr);
1836 }
1837
1838 // <rdar://problem/7457925> Local-only questions don't get remove events for unique records
1839 // We may want to consider changing this code so that we generate local-only question "rmv"
1840 // events (and maybe goodbye packets too) for unique records as well as for shared records
1841 // Note: If we change the logic for this "if" statement, need to ensure that the code in
1842 // CompleteDeregistration() sets the appropriate state variables to gaurantee that "else"
1843 // clause will execute here and the record will be cut from the list.
1844 if (rr->WakeUp.HMAC.l[0] ||
1845 (RecordType == kDNSRecordTypeShared && (rr->RequireGoodbye || rr->AnsweredLocalQ)))
1846 {
1847 verbosedebugf("mDNS_Deregister_internal: Starting deregistration for %s", ARDisplayString(m, rr));
1848 rr->resrec.RecordType = kDNSRecordTypeDeregistering;
1849 rr->resrec.rroriginalttl = 0;
1850 rr->AnnounceCount = rr->WakeUp.HMAC.l[0] ? WakeupCount : (drt == mDNS_Dereg_rapid) ? 1 : GoodbyeCount;
1851 rr->ThisAPInterval = mDNSPlatformOneSecond * 2;
1852 rr->LastAPTime = m->timenow - rr->ThisAPInterval;
1853 m->LocalRemoveEvents = mDNStrue;
1854 if (m->NextScheduledResponse - (m->timenow + mDNSPlatformOneSecond/10) >= 0)
1855 m->NextScheduledResponse = (m->timenow + mDNSPlatformOneSecond/10);
1856 }
1857 else
1858 {
1859 if (!dupList && RRLocalOnly(rr))
1860 {
1861 AuthGroup *ag = RemoveAuthRecord(m, &m->rrauth, rr);
1862 if (ag->NewLocalOnlyRecords == rr) ag->NewLocalOnlyRecords = rr->next;
1863 }
1864 else
1865 {
1866 *p = rr->next; // Cut this record from the list
1867 if (m->NewLocalRecords == rr) m->NewLocalRecords = rr->next;
1868 DecrementAutoTargetServices(m, rr);
1869 }
1870 // If someone is about to look at this, bump the pointer forward
1871 if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
1872 rr->next = mDNSNULL;
1873
1874 // Should we generate local remove events here?
1875 // i.e. something like:
1876 // if (rr->AnsweredLocalQ) { AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse); rr->AnsweredLocalQ = mDNSfalse; }
1877
1878 verbosedebugf("mDNS_Deregister_internal: Deleting record for %s", ARDisplayString(m, rr));
1879 rr->resrec.RecordType = kDNSRecordTypeUnregistered;
1880
1881 if ((drt == mDNS_Dereg_conflict || drt == mDNS_Dereg_repeat) && RecordType == kDNSRecordTypeShared)
1882 debugf("mDNS_Deregister_internal: Cannot have a conflict on a shared record! %##s (%s)",
1883 rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1884
1885 // If we have an update queued up which never executed, give the client a chance to free that memory
1886 if (rr->NewRData) CompleteRDataUpdate(m, rr); // Update our rdata, clear the NewRData pointer, and return memory to the client
1887
1888
1889 // CAUTION: MUST NOT do anything more with rr after calling rr->Callback(), because the client's callback function
1890 // is allowed to do anything, including starting/stopping queries, registering/deregistering records, etc.
1891 // In this case the likely client action to the mStatus_MemFree message is to free the memory,
1892 // so any attempt to touch rr after this is likely to lead to a crash.
1893 if (drt != mDNS_Dereg_conflict)
1894 {
1895 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
1896 LogInfo("mDNS_Deregister_internal: callback with mStatus_MemFree for %s", ARDisplayString(m, rr));
1897 if (rr->RecordCallback)
1898 rr->RecordCallback(m, rr, mStatus_MemFree); // MUST NOT touch rr after this
1899 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
1900 }
1901 else
1902 {
1903 RecordProbeFailure(m, rr);
1904 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
1905 if (rr->RecordCallback)
1906 rr->RecordCallback(m, rr, mStatus_NameConflict); // MUST NOT touch rr after this
1907 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
1908 // Now that we've finished deregistering rr, check our DuplicateRecords list for any that we marked previously.
1909 // Note that with all the client callbacks going on, by the time we get here all the
1910 // records we marked may have been explicitly deregistered by the client anyway.
1911 r2 = m->DuplicateRecords;
1912 while (r2)
1913 {
1914 if (r2->ProbeCount != 0xFF)
1915 {
1916 r2 = r2->next;
1917 }
1918 else
1919 {
1920 #if APPLE_OSX_mDNSResponder
1921 // See if this record was also registered with any D2D plugins.
1922 D2D_stop_advertising_record(r2);
1923 #endif
1924 mDNS_Deregister_internal(m, r2, mDNS_Dereg_conflict);
1925 // As this is a duplicate record, it will be unlinked from the list
1926 // immediately
1927 r2 = m->DuplicateRecords;
1928 }
1929 }
1930 }
1931 }
1932 mDNS_UpdateAllowSleep(m);
1933 return(mStatus_NoError);
1934 }
1935
1936 // ***************************************************************************
1937 #if COMPILER_LIKES_PRAGMA_MARK
1938 #pragma mark -
1939 #pragma mark - Packet Sending Functions
1940 #endif
1941
1942 mDNSlocal void AddRecordToResponseList(AuthRecord ***nrpp, AuthRecord *rr, AuthRecord *add)
1943 {
1944 if (rr->NextResponse == mDNSNULL && *nrpp != &rr->NextResponse)
1945 {
1946 **nrpp = rr;
1947 // NR_AdditionalTo must point to a record with NR_AnswerTo set (and not NR_AdditionalTo)
1948 // If 'add' does not meet this requirement, then follow its NR_AdditionalTo pointer to a record that does
1949 // The referenced record will definitely be acceptable (by recursive application of this rule)
1950 if (add && add->NR_AdditionalTo) add = add->NR_AdditionalTo;
1951 rr->NR_AdditionalTo = add;
1952 *nrpp = &rr->NextResponse;
1953 }
1954 debugf("AddRecordToResponseList: %##s (%s) already in list", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
1955 }
1956
1957 mDNSlocal void AddAdditionalsToResponseList(mDNS *const m, AuthRecord *ResponseRecords, AuthRecord ***nrpp, const mDNSInterfaceID InterfaceID)
1958 {
1959 AuthRecord *rr, *rr2;
1960 for (rr=ResponseRecords; rr; rr=rr->NextResponse) // For each record we plan to put
1961 {
1962 // (Note: This is an "if", not a "while". If we add a record, we'll find it again
1963 // later in the "for" loop, and we will follow further "additional" links then.)
1964 if (rr->Additional1 && ResourceRecordIsValidInterfaceAnswer(rr->Additional1, InterfaceID))
1965 AddRecordToResponseList(nrpp, rr->Additional1, rr);
1966
1967 if (rr->Additional2 && ResourceRecordIsValidInterfaceAnswer(rr->Additional2, InterfaceID))
1968 AddRecordToResponseList(nrpp, rr->Additional2, rr);
1969
1970 // For SRV records, automatically add the Address record(s) for the target host
1971 if (rr->resrec.rrtype == kDNSType_SRV)
1972 {
1973 for (rr2=m->ResourceRecords; rr2; rr2=rr2->next) // Scan list of resource records
1974 if (RRTypeIsAddressType(rr2->resrec.rrtype) && // For all address records (A/AAAA) ...
1975 ResourceRecordIsValidInterfaceAnswer(rr2, InterfaceID) && // ... which are valid for answer ...
1976 rr->resrec.rdatahash == rr2->resrec.namehash && // ... whose name is the name of the SRV target
1977 SameDomainName(&rr->resrec.rdata->u.srv.target, rr2->resrec.name))
1978 AddRecordToResponseList(nrpp, rr2, rr);
1979 }
1980 else if (RRTypeIsAddressType(rr->resrec.rrtype)) // For A or AAAA, put counterpart as additional
1981 {
1982 for (rr2=m->ResourceRecords; rr2; rr2=rr2->next) // Scan list of resource records
1983 if (RRTypeIsAddressType(rr2->resrec.rrtype) && // For all address records (A/AAAA) ...
1984 ResourceRecordIsValidInterfaceAnswer(rr2, InterfaceID) && // ... which are valid for answer ...
1985 rr->resrec.namehash == rr2->resrec.namehash && // ... and have the same name
1986 SameDomainName(rr->resrec.name, rr2->resrec.name))
1987 AddRecordToResponseList(nrpp, rr2, rr);
1988 }
1989 else if (rr->resrec.rrtype == kDNSType_PTR) // For service PTR, see if we want to add DeviceInfo record
1990 {
1991 if (ResourceRecordIsValidInterfaceAnswer(&m->DeviceInfo, InterfaceID) &&
1992 SameDomainLabel(rr->resrec.rdata->u.name.c, m->DeviceInfo.resrec.name->c))
1993 AddRecordToResponseList(nrpp, &m->DeviceInfo, rr);
1994 }
1995 }
1996 }
1997
1998 mDNSlocal int AnonInfoSpace(AnonymousInfo *info)
1999 {
2000 ResourceRecord *rr = info->nsec3RR;
2001
2002 // 2 bytes for compressed name + type (2) class (2) TTL (4) rdlength (2) rdata (n)
2003 return (2 + 10 + rr->rdlength);
2004 }
2005
2006 mDNSlocal void SendDelayedUnicastResponse(mDNS *const m, const mDNSAddr *const dest, const mDNSInterfaceID InterfaceID)
2007 {
2008 AuthRecord *rr;
2009 AuthRecord *ResponseRecords = mDNSNULL;
2010 AuthRecord **nrp = &ResponseRecords;
2011 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
2012 int AnoninfoSpace = 0;
2013
2014 // Make a list of all our records that need to be unicast to this destination
2015 for (rr = m->ResourceRecords; rr; rr=rr->next)
2016 {
2017 // If we find we can no longer unicast this answer, clear ImmedUnicast
2018 if (rr->ImmedAnswer == mDNSInterfaceMark ||
2019 mDNSSameIPv4Address(rr->v4Requester, onesIPv4Addr) ||
2020 mDNSSameIPv6Address(rr->v6Requester, onesIPv6Addr) )
2021 rr->ImmedUnicast = mDNSfalse;
2022
2023 if (rr->ImmedUnicast && rr->ImmedAnswer == InterfaceID)
2024 {
2025 if ((dest->type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->v4Requester, dest->ip.v4)) ||
2026 (dest->type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->v6Requester, dest->ip.v6)))
2027 {
2028 rr->ImmedAnswer = mDNSNULL; // Clear the state fields
2029 rr->ImmedUnicast = mDNSfalse;
2030 rr->v4Requester = zerov4Addr;
2031 rr->v6Requester = zerov6Addr;
2032
2033 // Only sent records registered for P2P over P2P interfaces
2034 if (intf && !mDNSPlatformValidRecordForInterface(rr, intf->InterfaceID))
2035 {
2036 continue;
2037 }
2038
2039 if (rr->NextResponse == mDNSNULL && nrp != &rr->NextResponse) // rr->NR_AnswerTo
2040 {
2041 rr->NR_AnswerTo = NR_AnswerMulticast;
2042 *nrp = rr;
2043 nrp = &rr->NextResponse;
2044 }
2045 }
2046 }
2047 }
2048
2049 AddAdditionalsToResponseList(m, ResponseRecords, &nrp, InterfaceID);
2050
2051 while (ResponseRecords)
2052 {
2053 mDNSu8 *responseptr = m->omsg.data;
2054 mDNSu8 *newptr;
2055 InitializeDNSMessage(&m->omsg.h, zeroID, ResponseFlags);
2056
2057 // Put answers in the packet
2058 while (ResponseRecords && ResponseRecords->NR_AnswerTo)
2059 {
2060 rr = ResponseRecords;
2061 if (rr->resrec.AnonInfo)
2062 {
2063 AnoninfoSpace += AnonInfoSpace(rr->resrec.AnonInfo);
2064 rr->resrec.AnonInfo->SendNow = mDNSInterfaceMark;
2065 }
2066 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
2067 rr->resrec.rrclass |= kDNSClass_UniqueRRSet; // Temporarily set the cache flush bit so PutResourceRecord will set it
2068
2069 // Retract the limit by AnoninfoSpace which we need to put the AnoInfo option.
2070 newptr = PutResourceRecordTTLWithLimit(&m->omsg, responseptr, &m->omsg.h.numAnswers, &rr->resrec, rr->resrec.rroriginalttl,
2071 m->omsg.data + (AllowedRRSpace(&m->omsg) - AnoninfoSpace));
2072
2073 rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; // Make sure to clear cache flush bit back to normal state
2074 if (!newptr && m->omsg.h.numAnswers)
2075 {
2076 break; // If packet full, send it now
2077 }
2078 if (newptr) responseptr = newptr;
2079 ResponseRecords = rr->NextResponse;
2080 rr->NextResponse = mDNSNULL;
2081 rr->NR_AnswerTo = mDNSNULL;
2082 rr->NR_AdditionalTo = mDNSNULL;
2083 rr->RequireGoodbye = mDNStrue;
2084 }
2085
2086 // We have reserved the space for AnonInfo option. PutResourceRecord uses the
2087 // standard limit (AllowedRRSpace) and we should have space now.
2088 for (rr = m->ResourceRecords; rr; rr=rr->next)
2089 {
2090 if (rr->resrec.AnonInfo && rr->resrec.AnonInfo->SendNow == mDNSInterfaceMark)
2091 {
2092 ResourceRecord *nsec3RR = rr->resrec.AnonInfo->nsec3RR;
2093
2094 newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAuthorities, nsec3RR);
2095 if (newptr)
2096 {
2097 responseptr = newptr;
2098 debugf("SendDelayedUnicastResponse: Added NSEC3 Record %s on %p", RRDisplayString(m, nsec3RR), intf->InterfaceID);
2099 }
2100 else
2101 {
2102 // We allocated space and we should not fail. Don't break, we need to clear the SendNow flag.
2103 LogMsg("SendDelayedUnicastResponse: ERROR!! Cannot Add NSEC3 Record %s on %p", RRDisplayString(m, nsec3RR), intf->InterfaceID);
2104 }
2105 rr->resrec.AnonInfo->SendNow = mDNSNULL;
2106 }
2107 }
2108
2109 // Add additionals, if there's space
2110 while (ResponseRecords && !ResponseRecords->NR_AnswerTo)
2111 {
2112 rr = ResponseRecords;
2113 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
2114 rr->resrec.rrclass |= kDNSClass_UniqueRRSet; // Temporarily set the cache flush bit so PutResourceRecord will set it
2115 newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAdditionals, &rr->resrec);
2116 rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; // Make sure to clear cache flush bit back to normal state
2117
2118 if (newptr) responseptr = newptr;
2119 if (newptr && m->omsg.h.numAnswers) rr->RequireGoodbye = mDNStrue;
2120 else if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask) rr->ImmedAnswer = mDNSInterfaceMark;
2121 ResponseRecords = rr->NextResponse;
2122 rr->NextResponse = mDNSNULL;
2123 rr->NR_AnswerTo = mDNSNULL;
2124 rr->NR_AdditionalTo = mDNSNULL;
2125 }
2126
2127 if (m->omsg.h.numAnswers)
2128 mDNSSendDNSMessage(m, &m->omsg, responseptr, InterfaceID, mDNSNULL, dest, MulticastDNSPort, mDNSNULL, mDNSNULL, mDNSfalse);
2129 }
2130 }
2131
2132 // CompleteDeregistration guarantees that on exit the record will have been cut from the m->ResourceRecords list
2133 // and the client's mStatus_MemFree callback will have been invoked
2134 mDNSexport void CompleteDeregistration(mDNS *const m, AuthRecord *rr)
2135 {
2136 LogInfo("CompleteDeregistration: called for Resource record %s", ARDisplayString(m, rr));
2137 // Clearing rr->RequireGoodbye signals mDNS_Deregister_internal() that
2138 // it should go ahead and immediately dispose of this registration
2139 rr->resrec.RecordType = kDNSRecordTypeShared;
2140 rr->RequireGoodbye = mDNSfalse;
2141 rr->WakeUp.HMAC = zeroEthAddr;
2142 if (rr->AnsweredLocalQ) { AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse); rr->AnsweredLocalQ = mDNSfalse; }
2143 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal); // Don't touch rr after this
2144 }
2145
2146 // DiscardDeregistrations is used on shutdown and sleep to discard (forcibly and immediately)
2147 // any deregistering records that remain in the m->ResourceRecords list.
2148 // DiscardDeregistrations calls mDNS_Deregister_internal which can call a user callback,
2149 // which may change the record list and/or question list.
2150 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
2151 mDNSlocal void DiscardDeregistrations(mDNS *const m)
2152 {
2153 if (m->CurrentRecord)
2154 LogMsg("DiscardDeregistrations ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
2155 m->CurrentRecord = m->ResourceRecords;
2156
2157 while (m->CurrentRecord)
2158 {
2159 AuthRecord *rr = m->CurrentRecord;
2160 if (!AuthRecord_uDNS(rr) && rr->resrec.RecordType == kDNSRecordTypeDeregistering)
2161 CompleteDeregistration(m, rr); // Don't touch rr after this
2162 else
2163 m->CurrentRecord = rr->next;
2164 }
2165 }
2166
2167 mDNSlocal mStatus GetLabelDecimalValue(const mDNSu8 *const src, mDNSu8 *dst)
2168 {
2169 int i, val = 0;
2170 if (src[0] < 1 || src[0] > 3) return(mStatus_Invalid);
2171 for (i=1; i<=src[0]; i++)
2172 {
2173 if (src[i] < '0' || src[i] > '9') return(mStatus_Invalid);
2174 val = val * 10 + src[i] - '0';
2175 }
2176 if (val > 255) return(mStatus_Invalid);
2177 *dst = (mDNSu8)val;
2178 return(mStatus_NoError);
2179 }
2180
2181 mDNSlocal mStatus GetIPv4FromName(mDNSAddr *const a, const domainname *const name)
2182 {
2183 int skip = CountLabels(name) - 6;
2184 if (skip < 0) { LogMsg("GetIPFromName: Need six labels in IPv4 reverse mapping name %##s", name); return mStatus_Invalid; }
2185 if (GetLabelDecimalValue(SkipLeadingLabels(name, skip+3)->c, &a->ip.v4.b[0]) ||
2186 GetLabelDecimalValue(SkipLeadingLabels(name, skip+2)->c, &a->ip.v4.b[1]) ||
2187 GetLabelDecimalValue(SkipLeadingLabels(name, skip+1)->c, &a->ip.v4.b[2]) ||
2188 GetLabelDecimalValue(SkipLeadingLabels(name, skip+0)->c, &a->ip.v4.b[3])) return mStatus_Invalid;
2189 a->type = mDNSAddrType_IPv4;
2190 return(mStatus_NoError);
2191 }
2192
2193 #define HexVal(X) ( ((X) >= '0' && (X) <= '9') ? ((X) - '0' ) : \
2194 ((X) >= 'A' && (X) <= 'F') ? ((X) - 'A' + 10) : \
2195 ((X) >= 'a' && (X) <= 'f') ? ((X) - 'a' + 10) : -1)
2196
2197 mDNSlocal mStatus GetIPv6FromName(mDNSAddr *const a, const domainname *const name)
2198 {
2199 int i, h, l;
2200 const domainname *n;
2201
2202 int skip = CountLabels(name) - 34;
2203 if (skip < 0) { LogMsg("GetIPFromName: Need 34 labels in IPv6 reverse mapping name %##s", name); return mStatus_Invalid; }
2204
2205 n = SkipLeadingLabels(name, skip);
2206 for (i=0; i<16; i++)
2207 {
2208 if (n->c[0] != 1) return mStatus_Invalid;
2209 l = HexVal(n->c[1]);
2210 n = (const domainname *)(n->c + 2);
2211
2212 if (n->c[0] != 1) return mStatus_Invalid;
2213 h = HexVal(n->c[1]);
2214 n = (const domainname *)(n->c + 2);
2215
2216 if (l<0 || h<0) return mStatus_Invalid;
2217 a->ip.v6.b[15-i] = (mDNSu8)((h << 4) | l);
2218 }
2219
2220 a->type = mDNSAddrType_IPv6;
2221 return(mStatus_NoError);
2222 }
2223
2224 mDNSlocal mDNSs32 ReverseMapDomainType(const domainname *const name)
2225 {
2226 int skip = CountLabels(name) - 2;
2227 if (skip >= 0)
2228 {
2229 const domainname *suffix = SkipLeadingLabels(name, skip);
2230 if (SameDomainName(suffix, (const domainname*)"\x7" "in-addr" "\x4" "arpa")) return mDNSAddrType_IPv4;
2231 if (SameDomainName(suffix, (const domainname*)"\x3" "ip6" "\x4" "arpa")) return mDNSAddrType_IPv6;
2232 }
2233 return(mDNSAddrType_None);
2234 }
2235
2236 mDNSlocal void SendARP(mDNS *const m, const mDNSu8 op, const AuthRecord *const rr,
2237 const mDNSv4Addr *const spa, const mDNSEthAddr *const tha, const mDNSv4Addr *const tpa, const mDNSEthAddr *const dst)
2238 {
2239 int i;
2240 mDNSu8 *ptr = m->omsg.data;
2241 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
2242 if (!intf) { LogMsg("SendARP: No interface with InterfaceID %p found %s", rr->resrec.InterfaceID, ARDisplayString(m,rr)); return; }
2243
2244 // 0x00 Destination address
2245 for (i=0; i<6; i++) *ptr++ = dst->b[i];
2246
2247 // 0x06 Source address (Note: Since we don't currently set the BIOCSHDRCMPLT option, BPF will fill in the real interface address for us)
2248 for (i=0; i<6; i++) *ptr++ = intf->MAC.b[0];
2249
2250 // 0x0C ARP Ethertype (0x0806)
2251 *ptr++ = 0x08; *ptr++ = 0x06;
2252
2253 // 0x0E ARP header
2254 *ptr++ = 0x00; *ptr++ = 0x01; // Hardware address space; Ethernet = 1
2255 *ptr++ = 0x08; *ptr++ = 0x00; // Protocol address space; IP = 0x0800
2256 *ptr++ = 6; // Hardware address length
2257 *ptr++ = 4; // Protocol address length
2258 *ptr++ = 0x00; *ptr++ = op; // opcode; Request = 1, Response = 2
2259
2260 // 0x16 Sender hardware address (our MAC address)
2261 for (i=0; i<6; i++) *ptr++ = intf->MAC.b[i];
2262
2263 // 0x1C Sender protocol address
2264 for (i=0; i<4; i++) *ptr++ = spa->b[i];
2265
2266 // 0x20 Target hardware address
2267 for (i=0; i<6; i++) *ptr++ = tha->b[i];
2268
2269 // 0x26 Target protocol address
2270 for (i=0; i<4; i++) *ptr++ = tpa->b[i];
2271
2272 // 0x2A Total ARP Packet length 42 bytes
2273 mDNSPlatformSendRawPacket(m->omsg.data, ptr, rr->resrec.InterfaceID);
2274 }
2275
2276 mDNSlocal mDNSu16 CheckSum(const void *const data, mDNSs32 length, mDNSu32 sum)
2277 {
2278 const mDNSu16 *ptr = data;
2279 while (length > 0) { length -= 2; sum += *ptr++; }
2280 sum = (sum & 0xFFFF) + (sum >> 16);
2281 sum = (sum & 0xFFFF) + (sum >> 16);
2282 return(sum != 0xFFFF ? sum : 0);
2283 }
2284
2285 mDNSlocal mDNSu16 IPv6CheckSum(const mDNSv6Addr *const src, const mDNSv6Addr *const dst, const mDNSu8 protocol, const void *const data, const mDNSu32 length)
2286 {
2287 IPv6PseudoHeader ph;
2288 ph.src = *src;
2289 ph.dst = *dst;
2290 ph.len.b[0] = length >> 24;
2291 ph.len.b[1] = length >> 16;
2292 ph.len.b[2] = length >> 8;
2293 ph.len.b[3] = length;
2294 ph.pro.b[0] = 0;
2295 ph.pro.b[1] = 0;
2296 ph.pro.b[2] = 0;
2297 ph.pro.b[3] = protocol;
2298 return CheckSum(&ph, sizeof(ph), CheckSum(data, length, 0));
2299 }
2300
2301 mDNSlocal void SendNDP(mDNS *const m, const mDNSu8 op, const mDNSu8 flags, const AuthRecord *const rr,
2302 const mDNSv6Addr *const spa, const mDNSEthAddr *const tha, const mDNSv6Addr *const tpa, const mDNSEthAddr *const dst)
2303 {
2304 int i;
2305 mDNSOpaque16 checksum;
2306 mDNSu8 *ptr = m->omsg.data;
2307 // Some recipient hosts seem to ignore Neighbor Solicitations if the IPv6-layer destination address is not the
2308 // appropriate IPv6 solicited node multicast address, so we use that IPv6-layer destination address, even though
2309 // at the Ethernet-layer we unicast the packet to the intended target, to avoid wasting network bandwidth.
2310 const mDNSv6Addr mc = { { 0xFF,0x02,0x00,0x00, 0,0,0,0, 0,0,0,1, 0xFF,tpa->b[0xD],tpa->b[0xE],tpa->b[0xF] } };
2311 const mDNSv6Addr *const v6dst = (op == NDP_Sol) ? &mc : tpa;
2312 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
2313 if (!intf) { LogMsg("SendNDP: No interface with InterfaceID %p found %s", rr->resrec.InterfaceID, ARDisplayString(m,rr)); return; }
2314
2315 // 0x00 Destination address
2316 for (i=0; i<6; i++) *ptr++ = dst->b[i];
2317 // Right now we only send Neighbor Solicitations to verify whether the host we're proxying for has gone to sleep yet.
2318 // Since we know who we're looking for, we send it via Ethernet-layer unicast, rather than bothering every host on the
2319 // link with a pointless link-layer multicast.
2320 // Should we want to send traditional Neighbor Solicitations in the future, where we really don't know in advance what
2321 // Ethernet-layer address we're looking for, we'll need to send to the appropriate Ethernet-layer multicast address:
2322 // *ptr++ = 0x33;
2323 // *ptr++ = 0x33;
2324 // *ptr++ = 0xFF;
2325 // *ptr++ = tpa->b[0xD];
2326 // *ptr++ = tpa->b[0xE];
2327 // *ptr++ = tpa->b[0xF];
2328
2329 // 0x06 Source address (Note: Since we don't currently set the BIOCSHDRCMPLT option, BPF will fill in the real interface address for us)
2330 for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
2331
2332 // 0x0C IPv6 Ethertype (0x86DD)
2333 *ptr++ = 0x86; *ptr++ = 0xDD;
2334
2335 // 0x0E IPv6 header
2336 *ptr++ = 0x60; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00; // Version, Traffic Class, Flow Label
2337 *ptr++ = 0x00; *ptr++ = 0x20; // Length
2338 *ptr++ = 0x3A; // Protocol == ICMPv6
2339 *ptr++ = 0xFF; // Hop Limit
2340
2341 // 0x16 Sender IPv6 address
2342 for (i=0; i<16; i++) *ptr++ = spa->b[i];
2343
2344 // 0x26 Destination IPv6 address
2345 for (i=0; i<16; i++) *ptr++ = v6dst->b[i];
2346
2347 // 0x36 NDP header
2348 *ptr++ = op; // 0x87 == Neighbor Solicitation, 0x88 == Neighbor Advertisement
2349 *ptr++ = 0x00; // Code
2350 *ptr++ = 0x00; *ptr++ = 0x00; // Checksum placeholder (0x38, 0x39)
2351 *ptr++ = flags;
2352 *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00;
2353
2354 if (op == NDP_Sol) // Neighbor Solicitation. The NDP "target" is the address we seek.
2355 {
2356 // 0x3E NDP target.
2357 for (i=0; i<16; i++) *ptr++ = tpa->b[i];
2358 // 0x4E Source Link-layer Address
2359 // <http://www.ietf.org/rfc/rfc2461.txt>
2360 // MUST NOT be included when the source IP address is the unspecified address.
2361 // Otherwise, on link layers that have addresses this option MUST be included
2362 // in multicast solicitations and SHOULD be included in unicast solicitations.
2363 if (!mDNSIPv6AddressIsZero(*spa))
2364 {
2365 *ptr++ = NDP_SrcLL; // Option Type 1 == Source Link-layer Address
2366 *ptr++ = 0x01; // Option length 1 (in units of 8 octets)
2367 for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
2368 }
2369 }
2370 else // Neighbor Advertisement. The NDP "target" is the address we're giving information about.
2371 {
2372 // 0x3E NDP target.
2373 for (i=0; i<16; i++) *ptr++ = spa->b[i];
2374 // 0x4E Target Link-layer Address
2375 *ptr++ = NDP_TgtLL; // Option Type 2 == Target Link-layer Address
2376 *ptr++ = 0x01; // Option length 1 (in units of 8 octets)
2377 for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
2378 }
2379
2380 // 0x4E or 0x56 Total NDP Packet length 78 or 86 bytes
2381 m->omsg.data[0x13] = ptr - &m->omsg.data[0x36]; // Compute actual length
2382 checksum.NotAnInteger = ~IPv6CheckSum(spa, v6dst, 0x3A, &m->omsg.data[0x36], m->omsg.data[0x13]);
2383 m->omsg.data[0x38] = checksum.b[0];
2384 m->omsg.data[0x39] = checksum.b[1];
2385
2386 mDNSPlatformSendRawPacket(m->omsg.data, ptr, rr->resrec.InterfaceID);
2387 }
2388
2389 mDNSlocal void SetupTracerOpt(const mDNS *const m, rdataOPT *const Trace)
2390 {
2391 mDNSu32 DNS_VERS = _DNS_SD_H;
2392 Trace->u.tracer.platf = m->mDNS_plat;
2393 Trace->u.tracer.mDNSv = DNS_VERS;
2394
2395 Trace->opt = kDNSOpt_Trace;
2396 Trace->optlen = DNSOpt_TraceData_Space - 4;
2397 }
2398
2399 mDNSlocal void SetupOwnerOpt(const mDNS *const m, const NetworkInterfaceInfo *const intf, rdataOPT *const owner)
2400 {
2401 owner->u.owner.vers = 0;
2402 owner->u.owner.seq = m->SleepSeqNum;
2403 owner->u.owner.HMAC = m->PrimaryMAC;
2404 owner->u.owner.IMAC = intf->MAC;
2405 owner->u.owner.password = zeroEthAddr;
2406
2407 // Don't try to compute the optlen until *after* we've set up the data fields
2408 // Right now the DNSOpt_Owner_Space macro does not depend on the owner->u.owner being set up correctly, but in the future it might
2409 owner->opt = kDNSOpt_Owner;
2410 owner->optlen = DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) - 4;
2411 }
2412
2413 mDNSlocal void GrantUpdateCredit(AuthRecord *rr)
2414 {
2415 if (++rr->UpdateCredits >= kMaxUpdateCredits) rr->NextUpdateCredit = 0;
2416 else rr->NextUpdateCredit = NonZeroTime(rr->NextUpdateCredit + kUpdateCreditRefreshInterval);
2417 }
2418
2419 mDNSlocal mDNSBool ShouldSendGoodbyesBeforeSleep(mDNS *const m, const NetworkInterfaceInfo *intf, AuthRecord *rr)
2420 {
2421 // If there are no sleep proxies, we set the state to SleepState_Sleeping explicitly
2422 // and hence there is no need to check for Transfering state. But if we have sleep
2423 // proxies and partially sending goodbyes for some records, we will be in Transfering
2424 // state and hence need to make sure that we send goodbyes in that case too. Checking whether
2425 // we are not awake handles both cases.
2426 if ((rr->AuthFlags & AuthFlagsWakeOnly) && (m->SleepState != SleepState_Awake))
2427 {
2428 debugf("ShouldSendGoodbyesBeforeSleep: marking for goodbye", ARDisplayString(m, rr));
2429 return mDNStrue;
2430 }
2431
2432 if (m->SleepState != SleepState_Sleeping)
2433 return mDNSfalse;
2434
2435 // If we are going to sleep and in SleepState_Sleeping, SendGoodbyes on the interface tell you
2436 // whether you can send goodbyes or not.
2437 if (!intf->SendGoodbyes)
2438 {
2439 debugf("ShouldSendGoodbyesBeforeSleep: not sending goodbye %s, int %p", ARDisplayString(m, rr), intf->InterfaceID);
2440 return mDNSfalse;
2441 }
2442 else
2443 {
2444 debugf("ShouldSendGoodbyesBeforeSleep: sending goodbye %s, int %p", ARDisplayString(m, rr), intf->InterfaceID);
2445 return mDNStrue;
2446 }
2447 }
2448
2449 mDNSlocal mDNSBool IsInterfaceValidForAuthRecord(const AuthRecord *ar, mDNSInterfaceID InterfaceID)
2450 {
2451 mDNSBool result;
2452
2453 if (ar->resrec.InterfaceID == mDNSInterface_Any)
2454 {
2455 result = mDNSPlatformValidRecordForInterface(ar, InterfaceID);
2456 }
2457 else
2458 {
2459 result = (ar->resrec.InterfaceID == InterfaceID);
2460 }
2461
2462 return(result);
2463 }
2464
2465 // Note about acceleration of announcements to facilitate automatic coalescing of
2466 // multiple independent threads of announcements into a single synchronized thread:
2467 // The announcements in the packet may be at different stages of maturity;
2468 // One-second interval, two-second interval, four-second interval, and so on.
2469 // After we've put in all the announcements that are due, we then consider
2470 // whether there are other nearly-due announcements that are worth accelerating.
2471 // To be eligible for acceleration, a record MUST NOT be older (further along
2472 // its timeline) than the most mature record we've already put in the packet.
2473 // In other words, younger records can have their timelines accelerated to catch up
2474 // with their elder bretheren; this narrows the age gap and helps them eventually get in sync.
2475 // Older records cannot have their timelines accelerated; this would just widen
2476 // the gap between them and their younger bretheren and get them even more out of sync.
2477
2478 // Note: SendResponses calls mDNS_Deregister_internal which can call a user callback, which may change
2479 // the record list and/or question list.
2480 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
2481 mDNSlocal void SendResponses(mDNS *const m)
2482 {
2483 int pktcount = 0;
2484 AuthRecord *rr, *r2;
2485 mDNSs32 maxExistingAnnounceInterval = 0;
2486 const NetworkInterfaceInfo *intf = GetFirstActiveInterface(m->HostInterfaces);
2487
2488 m->NextScheduledResponse = m->timenow + 0x78000000;
2489
2490 if (m->SleepState == SleepState_Transferring) RetrySPSRegistrations(m);
2491
2492 for (rr = m->ResourceRecords; rr; rr=rr->next)
2493 if (rr->ImmedUnicast)
2494 {
2495 mDNSAddr v4 = { mDNSAddrType_IPv4, {{{0}}} };
2496 mDNSAddr v6 = { mDNSAddrType_IPv6, {{{0}}} };
2497 v4.ip.v4 = rr->v4Requester;
2498 v6.ip.v6 = rr->v6Requester;
2499 if (!mDNSIPv4AddressIsZero(rr->v4Requester)) SendDelayedUnicastResponse(m, &v4, rr->ImmedAnswer);
2500 if (!mDNSIPv6AddressIsZero(rr->v6Requester)) SendDelayedUnicastResponse(m, &v6, rr->ImmedAnswer);
2501 if (rr->ImmedUnicast)
2502 {
2503 LogMsg("SendResponses: ERROR: rr->ImmedUnicast still set: %s", ARDisplayString(m, rr));
2504 rr->ImmedUnicast = mDNSfalse;
2505 }
2506 }
2507
2508 // ***
2509 // *** 1. Setup: Set the SendRNow and ImmedAnswer fields to indicate which interface(s) the records need to be sent on
2510 // ***
2511
2512 // Run through our list of records, and decide which ones we're going to announce on all interfaces
2513 for (rr = m->ResourceRecords; rr; rr=rr->next)
2514 {
2515 while (rr->NextUpdateCredit && m->timenow - rr->NextUpdateCredit >= 0) GrantUpdateCredit(rr);
2516 if (TimeToAnnounceThisRecord(rr, m->timenow))
2517 {
2518 if (rr->resrec.RecordType == kDNSRecordTypeDeregistering)
2519 {
2520 if (!rr->WakeUp.HMAC.l[0])
2521 {
2522 if (rr->AnnounceCount) rr->ImmedAnswer = mDNSInterfaceMark; // Send goodbye packet on all interfaces
2523 }
2524 else
2525 {
2526 mDNSBool unicastOnly;
2527 LogSPS("SendResponses: Sending wakeup %2d for %.6a %s", rr->AnnounceCount-3, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
2528 unicastOnly = ((rr->AnnounceCount == WakeupCount) || (rr->AnnounceCount == WakeupCount - 1)) ? mDNStrue : mDNSfalse;
2529 SendWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.IMAC, &rr->WakeUp.password, unicastOnly);
2530 for (r2 = rr; r2; r2=r2->next)
2531 if ((r2->resrec.RecordType == kDNSRecordTypeDeregistering) && r2->AnnounceCount && (r2->resrec.InterfaceID == rr->resrec.InterfaceID) &&
2532 mDNSSameEthAddress(&r2->WakeUp.IMAC, &rr->WakeUp.IMAC) && !mDNSSameEthAddress(&zeroEthAddr, &r2->WakeUp.HMAC))
2533 {
2534 // For now we only want to send a single Unsolicited Neighbor Advertisement restoring the address to the original
2535 // owner, because these packets can cause some IPv6 stacks to falsely conclude that there's an address conflict.
2536 if (r2->AddressProxy.type == mDNSAddrType_IPv6 && r2->AnnounceCount == WakeupCount)
2537 {
2538 LogSPS("NDP Announcement %2d Releasing traffic for H-MAC %.6a I-MAC %.6a %s",
2539 r2->AnnounceCount-3, &r2->WakeUp.HMAC, &r2->WakeUp.IMAC, ARDisplayString(m,r2));
2540 SendNDP(m, NDP_Adv, NDP_Override, r2, &r2->AddressProxy.ip.v6, &r2->WakeUp.IMAC, &AllHosts_v6, &AllHosts_v6_Eth);
2541 }
2542 r2->LastAPTime = m->timenow;
2543 // After 15 wakeups without success (maybe host has left the network) send three goodbyes instead
2544 if (--r2->AnnounceCount <= GoodbyeCount) r2->WakeUp.HMAC = zeroEthAddr;
2545 }
2546 }
2547 }
2548 else if (ResourceRecordIsValidAnswer(rr))
2549 {
2550 if (rr->AddressProxy.type)
2551 {
2552 if (!mDNSSameEthAddress(&zeroEthAddr, &rr->WakeUp.HMAC))
2553 {
2554 rr->AnnounceCount--;
2555 rr->ThisAPInterval *= 2;
2556 rr->LastAPTime = m->timenow;
2557 if (rr->AddressProxy.type == mDNSAddrType_IPv4)
2558 {
2559 LogSPS("ARP Announcement %2d Capturing traffic for H-MAC %.6a I-MAC %.6a %s",
2560 rr->AnnounceCount, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
2561 SendARP(m, 1, rr, &rr->AddressProxy.ip.v4, &zeroEthAddr, &rr->AddressProxy.ip.v4, &onesEthAddr);
2562 }
2563 else if (rr->AddressProxy.type == mDNSAddrType_IPv6)
2564 {
2565 LogSPS("NDP Announcement %2d Capturing traffic for H-MAC %.6a I-MAC %.6a %s",
2566 rr->AnnounceCount, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
2567 SendNDP(m, NDP_Adv, NDP_Override, rr, &rr->AddressProxy.ip.v6, mDNSNULL, &AllHosts_v6, &AllHosts_v6_Eth);
2568 }
2569 }
2570 }
2571 else
2572 {
2573 rr->ImmedAnswer = mDNSInterfaceMark; // Send on all interfaces
2574 if (maxExistingAnnounceInterval < rr->ThisAPInterval)
2575 maxExistingAnnounceInterval = rr->ThisAPInterval;
2576 if (rr->UpdateBlocked) rr->UpdateBlocked = 0;
2577 }
2578 }
2579 }
2580 }
2581
2582 // Any interface-specific records we're going to send are marked as being sent on all appropriate interfaces (which is just one)
2583 // Eligible records that are more than half-way to their announcement time are accelerated
2584 for (rr = m->ResourceRecords; rr; rr=rr->next)
2585 if ((rr->resrec.InterfaceID && rr->ImmedAnswer) ||
2586 (rr->ThisAPInterval <= maxExistingAnnounceInterval &&
2587 TimeToAnnounceThisRecord(rr, m->timenow + rr->ThisAPInterval/2) &&
2588 !rr->AddressProxy.type && // Don't include ARP Annoucements when considering which records to accelerate
2589 ResourceRecordIsValidAnswer(rr)))
2590 rr->ImmedAnswer = mDNSInterfaceMark; // Send on all interfaces
2591
2592 // When sending SRV records (particularly when announcing a new service) automatically add related Address record(s) as additionals
2593 // Note: Currently all address records are interface-specific, so it's safe to set ImmedAdditional to their InterfaceID,
2594 // which will be non-null. If by some chance there is an address record that's not interface-specific (should never happen)
2595 // then all that means is that it won't get sent -- which would not be the end of the world.
2596 for (rr = m->ResourceRecords; rr; rr=rr->next)
2597 {
2598 if (rr->ImmedAnswer && rr->resrec.rrtype == kDNSType_SRV)
2599 for (r2=m->ResourceRecords; r2; r2=r2->next) // Scan list of resource records
2600 if (RRTypeIsAddressType(r2->resrec.rrtype) && // For all address records (A/AAAA) ...
2601 ResourceRecordIsValidAnswer(r2) && // ... which are valid for answer ...
2602 rr->LastMCTime - r2->LastMCTime >= 0 && // ... which we have not sent recently ...
2603 rr->resrec.rdatahash == r2->resrec.namehash && // ... whose name is the name of the SRV target
2604 SameDomainName(&rr->resrec.rdata->u.srv.target, r2->resrec.name) &&
2605 (rr->ImmedAnswer == mDNSInterfaceMark || rr->ImmedAnswer == r2->resrec.InterfaceID))
2606 r2->ImmedAdditional = r2->resrec.InterfaceID; // ... then mark this address record for sending too
2607 // We also make sure we send the DeviceInfo TXT record too, if necessary
2608 // We check for RecordType == kDNSRecordTypeShared because we don't want to tag the
2609 // DeviceInfo TXT record onto a goodbye packet (RecordType == kDNSRecordTypeDeregistering).
2610 if (rr->ImmedAnswer && rr->resrec.RecordType == kDNSRecordTypeShared && rr->resrec.rrtype == kDNSType_PTR)
2611 if (ResourceRecordIsValidAnswer(&m->DeviceInfo) && SameDomainLabel(rr->resrec.rdata->u.name.c, m->DeviceInfo.resrec.name->c))
2612 {
2613 if (!m->DeviceInfo.ImmedAnswer) m->DeviceInfo.ImmedAnswer = rr->ImmedAnswer;
2614 else m->DeviceInfo.ImmedAnswer = mDNSInterfaceMark;
2615 }
2616 }
2617
2618 // If there's a record which is supposed to be unique that we're going to send, then make sure that we give
2619 // the whole RRSet as an atomic unit. That means that if we have any other records with the same name/type/class
2620 // then we need to mark them for sending too. Otherwise, if we set the kDNSClass_UniqueRRSet bit on a
2621 // record, then other RRSet members that have not been sent recently will get flushed out of client caches.
2622 // -- If a record is marked to be sent on a certain interface, make sure the whole set is marked to be sent on that interface
2623 // -- If any record is marked to be sent on all interfaces, make sure the whole set is marked to be sent on all interfaces
2624 for (rr = m->ResourceRecords; rr; rr=rr->next)
2625 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
2626 {
2627 if (rr->ImmedAnswer) // If we're sending this as answer, see that its whole RRSet is similarly marked
2628 {
2629 for (r2 = m->ResourceRecords; r2; r2=r2->next)
2630 {
2631 if ((r2->resrec.RecordType & kDNSRecordTypeUniqueMask) && ResourceRecordIsValidAnswer(r2) &&
2632 (r2->ImmedAnswer != mDNSInterfaceMark) && (r2->ImmedAnswer != rr->ImmedAnswer) &&
2633 SameResourceRecordSignature(r2, rr) &&
2634 ((rr->ImmedAnswer == mDNSInterfaceMark) || IsInterfaceValidForAuthRecord(r2, rr->ImmedAnswer)))
2635 {
2636 r2->ImmedAnswer = !r2->ImmedAnswer ? rr->ImmedAnswer : mDNSInterfaceMark;
2637 }
2638 }
2639 }
2640 else if (rr->ImmedAdditional) // If we're sending this as additional, see that its whole RRSet is similarly marked
2641 {
2642 for (r2 = m->ResourceRecords; r2; r2=r2->next)
2643 {
2644 if ((r2->resrec.RecordType & kDNSRecordTypeUniqueMask) && ResourceRecordIsValidAnswer(r2) &&
2645 (r2->ImmedAdditional != rr->ImmedAdditional) &&
2646 SameResourceRecordSignature(r2, rr) &&
2647 IsInterfaceValidForAuthRecord(r2, rr->ImmedAdditional))
2648 {
2649 r2->ImmedAdditional = rr->ImmedAdditional;
2650 }
2651 }
2652 }
2653 }
2654
2655 // Now set SendRNow state appropriately
2656 for (rr = m->ResourceRecords; rr; rr=rr->next)
2657 {
2658 if (rr->ImmedAnswer == mDNSInterfaceMark) // Sending this record on all appropriate interfaces
2659 {
2660 rr->SendRNow = !intf ? mDNSNULL : (rr->resrec.InterfaceID) ? rr->resrec.InterfaceID : intf->InterfaceID;
2661 rr->ImmedAdditional = mDNSNULL; // No need to send as additional if sending as answer
2662 rr->LastMCTime = m->timenow;
2663 rr->LastMCInterface = rr->ImmedAnswer;
2664 rr->ProbeRestartCount = 0; // Reset the probe restart count
2665 // If we're announcing this record, and it's at least half-way to its ordained time, then consider this announcement done
2666 if (TimeToAnnounceThisRecord(rr, m->timenow + rr->ThisAPInterval/2))
2667 {
2668 rr->AnnounceCount--;
2669 if (rr->resrec.RecordType != kDNSRecordTypeDeregistering)
2670 rr->ThisAPInterval *= 2;
2671 rr->LastAPTime = m->timenow;
2672 debugf("Announcing %##s (%s) %d", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), rr->AnnounceCount);
2673 }
2674 }
2675 else if (rr->ImmedAnswer) // Else, just respond to a single query on single interface:
2676 {
2677 rr->SendRNow = rr->ImmedAnswer; // Just respond on that interface
2678 rr->ImmedAdditional = mDNSNULL; // No need to send as additional too
2679 rr->LastMCTime = m->timenow;
2680 rr->LastMCInterface = rr->ImmedAnswer;
2681 }
2682 SetNextAnnounceProbeTime(m, rr);
2683 //if (rr->SendRNow) LogMsg("%-15.4a %s", &rr->v4Requester, ARDisplayString(m, rr));
2684 }
2685
2686 // ***
2687 // *** 2. Loop through interface list, sending records as appropriate
2688 // ***
2689
2690 while (intf)
2691 {
2692 int OwnerRecordSpace = (m->AnnounceOwner && intf->MAC.l[0]) ? DNSOpt_Header_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) : 0;
2693 int TraceRecordSpace = (mDNS_McastTracingEnabled && MDNS_TRACER) ? DNSOpt_Header_Space + DNSOpt_TraceData_Space : 0;
2694 int numDereg = 0;
2695 int numAnnounce = 0;
2696 int numAnswer = 0;
2697 int AnoninfoSpace = 0;
2698 mDNSu8 *responseptr = m->omsg.data;
2699 mDNSu8 *newptr;
2700 InitializeDNSMessage(&m->omsg.h, zeroID, ResponseFlags);
2701
2702 // First Pass. Look for:
2703 // 1. Deregistering records that need to send their goodbye packet
2704 // 2. Updated records that need to retract their old data
2705 // 3. Answers and announcements we need to send
2706 for (rr = m->ResourceRecords; rr; rr=rr->next)
2707 {
2708
2709 // Skip this interface if the record InterfaceID is *Any and the record is not
2710 // appropriate for the interface type.
2711 if ((rr->SendRNow == intf->InterfaceID) &&
2712 ((rr->resrec.InterfaceID == mDNSInterface_Any) && !mDNSPlatformValidRecordForInterface(rr, intf->InterfaceID)))
2713 {
2714 rr->SendRNow = GetNextActiveInterfaceID(intf);
2715 }
2716 else if (rr->SendRNow == intf->InterfaceID)
2717 {
2718 RData *OldRData = rr->resrec.rdata;
2719 mDNSu16 oldrdlength = rr->resrec.rdlength;
2720 mDNSu8 active = (mDNSu8)
2721 (rr->resrec.RecordType != kDNSRecordTypeDeregistering && !ShouldSendGoodbyesBeforeSleep(m, intf, rr));
2722 newptr = mDNSNULL;
2723 if (rr->NewRData && active)
2724 {
2725 // See if we should send a courtesy "goodbye" for the old data before we replace it.
2726 if (ResourceRecordIsValidAnswer(rr) && rr->resrec.RecordType == kDNSRecordTypeShared && rr->RequireGoodbye)
2727 {
2728 newptr = PutRR_OS_TTL(responseptr, &m->omsg.h.numAnswers, &rr->resrec, 0);
2729 if (newptr) { responseptr = newptr; numDereg++; rr->RequireGoodbye = mDNSfalse; }
2730 else continue; // If this packet is already too full to hold the goodbye for this record, skip it for now and we'll retry later
2731 }
2732 SetNewRData(&rr->resrec, rr->NewRData, rr->newrdlength);
2733 }
2734
2735 if (rr->resrec.AnonInfo)
2736 {
2737 int tmp = AnonInfoSpace(rr->resrec.AnonInfo);
2738
2739 AnoninfoSpace += tmp;
2740 // Adjust OwnerRecordSpace/TraceRecordSpace which is used by PutRR_OS_TTL below so that
2741 // we have space to put in the NSEC3 record in the authority section.
2742 OwnerRecordSpace += tmp;
2743 TraceRecordSpace += tmp;
2744 }
2745
2746 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
2747 rr->resrec.rrclass |= kDNSClass_UniqueRRSet; // Temporarily set the cache flush bit so PutResourceRecord will set it
2748 newptr = PutRR_OS_TTL(responseptr, &m->omsg.h.numAnswers, &rr->resrec, active ? rr->resrec.rroriginalttl : 0);
2749 rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; // Make sure to clear cache flush bit back to normal state
2750 if (newptr)
2751 {
2752 responseptr = newptr;
2753 rr->RequireGoodbye = active;
2754 if (rr->resrec.RecordType == kDNSRecordTypeDeregistering) numDereg++;
2755 else if (rr->LastAPTime == m->timenow) numAnnounce++;else numAnswer++;
2756 }
2757
2758 if (rr->NewRData && active)
2759 SetNewRData(&rr->resrec, OldRData, oldrdlength);
2760
2761 // The first time through (pktcount==0), if this record is verified unique
2762 // (i.e. typically A, AAAA, SRV, TXT and reverse-mapping PTR), set the flag to add an NSEC too.
2763 if (!pktcount && active && (rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && !rr->SendNSECNow)
2764 rr->SendNSECNow = mDNSInterfaceMark;
2765
2766 if (newptr) // If succeeded in sending, advance to next interface
2767 {
2768 if (rr->resrec.AnonInfo)
2769 {
2770 debugf("SendResponses: Marking %s, OwnerRecordSpace %d, TraceRecordSpace %d, limit %p", ARDisplayString(m, rr), OwnerRecordSpace,
2771 TraceRecordSpace, m->omsg.data + AllowedRRSpace(&m->omsg) - OwnerRecordSpace - TraceRecordSpace);
2772 rr->resrec.AnonInfo->SendNow = intf->InterfaceID;
2773 }
2774
2775 // If sending on all interfaces, go to next interface; else we're finished now
2776 if (rr->ImmedAnswer == mDNSInterfaceMark && rr->resrec.InterfaceID == mDNSInterface_Any)
2777 rr->SendRNow = GetNextActiveInterfaceID(intf);
2778 else
2779 rr->SendRNow = mDNSNULL;
2780 }
2781 }
2782 }
2783
2784 // Get the reserved space back
2785 OwnerRecordSpace -= AnoninfoSpace;
2786 TraceRecordSpace -= AnoninfoSpace;
2787 newptr = responseptr;
2788 for (rr = m->ResourceRecords; rr; rr=rr->next)
2789 {
2790 if (rr->resrec.AnonInfo && rr->resrec.AnonInfo->SendNow == intf->InterfaceID)
2791 {
2792 ResourceRecord *nsec3RR = rr->resrec.AnonInfo->nsec3RR;
2793
2794 newptr = PutRR_OS_TTL(newptr, &m->omsg.h.numAuthorities, nsec3RR, nsec3RR->rroriginalttl);
2795 if (newptr)
2796 {
2797 responseptr = newptr;
2798 debugf("SendResponses: Added NSEC3 %s, OwnerRecordSpace %d, TraceRecordSpace %d, limit %p", ARDisplayString(m, rr), OwnerRecordSpace,
2799 TraceRecordSpace, m->omsg.data + AllowedRRSpace(&m->omsg) - OwnerRecordSpace - TraceRecordSpace);
2800 }
2801 else
2802 {
2803 LogMsg("SendResponses: Cannot add NSEC3 %s, OwnerRecordSpace %d, TraceRecordSpace %d, limit %p", ARDisplayString(m, rr), OwnerRecordSpace,
2804 TraceRecordSpace, m->omsg.data + AllowedRRSpace(&m->omsg) - OwnerRecordSpace - TraceRecordSpace);
2805 }
2806 rr->resrec.AnonInfo->SendNow = mDNSNULL;
2807 }
2808 }
2809 // Second Pass. Add additional records, if there's space.
2810 newptr = responseptr;
2811 for (rr = m->ResourceRecords; rr; rr=rr->next)
2812 if (rr->ImmedAdditional == intf->InterfaceID)
2813 if (ResourceRecordIsValidAnswer(rr))
2814 {
2815 // If we have at least one answer already in the packet, then plan to add additionals too
2816 mDNSBool SendAdditional = (m->omsg.h.numAnswers > 0);
2817
2818 // If we're not planning to send any additionals, but this record is a unique one, then
2819 // make sure we haven't already sent any other members of its RRSet -- if we have, then they
2820 // will have had the cache flush bit set, so now we need to finish the job and send the rest.
2821 if (!SendAdditional && (rr->resrec.RecordType & kDNSRecordTypeUniqueMask))
2822 {
2823 const AuthRecord *a;
2824 for (a = m->ResourceRecords; a; a=a->next)
2825 if (a->LastMCTime == m->timenow &&
2826 a->LastMCInterface == intf->InterfaceID &&
2827 SameResourceRecordSignature(a, rr)) { SendAdditional = mDNStrue; break; }
2828 }
2829 if (!SendAdditional) // If we don't want to send this after all,
2830 rr->ImmedAdditional = mDNSNULL; // then cancel its ImmedAdditional field
2831 else if (newptr) // Else, try to add it if we can
2832 {
2833 // The first time through (pktcount==0), if this record is verified unique
2834 // (i.e. typically A, AAAA, SRV, TXT and reverse-mapping PTR), set the flag to add an NSEC too.
2835 if (!pktcount && (rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && !rr->SendNSECNow)
2836 rr->SendNSECNow = mDNSInterfaceMark;
2837
2838 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
2839 rr->resrec.rrclass |= kDNSClass_UniqueRRSet; // Temporarily set the cache flush bit so PutResourceRecord will set it
2840 newptr = PutRR_OS(newptr, &m->omsg.h.numAdditionals, &rr->resrec);
2841 rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; // Make sure to clear cache flush bit back to normal state
2842 if (newptr)
2843 {
2844 responseptr = newptr;
2845 rr->ImmedAdditional = mDNSNULL;
2846 rr->RequireGoodbye = mDNStrue;
2847 // If we successfully put this additional record in the packet, we record LastMCTime & LastMCInterface.
2848 // This matters particularly in the case where we have more than one IPv6 (or IPv4) address, because otherwise,
2849 // when we see our own multicast with the cache flush bit set, if we haven't set LastMCTime, then we'll get
2850 // all concerned and re-announce our record again to make sure it doesn't get flushed from peer caches.
2851 rr->LastMCTime = m->timenow;
2852 rr->LastMCInterface = intf->InterfaceID;
2853 }
2854 }
2855 }
2856
2857 // Third Pass. Add NSEC records, if there's space.
2858 // When we're generating an NSEC record in response to a specify query for that type
2859 // (recognized by rr->SendNSECNow == intf->InterfaceID) we should really put the NSEC in the Answer Section,
2860 // not Additional Section, but for now it's easier to handle both cases in this Additional Section loop here.
2861 for (rr = m->ResourceRecords; rr; rr=rr->next)
2862 if (rr->SendNSECNow == mDNSInterfaceMark || rr->SendNSECNow == intf->InterfaceID)
2863 {
2864 AuthRecord nsec;
2865 mDNSu8 *ptr;
2866 int len;
2867 mDNS_SetupResourceRecord(&nsec, mDNSNULL, mDNSInterface_Any, kDNSType_NSEC, rr->resrec.rroriginalttl, kDNSRecordTypeUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
2868 nsec.resrec.rrclass |= kDNSClass_UniqueRRSet;
2869 AssignDomainName(&nsec.namestorage, rr->resrec.name);
2870 ptr = nsec.rdatastorage.u.data;
2871 len = DomainNameLength(rr->resrec.name);
2872 // We have a nxt name followed by window number, window length and a window bitmap
2873 nsec.resrec.rdlength = len + 2 + NSEC_MCAST_WINDOW_SIZE;
2874 if (nsec.resrec.rdlength <= StandardAuthRDSize)
2875 {
2876 mDNSPlatformMemZero(ptr, nsec.resrec.rdlength);
2877 AssignDomainName((domainname *)ptr, rr->resrec.name);
2878 ptr += len;
2879 *ptr++ = 0; // window number
2880 *ptr++ = NSEC_MCAST_WINDOW_SIZE; // window length
2881 for (r2 = m->ResourceRecords; r2; r2=r2->next)
2882 if (ResourceRecordIsValidAnswer(r2) && SameResourceRecordNameClassInterface(r2, rr))
2883 {
2884 if (r2->resrec.rrtype >= kDNSQType_ANY) { LogMsg("SendResponses: Can't create NSEC for record %s", ARDisplayString(m, r2)); break; }
2885 else ptr[r2->resrec.rrtype >> 3] |= 128 >> (r2->resrec.rrtype & 7);
2886 }
2887 newptr = responseptr;
2888 if (!r2) // If we successfully built our NSEC record, add it to the packet now
2889 {
2890 newptr = PutRR_OS(responseptr, &m->omsg.h.numAdditionals, &nsec.resrec);
2891 if (newptr) responseptr = newptr;
2892 }
2893 }
2894 else LogMsg("SendResponses: not enough space (%d) in authrecord for nsec", nsec.resrec.rdlength);
2895
2896 // If we successfully put the NSEC record, clear the SendNSECNow flag
2897 // If we consider this NSEC optional, then we unconditionally clear the SendNSECNow flag, even if we fail to put this additional record
2898 if (newptr || rr->SendNSECNow == mDNSInterfaceMark)
2899 {
2900 rr->SendNSECNow = mDNSNULL;
2901 // Run through remainder of list clearing SendNSECNow flag for all other records which would generate the same NSEC
2902 for (r2 = rr->next; r2; r2=r2->next)
2903 if (SameResourceRecordNameClassInterface(r2, rr))
2904 if (r2->SendNSECNow == mDNSInterfaceMark || r2->SendNSECNow == intf->InterfaceID)
2905 r2->SendNSECNow = mDNSNULL;
2906 }
2907 }
2908
2909 if (m->omsg.h.numAnswers || m->omsg.h.numAdditionals)
2910 {
2911 // If we have data to send, add OWNER/TRACER/OWNER+TRACER option if necessary, then send packet
2912 if (OwnerRecordSpace || TraceRecordSpace)
2913 {
2914 AuthRecord opt;
2915 mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
2916 opt.resrec.rrclass = NormalMaxDNSMessageData;
2917 opt.resrec.rdlength = sizeof(rdataOPT);
2918 opt.resrec.rdestimate = sizeof(rdataOPT);
2919 if (OwnerRecordSpace && TraceRecordSpace)
2920 {
2921 opt.resrec.rdlength += sizeof(rdataOPT); // Two options in this OPT record
2922 opt.resrec.rdestimate += sizeof(rdataOPT);
2923 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
2924 SetupTracerOpt(m, &opt.resrec.rdata->u.opt[1]);
2925 }
2926 else if (OwnerRecordSpace)
2927 {
2928 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
2929 }
2930 else if (TraceRecordSpace)
2931 {
2932 SetupTracerOpt(m, &opt.resrec.rdata->u.opt[0]);
2933 }
2934 newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAdditionals, &opt.resrec);
2935 if (newptr)
2936 {
2937 responseptr = newptr;
2938 }
2939 else if (m->omsg.h.numAnswers + m->omsg.h.numAuthorities + m->omsg.h.numAdditionals == 1)
2940 {
2941 LogInfo("SendResponses: No space in packet for %s %s OPT record (%d/%d/%d/%d) %s", OwnerRecordSpace ? "OWNER" : "", TraceRecordSpace ? "TRACER" : "",
2942 m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
2943 }
2944 else
2945 {
2946 LogMsg("SendResponses: How did we fail to have space for %s %s OPT record (%d/%d/%d/%d) %s", OwnerRecordSpace ? "OWNER" : "", TraceRecordSpace ? "TRACER" : "",
2947 m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
2948 }
2949 }
2950
2951 debugf("SendResponses: Sending %d Deregistration%s, %d Announcement%s, %d Answer%s, %d Additional%s on %p",
2952 numDereg, numDereg == 1 ? "" : "s",
2953 numAnnounce, numAnnounce == 1 ? "" : "s",
2954 numAnswer, numAnswer == 1 ? "" : "s",
2955 m->omsg.h.numAdditionals, m->omsg.h.numAdditionals == 1 ? "" : "s", intf->InterfaceID);
2956
2957 if (intf->IPv4Available) mDNSSendDNSMessage(m, &m->omsg, responseptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v4, MulticastDNSPort, mDNSNULL, mDNSNULL, mDNSfalse);
2958 if (intf->IPv6Available) mDNSSendDNSMessage(m, &m->omsg, responseptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v6, MulticastDNSPort, mDNSNULL, mDNSNULL, mDNSfalse);
2959 if (!m->SuppressSending) m->SuppressSending = NonZeroTime(m->timenow + (mDNSPlatformOneSecond+9)/10);
2960 if (++pktcount >= 1000) { LogMsg("SendResponses exceeded loop limit %d: giving up", pktcount); break; }
2961 // There might be more things to send on this interface, so go around one more time and try again.
2962 }
2963 else // Nothing more to send on this interface; go to next
2964 {
2965 const NetworkInterfaceInfo *next = GetFirstActiveInterface(intf->next);
2966 #if MDNS_DEBUGMSGS && 0
2967 const char *const msg = next ? "SendResponses: Nothing more on %p; moving to %p" : "SendResponses: Nothing more on %p";
2968 debugf(msg, intf, next);
2969 #endif
2970 intf = next;
2971 pktcount = 0; // When we move to a new interface, reset packet count back to zero -- NSEC generation logic uses it
2972 }
2973 }
2974
2975 // ***
2976 // *** 3. Cleanup: Now that everything is sent, call client callback functions, and reset state variables
2977 // ***
2978
2979 if (m->CurrentRecord)
2980 LogMsg("SendResponses ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
2981 m->CurrentRecord = m->ResourceRecords;
2982 while (m->CurrentRecord)
2983 {
2984 rr = m->CurrentRecord;
2985 m->CurrentRecord = rr->next;
2986
2987 if (rr->SendRNow)
2988 {
2989 if (rr->ARType != AuthRecordLocalOnly && rr->ARType != AuthRecordP2P)
2990 LogInfo("SendResponses: No active interface %d to send: %d %02X %s",
2991 (uint32_t)rr->SendRNow, (uint32_t)rr->resrec.InterfaceID, rr->resrec.RecordType, ARDisplayString(m, rr));
2992 rr->SendRNow = mDNSNULL;
2993 }
2994
2995 if (rr->ImmedAnswer || rr->resrec.RecordType == kDNSRecordTypeDeregistering)
2996 {
2997 if (rr->NewRData) CompleteRDataUpdate(m, rr); // Update our rdata, clear the NewRData pointer, and return memory to the client
2998
2999 if (rr->resrec.RecordType == kDNSRecordTypeDeregistering && rr->AnnounceCount == 0)
3000 {
3001 // For Unicast, when we get the response from the server, we will call CompleteDeregistration
3002 if (!AuthRecord_uDNS(rr)) CompleteDeregistration(m, rr); // Don't touch rr after this
3003 }
3004 else
3005 {
3006 rr->ImmedAnswer = mDNSNULL;
3007 rr->ImmedUnicast = mDNSfalse;
3008 rr->v4Requester = zerov4Addr;
3009 rr->v6Requester = zerov6Addr;
3010 }
3011 }
3012 }
3013 verbosedebugf("SendResponses: Next in %ld ticks", m->NextScheduledResponse - m->timenow);
3014 }
3015
3016 // Calling CheckCacheExpiration() is an expensive operation because it has to look at the entire cache,
3017 // so we want to be lazy about how frequently we do it.
3018 // 1. If a cache record is currently referenced by *no* active questions,
3019 // then we don't mind expiring it up to a minute late (who will know?)
3020 // 2. Else, if a cache record is due for some of its final expiration queries,
3021 // we'll allow them to be late by up to 2% of the TTL
3022 // 3. Else, if a cache record has completed all its final expiration queries without success,
3023 // and is expiring, and had an original TTL more than ten seconds, we'll allow it to be one second late
3024 // 4. Else, it is expiring and had an original TTL of ten seconds or less (includes explicit goodbye packets),
3025 // so allow at most 1/10 second lateness
3026 // 5. For records with rroriginalttl set to zero, that means we really want to delete them immediately
3027 // (we have a new record with DelayDelivery set, waiting for the old record to go away before we can notify clients).
3028 #define CacheCheckGracePeriod(RR) ( \
3029 ((RR)->CRActiveQuestion == mDNSNULL ) ? (60 * mDNSPlatformOneSecond) : \
3030 ((RR)->UnansweredQueries < MaxUnansweredQueries) ? (TicksTTL(rr)/50) : \
3031 ((RR)->resrec.rroriginalttl > 10 ) ? (mDNSPlatformOneSecond) : \
3032 ((RR)->resrec.rroriginalttl > 0 ) ? (mDNSPlatformOneSecond/10) : 0)
3033
3034 #define NextCacheCheckEvent(RR) ((RR)->NextRequiredQuery + CacheCheckGracePeriod(RR))
3035
3036 mDNSexport void ScheduleNextCacheCheckTime(mDNS *const m, const mDNSu32 slot, const mDNSs32 event)
3037 {
3038 if (m->rrcache_nextcheck[slot] - event > 0)
3039 m->rrcache_nextcheck[slot] = event;
3040 if (m->NextCacheCheck - event > 0)
3041 m->NextCacheCheck = event;
3042 }
3043
3044 // Note: MUST call SetNextCacheCheckTimeForRecord any time we change:
3045 // rr->TimeRcvd
3046 // rr->resrec.rroriginalttl
3047 // rr->UnansweredQueries
3048 // rr->CRActiveQuestion
3049 mDNSexport void SetNextCacheCheckTimeForRecord(mDNS *const m, CacheRecord *const rr)
3050 {
3051 rr->NextRequiredQuery = RRExpireTime(rr);
3052
3053 // If we have an active question, then see if we want to schedule a refresher query for this record.
3054 // Usually we expect to do four queries, at 80-82%, 85-87%, 90-92% and then 95-97% of the TTL.
3055 if (rr->CRActiveQuestion && rr->UnansweredQueries < MaxUnansweredQueries)
3056 {
3057 rr->NextRequiredQuery -= TicksTTL(rr)/20 * (MaxUnansweredQueries - rr->UnansweredQueries);
3058 rr->NextRequiredQuery += mDNSRandom((mDNSu32)TicksTTL(rr)/50);
3059 verbosedebugf("SetNextCacheCheckTimeForRecord: NextRequiredQuery in %ld sec CacheCheckGracePeriod %d ticks for %s",
3060 (rr->NextRequiredQuery - m->timenow) / mDNSPlatformOneSecond, CacheCheckGracePeriod(rr), CRDisplayString(m,rr));
3061 }
3062 ScheduleNextCacheCheckTime(m, HashSlot(rr->resrec.name), NextCacheCheckEvent(rr));
3063 }
3064
3065 #define kMinimumReconfirmTime ((mDNSu32)mDNSPlatformOneSecond * 5)
3066 #define kDefaultReconfirmTimeForWake ((mDNSu32)mDNSPlatformOneSecond * 5)
3067 #define kDefaultReconfirmTimeForNoAnswer ((mDNSu32)mDNSPlatformOneSecond * 5)
3068 #define kDefaultReconfirmTimeForFlappingInterface ((mDNSu32)mDNSPlatformOneSecond * 5)
3069
3070 mDNSexport mStatus mDNS_Reconfirm_internal(mDNS *const m, CacheRecord *const rr, mDNSu32 interval)
3071 {
3072 if (interval < kMinimumReconfirmTime)
3073 interval = kMinimumReconfirmTime;
3074 if (interval > 0x10000000) // Make sure interval doesn't overflow when we multiply by four below
3075 interval = 0x10000000;
3076
3077 // If the expected expiration time for this record is more than interval+33%, then accelerate its expiration
3078 if (RRExpireTime(rr) - m->timenow > (mDNSs32)((interval * 4) / 3))
3079 {
3080 // Add a 33% random amount to the interval, to avoid synchronization between multiple hosts
3081 // For all the reconfirmations in a given batch, we want to use the same random value
3082 // so that the reconfirmation questions can be grouped into a single query packet
3083 if (!m->RandomReconfirmDelay) m->RandomReconfirmDelay = 1 + mDNSRandom(0x3FFFFFFF);
3084 interval += m->RandomReconfirmDelay % ((interval/3) + 1);
3085 rr->TimeRcvd = m->timenow - (mDNSs32)interval * 3;
3086 rr->resrec.rroriginalttl = (interval * 4 + mDNSPlatformOneSecond - 1) / mDNSPlatformOneSecond;
3087 SetNextCacheCheckTimeForRecord(m, rr);
3088 }
3089 debugf("mDNS_Reconfirm_internal:%6ld ticks to go for %s %p",
3090 RRExpireTime(rr) - m->timenow, CRDisplayString(m, rr), rr->CRActiveQuestion);
3091 return(mStatus_NoError);
3092 }
3093
3094 // BuildQuestion puts a question into a DNS Query packet and if successful, updates the value of queryptr.
3095 // It also appends to the list of known answer records that need to be included,
3096 // and updates the forcast for the size of the known answer section.
3097 mDNSlocal mDNSBool BuildQuestion(mDNS *const m, const NetworkInterfaceInfo *intf, DNSMessage *query, mDNSu8 **queryptr,
3098 DNSQuestion *q, CacheRecord ***kalistptrptr, mDNSu32 *answerforecast)
3099 {
3100 mDNSBool ucast = (q->LargeAnswers || q->RequestUnicast) && m->CanReceiveUnicastOn5353 && intf->SupportsUnicastMDNSResponse;
3101 mDNSu16 ucbit = (mDNSu16)(ucast ? kDNSQClass_UnicastResponse : 0);
3102 const mDNSu8 *const limit = query->data + NormalMaxDNSMessageData;
3103 mDNSu8 anoninfo_space = q->AnonInfo ? AnonInfoSpace(q->AnonInfo) : 0;
3104 mDNSu8 *newptr = putQuestion(query, *queryptr, limit - *answerforecast - anoninfo_space, &q->qname, q->qtype, (mDNSu16)(q->qclass | ucbit));
3105 if (!newptr)
3106 {
3107 debugf("BuildQuestion: No more space in this packet for question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
3108 return(mDNSfalse);
3109 }
3110 else
3111 {
3112 mDNSu32 forecast = *answerforecast + anoninfo_space;
3113 const mDNSu32 slot = HashSlot(&q->qname);
3114 const CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
3115 CacheRecord *rr;
3116 CacheRecord **ka = *kalistptrptr; // Make a working copy of the pointer we're going to update
3117
3118 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) // If we have a resource record in our cache,
3119 if (rr->resrec.InterfaceID == q->SendQNow && // received on this interface
3120 !(rr->resrec.RecordType & kDNSRecordTypeUniqueMask) && // which is a shared (i.e. not unique) record type
3121 rr->NextInKAList == mDNSNULL && ka != &rr->NextInKAList && // which is not already in the known answer list
3122 rr->resrec.rdlength <= SmallRecordLimit && // which is small enough to sensibly fit in the packet
3123 SameNameRecordAnswersQuestion(&rr->resrec, q) && // which answers our question
3124 rr->TimeRcvd + TicksTTL(rr)/2 - m->timenow > // and its half-way-to-expiry time is at least 1 second away
3125 mDNSPlatformOneSecond) // (also ensures we never include goodbye records with TTL=1)
3126 {
3127 // We don't want to include unique records in the Known Answer section. The Known Answer section
3128 // is intended to suppress floods of shared-record replies from many other devices on the network.
3129 // That concept really does not apply to unique records, and indeed if we do send a query for
3130 // which we have a unique record already in our cache, then including that unique record as a
3131 // Known Answer, so as to suppress the only answer we were expecting to get, makes little sense.
3132
3133 *ka = rr; // Link this record into our known answer chain
3134 ka = &rr->NextInKAList;
3135 // We forecast: compressed name (2) type (2) class (2) TTL (4) rdlength (2) rdata (n)
3136 forecast += 12 + rr->resrec.rdestimate;
3137 // If we're trying to put more than one question in this packet, and it doesn't fit
3138 // then undo that last question and try again next time
3139 if (query->h.numQuestions > 1 && newptr + forecast >= limit)
3140 {
3141 query->h.numQuestions--;
3142 debugf("BuildQuestion: Retracting question %##s (%s) new forecast total %d, total questions %d",
3143 q->qname.c, DNSTypeName(q->qtype), newptr + forecast - query->data, query->h.numQuestions);
3144 ka = *kalistptrptr; // Go back to where we started and retract these answer records
3145 while (*ka) { CacheRecord *c = *ka; *ka = mDNSNULL; ka = &c->NextInKAList; }
3146 return(mDNSfalse); // Return false, so we'll try again in the next packet
3147 }
3148 }
3149
3150 // Success! Update our state pointers, increment UnansweredQueries as appropriate, and return
3151 *queryptr = newptr; // Update the packet pointer
3152 *answerforecast = forecast; // Update the forecast
3153 *kalistptrptr = ka; // Update the known answer list pointer
3154 if (ucast) q->ExpectUnicastResp = NonZeroTime(m->timenow);
3155
3156 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) // For every resource record in our cache,
3157 if (rr->resrec.InterfaceID == q->SendQNow && // received on this interface
3158 rr->NextInKAList == mDNSNULL && ka != &rr->NextInKAList && // which is not in the known answer list
3159 SameNameRecordAnswersQuestion(&rr->resrec, q)) // which answers our question
3160 {
3161 rr->UnansweredQueries++; // indicate that we're expecting a response
3162 rr->LastUnansweredTime = m->timenow;
3163 SetNextCacheCheckTimeForRecord(m, rr);
3164 }
3165
3166 return(mDNStrue);
3167 }
3168 }
3169
3170 // When we have a query looking for a specified name, but there appear to be no answers with
3171 // that name, ReconfirmAntecedents() is called with depth=0 to start the reconfirmation process
3172 // for any records in our cache that reference the given name (e.g. PTR and SRV records).
3173 // For any such cache record we find, we also recursively call ReconfirmAntecedents() for *its* name.
3174 // We increment depth each time we recurse, to guard against possible infinite loops, with a limit of 5.
3175 // A typical reconfirmation scenario might go like this:
3176 // Depth 0: Name "myhost.local" has no address records
3177 // Depth 1: SRV "My Service._example._tcp.local." refers to "myhost.local"; may be stale
3178 // Depth 2: PTR "_example._tcp.local." refers to "My Service"; may be stale
3179 // Depth 3: PTR "_services._dns-sd._udp.local." refers to "_example._tcp.local."; may be stale
3180 // Currently depths 4 and 5 are not expected to occur; if we did get to depth 5 we'd reconfim any records we
3181 // found referring to the given name, but not recursively descend any further reconfirm *their* antecedents.
3182 mDNSlocal void ReconfirmAntecedents(mDNS *const m, const domainname *const name, const mDNSu32 namehash, const int depth)
3183 {
3184 mDNSu32 slot;
3185 CacheGroup *cg;
3186 CacheRecord *cr;
3187 debugf("ReconfirmAntecedents (depth=%d) for %##s", depth, name->c);
3188 FORALL_CACHERECORDS(slot, cg, cr)
3189 {
3190 domainname *crtarget = GetRRDomainNameTarget(&cr->resrec);
3191 if (crtarget && cr->resrec.rdatahash == namehash && SameDomainName(crtarget, name))
3192 {
3193 LogInfo("ReconfirmAntecedents: Reconfirming (depth=%d) %s", depth, CRDisplayString(m, cr));
3194 mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
3195 if (depth < 5)
3196 ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, depth+1);
3197 }
3198 }
3199 }
3200
3201 // If we get no answer for a AAAA query, then before doing an automatic implicit ReconfirmAntecedents
3202 // we check if we have an address record for the same name. If we do have an IPv4 address for a given
3203 // name but not an IPv6 address, that's okay (it just means the device doesn't do IPv6) so the failure
3204 // to get a AAAA response is not grounds to doubt the PTR/SRV chain that lead us to that name.
3205 mDNSlocal const CacheRecord *CacheHasAddressTypeForName(mDNS *const m, const domainname *const name, const mDNSu32 namehash)
3206 {
3207 CacheGroup *const cg = CacheGroupForName(m, HashSlot(name), namehash, name);
3208 const CacheRecord *cr = cg ? cg->members : mDNSNULL;
3209 while (cr && !RRTypeIsAddressType(cr->resrec.rrtype)) cr=cr->next;
3210 return(cr);
3211 }
3212
3213
3214 mDNSlocal const CacheRecord *FindSPSInCache1(mDNS *const m, const DNSQuestion *const q, const CacheRecord *const c0, const CacheRecord *const c1)
3215 {
3216 #ifndef SPC_DISABLED
3217 CacheGroup *const cg = CacheGroupForName(m, HashSlot(&q->qname), q->qnamehash, &q->qname);
3218 const CacheRecord *cr, *bestcr = mDNSNULL;
3219 mDNSu32 bestmetric = 1000000;
3220 for (cr = cg ? cg->members : mDNSNULL; cr; cr=cr->next)
3221 if (cr->resrec.rrtype == kDNSType_PTR && cr->resrec.rdlength >= 6) // If record is PTR type, with long enough name,
3222 if (cr != c0 && cr != c1) // that's not one we've seen before,
3223 if (SameNameRecordAnswersQuestion(&cr->resrec, q)) // and answers our browse query,
3224 if (!IdenticalSameNameRecord(&cr->resrec, &m->SPSRecords.RR_PTR.resrec)) // and is not our own advertised service...
3225 {
3226 mDNSu32 metric = SPSMetric(cr->resrec.rdata->u.name.c);
3227 if (bestmetric > metric) { bestmetric = metric; bestcr = cr; }
3228 }
3229 return(bestcr);
3230 #else // SPC_DISABLED
3231 (void) m;
3232 (void) q;
3233 (void) c0;
3234 (void) c1;
3235 (void) c1;
3236 return mDNSNULL;
3237 #endif // SPC_DISABLED
3238 }
3239
3240 mDNSlocal void CheckAndSwapSPS(const CacheRecord **sps1, const CacheRecord **sps2)
3241 {
3242 const CacheRecord *swap_sps;
3243 mDNSu32 metric1, metric2;
3244
3245 if (!(*sps1) || !(*sps2)) return;
3246 metric1 = SPSMetric((*sps1)->resrec.rdata->u.name.c);
3247 metric2 = SPSMetric((*sps2)->resrec.rdata->u.name.c);
3248 if (!SPSFeatures((*sps1)->resrec.rdata->u.name.c) && SPSFeatures((*sps2)->resrec.rdata->u.name.c) && (metric2 >= metric1))
3249 {
3250 swap_sps = *sps1;
3251 *sps1 = *sps2;
3252 *sps2 = swap_sps;
3253 }
3254 }
3255
3256 mDNSlocal void ReorderSPSByFeature(const CacheRecord *sps[3])
3257 {
3258 CheckAndSwapSPS(&sps[0], &sps[1]);
3259 CheckAndSwapSPS(&sps[0], &sps[2]);
3260 CheckAndSwapSPS(&sps[1], &sps[2]);
3261 }
3262
3263
3264 // Finds the three best Sleep Proxies we currently have in our cache
3265 mDNSexport void FindSPSInCache(mDNS *const m, const DNSQuestion *const q, const CacheRecord *sps[3])
3266 {
3267 sps[0] = FindSPSInCache1(m, q, mDNSNULL, mDNSNULL);
3268 sps[1] = !sps[0] ? mDNSNULL : FindSPSInCache1(m, q, sps[0], mDNSNULL);
3269 sps[2] = !sps[1] ? mDNSNULL : FindSPSInCache1(m, q, sps[0], sps[1]);
3270
3271 // SPS is already sorted by metric. We want to move the entries to the beginning of the array
3272 // only if they have equally good metric and support features.
3273 ReorderSPSByFeature(sps);
3274 }
3275
3276 // Only DupSuppressInfos newer than the specified 'time' are allowed to remain active
3277 mDNSlocal void ExpireDupSuppressInfo(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 time)
3278 {
3279 int i;
3280 for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].Time - time < 0) ds[i].InterfaceID = mDNSNULL;
3281 }
3282
3283 mDNSlocal void ExpireDupSuppressInfoOnInterface(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 time, mDNSInterfaceID InterfaceID)
3284 {
3285 int i;
3286 for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].InterfaceID == InterfaceID && ds[i].Time - time < 0) ds[i].InterfaceID = mDNSNULL;
3287 }
3288
3289 mDNSlocal mDNSBool SuppressOnThisInterface(const DupSuppressInfo ds[DupSuppressInfoSize], const NetworkInterfaceInfo * const intf)
3290 {
3291 int i;
3292 mDNSBool v4 = !intf->IPv4Available; // If this interface doesn't do v4, we don't need to find a v4 duplicate of this query
3293 mDNSBool v6 = !intf->IPv6Available; // If this interface doesn't do v6, we don't need to find a v6 duplicate of this query
3294 for (i=0; i<DupSuppressInfoSize; i++)
3295 if (ds[i].InterfaceID == intf->InterfaceID)
3296 {
3297 if (ds[i].Type == mDNSAddrType_IPv4) v4 = mDNStrue;
3298 else if (ds[i].Type == mDNSAddrType_IPv6) v6 = mDNStrue;
3299 if (v4 && v6) return(mDNStrue);
3300 }
3301 return(mDNSfalse);
3302 }
3303
3304 mDNSlocal void RecordDupSuppressInfo(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 Time, mDNSInterfaceID InterfaceID, mDNSs32 Type)
3305 {
3306 int i, j;
3307
3308 // See if we have this one in our list somewhere already
3309 for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].InterfaceID == InterfaceID && ds[i].Type == Type) break;
3310
3311 // If not, find a slot we can re-use
3312 if (i >= DupSuppressInfoSize)
3313 {
3314 i = 0;
3315 for (j=1; j<DupSuppressInfoSize && ds[i].InterfaceID; j++)
3316 if (!ds[j].InterfaceID || ds[j].Time - ds[i].Time < 0)
3317 i = j;
3318 }
3319
3320 // Record the info about this query we saw
3321 ds[i].Time = Time;
3322 ds[i].InterfaceID = InterfaceID;
3323 ds[i].Type = Type;
3324 }
3325
3326 mDNSlocal void mDNSSendWakeOnResolve(mDNS *const m, DNSQuestion *q)
3327 {
3328 int len, i, cnt;
3329 mDNSInterfaceID InterfaceID = q->InterfaceID;
3330 domainname *d = &q->qname;
3331
3332 // We can't send magic packets without knowing which interface to send it on.
3333 if (InterfaceID == mDNSInterface_Any || LocalOnlyOrP2PInterface(InterfaceID))
3334 {
3335 LogMsg("mDNSSendWakeOnResolve: ERROR!! Invalid InterfaceID %p for question %##s", InterfaceID, q->qname.c);
3336 return;
3337 }
3338
3339 // Split MAC@IPAddress and pass them separately
3340 len = d->c[0];
3341 cnt = 0;
3342 for (i = 1; i < len; i++)
3343 {
3344 if (d->c[i] == '@')
3345 {
3346 char EthAddr[18]; // ethernet adddress : 12 bytes + 5 ":" + 1 NULL byte
3347 char IPAddr[47]; // Max IP address len: 46 bytes (IPv6) + 1 NULL byte
3348 if (cnt != 5)
3349 {
3350 LogMsg("mDNSSendWakeOnResolve: ERROR!! Malformed Ethernet address %##s, cnt %d", q->qname.c, cnt);
3351 return;
3352 }
3353 if ((i - 1) > (int) (sizeof(EthAddr) - 1))
3354 {
3355 LogMsg("mDNSSendWakeOnResolve: ERROR!! Malformed Ethernet address %##s, length %d", q->qname.c, i - 1);
3356 return;
3357 }
3358 if ((len - i) > (int)(sizeof(IPAddr) - 1))
3359 {
3360 LogMsg("mDNSSendWakeOnResolve: ERROR!! Malformed IP address %##s, length %d", q->qname.c, len - i);
3361 return;
3362 }
3363 mDNSPlatformMemCopy(EthAddr, &d->c[1], i - 1);
3364 EthAddr[i - 1] = 0;
3365 mDNSPlatformMemCopy(IPAddr, &d->c[i + 1], len - i);
3366 IPAddr[len - i] = 0;
3367 m->mDNSStats.WakeOnResolves++;
3368 mDNSPlatformSendWakeupPacket(m, InterfaceID, EthAddr, IPAddr, InitialWakeOnResolveCount - q->WakeOnResolveCount);
3369 return;
3370 }
3371 else if (d->c[i] == ':')
3372 cnt++;
3373 }
3374 LogMsg("mDNSSendWakeOnResolve: ERROR!! Malformed WakeOnResolve name %##s", q->qname.c);
3375 }
3376
3377
3378 mDNSlocal mDNSBool AccelerateThisQuery(mDNS *const m, DNSQuestion *q)
3379 {
3380 // If more than 90% of the way to the query time, we should unconditionally accelerate it
3381 if (TimeToSendThisQuestion(q, m->timenow + q->ThisQInterval/10))
3382 return(mDNStrue);
3383
3384 // If half-way to next scheduled query time, only accelerate if it will add less than 512 bytes to the packet
3385 if (TimeToSendThisQuestion(q, m->timenow + q->ThisQInterval/2))
3386 {
3387 // We forecast: qname (n) type (2) class (2)
3388 mDNSu32 forecast = (mDNSu32)DomainNameLength(&q->qname) + 4;
3389 const mDNSu32 slot = HashSlot(&q->qname);
3390 const CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
3391 const CacheRecord *rr;
3392 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) // If we have a resource record in our cache,
3393 if (rr->resrec.rdlength <= SmallRecordLimit && // which is small enough to sensibly fit in the packet
3394 SameNameRecordAnswersQuestion(&rr->resrec, q) && // which answers our question
3395 rr->TimeRcvd + TicksTTL(rr)/2 - m->timenow >= 0 && // and it is less than half-way to expiry
3396 rr->NextRequiredQuery - (m->timenow + q->ThisQInterval) > 0) // and we'll ask at least once again before NextRequiredQuery
3397 {
3398 // We forecast: compressed name (2) type (2) class (2) TTL (4) rdlength (2) rdata (n)
3399 forecast += 12 + rr->resrec.rdestimate;
3400 if (forecast >= 512) return(mDNSfalse); // If this would add 512 bytes or more to the packet, don't accelerate
3401 }
3402 return(mDNStrue);
3403 }
3404
3405 return(mDNSfalse);
3406 }
3407
3408 // How Standard Queries are generated:
3409 // 1. The Question Section contains the question
3410 // 2. The Additional Section contains answers we already know, to suppress duplicate responses
3411
3412 // How Probe Queries are generated:
3413 // 1. The Question Section contains queries for the name we intend to use, with QType=ANY because
3414 // if some other host is already using *any* records with this name, we want to know about it.
3415 // 2. The Authority Section contains the proposed values we intend to use for one or more
3416 // of our records with that name (analogous to the Update section of DNS Update packets)
3417 // because if some other host is probing at the same time, we each want to know what the other is
3418 // planning, in order to apply the tie-breaking rule to see who gets to use the name and who doesn't.
3419
3420 mDNSlocal void SendQueries(mDNS *const m)
3421 {
3422 mDNSu32 slot;
3423 CacheGroup *cg;
3424 CacheRecord *cr;
3425 AuthRecord *ar;
3426 int pktcount = 0;
3427 DNSQuestion *q;
3428 // For explanation of maxExistingQuestionInterval logic, see comments for maxExistingAnnounceInterval
3429 mDNSs32 maxExistingQuestionInterval = 0;
3430 const NetworkInterfaceInfo *intf = GetFirstActiveInterface(m->HostInterfaces);
3431 CacheRecord *KnownAnswerList = mDNSNULL;
3432
3433 // 1. If time for a query, work out what we need to do
3434
3435 // We're expecting to send a query anyway, so see if any expiring cache records are close enough
3436 // to their NextRequiredQuery to be worth batching them together with this one
3437 FORALL_CACHERECORDS(slot, cg, cr)
3438 {
3439 if (cr->CRActiveQuestion && cr->UnansweredQueries < MaxUnansweredQueries)
3440 {
3441 if (m->timenow + TicksTTL(cr)/50 - cr->NextRequiredQuery >= 0)
3442 {
3443 debugf("Sending %d%% cache expiration query for %s", 80 + 5 * cr->UnansweredQueries, CRDisplayString(m, cr));
3444 q = cr->CRActiveQuestion;
3445 ExpireDupSuppressInfoOnInterface(q->DupSuppress, m->timenow - TicksTTL(cr)/20, cr->resrec.InterfaceID);
3446 // For uDNS queries (TargetQID non-zero) we adjust LastQTime,
3447 // and bump UnansweredQueries so that we don't spin trying to send the same cache expiration query repeatedly
3448 if (q->Target.type)
3449 {
3450 q->SendQNow = mDNSInterfaceMark; // If targeted query, mark it
3451 }
3452 else if (!mDNSOpaque16IsZero(q->TargetQID))
3453 {
3454 q->LastQTime = m->timenow - q->ThisQInterval;
3455 cr->UnansweredQueries++;
3456 m->mDNSStats.CacheRefreshQueries++;
3457 }
3458 else if (q->SendQNow == mDNSNULL)
3459 {
3460 q->SendQNow = cr->resrec.InterfaceID;
3461 }
3462 else if (q->SendQNow != cr->resrec.InterfaceID)
3463 {
3464 q->SendQNow = mDNSInterfaceMark;
3465 }
3466
3467 // Indicate that this question was marked for sending
3468 // to update an existing cached answer record.
3469 // The browse throttling logic below uses this to determine
3470 // if the query should be sent.
3471 if (mDNSOpaque16IsZero(q->TargetQID))
3472 q->CachedAnswerNeedsUpdate = mDNStrue;
3473 }
3474 }
3475 }
3476
3477 // Scan our list of questions to see which:
3478 // *WideArea* queries need to be sent
3479 // *unicast* queries need to be sent
3480 // *multicast* queries we're definitely going to send
3481 if (m->CurrentQuestion)
3482 LogMsg("SendQueries ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
3483 m->CurrentQuestion = m->Questions;
3484 while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
3485 {
3486 q = m->CurrentQuestion;
3487 if (q->Target.type && (q->SendQNow || TimeToSendThisQuestion(q, m->timenow)))
3488 {
3489 mDNSu8 *qptr = m->omsg.data;
3490 const mDNSu8 *const limit = m->omsg.data + sizeof(m->omsg.data);
3491
3492 // If we fail to get a new on-demand socket (should only happen cases of the most extreme resource exhaustion), we'll try again next time
3493 if (!q->LocalSocket) q->LocalSocket = mDNSPlatformUDPSocket(m, zeroIPPort);
3494 if (q->LocalSocket)
3495 {
3496 InitializeDNSMessage(&m->omsg.h, q->TargetQID, QueryFlags);
3497 qptr = putQuestion(&m->omsg, qptr, limit, &q->qname, q->qtype, q->qclass);
3498 mDNSSendDNSMessage(m, &m->omsg, qptr, mDNSInterface_Any, q->LocalSocket, &q->Target, q->TargetPort, mDNSNULL, mDNSNULL, q->UseBackgroundTrafficClass);
3499 q->ThisQInterval *= QuestionIntervalStep;
3500 }
3501 if (q->ThisQInterval > MaxQuestionInterval)
3502 q->ThisQInterval = MaxQuestionInterval;
3503 q->LastQTime = m->timenow;
3504 q->LastQTxTime = m->timenow;
3505 q->RecentAnswerPkts = 0;
3506 q->SendQNow = mDNSNULL;
3507 q->ExpectUnicastResp = NonZeroTime(m->timenow);
3508 }
3509 else if (mDNSOpaque16IsZero(q->TargetQID) && !q->Target.type && TimeToSendThisQuestion(q, m->timenow))
3510 {
3511 //LogInfo("Time to send %##s (%s) %d", q->qname.c, DNSTypeName(q->qtype), m->timenow - NextQSendTime(q));
3512 q->SendQNow = mDNSInterfaceMark; // Mark this question for sending on all interfaces
3513 if (maxExistingQuestionInterval < q->ThisQInterval)
3514 maxExistingQuestionInterval = q->ThisQInterval;
3515 }
3516 // If m->CurrentQuestion wasn't modified out from under us, advance it now
3517 // We can't do this at the start of the loop because uDNS_CheckCurrentQuestion() depends on having
3518 // m->CurrentQuestion point to the right question
3519 if (q == m->CurrentQuestion) m->CurrentQuestion = m->CurrentQuestion->next;
3520 }
3521 while (m->CurrentQuestion)
3522 {
3523 LogInfo("SendQueries question loop 1: Skipping NewQuestion %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
3524 m->CurrentQuestion = m->CurrentQuestion->next;
3525 }
3526 m->CurrentQuestion = mDNSNULL;
3527
3528 // Scan our list of questions
3529 // (a) to see if there are any more that are worth accelerating, and
3530 // (b) to update the state variables for *all* the questions we're going to send
3531 // Note: Don't set NextScheduledQuery until here, because uDNS_CheckCurrentQuestion in the loop above can add new questions to the list,
3532 // which causes NextScheduledQuery to get (incorrectly) set to m->timenow. Setting it here is the right place, because the very
3533 // next thing we do is scan the list and call SetNextQueryTime() for every question we find, so we know we end up with the right value.
3534 m->NextScheduledQuery = m->timenow + 0x78000000;
3535 for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
3536 {
3537 if (mDNSOpaque16IsZero(q->TargetQID)
3538 && (q->SendQNow || (!q->Target.type && ActiveQuestion(q) && q->ThisQInterval <= maxExistingQuestionInterval && AccelerateThisQuery(m,q))))
3539 {
3540 // If at least halfway to next query time, advance to next interval
3541 // If less than halfway to next query time, then
3542 // treat this as logically a repeat of the last transmission, without advancing the interval
3543 if (m->timenow - (q->LastQTime + (q->ThisQInterval/2)) >= 0)
3544 {
3545 // If we have reached the answer threshold for this question,
3546 // don't send it again until MaxQuestionInterval unless:
3547 // one of its cached answers needs to be refreshed,
3548 // or it's the initial query for a kDNSServiceFlagsThresholdFinder mode browse.
3549 if (q->BrowseThreshold
3550 && (q->CurrentAnswers >= q->BrowseThreshold)
3551 && (q->CachedAnswerNeedsUpdate == mDNSfalse)
3552 && !((q->flags & kDNSServiceFlagsThresholdFinder) && (q->ThisQInterval == InitialQuestionInterval)))
3553 {
3554 q->SendQNow = mDNSNULL;
3555 q->ThisQInterval = MaxQuestionInterval;
3556 q->LastQTime = m->timenow;
3557 q->RequestUnicast = 0;
3558 LogInfo("SendQueries: (%s) %##s reached threshold of %d answers",
3559 DNSTypeName(q->qtype), q->qname.c, q->BrowseThreshold);
3560 }
3561 else
3562 {
3563 // Mark this question for sending on all interfaces
3564 q->SendQNow = mDNSInterfaceMark;
3565 q->ThisQInterval *= QuestionIntervalStep;
3566 }
3567
3568 debugf("SendQueries: %##s (%s) next interval %d seconds RequestUnicast = %d",
3569 q->qname.c, DNSTypeName(q->qtype), q->ThisQInterval / InitialQuestionInterval, q->RequestUnicast);
3570
3571 if (q->ThisQInterval > MaxQuestionInterval)
3572 {
3573 q->ThisQInterval = MaxQuestionInterval;
3574 }
3575 else if (q->CurrentAnswers == 0 && q->ThisQInterval == InitialQuestionInterval * QuestionIntervalStep3 && !q->RequestUnicast &&
3576 !(RRTypeIsAddressType(q->qtype) && CacheHasAddressTypeForName(m, &q->qname, q->qnamehash)))
3577 {
3578 // Generally don't need to log this.
3579 // It's not especially noteworthy if a query finds no results -- this usually happens for domain
3580 // enumeration queries in the LL subdomain (e.g. "db._dns-sd._udp.0.0.254.169.in-addr.arpa")
3581 // and when there simply happen to be no instances of the service the client is looking
3582 // for (e.g. iTunes is set to look for RAOP devices, and the current network has none).
3583 debugf("SendQueries: Zero current answers for %##s (%s); will reconfirm antecedents",
3584 q->qname.c, DNSTypeName(q->qtype));
3585 // Sending third query, and no answers yet; time to begin doubting the source
3586 ReconfirmAntecedents(m, &q->qname, q->qnamehash, 0);
3587 }
3588 }
3589
3590 // Mark for sending. (If no active interfaces, then don't even try.)
3591 q->SendOnAll = (q->SendQNow == mDNSInterfaceMark);
3592 if (q->SendOnAll)
3593 {
3594 q->SendQNow = !intf ? mDNSNULL : (q->InterfaceID) ? q->InterfaceID : intf->InterfaceID;
3595 q->LastQTime = m->timenow;
3596 }
3597
3598 // If we recorded a duplicate suppression for this question less than half an interval ago,
3599 // then we consider it recent enough that we don't need to do an identical query ourselves.
3600 ExpireDupSuppressInfo(q->DupSuppress, m->timenow - q->ThisQInterval/2);
3601
3602 q->LastQTxTime = m->timenow;
3603 q->RecentAnswerPkts = 0;
3604 if (q->RequestUnicast) q->RequestUnicast--;
3605 }
3606 // For all questions (not just the ones we're sending) check what the next scheduled event will be
3607 // We don't need to consider NewQuestions here because for those we'll set m->NextScheduledQuery in AnswerNewQuestion
3608 SetNextQueryTime(m,q);
3609 }
3610
3611 // 2. Scan our authoritative RR list to see what probes we might need to send
3612
3613 m->NextScheduledProbe = m->timenow + 0x78000000;
3614
3615 if (m->CurrentRecord)
3616 LogMsg("SendQueries ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
3617 m->CurrentRecord = m->ResourceRecords;
3618 while (m->CurrentRecord)
3619 {
3620 ar = m->CurrentRecord;
3621 m->CurrentRecord = ar->next;
3622 if (!AuthRecord_uDNS(ar) && ar->resrec.RecordType == kDNSRecordTypeUnique) // For all records that are still probing...
3623 {
3624 // 1. If it's not reached its probe time, just make sure we update m->NextScheduledProbe correctly
3625 if (m->timenow - (ar->LastAPTime + ar->ThisAPInterval) < 0)
3626 {
3627 SetNextAnnounceProbeTime(m, ar);
3628 }
3629 // 2. else, if it has reached its probe time, mark it for sending and then update m->NextScheduledProbe correctly
3630 else if (ar->ProbeCount)
3631 {
3632 if (ar->AddressProxy.type == mDNSAddrType_IPv4)
3633 {
3634 // There's a problem here. If a host is waking up, and we probe to see if it responds, then
3635 // it will see those ARP probes as signalling intent to use the address, so it picks a different one.
3636 // A more benign way to find out if a host is responding to ARPs might be send a standard ARP *request*
3637 // (using our sender IP address) instead of an ARP *probe* (using all-zero sender IP address).
3638 // A similar concern may apply to the NDP Probe too. -- SC
3639 LogSPS("SendQueries ARP Probe %d %s %s", ar->ProbeCount, InterfaceNameForID(m, ar->resrec.InterfaceID), ARDisplayString(m,ar));
3640 SendARP(m, 1, ar, &zerov4Addr, &zeroEthAddr, &ar->AddressProxy.ip.v4, &ar->WakeUp.IMAC);
3641 }
3642 else if (ar->AddressProxy.type == mDNSAddrType_IPv6)
3643 {
3644 LogSPS("SendQueries NDP Probe %d %s %s", ar->ProbeCount, InterfaceNameForID(m, ar->resrec.InterfaceID), ARDisplayString(m,ar));
3645 // IPv6 source = zero
3646 // No target hardware address
3647 // IPv6 target address is address we're probing
3648 // Ethernet destination address is Ethernet interface address of the Sleep Proxy client we're probing
3649 SendNDP(m, NDP_Sol, 0, ar, &zerov6Addr, mDNSNULL, &ar->AddressProxy.ip.v6, &ar->WakeUp.IMAC);
3650 }
3651 // Mark for sending. (If no active interfaces, then don't even try.)
3652 ar->SendRNow = (!intf || ar->WakeUp.HMAC.l[0]) ? mDNSNULL : ar->resrec.InterfaceID ? ar->resrec.InterfaceID : intf->InterfaceID;
3653 ar->LastAPTime = m->timenow;
3654 // When we have a late conflict that resets a record to probing state we use a special marker value greater
3655 // than DefaultProbeCountForTypeUnique. Here we detect that state and reset ar->ProbeCount back to the right value.
3656 if (ar->ProbeCount > DefaultProbeCountForTypeUnique)
3657 ar->ProbeCount = DefaultProbeCountForTypeUnique;
3658 ar->ProbeCount--;
3659 SetNextAnnounceProbeTime(m, ar);
3660 if (ar->ProbeCount == 0)
3661 {
3662 // If this is the last probe for this record, then see if we have any matching records
3663 // on our duplicate list which should similarly have their ProbeCount cleared to zero...
3664 AuthRecord *r2;
3665 for (r2 = m->DuplicateRecords; r2; r2=r2->next)
3666 if (r2->resrec.RecordType == kDNSRecordTypeUnique && RecordIsLocalDuplicate(r2, ar))
3667 r2->ProbeCount = 0;
3668 // ... then acknowledge this record to the client.
3669 // We do this optimistically, just as we're about to send the third probe.
3670 // This helps clients that both advertise and browse, and want to filter themselves
3671 // from the browse results list, because it helps ensure that the registration
3672 // confirmation will be delivered 1/4 second *before* the browse "add" event.
3673 // A potential downside is that we could deliver a registration confirmation and then find out
3674 // moments later that there's a name conflict, but applications have to be prepared to handle
3675 // late conflicts anyway (e.g. on connection of network cable, etc.), so this is nothing new.
3676 if (!ar->Acknowledged) AcknowledgeRecord(m, ar);
3677 }
3678 }
3679 // else, if it has now finished probing, move it to state Verified,
3680 // and update m->NextScheduledResponse so it will be announced
3681 else
3682 {
3683 if (!ar->Acknowledged) AcknowledgeRecord(m, ar); // Defensive, just in case it got missed somehow
3684 ar->resrec.RecordType = kDNSRecordTypeVerified;
3685 ar->ThisAPInterval = DefaultAnnounceIntervalForTypeUnique;
3686 ar->LastAPTime = m->timenow - DefaultAnnounceIntervalForTypeUnique;
3687 SetNextAnnounceProbeTime(m, ar);
3688 }
3689 }
3690 }
3691 m->CurrentRecord = m->DuplicateRecords;
3692 while (m->CurrentRecord)
3693 {
3694 ar = m->CurrentRecord;
3695 m->CurrentRecord = ar->next;
3696 if (ar->resrec.RecordType == kDNSRecordTypeUnique && ar->ProbeCount == 0 && !ar->Acknowledged)
3697 AcknowledgeRecord(m, ar);
3698 }
3699
3700 // 3. Now we know which queries and probes we're sending,
3701 // go through our interface list sending the appropriate queries on each interface
3702 while (intf)
3703 {
3704 int OwnerRecordSpace = (m->AnnounceOwner && intf->MAC.l[0]) ? DNSOpt_Header_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) : 0;
3705 int TraceRecordSpace = (mDNS_McastTracingEnabled && MDNS_TRACER) ? DNSOpt_Header_Space + DNSOpt_TraceData_Space : 0;
3706 mDNSu8 *queryptr = m->omsg.data;
3707 mDNSBool useBackgroundTrafficClass = mDNSfalse; // set if we should use background traffic class
3708
3709 InitializeDNSMessage(&m->omsg.h, zeroID, QueryFlags);
3710 if (KnownAnswerList) verbosedebugf("SendQueries: KnownAnswerList set... Will continue from previous packet");
3711 if (!KnownAnswerList)
3712 {
3713 // Start a new known-answer list
3714 CacheRecord **kalistptr = &KnownAnswerList;
3715 mDNSu32 answerforecast = OwnerRecordSpace + TraceRecordSpace; // Start by assuming we'll need at least enough space to put the Owner+Tracer Option
3716
3717 // Put query questions in this packet
3718 for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
3719 {
3720 if (mDNSOpaque16IsZero(q->TargetQID) && (q->SendQNow == intf->InterfaceID))
3721 {
3722 mDNSBool Suppress = mDNSfalse;
3723 debugf("SendQueries: %s question for %##s (%s) at %d forecast total %d",
3724 SuppressOnThisInterface(q->DupSuppress, intf) ? "Suppressing" : "Putting ",
3725 q->qname.c, DNSTypeName(q->qtype), queryptr - m->omsg.data, queryptr + answerforecast - m->omsg.data);
3726
3727 // If interface is P2P type, verify that query should be sent over it.
3728 if (!mDNSPlatformValidQuestionForInterface(q, intf))
3729 {
3730 q->SendQNow = (q->InterfaceID || !q->SendOnAll) ? mDNSNULL : GetNextActiveInterfaceID(intf);
3731 }
3732 // If we're suppressing this question, or we successfully put it, update its SendQNow state
3733 else if ((Suppress = SuppressOnThisInterface(q->DupSuppress, intf)) ||
3734 BuildQuestion(m, intf, &m->omsg, &queryptr, q, &kalistptr, &answerforecast))
3735 {
3736 // We successfully added the question to the packet. Make sure that
3737 // we also send the NSEC3 record if required. BuildQuestion accounted for
3738 // the space.
3739 //
3740 // Note: We don't suppress anonymous questions and hence Suppress should always
3741 // be zero.
3742
3743 if (Suppress)
3744 m->mDNSStats.DupQuerySuppressions++;
3745
3746 if (!Suppress && q->AnonInfo)
3747 {
3748 debugf("SendQueries: marking for question %##s, Suppress %d", q->qname.c, Suppress);
3749 q->AnonInfo->SendNow = intf->InterfaceID;
3750 }
3751 q->SendQNow = (q->InterfaceID || !q->SendOnAll) ? mDNSNULL : GetNextActiveInterfaceID(intf);
3752 if (q->WakeOnResolveCount)
3753 {
3754 mDNSSendWakeOnResolve(m, q);
3755 q->WakeOnResolveCount--;
3756 }
3757
3758 // use background traffic class if any included question requires it
3759 if (q->UseBackgroundTrafficClass)
3760 {
3761 useBackgroundTrafficClass = mDNStrue;
3762 }
3763 }
3764 }
3765 }
3766
3767 // Put probe questions in this packet
3768 for (ar = m->ResourceRecords; ar; ar=ar->next)
3769 {
3770 if (ar->SendRNow != intf->InterfaceID)
3771 continue;
3772
3773 // If interface is a P2P variant, verify that the probe should be sent over it.
3774 if (!mDNSPlatformValidRecordForInterface(ar, intf->InterfaceID))
3775 {
3776 ar->SendRNow = (ar->resrec.InterfaceID) ? mDNSNULL : GetNextActiveInterfaceID(intf);
3777 ar->IncludeInProbe = mDNSfalse;
3778 }
3779 else
3780 {
3781 mDNSBool ucast = (ar->ProbeCount >= DefaultProbeCountForTypeUnique-1) && m->CanReceiveUnicastOn5353 && intf->SupportsUnicastMDNSResponse;
3782 mDNSu16 ucbit = (mDNSu16)(ucast ? kDNSQClass_UnicastResponse : 0);
3783 const mDNSu8 *const limit = m->omsg.data + (m->omsg.h.numQuestions ? NormalMaxDNSMessageData : AbsoluteMaxDNSMessageData);
3784 // We forecast: compressed name (2) type (2) class (2) TTL (4) rdlength (2) rdata (n)
3785 mDNSu32 forecast = answerforecast + 12 + ar->resrec.rdestimate;
3786 mDNSBool putProbe = mDNStrue;
3787 mDNSu16 qclass = ar->resrec.rrclass | ucbit;
3788
3789 {// Determine if this probe question is already in packet's dns message
3790 const mDNSu8 *questionptr = m->omsg.data;
3791 DNSQuestion question;
3792 mDNSu16 n;
3793 for (n = 0; n < m->omsg.h.numQuestions && questionptr; n++)
3794 {
3795 questionptr = getQuestion(&m->omsg, questionptr, limit, mDNSInterface_Any, &question);
3796 if (questionptr && (question.qtype == kDNSQType_ANY) && (question.qclass == qclass) &&
3797 (question.qnamehash == ar->resrec.namehash) && SameDomainName(&question.qname, ar->resrec.name))
3798 {
3799 putProbe = mDNSfalse; // set to false if already in message
3800 break;
3801 }
3802 }
3803 }
3804
3805 if (putProbe)
3806 {
3807 mDNSu8 *newptr = putQuestion(&m->omsg, queryptr, limit - forecast, ar->resrec.name, kDNSQType_ANY, qclass);
3808 if (newptr)
3809 {
3810 queryptr = newptr;
3811 answerforecast = forecast;
3812 ar->SendRNow = (ar->resrec.InterfaceID) ? mDNSNULL : GetNextActiveInterfaceID(intf);
3813 ar->IncludeInProbe = mDNStrue;
3814 verbosedebugf("SendQueries: Put Question %##s (%s) probecount %d InterfaceID= %d %d %d",
3815 ar->resrec.name->c, DNSTypeName(ar->resrec.rrtype), ar->ProbeCount, ar->resrec.InterfaceID, ar->resrec.rdestimate, answerforecast);
3816 }
3817 }
3818 else
3819 {
3820 ar->SendRNow = (ar->resrec.InterfaceID) ? mDNSNULL : GetNextActiveInterfaceID(intf);
3821 ar->IncludeInProbe = mDNStrue;
3822 }
3823 }
3824 }
3825 }
3826
3827 // Put our known answer list (either new one from this question or questions, or remainder of old one from last time)
3828 while (KnownAnswerList)
3829 {
3830 CacheRecord *ka = KnownAnswerList;
3831 mDNSu32 SecsSinceRcvd = ((mDNSu32)(m->timenow - ka->TimeRcvd)) / mDNSPlatformOneSecond;
3832 mDNSu8 *newptr = PutResourceRecordTTLWithLimit(&m->omsg, queryptr, &m->omsg.h.numAnswers, &ka->resrec, ka->resrec.rroriginalttl - SecsSinceRcvd,
3833 m->omsg.data + NormalMaxDNSMessageData - OwnerRecordSpace - TraceRecordSpace);
3834 if (newptr)
3835 {
3836 verbosedebugf("SendQueries: Put %##s (%s) at %d - %d",
3837 ka->resrec.name->c, DNSTypeName(ka->resrec.rrtype), queryptr - m->omsg.data, newptr - m->omsg.data);
3838 queryptr = newptr;
3839 KnownAnswerList = ka->NextInKAList;
3840 ka->NextInKAList = mDNSNULL;
3841 }
3842 else
3843 {
3844 // If we ran out of space and we have more than one question in the packet, that's an error --
3845 // we shouldn't have put more than one question if there was a risk of us running out of space.
3846 if (m->omsg.h.numQuestions > 1)
3847 LogMsg("SendQueries: Put %d answers; No more space for known answers", m->omsg.h.numAnswers);
3848 m->omsg.h.flags.b[0] |= kDNSFlag0_TC;
3849 break;
3850 }
3851 }
3852
3853 for (ar = m->ResourceRecords; ar; ar=ar->next)
3854 {
3855 if (ar->IncludeInProbe)
3856 {
3857 mDNSu8 *newptr = PutResourceRecord(&m->omsg, queryptr, &m->omsg.h.numAuthorities, &ar->resrec);
3858 ar->IncludeInProbe = mDNSfalse;
3859 if (newptr) queryptr = newptr;
3860 else LogMsg("SendQueries: How did we fail to have space for the Update record %s", ARDisplayString(m,ar));
3861 }
3862 }
3863
3864 for (q = m->Questions; q; q = q->next)
3865 {
3866 if (q->AnonInfo && q->AnonInfo->SendNow == intf->InterfaceID)
3867 {
3868 mDNSu8 *newptr = PutResourceRecord(&m->omsg, queryptr, &m->omsg.h.numAuthorities, q->AnonInfo->nsec3RR);
3869 if (newptr)
3870 {
3871 debugf("SendQueries: Added NSEC3 record %s on InterfaceID %p", RRDisplayString(m, q->AnonInfo->nsec3RR), intf->InterfaceID);
3872 queryptr = newptr;
3873 }
3874 else
3875 {
3876 LogMsg("SendQueries: ERROR!! Cannot add NSEC3 record %s on InterfaceID %p", RRDisplayString(m, q->AnonInfo->nsec3RR), intf->InterfaceID);
3877 }
3878 q->AnonInfo->SendNow = mDNSNULL;
3879 }
3880 }
3881
3882 if (queryptr > m->omsg.data)
3883 {
3884 // If we have data to send, add OWNER/TRACER/OWNER+TRACER option if necessary, then send packet
3885 if (OwnerRecordSpace || TraceRecordSpace)
3886 {
3887 AuthRecord opt;
3888 mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
3889 opt.resrec.rrclass = NormalMaxDNSMessageData;
3890 opt.resrec.rdlength = sizeof(rdataOPT);
3891 opt.resrec.rdestimate = sizeof(rdataOPT);
3892 if (OwnerRecordSpace && TraceRecordSpace)
3893 {
3894 opt.resrec.rdlength += sizeof(rdataOPT); // Two options in this OPT record
3895 opt.resrec.rdestimate += sizeof(rdataOPT);
3896 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
3897 SetupTracerOpt(m, &opt.resrec.rdata->u.opt[1]);
3898 }
3899 else if (OwnerRecordSpace)
3900 {
3901 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
3902 }
3903 else if (TraceRecordSpace)
3904 {
3905 SetupTracerOpt(m, &opt.resrec.rdata->u.opt[0]);
3906 }
3907 queryptr = PutResourceRecordTTLWithLimit(&m->omsg, queryptr, &m->omsg.h.numAdditionals,
3908 &opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
3909 if (!queryptr)
3910 {
3911 LogMsg("SendQueries: How did we fail to have space for %s %s OPT record (%d/%d/%d/%d) %s", OwnerRecordSpace ? "OWNER" : "", TraceRecordSpace ? "TRACER" : "",
3912 m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
3913 }
3914 if (queryptr > m->omsg.data + NormalMaxDNSMessageData)
3915 {
3916 if (m->omsg.h.numQuestions != 1 || m->omsg.h.numAnswers != 0 || m->omsg.h.numAuthorities != 1 || m->omsg.h.numAdditionals != 1)
3917 LogMsg("SendQueries: Why did we generate oversized packet with %s %s OPT record %p %p %p (%d/%d/%d/%d) %s", OwnerRecordSpace ? "OWNER" : "",
3918 TraceRecordSpace ? "TRACER" : "", m->omsg.data, m->omsg.data + NormalMaxDNSMessageData, queryptr, m->omsg.h.numQuestions, m->omsg.h.numAnswers,
3919 m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
3920 }
3921 }
3922
3923 if ((m->omsg.h.flags.b[0] & kDNSFlag0_TC) && m->omsg.h.numQuestions > 1)
3924 LogMsg("SendQueries: Should not have more than one question (%d) in a truncated packet", m->omsg.h.numQuestions);
3925 debugf("SendQueries: Sending %d Question%s %d Answer%s %d Update%s on %p",
3926 m->omsg.h.numQuestions, m->omsg.h.numQuestions == 1 ? "" : "s",
3927 m->omsg.h.numAnswers, m->omsg.h.numAnswers == 1 ? "" : "s",
3928 m->omsg.h.numAuthorities, m->omsg.h.numAuthorities == 1 ? "" : "s", intf->InterfaceID);
3929 if (intf->IPv4Available) mDNSSendDNSMessage(m, &m->omsg, queryptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v4, MulticastDNSPort, mDNSNULL, mDNSNULL, useBackgroundTrafficClass);
3930 if (intf->IPv6Available) mDNSSendDNSMessage(m, &m->omsg, queryptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v6, MulticastDNSPort, mDNSNULL, mDNSNULL, useBackgroundTrafficClass);
3931 if (!m->SuppressSending) m->SuppressSending = NonZeroTime(m->timenow + (mDNSPlatformOneSecond+9)/10);
3932 if (++pktcount >= 1000)
3933 { LogMsg("SendQueries exceeded loop limit %d: giving up", pktcount); break; }
3934 // There might be more records left in the known answer list, or more questions to send
3935 // on this interface, so go around one more time and try again.
3936 }
3937 else // Nothing more to send on this interface; go to next
3938 {
3939 const NetworkInterfaceInfo *next = GetFirstActiveInterface(intf->next);
3940 #if MDNS_DEBUGMSGS && 0
3941 const char *const msg = next ? "SendQueries: Nothing more on %p; moving to %p" : "SendQueries: Nothing more on %p";
3942 debugf(msg, intf, next);
3943 #endif
3944 intf = next;
3945 }
3946 }
3947
3948 // 4. Final housekeeping
3949
3950 // 4a. Debugging check: Make sure we announced all our records
3951 for (ar = m->ResourceRecords; ar; ar=ar->next)
3952 if (ar->SendRNow)
3953 {
3954 if (ar->ARType != AuthRecordLocalOnly && ar->ARType != AuthRecordP2P)
3955 LogInfo("SendQueries: No active interface %d to send probe: %d %s",
3956 (uint32_t)ar->SendRNow, (uint32_t)ar->resrec.InterfaceID, ARDisplayString(m, ar));
3957 ar->SendRNow = mDNSNULL;
3958 }
3959
3960 // 4b. When we have lingering cache records that we're keeping around for a few seconds in the hope
3961 // that their interface which went away might come back again, the logic will want to send queries
3962 // for those records, but we can't because their interface isn't here any more, so to keep the
3963 // state machine ticking over we just pretend we did so.
3964 // If the interface does not come back in time, the cache record will expire naturally
3965 FORALL_CACHERECORDS(slot, cg, cr)
3966 {
3967 if (cr->CRActiveQuestion && cr->UnansweredQueries < MaxUnansweredQueries)
3968 {
3969 if (m->timenow + TicksTTL(cr)/50 - cr->NextRequiredQuery >= 0)
3970 {
3971 cr->UnansweredQueries++;
3972 cr->CRActiveQuestion->SendQNow = mDNSNULL;
3973 SetNextCacheCheckTimeForRecord(m, cr);
3974 }
3975 }
3976 }
3977
3978 // 4c. Debugging check: Make sure we sent all our planned questions
3979 // Do this AFTER the lingering cache records check above, because that will prevent spurious warnings for questions
3980 // we legitimately couldn't send because the interface is no longer available
3981 for (q = m->Questions; q; q=q->next)
3982 {
3983 if (q->SendQNow)
3984 {
3985 DNSQuestion *x;
3986 for (x = m->NewQuestions; x; x=x->next) if (x == q) break; // Check if this question is a NewQuestion
3987 // There will not be an active interface for questions applied to mDNSInterface_BLE
3988 // so don't log the warning in that case.
3989 if (q->InterfaceID != mDNSInterface_BLE)
3990 LogInfo("SendQueries: No active interface %d to send %s question: %d %##s (%s)",
3991 (uint32_t)q->SendQNow, x ? "new" : "old", (uint32_t)q->InterfaceID, q->qname.c, DNSTypeName(q->qtype));
3992 q->SendQNow = mDNSNULL;
3993 }
3994 q->CachedAnswerNeedsUpdate = mDNSfalse;
3995 }
3996 }
3997
3998 mDNSlocal void SendWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *EthAddr, mDNSOpaque48 *password, mDNSBool unicastOnly)
3999 {
4000 int i, j;
4001
4002 mDNSu8 *ptr = m->omsg.data;
4003 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
4004 if (!intf) { LogMsg("SendARP: No interface with InterfaceID %p found", InterfaceID); return; }
4005
4006 // 0x00 Destination address
4007 for (i=0; i<6; i++) *ptr++ = EthAddr->b[i];
4008
4009 // 0x06 Source address (Note: Since we don't currently set the BIOCSHDRCMPLT option, BPF will fill in the real interface address for us)
4010 for (i=0; i<6; i++) *ptr++ = intf->MAC.b[0];
4011
4012 // 0x0C Ethertype (0x0842)
4013 *ptr++ = 0x08;
4014 *ptr++ = 0x42;
4015
4016 // 0x0E Wakeup sync sequence
4017 for (i=0; i<6; i++) *ptr++ = 0xFF;
4018
4019 // 0x14 Wakeup data
4020 for (j=0; j<16; j++) for (i=0; i<6; i++) *ptr++ = EthAddr->b[i];
4021
4022 // 0x74 Password
4023 for (i=0; i<6; i++) *ptr++ = password->b[i];
4024
4025 mDNSPlatformSendRawPacket(m->omsg.data, ptr, InterfaceID);
4026
4027 if (!unicastOnly)
4028 {
4029 // For Ethernet switches that don't flood-foward packets with unknown unicast destination MAC addresses,
4030 // broadcast is the only reliable way to get a wakeup packet to the intended target machine.
4031 // For 802.11 WPA networks, where a sleeping target machine may have missed a broadcast/multicast
4032 // key rotation, unicast is the only way to get a wakeup packet to the intended target machine.
4033 // So, we send one of each, unicast first, then broadcast second.
4034 for (i=0; i<6; i++) m->omsg.data[i] = 0xFF;
4035 mDNSPlatformSendRawPacket(m->omsg.data, ptr, InterfaceID);
4036 }
4037 }
4038
4039 // ***************************************************************************
4040 #if COMPILER_LIKES_PRAGMA_MARK
4041 #pragma mark -
4042 #pragma mark - RR List Management & Task Management
4043 #endif
4044
4045 // Whenever a question is answered, reset its state so that we don't query
4046 // the network repeatedly. This happens first time when we answer the question and
4047 // and later when we refresh the cache.
4048 mDNSlocal void ResetQuestionState(mDNS *const m, DNSQuestion *q)
4049 {
4050 q->LastQTime = m->timenow;
4051 q->LastQTxTime = m->timenow;
4052 q->RecentAnswerPkts = 0;
4053 q->ThisQInterval = MaxQuestionInterval;
4054 q->RequestUnicast = 0;
4055 // Reset unansweredQueries so that we don't penalize this server later when we
4056 // start sending queries when the cache expires.
4057 q->unansweredQueries = 0;
4058 debugf("ResetQuestionState: Set MaxQuestionInterval for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4059 }
4060
4061 // Note: AnswerCurrentQuestionWithResourceRecord can call a user callback, which may change the record list and/or question list.
4062 // Any code walking either list must use the m->CurrentQuestion (and possibly m->CurrentRecord) mechanism to protect against this.
4063 // In fact, to enforce this, the routine will *only* answer the question currently pointed to by m->CurrentQuestion,
4064 // which will be auto-advanced (possibly to NULL) if the client callback cancels the question.
4065 mDNSexport void AnswerCurrentQuestionWithResourceRecord(mDNS *const m, CacheRecord *const rr, const QC_result AddRecord)
4066 {
4067 DNSQuestion *const q = m->CurrentQuestion;
4068 const mDNSBool followcname = FollowCNAME(q, &rr->resrec, AddRecord);
4069
4070 verbosedebugf("AnswerCurrentQuestionWithResourceRecord:%4lu %s TTL %d %s",
4071 q->CurrentAnswers, AddRecord ? "Add" : "Rmv", rr->resrec.rroriginalttl, CRDisplayString(m, rr));
4072
4073 // When the response for the question was validated, the entire rrset was validated. If we deliver
4074 // a RMV for a single record in the rrset, we invalidate the response. If we deliver another add
4075 // in the future, we will do the revalidation again.
4076 //
4077 // Also, if we deliver an ADD for a negative cache record and it has no NSEC/NSEC3, the ValidationStatus needs
4078 // to be reset. This happens normally when we deliver a "secure" negative response followed by an insecure
4079 // negative response which can happen e.g., when disconnecting from network that leads to a negative response
4080 // due to no DNS servers. As we don't deliver RMVs for negative responses that were delivered before, we need
4081 // to do it on the next ADD of a negative cache record. This ADD could be the result of a timeout, no DNS servers
4082 // etc. in which case we need to reset the state to make sure we don't deliver them as secure. If this is
4083 // a real negative response, we would reset the state here and validate the results at the end of this function.
4084 // or the real response again if we purge the cache.
4085 if (q->ValidationRequired && ((AddRecord == QC_rmv) ||
4086 (rr->resrec.RecordType == kDNSRecordTypePacketNegative && (AddRecord == QC_add))))
4087 {
4088 q->ValidationStatus = 0;
4089 q->ValidationState = DNSSECValRequired;
4090 }
4091
4092 // Normally we don't send out the unicast query if we have answered using our local only auth records e.g., /etc/hosts.
4093 // But if the query for "A" record has a local answer but query for "AAAA" record has no local answer, we might
4094 // send the AAAA query out which will come back with CNAME and will also answer the "A" query. To prevent that,
4095 // we check to see if that query already has a unique local answer.
4096 if (q->LOAddressAnswers)
4097 {
4098 LogInfo("AnswerCurrentQuestionWithResourceRecord: Question %p %##s (%s) not answering with record %s due to "
4099 "LOAddressAnswers %d", q, q->qname.c, DNSTypeName(q->qtype), ARDisplayString(m, rr),
4100 q->LOAddressAnswers);
4101 return;
4102 }
4103
4104 if (QuerySuppressed(q))
4105 {
4106 // If the query is suppressed, then we don't want to answer from the cache. But if this query is
4107 // supposed to time out, we still want to callback the clients. We do this only for TimeoutQuestions
4108 // that are timing out, which we know are answered with negative cache record when timing out.
4109 if (!q->TimeoutQuestion || rr->resrec.RecordType != kDNSRecordTypePacketNegative || (m->timenow - q->StopTime < 0))
4110 return;
4111 }
4112
4113 #if TARGET_OS_EMBEDDED
4114 if ((AddRecord == QC_add) && Question_uDNS(q) && !followcname)
4115 {
4116 const domainname * queryName;
4117 mDNSu32 responseLatencyMs;
4118 mDNSBool isForCellular;
4119
4120 queryName = q->metrics.originalQName ? q->metrics.originalQName : &q->qname;
4121 isForCellular = (q->qDNSServer && q->qDNSServer->cellIntf);
4122 if (!q->metrics.answered)
4123 {
4124 if (q->metrics.querySendCount > 0)
4125 {
4126 responseLatencyMs = ((m->timenow - q->metrics.firstQueryTime) * 1000) / mDNSPlatformOneSecond;
4127 }
4128 else
4129 {
4130 responseLatencyMs = 0;
4131 }
4132
4133 MetricsUpdateUDNSQueryStats(queryName, q->qtype, &rr->resrec, q->metrics.querySendCount, responseLatencyMs, isForCellular);
4134 q->metrics.answered = mDNStrue;
4135 }
4136 if (q->metrics.querySendCount > 0)
4137 {
4138 MetricsUpdateUDNSResolveStats(queryName, &rr->resrec, isForCellular);
4139 }
4140 }
4141 #endif
4142 // Note: Use caution here. In the case of records with rr->DelayDelivery set, AnswerCurrentQuestionWithResourceRecord(... mDNStrue)
4143 // may be called twice, once when the record is received, and again when it's time to notify local clients.
4144 // If any counters or similar are added here, care must be taken to ensure that they are not double-incremented by this.
4145
4146 rr->LastUsed = m->timenow;
4147 if (AddRecord == QC_add && !q->DuplicateOf && rr->CRActiveQuestion != q)
4148 {
4149 if (!rr->CRActiveQuestion) m->rrcache_active++; // If not previously active, increment rrcache_active count
4150 debugf("AnswerCurrentQuestionWithResourceRecord: Updating CRActiveQuestion from %p to %p for cache record %s, CurrentAnswer %d",
4151 rr->CRActiveQuestion, q, CRDisplayString(m,rr), q->CurrentAnswers);
4152 rr->CRActiveQuestion = q; // We know q is non-null
4153 SetNextCacheCheckTimeForRecord(m, rr);
4154 }
4155
4156 // If this is:
4157 // (a) a no-cache add, where we've already done at least one 'QM' query, or
4158 // (b) a normal add, where we have at least one unique-type answer,
4159 // then there's no need to keep polling the network.
4160 // (If we have an answer in the cache, then we'll automatically ask again in time to stop it expiring.)
4161 // We do this for mDNS questions and uDNS one-shot questions, but not for
4162 // uDNS LongLived questions, because that would mess up our LLQ lease renewal timing.
4163 if ((AddRecord == QC_addnocache && !q->RequestUnicast) ||
4164 (AddRecord == QC_add && (q->ExpectUnique || (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask))))
4165 if (ActiveQuestion(q) && (mDNSOpaque16IsZero(q->TargetQID) || !q->LongLived))
4166 {
4167 ResetQuestionState(m, q);
4168 }
4169
4170 if (rr->DelayDelivery) return; // We'll come back later when CacheRecordDeferredAdd() calls us
4171
4172 #ifdef USE_LIBIDN
4173 if (rr->resrec.RecordType == kDNSRecordTypePacketNegative) // If negative answer, check if we need to try Punycode conversion
4174 {
4175 domainname newname;
4176 if (PerformNextPunycodeConversion(q, &newname)) // Itertative Punycode conversion succeeded, so reissue question with new name
4177 {
4178 UDPSocket *const sock = q->LocalSocket; // Save old socket and transaction ID
4179 const mDNSOpaque16 id = q->TargetQID;
4180 q->LocalSocket = mDNSNULL;
4181 mDNS_StopQuery_internal(m, q); // Stop old query
4182 AssignDomainName(&q->qname, &newname); // Update qname
4183 q->qnamehash = DomainNameHashValue(&q->qname); // and namehash
4184 mDNS_StartQuery_internal(m, q); // Start new query
4185
4186 if (sock) // Transplant saved socket, if appropriate
4187 {
4188 if (q->DuplicateOf) mDNSPlatformUDPClose(sock);
4189 else { q->LocalSocket = sock; q->TargetQID = id; }
4190 }
4191 return; // All done for now; wait until we get the next answer
4192 }
4193 }
4194 #endif // USE_LIBIDN
4195
4196 // Only deliver negative answers if client has explicitly requested them except when we are forcing a negative response
4197 // for the purpose of retrying search domains/timeout OR the question is suppressed
4198 if (rr->resrec.RecordType == kDNSRecordTypePacketNegative || (q->qtype != kDNSType_NSEC && RRAssertsNonexistence(&rr->resrec, q->qtype)))
4199 if (!AddRecord || (AddRecord != QC_suppressed && AddRecord != QC_forceresponse && !q->ReturnIntermed)) return;
4200
4201 // For CNAME results to non-CNAME questions, only inform the client if they explicitly requested that
4202 if (q->QuestionCallback && !q->NoAnswer && (!followcname || q->ReturnIntermed))
4203 {
4204 mDNS_DropLockBeforeCallback(); // Allow client (and us) to legally make mDNS API calls
4205 if (q->qtype != kDNSType_NSEC && RRAssertsNonexistence(&rr->resrec, q->qtype))
4206 {
4207 CacheRecord neg;
4208 MakeNegativeCacheRecord(m, &neg, &q->qname, q->qnamehash, q->qtype, q->qclass, 1, rr->resrec.InterfaceID, q->qDNSServer);
4209 q->QuestionCallback(m, q, &neg.resrec, AddRecord);
4210 }
4211 else
4212 q->QuestionCallback(m, q, &rr->resrec, AddRecord);
4213 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
4214 }
4215 // If this is an "Add" operation and this question needs validation, validate the response.
4216 // In the case of negative responses, extra care should be taken. Negative cache records are
4217 // used for many purposes. For example,
4218 //
4219 // 1) Suppressing questions (SuppressUnusable)
4220 // 2) Timeout questions
4221 // 3) The name does not exist
4222 // 4) No DNS servers are available and we need a quick response for the application
4223 //
4224 // (1) and (2) are handled by "QC_add" check as AddRecord would be "QC_forceresponse" or "QC_suppressed"
4225 // in that case. For (3), it is possible that we don't get nsecs back but we still need to call
4226 // VerifySignature so that we can deliver the appropriate DNSSEC result. There is no point in verifying
4227 // signature for (4) and hence the explicit check for q->qDNSServer.
4228 //
4229 if (m->CurrentQuestion == q && (AddRecord == QC_add) && !q->ValidatingResponse && q->ValidationRequired &&
4230 q->ValidationState == DNSSECValRequired && q->qDNSServer)
4231 {
4232 q->ValidationState = DNSSECValInProgress;
4233 // Treat it as callback call as that's what dnssec code expects
4234 mDNS_DropLockBeforeCallback(); // Allow client (and us) to legally make mDNS API calls
4235 VerifySignature(m, mDNSNULL, q);
4236 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
4237 return;
4238 }
4239
4240 // Note: Proceed with caution here because client callback function is allowed to do anything,
4241 // including starting/stopping queries, registering/deregistering records, etc.
4242 //
4243 // If we get a CNAME back while we are validating the response (i.e., CNAME for DS, DNSKEY, RRSIG),
4244 // don't follow them. If it is a ValidationRequired question, wait for the CNAME to be validated
4245 // first before following it
4246 if ((m->CurrentQuestion == q) && followcname && !ValidatingQuestion(q))
4247 AnswerQuestionByFollowingCNAME(m, q, &rr->resrec);
4248 }
4249
4250 mDNSlocal void CacheRecordDeferredAdd(mDNS *const m, CacheRecord *rr)
4251 {
4252 rr->DelayDelivery = 0;
4253 if (m->CurrentQuestion)
4254 LogMsg("CacheRecordDeferredAdd ERROR m->CurrentQuestion already set: %##s (%s)",
4255 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4256 m->CurrentQuestion = m->Questions;
4257 while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
4258 {
4259 DNSQuestion *q = m->CurrentQuestion;
4260 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
4261 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
4262 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
4263 m->CurrentQuestion = q->next;
4264 }
4265 m->CurrentQuestion = mDNSNULL;
4266 }
4267
4268 mDNSlocal mDNSs32 CheckForSoonToExpireRecords(mDNS *const m, const domainname *const name, const mDNSu32 namehash, const mDNSu32 slot, mDNSBool *purge)
4269 {
4270 const mDNSs32 threshhold = m->timenow + mDNSPlatformOneSecond; // See if there are any records expiring within one second
4271 const mDNSs32 start = m->timenow - 0x10000000;
4272 mDNSs32 delay = start;
4273 CacheGroup *cg = CacheGroupForName(m, slot, namehash, name);
4274 const CacheRecord *rr;
4275
4276 if (purge)
4277 *purge = mDNSfalse;
4278 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
4279 {
4280 // If there are records that will expire soon, there are cases that need delayed
4281 // delivery of events:
4282 //
4283 // 1) A new cache entry is about to be added as a replacement. The caller needs to
4284 // deliver a RMV (for the current old entry) followed by ADD (for the new entry).
4285 // It needs to schedule the timer for the next cache expiry (ScheduleNextCacheCheckTime),
4286 // so that the cache entry can be purged (purging causes the RMV followed by ADD)
4287 //
4288 // 2) A new question is about to be answered and the caller needs to know whether it's
4289 // scheduling should be delayed so that the question is not answered with this record.
4290 // Instead of delivering an ADD (old entry) followed by RMV (old entry) and another ADD
4291 // (new entry), a single ADD can be delivered by delaying the scheduling of the question
4292 // immediately.
4293 //
4294 // When the unicast cache record is created, it's TTL has been extended beyond its value
4295 // given in the resource record (See RRAdjustTTL). If it is in the "extended" time, the
4296 // cache is already expired and we set "purge" to indicate that. When "purge" is set, the
4297 // return value of the function should be ignored by the callers.
4298 //
4299 // Note: For case (1), "purge" argument is NULL and hence the following checks are skipped.
4300 // It is okay to skip in that case because the cache records have been set to expire almost
4301 // immediately and the extended time does not apply.
4302 //
4303 // Also, if there is already an active question we don't try to optimize as purging the cache
4304 // would end up delivering RMV for the active question and hence we avoid that.
4305
4306 if (purge && !rr->resrec.InterfaceID && !rr->CRActiveQuestion && rr->resrec.rroriginalttl)
4307 {
4308 mDNSu32 uTTL = RRUnadjustedTTL(rr->resrec.rroriginalttl);
4309 if (m->timenow - (rr->TimeRcvd + ((mDNSs32)uTTL * mDNSPlatformOneSecond)) >= 0)
4310 {
4311 LogInfo("CheckForSoonToExpireRecords: %s: rroriginalttl %u, unadjustedTTL %u, currentTTL %u",
4312 CRDisplayString(m, rr), rr->resrec.rroriginalttl, uTTL, (m->timenow - rr->TimeRcvd)/mDNSPlatformOneSecond);
4313 *purge = mDNStrue;
4314 continue;
4315 }
4316 }
4317 if (threshhold - RRExpireTime(rr) >= 0) // If we have records about to expire within a second
4318 {
4319 if (delay - RRExpireTime(rr) < 0) // then delay until after they've been deleted
4320 delay = RRExpireTime(rr);
4321 }
4322 }
4323 if (delay - start > 0)
4324 return(NonZeroTime(delay));
4325 else
4326 return(0);
4327 }
4328
4329 // CacheRecordAdd is only called from CreateNewCacheEntry, *never* directly as a result of a client API call.
4330 // If new questions are created as a result of invoking client callbacks, they will be added to
4331 // the end of the question list, and m->NewQuestions will be set to indicate the first new question.
4332 // rr is a new CacheRecord just received into our cache
4333 // (kDNSRecordTypePacketAns/PacketAnsUnique/PacketAdd/PacketAddUnique).
4334 // Note: CacheRecordAdd calls AnswerCurrentQuestionWithResourceRecord which can call a user callback,
4335 // which may change the record list and/or question list.
4336 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
4337 mDNSlocal void CacheRecordAdd(mDNS *const m, CacheRecord *rr)
4338 {
4339 DNSQuestion *q;
4340
4341 // We stop when we get to NewQuestions -- if we increment their CurrentAnswers/LargeAnswers/UniqueAnswers
4342 // counters here we'll end up double-incrementing them when we do it again in AnswerNewQuestion().
4343 for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
4344 {
4345 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
4346 {
4347 // If this question is one that's actively sending queries, and it's received ten answers within one
4348 // second of sending the last query packet, then that indicates some radical network topology change,
4349 // so reset its exponential backoff back to the start. We must be at least at the eight-second interval
4350 // to do this. If we're at the four-second interval, or less, there's not much benefit accelerating
4351 // because we will anyway send another query within a few seconds. The first reset query is sent out
4352 // randomized over the next four seconds to reduce possible synchronization between machines.
4353 if (q->LastAnswerPktNum != m->PktNum)
4354 {
4355 q->LastAnswerPktNum = m->PktNum;
4356 if (mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q) && ++q->RecentAnswerPkts >= 10 &&
4357 q->ThisQInterval > InitialQuestionInterval * QuestionIntervalStep3 && m->timenow - q->LastQTxTime < mDNSPlatformOneSecond)
4358 {
4359 LogMsg("CacheRecordAdd: %##s (%s) got immediate answer burst (%d); restarting exponential backoff sequence (%d)",
4360 q->qname.c, DNSTypeName(q->qtype), q->RecentAnswerPkts, q->ThisQInterval);
4361 q->LastQTime = m->timenow - InitialQuestionInterval + (mDNSs32)mDNSRandom((mDNSu32)mDNSPlatformOneSecond*4);
4362 q->ThisQInterval = InitialQuestionInterval;
4363 SetNextQueryTime(m,q);
4364 }
4365 }
4366 verbosedebugf("CacheRecordAdd %p %##s (%s) %lu %#a:%d question %p", rr, rr->resrec.name->c,
4367 DNSTypeName(rr->resrec.rrtype), rr->resrec.rroriginalttl, rr->resrec.rDNSServer ?
4368 &rr->resrec.rDNSServer->addr : mDNSNULL, mDNSVal16(rr->resrec.rDNSServer ?
4369 rr->resrec.rDNSServer->port : zeroIPPort), q);
4370 q->CurrentAnswers++;
4371
4372 q->unansweredQueries = 0;
4373 if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers++;
4374 if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers++;
4375 if (q->CurrentAnswers > 4000)
4376 {
4377 static int msgcount = 0;
4378 if (msgcount++ < 10)
4379 LogMsg("CacheRecordAdd: %##s (%s) has %d answers; shedding records to resist DOS attack",
4380 q->qname.c, DNSTypeName(q->qtype), q->CurrentAnswers);
4381 rr->resrec.rroriginalttl = 0;
4382 rr->UnansweredQueries = MaxUnansweredQueries;
4383 }
4384 }
4385 }
4386
4387 if (!rr->DelayDelivery)
4388 {
4389 if (m->CurrentQuestion)
4390 LogMsg("CacheRecordAdd ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4391 m->CurrentQuestion = m->Questions;
4392 while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
4393 {
4394 q = m->CurrentQuestion;
4395 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
4396 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
4397 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
4398 m->CurrentQuestion = q->next;
4399 }
4400 m->CurrentQuestion = mDNSNULL;
4401 }
4402
4403 SetNextCacheCheckTimeForRecord(m, rr);
4404 }
4405
4406 // NoCacheAnswer is only called from mDNSCoreReceiveResponse, *never* directly as a result of a client API call.
4407 // If new questions are created as a result of invoking client callbacks, they will be added to
4408 // the end of the question list, and m->NewQuestions will be set to indicate the first new question.
4409 // rr is a new CacheRecord just received from the wire (kDNSRecordTypePacketAns/AnsUnique/Add/AddUnique)
4410 // but we don't have any place to cache it. We'll deliver question 'add' events now, but we won't have any
4411 // way to deliver 'remove' events in future, nor will we be able to include this in known-answer lists,
4412 // so we immediately bump ThisQInterval up to MaxQuestionInterval to avoid pounding the network.
4413 // Note: NoCacheAnswer calls AnswerCurrentQuestionWithResourceRecord which can call a user callback,
4414 // which may change the record list and/or question list.
4415 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
4416 mDNSlocal void NoCacheAnswer(mDNS *const m, CacheRecord *rr)
4417 {
4418 LogMsg("No cache space: Delivering non-cached result for %##s", m->rec.r.resrec.name->c);
4419 if (m->CurrentQuestion)
4420 LogMsg("NoCacheAnswer ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4421 m->CurrentQuestion = m->Questions;
4422 // We do this for *all* questions, not stopping when we get to m->NewQuestions,
4423 // since we're not caching the record and we'll get no opportunity to do this later
4424 while (m->CurrentQuestion)
4425 {
4426 DNSQuestion *q = m->CurrentQuestion;
4427 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
4428 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_addnocache); // QC_addnocache means "don't expect remove events for this"
4429 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
4430 m->CurrentQuestion = q->next;
4431 }
4432 m->CurrentQuestion = mDNSNULL;
4433 }
4434
4435 // CacheRecordRmv is only called from CheckCacheExpiration, which is called from mDNS_Execute.
4436 // Note that CacheRecordRmv is *only* called for records that are referenced by at least one active question.
4437 // If new questions are created as a result of invoking client callbacks, they will be added to
4438 // the end of the question list, and m->NewQuestions will be set to indicate the first new question.
4439 // rr is an existing cache CacheRecord that just expired and is being deleted
4440 // (kDNSRecordTypePacketAns/PacketAnsUnique/PacketAdd/PacketAddUnique).
4441 // Note: CacheRecordRmv calls AnswerCurrentQuestionWithResourceRecord which can call a user callback,
4442 // which may change the record list and/or question list.
4443 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
4444 mDNSlocal void CacheRecordRmv(mDNS *const m, CacheRecord *rr)
4445 {
4446 if (m->CurrentQuestion)
4447 LogMsg("CacheRecordRmv ERROR m->CurrentQuestion already set: %##s (%s)",
4448 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4449 m->CurrentQuestion = m->Questions;
4450
4451 // We stop when we get to NewQuestions -- for new questions their CurrentAnswers/LargeAnswers/UniqueAnswers counters
4452 // will all still be zero because we haven't yet gone through the cache counting how many answers we have for them.
4453 while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
4454 {
4455 DNSQuestion *q = m->CurrentQuestion;
4456 // When a question enters suppressed state, we generate RMV events and generate a negative
4457 // response. A cache may be present that answers this question e.g., cache entry generated
4458 // before the question became suppressed. We need to skip the suppressed questions here as
4459 // the RMV event has already been generated.
4460 if (!QuerySuppressed(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
4461 {
4462 verbosedebugf("CacheRecordRmv %p %s", rr, CRDisplayString(m, rr));
4463 q->FlappingInterface1 = mDNSNULL;
4464 q->FlappingInterface2 = mDNSNULL;
4465
4466 if (q->CurrentAnswers == 0)
4467 LogMsg("CacheRecordRmv ERROR!!: How can CurrentAnswers already be zero for %p %##s (%s) DNSServer %#a:%d",
4468 q, q->qname.c, DNSTypeName(q->qtype), q->qDNSServer ? &q->qDNSServer->addr : mDNSNULL,
4469 mDNSVal16(q->qDNSServer ? q->qDNSServer->port : zeroIPPort));
4470 else
4471 {
4472 q->CurrentAnswers--;
4473 if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers--;
4474 if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers--;
4475 }
4476
4477 // If we have dropped below the answer threshold for this mDNS question,
4478 // restart the queries at InitialQuestionInterval.
4479 if (mDNSOpaque16IsZero(q->TargetQID) && (q->BrowseThreshold > 0) && (q->CurrentAnswers < q->BrowseThreshold))
4480 {
4481 q->ThisQInterval = InitialQuestionInterval;
4482 q->LastQTime = m->timenow - q->ThisQInterval;
4483 SetNextQueryTime(m,q);
4484 LogInfo("CacheRecordRmv: (%s) %##s dropped below threshold of %d answers",
4485 DNSTypeName(q->qtype), q->qname.c, q->BrowseThreshold);
4486 }
4487 if (rr->resrec.rdata->MaxRDLength) // Never generate "remove" events for negative results
4488 {
4489 if (q->CurrentAnswers == 0)
4490 {
4491 LogInfo("CacheRecordRmv: Last answer for %##s (%s) expired from cache; will reconfirm antecedents",
4492 q->qname.c, DNSTypeName(q->qtype));
4493 ReconfirmAntecedents(m, &q->qname, q->qnamehash, 0);
4494 }
4495 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_rmv);
4496 }
4497 }
4498 if (m->CurrentQuestion == q) // If m->CurrentQuestion was not auto-advanced, do it ourselves now
4499 m->CurrentQuestion = q->next;
4500 }
4501 m->CurrentQuestion = mDNSNULL;
4502 }
4503
4504 mDNSlocal void ReleaseCacheEntity(mDNS *const m, CacheEntity *e)
4505 {
4506 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING >= 1
4507 unsigned int i;
4508 for (i=0; i<sizeof(*e); i++) ((char*)e)[i] = 0xFF;
4509 #endif
4510 e->next = m->rrcache_free;
4511 m->rrcache_free = e;
4512 m->rrcache_totalused--;
4513 }
4514
4515 mDNSlocal void ReleaseCacheGroup(mDNS *const m, CacheGroup **cp)
4516 {
4517 CacheEntity *e = (CacheEntity *)(*cp);
4518 //LogMsg("ReleaseCacheGroup: Releasing CacheGroup for %p, %##s", (*cp)->name->c, (*cp)->name->c);
4519 if ((*cp)->rrcache_tail != &(*cp)->members)
4520 LogMsg("ERROR: (*cp)->members == mDNSNULL but (*cp)->rrcache_tail != &(*cp)->members)");
4521 //if ((*cp)->name != (domainname*)((*cp)->namestorage))
4522 // LogMsg("ReleaseCacheGroup: %##s, %p %p", (*cp)->name->c, (*cp)->name, (domainname*)((*cp)->namestorage));
4523 if ((*cp)->name != (domainname*)((*cp)->namestorage)) mDNSPlatformMemFree((*cp)->name);
4524 (*cp)->name = mDNSNULL;
4525 *cp = (*cp)->next; // Cut record from list
4526 ReleaseCacheEntity(m, e);
4527 }
4528
4529 mDNSlocal void ReleaseAdditionalCacheRecords(mDNS *const m, CacheRecord **rp)
4530 {
4531 while (*rp)
4532 {
4533 CacheRecord *rr = *rp;
4534 *rp = (*rp)->next; // Cut record from list
4535 if (rr->resrec.rdata && rr->resrec.rdata != (RData*)&rr->smallrdatastorage)
4536 {
4537 mDNSPlatformMemFree(rr->resrec.rdata);
4538 rr->resrec.rdata = mDNSNULL;
4539 }
4540 // NSEC or SOA records that are not added to the CacheGroup do not share the name
4541 // of the CacheGroup.
4542 if (rr->resrec.name)
4543 {
4544 debugf("ReleaseAdditionalCacheRecords: freeing cached record %##s (%s)", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
4545 mDNSPlatformMemFree((void *)rr->resrec.name);
4546 rr->resrec.name = mDNSNULL;
4547 }
4548 // Don't count the NSEC3 records used by anonymous browse/reg
4549 if (!rr->resrec.InterfaceID)
4550 {
4551 m->rrcache_totalused_unicast -= rr->resrec.rdlength;
4552 if (DNSSECRecordType(rr->resrec.rrtype))
4553 BumpDNSSECStats(m, kStatsActionDecrement, kStatsTypeMemoryUsage, rr->resrec.rdlength);
4554 }
4555 ReleaseCacheEntity(m, (CacheEntity *)rr);
4556 }
4557 }
4558
4559 mDNSexport void ReleaseCacheRecord(mDNS *const m, CacheRecord *r)
4560 {
4561 CacheGroup *cg;
4562 const mDNSu32 slot = HashSlot(r->resrec.name);
4563
4564 //LogMsg("ReleaseCacheRecord: Releasing %s", CRDisplayString(m, r));
4565 if (r->resrec.rdata && r->resrec.rdata != (RData*)&r->smallrdatastorage) mDNSPlatformMemFree(r->resrec.rdata);
4566 r->resrec.rdata = mDNSNULL;
4567
4568 cg = CacheGroupForRecord(m, slot, &r->resrec);
4569
4570 if (!cg)
4571 {
4572 // It is okay to have this printed for NSEC/NSEC3s
4573 LogInfo("ReleaseCacheRecord: ERROR!! cg NULL for %##s (%s)", r->resrec.name->c, DNSTypeName(r->resrec.rrtype));
4574 }
4575 // When NSEC records are not added to the cache, it is usually cached at the "nsec" list
4576 // of the CacheRecord. But sometimes they may be freed without adding to the "nsec" list
4577 // (which is handled below) and in that case it should be freed here.
4578 if (r->resrec.name && cg && r->resrec.name != cg->name)
4579 {
4580 debugf("ReleaseCacheRecord: freeing %##s (%s)", r->resrec.name->c, DNSTypeName(r->resrec.rrtype));
4581 mDNSPlatformMemFree((void *)r->resrec.name);
4582 }
4583 r->resrec.name = mDNSNULL;
4584
4585 if (r->resrec.AnonInfo)
4586 {
4587 debugf("ReleaseCacheRecord: freeing AnonInfo for %##s (%s)", r->resrec.name->c, DNSTypeName(r->resrec.rrtype));
4588 FreeAnonInfo((void *)r->resrec.AnonInfo);
4589 }
4590 r->resrec.AnonInfo = mDNSNULL;
4591
4592 if (!r->resrec.InterfaceID)
4593 {
4594 m->rrcache_totalused_unicast -= r->resrec.rdlength;
4595 if (DNSSECRecordType(r->resrec.rrtype))
4596 BumpDNSSECStats(m, kStatsActionDecrement, kStatsTypeMemoryUsage, r->resrec.rdlength);
4597 }
4598
4599 ReleaseAdditionalCacheRecords(m, &r->nsec);
4600 ReleaseAdditionalCacheRecords(m, &r->soa);
4601
4602 ReleaseCacheEntity(m, (CacheEntity *)r);
4603 }
4604
4605 // Note: We want to be careful that we deliver all the CacheRecordRmv calls before delivering
4606 // CacheRecordDeferredAdd calls. The in-order nature of the cache lists ensures that all
4607 // callbacks for old records are delivered before callbacks for newer records.
4608 mDNSlocal void CheckCacheExpiration(mDNS *const m, const mDNSu32 slot, CacheGroup *const cg)
4609 {
4610 CacheRecord **rp = &cg->members;
4611
4612 if (m->lock_rrcache) { LogMsg("CheckCacheExpiration ERROR! Cache already locked!"); return; }
4613 m->lock_rrcache = 1;
4614
4615 while (*rp)
4616 {
4617 CacheRecord *const rr = *rp;
4618 mDNSs32 event = RRExpireTime(rr);
4619 if (m->timenow - event >= 0) // If expired, delete it
4620 {
4621 *rp = rr->next; // Cut it from the list
4622
4623 verbosedebugf("CheckCacheExpiration: Deleting%7d %7d %p %s",
4624 m->timenow - rr->TimeRcvd, rr->resrec.rroriginalttl, rr->CRActiveQuestion, CRDisplayString(m, rr));
4625 if (rr->CRActiveQuestion) // If this record has one or more active questions, tell them it's going away
4626 {
4627 DNSQuestion *q = rr->CRActiveQuestion;
4628 // When a cache record is about to expire, we expect to do four queries at 80-82%, 85-87%, 90-92% and
4629 // then 95-97% of the TTL. If the DNS server does not respond, then we will remove the cache entry
4630 // before we pick a new DNS server. As the question interval is set to MaxQuestionInterval, we may
4631 // not send out a query anytime soon. Hence, we need to reset the question interval. If this is
4632 // a normal deferred ADD case, then AnswerCurrentQuestionWithResourceRecord will reset it to
4633 // MaxQuestionInterval. If we have inactive questions referring to negative cache entries,
4634 // don't ressurect them as they will deliver duplicate "No such Record" ADD events
4635 if (!mDNSOpaque16IsZero(q->TargetQID) && !q->LongLived && ActiveQuestion(q))
4636 {
4637 q->ThisQInterval = InitialQuestionInterval;
4638 q->LastQTime = m->timenow - q->ThisQInterval;
4639 SetNextQueryTime(m, q);
4640 }
4641 CacheRecordRmv(m, rr);
4642 m->rrcache_active--;
4643 }
4644 ReleaseCacheRecord(m, rr);
4645 }
4646 else // else, not expired; see if we need to query
4647 {
4648 // If waiting to delay delivery, do nothing until then
4649 if (rr->DelayDelivery && rr->DelayDelivery - m->timenow > 0)
4650 event = rr->DelayDelivery;
4651 else
4652 {
4653 if (rr->DelayDelivery) CacheRecordDeferredAdd(m, rr);
4654 if (rr->CRActiveQuestion && rr->UnansweredQueries < MaxUnansweredQueries)
4655 {
4656 if (m->timenow - rr->NextRequiredQuery < 0) // If not yet time for next query
4657 event = NextCacheCheckEvent(rr); // then just record when we want the next query
4658 else // else trigger our question to go out now
4659 {
4660 // Set NextScheduledQuery to timenow so that SendQueries() will run.
4661 // SendQueries() will see that we have records close to expiration, and send FEQs for them.
4662 m->NextScheduledQuery = m->timenow;
4663 // After sending the query we'll increment UnansweredQueries and call SetNextCacheCheckTimeForRecord(),
4664 // which will correctly update m->NextCacheCheck for us.
4665 event = m->timenow + 0x3FFFFFFF;
4666 }
4667 }
4668 }
4669 verbosedebugf("CheckCacheExpiration:%6d %5d %s",
4670 (event - m->timenow) / mDNSPlatformOneSecond, CacheCheckGracePeriod(rr), CRDisplayString(m, rr));
4671 if (m->rrcache_nextcheck[slot] - event > 0)
4672 m->rrcache_nextcheck[slot] = event;
4673 rp = &rr->next;
4674 }
4675 }
4676 if (cg->rrcache_tail != rp) verbosedebugf("CheckCacheExpiration: Updating CacheGroup tail from %p to %p", cg->rrcache_tail, rp);
4677 cg->rrcache_tail = rp;
4678 m->lock_rrcache = 0;
4679 }
4680
4681 // "LORecord" includes both LocalOnly and P2P record. This function assumes m->CurrentQuestion is pointing to "q".
4682 //
4683 // If "CheckOnly" is set to "true", the question won't be answered but just check to see if there is an answer and
4684 // returns true if there is an answer.
4685 //
4686 // If "CheckOnly" is set to "false", the question will be answered if there is a LocalOnly/P2P record and
4687 // returns true to indicate the same.
4688 mDNSlocal mDNSBool AnswerQuestionWithLORecord(mDNS *const m, DNSQuestion *q, mDNSBool checkOnly)
4689 {
4690 mDNSu32 slot;
4691 AuthRecord *lr;
4692 AuthGroup *ag;
4693
4694 if (m->CurrentRecord)
4695 LogMsg("AnswerQuestionWithLORecord ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
4696
4697 slot = AuthHashSlot(&q->qname);
4698 ag = AuthGroupForName(&m->rrauth, slot, q->qnamehash, &q->qname);
4699 if (ag)
4700 {
4701 m->CurrentRecord = ag->members;
4702 while (m->CurrentRecord && m->CurrentRecord != ag->NewLocalOnlyRecords)
4703 {
4704 AuthRecord *rr = m->CurrentRecord;
4705 m->CurrentRecord = rr->next;
4706 //
4707 // If the question is mDNSInterface_LocalOnly, all records local to the machine should be used
4708 // to answer the query. This is handled in AnswerNewLocalOnlyQuestion.
4709 //
4710 // We handle mDNSInterface_Any and scoped questions here. See LocalOnlyRecordAnswersQuestion for more
4711 // details on how we handle this case. For P2P we just handle "Interface_Any" questions. For LocalOnly
4712 // we handle both mDNSInterface_Any and scoped questions.
4713
4714 if (rr->ARType == AuthRecordLocalOnly || (rr->ARType == AuthRecordP2P && (q->InterfaceID == mDNSInterface_Any || q->InterfaceID == mDNSInterface_BLE)))
4715 if (LocalOnlyRecordAnswersQuestion(rr, q))
4716 {
4717 if (checkOnly)
4718 {
4719 LogInfo("AnswerQuestionWithLORecord: question %##s (%s) answered by %s", q->qname.c, DNSTypeName(q->qtype),
4720 ARDisplayString(m, rr));
4721 m->CurrentRecord = mDNSNULL;
4722 return mDNStrue;
4723 }
4724 AnswerLocalQuestionWithLocalAuthRecord(m, rr, mDNStrue);
4725 if (m->CurrentQuestion != q)
4726 break; // If callback deleted q, then we're finished here
4727 }
4728 }
4729 }
4730 m->CurrentRecord = mDNSNULL;
4731
4732 if (m->CurrentQuestion != q)
4733 {
4734 LogInfo("AnswerQuestionWithLORecord: Question deleted while while answering LocalOnly record answers");
4735 return mDNStrue;
4736 }
4737
4738 if (q->LOAddressAnswers)
4739 {
4740 LogInfo("AnswerQuestionWithLORecord: Question %p %##s (%s) answered using local auth records LOAddressAnswers %d",
4741 q, q->qname.c, DNSTypeName(q->qtype), q->LOAddressAnswers);
4742 return mDNStrue;
4743 }
4744
4745 // Before we go check the cache and ship this query on the wire, we have to be sure that there are
4746 // no local records that could possibly answer this question. As we did not check the NewLocalRecords, we
4747 // need to just peek at them to see whether it will answer this question. If it would answer, pretend
4748 // that we answered. AnswerAllLocalQuestionsWithLocalAuthRecord will answer shortly. This happens normally
4749 // when we add new /etc/hosts entries and restart the question. It is a new question and also a new record.
4750 if (ag)
4751 {
4752 lr = ag->NewLocalOnlyRecords;
4753 while (lr)
4754 {
4755 if (UniqueLocalOnlyRecord(lr) && LocalOnlyRecordAnswersQuestion(lr, q))
4756 {
4757 LogInfo("AnswerQuestionWithLORecord: Question %p %##s (%s) will be answered using new local auth records "
4758 " LOAddressAnswers %d", q, q->qname.c, DNSTypeName(q->qtype), q->LOAddressAnswers);
4759 return mDNStrue;
4760 }
4761 lr = lr->next;
4762 }
4763 }
4764 return mDNSfalse;
4765 }
4766
4767 // Today, we suppress questions (not send them on the wire) for several reasons e.g.,
4768 // AAAA query is suppressed because no IPv6 capability or PID is not allowed to make
4769 // DNS requests. We need to temporarily suspend the suppress status so that we can
4770 // deliver a negative response (AnswerCurrentQuestionWithResourceRecord does not answer
4771 // suppressed questions) and reset it back. In the future, if there are other
4772 // reasons for suppressing the query, this function should be updated.
4773 mDNSlocal void AnswerSuppressedQuestion(mDNS *const m, DNSQuestion *q)
4774 {
4775 mDNSBool SuppressQuery = q->SuppressQuery;
4776 mDNSBool DisallowPID = q->DisallowPID;
4777
4778 // make sure that QuerySuppressed() returns false
4779 q->SuppressQuery = mDNSfalse;
4780 q->DisallowPID = mDNSfalse;
4781
4782 GenerateNegativeResponse(m, mDNSInterface_Any, QC_suppressed);
4783
4784 q->SuppressQuery = SuppressQuery;
4785 q->DisallowPID = DisallowPID;
4786 }
4787
4788 mDNSlocal void AnswerNewQuestion(mDNS *const m)
4789 {
4790 mDNSBool ShouldQueryImmediately = mDNStrue;
4791 DNSQuestion *const q = m->NewQuestions; // Grab the question we're going to answer
4792 mDNSu32 slot = HashSlot(&q->qname);
4793 CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
4794 mDNSBool AnsweredFromCache = mDNSfalse;
4795
4796 verbosedebugf("AnswerNewQuestion: Answering %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4797
4798 if (cg) CheckCacheExpiration(m, slot, cg);
4799 if (m->NewQuestions != q) { LogInfo("AnswerNewQuestion: Question deleted while doing CheckCacheExpiration"); goto exit; }
4800 m->NewQuestions = q->next;
4801 // Advance NewQuestions to the next *after* calling CheckCacheExpiration, because if we advance it first
4802 // then CheckCacheExpiration may give this question add/remove callbacks, and it's not yet ready for that.
4803 //
4804 // Also, CheckCacheExpiration() calls CacheRecordDeferredAdd() and CacheRecordRmv(), which invoke
4805 // client callbacks, which may delete their own or any other question. Our mechanism for detecting
4806 // whether our current m->NewQuestions question got deleted by one of these callbacks is to store the
4807 // value of m->NewQuestions in 'q' before calling CheckCacheExpiration(), and then verify afterwards
4808 // that they're still the same. If m->NewQuestions has changed (because mDNS_StopQuery_internal
4809 // advanced it), that means the question was deleted, so we no longer need to worry about answering
4810 // it (and indeed 'q' is now a dangling pointer, so dereferencing it at all would be bad, and the
4811 // values we computed for slot and cg are now stale and relate to a question that no longer exists).
4812 //
4813 // We can't use the usual m->CurrentQuestion mechanism for this because CacheRecordDeferredAdd() and
4814 // CacheRecordRmv() both use that themselves when walking the list of (non-new) questions generating callbacks.
4815 // Fortunately mDNS_StopQuery_internal auto-advances both m->CurrentQuestion *AND* m->NewQuestions when
4816 // deleting a question, so luckily we have an easy alternative way of detecting if our question got deleted.
4817
4818 if (m->lock_rrcache) LogMsg("AnswerNewQuestion ERROR! Cache already locked!");
4819 // This should be safe, because calling the client's question callback may cause the
4820 // question list to be modified, but should not ever cause the rrcache list to be modified.
4821 // If the client's question callback deletes the question, then m->CurrentQuestion will
4822 // be advanced, and we'll exit out of the loop
4823 m->lock_rrcache = 1;
4824 if (m->CurrentQuestion)
4825 LogMsg("AnswerNewQuestion ERROR m->CurrentQuestion already set: %##s (%s)",
4826 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4827 m->CurrentQuestion = q; // Indicate which question we're answering, so we'll know if it gets deleted
4828
4829 if (q->NoAnswer == NoAnswer_Fail)
4830 {
4831 LogMsg("AnswerNewQuestion: NoAnswer_Fail %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4832 MakeNegativeCacheRecord(m, &m->rec.r, &q->qname, q->qnamehash, q->qtype, q->qclass, 60, mDNSInterface_Any, q->qDNSServer);
4833 q->NoAnswer = NoAnswer_Normal; // Temporarily turn off answer suppression
4834 AnswerCurrentQuestionWithResourceRecord(m, &m->rec.r, QC_addnocache);
4835 // Don't touch the question if it has been stopped already
4836 if (m->CurrentQuestion == q) q->NoAnswer = NoAnswer_Fail; // Restore NoAnswer state
4837 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
4838 }
4839
4840 if (m->CurrentQuestion != q)
4841 {
4842 LogInfo("AnswerNewQuestion: Question deleted while generating NoAnswer_Fail response");
4843 goto exit;
4844 }
4845
4846 // See if we want to tell it about LocalOnly/P2P records. If we answered them using LocalOnly
4847 // or P2P record, then we are done.
4848 if (AnswerQuestionWithLORecord(m, q, mDNSfalse))
4849 goto exit;
4850
4851 // If we are not supposed to answer this question, generate a negative response.
4852 // Temporarily suspend the SuppressQuery so that AnswerCurrentQuestionWithResourceRecord can answer the question
4853 //
4854 // If it is a question trying to validate some response, it already checked the cache for a response. If it still
4855 // reissues a question it means it could not find the RRSIGs. So, we need to bypass the cache check and send
4856 // the question out.
4857 if (QuerySuppressed(q))
4858 {
4859 AnswerSuppressedQuestion(m, q);
4860 }
4861 else if (!q->ValidatingResponse)
4862 {
4863 CacheRecord *rr;
4864 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
4865 if (SameNameRecordAnswersQuestion(&rr->resrec, q))
4866 {
4867 // SecsSinceRcvd is whole number of elapsed seconds, rounded down
4868 mDNSu32 SecsSinceRcvd = ((mDNSu32)(m->timenow - rr->TimeRcvd)) / mDNSPlatformOneSecond;
4869 if (rr->resrec.rroriginalttl <= SecsSinceRcvd)
4870 {
4871 LogMsg("AnswerNewQuestion: How is rr->resrec.rroriginalttl %lu <= SecsSinceRcvd %lu for %s %d %d",
4872 rr->resrec.rroriginalttl, SecsSinceRcvd, CRDisplayString(m, rr), m->timenow, rr->TimeRcvd);
4873 continue; // Go to next one in loop
4874 }
4875
4876 // If this record set is marked unique, then that means we can reasonably assume we have the whole set
4877 // -- we don't need to rush out on the network and query immediately to see if there are more answers out there
4878 if ((rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) || (q->ExpectUnique))
4879 ShouldQueryImmediately = mDNSfalse;
4880 q->CurrentAnswers++;
4881 if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers++;
4882 if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers++;
4883 AnsweredFromCache = mDNStrue;
4884 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
4885 if (m->CurrentQuestion != q) break; // If callback deleted q, then we're finished here
4886 }
4887 else if (mDNSOpaque16IsZero(q->TargetQID) && RRTypeIsAddressType(rr->resrec.rrtype) && RRTypeIsAddressType(q->qtype))
4888 ShouldQueryImmediately = mDNSfalse;
4889 }
4890 // We don't use LogInfo for this "Question deleted" message because it happens so routinely that
4891 // it's not remotely remarkable, and therefore unlikely to be of much help tracking down bugs.
4892 if (m->CurrentQuestion != q) { debugf("AnswerNewQuestion: Question deleted while giving cache answers"); goto exit; }
4893
4894 // Neither a local record nor a cache entry could answer this question. If this question need to be retried
4895 // with search domains, generate a negative response which will now retry after appending search domains.
4896 // If the query was suppressed above, we already generated a negative response. When it gets unsuppressed,
4897 // we will retry with search domains.
4898 if (!QuerySuppressed(q) && !AnsweredFromCache && q->RetryWithSearchDomains)
4899 {
4900 LogInfo("AnswerNewQuestion: Generating response for retrying with search domains %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4901 GenerateNegativeResponse(m, mDNSInterface_Any, QC_forceresponse);
4902 }
4903
4904 if (m->CurrentQuestion != q) { debugf("AnswerNewQuestion: Question deleted while giving negative answer"); goto exit; }
4905
4906 // Note: When a query gets suppressed or retried with search domains, we de-activate the question.
4907 // Hence we don't execute the following block of code for those cases.
4908 if (ShouldQueryImmediately && ActiveQuestion(q))
4909 {
4910 debugf("AnswerNewQuestion: ShouldQueryImmediately %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4911 q->ThisQInterval = InitialQuestionInterval;
4912 q->LastQTime = m->timenow - q->ThisQInterval;
4913 if (mDNSOpaque16IsZero(q->TargetQID)) // For mDNS, spread packets to avoid a burst of simultaneous queries
4914 {
4915 // Compute random delay in the range 1-6 seconds, then divide by 50 to get 20-120ms
4916 if (!m->RandomQueryDelay)
4917 m->RandomQueryDelay = (mDNSPlatformOneSecond + mDNSRandom(mDNSPlatformOneSecond*5) - 1) / 50 + 1;
4918 q->LastQTime += m->RandomQueryDelay;
4919 }
4920 }
4921
4922 // IN ALL CASES make sure that m->NextScheduledQuery is set appropriately.
4923 // In cases where m->NewQuestions->DelayAnswering is set, we may have delayed generating our
4924 // answers for this question until *after* its scheduled transmission time, in which case
4925 // m->NextScheduledQuery may now be set to 'never', and in that case -- even though we're *not* doing
4926 // ShouldQueryImmediately -- we still need to make sure we set m->NextScheduledQuery correctly.
4927 SetNextQueryTime(m,q);
4928
4929 exit:
4930 m->CurrentQuestion = mDNSNULL;
4931 m->lock_rrcache = 0;
4932 }
4933
4934 // When a NewLocalOnlyQuestion is created, AnswerNewLocalOnlyQuestion runs though our ResourceRecords delivering any
4935 // appropriate answers, stopping if it reaches a NewLocalOnlyRecord -- these will be handled by AnswerAllLocalQuestionsWithLocalAuthRecord
4936 mDNSlocal void AnswerNewLocalOnlyQuestion(mDNS *const m)
4937 {
4938 mDNSu32 slot;
4939 AuthGroup *ag;
4940 DNSQuestion *q = m->NewLocalOnlyQuestions; // Grab the question we're going to answer
4941 m->NewLocalOnlyQuestions = q->next; // Advance NewLocalOnlyQuestions to the next (if any)
4942 mDNSBool retEv = mDNSfalse;
4943
4944 debugf("AnswerNewLocalOnlyQuestion: Answering %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
4945
4946 if (m->CurrentQuestion)
4947 LogMsg("AnswerNewLocalOnlyQuestion ERROR m->CurrentQuestion already set: %##s (%s)",
4948 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
4949 m->CurrentQuestion = q; // Indicate which question we're answering, so we'll know if it gets deleted
4950
4951 if (m->CurrentRecord)
4952 LogMsg("AnswerNewLocalOnlyQuestion ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
4953
4954 // 1. First walk the LocalOnly records answering the LocalOnly question
4955 // 2. As LocalOnly questions should also be answered by any other Auth records local to the machine,
4956 // walk the ResourceRecords list delivering the answers
4957 slot = AuthHashSlot(&q->qname);
4958 ag = AuthGroupForName(&m->rrauth, slot, q->qnamehash, &q->qname);
4959 if (ag)
4960 {
4961 m->CurrentRecord = ag->members;
4962 while (m->CurrentRecord && m->CurrentRecord != ag->NewLocalOnlyRecords)
4963 {
4964 AuthRecord *rr = m->CurrentRecord;
4965 m->CurrentRecord = rr->next;
4966 if (LocalOnlyRecordAnswersQuestion(rr, q))
4967 {
4968 retEv = mDNStrue;
4969 AnswerLocalQuestionWithLocalAuthRecord(m, rr, mDNStrue);
4970 if (m->CurrentQuestion != q) break; // If callback deleted q, then we're finished here
4971 }
4972 }
4973 }
4974
4975 if (m->CurrentQuestion == q)
4976 {
4977 m->CurrentRecord = m->ResourceRecords;
4978
4979 while (m->CurrentRecord && m->CurrentRecord != m->NewLocalRecords)
4980 {
4981 AuthRecord *rr = m->CurrentRecord;
4982 m->CurrentRecord = rr->next;
4983 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
4984 {
4985 retEv = mDNStrue;
4986 AnswerLocalQuestionWithLocalAuthRecord(m, rr, mDNStrue);
4987 if (m->CurrentQuestion != q) break; // If callback deleted q, then we're finished here
4988 }
4989 }
4990 }
4991
4992 // The local host is the authoritative source for LocalOnly questions
4993 // so if no records exist and client requested intermediates, then generate a negative response
4994 if (!retEv && (m->CurrentQuestion == q) && q->ReturnIntermed)
4995 GenerateNegativeResponse(m, mDNSInterface_LocalOnly, QC_forceresponse);
4996
4997 m->CurrentQuestion = mDNSNULL;
4998 m->CurrentRecord = mDNSNULL;
4999 }
5000
5001 mDNSlocal CacheEntity *GetCacheEntity(mDNS *const m, const CacheGroup *const PreserveCG)
5002 {
5003 CacheEntity *e = mDNSNULL;
5004
5005 if (m->lock_rrcache) { LogMsg("GetFreeCacheRR ERROR! Cache already locked!"); return(mDNSNULL); }
5006 m->lock_rrcache = 1;
5007
5008 // If we have no free records, ask the client layer to give us some more memory
5009 if (!m->rrcache_free && m->MainCallback)
5010 {
5011 if (m->rrcache_totalused != m->rrcache_size)
5012 LogMsg("GetFreeCacheRR: count mismatch: m->rrcache_totalused %lu != m->rrcache_size %lu",
5013 m->rrcache_totalused, m->rrcache_size);
5014
5015 // We don't want to be vulnerable to a malicious attacker flooding us with an infinite
5016 // number of bogus records so that we keep growing our cache until the machine runs out of memory.
5017 // To guard against this, if our cache grows above 512kB (approx 3168 records at 164 bytes each),
5018 // and we're actively using less than 1/32 of that cache, then we purge all the unused records
5019 // and recycle them, instead of allocating more memory.
5020 if (m->rrcache_size > 5000 && m->rrcache_size / 32 > m->rrcache_active)
5021 LogInfo("Possible denial-of-service attack in progress: m->rrcache_size %lu; m->rrcache_active %lu",
5022 m->rrcache_size, m->rrcache_active);
5023 else
5024 {
5025 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
5026 m->MainCallback(m, mStatus_GrowCache);
5027 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
5028 }
5029 }
5030
5031 // If we still have no free records, recycle all the records we can.
5032 // Enumerating the entire cache is moderately expensive, so when we do it, we reclaim all the records we can in one pass.
5033 if (!m->rrcache_free)
5034 {
5035 mDNSu32 oldtotalused = m->rrcache_totalused;
5036 mDNSu32 slot;
5037 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
5038 {
5039 CacheGroup **cp = &m->rrcache_hash[slot];
5040 while (*cp)
5041 {
5042 CacheRecord **rp = &(*cp)->members;
5043 while (*rp)
5044 {
5045 // Records that answer still-active questions are not candidates for recycling
5046 // Records that are currently linked into the CacheFlushRecords list may not be recycled, or we'll crash
5047 if ((*rp)->CRActiveQuestion || (*rp)->NextInCFList)
5048 rp=&(*rp)->next;
5049 else
5050 {
5051 CacheRecord *rr = *rp;
5052 *rp = (*rp)->next; // Cut record from list
5053 ReleaseCacheRecord(m, rr);
5054 }
5055 }
5056 if ((*cp)->rrcache_tail != rp)
5057 verbosedebugf("GetFreeCacheRR: Updating rrcache_tail[%lu] from %p to %p", slot, (*cp)->rrcache_tail, rp);
5058 (*cp)->rrcache_tail = rp;
5059 if ((*cp)->members || (*cp)==PreserveCG) cp=&(*cp)->next;
5060 else ReleaseCacheGroup(m, cp);
5061 }
5062 }
5063 LogInfo("GetCacheEntity recycled %d records to reduce cache from %d to %d",
5064 oldtotalused - m->rrcache_totalused, oldtotalused, m->rrcache_totalused);
5065 }
5066
5067 if (m->rrcache_free) // If there are records in the free list, take one
5068 {
5069 e = m->rrcache_free;
5070 m->rrcache_free = e->next;
5071 if (++m->rrcache_totalused >= m->rrcache_report)
5072 {
5073 LogInfo("RR Cache now using %ld objects", m->rrcache_totalused);
5074 if (m->rrcache_report < 100) m->rrcache_report += 10;
5075 else if (m->rrcache_report < 1000) m->rrcache_report += 100;
5076 else m->rrcache_report += 1000;
5077 }
5078 mDNSPlatformMemZero(e, sizeof(*e));
5079 }
5080
5081 m->lock_rrcache = 0;
5082
5083 return(e);
5084 }
5085
5086 mDNSlocal CacheRecord *GetCacheRecord(mDNS *const m, CacheGroup *cg, mDNSu16 RDLength)
5087 {
5088 CacheRecord *r = (CacheRecord *)GetCacheEntity(m, cg);
5089 if (r)
5090 {
5091 r->resrec.rdata = (RData*)&r->smallrdatastorage; // By default, assume we're usually going to be using local storage
5092 if (RDLength > InlineCacheRDSize) // If RDLength is too big, allocate extra storage
5093 {
5094 r->resrec.rdata = (RData*)mDNSPlatformMemAllocate(sizeofRDataHeader + RDLength);
5095 if (r->resrec.rdata) r->resrec.rdata->MaxRDLength = r->resrec.rdlength = RDLength;
5096 else { ReleaseCacheEntity(m, (CacheEntity*)r); r = mDNSNULL; }
5097 }
5098 }
5099 return(r);
5100 }
5101
5102 mDNSlocal CacheGroup *GetCacheGroup(mDNS *const m, const mDNSu32 slot, const ResourceRecord *const rr)
5103 {
5104 mDNSu16 namelen = DomainNameLength(rr->name);
5105 CacheGroup *cg = (CacheGroup*)GetCacheEntity(m, mDNSNULL);
5106 if (!cg) { LogMsg("GetCacheGroup: Failed to allocate memory for %##s", rr->name->c); return(mDNSNULL); }
5107 cg->next = m->rrcache_hash[slot];
5108 cg->namehash = rr->namehash;
5109 cg->members = mDNSNULL;
5110 cg->rrcache_tail = &cg->members;
5111 if (namelen > sizeof(cg->namestorage))
5112 cg->name = mDNSPlatformMemAllocate(namelen);
5113 else
5114 cg->name = (domainname*)cg->namestorage;
5115 if (!cg->name)
5116 {
5117 LogMsg("GetCacheGroup: Failed to allocate name storage for %##s", rr->name->c);
5118 ReleaseCacheEntity(m, (CacheEntity*)cg);
5119 return(mDNSNULL);
5120 }
5121 AssignDomainName(cg->name, rr->name);
5122
5123 if (CacheGroupForRecord(m, slot, rr)) LogMsg("GetCacheGroup: Already have CacheGroup for %##s", rr->name->c);
5124 m->rrcache_hash[slot] = cg;
5125 if (CacheGroupForRecord(m, slot, rr) != cg) LogMsg("GetCacheGroup: Not finding CacheGroup for %##s", rr->name->c);
5126
5127 return(cg);
5128 }
5129
5130 mDNSexport void mDNS_PurgeCacheResourceRecord(mDNS *const m, CacheRecord *rr)
5131 {
5132 mDNS_CheckLock(m);
5133
5134 // Make sure we mark this record as thoroughly expired -- we don't ever want to give
5135 // a positive answer using an expired record (e.g. from an interface that has gone away).
5136 // We don't want to clear CRActiveQuestion here, because that would leave the record subject to
5137 // summary deletion without giving the proper callback to any questions that are monitoring it.
5138 // By setting UnansweredQueries to MaxUnansweredQueries we ensure it won't trigger any further expiration queries.
5139 rr->TimeRcvd = m->timenow - mDNSPlatformOneSecond * 60;
5140 rr->UnansweredQueries = MaxUnansweredQueries;
5141 rr->resrec.rroriginalttl = 0;
5142 SetNextCacheCheckTimeForRecord(m, rr);
5143 }
5144
5145 mDNSexport mDNSs32 mDNS_TimeNow(const mDNS *const m)
5146 {
5147 mDNSs32 time;
5148 mDNSPlatformLock(m);
5149 if (m->mDNS_busy)
5150 {
5151 LogMsg("mDNS_TimeNow called while holding mDNS lock. This is incorrect. Code protected by lock should just use m->timenow.");
5152 if (!m->timenow) LogMsg("mDNS_TimeNow: m->mDNS_busy is %ld but m->timenow not set", m->mDNS_busy);
5153 }
5154
5155 if (m->timenow) time = m->timenow;
5156 else time = mDNS_TimeNow_NoLock(m);
5157 mDNSPlatformUnlock(m);
5158 return(time);
5159 }
5160
5161 // To avoid pointless CPU thrash, we use SetSPSProxyListChanged(X) to record the last interface that
5162 // had its Sleep Proxy client list change, and defer to actual BPF reconfiguration to mDNS_Execute().
5163 // (GetNextScheduledEvent() returns "now" when m->SPSProxyListChanged is set)
5164 #define SetSPSProxyListChanged(X) do { \
5165 if (m->SPSProxyListChanged && m->SPSProxyListChanged != (X)) mDNSPlatformUpdateProxyList(m, m->SPSProxyListChanged); \
5166 m->SPSProxyListChanged = (X); } while(0)
5167
5168 // Called from mDNS_Execute() to expire stale proxy records
5169 mDNSlocal void CheckProxyRecords(mDNS *const m, AuthRecord *list)
5170 {
5171 m->CurrentRecord = list;
5172 while (m->CurrentRecord)
5173 {
5174 AuthRecord *rr = m->CurrentRecord;
5175 if (rr->resrec.RecordType != kDNSRecordTypeDeregistering && rr->WakeUp.HMAC.l[0])
5176 {
5177 // If m->SPSSocket is NULL that means we're not acting as a sleep proxy any more,
5178 // so we need to cease proxying for *all* records we may have, expired or not.
5179 if (m->SPSSocket && m->timenow - rr->TimeExpire < 0) // If proxy record not expired yet, update m->NextScheduledSPS
5180 {
5181 if (m->NextScheduledSPS - rr->TimeExpire > 0)
5182 m->NextScheduledSPS = rr->TimeExpire;
5183 }
5184 else // else proxy record expired, so remove it
5185 {
5186 LogSPS("CheckProxyRecords: Removing %d H-MAC %.6a I-MAC %.6a %d %s",
5187 m->ProxyRecords, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, ARDisplayString(m, rr));
5188 SetSPSProxyListChanged(rr->resrec.InterfaceID);
5189 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
5190 // Don't touch rr after this -- memory may have been free'd
5191 }
5192 }
5193 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
5194 // new records could have been added to the end of the list as a result of that call.
5195 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
5196 m->CurrentRecord = rr->next;
5197 }
5198 }
5199
5200 mDNSlocal void CheckRmvEventsForLocalRecords(mDNS *const m)
5201 {
5202 while (m->CurrentRecord)
5203 {
5204 AuthRecord *rr = m->CurrentRecord;
5205 if (rr->AnsweredLocalQ && rr->resrec.RecordType == kDNSRecordTypeDeregistering)
5206 {
5207 debugf("CheckRmvEventsForLocalRecords: Generating local RMV events for %s", ARDisplayString(m, rr));
5208 rr->resrec.RecordType = kDNSRecordTypeShared;
5209 AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse);
5210 if (m->CurrentRecord == rr) // If rr still exists in list, restore its state now
5211 {
5212 rr->resrec.RecordType = kDNSRecordTypeDeregistering;
5213 rr->AnsweredLocalQ = mDNSfalse;
5214 // SendResponses normally calls CompleteDeregistration after sending goodbyes.
5215 // For LocalOnly records, we don't do that and hence we need to do that here.
5216 if (RRLocalOnly(rr)) CompleteDeregistration(m, rr);
5217 }
5218 }
5219 if (m->CurrentRecord == rr) // If m->CurrentRecord was not auto-advanced, do it ourselves now
5220 m->CurrentRecord = rr->next;
5221 }
5222 }
5223
5224 mDNSlocal void TimeoutQuestions(mDNS *const m)
5225 {
5226 m->NextScheduledStopTime = m->timenow + 0x3FFFFFFF;
5227 if (m->CurrentQuestion)
5228 LogMsg("TimeoutQuestions ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c,
5229 DNSTypeName(m->CurrentQuestion->qtype));
5230 m->CurrentQuestion = m->Questions;
5231 while (m->CurrentQuestion)
5232 {
5233 DNSQuestion *const q = m->CurrentQuestion;
5234 if (q->StopTime)
5235 {
5236 if (!q->TimeoutQuestion)
5237 LogMsg("TimeoutQuestions: ERROR!! TimeoutQuestion not set, but StopTime set for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
5238
5239 if (m->timenow - q->StopTime >= 0)
5240 {
5241 LogInfo("TimeoutQuestions: question %p %##s timed out, time %d", q, q->qname.c, m->timenow - q->StopTime);
5242 GenerateNegativeResponse(m, mDNSInterface_Any, QC_forceresponse);
5243 if (m->CurrentQuestion == q) q->StopTime = 0;
5244 }
5245 else
5246 {
5247 if (m->NextScheduledStopTime - q->StopTime > 0)
5248 m->NextScheduledStopTime = q->StopTime;
5249 }
5250 }
5251 // If m->CurrentQuestion wasn't modified out from under us, advance it now
5252 // We can't do this at the start of the loop because GenerateNegativeResponse
5253 // depends on having m->CurrentQuestion point to the right question
5254 if (m->CurrentQuestion == q)
5255 m->CurrentQuestion = q->next;
5256 }
5257 m->CurrentQuestion = mDNSNULL;
5258 }
5259
5260 mDNSlocal void mDNSCoreFreeProxyRR(mDNS *const m)
5261 {
5262 AuthRecord *rrPtr = m->SPSRRSet, *rrNext = mDNSNULL;
5263 LogSPS("%s : Freeing stored sleep proxy A/AAAA records", __func__);
5264 while (rrPtr)
5265 {
5266 rrNext = rrPtr->next;
5267 mDNSPlatformMemFree(rrPtr);
5268 rrPtr = rrNext;
5269 }
5270 m->SPSRRSet = mDNSNULL;
5271 }
5272
5273 mDNSexport mDNSs32 mDNS_Execute(mDNS *const m)
5274 {
5275 mDNS_Lock(m); // Must grab lock before trying to read m->timenow
5276
5277 #if APPLE_OSX_mDNSResponder
5278 mDNSLogStatistics(m);
5279 #endif // APPLE_OSX_mDNSResponder
5280
5281 if (m->timenow - m->NextScheduledEvent >= 0)
5282 {
5283 int i;
5284 AuthRecord *head, *tail;
5285 mDNSu32 slot;
5286 AuthGroup *ag;
5287
5288 verbosedebugf("mDNS_Execute");
5289
5290 if (m->CurrentQuestion)
5291 LogMsg("mDNS_Execute: ERROR m->CurrentQuestion already set: %##s (%s)",
5292 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
5293
5294 if (m->CurrentRecord)
5295 LogMsg("mDNS_Execute: ERROR m->CurrentRecord already set: %s", ARDisplayString(m, m->CurrentRecord));
5296
5297 // 1. If we're past the probe suppression time, we can clear it
5298 if (m->SuppressProbes && m->timenow - m->SuppressProbes >= 0) m->SuppressProbes = 0;
5299
5300 // 2. If it's been more than ten seconds since the last probe failure, we can clear the counter
5301 if (m->NumFailedProbes && m->timenow - m->ProbeFailTime >= mDNSPlatformOneSecond * 10) m->NumFailedProbes = 0;
5302
5303 // 3. Purge our cache of stale old records
5304 if (m->rrcache_size && m->timenow - m->NextCacheCheck >= 0)
5305 {
5306 mDNSu32 numchecked = 0;
5307 m->NextCacheCheck = m->timenow + 0x3FFFFFFF;
5308 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
5309 {
5310 if (m->timenow - m->rrcache_nextcheck[slot] >= 0)
5311 {
5312 CacheGroup **cp = &m->rrcache_hash[slot];
5313 m->rrcache_nextcheck[slot] = m->timenow + 0x3FFFFFFF;
5314 while (*cp)
5315 {
5316 debugf("m->NextCacheCheck %4d Slot %3d %##s", numchecked, slot, *cp ? (*cp)->name : (domainname*)"\x04NULL");
5317 numchecked++;
5318 CheckCacheExpiration(m, slot, *cp);
5319 if ((*cp)->members) cp=&(*cp)->next;
5320 else ReleaseCacheGroup(m, cp);
5321 }
5322 }
5323 // Even if we didn't need to actually check this slot yet, still need to
5324 // factor its nextcheck time into our overall NextCacheCheck value
5325 if (m->NextCacheCheck - m->rrcache_nextcheck[slot] > 0)
5326 m->NextCacheCheck = m->rrcache_nextcheck[slot];
5327 }
5328 debugf("m->NextCacheCheck %4d checked, next in %d", numchecked, m->NextCacheCheck - m->timenow);
5329 }
5330
5331 if (m->timenow - m->NextScheduledSPS >= 0)
5332 {
5333 m->NextScheduledSPS = m->timenow + 0x3FFFFFFF;
5334 CheckProxyRecords(m, m->DuplicateRecords); // Clear m->DuplicateRecords first, then m->ResourceRecords
5335 CheckProxyRecords(m, m->ResourceRecords);
5336 }
5337
5338 SetSPSProxyListChanged(mDNSNULL); // Perform any deferred BPF reconfiguration now
5339
5340 // Check to see if we need to send any keepalives. Do this after we called CheckProxyRecords above
5341 // as records could have expired during that check
5342 if (m->timenow - m->NextScheduledKA >= 0)
5343 {
5344 m->NextScheduledKA = m->timenow + 0x3FFFFFFF;
5345 mDNS_SendKeepalives(m);
5346 }
5347
5348 #if BONJOUR_ON_DEMAND
5349 if (m->NextBonjourDisableTime && (m->timenow - m->NextBonjourDisableTime >= 0))
5350 {
5351 // Schedule immediate network change processing to leave the multicast group
5352 // since the delay time has expired since the previous active registration or query.
5353 m->NetworkChanged = m->timenow;
5354 m->NextBonjourDisableTime = 0;
5355 m->BonjourEnabled = 0;
5356
5357 LogInfo("mDNS_Execute: Scheduled network changed processing to leave multicast group.");
5358 }
5359 #endif // BONJOUR_ON_DEMAND
5360
5361 // Clear AnnounceOwner if necessary. (Do this *before* SendQueries() and SendResponses().)
5362 if (m->AnnounceOwner && m->timenow - m->AnnounceOwner >= 0)
5363 {
5364 m->AnnounceOwner = 0;
5365
5366 // This is a good time to reset the delay counter used to prevent spurious conflicts
5367 m->DelayConflictProcessing = 0;
5368 }
5369
5370 if (m->DelaySleep && m->timenow - m->DelaySleep >= 0)
5371 {
5372 m->DelaySleep = 0;
5373 if (m->SleepState == SleepState_Transferring)
5374 {
5375 LogSPS("Re-sleep delay passed; now checking for Sleep Proxy Servers");
5376 BeginSleepProcessing(m);
5377 }
5378 }
5379
5380 // 4. See if we can answer any of our new local questions from the cache
5381 for (i=0; m->NewQuestions && i<1000; i++)
5382 {
5383 if (m->NewQuestions->DelayAnswering && m->timenow - m->NewQuestions->DelayAnswering < 0) break;
5384 AnswerNewQuestion(m);
5385 }
5386 if (i >= 1000) LogMsg("mDNS_Execute: AnswerNewQuestion exceeded loop limit");
5387
5388 // Make sure we deliver *all* local RMV events, and clear the corresponding rr->AnsweredLocalQ flags, *before*
5389 // we begin generating *any* new ADD events in the m->NewLocalOnlyQuestions and m->NewLocalRecords loops below.
5390 for (i=0; i<1000 && m->LocalRemoveEvents; i++)
5391 {
5392 m->LocalRemoveEvents = mDNSfalse;
5393 m->CurrentRecord = m->ResourceRecords;
5394 CheckRmvEventsForLocalRecords(m);
5395 // Walk the LocalOnly records and deliver the RMV events
5396 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
5397 for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next)
5398 {
5399 m->CurrentRecord = ag->members;
5400 if (m->CurrentRecord) CheckRmvEventsForLocalRecords(m);
5401 }
5402 }
5403
5404 if (i >= 1000) LogMsg("mDNS_Execute: m->LocalRemoveEvents exceeded loop limit");
5405
5406 for (i=0; m->NewLocalOnlyQuestions && i<1000; i++) AnswerNewLocalOnlyQuestion(m);
5407 if (i >= 1000) LogMsg("mDNS_Execute: AnswerNewLocalOnlyQuestion exceeded loop limit");
5408
5409 head = tail = mDNSNULL;
5410 for (i=0; i<1000 && m->NewLocalRecords && m->NewLocalRecords != head; i++)
5411 {
5412 AuthRecord *rr = m->NewLocalRecords;
5413 m->NewLocalRecords = m->NewLocalRecords->next;
5414 if (LocalRecordReady(rr))
5415 {
5416 debugf("mDNS_Execute: Delivering Add event with LocalAuthRecord %s", ARDisplayString(m, rr));
5417 AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNStrue);
5418 }
5419 else if (!rr->next)
5420 {
5421 // If we have just one record that is not ready, we don't have to unlink and
5422 // reinsert. As the NewLocalRecords will be NULL for this case, the loop will
5423 // terminate and set the NewLocalRecords to rr.
5424 debugf("mDNS_Execute: Just one LocalAuthRecord %s, breaking out of the loop early", ARDisplayString(m, rr));
5425 if (head != mDNSNULL || m->NewLocalRecords != mDNSNULL)
5426 LogMsg("mDNS_Execute: ERROR!!: head %p, NewLocalRecords %p", head, m->NewLocalRecords);
5427
5428 head = rr;
5429 }
5430 else
5431 {
5432 AuthRecord **p = &m->ResourceRecords; // Find this record in our list of active records
5433 debugf("mDNS_Execute: Skipping LocalAuthRecord %s", ARDisplayString(m, rr));
5434 // if this is the first record we are skipping, move to the end of the list.
5435 // if we have already skipped records before, append it at the end.
5436 while (*p && *p != rr) p=&(*p)->next;
5437 if (*p) *p = rr->next; // Cut this record from the list
5438 else { LogMsg("mDNS_Execute: ERROR!! Cannot find record %s in ResourceRecords list", ARDisplayString(m, rr)); break; }
5439 if (!head)
5440 {
5441 while (*p) p=&(*p)->next;
5442 *p = rr;
5443 head = tail = rr;
5444 }
5445 else
5446 {
5447 tail->next = rr;
5448 tail = rr;
5449 }
5450 rr->next = mDNSNULL;
5451 }
5452 }
5453 m->NewLocalRecords = head;
5454 debugf("mDNS_Execute: Setting NewLocalRecords to %s", (head ? ARDisplayString(m, head) : "NULL"));
5455
5456 if (i >= 1000) LogMsg("mDNS_Execute: m->NewLocalRecords exceeded loop limit");
5457
5458 // Check to see if we have any new LocalOnly/P2P records to examine for delivering
5459 // to our local questions
5460 if (m->NewLocalOnlyRecords)
5461 {
5462 m->NewLocalOnlyRecords = mDNSfalse;
5463 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
5464 for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next)
5465 {
5466 for (i=0; i<100 && ag->NewLocalOnlyRecords; i++)
5467 {
5468 AuthRecord *rr = ag->NewLocalOnlyRecords;
5469 ag->NewLocalOnlyRecords = ag->NewLocalOnlyRecords->next;
5470 // LocalOnly records should always be ready as they never probe
5471 if (LocalRecordReady(rr))
5472 {
5473 debugf("mDNS_Execute: Delivering Add event with LocalAuthRecord %s", ARDisplayString(m, rr));
5474 AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNStrue);
5475 }
5476 else LogMsg("mDNS_Execute: LocalOnlyRecord %s not ready", ARDisplayString(m, rr));
5477 }
5478 // We limit about 100 per AuthGroup that can be serviced at a time
5479 if (i >= 100) LogMsg("mDNS_Execute: ag->NewLocalOnlyRecords exceeded loop limit");
5480 }
5481 }
5482
5483 // 5. See what packets we need to send
5484 if (m->mDNSPlatformStatus != mStatus_NoError || (m->SleepState == SleepState_Sleeping))
5485 DiscardDeregistrations(m);
5486 if (m->mDNSPlatformStatus == mStatus_NoError && (m->SuppressSending == 0 || m->timenow - m->SuppressSending >= 0))
5487 {
5488 // If the platform code is ready, and we're not suppressing packet generation right now
5489 // then send our responses, probes, and questions.
5490 // We check the cache first, because there might be records close to expiring that trigger questions to refresh them.
5491 // We send queries next, because there might be final-stage probes that complete their probing here, causing
5492 // them to advance to announcing state, and we want those to be included in any announcements we send out.
5493 // Finally, we send responses, including the previously mentioned records that just completed probing.
5494 m->SuppressSending = 0;
5495
5496 // 6. Send Query packets. This may cause some probing records to advance to announcing state
5497 if (m->timenow - m->NextScheduledQuery >= 0 || m->timenow - m->NextScheduledProbe >= 0) SendQueries(m);
5498 if (m->timenow - m->NextScheduledQuery >= 0)
5499 {
5500 DNSQuestion *q;
5501 LogMsg("mDNS_Execute: SendQueries didn't send all its queries (%d - %d = %d) will try again in one second",
5502 m->timenow, m->NextScheduledQuery, m->timenow - m->NextScheduledQuery);
5503 m->NextScheduledQuery = m->timenow + mDNSPlatformOneSecond;
5504 for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
5505 if (ActiveQuestion(q) && m->timenow - NextQSendTime(q) >= 0)
5506 LogMsg("mDNS_Execute: SendQueries didn't send %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
5507 }
5508 if (m->timenow - m->NextScheduledProbe >= 0)
5509 {
5510 LogMsg("mDNS_Execute: SendQueries didn't send all its probes (%d - %d = %d) will try again in one second",
5511 m->timenow, m->NextScheduledProbe, m->timenow - m->NextScheduledProbe);
5512 m->NextScheduledProbe = m->timenow + mDNSPlatformOneSecond;
5513 }
5514
5515 // 7. Send Response packets, including probing records just advanced to announcing state
5516 if (m->timenow - m->NextScheduledResponse >= 0) SendResponses(m);
5517 if (m->timenow - m->NextScheduledResponse >= 0)
5518 {
5519 LogMsg("mDNS_Execute: SendResponses didn't send all its responses; will try again in one second");
5520 m->NextScheduledResponse = m->timenow + mDNSPlatformOneSecond;
5521 }
5522 }
5523
5524 // Clear RandomDelay values, ready to pick a new different value next time
5525 m->RandomQueryDelay = 0;
5526 m->RandomReconfirmDelay = 0;
5527
5528 if (m->NextScheduledStopTime && m->timenow - m->NextScheduledStopTime >= 0) TimeoutQuestions(m);
5529 #ifndef UNICAST_DISABLED
5530 if (m->NextSRVUpdate && m->timenow - m->NextSRVUpdate >= 0) UpdateAllSRVRecords(m);
5531 if (m->timenow - m->NextScheduledNATOp >= 0) CheckNATMappings(m);
5532 if (m->timenow - m->NextuDNSEvent >= 0) uDNS_Tasks(m);
5533 #endif
5534 #if APPLE_OSX_mDNSResponder
5535 extern void serviceBLE();
5536 if (m->NextBLEServiceTime && (m->timenow - m->NextBLEServiceTime >= 0)) serviceBLE();
5537 #endif // APPLE_OSX_mDNSResponder
5538 }
5539
5540 // Note about multi-threaded systems:
5541 // On a multi-threaded system, some other thread could run right after the mDNS_Unlock(),
5542 // performing mDNS API operations that change our next scheduled event time.
5543 //
5544 // On multi-threaded systems (like the current Windows implementation) that have a single main thread
5545 // calling mDNS_Execute() (and other threads allowed to call mDNS API routines) it is the responsibility
5546 // of the mDNSPlatformUnlock() routine to signal some kind of stateful condition variable that will
5547 // signal whatever blocking primitive the main thread is using, so that it will wake up and execute one
5548 // more iteration of its loop, and immediately call mDNS_Execute() again. The signal has to be stateful
5549 // in the sense that if the main thread has not yet entered its blocking primitive, then as soon as it
5550 // does, the state of the signal will be noticed, causing the blocking primitive to return immediately
5551 // without blocking. This avoids the race condition between the signal from the other thread arriving
5552 // just *before* or just *after* the main thread enters the blocking primitive.
5553 //
5554 // On multi-threaded systems (like the current Mac OS 9 implementation) that are entirely timer-driven,
5555 // with no main mDNS_Execute() thread, it is the responsibility of the mDNSPlatformUnlock() routine to
5556 // set the timer according to the m->NextScheduledEvent value, and then when the timer fires, the timer
5557 // callback function should call mDNS_Execute() (and ignore the return value, which may already be stale
5558 // by the time it gets to the timer callback function).
5559
5560 mDNS_Unlock(m); // Calling mDNS_Unlock is what gives m->NextScheduledEvent its new value
5561 return(m->NextScheduledEvent);
5562 }
5563
5564 #ifndef UNICAST_DISABLED
5565 mDNSlocal void SuspendLLQs(mDNS *m)
5566 {
5567 DNSQuestion *q;
5568 for (q = m->Questions; q; q = q->next)
5569 if (ActiveQuestion(q) && !mDNSOpaque16IsZero(q->TargetQID) && q->LongLived && q->state == LLQ_Established)
5570 { q->ReqLease = 0; sendLLQRefresh(m, q); }
5571 }
5572 #endif // UNICAST_DISABLED
5573
5574 mDNSlocal mDNSBool QuestionHasLocalAnswers(mDNS *const m, DNSQuestion *q)
5575 {
5576 AuthRecord *rr;
5577 mDNSu32 slot;
5578 AuthGroup *ag;
5579
5580 slot = AuthHashSlot(&q->qname);
5581 ag = AuthGroupForName(&m->rrauth, slot, q->qnamehash, &q->qname);
5582 if (ag)
5583 {
5584 for (rr = ag->members; rr; rr=rr->next)
5585 // Filter the /etc/hosts records - LocalOnly, Unique, A/AAAA/CNAME
5586 if (UniqueLocalOnlyRecord(rr) && LocalOnlyRecordAnswersQuestion(rr, q))
5587 {
5588 LogInfo("QuestionHasLocalAnswers: Question %p %##s (%s) has local answer %s", q, q->qname.c, DNSTypeName(q->qtype), ARDisplayString(m, rr));
5589 return mDNStrue;
5590 }
5591 }
5592 return mDNSfalse;
5593 }
5594
5595 // ActivateUnicastQuery() is called from three places:
5596 // 1. When a new question is created
5597 // 2. On wake from sleep
5598 // 3. When the DNS configuration changes
5599 // In case 1 we don't want to mess with our established ThisQInterval and LastQTime (ScheduleImmediately is false)
5600 // In cases 2 and 3 we do want to cause the question to be resent immediately (ScheduleImmediately is true)
5601 mDNSlocal void ActivateUnicastQuery(mDNS *const m, DNSQuestion *const question, mDNSBool ScheduleImmediately)
5602 {
5603 // For now this AutoTunnel stuff is specific to Mac OS X.
5604 // In the future, if there's demand, we may see if we can abstract it out cleanly into the platform layer
5605 #if APPLE_OSX_mDNSResponder
5606 // Even though BTMM client tunnels are only useful for AAAA queries, we need to treat v4 and v6 queries equally.
5607 // Otherwise we can get the situation where the A query completes really fast (with an NXDOMAIN result) and the
5608 // caller then gives up waiting for the AAAA result while we're still in the process of setting up the tunnel.
5609 // To level the playing field, we block both A and AAAA queries while tunnel setup is in progress, and then
5610 // returns results for both at the same time. If we are looking for the _autotunnel6 record, then skip this logic
5611 // as this would trigger looking up _autotunnel6._autotunnel6 and end up failing the original query.
5612
5613 if (RRTypeIsAddressType(question->qtype) && PrivateQuery(question) &&
5614 !SameDomainLabel(question->qname.c, (const mDNSu8 *)"\x0c_autotunnel6")&& question->QuestionCallback != AutoTunnelCallback)
5615 {
5616 question->NoAnswer = NoAnswer_Suspended;
5617 AddNewClientTunnel(m, question);
5618 return;
5619 }
5620 #endif // APPLE_OSX_mDNSResponder
5621
5622 if (!question->DuplicateOf)
5623 {
5624 debugf("ActivateUnicastQuery: %##s %s%s%s",
5625 question->qname.c, DNSTypeName(question->qtype), PrivateQuery(question) ? " (Private)" : "", ScheduleImmediately ? " ScheduleImmediately" : "");
5626 question->CNAMEReferrals = 0;
5627 if (question->nta) { CancelGetZoneData(m, question->nta); question->nta = mDNSNULL; }
5628 if (question->LongLived)
5629 {
5630 question->state = LLQ_InitialRequest;
5631 question->id = zeroOpaque64;
5632 question->servPort = zeroIPPort;
5633 if (question->tcp) { DisposeTCPConn(question->tcp); question->tcp = mDNSNULL; }
5634 }
5635 // If the question has local answers, then we don't want answers from outside
5636 if (ScheduleImmediately && !QuestionHasLocalAnswers(m, question))
5637 {
5638 question->ThisQInterval = InitialQuestionInterval;
5639 question->LastQTime = m->timenow - question->ThisQInterval;
5640 SetNextQueryTime(m, question);
5641 }
5642 }
5643 }
5644
5645 // Caller should hold the lock
5646 mDNSexport void mDNSCoreRestartAddressQueries(mDNS *const m, mDNSBool SearchDomainsChanged, FlushCache flushCacheRecords,
5647 CallbackBeforeStartQuery BeforeStartCallback, void *context)
5648 {
5649 DNSQuestion *q;
5650 DNSQuestion *restart = mDNSNULL;
5651
5652 mDNS_CheckLock(m);
5653
5654 // 1. Flush the cache records
5655 if (flushCacheRecords) flushCacheRecords(m);
5656
5657 // 2. Even though we may have purged the cache records above, before it can generate RMV event
5658 // we are going to stop the question. Hence we need to deliver the RMV event before we
5659 // stop the question.
5660 //
5661 // CurrentQuestion is used by RmvEventsForQuestion below. While delivering RMV events, the
5662 // application callback can potentially stop the current question (detected by CurrentQuestion) or
5663 // *any* other question which could be the next one that we may process here. RestartQuestion
5664 // points to the "next" question which will be automatically advanced in mDNS_StopQuery_internal
5665 // if the "next" question is stopped while the CurrentQuestion is stopped
5666
5667 if (m->RestartQuestion)
5668 LogMsg("mDNSCoreRestartAddressQueries: ERROR!! m->RestartQuestion already set: %##s (%s)",
5669 m->RestartQuestion->qname.c, DNSTypeName(m->RestartQuestion->qtype));
5670
5671 m->RestartQuestion = m->Questions;
5672 while (m->RestartQuestion)
5673 {
5674 q = m->RestartQuestion;
5675 m->RestartQuestion = q->next;
5676 // GetZoneData questions are referenced by other questions (original query that started the GetZoneData
5677 // question) through their "nta" pointer. Normally when the original query stops, it stops the
5678 // GetZoneData question and also frees the memory (See CancelGetZoneData). If we stop the GetZoneData
5679 // question followed by the original query that refers to this GetZoneData question, we will end up
5680 // freeing the GetZoneData question and then start the "freed" question at the end.
5681
5682 if (IsGetZoneDataQuestion(q))
5683 {
5684 DNSQuestion *refq = q->next;
5685 LogInfo("mDNSCoreRestartAddressQueries: Skipping GetZoneDataQuestion %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
5686 // debug stuff, we just try to find the referencing question and don't do much with it
5687 while (refq)
5688 {
5689 if (q == &refq->nta->question)
5690 {
5691 LogInfo("mDNSCoreRestartAddressQueries: Question %p %##s (%s) referring to GetZoneDataQuestion %p, not stopping", refq, refq->qname.c, DNSTypeName(refq->qtype), q);
5692 }
5693 refq = refq->next;
5694 }
5695 continue;
5696 }
5697
5698 // This function is called when /etc/hosts changes and that could affect A, AAAA and CNAME queries
5699 if (q->qtype != kDNSType_A && q->qtype != kDNSType_AAAA && q->qtype != kDNSType_CNAME) continue;
5700
5701 // If the search domains did not change, then we restart all the queries. Otherwise, only
5702 // for queries for which we "might" have appended search domains ("might" because we may
5703 // find results before we apply search domains even though AppendSearchDomains is set to 1)
5704 if (!SearchDomainsChanged || q->AppendSearchDomains)
5705 {
5706 // NOTE: CacheRecordRmvEventsForQuestion will not generate RMV events for queries that have non-zero
5707 // LOAddressAnswers. Hence it is important that we call CacheRecordRmvEventsForQuestion before
5708 // LocalRecordRmvEventsForQuestion (which decrements LOAddressAnswers). Let us say that
5709 // /etc/hosts has an A Record for web.apple.com. Any queries for web.apple.com will be answered locally.
5710 // But this can't prevent a CNAME/AAAA query to not to be sent on the wire. When it is sent on the wire,
5711 // it could create cache entries. When we are restarting queries, we can't deliver the cache RMV events
5712 // for the original query using these cache entries as ADDs were never delivered using these cache
5713 // entries and hence this order is needed.
5714
5715 // If the query is suppressed, the RMV events won't be delivered
5716 if (!CacheRecordRmvEventsForQuestion(m, q)) { LogInfo("mDNSCoreRestartAddressQueries: Question deleted while delivering Cache Record RMV events"); continue; }
5717
5718 // SuppressQuery status does not affect questions that are answered using local records
5719 if (!LocalRecordRmvEventsForQuestion(m, q)) { LogInfo("mDNSCoreRestartAddressQueries: Question deleted while delivering Local Record RMV events"); continue; }
5720
5721 LogInfo("mDNSCoreRestartAddressQueries: Stop question %p %##s (%s), AppendSearchDomains %d, qnameOrig %p", q,
5722 q->qname.c, DNSTypeName(q->qtype), q->AppendSearchDomains, q->qnameOrig);
5723 mDNS_StopQuery_internal(m, q);
5724 // Reset state so that it looks like it was in the beginning i.e it should look at /etc/hosts, cache
5725 // and then search domains should be appended. At the beginning, qnameOrig was NULL.
5726 if (q->qnameOrig)
5727 {
5728 LogInfo("mDNSCoreRestartAddressQueries: qnameOrig %##s", q->qnameOrig);
5729 AssignDomainName(&q->qname, q->qnameOrig);
5730 mDNSPlatformMemFree(q->qnameOrig);
5731 q->qnameOrig = mDNSNULL;
5732 q->RetryWithSearchDomains = ApplySearchDomainsFirst(q) ? 1 : 0;
5733 }
5734 q->SearchListIndex = 0;
5735 q->next = restart;
5736 restart = q;
5737 }
5738 }
5739
5740 // 3. Callback before we start the query
5741 if (BeforeStartCallback) BeforeStartCallback(m, context);
5742
5743 // 4. Restart all the stopped queries
5744 while (restart)
5745 {
5746 q = restart;
5747 restart = restart->next;
5748 q->next = mDNSNULL;
5749 LogInfo("mDNSCoreRestartAddressQueries: Start question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
5750 mDNS_StartQuery_internal(m, q);
5751 }
5752 }
5753
5754 mDNSexport void mDNSCoreRestartQueries(mDNS *const m)
5755 {
5756 DNSQuestion *q;
5757
5758 #ifndef UNICAST_DISABLED
5759 // Retrigger all our uDNS questions
5760 if (m->CurrentQuestion)
5761 LogMsg("mDNSCoreRestartQueries: ERROR m->CurrentQuestion already set: %##s (%s)",
5762 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
5763 m->CurrentQuestion = m->Questions;
5764 while (m->CurrentQuestion)
5765 {
5766 q = m->CurrentQuestion;
5767 m->CurrentQuestion = m->CurrentQuestion->next;
5768 if (!mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q)) ActivateUnicastQuery(m, q, mDNStrue);
5769 }
5770 #endif
5771
5772 // Retrigger all our mDNS questions
5773 for (q = m->Questions; q; q=q->next) // Scan our list of questions
5774 mDNSCoreRestartQuestion(m, q);
5775 }
5776
5777 // restart question if it's multicast and currently active
5778 mDNSexport void mDNSCoreRestartQuestion(mDNS *const m, DNSQuestion *q)
5779 {
5780 if (mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q))
5781 {
5782 q->ThisQInterval = InitialQuestionInterval; // MUST be > zero for an active question
5783 q->RequestUnicast = kDefaultRequestUnicastCount;
5784 q->LastQTime = m->timenow - q->ThisQInterval;
5785 q->RecentAnswerPkts = 0;
5786 ExpireDupSuppressInfo(q->DupSuppress, m->timenow);
5787 m->NextScheduledQuery = m->timenow;
5788 }
5789 }
5790
5791 // restart the probe/announce cycle for multicast record
5792 mDNSexport void mDNSCoreRestartRegistration(mDNS *const m, AuthRecord *rr, int announceCount)
5793 {
5794 if (!AuthRecord_uDNS(rr))
5795 {
5796 if (rr->resrec.RecordType == kDNSRecordTypeVerified && !rr->DependentOn) rr->resrec.RecordType = kDNSRecordTypeUnique;
5797 rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
5798
5799 if (mDNS_KeepaliveRecord(&rr->resrec))
5800 {
5801 rr->AnnounceCount = 0; // Do not announce keepalive records
5802 }
5803 else
5804 {
5805 // announceCount < 0 indicates default announce count should be used
5806 if (announceCount < 0)
5807 announceCount = InitialAnnounceCount;
5808 if (rr->AnnounceCount < (mDNSu8)announceCount)
5809 rr->AnnounceCount = (mDNSu8)announceCount;
5810 }
5811
5812 rr->SendNSECNow = mDNSNULL;
5813 InitializeLastAPTime(m, rr);
5814 }
5815 }
5816
5817 // ***************************************************************************
5818 #if COMPILER_LIKES_PRAGMA_MARK
5819 #pragma mark -
5820 #pragma mark - Power Management (Sleep/Wake)
5821 #endif
5822
5823 mDNSexport void mDNS_UpdateAllowSleep(mDNS *const m)
5824 {
5825 #ifndef IDLESLEEPCONTROL_DISABLED
5826 mDNSBool allowSleep = mDNStrue;
5827 char reason[128];
5828
5829 reason[0] = 0;
5830
5831 if (m->SystemSleepOnlyIfWakeOnLAN)
5832 {
5833 // Don't sleep if we are a proxy for any services
5834 if (m->ProxyRecords)
5835 {
5836 allowSleep = mDNSfalse;
5837 mDNS_snprintf(reason, sizeof(reason), "sleep proxy for %d records", m->ProxyRecords);
5838 LogInfo("mDNS_UpdateAllowSleep: Sleep disabled because we are proxying %d records", m->ProxyRecords);
5839 }
5840
5841 if (allowSleep && mDNSCoreHaveAdvertisedMulticastServices(m))
5842 {
5843 // Scan the list of active interfaces
5844 NetworkInterfaceInfo *intf;
5845 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
5846 {
5847 if (intf->McastTxRx && !intf->Loopback && !mDNSPlatformInterfaceIsD2D(intf->InterfaceID))
5848 {
5849 // Disallow sleep if this interface doesn't support NetWake
5850 if (!intf->NetWake)
5851 {
5852 allowSleep = mDNSfalse;
5853 mDNS_snprintf(reason, sizeof(reason), "%s does not support NetWake", intf->ifname);
5854 LogInfo("mDNS_UpdateAllowSleep: Sleep disabled because %s does not support NetWake", intf->ifname);
5855 break;
5856 }
5857
5858 // If the interface can be an in-NIC Proxy, we should check if it can accomodate all the records
5859 // that will be offloaded. If not, we should prevent sleep.
5860 // This check will be possible once the lower layers provide an API to query the space available for offloads on the NIC.
5861 #if APPLE_OSX_mDNSResponder
5862 if (!SupportsInNICProxy(intf))
5863 #endif
5864 {
5865 // Disallow sleep if there is no sleep proxy server
5866 const CacheRecord *cr = FindSPSInCache1(m, &intf->NetWakeBrowse, mDNSNULL, mDNSNULL);
5867 if ( cr == mDNSNULL)
5868 {
5869 allowSleep = mDNSfalse;
5870 mDNS_snprintf(reason, sizeof(reason), "No sleep proxy server on %s", intf->ifname);
5871 LogInfo("mDNS_UpdateAllowSleep: Sleep disabled because %s has no sleep proxy server", intf->ifname);
5872 break;
5873 }
5874 else if (m->SPSType != 0)
5875 {
5876 mDNSu32 mymetric = LocalSPSMetric(m);
5877 mDNSu32 metric = SPSMetric(cr->resrec.rdata->u.name.c);
5878 if (metric >= mymetric)
5879 {
5880 allowSleep = mDNSfalse;
5881 mDNS_snprintf(reason, sizeof(reason), "No sleep proxy server with better metric on %s", intf->ifname);
5882 LogInfo("mDNS_UpdateAllowSleep: Sleep disabled because %s has no sleep proxy server with a better metric", intf->ifname);
5883 break;
5884 }
5885 }
5886 }
5887 }
5888 }
5889 }
5890 }
5891
5892 // Call the platform code to enable/disable sleep
5893 mDNSPlatformSetAllowSleep(m, allowSleep, reason);
5894 #else
5895 (void) m;
5896 #endif /* !defined(IDLESLEEPCONTROL_DISABLED) */
5897 }
5898
5899 mDNSlocal mDNSBool mDNSUpdateOkToSend(mDNS *const m, AuthRecord *rr, NetworkInterfaceInfo *const intf, mDNSu32 scopeid)
5900 {
5901 // If it is not a uDNS record, check to see if the updateid is zero. "updateid" is cleared when we have
5902 // sent the resource record on all the interfaces. If the update id is not zero, check to see if it is time
5903 // to send.
5904 if (AuthRecord_uDNS(rr) || (rr->AuthFlags & AuthFlagsWakeOnly) || mDNSOpaque16IsZero(rr->updateid) ||
5905 m->timenow - (rr->LastAPTime + rr->ThisAPInterval) < 0)
5906 {
5907 return mDNSfalse;
5908 }
5909
5910 // If we have a pending registration for "scopeid", it is ok to send the update on that interface.
5911 // If the scopeid is too big to check for validity, we don't check against updateIntID. When
5912 // we successfully update on all the interfaces (with whatever set in "rr->updateIntID"), we clear
5913 // updateid and we should have returned from above.
5914 //
5915 // Note: scopeid is the same as intf->InterfaceID. It is passed in so that we don't have to call the
5916 // platform function to extract the value from "intf" every time.
5917
5918 if ((scopeid >= (sizeof(rr->updateIntID) * mDNSNBBY) || bit_get_opaque64(rr->updateIntID, scopeid)) &&
5919 (!rr->resrec.InterfaceID || rr->resrec.InterfaceID == intf->InterfaceID))
5920 return mDNStrue;
5921
5922 return mDNSfalse;
5923 }
5924
5925 mDNSexport void UpdateRMAC(mDNS *const m, void *context)
5926 {
5927 IPAddressMACMapping *addrmap = (IPAddressMACMapping *)context ;
5928 m->CurrentRecord = m->ResourceRecords;
5929
5930 if (!addrmap)
5931 {
5932 LogMsg("UpdateRMAC: Address mapping is NULL");
5933 return;
5934 }
5935
5936 while (m->CurrentRecord)
5937 {
5938 AuthRecord *rr = m->CurrentRecord;
5939 // If this is a non-sleep proxy keepalive record and the remote IP address matches, update the RData
5940 if (!rr->WakeUp.HMAC.l[0] && mDNS_KeepaliveRecord(&rr->resrec))
5941 {
5942 mDNSAddr raddr;
5943 getKeepaliveRaddr(m, rr, &raddr);
5944 if (mDNSSameAddress(&raddr, &addrmap->ipaddr))
5945 {
5946 // Update the MAC address only if it is not a zero MAC address
5947 mDNSEthAddr macAddr;
5948 mDNSu8 *ptr = GetValueForMACAddr((mDNSu8 *)(addrmap->ethaddr), (mDNSu8 *) (addrmap->ethaddr + sizeof(addrmap->ethaddr)), &macAddr);
5949 if (ptr != mDNSNULL && !mDNSEthAddressIsZero(macAddr))
5950 {
5951 UpdateKeepaliveRData(m, rr, mDNSNULL, mDNStrue, (char *)(addrmap->ethaddr));
5952 }
5953 }
5954 }
5955 m->CurrentRecord = rr->next;
5956 }
5957
5958 if (addrmap)
5959 mDNSPlatformMemFree(addrmap);
5960
5961 }
5962
5963 mDNSexport mStatus UpdateKeepaliveRData(mDNS *const m, AuthRecord *rr, NetworkInterfaceInfo *const intf, mDNSBool updateMac, char *ethAddr)
5964 {
5965 mDNSu16 newrdlength;
5966 mDNSAddr laddr = zeroAddr;
5967 mDNSAddr raddr = zeroAddr;
5968 mDNSEthAddr eth = zeroEthAddr;
5969 mDNSIPPort lport = zeroIPPort;
5970 mDNSIPPort rport = zeroIPPort;
5971 mDNSu32 timeout = 0;
5972 mDNSu32 seq = 0;
5973 mDNSu32 ack = 0;
5974 mDNSu16 win = 0;
5975 UTF8str255 txt;
5976 int rdsize;
5977 RData *newrd;
5978 mDNSTCPInfo mti;
5979 mStatus ret;
5980
5981 // Note: If we fail to update the DNS NULL record with additional information in this function, it will be registered
5982 // with the SPS like any other record. SPS will not send keepalives if it does not have additional information.
5983 mDNS_ExtractKeepaliveInfo(rr, &timeout, &laddr, &raddr, &eth, &seq, &ack, &lport, &rport, &win);
5984 if (!timeout || mDNSAddressIsZero(&laddr) || mDNSAddressIsZero(&raddr) || mDNSIPPortIsZero(lport) || mDNSIPPortIsZero(rport))
5985 {
5986 LogMsg("UpdateKeepaliveRData: not a valid record %s for keepalive %#a:%d %#a:%d", ARDisplayString(m, rr), &laddr, lport.NotAnInteger, &raddr, rport.NotAnInteger);
5987 return mStatus_UnknownErr;
5988 }
5989
5990 if (updateMac)
5991 {
5992 if (laddr.type == mDNSAddrType_IPv4)
5993 newrdlength = mDNS_snprintf((char *)&txt.c[1], sizeof(txt.c) - 1, "t=%d i=%d c=%d h=%#a d=%#a l=%u r=%u m=%s", timeout, kKeepaliveRetryInterval, kKeepaliveRetryCount, &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport), ethAddr);
5994 else
5995 newrdlength = mDNS_snprintf((char *)&txt.c[1], sizeof(txt.c) - 1, "t=%d i=%d c=%d H=%#a D=%#a l=%u r=%u m=%s", timeout, kKeepaliveRetryInterval, kKeepaliveRetryCount, &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport), ethAddr);
5996
5997 }
5998 else
5999 {
6000 // If this keepalive packet would be sent on a different interface than the current one that we are processing
6001 // now, then we don't update the DNS NULL record. But we do not prevent it from registering with the SPS. When SPS sees
6002 // this DNS NULL record, it does not send any keepalives as it does not have all the information
6003 mDNSPlatformMemZero(&mti, sizeof (mDNSTCPInfo));
6004 ret = mDNSPlatformRetrieveTCPInfo(m, &laddr, &lport, &raddr, &rport, &mti);
6005 if (ret != mStatus_NoError)
6006 {
6007 LogMsg("mDNSPlatformRetrieveTCPInfo: mDNSPlatformRetrieveTCPInfo failed %d", ret);
6008 return ret;
6009 }
6010 if ((intf != mDNSNULL) && (mti.IntfId != intf->InterfaceID))
6011 {
6012 LogInfo("mDNSPlatformRetrieveTCPInfo: InterfaceID mismatch mti.IntfId = %p InterfaceID = %p", mti.IntfId, intf->InterfaceID);
6013 return mStatus_BadParamErr;
6014 }
6015
6016 if (laddr.type == mDNSAddrType_IPv4)
6017 newrdlength = mDNS_snprintf((char *)&txt.c[1], sizeof(txt.c) - 1, "t=%d i=%d c=%d h=%#a d=%#a l=%u r=%u m=%.6a s=%u a=%u w=%u", timeout, kKeepaliveRetryInterval, kKeepaliveRetryCount, &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport), &eth, mti.seq, mti.ack, mti.window);
6018 else
6019 newrdlength = mDNS_snprintf((char *)&txt.c[1], sizeof(txt.c) - 1, "t=%d i=%d c=%d H=%#a D=%#a l=%u r=%u m=%.6a s=%u a=%u w=%u", timeout, kKeepaliveRetryInterval, kKeepaliveRetryCount, &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport), &eth, mti.seq, mti.ack, mti.window);
6020 }
6021
6022 // Did we insert a null byte at the end ?
6023 if (newrdlength == (sizeof(txt.c) - 1))
6024 {
6025 LogMsg("UpdateKeepaliveRData: could not allocate memory %s", ARDisplayString(m, rr));
6026 return mStatus_NoMemoryErr;
6027 }
6028
6029 // Include the length for the null byte at the end
6030 txt.c[0] = newrdlength + 1;
6031 // Account for the first length byte and the null byte at the end
6032 newrdlength += 2;
6033
6034 rdsize = newrdlength > sizeof(RDataBody) ? newrdlength : sizeof(RDataBody);
6035 newrd = mDNSPlatformMemAllocate(sizeof(RData) - sizeof(RDataBody) + rdsize);
6036 if (!newrd) { LogMsg("UpdateKeepaliveRData: ptr NULL"); return mStatus_NoMemoryErr; }
6037
6038 newrd->MaxRDLength = (mDNSu16) rdsize;
6039 mDNSPlatformMemCopy(&newrd->u, txt.c, newrdlength);
6040
6041 // If we are updating the record for the first time, rdata points to rdatastorage as the rdata memory
6042 // was allocated as part of the AuthRecord itself. We allocate memory when we update the AuthRecord.
6043 // If the resource record has data that we allocated in a previous pass (to update MAC address),
6044 // free that memory here before copying in the new data.
6045 if ( rr->resrec.rdata != &rr->rdatastorage)
6046 {
6047 LogSPS("UpdateKeepaliveRData: Freed allocated memory for keep alive packet: %s ", ARDisplayString(m, rr));
6048 mDNSPlatformMemFree(rr->resrec.rdata);
6049 }
6050 SetNewRData(&rr->resrec, newrd, newrdlength); // Update our rdata
6051
6052 LogSPS("UpdateKeepaliveRData: successfully updated the record %s", ARDisplayString(m, rr));
6053 return mStatus_NoError;
6054 }
6055
6056 mDNSlocal void SendSPSRegistrationForOwner(mDNS *const m, NetworkInterfaceInfo *const intf, const mDNSOpaque16 id, const OwnerOptData *const owner)
6057 {
6058 const int optspace = DNSOpt_Header_Space + DNSOpt_LeaseData_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC);
6059 const int sps = intf->NextSPSAttempt / 3;
6060 AuthRecord *rr;
6061 mDNSOpaque16 msgid;
6062 mDNSu32 scopeid;
6063
6064 scopeid = mDNSPlatformInterfaceIndexfromInterfaceID(m, intf->InterfaceID, mDNStrue);
6065 if (!intf->SPSAddr[sps].type)
6066 {
6067 intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond;
6068 if (m->NextScheduledSPRetry - intf->NextSPSAttemptTime > 0)
6069 m->NextScheduledSPRetry = intf->NextSPSAttemptTime;
6070 LogSPS("SendSPSRegistration: %s SPS %d (%d) %##s not yet resolved", intf->ifname, intf->NextSPSAttempt, sps, intf->NetWakeResolve[sps].qname.c);
6071 goto exit;
6072 }
6073
6074 // Mark our mDNS records (not unicast records) for transfer to SPS
6075 if (mDNSOpaque16IsZero(id))
6076 {
6077 // We may have to register this record over multiple interfaces and we don't want to
6078 // overwrite the id. We send the registration over interface X with id "IDX" and before
6079 // we get a response, we overwrite with id "IDY" for interface Y and we won't accept responses
6080 // for "IDX". Hence, we want to use the same ID across all interfaces.
6081 //
6082 // In the case of sleep proxy server transfering its records when it goes to sleep, the owner
6083 // option check below will set the same ID across the records from the same owner. Records
6084 // with different owner option gets different ID.
6085 msgid = mDNS_NewMessageID(m);
6086 for (rr = m->ResourceRecords; rr; rr=rr->next)
6087 {
6088 if (!(rr->AuthFlags & AuthFlagsWakeOnly) && rr->resrec.RecordType > kDNSRecordTypeDeregistering)
6089 {
6090 if (rr->resrec.InterfaceID == intf->InterfaceID || (!rr->resrec.InterfaceID && (rr->ForceMCast || IsLocalDomain(rr->resrec.name))))
6091 {
6092 if (mDNSPlatformMemSame(owner, &rr->WakeUp, sizeof(*owner)))
6093 {
6094 rr->SendRNow = mDNSInterfaceMark; // mark it now
6095 // When we are registering on the first interface, rr->updateid is zero in which case
6096 // initialize with the new ID. For subsequent interfaces, we want to use the same ID.
6097 // At the end, all the updates sent across all the interfaces with the same ID.
6098 if (mDNSOpaque16IsZero(rr->updateid))
6099 rr->updateid = msgid;
6100 else
6101 msgid = rr->updateid;
6102 }
6103 }
6104 }
6105 }
6106 }
6107 else
6108 msgid = id;
6109
6110 while (1)
6111 {
6112 mDNSu8 *p = m->omsg.data;
6113 // To comply with RFC 2782, PutResourceRecord suppresses name compression for SRV records in unicast updates.
6114 // For now we follow that same logic for SPS registrations too.
6115 // If we decide to compress SRV records in SPS registrations in the future, we can achieve that by creating our
6116 // initial DNSMessage with h.flags set to zero, and then update it to UpdateReqFlags right before sending the packet.
6117 InitializeDNSMessage(&m->omsg.h, msgid, UpdateReqFlags);
6118
6119 for (rr = m->ResourceRecords; rr; rr=rr->next)
6120 if (rr->SendRNow || mDNSUpdateOkToSend(m, rr, intf, scopeid))
6121 {
6122 if (mDNSPlatformMemSame(owner, &rr->WakeUp, sizeof(*owner)))
6123 {
6124 mDNSu8 *newptr;
6125 const mDNSu8 *const limit = m->omsg.data + (m->omsg.h.mDNS_numUpdates ? NormalMaxDNSMessageData : AbsoluteMaxDNSMessageData) - optspace;
6126
6127 // If we can't update the keepalive record, don't send it
6128 if (mDNS_KeepaliveRecord(&rr->resrec) && (UpdateKeepaliveRData(m, rr, intf, mDNSfalse, mDNSNULL) != mStatus_NoError))
6129 {
6130 if (scopeid < (sizeof(rr->updateIntID) * mDNSNBBY))
6131 {
6132 bit_clr_opaque64(rr->updateIntID, scopeid);
6133 }
6134 rr->SendRNow = mDNSNULL;
6135 continue;
6136 }
6137
6138 if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
6139 rr->resrec.rrclass |= kDNSClass_UniqueRRSet; // Temporarily set the 'unique' bit so PutResourceRecord will set it
6140 newptr = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.mDNS_numUpdates, &rr->resrec, rr->resrec.rroriginalttl, limit);
6141 rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; // Make sure to clear 'unique' bit back to normal state
6142 if (!newptr)
6143 LogSPS("SendSPSRegistration put %s FAILED %d/%d %s", intf->ifname, p - m->omsg.data, limit - m->omsg.data, ARDisplayString(m, rr));
6144 else
6145 {
6146 LogSPS("SendSPSRegistration put %s 0x%x 0x%x (updateid %d) %s", intf->ifname, rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(m->omsg.h.id), ARDisplayString(m, rr));
6147 rr->SendRNow = mDNSNULL;
6148 rr->ThisAPInterval = mDNSPlatformOneSecond;
6149 rr->LastAPTime = m->timenow;
6150 // should be initialized above
6151 if (mDNSOpaque16IsZero(rr->updateid)) LogMsg("SendSPSRegistration: ERROR!! rr %s updateid is zero", ARDisplayString(m, rr));
6152 if (m->NextScheduledResponse - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
6153 m->NextScheduledResponse = (rr->LastAPTime + rr->ThisAPInterval);
6154 p = newptr;
6155 }
6156 }
6157 }
6158
6159 if (!m->omsg.h.mDNS_numUpdates) break;
6160 else
6161 {
6162 AuthRecord opt;
6163 mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
6164 opt.resrec.rrclass = NormalMaxDNSMessageData;
6165 opt.resrec.rdlength = sizeof(rdataOPT) * 2; // Two options in this OPT record
6166 opt.resrec.rdestimate = sizeof(rdataOPT) * 2;
6167 opt.resrec.rdata->u.opt[0].opt = kDNSOpt_Lease;
6168 opt.resrec.rdata->u.opt[0].optlen = DNSOpt_LeaseData_Space - 4;
6169 opt.resrec.rdata->u.opt[0].u.updatelease = DEFAULT_UPDATE_LEASE;
6170 if (!owner->HMAC.l[0]) // If no owner data,
6171 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[1]); // use our own interface information
6172 else // otherwise, use the owner data we were given
6173 {
6174 opt.resrec.rdata->u.opt[1].u.owner = *owner;
6175 opt.resrec.rdata->u.opt[1].opt = kDNSOpt_Owner;
6176 opt.resrec.rdata->u.opt[1].optlen = DNSOpt_Owner_Space(&owner->HMAC, &owner->IMAC) - 4;
6177 }
6178 LogSPS("SendSPSRegistration put %s %s", intf->ifname, ARDisplayString(m, &opt));
6179 p = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.numAdditionals, &opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
6180 if (!p)
6181 LogMsg("SendSPSRegistration: Failed to put OPT record (%d updates) %s", m->omsg.h.mDNS_numUpdates, ARDisplayString(m, &opt));
6182 else
6183 {
6184 mStatus err;
6185
6186 LogSPS("SendSPSRegistration: Sending Update %s %d (%d) id %5d with %d records %d bytes to %#a:%d", intf->ifname, intf->NextSPSAttempt, sps,
6187 mDNSVal16(m->omsg.h.id), m->omsg.h.mDNS_numUpdates, p - m->omsg.data, &intf->SPSAddr[sps], mDNSVal16(intf->SPSPort[sps]));
6188 // if (intf->NextSPSAttempt < 5) m->omsg.h.flags = zeroID; // For simulating packet loss
6189 err = mDNSSendDNSMessage(m, &m->omsg, p, intf->InterfaceID, mDNSNULL, &intf->SPSAddr[sps], intf->SPSPort[sps], mDNSNULL, mDNSNULL, mDNSfalse);
6190 if (err) LogSPS("SendSPSRegistration: mDNSSendDNSMessage err %d", err);
6191 if (err && intf->SPSAddr[sps].type == mDNSAddrType_IPv4 && intf->NetWakeResolve[sps].ThisQInterval == -1)
6192 {
6193 LogSPS("SendSPSRegistration %d %##s failed to send to IPv4 address; will try IPv6 instead", sps, intf->NetWakeResolve[sps].qname.c);
6194 intf->NetWakeResolve[sps].qtype = kDNSType_AAAA;
6195 mDNS_StartQuery_internal(m, &intf->NetWakeResolve[sps]);
6196 return;
6197 }
6198 }
6199 }
6200 }
6201
6202 intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond * 10; // If successful, update NextSPSAttemptTime
6203
6204 exit:
6205 if (mDNSOpaque16IsZero(id) && intf->NextSPSAttempt < 8) intf->NextSPSAttempt++;
6206 }
6207
6208 mDNSlocal mDNSBool RecordIsFirstOccurrenceOfOwner(mDNS *const m, const AuthRecord *const rr)
6209 {
6210 AuthRecord *ar;
6211 for (ar = m->ResourceRecords; ar && ar != rr; ar=ar->next)
6212 if (mDNSPlatformMemSame(&rr->WakeUp, &ar->WakeUp, sizeof(rr->WakeUp))) return mDNSfalse;
6213 return mDNStrue;
6214 }
6215
6216 mDNSlocal void mDNSCoreStoreProxyRR(mDNS *const m, const mDNSInterfaceID InterfaceID, AuthRecord *const rr)
6217 {
6218 AuthRecord *newRR = mDNSPlatformMemAllocate(sizeof(AuthRecord));
6219
6220 if (newRR == mDNSNULL)
6221 {
6222 LogSPS("%s : could not allocate memory for new resource record", __func__);
6223 return;
6224 }
6225
6226 mDNSPlatformMemZero(newRR, sizeof(AuthRecord));
6227 mDNS_SetupResourceRecord(newRR, mDNSNULL, InterfaceID, rr->resrec.rrtype,
6228 rr->resrec.rroriginalttl, rr->resrec.RecordType,
6229 rr->ARType, mDNSNULL, mDNSNULL);
6230
6231 AssignDomainName(&newRR->namestorage, &rr->namestorage);
6232 newRR->resrec.rdlength = DomainNameLength(rr->resrec.name);
6233 newRR->resrec.namehash = DomainNameHashValue(newRR->resrec.name);
6234 newRR->resrec.rrclass = rr->resrec.rrclass;
6235
6236 if (rr->resrec.rrtype == kDNSType_A)
6237 {
6238 newRR->resrec.rdata->u.ipv4 = rr->resrec.rdata->u.ipv4;
6239 }
6240 else if (rr->resrec.rrtype == kDNSType_AAAA)
6241 {
6242 newRR->resrec.rdata->u.ipv6 = rr->resrec.rdata->u.ipv6;
6243 }
6244 SetNewRData(&newRR->resrec, mDNSNULL, 0);
6245
6246 // Insert the new node at the head of the list.
6247 newRR->next = m->SPSRRSet;
6248 m->SPSRRSet = newRR;
6249 LogSPS("%s : Storing proxy record : %s ", __func__, ARDisplayString(m, rr));
6250 }
6251
6252 // Some records are interface specific and some are not. The ones that are supposed to be registered
6253 // on multiple interfaces need to be initialized with all the valid interfaces on which it will be sent.
6254 // updateIntID bit field tells us on which interfaces we need to register this record. When we get an
6255 // ack from the sleep proxy server, we clear the interface bit. This way, we know when a record completes
6256 // registration on all the interfaces
6257 mDNSlocal void SPSInitRecordsBeforeUpdate(mDNS *const m, mDNSOpaque64 updateIntID, mDNSBool *WakeOnlyService)
6258 {
6259 AuthRecord *ar;
6260 LogSPS("SPSInitRecordsBeforeUpdate: UpdateIntID 0x%x 0x%x", updateIntID.l[1], updateIntID.l[0]);
6261
6262 *WakeOnlyService = mDNSfalse;
6263
6264 // Before we store the A and AAAA records that we are going to register with the sleep proxy,
6265 // make sure that the old sleep proxy records are removed.
6266 mDNSCoreFreeProxyRR(m);
6267
6268 // For records that are registered only on a specific interface, mark only that bit as it will
6269 // never be registered on any other interface. For others, it should be sent on all interfaces.
6270 for (ar = m->ResourceRecords; ar; ar=ar->next)
6271 {
6272 ar->updateIntID = zeroOpaque64;
6273 ar->updateid = zeroID;
6274 if (AuthRecord_uDNS(ar))
6275 {
6276 continue;
6277 }
6278 if (ar->AuthFlags & AuthFlagsWakeOnly)
6279 {
6280 if (ar->resrec.RecordType == kDNSRecordTypeShared && ar->RequireGoodbye)
6281 {
6282 ar->ImmedAnswer = mDNSInterfaceMark;
6283 *WakeOnlyService = mDNStrue;
6284 continue;
6285 }
6286 }
6287 if (!ar->resrec.InterfaceID)
6288 {
6289 LogSPS("Setting scopeid (ALL) 0x%x 0x%x for %s", updateIntID.l[1], updateIntID.l[0], ARDisplayString(m, ar));
6290 ar->updateIntID = updateIntID;
6291 }
6292 else
6293 {
6294 // Filter records that belong to interfaces that we won't register the records on. UpdateIntID captures
6295 // exactly this.
6296 mDNSu32 scopeid = mDNSPlatformInterfaceIndexfromInterfaceID(m, ar->resrec.InterfaceID, mDNStrue);
6297 if ((scopeid < (sizeof(updateIntID) * mDNSNBBY)) && bit_get_opaque64(updateIntID, scopeid))
6298 {
6299 bit_set_opaque64(ar->updateIntID, scopeid);
6300 LogSPS("SPSInitRecordsBeforeUpdate: Setting scopeid(%d) 0x%x 0x%x for %s", scopeid, ar->updateIntID.l[1],
6301 ar->updateIntID.l[0], ARDisplayString(m, ar));
6302 }
6303 else
6304 {
6305 LogSPS("SPSInitRecordsBeforeUpdate: scopeid %d beyond range or not valid for SPS registration", scopeid);
6306 }
6307 }
6308 // Store the A and AAAA records that we registered with the sleep proxy.
6309 // We will use this to prevent spurious name conflicts that may occur when we wake up
6310 if (ar->resrec.rrtype == kDNSType_A || ar->resrec.rrtype == kDNSType_AAAA)
6311 {
6312 mDNSCoreStoreProxyRR(m, ar->resrec.InterfaceID, ar);
6313 }
6314 }
6315 }
6316
6317 mDNSlocal void SendSPSRegistration(mDNS *const m, NetworkInterfaceInfo *const intf, const mDNSOpaque16 id)
6318 {
6319 AuthRecord *ar;
6320 OwnerOptData owner = zeroOwner;
6321
6322 SendSPSRegistrationForOwner(m, intf, id, &owner);
6323
6324 for (ar = m->ResourceRecords; ar; ar=ar->next)
6325 {
6326 if (!mDNSPlatformMemSame(&owner, &ar->WakeUp, sizeof(owner)) && RecordIsFirstOccurrenceOfOwner(m, ar))
6327 {
6328 owner = ar->WakeUp;
6329 SendSPSRegistrationForOwner(m, intf, id, &owner);
6330 }
6331 }
6332 }
6333
6334 // RetrySPSRegistrations is called from SendResponses, with the lock held
6335 mDNSlocal void RetrySPSRegistrations(mDNS *const m)
6336 {
6337 AuthRecord *rr;
6338 NetworkInterfaceInfo *intf;
6339
6340 // First make sure none of our interfaces' NextSPSAttemptTimes are inadvertently set to m->timenow + mDNSPlatformOneSecond * 10
6341 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6342 if (intf->NextSPSAttempt && intf->NextSPSAttemptTime == m->timenow + mDNSPlatformOneSecond * 10)
6343 intf->NextSPSAttemptTime++;
6344
6345 // Retry any record registrations that are due
6346 for (rr = m->ResourceRecords; rr; rr=rr->next)
6347 if (!AuthRecord_uDNS(rr) && !mDNSOpaque16IsZero(rr->updateid) && m->timenow - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
6348 {
6349 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6350 {
6351 // If we still have registrations pending on this interface, send it now
6352 mDNSu32 scopeid = mDNSPlatformInterfaceIndexfromInterfaceID(m, intf->InterfaceID, mDNStrue);
6353 if ((scopeid >= (sizeof(rr->updateIntID) * mDNSNBBY) || bit_get_opaque64(rr->updateIntID, scopeid)) &&
6354 (!rr->resrec.InterfaceID || rr->resrec.InterfaceID == intf->InterfaceID))
6355 {
6356 LogSPS("RetrySPSRegistrations: 0x%x 0x%x (updateid %d) %s", rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(rr->updateid), ARDisplayString(m, rr));
6357 SendSPSRegistration(m, intf, rr->updateid);
6358 }
6359 }
6360 }
6361
6362 // For interfaces where we did an SPS registration attempt, increment intf->NextSPSAttempt
6363 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6364 if (intf->NextSPSAttempt && intf->NextSPSAttemptTime == m->timenow + mDNSPlatformOneSecond * 10 && intf->NextSPSAttempt < 8)
6365 intf->NextSPSAttempt++;
6366 }
6367
6368 mDNSlocal void NetWakeResolve(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
6369 {
6370 NetworkInterfaceInfo *intf = (NetworkInterfaceInfo *)question->QuestionContext;
6371 int sps = (int)(question - intf->NetWakeResolve);
6372 (void)m; // Unused
6373 LogSPS("NetWakeResolve: SPS: %d Add: %d %s", sps, AddRecord, RRDisplayString(m, answer));
6374
6375 if (!AddRecord) return; // Don't care about REMOVE events
6376 if (answer->rrtype != question->qtype) return; // Don't care about CNAMEs
6377
6378 // if (answer->rrtype == kDNSType_AAAA && sps == 0) return; // To test failing to resolve sleep proxy's address
6379
6380 if (answer->rrtype == kDNSType_SRV)
6381 {
6382 // 1. Got the SRV record; now look up the target host's IP address
6383 mDNS_StopQuery(m, question);
6384 intf->SPSPort[sps] = answer->rdata->u.srv.port;
6385 AssignDomainName(&question->qname, &answer->rdata->u.srv.target);
6386 question->qtype = kDNSType_A;
6387 mDNS_StartQuery(m, question);
6388 }
6389 else if (answer->rrtype == kDNSType_A && answer->rdlength == sizeof(mDNSv4Addr))
6390 {
6391 // 2. Got an IPv4 address for the target host; record address and initiate an SPS registration if appropriate
6392 mDNS_StopQuery(m, question);
6393 question->ThisQInterval = -1;
6394 intf->SPSAddr[sps].type = mDNSAddrType_IPv4;
6395 intf->SPSAddr[sps].ip.v4 = answer->rdata->u.ipv4;
6396 mDNS_Lock(m);
6397 if (sps == intf->NextSPSAttempt/3) SendSPSRegistration(m, intf, zeroID); // If we're ready for this result, use it now
6398 mDNS_Unlock(m);
6399 }
6400 else if (answer->rrtype == kDNSType_A && answer->rdlength == 0)
6401 {
6402 // 3. Got negative response -- target host apparently has IPv6 disabled -- so try looking up the target host's IPv4 address(es) instead
6403 mDNS_StopQuery(m, question);
6404 LogSPS("NetWakeResolve: SPS %d %##s has no IPv4 address, will try IPv6 instead", sps, question->qname.c);
6405 question->qtype = kDNSType_AAAA;
6406 mDNS_StartQuery(m, question);
6407 }
6408 else if (answer->rrtype == kDNSType_AAAA && answer->rdlength == sizeof(mDNSv6Addr) && mDNSv6AddressIsLinkLocal(&answer->rdata->u.ipv6))
6409 {
6410 // 4. Got the target host's IPv6 link-local address; record address and initiate an SPS registration if appropriate
6411 mDNS_StopQuery(m, question);
6412 question->ThisQInterval = -1;
6413 intf->SPSAddr[sps].type = mDNSAddrType_IPv6;
6414 intf->SPSAddr[sps].ip.v6 = answer->rdata->u.ipv6;
6415 mDNS_Lock(m);
6416 if (sps == intf->NextSPSAttempt/3) SendSPSRegistration(m, intf, zeroID); // If we're ready for this result, use it now
6417 mDNS_Unlock(m);
6418 }
6419 }
6420
6421 mDNSexport mDNSBool mDNSCoreHaveAdvertisedMulticastServices(mDNS *const m)
6422 {
6423 AuthRecord *rr;
6424 for (rr = m->ResourceRecords; rr; rr=rr->next)
6425 if (mDNS_KeepaliveRecord(&rr->resrec) || (rr->resrec.rrtype == kDNSType_SRV && !AuthRecord_uDNS(rr) && !mDNSSameIPPort(rr->resrec.rdata->u.srv.port, DiscardPort)))
6426 return mDNStrue;
6427 return mDNSfalse;
6428 }
6429
6430 #define WAKE_ONLY_SERVICE 1
6431 #define AC_ONLY_SERVICE 2
6432
6433 #ifdef APPLE_OSX_mDNSResponder
6434 mDNSlocal void SendGoodbyesForSelectServices(mDNS *const m, mDNSBool *servicePresent, mDNSu32 serviceType)
6435 {
6436 AuthRecord *rr;
6437 *servicePresent = mDNSfalse;
6438
6439 // Mark all the records we need to deregister and send them
6440 for (rr = m->ResourceRecords; rr; rr=rr->next)
6441 {
6442 // If the service type is wake only service and the auth flags match and requires a goodbye
6443 // OR if the service type is AC only and it is not a keepalive record,
6444 // mark the records we need to deregister and send them
6445 if ((serviceType == WAKE_ONLY_SERVICE && (rr->AuthFlags & AuthFlagsWakeOnly) &&
6446 rr->resrec.RecordType == kDNSRecordTypeShared && rr->RequireGoodbye) ||
6447 (serviceType == AC_ONLY_SERVICE && !mDNS_KeepaliveRecord(&rr->resrec)))
6448 {
6449 rr->ImmedAnswer = mDNSInterfaceMark;
6450 *servicePresent = mDNStrue;
6451 }
6452 }
6453 }
6454 #endif
6455
6456 #ifdef APPLE_OSX_mDNSResponder
6457 // This function is used only in the case of local NIC proxy. For external
6458 // sleep proxy server, we do this in SPSInitRecordsBeforeUpdate when we
6459 // walk the resource records.
6460 mDNSlocal void SendGoodbyesForWakeOnlyService(mDNS *const m, mDNSBool *WakeOnlyService)
6461 {
6462 return SendGoodbyesForSelectServices(m, WakeOnlyService, WAKE_ONLY_SERVICE);
6463 }
6464 #endif // APPLE_OSX_mDNSResponder
6465
6466
6467 mDNSlocal void SendSleepGoodbyes(mDNS *const m, mDNSBool AllInterfaces, mDNSBool unicast)
6468 {
6469 AuthRecord *rr;
6470 m->SleepState = SleepState_Sleeping;
6471
6472 // If AllInterfaces is not set, the caller has already marked it appropriately
6473 // on which interfaces this should be sent.
6474 if (AllInterfaces)
6475 {
6476 NetworkInterfaceInfo *intf;
6477 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6478 {
6479 intf->SendGoodbyes = 1;
6480 }
6481 }
6482 if (unicast)
6483 {
6484 #ifndef UNICAST_DISABLED
6485 SleepRecordRegistrations(m); // If we have no SPS, need to deregister our uDNS records
6486 #endif /* UNICAST_DISABLED */
6487 }
6488
6489 // Mark all the records we need to deregister and send them
6490 for (rr = m->ResourceRecords; rr; rr=rr->next)
6491 if (rr->resrec.RecordType == kDNSRecordTypeShared && rr->RequireGoodbye)
6492 rr->ImmedAnswer = mDNSInterfaceMark;
6493 SendResponses(m);
6494 }
6495
6496 /*
6497 * This function attempts to detect if multiple interfaces are on the same subnet.
6498 * It makes this determination based only on the IPv4 Addresses and subnet masks.
6499 * IPv6 link local addresses that are configured by default on all interfaces make
6500 * it hard to make this determination
6501 *
6502 * The 'real' fix for this would be to send out multicast packets over one interface
6503 * and conclude that multiple interfaces are on the same subnet only if these packets
6504 * are seen on other interfaces on the same system
6505 */
6506 mDNSlocal mDNSBool skipSameSubnetRegistration(mDNS *const m, mDNSInterfaceID *regID, mDNSu32 count, mDNSInterfaceID intfid)
6507 {
6508 NetworkInterfaceInfo *intf;
6509 NetworkInterfaceInfo *newIntf;
6510 mDNSu32 i;
6511
6512 for (newIntf = FirstInterfaceForID(m, intfid); newIntf; newIntf = newIntf->next)
6513 {
6514 if ((newIntf->InterfaceID != intfid) ||
6515 (newIntf->ip.type != mDNSAddrType_IPv4))
6516 {
6517 continue;
6518 }
6519 for ( i = 0; i < count; i++)
6520 {
6521 for (intf = FirstInterfaceForID(m, regID[i]); intf; intf = intf->next)
6522 {
6523 if ((intf->InterfaceID != regID[i]) ||
6524 (intf->ip.type != mDNSAddrType_IPv4))
6525 {
6526 continue;
6527 }
6528 if ((intf->ip.ip.v4.NotAnInteger & intf->mask.ip.v4.NotAnInteger) == (newIntf->ip.ip.v4.NotAnInteger & newIntf->mask.ip.v4.NotAnInteger))
6529 {
6530 LogSPS("%s : Already registered for the same subnet (IPv4) for interface %s", __func__, intf->ifname);
6531 return (mDNStrue);
6532 }
6533 }
6534 }
6535 }
6536 return (mDNSfalse);
6537 }
6538
6539 mDNSlocal void DoKeepaliveCallbacks(mDNS *m)
6540 {
6541 // Loop through the keepalive records and callback with an error
6542 m->CurrentRecord = m->ResourceRecords;
6543 while (m->CurrentRecord)
6544 {
6545 AuthRecord *const rr = m->CurrentRecord;
6546 if ((mDNS_KeepaliveRecord(&rr->resrec)) && (rr->resrec.RecordType != kDNSRecordTypeDeregistering))
6547 {
6548 LogSPS("DoKeepaliveCallbacks: Invoking the callback for %s", ARDisplayString(m, rr));
6549 if (rr->RecordCallback)
6550 rr->RecordCallback(m, rr, mStatus_BadStateErr);
6551 }
6552 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
6553 m->CurrentRecord = rr->next;
6554 }
6555 }
6556
6557 // BeginSleepProcessing is called, with the lock held, from either mDNS_Execute or mDNSCoreMachineSleep
6558 mDNSlocal void BeginSleepProcessing(mDNS *const m)
6559 {
6560 mDNSBool SendGoodbyes = mDNStrue;
6561 mDNSBool WakeOnlyService = mDNSfalse;
6562 mDNSBool invokeKACallback = mDNStrue;
6563 const CacheRecord *sps[3] = { mDNSNULL };
6564 mDNSOpaque64 updateIntID = zeroOpaque64;
6565 mDNSInterfaceID registeredIntfIDS[128] = { 0 };
6566 mDNSu32 registeredCount = 0;
6567 int skippedRegistrations = 0;
6568
6569 m->NextScheduledSPRetry = m->timenow;
6570
6571 if (!m->SystemWakeOnLANEnabled) LogSPS("BeginSleepProcessing: m->SystemWakeOnLANEnabled is false");
6572 else if (!mDNSCoreHaveAdvertisedMulticastServices(m)) LogSPS("BeginSleepProcessing: No advertised services");
6573 else // If we have at least one advertised service
6574 {
6575 NetworkInterfaceInfo *intf;
6576
6577 // Clear out the SCDynamic entry that stores the external SPS information
6578 mDNSPlatformClearSPSData();
6579
6580 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6581 {
6582 // Intialize it to false. These values make sense only when SleepState is set to Sleeping.
6583 intf->SendGoodbyes = 0;
6584
6585 // If it is not multicast capable, we could not have possibly discovered sleep proxy
6586 // servers.
6587 if (!intf->McastTxRx || mDNSPlatformInterfaceIsD2D(intf->InterfaceID))
6588 {
6589 LogSPS("BeginSleepProcessing: %-6s Ignoring for registrations", intf->ifname);
6590 continue;
6591 }
6592
6593 // If we are not capable of WOMP, then don't register with sleep proxy.
6594 //
6595 // Note: If we are not NetWake capable, we don't browse for the sleep proxy server.
6596 // We might find sleep proxy servers in the cache and start a resolve on them.
6597 // But then if the interface goes away, we won't stop these questions because
6598 // mDNS_DeactivateNetWake_internal assumes that a browse has been started for it
6599 // to stop both the browse and resolve questions.
6600 if (!intf->NetWake)
6601 {
6602 LogSPS("BeginSleepProcessing: %-6s not capable of magic packet wakeup", intf->ifname);
6603 intf->SendGoodbyes = 1;
6604 skippedRegistrations++;
6605 continue;
6606 }
6607
6608 // Check if we have already registered with a sleep proxy for this subnet
6609 if (skipSameSubnetRegistration(m, registeredIntfIDS, registeredCount, intf->InterfaceID))
6610 {
6611 LogSPS("%s : Skipping sleep proxy registration on %s", __func__, intf->ifname);
6612 continue;
6613 }
6614
6615 #if APPLE_OSX_mDNSResponder
6616 else if (SupportsInNICProxy(intf))
6617 {
6618 mDNSBool keepaliveOnly = mDNSfalse;
6619 if (ActivateLocalProxy(m, intf, &keepaliveOnly) == mStatus_NoError)
6620 {
6621 SendGoodbyesForWakeOnlyService(m, &WakeOnlyService);
6622
6623 // Send goodbyes for all advertised services if the only record offloaded was the keepalive record.
6624 SendGoodbyes = (keepaliveOnly) ? mDNStrue: mDNSfalse;
6625 invokeKACallback = mDNSfalse;
6626 LogSPS("BeginSleepProcessing: %-6s using local proxy", intf->ifname);
6627 // This will leave m->SleepState set to SleepState_Transferring,
6628 // which is okay because with no outstanding resolves, or updates in flight,
6629 // mDNSCoreReadyForSleep() will conclude correctly that all the updates have already completed
6630
6631 // Setting this flag activates the SleepLimit which delays sleep by 5 seconds and
6632 // will allow the system to deregister any BTMM records.
6633 m->NextScheduledSPRetry = m->timenow + (5 * mDNSPlatformOneSecond);
6634 registeredIntfIDS[registeredCount] = intf->InterfaceID;
6635 registeredCount++;
6636 }
6637 }
6638 #endif // APPLE_OSX_mDNSResponder
6639 else
6640 {
6641 #if APPLE_OSX_mDNSResponder
6642 // If on battery, do not attempt to offload to external sleep proxies
6643 if (m->SystemWakeOnLANEnabled == mDNS_WakeOnBattery)
6644 {
6645 LogSPS("BegingSleepProcessing: Not connected to AC power - Not registering with an external sleep proxy.");
6646 return;
6647 }
6648 #endif // APPLE_OSX_mDNSResponder
6649 FindSPSInCache(m, &intf->NetWakeBrowse, sps);
6650 if (!sps[0]) LogSPS("BeginSleepProcessing: %-6s %#a No Sleep Proxy Server found (Next Browse Q in %d, interval %d)",
6651 intf->ifname, &intf->ip, NextQSendTime(&intf->NetWakeBrowse) - m->timenow, intf->NetWakeBrowse.ThisQInterval);
6652 else
6653 {
6654 int i;
6655 mDNSu32 scopeid;
6656 SendGoodbyes = mDNSfalse;
6657 intf->NextSPSAttempt = 0;
6658 intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond;
6659
6660 scopeid = mDNSPlatformInterfaceIndexfromInterfaceID(m, intf->InterfaceID, mDNStrue);
6661 // Now we know for sure that we have to wait for registration to complete on this interface.
6662 if (scopeid < (sizeof(updateIntID) * mDNSNBBY))
6663 bit_set_opaque64(updateIntID, scopeid);
6664
6665 // Don't need to set m->NextScheduledSPRetry here because we already set "m->NextScheduledSPRetry = m->timenow" above
6666 for (i=0; i<3; i++)
6667 {
6668 #if ForceAlerts
6669 if (intf->SPSAddr[i].type)
6670 LogFatalError("BeginSleepProcessing: %s %d intf->SPSAddr[i].type %d", intf->ifname, i, intf->SPSAddr[i].type);
6671 if (intf->NetWakeResolve[i].ThisQInterval >= 0)
6672 LogFatalError("BeginSleepProcessing: %s %d intf->NetWakeResolve[i].ThisQInterval %d", intf->ifname, i, intf->NetWakeResolve[i].ThisQInterval);
6673 #endif
6674 intf->SPSAddr[i].type = mDNSAddrType_None;
6675 if (intf->NetWakeResolve[i].ThisQInterval >= 0) mDNS_StopQuery(m, &intf->NetWakeResolve[i]);
6676 intf->NetWakeResolve[i].ThisQInterval = -1;
6677 if (sps[i])
6678 {
6679 LogSPS("BeginSleepProcessing: %-6s Found Sleep Proxy Server %d TTL %d %s", intf->ifname, i, sps[i]->resrec.rroriginalttl, CRDisplayString(m, sps[i]));
6680 mDNS_SetupQuestion(&intf->NetWakeResolve[i], intf->InterfaceID, &sps[i]->resrec.rdata->u.name, kDNSType_SRV, NetWakeResolve, intf);
6681 intf->NetWakeResolve[i].ReturnIntermed = mDNStrue;
6682 mDNS_StartQuery_internal(m, &intf->NetWakeResolve[i]);
6683
6684 // If we are registering with a Sleep Proxy for a new subnet, add it to our list
6685 registeredIntfIDS[registeredCount] = intf->InterfaceID;
6686 registeredCount++;
6687 }
6688 }
6689 }
6690 }
6691 }
6692 }
6693
6694 // If we have at least one interface on which we are registering with an external sleep proxy,
6695 // initialize all the records appropriately.
6696 if (!mDNSOpaque64IsZero(&updateIntID))
6697 SPSInitRecordsBeforeUpdate(m, updateIntID, &WakeOnlyService);
6698
6699 // Call the applicaitons that registered a keepalive record to inform them that we failed to offload
6700 // the records to a sleep proxy.
6701 if (invokeKACallback)
6702 {
6703 LogSPS("BeginSleepProcessing: Did not register with an in-NIC proxy - invoking the callbacks for KA records");
6704 DoKeepaliveCallbacks(m);
6705 }
6706
6707 // SendSleepGoodbyes last two arguments control whether we send goodbyes on all
6708 // interfaces and also deregister unicast registrations.
6709 //
6710 // - If there are no sleep proxy servers, then send goodbyes on all interfaces
6711 // for both multicast and unicast.
6712 //
6713 // - If we skipped registrations on some interfaces, then we have already marked
6714 // them appropriately above. We don't need to send goodbyes for unicast as
6715 // we have registered with at least one sleep proxy.
6716 //
6717 // - If we are not planning to send any goodbyes, then check for WakeOnlyServices.
6718 //
6719 // Note: If we are planning to send goodbyes, we mark the record with mDNSInterfaceAny
6720 // and call SendResponses which inturn calls ShouldSendGoodbyesBeforeSleep which looks
6721 // at WakeOnlyServices first.
6722 if (SendGoodbyes)
6723 {
6724 LogSPS("BeginSleepProcessing: Not registering with Sleep Proxy Server");
6725 SendSleepGoodbyes(m, mDNStrue, mDNStrue);
6726 }
6727 else if (skippedRegistrations)
6728 {
6729 LogSPS("BeginSleepProcessing: Not registering with Sleep Proxy Server on all interfaces");
6730 SendSleepGoodbyes(m, mDNSfalse, mDNSfalse);
6731 }
6732 else if (WakeOnlyService)
6733 {
6734 // If we saw WakeOnly service above, send the goodbyes now.
6735 LogSPS("BeginSleepProcessing: Sending goodbyes for WakeOnlyService");
6736 SendResponses(m);
6737 }
6738 }
6739
6740 // Call mDNSCoreMachineSleep(m, mDNStrue) when the machine is about to go to sleep.
6741 // Call mDNSCoreMachineSleep(m, mDNSfalse) when the machine is has just woken up.
6742 // Normally, the platform support layer below mDNSCore should call this, not the client layer above.
6743 mDNSexport void mDNSCoreMachineSleep(mDNS *const m, mDNSBool sleep)
6744 {
6745 AuthRecord *rr;
6746
6747 LogSPS("%s (old state %d) at %ld", sleep ? "Sleeping" : "Waking", m->SleepState, m->timenow);
6748
6749 if (sleep && !m->SleepState) // Going to sleep
6750 {
6751 mDNS_Lock(m);
6752 // If we're going to sleep, need to stop advertising that we're a Sleep Proxy Server
6753 if (m->SPSSocket)
6754 {
6755 mDNSu8 oldstate = m->SPSState;
6756 mDNS_DropLockBeforeCallback(); // mDNS_DeregisterService expects to be called without the lock held, so we emulate that here
6757 m->SPSState = 2;
6758 #ifndef SPC_DISABLED
6759 if (oldstate == 1) mDNS_DeregisterService(m, &m->SPSRecords);
6760 #else
6761 (void)oldstate;
6762 #endif
6763 mDNS_ReclaimLockAfterCallback();
6764 }
6765
6766 m->SleepState = SleepState_Transferring;
6767 if (m->SystemWakeOnLANEnabled && m->DelaySleep)
6768 {
6769 // If we just woke up moments ago, allow ten seconds for networking to stabilize before going back to sleep
6770 LogSPS("mDNSCoreMachineSleep: Re-sleeping immediately after waking; will delay for %d ticks", m->DelaySleep - m->timenow);
6771 m->SleepLimit = NonZeroTime(m->DelaySleep + mDNSPlatformOneSecond * 10);
6772 }
6773 else
6774 {
6775 m->DelaySleep = 0;
6776 m->SleepLimit = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 10);
6777 m->mDNSStats.Sleeps++;
6778 BeginSleepProcessing(m);
6779 }
6780
6781 #ifndef UNICAST_DISABLED
6782 SuspendLLQs(m);
6783 #endif
6784 #if APPLE_OSX_mDNSResponder
6785 RemoveAutoTunnel6Record(m);
6786 #endif
6787 LogSPS("mDNSCoreMachineSleep: m->SleepState %d (%s) seq %d", m->SleepState,
6788 m->SleepState == SleepState_Transferring ? "Transferring" :
6789 m->SleepState == SleepState_Sleeping ? "Sleeping" : "?", m->SleepSeqNum);
6790 mDNS_Unlock(m);
6791 }
6792 else if (!sleep) // Waking up
6793 {
6794 mDNSu32 slot;
6795 CacheGroup *cg;
6796 CacheRecord *cr;
6797 NetworkInterfaceInfo *intf;
6798 mDNSs32 currtime, diff;
6799
6800 mDNS_Lock(m);
6801 // Reset SleepLimit back to 0 now that we're awake again.
6802 m->SleepLimit = 0;
6803
6804 // If we were previously sleeping, but now we're not, increment m->SleepSeqNum to indicate that we're entering a new period of wakefulness
6805 if (m->SleepState != SleepState_Awake)
6806 {
6807 m->SleepState = SleepState_Awake;
6808 m->SleepSeqNum++;
6809 // If the machine wakes and then immediately tries to sleep again (e.g. a maintenance wake)
6810 // then we enforce a minimum delay of five seconds before we begin sleep processing.
6811 // This is to allow time for the Ethernet link to come up, DHCP to get an address, mDNS to issue queries, etc.,
6812 // before we make our determination of whether there's a Sleep Proxy out there we should register with.
6813 m->DelaySleep = NonZeroTime(m->timenow + kDarkWakeDelaySleep);
6814 }
6815
6816 if (m->SPSState == 3)
6817 {
6818 m->SPSState = 0;
6819 mDNSCoreBeSleepProxyServer_internal(m, m->SPSType, m->SPSPortability, m->SPSMarginalPower, m->SPSTotalPower, m->SPSFeatureFlags);
6820 }
6821 m->mDNSStats.Wakes++;
6822 m->DelayConflictProcessing = MAX_CONFLICT_PROCESSING_DELAYS;
6823 // ... and the same for NextSPSAttempt
6824 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next)) intf->NextSPSAttempt = -1;
6825
6826 // Restart unicast and multicast queries
6827 mDNSCoreRestartQueries(m);
6828
6829 // and reactivtate service registrations
6830 m->NextSRVUpdate = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
6831 LogInfo("mDNSCoreMachineSleep waking: NextSRVUpdate in %d %d", m->NextSRVUpdate - m->timenow, m->timenow);
6832
6833 // 2. Re-validate our cache records
6834 currtime = mDNSPlatformUTC();
6835
6836 #if APPLE_OSX_mDNSResponder
6837 // start time of this statistics gathering interval
6838 m->StatStartTime = currtime;
6839 #endif // APPLE_OSX_mDNSResponder
6840
6841 diff = currtime - m->TimeSlept;
6842 FORALL_CACHERECORDS(slot, cg, cr)
6843 {
6844 // Temporary fix: For unicast cache records, look at how much time we slept.
6845 // Adjust the RecvTime by the amount of time we slept so that we age the
6846 // cache record appropriately. If it is expired already, purge. If there
6847 // is a network change that happens after the wakeup, we might purge the
6848 // cache anyways and this helps only in the case where there are no network
6849 // changes across sleep/wakeup transition.
6850 //
6851 // Note: If there is a network/DNS server change that already happened and
6852 // these cache entries are already refreshed and we are getting a delayed
6853 // wake up notification, we might adjust the TimeRcvd based on the time slept
6854 // now which can cause the cache to purge pre-maturely. As this is not a very
6855 // common case, this should happen rarely.
6856 if (!cr->resrec.InterfaceID)
6857 {
6858 if (diff > 0)
6859 {
6860 mDNSu32 uTTL = RRUnadjustedTTL(cr->resrec.rroriginalttl);
6861 const mDNSs32 remain = uTTL - (m->timenow - cr->TimeRcvd) / mDNSPlatformOneSecond;
6862
6863 // -if we have slept longer than the remaining TTL, purge and start fresh.
6864 // -if we have been sleeping for a long time, we could reduce TimeRcvd below by
6865 // a sufficiently big value which could cause the value to go into the future
6866 // because of the signed comparison of time. For this to happen, we should have been
6867 // sleeping really long (~24 days). For now, we want to be conservative and flush even
6868 // if we have slept for more than two days.
6869
6870 if (diff >= remain || diff > (2 * 24 * 3600))
6871 {
6872 LogInfo("mDNSCoreMachineSleep: %s: Purging cache entry SleptTime %d, Remaining TTL %d",
6873 CRDisplayString(m, cr), diff, remain);
6874 mDNS_PurgeCacheResourceRecord(m, cr);
6875 continue;
6876 }
6877 cr->TimeRcvd -= (diff * mDNSPlatformOneSecond);
6878 if (m->timenow - (cr->TimeRcvd + ((mDNSs32)uTTL * mDNSPlatformOneSecond)) >= 0)
6879 {
6880 LogInfo("mDNSCoreMachineSleep: %s: Purging after adjusting the remaining TTL %d by %d seconds",
6881 CRDisplayString(m, cr), remain, diff);
6882 mDNS_PurgeCacheResourceRecord(m, cr);
6883 }
6884 else
6885 {
6886 LogInfo("mDNSCoreMachineSleep: %s: Adjusted the remain ttl %u by %d seconds", CRDisplayString(m, cr), remain, diff);
6887 }
6888 }
6889 }
6890 else
6891 {
6892 mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForWake);
6893 }
6894 }
6895
6896 // 3. Retrigger probing and announcing for all our authoritative records
6897 for (rr = m->ResourceRecords; rr; rr=rr->next)
6898 {
6899 if (AuthRecord_uDNS(rr))
6900 {
6901 ActivateUnicastRegistration(m, rr);
6902 }
6903 else
6904 {
6905 mDNSCoreRestartRegistration(m, rr, -1);
6906 }
6907 }
6908
6909 // 4. Refresh NAT mappings
6910 // We don't want to have to assume that all hardware can necessarily keep accurate
6911 // track of passage of time while asleep, so on wake we refresh our NAT mappings.
6912 // We typically wake up with no interfaces active, so there's no need to rush to try to find our external address.
6913 // But if we do get a network configuration change, mDNSMacOSXNetworkChanged will call uDNS_SetupDNSConfig, which
6914 // will call mDNS_SetPrimaryInterfaceInfo, which will call RecreateNATMappings to refresh them, potentially sooner
6915 // than five seconds from now.
6916 LogInfo("mDNSCoreMachineSleep: recreating NAT mappings in 5 seconds");
6917 RecreateNATMappings(m, mDNSPlatformOneSecond * 5);
6918 mDNS_Unlock(m);
6919 }
6920 }
6921
6922 mDNSexport mDNSBool mDNSCoreReadyForSleep(mDNS *m, mDNSs32 now)
6923 {
6924 DNSQuestion *q;
6925 AuthRecord *rr;
6926 NetworkInterfaceInfo *intf;
6927
6928 mDNS_Lock(m);
6929
6930 if (m->DelaySleep) goto notready;
6931
6932 // If we've not hit the sleep limit time, and it's not time for our next retry, we can skip these checks
6933 if (m->SleepLimit - now > 0 && m->NextScheduledSPRetry - now > 0) goto notready;
6934
6935 m->NextScheduledSPRetry = now + 0x40000000UL;
6936
6937 // See if we might need to retransmit any lost Sleep Proxy Registrations
6938 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6939 if (intf->NextSPSAttempt >= 0)
6940 {
6941 if (now - intf->NextSPSAttemptTime >= 0)
6942 {
6943 LogSPS("mDNSCoreReadyForSleep: retrying for %s SPS %d try %d",
6944 intf->ifname, intf->NextSPSAttempt/3, intf->NextSPSAttempt);
6945 SendSPSRegistration(m, intf, zeroID);
6946 // Don't need to "goto notready" here, because if we do still have record registrations
6947 // that have not been acknowledged yet, we'll catch that in the record list scan below.
6948 }
6949 else
6950 if (m->NextScheduledSPRetry - intf->NextSPSAttemptTime > 0)
6951 m->NextScheduledSPRetry = intf->NextSPSAttemptTime;
6952 }
6953
6954 // Scan list of interfaces, and see if we're still waiting for any sleep proxy resolves to complete
6955 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
6956 {
6957 int sps = (intf->NextSPSAttempt == 0) ? 0 : (intf->NextSPSAttempt-1)/3;
6958 if (intf->NetWakeResolve[sps].ThisQInterval >= 0)
6959 {
6960 LogSPS("mDNSCoreReadyForSleep: waiting for SPS Resolve %s %##s (%s)",
6961 intf->ifname, intf->NetWakeResolve[sps].qname.c, DNSTypeName(intf->NetWakeResolve[sps].qtype));
6962 goto spsnotready;
6963 }
6964 }
6965
6966 // Scan list of registered records
6967 for (rr = m->ResourceRecords; rr; rr = rr->next)
6968 if (!AuthRecord_uDNS(rr))
6969 if (!mDNSOpaque64IsZero(&rr->updateIntID))
6970 { LogSPS("mDNSCoreReadyForSleep: waiting for SPS updateIntID 0x%x 0x%x (updateid %d) %s", rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(rr->updateid), ARDisplayString(m,rr)); goto spsnotready; }
6971
6972 // Scan list of private LLQs, and make sure they've all completed their handshake with the server
6973 for (q = m->Questions; q; q = q->next)
6974 if (!mDNSOpaque16IsZero(q->TargetQID) && q->LongLived && q->ReqLease == 0 && q->tcp)
6975 {
6976 LogSPS("mDNSCoreReadyForSleep: waiting for LLQ %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
6977 goto notready;
6978 }
6979
6980 // Scan list of registered records
6981 for (rr = m->ResourceRecords; rr; rr = rr->next)
6982 if (AuthRecord_uDNS(rr))
6983 {
6984 if (rr->state == regState_Refresh && rr->tcp)
6985 { LogSPS("mDNSCoreReadyForSleep: waiting for Record updateIntID 0x%x 0x%x (updateid %d) %s", rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(rr->updateid), ARDisplayString(m,rr)); goto notready; }
6986 #if APPLE_OSX_mDNSResponder
6987 if (!RecordReadyForSleep(m, rr)) { LogSPS("mDNSCoreReadyForSleep: waiting for %s", ARDisplayString(m, rr)); goto notready; }
6988 #endif
6989 }
6990
6991 mDNS_Unlock(m);
6992 return mDNStrue;
6993
6994 spsnotready:
6995
6996 // If we failed to complete sleep proxy registration within ten seconds, we give up on that
6997 // and allow up to ten seconds more to complete wide-area deregistration instead
6998 if (now - m->SleepLimit >= 0)
6999 {
7000 LogMsg("Failed to register with SPS, now sending goodbyes");
7001
7002 for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
7003 if (intf->NetWakeBrowse.ThisQInterval >= 0)
7004 {
7005 LogSPS("ReadyForSleep mDNS_DeactivateNetWake %s %##s (%s)",
7006 intf->ifname, intf->NetWakeResolve[0].qname.c, DNSTypeName(intf->NetWakeResolve[0].qtype));
7007 mDNS_DeactivateNetWake_internal(m, intf);
7008 }
7009
7010 for (rr = m->ResourceRecords; rr; rr = rr->next)
7011 if (!AuthRecord_uDNS(rr))
7012 if (!mDNSOpaque64IsZero(&rr->updateIntID))
7013 {
7014 LogSPS("ReadyForSleep clearing updateIntID 0x%x 0x%x (updateid %d) for %s", rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(rr->updateid), ARDisplayString(m, rr));
7015 rr->updateIntID = zeroOpaque64;
7016 }
7017
7018 // We'd really like to allow up to ten seconds more here,
7019 // but if we don't respond to the sleep notification within 30 seconds
7020 // we'll be put back to sleep forcibly without the chance to schedule the next maintenance wake.
7021 // Right now we wait 16 sec after wake for all the interfaces to come up, then we wait up to 10 seconds
7022 // more for SPS resolves and record registrations to complete, which puts us at 26 seconds.
7023 // If we allow just one more second to send our goodbyes, that puts us at 27 seconds.
7024 m->SleepLimit = now + mDNSPlatformOneSecond * 1;
7025
7026 SendSleepGoodbyes(m, mDNStrue, mDNStrue);
7027 }
7028
7029 notready:
7030 mDNS_Unlock(m);
7031 return mDNSfalse;
7032 }
7033
7034 mDNSexport mDNSs32 mDNSCoreIntervalToNextWake(mDNS *const m, mDNSs32 now)
7035 {
7036 AuthRecord *ar;
7037
7038 // Even when we have no wake-on-LAN-capable interfaces, or we failed to find a sleep proxy, or we have other
7039 // failure scenarios, we still want to wake up in at most 120 minutes, to see if the network environment has changed.
7040 // E.g. we might wake up and find no wireless network because the base station got rebooted just at that moment,
7041 // and if that happens we don't want to just give up and go back to sleep and never try again.
7042 mDNSs32 e = now + (120 * 60 * mDNSPlatformOneSecond); // Sleep for at most 120 minutes
7043
7044 NATTraversalInfo *nat;
7045 for (nat = m->NATTraversals; nat; nat=nat->next)
7046 if (nat->Protocol && nat->ExpiryTime && nat->ExpiryTime - now > mDNSPlatformOneSecond*4)
7047 {
7048 mDNSs32 t = nat->ExpiryTime - (nat->ExpiryTime - now) / 10; // Wake up when 90% of the way to the expiry time
7049 if (e - t > 0) e = t;
7050 LogSPS("ComputeWakeTime: %p %s Int %5d Ext %5d Err %d Retry %5d Interval %5d Expire %5d Wake %5d",
7051 nat, nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP",
7052 mDNSVal16(nat->IntPort), mDNSVal16(nat->ExternalPort), nat->Result,
7053 nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0,
7054 nat->retryInterval / mDNSPlatformOneSecond,
7055 nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0,
7056 (t - now) / mDNSPlatformOneSecond);
7057 }
7058
7059 // This loop checks both the time we need to renew wide-area registrations,
7060 // and the time we need to renew Sleep Proxy registrations
7061 for (ar = m->ResourceRecords; ar; ar = ar->next)
7062 if (ar->expire && ar->expire - now > mDNSPlatformOneSecond*4)
7063 {
7064 mDNSs32 t = ar->expire - (ar->expire - now) / 10; // Wake up when 90% of the way to the expiry time
7065 if (e - t > 0) e = t;
7066 LogSPS("ComputeWakeTime: %p Int %7d Next %7d Expire %7d Wake %7d %s",
7067 ar, ar->ThisAPInterval / mDNSPlatformOneSecond,
7068 (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond,
7069 ar->expire ? (ar->expire - now) / mDNSPlatformOneSecond : 0,
7070 (t - now) / mDNSPlatformOneSecond, ARDisplayString(m, ar));
7071 }
7072
7073 return(e - now);
7074 }
7075
7076 // ***************************************************************************
7077 #if COMPILER_LIKES_PRAGMA_MARK
7078 #pragma mark -
7079 #pragma mark - Packet Reception Functions
7080 #endif
7081
7082 #define MustSendRecord(RR) ((RR)->NR_AnswerTo || (RR)->NR_AdditionalTo)
7083
7084 mDNSlocal mDNSu8 *GenerateUnicastResponse(const DNSMessage *const query, const mDNSu8 *const end,
7085 const mDNSInterfaceID InterfaceID, mDNSBool LegacyQuery, DNSMessage *const response, AuthRecord *ResponseRecords)
7086 {
7087 mDNSu8 *responseptr = response->data;
7088 const mDNSu8 *const limit = response->data + sizeof(response->data);
7089 const mDNSu8 *ptr = query->data;
7090 AuthRecord *rr;
7091 mDNSu32 maxttl = 0x70000000;
7092 int i;
7093
7094 // Initialize the response fields so we can answer the questions
7095 InitializeDNSMessage(&response->h, query->h.id, ResponseFlags);
7096
7097 // ***
7098 // *** 1. Write out the list of questions we are actually going to answer with this packet
7099 // ***
7100 if (LegacyQuery)
7101 {
7102 maxttl = kStaticCacheTTL;
7103 for (i=0; i<query->h.numQuestions; i++) // For each question...
7104 {
7105 DNSQuestion q;
7106 ptr = getQuestion(query, ptr, end, InterfaceID, &q); // get the question...
7107 if (!ptr) return(mDNSNULL);
7108
7109 for (rr=ResponseRecords; rr; rr=rr->NextResponse) // and search our list of proposed answers
7110 {
7111 if (rr->NR_AnswerTo == ptr) // If we're going to generate a record answering this question
7112 { // then put the question in the question section
7113 responseptr = putQuestion(response, responseptr, limit, &q.qname, q.qtype, q.qclass);
7114 if (!responseptr) { debugf("GenerateUnicastResponse: Ran out of space for questions!"); return(mDNSNULL); }
7115 break; // break out of the ResponseRecords loop, and go on to the next question
7116 }
7117 }
7118 }
7119
7120 if (response->h.numQuestions == 0) { LogMsg("GenerateUnicastResponse: ERROR! Why no questions?"); return(mDNSNULL); }
7121 }
7122
7123 // ***
7124 // *** 2. Write Answers
7125 // ***
7126 for (rr=ResponseRecords; rr; rr=rr->NextResponse)
7127 if (rr->NR_AnswerTo)
7128 {
7129 mDNSu8 *p = PutResourceRecordTTL(response, responseptr, &response->h.numAnswers, &rr->resrec,
7130 maxttl < rr->resrec.rroriginalttl ? maxttl : rr->resrec.rroriginalttl);
7131 if (p) responseptr = p;
7132 else { debugf("GenerateUnicastResponse: Ran out of space for answers!"); response->h.flags.b[0] |= kDNSFlag0_TC; }
7133 }
7134
7135 // ***
7136 // *** 3. Write Additionals
7137 // ***
7138 for (rr=ResponseRecords; rr; rr=rr->NextResponse)
7139 if (rr->NR_AdditionalTo && !rr->NR_AnswerTo)
7140 {
7141 mDNSu8 *p = PutResourceRecordTTL(response, responseptr, &response->h.numAdditionals, &rr->resrec,
7142 maxttl < rr->resrec.rroriginalttl ? maxttl : rr->resrec.rroriginalttl);
7143 if (p) responseptr = p;
7144 else debugf("GenerateUnicastResponse: No more space for additionals");
7145 }
7146
7147 return(responseptr);
7148 }
7149
7150 // AuthRecord *our is our Resource Record
7151 // CacheRecord *pkt is the Resource Record from the response packet we've witnessed on the network
7152 // Returns 0 if there is no conflict
7153 // Returns +1 if there was a conflict and we won
7154 // Returns -1 if there was a conflict and we lost and have to rename
7155 mDNSlocal int CompareRData(const AuthRecord *const our, const CacheRecord *const pkt)
7156 {
7157 mDNSu8 ourdata[256], *ourptr = ourdata, *ourend;
7158 mDNSu8 pktdata[256], *pktptr = pktdata, *pktend;
7159 if (!our) { LogMsg("CompareRData ERROR: our is NULL"); return(+1); }
7160 if (!pkt) { LogMsg("CompareRData ERROR: pkt is NULL"); return(+1); }
7161
7162 ourend = putRData(mDNSNULL, ourdata, ourdata + sizeof(ourdata), &our->resrec);
7163 pktend = putRData(mDNSNULL, pktdata, pktdata + sizeof(pktdata), &pkt->resrec);
7164 while (ourptr < ourend && pktptr < pktend && *ourptr == *pktptr) { ourptr++; pktptr++; }
7165 if (ourptr >= ourend && pktptr >= pktend) return(0); // If data identical, not a conflict
7166
7167 if (ourptr >= ourend) return(-1); // Our data ran out first; We lost
7168 if (pktptr >= pktend) return(+1); // Packet data ran out first; We won
7169 if (*pktptr > *ourptr) return(-1); // Our data is numerically lower; We lost
7170 if (*pktptr < *ourptr) return(+1); // Packet data is numerically lower; We won
7171
7172 LogMsg("CompareRData ERROR: Invalid state");
7173 return(-1);
7174 }
7175
7176 // See if we have an authoritative record that's identical to this packet record,
7177 // whose canonical DependentOn record is the specified master record.
7178 // The DependentOn pointer is typically used for the TXT record of service registrations
7179 // It indicates that there is no inherent conflict detection for the TXT record
7180 // -- it depends on the SRV record to resolve name conflicts
7181 // If we find any identical ResourceRecords in our authoritative list, then follow their DependentOn
7182 // pointer chain (if any) to make sure we reach the canonical DependentOn record
7183 // If the record has no DependentOn, then just return that record's pointer
7184 // Returns NULL if we don't have any local RRs that are identical to the one from the packet
7185 mDNSlocal mDNSBool MatchDependentOn(const mDNS *const m, const CacheRecord *const pktrr, const AuthRecord *const master)
7186 {
7187 const AuthRecord *r1;
7188 for (r1 = m->ResourceRecords; r1; r1=r1->next)
7189 {
7190 if (IdenticalResourceRecord(&r1->resrec, &pktrr->resrec))
7191 {
7192 const AuthRecord *r2 = r1;
7193 while (r2->DependentOn) r2 = r2->DependentOn;
7194 if (r2 == master) return(mDNStrue);
7195 }
7196 }
7197 for (r1 = m->DuplicateRecords; r1; r1=r1->next)
7198 {
7199 if (IdenticalResourceRecord(&r1->resrec, &pktrr->resrec))
7200 {
7201 const AuthRecord *r2 = r1;
7202 while (r2->DependentOn) r2 = r2->DependentOn;
7203 if (r2 == master) return(mDNStrue);
7204 }
7205 }
7206 return(mDNSfalse);
7207 }
7208
7209 // Find the canonical RRSet pointer for this RR received in a packet.
7210 // If we find any identical AuthRecord in our authoritative list, then follow its RRSet
7211 // pointers (if any) to make sure we return the canonical member of this name/type/class
7212 // Returns NULL if we don't have any local RRs that are identical to the one from the packet
7213 mDNSlocal const AuthRecord *FindRRSet(const mDNS *const m, const CacheRecord *const pktrr)
7214 {
7215 const AuthRecord *rr;
7216 for (rr = m->ResourceRecords; rr; rr=rr->next)
7217 {
7218 if (IdenticalResourceRecord(&rr->resrec, &pktrr->resrec))
7219 {
7220 while (rr->RRSet && rr != rr->RRSet) rr = rr->RRSet;
7221 return(rr);
7222 }
7223 }
7224 return(mDNSNULL);
7225 }
7226
7227 // PacketRRConflict is called when we've received an RR (pktrr) which has the same name
7228 // as one of our records (our) but different rdata.
7229 // 1. If our record is not a type that's supposed to be unique, we don't care.
7230 // 2a. If our record is marked as dependent on some other record for conflict detection, ignore this one.
7231 // 2b. If the packet rr exactly matches one of our other RRs, and *that* record's DependentOn pointer
7232 // points to our record, ignore this conflict (e.g. the packet record matches one of our
7233 // TXT records, and that record is marked as dependent on 'our', its SRV record).
7234 // 3. If we have some *other* RR that exactly matches the one from the packet, and that record and our record
7235 // are members of the same RRSet, then this is not a conflict.
7236 mDNSlocal mDNSBool PacketRRConflict(const mDNS *const m, const AuthRecord *const our, const CacheRecord *const pktrr)
7237 {
7238 // If not supposed to be unique, not a conflict
7239 if (!(our->resrec.RecordType & kDNSRecordTypeUniqueMask)) return(mDNSfalse);
7240
7241 // If a dependent record, not a conflict
7242 if (our->DependentOn || MatchDependentOn(m, pktrr, our)) return(mDNSfalse);
7243 else
7244 {
7245 // If the pktrr matches a member of ourset, not a conflict
7246 const AuthRecord *ourset = our->RRSet ? our->RRSet : our;
7247 const AuthRecord *pktset = FindRRSet(m, pktrr);
7248 if (pktset == ourset) return(mDNSfalse);
7249
7250 // For records we're proxying, where we don't know the full
7251 // relationship between the records, having any matching record
7252 // in our AuthRecords list is sufficient evidence of non-conflict
7253 if (our->WakeUp.HMAC.l[0] && pktset) return(mDNSfalse);
7254 }
7255
7256 // Okay, this is a conflict
7257 return(mDNStrue);
7258 }
7259
7260 // Note: ResolveSimultaneousProbe calls mDNS_Deregister_internal which can call a user callback, which may change
7261 // the record list and/or question list.
7262 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
7263 mDNSlocal void ResolveSimultaneousProbe(mDNS *const m, const DNSMessage *const query, const mDNSu8 *const end,
7264 DNSQuestion *q, AuthRecord *our)
7265 {
7266 int i;
7267 const mDNSu8 *ptr = LocateAuthorities(query, end);
7268 mDNSBool FoundUpdate = mDNSfalse;
7269
7270 for (i = 0; i < query->h.numAuthorities; i++)
7271 {
7272 ptr = GetLargeResourceRecord(m, query, ptr, end, q->InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
7273 if (!ptr) break;
7274 if (m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && ResourceRecordAnswersQuestion(&m->rec.r.resrec, q))
7275 {
7276 FoundUpdate = mDNStrue;
7277 if (PacketRRConflict(m, our, &m->rec.r))
7278 {
7279 int result = (int)our->resrec.rrclass - (int)m->rec.r.resrec.rrclass;
7280 if (!result) result = (int)our->resrec.rrtype - (int)m->rec.r.resrec.rrtype;
7281 if (!result) result = CompareRData(our, &m->rec.r);
7282 if (result)
7283 {
7284 const char *const msg = (result < 0) ? "lost:" : (result > 0) ? "won: " : "tie: ";
7285 LogMsg("ResolveSimultaneousProbe: %p Pkt Record: %08lX %s", q->InterfaceID, m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
7286 LogMsg("ResolveSimultaneousProbe: %p Our Record %d %s %08lX %s", our->resrec.InterfaceID, our->ProbeCount, msg, our->resrec.rdatahash, ARDisplayString(m, our));
7287 }
7288 // If we lost the tie-break for simultaneous probes, we don't immediately give up, because we might be seeing stale packets on the network.
7289 // Instead we pause for one second, to give the other host (if real) a chance to establish its name, and then try probing again.
7290 // If there really is another live host out there with the same name, it will answer our probes and we'll then rename.
7291 if (result < 0)
7292 {
7293 m->SuppressProbes = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
7294 our->ProbeCount = DefaultProbeCountForTypeUnique;
7295 our->AnnounceCount = InitialAnnounceCount;
7296 InitializeLastAPTime(m, our);
7297 goto exit;
7298 }
7299 }
7300 #if 0
7301 else
7302 {
7303 LogMsg("ResolveSimultaneousProbe: %p Pkt Record: %08lX %s", q->InterfaceID, m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
7304 LogMsg("ResolveSimultaneousProbe: %p Our Record %d ign: %08lX %s", our->resrec.InterfaceID, our->ProbeCount, our->resrec.rdatahash, ARDisplayString(m, our));
7305 }
7306 #endif
7307 }
7308 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
7309 }
7310 if (!FoundUpdate)
7311 LogInfo("ResolveSimultaneousProbe: %##s (%s): No Update Record found", our->resrec.name->c, DNSTypeName(our->resrec.rrtype));
7312 exit:
7313 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
7314 }
7315
7316 mDNSlocal CacheRecord *FindIdenticalRecordInCache(const mDNS *const m, const ResourceRecord *const pktrr)
7317 {
7318 mDNSu32 slot = HashSlot(pktrr->name);
7319 CacheGroup *cg = CacheGroupForRecord(m, slot, pktrr);
7320 CacheRecord *rr;
7321 mDNSBool match;
7322 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
7323 {
7324 if (!pktrr->InterfaceID)
7325 {
7326 mDNSu16 id1 = (pktrr->rDNSServer ? pktrr->rDNSServer->resGroupID : 0);
7327 mDNSu16 id2 = (rr->resrec.rDNSServer ? rr->resrec.rDNSServer->resGroupID : 0);
7328 match = (id1 == id2);
7329 }
7330 else match = (pktrr->InterfaceID == rr->resrec.InterfaceID);
7331
7332 if (match && IdenticalSameNameRecord(pktrr, &rr->resrec)) break;
7333 }
7334 return(rr);
7335 }
7336 mDNSlocal void DeregisterProxyRecord(mDNS *const m, AuthRecord *const rr)
7337 {
7338 rr->WakeUp.HMAC = zeroEthAddr; // Clear HMAC so that mDNS_Deregister_internal doesn't waste packets trying to wake this host
7339 rr->RequireGoodbye = mDNSfalse; // and we don't want to send goodbye for it
7340 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
7341 SetSPSProxyListChanged(m->rec.r.resrec.InterfaceID);
7342 }
7343
7344 mDNSlocal void ClearKeepaliveProxyRecords(mDNS *const m, const OwnerOptData *const owner, AuthRecord *const thelist, const mDNSInterfaceID InterfaceID)
7345 {
7346 if (m->CurrentRecord)
7347 LogMsg("ClearKeepaliveProxyRecords ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
7348 m->CurrentRecord = thelist;
7349
7350 // Normally, the RDATA of the keepalive record will be different each time and hence we always
7351 // clean up the keepalive record.
7352 while (m->CurrentRecord)
7353 {
7354 AuthRecord *const rr = m->CurrentRecord;
7355 if (InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&owner->HMAC, &rr->WakeUp.HMAC))
7356 {
7357 if (mDNS_KeepaliveRecord(&m->rec.r.resrec))
7358 {
7359 LogSPS("ClearKeepaliveProxyRecords: Removing %3d H-MAC %.6a I-MAC %.6a %d %d %s",
7360 m->ProxyRecords, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, owner->seq, ARDisplayString(m, rr));
7361 DeregisterProxyRecord(m, rr);
7362 }
7363 }
7364 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
7365 // new records could have been added to the end of the list as a result of that call.
7366 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
7367 m->CurrentRecord = rr->next;
7368 }
7369 }
7370
7371 // Called from mDNSCoreReceiveUpdate when we get a sleep proxy registration request,
7372 // to check our lists and discard any stale duplicates of this record we already have
7373 mDNSlocal void ClearIdenticalProxyRecords(mDNS *const m, const OwnerOptData *const owner, AuthRecord *const thelist)
7374 {
7375 if (m->CurrentRecord)
7376 LogMsg("ClearIdenticalProxyRecords ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
7377 m->CurrentRecord = thelist;
7378 while (m->CurrentRecord)
7379 {
7380 AuthRecord *const rr = m->CurrentRecord;
7381 if (m->rec.r.resrec.InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&owner->HMAC, &rr->WakeUp.HMAC))
7382 if (IdenticalResourceRecord(&rr->resrec, &m->rec.r.resrec))
7383 {
7384 LogSPS("ClearIdenticalProxyRecords: Removing %3d H-MAC %.6a I-MAC %.6a %d %d %s",
7385 m->ProxyRecords, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, owner->seq, ARDisplayString(m, rr));
7386 DeregisterProxyRecord(m, rr);
7387 }
7388 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
7389 // new records could have been added to the end of the list as a result of that call.
7390 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
7391 m->CurrentRecord = rr->next;
7392 }
7393 }
7394
7395 // Called from ProcessQuery when we get an mDNS packet with an owner record in it
7396 mDNSlocal void ClearProxyRecords(mDNS *const m, const OwnerOptData *const owner, AuthRecord *const thelist)
7397 {
7398 if (m->CurrentRecord)
7399 LogMsg("ClearProxyRecords ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
7400 m->CurrentRecord = thelist;
7401 while (m->CurrentRecord)
7402 {
7403 AuthRecord *const rr = m->CurrentRecord;
7404 if (m->rec.r.resrec.InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&owner->HMAC, &rr->WakeUp.HMAC))
7405 if (owner->seq != rr->WakeUp.seq || m->timenow - rr->TimeRcvd > mDNSPlatformOneSecond * 60)
7406 {
7407 if (rr->AddressProxy.type == mDNSAddrType_IPv6)
7408 {
7409 // We don't do this here because we know that the host is waking up at this point, so we don't send
7410 // Unsolicited Neighbor Advertisements -- even Neighbor Advertisements agreeing with what the host should be
7411 // saying itself -- because it can cause some IPv6 stacks to falsely conclude that there's an address conflict.
7412 #if MDNS_USE_Unsolicited_Neighbor_Advertisements
7413 LogSPS("NDP Announcement -- Releasing traffic for H-MAC %.6a I-MAC %.6a %s",
7414 &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
7415 SendNDP(m, NDP_Adv, NDP_Override, rr, &rr->AddressProxy.ip.v6, &rr->WakeUp.IMAC, &AllHosts_v6, &AllHosts_v6_Eth);
7416 #endif
7417 }
7418 LogSPS("ClearProxyRecords: Removing %3d AC %2d %02X H-MAC %.6a I-MAC %.6a %d %d %s",
7419 m->ProxyRecords, rr->AnnounceCount, rr->resrec.RecordType,
7420 &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, owner->seq, ARDisplayString(m, rr));
7421 if (rr->resrec.RecordType == kDNSRecordTypeDeregistering) rr->resrec.RecordType = kDNSRecordTypeShared;
7422 rr->WakeUp.HMAC = zeroEthAddr; // Clear HMAC so that mDNS_Deregister_internal doesn't waste packets trying to wake this host
7423 rr->RequireGoodbye = mDNSfalse; // and we don't want to send goodbye for it, since real host is now back and functional
7424 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
7425 SetSPSProxyListChanged(m->rec.r.resrec.InterfaceID);
7426 }
7427 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
7428 // new records could have been added to the end of the list as a result of that call.
7429 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
7430 m->CurrentRecord = rr->next;
7431 }
7432 }
7433
7434 // ProcessQuery examines a received query to see if we have any answers to give
7435 mDNSlocal mDNSu8 *ProcessQuery(mDNS *const m, const DNSMessage *const query, const mDNSu8 *const end,
7436 const mDNSAddr *srcaddr, const mDNSInterfaceID InterfaceID, mDNSBool LegacyQuery, mDNSBool QueryWasMulticast,
7437 mDNSBool QueryWasLocalUnicast, DNSMessage *const response)
7438 {
7439 mDNSBool FromLocalSubnet = srcaddr && mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
7440 AuthRecord *ResponseRecords = mDNSNULL;
7441 AuthRecord **nrp = &ResponseRecords;
7442
7443 #if POOF_ENABLED
7444 CacheRecord *ExpectedAnswers = mDNSNULL; // Records in our cache we expect to see updated
7445 CacheRecord **eap = &ExpectedAnswers;
7446 #endif // POOF_ENABLED
7447
7448 DNSQuestion *DupQuestions = mDNSNULL; // Our questions that are identical to questions in this packet
7449 DNSQuestion **dqp = &DupQuestions;
7450 mDNSs32 delayresponse = 0;
7451 mDNSBool SendLegacyResponse = mDNSfalse;
7452 const mDNSu8 *ptr;
7453 mDNSu8 *responseptr = mDNSNULL;
7454 AuthRecord *rr;
7455 int i;
7456 CacheRecord *McastNSEC3Records = mDNSNULL;
7457
7458 // ***
7459 // *** 1. Look in Additional Section for an OPT record
7460 // ***
7461 ptr = LocateOptRR(query, end, DNSOpt_OwnerData_ID_Space);
7462 if (ptr)
7463 {
7464 ptr = GetLargeResourceRecord(m, query, ptr, end, InterfaceID, kDNSRecordTypePacketAdd, &m->rec);
7465 if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
7466 {
7467 const rdataOPT *opt;
7468 const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
7469 // Find owner sub-option(s). We verify that the MAC is non-zero, otherwise we could inadvertently
7470 // delete all our own AuthRecords (which are identified by having zero MAC tags on them).
7471 for (opt = &m->rec.r.resrec.rdata->u.opt[0]; opt < e; opt++)
7472 if (opt->opt == kDNSOpt_Owner && opt->u.owner.vers == 0 && opt->u.owner.HMAC.l[0])
7473 {
7474 ClearProxyRecords(m, &opt->u.owner, m->DuplicateRecords);
7475 ClearProxyRecords(m, &opt->u.owner, m->ResourceRecords);
7476 }
7477 }
7478 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
7479 }
7480
7481 //
7482 // Look in Authority Section for NSEC3 record
7483 //
7484
7485 mDNSParseNSEC3Records(m, query, end, InterfaceID, &McastNSEC3Records);
7486
7487 // ***
7488 // *** 2. Parse Question Section and mark potential answers
7489 // ***
7490 ptr = query->data;
7491 for (i=0; i<query->h.numQuestions; i++) // For each question...
7492 {
7493 mDNSBool QuestionNeedsMulticastResponse;
7494 int NumAnswersForThisQuestion = 0;
7495 AuthRecord *NSECAnswer = mDNSNULL;
7496 DNSQuestion pktq, *q;
7497 ptr = getQuestion(query, ptr, end, InterfaceID, &pktq); // get the question...
7498 if (!ptr) goto exit;
7499
7500 pktq.AnonInfo = mDNSNULL;
7501 if (McastNSEC3Records)
7502 InitializeAnonInfoForQuestion(m, &McastNSEC3Records, &pktq);
7503 // The only queries that *need* a multicast response are:
7504 // * Queries sent via multicast
7505 // * from port 5353
7506 // * that don't have the kDNSQClass_UnicastResponse bit set
7507 // These queries need multicast responses because other clients will:
7508 // * suppress their own identical questions when they see these questions, and
7509 // * expire their cache records if they don't see the expected responses
7510 // For other queries, we may still choose to send the occasional multicast response anyway,
7511 // to keep our neighbours caches warm, and for ongoing conflict detection.
7512 QuestionNeedsMulticastResponse = QueryWasMulticast && !LegacyQuery && !(pktq.qclass & kDNSQClass_UnicastResponse);
7513
7514 if (pktq.qclass & kDNSQClass_UnicastResponse)
7515 m->mDNSStats.UnicastBitInQueries++;
7516 else
7517 m->mDNSStats.NormalQueries++;
7518
7519 // Clear the UnicastResponse flag -- don't want to confuse the rest of the code that follows later
7520 pktq.qclass &= ~kDNSQClass_UnicastResponse;
7521
7522 // Note: We use the m->CurrentRecord mechanism here because calling ResolveSimultaneousProbe
7523 // can result in user callbacks which may change the record list and/or question list.
7524 // Also note: we just mark potential answer records here, without trying to build the
7525 // "ResponseRecords" list, because we don't want to risk user callbacks deleting records
7526 // from that list while we're in the middle of trying to build it.
7527 if (m->CurrentRecord)
7528 LogMsg("ProcessQuery ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
7529 m->CurrentRecord = m->ResourceRecords;
7530 while (m->CurrentRecord)
7531 {
7532 rr = m->CurrentRecord;
7533 m->CurrentRecord = rr->next;
7534 if (AnyTypeRecordAnswersQuestion(&rr->resrec, &pktq) && (QueryWasMulticast || QueryWasLocalUnicast || rr->AllowRemoteQuery))
7535 {
7536 m->mDNSStats.MatchingAnswersForQueries++;
7537 if (RRTypeAnswersQuestionType(&rr->resrec, pktq.qtype))
7538 {
7539 if (rr->resrec.RecordType == kDNSRecordTypeUnique)
7540 ResolveSimultaneousProbe(m, query, end, &pktq, rr);
7541 else if (ResourceRecordIsValidAnswer(rr))
7542 {
7543 NumAnswersForThisQuestion++;
7544 // As we have verified this question to be part of the same subset,
7545 // set the anonymous data which is needed below when walk the cache
7546 // records to see what answers we should be expecting. The cache records
7547 // may cache only the nsec3RR and not the anonymous data itself.
7548 if (pktq.AnonInfo && rr->resrec.AnonInfo)
7549 SetAnonData(&pktq, &rr->resrec, mDNStrue);
7550
7551 // Note: We should check here if this is a probe-type query, and if so, generate an immediate
7552 // unicast answer back to the source, because timeliness in answering probes is important.
7553
7554 // Notes:
7555 // NR_AnswerTo pointing into query packet means "answer via immediate legacy unicast" (may *also* choose to multicast)
7556 // NR_AnswerTo == NR_AnswerUnicast means "answer via delayed unicast" (to modern querier; may promote to multicast instead)
7557 // NR_AnswerTo == NR_AnswerMulticast means "definitely answer via multicast" (can't downgrade to unicast later)
7558 // If we're not multicasting this record because the kDNSQClass_UnicastResponse bit was set,
7559 // but the multicast querier is not on a matching subnet (e.g. because of overlaid subnets on one link)
7560 // then we'll multicast it anyway (if we unicast, the receiver will ignore it because it has an apparently non-local source)
7561 if (QuestionNeedsMulticastResponse || (!FromLocalSubnet && QueryWasMulticast && !LegacyQuery))
7562 {
7563 // We only mark this question for sending if it is at least one second since the last time we multicast it
7564 // on this interface. If it is more than a second, or LastMCInterface is different, then we may multicast it.
7565 // This is to guard against the case where someone blasts us with queries as fast as they can.
7566 if (m->timenow - (rr->LastMCTime + mDNSPlatformOneSecond) >= 0 ||
7567 (rr->LastMCInterface != mDNSInterfaceMark && rr->LastMCInterface != InterfaceID))
7568 rr->NR_AnswerTo = NR_AnswerMulticast;
7569 }
7570 else if (!rr->NR_AnswerTo) rr->NR_AnswerTo = LegacyQuery ? ptr : NR_AnswerUnicast;
7571 }
7572 }
7573 else if ((rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && ResourceRecordIsValidAnswer(rr))
7574 {
7575 // If we don't have any answers for this question, but we do own another record with the same name,
7576 // then we'll want to mark it to generate an NSEC record on this interface
7577 if (!NSECAnswer) NSECAnswer = rr;
7578 }
7579 }
7580 }
7581
7582 if (NumAnswersForThisQuestion == 0 && NSECAnswer)
7583 {
7584 NumAnswersForThisQuestion++;
7585 NSECAnswer->SendNSECNow = InterfaceID;
7586 m->NextScheduledResponse = m->timenow;
7587 }
7588
7589 // If we couldn't answer this question, someone else might be able to,
7590 // so use random delay on response to reduce collisions
7591 if (NumAnswersForThisQuestion == 0) delayresponse = mDNSPlatformOneSecond; // Divided by 50 = 20ms
7592
7593 if (query->h.flags.b[0] & kDNSFlag0_TC)
7594 m->mDNSStats.KnownAnswerMultiplePkts++;
7595 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7596 if (QuestionNeedsMulticastResponse)
7597 #else
7598 // We only do the following accelerated cache expiration and duplicate question suppression processing
7599 // for non-truncated multicast queries with multicast responses.
7600 // For any query generating a unicast response we don't do this because we can't assume we will see the response.
7601 // For truncated queries we don't do this because a response we're expecting might be suppressed by a subsequent
7602 // known-answer packet, and when there's packet loss we can't safely assume we'll receive *all* known-answer packets.
7603 if (QuestionNeedsMulticastResponse && !(query->h.flags.b[0] & kDNSFlag0_TC))
7604 #endif
7605 {
7606 #if POOF_ENABLED
7607 const mDNSu32 slot = HashSlot(&pktq.qname);
7608 CacheGroup *cg = CacheGroupForName(m, slot, pktq.qnamehash, &pktq.qname);
7609 CacheRecord *cr;
7610
7611 // Make a list indicating which of our own cache records we expect to see updated as a result of this query
7612 // Note: Records larger than 1K are not habitually multicast, so don't expect those to be updated
7613 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7614 if (!(query->h.flags.b[0] & kDNSFlag0_TC))
7615 #endif // ENABLE_MULTI_PACKET_QUERY_SNOOPING
7616 for (cr = cg ? cg->members : mDNSNULL; cr; cr=cr->next)
7617 if (SameNameRecordAnswersQuestion(&cr->resrec, &pktq) && cr->resrec.rdlength <= SmallRecordLimit)
7618 if (!cr->NextInKAList && eap != &cr->NextInKAList)
7619 {
7620 *eap = cr;
7621 eap = &cr->NextInKAList;
7622 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7623 if (cr->MPUnansweredQ == 0 || m->timenow - cr->MPLastUnansweredQT >= mDNSPlatformOneSecond)
7624 {
7625 // Although MPUnansweredQ is only really used for multi-packet query processing,
7626 // we increment it for both single-packet and multi-packet queries, so that it stays in sync
7627 // with the MPUnansweredKA value, which by necessity is incremented for both query types.
7628 cr->MPUnansweredQ++;
7629 cr->MPLastUnansweredQT = m->timenow;
7630 cr->MPExpectingKA = mDNStrue;
7631 }
7632 #endif // ENABLE_MULTI_PACKET_QUERY_SNOOPING
7633 }
7634 #endif // POOF_ENABLED
7635
7636 // Check if this question is the same as any of mine.
7637 // We only do this for non-truncated queries. Right now it would be too complicated to try
7638 // to keep track of duplicate suppression state between multiple packets, especially when we
7639 // can't guarantee to receive all of the Known Answer packets that go with a particular query.
7640 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7641 if (!(query->h.flags.b[0] & kDNSFlag0_TC))
7642 #endif
7643 // For anonymous question, the duplicate suppressesion should happen if the
7644 // question belongs in the same group. As the group is expected to be
7645 // small, we don't do the optimization for now.
7646 if (!pktq.AnonInfo)
7647 {
7648 for (q = m->Questions; q; q=q->next)
7649 if (!q->Target.type && ActiveQuestion(q) && m->timenow - q->LastQTxTime > mDNSPlatformOneSecond / 4)
7650 if (!q->InterfaceID || q->InterfaceID == InterfaceID)
7651 if (q->NextInDQList == mDNSNULL && dqp != &q->NextInDQList)
7652 if (q->qtype == pktq.qtype &&
7653 q->qclass == pktq.qclass &&
7654 q->qnamehash == pktq.qnamehash && SameDomainName(&q->qname, &pktq.qname))
7655 { *dqp = q; dqp = &q->NextInDQList; }
7656 }
7657 }
7658 if (pktq.AnonInfo)
7659 {
7660 FreeAnonInfo(pktq.AnonInfo);
7661 }
7662 }
7663
7664 // ***
7665 // *** 3. Now we can safely build the list of marked answers
7666 // ***
7667 for (rr = m->ResourceRecords; rr; rr=rr->next) // Now build our list of potential answers
7668 if (rr->NR_AnswerTo) // If we marked the record...
7669 AddRecordToResponseList(&nrp, rr, mDNSNULL); // ... add it to the list
7670
7671 // ***
7672 // *** 4. Add additional records
7673 // ***
7674 AddAdditionalsToResponseList(m, ResponseRecords, &nrp, InterfaceID);
7675
7676 // ***
7677 // *** 5. Parse Answer Section and cancel any records disallowed by Known-Answer list
7678 // ***
7679 for (i=0; i<query->h.numAnswers; i++) // For each record in the query's answer section...
7680 {
7681 // Get the record...
7682 CacheRecord *ourcacherr;
7683 ptr = GetLargeResourceRecord(m, query, ptr, end, InterfaceID, kDNSRecordTypePacketAns, &m->rec);
7684 if (!ptr) goto exit;
7685 if (m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative)
7686 {
7687 // See if this Known-Answer suppresses any of our currently planned answers
7688 for (rr=ResponseRecords; rr; rr=rr->NextResponse)
7689 {
7690 if (MustSendRecord(rr) && ShouldSuppressKnownAnswer(&m->rec.r, rr))
7691 {
7692 m->mDNSStats.KnownAnswerSuppressions++;
7693 rr->NR_AnswerTo = mDNSNULL;
7694 rr->NR_AdditionalTo = mDNSNULL;
7695 }
7696 }
7697
7698 // See if this Known-Answer suppresses any previously scheduled answers (for multi-packet KA suppression)
7699 for (rr=m->ResourceRecords; rr; rr=rr->next)
7700 {
7701 // If we're planning to send this answer on this interface, and only on this interface, then allow KA suppression
7702 if (rr->ImmedAnswer == InterfaceID && ShouldSuppressKnownAnswer(&m->rec.r, rr))
7703 {
7704 if (srcaddr->type == mDNSAddrType_IPv4)
7705 {
7706 if (mDNSSameIPv4Address(rr->v4Requester, srcaddr->ip.v4)) rr->v4Requester = zerov4Addr;
7707 }
7708 else if (srcaddr->type == mDNSAddrType_IPv6)
7709 {
7710 if (mDNSSameIPv6Address(rr->v6Requester, srcaddr->ip.v6)) rr->v6Requester = zerov6Addr;
7711 }
7712 if (mDNSIPv4AddressIsZero(rr->v4Requester) && mDNSIPv6AddressIsZero(rr->v6Requester))
7713 {
7714 m->mDNSStats.KnownAnswerSuppressions++;
7715 rr->ImmedAnswer = mDNSNULL;
7716 rr->ImmedUnicast = mDNSfalse;
7717 #if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
7718 LogMsg("Suppressed after%4d: %s", m->timenow - rr->ImmedAnswerMarkTime, ARDisplayString(m, rr));
7719 #endif
7720 }
7721 }
7722 }
7723
7724 ourcacherr = FindIdenticalRecordInCache(m, &m->rec.r.resrec);
7725
7726 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7727 // See if this Known-Answer suppresses any answers we were expecting for our cache records. We do this always,
7728 // even if the TC bit is not set (the TC bit will *not* be set in the *last* packet of a multi-packet KA list).
7729 if (ourcacherr && ourcacherr->MPExpectingKA && m->timenow - ourcacherr->MPLastUnansweredQT < mDNSPlatformOneSecond)
7730 {
7731 ourcacherr->MPUnansweredKA++;
7732 ourcacherr->MPExpectingKA = mDNSfalse;
7733 }
7734 #endif
7735
7736 #if POOF_ENABLED
7737 // Having built our ExpectedAnswers list from the questions in this packet, we then remove
7738 // any records that are suppressed by the Known Answer list in this packet.
7739 eap = &ExpectedAnswers;
7740 while (*eap)
7741 {
7742 CacheRecord *cr = *eap;
7743 if (cr->resrec.InterfaceID == InterfaceID && IdenticalResourceRecord(&m->rec.r.resrec, &cr->resrec))
7744 { *eap = cr->NextInKAList; cr->NextInKAList = mDNSNULL; }
7745 else eap = &cr->NextInKAList;
7746 }
7747 #endif // POOF_ENABLED
7748
7749 // See if this Known-Answer is a surprise to us. If so, we shouldn't suppress our own query.
7750 if (!ourcacherr)
7751 {
7752 dqp = &DupQuestions;
7753 while (*dqp)
7754 {
7755 DNSQuestion *q = *dqp;
7756 if (ResourceRecordAnswersQuestion(&m->rec.r.resrec, q))
7757 { *dqp = q->NextInDQList; q->NextInDQList = mDNSNULL; }
7758 else dqp = &q->NextInDQList;
7759 }
7760 }
7761 }
7762 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
7763 }
7764
7765 // ***
7766 // *** 6. Cancel any additionals that were added because of now-deleted records
7767 // ***
7768 for (rr=ResponseRecords; rr; rr=rr->NextResponse)
7769 if (rr->NR_AdditionalTo && !MustSendRecord(rr->NR_AdditionalTo))
7770 { rr->NR_AnswerTo = mDNSNULL; rr->NR_AdditionalTo = mDNSNULL; }
7771
7772 // ***
7773 // *** 7. Mark the send flags on the records we plan to send
7774 // ***
7775 for (rr=ResponseRecords; rr; rr=rr->NextResponse)
7776 {
7777 if (rr->NR_AnswerTo)
7778 {
7779 mDNSBool SendMulticastResponse = mDNSfalse; // Send modern multicast response
7780 mDNSBool SendUnicastResponse = mDNSfalse; // Send modern unicast response (not legacy unicast response)
7781
7782 #if !TARGET_OS_EMBEDDED
7783 // always honor kDNSQClass_UnicastResponse in embedded environment to increase reliability
7784 // in high multicast packet loss environments.
7785
7786 // If it's been one TTL/4 since we multicast this, then send a multicast response
7787 // for conflict detection, etc.
7788 if (m->timenow - (rr->LastMCTime + TicksTTL(rr)/4) >= 0)
7789 {
7790 SendMulticastResponse = mDNStrue;
7791 // If this record was marked for modern (delayed) unicast response, then mark it as promoted to
7792 // multicast response instead (don't want to end up ALSO setting SendUnicastResponse in the check below).
7793 // If this record was marked for legacy unicast response, then we mustn't change the NR_AnswerTo value.
7794 if (rr->NR_AnswerTo == NR_AnswerUnicast)
7795 {
7796 m->mDNSStats.UnicastDemotedToMulticast++;
7797 rr->NR_AnswerTo = NR_AnswerMulticast;
7798 }
7799 }
7800 #endif // !TARGET_OS_EMBEDDED
7801
7802 // If the client insists on a multicast response, then we'd better send one
7803 if (rr->NR_AnswerTo == NR_AnswerMulticast)
7804 {
7805 m->mDNSStats.MulticastResponses++;
7806 SendMulticastResponse = mDNStrue;
7807 }
7808 else if (rr->NR_AnswerTo == NR_AnswerUnicast)
7809 {
7810 m->mDNSStats.UnicastResponses++;
7811 SendUnicastResponse = mDNStrue;
7812 }
7813 else if (rr->NR_AnswerTo)
7814 {
7815 SendLegacyResponse = mDNStrue;
7816 }
7817
7818 if (SendMulticastResponse || SendUnicastResponse)
7819 {
7820 #if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
7821 rr->ImmedAnswerMarkTime = m->timenow;
7822 #endif
7823 m->NextScheduledResponse = m->timenow;
7824 // If we're already planning to send this on another interface, just send it on all interfaces
7825 if (rr->ImmedAnswer && rr->ImmedAnswer != InterfaceID)
7826 rr->ImmedAnswer = mDNSInterfaceMark;
7827 else
7828 {
7829 rr->ImmedAnswer = InterfaceID; // Record interface to send it on
7830 if (SendUnicastResponse) rr->ImmedUnicast = mDNStrue;
7831 if (srcaddr->type == mDNSAddrType_IPv4)
7832 {
7833 if (mDNSIPv4AddressIsZero(rr->v4Requester)) rr->v4Requester = srcaddr->ip.v4;
7834 else if (!mDNSSameIPv4Address(rr->v4Requester, srcaddr->ip.v4)) rr->v4Requester = onesIPv4Addr;
7835 }
7836 else if (srcaddr->type == mDNSAddrType_IPv6)
7837 {
7838 if (mDNSIPv6AddressIsZero(rr->v6Requester)) rr->v6Requester = srcaddr->ip.v6;
7839 else if (!mDNSSameIPv6Address(rr->v6Requester, srcaddr->ip.v6)) rr->v6Requester = onesIPv6Addr;
7840 }
7841 }
7842 }
7843 // If TC flag is set, it means we should expect that additional known answers may be coming in another packet,
7844 // so we allow roughly half a second before deciding to reply (we've observed inter-packet delays of 100-200ms on 802.11)
7845 // else, if record is a shared one, spread responses over 100ms to avoid implosion of simultaneous responses
7846 // else, for a simple unique record reply, we can reply immediately; no need for delay
7847 if (query->h.flags.b[0] & kDNSFlag0_TC) delayresponse = mDNSPlatformOneSecond * 20; // Divided by 50 = 400ms
7848 else if (rr->resrec.RecordType == kDNSRecordTypeShared) delayresponse = mDNSPlatformOneSecond; // Divided by 50 = 20ms
7849 }
7850 else if (rr->NR_AdditionalTo && rr->NR_AdditionalTo->NR_AnswerTo == NR_AnswerMulticast)
7851 {
7852 // Since additional records are an optimization anyway, we only ever send them on one interface at a time
7853 // If two clients on different interfaces do queries that invoke the same optional additional answer,
7854 // then the earlier client is out of luck
7855 rr->ImmedAdditional = InterfaceID;
7856 // No need to set m->NextScheduledResponse here
7857 // We'll send these additional records when we send them, or not, as the case may be
7858 }
7859 }
7860
7861 // ***
7862 // *** 8. If we think other machines are likely to answer these questions, set our packet suppression timer
7863 // ***
7864 if (delayresponse && (!m->SuppressSending || (m->SuppressSending - m->timenow) < (delayresponse + 49) / 50))
7865 {
7866 #if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
7867 mDNSs32 oldss = m->SuppressSending;
7868 if (oldss && delayresponse)
7869 LogMsg("Current SuppressSending delay%5ld; require%5ld", m->SuppressSending - m->timenow, (delayresponse + 49) / 50);
7870 #endif
7871 // Pick a random delay:
7872 // We start with the base delay chosen above (typically either 1 second or 20 seconds),
7873 // and add a random value in the range 0-5 seconds (making 1-6 seconds or 20-25 seconds).
7874 // This is an integer value, with resolution determined by the platform clock rate.
7875 // We then divide that by 50 to get the delay value in ticks. We defer the division until last
7876 // to get better results on platforms with coarse clock granularity (e.g. ten ticks per second).
7877 // The +49 before dividing is to ensure we round up, not down, to ensure that even
7878 // on platforms where the native clock rate is less than fifty ticks per second,
7879 // we still guarantee that the final calculated delay is at least one platform tick.
7880 // We want to make sure we don't ever allow the delay to be zero ticks,
7881 // because if that happens we'll fail the Bonjour Conformance Test.
7882 // Our final computed delay is 20-120ms for normal delayed replies,
7883 // or 400-500ms in the case of multi-packet known-answer lists.
7884 m->SuppressSending = m->timenow + (delayresponse + (mDNSs32)mDNSRandom((mDNSu32)mDNSPlatformOneSecond*5) + 49) / 50;
7885 if (m->SuppressSending == 0) m->SuppressSending = 1;
7886 #if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
7887 if (oldss && delayresponse)
7888 LogMsg("Set SuppressSending to %5ld", m->SuppressSending - m->timenow);
7889 #endif
7890 }
7891
7892 // ***
7893 // *** 9. If query is from a legacy client, or from a new client requesting a unicast reply, then generate a unicast response too
7894 // ***
7895 if (SendLegacyResponse)
7896 responseptr = GenerateUnicastResponse(query, end, InterfaceID, LegacyQuery, response, ResponseRecords);
7897
7898 exit:
7899 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
7900
7901 // ***
7902 // *** 10. Finally, clear our link chains ready for use next time
7903 // ***
7904 while (ResponseRecords)
7905 {
7906 rr = ResponseRecords;
7907 ResponseRecords = rr->NextResponse;
7908 rr->NextResponse = mDNSNULL;
7909 rr->NR_AnswerTo = mDNSNULL;
7910 rr->NR_AdditionalTo = mDNSNULL;
7911 }
7912
7913 #if POOF_ENABLED
7914 while (ExpectedAnswers)
7915 {
7916 CacheRecord *cr = ExpectedAnswers;
7917 ExpectedAnswers = cr->NextInKAList;
7918 cr->NextInKAList = mDNSNULL;
7919
7920 // For non-truncated queries, we can definitively say that we should expect
7921 // to be seeing a response for any records still left in the ExpectedAnswers list
7922 if (!(query->h.flags.b[0] & kDNSFlag0_TC))
7923 if (cr->UnansweredQueries == 0 || m->timenow - cr->LastUnansweredTime >= mDNSPlatformOneSecond * 3/4)
7924 {
7925 cr->UnansweredQueries++;
7926 cr->LastUnansweredTime = m->timenow;
7927 if (cr->UnansweredQueries > 1)
7928 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7929 debugf("ProcessQuery: (!TC) UAQ %lu MPQ %lu MPKA %lu %s",
7930 cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
7931 #else
7932 debugf("ProcessQuery: UnansweredQueries %lu %s", cr->UnansweredQueries, CRDisplayString(m, cr));
7933 #endif // ENABLE_MULTI_PACKET_QUERY_SNOOPING
7934 SetNextCacheCheckTimeForRecord(m, cr);
7935 }
7936
7937 // If we've seen multiple unanswered queries for this record,
7938 // then mark it to expire in five seconds if we don't get a response by then.
7939 if (cr->UnansweredQueries >= MaxUnansweredQueries)
7940 {
7941 // Only show debugging message if this record was not about to expire anyway
7942 if (RRExpireTime(cr) - m->timenow > (mDNSs32) kDefaultReconfirmTimeForNoAnswer * 4 / 3 + mDNSPlatformOneSecond)
7943 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7944 debugf("ProcessQuery: (Max) UAQ %lu MPQ %lu MPKA %lu mDNS_Reconfirm() for %s",
7945 cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
7946 #else
7947 LogInfo("ProcessQuery: UnansweredQueries %lu TTL %lu mDNS_Reconfirm() for %s",
7948 cr->UnansweredQueries, (RRExpireTime(cr) - m->timenow + mDNSPlatformOneSecond-1) / mDNSPlatformOneSecond, CRDisplayString(m, cr));
7949 #endif // ENABLE_MULTI_PACKET_QUERY_SNOOPING
7950
7951 m->mDNSStats.PoofCacheDeletions++;
7952 mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
7953 }
7954 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
7955 // Make a guess, based on the multi-packet query / known answer counts, whether we think we
7956 // should have seen an answer for this. (We multiply MPQ by 4 and MPKA by 5, to allow for
7957 // possible packet loss of up to 20% of the additional KA packets.)
7958 else if (cr->MPUnansweredQ * 4 > cr->MPUnansweredKA * 5 + 8)
7959 {
7960 // We want to do this conservatively.
7961 // If there are so many machines on the network that they have to use multi-packet known-answer lists,
7962 // then we don't want them to all hit the network simultaneously with their final expiration queries.
7963 // By setting the record to expire in four minutes, we achieve two things:
7964 // (a) the 90-95% final expiration queries will be less bunched together
7965 // (b) we allow some time for us to witness enough other failed queries that we don't have to do our own
7966 mDNSu32 remain = (mDNSu32)(RRExpireTime(cr) - m->timenow) / 4;
7967 if (remain > 240 * (mDNSu32)mDNSPlatformOneSecond)
7968 remain = 240 * (mDNSu32)mDNSPlatformOneSecond;
7969
7970 // Only show debugging message if this record was not about to expire anyway
7971 if (RRExpireTime(cr) - m->timenow > 4 * mDNSPlatformOneSecond)
7972 debugf("ProcessQuery: (MPQ) UAQ %lu MPQ %lu MPKA %lu mDNS_Reconfirm() for %s",
7973 cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
7974
7975 if (remain <= 60 * (mDNSu32)mDNSPlatformOneSecond)
7976 cr->UnansweredQueries++; // Treat this as equivalent to one definite unanswered query
7977 cr->MPUnansweredQ = 0; // Clear MPQ/MPKA statistics
7978 cr->MPUnansweredKA = 0;
7979 cr->MPExpectingKA = mDNSfalse;
7980
7981 if (remain < kDefaultReconfirmTimeForNoAnswer)
7982 remain = kDefaultReconfirmTimeForNoAnswer;
7983 mDNS_Reconfirm_internal(m, cr, remain);
7984 }
7985 #endif // ENABLE_MULTI_PACKET_QUERY_SNOOPING
7986 }
7987 #endif // POOF_ENABLED
7988
7989 while (DupQuestions)
7990 {
7991 DNSQuestion *q = DupQuestions;
7992 DupQuestions = q->NextInDQList;
7993 q->NextInDQList = mDNSNULL;
7994 RecordDupSuppressInfo(q->DupSuppress, m->timenow, InterfaceID, srcaddr->type);
7995 debugf("ProcessQuery: Recorded DSI for %##s (%s) on %p/%s", q->qname.c, DNSTypeName(q->qtype), InterfaceID,
7996 srcaddr->type == mDNSAddrType_IPv4 ? "v4" : "v6");
7997 }
7998
7999 if (McastNSEC3Records)
8000 {
8001 debugf("ProcessQuery: McastNSEC3Records not used");
8002 FreeNSECRecords(m, McastNSEC3Records);
8003 }
8004
8005 return(responseptr);
8006 }
8007
8008 mDNSlocal void mDNSCoreReceiveQuery(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *const end,
8009 const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
8010 const mDNSInterfaceID InterfaceID)
8011 {
8012 mDNSu8 *responseend = mDNSNULL;
8013 mDNSBool QueryWasLocalUnicast = srcaddr && dstaddr &&
8014 !mDNSAddrIsDNSMulticast(dstaddr) && mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
8015
8016 if (!dstaddr || (!InterfaceID && mDNSAddrIsDNSMulticast(dstaddr)))
8017 {
8018 const char *const reason = !dstaddr ? "Received over TCP connection" : "Multicast, but no InterfaceID";
8019 LogMsg("Ignoring Query from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
8020 "%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes (%s)",
8021 srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
8022 msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
8023 msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
8024 msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
8025 msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data, reason);
8026 return;
8027 }
8028
8029 verbosedebugf("Received Query from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
8030 "%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes",
8031 srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
8032 msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
8033 msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
8034 msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
8035 msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data);
8036
8037 responseend = ProcessQuery(m, msg, end, srcaddr, InterfaceID,
8038 !mDNSSameIPPort(srcport, MulticastDNSPort), mDNSAddrIsDNSMulticast(dstaddr), QueryWasLocalUnicast, &m->omsg);
8039
8040 if (responseend) // If responseend is non-null, that means we built a unicast response packet
8041 {
8042 debugf("Unicast Response: %d Question%s, %d Answer%s, %d Additional%s to %#-15a:%d on %p/%ld",
8043 m->omsg.h.numQuestions, m->omsg.h.numQuestions == 1 ? "" : "s",
8044 m->omsg.h.numAnswers, m->omsg.h.numAnswers == 1 ? "" : "s",
8045 m->omsg.h.numAdditionals, m->omsg.h.numAdditionals == 1 ? "" : "s",
8046 srcaddr, mDNSVal16(srcport), InterfaceID, srcaddr->type);
8047 mDNSSendDNSMessage(m, &m->omsg, responseend, InterfaceID, mDNSNULL, srcaddr, srcport, mDNSNULL, mDNSNULL, mDNSfalse);
8048 }
8049 }
8050
8051 #if 0
8052 mDNSlocal mDNSBool TrustedSource(const mDNS *const m, const mDNSAddr *const srcaddr)
8053 {
8054 DNSServer *s;
8055 (void)m; // Unused
8056 (void)srcaddr; // Unused
8057 for (s = m->DNSServers; s; s = s->next)
8058 if (mDNSSameAddress(srcaddr, &s->addr)) return(mDNStrue);
8059 return(mDNSfalse);
8060 }
8061 #endif
8062
8063 struct UDPSocket_struct
8064 {
8065 mDNSIPPort port; // MUST BE FIRST FIELD -- mDNSCoreReceive expects every UDPSocket_struct to begin with mDNSIPPort port
8066 };
8067
8068 mDNSlocal DNSQuestion *ExpectingUnicastResponseForQuestion(const mDNS *const m, const mDNSIPPort port, const mDNSOpaque16 id, const DNSQuestion *const question, mDNSBool tcp)
8069 {
8070 DNSQuestion *q;
8071 for (q = m->Questions; q; q=q->next)
8072 {
8073 if (!tcp && !q->LocalSocket) continue;
8074 if (mDNSSameIPPort(tcp ? q->tcpSrcPort : q->LocalSocket->port, port) &&
8075 mDNSSameOpaque16(q->TargetQID, id) &&
8076 q->qtype == question->qtype &&
8077 q->qclass == question->qclass &&
8078 q->qnamehash == question->qnamehash &&
8079 SameDomainName(&q->qname, &question->qname))
8080 return(q);
8081 }
8082 return(mDNSNULL);
8083 }
8084
8085 // This function is called when we receive a unicast response. This could be the case of a unicast response from the
8086 // DNS server or a response to the QU query. Hence, the cache record's InterfaceId can be both NULL or non-NULL (QU case)
8087 mDNSlocal DNSQuestion *ExpectingUnicastResponseForRecord(mDNS *const m,
8088 const mDNSAddr *const srcaddr, const mDNSBool SrcLocal, const mDNSIPPort port, const mDNSOpaque16 id, const CacheRecord *const rr, mDNSBool tcp)
8089 {
8090 DNSQuestion *q;
8091 (void)id;
8092 (void)srcaddr;
8093
8094 for (q = m->Questions; q; q=q->next)
8095 {
8096 if (!q->DuplicateOf && ResourceRecordAnswersUnicastResponse(&rr->resrec, q))
8097 {
8098 if (!mDNSOpaque16IsZero(q->TargetQID))
8099 {
8100 debugf("ExpectingUnicastResponseForRecord msg->h.id %d q->TargetQID %d for %s", mDNSVal16(id), mDNSVal16(q->TargetQID), CRDisplayString(m, rr));
8101
8102 if (mDNSSameOpaque16(q->TargetQID, id))
8103 {
8104 mDNSIPPort srcp;
8105 if (!tcp)
8106 {
8107 srcp = q->LocalSocket ? q->LocalSocket->port : zeroIPPort;
8108 }
8109 else
8110 {
8111 srcp = q->tcpSrcPort;
8112 }
8113 if (mDNSSameIPPort(srcp, port)) return(q);
8114
8115 // if (mDNSSameAddress(srcaddr, &q->Target)) return(mDNStrue);
8116 // if (q->LongLived && mDNSSameAddress(srcaddr, &q->servAddr)) return(mDNStrue); Shouldn't need this now that we have LLQType checking
8117 // if (TrustedSource(m, srcaddr)) return(mDNStrue);
8118 LogInfo("WARNING: Ignoring suspect uDNS response for %##s (%s) [q->Target %#a:%d] from %#a:%d %s",
8119 q->qname.c, DNSTypeName(q->qtype), &q->Target, mDNSVal16(srcp), srcaddr, mDNSVal16(port), CRDisplayString(m, rr));
8120 return(mDNSNULL);
8121 }
8122 }
8123 else
8124 {
8125 if (SrcLocal && q->ExpectUnicastResp && (mDNSu32)(m->timenow - q->ExpectUnicastResp) < (mDNSu32)(mDNSPlatformOneSecond*2))
8126 return(q);
8127 }
8128 }
8129 }
8130 return(mDNSNULL);
8131 }
8132
8133 // Certain data types need more space for in-memory storage than their in-packet rdlength would imply
8134 // Currently this applies only to rdata types containing more than one domainname,
8135 // or types where the domainname is not the last item in the structure.
8136 mDNSlocal mDNSu16 GetRDLengthMem(const ResourceRecord *const rr)
8137 {
8138 switch (rr->rrtype)
8139 {
8140 case kDNSType_SOA: return sizeof(rdataSOA);
8141 case kDNSType_RP: return sizeof(rdataRP);
8142 case kDNSType_PX: return sizeof(rdataPX);
8143 default: return rr->rdlength;
8144 }
8145 }
8146
8147 mDNSexport CacheRecord *CreateNewCacheEntry(mDNS *const m, const mDNSu32 slot, CacheGroup *cg, mDNSs32 delay, mDNSBool Add, const mDNSAddr *sourceAddress)
8148 {
8149 CacheRecord *rr = mDNSNULL;
8150 mDNSu16 RDLength = GetRDLengthMem(&m->rec.r.resrec);
8151
8152 if (!m->rec.r.resrec.InterfaceID) debugf("CreateNewCacheEntry %s", CRDisplayString(m, &m->rec.r));
8153
8154 //if (RDLength > InlineCacheRDSize)
8155 // LogInfo("Rdata len %4d > InlineCacheRDSize %d %s", RDLength, InlineCacheRDSize, CRDisplayString(m, &m->rec.r));
8156
8157 if (!cg) cg = GetCacheGroup(m, slot, &m->rec.r.resrec); // If we don't have a CacheGroup for this name, make one now
8158 if (cg) rr = GetCacheRecord(m, cg, RDLength); // Make a cache record, being careful not to recycle cg
8159 if (!rr) NoCacheAnswer(m, &m->rec.r);
8160 else
8161 {
8162 RData *saveptr = rr->resrec.rdata; // Save the rr->resrec.rdata pointer
8163 *rr = m->rec.r; // Block copy the CacheRecord object
8164 rr->resrec.rdata = saveptr; // Restore rr->resrec.rdata after the structure assignment
8165 rr->resrec.name = cg->name; // And set rr->resrec.name to point into our CacheGroup header
8166
8167 // We need to add the anonymous info before we call CacheRecordAdd so that
8168 // if it finds a matching question with this record, it bumps up the counters like
8169 // CurrentAnswers etc. Otherwise, when a cache entry gets removed, CacheRecordRmv
8170 // will complain.
8171 if (m->rec.r.resrec.AnonInfo)
8172 {
8173 rr->resrec.AnonInfo = m->rec.r.resrec.AnonInfo;
8174 m->rec.r.resrec.AnonInfo = mDNSNULL;
8175 }
8176 rr->DelayDelivery = delay;
8177
8178 // If this is an oversized record with external storage allocated, copy rdata to external storage
8179 if (rr->resrec.rdata == (RData*)&rr->smallrdatastorage && RDLength > InlineCacheRDSize)
8180 LogMsg("rr->resrec.rdata == &rr->rdatastorage but length > InlineCacheRDSize %##s", m->rec.r.resrec.name->c);
8181 else if (rr->resrec.rdata != (RData*)&rr->smallrdatastorage && RDLength <= InlineCacheRDSize)
8182 LogMsg("rr->resrec.rdata != &rr->rdatastorage but length <= InlineCacheRDSize %##s", m->rec.r.resrec.name->c);
8183 if (RDLength > InlineCacheRDSize)
8184 mDNSPlatformMemCopy(rr->resrec.rdata, m->rec.r.resrec.rdata, sizeofRDataHeader + RDLength);
8185
8186 rr->next = mDNSNULL; // Clear 'next' pointer
8187 rr->nsec = mDNSNULL;
8188 rr->soa = mDNSNULL;
8189
8190 if (sourceAddress)
8191 rr->sourceAddress = *sourceAddress;
8192
8193 if (!rr->resrec.InterfaceID)
8194 {
8195 m->rrcache_totalused_unicast += rr->resrec.rdlength;
8196 if (DNSSECRecordType(rr->resrec.rrtype))
8197 BumpDNSSECStats(m, kStatsActionIncrement, kStatsTypeMemoryUsage, rr->resrec.rdlength);
8198 }
8199
8200 if (Add)
8201 {
8202 *(cg->rrcache_tail) = rr; // Append this record to tail of cache slot list
8203 cg->rrcache_tail = &(rr->next); // Advance tail pointer
8204 CacheRecordAdd(m, rr); // CacheRecordAdd calls SetNextCacheCheckTimeForRecord(m, rr); for us
8205 }
8206 else
8207 {
8208 // Can't use the "cg->name" if we are not adding to the cache as the
8209 // CacheGroup may be released anytime if it is empty
8210 domainname *name = mDNSPlatformMemAllocate(DomainNameLength(cg->name));
8211 if (name)
8212 {
8213 AssignDomainName(name, cg->name);
8214 rr->resrec.name = name;
8215 }
8216 else
8217 {
8218 ReleaseCacheRecord(m, rr);
8219 NoCacheAnswer(m, &m->rec.r);
8220 rr = mDNSNULL;
8221 }
8222 }
8223 }
8224 return(rr);
8225 }
8226
8227 mDNSlocal void RefreshCacheRecord(mDNS *const m, CacheRecord *rr, mDNSu32 ttl)
8228 {
8229 rr->TimeRcvd = m->timenow;
8230 rr->resrec.rroriginalttl = ttl;
8231 rr->UnansweredQueries = 0;
8232 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
8233 rr->MPUnansweredQ = 0;
8234 rr->MPUnansweredKA = 0;
8235 rr->MPExpectingKA = mDNSfalse;
8236 #endif
8237 SetNextCacheCheckTimeForRecord(m, rr);
8238 }
8239
8240 mDNSexport void GrantCacheExtensions(mDNS *const m, DNSQuestion *q, mDNSu32 lease)
8241 {
8242 CacheRecord *rr;
8243 const mDNSu32 slot = HashSlot(&q->qname);
8244 CacheGroup *cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
8245 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
8246 if (rr->CRActiveQuestion == q)
8247 {
8248 //LogInfo("GrantCacheExtensions: new lease %d / %s", lease, CRDisplayString(m, rr));
8249 RefreshCacheRecord(m, rr, lease);
8250 }
8251 }
8252
8253 mDNSlocal mDNSu32 GetEffectiveTTL(const uDNS_LLQType LLQType, mDNSu32 ttl) // TTL in seconds
8254 {
8255 if (LLQType == uDNS_LLQ_Entire) ttl = kLLQ_DefLease;
8256 else if (LLQType == uDNS_LLQ_Events)
8257 {
8258 // If the TTL is -1 for uDNS LLQ event packet, that means "remove"
8259 if (ttl == 0xFFFFFFFF) ttl = 0;
8260 else ttl = kLLQ_DefLease;
8261 }
8262 else // else not LLQ (standard uDNS response)
8263 {
8264 // The TTL is already capped to a maximum value in GetLargeResourceRecord, but just to be extra safe we
8265 // also do this check here to make sure we can't get overflow below when we add a quarter to the TTL
8266 if (ttl > 0x60000000UL / mDNSPlatformOneSecond) ttl = 0x60000000UL / mDNSPlatformOneSecond;
8267
8268 ttl = RRAdjustTTL(ttl);
8269
8270 // For mDNS, TTL zero means "delete this record"
8271 // For uDNS, TTL zero means: this data is true at this moment, but don't cache it.
8272 // For the sake of network efficiency, we impose a minimum effective TTL of 15 seconds.
8273 // This means that we'll do our 80, 85, 90, 95% queries at 12.00, 12.75, 13.50, 14.25 seconds
8274 // respectively, and then if we get no response, delete the record from the cache at 15 seconds.
8275 // This gives the server up to three seconds to respond between when we send our 80% query at 12 seconds
8276 // and when we delete the record at 15 seconds. Allowing cache lifetimes less than 15 seconds would
8277 // (with the current code) result in the server having even less than three seconds to respond
8278 // before we deleted the record and reported a "remove" event to any active questions.
8279 // Furthermore, with the current code, if we were to allow a TTL of less than 2 seconds
8280 // then things really break (e.g. we end up making a negative cache entry).
8281 // In the future we may want to revisit this and consider properly supporting non-cached (TTL=0) uDNS answers.
8282 if (ttl < 15) ttl = 15;
8283 }
8284
8285 return ttl;
8286 }
8287
8288 // When the response does not match the question directly, we still want to cache them sometimes. The current response is
8289 // in m->rec.
8290 mDNSlocal mDNSBool IsResponseAcceptable(mDNS *const m, const CacheRecord *crlist, DNSQuestion *q, mDNSBool *nseclist)
8291 {
8292 CacheRecord *const newcr = &m->rec.r;
8293 ResourceRecord *rr = &newcr->resrec;
8294 const CacheRecord *cr;
8295
8296 *nseclist = mDNSfalse;
8297 for (cr = crlist; cr != (CacheRecord*)1; cr = cr->NextInCFList)
8298 {
8299 domainname *target = GetRRDomainNameTarget(&cr->resrec);
8300 // When we issue a query for A record, the response might contain both a CNAME and A records. Only the CNAME would
8301 // match the question and we already created a cache entry in the previous pass of this loop. Now when we process
8302 // the A record, it does not match the question because the record name here is the CNAME. Hence we try to
8303 // match with the previous records to make it an AcceptableResponse. We have to be careful about setting the
8304 // DNSServer value that we got in the previous pass. This can happen for other record types like SRV also.
8305
8306 if (target && cr->resrec.rdatahash == rr->namehash && SameDomainName(target, rr->name))
8307 {
8308 LogInfo("IsResponseAcceptable: Found a matching entry for %##s in the CacheFlushRecords %s", rr->name->c, CRDisplayString(m, cr));
8309 return (mDNStrue);
8310 }
8311 }
8312
8313 // Either the question requires validation or we are validating a response with DNSSEC in which case
8314 // we need to accept the RRSIGs also so that we can validate the response. It is also possible that
8315 // we receive NSECs for our query which does not match the qname and we need to cache in that case
8316 // too. nseclist is set if they have to be cached as part of the negative cache record.
8317 if (q && DNSSECQuestion(q))
8318 {
8319 mDNSBool same = SameDomainName(&q->qname, rr->name);
8320 if (same && (q->qtype == rr->rrtype || rr->rrtype == kDNSType_CNAME))
8321 {
8322 LogInfo("IsResponseAcceptable: Accepting, same name and qtype %s, CR %s", DNSTypeName(q->qtype),
8323 CRDisplayString(m, newcr));
8324 return mDNStrue;
8325 }
8326 // We cache RRSIGS if it covers the question type or NSEC. If it covers a NSEC,
8327 // "nseclist" is set
8328 if (rr->rrtype == kDNSType_RRSIG)
8329 {
8330 RDataBody2 *const rdb = (RDataBody2 *)newcr->smallrdatastorage.data;
8331 rdataRRSig *rrsig = &rdb->rrsig;
8332 mDNSu16 typeCovered = swap16(rrsig->typeCovered);
8333
8334 // Note the ordering. If we are looking up the NSEC record, then the RRSIG's typeCovered
8335 // would match the qtype and they are cached normally as they are not used to prove the
8336 // non-existence of any name. In that case, it is like any other normal dnssec validation
8337 // and hence nseclist should not be set.
8338
8339 if (same && ((typeCovered == q->qtype) || (typeCovered == kDNSType_CNAME)))
8340 {
8341 LogInfo("IsResponseAcceptable: Accepting RRSIG %s matches question type %s", CRDisplayString(m, newcr),
8342 DNSTypeName(q->qtype));
8343 return mDNStrue;
8344 }
8345 else if (typeCovered == kDNSType_NSEC || typeCovered == kDNSType_NSEC3)
8346 {
8347 LogInfo("IsResponseAcceptable: Accepting RRSIG %s matches %s type (nseclist = 1)", CRDisplayString(m, newcr), DNSTypeName(typeCovered));
8348 *nseclist = mDNStrue;
8349 return mDNStrue;
8350 }
8351 else if (typeCovered == kDNSType_SOA)
8352 {
8353 LogInfo("IsResponseAcceptable: Accepting RRSIG %s matches SOA type (nseclist = 1)", CRDisplayString(m, newcr));
8354 *nseclist = mDNStrue;
8355 return mDNStrue;
8356 }
8357 else return mDNSfalse;
8358 }
8359 if (rr->rrtype == kDNSType_NSEC)
8360 {
8361 if (!UNICAST_NSEC(rr))
8362 {
8363 LogMsg("IsResponseAcceptable: ERROR!! Not a unicast NSEC %s", CRDisplayString(m, newcr));
8364 return mDNSfalse;
8365 }
8366 LogInfo("IsResponseAcceptable: Accepting NSEC %s (nseclist = 1)", CRDisplayString(m, newcr));
8367 *nseclist = mDNStrue;
8368 return mDNStrue;
8369 }
8370 if (rr->rrtype == kDNSType_SOA)
8371 {
8372 LogInfo("IsResponseAcceptable: Accepting SOA %s (nseclist = 1)", CRDisplayString(m, newcr));
8373 *nseclist = mDNStrue;
8374 return mDNStrue;
8375 }
8376 else if (rr->rrtype == kDNSType_NSEC3)
8377 {
8378 LogInfo("IsResponseAcceptable: Accepting NSEC3 %s (nseclist = 1)", CRDisplayString(m, newcr));
8379 *nseclist = mDNStrue;
8380 return mDNStrue;
8381 }
8382 }
8383 return mDNSfalse;
8384 }
8385
8386 mDNSlocal void FreeNSECRecords(mDNS *const m, CacheRecord *NSECRecords)
8387 {
8388 CacheRecord *rp, *next;
8389
8390 for (rp = NSECRecords; rp; rp = next)
8391 {
8392 next = rp->next;
8393 ReleaseCacheRecord(m, rp);
8394 }
8395 }
8396
8397 // If we received zero DNSSEC records even when the DO/EDNS0 bit was set, we need to provide this
8398 // information to ValidatingResponse question to indicate the DNSSEC status to the application
8399 mDNSlocal void mDNSCoreReceiveNoDNSSECAnswers(mDNS *const m, const DNSMessage *const response, const mDNSu8 *end, const mDNSAddr *dstaddr,
8400 mDNSIPPort dstport, const mDNSInterfaceID InterfaceID)
8401 {
8402 int i;
8403 const mDNSu8 *ptr = response->data;
8404
8405 for (i = 0; i < response->h.numQuestions && ptr && ptr < end; i++)
8406 {
8407 DNSQuestion pktq;
8408 DNSQuestion *qptr = mDNSNULL;
8409 ptr = getQuestion(response, ptr, end, InterfaceID, &pktq);
8410 if (ptr && (qptr = ExpectingUnicastResponseForQuestion(m, dstport, response->h.id, &pktq, !dstaddr)) &&
8411 qptr->ValidatingResponse)
8412 {
8413 DNSQuestion *next, *q;
8414
8415 if (qptr->DuplicateOf)
8416 LogMsg("mDNSCoreReceiveNoDNSSECAnswers: ERROR!! qptr %##s (%s) Duplicate question matching response", qptr->qname.c, DNSTypeName(qptr->qtype));
8417
8418 // Be careful to call the callback for duplicate questions first and then the original
8419 // question. If we called the callback on the original question, it could stop and
8420 // a duplicate question would become the original question.
8421 mDNS_DropLockBeforeCallback(); // Allow client (and us) to legally make mDNS API calls
8422 for (q = qptr->next ; q && q != m->NewQuestions; q = next)
8423 {
8424 next = q->next;
8425 if (q->DuplicateOf == qptr)
8426 {
8427 if (q->ValidatingResponse)
8428 LogInfo("mDNSCoreReceiveNoDNSSECAnswers: qptr %##s (%s) Duplicate question found", q->qname.c, DNSTypeName(q->qtype));
8429 else
8430 LogMsg("mDNSCoreReceiveNoDNSSECAnswers: ERROR!! qptr %##s (%s) Duplicate question not ValidatingResponse", q->qname.c, DNSTypeName(q->qtype));
8431 if (q->QuestionCallback)
8432 q->QuestionCallback(m, q, mDNSNULL, QC_nodnssec);
8433 }
8434 }
8435 if (qptr->QuestionCallback)
8436 qptr->QuestionCallback(m, qptr, mDNSNULL, QC_nodnssec);
8437 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
8438 }
8439 }
8440 }
8441
8442 mDNSlocal void mDNSCoreReceiveNoUnicastAnswers(mDNS *const m, const DNSMessage *const response, const mDNSu8 *end, const mDNSAddr *dstaddr,
8443 mDNSIPPort dstport, const mDNSInterfaceID InterfaceID, uDNS_LLQType LLQType, mDNSu8 rcode, CacheRecord *NSECRecords)
8444 {
8445 int i;
8446 const mDNSu8 *ptr = response->data;
8447 CacheRecord *SOARecord = mDNSNULL;
8448
8449 for (i = 0; i < response->h.numQuestions && ptr && ptr < end; i++)
8450 {
8451 DNSQuestion q;
8452 DNSQuestion *qptr = mDNSNULL;
8453 ptr = getQuestion(response, ptr, end, InterfaceID, &q);
8454 if (ptr && (qptr = ExpectingUnicastResponseForQuestion(m, dstport, response->h.id, &q, !dstaddr)))
8455 {
8456 CacheRecord *rr, *neg = mDNSNULL;
8457 mDNSu32 slot = HashSlot(&q.qname);
8458 CacheGroup *cg = CacheGroupForName(m, slot, q.qnamehash, &q.qname);
8459 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
8460 if (SameNameRecordAnswersQuestion(&rr->resrec, qptr))
8461 {
8462 // 1. If we got a fresh answer to this query, then don't need to generate a negative entry
8463 if (RRExpireTime(rr) - m->timenow > 0) break;
8464 // 2. If we already had a negative entry, keep track of it so we can resurrect it instead of creating a new one
8465 if (rr->resrec.RecordType == kDNSRecordTypePacketNegative) neg = rr;
8466 }
8467 // When we're doing parallel unicast and multicast queries for dot-local names (for supporting Microsoft
8468 // Active Directory sites) we don't want to waste memory making negative cache entries for all the unicast answers.
8469 // Otherwise we just fill up our cache with negative entries for just about every single multicast name we ever look up
8470 // (since the Microsoft Active Directory server is going to assert that pretty much every single multicast name doesn't exist).
8471 // This is not only a waste of memory, but there's also the problem of those negative entries confusing us later -- e.g. we
8472 // suppress sending our mDNS query packet because we think we already have a valid (negative) answer to that query in our cache.
8473 // The one exception is that we *DO* want to make a negative cache entry for "local. SOA", for the (common) case where we're
8474 // *not* on a Microsoft Active Directory network, and there is no authoritative server for "local". Note that this is not
8475 // in conflict with the mDNS spec, because that spec says, "Multicast DNS Zones have no SOA record," so it's okay to cache
8476 // negative answers for "local. SOA" from a uDNS server, because the mDNS spec already says that such records do not exist :-)
8477 //
8478 // By suppressing negative responses, it might take longer to timeout a .local question as it might be expecting a
8479 // response e.g., we deliver a positive "A" response and suppress negative "AAAA" response and the upper layer may
8480 // be waiting longer to get the AAAA response before returning the "A" response to the application. To handle this
8481 // case without creating the negative cache entries, we generate a negative response and let the layer above us
8482 // do the appropriate thing. This negative response is also needed for appending new search domains.
8483 if (!InterfaceID && q.qtype != kDNSType_SOA && IsLocalDomain(&q.qname))
8484 {
8485 if (!rr)
8486 {
8487 LogInfo("mDNSCoreReceiveNoUnicastAnswers: Generate negative response for %##s (%s)", q.qname.c, DNSTypeName(q.qtype));
8488 m->CurrentQuestion = qptr;
8489 // We are not creating a cache record in this case, we need to pass back
8490 // the error we got so that the proxy code can return the right one to
8491 // the application
8492 if (qptr->ProxyQuestion)
8493 qptr->responseFlags = response->h.flags;
8494 GenerateNegativeResponse(m, mDNSInterface_Any, QC_forceresponse);
8495 m->CurrentQuestion = mDNSNULL;
8496 }
8497 else
8498 {
8499 LogInfo("mDNSCoreReceiveNoUnicastAnswers: Skipping check and not creating a negative cache entry for %##s (%s)", q.qname.c, DNSTypeName(q.qtype));
8500 }
8501 }
8502 else
8503 {
8504 if (!rr)
8505 {
8506 // We start off assuming a negative caching TTL of 60 seconds
8507 // but then look to see if we can find an SOA authority record to tell us a better value we should be using
8508 mDNSu32 negttl = 60;
8509 int repeat = 0;
8510 const domainname *name = &q.qname;
8511 mDNSu32 hash = q.qnamehash;
8512
8513 // Special case for our special Microsoft Active Directory "local SOA" check.
8514 // Some cheap home gateways don't include an SOA record in the authority section when
8515 // they send negative responses, so we don't know how long to cache the negative result.
8516 // Because we don't want to keep hitting the root name servers with our query to find
8517 // if we're on a network using Microsoft Active Directory using "local" as a private
8518 // internal top-level domain, we make sure to cache the negative result for at least one day.
8519 if (q.qtype == kDNSType_SOA && SameDomainName(&q.qname, &localdomain)) negttl = 60 * 60 * 24;
8520
8521 // If we're going to make (or update) a negative entry, then look for the appropriate TTL from the SOA record
8522 if (response->h.numAuthorities && (ptr = LocateAuthorities(response, end)) != mDNSNULL)
8523 {
8524 ptr = GetLargeResourceRecord(m, response, ptr, end, InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
8525 if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_SOA)
8526 {
8527 const mDNSu32 s = HashSlot(m->rec.r.resrec.name);
8528 CacheGroup *cgSOA = CacheGroupForRecord(m, s, &m->rec.r.resrec);
8529 const rdataSOA *const soa = (const rdataSOA *)m->rec.r.resrec.rdata->u.data;
8530 mDNSu32 ttl_s = soa->min;
8531 // We use the lesser of the SOA.MIN field and the SOA record's TTL, *except*
8532 // for the SOA record for ".", where the record is reported as non-cacheable
8533 // (TTL zero) for some reason, so in this case we just take the SOA record's TTL as-is
8534 if (ttl_s > m->rec.r.resrec.rroriginalttl && m->rec.r.resrec.name->c[0])
8535 ttl_s = m->rec.r.resrec.rroriginalttl;
8536 if (negttl < ttl_s) negttl = ttl_s;
8537
8538 // Create the SOA record as we may have to return this to the questions
8539 // that we are acting as a proxy for currently or in the future.
8540 SOARecord = CreateNewCacheEntry(m, s, cgSOA, 1, mDNSfalse, mDNSNULL);
8541
8542 // Special check for SOA queries: If we queried for a.b.c.d.com, and got no answer,
8543 // with an Authority Section SOA record for d.com, then this is a hint that the authority
8544 // is d.com, and consequently SOA records b.c.d.com and c.d.com don't exist either.
8545 // To do this we set the repeat count so the while loop below will make a series of negative cache entries for us
8546 //
8547 // For ProxyQuestions, we don't do this as we need to create additional SOA records to cache them
8548 // along with the negative cache record. For simplicity, we don't create the additional records.
8549 if (!qptr->ProxyQuestion && q.qtype == kDNSType_SOA)
8550 {
8551 int qcount = CountLabels(&q.qname);
8552 int scount = CountLabels(m->rec.r.resrec.name);
8553 if (qcount - 1 > scount)
8554 if (SameDomainName(SkipLeadingLabels(&q.qname, qcount - scount), m->rec.r.resrec.name))
8555 repeat = qcount - 1 - scount;
8556 }
8557 }
8558 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
8559 }
8560
8561 // If we already had a negative entry in the cache, then we double our existing negative TTL. This is to avoid
8562 // the case where the record doesn't exist (e.g. particularly for things like our lb._dns-sd._udp.<domain> query),
8563 // and the server returns no SOA record (or an SOA record with a small MIN TTL) so we assume a TTL
8564 // of 60 seconds, and we end up polling the server every minute for a record that doesn't exist.
8565 // With this fix in place, when this happens, we double the effective TTL each time (up to one hour),
8566 // so that we back off our polling rate and don't keep hitting the server continually.
8567 if (neg)
8568 {
8569 if (negttl < neg->resrec.rroriginalttl * 2)
8570 negttl = neg->resrec.rroriginalttl * 2;
8571 if (negttl > 3600)
8572 negttl = 3600;
8573 }
8574
8575 negttl = GetEffectiveTTL(LLQType, negttl); // Add 25% grace period if necessary
8576
8577 // If we already had a negative cache entry just update it, else make one or more new negative cache entries.
8578 if (neg)
8579 {
8580 LogInfo("mDNSCoreReceiveNoUnicastAnswers: Renewing negative TTL from %d to %d %s", neg->resrec.rroriginalttl, negttl, CRDisplayString(m, neg));
8581 RefreshCacheRecord(m, neg, negttl);
8582 // When we created the cache for the first time and answered the question, the question's
8583 // interval was set to MaxQuestionInterval. If the cache is about to expire and we are resending
8584 // the queries, the interval should still be at MaxQuestionInterval. If the query is being
8585 // restarted (setting it to InitialQuestionInterval) for other reasons e.g., wakeup,
8586 // we should reset its question interval here to MaxQuestionInterval.
8587 ResetQuestionState(m, qptr);
8588 if (DNSSECQuestion(qptr))
8589 neg->CRDNSSECQuestion = 1;
8590 // Update the NSEC records again.
8591 // TBD: Need to purge and revalidate if the cached NSECS and the new set are not same.
8592 if (NSECRecords)
8593 {
8594 if (!AddNSECSForCacheRecord(m, NSECRecords, neg, rcode))
8595 {
8596 // We might just have an SOA record for zones that are not signed and hence don't log
8597 // this as an error
8598 LogInfo("mDNSCoreReceiveNoUnicastAnswers: AddNSECSForCacheRecord failed to add NSEC for negcr %s during refresh", CRDisplayString(m, neg));
8599 FreeNSECRecords(m, NSECRecords);
8600 neg->CRDNSSECQuestion = 0;
8601 }
8602 NSECRecords = mDNSNULL;
8603 }
8604 if (SOARecord)
8605 {
8606 if (neg->soa)
8607 ReleaseCacheRecord(m, neg->soa);
8608 neg->soa = SOARecord;
8609 SOARecord = mDNSNULL;
8610 }
8611 }
8612 else while (1)
8613 {
8614 CacheRecord *negcr;
8615 debugf("mDNSCoreReceiveNoUnicastAnswers making negative cache entry TTL %d for %##s (%s)", negttl, name->c, DNSTypeName(q.qtype));
8616 MakeNegativeCacheRecord(m, &m->rec.r, name, hash, q.qtype, q.qclass, negttl, mDNSInterface_Any, qptr->qDNSServer);
8617 m->rec.r.responseFlags = response->h.flags;
8618 // We create SOA records above which might create new cache groups. Earlier
8619 // in the function we looked up the cache group for the name and it could have
8620 // been NULL. If we pass NULL cg to new cache entries that we create below,
8621 // it will create additional cache groups for the same name. To avoid that,
8622 // look up the cache group again to re-initialize cg again.
8623 cg = CacheGroupForName(m, slot, hash, name);
8624 if (NSECRecords && DNSSECQuestion(qptr))
8625 {
8626 // Create the cache entry with delay and then add the NSEC records
8627 // to it and add it immediately.
8628 negcr = CreateNewCacheEntry(m, slot, cg, 1, mDNStrue, mDNSNULL);
8629 if (negcr)
8630 {
8631 negcr->CRDNSSECQuestion = 0;
8632 if (!AddNSECSForCacheRecord(m, NSECRecords, negcr, rcode))
8633 {
8634 LogInfo("mDNSCoreReceiveNoUnicastAnswers: AddNSECSForCacheRecord failed to add NSEC for negcr %s",
8635 CRDisplayString(m, negcr));
8636 FreeNSECRecords(m, NSECRecords);
8637 }
8638 else
8639 {
8640 negcr->CRDNSSECQuestion = 1;
8641 LogInfo("mDNSCoreReceiveNoUnicastAnswers: AddNSECSForCacheRecord added neg NSEC for %s", CRDisplayString(m, negcr));
8642 }
8643 NSECRecords = mDNSNULL;
8644 negcr->DelayDelivery = 0;
8645 CacheRecordDeferredAdd(m, negcr);
8646 }
8647 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
8648 break;
8649 }
8650 else
8651 {
8652 // Need to add with a delay so that we can tag the SOA record
8653 negcr = CreateNewCacheEntry(m, slot, cg, 1, mDNStrue, mDNSNULL);
8654 if (negcr)
8655 {
8656 negcr->CRDNSSECQuestion = 0;
8657 if (DNSSECQuestion(qptr))
8658 negcr->CRDNSSECQuestion = 1;
8659 negcr->DelayDelivery = 0;
8660
8661 if (SOARecord)
8662 {
8663 if (negcr->soa)
8664 ReleaseCacheRecord(m, negcr->soa);
8665 negcr->soa = SOARecord;
8666 SOARecord = mDNSNULL;
8667 }
8668 CacheRecordDeferredAdd(m, negcr);
8669 }
8670 }
8671 m->rec.r.responseFlags = zeroID;
8672 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
8673 if (!repeat) break;
8674 repeat--;
8675 name = (const domainname *)(name->c + 1 + name->c[0]);
8676 hash = DomainNameHashValue(name);
8677 slot = HashSlot(name);
8678 // For now, we don't need to update cg here, because we'll do it again immediately, back up at the start of this loop
8679 //cg = CacheGroupForName(m, slot, hash, name);
8680 }
8681 }
8682 }
8683 }
8684 }
8685 if (NSECRecords) { LogInfo("mDNSCoreReceiveNoUnicastAnswers: NSECRecords not used"); FreeNSECRecords(m, NSECRecords); }
8686 if (SOARecord) { LogInfo("mDNSCoreReceiveNoUnicastAnswers: SOARecord not used"); ReleaseCacheRecord(m, SOARecord); }
8687 }
8688
8689 mDNSlocal void mDNSCorePrintStoredProxyRecords(mDNS *const m)
8690 {
8691 AuthRecord *rrPtr = mDNSNULL;
8692 LogSPS("Stored Proxy records :");
8693 for (rrPtr = m->SPSRRSet; rrPtr; rrPtr = rrPtr->next)
8694 {
8695 LogSPS("%s", ARDisplayString(m, rrPtr));
8696 }
8697 }
8698
8699 mDNSlocal mDNSBool mDNSCoreRegisteredProxyRecord(mDNS *const m, AuthRecord *rr)
8700 {
8701 AuthRecord *rrPtr = mDNSNULL;
8702
8703 for (rrPtr = m->SPSRRSet; rrPtr; rrPtr = rrPtr->next)
8704 {
8705 if (IdenticalResourceRecord(&rrPtr->resrec, &rr->resrec))
8706 {
8707 LogSPS("mDNSCoreRegisteredProxyRecord: Ignoring packet registered with sleep proxy : %s ", ARDisplayString(m, rr));
8708 return mDNStrue;
8709 }
8710 }
8711 mDNSCorePrintStoredProxyRecords(m);
8712 return mDNSfalse;
8713 }
8714
8715 mDNSlocal CacheRecord* mDNSCoreReceiveCacheCheck(mDNS *const m, const DNSMessage *const response, uDNS_LLQType LLQType,
8716 const mDNSu32 slot, CacheGroup *cg, DNSQuestion *unicastQuestion, CacheRecord ***cfp, CacheRecord **NSECCachePtr,
8717 mDNSInterfaceID InterfaceID)
8718 {
8719 CacheRecord *rr;
8720 CacheRecord **cflocal = *cfp;
8721
8722 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
8723 {
8724 mDNSBool match;
8725 // Resource record received via unicast, the resGroupID should match ?
8726 if (!InterfaceID)
8727 {
8728 mDNSu16 id1 = (rr->resrec.rDNSServer ? rr->resrec.rDNSServer->resGroupID : 0);
8729 mDNSu16 id2 = (m->rec.r.resrec.rDNSServer ? m->rec.r.resrec.rDNSServer->resGroupID : 0);
8730 match = (id1 == id2);
8731 }
8732 else
8733 match = (rr->resrec.InterfaceID == InterfaceID);
8734 // If we found this exact resource record, refresh its TTL
8735 if (match && IdenticalSameNameRecord(&m->rec.r.resrec, &rr->resrec))
8736 {
8737 if (m->rec.r.resrec.rdlength > InlineCacheRDSize)
8738 verbosedebugf("mDNSCoreReceiveCacheCheck: Found record size %5d interface %p already in cache: %s",
8739 m->rec.r.resrec.rdlength, InterfaceID, CRDisplayString(m, &m->rec.r));
8740
8741 if (m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask)
8742 {
8743 // If this packet record has the kDNSClass_UniqueRRSet flag set, then add it to our cache flushing list
8744 if (rr->NextInCFList == mDNSNULL && *cfp != &rr->NextInCFList && LLQType != uDNS_LLQ_Events)
8745 {
8746 *cflocal = rr;
8747 cflocal = &rr->NextInCFList;
8748 *cflocal = (CacheRecord*)1;
8749 *cfp = &rr->NextInCFList;
8750 }
8751
8752 // If this packet record is marked unique, and our previous cached copy was not, then fix it
8753 if (!(rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask))
8754 {
8755 DNSQuestion *q;
8756 for (q = m->Questions; q; q=q->next)
8757 {
8758 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
8759 q->UniqueAnswers++;
8760 }
8761 rr->resrec.RecordType = m->rec.r.resrec.RecordType;
8762 }
8763 }
8764
8765 if (!SameRDataBody(&m->rec.r.resrec, &rr->resrec.rdata->u, SameDomainNameCS))
8766 {
8767 // If the rdata of the packet record differs in name capitalization from the record in our cache
8768 // then mDNSPlatformMemSame will detect this. In this case, throw the old record away, so that clients get
8769 // a 'remove' event for the record with the old capitalization, and then an 'add' event for the new one.
8770 // <rdar://problem/4015377> mDNS -F returns the same domain multiple times with different casing
8771 rr->resrec.rroriginalttl = 0;
8772 rr->TimeRcvd = m->timenow;
8773 rr->UnansweredQueries = MaxUnansweredQueries;
8774 SetNextCacheCheckTimeForRecord(m, rr);
8775 LogInfo("mDNSCoreReceiveCacheCheck: Discarding due to domainname case change old: %s", CRDisplayString(m, rr));
8776 LogInfo("mDNSCoreReceiveCacheCheck: Discarding due to domainname case change new: %s", CRDisplayString(m, &m->rec.r));
8777 LogInfo("mDNSCoreReceiveCacheCheck: Discarding due to domainname case change in %d slot %3d in %d %d",
8778 NextCacheCheckEvent(rr) - m->timenow, slot, m->rrcache_nextcheck[slot] - m->timenow, m->NextCacheCheck - m->timenow);
8779 // DO NOT break out here -- we want to continue as if we never found it
8780 }
8781 else if (!IdenticalAnonInfo(m->rec.r.resrec.AnonInfo, rr->resrec.AnonInfo))
8782 {
8783 // If the NSEC3 record changed, a few possibilities
8784 //
8785 // 1) the peer reinitialized e.g., after network change and still part of the
8786 // same set.
8787 // 2) the peer went to a different set but we did not see the goodbyes. If we just
8788 // update the nsec3 record, it would be incorrect. Flush the cache so that we
8789 // can deliver a RMV followed by ADD.
8790 // 3) if the peer is ourselves and we see the goodbye when moving to a different set
8791 // and so we flush the cache and create a new cache record with the new set information.
8792 // Now we move back to the original set. In this case, we can't just update the
8793 // NSEC3 record alone. We need to flush so that we can deliver an RMV followed by ADD
8794 // when we create the new cache entry.
8795 //
8796 // Note: For case (1), we could avoid flushing the cache but we can't tell the difference
8797 // from the other cases.
8798 rr->resrec.rroriginalttl = 0;
8799 rr->TimeRcvd = m->timenow;
8800 rr->UnansweredQueries = MaxUnansweredQueries;
8801 SetNextCacheCheckTimeForRecord(m, rr);
8802 LogInfo("mDNSCoreReceiveCacheCheck: AnonInfo changed for %s", CRDisplayString(m, rr));
8803 // DO NOT break out here -- we want to continue as if we never found it. When we return
8804 // from this function, we will create a new cache entry with the new NSEC3 record
8805 }
8806 else if (m->rec.r.resrec.rroriginalttl > 0)
8807 {
8808 DNSQuestion *q;
8809
8810 m->mDNSStats.CacheRefreshed++;
8811
8812 if (rr->resrec.rroriginalttl == 0) debugf("uDNS rescuing %s", CRDisplayString(m, rr));
8813 RefreshCacheRecord(m, rr, m->rec.r.resrec.rroriginalttl);
8814 rr->responseFlags = response->h.flags;
8815
8816 // If we may have NSEC records returned with the answer (which we don't know yet as it
8817 // has not been processed), we need to cache them along with the first cache
8818 // record in the list that answers the question so that it can be used for validation
8819 // later. The "type" check below is to make sure that we cache on the cache record
8820 // that would answer the question. It is possible that we might cache additional things
8821 // e.g., MX question might cache A records also, and we want to cache the NSEC on
8822 // the record that answers the question.
8823 if (response->h.numAnswers && unicastQuestion && unicastQuestion->qtype == rr->resrec.rrtype
8824 && !(*NSECCachePtr))
8825 {
8826 LogInfo("mDNSCoreReceiveCacheCheck: rescuing RR %s", CRDisplayString(m, rr));
8827 *NSECCachePtr = rr;
8828 }
8829 // We have to reset the question interval to MaxQuestionInterval so that we don't keep
8830 // polling the network once we get a valid response back. For the first time when a new
8831 // cache entry is created, AnswerCurrentQuestionWithResourceRecord does that.
8832 // Subsequently, if we reissue questions from within the mDNSResponder e.g., DNS server
8833 // configuration changed, without flushing the cache, we reset the question interval here.
8834 // Currently, we do this for for both multicast and unicast questions as long as the record
8835 // type is unique. For unicast, resource record is always unique and for multicast it is
8836 // true for records like A etc. but not for PTR.
8837 if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask)
8838 {
8839 for (q = m->Questions; q; q=q->next)
8840 {
8841 if (!q->DuplicateOf && !q->LongLived &&
8842 ActiveQuestion(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
8843 {
8844 ResetQuestionState(m, q);
8845 debugf("mDNSCoreReceiveCacheCheck: Set MaxQuestionInterval for %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
8846 break; // Why break here? Aren't there other questions we might want to look at?-- SC July 2010
8847 }
8848 }
8849 }
8850 break;
8851 }
8852 else
8853 {
8854 // If the packet TTL is zero, that means we're deleting this record.
8855 // To give other hosts on the network a chance to protest, we push the deletion
8856 // out one second into the future. Also, we set UnansweredQueries to MaxUnansweredQueries.
8857 // Otherwise, we'll do final queries for this record at 80% and 90% of its apparent
8858 // lifetime (800ms and 900ms from now) which is a pointless waste of network bandwidth.
8859 // If record's current expiry time is more than a second from now, we set it to expire in one second.
8860 // If the record is already going to expire in less than one second anyway, we leave it alone --
8861 // we don't want to let the goodbye packet *extend* the record's lifetime in our cache.
8862 debugf("DE for %s", CRDisplayString(m, rr));
8863 if (RRExpireTime(rr) - m->timenow > mDNSPlatformOneSecond)
8864 {
8865 rr->resrec.rroriginalttl = 1;
8866 rr->TimeRcvd = m->timenow;
8867 rr->UnansweredQueries = MaxUnansweredQueries;
8868 SetNextCacheCheckTimeForRecord(m, rr);
8869 }
8870 break;
8871 }
8872 }
8873 }
8874 return rr;
8875 }
8876
8877 mDNSlocal void mDNSParseNSEC3Records(mDNS *const m, const DNSMessage *const response, const mDNSu8 *end,
8878 const mDNSInterfaceID InterfaceID, CacheRecord **NSEC3Records)
8879 {
8880 const mDNSu8 *ptr;
8881 CacheRecord *rr;
8882 int i;
8883
8884 if (!response->h.numAuthorities)
8885 return;
8886 ptr = LocateAuthorities(response, end);
8887 if (!ptr)
8888 {
8889 LogInfo("mDNSParseNSEC3Records: ERROR can't locate authorities");
8890 return;
8891 }
8892 for (i = 0; i < response->h.numAuthorities && ptr && ptr < end; i++)
8893 {
8894 mDNSu32 slot;
8895 CacheGroup *cg;
8896
8897 ptr = GetLargeResourceRecord(m, response, ptr, end, InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
8898 if (!ptr || m->rec.r.resrec.RecordType == kDNSRecordTypePacketNegative || m->rec.r.resrec.rrtype != kDNSType_NSEC3)
8899 {
8900 debugf("mDNSParseNSEC3Records: ptr %p, Record %s, ignoring", ptr, CRDisplayString(m, &m->rec.r));
8901 m->rec.r.resrec.RecordType = 0;
8902 continue;
8903 }
8904 slot = HashSlot(m->rec.r.resrec.name);
8905 cg = CacheGroupForRecord(m, slot, &m->rec.r.resrec);
8906 // Create the cache entry but don't add it to the cache it. We need
8907 // to cache this along with the main cache record.
8908 rr = CreateNewCacheEntry(m, slot, cg, 0, mDNSfalse, mDNSNULL);
8909 if (rr)
8910 {
8911 debugf("mDNSParseNSEC3Records: %s", CRDisplayString(m, rr));
8912 *NSEC3Records = rr;
8913 NSEC3Records = &rr->next;
8914 }
8915 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
8916 }
8917 }
8918
8919 mDNSlocal void mDNSCoreResetRecord(mDNS *const m)
8920 {
8921 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
8922 if (m->rec.r.resrec.AnonInfo)
8923 {
8924 FreeAnonInfo(m->rec.r.resrec.AnonInfo);
8925 m->rec.r.resrec.AnonInfo = mDNSNULL;
8926 }
8927 }
8928
8929 // Note: mDNSCoreReceiveResponse calls mDNS_Deregister_internal which can call a user callback, which may change
8930 // the record list and/or question list.
8931 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
8932 // InterfaceID non-NULL tells us the interface this multicast response was received on
8933 // InterfaceID NULL tells us this was a unicast response
8934 // dstaddr NULL tells us we received this over an outgoing TCP connection we made
8935 mDNSlocal void mDNSCoreReceiveResponse(mDNS *const m,
8936 const DNSMessage *const response, const mDNSu8 *end,
8937 const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
8938 const mDNSInterfaceID InterfaceID)
8939 {
8940 int i;
8941 mDNSBool ResponseMCast = dstaddr && mDNSAddrIsDNSMulticast(dstaddr);
8942 mDNSBool ResponseSrcLocal = !srcaddr || mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
8943 DNSQuestion *llqMatch = mDNSNULL;
8944 DNSQuestion *unicastQuestion = mDNSNULL;
8945 uDNS_LLQType LLQType = uDNS_recvLLQResponse(m, response, end, srcaddr, srcport, &llqMatch);
8946
8947 // "(CacheRecord*)1" is a special (non-zero) end-of-list marker
8948 // We use this non-zero marker so that records in our CacheFlushRecords list will always have NextInCFList
8949 // set non-zero, and that tells GetCacheEntity() that they're not, at this moment, eligible for recycling.
8950 CacheRecord *CacheFlushRecords = (CacheRecord*)1;
8951 CacheRecord **cfp = &CacheFlushRecords;
8952 CacheRecord *NSECRecords = mDNSNULL;
8953 CacheRecord *NSECCachePtr = mDNSNULL;
8954 CacheRecord **nsecp = &NSECRecords;
8955 CacheRecord *McastNSEC3Records = mDNSNULL;
8956 mDNSBool nseclist;
8957 mDNSu8 rcode = '\0';
8958 mDNSBool rrsigsCreated = mDNSfalse;
8959 mDNSBool DNSSECQuestion = mDNSfalse;
8960 NetworkInterfaceInfo *llintf = FirstIPv4LLInterfaceForID(m, InterfaceID);
8961 mDNSBool recordAcceptedInResponse = mDNSfalse; // Set if a record is accepted from a unicast mDNS response that answers an existing question.
8962
8963 // All records in a DNS response packet are treated as equally valid statements of truth. If we want
8964 // to guard against spoof responses, then the only credible protection against that is cryptographic
8965 // security, e.g. DNSSEC., not worrying about which section in the spoof packet contained the record.
8966 int firstauthority = response->h.numAnswers;
8967 int firstadditional = firstauthority + response->h.numAuthorities;
8968 int totalrecords = firstadditional + response->h.numAdditionals;
8969 const mDNSu8 *ptr = response->data;
8970 DNSServer *uDNSServer = mDNSNULL;
8971
8972 debugf("Received Response from %#-15a addressed to %#-15a on %p with "
8973 "%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes LLQType %d",
8974 srcaddr, dstaddr, InterfaceID,
8975 response->h.numQuestions, response->h.numQuestions == 1 ? ", " : "s,",
8976 response->h.numAnswers, response->h.numAnswers == 1 ? ", " : "s,",
8977 response->h.numAuthorities, response->h.numAuthorities == 1 ? "y, " : "ies,",
8978 response->h.numAdditionals, response->h.numAdditionals == 1 ? " " : "s", end - response->data, LLQType);
8979
8980 // According to RFC 2181 <http://www.ietf.org/rfc/rfc2181.txt>
8981 // When a DNS client receives a reply with TC
8982 // set, it should ignore that response, and query again, using a
8983 // mechanism, such as a TCP connection, that will permit larger replies.
8984 // It feels wrong to be throwing away data after the network went to all the trouble of delivering it to us, but
8985 // delivering some records of the RRSet first and then the remainder a couple of milliseconds later was causing
8986 // failures in our Microsoft Active Directory client, which expects to get the entire set of answers at once.
8987 // <rdar://problem/6690034> Can't bind to Active Directory
8988 // In addition, if the client immediately canceled its query after getting the initial partial response, then we'll
8989 // abort our TCP connection, and not complete the operation, and end up with an incomplete RRSet in our cache.
8990 // Next time there's a query for this RRSet we'll see answers in our cache, and assume we have the whole RRSet already,
8991 // and not even do the TCP query.
8992 // Accordingly, if we get a uDNS reply with kDNSFlag0_TC set, we bail out and wait for the TCP response containing the entire RRSet.
8993 if (!InterfaceID && (response->h.flags.b[0] & kDNSFlag0_TC)) return;
8994
8995 if (LLQType == uDNS_LLQ_Ignore) return;
8996
8997 // 1. We ignore questions (if any) in mDNS response packets
8998 // 2. If this is an LLQ response, we handle it much the same
8999 // 3. If we get a uDNS UDP response with the TC (truncated) bit set, then we can't treat this
9000 // answer as being the authoritative complete RRSet, and respond by deleting all other
9001 // matching cache records that don't appear in this packet.
9002 // Otherwise, this is a authoritative uDNS answer, so arrange for any stale records to be purged
9003 if (ResponseMCast || LLQType == uDNS_LLQ_Events || (response->h.flags.b[0] & kDNSFlag0_TC))
9004 ptr = LocateAnswers(response, end);
9005 // Otherwise, for one-shot queries, any answers in our cache that are not also contained
9006 // in this response packet are immediately deemed to be invalid.
9007 else
9008 {
9009 mDNSBool failure, returnEarly;
9010 rcode = (mDNSu8)(response->h.flags.b[1] & kDNSFlag1_RC_Mask);
9011 failure = !(rcode == kDNSFlag1_RC_NoErr || rcode == kDNSFlag1_RC_NXDomain || rcode == kDNSFlag1_RC_NotAuth);
9012 returnEarly = mDNSfalse;
9013 // We could possibly combine this with the similar loop at the end of this function --
9014 // instead of tagging cache records here and then rescuing them if we find them in the answer section,
9015 // we could instead use the "m->PktNum" mechanism to tag each cache record with the packet number in
9016 // which it was received (or refreshed), and then at the end if we find any cache records which
9017 // answer questions in this packet's question section, but which aren't tagged with this packet's
9018 // packet number, then we deduce they are old and delete them
9019 for (i = 0; i < response->h.numQuestions && ptr && ptr < end; i++)
9020 {
9021 DNSQuestion q, *qptr = mDNSNULL;
9022 ptr = getQuestion(response, ptr, end, InterfaceID, &q);
9023 if (ptr && (qptr = ExpectingUnicastResponseForQuestion(m, dstport, response->h.id, &q, !dstaddr)))
9024 {
9025 if (!failure)
9026 {
9027 CacheRecord *rr;
9028 // Remember the unicast question that we found, which we use to make caching
9029 // decisions later on in this function
9030 const mDNSu32 slot = HashSlot(&q.qname);
9031 CacheGroup *cg = CacheGroupForName(m, slot, q.qnamehash, &q.qname);
9032 if (!mDNSOpaque16IsZero(response->h.id))
9033 {
9034 unicastQuestion = qptr;
9035 if (qptr->qDNSServer && DNSSECQuestion(qptr))
9036 {
9037 LogInfo("mDNSCoreReceiveResponse: Setting aware for %##s (%s) on %#a", qptr->qname.c,
9038 DNSTypeName(qptr->qtype), &qptr->qDNSServer->addr);
9039 qptr->qDNSServer->DNSSECAware = mDNStrue;
9040 qptr->qDNSServer->req_DO = mDNStrue;
9041 }
9042 if (qptr->ValidatingResponse)
9043 DNSSECQuestion = mDNStrue;
9044 }
9045 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
9046 if (SameNameRecordAnswersQuestion(&rr->resrec, qptr))
9047 {
9048 debugf("uDNS marking %p %##s (%s) %p %s", q.InterfaceID, q.qname.c, DNSTypeName(q.qtype),
9049 rr->resrec.InterfaceID, CRDisplayString(m, rr));
9050 // Don't want to disturb rroriginalttl here, because code below might need it for the exponential backoff doubling algorithm
9051 rr->TimeRcvd = m->timenow - TicksTTL(rr) - 1;
9052 rr->UnansweredQueries = MaxUnansweredQueries;
9053 rr->CRDNSSECQuestion = 0;
9054 if (unicastQuestion && DNSSECQuestion(unicastQuestion))
9055 {
9056 LogInfo("mDNSCoreReceiveResponse: CRDNSSECQuestion set for record %s, question %##s (%s)", CRDisplayString(m, rr),
9057 unicastQuestion->qname.c, DNSTypeName(unicastQuestion->qtype));
9058 rr->CRDNSSECQuestion = 1;
9059 }
9060 }
9061 }
9062 else
9063 {
9064 if (qptr)
9065 {
9066 // If we recv any error from the DNSServer for a DNSSEC Query and if we know that the server
9067 // is not DNSSEC aware, stop doing DNSSEC for that DNSServer. Note that by setting the
9068 // req_DO to false here, the next retransmission for this question will turn off validation
9069 // and hence retransmit without the EDNS0/DOK option.
9070 if (DNSSECOptionalQuestion(qptr) && qptr->qDNSServer && !qptr->qDNSServer->DNSSECAware)
9071 {
9072 LogInfo("mDNSCoreReceiveResponse: Server %p responded with code %d to DNSSEC Query %##s (%s), clear DO flag",
9073 qptr->qDNSServer, rcode, q.qname.c, DNSTypeName(q.qtype));
9074 qptr->qDNSServer->req_DO = mDNSfalse;
9075 }
9076 // For Unicast DNS Queries, penalize the DNSServer
9077 else
9078 {
9079 LogInfo("mDNSCoreReceiveResponse: Server %p responded with code %d to query %##s (%s)",
9080 qptr->qDNSServer, rcode, q.qname.c, DNSTypeName(q.qtype));
9081 PenalizeDNSServer(m, qptr, response->h.flags);
9082 }
9083 }
9084 returnEarly = mDNStrue;
9085 }
9086 }
9087 }
9088 if (returnEarly)
9089 {
9090 LogInfo("Ignoring %2d Answer%s %2d Authorit%s %2d Additional%s",
9091 response->h.numAnswers, response->h.numAnswers == 1 ? ", " : "s,",
9092 response->h.numAuthorities, response->h.numAuthorities == 1 ? "y, " : "ies,",
9093 response->h.numAdditionals, response->h.numAdditionals == 1 ? "" : "s");
9094 // not goto exit because we won't have any CacheFlushRecords and we do not want to
9095 // generate negative cache entries (we want to query the next server)
9096 return;
9097 }
9098 if (unicastQuestion && DNSSECQuestion(unicastQuestion))
9099 {
9100 BumpDNSSECStats(m, kStatsActionSet, kStatsTypeMsgSize, (end - response->data));
9101 }
9102 }
9103
9104 // Parse the NSEC3 records from the Authority section before we process
9105 // the Answer section so that we can cache them along with the proper
9106 // cache records we create.
9107 if (mDNSOpaque16IsZero(response->h.id))
9108 mDNSParseNSEC3Records(m, response, end, InterfaceID, &McastNSEC3Records);
9109
9110 for (i = 0; i < totalrecords && ptr && ptr < end; i++)
9111 {
9112 // All responses sent via LL multicast are acceptable for caching
9113 // All responses received over our outbound TCP connections are acceptable for caching
9114 // We accept all records in a unicast response to a multicast query once we find one that
9115 // answers an active question.
9116 mDNSBool AcceptableResponse = ResponseMCast || !dstaddr || LLQType || recordAcceptedInResponse;
9117 // (Note that just because we are willing to cache something, that doesn't necessarily make it a trustworthy answer
9118 // to any specific question -- any code reading records from the cache needs to make that determination for itself.)
9119
9120 const mDNSu8 RecordType =
9121 (i < firstauthority ) ? (mDNSu8)kDNSRecordTypePacketAns :
9122 (i < firstadditional) ? (mDNSu8)kDNSRecordTypePacketAuth : (mDNSu8)kDNSRecordTypePacketAdd;
9123 ptr = GetLargeResourceRecord(m, response, ptr, end, InterfaceID, RecordType, &m->rec);
9124 if (!ptr) goto exit; // Break out of the loop and clean up our CacheFlushRecords list before exiting
9125
9126 if (m->rec.r.resrec.RecordType == kDNSRecordTypePacketNegative)
9127 {
9128 mDNSCoreResetRecord(m);
9129 continue;
9130 }
9131
9132 // We have already parsed the NSEC3 records and cached them approrpriately for
9133 // multicast responses.
9134 if (mDNSOpaque16IsZero(response->h.id) && m->rec.r.resrec.rrtype == kDNSType_NSEC3)
9135 {
9136 mDNSCoreResetRecord(m);
9137 continue;
9138 }
9139 // Don't want to cache OPT or TSIG pseudo-RRs
9140 if (m->rec.r.resrec.rrtype == kDNSType_TSIG)
9141 {
9142 mDNSCoreResetRecord(m);
9143 continue;
9144 }
9145 if (m->rec.r.resrec.rrtype == kDNSType_OPT)
9146 {
9147 const rdataOPT *opt;
9148 const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
9149 // Find owner sub-option(s). We verify that the MAC is non-zero, otherwise we could inadvertently
9150 // delete all our own AuthRecords (which are identified by having zero MAC tags on them).
9151 for (opt = &m->rec.r.resrec.rdata->u.opt[0]; opt < e; opt++)
9152 if (opt->opt == kDNSOpt_Owner && opt->u.owner.vers == 0 && opt->u.owner.HMAC.l[0])
9153 {
9154 ClearProxyRecords(m, &opt->u.owner, m->DuplicateRecords);
9155 ClearProxyRecords(m, &opt->u.owner, m->ResourceRecords);
9156 }
9157 mDNSCoreResetRecord(m);
9158 continue;
9159 }
9160 // if a CNAME record points to itself, then don't add it to the cache
9161 if ((m->rec.r.resrec.rrtype == kDNSType_CNAME) && SameDomainName(m->rec.r.resrec.name, &m->rec.r.resrec.rdata->u.name))
9162 {
9163 LogInfo("mDNSCoreReceiveResponse: CNAME loop domain name %##s", m->rec.r.resrec.name->c);
9164 mDNSCoreResetRecord(m);
9165 continue;
9166 }
9167
9168 // When we receive uDNS LLQ responses, we assume a long cache lifetime --
9169 // In the case of active LLQs, we'll get remove events when the records actually do go away
9170 // In the case of polling LLQs, we assume the record remains valid until the next poll
9171 if (!mDNSOpaque16IsZero(response->h.id))
9172 m->rec.r.resrec.rroriginalttl = GetEffectiveTTL(LLQType, m->rec.r.resrec.rroriginalttl);
9173
9174 // If response was not sent via LL multicast,
9175 // then see if it answers a recent query of ours, which would also make it acceptable for caching.
9176 if (!ResponseMCast)
9177 {
9178 if (LLQType)
9179 {
9180 // For Long Lived queries that are both sent over UDP and Private TCP, LLQType is set.
9181 // Even though it is AcceptableResponse, we need a matching DNSServer pointer for the
9182 // queries to get ADD/RMV events. To lookup the question, we can't use
9183 // ExpectingUnicastResponseForRecord as the port numbers don't match. uDNS_recvLLQRespose
9184 // has already matched the question using the 64 bit Id in the packet and we use that here.
9185
9186 if (llqMatch != mDNSNULL) m->rec.r.resrec.rDNSServer = uDNSServer = llqMatch->qDNSServer;
9187
9188 // If this is a DNSSEC question that is also LongLived, don't accept records from the
9189 // Additional/Authority section blindly. We need to go through IsAcceptableResponse below
9190 // so that NSEC/NSEC3 record are cached in the nseclist if we accept them. This can happen
9191 // for both negative responses and wildcard expanded positive responses as both of come
9192 // back with NSEC/NSEC3s.
9193 if (unicastQuestion && DNSSECQuestion(unicastQuestion))
9194 AcceptableResponse = mDNSfalse;
9195 }
9196 else if (!AcceptableResponse || !dstaddr)
9197 {
9198 // For responses that come over TCP (Responses that can't fit within UDP) or TLS (Private queries
9199 // that are not long lived e.g., AAAA lookup in a Private domain), it is indicated by !dstaddr.
9200 // Even though it is AcceptableResponse, we still need a DNSServer pointer for the resource records that
9201 // we create.
9202
9203 DNSQuestion *q = ExpectingUnicastResponseForRecord(m, srcaddr, ResponseSrcLocal, dstport, response->h.id, &m->rec.r, !dstaddr);
9204
9205 // Initialize the DNS server on the resource record which will now filter what questions we answer with
9206 // this record.
9207 //
9208 // We could potentially lookup the DNS server based on the source address, but that may not work always
9209 // and that's why ExpectingUnicastResponseForRecord does not try to verify whether the response came
9210 // from the DNS server that queried. We follow the same logic here. If we can find a matching quetion based
9211 // on the "id" and "source port", then this response answers the question and assume the response
9212 // came from the same DNS server that we sent the query to.
9213
9214 if (q != mDNSNULL)
9215 {
9216 AcceptableResponse = mDNStrue;
9217 if (!InterfaceID)
9218 {
9219 debugf("mDNSCoreReceiveResponse: InterfaceID %p %##s (%s)", q->InterfaceID, q->qname.c, DNSTypeName(q->qtype));
9220 m->rec.r.resrec.rDNSServer = uDNSServer = q->qDNSServer;
9221 }
9222 else
9223 {
9224 // Accept all remaining records in this unicast response to an mDNS query.
9225 recordAcceptedInResponse = mDNStrue;
9226 LogInfo("mDNSCoreReceiveResponse: Accepting response for query: %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
9227 }
9228 }
9229 else
9230 {
9231 // If we can't find a matching question, we need to see whether we have seen records earlier that matched
9232 // the question. The code below does that. So, make this record unacceptable for now
9233 if (!InterfaceID)
9234 {
9235 debugf("mDNSCoreReceiveResponse: Can't find question for record name %##s", m->rec.r.resrec.name->c);
9236 AcceptableResponse = mDNSfalse;
9237 }
9238 }
9239 }
9240 }
9241 else if (llintf && llintf->IgnoreIPv4LL && m->rec.r.resrec.rrtype == kDNSType_A)
9242 {
9243 // There are some routers (rare, thankfully) that generate bogus ARP responses for
9244 // any IPv4 address they don’t recognize, including RFC 3927 IPv4 link-local addresses.
9245 // To work with these broken routers, client devices need to blacklist these broken
9246 // routers and ignore their bogus ARP responses. Some devices implement a technique
9247 // such as the one described in US Patent 7436783, which lets clients detect and
9248 // ignore these broken routers: <https://www.google.com/patents/US7436783>
9249
9250 // OS X and iOS do not implement this defensive mechanism, instead taking a simpler
9251 // approach of just detecting these broken routers and completely disabling IPv4
9252 // link-local communication on interfaces where a broken router is detected.
9253 // OS X and iOS set the IFEF_ARPLL interface flag on interfaces
9254 // that are deemed “safe” for IPv4 link-local communication;
9255 // the flag is cleared on interfaces where a broken router is detected.
9256
9257 // OS X and iOS will not even try to communicate with an IPv4
9258 // link-local destination on an interface without the IFEF_ARPLL flag set.
9259 // This can cause some badly written applications to freeze for a long time if they
9260 // attempt to connect to an IPv4 link-local destination address and then wait for
9261 // that connection attempt to time out before trying other candidate addresses.
9262
9263 // To mask this client bug, we suppress acceptance of IPv4 link-local address
9264 // records on interfaces where we know the OS will be unwilling even to attempt
9265 // communication with those IPv4 link-local destination addresses.
9266 // <rdar://problem/9400639> kSuppress IPv4LL answers on interfaces without IFEF_ARPLL
9267
9268 const CacheRecord *const rr = &m->rec.r;
9269 const RDataBody2 *const rdb = (RDataBody2 *)rr->smallrdatastorage.data;
9270 if (mDNSv4AddressIsLinkLocal(&rdb->ipv4))
9271 {
9272 LogInfo("mDNSResponder: Dropping LinkLocal packet %s", CRDisplayString(m, &m->rec.r));
9273 mDNSCoreResetRecord(m);
9274 continue;
9275 }
9276 }
9277
9278 // 1. Check that this packet resource record does not conflict with any of ours
9279 if (mDNSOpaque16IsZero(response->h.id) && m->rec.r.resrec.rrtype != kDNSType_NSEC)
9280 {
9281 if (m->CurrentRecord)
9282 LogMsg("mDNSCoreReceiveResponse ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
9283 m->CurrentRecord = m->ResourceRecords;
9284 while (m->CurrentRecord)
9285 {
9286 AuthRecord *rr = m->CurrentRecord;
9287 m->CurrentRecord = rr->next;
9288 // We accept all multicast responses, and unicast responses resulting from queries we issued
9289 // For other unicast responses, this code accepts them only for responses with an
9290 // (apparently) local source address that pertain to a record of our own that's in probing state
9291 if (!AcceptableResponse && !(ResponseSrcLocal && rr->resrec.RecordType == kDNSRecordTypeUnique)) continue;
9292
9293 if (PacketRRMatchesSignature(&m->rec.r, rr)) // If interface, name, type (if shared record) and class match...
9294 {
9295 // ... check to see if type and rdata are identical
9296 if (IdenticalSameNameRecord(&m->rec.r.resrec, &rr->resrec))
9297 {
9298 // If the RR in the packet is identical to ours, just check they're not trying to lower the TTL on us
9299 if (m->rec.r.resrec.rroriginalttl >= rr->resrec.rroriginalttl/2 || m->SleepState)
9300 {
9301 // If we were planning to send on this -- and only this -- interface, then we don't need to any more
9302 if (rr->ImmedAnswer == InterfaceID) { rr->ImmedAnswer = mDNSNULL; rr->ImmedUnicast = mDNSfalse; }
9303 }
9304 else
9305 {
9306 if (rr->ImmedAnswer == mDNSNULL) { rr->ImmedAnswer = InterfaceID; m->NextScheduledResponse = m->timenow; }
9307 else if (rr->ImmedAnswer != InterfaceID) { rr->ImmedAnswer = mDNSInterfaceMark; m->NextScheduledResponse = m->timenow; }
9308 }
9309 }
9310 // else, the packet RR has different type or different rdata -- check to see if this is a conflict
9311 else if (m->rec.r.resrec.rroriginalttl > 0 && PacketRRConflict(m, rr, &m->rec.r))
9312 {
9313 LogInfo("mDNSCoreReceiveResponse: Pkt Record: %08lX %s", m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
9314 LogInfo("mDNSCoreReceiveResponse: Our Record: %08lX %s", rr->resrec.rdatahash, ARDisplayString(m, rr));
9315
9316 // If this record is marked DependentOn another record for conflict detection purposes,
9317 // then *that* record has to be bumped back to probing state to resolve the conflict
9318 if (rr->DependentOn)
9319 {
9320 while (rr->DependentOn) rr = rr->DependentOn;
9321 LogInfo("mDNSCoreReceiveResponse: Dep Record: %08lX %s", rr->resrec.rdatahash, ARDisplayString(m, rr));
9322 }
9323
9324 // If we've just whacked this record's ProbeCount, don't need to do it again
9325 if (rr->ProbeCount > DefaultProbeCountForTypeUnique)
9326 LogInfo("mDNSCoreReceiveResponse: Already reset to Probing: %s", ARDisplayString(m, rr));
9327 else if (rr->ProbeCount == DefaultProbeCountForTypeUnique)
9328 LogInfo("mDNSCoreReceiveResponse: Ignoring response received before we even began probing: %s", ARDisplayString(m, rr));
9329 else
9330 {
9331 LogMsg("mDNSCoreReceiveResponse: Received from %#a:%d %s", srcaddr, mDNSVal16(srcport), CRDisplayString(m, &m->rec.r));
9332 // If we'd previously verified this record, put it back to probing state and try again
9333 if (rr->resrec.RecordType == kDNSRecordTypeVerified)
9334 {
9335 LogMsg("mDNSCoreReceiveResponse: Resetting to Probing: %s", ARDisplayString(m, rr));
9336 rr->resrec.RecordType = kDNSRecordTypeUnique;
9337 // We set ProbeCount to one more than the usual value so we know we've already touched this record.
9338 // This is because our single probe for "example-name.local" could yield a response with (say) two A records and
9339 // three AAAA records in it, and we don't want to call RecordProbeFailure() five times and count that as five conflicts.
9340 // This special value is recognised and reset to DefaultProbeCountForTypeUnique in SendQueries().
9341 rr->ProbeCount = DefaultProbeCountForTypeUnique + 1;
9342 rr->AnnounceCount = InitialAnnounceCount;
9343 InitializeLastAPTime(m, rr);
9344 RecordProbeFailure(m, rr); // Repeated late conflicts also cause us to back off to the slower probing rate
9345 }
9346 // If we're probing for this record, we just failed
9347 else if (rr->resrec.RecordType == kDNSRecordTypeUnique)
9348 {
9349 // At this point in the code, we're probing for uniqueness.
9350 // We've sent at least one probe (rr->ProbeCount < DefaultProbeCountForTypeUnique)
9351 // but we haven't completed probing yet (rr->resrec.RecordType == kDNSRecordTypeUnique).
9352 // Before we call deregister, check if this is a packet we registered with the sleep proxy.
9353 if (!mDNSCoreRegisteredProxyRecord(m, rr))
9354 {
9355 // This may be a conflict due to stale packets on the network. Delay probing by a second.
9356 // If there are conflicts after 3 such attempts, then it is a true conflict.
9357 if (m->DelayConflictProcessing)
9358 {
9359 m->DelayConflictProcessing--;
9360 LogMsg("Possible spurious conflict for %s. Attempt %d at suppressing probes for one second",
9361 ARDisplayString(m, rr), (MAX_CONFLICT_PROCESSING_DELAYS - m->DelayConflictProcessing));
9362 rr->ProbeCount = DefaultProbeCountForTypeUnique + 1;
9363 rr->AnnounceCount = InitialAnnounceCount;
9364 m->SuppressProbes = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
9365 InitializeLastAPTime(m, rr);
9366 RecordProbeFailure(m, rr); // Repeated late conflicts also cause us to back off to the slower probing rate
9367 }
9368 else
9369 {
9370 LogMsg("mDNSCoreReceiveResponse: ProbeCount %d; will deregister %s", rr->ProbeCount, ARDisplayString(m, rr));
9371 m->mDNSStats.NameConflicts++;
9372 #if APPLE_OSX_mDNSResponder
9373 // See if this record was also registered with any D2D plugins.
9374 D2D_stop_advertising_record(rr);
9375 #endif
9376 mDNS_Deregister_internal(m, rr, mDNS_Dereg_conflict);
9377 }
9378
9379 }
9380 }
9381 // We assumed this record must be unique, but we were wrong. (e.g. There are two mDNSResponders on the
9382 // same machine giving different answers for the reverse mapping record, or there are two machines on the
9383 // network using the same IP address.) This is simply a misconfiguration, and there's nothing we can do
9384 // to fix it -- e.g. it's not our job to be trying to change the machine's IP address. We just discard our
9385 // record to avoid continued conflicts (as we do for a conflict on our Unique records) and get on with life.
9386 else if (rr->resrec.RecordType == kDNSRecordTypeKnownUnique)
9387 {
9388 LogMsg("mDNSCoreReceiveResponse: Unexpected conflict discarding %s", ARDisplayString(m, rr));
9389 m->mDNSStats.KnownUniqueNameConflicts++;
9390 #if APPLE_OSX_mDNSResponder
9391 D2D_stop_advertising_record(rr);
9392 #endif
9393 mDNS_Deregister_internal(m, rr, mDNS_Dereg_conflict);
9394 }
9395 else
9396 LogMsg("mDNSCoreReceiveResponse: Unexpected record type %X %s", rr->resrec.RecordType, ARDisplayString(m, rr));
9397 }
9398 }
9399 // Else, matching signature, different type or rdata, but not a considered a conflict.
9400 // If the packet record has the cache-flush bit set, then we check to see if we
9401 // have any record(s) of the same type that we should re-assert to rescue them
9402 // (see note about "multi-homing and bridged networks" at the end of this function).
9403 else if (m->rec.r.resrec.rrtype == rr->resrec.rrtype)
9404 if ((m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask) && m->timenow - rr->LastMCTime > mDNSPlatformOneSecond/2)
9405 { rr->ImmedAnswer = mDNSInterfaceMark; m->NextScheduledResponse = m->timenow; }
9406 }
9407 }
9408 }
9409
9410 nseclist = mDNSfalse;
9411 if (!AcceptableResponse)
9412 {
9413 AcceptableResponse = IsResponseAcceptable(m, CacheFlushRecords, unicastQuestion, &nseclist);
9414 if (AcceptableResponse) m->rec.r.resrec.rDNSServer = uDNSServer;
9415 }
9416
9417 // 2. See if we want to add this packet resource record to our cache
9418 // We only try to cache answers if we have a cache to put them in
9419 // Also, we ignore any apparent attempts at cache poisoning unicast to us that do not answer any outstanding active query
9420 if (!AcceptableResponse) LogInfo("mDNSCoreReceiveResponse ignoring %s", CRDisplayString(m, &m->rec.r));
9421 if (m->rrcache_size && AcceptableResponse)
9422 {
9423 const mDNSu32 slot = HashSlot(m->rec.r.resrec.name);
9424 CacheGroup *cg = CacheGroupForRecord(m, slot, &m->rec.r.resrec);
9425 CacheRecord *rr = mDNSNULL;
9426
9427 if (McastNSEC3Records)
9428 InitializeAnonInfoForCR(m, &McastNSEC3Records, &m->rec.r);
9429
9430 // 2a. Check if this packet resource record is already in our cache.
9431 //
9432 // If this record should go in the nseclist, don't look in the cache for updating it.
9433 // They are supposed to be cached under the "nsec" field of the cache record for
9434 // validation. Just create the cache record.
9435 if (!nseclist)
9436 {
9437 rr = mDNSCoreReceiveCacheCheck(m, response, LLQType, slot, cg, unicastQuestion, &cfp, &NSECCachePtr, InterfaceID);
9438 }
9439
9440 // If packet resource record not in our cache, add it now
9441 // (unless it is just a deletion of a record we never had, in which case we don't care)
9442 if (!rr && m->rec.r.resrec.rroriginalttl > 0)
9443 {
9444 const mDNSBool AddToCFList = (m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask) && (LLQType != uDNS_LLQ_Events);
9445 mDNSs32 delay;
9446
9447 if (AddToCFList)
9448 delay = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
9449 else
9450 delay = CheckForSoonToExpireRecords(m, m->rec.r.resrec.name, m->rec.r.resrec.namehash, slot, mDNSNULL);
9451
9452 // If unique, assume we may have to delay delivery of this 'add' event.
9453 // Below, where we walk the CacheFlushRecords list, we either call CacheRecordDeferredAdd()
9454 // to immediately to generate answer callbacks, or we call ScheduleNextCacheCheckTime()
9455 // to schedule an mDNS_Execute task at the appropriate time.
9456 rr = CreateNewCacheEntry(m, slot, cg, delay, !nseclist, srcaddr);
9457 if (rr)
9458 {
9459 rr->responseFlags = response->h.flags;
9460 // If we are not creating signatures, then we need to inform DNSSEC so that
9461 // it does not wait forever. Don't do this if we got NSEC records
9462 // as it indicates that this name does not exist.
9463 if (rr->resrec.rrtype == kDNSType_RRSIG && !nseclist)
9464 {
9465 rrsigsCreated = mDNStrue;
9466 }
9467 // Remember whether we created a cache record in response to a DNSSEC question.
9468 // This helps DNSSEC code not to reissue the question to fetch the DNSSEC records.
9469 rr->CRDNSSECQuestion = 0;
9470 if (unicastQuestion && DNSSECQuestion(unicastQuestion))
9471 {
9472 LogInfo("mDNSCoreReceiveResponse: CRDNSSECQuestion set for new record %s, question %##s (%s)", CRDisplayString(m, rr),
9473 unicastQuestion->qname.c, DNSTypeName(unicastQuestion->qtype));
9474 rr->CRDNSSECQuestion = 1;
9475 }
9476 // NSEC/NSEC3 records and its signatures are cached with the negative cache entry
9477 // which we should be creating below. It is also needed in the wildcard
9478 // expanded answer case and in that case it is cached along with the answer.
9479 if (nseclist)
9480 {
9481 rr->TimeRcvd = m->timenow;
9482 *nsecp = rr;
9483 nsecp = &rr->next;
9484 }
9485 else if (AddToCFList)
9486 {
9487 *cfp = rr;
9488 cfp = &rr->NextInCFList;
9489 *cfp = (CacheRecord*)1;
9490 }
9491 else if (rr->DelayDelivery)
9492 {
9493 ScheduleNextCacheCheckTime(m, slot, rr->DelayDelivery);
9494 }
9495 }
9496 }
9497 else
9498 {
9499 if (rr && rr->resrec.AnonInfo && m->rec.r.resrec.AnonInfo)
9500 {
9501 CopyAnonInfoForCR(m, rr, &m->rec.r);
9502 }
9503 }
9504 }
9505 mDNSCoreResetRecord(m);
9506 }
9507
9508 exit:
9509 mDNSCoreResetRecord(m);
9510
9511 // If we've just received one or more records with their cache flush bits set,
9512 // then scan that cache slot to see if there are any old stale records we need to flush
9513 while (CacheFlushRecords != (CacheRecord*)1)
9514 {
9515 CacheRecord *r1 = CacheFlushRecords, *r2;
9516 const mDNSu32 slot = HashSlot(r1->resrec.name);
9517 const CacheGroup *cg = CacheGroupForRecord(m, slot, &r1->resrec);
9518 CacheFlushRecords = CacheFlushRecords->NextInCFList;
9519 r1->NextInCFList = mDNSNULL;
9520
9521 // Look for records in the cache with the same signature as this new one with the cache flush
9522 // bit set, and either (a) if they're fresh, just make sure the whole RRSet has the same TTL
9523 // (as required by DNS semantics) or (b) if they're old, mark them for deletion in one second.
9524 // We make these TTL adjustments *only* for records that still have *more* than one second
9525 // remaining to live. Otherwise, a record that we tagged for deletion half a second ago
9526 // (and now has half a second remaining) could inadvertently get its life extended, by either
9527 // (a) if we got an explicit goodbye packet half a second ago, the record would be considered
9528 // "fresh" and would be incorrectly resurrected back to the same TTL as the rest of the RRSet,
9529 // or (b) otherwise, the record would not be fully resurrected, but would be reset to expire
9530 // in one second, thereby inadvertently delaying its actual expiration, instead of hastening it.
9531 // If this were to happen repeatedly, the record's expiration could be deferred indefinitely.
9532 // To avoid this, we need to ensure that the cache flushing operation will only act to
9533 // *decrease* a record's remaining lifetime, never *increase* it.
9534 for (r2 = cg ? cg->members : mDNSNULL; r2; r2=r2->next)
9535 {
9536 mDNSu16 id1;
9537 mDNSu16 id2;
9538 if (!r1->resrec.InterfaceID)
9539 {
9540 id1 = (r1->resrec.rDNSServer ? r1->resrec.rDNSServer->resGroupID : 0);
9541 id2 = (r2->resrec.rDNSServer ? r2->resrec.rDNSServer->resGroupID : 0);
9542 }
9543 else
9544 {
9545 id1 = id2 = 0;
9546 }
9547 // When we receive new RRSIGs e.g., for DNSKEY record, we should not flush the old
9548 // RRSIGS e.g., for TXT record. To do so, we need to look at the typeCovered field of
9549 // the new RRSIG that we received. Process only if the typeCovered matches.
9550 if ((r1->resrec.rrtype == r2->resrec.rrtype) && (r1->resrec.rrtype == kDNSType_RRSIG))
9551 {
9552 rdataRRSig *rrsig1 = (rdataRRSig *)(((RDataBody2 *)(r1->resrec.rdata->u.data))->data);
9553 rdataRRSig *rrsig2 = (rdataRRSig *)(((RDataBody2 *)(r2->resrec.rdata->u.data))->data);
9554 if (swap16(rrsig1->typeCovered) != swap16(rrsig2->typeCovered))
9555 {
9556 debugf("mDNSCoreReceiveResponse: Received RRSIG typeCovered %s, found %s, not processing",
9557 DNSTypeName(swap16(rrsig1->typeCovered)), DNSTypeName(swap16(rrsig2->typeCovered)));
9558 continue;
9559 }
9560 }
9561
9562 // For Unicast (null InterfaceID) the resolver IDs should also match
9563 if ((r1->resrec.InterfaceID == r2->resrec.InterfaceID) &&
9564 (r1->resrec.InterfaceID || (id1 == id2)) &&
9565 r1->resrec.rrtype == r2->resrec.rrtype &&
9566 r1->resrec.rrclass == r2->resrec.rrclass)
9567 {
9568 // If record is recent, just ensure the whole RRSet has the same TTL (as required by DNS semantics)
9569 // else, if record is old, mark it to be flushed
9570 if (m->timenow - r2->TimeRcvd < mDNSPlatformOneSecond && RRExpireTime(r2) - m->timenow > mDNSPlatformOneSecond)
9571 {
9572 // If we find mismatched TTLs in an RRSet, correct them.
9573 // We only do this for records with a TTL of 2 or higher. It's possible to have a
9574 // goodbye announcement with the cache flush bit set (or a case-change on record rdata,
9575 // which we treat as a goodbye followed by an addition) and in that case it would be
9576 // inappropriate to synchronize all the other records to a TTL of 0 (or 1).
9577
9578 // We suppress the message for the specific case of correcting from 240 to 60 for type TXT,
9579 // because certain early Bonjour devices are known to have this specific mismatch, and
9580 // there's no point filling syslog with messages about something we already know about.
9581 // We also don't log this for uDNS responses, since a caching name server is obliged
9582 // to give us an aged TTL to correct for how long it has held the record,
9583 // so our received TTLs are expected to vary in that case
9584
9585 // We also suppress log message in the case of SRV records that are received
9586 // with a TTL of 4500 that are already cached with a TTL of 120 seconds, since
9587 // this behavior was observed for a number of discoveryd based AppleTV's in iOS 8
9588 // GM builds.
9589 if (r2->resrec.rroriginalttl != r1->resrec.rroriginalttl && r1->resrec.rroriginalttl > 1)
9590 {
9591 if (!(r2->resrec.rroriginalttl == 240 && r1->resrec.rroriginalttl == 60 && r2->resrec.rrtype == kDNSType_TXT) &&
9592 !(r2->resrec.rroriginalttl == 120 && r1->resrec.rroriginalttl == 4500 && r2->resrec.rrtype == kDNSType_SRV) &&
9593 mDNSOpaque16IsZero(response->h.id))
9594 LogInfo("Correcting TTL from %4d to %4d for %s",
9595 r2->resrec.rroriginalttl, r1->resrec.rroriginalttl, CRDisplayString(m, r2));
9596 r2->resrec.rroriginalttl = r1->resrec.rroriginalttl;
9597 }
9598 r2->TimeRcvd = m->timenow;
9599 }
9600 else // else, if record is old, mark it to be flushed
9601 {
9602 verbosedebugf("Cache flush new %p age %d expire in %d %s", r1, m->timenow - r1->TimeRcvd, RRExpireTime(r1) - m->timenow, CRDisplayString(m, r1));
9603 verbosedebugf("Cache flush old %p age %d expire in %d %s", r2, m->timenow - r2->TimeRcvd, RRExpireTime(r2) - m->timenow, CRDisplayString(m, r2));
9604 // We set stale records to expire in one second.
9605 // This gives the owner a chance to rescue it if necessary.
9606 // This is important in the case of multi-homing and bridged networks:
9607 // Suppose host X is on Ethernet. X then connects to an AirPort base station, which happens to be
9608 // bridged onto the same Ethernet. When X announces its AirPort IP address with the cache-flush bit
9609 // set, the AirPort packet will be bridged onto the Ethernet, and all other hosts on the Ethernet
9610 // will promptly delete their cached copies of the (still valid) Ethernet IP address record.
9611 // By delaying the deletion by one second, we give X a change to notice that this bridging has
9612 // happened, and re-announce its Ethernet IP address to rescue it from deletion from all our caches.
9613
9614 // We set UnansweredQueries to MaxUnansweredQueries to avoid expensive and unnecessary
9615 // final expiration queries for this record.
9616
9617 // If a record is deleted twice, first with an explicit DE record, then a second time by virtue of the cache
9618 // flush bit on the new record replacing it, then we allow the record to be deleted immediately, without the usual
9619 // one-second grace period. This improves responsiveness for mDNS_Update(), as used for things like iChat status updates.
9620 // <rdar://problem/5636422> Updating TXT records is too slow
9621 // We check for "rroriginalttl == 1" because we want to include records tagged by the "packet TTL is zero" check above,
9622 // which sets rroriginalttl to 1, but not records tagged by the rdata case-change check, which sets rroriginalttl to 0.
9623 if (r2->TimeRcvd == m->timenow && r2->resrec.rroriginalttl == 1 && r2->UnansweredQueries == MaxUnansweredQueries)
9624 {
9625 LogInfo("Cache flush for DE record %s", CRDisplayString(m, r2));
9626 r2->resrec.rroriginalttl = 0;
9627 }
9628 else if (RRExpireTime(r2) - m->timenow > mDNSPlatformOneSecond)
9629 {
9630 // We only set a record to expire in one second if it currently has *more* than a second to live
9631 // If it's already due to expire in a second or less, we just leave it alone
9632 r2->resrec.rroriginalttl = 1;
9633 r2->UnansweredQueries = MaxUnansweredQueries;
9634 r2->TimeRcvd = m->timenow - 1;
9635 // We use (m->timenow - 1) instead of m->timenow, because we use that to identify records
9636 // that we marked for deletion via an explicit DE record
9637 }
9638 }
9639 SetNextCacheCheckTimeForRecord(m, r2);
9640 }
9641 }
9642
9643 if (r1->DelayDelivery) // If we were planning to delay delivery of this record, see if we still need to
9644 {
9645 // If we had a unicast question for this response with at least one positive answer and we
9646 // have NSECRecords, it is most likely a wildcard expanded answer. Cache the NSEC and its
9647 // signatures along with the cache record which will be used for validation later. If
9648 // we rescued a few records earlier in this function, then NSECCachePtr would be set. In that
9649 // use that instead.
9650 if (response->h.numAnswers && unicastQuestion && NSECRecords)
9651 {
9652 if (!NSECCachePtr)
9653 {
9654 LogInfo("mDNSCoreReceiveResponse: Updating NSECCachePtr to %s", CRDisplayString(m, r1));
9655 NSECCachePtr = r1;
9656 }
9657 // Note: We need to do this before we call CacheRecordDeferredAdd as this
9658 // might start the verification process which needs these NSEC records
9659 if (!AddNSECSForCacheRecord(m, NSECRecords, NSECCachePtr, rcode))
9660 {
9661 LogInfo("mDNSCoreReceiveResponse: AddNSECSForCacheRecord failed to add NSEC for %s", CRDisplayString(m, NSECCachePtr));
9662 FreeNSECRecords(m, NSECRecords);
9663 }
9664 NSECRecords = mDNSNULL;
9665 NSECCachePtr = mDNSNULL;
9666 }
9667 r1->DelayDelivery = CheckForSoonToExpireRecords(m, r1->resrec.name, r1->resrec.namehash, slot, mDNSNULL);
9668 // If no longer delaying, deliver answer now, else schedule delivery for the appropriate time
9669 if (!r1->DelayDelivery) CacheRecordDeferredAdd(m, r1);
9670 else ScheduleNextCacheCheckTime(m, slot, r1->DelayDelivery);
9671 }
9672 }
9673
9674 // If we have not consumed the NSEC records yet e.g., just refreshing the cache,
9675 // update them now for future validations.
9676 if (NSECRecords && NSECCachePtr)
9677 {
9678 LogInfo("mDNSCoreReceieveResponse: Updating NSEC records in %s", CRDisplayString(m, NSECCachePtr));
9679 if (!AddNSECSForCacheRecord(m, NSECRecords, NSECCachePtr, rcode))
9680 {
9681 LogInfo("mDNSCoreReceiveResponse: AddNSECSForCacheRecord failed to add NSEC for %s", CRDisplayString(m, NSECCachePtr));
9682 FreeNSECRecords(m, NSECRecords);
9683 }
9684 NSECRecords = mDNSNULL;
9685 NSECCachePtr = mDNSNULL;
9686 }
9687
9688 // If there is at least one answer and we did not create RRSIGs and there was a
9689 // ValidatingResponse question waiting for this response, give a hint that no RRSIGs
9690 // were created. We don't need to give a hint:
9691 //
9692 // - if we have no answers, the mDNSCoreReceiveNoUnicastAnswers below should
9693 // generate a negative response
9694 //
9695 // - if we have NSECRecords, it means we might have a potential proof for
9696 // non-existence of name that we are looking for
9697 //
9698 if (response->h.numAnswers && !rrsigsCreated && DNSSECQuestion && !NSECRecords)
9699 mDNSCoreReceiveNoDNSSECAnswers(m, response, end, dstaddr, dstport, InterfaceID);
9700
9701 // See if we need to generate negative cache entries for unanswered unicast questions
9702 mDNSCoreReceiveNoUnicastAnswers(m, response, end, dstaddr, dstport, InterfaceID, LLQType, rcode, NSECRecords);
9703
9704 if (McastNSEC3Records)
9705 {
9706 debugf("mDNSCoreReceiveResponse: McastNSEC3Records not used");
9707 FreeNSECRecords(m, McastNSEC3Records);
9708 }
9709 }
9710
9711 // ScheduleWakeup causes all proxy records with WakeUp.HMAC matching mDNSEthAddr 'e' to be deregistered, causing
9712 // multiple wakeup magic packets to be sent if appropriate, and all records to be ultimately freed after a few seconds.
9713 // ScheduleWakeup is called on mDNS record conflicts, ARP conflicts, NDP conflicts, or reception of trigger traffic
9714 // that warrants waking the sleeping host.
9715 // ScheduleWakeup must be called with the lock held (ScheduleWakeupForList uses mDNS_Deregister_internal)
9716
9717 mDNSlocal void ScheduleWakeupForList(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *e, AuthRecord *const thelist)
9718 {
9719 // We need to use the m->CurrentRecord mechanism here when dealing with DuplicateRecords list as
9720 // mDNS_Deregister_internal deregisters duplicate records immediately as they are not used
9721 // to send wakeups or goodbyes. See the comment in that function for more details. To keep it
9722 // simple, we use the same mechanism for both lists.
9723 if (!e->l[0])
9724 {
9725 LogMsg("ScheduleWakeupForList ERROR: Target HMAC is zero");
9726 return;
9727 }
9728 m->CurrentRecord = thelist;
9729 while (m->CurrentRecord)
9730 {
9731 AuthRecord *const rr = m->CurrentRecord;
9732 if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering && mDNSSameEthAddress(&rr->WakeUp.HMAC, e))
9733 {
9734 LogInfo("ScheduleWakeupForList: Scheduling wakeup packets for %s", ARDisplayString(m, rr));
9735 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
9736 }
9737 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
9738 m->CurrentRecord = rr->next;
9739 }
9740 }
9741
9742 mDNSlocal void ScheduleWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *e)
9743 {
9744 if (!e->l[0]) { LogMsg("ScheduleWakeup ERROR: Target HMAC is zero"); return; }
9745 ScheduleWakeupForList(m, InterfaceID, e, m->DuplicateRecords);
9746 ScheduleWakeupForList(m, InterfaceID, e, m->ResourceRecords);
9747 }
9748
9749 mDNSlocal void SPSRecordCallback(mDNS *const m, AuthRecord *const ar, mStatus result)
9750 {
9751 if (result && result != mStatus_MemFree)
9752 LogInfo("SPS Callback %d %s", result, ARDisplayString(m, ar));
9753
9754 if (result == mStatus_NameConflict)
9755 {
9756 mDNS_Lock(m);
9757 LogMsg("%-7s Conflicting mDNS -- waking %.6a %s", InterfaceNameForID(m, ar->resrec.InterfaceID), &ar->WakeUp.HMAC, ARDisplayString(m, ar));
9758 if (ar->WakeUp.HMAC.l[0])
9759 {
9760 SendWakeup(m, ar->resrec.InterfaceID, &ar->WakeUp.IMAC, &ar->WakeUp.password, mDNSfalse); // Send one wakeup magic packet
9761 ScheduleWakeup(m, ar->resrec.InterfaceID, &ar->WakeUp.HMAC); // Schedule all other records with the same owner to be woken
9762 }
9763 mDNS_Unlock(m);
9764 }
9765
9766 if (result == mStatus_NameConflict || result == mStatus_MemFree)
9767 {
9768 m->ProxyRecords--;
9769 mDNSPlatformMemFree(ar);
9770 mDNS_UpdateAllowSleep(m);
9771 }
9772 }
9773
9774 mDNSlocal mDNSu8 *GetValueForMACAddr(mDNSu8 *ptr, mDNSu8 *limit, mDNSEthAddr *eth)
9775 {
9776 int i;
9777 mDNSs8 hval = 0;
9778 int colons = 0;
9779 mDNSu8 val = 0;
9780
9781 for (i = 0; ptr < limit && *ptr != ' ' && i < 17; i++, ptr++)
9782 {
9783 hval = HexVal(*ptr);
9784 if (hval != -1)
9785 {
9786 val <<= 4;
9787 val |= hval;
9788 }
9789 else if (*ptr == ':')
9790 {
9791 if (colons >=5)
9792 {
9793 LogMsg("GetValueForMACAddr: Address malformed colons %d val %d", colons, val);
9794 return mDNSNULL;
9795 }
9796 eth->b[colons] = val;
9797 colons++;
9798 val = 0;
9799 }
9800 }
9801 if (colons != 5)
9802 {
9803 LogMsg("GetValueForMACAddr: Address malformed colons %d", colons);
9804 return mDNSNULL;
9805 }
9806 eth->b[colons] = val;
9807 return ptr;
9808 }
9809
9810 mDNSlocal mDNSu8 *GetValueForIPv6Addr(mDNSu8 *ptr, mDNSu8 *limit, mDNSv6Addr *v6)
9811 {
9812 int hval;
9813 int value;
9814 int numBytes;
9815 int digitsProcessed;
9816 int zeroFillStart;
9817 int numColons;
9818 mDNSu8 v6addr[16];
9819
9820 // RFC 3513: Section 2.2 specifies IPv6 presentation format. The following parsing
9821 // handles both (1) and (2) and does not handle embedded IPv4 addresses.
9822 //
9823 // First forms a address in "v6addr", then expands to fill the zeroes in and returns
9824 // the result in "v6"
9825
9826 numColons = numBytes = value = digitsProcessed = zeroFillStart = 0;
9827 while (ptr < limit && *ptr != ' ')
9828 {
9829 hval = HexVal(*ptr);
9830 if (hval != -1)
9831 {
9832 value <<= 4;
9833 value |= hval;
9834 digitsProcessed = 1;
9835 }
9836 else if (*ptr == ':')
9837 {
9838 if (!digitsProcessed)
9839 {
9840 // If we have already seen a "::", we should not see one more. Handle the special
9841 // case of "::"
9842 if (numColons)
9843 {
9844 // if we never filled any bytes and the next character is space (we have reached the end)
9845 // we are done
9846 if (!numBytes && (ptr + 1) < limit && *(ptr + 1) == ' ')
9847 {
9848 mDNSPlatformMemZero(v6->b, 16);
9849 return ptr + 1;
9850 }
9851 LogMsg("GetValueForIPv6Addr: zeroFillStart non-zero %d", zeroFillStart);
9852 return mDNSNULL;
9853 }
9854
9855 // We processed "::". We need to fill zeroes later. For now, mark the
9856 // point where we will start filling zeroes from.
9857 zeroFillStart = numBytes;
9858 numColons++;
9859 }
9860 else if ((ptr + 1) < limit && *(ptr + 1) == ' ')
9861 {
9862 // We have a trailing ":" i.e., no more characters after ":"
9863 LogMsg("GetValueForIPv6Addr: Trailing colon");
9864 return mDNSNULL;
9865 }
9866 else
9867 {
9868 // For a fully expanded IPv6 address, we fill the 14th and 15th byte outside of this while
9869 // loop below as there is no ":" at the end. Hence, the last two bytes that can possibly
9870 // filled here is 12 and 13.
9871 if (numBytes > 13) { LogMsg("GetValueForIPv6Addr:1: numBytes is %d", numBytes); return mDNSNULL; }
9872
9873 v6addr[numBytes++] = (mDNSu8) ((value >> 8) & 0xFF);
9874 v6addr[numBytes++] = (mDNSu8) (value & 0xFF);
9875 digitsProcessed = value = 0;
9876
9877 // Make sure that we did not fill the 13th and 14th byte above
9878 if (numBytes > 14) { LogMsg("GetValueForIPv6Addr:2: numBytes is %d", numBytes); return mDNSNULL; }
9879 }
9880 }
9881 ptr++;
9882 }
9883
9884 // We should be processing the last set of bytes following the last ":" here
9885 if (!digitsProcessed)
9886 {
9887 LogMsg("GetValueForIPv6Addr: no trailing bytes after colon, numBytes is %d", numBytes);
9888 return mDNSNULL;
9889 }
9890
9891 if (numBytes > 14) { LogMsg("GetValueForIPv6Addr:3: numBytes is %d", numBytes); return mDNSNULL; }
9892 v6addr[numBytes++] = (mDNSu8) ((value >> 8) & 0xFF);
9893 v6addr[numBytes++] = (mDNSu8) (value & 0xFF);
9894
9895 if (zeroFillStart)
9896 {
9897 int i, j, n;
9898 for (i = 0; i < zeroFillStart; i++)
9899 v6->b[i] = v6addr[i];
9900 for (j = i, n = 0; n < 16 - numBytes; j++, n++)
9901 v6->b[j] = 0;
9902 for (; j < 16; i++, j++)
9903 v6->b[j] = v6addr[i];
9904 }
9905 else if (numBytes == 16)
9906 mDNSPlatformMemCopy(v6->b, v6addr, 16);
9907 else
9908 {
9909 LogMsg("GetValueForIPv6addr: Not enough bytes for IPv6 address, numBytes is %d", numBytes);
9910 return mDNSNULL;
9911 }
9912 return ptr;
9913 }
9914
9915 mDNSlocal mDNSu8 *GetValueForIPv4Addr(mDNSu8 *ptr, mDNSu8 *limit, mDNSv4Addr *v4)
9916 {
9917 mDNSu32 val;
9918 int dots = 0;
9919 val = 0;
9920
9921 for ( ; ptr < limit && *ptr != ' '; ptr++)
9922 {
9923 if (*ptr >= '0' && *ptr <= '9')
9924 val = val * 10 + *ptr - '0';
9925 else if (*ptr == '.')
9926 {
9927 if (val > 255 || dots >= 3)
9928 {
9929 LogMsg("GetValueForIPv4Addr: something wrong ptr(%p) %c, limit %p, dots %d", ptr, *ptr, limit, dots);
9930 return mDNSNULL;
9931 }
9932 v4->b[dots++] = val;
9933 val = 0;
9934 }
9935 else
9936 {
9937 // We have a zero at the end and if we reached that, then we are done.
9938 if (*ptr == 0 && ptr == limit - 1 && dots == 3)
9939 {
9940 v4->b[dots] = val;
9941 return ptr + 1;
9942 }
9943 else { LogMsg("GetValueForIPv4Addr: something wrong ptr(%p) %c, limit %p, dots %d", ptr, *ptr, limit, dots); return mDNSNULL; }
9944 }
9945 }
9946 if (dots != 3) { LogMsg("GetValueForIPv4Addr: Address malformed dots %d", dots); return mDNSNULL; }
9947 v4->b[dots] = val;
9948 return ptr;
9949 }
9950
9951 mDNSlocal mDNSu8 *GetValueForKeepalive(mDNSu8 *ptr, mDNSu8 *limit, mDNSu32 *value)
9952 {
9953 mDNSu32 val;
9954
9955 val = 0;
9956 for ( ; ptr < limit && *ptr != ' '; ptr++)
9957 {
9958 if (*ptr < '0' || *ptr > '9')
9959 {
9960 // We have a zero at the end and if we reached that, then we are done.
9961 if (*ptr == 0 && ptr == limit - 1)
9962 {
9963 *value = val;
9964 return ptr + 1;
9965 }
9966 else { LogMsg("GetValueForKeepalive: *ptr %d, ptr %p, limit %p, ptr +1 %d", *ptr, ptr, limit, *(ptr + 1)); return mDNSNULL; }
9967 }
9968 val = val * 10 + *ptr - '0';
9969 }
9970 *value = val;
9971 return ptr;
9972 }
9973
9974 mDNSexport mDNSBool mDNSValidKeepAliveRecord(AuthRecord *rr)
9975 {
9976 mDNSAddr laddr, raddr;
9977 mDNSEthAddr eth;
9978 mDNSIPPort lport, rport;
9979 mDNSu32 timeout, seq, ack;
9980 mDNSu16 win;
9981
9982 if (!mDNS_KeepaliveRecord(&rr->resrec))
9983 {
9984 return mDNSfalse;
9985 }
9986
9987 timeout = seq = ack = 0;
9988 win = 0;
9989 laddr = raddr = zeroAddr;
9990 lport = rport = zeroIPPort;
9991 eth = zeroEthAddr;
9992
9993 mDNS_ExtractKeepaliveInfo(rr, &timeout, &laddr, &raddr, &eth, &seq, &ack, &lport, &rport, &win);
9994
9995 if (mDNSAddressIsZero(&laddr) || mDNSIPPortIsZero(lport) ||
9996 mDNSAddressIsZero(&raddr) || mDNSIPPortIsZero(rport) ||
9997 mDNSEthAddressIsZero(eth))
9998 {
9999 return mDNSfalse;
10000 }
10001
10002 return mDNStrue;
10003 }
10004
10005
10006 mDNSlocal void mDNS_ExtractKeepaliveInfo(AuthRecord *ar, mDNSu32 *timeout, mDNSAddr *laddr, mDNSAddr *raddr, mDNSEthAddr *eth, mDNSu32 *seq,
10007 mDNSu32 *ack, mDNSIPPort *lport, mDNSIPPort *rport, mDNSu16 *win)
10008 {
10009 if (ar->resrec.rrtype != kDNSType_NULL)
10010 return;
10011
10012 if (mDNS_KeepaliveRecord(&ar->resrec))
10013 {
10014 int len = ar->resrec.rdlength;
10015 mDNSu8 *ptr = &ar->resrec.rdata->u.txt.c[1];
10016 mDNSu8 *limit = ptr + len - 1; // Exclude the first byte that is the length
10017 mDNSu32 value = 0;
10018
10019 while (ptr < limit)
10020 {
10021 mDNSu8 param = *ptr;
10022 ptr += 2; // Skip the letter and the "="
10023 if (param == 'h')
10024 {
10025 laddr->type = mDNSAddrType_IPv4;
10026 ptr = GetValueForIPv4Addr(ptr, limit, &laddr->ip.v4);
10027 }
10028 else if (param == 'd')
10029 {
10030 raddr->type = mDNSAddrType_IPv4;
10031 ptr = GetValueForIPv4Addr(ptr, limit, &raddr->ip.v4);
10032 }
10033 else if (param == 'H')
10034 {
10035 laddr->type = mDNSAddrType_IPv6;
10036 ptr = GetValueForIPv6Addr(ptr, limit, &laddr->ip.v6);
10037 }
10038 else if (param == 'D')
10039 {
10040 raddr->type = mDNSAddrType_IPv6;
10041 ptr = GetValueForIPv6Addr(ptr, limit, &raddr->ip.v6);
10042 }
10043 else if (param == 'm')
10044 {
10045 ptr = GetValueForMACAddr(ptr, limit, eth);
10046 }
10047 else
10048 {
10049 ptr = GetValueForKeepalive(ptr, limit, &value);
10050 }
10051 if (!ptr) { LogMsg("mDNS_ExtractKeepaliveInfo: Cannot parse\n"); return; }
10052
10053 // Extract everything in network order so that it is easy for sending a keepalive and also
10054 // for matching incoming TCP packets
10055 switch (param)
10056 {
10057 case 't':
10058 *timeout = value;
10059 //if (*timeout < 120) *timeout = 120;
10060 break;
10061 case 'h':
10062 case 'H':
10063 case 'd':
10064 case 'D':
10065 case 'm':
10066 case 'i':
10067 case 'c':
10068 break;
10069 case 'l':
10070 lport->NotAnInteger = swap16((mDNSu16)value);
10071 break;
10072 case 'r':
10073 rport->NotAnInteger = swap16((mDNSu16)value);
10074 break;
10075 case 's':
10076 *seq = swap32(value);
10077 break;
10078 case 'a':
10079 *ack = swap32(value);
10080 break;
10081 case 'w':
10082 *win = swap16((mDNSu16)value);
10083 break;
10084 default:
10085 LogMsg("mDNS_ExtractKeepaliveInfo: unknown value %c\n", param);
10086 ptr = limit;
10087 break;
10088 }
10089 ptr++; // skip the space
10090 }
10091 }
10092 }
10093
10094 // Matches the proxied auth records to the incoming TCP packet and returns the match and its sequence and ack in "rseq" and "rack" so that
10095 // the clients need not retrieve this information from the auth record again.
10096 mDNSlocal AuthRecord* mDNS_MatchKeepaliveInfo(mDNS *const m, const mDNSAddr* pladdr, const mDNSAddr* praddr, const mDNSIPPort plport,
10097 const mDNSIPPort prport, mDNSu32 *rseq, mDNSu32 *rack)
10098 {
10099 AuthRecord *ar;
10100 mDNSAddr laddr, raddr;
10101 mDNSEthAddr eth;
10102 mDNSIPPort lport, rport;
10103 mDNSu32 timeout, seq, ack;
10104 mDNSu16 win;
10105
10106 for (ar = m->ResourceRecords; ar; ar=ar->next)
10107 {
10108 timeout = seq = ack = 0;
10109 win = 0;
10110 laddr = raddr = zeroAddr;
10111 lport = rport = zeroIPPort;
10112
10113 if (!ar->WakeUp.HMAC.l[0]) continue;
10114
10115 mDNS_ExtractKeepaliveInfo(ar, &timeout, &laddr, &raddr, &eth, &seq, &ack, &lport, &rport, &win);
10116
10117 // Did we parse correctly ?
10118 if (!timeout || mDNSAddressIsZero(&laddr) || mDNSAddressIsZero(&raddr) || !seq || !ack || mDNSIPPortIsZero(lport) || mDNSIPPortIsZero(rport) || !win)
10119 {
10120 debugf("mDNS_MatchKeepaliveInfo: not a valid record %s for keepalive", ARDisplayString(m, ar));
10121 continue;
10122 }
10123
10124 debugf("mDNS_MatchKeepaliveInfo: laddr %#a pladdr %#a, raddr %#a praddr %#a, lport %d plport %d, rport %d prport %d",
10125 &laddr, pladdr, &raddr, praddr, mDNSVal16(lport), mDNSVal16(plport), mDNSVal16(rport), mDNSVal16(prport));
10126
10127 // Does it match the incoming TCP packet ?
10128 if (mDNSSameAddress(&laddr, pladdr) && mDNSSameAddress(&raddr, praddr) && mDNSSameIPPort(lport, plport) && mDNSSameIPPort(rport, prport))
10129 {
10130 // returning in network order
10131 *rseq = seq;
10132 *rack = ack;
10133 return ar;
10134 }
10135 }
10136 return mDNSNULL;
10137 }
10138
10139 mDNSlocal void mDNS_SendKeepalives(mDNS *const m)
10140 {
10141 AuthRecord *ar;
10142
10143 for (ar = m->ResourceRecords; ar; ar=ar->next)
10144 {
10145 mDNSu32 timeout, seq, ack;
10146 mDNSu16 win;
10147 mDNSAddr laddr, raddr;
10148 mDNSEthAddr eth;
10149 mDNSIPPort lport, rport;
10150
10151 timeout = seq = ack = 0;
10152 win = 0;
10153
10154 laddr = raddr = zeroAddr;
10155 lport = rport = zeroIPPort;
10156
10157 if (!ar->WakeUp.HMAC.l[0]) continue;
10158
10159 mDNS_ExtractKeepaliveInfo(ar, &timeout, &laddr, &raddr, &eth, &seq, &ack, &lport, &rport, &win);
10160
10161 if (!timeout || mDNSAddressIsZero(&laddr) || mDNSAddressIsZero(&raddr) || !seq || !ack || mDNSIPPortIsZero(lport) || mDNSIPPortIsZero(rport) || !win)
10162 {
10163 debugf("mDNS_SendKeepalives: not a valid record %s for keepalive", ARDisplayString(m, ar));
10164 continue;
10165 }
10166 LogMsg("mDNS_SendKeepalives: laddr %#a raddr %#a lport %d rport %d", &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport));
10167
10168 // When we receive a proxy update, we set KATimeExpire to zero so that we always send a keepalive
10169 // immediately (to detect any potential problems). After that we always set it to a non-zero value.
10170 if (!ar->KATimeExpire || (m->timenow - ar->KATimeExpire >= 0))
10171 {
10172 mDNSPlatformSendKeepalive(&laddr, &raddr, &lport, &rport, seq, ack, win);
10173 ar->KATimeExpire = NonZeroTime(m->timenow + timeout * mDNSPlatformOneSecond);
10174 }
10175 if (m->NextScheduledKA - ar->KATimeExpire > 0)
10176 m->NextScheduledKA = ar->KATimeExpire;
10177 }
10178 }
10179
10180 mDNSlocal void mDNS_SendKeepaliveACK(mDNS *const m, AuthRecord *ar)
10181 {
10182 mDNSu32 timeout, seq, ack, seqInc;
10183 mDNSu16 win;
10184 mDNSAddr laddr, raddr;
10185 mDNSEthAddr eth;
10186 mDNSIPPort lport, rport;
10187 mDNSu8 *ptr;
10188
10189 if (ar == mDNSNULL)
10190 {
10191 LogInfo("mDNS_SendKeepalivesACK: AuthRecord is NULL");
10192 return;
10193 }
10194
10195 timeout = seq = ack = 0;
10196 win = 0;
10197
10198 laddr = raddr = zeroAddr;
10199 lport = rport = zeroIPPort;
10200
10201 mDNS_ExtractKeepaliveInfo(ar, &timeout, &laddr, &raddr, &eth, &seq, &ack, &lport, &rport, &win);
10202
10203 if (!timeout || mDNSAddressIsZero(&laddr) || mDNSAddressIsZero(&raddr) || !seq || !ack || mDNSIPPortIsZero(lport) || mDNSIPPortIsZero(rport) || !win)
10204 {
10205 LogInfo("mDNS_SendKeepaliveACK: not a valid record %s for keepalive", ARDisplayString(m, ar));
10206 return;
10207 }
10208
10209 // To send a keepalive ACK, we need to add one to the sequence number from the keepalive
10210 // record, which is the TCP connection's "next" sequence number minus one. Otherwise, the
10211 // keepalive ACK also ends up being a keepalive probe. Also, seq is in network byte order, so
10212 // it's converted to host byte order before incrementing it by one.
10213 ptr = (mDNSu8 *)&seq;
10214 seqInc = (mDNSu32)((ptr[0] << 24) | (ptr[1] << 16) | (ptr[2] << 8) | ptr[3]) + 1;
10215 ptr[0] = (mDNSu8)((seqInc >> 24) & 0xFF);
10216 ptr[1] = (mDNSu8)((seqInc >> 16) & 0xFF);
10217 ptr[2] = (mDNSu8)((seqInc >> 8) & 0xFF);
10218 ptr[3] = (mDNSu8)((seqInc ) & 0xFF);
10219 LogMsg("mDNS_SendKeepaliveACK: laddr %#a raddr %#a lport %d rport %d", &laddr, &raddr, mDNSVal16(lport), mDNSVal16(rport));
10220 mDNSPlatformSendKeepalive(&laddr, &raddr, &lport, &rport, seq, ack, win);
10221 }
10222
10223 mDNSlocal void mDNSCoreReceiveUpdate(mDNS *const m,
10224 const DNSMessage *const msg, const mDNSu8 *end,
10225 const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
10226 const mDNSInterfaceID InterfaceID)
10227 {
10228 int i;
10229 AuthRecord opt;
10230 mDNSu8 *p = m->omsg.data;
10231 OwnerOptData owner = zeroOwner; // Need to zero this, so we'll know if this Update packet was missing its Owner option
10232 mDNSu32 updatelease = 0;
10233 const mDNSu8 *ptr;
10234
10235 LogSPS("Received Update from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
10236 "%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes",
10237 srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
10238 msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
10239 msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
10240 msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
10241 msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data);
10242
10243 if (!InterfaceID || !m->SPSSocket || !mDNSSameIPPort(dstport, m->SPSSocket->port)) return;
10244
10245 if (mDNS_PacketLoggingEnabled)
10246 DumpPacket(m, mStatus_NoError, mDNSfalse, "UDP", srcaddr, srcport, dstaddr, dstport, msg, end);
10247
10248 ptr = LocateOptRR(msg, end, DNSOpt_LeaseData_Space + DNSOpt_OwnerData_ID_Space);
10249 if (ptr)
10250 {
10251 ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
10252 if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
10253 {
10254 const rdataOPT *o;
10255 const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
10256 for (o = &m->rec.r.resrec.rdata->u.opt[0]; o < e; o++)
10257 {
10258 if (o->opt == kDNSOpt_Lease) updatelease = o->u.updatelease;
10259 else if (o->opt == kDNSOpt_Owner && o->u.owner.vers == 0) owner = o->u.owner;
10260 }
10261 }
10262 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
10263 }
10264
10265 InitializeDNSMessage(&m->omsg.h, msg->h.id, UpdateRespFlags);
10266
10267 if (!updatelease || !owner.HMAC.l[0])
10268 {
10269 static int msgs = 0;
10270 if (msgs < 100)
10271 {
10272 msgs++;
10273 LogMsg("Refusing sleep proxy registration from %#a:%d:%s%s", srcaddr, mDNSVal16(srcport),
10274 !updatelease ? " No lease" : "", !owner.HMAC.l[0] ? " No owner" : "");
10275 }
10276 m->omsg.h.flags.b[1] |= kDNSFlag1_RC_FormErr;
10277 }
10278 else if (m->ProxyRecords + msg->h.mDNS_numUpdates > MAX_PROXY_RECORDS)
10279 {
10280 static int msgs = 0;
10281 if (msgs < 100)
10282 {
10283 msgs++;
10284 LogMsg("Refusing sleep proxy registration from %#a:%d: Too many records %d + %d = %d > %d", srcaddr, mDNSVal16(srcport),
10285 m->ProxyRecords, msg->h.mDNS_numUpdates, m->ProxyRecords + msg->h.mDNS_numUpdates, MAX_PROXY_RECORDS);
10286 }
10287 m->omsg.h.flags.b[1] |= kDNSFlag1_RC_Refused;
10288 }
10289 else
10290 {
10291 LogSPS("Received Update for H-MAC %.6a I-MAC %.6a Password %.6a seq %d", &owner.HMAC, &owner.IMAC, &owner.password, owner.seq);
10292
10293 if (updatelease > 24 * 60 * 60)
10294 updatelease = 24 * 60 * 60;
10295
10296 if (updatelease > 0x40000000UL / mDNSPlatformOneSecond)
10297 updatelease = 0x40000000UL / mDNSPlatformOneSecond;
10298
10299 ptr = LocateAuthorities(msg, end);
10300
10301 // Clear any stale TCP keepalive records that may exist
10302 ClearKeepaliveProxyRecords(m, &owner, m->DuplicateRecords, InterfaceID);
10303 ClearKeepaliveProxyRecords(m, &owner, m->ResourceRecords, InterfaceID);
10304
10305 for (i = 0; i < msg->h.mDNS_numUpdates && ptr && ptr < end; i++)
10306 {
10307 ptr = GetLargeResourceRecord(m, msg, ptr, end, InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
10308 if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative)
10309 {
10310 mDNSu16 RDLengthMem = GetRDLengthMem(&m->rec.r.resrec);
10311 AuthRecord *ar = mDNSPlatformMemAllocate(sizeof(AuthRecord) - sizeof(RDataBody) + RDLengthMem);
10312 if (!ar)
10313 {
10314 m->omsg.h.flags.b[1] |= kDNSFlag1_RC_Refused;
10315 break;
10316 }
10317 else
10318 {
10319 mDNSu8 RecordType = m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask ? kDNSRecordTypeUnique : kDNSRecordTypeShared;
10320 m->rec.r.resrec.rrclass &= ~kDNSClass_UniqueRRSet;
10321 // All stale keepalive records have been flushed prior to this loop.
10322 if (!mDNS_KeepaliveRecord(&m->rec.r.resrec))
10323 {
10324 ClearIdenticalProxyRecords(m, &owner, m->DuplicateRecords); // Make sure we don't have any old stale duplicates of this record
10325 ClearIdenticalProxyRecords(m, &owner, m->ResourceRecords);
10326 }
10327 mDNS_SetupResourceRecord(ar, mDNSNULL, InterfaceID, m->rec.r.resrec.rrtype, m->rec.r.resrec.rroriginalttl, RecordType, AuthRecordAny, SPSRecordCallback, ar);
10328 AssignDomainName(&ar->namestorage, m->rec.r.resrec.name);
10329 ar->resrec.rdlength = GetRDLength(&m->rec.r.resrec, mDNSfalse);
10330 ar->resrec.rdata->MaxRDLength = RDLengthMem;
10331 mDNSPlatformMemCopy(ar->resrec.rdata->u.data, m->rec.r.resrec.rdata->u.data, RDLengthMem);
10332 ar->ForceMCast = mDNStrue;
10333 ar->WakeUp = owner;
10334 if (m->rec.r.resrec.rrtype == kDNSType_PTR)
10335 {
10336 mDNSs32 t = ReverseMapDomainType(m->rec.r.resrec.name);
10337 if (t == mDNSAddrType_IPv4) GetIPv4FromName(&ar->AddressProxy, m->rec.r.resrec.name);
10338 else if (t == mDNSAddrType_IPv6) GetIPv6FromName(&ar->AddressProxy, m->rec.r.resrec.name);
10339 debugf("mDNSCoreReceiveUpdate: PTR %d %d %#a %s", t, ar->AddressProxy.type, &ar->AddressProxy, ARDisplayString(m, ar));
10340 if (ar->AddressProxy.type) SetSPSProxyListChanged(InterfaceID);
10341 }
10342 ar->TimeRcvd = m->timenow;
10343 ar->TimeExpire = m->timenow + updatelease * mDNSPlatformOneSecond;
10344 if (m->NextScheduledSPS - ar->TimeExpire > 0)
10345 m->NextScheduledSPS = ar->TimeExpire;
10346 ar->KATimeExpire = 0;
10347 mDNS_Register_internal(m, ar);
10348
10349 m->ProxyRecords++;
10350 mDNS_UpdateAllowSleep(m);
10351 LogSPS("SPS Registered %4d %X %s", m->ProxyRecords, RecordType, ARDisplayString(m,ar));
10352 }
10353 }
10354 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
10355 }
10356
10357 if (m->omsg.h.flags.b[1] & kDNSFlag1_RC_Mask)
10358 {
10359 LogMsg("Refusing sleep proxy registration from %#a:%d: Out of memory", srcaddr, mDNSVal16(srcport));
10360 ClearProxyRecords(m, &owner, m->DuplicateRecords);
10361 ClearProxyRecords(m, &owner, m->ResourceRecords);
10362 }
10363 else
10364 {
10365 mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
10366 opt.resrec.rrclass = NormalMaxDNSMessageData;
10367 opt.resrec.rdlength = sizeof(rdataOPT); // One option in this OPT record
10368 opt.resrec.rdestimate = sizeof(rdataOPT);
10369 opt.resrec.rdata->u.opt[0].opt = kDNSOpt_Lease;
10370 opt.resrec.rdata->u.opt[0].u.updatelease = updatelease;
10371 p = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.numAdditionals, &opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
10372 }
10373 }
10374
10375 if (p) mDNSSendDNSMessage(m, &m->omsg, p, InterfaceID, m->SPSSocket, srcaddr, srcport, mDNSNULL, mDNSNULL, mDNSfalse);
10376 mDNS_SendKeepalives(m);
10377 }
10378
10379 mDNSlocal mDNSu32 mDNSGenerateOwnerOptForInterface(mDNS *const m, const mDNSInterfaceID InterfaceID, DNSMessage *msg)
10380 {
10381 mDNSu8 *ptr = msg->data;
10382 mDNSu8 *end = mDNSNULL;
10383 mDNSu32 length = 0;
10384 AuthRecord opt;
10385
10386 mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
10387 opt.resrec.rrclass = NormalMaxDNSMessageData;
10388 opt.resrec.rdlength = sizeof(rdataOPT);
10389 opt.resrec.rdestimate = sizeof(rdataOPT);
10390
10391 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
10392 SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
10393
10394 LogSPS("Generated OPT record : %s", ARDisplayString(m, &opt));
10395 end = PutResourceRecord(msg, ptr, &msg->h.numAdditionals, &opt.resrec);
10396 if (end != mDNSNULL)
10397 {
10398 // Put all the integer values in IETF byte-order (MSB first, LSB second)
10399 SwapDNSHeaderBytes(msg);
10400 length = (end - msg->data);
10401 }
10402 else
10403 LogSPS("mDNSGenerateOwnerOptForInterface: Failed to generate owner OPT record");
10404
10405 return length;
10406 }
10407
10408 mDNSlocal void mDNSCoreReceiveUpdateR(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *end, const mDNSAddr *srcaddr, const mDNSInterfaceID InterfaceID)
10409 {
10410 if (InterfaceID)
10411 {
10412 mDNSu32 updatelease = 60 * 60; // If SPS fails to indicate lease time, assume one hour
10413 const mDNSu8 *ptr = LocateOptRR(msg, end, DNSOpt_LeaseData_Space);
10414 mDNSAddr spsaddr;
10415 char *ifname;
10416 if (ptr)
10417 {
10418 ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
10419 if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
10420 {
10421 const rdataOPT *o;
10422 const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
10423 for (o = &m->rec.r.resrec.rdata->u.opt[0]; o < e; o++)
10424 if (o->opt == kDNSOpt_Lease)
10425 {
10426 updatelease = o->u.updatelease;
10427 LogSPS("Sleep Proxy granted lease time %4d seconds, updateid %d, InterfaceID %p", updatelease, mDNSVal16(msg->h.id), InterfaceID);
10428 }
10429 }
10430 m->rec.r.resrec.RecordType = 0; // Clear RecordType to show we're not still using it
10431 }
10432
10433 if (m->CurrentRecord)
10434 LogMsg("mDNSCoreReceiveUpdateR ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
10435 m->CurrentRecord = m->ResourceRecords;
10436 while (m->CurrentRecord)
10437 {
10438 AuthRecord *const rr = m->CurrentRecord;
10439 if (rr->resrec.InterfaceID == InterfaceID || (!rr->resrec.InterfaceID && (rr->ForceMCast || IsLocalDomain(rr->resrec.name))))
10440 if (mDNSSameOpaque16(rr->updateid, msg->h.id))
10441 {
10442 // We successfully completed this record's registration on this "InterfaceID". Clear that bit.
10443 // Clear the updateid when we are done sending on all interfaces.
10444 mDNSu32 scopeid = mDNSPlatformInterfaceIndexfromInterfaceID(m, InterfaceID, mDNStrue);
10445 if (scopeid < (sizeof(rr->updateIntID) * mDNSNBBY))
10446 bit_clr_opaque64(rr->updateIntID, scopeid);
10447 if (mDNSOpaque64IsZero(&rr->updateIntID))
10448 rr->updateid = zeroID;
10449 rr->expire = NonZeroTime(m->timenow + updatelease * mDNSPlatformOneSecond);
10450 LogSPS("Sleep Proxy %s record %5d 0x%x 0x%x (%d) %s", rr->WakeUp.HMAC.l[0] ? "transferred" : "registered", updatelease, rr->updateIntID.l[1], rr->updateIntID.l[0], mDNSVal16(rr->updateid), ARDisplayString(m,rr));
10451 if (rr->WakeUp.HMAC.l[0])
10452 {
10453 rr->WakeUp.HMAC = zeroEthAddr; // Clear HMAC so that mDNS_Deregister_internal doesn't waste packets trying to wake this host
10454 rr->RequireGoodbye = mDNSfalse; // and we don't want to send goodbye for it
10455 mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
10456 }
10457 }
10458 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
10459 // new records could have been added to the end of the list as a result of that call.
10460 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
10461 m->CurrentRecord = rr->next;
10462 }
10463
10464 // Update the dynamic store with the IP Address and MAC address of the sleep proxy
10465 ifname = InterfaceNameForID(m, InterfaceID);
10466 mDNSPlatformMemCopy(&spsaddr, srcaddr, sizeof (mDNSAddr));
10467 mDNSPlatformStoreSPSMACAddr(&spsaddr, ifname);
10468
10469 // Store the Owner OPT record for this interface.
10470 // Configd may use the OPT record if it detects a conflict with the BSP when the system wakes up
10471 DNSMessage optMsg;
10472 int length = 0;
10473 InitializeDNSMessage(&optMsg.h, zeroID, ResponseFlags);
10474 length = mDNSGenerateOwnerOptForInterface(m, InterfaceID, &optMsg);
10475 if (length != 0)
10476 {
10477 length += sizeof(DNSMessageHeader);
10478 mDNSPlatformStoreOwnerOptRecord(ifname, &optMsg, length);
10479 }
10480 }
10481 // If we were waiting to go to sleep, then this SPS registration or wide-area record deletion
10482 // may have been the thing we were waiting for, so schedule another check to see if we can sleep now.
10483 if (m->SleepLimit) m->NextScheduledSPRetry = m->timenow;
10484 }
10485
10486 mDNSexport void MakeNegativeCacheRecord(mDNS *const m, CacheRecord *const cr,
10487 const domainname *const name, const mDNSu32 namehash, const mDNSu16 rrtype, const mDNSu16 rrclass, mDNSu32 ttl_seconds, mDNSInterfaceID InterfaceID, DNSServer *dnsserver)
10488 {
10489 if (cr == &m->rec.r && m->rec.r.resrec.RecordType)
10490 LogFatalError("MakeNegativeCacheRecord: m->rec appears to be already in use for %s", CRDisplayString(m, &m->rec.r));
10491
10492 // Create empty resource record
10493 cr->resrec.RecordType = kDNSRecordTypePacketNegative;
10494 cr->resrec.InterfaceID = InterfaceID;
10495 cr->resrec.rDNSServer = dnsserver;
10496 cr->resrec.name = name; // Will be updated to point to cg->name when we call CreateNewCacheEntry
10497 cr->resrec.rrtype = rrtype;
10498 cr->resrec.rrclass = rrclass;
10499 cr->resrec.rroriginalttl = ttl_seconds;
10500 cr->resrec.rdlength = 0;
10501 cr->resrec.rdestimate = 0;
10502 cr->resrec.namehash = namehash;
10503 cr->resrec.rdatahash = 0;
10504 cr->resrec.rdata = (RData*)&cr->smallrdatastorage;
10505 cr->resrec.rdata->MaxRDLength = 0;
10506
10507 cr->NextInKAList = mDNSNULL;
10508 cr->TimeRcvd = m->timenow;
10509 cr->DelayDelivery = 0;
10510 cr->NextRequiredQuery = m->timenow;
10511 cr->LastUsed = m->timenow;
10512 cr->CRActiveQuestion = mDNSNULL;
10513 cr->UnansweredQueries = 0;
10514 cr->LastUnansweredTime = 0;
10515 #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
10516 cr->MPUnansweredQ = 0;
10517 cr->MPLastUnansweredQT = 0;
10518 cr->MPUnansweredKA = 0;
10519 cr->MPExpectingKA = mDNSfalse;
10520 #endif
10521 cr->NextInCFList = mDNSNULL;
10522 cr->nsec = mDNSNULL;
10523 cr->soa = mDNSNULL;
10524 cr->CRDNSSECQuestion = 0;
10525 // Initialize to the basic one and the caller can set it to more
10526 // specific based on the response if any
10527 cr->responseFlags = ResponseFlags;
10528 }
10529
10530 mDNSexport void mDNSCoreReceive(mDNS *const m, DNSMessage *const msg, const mDNSu8 *const end,
10531 const mDNSAddr *const srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, const mDNSIPPort dstport,
10532 const mDNSInterfaceID InterfaceID)
10533 {
10534 mDNSInterfaceID ifid = InterfaceID;
10535 const mDNSu8 *const pkt = (mDNSu8 *)msg;
10536 const mDNSu8 StdQ = kDNSFlag0_QR_Query | kDNSFlag0_OP_StdQuery;
10537 const mDNSu8 StdR = kDNSFlag0_QR_Response | kDNSFlag0_OP_StdQuery;
10538 const mDNSu8 UpdQ = kDNSFlag0_QR_Query | kDNSFlag0_OP_Update;
10539 const mDNSu8 UpdR = kDNSFlag0_QR_Response | kDNSFlag0_OP_Update;
10540 mDNSu8 QR_OP;
10541 mDNSu8 *ptr = mDNSNULL;
10542 mDNSBool TLS = (dstaddr == (mDNSAddr *)1); // For debug logs: dstaddr = 0 means TCP; dstaddr = 1 means TLS
10543 if (TLS) dstaddr = mDNSNULL;
10544
10545 #ifndef UNICAST_DISABLED
10546 if (mDNSSameAddress(srcaddr, &m->Router))
10547 {
10548 #ifdef _LEGACY_NAT_TRAVERSAL_
10549 if (mDNSSameIPPort(srcport, SSDPPort) || (m->SSDPSocket && mDNSSameIPPort(dstport, m->SSDPSocket->port)))
10550 {
10551 mDNS_Lock(m);
10552 LNT_ConfigureRouterInfo(m, InterfaceID, (mDNSu8 *)msg, (mDNSu16)(end - pkt));
10553 mDNS_Unlock(m);
10554 return;
10555 }
10556 #endif
10557 if (mDNSSameIPPort(srcport, NATPMPPort))
10558 {
10559 mDNS_Lock(m);
10560 uDNS_ReceiveNATPacket(m, InterfaceID, (mDNSu8 *)msg, (mDNSu16)(end - pkt));
10561 mDNS_Unlock(m);
10562 return;
10563 }
10564 }
10565 #ifdef _LEGACY_NAT_TRAVERSAL_
10566 else if (m->SSDPSocket && mDNSSameIPPort(dstport, m->SSDPSocket->port)) { debugf("Ignoring SSDP response from %#a:%d", srcaddr, mDNSVal16(srcport)); return; }
10567 #endif
10568
10569 #endif
10570 if ((unsigned)(end - pkt) < sizeof(DNSMessageHeader))
10571 {
10572 LogMsg("DNS Message from %#a:%d to %#a:%d length %d too short", srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), (int)(end - pkt));
10573 return;
10574 }
10575 QR_OP = (mDNSu8)(msg->h.flags.b[0] & kDNSFlag0_QROP_Mask);
10576 // Read the integer parts which are in IETF byte-order (MSB first, LSB second)
10577 ptr = (mDNSu8 *)&msg->h.numQuestions;
10578 msg->h.numQuestions = (mDNSu16)((mDNSu16)ptr[0] << 8 | ptr[1]);
10579 msg->h.numAnswers = (mDNSu16)((mDNSu16)ptr[2] << 8 | ptr[3]);
10580 msg->h.numAuthorities = (mDNSu16)((mDNSu16)ptr[4] << 8 | ptr[5]);
10581 msg->h.numAdditionals = (mDNSu16)((mDNSu16)ptr[6] << 8 | ptr[7]);
10582
10583 if (!m) { LogMsg("mDNSCoreReceive ERROR m is NULL"); return; }
10584
10585 // We use zero addresses and all-ones addresses at various places in the code to indicate special values like "no address"
10586 // If we accept and try to process a packet with zero or all-ones source address, that could really mess things up
10587 if (srcaddr && !mDNSAddressIsValid(srcaddr)) { debugf("mDNSCoreReceive ignoring packet from %#a", srcaddr); return; }
10588
10589 mDNS_Lock(m);
10590 m->PktNum++;
10591 if (mDNSOpaque16IsZero(msg->h.id))
10592 {
10593 m->MPktNum++;
10594 #if APPLE_OSX_mDNSResponder
10595 // Track the number of multicast packets received from a source outside our subnet.
10596 // Check the destination address to avoid accounting for spurious packets that
10597 // comes in with message id zero.
10598 if (!mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr) &&
10599 mDNSAddressIsAllDNSLinkGroup(dstaddr))
10600 {
10601 m->RemoteSubnet++;
10602 }
10603 #endif // #if APPLE_OSX_mDNSResponder
10604 }
10605
10606 #ifndef UNICAST_DISABLED
10607 if (!dstaddr || (!mDNSAddressIsAllDNSLinkGroup(dstaddr) && (QR_OP == StdR || QR_OP == UpdR)))
10608 if (!mDNSOpaque16IsZero(msg->h.id)) // uDNS_ReceiveMsg only needs to get real uDNS responses, not "QU" mDNS responses
10609 {
10610 ifid = mDNSInterface_Any;
10611 if (mDNS_PacketLoggingEnabled)
10612 DumpPacket(m, mStatus_NoError, mDNSfalse, TLS ? "TLS" : !dstaddr ? "TCP" : "UDP", srcaddr, srcport, dstaddr, dstport, msg, end);
10613 uDNS_ReceiveMsg(m, msg, end, srcaddr, srcport);
10614 // Note: mDNSCore also needs to get access to received unicast responses
10615 }
10616 #endif
10617 if (QR_OP == StdQ) mDNSCoreReceiveQuery (m, msg, end, srcaddr, srcport, dstaddr, dstport, ifid);
10618 else if (QR_OP == StdR) mDNSCoreReceiveResponse(m, msg, end, srcaddr, srcport, dstaddr, dstport, ifid);
10619 else if (QR_OP == UpdQ) mDNSCoreReceiveUpdate (m, msg, end, srcaddr, srcport, dstaddr, dstport, InterfaceID);
10620 else if (QR_OP == UpdR) mDNSCoreReceiveUpdateR (m, msg, end, srcaddr, InterfaceID);
10621 else
10622 {
10623 if (mDNS_LoggingEnabled)
10624 {
10625 static int msgCount = 0;
10626 if (msgCount < 1000) {
10627 msgCount++;
10628 int i = 0;
10629 LogInfo("Unknown DNS packet type %02X%02X from %#-15a:%-5d to %#-15a:%-5d length %d on %p (ignored)",
10630 msg->h.flags.b[0], msg->h.flags.b[1], srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), (int)(end - pkt), InterfaceID);
10631 while (i < (int)(end - pkt))
10632 {
10633 char buffer[128];
10634 char *p = buffer + mDNS_snprintf(buffer, sizeof(buffer), "%04X", i);
10635 do if (i < (int)(end - pkt)) p += mDNS_snprintf(p, sizeof(buffer), " %02X", pkt[i]);while (++i & 15);
10636 LogInfo("%s", buffer);
10637 }
10638 }
10639 }
10640 }
10641 // Packet reception often causes a change to the task list:
10642 // 1. Inbound queries can cause us to need to send responses
10643 // 2. Conflicing response packets received from other hosts can cause us to need to send defensive responses
10644 // 3. Other hosts announcing deletion of shared records can cause us to need to re-assert those records
10645 // 4. Response packets that answer questions may cause our client to issue new questions
10646 mDNS_Unlock(m);
10647 }
10648
10649 // ***************************************************************************
10650 #if COMPILER_LIKES_PRAGMA_MARK
10651 #pragma mark -
10652 #pragma mark - Searcher Functions
10653 #endif
10654
10655 // Targets are considered the same if both queries are untargeted, or
10656 // if both are targeted to the same address+port
10657 // (If Target address is zero, TargetPort is undefined)
10658 #define SameQTarget(A,B) (((A)->Target.type == mDNSAddrType_None && (B)->Target.type == mDNSAddrType_None) || \
10659 (mDNSSameAddress(& (A)->Target, & (B)->Target) && mDNSSameIPPort((A)->TargetPort, (B)->TargetPort)))
10660
10661 // Note: We explicitly disallow making a public query be a duplicate of a private one. This is to avoid the
10662 // circular deadlock where a client does a query for something like "dns-sd -Q _dns-query-tls._tcp.company.com SRV"
10663 // and we have a key for company.com, so we try to locate the private query server for company.com, which necessarily entails
10664 // doing a standard DNS query for the _dns-query-tls._tcp SRV record for company.com. If we make the latter (public) query
10665 // a duplicate of the former (private) query, then it will block forever waiting for an answer that will never come.
10666 //
10667 // We keep SuppressUnusable questions separate so that we can return a quick response to them and not get blocked behind
10668 // the queries that are not marked SuppressUnusable. But if the query is not suppressed, they are treated the same as
10669 // non-SuppressUnusable questions. This should be fine as the goal of SuppressUnusable is to return quickly only if it
10670 // is suppressed. If it is not suppressed, we do try all the DNS servers for valid answers like any other question.
10671 // The main reason for this design is that cache entries point to a *single* question and that question is responsible
10672 // for keeping the cache fresh as long as it is active. Having multiple active question for a single cache entry
10673 // breaks this design principle.
10674 //
10675
10676 // If IsLLQ(Q) is true, it means the question is both:
10677 // (a) long-lived and
10678 // (b) being performed by a unicast DNS long-lived query (either full LLQ, or polling)
10679 // for multicast questions, we don't want to treat LongLived as anything special
10680 #define IsLLQ(Q) ((Q)->LongLived && !mDNSOpaque16IsZero((Q)->TargetQID))
10681 #define IsAWDLIncluded(Q) (((Q)->flags & kDNSServiceFlagsIncludeAWDL) != 0)
10682
10683 mDNSlocal DNSQuestion *FindDuplicateQuestion(const mDNS *const m, const DNSQuestion *const question)
10684 {
10685 DNSQuestion *q;
10686 // Note: A question can only be marked as a duplicate of one that occurs *earlier* in the list.
10687 // This prevents circular references, where two questions are each marked as a duplicate of the other.
10688 // Accordingly, we break out of the loop when we get to 'question', because there's no point searching
10689 // further in the list.
10690 for (q = m->Questions; q && q != question; q=q->next) // Scan our list for another question
10691 if (q->InterfaceID == question->InterfaceID && // with the same InterfaceID,
10692 SameQTarget(q, question) && // and same unicast/multicast target settings
10693 q->qtype == question->qtype && // type,
10694 q->qclass == question->qclass && // class,
10695 IsLLQ(q) == IsLLQ(question) && // and long-lived status matches
10696 (!q->AuthInfo || question->AuthInfo) && // to avoid deadlock, don't make public query dup of a private one
10697 (q->AnonInfo == question->AnonInfo) && // Anonymous query not a dup of normal query
10698 (q->SuppressQuery == question->SuppressQuery) && // Questions that are suppressed/not suppressed
10699 (q->ValidationRequired == question->ValidationRequired) && // Questions that require DNSSEC validation
10700 (q->ValidatingResponse == question->ValidatingResponse) && // Questions that are validating responses using DNSSEC
10701 (q->DisallowPID == question->DisallowPID) && // Disallowing a PID should not affect a PID that is allowed
10702 (q->BrowseThreshold == question->BrowseThreshold) && // browse thresholds must match
10703 q->qnamehash == question->qnamehash &&
10704 (IsAWDLIncluded(q) == IsAWDLIncluded(question)) && // Inclusion of AWDL interface must match
10705 SameDomainName(&q->qname, &question->qname)) // and name
10706 return(q);
10707 return(mDNSNULL);
10708 }
10709
10710 // This is called after a question is deleted, in case other identical questions were being suppressed as duplicates
10711 mDNSlocal void UpdateQuestionDuplicates(mDNS *const m, DNSQuestion *const question)
10712 {
10713 DNSQuestion *q;
10714 DNSQuestion *first = mDNSNULL;
10715
10716 // This is referring to some other question as duplicate. No other question can refer to this
10717 // question as a duplicate.
10718 if (question->DuplicateOf)
10719 {
10720 LogInfo("UpdateQuestionDuplicates: question %p %##s (%s) duplicate of %p %##s (%s)",
10721 question, question->qname.c, DNSTypeName(question->qtype),
10722 question->DuplicateOf, question->DuplicateOf->qname.c, DNSTypeName(question->DuplicateOf->qtype));
10723 return;
10724 }
10725
10726 for (q = m->Questions; q; q=q->next) // Scan our list of questions
10727 if (q->DuplicateOf == question) // To see if any questions were referencing this as their duplicate
10728 {
10729 q->DuplicateOf = first;
10730 if (!first)
10731 {
10732 first = q;
10733 // If q used to be a duplicate, but now is not,
10734 // then inherit the state from the question that's going away
10735 q->LastQTime = question->LastQTime;
10736 q->ThisQInterval = question->ThisQInterval;
10737 q->ExpectUnicastResp = question->ExpectUnicastResp;
10738 q->LastAnswerPktNum = question->LastAnswerPktNum;
10739 q->RecentAnswerPkts = question->RecentAnswerPkts;
10740 q->RequestUnicast = question->RequestUnicast;
10741 q->LastQTxTime = question->LastQTxTime;
10742 q->CNAMEReferrals = question->CNAMEReferrals;
10743 q->nta = question->nta;
10744 q->servAddr = question->servAddr;
10745 q->servPort = question->servPort;
10746 q->qDNSServer = question->qDNSServer;
10747 q->validDNSServers = question->validDNSServers;
10748 q->unansweredQueries = question->unansweredQueries;
10749 q->noServerResponse = question->noServerResponse;
10750 q->triedAllServersOnce = question->triedAllServersOnce;
10751
10752 q->TargetQID = question->TargetQID;
10753 q->LocalSocket = question->LocalSocket;
10754 // No need to close old q->LocalSocket first -- duplicate questions can't have their own sockets
10755
10756 q->state = question->state;
10757 // q->tcp = question->tcp;
10758 q->ReqLease = question->ReqLease;
10759 q->expire = question->expire;
10760 q->ntries = question->ntries;
10761 q->id = question->id;
10762
10763 question->LocalSocket = mDNSNULL;
10764 question->nta = mDNSNULL; // If we've got a GetZoneData in progress, transfer it to the newly active question
10765 // question->tcp = mDNSNULL;
10766
10767 if (q->LocalSocket)
10768 debugf("UpdateQuestionDuplicates transferred LocalSocket pointer for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
10769
10770 if (q->nta)
10771 {
10772 LogInfo("UpdateQuestionDuplicates transferred nta pointer for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
10773 q->nta->ZoneDataContext = q;
10774 }
10775
10776 // Need to work out how to safely transfer this state too -- appropriate context pointers need to be updated or the code will crash
10777 if (question->tcp) LogInfo("UpdateQuestionDuplicates did not transfer tcp pointer");
10778
10779 if (question->state == LLQ_Established)
10780 {
10781 LogInfo("UpdateQuestionDuplicates transferred LLQ state for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
10782 question->state = 0; // Must zero question->state, or mDNS_StopQuery_internal will clean up and cancel our LLQ from the server
10783 }
10784
10785 SetNextQueryTime(m,q);
10786 }
10787 }
10788 }
10789
10790 mDNSexport McastResolver *mDNS_AddMcastResolver(mDNS *const m, const domainname *d, const mDNSInterfaceID interface, mDNSu32 timeout)
10791 {
10792 McastResolver **p = &m->McastResolvers;
10793 McastResolver *tmp = mDNSNULL;
10794
10795 if (!d) d = (const domainname *)"";
10796
10797 LogInfo("mDNS_AddMcastResolver: Adding %##s, InterfaceID %p, timeout %u", d->c, interface, timeout);
10798
10799 mDNS_CheckLock(m);
10800
10801 while (*p) // Check if we already have this {interface, domain} tuple registered
10802 {
10803 if ((*p)->interface == interface && SameDomainName(&(*p)->domain, d))
10804 {
10805 if (!((*p)->flags & McastResolver_FlagDelete)) LogMsg("Note: Mcast Resolver domain %##s (%p) registered more than once", d->c, interface);
10806 (*p)->flags &= ~McastResolver_FlagDelete;
10807 tmp = *p;
10808 *p = tmp->next;
10809 tmp->next = mDNSNULL;
10810 }
10811 else
10812 p=&(*p)->next;
10813 }
10814
10815 if (tmp) *p = tmp; // move to end of list, to ensure ordering from platform layer
10816 else
10817 {
10818 // allocate, add to list
10819 *p = mDNSPlatformMemAllocate(sizeof(**p));
10820 if (!*p) LogMsg("mDNS_AddMcastResolver: ERROR!! - malloc");
10821 else
10822 {
10823 (*p)->interface = interface;
10824 (*p)->flags = McastResolver_FlagNew;
10825 (*p)->timeout = timeout;
10826 AssignDomainName(&(*p)->domain, d);
10827 (*p)->next = mDNSNULL;
10828 }
10829 }
10830 return(*p);
10831 }
10832
10833 mDNSinline mDNSs32 PenaltyTimeForServer(mDNS *m, DNSServer *server)
10834 {
10835 mDNSs32 ptime = 0;
10836 if (server->penaltyTime != 0)
10837 {
10838 ptime = server->penaltyTime - m->timenow;
10839 if (ptime < 0)
10840 {
10841 // This should always be a positive value between 0 and DNSSERVER_PENALTY_TIME
10842 // If it does not get reset in ResetDNSServerPenalties for some reason, we do it
10843 // here
10844 LogMsg("PenaltyTimeForServer: PenaltyTime negative %d, (server penaltyTime %d, timenow %d) resetting the penalty",
10845 ptime, server->penaltyTime, m->timenow);
10846 server->penaltyTime = 0;
10847 ptime = 0;
10848 }
10849 }
10850 return ptime;
10851 }
10852
10853 //Checks to see whether the newname is a better match for the name, given the best one we have
10854 //seen so far (given in bestcount).
10855 //Returns -1 if the newname is not a better match
10856 //Returns 0 if the newname is the same as the old match
10857 //Returns 1 if the newname is a better match
10858 mDNSlocal int BetterMatchForName(const domainname *name, int namecount, const domainname *newname, int newcount,
10859 int bestcount)
10860 {
10861 // If the name contains fewer labels than the new server's domain or the new name
10862 // contains fewer labels than the current best, then it can't possibly be a better match
10863 if (namecount < newcount || newcount < bestcount) return -1;
10864
10865 // If there is no match, return -1 and the caller will skip this newname for
10866 // selection
10867 //
10868 // If we find a match and the number of labels is the same as bestcount, then
10869 // we return 0 so that the caller can do additional logic to pick one of
10870 // the best based on some other factors e.g., penaltyTime
10871 //
10872 // If we find a match and the number of labels is more than bestcount, then we
10873 // return 1 so that the caller can pick this over the old one.
10874 //
10875 // Note: newcount can either be equal or greater than bestcount beause of the
10876 // check above.
10877
10878 if (SameDomainName(SkipLeadingLabels(name, namecount - newcount), newname))
10879 return bestcount == newcount ? 0 : 1;
10880 else
10881 return -1;
10882 }
10883
10884 // Normally, we have McastResolvers for .local, in-addr.arpa and ip6.arpa. But there
10885 // can be queries that can forced to multicast (ForceMCast) even though they don't end in these
10886 // names. In that case, we give a default timeout of 5 seconds
10887 #define DEFAULT_MCAST_TIMEOUT 5
10888 mDNSlocal mDNSu32 GetTimeoutForMcastQuestion(mDNS *m, DNSQuestion *question)
10889 {
10890 McastResolver *curmatch = mDNSNULL;
10891 int bestmatchlen = -1, namecount = CountLabels(&question->qname);
10892 McastResolver *curr;
10893 int bettermatch, currcount;
10894 for (curr = m->McastResolvers; curr; curr = curr->next)
10895 {
10896 currcount = CountLabels(&curr->domain);
10897 bettermatch = BetterMatchForName(&question->qname, namecount, &curr->domain, currcount, bestmatchlen);
10898 // Take the first best match. If there are multiple equally good matches (bettermatch = 0), we take
10899 // the timeout value from the first one
10900 if (bettermatch == 1)
10901 {
10902 curmatch = curr;
10903 bestmatchlen = currcount;
10904 }
10905 }
10906 LogInfo("GetTimeoutForMcastQuestion: question %##s curmatch %p, Timeout %d", question->qname.c, curmatch,
10907 curmatch ? curmatch->timeout : DEFAULT_MCAST_TIMEOUT);
10908 return ( curmatch ? curmatch->timeout : DEFAULT_MCAST_TIMEOUT);
10909 }
10910
10911 // Returns true if it is a Domain Enumeration Query
10912 mDNSexport mDNSBool DomainEnumQuery(const domainname *qname)
10913 {
10914 const mDNSu8 *mDNS_DEQLabels[] = { (const mDNSu8 *)"\001b", (const mDNSu8 *)"\002db", (const mDNSu8 *)"\002lb",
10915 (const mDNSu8 *)"\001r", (const mDNSu8 *)"\002dr", (const mDNSu8 *)mDNSNULL, };
10916 const domainname *d = qname;
10917 const mDNSu8 *label;
10918 int i = 0;
10919
10920 // We need at least 3 labels (DEQ prefix) + one more label to make a meaningful DE query
10921 if (CountLabels(qname) < 4) { debugf("DomainEnumQuery: question %##s, not enough labels", qname->c); return mDNSfalse; }
10922
10923 label = (const mDNSu8 *)d;
10924 while (mDNS_DEQLabels[i] != (const mDNSu8 *)mDNSNULL)
10925 {
10926 if (SameDomainLabel(mDNS_DEQLabels[i], label)) {debugf("DomainEnumQuery: DEQ %##s, label1 match", qname->c); break;}
10927 i++;
10928 }
10929 if (mDNS_DEQLabels[i] == (const mDNSu8 *)mDNSNULL)
10930 {
10931 debugf("DomainEnumQuery: Not a DEQ %##s, label1 mismatch", qname->c);
10932 return mDNSfalse;
10933 }
10934 debugf("DomainEnumQuery: DEQ %##s, label1 match", qname->c);
10935
10936 // CountLabels already verified the number of labels
10937 d = (const domainname *)(d->c + 1 + d->c[0]); // Second Label
10938 label = (const mDNSu8 *)d;
10939 if (!SameDomainLabel(label, (const mDNSu8 *)"\007_dns-sd"))
10940 {
10941 debugf("DomainEnumQuery: Not a DEQ %##s, label2 mismatch", qname->c);
10942 return(mDNSfalse);
10943 }
10944 debugf("DomainEnumQuery: DEQ %##s, label2 match", qname->c);
10945
10946 d = (const domainname *)(d->c + 1 + d->c[0]); // Third Label
10947 label = (const mDNSu8 *)d;
10948 if (!SameDomainLabel(label, (const mDNSu8 *)"\004_udp"))
10949 {
10950 debugf("DomainEnumQuery: Not a DEQ %##s, label3 mismatch", qname->c);
10951 return(mDNSfalse);
10952 }
10953 debugf("DomainEnumQuery: DEQ %##s, label3 match", qname->c);
10954
10955 debugf("DomainEnumQuery: Question %##s is a Domain Enumeration query", qname->c);
10956
10957 return mDNStrue;
10958 }
10959
10960 // Note: InterfaceID is the InterfaceID of the question
10961 mDNSlocal mDNSBool DNSServerMatch(DNSServer *d, mDNSInterfaceID InterfaceID, mDNSs32 ServiceID)
10962 {
10963 // 1) Unscoped questions (NULL InterfaceID) should consider *only* unscoped DNSServers ( DNSServer
10964 // with "scoped" set to kScopeNone)
10965 //
10966 // 2) Scoped questions (non-NULL InterfaceID) should consider *only* scoped DNSServers (DNSServer
10967 // with "scoped" set to kScopeInterfaceId) and their InterfaceIDs should match.
10968 //
10969 // 3) Scoped questions (non-zero ServiceID) should consider *only* scoped DNSServers (DNSServer
10970 // with "scoped" set to kScopeServiceID) and their ServiceIDs should match.
10971 //
10972 // The first condition in the "if" statement checks to see if both the question and the DNSServer are
10973 // unscoped. The question is unscoped only if InterfaceID is zero and ServiceID is -1.
10974 //
10975 // If the first condition fails, following are the possible cases (the notes below are using
10976 // InterfaceID for discussion and the same holds good for ServiceID):
10977 //
10978 // - DNSServer is not scoped, InterfaceID is not NULL - we should skip the current DNSServer entry
10979 // as scoped questions should not pick non-scoped DNSServer entry (Refer to (2) above).
10980 //
10981 // - DNSServer is scoped, InterfaceID is NULL - we should skip the current DNSServer entry as
10982 // unscoped question should not match scoped DNSServer (Refer to (1) above). The InterfaceID check
10983 // would fail in this case.
10984 //
10985 // - DNSServer is scoped and InterfaceID is not NULL - the InterfaceID of the question and the DNSServer
10986 // should match (Refer to (2) above).
10987 //
10988 // Note: mDNSInterface_Unicast is used only by .local unicast questions and are treated as unscoped.
10989 // If a question is scoped both to InterfaceID and ServiceID, the question will be scoped to InterfaceID.
10990
10991 if (((d->scoped == kScopeNone) && ((!InterfaceID && ServiceID == -1) || InterfaceID == mDNSInterface_Unicast)) ||
10992 ((d->scoped == kScopeInterfaceID) && d->interface == InterfaceID) ||
10993 ((d->scoped == kScopeServiceID) && d->serviceID == ServiceID))
10994 {
10995 return mDNStrue;
10996 }
10997 return mDNSfalse;
10998 }
10999
11000 // Sets all the Valid DNS servers for a question
11001 mDNSexport mDNSu32 SetValidDNSServers(mDNS *m, DNSQuestion *question)
11002 {
11003 int bestmatchlen = -1, namecount = CountLabels(&question->qname);
11004 DNSServer *curr;
11005 int bettermatch, currcount;
11006 int index = 0;
11007 mDNSu32 timeout = 0;
11008 mDNSBool DEQuery;
11009
11010 question->validDNSServers = zeroOpaque64;
11011 DEQuery = DomainEnumQuery(&question->qname);
11012 for (curr = m->DNSServers; curr; curr = curr->next)
11013 {
11014 debugf("SetValidDNSServers: Parsing DNS server Address %#a (Domain %##s), Scope: %d", &curr->addr, curr->domain.c, curr->scoped);
11015 // skip servers that will soon be deleted
11016 if (curr->flags & DNSServer_FlagDelete)
11017 {
11018 debugf("SetValidDNSServers: Delete set for index %d, DNS server %#a (Domain %##s), scoped %d", index, &curr->addr, curr->domain.c, curr->scoped);
11019 continue;
11020 }
11021
11022 // This happens normally when you unplug the interface where we reset the interfaceID to mDNSInterface_Any for all
11023 // the DNS servers whose scope match the interfaceID. Few seconds later, we also receive the updated DNS configuration.
11024 // But any questions that has mDNSInterface_Any scope that are started/restarted before we receive the update
11025 // (e.g., CheckSuppressUnusableQuestions is called when interfaces are deregistered with the core) should not
11026 // match the scoped entries by mistake.
11027 //
11028 // Note: DNS configuration change will help pick the new dns servers but currently it does not affect the timeout
11029
11030 // Skip DNSServers that are InterfaceID Scoped but have no valid interfaceid set OR DNSServers that are ServiceID Scoped but have no valid serviceid set
11031 if ((curr->scoped == kScopeInterfaceID && curr->interface == mDNSInterface_Any) || (curr->scoped == kScopeServiceID && curr->serviceID <= 0))
11032 {
11033 LogInfo("SetValidDNSServers: ScopeType[%d] Skipping DNS server %#a (Domain %##s) Interface:[%p] Serviceid:[%d]", curr->scoped, &curr->addr, curr->domain.c, curr->interface, curr->serviceID);
11034 continue;
11035 }
11036
11037 currcount = CountLabels(&curr->domain);
11038 if ((!curr->cellIntf || (!DEQuery && !(question->flags & kDNSServiceFlagsDenyCellular))) &&
11039 DNSServerMatch(curr, question->InterfaceID, question->ServiceID))
11040 {
11041 bettermatch = BetterMatchForName(&question->qname, namecount, &curr->domain, currcount, bestmatchlen);
11042
11043 // If we found a better match (bettermatch == 1) then clear all the bits
11044 // corresponding to the old DNSServers that we have may set before and start fresh.
11045 // If we find an equal match, then include that DNSServer also by setting the corresponding
11046 // bit
11047 if ((bettermatch == 1) || (bettermatch == 0))
11048 {
11049 bestmatchlen = currcount;
11050 if (bettermatch)
11051 {
11052 debugf("SetValidDNSServers: Resetting all the bits");
11053 question->validDNSServers = zeroOpaque64;
11054 timeout = 0;
11055 }
11056 debugf("SetValidDNSServers: question %##s Setting the bit for DNS server Address %#a (Domain %##s), Scoped:%d index %d,"
11057 " Timeout %d, interface %p", question->qname.c, &curr->addr, curr->domain.c, curr->scoped, index, curr->timeout,
11058 curr->interface);
11059 timeout += curr->timeout;
11060 if (DEQuery)
11061 debugf("DomainEnumQuery: Question %##s, DNSServer %#a, cell %d", question->qname.c, &curr->addr, curr->cellIntf);
11062 bit_set_opaque64(question->validDNSServers, index);
11063 }
11064 }
11065 index++;
11066 }
11067 question->noServerResponse = 0;
11068
11069 debugf("SetValidDNSServers: ValidDNSServer bits 0x%x%x for question %p %##s (%s)",
11070 question->validDNSServers.l[1], question->validDNSServers.l[0], question, question->qname.c, DNSTypeName(question->qtype));
11071 // If there are no matching resolvers, then use the default timeout value.
11072 // For ProxyQuestion, shorten the timeout so that dig does not timeout on us in case of no response.
11073 return ((question->ProxyQuestion || question->ValidatingResponse) ? DEFAULT_UDNSSEC_TIMEOUT : timeout ? timeout : DEFAULT_UDNS_TIMEOUT);
11074 }
11075
11076 // Get the Best server that matches a name. If you find penalized servers, look for the one
11077 // that will come out of the penalty box soon
11078 mDNSlocal DNSServer *GetBestServer(mDNS *m, const domainname *name, mDNSInterfaceID InterfaceID, mDNSs32 ServiceID, mDNSOpaque64 validBits,
11079 int *selected, mDNSBool nameMatch)
11080 {
11081 DNSServer *curmatch = mDNSNULL;
11082 int bestmatchlen = -1, namecount = name ? CountLabels(name) : 0;
11083 DNSServer *curr;
11084 mDNSs32 bestPenaltyTime, currPenaltyTime;
11085 int bettermatch, currcount;
11086 int index = 0;
11087 int currindex = -1;
11088
11089 debugf("GetBestServer: ValidDNSServer bits 0x%x%x", validBits.l[1], validBits.l[0]);
11090 bestPenaltyTime = DNSSERVER_PENALTY_TIME + 1;
11091 for (curr = m->DNSServers; curr; curr = curr->next)
11092 {
11093 // skip servers that will soon be deleted
11094 if (curr->flags & DNSServer_FlagDelete)
11095 {
11096 debugf("GetBestServer: Delete set for index %d, DNS server %#a (Domain %##s), scoped %d", index, &curr->addr, curr->domain.c, curr->scoped);
11097 continue;
11098 }
11099
11100 // Check if this is a valid DNSServer
11101 if (!bit_get_opaque64(validBits, index))
11102 {
11103 debugf("GetBestServer: continuing for index %d", index);
11104 index++;
11105 continue;
11106 }
11107
11108 currcount = CountLabels(&curr->domain);
11109 currPenaltyTime = PenaltyTimeForServer(m, curr);
11110
11111 debugf("GetBestServer: Address %#a (Domain %##s), PenaltyTime(abs) %d, PenaltyTime(rel) %d",
11112 &curr->addr, curr->domain.c, curr->penaltyTime, currPenaltyTime);
11113
11114 // If there are multiple best servers for a given question, we will pick the first one
11115 // if none of them are penalized. If some of them are penalized in that list, we pick
11116 // the least penalized one. BetterMatchForName walks through all best matches and
11117 // "currPenaltyTime < bestPenaltyTime" check lets us either pick the first best server
11118 // in the list when there are no penalized servers and least one among them
11119 // when there are some penalized servers.
11120
11121 if (DNSServerMatch(curr, InterfaceID, ServiceID))
11122 {
11123
11124 // If we know that all the names are already equally good matches, then skip calling BetterMatchForName.
11125 // This happens when we initially walk all the DNS servers and set the validity bit on the question.
11126 // Actually we just need PenaltyTime match, but for the sake of readability we just skip the expensive
11127 // part and still do some redundant steps e.g., InterfaceID match
11128
11129 if (nameMatch)
11130 bettermatch = BetterMatchForName(name, namecount, &curr->domain, currcount, bestmatchlen);
11131 else
11132 bettermatch = 0;
11133
11134 // If we found a better match (bettermatch == 1) then we don't need to
11135 // compare penalty times. But if we found an equal match, then we compare
11136 // the penalty times to pick a better match
11137
11138 if ((bettermatch == 1) || ((bettermatch == 0) && currPenaltyTime < bestPenaltyTime))
11139 {
11140 currindex = index;
11141 curmatch = curr;
11142 bestmatchlen = currcount;
11143 bestPenaltyTime = currPenaltyTime;
11144 }
11145 }
11146 index++;
11147 }
11148 if (selected) *selected = currindex;
11149 return curmatch;
11150 }
11151
11152 // Look up a DNS Server, matching by name and InterfaceID
11153 mDNSlocal DNSServer *GetServerForName(mDNS *m, const domainname *name, mDNSInterfaceID InterfaceID, mDNSs32 ServiceID)
11154 {
11155 DNSServer *curmatch = mDNSNULL;
11156 char *ifname = mDNSNULL; // for logging purposes only
11157 mDNSOpaque64 allValid;
11158
11159 if ((InterfaceID == mDNSInterface_Unicast) || (InterfaceID == mDNSInterface_LocalOnly))
11160 InterfaceID = mDNSNULL;
11161
11162 if (InterfaceID) ifname = InterfaceNameForID(m, InterfaceID);
11163
11164 // By passing in all ones, we make sure that every DNS server is considered
11165 allValid.l[0] = allValid.l[1] = 0xFFFFFFFF;
11166
11167 curmatch = GetBestServer(m, name, InterfaceID, ServiceID, allValid, mDNSNULL, mDNStrue);
11168
11169 if (curmatch != mDNSNULL)
11170 LogInfo("GetServerForName: DNS server %#a:%d (Penalty Time Left %d) (Scope %s:%p) found for name %##s", &curmatch->addr,
11171 mDNSVal16(curmatch->port), (curmatch->penaltyTime ? (curmatch->penaltyTime - m->timenow) : 0), ifname ? ifname : "None",
11172 InterfaceID, name);
11173 else
11174 LogInfo("GetServerForName: no DNS server (Scope %s:%p) found for name %##s", ifname ? ifname : "None", InterfaceID, name);
11175
11176 return(curmatch);
11177 }
11178
11179 // Look up a DNS Server for a question within its valid DNSServer bits
11180 mDNSexport DNSServer *GetServerForQuestion(mDNS *m, DNSQuestion *question)
11181 {
11182 DNSServer *curmatch = mDNSNULL;
11183 char *ifname = mDNSNULL; // for logging purposes only
11184 mDNSInterfaceID InterfaceID = question->InterfaceID;
11185 const domainname *name = &question->qname;
11186 int currindex;
11187
11188 if ((InterfaceID == mDNSInterface_Unicast) || (InterfaceID == mDNSInterface_LocalOnly))
11189 InterfaceID = mDNSNULL;
11190
11191 if (InterfaceID)
11192 ifname = InterfaceNameForID(m, InterfaceID);
11193
11194 if (!mDNSOpaque64IsZero(&question->validDNSServers))
11195 {
11196 curmatch = GetBestServer(m, name, InterfaceID, question->ServiceID, question->validDNSServers, &currindex, mDNSfalse);
11197 if (currindex != -1)
11198 bit_clr_opaque64(question->validDNSServers, currindex);
11199 }
11200
11201 if (curmatch != mDNSNULL)
11202 {
11203 LogInfo("GetServerForQuestion: %p DNS server (%p) %#a:%d (Penalty Time Left %d) (Scope %s:%p:%d) found for name %##s (%s)",
11204 question, curmatch, &curmatch->addr, mDNSVal16(curmatch->port),
11205 (curmatch->penaltyTime ? (curmatch->penaltyTime - m->timenow) : 0), ifname ? ifname : "None",
11206 InterfaceID, question->ServiceID, name, DNSTypeName(question->qtype));
11207 }
11208 else
11209 {
11210 LogInfo("GetServerForQuestion: %p no DNS server (Scope %s:%p:%d) found for name %##s (%s)",
11211 question, ifname ? ifname : "None", InterfaceID, question->ServiceID, name, DNSTypeName(question->qtype));
11212 }
11213
11214 return(curmatch);
11215 }
11216
11217
11218 #define ValidQuestionTarget(Q) (((Q)->Target.type == mDNSAddrType_IPv4 || (Q)->Target.type == mDNSAddrType_IPv6) && \
11219 (mDNSSameIPPort((Q)->TargetPort, UnicastDNSPort) || mDNSSameIPPort((Q)->TargetPort, MulticastDNSPort)))
11220
11221 // Called in normal client context (lock not held)
11222 mDNSlocal void LLQNATCallback(mDNS *m, NATTraversalInfo *n)
11223 {
11224 DNSQuestion *q;
11225 mDNS_Lock(m);
11226 LogInfo("LLQNATCallback external address:port %.4a:%u, NAT result %d", &n->ExternalAddress, mDNSVal16(n->ExternalPort), n->Result);
11227 n->clientContext = mDNSNULL; // we received at least one callback since starting this NAT-T
11228 for (q = m->Questions; q; q=q->next)
11229 if (ActiveQuestion(q) && !mDNSOpaque16IsZero(q->TargetQID) && q->LongLived)
11230 startLLQHandshake(m, q); // If ExternalPort is zero, will do StartLLQPolling instead
11231 #if APPLE_OSX_mDNSResponder
11232 UpdateAutoTunnelDomainStatuses(m);
11233 #endif
11234 mDNS_Unlock(m);
11235 }
11236
11237 mDNSlocal mDNSBool IsPrivateDomain(mDNS *const m, DNSQuestion *q)
11238 {
11239 DomainAuthInfo *AuthInfo;
11240 // Skip Private domains as we have special addresses to get the hosts in the Private domain
11241 AuthInfo = GetAuthInfoForName_internal(m, &q->qname);
11242 if (AuthInfo && !AuthInfo->deltime && AuthInfo->AutoTunnel)
11243 {
11244 debugf("IsPrivateDomain: %##s true", q->qname.c);
11245 return mDNStrue;
11246 }
11247 else
11248 {
11249 debugf("IsPrivateDomain: %##s false", q->qname.c);
11250 return mDNSfalse;
11251 }
11252 }
11253
11254 // This function takes the DNSServer as a separate argument because sometimes the
11255 // caller has not yet assigned the DNSServer, but wants to evaluate the SuppressQuery
11256 // status before switching to it.
11257 mDNSlocal mDNSBool ShouldSuppressUnicastQuery(mDNS *const m, DNSQuestion *q, DNSServer *d)
11258 {
11259 // Some callers don't check for the qtype
11260 if (q->qtype != kDNSType_A && q->qtype != kDNSType_AAAA)
11261 {
11262 LogInfo("ShouldSuppressUnicastQuery: Query not suppressed for %##s, qtype %s, not A/AAAA type", q->qname.c, DNSTypeName(q->qtype));
11263 return mDNSfalse;
11264 }
11265
11266 // Private domains are exempted irrespective of what the DNSServer says
11267 if (IsPrivateDomain(m, q))
11268 {
11269 LogInfo("ShouldSuppressUnicastQuery: Query not suppressed for %##s, qtype %s, Private Domain", q->qname.c, DNSTypeName(q->qtype));
11270 return mDNSfalse;
11271 }
11272
11273 if (!d)
11274 {
11275 LogInfo("ShouldSuppressUnicastQuery: Query suppressed for %##s, qtype %s, as the DNS server is NULL", q->qname.c, DNSTypeName(q->qtype));
11276 return mDNStrue;
11277 }
11278
11279 // Check if the DNS Configuration allows A/AAAA queries to be sent
11280 if ((q->qtype == kDNSType_A) && (d->req_A))
11281 {
11282 LogInfo("ShouldSuppressUnicastQuery: Query not suppressed for %##s, qtype %s, DNSServer %##s %#a:%d allows A queries", q->qname.c,
11283 DNSTypeName(q->qtype), d->domain.c, &d->addr, mDNSVal16(d->port));
11284 return mDNSfalse;
11285 }
11286 if ((q->qtype == kDNSType_AAAA) && (d->req_AAAA))
11287 {
11288 LogInfo("ShouldSuppressUnicastQuery: Query not suppressed for %##s, qtype %s, DNSServer %##s %#a:%d allows AAAA queries", q->qname.c,
11289 DNSTypeName(q->qtype), d->domain.c, &d->addr, mDNSVal16(d->port));
11290 return mDNSfalse;
11291 }
11292
11293 LogInfo("ShouldSuppressUnicastQuery: Query suppressed for %##s, qtype %s, since DNS Configuration does not allow (req_A is %s and req_AAAA is %s)",
11294 q->qname.c, DNSTypeName(q->qtype), d->req_A ? "true" : "false", d->req_AAAA ? "true" : "false");
11295
11296 return mDNStrue;
11297 }
11298
11299 mDNSlocal mDNSBool ShouldSuppressDotLocalQuery(mDNS *const m, DNSQuestion *q)
11300 {
11301 NetworkInterfaceInfo *intf;
11302 AuthRecord *rr;
11303 mDNSBool ret;
11304
11305 // Check to see if there is at least one interface other than loopback and don't suppress
11306 // .local questions if you find one. If we have at least one interface, it means that
11307 // we can send unicast queries for the .local name and we don't want to suppress
11308 // multicast in that case as upper layers don't know how to handle if we return a
11309 // negative response for multicast followed by a positive response for unicast.
11310 //
11311 // Note: we used to check for multicast capable interfaces instead of just any interface
11312 // present. That did not work in the case where we have a valid interface for unicast
11313 // but not multicast capable e.g., cellular, as we ended up delivering a negative response
11314 // first and the upper layer did not wait for the positive response that came later.
11315 for (intf = m->HostInterfaces; intf; intf = intf->next)
11316 {
11317 if (intf->InterfaceActive && !intf->Loopback)
11318 {
11319 LogInfo("ShouldSuppressDotLocalQuery: Found interface %s, not suppressing", intf->ifname);
11320 return mDNSfalse;
11321 }
11322 }
11323
11324 // 1. If we find a LocalOnly or P2P record answering this question, then don't suppress it.
11325 // Set m->CurrentQuestion as it is required by AnswerQuestionWithLORecord.
11326 m->CurrentQuestion = q;
11327 ret = AnswerQuestionWithLORecord(m, q, mDNStrue);
11328 m->CurrentQuestion = mDNSNULL;
11329
11330 if (ret)
11331 {
11332 LogInfo("ShouldSuppressDotLocalQuery: Found LocalOnly record for %##s (%s), not suppressing", q->qname.c,
11333 DNSTypeName(q->qtype));
11334 return mDNSfalse;
11335 }
11336
11337 // 2. If we find a local AuthRecord answering this question, then don't suppress it.
11338 for (rr = m->ResourceRecords; rr; rr = rr->next)
11339 {
11340 if (ResourceRecordAnswersQuestion(&rr->resrec, q))
11341 {
11342 LogInfo("ShouldSuppressDotLocalQuery: Found resource record %s for %##s (%s) not suppressing", ARDisplayString(m, rr),
11343 q->qname.c, DNSTypeName(q->qtype));
11344 return mDNSfalse;
11345 }
11346 }
11347 return mDNStrue;
11348 }
11349
11350 mDNSlocal mDNSBool ShouldSuppressQuery(mDNS *const m, DNSQuestion *q)
11351 {
11352 if (q->InterfaceID == mDNSInterface_LocalOnly)
11353 {
11354 LogInfo("ShouldSuppressQuery: LocalOnly query not suppressed for %##s, qtype %s", q->qname.c, DNSTypeName(q->qtype));
11355 return mDNSfalse;
11356 }
11357
11358 if (q->qtype != kDNSType_A && q->qtype != kDNSType_AAAA)
11359 {
11360 LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, not A/AAAA type", q->qname.c, DNSTypeName(q->qtype));
11361 return mDNSfalse;
11362 }
11363
11364 // We still want the ability to be able to listen to the local services and hence
11365 // don't fail .local query if we have local records that can potentially answer
11366 // the question.
11367 if (q->InterfaceID != mDNSInterface_Unicast && IsLocalDomain(&q->qname))
11368 {
11369 if (!ShouldSuppressDotLocalQuery(m, q))
11370 {
11371 LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, Local question", q->qname.c, DNSTypeName(q->qtype));
11372 return mDNSfalse;
11373 }
11374 else
11375 {
11376 LogInfo("ShouldSuppressQuery: Query suppressed for %##s, qtype %s, Local question", q->qname.c, DNSTypeName(q->qtype));
11377 return mDNStrue;
11378 }
11379 }
11380
11381 return (ShouldSuppressUnicastQuery(m, q, q->qDNSServer));
11382 }
11383
11384 mDNSlocal void CacheRecordRmvEventsForCurrentQuestion(mDNS *const m, DNSQuestion *q)
11385 {
11386 CacheRecord *rr;
11387 mDNSu32 slot;
11388 CacheGroup *cg;
11389
11390 slot = HashSlot(&q->qname);
11391 cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
11392 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
11393 {
11394 // Don't deliver RMV events for negative records
11395 if (rr->resrec.RecordType == kDNSRecordTypePacketNegative)
11396 {
11397 LogInfo("CacheRecordRmvEventsForCurrentQuestion: CacheRecord %s Suppressing RMV events for question %p %##s (%s), CRActiveQuestion %p, CurrentAnswers %d",
11398 CRDisplayString(m, rr), q, q->qname.c, DNSTypeName(q->qtype), rr->CRActiveQuestion, q->CurrentAnswers);
11399 continue;
11400 }
11401
11402 if (SameNameRecordAnswersQuestion(&rr->resrec, q))
11403 {
11404 LogInfo("CacheRecordRmvEventsForCurrentQuestion: Calling AnswerCurrentQuestionWithResourceRecord (RMV) for question %##s using resource record %s LocalAnswers %d",
11405 q->qname.c, CRDisplayString(m, rr), q->LOAddressAnswers);
11406
11407 q->CurrentAnswers--;
11408 if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers--;
11409 if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers--;
11410
11411 if (rr->CRActiveQuestion == q)
11412 {
11413 DNSQuestion *qptr;
11414 // If this was the active question for this cache entry, it was the one that was
11415 // responsible for keeping the cache entry fresh when the cache entry was reaching
11416 // its expiry. We need to handover the responsibility to someone else. Otherwise,
11417 // when the cache entry is about to expire, we won't find an active question
11418 // (pointed by CRActiveQuestion) to refresh the cache.
11419 for (qptr = m->Questions; qptr; qptr=qptr->next)
11420 if (qptr != q && ActiveQuestion(qptr) && ResourceRecordAnswersQuestion(&rr->resrec, qptr))
11421 break;
11422
11423 if (qptr)
11424 LogInfo("CacheRecordRmvEventsForCurrentQuestion: Updating CRActiveQuestion to %p for cache record %s, "
11425 "Original question CurrentAnswers %d, new question CurrentAnswers %d, SuppressUnusable %d, SuppressQuery %d",
11426 qptr, CRDisplayString(m,rr), q->CurrentAnswers, qptr->CurrentAnswers, qptr->SuppressUnusable, qptr->SuppressQuery);
11427
11428 rr->CRActiveQuestion = qptr; // Question used to be active; new value may or may not be null
11429 if (!qptr) m->rrcache_active--; // If no longer active, decrement rrcache_active count
11430 }
11431 AnswerCurrentQuestionWithResourceRecord(m, rr, QC_rmv);
11432 if (m->CurrentQuestion != q) break; // If callback deleted q, then we're finished here
11433 }
11434 }
11435 }
11436
11437 mDNSlocal mDNSBool IsQuestionNew(mDNS *const m, DNSQuestion *question)
11438 {
11439 DNSQuestion *q;
11440 for (q = m->NewQuestions; q; q = q->next)
11441 if (q == question) return mDNStrue;
11442 return mDNSfalse;
11443 }
11444
11445 mDNSlocal mDNSBool LocalRecordRmvEventsForQuestion(mDNS *const m, DNSQuestion *q)
11446 {
11447 AuthRecord *rr;
11448 mDNSu32 slot;
11449 AuthGroup *ag;
11450
11451 if (m->CurrentQuestion)
11452 LogMsg("LocalRecordRmvEventsForQuestion: ERROR m->CurrentQuestion already set: %##s (%s)",
11453 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
11454
11455 if (IsQuestionNew(m, q))
11456 {
11457 LogInfo("LocalRecordRmvEventsForQuestion: New Question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
11458 return mDNStrue;
11459 }
11460 m->CurrentQuestion = q;
11461 slot = AuthHashSlot(&q->qname);
11462 ag = AuthGroupForName(&m->rrauth, slot, q->qnamehash, &q->qname);
11463 if (ag)
11464 {
11465 for (rr = ag->members; rr; rr=rr->next)
11466 // Filter the /etc/hosts records - LocalOnly, Unique, A/AAAA/CNAME
11467 if (UniqueLocalOnlyRecord(rr) && LocalOnlyRecordAnswersQuestion(rr, q))
11468 {
11469 LogInfo("LocalRecordRmvEventsForQuestion: Delivering possible Rmv events with record %s",
11470 ARDisplayString(m, rr));
11471 if (q->CurrentAnswers <= 0 || q->LOAddressAnswers <= 0)
11472 {
11473 LogMsg("LocalRecordRmvEventsForQuestion: ERROR!! CurrentAnswers or LOAddressAnswers is zero %p %##s"
11474 " (%s) CurrentAnswers %d, LOAddressAnswers %d", q, q->qname.c, DNSTypeName(q->qtype),
11475 q->CurrentAnswers, q->LOAddressAnswers);
11476 continue;
11477 }
11478 AnswerLocalQuestionWithLocalAuthRecord(m, rr, QC_rmv); // MUST NOT dereference q again
11479 if (m->CurrentQuestion != q) { m->CurrentQuestion = mDNSNULL; return mDNSfalse; }
11480 }
11481 }
11482 m->CurrentQuestion = mDNSNULL;
11483 return mDNStrue;
11484 }
11485
11486 // Returns false if the question got deleted while delivering the RMV events
11487 // The caller should handle the case
11488 mDNSlocal mDNSBool CacheRecordRmvEventsForQuestion(mDNS *const m, DNSQuestion *q)
11489 {
11490 if (m->CurrentQuestion)
11491 LogMsg("CacheRecordRmvEventsForQuestion: ERROR m->CurrentQuestion already set: %##s (%s)",
11492 m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
11493
11494 // If it is a new question, we have not delivered any ADD events yet. So, don't deliver RMV events.
11495 // If this question was answered using local auth records, then you can't deliver RMVs using cache
11496 if (!IsQuestionNew(m, q) && !q->LOAddressAnswers)
11497 {
11498 m->CurrentQuestion = q;
11499 CacheRecordRmvEventsForCurrentQuestion(m, q);
11500 if (m->CurrentQuestion != q) { m->CurrentQuestion = mDNSNULL; return mDNSfalse; }
11501 m->CurrentQuestion = mDNSNULL;
11502 }
11503 else { LogInfo("CacheRecordRmvEventsForQuestion: Question %p %##s (%s) is a new question", q, q->qname.c, DNSTypeName(q->qtype)); }
11504 return mDNStrue;
11505 }
11506
11507 mDNSlocal void SuppressStatusChanged(mDNS *const m, DNSQuestion *q, DNSQuestion **restart)
11508 {
11509 // NOTE: CacheRecordRmvEventsForQuestion will not generate RMV events for queries that have non-zero
11510 // LOAddressAnswers. Hence it is important that we call CacheRecordRmvEventsForQuestion before
11511 // LocalRecordRmvEventsForQuestion (which decrements LOAddressAnswers)
11512 if (q->SuppressQuery)
11513 {
11514 q->SuppressQuery = mDNSfalse;
11515 if (!CacheRecordRmvEventsForQuestion(m, q))
11516 {
11517 LogInfo("SuppressStatusChanged: Question deleted while delivering RMV events from cache");
11518 return;
11519 }
11520 q->SuppressQuery = mDNStrue;
11521 }
11522
11523 // SuppressUnusable does not affect questions that are answered from the local records (/etc/hosts)
11524 // and SuppressQuery status does not mean anything for these questions. As we are going to stop the
11525 // question below, we need to deliver the RMV events so that the ADDs that will be delivered during
11526 // the restart will not be a duplicate ADD
11527 if (!LocalRecordRmvEventsForQuestion(m, q))
11528 {
11529 LogInfo("SuppressStatusChanged: Question deleted while delivering RMV events from Local AuthRecords");
11530 return;
11531 }
11532
11533 // There are two cases here.
11534 //
11535 // 1. Previously it was suppressed and now it is not suppressed, restart the question so
11536 // that it will start as a new question. Note that we can't just call ActivateUnicastQuery
11537 // because when we get the response, if we had entries in the cache already, it will not answer
11538 // this question if the cache entry did not change. Hence, we need to restart
11539 // the query so that it can be answered from the cache.
11540 //
11541 // 2. Previously it was not suppressed and now it is suppressed. We need to restart the questions
11542 // so that we redo the duplicate checks in mDNS_StartQuery_internal. A SuppressUnusable question
11543 // is a duplicate of non-SuppressUnusable question if it is not suppressed (SuppressQuery is false).
11544 // A SuppressUnusable question is not a duplicate of non-SuppressUnusable question if it is suppressed
11545 // (SuppressQuery is true). The reason for this is that when a question is suppressed, we want an
11546 // immediate response and not want to be blocked behind a question that is querying DNS servers. When
11547 // the question is not suppressed, we don't want two active questions sending packets on the wire.
11548 // This affects both efficiency and also the current design where there is only one active question
11549 // pointed to from a cache entry.
11550 //
11551 // We restart queries in a two step process by first calling stop and build a temporary list which we
11552 // will restart at the end. The main reason for the two step process is to handle duplicate questions.
11553 // If there are duplicate questions, calling stop inherits the values from another question on the list (which
11554 // will soon become the real question) including q->ThisQInterval which might be zero if it was
11555 // suppressed before. At the end when we have restarted all questions, none of them is active as each
11556 // inherits from one another and we need to reactivate one of the questions here which is a little hacky.
11557 //
11558 // It is much cleaner and less error prone to build a list of questions and restart at the end.
11559
11560 LogInfo("SuppressStatusChanged: Stop question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
11561 mDNS_StopQuery_internal(m, q);
11562 q->next = *restart;
11563 *restart = q;
11564 }
11565
11566 // The caller should hold the lock
11567 mDNSexport void CheckSuppressUnusableQuestions(mDNS *const m)
11568 {
11569 DNSQuestion *q;
11570 DNSQuestion *restart = mDNSNULL;
11571
11572 // We look through all questions including new questions. During network change events,
11573 // we potentially restart questions here in this function that ends up as new questions,
11574 // which may be suppressed at this instance. Before it is handled we get another network
11575 // event that changes the status e.g., address becomes available. If we did not process
11576 // new questions, we would never change its SuppressQuery status.
11577 //
11578 // CurrentQuestion is used by RmvEventsForQuestion below. While delivering RMV events, the
11579 // application callback can potentially stop the current question (detected by CurrentQuestion) or
11580 // *any* other question which could be the next one that we may process here. RestartQuestion
11581 // points to the "next" question which will be automatically advanced in mDNS_StopQuery_internal
11582 // if the "next" question is stopped while the CurrentQuestion is stopped
11583 if (m->RestartQuestion)
11584 LogMsg("CheckSuppressUnusableQuestions: ERROR!! m->RestartQuestion already set: %##s (%s)",
11585 m->RestartQuestion->qname.c, DNSTypeName(m->RestartQuestion->qtype));
11586 m->RestartQuestion = m->Questions;
11587 while (m->RestartQuestion)
11588 {
11589 q = m->RestartQuestion;
11590 m->RestartQuestion = q->next;
11591 if (q->SuppressUnusable)
11592 {
11593 mDNSBool old = q->SuppressQuery;
11594 q->SuppressQuery = ShouldSuppressQuery(m, q);
11595 if (q->SuppressQuery != old)
11596 {
11597 // Previously it was not suppressed, Generate RMV events for the ADDs that we might have delivered before
11598 // followed by a negative cache response. Temporarily turn off suppression so that
11599 // AnswerCurrentQuestionWithResourceRecord can answer the question
11600 SuppressStatusChanged(m, q, &restart);
11601 }
11602 }
11603 }
11604 while (restart)
11605 {
11606 q = restart;
11607 restart = restart->next;
11608 q->next = mDNSNULL;
11609 LogInfo("CheckSuppressUnusableQuestions: Start question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
11610 mDNS_StartQuery_internal(m, q);
11611 }
11612 }
11613
11614 mDNSlocal void RestartUnicastQuestions(mDNS *const m)
11615 {
11616 DNSQuestion *q;
11617 DNSQuestion *restart = mDNSNULL;
11618
11619 if (m->RestartQuestion)
11620 LogMsg("RestartUnicastQuestions: ERROR!! m->RestartQuestion already set: %##s (%s)",
11621 m->RestartQuestion->qname.c, DNSTypeName(m->RestartQuestion->qtype));
11622 m->RestartQuestion = m->Questions;
11623 while (m->RestartQuestion)
11624 {
11625 q = m->RestartQuestion;
11626 m->RestartQuestion = q->next;
11627 if (q->Restart)
11628 {
11629 if (mDNSOpaque16IsZero(q->TargetQID))
11630 LogMsg("RestartUnicastQuestions: ERROR!! Restart set for multicast question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
11631
11632 q->Restart = 0;
11633 SuppressStatusChanged(m, q, &restart);
11634 }
11635 }
11636 while (restart)
11637 {
11638 q = restart;
11639 restart = restart->next;
11640 q->next = mDNSNULL;
11641 LogInfo("RestartUnicastQuestions: Start question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
11642 mDNS_StartQuery_internal(m, q);
11643 }
11644 }
11645
11646
11647 // ValidateParameters() is called by mDNS_StartQuery_internal() to check the client parameters of
11648 // DNS Question that are already set by the client before calling mDNS_StartQuery()
11649 mDNSlocal mStatus ValidateParameters(mDNS *const m, DNSQuestion *const question)
11650 {
11651
11652 if (question->Target.type && !ValidQuestionTarget(question))
11653 {
11654 LogMsg("ValidateParameters: Warning! Target.type = %ld port = %u (Client forgot to initialize before calling mDNS_StartQuery? for question %##s)",
11655 question->Target.type, mDNSVal16(question->TargetPort), question->qname.c);
11656 question->Target.type = mDNSAddrType_None;
11657 }
11658
11659 // If no question->Target specified, clear TargetPort
11660 if (!question->Target.type)
11661 question->TargetPort = zeroIPPort;
11662
11663 if (!ValidateDomainName(&question->qname))
11664 {
11665 LogMsg("ValidateParameters: Attempt to start query with invalid qname %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
11666 return(mStatus_Invalid);
11667 }
11668
11669 // If this question is referencing a specific interface, verify it exists
11670 if (question->InterfaceID && !LocalOnlyOrP2PInterface(question->InterfaceID) && question->InterfaceID != mDNSInterface_Unicast)
11671 {
11672 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, question->InterfaceID);
11673 if (!intf)
11674 LogInfo("ValidateParameters: Note: InterfaceID %d for question %##s (%s) not currently found in active interface list",
11675 (uint32_t)question->InterfaceID, question->qname.c, DNSTypeName(question->qtype));
11676 }
11677
11678 return(mStatus_NoError);
11679 }
11680
11681 // InitDNSConfig() is called by InitCommonState() to initialize the DNS configuration of the Question.
11682 // These are a subset of the internal uDNS fields. Must be done before ShouldSuppressQuery() & mDNS_PurgeForQuestion()
11683 mDNSlocal void InitDNSConfig(mDNS *const m, DNSQuestion *const question)
11684 {
11685 // First reset all DNS Configuration
11686 question->qDNSServer = mDNSNULL;
11687 question->validDNSServers = zeroOpaque64;
11688 question->triedAllServersOnce = 0;
11689 question->noServerResponse = 0;
11690 question->StopTime = 0;
11691 #if TARGET_OS_EMBEDDED
11692 mDNSPlatformMemZero(&question->metrics, sizeof(question->metrics));
11693 #endif
11694
11695 // Need not initialize the DNS Configuration for Local Only OR P2P Questions
11696 if (LocalOnlyOrP2PInterface(question->InterfaceID))
11697 return;
11698 // Proceed to initialize DNS Configuration (some are set in SetValidDNSServers())
11699 if (!mDNSOpaque16IsZero(question->TargetQID))
11700 {
11701 mDNSu32 timeout = SetValidDNSServers(m, question);
11702 // We set the timeout whenever mDNS_StartQuery_internal is called. This means if we have
11703 // a networking change/search domain change that calls this function again we keep
11704 // reinitializing the timeout value which means it may never timeout. If this becomes
11705 // a common case in the future, we can easily fix this by adding extra state that
11706 // indicates that we have already set the StopTime.
11707 //
11708 // Note that we set the timeout for all questions. If this turns out to be a duplicate,
11709 // it gets a full timeout value even if the original question times out earlier.
11710 if (question->TimeoutQuestion)
11711 {
11712 question->StopTime = NonZeroTime(m->timenow + timeout * mDNSPlatformOneSecond);
11713 LogInfo("InitDNSConfig: Setting StopTime on question %p %##s (%s)", question, question->qname.c, DNSTypeName(question->qtype));
11714 }
11715
11716 question->qDNSServer = GetServerForQuestion(m, question);
11717 LogInfo("InitDNSConfig: question %p %##s (%s) Timeout %d, DNS Server %#a:%d",
11718 question, question->qname.c, DNSTypeName(question->qtype), timeout,
11719 question->qDNSServer ? &question->qDNSServer->addr : mDNSNULL,
11720 mDNSVal16(question->qDNSServer ? question->qDNSServer->port : zeroIPPort));
11721 }
11722 else
11723 {
11724 if (question->TimeoutQuestion)
11725 question->StopTime = NonZeroTime(m->timenow + GetTimeoutForMcastQuestion(m, question) * mDNSPlatformOneSecond);
11726 }
11727 // Set StopTime here since it is a part of DNS Configuration
11728 if (question->StopTime)
11729 SetNextQueryStopTime(m, question);
11730 // SetNextQueryTime() need not be initialized for LocalOnly OR P2P Questions since those questions
11731 // will never be transmitted on the wire. Hence we call SetNextQueryTime() here.
11732 SetNextQueryTime(m,question);
11733 }
11734
11735 // InitCommonState() is called by mDNS_StartQuery_internal() to initialize the common(uDNS/mDNS) internal
11736 // state fields of the DNS Question. These are independent of the Client layer.
11737 mDNSlocal mDNSBool InitCommonState(mDNS *const m, DNSQuestion *const question)
11738 {
11739 mDNSBool purge;
11740 int i;
11741 mDNSBool isBlocked = mDNSfalse;
11742
11743 // Note: In the case where we already have the answer to this question in our cache, that may be all the client
11744 // wanted, and they may immediately cancel their question. In this case, sending an actual query on the wire would
11745 // be a waste. For that reason, we schedule our first query to go out in half a second (InitialQuestionInterval).
11746 // If AnswerNewQuestion() finds that we have *no* relevant answers currently in our cache, then it will accelerate
11747 // that to go out immediately.
11748 question->next = mDNSNULL;
11749 // ThisQInterval should be initialized before any memory allocations occur. If malloc
11750 // debugging is turned on within mDNSResponder (see mDNSDebug.h for details) it validates
11751 // the question list to check if ThisQInterval is negative which means the question has been
11752 // stopped and can't be on the list. The question is already on the list and ThisQInterval
11753 // can be negative if the caller just stopped it and starting it again. Hence, it always has to
11754 // be initialized. CheckForSoonToExpireRecords below prints the cache records when logging is
11755 // turned ON which can allocate memory e.g., base64 encoding, in the case of DNSSEC.
11756 question->ThisQInterval = InitialQuestionInterval; // MUST be > zero for an active question
11757 question->qnamehash = DomainNameHashValue(&question->qname);
11758 question->DelayAnswering = CheckForSoonToExpireRecords(m, &question->qname, question->qnamehash, HashSlot(&question->qname), &purge);
11759 question->LastQTime = m->timenow;
11760 question->ExpectUnicastResp = 0;
11761 question->LastAnswerPktNum = m->PktNum;
11762 question->RecentAnswerPkts = 0;
11763 question->CurrentAnswers = 0;
11764
11765 #if APPLE_OSX_mDNSResponder
11766
11767 // Initial browse threshold used by Finder.
11768 #define mDNSFinderBrowseThreshold 20
11769
11770 // Set the threshold at which we move to a passive browse state,
11771 // not actively sending queries.
11772 if (question->flags & kDNSServiceFlagsThresholdOne)
11773 question->BrowseThreshold = 1;
11774 else if (question->flags & kDNSServiceFlagsThresholdFinder)
11775 question->BrowseThreshold = mDNSFinderBrowseThreshold;
11776 else
11777 question->BrowseThreshold = 0;
11778
11779 #else // APPLE_OSX_mDNSResponder
11780 question->BrowseThreshold = 0;
11781 #endif // APPLE_OSX_mDNSResponder
11782 question->CachedAnswerNeedsUpdate = mDNSfalse;
11783
11784 question->LargeAnswers = 0;
11785 question->UniqueAnswers = 0;
11786 question->LOAddressAnswers = 0;
11787 question->FlappingInterface1 = mDNSNULL;
11788 question->FlappingInterface2 = mDNSNULL;
11789
11790 // if kDNSServiceFlagsServiceIndex flag is SET by the client, then do NOT call mDNSPlatformGetDNSRoutePolicy()
11791 // since we would already have the question->ServiceID in that case.
11792 if (!(question->flags & kDNSServiceFlagsServiceIndex))
11793 {
11794 #if APPLE_OSX_mDNSResponder
11795 mDNSPlatformGetDNSRoutePolicy(m, question, &isBlocked);
11796 #else
11797 question->ServiceID = -1;
11798 #endif
11799 }
11800 else
11801 LogInfo("InitCommonState: Query for %##s (%s), PID[%d], EUID[%d], ServiceID[%d] is already set by client", question->qname.c,
11802 DNSTypeName(question->qtype), question->pid, question->euid, question->ServiceID);
11803
11804 InitDNSConfig(m, question);
11805
11806 question->AuthInfo = GetAuthInfoForQuestion(m, question);
11807 question->SuppressQuery = 0;
11808 if (question->SuppressUnusable)
11809 question->SuppressQuery = ShouldSuppressQuery(m, question);
11810
11811 // If ServiceID is 0 or the policy disallows making DNS requests,
11812 // set DisallowPID
11813 question->DisallowPID = (question->ServiceID == 0 || isBlocked);
11814 if (question->DisallowPID)
11815 LogInfo("InitCommonState: Query suppressed for %##s (%s), PID %d/ServiceID %d not allowed", question->qname.c,
11816 DNSTypeName(question->qtype), question->pid, question->ServiceID);
11817
11818 question->NextInDQList = mDNSNULL;
11819 question->SendQNow = mDNSNULL;
11820 question->SendOnAll = mDNSfalse;
11821 question->RequestUnicast = kDefaultRequestUnicastCount;
11822
11823 #if APPLE_OSX_mDNSResponder
11824 // Set the QU bit in the first query for the following options.
11825 if ((question->flags & kDNSServiceFlagsUnicastResponse) || (question->flags & kDNSServiceFlagsThresholdFinder))
11826 {
11827 question->RequestUnicast = SET_QU_IN_FIRST_QUERY;
11828 LogInfo("InitCommonState: setting RequestUnicast = %d for %##s (%s)", question->RequestUnicast, question->qname.c,
11829 DNSTypeName(question->qtype));
11830 }
11831 #endif // APPLE_OSX_mDNSResponder
11832
11833 question->LastQTxTime = m->timenow;
11834 question->CNAMEReferrals = 0;
11835
11836 question->WakeOnResolveCount = 0;
11837 if (question->WakeOnResolve)
11838 {
11839 question->WakeOnResolveCount = InitialWakeOnResolveCount;
11840 purge = mDNStrue;
11841 }
11842
11843 for (i=0; i<DupSuppressInfoSize; i++)
11844 question->DupSuppress[i].InterfaceID = mDNSNULL;
11845
11846 question->Restart = 0;
11847
11848 debugf("InitCommonState: Question %##s (%s) Interface %p Now %d Send in %d Answer in %d (%p) %s (%p)",
11849 question->qname.c, DNSTypeName(question->qtype), question->InterfaceID, m->timenow,
11850 NextQSendTime(question) - m->timenow,
11851 question->DelayAnswering ? question->DelayAnswering - m->timenow : 0,
11852 question, question->DuplicateOf ? "duplicate of" : "not duplicate", question->DuplicateOf);
11853
11854 if (question->DelayAnswering)
11855 LogInfo("InitCommonState: Delaying answering for %d ticks while cache stabilizes for %##s (%s)",
11856 question->DelayAnswering - m->timenow, question->qname.c, DNSTypeName(question->qtype));
11857
11858 return(purge);
11859 }
11860
11861 // Excludes the DNS Config fields which are already handled by InitDNSConfig()
11862 mDNSlocal void InitWABState(DNSQuestion *const question)
11863 {
11864 // We'll create our question->LocalSocket on demand, if needed.
11865 // We won't need one for duplicate questions, or from questions answered immediately out of the cache.
11866 // We also don't need one for LLQs because (when we're using NAT) we want them all to share a single
11867 // NAT mapping for receiving inbound add/remove events.
11868 question->LocalSocket = mDNSNULL;
11869 question->unansweredQueries = 0;
11870 question->nta = mDNSNULL;
11871 question->servAddr = zeroAddr;
11872 question->servPort = zeroIPPort;
11873 question->tcp = mDNSNULL;
11874 question->NoAnswer = NoAnswer_Normal;
11875 }
11876
11877 mDNSlocal void InitLLQNATState(mDNS *const m)
11878 {
11879 // If we don't have our NAT mapping active, start it now
11880 if (!m->LLQNAT.clientCallback)
11881 {
11882 m->LLQNAT.Protocol = NATOp_MapUDP;
11883 m->LLQNAT.IntPort = m->UnicastPort4;
11884 m->LLQNAT.RequestedPort = m->UnicastPort4;
11885 m->LLQNAT.clientCallback = LLQNATCallback;
11886 m->LLQNAT.clientContext = (void*)1; // Means LLQ NAT Traversal just started
11887 mDNS_StartNATOperation_internal(m, &m->LLQNAT);
11888 }
11889 }
11890
11891 mDNSlocal void InitLLQState(DNSQuestion *const question)
11892 {
11893 question->state = LLQ_InitialRequest;
11894 question->ReqLease = 0;
11895 question->expire = 0;
11896 question->ntries = 0;
11897 question->id = zeroOpaque64;
11898 }
11899
11900 mDNSlocal void InitDNSPNState(DNSQuestion *const question)
11901 {
11902 question->dnsPushState = DNSPUSH_INIT;
11903 }
11904
11905 // InitDNSSECProxyState() is called by mDNS_StartQuery_internal() to initialize
11906 // DNSSEC & DNS Proxy fields of the DNS Question.
11907 mDNSlocal void InitDNSSECProxyState(mDNS *const m, DNSQuestion *const question)
11908 {
11909 (void) m;
11910
11911 // DNS server selection affects DNSSEC. Turn off validation if req_DO is not set
11912 // or the request is going over cellular interface.
11913 //
11914 // Note: This needs to be done here before we call FindDuplicateQuestion as it looks
11915 // at ValidationRequired setting also.
11916 if (question->qDNSServer)
11917 {
11918 if (question->qDNSServer->cellIntf)
11919 {
11920 debugf("InitDNSSECProxyState: Turning off validation for %##s (%s); going over cell", question->qname.c, DNSTypeName(question->qtype));
11921 question->ValidationRequired = mDNSfalse;
11922 }
11923 if (DNSSECOptionalQuestion(question) && !(question->qDNSServer->req_DO))
11924 {
11925 LogInfo("InitDNSSECProxyState: Turning off validation for %##s (%s); req_DO false",
11926 question->qname.c, DNSTypeName(question->qtype));
11927 question->ValidationRequired = DNSSEC_VALIDATION_NONE;
11928 }
11929 }
11930 question->ValidationState = (question->ValidationRequired ? DNSSECValRequired : DNSSECValNotRequired);
11931 question->ValidationStatus = 0;
11932 question->responseFlags = zeroID;
11933 }
11934
11935 // Once the question is completely initialized including the duplicate logic, this function
11936 // is called to finalize the unicast question which requires flushing the cache if needed,
11937 // activating the query etc.
11938 mDNSlocal void FinalizeUnicastQuestion(mDNS *const m, DNSQuestion *question, mDNSBool purge)
11939 {
11940 // Ensure DNS related info of duplicate question is same as the orig question
11941 if (question->DuplicateOf)
11942 {
11943 question->validDNSServers = question->DuplicateOf->validDNSServers;
11944 question->qDNSServer = question->DuplicateOf->qDNSServer;
11945 LogInfo("FinalizeUnicastQuestion: Duplicate question %p (%p) %##s (%s), DNS Server %#a:%d",
11946 question, question->DuplicateOf, question->qname.c, DNSTypeName(question->qtype),
11947 question->qDNSServer ? &question->qDNSServer->addr : mDNSNULL,
11948 mDNSVal16(question->qDNSServer ? question->qDNSServer->port : zeroIPPort));
11949 }
11950
11951 ActivateUnicastQuery(m, question, mDNSfalse);
11952
11953 // If purge was set above, flush the cache. Need to do this after we set the
11954 // DNS server on the question
11955 if (purge)
11956 {
11957 question->DelayAnswering = 0;
11958 mDNS_PurgeForQuestion(m, question);
11959 }
11960 else if (!question->DuplicateOf && DNSSECQuestion(question))
11961 {
11962 // For DNSSEC questions, we need to have the RRSIGs also for verification.
11963 CheckForDNSSECRecords(m, question);
11964 }
11965 if (question->LongLived)
11966 {
11967 // Unlike other initializations, InitLLQNATState should be done after
11968 // we determine that it is a unicast question. LongLived is set for
11969 // both multicast and unicast browse questions but we should initialize
11970 // the LLQ NAT state only for unicast. Otherwise we will unnecessarily
11971 // start the NAT traversal that is not needed.
11972 InitLLQNATState(m);
11973 #if APPLE_OSX_mDNSResponder
11974 UpdateAutoTunnelDomainStatuses(m);
11975 #endif
11976 }
11977 }
11978
11979 mDNSexport mStatus mDNS_StartQuery_internal(mDNS *const m, DNSQuestion *const question)
11980 {
11981 DNSQuestion **q;
11982 mStatus vStatus;
11983 mDNSBool purge;
11984
11985 // First check for cache space (can't do queries if there is no cache space allocated)
11986 if (m->rrcache_size == 0)
11987 return(mStatus_NoCache);
11988
11989 vStatus = ValidateParameters(m, question);
11990 if (vStatus)
11991 return(vStatus);
11992
11993 #ifdef USE_LIBIDN
11994 // If the TLD includes high-ascii bytes, assume it will need to be converted to Punycode.
11995 // (In the future the root name servers may answer UTF-8 queries directly, but for now they do not.)
11996 if (IsHighASCIILabel(LastLabel(&question->qname)))
11997 {
11998 domainname newname;
11999 if (PerformNextPunycodeConversion(question, &newname))
12000 AssignDomainName(&question->qname, &newname);
12001 }
12002 #endif // USE_LIBIDN
12003
12004 question->TargetQID =
12005 #ifndef UNICAST_DISABLED
12006 (question->Target.type || Question_uDNS(question)) ? mDNS_NewMessageID(m) :
12007 #endif // UNICAST_DISABLED
12008 zeroID;
12009 debugf("mDNS_StartQuery_internal: %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
12010
12011 // Note: It important that new questions are appended at the *end* of the list, not prepended at the start
12012 q = &m->Questions;
12013 if (LocalOnlyOrP2PInterface(question->InterfaceID))
12014 q = &m->LocalOnlyQuestions;
12015 while (*q && *q != question)
12016 q=&(*q)->next;
12017
12018 if (*q)
12019 {
12020 LogMsg("mDNS_StartQuery_internal: Error! Tried to add a question %##s (%s) %p that's already in the active list",
12021 question->qname.c, DNSTypeName(question->qtype), question);
12022 return(mStatus_AlreadyRegistered);
12023 }
12024 *q = question;
12025
12026 // Intialize the question. The only ordering constraint we have today is that
12027 // InitDNSSECProxyState should be called after the DNS server is selected (in
12028 // InitCommonState -> InitDNSConfig) as DNS server selection affects DNSSEC
12029 // validation.
12030
12031 purge = InitCommonState(m, question);
12032 InitWABState(question);
12033 InitLLQState(question);
12034 InitDNSPNState(question);
12035 InitDNSSECProxyState(m, question);
12036
12037 // FindDuplicateQuestion should be called last after all the intialization
12038 // as the duplicate logic could be potentially based on any field in the
12039 // question.
12040 question->DuplicateOf = FindDuplicateQuestion(m, question);
12041 if (question->DuplicateOf)
12042 question->AuthInfo = question->DuplicateOf->AuthInfo;
12043
12044 if (LocalOnlyOrP2PInterface(question->InterfaceID))
12045 {
12046 if (!m->NewLocalOnlyQuestions)
12047 m->NewLocalOnlyQuestions = question;
12048 }
12049 else
12050 {
12051 if (!m->NewQuestions)
12052 m->NewQuestions = question;
12053
12054 // If the question's id is non-zero, then it's Wide Area
12055 // MUST NOT do this Wide Area setup until near the end of
12056 // mDNS_StartQuery_internal -- this code may itself issue queries (e.g. SOA,
12057 // NS, etc.) and if we haven't finished setting up our own question and setting
12058 // m->NewQuestions if necessary then we could end up recursively re-entering
12059 // this routine with the question list data structures in an inconsistent state.
12060 if (!mDNSOpaque16IsZero(question->TargetQID))
12061 {
12062 FinalizeUnicastQuestion(m, question, purge);
12063 }
12064 else
12065 {
12066 #if BONJOUR_ON_DEMAND
12067 m->NumAllInterfaceQuestions++;
12068 LogInfo("mDNS_StartQuery_internal: NumAllInterfaceRecords %d NumAllInterfaceQuestions %d %##s (%s)",
12069 m->NumAllInterfaceRecords, m->NumAllInterfaceQuestions, question->qname.c, DNSTypeName(question->qtype));
12070 if (m->NumAllInterfaceRecords + m->NumAllInterfaceQuestions == 1)
12071 {
12072 m->NextBonjourDisableTime = 0;
12073 if (m->BonjourEnabled == 0)
12074 {
12075 // Enable Bonjour immediately by scheduling network changed processing where
12076 // we will join the multicast group on each active interface.
12077 m->BonjourEnabled = 1;
12078 m->NetworkChanged = m->timenow;
12079 }
12080 }
12081 #endif // BONJOUR_ON_DEMAND
12082 if (purge)
12083 {
12084 LogInfo("mDNS_StartQuery_internal: Purging for %##s", question->qname.c);
12085 mDNS_PurgeForQuestion(m, question);
12086 }
12087 }
12088 }
12089
12090 return(mStatus_NoError);
12091 }
12092
12093 // CancelGetZoneData is an internal routine (i.e. must be called with the lock already held)
12094 mDNSexport void CancelGetZoneData(mDNS *const m, ZoneData *nta)
12095 {
12096 debugf("CancelGetZoneData %##s (%s)", nta->question.qname.c, DNSTypeName(nta->question.qtype));
12097 // This function may be called anytime to free the zone information.The question may or may not have stopped.
12098 // If it was already stopped, mDNS_StopQuery_internal would have set q->ThisQInterval to -1 and should not
12099 // call it again
12100 if (nta->question.ThisQInterval != -1)
12101 {
12102 mDNS_StopQuery_internal(m, &nta->question);
12103 if (nta->question.ThisQInterval != -1)
12104 LogMsg("CancelGetZoneData: Question %##s (%s) ThisQInterval %d not -1", nta->question.qname.c, DNSTypeName(nta->question.qtype), nta->question.ThisQInterval);
12105 }
12106 mDNSPlatformMemFree(nta);
12107 }
12108
12109 mDNSexport mStatus mDNS_StopQuery_internal(mDNS *const m, DNSQuestion *const question)
12110 {
12111 const mDNSu32 slot = HashSlot(&question->qname);
12112 CacheGroup *cg = CacheGroupForName(m, slot, question->qnamehash, &question->qname);
12113 CacheRecord *rr;
12114 DNSQuestion **qp = &m->Questions;
12115
12116 //LogInfo("mDNS_StopQuery_internal %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
12117
12118 if (LocalOnlyOrP2PInterface(question->InterfaceID))
12119 qp = &m->LocalOnlyQuestions;
12120 while (*qp && *qp != question) qp=&(*qp)->next;
12121 if (*qp) *qp = (*qp)->next;
12122 else
12123 {
12124 #if !ForceAlerts
12125 if (question->ThisQInterval >= 0) // Only log error message if the query was supposed to be active
12126 #endif
12127 LogFatalError("mDNS_StopQuery_internal: Question %##s (%s) not found in active list", question->qname.c, DNSTypeName(question->qtype));
12128 return(mStatus_BadReferenceErr);
12129 }
12130
12131 #if BONJOUR_ON_DEMAND
12132 if (!LocalOnlyOrP2PInterface(question->InterfaceID) && mDNSOpaque16IsZero(question->TargetQID))
12133 {
12134 if (m->NumAllInterfaceRecords + m->NumAllInterfaceQuestions == 1)
12135 m->NextBonjourDisableTime = NonZeroTime(m->timenow + (BONJOUR_DISABLE_DELAY * mDNSPlatformOneSecond));
12136 m->NumAllInterfaceQuestions--;
12137 LogInfo("mDNS_StopQuery_internal: NumAllInterfaceRecords %d NumAllInterfaceQuestions %d %##s (%s)",
12138 m->NumAllInterfaceRecords, m->NumAllInterfaceQuestions, question->qname.c, DNSTypeName(question->qtype));
12139 }
12140 #endif // BONJOUR_ON_DEMAND
12141
12142 #if TARGET_OS_EMBEDDED
12143 if (Question_uDNS(question) && !question->metrics.answered && (question->metrics.querySendCount > 0))
12144 {
12145 const domainname * queryName;
12146 mDNSBool isForCell;
12147 mDNSu32 durationMs;
12148
12149 queryName = question->metrics.originalQName ? question->metrics.originalQName : &question->qname;
12150 isForCell = (question->qDNSServer && question->qDNSServer->cellIntf);
12151
12152 if (question->metrics.querySendCount > 0)
12153 {
12154 durationMs = ((m->timenow - question->metrics.firstQueryTime) * 1000) / mDNSPlatformOneSecond;
12155 }
12156 else
12157 {
12158 durationMs = 0;
12159 }
12160 MetricsUpdateUDNSQueryStats(queryName, question->qtype, mDNSNULL, question->metrics.querySendCount, durationMs, isForCell);
12161 }
12162 #endif
12163 // Take care to cut question from list *before* calling UpdateQuestionDuplicates
12164 UpdateQuestionDuplicates(m, question);
12165 // But don't trash ThisQInterval until afterwards.
12166 question->ThisQInterval = -1;
12167
12168 // If there are any cache records referencing this as their active question, then see if there is any
12169 // other question that is also referencing them, else their CRActiveQuestion needs to get set to NULL.
12170 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
12171 {
12172 if (rr->CRActiveQuestion == question)
12173 {
12174 DNSQuestion *q;
12175 DNSQuestion *replacement = mDNSNULL;
12176 // If we find an active question that is answered by this cached record, use it as the cache record's
12177 // CRActiveQuestion replacement. If there are no such questions, but there's at least one unsuppressed inactive
12178 // question that is answered by this cache record, then use an inactive one to not forgo generating RMV events
12179 // via CacheRecordRmv() when the cache record expires.
12180 for (q = m->Questions; q && (q != m->NewQuestions); q = q->next)
12181 {
12182 if (!q->DuplicateOf && !QuerySuppressed(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
12183 {
12184 if (q->ThisQInterval > 0)
12185 {
12186 replacement = q;
12187 break;
12188 }
12189 else if (!replacement)
12190 {
12191 replacement = q;
12192 }
12193 }
12194 }
12195 if (replacement)
12196 debugf("mDNS_StopQuery_internal: Updating CRActiveQuestion to %p for cache record %s, Original question CurrentAnswers %d, new question "
12197 "CurrentAnswers %d, SuppressQuery %d", replacement, CRDisplayString(m,rr), question->CurrentAnswers, replacement->CurrentAnswers, replacement->SuppressQuery);
12198 rr->CRActiveQuestion = replacement; // Question used to be active; new value may or may not be null
12199 if (!replacement) m->rrcache_active--; // If no longer active, decrement rrcache_active count
12200 }
12201 }
12202
12203 // If we just deleted the question that CacheRecordAdd() or CacheRecordRmv() is about to look at,
12204 // bump its pointer forward one question.
12205 if (m->CurrentQuestion == question)
12206 {
12207 debugf("mDNS_StopQuery_internal: Just deleted the currently active question: %##s (%s)",
12208 question->qname.c, DNSTypeName(question->qtype));
12209 m->CurrentQuestion = question->next;
12210 }
12211
12212 if (m->NewQuestions == question)
12213 {
12214 debugf("mDNS_StopQuery_internal: Just deleted a new question that wasn't even answered yet: %##s (%s)",
12215 question->qname.c, DNSTypeName(question->qtype));
12216 m->NewQuestions = question->next;
12217 }
12218
12219 if (m->NewLocalOnlyQuestions == question) m->NewLocalOnlyQuestions = question->next;
12220
12221 if (m->RestartQuestion == question)
12222 {
12223 LogMsg("mDNS_StopQuery_internal: Just deleted the current restart question: %##s (%s)",
12224 question->qname.c, DNSTypeName(question->qtype));
12225 m->RestartQuestion = question->next;
12226 }
12227
12228 if (m->ValidationQuestion == question)
12229 {
12230 LogInfo("mDNS_StopQuery_internal: Just deleted the current Validation question: %##s (%s)",
12231 question->qname.c, DNSTypeName(question->qtype));
12232 m->ValidationQuestion = question->next;
12233 }
12234
12235 // Take care not to trash question->next until *after* we've updated m->CurrentQuestion and m->NewQuestions
12236 question->next = mDNSNULL;
12237
12238 // LogMsg("mDNS_StopQuery_internal: Question %##s (%s) removed", question->qname.c, DNSTypeName(question->qtype));
12239
12240 // And finally, cancel any associated GetZoneData operation that's still running.
12241 // Must not do this until last, because there's a good chance the GetZoneData question is the next in the list,
12242 // so if we delete it earlier in this routine, we could find that our "question->next" pointer above is already
12243 // invalid before we even use it. By making sure that we update m->CurrentQuestion and m->NewQuestions if necessary
12244 // *first*, then they're all ready to be updated a second time if necessary when we cancel our GetZoneData query.
12245 if (question->tcp) { DisposeTCPConn(question->tcp); question->tcp = mDNSNULL; }
12246 if (question->LocalSocket) { mDNSPlatformUDPClose(question->LocalSocket); question->LocalSocket = mDNSNULL; }
12247 if (!mDNSOpaque16IsZero(question->TargetQID) && question->LongLived)
12248 {
12249 // Scan our list to see if any more wide-area LLQs remain. If not, stop our NAT Traversal.
12250 DNSQuestion *q;
12251 for (q = m->Questions; q; q=q->next)
12252 if (!mDNSOpaque16IsZero(q->TargetQID) && q->LongLived) break;
12253 if (!q)
12254 {
12255 if (!m->LLQNAT.clientCallback) // Should never happen, but just in case...
12256 {
12257 LogMsg("mDNS_StopQuery ERROR LLQNAT.clientCallback NULL");
12258 }
12259 else
12260 {
12261 LogInfo("Stopping LLQNAT");
12262 mDNS_StopNATOperation_internal(m, &m->LLQNAT);
12263 m->LLQNAT.clientCallback = mDNSNULL; // Means LLQ NAT Traversal not running
12264 }
12265 }
12266
12267 // If necessary, tell server it can delete this LLQ state
12268 if (question->state == LLQ_Established)
12269 {
12270 question->ReqLease = 0;
12271 sendLLQRefresh(m, question);
12272 // If we need need to make a TCP connection to cancel the LLQ, that's going to take a little while.
12273 // We clear the tcp->question backpointer so that when the TCP connection completes, it doesn't
12274 // crash trying to access our cancelled question, but we don't cancel the TCP operation itself --
12275 // we let that run out its natural course and complete asynchronously.
12276 if (question->tcp)
12277 {
12278 question->tcp->question = mDNSNULL;
12279 question->tcp = mDNSNULL;
12280 }
12281 }
12282 else if (question->dnsPushState == DNSPUSH_ESTABLISHED)
12283 {
12284 if (question->tcp)
12285 {
12286 UnSubscribeToDNSPushNotificationServer(m, q);
12287 question->tcp->question = mDNSNULL;
12288 question->tcp = mDNSNULL;
12289 }
12290 }
12291 #if APPLE_OSX_mDNSResponder
12292 UpdateAutoTunnelDomainStatuses(m);
12293 #endif
12294 }
12295 // wait until we send the refresh above which needs the nta
12296 if (question->nta) { CancelGetZoneData(m, question->nta); question->nta = mDNSNULL; }
12297
12298 if (question->ValidationRequired && question->DNSSECAuthInfo)
12299 {
12300 LogInfo("mDNS_StopQuery_internal: freeing DNSSECAuthInfo %##s", question->qname.c);
12301 question->DAIFreeCallback(m, question->DNSSECAuthInfo);
12302 question->DNSSECAuthInfo = mDNSNULL;
12303 }
12304 if (question->AnonInfo)
12305 {
12306 FreeAnonInfo(question->AnonInfo);
12307 question->AnonInfo = mDNSNULL;
12308 }
12309 #if TARGET_OS_EMBEDDED
12310 if (question->metrics.originalQName)
12311 {
12312 mDNSPlatformMemFree(question->metrics.originalQName);
12313 question->metrics.originalQName = mDNSNULL;
12314 }
12315 #endif
12316
12317 return(mStatus_NoError);
12318 }
12319
12320 mDNSexport mStatus mDNS_StartQuery(mDNS *const m, DNSQuestion *const question)
12321 {
12322 mStatus status;
12323 mDNS_Lock(m);
12324 status = mDNS_StartQuery_internal(m, question);
12325 mDNS_Unlock(m);
12326 return(status);
12327 }
12328
12329 mDNSexport mStatus mDNS_StopQuery(mDNS *const m, DNSQuestion *const question)
12330 {
12331 mStatus status;
12332 mDNS_Lock(m);
12333 status = mDNS_StopQuery_internal(m, question);
12334 mDNS_Unlock(m);
12335 return(status);
12336 }
12337
12338 // Note that mDNS_StopQueryWithRemoves() does not currently implement the full generality of the other APIs
12339 // Specifically, question callbacks invoked as a result of this call cannot themselves make API calls.
12340 // We invoke the callback without using mDNS_DropLockBeforeCallback/mDNS_ReclaimLockAfterCallback
12341 // specifically to catch and report if the client callback does try to make API calls
12342 mDNSexport mStatus mDNS_StopQueryWithRemoves(mDNS *const m, DNSQuestion *const question)
12343 {
12344 mStatus status;
12345 DNSQuestion *qq;
12346 mDNS_Lock(m);
12347
12348 // Check if question is new -- don't want to give remove events for a question we haven't even answered yet
12349 for (qq = m->NewQuestions; qq; qq=qq->next) if (qq == question) break;
12350
12351 status = mDNS_StopQuery_internal(m, question);
12352 if (status == mStatus_NoError && !qq)
12353 {
12354 const CacheRecord *rr;
12355 const mDNSu32 slot = HashSlot(&question->qname);
12356 CacheGroup *const cg = CacheGroupForName(m, slot, question->qnamehash, &question->qname);
12357 LogInfo("Generating terminal removes for %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
12358 for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
12359 if (rr->resrec.RecordType != kDNSRecordTypePacketNegative && SameNameRecordAnswersQuestion(&rr->resrec, question))
12360 {
12361 // Don't use mDNS_DropLockBeforeCallback() here, since we don't allow API calls
12362 if (question->QuestionCallback)
12363 question->QuestionCallback(m, question, &rr->resrec, mDNSfalse);
12364 }
12365 }
12366 mDNS_Unlock(m);
12367 return(status);
12368 }
12369
12370 mDNSexport mStatus mDNS_Reconfirm(mDNS *const m, CacheRecord *const cr)
12371 {
12372 mStatus status;
12373 mDNS_Lock(m);
12374 status = mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
12375 if (status == mStatus_NoError) ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, 0);
12376 mDNS_Unlock(m);
12377 return(status);
12378 }
12379
12380 mDNSexport mStatus mDNS_ReconfirmByValue(mDNS *const m, ResourceRecord *const rr)
12381 {
12382 mStatus status = mStatus_BadReferenceErr;
12383 CacheRecord *cr;
12384 mDNS_Lock(m);
12385 cr = FindIdenticalRecordInCache(m, rr);
12386 debugf("mDNS_ReconfirmByValue: %p %s", cr, RRDisplayString(m, rr));
12387 if (cr) status = mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
12388 if (status == mStatus_NoError) ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, 0);
12389 mDNS_Unlock(m);
12390 return(status);
12391 }
12392
12393 mDNSlocal mStatus mDNS_StartBrowse_internal(mDNS *const m, DNSQuestion *const question,
12394 const domainname *const srv, const domainname *const domain,
12395 const mDNSu8 *anondata, const mDNSInterfaceID InterfaceID, mDNSu32 flags,
12396 mDNSBool ForceMCast, mDNSBool useBackgroundTrafficClass,
12397 mDNSQuestionCallback *Callback, void *Context)
12398 {
12399 question->InterfaceID = InterfaceID;
12400 question->flags = flags;
12401 question->Target = zeroAddr;
12402 question->qtype = kDNSType_PTR;
12403 question->qclass = kDNSClass_IN;
12404 question->LongLived = mDNStrue;
12405 question->ExpectUnique = mDNSfalse;
12406 question->ForceMCast = ForceMCast;
12407 question->ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
12408 question->SuppressUnusable = mDNSfalse;
12409 question->SearchListIndex = 0;
12410 question->AppendSearchDomains = 0;
12411 question->RetryWithSearchDomains = mDNSfalse;
12412 question->TimeoutQuestion = 0;
12413 question->WakeOnResolve = 0;
12414 question->UseBackgroundTrafficClass = useBackgroundTrafficClass;
12415 question->ValidationRequired = 0;
12416 question->ValidatingResponse = 0;
12417 question->ProxyQuestion = 0;
12418 question->qnameOrig = mDNSNULL;
12419 question->AnonInfo = mDNSNULL;
12420 question->QuestionCallback = Callback;
12421 question->QuestionContext = Context;
12422
12423 if (!ConstructServiceName(&question->qname, mDNSNULL, srv, domain))
12424 return(mStatus_BadParamErr);
12425
12426 if (anondata)
12427 {
12428 question->AnonInfo = AllocateAnonInfo(&question->qname, anondata, mDNSPlatformStrLen(anondata), mDNSNULL);
12429 if (!question->AnonInfo)
12430 return(mStatus_BadParamErr);
12431 }
12432
12433 return(mDNS_StartQuery_internal(m, question));
12434 }
12435
12436 mDNSexport mStatus mDNS_StartBrowse(mDNS *const m, DNSQuestion *const question,
12437 const domainname *const srv, const domainname *const domain,
12438 const mDNSu8 *anondata, const mDNSInterfaceID InterfaceID, mDNSu32 flags,
12439 mDNSBool ForceMCast, mDNSBool useBackgroundTrafficClass,
12440 mDNSQuestionCallback *Callback, void *Context)
12441 {
12442 mStatus status;
12443 mDNS_Lock(m);
12444 status = mDNS_StartBrowse_internal(m, question, srv, domain, anondata, InterfaceID, flags, ForceMCast, useBackgroundTrafficClass, Callback, Context);
12445 mDNS_Unlock(m);
12446 return(status);
12447 }
12448
12449
12450 mDNSexport mStatus mDNS_GetDomains(mDNS *const m, DNSQuestion *const question, mDNS_DomainType DomainType, const domainname *dom,
12451 const mDNSInterfaceID InterfaceID, mDNSQuestionCallback *Callback, void *Context)
12452 {
12453 question->InterfaceID = InterfaceID;
12454 question->flags = 0;
12455 question->Target = zeroAddr;
12456 question->qtype = kDNSType_PTR;
12457 question->qclass = kDNSClass_IN;
12458 question->LongLived = mDNSfalse;
12459 question->ExpectUnique = mDNSfalse;
12460 question->ForceMCast = mDNSfalse;
12461 question->ReturnIntermed = mDNSfalse;
12462 question->SuppressUnusable = mDNSfalse;
12463 question->SearchListIndex = 0;
12464 question->AppendSearchDomains = 0;
12465 question->RetryWithSearchDomains = mDNSfalse;
12466 question->TimeoutQuestion = 0;
12467 question->WakeOnResolve = 0;
12468 question->UseBackgroundTrafficClass = mDNSfalse;
12469 question->ValidationRequired = 0;
12470 question->ValidatingResponse = 0;
12471 question->ProxyQuestion = 0;
12472 question->qnameOrig = mDNSNULL;
12473 question->AnonInfo = mDNSNULL;
12474 question->pid = mDNSPlatformGetPID();
12475 question->euid = 0;
12476 question->QuestionCallback = Callback;
12477 question->QuestionContext = Context;
12478 if (DomainType > mDNS_DomainTypeMax) return(mStatus_BadParamErr);
12479 if (!MakeDomainNameFromDNSNameString(&question->qname, mDNS_DomainTypeNames[DomainType])) return(mStatus_BadParamErr);
12480 if (!dom) dom = &localdomain;
12481 if (!AppendDomainName(&question->qname, dom)) return(mStatus_BadParamErr);
12482 return(mDNS_StartQuery(m, question));
12483 }
12484
12485 // ***************************************************************************
12486 #if COMPILER_LIKES_PRAGMA_MARK
12487 #pragma mark -
12488 #pragma mark - Responder Functions
12489 #endif
12490
12491 mDNSexport mStatus mDNS_Register(mDNS *const m, AuthRecord *const rr)
12492 {
12493 mStatus status;
12494 mDNS_Lock(m);
12495 status = mDNS_Register_internal(m, rr);
12496 mDNS_Unlock(m);
12497 return(status);
12498 }
12499
12500 mDNSexport mStatus mDNS_Update(mDNS *const m, AuthRecord *const rr, mDNSu32 newttl,
12501 const mDNSu16 newrdlength, RData *const newrdata, mDNSRecordUpdateCallback *Callback)
12502 {
12503 if (!ValidateRData(rr->resrec.rrtype, newrdlength, newrdata))
12504 {
12505 LogMsg("Attempt to update record with invalid rdata: %s", GetRRDisplayString_rdb(&rr->resrec, &newrdata->u, m->MsgBuffer));
12506 return(mStatus_Invalid);
12507 }
12508
12509 mDNS_Lock(m);
12510
12511 // If TTL is unspecified, leave TTL unchanged
12512 if (newttl == 0) newttl = rr->resrec.rroriginalttl;
12513
12514 // If we already have an update queued up which has not gone through yet, give the client a chance to free that memory
12515 if (rr->NewRData)
12516 {
12517 RData *n = rr->NewRData;
12518 rr->NewRData = mDNSNULL; // Clear the NewRData pointer ...
12519 if (rr->UpdateCallback)
12520 rr->UpdateCallback(m, rr, n, rr->newrdlength); // ...and let the client free this memory, if necessary
12521 }
12522
12523 rr->NewRData = newrdata;
12524 rr->newrdlength = newrdlength;
12525 rr->UpdateCallback = Callback;
12526
12527 #ifndef UNICAST_DISABLED
12528 if (rr->ARType != AuthRecordLocalOnly && rr->ARType != AuthRecordP2P && !IsLocalDomain(rr->resrec.name))
12529 {
12530 mStatus status = uDNS_UpdateRecord(m, rr);
12531 // The caller frees the memory on error, don't retain stale pointers
12532 if (status != mStatus_NoError) { rr->NewRData = mDNSNULL; rr->newrdlength = 0; }
12533 mDNS_Unlock(m);
12534 return(status);
12535 }
12536 #endif
12537
12538 if (RRLocalOnly(rr) || (rr->resrec.rroriginalttl == newttl &&
12539 rr->resrec.rdlength == newrdlength && mDNSPlatformMemSame(rr->resrec.rdata->u.data, newrdata->u.data, newrdlength)))
12540 CompleteRDataUpdate(m, rr);
12541 else
12542 {
12543 rr->AnnounceCount = InitialAnnounceCount;
12544 InitializeLastAPTime(m, rr);
12545 while (rr->NextUpdateCredit && m->timenow - rr->NextUpdateCredit >= 0) GrantUpdateCredit(rr);
12546 if (!rr->UpdateBlocked && rr->UpdateCredits) rr->UpdateCredits--;
12547 if (!rr->NextUpdateCredit) rr->NextUpdateCredit = NonZeroTime(m->timenow + kUpdateCreditRefreshInterval);
12548 if (rr->AnnounceCount > rr->UpdateCredits + 1) rr->AnnounceCount = (mDNSu8)(rr->UpdateCredits + 1);
12549 if (rr->UpdateCredits <= 5)
12550 {
12551 mDNSu32 delay = 6 - rr->UpdateCredits; // Delay 1 second, then 2, then 3, etc. up to 6 seconds maximum
12552 if (!rr->UpdateBlocked) rr->UpdateBlocked = NonZeroTime(m->timenow + (mDNSs32)delay * mDNSPlatformOneSecond);
12553 rr->ThisAPInterval *= 4;
12554 rr->LastAPTime = rr->UpdateBlocked - rr->ThisAPInterval;
12555 LogMsg("Excessive update rate for %##s; delaying announcement by %ld second%s",
12556 rr->resrec.name->c, delay, delay > 1 ? "s" : "");
12557 }
12558 rr->resrec.rroriginalttl = newttl;
12559 }
12560
12561 mDNS_Unlock(m);
12562 return(mStatus_NoError);
12563 }
12564
12565 // Note: mDNS_Deregister calls mDNS_Deregister_internal which can call a user callback, which may change
12566 // the record list and/or question list.
12567 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
12568 mDNSexport mStatus mDNS_Deregister(mDNS *const m, AuthRecord *const rr)
12569 {
12570 mStatus status;
12571 mDNS_Lock(m);
12572 status = mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
12573 mDNS_Unlock(m);
12574 return(status);
12575 }
12576
12577 // Circular reference: AdvertiseInterface references mDNS_HostNameCallback, which calls mDNS_SetFQDN, which call AdvertiseInterface
12578 mDNSlocal void mDNS_HostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result);
12579
12580 mDNSlocal NetworkInterfaceInfo *FindFirstAdvertisedInterface(mDNS *const m)
12581 {
12582 NetworkInterfaceInfo *intf;
12583 for (intf = m->HostInterfaces; intf; intf = intf->next)
12584 if (intf->Advertise) break;
12585 return(intf);
12586 }
12587
12588 mDNSlocal void AdvertiseInterface(mDNS *const m, NetworkInterfaceInfo *set)
12589 {
12590 char buffer[MAX_REVERSE_MAPPING_NAME];
12591 NetworkInterfaceInfo *primary;
12592 mDNSu8 recordType;
12593
12594 if (m->AutoTargetServices == 0)
12595 {
12596 LogInfo("AdvertiseInterface: Returning due to AutoTargetServices zero for %s", set->ifname);
12597 return;
12598 }
12599
12600 primary = FindFirstAdvertisedInterface(m);
12601 if (!primary) primary = set; // If no existing advertised interface, this new NetworkInterfaceInfo becomes our new primary
12602
12603 // If interface is marked as a direct link, we can assume the address record is unique
12604 // and does not need to go through the probe phase of the probe/announce packet sequence.
12605 recordType = (set->DirectLink ? kDNSRecordTypeKnownUnique : kDNSRecordTypeUnique);
12606
12607 if (set->DirectLink)
12608 LogInfo("AdvertiseInterface: Marking address record as kDNSRecordTypeKnownUnique for %s", set->ifname);
12609
12610 // Send dynamic update for non-linklocal IPv4 Addresses
12611 mDNS_SetupResourceRecord(&set->RR_A, mDNSNULL, set->InterfaceID, kDNSType_A, kHostNameTTL, recordType, AuthRecordAny, mDNS_HostNameCallback, set);
12612 mDNS_SetupResourceRecord(&set->RR_PTR, mDNSNULL, set->InterfaceID, kDNSType_PTR, kHostNameTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
12613 mDNS_SetupResourceRecord(&set->RR_HINFO, mDNSNULL, set->InterfaceID, kDNSType_HINFO, kHostNameTTL, kDNSRecordTypeUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
12614
12615 #if ANSWER_REMOTE_HOSTNAME_QUERIES
12616 set->RR_A.AllowRemoteQuery = mDNStrue;
12617 set->RR_PTR.AllowRemoteQuery = mDNStrue;
12618 set->RR_HINFO.AllowRemoteQuery = mDNStrue;
12619 #endif
12620 // 1. Set up Address record to map from host name ("foo.local.") to IP address
12621 // 2. Set up reverse-lookup PTR record to map from our address back to our host name
12622 AssignDomainName(&set->RR_A.namestorage, &m->MulticastHostname);
12623 if (set->ip.type == mDNSAddrType_IPv4)
12624 {
12625 set->RR_A.resrec.rrtype = kDNSType_A;
12626 set->RR_A.resrec.rdata->u.ipv4 = set->ip.ip.v4;
12627 // Note: This is reverse order compared to a normal dotted-decimal IP address, so we can't use our customary "%.4a" format code
12628 mDNS_snprintf(buffer, sizeof(buffer), "%d.%d.%d.%d.in-addr.arpa.",
12629 set->ip.ip.v4.b[3], set->ip.ip.v4.b[2], set->ip.ip.v4.b[1], set->ip.ip.v4.b[0]);
12630 }
12631 else if (set->ip.type == mDNSAddrType_IPv6)
12632 {
12633 int i;
12634 set->RR_A.resrec.rrtype = kDNSType_AAAA;
12635 set->RR_A.resrec.rdata->u.ipv6 = set->ip.ip.v6;
12636 for (i = 0; i < 16; i++)
12637 {
12638 static const char hexValues[] = "0123456789ABCDEF";
12639 buffer[i * 4 ] = hexValues[set->ip.ip.v6.b[15 - i] & 0x0F];
12640 buffer[i * 4 + 1] = '.';
12641 buffer[i * 4 + 2] = hexValues[set->ip.ip.v6.b[15 - i] >> 4];
12642 buffer[i * 4 + 3] = '.';
12643 }
12644 mDNS_snprintf(&buffer[64], sizeof(buffer)-64, "ip6.arpa.");
12645 }
12646
12647 MakeDomainNameFromDNSNameString(&set->RR_PTR.namestorage, buffer);
12648 set->RR_PTR.AutoTarget = Target_AutoHost; // Tell mDNS that the target of this PTR is to be kept in sync with our host name
12649 set->RR_PTR.ForceMCast = mDNStrue; // This PTR points to our dot-local name, so don't ever try to write it into a uDNS server
12650
12651 set->RR_A.RRSet = &primary->RR_A; // May refer to self
12652
12653 mDNS_Register_internal(m, &set->RR_A);
12654 mDNS_Register_internal(m, &set->RR_PTR);
12655
12656 #if APPLE_OSX_mDNSResponder
12657 // must be after the mDNS_Register_internal() calls so that records have complete rdata fields, etc
12658 D2D_start_advertising_interface(set);
12659 #endif // APPLE_OSX_mDNSResponder
12660
12661 if (!NO_HINFO && m->HIHardware.c[0] > 0 && m->HISoftware.c[0] > 0 && m->HIHardware.c[0] + m->HISoftware.c[0] <= 254)
12662 {
12663 mDNSu8 *p = set->RR_HINFO.resrec.rdata->u.data;
12664 AssignDomainName(&set->RR_HINFO.namestorage, &m->MulticastHostname);
12665 set->RR_HINFO.DependentOn = &set->RR_A;
12666 mDNSPlatformMemCopy(p, &m->HIHardware, 1 + (mDNSu32)m->HIHardware.c[0]);
12667 p += 1 + (int)p[0];
12668 mDNSPlatformMemCopy(p, &m->HISoftware, 1 + (mDNSu32)m->HISoftware.c[0]);
12669 mDNS_Register_internal(m, &set->RR_HINFO);
12670 }
12671 else
12672 {
12673 debugf("Not creating HINFO record: platform support layer provided no information");
12674 set->RR_HINFO.resrec.RecordType = kDNSRecordTypeUnregistered;
12675 }
12676 }
12677
12678 mDNSlocal void DeadvertiseInterface(mDNS *const m, NetworkInterfaceInfo *set)
12679 {
12680 if (m->AutoTargetServices == 0)
12681 {
12682 LogInfo("DeadvertiseInterface: Returning due to AutoTargetServices zero for %s", set->ifname);
12683 return;
12684 }
12685
12686 #if APPLE_OSX_mDNSResponder
12687 D2D_stop_advertising_interface(set);
12688 #endif // APPLE_OSX_mDNSResponder
12689
12690 // Unregister these records.
12691 // When doing the mDNS_Exit processing, we first call DeadvertiseInterface for each interface, so by the time the platform
12692 // support layer gets to call mDNS_DeregisterInterface, the address and PTR records have already been deregistered for it.
12693 // Also, in the event of a name conflict, one or more of our records will have been forcibly deregistered.
12694 // To avoid unnecessary and misleading warning messages, we check the RecordType before calling mDNS_Deregister_internal().
12695 if (set->RR_A .resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_A, mDNS_Dereg_normal);
12696 if (set->RR_PTR .resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_PTR, mDNS_Dereg_normal);
12697 if (set->RR_HINFO.resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_HINFO, mDNS_Dereg_normal);
12698 }
12699
12700 mDNSlocal void AdvertiseAllInterfaceRecords(mDNS *const m)
12701 {
12702 NetworkInterfaceInfo *intf;
12703 for (intf = m->HostInterfaces; intf; intf = intf->next)
12704 {
12705 if (intf->Advertise)
12706 {
12707 LogInfo("AdvertiseInterface: Advertising for ifname %s", intf->ifname);
12708 AdvertiseInterface(m, intf);
12709 }
12710 }
12711 }
12712
12713 mDNSlocal void DeadvertiseAllInterfaceRecords(mDNS *const m)
12714 {
12715 NetworkInterfaceInfo *intf;
12716 for (intf = m->HostInterfaces; intf; intf = intf->next)
12717 {
12718 if (intf->Advertise)
12719 {
12720 LogInfo("DeadvertiseInterface: Deadvertising for ifname %s", intf->ifname);
12721 DeadvertiseInterface(m, intf);
12722 }
12723 }
12724 }
12725
12726 // Change target host name for record.
12727 mDNSlocal void UpdateTargetHostName(mDNS *const m, AuthRecord *const rr)
12728 {
12729 #if APPLE_OSX_mDNSResponder
12730 // If this record was also registered with any D2D plugins, stop advertising
12731 // the version with the old host name.
12732 D2D_stop_advertising_record(rr);
12733 #endif
12734
12735 SetTargetToHostName(m, rr);
12736
12737 #if APPLE_OSX_mDNSResponder
12738 // Advertise the record with the updated host name with the D2D plugins if appropriate.
12739 D2D_start_advertising_record(rr);
12740 #endif
12741 }
12742
12743 mDNSexport void mDNS_SetFQDN(mDNS *const m)
12744 {
12745 domainname newmname;
12746 AuthRecord *rr;
12747 newmname.c[0] = 0;
12748
12749 if (!AppendDomainLabel(&newmname, &m->hostlabel)) { LogMsg("ERROR: mDNS_SetFQDN: Cannot create MulticastHostname"); return; }
12750 if (!AppendLiteralLabelString(&newmname, "local")) { LogMsg("ERROR: mDNS_SetFQDN: Cannot create MulticastHostname"); return; }
12751
12752 mDNS_Lock(m);
12753
12754 if (SameDomainNameCS(&m->MulticastHostname, &newmname)) debugf("mDNS_SetFQDN - hostname unchanged");
12755 else
12756 {
12757 AssignDomainName(&m->MulticastHostname, &newmname);
12758 DeadvertiseAllInterfaceRecords(m);
12759 AdvertiseAllInterfaceRecords(m);
12760 }
12761
12762 // 3. Make sure that any AutoTarget SRV records (and the like) get updated
12763 for (rr = m->ResourceRecords; rr; rr=rr->next) if (rr->AutoTarget) UpdateTargetHostName(m, rr);
12764 for (rr = m->DuplicateRecords; rr; rr=rr->next) if (rr->AutoTarget) UpdateTargetHostName(m, rr);
12765
12766 mDNS_Unlock(m);
12767 }
12768
12769 mDNSlocal void mDNS_HostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
12770 {
12771 (void)rr; // Unused parameter
12772
12773 #if MDNS_DEBUGMSGS
12774 {
12775 char *msg = "Unknown result";
12776 if (result == mStatus_NoError) msg = "Name registered";
12777 else if (result == mStatus_NameConflict) msg = "Name conflict";
12778 debugf("mDNS_HostNameCallback: %##s (%s) %s (%ld)", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), msg, result);
12779 }
12780 #endif
12781
12782 if (result == mStatus_NoError)
12783 {
12784 // Notify the client that the host name is successfully registered
12785 if (m->MainCallback)
12786 m->MainCallback(m, mStatus_NoError);
12787 }
12788 else if (result == mStatus_NameConflict)
12789 {
12790 domainlabel oldlabel = m->hostlabel;
12791
12792 // 1. First give the client callback a chance to pick a new name
12793 if (m->MainCallback)
12794 m->MainCallback(m, mStatus_NameConflict);
12795
12796 // 2. If the client callback didn't do it, add (or increment) an index ourselves
12797 // This needs to be case-INSENSITIVE compare, because we need to know that the name has been changed so as to
12798 // remedy the conflict, and a name that differs only in capitalization will just suffer the exact same conflict again.
12799 if (SameDomainLabel(m->hostlabel.c, oldlabel.c))
12800 IncrementLabelSuffix(&m->hostlabel, mDNSfalse);
12801
12802 // 3. Generate the FQDNs from the hostlabel,
12803 // and make sure all SRV records, etc., are updated to reference our new hostname
12804 mDNS_SetFQDN(m);
12805 LogMsg("Local Hostname %#s.local already in use; will try %#s.local instead", oldlabel.c, m->hostlabel.c);
12806 }
12807 else if (result == mStatus_MemFree)
12808 {
12809 // .local hostnames do not require goodbyes - we ignore the MemFree (which is sent directly by
12810 // mDNS_Deregister_internal), and allow the caller to deallocate immediately following mDNS_DeadvertiseInterface
12811 debugf("mDNS_HostNameCallback: MemFree (ignored)");
12812 }
12813 else
12814 LogMsg("mDNS_HostNameCallback: Unknown error %d for registration of record %s", result, rr->resrec.name->c);
12815 }
12816
12817 mDNSlocal void UpdateInterfaceProtocols(mDNS *const m, NetworkInterfaceInfo *active)
12818 {
12819 NetworkInterfaceInfo *intf;
12820 active->IPv4Available = mDNSfalse;
12821 active->IPv6Available = mDNSfalse;
12822 for (intf = m->HostInterfaces; intf; intf = intf->next)
12823 if (intf->InterfaceID == active->InterfaceID)
12824 {
12825 if (intf->ip.type == mDNSAddrType_IPv4 && intf->McastTxRx) active->IPv4Available = mDNStrue;
12826 if (intf->ip.type == mDNSAddrType_IPv6 && intf->McastTxRx) active->IPv6Available = mDNStrue;
12827 }
12828 }
12829
12830 mDNSlocal void RestartRecordGetZoneData(mDNS * const m)
12831 {
12832 AuthRecord *rr;
12833 LogInfo("RestartRecordGetZoneData: ResourceRecords");
12834 for (rr = m->ResourceRecords; rr; rr=rr->next)
12835 if (AuthRecord_uDNS(rr) && rr->state != regState_NoTarget)
12836 {
12837 debugf("RestartRecordGetZoneData: StartGetZoneData for %##s", rr->resrec.name->c);
12838 // Zero out the updateid so that if we have a pending response from the server, it won't
12839 // be accepted as a valid response. If we accept the response, we might free the new "nta"
12840 if (rr->nta) { rr->updateid = zeroID; CancelGetZoneData(m, rr->nta); }
12841 rr->nta = StartGetZoneData(m, rr->resrec.name, ZoneServiceUpdate, RecordRegistrationGotZoneData, rr);
12842 }
12843 }
12844
12845 mDNSlocal void InitializeNetWakeState(mDNS *const m, NetworkInterfaceInfo *set)
12846 {
12847 int i;
12848 // We initialize ThisQInterval to -1 indicating that the question has not been started
12849 // yet. If the question (browse) is started later during interface registration, it will
12850 // be stopped during interface deregistration. We can't sanity check to see if the
12851 // question has been stopped or not before initializing it to -1 because we need to
12852 // initialize it to -1 the very first time.
12853
12854 set->NetWakeBrowse.ThisQInterval = -1;
12855 for (i=0; i<3; i++)
12856 {
12857 set->NetWakeResolve[i].ThisQInterval = -1;
12858 set->SPSAddr[i].type = mDNSAddrType_None;
12859 }
12860 set->NextSPSAttempt = -1;
12861 set->NextSPSAttemptTime = m->timenow;
12862 }
12863
12864 mDNSexport void mDNS_ActivateNetWake_internal(mDNS *const m, NetworkInterfaceInfo *set)
12865 {
12866 NetworkInterfaceInfo *p = m->HostInterfaces;
12867 while (p && p != set) p=p->next;
12868 if (!p) { LogMsg("mDNS_ActivateNetWake_internal: NetworkInterfaceInfo %p not found in active list", set); return; }
12869
12870 if (set->InterfaceActive)
12871 {
12872 LogSPS("ActivateNetWake for %s (%#a)", set->ifname, &set->ip);
12873 mDNS_StartBrowse_internal(m, &set->NetWakeBrowse, &SleepProxyServiceType, &localdomain, mDNSNULL, set->InterfaceID, 0, mDNSfalse, mDNSfalse, m->SPSBrowseCallback, set);
12874 }
12875 }
12876
12877 mDNSexport void mDNS_DeactivateNetWake_internal(mDNS *const m, NetworkInterfaceInfo *set)
12878 {
12879 NetworkInterfaceInfo *p = m->HostInterfaces;
12880 while (p && p != set) p=p->next;
12881 if (!p) { LogMsg("mDNS_DeactivateNetWake_internal: NetworkInterfaceInfo %p not found in active list", set); return; }
12882
12883 // Note: We start the browse only if the interface is NetWake capable and we use this to
12884 // stop the resolves also. Hence, the resolves should not be started without the browse
12885 // being started i.e, resolves should not happen unless NetWake capable which is
12886 // guaranteed by BeginSleepProcessing.
12887 if (set->NetWakeBrowse.ThisQInterval >= 0)
12888 {
12889 int i;
12890 LogSPS("DeactivateNetWake for %s (%#a)", set->ifname, &set->ip);
12891
12892 // Stop our browse and resolve operations
12893 mDNS_StopQuery_internal(m, &set->NetWakeBrowse);
12894 for (i=0; i<3; i++) if (set->NetWakeResolve[i].ThisQInterval >= 0) mDNS_StopQuery_internal(m, &set->NetWakeResolve[i]);
12895
12896 // Make special call to the browse callback to let it know it can to remove all records for this interface
12897 if (m->SPSBrowseCallback)
12898 {
12899 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
12900 m->SPSBrowseCallback(m, &set->NetWakeBrowse, mDNSNULL, mDNSfalse);
12901 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
12902 }
12903
12904 // Reset our variables back to initial state, so we're ready for when NetWake is turned back on
12905 // (includes resetting NetWakeBrowse.ThisQInterval back to -1)
12906 InitializeNetWakeState(m, set);
12907 }
12908 }
12909
12910 mDNSexport mStatus mDNS_RegisterInterface(mDNS *const m, NetworkInterfaceInfo *set, mDNSBool flapping)
12911 {
12912 AuthRecord *rr;
12913 mDNSBool FirstOfType = mDNStrue;
12914 NetworkInterfaceInfo **p = &m->HostInterfaces;
12915
12916 if (!set->InterfaceID)
12917 { LogMsg("mDNS_RegisterInterface: Error! Tried to register a NetworkInterfaceInfo %#a with zero InterfaceID", &set->ip); return(mStatus_Invalid); }
12918
12919 if (!mDNSAddressIsValidNonZero(&set->mask))
12920 { LogMsg("mDNS_RegisterInterface: Error! Tried to register a NetworkInterfaceInfo %#a with invalid mask %#a", &set->ip, &set->mask); return(mStatus_Invalid); }
12921
12922 mDNS_Lock(m);
12923
12924 // Assume this interface will be active now, unless we find a duplicate already in the list
12925 set->InterfaceActive = mDNStrue;
12926 set->IPv4Available = (mDNSu8)(set->ip.type == mDNSAddrType_IPv4 && set->McastTxRx);
12927 set->IPv6Available = (mDNSu8)(set->ip.type == mDNSAddrType_IPv6 && set->McastTxRx);
12928
12929 InitializeNetWakeState(m, set);
12930
12931 // Scan list to see if this InterfaceID is already represented
12932 while (*p)
12933 {
12934 if (*p == set)
12935 {
12936 LogMsg("mDNS_RegisterInterface: Error! Tried to register a NetworkInterfaceInfo that's already in the list");
12937 mDNS_Unlock(m);
12938 return(mStatus_AlreadyRegistered);
12939 }
12940
12941 if ((*p)->InterfaceID == set->InterfaceID)
12942 {
12943 // This InterfaceID already represented by a different interface in the list, so mark this instance inactive for now
12944 set->InterfaceActive = mDNSfalse;
12945 if (set->ip.type == (*p)->ip.type) FirstOfType = mDNSfalse;
12946 if (set->ip.type == mDNSAddrType_IPv4 && set->McastTxRx) (*p)->IPv4Available = mDNStrue;
12947 if (set->ip.type == mDNSAddrType_IPv6 && set->McastTxRx) (*p)->IPv6Available = mDNStrue;
12948 }
12949
12950 p=&(*p)->next;
12951 }
12952
12953 set->next = mDNSNULL;
12954 *p = set;
12955
12956 if (set->Advertise)
12957 AdvertiseInterface(m, set);
12958
12959 LogInfo("mDNS_RegisterInterface: InterfaceID %d %s (%#a) %s",
12960 (uint32_t)set->InterfaceID, set->ifname, &set->ip,
12961 set->InterfaceActive ?
12962 "not represented in list; marking active and retriggering queries" :
12963 "already represented in list; marking inactive for now");
12964
12965 if (set->NetWake) mDNS_ActivateNetWake_internal(m, set);
12966
12967 // In early versions of OS X the IPv6 address remains on an interface even when the interface is turned off,
12968 // giving the false impression that there's an active representative of this interface when there really isn't.
12969 // Therefore, when registering an interface, we want to re-trigger our questions and re-probe our Resource Records,
12970 // even if we believe that we previously had an active representative of this interface.
12971 if (set->McastTxRx && (FirstOfType || set->InterfaceActive))
12972 {
12973 DNSQuestion *q;
12974 // Normally, after an interface comes up, we pause half a second before beginning probing.
12975 // This is to guard against cases where there's rapid interface changes, where we could be confused by
12976 // seeing packets we ourselves sent just moments ago (perhaps when this interface had a different address)
12977 // which are then echoed back after a short delay by some Ethernet switches and some 802.11 base stations.
12978 // We don't want to do a probe, and then see a stale echo of an announcement we ourselves sent,
12979 // and think it's a conflicting answer to our probe.
12980 // In the case of a flapping interface, we pause for five seconds, and reduce the announcement count to one packet.
12981 const mDNSs32 probedelay = flapping ? mDNSPlatformOneSecond * 5 : mDNSPlatformOneSecond / 2;
12982 const mDNSu8 numannounce = flapping ? (mDNSu8)1 : InitialAnnounceCount;
12983
12984 // Use a small amount of randomness:
12985 // In the case of a network administrator turning on an Ethernet hub so that all the
12986 // connected machines establish link at exactly the same time, we don't want them all
12987 // to go and hit the network with identical queries at exactly the same moment.
12988 // We set a random delay of up to InitialQuestionInterval (1/3 second).
12989 // We must *never* set m->SuppressSending to more than that (or set it repeatedly in a way
12990 // that causes mDNSResponder to remain in a prolonged state of SuppressSending, because
12991 // suppressing packet sending for more than about 1/3 second can cause protocol correctness
12992 // to start to break down (e.g. we don't answer probes fast enough, and get name conflicts).
12993 // See <rdar://problem/4073853> mDNS: m->SuppressSending set too enthusiastically
12994 if (!m->SuppressSending) m->SuppressSending = m->timenow + (mDNSs32)mDNSRandom((mDNSu32)InitialQuestionInterval);
12995
12996 if (flapping)
12997 {
12998 LogMsg("mDNS_RegisterInterface: Frequent transitions for interface %s (%#a)", set->ifname, &set->ip);
12999 m->mDNSStats.InterfaceUpFlap++;
13000 }
13001
13002 LogInfo("mDNS_RegisterInterface: %s (%#a) probedelay %d", set->ifname, &set->ip, probedelay);
13003
13004 if (m->SuppressProbes == 0 ||
13005 m->SuppressProbes - NonZeroTime(m->timenow + probedelay) < 0)
13006 m->SuppressProbes = NonZeroTime(m->timenow + probedelay);
13007
13008 // Include OWNER option in packets for 60 seconds after connecting to the network. Setting
13009 // it here also handles the wake up case as the network link comes UP after waking causing
13010 // us to reconnect to the network. If we do this as part of the wake up code, it is possible
13011 // that the network link comes UP after 60 seconds and we never set the OWNER option
13012 m->AnnounceOwner = NonZeroTime(m->timenow + 60 * mDNSPlatformOneSecond);
13013 LogInfo("mDNS_RegisterInterface: Setting AnnounceOwner");
13014
13015 m->mDNSStats.InterfaceUp++;
13016 for (q = m->Questions; q; q=q->next) // Scan our list of questions
13017 {
13018 if (mDNSOpaque16IsZero(q->TargetQID))
13019 {
13020 if (!q->InterfaceID || q->InterfaceID == set->InterfaceID) // If non-specific Q, or Q on this specific interface,
13021 { // then reactivate this question
13022 // If flapping, delay between first and second queries is nine seconds instead of one second
13023 mDNSBool dodelay = flapping && (q->FlappingInterface1 == set->InterfaceID || q->FlappingInterface2 == set->InterfaceID);
13024 mDNSs32 initial = dodelay ? InitialQuestionInterval * QuestionIntervalStep2 : InitialQuestionInterval;
13025 mDNSs32 qdelay = dodelay ? mDNSPlatformOneSecond * 5 : 0;
13026 if (dodelay) LogInfo("No cache records expired for %##s (%s); okay to delay questions a little", q->qname.c, DNSTypeName(q->qtype));
13027
13028 if (!q->ThisQInterval || q->ThisQInterval > initial)
13029 {
13030 q->ThisQInterval = initial;
13031 q->RequestUnicast = kDefaultRequestUnicastCount;
13032 }
13033 q->LastQTime = m->timenow - q->ThisQInterval + qdelay;
13034 q->RecentAnswerPkts = 0;
13035 // Change the salt
13036 ReInitAnonInfo(&q->AnonInfo, &q->qname);
13037 SetNextQueryTime(m,q);
13038 }
13039 }
13040 }
13041
13042 // For all our non-specific authoritative resource records (and any dormant records specific to this interface)
13043 // we now need them to re-probe if necessary, and then re-announce.
13044 for (rr = m->ResourceRecords; rr; rr=rr->next)
13045 {
13046 if (!rr->resrec.InterfaceID || rr->resrec.InterfaceID == set->InterfaceID)
13047 {
13048 // Change the salt
13049 ReInitAnonInfo(&rr->resrec.AnonInfo, rr->resrec.name);
13050 mDNSCoreRestartRegistration(m, rr, numannounce);
13051 }
13052 }
13053 #if APPLE_OSX_mDNSResponder && !TARGET_OS_IPHONE
13054 DNSSECProbe(m);
13055 #endif
13056 }
13057
13058 RestartRecordGetZoneData(m);
13059
13060 mDNS_UpdateAllowSleep(m);
13061
13062 mDNS_Unlock(m);
13063 return(mStatus_NoError);
13064 }
13065
13066 // Note: mDNS_DeregisterInterface calls mDNS_Deregister_internal which can call a user callback, which may change
13067 // the record list and/or question list.
13068 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
13069 mDNSexport void mDNS_DeregisterInterface(mDNS *const m, NetworkInterfaceInfo *set, mDNSBool flapping)
13070 {
13071 NetworkInterfaceInfo **p = &m->HostInterfaces;
13072 mDNSBool revalidate = mDNSfalse;
13073 NetworkInterfaceInfo *primary;
13074 NetworkInterfaceInfo *intf;
13075 AuthRecord *A;
13076
13077 mDNS_Lock(m);
13078
13079 // Find this record in our list
13080 while (*p && *p != set) p=&(*p)->next;
13081 if (!*p) { debugf("mDNS_DeregisterInterface: NetworkInterfaceInfo not found in list"); mDNS_Unlock(m); return; }
13082
13083 mDNS_DeactivateNetWake_internal(m, set);
13084
13085 // Unlink this record from our list
13086 *p = (*p)->next;
13087 set->next = mDNSNULL;
13088
13089 if (!set->InterfaceActive)
13090 {
13091 // If this interface not the active member of its set, update the v4/v6Available flags for the active member
13092 for (intf = m->HostInterfaces; intf; intf = intf->next)
13093 if (intf->InterfaceActive && intf->InterfaceID == set->InterfaceID)
13094 UpdateInterfaceProtocols(m, intf);
13095 }
13096 else
13097 {
13098 intf = FirstInterfaceForID(m, set->InterfaceID);
13099 if (intf)
13100 {
13101 LogInfo("mDNS_DeregisterInterface: Another representative of InterfaceID %d %s (%#a) exists;"
13102 " making it active", (uint32_t)set->InterfaceID, set->ifname, &set->ip);
13103 if (intf->InterfaceActive)
13104 LogMsg("mDNS_DeregisterInterface: ERROR intf->InterfaceActive already set for %s (%#a)", set->ifname, &set->ip);
13105 intf->InterfaceActive = mDNStrue;
13106 UpdateInterfaceProtocols(m, intf);
13107
13108 if (intf->NetWake) mDNS_ActivateNetWake_internal(m, intf);
13109
13110 // See if another representative *of the same type* exists. If not, we mave have gone from
13111 // dual-stack to v6-only (or v4-only) so we need to reconfirm which records are still valid.
13112 for (intf = m->HostInterfaces; intf; intf = intf->next)
13113 if (intf->InterfaceID == set->InterfaceID && intf->ip.type == set->ip.type)
13114 break;
13115 if (!intf) revalidate = mDNStrue;
13116 }
13117 else
13118 {
13119 mDNSu32 slot;
13120 CacheGroup *cg;
13121 CacheRecord *rr;
13122 DNSQuestion *q;
13123
13124 LogInfo("mDNS_DeregisterInterface: Last representative of InterfaceID %d %s (%#a) deregistered;"
13125 " marking questions etc. dormant", (uint32_t)set->InterfaceID, set->ifname, &set->ip);
13126
13127 m->mDNSStats.InterfaceDown++;
13128
13129 if (set->McastTxRx && flapping)
13130 {
13131 LogMsg("mDNS_DeregisterInterface: Frequent transitions for interface %s (%#a)", set->ifname, &set->ip);
13132 m->mDNSStats.InterfaceDownFlap++;
13133 }
13134
13135 // 1. Deactivate any questions specific to this interface, and tag appropriate questions
13136 // so that mDNS_RegisterInterface() knows how swiftly it needs to reactivate them
13137 for (q = m->Questions; q; q=q->next)
13138 {
13139 if (q->InterfaceID == set->InterfaceID) q->ThisQInterval = 0;
13140 if (!q->InterfaceID || q->InterfaceID == set->InterfaceID)
13141 {
13142 q->FlappingInterface2 = q->FlappingInterface1;
13143 q->FlappingInterface1 = set->InterfaceID; // Keep history of the last two interfaces to go away
13144 }
13145 }
13146
13147 // 2. Flush any cache records received on this interface
13148 revalidate = mDNSfalse; // Don't revalidate if we're flushing the records
13149 FORALL_CACHERECORDS(slot, cg, rr)
13150 {
13151 if (rr->resrec.InterfaceID == set->InterfaceID)
13152 {
13153 // If this interface is deemed flapping,
13154 // postpone deleting the cache records in case the interface comes back again
13155 if (set->McastTxRx && flapping)
13156 {
13157 // For a flapping interface we want these record to go away after 30 seconds
13158 mDNS_Reconfirm_internal(m, rr, kDefaultReconfirmTimeForFlappingInterface);
13159 // We set UnansweredQueries = MaxUnansweredQueries so we don't waste time doing any queries for them --
13160 // if the interface does come back, any relevant questions will be reactivated anyway
13161 rr->UnansweredQueries = MaxUnansweredQueries;
13162 }
13163 else
13164 {
13165 mDNS_PurgeCacheResourceRecord(m, rr);
13166 }
13167 }
13168 }
13169 }
13170 }
13171
13172 // If we still have address records referring to this one, update them.
13173 // This is safe, because this NetworkInterfaceInfo has already been unlinked from the list,
13174 // so the call to FindFirstAdvertisedInterface() won’t accidentally find it.
13175 primary = FindFirstAdvertisedInterface(m);
13176 A = primary ? &primary->RR_A : mDNSNULL;
13177 for (intf = m->HostInterfaces; intf; intf = intf->next)
13178 if (intf->RR_A.RRSet == &set->RR_A)
13179 intf->RR_A.RRSet = A;
13180
13181 // If we were advertising on this interface, deregister those address and reverse-lookup records now
13182 if (set->Advertise) DeadvertiseInterface(m, set);
13183
13184 // If we have any cache records received on this interface that went away, then re-verify them.
13185 // In some versions of OS X the IPv6 address remains on an interface even when the interface is turned off,
13186 // giving the false impression that there's an active representative of this interface when there really isn't.
13187 // Don't need to do this when shutting down, because *all* interfaces are about to go away
13188 if (revalidate && !m->ShutdownTime)
13189 {
13190 mDNSu32 slot;
13191 CacheGroup *cg;
13192 CacheRecord *rr;
13193 FORALL_CACHERECORDS(slot, cg, rr)
13194 if (rr->resrec.InterfaceID == set->InterfaceID)
13195 mDNS_Reconfirm_internal(m, rr, kDefaultReconfirmTimeForFlappingInterface);
13196 }
13197
13198 mDNS_UpdateAllowSleep(m);
13199
13200 mDNS_Unlock(m);
13201 }
13202
13203 mDNSlocal void SetAnonInfoSRS(ServiceRecordSet *sr, int NumSubTypes)
13204 {
13205 int i, len;
13206
13207 if (!sr->AnonData)
13208 return;
13209
13210 len = mDNSPlatformStrLen(sr->AnonData);
13211 if (sr->RR_PTR.resrec.AnonInfo)
13212 {
13213 LogMsg("SetAnonInfoSRS: Freeing AnonInfo for PTR record %##s, should have been freed already", sr->RR_PTR.resrec.name->c);
13214 FreeAnonInfo(sr->RR_PTR.resrec.AnonInfo);
13215 }
13216 sr->RR_PTR.resrec.AnonInfo = AllocateAnonInfo(sr->RR_PTR.resrec.name, sr->AnonData, len, mDNSNULL);
13217 for (i=0; i<NumSubTypes; i++)
13218 {
13219 if (sr->SubTypes[i].resrec.AnonInfo)
13220 {
13221 LogMsg("SetAnonInfoSRS: Freeing AnonInfo for subtype record %##s, should have been freed already", sr->SubTypes[i].resrec.name->c);
13222 FreeAnonInfo(sr->SubTypes[i].resrec.AnonInfo);
13223 }
13224 sr->SubTypes[i].resrec.AnonInfo = AllocateAnonInfo(sr->SubTypes[i].resrec.name, sr->AnonData, len, mDNSNULL);
13225 }
13226 }
13227
13228 mDNSlocal void ResetAnonInfoSRS(ServiceRecordSet *sr, int NumSubTypes)
13229 {
13230 int i;
13231
13232 if (!sr->AnonData)
13233 return;
13234 if (sr->RR_PTR.resrec.AnonInfo)
13235 {
13236 FreeAnonInfo(sr->RR_PTR.resrec.AnonInfo);
13237 sr->RR_PTR.resrec.AnonInfo = mDNSNULL;
13238 }
13239 for (i=0; i<NumSubTypes; i++)
13240 {
13241 if (sr->SubTypes[i].resrec.AnonInfo)
13242 {
13243 FreeAnonInfo(sr->SubTypes[i].resrec.AnonInfo);
13244 sr->SubTypes[i].resrec.AnonInfo = mDNSNULL;
13245 }
13246 }
13247 }
13248
13249 mDNSlocal void ServiceCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
13250 {
13251 ServiceRecordSet *sr = (ServiceRecordSet *)rr->RecordContext;
13252 (void)m; // Unused parameter
13253
13254 #if MDNS_DEBUGMSGS
13255 {
13256 char *msg = "Unknown result";
13257 if (result == mStatus_NoError) msg = "Name Registered";
13258 else if (result == mStatus_NameConflict) msg = "Name Conflict";
13259 else if (result == mStatus_MemFree) msg = "Memory Free";
13260 debugf("ServiceCallback: %##s (%s) %s (%d)", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), msg, result);
13261 }
13262 #endif
13263
13264 // Only pass on the NoError acknowledgement for the SRV record (when it finishes probing)
13265 if (result == mStatus_NoError && rr != &sr->RR_SRV) return;
13266
13267 // If we got a name conflict on either SRV or TXT, forcibly deregister this service, and record that we did that
13268 if (result == mStatus_NameConflict)
13269 {
13270 sr->Conflict = mDNStrue; // Record that this service set had a conflict
13271 mDNS_DeregisterService(m, sr); // Unlink the records from our list
13272 return;
13273 }
13274
13275 if (result == mStatus_MemFree)
13276 {
13277 // If the SRV/TXT/PTR records, or the _services._dns-sd._udp record, or any of the subtype PTR records,
13278 // are still in the process of deregistering, don't pass on the NameConflict/MemFree message until
13279 // every record is finished cleaning up.
13280 mDNSu32 i;
13281 ExtraResourceRecord *e = sr->Extras;
13282
13283 if (sr->RR_SRV.resrec.RecordType != kDNSRecordTypeUnregistered) return;
13284 if (sr->RR_TXT.resrec.RecordType != kDNSRecordTypeUnregistered) return;
13285 if (sr->RR_PTR.resrec.RecordType != kDNSRecordTypeUnregistered) return;
13286 if (sr->RR_ADV.resrec.RecordType != kDNSRecordTypeUnregistered) return;
13287 for (i=0; i<sr->NumSubTypes; i++) if (sr->SubTypes[i].resrec.RecordType != kDNSRecordTypeUnregistered) return;
13288
13289 while (e)
13290 {
13291 if (e->r.resrec.RecordType != kDNSRecordTypeUnregistered) return;
13292 e = e->next;
13293 }
13294 ResetAnonInfoSRS(sr, sr->NumSubTypes);
13295
13296 // If this ServiceRecordSet was forcibly deregistered, and now its memory is ready for reuse,
13297 // then we can now report the NameConflict to the client
13298 if (sr->Conflict) result = mStatus_NameConflict;
13299
13300 }
13301
13302 LogInfo("ServiceCallback: All records %s for %##s", (result == mStatus_MemFree ? "Unregistered" : "Registered"), sr->RR_PTR.resrec.name->c);
13303 // CAUTION: MUST NOT do anything more with sr after calling sr->Callback(), because the client's callback
13304 // function is allowed to do anything, including deregistering this service and freeing its memory.
13305 if (sr->ServiceCallback)
13306 sr->ServiceCallback(m, sr, result);
13307 }
13308
13309 mDNSlocal void NSSCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
13310 {
13311 ServiceRecordSet *sr = (ServiceRecordSet *)rr->RecordContext;
13312 if (sr->ServiceCallback)
13313 sr->ServiceCallback(m, sr, result);
13314 }
13315
13316
13317 // Derive AuthRecType from the coreFlag* values.
13318 // Note, this is not using the external flags values, kDNSServiceFlags*, defined in dns_sd.h.
13319 // It should be changed to do so once the use of coreFlag* is completely replaced with
13320 // the use the kDNSServiceFlags* definitions within mDNSResponder.
13321 mDNSlocal AuthRecType setAuthRecType(mDNSInterfaceID InterfaceID, mDNSu32 flags)
13322 {
13323 AuthRecType artype;
13324
13325 if (InterfaceID == mDNSInterface_LocalOnly)
13326 artype = AuthRecordLocalOnly;
13327 else if (InterfaceID == mDNSInterface_P2P || InterfaceID == mDNSInterface_BLE)
13328 artype = AuthRecordP2P;
13329 else if ((InterfaceID == mDNSInterface_Any) && (flags & coreFlagIncludeP2P)
13330 && (flags & coreFlagIncludeAWDL))
13331 artype = AuthRecordAnyIncludeAWDLandP2P;
13332 else if ((InterfaceID == mDNSInterface_Any) && (flags & coreFlagIncludeP2P))
13333 artype = AuthRecordAnyIncludeP2P;
13334 else if ((InterfaceID == mDNSInterface_Any) && (flags & coreFlagIncludeAWDL))
13335 artype = AuthRecordAnyIncludeAWDL;
13336 else
13337 artype = AuthRecordAny;
13338
13339 return artype;
13340 }
13341
13342 // Used to derive the original D2D specific flags specified by the client in the registration
13343 // when we don't have access to the original flag (kDNSServiceFlags*) values.
13344 mDNSexport mDNSu32 deriveD2DFlagsFromAuthRecType(AuthRecType authRecType)
13345 {
13346 mDNSu32 flags = 0;
13347 if ((authRecType == AuthRecordAnyIncludeP2P) || (authRecType == AuthRecordAnyIncludeAWDLandP2P))
13348 flags |= kDNSServiceFlagsIncludeP2P;
13349 else if ((authRecType == AuthRecordAnyIncludeAWDL) || (authRecType == AuthRecordAnyIncludeAWDLandP2P))
13350 flags |= kDNSServiceFlagsIncludeAWDL;
13351 return flags;
13352 }
13353
13354 // Note:
13355 // Name is first label of domain name (any dots in the name are actual dots, not label separators)
13356 // Type is service type (e.g. "_ipp._tcp.")
13357 // Domain is fully qualified domain name (i.e. ending with a null label)
13358 // We always register a TXT, even if it is empty (so that clients are not
13359 // left waiting forever looking for a nonexistent record.)
13360 // If the host parameter is mDNSNULL or the root domain (ASCII NUL),
13361 // then the default host name (m->MulticastHostname) is automatically used
13362 // If the optional target host parameter is set, then the storage it points to must remain valid for the lifetime of the service registration
13363 mDNSexport mStatus mDNS_RegisterService(mDNS *const m, ServiceRecordSet *sr,
13364 const domainlabel *const name, const domainname *const type, const domainname *const domain,
13365 const domainname *const host, mDNSIPPort port, const mDNSu8 txtinfo[], mDNSu16 txtlen,
13366 AuthRecord *SubTypes, mDNSu32 NumSubTypes,
13367 mDNSInterfaceID InterfaceID, mDNSServiceCallback Callback, void *Context, mDNSu32 flags)
13368 {
13369 mStatus err;
13370 mDNSu32 i;
13371 mDNSu32 hostTTL;
13372 AuthRecType artype;
13373 mDNSu8 recordType = (flags & coreFlagKnownUnique) ? kDNSRecordTypeKnownUnique : kDNSRecordTypeUnique;
13374
13375 sr->ServiceCallback = Callback;
13376 sr->ServiceContext = Context;
13377 sr->Conflict = mDNSfalse;
13378
13379 sr->Extras = mDNSNULL;
13380 sr->NumSubTypes = NumSubTypes;
13381 sr->SubTypes = SubTypes;
13382 sr->flags = flags;
13383
13384 artype = setAuthRecType(InterfaceID, flags);
13385
13386 // Initialize the AuthRecord objects to sane values
13387 // Need to initialize everything correctly *before* making the decision whether to do a RegisterNoSuchService and bail out
13388 mDNS_SetupResourceRecord(&sr->RR_ADV, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeAdvisory, artype, ServiceCallback, sr);
13389 mDNS_SetupResourceRecord(&sr->RR_PTR, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, artype, ServiceCallback, sr);
13390
13391 if (flags & coreFlagWakeOnly)
13392 {
13393 sr->RR_PTR.AuthFlags = AuthFlagsWakeOnly;
13394 }
13395
13396 if (SameDomainName(type, (const domainname *) "\x4" "_ubd" "\x4" "_tcp"))
13397 hostTTL = kHostNameSmallTTL;
13398 else
13399 hostTTL = kHostNameTTL;
13400
13401 mDNS_SetupResourceRecord(&sr->RR_SRV, mDNSNULL, InterfaceID, kDNSType_SRV, hostTTL, recordType, artype, ServiceCallback, sr);
13402 mDNS_SetupResourceRecord(&sr->RR_TXT, mDNSNULL, InterfaceID, kDNSType_TXT, kStandardTTL, kDNSRecordTypeUnique, artype, ServiceCallback, sr);
13403
13404 // If port number is zero, that means the client is really trying to do a RegisterNoSuchService
13405 if (mDNSIPPortIsZero(port))
13406 return(mDNS_RegisterNoSuchService(m, &sr->RR_SRV, name, type, domain, mDNSNULL, InterfaceID, NSSCallback, sr, flags));
13407
13408 // If the caller is registering an oversized TXT record,
13409 // it is the caller's responsibility to allocate a ServiceRecordSet structure that is large enough for it
13410 if (sr->RR_TXT.resrec.rdata->MaxRDLength < txtlen)
13411 sr->RR_TXT.resrec.rdata->MaxRDLength = txtlen;
13412
13413 // Set up the record names
13414 // For now we only create an advisory record for the main type, not for subtypes
13415 // We need to gain some operational experience before we decide if there's a need to create them for subtypes too
13416 if (ConstructServiceName(&sr->RR_ADV.namestorage, (const domainlabel*)"\x09_services", (const domainname*)"\x07_dns-sd\x04_udp", domain) == mDNSNULL)
13417 return(mStatus_BadParamErr);
13418 if (ConstructServiceName(&sr->RR_PTR.namestorage, mDNSNULL, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
13419 if (ConstructServiceName(&sr->RR_SRV.namestorage, name, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
13420 AssignDomainName(&sr->RR_TXT.namestorage, sr->RR_SRV.resrec.name);
13421
13422 // 1. Set up the ADV record rdata to advertise our service type
13423 AssignDomainName(&sr->RR_ADV.resrec.rdata->u.name, sr->RR_PTR.resrec.name);
13424
13425 // 2. Set up the PTR record rdata to point to our service name
13426 // We set up two additionals, so when a client asks for this PTR we automatically send the SRV and the TXT too
13427 // Note: uDNS registration code assumes that Additional1 points to the SRV record
13428 AssignDomainName(&sr->RR_PTR.resrec.rdata->u.name, sr->RR_SRV.resrec.name);
13429 sr->RR_PTR.Additional1 = &sr->RR_SRV;
13430 sr->RR_PTR.Additional2 = &sr->RR_TXT;
13431
13432 // 2a. Set up any subtype PTRs to point to our service name
13433 // If the client is using subtypes, it is the client's responsibility to have
13434 // already set the first label of the record name to the subtype being registered
13435 for (i=0; i<NumSubTypes; i++)
13436 {
13437 domainname st;
13438 AssignDomainName(&st, sr->SubTypes[i].resrec.name);
13439 st.c[1+st.c[0]] = 0; // Only want the first label, not the whole FQDN (particularly for mDNS_RenameAndReregisterService())
13440 AppendDomainName(&st, type);
13441 mDNS_SetupResourceRecord(&sr->SubTypes[i], mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, artype, ServiceCallback, sr);
13442 if (ConstructServiceName(&sr->SubTypes[i].namestorage, mDNSNULL, &st, domain) == mDNSNULL) return(mStatus_BadParamErr);
13443 AssignDomainName(&sr->SubTypes[i].resrec.rdata->u.name, &sr->RR_SRV.namestorage);
13444 sr->SubTypes[i].Additional1 = &sr->RR_SRV;
13445 sr->SubTypes[i].Additional2 = &sr->RR_TXT;
13446 }
13447
13448 SetAnonInfoSRS(sr, NumSubTypes);
13449
13450 // 3. Set up the SRV record rdata.
13451 sr->RR_SRV.resrec.rdata->u.srv.priority = 0;
13452 sr->RR_SRV.resrec.rdata->u.srv.weight = 0;
13453 sr->RR_SRV.resrec.rdata->u.srv.port = port;
13454
13455 // Setting AutoTarget tells DNS that the target of this SRV is to be automatically kept in sync with our host name
13456 if (host && host->c[0]) AssignDomainName(&sr->RR_SRV.resrec.rdata->u.srv.target, host);
13457 else { sr->RR_SRV.AutoTarget = Target_AutoHost; sr->RR_SRV.resrec.rdata->u.srv.target.c[0] = '\0'; }
13458
13459 // 4. Set up the TXT record rdata,
13460 // and set DependentOn because we're depending on the SRV record to find and resolve conflicts for us
13461 // Note: uDNS registration code assumes that DependentOn points to the SRV record
13462 if (txtinfo == mDNSNULL) sr->RR_TXT.resrec.rdlength = 0;
13463 else if (txtinfo != sr->RR_TXT.resrec.rdata->u.txt.c)
13464 {
13465 sr->RR_TXT.resrec.rdlength = txtlen;
13466 if (sr->RR_TXT.resrec.rdlength > sr->RR_TXT.resrec.rdata->MaxRDLength) return(mStatus_BadParamErr);
13467 mDNSPlatformMemCopy(sr->RR_TXT.resrec.rdata->u.txt.c, txtinfo, txtlen);
13468 }
13469 sr->RR_TXT.DependentOn = &sr->RR_SRV;
13470
13471 mDNS_Lock(m);
13472 // It is important that we register SRV first. uDNS assumes that SRV is registered first so
13473 // that if the SRV cannot find a target, rest of the records that belong to this service
13474 // will not be activated.
13475 err = mDNS_Register_internal(m, &sr->RR_SRV);
13476 // If we can't register the SRV record due to errors, bail out. It has not been inserted in
13477 // any list and hence no need to deregister. We could probably do similar checks for other
13478 // records below and bail out. For now, this seems to be sufficient to address rdar://9304275
13479 if (err)
13480 {
13481 mDNS_Unlock(m);
13482 return err;
13483 }
13484 if (!err) err = mDNS_Register_internal(m, &sr->RR_TXT);
13485 // We register the RR_PTR last, because we want to be sure that in the event of a forced call to
13486 // mDNS_StartExit, the RR_PTR will be the last one to be forcibly deregistered, since that is what triggers
13487 // the mStatus_MemFree callback to ServiceCallback, which in turn passes on the mStatus_MemFree back to
13488 // the client callback, which is then at liberty to free the ServiceRecordSet memory at will. We need to
13489 // make sure we've deregistered all our records and done any other necessary cleanup before that happens.
13490 if (!err) err = mDNS_Register_internal(m, &sr->RR_ADV);
13491 for (i=0; i<NumSubTypes; i++) if (!err) err = mDNS_Register_internal(m, &sr->SubTypes[i]);
13492 if (!err) err = mDNS_Register_internal(m, &sr->RR_PTR);
13493
13494 mDNS_Unlock(m);
13495
13496 if (err) mDNS_DeregisterService(m, sr);
13497 return(err);
13498 }
13499
13500 mDNSexport mStatus mDNS_AddRecordToService(mDNS *const m, ServiceRecordSet *sr,
13501 ExtraResourceRecord *extra, RData *rdata, mDNSu32 ttl, mDNSu32 flags)
13502 {
13503 ExtraResourceRecord **e;
13504 mStatus status;
13505 AuthRecType artype;
13506 mDNSInterfaceID InterfaceID = sr->RR_PTR.resrec.InterfaceID;
13507 ResourceRecord *rr;
13508
13509 artype = setAuthRecType(InterfaceID, flags);
13510
13511 extra->next = mDNSNULL;
13512 mDNS_SetupResourceRecord(&extra->r, rdata, sr->RR_PTR.resrec.InterfaceID,
13513 extra->r.resrec.rrtype, ttl, kDNSRecordTypeUnique, artype, ServiceCallback, sr);
13514 AssignDomainName(&extra->r.namestorage, sr->RR_SRV.resrec.name);
13515
13516 mDNS_Lock(m);
13517 rr = mDNSNULL;
13518 if (extra->r.resrec.rrtype == kDNSType_TXT)
13519 {
13520 if (sr->RR_TXT.resrec.RecordType & kDNSRecordTypeUniqueMask) rr = &sr->RR_TXT.resrec;
13521 }
13522 else if (extra->r.resrec.rrtype == kDNSType_SRV)
13523 {
13524 if (sr->RR_SRV.resrec.RecordType & kDNSRecordTypeUniqueMask) rr = &sr->RR_SRV.resrec;
13525 }
13526
13527 if (!rr)
13528 {
13529 ExtraResourceRecord *srExtra;
13530
13531 for (srExtra = sr->Extras; srExtra; srExtra = srExtra->next)
13532 {
13533 if ((srExtra->r.resrec.rrtype == extra->r.resrec.rrtype) && (srExtra->r.resrec.RecordType & kDNSRecordTypeUniqueMask))
13534 {
13535 rr = &srExtra->r.resrec;
13536 break;
13537 }
13538 }
13539 }
13540
13541 if (rr && (extra->r.resrec.rroriginalttl != rr->rroriginalttl))
13542 {
13543 LogMsg("mDNS_AddRecordToService: Correcting TTL from %4d to %4d for %s",
13544 extra->r.resrec.rroriginalttl, rr->rroriginalttl, RRDisplayString(m, &extra->r.resrec));
13545 extra->r.resrec.rroriginalttl = rr->rroriginalttl;
13546 }
13547
13548 e = &sr->Extras;
13549 while (*e) e = &(*e)->next;
13550
13551 extra->r.DependentOn = &sr->RR_SRV;
13552
13553 debugf("mDNS_AddRecordToService adding record to %##s %s %d",
13554 extra->r.resrec.name->c, DNSTypeName(extra->r.resrec.rrtype), extra->r.resrec.rdlength);
13555
13556 status = mDNS_Register_internal(m, &extra->r);
13557 if (status == mStatus_NoError) *e = extra;
13558
13559 mDNS_Unlock(m);
13560 return(status);
13561 }
13562
13563 mDNSexport mStatus mDNS_RemoveRecordFromService(mDNS *const m, ServiceRecordSet *sr, ExtraResourceRecord *extra,
13564 mDNSRecordCallback MemFreeCallback, void *Context)
13565 {
13566 ExtraResourceRecord **e;
13567 mStatus status;
13568
13569 mDNS_Lock(m);
13570 e = &sr->Extras;
13571 while (*e && *e != extra) e = &(*e)->next;
13572 if (!*e)
13573 {
13574 debugf("mDNS_RemoveRecordFromService failed to remove record from %##s", extra->r.resrec.name->c);
13575 status = mStatus_BadReferenceErr;
13576 }
13577 else
13578 {
13579 debugf("mDNS_RemoveRecordFromService removing record from %##s", extra->r.resrec.name->c);
13580 extra->r.RecordCallback = MemFreeCallback;
13581 extra->r.RecordContext = Context;
13582 *e = (*e)->next;
13583 status = mDNS_Deregister_internal(m, &extra->r, mDNS_Dereg_normal);
13584 }
13585 mDNS_Unlock(m);
13586 return(status);
13587 }
13588
13589 mDNSexport mStatus mDNS_RenameAndReregisterService(mDNS *const m, ServiceRecordSet *const sr, const domainlabel *newname)
13590 {
13591 // Note: Don't need to use mDNS_Lock(m) here, because this code is just using public routines
13592 // mDNS_RegisterService() and mDNS_AddRecordToService(), which do the right locking internally.
13593 domainlabel name1, name2;
13594 domainname type, domain;
13595 const domainname *host = sr->RR_SRV.AutoTarget ? mDNSNULL : &sr->RR_SRV.resrec.rdata->u.srv.target;
13596 ExtraResourceRecord *extras = sr->Extras;
13597 mStatus err;
13598
13599 DeconstructServiceName(sr->RR_SRV.resrec.name, &name1, &type, &domain);
13600 if (!newname)
13601 {
13602 name2 = name1;
13603 IncrementLabelSuffix(&name2, mDNStrue);
13604 newname = &name2;
13605 }
13606
13607 if (SameDomainName(&domain, &localdomain))
13608 debugf("%##s service renamed from \"%#s\" to \"%#s\"", type.c, name1.c, newname->c);
13609 else debugf("%##s service (domain %##s) renamed from \"%#s\" to \"%#s\"",type.c, domain.c, name1.c, newname->c);
13610
13611 err = mDNS_RegisterService(m, sr, newname, &type, &domain,
13612 host, sr->RR_SRV.resrec.rdata->u.srv.port, sr->RR_TXT.resrec.rdata->u.txt.c, sr->RR_TXT.resrec.rdlength,
13613 sr->SubTypes, sr->NumSubTypes,
13614 sr->RR_PTR.resrec.InterfaceID, sr->ServiceCallback, sr->ServiceContext, sr->flags);
13615
13616 // mDNS_RegisterService() just reset sr->Extras to NULL.
13617 // Fortunately we already grabbed ourselves a copy of this pointer (above), so we can now run
13618 // through the old list of extra records, and re-add them to our freshly created service registration
13619 while (!err && extras)
13620 {
13621 ExtraResourceRecord *e = extras;
13622 extras = extras->next;
13623 err = mDNS_AddRecordToService(m, sr, e, e->r.resrec.rdata, e->r.resrec.rroriginalttl, 0);
13624 }
13625
13626 return(err);
13627 }
13628
13629 // Note: mDNS_DeregisterService calls mDNS_Deregister_internal which can call a user callback,
13630 // which may change the record list and/or question list.
13631 // Any code walking either list must use the CurrentQuestion and/or CurrentRecord mechanism to protect against this.
13632 mDNSexport mStatus mDNS_DeregisterService_drt(mDNS *const m, ServiceRecordSet *sr, mDNS_Dereg_type drt)
13633 {
13634 // If port number is zero, that means this was actually registered using mDNS_RegisterNoSuchService()
13635 if (mDNSIPPortIsZero(sr->RR_SRV.resrec.rdata->u.srv.port)) return(mDNS_DeregisterNoSuchService(m, &sr->RR_SRV));
13636
13637 if (sr->RR_PTR.resrec.RecordType == kDNSRecordTypeUnregistered)
13638 {
13639 debugf("Service set for %##s already deregistered", sr->RR_SRV.resrec.name->c);
13640 return(mStatus_BadReferenceErr);
13641 }
13642 else if (sr->RR_PTR.resrec.RecordType == kDNSRecordTypeDeregistering)
13643 {
13644 LogInfo("Service set for %##s already in the process of deregistering", sr->RR_SRV.resrec.name->c);
13645 // Avoid race condition:
13646 // If a service gets a conflict, then we set the Conflict flag to tell us to generate
13647 // an mStatus_NameConflict message when we get the mStatus_MemFree for our PTR record.
13648 // If the client happens to deregister the service in the middle of that process, then
13649 // we clear the flag back to the normal state, so that we deliver a plain mStatus_MemFree
13650 // instead of incorrectly promoting it to mStatus_NameConflict.
13651 // This race condition is exposed particularly when the conformance test generates
13652 // a whole batch of simultaneous conflicts across a range of services all advertised
13653 // using the same system default name, and if we don't take this precaution then
13654 // we end up incrementing m->nicelabel multiple times instead of just once.
13655 // <rdar://problem/4060169> Bug when auto-renaming Computer Name after name collision
13656 sr->Conflict = mDNSfalse;
13657 return(mStatus_NoError);
13658 }
13659 else
13660 {
13661 mDNSu32 i;
13662 mStatus status;
13663 ExtraResourceRecord *e;
13664 mDNS_Lock(m);
13665 e = sr->Extras;
13666
13667 // We use mDNS_Dereg_repeat because, in the event of a collision, some or all of the
13668 // SRV, TXT, or Extra records could have already been automatically deregistered, and that's okay
13669 mDNS_Deregister_internal(m, &sr->RR_SRV, mDNS_Dereg_repeat);
13670 mDNS_Deregister_internal(m, &sr->RR_TXT, mDNS_Dereg_repeat);
13671
13672 mDNS_Deregister_internal(m, &sr->RR_ADV, drt);
13673
13674 // We deregister all of the extra records, but we leave the sr->Extras list intact
13675 // in case the client wants to do a RenameAndReregister and reinstate the registration
13676 while (e)
13677 {
13678 mDNS_Deregister_internal(m, &e->r, mDNS_Dereg_repeat);
13679 e = e->next;
13680 }
13681
13682 for (i=0; i<sr->NumSubTypes; i++)
13683 mDNS_Deregister_internal(m, &sr->SubTypes[i], drt);
13684
13685 status = mDNS_Deregister_internal(m, &sr->RR_PTR, drt);
13686 mDNS_Unlock(m);
13687 return(status);
13688 }
13689 }
13690
13691 // Create a registration that asserts that no such service exists with this name.
13692 // This can be useful where there is a given function is available through several protocols.
13693 // For example, a printer called "Stuart's Printer" may implement printing via the "pdl-datastream" and "IPP"
13694 // protocols, but not via "LPR". In this case it would be prudent for the printer to assert the non-existence of an
13695 // "LPR" service called "Stuart's Printer". Without this precaution, another printer than offers only "LPR" printing
13696 // could inadvertently advertise its service under the same name "Stuart's Printer", which might be confusing for users.
13697 mDNSexport mStatus mDNS_RegisterNoSuchService(mDNS *const m, AuthRecord *const rr,
13698 const domainlabel *const name, const domainname *const type, const domainname *const domain,
13699 const domainname *const host,
13700 const mDNSInterfaceID InterfaceID, mDNSRecordCallback Callback, void *Context, mDNSu32 flags)
13701 {
13702 AuthRecType artype;
13703
13704 artype = setAuthRecType(InterfaceID, flags);
13705
13706 mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, kDNSType_SRV, kHostNameTTL, kDNSRecordTypeUnique, artype, Callback, Context);
13707 if (ConstructServiceName(&rr->namestorage, name, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
13708 rr->resrec.rdata->u.srv.priority = 0;
13709 rr->resrec.rdata->u.srv.weight = 0;
13710 rr->resrec.rdata->u.srv.port = zeroIPPort;
13711 if (host && host->c[0]) AssignDomainName(&rr->resrec.rdata->u.srv.target, host);
13712 else rr->AutoTarget = Target_AutoHost;
13713 return(mDNS_Register(m, rr));
13714 }
13715
13716 mDNSexport mStatus mDNS_AdvertiseDomains(mDNS *const m, AuthRecord *rr,
13717 mDNS_DomainType DomainType, const mDNSInterfaceID InterfaceID, char *domname)
13718 {
13719 AuthRecType artype;
13720
13721 if (InterfaceID == mDNSInterface_LocalOnly)
13722 artype = AuthRecordLocalOnly;
13723 else if (InterfaceID == mDNSInterface_P2P || InterfaceID == mDNSInterface_BLE)
13724 artype = AuthRecordP2P;
13725 else
13726 artype = AuthRecordAny;
13727 mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, artype, mDNSNULL, mDNSNULL);
13728 if (!MakeDomainNameFromDNSNameString(&rr->namestorage, mDNS_DomainTypeNames[DomainType])) return(mStatus_BadParamErr);
13729 if (!MakeDomainNameFromDNSNameString(&rr->resrec.rdata->u.name, domname)) return(mStatus_BadParamErr);
13730 return(mDNS_Register(m, rr));
13731 }
13732
13733 mDNSlocal mDNSBool mDNS_IdUsedInResourceRecordsList(mDNS * const m, mDNSOpaque16 id)
13734 {
13735 AuthRecord *r;
13736 for (r = m->ResourceRecords; r; r=r->next) if (mDNSSameOpaque16(id, r->updateid)) return mDNStrue;
13737 return mDNSfalse;
13738 }
13739
13740 mDNSlocal mDNSBool mDNS_IdUsedInQuestionsList(mDNS * const m, mDNSOpaque16 id)
13741 {
13742 DNSQuestion *q;
13743 for (q = m->Questions; q; q=q->next) if (mDNSSameOpaque16(id, q->TargetQID)) return mDNStrue;
13744 return mDNSfalse;
13745 }
13746
13747 mDNSexport mDNSOpaque16 mDNS_NewMessageID(mDNS * const m)
13748 {
13749 mDNSOpaque16 id;
13750 int i;
13751
13752 for (i=0; i<10; i++)
13753 {
13754 id = mDNSOpaque16fromIntVal(1 + (mDNSu16)mDNSRandom(0xFFFE));
13755 if (!mDNS_IdUsedInResourceRecordsList(m, id) && !mDNS_IdUsedInQuestionsList(m, id)) break;
13756 }
13757
13758 debugf("mDNS_NewMessageID: %5d", mDNSVal16(id));
13759
13760 return id;
13761 }
13762
13763 // ***************************************************************************
13764 #if COMPILER_LIKES_PRAGMA_MARK
13765 #pragma mark -
13766 #pragma mark - Sleep Proxy Server
13767 #endif
13768
13769 mDNSlocal void RestartARPProbing(mDNS *const m, AuthRecord *const rr)
13770 {
13771 // If we see an ARP from a machine we think is sleeping, then either
13772 // (i) the machine has woken, or
13773 // (ii) it's just a stray old packet from before the machine slept
13774 // To handle the second case, we reset ProbeCount, so we'll suppress our own answers for a while, to avoid
13775 // generating ARP conflicts with a waking machine, and set rr->LastAPTime so we'll start probing again in 10 seconds.
13776 // If the machine has just woken then we'll discard our records when we see the first new mDNS probe from that machine.
13777 // If it was a stray old packet, then after 10 seconds we'll probe again and then start answering ARPs again. In this case we *do*
13778 // need to send new ARP Announcements, because the owner's ARP broadcasts will have updated neighboring ARP caches, so we need to
13779 // re-assert our (temporary) ownership of that IP address in order to receive subsequent packets addressed to that IPv4 address.
13780
13781 rr->resrec.RecordType = kDNSRecordTypeUnique;
13782 rr->ProbeCount = DefaultProbeCountForTypeUnique;
13783 rr->ProbeRestartCount++;
13784
13785 // If we haven't started announcing yet (and we're not already in ten-second-delay mode) the machine is probably
13786 // still going to sleep, so we just reset rr->ProbeCount so we'll continue probing until it stops responding.
13787 // If we *have* started announcing, the machine is probably in the process of waking back up, so in that case
13788 // we're more cautious and we wait ten seconds before probing it again. We do this because while waking from
13789 // sleep, some network interfaces tend to lose or delay inbound packets, and without this delay, if the waking machine
13790 // didn't answer our three probes within three seconds then we'd announce and cause it an unnecessary address conflict.
13791 if (rr->AnnounceCount == InitialAnnounceCount && m->timenow - rr->LastAPTime >= 0)
13792 InitializeLastAPTime(m, rr);
13793 else
13794 {
13795 rr->AnnounceCount = InitialAnnounceCount;
13796 rr->ThisAPInterval = mDNSPlatformOneSecond;
13797 rr->LastAPTime = m->timenow + mDNSPlatformOneSecond * 9; // Send first packet at rr->LastAPTime + rr->ThisAPInterval, i.e. 10 seconds from now
13798 SetNextAnnounceProbeTime(m, rr);
13799 }
13800 }
13801
13802 mDNSlocal void mDNSCoreReceiveRawARP(mDNS *const m, const ARP_EthIP *const arp, const mDNSInterfaceID InterfaceID)
13803 {
13804 static const mDNSOpaque16 ARP_op_request = { { 0, 1 } };
13805 AuthRecord *rr;
13806 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
13807 if (!intf) return;
13808
13809 mDNS_Lock(m);
13810
13811 // Pass 1:
13812 // Process ARP Requests and Probes (but not Announcements), and generate an ARP Reply if necessary.
13813 // We also process ARPs from our own kernel (and 'answer' them by injecting a local ARP table entry)
13814 // We ignore ARP Announcements here -- Announcements are not questions, they're assertions, so we don't need to answer them.
13815 // The times we might need to react to an ARP Announcement are:
13816 // (i) as an indication that the host in question has not gone to sleep yet (so we should delay beginning to proxy for it) or
13817 // (ii) if it's a conflicting Announcement from another host
13818 // -- and we check for these in Pass 2 below.
13819 if (mDNSSameOpaque16(arp->op, ARP_op_request) && !mDNSSameIPv4Address(arp->spa, arp->tpa))
13820 {
13821 for (rr = m->ResourceRecords; rr; rr=rr->next)
13822 if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
13823 rr->AddressProxy.type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->AddressProxy.ip.v4, arp->tpa))
13824 {
13825 static const char msg1[] = "ARP Req from owner -- re-probing";
13826 static const char msg2[] = "Ignoring ARP Request from ";
13827 static const char msg3[] = "Creating Local ARP Cache entry ";
13828 static const char msg4[] = "Answering ARP Request from ";
13829 const char *const msg = mDNSSameEthAddress(&arp->sha, &rr->WakeUp.IMAC) ? msg1 :
13830 (rr->AnnounceCount == InitialAnnounceCount) ? msg2 :
13831 mDNSSameEthAddress(&arp->sha, &intf->MAC) ? msg3 : msg4;
13832 LogMsg("Arp %-7s %s %.6a %.4a for %.4a -- H-MAC %.6a I-MAC %.6a %s",
13833 intf->ifname, msg, arp->sha.b, arp->spa.b, arp->tpa.b,
13834 &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13835 if (msg == msg1)
13836 {
13837 if ( rr->ProbeRestartCount < MAX_PROBE_RESTARTS)
13838 RestartARPProbing(m, rr);
13839 else
13840 LogSPS("Reached maximum number of restarts for probing - %s", ARDisplayString(m,rr));
13841 }
13842 else if (msg == msg3)
13843 {
13844 mDNSPlatformSetLocalAddressCacheEntry(m, &rr->AddressProxy, &rr->WakeUp.IMAC, InterfaceID);
13845 }
13846 else if (msg == msg4)
13847 {
13848 SendARP(m, 2, rr, (mDNSv4Addr *)arp->tpa.b, &arp->sha, (mDNSv4Addr *)arp->spa.b, &arp->sha);
13849 }
13850 }
13851 }
13852
13853 // Pass 2:
13854 // For all types of ARP packet we check the Sender IP address to make sure it doesn't conflict with any AddressProxy record we're holding.
13855 // (Strictly speaking we're only checking Announcement/Request/Reply packets, since ARP Probes have zero Sender IP address,
13856 // so by definition (and by design) they can never conflict with any real (i.e. non-zero) IP address).
13857 // We ignore ARPs we sent ourselves (Sender MAC address is our MAC address) because our own proxy ARPs do not constitute a conflict that we need to handle.
13858 // If we see an apparently conflicting ARP, we check the sender hardware address:
13859 // If the sender hardware address is the original owner this is benign, so we just suppress our own proxy answering for a while longer.
13860 // If the sender hardware address is *not* the original owner, then this is a conflict, and we need to wake the sleeping machine to handle it.
13861 if (mDNSSameEthAddress(&arp->sha, &intf->MAC))
13862 debugf("ARP from self for %.4a", arp->tpa.b);
13863 else
13864 {
13865 if (!mDNSSameIPv4Address(arp->spa, zerov4Addr))
13866 for (rr = m->ResourceRecords; rr; rr=rr->next)
13867 if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
13868 rr->AddressProxy.type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->AddressProxy.ip.v4, arp->spa) && (rr->ProbeRestartCount < MAX_PROBE_RESTARTS))
13869 {
13870 if (mDNSSameEthAddress(&zeroEthAddr, &rr->WakeUp.HMAC))
13871 {
13872 LogMsg("%-7s ARP from %.6a %.4a for %.4a -- Invalid H-MAC %.6a I-MAC %.6a %s", intf->ifname,
13873 arp->sha.b, arp->spa.b, arp->tpa.b, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13874 }
13875 else
13876 {
13877 RestartARPProbing(m, rr);
13878 if (mDNSSameEthAddress(&arp->sha, &rr->WakeUp.IMAC))
13879 {
13880 LogMsg("%-7s ARP %s from owner %.6a %.4a for %-15.4a -- re-starting probing for %s", intf->ifname,
13881 mDNSSameIPv4Address(arp->spa, arp->tpa) ? "Announcement " : mDNSSameOpaque16(arp->op, ARP_op_request) ? "Request " : "Response ",
13882 arp->sha.b, arp->spa.b, arp->tpa.b, ARDisplayString(m, rr));
13883 }
13884 else
13885 {
13886 LogMsg("%-7s Conflicting ARP from %.6a %.4a for %.4a -- waking H-MAC %.6a I-MAC %.6a %s", intf->ifname,
13887 arp->sha.b, arp->spa.b, arp->tpa.b, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13888 ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
13889 }
13890 }
13891 }
13892 }
13893
13894 mDNS_Unlock(m);
13895 }
13896
13897 /*
13898 // Option 1 is Source Link Layer Address Option
13899 // Option 2 is Target Link Layer Address Option
13900 mDNSlocal const mDNSEthAddr *GetLinkLayerAddressOption(const IPv6NDP *const ndp, const mDNSu8 *const end, mDNSu8 op)
13901 {
13902 const mDNSu8 *options = (mDNSu8 *)(ndp+1);
13903 while (options < end)
13904 {
13905 debugf("NDP Option %02X len %2d %d", options[0], options[1], end - options);
13906 if (options[0] == op && options[1] == 1) return (const mDNSEthAddr*)(options+2);
13907 options += options[1] * 8;
13908 }
13909 return mDNSNULL;
13910 }
13911 */
13912
13913 mDNSlocal void mDNSCoreReceiveRawND(mDNS *const m, const mDNSEthAddr *const sha, const mDNSv6Addr *spa,
13914 const IPv6NDP *const ndp, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID)
13915 {
13916 AuthRecord *rr;
13917 NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
13918 if (!intf) return;
13919
13920 mDNS_Lock(m);
13921
13922 // Pass 1: Process Neighbor Solicitations, and generate a Neighbor Advertisement if necessary.
13923 if (ndp->type == NDP_Sol)
13924 {
13925 //const mDNSEthAddr *const sha = GetLinkLayerAddressOption(ndp, end, NDP_SrcLL);
13926 (void)end;
13927 for (rr = m->ResourceRecords; rr; rr=rr->next)
13928 if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
13929 rr->AddressProxy.type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->AddressProxy.ip.v6, ndp->target))
13930 {
13931 static const char msg1[] = "NDP Req from owner -- re-probing";
13932 static const char msg2[] = "Ignoring NDP Request from ";
13933 static const char msg3[] = "Creating Local NDP Cache entry ";
13934 static const char msg4[] = "Answering NDP Request from ";
13935 static const char msg5[] = "Answering NDP Probe from ";
13936 const char *const msg = sha && mDNSSameEthAddress(sha, &rr->WakeUp.IMAC) ? msg1 :
13937 (rr->AnnounceCount == InitialAnnounceCount) ? msg2 :
13938 sha && mDNSSameEthAddress(sha, &intf->MAC) ? msg3 :
13939 spa && mDNSIPv6AddressIsZero(*spa) ? msg4 : msg5;
13940 LogSPS("%-7s %s %.6a %.16a for %.16a -- H-MAC %.6a I-MAC %.6a %s",
13941 intf->ifname, msg, sha, spa, &ndp->target, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13942 if (msg == msg1)
13943 {
13944 if (rr->ProbeRestartCount < MAX_PROBE_RESTARTS)
13945 RestartARPProbing(m, rr);
13946 else
13947 LogSPS("Reached maximum number of restarts for probing - %s", ARDisplayString(m,rr));
13948 }
13949 else if (msg == msg3)
13950 mDNSPlatformSetLocalAddressCacheEntry(m, &rr->AddressProxy, &rr->WakeUp.IMAC, InterfaceID);
13951 else if (msg == msg4)
13952 SendNDP(m, NDP_Adv, NDP_Solicited, rr, &ndp->target, mDNSNULL, spa, sha);
13953 else if (msg == msg5)
13954 SendNDP(m, NDP_Adv, 0, rr, &ndp->target, mDNSNULL, &AllHosts_v6, &AllHosts_v6_Eth);
13955 }
13956 }
13957
13958 // Pass 2: For all types of NDP packet we check the Sender IP address to make sure it doesn't conflict with any AddressProxy record we're holding.
13959 if (mDNSSameEthAddress(sha, &intf->MAC))
13960 debugf("NDP from self for %.16a", &ndp->target);
13961 else
13962 {
13963 // For Neighbor Advertisements we check the Target address field, not the actual IPv6 source address.
13964 // When a machine has both link-local and routable IPv6 addresses, it may send NDP packets making assertions
13965 // about its routable IPv6 address, using its link-local address as the source address for all NDP packets.
13966 // Hence it is the NDP target address we care about, not the actual packet source address.
13967 if (ndp->type == NDP_Adv) spa = &ndp->target;
13968 if (!mDNSSameIPv6Address(*spa, zerov6Addr))
13969 for (rr = m->ResourceRecords; rr; rr=rr->next)
13970 if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
13971 rr->AddressProxy.type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->AddressProxy.ip.v6, *spa) && (rr->ProbeRestartCount < MAX_PROBE_RESTARTS))
13972 {
13973 if (mDNSSameEthAddress(&zeroEthAddr, &rr->WakeUp.HMAC))
13974 {
13975 LogSPS("%-7s NDP from %.6a %.16a for %.16a -- Invalid H-MAC %.6a I-MAC %.6a %s", intf->ifname,
13976 sha, spa, &ndp->target, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13977 }
13978 else
13979 {
13980 RestartARPProbing(m, rr);
13981 if (mDNSSameEthAddress(sha, &rr->WakeUp.IMAC))
13982 {
13983 LogSPS("%-7s NDP %s from owner %.6a %.16a for %.16a -- re-starting probing for %s", intf->ifname,
13984 ndp->type == NDP_Sol ? "Solicitation " : "Advertisement", sha, spa, &ndp->target, ARDisplayString(m, rr));
13985 }
13986 else
13987 {
13988 LogMsg("%-7s Conflicting NDP from %.6a %.16a for %.16a -- waking H-MAC %.6a I-MAC %.6a %s", intf->ifname,
13989 sha, spa, &ndp->target, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
13990 ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
13991 }
13992 }
13993 }
13994 }
13995
13996 mDNS_Unlock(m);
13997 }
13998
13999 mDNSlocal void mDNSCoreReceiveRawTransportPacket(mDNS *const m, const mDNSEthAddr *const sha, const mDNSAddr *const src, const mDNSAddr *const dst, const mDNSu8 protocol,
14000 const mDNSu8 *const p, const TransportLayerPacket *const t, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID, const mDNSu16 len)
14001 {
14002 const mDNSIPPort port = (protocol == 0x06) ? t->tcp.dst : (protocol == 0x11) ? t->udp.dst : zeroIPPort;
14003 mDNSBool wake = mDNSfalse;
14004 mDNSBool kaWake = mDNSfalse;
14005
14006 switch (protocol)
14007 {
14008 #define XX wake ? "Received" : "Ignoring", end-p
14009 case 0x01: LogSPS("Ignoring %d-byte ICMP from %#a to %#a", end-p, src, dst);
14010 break;
14011
14012 case 0x06: {
14013 AuthRecord *kr;
14014 mDNSu32 seq, ack;
14015 #define TH_FIN 0x01
14016 #define TH_SYN 0x02
14017 #define TH_RST 0x04
14018 #define TH_ACK 0x10
14019
14020 kr = mDNS_MatchKeepaliveInfo(m, dst, src, port, t->tcp.src, &seq, &ack);
14021 if (kr)
14022 {
14023 LogSPS("mDNSCoreReceiveRawTransportPacket: Found a Keepalive record from %#a:%d to %#a:%d", src, mDNSVal16(t->tcp.src), dst, mDNSVal16(port));
14024 // Plan to wake if
14025 // (a) RST or FIN is set (the keepalive that we sent could have caused a reset)
14026 // (b) packet that contains new data and acks a sequence number higher than the one
14027 // we have been sending in the keepalive
14028
14029 wake = ((t->tcp.flags & TH_RST) || (t->tcp.flags & TH_FIN)) ;
14030 if (!wake)
14031 {
14032 mDNSu8 *ptr;
14033 mDNSu32 pseq, pack;
14034 mDNSBool data = mDNSfalse;
14035 mDNSu8 tcphlen;
14036
14037 // Convert to host order
14038 ptr = (mDNSu8 *)&seq;
14039 seq = ptr[0] << 24 | ptr[1] << 16 | ptr[2] << 8 | ptr[3];
14040
14041 ptr = (mDNSu8 *)&ack;
14042 ack = ptr[0] << 24 | ptr[1] << 16 | ptr[2] << 8 | ptr[3];
14043
14044 pseq = t->tcp.seq;
14045 ptr = (mDNSu8 *)&pseq;
14046 pseq = ptr[0] << 24 | ptr[1] << 16 | ptr[2] << 8 | ptr[3];
14047
14048 pack = t->tcp.ack;
14049 ptr = (mDNSu8 *)&pack;
14050 pack = ptr[0] << 24 | ptr[1] << 16 | ptr[2] << 8 | ptr[3];
14051
14052 // If the other side is acking one more than our sequence number (keepalive is one
14053 // less than the last valid sequence sent) and it's sequence is more than what we
14054 // acked before
14055 //if (end - p - 34 - ((t->tcp.offset >> 4) * 4) > 0) data = mDNStrue;
14056 tcphlen = ((t->tcp.offset >> 4) * 4);
14057 if (end - ((mDNSu8 *)t + tcphlen) > 0) data = mDNStrue;
14058 wake = ((int)(pack - seq) > 0) && ((int)(pseq - ack) >= 0) && data;
14059
14060 // If we got a regular keepalive on a connection that was registed with the KeepAlive API, respond with an ACK
14061 if ((t->tcp.flags & TH_ACK) && (data == mDNSfalse) &&
14062 ((int)(ack - pseq) == 1))
14063 {
14064 // Send an ACK;
14065 mDNS_SendKeepaliveACK(m, kr);
14066 }
14067 LogSPS("mDNSCoreReceiveRawTransportPacket: End %p, hlen %d, Datalen %d, pack %u, seq %u, pseq %u, ack %u, wake %d",
14068 end, tcphlen, end - ((mDNSu8 *)t + tcphlen), pack, seq, pseq, ack, wake);
14069 }
14070 else { LogSPS("mDNSCoreReceiveRawTransportPacket: waking because of RST or FIN th_flags %d", t->tcp.flags); }
14071 kaWake = wake;
14072 }
14073 else
14074 {
14075 // Plan to wake if
14076 // (a) RST is not set, AND
14077 // (b) packet is SYN, SYN+FIN, or plain data packet (no SYN or FIN). We won't wake for FIN alone.
14078 wake = (!(t->tcp.flags & TH_RST) && (t->tcp.flags & (TH_FIN|TH_SYN)) != TH_FIN);
14079
14080 // For now, to reduce spurious wakeups, we wake only for TCP SYN,
14081 // except for ssh connections, where we'll wake for plain data packets too
14082 if (!mDNSSameIPPort(port, SSHPort) && !(t->tcp.flags & 2)) wake = mDNSfalse;
14083
14084 LogSPS("%s %d-byte TCP from %#a:%d to %#a:%d%s%s%s", XX,
14085 src, mDNSVal16(t->tcp.src), dst, mDNSVal16(port),
14086 (t->tcp.flags & 2) ? " SYN" : "",
14087 (t->tcp.flags & 1) ? " FIN" : "",
14088 (t->tcp.flags & 4) ? " RST" : "");
14089 }
14090 break;
14091 }
14092
14093 case 0x11: {
14094 #define ARD_AsNumber 3283
14095 static const mDNSIPPort ARD = { { ARD_AsNumber >> 8, ARD_AsNumber & 0xFF } };
14096 const mDNSu16 udplen = (mDNSu16)((mDNSu16)t->bytes[4] << 8 | t->bytes[5]); // Length *including* 8-byte UDP header
14097 if (udplen >= sizeof(UDPHeader))
14098 {
14099 const mDNSu16 datalen = udplen - sizeof(UDPHeader);
14100 wake = mDNStrue;
14101
14102 // For Back to My Mac UDP port 4500 (IPSEC) packets, we do some special handling
14103 if (mDNSSameIPPort(port, IPSECPort))
14104 {
14105 // Specifically ignore NAT keepalive packets
14106 if (datalen == 1 && end >= &t->bytes[9] && t->bytes[8] == 0xFF) wake = mDNSfalse;
14107 else
14108 {
14109 // Skip over the Non-ESP Marker if present
14110 const mDNSBool NonESP = (end >= &t->bytes[12] && t->bytes[8] == 0 && t->bytes[9] == 0 && t->bytes[10] == 0 && t->bytes[11] == 0);
14111 const IKEHeader *const ike = (IKEHeader *)(t + (NonESP ? 12 : 8));
14112 const mDNSu16 ikelen = datalen - (NonESP ? 4 : 0);
14113 if (ikelen >= sizeof(IKEHeader) && end >= ((mDNSu8 *)ike) + sizeof(IKEHeader))
14114 if ((ike->Version & 0x10) == 0x10)
14115 {
14116 // ExchangeType == 5 means 'Informational' <http://www.ietf.org/rfc/rfc2408.txt>
14117 // ExchangeType == 34 means 'IKE_SA_INIT' <http://www.iana.org/assignments/ikev2-parameters>
14118 if (ike->ExchangeType == 5 || ike->ExchangeType == 34) wake = mDNSfalse;
14119 LogSPS("%s %d-byte IKE ExchangeType %d", XX, ike->ExchangeType);
14120 }
14121 }
14122 }
14123
14124 // For now, because we haven't yet worked out a clean elegant way to do this, we just special-case the
14125 // Apple Remote Desktop port number -- we ignore all packets to UDP 3283 (the "Net Assistant" port),
14126 // except for Apple Remote Desktop's explicit manual wakeup packet, which looks like this:
14127 // UDP header (8 bytes)
14128 // Payload: 13 88 00 6a 41 4e 41 20 (8 bytes) ffffffffffff (6 bytes) 16xMAC (96 bytes) = 110 bytes total
14129 if (mDNSSameIPPort(port, ARD)) wake = (datalen >= 110 && end >= &t->bytes[10] && t->bytes[8] == 0x13 && t->bytes[9] == 0x88);
14130
14131 LogSPS("%s %d-byte UDP from %#a:%d to %#a:%d", XX, src, mDNSVal16(t->udp.src), dst, mDNSVal16(port));
14132 }
14133 }
14134 break;
14135
14136 case 0x3A: if (&t->bytes[len] <= end)
14137 {
14138 mDNSu16 checksum = IPv6CheckSum(&src->ip.v6, &dst->ip.v6, protocol, t->bytes, len);
14139 if (!checksum) mDNSCoreReceiveRawND(m, sha, &src->ip.v6, &t->ndp, &t->bytes[len], InterfaceID);
14140 else LogInfo("IPv6CheckSum bad %04X %02X%02X from %#a to %#a", checksum, t->bytes[2], t->bytes[3], src, dst);
14141 }
14142 break;
14143
14144 default: LogSPS("Ignoring %d-byte IP packet unknown protocol %d from %#a to %#a", end-p, protocol, src, dst);
14145 break;
14146 }
14147
14148 if (wake)
14149 {
14150 AuthRecord *rr, *r2;
14151
14152 mDNS_Lock(m);
14153 for (rr = m->ResourceRecords; rr; rr=rr->next)
14154 if (rr->resrec.InterfaceID == InterfaceID &&
14155 rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
14156 rr->AddressProxy.type && mDNSSameAddress(&rr->AddressProxy, dst))
14157 {
14158 const mDNSu8 *const tp = (protocol == 6) ? (const mDNSu8 *)"\x4_tcp" : (const mDNSu8 *)"\x4_udp";
14159 for (r2 = m->ResourceRecords; r2; r2=r2->next)
14160 if (r2->resrec.InterfaceID == InterfaceID && mDNSSameEthAddress(&r2->WakeUp.HMAC, &rr->WakeUp.HMAC) &&
14161 r2->resrec.RecordType != kDNSRecordTypeDeregistering &&
14162 r2->resrec.rrtype == kDNSType_SRV && mDNSSameIPPort(r2->resrec.rdata->u.srv.port, port) &&
14163 SameDomainLabel(ThirdLabel(r2->resrec.name)->c, tp))
14164 break;
14165 if (!r2 && mDNSSameIPPort(port, IPSECPort)) r2 = rr; // So that we wake for BTMM IPSEC packets, even without a matching SRV record
14166 if (!r2 && kaWake) r2 = rr; // So that we wake for keepalive packets, even without a matching SRV record
14167 if (r2)
14168 {
14169 LogMsg("Waking host at %s %#a H-MAC %.6a I-MAC %.6a for %s",
14170 InterfaceNameForID(m, rr->resrec.InterfaceID), dst, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, r2));
14171 ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
14172 }
14173 else
14174 LogSPS("Sleeping host at %s %#a %.6a has no service on %#s %d",
14175 InterfaceNameForID(m, rr->resrec.InterfaceID), dst, &rr->WakeUp.HMAC, tp, mDNSVal16(port));
14176 }
14177 mDNS_Unlock(m);
14178 }
14179 }
14180
14181 mDNSexport void mDNSCoreReceiveRawPacket(mDNS *const m, const mDNSu8 *const p, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID)
14182 {
14183 static const mDNSOpaque16 Ethertype_ARP = { { 0x08, 0x06 } }; // Ethertype 0x0806 = ARP
14184 static const mDNSOpaque16 Ethertype_IPv4 = { { 0x08, 0x00 } }; // Ethertype 0x0800 = IPv4
14185 static const mDNSOpaque16 Ethertype_IPv6 = { { 0x86, 0xDD } }; // Ethertype 0x86DD = IPv6
14186 static const mDNSOpaque16 ARP_hrd_eth = { { 0x00, 0x01 } }; // Hardware address space (Ethernet = 1)
14187 static const mDNSOpaque16 ARP_pro_ip = { { 0x08, 0x00 } }; // Protocol address space (IP = 0x0800)
14188
14189 // Note: BPF guarantees that the NETWORK LAYER header will be word aligned, not the link-layer header.
14190 // In other words, we can safely assume that pkt below (ARP, IPv4 or IPv6) is properly word aligned,
14191 // but if pkt is 4-byte aligned, that necessarily means that eth CANNOT also be 4-byte aligned
14192 // since it points to a an address 14 bytes before pkt.
14193 const EthernetHeader *const eth = (const EthernetHeader *)p;
14194 const NetworkLayerPacket *const pkt = (const NetworkLayerPacket *)(eth+1);
14195 mDNSAddr src, dst;
14196 #define RequiredCapLen(P) ((P)==0x01 ? 4 : (P)==0x06 ? 20 : (P)==0x11 ? 8 : (P)==0x3A ? 24 : 0)
14197
14198 // Is ARP? Length must be at least 14 + 28 = 42 bytes
14199 if (end >= p+42 && mDNSSameOpaque16(eth->ethertype, Ethertype_ARP) && mDNSSameOpaque16(pkt->arp.hrd, ARP_hrd_eth) && mDNSSameOpaque16(pkt->arp.pro, ARP_pro_ip))
14200 mDNSCoreReceiveRawARP(m, &pkt->arp, InterfaceID);
14201 // Is IPv4 with zero fragmentation offset? Length must be at least 14 + 20 = 34 bytes
14202 else if (end >= p+34 && mDNSSameOpaque16(eth->ethertype, Ethertype_IPv4) && (pkt->v4.flagsfrags.b[0] & 0x1F) == 0 && pkt->v4.flagsfrags.b[1] == 0)
14203 {
14204 const mDNSu8 *const trans = p + 14 + (pkt->v4.vlen & 0xF) * 4;
14205 const mDNSu8 * transEnd = p + 14 + mDNSVal16(pkt->v4.totlen);
14206 if (transEnd > end) transEnd = end;
14207 debugf("Got IPv4 %02X from %.4a to %.4a", pkt->v4.protocol, &pkt->v4.src, &pkt->v4.dst);
14208 src.type = mDNSAddrType_IPv4; src.ip.v4 = pkt->v4.src;
14209 dst.type = mDNSAddrType_IPv4; dst.ip.v4 = pkt->v4.dst;
14210 if (transEnd >= trans + RequiredCapLen(pkt->v4.protocol))
14211 mDNSCoreReceiveRawTransportPacket(m, &eth->src, &src, &dst, pkt->v4.protocol, p, (TransportLayerPacket*)trans, transEnd, InterfaceID, 0);
14212 }
14213 // Is IPv6? Length must be at least 14 + 28 = 42 bytes
14214 else if (end >= p+54 && mDNSSameOpaque16(eth->ethertype, Ethertype_IPv6))
14215 {
14216 const mDNSu8 *const trans = p + 54;
14217 debugf("Got IPv6 %02X from %.16a to %.16a", pkt->v6.pro, &pkt->v6.src, &pkt->v6.dst);
14218 src.type = mDNSAddrType_IPv6; src.ip.v6 = pkt->v6.src;
14219 dst.type = mDNSAddrType_IPv6; dst.ip.v6 = pkt->v6.dst;
14220 if (end >= trans + RequiredCapLen(pkt->v6.pro))
14221 mDNSCoreReceiveRawTransportPacket(m, &eth->src, &src, &dst, pkt->v6.pro, p, (TransportLayerPacket*)trans, end, InterfaceID,
14222 (mDNSu16)pkt->bytes[4] << 8 | pkt->bytes[5]);
14223 }
14224 }
14225
14226 mDNSlocal void ConstructSleepProxyServerName(mDNS *const m, domainlabel *name)
14227 {
14228 name->c[0] = (mDNSu8)mDNS_snprintf((char*)name->c+1, 62, "%d-%d-%d-%d.%d %#s",
14229 m->SPSType, m->SPSPortability, m->SPSMarginalPower, m->SPSTotalPower, m->SPSFeatureFlags, &m->nicelabel);
14230 }
14231
14232 #ifndef SPC_DISABLED
14233 mDNSlocal void SleepProxyServerCallback(mDNS *const m, ServiceRecordSet *const srs, mStatus result)
14234 {
14235 if (result == mStatus_NameConflict)
14236 mDNS_RenameAndReregisterService(m, srs, mDNSNULL);
14237 else if (result == mStatus_MemFree)
14238 {
14239 if (m->SleepState)
14240 m->SPSState = 3;
14241 else
14242 {
14243 m->SPSState = (mDNSu8)(m->SPSSocket != mDNSNULL);
14244 if (m->SPSState)
14245 {
14246 domainlabel name;
14247 ConstructSleepProxyServerName(m, &name);
14248 mDNS_RegisterService(m, srs,
14249 &name, &SleepProxyServiceType, &localdomain,
14250 mDNSNULL, m->SPSSocket->port, // Host, port
14251 (mDNSu8 *)"", 1, // TXT data, length
14252 mDNSNULL, 0, // Subtypes (none)
14253 mDNSInterface_Any, // Interface ID
14254 SleepProxyServerCallback, mDNSNULL, 0); // Callback, context, flags
14255 }
14256 LogSPS("Sleep Proxy Server %#s %s", srs->RR_SRV.resrec.name->c, m->SPSState ? "started" : "stopped");
14257 }
14258 }
14259 }
14260 #endif
14261
14262 // Called with lock held
14263 mDNSexport void mDNSCoreBeSleepProxyServer_internal(mDNS *const m, mDNSu8 sps, mDNSu8 port, mDNSu8 marginalpower, mDNSu8 totpower, mDNSu8 features)
14264 {
14265 // This routine uses mDNS_DeregisterService and calls SleepProxyServerCallback, so we execute in user callback context
14266 mDNS_DropLockBeforeCallback();
14267
14268 // If turning off SPS, close our socket
14269 // (Do this first, BEFORE calling mDNS_DeregisterService below)
14270 if (!sps && m->SPSSocket) { mDNSPlatformUDPClose(m->SPSSocket); m->SPSSocket = mDNSNULL; }
14271
14272 // If turning off, or changing type, deregister old name
14273 #ifndef SPC_DISABLED
14274 if (m->SPSState == 1 && sps != m->SPSType)
14275 { m->SPSState = 2; mDNS_DeregisterService_drt(m, &m->SPSRecords, sps ? mDNS_Dereg_rapid : mDNS_Dereg_normal); }
14276 #endif // SPC_DISABLED
14277
14278 // Record our new SPS parameters
14279 m->SPSType = sps;
14280 m->SPSPortability = port;
14281 m->SPSMarginalPower = marginalpower;
14282 m->SPSTotalPower = totpower;
14283 m->SPSFeatureFlags = features;
14284 // If turning on, open socket and advertise service
14285 if (sps)
14286 {
14287 if (!m->SPSSocket)
14288 {
14289 m->SPSSocket = mDNSPlatformUDPSocket(m, zeroIPPort);
14290 if (!m->SPSSocket) { LogMsg("mDNSCoreBeSleepProxyServer: Failed to allocate SPSSocket"); goto fail; }
14291 }
14292 #ifndef SPC_DISABLED
14293 if (m->SPSState == 0) SleepProxyServerCallback(m, &m->SPSRecords, mStatus_MemFree);
14294 #endif // SPC_DISABLED
14295 }
14296 else if (m->SPSState)
14297 {
14298 LogSPS("mDNSCoreBeSleepProxyServer turning off from state %d; will wake clients", m->SPSState);
14299 m->NextScheduledSPS = m->timenow;
14300 }
14301 fail:
14302 mDNS_ReclaimLockAfterCallback();
14303 }
14304
14305 // ***************************************************************************
14306 #if COMPILER_LIKES_PRAGMA_MARK
14307 #pragma mark -
14308 #pragma mark - Startup and Shutdown
14309 #endif
14310
14311 mDNSlocal void mDNS_GrowCache_internal(mDNS *const m, CacheEntity *storage, mDNSu32 numrecords)
14312 {
14313 if (storage && numrecords)
14314 {
14315 mDNSu32 i;
14316 debugf("Adding cache storage for %d more records (%d bytes)", numrecords, numrecords*sizeof(CacheEntity));
14317 for (i=0; i<numrecords; i++) storage[i].next = &storage[i+1];
14318 storage[numrecords-1].next = m->rrcache_free;
14319 m->rrcache_free = storage;
14320 m->rrcache_size += numrecords;
14321 }
14322 }
14323
14324 mDNSexport void mDNS_GrowCache(mDNS *const m, CacheEntity *storage, mDNSu32 numrecords)
14325 {
14326 mDNS_Lock(m);
14327 mDNS_GrowCache_internal(m, storage, numrecords);
14328 mDNS_Unlock(m);
14329 }
14330
14331 mDNSexport mStatus mDNS_Init(mDNS *const m, mDNS_PlatformSupport *const p,
14332 CacheEntity *rrcachestorage, mDNSu32 rrcachesize,
14333 mDNSBool AdvertiseLocalAddresses, mDNSCallback *Callback, void *Context)
14334 {
14335 mDNSu32 slot;
14336 mDNSs32 timenow;
14337 mStatus result;
14338
14339 if (!rrcachestorage) rrcachesize = 0;
14340
14341 m->p = p;
14342 m->NetworkChanged = 0;
14343 m->CanReceiveUnicastOn5353 = mDNSfalse; // Assume we can't receive unicasts on 5353, unless platform layer tells us otherwise
14344 m->AdvertiseLocalAddresses = AdvertiseLocalAddresses;
14345 m->DivertMulticastAdvertisements = mDNSfalse;
14346 m->mDNSPlatformStatus = mStatus_Waiting;
14347 m->UnicastPort4 = zeroIPPort;
14348 m->UnicastPort6 = zeroIPPort;
14349 m->PrimaryMAC = zeroEthAddr;
14350 m->MainCallback = Callback;
14351 m->MainContext = Context;
14352 m->rec.r.resrec.RecordType = 0;
14353 m->rec.r.resrec.AnonInfo = mDNSNULL;
14354
14355 // For debugging: To catch and report locking failures
14356 m->mDNS_busy = 0;
14357 m->mDNS_reentrancy = 0;
14358 m->ShutdownTime = 0;
14359 m->lock_rrcache = 0;
14360 m->lock_Questions = 0;
14361 m->lock_Records = 0;
14362
14363 // Task Scheduling variables
14364 result = mDNSPlatformTimeInit();
14365 if (result != mStatus_NoError) return(result);
14366 m->timenow_adjust = (mDNSs32)mDNSRandom(0xFFFFFFFF);
14367 timenow = mDNS_TimeNow_NoLock(m);
14368
14369 m->timenow = 0; // MUST only be set within mDNS_Lock/mDNS_Unlock section
14370 m->timenow_last = timenow;
14371 m->NextScheduledEvent = timenow;
14372 m->SuppressSending = timenow;
14373 m->NextCacheCheck = timenow + 0x78000000;
14374 m->NextScheduledQuery = timenow + 0x78000000;
14375 m->NextScheduledProbe = timenow + 0x78000000;
14376 m->NextScheduledResponse = timenow + 0x78000000;
14377 m->NextScheduledNATOp = timenow + 0x78000000;
14378 m->NextScheduledSPS = timenow + 0x78000000;
14379 m->NextScheduledKA = timenow + 0x78000000;
14380 m->NextScheduledStopTime = timenow + 0x78000000;
14381 m->NextBLEServiceTime = 0; // zero indicates inactive
14382
14383 #if BONJOUR_ON_DEMAND
14384 m->NextBonjourDisableTime = 0; // Timer active when non zero.
14385 m->BonjourEnabled = 0; // Set when Bonjour on Demand is enabled and Bonjour is currently enabled.
14386 #endif // BONJOUR_ON_DEMAND
14387
14388 m->DelayConflictProcessing = MAX_CONFLICT_PROCESSING_DELAYS;
14389 m->RandomQueryDelay = 0;
14390 m->RandomReconfirmDelay = 0;
14391 m->PktNum = 0;
14392 m->MPktNum = 0;
14393 m->LocalRemoveEvents = mDNSfalse;
14394 m->SleepState = SleepState_Awake;
14395 m->SleepSeqNum = 0;
14396 m->SystemWakeOnLANEnabled = mDNSfalse;
14397 m->AnnounceOwner = NonZeroTime(timenow + 60 * mDNSPlatformOneSecond);
14398 m->DelaySleep = 0;
14399 m->SleepLimit = 0;
14400
14401 #if APPLE_OSX_mDNSResponder
14402 m->StatStartTime = mDNSPlatformUTC();
14403 m->NextStatLogTime = m->StatStartTime + kDefaultNextStatsticsLogTime;
14404 m->ActiveStatTime = 0;
14405 m->UnicastPacketsSent = 0;
14406 m->MulticastPacketsSent = 0;
14407 m->RemoteSubnet = 0;
14408 #endif // APPLE_OSX_mDNSResponder
14409
14410 // These fields only required for mDNS Searcher...
14411 m->Questions = mDNSNULL;
14412 m->NewQuestions = mDNSNULL;
14413 m->CurrentQuestion = mDNSNULL;
14414 m->LocalOnlyQuestions = mDNSNULL;
14415 m->NewLocalOnlyQuestions = mDNSNULL;
14416 m->RestartQuestion = mDNSNULL;
14417 m->ValidationQuestion = mDNSNULL;
14418 m->rrcache_size = 0;
14419 m->rrcache_totalused = 0;
14420 m->rrcache_active = 0;
14421 m->rrcache_report = 10;
14422 m->rrcache_free = mDNSNULL;
14423
14424 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
14425 {
14426 m->rrcache_hash[slot] = mDNSNULL;
14427 m->rrcache_nextcheck[slot] = timenow + 0x78000000;;
14428 }
14429
14430 mDNS_GrowCache_internal(m, rrcachestorage, rrcachesize);
14431 m->rrauth.rrauth_free = mDNSNULL;
14432
14433 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
14434 m->rrauth.rrauth_hash[slot] = mDNSNULL;
14435
14436 // Fields below only required for mDNS Responder...
14437 m->hostlabel.c[0] = 0;
14438 m->nicelabel.c[0] = 0;
14439 m->MulticastHostname.c[0] = 0;
14440 m->HIHardware.c[0] = 0;
14441 m->HISoftware.c[0] = 0;
14442 m->ResourceRecords = mDNSNULL;
14443 m->DuplicateRecords = mDNSNULL;
14444 m->NewLocalRecords = mDNSNULL;
14445 m->NewLocalOnlyRecords = mDNSfalse;
14446 m->CurrentRecord = mDNSNULL;
14447 m->HostInterfaces = mDNSNULL;
14448 m->ProbeFailTime = 0;
14449 m->NumFailedProbes = 0;
14450 m->SuppressProbes = 0;
14451
14452 #ifndef UNICAST_DISABLED
14453 m->NextuDNSEvent = timenow + 0x78000000;
14454 m->NextSRVUpdate = timenow + 0x78000000;
14455
14456 m->DNSServers = mDNSNULL;
14457
14458 m->Router = zeroAddr;
14459 m->AdvertisedV4 = zeroAddr;
14460 m->AdvertisedV6 = zeroAddr;
14461
14462 m->AuthInfoList = mDNSNULL;
14463
14464 m->ReverseMap.ThisQInterval = -1;
14465 m->StaticHostname.c[0] = 0;
14466 m->FQDN.c[0] = 0;
14467 m->Hostnames = mDNSNULL;
14468 m->AutoTunnelNAT.clientContext = mDNSNULL;
14469
14470 m->WABBrowseQueriesCount = 0;
14471 m->WABLBrowseQueriesCount = 0;
14472 m->WABRegQueriesCount = 0;
14473 m->AutoTargetServices = 0;
14474
14475 #if BONJOUR_ON_DEMAND
14476 m->NumAllInterfaceRecords = 0;
14477 m->NumAllInterfaceQuestions = 0;
14478 #endif
14479 // NAT traversal fields
14480 m->LLQNAT.clientCallback = mDNSNULL;
14481 m->LLQNAT.clientContext = mDNSNULL;
14482 m->NATTraversals = mDNSNULL;
14483 m->CurrentNATTraversal = mDNSNULL;
14484 m->retryIntervalGetAddr = 0; // delta between time sent and retry
14485 m->retryGetAddr = timenow + 0x78000000; // absolute time when we retry
14486 m->ExtAddress = zerov4Addr;
14487 m->PCPNonce[0] = mDNSRandom(-1);
14488 m->PCPNonce[1] = mDNSRandom(-1);
14489 m->PCPNonce[2] = mDNSRandom(-1);
14490
14491 m->NATMcastRecvskt = mDNSNULL;
14492 m->LastNATupseconds = 0;
14493 m->LastNATReplyLocalTime = timenow;
14494 m->LastNATMapResultCode = NATErr_None;
14495
14496 m->UPnPInterfaceID = 0;
14497 m->SSDPSocket = mDNSNULL;
14498 m->SSDPWANPPPConnection = mDNSfalse;
14499 m->UPnPRouterPort = zeroIPPort;
14500 m->UPnPSOAPPort = zeroIPPort;
14501 m->UPnPRouterURL = mDNSNULL;
14502 m->UPnPWANPPPConnection = mDNSfalse;
14503 m->UPnPSOAPURL = mDNSNULL;
14504 m->UPnPRouterAddressString = mDNSNULL;
14505 m->UPnPSOAPAddressString = mDNSNULL;
14506 m->SPSType = 0;
14507 m->SPSPortability = 0;
14508 m->SPSMarginalPower = 0;
14509 m->SPSTotalPower = 0;
14510 m->SPSFeatureFlags = 0;
14511 m->SPSState = 0;
14512 m->SPSProxyListChanged = mDNSNULL;
14513 m->SPSSocket = mDNSNULL;
14514 m->SPSBrowseCallback = mDNSNULL;
14515 m->ProxyRecords = 0;
14516
14517 m->DNSPushServers = mDNSNULL;
14518 m->DNSPushZones = mDNSNULL;
14519 #endif
14520
14521 #if APPLE_OSX_mDNSResponder
14522 m->TunnelClients = mDNSNULL;
14523
14524 #if !NO_WCF
14525 CHECK_WCF_FUNCTION(WCFConnectionNew)
14526 {
14527 m->WCF = WCFConnectionNew();
14528 if (!m->WCF) { LogMsg("WCFConnectionNew failed"); return -1; }
14529 }
14530 #endif
14531
14532 #endif
14533
14534 result = mDNSPlatformInit(m);
14535
14536 #ifndef UNICAST_DISABLED
14537 // It's better to do this *after* the platform layer has set up the
14538 // interface list and security credentials
14539 uDNS_SetupDNSConfig(m); // Get initial DNS configuration
14540 #endif
14541
14542 return(result);
14543 }
14544
14545 mDNSexport void mDNS_ConfigChanged(mDNS *const m)
14546 {
14547 if (m->SPSState == 1)
14548 {
14549 domainlabel name, newname;
14550 #ifndef SPC_DISABLED
14551 domainname type, domain;
14552 DeconstructServiceName(m->SPSRecords.RR_SRV.resrec.name, &name, &type, &domain);
14553 #endif // SPC_DISABLED
14554 ConstructSleepProxyServerName(m, &newname);
14555 if (!SameDomainLabelCS(name.c, newname.c))
14556 {
14557 LogSPS("Renaming SPS from “%#s” to “%#s”", name.c, newname.c);
14558 // When SleepProxyServerCallback gets the mStatus_MemFree message,
14559 // it will reregister the service under the new name
14560 m->SPSState = 2;
14561 #ifndef SPC_DISABLED
14562 mDNS_DeregisterService_drt(m, &m->SPSRecords, mDNS_Dereg_rapid);
14563 #endif // SPC_DISABLED
14564 }
14565 }
14566
14567 if (m->MainCallback)
14568 m->MainCallback(m, mStatus_ConfigChanged);
14569 }
14570
14571 mDNSlocal void DynDNSHostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
14572 {
14573 (void)m; // unused
14574 debugf("NameStatusCallback: result %d for registration of name %##s", result, rr->resrec.name->c);
14575 mDNSPlatformDynDNSHostNameStatusChanged(rr->resrec.name, result);
14576 }
14577
14578 mDNSlocal void PurgeOrReconfirmCacheRecord(mDNS *const m, CacheRecord *cr, const DNSServer * const ptr, mDNSBool lameduck)
14579 {
14580 mDNSBool purge = cr->resrec.RecordType == kDNSRecordTypePacketNegative ||
14581 cr->resrec.rrtype == kDNSType_A ||
14582 cr->resrec.rrtype == kDNSType_AAAA ||
14583 cr->resrec.rrtype == kDNSType_SRV ||
14584 cr->resrec.rrtype == kDNSType_CNAME;
14585
14586 (void) lameduck;
14587 (void) ptr;
14588 debugf("PurgeOrReconfirmCacheRecord: %s cache record due to %s server %p %#a:%d (%##s): %s",
14589 purge ? "purging" : "reconfirming",
14590 lameduck ? "lame duck" : "new",
14591 ptr, &ptr->addr, mDNSVal16(ptr->port), ptr->domain.c, CRDisplayString(m, cr));
14592
14593 if (purge)
14594 {
14595 LogInfo("PurgeorReconfirmCacheRecord: Purging Resourcerecord %s, RecordType %x", CRDisplayString(m, cr), cr->resrec.RecordType);
14596 mDNS_PurgeCacheResourceRecord(m, cr);
14597 }
14598 else
14599 {
14600 LogInfo("PurgeorReconfirmCacheRecord: Reconfirming Resourcerecord %s, RecordType %x", CRDisplayString(m, cr), cr->resrec.RecordType);
14601 mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
14602 }
14603 }
14604
14605 mDNSlocal void mDNS_PurgeForQuestion(mDNS *const m, DNSQuestion *q)
14606 {
14607 const mDNSu32 slot = HashSlot(&q->qname);
14608 CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
14609 CacheRecord *rp;
14610 mDNSu8 validatingResponse = 0;
14611
14612 // For DNSSEC questions, purge the corresponding RRSIGs also.
14613 if (DNSSECQuestion(q))
14614 {
14615 validatingResponse = q->ValidatingResponse;
14616 q->ValidatingResponse = mDNStrue;
14617 }
14618 for (rp = cg ? cg->members : mDNSNULL; rp; rp = rp->next)
14619 {
14620 if (SameNameRecordAnswersQuestion(&rp->resrec, q))
14621 {
14622 LogInfo("mDNS_PurgeForQuestion: Flushing %s", CRDisplayString(m, rp));
14623 mDNS_PurgeCacheResourceRecord(m, rp);
14624 }
14625 }
14626 if (DNSSECQuestion(q))
14627 {
14628 q->ValidatingResponse = validatingResponse;
14629 }
14630 }
14631
14632 // For DNSSEC question, we need the DNSSEC records also. If the cache does not
14633 // have the DNSSEC records, we need to re-issue the question with EDNS0/DO bit set.
14634 // Just re-issuing the question for RRSIGs does not work in practice as the response
14635 // may not contain the RRSIGs whose typeCovered field matches the question's qtype.
14636 //
14637 // For negative responses, we need the NSECs to prove the non-existence. If we don't
14638 // have the cached NSECs, purge them. For positive responses, if we don't have the
14639 // RRSIGs and if we have not already issued the question with EDNS0/DO bit set, purge
14640 // them.
14641 mDNSlocal void CheckForDNSSECRecords(mDNS *const m, DNSQuestion *q)
14642 {
14643 const mDNSu32 slot = HashSlot(&q->qname);
14644 CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
14645 CacheRecord *rp;
14646
14647 for (rp = cg ? cg->members : mDNSNULL; rp; rp = rp->next)
14648 {
14649 if (SameNameRecordAnswersQuestion(&rp->resrec, q))
14650 {
14651 if (rp->resrec.RecordType != kDNSRecordTypePacketNegative || !rp->nsec)
14652 {
14653 if (!rp->CRDNSSECQuestion)
14654 {
14655 LogInfo("CheckForDNSSECRecords: Flushing %s", CRDisplayString(m, rp));
14656 mDNS_PurgeCacheResourceRecord(m, rp);
14657 }
14658 }
14659 }
14660 }
14661 }
14662
14663 // Check for a positive unicast response to the question but with qtype
14664 mDNSexport mDNSBool mDNS_CheckForCacheRecord(mDNS *const m, DNSQuestion *q, mDNSu16 qtype)
14665 {
14666 DNSQuestion question;
14667 const mDNSu32 slot = HashSlot(&q->qname);
14668 CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
14669 CacheRecord *rp;
14670
14671 // Create an identical question but with qtype
14672 mDNS_SetupQuestion(&question, q->InterfaceID, &q->qname, qtype, mDNSNULL, mDNSNULL);
14673 question.qDNSServer = q->qDNSServer;
14674
14675 for (rp = cg ? cg->members : mDNSNULL; rp; rp = rp->next)
14676 {
14677 if (!rp->resrec.InterfaceID && rp->resrec.RecordType != kDNSRecordTypePacketNegative &&
14678 SameNameRecordAnswersQuestion(&rp->resrec, &question))
14679 {
14680 LogInfo("mDNS_CheckForCacheRecord: Found %s", CRDisplayString(m, rp));
14681 return mDNStrue;
14682 }
14683 }
14684 return mDNSfalse;
14685 }
14686
14687 mDNSexport void DNSServerChangeForQuestion(mDNS *const m, DNSQuestion *q, DNSServer *new)
14688 {
14689 DNSQuestion *qptr;
14690
14691 (void) m;
14692
14693 if (q->DuplicateOf)
14694 LogMsg("DNSServerChangeForQuestion: ERROR: Called for duplicate question %##s", q->qname.c);
14695
14696 // Make sure all the duplicate questions point to the same DNSServer so that delivery
14697 // of events for all of them are consistent. Duplicates for a question are always inserted
14698 // after in the list.
14699 q->qDNSServer = new;
14700 for (qptr = q->next ; qptr; qptr = qptr->next)
14701 {
14702 if (qptr->DuplicateOf == q) { qptr->validDNSServers = q->validDNSServers; qptr->qDNSServer = new; }
14703 }
14704 }
14705
14706 mDNSlocal void SetConfigState(mDNS *const m, mDNSBool delete)
14707 {
14708 McastResolver *mr;
14709 DNSServer *ptr;
14710
14711 if (delete)
14712 {
14713 for (ptr = m->DNSServers; ptr; ptr = ptr->next)
14714 {
14715 ptr->penaltyTime = 0;
14716 NumUnicastDNSServers--;
14717 ptr->flags |= DNSServer_FlagDelete;
14718 #if APPLE_OSX_mDNSResponder
14719 if (ptr->flags & DNSServer_FlagUnreachable)
14720 NumUnreachableDNSServers--;
14721 #endif
14722 }
14723 // We handle the mcast resolvers here itself as mDNSPlatformSetDNSConfig looks at
14724 // mcast resolvers. Today we get both mcast and ucast configuration using the same
14725 // API
14726 for (mr = m->McastResolvers; mr; mr = mr->next)
14727 mr->flags |= McastResolver_FlagDelete;
14728 }
14729 else
14730 {
14731 for (ptr = m->DNSServers; ptr; ptr = ptr->next)
14732 {
14733 ptr->penaltyTime = 0;
14734 NumUnicastDNSServers++;
14735 ptr->flags &= ~DNSServer_FlagDelete;
14736 #if APPLE_OSX_mDNSResponder
14737 if (ptr->flags & DNSServer_FlagUnreachable)
14738 NumUnreachableDNSServers++;
14739 #endif
14740 }
14741 for (mr = m->McastResolvers; mr; mr = mr->next)
14742 mr->flags &= ~McastResolver_FlagDelete;
14743 }
14744 }
14745
14746 mDNSlocal void SetDynDNSHostNameIfChanged(mDNS *const m, domainname *const fqdn)
14747 {
14748 // Did our FQDN change?
14749 if (!SameDomainName(fqdn, &m->FQDN))
14750 {
14751 if (m->FQDN.c[0]) mDNS_RemoveDynDNSHostName(m, &m->FQDN);
14752
14753 AssignDomainName(&m->FQDN, fqdn);
14754
14755 if (m->FQDN.c[0])
14756 {
14757 mDNSPlatformDynDNSHostNameStatusChanged(&m->FQDN, 1);
14758 mDNS_AddDynDNSHostName(m, &m->FQDN, DynDNSHostNameCallback, mDNSNULL);
14759 }
14760 }
14761 }
14762
14763 mDNSexport mStatus uDNS_SetupDNSConfig(mDNS *const m)
14764 {
14765 mDNSu32 slot;
14766 CacheGroup *cg;
14767 CacheRecord *cr;
14768 mDNSBool Restart = mDNSfalse;
14769 mDNSAddr v4, v6, r;
14770 domainname fqdn;
14771 DNSServer *ptr, **p = &m->DNSServers;
14772 const DNSServer *oldServers = m->DNSServers;
14773 DNSQuestion *q;
14774 McastResolver *mr, **mres = &m->McastResolvers;
14775
14776 debugf("uDNS_SetupDNSConfig: entry");
14777
14778 // Let the platform layer get the current DNS information and setup the WAB queries if needed.
14779 uDNS_SetupWABQueries(m);
14780
14781 mDNS_Lock(m);
14782
14783 // We need to first mark all the entries to be deleted. If the configuration changed, then
14784 // the entries would be undeleted appropriately. Otherwise, we need to clear them.
14785 //
14786 // Note: The last argument to mDNSPlatformSetDNSConfig is "mDNStrue" which means ack the
14787 // configuration. We already processed search domains in uDNS_SetupWABQueries above and
14788 // hence we are ready to ack the configuration as this is the last call to mDNSPlatformSetConfig
14789 // for the dns configuration change notification.
14790 SetConfigState(m, mDNStrue);
14791 if (!mDNSPlatformSetDNSConfig(m, mDNStrue, mDNSfalse, &fqdn, mDNSNULL, mDNSNULL, mDNStrue))
14792 {
14793 SetDynDNSHostNameIfChanged(m, &fqdn);
14794 SetConfigState(m, mDNSfalse);
14795 mDNS_Unlock(m);
14796 LogInfo("uDNS_SetupDNSConfig: No configuration change");
14797 return mStatus_NoError;
14798 }
14799
14800 // For now, we just delete the mcast resolvers. We don't deal with cache or
14801 // questions here. Neither question nor cache point to mcast resolvers. Questions
14802 // do inherit the timeout values from mcast resolvers. But we don't bother
14803 // affecting them as they never change.
14804 while (*mres)
14805 {
14806 if (((*mres)->flags & McastResolver_FlagDelete) != 0)
14807 {
14808 mr = *mres;
14809 *mres = (*mres)->next;
14810 debugf("uDNS_SetupDNSConfig: Deleting mcast resolver %##s", mr, mr->domain.c);
14811 mDNSPlatformMemFree(mr);
14812 }
14813 else
14814 {
14815 (*mres)->flags &= ~McastResolver_FlagNew;
14816 mres = &(*mres)->next;
14817 }
14818 }
14819
14820 // Update our qDNSServer pointers before we go and free the DNSServer object memory
14821 //
14822 // All non-scoped resolvers share the same resGroupID. At no point in time a cache entry using DNSServer
14823 // from scoped resolver will be used to answer non-scoped questions and vice versa, as scoped and non-scoped
14824 // resolvers don't share the same resGroupID. A few examples to describe the interaction with how we pick
14825 // DNSServers and flush the cache.
14826 //
14827 // - A non-scoped question picks DNSServer X, creates a cache entry with X. If a new resolver gets added later that
14828 // is a better match, we pick the new DNSServer for the question and activate the unicast query. We may or may not
14829 // flush the cache (See PurgeOrReconfirmCacheRecord). In either case, we don't change the cache record's DNSServer
14830 // pointer immediately (qDNSServer and rDNSServer may be different but still share the same resGroupID). If we don't
14831 // flush the cache immediately, the record's rDNSServer pointer will be updated (in mDNSCoreReceiveResponse)
14832 // later when we get the response. If we purge the cache, we still deliver a RMV when it is purged even though
14833 // we don't update the cache record's DNSServer pointer to match the question's DNSSever, as they both point to
14834 // the same resGroupID.
14835 //
14836 // Note: If the new DNSServer comes back with a different response than what we have in the cache, we will deliver a RMV
14837 // of the old followed by ADD of the new records.
14838 //
14839 // - A non-scoped question picks DNSServer X, creates a cache entry with X. If the resolver gets removed later, we will
14840 // pick a new DNSServer for the question which may or may not be NULL and set the cache record's pointer to the same
14841 // as in question's qDNSServer if the cache record is not flushed. If there is no active question, it will be set to NULL.
14842 //
14843 // - Two questions scoped and non-scoped for the same name will pick two different DNSServer and will end up creating separate
14844 // cache records and as the resGroupID is different, you can't use the cache record from the scoped DNSServer to answer the
14845 // non-scoped question and vice versa.
14846 //
14847 for (q = m->Questions; q; q=q->next)
14848 {
14849 if (!mDNSOpaque16IsZero(q->TargetQID))
14850 {
14851 DNSServer *s, *t;
14852 DNSQuestion *qptr;
14853 if (q->DuplicateOf) continue;
14854 SetValidDNSServers(m, q);
14855 q->triedAllServersOnce = 0;
14856 s = GetServerForQuestion(m, q);
14857 t = q->qDNSServer;
14858 if (t != s)
14859 {
14860 mDNSBool old, new;
14861 // If DNS Server for this question has changed, reactivate it
14862 LogInfo("uDNS_SetupDNSConfig: Updating DNS Server from %#a:%d (%##s) to %#a:%d (%##s) for question %##s (%s) (scope:%p)",
14863 t ? &t->addr : mDNSNULL, mDNSVal16(t ? t->port : zeroIPPort), t ? t->domain.c : (mDNSu8*)"",
14864 s ? &s->addr : mDNSNULL, mDNSVal16(s ? s->port : zeroIPPort), s ? s->domain.c : (mDNSu8*)"",
14865 q->qname.c, DNSTypeName(q->qtype), q->InterfaceID);
14866
14867 old = q->SuppressQuery;
14868 new = ShouldSuppressUnicastQuery(m, q, s);
14869 if (old != new)
14870 {
14871 // Changing the DNS server affected the SuppressQuery status. We need to
14872 // deliver RMVs for the previous ADDs (if any) before switching to the new
14873 // DNSServer. To keep it simple, we walk all the questions and mark them
14874 // to be restarted and then handle all of them at once.
14875 q->Restart = 1;
14876 q->SuppressQuery = new;
14877 for (qptr = q->next ; qptr; qptr = qptr->next)
14878 {
14879 if (qptr->DuplicateOf == q)
14880 qptr->Restart = 1;
14881 }
14882 Restart = mDNStrue;
14883 }
14884 else
14885 {
14886 DNSServerChangeForQuestion(m, q, s);
14887 q->unansweredQueries = 0;
14888
14889 // If we had sent a query out to DNSServer "t" and we are changing to "s", we
14890 // need to ignore the responses coming back from "t" as the DNS configuration
14891 // has changed e.g., when a new interface is coming up and that becomes the primary
14892 // interface, we switch to the DNS servers configured for the primary interface. In
14893 // this case, we should not accept responses associated with the previous interface as
14894 // the "name" could resolve differently on this new primary interface. Hence, discard
14895 // in-flight responses.
14896 q->TargetQID = mDNS_NewMessageID(m);
14897
14898 if (!QuerySuppressed(q))
14899 {
14900 debugf("uDNS_SetupDNSConfig: Activating query %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
14901 ActivateUnicastQuery(m, q, mDNStrue);
14902 // ActivateUnicastQuery is called for duplicate questions also as it does something
14903 // special for AutoTunnel questions
14904 for (qptr = q->next ; qptr; qptr = qptr->next)
14905 {
14906 if (qptr->DuplicateOf == q) ActivateUnicastQuery(m, qptr, mDNStrue);
14907 }
14908 }
14909 }
14910 }
14911 else
14912 {
14913 debugf("uDNS_SetupDNSConfig: Not Updating DNS server question %p %##s (%s) DNS server %#a:%d %p %d",
14914 q, q->qname.c, DNSTypeName(q->qtype), t ? &t->addr : mDNSNULL, mDNSVal16(t ? t->port : zeroIPPort), q->DuplicateOf, q->SuppressUnusable);
14915 for (qptr = q->next ; qptr; qptr = qptr->next)
14916 if (qptr->DuplicateOf == q) { qptr->validDNSServers = q->validDNSServers; qptr->qDNSServer = q->qDNSServer; }
14917 }
14918 }
14919 }
14920 if (Restart)
14921 RestartUnicastQuestions(m);
14922
14923 FORALL_CACHERECORDS(slot, cg, cr)
14924 {
14925 if (cr->resrec.InterfaceID)
14926 continue;
14927
14928 // We already walked the questions and restarted/reactivated them if the dns server
14929 // change affected the question. That should take care of updating the cache. But
14930 // what if there is no active question at this point when the DNS server change
14931 // happened ? There could be old cache entries lying around and if we don't flush
14932 // them, a new question after the DNS server change could pick up these stale
14933 // entries and get a wrong answer.
14934 //
14935 // For cache entries that have active questions we might have skipped rescheduling
14936 // the questions if they were suppressed (see above). To keep it simple, we walk
14937 // all the cache entries to make sure that there are no stale entries. We use the
14938 // active question's InterfaceID/ServiceID for looking up the right DNS server.
14939 // Note that the unscoped value for ServiceID is -1.
14940 //
14941 // Note: If GetServerForName returns NULL, it could either mean that there are no
14942 // DNS servers or no matching DNS servers for this question. In either case,
14943 // the cache should get purged below when we process deleted DNS servers.
14944
14945 ptr = GetServerForName(m, cr->resrec.name,
14946 (cr->CRActiveQuestion ? cr->CRActiveQuestion->InterfaceID : mDNSNULL),
14947 (cr->CRActiveQuestion ? cr->CRActiveQuestion->ServiceID : -1));
14948
14949 // Purge or Reconfirm if this cache entry would use the new DNS server
14950 if (ptr && (ptr != cr->resrec.rDNSServer))
14951 {
14952 // As the DNSServers for this cache record is not the same anymore, we don't
14953 // want any new questions to pick this old value. If there is no active question,
14954 // we can't possibly re-confirm, so purge in that case. If it is a DNSSEC question,
14955 // purge the cache as the DNSSEC capabilities of the DNS server may have changed.
14956
14957 if (cr->CRActiveQuestion == mDNSNULL || DNSSECQuestion(cr->CRActiveQuestion))
14958 {
14959 LogInfo("uDNS_SetupDNSConfig: Purging Resourcerecord %s, New DNS server %#a , Old DNS server %#a", CRDisplayString(m, cr),
14960 &ptr->addr, (cr->resrec.rDNSServer != mDNSNULL ? &cr->resrec.rDNSServer->addr : mDNSNULL));
14961 mDNS_PurgeCacheResourceRecord(m, cr);
14962 }
14963 else
14964 {
14965 LogInfo("uDNS_SetupDNSConfig: Purging/Reconfirming Resourcerecord %s, New DNS server %#a, Old DNS server %#a", CRDisplayString(m, cr),
14966 &ptr->addr, (cr->resrec.rDNSServer != mDNSNULL ? &cr->resrec.rDNSServer->addr : mDNSNULL));
14967 PurgeOrReconfirmCacheRecord(m, cr, ptr, mDNSfalse);
14968 }
14969 }
14970
14971 // If a cache record's DNSServer pointer is NULL, but its active question got a DNSServer in this DNS configuration
14972 // update, then use its DNSServer. This way, the active question and its duplicates don't miss out on RMV events.
14973 if (!cr->resrec.rDNSServer && cr->CRActiveQuestion && cr->CRActiveQuestion->qDNSServer)
14974 {
14975 cr->resrec.rDNSServer = cr->CRActiveQuestion->qDNSServer;
14976 LogInfo("uDNS_SetupDNSConfig: Using active question's DNS server %#a for cache record %s", &cr->resrec.rDNSServer->addr, CRDisplayString(m, cr));
14977 }
14978 }
14979
14980 while (*p)
14981 {
14982 if (((*p)->flags & DNSServer_FlagDelete) != 0)
14983 {
14984 // Scan our cache, looking for uDNS records that we would have queried this server for.
14985 // We reconfirm any records that match, because in this world of split DNS, firewalls, etc.
14986 // different DNS servers can give different answers to the same question.
14987 ptr = *p;
14988 FORALL_CACHERECORDS(slot, cg, cr)
14989 {
14990 if (cr->resrec.InterfaceID) continue;
14991 if (cr->resrec.rDNSServer == ptr)
14992 {
14993 // If we don't have an active question for this cache record, neither Purge can
14994 // generate RMV events nor Reconfirm can send queries out. Just set the DNSServer
14995 // pointer on the record NULL so that we don't point to freed memory (We might dereference
14996 // DNSServer pointers from resource record for logging purposes).
14997 //
14998 // If there is an active question, point to its DNSServer as long as it does not point to the
14999 // freed one. We already went through the questions above and made them point at either the
15000 // new server or NULL if there is no server.
15001
15002 if (cr->CRActiveQuestion)
15003 {
15004 DNSQuestion *qptr = cr->CRActiveQuestion;
15005
15006 if (qptr->qDNSServer == ptr)
15007 {
15008 LogMsg("uDNS_SetupDNSConfig: ERROR!! Cache Record %s Active question %##s (%s) (scope:%p) pointing to DNSServer Address %#a"
15009 " to be freed", CRDisplayString(m, cr), qptr->qname.c, DNSTypeName(qptr->qtype), qptr->InterfaceID, &ptr->addr);
15010 qptr->validDNSServers = zeroOpaque64;
15011 qptr->qDNSServer = mDNSNULL;
15012 cr->resrec.rDNSServer = mDNSNULL;
15013 }
15014 else
15015 {
15016 LogInfo("uDNS_SetupDNSConfig: Cache Record %s, Active question %##s (%s) (scope:%p), pointing to DNSServer %#a (to be deleted),"
15017 " resetting to question's DNSServer Address %#a", CRDisplayString(m, cr), qptr->qname.c, DNSTypeName(qptr->qtype),
15018 qptr->InterfaceID, &ptr->addr, (qptr->qDNSServer) ? &qptr->qDNSServer->addr : mDNSNULL);
15019 cr->resrec.rDNSServer = qptr->qDNSServer;
15020 }
15021 }
15022 else
15023 {
15024 LogInfo("uDNS_SetupDNSConfig: Cache Record %##s has no Active question, Record's DNSServer Address %#a, Server to be deleted %#a",
15025 cr->resrec.name, &cr->resrec.rDNSServer->addr, &ptr->addr);
15026 cr->resrec.rDNSServer = mDNSNULL;
15027 }
15028
15029 PurgeOrReconfirmCacheRecord(m, cr, ptr, mDNStrue);
15030 }
15031 }
15032 *p = (*p)->next;
15033 LogInfo("uDNS_SetupDNSConfig: Deleting server %p %#a:%d (%##s) %d", ptr, &ptr->addr, mDNSVal16(ptr->port), ptr->domain.c, NumUnicastDNSServers);
15034 mDNSPlatformMemFree(ptr);
15035 }
15036 else
15037 {
15038 (*p)->flags &= ~DNSServer_FlagNew;
15039 p = &(*p)->next;
15040 }
15041 }
15042
15043 // If we now have no DNS servers at all and we used to have some, then immediately purge all unicast cache records (including for LLQs).
15044 // This is important for giving prompt remove events when the user disconnects the Ethernet cable or turns off wireless.
15045 // Otherwise, stale data lingers for 5-10 seconds, which is not the user-experience people expect from Bonjour.
15046 // Similarly, if we now have some DNS servers and we used to have none, we want to purge any fake negative results we may have generated.
15047 if ((m->DNSServers != mDNSNULL) != (oldServers != mDNSNULL))
15048 {
15049 int count = 0;
15050 FORALL_CACHERECORDS(slot, cg, cr)
15051 {
15052 if (!cr->resrec.InterfaceID)
15053 {
15054 mDNS_PurgeCacheResourceRecord(m, cr);
15055 count++;
15056 }
15057 }
15058 LogInfo("uDNS_SetupDNSConfig: %s available; purged %d unicast DNS records from cache",
15059 m->DNSServers ? "DNS server became" : "No DNS servers", count);
15060
15061 // Force anything that needs to get zone data to get that information again
15062 RestartRecordGetZoneData(m);
15063 }
15064
15065 SetDynDNSHostNameIfChanged(m, &fqdn);
15066
15067 mDNS_Unlock(m);
15068
15069 // handle router and primary interface changes
15070 v4 = v6 = r = zeroAddr;
15071 v4.type = r.type = mDNSAddrType_IPv4;
15072
15073 if (mDNSPlatformGetPrimaryInterface(m, &v4, &v6, &r) == mStatus_NoError && !mDNSv4AddressIsLinkLocal(&v4.ip.v4))
15074 {
15075 mDNS_SetPrimaryInterfaceInfo(m,
15076 !mDNSIPv4AddressIsZero(v4.ip.v4) ? &v4 : mDNSNULL,
15077 !mDNSIPv6AddressIsZero(v6.ip.v6) ? &v6 : mDNSNULL,
15078 !mDNSIPv4AddressIsZero(r.ip.v4) ? &r : mDNSNULL);
15079 }
15080 else
15081 {
15082 mDNS_SetPrimaryInterfaceInfo(m, mDNSNULL, mDNSNULL, mDNSNULL);
15083 if (m->FQDN.c[0]) mDNSPlatformDynDNSHostNameStatusChanged(&m->FQDN, 1); // Set status to 1 to indicate temporary failure
15084 }
15085
15086 debugf("uDNS_SetupDNSConfig: number of unicast DNS servers %d", NumUnicastDNSServers);
15087 return mStatus_NoError;
15088 }
15089
15090 mDNSexport void mDNSCoreInitComplete(mDNS *const m, mStatus result)
15091 {
15092 m->mDNSPlatformStatus = result;
15093 if (m->MainCallback)
15094 {
15095 mDNS_Lock(m);
15096 mDNS_DropLockBeforeCallback(); // Allow client to legally make mDNS API calls from the callback
15097 m->MainCallback(m, mStatus_NoError);
15098 mDNS_ReclaimLockAfterCallback(); // Decrement mDNS_reentrancy to block mDNS API calls again
15099 mDNS_Unlock(m);
15100 }
15101 }
15102
15103 mDNSlocal void DeregLoop(mDNS *const m, AuthRecord *const start)
15104 {
15105 m->CurrentRecord = start;
15106 while (m->CurrentRecord)
15107 {
15108 AuthRecord *rr = m->CurrentRecord;
15109 LogInfo("DeregLoop: %s deregistration for %p %02X %s",
15110 (rr->resrec.RecordType != kDNSRecordTypeDeregistering) ? "Initiating " : "Accelerating",
15111 rr, rr->resrec.RecordType, ARDisplayString(m, rr));
15112 if (rr->resrec.RecordType != kDNSRecordTypeDeregistering)
15113 mDNS_Deregister_internal(m, rr, mDNS_Dereg_rapid);
15114 else if (rr->AnnounceCount > 1)
15115 {
15116 rr->AnnounceCount = 1;
15117 rr->LastAPTime = m->timenow - rr->ThisAPInterval;
15118 }
15119 // Mustn't advance m->CurrentRecord until *after* mDNS_Deregister_internal, because
15120 // new records could have been added to the end of the list as a result of that call.
15121 if (m->CurrentRecord == rr) // If m->CurrentRecord was not advanced for us, do it now
15122 m->CurrentRecord = rr->next;
15123 }
15124 }
15125
15126 mDNSexport void mDNS_StartExit(mDNS *const m)
15127 {
15128 AuthRecord *rr;
15129
15130 mDNS_Lock(m);
15131
15132 LogInfo("mDNS_StartExit");
15133 m->ShutdownTime = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 5);
15134
15135 mDNSCoreBeSleepProxyServer_internal(m, 0, 0, 0, 0, 0);
15136
15137 #if APPLE_OSX_mDNSResponder
15138 #if !NO_WCF
15139 CHECK_WCF_FUNCTION(WCFConnectionDealloc)
15140 {
15141 if (m->WCF) WCFConnectionDealloc((WCFConnection *)m->WCF);
15142 }
15143 #endif
15144 #endif
15145
15146 #ifndef UNICAST_DISABLED
15147 {
15148 SearchListElem *s;
15149 SuspendLLQs(m);
15150 // Don't need to do SleepRecordRegistrations() here
15151 // because we deregister all records and services later in this routine
15152 while (m->Hostnames) mDNS_RemoveDynDNSHostName(m, &m->Hostnames->fqdn);
15153
15154 // For each member of our SearchList, deregister any records it may have created, and cut them from the list.
15155 // Otherwise they'll be forcibly deregistered for us (without being cut them from the appropriate list)
15156 // and we may crash because the list still contains dangling pointers.
15157 for (s = SearchList; s; s = s->next)
15158 while (s->AuthRecs)
15159 {
15160 ARListElem *dereg = s->AuthRecs;
15161 s->AuthRecs = s->AuthRecs->next;
15162 mDNS_Deregister_internal(m, &dereg->ar, mDNS_Dereg_normal); // Memory will be freed in the FreeARElemCallback
15163 }
15164 }
15165 #endif
15166
15167 DeadvertiseAllInterfaceRecords(m);
15168
15169 // Shut down all our active NAT Traversals
15170 while (m->NATTraversals)
15171 {
15172 NATTraversalInfo *t = m->NATTraversals;
15173 mDNS_StopNATOperation_internal(m, t); // This will cut 't' from the list, thereby advancing m->NATTraversals in the process
15174
15175 // After stopping the NAT Traversal, we zero out the fields.
15176 // This has particularly important implications for our AutoTunnel records --
15177 // when we deregister our AutoTunnel records below, we don't want their mStatus_MemFree
15178 // handlers to just turn around and attempt to re-register those same records.
15179 // Clearing t->ExternalPort/t->RequestedPort will cause the mStatus_MemFree callback handlers
15180 // to not do this.
15181 t->ExternalAddress = zerov4Addr;
15182 t->NewAddress = zerov4Addr;
15183 t->ExternalPort = zeroIPPort;
15184 t->RequestedPort = zeroIPPort;
15185 t->Lifetime = 0;
15186 t->Result = mStatus_NoError;
15187 }
15188
15189 // Make sure there are nothing but deregistering records remaining in the list
15190 if (m->CurrentRecord)
15191 LogMsg("mDNS_StartExit: ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
15192
15193 // We're in the process of shutting down, so queries, etc. are no longer available.
15194 // Consequently, determining certain information, e.g. the uDNS update server's IP
15195 // address, will not be possible. The records on the main list are more likely to
15196 // already contain such information, so we deregister the duplicate records first.
15197 LogInfo("mDNS_StartExit: Deregistering duplicate resource records");
15198 DeregLoop(m, m->DuplicateRecords);
15199 LogInfo("mDNS_StartExit: Deregistering resource records");
15200 DeregLoop(m, m->ResourceRecords);
15201
15202 // If we scheduled a response to send goodbye packets, we set NextScheduledResponse to now. Normally when deregistering records,
15203 // we allow up to 100ms delay (to help improve record grouping) but when shutting down we don't want any such delay.
15204 if (m->NextScheduledResponse - m->timenow < mDNSPlatformOneSecond)
15205 {
15206 m->NextScheduledResponse = m->timenow;
15207 m->SuppressSending = 0;
15208 }
15209
15210 if (m->ResourceRecords) LogInfo("mDNS_StartExit: Sending final record deregistrations");
15211 else LogInfo("mDNS_StartExit: No deregistering records remain");
15212
15213 for (rr = m->DuplicateRecords; rr; rr = rr->next)
15214 LogMsg("mDNS_StartExit: Should not still have Duplicate Records remaining: %02X %s", rr->resrec.RecordType, ARDisplayString(m, rr));
15215
15216 // If any deregistering records remain, send their deregistration announcements before we exit
15217 if (m->mDNSPlatformStatus != mStatus_NoError) DiscardDeregistrations(m);
15218
15219 mDNS_Unlock(m);
15220
15221 LogInfo("mDNS_StartExit: done");
15222 }
15223
15224 mDNSexport void mDNS_FinalExit(mDNS *const m)
15225 {
15226 mDNSu32 rrcache_active = 0;
15227 mDNSu32 rrcache_totalused = m->rrcache_totalused;
15228 mDNSu32 slot;
15229 AuthRecord *rr;
15230
15231 LogInfo("mDNS_FinalExit: mDNSPlatformClose");
15232 mDNSPlatformClose(m);
15233
15234 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
15235 {
15236 while (m->rrcache_hash[slot])
15237 {
15238 CacheGroup *cg = m->rrcache_hash[slot];
15239 while (cg->members)
15240 {
15241 CacheRecord *cr = cg->members;
15242 cg->members = cg->members->next;
15243 if (cr->CRActiveQuestion) rrcache_active++;
15244 ReleaseCacheRecord(m, cr);
15245 }
15246 cg->rrcache_tail = &cg->members;
15247 ReleaseCacheGroup(m, &m->rrcache_hash[slot]);
15248 }
15249 }
15250 debugf("mDNS_FinalExit: RR Cache was using %ld records, %lu active", rrcache_totalused, rrcache_active);
15251 if (rrcache_active != m->rrcache_active)
15252 LogMsg("*** ERROR *** rrcache_totalused %lu; rrcache_active %lu != m->rrcache_active %lu", rrcache_totalused, rrcache_active, m->rrcache_active);
15253
15254 for (rr = m->ResourceRecords; rr; rr = rr->next)
15255 LogMsg("mDNS_FinalExit failed to send goodbye for: %p %02X %s", rr, rr->resrec.RecordType, ARDisplayString(m, rr));
15256
15257 LogInfo("mDNS_FinalExit: done");
15258 }