#include "DNSCommon.h" // Defines general DNS untility routines
#include "uDNS.h" // Defines entry points into unicast-specific routines
#if(defined(_MSC_VER))
#pragma warning(disable:4127)
#pragma warning(disable:4706)
#endif
#if APPLE_OSX_mDNSResponder
#include <WebFilterDNS/WebFilterDNS.h>
#if ! NO_WCF
WCFConnection *WCFConnectionNew(void) __attribute__((weak_import));
void WCFConnectionDealloc(WCFConnection* c) __attribute__((weak_import));
#define CHECK_WCF_FUNCTION(X) if (X)
#endif // ! NO_WCF
#else
#define NO_WCF 1
#endif // APPLE_OSX_mDNSResponder
mDNSlocal void BeginSleepProcessing(mDNS *const m);
mDNSlocal void RetrySPSRegistrations(mDNS *const m);
mDNSlocal void SendWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *EthAddr, mDNSOpaque48 *password);
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark - Program Constants
#endif
#define NO_HINFO 1
#define SmallRecordLimit 1024
#define kMaxUpdateCredits 10
#define kUpdateCreditRefreshInterval (mDNSPlatformOneSecond * 6)
mDNSexport const char *const mDNS_DomainTypeNames[] =
{
"b._dns-sd._udp.", "db._dns-sd._udp.", "lb._dns-sd._udp.", "r._dns-sd._udp.", "dr._dns-sd._udp." };
#ifdef UNICAST_DISABLED
#define uDNS_IsActiveQuery(q, u) mDNSfalse
#endif
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - General Utility Functions
#endif
mDNSexport void SetNextQueryTime(mDNS *const m, const DNSQuestion *const q)
{
if (m->mDNS_busy != m->mDNS_reentrancy+1)
LogMsg("SetNextQueryTime: Lock not held! mDNS_busy (%ld) mDNS_reentrancy (%ld)", m->mDNS_busy, m->mDNS_reentrancy);
#if ForceAlerts
if (m->mDNS_busy != m->mDNS_reentrancy+1) *(long*)0 = 0;
#endif
if (ActiveQuestion(q))
{
mDNSs32 *const timer = mDNSOpaque16IsZero(q->TargetQID) ? &m->NextScheduledQuery : &m->NextuDNSEvent;
if (*timer - NextQSendTime(q) > 0)
*timer = NextQSendTime(q);
}
}
mDNSexport CacheGroup *CacheGroupForName(const mDNS *const m, const mDNSu32 slot, const mDNSu32 namehash, const domainname *const name)
{
CacheGroup *cg;
for (cg = m->rrcache_hash[slot]; cg; cg=cg->next)
if (cg->namehash == namehash && SameDomainName(cg->name, name))
break;
return(cg);
}
mDNSlocal CacheGroup *CacheGroupForRecord(const mDNS *const m, const mDNSu32 slot, const ResourceRecord *const rr)
{
return(CacheGroupForName(m, slot, rr->namehash, rr->name));
}
mDNSexport mDNSBool mDNS_AddressIsLocalSubnet(mDNS *const m, const mDNSInterfaceID InterfaceID, const mDNSAddr *addr)
{
NetworkInterfaceInfo *intf;
if (addr->type == mDNSAddrType_IPv4)
{
if (mDNSv4AddressIsLinkLocal(&addr->ip.v4)) return(mDNStrue);
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->ip.type == addr->type && intf->InterfaceID == InterfaceID && intf->McastTxRx)
if (((intf->ip.ip.v4.NotAnInteger ^ addr->ip.v4.NotAnInteger) & intf->mask.ip.v4.NotAnInteger) == 0)
return(mDNStrue);
}
if (addr->type == mDNSAddrType_IPv6)
{
if (mDNSv6AddressIsLinkLocal(&addr->ip.v4)) return(mDNStrue);
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->ip.type == addr->type && intf->InterfaceID == InterfaceID && intf->McastTxRx)
if ((((intf->ip.ip.v6.l[0] ^ addr->ip.v6.l[0]) & intf->mask.ip.v6.l[0]) == 0) &&
(((intf->ip.ip.v6.l[1] ^ addr->ip.v6.l[1]) & intf->mask.ip.v6.l[1]) == 0) &&
(((intf->ip.ip.v6.l[2] ^ addr->ip.v6.l[2]) & intf->mask.ip.v6.l[2]) == 0) &&
(((intf->ip.ip.v6.l[3] ^ addr->ip.v6.l[3]) & intf->mask.ip.v6.l[3]) == 0))
return(mDNStrue);
}
return(mDNSfalse);
}
mDNSlocal NetworkInterfaceInfo *FirstInterfaceForID(mDNS *const m, const mDNSInterfaceID InterfaceID)
{
NetworkInterfaceInfo *intf = m->HostInterfaces;
while (intf && intf->InterfaceID != InterfaceID) intf = intf->next;
return(intf);
}
mDNSexport char *InterfaceNameForID(mDNS *const m, const mDNSInterfaceID InterfaceID)
{
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
return(intf ? intf->ifname : mDNSNULL);
}
mDNSlocal void AnswerLocalQuestionWithLocalAuthRecord(mDNS *const m, DNSQuestion *q, AuthRecord *rr, QC_result AddRecord)
{
if (!(rr->resrec.RecordType & kDNSRecordTypeActiveMask))
{
LogMsg("AnswerLocalQuestionWithLocalAuthRecord: *NOT* delivering %s event for local record type %X %s",
AddRecord ? "Add" : "Rmv", rr->resrec.RecordType, ARDisplayString(m, rr));
return;
}
if (AddRecord) rr->AnsweredLocalQ = mDNStrue;
mDNS_DropLockBeforeCallback(); if (q->QuestionCallback && !q->NoAnswer)
{
q->CurrentAnswers += AddRecord ? 1 : -1;
q->QuestionCallback(m, q, &rr->resrec, AddRecord);
}
mDNS_ReclaimLockAfterCallback(); }
mDNSlocal void AnswerAllLocalQuestionsWithLocalAuthRecord(mDNS *const m, AuthRecord *rr, QC_result AddRecord)
{
if (m->CurrentQuestion)
LogMsg("AnswerAllLocalQuestionsWithLocalAuthRecord ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->LocalOnlyQuestions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewLocalOnlyQuestions)
{
DNSQuestion *q = m->CurrentQuestion;
m->CurrentQuestion = q->next;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
AnswerLocalQuestionWithLocalAuthRecord(m, q, rr, AddRecord); }
if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly || rr->resrec.InterfaceID == mDNSInterface_P2P)
{
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
{
DNSQuestion *q = m->CurrentQuestion;
m->CurrentQuestion = q->next;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
AnswerLocalQuestionWithLocalAuthRecord(m, q, rr, AddRecord); }
}
m->CurrentQuestion = mDNSNULL;
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Resource Record Utility Functions
#endif
#define RRTypeIsAddressType(T) ((T) == kDNSType_A || (T) == kDNSType_AAAA)
#define ResourceRecordIsValidAnswer(RR) ( ((RR)-> resrec.RecordType & kDNSRecordTypeActiveMask) && \
((RR)->Additional1 == mDNSNULL || ((RR)->Additional1->resrec.RecordType & kDNSRecordTypeActiveMask)) && \
((RR)->Additional2 == mDNSNULL || ((RR)->Additional2->resrec.RecordType & kDNSRecordTypeActiveMask)) && \
((RR)->DependentOn == mDNSNULL || ((RR)->DependentOn->resrec.RecordType & kDNSRecordTypeActiveMask)) )
#define ResourceRecordIsValidInterfaceAnswer(RR, INTID) \
(ResourceRecordIsValidAnswer(RR) && \
((RR)->resrec.InterfaceID == mDNSInterface_Any || (RR)->resrec.InterfaceID == (INTID)))
#define DefaultProbeCountForTypeUnique ((mDNSu8)3)
#define DefaultProbeCountForRecordType(X) ((X) == kDNSRecordTypeUnique ? DefaultProbeCountForTypeUnique : (mDNSu8)0)
#define InitialAnnounceCount ((mDNSu8)8)
#define GoodbyeCount ((mDNSu8)3)
#define WakeupCount ((mDNSu8)18)
#define DefaultProbeIntervalForTypeUnique (mDNSPlatformOneSecond/4)
#define DefaultAnnounceIntervalForTypeShared (mDNSPlatformOneSecond/2)
#define DefaultAnnounceIntervalForTypeUnique (mDNSPlatformOneSecond/2)
#define DefaultAPIntervalForRecordType(X) ((X) & kDNSRecordTypeActiveSharedMask ? DefaultAnnounceIntervalForTypeShared : \
(X) & kDNSRecordTypeUnique ? DefaultProbeIntervalForTypeUnique : \
(X) & kDNSRecordTypeActiveUniqueMask ? DefaultAnnounceIntervalForTypeUnique : 0)
#define TimeToAnnounceThisRecord(RR,time) ((RR)->AnnounceCount && (time) - ((RR)->LastAPTime + (RR)->ThisAPInterval) >= 0)
#define TimeToSendThisRecord(RR,time) ((TimeToAnnounceThisRecord(RR,time) || (RR)->ImmedAnswer) && ResourceRecordIsValidAnswer(RR))
#define TicksTTL(RR) ((mDNSs32)(RR)->resrec.rroriginalttl * mDNSPlatformOneSecond)
#define RRExpireTime(RR) ((RR)->TimeRcvd + TicksTTL(RR))
#define MaxUnansweredQueries 4
#define SameResourceRecordSignature(A,B) (A)->resrec.rrtype == (B)->resrec.rrtype && SameResourceRecordNameClassInterface((A),(B))
mDNSlocal mDNSBool SameResourceRecordNameClassInterface(const AuthRecord *const r1, const AuthRecord *const r2)
{
if (!r1) { LogMsg("SameResourceRecordSignature ERROR: r1 is NULL"); return(mDNSfalse); }
if (!r2) { LogMsg("SameResourceRecordSignature ERROR: r2 is NULL"); return(mDNSfalse); }
if (r1->resrec.InterfaceID &&
r2->resrec.InterfaceID &&
r1->resrec.InterfaceID != r2->resrec.InterfaceID) return(mDNSfalse);
return(mDNSBool)(
r1->resrec.rrclass == r2->resrec.rrclass &&
r1->resrec.namehash == r2->resrec.namehash &&
SameDomainName(r1->resrec.name, r2->resrec.name));
}
mDNSlocal mDNSBool PacketRRMatchesSignature(const CacheRecord *const pktrr, const AuthRecord *const authrr)
{
if (!pktrr) { LogMsg("PacketRRMatchesSignature ERROR: pktrr is NULL"); return(mDNSfalse); }
if (!authrr) { LogMsg("PacketRRMatchesSignature ERROR: authrr is NULL"); return(mDNSfalse); }
if (pktrr->resrec.InterfaceID &&
authrr->resrec.InterfaceID &&
pktrr->resrec.InterfaceID != authrr->resrec.InterfaceID) return(mDNSfalse);
if (!(authrr->resrec.RecordType & kDNSRecordTypeUniqueMask) || authrr->WakeUp.HMAC.l[0])
if (pktrr->resrec.rrtype != authrr->resrec.rrtype) return(mDNSfalse);
return(mDNSBool)(
pktrr->resrec.rrclass == authrr->resrec.rrclass &&
pktrr->resrec.namehash == authrr->resrec.namehash &&
SameDomainName(pktrr->resrec.name, authrr->resrec.name));
}
mDNSlocal mDNSBool ShouldSuppressKnownAnswer(const CacheRecord *const ka, const AuthRecord *const rr)
{
if (!IdenticalResourceRecord(&ka->resrec, &rr->resrec)) return(mDNSfalse);
return(mDNSBool)(ka->resrec.rroriginalttl >= rr->resrec.rroriginalttl / 2);
}
mDNSlocal void SetNextAnnounceProbeTime(mDNS *const m, const AuthRecord *const rr)
{
if (rr->resrec.RecordType == kDNSRecordTypeUnique)
{
if ((rr->LastAPTime + rr->ThisAPInterval) - m->timenow > mDNSPlatformOneSecond * 10)
{
LogMsg("SetNextAnnounceProbeTime: ProbeCount %d Next in %d %s", rr->ProbeCount, (rr->LastAPTime + rr->ThisAPInterval) - m->timenow, ARDisplayString(m, rr));
LogMsg("SetNextAnnounceProbeTime: m->SuppressProbes %d m->timenow %d diff %d", m->SuppressProbes, m->timenow, m->SuppressProbes - m->timenow);
}
if (m->NextScheduledProbe - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
m->NextScheduledProbe = (rr->LastAPTime + rr->ThisAPInterval);
if (m->NextScheduledProbe - m->timenow < 0)
m->NextScheduledProbe = m->timenow;
}
else if (rr->AnnounceCount && (ResourceRecordIsValidAnswer(rr) || rr->resrec.RecordType == kDNSRecordTypeDeregistering))
{
if (m->NextScheduledResponse - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
m->NextScheduledResponse = (rr->LastAPTime + rr->ThisAPInterval);
}
}
mDNSlocal void InitializeLastAPTime(mDNS *const m, AuthRecord *const rr)
{
rr->ThisAPInterval = rr->AddressProxy.type ? mDNSPlatformOneSecond : DefaultAPIntervalForRecordType(rr->resrec.RecordType);
if (rr->ProbeCount)
{
if (m->SuppressProbes == 0 || m->SuppressProbes - m->timenow < 0)
{
m->SuppressProbes = NonZeroTime(m->timenow + DefaultProbeIntervalForTypeUnique/2 + mDNSRandom(DefaultProbeIntervalForTypeUnique/2));
if (m->SuppressProbes - m->NextScheduledProbe >= 0)
m->SuppressProbes = NonZeroTime(m->NextScheduledProbe);
if (m->SuppressProbes - m->timenow < 0) m->SuppressProbes = m->timenow;
if (m->SuppressProbes - m->NextScheduledQuery >= 0)
m->SuppressProbes = NonZeroTime(m->NextScheduledQuery);
if (m->SuppressProbes - m->timenow < 0) m->SuppressProbes = m->timenow;
if (m->SuppressSending && m->SuppressProbes - m->SuppressSending < 0)
m->SuppressProbes = NonZeroTime(m->SuppressSending);
if (m->SuppressProbes - m->timenow > mDNSPlatformOneSecond * 8)
{
LogMsg("InitializeLastAPTime ERROR m->SuppressProbes %d m->NextScheduledProbe %d m->NextScheduledQuery %d m->SuppressSending %d %d",
m->SuppressProbes - m->timenow,
m->NextScheduledProbe - m->timenow,
m->NextScheduledQuery - m->timenow,
m->SuppressSending,
m->SuppressSending - m->timenow);
m->SuppressProbes = NonZeroTime(m->timenow + DefaultProbeIntervalForTypeUnique/2 + mDNSRandom(DefaultProbeIntervalForTypeUnique/2));
}
}
rr->LastAPTime = m->SuppressProbes - rr->ThisAPInterval;
}
else if (m->SuppressProbes && m->SuppressProbes - m->timenow >= 0)
rr->LastAPTime = m->SuppressProbes - rr->ThisAPInterval + DefaultProbeIntervalForTypeUnique * DefaultProbeCountForTypeUnique + rr->ThisAPInterval / 2;
else
rr->LastAPTime = m->timenow - rr->ThisAPInterval;
if (rr->AddressProxy.type) rr->LastAPTime = m->timenow;
if (m->KnownBugs & mDNS_KnownBug_LimitedIPv6)
if (rr->WakeUp.HMAC.l[0] && rr->resrec.rrtype == kDNSType_AAAA)
rr->LastAPTime = m->timenow - rr->ThisAPInterval + mDNSPlatformOneSecond * 10;
rr->LastMCTime = m->timenow;
rr->LastMCInterface = mDNSInterfaceMark;
SetNextAnnounceProbeTime(m, rr);
}
mDNSlocal const domainname *SetUnicastTargetToHostName(mDNS *const m, AuthRecord *rr)
{
const domainname *target;
if (rr->AutoTarget)
{
DomainAuthInfo *AuthInfo = GetAuthInfoForName_internal(m, rr->resrec.name);
if (!AuthInfo || !AuthInfo->AutoTunnel) rr->AutoTarget = Target_AutoHostAndNATMAP;
}
target = GetServiceTarget(m, rr);
if (!target || target->c[0] == 0)
{
LogInfo("SetUnicastTargetToHostName No target for %s", ARDisplayString(m, rr));
rr->state = regState_NoTarget;
return mDNSNULL;
}
else
{
LogInfo("SetUnicastTargetToHostName target %##s for resource record %s", target->c, ARDisplayString(m,rr));
return target;
}
}
mDNSlocal void SetTargetToHostName(mDNS *const m, AuthRecord *const rr)
{
domainname *const target = GetRRDomainNameTarget(&rr->resrec);
const domainname *newname = &m->MulticastHostname;
if (!target) LogInfo("SetTargetToHostName: Don't know how to set the target of rrtype %s", DNSTypeName(rr->resrec.rrtype));
if (!(rr->ForceMCast || rr->resrec.InterfaceID == mDNSInterface_LocalOnly || rr->resrec.InterfaceID == mDNSInterface_P2P || IsLocalDomain(&rr->namestorage)))
{
const domainname *const n = SetUnicastTargetToHostName(m, rr);
if (n) newname = n;
else { target->c[0] = 0; SetNewRData(&rr->resrec, mDNSNULL, 0); return; }
}
if (target && SameDomainName(target, newname))
debugf("SetTargetToHostName: Target of %##s is already %##s", rr->resrec.name->c, target->c);
if (target && !SameDomainName(target, newname))
{
AssignDomainName(target, newname);
SetNewRData(&rr->resrec, mDNSNULL, 0);
rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
if (rr->RequireGoodbye && rr->resrec.RecordType == kDNSRecordTypeShared)
debugf("Have announced shared record %##s (%s) at least once: should have sent a goodbye packet before updating",
rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
rr->AnnounceCount = InitialAnnounceCount;
rr->RequireGoodbye = mDNSfalse;
InitializeLastAPTime(m, rr);
}
}
mDNSlocal void AcknowledgeRecord(mDNS *const m, AuthRecord *const rr)
{
if (rr->RecordCallback)
{
rr->Acknowledged = mDNStrue;
mDNS_DropLockBeforeCallback(); rr->RecordCallback(m, rr, mStatus_NoError);
mDNS_ReclaimLockAfterCallback(); }
}
mDNSexport void ActivateUnicastRegistration(mDNS *const m, AuthRecord *const rr)
{
if (rr->resrec.rrtype != kDNSType_SRV)
{
AuthRecord *srvRR = mDNSNULL;
if (rr->resrec.rrtype == kDNSType_PTR)
srvRR = rr->Additional1;
else if (rr->resrec.rrtype == kDNSType_TXT)
srvRR = rr->DependentOn;
if (srvRR)
{
if (srvRR->resrec.rrtype != kDNSType_SRV)
{
LogMsg("ActivateUnicastRegistration: ERROR!! Resource record %s wrong, expecting SRV type", ARDisplayString(m, srvRR));
}
else
{
LogInfo("ActivateUnicastRegistration: Found Service Record %s in state %d for %##s (%s)",
ARDisplayString(m, srvRR), srvRR->state, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
rr->state = srvRR->state;
}
}
}
if (rr->state == regState_NoTarget)
{
LogInfo("ActivateUnicastRegistration record %s in regState_NoTarget, not activating", ARDisplayString(m, rr));
return;
}
if (rr->resrec.RecordType == kDNSRecordTypeDeregistering)
{
LogInfo("ActivateUnicastRegistration: Resource record %s, current state %d, moving to DeregPending", ARDisplayString(m, rr), rr->state);
rr->state = regState_DeregPending;
}
else
{
LogInfo("ActivateUnicastRegistration: Resource record %s, current state %d, moving to Pending", ARDisplayString(m, rr), rr->state);
rr->state = regState_Pending;
}
rr->ProbeCount = 0;
rr->AnnounceCount = 0;
rr->ThisAPInterval = INIT_RECORD_REG_INTERVAL;
rr->LastAPTime = m->timenow - rr->ThisAPInterval;
rr->expire = 0; rr->uselease = mDNStrue;
rr->updateid = zeroID;
rr->SRVChanged = mDNSfalse;
rr->updateError = mStatus_NoError;
if (rr->NATinfo.clientContext)
{
mDNS_StopNATOperation_internal(m, &rr->NATinfo);
rr->NATinfo.clientContext = mDNSNULL;
}
if (rr->nta) { CancelGetZoneData(m, rr->nta); rr->nta = mDNSNULL; }
if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
if (m->NextuDNSEvent - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
m->NextuDNSEvent = (rr->LastAPTime + rr->ThisAPInterval);
}
#define RecordLDT(A,B) ((A)->resrec.RecordType == (B)->resrec.RecordType || \
((A)->resrec.RecordType | (B)->resrec.RecordType) == (kDNSRecordTypeUnique | kDNSRecordTypeVerified) || \
((A)->resrec.RecordType == kDNSRecordTypeDeregistering || (B)->resrec.RecordType == kDNSRecordTypeDeregistering))
#define RecordIsLocalDuplicate(A,B) \
((A)->resrec.InterfaceID == (B)->resrec.InterfaceID && RecordLDT((A),(B)) && IdenticalResourceRecord(&(A)->resrec, &(B)->resrec))
mDNSexport mStatus mDNS_Register_internal(mDNS *const m, AuthRecord *const rr)
{
domainname *target = GetRRDomainNameTarget(&rr->resrec);
AuthRecord *r;
AuthRecord **p = &m->ResourceRecords;
AuthRecord **d = &m->DuplicateRecords;
if ((mDNSs32)rr->resrec.rroriginalttl <= 0)
{ LogMsg("mDNS_Register_internal: TTL %X should be 1 - 0x7FFFFFFF %s", rr->resrec.rroriginalttl, ARDisplayString(m, rr)); return(mStatus_BadParamErr); }
if (!rr->resrec.RecordType)
{ LogMsg("mDNS_Register_internal: RecordType must be non-zero %s", ARDisplayString(m, rr)); return(mStatus_BadParamErr); }
if (m->ShutdownTime)
{ LogMsg("mDNS_Register_internal: Shutting down, can't register %s", ARDisplayString(m, rr)); return(mStatus_ServiceNotRunning); }
if (m->DivertMulticastAdvertisements && !AuthRecord_uDNS(rr))
{
mDNSInterfaceID previousID = rr->resrec.InterfaceID;
if (rr->resrec.InterfaceID == mDNSInterface_Any || rr->resrec.InterfaceID == mDNSInterface_P2P) rr->resrec.InterfaceID = mDNSInterface_LocalOnly;
if (rr->resrec.InterfaceID != mDNSInterface_LocalOnly)
{
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
if (intf && !intf->Advertise) rr->resrec.InterfaceID = mDNSInterface_LocalOnly;
}
if (rr->resrec.InterfaceID != previousID)
LogInfo("mDNS_Register_internal: Diverting record to local-only %s", ARDisplayString(m, rr));
}
while (*p && *p != rr) p=&(*p)->next;
while (*d && *d != rr) d=&(*d)->next;
if (*d || *p)
{
LogMsg("Error! Tried to register AuthRecord %p %##s (%s) that's already in the list",
rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
return(mStatus_AlreadyRegistered);
}
if (rr->DependentOn)
{
if (rr->resrec.RecordType == kDNSRecordTypeUnique)
rr->resrec.RecordType = kDNSRecordTypeVerified;
else
{
LogMsg("mDNS_Register_internal: ERROR! %##s (%s): rr->DependentOn && RecordType != kDNSRecordTypeUnique",
rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
return(mStatus_Invalid);
}
if (!(rr->DependentOn->resrec.RecordType & (kDNSRecordTypeUnique | kDNSRecordTypeVerified)))
{
LogMsg("mDNS_Register_internal: ERROR! %##s (%s): rr->DependentOn->RecordType bad type %X",
rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), rr->DependentOn->resrec.RecordType);
return(mStatus_Invalid);
}
}
if (rr->resrec.InterfaceID && rr->resrec.InterfaceID != mDNSInterface_LocalOnly && rr->resrec.InterfaceID != mDNSInterface_P2P)
{
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
if (!intf)
{
debugf("mDNS_Register_internal: Bogus InterfaceID %p in resource record", rr->resrec.InterfaceID);
return(mStatus_BadReferenceErr);
}
}
rr->next = mDNSNULL;
if (rr->AutoTarget && target) target->c[0] = 0;
rr->Acknowledged = mDNSfalse;
rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
rr->AnnounceCount = InitialAnnounceCount;
rr->RequireGoodbye = mDNSfalse;
rr->AnsweredLocalQ = mDNSfalse;
rr->IncludeInProbe = mDNSfalse;
rr->ImmedUnicast = mDNSfalse;
rr->SendNSECNow = mDNSNULL;
rr->ImmedAnswer = mDNSNULL;
rr->ImmedAdditional = mDNSNULL;
rr->SendRNow = mDNSNULL;
rr->v4Requester = zerov4Addr;
rr->v6Requester = zerov6Addr;
rr->NextResponse = mDNSNULL;
rr->NR_AnswerTo = mDNSNULL;
rr->NR_AdditionalTo = mDNSNULL;
if (!rr->AutoTarget) InitializeLastAPTime(m, rr);
rr->NewRData = mDNSNULL;
rr->newrdlength = 0;
rr->UpdateCallback = mDNSNULL;
rr->UpdateCredits = kMaxUpdateCredits;
rr->NextUpdateCredit = 0;
rr->UpdateBlocked = 0;
if (rr->WakeUp.HMAC.l[0] && !rr->AddressProxy.type) rr->AnnounceCount = 2;
rr->state = regState_Zero;
rr->uselease = 0;
rr->expire = 0;
rr->Private = 0;
rr->updateid = zeroID;
rr->zone = rr->resrec.name;
rr->nta = mDNSNULL;
rr->tcp = mDNSNULL;
rr->OrigRData = 0;
rr->OrigRDLen = 0;
rr->InFlightRData = 0;
rr->InFlightRDLen = 0;
rr->QueuedRData = 0;
rr->QueuedRDLen = 0;
if (rr->resrec.rrtype == kDNSType_TXT && rr->resrec.rdlength == 0) { rr->resrec.rdlength = 1; rr->resrec.rdata->u.txt.c[0] = 0; }
if (rr->AutoTarget)
{
SetTargetToHostName(m, rr); #ifndef UNICAST_DISABLED
if (rr->state == regState_NoTarget)
{
domainname *tar = GetRRDomainNameTarget(&rr->resrec);
if (tar) tar->c[0] = 0;
LogInfo("mDNS_Register_internal: record %s in NoTarget state", ARDisplayString(m, rr));
}
#endif
}
else
{
rr->resrec.rdlength = GetRDLength(&rr->resrec, mDNSfalse);
rr->resrec.rdestimate = GetRDLength(&rr->resrec, mDNStrue);
}
if (!ValidateDomainName(rr->resrec.name))
{ LogMsg("Attempt to register record with invalid name: %s", ARDisplayString(m, rr)); return(mStatus_Invalid); }
if (!ValidateRData(rr->resrec.rrtype, rr->resrec.rdlength, rr->resrec.rdata))
{ LogMsg("Attempt to register record with invalid rdata: %s", ARDisplayString(m, rr)); return(mStatus_Invalid); }
rr->resrec.namehash = DomainNameHashValue(rr->resrec.name);
rr->resrec.rdatahash = target ? DomainNameHashValue(target) : RDataHashValue(&rr->resrec);
if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly || rr->resrec.InterfaceID == mDNSInterface_P2P)
{
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
{
const AuthRecord *s1 = rr->RRSet ? rr->RRSet : rr;
for (r = m->ResourceRecords; r; r=r->next)
{
const AuthRecord *s2 = r->RRSet ? r->RRSet : r;
if (s1 != s2 && SameResourceRecordSignature(r, rr) && !IdenticalSameNameRecord(&r->resrec, &rr->resrec))
break;
}
if (r) {
debugf("Name conflict %p %##s (%s)", rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
rr->resrec.RecordType = kDNSRecordTypeDeregistering;
rr->resrec.rroriginalttl = 0;
rr->ImmedAnswer = mDNSInterfaceMark;
m->LocalRemoveEvents = mDNStrue;
m->NextScheduledResponse = m->timenow;
}
}
}
#ifndef UNICAST_DISABLED
if (AuthRecord_uDNS(rr))
{
if (!m->NewLocalRecords) m->NewLocalRecords = rr;
while (*p) p=&(*p)->next;
*p = rr;
if (rr->resrec.RecordType == kDNSRecordTypeUnique) rr->resrec.RecordType = kDNSRecordTypeVerified;
rr->ProbeCount = 0;
rr->AnnounceCount = 0;
if (rr->state != regState_NoTarget) ActivateUnicastRegistration(m, rr);
return(mStatus_NoError); }
#endif
for (r = m->ResourceRecords; r; r=r->next)
if (RecordIsLocalDuplicate(r, rr))
{
if (r->resrec.RecordType == kDNSRecordTypeDeregistering) r->AnnounceCount = 0;
else break;
}
if (r)
{
debugf("mDNS_Register_internal:Adding to duplicate list %s", ARDisplayString(m,rr));
*d = rr;
if (rr->resrec.RecordType == kDNSRecordTypeUnique && r->resrec.RecordType == kDNSRecordTypeVerified)
rr->ProbeCount = 0;
}
else
{
debugf("mDNS_Register_internal: Adding to active record list %s", ARDisplayString(m,rr));
if (!m->NewLocalRecords) m->NewLocalRecords = rr;
*p = rr;
}
if (!AuthRecord_uDNS(rr)) {
if (rr->resrec.RecordType != kDNSRecordTypeUnique && rr->resrec.RecordType != kDNSRecordTypeDeregistering)
AcknowledgeRecord(m, rr);
mDNS_UpdateAllowSleep(m);
}
return(mStatus_NoError);
}
mDNSlocal void RecordProbeFailure(mDNS *const m, const AuthRecord *const rr)
{
m->ProbeFailTime = m->timenow;
m->NumFailedProbes++;
if (m->NumFailedProbes >= 15)
{
m->SuppressProbes = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 5);
LogMsg("Excessive name conflicts (%lu) for %##s (%s); rate limiting in effect",
m->NumFailedProbes, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
}
}
mDNSlocal void CompleteRDataUpdate(mDNS *const m, AuthRecord *const rr)
{
RData *OldRData = rr->resrec.rdata;
mDNSu16 OldRDLen = rr->resrec.rdlength;
SetNewRData(&rr->resrec, rr->NewRData, rr->newrdlength); rr->NewRData = mDNSNULL; if (rr->UpdateCallback)
rr->UpdateCallback(m, rr, OldRData, OldRDLen); }
mDNSexport mStatus mDNS_Deregister_internal(mDNS *const m, AuthRecord *const rr, mDNS_Dereg_type drt)
{
AuthRecord *r2;
mDNSu8 RecordType = rr->resrec.RecordType;
AuthRecord **p = &m->ResourceRecords;
while (*p && *p != rr) p=&(*p)->next;
if (*p)
{
if (drt == mDNS_Dereg_conflict) {
for (r2 = m->DuplicateRecords; r2; r2=r2->next) if (RecordIsLocalDuplicate(r2, rr)) r2->ProbeCount = 0xFF;
}
else
{
AuthRecord **d = &m->DuplicateRecords;
while (*d && !RecordIsLocalDuplicate(*d, rr)) d=&(*d)->next;
if (*d)
{
AuthRecord *dup = *d;
debugf("mDNS_Register_internal: Duplicate record %p taking over from %p %##s (%s)",
dup, rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
*d = dup->next; dup->next = rr->next; rr->next = dup; dup->resrec.RecordType = rr->resrec.RecordType;
dup->ProbeCount = rr->ProbeCount;
dup->AnnounceCount = rr->AnnounceCount;
dup->RequireGoodbye = rr->RequireGoodbye;
dup->AnsweredLocalQ = rr->AnsweredLocalQ;
dup->ImmedAnswer = rr->ImmedAnswer;
dup->ImmedUnicast = rr->ImmedUnicast;
dup->ImmedAdditional = rr->ImmedAdditional;
dup->v4Requester = rr->v4Requester;
dup->v6Requester = rr->v6Requester;
dup->ThisAPInterval = rr->ThisAPInterval;
dup->LastAPTime = rr->LastAPTime;
dup->LastMCTime = rr->LastMCTime;
dup->LastMCInterface = rr->LastMCInterface;
dup->Private = rr->Private;
dup->state = rr->state;
rr->RequireGoodbye = mDNSfalse;
rr->AnsweredLocalQ = mDNSfalse;
}
}
}
else
{
p = &m->DuplicateRecords;
while (*p && *p != rr) p=&(*p)->next;
if (*p) rr->RequireGoodbye = mDNSfalse;
if (*p) debugf("mDNS_Deregister_internal: Deleting DuplicateRecord %p %##s (%s)",
rr, rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
}
if (!*p)
{
if (drt != mDNS_Dereg_repeat)
LogMsg("mDNS_Deregister_internal: Record %p not found in list %s", rr, ARDisplayString(m,rr));
return(mStatus_BadReferenceErr);
}
#ifndef UNICAST_DISABLED
if (AuthRecord_uDNS(rr))
{
if (rr->RequireGoodbye)
{
if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
rr->resrec.RecordType = kDNSRecordTypeDeregistering;
m->LocalRemoveEvents = mDNStrue;
uDNS_DeregisterRecord(m, rr);
return(mStatus_NoError);
}
rr->updateid = zeroID;
if (rr->NATinfo.clientContext)
{
mDNS_StopNATOperation_internal(m, &rr->NATinfo);
rr->NATinfo.clientContext = mDNSNULL;
}
if (rr->nta) { CancelGetZoneData(m, rr->nta); rr->nta = mDNSNULL; }
if (rr->tcp) { DisposeTCPConn(rr->tcp); rr->tcp = mDNSNULL; }
}
#endif // UNICAST_DISABLED
if (RecordType == kDNSRecordTypeUnregistered)
LogMsg("mDNS_Deregister_internal: %s already marked kDNSRecordTypeUnregistered", ARDisplayString(m, rr));
else if (RecordType == kDNSRecordTypeDeregistering)
{
LogMsg("mDNS_Deregister_internal: %s already marked kDNSRecordTypeDeregistering", ARDisplayString(m, rr));
return(mStatus_BadReferenceErr);
}
if (rr->WakeUp.HMAC.l[0] ||
(RecordType == kDNSRecordTypeShared && (rr->RequireGoodbye || rr->AnsweredLocalQ)))
{
verbosedebugf("mDNS_Deregister_internal: Starting deregistration for %s", ARDisplayString(m, rr));
rr->resrec.RecordType = kDNSRecordTypeDeregistering;
rr->resrec.rroriginalttl = 0;
rr->AnnounceCount = rr->WakeUp.HMAC.l[0] ? WakeupCount : (drt == mDNS_Dereg_rapid) ? 1 : GoodbyeCount;
rr->ThisAPInterval = mDNSPlatformOneSecond * 2;
rr->LastAPTime = m->timenow - rr->ThisAPInterval;
m->LocalRemoveEvents = mDNStrue;
if (m->NextScheduledResponse - (m->timenow + mDNSPlatformOneSecond/10) >= 0)
m->NextScheduledResponse = (m->timenow + mDNSPlatformOneSecond/10);
}
else
{
*p = rr->next; if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
if (m->NewLocalRecords == rr) m->NewLocalRecords = rr->next;
rr->next = mDNSNULL;
verbosedebugf("mDNS_Deregister_internal: Deleting record for %s", ARDisplayString(m, rr));
rr->resrec.RecordType = kDNSRecordTypeUnregistered;
if ((drt == mDNS_Dereg_conflict || drt == mDNS_Dereg_repeat) && RecordType == kDNSRecordTypeShared)
debugf("mDNS_Deregister_internal: Cannot have a conflict on a shared record! %##s (%s)",
rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
if (rr->NewRData) CompleteRDataUpdate(m, rr);
if (drt != mDNS_Dereg_conflict)
{
mDNS_DropLockBeforeCallback(); LogInfo("mDNS_Deregister_internal: mStatus_MemFree for %s", ARDisplayString(m, rr));
if (rr->RecordCallback)
rr->RecordCallback(m, rr, mStatus_MemFree); mDNS_ReclaimLockAfterCallback(); }
else
{
RecordProbeFailure(m, rr);
mDNS_DropLockBeforeCallback(); if (rr->RecordCallback)
rr->RecordCallback(m, rr, mStatus_NameConflict); mDNS_ReclaimLockAfterCallback(); r2 = m->DuplicateRecords;
while (r2)
{
if (r2->ProbeCount != 0xFF) r2 = r2->next;
else { mDNS_Deregister_internal(m, r2, mDNS_Dereg_conflict); r2 = m->DuplicateRecords; }
}
}
}
mDNS_UpdateAllowSleep(m);
return(mStatus_NoError);
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Packet Sending Functions
#endif
mDNSlocal void AddRecordToResponseList(AuthRecord ***nrpp, AuthRecord *rr, AuthRecord *add)
{
if (rr->NextResponse == mDNSNULL && *nrpp != &rr->NextResponse)
{
**nrpp = rr;
if (add && add->NR_AdditionalTo) add = add->NR_AdditionalTo;
rr->NR_AdditionalTo = add;
*nrpp = &rr->NextResponse;
}
debugf("AddRecordToResponseList: %##s (%s) already in list", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype));
}
mDNSlocal void AddAdditionalsToResponseList(mDNS *const m, AuthRecord *ResponseRecords, AuthRecord ***nrpp, const mDNSInterfaceID InterfaceID)
{
AuthRecord *rr, *rr2;
for (rr=ResponseRecords; rr; rr=rr->NextResponse) {
if (rr->Additional1 && ResourceRecordIsValidInterfaceAnswer(rr->Additional1, InterfaceID))
AddRecordToResponseList(nrpp, rr->Additional1, rr);
if (rr->Additional2 && ResourceRecordIsValidInterfaceAnswer(rr->Additional2, InterfaceID))
AddRecordToResponseList(nrpp, rr->Additional2, rr);
if (rr->resrec.rrtype == kDNSType_SRV)
{
for (rr2=m->ResourceRecords; rr2; rr2=rr2->next) if (RRTypeIsAddressType(rr2->resrec.rrtype) && ResourceRecordIsValidInterfaceAnswer(rr2, InterfaceID) && rr->resrec.rdatahash == rr2->resrec.namehash && SameDomainName(&rr->resrec.rdata->u.srv.target, rr2->resrec.name))
AddRecordToResponseList(nrpp, rr2, rr);
}
else if (RRTypeIsAddressType(rr->resrec.rrtype)) {
for (rr2=m->ResourceRecords; rr2; rr2=rr2->next) if (RRTypeIsAddressType(rr2->resrec.rrtype) && ResourceRecordIsValidInterfaceAnswer(rr2, InterfaceID) && rr->resrec.namehash == rr2->resrec.namehash && SameDomainName(rr->resrec.name, rr2->resrec.name))
AddRecordToResponseList(nrpp, rr2, rr);
}
else if (rr->resrec.rrtype == kDNSType_PTR) {
if (ResourceRecordIsValidInterfaceAnswer(&m->DeviceInfo, InterfaceID) &&
SameDomainLabel(rr->resrec.rdata->u.name.c, m->DeviceInfo.resrec.name->c))
AddRecordToResponseList(nrpp, &m->DeviceInfo, rr);
}
}
}
mDNSlocal void SendDelayedUnicastResponse(mDNS *const m, const mDNSAddr *const dest, const mDNSInterfaceID InterfaceID)
{
AuthRecord *rr;
AuthRecord *ResponseRecords = mDNSNULL;
AuthRecord **nrp = &ResponseRecords;
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
if (rr->ImmedAnswer == mDNSInterfaceMark ||
mDNSSameIPv4Address(rr->v4Requester, onesIPv4Addr) ||
mDNSSameIPv6Address(rr->v6Requester, onesIPv6Addr) )
rr->ImmedUnicast = mDNSfalse;
if (rr->ImmedUnicast && rr->ImmedAnswer == InterfaceID)
if ((dest->type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->v4Requester, dest->ip.v4)) ||
(dest->type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->v6Requester, dest->ip.v6)))
{
rr->ImmedAnswer = mDNSNULL; rr->ImmedUnicast = mDNSfalse;
rr->v4Requester = zerov4Addr;
rr->v6Requester = zerov6Addr;
if (rr->NextResponse == mDNSNULL && nrp != &rr->NextResponse) { rr->NR_AnswerTo = (mDNSu8*)~0; *nrp = rr; nrp = &rr->NextResponse; }
}
}
AddAdditionalsToResponseList(m, ResponseRecords, &nrp, InterfaceID);
while (ResponseRecords)
{
mDNSu8 *responseptr = m->omsg.data;
mDNSu8 *newptr;
InitializeDNSMessage(&m->omsg.h, zeroID, ResponseFlags);
while (ResponseRecords && ResponseRecords->NR_AnswerTo)
{
rr = ResponseRecords;
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
rr->resrec.rrclass |= kDNSClass_UniqueRRSet; newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAnswers, &rr->resrec);
rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; if (!newptr && m->omsg.h.numAnswers) break; if (newptr) responseptr = newptr;
ResponseRecords = rr->NextResponse;
rr->NextResponse = mDNSNULL;
rr->NR_AnswerTo = mDNSNULL;
rr->NR_AdditionalTo = mDNSNULL;
rr->RequireGoodbye = mDNStrue;
}
while (ResponseRecords && !ResponseRecords->NR_AnswerTo)
{
rr = ResponseRecords;
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
rr->resrec.rrclass |= kDNSClass_UniqueRRSet; newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAdditionals, &rr->resrec);
rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet;
if (newptr) responseptr = newptr;
if (newptr && m->omsg.h.numAnswers) rr->RequireGoodbye = mDNStrue;
else if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask) rr->ImmedAnswer = mDNSInterfaceMark;
ResponseRecords = rr->NextResponse;
rr->NextResponse = mDNSNULL;
rr->NR_AnswerTo = mDNSNULL;
rr->NR_AdditionalTo = mDNSNULL;
}
if (m->omsg.h.numAnswers)
mDNSSendDNSMessage(m, &m->omsg, responseptr, mDNSInterface_Any, mDNSNULL, dest, MulticastDNSPort, mDNSNULL, mDNSNULL);
}
}
mDNSexport void CompleteDeregistration(mDNS *const m, AuthRecord *rr)
{
LogInfo("CompleteDeregistration: called for Resource record %s", ARDisplayString(m, rr));
rr->resrec.RecordType = kDNSRecordTypeShared;
rr->RequireGoodbye = mDNSfalse;
rr->WakeUp.HMAC = zeroEthAddr;
if (rr->AnsweredLocalQ) { AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse); rr->AnsweredLocalQ = mDNSfalse; }
mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal); }
mDNSlocal void DiscardDeregistrations(mDNS *const m)
{
if (m->CurrentRecord)
LogMsg("DiscardDeregistrations ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
AuthRecord *rr = m->CurrentRecord;
if (!AuthRecord_uDNS(rr) && rr->resrec.RecordType == kDNSRecordTypeDeregistering)
CompleteDeregistration(m, rr); else
m->CurrentRecord = rr->next;
}
}
mDNSlocal mStatus GetLabelDecimalValue(const mDNSu8 *const src, mDNSu8 *dst)
{
int i, val = 0;
if (src[0] < 1 || src[0] > 3) return(mStatus_Invalid);
for (i=1; i<=src[0]; i++)
{
if (src[i] < '0' || src[i] > '9') return(mStatus_Invalid);
val = val * 10 + src[i] - '0';
}
if (val > 255) return(mStatus_Invalid);
*dst = (mDNSu8)val;
return(mStatus_NoError);
}
mDNSlocal mStatus GetIPv4FromName(mDNSAddr *const a, const domainname *const name)
{
int skip = CountLabels(name) - 6;
if (skip < 0) { LogMsg("GetIPFromName: Need six labels in IPv4 reverse mapping name %##s", name); return mStatus_Invalid; }
if (GetLabelDecimalValue(SkipLeadingLabels(name, skip+3)->c, &a->ip.v4.b[0]) ||
GetLabelDecimalValue(SkipLeadingLabels(name, skip+2)->c, &a->ip.v4.b[1]) ||
GetLabelDecimalValue(SkipLeadingLabels(name, skip+1)->c, &a->ip.v4.b[2]) ||
GetLabelDecimalValue(SkipLeadingLabels(name, skip+0)->c, &a->ip.v4.b[3])) return mStatus_Invalid;
a->type = mDNSAddrType_IPv4;
return(mStatus_NoError);
}
#define HexVal(X) ( ((X) >= '0' && (X) <= '9') ? ((X) - '0' ) : \
((X) >= 'A' && (X) <= 'F') ? ((X) - 'A' + 10) : \
((X) >= 'a' && (X) <= 'f') ? ((X) - 'a' + 10) : -1)
mDNSlocal mStatus GetIPv6FromName(mDNSAddr *const a, const domainname *const name)
{
int i, h, l;
const domainname *n;
int skip = CountLabels(name) - 34;
if (skip < 0) { LogMsg("GetIPFromName: Need 34 labels in IPv6 reverse mapping name %##s", name); return mStatus_Invalid; }
n = SkipLeadingLabels(name, skip);
for (i=0; i<16; i++)
{
if (n->c[0] != 1) return mStatus_Invalid;
l = HexVal(n->c[1]);
n = (const domainname *)(n->c + 2);
if (n->c[0] != 1) return mStatus_Invalid;
h = HexVal(n->c[1]);
n = (const domainname *)(n->c + 2);
if (l<0 || h<0) return mStatus_Invalid;
a->ip.v6.b[15-i] = (mDNSu8)((h << 4) | l);
}
a->type = mDNSAddrType_IPv6;
return(mStatus_NoError);
}
mDNSlocal mDNSs32 ReverseMapDomainType(const domainname *const name)
{
int skip = CountLabels(name) - 2;
if (skip >= 0)
{
const domainname *suffix = SkipLeadingLabels(name, skip);
if (SameDomainName(suffix, (const domainname*)"\x7" "in-addr" "\x4" "arpa")) return mDNSAddrType_IPv4;
if (SameDomainName(suffix, (const domainname*)"\x3" "ip6" "\x4" "arpa")) return mDNSAddrType_IPv6;
}
return(mDNSAddrType_None);
}
mDNSlocal void SendARP(mDNS *const m, const mDNSu8 op, const AuthRecord *const rr,
const mDNSv4Addr *const spa, const mDNSEthAddr *const tha, const mDNSv4Addr *const tpa, const mDNSEthAddr *const dst)
{
int i;
mDNSu8 *ptr = m->omsg.data;
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
if (!intf) { LogMsg("SendARP: No interface with InterfaceID %p found %s", rr->resrec.InterfaceID, ARDisplayString(m,rr)); return; }
for (i=0; i<6; i++) *ptr++ = dst->b[i];
for (i=0; i<6; i++) *ptr++ = intf->MAC.b[0];
*ptr++ = 0x08; *ptr++ = 0x06;
*ptr++ = 0x00; *ptr++ = 0x01; *ptr++ = 0x08; *ptr++ = 0x00; *ptr++ = 6; *ptr++ = 4; *ptr++ = 0x00; *ptr++ = op;
for (i=0; i<6; i++) *ptr++ = intf->MAC.b[i];
for (i=0; i<4; i++) *ptr++ = spa->b[i];
for (i=0; i<6; i++) *ptr++ = tha->b[i];
for (i=0; i<4; i++) *ptr++ = tpa->b[i];
mDNSPlatformSendRawPacket(m->omsg.data, ptr, rr->resrec.InterfaceID);
}
mDNSlocal mDNSu16 CheckSum(const void *const data, mDNSs32 length, mDNSu32 sum)
{
const mDNSu16 *ptr = data;
while (length > 0) { length -= 2; sum += *ptr++; }
sum = (sum & 0xFFFF) + (sum >> 16);
sum = (sum & 0xFFFF) + (sum >> 16);
return(sum != 0xFFFF ? sum : 0);
}
mDNSlocal mDNSu16 IPv6CheckSum(const mDNSv6Addr *const src, const mDNSv6Addr *const dst, const mDNSu8 protocol, const void *const data, const mDNSu32 length)
{
IPv6PseudoHeader ph;
ph.src = *src;
ph.dst = *dst;
ph.len.b[0] = length >> 24;
ph.len.b[1] = length >> 16;
ph.len.b[2] = length >> 8;
ph.len.b[3] = length;
ph.pro.b[0] = 0;
ph.pro.b[1] = 0;
ph.pro.b[2] = 0;
ph.pro.b[3] = protocol;
return CheckSum(&ph, sizeof(ph), CheckSum(data, length, 0));
}
mDNSlocal void SendNDP(mDNS *const m, const mDNSu8 op, const mDNSu8 flags, const AuthRecord *const rr,
const mDNSv6Addr *const spa, const mDNSEthAddr *const tha, const mDNSv6Addr *const tpa, const mDNSEthAddr *const dst)
{
int i;
mDNSOpaque16 checksum;
mDNSu8 *ptr = m->omsg.data;
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] } };
const mDNSv6Addr *const v6dst = (op == NDP_Sol) ? &mc : tpa;
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, rr->resrec.InterfaceID);
if (!intf) { LogMsg("SendNDP: No interface with InterfaceID %p found %s", rr->resrec.InterfaceID, ARDisplayString(m,rr)); return; }
for (i=0; i<6; i++) *ptr++ = dst->b[i];
for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
*ptr++ = 0x86; *ptr++ = 0xDD;
*ptr++ = 0x60; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x20; *ptr++ = 0x3A; *ptr++ = 0xFF;
for (i=0; i<16; i++) *ptr++ = spa->b[i];
for (i=0; i<16; i++) *ptr++ = v6dst->b[i];
*ptr++ = op; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = flags;
*ptr++ = 0x00; *ptr++ = 0x00; *ptr++ = 0x00;
if (op == NDP_Sol) {
for (i=0; i<16; i++) *ptr++ = tpa->b[i];
if (!mDNSIPv6AddressIsZero(*spa))
{
*ptr++ = NDP_SrcLL; *ptr++ = 0x01; for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
}
}
else {
for (i=0; i<16; i++) *ptr++ = spa->b[i];
*ptr++ = NDP_TgtLL; *ptr++ = 0x01; for (i=0; i<6; i++) *ptr++ = (tha ? *tha : intf->MAC).b[i];
}
m->omsg.data[0x13] = ptr - &m->omsg.data[0x36]; checksum.NotAnInteger = ~IPv6CheckSum(spa, v6dst, 0x3A, &m->omsg.data[0x36], m->omsg.data[0x13]);
m->omsg.data[0x38] = checksum.b[0];
m->omsg.data[0x39] = checksum.b[1];
mDNSPlatformSendRawPacket(m->omsg.data, ptr, rr->resrec.InterfaceID);
}
mDNSlocal void SetupOwnerOpt(const mDNS *const m, const NetworkInterfaceInfo *const intf, rdataOPT *const owner)
{
owner->u.owner.vers = 0;
owner->u.owner.seq = m->SleepSeqNum;
owner->u.owner.HMAC = m->PrimaryMAC;
owner->u.owner.IMAC = intf->MAC;
owner->u.owner.password = zeroEthAddr;
owner->opt = kDNSOpt_Owner;
owner->optlen = DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) - 4;
}
mDNSlocal void GrantUpdateCredit(AuthRecord *rr)
{
if (++rr->UpdateCredits >= kMaxUpdateCredits) rr->NextUpdateCredit = 0;
else rr->NextUpdateCredit = NonZeroTime(rr->NextUpdateCredit + kUpdateCreditRefreshInterval);
}
mDNSlocal void SendResponses(mDNS *const m)
{
int pktcount = 0;
AuthRecord *rr, *r2;
mDNSs32 maxExistingAnnounceInterval = 0;
const NetworkInterfaceInfo *intf = GetFirstActiveInterface(m->HostInterfaces);
m->NextScheduledResponse = m->timenow + 0x78000000;
if (m->SleepState == SleepState_Transferring) RetrySPSRegistrations(m);
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->ImmedUnicast)
{
mDNSAddr v4 = { mDNSAddrType_IPv4, {{{0}}} };
mDNSAddr v6 = { mDNSAddrType_IPv6, {{{0}}} };
v4.ip.v4 = rr->v4Requester;
v6.ip.v6 = rr->v6Requester;
if (!mDNSIPv4AddressIsZero(rr->v4Requester)) SendDelayedUnicastResponse(m, &v4, rr->ImmedAnswer);
if (!mDNSIPv6AddressIsZero(rr->v6Requester)) SendDelayedUnicastResponse(m, &v6, rr->ImmedAnswer);
if (rr->ImmedUnicast)
{
LogMsg("SendResponses: ERROR: rr->ImmedUnicast still set: %s", ARDisplayString(m, rr));
rr->ImmedUnicast = mDNSfalse;
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
while (rr->NextUpdateCredit && m->timenow - rr->NextUpdateCredit >= 0) GrantUpdateCredit(rr);
if (TimeToAnnounceThisRecord(rr, m->timenow))
{
if (rr->resrec.RecordType == kDNSRecordTypeDeregistering)
{
if (!rr->WakeUp.HMAC.l[0])
{
if (rr->AnnounceCount) rr->ImmedAnswer = mDNSInterfaceMark; }
else
{
LogSPS("SendResponses: Sending wakeup %2d for %.6a %s", rr->AnnounceCount-3, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
SendWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.IMAC, &rr->WakeUp.password);
for (r2 = rr; r2; r2=r2->next)
if (r2->AnnounceCount && r2->resrec.InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&r2->WakeUp.IMAC, &rr->WakeUp.IMAC))
{
if (r2->AddressProxy.type == mDNSAddrType_IPv6 && r2->AnnounceCount == WakeupCount)
{
LogSPS("NDP Announcement %2d Releasing traffic for H-MAC %.6a I-MAC %.6a %s",
r2->AnnounceCount-3, &r2->WakeUp.HMAC, &r2->WakeUp.IMAC, ARDisplayString(m,r2));
SendNDP(m, NDP_Adv, NDP_Override, r2, &r2->AddressProxy.ip.v6, &r2->WakeUp.IMAC, &AllHosts_v6, &AllHosts_v6_Eth);
}
r2->LastAPTime = m->timenow;
if (--r2->AnnounceCount <= GoodbyeCount) r2->WakeUp.HMAC = zeroEthAddr;
}
}
}
else if (ResourceRecordIsValidAnswer(rr))
{
if (rr->AddressProxy.type)
{
rr->AnnounceCount--;
rr->ThisAPInterval *= 2;
rr->LastAPTime = m->timenow;
if (rr->AddressProxy.type == mDNSAddrType_IPv4)
{
LogSPS("ARP Announcement %2d Capturing traffic for H-MAC %.6a I-MAC %.6a %s",
rr->AnnounceCount, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
SendARP(m, 1, rr, &rr->AddressProxy.ip.v4, &zeroEthAddr, &rr->AddressProxy.ip.v4, &onesEthAddr);
}
else if (rr->AddressProxy.type == mDNSAddrType_IPv6)
{
LogSPS("NDP Announcement %2d Capturing traffic for H-MAC %.6a I-MAC %.6a %s",
rr->AnnounceCount, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
SendNDP(m, NDP_Adv, NDP_Override, rr, &rr->AddressProxy.ip.v6, mDNSNULL, &AllHosts_v6, &AllHosts_v6_Eth);
}
}
else
{
rr->ImmedAnswer = mDNSInterfaceMark; if (maxExistingAnnounceInterval < rr->ThisAPInterval)
maxExistingAnnounceInterval = rr->ThisAPInterval;
if (rr->UpdateBlocked) rr->UpdateBlocked = 0;
}
}
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
if ((rr->resrec.InterfaceID && rr->ImmedAnswer) ||
(rr->ThisAPInterval <= maxExistingAnnounceInterval &&
TimeToAnnounceThisRecord(rr, m->timenow + rr->ThisAPInterval/2) &&
!rr->AddressProxy.type && ResourceRecordIsValidAnswer(rr)))
rr->ImmedAnswer = mDNSInterfaceMark;
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
if (rr->ImmedAnswer && rr->resrec.rrtype == kDNSType_SRV)
for (r2=m->ResourceRecords; r2; r2=r2->next) if (RRTypeIsAddressType(r2->resrec.rrtype) && ResourceRecordIsValidAnswer(r2) && rr->LastMCTime - r2->LastMCTime >= 0 && rr->resrec.rdatahash == r2->resrec.namehash && SameDomainName(&rr->resrec.rdata->u.srv.target, r2->resrec.name) &&
(rr->ImmedAnswer == mDNSInterfaceMark || rr->ImmedAnswer == r2->resrec.InterfaceID))
r2->ImmedAdditional = r2->resrec.InterfaceID; if (rr->ImmedAnswer && rr->resrec.RecordType == kDNSRecordTypeShared && rr->resrec.rrtype == kDNSType_PTR)
if (ResourceRecordIsValidAnswer(&m->DeviceInfo) && SameDomainLabel(rr->resrec.rdata->u.name.c, m->DeviceInfo.resrec.name->c))
{
if (!m->DeviceInfo.ImmedAnswer) m->DeviceInfo.ImmedAnswer = rr->ImmedAnswer;
else m->DeviceInfo.ImmedAnswer = mDNSInterfaceMark;
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
{
if (rr->ImmedAnswer) {
for (r2 = m->ResourceRecords; r2; r2=r2->next)
if (ResourceRecordIsValidAnswer(r2))
if (r2->ImmedAnswer != mDNSInterfaceMark &&
r2->ImmedAnswer != rr->ImmedAnswer && SameResourceRecordSignature(r2, rr))
r2->ImmedAnswer = !r2->ImmedAnswer ? rr->ImmedAnswer : mDNSInterfaceMark;
}
else if (rr->ImmedAdditional) {
for (r2 = m->ResourceRecords; r2; r2=r2->next)
if (ResourceRecordIsValidAnswer(r2))
if (r2->ImmedAdditional != rr->ImmedAdditional && SameResourceRecordSignature(r2, rr))
r2->ImmedAdditional = rr->ImmedAdditional;
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
if (rr->ImmedAnswer == mDNSInterfaceMark) {
rr->SendRNow = !intf ? mDNSNULL : (rr->resrec.InterfaceID) ? rr->resrec.InterfaceID : intf->InterfaceID;
rr->ImmedAdditional = mDNSNULL; rr->LastMCTime = m->timenow;
rr->LastMCInterface = rr->ImmedAnswer;
if (TimeToAnnounceThisRecord(rr, m->timenow + rr->ThisAPInterval/2))
{
rr->AnnounceCount--;
if (rr->resrec.RecordType != kDNSRecordTypeDeregistering)
rr->ThisAPInterval *= 2;
rr->LastAPTime = m->timenow;
debugf("Announcing %##s (%s) %d", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), rr->AnnounceCount);
}
}
else if (rr->ImmedAnswer) {
rr->SendRNow = rr->ImmedAnswer; rr->ImmedAdditional = mDNSNULL; rr->LastMCTime = m->timenow;
rr->LastMCInterface = rr->ImmedAnswer;
}
SetNextAnnounceProbeTime(m, rr);
}
while (intf)
{
const int OwnerRecordSpace = (m->AnnounceOwner && intf->MAC.l[0]) ? DNSOpt_Header_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) : 0;
int numDereg = 0;
int numAnnounce = 0;
int numAnswer = 0;
mDNSu8 *responseptr = m->omsg.data;
mDNSu8 *newptr;
InitializeDNSMessage(&m->omsg.h, zeroID, ResponseFlags);
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
if (rr->SendRNow == intf->InterfaceID)
{
RData *OldRData = rr->resrec.rdata;
mDNSu16 oldrdlength = rr->resrec.rdlength;
mDNSu8 active = (mDNSu8)
(rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
(m->SleepState != SleepState_Sleeping || intf->SPSAddr[0].type || intf->SPSAddr[1].type || intf->SPSAddr[2].type));
newptr = mDNSNULL;
if (rr->NewRData && active)
{
if (ResourceRecordIsValidAnswer(rr) && rr->resrec.RecordType == kDNSRecordTypeShared && rr->RequireGoodbye)
{
newptr = PutRR_OS_TTL(responseptr, &m->omsg.h.numAnswers, &rr->resrec, 0);
if (newptr) { responseptr = newptr; numDereg++; rr->RequireGoodbye = mDNSfalse; }
else continue; }
SetNewRData(&rr->resrec, rr->NewRData, rr->newrdlength);
}
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
rr->resrec.rrclass |= kDNSClass_UniqueRRSet; newptr = PutRR_OS_TTL(responseptr, &m->omsg.h.numAnswers, &rr->resrec, active ? rr->resrec.rroriginalttl : 0);
rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; if (newptr)
{
responseptr = newptr;
rr->RequireGoodbye = active;
if (rr->resrec.RecordType == kDNSRecordTypeDeregistering) numDereg++;
else if (rr->LastAPTime == m->timenow) numAnnounce++; else numAnswer++;
}
if (rr->NewRData && active)
SetNewRData(&rr->resrec, OldRData, oldrdlength);
if (!pktcount && active && (rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && !rr->SendNSECNow)
rr->SendNSECNow = mDNSInterfaceMark;
if (newptr) {
if (rr->ImmedAnswer == mDNSInterfaceMark && rr->resrec.InterfaceID == mDNSInterface_Any)
rr->SendRNow = GetNextActiveInterfaceID(intf);
else
rr->SendRNow = mDNSNULL;
}
}
}
newptr = responseptr;
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->ImmedAdditional == intf->InterfaceID)
if (ResourceRecordIsValidAnswer(rr))
{
mDNSBool SendAdditional = (m->omsg.h.numAnswers > 0);
if (!SendAdditional && (rr->resrec.RecordType & kDNSRecordTypeUniqueMask))
{
const AuthRecord *a;
for (a = m->ResourceRecords; a; a=a->next)
if (a->LastMCTime == m->timenow &&
a->LastMCInterface == intf->InterfaceID &&
SameResourceRecordSignature(a, rr)) { SendAdditional = mDNStrue; break; }
}
if (!SendAdditional) rr->ImmedAdditional = mDNSNULL; else if (newptr) {
if (!pktcount && (rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && !rr->SendNSECNow)
rr->SendNSECNow = mDNSInterfaceMark;
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
rr->resrec.rrclass |= kDNSClass_UniqueRRSet; newptr = PutRR_OS(newptr, &m->omsg.h.numAdditionals, &rr->resrec);
rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; if (newptr)
{
responseptr = newptr;
rr->ImmedAdditional = mDNSNULL;
rr->RequireGoodbye = mDNStrue;
rr->LastMCTime = m->timenow;
rr->LastMCInterface = intf->InterfaceID;
}
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->SendNSECNow == mDNSInterfaceMark || rr->SendNSECNow == intf->InterfaceID)
{
AuthRecord nsec;
mDNS_SetupResourceRecord(&nsec, mDNSNULL, mDNSInterface_Any, kDNSType_NSEC, rr->resrec.rroriginalttl, kDNSRecordTypeUnique, mDNSNULL, mDNSNULL);
nsec.resrec.rrclass |= kDNSClass_UniqueRRSet;
AssignDomainName(&nsec.namestorage, rr->resrec.name);
mDNSPlatformMemZero(nsec.rdatastorage.u.nsec.bitmap, sizeof(nsec.rdatastorage.u.nsec.bitmap));
for (r2 = m->ResourceRecords; r2; r2=r2->next)
if (ResourceRecordIsValidAnswer(r2) && SameResourceRecordNameClassInterface(r2, rr))
{
if (r2->resrec.rrtype >= kDNSQType_ANY) { LogMsg("Can't create NSEC for record %s", ARDisplayString(m, r2)); break; }
else nsec.rdatastorage.u.nsec.bitmap[r2->resrec.rrtype >> 3] |= 128 >> (r2->resrec.rrtype & 7);
}
newptr = responseptr;
if (!r2) {
newptr = PutRR_OS(responseptr, &m->omsg.h.numAdditionals, &nsec.resrec);
if (newptr) responseptr = newptr;
}
if (newptr || rr->SendNSECNow == mDNSInterfaceMark)
{
rr->SendNSECNow = mDNSNULL;
for (r2 = rr->next; r2; r2=r2->next)
if (SameResourceRecordNameClassInterface(r2, rr))
if (r2->SendNSECNow == mDNSInterfaceMark || r2->SendNSECNow == intf->InterfaceID)
r2->SendNSECNow = mDNSNULL;
}
}
if (m->omsg.h.numAnswers || m->omsg.h.numAdditionals)
{
if (OwnerRecordSpace)
{
AuthRecord opt;
mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, mDNSNULL, mDNSNULL);
opt.resrec.rrclass = NormalMaxDNSMessageData;
opt.resrec.rdlength = sizeof(rdataOPT); opt.resrec.rdestimate = sizeof(rdataOPT);
SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
newptr = PutResourceRecord(&m->omsg, responseptr, &m->omsg.h.numAdditionals, &opt.resrec);
if (newptr) { responseptr = newptr; LogSPS("SendResponses put %s", ARDisplayString(m, &opt)); }
else if (m->omsg.h.numAnswers + m->omsg.h.numAuthorities + m->omsg.h.numAdditionals == 1)
LogSPS("SendResponses: No space in packet for Owner OPT record (%d/%d/%d/%d) %s",
m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
else
LogMsg("SendResponses: How did we fail to have space for Owner OPT record (%d/%d/%d/%d) %s",
m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
}
debugf("SendResponses: Sending %d Deregistration%s, %d Announcement%s, %d Answer%s, %d Additional%s on %p",
numDereg, numDereg == 1 ? "" : "s",
numAnnounce, numAnnounce == 1 ? "" : "s",
numAnswer, numAnswer == 1 ? "" : "s",
m->omsg.h.numAdditionals, m->omsg.h.numAdditionals == 1 ? "" : "s", intf->InterfaceID);
if (intf->IPv4Available) mDNSSendDNSMessage(m, &m->omsg, responseptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v4, MulticastDNSPort, mDNSNULL, mDNSNULL);
if (intf->IPv6Available) mDNSSendDNSMessage(m, &m->omsg, responseptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v6, MulticastDNSPort, mDNSNULL, mDNSNULL);
if (!m->SuppressSending) m->SuppressSending = NonZeroTime(m->timenow + (mDNSPlatformOneSecond+9)/10);
if (++pktcount >= 1000) { LogMsg("SendResponses exceeded loop limit %d: giving up", pktcount); break; }
}
else {
const NetworkInterfaceInfo *next = GetFirstActiveInterface(intf->next);
#if MDNS_DEBUGMSGS && 0
const char *const msg = next ? "SendResponses: Nothing more on %p; moving to %p" : "SendResponses: Nothing more on %p";
debugf(msg, intf, next);
#endif
intf = next;
pktcount = 0; }
}
if (m->CurrentRecord)
LogMsg("SendResponses ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
rr = m->CurrentRecord;
m->CurrentRecord = rr->next;
if (rr->SendRNow)
{
if (rr->resrec.InterfaceID != mDNSInterface_LocalOnly && rr->resrec.InterfaceID != mDNSInterface_P2P)
LogMsg("SendResponses: No active interface %p to send: %p %02X %s", rr->SendRNow, rr->resrec.InterfaceID, rr->resrec.RecordType, ARDisplayString(m, rr));
rr->SendRNow = mDNSNULL;
}
if (rr->ImmedAnswer || rr->resrec.RecordType == kDNSRecordTypeDeregistering)
{
if (rr->NewRData) CompleteRDataUpdate(m, rr);
if (rr->resrec.RecordType == kDNSRecordTypeDeregistering && rr->AnnounceCount == 0)
{
if (!AuthRecord_uDNS(rr)) CompleteDeregistration(m, rr); }
else
{
rr->ImmedAnswer = mDNSNULL;
rr->ImmedUnicast = mDNSfalse;
rr->v4Requester = zerov4Addr;
rr->v6Requester = zerov6Addr;
}
}
}
verbosedebugf("SendResponses: Next in %ld ticks", m->NextScheduledResponse - m->timenow);
}
#define CacheCheckGracePeriod(RR) ( \
((RR)->CRActiveQuestion == mDNSNULL ) ? (60 * mDNSPlatformOneSecond) : \
((RR)->UnansweredQueries < MaxUnansweredQueries) ? (TicksTTL(rr)/50) : \
((RR)->resrec.rroriginalttl > 10 ) ? (mDNSPlatformOneSecond) : \
((RR)->resrec.rroriginalttl > 0 ) ? (mDNSPlatformOneSecond/10) : 0)
#define NextCacheCheckEvent(RR) ((RR)->NextRequiredQuery + CacheCheckGracePeriod(RR))
mDNSexport void ScheduleNextCacheCheckTime(mDNS *const m, const mDNSu32 slot, const mDNSs32 event)
{
if (m->rrcache_nextcheck[slot] - event > 0)
m->rrcache_nextcheck[slot] = event;
if (m->NextCacheCheck - event > 0)
m->NextCacheCheck = event;
}
mDNSlocal void SetNextCacheCheckTimeForRecord(mDNS *const m, CacheRecord *const rr)
{
rr->NextRequiredQuery = RRExpireTime(rr);
if (rr->CRActiveQuestion && rr->UnansweredQueries < MaxUnansweredQueries)
{
rr->NextRequiredQuery -= TicksTTL(rr)/20 * (MaxUnansweredQueries - rr->UnansweredQueries);
rr->NextRequiredQuery += mDNSRandom((mDNSu32)TicksTTL(rr)/50);
verbosedebugf("SetNextCacheCheckTimeForRecord: NextRequiredQuery in %ld sec CacheCheckGracePeriod %d ticks for %s",
(rr->NextRequiredQuery - m->timenow) / mDNSPlatformOneSecond, CacheCheckGracePeriod(rr), CRDisplayString(m,rr));
}
ScheduleNextCacheCheckTime(m, HashSlot(rr->resrec.name), NextCacheCheckEvent(rr));
}
#define kMinimumReconfirmTime ((mDNSu32)mDNSPlatformOneSecond * 5)
#define kDefaultReconfirmTimeForWake ((mDNSu32)mDNSPlatformOneSecond * 5)
#define kDefaultReconfirmTimeForNoAnswer ((mDNSu32)mDNSPlatformOneSecond * 5)
#define kDefaultReconfirmTimeForFlappingInterface ((mDNSu32)mDNSPlatformOneSecond * 30)
mDNSlocal mStatus mDNS_Reconfirm_internal(mDNS *const m, CacheRecord *const rr, mDNSu32 interval)
{
if (interval < kMinimumReconfirmTime)
interval = kMinimumReconfirmTime;
if (interval > 0x10000000) interval = 0x10000000;
if (RRExpireTime(rr) - m->timenow > (mDNSs32)((interval * 4) / 3))
{
if (!m->RandomReconfirmDelay) m->RandomReconfirmDelay = 1 + mDNSRandom(0x3FFFFFFF);
interval += m->RandomReconfirmDelay % ((interval/3) + 1);
rr->TimeRcvd = m->timenow - (mDNSs32)interval * 3;
rr->resrec.rroriginalttl = (interval * 4 + mDNSPlatformOneSecond - 1) / mDNSPlatformOneSecond;
SetNextCacheCheckTimeForRecord(m, rr);
}
debugf("mDNS_Reconfirm_internal:%6ld ticks to go for %s %p",
RRExpireTime(rr) - m->timenow, CRDisplayString(m, rr), rr->CRActiveQuestion);
return(mStatus_NoError);
}
#define MaxQuestionInterval (3600 * mDNSPlatformOneSecond)
mDNSlocal mDNSBool BuildQuestion(mDNS *const m, DNSMessage *query, mDNSu8 **queryptr, DNSQuestion *q,
CacheRecord ***kalistptrptr, mDNSu32 *answerforecast)
{
mDNSBool ucast = (q->LargeAnswers || q->RequestUnicast) && m->CanReceiveUnicastOn5353;
mDNSu16 ucbit = (mDNSu16)(ucast ? kDNSQClass_UnicastResponse : 0);
const mDNSu8 *const limit = query->data + NormalMaxDNSMessageData;
mDNSu8 *newptr = putQuestion(query, *queryptr, limit - *answerforecast, &q->qname, q->qtype, (mDNSu16)(q->qclass | ucbit));
if (!newptr)
{
debugf("BuildQuestion: No more space in this packet for question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
return(mDNSfalse);
}
else
{
mDNSu32 forecast = *answerforecast;
const mDNSu32 slot = HashSlot(&q->qname);
const CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
CacheRecord *rr;
CacheRecord **ka = *kalistptrptr;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) if (rr->resrec.InterfaceID == q->SendQNow && !(rr->resrec.RecordType & kDNSRecordTypeUniqueMask) && rr->NextInKAList == mDNSNULL && ka != &rr->NextInKAList && rr->resrec.rdlength <= SmallRecordLimit && SameNameRecordAnswersQuestion(&rr->resrec, q) && rr->TimeRcvd + TicksTTL(rr)/2 - m->timenow > mDNSPlatformOneSecond) {
*ka = rr; ka = &rr->NextInKAList;
forecast += 12 + rr->resrec.rdestimate;
if (query->h.numQuestions > 1 && newptr + forecast >= limit)
{
debugf("BuildQuestion: Retracting question %##s (%s) new forecast total %d",
q->qname.c, DNSTypeName(q->qtype), newptr + forecast - query->data);
query->h.numQuestions--;
ka = *kalistptrptr; while (*ka) { CacheRecord *c = *ka; *ka = mDNSNULL; ka = &c->NextInKAList; }
return(mDNSfalse); }
}
*queryptr = newptr; *answerforecast = forecast; *kalistptrptr = ka; if (ucast) q->ExpectUnicastResp = NonZeroTime(m->timenow);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) if (rr->resrec.InterfaceID == q->SendQNow && rr->NextInKAList == mDNSNULL && ka != &rr->NextInKAList && SameNameRecordAnswersQuestion(&rr->resrec, q)) {
rr->UnansweredQueries++; rr->LastUnansweredTime = m->timenow;
SetNextCacheCheckTimeForRecord(m, rr);
}
return(mDNStrue);
}
}
mDNSlocal void ReconfirmAntecedents(mDNS *const m, const domainname *const name, const mDNSu32 namehash, const int depth)
{
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *cr;
debugf("ReconfirmAntecedents (depth=%d) for %##s", depth, name->c);
FORALL_CACHERECORDS(slot, cg, cr)
{
domainname *crtarget = GetRRDomainNameTarget(&cr->resrec);
if (crtarget && cr->resrec.rdatahash == namehash && SameDomainName(crtarget, name))
{
LogInfo("ReconfirmAntecedents: Reconfirming (depth=%d) %s", depth, CRDisplayString(m, cr));
mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
if (depth < 5) ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, depth+1);
}
}
}
mDNSlocal const CacheRecord *CacheHasAddressTypeForName(mDNS *const m, const domainname *const name, const mDNSu32 namehash)
{
CacheGroup *const cg = CacheGroupForName(m, HashSlot(name), namehash, name);
const CacheRecord *cr = cg ? cg->members : mDNSNULL;
while (cr && !RRTypeIsAddressType(cr->resrec.rrtype)) cr=cr->next;
return(cr);
}
mDNSlocal const CacheRecord *FindSPSInCache1(mDNS *const m, const DNSQuestion *const q, const CacheRecord *const c0, const CacheRecord *const c1)
{
CacheGroup *const cg = CacheGroupForName(m, HashSlot(&q->qname), q->qnamehash, &q->qname);
const CacheRecord *cr, *bestcr = mDNSNULL;
mDNSu32 bestmetric = 1000000;
for (cr = cg ? cg->members : mDNSNULL; cr; cr=cr->next)
if (cr->resrec.rrtype == kDNSType_PTR && cr->resrec.rdlength >= 6) if (cr != c0 && cr != c1) if (SameNameRecordAnswersQuestion(&cr->resrec, q)) if (!IdenticalSameNameRecord(&cr->resrec, &m->SPSRecords.RR_PTR.resrec)) {
mDNSu32 metric = SPSMetric(cr->resrec.rdata->u.name.c);
if (bestmetric > metric) { bestmetric = metric; bestcr = cr; }
}
return(bestcr);
}
mDNSexport void FindSPSInCache(mDNS *const m, const DNSQuestion *const q, const CacheRecord *sps[3])
{
sps[0] = FindSPSInCache1(m, q, mDNSNULL, mDNSNULL);
sps[1] = !sps[0] ? mDNSNULL : FindSPSInCache1(m, q, sps[0], mDNSNULL);
sps[2] = !sps[1] ? mDNSNULL : FindSPSInCache1(m, q, sps[0], sps[1]);
}
mDNSlocal void ExpireDupSuppressInfo(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 time)
{
int i;
for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].Time - time < 0) ds[i].InterfaceID = mDNSNULL;
}
mDNSlocal void ExpireDupSuppressInfoOnInterface(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 time, mDNSInterfaceID InterfaceID)
{
int i;
for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].InterfaceID == InterfaceID && ds[i].Time - time < 0) ds[i].InterfaceID = mDNSNULL;
}
mDNSlocal mDNSBool SuppressOnThisInterface(const DupSuppressInfo ds[DupSuppressInfoSize], const NetworkInterfaceInfo * const intf)
{
int i;
mDNSBool v4 = !intf->IPv4Available; mDNSBool v6 = !intf->IPv6Available; for (i=0; i<DupSuppressInfoSize; i++)
if (ds[i].InterfaceID == intf->InterfaceID)
{
if (ds[i].Type == mDNSAddrType_IPv4) v4 = mDNStrue;
else if (ds[i].Type == mDNSAddrType_IPv6) v6 = mDNStrue;
if (v4 && v6) return(mDNStrue);
}
return(mDNSfalse);
}
mDNSlocal int RecordDupSuppressInfo(DupSuppressInfo ds[DupSuppressInfoSize], mDNSs32 Time, mDNSInterfaceID InterfaceID, mDNSs32 Type)
{
int i, j;
for (i=0; i<DupSuppressInfoSize; i++) if (ds[i].InterfaceID == InterfaceID && ds[i].Type == Type) break;
if (i >= DupSuppressInfoSize)
{
i = 0;
for (j=1; j<DupSuppressInfoSize && ds[i].InterfaceID; j++)
if (!ds[j].InterfaceID || ds[j].Time - ds[i].Time < 0)
i = j;
}
ds[i].Time = Time;
ds[i].InterfaceID = InterfaceID;
ds[i].Type = Type;
return(i);
}
mDNSlocal mDNSBool AccelerateThisQuery(mDNS *const m, DNSQuestion *q)
{
if (TimeToSendThisQuestion(q, m->timenow + q->ThisQInterval/10))
return(mDNStrue);
if (TimeToSendThisQuestion(q, m->timenow + q->ThisQInterval/2))
{
mDNSu32 forecast = (mDNSu32)DomainNameLength(&q->qname) + 4;
const mDNSu32 slot = HashSlot(&q->qname);
const CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
const CacheRecord *rr;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next) if (rr->resrec.rdlength <= SmallRecordLimit && SameNameRecordAnswersQuestion(&rr->resrec, q) && rr->TimeRcvd + TicksTTL(rr)/2 - m->timenow >= 0 && rr->NextRequiredQuery - (m->timenow + q->ThisQInterval) > 0) {
forecast += 12 + rr->resrec.rdestimate;
if (forecast >= 512) return(mDNSfalse); }
return(mDNStrue);
}
return(mDNSfalse);
}
mDNSlocal void SendQueries(mDNS *const m)
{
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *cr;
AuthRecord *ar;
int pktcount = 0;
DNSQuestion *q;
mDNSs32 maxExistingQuestionInterval = 0;
const NetworkInterfaceInfo *intf = GetFirstActiveInterface(m->HostInterfaces);
CacheRecord *KnownAnswerList = mDNSNULL;
FORALL_CACHERECORDS(slot, cg, cr)
if (cr->CRActiveQuestion && cr->UnansweredQueries < MaxUnansweredQueries)
if (m->timenow + TicksTTL(cr)/50 - cr->NextRequiredQuery >= 0)
{
debugf("Sending %d%% cache expiration query for %s", 80 + 5 * cr->UnansweredQueries, CRDisplayString(m, cr));
q = cr->CRActiveQuestion;
ExpireDupSuppressInfoOnInterface(q->DupSuppress, m->timenow - TicksTTL(cr)/20, cr->resrec.InterfaceID);
if (q->Target.type) q->SendQNow = mDNSInterfaceMark; else if (!mDNSOpaque16IsZero(q->TargetQID)) { q->LastQTime = m->timenow - q->ThisQInterval; cr->UnansweredQueries++; }
else if (q->SendQNow == mDNSNULL) q->SendQNow = cr->resrec.InterfaceID;
else if (q->SendQNow != cr->resrec.InterfaceID) q->SendQNow = mDNSInterfaceMark;
}
if (m->CurrentQuestion)
LogMsg("SendQueries ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
{
q = m->CurrentQuestion;
if (q->Target.type && (q->SendQNow || TimeToSendThisQuestion(q, m->timenow)))
{
mDNSu8 *qptr = m->omsg.data;
const mDNSu8 *const limit = m->omsg.data + sizeof(m->omsg.data);
if (!q->LocalSocket) q->LocalSocket = mDNSPlatformUDPSocket(m, zeroIPPort);
if (q->LocalSocket)
{
InitializeDNSMessage(&m->omsg.h, q->TargetQID, QueryFlags);
qptr = putQuestion(&m->omsg, qptr, limit, &q->qname, q->qtype, q->qclass);
mDNSSendDNSMessage(m, &m->omsg, qptr, mDNSInterface_Any, q->LocalSocket, &q->Target, q->TargetPort, mDNSNULL, mDNSNULL);
q->ThisQInterval *= QuestionIntervalStep;
}
if (q->ThisQInterval > MaxQuestionInterval)
q->ThisQInterval = MaxQuestionInterval;
q->LastQTime = m->timenow;
q->LastQTxTime = m->timenow;
q->RecentAnswerPkts = 0;
q->SendQNow = mDNSNULL;
q->ExpectUnicastResp = NonZeroTime(m->timenow);
}
else if (mDNSOpaque16IsZero(q->TargetQID) && !q->Target.type && TimeToSendThisQuestion(q, m->timenow))
{
q->SendQNow = mDNSInterfaceMark; if (maxExistingQuestionInterval < q->ThisQInterval)
maxExistingQuestionInterval = q->ThisQInterval;
}
if (q == m->CurrentQuestion) m->CurrentQuestion = m->CurrentQuestion->next;
}
while (m->CurrentQuestion)
{
LogInfo("SendQueries question loop 1: Skipping NewQuestion %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->CurrentQuestion->next;
}
m->CurrentQuestion = mDNSNULL;
m->NextScheduledQuery = m->timenow + 0x78000000;
for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
{
if (mDNSOpaque16IsZero(q->TargetQID) && (q->SendQNow ||
(!q->Target.type && ActiveQuestion(q) && q->ThisQInterval <= maxExistingQuestionInterval && AccelerateThisQuery(m,q))))
{
if (m->timenow - (q->LastQTime + (q->ThisQInterval/2)) >= 0)
{
q->SendQNow = mDNSInterfaceMark; debugf("SendQueries: %##s (%s) next interval %d seconds RequestUnicast = %d",
q->qname.c, DNSTypeName(q->qtype), q->ThisQInterval / InitialQuestionInterval, q->RequestUnicast);
q->ThisQInterval *= QuestionIntervalStep;
if (q->ThisQInterval > MaxQuestionInterval)
q->ThisQInterval = MaxQuestionInterval;
else if (q->CurrentAnswers == 0 && q->ThisQInterval == InitialQuestionInterval * QuestionIntervalStep3 && !q->RequestUnicast &&
!(RRTypeIsAddressType(q->qtype) && CacheHasAddressTypeForName(m, &q->qname, q->qnamehash)))
{
debugf("SendQueries: Zero current answers for %##s (%s); will reconfirm antecedents",
q->qname.c, DNSTypeName(q->qtype));
ReconfirmAntecedents(m, &q->qname, q->qnamehash, 0);
}
}
q->SendOnAll = (q->SendQNow == mDNSInterfaceMark);
if (q->SendOnAll)
{
q->SendQNow = !intf ? mDNSNULL : (q->InterfaceID) ? q->InterfaceID : intf->InterfaceID;
q->LastQTime = m->timenow;
}
ExpireDupSuppressInfo(q->DupSuppress, m->timenow - q->ThisQInterval/2);
q->LastQTxTime = m->timenow;
q->RecentAnswerPkts = 0;
if (q->RequestUnicast) q->RequestUnicast--;
}
SetNextQueryTime(m,q);
}
m->NextScheduledProbe = m->timenow + 0x78000000;
if (m->CurrentRecord)
LogMsg("SendQueries ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
ar = m->CurrentRecord;
m->CurrentRecord = ar->next;
if (!AuthRecord_uDNS(ar) && ar->resrec.RecordType == kDNSRecordTypeUnique) {
if (m->timenow - (ar->LastAPTime + ar->ThisAPInterval) < 0)
{
SetNextAnnounceProbeTime(m, ar);
}
else if (ar->ProbeCount)
{
if (ar->AddressProxy.type == mDNSAddrType_IPv4)
{
LogSPS("SendQueries ARP Probe %d %s %s", ar->ProbeCount, InterfaceNameForID(m, ar->resrec.InterfaceID), ARDisplayString(m,ar));
SendARP(m, 1, ar, &zerov4Addr, &zeroEthAddr, &ar->AddressProxy.ip.v4, &ar->WakeUp.IMAC);
}
else if (ar->AddressProxy.type == mDNSAddrType_IPv6)
{
LogSPS("SendQueries NDP Probe %d %s %s", ar->ProbeCount, InterfaceNameForID(m, ar->resrec.InterfaceID), ARDisplayString(m,ar));
SendNDP(m, NDP_Sol, 0, ar, &zerov6Addr, mDNSNULL, &ar->AddressProxy.ip.v6, &ar->WakeUp.IMAC);
}
ar->SendRNow = (!intf || ar->WakeUp.HMAC.l[0]) ? mDNSNULL : ar->resrec.InterfaceID ? ar->resrec.InterfaceID : intf->InterfaceID;
ar->LastAPTime = m->timenow;
if (ar->ProbeCount > DefaultProbeCountForTypeUnique)
ar->ProbeCount = DefaultProbeCountForTypeUnique;
ar->ProbeCount--;
SetNextAnnounceProbeTime(m, ar);
if (ar->ProbeCount == 0)
{
AuthRecord *r2;
for (r2 = m->DuplicateRecords; r2; r2=r2->next)
if (r2->resrec.RecordType == kDNSRecordTypeUnique && RecordIsLocalDuplicate(r2, ar))
r2->ProbeCount = 0;
if (!ar->Acknowledged) AcknowledgeRecord(m, ar);
}
}
else
{
if (!ar->Acknowledged) AcknowledgeRecord(m, ar); ar->resrec.RecordType = kDNSRecordTypeVerified;
ar->ThisAPInterval = DefaultAnnounceIntervalForTypeUnique;
ar->LastAPTime = m->timenow - DefaultAnnounceIntervalForTypeUnique;
SetNextAnnounceProbeTime(m, ar);
}
}
}
m->CurrentRecord = m->DuplicateRecords;
while (m->CurrentRecord)
{
ar = m->CurrentRecord;
m->CurrentRecord = ar->next;
if (ar->resrec.RecordType == kDNSRecordTypeUnique && ar->ProbeCount == 0 && !ar->Acknowledged)
AcknowledgeRecord(m, ar);
}
while (intf)
{
const int OwnerRecordSpace = (m->AnnounceOwner && intf->MAC.l[0]) ? DNSOpt_Header_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC) : 0;
mDNSu8 *queryptr = m->omsg.data;
InitializeDNSMessage(&m->omsg.h, zeroID, QueryFlags);
if (KnownAnswerList) verbosedebugf("SendQueries: KnownAnswerList set... Will continue from previous packet");
if (!KnownAnswerList)
{
CacheRecord **kalistptr = &KnownAnswerList;
mDNSu32 answerforecast = OwnerRecordSpace;
for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
{
if (mDNSOpaque16IsZero(q->TargetQID) && (q->SendQNow == intf->InterfaceID))
{
debugf("SendQueries: %s question for %##s (%s) at %d forecast total %d",
SuppressOnThisInterface(q->DupSuppress, intf) ? "Suppressing" : "Putting ",
q->qname.c, DNSTypeName(q->qtype), queryptr - m->omsg.data, queryptr + answerforecast - m->omsg.data);
if (SuppressOnThisInterface(q->DupSuppress, intf) ||
BuildQuestion(m, &m->omsg, &queryptr, q, &kalistptr, &answerforecast))
q->SendQNow = (q->InterfaceID || !q->SendOnAll) ? mDNSNULL : GetNextActiveInterfaceID(intf);
}
}
for (ar = m->ResourceRecords; ar; ar=ar->next)
if (ar->SendRNow == intf->InterfaceID)
{
mDNSBool ucast = (ar->ProbeCount >= DefaultProbeCountForTypeUnique-1) && m->CanReceiveUnicastOn5353;
mDNSu16 ucbit = (mDNSu16)(ucast ? kDNSQClass_UnicastResponse : 0);
const mDNSu8 *const limit = m->omsg.data + (m->omsg.h.numQuestions ? NormalMaxDNSMessageData : AbsoluteMaxDNSMessageData);
mDNSu32 forecast = answerforecast + 12 + ar->resrec.rdestimate;
mDNSu8 *newptr = putQuestion(&m->omsg, queryptr, limit - forecast, ar->resrec.name, kDNSQType_ANY, (mDNSu16)(ar->resrec.rrclass | ucbit));
if (newptr)
{
queryptr = newptr;
answerforecast = forecast;
ar->SendRNow = (ar->resrec.InterfaceID) ? mDNSNULL : GetNextActiveInterfaceID(intf);
ar->IncludeInProbe = mDNStrue;
verbosedebugf("SendQueries: Put Question %##s (%s) probecount %d",
ar->resrec.name->c, DNSTypeName(ar->resrec.rrtype), ar->ProbeCount);
}
}
}
while (KnownAnswerList)
{
CacheRecord *ka = KnownAnswerList;
mDNSu32 SecsSinceRcvd = ((mDNSu32)(m->timenow - ka->TimeRcvd)) / mDNSPlatformOneSecond;
mDNSu8 *newptr = PutResourceRecordTTLWithLimit(&m->omsg, queryptr, &m->omsg.h.numAnswers,
&ka->resrec, ka->resrec.rroriginalttl - SecsSinceRcvd, m->omsg.data + NormalMaxDNSMessageData - OwnerRecordSpace);
if (newptr)
{
verbosedebugf("SendQueries: Put %##s (%s) at %d - %d",
ka->resrec.name->c, DNSTypeName(ka->resrec.rrtype), queryptr - m->omsg.data, newptr - m->omsg.data);
queryptr = newptr;
KnownAnswerList = ka->NextInKAList;
ka->NextInKAList = mDNSNULL;
}
else
{
if (m->omsg.h.numQuestions > 1)
LogMsg("SendQueries: Put %d answers; No more space for known answers", m->omsg.h.numAnswers);
m->omsg.h.flags.b[0] |= kDNSFlag0_TC;
break;
}
}
for (ar = m->ResourceRecords; ar; ar=ar->next)
if (ar->IncludeInProbe)
{
mDNSu8 *newptr = PutResourceRecord(&m->omsg, queryptr, &m->omsg.h.numAuthorities, &ar->resrec);
ar->IncludeInProbe = mDNSfalse;
if (newptr) queryptr = newptr;
else LogMsg("SendQueries: How did we fail to have space for the Update record %s", ARDisplayString(m,ar));
}
if (queryptr > m->omsg.data)
{
if (OwnerRecordSpace)
{
AuthRecord opt;
mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, mDNSNULL, mDNSNULL);
opt.resrec.rrclass = NormalMaxDNSMessageData;
opt.resrec.rdlength = sizeof(rdataOPT); opt.resrec.rdestimate = sizeof(rdataOPT);
SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[0]);
LogSPS("SendQueries putting %s", ARDisplayString(m, &opt));
queryptr = PutResourceRecordTTLWithLimit(&m->omsg, queryptr, &m->omsg.h.numAdditionals,
&opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
if (!queryptr)
LogMsg("SendQueries: How did we fail to have space for the OPT record (%d/%d/%d/%d) %s",
m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
if (queryptr > m->omsg.data + NormalMaxDNSMessageData)
if (m->omsg.h.numQuestions != 1 || m->omsg.h.numAnswers != 0 || m->omsg.h.numAuthorities != 1 || m->omsg.h.numAdditionals != 1)
LogMsg("SendQueries: Why did we generate oversized packet with OPT record %p %p %p (%d/%d/%d/%d) %s",
m->omsg.data, m->omsg.data + NormalMaxDNSMessageData, queryptr,
m->omsg.h.numQuestions, m->omsg.h.numAnswers, m->omsg.h.numAuthorities, m->omsg.h.numAdditionals, ARDisplayString(m, &opt));
}
if ((m->omsg.h.flags.b[0] & kDNSFlag0_TC) && m->omsg.h.numQuestions > 1)
LogMsg("SendQueries: Should not have more than one question (%d) in a truncated packet", m->omsg.h.numQuestions);
debugf("SendQueries: Sending %d Question%s %d Answer%s %d Update%s on %p",
m->omsg.h.numQuestions, m->omsg.h.numQuestions == 1 ? "" : "s",
m->omsg.h.numAnswers, m->omsg.h.numAnswers == 1 ? "" : "s",
m->omsg.h.numAuthorities, m->omsg.h.numAuthorities == 1 ? "" : "s", intf->InterfaceID);
if (intf->IPv4Available) mDNSSendDNSMessage(m, &m->omsg, queryptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v4, MulticastDNSPort, mDNSNULL, mDNSNULL);
if (intf->IPv6Available) mDNSSendDNSMessage(m, &m->omsg, queryptr, intf->InterfaceID, mDNSNULL, &AllDNSLinkGroup_v6, MulticastDNSPort, mDNSNULL, mDNSNULL);
if (!m->SuppressSending) m->SuppressSending = NonZeroTime(m->timenow + (mDNSPlatformOneSecond+9)/10);
if (++pktcount >= 1000)
{ LogMsg("SendQueries exceeded loop limit %d: giving up", pktcount); break; }
}
else {
const NetworkInterfaceInfo *next = GetFirstActiveInterface(intf->next);
#if MDNS_DEBUGMSGS && 0
const char *const msg = next ? "SendQueries: Nothing more on %p; moving to %p" : "SendQueries: Nothing more on %p";
debugf(msg, intf, next);
#endif
intf = next;
}
}
for (ar = m->ResourceRecords; ar; ar=ar->next)
if (ar->SendRNow)
{
if (ar->resrec.InterfaceID != mDNSInterface_LocalOnly && ar->resrec.InterfaceID != mDNSInterface_P2P)
LogMsg("SendQueries: No active interface %p to send probe: %p %s", ar->SendRNow, ar->resrec.InterfaceID, ARDisplayString(m, ar));
ar->SendRNow = mDNSNULL;
}
FORALL_CACHERECORDS(slot, cg, cr)
if (cr->CRActiveQuestion && cr->UnansweredQueries < MaxUnansweredQueries)
if (m->timenow + TicksTTL(cr)/50 - cr->NextRequiredQuery >= 0)
{
cr->UnansweredQueries++;
cr->CRActiveQuestion->SendQNow = mDNSNULL;
SetNextCacheCheckTimeForRecord(m, cr);
}
for (q = m->Questions; q; q=q->next)
if (q->SendQNow)
{
DNSQuestion *x;
for (x = m->NewQuestions; x; x=x->next) if (x == q) break; LogMsg("SendQueries: No active interface %p to send %s question: %p %##s (%s)", q->SendQNow, x ? "new" : "old", q->InterfaceID, q->qname.c, DNSTypeName(q->qtype));
q->SendQNow = mDNSNULL;
}
}
mDNSlocal void SendWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *EthAddr, mDNSOpaque48 *password)
{
int i, j;
mDNSu8 *ptr = m->omsg.data;
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
if (!intf) { LogMsg("SendARP: No interface with InterfaceID %p found", InterfaceID); return; }
for (i=0; i<6; i++) *ptr++ = EthAddr->b[i];
for (i=0; i<6; i++) *ptr++ = intf->MAC.b[0];
*ptr++ = 0x08;
*ptr++ = 0x42;
for (i=0; i<6; i++) *ptr++ = 0xFF;
for (j=0; j<16; j++) for (i=0; i<6; i++) *ptr++ = EthAddr->b[i];
for (i=0; i<6; i++) *ptr++ = password->b[i];
mDNSPlatformSendRawPacket(m->omsg.data, ptr, InterfaceID);
for (i=0; i<6; i++) m->omsg.data[i] = 0xFF;
mDNSPlatformSendRawPacket(m->omsg.data, ptr, InterfaceID);
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - RR List Management & Task Management
#endif
mDNSexport void AnswerCurrentQuestionWithResourceRecord(mDNS *const m, CacheRecord *const rr, const QC_result AddRecord)
{
DNSQuestion *const q = m->CurrentQuestion;
mDNSBool followcname = rr->resrec.RecordType != kDNSRecordTypePacketNegative && AddRecord &&
rr->resrec.rrtype == kDNSType_CNAME && q->qtype != kDNSType_CNAME;
verbosedebugf("AnswerCurrentQuestionWithResourceRecord:%4lu %s TTL %d %s",
q->CurrentAnswers, AddRecord ? "Add" : "Rmv", rr->resrec.rroriginalttl, CRDisplayString(m, rr));
if (QuerySuppressed(q)) return;
rr->LastUsed = m->timenow;
if (AddRecord == QC_add && !q->DuplicateOf && rr->CRActiveQuestion != q)
{
if (!rr->CRActiveQuestion) m->rrcache_active++; debugf("AnswerCurrentQuestionWithResourceRecord: Updating CRActiveQuestion from %p to %p for cache record %s, CurrentAnswer %d",
rr->CRActiveQuestion, q, CRDisplayString(m,rr), q->CurrentAnswers);
rr->CRActiveQuestion = q; SetNextCacheCheckTimeForRecord(m, rr);
}
if ((AddRecord == QC_addnocache && !q->RequestUnicast) ||
(AddRecord == QC_add && (q->ExpectUnique || (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask))))
if (ActiveQuestion(q) && (mDNSOpaque16IsZero(q->TargetQID) || !q->LongLived))
{
q->LastQTime = m->timenow;
q->LastQTxTime = m->timenow;
q->RecentAnswerPkts = 0;
q->ThisQInterval = MaxQuestionInterval;
q->RequestUnicast = mDNSfalse;
debugf("AnswerCurrentQuestionWithResourceRecord: Set MaxQuestionInterval for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
}
if (rr->DelayDelivery) return;
if (rr->resrec.RecordType == kDNSRecordTypePacketNegative || (q->qtype != kDNSType_NSEC && RRAssertsNonexistence(&rr->resrec, q->qtype)))
if (!AddRecord || !q->ReturnIntermed) return;
if (q->QuestionCallback && !q->NoAnswer && (!followcname || q->ReturnIntermed))
{
mDNS_DropLockBeforeCallback(); if (q->qtype != kDNSType_NSEC && RRAssertsNonexistence(&rr->resrec, q->qtype))
{
CacheRecord neg;
MakeNegativeCacheRecord(m, &neg, &q->qname, q->qnamehash, q->qtype, q->qclass, 1, rr->resrec.InterfaceID, q->qDNSServer);
q->QuestionCallback(m, q, &neg.resrec, AddRecord);
}
else
q->QuestionCallback(m, q, &rr->resrec, AddRecord);
mDNS_ReclaimLockAfterCallback(); }
if (followcname && m->CurrentQuestion == q)
{
const mDNSBool selfref = SameDomainName(&q->qname, &rr->resrec.rdata->u.name);
if (q->CNAMEReferrals >= 10 || selfref)
LogMsg("AnswerCurrentQuestionWithResourceRecord: %p %##s (%s) NOT following CNAME referral %d%s for %s",
q, q->qname.c, DNSTypeName(q->qtype), q->CNAMEReferrals, selfref ? " (Self-Referential)" : "", CRDisplayString(m, rr));
else
{
const mDNSu32 c = q->CNAMEReferrals + 1;
LogInfo("AnswerCurrentQuestionWithResourceRecord: %p %##s (%s) following CNAME referral %d for %s",
q, q->qname.c, DNSTypeName(q->qtype), q->CNAMEReferrals, CRDisplayString(m, rr));
mDNS_StopQuery_internal(m, q); AssignDomainName(&q->qname, &rr->resrec.rdata->u.name); q->qnamehash = DomainNameHashValue(&q->qname); mDNS_StartQuery_internal(m, q); q->CNAMEReferrals = c;
}
}
}
mDNSlocal void AnswerQuestionsForDNSServerChanges(mDNS *const m)
{
DNSQuestion *q;
DNSQuestion *qnext;
CacheRecord *rr;
mDNSu32 slot;
CacheGroup *cg;
if (m->CurrentQuestion)
LogMsg("AnswerQuestionsForDNSServerChanges: ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
for (q = m->Questions; q && q != m->NewQuestions; q = qnext)
{
qnext = q->next;
if (mDNSOpaque16IsZero(q->TargetQID)) continue;
if (!q->deliverAddEvents) continue;
q->deliverAddEvents = mDNSfalse;
if (QuerySuppressed(q)) continue;
m->CurrentQuestion = q;
slot = HashSlot(&q->qname);
cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
{
if (SameNameRecordAnswersQuestion(&rr->resrec, q))
{
LogInfo("AnswerQuestionsForDNSServerChanges: Calling AnswerCurrentQuestionWithResourceRecord for question %p %##s using resource record %s",
q, q->qname.c, CRDisplayString(m, rr));
if (!q->qDNSServer && !q->DuplicateOf && rr->resrec.RecordType == kDNSRecordTypePacketNegative)
{
DNSQuestion *qptr;
SetValidDNSServers(m, q);
q->qDNSServer = GetServerForQuestion(m, q);
for (qptr = q->next ; qptr; qptr = qptr->next)
if (qptr->DuplicateOf == q) { qptr->validDNSServers = q->validDNSServers; qptr->qDNSServer = q->qDNSServer; }
}
q->CurrentAnswers++;
if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers++;
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers++;
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
if (m->CurrentQuestion != q) break; }
}
}
m->CurrentQuestion = mDNSNULL;
}
mDNSlocal void CacheRecordDeferredAdd(mDNS *const m, CacheRecord *rr)
{
rr->DelayDelivery = 0;
if (m->CurrentQuestion)
LogMsg("CacheRecordDeferredAdd ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
{
DNSQuestion *q = m->CurrentQuestion;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
if (m->CurrentQuestion == q) m->CurrentQuestion = q->next;
}
m->CurrentQuestion = mDNSNULL;
}
mDNSlocal mDNSs32 CheckForSoonToExpireRecords(mDNS *const m, const domainname *const name, const mDNSu32 namehash, const mDNSu32 slot)
{
const mDNSs32 threshhold = m->timenow + mDNSPlatformOneSecond; const mDNSs32 start = m->timenow - 0x10000000;
mDNSs32 delay = start;
CacheGroup *cg = CacheGroupForName(m, slot, namehash, name);
const CacheRecord *rr;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (threshhold - RRExpireTime(rr) >= 0) if (delay - RRExpireTime(rr) < 0) delay = RRExpireTime(rr);
if (delay - start > 0) return(NonZeroTime(delay));
else return(0);
}
mDNSlocal void CacheRecordAdd(mDNS *const m, CacheRecord *rr)
{
DNSQuestion *q;
for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
{
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
{
if (q->LastAnswerPktNum != m->PktNum)
{
q->LastAnswerPktNum = m->PktNum;
if (mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q) && ++q->RecentAnswerPkts >= 10 &&
q->ThisQInterval > InitialQuestionInterval * QuestionIntervalStep3 && m->timenow - q->LastQTxTime < mDNSPlatformOneSecond)
{
LogMsg("CacheRecordAdd: %##s (%s) got immediate answer burst (%d); restarting exponential backoff sequence (%d)",
q->qname.c, DNSTypeName(q->qtype), q->RecentAnswerPkts, q->ThisQInterval);
q->LastQTime = m->timenow - InitialQuestionInterval + (mDNSs32)mDNSRandom((mDNSu32)mDNSPlatformOneSecond*4);
q->ThisQInterval = InitialQuestionInterval;
SetNextQueryTime(m,q);
}
}
verbosedebugf("CacheRecordAdd %p %##s (%s) %lu %#a:%d question %p", rr, rr->resrec.name->c,
DNSTypeName(rr->resrec.rrtype), rr->resrec.rroriginalttl, rr->resrec.rDNSServer ?
&rr->resrec.rDNSServer->addr : mDNSNULL, mDNSVal16(rr->resrec.rDNSServer ?
rr->resrec.rDNSServer->port : zeroIPPort), q);
q->CurrentAnswers++;
q->unansweredQueries = 0;
if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers++;
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers++;
if (q->CurrentAnswers > 4000)
{
static int msgcount = 0;
if (msgcount++ < 10)
LogMsg("CacheRecordAdd: %##s (%s) has %d answers; shedding records to resist DOS attack",
q->qname.c, DNSTypeName(q->qtype), q->CurrentAnswers);
rr->resrec.rroriginalttl = 0;
rr->UnansweredQueries = MaxUnansweredQueries;
}
}
}
if (!rr->DelayDelivery)
{
if (m->CurrentQuestion)
LogMsg("CacheRecordAdd ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
{
q = m->CurrentQuestion;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
if (m->CurrentQuestion == q) m->CurrentQuestion = q->next;
}
m->CurrentQuestion = mDNSNULL;
}
SetNextCacheCheckTimeForRecord(m, rr);
}
mDNSlocal void NoCacheAnswer(mDNS *const m, CacheRecord *rr)
{
LogMsg("No cache space: Delivering non-cached result for %##s", m->rec.r.resrec.name->c);
if (m->CurrentQuestion)
LogMsg("NoCacheAnswer ERROR m->CurrentQuestion already set: %##s (%s)", m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion)
{
DNSQuestion *q = m->CurrentQuestion;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_addnocache); if (m->CurrentQuestion == q) m->CurrentQuestion = q->next;
}
m->CurrentQuestion = mDNSNULL;
}
mDNSlocal void CacheRecordRmv(mDNS *const m, CacheRecord *rr)
{
if (m->CurrentQuestion)
LogMsg("CacheRecordRmv ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion && m->CurrentQuestion != m->NewQuestions)
{
DNSQuestion *q = m->CurrentQuestion;
if (!QuerySuppressed(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
{
verbosedebugf("CacheRecordRmv %p %s", rr, CRDisplayString(m, rr));
q->FlappingInterface1 = mDNSNULL;
q->FlappingInterface2 = mDNSNULL;
if (q->deliverAddEvents && !q->CurrentAnswers)
{
LogInfo("CacheRecordRmv: Question %p %##s (%s) deliverAddEvents set, DNSServer %#a:%d",
q, q->qname.c, DNSTypeName(q->qtype), q->qDNSServer ? &q->qDNSServer->addr : mDNSNULL,
mDNSVal16(q->qDNSServer ? q->qDNSServer->port : zeroIPPort));
m->CurrentQuestion = q->next;
continue;
}
if (q->CurrentAnswers == 0)
LogMsg("CacheRecordRmv ERROR!!: How can CurrentAnswers already be zero for %p %##s (%s) DNSServer %#a:%d",
q, q->qname.c, DNSTypeName(q->qtype), q->qDNSServer ? &q->qDNSServer->addr : mDNSNULL,
mDNSVal16(q->qDNSServer ? q->qDNSServer->port : zeroIPPort));
else
{
q->CurrentAnswers--;
if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers--;
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers--;
}
if (rr->resrec.rdata->MaxRDLength) {
if (q->CurrentAnswers == 0)
{
LogInfo("CacheRecordRmv: Last answer for %##s (%s) expired from cache; will reconfirm antecedents",
q->qname.c, DNSTypeName(q->qtype));
ReconfirmAntecedents(m, &q->qname, q->qnamehash, 0);
}
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_rmv);
}
}
if (m->CurrentQuestion == q) m->CurrentQuestion = q->next;
}
m->CurrentQuestion = mDNSNULL;
}
mDNSlocal void ReleaseCacheEntity(mDNS *const m, CacheEntity *e)
{
#if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING >= 1
unsigned int i;
for (i=0; i<sizeof(*e); i++) ((char*)e)[i] = 0xFF;
#endif
e->next = m->rrcache_free;
m->rrcache_free = e;
m->rrcache_totalused--;
}
mDNSlocal void ReleaseCacheGroup(mDNS *const m, CacheGroup **cp)
{
CacheEntity *e = (CacheEntity *)(*cp);
if ((*cp)->rrcache_tail != &(*cp)->members)
LogMsg("ERROR: (*cp)->members == mDNSNULL but (*cp)->rrcache_tail != &(*cp)->members)");
if ((*cp)->name != (domainname*)((*cp)->namestorage)) mDNSPlatformMemFree((*cp)->name);
(*cp)->name = mDNSNULL;
*cp = (*cp)->next; ReleaseCacheEntity(m, e);
}
mDNSlocal void ReleaseCacheRecord(mDNS *const m, CacheRecord *r)
{
if (r->resrec.rdata && r->resrec.rdata != (RData*)&r->smallrdatastorage) mDNSPlatformMemFree(r->resrec.rdata);
r->resrec.rdata = mDNSNULL;
ReleaseCacheEntity(m, (CacheEntity *)r);
}
mDNSlocal void CheckCacheExpiration(mDNS *const m, const mDNSu32 slot, CacheGroup *const cg)
{
CacheRecord **rp = &cg->members;
if (m->lock_rrcache) { LogMsg("CheckCacheExpiration ERROR! Cache already locked!"); return; }
m->lock_rrcache = 1;
while (*rp)
{
CacheRecord *const rr = *rp;
mDNSs32 event = RRExpireTime(rr);
if (m->timenow - event >= 0) {
*rp = rr->next; verbosedebugf("CheckCacheExpiration: Deleting%7d %7d %p %s",
m->timenow - rr->TimeRcvd, rr->resrec.rroriginalttl, rr->CRActiveQuestion, CRDisplayString(m, rr));
if (rr->CRActiveQuestion) {
DNSQuestion *q = rr->CRActiveQuestion;
if (!mDNSOpaque16IsZero(q->TargetQID) && !q->LongLived && ActiveQuestion(q))
{
q->ThisQInterval = InitialQuestionInterval;
q->LastQTime = m->timenow - q->ThisQInterval;
SetNextQueryTime(m, q);
}
CacheRecordRmv(m, rr);
m->rrcache_active--;
}
ReleaseCacheRecord(m, rr);
}
else {
if (rr->DelayDelivery && rr->DelayDelivery - m->timenow > 0)
event = rr->DelayDelivery;
else
{
if (rr->DelayDelivery) CacheRecordDeferredAdd(m, rr);
if (rr->CRActiveQuestion && rr->UnansweredQueries < MaxUnansweredQueries)
{
if (m->timenow - rr->NextRequiredQuery < 0) event = NextCacheCheckEvent(rr); else {
m->NextScheduledQuery = m->timenow;
event = m->timenow + 0x3FFFFFFF;
}
}
}
verbosedebugf("CheckCacheExpiration:%6d %5d %s",
(event - m->timenow) / mDNSPlatformOneSecond, CacheCheckGracePeriod(rr), CRDisplayString(m, rr));
if (m->rrcache_nextcheck[slot] - event > 0)
m->rrcache_nextcheck[slot] = event;
rp = &rr->next;
}
}
if (cg->rrcache_tail != rp) verbosedebugf("CheckCacheExpiration: Updating CacheGroup tail from %p to %p", cg->rrcache_tail, rp);
cg->rrcache_tail = rp;
m->lock_rrcache = 0;
}
mDNSlocal void AnswerSuppressUnusableQuestion(mDNS *const m, DNSQuestion *q)
{
LogInfo("AnswerSuppressUnusableQuestion: Generating negative response for question %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
if (!m->CurrentQuestion) LogMsg("AnswerSuppressUnusableQuestion: ERROR!! CurrentQuestion not set");
MakeNegativeCacheRecord(m, &m->rec.r, &q->qname, q->qnamehash, q->qtype, q->qclass, 60, mDNSInterface_Any, mDNSNULL);
AnswerCurrentQuestionWithResourceRecord(m, &m->rec.r, QC_addnocache);
if (m->CurrentQuestion == q) q->ThisQInterval = 0; m->rec.r.resrec.RecordType = 0; }
mDNSlocal void AnswerNewQuestion(mDNS *const m)
{
mDNSBool ShouldQueryImmediately = mDNStrue;
DNSQuestion *const q = m->NewQuestions; const mDNSu32 slot = HashSlot(&q->qname);
CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
verbosedebugf("AnswerNewQuestion: Answering %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
if (cg) CheckCacheExpiration(m, slot, cg);
if (m->NewQuestions != q) { LogInfo("AnswerNewQuestion: Question deleted while doing CheckCacheExpiration"); goto exit; }
m->NewQuestions = q->next;
if (m->lock_rrcache) LogMsg("AnswerNewQuestion ERROR! Cache already locked!");
m->lock_rrcache = 1;
if (m->CurrentQuestion)
LogMsg("AnswerNewQuestion ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = q;
if (q->NoAnswer == NoAnswer_Fail)
{
LogMsg("AnswerNewQuestion: NoAnswer_Fail %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
MakeNegativeCacheRecord(m, &m->rec.r, &q->qname, q->qnamehash, q->qtype, q->qclass, 60, mDNSInterface_Any, q->qDNSServer);
q->NoAnswer = NoAnswer_Normal; AnswerCurrentQuestionWithResourceRecord(m, &m->rec.r, QC_addnocache);
q->NoAnswer = NoAnswer_Fail; m->rec.r.resrec.RecordType = 0; }
if (m->CurrentQuestion != q) { LogInfo("AnswerNewQuestion: Question deleted while generating NoAnswer_Fail response"); goto exit; }
if (q->InterfaceID == mDNSInterface_Any)
{
if (m->CurrentRecord)
LogMsg("AnswerNewQuestion ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord && m->CurrentRecord != m->NewLocalRecords)
{
AuthRecord *rr = m->CurrentRecord;
m->CurrentRecord = rr->next;
if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly || rr->resrec.InterfaceID == mDNSInterface_P2P)
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
{
AnswerLocalQuestionWithLocalAuthRecord(m, q, rr, mDNStrue);
if (m->CurrentQuestion != q) break; }
}
m->CurrentRecord = mDNSNULL;
}
if (m->CurrentQuestion != q) { LogInfo("AnswerNewQuestion: Question deleted while while giving LocalOnly record answers"); goto exit; }
if (QuerySuppressed(q)) { q->SuppressQuery = mDNSfalse; AnswerSuppressUnusableQuestion(m, q); q->SuppressQuery = mDNStrue; }
else
{
CacheRecord *rr;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (SameNameRecordAnswersQuestion(&rr->resrec, q))
{
mDNSu32 SecsSinceRcvd = ((mDNSu32)(m->timenow - rr->TimeRcvd)) / mDNSPlatformOneSecond;
if (rr->resrec.rroriginalttl <= SecsSinceRcvd)
{
LogMsg("AnswerNewQuestion: How is rr->resrec.rroriginalttl %lu <= SecsSinceRcvd %lu for %s %d %d",
rr->resrec.rroriginalttl, SecsSinceRcvd, CRDisplayString(m, rr), m->timenow, rr->TimeRcvd);
continue; }
if ((rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) || (q->ExpectUnique))
ShouldQueryImmediately = mDNSfalse;
q->CurrentAnswers++;
if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers++;
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers++;
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_add);
if (m->CurrentQuestion != q) break; }
else if (RRTypeIsAddressType(rr->resrec.rrtype) && RRTypeIsAddressType(q->qtype))
ShouldQueryImmediately = mDNSfalse;
}
if (m->CurrentQuestion != q) { debugf("AnswerNewQuestion: Question deleted while giving cache answers"); goto exit; }
if (ShouldQueryImmediately && ActiveQuestion(q))
{
debugf("AnswerNewQuestion: ShouldQueryImmediately %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
q->ThisQInterval = InitialQuestionInterval;
q->LastQTime = m->timenow - q->ThisQInterval;
if (mDNSOpaque16IsZero(q->TargetQID)) {
if (!m->RandomQueryDelay)
m->RandomQueryDelay = (mDNSPlatformOneSecond + mDNSRandom(mDNSPlatformOneSecond*5) - 1) / 50 + 1;
q->LastQTime += m->RandomQueryDelay;
}
}
SetNextQueryTime(m,q);
exit:
m->CurrentQuestion = mDNSNULL;
m->lock_rrcache = 0;
}
mDNSlocal void AnswerNewLocalOnlyQuestion(mDNS *const m)
{
DNSQuestion *q = m->NewLocalOnlyQuestions; m->NewLocalOnlyQuestions = q->next;
debugf("AnswerNewLocalOnlyQuestion: Answering %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
if (m->CurrentQuestion)
LogMsg("AnswerNewLocalOnlyQuestion ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = q;
if (m->CurrentRecord)
LogMsg("AnswerNewLocalOnlyQuestion ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord && m->CurrentRecord != m->NewLocalRecords)
{
AuthRecord *rr = m->CurrentRecord;
m->CurrentRecord = rr->next;
if (ResourceRecordAnswersQuestion(&rr->resrec, q))
{
AnswerLocalQuestionWithLocalAuthRecord(m, q, rr, mDNStrue);
if (m->CurrentQuestion != q) break; }
}
m->CurrentQuestion = mDNSNULL;
m->CurrentRecord = mDNSNULL;
}
mDNSlocal CacheEntity *GetCacheEntity(mDNS *const m, const CacheGroup *const PreserveCG)
{
CacheEntity *e = mDNSNULL;
if (m->lock_rrcache) { LogMsg("GetFreeCacheRR ERROR! Cache already locked!"); return(mDNSNULL); }
m->lock_rrcache = 1;
if (!m->rrcache_free && m->MainCallback)
{
if (m->rrcache_totalused != m->rrcache_size)
LogMsg("GetFreeCacheRR: count mismatch: m->rrcache_totalused %lu != m->rrcache_size %lu",
m->rrcache_totalused, m->rrcache_size);
if (m->rrcache_size > 5000 && m->rrcache_size / 32 > m->rrcache_active)
LogInfo("Possible denial-of-service attack in progress: m->rrcache_size %lu; m->rrcache_active %lu",
m->rrcache_size, m->rrcache_active);
else
{
mDNS_DropLockBeforeCallback(); m->MainCallback(m, mStatus_GrowCache);
mDNS_ReclaimLockAfterCallback(); }
}
if (!m->rrcache_free)
{
mDNSu32 oldtotalused = m->rrcache_totalused;
mDNSu32 slot;
for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
{
CacheGroup **cp = &m->rrcache_hash[slot];
while (*cp)
{
CacheRecord **rp = &(*cp)->members;
while (*rp)
{
if ((*rp)->CRActiveQuestion || (*rp)->NextInCFList)
rp=&(*rp)->next;
else
{
CacheRecord *rr = *rp;
*rp = (*rp)->next; ReleaseCacheRecord(m, rr);
}
}
if ((*cp)->rrcache_tail != rp)
verbosedebugf("GetFreeCacheRR: Updating rrcache_tail[%lu] from %p to %p", slot, (*cp)->rrcache_tail, rp);
(*cp)->rrcache_tail = rp;
if ((*cp)->members || (*cp)==PreserveCG) cp=&(*cp)->next;
else ReleaseCacheGroup(m, cp);
}
}
LogInfo("GetCacheEntity recycled %d records to reduce cache from %d to %d",
oldtotalused - m->rrcache_totalused, oldtotalused, m->rrcache_totalused);
}
if (m->rrcache_free) {
e = m->rrcache_free;
m->rrcache_free = e->next;
if (++m->rrcache_totalused >= m->rrcache_report)
{
LogInfo("RR Cache now using %ld objects", m->rrcache_totalused);
if (m->rrcache_report < 100) m->rrcache_report += 10;
else if (m->rrcache_report < 1000) m->rrcache_report += 100;
else m->rrcache_report += 1000;
}
mDNSPlatformMemZero(e, sizeof(*e));
}
m->lock_rrcache = 0;
return(e);
}
mDNSlocal CacheRecord *GetCacheRecord(mDNS *const m, CacheGroup *cg, mDNSu16 RDLength)
{
CacheRecord *r = (CacheRecord *)GetCacheEntity(m, cg);
if (r)
{
r->resrec.rdata = (RData*)&r->smallrdatastorage; if (RDLength > InlineCacheRDSize) {
r->resrec.rdata = (RData*)mDNSPlatformMemAllocate(sizeofRDataHeader + RDLength);
if (r->resrec.rdata) r->resrec.rdata->MaxRDLength = r->resrec.rdlength = RDLength;
else { ReleaseCacheEntity(m, (CacheEntity*)r); r = mDNSNULL; }
}
}
return(r);
}
mDNSlocal CacheGroup *GetCacheGroup(mDNS *const m, const mDNSu32 slot, const ResourceRecord *const rr)
{
mDNSu16 namelen = DomainNameLength(rr->name);
CacheGroup *cg = (CacheGroup*)GetCacheEntity(m, mDNSNULL);
if (!cg) { LogMsg("GetCacheGroup: Failed to allocate memory for %##s", rr->name->c); return(mDNSNULL); }
cg->next = m->rrcache_hash[slot];
cg->namehash = rr->namehash;
cg->members = mDNSNULL;
cg->rrcache_tail = &cg->members;
cg->name = (domainname*)cg->namestorage;
if (namelen > InlineCacheGroupNameSize) cg->name = mDNSPlatformMemAllocate(namelen);
if (!cg->name)
{
LogMsg("GetCacheGroup: Failed to allocate name storage for %##s", rr->name->c);
ReleaseCacheEntity(m, (CacheEntity*)cg);
return(mDNSNULL);
}
AssignDomainName(cg->name, rr->name);
if (CacheGroupForRecord(m, slot, rr)) LogMsg("GetCacheGroup: Already have CacheGroup for %##s", rr->name->c);
m->rrcache_hash[slot] = cg;
if (CacheGroupForRecord(m, slot, rr) != cg) LogMsg("GetCacheGroup: Not finding CacheGroup for %##s", rr->name->c);
return(cg);
}
mDNSexport void mDNS_PurgeCacheResourceRecord(mDNS *const m, CacheRecord *rr)
{
if (m->mDNS_busy != m->mDNS_reentrancy+1)
LogMsg("mDNS_PurgeCacheResourceRecord: Lock not held! mDNS_busy (%ld) mDNS_reentrancy (%ld)", m->mDNS_busy, m->mDNS_reentrancy);
rr->TimeRcvd = m->timenow - mDNSPlatformOneSecond * 60;
rr->UnansweredQueries = MaxUnansweredQueries;
rr->resrec.rroriginalttl = 0;
SetNextCacheCheckTimeForRecord(m, rr);
}
mDNSexport mDNSs32 mDNS_TimeNow(const mDNS *const m)
{
mDNSs32 time;
mDNSPlatformLock(m);
if (m->mDNS_busy)
{
LogMsg("mDNS_TimeNow called while holding mDNS lock. This is incorrect. Code protected by lock should just use m->timenow.");
if (!m->timenow) LogMsg("mDNS_TimeNow: m->mDNS_busy is %ld but m->timenow not set", m->mDNS_busy);
}
if (m->timenow) time = m->timenow;
else time = mDNS_TimeNow_NoLock(m);
mDNSPlatformUnlock(m);
return(time);
}
#define SetSPSProxyListChanged(X) do { \
if (m->SPSProxyListChanged && m->SPSProxyListChanged != (X)) mDNSPlatformUpdateProxyList(m, m->SPSProxyListChanged); \
m->SPSProxyListChanged = (X); } while(0)
mDNSlocal void CheckProxyRecords(mDNS *const m, AuthRecord *list)
{
m->CurrentRecord = list;
while (m->CurrentRecord)
{
AuthRecord *rr = m->CurrentRecord;
if (rr->resrec.RecordType != kDNSRecordTypeDeregistering && rr->WakeUp.HMAC.l[0])
{
if (m->SPSSocket && m->timenow - rr->TimeExpire < 0) {
if (m->NextScheduledSPS - rr->TimeExpire > 0)
m->NextScheduledSPS = rr->TimeExpire;
}
else {
LogSPS("CheckProxyRecords: Removing %d H-MAC %.6a I-MAC %.6a %d %s",
m->ProxyRecords, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, ARDisplayString(m, rr));
SetSPSProxyListChanged(rr->resrec.InterfaceID);
mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
}
}
if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
}
}
mDNSexport mDNSs32 mDNS_Execute(mDNS *const m)
{
mDNS_Lock(m);
if (m->timenow - m->NextScheduledEvent >= 0)
{
int i;
AuthRecord *head, *tail;
verbosedebugf("mDNS_Execute");
if (m->CurrentQuestion)
LogMsg("mDNS_Execute: ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
if (m->CurrentRecord)
LogMsg("mDNS_Execute: ERROR m->CurrentRecord already set: %s", ARDisplayString(m, m->CurrentRecord));
if (m->SuppressProbes && m->timenow - m->SuppressProbes >= 0) m->SuppressProbes = 0;
if (m->NumFailedProbes && m->timenow - m->ProbeFailTime >= mDNSPlatformOneSecond * 10) m->NumFailedProbes = 0;
if (m->rrcache_size && m->timenow - m->NextCacheCheck >= 0)
{
mDNSu32 slot, numchecked = 0;
m->NextCacheCheck = m->timenow + 0x3FFFFFFF;
for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
{
if (m->timenow - m->rrcache_nextcheck[slot] >= 0)
{
CacheGroup **cp = &m->rrcache_hash[slot];
m->rrcache_nextcheck[slot] = m->timenow + 0x3FFFFFFF;
while (*cp)
{
debugf("m->NextCacheCheck %4d Slot %3d %##s", numchecked, slot, *cp ? (*cp)->name : (domainname*)"\x04NULL");
numchecked++;
CheckCacheExpiration(m, slot, *cp);
if ((*cp)->members) cp=&(*cp)->next;
else ReleaseCacheGroup(m, cp);
}
}
if (m->NextCacheCheck - m->rrcache_nextcheck[slot] > 0)
m->NextCacheCheck = m->rrcache_nextcheck[slot];
}
debugf("m->NextCacheCheck %4d checked, next in %d", numchecked, m->NextCacheCheck - m->timenow);
}
if (m->timenow - m->NextScheduledSPS >= 0)
{
m->NextScheduledSPS = m->timenow + 0x3FFFFFFF;
CheckProxyRecords(m, m->DuplicateRecords); CheckProxyRecords(m, m->ResourceRecords);
}
SetSPSProxyListChanged(mDNSNULL);
if (m->AnnounceOwner && m->timenow - m->AnnounceOwner >= 0) m->AnnounceOwner = 0;
if (m->DelaySleep && m->timenow - m->DelaySleep >= 0)
{
m->DelaySleep = 0;
if (m->SleepState == SleepState_Transferring)
{
LogSPS("Re-sleep delay passed; now checking for Sleep Proxy Servers");
BeginSleepProcessing(m);
}
}
for (i=0; m->NewQuestions && i<1000; i++)
{
if (m->NewQuestions->DelayAnswering && m->timenow - m->NewQuestions->DelayAnswering < 0) break;
AnswerNewQuestion(m);
}
if (i >= 1000) LogMsg("mDNS_Execute: AnswerNewQuestion exceeded loop limit");
for (i=0; i<1000 && m->LocalRemoveEvents; i++)
{
m->LocalRemoveEvents = mDNSfalse;
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
AuthRecord *rr = m->CurrentRecord;
if (rr->AnsweredLocalQ && rr->resrec.RecordType == kDNSRecordTypeDeregistering)
{
debugf("mDNS_Execute: Generating local RMV events for %s", ARDisplayString(m, rr));
rr->resrec.RecordType = kDNSRecordTypeShared;
AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNSfalse);
if (m->CurrentRecord == rr) {
rr->resrec.RecordType = kDNSRecordTypeDeregistering;
rr->AnsweredLocalQ = mDNSfalse;
}
}
if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
}
}
if (i >= 1000) LogMsg("mDNS_Execute: m->LocalRemoveEvents exceeded loop limit");
for (i=0; m->NewLocalOnlyQuestions && i<1000; i++) AnswerNewLocalOnlyQuestion(m);
if (i >= 1000) LogMsg("mDNS_Execute: AnswerNewLocalOnlyQuestion exceeded loop limit");
head = tail = mDNSNULL;
for (i=0; i<1000 && m->NewLocalRecords && m->NewLocalRecords != head; i++)
{
AuthRecord *rr = m->NewLocalRecords;
m->NewLocalRecords = m->NewLocalRecords->next;
if (LocalRecordReady(rr))
{
debugf("mDNS_Execute: Delivering Add event with LocalAuthRecord %s", ARDisplayString(m, rr));
AnswerAllLocalQuestionsWithLocalAuthRecord(m, rr, mDNStrue);
}
else if (!rr->next)
{
debugf("mDNS_Execute: Just one LocalAuthRecord %s, breaking out of the loop early", ARDisplayString(m, rr));
if (head != mDNSNULL || m->NewLocalRecords != mDNSNULL)
LogMsg("mDNS_Execute: ERROR!!: head %p, NewLocalRecords %p", head, m->NewLocalRecords);
head = rr;
}
else
{
AuthRecord **p = &m->ResourceRecords; debugf("mDNS_Execute: Skipping LocalAuthRecord %s", ARDisplayString(m, rr));
while (*p && *p != rr) p=&(*p)->next;
if (*p) *p = rr->next; else { LogMsg("mDNS_Execute: ERROR!! Cannot find record %s in ResourceRecords list", ARDisplayString(m, rr)); break; }
if (!head)
{
while (*p) p=&(*p)->next;
*p = rr;
head = tail = rr;
}
else
{
tail->next = rr;
tail = rr;
}
rr->next = mDNSNULL;
}
}
m->NewLocalRecords = head;
debugf("mDNS_Execute: Setting NewLocalRecords to %s", (head ? ARDisplayString(m, head) : "NULL"));
if (i >= 1000) LogMsg("mDNS_Execute: m->NewLocalRecords exceeded loop limit");
AnswerQuestionsForDNSServerChanges(m);
if (m->mDNSPlatformStatus != mStatus_NoError || (m->SleepState == SleepState_Sleeping))
DiscardDeregistrations(m);
if (m->mDNSPlatformStatus == mStatus_NoError && (m->SuppressSending == 0 || m->timenow - m->SuppressSending >= 0))
{
m->SuppressSending = 0;
if (m->timenow - m->NextScheduledQuery >= 0 || m->timenow - m->NextScheduledProbe >= 0) SendQueries(m);
if (m->timenow - m->NextScheduledQuery >= 0)
{
DNSQuestion *q;
LogMsg("mDNS_Execute: SendQueries didn't send all its queries (%d - %d = %d) will try again in one second",
m->timenow, m->NextScheduledQuery, m->timenow - m->NextScheduledQuery);
m->NextScheduledQuery = m->timenow + mDNSPlatformOneSecond;
for (q = m->Questions; q && q != m->NewQuestions; q=q->next)
if (ActiveQuestion(q) && m->timenow - NextQSendTime(q) >= 0)
LogMsg("mDNS_Execute: SendQueries didn't send %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
}
if (m->timenow - m->NextScheduledProbe >= 0)
{
LogMsg("mDNS_Execute: SendQueries didn't send all its probes (%d - %d = %d) will try again in one second",
m->timenow, m->NextScheduledProbe, m->timenow - m->NextScheduledProbe);
m->NextScheduledProbe = m->timenow + mDNSPlatformOneSecond;
}
if (m->timenow - m->NextScheduledResponse >= 0) SendResponses(m);
if (m->timenow - m->NextScheduledResponse >= 0)
{
LogMsg("mDNS_Execute: SendResponses didn't send all its responses; will try again in one second");
m->NextScheduledResponse = m->timenow + mDNSPlatformOneSecond;
}
}
m->RandomQueryDelay = 0;
m->RandomReconfirmDelay = 0;
#ifndef UNICAST_DISABLED
if (m->NextSRVUpdate && m->timenow - m->NextSRVUpdate >= 0) UpdateAllSRVRecords(m);
if (m->timenow - m->NextScheduledNATOp >= 0) CheckNATMappings(m);
if (m->timenow - m->NextuDNSEvent >= 0) uDNS_Tasks(m);
#endif
}
mDNS_Unlock(m); return(m->NextScheduledEvent);
}
mDNSlocal void SuspendLLQs(mDNS *m)
{
DNSQuestion *q;
for (q = m->Questions; q; q = q->next)
if (ActiveQuestion(q) && !mDNSOpaque16IsZero(q->TargetQID) && q->LongLived && q->state == LLQ_Established)
{ q->ReqLease = 0; sendLLQRefresh(m, q); }
}
mDNSlocal void ActivateUnicastQuery(mDNS *const m, DNSQuestion *const question, mDNSBool ScheduleImmediately)
{
#if APPLE_OSX_mDNSResponder
if (RRTypeIsAddressType(question->qtype) && PrivateQuery(question) &&
!SameDomainLabel(question->qname.c, (const mDNSu8 *)"\x0c_autotunnel6")&& question->QuestionCallback != AutoTunnelCallback)
{
question->NoAnswer = NoAnswer_Suspended;
AddNewClientTunnel(m, question);
return;
}
#endif // APPLE_OSX_mDNSResponder
if (!question->DuplicateOf)
{
debugf("ActivateUnicastQuery: %##s %s%s%s",
question->qname.c, DNSTypeName(question->qtype), PrivateQuery(question) ? " (Private)" : "", ScheduleImmediately ? " ScheduleImmediately" : "");
question->CNAMEReferrals = 0;
if (question->nta) { CancelGetZoneData(m, question->nta); question->nta = mDNSNULL; }
if (question->LongLived)
{
question->state = LLQ_InitialRequest;
question->id = zeroOpaque64;
question->servPort = zeroIPPort;
if (question->tcp) { DisposeTCPConn(question->tcp); question->tcp = mDNSNULL; }
}
if (ScheduleImmediately)
{
question->ThisQInterval = InitialQuestionInterval;
question->LastQTime = m->timenow - question->ThisQInterval;
SetNextQueryTime(m, question);
}
}
}
mDNSexport void mDNSCoreRestartQueries(mDNS *const m)
{
DNSQuestion *q;
#ifndef UNICAST_DISABLED
if (m->CurrentQuestion)
LogMsg("mDNSCoreRestartQueries: ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
m->CurrentQuestion = m->Questions;
while (m->CurrentQuestion)
{
q = m->CurrentQuestion;
m->CurrentQuestion = m->CurrentQuestion->next;
if (!mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q)) ActivateUnicastQuery(m, q, mDNStrue);
}
#endif
for (q = m->Questions; q; q=q->next) if (mDNSOpaque16IsZero(q->TargetQID) && ActiveQuestion(q))
{
q->ThisQInterval = InitialQuestionInterval; q->RequestUnicast = 2; q->LastQTime = m->timenow - q->ThisQInterval;
q->RecentAnswerPkts = 0;
ExpireDupSuppressInfo(q->DupSuppress, m->timenow);
m->NextScheduledQuery = m->timenow;
}
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Power Management (Sleep/Wake)
#endif
mDNSexport void mDNS_UpdateAllowSleep(mDNS *const m)
{
#ifndef IDLESLEEPCONTROL_DISABLED
mDNSBool allowSleep = mDNStrue;
if (m->SystemSleepOnlyIfWakeOnLAN)
{
if (m->ProxyRecords)
{
allowSleep = mDNSfalse;
LogInfo("Sleep disabled because we are proxying %d records", m->ProxyRecords);
}
if (allowSleep && mDNSCoreHaveAdvertisedMulticastServices(m))
{
NetworkInterfaceInfo *intf;
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
{
if (intf->McastTxRx)
{
if (!intf->NetWake)
{
allowSleep = mDNSfalse;
LogInfo("Sleep disabled because %s does not support NetWake", intf->ifname);
break;
}
if (FindSPSInCache1(m, &intf->NetWakeBrowse, mDNSNULL, mDNSNULL) == mDNSNULL)
{
allowSleep = mDNSfalse;
LogInfo("Sleep disabled because %s has no sleep proxy", intf->ifname);
break;
}
}
}
}
#endif
}
mDNSPlatformSetAllowSleep(m, allowSleep);
}
mDNSlocal void SendSPSRegistration(mDNS *const m, NetworkInterfaceInfo *intf, const mDNSOpaque16 id)
{
const int optspace = DNSOpt_Header_Space + DNSOpt_LeaseData_Space + DNSOpt_Owner_Space(&m->PrimaryMAC, &intf->MAC);
const int sps = intf->NextSPSAttempt / 3;
AuthRecord *rr;
if (!intf->SPSAddr[sps].type)
{
intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond;
if (m->NextScheduledSPRetry - intf->NextSPSAttemptTime > 0)
m->NextScheduledSPRetry = intf->NextSPSAttemptTime;
LogSPS("SendSPSRegistration: %s SPS %d (%d) %##s not yet resolved", intf->ifname, intf->NextSPSAttempt, sps, intf->NetWakeResolve[sps].qname.c);
goto exit;
}
if (mDNSOpaque16IsZero(id))
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.RecordType > kDNSRecordTypeDeregistering)
if (rr->resrec.InterfaceID == intf->InterfaceID || (!rr->resrec.InterfaceID && (rr->ForceMCast || IsLocalDomain(rr->resrec.name))))
rr->SendRNow = mDNSInterfaceMark;
while (1)
{
mDNSu8 *p = m->omsg.data;
InitializeDNSMessage(&m->omsg.h, mDNSOpaque16IsZero(id) ? mDNS_NewMessageID(m) : id, UpdateReqFlags);
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->SendRNow || (!mDNSOpaque16IsZero(id) && !AuthRecord_uDNS(rr) && mDNSSameOpaque16(rr->updateid, id) && m->timenow - (rr->LastAPTime + rr->ThisAPInterval) >= 0))
{
mDNSu8 *newptr;
const mDNSu8 *const limit = m->omsg.data + (m->omsg.h.mDNS_numUpdates ? NormalMaxDNSMessageData : AbsoluteMaxDNSMessageData) - optspace;
if (rr->resrec.RecordType & kDNSRecordTypeUniqueMask)
rr->resrec.rrclass |= kDNSClass_UniqueRRSet; newptr = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.mDNS_numUpdates, &rr->resrec, rr->resrec.rroriginalttl, limit);
rr->resrec.rrclass &= ~kDNSClass_UniqueRRSet; if (!newptr)
LogSPS("SendSPSRegistration put %s FAILED %d/%d %s", intf->ifname, p - m->omsg.data, limit - m->omsg.data, ARDisplayString(m, rr));
else
{
LogSPS("SendSPSRegistration put %s %s", intf->ifname, ARDisplayString(m, rr));
rr->SendRNow = mDNSNULL;
rr->ThisAPInterval = mDNSPlatformOneSecond;
rr->LastAPTime = m->timenow;
rr->updateid = m->omsg.h.id;
if (m->NextScheduledResponse - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
m->NextScheduledResponse = (rr->LastAPTime + rr->ThisAPInterval);
p = newptr;
}
}
if (!m->omsg.h.mDNS_numUpdates) break;
else
{
AuthRecord opt;
mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, mDNSNULL, mDNSNULL);
opt.resrec.rrclass = NormalMaxDNSMessageData;
opt.resrec.rdlength = sizeof(rdataOPT) * 2; opt.resrec.rdestimate = sizeof(rdataOPT) * 2;
opt.resrec.rdata->u.opt[0].opt = kDNSOpt_Lease;
opt.resrec.rdata->u.opt[0].optlen = DNSOpt_LeaseData_Space - 4;
opt.resrec.rdata->u.opt[0].u.updatelease = DEFAULT_UPDATE_LEASE;
SetupOwnerOpt(m, intf, &opt.resrec.rdata->u.opt[1]);
LogSPS("SendSPSRegistration put %s %s", intf->ifname, ARDisplayString(m, &opt));
p = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.numAdditionals, &opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
if (!p)
LogMsg("SendSPSRegistration: Failed to put OPT record (%d updates) %s", m->omsg.h.mDNS_numUpdates, ARDisplayString(m, &opt));
else
{
mStatus err;
m->SentSleepProxyRegistration = mDNStrue;
LogSPS("SendSPSRegistration: Sending Update %s %d (%d) id %5d with %d records %d bytes to %#a:%d", intf->ifname, intf->NextSPSAttempt, sps,
mDNSVal16(m->omsg.h.id), m->omsg.h.mDNS_numUpdates, p - m->omsg.data, &intf->SPSAddr[sps], mDNSVal16(intf->SPSPort[sps]));
err = mDNSSendDNSMessage(m, &m->omsg, p, intf->InterfaceID, mDNSNULL, &intf->SPSAddr[sps], intf->SPSPort[sps], mDNSNULL, mDNSNULL);
if (err) LogSPS("SendSPSRegistration: mDNSSendDNSMessage err %d", err);
if (err && intf->SPSAddr[sps].type == mDNSAddrType_IPv6 && intf->NetWakeResolve[sps].ThisQInterval == -1)
{
LogSPS("SendSPSRegistration %d %##s failed to send to IPv6 address; will try IPv4 instead", sps, intf->NetWakeResolve[sps].qname.c);
intf->NetWakeResolve[sps].qtype = kDNSType_A;
mDNS_StartQuery_internal(m, &intf->NetWakeResolve[sps]);
return;
}
}
}
}
intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond * 10;
exit:
if (mDNSOpaque16IsZero(id) && intf->NextSPSAttempt < 8) intf->NextSPSAttempt++;
}
mDNSlocal void RetrySPSRegistrations(mDNS *const m)
{
AuthRecord *rr;
NetworkInterfaceInfo *intf;
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
if (intf->NextSPSAttempt && intf->NextSPSAttemptTime == m->timenow + mDNSPlatformOneSecond * 10)
intf->NextSPSAttemptTime++;
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (!AuthRecord_uDNS(rr) && !mDNSOpaque16IsZero(rr->updateid) && m->timenow - (rr->LastAPTime + rr->ThisAPInterval) >= 0)
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
if (!rr->resrec.InterfaceID || rr->resrec.InterfaceID == intf->InterfaceID)
{
LogSPS("RetrySPSRegistrations: %s", ARDisplayString(m, rr));
SendSPSRegistration(m, intf, rr->updateid);
}
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
if (intf->NextSPSAttempt && intf->NextSPSAttemptTime == m->timenow + mDNSPlatformOneSecond * 10 && intf->NextSPSAttempt < 8)
intf->NextSPSAttempt++;
}
mDNSlocal void NetWakeResolve(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
{
NetworkInterfaceInfo *intf = (NetworkInterfaceInfo *)question->QuestionContext;
int sps = (int)(question - intf->NetWakeResolve);
(void)m; LogSPS("NetWakeResolve: SPS: %d Add: %d %s", sps, AddRecord, RRDisplayString(m, answer));
if (!AddRecord) return; if (answer->rrtype != question->qtype) return;
if (answer->rrtype == kDNSType_SRV)
{
mDNS_StopQuery(m, question);
intf->SPSPort[sps] = answer->rdata->u.srv.port;
AssignDomainName(&question->qname, &answer->rdata->u.srv.target);
question->qtype = kDNSType_AAAA;
mDNS_StartQuery(m, question);
}
else if (answer->rrtype == kDNSType_AAAA && answer->rdlength == sizeof(mDNSv6Addr) && mDNSv6AddressIsLinkLocal(&answer->rdata->u.ipv6))
{
mDNS_StopQuery(m, question);
question->ThisQInterval = -1;
intf->SPSAddr[sps].type = mDNSAddrType_IPv6;
intf->SPSAddr[sps].ip.v6 = answer->rdata->u.ipv6;
mDNS_Lock(m);
if (sps == intf->NextSPSAttempt/3) SendSPSRegistration(m, intf, zeroID); mDNS_Unlock(m);
}
else if (answer->rrtype == kDNSType_AAAA && answer->rdlength == 0)
{
mDNS_StopQuery(m, question);
LogSPS("NetWakeResolve: SPS %d %##s has no IPv6 address, will try IPv4 instead", sps, question->qname.c);
question->qtype = kDNSType_A;
mDNS_StartQuery(m, question);
}
else if (answer->rrtype == kDNSType_A && answer->rdlength == sizeof(mDNSv4Addr))
{
mDNS_StopQuery(m, question);
question->ThisQInterval = -1;
intf->SPSAddr[sps].type = mDNSAddrType_IPv4;
intf->SPSAddr[sps].ip.v4 = answer->rdata->u.ipv4;
mDNS_Lock(m);
if (sps == intf->NextSPSAttempt/3) SendSPSRegistration(m, intf, zeroID); mDNS_Unlock(m);
}
}
mDNSexport mDNSBool mDNSCoreHaveAdvertisedMulticastServices(mDNS *const m)
{
AuthRecord *rr;
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.rrtype == kDNSType_SRV && !AuthRecord_uDNS(rr) && !mDNSSameIPPort(rr->resrec.rdata->u.srv.port, DiscardPort))
return mDNStrue;
return mDNSfalse;
}
mDNSlocal void SendSleepGoodbyes(mDNS *const m)
{
AuthRecord *rr;
m->SleepState = SleepState_Sleeping;
#ifndef UNICAST_DISABLED
SleepRecordRegistrations(m); #endif
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.RecordType == kDNSRecordTypeShared && rr->RequireGoodbye)
rr->ImmedAnswer = mDNSInterfaceMark;
SendResponses(m);
}
mDNSlocal void BeginSleepProcessing(mDNS *const m)
{
mDNSBool SendGoodbyes = mDNStrue;
const CacheRecord *sps[3] = { mDNSNULL };
m->NextScheduledSPRetry = m->timenow;
if (!m->SystemWakeOnLANEnabled) LogSPS("BeginSleepProcessing: m->SystemWakeOnLANEnabled is false");
else if (!mDNSCoreHaveAdvertisedMulticastServices(m)) LogSPS("BeginSleepProcessing: No advertised services");
else {
NetworkInterfaceInfo *intf;
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
{
if (!intf->NetWake) LogSPS("BeginSleepProcessing: %-6s not capable of magic packet wakeup", intf->ifname);
#if APPLE_OSX_mDNSResponder
else if (ActivateLocalProxy(m, intf->ifname) == mStatus_NoError)
{
SendGoodbyes = mDNSfalse;
LogSPS("BeginSleepProcessing: %-6s using local proxy", intf->ifname);
}
#endif // APPLE_OSX_mDNSResponder
else
{
FindSPSInCache(m, &intf->NetWakeBrowse, sps);
if (!sps[0]) LogSPS("BeginSleepProcessing: %-6s %#a No Sleep Proxy Server found (Next Browse Q in %d, interval %d)",
intf->ifname, &intf->ip, NextQSendTime(&intf->NetWakeBrowse) - m->timenow, intf->NetWakeBrowse.ThisQInterval);
else
{
int i;
SendGoodbyes = mDNSfalse;
intf->NextSPSAttempt = 0;
intf->NextSPSAttemptTime = m->timenow + mDNSPlatformOneSecond;
for (i=0; i<3; i++)
{
#if ForceAlerts
if (intf->SPSAddr[i].type)
{ LogMsg("BeginSleepProcessing: %s %d intf->SPSAddr[i].type %d", intf->ifname, i, intf->SPSAddr[i].type); *(long*)0 = 0; }
if (intf->NetWakeResolve[i].ThisQInterval >= 0)
{ LogMsg("BeginSleepProcessing: %s %d intf->NetWakeResolve[i].ThisQInterval %d", intf->ifname, i, intf->NetWakeResolve[i].ThisQInterval); *(long*)0 = 0; }
#endif
intf->SPSAddr[i].type = mDNSAddrType_None;
if (intf->NetWakeResolve[i].ThisQInterval >= 0) mDNS_StopQuery(m, &intf->NetWakeResolve[i]);
intf->NetWakeResolve[i].ThisQInterval = -1;
if (sps[i])
{
LogSPS("BeginSleepProcessing: %-6s Found Sleep Proxy Server %d TTL %d %s", intf->ifname, i, sps[i]->resrec.rroriginalttl, CRDisplayString(m, sps[i]));
mDNS_SetupQuestion(&intf->NetWakeResolve[i], intf->InterfaceID, &sps[i]->resrec.rdata->u.name, kDNSType_SRV, NetWakeResolve, intf);
intf->NetWakeResolve[i].ReturnIntermed = mDNStrue;
mDNS_StartQuery_internal(m, &intf->NetWakeResolve[i]);
}
}
}
}
}
}
if (SendGoodbyes) {
LogSPS("BeginSleepProcessing: Not registering with Sleep Proxy Server");
SendSleepGoodbyes(m);
}
}
mDNSexport void mDNSCoreMachineSleep(mDNS *const m, mDNSBool sleep)
{
AuthRecord *rr;
LogSPS("%s (old state %d) at %ld", sleep ? "Sleeping" : "Waking", m->SleepState, m->timenow);
if (sleep && !m->SleepState) {
mDNS_Lock(m);
if (m->SPSSocket)
{
mDNSu8 oldstate = m->SPSState;
mDNS_DropLockBeforeCallback(); m->SPSState = 2;
if (oldstate == 1) mDNS_DeregisterService(m, &m->SPSRecords);
mDNS_ReclaimLockAfterCallback();
}
m->SleepState = SleepState_Transferring;
if (m->SystemWakeOnLANEnabled && m->DelaySleep)
{
LogSPS("mDNSCoreMachineSleep: Re-sleeping immediately after waking; will delay for %d ticks", m->DelaySleep - m->timenow);
m->SleepLimit = NonZeroTime(m->DelaySleep + mDNSPlatformOneSecond * 10);
}
else
{
m->DelaySleep = 0;
m->SleepLimit = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 10);
BeginSleepProcessing(m);
}
#ifndef UNICAST_DISABLED
SuspendLLQs(m);
#endif
mDNS_Unlock(m);
#if APPLE_OSX_mDNSResponder
RemoveAutoTunnel6Record(m);
#endif
LogSPS("mDNSCoreMachineSleep: m->SleepState %d (%s) seq %d", m->SleepState,
m->SleepState == SleepState_Transferring ? "Transferring" :
m->SleepState == SleepState_Sleeping ? "Sleeping" : "?", m->SleepSeqNum);
}
else if (!sleep) {
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *cr;
NetworkInterfaceInfo *intf;
mDNS_Lock(m);
m->SleepLimit = 0;
if (m->SleepState != SleepState_Awake)
{
m->SleepState = SleepState_Awake;
m->SleepSeqNum++;
if (m->SentSleepProxyRegistration) {
m->SentSleepProxyRegistration = mDNSfalse;
m->AnnounceOwner = NonZeroTime(m->timenow + 60 * mDNSPlatformOneSecond);
}
m->DelaySleep = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 16);
}
if (m->SPSState == 3)
{
m->SPSState = 0;
mDNSCoreBeSleepProxyServer_internal(m, m->SPSType, m->SPSPortability, m->SPSMarginalPower, m->SPSTotalPower);
}
for (rr = m->ResourceRecords; rr; rr=rr->next) rr->updateid = zeroID;
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next)) intf->NextSPSAttempt = -1;
mDNSCoreRestartQueries(m);
m->NextSRVUpdate = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
LogInfo("mDNSCoreMachineSleep waking: NextSRVUpdate in %d %d", m->NextSRVUpdate - m->timenow, m->timenow);
FORALL_CACHERECORDS(slot, cg, cr)
mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForWake);
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (AuthRecord_uDNS(rr))
{
ActivateUnicastRegistration(m, rr);
}
else
{
if (rr->resrec.RecordType == kDNSRecordTypeVerified && !rr->DependentOn) rr->resrec.RecordType = kDNSRecordTypeUnique;
rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
rr->AnnounceCount = InitialAnnounceCount;
rr->SendNSECNow = mDNSNULL;
InitializeLastAPTime(m, rr);
}
m->retryIntervalGetAddr = NATMAP_INIT_RETRY;
m->retryGetAddr = m->timenow + mDNSPlatformOneSecond * 5;
LogInfo("mDNSCoreMachineSleep: retryGetAddr in %d %d", m->retryGetAddr - m->timenow, m->timenow);
RecreateNATMappings(m);
mDNS_Unlock(m);
}
}
mDNSexport mDNSBool mDNSCoreReadyForSleep(mDNS *m, mDNSs32 now)
{
DNSQuestion *q;
AuthRecord *rr;
NetworkInterfaceInfo *intf;
mDNS_Lock(m);
if (m->DelaySleep) goto notready;
if (m->SleepLimit - now > 0 && m->NextScheduledSPRetry - now > 0) goto notready;
m->NextScheduledSPRetry = now + 0x40000000UL;
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
if (intf->NextSPSAttempt >= 0)
{
if (now - intf->NextSPSAttemptTime >= 0)
{
LogSPS("mDNSCoreReadyForSleep: retrying for %s SPS %d try %d",
intf->ifname, intf->NextSPSAttempt/3, intf->NextSPSAttempt);
SendSPSRegistration(m, intf, zeroID);
}
else
if (m->NextScheduledSPRetry - intf->NextSPSAttemptTime > 0)
m->NextScheduledSPRetry = intf->NextSPSAttemptTime;
}
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
{
int sps = (intf->NextSPSAttempt == 0) ? 0 : (intf->NextSPSAttempt-1)/3;
if (intf->NetWakeResolve[sps].ThisQInterval >= 0)
{
LogSPS("mDNSCoreReadyForSleep: waiting for SPS Resolve %s %##s (%s)",
intf->ifname, intf->NetWakeResolve[sps].qname.c, DNSTypeName(intf->NetWakeResolve[sps].qtype));
goto spsnotready;
}
}
for (rr = m->ResourceRecords; rr; rr = rr->next)
if (!AuthRecord_uDNS(rr))
if (!mDNSOpaque16IsZero(rr->updateid))
{ LogSPS("mDNSCoreReadyForSleep: waiting for SPS Update ID %d %s", mDNSVal16(rr->updateid), ARDisplayString(m,rr)); goto spsnotready; }
for (q = m->Questions; q; q = q->next)
if (!mDNSOpaque16IsZero(q->TargetQID) && q->LongLived && q->ReqLease == 0 && q->tcp)
{
LogSPS("mDNSCoreReadyForSleep: waiting for LLQ %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
goto notready;
}
for (rr = m->ResourceRecords; rr; rr = rr->next)
if (AuthRecord_uDNS(rr))
{
if (rr->state == regState_Refresh && rr->tcp)
{ LogSPS("mDNSCoreReadyForSleep: waiting for Record Update ID %d %s", mDNSVal16(rr->updateid), ARDisplayString(m,rr)); goto notready; }
#if APPLE_OSX_mDNSResponder
if (!RecordReadyForSleep(m, rr)) { LogSPS("mDNSCoreReadyForSleep: waiting for %s", ARDisplayString(m, rr)); goto notready; }
#endif
}
mDNS_Unlock(m);
return mDNStrue;
spsnotready:
if (now - m->SleepLimit >= 0)
{
LogMsg("Failed to register with SPS, now sending goodbyes");
for (intf = GetFirstActiveInterface(m->HostInterfaces); intf; intf = GetFirstActiveInterface(intf->next))
if (intf->NetWakeBrowse.ThisQInterval >= 0)
{
LogSPS("ReadyForSleep mDNS_DeactivateNetWake %s %##s (%s)",
intf->ifname, intf->NetWakeResolve[0].qname.c, DNSTypeName(intf->NetWakeResolve[0].qtype));
mDNS_DeactivateNetWake_internal(m, intf);
}
for (rr = m->ResourceRecords; rr; rr = rr->next)
if (!AuthRecord_uDNS(rr))
if (!mDNSOpaque16IsZero(rr->updateid))
{
LogSPS("ReadyForSleep clearing updateid for %s", ARDisplayString(m, rr));
rr->updateid = zeroID;
}
m->SleepLimit = now + mDNSPlatformOneSecond * 1;
SendSleepGoodbyes(m);
}
notready:
mDNS_Unlock(m);
return mDNSfalse;
}
mDNSexport mDNSs32 mDNSCoreIntervalToNextWake(mDNS *const m, mDNSs32 now)
{
AuthRecord *ar;
mDNSs32 e = now + (120 * 60 * mDNSPlatformOneSecond);
NATTraversalInfo *nat;
for (nat = m->NATTraversals; nat; nat=nat->next)
if (nat->Protocol && nat->ExpiryTime && nat->ExpiryTime - now > mDNSPlatformOneSecond*4)
{
mDNSs32 t = nat->ExpiryTime - (nat->ExpiryTime - now) / 10; if (e - t > 0) e = t;
LogSPS("ComputeWakeTime: %p %s Int %5d Ext %5d Err %d Retry %5d Interval %5d Expire %5d Wake %5d",
nat, nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP",
mDNSVal16(nat->IntPort), mDNSVal16(nat->ExternalPort), nat->Result,
nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0,
nat->retryInterval / mDNSPlatformOneSecond,
nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0,
(t - now) / mDNSPlatformOneSecond);
}
for (ar = m->ResourceRecords; ar; ar = ar->next)
if (ar->expire && ar->expire - now > mDNSPlatformOneSecond*4)
{
mDNSs32 t = ar->expire - (ar->expire - now) / 10; if (e - t > 0) e = t;
LogSPS("ComputeWakeTime: %p Int %7d Next %7d Expire %7d Wake %7d %s",
ar, ar->ThisAPInterval / mDNSPlatformOneSecond,
(ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond,
ar->expire ? (ar->expire - now) / mDNSPlatformOneSecond : 0,
(t - now) / mDNSPlatformOneSecond, ARDisplayString(m, ar));
}
return(e - now);
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Packet Reception Functions
#endif
#define MustSendRecord(RR) ((RR)->NR_AnswerTo || (RR)->NR_AdditionalTo)
mDNSlocal mDNSu8 *GenerateUnicastResponse(const DNSMessage *const query, const mDNSu8 *const end,
const mDNSInterfaceID InterfaceID, mDNSBool LegacyQuery, DNSMessage *const response, AuthRecord *ResponseRecords)
{
mDNSu8 *responseptr = response->data;
const mDNSu8 *const limit = response->data + sizeof(response->data);
const mDNSu8 *ptr = query->data;
AuthRecord *rr;
mDNSu32 maxttl = 0x70000000;
int i;
InitializeDNSMessage(&response->h, query->h.id, ResponseFlags);
if (LegacyQuery)
{
maxttl = kStaticCacheTTL;
for (i=0; i<query->h.numQuestions; i++) {
DNSQuestion q;
ptr = getQuestion(query, ptr, end, InterfaceID, &q); if (!ptr) return(mDNSNULL);
for (rr=ResponseRecords; rr; rr=rr->NextResponse) {
if (rr->NR_AnswerTo == ptr) { responseptr = putQuestion(response, responseptr, limit, &q.qname, q.qtype, q.qclass);
if (!responseptr) { debugf("GenerateUnicastResponse: Ran out of space for questions!"); return(mDNSNULL); }
break; }
}
}
if (response->h.numQuestions == 0) { LogMsg("GenerateUnicastResponse: ERROR! Why no questions?"); return(mDNSNULL); }
}
for (rr=ResponseRecords; rr; rr=rr->NextResponse)
if (rr->NR_AnswerTo)
{
mDNSu8 *p = PutResourceRecordTTL(response, responseptr, &response->h.numAnswers, &rr->resrec,
maxttl < rr->resrec.rroriginalttl ? maxttl : rr->resrec.rroriginalttl);
if (p) responseptr = p;
else { debugf("GenerateUnicastResponse: Ran out of space for answers!"); response->h.flags.b[0] |= kDNSFlag0_TC; }
}
for (rr=ResponseRecords; rr; rr=rr->NextResponse)
if (rr->NR_AdditionalTo && !rr->NR_AnswerTo)
{
mDNSu8 *p = PutResourceRecordTTL(response, responseptr, &response->h.numAdditionals, &rr->resrec,
maxttl < rr->resrec.rroriginalttl ? maxttl : rr->resrec.rroriginalttl);
if (p) responseptr = p;
else debugf("GenerateUnicastResponse: No more space for additionals");
}
return(responseptr);
}
mDNSlocal int CompareRData(const AuthRecord *const our, const CacheRecord *const pkt)
{
mDNSu8 ourdata[256], *ourptr = ourdata, *ourend;
mDNSu8 pktdata[256], *pktptr = pktdata, *pktend;
if (!our) { LogMsg("CompareRData ERROR: our is NULL"); return(+1); }
if (!pkt) { LogMsg("CompareRData ERROR: pkt is NULL"); return(+1); }
ourend = putRData(mDNSNULL, ourdata, ourdata + sizeof(ourdata), &our->resrec);
pktend = putRData(mDNSNULL, pktdata, pktdata + sizeof(pktdata), &pkt->resrec);
while (ourptr < ourend && pktptr < pktend && *ourptr == *pktptr) { ourptr++; pktptr++; }
if (ourptr >= ourend && pktptr >= pktend) return(0);
if (ourptr >= ourend) return(-1); if (pktptr >= pktend) return(+1); if (*pktptr > *ourptr) return(-1); if (*pktptr < *ourptr) return(+1);
LogMsg("CompareRData ERROR: Invalid state");
return(-1);
}
mDNSlocal mDNSBool MatchDependentOn(const mDNS *const m, const CacheRecord *const pktrr, const AuthRecord *const master)
{
const AuthRecord *r1;
for (r1 = m->ResourceRecords; r1; r1=r1->next)
{
if (IdenticalResourceRecord(&r1->resrec, &pktrr->resrec))
{
const AuthRecord *r2 = r1;
while (r2->DependentOn) r2 = r2->DependentOn;
if (r2 == master) return(mDNStrue);
}
}
for (r1 = m->DuplicateRecords; r1; r1=r1->next)
{
if (IdenticalResourceRecord(&r1->resrec, &pktrr->resrec))
{
const AuthRecord *r2 = r1;
while (r2->DependentOn) r2 = r2->DependentOn;
if (r2 == master) return(mDNStrue);
}
}
return(mDNSfalse);
}
mDNSlocal const AuthRecord *FindRRSet(const mDNS *const m, const CacheRecord *const pktrr)
{
const AuthRecord *rr;
for (rr = m->ResourceRecords; rr; rr=rr->next)
{
if (IdenticalResourceRecord(&rr->resrec, &pktrr->resrec))
{
while (rr->RRSet && rr != rr->RRSet) rr = rr->RRSet;
return(rr);
}
}
return(mDNSNULL);
}
mDNSlocal mDNSBool PacketRRConflict(const mDNS *const m, const AuthRecord *const our, const CacheRecord *const pktrr)
{
if (!(our->resrec.RecordType & kDNSRecordTypeUniqueMask)) return(mDNSfalse);
if (our->DependentOn || MatchDependentOn(m, pktrr, our)) return(mDNSfalse);
else
{
const AuthRecord *ourset = our->RRSet ? our->RRSet : our;
const AuthRecord *pktset = FindRRSet(m, pktrr);
if (pktset == ourset) return(mDNSfalse);
if (our->WakeUp.HMAC.l[0] && pktset) return(mDNSfalse);
}
return(mDNStrue);
}
mDNSlocal void ResolveSimultaneousProbe(mDNS *const m, const DNSMessage *const query, const mDNSu8 *const end,
DNSQuestion *q, AuthRecord *our)
{
int i;
const mDNSu8 *ptr = LocateAuthorities(query, end);
mDNSBool FoundUpdate = mDNSfalse;
for (i = 0; i < query->h.numAuthorities; i++)
{
ptr = GetLargeResourceRecord(m, query, ptr, end, q->InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
if (!ptr) break;
if (m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && ResourceRecordAnswersQuestion(&m->rec.r.resrec, q))
{
FoundUpdate = mDNStrue;
if (PacketRRConflict(m, our, &m->rec.r))
{
int result = (int)our->resrec.rrclass - (int)m->rec.r.resrec.rrclass;
if (!result) result = (int)our->resrec.rrtype - (int)m->rec.r.resrec.rrtype;
if (!result) result = CompareRData(our, &m->rec.r);
if (result)
{
const char *const msg = (result < 0) ? "lost:" : (result > 0) ? "won: " : "tie: ";
LogMsg("ResolveSimultaneousProbe: %p Pkt Record: %08lX %s", q->InterfaceID, m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
LogMsg("ResolveSimultaneousProbe: %p Our Record %d %s %08lX %s", our->resrec.InterfaceID, our->ProbeCount, msg, our->resrec.rdatahash, ARDisplayString(m, our));
}
if (result < 0)
{
m->SuppressProbes = NonZeroTime(m->timenow + mDNSPlatformOneSecond);
our->ProbeCount = DefaultProbeCountForTypeUnique;
our->AnnounceCount = InitialAnnounceCount;
InitializeLastAPTime(m, our);
goto exit;
}
}
#if 0
else
{
LogMsg("ResolveSimultaneousProbe: %p Pkt Record: %08lX %s", q->InterfaceID, m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
LogMsg("ResolveSimultaneousProbe: %p Our Record %d ign: %08lX %s", our->resrec.InterfaceID, our->ProbeCount, our->resrec.rdatahash, ARDisplayString(m, our));
}
#endif
}
m->rec.r.resrec.RecordType = 0; }
if (!FoundUpdate)
LogInfo("ResolveSimultaneousProbe: %##s (%s): No Update Record found", our->resrec.name->c, DNSTypeName(our->resrec.rrtype));
exit:
m->rec.r.resrec.RecordType = 0; }
mDNSlocal CacheRecord *FindIdenticalRecordInCache(const mDNS *const m, const ResourceRecord *const pktrr)
{
mDNSu32 slot = HashSlot(pktrr->name);
CacheGroup *cg = CacheGroupForRecord(m, slot, pktrr);
CacheRecord *rr;
mDNSBool match;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
{
match = !pktrr->InterfaceID ? pktrr->rDNSServer == rr->resrec.rDNSServer : pktrr->InterfaceID == rr->resrec.InterfaceID;
if (match && IdenticalSameNameRecord(pktrr, &rr->resrec)) break;
}
return(rr);
}
mDNSlocal void ClearIdenticalProxyRecords(mDNS *const m, const OwnerOptData *const owner, AuthRecord *const thelist)
{
if (m->CurrentRecord)
LogMsg("ClearIdenticalProxyRecords ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = thelist;
while (m->CurrentRecord)
{
AuthRecord *const rr = m->CurrentRecord;
if (m->rec.r.resrec.InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&owner->HMAC, &rr->WakeUp.HMAC))
if (IdenticalResourceRecord(&rr->resrec, &m->rec.r.resrec))
{
LogSPS("ClearIdenticalProxyRecords: Removing %3d H-MAC %.6a I-MAC %.6a %d %d %s",
m->ProxyRecords, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, owner->seq, ARDisplayString(m, rr));
rr->WakeUp.HMAC = zeroEthAddr; rr->RequireGoodbye = mDNSfalse; mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
SetSPSProxyListChanged(m->rec.r.resrec.InterfaceID);
}
if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
}
}
mDNSlocal void ClearProxyRecords(mDNS *const m, const OwnerOptData *const owner, AuthRecord *const thelist)
{
if (m->CurrentRecord)
LogMsg("ClearProxyRecords ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = thelist;
while (m->CurrentRecord)
{
AuthRecord *const rr = m->CurrentRecord;
if (m->rec.r.resrec.InterfaceID == rr->resrec.InterfaceID && mDNSSameEthAddress(&owner->HMAC, &rr->WakeUp.HMAC))
if (owner->seq != rr->WakeUp.seq || m->timenow - rr->TimeRcvd > mDNSPlatformOneSecond * 60)
{
if (rr->AddressProxy.type == mDNSAddrType_IPv6)
{
#if MDNS_USE_Unsolicited_Neighbor_Advertisements
LogSPS("NDP Announcement -- Releasing traffic for H-MAC %.6a I-MAC %.6a %s",
&rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m,rr));
SendNDP(m, NDP_Adv, NDP_Override, rr, &rr->AddressProxy.ip.v6, &rr->WakeUp.IMAC, &AllHosts_v6, &AllHosts_v6_Eth);
#endif
}
LogSPS("ClearProxyRecords: Removing %3d AC %2d %02X H-MAC %.6a I-MAC %.6a %d %d %s",
m->ProxyRecords, rr->AnnounceCount, rr->resrec.RecordType,
&rr->WakeUp.HMAC, &rr->WakeUp.IMAC, rr->WakeUp.seq, owner->seq, ARDisplayString(m, rr));
if (rr->resrec.RecordType == kDNSRecordTypeDeregistering) rr->resrec.RecordType = kDNSRecordTypeShared;
rr->WakeUp.HMAC = zeroEthAddr; rr->RequireGoodbye = mDNSfalse; mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
SetSPSProxyListChanged(m->rec.r.resrec.InterfaceID);
}
if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
}
}
mDNSlocal mDNSu8 *ProcessQuery(mDNS *const m, const DNSMessage *const query, const mDNSu8 *const end,
const mDNSAddr *srcaddr, const mDNSInterfaceID InterfaceID, mDNSBool LegacyQuery, mDNSBool QueryWasMulticast,
mDNSBool QueryWasLocalUnicast, DNSMessage *const response)
{
mDNSBool FromLocalSubnet = srcaddr && mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
AuthRecord *ResponseRecords = mDNSNULL;
AuthRecord **nrp = &ResponseRecords;
CacheRecord *ExpectedAnswers = mDNSNULL; CacheRecord **eap = &ExpectedAnswers;
DNSQuestion *DupQuestions = mDNSNULL; DNSQuestion **dqp = &DupQuestions;
mDNSs32 delayresponse = 0;
mDNSBool SendLegacyResponse = mDNSfalse;
const mDNSu8 *ptr;
mDNSu8 *responseptr = mDNSNULL;
AuthRecord *rr;
int i;
ptr = LocateOptRR(query, end, DNSOpt_OwnerData_ID_Space);
if (ptr)
{
ptr = GetLargeResourceRecord(m, query, ptr, end, InterfaceID, kDNSRecordTypePacketAdd, &m->rec);
if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
{
const rdataOPT *opt;
const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
for (opt = &m->rec.r.resrec.rdata->u.opt[0]; opt < e; opt++)
if (opt->opt == kDNSOpt_Owner && opt->u.owner.vers == 0 && opt->u.owner.HMAC.l[0])
{
ClearProxyRecords(m, &opt->u.owner, m->DuplicateRecords);
ClearProxyRecords(m, &opt->u.owner, m->ResourceRecords);
}
}
m->rec.r.resrec.RecordType = 0; }
ptr = query->data;
for (i=0; i<query->h.numQuestions; i++) {
mDNSBool QuestionNeedsMulticastResponse;
int NumAnswersForThisQuestion = 0;
AuthRecord *NSECAnswer = mDNSNULL;
DNSQuestion pktq, *q;
ptr = getQuestion(query, ptr, end, InterfaceID, &pktq); if (!ptr) goto exit;
QuestionNeedsMulticastResponse = QueryWasMulticast && !LegacyQuery && !(pktq.qclass & kDNSQClass_UnicastResponse);
pktq.qclass &= ~kDNSQClass_UnicastResponse;
if (m->CurrentRecord)
LogMsg("ProcessQuery ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
rr = m->CurrentRecord;
m->CurrentRecord = rr->next;
if (AnyTypeRecordAnswersQuestion(&rr->resrec, &pktq) && (QueryWasMulticast || QueryWasLocalUnicast || rr->AllowRemoteQuery))
{
if (RRTypeAnswersQuestionType(&rr->resrec, pktq.qtype))
{
if (rr->resrec.RecordType == kDNSRecordTypeUnique)
ResolveSimultaneousProbe(m, query, end, &pktq, rr);
else if (ResourceRecordIsValidAnswer(rr))
{
NumAnswersForThisQuestion++;
if (QuestionNeedsMulticastResponse || (!FromLocalSubnet && QueryWasMulticast && !LegacyQuery))
{
if (m->timenow - (rr->LastMCTime + mDNSPlatformOneSecond) >= 0 ||
(rr->LastMCInterface != mDNSInterfaceMark && rr->LastMCInterface != InterfaceID))
rr->NR_AnswerTo = (mDNSu8*)~0;
}
else if (!rr->NR_AnswerTo) rr->NR_AnswerTo = LegacyQuery ? ptr : (mDNSu8*)~1;
}
}
else if ((rr->resrec.RecordType & kDNSRecordTypeActiveUniqueMask) && ResourceRecordIsValidAnswer(rr))
{
if (!NSECAnswer) NSECAnswer = rr;
}
}
}
if (NumAnswersForThisQuestion == 0 && NSECAnswer)
{
NumAnswersForThisQuestion++;
NSECAnswer->SendNSECNow = InterfaceID;
m->NextScheduledResponse = m->timenow;
}
if (NumAnswersForThisQuestion == 0) delayresponse = mDNSPlatformOneSecond;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (QuestionNeedsMulticastResponse)
#else
if (QuestionNeedsMulticastResponse && !(query->h.flags.b[0] & kDNSFlag0_TC))
#endif
{
const mDNSu32 slot = HashSlot(&pktq.qname);
CacheGroup *cg = CacheGroupForName(m, slot, pktq.qnamehash, &pktq.qname);
CacheRecord *cr;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (!(query->h.flags.b[0] & kDNSFlag0_TC))
#endif
for (cr = cg ? cg->members : mDNSNULL; cr; cr=cr->next)
if (SameNameRecordAnswersQuestion(&cr->resrec, &pktq) && cr->resrec.rdlength <= SmallRecordLimit)
if (!cr->NextInKAList && eap != &cr->NextInKAList)
{
*eap = cr;
eap = &cr->NextInKAList;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (cr->MPUnansweredQ == 0 || m->timenow - cr->MPLastUnansweredQT >= mDNSPlatformOneSecond)
{
cr->MPUnansweredQ++;
cr->MPLastUnansweredQT = m->timenow;
cr->MPExpectingKA = mDNStrue;
}
#endif
}
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (!(query->h.flags.b[0] & kDNSFlag0_TC))
#endif
for (q = m->Questions; q; q=q->next)
if (!q->Target.type && ActiveQuestion(q) && m->timenow - q->LastQTxTime > mDNSPlatformOneSecond / 4)
if (!q->InterfaceID || q->InterfaceID == InterfaceID)
if (q->NextInDQList == mDNSNULL && dqp != &q->NextInDQList)
if (q->qtype == pktq.qtype &&
q->qclass == pktq.qclass &&
q->qnamehash == pktq.qnamehash && SameDomainName(&q->qname, &pktq.qname))
{ *dqp = q; dqp = &q->NextInDQList; }
}
}
for (rr = m->ResourceRecords; rr; rr=rr->next) if (rr->NR_AnswerTo) AddRecordToResponseList(&nrp, rr, mDNSNULL);
AddAdditionalsToResponseList(m, ResponseRecords, &nrp, InterfaceID);
for (i=0; i<query->h.numAnswers; i++) {
CacheRecord *ourcacherr;
ptr = GetLargeResourceRecord(m, query, ptr, end, InterfaceID, kDNSRecordTypePacketAns, &m->rec);
if (!ptr) goto exit;
if (m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative)
{
for (rr=ResponseRecords; rr; rr=rr->NextResponse)
if (MustSendRecord(rr) && ShouldSuppressKnownAnswer(&m->rec.r, rr))
{ rr->NR_AnswerTo = mDNSNULL; rr->NR_AdditionalTo = mDNSNULL; }
for (rr=m->ResourceRecords; rr; rr=rr->next)
{
if (rr->ImmedAnswer == InterfaceID && ShouldSuppressKnownAnswer(&m->rec.r, rr))
{
if (srcaddr->type == mDNSAddrType_IPv4)
{
if (mDNSSameIPv4Address(rr->v4Requester, srcaddr->ip.v4)) rr->v4Requester = zerov4Addr;
}
else if (srcaddr->type == mDNSAddrType_IPv6)
{
if (mDNSSameIPv6Address(rr->v6Requester, srcaddr->ip.v6)) rr->v6Requester = zerov6Addr;
}
if (mDNSIPv4AddressIsZero(rr->v4Requester) && mDNSIPv6AddressIsZero(rr->v6Requester))
{
rr->ImmedAnswer = mDNSNULL;
rr->ImmedUnicast = mDNSfalse;
#if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
LogMsg("Suppressed after%4d: %s", m->timenow - rr->ImmedAnswerMarkTime, ARDisplayString(m, rr));
#endif
}
}
}
ourcacherr = FindIdenticalRecordInCache(m, &m->rec.r.resrec);
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (ourcacherr && ourcacherr->MPExpectingKA && m->timenow - ourcacherr->MPLastUnansweredQT < mDNSPlatformOneSecond)
{
ourcacherr->MPUnansweredKA++;
ourcacherr->MPExpectingKA = mDNSfalse;
}
#endif
eap = &ExpectedAnswers;
while (*eap)
{
CacheRecord *cr = *eap;
if (cr->resrec.InterfaceID == InterfaceID && IdenticalResourceRecord(&m->rec.r.resrec, &cr->resrec))
{ *eap = cr->NextInKAList; cr->NextInKAList = mDNSNULL; }
else eap = &cr->NextInKAList;
}
if (!ourcacherr)
{
dqp = &DupQuestions;
while (*dqp)
{
DNSQuestion *q = *dqp;
if (ResourceRecordAnswersQuestion(&m->rec.r.resrec, q))
{ *dqp = q->NextInDQList; q->NextInDQList = mDNSNULL; }
else dqp = &q->NextInDQList;
}
}
}
m->rec.r.resrec.RecordType = 0; }
for (rr=ResponseRecords; rr; rr=rr->NextResponse)
if (rr->NR_AdditionalTo && !MustSendRecord(rr->NR_AdditionalTo))
{ rr->NR_AnswerTo = mDNSNULL; rr->NR_AdditionalTo = mDNSNULL; }
for (rr=ResponseRecords; rr; rr=rr->NextResponse)
{
if (rr->NR_AnswerTo)
{
mDNSBool SendMulticastResponse = mDNSfalse; mDNSBool SendUnicastResponse = mDNSfalse;
if (m->timenow - (rr->LastMCTime + TicksTTL(rr)/4) >= 0)
{
SendMulticastResponse = mDNStrue;
if (rr->NR_AnswerTo == (mDNSu8*)~1) rr->NR_AnswerTo = (mDNSu8*)~0;
}
if (rr->NR_AnswerTo == (mDNSu8*)~0) SendMulticastResponse = mDNStrue;
else if (rr->NR_AnswerTo == (mDNSu8*)~1) SendUnicastResponse = mDNStrue;
else if (rr->NR_AnswerTo) SendLegacyResponse = mDNStrue;
if (SendMulticastResponse || SendUnicastResponse)
{
#if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
rr->ImmedAnswerMarkTime = m->timenow;
#endif
m->NextScheduledResponse = m->timenow;
if (rr->ImmedAnswer && rr->ImmedAnswer != InterfaceID)
rr->ImmedAnswer = mDNSInterfaceMark;
else
{
rr->ImmedAnswer = InterfaceID; if (SendUnicastResponse) rr->ImmedUnicast = mDNStrue;
if (srcaddr->type == mDNSAddrType_IPv4)
{
if (mDNSIPv4AddressIsZero(rr->v4Requester)) rr->v4Requester = srcaddr->ip.v4;
else if (!mDNSSameIPv4Address(rr->v4Requester, srcaddr->ip.v4)) rr->v4Requester = onesIPv4Addr;
}
else if (srcaddr->type == mDNSAddrType_IPv6)
{
if (mDNSIPv6AddressIsZero(rr->v6Requester)) rr->v6Requester = srcaddr->ip.v6;
else if (!mDNSSameIPv6Address(rr->v6Requester, srcaddr->ip.v6)) rr->v6Requester = onesIPv6Addr;
}
}
}
if (query->h.flags.b[0] & kDNSFlag0_TC) delayresponse = mDNSPlatformOneSecond * 20; else if (rr->resrec.RecordType == kDNSRecordTypeShared) delayresponse = mDNSPlatformOneSecond; }
else if (rr->NR_AdditionalTo && rr->NR_AdditionalTo->NR_AnswerTo == (mDNSu8*)~0)
{
rr->ImmedAdditional = InterfaceID;
}
}
if (delayresponse && (!m->SuppressSending || (m->SuppressSending - m->timenow) < (delayresponse + 49) / 50))
{
#if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
mDNSs32 oldss = m->SuppressSending;
if (oldss && delayresponse)
LogMsg("Current SuppressSending delay%5ld; require%5ld", m->SuppressSending - m->timenow, (delayresponse + 49) / 50);
#endif
m->SuppressSending = m->timenow + (delayresponse + (mDNSs32)mDNSRandom((mDNSu32)mDNSPlatformOneSecond*5) + 49) / 50;
if (m->SuppressSending == 0) m->SuppressSending = 1;
#if MDNS_LOG_ANSWER_SUPPRESSION_TIMES
if (oldss && delayresponse)
LogMsg("Set SuppressSending to %5ld", m->SuppressSending - m->timenow);
#endif
}
if (SendLegacyResponse)
responseptr = GenerateUnicastResponse(query, end, InterfaceID, LegacyQuery, response, ResponseRecords);
exit:
m->rec.r.resrec.RecordType = 0;
while (ResponseRecords)
{
rr = ResponseRecords;
ResponseRecords = rr->NextResponse;
rr->NextResponse = mDNSNULL;
rr->NR_AnswerTo = mDNSNULL;
rr->NR_AdditionalTo = mDNSNULL;
}
while (ExpectedAnswers)
{
CacheRecord *cr = ExpectedAnswers;
ExpectedAnswers = cr->NextInKAList;
cr->NextInKAList = mDNSNULL;
if (!(query->h.flags.b[0] & kDNSFlag0_TC))
if (cr->UnansweredQueries == 0 || m->timenow - cr->LastUnansweredTime >= mDNSPlatformOneSecond)
{
cr->UnansweredQueries++;
cr->LastUnansweredTime = m->timenow;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (cr->UnansweredQueries > 1)
debugf("ProcessQuery: (!TC) UAQ %lu MPQ %lu MPKA %lu %s",
cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
#endif
SetNextCacheCheckTimeForRecord(m, cr);
}
if (cr->UnansweredQueries >= MaxUnansweredQueries)
{
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
if (RRExpireTime(cr) - m->timenow > 4 * mDNSPlatformOneSecond)
debugf("ProcessQuery: (Max) UAQ %lu MPQ %lu MPKA %lu mDNS_Reconfirm() for %s",
cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
#endif
mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
}
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
else if (cr->MPUnansweredQ * 4 > cr->MPUnansweredKA * 5 + 8)
{
mDNSu32 remain = (mDNSu32)(RRExpireTime(cr) - m->timenow) / 4;
if (remain > 240 * (mDNSu32)mDNSPlatformOneSecond)
remain = 240 * (mDNSu32)mDNSPlatformOneSecond;
if (RRExpireTime(cr) - m->timenow > 4 * mDNSPlatformOneSecond)
debugf("ProcessQuery: (MPQ) UAQ %lu MPQ %lu MPKA %lu mDNS_Reconfirm() for %s",
cr->UnansweredQueries, cr->MPUnansweredQ, cr->MPUnansweredKA, CRDisplayString(m, cr));
if (remain <= 60 * (mDNSu32)mDNSPlatformOneSecond)
cr->UnansweredQueries++; cr->MPUnansweredQ = 0; cr->MPUnansweredKA = 0;
cr->MPExpectingKA = mDNSfalse;
if (remain < kDefaultReconfirmTimeForNoAnswer)
remain = kDefaultReconfirmTimeForNoAnswer;
mDNS_Reconfirm_internal(m, cr, remain);
}
#endif
}
while (DupQuestions)
{
DNSQuestion *q = DupQuestions;
DupQuestions = q->NextInDQList;
q->NextInDQList = mDNSNULL;
i = RecordDupSuppressInfo(q->DupSuppress, m->timenow, InterfaceID, srcaddr->type);
debugf("ProcessQuery: Recorded DSI for %##s (%s) on %p/%s %d", q->qname.c, DNSTypeName(q->qtype), InterfaceID,
srcaddr->type == mDNSAddrType_IPv4 ? "v4" : "v6", i);
}
return(responseptr);
}
mDNSlocal void mDNSCoreReceiveQuery(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *const end,
const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
const mDNSInterfaceID InterfaceID)
{
mDNSu8 *responseend = mDNSNULL;
mDNSBool QueryWasLocalUnicast = srcaddr && dstaddr &&
!mDNSAddrIsDNSMulticast(dstaddr) && mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
if (!InterfaceID && dstaddr && mDNSAddrIsDNSMulticast(dstaddr))
{
LogMsg("Ignoring Query from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
"%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes (Multicast, but no InterfaceID)",
srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data);
return;
}
verbosedebugf("Received Query from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
"%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes",
srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data);
responseend = ProcessQuery(m, msg, end, srcaddr, InterfaceID,
!mDNSSameIPPort(srcport, MulticastDNSPort), mDNSAddrIsDNSMulticast(dstaddr), QueryWasLocalUnicast, &m->omsg);
if (responseend) {
debugf("Unicast Response: %d Question%s, %d Answer%s, %d Additional%s to %#-15a:%d on %p/%ld",
m->omsg.h.numQuestions, m->omsg.h.numQuestions == 1 ? "" : "s",
m->omsg.h.numAnswers, m->omsg.h.numAnswers == 1 ? "" : "s",
m->omsg.h.numAdditionals, m->omsg.h.numAdditionals == 1 ? "" : "s",
srcaddr, mDNSVal16(srcport), InterfaceID, srcaddr->type);
mDNSSendDNSMessage(m, &m->omsg, responseend, InterfaceID, mDNSNULL, srcaddr, srcport, mDNSNULL, mDNSNULL);
}
}
#if 0
mDNSlocal mDNSBool TrustedSource(const mDNS *const m, const mDNSAddr *const srcaddr)
{
DNSServer *s;
(void)m; (void)srcaddr; for (s = m->DNSServers; s; s = s->next)
if (mDNSSameAddress(srcaddr, &s->addr)) return(mDNStrue);
return(mDNSfalse);
}
#endif
struct UDPSocket_struct
{
mDNSIPPort port; };
mDNSlocal DNSQuestion *ExpectingUnicastResponseForQuestion(const mDNS *const m, const mDNSIPPort port, const mDNSOpaque16 id, const DNSQuestion *const question, mDNSBool tcp)
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next)
{
if (!tcp && !q->LocalSocket) continue;
if (mDNSSameIPPort(tcp ? q->tcpSrcPort : q->LocalSocket->port, port) &&
mDNSSameOpaque16(q->TargetQID, id) &&
q->qtype == question->qtype &&
q->qclass == question->qclass &&
q->qnamehash == question->qnamehash &&
SameDomainName(&q->qname, &question->qname))
return(q);
}
return(mDNSNULL);
}
mDNSlocal DNSQuestion *ExpectingUnicastResponseForRecord(mDNS *const m,
const mDNSAddr *const srcaddr, const mDNSBool SrcLocal, const mDNSIPPort port, const mDNSOpaque16 id, const CacheRecord *const rr, mDNSBool tcp)
{
DNSQuestion *q;
(void)id;
(void)srcaddr;
if (rr->resrec.InterfaceID) return mDNSNULL;
for (q = m->Questions; q; q=q->next)
{
if (!q->DuplicateOf && UnicastResourceRecordAnswersQuestion(&rr->resrec, q))
{
if (!mDNSOpaque16IsZero(q->TargetQID))
{
debugf("ExpectingUnicastResponseForRecord msg->h.id %d q->TargetQID %d for %s", mDNSVal16(id), mDNSVal16(q->TargetQID), CRDisplayString(m, rr));
if (mDNSSameOpaque16(q->TargetQID, id))
{
mDNSIPPort srcp;
if (!tcp)
{
srcp = q->LocalSocket ? q->LocalSocket->port : zeroIPPort;
}
else
{
srcp = q->tcpSrcPort;
}
if (mDNSSameIPPort(srcp, port)) return(q);
LogInfo("WARNING: Ignoring suspect uDNS response for %##s (%s) [q->Target %#a:%d] from %#a:%d %s",
q->qname.c, DNSTypeName(q->qtype), &q->Target, mDNSVal16(srcp), srcaddr, mDNSVal16(port), CRDisplayString(m, rr));
return(mDNSNULL);
}
}
else
{
if (SrcLocal && q->ExpectUnicastResp && (mDNSu32)(m->timenow - q->ExpectUnicastResp) < (mDNSu32)(mDNSPlatformOneSecond*2))
return(q);
}
}
}
return(mDNSNULL);
}
mDNSlocal mDNSu16 GetRDLengthMem(const ResourceRecord *const rr)
{
switch (rr->rrtype)
{
case kDNSType_SOA: return sizeof(rdataSOA);
case kDNSType_RP: return sizeof(rdataRP);
case kDNSType_PX: return sizeof(rdataPX);
case kDNSType_NSEC:return sizeof(rdataNSEC);
default: return rr->rdlength;
}
}
mDNSexport CacheRecord *CreateNewCacheEntry(mDNS *const m, const mDNSu32 slot, CacheGroup *cg, mDNSs32 delay)
{
CacheRecord *rr = mDNSNULL;
mDNSu16 RDLength = GetRDLengthMem(&m->rec.r.resrec);
if (!m->rec.r.resrec.InterfaceID) debugf("CreateNewCacheEntry %s", CRDisplayString(m, &m->rec.r));
if (!cg) cg = GetCacheGroup(m, slot, &m->rec.r.resrec); if (cg) rr = GetCacheRecord(m, cg, RDLength); if (!rr) NoCacheAnswer(m, &m->rec.r);
else
{
RData *saveptr = rr->resrec.rdata; *rr = m->rec.r; rr->resrec.rdata = saveptr; rr->resrec.name = cg->name; rr->DelayDelivery = delay;
if (rr->resrec.rdata == (RData*)&rr->smallrdatastorage && RDLength > InlineCacheRDSize)
LogMsg("rr->resrec.rdata == &rr->rdatastorage but length > InlineCacheRDSize %##s", m->rec.r.resrec.name->c);
else if (rr->resrec.rdata != (RData*)&rr->smallrdatastorage && RDLength <= InlineCacheRDSize)
LogMsg("rr->resrec.rdata != &rr->rdatastorage but length <= InlineCacheRDSize %##s", m->rec.r.resrec.name->c);
if (RDLength > InlineCacheRDSize)
mDNSPlatformMemCopy(rr->resrec.rdata, m->rec.r.resrec.rdata, sizeofRDataHeader + RDLength);
rr->next = mDNSNULL; *(cg->rrcache_tail) = rr; cg->rrcache_tail = &(rr->next);
CacheRecordAdd(m, rr); }
return(rr);
}
mDNSlocal void RefreshCacheRecord(mDNS *const m, CacheRecord *rr, mDNSu32 ttl)
{
rr->TimeRcvd = m->timenow;
rr->resrec.rroriginalttl = ttl;
rr->UnansweredQueries = 0;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
rr->MPUnansweredQ = 0;
rr->MPUnansweredKA = 0;
rr->MPExpectingKA = mDNSfalse;
#endif
SetNextCacheCheckTimeForRecord(m, rr);
}
mDNSexport void GrantCacheExtensions(mDNS *const m, DNSQuestion *q, mDNSu32 lease)
{
CacheRecord *rr;
const mDNSu32 slot = HashSlot(&q->qname);
CacheGroup *cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (rr->CRActiveQuestion == q)
{
RefreshCacheRecord(m, rr, lease);
}
}
mDNSlocal mDNSu32 GetEffectiveTTL(const uDNS_LLQType LLQType, mDNSu32 ttl) {
if (LLQType == uDNS_LLQ_Entire) ttl = kLLQ_DefLease;
else if (LLQType == uDNS_LLQ_Events)
{
if (ttl == 0xFFFFFFFF) ttl = 0;
else ttl = kLLQ_DefLease;
}
else {
if (ttl > 0x60000000UL / mDNSPlatformOneSecond) ttl = 0x60000000UL / mDNSPlatformOneSecond;
ttl += ttl/4 + 2;
if (ttl < 15) ttl = 15;
}
return ttl;
}
mDNSlocal void mDNSCoreReceiveResponse(mDNS *const m,
const DNSMessage *const response, const mDNSu8 *end,
const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
const mDNSInterfaceID InterfaceID)
{
int i;
mDNSBool ResponseMCast = dstaddr && mDNSAddrIsDNSMulticast(dstaddr);
mDNSBool ResponseSrcLocal = !srcaddr || mDNS_AddressIsLocalSubnet(m, InterfaceID, srcaddr);
DNSQuestion *llqMatch = mDNSNULL;
uDNS_LLQType LLQType = uDNS_recvLLQResponse(m, response, end, srcaddr, srcport, &llqMatch);
CacheRecord *CacheFlushRecords = (CacheRecord*)1;
CacheRecord **cfp = &CacheFlushRecords;
int firstauthority = response->h.numAnswers;
int firstadditional = firstauthority + response->h.numAuthorities;
int totalrecords = firstadditional + response->h.numAdditionals;
const mDNSu8 *ptr = response->data;
DNSServer *uDNSServer = mDNSNULL;
debugf("Received Response from %#-15a addressed to %#-15a on %p with "
"%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes LLQType %d",
srcaddr, dstaddr, InterfaceID,
response->h.numQuestions, response->h.numQuestions == 1 ? ", " : "s,",
response->h.numAnswers, response->h.numAnswers == 1 ? ", " : "s,",
response->h.numAuthorities, response->h.numAuthorities == 1 ? "y, " : "ies,",
response->h.numAdditionals, response->h.numAdditionals == 1 ? " " : "s", end - response->data, LLQType);
if (!InterfaceID && (response->h.flags.b[0] & kDNSFlag0_TC)) return;
if (LLQType == uDNS_LLQ_Ignore) return;
if (ResponseMCast || LLQType == uDNS_LLQ_Events || (response->h.flags.b[0] & kDNSFlag0_TC))
ptr = LocateAnswers(response, end);
else
{
mDNSu8 rcode = (mDNSu8)(response->h.flags.b[1] & kDNSFlag1_RC_Mask);
mDNSBool failure = !(rcode == kDNSFlag1_RC_NoErr || rcode == kDNSFlag1_RC_NXDomain || rcode == kDNSFlag1_RC_NotAuth);
mDNSBool returnEarly = mDNSfalse;
for (i = 0; i < response->h.numQuestions && ptr && ptr < end; i++)
{
DNSQuestion q, *qptr = mDNSNULL;
ptr = getQuestion(response, ptr, end, InterfaceID, &q);
if (ptr && (qptr = ExpectingUnicastResponseForQuestion(m, dstport, response->h.id, &q, !dstaddr)))
{
if (!failure)
{
CacheRecord *rr;
const mDNSu32 slot = HashSlot(&q.qname);
CacheGroup *cg = CacheGroupForName(m, slot, q.qnamehash, &q.qname);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (SameNameRecordAnswersQuestion(&rr->resrec, qptr))
{
debugf("uDNS marking %p %##s (%s) %p %s", q.InterfaceID, q.qname.c, DNSTypeName(q.qtype),
rr->resrec.InterfaceID, CRDisplayString(m, rr));
rr->TimeRcvd = m->timenow - TicksTTL(rr) - 1;
rr->UnansweredQueries = MaxUnansweredQueries;
}
}
else
{
if (qptr)
{
LogInfo("mDNSCoreReceiveResponse: Server %p responded with code %d to query %##s (%s)", qptr->qDNSServer, rcode, q.qname.c, DNSTypeName(q.qtype));
PenalizeDNSServer(m, qptr);
}
returnEarly = mDNStrue;
}
}
}
if (returnEarly)
{
LogInfo("Ignoring %2d Answer%s %2d Authorit%s %2d Additional%s",
response->h.numAnswers, response->h.numAnswers == 1 ? ", " : "s,",
response->h.numAuthorities, response->h.numAuthorities == 1 ? "y, " : "ies,",
response->h.numAdditionals, response->h.numAdditionals == 1 ? "" : "s");
return;
}
}
for (i = 0; i < totalrecords && ptr && ptr < end; i++)
{
mDNSBool AcceptableResponse = ResponseMCast || !dstaddr || LLQType;
const mDNSu8 RecordType =
(i < firstauthority ) ? (mDNSu8)kDNSRecordTypePacketAns :
(i < firstadditional) ? (mDNSu8)kDNSRecordTypePacketAuth : (mDNSu8)kDNSRecordTypePacketAdd;
ptr = GetLargeResourceRecord(m, response, ptr, end, InterfaceID, RecordType, &m->rec);
if (!ptr) goto exit; if (m->rec.r.resrec.RecordType == kDNSRecordTypePacketNegative) { m->rec.r.resrec.RecordType = 0; continue; }
if (m->rec.r.resrec.rrtype == kDNSType_TSIG) { m->rec.r.resrec.RecordType = 0; continue; }
if (m->rec.r.resrec.rrtype == kDNSType_OPT)
{
const rdataOPT *opt;
const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
for (opt = &m->rec.r.resrec.rdata->u.opt[0]; opt < e; opt++)
if (opt->opt == kDNSOpt_Owner && opt->u.owner.vers == 0 && opt->u.owner.HMAC.l[0])
{
ClearProxyRecords(m, &opt->u.owner, m->DuplicateRecords);
ClearProxyRecords(m, &opt->u.owner, m->ResourceRecords);
}
m->rec.r.resrec.RecordType = 0;
continue;
}
if ((m->rec.r.resrec.rrtype == kDNSType_CNAME) && SameDomainName(m->rec.r.resrec.name, &m->rec.r.resrec.rdata->u.name))
{
LogInfo("mDNSCoreReceiveResponse: CNAME loop domain name %##s", m->rec.r.resrec.name->c);
m->rec.r.resrec.RecordType = 0;
continue;
}
if (!mDNSOpaque16IsZero(response->h.id))
m->rec.r.resrec.rroriginalttl = GetEffectiveTTL(LLQType, m->rec.r.resrec.rroriginalttl);
if (!ResponseMCast)
{
if (LLQType)
{
if (llqMatch != mDNSNULL) m->rec.r.resrec.rDNSServer = uDNSServer = llqMatch->qDNSServer;
}
else if (!AcceptableResponse || !dstaddr)
{
DNSQuestion *q = ExpectingUnicastResponseForRecord(m, srcaddr, ResponseSrcLocal, dstport, response->h.id, &m->rec.r, !dstaddr);
if (q != mDNSNULL)
{
AcceptableResponse = mDNStrue;
if (!InterfaceID)
{
debugf("mDNSCoreReceiveResponse: InterfaceID %p %##s (%s)", q->InterfaceID, q->qname.c, DNSTypeName(q->qtype));
m->rec.r.resrec.rDNSServer = uDNSServer = q->qDNSServer;
}
}
else
{
if (!InterfaceID)
{
debugf("mDNSCoreReceiveResponse: Can't find question for record name %##s", m->rec.r.resrec.name->c);
AcceptableResponse = mDNSfalse;
}
}
}
}
if (mDNSOpaque16IsZero(response->h.id) && m->rec.r.resrec.rrtype != kDNSType_NSEC)
{
if (m->CurrentRecord)
LogMsg("mDNSCoreReceiveResponse ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
m->CurrentRecord = m->ResourceRecords;
while (m->CurrentRecord)
{
AuthRecord *rr = m->CurrentRecord;
m->CurrentRecord = rr->next;
if (!AcceptableResponse && !(ResponseSrcLocal && rr->resrec.RecordType == kDNSRecordTypeUnique)) continue;
if (PacketRRMatchesSignature(&m->rec.r, rr)) {
if (IdenticalSameNameRecord(&m->rec.r.resrec, &rr->resrec))
{
if (m->rec.r.resrec.rroriginalttl >= rr->resrec.rroriginalttl/2 || m->SleepState)
{
if (rr->ImmedAnswer == InterfaceID) { rr->ImmedAnswer = mDNSNULL; rr->ImmedUnicast = mDNSfalse; }
}
else
{
if (rr->ImmedAnswer == mDNSNULL) { rr->ImmedAnswer = InterfaceID; m->NextScheduledResponse = m->timenow; }
else if (rr->ImmedAnswer != InterfaceID) { rr->ImmedAnswer = mDNSInterfaceMark; m->NextScheduledResponse = m->timenow; }
}
}
else if (m->rec.r.resrec.rroriginalttl > 0 && PacketRRConflict(m, rr, &m->rec.r))
{
LogInfo("mDNSCoreReceiveResponse: Pkt Record: %08lX %s", m->rec.r.resrec.rdatahash, CRDisplayString(m, &m->rec.r));
LogInfo("mDNSCoreReceiveResponse: Our Record: %08lX %s", rr-> resrec.rdatahash, ARDisplayString(m, rr));
if (rr->DependentOn)
{
while (rr->DependentOn) rr = rr->DependentOn;
LogInfo("mDNSCoreReceiveResponse: Dep Record: %08lX %s", rr-> resrec.rdatahash, ARDisplayString(m, rr));
}
if (rr->ProbeCount > DefaultProbeCountForTypeUnique)
LogInfo("mDNSCoreReceiveResponse: Already reset to Probing: %s", ARDisplayString(m, rr));
else if (rr->ProbeCount == DefaultProbeCountForTypeUnique)
LogMsg("mDNSCoreReceiveResponse: Ignoring response received before we even began probing: %s", ARDisplayString(m, rr));
else
{
LogMsg("mDNSCoreReceiveResponse: Received from %#a:%d %s", srcaddr, mDNSVal16(srcport), CRDisplayString(m, &m->rec.r));
if (rr->resrec.RecordType == kDNSRecordTypeVerified)
{
LogMsg("mDNSCoreReceiveResponse: Reseting to Probing: %s", ARDisplayString(m, rr));
rr->resrec.RecordType = kDNSRecordTypeUnique;
rr->ProbeCount = DefaultProbeCountForTypeUnique + 1;
rr->AnnounceCount = InitialAnnounceCount;
InitializeLastAPTime(m, rr);
RecordProbeFailure(m, rr); }
else if (rr->resrec.RecordType == kDNSRecordTypeUnique)
{
LogMsg("mDNSCoreReceiveResponse: ProbeCount %d; will rename %s", rr->ProbeCount, ARDisplayString(m, rr));
mDNS_Deregister_internal(m, rr, mDNS_Dereg_conflict);
}
else if (rr->resrec.RecordType == kDNSRecordTypeKnownUnique)
{
LogMsg("mDNSCoreReceiveResponse: Unexpected conflict discarding %s", ARDisplayString(m, rr));
mDNS_Deregister_internal(m, rr, mDNS_Dereg_conflict);
}
else
LogMsg("mDNSCoreReceiveResponse: Unexpected record type %X %s", rr->resrec.RecordType, ARDisplayString(m, rr));
}
}
else if (m->rec.r.resrec.rrtype == rr->resrec.rrtype)
if ((m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask) && m->timenow - rr->LastMCTime > mDNSPlatformOneSecond/2)
{ rr->ImmedAnswer = mDNSInterfaceMark; m->NextScheduledResponse = m->timenow; }
}
}
}
if (!AcceptableResponse)
{
const CacheRecord *cr;
for (cr = CacheFlushRecords; cr != (CacheRecord*)1; cr = cr->NextInCFList)
{
domainname *target = GetRRDomainNameTarget(&cr->resrec);
if (target && cr->resrec.rdatahash == m->rec.r.resrec.namehash && SameDomainName(target, m->rec.r.resrec.name))
{
debugf("mDNSCoreReceiveResponse: Found a matching entry for %##s in the CacheFlushRecords", m->rec.r.resrec.name->c);
AcceptableResponse = mDNStrue;
m->rec.r.resrec.rDNSServer = uDNSServer;
break;
}
}
}
if (!AcceptableResponse) LogInfo("mDNSCoreReceiveResponse ignoring %s", CRDisplayString(m, &m->rec.r));
if (m->rrcache_size && AcceptableResponse)
{
const mDNSu32 slot = HashSlot(m->rec.r.resrec.name);
CacheGroup *cg = CacheGroupForRecord(m, slot, &m->rec.r.resrec);
CacheRecord *rr;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
{
mDNSBool match = !InterfaceID ? m->rec.r.resrec.rDNSServer == rr->resrec.rDNSServer : rr->resrec.InterfaceID == InterfaceID;
if (match && IdenticalSameNameRecord(&m->rec.r.resrec, &rr->resrec))
{
if (m->rec.r.resrec.rdlength > InlineCacheRDSize)
verbosedebugf("Found record size %5d interface %p already in cache: %s",
m->rec.r.resrec.rdlength, InterfaceID, CRDisplayString(m, &m->rec.r));
if (m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask)
{
if (rr->NextInCFList == mDNSNULL && cfp != &rr->NextInCFList && LLQType != uDNS_LLQ_Events)
{ *cfp = rr; cfp = &rr->NextInCFList; *cfp = (CacheRecord*)1; }
if (!(rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask))
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next) if (ResourceRecordAnswersQuestion(&rr->resrec, q)) q->UniqueAnswers++;
rr->resrec.RecordType = m->rec.r.resrec.RecordType;
}
}
if (!SameRDataBody(&m->rec.r.resrec, &rr->resrec.rdata->u, SameDomainNameCS))
{
rr->resrec.rroriginalttl = 0;
rr->TimeRcvd = m->timenow;
rr->UnansweredQueries = MaxUnansweredQueries;
SetNextCacheCheckTimeForRecord(m, rr);
LogInfo("Discarding due to domainname case change old: %s", CRDisplayString(m,rr));
LogInfo("Discarding due to domainname case change new: %s", CRDisplayString(m,&m->rec.r));
LogInfo("Discarding due to domainname case change in %d slot %3d in %d %d",
NextCacheCheckEvent(rr) - m->timenow, slot, m->rrcache_nextcheck[slot] - m->timenow, m->NextCacheCheck - m->timenow);
}
else if (m->rec.r.resrec.rroriginalttl > 0)
{
DNSQuestion *q;
RefreshCacheRecord(m, rr, m->rec.r.resrec.rroriginalttl);
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask)
{
for (q = m->Questions; q; q=q->next)
{
if (!q->DuplicateOf && !q->LongLived &&
ActiveQuestion(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
{
q->LastQTime = m->timenow;
q->LastQTxTime = m->timenow;
q->RecentAnswerPkts = 0;
q->ThisQInterval = MaxQuestionInterval;
q->RequestUnicast = mDNSfalse;
q->unansweredQueries = 0;
debugf("mDNSCoreReceiveResponse: Set MaxQuestionInterval for %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
break; }
}
}
break;
}
else
{
debugf("DE for %s", CRDisplayString(m, rr));
if (RRExpireTime(rr) - m->timenow > mDNSPlatformOneSecond)
{
rr->resrec.rroriginalttl = 1;
rr->TimeRcvd = m->timenow;
rr->UnansweredQueries = MaxUnansweredQueries;
SetNextCacheCheckTimeForRecord(m, rr);
}
break;
}
}
}
if (!rr && m->rec.r.resrec.rroriginalttl > 0)
{
const mDNSBool AddToCFList = (m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask) && (LLQType != uDNS_LLQ_Events);
const mDNSs32 delay = AddToCFList ? NonZeroTime(m->timenow + mDNSPlatformOneSecond) :
CheckForSoonToExpireRecords(m, m->rec.r.resrec.name, m->rec.r.resrec.namehash, slot);
rr = CreateNewCacheEntry(m, slot, cg, delay);
if (rr)
{
if (AddToCFList) { *cfp = rr; cfp = &rr->NextInCFList; *cfp = (CacheRecord*)1; }
else if (rr->DelayDelivery) ScheduleNextCacheCheckTime(m, slot, rr->DelayDelivery);
}
}
}
m->rec.r.resrec.RecordType = 0; }
exit:
m->rec.r.resrec.RecordType = 0;
while (CacheFlushRecords != (CacheRecord*)1)
{
CacheRecord *r1 = CacheFlushRecords, *r2;
const mDNSu32 slot = HashSlot(r1->resrec.name);
const CacheGroup *cg = CacheGroupForRecord(m, slot, &r1->resrec);
CacheFlushRecords = CacheFlushRecords->NextInCFList;
r1->NextInCFList = mDNSNULL;
for (r2 = cg ? cg->members : mDNSNULL; r2; r2=r2->next)
if ((r1->resrec.InterfaceID == r2->resrec.InterfaceID) &&
(r1->resrec.InterfaceID || (r1->resrec.rDNSServer == r2->resrec.rDNSServer)) &&
r1->resrec.rrtype == r2->resrec.rrtype &&
r1->resrec.rrclass == r2->resrec.rrclass)
{
if (m->timenow - r2->TimeRcvd < mDNSPlatformOneSecond && RRExpireTime(r2) - m->timenow > mDNSPlatformOneSecond)
{
if (r2->resrec.rroriginalttl != r1->resrec.rroriginalttl && r1->resrec.rroriginalttl > 1)
{
if (!(r2->resrec.rroriginalttl == 240 && r1->resrec.rroriginalttl == 60 && r2->resrec.rrtype == kDNSType_TXT) &&
mDNSOpaque16IsZero(response->h.id))
LogInfo("Correcting TTL from %4d to %4d for %s",
r2->resrec.rroriginalttl, r1->resrec.rroriginalttl, CRDisplayString(m, r2));
r2->resrec.rroriginalttl = r1->resrec.rroriginalttl;
}
r2->TimeRcvd = m->timenow;
}
else {
verbosedebugf("Cache flush new %p age %d expire in %d %s", r1, m->timenow - r1->TimeRcvd, RRExpireTime(r1) - m->timenow, CRDisplayString(m, r1));
verbosedebugf("Cache flush old %p age %d expire in %d %s", r2, m->timenow - r2->TimeRcvd, RRExpireTime(r2) - m->timenow, CRDisplayString(m, r2));
if (r2->TimeRcvd == m->timenow && r2->resrec.rroriginalttl == 1 && r2->UnansweredQueries == MaxUnansweredQueries)
{
LogInfo("Cache flush for DE record %s", CRDisplayString(m, r2));
r2->resrec.rroriginalttl = 0;
}
else if (RRExpireTime(r2) - m->timenow > mDNSPlatformOneSecond)
{
r2->resrec.rroriginalttl = 1;
r2->UnansweredQueries = MaxUnansweredQueries;
r2->TimeRcvd = m->timenow - 1;
}
}
SetNextCacheCheckTimeForRecord(m, r2);
}
if (r1->DelayDelivery) {
r1->DelayDelivery = CheckForSoonToExpireRecords(m, r1->resrec.name, r1->resrec.namehash, slot);
if (!r1->DelayDelivery) CacheRecordDeferredAdd(m, r1);
else ScheduleNextCacheCheckTime(m, slot, r1->DelayDelivery);
}
}
ptr = response->data;
for (i = 0; i < response->h.numQuestions && ptr && ptr < end; i++)
{
DNSQuestion q;
DNSQuestion *qptr = mDNSNULL;
ptr = getQuestion(response, ptr, end, InterfaceID, &q);
if (ptr && (qptr = ExpectingUnicastResponseForQuestion(m, dstport, response->h.id, &q, !dstaddr)))
{
if (!InterfaceID && q.qtype != kDNSType_SOA && IsLocalDomain(&q.qname))
LogInfo("Skipping check to see if we need to generate a negative cache entry for %##s (%s)", q.qname.c, DNSTypeName(q.qtype));
else
{
CacheRecord *rr, *neg = mDNSNULL;
mDNSu32 slot = HashSlot(&q.qname);
CacheGroup *cg = CacheGroupForName(m, slot, q.qnamehash, &q.qname);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (SameNameRecordAnswersQuestion(&rr->resrec, qptr))
{
if (RRExpireTime(rr) - m->timenow > 0) break;
if (rr->resrec.RecordType == kDNSRecordTypePacketNegative) neg = rr;
}
if (!rr)
{
mDNSu32 negttl = 60;
int repeat = 0;
const domainname *name = &q.qname;
mDNSu32 hash = q.qnamehash;
if (q.qtype == kDNSType_SOA && SameDomainName(&q.qname, &localdomain)) negttl = 60 * 60 * 24;
if (response->h.numAuthorities && (ptr = LocateAuthorities(response, end)) != mDNSNULL)
{
ptr = GetLargeResourceRecord(m, response, ptr, end, InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_SOA)
{
const rdataSOA *const soa = (const rdataSOA *)m->rec.r.resrec.rdata->u.data;
mDNSu32 ttl_s = soa->min;
if (ttl_s > m->rec.r.resrec.rroriginalttl && m->rec.r.resrec.name->c[0])
ttl_s = m->rec.r.resrec.rroriginalttl;
if (negttl < ttl_s) negttl = ttl_s;
if (q.qtype == kDNSType_SOA)
{
int qcount = CountLabels(&q.qname);
int scount = CountLabels(m->rec.r.resrec.name);
if (qcount - 1 > scount)
if (SameDomainName(SkipLeadingLabels(&q.qname, qcount - scount), m->rec.r.resrec.name))
repeat = qcount - 1 - scount;
}
}
m->rec.r.resrec.RecordType = 0; }
if (neg)
{
if (negttl < neg->resrec.rroriginalttl * 2)
negttl = neg->resrec.rroriginalttl * 2;
if (negttl > 3600)
negttl = 3600;
}
negttl = GetEffectiveTTL(LLQType, negttl);
if (neg)
{
debugf("Renewing negative TTL from %d to %d %s", neg->resrec.rroriginalttl, negttl, CRDisplayString(m, neg));
RefreshCacheRecord(m, neg, negttl);
}
else while (1)
{
debugf("mDNSCoreReceiveResponse making negative cache entry TTL %d for %##s (%s)", negttl, name->c, DNSTypeName(q.qtype));
MakeNegativeCacheRecord(m, &m->rec.r, name, hash, q.qtype, q.qclass, negttl, mDNSInterface_Any, qptr->qDNSServer);
CreateNewCacheEntry(m, slot, cg, 0); m->rec.r.resrec.RecordType = 0; if (!repeat) break;
repeat--;
name = (const domainname *)(name->c + 1 + name->c[0]);
hash = DomainNameHashValue(name);
slot = HashSlot(name);
cg = CacheGroupForName(m, slot, hash, name);
}
}
}
}
}
}
mDNSlocal void ScheduleWakeupForList(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *e, AuthRecord *const thelist)
{
AuthRecord *rr;
for (rr = thelist; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering && mDNSSameEthAddress(&rr->WakeUp.HMAC, e))
mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
}
mDNSlocal void ScheduleWakeup(mDNS *const m, mDNSInterfaceID InterfaceID, mDNSEthAddr *e)
{
ScheduleWakeupForList(m, InterfaceID, e, m->DuplicateRecords);
ScheduleWakeupForList(m, InterfaceID, e, m->ResourceRecords);
}
mDNSlocal void SPSRecordCallback(mDNS *const m, AuthRecord *const ar, mStatus result)
{
if (result && result != mStatus_MemFree)
LogInfo("SPS Callback %d %s", result, ARDisplayString(m, ar));
if (result == mStatus_NameConflict)
{
LogMsg("Received Conflicting mDNS -- waking %s %.6a %s", InterfaceNameForID(m, ar->resrec.InterfaceID), &ar->WakeUp.HMAC, ARDisplayString(m, ar));
SendWakeup(m, ar->resrec.InterfaceID, &ar->WakeUp.IMAC, &ar->WakeUp.password);
ScheduleWakeup(m, ar->resrec.InterfaceID, &ar->WakeUp.HMAC);
}
else if (result == mStatus_MemFree)
{
m->ProxyRecords--;
mDNSPlatformMemFree(ar);
mDNS_UpdateAllowSleep(m);
}
}
mDNSlocal void mDNSCoreReceiveUpdate(mDNS *const m,
const DNSMessage *const msg, const mDNSu8 *end,
const mDNSAddr *srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, mDNSIPPort dstport,
const mDNSInterfaceID InterfaceID)
{
int i;
AuthRecord opt;
mDNSu8 *p = m->omsg.data;
OwnerOptData owner = zeroOwner; mDNSu32 updatelease = 0;
const mDNSu8 *ptr;
LogSPS("Received Update from %#-15a:%-5d to %#-15a:%-5d on 0x%p with "
"%2d Question%s %2d Answer%s %2d Authorit%s %2d Additional%s %d bytes",
srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), InterfaceID,
msg->h.numQuestions, msg->h.numQuestions == 1 ? ", " : "s,",
msg->h.numAnswers, msg->h.numAnswers == 1 ? ", " : "s,",
msg->h.numAuthorities, msg->h.numAuthorities == 1 ? "y, " : "ies,",
msg->h.numAdditionals, msg->h.numAdditionals == 1 ? " " : "s", end - msg->data);
if (!InterfaceID || !m->SPSSocket || !mDNSSameIPPort(dstport, m->SPSSocket->port)) return;
if (mDNS_PacketLoggingEnabled)
DumpPacket(m, mStatus_NoError, mDNSfalse, "UDP", srcaddr, srcport, dstaddr, dstport, msg, end);
ptr = LocateOptRR(msg, end, DNSOpt_LeaseData_Space + DNSOpt_OwnerData_ID_Space);
if (ptr)
{
ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
{
const rdataOPT *o;
const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
for (o = &m->rec.r.resrec.rdata->u.opt[0]; o < e; o++)
{
if (o->opt == kDNSOpt_Lease) updatelease = o->u.updatelease;
else if (o->opt == kDNSOpt_Owner && o->u.owner.vers == 0) owner = o->u.owner;
}
}
m->rec.r.resrec.RecordType = 0; }
InitializeDNSMessage(&m->omsg.h, msg->h.id, UpdateRespFlags);
if (!updatelease || !owner.HMAC.l[0])
{
static int msgs = 0;
if (msgs < 100)
{
msgs++;
LogMsg("Refusing sleep proxy registration from %#a:%d:%s%s", srcaddr, mDNSVal16(srcport),
!updatelease ? " No lease" : "", !owner.HMAC.l[0] ? " No owner" : "");
}
m->omsg.h.flags.b[1] |= kDNSFlag1_RC_FormErr;
}
else if (m->ProxyRecords + msg->h.mDNS_numUpdates > MAX_PROXY_RECORDS)
{
static int msgs = 0;
if (msgs < 100)
{
msgs++;
LogMsg("Refusing sleep proxy registration from %#a:%d: Too many records %d + %d = %d > %d", srcaddr, mDNSVal16(srcport),
m->ProxyRecords, msg->h.mDNS_numUpdates, m->ProxyRecords + msg->h.mDNS_numUpdates, MAX_PROXY_RECORDS);
}
m->omsg.h.flags.b[1] |= kDNSFlag1_RC_Refused;
}
else
{
LogSPS("Received Update for H-MAC %.6a I-MAC %.6a Password %.6a seq %d", &owner.HMAC, &owner.IMAC, &owner.password, owner.seq);
if (updatelease > 24 * 60 * 60)
updatelease = 24 * 60 * 60;
if (updatelease > 0x40000000UL / mDNSPlatformOneSecond)
updatelease = 0x40000000UL / mDNSPlatformOneSecond;
ptr = LocateAuthorities(msg, end);
for (i = 0; i < msg->h.mDNS_numUpdates && ptr && ptr < end; i++)
{
ptr = GetLargeResourceRecord(m, msg, ptr, end, InterfaceID, kDNSRecordTypePacketAuth, &m->rec);
if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative)
{
mDNSu16 RDLengthMem = GetRDLengthMem(&m->rec.r.resrec);
AuthRecord *ar = mDNSPlatformMemAllocate(sizeof(AuthRecord) - sizeof(RDataBody) + RDLengthMem);
if (!ar) { m->omsg.h.flags.b[1] |= kDNSFlag1_RC_Refused; break; }
else
{
mDNSu8 RecordType = m->rec.r.resrec.RecordType & kDNSRecordTypePacketUniqueMask ? kDNSRecordTypeUnique : kDNSRecordTypeShared;
m->rec.r.resrec.rrclass &= ~kDNSClass_UniqueRRSet;
ClearIdenticalProxyRecords(m, &owner, m->DuplicateRecords); ClearIdenticalProxyRecords(m, &owner, m->ResourceRecords);
mDNS_SetupResourceRecord(ar, mDNSNULL, InterfaceID, m->rec.r.resrec.rrtype, m->rec.r.resrec.rroriginalttl, RecordType, SPSRecordCallback, ar);
AssignDomainName(&ar->namestorage, m->rec.r.resrec.name);
ar->resrec.rdlength = GetRDLength(&m->rec.r.resrec, mDNSfalse);
ar->resrec.rdata->MaxRDLength = RDLengthMem;
mDNSPlatformMemCopy(ar->resrec.rdata->u.data, m->rec.r.resrec.rdata->u.data, RDLengthMem);
ar->ForceMCast = mDNStrue;
ar->WakeUp = owner;
if (m->rec.r.resrec.rrtype == kDNSType_PTR)
{
mDNSs32 t = ReverseMapDomainType(m->rec.r.resrec.name);
if (t == mDNSAddrType_IPv4) GetIPv4FromName(&ar->AddressProxy, m->rec.r.resrec.name);
else if (t == mDNSAddrType_IPv6) GetIPv6FromName(&ar->AddressProxy, m->rec.r.resrec.name);
debugf("mDNSCoreReceiveUpdate: PTR %d %d %#a %s", t, ar->AddressProxy.type, &ar->AddressProxy, ARDisplayString(m, ar));
if (ar->AddressProxy.type) SetSPSProxyListChanged(InterfaceID);
}
ar->TimeRcvd = m->timenow;
ar->TimeExpire = m->timenow + updatelease * mDNSPlatformOneSecond;
if (m->NextScheduledSPS - ar->TimeExpire > 0)
m->NextScheduledSPS = ar->TimeExpire;
mDNS_Register_internal(m, ar);
if (m->KnownBugs & mDNS_KnownBug_LimitedIPv6)
if (ar->resrec.rrtype == kDNSType_AAAA) ar->resrec.rroriginalttl = 0;
m->ProxyRecords++;
mDNS_UpdateAllowSleep(m);
LogSPS("SPS Registered %4d %X %s", m->ProxyRecords, RecordType, ARDisplayString(m,ar));
}
}
m->rec.r.resrec.RecordType = 0; }
if (m->omsg.h.flags.b[1] & kDNSFlag1_RC_Mask)
{
LogMsg("Refusing sleep proxy registration from %#a:%d: Out of memory", srcaddr, mDNSVal16(srcport));
ClearProxyRecords(m, &owner, m->DuplicateRecords);
ClearProxyRecords(m, &owner, m->ResourceRecords);
}
else
{
mDNS_SetupResourceRecord(&opt, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, mDNSNULL, mDNSNULL);
opt.resrec.rrclass = NormalMaxDNSMessageData;
opt.resrec.rdlength = sizeof(rdataOPT); opt.resrec.rdestimate = sizeof(rdataOPT);
opt.resrec.rdata->u.opt[0].opt = kDNSOpt_Lease;
opt.resrec.rdata->u.opt[0].u.updatelease = updatelease;
p = PutResourceRecordTTLWithLimit(&m->omsg, p, &m->omsg.h.numAdditionals, &opt.resrec, opt.resrec.rroriginalttl, m->omsg.data + AbsoluteMaxDNSMessageData);
}
}
if (p) mDNSSendDNSMessage(m, &m->omsg, p, InterfaceID, m->SPSSocket, srcaddr, srcport, mDNSNULL, mDNSNULL);
}
mDNSlocal void mDNSCoreReceiveUpdateR(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *end, const mDNSInterfaceID InterfaceID)
{
if (InterfaceID)
{
AuthRecord *rr;
mDNSu32 updatelease = 60 * 60; const mDNSu8 *ptr = LocateOptRR(msg, end, DNSOpt_LeaseData_Space);
if (ptr)
{
ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative && m->rec.r.resrec.rrtype == kDNSType_OPT)
{
const rdataOPT *o;
const rdataOPT *const e = (const rdataOPT *)&m->rec.r.resrec.rdata->u.data[m->rec.r.resrec.rdlength];
for (o = &m->rec.r.resrec.rdata->u.opt[0]; o < e; o++)
if (o->opt == kDNSOpt_Lease)
{
updatelease = o->u.updatelease;
LogSPS("Sleep Proxy granted lease time %4d seconds", updatelease);
}
}
m->rec.r.resrec.RecordType = 0; }
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID || (!rr->resrec.InterfaceID && (rr->ForceMCast || IsLocalDomain(rr->resrec.name))))
if (mDNSSameOpaque16(rr->updateid, msg->h.id))
{
rr->updateid = zeroID;
rr->expire = NonZeroTime(m->timenow + updatelease * mDNSPlatformOneSecond);
LogSPS("Sleep Proxy registered record %5d %s", updatelease, ARDisplayString(m,rr));
}
}
if (m->SleepLimit) m->NextScheduledSPRetry = m->timenow;
}
mDNSexport void MakeNegativeCacheRecord(mDNS *const m, CacheRecord *const cr,
const domainname *const name, const mDNSu32 namehash, const mDNSu16 rrtype, const mDNSu16 rrclass, mDNSu32 ttl_seconds, mDNSInterfaceID InterfaceID, DNSServer *dnsserver)
{
if (cr == &m->rec.r && m->rec.r.resrec.RecordType)
{
LogMsg("MakeNegativeCacheRecord: m->rec appears to be already in use for %s", CRDisplayString(m, &m->rec.r));
#if ForceAlerts
*(long*)0 = 0;
#endif
}
cr->resrec.RecordType = kDNSRecordTypePacketNegative;
cr->resrec.InterfaceID = InterfaceID;
cr->resrec.rDNSServer = dnsserver;
cr->resrec.name = name; cr->resrec.rrtype = rrtype;
cr->resrec.rrclass = rrclass;
cr->resrec.rroriginalttl = ttl_seconds;
cr->resrec.rdlength = 0;
cr->resrec.rdestimate = 0;
cr->resrec.namehash = namehash;
cr->resrec.rdatahash = 0;
cr->resrec.rdata = (RData*)&cr->smallrdatastorage;
cr->resrec.rdata->MaxRDLength = 0;
cr->NextInKAList = mDNSNULL;
cr->TimeRcvd = m->timenow;
cr->DelayDelivery = 0;
cr->NextRequiredQuery = m->timenow;
cr->LastUsed = m->timenow;
cr->CRActiveQuestion = mDNSNULL;
cr->UnansweredQueries = 0;
cr->LastUnansweredTime = 0;
#if ENABLE_MULTI_PACKET_QUERY_SNOOPING
cr->MPUnansweredQ = 0;
cr->MPLastUnansweredQT = 0;
cr->MPUnansweredKA = 0;
cr->MPExpectingKA = mDNSfalse;
#endif
cr->NextInCFList = mDNSNULL;
}
mDNSexport void mDNSCoreReceive(mDNS *const m, void *const pkt, const mDNSu8 *const end,
const mDNSAddr *const srcaddr, const mDNSIPPort srcport, const mDNSAddr *dstaddr, const mDNSIPPort dstport,
const mDNSInterfaceID InterfaceID)
{
mDNSInterfaceID ifid = InterfaceID;
DNSMessage *msg = (DNSMessage *)pkt;
const mDNSu8 StdQ = kDNSFlag0_QR_Query | kDNSFlag0_OP_StdQuery;
const mDNSu8 StdR = kDNSFlag0_QR_Response | kDNSFlag0_OP_StdQuery;
const mDNSu8 UpdQ = kDNSFlag0_QR_Query | kDNSFlag0_OP_Update;
const mDNSu8 UpdR = kDNSFlag0_QR_Response | kDNSFlag0_OP_Update;
mDNSu8 QR_OP;
mDNSu8 *ptr = mDNSNULL;
mDNSBool TLS = (dstaddr == (mDNSAddr *)1); if (TLS) dstaddr = mDNSNULL;
#ifndef UNICAST_DISABLED
if (mDNSSameAddress(srcaddr, &m->Router))
{
#ifdef _LEGACY_NAT_TRAVERSAL_
if (mDNSSameIPPort(srcport, SSDPPort) || (m->SSDPSocket && mDNSSameIPPort(dstport, m->SSDPSocket->port)))
{
mDNS_Lock(m);
LNT_ConfigureRouterInfo(m, InterfaceID, pkt, (mDNSu16)(end - (mDNSu8 *)pkt));
mDNS_Unlock(m);
return;
}
#endif
if (mDNSSameIPPort(srcport, NATPMPPort))
{
mDNS_Lock(m);
uDNS_ReceiveNATPMPPacket(m, InterfaceID, pkt, (mDNSu16)(end - (mDNSu8 *)pkt));
mDNS_Unlock(m);
return;
}
}
#ifdef _LEGACY_NAT_TRAVERSAL_
else if (m->SSDPSocket && mDNSSameIPPort(dstport, m->SSDPSocket->port)) { debugf("Ignoring SSDP response from %#a:%d", srcaddr, mDNSVal16(srcport)); return; }
#endif
#endif
if ((unsigned)(end - (mDNSu8 *)pkt) < sizeof(DNSMessageHeader))
{
LogMsg("DNS Message from %#a:%d to %#a:%d length %d too short", srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), end - (mDNSu8 *)pkt);
return;
}
QR_OP = (mDNSu8)(msg->h.flags.b[0] & kDNSFlag0_QROP_Mask);
ptr = (mDNSu8 *)&msg->h.numQuestions;
msg->h.numQuestions = (mDNSu16)((mDNSu16)ptr[0] << 8 | ptr[1]);
msg->h.numAnswers = (mDNSu16)((mDNSu16)ptr[2] << 8 | ptr[3]);
msg->h.numAuthorities = (mDNSu16)((mDNSu16)ptr[4] << 8 | ptr[5]);
msg->h.numAdditionals = (mDNSu16)((mDNSu16)ptr[6] << 8 | ptr[7]);
if (!m) { LogMsg("mDNSCoreReceive ERROR m is NULL"); return; }
if (srcaddr && !mDNSAddressIsValid(srcaddr)) { debugf("mDNSCoreReceive ignoring packet from %#a", srcaddr); return; }
mDNS_Lock(m);
m->PktNum++;
#ifndef UNICAST_DISABLED
if (!dstaddr || (!mDNSAddressIsAllDNSLinkGroup(dstaddr) && (QR_OP == StdR || QR_OP == UpdR)))
if (!mDNSOpaque16IsZero(msg->h.id)) {
ifid = mDNSInterface_Any;
if (mDNS_PacketLoggingEnabled)
DumpPacket(m, mStatus_NoError, mDNSfalse, TLS ? "TLS" : !dstaddr ? "TCP" : "UDP", srcaddr, srcport, dstaddr, dstport, msg, end);
uDNS_ReceiveMsg(m, msg, end, srcaddr, srcport);
}
#endif
if (QR_OP == StdQ) mDNSCoreReceiveQuery (m, msg, end, srcaddr, srcport, dstaddr, dstport, ifid);
else if (QR_OP == StdR) mDNSCoreReceiveResponse(m, msg, end, srcaddr, srcport, dstaddr, dstport, ifid);
else if (QR_OP == UpdQ) mDNSCoreReceiveUpdate (m, msg, end, srcaddr, srcport, dstaddr, dstport, InterfaceID);
else if (QR_OP == UpdR) mDNSCoreReceiveUpdateR (m, msg, end, InterfaceID);
else
{
LogMsg("Unknown DNS packet type %02X%02X from %#-15a:%-5d to %#-15a:%-5d length %d on %p (ignored)",
msg->h.flags.b[0], msg->h.flags.b[1], srcaddr, mDNSVal16(srcport), dstaddr, mDNSVal16(dstport), end - (mDNSu8 *)pkt, InterfaceID);
if (mDNS_LoggingEnabled)
{
int i = 0;
while (i<end - (mDNSu8 *)pkt)
{
char buffer[128];
char *p = buffer + mDNS_snprintf(buffer, sizeof(buffer), "%04X", i);
do if (i<end - (mDNSu8 *)pkt) p += mDNS_snprintf(p, sizeof(buffer), " %02X", ((mDNSu8 *)pkt)[i]); while (++i & 15);
LogInfo("%s", buffer);
}
}
}
mDNS_Unlock(m);
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Searcher Functions
#endif
#define SameQTarget(A,B) (((A)->Target.type == mDNSAddrType_None && (B)->Target.type == mDNSAddrType_None) || \
(mDNSSameAddress(&(A)->Target, &(B)->Target) && mDNSSameIPPort((A)->TargetPort, (B)->TargetPort)))
#define IsLLQ(Q) ((Q)->LongLived && !mDNSOpaque16IsZero((Q)->TargetQID))
mDNSlocal DNSQuestion *FindDuplicateQuestion(const mDNS *const m, const DNSQuestion *const question)
{
DNSQuestion *q;
for (q = m->Questions; q && q != question; q=q->next) if (q->InterfaceID == question->InterfaceID && SameQTarget(q, question) && q->qtype == question->qtype && q->qclass == question->qclass && IsLLQ(q) == IsLLQ(question) && (!q->AuthInfo || question->AuthInfo) && (q->SuppressQuery == question->SuppressQuery) && q->qnamehash == question->qnamehash &&
SameDomainName(&q->qname, &question->qname)) return(q);
return(mDNSNULL);
}
mDNSlocal void UpdateQuestionDuplicates(mDNS *const m, DNSQuestion *const question)
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next) if (q->DuplicateOf == question) if ((q->DuplicateOf = FindDuplicateQuestion(m, q)) == mDNSNULL)
{
q->LastQTime = question->LastQTime;
q->ThisQInterval = question->ThisQInterval;
q->ExpectUnicastResp = question->ExpectUnicastResp;
q->LastAnswerPktNum = question->LastAnswerPktNum;
q->RecentAnswerPkts = question->RecentAnswerPkts;
q->RequestUnicast = question->RequestUnicast;
q->LastQTxTime = question->LastQTxTime;
q->CNAMEReferrals = question->CNAMEReferrals;
q->nta = question->nta;
q->servAddr = question->servAddr;
q->servPort = question->servPort;
q->qDNSServer = question->qDNSServer;
q->validDNSServers = question->validDNSServers;
q->unansweredQueries = question->unansweredQueries;
q->noServerResponse = question->noServerResponse;
q->triedAllServersOnce = question->triedAllServersOnce;
q->TargetQID = question->TargetQID;
q->LocalSocket = question->LocalSocket;
q->state = question->state;
q->ReqLease = question->ReqLease;
q->expire = question->expire;
q->ntries = question->ntries;
q->id = question->id;
question->LocalSocket = mDNSNULL;
question->nta = mDNSNULL;
if (q->LocalSocket)
debugf("UpdateQuestionDuplicates transferred LocalSocket pointer for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
if (q->nta)
{
LogInfo("UpdateQuestionDuplicates transferred nta pointer for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
q->nta->ZoneDataContext = q;
}
if (question->tcp) LogInfo("UpdateQuestionDuplicates did not transfer tcp pointer");
if (question->state == LLQ_Established)
{
LogInfo("UpdateQuestionDuplicates transferred LLQ state for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
question->state = 0; }
SetNextQueryTime(m,q);
}
}
mDNSinline mDNSs32 PenaltyTimeForServer(mDNS *m, DNSServer *server)
{
mDNSs32 ptime = 0;
if (server->penaltyTime != 0)
{
ptime = server->penaltyTime - m->timenow;
if (ptime < 0)
{
LogMsg("PenaltyTimeForServer: PenaltyTime negative %d, (server penaltyTime %d, timenow %d) resetting the penalty",
ptime, server->penaltyTime, m->timenow);
server->penaltyTime = 0;
ptime = 0;
}
}
return ptime;
}
mDNSlocal int BetterMatchForName(const domainname *name, int namecount, const domainname *newname, int newcount,
int bestcount)
{
if (namecount < newcount || newcount < bestcount) return -1;
if (SameDomainName(SkipLeadingLabels(name, namecount - newcount), newname))
return bestcount == newcount ? 0 : 1;
else
return -1;
}
mDNSexport void SetValidDNSServers(mDNS *m, DNSQuestion *question)
{
DNSServer *curmatch = mDNSNULL;
int bestmatchlen = -1, namecount = CountLabels(&question->qname);
DNSServer *curr;
int bettermatch, currcount;
int index = 0;
question->validDNSServers = zeroOpaque64;
for (curr = m->DNSServers; curr; curr = curr->next)
{
debugf("SetValidDNSServers: Parsing DNS server Address %#a (Domain %##s), Scope: %d", &curr->addr, curr->domain.c, curr->scoped);
if (curr->flags & DNSServer_FlagDelete)
{ debugf("SetValidDNSServers: Delete set for index %d, DNS server %#a (Domain %##s), scoped %d", index, &curr->addr, curr->domain.c, curr->scoped); continue; }
currcount = CountLabels(&curr->domain);
if ((!curr->scoped && (!question->InterfaceID || (question->InterfaceID == mDNSInterface_Unicast))) || (curr->interface == question->InterfaceID))
{
bettermatch = BetterMatchForName(&question->qname, namecount, &curr->domain, currcount, bestmatchlen);
if ((bettermatch == 1) || (bettermatch == 0))
{
curmatch = curr;
bestmatchlen = currcount;
if (bettermatch) { debugf("SetValidDNSServers: Resetting all the bits"); question->validDNSServers = zeroOpaque64; }
debugf("SetValidDNSServers: Setting the bit for DNS server Address %#a (Domain %##s), Scoped:%d index %d", &curr->addr, curr->domain.c, curr->scoped, index);
bit_set_opaque64(question->validDNSServers, index);
}
}
index++;
}
question->noServerResponse = 0;
debugf("SetValidDNSServers: ValidDNSServer bits 0x%x%x for question %p %##s (%s)",
question->validDNSServers.l[1], question->validDNSServers.l[0], question, question->qname.c, DNSTypeName(question->qtype));
}
mDNSlocal DNSServer *GetBestServer(mDNS *m, const domainname *name, mDNSInterfaceID InterfaceID, mDNSOpaque64 validBits, int *selected, mDNSBool nameMatch)
{
DNSServer *curmatch = mDNSNULL;
int bestmatchlen = -1, namecount = name ? CountLabels(name) : 0;
DNSServer *curr;
mDNSs32 bestPenaltyTime, currPenaltyTime;
int bettermatch, currcount;
int index = 0;
int currindex = -1;
debugf("GetBestServer: ValidDNSServer bits 0x%x%x", validBits.l[1], validBits.l[0]);
bestPenaltyTime = DNSSERVER_PENALTY_TIME + 1;
for (curr = m->DNSServers; curr; curr = curr->next)
{
if (curr->flags & DNSServer_FlagDelete)
{ debugf("GetBestServer: Delete set for index %d, DNS server %#a (Domain %##s), scoped %d", index, &curr->addr, curr->domain.c, curr->scoped); continue; }
if (!bit_get_opaque64(validBits, index)) { debugf("GetBestServer: continuing for index %d", index); index++; continue; }
currcount = CountLabels(&curr->domain);
currPenaltyTime = PenaltyTimeForServer(m, curr);
debugf("GetBestServer: Address %#a (Domain %##s), PenaltyTime(abs) %d, PenaltyTime(rel) %d",
&curr->addr, curr->domain.c, curr->penaltyTime, currPenaltyTime);
if ((!curr->scoped && !InterfaceID) || (curr->interface == InterfaceID))
{
if (nameMatch) bettermatch = BetterMatchForName(name, namecount, &curr->domain, currcount, bestmatchlen);
else bettermatch = 0;
if ((bettermatch == 1) || ((bettermatch == 0) && currPenaltyTime < bestPenaltyTime))
{ currindex = index; curmatch = curr; bestmatchlen = currcount; bestPenaltyTime = currPenaltyTime; }
}
index++;
}
if (selected) *selected = currindex;
return curmatch;
}
mDNSexport DNSServer *GetServerForName(mDNS *m, const domainname *name, mDNSInterfaceID InterfaceID)
{
DNSServer *curmatch = mDNSNULL;
char *ifname = mDNSNULL; mDNSOpaque64 allValid;
if ((InterfaceID == mDNSInterface_Unicast) || (InterfaceID == mDNSInterface_LocalOnly))
InterfaceID = mDNSNULL;
if (InterfaceID) ifname = InterfaceNameForID(m, InterfaceID);
allValid.l[0] = allValid.l[1] = 0xFFFFFFFF;
curmatch = GetBestServer(m, name, InterfaceID, allValid, mDNSNULL, mDNStrue);
if (curmatch != mDNSNULL)
LogInfo("GetServerForName: DNS server %#a:%d (Penalty Time Left %d) (Scope %s:%p) found for name %##s", &curmatch->addr,
mDNSVal16(curmatch->port), (curmatch->penaltyTime ? (curmatch->penaltyTime - m->timenow) : 0), ifname ? ifname : "None",
InterfaceID, name);
else
LogInfo("GetServerForName: no DNS server (Scope %s:%p) found for name %##s", ifname ? ifname : "None", InterfaceID, name);
return(curmatch);
}
mDNSexport DNSServer *GetServerForQuestion(mDNS *m, DNSQuestion *question)
{
DNSServer *curmatch = mDNSNULL;
char *ifname = mDNSNULL; mDNSInterfaceID InterfaceID = question->InterfaceID;
const domainname *name = &question->qname;
int currindex;
if ((InterfaceID == mDNSInterface_Unicast) || (InterfaceID == mDNSInterface_LocalOnly))
InterfaceID = mDNSNULL;
if (InterfaceID) ifname = InterfaceNameForID(m, InterfaceID);
if (!mDNSOpaque64IsZero(&question->validDNSServers))
{
curmatch = GetBestServer(m, name, InterfaceID, question->validDNSServers, &currindex, mDNSfalse);
if (currindex != -1) bit_clr_opaque64(question->validDNSServers, currindex);
}
if (curmatch != mDNSNULL)
LogInfo("GetServerForQuestion: %p DNS server %#a:%d (Penalty Time Left %d) (Scope %s:%p) found for name %##s (%s)", question, &curmatch->addr,
mDNSVal16(curmatch->port), (curmatch->penaltyTime ? (curmatch->penaltyTime - m->timenow) : 0), ifname ? ifname : "None",
InterfaceID, name, DNSTypeName(question->qtype));
else
LogInfo("GetServerForQuestion: %p no DNS server (Scope %s:%p) found for name %##s (%s)", question, ifname ? ifname : "None", InterfaceID, name, DNSTypeName(question->qtype));
return(curmatch);
}
#define ValidQuestionTarget(Q) (((Q)->Target.type == mDNSAddrType_IPv4 || (Q)->Target.type == mDNSAddrType_IPv6) && \
(mDNSSameIPPort((Q)->TargetPort, UnicastDNSPort) || mDNSSameIPPort((Q)->TargetPort, MulticastDNSPort)))
mDNSlocal void LLQNATCallback(mDNS *m, NATTraversalInfo *n)
{
DNSQuestion *q;
(void)n; mDNS_Lock(m);
LogInfo("LLQNATCallback external address:port %.4a:%u, NAT result %d", &n->ExternalAddress, mDNSVal16(n->ExternalPort), n->Result);
for (q = m->Questions; q; q=q->next)
if (ActiveQuestion(q) && !mDNSOpaque16IsZero(q->TargetQID) && q->LongLived)
startLLQHandshake(m, q); #if APPLE_OSX_mDNSResponder
UpdateAutoTunnelDomainStatuses(m);
#endif
mDNS_Unlock(m);
}
mDNSlocal mDNSBool ShouldSuppressQuery(mDNS *const m, domainname *qname, mDNSu16 qtype, mDNSInterfaceID InterfaceID)
{
NetworkInterfaceInfo *i;
mDNSs32 iptype;
DomainAuthInfo *AuthInfo;
if (qtype == kDNSType_A) iptype = mDNSAddrType_IPv4;
else if (qtype == kDNSType_AAAA) iptype = mDNSAddrType_IPv6;
else { LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, not A/AAAA type", qname, DNSTypeName(qtype)); return mDNSfalse; }
if (IsLocalDomain(qname)) { LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, Local question", qname, DNSTypeName(qtype)); return mDNSfalse; }
AuthInfo = GetAuthInfoForName_internal(m, qname);
if (AuthInfo && !AuthInfo->deltime && AuthInfo->AutoTunnel)
{ LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, Private Domain", qname, DNSTypeName(qtype)); return mDNSfalse; }
for (i = m->HostInterfaces; i; i = i->next)
{
if (i->ip.type != iptype) continue;
if (!InterfaceID || (InterfaceID == mDNSInterface_LocalOnly) || (InterfaceID == mDNSInterface_P2P) ||
(i->InterfaceID == InterfaceID))
{
if (iptype == mDNSAddrType_IPv4 && !mDNSv4AddressIsLoopback(&i->ip.ip.v4) && !mDNSv4AddressIsLinkLocal(&i->ip.ip.v4))
{
LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, Local Address %.4a found", qname, DNSTypeName(qtype),
&i->ip.ip.v4);
return mDNSfalse;
}
else if (iptype == mDNSAddrType_IPv6 &&
!mDNSv6AddressIsLoopback(&i->ip.ip.v6) &&
!mDNSv6AddressIsLinkLocal(&i->ip.ip.v6) &&
!mDNSSameIPv6Address(i->ip.ip.v6, m->AutoTunnelHostAddr) &&
!mDNSSameIPv6Address(i->ip.ip.v6, m->AutoTunnelRelayAddr))
{
LogInfo("ShouldSuppressQuery: Query not suppressed for %##s, qtype %s, Local Address %.16a found", qname, DNSTypeName(qtype),
&i->ip.ip.v6);
return mDNSfalse;
}
}
}
LogInfo("ShouldSuppressQuery: Query suppressed for %##s, qtype %s, because no matching interface found", qname, DNSTypeName(qtype));
return mDNStrue;
}
mDNSlocal void CheckSuppressedCurrentQuestion(mDNS *const m, DNSQuestion *q)
{
CacheRecord *rr;
mDNSu32 slot;
CacheGroup *cg;
q->SuppressQuery = mDNSfalse;
slot = HashSlot(&q->qname);
cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
{
if (rr->resrec.RecordType == kDNSRecordTypePacketNegative)
{
LogInfo("CheckSuppressedCurrentQuestion: CacheRecord %s Suppressing RMV events for question %p %##s (%s), CRActiveQuestion %p, CurrentAnswers %d",
CRDisplayString(m, rr), q, q->qname.c, DNSTypeName(q->qtype), rr->CRActiveQuestion, q->CurrentAnswers);
continue;
}
if (SameNameRecordAnswersQuestion(&rr->resrec, q))
{
LogInfo("CheckSuppressedCurrentQuestion: Calling AnswerCurrentQuestionWithResourceRecord (RMV) for question %##s using resource record %s",
q->qname.c, CRDisplayString(m, rr));
q->CurrentAnswers--;
if (rr->resrec.rdlength > SmallRecordLimit) q->LargeAnswers--;
if (rr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) q->UniqueAnswers--;
if (rr->CRActiveQuestion == q)
{
DNSQuestion *qptr;
for (qptr = m->Questions; qptr; qptr=qptr->next)
if (ActiveQuestion(qptr) && ResourceRecordAnswersQuestion(&rr->resrec, qptr))
break;
if (qptr)
LogInfo("CheckSuppressedCurrentQuestion: Updating CRActiveQuestion to %p for cache record %s, "
"Original question CurrentAnswers %d, new question CurrentAnswers %d, SuppressUnusable %d, SuppressQuery %d",
qptr, CRDisplayString(m,rr), q->CurrentAnswers, qptr->CurrentAnswers, qptr->SuppressUnusable, qptr->SuppressQuery);
rr->CRActiveQuestion = qptr; if (!qptr) m->rrcache_active--; }
AnswerCurrentQuestionWithResourceRecord(m, rr, QC_rmv);
if (m->CurrentQuestion != q) break; }
}
if (m->CurrentQuestion == q) q->SuppressQuery = mDNStrue;
}
mDNSlocal mDNSBool IsQuestionNew(mDNS *const m, DNSQuestion *question)
{
DNSQuestion *q;
for (q = m->NewQuestions; q; q = q->next)
if (q == question) return mDNStrue;
return mDNSfalse;
}
mDNSexport void CheckSuppressUnusableQuestions(mDNS *const m)
{
DNSQuestion *q, *qnext;
DNSQuestion *restart = mDNSNULL;
for (q = m->Questions; q ; q = qnext)
{
qnext = q->next;
if (!mDNSOpaque16IsZero(q->TargetQID) && q->SuppressUnusable)
{
mDNSBool old = q->SuppressQuery;
q->SuppressQuery = ShouldSuppressQuery(m, &q->qname, q->qtype, q->InterfaceID);
if (q->SuppressQuery != old)
{
if (q->SuppressQuery)
{
if (m->CurrentQuestion)
LogMsg("CheckSuppressUnusableQuestions: ERROR m->CurrentQuestion already set: %##s (%s)",
m->CurrentQuestion->qname.c, DNSTypeName(m->CurrentQuestion->qtype));
if (!IsQuestionNew(m, q))
{
m->CurrentQuestion = q;
CheckSuppressedCurrentQuestion(m, q);
if (m->CurrentQuestion != q)
{
m->CurrentQuestion = mDNSNULL;
LogInfo("CheckSuppressUnusableQuestions: Question deleted while giving RMV events");
continue;
}
m->CurrentQuestion = mDNSNULL;
}
else { debugf("CheckSuppressUnusableQuestion: Question %p %##s (%s) is a new question", q, q->qname.c, DNSTypeName(q->qtype)); }
}
LogInfo("CheckSuppressUnusableQuestions: Stop question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
mDNS_StopQuery_internal(m, q);
q->next = restart;
restart = q;
}
}
}
while (restart)
{
q = restart;
restart = restart->next;
q->next = mDNSNULL;
LogInfo("CheckSuppressUnusableQuestions: Start question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
mDNS_StartQuery_internal(m, q);
}
}
mDNSexport mStatus mDNS_StartQuery_internal(mDNS *const m, DNSQuestion *const question)
{
if (question->Target.type && !ValidQuestionTarget(question))
{
LogMsg("Warning! Target.type = %ld port = %u (Client forgot to initialize before calling mDNS_StartQuery?)",
question->Target.type, mDNSVal16(question->TargetPort));
question->Target.type = mDNSAddrType_None;
}
if (!question->Target.type) question->TargetPort = zeroIPPort;
question->TargetQID =
#ifndef UNICAST_DISABLED
(question->Target.type || Question_uDNS(question)) ? mDNS_NewMessageID(m) :
#endif // UNICAST_DISABLED
zeroID;
debugf("mDNS_StartQuery: %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
if (m->rrcache_size == 0) return(mStatus_NoCache);
else
{
int i;
DNSQuestion **q;
if (!ValidateDomainName(&question->qname))
{
LogMsg("Attempt to start query with invalid qname %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
return(mStatus_Invalid);
}
q = &m->Questions;
if (question->InterfaceID == mDNSInterface_LocalOnly || question->InterfaceID == mDNSInterface_P2P) q = &m->LocalOnlyQuestions;
while (*q && *q != question) q=&(*q)->next;
if (*q)
{
LogMsg("Error! Tried to add a question %##s (%s) %p that's already in the active list",
question->qname.c, DNSTypeName(question->qtype), question);
return(mStatus_AlreadyRegistered);
}
*q = question;
if (question->InterfaceID && question->InterfaceID != mDNSInterface_LocalOnly && question->InterfaceID != mDNSInterface_Unicast && question->InterfaceID != mDNSInterface_P2P)
{
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, question->InterfaceID);
if (!intf)
LogMsg("Note: InterfaceID %p for question %##s (%s) not currently found in active interface list",
question->InterfaceID, question->qname.c, DNSTypeName(question->qtype));
}
question->next = mDNSNULL;
question->qnamehash = DomainNameHashValue(&question->qname); question->DelayAnswering = CheckForSoonToExpireRecords(m, &question->qname, question->qnamehash, HashSlot(&question->qname));
question->LastQTime = m->timenow;
question->ThisQInterval = InitialQuestionInterval; question->ExpectUnicastResp = 0;
question->LastAnswerPktNum = m->PktNum;
question->RecentAnswerPkts = 0;
question->CurrentAnswers = 0;
question->LargeAnswers = 0;
question->UniqueAnswers = 0;
question->FlappingInterface1 = mDNSNULL;
question->FlappingInterface2 = mDNSNULL;
question->AuthInfo = GetAuthInfoForQuestion(m, question);
if (question->SuppressUnusable)
question->SuppressQuery = ShouldSuppressQuery(m, &question->qname, question->qtype, question->InterfaceID);
else
question->SuppressQuery = 0;
question->DuplicateOf = FindDuplicateQuestion(m, question);
question->NextInDQList = mDNSNULL;
question->SendQNow = mDNSNULL;
question->SendOnAll = mDNSfalse;
question->RequestUnicast = 0;
question->LastQTxTime = m->timenow;
question->CNAMEReferrals = 0;
question->LocalSocket = mDNSNULL;
question->deliverAddEvents = mDNSfalse;
question->qDNSServer = mDNSNULL;
question->unansweredQueries = 0;
question->nta = mDNSNULL;
question->servAddr = zeroAddr;
question->servPort = zeroIPPort;
question->tcp = mDNSNULL;
question->NoAnswer = NoAnswer_Normal;
question->state = LLQ_InitialRequest;
question->ReqLease = 0;
question->expire = 0;
question->ntries = 0;
question->id = zeroOpaque64;
question->validDNSServers = zeroOpaque64;
question->triedAllServersOnce = 0;
question->noServerResponse = 0;
if (question->DuplicateOf) question->AuthInfo = question->DuplicateOf->AuthInfo;
for (i=0; i<DupSuppressInfoSize; i++)
question->DupSuppress[i].InterfaceID = mDNSNULL;
debugf("mDNS_StartQuery: Question %##s (%s) Interface %p Now %d Send in %d Answer in %d (%p) %s (%p)",
question->qname.c, DNSTypeName(question->qtype), question->InterfaceID, m->timenow,
NextQSendTime(question) - m->timenow,
question->DelayAnswering ? question->DelayAnswering - m->timenow : 0,
question, question->DuplicateOf ? "duplicate of" : "not duplicate", question->DuplicateOf);
if (question->DelayAnswering)
LogInfo("mDNS_StartQuery_internal: Delaying answering for %d ticks while cache stabilizes for %##s (%s)",
question->DelayAnswering - m->timenow, question->qname.c, DNSTypeName(question->qtype));
if (question->InterfaceID == mDNSInterface_LocalOnly || question->InterfaceID == mDNSInterface_P2P)
{
if (!m->NewLocalOnlyQuestions) m->NewLocalOnlyQuestions = question;
}
else
{
if (!m->NewQuestions) m->NewQuestions = question;
if (!mDNSOpaque16IsZero(question->TargetQID))
{
if (question->DuplicateOf)
{
question->validDNSServers = question->DuplicateOf->validDNSServers;
question->qDNSServer = question->DuplicateOf->qDNSServer;
LogInfo("mDNS_StartQuery_internal: Duplicate question %p (%p) %##s (%s), DNS Server %#a:%d",
question, question->DuplicateOf, question->qname.c, DNSTypeName(question->qtype),
question->qDNSServer ? &question->qDNSServer->addr : mDNSNULL,
mDNSVal16(question->qDNSServer ? question->qDNSServer->port : zeroIPPort));
}
else
{
SetValidDNSServers(m, question);
question->qDNSServer = GetServerForQuestion(m, question);
LogInfo("mDNS_StartQuery_internal: question %p %##s (%s), DNS Server %#a:%d",
question, question->qname.c, DNSTypeName(question->qtype),
question->qDNSServer ? &question->qDNSServer->addr : mDNSNULL,
mDNSVal16(question->qDNSServer ? question->qDNSServer->port : zeroIPPort));
}
ActivateUnicastQuery(m, question, mDNSfalse);
if (question->LongLived && !m->LLQNAT.clientContext)
{
m->LLQNAT.Protocol = NATOp_MapUDP;
m->LLQNAT.IntPort = m->UnicastPort4;
m->LLQNAT.RequestedPort = m->UnicastPort4;
m->LLQNAT.clientCallback = LLQNATCallback;
m->LLQNAT.clientContext = (void*)1; mDNS_StartNATOperation_internal(m, &m->LLQNAT);
}
#if APPLE_OSX_mDNSResponder
if (question->LongLived)
UpdateAutoTunnelDomainStatuses(m);
#endif
}
SetNextQueryTime(m,question);
}
return(mStatus_NoError);
}
}
mDNSexport void CancelGetZoneData(mDNS *const m, ZoneData *nta)
{
debugf("CancelGetZoneData %##s (%s)", nta->question.qname.c, DNSTypeName(nta->question.qtype));
if (nta->question.ThisQInterval != -1)
{
mDNS_StopQuery_internal(m, &nta->question);
if (nta->question.ThisQInterval != -1)
LogMsg("CancelGetZoneData: Question %##s (%s) ThisQInterval %d not -1", nta->question.qname.c, DNSTypeName(nta->question.qtype), nta->question.ThisQInterval);
}
mDNSPlatformMemFree(nta);
}
mDNSexport mStatus mDNS_StopQuery_internal(mDNS *const m, DNSQuestion *const question)
{
const mDNSu32 slot = HashSlot(&question->qname);
CacheGroup *cg = CacheGroupForName(m, slot, question->qnamehash, &question->qname);
CacheRecord *rr;
DNSQuestion **qp = &m->Questions;
if (question->InterfaceID == mDNSInterface_LocalOnly || question->InterfaceID == mDNSInterface_P2P) qp = &m->LocalOnlyQuestions;
while (*qp && *qp != question) qp=&(*qp)->next;
if (*qp) *qp = (*qp)->next;
else
{
#if !ForceAlerts
if (question->ThisQInterval >= 0) #endif
LogMsg("mDNS_StopQuery_internal: Question %##s (%s) not found in active list",
question->qname.c, DNSTypeName(question->qtype));
#if ForceAlerts
*(long*)0 = 0;
#endif
return(mStatus_BadReferenceErr);
}
UpdateQuestionDuplicates(m, question);
question->ThisQInterval = -1;
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
{
if (rr->CRActiveQuestion == question)
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next) if (ActiveQuestion(q) && ResourceRecordAnswersQuestion(&rr->resrec, q))
break;
if (q)
debugf("mDNS_StopQuery_internal: Updating CRActiveQuestion to %p for cache record %s, Original question CurrentAnswers %d, new question "
"CurrentAnswers %d, SuppressQuery %d", q, CRDisplayString(m,rr), question->CurrentAnswers, q->CurrentAnswers, q->SuppressQuery);
rr->CRActiveQuestion = q; if (!q) m->rrcache_active--; }
}
if (m->CurrentQuestion == question)
{
debugf("mDNS_StopQuery_internal: Just deleted the currently active question: %##s (%s)",
question->qname.c, DNSTypeName(question->qtype));
m->CurrentQuestion = question->next;
}
if (m->NewQuestions == question)
{
debugf("mDNS_StopQuery_internal: Just deleted a new question that wasn't even answered yet: %##s (%s)",
question->qname.c, DNSTypeName(question->qtype));
m->NewQuestions = question->next;
}
if (m->NewLocalOnlyQuestions == question) m->NewLocalOnlyQuestions = question->next;
question->next = mDNSNULL;
if (question->tcp) { DisposeTCPConn(question->tcp); question->tcp = mDNSNULL; }
if (question->LocalSocket) { mDNSPlatformUDPClose(question->LocalSocket); question->LocalSocket = mDNSNULL; }
if (!mDNSOpaque16IsZero(question->TargetQID) && question->LongLived)
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next)
if (!mDNSOpaque16IsZero(q->TargetQID) && q->LongLived) break;
if (!q)
{
if (!m->LLQNAT.clientContext) LogMsg("mDNS_StopQuery ERROR LLQNAT.clientContext NULL");
else
{
LogInfo("Stopping LLQNAT");
mDNS_StopNATOperation_internal(m, &m->LLQNAT);
m->LLQNAT.clientContext = mDNSNULL; }
}
if (question->state == LLQ_Established)
{
question->ReqLease = 0;
sendLLQRefresh(m, question);
if (question->tcp)
{
question->tcp->question = mDNSNULL;
question->tcp = mDNSNULL;
}
}
#if APPLE_OSX_mDNSResponder
UpdateAutoTunnelDomainStatuses(m);
#endif
}
if (question->nta) { CancelGetZoneData(m, question->nta); question->nta = mDNSNULL; }
return(mStatus_NoError);
}
mDNSexport mStatus mDNS_StartQuery(mDNS *const m, DNSQuestion *const question)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_StartQuery_internal(m, question);
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_StopQuery(mDNS *const m, DNSQuestion *const question)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_StopQuery_internal(m, question);
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_StopQueryWithRemoves(mDNS *const m, DNSQuestion *const question)
{
mStatus status;
DNSQuestion *qq;
mDNS_Lock(m);
for (qq = m->NewQuestions; qq; qq=qq->next) if (qq == question) break;
status = mDNS_StopQuery_internal(m, question);
if (status == mStatus_NoError && !qq)
{
const CacheRecord *rr;
const mDNSu32 slot = HashSlot(&question->qname);
CacheGroup *const cg = CacheGroupForName(m, slot, question->qnamehash, &question->qname);
LogInfo("Generating terminal removes for %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
for (rr = cg ? cg->members : mDNSNULL; rr; rr=rr->next)
if (rr->resrec.RecordType != kDNSRecordTypePacketNegative && SameNameRecordAnswersQuestion(&rr->resrec, question))
{
if (question->QuestionCallback)
question->QuestionCallback(m, question, &rr->resrec, mDNSfalse);
}
}
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_Reconfirm(mDNS *const m, CacheRecord *const cr)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
if (status == mStatus_NoError) ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, 0);
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_ReconfirmByValue(mDNS *const m, ResourceRecord *const rr)
{
mStatus status = mStatus_BadReferenceErr;
CacheRecord *cr;
mDNS_Lock(m);
cr = FindIdenticalRecordInCache(m, rr);
debugf("mDNS_ReconfirmByValue: %p %s", cr, RRDisplayString(m, rr));
if (cr) status = mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
if (status == mStatus_NoError) ReconfirmAntecedents(m, cr->resrec.name, cr->resrec.namehash, 0);
mDNS_Unlock(m);
return(status);
}
mDNSlocal mStatus mDNS_StartBrowse_internal(mDNS *const m, DNSQuestion *const question,
const domainname *const srv, const domainname *const domain,
const mDNSInterfaceID InterfaceID, mDNSBool ForceMCast, mDNSQuestionCallback *Callback, void *Context)
{
question->InterfaceID = InterfaceID;
question->Target = zeroAddr;
question->qtype = kDNSType_PTR;
question->qclass = kDNSClass_IN;
question->LongLived = mDNStrue;
question->ExpectUnique = mDNSfalse;
question->ForceMCast = ForceMCast;
question->ReturnIntermed = mDNSfalse;
question->SuppressUnusable = mDNSfalse;
question->QuestionCallback = Callback;
question->QuestionContext = Context;
if (!ConstructServiceName(&question->qname, mDNSNULL, srv, domain)) return(mStatus_BadParamErr);
return(mDNS_StartQuery_internal(m, question));
}
mDNSexport mStatus mDNS_StartBrowse(mDNS *const m, DNSQuestion *const question,
const domainname *const srv, const domainname *const domain,
const mDNSInterfaceID InterfaceID, mDNSBool ForceMCast, mDNSQuestionCallback *Callback, void *Context)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_StartBrowse_internal(m, question, srv, domain, InterfaceID, ForceMCast, Callback, Context);
mDNS_Unlock(m);
return(status);
}
mDNSlocal mDNSBool MachineHasActiveIPv6(mDNS *const m)
{
NetworkInterfaceInfo *intf;
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->ip.type == mDNSAddrType_IPv6) return(mDNStrue);
return(mDNSfalse);
}
mDNSlocal void FoundServiceInfoSRV(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
{
ServiceInfoQuery *query = (ServiceInfoQuery *)question->QuestionContext;
mDNSBool PortChanged = !mDNSSameIPPort(query->info->port, answer->rdata->u.srv.port);
if (!AddRecord) return;
if (answer->rrtype != kDNSType_SRV) return;
query->info->port = answer->rdata->u.srv.port;
if (!query->GotSRV)
{
query->GotSRV = mDNStrue;
query->qAv4.InterfaceID = answer->InterfaceID;
AssignDomainName(&query->qAv4.qname, &answer->rdata->u.srv.target);
query->qAv6.InterfaceID = answer->InterfaceID;
AssignDomainName(&query->qAv6.qname, &answer->rdata->u.srv.target);
mDNS_StartQuery(m, &query->qAv4);
if (MachineHasActiveIPv6(m)) mDNS_StartQuery(m, &query->qAv6);
}
else if ((query->qAv4.InterfaceID != query->qSRV.InterfaceID && query->qAv4.InterfaceID != answer->InterfaceID) ||
!SameDomainName(&query->qAv4.qname, &answer->rdata->u.srv.target))
{
mDNS_StopQuery(m, &query->qAv4);
if (query->qAv6.ThisQInterval >= 0) mDNS_StopQuery(m, &query->qAv6);
if (SameDomainName(&query->qAv4.qname, &answer->rdata->u.srv.target) && !PortChanged)
{
query->qAv4.InterfaceID = query->qSRV.InterfaceID; query->qAv6.InterfaceID = query->qSRV.InterfaceID;
}
else
{
query->qAv4.InterfaceID = answer->InterfaceID;
AssignDomainName(&query->qAv4.qname, &answer->rdata->u.srv.target);
query->qAv6.InterfaceID = answer->InterfaceID;
AssignDomainName(&query->qAv6.qname, &answer->rdata->u.srv.target);
}
debugf("FoundServiceInfoSRV: Restarting address queries for %##s (%s)", query->qAv4.qname.c, DNSTypeName(query->qAv4.qtype));
mDNS_StartQuery(m, &query->qAv4);
if (MachineHasActiveIPv6(m)) mDNS_StartQuery(m, &query->qAv6);
}
else if (query->ServiceInfoQueryCallback && query->GotADD && query->GotTXT && PortChanged)
{
if (++query->Answers >= 100)
debugf("**** WARNING **** Have given %lu answers for %##s (SRV) %##s %u",
query->Answers, query->qSRV.qname.c, answer->rdata->u.srv.target.c,
mDNSVal16(answer->rdata->u.srv.port));
query->ServiceInfoQueryCallback(m, query);
}
}
mDNSlocal void FoundServiceInfoTXT(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
{
ServiceInfoQuery *query = (ServiceInfoQuery *)question->QuestionContext;
if (!AddRecord) return;
if (answer->rrtype != kDNSType_TXT) return;
if (answer->rdlength > sizeof(query->info->TXTinfo)) return;
query->GotTXT = mDNStrue;
query->info->TXTlen = answer->rdlength;
query->info->TXTinfo[0] = 0; mDNSPlatformMemCopy(query->info->TXTinfo, answer->rdata->u.txt.c, answer->rdlength);
verbosedebugf("FoundServiceInfoTXT: %##s GotADD=%d", query->info->name.c, query->GotADD);
if (query->ServiceInfoQueryCallback && query->GotADD)
{
if (++query->Answers >= 100)
debugf("**** WARNING **** have given %lu answers for %##s (TXT) %#s...",
query->Answers, query->qSRV.qname.c, answer->rdata->u.txt.c);
query->ServiceInfoQueryCallback(m, query);
}
}
mDNSlocal void FoundServiceInfo(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
{
ServiceInfoQuery *query = (ServiceInfoQuery *)question->QuestionContext;
if (!AddRecord) return;
if (answer->rrtype == kDNSType_A)
{
query->info->ip.type = mDNSAddrType_IPv4;
query->info->ip.ip.v4 = answer->rdata->u.ipv4;
}
else if (answer->rrtype == kDNSType_AAAA)
{
query->info->ip.type = mDNSAddrType_IPv6;
query->info->ip.ip.v6 = answer->rdata->u.ipv6;
}
else
{
debugf("FoundServiceInfo: answer %##s type %d (%s) unexpected", answer->name->c, answer->rrtype, DNSTypeName(answer->rrtype));
return;
}
query->GotADD = mDNStrue;
query->info->InterfaceID = answer->InterfaceID;
verbosedebugf("FoundServiceInfo v%ld: %##s GotTXT=%d", query->info->ip.type, query->info->name.c, query->GotTXT);
if (query->ServiceInfoQueryCallback && query->GotTXT)
{
if (++query->Answers >= 100)
debugf(answer->rrtype == kDNSType_A ?
"**** WARNING **** have given %lu answers for %##s (A) %.4a" :
"**** WARNING **** have given %lu answers for %##s (AAAA) %.16a",
query->Answers, query->qSRV.qname.c, &answer->rdata->u.data);
query->ServiceInfoQueryCallback(m, query);
}
}
mDNSexport mStatus mDNS_StartResolveService(mDNS *const m,
ServiceInfoQuery *query, ServiceInfo *info, mDNSServiceInfoQueryCallback *Callback, void *Context)
{
mStatus status;
mDNS_Lock(m);
query->qSRV.ThisQInterval = -1; query->qSRV.InterfaceID = info->InterfaceID;
query->qSRV.Target = zeroAddr;
AssignDomainName(&query->qSRV.qname, &info->name);
query->qSRV.qtype = kDNSType_SRV;
query->qSRV.qclass = kDNSClass_IN;
query->qSRV.LongLived = mDNSfalse;
query->qSRV.ExpectUnique = mDNStrue;
query->qSRV.ForceMCast = mDNSfalse;
query->qSRV.ReturnIntermed = mDNSfalse;
query->qSRV.SuppressUnusable = mDNSfalse;
query->qSRV.QuestionCallback = FoundServiceInfoSRV;
query->qSRV.QuestionContext = query;
query->qTXT.ThisQInterval = -1; query->qTXT.InterfaceID = info->InterfaceID;
query->qTXT.Target = zeroAddr;
AssignDomainName(&query->qTXT.qname, &info->name);
query->qTXT.qtype = kDNSType_TXT;
query->qTXT.qclass = kDNSClass_IN;
query->qTXT.LongLived = mDNSfalse;
query->qTXT.ExpectUnique = mDNStrue;
query->qTXT.ForceMCast = mDNSfalse;
query->qTXT.ReturnIntermed = mDNSfalse;
query->qTXT.SuppressUnusable = mDNSfalse;
query->qTXT.QuestionCallback = FoundServiceInfoTXT;
query->qTXT.QuestionContext = query;
query->qAv4.ThisQInterval = -1; query->qAv4.InterfaceID = info->InterfaceID;
query->qAv4.Target = zeroAddr;
query->qAv4.qname.c[0] = 0;
query->qAv4.qtype = kDNSType_A;
query->qAv4.qclass = kDNSClass_IN;
query->qAv4.LongLived = mDNSfalse;
query->qAv4.ExpectUnique = mDNStrue;
query->qAv4.ForceMCast = mDNSfalse;
query->qAv4.ReturnIntermed = mDNSfalse;
query->qAv4.SuppressUnusable = mDNSfalse;
query->qAv4.QuestionCallback = FoundServiceInfo;
query->qAv4.QuestionContext = query;
query->qAv6.ThisQInterval = -1; query->qAv6.InterfaceID = info->InterfaceID;
query->qAv6.Target = zeroAddr;
query->qAv6.qname.c[0] = 0;
query->qAv6.qtype = kDNSType_AAAA;
query->qAv6.qclass = kDNSClass_IN;
query->qAv6.LongLived = mDNSfalse;
query->qAv6.ExpectUnique = mDNStrue;
query->qAv6.ForceMCast = mDNSfalse;
query->qAv6.ReturnIntermed = mDNSfalse;
query->qAv6.SuppressUnusable = mDNSfalse;
query->qAv6.QuestionCallback = FoundServiceInfo;
query->qAv6.QuestionContext = query;
query->GotSRV = mDNSfalse;
query->GotTXT = mDNSfalse;
query->GotADD = mDNSfalse;
query->Answers = 0;
query->info = info;
query->ServiceInfoQueryCallback = Callback;
query->ServiceInfoQueryContext = Context;
info->ip = zeroAddr;
info->port = zeroIPPort;
info->TXTlen = 0;
status = mDNS_StartQuery_internal(m, &query->qSRV);
if (status == mStatus_NoError) status = mDNS_StartQuery_internal(m, &query->qTXT);
if (status != mStatus_NoError) mDNS_StopResolveService(m, query);
mDNS_Unlock(m);
return(status);
}
mDNSexport void mDNS_StopResolveService (mDNS *const m, ServiceInfoQuery *q)
{
mDNS_Lock(m);
if (q->qSRV.ThisQInterval >= 0) mDNS_StopQuery_internal(m, &q->qSRV);
if (q->qTXT.ThisQInterval >= 0) mDNS_StopQuery_internal(m, &q->qTXT);
if (q->qAv4.ThisQInterval >= 0) mDNS_StopQuery_internal(m, &q->qAv4);
if (q->qAv6.ThisQInterval >= 0) mDNS_StopQuery_internal(m, &q->qAv6);
mDNS_Unlock(m);
}
mDNSexport mStatus mDNS_GetDomains(mDNS *const m, DNSQuestion *const question, mDNS_DomainType DomainType, const domainname *dom,
const mDNSInterfaceID InterfaceID, mDNSQuestionCallback *Callback, void *Context)
{
question->InterfaceID = InterfaceID;
question->Target = zeroAddr;
question->qtype = kDNSType_PTR;
question->qclass = kDNSClass_IN;
question->LongLived = mDNSfalse;
question->ExpectUnique = mDNSfalse;
question->ForceMCast = mDNSfalse;
question->ReturnIntermed = mDNSfalse;
question->SuppressUnusable = mDNSfalse;
question->QuestionCallback = Callback;
question->QuestionContext = Context;
if (DomainType > mDNS_DomainTypeMax) return(mStatus_BadParamErr);
if (!MakeDomainNameFromDNSNameString(&question->qname, mDNS_DomainTypeNames[DomainType])) return(mStatus_BadParamErr);
if (!dom) dom = &localdomain;
if (!AppendDomainName(&question->qname, dom)) return(mStatus_BadParamErr);
return(mDNS_StartQuery(m, question));
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Responder Functions
#endif
mDNSexport mStatus mDNS_Register(mDNS *const m, AuthRecord *const rr)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_Register_internal(m, rr);
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_Update(mDNS *const m, AuthRecord *const rr, mDNSu32 newttl,
const mDNSu16 newrdlength, RData *const newrdata, mDNSRecordUpdateCallback *Callback)
{
if (!ValidateRData(rr->resrec.rrtype, newrdlength, newrdata))
{
LogMsg("Attempt to update record with invalid rdata: %s", GetRRDisplayString_rdb(&rr->resrec, &newrdata->u, m->MsgBuffer));
return(mStatus_Invalid);
}
mDNS_Lock(m);
if (newttl == 0) newttl = rr->resrec.rroriginalttl;
if (rr->NewRData)
{
RData *n = rr->NewRData;
rr->NewRData = mDNSNULL; if (rr->UpdateCallback)
rr->UpdateCallback(m, rr, n, rr->newrdlength); }
rr->NewRData = newrdata;
rr->newrdlength = newrdlength;
rr->UpdateCallback = Callback;
#ifndef UNICAST_DISABLED
if (rr->resrec.InterfaceID != mDNSInterface_LocalOnly && rr->resrec.InterfaceID != mDNSInterface_P2P && !IsLocalDomain(rr->resrec.name))
{
mStatus status = uDNS_UpdateRecord(m, rr);
if (status != mStatus_NoError) { rr->NewRData = mDNSNULL; rr->newrdlength = 0; }
mDNS_Unlock(m);
return(status);
}
#endif
if (rr->resrec.rroriginalttl == newttl &&
rr->resrec.rdlength == newrdlength && mDNSPlatformMemSame(rr->resrec.rdata->u.data, newrdata->u.data, newrdlength))
CompleteRDataUpdate(m, rr);
else
{
rr->AnnounceCount = InitialAnnounceCount;
InitializeLastAPTime(m, rr);
while (rr->NextUpdateCredit && m->timenow - rr->NextUpdateCredit >= 0) GrantUpdateCredit(rr);
if (!rr->UpdateBlocked && rr->UpdateCredits) rr->UpdateCredits--;
if (!rr->NextUpdateCredit) rr->NextUpdateCredit = NonZeroTime(m->timenow + kUpdateCreditRefreshInterval);
if (rr->AnnounceCount > rr->UpdateCredits + 1) rr->AnnounceCount = (mDNSu8)(rr->UpdateCredits + 1);
if (rr->UpdateCredits <= 5)
{
mDNSu32 delay = 6 - rr->UpdateCredits; if (!rr->UpdateBlocked) rr->UpdateBlocked = NonZeroTime(m->timenow + (mDNSs32)delay * mDNSPlatformOneSecond);
rr->ThisAPInterval *= 4;
rr->LastAPTime = rr->UpdateBlocked - rr->ThisAPInterval;
LogMsg("Excessive update rate for %##s; delaying announcement by %ld second%s",
rr->resrec.name->c, delay, delay > 1 ? "s" : "");
}
rr->resrec.rroriginalttl = newttl;
}
mDNS_Unlock(m);
return(mStatus_NoError);
}
mDNSexport mStatus mDNS_Deregister(mDNS *const m, AuthRecord *const rr)
{
mStatus status;
mDNS_Lock(m);
status = mDNS_Deregister_internal(m, rr, mDNS_Dereg_normal);
mDNS_Unlock(m);
return(status);
}
mDNSlocal void mDNS_HostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result);
mDNSlocal NetworkInterfaceInfo *FindFirstAdvertisedInterface(mDNS *const m)
{
NetworkInterfaceInfo *intf;
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->Advertise) break;
return(intf);
}
mDNSlocal void AdvertiseInterface(mDNS *const m, NetworkInterfaceInfo *set)
{
char buffer[MAX_REVERSE_MAPPING_NAME];
NetworkInterfaceInfo *primary = FindFirstAdvertisedInterface(m);
if (!primary) primary = set;
mDNS_SetupResourceRecord(&set->RR_A, mDNSNULL, set->InterfaceID, kDNSType_A, kHostNameTTL, kDNSRecordTypeUnique, mDNS_HostNameCallback, set);
mDNS_SetupResourceRecord(&set->RR_PTR, mDNSNULL, set->InterfaceID, kDNSType_PTR, kHostNameTTL, kDNSRecordTypeKnownUnique, mDNSNULL, mDNSNULL);
mDNS_SetupResourceRecord(&set->RR_HINFO, mDNSNULL, set->InterfaceID, kDNSType_HINFO, kHostNameTTL, kDNSRecordTypeUnique, mDNSNULL, mDNSNULL);
#if ANSWER_REMOTE_HOSTNAME_QUERIES
set->RR_A .AllowRemoteQuery = mDNStrue;
set->RR_PTR .AllowRemoteQuery = mDNStrue;
set->RR_HINFO.AllowRemoteQuery = mDNStrue;
#endif
AssignDomainName(&set->RR_A.namestorage, &m->MulticastHostname);
if (set->ip.type == mDNSAddrType_IPv4)
{
set->RR_A.resrec.rrtype = kDNSType_A;
set->RR_A.resrec.rdata->u.ipv4 = set->ip.ip.v4;
mDNS_snprintf(buffer, sizeof(buffer), "%d.%d.%d.%d.in-addr.arpa.",
set->ip.ip.v4.b[3], set->ip.ip.v4.b[2], set->ip.ip.v4.b[1], set->ip.ip.v4.b[0]);
}
else if (set->ip.type == mDNSAddrType_IPv6)
{
int i;
set->RR_A.resrec.rrtype = kDNSType_AAAA;
set->RR_A.resrec.rdata->u.ipv6 = set->ip.ip.v6;
for (i = 0; i < 16; i++)
{
static const char hexValues[] = "0123456789ABCDEF";
buffer[i * 4 ] = hexValues[set->ip.ip.v6.b[15 - i] & 0x0F];
buffer[i * 4 + 1] = '.';
buffer[i * 4 + 2] = hexValues[set->ip.ip.v6.b[15 - i] >> 4];
buffer[i * 4 + 3] = '.';
}
mDNS_snprintf(&buffer[64], sizeof(buffer)-64, "ip6.arpa.");
}
MakeDomainNameFromDNSNameString(&set->RR_PTR.namestorage, buffer);
set->RR_PTR.AutoTarget = Target_AutoHost; set->RR_PTR.ForceMCast = mDNStrue;
set->RR_A.RRSet = &primary->RR_A;
mDNS_Register_internal(m, &set->RR_A);
mDNS_Register_internal(m, &set->RR_PTR);
if (!NO_HINFO && m->HIHardware.c[0] > 0 && m->HISoftware.c[0] > 0 && m->HIHardware.c[0] + m->HISoftware.c[0] <= 254)
{
mDNSu8 *p = set->RR_HINFO.resrec.rdata->u.data;
AssignDomainName(&set->RR_HINFO.namestorage, &m->MulticastHostname);
set->RR_HINFO.DependentOn = &set->RR_A;
mDNSPlatformMemCopy(p, &m->HIHardware, 1 + (mDNSu32)m->HIHardware.c[0]);
p += 1 + (int)p[0];
mDNSPlatformMemCopy(p, &m->HISoftware, 1 + (mDNSu32)m->HISoftware.c[0]);
mDNS_Register_internal(m, &set->RR_HINFO);
}
else
{
debugf("Not creating HINFO record: platform support layer provided no information");
set->RR_HINFO.resrec.RecordType = kDNSRecordTypeUnregistered;
}
}
mDNSlocal void DeadvertiseInterface(mDNS *const m, NetworkInterfaceInfo *set)
{
NetworkInterfaceInfo *intf;
NetworkInterfaceInfo *primary = FindFirstAdvertisedInterface(m);
AuthRecord *A = primary ? &primary->RR_A : mDNSNULL;
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->RR_A.RRSet == &set->RR_A)
intf->RR_A.RRSet = A;
if (set->RR_A. resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_A, mDNS_Dereg_normal);
if (set->RR_PTR. resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_PTR, mDNS_Dereg_normal);
if (set->RR_HINFO.resrec.RecordType) mDNS_Deregister_internal(m, &set->RR_HINFO, mDNS_Dereg_normal);
}
mDNSexport void mDNS_SetFQDN(mDNS *const m)
{
domainname newmname;
NetworkInterfaceInfo *intf;
AuthRecord *rr;
newmname.c[0] = 0;
if (!AppendDomainLabel(&newmname, &m->hostlabel)) { LogMsg("ERROR: mDNS_SetFQDN: Cannot create MulticastHostname"); return; }
if (!AppendLiteralLabelString(&newmname, "local")) { LogMsg("ERROR: mDNS_SetFQDN: Cannot create MulticastHostname"); return; }
mDNS_Lock(m);
if (SameDomainNameCS(&m->MulticastHostname, &newmname)) debugf("mDNS_SetFQDN - hostname unchanged");
else
{
AssignDomainName(&m->MulticastHostname, &newmname);
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->Advertise) DeadvertiseInterface(m, intf);
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->Advertise) AdvertiseInterface(m, intf);
}
for (rr = m->ResourceRecords; rr; rr=rr->next) if (rr->AutoTarget) SetTargetToHostName(m, rr);
for (rr = m->DuplicateRecords; rr; rr=rr->next) if (rr->AutoTarget) SetTargetToHostName(m, rr);
mDNS_Unlock(m);
}
mDNSlocal void mDNS_HostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
{
(void)rr;
#if MDNS_DEBUGMSGS
{
char *msg = "Unknown result";
if (result == mStatus_NoError) msg = "Name registered";
else if (result == mStatus_NameConflict) msg = "Name conflict";
debugf("mDNS_HostNameCallback: %##s (%s) %s (%ld)", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), msg, result);
}
#endif
if (result == mStatus_NoError)
{
if (m->MainCallback)
m->MainCallback(m, mStatus_NoError);
}
else if (result == mStatus_NameConflict)
{
domainlabel oldlabel = m->hostlabel;
if (m->MainCallback)
m->MainCallback(m, mStatus_NameConflict);
if (SameDomainLabel(m->hostlabel.c, oldlabel.c))
IncrementLabelSuffix(&m->hostlabel, mDNSfalse);
mDNS_SetFQDN(m);
LogMsg("Local Hostname %#s.local already in use; will try %#s.local instead", oldlabel.c, m->hostlabel.c);
}
else if (result == mStatus_MemFree)
{
debugf("mDNS_HostNameCallback: MemFree (ignored)");
}
else
LogMsg("mDNS_HostNameCallback: Unknown error %d for registration of record %s", result, rr->resrec.name->c);
}
mDNSlocal void UpdateInterfaceProtocols(mDNS *const m, NetworkInterfaceInfo *active)
{
NetworkInterfaceInfo *intf;
active->IPv4Available = mDNSfalse;
active->IPv6Available = mDNSfalse;
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->InterfaceID == active->InterfaceID)
{
if (intf->ip.type == mDNSAddrType_IPv4 && intf->McastTxRx) active->IPv4Available = mDNStrue;
if (intf->ip.type == mDNSAddrType_IPv6 && intf->McastTxRx) active->IPv6Available = mDNStrue;
}
}
mDNSlocal void RestartRecordGetZoneData(mDNS * const m)
{
AuthRecord *rr;
LogInfo("RestartRecordGetZoneData: ResourceRecords");
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (AuthRecord_uDNS(rr) && rr->state != regState_NoTarget)
{
debugf("RestartRecordGetZoneData: StartGetZoneData for %##s", rr->resrec.name->c);
if (rr->nta) { rr->updateid = zeroID; CancelGetZoneData(m, rr->nta); }
rr->nta = StartGetZoneData(m, rr->resrec.name, ZoneServiceUpdate, RecordRegistrationGotZoneData, rr);
}
}
mDNSlocal void InitializeNetWakeState(mDNS *const m, NetworkInterfaceInfo *set)
{
int i;
set->NetWakeBrowse.ThisQInterval = -1;
for (i=0; i<3; i++)
{
set->NetWakeResolve[i].ThisQInterval = -1;
set->SPSAddr[i].type = mDNSAddrType_None;
}
set->NextSPSAttempt = -1;
set->NextSPSAttemptTime = m->timenow;
}
mDNSexport void mDNS_ActivateNetWake_internal(mDNS *const m, NetworkInterfaceInfo *set)
{
NetworkInterfaceInfo *p = m->HostInterfaces;
while (p && p != set) p=p->next;
if (!p) { LogMsg("mDNS_ActivateNetWake_internal: NetworkInterfaceInfo %p not found in active list", set); return; }
if (set->InterfaceActive)
{
LogSPS("ActivateNetWake for %s (%#a)", set->ifname, &set->ip);
mDNS_StartBrowse_internal(m, &set->NetWakeBrowse, &SleepProxyServiceType, &localdomain, set->InterfaceID, mDNSfalse, m->SPSBrowseCallback, set);
}
}
mDNSexport void mDNS_DeactivateNetWake_internal(mDNS *const m, NetworkInterfaceInfo *set)
{
NetworkInterfaceInfo *p = m->HostInterfaces;
while (p && p != set) p=p->next;
if (!p) { LogMsg("mDNS_DeactivateNetWake_internal: NetworkInterfaceInfo %p not found in active list", set); return; }
if (set->NetWakeBrowse.ThisQInterval >= 0)
{
int i;
LogSPS("DeactivateNetWake for %s (%#a)", set->ifname, &set->ip);
mDNS_StopQuery_internal(m, &set->NetWakeBrowse);
for (i=0; i<3; i++) if (set->NetWakeResolve[i].ThisQInterval >= 0) mDNS_StopQuery_internal(m, &set->NetWakeResolve[i]);
if (m->SPSBrowseCallback)
{
mDNS_DropLockBeforeCallback(); m->SPSBrowseCallback(m, &set->NetWakeBrowse, mDNSNULL, mDNSfalse);
mDNS_ReclaimLockAfterCallback(); }
InitializeNetWakeState(m, set);
}
}
mDNSexport mStatus mDNS_RegisterInterface(mDNS *const m, NetworkInterfaceInfo *set, mDNSBool flapping)
{
AuthRecord *rr;
mDNSBool FirstOfType = mDNStrue;
NetworkInterfaceInfo **p = &m->HostInterfaces;
if (!set->InterfaceID)
{ LogMsg("Error! Tried to register a NetworkInterfaceInfo %#a with zero InterfaceID", &set->ip); return(mStatus_Invalid); }
if (!mDNSAddressIsValidNonZero(&set->mask))
{ LogMsg("Error! Tried to register a NetworkInterfaceInfo %#a with invalid mask %#a", &set->ip, &set->mask); return(mStatus_Invalid); }
mDNS_Lock(m);
set->InterfaceActive = mDNStrue;
set->IPv4Available = (mDNSu8)(set->ip.type == mDNSAddrType_IPv4 && set->McastTxRx);
set->IPv6Available = (mDNSu8)(set->ip.type == mDNSAddrType_IPv6 && set->McastTxRx);
InitializeNetWakeState(m, set);
while (*p)
{
if (*p == set)
{
LogMsg("Error! Tried to register a NetworkInterfaceInfo that's already in the list");
mDNS_Unlock(m);
return(mStatus_AlreadyRegistered);
}
if ((*p)->InterfaceID == set->InterfaceID)
{
set->InterfaceActive = mDNSfalse;
if (set->ip.type == (*p)->ip.type) FirstOfType = mDNSfalse;
if (set->ip.type == mDNSAddrType_IPv4 && set->McastTxRx) (*p)->IPv4Available = mDNStrue;
if (set->ip.type == mDNSAddrType_IPv6 && set->McastTxRx) (*p)->IPv6Available = mDNStrue;
}
p=&(*p)->next;
}
set->next = mDNSNULL;
*p = set;
if (set->Advertise)
AdvertiseInterface(m, set);
LogInfo("mDNS_RegisterInterface: InterfaceID %p %s (%#a) %s", set->InterfaceID, set->ifname, &set->ip,
set->InterfaceActive ?
"not represented in list; marking active and retriggering queries" :
"already represented in list; marking inactive for now");
if (set->NetWake) mDNS_ActivateNetWake_internal(m, set);
if (set->McastTxRx && ((m->KnownBugs & mDNS_KnownBug_PhantomInterfaces) || FirstOfType || set->InterfaceActive))
{
DNSQuestion *q;
const mDNSs32 probedelay = flapping ? mDNSPlatformOneSecond * 5 : mDNSPlatformOneSecond / 2;
const mDNSu8 numannounce = flapping ? (mDNSu8)1 : InitialAnnounceCount;
if (!m->SuppressSending) m->SuppressSending = m->timenow + (mDNSs32)mDNSRandom((mDNSu32)InitialQuestionInterval);
if (flapping) LogMsg("RegisterInterface: Frequent transitions for interface %s (%#a)", set->ifname, &set->ip);
LogInfo("RegisterInterface: %s (%#a) probedelay %d", set->ifname, &set->ip, probedelay);
if (m->SuppressProbes == 0 ||
m->SuppressProbes - NonZeroTime(m->timenow + probedelay) < 0)
m->SuppressProbes = NonZeroTime(m->timenow + probedelay);
for (q = m->Questions; q; q=q->next) if (mDNSOpaque16IsZero(q->TargetQID))
if (!q->InterfaceID || q->InterfaceID == set->InterfaceID) { mDNSBool dodelay = flapping && (q->FlappingInterface1 == set->InterfaceID || q->FlappingInterface2 == set->InterfaceID);
mDNSs32 initial = dodelay ? InitialQuestionInterval * QuestionIntervalStep2 : InitialQuestionInterval;
mDNSs32 qdelay = dodelay ? mDNSPlatformOneSecond * 5 : 0;
if (dodelay) LogInfo("No cache records expired for %##s (%s); okay to delay questions a little", q->qname.c, DNSTypeName(q->qtype));
if (!q->ThisQInterval || q->ThisQInterval > initial)
{
q->ThisQInterval = initial;
q->RequestUnicast = 2; }
q->LastQTime = m->timenow - q->ThisQInterval + qdelay;
q->RecentAnswerPkts = 0;
SetNextQueryTime(m,q);
}
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (!AuthRecord_uDNS(rr))
if (!rr->resrec.InterfaceID || rr->resrec.InterfaceID == set->InterfaceID)
{
if (rr->resrec.RecordType == kDNSRecordTypeVerified && !rr->DependentOn) rr->resrec.RecordType = kDNSRecordTypeUnique;
rr->ProbeCount = DefaultProbeCountForRecordType(rr->resrec.RecordType);
if (rr->AnnounceCount < numannounce) rr->AnnounceCount = numannounce;
rr->SendNSECNow = mDNSNULL;
InitializeLastAPTime(m, rr);
}
}
RestartRecordGetZoneData(m);
CheckSuppressUnusableQuestions(m);
mDNS_UpdateAllowSleep(m);
mDNS_Unlock(m);
return(mStatus_NoError);
}
mDNSexport void mDNS_DeregisterInterface(mDNS *const m, NetworkInterfaceInfo *set, mDNSBool flapping)
{
NetworkInterfaceInfo **p = &m->HostInterfaces;
mDNSBool revalidate = mDNSfalse;
if (m->KnownBugs & mDNS_KnownBug_PhantomInterfaces) revalidate = mDNStrue;
mDNS_Lock(m);
while (*p && *p != set) p=&(*p)->next;
if (!*p) { debugf("mDNS_DeregisterInterface: NetworkInterfaceInfo not found in list"); mDNS_Unlock(m); return; }
mDNS_DeactivateNetWake_internal(m, set);
*p = (*p)->next;
set->next = mDNSNULL;
if (!set->InterfaceActive)
{
NetworkInterfaceInfo *intf;
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->InterfaceActive && intf->InterfaceID == set->InterfaceID)
UpdateInterfaceProtocols(m, intf);
}
else
{
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, set->InterfaceID);
if (intf)
{
LogInfo("mDNS_DeregisterInterface: Another representative of InterfaceID %p %s (%#a) exists;"
" making it active", set->InterfaceID, set->ifname, &set->ip);
if (intf->InterfaceActive)
LogMsg("mDNS_DeregisterInterface: ERROR intf->InterfaceActive already set for %s (%#a)", set->ifname, &set->ip);
intf->InterfaceActive = mDNStrue;
UpdateInterfaceProtocols(m, intf);
if (intf->NetWake) mDNS_ActivateNetWake_internal(m, intf);
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->InterfaceID == set->InterfaceID && intf->ip.type == set->ip.type)
break;
if (!intf) revalidate = mDNStrue;
}
else
{
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *rr;
DNSQuestion *q;
DNSServer *s;
LogInfo("mDNS_DeregisterInterface: Last representative of InterfaceID %p %s (%#a) deregistered;"
" marking questions etc. dormant", set->InterfaceID, set->ifname, &set->ip);
if (set->McastTxRx && flapping)
LogMsg("DeregisterInterface: Frequent transitions for interface %s (%#a)", set->ifname, &set->ip);
for (q = m->Questions; q; q=q->next)
{
if (q->InterfaceID == set->InterfaceID) q->ThisQInterval = 0;
if (!q->InterfaceID || q->InterfaceID == set->InterfaceID)
{
q->FlappingInterface2 = q->FlappingInterface1;
q->FlappingInterface1 = set->InterfaceID; }
}
revalidate = mDNSfalse; FORALL_CACHERECORDS(slot, cg, rr)
if (rr->resrec.InterfaceID == set->InterfaceID)
{
if (set->McastTxRx && flapping)
{
mDNS_Reconfirm_internal(m, rr, kDefaultReconfirmTimeForFlappingInterface);
rr->UnansweredQueries = MaxUnansweredQueries;
}
else
mDNS_PurgeCacheResourceRecord(m, rr);
}
for (s = m->DNSServers; s; s = s->next)
if (s->interface == set->InterfaceID)
{
s->interface = mDNSInterface_Any;
s->teststate = DNSServer_Disabled;
}
}
}
if (set->Advertise) DeadvertiseInterface(m, set);
if (revalidate && !m->ShutdownTime)
{
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *rr;
FORALL_CACHERECORDS(slot, cg, rr)
if (rr->resrec.InterfaceID == set->InterfaceID)
mDNS_Reconfirm_internal(m, rr, kDefaultReconfirmTimeForFlappingInterface);
}
CheckSuppressUnusableQuestions(m);
mDNS_UpdateAllowSleep(m);
mDNS_Unlock(m);
}
mDNSlocal void ServiceCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
{
ServiceRecordSet *sr = (ServiceRecordSet *)rr->RecordContext;
(void)m;
#if MDNS_DEBUGMSGS
{
char *msg = "Unknown result";
if (result == mStatus_NoError) msg = "Name Registered";
else if (result == mStatus_NameConflict) msg = "Name Conflict";
else if (result == mStatus_MemFree) msg = "Memory Free";
debugf("ServiceCallback: %##s (%s) %s (%d)", rr->resrec.name->c, DNSTypeName(rr->resrec.rrtype), msg, result);
}
#endif
if (result == mStatus_NoError && rr != &sr->RR_SRV) return;
if (result == mStatus_NameConflict)
{
sr->Conflict = mDNStrue; mDNS_DeregisterService(m, sr); return;
}
if (result == mStatus_MemFree)
{
mDNSu32 i;
ExtraResourceRecord *e = sr->Extras;
if (sr->RR_SRV.resrec.RecordType != kDNSRecordTypeUnregistered) return;
if (sr->RR_TXT.resrec.RecordType != kDNSRecordTypeUnregistered) return;
if (sr->RR_PTR.resrec.RecordType != kDNSRecordTypeUnregistered) return;
if (sr->RR_ADV.resrec.RecordType != kDNSRecordTypeUnregistered) return;
for (i=0; i<sr->NumSubTypes; i++) if (sr->SubTypes[i].resrec.RecordType != kDNSRecordTypeUnregistered) return;
while (e)
{
if (e->r.resrec.RecordType != kDNSRecordTypeUnregistered) return;
e = e->next;
}
if (sr->Conflict) result = mStatus_NameConflict;
}
LogInfo("ServiceCallback: All records %s for %##s", (result == mStatus_MemFree ? "Unregistered": "Registered"), sr->RR_PTR.resrec.name->c);
if (sr->ServiceCallback)
sr->ServiceCallback(m, sr, result);
}
mDNSlocal void NSSCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
{
ServiceRecordSet *sr = (ServiceRecordSet *)rr->RecordContext;
if (sr->ServiceCallback)
sr->ServiceCallback(m, sr, result);
}
mDNSexport mStatus mDNS_RegisterService(mDNS *const m, ServiceRecordSet *sr,
const domainlabel *const name, const domainname *const type, const domainname *const domain,
const domainname *const host, mDNSIPPort port, const mDNSu8 txtinfo[], mDNSu16 txtlen,
AuthRecord *SubTypes, mDNSu32 NumSubTypes,
const mDNSInterfaceID InterfaceID, mDNSServiceCallback Callback, void *Context)
{
mStatus err;
mDNSu32 i;
sr->ServiceCallback = Callback;
sr->ServiceContext = Context;
sr->Conflict = mDNSfalse;
sr->Extras = mDNSNULL;
sr->NumSubTypes = NumSubTypes;
sr->SubTypes = SubTypes;
mDNS_SetupResourceRecord(&sr->RR_ADV, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeAdvisory, ServiceCallback, sr);
mDNS_SetupResourceRecord(&sr->RR_PTR, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, ServiceCallback, sr);
mDNS_SetupResourceRecord(&sr->RR_SRV, mDNSNULL, InterfaceID, kDNSType_SRV, kHostNameTTL, kDNSRecordTypeUnique, ServiceCallback, sr);
mDNS_SetupResourceRecord(&sr->RR_TXT, mDNSNULL, InterfaceID, kDNSType_TXT, kStandardTTL, kDNSRecordTypeUnique, ServiceCallback, sr);
if (mDNSIPPortIsZero(port))
return(mDNS_RegisterNoSuchService(m, &sr->RR_SRV, name, type, domain, mDNSNULL, mDNSInterface_Any, NSSCallback, sr));
if (sr->RR_TXT.resrec.rdata->MaxRDLength < txtlen)
sr->RR_TXT.resrec.rdata->MaxRDLength = txtlen;
if (ConstructServiceName(&sr->RR_ADV.namestorage, (const domainlabel*)"\x09_services", (const domainname*)"\x07_dns-sd\x04_udp", domain) == mDNSNULL)
return(mStatus_BadParamErr);
if (ConstructServiceName(&sr->RR_PTR.namestorage, mDNSNULL, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
if (ConstructServiceName(&sr->RR_SRV.namestorage, name, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
AssignDomainName(&sr->RR_TXT.namestorage, sr->RR_SRV.resrec.name);
AssignDomainName(&sr->RR_ADV.resrec.rdata->u.name, sr->RR_PTR.resrec.name);
AssignDomainName(&sr->RR_PTR.resrec.rdata->u.name, sr->RR_SRV.resrec.name);
sr->RR_PTR.Additional1 = &sr->RR_SRV;
sr->RR_PTR.Additional2 = &sr->RR_TXT;
for (i=0; i<NumSubTypes; i++)
{
domainname st;
AssignDomainName(&st, sr->SubTypes[i].resrec.name);
st.c[1+st.c[0]] = 0; AppendDomainName(&st, type);
mDNS_SetupResourceRecord(&sr->SubTypes[i], mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, ServiceCallback, sr);
if (ConstructServiceName(&sr->SubTypes[i].namestorage, mDNSNULL, &st, domain) == mDNSNULL) return(mStatus_BadParamErr);
AssignDomainName(&sr->SubTypes[i].resrec.rdata->u.name, &sr->RR_SRV.namestorage);
sr->SubTypes[i].Additional1 = &sr->RR_SRV;
sr->SubTypes[i].Additional2 = &sr->RR_TXT;
}
sr->RR_SRV.resrec.rdata->u.srv.priority = 0;
sr->RR_SRV.resrec.rdata->u.srv.weight = 0;
sr->RR_SRV.resrec.rdata->u.srv.port = port;
if (host && host->c[0]) AssignDomainName(&sr->RR_SRV.resrec.rdata->u.srv.target, host);
else { sr->RR_SRV.AutoTarget = Target_AutoHost; sr->RR_SRV.resrec.rdata->u.srv.target.c[0] = '\0'; }
if (txtinfo == mDNSNULL) sr->RR_TXT.resrec.rdlength = 0;
else if (txtinfo != sr->RR_TXT.resrec.rdata->u.txt.c)
{
sr->RR_TXT.resrec.rdlength = txtlen;
if (sr->RR_TXT.resrec.rdlength > sr->RR_TXT.resrec.rdata->MaxRDLength) return(mStatus_BadParamErr);
mDNSPlatformMemCopy(sr->RR_TXT.resrec.rdata->u.txt.c, txtinfo, txtlen);
}
sr->RR_TXT.DependentOn = &sr->RR_SRV;
mDNS_Lock(m);
err = mDNS_Register_internal(m, &sr->RR_SRV);
if (!err) err = mDNS_Register_internal(m, &sr->RR_TXT);
if (!err) err = mDNS_Register_internal(m, &sr->RR_ADV);
for (i=0; i<NumSubTypes; i++) if (!err) err = mDNS_Register_internal(m, &sr->SubTypes[i]);
if (!err) err = mDNS_Register_internal(m, &sr->RR_PTR);
mDNS_Unlock(m);
if (err) mDNS_DeregisterService(m, sr);
return(err);
}
mDNSexport mStatus mDNS_AddRecordToService(mDNS *const m, ServiceRecordSet *sr,
ExtraResourceRecord *extra, RData *rdata, mDNSu32 ttl)
{
ExtraResourceRecord **e;
mStatus status;
extra->next = mDNSNULL;
mDNS_SetupResourceRecord(&extra->r, rdata, sr->RR_PTR.resrec.InterfaceID,
extra->r.resrec.rrtype, ttl, kDNSRecordTypeUnique, ServiceCallback, sr);
AssignDomainName(&extra->r.namestorage, sr->RR_SRV.resrec.name);
mDNS_Lock(m);
e = &sr->Extras;
while (*e) e = &(*e)->next;
if (ttl == 0) ttl = kStandardTTL;
extra->r.DependentOn = &sr->RR_SRV;
debugf("mDNS_AddRecordToService adding record to %##s %s %d",
extra->r.resrec.name->c, DNSTypeName(extra->r.resrec.rrtype), extra->r.resrec.rdlength);
status = mDNS_Register_internal(m, &extra->r);
if (status == mStatus_NoError) *e = extra;
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_RemoveRecordFromService(mDNS *const m, ServiceRecordSet *sr, ExtraResourceRecord *extra,
mDNSRecordCallback MemFreeCallback, void *Context)
{
ExtraResourceRecord **e;
mStatus status;
mDNS_Lock(m);
e = &sr->Extras;
while (*e && *e != extra) e = &(*e)->next;
if (!*e)
{
debugf("mDNS_RemoveRecordFromService failed to remove record from %##s", extra->r.resrec.name->c);
status = mStatus_BadReferenceErr;
}
else
{
debugf("mDNS_RemoveRecordFromService removing record from %##s", extra->r.resrec.name->c);
extra->r.RecordCallback = MemFreeCallback;
extra->r.RecordContext = Context;
*e = (*e)->next;
status = mDNS_Deregister_internal(m, &extra->r, mDNS_Dereg_normal);
}
mDNS_Unlock(m);
return(status);
}
mDNSexport mStatus mDNS_RenameAndReregisterService(mDNS *const m, ServiceRecordSet *const sr, const domainlabel *newname)
{
domainlabel name1, name2;
domainname type, domain;
const domainname *host = sr->RR_SRV.AutoTarget ? mDNSNULL : &sr->RR_SRV.resrec.rdata->u.srv.target;
ExtraResourceRecord *extras = sr->Extras;
mStatus err;
DeconstructServiceName(sr->RR_SRV.resrec.name, &name1, &type, &domain);
if (!newname)
{
name2 = name1;
IncrementLabelSuffix(&name2, mDNStrue);
newname = &name2;
}
if (SameDomainName(&domain, &localdomain))
debugf("%##s service renamed from \"%#s\" to \"%#s\"", type.c, name1.c, newname->c);
else debugf("%##s service (domain %##s) renamed from \"%#s\" to \"%#s\"",type.c, domain.c, name1.c, newname->c);
err = mDNS_RegisterService(m, sr, newname, &type, &domain,
host, sr->RR_SRV.resrec.rdata->u.srv.port, sr->RR_TXT.resrec.rdata->u.txt.c, sr->RR_TXT.resrec.rdlength,
sr->SubTypes, sr->NumSubTypes,
sr->RR_PTR.resrec.InterfaceID, sr->ServiceCallback, sr->ServiceContext);
while (!err && extras)
{
ExtraResourceRecord *e = extras;
extras = extras->next;
err = mDNS_AddRecordToService(m, sr, e, e->r.resrec.rdata, e->r.resrec.rroriginalttl);
}
return(err);
}
mDNSexport mStatus mDNS_DeregisterService_drt(mDNS *const m, ServiceRecordSet *sr, mDNS_Dereg_type drt)
{
if (mDNSIPPortIsZero(sr->RR_SRV.resrec.rdata->u.srv.port)) return(mDNS_DeregisterNoSuchService(m, &sr->RR_SRV));
if (sr->RR_PTR.resrec.RecordType == kDNSRecordTypeUnregistered)
{
debugf("Service set for %##s already deregistered", sr->RR_SRV.resrec.name->c);
return(mStatus_BadReferenceErr);
}
else if (sr->RR_PTR.resrec.RecordType == kDNSRecordTypeDeregistering)
{
LogInfo("Service set for %##s already in the process of deregistering", sr->RR_SRV.resrec.name->c);
sr->Conflict = mDNSfalse;
return(mStatus_NoError);
}
else
{
mDNSu32 i;
mStatus status;
ExtraResourceRecord *e;
mDNS_Lock(m);
e = sr->Extras;
mDNS_Deregister_internal(m, &sr->RR_SRV, mDNS_Dereg_repeat);
mDNS_Deregister_internal(m, &sr->RR_TXT, mDNS_Dereg_repeat);
mDNS_Deregister_internal(m, &sr->RR_ADV, drt);
while (e)
{
mDNS_Deregister_internal(m, &e->r, mDNS_Dereg_repeat);
e = e->next;
}
for (i=0; i<sr->NumSubTypes; i++)
mDNS_Deregister_internal(m, &sr->SubTypes[i], drt);
status = mDNS_Deregister_internal(m, &sr->RR_PTR, drt);
mDNS_Unlock(m);
return(status);
}
}
mDNSexport mStatus mDNS_RegisterNoSuchService(mDNS *const m, AuthRecord *const rr,
const domainlabel *const name, const domainname *const type, const domainname *const domain,
const domainname *const host,
const mDNSInterfaceID InterfaceID, mDNSRecordCallback Callback, void *Context)
{
mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, kDNSType_SRV, kHostNameTTL, kDNSRecordTypeUnique, Callback, Context);
if (ConstructServiceName(&rr->namestorage, name, type, domain) == mDNSNULL) return(mStatus_BadParamErr);
rr->resrec.rdata->u.srv.priority = 0;
rr->resrec.rdata->u.srv.weight = 0;
rr->resrec.rdata->u.srv.port = zeroIPPort;
if (host && host->c[0]) AssignDomainName(&rr->resrec.rdata->u.srv.target, host);
else rr->AutoTarget = Target_AutoHost;
return(mDNS_Register(m, rr));
}
mDNSexport mStatus mDNS_AdvertiseDomains(mDNS *const m, AuthRecord *rr,
mDNS_DomainType DomainType, const mDNSInterfaceID InterfaceID, char *domname)
{
mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, kDNSType_PTR, kStandardTTL, kDNSRecordTypeShared, mDNSNULL, mDNSNULL);
if (!MakeDomainNameFromDNSNameString(&rr->namestorage, mDNS_DomainTypeNames[DomainType])) return(mStatus_BadParamErr);
if (!MakeDomainNameFromDNSNameString(&rr->resrec.rdata->u.name, domname)) return(mStatus_BadParamErr);
return(mDNS_Register(m, rr));
}
mDNSlocal mDNSBool mDNS_IdUsedInResourceRecordsList(mDNS * const m, mDNSOpaque16 id)
{
AuthRecord *r;
for (r = m->ResourceRecords; r; r=r->next) if (mDNSSameOpaque16(id, r->updateid)) return mDNStrue;
return mDNSfalse;
}
mDNSlocal mDNSBool mDNS_IdUsedInQuestionsList(mDNS * const m, mDNSOpaque16 id)
{
DNSQuestion *q;
for (q = m->Questions; q; q=q->next) if (mDNSSameOpaque16(id, q->TargetQID)) return mDNStrue;
return mDNSfalse;
}
mDNSexport mDNSOpaque16 mDNS_NewMessageID(mDNS * const m)
{
mDNSOpaque16 id;
int i;
for (i=0; i<10; i++)
{
id = mDNSOpaque16fromIntVal(1 + (mDNSu16)mDNSRandom(0xFFFE));
if (!mDNS_IdUsedInResourceRecordsList(m, id) && !mDNS_IdUsedInQuestionsList(m, id)) break;
}
debugf("mDNS_NewMessageID: %5d", mDNSVal16(id));
return id;
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Sleep Proxy Server
#endif
mDNSlocal void RestartARPProbing(mDNS *const m, AuthRecord *const rr)
{
rr->resrec.RecordType = kDNSRecordTypeUnique;
rr->ProbeCount = DefaultProbeCountForTypeUnique;
if (rr->AnnounceCount == InitialAnnounceCount && m->timenow - rr->LastAPTime >= 0)
InitializeLastAPTime(m, rr);
else
{
rr->AnnounceCount = InitialAnnounceCount;
rr->ThisAPInterval = mDNSPlatformOneSecond;
rr->LastAPTime = m->timenow + mDNSPlatformOneSecond * 9; SetNextAnnounceProbeTime(m, rr);
}
}
mDNSlocal void mDNSCoreReceiveRawARP(mDNS *const m, const ARP_EthIP *const arp, const mDNSInterfaceID InterfaceID)
{
static const mDNSOpaque16 ARP_op_request = { { 0, 1 } };
AuthRecord *rr;
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
if (!intf) return;
mDNS_Lock(m);
if (mDNSSameOpaque16(arp->op, ARP_op_request) && !mDNSSameIPv4Address(arp->spa, arp->tpa))
{
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
rr->AddressProxy.type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->AddressProxy.ip.v4, arp->tpa))
{
static const char msg1[] = "ARP Req from owner -- re-probing";
static const char msg2[] = "Ignoring ARP Request from ";
static const char msg3[] = "Creating Local ARP Cache entry ";
static const char msg4[] = "Answering ARP Request from ";
const char *const msg = mDNSSameEthAddress(&arp->sha, &rr->WakeUp.IMAC) ? msg1 :
(rr->AnnounceCount == InitialAnnounceCount) ? msg2 :
mDNSSameEthAddress(&arp->sha, &intf->MAC) ? msg3 : msg4;
LogSPS("%-7s %s %.6a %.4a for %.4a -- H-MAC %.6a I-MAC %.6a %s",
intf->ifname, msg, &arp->sha, &arp->spa, &arp->tpa, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
if (msg == msg1) RestartARPProbing(m, rr);
else if (msg == msg3) mDNSPlatformSetLocalAddressCacheEntry(m, &rr->AddressProxy, &rr->WakeUp.IMAC, InterfaceID);
else if (msg == msg4) SendARP(m, 2, rr, &arp->tpa, &arp->sha, &arp->spa, &arp->sha);
}
}
if (mDNSSameEthAddress(&arp->sha, &intf->MAC))
debugf("ARP from self for %.4a", &arp->tpa);
else
{
if (!mDNSSameIPv4Address(arp->spa, zerov4Addr))
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
rr->AddressProxy.type == mDNSAddrType_IPv4 && mDNSSameIPv4Address(rr->AddressProxy.ip.v4, arp->spa))
{
RestartARPProbing(m, rr);
if (mDNSSameEthAddress(&arp->sha, &rr->WakeUp.IMAC))
LogSPS("%-7s ARP %s from owner %.6a %.4a for %-15.4a -- re-starting probing for %s", intf->ifname,
mDNSSameIPv4Address(arp->spa, arp->tpa) ? "Announcement " : mDNSSameOpaque16(arp->op, ARP_op_request) ? "Request " : "Response ",
&arp->sha, &arp->spa, &arp->tpa, ARDisplayString(m, rr));
else
{
LogMsg("%-7s Conflicting ARP from %.6a %.4a for %.4a -- waking H-MAC %.6a I-MAC %.6a %s", intf->ifname,
&arp->sha, &arp->spa, &arp->tpa, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
}
}
}
mDNS_Unlock(m);
}
mDNSlocal void mDNSCoreReceiveRawND(mDNS *const m, const mDNSEthAddr *const sha, const mDNSv6Addr *spa,
const IPv6NDP *const ndp, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID)
{
AuthRecord *rr;
NetworkInterfaceInfo *intf = FirstInterfaceForID(m, InterfaceID);
if (!intf) return;
mDNS_Lock(m);
if (ndp->type == NDP_Sol)
{
(void)end;
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
rr->AddressProxy.type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->AddressProxy.ip.v6, ndp->target))
{
static const char msg1[] = "NDP Req from owner -- re-probing";
static const char msg2[] = "Ignoring NDP Request from ";
static const char msg3[] = "Creating Local NDP Cache entry ";
static const char msg4[] = "Answering NDP Request from ";
static const char msg5[] = "Answering NDP Probe from ";
const char *const msg = sha && mDNSSameEthAddress(sha, &rr->WakeUp.IMAC) ? msg1 :
(rr->AnnounceCount == InitialAnnounceCount) ? msg2 :
sha && mDNSSameEthAddress(sha, &intf->MAC) ? msg3 :
spa && mDNSIPv6AddressIsZero(*spa) ? msg4 : msg5;
LogSPS("%-7s %s %.6a %.16a for %.16a -- H-MAC %.6a I-MAC %.6a %s",
intf->ifname, msg, sha, spa, &ndp->target, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
if (msg == msg1) RestartARPProbing(m, rr);
else if (msg == msg3)
{
if (!(m->KnownBugs & mDNS_KnownBug_LimitedIPv6))
mDNSPlatformSetLocalAddressCacheEntry(m, &rr->AddressProxy, &rr->WakeUp.IMAC, InterfaceID);
}
else if (msg == msg4) SendNDP(m, NDP_Adv, NDP_Solicited, rr, &ndp->target, mDNSNULL, spa, sha );
else if (msg == msg5) SendNDP(m, NDP_Adv, 0, rr, &ndp->target, mDNSNULL, &AllHosts_v6, &AllHosts_v6_Eth);
}
}
if (mDNSSameEthAddress(sha, &intf->MAC))
debugf("NDP from self for %.16a", &ndp->target);
else
{
if (ndp->type == NDP_Adv) spa = &ndp->target;
if (!mDNSSameIPv6Address(*spa, zerov6Addr))
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID && rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
rr->AddressProxy.type == mDNSAddrType_IPv6 && mDNSSameIPv6Address(rr->AddressProxy.ip.v6, *spa))
{
RestartARPProbing(m, rr);
if (mDNSSameEthAddress(sha, &rr->WakeUp.IMAC))
LogSPS("%-7s NDP %s from owner %.6a %.16a for %.16a -- re-starting probing for %s", intf->ifname,
ndp->type == NDP_Sol ? "Solicitation " : "Advertisement", sha, spa, &ndp->target, ARDisplayString(m, rr));
else
{
LogMsg("%-7s Conflicting NDP from %.6a %.16a for %.16a -- waking H-MAC %.6a I-MAC %.6a %s", intf->ifname,
sha, spa, &ndp->target, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, rr));
ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
}
}
}
mDNS_Unlock(m);
}
mDNSlocal void mDNSCoreReceiveRawTransportPacket(mDNS *const m, const mDNSEthAddr *const sha, const mDNSAddr *const src, const mDNSAddr *const dst, const mDNSu8 protocol,
const mDNSu8 *const p, const TransportLayerPacket *const t, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID, const mDNSu16 len)
{
const mDNSIPPort port = (protocol == 0x06) ? t->tcp.dst : (protocol == 0x11) ? t->udp.dst : zeroIPPort;
mDNSBool wake = mDNSfalse;
switch (protocol)
{
#define XX wake ? "Received" : "Ignoring", end-p
case 0x01: LogSPS("Ignoring %d-byte ICMP from %#a to %#a", end-p, src, dst);
break;
case 0x06: {
#define SSH_AsNumber 22
static const mDNSIPPort SSH = { { SSH_AsNumber >> 8, SSH_AsNumber & 0xFF } };
wake = (!(t->tcp.flags & 4) && (t->tcp.flags & 3) != 1);
if (!mDNSSameIPPort(port, SSH) && !(t->tcp.flags & 2)) wake = mDNSfalse;
LogSPS("%s %d-byte TCP from %#a:%d to %#a:%d%s%s%s", XX,
src, mDNSVal16(t->tcp.src), dst, mDNSVal16(port),
(t->tcp.flags & 2) ? " SYN" : "",
(t->tcp.flags & 1) ? " FIN" : "",
(t->tcp.flags & 4) ? " RST" : "");
}
break;
case 0x11: {
#define ARD_AsNumber 3283
static const mDNSIPPort ARD = { { ARD_AsNumber >> 8, ARD_AsNumber & 0xFF } };
const mDNSu16 udplen = (mDNSu16)((mDNSu16)t->bytes[4] << 8 | t->bytes[5]); if (udplen >= sizeof(UDPHeader))
{
const mDNSu16 datalen = udplen - sizeof(UDPHeader);
wake = mDNStrue;
if (mDNSSameIPPort(port, IPSECPort))
{
if (datalen == 1 && end >= &t->bytes[9] && t->bytes[8] == 0xFF) wake = mDNSfalse;
else
{
const mDNSBool NonESP = (end >= &t->bytes[12] && t->bytes[8] == 0 && t->bytes[9] == 0 && t->bytes[10] == 0 && t->bytes[11] == 0);
const IKEHeader *const ike = (IKEHeader *)(t + (NonESP ? 12 : 8));
const mDNSu16 ikelen = datalen - (NonESP ? 4 : 0);
if (ikelen >= sizeof(IKEHeader) && end >= ((mDNSu8 *)ike) + sizeof(IKEHeader))
if ((ike->Version & 0x10) == 0x10)
{
if (ike->ExchangeType == 5 || ike->ExchangeType == 34) wake = mDNSfalse;
LogSPS("%s %d-byte IKE ExchangeType %d", XX, ike->ExchangeType);
}
}
}
if (mDNSSameIPPort(port, ARD)) wake = (datalen >= 110 && end >= &t->bytes[10] && t->bytes[8] == 0x13 && t->bytes[9] == 0x88);
LogSPS("%s %d-byte UDP from %#a:%d to %#a:%d", XX, src, mDNSVal16(t->udp.src), dst, mDNSVal16(port));
}
}
break;
case 0x3A: if (&t->bytes[len] <= end)
{
mDNSu16 checksum = IPv6CheckSum(&src->ip.v6, &dst->ip.v6, protocol, t->bytes, len);
if (!checksum) mDNSCoreReceiveRawND(m, sha, &src->ip.v6, &t->ndp, &t->bytes[len], InterfaceID);
else LogInfo("IPv6CheckSum bad %04X %02X%02X from %#a to %#a", checksum, t->bytes[2], t->bytes[3], src, dst);
}
break;
default: LogSPS("Ignoring %d-byte IP packet unknown protocol %d from %#a to %#a", end-p, protocol, src, dst);
break;
}
if (wake)
{
AuthRecord *rr, *r2;
mDNS_Lock(m);
for (rr = m->ResourceRecords; rr; rr=rr->next)
if (rr->resrec.InterfaceID == InterfaceID &&
rr->resrec.RecordType != kDNSRecordTypeDeregistering &&
rr->AddressProxy.type && mDNSSameAddress(&rr->AddressProxy, dst))
{
const mDNSu8 *const tp = (protocol == 6) ? (const mDNSu8 *)"\x4_tcp" : (const mDNSu8 *)"\x4_udp";
for (r2 = m->ResourceRecords; r2; r2=r2->next)
if (r2->resrec.InterfaceID == InterfaceID && mDNSSameEthAddress(&r2->WakeUp.HMAC, &rr->WakeUp.HMAC) &&
r2->resrec.RecordType != kDNSRecordTypeDeregistering &&
r2->resrec.rrtype == kDNSType_SRV && mDNSSameIPPort(r2->resrec.rdata->u.srv.port, port) &&
SameDomainLabel(ThirdLabel(r2->resrec.name)->c, tp))
break;
if (!r2 && mDNSSameIPPort(port, IPSECPort)) r2 = rr; if (r2)
{
LogMsg("Waking host at %s %#a H-MAC %.6a I-MAC %.6a for %s",
InterfaceNameForID(m, rr->resrec.InterfaceID), dst, &rr->WakeUp.HMAC, &rr->WakeUp.IMAC, ARDisplayString(m, r2));
ScheduleWakeup(m, rr->resrec.InterfaceID, &rr->WakeUp.HMAC);
}
else
LogSPS("Sleeping host at %s %#a %.6a has no service on %#s %d",
InterfaceNameForID(m, rr->resrec.InterfaceID), dst, &rr->WakeUp.HMAC, tp, mDNSVal16(port));
}
mDNS_Unlock(m);
}
}
mDNSexport void mDNSCoreReceiveRawPacket(mDNS *const m, const mDNSu8 *const p, const mDNSu8 *const end, const mDNSInterfaceID InterfaceID)
{
static const mDNSOpaque16 Ethertype_ARP = { { 0x08, 0x06 } }; static const mDNSOpaque16 Ethertype_IPv4 = { { 0x08, 0x00 } }; static const mDNSOpaque16 Ethertype_IPv6 = { { 0x86, 0xDD } }; static const mDNSOpaque16 ARP_hrd_eth = { { 0x00, 0x01 } }; static const mDNSOpaque16 ARP_pro_ip = { { 0x08, 0x00 } };
const EthernetHeader *const eth = (const EthernetHeader *)p;
const NetworkLayerPacket *const pkt = (const NetworkLayerPacket *)(eth+1);
mDNSAddr src, dst;
#define RequiredCapLen(P) ((P)==0x01 ? 4 : (P)==0x06 ? 20 : (P)==0x11 ? 8 : (P)==0x3A ? 24 : 0)
if (end >= p+42 && mDNSSameOpaque16(eth->ethertype, Ethertype_ARP) && mDNSSameOpaque16(pkt->arp.hrd, ARP_hrd_eth) && mDNSSameOpaque16(pkt->arp.pro, ARP_pro_ip))
mDNSCoreReceiveRawARP(m, &pkt->arp, InterfaceID);
else if (end >= p+34 && mDNSSameOpaque16(eth->ethertype, Ethertype_IPv4) && (pkt->v4.flagsfrags.b[0] & 0x1F) == 0 && pkt->v4.flagsfrags.b[1] == 0)
{
const mDNSu8 *const trans = p + 14 + (pkt->v4.vlen & 0xF) * 4;
debugf("Got IPv4 %02X from %.4a to %.4a", pkt->v4.protocol, &pkt->v4.src, &pkt->v4.dst);
src.type = mDNSAddrType_IPv4; src.ip.v4 = pkt->v4.src;
dst.type = mDNSAddrType_IPv4; dst.ip.v4 = pkt->v4.dst;
if (end >= trans + RequiredCapLen(pkt->v4.protocol))
mDNSCoreReceiveRawTransportPacket(m, ð->src, &src, &dst, pkt->v4.protocol, p, (TransportLayerPacket*)trans, end, InterfaceID, 0);
}
else if (end >= p+54 && mDNSSameOpaque16(eth->ethertype, Ethertype_IPv6))
{
const mDNSu8 *const trans = p + 54;
debugf("Got IPv6 %02X from %.16a to %.16a", pkt->v6.pro, &pkt->v6.src, &pkt->v6.dst);
src.type = mDNSAddrType_IPv6; src.ip.v6 = pkt->v6.src;
dst.type = mDNSAddrType_IPv6; dst.ip.v6 = pkt->v6.dst;
if (end >= trans + RequiredCapLen(pkt->v6.pro))
mDNSCoreReceiveRawTransportPacket(m, ð->src, &src, &dst, pkt->v6.pro, p, (TransportLayerPacket*)trans, end, InterfaceID,
(mDNSu16)pkt->bytes[4] << 8 | pkt->bytes[5]);
}
}
mDNSlocal void ConstructSleepProxyServerName(mDNS *const m, domainlabel *name)
{
name->c[0] = (mDNSu8)mDNS_snprintf((char*)name->c+1, 62, "%d-%d-%d-%d %#s",
m->SPSType, m->SPSPortability, m->SPSMarginalPower, m->SPSTotalPower, &m->nicelabel);
}
mDNSlocal void SleepProxyServerCallback(mDNS *const m, ServiceRecordSet *const srs, mStatus result)
{
if (result == mStatus_NameConflict)
mDNS_RenameAndReregisterService(m, srs, mDNSNULL);
else if (result == mStatus_MemFree)
{
if (m->SleepState)
m->SPSState = 3;
else
{
m->SPSState = (mDNSu8)(m->SPSSocket != mDNSNULL);
if (m->SPSState)
{
domainlabel name;
ConstructSleepProxyServerName(m, &name);
mDNS_RegisterService(m, srs,
&name, &SleepProxyServiceType, &localdomain,
mDNSNULL, m->SPSSocket->port, (mDNSu8 *)"", 1, mDNSNULL, 0, mDNSInterface_Any, SleepProxyServerCallback, mDNSNULL); }
LogSPS("Sleep Proxy Server %#s %s", srs->RR_SRV.resrec.name->c, m->SPSState ? "started" : "stopped");
}
}
}
mDNSexport void mDNSCoreBeSleepProxyServer_internal(mDNS *const m, mDNSu8 sps, mDNSu8 port, mDNSu8 marginalpower, mDNSu8 totpower)
{
mDNS_DropLockBeforeCallback();
if (!sps && m->SPSSocket) { mDNSPlatformUDPClose(m->SPSSocket); m->SPSSocket = mDNSNULL; }
if (m->SPSState == 1 && sps != m->SPSType)
{ m->SPSState = 2; mDNS_DeregisterService_drt(m, &m->SPSRecords, sps ? mDNS_Dereg_rapid : mDNS_Dereg_normal); }
m->SPSType = sps;
m->SPSPortability = port;
m->SPSMarginalPower = marginalpower;
m->SPSTotalPower = totpower;
if (sps)
{
if (!m->SPSSocket)
{
m->SPSSocket = mDNSPlatformUDPSocket(m, zeroIPPort);
if (!m->SPSSocket) { LogMsg("mDNSCoreBeSleepProxyServer: Failed to allocate SPSSocket"); goto fail; }
}
if (m->SPSState == 0) SleepProxyServerCallback(m, &m->SPSRecords, mStatus_MemFree);
}
else if (m->SPSState)
{
LogSPS("mDNSCoreBeSleepProxyServer turning off from state %d; will wake clients", m->SPSState);
m->NextScheduledSPS = m->timenow;
}
fail:
mDNS_ReclaimLockAfterCallback();
}
#if COMPILER_LIKES_PRAGMA_MARK
#pragma mark -
#pragma mark - Startup and Shutdown
#endif
mDNSlocal void mDNS_GrowCache_internal(mDNS *const m, CacheEntity *storage, mDNSu32 numrecords)
{
if (storage && numrecords)
{
mDNSu32 i;
debugf("Adding cache storage for %d more records (%d bytes)", numrecords, numrecords*sizeof(CacheEntity));
for (i=0; i<numrecords; i++) storage[i].next = &storage[i+1];
storage[numrecords-1].next = m->rrcache_free;
m->rrcache_free = storage;
m->rrcache_size += numrecords;
}
}
mDNSexport void mDNS_GrowCache(mDNS *const m, CacheEntity *storage, mDNSu32 numrecords)
{
mDNS_Lock(m);
mDNS_GrowCache_internal(m, storage, numrecords);
mDNS_Unlock(m);
}
mDNSexport mStatus mDNS_Init(mDNS *const m, mDNS_PlatformSupport *const p,
CacheEntity *rrcachestorage, mDNSu32 rrcachesize,
mDNSBool AdvertiseLocalAddresses, mDNSCallback *Callback, void *Context)
{
mDNSu32 slot;
mDNSs32 timenow;
mStatus result;
if (!rrcachestorage) rrcachesize = 0;
m->p = p;
m->KnownBugs = 0;
m->CanReceiveUnicastOn5353 = mDNSfalse; m->AdvertiseLocalAddresses = AdvertiseLocalAddresses;
m->DivertMulticastAdvertisements = mDNSfalse;
m->mDNSPlatformStatus = mStatus_Waiting;
m->UnicastPort4 = zeroIPPort;
m->UnicastPort6 = zeroIPPort;
m->PrimaryMAC = zeroEthAddr;
m->MainCallback = Callback;
m->MainContext = Context;
m->rec.r.resrec.RecordType = 0;
m->mDNS_busy = 0;
m->mDNS_reentrancy = 0;
m->ShutdownTime = 0;
m->lock_rrcache = 0;
m->lock_Questions = 0;
m->lock_Records = 0;
result = mDNSPlatformTimeInit();
if (result != mStatus_NoError) return(result);
m->timenow_adjust = (mDNSs32)mDNSRandom(0xFFFFFFFF);
timenow = mDNS_TimeNow_NoLock(m);
m->timenow = 0; m->timenow_last = timenow;
m->NextScheduledEvent = timenow;
m->SuppressSending = timenow;
m->NextCacheCheck = timenow + 0x78000000;
m->NextScheduledQuery = timenow + 0x78000000;
m->NextScheduledProbe = timenow + 0x78000000;
m->NextScheduledResponse = timenow + 0x78000000;
m->NextScheduledNATOp = timenow + 0x78000000;
m->NextScheduledSPS = timenow + 0x78000000;
m->RandomQueryDelay = 0;
m->RandomReconfirmDelay = 0;
m->PktNum = 0;
m->LocalRemoveEvents = mDNSfalse;
m->SleepState = SleepState_Awake;
m->SleepSeqNum = 0;
m->SystemWakeOnLANEnabled = mDNSfalse;
m->SentSleepProxyRegistration = mDNSfalse;
m->AnnounceOwner = NonZeroTime(timenow + 60 * mDNSPlatformOneSecond);
m->DelaySleep = 0;
m->SleepLimit = 0;
m->Questions = mDNSNULL;
m->NewQuestions = mDNSNULL;
m->CurrentQuestion = mDNSNULL;
m->LocalOnlyQuestions = mDNSNULL;
m->NewLocalOnlyQuestions = mDNSNULL;
m->rrcache_size = 0;
m->rrcache_totalused = 0;
m->rrcache_active = 0;
m->rrcache_report = 10;
m->rrcache_free = mDNSNULL;
for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
{
m->rrcache_hash[slot] = mDNSNULL;
m->rrcache_nextcheck[slot] = timenow + 0x78000000;;
}
mDNS_GrowCache_internal(m, rrcachestorage, rrcachesize);
m->hostlabel.c[0] = 0;
m->nicelabel.c[0] = 0;
m->MulticastHostname.c[0] = 0;
m->HIHardware.c[0] = 0;
m->HISoftware.c[0] = 0;
m->ResourceRecords = mDNSNULL;
m->DuplicateRecords = mDNSNULL;
m->NewLocalRecords = mDNSNULL;
m->CurrentRecord = mDNSNULL;
m->HostInterfaces = mDNSNULL;
m->ProbeFailTime = 0;
m->NumFailedProbes = 0;
m->SuppressProbes = 0;
#ifndef UNICAST_DISABLED
m->NextuDNSEvent = timenow + 0x78000000;
m->NextSRVUpdate = timenow + 0x78000000;
m->DNSServers = mDNSNULL;
m->Router = zeroAddr;
m->AdvertisedV4 = zeroAddr;
m->AdvertisedV6 = zeroAddr;
m->AuthInfoList = mDNSNULL;
m->ReverseMap.ThisQInterval = -1;
m->StaticHostname.c[0] = 0;
m->FQDN.c[0] = 0;
m->Hostnames = mDNSNULL;
m->AutoTunnelHostAddr.b[0] = 0;
m->AutoTunnelHostAddrActive = mDNSfalse;
m->AutoTunnelLabel.c[0] = 0;
m->RegisterSearchDomains = mDNSfalse;
m->RegisterAutoTunnel6 = mDNStrue;
m->NATTraversals = mDNSNULL;
m->CurrentNATTraversal = mDNSNULL;
m->retryIntervalGetAddr = 0; m->retryGetAddr = timenow + 0x78000000; m->ExternalAddress = zerov4Addr;
m->NATMcastRecvskt = mDNSNULL;
m->LastNATupseconds = 0;
m->LastNATReplyLocalTime = timenow;
m->LastNATMapResultCode = NATErr_None;
m->UPnPInterfaceID = 0;
m->SSDPSocket = mDNSNULL;
m->SSDPWANPPPConnection = mDNSfalse;
m->UPnPRouterPort = zeroIPPort;
m->UPnPSOAPPort = zeroIPPort;
m->UPnPRouterURL = mDNSNULL;
m->UPnPWANPPPConnection = mDNSfalse;
m->UPnPSOAPURL = mDNSNULL;
m->UPnPRouterAddressString = mDNSNULL;
m->UPnPSOAPAddressString = mDNSNULL;
m->SPSType = 0;
m->SPSPortability = 0;
m->SPSMarginalPower = 0;
m->SPSTotalPower = 0;
m->SPSState = 0;
m->SPSProxyListChanged = mDNSNULL;
m->SPSSocket = mDNSNULL;
m->SPSBrowseCallback = mDNSNULL;
m->ProxyRecords = 0;
#endif
#if APPLE_OSX_mDNSResponder
m->TunnelClients = mDNSNULL;
#if ! NO_WCF
CHECK_WCF_FUNCTION(WCFConnectionNew)
{
m->WCF = WCFConnectionNew();
if (!m->WCF) { LogMsg("WCFConnectionNew failed"); return -1; }
}
#endif
#endif
result = mDNSPlatformInit(m);
#ifndef UNICAST_DISABLED
uDNS_SetupDNSConfig(m); #endif
return(result);
}
mDNSexport void mDNS_ConfigChanged(mDNS *const m)
{
if (m->SPSState == 1)
{
domainlabel name, newname;
domainname type, domain;
DeconstructServiceName(m->SPSRecords.RR_SRV.resrec.name, &name, &type, &domain);
ConstructSleepProxyServerName(m, &newname);
if (!SameDomainLabelCS(name.c, newname.c))
{
LogSPS("Renaming SPS from ā%#sā to ā%#sā", name.c, newname.c);
m->SPSState = 2;
mDNS_DeregisterService_drt(m, &m->SPSRecords, mDNS_Dereg_rapid);
}
}
if (m->MainCallback)
m->MainCallback(m, mStatus_ConfigChanged);
}
mDNSlocal void DynDNSHostNameCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
{
(void)m; debugf("NameStatusCallback: result %d for registration of name %##s", result, rr->resrec.name->c);
mDNSPlatformDynDNSHostNameStatusChanged(rr->resrec.name, result);
}
mDNSlocal void PurgeOrReconfirmCacheRecord(mDNS *const m, CacheRecord *cr, const DNSServer * const ptr, mDNSBool lameduck)
{
mDNSBool purge = cr->resrec.RecordType == kDNSRecordTypePacketNegative ||
cr->resrec.rrtype == kDNSType_A ||
cr->resrec.rrtype == kDNSType_AAAA ||
cr->resrec.rrtype == kDNSType_SRV;
(void) lameduck;
(void) ptr;
debugf("PurgeOrReconfirmCacheRecord: %s cache record due to %s server %p %#a:%d (%##s): %s",
purge ? "purging" : "reconfirming",
lameduck ? "lame duck" : "new",
ptr, &ptr->addr, mDNSVal16(ptr->port), ptr->domain.c, CRDisplayString(m, cr));
if (purge)
{
LogInfo("PurgeorReconfirmCacheRecord: Purging Resourcerecord %s, RecordType %x", CRDisplayString(m, cr), cr->resrec.RecordType);
mDNS_PurgeCacheResourceRecord(m, cr);
}
else
{
LogInfo("PurgeorReconfirmCacheRecord: Reconfirming Resourcerecord %s, RecordType %x", CRDisplayString(m, cr), cr->resrec.RecordType);
mDNS_Reconfirm_internal(m, cr, kDefaultReconfirmTimeForNoAnswer);
}
}
mDNSlocal void CacheRecordResetDNSServer(mDNS *const m, DNSQuestion *q, DNSServer *new)
{
const mDNSu32 slot = HashSlot(&q->qname);
CacheGroup *const cg = CacheGroupForName(m, slot, q->qnamehash, &q->qname);
CacheRecord *rp;
mDNSBool found = mDNSfalse;
mDNSBool foundNew = mDNSfalse;
DNSServer *old = q->qDNSServer;
mDNSBool newQuestion = IsQuestionNew(m, q);
DNSQuestion *qptr;
q->qDNSServer = new;
for (rp = cg ? cg->members : mDNSNULL; rp; rp = rp->next)
{
if (SameNameRecordAnswersQuestion(&rp->resrec, q))
{
LogInfo("CacheRecordResetDNSServer: Found cache record %##s for new DNSServer address: %#a", rp->resrec.name->c,
(rp->resrec.rDNSServer != mDNSNULL ? &rp->resrec.rDNSServer->addr : mDNSNULL));
foundNew = mDNStrue;
break;
}
}
q->qDNSServer = old;
for (rp = cg ? cg->members : mDNSNULL; rp; rp = rp->next)
{
if (SameNameRecordAnswersQuestion(&rp->resrec, q))
{
found = mDNStrue;
if (foundNew)
{
LogInfo("CacheRecordResetDNSServer: Flushing Resourcerecord %##s, before:%#a, after:%#a", rp->resrec.name->c,
(rp->resrec.rDNSServer != mDNSNULL ? &rp->resrec.rDNSServer->addr : mDNSNULL),
(new != mDNSNULL ? &new->addr : mDNSNULL));
mDNS_PurgeCacheResourceRecord(m, rp);
if (newQuestion)
{
if (rp->CRActiveQuestion != mDNSNULL)
{
LogMsg("CacheRecordResetDNSServer: ERROR!!: CRActiveQuestion %p set, current question %p, name %##s", rp->CRActiveQuestion, q, q->qname.c);
rp->CRActiveQuestion = mDNSNULL;
}
continue;
}
}
LogInfo("CacheRecordResetDNSServer: resetting cache record %##s DNSServer address before:%#a,"
" after:%#a, CRActiveQuestion %p", rp->resrec.name->c, (rp->resrec.rDNSServer != mDNSNULL ?
&rp->resrec.rDNSServer->addr : mDNSNULL), (new != mDNSNULL ? &new->addr : mDNSNULL),
rp->CRActiveQuestion);
rp->resrec.rDNSServer = new;
}
}
if ((found && foundNew) || (!found && foundNew))
{
if (newQuestion)
LogInfo("CacheRecordResetDNSServer: deliverAddEvents not set for question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
else if (QuerySuppressed(q))
LogInfo("CacheRecordResetDNSServer: deliverAddEvents not set for suppressed question %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
else
{
LogInfo("CacheRecordResetDNSServer: deliverAddEvents set for %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
q->deliverAddEvents = mDNStrue;
for (qptr = q->next; qptr; qptr = qptr->next)
if (qptr->DuplicateOf == q) qptr->deliverAddEvents = mDNStrue;
}
return;
}
return;
}
mDNSexport void DNSServerChangeForQuestion(mDNS *const m, DNSQuestion *q, DNSServer *new)
{
DNSQuestion *qptr;
q->TargetQID = mDNS_NewMessageID(m);
if (q->DuplicateOf)
LogMsg("DNSServerChangeForQuestion: ERROR: Called for duplicate question %##s", q->qname.c);
else
CacheRecordResetDNSServer(m, q, new);
q->qDNSServer = new;
for (qptr = q->next ; qptr; qptr = qptr->next)
{
if (qptr->DuplicateOf == q) { qptr->validDNSServers = q->validDNSServers; qptr->qDNSServer = new; }
}
}
mDNSexport mStatus uDNS_SetupDNSConfig(mDNS *const m)
{
mDNSu32 slot;
CacheGroup *cg;
CacheRecord *cr;
mDNSAddr v4, v6, r;
domainname fqdn;
DNSServer *ptr, **p = &m->DNSServers;
const DNSServer *oldServers = m->DNSServers;
DNSQuestion *q;
debugf("uDNS_SetupDNSConfig: entry");
if (m->RegisterSearchDomains) uDNS_RegisterSearchDomains(m);
mDNS_Lock(m);
for (ptr = m->DNSServers; ptr; ptr = ptr->next)
{
ptr->penaltyTime = 0;
ptr->flags |= DNSServer_FlagDelete;
}
mDNSPlatformSetDNSConfig(m, mDNStrue, mDNSfalse, &fqdn, mDNSNULL, mDNSNULL);
FORALL_CACHERECORDS(slot, cg, cr)
{
if (cr->resrec.InterfaceID) continue;
ptr = GetServerForName(m, cr->resrec.name, cr->CRActiveQuestion ? cr->CRActiveQuestion->InterfaceID : mDNSNULL);
if (ptr && (ptr != cr->resrec.rDNSServer))
{
if (cr->CRActiveQuestion == mDNSNULL)
{
LogInfo("uDNS_SetupDNSConfig: Purging Resourcerecord %s", CRDisplayString(m, cr));
mDNS_PurgeCacheResourceRecord(m, cr);
}
else
PurgeOrReconfirmCacheRecord(m, cr, ptr, mDNSfalse);
}
}
for (q = m->Questions; q; q=q->next)
if (!mDNSOpaque16IsZero(q->TargetQID))
{
DNSServer *s, *t;
DNSQuestion *qptr;
if (q->DuplicateOf) continue;
SetValidDNSServers(m, q);
q->triedAllServersOnce = 0;
s = GetServerForQuestion(m, q);
t = q->qDNSServer;
if (t != s)
{
debugf("uDNS_SetupDNSConfig: Updating DNS Server from %p %#a:%d (%##s) to %p %#a:%d (%##s) for %##s (%s)",
t, t ? &t->addr : mDNSNULL, mDNSVal16(t ? t->port : zeroIPPort), t ? t->domain.c : (mDNSu8*)"",
s, s ? &s->addr : mDNSNULL, mDNSVal16(s ? s->port : zeroIPPort), s ? s->domain.c : (mDNSu8*)"",
q->qname.c, DNSTypeName(q->qtype));
DNSServerChangeForQuestion(m, q, s);
q->unansweredQueries = 0;
if (!QuerySuppressed(q))
{
debugf("uDNS_SetupDNSConfig: Activating query %p %##s (%s)", q, q->qname.c, DNSTypeName(q->qtype));
ActivateUnicastQuery(m, q, mDNStrue);
for (qptr = q->next ; qptr; qptr = qptr->next)
{
if (qptr->DuplicateOf == q) ActivateUnicastQuery(m, qptr, mDNStrue);
}
}
}
else
{
debugf("uDNS_SetupDNSConfig: Not Updating DNS server question %p %##s (%s) DNS server %#a:%d %p %d",
q, q->qname.c, DNSTypeName(q->qtype), t ? &t->addr : mDNSNULL, mDNSVal16(t ? t->port : zeroIPPort), q->DuplicateOf, q->SuppressUnusable);
for (qptr = q->next ; qptr; qptr = qptr->next)
if (qptr->DuplicateOf == q) { qptr->validDNSServers = q->validDNSServers; qptr->qDNSServer = q->qDNSServer; }
}
}
while (*p)
{
if (((*p)->flags & DNSServer_FlagDelete) != 0)
{
ptr = *p;
FORALL_CACHERECORDS(slot, cg, cr)
{
if (cr->resrec.InterfaceID) continue;
if (cr->resrec.rDNSServer == ptr)
{
if (cr->CRActiveQuestion)
{
DNSQuestion *qptr = cr->CRActiveQuestion;
if (qptr->qDNSServer == mDNSNULL)
LogMsg("uDNS_SetupDNSConfig: Cache Record %s match: Active question %##s (%s) with DNSServer Address NULL, Server to be deleted %#a",
CRDisplayString(m, cr), qptr->qname.c, DNSTypeName(qptr->qtype), &ptr->addr);
else
LogMsg("uDNS_SetupDNSConfig: Cache Record %s match: Active question %##s (%s) DNSServer Address %#a, Server to be deleted %#a",
CRDisplayString(m, cr), qptr->qname.c, DNSTypeName(qptr->qtype), &qptr->qDNSServer->addr, &ptr->addr);
if (qptr->qDNSServer == ptr)
{
qptr->validDNSServers = zeroOpaque64;
qptr->qDNSServer = mDNSNULL;
cr->resrec.rDNSServer = mDNSNULL;
}
else
{
cr->resrec.rDNSServer = qptr->qDNSServer;
}
}
else
{
LogInfo("uDNS_SetupDNSConfig: Cache Record %##s has no Active question, Record's DNSServer Address %#a, Server to be deleted %#a",
cr->resrec.name, &cr->resrec.rDNSServer->addr, &ptr->addr);
cr->resrec.rDNSServer = mDNSNULL;
}
PurgeOrReconfirmCacheRecord(m, cr, ptr, mDNStrue);
}
}
*p = (*p)->next;
debugf("uDNS_SetupDNSConfig: Deleting server %p %#a:%d (%##s)", ptr, &ptr->addr, mDNSVal16(ptr->port), ptr->domain.c);
mDNSPlatformMemFree(ptr);
NumUnicastDNSServers--;
}
else
{
(*p)->flags &= ~DNSServer_FlagNew;
p = &(*p)->next;
}
}
if ((m->DNSServers != mDNSNULL) != (oldServers != mDNSNULL))
{
int count = 0;
FORALL_CACHERECORDS(slot, cg, cr) if (!cr->resrec.InterfaceID) { mDNS_PurgeCacheResourceRecord(m, cr); count++; }
LogInfo("uDNS_SetupDNSConfig: %s available; purged %d unicast DNS records from cache",
m->DNSServers ? "DNS server became" : "No DNS servers", count);
RestartRecordGetZoneData(m);
}
if (!SameDomainName(&fqdn, &m->FQDN))
{
if (m->FQDN.c[0]) mDNS_RemoveDynDNSHostName(m, &m->FQDN);
AssignDomainName(&m->FQDN, &fqdn);
if (m->FQDN.c[0])
{
mDNSPlatformDynDNSHostNameStatusChanged(&m->FQDN, 1);
mDNS_AddDynDNSHostName(m, &m->FQDN, DynDNSHostNameCallback, mDNSNULL);
}
}
mDNS_Unlock(m);
v4 = v6 = r = zeroAddr;
v4.type = r.type = mDNSAddrType_IPv4;
if (mDNSPlatformGetPrimaryInterface(m, &v4, &v6, &r) == mStatus_NoError && !mDNSv4AddressIsLinkLocal(&v4.ip.v4))
{
mDNS_SetPrimaryInterfaceInfo(m,
!mDNSIPv4AddressIsZero(v4.ip.v4) ? &v4 : mDNSNULL,
!mDNSIPv6AddressIsZero(v6.ip.v6) ? &v6 : mDNSNULL,
!mDNSIPv4AddressIsZero(r .ip.v4) ? &r : mDNSNULL);
}
else
{
mDNS_SetPrimaryInterfaceInfo(m, mDNSNULL, mDNSNULL, mDNSNULL);
if (m->FQDN.c[0]) mDNSPlatformDynDNSHostNameStatusChanged(&m->FQDN, 1); }
debugf("uDNS_SetupDNSConfig: number of unicast DNS servers %d", NumUnicastDNSServers);
return mStatus_NoError;
}
mDNSexport void mDNSCoreInitComplete(mDNS *const m, mStatus result)
{
m->mDNSPlatformStatus = result;
if (m->MainCallback)
{
mDNS_Lock(m);
mDNS_DropLockBeforeCallback(); m->MainCallback(m, mStatus_NoError);
mDNS_ReclaimLockAfterCallback(); mDNS_Unlock(m);
}
}
mDNSlocal void DeregLoop(mDNS *const m, AuthRecord *const start)
{
m->CurrentRecord = start;
while (m->CurrentRecord)
{
AuthRecord *rr = m->CurrentRecord;
LogInfo("DeregLoop: %s deregistration for %p %02X %s",
(rr->resrec.RecordType != kDNSRecordTypeDeregistering) ? "Initiating " : "Accelerating",
rr, rr->resrec.RecordType, ARDisplayString(m, rr));
if (rr->resrec.RecordType != kDNSRecordTypeDeregistering)
mDNS_Deregister_internal(m, rr, mDNS_Dereg_rapid);
else if (rr->AnnounceCount > 1)
{
rr->AnnounceCount = 1;
rr->LastAPTime = m->timenow - rr->ThisAPInterval;
}
if (m->CurrentRecord == rr) m->CurrentRecord = rr->next;
}
}
mDNSexport void mDNS_StartExit(mDNS *const m)
{
NetworkInterfaceInfo *intf;
AuthRecord *rr;
mDNS_Lock(m);
LogInfo("mDNS_StartExit");
m->ShutdownTime = NonZeroTime(m->timenow + mDNSPlatformOneSecond * 5);
mDNSCoreBeSleepProxyServer_internal(m, 0, 0, 0, 0);
#if APPLE_OSX_mDNSResponder
#if ! NO_WCF
CHECK_WCF_FUNCTION(WCFConnectionDealloc)
{
if (m->WCF) WCFConnectionDealloc((WCFConnection *)m->WCF);
}
#endif
#endif
#ifndef UNICAST_DISABLED
{
SearchListElem *s;
SuspendLLQs(m);
while (m->Hostnames) mDNS_RemoveDynDNSHostName(m, &m->Hostnames->fqdn);
for (s = SearchList; s; s = s->next)
while (s->AuthRecs)
{
ARListElem *dereg = s->AuthRecs;
s->AuthRecs = s->AuthRecs->next;
mDNS_Deregister_internal(m, &dereg->ar, mDNS_Dereg_normal); }
}
#endif
for (intf = m->HostInterfaces; intf; intf = intf->next)
if (intf->Advertise)
DeadvertiseInterface(m, intf);
while (m->NATTraversals)
{
NATTraversalInfo *t = m->NATTraversals;
mDNS_StopNATOperation_internal(m, t);
t->ExternalAddress = zerov4Addr;
t->ExternalPort = zeroIPPort;
t->RequestedPort = zeroIPPort;
t->Lifetime = 0;
t->Result = mStatus_NoError;
}
if (m->CurrentRecord)
LogMsg("mDNS_StartExit: ERROR m->CurrentRecord already set %s", ARDisplayString(m, m->CurrentRecord));
LogInfo("mDNS_StartExit: Deregistering duplicate resource records");
DeregLoop(m, m->DuplicateRecords);
LogInfo("mDNS_StartExit: Deregistering resource records");
DeregLoop(m, m->ResourceRecords);
if (m->NextScheduledResponse - m->timenow < mDNSPlatformOneSecond)
{
m->NextScheduledResponse = m->timenow;
m->SuppressSending = 0;
}
if (m->ResourceRecords) LogInfo("mDNS_StartExit: Sending final record deregistrations");
else LogInfo("mDNS_StartExit: No deregistering records remain");
for (rr = m->DuplicateRecords; rr; rr = rr->next)
LogMsg("mDNS_StartExit: Should not still have Duplicate Records remaining: %02X %s", rr->resrec.RecordType, ARDisplayString(m, rr));
if (m->mDNSPlatformStatus != mStatus_NoError) DiscardDeregistrations(m);
mDNS_Unlock(m);
LogInfo("mDNS_StartExit: done");
}
mDNSexport void mDNS_FinalExit(mDNS *const m)
{
mDNSu32 rrcache_active = 0;
mDNSu32 rrcache_totalused = 0;
mDNSu32 slot;
AuthRecord *rr;
LogInfo("mDNS_FinalExit: mDNSPlatformClose");
mDNSPlatformClose(m);
rrcache_totalused = m->rrcache_totalused;
for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
{
while (m->rrcache_hash[slot])
{
CacheGroup *cg = m->rrcache_hash[slot];
while (cg->members)
{
CacheRecord *cr = cg->members;
cg->members = cg->members->next;
if (cr->CRActiveQuestion) rrcache_active++;
ReleaseCacheRecord(m, cr);
}
cg->rrcache_tail = &cg->members;
ReleaseCacheGroup(m, &m->rrcache_hash[slot]);
}
}
debugf("mDNS_FinalExit: RR Cache was using %ld records, %lu active", rrcache_totalused, rrcache_active);
if (rrcache_active != m->rrcache_active)
LogMsg("*** ERROR *** rrcache_active %lu != m->rrcache_active %lu", rrcache_active, m->rrcache_active);
for (rr = m->ResourceRecords; rr; rr = rr->next)
LogMsg("mDNS_FinalExit failed to send goodbye for: %p %02X %s", rr, rr->resrec.RecordType, ARDisplayString(m, rr));
LogInfo("mDNS_FinalExit: done");
}