xref: /aosp_15_r20/external/mdnsresponder/mDNSCore/DNSCommon.c (revision 48a54d368dc4fa860885eef7b70b6c53499e7c25)
1*48a54d36SAndroid Build Coastguard Worker /* -*- Mode: C; tab-width: 4 -*-
2*48a54d36SAndroid Build Coastguard Worker  *
3*48a54d36SAndroid Build Coastguard Worker  * Copyright (c) 2002-2003 Apple Computer, Inc. All rights reserved.
4*48a54d36SAndroid Build Coastguard Worker  *
5*48a54d36SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
6*48a54d36SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
7*48a54d36SAndroid Build Coastguard Worker  * You may obtain a copy of the License at
8*48a54d36SAndroid Build Coastguard Worker  *
9*48a54d36SAndroid Build Coastguard Worker  *     http://www.apache.org/licenses/LICENSE-2.0
10*48a54d36SAndroid Build Coastguard Worker  *
11*48a54d36SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
12*48a54d36SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
13*48a54d36SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*48a54d36SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
15*48a54d36SAndroid Build Coastguard Worker  * limitations under the License.
16*48a54d36SAndroid Build Coastguard Worker  */
17*48a54d36SAndroid Build Coastguard Worker 
18*48a54d36SAndroid Build Coastguard Worker // Set mDNS_InstantiateInlines to tell mDNSEmbeddedAPI.h to instantiate inline functions, if necessary
19*48a54d36SAndroid Build Coastguard Worker #define mDNS_InstantiateInlines 1
20*48a54d36SAndroid Build Coastguard Worker #include "DNSCommon.h"
21*48a54d36SAndroid Build Coastguard Worker 
22*48a54d36SAndroid Build Coastguard Worker // Disable certain benign warnings with Microsoft compilers
23*48a54d36SAndroid Build Coastguard Worker #if (defined(_MSC_VER))
24*48a54d36SAndroid Build Coastguard Worker 	// Disable "conditional expression is constant" warning for debug macros.
25*48a54d36SAndroid Build Coastguard Worker 	// Otherwise, this generates warnings for the perfectly natural construct "while(1)"
26*48a54d36SAndroid Build Coastguard Worker 	// If someone knows a variant way of writing "while(1)" that doesn't generate warning messages, please let us know
27*48a54d36SAndroid Build Coastguard Worker 	#pragma warning(disable:4127)
28*48a54d36SAndroid Build Coastguard Worker 	// Disable "array is too small to include a terminating null character" warning
29*48a54d36SAndroid Build Coastguard Worker 	// -- domain labels have an initial length byte, not a terminating null character
30*48a54d36SAndroid Build Coastguard Worker 	#pragma warning(disable:4295)
31*48a54d36SAndroid Build Coastguard Worker #endif
32*48a54d36SAndroid Build Coastguard Worker 
33*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
34*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
35*48a54d36SAndroid Build Coastguard Worker #pragma mark - Program Constants
36*48a54d36SAndroid Build Coastguard Worker #endif
37*48a54d36SAndroid Build Coastguard Worker 
38*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSInterfaceID mDNSInterface_Any       = 0;
39*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSInterfaceID mDNSInterfaceMark       = (mDNSInterfaceID)-1;
40*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSInterfaceID mDNSInterface_LocalOnly = (mDNSInterfaceID)-2;
41*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSInterfaceID mDNSInterface_Unicast   = (mDNSInterfaceID)-3;
42*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSInterfaceID mDNSInterface_P2P       = (mDNSInterfaceID)-4;
43*48a54d36SAndroid Build Coastguard Worker 
44*48a54d36SAndroid Build Coastguard Worker // Note: Microsoft's proposed "Link Local Multicast Name Resolution Protocol" (LLMNR) is essentially a limited version of
45*48a54d36SAndroid Build Coastguard Worker // Multicast DNS, using the same packet formats, naming syntax, and record types as Multicast DNS, but on a different UDP
46*48a54d36SAndroid Build Coastguard Worker // port and multicast address, which means it won't interoperate with the existing installed base of Multicast DNS responders.
47*48a54d36SAndroid Build Coastguard Worker // LLMNR uses IPv4 multicast address 224.0.0.252, IPv6 multicast address FF02::0001:0003, and UDP port 5355.
48*48a54d36SAndroid Build Coastguard Worker // Uncomment the appropriate lines below to build a special Multicast DNS responder for testing interoperability
49*48a54d36SAndroid Build Coastguard Worker // with Microsoft's LLMNR client code.
50*48a54d36SAndroid Build Coastguard Worker 
51*48a54d36SAndroid Build Coastguard Worker #define   DiscardPortAsNumber               9
52*48a54d36SAndroid Build Coastguard Worker #define   SSHPortAsNumber                  22
53*48a54d36SAndroid Build Coastguard Worker #define   UnicastDNSPortAsNumber           53
54*48a54d36SAndroid Build Coastguard Worker #define   SSDPPortAsNumber               1900
55*48a54d36SAndroid Build Coastguard Worker #define   IPSECPortAsNumber              4500
56*48a54d36SAndroid Build Coastguard Worker #define   NSIPCPortAsNumber              5030		// Port used for dnsextd to talk to local nameserver bound to loopback
57*48a54d36SAndroid Build Coastguard Worker #define   NATPMPAnnouncementPortAsNumber 5350
58*48a54d36SAndroid Build Coastguard Worker #define   NATPMPPortAsNumber             5351
59*48a54d36SAndroid Build Coastguard Worker #define   DNSEXTPortAsNumber             5352		// Port used for end-to-end DNS operations like LLQ, Updates with Leases, etc.
60*48a54d36SAndroid Build Coastguard Worker #define   MulticastDNSPortAsNumber       5353
61*48a54d36SAndroid Build Coastguard Worker #define   LoopbackIPCPortAsNumber        5354
62*48a54d36SAndroid Build Coastguard Worker //#define MulticastDNSPortAsNumber       5355		// LLMNR
63*48a54d36SAndroid Build Coastguard Worker #define   PrivateDNSPortAsNumber         5533
64*48a54d36SAndroid Build Coastguard Worker 
65*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort DiscardPort            = { { DiscardPortAsNumber            >> 8, DiscardPortAsNumber            & 0xFF } };
66*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort SSHPort                = { { SSHPortAsNumber                >> 8, SSHPortAsNumber                & 0xFF } };
67*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort UnicastDNSPort         = { { UnicastDNSPortAsNumber         >> 8, UnicastDNSPortAsNumber         & 0xFF } };
68*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort SSDPPort               = { { SSDPPortAsNumber               >> 8, SSDPPortAsNumber               & 0xFF } };
69*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort IPSECPort              = { { IPSECPortAsNumber              >> 8, IPSECPortAsNumber              & 0xFF } };
70*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort NSIPCPort              = { { NSIPCPortAsNumber              >> 8, NSIPCPortAsNumber              & 0xFF } };
71*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort NATPMPAnnouncementPort = { { NATPMPAnnouncementPortAsNumber >> 8, NATPMPAnnouncementPortAsNumber & 0xFF } };
72*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort NATPMPPort             = { { NATPMPPortAsNumber             >> 8, NATPMPPortAsNumber             & 0xFF } };
73*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort DNSEXTPort             = { { DNSEXTPortAsNumber             >> 8, DNSEXTPortAsNumber             & 0xFF } };
74*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort MulticastDNSPort       = { { MulticastDNSPortAsNumber       >> 8, MulticastDNSPortAsNumber       & 0xFF } };
75*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort LoopbackIPCPort        = { { LoopbackIPCPortAsNumber        >> 8, LoopbackIPCPortAsNumber        & 0xFF } };
76*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort PrivateDNSPort         = { { PrivateDNSPortAsNumber         >> 8, PrivateDNSPortAsNumber         & 0xFF } };
77*48a54d36SAndroid Build Coastguard Worker 
78*48a54d36SAndroid Build Coastguard Worker mDNSexport const OwnerOptData    zeroOwner         = { 0, 0, { { 0 } }, { { 0 } }, { { 0 } } };
79*48a54d36SAndroid Build Coastguard Worker 
80*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSIPPort      zeroIPPort        = { { 0 } };
81*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv4Addr      zerov4Addr        = { { 0 } };
82*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv6Addr      zerov6Addr        = { { 0 } };
83*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSEthAddr     zeroEthAddr       = { { 0 } };
84*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv4Addr      onesIPv4Addr      = { { 255, 255, 255, 255 } };
85*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv6Addr      onesIPv6Addr      = { { 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255 } };
86*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSEthAddr     onesEthAddr       = { { 255, 255, 255, 255, 255, 255 } };
87*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSAddr        zeroAddr          = { mDNSAddrType_None, {{{ 0 }}} };
88*48a54d36SAndroid Build Coastguard Worker 
89*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv4Addr  AllDNSAdminGroup   = { { 239, 255, 255, 251 } };
90*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv4Addr  AllHosts_v4        = { { 224,   0,   0,   1 } }; // For NAT-PMP Annoucements
91*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv6Addr  AllHosts_v6        = { { 0xFF,0x02,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x01 } };
92*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSv6Addr  NDP_prefix         = { { 0xFF,0x02,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x01, 0xFF,0x00,0x00,0xFB } }; // FF02:0:0:0:0:1:FF00::/104
93*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSEthAddr AllHosts_v6_Eth    = { { 0x33, 0x33, 0x00, 0x00, 0x00, 0x01 } };
94*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSAddr    AllDNSLinkGroup_v4 = { mDNSAddrType_IPv4, { { { 224,   0,   0, 251 } } } };
95*48a54d36SAndroid Build Coastguard Worker //mDNSexport const mDNSAddr  AllDNSLinkGroup_v4 = { mDNSAddrType_IPv4, { { { 224,   0,   0, 252 } } } }; // LLMNR
96*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSAddr    AllDNSLinkGroup_v6 = { mDNSAddrType_IPv6, { { { 0xFF,0x02,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0xFB } } } };
97*48a54d36SAndroid Build Coastguard Worker //mDNSexport const mDNSAddr  AllDNSLinkGroup_v6 = { mDNSAddrType_IPv6, { { { 0xFF,0x02,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00, 0x00,0x01,0x00,0x03 } } } }; // LLMNR
98*48a54d36SAndroid Build Coastguard Worker 
99*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 zeroID          = { { 0, 0 } };
100*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 onesID          = { { 255, 255 } };
101*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 QueryFlags      = { { kDNSFlag0_QR_Query    | kDNSFlag0_OP_StdQuery,                0 } };
102*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 uQueryFlags     = { { kDNSFlag0_QR_Query    | kDNSFlag0_OP_StdQuery | kDNSFlag0_RD, 0 } };
103*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 ResponseFlags   = { { kDNSFlag0_QR_Response | kDNSFlag0_OP_StdQuery | kDNSFlag0_AA, 0 } };
104*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 UpdateReqFlags  = { { kDNSFlag0_QR_Query    | kDNSFlag0_OP_Update,                  0 } };
105*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque16 UpdateRespFlags = { { kDNSFlag0_QR_Response | kDNSFlag0_OP_Update,                  0 } };
106*48a54d36SAndroid Build Coastguard Worker 
107*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSOpaque64 zeroOpaque64    = { { 0 } };
108*48a54d36SAndroid Build Coastguard Worker 
109*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
110*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
111*48a54d36SAndroid Build Coastguard Worker #pragma mark -
112*48a54d36SAndroid Build Coastguard Worker #pragma mark - General Utility Functions
113*48a54d36SAndroid Build Coastguard Worker #endif
114*48a54d36SAndroid Build Coastguard Worker 
115*48a54d36SAndroid Build Coastguard Worker // return true for RFC1918 private addresses
mDNSv4AddrIsRFC1918(mDNSv4Addr * addr)116*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool mDNSv4AddrIsRFC1918(mDNSv4Addr *addr)
117*48a54d36SAndroid Build Coastguard Worker 	{
118*48a54d36SAndroid Build Coastguard Worker 	return ((addr->b[0] == 10) ||                                 // 10/8 prefix
119*48a54d36SAndroid Build Coastguard Worker 			(addr->b[0] == 172 && (addr->b[1] & 0xF0) == 16) ||   // 172.16/12
120*48a54d36SAndroid Build Coastguard Worker 			(addr->b[0] == 192 && addr->b[1] == 168));            // 192.168/16
121*48a54d36SAndroid Build Coastguard Worker 	}
122*48a54d36SAndroid Build Coastguard Worker 
GetFirstActiveInterface(NetworkInterfaceInfo * intf)123*48a54d36SAndroid Build Coastguard Worker mDNSexport NetworkInterfaceInfo *GetFirstActiveInterface(NetworkInterfaceInfo *intf)
124*48a54d36SAndroid Build Coastguard Worker 	{
125*48a54d36SAndroid Build Coastguard Worker 	while (intf && !intf->InterfaceActive) intf = intf->next;
126*48a54d36SAndroid Build Coastguard Worker 	return(intf);
127*48a54d36SAndroid Build Coastguard Worker 	}
128*48a54d36SAndroid Build Coastguard Worker 
GetNextActiveInterfaceID(const NetworkInterfaceInfo * intf)129*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSInterfaceID GetNextActiveInterfaceID(const NetworkInterfaceInfo *intf)
130*48a54d36SAndroid Build Coastguard Worker 	{
131*48a54d36SAndroid Build Coastguard Worker 	const NetworkInterfaceInfo *next = GetFirstActiveInterface(intf->next);
132*48a54d36SAndroid Build Coastguard Worker 	if (next) return(next->InterfaceID); else return(mDNSNULL);
133*48a54d36SAndroid Build Coastguard Worker 	}
134*48a54d36SAndroid Build Coastguard Worker 
NumCacheRecordsForInterfaceID(const mDNS * const m,mDNSInterfaceID id)135*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 NumCacheRecordsForInterfaceID(const mDNS *const m, mDNSInterfaceID id)
136*48a54d36SAndroid Build Coastguard Worker 	{
137*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 slot, used = 0;
138*48a54d36SAndroid Build Coastguard Worker 	CacheGroup *cg;
139*48a54d36SAndroid Build Coastguard Worker 	const CacheRecord *rr;
140*48a54d36SAndroid Build Coastguard Worker 	FORALL_CACHERECORDS(slot, cg, rr)
141*48a54d36SAndroid Build Coastguard Worker 		if (rr->resrec.InterfaceID == id) used++;
142*48a54d36SAndroid Build Coastguard Worker 	return(used);
143*48a54d36SAndroid Build Coastguard Worker 	}
144*48a54d36SAndroid Build Coastguard Worker 
DNSTypeName(mDNSu16 rrtype)145*48a54d36SAndroid Build Coastguard Worker mDNSexport char *DNSTypeName(mDNSu16 rrtype)
146*48a54d36SAndroid Build Coastguard Worker 	{
147*48a54d36SAndroid Build Coastguard Worker 	switch (rrtype)
148*48a54d36SAndroid Build Coastguard Worker 		{
149*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:    return("Addr");
150*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:   return("NS");
151*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:return("CNAME");
152*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:  return("SOA");
153*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NULL: return("NULL");
154*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:  return("PTR");
155*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:return("HINFO");
156*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:  return("TXT");
157*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA: return("AAAA");
158*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:  return("SRV");
159*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:  return("OPT");
160*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: return("NSEC");
161*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TSIG: return("TSIG");
162*48a54d36SAndroid Build Coastguard Worker 		case kDNSQType_ANY: return("ANY");
163*48a54d36SAndroid Build Coastguard Worker 		default:			{
164*48a54d36SAndroid Build Coastguard Worker 							static char buffer[16];
165*48a54d36SAndroid Build Coastguard Worker 							mDNS_snprintf(buffer, sizeof(buffer), "(%d)", rrtype);
166*48a54d36SAndroid Build Coastguard Worker 							return(buffer);
167*48a54d36SAndroid Build Coastguard Worker 							}
168*48a54d36SAndroid Build Coastguard Worker 		}
169*48a54d36SAndroid Build Coastguard Worker 	}
170*48a54d36SAndroid Build Coastguard Worker 
171*48a54d36SAndroid Build Coastguard Worker // Note slight bug: this code uses the rdlength from the ResourceRecord object, to display
172*48a54d36SAndroid Build Coastguard Worker // the rdata from the RDataBody object. Sometimes this could be the wrong length -- but as
173*48a54d36SAndroid Build Coastguard Worker // long as this routine is only used for debugging messages, it probably isn't a big problem.
GetRRDisplayString_rdb(const ResourceRecord * const rr,const RDataBody * const rd1,char * const buffer)174*48a54d36SAndroid Build Coastguard Worker mDNSexport char *GetRRDisplayString_rdb(const ResourceRecord *const rr, const RDataBody *const rd1, char *const buffer)
175*48a54d36SAndroid Build Coastguard Worker 	{
176*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const rd = (RDataBody2 *)rd1;
177*48a54d36SAndroid Build Coastguard Worker 	#define RemSpc (MaxMsg-1-length)
178*48a54d36SAndroid Build Coastguard Worker 	char *ptr = buffer;
179*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 length = mDNS_snprintf(buffer, MaxMsg-1, "%4d %##s %s ", rr->rdlength, rr->name->c, DNSTypeName(rr->rrtype));
180*48a54d36SAndroid Build Coastguard Worker 	if (rr->RecordType == kDNSRecordTypePacketNegative) return(buffer);
181*48a54d36SAndroid Build Coastguard Worker 	if (!rr->rdlength) { mDNS_snprintf(buffer+length, RemSpc, "<< ZERO RDATA LENGTH >>"); return(buffer); }
182*48a54d36SAndroid Build Coastguard Worker 
183*48a54d36SAndroid Build Coastguard Worker 	switch (rr->rrtype)
184*48a54d36SAndroid Build Coastguard Worker 		{
185*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:	mDNS_snprintf(buffer+length, RemSpc, "%.4a", &rd->ipv4);          break;
186*48a54d36SAndroid Build Coastguard Worker 
187*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:	// Same as PTR
188*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:// Same as PTR
189*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:	mDNS_snprintf(buffer+length, RemSpc, "%##s", rd->name.c);       break;
190*48a54d36SAndroid Build Coastguard Worker 
191*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:  mDNS_snprintf(buffer+length, RemSpc, "%##s %##s %d %d %d %d %d",
192*48a54d36SAndroid Build Coastguard Worker 								rd->soa.mname.c, rd->soa.rname.c,
193*48a54d36SAndroid Build Coastguard Worker 								rd->soa.serial, rd->soa.refresh, rd->soa.retry, rd->soa.expire, rd->soa.min);
194*48a54d36SAndroid Build Coastguard Worker 							break;
195*48a54d36SAndroid Build Coastguard Worker 
196*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:// Display this the same as TXT (show all constituent strings)
197*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:  {
198*48a54d36SAndroid Build Coastguard Worker 							const mDNSu8 *t = rd->txt.c;
199*48a54d36SAndroid Build Coastguard Worker 							while (t < rd->txt.c + rr->rdlength)
200*48a54d36SAndroid Build Coastguard Worker 								{
201*48a54d36SAndroid Build Coastguard Worker 								length += mDNS_snprintf(buffer+length, RemSpc, "%s%#s", t > rd->txt.c ? "¦" : "", t);
202*48a54d36SAndroid Build Coastguard Worker 								t += 1 + t[0];
203*48a54d36SAndroid Build Coastguard Worker 								}
204*48a54d36SAndroid Build Coastguard Worker 							} break;
205*48a54d36SAndroid Build Coastguard Worker 
206*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA:	mDNS_snprintf(buffer+length, RemSpc, "%.16a", &rd->ipv6);       break;
207*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	mDNS_snprintf(buffer+length, RemSpc, "%u %u %u %##s",
208*48a54d36SAndroid Build Coastguard Worker 								rd->srv.priority, rd->srv.weight, mDNSVal16(rd->srv.port), rd->srv.target.c); break;
209*48a54d36SAndroid Build Coastguard Worker 
210*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:  {
211*48a54d36SAndroid Build Coastguard Worker 							const rdataOPT *opt;
212*48a54d36SAndroid Build Coastguard Worker 							const rdataOPT *const end = (const rdataOPT *)&rd->data[rr->rdlength];
213*48a54d36SAndroid Build Coastguard Worker 							length += mDNS_snprintf(buffer+length, RemSpc, "Max %d", rr->rrclass);
214*48a54d36SAndroid Build Coastguard Worker 							for (opt = &rd->opt[0]; opt < end; opt++)
215*48a54d36SAndroid Build Coastguard Worker 								{
216*48a54d36SAndroid Build Coastguard Worker 								switch(opt->opt)
217*48a54d36SAndroid Build Coastguard Worker 									{
218*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_LLQ:
219*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Vers %d",     opt->u.llq.vers);
220*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Op %d",       opt->u.llq.llqOp);
221*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Err/Port %d", opt->u.llq.err);
222*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " ID %08X%08X", opt->u.llq.id.l[0], opt->u.llq.id.l[1]);
223*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Lease %d",    opt->u.llq.llqlease);
224*48a54d36SAndroid Build Coastguard Worker 										break;
225*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Lease:
226*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Lease %d",    opt->u.updatelease);
227*48a54d36SAndroid Build Coastguard Worker 										break;
228*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Owner:
229*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Vers %d",     opt->u.owner.vers);
230*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Seq %3d", (mDNSu8)opt->u.owner.seq);	// Display as unsigned
231*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " MAC %.6a",    opt->u.owner.HMAC.b);
232*48a54d36SAndroid Build Coastguard Worker 										if (opt->optlen >= DNSOpt_OwnerData_ID_Wake_Space-4)
233*48a54d36SAndroid Build Coastguard Worker 											{
234*48a54d36SAndroid Build Coastguard Worker 											length += mDNS_snprintf(buffer+length, RemSpc, " I-MAC %.6a", opt->u.owner.IMAC.b);
235*48a54d36SAndroid Build Coastguard Worker 											if (opt->optlen > DNSOpt_OwnerData_ID_Wake_Space-4)
236*48a54d36SAndroid Build Coastguard Worker 												length += mDNS_snprintf(buffer+length, RemSpc, " Password %.6a", opt->u.owner.password.b);
237*48a54d36SAndroid Build Coastguard Worker 											}
238*48a54d36SAndroid Build Coastguard Worker 										break;
239*48a54d36SAndroid Build Coastguard Worker 									default:
240*48a54d36SAndroid Build Coastguard Worker 										length += mDNS_snprintf(buffer+length, RemSpc, " Unknown %d",  opt->opt);
241*48a54d36SAndroid Build Coastguard Worker 										break;
242*48a54d36SAndroid Build Coastguard Worker 									}
243*48a54d36SAndroid Build Coastguard Worker 								}
244*48a54d36SAndroid Build Coastguard Worker 							}
245*48a54d36SAndroid Build Coastguard Worker 							break;
246*48a54d36SAndroid Build Coastguard Worker 
247*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: {
248*48a54d36SAndroid Build Coastguard Worker 							mDNSu16 i;
249*48a54d36SAndroid Build Coastguard Worker 							for (i=0; i<255; i++)
250*48a54d36SAndroid Build Coastguard Worker 								if (rd->nsec.bitmap[i>>3] & (128 >> (i&7)))
251*48a54d36SAndroid Build Coastguard Worker 									length += mDNS_snprintf(buffer+length, RemSpc, "%s ", DNSTypeName(i));
252*48a54d36SAndroid Build Coastguard Worker 							}
253*48a54d36SAndroid Build Coastguard Worker 							break;
254*48a54d36SAndroid Build Coastguard Worker 
255*48a54d36SAndroid Build Coastguard Worker 		default:			mDNS_snprintf(buffer+length, RemSpc, "RDLen %d: %s", rr->rdlength, rd->data);
256*48a54d36SAndroid Build Coastguard Worker 							// Really should scan buffer to check if text is valid UTF-8 and only replace with dots if not
257*48a54d36SAndroid Build Coastguard Worker 							for (ptr = buffer; *ptr; ptr++) if (*ptr < ' ') *ptr = '.';
258*48a54d36SAndroid Build Coastguard Worker 							break;
259*48a54d36SAndroid Build Coastguard Worker 		}
260*48a54d36SAndroid Build Coastguard Worker 	return(buffer);
261*48a54d36SAndroid Build Coastguard Worker 	}
262*48a54d36SAndroid Build Coastguard Worker 
263*48a54d36SAndroid Build Coastguard Worker // See comments in mDNSEmbeddedAPI.h
264*48a54d36SAndroid Build Coastguard Worker #if _PLATFORM_HAS_STRONG_PRNG_
265*48a54d36SAndroid Build Coastguard Worker #define mDNSRandomNumber mDNSPlatformRandomNumber
266*48a54d36SAndroid Build Coastguard Worker #else
mDNSRandomFromSeed(mDNSu32 seed)267*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu32 mDNSRandomFromSeed(mDNSu32 seed)
268*48a54d36SAndroid Build Coastguard Worker 	{
269*48a54d36SAndroid Build Coastguard Worker 	return seed * 21 + 1;
270*48a54d36SAndroid Build Coastguard Worker 	}
271*48a54d36SAndroid Build Coastguard Worker 
mDNSMixRandomSeed(mDNSu32 seed,mDNSu8 iteration)272*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu32 mDNSMixRandomSeed(mDNSu32 seed, mDNSu8 iteration)
273*48a54d36SAndroid Build Coastguard Worker 	{
274*48a54d36SAndroid Build Coastguard Worker 	return iteration ? mDNSMixRandomSeed(mDNSRandomFromSeed(seed), --iteration) : seed;
275*48a54d36SAndroid Build Coastguard Worker 	}
276*48a54d36SAndroid Build Coastguard Worker 
mDNSRandomNumber()277*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu32 mDNSRandomNumber()
278*48a54d36SAndroid Build Coastguard Worker 	{
279*48a54d36SAndroid Build Coastguard Worker 	static mDNSBool seeded = mDNSfalse;
280*48a54d36SAndroid Build Coastguard Worker 	static mDNSu32 seed = 0;
281*48a54d36SAndroid Build Coastguard Worker 	if (!seeded)
282*48a54d36SAndroid Build Coastguard Worker 		{
283*48a54d36SAndroid Build Coastguard Worker 		seed = mDNSMixRandomSeed(mDNSPlatformRandomSeed(), 100);
284*48a54d36SAndroid Build Coastguard Worker 		seeded = mDNStrue;
285*48a54d36SAndroid Build Coastguard Worker 		}
286*48a54d36SAndroid Build Coastguard Worker 	return (seed = mDNSRandomFromSeed(seed));
287*48a54d36SAndroid Build Coastguard Worker 	}
288*48a54d36SAndroid Build Coastguard Worker #endif // ! _PLATFORM_HAS_STRONG_PRNG_
289*48a54d36SAndroid Build Coastguard Worker 
mDNSRandom(mDNSu32 max)290*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 mDNSRandom(mDNSu32 max)		// Returns pseudo-random result from zero to max inclusive
291*48a54d36SAndroid Build Coastguard Worker 	{
292*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 ret = 0;
293*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 mask = 1;
294*48a54d36SAndroid Build Coastguard Worker 
295*48a54d36SAndroid Build Coastguard Worker 	while (mask < max) mask = (mask << 1) | 1;
296*48a54d36SAndroid Build Coastguard Worker 
297*48a54d36SAndroid Build Coastguard Worker 	do ret = mDNSRandomNumber() & mask;
298*48a54d36SAndroid Build Coastguard Worker 	while (ret > max);
299*48a54d36SAndroid Build Coastguard Worker 
300*48a54d36SAndroid Build Coastguard Worker 	return ret;
301*48a54d36SAndroid Build Coastguard Worker 	}
302*48a54d36SAndroid Build Coastguard Worker 
mDNSSameAddress(const mDNSAddr * ip1,const mDNSAddr * ip2)303*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool mDNSSameAddress(const mDNSAddr *ip1, const mDNSAddr *ip2)
304*48a54d36SAndroid Build Coastguard Worker 	{
305*48a54d36SAndroid Build Coastguard Worker 	if (ip1->type == ip2->type)
306*48a54d36SAndroid Build Coastguard Worker 		{
307*48a54d36SAndroid Build Coastguard Worker 		switch (ip1->type)
308*48a54d36SAndroid Build Coastguard Worker 			{
309*48a54d36SAndroid Build Coastguard Worker 			case mDNSAddrType_None : return(mDNStrue); // Empty addresses have no data and are therefore always equal
310*48a54d36SAndroid Build Coastguard Worker 			case mDNSAddrType_IPv4 : return(mDNSBool)(mDNSSameIPv4Address(ip1->ip.v4, ip2->ip.v4));
311*48a54d36SAndroid Build Coastguard Worker 			case mDNSAddrType_IPv6 : return(mDNSBool)(mDNSSameIPv6Address(ip1->ip.v6, ip2->ip.v6));
312*48a54d36SAndroid Build Coastguard Worker 			}
313*48a54d36SAndroid Build Coastguard Worker 		}
314*48a54d36SAndroid Build Coastguard Worker 	return(mDNSfalse);
315*48a54d36SAndroid Build Coastguard Worker 	}
316*48a54d36SAndroid Build Coastguard Worker 
mDNSAddrIsDNSMulticast(const mDNSAddr * ip)317*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool mDNSAddrIsDNSMulticast(const mDNSAddr *ip)
318*48a54d36SAndroid Build Coastguard Worker 	{
319*48a54d36SAndroid Build Coastguard Worker 	switch(ip->type)
320*48a54d36SAndroid Build Coastguard Worker 		{
321*48a54d36SAndroid Build Coastguard Worker 		case mDNSAddrType_IPv4: return(mDNSBool)(mDNSSameIPv4Address(ip->ip.v4, AllDNSLinkGroup_v4.ip.v4));
322*48a54d36SAndroid Build Coastguard Worker 		case mDNSAddrType_IPv6: return(mDNSBool)(mDNSSameIPv6Address(ip->ip.v6, AllDNSLinkGroup_v6.ip.v6));
323*48a54d36SAndroid Build Coastguard Worker 		default: return(mDNSfalse);
324*48a54d36SAndroid Build Coastguard Worker 		}
325*48a54d36SAndroid Build Coastguard Worker 	}
326*48a54d36SAndroid Build Coastguard Worker 
327*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
328*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
329*48a54d36SAndroid Build Coastguard Worker #pragma mark -
330*48a54d36SAndroid Build Coastguard Worker #pragma mark - Domain Name Utility Functions
331*48a54d36SAndroid Build Coastguard Worker #endif
332*48a54d36SAndroid Build Coastguard Worker 
SameDomainLabel(const mDNSu8 * a,const mDNSu8 * b)333*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool SameDomainLabel(const mDNSu8 *a, const mDNSu8 *b)
334*48a54d36SAndroid Build Coastguard Worker 	{
335*48a54d36SAndroid Build Coastguard Worker 	int i;
336*48a54d36SAndroid Build Coastguard Worker 	const int len = *a++;
337*48a54d36SAndroid Build Coastguard Worker 
338*48a54d36SAndroid Build Coastguard Worker 	if (len > MAX_DOMAIN_LABEL)
339*48a54d36SAndroid Build Coastguard Worker 		{ debugf("Malformed label (too long)"); return(mDNSfalse); }
340*48a54d36SAndroid Build Coastguard Worker 
341*48a54d36SAndroid Build Coastguard Worker 	if (len != *b++) return(mDNSfalse);
342*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<len; i++)
343*48a54d36SAndroid Build Coastguard Worker 		{
344*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 ac = *a++;
345*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 bc = *b++;
346*48a54d36SAndroid Build Coastguard Worker 		if (mDNSIsUpperCase(ac)) ac += 'a' - 'A';
347*48a54d36SAndroid Build Coastguard Worker 		if (mDNSIsUpperCase(bc)) bc += 'a' - 'A';
348*48a54d36SAndroid Build Coastguard Worker 		if (ac != bc) return(mDNSfalse);
349*48a54d36SAndroid Build Coastguard Worker 		}
350*48a54d36SAndroid Build Coastguard Worker 	return(mDNStrue);
351*48a54d36SAndroid Build Coastguard Worker 	}
352*48a54d36SAndroid Build Coastguard Worker 
SameDomainName(const domainname * const d1,const domainname * const d2)353*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool SameDomainName(const domainname *const d1, const domainname *const d2)
354*48a54d36SAndroid Build Coastguard Worker 	{
355*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      a   = d1->c;
356*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      b   = d2->c;
357*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const max = d1->c + MAX_DOMAIN_NAME;			// Maximum that's valid
358*48a54d36SAndroid Build Coastguard Worker 
359*48a54d36SAndroid Build Coastguard Worker 	while (*a || *b)
360*48a54d36SAndroid Build Coastguard Worker 		{
361*48a54d36SAndroid Build Coastguard Worker 		if (a + 1 + *a >= max)
362*48a54d36SAndroid Build Coastguard Worker 			{ debugf("Malformed domain name (more than 256 characters)"); return(mDNSfalse); }
363*48a54d36SAndroid Build Coastguard Worker 		if (!SameDomainLabel(a, b)) return(mDNSfalse);
364*48a54d36SAndroid Build Coastguard Worker 		a += 1 + *a;
365*48a54d36SAndroid Build Coastguard Worker 		b += 1 + *b;
366*48a54d36SAndroid Build Coastguard Worker 		}
367*48a54d36SAndroid Build Coastguard Worker 
368*48a54d36SAndroid Build Coastguard Worker 	return(mDNStrue);
369*48a54d36SAndroid Build Coastguard Worker 	}
370*48a54d36SAndroid Build Coastguard Worker 
SameDomainNameCS(const domainname * const d1,const domainname * const d2)371*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool SameDomainNameCS(const domainname *const d1, const domainname *const d2)
372*48a54d36SAndroid Build Coastguard Worker 	{
373*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 l1 = DomainNameLength(d1);
374*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 l2 = DomainNameLength(d2);
375*48a54d36SAndroid Build Coastguard Worker 	return(l1 <= MAX_DOMAIN_NAME && l1 == l2 && mDNSPlatformMemSame(d1, d2, l1));
376*48a54d36SAndroid Build Coastguard Worker 	}
377*48a54d36SAndroid Build Coastguard Worker 
IsLocalDomain(const domainname * d)378*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool IsLocalDomain(const domainname *d)
379*48a54d36SAndroid Build Coastguard Worker 	{
380*48a54d36SAndroid Build Coastguard Worker 	// Domains that are defined to be resolved via link-local multicast are:
381*48a54d36SAndroid Build Coastguard Worker 	// local., 254.169.in-addr.arpa., and {8,9,A,B}.E.F.ip6.arpa.
382*48a54d36SAndroid Build Coastguard Worker 	static const domainname *nL = (const domainname*)"\x5" "local";
383*48a54d36SAndroid Build Coastguard Worker 	static const domainname *nR = (const domainname*)"\x3" "254" "\x3" "169"         "\x7" "in-addr" "\x4" "arpa";
384*48a54d36SAndroid Build Coastguard Worker 	static const domainname *n8 = (const domainname*)"\x1" "8"   "\x1" "e" "\x1" "f" "\x3" "ip6"     "\x4" "arpa";
385*48a54d36SAndroid Build Coastguard Worker 	static const domainname *n9 = (const domainname*)"\x1" "9"   "\x1" "e" "\x1" "f" "\x3" "ip6"     "\x4" "arpa";
386*48a54d36SAndroid Build Coastguard Worker 	static const domainname *nA = (const domainname*)"\x1" "a"   "\x1" "e" "\x1" "f" "\x3" "ip6"     "\x4" "arpa";
387*48a54d36SAndroid Build Coastguard Worker 	static const domainname *nB = (const domainname*)"\x1" "b"   "\x1" "e" "\x1" "f" "\x3" "ip6"     "\x4" "arpa";
388*48a54d36SAndroid Build Coastguard Worker 
389*48a54d36SAndroid Build Coastguard Worker 	const domainname *d1, *d2, *d3, *d4, *d5;	// Top-level domain, second-level domain, etc.
390*48a54d36SAndroid Build Coastguard Worker 	d1 = d2 = d3 = d4 = d5 = mDNSNULL;
391*48a54d36SAndroid Build Coastguard Worker 	while (d->c[0])
392*48a54d36SAndroid Build Coastguard Worker 		{
393*48a54d36SAndroid Build Coastguard Worker 		d5 = d4; d4 = d3; d3 = d2; d2 = d1; d1 = d;
394*48a54d36SAndroid Build Coastguard Worker 		d = (const domainname*)(d->c + 1 + d->c[0]);
395*48a54d36SAndroid Build Coastguard Worker 		}
396*48a54d36SAndroid Build Coastguard Worker 
397*48a54d36SAndroid Build Coastguard Worker 	if (d1 && SameDomainName(d1, nL)) return(mDNStrue);
398*48a54d36SAndroid Build Coastguard Worker 	if (d4 && SameDomainName(d4, nR)) return(mDNStrue);
399*48a54d36SAndroid Build Coastguard Worker 	if (d5 && SameDomainName(d5, n8)) return(mDNStrue);
400*48a54d36SAndroid Build Coastguard Worker 	if (d5 && SameDomainName(d5, n9)) return(mDNStrue);
401*48a54d36SAndroid Build Coastguard Worker 	if (d5 && SameDomainName(d5, nA)) return(mDNStrue);
402*48a54d36SAndroid Build Coastguard Worker 	if (d5 && SameDomainName(d5, nB)) return(mDNStrue);
403*48a54d36SAndroid Build Coastguard Worker 	return(mDNSfalse);
404*48a54d36SAndroid Build Coastguard Worker 	}
405*48a54d36SAndroid Build Coastguard Worker 
LastLabel(const domainname * d)406*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *LastLabel(const domainname *d)
407*48a54d36SAndroid Build Coastguard Worker 	{
408*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *p = d->c;
409*48a54d36SAndroid Build Coastguard Worker 	while (d->c[0])
410*48a54d36SAndroid Build Coastguard Worker 		{
411*48a54d36SAndroid Build Coastguard Worker 		p = d->c;
412*48a54d36SAndroid Build Coastguard Worker 		d = (const domainname*)(d->c + 1 + d->c[0]);
413*48a54d36SAndroid Build Coastguard Worker 		}
414*48a54d36SAndroid Build Coastguard Worker 	return(p);
415*48a54d36SAndroid Build Coastguard Worker 	}
416*48a54d36SAndroid Build Coastguard Worker 
417*48a54d36SAndroid Build Coastguard Worker // Returns length of a domain name INCLUDING the byte for the final null label
418*48a54d36SAndroid Build Coastguard Worker // e.g. for the root label "." it returns one
419*48a54d36SAndroid Build Coastguard Worker // For the FQDN "com." it returns 5 (length byte, three data bytes, final zero)
420*48a54d36SAndroid Build Coastguard Worker // Legal results are 1 (just root label) to 256 (MAX_DOMAIN_NAME)
421*48a54d36SAndroid Build Coastguard Worker // If the given domainname is invalid, result is 257 (MAX_DOMAIN_NAME+1)
DomainNameLengthLimit(const domainname * const name,const mDNSu8 * limit)422*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu16 DomainNameLengthLimit(const domainname *const name, const mDNSu8 *limit)
423*48a54d36SAndroid Build Coastguard Worker 	{
424*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *src = name->c;
425*48a54d36SAndroid Build Coastguard Worker 	while (src < limit && *src <= MAX_DOMAIN_LABEL)
426*48a54d36SAndroid Build Coastguard Worker 		{
427*48a54d36SAndroid Build Coastguard Worker 		if (*src == 0) return((mDNSu16)(src - name->c + 1));
428*48a54d36SAndroid Build Coastguard Worker 		src += 1 + *src;
429*48a54d36SAndroid Build Coastguard Worker 		}
430*48a54d36SAndroid Build Coastguard Worker 	return(MAX_DOMAIN_NAME+1);
431*48a54d36SAndroid Build Coastguard Worker 	}
432*48a54d36SAndroid Build Coastguard Worker 
433*48a54d36SAndroid Build Coastguard Worker // CompressedDomainNameLength returns the length of a domain name INCLUDING the byte
434*48a54d36SAndroid Build Coastguard Worker // for the final null label, e.g. for the root label "." it returns one.
435*48a54d36SAndroid Build Coastguard Worker // E.g. for the FQDN "foo.com." it returns 9
436*48a54d36SAndroid Build Coastguard Worker // (length, three data bytes, length, three more data bytes, final zero).
437*48a54d36SAndroid Build Coastguard Worker // In the case where a parent domain name is provided, and the given name is a child
438*48a54d36SAndroid Build Coastguard Worker // of that parent, CompressedDomainNameLength returns the length of the prefix portion
439*48a54d36SAndroid Build Coastguard Worker // of the child name, plus TWO bytes for the compression pointer.
440*48a54d36SAndroid Build Coastguard Worker // E.g. for the name "foo.com." with parent "com.", it returns 6
441*48a54d36SAndroid Build Coastguard Worker // (length, three data bytes, two-byte compression pointer).
CompressedDomainNameLength(const domainname * const name,const domainname * parent)442*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu16 CompressedDomainNameLength(const domainname *const name, const domainname *parent)
443*48a54d36SAndroid Build Coastguard Worker 	{
444*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *src = name->c;
445*48a54d36SAndroid Build Coastguard Worker 	if (parent && parent->c[0] == 0) parent = mDNSNULL;
446*48a54d36SAndroid Build Coastguard Worker 	while (*src)
447*48a54d36SAndroid Build Coastguard Worker 		{
448*48a54d36SAndroid Build Coastguard Worker 		if (*src > MAX_DOMAIN_LABEL) return(MAX_DOMAIN_NAME+1);
449*48a54d36SAndroid Build Coastguard Worker 		if (parent && SameDomainName((const domainname *)src, parent)) return((mDNSu16)(src - name->c + 2));
450*48a54d36SAndroid Build Coastguard Worker 		src += 1 + *src;
451*48a54d36SAndroid Build Coastguard Worker 		if (src - name->c >= MAX_DOMAIN_NAME) return(MAX_DOMAIN_NAME+1);
452*48a54d36SAndroid Build Coastguard Worker 		}
453*48a54d36SAndroid Build Coastguard Worker 	return((mDNSu16)(src - name->c + 1));
454*48a54d36SAndroid Build Coastguard Worker 	}
455*48a54d36SAndroid Build Coastguard Worker 
456*48a54d36SAndroid Build Coastguard Worker // CountLabels() returns number of labels in name, excluding final root label
457*48a54d36SAndroid Build Coastguard Worker // (e.g. for "apple.com." CountLabels returns 2.)
CountLabels(const domainname * d)458*48a54d36SAndroid Build Coastguard Worker mDNSexport int CountLabels(const domainname *d)
459*48a54d36SAndroid Build Coastguard Worker 	{
460*48a54d36SAndroid Build Coastguard Worker 	int count = 0;
461*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr;
462*48a54d36SAndroid Build Coastguard Worker 	for (ptr = d->c; *ptr; ptr = ptr + ptr[0] + 1) count++;
463*48a54d36SAndroid Build Coastguard Worker 	return count;
464*48a54d36SAndroid Build Coastguard Worker 	}
465*48a54d36SAndroid Build Coastguard Worker 
466*48a54d36SAndroid Build Coastguard Worker // SkipLeadingLabels skips over the first 'skip' labels in the domainname,
467*48a54d36SAndroid Build Coastguard Worker // returning a pointer to the suffix with 'skip' labels removed.
SkipLeadingLabels(const domainname * d,int skip)468*48a54d36SAndroid Build Coastguard Worker mDNSexport const domainname *SkipLeadingLabels(const domainname *d, int skip)
469*48a54d36SAndroid Build Coastguard Worker 	{
470*48a54d36SAndroid Build Coastguard Worker 	while (skip > 0 && d->c[0]) { d = (const domainname *)(d->c + 1 + d->c[0]); skip--; }
471*48a54d36SAndroid Build Coastguard Worker 	return(d);
472*48a54d36SAndroid Build Coastguard Worker 	}
473*48a54d36SAndroid Build Coastguard Worker 
474*48a54d36SAndroid Build Coastguard Worker // AppendLiteralLabelString appends a single label to an existing (possibly empty) domainname.
475*48a54d36SAndroid Build Coastguard Worker // The C string contains the label as-is, with no escaping, etc.
476*48a54d36SAndroid Build Coastguard Worker // Any dots in the name are literal dots, not label separators
477*48a54d36SAndroid Build Coastguard Worker // If successful, AppendLiteralLabelString returns a pointer to the next unused byte
478*48a54d36SAndroid Build Coastguard Worker // in the domainname bufer (i.e. the next byte after the terminating zero).
479*48a54d36SAndroid Build Coastguard Worker // If unable to construct a legal domain name (i.e. label more than 63 bytes, or total more than 256 bytes)
480*48a54d36SAndroid Build Coastguard Worker // AppendLiteralLabelString returns mDNSNULL.
AppendLiteralLabelString(domainname * const name,const char * cstr)481*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *AppendLiteralLabelString(domainname *const name, const char *cstr)
482*48a54d36SAndroid Build Coastguard Worker 	{
483*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *      ptr  = name->c + DomainNameLength(name) - 1;	// Find end of current name
484*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim1 = name->c + MAX_DOMAIN_NAME - 1;			// Limit of how much we can add (not counting final zero)
485*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim2 = ptr + 1 + MAX_DOMAIN_LABEL;
486*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim  = (lim1 < lim2) ? lim1 : lim2;
487*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *lengthbyte = ptr++;									// Record where the length is going to go
488*48a54d36SAndroid Build Coastguard Worker 
489*48a54d36SAndroid Build Coastguard Worker 	while (*cstr && ptr < lim) *ptr++ = (mDNSu8)*cstr++;	// Copy the data
490*48a54d36SAndroid Build Coastguard Worker 	*lengthbyte = (mDNSu8)(ptr - lengthbyte - 1);			// Fill in the length byte
491*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = 0;												// Put the null root label on the end
492*48a54d36SAndroid Build Coastguard Worker 	if (*cstr) return(mDNSNULL);							// Failure: We didn't successfully consume all input
493*48a54d36SAndroid Build Coastguard Worker 	else return(ptr);										// Success: return new value of ptr
494*48a54d36SAndroid Build Coastguard Worker 	}
495*48a54d36SAndroid Build Coastguard Worker 
496*48a54d36SAndroid Build Coastguard Worker // AppendDNSNameString appends zero or more labels to an existing (possibly empty) domainname.
497*48a54d36SAndroid Build Coastguard Worker // The C string is in conventional DNS syntax:
498*48a54d36SAndroid Build Coastguard Worker // Textual labels, escaped as necessary using the usual DNS '\' notation, separated by dots.
499*48a54d36SAndroid Build Coastguard Worker // If successful, AppendDNSNameString returns a pointer to the next unused byte
500*48a54d36SAndroid Build Coastguard Worker // in the domainname bufer (i.e. the next byte after the terminating zero).
501*48a54d36SAndroid Build Coastguard Worker // If unable to construct a legal domain name (i.e. label more than 63 bytes, or total more than 256 bytes)
502*48a54d36SAndroid Build Coastguard Worker // AppendDNSNameString returns mDNSNULL.
AppendDNSNameString(domainname * const name,const char * cstring)503*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *AppendDNSNameString(domainname *const name, const char *cstring)
504*48a54d36SAndroid Build Coastguard Worker 	{
505*48a54d36SAndroid Build Coastguard Worker 	const char   *cstr      = cstring;
506*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *      ptr = name->c + DomainNameLength(name) - 1;	// Find end of current name
507*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim = name->c + MAX_DOMAIN_NAME - 1;		// Limit of how much we can add (not counting final zero)
508*48a54d36SAndroid Build Coastguard Worker 	while (*cstr && ptr < lim)										// While more characters, and space to put them...
509*48a54d36SAndroid Build Coastguard Worker 		{
510*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 *lengthbyte = ptr++;									// Record where the length is going to go
511*48a54d36SAndroid Build Coastguard Worker 		if (*cstr == '.') { LogMsg("AppendDNSNameString: Illegal empty label in name \"%s\"", cstring); return(mDNSNULL); }
512*48a54d36SAndroid Build Coastguard Worker 		while (*cstr && *cstr != '.' && ptr < lim)					// While we have characters in the label...
513*48a54d36SAndroid Build Coastguard Worker 			{
514*48a54d36SAndroid Build Coastguard Worker 			mDNSu8 c = (mDNSu8)*cstr++;								// Read the character
515*48a54d36SAndroid Build Coastguard Worker 			if (c == '\\')											// If escape character, check next character
516*48a54d36SAndroid Build Coastguard Worker 				{
517*48a54d36SAndroid Build Coastguard Worker 				c = (mDNSu8)*cstr++;								// Assume we'll just take the next character
518*48a54d36SAndroid Build Coastguard Worker 				if (mDNSIsDigit(cstr[-1]) && mDNSIsDigit(cstr[0]) && mDNSIsDigit(cstr[1]))
519*48a54d36SAndroid Build Coastguard Worker 					{												// If three decimal digits,
520*48a54d36SAndroid Build Coastguard Worker 					int v0 = cstr[-1] - '0';						// then interpret as three-digit decimal
521*48a54d36SAndroid Build Coastguard Worker 					int v1 = cstr[ 0] - '0';
522*48a54d36SAndroid Build Coastguard Worker 					int v2 = cstr[ 1] - '0';
523*48a54d36SAndroid Build Coastguard Worker 					int val = v0 * 100 + v1 * 10 + v2;
524*48a54d36SAndroid Build Coastguard Worker 					if (val <= 255) { c = (mDNSu8)val; cstr += 2; }	// If valid three-digit decimal value, use it
525*48a54d36SAndroid Build Coastguard Worker 					}
526*48a54d36SAndroid Build Coastguard Worker 				}
527*48a54d36SAndroid Build Coastguard Worker 			*ptr++ = c;												// Write the character
528*48a54d36SAndroid Build Coastguard Worker 			}
529*48a54d36SAndroid Build Coastguard Worker 		if (*cstr) cstr++;											// Skip over the trailing dot (if present)
530*48a54d36SAndroid Build Coastguard Worker 		if (ptr - lengthbyte - 1 > MAX_DOMAIN_LABEL)				// If illegal label, abort
531*48a54d36SAndroid Build Coastguard Worker 			return(mDNSNULL);
532*48a54d36SAndroid Build Coastguard Worker 		*lengthbyte = (mDNSu8)(ptr - lengthbyte - 1);				// Fill in the length byte
533*48a54d36SAndroid Build Coastguard Worker 		}
534*48a54d36SAndroid Build Coastguard Worker 
535*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = 0;														// Put the null root label on the end
536*48a54d36SAndroid Build Coastguard Worker 	if (*cstr) return(mDNSNULL);									// Failure: We didn't successfully consume all input
537*48a54d36SAndroid Build Coastguard Worker 	else return(ptr);												// Success: return new value of ptr
538*48a54d36SAndroid Build Coastguard Worker 	}
539*48a54d36SAndroid Build Coastguard Worker 
540*48a54d36SAndroid Build Coastguard Worker // AppendDomainLabel appends a single label to a name.
541*48a54d36SAndroid Build Coastguard Worker // If successful, AppendDomainLabel returns a pointer to the next unused byte
542*48a54d36SAndroid Build Coastguard Worker // in the domainname bufer (i.e. the next byte after the terminating zero).
543*48a54d36SAndroid Build Coastguard Worker // If unable to construct a legal domain name (i.e. label more than 63 bytes, or total more than 256 bytes)
544*48a54d36SAndroid Build Coastguard Worker // AppendDomainLabel returns mDNSNULL.
AppendDomainLabel(domainname * const name,const domainlabel * const label)545*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *AppendDomainLabel(domainname *const name, const domainlabel *const label)
546*48a54d36SAndroid Build Coastguard Worker 	{
547*48a54d36SAndroid Build Coastguard Worker 	int i;
548*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *ptr = name->c + DomainNameLength(name) - 1;
549*48a54d36SAndroid Build Coastguard Worker 
550*48a54d36SAndroid Build Coastguard Worker 	// Check label is legal
551*48a54d36SAndroid Build Coastguard Worker 	if (label->c[0] > MAX_DOMAIN_LABEL) return(mDNSNULL);
552*48a54d36SAndroid Build Coastguard Worker 
553*48a54d36SAndroid Build Coastguard Worker 	// Check that ptr + length byte + data bytes + final zero does not exceed our limit
554*48a54d36SAndroid Build Coastguard Worker 	if (ptr + 1 + label->c[0] + 1 > name->c + MAX_DOMAIN_NAME) return(mDNSNULL);
555*48a54d36SAndroid Build Coastguard Worker 
556*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=label->c[0]; i++) *ptr++ = label->c[i];	// Copy the label data
557*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = 0;								// Put the null root label on the end
558*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
559*48a54d36SAndroid Build Coastguard Worker 	}
560*48a54d36SAndroid Build Coastguard Worker 
AppendDomainName(domainname * const name,const domainname * const append)561*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *AppendDomainName(domainname *const name, const domainname *const append)
562*48a54d36SAndroid Build Coastguard Worker 	{
563*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *      ptr = name->c + DomainNameLength(name) - 1;	// Find end of current name
564*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim = name->c + MAX_DOMAIN_NAME - 1;		// Limit of how much we can add (not counting final zero)
565*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      src = append->c;
566*48a54d36SAndroid Build Coastguard Worker 	while (src[0])
567*48a54d36SAndroid Build Coastguard Worker 		{
568*48a54d36SAndroid Build Coastguard Worker 		int i;
569*48a54d36SAndroid Build Coastguard Worker 		if (ptr + 1 + src[0] > lim) return(mDNSNULL);
570*48a54d36SAndroid Build Coastguard Worker 		for (i=0; i<=src[0]; i++) *ptr++ = src[i];
571*48a54d36SAndroid Build Coastguard Worker 		*ptr = 0;	// Put the null root label on the end
572*48a54d36SAndroid Build Coastguard Worker 		src += i;
573*48a54d36SAndroid Build Coastguard Worker 		}
574*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
575*48a54d36SAndroid Build Coastguard Worker 	}
576*48a54d36SAndroid Build Coastguard Worker 
577*48a54d36SAndroid Build Coastguard Worker // MakeDomainLabelFromLiteralString makes a single domain label from a single literal C string (with no escaping).
578*48a54d36SAndroid Build Coastguard Worker // If successful, MakeDomainLabelFromLiteralString returns mDNStrue.
579*48a54d36SAndroid Build Coastguard Worker // If unable to convert the whole string to a legal domain label (i.e. because length is more than 63 bytes) then
580*48a54d36SAndroid Build Coastguard Worker // MakeDomainLabelFromLiteralString makes a legal domain label from the first 63 bytes of the string and returns mDNSfalse.
581*48a54d36SAndroid Build Coastguard Worker // In some cases silently truncated oversized names to 63 bytes is acceptable, so the return result may be ignored.
582*48a54d36SAndroid Build Coastguard Worker // In other cases silent truncation may not be acceptable, so in those cases the calling function needs to check the return result.
MakeDomainLabelFromLiteralString(domainlabel * const label,const char * cstr)583*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool MakeDomainLabelFromLiteralString(domainlabel *const label, const char *cstr)
584*48a54d36SAndroid Build Coastguard Worker 	{
585*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *      ptr   = label->c + 1;						// Where we're putting it
586*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const limit = label->c + 1 + MAX_DOMAIN_LABEL;	// The maximum we can put
587*48a54d36SAndroid Build Coastguard Worker 	while (*cstr && ptr < limit) *ptr++ = (mDNSu8)*cstr++;			// Copy the label
588*48a54d36SAndroid Build Coastguard Worker 	label->c[0] = (mDNSu8)(ptr - label->c - 1);						// Set the length byte
589*48a54d36SAndroid Build Coastguard Worker 	return(*cstr == 0);												// Return mDNStrue if we successfully consumed all input
590*48a54d36SAndroid Build Coastguard Worker 	}
591*48a54d36SAndroid Build Coastguard Worker 
592*48a54d36SAndroid Build Coastguard Worker // MakeDomainNameFromDNSNameString makes a native DNS-format domainname from a C string.
593*48a54d36SAndroid Build Coastguard Worker // The C string is in conventional DNS syntax:
594*48a54d36SAndroid Build Coastguard Worker // Textual labels, escaped as necessary using the usual DNS '\' notation, separated by dots.
595*48a54d36SAndroid Build Coastguard Worker // If successful, MakeDomainNameFromDNSNameString returns a pointer to the next unused byte
596*48a54d36SAndroid Build Coastguard Worker // in the domainname bufer (i.e. the next byte after the terminating zero).
597*48a54d36SAndroid Build Coastguard Worker // If unable to construct a legal domain name (i.e. label more than 63 bytes, or total more than 256 bytes)
598*48a54d36SAndroid Build Coastguard Worker // MakeDomainNameFromDNSNameString returns mDNSNULL.
MakeDomainNameFromDNSNameString(domainname * const name,const char * cstr)599*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *MakeDomainNameFromDNSNameString(domainname *const name, const char *cstr)
600*48a54d36SAndroid Build Coastguard Worker 	{
601*48a54d36SAndroid Build Coastguard Worker 	name->c[0] = 0;									// Make an empty domain name
602*48a54d36SAndroid Build Coastguard Worker 	return(AppendDNSNameString(name, cstr));		// And then add this string to it
603*48a54d36SAndroid Build Coastguard Worker 	}
604*48a54d36SAndroid Build Coastguard Worker 
ConvertDomainLabelToCString_withescape(const domainlabel * const label,char * ptr,char esc)605*48a54d36SAndroid Build Coastguard Worker mDNSexport char *ConvertDomainLabelToCString_withescape(const domainlabel *const label, char *ptr, char esc)
606*48a54d36SAndroid Build Coastguard Worker 	{
607*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      src = label->c;							// Domain label we're reading
608*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8        len = *src++;							// Read length of this (non-null) label
609*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const end = src + len;						// Work out where the label ends
610*48a54d36SAndroid Build Coastguard Worker 	if (len > MAX_DOMAIN_LABEL) return(mDNSNULL);				// If illegal label, abort
611*48a54d36SAndroid Build Coastguard Worker 	while (src < end)											// While we have characters in the label
612*48a54d36SAndroid Build Coastguard Worker 		{
613*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 c = *src++;
614*48a54d36SAndroid Build Coastguard Worker 		if (esc)
615*48a54d36SAndroid Build Coastguard Worker 			{
616*48a54d36SAndroid Build Coastguard Worker 			if (c == '.' || c == esc)							// If character is a dot or the escape character
617*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = esc;									// Output escape character
618*48a54d36SAndroid Build Coastguard Worker 			else if (c <= ' ')									// If non-printing ascii,
619*48a54d36SAndroid Build Coastguard Worker 				{												// Output decimal escape sequence
620*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = esc;
621*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = (char)  ('0' + (c / 100)     );
622*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = (char)  ('0' + (c /  10) % 10);
623*48a54d36SAndroid Build Coastguard Worker 				c      = (mDNSu8)('0' + (c      ) % 10);
624*48a54d36SAndroid Build Coastguard Worker 				}
625*48a54d36SAndroid Build Coastguard Worker 			}
626*48a54d36SAndroid Build Coastguard Worker 		*ptr++ = (char)c;										// Copy the character
627*48a54d36SAndroid Build Coastguard Worker 		}
628*48a54d36SAndroid Build Coastguard Worker 	*ptr = 0;													// Null-terminate the string
629*48a54d36SAndroid Build Coastguard Worker 	return(ptr);												// and return
630*48a54d36SAndroid Build Coastguard Worker 	}
631*48a54d36SAndroid Build Coastguard Worker 
632*48a54d36SAndroid Build Coastguard Worker // Note: To guarantee that there will be no possible overrun, cstr must be at least MAX_ESCAPED_DOMAIN_NAME (1009 bytes)
ConvertDomainNameToCString_withescape(const domainname * const name,char * ptr,char esc)633*48a54d36SAndroid Build Coastguard Worker mDNSexport char *ConvertDomainNameToCString_withescape(const domainname *const name, char *ptr, char esc)
634*48a54d36SAndroid Build Coastguard Worker 	{
635*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *src         = name->c;							// Domain name we're reading
636*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const max   = name->c + MAX_DOMAIN_NAME;			// Maximum that's valid
637*48a54d36SAndroid Build Coastguard Worker 
638*48a54d36SAndroid Build Coastguard Worker 	if (*src == 0) *ptr++ = '.';									// Special case: For root, just write a dot
639*48a54d36SAndroid Build Coastguard Worker 
640*48a54d36SAndroid Build Coastguard Worker 	while (*src)													// While more characters in the domain name
641*48a54d36SAndroid Build Coastguard Worker 		{
642*48a54d36SAndroid Build Coastguard Worker 		if (src + 1 + *src >= max) return(mDNSNULL);
643*48a54d36SAndroid Build Coastguard Worker 		ptr = ConvertDomainLabelToCString_withescape((const domainlabel *)src, ptr, esc);
644*48a54d36SAndroid Build Coastguard Worker 		if (!ptr) return(mDNSNULL);
645*48a54d36SAndroid Build Coastguard Worker 		src += 1 + *src;
646*48a54d36SAndroid Build Coastguard Worker 		*ptr++ = '.';												// Write the dot after the label
647*48a54d36SAndroid Build Coastguard Worker 		}
648*48a54d36SAndroid Build Coastguard Worker 
649*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = 0;														// Null-terminate the string
650*48a54d36SAndroid Build Coastguard Worker 	return(ptr);													// and return
651*48a54d36SAndroid Build Coastguard Worker 	}
652*48a54d36SAndroid Build Coastguard Worker 
653*48a54d36SAndroid Build Coastguard Worker // RFC 1034 rules:
654*48a54d36SAndroid Build Coastguard Worker // Host names must start with a letter, end with a letter or digit,
655*48a54d36SAndroid Build Coastguard Worker // and have as interior characters only letters, digits, and hyphen.
656*48a54d36SAndroid Build Coastguard Worker // This was subsequently modified in RFC 1123 to allow the first character to be either a letter or a digit
657*48a54d36SAndroid Build Coastguard Worker 
ConvertUTF8PstringToRFC1034HostLabel(const mDNSu8 UTF8Name[],domainlabel * const hostlabel)658*48a54d36SAndroid Build Coastguard Worker mDNSexport void ConvertUTF8PstringToRFC1034HostLabel(const mDNSu8 UTF8Name[], domainlabel *const hostlabel)
659*48a54d36SAndroid Build Coastguard Worker 	{
660*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      src  = &UTF8Name[1];
661*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const end  = &UTF8Name[1] + UTF8Name[0];
662*48a54d36SAndroid Build Coastguard Worker 	      mDNSu8 *      ptr  = &hostlabel->c[1];
663*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const lim  = &hostlabel->c[1] + MAX_DOMAIN_LABEL;
664*48a54d36SAndroid Build Coastguard Worker 	while (src < end)
665*48a54d36SAndroid Build Coastguard Worker 		{
666*48a54d36SAndroid Build Coastguard Worker 		// Delete apostrophes from source name
667*48a54d36SAndroid Build Coastguard Worker 		if (src[0] == '\'') { src++; continue; }		// Standard straight single quote
668*48a54d36SAndroid Build Coastguard Worker 		if (src + 2 < end && src[0] == 0xE2 && src[1] == 0x80 && src[2] == 0x99)
669*48a54d36SAndroid Build Coastguard Worker 			{ src += 3; continue; }	// Unicode curly apostrophe
670*48a54d36SAndroid Build Coastguard Worker 		if (ptr < lim)
671*48a54d36SAndroid Build Coastguard Worker 			{
672*48a54d36SAndroid Build Coastguard Worker 			if (mDNSValidHostChar(*src, (ptr > &hostlabel->c[1]), (src < end-1))) *ptr++ = *src;
673*48a54d36SAndroid Build Coastguard Worker 			else if (ptr > &hostlabel->c[1] && ptr[-1] != '-') *ptr++ = '-';
674*48a54d36SAndroid Build Coastguard Worker 			}
675*48a54d36SAndroid Build Coastguard Worker 		src++;
676*48a54d36SAndroid Build Coastguard Worker 		}
677*48a54d36SAndroid Build Coastguard Worker 	while (ptr > &hostlabel->c[1] && ptr[-1] == '-') ptr--;	// Truncate trailing '-' marks
678*48a54d36SAndroid Build Coastguard Worker 	hostlabel->c[0] = (mDNSu8)(ptr - &hostlabel->c[1]);
679*48a54d36SAndroid Build Coastguard Worker 	}
680*48a54d36SAndroid Build Coastguard Worker 
681*48a54d36SAndroid Build Coastguard Worker #define ValidTransportProtocol(X) ( (X)[0] == 4 && (X)[1] == '_' && \
682*48a54d36SAndroid Build Coastguard Worker 	((((X)[2] | 0x20) == 'u' && ((X)[3] | 0x20) == 'd') || (((X)[2] | 0x20) == 't' && ((X)[3] | 0x20) == 'c')) && \
683*48a54d36SAndroid Build Coastguard Worker 	((X)[4] | 0x20) == 'p')
684*48a54d36SAndroid Build Coastguard Worker 
ConstructServiceName(domainname * const fqdn,const domainlabel * name,const domainname * type,const domainname * const domain)685*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *ConstructServiceName(domainname *const fqdn,
686*48a54d36SAndroid Build Coastguard Worker 	const domainlabel *name, const domainname *type, const domainname *const domain)
687*48a54d36SAndroid Build Coastguard Worker 	{
688*48a54d36SAndroid Build Coastguard Worker 	int i, len;
689*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *dst = fqdn->c;
690*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *src;
691*48a54d36SAndroid Build Coastguard Worker 	const char *errormsg;
692*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
693*48a54d36SAndroid Build Coastguard Worker 	mDNSBool	loggedUnderscore = mDNSfalse;
694*48a54d36SAndroid Build Coastguard Worker 	static char typeBuf[MAX_ESCAPED_DOMAIN_NAME];
695*48a54d36SAndroid Build Coastguard Worker #endif
696*48a54d36SAndroid Build Coastguard Worker 
697*48a54d36SAndroid Build Coastguard Worker 	// In the case where there is no name (and ONLY in that case),
698*48a54d36SAndroid Build Coastguard Worker 	// a single-label subtype is allowed as the first label of a three-part "type"
699*48a54d36SAndroid Build Coastguard Worker 	if (!name && type)
700*48a54d36SAndroid Build Coastguard Worker 		{
701*48a54d36SAndroid Build Coastguard Worker 		const mDNSu8 *s0 = type->c;
702*48a54d36SAndroid Build Coastguard Worker 		if (s0[0] && s0[0] < 0x40)		// If legal first label (at least one character, and no more than 63)
703*48a54d36SAndroid Build Coastguard Worker 			{
704*48a54d36SAndroid Build Coastguard Worker 			const mDNSu8 * s1 = s0 + 1 + s0[0];
705*48a54d36SAndroid Build Coastguard Worker 			if (s1[0] && s1[0] < 0x40)	// and legal second label (at least one character, and no more than 63)
706*48a54d36SAndroid Build Coastguard Worker 				{
707*48a54d36SAndroid Build Coastguard Worker 				const mDNSu8 *s2 = s1 + 1 + s1[0];
708*48a54d36SAndroid Build Coastguard Worker 				if (s2[0] && s2[0] < 0x40 && s2[1+s2[0]] == 0)	// and we have three and only three labels
709*48a54d36SAndroid Build Coastguard Worker 					{
710*48a54d36SAndroid Build Coastguard Worker 					static const mDNSu8 SubTypeLabel[5] = "\x04_sub";
711*48a54d36SAndroid Build Coastguard Worker 					src = s0;									// Copy the first label
712*48a54d36SAndroid Build Coastguard Worker 					len = *src;
713*48a54d36SAndroid Build Coastguard Worker 					for (i=0; i <= len;                      i++) *dst++ = *src++;
714*48a54d36SAndroid Build Coastguard Worker 					for (i=0; i < (int)sizeof(SubTypeLabel); i++) *dst++ = SubTypeLabel[i];
715*48a54d36SAndroid Build Coastguard Worker 					type = (const domainname *)s1;
716*48a54d36SAndroid Build Coastguard Worker 
717*48a54d36SAndroid Build Coastguard Worker 					// Special support to enable the DNSServiceBrowse call made by Bonjour Browser
718*48a54d36SAndroid Build Coastguard Worker 					// For these queries, we retract the "._sub" we just added between the subtype and the main type
719*48a54d36SAndroid Build Coastguard Worker 					// Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse
720*48a54d36SAndroid Build Coastguard Worker 					if (SameDomainName((domainname*)s0, (const domainname*)"\x09_services\x07_dns-sd\x04_udp"))
721*48a54d36SAndroid Build Coastguard Worker 						dst -= sizeof(SubTypeLabel);
722*48a54d36SAndroid Build Coastguard Worker 					}
723*48a54d36SAndroid Build Coastguard Worker 				}
724*48a54d36SAndroid Build Coastguard Worker 			}
725*48a54d36SAndroid Build Coastguard Worker 		}
726*48a54d36SAndroid Build Coastguard Worker 
727*48a54d36SAndroid Build Coastguard Worker 	if (name && name->c[0])
728*48a54d36SAndroid Build Coastguard Worker 		{
729*48a54d36SAndroid Build Coastguard Worker 		src = name->c;									// Put the service name into the domain name
730*48a54d36SAndroid Build Coastguard Worker 		len = *src;
731*48a54d36SAndroid Build Coastguard Worker 		if (len >= 0x40) { errormsg = "Service instance name too long"; goto fail; }
732*48a54d36SAndroid Build Coastguard Worker 		for (i=0; i<=len; i++) *dst++ = *src++;
733*48a54d36SAndroid Build Coastguard Worker 		}
734*48a54d36SAndroid Build Coastguard Worker 	else
735*48a54d36SAndroid Build Coastguard Worker 		name = (domainlabel*)"";	// Set this up to be non-null, to avoid errors if we have to call LogMsg() below
736*48a54d36SAndroid Build Coastguard Worker 
737*48a54d36SAndroid Build Coastguard Worker 	src = type->c;										// Put the service type into the domain name
738*48a54d36SAndroid Build Coastguard Worker 	len = *src;
739*48a54d36SAndroid Build Coastguard Worker 	if (len < 2 || len > 16)
740*48a54d36SAndroid Build Coastguard Worker 		{
741*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Bad service type in %#s.%##s%##s Application protocol name must be underscore plus 1-15 characters. "
742*48a54d36SAndroid Build Coastguard Worker 			"See <http://www.dns-sd.org/ServiceTypes.html>", name->c, type->c, domain->c);
743*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
744*48a54d36SAndroid Build Coastguard Worker 		ConvertDomainNameToCString(type, typeBuf);
745*48a54d36SAndroid Build Coastguard Worker 		mDNSASLLog(mDNSNULL, "serviceType.nameTooLong", "noop", typeBuf, "");
746*48a54d36SAndroid Build Coastguard Worker #endif
747*48a54d36SAndroid Build Coastguard Worker 		}
748*48a54d36SAndroid Build Coastguard Worker 	if (len < 2 || len >= 0x40 || (len > 16 && !SameDomainName(domain, &localdomain))) return(mDNSNULL);
749*48a54d36SAndroid Build Coastguard Worker 	if (src[1] != '_') { errormsg = "Application protocol name must begin with underscore"; goto fail; }
750*48a54d36SAndroid Build Coastguard Worker 	for (i=2; i<=len; i++)
751*48a54d36SAndroid Build Coastguard Worker 		{
752*48a54d36SAndroid Build Coastguard Worker 		// Letters and digits are allowed anywhere
753*48a54d36SAndroid Build Coastguard Worker 		if (mDNSIsLetter(src[i]) || mDNSIsDigit(src[i])) continue;
754*48a54d36SAndroid Build Coastguard Worker 		// Hyphens are only allowed as interior characters
755*48a54d36SAndroid Build Coastguard Worker 		// Underscores are not supposed to be allowed at all, but for backwards compatibility with some old products we do allow them,
756*48a54d36SAndroid Build Coastguard Worker 		// with the same rule as hyphens
757*48a54d36SAndroid Build Coastguard Worker 		if ((src[i] == '-' || src[i] == '_') && i > 2 && i < len)
758*48a54d36SAndroid Build Coastguard Worker 			{
759*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
760*48a54d36SAndroid Build Coastguard Worker 			if (src[i] == '_' && loggedUnderscore == mDNSfalse)
761*48a54d36SAndroid Build Coastguard Worker 				{
762*48a54d36SAndroid Build Coastguard Worker 				ConvertDomainNameToCString(type, typeBuf);
763*48a54d36SAndroid Build Coastguard Worker 				mDNSASLLog(mDNSNULL, "serviceType.nameWithUnderscore", "noop", typeBuf, "");
764*48a54d36SAndroid Build Coastguard Worker 				loggedUnderscore = mDNStrue;
765*48a54d36SAndroid Build Coastguard Worker 				}
766*48a54d36SAndroid Build Coastguard Worker #endif
767*48a54d36SAndroid Build Coastguard Worker 			continue;
768*48a54d36SAndroid Build Coastguard Worker 			}
769*48a54d36SAndroid Build Coastguard Worker 		errormsg = "Application protocol name must contain only letters, digits, and hyphens";
770*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
771*48a54d36SAndroid Build Coastguard Worker 		{
772*48a54d36SAndroid Build Coastguard Worker 		ConvertDomainNameToCString(type, typeBuf);
773*48a54d36SAndroid Build Coastguard Worker 		mDNSASLLog(mDNSNULL, "serviceType.nameWithIllegalCharacters", "noop", typeBuf, "");
774*48a54d36SAndroid Build Coastguard Worker 		}
775*48a54d36SAndroid Build Coastguard Worker #endif
776*48a54d36SAndroid Build Coastguard Worker 		 goto fail;
777*48a54d36SAndroid Build Coastguard Worker 		}
778*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=len; i++) *dst++ = *src++;
779*48a54d36SAndroid Build Coastguard Worker 
780*48a54d36SAndroid Build Coastguard Worker 	len = *src;
781*48a54d36SAndroid Build Coastguard Worker 	if (!ValidTransportProtocol(src)) { errormsg = "Transport protocol name must be _udp or _tcp"; goto fail; }
782*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=len; i++) *dst++ = *src++;
783*48a54d36SAndroid Build Coastguard Worker 
784*48a54d36SAndroid Build Coastguard Worker 	if (*src) { errormsg = "Service type must have only two labels"; goto fail; }
785*48a54d36SAndroid Build Coastguard Worker 
786*48a54d36SAndroid Build Coastguard Worker 	*dst = 0;
787*48a54d36SAndroid Build Coastguard Worker 	if (!domain->c[0]) { errormsg = "Service domain must be non-empty"; goto fail; }
788*48a54d36SAndroid Build Coastguard Worker 	if (SameDomainName(domain, (const domainname*)"\x05" "local" "\x04" "arpa"))
789*48a54d36SAndroid Build Coastguard Worker 		{ errormsg = "Illegal domain \"local.arpa.\" Use \"local.\" (or empty string)"; goto fail; }
790*48a54d36SAndroid Build Coastguard Worker 	dst = AppendDomainName(fqdn, domain);
791*48a54d36SAndroid Build Coastguard Worker 	if (!dst) { errormsg = "Service domain too long"; goto fail; }
792*48a54d36SAndroid Build Coastguard Worker 	return(dst);
793*48a54d36SAndroid Build Coastguard Worker 
794*48a54d36SAndroid Build Coastguard Worker fail:
795*48a54d36SAndroid Build Coastguard Worker 	LogMsg("ConstructServiceName: %s: %#s.%##s%##s", errormsg, name->c, type->c, domain->c);
796*48a54d36SAndroid Build Coastguard Worker 	return(mDNSNULL);
797*48a54d36SAndroid Build Coastguard Worker 	}
798*48a54d36SAndroid Build Coastguard Worker 
799*48a54d36SAndroid Build Coastguard Worker // A service name has the form: instance.application-protocol.transport-protocol.domain
800*48a54d36SAndroid Build Coastguard Worker // DeconstructServiceName is currently fairly forgiving: It doesn't try to enforce character
801*48a54d36SAndroid Build Coastguard Worker // set or length limits for the protocol names, and the final domain is allowed to be empty.
802*48a54d36SAndroid Build Coastguard Worker // However, if the given FQDN doesn't contain at least three labels,
803*48a54d36SAndroid Build Coastguard Worker // DeconstructServiceName will reject it and return mDNSfalse.
DeconstructServiceName(const domainname * const fqdn,domainlabel * const name,domainname * const type,domainname * const domain)804*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool DeconstructServiceName(const domainname *const fqdn,
805*48a54d36SAndroid Build Coastguard Worker 	domainlabel *const name, domainname *const type, domainname *const domain)
806*48a54d36SAndroid Build Coastguard Worker 	{
807*48a54d36SAndroid Build Coastguard Worker 	int i, len;
808*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *src = fqdn->c;
809*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *max = fqdn->c + MAX_DOMAIN_NAME;
810*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *dst;
811*48a54d36SAndroid Build Coastguard Worker 
812*48a54d36SAndroid Build Coastguard Worker 	dst = name->c;										// Extract the service name
813*48a54d36SAndroid Build Coastguard Worker 	len = *src;
814*48a54d36SAndroid Build Coastguard Worker 	if (!len)         { debugf("DeconstructServiceName: FQDN empty!");                             return(mDNSfalse); }
815*48a54d36SAndroid Build Coastguard Worker 	if (len >= 0x40)  { debugf("DeconstructServiceName: Instance name too long");                  return(mDNSfalse); }
816*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=len; i++) *dst++ = *src++;
817*48a54d36SAndroid Build Coastguard Worker 
818*48a54d36SAndroid Build Coastguard Worker 	dst = type->c;										// Extract the service type
819*48a54d36SAndroid Build Coastguard Worker 	len = *src;
820*48a54d36SAndroid Build Coastguard Worker 	if (!len)         { debugf("DeconstructServiceName: FQDN contains only one label!");           return(mDNSfalse); }
821*48a54d36SAndroid Build Coastguard Worker 	if (len >= 0x40)  { debugf("DeconstructServiceName: Application protocol name too long");      return(mDNSfalse); }
822*48a54d36SAndroid Build Coastguard Worker 	if (src[1] != '_'){ debugf("DeconstructServiceName: No _ at start of application protocol");   return(mDNSfalse); }
823*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=len; i++) *dst++ = *src++;
824*48a54d36SAndroid Build Coastguard Worker 
825*48a54d36SAndroid Build Coastguard Worker 	len = *src;
826*48a54d36SAndroid Build Coastguard Worker 	if (!len)         { debugf("DeconstructServiceName: FQDN contains only two labels!");          return(mDNSfalse); }
827*48a54d36SAndroid Build Coastguard Worker 	if (!ValidTransportProtocol(src))
828*48a54d36SAndroid Build Coastguard Worker 	                  { debugf("DeconstructServiceName: Transport protocol must be _udp or _tcp"); return(mDNSfalse); }
829*48a54d36SAndroid Build Coastguard Worker 	for (i=0; i<=len; i++) *dst++ = *src++;
830*48a54d36SAndroid Build Coastguard Worker 	*dst++ = 0;											// Put terminator on the end of service type
831*48a54d36SAndroid Build Coastguard Worker 
832*48a54d36SAndroid Build Coastguard Worker 	dst = domain->c;									// Extract the service domain
833*48a54d36SAndroid Build Coastguard Worker 	while (*src)
834*48a54d36SAndroid Build Coastguard Worker 		{
835*48a54d36SAndroid Build Coastguard Worker 		len = *src;
836*48a54d36SAndroid Build Coastguard Worker 		if (len >= 0x40)
837*48a54d36SAndroid Build Coastguard Worker 			{ debugf("DeconstructServiceName: Label in service domain too long"); return(mDNSfalse); }
838*48a54d36SAndroid Build Coastguard Worker 		if (src + 1 + len + 1 >= max)
839*48a54d36SAndroid Build Coastguard Worker 			{ debugf("DeconstructServiceName: Total service domain too long"); return(mDNSfalse); }
840*48a54d36SAndroid Build Coastguard Worker 		for (i=0; i<=len; i++) *dst++ = *src++;
841*48a54d36SAndroid Build Coastguard Worker 		}
842*48a54d36SAndroid Build Coastguard Worker 	*dst++ = 0;		// Put the null root label on the end
843*48a54d36SAndroid Build Coastguard Worker 
844*48a54d36SAndroid Build Coastguard Worker 	return(mDNStrue);
845*48a54d36SAndroid Build Coastguard Worker 	}
846*48a54d36SAndroid Build Coastguard Worker 
847*48a54d36SAndroid Build Coastguard Worker // Notes on UTF-8:
848*48a54d36SAndroid Build Coastguard Worker // 0xxxxxxx represents a 7-bit ASCII value from 0x00 to 0x7F
849*48a54d36SAndroid Build Coastguard Worker // 10xxxxxx is a continuation byte of a multi-byte character
850*48a54d36SAndroid Build Coastguard Worker // 110xxxxx is the first byte of a 2-byte character (11 effective bits; values 0x     80 - 0x     800-1)
851*48a54d36SAndroid Build Coastguard Worker // 1110xxxx is the first byte of a 3-byte character (16 effective bits; values 0x    800 - 0x   10000-1)
852*48a54d36SAndroid Build Coastguard Worker // 11110xxx is the first byte of a 4-byte character (21 effective bits; values 0x  10000 - 0x  200000-1)
853*48a54d36SAndroid Build Coastguard Worker // 111110xx is the first byte of a 5-byte character (26 effective bits; values 0x 200000 - 0x 4000000-1)
854*48a54d36SAndroid Build Coastguard Worker // 1111110x is the first byte of a 6-byte character (31 effective bits; values 0x4000000 - 0x80000000-1)
855*48a54d36SAndroid Build Coastguard Worker //
856*48a54d36SAndroid Build Coastguard Worker // UTF-16 surrogate pairs are used in UTF-16 to encode values larger than 0xFFFF.
857*48a54d36SAndroid Build Coastguard Worker // Although UTF-16 surrogate pairs are not supposed to appear in legal UTF-8, we want to be defensive
858*48a54d36SAndroid Build Coastguard Worker // about that too. (See <http://www.unicode.org/faq/utf_bom.html#34>, "What are surrogates?")
859*48a54d36SAndroid Build Coastguard Worker // The first of pair is a UTF-16 value in the range 0xD800-0xDBFF (11101101 1010xxxx 10xxxxxx in UTF-8),
860*48a54d36SAndroid Build Coastguard Worker // and the second    is a UTF-16 value in the range 0xDC00-0xDFFF (11101101 1011xxxx 10xxxxxx in UTF-8).
861*48a54d36SAndroid Build Coastguard Worker 
TruncateUTF8ToLength(mDNSu8 * string,mDNSu32 length,mDNSu32 max)862*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 TruncateUTF8ToLength(mDNSu8 *string, mDNSu32 length, mDNSu32 max)
863*48a54d36SAndroid Build Coastguard Worker 	{
864*48a54d36SAndroid Build Coastguard Worker 	if (length > max)
865*48a54d36SAndroid Build Coastguard Worker 		{
866*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 c1 = string[max];										// First byte after cut point
867*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 c2 = (max+1 < length) ? string[max+1] : (mDNSu8)0xB0;	// Second byte after cut point
868*48a54d36SAndroid Build Coastguard Worker 		length = max;	// Trim length down
869*48a54d36SAndroid Build Coastguard Worker 		while (length > 0)
870*48a54d36SAndroid Build Coastguard Worker 			{
871*48a54d36SAndroid Build Coastguard Worker 			// Check if the byte right after the chop point is a UTF-8 continuation byte,
872*48a54d36SAndroid Build Coastguard Worker 			// or if the character right after the chop point is the second of a UTF-16 surrogate pair.
873*48a54d36SAndroid Build Coastguard Worker 			// If so, then we continue to chop more bytes until we get to a legal chop point.
874*48a54d36SAndroid Build Coastguard Worker 			mDNSBool continuation    = ((c1 & 0xC0) == 0x80);
875*48a54d36SAndroid Build Coastguard Worker 			mDNSBool secondsurrogate = (c1 == 0xED && (c2 & 0xF0) == 0xB0);
876*48a54d36SAndroid Build Coastguard Worker 			if (!continuation && !secondsurrogate) break;
877*48a54d36SAndroid Build Coastguard Worker 			c2 = c1;
878*48a54d36SAndroid Build Coastguard Worker 			c1 = string[--length];
879*48a54d36SAndroid Build Coastguard Worker 			}
880*48a54d36SAndroid Build Coastguard Worker 		// Having truncated characters off the end of our string, also cut off any residual white space
881*48a54d36SAndroid Build Coastguard Worker 		while (length > 0 && string[length-1] <= ' ') length--;
882*48a54d36SAndroid Build Coastguard Worker 		}
883*48a54d36SAndroid Build Coastguard Worker 	return(length);
884*48a54d36SAndroid Build Coastguard Worker 	}
885*48a54d36SAndroid Build Coastguard Worker 
886*48a54d36SAndroid Build Coastguard Worker // Returns true if a rich text label ends in " (nnn)", or if an RFC 1034
887*48a54d36SAndroid Build Coastguard Worker // name ends in "-nnn", where n is some decimal number.
LabelContainsSuffix(const domainlabel * const name,const mDNSBool RichText)888*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool LabelContainsSuffix(const domainlabel *const name, const mDNSBool RichText)
889*48a54d36SAndroid Build Coastguard Worker 	{
890*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 l = name->c[0];
891*48a54d36SAndroid Build Coastguard Worker 
892*48a54d36SAndroid Build Coastguard Worker 	if (RichText)
893*48a54d36SAndroid Build Coastguard Worker 		{
894*48a54d36SAndroid Build Coastguard Worker 		if (l < 4) return mDNSfalse;							// Need at least " (2)"
895*48a54d36SAndroid Build Coastguard Worker 		if (name->c[l--] != ')') return mDNSfalse;				// Last char must be ')'
896*48a54d36SAndroid Build Coastguard Worker 		if (!mDNSIsDigit(name->c[l])) return mDNSfalse;			// Preceeded by a digit
897*48a54d36SAndroid Build Coastguard Worker 		l--;
898*48a54d36SAndroid Build Coastguard Worker 		while (l > 2 && mDNSIsDigit(name->c[l])) l--;			// Strip off digits
899*48a54d36SAndroid Build Coastguard Worker 		return (name->c[l] == '(' && name->c[l - 1] == ' ');
900*48a54d36SAndroid Build Coastguard Worker 		}
901*48a54d36SAndroid Build Coastguard Worker 	else
902*48a54d36SAndroid Build Coastguard Worker 		{
903*48a54d36SAndroid Build Coastguard Worker 		if (l < 2) return mDNSfalse;							// Need at least "-2"
904*48a54d36SAndroid Build Coastguard Worker 		if (!mDNSIsDigit(name->c[l])) return mDNSfalse;			// Last char must be a digit
905*48a54d36SAndroid Build Coastguard Worker 		l--;
906*48a54d36SAndroid Build Coastguard Worker 		while (l > 2 && mDNSIsDigit(name->c[l])) l--;			// Strip off digits
907*48a54d36SAndroid Build Coastguard Worker 		return (name->c[l] == '-');
908*48a54d36SAndroid Build Coastguard Worker 		}
909*48a54d36SAndroid Build Coastguard Worker 	}
910*48a54d36SAndroid Build Coastguard Worker 
911*48a54d36SAndroid Build Coastguard Worker // removes an auto-generated suffix (appended on a name collision) from a label.  caller is
912*48a54d36SAndroid Build Coastguard Worker // responsible for ensuring that the label does indeed contain a suffix.  returns the number
913*48a54d36SAndroid Build Coastguard Worker // from the suffix that was removed.
RemoveLabelSuffix(domainlabel * name,mDNSBool RichText)914*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 RemoveLabelSuffix(domainlabel *name, mDNSBool RichText)
915*48a54d36SAndroid Build Coastguard Worker 	{
916*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 val = 0, multiplier = 1;
917*48a54d36SAndroid Build Coastguard Worker 
918*48a54d36SAndroid Build Coastguard Worker 	// Chop closing parentheses from RichText suffix
919*48a54d36SAndroid Build Coastguard Worker 	if (RichText && name->c[0] >= 1 && name->c[name->c[0]] == ')') name->c[0]--;
920*48a54d36SAndroid Build Coastguard Worker 
921*48a54d36SAndroid Build Coastguard Worker 	// Get any existing numerical suffix off the name
922*48a54d36SAndroid Build Coastguard Worker 	while (mDNSIsDigit(name->c[name->c[0]]))
923*48a54d36SAndroid Build Coastguard Worker 		{ val += (name->c[name->c[0]] - '0') * multiplier; multiplier *= 10; name->c[0]--; }
924*48a54d36SAndroid Build Coastguard Worker 
925*48a54d36SAndroid Build Coastguard Worker 	// Chop opening parentheses or dash from suffix
926*48a54d36SAndroid Build Coastguard Worker 	if (RichText)
927*48a54d36SAndroid Build Coastguard Worker 		{
928*48a54d36SAndroid Build Coastguard Worker 		if (name->c[0] >= 2 && name->c[name->c[0]] == '(' && name->c[name->c[0]-1] == ' ') name->c[0] -= 2;
929*48a54d36SAndroid Build Coastguard Worker 		}
930*48a54d36SAndroid Build Coastguard Worker 	else
931*48a54d36SAndroid Build Coastguard Worker 		{
932*48a54d36SAndroid Build Coastguard Worker 		if (name->c[0] >= 1 && name->c[name->c[0]] == '-') name->c[0] -= 1;
933*48a54d36SAndroid Build Coastguard Worker 		}
934*48a54d36SAndroid Build Coastguard Worker 
935*48a54d36SAndroid Build Coastguard Worker 	return(val);
936*48a54d36SAndroid Build Coastguard Worker 	}
937*48a54d36SAndroid Build Coastguard Worker 
938*48a54d36SAndroid Build Coastguard Worker // appends a numerical suffix to a label, with the number following a whitespace and enclosed
939*48a54d36SAndroid Build Coastguard Worker // in parentheses (rich text) or following two consecutive hyphens (RFC 1034 domain label).
AppendLabelSuffix(domainlabel * const name,mDNSu32 val,const mDNSBool RichText)940*48a54d36SAndroid Build Coastguard Worker mDNSexport void AppendLabelSuffix(domainlabel *const name, mDNSu32 val, const mDNSBool RichText)
941*48a54d36SAndroid Build Coastguard Worker 	{
942*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 divisor = 1, chars = 2;	// Shortest possible RFC1034 name suffix is 2 characters ("-2")
943*48a54d36SAndroid Build Coastguard Worker 	if (RichText) chars = 4;		// Shortest possible RichText suffix is 4 characters (" (2)")
944*48a54d36SAndroid Build Coastguard Worker 
945*48a54d36SAndroid Build Coastguard Worker 	// Truncate trailing spaces from RichText names
946*48a54d36SAndroid Build Coastguard Worker 	if (RichText) while (name->c[name->c[0]] == ' ') name->c[0]--;
947*48a54d36SAndroid Build Coastguard Worker 
948*48a54d36SAndroid Build Coastguard Worker 	while (divisor < 0xFFFFFFFFUL/10 && val >= divisor * 10) { divisor *= 10; chars++; }
949*48a54d36SAndroid Build Coastguard Worker 
950*48a54d36SAndroid Build Coastguard Worker 	name->c[0] = (mDNSu8) TruncateUTF8ToLength(name->c+1, name->c[0], MAX_DOMAIN_LABEL - chars);
951*48a54d36SAndroid Build Coastguard Worker 
952*48a54d36SAndroid Build Coastguard Worker 	if (RichText) { name->c[++name->c[0]] = ' '; name->c[++name->c[0]] = '('; }
953*48a54d36SAndroid Build Coastguard Worker 	else          { name->c[++name->c[0]] = '-'; }
954*48a54d36SAndroid Build Coastguard Worker 
955*48a54d36SAndroid Build Coastguard Worker 	while (divisor)
956*48a54d36SAndroid Build Coastguard Worker 		{
957*48a54d36SAndroid Build Coastguard Worker 		name->c[++name->c[0]] = (mDNSu8)('0' + val / divisor);
958*48a54d36SAndroid Build Coastguard Worker 		val     %= divisor;
959*48a54d36SAndroid Build Coastguard Worker 		divisor /= 10;
960*48a54d36SAndroid Build Coastguard Worker 		}
961*48a54d36SAndroid Build Coastguard Worker 
962*48a54d36SAndroid Build Coastguard Worker 	if (RichText) name->c[++name->c[0]] = ')';
963*48a54d36SAndroid Build Coastguard Worker 	}
964*48a54d36SAndroid Build Coastguard Worker 
IncrementLabelSuffix(domainlabel * name,mDNSBool RichText)965*48a54d36SAndroid Build Coastguard Worker mDNSexport void IncrementLabelSuffix(domainlabel *name, mDNSBool RichText)
966*48a54d36SAndroid Build Coastguard Worker 	{
967*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 val = 0;
968*48a54d36SAndroid Build Coastguard Worker 
969*48a54d36SAndroid Build Coastguard Worker 	if (LabelContainsSuffix(name, RichText))
970*48a54d36SAndroid Build Coastguard Worker 		val = RemoveLabelSuffix(name, RichText);
971*48a54d36SAndroid Build Coastguard Worker 
972*48a54d36SAndroid Build Coastguard Worker 	// If no existing suffix, start by renaming "Foo" as "Foo (2)" or "Foo-2" as appropriate.
973*48a54d36SAndroid Build Coastguard Worker 	// If existing suffix in the range 2-9, increment it.
974*48a54d36SAndroid Build Coastguard Worker 	// If we've had ten conflicts already, there are probably too many hosts trying to use the same name,
975*48a54d36SAndroid Build Coastguard Worker 	// so add a random increment to improve the chances of finding an available name next time.
976*48a54d36SAndroid Build Coastguard Worker 	if      (val == 0) val = 2;
977*48a54d36SAndroid Build Coastguard Worker 	else if (val < 10) val++;
978*48a54d36SAndroid Build Coastguard Worker 	else               val += 1 + mDNSRandom(99);
979*48a54d36SAndroid Build Coastguard Worker 
980*48a54d36SAndroid Build Coastguard Worker 	AppendLabelSuffix(name, val, RichText);
981*48a54d36SAndroid Build Coastguard Worker 	}
982*48a54d36SAndroid Build Coastguard Worker 
983*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
984*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
985*48a54d36SAndroid Build Coastguard Worker #pragma mark -
986*48a54d36SAndroid Build Coastguard Worker #pragma mark - Resource Record Utility Functions
987*48a54d36SAndroid Build Coastguard Worker #endif
988*48a54d36SAndroid Build Coastguard Worker 
989*48a54d36SAndroid Build Coastguard Worker // Set up a AuthRecord with sensible default values.
990*48a54d36SAndroid Build Coastguard Worker // These defaults may be overwritten with new values before mDNS_Register is called
mDNS_SetupResourceRecord(AuthRecord * rr,RData * RDataStorage,mDNSInterfaceID InterfaceID,mDNSu16 rrtype,mDNSu32 ttl,mDNSu8 RecordType,AuthRecType artype,mDNSRecordCallback Callback,void * Context)991*48a54d36SAndroid Build Coastguard Worker mDNSexport void mDNS_SetupResourceRecord(AuthRecord *rr, RData *RDataStorage, mDNSInterfaceID InterfaceID,
992*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 rrtype, mDNSu32 ttl, mDNSu8 RecordType, AuthRecType artype, mDNSRecordCallback Callback, void *Context)
993*48a54d36SAndroid Build Coastguard Worker 	{
994*48a54d36SAndroid Build Coastguard Worker 	//
995*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly auth record can be created with LocalOnly InterfaceID or a valid InterfaceID.
996*48a54d36SAndroid Build Coastguard Worker 	// Most of the applications normally create with LocalOnly InterfaceID and we store them as
997*48a54d36SAndroid Build Coastguard Worker 	// such, so that we can deliver the response to questions that specify LocalOnly InterfaceID.
998*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly resource records can also be created with valid InterfaceID which happens today
999*48a54d36SAndroid Build Coastguard Worker 	// when we create LocalOnly records for /etc/hosts.
1000*48a54d36SAndroid Build Coastguard Worker 
1001*48a54d36SAndroid Build Coastguard Worker 	if (InterfaceID == mDNSInterface_LocalOnly && artype != AuthRecordLocalOnly)
1002*48a54d36SAndroid Build Coastguard Worker 		{
1003*48a54d36SAndroid Build Coastguard Worker 		LogMsg("mDNS_SetupResourceRecord: ERROR!! Mismatch LocalOnly record InterfaceID %p called with artype %d", InterfaceID, artype);
1004*48a54d36SAndroid Build Coastguard Worker 		return;
1005*48a54d36SAndroid Build Coastguard Worker 		}
1006*48a54d36SAndroid Build Coastguard Worker 	else if (InterfaceID == mDNSInterface_P2P && artype != AuthRecordP2P)
1007*48a54d36SAndroid Build Coastguard Worker 		{
1008*48a54d36SAndroid Build Coastguard Worker 		LogMsg("mDNS_SetupResourceRecord: ERROR!! Mismatch P2P record InterfaceID %p called with artype %d", InterfaceID, artype);
1009*48a54d36SAndroid Build Coastguard Worker 		return;
1010*48a54d36SAndroid Build Coastguard Worker 		}
1011*48a54d36SAndroid Build Coastguard Worker 	else if (!InterfaceID && (artype == AuthRecordP2P || artype == AuthRecordLocalOnly))
1012*48a54d36SAndroid Build Coastguard Worker 		{
1013*48a54d36SAndroid Build Coastguard Worker 		LogMsg("mDNS_SetupResourceRecord: ERROR!! Mismatch InterfaceAny record InterfaceID %p called with artype %d", InterfaceID, artype);
1014*48a54d36SAndroid Build Coastguard Worker 		return;
1015*48a54d36SAndroid Build Coastguard Worker 		}
1016*48a54d36SAndroid Build Coastguard Worker 
1017*48a54d36SAndroid Build Coastguard Worker 	// Don't try to store a TTL bigger than we can represent in platform time units
1018*48a54d36SAndroid Build Coastguard Worker 	if (ttl > 0x7FFFFFFFUL / mDNSPlatformOneSecond)
1019*48a54d36SAndroid Build Coastguard Worker 		ttl = 0x7FFFFFFFUL / mDNSPlatformOneSecond;
1020*48a54d36SAndroid Build Coastguard Worker 	else if (ttl == 0)		// And Zero TTL is illegal
1021*48a54d36SAndroid Build Coastguard Worker 		ttl = DefaultTTLforRRType(rrtype);
1022*48a54d36SAndroid Build Coastguard Worker 
1023*48a54d36SAndroid Build Coastguard Worker 	// Field Group 1: The actual information pertaining to this resource record
1024*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.RecordType        = RecordType;
1025*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.InterfaceID       = InterfaceID;
1026*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.name              = &rr->namestorage;
1027*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rrtype            = rrtype;
1028*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rrclass           = kDNSClass_IN;
1029*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rroriginalttl     = ttl;
1030*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rDNSServer		 = mDNSNULL;
1031*48a54d36SAndroid Build Coastguard Worker //	rr->resrec.rdlength          = MUST set by client and/or in mDNS_Register_internal
1032*48a54d36SAndroid Build Coastguard Worker //	rr->resrec.rdestimate        = set in mDNS_Register_internal
1033*48a54d36SAndroid Build Coastguard Worker //	rr->resrec.rdata             = MUST be set by client
1034*48a54d36SAndroid Build Coastguard Worker 
1035*48a54d36SAndroid Build Coastguard Worker 	if (RDataStorage)
1036*48a54d36SAndroid Build Coastguard Worker 		rr->resrec.rdata = RDataStorage;
1037*48a54d36SAndroid Build Coastguard Worker 	else
1038*48a54d36SAndroid Build Coastguard Worker 		{
1039*48a54d36SAndroid Build Coastguard Worker 		rr->resrec.rdata = &rr->rdatastorage;
1040*48a54d36SAndroid Build Coastguard Worker 		rr->resrec.rdata->MaxRDLength = sizeof(RDataBody);
1041*48a54d36SAndroid Build Coastguard Worker 		}
1042*48a54d36SAndroid Build Coastguard Worker 
1043*48a54d36SAndroid Build Coastguard Worker 	// Field Group 2: Persistent metadata for Authoritative Records
1044*48a54d36SAndroid Build Coastguard Worker 	rr->Additional1       = mDNSNULL;
1045*48a54d36SAndroid Build Coastguard Worker 	rr->Additional2       = mDNSNULL;
1046*48a54d36SAndroid Build Coastguard Worker 	rr->DependentOn       = mDNSNULL;
1047*48a54d36SAndroid Build Coastguard Worker 	rr->RRSet             = mDNSNULL;
1048*48a54d36SAndroid Build Coastguard Worker 	rr->RecordCallback    = Callback;
1049*48a54d36SAndroid Build Coastguard Worker 	rr->RecordContext     = Context;
1050*48a54d36SAndroid Build Coastguard Worker 
1051*48a54d36SAndroid Build Coastguard Worker 	rr->AutoTarget        = Target_Manual;
1052*48a54d36SAndroid Build Coastguard Worker 	rr->AllowRemoteQuery  = mDNSfalse;
1053*48a54d36SAndroid Build Coastguard Worker 	rr->ForceMCast        = mDNSfalse;
1054*48a54d36SAndroid Build Coastguard Worker 
1055*48a54d36SAndroid Build Coastguard Worker 	rr->WakeUp            = zeroOwner;
1056*48a54d36SAndroid Build Coastguard Worker 	rr->AddressProxy      = zeroAddr;
1057*48a54d36SAndroid Build Coastguard Worker 	rr->TimeRcvd          = 0;
1058*48a54d36SAndroid Build Coastguard Worker 	rr->TimeExpire        = 0;
1059*48a54d36SAndroid Build Coastguard Worker 	rr->ARType            = artype;
1060*48a54d36SAndroid Build Coastguard Worker 
1061*48a54d36SAndroid Build Coastguard Worker 	// Field Group 3: Transient state for Authoritative Records (set in mDNS_Register_internal)
1062*48a54d36SAndroid Build Coastguard Worker 	// Field Group 4: Transient uDNS state for Authoritative Records (set in mDNS_Register_internal)
1063*48a54d36SAndroid Build Coastguard Worker 
1064*48a54d36SAndroid Build Coastguard Worker 	// For now, until the uDNS code is fully integrated, it's helpful to zero the uDNS state fields here too, just in case
1065*48a54d36SAndroid Build Coastguard Worker 	// (e.g. uDNS_RegisterService short-circuits the usual mDNS_Register_internal record registration calls, so a bunch
1066*48a54d36SAndroid Build Coastguard Worker 	// of fields don't get set up properly. In particular, if we don't zero rr->QueuedRData then the uDNS code crashes.)
1067*48a54d36SAndroid Build Coastguard Worker 	rr->state             = regState_Zero;
1068*48a54d36SAndroid Build Coastguard Worker 	rr->uselease          = 0;
1069*48a54d36SAndroid Build Coastguard Worker 	rr->expire            = 0;
1070*48a54d36SAndroid Build Coastguard Worker 	rr->Private           = 0;
1071*48a54d36SAndroid Build Coastguard Worker 	rr->updateid          = zeroID;
1072*48a54d36SAndroid Build Coastguard Worker 	rr->zone              = rr->resrec.name;
1073*48a54d36SAndroid Build Coastguard Worker 	rr->nta               = mDNSNULL;
1074*48a54d36SAndroid Build Coastguard Worker 	rr->tcp               = mDNSNULL;
1075*48a54d36SAndroid Build Coastguard Worker 	rr->OrigRData         = 0;
1076*48a54d36SAndroid Build Coastguard Worker 	rr->OrigRDLen         = 0;
1077*48a54d36SAndroid Build Coastguard Worker 	rr->InFlightRData     = 0;
1078*48a54d36SAndroid Build Coastguard Worker 	rr->InFlightRDLen     = 0;
1079*48a54d36SAndroid Build Coastguard Worker 	rr->QueuedRData       = 0;
1080*48a54d36SAndroid Build Coastguard Worker 	rr->QueuedRDLen       = 0;
1081*48a54d36SAndroid Build Coastguard Worker 	mDNSPlatformMemZero(&rr->NATinfo, sizeof(rr->NATinfo));
1082*48a54d36SAndroid Build Coastguard Worker 	rr->SRVChanged = mDNSfalse;
1083*48a54d36SAndroid Build Coastguard Worker 	rr->mState = mergeState_Zero;
1084*48a54d36SAndroid Build Coastguard Worker 
1085*48a54d36SAndroid Build Coastguard Worker 	rr->namestorage.c[0]  = 0;		// MUST be set by client before calling mDNS_Register()
1086*48a54d36SAndroid Build Coastguard Worker 	}
1087*48a54d36SAndroid Build Coastguard Worker 
mDNS_SetupQuestion(DNSQuestion * const q,const mDNSInterfaceID InterfaceID,const domainname * const name,const mDNSu16 qtype,mDNSQuestionCallback * const callback,void * const context)1088*48a54d36SAndroid Build Coastguard Worker mDNSexport void mDNS_SetupQuestion(DNSQuestion *const q, const mDNSInterfaceID InterfaceID, const domainname *const name,
1089*48a54d36SAndroid Build Coastguard Worker                const mDNSu16 qtype, mDNSQuestionCallback *const callback, void *const context)
1090*48a54d36SAndroid Build Coastguard Worker 	{
1091*48a54d36SAndroid Build Coastguard Worker 	q->InterfaceID         = InterfaceID;
1092*48a54d36SAndroid Build Coastguard Worker 	q->Target              = zeroAddr;
1093*48a54d36SAndroid Build Coastguard Worker 	AssignDomainName(&q->qname, name);
1094*48a54d36SAndroid Build Coastguard Worker 	q->qtype               = qtype;
1095*48a54d36SAndroid Build Coastguard Worker 	q->qclass              = kDNSClass_IN;
1096*48a54d36SAndroid Build Coastguard Worker 	q->LongLived           = (qtype == kDNSType_PTR);
1097*48a54d36SAndroid Build Coastguard Worker 	q->ExpectUnique        = (qtype != kDNSType_PTR);
1098*48a54d36SAndroid Build Coastguard Worker 	q->ForceMCast          = mDNSfalse;
1099*48a54d36SAndroid Build Coastguard Worker 	q->ReturnIntermed      = mDNSfalse;
1100*48a54d36SAndroid Build Coastguard Worker 	q->SuppressUnusable    = mDNSfalse;
1101*48a54d36SAndroid Build Coastguard Worker 	q->SearchListIndex     = 0;
1102*48a54d36SAndroid Build Coastguard Worker 	q->AppendSearchDomains = 0;
1103*48a54d36SAndroid Build Coastguard Worker 	q->RetryWithSearchDomains = mDNSfalse;
1104*48a54d36SAndroid Build Coastguard Worker 	q->TimeoutQuestion     = 0;
1105*48a54d36SAndroid Build Coastguard Worker 	q->WakeOnResolve       = 0;
1106*48a54d36SAndroid Build Coastguard Worker 	q->qnameOrig           = mDNSNULL;
1107*48a54d36SAndroid Build Coastguard Worker 	q->QuestionCallback    = callback;
1108*48a54d36SAndroid Build Coastguard Worker 	q->QuestionContext     = context;
1109*48a54d36SAndroid Build Coastguard Worker 	}
1110*48a54d36SAndroid Build Coastguard Worker 
RDataHashValue(const ResourceRecord * const rr)1111*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 RDataHashValue(const ResourceRecord *const rr)
1112*48a54d36SAndroid Build Coastguard Worker 	{
1113*48a54d36SAndroid Build Coastguard Worker 	int len = rr->rdlength;
1114*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const rdb = (RDataBody2 *)rr->rdata->u.data;
1115*48a54d36SAndroid Build Coastguard Worker 	switch(rr->rrtype)
1116*48a54d36SAndroid Build Coastguard Worker 		{
1117*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:
1118*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:
1119*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:
1120*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DNAME: return DomainNameHashValue(&rdb->name);
1121*48a54d36SAndroid Build Coastguard Worker 
1122*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:   return rdb->soa.serial  +
1123*48a54d36SAndroid Build Coastguard Worker 									rdb->soa.refresh +
1124*48a54d36SAndroid Build Coastguard Worker 									rdb->soa.retry   +
1125*48a54d36SAndroid Build Coastguard Worker 									rdb->soa.expire  +
1126*48a54d36SAndroid Build Coastguard Worker 									rdb->soa.min     +
1127*48a54d36SAndroid Build Coastguard Worker 									DomainNameHashValue(&rdb->soa.mname) +
1128*48a54d36SAndroid Build Coastguard Worker 									DomainNameHashValue(&rdb->soa.rname);
1129*48a54d36SAndroid Build Coastguard Worker 
1130*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:
1131*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AFSDB:
1132*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RT:
1133*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_KX:	 return DomainNameHashValue(&rdb->mx.exchange);
1134*48a54d36SAndroid Build Coastguard Worker 
1135*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RP:	 return DomainNameHashValue(&rdb->rp.mbox)   + DomainNameHashValue(&rdb->rp.txt);
1136*48a54d36SAndroid Build Coastguard Worker 
1137*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PX:	 return DomainNameHashValue(&rdb->px.map822) + DomainNameHashValue(&rdb->px.mapx400);
1138*48a54d36SAndroid Build Coastguard Worker 
1139*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	 return DomainNameHashValue(&rdb->srv.target);
1140*48a54d36SAndroid Build Coastguard Worker 
1141*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:	 return 0;	// OPT is a pseudo-RR container structure; makes no sense to compare
1142*48a54d36SAndroid Build Coastguard Worker 
1143*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC:	 len = sizeof(rdataNSEC);	// Use in-memory length of 32, and fall through default checksum computation below
1144*48a54d36SAndroid Build Coastguard Worker 
1145*48a54d36SAndroid Build Coastguard Worker 		default:
1146*48a54d36SAndroid Build Coastguard Worker 			{
1147*48a54d36SAndroid Build Coastguard Worker 			mDNSu32 sum = 0;
1148*48a54d36SAndroid Build Coastguard Worker 			int i;
1149*48a54d36SAndroid Build Coastguard Worker 			for (i=0; i+1 < len; i+=2)
1150*48a54d36SAndroid Build Coastguard Worker 				{
1151*48a54d36SAndroid Build Coastguard Worker 				sum += (((mDNSu32)(rdb->data[i])) << 8) | rdb->data[i+1];
1152*48a54d36SAndroid Build Coastguard Worker 				sum = (sum<<3) | (sum>>29);
1153*48a54d36SAndroid Build Coastguard Worker 				}
1154*48a54d36SAndroid Build Coastguard Worker 			if (i < len)
1155*48a54d36SAndroid Build Coastguard Worker 				{
1156*48a54d36SAndroid Build Coastguard Worker 				sum += ((mDNSu32)(rdb->data[i])) << 8;
1157*48a54d36SAndroid Build Coastguard Worker 				}
1158*48a54d36SAndroid Build Coastguard Worker 			return(sum);
1159*48a54d36SAndroid Build Coastguard Worker 			}
1160*48a54d36SAndroid Build Coastguard Worker 		}
1161*48a54d36SAndroid Build Coastguard Worker 	}
1162*48a54d36SAndroid Build Coastguard Worker 
1163*48a54d36SAndroid Build Coastguard Worker // r1 has to be a full ResourceRecord including rrtype and rdlength
1164*48a54d36SAndroid Build Coastguard Worker // r2 is just a bare RDataBody, which MUST be the same rrtype and rdlength as r1
SameRDataBody(const ResourceRecord * const r1,const RDataBody * const r2,DomainNameComparisonFn * samename)1165*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool SameRDataBody(const ResourceRecord *const r1, const RDataBody *const r2, DomainNameComparisonFn *samename)
1166*48a54d36SAndroid Build Coastguard Worker 	{
1167*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const b1 = (RDataBody2 *)r1->rdata->u.data;
1168*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const b2 = (RDataBody2 *)r2;
1169*48a54d36SAndroid Build Coastguard Worker 	switch(r1->rrtype)
1170*48a54d36SAndroid Build Coastguard Worker 		{
1171*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:
1172*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:
1173*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:
1174*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DNAME:return(SameDomainName(&b1->name, &b2->name));
1175*48a54d36SAndroid Build Coastguard Worker 
1176*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:	return(mDNSBool)(  	b1->soa.serial   == b2->soa.serial             &&
1177*48a54d36SAndroid Build Coastguard Worker 												b1->soa.refresh  == b2->soa.refresh            &&
1178*48a54d36SAndroid Build Coastguard Worker 												b1->soa.retry    == b2->soa.retry              &&
1179*48a54d36SAndroid Build Coastguard Worker 												b1->soa.expire   == b2->soa.expire             &&
1180*48a54d36SAndroid Build Coastguard Worker 												b1->soa.min      == b2->soa.min                &&
1181*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->soa.mname, &b2->soa.mname) &&
1182*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->soa.rname, &b2->soa.rname));
1183*48a54d36SAndroid Build Coastguard Worker 
1184*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:
1185*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AFSDB:
1186*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RT:
1187*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_KX:	return(mDNSBool)(  	b1->mx.preference == b2->mx.preference &&
1188*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->mx.exchange, &b2->mx.exchange));
1189*48a54d36SAndroid Build Coastguard Worker 
1190*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RP:	return(mDNSBool)(  	samename(&b1->rp.mbox, &b2->rp.mbox) &&
1191*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->rp.txt,  &b2->rp.txt));
1192*48a54d36SAndroid Build Coastguard Worker 
1193*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PX:	return(mDNSBool)(  	b1->px.preference == b2->px.preference          &&
1194*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->px.map822,  &b2->px.map822) &&
1195*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->px.mapx400, &b2->px.mapx400));
1196*48a54d36SAndroid Build Coastguard Worker 
1197*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	return(mDNSBool)(  	b1->srv.priority == b2->srv.priority       &&
1198*48a54d36SAndroid Build Coastguard Worker 												b1->srv.weight   == b2->srv.weight         &&
1199*48a54d36SAndroid Build Coastguard Worker 												mDNSSameIPPort(b1->srv.port, b2->srv.port) &&
1200*48a54d36SAndroid Build Coastguard Worker 												samename(&b1->srv.target, &b2->srv.target));
1201*48a54d36SAndroid Build Coastguard Worker 
1202*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:	return mDNSfalse;	// OPT is a pseudo-RR container structure; makes no sense to compare
1203*48a54d36SAndroid Build Coastguard Worker 
1204*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: return(mDNSPlatformMemSame(b1->data, b2->data, sizeof(rdataNSEC)));
1205*48a54d36SAndroid Build Coastguard Worker 
1206*48a54d36SAndroid Build Coastguard Worker 		default:			return(mDNSPlatformMemSame(b1->data, b2->data, r1->rdlength));
1207*48a54d36SAndroid Build Coastguard Worker 		}
1208*48a54d36SAndroid Build Coastguard Worker 	}
1209*48a54d36SAndroid Build Coastguard Worker 
1210*48a54d36SAndroid Build Coastguard Worker // ResourceRecordAnswersQuestion returns mDNStrue if the given resource record is a valid answer to the given question.
1211*48a54d36SAndroid Build Coastguard Worker // SameNameRecordAnswersQuestion is the same, except it skips the expensive SameDomainName() call.
1212*48a54d36SAndroid Build Coastguard Worker // SameDomainName() is generally cheap when the names don't match, but expensive when they do match,
1213*48a54d36SAndroid Build Coastguard Worker // because it has to check all the way to the end of the names to be sure.
1214*48a54d36SAndroid Build Coastguard Worker // In cases where we know in advance that the names match it's especially advantageous to skip the
1215*48a54d36SAndroid Build Coastguard Worker // SameDomainName() call because that's precisely the time when it's most expensive and least useful.
1216*48a54d36SAndroid Build Coastguard Worker 
SameNameRecordAnswersQuestion(const ResourceRecord * const rr,const DNSQuestion * const q)1217*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool SameNameRecordAnswersQuestion(const ResourceRecord *const rr, const DNSQuestion *const q)
1218*48a54d36SAndroid Build Coastguard Worker 	{
1219*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly/P2P questions can be answered with AuthRecordAny in this function. LocalOnly/P2P records
1220*48a54d36SAndroid Build Coastguard Worker 	// are handled in LocalOnlyRecordAnswersQuestion
1221*48a54d36SAndroid Build Coastguard Worker 	if ((rr->InterfaceID == mDNSInterface_LocalOnly) || (rr->InterfaceID == mDNSInterface_P2P))
1222*48a54d36SAndroid Build Coastguard Worker 		{
1223*48a54d36SAndroid Build Coastguard Worker 		LogMsg("SameNameRecordAnswersQuestion: ERROR!! called with LocalOnly ResourceRecord %p, Question %p", rr->InterfaceID, q->InterfaceID);
1224*48a54d36SAndroid Build Coastguard Worker 		return mDNSfalse;
1225*48a54d36SAndroid Build Coastguard Worker 		}
1226*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID &&
1227*48a54d36SAndroid Build Coastguard Worker 		q ->InterfaceID && q->InterfaceID != mDNSInterface_LocalOnly &&
1228*48a54d36SAndroid Build Coastguard Worker 		rr->InterfaceID != q->InterfaceID) return(mDNSfalse);
1229*48a54d36SAndroid Build Coastguard Worker 
1230*48a54d36SAndroid Build Coastguard Worker 	// Resource record received via unicast, the DNSServer entries should match ?
1231*48a54d36SAndroid Build Coastguard Worker 	if (!rr->InterfaceID && rr->rDNSServer != q->qDNSServer) return(mDNSfalse);
1232*48a54d36SAndroid Build Coastguard Worker 
1233*48a54d36SAndroid Build Coastguard Worker 	// If ResourceRecord received via multicast, but question was unicast, then shouldn't use record to answer this question
1234*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID && !mDNSOpaque16IsZero(q->TargetQID)) return(mDNSfalse);
1235*48a54d36SAndroid Build Coastguard Worker 
1236*48a54d36SAndroid Build Coastguard Worker 	// RR type CNAME matches any query type. QTYPE ANY matches any RR type. QCLASS ANY matches any RR class.
1237*48a54d36SAndroid Build Coastguard Worker 	if (!RRTypeAnswersQuestionType(rr,q->qtype)) return(mDNSfalse);
1238*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass != q->qclass && q->qclass != kDNSQClass_ANY) return(mDNSfalse);
1239*48a54d36SAndroid Build Coastguard Worker 
1240*48a54d36SAndroid Build Coastguard Worker 	return(mDNStrue);
1241*48a54d36SAndroid Build Coastguard Worker 	}
1242*48a54d36SAndroid Build Coastguard Worker 
ResourceRecordAnswersQuestion(const ResourceRecord * const rr,const DNSQuestion * const q)1243*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool ResourceRecordAnswersQuestion(const ResourceRecord *const rr, const DNSQuestion *const q)
1244*48a54d36SAndroid Build Coastguard Worker 	{
1245*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly/P2P questions can be answered with AuthRecordAny in this function. LocalOnly/P2P records
1246*48a54d36SAndroid Build Coastguard Worker 	// are handled in LocalOnlyRecordAnswersQuestion
1247*48a54d36SAndroid Build Coastguard Worker 	if ((rr->InterfaceID == mDNSInterface_LocalOnly) || (rr->InterfaceID == mDNSInterface_P2P))
1248*48a54d36SAndroid Build Coastguard Worker 		{
1249*48a54d36SAndroid Build Coastguard Worker 		LogMsg("ResourceRecordAnswersQuestion: ERROR!! called with LocalOnly/P2P ResourceRecord %p, Question %p", rr->InterfaceID, q->InterfaceID);
1250*48a54d36SAndroid Build Coastguard Worker 		return mDNSfalse;
1251*48a54d36SAndroid Build Coastguard Worker 		}
1252*48a54d36SAndroid Build Coastguard Worker 
1253*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID &&
1254*48a54d36SAndroid Build Coastguard Worker 		q ->InterfaceID && q->InterfaceID != mDNSInterface_LocalOnly &&
1255*48a54d36SAndroid Build Coastguard Worker 		rr->InterfaceID != q->InterfaceID) return(mDNSfalse);
1256*48a54d36SAndroid Build Coastguard Worker 
1257*48a54d36SAndroid Build Coastguard Worker 	// Resource record received via unicast, the DNSServer entries should match ?
1258*48a54d36SAndroid Build Coastguard Worker 	if (!rr->InterfaceID && rr->rDNSServer != q->qDNSServer) return(mDNSfalse);
1259*48a54d36SAndroid Build Coastguard Worker 
1260*48a54d36SAndroid Build Coastguard Worker 	// If ResourceRecord received via multicast, but question was unicast, then shouldn't use record to answer this question.
1261*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID && !mDNSOpaque16IsZero(q->TargetQID)) return(mDNSfalse);
1262*48a54d36SAndroid Build Coastguard Worker 
1263*48a54d36SAndroid Build Coastguard Worker 	// RR type CNAME matches any query type. QTYPE ANY matches any RR type. QCLASS ANY matches any RR class.
1264*48a54d36SAndroid Build Coastguard Worker 	if (!RRTypeAnswersQuestionType(rr,q->qtype)) return(mDNSfalse);
1265*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass != q->qclass && q->qclass != kDNSQClass_ANY) return(mDNSfalse);
1266*48a54d36SAndroid Build Coastguard Worker 
1267*48a54d36SAndroid Build Coastguard Worker 	return(rr->namehash == q->qnamehash && SameDomainName(rr->name, &q->qname));
1268*48a54d36SAndroid Build Coastguard Worker 	}
1269*48a54d36SAndroid Build Coastguard Worker 
1270*48a54d36SAndroid Build Coastguard Worker // We have a separate function to handle LocalOnly AuthRecords because they can be created with
1271*48a54d36SAndroid Build Coastguard Worker // a valid InterfaceID (e.g., scoped /etc/hosts) and can be used to answer unicast questions unlike
1272*48a54d36SAndroid Build Coastguard Worker // multicast resource records (which has a valid InterfaceID) which can't be used to answer
1273*48a54d36SAndroid Build Coastguard Worker // unicast questions. ResourceRecordAnswersQuestion/SameNameRecordAnswersQuestion can't tell whether
1274*48a54d36SAndroid Build Coastguard Worker // a resource record is multicast or LocalOnly by just looking at the ResourceRecord because
1275*48a54d36SAndroid Build Coastguard Worker // LocalOnly records are truly identified by ARType in the AuthRecord.  As P2P and LocalOnly record
1276*48a54d36SAndroid Build Coastguard Worker // are kept in the same hash table, we use the same function to make it easy for the callers when
1277*48a54d36SAndroid Build Coastguard Worker // they walk the hash table to answer LocalOnly/P2P questions
1278*48a54d36SAndroid Build Coastguard Worker //
LocalOnlyRecordAnswersQuestion(AuthRecord * const ar,const DNSQuestion * const q)1279*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool LocalOnlyRecordAnswersQuestion(AuthRecord *const ar, const DNSQuestion *const q)
1280*48a54d36SAndroid Build Coastguard Worker 	{
1281*48a54d36SAndroid Build Coastguard Worker 	ResourceRecord *rr = &ar->resrec;
1282*48a54d36SAndroid Build Coastguard Worker 
1283*48a54d36SAndroid Build Coastguard Worker 	// mDNSInterface_Any questions can be answered with LocalOnly/P2P records in this function. AuthRecord_Any
1284*48a54d36SAndroid Build Coastguard Worker 	// records are handled in ResourceRecordAnswersQuestion/SameNameRecordAnswersQuestion
1285*48a54d36SAndroid Build Coastguard Worker 	if (RRAny(ar))
1286*48a54d36SAndroid Build Coastguard Worker 		{
1287*48a54d36SAndroid Build Coastguard Worker 		LogMsg("LocalOnlyRecordAnswersQuestion: ERROR!! called with regular AuthRecordAny %##s", rr->name->c);
1288*48a54d36SAndroid Build Coastguard Worker 		return mDNSfalse;
1289*48a54d36SAndroid Build Coastguard Worker 		}
1290*48a54d36SAndroid Build Coastguard Worker 
1291*48a54d36SAndroid Build Coastguard Worker 	// Questions with mDNSInterface_LocalOnly InterfaceID should be answered with all resource records that are
1292*48a54d36SAndroid Build Coastguard Worker 	// *local* to the machine. These include resource records that have InterfaceID set to mDNSInterface_LocalOnly,
1293*48a54d36SAndroid Build Coastguard Worker 	// mDNSInterface_Any and any other real InterfaceID. Hence, LocalOnly questions should not be checked against
1294*48a54d36SAndroid Build Coastguard Worker 	// the InterfaceID in the resource record.
1295*48a54d36SAndroid Build Coastguard Worker 	//
1296*48a54d36SAndroid Build Coastguard Worker 	// mDNSInterface_Unicast does not indicate any scope and hence treat them like mDNSInterface_Any.
1297*48a54d36SAndroid Build Coastguard Worker 
1298*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID &&
1299*48a54d36SAndroid Build Coastguard Worker 		q->InterfaceID && q->InterfaceID != mDNSInterface_LocalOnly && q->InterfaceID != mDNSInterface_Unicast &&
1300*48a54d36SAndroid Build Coastguard Worker 		rr->InterfaceID != q->InterfaceID) return(mDNSfalse);
1301*48a54d36SAndroid Build Coastguard Worker 
1302*48a54d36SAndroid Build Coastguard Worker 	// Entries in /etc/hosts are added as LocalOnly resource records. The LocalOnly resource records
1303*48a54d36SAndroid Build Coastguard Worker 	// may have a scope e.g., fe80::1%en0. The question may be scoped or not: the InterfaceID may be set
1304*48a54d36SAndroid Build Coastguard Worker 	// to mDNSInterface_Any, mDNSInterface_LocalOnly or a real InterfaceID (scoped).
1305*48a54d36SAndroid Build Coastguard Worker 	//
1306*48a54d36SAndroid Build Coastguard Worker 	// 1) Question: Any, LocalOnly Record: no scope. This question should be answered with this record.
1307*48a54d36SAndroid Build Coastguard Worker 	//
1308*48a54d36SAndroid Build Coastguard Worker 	// 2) Question: Any, LocalOnly Record: scoped.  This question should be answered with the record because
1309*48a54d36SAndroid Build Coastguard Worker 	//    traditionally applications never specify scope e.g., getaddrinfo, but need to be able
1310*48a54d36SAndroid Build Coastguard Worker 	//    to get to /etc/hosts entries.
1311*48a54d36SAndroid Build Coastguard Worker 	//
1312*48a54d36SAndroid Build Coastguard Worker 	// 3) Question: Scoped (LocalOnly or InterfaceID), LocalOnly Record: no scope. This is the inverse of (2).
1313*48a54d36SAndroid Build Coastguard Worker 	//    If we register a LocalOnly record, we need to answer a LocalOnly question. If the /etc/hosts has a
1314*48a54d36SAndroid Build Coastguard Worker 	//    non scoped entry, it may not make sense to answer a scoped question. But we can't tell these two
1315*48a54d36SAndroid Build Coastguard Worker 	//    cases apart. As we currently answer LocalOnly question with LocalOnly record, we continue to do so.
1316*48a54d36SAndroid Build Coastguard Worker 	//
1317*48a54d36SAndroid Build Coastguard Worker 	// 4) Question: Scoped (LocalOnly or InterfaceID), LocalOnly Record: scoped. LocalOnly questions should be
1318*48a54d36SAndroid Build Coastguard Worker 	//    answered with any resource record where as if it has a valid InterfaceID, the scope should match.
1319*48a54d36SAndroid Build Coastguard Worker 	//
1320*48a54d36SAndroid Build Coastguard Worker 	// (1) and (2) is bypassed because we check for a non-NULL InterfaceID above. For (3), the InterfaceID is NULL
1321*48a54d36SAndroid Build Coastguard Worker 	// and hence bypassed above. For (4) we bypassed LocalOnly questions and checked the scope of the record
1322*48a54d36SAndroid Build Coastguard Worker 	// against the question.
1323*48a54d36SAndroid Build Coastguard Worker 	//
1324*48a54d36SAndroid Build Coastguard Worker 	// For P2P, InterfaceIDs of the question and the record should match.
1325*48a54d36SAndroid Build Coastguard Worker 
1326*48a54d36SAndroid Build Coastguard Worker 	// If ResourceRecord received via multicast, but question was unicast, then shouldn't use record to answer this question.
1327*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly authoritative answers are exempt. LocalOnly authoritative answers are used for /etc/host entries.
1328*48a54d36SAndroid Build Coastguard Worker 	// We don't want a local process to be able to create a fake LocalOnly address record for "www.bigbank.com" which would then
1329*48a54d36SAndroid Build Coastguard Worker 	// cause other applications (e.g. Safari) to connect to the wrong address. The rpc to register records filters out records
1330*48a54d36SAndroid Build Coastguard Worker 	// with names that don't end in local and have mDNSInterface_LocalOnly set.
1331*48a54d36SAndroid Build Coastguard Worker 	//
1332*48a54d36SAndroid Build Coastguard Worker 	// Note: The check is bypassed for LocalOnly and for P2P it is not needed as only .local records are registered and for
1333*48a54d36SAndroid Build Coastguard Worker 	// a question to match its names, it also has to end in .local and that question can't be a unicast question (See
1334*48a54d36SAndroid Build Coastguard Worker 	// Question_uDNS macro and its usage). As P2P does not enforce .local only registrations we still make this check
1335*48a54d36SAndroid Build Coastguard Worker 	// and also makes it future proof.
1336*48a54d36SAndroid Build Coastguard Worker 
1337*48a54d36SAndroid Build Coastguard Worker 	if (ar->ARType != AuthRecordLocalOnly && rr->InterfaceID && !mDNSOpaque16IsZero(q->TargetQID)) return(mDNSfalse);
1338*48a54d36SAndroid Build Coastguard Worker 
1339*48a54d36SAndroid Build Coastguard Worker 	// RR type CNAME matches any query type. QTYPE ANY matches any RR type. QCLASS ANY matches any RR class.
1340*48a54d36SAndroid Build Coastguard Worker 	if (!RRTypeAnswersQuestionType(rr,q->qtype)) return(mDNSfalse);
1341*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass != q->qclass && q->qclass != kDNSQClass_ANY) return(mDNSfalse);
1342*48a54d36SAndroid Build Coastguard Worker 
1343*48a54d36SAndroid Build Coastguard Worker 	return(rr->namehash == q->qnamehash && SameDomainName(rr->name, &q->qname));
1344*48a54d36SAndroid Build Coastguard Worker 	}
1345*48a54d36SAndroid Build Coastguard Worker 
AnyTypeRecordAnswersQuestion(const ResourceRecord * const rr,const DNSQuestion * const q)1346*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool AnyTypeRecordAnswersQuestion(const ResourceRecord *const rr, const DNSQuestion *const q)
1347*48a54d36SAndroid Build Coastguard Worker 	{
1348*48a54d36SAndroid Build Coastguard Worker 	// LocalOnly/P2P questions can be answered with AuthRecordAny in this function. LocalOnly/P2P records
1349*48a54d36SAndroid Build Coastguard Worker 	// are handled in LocalOnlyRecordAnswersQuestion
1350*48a54d36SAndroid Build Coastguard Worker 	if ((rr->InterfaceID == mDNSInterface_LocalOnly) || (rr->InterfaceID == mDNSInterface_P2P))
1351*48a54d36SAndroid Build Coastguard Worker 		{
1352*48a54d36SAndroid Build Coastguard Worker 		LogMsg("AnyTypeRecordAnswersQuestion: ERROR!! called with LocalOnly ResourceRecord %p, Question %p", rr->InterfaceID, q->InterfaceID);
1353*48a54d36SAndroid Build Coastguard Worker 		return mDNSfalse;
1354*48a54d36SAndroid Build Coastguard Worker 		}
1355*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID &&
1356*48a54d36SAndroid Build Coastguard Worker 		q ->InterfaceID && q->InterfaceID != mDNSInterface_LocalOnly &&
1357*48a54d36SAndroid Build Coastguard Worker 		rr->InterfaceID != q->InterfaceID) return(mDNSfalse);
1358*48a54d36SAndroid Build Coastguard Worker 
1359*48a54d36SAndroid Build Coastguard Worker 	// Resource record received via unicast, the DNSServer entries should match ?
1360*48a54d36SAndroid Build Coastguard Worker 	// Note that Auth Records are normally setup with NULL InterfaceID and
1361*48a54d36SAndroid Build Coastguard Worker 	// both the DNSServers are assumed to be NULL in that case
1362*48a54d36SAndroid Build Coastguard Worker 	if (!rr->InterfaceID && rr->rDNSServer != q->qDNSServer) return(mDNSfalse);
1363*48a54d36SAndroid Build Coastguard Worker 
1364*48a54d36SAndroid Build Coastguard Worker 	// If ResourceRecord received via multicast, but question was unicast, then shouldn't use record to answer this question
1365*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID && !mDNSOpaque16IsZero(q->TargetQID)) return(mDNSfalse);
1366*48a54d36SAndroid Build Coastguard Worker 
1367*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass != q->qclass && q->qclass != kDNSQClass_ANY) return(mDNSfalse);
1368*48a54d36SAndroid Build Coastguard Worker 
1369*48a54d36SAndroid Build Coastguard Worker 	return(rr->namehash == q->qnamehash && SameDomainName(rr->name, &q->qname));
1370*48a54d36SAndroid Build Coastguard Worker 	}
1371*48a54d36SAndroid Build Coastguard Worker 
1372*48a54d36SAndroid Build Coastguard Worker // This is called with both unicast resource record and multicast resource record. The question that
1373*48a54d36SAndroid Build Coastguard Worker // received the unicast response could be the regular unicast response from a DNS server or a response
1374*48a54d36SAndroid Build Coastguard Worker // to a mDNS QU query. The main reason we need this function is that we can't compare DNSServers between the
1375*48a54d36SAndroid Build Coastguard Worker // question and the resource record because the resource record is not completely initialized in
1376*48a54d36SAndroid Build Coastguard Worker // mDNSCoreReceiveResponse when this function is called.
ResourceRecordAnswersUnicastResponse(const ResourceRecord * const rr,const DNSQuestion * const q)1377*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool ResourceRecordAnswersUnicastResponse(const ResourceRecord *const rr, const DNSQuestion *const q)
1378*48a54d36SAndroid Build Coastguard Worker 	{
1379*48a54d36SAndroid Build Coastguard Worker 	// For resource records created using multicast, the InterfaceIDs have to match
1380*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID &&
1381*48a54d36SAndroid Build Coastguard Worker 		q->InterfaceID && rr->InterfaceID != q->InterfaceID) return(mDNSfalse);
1382*48a54d36SAndroid Build Coastguard Worker 
1383*48a54d36SAndroid Build Coastguard Worker 	// If ResourceRecord received via multicast, but question was unicast, then shouldn't use record to answer this question.
1384*48a54d36SAndroid Build Coastguard Worker 	if (rr->InterfaceID && !mDNSOpaque16IsZero(q->TargetQID)) return(mDNSfalse);
1385*48a54d36SAndroid Build Coastguard Worker 
1386*48a54d36SAndroid Build Coastguard Worker 	// RR type CNAME matches any query type. QTYPE ANY matches any RR type. QCLASS ANY matches any RR class.
1387*48a54d36SAndroid Build Coastguard Worker 	if (!RRTypeAnswersQuestionType(rr,q->qtype)) return(mDNSfalse);
1388*48a54d36SAndroid Build Coastguard Worker 
1389*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass != q->qclass && q->qclass != kDNSQClass_ANY) return(mDNSfalse);
1390*48a54d36SAndroid Build Coastguard Worker 
1391*48a54d36SAndroid Build Coastguard Worker 	return(rr->namehash == q->qnamehash && SameDomainName(rr->name, &q->qname));
1392*48a54d36SAndroid Build Coastguard Worker 	}
1393*48a54d36SAndroid Build Coastguard Worker 
GetRDLength(const ResourceRecord * const rr,mDNSBool estimate)1394*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu16 GetRDLength(const ResourceRecord *const rr, mDNSBool estimate)
1395*48a54d36SAndroid Build Coastguard Worker 	{
1396*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const rd = (RDataBody2 *)rr->rdata->u.data;
1397*48a54d36SAndroid Build Coastguard Worker 	const domainname *const name = estimate ? rr->name : mDNSNULL;
1398*48a54d36SAndroid Build Coastguard Worker 	if (rr->rrclass == kDNSQClass_ANY) return(rr->rdlength);	// Used in update packets to mean "Delete An RRset" (RFC 2136)
1399*48a54d36SAndroid Build Coastguard Worker 	else switch (rr->rrtype)
1400*48a54d36SAndroid Build Coastguard Worker 		{
1401*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:	return(sizeof(rd->ipv4));
1402*48a54d36SAndroid Build Coastguard Worker 
1403*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:
1404*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:
1405*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:
1406*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DNAME:return(CompressedDomainNameLength(&rd->name, name));
1407*48a54d36SAndroid Build Coastguard Worker 
1408*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:  return(mDNSu16)(CompressedDomainNameLength(&rd->soa.mname, name) +
1409*48a54d36SAndroid Build Coastguard Worker 											CompressedDomainNameLength(&rd->soa.rname, name) +
1410*48a54d36SAndroid Build Coastguard Worker 											5 * sizeof(mDNSOpaque32));
1411*48a54d36SAndroid Build Coastguard Worker 
1412*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NULL:
1413*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TSIG:
1414*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:
1415*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_X25:
1416*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_ISDN:
1417*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_LOC:
1418*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DHCID:return(rr->rdlength); // Not self-describing, so have to just trust rdlength
1419*48a54d36SAndroid Build Coastguard Worker 
1420*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:return(mDNSu16)(2 + (int)rd->data[0] + (int)rd->data[1 + (int)rd->data[0]]);
1421*48a54d36SAndroid Build Coastguard Worker 
1422*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:
1423*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AFSDB:
1424*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RT:
1425*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_KX:	return(mDNSu16)(2 + CompressedDomainNameLength(&rd->mx.exchange, name));
1426*48a54d36SAndroid Build Coastguard Worker 
1427*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RP:	return(mDNSu16)(CompressedDomainNameLength(&rd->rp.mbox, name) +
1428*48a54d36SAndroid Build Coastguard Worker 											CompressedDomainNameLength(&rd->rp.txt, name));
1429*48a54d36SAndroid Build Coastguard Worker 
1430*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PX:	return(mDNSu16)(2 + CompressedDomainNameLength(&rd->px.map822, name) +
1431*48a54d36SAndroid Build Coastguard Worker 												CompressedDomainNameLength(&rd->px.mapx400, name));
1432*48a54d36SAndroid Build Coastguard Worker 
1433*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA:	return(sizeof(rd->ipv6));
1434*48a54d36SAndroid Build Coastguard Worker 
1435*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	return(mDNSu16)(6 + CompressedDomainNameLength(&rd->srv.target, name));
1436*48a54d36SAndroid Build Coastguard Worker 
1437*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:  return(rr->rdlength);
1438*48a54d36SAndroid Build Coastguard Worker 
1439*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: {
1440*48a54d36SAndroid Build Coastguard Worker 							int i;
1441*48a54d36SAndroid Build Coastguard Worker 							for (i=sizeof(rdataNSEC); i>0; i--) if (rd->nsec.bitmap[i-1]) break;
1442*48a54d36SAndroid Build Coastguard Worker 							// For our simplified use of NSEC synthetic records:
1443*48a54d36SAndroid Build Coastguard Worker 							// nextname is always the record's own name,
1444*48a54d36SAndroid Build Coastguard Worker 							// and if we have at least one record type that exists,
1445*48a54d36SAndroid Build Coastguard Worker 							//  - the block number is always 0,
1446*48a54d36SAndroid Build Coastguard Worker 							//  - the count byte is a value in the range 1-32,
1447*48a54d36SAndroid Build Coastguard Worker 							//  - followed by the 1-32 data bytes
1448*48a54d36SAndroid Build Coastguard Worker 							return(mDNSu16)((estimate ? 2 : DomainNameLength(rr->name)) + (i ? (2 + i) : 0));
1449*48a54d36SAndroid Build Coastguard Worker 							}
1450*48a54d36SAndroid Build Coastguard Worker 
1451*48a54d36SAndroid Build Coastguard Worker 		default:			debugf("Warning! Don't know how to get length of resource type %d", rr->rrtype);
1452*48a54d36SAndroid Build Coastguard Worker 							return(rr->rdlength);
1453*48a54d36SAndroid Build Coastguard Worker 		}
1454*48a54d36SAndroid Build Coastguard Worker 	}
1455*48a54d36SAndroid Build Coastguard Worker 
1456*48a54d36SAndroid Build Coastguard Worker // When a local client registers (or updates) a record, we use this routine to do some simple validation checks
1457*48a54d36SAndroid Build Coastguard Worker // to help reduce the risk of bogus malformed data on the network
ValidateRData(const mDNSu16 rrtype,const mDNSu16 rdlength,const RData * const rd)1458*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSBool ValidateRData(const mDNSu16 rrtype, const mDNSu16 rdlength, const RData *const rd)
1459*48a54d36SAndroid Build Coastguard Worker 	{
1460*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 len;
1461*48a54d36SAndroid Build Coastguard Worker 
1462*48a54d36SAndroid Build Coastguard Worker 	switch(rrtype)
1463*48a54d36SAndroid Build Coastguard Worker 		{
1464*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:	return(rdlength == sizeof(mDNSv4Addr));
1465*48a54d36SAndroid Build Coastguard Worker 
1466*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:	// Same as PTR
1467*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MD:	// Same as PTR
1468*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MF:	// Same as PTR
1469*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:// Same as PTR
1470*48a54d36SAndroid Build Coastguard Worker 		//case kDNSType_SOA not checked
1471*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MB:	// Same as PTR
1472*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MG:	// Same as PTR
1473*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MR:	// Same as PTR
1474*48a54d36SAndroid Build Coastguard Worker 		//case kDNSType_NULL not checked (no specified format, so always valid)
1475*48a54d36SAndroid Build Coastguard Worker 		//case kDNSType_WKS not checked
1476*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:	len = DomainNameLengthLimit(&rd->u.name, rd->u.data + rdlength);
1477*48a54d36SAndroid Build Coastguard Worker 							return(len <= MAX_DOMAIN_NAME && rdlength == len);
1478*48a54d36SAndroid Build Coastguard Worker 
1479*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:// Same as TXT (roughly)
1480*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MINFO:// Same as TXT (roughly)
1481*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:  if (!rdlength) return(mDNSfalse); // TXT record has to be at least one byte (RFC 1035)
1482*48a54d36SAndroid Build Coastguard Worker 							{
1483*48a54d36SAndroid Build Coastguard Worker 							const mDNSu8 *ptr = rd->u.txt.c;
1484*48a54d36SAndroid Build Coastguard Worker 							const mDNSu8 *end = rd->u.txt.c + rdlength;
1485*48a54d36SAndroid Build Coastguard Worker 							while (ptr < end) ptr += 1 + ptr[0];
1486*48a54d36SAndroid Build Coastguard Worker 							return (ptr == end);
1487*48a54d36SAndroid Build Coastguard Worker 							}
1488*48a54d36SAndroid Build Coastguard Worker 
1489*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA:	return(rdlength == sizeof(mDNSv6Addr));
1490*48a54d36SAndroid Build Coastguard Worker 
1491*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:   // Must be at least two-byte preference, plus domainname
1492*48a54d36SAndroid Build Coastguard Worker 							// Call to DomainNameLengthLimit() implicitly enforces both requirements for us
1493*48a54d36SAndroid Build Coastguard Worker 							len = DomainNameLengthLimit(&rd->u.mx.exchange, rd->u.data + rdlength);
1494*48a54d36SAndroid Build Coastguard Worker 							return(len <= MAX_DOMAIN_NAME && rdlength == 2+len);
1495*48a54d36SAndroid Build Coastguard Worker 
1496*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	// Must be at least priority+weight+port, plus domainname
1497*48a54d36SAndroid Build Coastguard Worker 							// Call to DomainNameLengthLimit() implicitly enforces both requirements for us
1498*48a54d36SAndroid Build Coastguard Worker 							len = DomainNameLengthLimit(&rd->u.srv.target, rd->u.data + rdlength);
1499*48a54d36SAndroid Build Coastguard Worker 							return(len <= MAX_DOMAIN_NAME && rdlength == 6+len);
1500*48a54d36SAndroid Build Coastguard Worker 
1501*48a54d36SAndroid Build Coastguard Worker 		//case kDNSType_NSEC not checked
1502*48a54d36SAndroid Build Coastguard Worker 
1503*48a54d36SAndroid Build Coastguard Worker 		default:			return(mDNStrue);	// Allow all other types without checking
1504*48a54d36SAndroid Build Coastguard Worker 		}
1505*48a54d36SAndroid Build Coastguard Worker 	}
1506*48a54d36SAndroid Build Coastguard Worker 
1507*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
1508*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
1509*48a54d36SAndroid Build Coastguard Worker #pragma mark -
1510*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNS Message Creation Functions
1511*48a54d36SAndroid Build Coastguard Worker #endif
1512*48a54d36SAndroid Build Coastguard Worker 
InitializeDNSMessage(DNSMessageHeader * h,mDNSOpaque16 id,mDNSOpaque16 flags)1513*48a54d36SAndroid Build Coastguard Worker mDNSexport void InitializeDNSMessage(DNSMessageHeader *h, mDNSOpaque16 id, mDNSOpaque16 flags)
1514*48a54d36SAndroid Build Coastguard Worker 	{
1515*48a54d36SAndroid Build Coastguard Worker 	h->id             = id;
1516*48a54d36SAndroid Build Coastguard Worker 	h->flags          = flags;
1517*48a54d36SAndroid Build Coastguard Worker 	h->numQuestions   = 0;
1518*48a54d36SAndroid Build Coastguard Worker 	h->numAnswers     = 0;
1519*48a54d36SAndroid Build Coastguard Worker 	h->numAuthorities = 0;
1520*48a54d36SAndroid Build Coastguard Worker 	h->numAdditionals = 0;
1521*48a54d36SAndroid Build Coastguard Worker 	}
1522*48a54d36SAndroid Build Coastguard Worker 
FindCompressionPointer(const mDNSu8 * const base,const mDNSu8 * const end,const mDNSu8 * const domname)1523*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *FindCompressionPointer(const mDNSu8 *const base, const mDNSu8 *const end, const mDNSu8 *const domname)
1524*48a54d36SAndroid Build Coastguard Worker 	{
1525*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *result = end - *domname - 1;
1526*48a54d36SAndroid Build Coastguard Worker 
1527*48a54d36SAndroid Build Coastguard Worker 	if (*domname == 0) return(mDNSNULL);	// There's no point trying to match just the root label
1528*48a54d36SAndroid Build Coastguard Worker 
1529*48a54d36SAndroid Build Coastguard Worker 	// This loop examines each possible starting position in packet, starting end of the packet and working backwards
1530*48a54d36SAndroid Build Coastguard Worker 	while (result >= base)
1531*48a54d36SAndroid Build Coastguard Worker 		{
1532*48a54d36SAndroid Build Coastguard Worker 		// If the length byte and first character of the label match, then check further to see
1533*48a54d36SAndroid Build Coastguard Worker 		// if this location in the packet will yield a useful name compression pointer.
1534*48a54d36SAndroid Build Coastguard Worker 		if (result[0] == domname[0] && result[1] == domname[1])
1535*48a54d36SAndroid Build Coastguard Worker 			{
1536*48a54d36SAndroid Build Coastguard Worker 			const mDNSu8 *name = domname;
1537*48a54d36SAndroid Build Coastguard Worker 			const mDNSu8 *targ = result;
1538*48a54d36SAndroid Build Coastguard Worker 			while (targ + *name < end)
1539*48a54d36SAndroid Build Coastguard Worker 				{
1540*48a54d36SAndroid Build Coastguard Worker 				// First see if this label matches
1541*48a54d36SAndroid Build Coastguard Worker 				int i;
1542*48a54d36SAndroid Build Coastguard Worker 				const mDNSu8 *pointertarget;
1543*48a54d36SAndroid Build Coastguard Worker 				for (i=0; i <= *name; i++) if (targ[i] != name[i]) break;
1544*48a54d36SAndroid Build Coastguard Worker 				if (i <= *name) break;							// If label did not match, bail out
1545*48a54d36SAndroid Build Coastguard Worker 				targ += 1 + *name;								// Else, did match, so advance target pointer
1546*48a54d36SAndroid Build Coastguard Worker 				name += 1 + *name;								// and proceed to check next label
1547*48a54d36SAndroid Build Coastguard Worker 				if (*name == 0 && *targ == 0) return(result);	// If no more labels, we found a match!
1548*48a54d36SAndroid Build Coastguard Worker 				if (*name == 0) break;							// If no more labels to match, we failed, so bail out
1549*48a54d36SAndroid Build Coastguard Worker 
1550*48a54d36SAndroid Build Coastguard Worker 				// The label matched, so now follow the pointer (if appropriate) and then see if the next label matches
1551*48a54d36SAndroid Build Coastguard Worker 				if (targ[0] < 0x40) continue;					// If length value, continue to check next label
1552*48a54d36SAndroid Build Coastguard Worker 				if (targ[0] < 0xC0) break;						// If 40-BF, not valid
1553*48a54d36SAndroid Build Coastguard Worker 				if (targ+1 >= end) break;						// Second byte not present!
1554*48a54d36SAndroid Build Coastguard Worker 				pointertarget = base + (((mDNSu16)(targ[0] & 0x3F)) << 8) + targ[1];
1555*48a54d36SAndroid Build Coastguard Worker 				if (targ < pointertarget) break;				// Pointertarget must point *backwards* in the packet
1556*48a54d36SAndroid Build Coastguard Worker 				if (pointertarget[0] >= 0x40) break;			// Pointertarget must point to a valid length byte
1557*48a54d36SAndroid Build Coastguard Worker 				targ = pointertarget;
1558*48a54d36SAndroid Build Coastguard Worker 				}
1559*48a54d36SAndroid Build Coastguard Worker 			}
1560*48a54d36SAndroid Build Coastguard Worker 		result--;	// We failed to match at this search position, so back up the tentative result pointer and try again
1561*48a54d36SAndroid Build Coastguard Worker 		}
1562*48a54d36SAndroid Build Coastguard Worker 	return(mDNSNULL);
1563*48a54d36SAndroid Build Coastguard Worker 	}
1564*48a54d36SAndroid Build Coastguard Worker 
1565*48a54d36SAndroid Build Coastguard Worker // Put a string of dot-separated labels as length-prefixed labels
1566*48a54d36SAndroid Build Coastguard Worker // domainname is a fully-qualified name (i.e. assumed to be ending in a dot, even if it doesn't)
1567*48a54d36SAndroid Build Coastguard Worker // msg points to the message we're building (pass mDNSNULL if we don't want to use compression pointers)
1568*48a54d36SAndroid Build Coastguard Worker // end points to the end of the message so far
1569*48a54d36SAndroid Build Coastguard Worker // ptr points to where we want to put the name
1570*48a54d36SAndroid Build Coastguard Worker // limit points to one byte past the end of the buffer that we must not overrun
1571*48a54d36SAndroid Build Coastguard Worker // domainname is the name to put
putDomainNameAsLabels(const DNSMessage * const msg,mDNSu8 * ptr,const mDNSu8 * const limit,const domainname * const name)1572*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putDomainNameAsLabels(const DNSMessage *const msg,
1573*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *ptr, const mDNSu8 *const limit, const domainname *const name)
1574*48a54d36SAndroid Build Coastguard Worker 	{
1575*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const base        = (const mDNSu8 *)msg;
1576*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      np          = name->c;
1577*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const max         = name->c + MAX_DOMAIN_NAME;	// Maximum that's valid
1578*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *      pointer     = mDNSNULL;
1579*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const searchlimit = ptr;
1580*48a54d36SAndroid Build Coastguard Worker 
1581*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { LogMsg("putDomainNameAsLabels %##s ptr is null", name->c); return(mDNSNULL); }
1582*48a54d36SAndroid Build Coastguard Worker 
1583*48a54d36SAndroid Build Coastguard Worker 	if (!*np)		// If just writing one-byte root label, make sure we have space for that
1584*48a54d36SAndroid Build Coastguard Worker 		{
1585*48a54d36SAndroid Build Coastguard Worker 		if (ptr >= limit) return(mDNSNULL);
1586*48a54d36SAndroid Build Coastguard Worker 		}
1587*48a54d36SAndroid Build Coastguard Worker 	else			// else, loop through writing labels and/or a compression offset
1588*48a54d36SAndroid Build Coastguard Worker 		{
1589*48a54d36SAndroid Build Coastguard Worker 		do	{
1590*48a54d36SAndroid Build Coastguard Worker 			if (*np > MAX_DOMAIN_LABEL)
1591*48a54d36SAndroid Build Coastguard Worker 				{ LogMsg("Malformed domain name %##s (label more than 63 bytes)", name->c); return(mDNSNULL); }
1592*48a54d36SAndroid Build Coastguard Worker 
1593*48a54d36SAndroid Build Coastguard Worker 			// This check correctly allows for the final trailing root label:
1594*48a54d36SAndroid Build Coastguard Worker 			// e.g.
1595*48a54d36SAndroid Build Coastguard Worker 			// Suppose our domain name is exactly 256 bytes long, including the final trailing root label.
1596*48a54d36SAndroid Build Coastguard Worker 			// Suppose np is now at name->c[249], and we're about to write our last non-null label ("local").
1597*48a54d36SAndroid Build Coastguard Worker 			// We know that max will be at name->c[256]
1598*48a54d36SAndroid Build Coastguard Worker 			// That means that np + 1 + 5 == max - 1, so we (just) pass the "if" test below, write our
1599*48a54d36SAndroid Build Coastguard Worker 			// six bytes, then exit the loop, write the final terminating root label, and the domain
1600*48a54d36SAndroid Build Coastguard Worker 			// name we've written is exactly 256 bytes long, exactly at the correct legal limit.
1601*48a54d36SAndroid Build Coastguard Worker 			// If the name is one byte longer, then we fail the "if" test below, and correctly bail out.
1602*48a54d36SAndroid Build Coastguard Worker 			if (np + 1 + *np >= max)
1603*48a54d36SAndroid Build Coastguard Worker 				{ LogMsg("Malformed domain name %##s (more than 256 bytes)", name->c); return(mDNSNULL); }
1604*48a54d36SAndroid Build Coastguard Worker 
1605*48a54d36SAndroid Build Coastguard Worker 			if (base) pointer = FindCompressionPointer(base, searchlimit, np);
1606*48a54d36SAndroid Build Coastguard Worker 			if (pointer)					// Use a compression pointer if we can
1607*48a54d36SAndroid Build Coastguard Worker 				{
1608*48a54d36SAndroid Build Coastguard Worker 				const mDNSu16 offset = (mDNSu16)(pointer - base);
1609*48a54d36SAndroid Build Coastguard Worker 				if (ptr+2 > limit) return(mDNSNULL);	// If we don't have two bytes of space left, give up
1610*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = (mDNSu8)(0xC0 | (offset >> 8));
1611*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = (mDNSu8)(        offset &  0xFF);
1612*48a54d36SAndroid Build Coastguard Worker 				return(ptr);
1613*48a54d36SAndroid Build Coastguard Worker 				}
1614*48a54d36SAndroid Build Coastguard Worker 			else							// Else copy one label and try again
1615*48a54d36SAndroid Build Coastguard Worker 				{
1616*48a54d36SAndroid Build Coastguard Worker 				int i;
1617*48a54d36SAndroid Build Coastguard Worker 				mDNSu8 len = *np++;
1618*48a54d36SAndroid Build Coastguard Worker 				// If we don't at least have enough space for this label *plus* a terminating zero on the end, give up
1619*48a54d36SAndroid Build Coastguard Worker 				if (ptr + 1 + len >= limit) return(mDNSNULL);
1620*48a54d36SAndroid Build Coastguard Worker 				*ptr++ = len;
1621*48a54d36SAndroid Build Coastguard Worker 				for (i=0; i<len; i++) *ptr++ = *np++;
1622*48a54d36SAndroid Build Coastguard Worker 				}
1623*48a54d36SAndroid Build Coastguard Worker 			} while (*np);					// While we've got characters remaining in the name, continue
1624*48a54d36SAndroid Build Coastguard Worker 		}
1625*48a54d36SAndroid Build Coastguard Worker 
1626*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = 0;		// Put the final root label
1627*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
1628*48a54d36SAndroid Build Coastguard Worker 	}
1629*48a54d36SAndroid Build Coastguard Worker 
putVal16(mDNSu8 * ptr,mDNSu16 val)1630*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu8 *putVal16(mDNSu8 *ptr, mDNSu16 val)
1631*48a54d36SAndroid Build Coastguard Worker 	{
1632*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)((val >> 8 ) & 0xFF);
1633*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)((val      ) & 0xFF);
1634*48a54d36SAndroid Build Coastguard Worker 	return ptr + sizeof(mDNSOpaque16);
1635*48a54d36SAndroid Build Coastguard Worker 	}
1636*48a54d36SAndroid Build Coastguard Worker 
putVal32(mDNSu8 * ptr,mDNSu32 val)1637*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu8 *putVal32(mDNSu8 *ptr, mDNSu32 val)
1638*48a54d36SAndroid Build Coastguard Worker 	{
1639*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)((val >> 24) & 0xFF);
1640*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)((val >> 16) & 0xFF);
1641*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)((val >>  8) & 0xFF);
1642*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)((val      ) & 0xFF);
1643*48a54d36SAndroid Build Coastguard Worker 	return ptr + sizeof(mDNSu32);
1644*48a54d36SAndroid Build Coastguard Worker 	}
1645*48a54d36SAndroid Build Coastguard Worker 
1646*48a54d36SAndroid Build Coastguard Worker // msg points to the message we're building (pass mDNSNULL if we don't want to use compression pointers)
putRData(const DNSMessage * const msg,mDNSu8 * ptr,const mDNSu8 * const limit,const ResourceRecord * const rr)1647*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putRData(const DNSMessage *const msg, mDNSu8 *ptr, const mDNSu8 *const limit, const ResourceRecord *const rr)
1648*48a54d36SAndroid Build Coastguard Worker 	{
1649*48a54d36SAndroid Build Coastguard Worker 	const RDataBody2 *const rdb = (RDataBody2 *)rr->rdata->u.data;
1650*48a54d36SAndroid Build Coastguard Worker 	switch (rr->rrtype)
1651*48a54d36SAndroid Build Coastguard Worker 		{
1652*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:	if (rr->rdlength != 4)
1653*48a54d36SAndroid Build Coastguard Worker 								{ debugf("putRData: Illegal length %d for kDNSType_A", rr->rdlength); return(mDNSNULL); }
1654*48a54d36SAndroid Build Coastguard Worker 							if (ptr + 4 > limit) return(mDNSNULL);
1655*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->ipv4.b[0];
1656*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->ipv4.b[1];
1657*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->ipv4.b[2];
1658*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->ipv4.b[3];
1659*48a54d36SAndroid Build Coastguard Worker 							return(ptr);
1660*48a54d36SAndroid Build Coastguard Worker 
1661*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:
1662*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:
1663*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:
1664*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DNAME:return(putDomainNameAsLabels(msg, ptr, limit, &rdb->name));
1665*48a54d36SAndroid Build Coastguard Worker 
1666*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:  ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->soa.mname);
1667*48a54d36SAndroid Build Coastguard Worker 							if (!ptr) return(mDNSNULL);
1668*48a54d36SAndroid Build Coastguard Worker 							ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->soa.rname);
1669*48a54d36SAndroid Build Coastguard Worker 							if (!ptr || ptr + 20 > limit) return(mDNSNULL);
1670*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal32(ptr, rdb->soa.serial);
1671*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal32(ptr, rdb->soa.refresh);
1672*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal32(ptr, rdb->soa.retry);
1673*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal32(ptr, rdb->soa.expire);
1674*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal32(ptr, rdb->soa.min);
1675*48a54d36SAndroid Build Coastguard Worker 			                return(ptr);
1676*48a54d36SAndroid Build Coastguard Worker 
1677*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NULL:
1678*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:
1679*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TSIG:
1680*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:
1681*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_X25:
1682*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_ISDN:
1683*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_LOC:
1684*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DHCID:if (ptr + rr->rdlength > limit) return(mDNSNULL);
1685*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(ptr, rdb->data, rr->rdlength);
1686*48a54d36SAndroid Build Coastguard Worker 							return(ptr + rr->rdlength);
1687*48a54d36SAndroid Build Coastguard Worker 
1688*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:
1689*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AFSDB:
1690*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RT:
1691*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_KX:	if (ptr + 3 > limit) return(mDNSNULL);
1692*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal16(ptr, rdb->mx.preference);
1693*48a54d36SAndroid Build Coastguard Worker 							return(putDomainNameAsLabels(msg, ptr, limit, &rdb->mx.exchange));
1694*48a54d36SAndroid Build Coastguard Worker 
1695*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RP:	ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->rp.mbox);
1696*48a54d36SAndroid Build Coastguard Worker 							if (!ptr) return(mDNSNULL);
1697*48a54d36SAndroid Build Coastguard Worker 							ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->rp.txt);
1698*48a54d36SAndroid Build Coastguard Worker 			                return(ptr);
1699*48a54d36SAndroid Build Coastguard Worker 
1700*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PX:	if (ptr + 5 > limit) return(mDNSNULL);
1701*48a54d36SAndroid Build Coastguard Worker 							ptr = putVal16(ptr, rdb->px.preference);
1702*48a54d36SAndroid Build Coastguard Worker 							ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->px.map822);
1703*48a54d36SAndroid Build Coastguard Worker 							if (!ptr) return(mDNSNULL);
1704*48a54d36SAndroid Build Coastguard Worker 							ptr = putDomainNameAsLabels(msg, ptr, limit, &rdb->px.mapx400);
1705*48a54d36SAndroid Build Coastguard Worker 			                return(ptr);
1706*48a54d36SAndroid Build Coastguard Worker 
1707*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA:	if (rr->rdlength != sizeof(rdb->ipv6))
1708*48a54d36SAndroid Build Coastguard Worker 								{ debugf("putRData: Illegal length %d for kDNSType_AAAA", rr->rdlength); return(mDNSNULL); }
1709*48a54d36SAndroid Build Coastguard Worker 							if (ptr + sizeof(rdb->ipv6) > limit) return(mDNSNULL);
1710*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(ptr, &rdb->ipv6, sizeof(rdb->ipv6));
1711*48a54d36SAndroid Build Coastguard Worker 							return(ptr + sizeof(rdb->ipv6));
1712*48a54d36SAndroid Build Coastguard Worker 
1713*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	if (ptr + 7 > limit) return(mDNSNULL);
1714*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = (mDNSu8)(rdb->srv.priority >> 8);
1715*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = (mDNSu8)(rdb->srv.priority &  0xFF);
1716*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = (mDNSu8)(rdb->srv.weight   >> 8);
1717*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = (mDNSu8)(rdb->srv.weight   &  0xFF);
1718*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->srv.port.b[0];
1719*48a54d36SAndroid Build Coastguard Worker 							*ptr++ = rdb->srv.port.b[1];
1720*48a54d36SAndroid Build Coastguard Worker 							return(putDomainNameAsLabels(msg, ptr, limit, &rdb->srv.target));
1721*48a54d36SAndroid Build Coastguard Worker 
1722*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:	{
1723*48a54d36SAndroid Build Coastguard Worker 							int len = 0;
1724*48a54d36SAndroid Build Coastguard Worker 							const rdataOPT *opt;
1725*48a54d36SAndroid Build Coastguard Worker 							const rdataOPT *const end = (const rdataOPT *)&rr->rdata->u.data[rr->rdlength];
1726*48a54d36SAndroid Build Coastguard Worker 							for (opt = &rr->rdata->u.opt[0]; opt < end; opt++) len += DNSOpt_Data_Space(opt);
1727*48a54d36SAndroid Build Coastguard Worker 							if (ptr + len > limit) { LogMsg("ERROR: putOptRData - out of space"); return mDNSNULL; }
1728*48a54d36SAndroid Build Coastguard Worker 
1729*48a54d36SAndroid Build Coastguard Worker 							for (opt = &rr->rdata->u.opt[0]; opt < end; opt++)
1730*48a54d36SAndroid Build Coastguard Worker 								{
1731*48a54d36SAndroid Build Coastguard Worker 								const int space = DNSOpt_Data_Space(opt);
1732*48a54d36SAndroid Build Coastguard Worker 								ptr = putVal16(ptr, opt->opt);
1733*48a54d36SAndroid Build Coastguard Worker 								ptr = putVal16(ptr, (mDNSu16)space - 4);
1734*48a54d36SAndroid Build Coastguard Worker 								switch (opt->opt)
1735*48a54d36SAndroid Build Coastguard Worker 									{
1736*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_LLQ:
1737*48a54d36SAndroid Build Coastguard Worker 										ptr = putVal16(ptr, opt->u.llq.vers);
1738*48a54d36SAndroid Build Coastguard Worker 										ptr = putVal16(ptr, opt->u.llq.llqOp);
1739*48a54d36SAndroid Build Coastguard Worker 										ptr = putVal16(ptr, opt->u.llq.err);
1740*48a54d36SAndroid Build Coastguard Worker 										mDNSPlatformMemCopy(ptr, opt->u.llq.id.b, 8);  // 8-byte id
1741*48a54d36SAndroid Build Coastguard Worker 										ptr += 8;
1742*48a54d36SAndroid Build Coastguard Worker 										ptr = putVal32(ptr, opt->u.llq.llqlease);
1743*48a54d36SAndroid Build Coastguard Worker 										break;
1744*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Lease:
1745*48a54d36SAndroid Build Coastguard Worker 										ptr = putVal32(ptr, opt->u.updatelease);
1746*48a54d36SAndroid Build Coastguard Worker 										break;
1747*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Owner:
1748*48a54d36SAndroid Build Coastguard Worker 										*ptr++ = opt->u.owner.vers;
1749*48a54d36SAndroid Build Coastguard Worker 										*ptr++ = opt->u.owner.seq;
1750*48a54d36SAndroid Build Coastguard Worker 										mDNSPlatformMemCopy(ptr, opt->u.owner.HMAC.b, 6);  // 6-byte Host identifier
1751*48a54d36SAndroid Build Coastguard Worker 										ptr += 6;
1752*48a54d36SAndroid Build Coastguard Worker 										if (space >= DNSOpt_OwnerData_ID_Wake_Space)
1753*48a54d36SAndroid Build Coastguard Worker 											{
1754*48a54d36SAndroid Build Coastguard Worker 											mDNSPlatformMemCopy(ptr, opt->u.owner.IMAC.b, 6);	// 6-byte interface MAC
1755*48a54d36SAndroid Build Coastguard Worker 											ptr += 6;
1756*48a54d36SAndroid Build Coastguard Worker 											if (space > DNSOpt_OwnerData_ID_Wake_Space)
1757*48a54d36SAndroid Build Coastguard Worker 												{
1758*48a54d36SAndroid Build Coastguard Worker 												mDNSPlatformMemCopy(ptr, opt->u.owner.password.b, space - DNSOpt_OwnerData_ID_Wake_Space);
1759*48a54d36SAndroid Build Coastguard Worker 												ptr += space - DNSOpt_OwnerData_ID_Wake_Space;
1760*48a54d36SAndroid Build Coastguard Worker 												}
1761*48a54d36SAndroid Build Coastguard Worker 											}
1762*48a54d36SAndroid Build Coastguard Worker 										break;
1763*48a54d36SAndroid Build Coastguard Worker 									}
1764*48a54d36SAndroid Build Coastguard Worker 								}
1765*48a54d36SAndroid Build Coastguard Worker 							return ptr;
1766*48a54d36SAndroid Build Coastguard Worker 							}
1767*48a54d36SAndroid Build Coastguard Worker 
1768*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: {
1769*48a54d36SAndroid Build Coastguard Worker 							// For our simplified use of NSEC synthetic records:
1770*48a54d36SAndroid Build Coastguard Worker 							// nextname is always the record's own name,
1771*48a54d36SAndroid Build Coastguard Worker 							// the block number is always 0,
1772*48a54d36SAndroid Build Coastguard Worker 							// the count byte is a value in the range 1-32,
1773*48a54d36SAndroid Build Coastguard Worker 							// followed by the 1-32 data bytes
1774*48a54d36SAndroid Build Coastguard Worker 							int i, j;
1775*48a54d36SAndroid Build Coastguard Worker 							for (i=sizeof(rdataNSEC); i>0; i--) if (rdb->nsec.bitmap[i-1]) break;
1776*48a54d36SAndroid Build Coastguard Worker 							ptr = putDomainNameAsLabels(msg, ptr, limit, rr->name);
1777*48a54d36SAndroid Build Coastguard Worker 							if (!ptr) return(mDNSNULL);
1778*48a54d36SAndroid Build Coastguard Worker 							if (i)		// Only put a block if at least one type exists for this name
1779*48a54d36SAndroid Build Coastguard Worker 								{
1780*48a54d36SAndroid Build Coastguard Worker 								if (ptr + 2 + i > limit) return(mDNSNULL);
1781*48a54d36SAndroid Build Coastguard Worker 								*ptr++ = 0;
1782*48a54d36SAndroid Build Coastguard Worker 								*ptr++ = (mDNSu8)i;
1783*48a54d36SAndroid Build Coastguard Worker 								for (j=0; j<i; j++) *ptr++ = rdb->nsec.bitmap[j];
1784*48a54d36SAndroid Build Coastguard Worker 								}
1785*48a54d36SAndroid Build Coastguard Worker 							return ptr;
1786*48a54d36SAndroid Build Coastguard Worker 							}
1787*48a54d36SAndroid Build Coastguard Worker 
1788*48a54d36SAndroid Build Coastguard Worker 		default:			debugf("putRData: Warning! Writing unknown resource type %d as raw data", rr->rrtype);
1789*48a54d36SAndroid Build Coastguard Worker 							if (ptr + rr->rdlength > limit) return(mDNSNULL);
1790*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(ptr, rdb->data, rr->rdlength);
1791*48a54d36SAndroid Build Coastguard Worker 							return(ptr + rr->rdlength);
1792*48a54d36SAndroid Build Coastguard Worker 		}
1793*48a54d36SAndroid Build Coastguard Worker 	}
1794*48a54d36SAndroid Build Coastguard Worker 
1795*48a54d36SAndroid Build Coastguard Worker #define IsUnicastUpdate(X) (!mDNSOpaque16IsZero((X)->h.id) && ((X)->h.flags.b[0] & kDNSFlag0_OP_Mask) == kDNSFlag0_OP_Update)
1796*48a54d36SAndroid Build Coastguard Worker 
PutResourceRecordTTLWithLimit(DNSMessage * const msg,mDNSu8 * ptr,mDNSu16 * count,ResourceRecord * rr,mDNSu32 ttl,const mDNSu8 * limit)1797*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *PutResourceRecordTTLWithLimit(DNSMessage *const msg, mDNSu8 *ptr, mDNSu16 *count, ResourceRecord *rr, mDNSu32 ttl, const mDNSu8 *limit)
1798*48a54d36SAndroid Build Coastguard Worker 	{
1799*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *endofrdata;
1800*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 actualLength;
1801*48a54d36SAndroid Build Coastguard Worker 	// When sending SRV to conventional DNS server (i.e. in DNS update requests) we should not do name compression on the rdata (RFC 2782)
1802*48a54d36SAndroid Build Coastguard Worker 	const DNSMessage *const rdatacompressionbase = (IsUnicastUpdate(msg) && rr->rrtype == kDNSType_SRV) ? mDNSNULL : msg;
1803*48a54d36SAndroid Build Coastguard Worker 
1804*48a54d36SAndroid Build Coastguard Worker 	if (rr->RecordType == kDNSRecordTypeUnregistered)
1805*48a54d36SAndroid Build Coastguard Worker 		{
1806*48a54d36SAndroid Build Coastguard Worker 		LogMsg("PutResourceRecord ERROR! Attempt to put kDNSRecordTypeUnregistered %##s (%s)", rr->name->c, DNSTypeName(rr->rrtype));
1807*48a54d36SAndroid Build Coastguard Worker 		return(ptr);
1808*48a54d36SAndroid Build Coastguard Worker 		}
1809*48a54d36SAndroid Build Coastguard Worker 
1810*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { LogMsg("PutResourceRecordTTLWithLimit ptr is null"); return(mDNSNULL); }
1811*48a54d36SAndroid Build Coastguard Worker 
1812*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, rr->name);
1813*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr + 10 >= limit) return(mDNSNULL);	// If we're out-of-space, return mDNSNULL
1814*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)(rr->rrtype  >> 8);
1815*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)(rr->rrtype  &  0xFF);
1816*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)(rr->rrclass >> 8);
1817*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)(rr->rrclass &  0xFF);
1818*48a54d36SAndroid Build Coastguard Worker 	ptr[4] = (mDNSu8)((ttl >> 24) &  0xFF);
1819*48a54d36SAndroid Build Coastguard Worker 	ptr[5] = (mDNSu8)((ttl >> 16) &  0xFF);
1820*48a54d36SAndroid Build Coastguard Worker 	ptr[6] = (mDNSu8)((ttl >>  8) &  0xFF);
1821*48a54d36SAndroid Build Coastguard Worker 	ptr[7] = (mDNSu8)( ttl        &  0xFF);
1822*48a54d36SAndroid Build Coastguard Worker 	// ptr[8] and ptr[9] filled in *after* we find out how much space the rdata takes
1823*48a54d36SAndroid Build Coastguard Worker 
1824*48a54d36SAndroid Build Coastguard Worker 	endofrdata = putRData(rdatacompressionbase, ptr+10, limit, rr);
1825*48a54d36SAndroid Build Coastguard Worker 	if (!endofrdata) { verbosedebugf("Ran out of space in PutResourceRecord for %##s (%s)", rr->name->c, DNSTypeName(rr->rrtype)); return(mDNSNULL); }
1826*48a54d36SAndroid Build Coastguard Worker 
1827*48a54d36SAndroid Build Coastguard Worker 	// Go back and fill in the actual number of data bytes we wrote
1828*48a54d36SAndroid Build Coastguard Worker 	// (actualLength can be less than rdlength when domain name compression is used)
1829*48a54d36SAndroid Build Coastguard Worker 	actualLength = (mDNSu16)(endofrdata - ptr - 10);
1830*48a54d36SAndroid Build Coastguard Worker 	ptr[8] = (mDNSu8)(actualLength >> 8);
1831*48a54d36SAndroid Build Coastguard Worker 	ptr[9] = (mDNSu8)(actualLength &  0xFF);
1832*48a54d36SAndroid Build Coastguard Worker 
1833*48a54d36SAndroid Build Coastguard Worker 	if (count) (*count)++;
1834*48a54d36SAndroid Build Coastguard Worker 	else LogMsg("PutResourceRecordTTL: ERROR: No target count to update for %##s (%s)", rr->name->c, DNSTypeName(rr->rrtype));
1835*48a54d36SAndroid Build Coastguard Worker 	return(endofrdata);
1836*48a54d36SAndroid Build Coastguard Worker 	}
1837*48a54d36SAndroid Build Coastguard Worker 
putEmptyResourceRecord(DNSMessage * const msg,mDNSu8 * ptr,const mDNSu8 * const limit,mDNSu16 * count,const AuthRecord * rr)1838*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu8 *putEmptyResourceRecord(DNSMessage *const msg, mDNSu8 *ptr, const mDNSu8 *const limit, mDNSu16 *count, const AuthRecord *rr)
1839*48a54d36SAndroid Build Coastguard Worker 	{
1840*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, rr->resrec.name);
1841*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr + 10 > limit) return(mDNSNULL);		// If we're out-of-space, return mDNSNULL
1842*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)(rr->resrec.rrtype  >> 8);				// Put type
1843*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)(rr->resrec.rrtype  &  0xFF);
1844*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)(rr->resrec.rrclass >> 8);				// Put class
1845*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)(rr->resrec.rrclass &  0xFF);
1846*48a54d36SAndroid Build Coastguard Worker 	ptr[4] = ptr[5] = ptr[6] = ptr[7] = 0;				// TTL is zero
1847*48a54d36SAndroid Build Coastguard Worker 	ptr[8] = ptr[9] = 0;								// RDATA length is zero
1848*48a54d36SAndroid Build Coastguard Worker 	(*count)++;
1849*48a54d36SAndroid Build Coastguard Worker 	return(ptr + 10);
1850*48a54d36SAndroid Build Coastguard Worker 	}
1851*48a54d36SAndroid Build Coastguard Worker 
putQuestion(DNSMessage * const msg,mDNSu8 * ptr,const mDNSu8 * const limit,const domainname * const name,mDNSu16 rrtype,mDNSu16 rrclass)1852*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putQuestion(DNSMessage *const msg, mDNSu8 *ptr, const mDNSu8 *const limit, const domainname *const name, mDNSu16 rrtype, mDNSu16 rrclass)
1853*48a54d36SAndroid Build Coastguard Worker 	{
1854*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, name);
1855*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr+4 >= limit) return(mDNSNULL);			// If we're out-of-space, return mDNSNULL
1856*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)(rrtype  >> 8);
1857*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)(rrtype  &  0xFF);
1858*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)(rrclass >> 8);
1859*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)(rrclass &  0xFF);
1860*48a54d36SAndroid Build Coastguard Worker 	msg->h.numQuestions++;
1861*48a54d36SAndroid Build Coastguard Worker 	return(ptr+4);
1862*48a54d36SAndroid Build Coastguard Worker 	}
1863*48a54d36SAndroid Build Coastguard Worker 
1864*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putZone(DNSMessage * const msg,mDNSu8 * ptr,mDNSu8 * limit,const domainname * zone,mDNSOpaque16 zoneClass)1865*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putZone(DNSMessage *const msg, mDNSu8 *ptr, mDNSu8 *limit, const domainname *zone, mDNSOpaque16 zoneClass)
1866*48a54d36SAndroid Build Coastguard Worker 	{
1867*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, zone);
1868*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr + 4 > limit) return mDNSNULL;		// If we're out-of-space, return NULL
1869*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = (mDNSu8)(kDNSType_SOA  >> 8);
1870*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = (mDNSu8)(kDNSType_SOA  &  0xFF);
1871*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = zoneClass.b[0];
1872*48a54d36SAndroid Build Coastguard Worker 	*ptr++ = zoneClass.b[1];
1873*48a54d36SAndroid Build Coastguard Worker 	msg->h.mDNS_numZones++;
1874*48a54d36SAndroid Build Coastguard Worker 	return ptr;
1875*48a54d36SAndroid Build Coastguard Worker 	}
1876*48a54d36SAndroid Build Coastguard Worker 
1877*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putPrereqNameNotInUse(const domainname * const name,DNSMessage * const msg,mDNSu8 * const ptr,mDNSu8 * const end)1878*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putPrereqNameNotInUse(const domainname *const name, DNSMessage *const msg, mDNSu8 *const ptr, mDNSu8 *const end)
1879*48a54d36SAndroid Build Coastguard Worker 	{
1880*48a54d36SAndroid Build Coastguard Worker 	AuthRecord prereq;
1881*48a54d36SAndroid Build Coastguard Worker 	mDNS_SetupResourceRecord(&prereq, mDNSNULL, mDNSInterface_Any, kDNSQType_ANY, kStandardTTL, 0, AuthRecordAny, mDNSNULL, mDNSNULL);
1882*48a54d36SAndroid Build Coastguard Worker 	AssignDomainName(&prereq.namestorage, name);
1883*48a54d36SAndroid Build Coastguard Worker 	prereq.resrec.rrtype = kDNSQType_ANY;
1884*48a54d36SAndroid Build Coastguard Worker 	prereq.resrec.rrclass = kDNSClass_NONE;
1885*48a54d36SAndroid Build Coastguard Worker 	return putEmptyResourceRecord(msg, ptr, end, &msg->h.mDNS_numPrereqs, &prereq);
1886*48a54d36SAndroid Build Coastguard Worker 	}
1887*48a54d36SAndroid Build Coastguard Worker 
1888*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putDeletionRecord(DNSMessage * msg,mDNSu8 * ptr,ResourceRecord * rr)1889*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putDeletionRecord(DNSMessage *msg, mDNSu8 *ptr, ResourceRecord *rr)
1890*48a54d36SAndroid Build Coastguard Worker 	{
1891*48a54d36SAndroid Build Coastguard Worker 	// deletion: specify record w/ TTL 0, class NONE
1892*48a54d36SAndroid Build Coastguard Worker 	const mDNSu16 origclass = rr->rrclass;
1893*48a54d36SAndroid Build Coastguard Worker 	rr->rrclass = kDNSClass_NONE;
1894*48a54d36SAndroid Build Coastguard Worker 	ptr = PutResourceRecordTTLJumbo(msg, ptr, &msg->h.mDNS_numUpdates, rr, 0);
1895*48a54d36SAndroid Build Coastguard Worker 	rr->rrclass = origclass;
1896*48a54d36SAndroid Build Coastguard Worker 	return ptr;
1897*48a54d36SAndroid Build Coastguard Worker 	}
1898*48a54d36SAndroid Build Coastguard Worker 
1899*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putDeletionRecordWithLimit(DNSMessage * msg,mDNSu8 * ptr,ResourceRecord * rr,mDNSu8 * limit)1900*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putDeletionRecordWithLimit(DNSMessage *msg, mDNSu8 *ptr, ResourceRecord *rr, mDNSu8 *limit)
1901*48a54d36SAndroid Build Coastguard Worker 	{
1902*48a54d36SAndroid Build Coastguard Worker 	// deletion: specify record w/ TTL 0, class NONE
1903*48a54d36SAndroid Build Coastguard Worker 	const mDNSu16 origclass = rr->rrclass;
1904*48a54d36SAndroid Build Coastguard Worker 	rr->rrclass = kDNSClass_NONE;
1905*48a54d36SAndroid Build Coastguard Worker 	ptr = PutResourceRecordTTLWithLimit(msg, ptr, &msg->h.mDNS_numUpdates, rr, 0, limit);
1906*48a54d36SAndroid Build Coastguard Worker 	rr->rrclass = origclass;
1907*48a54d36SAndroid Build Coastguard Worker 	return ptr;
1908*48a54d36SAndroid Build Coastguard Worker 	}
1909*48a54d36SAndroid Build Coastguard Worker 
putDeleteRRSetWithLimit(DNSMessage * msg,mDNSu8 * ptr,const domainname * name,mDNSu16 rrtype,mDNSu8 * limit)1910*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putDeleteRRSetWithLimit(DNSMessage *msg, mDNSu8 *ptr, const domainname *name, mDNSu16 rrtype, mDNSu8 *limit)
1911*48a54d36SAndroid Build Coastguard Worker 	{
1912*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 class = kDNSQClass_ANY;
1913*48a54d36SAndroid Build Coastguard Worker 
1914*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, name);
1915*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr + 10 >= limit) return mDNSNULL;	// If we're out-of-space, return mDNSNULL
1916*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)(rrtype  >> 8);
1917*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)(rrtype  &  0xFF);
1918*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)(class >> 8);
1919*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)(class &  0xFF);
1920*48a54d36SAndroid Build Coastguard Worker 	ptr[4] = ptr[5] = ptr[6] = ptr[7] = 0; // zero ttl
1921*48a54d36SAndroid Build Coastguard Worker 	ptr[8] = ptr[9] = 0; // zero rdlength/rdata
1922*48a54d36SAndroid Build Coastguard Worker 
1923*48a54d36SAndroid Build Coastguard Worker 	msg->h.mDNS_numUpdates++;
1924*48a54d36SAndroid Build Coastguard Worker 	return ptr + 10;
1925*48a54d36SAndroid Build Coastguard Worker 	}
1926*48a54d36SAndroid Build Coastguard Worker 
1927*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putDeleteAllRRSets(DNSMessage * msg,mDNSu8 * ptr,const domainname * name)1928*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putDeleteAllRRSets(DNSMessage *msg, mDNSu8 *ptr, const domainname *name)
1929*48a54d36SAndroid Build Coastguard Worker 	{
1930*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *limit = msg->data + AbsoluteMaxDNSMessageData;
1931*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 class = kDNSQClass_ANY;
1932*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 rrtype = kDNSQType_ANY;
1933*48a54d36SAndroid Build Coastguard Worker 
1934*48a54d36SAndroid Build Coastguard Worker 	ptr = putDomainNameAsLabels(msg, ptr, limit, name);
1935*48a54d36SAndroid Build Coastguard Worker 	if (!ptr || ptr + 10 >= limit) return mDNSNULL;	// If we're out-of-space, return mDNSNULL
1936*48a54d36SAndroid Build Coastguard Worker 	ptr[0] = (mDNSu8)(rrtype >> 8);
1937*48a54d36SAndroid Build Coastguard Worker 	ptr[1] = (mDNSu8)(rrtype &  0xFF);
1938*48a54d36SAndroid Build Coastguard Worker 	ptr[2] = (mDNSu8)(class  >> 8);
1939*48a54d36SAndroid Build Coastguard Worker 	ptr[3] = (mDNSu8)(class  &  0xFF);
1940*48a54d36SAndroid Build Coastguard Worker 	ptr[4] = ptr[5] = ptr[6] = ptr[7] = 0; // zero ttl
1941*48a54d36SAndroid Build Coastguard Worker 	ptr[8] = ptr[9] = 0; // zero rdlength/rdata
1942*48a54d36SAndroid Build Coastguard Worker 
1943*48a54d36SAndroid Build Coastguard Worker 	msg->h.mDNS_numUpdates++;
1944*48a54d36SAndroid Build Coastguard Worker 	return ptr + 10;
1945*48a54d36SAndroid Build Coastguard Worker 	}
1946*48a54d36SAndroid Build Coastguard Worker 
1947*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putUpdateLease(DNSMessage * msg,mDNSu8 * end,mDNSu32 lease)1948*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putUpdateLease(DNSMessage *msg, mDNSu8 *end, mDNSu32 lease)
1949*48a54d36SAndroid Build Coastguard Worker 	{
1950*48a54d36SAndroid Build Coastguard Worker 	AuthRecord rr;
1951*48a54d36SAndroid Build Coastguard Worker 	mDNS_SetupResourceRecord(&rr, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
1952*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rrclass    = NormalMaxDNSMessageData;
1953*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdlength   = sizeof(rdataOPT);	// One option in this OPT record
1954*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdestimate = sizeof(rdataOPT);
1955*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdata->u.opt[0].opt           = kDNSOpt_Lease;
1956*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdata->u.opt[0].u.updatelease = lease;
1957*48a54d36SAndroid Build Coastguard Worker 	end = PutResourceRecordTTLJumbo(msg, end, &msg->h.numAdditionals, &rr.resrec, 0);
1958*48a54d36SAndroid Build Coastguard Worker 	if (!end) { LogMsg("ERROR: putUpdateLease - PutResourceRecordTTL"); return mDNSNULL; }
1959*48a54d36SAndroid Build Coastguard Worker 	return end;
1960*48a54d36SAndroid Build Coastguard Worker 	}
1961*48a54d36SAndroid Build Coastguard Worker 
1962*48a54d36SAndroid Build Coastguard Worker // for dynamic updates
putUpdateLeaseWithLimit(DNSMessage * msg,mDNSu8 * end,mDNSu32 lease,mDNSu8 * limit)1963*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putUpdateLeaseWithLimit(DNSMessage *msg, mDNSu8 *end, mDNSu32 lease, mDNSu8 *limit)
1964*48a54d36SAndroid Build Coastguard Worker 	{
1965*48a54d36SAndroid Build Coastguard Worker 	AuthRecord rr;
1966*48a54d36SAndroid Build Coastguard Worker 	mDNS_SetupResourceRecord(&rr, mDNSNULL, mDNSInterface_Any, kDNSType_OPT, kStandardTTL, kDNSRecordTypeKnownUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
1967*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rrclass    = NormalMaxDNSMessageData;
1968*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdlength   = sizeof(rdataOPT);	// One option in this OPT record
1969*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdestimate = sizeof(rdataOPT);
1970*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdata->u.opt[0].opt           = kDNSOpt_Lease;
1971*48a54d36SAndroid Build Coastguard Worker 	rr.resrec.rdata->u.opt[0].u.updatelease = lease;
1972*48a54d36SAndroid Build Coastguard Worker 	end = PutResourceRecordTTLWithLimit(msg, end, &msg->h.numAdditionals, &rr.resrec, 0, limit);
1973*48a54d36SAndroid Build Coastguard Worker 	if (!end) { LogMsg("ERROR: putUpdateLease - PutResourceRecordTTLWithLimit"); return mDNSNULL; }
1974*48a54d36SAndroid Build Coastguard Worker 	return end;
1975*48a54d36SAndroid Build Coastguard Worker 	}
1976*48a54d36SAndroid Build Coastguard Worker 
putHINFO(const mDNS * const m,DNSMessage * const msg,mDNSu8 * end,DomainAuthInfo * authInfo,mDNSu8 * limit)1977*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu8 *putHINFO(const mDNS *const m, DNSMessage *const msg, mDNSu8 *end, DomainAuthInfo *authInfo, mDNSu8 *limit)
1978*48a54d36SAndroid Build Coastguard Worker 	{
1979*48a54d36SAndroid Build Coastguard Worker 	if (authInfo && authInfo->AutoTunnel)
1980*48a54d36SAndroid Build Coastguard Worker 		{
1981*48a54d36SAndroid Build Coastguard Worker 		AuthRecord hinfo;
1982*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 *h = hinfo.rdatastorage.u.data;
1983*48a54d36SAndroid Build Coastguard Worker 		mDNSu16 len = 2 + m->HIHardware.c[0] + m->HISoftware.c[0];
1984*48a54d36SAndroid Build Coastguard Worker 		mDNSu8 *newptr;
1985*48a54d36SAndroid Build Coastguard Worker 		mDNS_SetupResourceRecord(&hinfo, mDNSNULL, mDNSInterface_Any, kDNSType_HINFO, 0, kDNSRecordTypeUnique, AuthRecordAny, mDNSNULL, mDNSNULL);
1986*48a54d36SAndroid Build Coastguard Worker 		AppendDomainLabel(&hinfo.namestorage, &m->hostlabel);
1987*48a54d36SAndroid Build Coastguard Worker 		AppendDomainName (&hinfo.namestorage, &authInfo->domain);
1988*48a54d36SAndroid Build Coastguard Worker 		hinfo.resrec.rroriginalttl = 0;
1989*48a54d36SAndroid Build Coastguard Worker 		mDNSPlatformMemCopy(h, &m->HIHardware, 1 + (mDNSu32)m->HIHardware.c[0]);
1990*48a54d36SAndroid Build Coastguard Worker 		h += 1 + (int)h[0];
1991*48a54d36SAndroid Build Coastguard Worker 		mDNSPlatformMemCopy(h, &m->HISoftware, 1 + (mDNSu32)m->HISoftware.c[0]);
1992*48a54d36SAndroid Build Coastguard Worker 		hinfo.resrec.rdlength   = len;
1993*48a54d36SAndroid Build Coastguard Worker 		hinfo.resrec.rdestimate = len;
1994*48a54d36SAndroid Build Coastguard Worker 		newptr = PutResourceRecordTTLWithLimit(msg, end, &msg->h.numAdditionals, &hinfo.resrec, 0, limit);
1995*48a54d36SAndroid Build Coastguard Worker 		return newptr;
1996*48a54d36SAndroid Build Coastguard Worker 		}
1997*48a54d36SAndroid Build Coastguard Worker 	else
1998*48a54d36SAndroid Build Coastguard Worker 		return end;
1999*48a54d36SAndroid Build Coastguard Worker 	}
2000*48a54d36SAndroid Build Coastguard Worker 
2001*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2002*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2003*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2004*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNS Message Parsing Functions
2005*48a54d36SAndroid Build Coastguard Worker #endif
2006*48a54d36SAndroid Build Coastguard Worker 
DomainNameHashValue(const domainname * const name)2007*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 DomainNameHashValue(const domainname *const name)
2008*48a54d36SAndroid Build Coastguard Worker 	{
2009*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 sum = 0;
2010*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *c;
2011*48a54d36SAndroid Build Coastguard Worker 
2012*48a54d36SAndroid Build Coastguard Worker 	for (c = name->c; c[0] != 0 && c[1] != 0; c += 2)
2013*48a54d36SAndroid Build Coastguard Worker 		{
2014*48a54d36SAndroid Build Coastguard Worker 		sum += ((mDNSIsUpperCase(c[0]) ? c[0] + 'a' - 'A' : c[0]) << 8) |
2015*48a54d36SAndroid Build Coastguard Worker 				(mDNSIsUpperCase(c[1]) ? c[1] + 'a' - 'A' : c[1]);
2016*48a54d36SAndroid Build Coastguard Worker 		sum = (sum<<3) | (sum>>29);
2017*48a54d36SAndroid Build Coastguard Worker 		}
2018*48a54d36SAndroid Build Coastguard Worker 	if (c[0]) sum += ((mDNSIsUpperCase(c[0]) ? c[0] + 'a' - 'A' : c[0]) << 8);
2019*48a54d36SAndroid Build Coastguard Worker 	return(sum);
2020*48a54d36SAndroid Build Coastguard Worker 	}
2021*48a54d36SAndroid Build Coastguard Worker 
SetNewRData(ResourceRecord * const rr,RData * NewRData,mDNSu16 rdlength)2022*48a54d36SAndroid Build Coastguard Worker mDNSexport void SetNewRData(ResourceRecord *const rr, RData *NewRData, mDNSu16 rdlength)
2023*48a54d36SAndroid Build Coastguard Worker 	{
2024*48a54d36SAndroid Build Coastguard Worker 	domainname *target;
2025*48a54d36SAndroid Build Coastguard Worker 	if (NewRData)
2026*48a54d36SAndroid Build Coastguard Worker 		{
2027*48a54d36SAndroid Build Coastguard Worker 		rr->rdata    = NewRData;
2028*48a54d36SAndroid Build Coastguard Worker 		rr->rdlength = rdlength;
2029*48a54d36SAndroid Build Coastguard Worker 		}
2030*48a54d36SAndroid Build Coastguard Worker 	// Must not try to get target pointer until after updating rr->rdata
2031*48a54d36SAndroid Build Coastguard Worker 	target = GetRRDomainNameTarget(rr);
2032*48a54d36SAndroid Build Coastguard Worker 	rr->rdlength   = GetRDLength(rr, mDNSfalse);
2033*48a54d36SAndroid Build Coastguard Worker 	rr->rdestimate = GetRDLength(rr, mDNStrue);
2034*48a54d36SAndroid Build Coastguard Worker 	rr->rdatahash  = target ? DomainNameHashValue(target) : RDataHashValue(rr);
2035*48a54d36SAndroid Build Coastguard Worker 	}
2036*48a54d36SAndroid Build Coastguard Worker 
skipDomainName(const DNSMessage * const msg,const mDNSu8 * ptr,const mDNSu8 * const end)2037*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *skipDomainName(const DNSMessage *const msg, const mDNSu8 *ptr, const mDNSu8 *const end)
2038*48a54d36SAndroid Build Coastguard Worker 	{
2039*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 total = 0;
2040*48a54d36SAndroid Build Coastguard Worker 
2041*48a54d36SAndroid Build Coastguard Worker 	if (ptr < (mDNSu8*)msg || ptr >= end)
2042*48a54d36SAndroid Build Coastguard Worker 		{ debugf("skipDomainName: Illegal ptr not within packet boundaries"); return(mDNSNULL); }
2043*48a54d36SAndroid Build Coastguard Worker 
2044*48a54d36SAndroid Build Coastguard Worker 	while (1)						// Read sequence of labels
2045*48a54d36SAndroid Build Coastguard Worker 		{
2046*48a54d36SAndroid Build Coastguard Worker 		const mDNSu8 len = *ptr++;	// Read length of this label
2047*48a54d36SAndroid Build Coastguard Worker 		if (len == 0) return(ptr);	// If length is zero, that means this name is complete
2048*48a54d36SAndroid Build Coastguard Worker 		switch (len & 0xC0)
2049*48a54d36SAndroid Build Coastguard Worker 			{
2050*48a54d36SAndroid Build Coastguard Worker 			case 0x00:	if (ptr + len >= end)					// Remember: expect at least one more byte for the root label
2051*48a54d36SAndroid Build Coastguard Worker 							{ debugf("skipDomainName: Malformed domain name (overruns packet end)"); return(mDNSNULL); }
2052*48a54d36SAndroid Build Coastguard Worker 						if (total + 1 + len >= MAX_DOMAIN_NAME)	// Remember: expect at least one more byte for the root label
2053*48a54d36SAndroid Build Coastguard Worker 							{ debugf("skipDomainName: Malformed domain name (more than 256 characters)"); return(mDNSNULL); }
2054*48a54d36SAndroid Build Coastguard Worker 						ptr += len;
2055*48a54d36SAndroid Build Coastguard Worker 						total += 1 + len;
2056*48a54d36SAndroid Build Coastguard Worker 						break;
2057*48a54d36SAndroid Build Coastguard Worker 
2058*48a54d36SAndroid Build Coastguard Worker 			case 0x40:	debugf("skipDomainName: Extended EDNS0 label types 0x%X not supported", len); return(mDNSNULL);
2059*48a54d36SAndroid Build Coastguard Worker 			case 0x80:	debugf("skipDomainName: Illegal label length 0x%X", len); return(mDNSNULL);
2060*48a54d36SAndroid Build Coastguard Worker 			case 0xC0:	return(ptr+1);
2061*48a54d36SAndroid Build Coastguard Worker 			}
2062*48a54d36SAndroid Build Coastguard Worker 		}
2063*48a54d36SAndroid Build Coastguard Worker 	}
2064*48a54d36SAndroid Build Coastguard Worker 
2065*48a54d36SAndroid Build Coastguard Worker // Routine to fetch an FQDN from the DNS message, following compression pointers if necessary.
getDomainName(const DNSMessage * const msg,const mDNSu8 * ptr,const mDNSu8 * const end,domainname * const name)2066*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *getDomainName(const DNSMessage *const msg, const mDNSu8 *ptr, const mDNSu8 *const end,
2067*48a54d36SAndroid Build Coastguard Worker 	domainname *const name)
2068*48a54d36SAndroid Build Coastguard Worker 	{
2069*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *nextbyte = mDNSNULL;					// Record where we got to before we started following pointers
2070*48a54d36SAndroid Build Coastguard Worker 	mDNSu8       *np = name->c;							// Name pointer
2071*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *const limit = np + MAX_DOMAIN_NAME;	// Limit so we don't overrun buffer
2072*48a54d36SAndroid Build Coastguard Worker 
2073*48a54d36SAndroid Build Coastguard Worker 	if (ptr < (mDNSu8*)msg || ptr >= end)
2074*48a54d36SAndroid Build Coastguard Worker 		{ debugf("getDomainName: Illegal ptr not within packet boundaries"); return(mDNSNULL); }
2075*48a54d36SAndroid Build Coastguard Worker 
2076*48a54d36SAndroid Build Coastguard Worker 	*np = 0;						// Tentatively place the root label here (may be overwritten if we have more labels)
2077*48a54d36SAndroid Build Coastguard Worker 
2078*48a54d36SAndroid Build Coastguard Worker 	while (1)						// Read sequence of labels
2079*48a54d36SAndroid Build Coastguard Worker 		{
2080*48a54d36SAndroid Build Coastguard Worker 		const mDNSu8 len = *ptr++;	// Read length of this label
2081*48a54d36SAndroid Build Coastguard Worker 		if (len == 0) break;		// If length is zero, that means this name is complete
2082*48a54d36SAndroid Build Coastguard Worker 		switch (len & 0xC0)
2083*48a54d36SAndroid Build Coastguard Worker 			{
2084*48a54d36SAndroid Build Coastguard Worker 			int i;
2085*48a54d36SAndroid Build Coastguard Worker 			mDNSu16 offset;
2086*48a54d36SAndroid Build Coastguard Worker 
2087*48a54d36SAndroid Build Coastguard Worker 			case 0x00:	if (ptr + len >= end)		// Remember: expect at least one more byte for the root label
2088*48a54d36SAndroid Build Coastguard Worker 							{ debugf("getDomainName: Malformed domain name (overruns packet end)"); return(mDNSNULL); }
2089*48a54d36SAndroid Build Coastguard Worker 						if (np + 1 + len >= limit)	// Remember: expect at least one more byte for the root label
2090*48a54d36SAndroid Build Coastguard Worker 							{ debugf("getDomainName: Malformed domain name (more than 256 characters)"); return(mDNSNULL); }
2091*48a54d36SAndroid Build Coastguard Worker 						*np++ = len;
2092*48a54d36SAndroid Build Coastguard Worker 						for (i=0; i<len; i++) *np++ = *ptr++;
2093*48a54d36SAndroid Build Coastguard Worker 						*np = 0;	// Tentatively place the root label here (may be overwritten if we have more labels)
2094*48a54d36SAndroid Build Coastguard Worker 						break;
2095*48a54d36SAndroid Build Coastguard Worker 
2096*48a54d36SAndroid Build Coastguard Worker 			case 0x40:	debugf("getDomainName: Extended EDNS0 label types 0x%X not supported in name %##s", len, name->c);
2097*48a54d36SAndroid Build Coastguard Worker 						return(mDNSNULL);
2098*48a54d36SAndroid Build Coastguard Worker 
2099*48a54d36SAndroid Build Coastguard Worker 			case 0x80:	debugf("getDomainName: Illegal label length 0x%X in domain name %##s", len, name->c); return(mDNSNULL);
2100*48a54d36SAndroid Build Coastguard Worker 
2101*48a54d36SAndroid Build Coastguard Worker 			case 0xC0:	offset = (mDNSu16)((((mDNSu16)(len & 0x3F)) << 8) | *ptr++);
2102*48a54d36SAndroid Build Coastguard Worker 						if (!nextbyte) nextbyte = ptr;	// Record where we got to before we started following pointers
2103*48a54d36SAndroid Build Coastguard Worker 						ptr = (mDNSu8 *)msg + offset;
2104*48a54d36SAndroid Build Coastguard Worker 						if (ptr < (mDNSu8*)msg || ptr >= end)
2105*48a54d36SAndroid Build Coastguard Worker 							{ debugf("getDomainName: Illegal compression pointer not within packet boundaries"); return(mDNSNULL); }
2106*48a54d36SAndroid Build Coastguard Worker 						if (*ptr & 0xC0)
2107*48a54d36SAndroid Build Coastguard Worker 							{ debugf("getDomainName: Compression pointer must point to real label"); return(mDNSNULL); }
2108*48a54d36SAndroid Build Coastguard Worker 						break;
2109*48a54d36SAndroid Build Coastguard Worker 			}
2110*48a54d36SAndroid Build Coastguard Worker 		}
2111*48a54d36SAndroid Build Coastguard Worker 
2112*48a54d36SAndroid Build Coastguard Worker 	if (nextbyte) return(nextbyte);
2113*48a54d36SAndroid Build Coastguard Worker 	else return(ptr);
2114*48a54d36SAndroid Build Coastguard Worker 	}
2115*48a54d36SAndroid Build Coastguard Worker 
skipResourceRecord(const DNSMessage * msg,const mDNSu8 * ptr,const mDNSu8 * end)2116*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *skipResourceRecord(const DNSMessage *msg, const mDNSu8 *ptr, const mDNSu8 *end)
2117*48a54d36SAndroid Build Coastguard Worker 	{
2118*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 pktrdlength;
2119*48a54d36SAndroid Build Coastguard Worker 
2120*48a54d36SAndroid Build Coastguard Worker 	ptr = skipDomainName(msg, ptr, end);
2121*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { debugf("skipResourceRecord: Malformed RR name"); return(mDNSNULL); }
2122*48a54d36SAndroid Build Coastguard Worker 
2123*48a54d36SAndroid Build Coastguard Worker 	if (ptr + 10 > end) { debugf("skipResourceRecord: Malformed RR -- no type/class/ttl/len!"); return(mDNSNULL); }
2124*48a54d36SAndroid Build Coastguard Worker 	pktrdlength = (mDNSu16)((mDNSu16)ptr[8] <<  8 | ptr[9]);
2125*48a54d36SAndroid Build Coastguard Worker 	ptr += 10;
2126*48a54d36SAndroid Build Coastguard Worker 	if (ptr + pktrdlength > end) { debugf("skipResourceRecord: RDATA exceeds end of packet"); return(mDNSNULL); }
2127*48a54d36SAndroid Build Coastguard Worker 
2128*48a54d36SAndroid Build Coastguard Worker 	return(ptr + pktrdlength);
2129*48a54d36SAndroid Build Coastguard Worker 	}
2130*48a54d36SAndroid Build Coastguard Worker 
GetLargeResourceRecord(mDNS * const m,const DNSMessage * const msg,const mDNSu8 * ptr,const mDNSu8 * end,const mDNSInterfaceID InterfaceID,mDNSu8 RecordType,LargeCacheRecord * const largecr)2131*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *GetLargeResourceRecord(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *ptr,
2132*48a54d36SAndroid Build Coastguard Worker     const mDNSu8 *end, const mDNSInterfaceID InterfaceID, mDNSu8 RecordType, LargeCacheRecord *const largecr)
2133*48a54d36SAndroid Build Coastguard Worker 	{
2134*48a54d36SAndroid Build Coastguard Worker 	CacheRecord *const rr = &largecr->r;
2135*48a54d36SAndroid Build Coastguard Worker 	RDataBody2 *const rdb = (RDataBody2 *)rr->smallrdatastorage.data;
2136*48a54d36SAndroid Build Coastguard Worker 	mDNSu16 pktrdlength;
2137*48a54d36SAndroid Build Coastguard Worker 
2138*48a54d36SAndroid Build Coastguard Worker 	if (largecr == &m->rec && m->rec.r.resrec.RecordType)
2139*48a54d36SAndroid Build Coastguard Worker 		{
2140*48a54d36SAndroid Build Coastguard Worker 		LogMsg("GetLargeResourceRecord: m->rec appears to be already in use for %s", CRDisplayString(m, &m->rec.r));
2141*48a54d36SAndroid Build Coastguard Worker #if ForceAlerts
2142*48a54d36SAndroid Build Coastguard Worker 		*(long*)0 = 0;
2143*48a54d36SAndroid Build Coastguard Worker #endif
2144*48a54d36SAndroid Build Coastguard Worker 		}
2145*48a54d36SAndroid Build Coastguard Worker 
2146*48a54d36SAndroid Build Coastguard Worker 	rr->next              = mDNSNULL;
2147*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.name       = &largecr->namestorage;
2148*48a54d36SAndroid Build Coastguard Worker 
2149*48a54d36SAndroid Build Coastguard Worker 	rr->NextInKAList      = mDNSNULL;
2150*48a54d36SAndroid Build Coastguard Worker 	rr->TimeRcvd          = m ? m->timenow : 0;
2151*48a54d36SAndroid Build Coastguard Worker 	rr->DelayDelivery     = 0;
2152*48a54d36SAndroid Build Coastguard Worker 	rr->NextRequiredQuery = m ? m->timenow : 0;		// Will be updated to the real value when we call SetNextCacheCheckTimeForRecord()
2153*48a54d36SAndroid Build Coastguard Worker 	rr->LastUsed          = m ? m->timenow : 0;
2154*48a54d36SAndroid Build Coastguard Worker 	rr->CRActiveQuestion  = mDNSNULL;
2155*48a54d36SAndroid Build Coastguard Worker 	rr->UnansweredQueries = 0;
2156*48a54d36SAndroid Build Coastguard Worker 	rr->LastUnansweredTime= 0;
2157*48a54d36SAndroid Build Coastguard Worker #if ENABLE_MULTI_PACKET_QUERY_SNOOPING
2158*48a54d36SAndroid Build Coastguard Worker 	rr->MPUnansweredQ     = 0;
2159*48a54d36SAndroid Build Coastguard Worker 	rr->MPLastUnansweredQT= 0;
2160*48a54d36SAndroid Build Coastguard Worker 	rr->MPUnansweredKA    = 0;
2161*48a54d36SAndroid Build Coastguard Worker 	rr->MPExpectingKA     = mDNSfalse;
2162*48a54d36SAndroid Build Coastguard Worker #endif
2163*48a54d36SAndroid Build Coastguard Worker 	rr->NextInCFList      = mDNSNULL;
2164*48a54d36SAndroid Build Coastguard Worker 
2165*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.InterfaceID       = InterfaceID;
2166*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rDNSServer = mDNSNULL;
2167*48a54d36SAndroid Build Coastguard Worker 
2168*48a54d36SAndroid Build Coastguard Worker 	ptr = getDomainName(msg, ptr, end, &largecr->namestorage);		// Will bail out correctly if ptr is NULL
2169*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { debugf("GetLargeResourceRecord: Malformed RR name"); return(mDNSNULL); }
2170*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.namehash = DomainNameHashValue(rr->resrec.name);
2171*48a54d36SAndroid Build Coastguard Worker 
2172*48a54d36SAndroid Build Coastguard Worker 	if (ptr + 10 > end) { debugf("GetLargeResourceRecord: Malformed RR -- no type/class/ttl/len!"); return(mDNSNULL); }
2173*48a54d36SAndroid Build Coastguard Worker 
2174*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rrtype            = (mDNSu16) ((mDNSu16)ptr[0] <<  8 | ptr[1]);
2175*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rrclass           = (mDNSu16)(((mDNSu16)ptr[2] <<  8 | ptr[3]) & kDNSClass_Mask);
2176*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rroriginalttl     = (mDNSu32) ((mDNSu32)ptr[4] << 24 | (mDNSu32)ptr[5] << 16 | (mDNSu32)ptr[6] << 8 | ptr[7]);
2177*48a54d36SAndroid Build Coastguard Worker 	if (rr->resrec.rroriginalttl > 0x70000000UL / mDNSPlatformOneSecond && (mDNSs32)rr->resrec.rroriginalttl != -1)
2178*48a54d36SAndroid Build Coastguard Worker 		rr->resrec.rroriginalttl = 0x70000000UL / mDNSPlatformOneSecond;
2179*48a54d36SAndroid Build Coastguard Worker 	// Note: We don't have to adjust m->NextCacheCheck here -- this is just getting a record into memory for
2180*48a54d36SAndroid Build Coastguard Worker 	// us to look at. If we decide to copy it into the cache, then we'll update m->NextCacheCheck accordingly.
2181*48a54d36SAndroid Build Coastguard Worker 	pktrdlength           = (mDNSu16)((mDNSu16)ptr[8] <<  8 | ptr[9]);
2182*48a54d36SAndroid Build Coastguard Worker 
2183*48a54d36SAndroid Build Coastguard Worker 	// If mDNS record has cache-flush bit set, we mark it unique
2184*48a54d36SAndroid Build Coastguard Worker 	// For uDNS records, all are implicitly deemed unique (a single DNS server is always
2185*48a54d36SAndroid Build Coastguard Worker 	// authoritative for the entire RRSet), unless this is a truncated response
2186*48a54d36SAndroid Build Coastguard Worker 	if (ptr[2] & (kDNSClass_UniqueRRSet >> 8) || (!InterfaceID && !(msg->h.flags.b[0] & kDNSFlag0_TC)))
2187*48a54d36SAndroid Build Coastguard Worker 		RecordType |= kDNSRecordTypePacketUniqueMask;
2188*48a54d36SAndroid Build Coastguard Worker 	ptr += 10;
2189*48a54d36SAndroid Build Coastguard Worker 	if (ptr + pktrdlength > end) { debugf("GetLargeResourceRecord: RDATA exceeds end of packet"); return(mDNSNULL); }
2190*48a54d36SAndroid Build Coastguard Worker 	end = ptr + pktrdlength;		// Adjust end to indicate the end of the rdata for this resource record
2191*48a54d36SAndroid Build Coastguard Worker 
2192*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rdata = (RData*)&rr->smallrdatastorage;
2193*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rdata->MaxRDLength = MaximumRDSize;
2194*48a54d36SAndroid Build Coastguard Worker 
2195*48a54d36SAndroid Build Coastguard Worker 	if (!RecordType) LogMsg("GetLargeResourceRecord: No RecordType for %##s", rr->resrec.name->c);
2196*48a54d36SAndroid Build Coastguard Worker 
2197*48a54d36SAndroid Build Coastguard Worker 	// IMPORTANT: Any record type we understand and unpack into a structure containing domainnames needs to have corresponding
2198*48a54d36SAndroid Build Coastguard Worker 	// cases in SameRDataBody() and RDataHashValue() to do a semantic comparison (or checksum) of the structure instead of a blind
2199*48a54d36SAndroid Build Coastguard Worker 	// bitwise memory compare (or sum). This is because a domainname is a fixed size structure holding variable-length data.
2200*48a54d36SAndroid Build Coastguard Worker 	// Any bytes past the logical end of the name are undefined, and a blind bitwise memory compare may indicate that
2201*48a54d36SAndroid Build Coastguard Worker 	// two domainnames are different when semantically they are the same name and it's only the unused bytes that differ.
2202*48a54d36SAndroid Build Coastguard Worker 	if (rr->resrec.rrclass == kDNSQClass_ANY && pktrdlength == 0)	// Used in update packets to mean "Delete An RRset" (RFC 2136)
2203*48a54d36SAndroid Build Coastguard Worker 		rr->resrec.rdlength = 0;
2204*48a54d36SAndroid Build Coastguard Worker 	else switch (rr->resrec.rrtype)
2205*48a54d36SAndroid Build Coastguard Worker 		{
2206*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_A:	if (pktrdlength != sizeof(mDNSv4Addr)) goto fail;
2207*48a54d36SAndroid Build Coastguard Worker 							rdb->ipv4.b[0] = ptr[0];
2208*48a54d36SAndroid Build Coastguard Worker 							rdb->ipv4.b[1] = ptr[1];
2209*48a54d36SAndroid Build Coastguard Worker 							rdb->ipv4.b[2] = ptr[2];
2210*48a54d36SAndroid Build Coastguard Worker 							rdb->ipv4.b[3] = ptr[3];
2211*48a54d36SAndroid Build Coastguard Worker 							break;
2212*48a54d36SAndroid Build Coastguard Worker 
2213*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NS:
2214*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_CNAME:
2215*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PTR:
2216*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DNAME:ptr = getDomainName(msg, ptr, end, &rdb->name);
2217*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed CNAME/PTR RDATA name"); goto fail; }
2218*48a54d36SAndroid Build Coastguard Worker 							//debugf("%##s PTR %##s rdlen %d", rr->resrec.name.c, rdb->name.c, pktrdlength);
2219*48a54d36SAndroid Build Coastguard Worker 							break;
2220*48a54d36SAndroid Build Coastguard Worker 
2221*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SOA:  ptr = getDomainName(msg, ptr, end, &rdb->soa.mname);
2222*48a54d36SAndroid Build Coastguard Worker 							if (!ptr)              { debugf("GetLargeResourceRecord: Malformed SOA RDATA mname"); goto fail; }
2223*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr, end, &rdb->soa.rname);
2224*48a54d36SAndroid Build Coastguard Worker 							if (!ptr)              { debugf("GetLargeResourceRecord: Malformed SOA RDATA rname"); goto fail; }
2225*48a54d36SAndroid Build Coastguard Worker 							if (ptr + 0x14 != end) { debugf("GetLargeResourceRecord: Malformed SOA RDATA");       goto fail; }
2226*48a54d36SAndroid Build Coastguard Worker 							rdb->soa.serial  = (mDNSs32) ((mDNSs32)ptr[0x00] << 24 | (mDNSs32)ptr[0x01] << 16 | (mDNSs32)ptr[0x02] << 8 | ptr[0x03]);
2227*48a54d36SAndroid Build Coastguard Worker 							rdb->soa.refresh = (mDNSu32) ((mDNSu32)ptr[0x04] << 24 | (mDNSu32)ptr[0x05] << 16 | (mDNSu32)ptr[0x06] << 8 | ptr[0x07]);
2228*48a54d36SAndroid Build Coastguard Worker 							rdb->soa.retry   = (mDNSu32) ((mDNSu32)ptr[0x08] << 24 | (mDNSu32)ptr[0x09] << 16 | (mDNSu32)ptr[0x0A] << 8 | ptr[0x0B]);
2229*48a54d36SAndroid Build Coastguard Worker 							rdb->soa.expire  = (mDNSu32) ((mDNSu32)ptr[0x0C] << 24 | (mDNSu32)ptr[0x0D] << 16 | (mDNSu32)ptr[0x0E] << 8 | ptr[0x0F]);
2230*48a54d36SAndroid Build Coastguard Worker 							rdb->soa.min     = (mDNSu32) ((mDNSu32)ptr[0x10] << 24 | (mDNSu32)ptr[0x11] << 16 | (mDNSu32)ptr[0x12] << 8 | ptr[0x13]);
2231*48a54d36SAndroid Build Coastguard Worker 							break;
2232*48a54d36SAndroid Build Coastguard Worker 
2233*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NULL:
2234*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_HINFO:
2235*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TSIG:
2236*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_TXT:
2237*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_X25:
2238*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_ISDN:
2239*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_LOC:
2240*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_DHCID:if (pktrdlength > rr->resrec.rdata->MaxRDLength)
2241*48a54d36SAndroid Build Coastguard Worker 								{
2242*48a54d36SAndroid Build Coastguard Worker 								debugf("GetLargeResourceRecord: %s rdata size (%d) exceeds storage (%d)",
2243*48a54d36SAndroid Build Coastguard Worker 									DNSTypeName(rr->resrec.rrtype), pktrdlength, rr->resrec.rdata->MaxRDLength);
2244*48a54d36SAndroid Build Coastguard Worker 								goto fail;
2245*48a54d36SAndroid Build Coastguard Worker 								}
2246*48a54d36SAndroid Build Coastguard Worker 							rr->resrec.rdlength = pktrdlength;
2247*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(rdb->data, ptr, pktrdlength);
2248*48a54d36SAndroid Build Coastguard Worker 							break;
2249*48a54d36SAndroid Build Coastguard Worker 
2250*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_MX:
2251*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AFSDB:
2252*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RT:
2253*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_KX:	if (pktrdlength < 3) goto fail;	// Preference + domainname
2254*48a54d36SAndroid Build Coastguard Worker 							rdb->mx.preference = (mDNSu16)((mDNSu16)ptr[0] <<  8 | ptr[1]);
2255*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr+2, end, &rdb->mx.exchange);
2256*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed MX name"); goto fail; }
2257*48a54d36SAndroid Build Coastguard Worker 							//debugf("%##s SRV %##s rdlen %d", rr->resrec.name.c, rdb->srv.target.c, pktrdlength);
2258*48a54d36SAndroid Build Coastguard Worker 							break;
2259*48a54d36SAndroid Build Coastguard Worker 
2260*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_RP:	ptr = getDomainName(msg, ptr, end, &rdb->rp.mbox);	// Domainname + domainname
2261*48a54d36SAndroid Build Coastguard Worker 							if (!ptr)       { debugf("GetLargeResourceRecord: Malformed RP mbox"); goto fail; }
2262*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr, end, &rdb->rp.txt);
2263*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed RP txt"); goto fail; }
2264*48a54d36SAndroid Build Coastguard Worker 							break;
2265*48a54d36SAndroid Build Coastguard Worker 
2266*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_PX:	if (pktrdlength < 4) goto fail;	// Preference + domainname + domainname
2267*48a54d36SAndroid Build Coastguard Worker 							rdb->px.preference = (mDNSu16)((mDNSu16)ptr[0] <<  8 | ptr[1]);
2268*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr, end, &rdb->px.map822);
2269*48a54d36SAndroid Build Coastguard Worker 							if (!ptr)       { debugf("GetLargeResourceRecord: Malformed PX map822"); goto fail; }
2270*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr, end, &rdb->px.mapx400);
2271*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed PX mapx400"); goto fail; }
2272*48a54d36SAndroid Build Coastguard Worker 							break;
2273*48a54d36SAndroid Build Coastguard Worker 
2274*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_AAAA:	if (pktrdlength != sizeof(mDNSv6Addr)) goto fail;
2275*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(&rdb->ipv6, ptr, sizeof(rdb->ipv6));
2276*48a54d36SAndroid Build Coastguard Worker 							break;
2277*48a54d36SAndroid Build Coastguard Worker 
2278*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_SRV:	if (pktrdlength < 7) goto fail;	// Priority + weight + port + domainname
2279*48a54d36SAndroid Build Coastguard Worker 							rdb->srv.priority = (mDNSu16)((mDNSu16)ptr[0] <<  8 | ptr[1]);
2280*48a54d36SAndroid Build Coastguard Worker 							rdb->srv.weight   = (mDNSu16)((mDNSu16)ptr[2] <<  8 | ptr[3]);
2281*48a54d36SAndroid Build Coastguard Worker 							rdb->srv.port.b[0] = ptr[4];
2282*48a54d36SAndroid Build Coastguard Worker 							rdb->srv.port.b[1] = ptr[5];
2283*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr+6, end, &rdb->srv.target);
2284*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed SRV RDATA name"); goto fail; }
2285*48a54d36SAndroid Build Coastguard Worker 							//debugf("%##s SRV %##s rdlen %d", rr->resrec.name.c, rdb->srv.target.c, pktrdlength);
2286*48a54d36SAndroid Build Coastguard Worker 							break;
2287*48a54d36SAndroid Build Coastguard Worker 
2288*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_OPT:	{
2289*48a54d36SAndroid Build Coastguard Worker 							rdataOPT *opt = rr->resrec.rdata->u.opt;
2290*48a54d36SAndroid Build Coastguard Worker 							rr->resrec.rdlength = 0;
2291*48a54d36SAndroid Build Coastguard Worker 							while (ptr < end && (mDNSu8 *)(opt+1) < &rr->resrec.rdata->u.data[MaximumRDSize])
2292*48a54d36SAndroid Build Coastguard Worker 								{
2293*48a54d36SAndroid Build Coastguard Worker 								const rdataOPT *const currentopt = opt;
2294*48a54d36SAndroid Build Coastguard Worker 								if (ptr + 4 > end) { LogInfo("GetLargeResourceRecord: OPT RDATA ptr + 4 > end"); goto fail; }
2295*48a54d36SAndroid Build Coastguard Worker 								opt->opt    = (mDNSu16)((mDNSu16)ptr[0] <<  8 | ptr[1]);
2296*48a54d36SAndroid Build Coastguard Worker 								opt->optlen = (mDNSu16)((mDNSu16)ptr[2] <<  8 | ptr[3]);
2297*48a54d36SAndroid Build Coastguard Worker 								ptr += 4;
2298*48a54d36SAndroid Build Coastguard Worker 								if (ptr + opt->optlen > end) { LogInfo("GetLargeResourceRecord: ptr + opt->optlen > end"); goto fail; }
2299*48a54d36SAndroid Build Coastguard Worker 								switch (opt->opt)
2300*48a54d36SAndroid Build Coastguard Worker 									{
2301*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_LLQ:
2302*48a54d36SAndroid Build Coastguard Worker 										if (opt->optlen == DNSOpt_LLQData_Space - 4)
2303*48a54d36SAndroid Build Coastguard Worker 											{
2304*48a54d36SAndroid Build Coastguard Worker 											opt->u.llq.vers  = (mDNSu16)((mDNSu16)ptr[0] <<  8 | ptr[1]);
2305*48a54d36SAndroid Build Coastguard Worker 											opt->u.llq.llqOp = (mDNSu16)((mDNSu16)ptr[2] <<  8 | ptr[3]);
2306*48a54d36SAndroid Build Coastguard Worker 											opt->u.llq.err   = (mDNSu16)((mDNSu16)ptr[4] <<  8 | ptr[5]);
2307*48a54d36SAndroid Build Coastguard Worker 											mDNSPlatformMemCopy(opt->u.llq.id.b, ptr+6, 8);
2308*48a54d36SAndroid Build Coastguard Worker 											opt->u.llq.llqlease = (mDNSu32) ((mDNSu32)ptr[14] << 24 | (mDNSu32)ptr[15] << 16 | (mDNSu32)ptr[16] << 8 | ptr[17]);
2309*48a54d36SAndroid Build Coastguard Worker 											if (opt->u.llq.llqlease > 0x70000000UL / mDNSPlatformOneSecond)
2310*48a54d36SAndroid Build Coastguard Worker 												opt->u.llq.llqlease = 0x70000000UL / mDNSPlatformOneSecond;
2311*48a54d36SAndroid Build Coastguard Worker 											opt++;
2312*48a54d36SAndroid Build Coastguard Worker 											}
2313*48a54d36SAndroid Build Coastguard Worker 										break;
2314*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Lease:
2315*48a54d36SAndroid Build Coastguard Worker 										if (opt->optlen == DNSOpt_LeaseData_Space - 4)
2316*48a54d36SAndroid Build Coastguard Worker 											{
2317*48a54d36SAndroid Build Coastguard Worker 											opt->u.updatelease = (mDNSu32) ((mDNSu32)ptr[0] << 24 | (mDNSu32)ptr[1] << 16 | (mDNSu32)ptr[2] << 8 | ptr[3]);
2318*48a54d36SAndroid Build Coastguard Worker 											if (opt->u.updatelease > 0x70000000UL / mDNSPlatformOneSecond)
2319*48a54d36SAndroid Build Coastguard Worker 												opt->u.updatelease = 0x70000000UL / mDNSPlatformOneSecond;
2320*48a54d36SAndroid Build Coastguard Worker 											opt++;
2321*48a54d36SAndroid Build Coastguard Worker 											}
2322*48a54d36SAndroid Build Coastguard Worker 										break;
2323*48a54d36SAndroid Build Coastguard Worker 									case kDNSOpt_Owner:
2324*48a54d36SAndroid Build Coastguard Worker 										if (ValidOwnerLength(opt->optlen))
2325*48a54d36SAndroid Build Coastguard Worker 											{
2326*48a54d36SAndroid Build Coastguard Worker 											opt->u.owner.vers = ptr[0];
2327*48a54d36SAndroid Build Coastguard Worker 											opt->u.owner.seq  = ptr[1];
2328*48a54d36SAndroid Build Coastguard Worker 											mDNSPlatformMemCopy(opt->u.owner.HMAC.b, ptr+2, 6);		// 6-byte MAC address
2329*48a54d36SAndroid Build Coastguard Worker 											mDNSPlatformMemCopy(opt->u.owner.IMAC.b, ptr+2, 6);		// 6-byte MAC address
2330*48a54d36SAndroid Build Coastguard Worker 											opt->u.owner.password = zeroEthAddr;
2331*48a54d36SAndroid Build Coastguard Worker 											if (opt->optlen >= DNSOpt_OwnerData_ID_Wake_Space-4)
2332*48a54d36SAndroid Build Coastguard Worker 												{
2333*48a54d36SAndroid Build Coastguard Worker 												mDNSPlatformMemCopy(opt->u.owner.IMAC.b, ptr+8, 6);	// 6-byte MAC address
2334*48a54d36SAndroid Build Coastguard Worker 												// This mDNSPlatformMemCopy is safe because the ValidOwnerLength(opt->optlen) check above
2335*48a54d36SAndroid Build Coastguard Worker 												// ensures that opt->optlen is no more than DNSOpt_OwnerData_ID_Wake_PW6_Space - 4
2336*48a54d36SAndroid Build Coastguard Worker 												if (opt->optlen > DNSOpt_OwnerData_ID_Wake_Space-4)
2337*48a54d36SAndroid Build Coastguard Worker 													mDNSPlatformMemCopy(opt->u.owner.password.b, ptr+14, opt->optlen - (DNSOpt_OwnerData_ID_Wake_Space-4));
2338*48a54d36SAndroid Build Coastguard Worker 												}
2339*48a54d36SAndroid Build Coastguard Worker 											opt++;
2340*48a54d36SAndroid Build Coastguard Worker 											}
2341*48a54d36SAndroid Build Coastguard Worker 										break;
2342*48a54d36SAndroid Build Coastguard Worker 									}
2343*48a54d36SAndroid Build Coastguard Worker 								ptr += currentopt->optlen;
2344*48a54d36SAndroid Build Coastguard Worker 								}
2345*48a54d36SAndroid Build Coastguard Worker 							rr->resrec.rdlength = (mDNSu16)((mDNSu8*)opt - rr->resrec.rdata->u.data);
2346*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { LogInfo("GetLargeResourceRecord: Malformed OptRdata"); goto fail; }
2347*48a54d36SAndroid Build Coastguard Worker 							break;
2348*48a54d36SAndroid Build Coastguard Worker 							}
2349*48a54d36SAndroid Build Coastguard Worker 
2350*48a54d36SAndroid Build Coastguard Worker 		case kDNSType_NSEC: {
2351*48a54d36SAndroid Build Coastguard Worker 							unsigned int i, j;
2352*48a54d36SAndroid Build Coastguard Worker 							domainname d;
2353*48a54d36SAndroid Build Coastguard Worker 							ptr = getDomainName(msg, ptr, end, &d);		// Ignored for our simplified use of NSEC synthetic records
2354*48a54d36SAndroid Build Coastguard Worker 							if (!ptr) { LogInfo("GetLargeResourceRecord: Malformed NSEC nextname"); goto fail; }
2355*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemZero(rdb->nsec.bitmap, sizeof(rdb->nsec.bitmap));
2356*48a54d36SAndroid Build Coastguard Worker 							if (ptr < end)
2357*48a54d36SAndroid Build Coastguard Worker 								{
2358*48a54d36SAndroid Build Coastguard Worker 								if (*ptr++ != 0) { debugf("GetLargeResourceRecord: We only handle block zero NSECs"); goto fail; }
2359*48a54d36SAndroid Build Coastguard Worker 								i = *ptr++;
2360*48a54d36SAndroid Build Coastguard Worker 								if (i > sizeof(rdataNSEC)) { debugf("GetLargeResourceRecord: invalid block length %d", i); goto fail; }
2361*48a54d36SAndroid Build Coastguard Worker 								for (j=0; j<i; j++) rdb->nsec.bitmap[j] = *ptr++;
2362*48a54d36SAndroid Build Coastguard Worker 								}
2363*48a54d36SAndroid Build Coastguard Worker 							if (ptr != end) { debugf("GetLargeResourceRecord: Malformed NSEC"); goto fail; }
2364*48a54d36SAndroid Build Coastguard Worker 							break;
2365*48a54d36SAndroid Build Coastguard Worker 							}
2366*48a54d36SAndroid Build Coastguard Worker 
2367*48a54d36SAndroid Build Coastguard Worker 		default:			if (pktrdlength > rr->resrec.rdata->MaxRDLength)
2368*48a54d36SAndroid Build Coastguard Worker 								{
2369*48a54d36SAndroid Build Coastguard Worker 								debugf("GetLargeResourceRecord: rdata %d (%s) size (%d) exceeds storage (%d)",
2370*48a54d36SAndroid Build Coastguard Worker 									rr->resrec.rrtype, DNSTypeName(rr->resrec.rrtype), pktrdlength, rr->resrec.rdata->MaxRDLength);
2371*48a54d36SAndroid Build Coastguard Worker 								goto fail;
2372*48a54d36SAndroid Build Coastguard Worker 								}
2373*48a54d36SAndroid Build Coastguard Worker 							debugf("GetLargeResourceRecord: Warning! Reading resource type %d (%s) as opaque data",
2374*48a54d36SAndroid Build Coastguard Worker 								rr->resrec.rrtype, DNSTypeName(rr->resrec.rrtype));
2375*48a54d36SAndroid Build Coastguard Worker 							// Note: Just because we don't understand the record type, that doesn't
2376*48a54d36SAndroid Build Coastguard Worker 							// mean we fail. The DNS protocol specifies rdlength, so we can
2377*48a54d36SAndroid Build Coastguard Worker 							// safely skip over unknown records and ignore them.
2378*48a54d36SAndroid Build Coastguard Worker 							// We also grab a binary copy of the rdata anyway, since the caller
2379*48a54d36SAndroid Build Coastguard Worker 							// might know how to interpret it even if we don't.
2380*48a54d36SAndroid Build Coastguard Worker 							rr->resrec.rdlength = pktrdlength;
2381*48a54d36SAndroid Build Coastguard Worker 							mDNSPlatformMemCopy(rdb->data, ptr, pktrdlength);
2382*48a54d36SAndroid Build Coastguard Worker 							break;
2383*48a54d36SAndroid Build Coastguard Worker 		}
2384*48a54d36SAndroid Build Coastguard Worker 
2385*48a54d36SAndroid Build Coastguard Worker 	SetNewRData(&rr->resrec, mDNSNULL, 0);		// Sets rdlength, rdestimate, rdatahash for us
2386*48a54d36SAndroid Build Coastguard Worker 
2387*48a54d36SAndroid Build Coastguard Worker 	// Success! Now fill in RecordType to show this record contains valid data
2388*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.RecordType = RecordType;
2389*48a54d36SAndroid Build Coastguard Worker 	return(end);
2390*48a54d36SAndroid Build Coastguard Worker 
2391*48a54d36SAndroid Build Coastguard Worker fail:
2392*48a54d36SAndroid Build Coastguard Worker 	// If we were unable to parse the rdata in this record, we indicate that by
2393*48a54d36SAndroid Build Coastguard Worker 	// returing a 'kDNSRecordTypePacketNegative' record with rdlength set to zero
2394*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.RecordType = kDNSRecordTypePacketNegative;
2395*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rdlength   = 0;
2396*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rdestimate = 0;
2397*48a54d36SAndroid Build Coastguard Worker 	rr->resrec.rdatahash  = 0;
2398*48a54d36SAndroid Build Coastguard Worker 	return(end);
2399*48a54d36SAndroid Build Coastguard Worker 	}
2400*48a54d36SAndroid Build Coastguard Worker 
skipQuestion(const DNSMessage * msg,const mDNSu8 * ptr,const mDNSu8 * end)2401*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *skipQuestion(const DNSMessage *msg, const mDNSu8 *ptr, const mDNSu8 *end)
2402*48a54d36SAndroid Build Coastguard Worker 	{
2403*48a54d36SAndroid Build Coastguard Worker 	ptr = skipDomainName(msg, ptr, end);
2404*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { debugf("skipQuestion: Malformed domain name in DNS question section"); return(mDNSNULL); }
2405*48a54d36SAndroid Build Coastguard Worker 	if (ptr+4 > end) { debugf("skipQuestion: Malformed DNS question section -- no query type and class!"); return(mDNSNULL); }
2406*48a54d36SAndroid Build Coastguard Worker 	return(ptr+4);
2407*48a54d36SAndroid Build Coastguard Worker 	}
2408*48a54d36SAndroid Build Coastguard Worker 
getQuestion(const DNSMessage * msg,const mDNSu8 * ptr,const mDNSu8 * end,const mDNSInterfaceID InterfaceID,DNSQuestion * question)2409*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *getQuestion(const DNSMessage *msg, const mDNSu8 *ptr, const mDNSu8 *end, const mDNSInterfaceID InterfaceID,
2410*48a54d36SAndroid Build Coastguard Worker 	DNSQuestion *question)
2411*48a54d36SAndroid Build Coastguard Worker 	{
2412*48a54d36SAndroid Build Coastguard Worker 	mDNSPlatformMemZero(question, sizeof(*question));
2413*48a54d36SAndroid Build Coastguard Worker 	question->InterfaceID = InterfaceID;
2414*48a54d36SAndroid Build Coastguard Worker 	if (!InterfaceID) question->TargetQID = onesID;	// In DNSQuestions we use TargetQID as the indicator of whether it's unicast or multicast
2415*48a54d36SAndroid Build Coastguard Worker 	ptr = getDomainName(msg, ptr, end, &question->qname);
2416*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) { debugf("Malformed domain name in DNS question section"); return(mDNSNULL); }
2417*48a54d36SAndroid Build Coastguard Worker 	if (ptr+4 > end) { debugf("Malformed DNS question section -- no query type and class!"); return(mDNSNULL); }
2418*48a54d36SAndroid Build Coastguard Worker 
2419*48a54d36SAndroid Build Coastguard Worker 	question->qnamehash = DomainNameHashValue(&question->qname);
2420*48a54d36SAndroid Build Coastguard Worker 	question->qtype  = (mDNSu16)((mDNSu16)ptr[0] << 8 | ptr[1]);			// Get type
2421*48a54d36SAndroid Build Coastguard Worker 	question->qclass = (mDNSu16)((mDNSu16)ptr[2] << 8 | ptr[3]);			// and class
2422*48a54d36SAndroid Build Coastguard Worker 	return(ptr+4);
2423*48a54d36SAndroid Build Coastguard Worker 	}
2424*48a54d36SAndroid Build Coastguard Worker 
LocateAnswers(const DNSMessage * const msg,const mDNSu8 * const end)2425*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *LocateAnswers(const DNSMessage *const msg, const mDNSu8 *const end)
2426*48a54d36SAndroid Build Coastguard Worker 	{
2427*48a54d36SAndroid Build Coastguard Worker 	int i;
2428*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = msg->data;
2429*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; i < msg->h.numQuestions && ptr; i++) ptr = skipQuestion(msg, ptr, end);
2430*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
2431*48a54d36SAndroid Build Coastguard Worker 	}
2432*48a54d36SAndroid Build Coastguard Worker 
LocateAuthorities(const DNSMessage * const msg,const mDNSu8 * const end)2433*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *LocateAuthorities(const DNSMessage *const msg, const mDNSu8 *const end)
2434*48a54d36SAndroid Build Coastguard Worker 	{
2435*48a54d36SAndroid Build Coastguard Worker 	int i;
2436*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = LocateAnswers(msg, end);
2437*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; i < msg->h.numAnswers && ptr; i++) ptr = skipResourceRecord(msg, ptr, end);
2438*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
2439*48a54d36SAndroid Build Coastguard Worker 	}
2440*48a54d36SAndroid Build Coastguard Worker 
LocateAdditionals(const DNSMessage * const msg,const mDNSu8 * const end)2441*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *LocateAdditionals(const DNSMessage *const msg, const mDNSu8 *const end)
2442*48a54d36SAndroid Build Coastguard Worker 	{
2443*48a54d36SAndroid Build Coastguard Worker 	int i;
2444*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = LocateAuthorities(msg, end);
2445*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; i < msg->h.numAuthorities; i++) ptr = skipResourceRecord(msg, ptr, end);
2446*48a54d36SAndroid Build Coastguard Worker 	return (ptr);
2447*48a54d36SAndroid Build Coastguard Worker 	}
2448*48a54d36SAndroid Build Coastguard Worker 
LocateOptRR(const DNSMessage * const msg,const mDNSu8 * const end,int minsize)2449*48a54d36SAndroid Build Coastguard Worker mDNSexport const mDNSu8 *LocateOptRR(const DNSMessage *const msg, const mDNSu8 *const end, int minsize)
2450*48a54d36SAndroid Build Coastguard Worker 	{
2451*48a54d36SAndroid Build Coastguard Worker 	int i;
2452*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = LocateAdditionals(msg, end);
2453*48a54d36SAndroid Build Coastguard Worker 
2454*48a54d36SAndroid Build Coastguard Worker 	// Locate the OPT record.
2455*48a54d36SAndroid Build Coastguard Worker 	// According to RFC 2671, "One OPT pseudo-RR can be added to the additional data section of either a request or a response."
2456*48a54d36SAndroid Build Coastguard Worker 	// This implies that there may be *at most* one OPT record per DNS message, in the Additional Section,
2457*48a54d36SAndroid Build Coastguard Worker 	// but not necessarily the *last* entry in the Additional Section.
2458*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; ptr && i < msg->h.numAdditionals; i++)
2459*48a54d36SAndroid Build Coastguard Worker 		{
2460*48a54d36SAndroid Build Coastguard Worker 		if (ptr + DNSOpt_Header_Space + minsize <= end &&	// Make sure we have 11+minsize bytes of data
2461*48a54d36SAndroid Build Coastguard Worker 			ptr[0] == 0                                &&	// Name must be root label
2462*48a54d36SAndroid Build Coastguard Worker 			ptr[1] == (kDNSType_OPT >> 8  )            &&	// rrtype OPT
2463*48a54d36SAndroid Build Coastguard Worker 			ptr[2] == (kDNSType_OPT & 0xFF)            &&
2464*48a54d36SAndroid Build Coastguard Worker 			((mDNSu16)ptr[9] << 8 | (mDNSu16)ptr[10]) >= (mDNSu16)minsize)
2465*48a54d36SAndroid Build Coastguard Worker 			return(ptr);
2466*48a54d36SAndroid Build Coastguard Worker 		else
2467*48a54d36SAndroid Build Coastguard Worker 			ptr = skipResourceRecord(msg, ptr, end);
2468*48a54d36SAndroid Build Coastguard Worker 		}
2469*48a54d36SAndroid Build Coastguard Worker 	return(mDNSNULL);
2470*48a54d36SAndroid Build Coastguard Worker 	}
2471*48a54d36SAndroid Build Coastguard Worker 
2472*48a54d36SAndroid Build Coastguard Worker // On success, GetLLQOptData returns pointer to storage within shared "m->rec";
2473*48a54d36SAndroid Build Coastguard Worker // it is caller's responsibilty to clear m->rec.r.resrec.RecordType after use
2474*48a54d36SAndroid Build Coastguard Worker // Note: An OPT RDataBody actually contains one or more variable-length rdataOPT objects packed together
2475*48a54d36SAndroid Build Coastguard Worker // The code that currently calls this assumes there's only one, instead of iterating through the set
GetLLQOptData(mDNS * const m,const DNSMessage * const msg,const mDNSu8 * const end)2476*48a54d36SAndroid Build Coastguard Worker mDNSexport const rdataOPT *GetLLQOptData(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *const end)
2477*48a54d36SAndroid Build Coastguard Worker 	{
2478*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = LocateOptRR(msg, end, DNSOpt_LLQData_Space);
2479*48a54d36SAndroid Build Coastguard Worker 	if (ptr)
2480*48a54d36SAndroid Build Coastguard Worker 		{
2481*48a54d36SAndroid Build Coastguard Worker 		ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
2482*48a54d36SAndroid Build Coastguard Worker 		if (ptr && m->rec.r.resrec.RecordType != kDNSRecordTypePacketNegative) return(&m->rec.r.resrec.rdata->u.opt[0]);
2483*48a54d36SAndroid Build Coastguard Worker 		}
2484*48a54d36SAndroid Build Coastguard Worker 	return(mDNSNULL);
2485*48a54d36SAndroid Build Coastguard Worker 	}
2486*48a54d36SAndroid Build Coastguard Worker 
2487*48a54d36SAndroid Build Coastguard Worker // Get the lease life of records in a dynamic update
2488*48a54d36SAndroid Build Coastguard Worker // returns 0 on error or if no lease present
GetPktLease(mDNS * m,DNSMessage * msg,const mDNSu8 * end)2489*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 GetPktLease(mDNS *m, DNSMessage *msg, const mDNSu8 *end)
2490*48a54d36SAndroid Build Coastguard Worker 	{
2491*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 result = 0;
2492*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = LocateOptRR(msg, end, DNSOpt_LeaseData_Space);
2493*48a54d36SAndroid Build Coastguard Worker 	if (ptr) ptr = GetLargeResourceRecord(m, msg, ptr, end, 0, kDNSRecordTypePacketAdd, &m->rec);
2494*48a54d36SAndroid Build Coastguard Worker 	if (ptr && m->rec.r.resrec.rdlength >= DNSOpt_LeaseData_Space && m->rec.r.resrec.rdata->u.opt[0].opt == kDNSOpt_Lease)
2495*48a54d36SAndroid Build Coastguard Worker 		result = m->rec.r.resrec.rdata->u.opt[0].u.updatelease;
2496*48a54d36SAndroid Build Coastguard Worker 	m->rec.r.resrec.RecordType = 0;		// Clear RecordType to show we're not still using it
2497*48a54d36SAndroid Build Coastguard Worker 	return(result);
2498*48a54d36SAndroid Build Coastguard Worker 	}
2499*48a54d36SAndroid Build Coastguard Worker 
DumpRecords(mDNS * const m,const DNSMessage * const msg,const mDNSu8 * ptr,const mDNSu8 * const end,int count,char * label)2500*48a54d36SAndroid Build Coastguard Worker mDNSlocal const mDNSu8 *DumpRecords(mDNS *const m, const DNSMessage *const msg, const mDNSu8 *ptr, const mDNSu8 *const end, int count, char *label)
2501*48a54d36SAndroid Build Coastguard Worker 	{
2502*48a54d36SAndroid Build Coastguard Worker 	int i;
2503*48a54d36SAndroid Build Coastguard Worker 	LogMsg("%2d %s", count, label);
2504*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; i < count && ptr; i++)
2505*48a54d36SAndroid Build Coastguard Worker 		{
2506*48a54d36SAndroid Build Coastguard Worker 		// This puts a LargeCacheRecord on the stack instead of using the shared m->rec storage,
2507*48a54d36SAndroid Build Coastguard Worker 		// but since it's only used for debugging (and probably only on OS X, not on
2508*48a54d36SAndroid Build Coastguard Worker 		// embedded systems) putting a 9kB object on the stack isn't a big problem.
2509*48a54d36SAndroid Build Coastguard Worker 		LargeCacheRecord largecr;
2510*48a54d36SAndroid Build Coastguard Worker 		ptr = GetLargeResourceRecord(m, msg, ptr, end, mDNSInterface_Any, kDNSRecordTypePacketAns, &largecr);
2511*48a54d36SAndroid Build Coastguard Worker 		if (ptr) LogMsg("%2d TTL%8d %s", i, largecr.r.resrec.rroriginalttl, CRDisplayString(m, &largecr.r));
2512*48a54d36SAndroid Build Coastguard Worker 		}
2513*48a54d36SAndroid Build Coastguard Worker 	if (!ptr) LogMsg("ERROR: Premature end of packet data");
2514*48a54d36SAndroid Build Coastguard Worker 	return(ptr);
2515*48a54d36SAndroid Build Coastguard Worker 	}
2516*48a54d36SAndroid Build Coastguard Worker 
2517*48a54d36SAndroid Build Coastguard Worker #define DNS_OP_Name(X) (                              \
2518*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_StdQuery ? ""         :       \
2519*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_Iquery   ? "Iquery "  :       \
2520*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_Status   ? "Status "  :       \
2521*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_Unused3  ? "Unused3 " :       \
2522*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_Notify   ? "Notify "  :       \
2523*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag0_OP_Update   ? "Update "  : "?? " )
2524*48a54d36SAndroid Build Coastguard Worker 
2525*48a54d36SAndroid Build Coastguard Worker #define DNS_RC_Name(X) (                             \
2526*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NoErr    ? "NoErr"    :      \
2527*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_FormErr  ? "FormErr"  :      \
2528*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_ServFail ? "ServFail" :      \
2529*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NXDomain ? "NXDomain" :      \
2530*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NotImpl  ? "NotImpl"  :      \
2531*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_Refused  ? "Refused"  :      \
2532*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_YXDomain ? "YXDomain" :      \
2533*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_YXRRSet  ? "YXRRSet"  :      \
2534*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NXRRSet  ? "NXRRSet"  :      \
2535*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NotAuth  ? "NotAuth"  :      \
2536*48a54d36SAndroid Build Coastguard Worker 	(X) == kDNSFlag1_RC_NotZone  ? "NotZone"  : "??" )
2537*48a54d36SAndroid Build Coastguard Worker 
2538*48a54d36SAndroid Build Coastguard Worker // Note: DumpPacket expects the packet header fields in host byte order, not network byte order
DumpPacket(mDNS * const m,mStatus status,mDNSBool sent,char * transport,const mDNSAddr * srcaddr,mDNSIPPort srcport,const mDNSAddr * dstaddr,mDNSIPPort dstport,const DNSMessage * const msg,const mDNSu8 * const end)2539*48a54d36SAndroid Build Coastguard Worker mDNSexport void DumpPacket(mDNS *const m, mStatus status, mDNSBool sent, char *transport,
2540*48a54d36SAndroid Build Coastguard Worker 	const mDNSAddr *srcaddr, mDNSIPPort srcport,
2541*48a54d36SAndroid Build Coastguard Worker 	const mDNSAddr *dstaddr, mDNSIPPort dstport, const DNSMessage *const msg, const mDNSu8 *const end)
2542*48a54d36SAndroid Build Coastguard Worker 	{
2543*48a54d36SAndroid Build Coastguard Worker 	mDNSBool IsUpdate = ((msg->h.flags.b[0] & kDNSFlag0_OP_Mask) == kDNSFlag0_OP_Update);
2544*48a54d36SAndroid Build Coastguard Worker 	const mDNSu8 *ptr = msg->data;
2545*48a54d36SAndroid Build Coastguard Worker 	int i;
2546*48a54d36SAndroid Build Coastguard Worker 	DNSQuestion q;
2547*48a54d36SAndroid Build Coastguard Worker 	char tbuffer[64], sbuffer[64], dbuffer[64] = "";
2548*48a54d36SAndroid Build Coastguard Worker 	if (!status) tbuffer[mDNS_snprintf(tbuffer, sizeof(tbuffer), sent ? "Sent" : "Received"                        )] = 0;
2549*48a54d36SAndroid Build Coastguard Worker 	else         tbuffer[mDNS_snprintf(tbuffer, sizeof(tbuffer), "ERROR %d %sing", status, sent ? "Send" : "Receiv")] = 0;
2550*48a54d36SAndroid Build Coastguard Worker 	if (sent) sbuffer[mDNS_snprintf(sbuffer, sizeof(sbuffer), "port "        )] = 0;
2551*48a54d36SAndroid Build Coastguard Worker 	else      sbuffer[mDNS_snprintf(sbuffer, sizeof(sbuffer), "%#a:", srcaddr)] = 0;
2552*48a54d36SAndroid Build Coastguard Worker 	if (dstaddr || !mDNSIPPortIsZero(dstport))
2553*48a54d36SAndroid Build Coastguard Worker 		dbuffer[mDNS_snprintf(dbuffer, sizeof(dbuffer), " to %#a:%d", dstaddr, mDNSVal16(dstport))] = 0;
2554*48a54d36SAndroid Build Coastguard Worker 
2555*48a54d36SAndroid Build Coastguard Worker 	LogMsg("-- %s %s DNS %s%s (flags %02X%02X) RCODE: %s (%d) %s%s%s%s%s%sID: %d %d bytes from %s%d%s%s --",
2556*48a54d36SAndroid Build Coastguard Worker 		tbuffer, transport,
2557*48a54d36SAndroid Build Coastguard Worker 		DNS_OP_Name(msg->h.flags.b[0] & kDNSFlag0_OP_Mask),
2558*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[0] & kDNSFlag0_QR_Response ? "Response" : "Query",
2559*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[0], msg->h.flags.b[1],
2560*48a54d36SAndroid Build Coastguard Worker 		DNS_RC_Name(msg->h.flags.b[1] & kDNSFlag1_RC_Mask),
2561*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[1] & kDNSFlag1_RC_Mask,
2562*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[0] & kDNSFlag0_AA ? "AA " : "",
2563*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[0] & kDNSFlag0_TC ? "TC " : "",
2564*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[0] & kDNSFlag0_RD ? "RD " : "",
2565*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[1] & kDNSFlag1_RA ? "RA " : "",
2566*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[1] & kDNSFlag1_AD ? "AD " : "",
2567*48a54d36SAndroid Build Coastguard Worker 		msg->h.flags.b[1] & kDNSFlag1_CD ? "CD " : "",
2568*48a54d36SAndroid Build Coastguard Worker 		mDNSVal16(msg->h.id),
2569*48a54d36SAndroid Build Coastguard Worker 		end - msg->data,
2570*48a54d36SAndroid Build Coastguard Worker 		sbuffer, mDNSVal16(srcport), dbuffer,
2571*48a54d36SAndroid Build Coastguard Worker 		(msg->h.flags.b[0] & kDNSFlag0_TC) ? " (truncated)" : ""
2572*48a54d36SAndroid Build Coastguard Worker 		);
2573*48a54d36SAndroid Build Coastguard Worker 
2574*48a54d36SAndroid Build Coastguard Worker 	LogMsg("%2d %s", msg->h.numQuestions, IsUpdate ? "Zone" : "Questions");
2575*48a54d36SAndroid Build Coastguard Worker 	for (i = 0; i < msg->h.numQuestions && ptr; i++)
2576*48a54d36SAndroid Build Coastguard Worker 		{
2577*48a54d36SAndroid Build Coastguard Worker 		ptr = getQuestion(msg, ptr, end, mDNSInterface_Any, &q);
2578*48a54d36SAndroid Build Coastguard Worker 		if (ptr) LogMsg("%2d %##s %s", i, q.qname.c, DNSTypeName(q.qtype));
2579*48a54d36SAndroid Build Coastguard Worker 		}
2580*48a54d36SAndroid Build Coastguard Worker 	ptr = DumpRecords(m, msg, ptr, end, msg->h.numAnswers,     IsUpdate ? "Prerequisites" : "Answers");
2581*48a54d36SAndroid Build Coastguard Worker 	ptr = DumpRecords(m, msg, ptr, end, msg->h.numAuthorities, IsUpdate ? "Updates"       : "Authorities");
2582*48a54d36SAndroid Build Coastguard Worker 	ptr = DumpRecords(m, msg, ptr, end, msg->h.numAdditionals, "Additionals");
2583*48a54d36SAndroid Build Coastguard Worker 	LogMsg("--------------");
2584*48a54d36SAndroid Build Coastguard Worker 	}
2585*48a54d36SAndroid Build Coastguard Worker 
2586*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2587*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2588*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2589*48a54d36SAndroid Build Coastguard Worker #pragma mark - Packet Sending Functions
2590*48a54d36SAndroid Build Coastguard Worker #endif
2591*48a54d36SAndroid Build Coastguard Worker 
2592*48a54d36SAndroid Build Coastguard Worker // Stub definition of TCPSocket_struct so we can access flags field. (Rest of TCPSocket_struct is platform-dependent.)
2593*48a54d36SAndroid Build Coastguard Worker struct TCPSocket_struct { TCPSocketFlags flags; /* ... */ };
2594*48a54d36SAndroid Build Coastguard Worker 
2595*48a54d36SAndroid Build Coastguard Worker struct UDPSocket_struct
2596*48a54d36SAndroid Build Coastguard Worker 	{
2597*48a54d36SAndroid Build Coastguard Worker 	mDNSIPPort port; // MUST BE FIRST FIELD -- mDNSCoreReceive expects every UDPSocket_struct to begin with mDNSIPPort port
2598*48a54d36SAndroid Build Coastguard Worker 	};
2599*48a54d36SAndroid Build Coastguard Worker 
2600*48a54d36SAndroid Build Coastguard Worker // Note: When we sign a DNS message using DNSDigest_SignMessage(), the current real-time clock value is used, which
2601*48a54d36SAndroid Build Coastguard Worker // is why we generally defer signing until we send the message, to ensure the signature is as fresh as possible.
mDNSSendDNSMessage(mDNS * const m,DNSMessage * const msg,mDNSu8 * end,mDNSInterfaceID InterfaceID,UDPSocket * src,const mDNSAddr * dst,mDNSIPPort dstport,TCPSocket * sock,DomainAuthInfo * authInfo)2602*48a54d36SAndroid Build Coastguard Worker mDNSexport mStatus mDNSSendDNSMessage(mDNS *const m, DNSMessage *const msg, mDNSu8 *end,
2603*48a54d36SAndroid Build Coastguard Worker     mDNSInterfaceID InterfaceID, UDPSocket *src, const mDNSAddr *dst, mDNSIPPort dstport, TCPSocket *sock, DomainAuthInfo *authInfo)
2604*48a54d36SAndroid Build Coastguard Worker 	{
2605*48a54d36SAndroid Build Coastguard Worker 	mStatus status = mStatus_NoError;
2606*48a54d36SAndroid Build Coastguard Worker 	const mDNSu16 numAdditionals = msg->h.numAdditionals;
2607*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *newend;
2608*48a54d36SAndroid Build Coastguard Worker 	mDNSu8 *limit = msg->data + AbsoluteMaxDNSMessageData;
2609*48a54d36SAndroid Build Coastguard Worker 
2610*48a54d36SAndroid Build Coastguard Worker 	// Zero-length message data is okay (e.g. for a DNS Update ack, where all we need is an ID and an error code
2611*48a54d36SAndroid Build Coastguard Worker 	if (end < msg->data || end - msg->data > AbsoluteMaxDNSMessageData)
2612*48a54d36SAndroid Build Coastguard Worker 		{
2613*48a54d36SAndroid Build Coastguard Worker 		LogMsg("mDNSSendDNSMessage: invalid message %p %p %d", msg->data, end, end - msg->data);
2614*48a54d36SAndroid Build Coastguard Worker 		return mStatus_BadParamErr;
2615*48a54d36SAndroid Build Coastguard Worker 		}
2616*48a54d36SAndroid Build Coastguard Worker 
2617*48a54d36SAndroid Build Coastguard Worker 	newend = putHINFO(m, msg, end, authInfo, limit);
2618*48a54d36SAndroid Build Coastguard Worker 	if (!newend) LogMsg("mDNSSendDNSMessage: putHINFO failed msg %p end %p, limit %p", msg->data, end, limit); // Not fatal
2619*48a54d36SAndroid Build Coastguard Worker 	else end = newend;
2620*48a54d36SAndroid Build Coastguard Worker 
2621*48a54d36SAndroid Build Coastguard Worker 	// Put all the integer values in IETF byte-order (MSB first, LSB second)
2622*48a54d36SAndroid Build Coastguard Worker 	SwapDNSHeaderBytes(msg);
2623*48a54d36SAndroid Build Coastguard Worker 
2624*48a54d36SAndroid Build Coastguard Worker 	if (authInfo) DNSDigest_SignMessage(msg, &end, authInfo, 0);	// DNSDigest_SignMessage operates on message in network byte order
2625*48a54d36SAndroid Build Coastguard Worker 	if (!end) { LogMsg("mDNSSendDNSMessage: DNSDigest_SignMessage failed"); status = mStatus_NoMemoryErr; }
2626*48a54d36SAndroid Build Coastguard Worker 	else
2627*48a54d36SAndroid Build Coastguard Worker 		{
2628*48a54d36SAndroid Build Coastguard Worker 		// Send the packet on the wire
2629*48a54d36SAndroid Build Coastguard Worker 		if (!sock)
2630*48a54d36SAndroid Build Coastguard Worker 			status = mDNSPlatformSendUDP(m, msg, end, InterfaceID, src, dst, dstport);
2631*48a54d36SAndroid Build Coastguard Worker 		else
2632*48a54d36SAndroid Build Coastguard Worker 			{
2633*48a54d36SAndroid Build Coastguard Worker 			mDNSu16 msglen = (mDNSu16)(end - (mDNSu8 *)msg);
2634*48a54d36SAndroid Build Coastguard Worker 			mDNSu8 lenbuf[2] = { (mDNSu8)(msglen >> 8), (mDNSu8)(msglen & 0xFF) };
2635*48a54d36SAndroid Build Coastguard Worker 			long nsent = mDNSPlatformWriteTCP(sock, (char*)lenbuf, 2);		// Should do scatter/gather here -- this is probably going out as two packets
2636*48a54d36SAndroid Build Coastguard Worker 			if (nsent != 2) { LogMsg("mDNSSendDNSMessage: write msg length failed %d/%d", nsent, 2); status = mStatus_ConnFailed; }
2637*48a54d36SAndroid Build Coastguard Worker 			else
2638*48a54d36SAndroid Build Coastguard Worker 				{
2639*48a54d36SAndroid Build Coastguard Worker 				nsent = mDNSPlatformWriteTCP(sock, (char *)msg, msglen);
2640*48a54d36SAndroid Build Coastguard Worker 				if (nsent != msglen) { LogMsg("mDNSSendDNSMessage: write msg body failed %d/%d", nsent, msglen); status = mStatus_ConnFailed; }
2641*48a54d36SAndroid Build Coastguard Worker 				}
2642*48a54d36SAndroid Build Coastguard Worker 			}
2643*48a54d36SAndroid Build Coastguard Worker 		}
2644*48a54d36SAndroid Build Coastguard Worker 
2645*48a54d36SAndroid Build Coastguard Worker 	// Swap the integer values back the way they were (remember that numAdditionals may have been changed by putHINFO and/or SignMessage)
2646*48a54d36SAndroid Build Coastguard Worker 	SwapDNSHeaderBytes(msg);
2647*48a54d36SAndroid Build Coastguard Worker 
2648*48a54d36SAndroid Build Coastguard Worker 	// Dump the packet with the HINFO and TSIG
2649*48a54d36SAndroid Build Coastguard Worker 	if (mDNS_PacketLoggingEnabled && !mDNSOpaque16IsZero(msg->h.id))
2650*48a54d36SAndroid Build Coastguard Worker 		DumpPacket(m, status, mDNStrue, sock && (sock->flags & kTCPSocketFlags_UseTLS) ? "TLS" : sock ? "TCP" : "UDP", mDNSNULL, src ? src->port : MulticastDNSPort, dst, dstport, msg, end);
2651*48a54d36SAndroid Build Coastguard Worker 
2652*48a54d36SAndroid Build Coastguard Worker 	// put the number of additionals back the way it was
2653*48a54d36SAndroid Build Coastguard Worker 	msg->h.numAdditionals = numAdditionals;
2654*48a54d36SAndroid Build Coastguard Worker 
2655*48a54d36SAndroid Build Coastguard Worker 	return(status);
2656*48a54d36SAndroid Build Coastguard Worker 	}
2657*48a54d36SAndroid Build Coastguard Worker 
2658*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2659*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2660*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2661*48a54d36SAndroid Build Coastguard Worker #pragma mark - RR List Management & Task Management
2662*48a54d36SAndroid Build Coastguard Worker #endif
2663*48a54d36SAndroid Build Coastguard Worker 
mDNS_Lock_(mDNS * const m,const char * const functionname)2664*48a54d36SAndroid Build Coastguard Worker mDNSexport void mDNS_Lock_(mDNS *const m, const char * const functionname)
2665*48a54d36SAndroid Build Coastguard Worker 	{
2666*48a54d36SAndroid Build Coastguard Worker 	// MUST grab the platform lock FIRST!
2667*48a54d36SAndroid Build Coastguard Worker 	mDNSPlatformLock(m);
2668*48a54d36SAndroid Build Coastguard Worker 
2669*48a54d36SAndroid Build Coastguard Worker 	// Normally, mDNS_reentrancy is zero and so is mDNS_busy
2670*48a54d36SAndroid Build Coastguard Worker 	// However, when we call a client callback mDNS_busy is one, and we increment mDNS_reentrancy too
2671*48a54d36SAndroid Build Coastguard Worker 	// If that client callback does mDNS API calls, mDNS_reentrancy and mDNS_busy will both be one
2672*48a54d36SAndroid Build Coastguard Worker 	// If mDNS_busy != mDNS_reentrancy that's a bad sign
2673*48a54d36SAndroid Build Coastguard Worker 	if (m->mDNS_busy != m->mDNS_reentrancy)
2674*48a54d36SAndroid Build Coastguard Worker 		{
2675*48a54d36SAndroid Build Coastguard Worker 		LogMsg("%s: mDNS_Lock: Locking failure! mDNS_busy (%ld) != mDNS_reentrancy (%ld)", functionname, m->mDNS_busy, m->mDNS_reentrancy);
2676*48a54d36SAndroid Build Coastguard Worker #if ForceAlerts
2677*48a54d36SAndroid Build Coastguard Worker 		*(long*)0 = 0;
2678*48a54d36SAndroid Build Coastguard Worker #endif
2679*48a54d36SAndroid Build Coastguard Worker 		}
2680*48a54d36SAndroid Build Coastguard Worker 
2681*48a54d36SAndroid Build Coastguard Worker 	// If this is an initial entry into the mDNSCore code, set m->timenow
2682*48a54d36SAndroid Build Coastguard Worker 	// else, if this is a re-entrant entry into the mDNSCore code, m->timenow should already be set
2683*48a54d36SAndroid Build Coastguard Worker 	if (m->mDNS_busy == 0)
2684*48a54d36SAndroid Build Coastguard Worker 		{
2685*48a54d36SAndroid Build Coastguard Worker 		if (m->timenow)
2686*48a54d36SAndroid Build Coastguard Worker 			LogMsg("%s: mDNS_Lock: m->timenow already set (%ld/%ld)", functionname, m->timenow, mDNS_TimeNow_NoLock(m));
2687*48a54d36SAndroid Build Coastguard Worker 		m->timenow = mDNS_TimeNow_NoLock(m);
2688*48a54d36SAndroid Build Coastguard Worker 		if (m->timenow == 0) m->timenow = 1;
2689*48a54d36SAndroid Build Coastguard Worker 		}
2690*48a54d36SAndroid Build Coastguard Worker 	else if (m->timenow == 0)
2691*48a54d36SAndroid Build Coastguard Worker 		{
2692*48a54d36SAndroid Build Coastguard Worker 		LogMsg("%s: mDNS_Lock: m->mDNS_busy is %ld but m->timenow not set", functionname, m->mDNS_busy);
2693*48a54d36SAndroid Build Coastguard Worker 		m->timenow = mDNS_TimeNow_NoLock(m);
2694*48a54d36SAndroid Build Coastguard Worker 		if (m->timenow == 0) m->timenow = 1;
2695*48a54d36SAndroid Build Coastguard Worker 		}
2696*48a54d36SAndroid Build Coastguard Worker 
2697*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow_last - m->timenow > 0)
2698*48a54d36SAndroid Build Coastguard Worker 		{
2699*48a54d36SAndroid Build Coastguard Worker 		m->timenow_adjust += m->timenow_last - m->timenow;
2700*48a54d36SAndroid Build Coastguard Worker 		LogMsg("%s: mDNSPlatformRawTime went backwards by %ld ticks; setting correction factor to %ld", functionname, m->timenow_last - m->timenow, m->timenow_adjust);
2701*48a54d36SAndroid Build Coastguard Worker 		m->timenow = m->timenow_last;
2702*48a54d36SAndroid Build Coastguard Worker 		}
2703*48a54d36SAndroid Build Coastguard Worker 	m->timenow_last = m->timenow;
2704*48a54d36SAndroid Build Coastguard Worker 
2705*48a54d36SAndroid Build Coastguard Worker 	// Increment mDNS_busy so we'll recognise re-entrant calls
2706*48a54d36SAndroid Build Coastguard Worker 	m->mDNS_busy++;
2707*48a54d36SAndroid Build Coastguard Worker 	}
2708*48a54d36SAndroid Build Coastguard Worker 
AnyLocalRecordReady(const mDNS * const m)2709*48a54d36SAndroid Build Coastguard Worker mDNSlocal AuthRecord *AnyLocalRecordReady(const mDNS *const m)
2710*48a54d36SAndroid Build Coastguard Worker 	{
2711*48a54d36SAndroid Build Coastguard Worker 	AuthRecord *rr;
2712*48a54d36SAndroid Build Coastguard Worker 	for (rr = m->NewLocalRecords; rr; rr = rr->next)
2713*48a54d36SAndroid Build Coastguard Worker 		if (LocalRecordReady(rr)) return rr;
2714*48a54d36SAndroid Build Coastguard Worker 	return mDNSNULL;
2715*48a54d36SAndroid Build Coastguard Worker 	}
2716*48a54d36SAndroid Build Coastguard Worker 
GetNextScheduledEvent(const mDNS * const m)2717*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSs32 GetNextScheduledEvent(const mDNS *const m)
2718*48a54d36SAndroid Build Coastguard Worker 	{
2719*48a54d36SAndroid Build Coastguard Worker 	mDNSs32 e = m->timenow + 0x78000000;
2720*48a54d36SAndroid Build Coastguard Worker 	if (m->mDNSPlatformStatus != mStatus_NoError) return(e);
2721*48a54d36SAndroid Build Coastguard Worker 	if (m->NewQuestions)
2722*48a54d36SAndroid Build Coastguard Worker 		{
2723*48a54d36SAndroid Build Coastguard Worker 		if (m->NewQuestions->DelayAnswering) e = m->NewQuestions->DelayAnswering;
2724*48a54d36SAndroid Build Coastguard Worker 		else return(m->timenow);
2725*48a54d36SAndroid Build Coastguard Worker 		}
2726*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalOnlyQuestions)                     return(m->timenow);
2727*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalRecords && AnyLocalRecordReady(m)) return(m->timenow);
2728*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalOnlyRecords)                       return(m->timenow);
2729*48a54d36SAndroid Build Coastguard Worker 	if (m->SPSProxyListChanged)                       return(m->timenow);
2730*48a54d36SAndroid Build Coastguard Worker 	if (m->LocalRemoveEvents)                         return(m->timenow);
2731*48a54d36SAndroid Build Coastguard Worker 
2732*48a54d36SAndroid Build Coastguard Worker #ifndef UNICAST_DISABLED
2733*48a54d36SAndroid Build Coastguard Worker 	if (e - m->NextuDNSEvent         > 0) e = m->NextuDNSEvent;
2734*48a54d36SAndroid Build Coastguard Worker 	if (e - m->NextScheduledNATOp    > 0) e = m->NextScheduledNATOp;
2735*48a54d36SAndroid Build Coastguard Worker 	if (m->NextSRVUpdate && e - m->NextSRVUpdate > 0) e = m->NextSRVUpdate;
2736*48a54d36SAndroid Build Coastguard Worker #endif
2737*48a54d36SAndroid Build Coastguard Worker 
2738*48a54d36SAndroid Build Coastguard Worker 	if (e - m->NextCacheCheck        > 0) e = m->NextCacheCheck;
2739*48a54d36SAndroid Build Coastguard Worker 	if (e - m->NextScheduledSPS      > 0) e = m->NextScheduledSPS;
2740*48a54d36SAndroid Build Coastguard Worker 	// NextScheduledSPRetry only valid when DelaySleep not set
2741*48a54d36SAndroid Build Coastguard Worker 	if (!m->DelaySleep && m->SleepLimit && e - m->NextScheduledSPRetry > 0) e = m->NextScheduledSPRetry;
2742*48a54d36SAndroid Build Coastguard Worker 	if (m->DelaySleep && e - m->DelaySleep > 0) e = m->DelaySleep;
2743*48a54d36SAndroid Build Coastguard Worker 
2744*48a54d36SAndroid Build Coastguard Worker 	if (m->SuppressSending)
2745*48a54d36SAndroid Build Coastguard Worker 		{
2746*48a54d36SAndroid Build Coastguard Worker 		if (e - m->SuppressSending       > 0) e = m->SuppressSending;
2747*48a54d36SAndroid Build Coastguard Worker 		}
2748*48a54d36SAndroid Build Coastguard Worker 	else
2749*48a54d36SAndroid Build Coastguard Worker 		{
2750*48a54d36SAndroid Build Coastguard Worker 		if (e - m->NextScheduledQuery    > 0) e = m->NextScheduledQuery;
2751*48a54d36SAndroid Build Coastguard Worker 		if (e - m->NextScheduledProbe    > 0) e = m->NextScheduledProbe;
2752*48a54d36SAndroid Build Coastguard Worker 		if (e - m->NextScheduledResponse > 0) e = m->NextScheduledResponse;
2753*48a54d36SAndroid Build Coastguard Worker 		}
2754*48a54d36SAndroid Build Coastguard Worker 	if (e - m->NextScheduledStopTime > 0) e = m->NextScheduledStopTime;
2755*48a54d36SAndroid Build Coastguard Worker 	return(e);
2756*48a54d36SAndroid Build Coastguard Worker 	}
2757*48a54d36SAndroid Build Coastguard Worker 
ShowTaskSchedulingError(mDNS * const m)2758*48a54d36SAndroid Build Coastguard Worker mDNSexport void ShowTaskSchedulingError(mDNS *const m)
2759*48a54d36SAndroid Build Coastguard Worker 	{
2760*48a54d36SAndroid Build Coastguard Worker 	AuthRecord *rr;
2761*48a54d36SAndroid Build Coastguard Worker 	mDNS_Lock(m);
2762*48a54d36SAndroid Build Coastguard Worker 
2763*48a54d36SAndroid Build Coastguard Worker 	LogMsg("Task Scheduling Error: Continuously busy for more than a second");
2764*48a54d36SAndroid Build Coastguard Worker 
2765*48a54d36SAndroid Build Coastguard Worker 	// Note: To accurately diagnose *why* we're busy, the debugging code here needs to mirror the logic in GetNextScheduledEvent above
2766*48a54d36SAndroid Build Coastguard Worker 
2767*48a54d36SAndroid Build Coastguard Worker 	if (m->NewQuestions && (!m->NewQuestions->DelayAnswering || m->timenow - m->NewQuestions->DelayAnswering >= 0))
2768*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: NewQuestion %##s (%s)",
2769*48a54d36SAndroid Build Coastguard Worker 			m->NewQuestions->qname.c, DNSTypeName(m->NewQuestions->qtype));
2770*48a54d36SAndroid Build Coastguard Worker 
2771*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalOnlyQuestions)
2772*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: NewLocalOnlyQuestions %##s (%s)",
2773*48a54d36SAndroid Build Coastguard Worker 			m->NewLocalOnlyQuestions->qname.c, DNSTypeName(m->NewLocalOnlyQuestions->qtype));
2774*48a54d36SAndroid Build Coastguard Worker 
2775*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalRecords)
2776*48a54d36SAndroid Build Coastguard Worker 		{
2777*48a54d36SAndroid Build Coastguard Worker 		rr = AnyLocalRecordReady(m);
2778*48a54d36SAndroid Build Coastguard Worker 		if (rr) LogMsg("Task Scheduling Error: NewLocalRecords %s", ARDisplayString(m, rr));
2779*48a54d36SAndroid Build Coastguard Worker 		}
2780*48a54d36SAndroid Build Coastguard Worker 
2781*48a54d36SAndroid Build Coastguard Worker 	if (m->NewLocalOnlyRecords) LogMsg("Task Scheduling Error: NewLocalOnlyRecords");
2782*48a54d36SAndroid Build Coastguard Worker 
2783*48a54d36SAndroid Build Coastguard Worker 	if (m->SPSProxyListChanged) LogMsg("Task Scheduling Error: SPSProxyListChanged");
2784*48a54d36SAndroid Build Coastguard Worker 	if (m->LocalRemoveEvents)   LogMsg("Task Scheduling Error: LocalRemoveEvents");
2785*48a54d36SAndroid Build Coastguard Worker 
2786*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledEvent    >= 0)
2787*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledEvent %d",    m->timenow - m->NextScheduledEvent);
2788*48a54d36SAndroid Build Coastguard Worker 
2789*48a54d36SAndroid Build Coastguard Worker #ifndef UNICAST_DISABLED
2790*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextuDNSEvent         >= 0)
2791*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextuDNSEvent %d",         m->timenow - m->NextuDNSEvent);
2792*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledNATOp    >= 0)
2793*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledNATOp %d",    m->timenow - m->NextScheduledNATOp);
2794*48a54d36SAndroid Build Coastguard Worker 	if (m->NextSRVUpdate && m->timenow - m->NextSRVUpdate >= 0)
2795*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextSRVUpdate %d",         m->timenow - m->NextSRVUpdate);
2796*48a54d36SAndroid Build Coastguard Worker #endif
2797*48a54d36SAndroid Build Coastguard Worker 
2798*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextCacheCheck        >= 0)
2799*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextCacheCheck %d",        m->timenow - m->NextCacheCheck);
2800*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledSPS      >= 0)
2801*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledSPS %d",      m->timenow - m->NextScheduledSPS);
2802*48a54d36SAndroid Build Coastguard Worker 	if (!m->DelaySleep && m->SleepLimit && m->timenow - m->NextScheduledSPRetry >= 0)
2803*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledSPRetry %d",  m->timenow - m->NextScheduledSPRetry);
2804*48a54d36SAndroid Build Coastguard Worker 	if (m->DelaySleep && m->timenow - m->DelaySleep >= 0)
2805*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->DelaySleep %d",            m->timenow - m->DelaySleep);
2806*48a54d36SAndroid Build Coastguard Worker 
2807*48a54d36SAndroid Build Coastguard Worker 	if (m->SuppressSending && m->timenow - m->SuppressSending >= 0)
2808*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->SuppressSending %d",       m->timenow - m->SuppressSending);
2809*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledQuery    >= 0)
2810*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledQuery %d",    m->timenow - m->NextScheduledQuery);
2811*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledProbe    >= 0)
2812*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledProbe %d",    m->timenow - m->NextScheduledProbe);
2813*48a54d36SAndroid Build Coastguard Worker 	if (m->timenow - m->NextScheduledResponse >= 0)
2814*48a54d36SAndroid Build Coastguard Worker 		LogMsg("Task Scheduling Error: m->NextScheduledResponse %d", m->timenow - m->NextScheduledResponse);
2815*48a54d36SAndroid Build Coastguard Worker 
2816*48a54d36SAndroid Build Coastguard Worker 	mDNS_Unlock(m);
2817*48a54d36SAndroid Build Coastguard Worker 	}
2818*48a54d36SAndroid Build Coastguard Worker 
mDNS_Unlock_(mDNS * const m,const char * const functionname)2819*48a54d36SAndroid Build Coastguard Worker mDNSexport void mDNS_Unlock_(mDNS *const m, const char * const functionname)
2820*48a54d36SAndroid Build Coastguard Worker 	{
2821*48a54d36SAndroid Build Coastguard Worker 	// Decrement mDNS_busy
2822*48a54d36SAndroid Build Coastguard Worker 	m->mDNS_busy--;
2823*48a54d36SAndroid Build Coastguard Worker 
2824*48a54d36SAndroid Build Coastguard Worker 	// Check for locking failures
2825*48a54d36SAndroid Build Coastguard Worker 	if (m->mDNS_busy != m->mDNS_reentrancy)
2826*48a54d36SAndroid Build Coastguard Worker 		{
2827*48a54d36SAndroid Build Coastguard Worker 		LogMsg("%s: mDNS_Unlock: Locking failure! mDNS_busy (%ld) != mDNS_reentrancy (%ld)", functionname, m->mDNS_busy, m->mDNS_reentrancy);
2828*48a54d36SAndroid Build Coastguard Worker #if ForceAlerts
2829*48a54d36SAndroid Build Coastguard Worker 		*(long*)0 = 0;
2830*48a54d36SAndroid Build Coastguard Worker #endif
2831*48a54d36SAndroid Build Coastguard Worker 		}
2832*48a54d36SAndroid Build Coastguard Worker 
2833*48a54d36SAndroid Build Coastguard Worker 	// If this is a final exit from the mDNSCore code, set m->NextScheduledEvent and clear m->timenow
2834*48a54d36SAndroid Build Coastguard Worker 	if (m->mDNS_busy == 0)
2835*48a54d36SAndroid Build Coastguard Worker 		{
2836*48a54d36SAndroid Build Coastguard Worker 		m->NextScheduledEvent = GetNextScheduledEvent(m);
2837*48a54d36SAndroid Build Coastguard Worker 		if (m->timenow == 0) LogMsg("%s: mDNS_Unlock: ERROR! m->timenow aready zero", functionname);
2838*48a54d36SAndroid Build Coastguard Worker 		m->timenow = 0;
2839*48a54d36SAndroid Build Coastguard Worker 		}
2840*48a54d36SAndroid Build Coastguard Worker 
2841*48a54d36SAndroid Build Coastguard Worker 	// MUST release the platform lock LAST!
2842*48a54d36SAndroid Build Coastguard Worker 	mDNSPlatformUnlock(m);
2843*48a54d36SAndroid Build Coastguard Worker 	}
2844*48a54d36SAndroid Build Coastguard Worker 
2845*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2846*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2847*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2848*48a54d36SAndroid Build Coastguard Worker #pragma mark - Specialized mDNS version of vsnprintf
2849*48a54d36SAndroid Build Coastguard Worker #endif
2850*48a54d36SAndroid Build Coastguard Worker 
2851*48a54d36SAndroid Build Coastguard Worker static const struct mDNSprintf_format
2852*48a54d36SAndroid Build Coastguard Worker 	{
2853*48a54d36SAndroid Build Coastguard Worker 	unsigned      leftJustify : 1;
2854*48a54d36SAndroid Build Coastguard Worker 	unsigned      forceSign : 1;
2855*48a54d36SAndroid Build Coastguard Worker 	unsigned      zeroPad : 1;
2856*48a54d36SAndroid Build Coastguard Worker 	unsigned      havePrecision : 1;
2857*48a54d36SAndroid Build Coastguard Worker 	unsigned      hSize : 1;
2858*48a54d36SAndroid Build Coastguard Worker 	unsigned      lSize : 1;
2859*48a54d36SAndroid Build Coastguard Worker 	char          altForm;
2860*48a54d36SAndroid Build Coastguard Worker 	char          sign;		// +, - or space
2861*48a54d36SAndroid Build Coastguard Worker 	unsigned int  fieldWidth;
2862*48a54d36SAndroid Build Coastguard Worker 	unsigned int  precision;
2863*48a54d36SAndroid Build Coastguard Worker 	} mDNSprintf_format_default = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 };
2864*48a54d36SAndroid Build Coastguard Worker 
mDNS_vsnprintf(char * sbuffer,mDNSu32 buflen,const char * fmt,va_list arg)2865*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 mDNS_vsnprintf(char *sbuffer, mDNSu32 buflen, const char *fmt, va_list arg)
2866*48a54d36SAndroid Build Coastguard Worker 	{
2867*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 nwritten = 0;
2868*48a54d36SAndroid Build Coastguard Worker 	int c;
2869*48a54d36SAndroid Build Coastguard Worker 	if (buflen == 0) return(0);
2870*48a54d36SAndroid Build Coastguard Worker 	buflen--;		// Pre-reserve one space in the buffer for the terminating null
2871*48a54d36SAndroid Build Coastguard Worker 	if (buflen == 0) goto exit;
2872*48a54d36SAndroid Build Coastguard Worker 
2873*48a54d36SAndroid Build Coastguard Worker 	for (c = *fmt; c != 0; c = *++fmt)
2874*48a54d36SAndroid Build Coastguard Worker 		{
2875*48a54d36SAndroid Build Coastguard Worker 		if (c != '%')
2876*48a54d36SAndroid Build Coastguard Worker 			{
2877*48a54d36SAndroid Build Coastguard Worker 			*sbuffer++ = (char)c;
2878*48a54d36SAndroid Build Coastguard Worker 			if (++nwritten >= buflen) goto exit;
2879*48a54d36SAndroid Build Coastguard Worker 			}
2880*48a54d36SAndroid Build Coastguard Worker 		else
2881*48a54d36SAndroid Build Coastguard Worker 			{
2882*48a54d36SAndroid Build Coastguard Worker 			unsigned int i=0, j;
2883*48a54d36SAndroid Build Coastguard Worker 			// The mDNS Vsprintf Argument Conversion Buffer is used as a temporary holding area for
2884*48a54d36SAndroid Build Coastguard Worker 			// generating decimal numbers, hexdecimal numbers, IP addresses, domain name strings, etc.
2885*48a54d36SAndroid Build Coastguard Worker 			// The size needs to be enough for a 256-byte domain name plus some error text.
2886*48a54d36SAndroid Build Coastguard Worker 			#define mDNS_VACB_Size 300
2887*48a54d36SAndroid Build Coastguard Worker 			char mDNS_VACB[mDNS_VACB_Size];
2888*48a54d36SAndroid Build Coastguard Worker 			#define mDNS_VACB_Lim (&mDNS_VACB[mDNS_VACB_Size])
2889*48a54d36SAndroid Build Coastguard Worker 			#define mDNS_VACB_Remain(s) ((mDNSu32)(mDNS_VACB_Lim - s))
2890*48a54d36SAndroid Build Coastguard Worker 			char *s = mDNS_VACB_Lim, *digits;
2891*48a54d36SAndroid Build Coastguard Worker 			struct mDNSprintf_format F = mDNSprintf_format_default;
2892*48a54d36SAndroid Build Coastguard Worker 
2893*48a54d36SAndroid Build Coastguard Worker 			while (1)	//  decode flags
2894*48a54d36SAndroid Build Coastguard Worker 				{
2895*48a54d36SAndroid Build Coastguard Worker 				c = *++fmt;
2896*48a54d36SAndroid Build Coastguard Worker 				if      (c == '-') F.leftJustify = 1;
2897*48a54d36SAndroid Build Coastguard Worker 				else if (c == '+') F.forceSign = 1;
2898*48a54d36SAndroid Build Coastguard Worker 				else if (c == ' ') F.sign = ' ';
2899*48a54d36SAndroid Build Coastguard Worker 				else if (c == '#') F.altForm++;
2900*48a54d36SAndroid Build Coastguard Worker 				else if (c == '0') F.zeroPad = 1;
2901*48a54d36SAndroid Build Coastguard Worker 				else break;
2902*48a54d36SAndroid Build Coastguard Worker 				}
2903*48a54d36SAndroid Build Coastguard Worker 
2904*48a54d36SAndroid Build Coastguard Worker 			if (c == '*')	//  decode field width
2905*48a54d36SAndroid Build Coastguard Worker 				{
2906*48a54d36SAndroid Build Coastguard Worker 				int f = va_arg(arg, int);
2907*48a54d36SAndroid Build Coastguard Worker 				if (f < 0) { f = -f; F.leftJustify = 1; }
2908*48a54d36SAndroid Build Coastguard Worker 				F.fieldWidth = (unsigned int)f;
2909*48a54d36SAndroid Build Coastguard Worker 				c = *++fmt;
2910*48a54d36SAndroid Build Coastguard Worker 				}
2911*48a54d36SAndroid Build Coastguard Worker 			else
2912*48a54d36SAndroid Build Coastguard Worker 				{
2913*48a54d36SAndroid Build Coastguard Worker 				for (; c >= '0' && c <= '9'; c = *++fmt)
2914*48a54d36SAndroid Build Coastguard Worker 					F.fieldWidth = (10 * F.fieldWidth) + (c - '0');
2915*48a54d36SAndroid Build Coastguard Worker 				}
2916*48a54d36SAndroid Build Coastguard Worker 
2917*48a54d36SAndroid Build Coastguard Worker 			if (c == '.')	//  decode precision
2918*48a54d36SAndroid Build Coastguard Worker 				{
2919*48a54d36SAndroid Build Coastguard Worker 				if ((c = *++fmt) == '*')
2920*48a54d36SAndroid Build Coastguard Worker 					{ F.precision = va_arg(arg, unsigned int); c = *++fmt; }
2921*48a54d36SAndroid Build Coastguard Worker 				else for (; c >= '0' && c <= '9'; c = *++fmt)
2922*48a54d36SAndroid Build Coastguard Worker 						F.precision = (10 * F.precision) + (c - '0');
2923*48a54d36SAndroid Build Coastguard Worker 				F.havePrecision = 1;
2924*48a54d36SAndroid Build Coastguard Worker 				}
2925*48a54d36SAndroid Build Coastguard Worker 
2926*48a54d36SAndroid Build Coastguard Worker 			if (F.leftJustify) F.zeroPad = 0;
2927*48a54d36SAndroid Build Coastguard Worker 
2928*48a54d36SAndroid Build Coastguard Worker 			conv:
2929*48a54d36SAndroid Build Coastguard Worker 			switch (c)	//  perform appropriate conversion
2930*48a54d36SAndroid Build Coastguard Worker 				{
2931*48a54d36SAndroid Build Coastguard Worker 				unsigned long n;
2932*48a54d36SAndroid Build Coastguard Worker 				case 'h' :	F.hSize = 1; c = *++fmt; goto conv;
2933*48a54d36SAndroid Build Coastguard Worker 				case 'l' :	// fall through
2934*48a54d36SAndroid Build Coastguard Worker 				case 'L' :	F.lSize = 1; c = *++fmt; goto conv;
2935*48a54d36SAndroid Build Coastguard Worker 				case 'd' :
2936*48a54d36SAndroid Build Coastguard Worker 				case 'i' :	if (F.lSize) n = (unsigned long)va_arg(arg, long);
2937*48a54d36SAndroid Build Coastguard Worker 							else n = (unsigned long)va_arg(arg, int);
2938*48a54d36SAndroid Build Coastguard Worker 							if (F.hSize) n = (short) n;
2939*48a54d36SAndroid Build Coastguard Worker 							if ((long) n < 0) { n = (unsigned long)-(long)n; F.sign = '-'; }
2940*48a54d36SAndroid Build Coastguard Worker 							else if (F.forceSign) F.sign = '+';
2941*48a54d36SAndroid Build Coastguard Worker 							goto decimal;
2942*48a54d36SAndroid Build Coastguard Worker 				case 'u' :	if (F.lSize) n = va_arg(arg, unsigned long);
2943*48a54d36SAndroid Build Coastguard Worker 							else n = va_arg(arg, unsigned int);
2944*48a54d36SAndroid Build Coastguard Worker 							if (F.hSize) n = (unsigned short) n;
2945*48a54d36SAndroid Build Coastguard Worker 							F.sign = 0;
2946*48a54d36SAndroid Build Coastguard Worker 							goto decimal;
2947*48a54d36SAndroid Build Coastguard Worker 				decimal:	if (!F.havePrecision)
2948*48a54d36SAndroid Build Coastguard Worker 								{
2949*48a54d36SAndroid Build Coastguard Worker 								if (F.zeroPad)
2950*48a54d36SAndroid Build Coastguard Worker 									{
2951*48a54d36SAndroid Build Coastguard Worker 									F.precision = F.fieldWidth;
2952*48a54d36SAndroid Build Coastguard Worker 									if (F.sign) --F.precision;
2953*48a54d36SAndroid Build Coastguard Worker 									}
2954*48a54d36SAndroid Build Coastguard Worker 								if (F.precision < 1) F.precision = 1;
2955*48a54d36SAndroid Build Coastguard Worker 								}
2956*48a54d36SAndroid Build Coastguard Worker 							if (F.precision > mDNS_VACB_Size - 1)
2957*48a54d36SAndroid Build Coastguard Worker 								F.precision = mDNS_VACB_Size - 1;
2958*48a54d36SAndroid Build Coastguard Worker 							for (i = 0; n; n /= 10, i++) *--s = (char)(n % 10 + '0');
2959*48a54d36SAndroid Build Coastguard Worker 							for (; i < F.precision; i++) *--s = '0';
2960*48a54d36SAndroid Build Coastguard Worker 							if (F.sign) { *--s = F.sign; i++; }
2961*48a54d36SAndroid Build Coastguard Worker 							break;
2962*48a54d36SAndroid Build Coastguard Worker 
2963*48a54d36SAndroid Build Coastguard Worker 				case 'o' :	if (F.lSize) n = va_arg(arg, unsigned long);
2964*48a54d36SAndroid Build Coastguard Worker 							else n = va_arg(arg, unsigned int);
2965*48a54d36SAndroid Build Coastguard Worker 							if (F.hSize) n = (unsigned short) n;
2966*48a54d36SAndroid Build Coastguard Worker 							if (!F.havePrecision)
2967*48a54d36SAndroid Build Coastguard Worker 								{
2968*48a54d36SAndroid Build Coastguard Worker 								if (F.zeroPad) F.precision = F.fieldWidth;
2969*48a54d36SAndroid Build Coastguard Worker 								if (F.precision < 1) F.precision = 1;
2970*48a54d36SAndroid Build Coastguard Worker 								}
2971*48a54d36SAndroid Build Coastguard Worker 							if (F.precision > mDNS_VACB_Size - 1)
2972*48a54d36SAndroid Build Coastguard Worker 								F.precision = mDNS_VACB_Size - 1;
2973*48a54d36SAndroid Build Coastguard Worker 							for (i = 0; n; n /= 8, i++) *--s = (char)(n % 8 + '0');
2974*48a54d36SAndroid Build Coastguard Worker 							if (F.altForm && i && *s != '0') { *--s = '0'; i++; }
2975*48a54d36SAndroid Build Coastguard Worker 							for (; i < F.precision; i++) *--s = '0';
2976*48a54d36SAndroid Build Coastguard Worker 							break;
2977*48a54d36SAndroid Build Coastguard Worker 
2978*48a54d36SAndroid Build Coastguard Worker 				case 'a' :	{
2979*48a54d36SAndroid Build Coastguard Worker 							unsigned char *a = va_arg(arg, unsigned char *);
2980*48a54d36SAndroid Build Coastguard Worker 							if (!a) { static char emsg[] = "<<NULL>>"; s = emsg; i = sizeof(emsg)-1; }
2981*48a54d36SAndroid Build Coastguard Worker 							else
2982*48a54d36SAndroid Build Coastguard Worker 								{
2983*48a54d36SAndroid Build Coastguard Worker 								s = mDNS_VACB;	// Adjust s to point to the start of the buffer, not the end
2984*48a54d36SAndroid Build Coastguard Worker 								if (F.altForm)
2985*48a54d36SAndroid Build Coastguard Worker 									{
2986*48a54d36SAndroid Build Coastguard Worker 									mDNSAddr *ip = (mDNSAddr*)a;
2987*48a54d36SAndroid Build Coastguard Worker 									switch (ip->type)
2988*48a54d36SAndroid Build Coastguard Worker 										{
2989*48a54d36SAndroid Build Coastguard Worker 										case mDNSAddrType_IPv4: F.precision =  4; a = (unsigned char *)&ip->ip.v4; break;
2990*48a54d36SAndroid Build Coastguard Worker 										case mDNSAddrType_IPv6: F.precision = 16; a = (unsigned char *)&ip->ip.v6; break;
2991*48a54d36SAndroid Build Coastguard Worker 										default:                F.precision =  0; break;
2992*48a54d36SAndroid Build Coastguard Worker 										}
2993*48a54d36SAndroid Build Coastguard Worker 									}
2994*48a54d36SAndroid Build Coastguard Worker 								if (F.altForm && !F.precision)
2995*48a54d36SAndroid Build Coastguard Worker 									i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB), "«ZERO ADDRESS»");
2996*48a54d36SAndroid Build Coastguard Worker 								else switch (F.precision)
2997*48a54d36SAndroid Build Coastguard Worker 									{
2998*48a54d36SAndroid Build Coastguard Worker 									case  4: i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB), "%d.%d.%d.%d",
2999*48a54d36SAndroid Build Coastguard Worker 														a[0], a[1], a[2], a[3]); break;
3000*48a54d36SAndroid Build Coastguard Worker 									case  6: i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB), "%02X:%02X:%02X:%02X:%02X:%02X",
3001*48a54d36SAndroid Build Coastguard Worker 														a[0], a[1], a[2], a[3], a[4], a[5]); break;
3002*48a54d36SAndroid Build Coastguard Worker 									case 16: i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB),
3003*48a54d36SAndroid Build Coastguard Worker 														"%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X",
3004*48a54d36SAndroid Build Coastguard Worker 														a[0x0], a[0x1], a[0x2], a[0x3], a[0x4], a[0x5], a[0x6], a[0x7],
3005*48a54d36SAndroid Build Coastguard Worker 														a[0x8], a[0x9], a[0xA], a[0xB], a[0xC], a[0xD], a[0xE], a[0xF]); break;
3006*48a54d36SAndroid Build Coastguard Worker 									default: i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB), "%s", "<< ERROR: Must specify"
3007*48a54d36SAndroid Build Coastguard Worker 														" address size (i.e. %.4a=IPv4, %.6a=Ethernet, %.16a=IPv6) >>"); break;
3008*48a54d36SAndroid Build Coastguard Worker 									}
3009*48a54d36SAndroid Build Coastguard Worker 								}
3010*48a54d36SAndroid Build Coastguard Worker 							}
3011*48a54d36SAndroid Build Coastguard Worker 							break;
3012*48a54d36SAndroid Build Coastguard Worker 
3013*48a54d36SAndroid Build Coastguard Worker 				case 'p' :	F.havePrecision = F.lSize = 1;
3014*48a54d36SAndroid Build Coastguard Worker 							F.precision = sizeof(void*) * 2;	// 8 characters on 32-bit; 16 characters on 64-bit
3015*48a54d36SAndroid Build Coastguard Worker 				case 'X' :	digits = "0123456789ABCDEF";
3016*48a54d36SAndroid Build Coastguard Worker 							goto hexadecimal;
3017*48a54d36SAndroid Build Coastguard Worker 				case 'x' :	digits = "0123456789abcdef";
3018*48a54d36SAndroid Build Coastguard Worker 				hexadecimal:if (F.lSize) n = va_arg(arg, unsigned long);
3019*48a54d36SAndroid Build Coastguard Worker 							else n = va_arg(arg, unsigned int);
3020*48a54d36SAndroid Build Coastguard Worker 							if (F.hSize) n = (unsigned short) n;
3021*48a54d36SAndroid Build Coastguard Worker 							if (!F.havePrecision)
3022*48a54d36SAndroid Build Coastguard Worker 								{
3023*48a54d36SAndroid Build Coastguard Worker 								if (F.zeroPad)
3024*48a54d36SAndroid Build Coastguard Worker 									{
3025*48a54d36SAndroid Build Coastguard Worker 									F.precision = F.fieldWidth;
3026*48a54d36SAndroid Build Coastguard Worker 									if (F.altForm) F.precision -= 2;
3027*48a54d36SAndroid Build Coastguard Worker 									}
3028*48a54d36SAndroid Build Coastguard Worker 								if (F.precision < 1) F.precision = 1;
3029*48a54d36SAndroid Build Coastguard Worker 								}
3030*48a54d36SAndroid Build Coastguard Worker 							if (F.precision > mDNS_VACB_Size - 1)
3031*48a54d36SAndroid Build Coastguard Worker 								F.precision = mDNS_VACB_Size - 1;
3032*48a54d36SAndroid Build Coastguard Worker 							for (i = 0; n; n /= 16, i++) *--s = digits[n % 16];
3033*48a54d36SAndroid Build Coastguard Worker 							for (; i < F.precision; i++) *--s = '0';
3034*48a54d36SAndroid Build Coastguard Worker 							if (F.altForm) { *--s = (char)c; *--s = '0'; i += 2; }
3035*48a54d36SAndroid Build Coastguard Worker 							break;
3036*48a54d36SAndroid Build Coastguard Worker 
3037*48a54d36SAndroid Build Coastguard Worker 				case 'c' :	*--s = (char)va_arg(arg, int); i = 1; break;
3038*48a54d36SAndroid Build Coastguard Worker 
3039*48a54d36SAndroid Build Coastguard Worker 				case 's' :	s = va_arg(arg, char *);
3040*48a54d36SAndroid Build Coastguard Worker 							if (!s) { static char emsg[] = "<<NULL>>"; s = emsg; i = sizeof(emsg)-1; }
3041*48a54d36SAndroid Build Coastguard Worker 							else switch (F.altForm)
3042*48a54d36SAndroid Build Coastguard Worker 								{
3043*48a54d36SAndroid Build Coastguard Worker 								case 0: i=0;
3044*48a54d36SAndroid Build Coastguard Worker 										if (!F.havePrecision)				// C string
3045*48a54d36SAndroid Build Coastguard Worker 											while (s[i]) i++;
3046*48a54d36SAndroid Build Coastguard Worker 										else
3047*48a54d36SAndroid Build Coastguard Worker 											{
3048*48a54d36SAndroid Build Coastguard Worker 											while ((i < F.precision) && s[i]) i++;
3049*48a54d36SAndroid Build Coastguard Worker 											// Make sure we don't truncate in the middle of a UTF-8 character
3050*48a54d36SAndroid Build Coastguard Worker 											// If last character we got was any kind of UTF-8 multi-byte character,
3051*48a54d36SAndroid Build Coastguard Worker 											// then see if we have to back up.
3052*48a54d36SAndroid Build Coastguard Worker 											// This is not as easy as the similar checks below, because
3053*48a54d36SAndroid Build Coastguard Worker 											// here we can't assume it's safe to examine the *next* byte, so we
3054*48a54d36SAndroid Build Coastguard Worker 											// have to confine ourselves to working only backwards in the string.
3055*48a54d36SAndroid Build Coastguard Worker 											j = i;		// Record where we got to
3056*48a54d36SAndroid Build Coastguard Worker 											// Now, back up until we find first non-continuation-char
3057*48a54d36SAndroid Build Coastguard Worker 											while (i>0 && (s[i-1] & 0xC0) == 0x80) i--;
3058*48a54d36SAndroid Build Coastguard Worker 											// Now s[i-1] is the first non-continuation-char
3059*48a54d36SAndroid Build Coastguard Worker 											// and (j-i) is the number of continuation-chars we found
3060*48a54d36SAndroid Build Coastguard Worker 											if (i>0 && (s[i-1] & 0xC0) == 0xC0)	// If we found a start-char
3061*48a54d36SAndroid Build Coastguard Worker 												{
3062*48a54d36SAndroid Build Coastguard Worker 												i--;		// Tentatively eliminate this start-char as well
3063*48a54d36SAndroid Build Coastguard Worker 												// Now (j-i) is the number of characters we're considering eliminating.
3064*48a54d36SAndroid Build Coastguard Worker 												// To be legal UTF-8, the start-char must contain (j-i) one-bits,
3065*48a54d36SAndroid Build Coastguard Worker 												// followed by a zero bit. If we shift it right by (7-(j-i)) bits
3066*48a54d36SAndroid Build Coastguard Worker 												// (with sign extension) then the result has to be 0xFE.
3067*48a54d36SAndroid Build Coastguard Worker 												// If this is right, then we reinstate the tentatively eliminated bytes.
3068*48a54d36SAndroid Build Coastguard Worker 												if (((j-i) < 7) && (((s[i] >> (7-(j-i))) & 0xFF) == 0xFE)) i = j;
3069*48a54d36SAndroid Build Coastguard Worker 												}
3070*48a54d36SAndroid Build Coastguard Worker 											}
3071*48a54d36SAndroid Build Coastguard Worker 										break;
3072*48a54d36SAndroid Build Coastguard Worker 								case 1: i = (unsigned char) *s++; break;	// Pascal string
3073*48a54d36SAndroid Build Coastguard Worker 								case 2: {									// DNS label-sequence name
3074*48a54d36SAndroid Build Coastguard Worker 										unsigned char *a = (unsigned char *)s;
3075*48a54d36SAndroid Build Coastguard Worker 										s = mDNS_VACB;	// Adjust s to point to the start of the buffer, not the end
3076*48a54d36SAndroid Build Coastguard Worker 										if (*a == 0) *s++ = '.';	// Special case for root DNS name
3077*48a54d36SAndroid Build Coastguard Worker 										while (*a)
3078*48a54d36SAndroid Build Coastguard Worker 											{
3079*48a54d36SAndroid Build Coastguard Worker 											char buf[63*4+1];
3080*48a54d36SAndroid Build Coastguard Worker 											if (*a > 63)
3081*48a54d36SAndroid Build Coastguard Worker 												{ s += mDNS_snprintf(s, mDNS_VACB_Remain(s), "<<INVALID LABEL LENGTH %u>>", *a); break; }
3082*48a54d36SAndroid Build Coastguard Worker 											if (s + *a >= &mDNS_VACB[254])
3083*48a54d36SAndroid Build Coastguard Worker 												{ s += mDNS_snprintf(s, mDNS_VACB_Remain(s), "<<NAME TOO LONG>>"); break; }
3084*48a54d36SAndroid Build Coastguard Worker 											// Need to use ConvertDomainLabelToCString to do proper escaping here,
3085*48a54d36SAndroid Build Coastguard Worker 											// so it's clear what's a literal dot and what's a label separator
3086*48a54d36SAndroid Build Coastguard Worker 											ConvertDomainLabelToCString((domainlabel*)a, buf);
3087*48a54d36SAndroid Build Coastguard Worker 											s += mDNS_snprintf(s, mDNS_VACB_Remain(s), "%s.", buf);
3088*48a54d36SAndroid Build Coastguard Worker 											a += 1 + *a;
3089*48a54d36SAndroid Build Coastguard Worker 											}
3090*48a54d36SAndroid Build Coastguard Worker 										i = (mDNSu32)(s - mDNS_VACB);
3091*48a54d36SAndroid Build Coastguard Worker 										s = mDNS_VACB;	// Reset s back to the start of the buffer
3092*48a54d36SAndroid Build Coastguard Worker 										break;
3093*48a54d36SAndroid Build Coastguard Worker 										}
3094*48a54d36SAndroid Build Coastguard Worker 								}
3095*48a54d36SAndroid Build Coastguard Worker 							// Make sure we don't truncate in the middle of a UTF-8 character (see similar comment below)
3096*48a54d36SAndroid Build Coastguard Worker 							if (F.havePrecision && i > F.precision)
3097*48a54d36SAndroid Build Coastguard Worker 								{ i = F.precision; while (i>0 && (s[i] & 0xC0) == 0x80) i--; }
3098*48a54d36SAndroid Build Coastguard Worker 							break;
3099*48a54d36SAndroid Build Coastguard Worker 
3100*48a54d36SAndroid Build Coastguard Worker 				case 'n' :	s = va_arg(arg, char *);
3101*48a54d36SAndroid Build Coastguard Worker 							if      (F.hSize) * (short *) s = (short)nwritten;
3102*48a54d36SAndroid Build Coastguard Worker 							else if (F.lSize) * (long  *) s = (long)nwritten;
3103*48a54d36SAndroid Build Coastguard Worker 							else              * (int   *) s = (int)nwritten;
3104*48a54d36SAndroid Build Coastguard Worker 							continue;
3105*48a54d36SAndroid Build Coastguard Worker 
3106*48a54d36SAndroid Build Coastguard Worker 				default:	s = mDNS_VACB;
3107*48a54d36SAndroid Build Coastguard Worker 							i = mDNS_snprintf(mDNS_VACB, sizeof(mDNS_VACB), "<<UNKNOWN FORMAT CONVERSION CODE %%%c>>", c);
3108*48a54d36SAndroid Build Coastguard Worker 
3109*48a54d36SAndroid Build Coastguard Worker 				case '%' :	*sbuffer++ = (char)c;
3110*48a54d36SAndroid Build Coastguard Worker 							if (++nwritten >= buflen) goto exit;
3111*48a54d36SAndroid Build Coastguard Worker 							break;
3112*48a54d36SAndroid Build Coastguard Worker 				}
3113*48a54d36SAndroid Build Coastguard Worker 
3114*48a54d36SAndroid Build Coastguard Worker 			if (i < F.fieldWidth && !F.leftJustify)			// Pad on the left
3115*48a54d36SAndroid Build Coastguard Worker 				do	{
3116*48a54d36SAndroid Build Coastguard Worker 					*sbuffer++ = ' ';
3117*48a54d36SAndroid Build Coastguard Worker 					if (++nwritten >= buflen) goto exit;
3118*48a54d36SAndroid Build Coastguard Worker 					} while (i < --F.fieldWidth);
3119*48a54d36SAndroid Build Coastguard Worker 
3120*48a54d36SAndroid Build Coastguard Worker 			// Make sure we don't truncate in the middle of a UTF-8 character.
3121*48a54d36SAndroid Build Coastguard Worker 			// Note: s[i] is the first eliminated character; i.e. the next character *after* the last character of the
3122*48a54d36SAndroid Build Coastguard Worker 			// allowed output. If s[i] is a UTF-8 continuation character, then we've cut a unicode character in half,
3123*48a54d36SAndroid Build Coastguard Worker 			// so back up 'i' until s[i] is no longer a UTF-8 continuation character. (if the input was proprly
3124*48a54d36SAndroid Build Coastguard Worker 			// formed, s[i] will now be the UTF-8 start character of the multi-byte character we just eliminated).
3125*48a54d36SAndroid Build Coastguard Worker 			if (i > buflen - nwritten)
3126*48a54d36SAndroid Build Coastguard Worker 				{ i = buflen - nwritten; while (i>0 && (s[i] & 0xC0) == 0x80) i--; }
3127*48a54d36SAndroid Build Coastguard Worker 			for (j=0; j<i; j++) *sbuffer++ = *s++;			// Write the converted result
3128*48a54d36SAndroid Build Coastguard Worker 			nwritten += i;
3129*48a54d36SAndroid Build Coastguard Worker 			if (nwritten >= buflen) goto exit;
3130*48a54d36SAndroid Build Coastguard Worker 
3131*48a54d36SAndroid Build Coastguard Worker 			for (; i < F.fieldWidth; i++)					// Pad on the right
3132*48a54d36SAndroid Build Coastguard Worker 				{
3133*48a54d36SAndroid Build Coastguard Worker 				*sbuffer++ = ' ';
3134*48a54d36SAndroid Build Coastguard Worker 				if (++nwritten >= buflen) goto exit;
3135*48a54d36SAndroid Build Coastguard Worker 				}
3136*48a54d36SAndroid Build Coastguard Worker 			}
3137*48a54d36SAndroid Build Coastguard Worker 		}
3138*48a54d36SAndroid Build Coastguard Worker 	exit:
3139*48a54d36SAndroid Build Coastguard Worker 	*sbuffer++ = 0;
3140*48a54d36SAndroid Build Coastguard Worker 	return(nwritten);
3141*48a54d36SAndroid Build Coastguard Worker 	}
3142*48a54d36SAndroid Build Coastguard Worker 
mDNS_snprintf(char * sbuffer,mDNSu32 buflen,const char * fmt,...)3143*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSu32 mDNS_snprintf(char *sbuffer, mDNSu32 buflen, const char *fmt, ...)
3144*48a54d36SAndroid Build Coastguard Worker 	{
3145*48a54d36SAndroid Build Coastguard Worker 	mDNSu32 length;
3146*48a54d36SAndroid Build Coastguard Worker 
3147*48a54d36SAndroid Build Coastguard Worker 	va_list ptr;
3148*48a54d36SAndroid Build Coastguard Worker 	va_start(ptr,fmt);
3149*48a54d36SAndroid Build Coastguard Worker 	length = mDNS_vsnprintf(sbuffer, buflen, fmt, ptr);
3150*48a54d36SAndroid Build Coastguard Worker 	va_end(ptr);
3151*48a54d36SAndroid Build Coastguard Worker 
3152*48a54d36SAndroid Build Coastguard Worker 	return(length);
3153*48a54d36SAndroid Build Coastguard Worker 	}
3154