1*48a54d36SAndroid Build Coastguard Worker /* -*- Mode: C; tab-width: 4 -*-
2*48a54d36SAndroid Build Coastguard Worker *
3*48a54d36SAndroid Build Coastguard Worker * Copyright (c) 2003-2006 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 #if defined(_WIN32)
19*48a54d36SAndroid Build Coastguard Worker #include <process.h>
20*48a54d36SAndroid Build Coastguard Worker #define usleep(X) Sleep(((X)+999)/1000)
21*48a54d36SAndroid Build Coastguard Worker #else
22*48a54d36SAndroid Build Coastguard Worker #include <fcntl.h>
23*48a54d36SAndroid Build Coastguard Worker #include <errno.h>
24*48a54d36SAndroid Build Coastguard Worker #include <sys/ioctl.h>
25*48a54d36SAndroid Build Coastguard Worker #include <sys/types.h>
26*48a54d36SAndroid Build Coastguard Worker #include <sys/time.h>
27*48a54d36SAndroid Build Coastguard Worker #include <sys/resource.h>
28*48a54d36SAndroid Build Coastguard Worker #endif
29*48a54d36SAndroid Build Coastguard Worker
30*48a54d36SAndroid Build Coastguard Worker #include <stdlib.h>
31*48a54d36SAndroid Build Coastguard Worker #include <stdio.h>
32*48a54d36SAndroid Build Coastguard Worker
33*48a54d36SAndroid Build Coastguard Worker #include "mDNSEmbeddedAPI.h"
34*48a54d36SAndroid Build Coastguard Worker #include "DNSCommon.h"
35*48a54d36SAndroid Build Coastguard Worker #include "uDNS.h"
36*48a54d36SAndroid Build Coastguard Worker #include "uds_daemon.h"
37*48a54d36SAndroid Build Coastguard Worker
38*48a54d36SAndroid Build Coastguard Worker #ifdef __ANDROID__
39*48a54d36SAndroid Build Coastguard Worker #include "cutils/sockets.h"
40*48a54d36SAndroid Build Coastguard Worker #endif
41*48a54d36SAndroid Build Coastguard Worker
42*48a54d36SAndroid Build Coastguard Worker // Normally we append search domains only for queries with a single label that are not
43*48a54d36SAndroid Build Coastguard Worker // fully qualified. This can be overridden to apply search domains for queries (that are
44*48a54d36SAndroid Build Coastguard Worker // not fully qualified) with any number of labels e.g., moon, moon.cs, moon.cs.be, etc.
45*48a54d36SAndroid Build Coastguard Worker mDNSBool AlwaysAppendSearchDomains = mDNSfalse;
46*48a54d36SAndroid Build Coastguard Worker
47*48a54d36SAndroid Build Coastguard Worker // Apple-specific functionality, not required for other platforms
48*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
49*48a54d36SAndroid Build Coastguard Worker #include <sys/ucred.h>
50*48a54d36SAndroid Build Coastguard Worker #ifndef PID_FILE
51*48a54d36SAndroid Build Coastguard Worker #define PID_FILE ""
52*48a54d36SAndroid Build Coastguard Worker #endif
53*48a54d36SAndroid Build Coastguard Worker #endif
54*48a54d36SAndroid Build Coastguard Worker
55*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
56*48a54d36SAndroid Build Coastguard Worker #include <WebFilterDNS/WebFilterDNS.h>
57*48a54d36SAndroid Build Coastguard Worker
58*48a54d36SAndroid Build Coastguard Worker #if ! NO_WCF
59*48a54d36SAndroid Build Coastguard Worker
60*48a54d36SAndroid Build Coastguard Worker int WCFIsServerRunning(WCFConnection *conn) __attribute__((weak_import));
61*48a54d36SAndroid Build Coastguard Worker int WCFNameResolvesToAddr(WCFConnection *conn, char* domainName, struct sockaddr* address, uid_t userid) __attribute__((weak_import));
62*48a54d36SAndroid Build Coastguard Worker int WCFNameResolvesToName(WCFConnection *conn, char* fromName, char* toName, uid_t userid) __attribute__((weak_import));
63*48a54d36SAndroid Build Coastguard Worker
64*48a54d36SAndroid Build Coastguard Worker // Do we really need to define a macro for "if"?
65*48a54d36SAndroid Build Coastguard Worker #define CHECK_WCF_FUNCTION(X) if (X)
66*48a54d36SAndroid Build Coastguard Worker #endif // ! NO_WCF
67*48a54d36SAndroid Build Coastguard Worker
68*48a54d36SAndroid Build Coastguard Worker #else
69*48a54d36SAndroid Build Coastguard Worker #define NO_WCF 1
70*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
71*48a54d36SAndroid Build Coastguard Worker
72*48a54d36SAndroid Build Coastguard Worker // User IDs 0-500 are system-wide processes, not actual users in the usual sense
73*48a54d36SAndroid Build Coastguard Worker // User IDs for real user accounts start at 501 and count up from there
74*48a54d36SAndroid Build Coastguard Worker #define SystemUID(X) ((X) <= 500)
75*48a54d36SAndroid Build Coastguard Worker
76*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
77*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
78*48a54d36SAndroid Build Coastguard Worker #pragma mark -
79*48a54d36SAndroid Build Coastguard Worker #pragma mark - Types and Data Structures
80*48a54d36SAndroid Build Coastguard Worker #endif
81*48a54d36SAndroid Build Coastguard Worker
82*48a54d36SAndroid Build Coastguard Worker typedef enum
83*48a54d36SAndroid Build Coastguard Worker {
84*48a54d36SAndroid Build Coastguard Worker t_uninitialized,
85*48a54d36SAndroid Build Coastguard Worker t_morecoming,
86*48a54d36SAndroid Build Coastguard Worker t_complete,
87*48a54d36SAndroid Build Coastguard Worker t_error,
88*48a54d36SAndroid Build Coastguard Worker t_terminated
89*48a54d36SAndroid Build Coastguard Worker } transfer_state;
90*48a54d36SAndroid Build Coastguard Worker
91*48a54d36SAndroid Build Coastguard Worker typedef struct request_state request_state;
92*48a54d36SAndroid Build Coastguard Worker
93*48a54d36SAndroid Build Coastguard Worker typedef void (*req_termination_fn)(request_state *request);
94*48a54d36SAndroid Build Coastguard Worker
95*48a54d36SAndroid Build Coastguard Worker typedef struct registered_record_entry
96*48a54d36SAndroid Build Coastguard Worker {
97*48a54d36SAndroid Build Coastguard Worker struct registered_record_entry *next;
98*48a54d36SAndroid Build Coastguard Worker mDNSu32 key;
99*48a54d36SAndroid Build Coastguard Worker client_context_t regrec_client_context;
100*48a54d36SAndroid Build Coastguard Worker request_state *request;
101*48a54d36SAndroid Build Coastguard Worker mDNSBool external_advertise;
102*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID origInterfaceID;
103*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr; // Pointer to variable-sized AuthRecord (Why a pointer? Why not just embed it here?)
104*48a54d36SAndroid Build Coastguard Worker } registered_record_entry;
105*48a54d36SAndroid Build Coastguard Worker
106*48a54d36SAndroid Build Coastguard Worker // A single registered service: ServiceRecordSet + bookkeeping
107*48a54d36SAndroid Build Coastguard Worker // Note that we duplicate some fields from parent service_info object
108*48a54d36SAndroid Build Coastguard Worker // to facilitate cleanup, when instances and parent may be deallocated at different times.
109*48a54d36SAndroid Build Coastguard Worker typedef struct service_instance
110*48a54d36SAndroid Build Coastguard Worker {
111*48a54d36SAndroid Build Coastguard Worker struct service_instance *next;
112*48a54d36SAndroid Build Coastguard Worker request_state *request;
113*48a54d36SAndroid Build Coastguard Worker AuthRecord *subtypes;
114*48a54d36SAndroid Build Coastguard Worker mDNSBool renameonmemfree; // Set on config change when we deregister original name
115*48a54d36SAndroid Build Coastguard Worker mDNSBool clientnotified; // Has client been notified of successful registration yet?
116*48a54d36SAndroid Build Coastguard Worker mDNSBool default_local; // is this the "local." from an empty-string registration?
117*48a54d36SAndroid Build Coastguard Worker mDNSBool external_advertise; // is this is being advertised externally?
118*48a54d36SAndroid Build Coastguard Worker domainname domain;
119*48a54d36SAndroid Build Coastguard Worker ServiceRecordSet srs; // note -- variable-sized object -- must be last field in struct
120*48a54d36SAndroid Build Coastguard Worker } service_instance;
121*48a54d36SAndroid Build Coastguard Worker
122*48a54d36SAndroid Build Coastguard Worker // for multi-domain default browsing
123*48a54d36SAndroid Build Coastguard Worker typedef struct browser_t
124*48a54d36SAndroid Build Coastguard Worker {
125*48a54d36SAndroid Build Coastguard Worker struct browser_t *next;
126*48a54d36SAndroid Build Coastguard Worker domainname domain;
127*48a54d36SAndroid Build Coastguard Worker DNSQuestion q;
128*48a54d36SAndroid Build Coastguard Worker } browser_t;
129*48a54d36SAndroid Build Coastguard Worker
130*48a54d36SAndroid Build Coastguard Worker struct request_state
131*48a54d36SAndroid Build Coastguard Worker {
132*48a54d36SAndroid Build Coastguard Worker request_state *next;
133*48a54d36SAndroid Build Coastguard Worker request_state *primary; // If this operation is on a shared socket, pointer to primary
134*48a54d36SAndroid Build Coastguard Worker // request_state for the original DNSServiceCreateConnection() operation
135*48a54d36SAndroid Build Coastguard Worker dnssd_sock_t sd;
136*48a54d36SAndroid Build Coastguard Worker dnssd_sock_t errsd;
137*48a54d36SAndroid Build Coastguard Worker mDNSu32 uid;
138*48a54d36SAndroid Build Coastguard Worker void * platform_data;
139*48a54d36SAndroid Build Coastguard Worker
140*48a54d36SAndroid Build Coastguard Worker // Note: On a shared connection these fields in the primary structure, including hdr, are re-used
141*48a54d36SAndroid Build Coastguard Worker // for each new request. This is because, until we've read the ipc_msg_hdr to find out what the
142*48a54d36SAndroid Build Coastguard Worker // operation is, we don't know if we're going to need to allocate a new request_state or not.
143*48a54d36SAndroid Build Coastguard Worker transfer_state ts;
144*48a54d36SAndroid Build Coastguard Worker mDNSu32 hdr_bytes; // bytes of header already read
145*48a54d36SAndroid Build Coastguard Worker ipc_msg_hdr hdr;
146*48a54d36SAndroid Build Coastguard Worker mDNSu32 data_bytes; // bytes of message data already read
147*48a54d36SAndroid Build Coastguard Worker char *msgbuf; // pointer to data storage to pass to free()
148*48a54d36SAndroid Build Coastguard Worker const char *msgptr; // pointer to data to be read from (may be modified)
149*48a54d36SAndroid Build Coastguard Worker char *msgend; // pointer to byte after last byte of message
150*48a54d36SAndroid Build Coastguard Worker
151*48a54d36SAndroid Build Coastguard Worker // reply, termination, error, and client context info
152*48a54d36SAndroid Build Coastguard Worker int no_reply; // don't send asynchronous replies to client
153*48a54d36SAndroid Build Coastguard Worker mDNSs32 time_blocked; // record time of a blocked client
154*48a54d36SAndroid Build Coastguard Worker int unresponsiveness_reports;
155*48a54d36SAndroid Build Coastguard Worker struct reply_state *replies; // corresponding (active) reply list
156*48a54d36SAndroid Build Coastguard Worker req_termination_fn terminate;
157*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags;
158*48a54d36SAndroid Build Coastguard Worker
159*48a54d36SAndroid Build Coastguard Worker union
160*48a54d36SAndroid Build Coastguard Worker {
161*48a54d36SAndroid Build Coastguard Worker registered_record_entry *reg_recs; // list of registrations for a connection-oriented request
162*48a54d36SAndroid Build Coastguard Worker struct
163*48a54d36SAndroid Build Coastguard Worker {
164*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID interface_id;
165*48a54d36SAndroid Build Coastguard Worker mDNSBool default_domain;
166*48a54d36SAndroid Build Coastguard Worker mDNSBool ForceMCast;
167*48a54d36SAndroid Build Coastguard Worker domainname regtype;
168*48a54d36SAndroid Build Coastguard Worker browser_t *browsers;
169*48a54d36SAndroid Build Coastguard Worker } browser;
170*48a54d36SAndroid Build Coastguard Worker struct
171*48a54d36SAndroid Build Coastguard Worker {
172*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID;
173*48a54d36SAndroid Build Coastguard Worker mDNSu16 txtlen;
174*48a54d36SAndroid Build Coastguard Worker void *txtdata;
175*48a54d36SAndroid Build Coastguard Worker mDNSIPPort port;
176*48a54d36SAndroid Build Coastguard Worker domainlabel name;
177*48a54d36SAndroid Build Coastguard Worker char type_as_string[MAX_ESCAPED_DOMAIN_NAME];
178*48a54d36SAndroid Build Coastguard Worker domainname type;
179*48a54d36SAndroid Build Coastguard Worker mDNSBool default_domain;
180*48a54d36SAndroid Build Coastguard Worker domainname host;
181*48a54d36SAndroid Build Coastguard Worker mDNSBool autoname; // Set if this name is tied to the Computer Name
182*48a54d36SAndroid Build Coastguard Worker mDNSBool autorename; // Set if this client wants us to automatically rename on conflict
183*48a54d36SAndroid Build Coastguard Worker mDNSBool allowremotequery; // Respond to unicast queries from outside the local link?
184*48a54d36SAndroid Build Coastguard Worker int num_subtypes;
185*48a54d36SAndroid Build Coastguard Worker service_instance *instances;
186*48a54d36SAndroid Build Coastguard Worker } servicereg;
187*48a54d36SAndroid Build Coastguard Worker struct
188*48a54d36SAndroid Build Coastguard Worker {
189*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID interface_id;
190*48a54d36SAndroid Build Coastguard Worker mDNSu32 flags;
191*48a54d36SAndroid Build Coastguard Worker mDNSu32 protocol;
192*48a54d36SAndroid Build Coastguard Worker DNSQuestion q4;
193*48a54d36SAndroid Build Coastguard Worker DNSQuestion *q42;
194*48a54d36SAndroid Build Coastguard Worker DNSQuestion q6;
195*48a54d36SAndroid Build Coastguard Worker DNSQuestion *q62;
196*48a54d36SAndroid Build Coastguard Worker } addrinfo;
197*48a54d36SAndroid Build Coastguard Worker struct
198*48a54d36SAndroid Build Coastguard Worker {
199*48a54d36SAndroid Build Coastguard Worker mDNSIPPort ReqExt; // External port we originally requested, for logging purposes
200*48a54d36SAndroid Build Coastguard Worker NATTraversalInfo NATinfo;
201*48a54d36SAndroid Build Coastguard Worker } pm;
202*48a54d36SAndroid Build Coastguard Worker struct
203*48a54d36SAndroid Build Coastguard Worker {
204*48a54d36SAndroid Build Coastguard Worker #if 0
205*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags;
206*48a54d36SAndroid Build Coastguard Worker #endif
207*48a54d36SAndroid Build Coastguard Worker DNSQuestion q_all;
208*48a54d36SAndroid Build Coastguard Worker DNSQuestion q_default;
209*48a54d36SAndroid Build Coastguard Worker } enumeration;
210*48a54d36SAndroid Build Coastguard Worker struct
211*48a54d36SAndroid Build Coastguard Worker {
212*48a54d36SAndroid Build Coastguard Worker DNSQuestion q;
213*48a54d36SAndroid Build Coastguard Worker DNSQuestion *q2;
214*48a54d36SAndroid Build Coastguard Worker } queryrecord;
215*48a54d36SAndroid Build Coastguard Worker struct
216*48a54d36SAndroid Build Coastguard Worker {
217*48a54d36SAndroid Build Coastguard Worker DNSQuestion qtxt;
218*48a54d36SAndroid Build Coastguard Worker DNSQuestion qsrv;
219*48a54d36SAndroid Build Coastguard Worker const ResourceRecord *txt;
220*48a54d36SAndroid Build Coastguard Worker const ResourceRecord *srv;
221*48a54d36SAndroid Build Coastguard Worker mDNSs32 ReportTime;
222*48a54d36SAndroid Build Coastguard Worker mDNSBool external_advertise;
223*48a54d36SAndroid Build Coastguard Worker } resolve;
224*48a54d36SAndroid Build Coastguard Worker } u;
225*48a54d36SAndroid Build Coastguard Worker };
226*48a54d36SAndroid Build Coastguard Worker
227*48a54d36SAndroid Build Coastguard Worker // struct physically sits between ipc message header and call-specific fields in the message buffer
228*48a54d36SAndroid Build Coastguard Worker typedef struct
229*48a54d36SAndroid Build Coastguard Worker {
230*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags; // Note: This field is in NETWORK byte order
231*48a54d36SAndroid Build Coastguard Worker mDNSu32 ifi; // Note: This field is in NETWORK byte order
232*48a54d36SAndroid Build Coastguard Worker DNSServiceErrorType error; // Note: This field is in NETWORK byte order
233*48a54d36SAndroid Build Coastguard Worker } reply_hdr;
234*48a54d36SAndroid Build Coastguard Worker
235*48a54d36SAndroid Build Coastguard Worker typedef struct reply_state
236*48a54d36SAndroid Build Coastguard Worker {
237*48a54d36SAndroid Build Coastguard Worker struct reply_state *next; // If there are multiple unsent replies
238*48a54d36SAndroid Build Coastguard Worker mDNSu32 totallen;
239*48a54d36SAndroid Build Coastguard Worker mDNSu32 nwriten;
240*48a54d36SAndroid Build Coastguard Worker ipc_msg_hdr mhdr[1];
241*48a54d36SAndroid Build Coastguard Worker reply_hdr rhdr[1];
242*48a54d36SAndroid Build Coastguard Worker } reply_state;
243*48a54d36SAndroid Build Coastguard Worker
244*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
245*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
246*48a54d36SAndroid Build Coastguard Worker #pragma mark -
247*48a54d36SAndroid Build Coastguard Worker #pragma mark - Globals
248*48a54d36SAndroid Build Coastguard Worker #endif
249*48a54d36SAndroid Build Coastguard Worker
250*48a54d36SAndroid Build Coastguard Worker // globals
251*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNS mDNSStorage;
252*48a54d36SAndroid Build Coastguard Worker mDNSexport const char ProgramName[] = "mDNSResponder";
253*48a54d36SAndroid Build Coastguard Worker
254*48a54d36SAndroid Build Coastguard Worker static dnssd_sock_t listenfd = dnssd_InvalidSocket;
255*48a54d36SAndroid Build Coastguard Worker static request_state *all_requests = NULL;
256*48a54d36SAndroid Build Coastguard Worker
257*48a54d36SAndroid Build Coastguard Worker // Note asymmetry here between registration and browsing.
258*48a54d36SAndroid Build Coastguard Worker // For service registrations we only automatically register in domains that explicitly appear in local configuration data
259*48a54d36SAndroid Build Coastguard Worker // (so AutoRegistrationDomains could equally well be called SCPrefRegDomains)
260*48a54d36SAndroid Build Coastguard Worker // For service browsing we also learn automatic browsing domains from the network, so for that case we have:
261*48a54d36SAndroid Build Coastguard Worker // 1. SCPrefBrowseDomains (local configuration data)
262*48a54d36SAndroid Build Coastguard Worker // 2. LocalDomainEnumRecords (locally-generated local-only PTR records -- equivalent to slElem->AuthRecs in uDNS.c)
263*48a54d36SAndroid Build Coastguard Worker // 3. AutoBrowseDomains, which is populated by tracking add/rmv events in AutomaticBrowseDomainChange, the callback function for our mDNS_GetDomains call.
264*48a54d36SAndroid Build Coastguard Worker // By creating and removing our own LocalDomainEnumRecords, we trigger AutomaticBrowseDomainChange callbacks just like domains learned from the network would.
265*48a54d36SAndroid Build Coastguard Worker
266*48a54d36SAndroid Build Coastguard Worker mDNSexport DNameListElem *AutoRegistrationDomains; // Domains where we automatically register for empty-string registrations
267*48a54d36SAndroid Build Coastguard Worker
268*48a54d36SAndroid Build Coastguard Worker static DNameListElem *SCPrefBrowseDomains; // List of automatic browsing domains read from SCPreferences for "empty string" browsing
269*48a54d36SAndroid Build Coastguard Worker static ARListElem *LocalDomainEnumRecords; // List of locally-generated PTR records to augment those we learn from the network
270*48a54d36SAndroid Build Coastguard Worker mDNSexport DNameListElem *AutoBrowseDomains; // List created from those local-only PTR records plus records we get from the network
271*48a54d36SAndroid Build Coastguard Worker
272*48a54d36SAndroid Build Coastguard Worker #define MSG_PAD_BYTES 5 // pad message buffer (read from client) with n zero'd bytes to guarantee
273*48a54d36SAndroid Build Coastguard Worker // n get_string() calls w/o buffer overrun
274*48a54d36SAndroid Build Coastguard Worker // initialization, setup/teardown functions
275*48a54d36SAndroid Build Coastguard Worker
276*48a54d36SAndroid Build Coastguard Worker // If a platform specifies its own PID file name, we use that
277*48a54d36SAndroid Build Coastguard Worker #ifndef PID_FILE
278*48a54d36SAndroid Build Coastguard Worker #define PID_FILE "/var/run/mDNSResponder.pid"
279*48a54d36SAndroid Build Coastguard Worker #endif
280*48a54d36SAndroid Build Coastguard Worker
281*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
282*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
283*48a54d36SAndroid Build Coastguard Worker #pragma mark -
284*48a54d36SAndroid Build Coastguard Worker #pragma mark - General Utility Functions
285*48a54d36SAndroid Build Coastguard Worker #endif
286*48a54d36SAndroid Build Coastguard Worker
FatalError(char * errmsg)287*48a54d36SAndroid Build Coastguard Worker mDNSlocal void FatalError(char *errmsg)
288*48a54d36SAndroid Build Coastguard Worker {
289*48a54d36SAndroid Build Coastguard Worker LogMsg("%s: %s", errmsg, dnssd_strerror(dnssd_errno));
290*48a54d36SAndroid Build Coastguard Worker *(volatile long*)0 = 0; // On OS X abort() doesn't generate a crash log, but writing to zero does
291*48a54d36SAndroid Build Coastguard Worker abort(); // On platforms where writing to zero doesn't generate an exception, abort instead
292*48a54d36SAndroid Build Coastguard Worker }
293*48a54d36SAndroid Build Coastguard Worker
dnssd_htonl(mDNSu32 l)294*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSu32 dnssd_htonl(mDNSu32 l)
295*48a54d36SAndroid Build Coastguard Worker {
296*48a54d36SAndroid Build Coastguard Worker mDNSu32 ret;
297*48a54d36SAndroid Build Coastguard Worker char *data = (char*) &ret;
298*48a54d36SAndroid Build Coastguard Worker put_uint32(l, &data);
299*48a54d36SAndroid Build Coastguard Worker return ret;
300*48a54d36SAndroid Build Coastguard Worker }
301*48a54d36SAndroid Build Coastguard Worker
302*48a54d36SAndroid Build Coastguard Worker // hack to search-replace perror's to LogMsg's
my_perror(char * errmsg)303*48a54d36SAndroid Build Coastguard Worker mDNSlocal void my_perror(char *errmsg)
304*48a54d36SAndroid Build Coastguard Worker {
305*48a54d36SAndroid Build Coastguard Worker LogMsg("%s: %d (%s)", errmsg, dnssd_errno, dnssd_strerror(dnssd_errno));
306*48a54d36SAndroid Build Coastguard Worker }
307*48a54d36SAndroid Build Coastguard Worker
abort_request(request_state * req)308*48a54d36SAndroid Build Coastguard Worker mDNSlocal void abort_request(request_state *req)
309*48a54d36SAndroid Build Coastguard Worker {
310*48a54d36SAndroid Build Coastguard Worker if (req->terminate == (req_termination_fn)~0)
311*48a54d36SAndroid Build Coastguard Worker { LogMsg("abort_request: ERROR: Attempt to abort operation %p with req->terminate %p", req, req->terminate); return; }
312*48a54d36SAndroid Build Coastguard Worker
313*48a54d36SAndroid Build Coastguard Worker // First stop whatever mDNSCore operation we were doing
314*48a54d36SAndroid Build Coastguard Worker // If this is actually a shared connection operation, then its req->terminate function will scan
315*48a54d36SAndroid Build Coastguard Worker // the all_requests list and terminate any subbordinate operations sharing this file descriptor
316*48a54d36SAndroid Build Coastguard Worker if (req->terminate) req->terminate(req);
317*48a54d36SAndroid Build Coastguard Worker
318*48a54d36SAndroid Build Coastguard Worker if (!dnssd_SocketValid(req->sd))
319*48a54d36SAndroid Build Coastguard Worker { LogMsg("abort_request: ERROR: Attempt to abort operation %p with invalid fd %d", req, req->sd); return; }
320*48a54d36SAndroid Build Coastguard Worker
321*48a54d36SAndroid Build Coastguard Worker // Now, if this request_state is not subordinate to some other primary, close file descriptor and discard replies
322*48a54d36SAndroid Build Coastguard Worker if (!req->primary)
323*48a54d36SAndroid Build Coastguard Worker {
324*48a54d36SAndroid Build Coastguard Worker if (req->errsd != req->sd) LogOperation("%3d: Removing FD and closing errsd %d", req->sd, req->errsd);
325*48a54d36SAndroid Build Coastguard Worker else LogOperation("%3d: Removing FD", req->sd);
326*48a54d36SAndroid Build Coastguard Worker udsSupportRemoveFDFromEventLoop(req->sd, req->platform_data); // Note: This also closes file descriptor req->sd for us
327*48a54d36SAndroid Build Coastguard Worker if (req->errsd != req->sd) { dnssd_close(req->errsd); req->errsd = req->sd; }
328*48a54d36SAndroid Build Coastguard Worker
329*48a54d36SAndroid Build Coastguard Worker while (req->replies) // free pending replies
330*48a54d36SAndroid Build Coastguard Worker {
331*48a54d36SAndroid Build Coastguard Worker reply_state *ptr = req->replies;
332*48a54d36SAndroid Build Coastguard Worker req->replies = req->replies->next;
333*48a54d36SAndroid Build Coastguard Worker freeL("reply_state (abort)", ptr);
334*48a54d36SAndroid Build Coastguard Worker }
335*48a54d36SAndroid Build Coastguard Worker }
336*48a54d36SAndroid Build Coastguard Worker
337*48a54d36SAndroid Build Coastguard Worker // Set req->sd to something invalid, so that udsserver_idle knows to unlink and free this structure
338*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING
339*48a54d36SAndroid Build Coastguard Worker // Don't use dnssd_InvalidSocket (-1) because that's the sentinel value MACOSX_MDNS_MALLOC_DEBUGGING uses
340*48a54d36SAndroid Build Coastguard Worker // for detecting when the memory for an object is inadvertently freed while the object is still on some list
341*48a54d36SAndroid Build Coastguard Worker req->sd = req->errsd = -2;
342*48a54d36SAndroid Build Coastguard Worker #else
343*48a54d36SAndroid Build Coastguard Worker req->sd = req->errsd = dnssd_InvalidSocket;
344*48a54d36SAndroid Build Coastguard Worker #endif
345*48a54d36SAndroid Build Coastguard Worker // We also set req->terminate to a bogus value so we know if abort_request() gets called again for this request
346*48a54d36SAndroid Build Coastguard Worker req->terminate = (req_termination_fn)~0;
347*48a54d36SAndroid Build Coastguard Worker }
348*48a54d36SAndroid Build Coastguard Worker
AbortUnlinkAndFree(request_state * req)349*48a54d36SAndroid Build Coastguard Worker mDNSlocal void AbortUnlinkAndFree(request_state *req)
350*48a54d36SAndroid Build Coastguard Worker {
351*48a54d36SAndroid Build Coastguard Worker request_state **p = &all_requests;
352*48a54d36SAndroid Build Coastguard Worker abort_request(req);
353*48a54d36SAndroid Build Coastguard Worker while (*p && *p != req) p=&(*p)->next;
354*48a54d36SAndroid Build Coastguard Worker if (*p) { *p = req->next; freeL("request_state/AbortUnlinkAndFree", req); }
355*48a54d36SAndroid Build Coastguard Worker else LogMsg("AbortUnlinkAndFree: ERROR: Attempt to abort operation %p not in list", req);
356*48a54d36SAndroid Build Coastguard Worker }
357*48a54d36SAndroid Build Coastguard Worker
create_reply(const reply_op_t op,const size_t datalen,request_state * const request)358*48a54d36SAndroid Build Coastguard Worker mDNSlocal reply_state *create_reply(const reply_op_t op, const size_t datalen, request_state *const request)
359*48a54d36SAndroid Build Coastguard Worker {
360*48a54d36SAndroid Build Coastguard Worker reply_state *reply;
361*48a54d36SAndroid Build Coastguard Worker
362*48a54d36SAndroid Build Coastguard Worker if ((unsigned)datalen < sizeof(reply_hdr))
363*48a54d36SAndroid Build Coastguard Worker {
364*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: create_reply - data length less than length of required fields");
365*48a54d36SAndroid Build Coastguard Worker return NULL;
366*48a54d36SAndroid Build Coastguard Worker }
367*48a54d36SAndroid Build Coastguard Worker
368*48a54d36SAndroid Build Coastguard Worker reply = mallocL("reply_state", sizeof(reply_state) + datalen - sizeof(reply_hdr));
369*48a54d36SAndroid Build Coastguard Worker if (!reply) FatalError("ERROR: malloc");
370*48a54d36SAndroid Build Coastguard Worker
371*48a54d36SAndroid Build Coastguard Worker reply->next = mDNSNULL;
372*48a54d36SAndroid Build Coastguard Worker reply->totallen = (mDNSu32)datalen + sizeof(ipc_msg_hdr);
373*48a54d36SAndroid Build Coastguard Worker reply->nwriten = 0;
374*48a54d36SAndroid Build Coastguard Worker
375*48a54d36SAndroid Build Coastguard Worker reply->mhdr->version = VERSION;
376*48a54d36SAndroid Build Coastguard Worker reply->mhdr->datalen = (mDNSu32)datalen;
377*48a54d36SAndroid Build Coastguard Worker reply->mhdr->ipc_flags = 0;
378*48a54d36SAndroid Build Coastguard Worker reply->mhdr->op = op;
379*48a54d36SAndroid Build Coastguard Worker reply->mhdr->client_context = request->hdr.client_context;
380*48a54d36SAndroid Build Coastguard Worker reply->mhdr->reg_index = 0;
381*48a54d36SAndroid Build Coastguard Worker
382*48a54d36SAndroid Build Coastguard Worker return reply;
383*48a54d36SAndroid Build Coastguard Worker }
384*48a54d36SAndroid Build Coastguard Worker
385*48a54d36SAndroid Build Coastguard Worker // Append a reply to the list in a request object
386*48a54d36SAndroid Build Coastguard Worker // If our request is sharing a connection, then we append our reply_state onto the primary's list
append_reply(request_state * req,reply_state * rep)387*48a54d36SAndroid Build Coastguard Worker mDNSlocal void append_reply(request_state *req, reply_state *rep)
388*48a54d36SAndroid Build Coastguard Worker {
389*48a54d36SAndroid Build Coastguard Worker request_state *r = req->primary ? req->primary : req;
390*48a54d36SAndroid Build Coastguard Worker reply_state **ptr = &r->replies;
391*48a54d36SAndroid Build Coastguard Worker while (*ptr) ptr = &(*ptr)->next;
392*48a54d36SAndroid Build Coastguard Worker *ptr = rep;
393*48a54d36SAndroid Build Coastguard Worker rep->next = NULL;
394*48a54d36SAndroid Build Coastguard Worker }
395*48a54d36SAndroid Build Coastguard Worker
396*48a54d36SAndroid Build Coastguard Worker // Generates a response message giving name, type, domain, plus interface index,
397*48a54d36SAndroid Build Coastguard Worker // suitable for a browse result or service registration result.
398*48a54d36SAndroid Build Coastguard Worker // On successful completion rep is set to point to a malloc'd reply_state struct
GenerateNTDResponse(const domainname * const servicename,const mDNSInterfaceID id,request_state * const request,reply_state ** const rep,reply_op_t op,DNSServiceFlags flags,mStatus err)399*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus GenerateNTDResponse(const domainname *const servicename, const mDNSInterfaceID id,
400*48a54d36SAndroid Build Coastguard Worker request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err)
401*48a54d36SAndroid Build Coastguard Worker {
402*48a54d36SAndroid Build Coastguard Worker domainlabel name;
403*48a54d36SAndroid Build Coastguard Worker domainname type, dom;
404*48a54d36SAndroid Build Coastguard Worker *rep = NULL;
405*48a54d36SAndroid Build Coastguard Worker if (!DeconstructServiceName(servicename, &name, &type, &dom))
406*48a54d36SAndroid Build Coastguard Worker return kDNSServiceErr_Invalid;
407*48a54d36SAndroid Build Coastguard Worker else
408*48a54d36SAndroid Build Coastguard Worker {
409*48a54d36SAndroid Build Coastguard Worker char namestr[MAX_DOMAIN_LABEL+1];
410*48a54d36SAndroid Build Coastguard Worker char typestr[MAX_ESCAPED_DOMAIN_NAME];
411*48a54d36SAndroid Build Coastguard Worker char domstr [MAX_ESCAPED_DOMAIN_NAME];
412*48a54d36SAndroid Build Coastguard Worker int len;
413*48a54d36SAndroid Build Coastguard Worker char *data;
414*48a54d36SAndroid Build Coastguard Worker
415*48a54d36SAndroid Build Coastguard Worker ConvertDomainLabelToCString_unescaped(&name, namestr);
416*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(&type, typestr);
417*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(&dom, domstr);
418*48a54d36SAndroid Build Coastguard Worker
419*48a54d36SAndroid Build Coastguard Worker // Calculate reply data length
420*48a54d36SAndroid Build Coastguard Worker len = sizeof(DNSServiceFlags);
421*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32); // if index
422*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceErrorType);
423*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(namestr) + 1);
424*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(typestr) + 1);
425*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(domstr) + 1);
426*48a54d36SAndroid Build Coastguard Worker
427*48a54d36SAndroid Build Coastguard Worker // Build reply header
428*48a54d36SAndroid Build Coastguard Worker *rep = create_reply(op, len, request);
429*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->flags = dnssd_htonl(flags);
430*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse));
431*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->error = dnssd_htonl(err);
432*48a54d36SAndroid Build Coastguard Worker
433*48a54d36SAndroid Build Coastguard Worker // Build reply body
434*48a54d36SAndroid Build Coastguard Worker data = (char *)&(*rep)->rhdr[1];
435*48a54d36SAndroid Build Coastguard Worker put_string(namestr, &data);
436*48a54d36SAndroid Build Coastguard Worker put_string(typestr, &data);
437*48a54d36SAndroid Build Coastguard Worker put_string(domstr, &data);
438*48a54d36SAndroid Build Coastguard Worker
439*48a54d36SAndroid Build Coastguard Worker return mStatus_NoError;
440*48a54d36SAndroid Build Coastguard Worker }
441*48a54d36SAndroid Build Coastguard Worker }
442*48a54d36SAndroid Build Coastguard Worker
443*48a54d36SAndroid Build Coastguard Worker // Special support to enable the DNSServiceBrowse call made by Bonjour Browser
444*48a54d36SAndroid Build Coastguard Worker // Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse
GenerateBonjourBrowserResponse(const domainname * const servicename,const mDNSInterfaceID id,request_state * const request,reply_state ** const rep,reply_op_t op,DNSServiceFlags flags,mStatus err)445*48a54d36SAndroid Build Coastguard Worker mDNSlocal void GenerateBonjourBrowserResponse(const domainname *const servicename, const mDNSInterfaceID id,
446*48a54d36SAndroid Build Coastguard Worker request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err)
447*48a54d36SAndroid Build Coastguard Worker {
448*48a54d36SAndroid Build Coastguard Worker char namestr[MAX_DOMAIN_LABEL+1];
449*48a54d36SAndroid Build Coastguard Worker char typestr[MAX_ESCAPED_DOMAIN_NAME];
450*48a54d36SAndroid Build Coastguard Worker static const char domstr[] = ".";
451*48a54d36SAndroid Build Coastguard Worker int len;
452*48a54d36SAndroid Build Coastguard Worker char *data;
453*48a54d36SAndroid Build Coastguard Worker
454*48a54d36SAndroid Build Coastguard Worker *rep = NULL;
455*48a54d36SAndroid Build Coastguard Worker
456*48a54d36SAndroid Build Coastguard Worker // 1. Put first label in namestr
457*48a54d36SAndroid Build Coastguard Worker ConvertDomainLabelToCString_unescaped((const domainlabel *)servicename, namestr);
458*48a54d36SAndroid Build Coastguard Worker
459*48a54d36SAndroid Build Coastguard Worker // 2. Put second label and "local" into typestr
460*48a54d36SAndroid Build Coastguard Worker mDNS_snprintf(typestr, sizeof(typestr), "%#s.local.", SecondLabel(servicename));
461*48a54d36SAndroid Build Coastguard Worker
462*48a54d36SAndroid Build Coastguard Worker // Calculate reply data length
463*48a54d36SAndroid Build Coastguard Worker len = sizeof(DNSServiceFlags);
464*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32); // if index
465*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceErrorType);
466*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(namestr) + 1);
467*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(typestr) + 1);
468*48a54d36SAndroid Build Coastguard Worker len += (int) (strlen(domstr) + 1);
469*48a54d36SAndroid Build Coastguard Worker
470*48a54d36SAndroid Build Coastguard Worker // Build reply header
471*48a54d36SAndroid Build Coastguard Worker *rep = create_reply(op, len, request);
472*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->flags = dnssd_htonl(flags);
473*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse));
474*48a54d36SAndroid Build Coastguard Worker (*rep)->rhdr->error = dnssd_htonl(err);
475*48a54d36SAndroid Build Coastguard Worker
476*48a54d36SAndroid Build Coastguard Worker // Build reply body
477*48a54d36SAndroid Build Coastguard Worker data = (char *)&(*rep)->rhdr[1];
478*48a54d36SAndroid Build Coastguard Worker put_string(namestr, &data);
479*48a54d36SAndroid Build Coastguard Worker put_string(typestr, &data);
480*48a54d36SAndroid Build Coastguard Worker put_string(domstr, &data);
481*48a54d36SAndroid Build Coastguard Worker }
482*48a54d36SAndroid Build Coastguard Worker
483*48a54d36SAndroid Build Coastguard Worker // Returns a resource record (allocated w/ malloc) containing the data found in an IPC message
484*48a54d36SAndroid Build Coastguard Worker // Data must be in the following format: flags, interfaceIndex, name, rrtype, rrclass, rdlen, rdata, (optional) ttl
485*48a54d36SAndroid Build Coastguard Worker // (ttl only extracted/set if ttl argument is non-zero). Returns NULL for a bad-parameter error
read_rr_from_ipc_msg(request_state * request,int GetTTL,int validate_flags)486*48a54d36SAndroid Build Coastguard Worker mDNSlocal AuthRecord *read_rr_from_ipc_msg(request_state *request, int GetTTL, int validate_flags)
487*48a54d36SAndroid Build Coastguard Worker {
488*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
489*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
490*48a54d36SAndroid Build Coastguard Worker char name[256];
491*48a54d36SAndroid Build Coastguard Worker int str_err = get_string(&request->msgptr, request->msgend, name, sizeof(name));
492*48a54d36SAndroid Build Coastguard Worker mDNSu16 type = get_uint16(&request->msgptr, request->msgend);
493*48a54d36SAndroid Build Coastguard Worker mDNSu16 class = get_uint16(&request->msgptr, request->msgend);
494*48a54d36SAndroid Build Coastguard Worker mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
495*48a54d36SAndroid Build Coastguard Worker const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen);
496*48a54d36SAndroid Build Coastguard Worker mDNSu32 ttl = GetTTL ? get_uint32(&request->msgptr, request->msgend) : 0;
497*48a54d36SAndroid Build Coastguard Worker int storage_size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody);
498*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr;
499*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID;
500*48a54d36SAndroid Build Coastguard Worker AuthRecType artype;
501*48a54d36SAndroid Build Coastguard Worker
502*48a54d36SAndroid Build Coastguard Worker request->flags = flags;
503*48a54d36SAndroid Build Coastguard Worker
504*48a54d36SAndroid Build Coastguard Worker if (str_err) { LogMsg("ERROR: read_rr_from_ipc_msg - get_string"); return NULL; }
505*48a54d36SAndroid Build Coastguard Worker
506*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("Error reading Resource Record from client"); return NULL; }
507*48a54d36SAndroid Build Coastguard Worker
508*48a54d36SAndroid Build Coastguard Worker if (validate_flags &&
509*48a54d36SAndroid Build Coastguard Worker !((flags & kDNSServiceFlagsShared) == kDNSServiceFlagsShared) &&
510*48a54d36SAndroid Build Coastguard Worker !((flags & kDNSServiceFlagsUnique) == kDNSServiceFlagsUnique))
511*48a54d36SAndroid Build Coastguard Worker {
512*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: Bad resource record flags (must be kDNSServiceFlagsShared or kDNSServiceFlagsUnique)");
513*48a54d36SAndroid Build Coastguard Worker return NULL;
514*48a54d36SAndroid Build Coastguard Worker }
515*48a54d36SAndroid Build Coastguard Worker
516*48a54d36SAndroid Build Coastguard Worker rr = mallocL("AuthRecord/read_rr_from_ipc_msg", sizeof(AuthRecord) - sizeof(RDataBody) + storage_size);
517*48a54d36SAndroid Build Coastguard Worker if (!rr) FatalError("ERROR: malloc");
518*48a54d36SAndroid Build Coastguard Worker
519*48a54d36SAndroid Build Coastguard Worker InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
520*48a54d36SAndroid Build Coastguard Worker if (InterfaceID == mDNSInterface_LocalOnly)
521*48a54d36SAndroid Build Coastguard Worker artype = AuthRecordLocalOnly;
522*48a54d36SAndroid Build Coastguard Worker else if (InterfaceID == mDNSInterface_P2P)
523*48a54d36SAndroid Build Coastguard Worker artype = AuthRecordP2P;
524*48a54d36SAndroid Build Coastguard Worker else if ((InterfaceID == mDNSInterface_Any) && (flags & kDNSServiceFlagsIncludeP2P))
525*48a54d36SAndroid Build Coastguard Worker artype = AuthRecordAnyIncludeP2P;
526*48a54d36SAndroid Build Coastguard Worker else
527*48a54d36SAndroid Build Coastguard Worker artype = AuthRecordAny;
528*48a54d36SAndroid Build Coastguard Worker
529*48a54d36SAndroid Build Coastguard Worker mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, type, 0,
530*48a54d36SAndroid Build Coastguard Worker (mDNSu8) ((flags & kDNSServiceFlagsShared) ? kDNSRecordTypeShared : kDNSRecordTypeUnique), artype, mDNSNULL, mDNSNULL);
531*48a54d36SAndroid Build Coastguard Worker
532*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&rr->namestorage, name))
533*48a54d36SAndroid Build Coastguard Worker {
534*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: bad name: %s", name);
535*48a54d36SAndroid Build Coastguard Worker freeL("AuthRecord/read_rr_from_ipc_msg", rr);
536*48a54d36SAndroid Build Coastguard Worker return NULL;
537*48a54d36SAndroid Build Coastguard Worker }
538*48a54d36SAndroid Build Coastguard Worker
539*48a54d36SAndroid Build Coastguard Worker if (flags & kDNSServiceFlagsAllowRemoteQuery) rr->AllowRemoteQuery = mDNStrue;
540*48a54d36SAndroid Build Coastguard Worker rr->resrec.rrclass = class;
541*48a54d36SAndroid Build Coastguard Worker rr->resrec.rdlength = rdlen;
542*48a54d36SAndroid Build Coastguard Worker rr->resrec.rdata->MaxRDLength = rdlen;
543*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(rr->resrec.rdata->u.data, rdata, rdlen);
544*48a54d36SAndroid Build Coastguard Worker if (GetTTL) rr->resrec.rroriginalttl = ttl;
545*48a54d36SAndroid Build Coastguard Worker rr->resrec.namehash = DomainNameHashValue(rr->resrec.name);
546*48a54d36SAndroid Build Coastguard Worker SetNewRData(&rr->resrec, mDNSNULL, 0); // Sets rr->rdatahash for us
547*48a54d36SAndroid Build Coastguard Worker return rr;
548*48a54d36SAndroid Build Coastguard Worker }
549*48a54d36SAndroid Build Coastguard Worker
build_domainname_from_strings(domainname * srv,char * name,char * regtype,char * domain)550*48a54d36SAndroid Build Coastguard Worker mDNSlocal int build_domainname_from_strings(domainname *srv, char *name, char *regtype, char *domain)
551*48a54d36SAndroid Build Coastguard Worker {
552*48a54d36SAndroid Build Coastguard Worker domainlabel n;
553*48a54d36SAndroid Build Coastguard Worker domainname d, t;
554*48a54d36SAndroid Build Coastguard Worker
555*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainLabelFromLiteralString(&n, name)) return -1;
556*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&t, regtype)) return -1;
557*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&d, domain)) return -1;
558*48a54d36SAndroid Build Coastguard Worker if (!ConstructServiceName(srv, &n, &t, &d)) return -1;
559*48a54d36SAndroid Build Coastguard Worker return 0;
560*48a54d36SAndroid Build Coastguard Worker }
561*48a54d36SAndroid Build Coastguard Worker
send_all(dnssd_sock_t s,const char * ptr,int len)562*48a54d36SAndroid Build Coastguard Worker mDNSlocal void send_all(dnssd_sock_t s, const char *ptr, int len)
563*48a54d36SAndroid Build Coastguard Worker {
564*48a54d36SAndroid Build Coastguard Worker int n = send(s, ptr, len, 0);
565*48a54d36SAndroid Build Coastguard Worker // On a freshly-created Unix Domain Socket, the kernel should *never* fail to buffer a small write for us
566*48a54d36SAndroid Build Coastguard Worker // (four bytes for a typical error code return, 12 bytes for DNSServiceGetProperty(DaemonVersion)).
567*48a54d36SAndroid Build Coastguard Worker // If it does fail, we don't attempt to handle this failure, but we do log it so we know something is wrong.
568*48a54d36SAndroid Build Coastguard Worker if (n < len)
569*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: send_all(%d) wrote %d of %d errno %d (%s)",
570*48a54d36SAndroid Build Coastguard Worker s, n, len, dnssd_errno, dnssd_strerror(dnssd_errno));
571*48a54d36SAndroid Build Coastguard Worker }
572*48a54d36SAndroid Build Coastguard Worker
573*48a54d36SAndroid Build Coastguard Worker #if 0
574*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSBool AuthorizedDomain(const request_state * const request, const domainname * const d, const DNameListElem * const doms)
575*48a54d36SAndroid Build Coastguard Worker {
576*48a54d36SAndroid Build Coastguard Worker const DNameListElem *delem = mDNSNULL;
577*48a54d36SAndroid Build Coastguard Worker int bestDelta = -1; // the delta of the best match, lower is better
578*48a54d36SAndroid Build Coastguard Worker int dLabels = 0;
579*48a54d36SAndroid Build Coastguard Worker mDNSBool allow = mDNSfalse;
580*48a54d36SAndroid Build Coastguard Worker
581*48a54d36SAndroid Build Coastguard Worker if (SystemUID(request->uid)) return mDNStrue;
582*48a54d36SAndroid Build Coastguard Worker
583*48a54d36SAndroid Build Coastguard Worker dLabels = CountLabels(d);
584*48a54d36SAndroid Build Coastguard Worker for (delem = doms; delem; delem = delem->next)
585*48a54d36SAndroid Build Coastguard Worker {
586*48a54d36SAndroid Build Coastguard Worker if (delem->uid)
587*48a54d36SAndroid Build Coastguard Worker {
588*48a54d36SAndroid Build Coastguard Worker int delemLabels = CountLabels(&delem->name);
589*48a54d36SAndroid Build Coastguard Worker int delta = dLabels - delemLabels;
590*48a54d36SAndroid Build Coastguard Worker if ((bestDelta == -1 || delta <= bestDelta) && SameDomainName(&delem->name, SkipLeadingLabels(d, delta)))
591*48a54d36SAndroid Build Coastguard Worker {
592*48a54d36SAndroid Build Coastguard Worker bestDelta = delta;
593*48a54d36SAndroid Build Coastguard Worker allow = (allow || (delem->uid == request->uid));
594*48a54d36SAndroid Build Coastguard Worker }
595*48a54d36SAndroid Build Coastguard Worker }
596*48a54d36SAndroid Build Coastguard Worker }
597*48a54d36SAndroid Build Coastguard Worker
598*48a54d36SAndroid Build Coastguard Worker return bestDelta == -1 ? mDNStrue : allow;
599*48a54d36SAndroid Build Coastguard Worker }
600*48a54d36SAndroid Build Coastguard Worker #endif
601*48a54d36SAndroid Build Coastguard Worker
602*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
603*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
604*48a54d36SAndroid Build Coastguard Worker #pragma mark -
605*48a54d36SAndroid Build Coastguard Worker #pragma mark - external helpers
606*48a54d36SAndroid Build Coastguard Worker #endif
607*48a54d36SAndroid Build Coastguard Worker
external_start_advertising_helper(service_instance * const instance)608*48a54d36SAndroid Build Coastguard Worker mDNSlocal void external_start_advertising_helper(service_instance *const instance)
609*48a54d36SAndroid Build Coastguard Worker {
610*48a54d36SAndroid Build Coastguard Worker AuthRecord *st = instance->subtypes;
611*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *e;
612*48a54d36SAndroid Build Coastguard Worker int i;
613*48a54d36SAndroid Build Coastguard Worker
614*48a54d36SAndroid Build Coastguard Worker if (mDNSIPPortIsZero(instance->request->u.servicereg.port))
615*48a54d36SAndroid Build Coastguard Worker {
616*48a54d36SAndroid Build Coastguard Worker LogInfo("external_start_advertising_helper: Not registering service with port number zero");
617*48a54d36SAndroid Build Coastguard Worker return;
618*48a54d36SAndroid Build Coastguard Worker }
619*48a54d36SAndroid Build Coastguard Worker
620*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
621*48a54d36SAndroid Build Coastguard Worker // Update packet filter if p2p interface already exists, otherwise,
622*48a54d36SAndroid Build Coastguard Worker // if will be updated when we get the KEV_DL_IF_ATTACHED event for
623*48a54d36SAndroid Build Coastguard Worker // the interface. Called here since we don't call external_start_advertising_service()
624*48a54d36SAndroid Build Coastguard Worker // with the SRV record when advertising a service.
625*48a54d36SAndroid Build Coastguard Worker mDNSInitPacketFilter();
626*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
627*48a54d36SAndroid Build Coastguard Worker
628*48a54d36SAndroid Build Coastguard Worker if (instance->external_advertise) LogMsg("external_start_advertising_helper: external_advertise already set!");
629*48a54d36SAndroid Build Coastguard Worker
630*48a54d36SAndroid Build Coastguard Worker for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++)
631*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&st[i].resrec);
632*48a54d36SAndroid Build Coastguard Worker
633*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&instance->srs.RR_PTR.resrec);
634*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&instance->srs.RR_TXT.resrec);
635*48a54d36SAndroid Build Coastguard Worker
636*48a54d36SAndroid Build Coastguard Worker for (e = instance->srs.Extras; e; e = e->next)
637*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&e->r.resrec);
638*48a54d36SAndroid Build Coastguard Worker
639*48a54d36SAndroid Build Coastguard Worker instance->external_advertise = mDNStrue;
640*48a54d36SAndroid Build Coastguard Worker }
641*48a54d36SAndroid Build Coastguard Worker
external_stop_advertising_helper(service_instance * const instance)642*48a54d36SAndroid Build Coastguard Worker mDNSlocal void external_stop_advertising_helper(service_instance *const instance)
643*48a54d36SAndroid Build Coastguard Worker {
644*48a54d36SAndroid Build Coastguard Worker AuthRecord *st = instance->subtypes;
645*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *e;
646*48a54d36SAndroid Build Coastguard Worker int i;
647*48a54d36SAndroid Build Coastguard Worker
648*48a54d36SAndroid Build Coastguard Worker if (!instance->external_advertise) return;
649*48a54d36SAndroid Build Coastguard Worker
650*48a54d36SAndroid Build Coastguard Worker LogInfo("external_stop_advertising_helper: calling external_stop_advertising_service");
651*48a54d36SAndroid Build Coastguard Worker
652*48a54d36SAndroid Build Coastguard Worker for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++)
653*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&st[i].resrec);
654*48a54d36SAndroid Build Coastguard Worker
655*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&instance->srs.RR_PTR.resrec);
656*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&instance->srs.RR_TXT.resrec);
657*48a54d36SAndroid Build Coastguard Worker
658*48a54d36SAndroid Build Coastguard Worker for (e = instance->srs.Extras; e; e = e->next)
659*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&e->r.resrec);
660*48a54d36SAndroid Build Coastguard Worker
661*48a54d36SAndroid Build Coastguard Worker instance->external_advertise = mDNSfalse;
662*48a54d36SAndroid Build Coastguard Worker }
663*48a54d36SAndroid Build Coastguard Worker
664*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
665*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
666*48a54d36SAndroid Build Coastguard Worker #pragma mark -
667*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceRegister
668*48a54d36SAndroid Build Coastguard Worker #endif
669*48a54d36SAndroid Build Coastguard Worker
FreeExtraRR(mDNS * const m,AuthRecord * const rr,mStatus result)670*48a54d36SAndroid Build Coastguard Worker mDNSexport void FreeExtraRR(mDNS *const m, AuthRecord *const rr, mStatus result)
671*48a54d36SAndroid Build Coastguard Worker {
672*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *extra = (ExtraResourceRecord *)rr->RecordContext;
673*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
674*48a54d36SAndroid Build Coastguard Worker
675*48a54d36SAndroid Build Coastguard Worker if (result != mStatus_MemFree) { LogMsg("Error: FreeExtraRR invoked with unexpected error %d", result); return; }
676*48a54d36SAndroid Build Coastguard Worker
677*48a54d36SAndroid Build Coastguard Worker LogInfo(" FreeExtraRR %s", RRDisplayString(m, &rr->resrec));
678*48a54d36SAndroid Build Coastguard Worker
679*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rdata != &rr->rdatastorage)
680*48a54d36SAndroid Build Coastguard Worker freeL("Extra RData", rr->resrec.rdata);
681*48a54d36SAndroid Build Coastguard Worker freeL("ExtraResourceRecord/FreeExtraRR", extra);
682*48a54d36SAndroid Build Coastguard Worker }
683*48a54d36SAndroid Build Coastguard Worker
unlink_and_free_service_instance(service_instance * srv)684*48a54d36SAndroid Build Coastguard Worker mDNSlocal void unlink_and_free_service_instance(service_instance *srv)
685*48a54d36SAndroid Build Coastguard Worker {
686*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *e = srv->srs.Extras, *tmp;
687*48a54d36SAndroid Build Coastguard Worker
688*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_helper(srv);
689*48a54d36SAndroid Build Coastguard Worker
690*48a54d36SAndroid Build Coastguard Worker // clear pointers from parent struct
691*48a54d36SAndroid Build Coastguard Worker if (srv->request)
692*48a54d36SAndroid Build Coastguard Worker {
693*48a54d36SAndroid Build Coastguard Worker service_instance **p = &srv->request->u.servicereg.instances;
694*48a54d36SAndroid Build Coastguard Worker while (*p)
695*48a54d36SAndroid Build Coastguard Worker {
696*48a54d36SAndroid Build Coastguard Worker if (*p == srv) { *p = (*p)->next; break; }
697*48a54d36SAndroid Build Coastguard Worker p = &(*p)->next;
698*48a54d36SAndroid Build Coastguard Worker }
699*48a54d36SAndroid Build Coastguard Worker }
700*48a54d36SAndroid Build Coastguard Worker
701*48a54d36SAndroid Build Coastguard Worker while (e)
702*48a54d36SAndroid Build Coastguard Worker {
703*48a54d36SAndroid Build Coastguard Worker e->r.RecordContext = e;
704*48a54d36SAndroid Build Coastguard Worker tmp = e;
705*48a54d36SAndroid Build Coastguard Worker e = e->next;
706*48a54d36SAndroid Build Coastguard Worker FreeExtraRR(&mDNSStorage, &tmp->r, mStatus_MemFree);
707*48a54d36SAndroid Build Coastguard Worker }
708*48a54d36SAndroid Build Coastguard Worker
709*48a54d36SAndroid Build Coastguard Worker if (srv->srs.RR_TXT.resrec.rdata != &srv->srs.RR_TXT.rdatastorage)
710*48a54d36SAndroid Build Coastguard Worker freeL("TXT RData", srv->srs.RR_TXT.resrec.rdata);
711*48a54d36SAndroid Build Coastguard Worker
712*48a54d36SAndroid Build Coastguard Worker if (srv->subtypes) { freeL("ServiceSubTypes", srv->subtypes); srv->subtypes = NULL; }
713*48a54d36SAndroid Build Coastguard Worker freeL("service_instance", srv);
714*48a54d36SAndroid Build Coastguard Worker }
715*48a54d36SAndroid Build Coastguard Worker
716*48a54d36SAndroid Build Coastguard Worker // Count how many other service records we have locally with the same name, but different rdata.
717*48a54d36SAndroid Build Coastguard Worker // For auto-named services, we can have at most one per machine -- if we allowed two auto-named services of
718*48a54d36SAndroid Build Coastguard Worker // the same type on the same machine, we'd get into an infinite autoimmune-response loop of continuous renaming.
CountPeerRegistrations(mDNS * const m,ServiceRecordSet * const srs)719*48a54d36SAndroid Build Coastguard Worker mDNSexport int CountPeerRegistrations(mDNS *const m, ServiceRecordSet *const srs)
720*48a54d36SAndroid Build Coastguard Worker {
721*48a54d36SAndroid Build Coastguard Worker int count = 0;
722*48a54d36SAndroid Build Coastguard Worker ResourceRecord *r = &srs->RR_SRV.resrec;
723*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr;
724*48a54d36SAndroid Build Coastguard Worker
725*48a54d36SAndroid Build Coastguard Worker for (rr = m->ResourceRecords; rr; rr=rr->next)
726*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rrtype == kDNSType_SRV && SameDomainName(rr->resrec.name, r->name) && !IdenticalSameNameRecord(&rr->resrec, r))
727*48a54d36SAndroid Build Coastguard Worker count++;
728*48a54d36SAndroid Build Coastguard Worker
729*48a54d36SAndroid Build Coastguard Worker verbosedebugf("%d peer registrations for %##s", count, r->name->c);
730*48a54d36SAndroid Build Coastguard Worker return(count);
731*48a54d36SAndroid Build Coastguard Worker }
732*48a54d36SAndroid Build Coastguard Worker
CountExistingRegistrations(domainname * srv,mDNSIPPort port)733*48a54d36SAndroid Build Coastguard Worker mDNSexport int CountExistingRegistrations(domainname *srv, mDNSIPPort port)
734*48a54d36SAndroid Build Coastguard Worker {
735*48a54d36SAndroid Build Coastguard Worker int count = 0;
736*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr;
737*48a54d36SAndroid Build Coastguard Worker for (rr = mDNSStorage.ResourceRecords; rr; rr=rr->next)
738*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rrtype == kDNSType_SRV &&
739*48a54d36SAndroid Build Coastguard Worker mDNSSameIPPort(rr->resrec.rdata->u.srv.port, port) &&
740*48a54d36SAndroid Build Coastguard Worker SameDomainName(rr->resrec.name, srv))
741*48a54d36SAndroid Build Coastguard Worker count++;
742*48a54d36SAndroid Build Coastguard Worker return(count);
743*48a54d36SAndroid Build Coastguard Worker }
744*48a54d36SAndroid Build Coastguard Worker
SendServiceRemovalNotification(ServiceRecordSet * const srs)745*48a54d36SAndroid Build Coastguard Worker mDNSlocal void SendServiceRemovalNotification(ServiceRecordSet *const srs)
746*48a54d36SAndroid Build Coastguard Worker {
747*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
748*48a54d36SAndroid Build Coastguard Worker service_instance *instance = srs->ServiceContext;
749*48a54d36SAndroid Build Coastguard Worker if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, 0, mStatus_NoError) != mStatus_NoError)
750*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: SendServiceRemovalNotification: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c);
751*48a54d36SAndroid Build Coastguard Worker else { append_reply(instance->request, rep); instance->clientnotified = mDNSfalse; }
752*48a54d36SAndroid Build Coastguard Worker }
753*48a54d36SAndroid Build Coastguard Worker
754*48a54d36SAndroid Build Coastguard Worker // service registration callback performs three duties - frees memory for deregistered services,
755*48a54d36SAndroid Build Coastguard Worker // handles name conflicts, and delivers completed registration information to the client
regservice_callback(mDNS * const m,ServiceRecordSet * const srs,mStatus result)756*48a54d36SAndroid Build Coastguard Worker mDNSlocal void regservice_callback(mDNS *const m, ServiceRecordSet *const srs, mStatus result)
757*48a54d36SAndroid Build Coastguard Worker {
758*48a54d36SAndroid Build Coastguard Worker mStatus err;
759*48a54d36SAndroid Build Coastguard Worker mDNSBool SuppressError = mDNSfalse;
760*48a54d36SAndroid Build Coastguard Worker service_instance *instance;
761*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
762*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
763*48a54d36SAndroid Build Coastguard Worker
764*48a54d36SAndroid Build Coastguard Worker if (!srs) { LogMsg("regservice_callback: srs is NULL %d", result); return; }
765*48a54d36SAndroid Build Coastguard Worker
766*48a54d36SAndroid Build Coastguard Worker instance = srs->ServiceContext;
767*48a54d36SAndroid Build Coastguard Worker if (!instance) { LogMsg("regservice_callback: srs->ServiceContext is NULL %d", result); return; }
768*48a54d36SAndroid Build Coastguard Worker
769*48a54d36SAndroid Build Coastguard Worker // don't send errors up to client for wide-area, empty-string registrations
770*48a54d36SAndroid Build Coastguard Worker if (instance->request &&
771*48a54d36SAndroid Build Coastguard Worker instance->request->u.servicereg.default_domain &&
772*48a54d36SAndroid Build Coastguard Worker !instance->default_local)
773*48a54d36SAndroid Build Coastguard Worker SuppressError = mDNStrue;
774*48a54d36SAndroid Build Coastguard Worker
775*48a54d36SAndroid Build Coastguard Worker if (mDNS_LoggingEnabled)
776*48a54d36SAndroid Build Coastguard Worker {
777*48a54d36SAndroid Build Coastguard Worker const char *const fmt =
778*48a54d36SAndroid Build Coastguard Worker (result == mStatus_NoError) ? "%s DNSServiceRegister(%##s, %u) REGISTERED" :
779*48a54d36SAndroid Build Coastguard Worker (result == mStatus_MemFree) ? "%s DNSServiceRegister(%##s, %u) DEREGISTERED" :
780*48a54d36SAndroid Build Coastguard Worker (result == mStatus_NameConflict) ? "%s DNSServiceRegister(%##s, %u) NAME CONFLICT" :
781*48a54d36SAndroid Build Coastguard Worker "%s DNSServiceRegister(%##s, %u) %s %d";
782*48a54d36SAndroid Build Coastguard Worker char prefix[16] = "---:";
783*48a54d36SAndroid Build Coastguard Worker if (instance->request) mDNS_snprintf(prefix, sizeof(prefix), "%3d:", instance->request->sd);
784*48a54d36SAndroid Build Coastguard Worker LogOperation(fmt, prefix, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port),
785*48a54d36SAndroid Build Coastguard Worker SuppressError ? "suppressed error" : "CALLBACK", result);
786*48a54d36SAndroid Build Coastguard Worker }
787*48a54d36SAndroid Build Coastguard Worker
788*48a54d36SAndroid Build Coastguard Worker if (!instance->request && result != mStatus_MemFree) { LogMsg("regservice_callback: instance->request is NULL %d", result); return; }
789*48a54d36SAndroid Build Coastguard Worker
790*48a54d36SAndroid Build Coastguard Worker if (result == mStatus_NoError)
791*48a54d36SAndroid Build Coastguard Worker {
792*48a54d36SAndroid Build Coastguard Worker if (instance->request->u.servicereg.allowremotequery)
793*48a54d36SAndroid Build Coastguard Worker {
794*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *e;
795*48a54d36SAndroid Build Coastguard Worker srs->RR_ADV.AllowRemoteQuery = mDNStrue;
796*48a54d36SAndroid Build Coastguard Worker srs->RR_PTR.AllowRemoteQuery = mDNStrue;
797*48a54d36SAndroid Build Coastguard Worker srs->RR_SRV.AllowRemoteQuery = mDNStrue;
798*48a54d36SAndroid Build Coastguard Worker srs->RR_TXT.AllowRemoteQuery = mDNStrue;
799*48a54d36SAndroid Build Coastguard Worker for (e = instance->srs.Extras; e; e = e->next) e->r.AllowRemoteQuery = mDNStrue;
800*48a54d36SAndroid Build Coastguard Worker }
801*48a54d36SAndroid Build Coastguard Worker
802*48a54d36SAndroid Build Coastguard Worker if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
803*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c);
804*48a54d36SAndroid Build Coastguard Worker else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
805*48a54d36SAndroid Build Coastguard Worker
806*48a54d36SAndroid Build Coastguard Worker if (instance->request->u.servicereg.InterfaceID == mDNSInterface_P2P || (!instance->request->u.servicereg.InterfaceID && SameDomainName(&instance->domain, &localdomain) && (instance->request->flags & kDNSServiceFlagsIncludeP2P)))
807*48a54d36SAndroid Build Coastguard Worker {
808*48a54d36SAndroid Build Coastguard Worker LogInfo("regservice_callback: calling external_start_advertising_helper()");
809*48a54d36SAndroid Build Coastguard Worker external_start_advertising_helper(instance);
810*48a54d36SAndroid Build Coastguard Worker }
811*48a54d36SAndroid Build Coastguard Worker if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0)
812*48a54d36SAndroid Build Coastguard Worker RecordUpdatedNiceLabel(m, 0); // Successfully got new name, tell user immediately
813*48a54d36SAndroid Build Coastguard Worker }
814*48a54d36SAndroid Build Coastguard Worker else if (result == mStatus_MemFree)
815*48a54d36SAndroid Build Coastguard Worker {
816*48a54d36SAndroid Build Coastguard Worker if (instance->request && instance->renameonmemfree)
817*48a54d36SAndroid Build Coastguard Worker {
818*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_helper(instance);
819*48a54d36SAndroid Build Coastguard Worker instance->renameonmemfree = 0;
820*48a54d36SAndroid Build Coastguard Worker err = mDNS_RenameAndReregisterService(m, srs, &instance->request->u.servicereg.name);
821*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("ERROR: regservice_callback - RenameAndReregisterService returned %d", err);
822*48a54d36SAndroid Build Coastguard Worker // error should never happen - safest to log and continue
823*48a54d36SAndroid Build Coastguard Worker }
824*48a54d36SAndroid Build Coastguard Worker else
825*48a54d36SAndroid Build Coastguard Worker unlink_and_free_service_instance(instance);
826*48a54d36SAndroid Build Coastguard Worker }
827*48a54d36SAndroid Build Coastguard Worker else if (result == mStatus_NameConflict)
828*48a54d36SAndroid Build Coastguard Worker {
829*48a54d36SAndroid Build Coastguard Worker if (instance->request->u.servicereg.autorename)
830*48a54d36SAndroid Build Coastguard Worker {
831*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_helper(instance);
832*48a54d36SAndroid Build Coastguard Worker if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0)
833*48a54d36SAndroid Build Coastguard Worker {
834*48a54d36SAndroid Build Coastguard Worker // On conflict for an autoname service, rename and reregister *all* autoname services
835*48a54d36SAndroid Build Coastguard Worker IncrementLabelSuffix(&m->nicelabel, mDNStrue);
836*48a54d36SAndroid Build Coastguard Worker mDNS_ConfigChanged(m); // Will call back into udsserver_handle_configchange()
837*48a54d36SAndroid Build Coastguard Worker }
838*48a54d36SAndroid Build Coastguard Worker else // On conflict for a non-autoname service, rename and reregister just that one service
839*48a54d36SAndroid Build Coastguard Worker {
840*48a54d36SAndroid Build Coastguard Worker if (instance->clientnotified) SendServiceRemovalNotification(srs);
841*48a54d36SAndroid Build Coastguard Worker mDNS_RenameAndReregisterService(m, srs, mDNSNULL);
842*48a54d36SAndroid Build Coastguard Worker }
843*48a54d36SAndroid Build Coastguard Worker }
844*48a54d36SAndroid Build Coastguard Worker else
845*48a54d36SAndroid Build Coastguard Worker {
846*48a54d36SAndroid Build Coastguard Worker if (!SuppressError)
847*48a54d36SAndroid Build Coastguard Worker {
848*48a54d36SAndroid Build Coastguard Worker if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
849*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c);
850*48a54d36SAndroid Build Coastguard Worker else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
851*48a54d36SAndroid Build Coastguard Worker }
852*48a54d36SAndroid Build Coastguard Worker unlink_and_free_service_instance(instance);
853*48a54d36SAndroid Build Coastguard Worker }
854*48a54d36SAndroid Build Coastguard Worker }
855*48a54d36SAndroid Build Coastguard Worker else // Not mStatus_NoError, mStatus_MemFree, or mStatus_NameConflict
856*48a54d36SAndroid Build Coastguard Worker {
857*48a54d36SAndroid Build Coastguard Worker if (!SuppressError)
858*48a54d36SAndroid Build Coastguard Worker {
859*48a54d36SAndroid Build Coastguard Worker if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError)
860*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c);
861*48a54d36SAndroid Build Coastguard Worker else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; }
862*48a54d36SAndroid Build Coastguard Worker }
863*48a54d36SAndroid Build Coastguard Worker }
864*48a54d36SAndroid Build Coastguard Worker }
865*48a54d36SAndroid Build Coastguard Worker
regrecord_callback(mDNS * const m,AuthRecord * rr,mStatus result)866*48a54d36SAndroid Build Coastguard Worker mDNSlocal void regrecord_callback(mDNS *const m, AuthRecord *rr, mStatus result)
867*48a54d36SAndroid Build Coastguard Worker {
868*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
869*48a54d36SAndroid Build Coastguard Worker if (!rr->RecordContext) // parent struct already freed by termination callback
870*48a54d36SAndroid Build Coastguard Worker {
871*48a54d36SAndroid Build Coastguard Worker if (result == mStatus_NoError)
872*48a54d36SAndroid Build Coastguard Worker LogMsg("Error: regrecord_callback: successful registration of orphaned record %s", ARDisplayString(m, rr));
873*48a54d36SAndroid Build Coastguard Worker else
874*48a54d36SAndroid Build Coastguard Worker {
875*48a54d36SAndroid Build Coastguard Worker if (result != mStatus_MemFree) LogMsg("regrecord_callback: error %d received after parent termination", result);
876*48a54d36SAndroid Build Coastguard Worker
877*48a54d36SAndroid Build Coastguard Worker // We come here when the record is being deregistered either from DNSServiceRemoveRecord or connection_termination.
878*48a54d36SAndroid Build Coastguard Worker // If the record has been updated, we need to free the rdata. Everytime we call mDNS_Update, it calls update_callback
879*48a54d36SAndroid Build Coastguard Worker // with the old rdata (so that we can free it) and stores the new rdata in "rr->resrec.rdata". This means, we need
880*48a54d36SAndroid Build Coastguard Worker // to free the latest rdata for which the update_callback was never called with.
881*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rdata != &rr->rdatastorage) freeL("RData/regrecord_callback", rr->resrec.rdata);
882*48a54d36SAndroid Build Coastguard Worker freeL("AuthRecord/regrecord_callback", rr);
883*48a54d36SAndroid Build Coastguard Worker }
884*48a54d36SAndroid Build Coastguard Worker }
885*48a54d36SAndroid Build Coastguard Worker else
886*48a54d36SAndroid Build Coastguard Worker {
887*48a54d36SAndroid Build Coastguard Worker registered_record_entry *re = rr->RecordContext;
888*48a54d36SAndroid Build Coastguard Worker request_state *request = re->request;
889*48a54d36SAndroid Build Coastguard Worker
890*48a54d36SAndroid Build Coastguard Worker if (mDNS_LoggingEnabled)
891*48a54d36SAndroid Build Coastguard Worker {
892*48a54d36SAndroid Build Coastguard Worker char *fmt = (result == mStatus_NoError) ? "%3d: DNSServiceRegisterRecord(%u %s) REGISTERED" :
893*48a54d36SAndroid Build Coastguard Worker (result == mStatus_MemFree) ? "%3d: DNSServiceRegisterRecord(%u %s) DEREGISTERED" :
894*48a54d36SAndroid Build Coastguard Worker (result == mStatus_NameConflict) ? "%3d: DNSServiceRegisterRecord(%u %s) NAME CONFLICT" :
895*48a54d36SAndroid Build Coastguard Worker "%3d: DNSServiceRegisterRecord(%u %s) %d";
896*48a54d36SAndroid Build Coastguard Worker LogOperation(fmt, request->sd, re->key, RRDisplayString(m, &rr->resrec), result);
897*48a54d36SAndroid Build Coastguard Worker }
898*48a54d36SAndroid Build Coastguard Worker
899*48a54d36SAndroid Build Coastguard Worker if (result != mStatus_MemFree)
900*48a54d36SAndroid Build Coastguard Worker {
901*48a54d36SAndroid Build Coastguard Worker int len = sizeof(DNSServiceFlags) + sizeof(mDNSu32) + sizeof(DNSServiceErrorType);
902*48a54d36SAndroid Build Coastguard Worker reply_state *reply = create_reply(reg_record_reply_op, len, request);
903*48a54d36SAndroid Build Coastguard Worker reply->mhdr->client_context = re->regrec_client_context;
904*48a54d36SAndroid Build Coastguard Worker reply->rhdr->flags = dnssd_htonl(0);
905*48a54d36SAndroid Build Coastguard Worker reply->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, rr->resrec.InterfaceID, mDNSfalse));
906*48a54d36SAndroid Build Coastguard Worker reply->rhdr->error = dnssd_htonl(result);
907*48a54d36SAndroid Build Coastguard Worker append_reply(request, reply);
908*48a54d36SAndroid Build Coastguard Worker }
909*48a54d36SAndroid Build Coastguard Worker
910*48a54d36SAndroid Build Coastguard Worker if (result)
911*48a54d36SAndroid Build Coastguard Worker {
912*48a54d36SAndroid Build Coastguard Worker // unlink from list, free memory
913*48a54d36SAndroid Build Coastguard Worker registered_record_entry **ptr = &request->u.reg_recs;
914*48a54d36SAndroid Build Coastguard Worker while (*ptr && (*ptr) != re) ptr = &(*ptr)->next;
915*48a54d36SAndroid Build Coastguard Worker if (!*ptr) { LogMsg("regrecord_callback - record not in list!"); return; }
916*48a54d36SAndroid Build Coastguard Worker *ptr = (*ptr)->next;
917*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry AuthRecord regrecord_callback", re->rr);
918*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry regrecord_callback", re);
919*48a54d36SAndroid Build Coastguard Worker }
920*48a54d36SAndroid Build Coastguard Worker else
921*48a54d36SAndroid Build Coastguard Worker {
922*48a54d36SAndroid Build Coastguard Worker if (re->external_advertise) LogMsg("regrecord_callback: external_advertise already set!");
923*48a54d36SAndroid Build Coastguard Worker
924*48a54d36SAndroid Build Coastguard Worker if (re->origInterfaceID == mDNSInterface_P2P || (!re->origInterfaceID && IsLocalDomain(&rr->namestorage) && (request->flags & kDNSServiceFlagsIncludeP2P)))
925*48a54d36SAndroid Build Coastguard Worker {
926*48a54d36SAndroid Build Coastguard Worker LogInfo("regrecord_callback: calling external_start_advertising_service");
927*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&rr->resrec);
928*48a54d36SAndroid Build Coastguard Worker re->external_advertise = mDNStrue;
929*48a54d36SAndroid Build Coastguard Worker }
930*48a54d36SAndroid Build Coastguard Worker }
931*48a54d36SAndroid Build Coastguard Worker }
932*48a54d36SAndroid Build Coastguard Worker }
933*48a54d36SAndroid Build Coastguard Worker
connection_termination(request_state * request)934*48a54d36SAndroid Build Coastguard Worker mDNSlocal void connection_termination(request_state *request)
935*48a54d36SAndroid Build Coastguard Worker {
936*48a54d36SAndroid Build Coastguard Worker // When terminating a shared connection, we need to scan the all_requests list
937*48a54d36SAndroid Build Coastguard Worker // and terminate any subbordinate operations sharing this file descriptor
938*48a54d36SAndroid Build Coastguard Worker request_state **req = &all_requests;
939*48a54d36SAndroid Build Coastguard Worker
940*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceCreateConnection STOP", request->sd);
941*48a54d36SAndroid Build Coastguard Worker
942*48a54d36SAndroid Build Coastguard Worker while (*req)
943*48a54d36SAndroid Build Coastguard Worker {
944*48a54d36SAndroid Build Coastguard Worker if ((*req)->primary == request)
945*48a54d36SAndroid Build Coastguard Worker {
946*48a54d36SAndroid Build Coastguard Worker // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
947*48a54d36SAndroid Build Coastguard Worker request_state *tmp = *req;
948*48a54d36SAndroid Build Coastguard Worker if (tmp->primary == tmp) LogMsg("connection_termination ERROR (*req)->primary == *req for %p %d", tmp, tmp->sd);
949*48a54d36SAndroid Build Coastguard Worker if (tmp->replies) LogMsg("connection_termination ERROR How can subordinate req %p %d have replies queued?", tmp, tmp->sd);
950*48a54d36SAndroid Build Coastguard Worker abort_request(tmp);
951*48a54d36SAndroid Build Coastguard Worker *req = tmp->next;
952*48a54d36SAndroid Build Coastguard Worker freeL("request_state/connection_termination", tmp);
953*48a54d36SAndroid Build Coastguard Worker }
954*48a54d36SAndroid Build Coastguard Worker else
955*48a54d36SAndroid Build Coastguard Worker req = &(*req)->next;
956*48a54d36SAndroid Build Coastguard Worker }
957*48a54d36SAndroid Build Coastguard Worker
958*48a54d36SAndroid Build Coastguard Worker while (request->u.reg_recs)
959*48a54d36SAndroid Build Coastguard Worker {
960*48a54d36SAndroid Build Coastguard Worker registered_record_entry *ptr = request->u.reg_recs;
961*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegisterRecord(%u %s) STOP", request->sd, ptr->key, RRDisplayString(&mDNSStorage, &ptr->rr->resrec));
962*48a54d36SAndroid Build Coastguard Worker request->u.reg_recs = request->u.reg_recs->next;
963*48a54d36SAndroid Build Coastguard Worker ptr->rr->RecordContext = NULL;
964*48a54d36SAndroid Build Coastguard Worker if (ptr->external_advertise)
965*48a54d36SAndroid Build Coastguard Worker {
966*48a54d36SAndroid Build Coastguard Worker ptr->external_advertise = mDNSfalse;
967*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&ptr->rr->resrec);
968*48a54d36SAndroid Build Coastguard Worker }
969*48a54d36SAndroid Build Coastguard Worker mDNS_Deregister(&mDNSStorage, ptr->rr); // Will free ptr->rr for us
970*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry/connection_termination", ptr);
971*48a54d36SAndroid Build Coastguard Worker }
972*48a54d36SAndroid Build Coastguard Worker }
973*48a54d36SAndroid Build Coastguard Worker
handle_cancel_request(request_state * request)974*48a54d36SAndroid Build Coastguard Worker mDNSlocal void handle_cancel_request(request_state *request)
975*48a54d36SAndroid Build Coastguard Worker {
976*48a54d36SAndroid Build Coastguard Worker request_state **req = &all_requests;
977*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: Cancel %08X %08X", request->sd, request->hdr.client_context.u32[1], request->hdr.client_context.u32[0]);
978*48a54d36SAndroid Build Coastguard Worker while (*req)
979*48a54d36SAndroid Build Coastguard Worker {
980*48a54d36SAndroid Build Coastguard Worker if ((*req)->primary == request &&
981*48a54d36SAndroid Build Coastguard Worker (*req)->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] &&
982*48a54d36SAndroid Build Coastguard Worker (*req)->hdr.client_context.u32[1] == request->hdr.client_context.u32[1])
983*48a54d36SAndroid Build Coastguard Worker {
984*48a54d36SAndroid Build Coastguard Worker // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
985*48a54d36SAndroid Build Coastguard Worker request_state *tmp = *req;
986*48a54d36SAndroid Build Coastguard Worker abort_request(tmp);
987*48a54d36SAndroid Build Coastguard Worker *req = tmp->next;
988*48a54d36SAndroid Build Coastguard Worker freeL("request_state/handle_cancel_request", tmp);
989*48a54d36SAndroid Build Coastguard Worker }
990*48a54d36SAndroid Build Coastguard Worker else
991*48a54d36SAndroid Build Coastguard Worker req = &(*req)->next;
992*48a54d36SAndroid Build Coastguard Worker }
993*48a54d36SAndroid Build Coastguard Worker }
994*48a54d36SAndroid Build Coastguard Worker
handle_regrecord_request(request_state * request)995*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_regrecord_request(request_state *request)
996*48a54d36SAndroid Build Coastguard Worker {
997*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_BadParamErr;
998*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr = read_rr_from_ipc_msg(request, 1, 1);
999*48a54d36SAndroid Build Coastguard Worker if (rr)
1000*48a54d36SAndroid Build Coastguard Worker {
1001*48a54d36SAndroid Build Coastguard Worker registered_record_entry *re;
1002*48a54d36SAndroid Build Coastguard Worker // Don't allow non-local domains to be regsitered as LocalOnly. Allowing this would permit
1003*48a54d36SAndroid Build Coastguard Worker // clients to register records such as www.bigbank.com A w.x.y.z to redirect Safari.
1004*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly && !IsLocalDomain(rr->resrec.name) &&
1005*48a54d36SAndroid Build Coastguard Worker rr->resrec.rrclass == kDNSClass_IN && (rr->resrec.rrtype == kDNSType_A || rr->resrec.rrtype == kDNSType_AAAA ||
1006*48a54d36SAndroid Build Coastguard Worker rr->resrec.rrtype == kDNSType_CNAME))
1007*48a54d36SAndroid Build Coastguard Worker {
1008*48a54d36SAndroid Build Coastguard Worker freeL("AuthRecord/handle_regrecord_request", rr);
1009*48a54d36SAndroid Build Coastguard Worker return (mStatus_BadParamErr);
1010*48a54d36SAndroid Build Coastguard Worker }
1011*48a54d36SAndroid Build Coastguard Worker // allocate registration entry, link into list
1012*48a54d36SAndroid Build Coastguard Worker re = mallocL("registered_record_entry", sizeof(registered_record_entry));
1013*48a54d36SAndroid Build Coastguard Worker if (!re) FatalError("ERROR: malloc");
1014*48a54d36SAndroid Build Coastguard Worker re->key = request->hdr.reg_index;
1015*48a54d36SAndroid Build Coastguard Worker re->rr = rr;
1016*48a54d36SAndroid Build Coastguard Worker re->regrec_client_context = request->hdr.client_context;
1017*48a54d36SAndroid Build Coastguard Worker re->request = request;
1018*48a54d36SAndroid Build Coastguard Worker re->external_advertise = mDNSfalse;
1019*48a54d36SAndroid Build Coastguard Worker rr->RecordContext = re;
1020*48a54d36SAndroid Build Coastguard Worker rr->RecordCallback = regrecord_callback;
1021*48a54d36SAndroid Build Coastguard Worker
1022*48a54d36SAndroid Build Coastguard Worker re->origInterfaceID = rr->resrec.InterfaceID;
1023*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.InterfaceID == mDNSInterface_P2P) rr->resrec.InterfaceID = mDNSInterface_Any;
1024*48a54d36SAndroid Build Coastguard Worker #if 0
1025*48a54d36SAndroid Build Coastguard Worker if (!AuthorizedDomain(request, rr->resrec.name, AutoRegistrationDomains)) return (mStatus_NoError);
1026*48a54d36SAndroid Build Coastguard Worker #endif
1027*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rroriginalttl == 0)
1028*48a54d36SAndroid Build Coastguard Worker rr->resrec.rroriginalttl = DefaultTTLforRRType(rr->resrec.rrtype);
1029*48a54d36SAndroid Build Coastguard Worker
1030*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegisterRecord(%u %s) START", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec));
1031*48a54d36SAndroid Build Coastguard Worker err = mDNS_Register(&mDNSStorage, rr);
1032*48a54d36SAndroid Build Coastguard Worker if (err)
1033*48a54d36SAndroid Build Coastguard Worker {
1034*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegisterRecord(%u %s) ERROR (%d)", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec), err);
1035*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry", re);
1036*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry/AuthRecord", rr);
1037*48a54d36SAndroid Build Coastguard Worker }
1038*48a54d36SAndroid Build Coastguard Worker else
1039*48a54d36SAndroid Build Coastguard Worker {
1040*48a54d36SAndroid Build Coastguard Worker re->next = request->u.reg_recs;
1041*48a54d36SAndroid Build Coastguard Worker request->u.reg_recs = re;
1042*48a54d36SAndroid Build Coastguard Worker }
1043*48a54d36SAndroid Build Coastguard Worker }
1044*48a54d36SAndroid Build Coastguard Worker return(err);
1045*48a54d36SAndroid Build Coastguard Worker }
1046*48a54d36SAndroid Build Coastguard Worker
1047*48a54d36SAndroid Build Coastguard Worker mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m);
1048*48a54d36SAndroid Build Coastguard Worker
regservice_termination_callback(request_state * request)1049*48a54d36SAndroid Build Coastguard Worker mDNSlocal void regservice_termination_callback(request_state *request)
1050*48a54d36SAndroid Build Coastguard Worker {
1051*48a54d36SAndroid Build Coastguard Worker if (!request) { LogMsg("regservice_termination_callback context is NULL"); return; }
1052*48a54d36SAndroid Build Coastguard Worker while (request->u.servicereg.instances)
1053*48a54d36SAndroid Build Coastguard Worker {
1054*48a54d36SAndroid Build Coastguard Worker service_instance *p = request->u.servicereg.instances;
1055*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.instances = request->u.servicereg.instances->next;
1056*48a54d36SAndroid Build Coastguard Worker // only safe to free memory if registration is not valid, i.e. deregister fails (which invalidates p)
1057*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegister(%##s, %u) STOP",
1058*48a54d36SAndroid Build Coastguard Worker request->sd, p->srs.RR_SRV.resrec.name->c, mDNSVal16(p->srs.RR_SRV.resrec.rdata->u.srv.port));
1059*48a54d36SAndroid Build Coastguard Worker
1060*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_helper(p);
1061*48a54d36SAndroid Build Coastguard Worker
1062*48a54d36SAndroid Build Coastguard Worker // Clear backpointer *before* calling mDNS_DeregisterService/unlink_and_free_service_instance
1063*48a54d36SAndroid Build Coastguard Worker // We don't need unlink_and_free_service_instance to cut its element from the list, because we're already advancing
1064*48a54d36SAndroid Build Coastguard Worker // request->u.servicereg.instances as we work our way through the list, implicitly cutting one element at a time
1065*48a54d36SAndroid Build Coastguard Worker // We can't clear p->request *after* the calling mDNS_DeregisterService/unlink_and_free_service_instance
1066*48a54d36SAndroid Build Coastguard Worker // because by then we might have already freed p
1067*48a54d36SAndroid Build Coastguard Worker p->request = NULL;
1068*48a54d36SAndroid Build Coastguard Worker if (mDNS_DeregisterService(&mDNSStorage, &p->srs)) unlink_and_free_service_instance(p);
1069*48a54d36SAndroid Build Coastguard Worker // Don't touch service_instance *p after this -- it's likely to have been freed already
1070*48a54d36SAndroid Build Coastguard Worker }
1071*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.txtdata)
1072*48a54d36SAndroid Build Coastguard Worker { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; }
1073*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.autoname)
1074*48a54d36SAndroid Build Coastguard Worker {
1075*48a54d36SAndroid Build Coastguard Worker // Clear autoname before calling UpdateDeviceInfoRecord() so it doesn't mistakenly include this in its count of active autoname registrations
1076*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.autoname = mDNSfalse;
1077*48a54d36SAndroid Build Coastguard Worker UpdateDeviceInfoRecord(&mDNSStorage);
1078*48a54d36SAndroid Build Coastguard Worker }
1079*48a54d36SAndroid Build Coastguard Worker }
1080*48a54d36SAndroid Build Coastguard Worker
LocateSubordinateRequest(request_state * request)1081*48a54d36SAndroid Build Coastguard Worker mDNSlocal request_state *LocateSubordinateRequest(request_state *request)
1082*48a54d36SAndroid Build Coastguard Worker {
1083*48a54d36SAndroid Build Coastguard Worker request_state *req;
1084*48a54d36SAndroid Build Coastguard Worker for (req = all_requests; req; req = req->next)
1085*48a54d36SAndroid Build Coastguard Worker if (req->primary == request &&
1086*48a54d36SAndroid Build Coastguard Worker req->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] &&
1087*48a54d36SAndroid Build Coastguard Worker req->hdr.client_context.u32[1] == request->hdr.client_context.u32[1]) return(req);
1088*48a54d36SAndroid Build Coastguard Worker return(request);
1089*48a54d36SAndroid Build Coastguard Worker }
1090*48a54d36SAndroid Build Coastguard Worker
add_record_to_service(request_state * request,service_instance * instance,mDNSu16 rrtype,mDNSu16 rdlen,const char * rdata,mDNSu32 ttl)1091*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus add_record_to_service(request_state *request, service_instance *instance, mDNSu16 rrtype, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl)
1092*48a54d36SAndroid Build Coastguard Worker {
1093*48a54d36SAndroid Build Coastguard Worker ServiceRecordSet *srs = &instance->srs;
1094*48a54d36SAndroid Build Coastguard Worker mStatus result;
1095*48a54d36SAndroid Build Coastguard Worker int size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody);
1096*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *extra = mallocL("ExtraResourceRecord", sizeof(*extra) - sizeof(RDataBody) + size);
1097*48a54d36SAndroid Build Coastguard Worker if (!extra) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; }
1098*48a54d36SAndroid Build Coastguard Worker
1099*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(extra, sizeof(ExtraResourceRecord)); // OK if oversized rdata not zero'd
1100*48a54d36SAndroid Build Coastguard Worker extra->r.resrec.rrtype = rrtype;
1101*48a54d36SAndroid Build Coastguard Worker extra->r.rdatastorage.MaxRDLength = (mDNSu16) size;
1102*48a54d36SAndroid Build Coastguard Worker extra->r.resrec.rdlength = rdlen;
1103*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(&extra->r.rdatastorage.u.data, rdata, rdlen);
1104*48a54d36SAndroid Build Coastguard Worker
1105*48a54d36SAndroid Build Coastguard Worker result = mDNS_AddRecordToService(&mDNSStorage, srs, extra, &extra->r.rdatastorage, ttl,
1106*48a54d36SAndroid Build Coastguard Worker (request->flags & kDNSServiceFlagsIncludeP2P) ? 1: 0);
1107*48a54d36SAndroid Build Coastguard Worker if (result) { freeL("ExtraResourceRecord/add_record_to_service", extra); return result; }
1108*48a54d36SAndroid Build Coastguard Worker
1109*48a54d36SAndroid Build Coastguard Worker extra->ClientID = request->hdr.reg_index;
1110*48a54d36SAndroid Build Coastguard Worker if (instance->external_advertise && (instance->request->u.servicereg.InterfaceID == mDNSInterface_P2P || (!instance->request->u.servicereg.InterfaceID && SameDomainName(&instance->domain, &localdomain) && (instance->request->flags & kDNSServiceFlagsIncludeP2P))))
1111*48a54d36SAndroid Build Coastguard Worker {
1112*48a54d36SAndroid Build Coastguard Worker LogInfo("add_record_to_service: calling external_start_advertising_service");
1113*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&extra->r.resrec);
1114*48a54d36SAndroid Build Coastguard Worker }
1115*48a54d36SAndroid Build Coastguard Worker return result;
1116*48a54d36SAndroid Build Coastguard Worker }
1117*48a54d36SAndroid Build Coastguard Worker
handle_add_request(request_state * request)1118*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_add_request(request_state *request)
1119*48a54d36SAndroid Build Coastguard Worker {
1120*48a54d36SAndroid Build Coastguard Worker service_instance *i;
1121*48a54d36SAndroid Build Coastguard Worker mStatus result = mStatus_UnknownErr;
1122*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend);
1123*48a54d36SAndroid Build Coastguard Worker mDNSu16 rrtype = get_uint16(&request->msgptr, request->msgend);
1124*48a54d36SAndroid Build Coastguard Worker mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
1125*48a54d36SAndroid Build Coastguard Worker const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen);
1126*48a54d36SAndroid Build Coastguard Worker mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend);
1127*48a54d36SAndroid Build Coastguard Worker if (!ttl) ttl = DefaultTTLforRRType(rrtype);
1128*48a54d36SAndroid Build Coastguard Worker (void)flags; // Unused
1129*48a54d36SAndroid Build Coastguard Worker
1130*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceAddRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1131*48a54d36SAndroid Build Coastguard Worker
1132*48a54d36SAndroid Build Coastguard Worker // If this is a shared connection, check if the operation actually applies to a subordinate request_state object
1133*48a54d36SAndroid Build Coastguard Worker if (request->terminate == connection_termination) request = LocateSubordinateRequest(request);
1134*48a54d36SAndroid Build Coastguard Worker
1135*48a54d36SAndroid Build Coastguard Worker if (request->terminate != regservice_termination_callback)
1136*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceAddRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); }
1137*48a54d36SAndroid Build Coastguard Worker
1138*48a54d36SAndroid Build Coastguard Worker // For a service registered with zero port, don't allow adding records. This mostly happens due to a bug
1139*48a54d36SAndroid Build Coastguard Worker // in the application. See radar://9165807.
1140*48a54d36SAndroid Build Coastguard Worker if (mDNSIPPortIsZero(request->u.servicereg.port))
1141*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceAddRecord: adding record to a service registered with zero port", request->sd); return(mStatus_BadParamErr); }
1142*48a54d36SAndroid Build Coastguard Worker
1143*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceAddRecord(%X, %##s, %s, %d)", request->sd, flags,
1144*48a54d36SAndroid Build Coastguard Worker (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, DNSTypeName(rrtype), rdlen);
1145*48a54d36SAndroid Build Coastguard Worker
1146*48a54d36SAndroid Build Coastguard Worker for (i = request->u.servicereg.instances; i; i = i->next)
1147*48a54d36SAndroid Build Coastguard Worker {
1148*48a54d36SAndroid Build Coastguard Worker result = add_record_to_service(request, i, rrtype, rdlen, rdata, ttl);
1149*48a54d36SAndroid Build Coastguard Worker if (result && i->default_local) break;
1150*48a54d36SAndroid Build Coastguard Worker else result = mStatus_NoError; // suppress non-local default errors
1151*48a54d36SAndroid Build Coastguard Worker }
1152*48a54d36SAndroid Build Coastguard Worker
1153*48a54d36SAndroid Build Coastguard Worker return(result);
1154*48a54d36SAndroid Build Coastguard Worker }
1155*48a54d36SAndroid Build Coastguard Worker
update_callback(mDNS * const m,AuthRecord * const rr,RData * oldrd,mDNSu16 oldrdlen)1156*48a54d36SAndroid Build Coastguard Worker mDNSlocal void update_callback(mDNS *const m, AuthRecord *const rr, RData *oldrd, mDNSu16 oldrdlen)
1157*48a54d36SAndroid Build Coastguard Worker {
1158*48a54d36SAndroid Build Coastguard Worker mDNSBool external_advertise = (rr->UpdateContext) ? *((mDNSBool *)rr->UpdateContext) : mDNSfalse;
1159*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
1160*48a54d36SAndroid Build Coastguard Worker
1161*48a54d36SAndroid Build Coastguard Worker // There are three cases.
1162*48a54d36SAndroid Build Coastguard Worker //
1163*48a54d36SAndroid Build Coastguard Worker // 1. We have updated the primary TXT record of the service
1164*48a54d36SAndroid Build Coastguard Worker // 2. We have updated the TXT record that was added to the service using DNSServiceAddRecord
1165*48a54d36SAndroid Build Coastguard Worker // 3. We have updated the TXT record that was registered using DNSServiceRegisterRecord
1166*48a54d36SAndroid Build Coastguard Worker //
1167*48a54d36SAndroid Build Coastguard Worker // external_advertise is set if we have advertised at least once during the initial addition
1168*48a54d36SAndroid Build Coastguard Worker // of the record in all of the three cases above. We should have checked for InterfaceID/LocalDomain
1169*48a54d36SAndroid Build Coastguard Worker // checks during the first time and hence we don't do any checks here
1170*48a54d36SAndroid Build Coastguard Worker if (external_advertise)
1171*48a54d36SAndroid Build Coastguard Worker {
1172*48a54d36SAndroid Build Coastguard Worker ResourceRecord ext = rr->resrec;
1173*48a54d36SAndroid Build Coastguard Worker if (ext.rdlength == oldrdlen && mDNSPlatformMemSame(&ext.rdata->u, &oldrd->u, oldrdlen)) goto exit;
1174*48a54d36SAndroid Build Coastguard Worker SetNewRData(&ext, oldrd, oldrdlen);
1175*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&ext);
1176*48a54d36SAndroid Build Coastguard Worker LogInfo("update_callback: calling external_start_advertising_service");
1177*48a54d36SAndroid Build Coastguard Worker external_start_advertising_service(&rr->resrec);
1178*48a54d36SAndroid Build Coastguard Worker }
1179*48a54d36SAndroid Build Coastguard Worker exit:
1180*48a54d36SAndroid Build Coastguard Worker if (oldrd != &rr->rdatastorage) freeL("RData/update_callback", oldrd);
1181*48a54d36SAndroid Build Coastguard Worker }
1182*48a54d36SAndroid Build Coastguard Worker
update_record(AuthRecord * rr,mDNSu16 rdlen,const char * rdata,mDNSu32 ttl,const mDNSBool * const external_advertise)1183*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus update_record(AuthRecord *rr, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl, const mDNSBool *const external_advertise)
1184*48a54d36SAndroid Build Coastguard Worker {
1185*48a54d36SAndroid Build Coastguard Worker mStatus result;
1186*48a54d36SAndroid Build Coastguard Worker const int rdsize = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody);
1187*48a54d36SAndroid Build Coastguard Worker RData *newrd = mallocL("RData/update_record", sizeof(RData) - sizeof(RDataBody) + rdsize);
1188*48a54d36SAndroid Build Coastguard Worker if (!newrd) FatalError("ERROR: malloc");
1189*48a54d36SAndroid Build Coastguard Worker newrd->MaxRDLength = (mDNSu16) rdsize;
1190*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(&newrd->u, rdata, rdlen);
1191*48a54d36SAndroid Build Coastguard Worker
1192*48a54d36SAndroid Build Coastguard Worker // BIND named (name daemon) doesn't allow TXT records with zero-length rdata. This is strictly speaking correct,
1193*48a54d36SAndroid Build Coastguard Worker // since RFC 1035 specifies a TXT record as "One or more <character-string>s", not "Zero or more <character-string>s".
1194*48a54d36SAndroid Build Coastguard Worker // Since some legacy apps try to create zero-length TXT records, we'll silently correct it here.
1195*48a54d36SAndroid Build Coastguard Worker if (rr->resrec.rrtype == kDNSType_TXT && rdlen == 0) { rdlen = 1; newrd->u.txt.c[0] = 0; }
1196*48a54d36SAndroid Build Coastguard Worker
1197*48a54d36SAndroid Build Coastguard Worker if (external_advertise) rr->UpdateContext = (void *)external_advertise;
1198*48a54d36SAndroid Build Coastguard Worker
1199*48a54d36SAndroid Build Coastguard Worker result = mDNS_Update(&mDNSStorage, rr, ttl, rdlen, newrd, update_callback);
1200*48a54d36SAndroid Build Coastguard Worker if (result) { LogMsg("update_record: Error %d for %s", (int)result, ARDisplayString(&mDNSStorage, rr)); freeL("RData/update_record", newrd); }
1201*48a54d36SAndroid Build Coastguard Worker return result;
1202*48a54d36SAndroid Build Coastguard Worker }
1203*48a54d36SAndroid Build Coastguard Worker
handle_update_request(request_state * request)1204*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_update_request(request_state *request)
1205*48a54d36SAndroid Build Coastguard Worker {
1206*48a54d36SAndroid Build Coastguard Worker const ipc_msg_hdr *const hdr = &request->hdr;
1207*48a54d36SAndroid Build Coastguard Worker mStatus result = mStatus_BadReferenceErr;
1208*48a54d36SAndroid Build Coastguard Worker service_instance *i;
1209*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr = NULL;
1210*48a54d36SAndroid Build Coastguard Worker
1211*48a54d36SAndroid Build Coastguard Worker // get the message data
1212*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend); // flags unused
1213*48a54d36SAndroid Build Coastguard Worker mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend);
1214*48a54d36SAndroid Build Coastguard Worker const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen);
1215*48a54d36SAndroid Build Coastguard Worker mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend);
1216*48a54d36SAndroid Build Coastguard Worker (void)flags; // Unused
1217*48a54d36SAndroid Build Coastguard Worker
1218*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceUpdateRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1219*48a54d36SAndroid Build Coastguard Worker
1220*48a54d36SAndroid Build Coastguard Worker // If this is a shared connection, check if the operation actually applies to a subordinate request_state object
1221*48a54d36SAndroid Build Coastguard Worker if (request->terminate == connection_termination) request = LocateSubordinateRequest(request);
1222*48a54d36SAndroid Build Coastguard Worker
1223*48a54d36SAndroid Build Coastguard Worker if (request->terminate == connection_termination)
1224*48a54d36SAndroid Build Coastguard Worker {
1225*48a54d36SAndroid Build Coastguard Worker // update an individually registered record
1226*48a54d36SAndroid Build Coastguard Worker registered_record_entry *reptr;
1227*48a54d36SAndroid Build Coastguard Worker for (reptr = request->u.reg_recs; reptr; reptr = reptr->next)
1228*48a54d36SAndroid Build Coastguard Worker {
1229*48a54d36SAndroid Build Coastguard Worker if (reptr->key == hdr->reg_index)
1230*48a54d36SAndroid Build Coastguard Worker {
1231*48a54d36SAndroid Build Coastguard Worker result = update_record(reptr->rr, rdlen, rdata, ttl, &reptr->external_advertise);
1232*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s)",
1233*48a54d36SAndroid Build Coastguard Worker request->sd, reptr->rr->resrec.name->c, reptr->rr ? DNSTypeName(reptr->rr->resrec.rrtype) : "<NONE>");
1234*48a54d36SAndroid Build Coastguard Worker goto end;
1235*48a54d36SAndroid Build Coastguard Worker }
1236*48a54d36SAndroid Build Coastguard Worker }
1237*48a54d36SAndroid Build Coastguard Worker result = mStatus_BadReferenceErr;
1238*48a54d36SAndroid Build Coastguard Worker goto end;
1239*48a54d36SAndroid Build Coastguard Worker }
1240*48a54d36SAndroid Build Coastguard Worker
1241*48a54d36SAndroid Build Coastguard Worker if (request->terminate != regservice_termination_callback)
1242*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceUpdateRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); }
1243*48a54d36SAndroid Build Coastguard Worker
1244*48a54d36SAndroid Build Coastguard Worker // For a service registered with zero port, only SRV record is initialized. Don't allow any updates.
1245*48a54d36SAndroid Build Coastguard Worker if (mDNSIPPortIsZero(request->u.servicereg.port))
1246*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceUpdateRecord: updating the record of a service registered with zero port", request->sd); return(mStatus_BadParamErr); }
1247*48a54d36SAndroid Build Coastguard Worker
1248*48a54d36SAndroid Build Coastguard Worker // update the saved off TXT data for the service
1249*48a54d36SAndroid Build Coastguard Worker if (hdr->reg_index == TXT_RECORD_INDEX)
1250*48a54d36SAndroid Build Coastguard Worker {
1251*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.txtdata)
1252*48a54d36SAndroid Build Coastguard Worker { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; }
1253*48a54d36SAndroid Build Coastguard Worker if (rdlen > 0)
1254*48a54d36SAndroid Build Coastguard Worker {
1255*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtdata = mallocL("service_info txtdata", rdlen);
1256*48a54d36SAndroid Build Coastguard Worker if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_update_request - malloc");
1257*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(request->u.servicereg.txtdata, rdata, rdlen);
1258*48a54d36SAndroid Build Coastguard Worker }
1259*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtlen = rdlen;
1260*48a54d36SAndroid Build Coastguard Worker }
1261*48a54d36SAndroid Build Coastguard Worker
1262*48a54d36SAndroid Build Coastguard Worker // update a record from a service record set
1263*48a54d36SAndroid Build Coastguard Worker for (i = request->u.servicereg.instances; i; i = i->next)
1264*48a54d36SAndroid Build Coastguard Worker {
1265*48a54d36SAndroid Build Coastguard Worker if (hdr->reg_index == TXT_RECORD_INDEX) rr = &i->srs.RR_TXT;
1266*48a54d36SAndroid Build Coastguard Worker else
1267*48a54d36SAndroid Build Coastguard Worker {
1268*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *e;
1269*48a54d36SAndroid Build Coastguard Worker for (e = i->srs.Extras; e; e = e->next)
1270*48a54d36SAndroid Build Coastguard Worker if (e->ClientID == hdr->reg_index) { rr = &e->r; break; }
1271*48a54d36SAndroid Build Coastguard Worker }
1272*48a54d36SAndroid Build Coastguard Worker
1273*48a54d36SAndroid Build Coastguard Worker if (!rr) { result = mStatus_BadReferenceErr; goto end; }
1274*48a54d36SAndroid Build Coastguard Worker result = update_record(rr, rdlen, rdata, ttl, &i->external_advertise);
1275*48a54d36SAndroid Build Coastguard Worker if (result && i->default_local) goto end;
1276*48a54d36SAndroid Build Coastguard Worker else result = mStatus_NoError; // suppress non-local default errors
1277*48a54d36SAndroid Build Coastguard Worker }
1278*48a54d36SAndroid Build Coastguard Worker
1279*48a54d36SAndroid Build Coastguard Worker end:
1280*48a54d36SAndroid Build Coastguard Worker if (request->terminate == regservice_termination_callback)
1281*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s)", request->sd,
1282*48a54d36SAndroid Build Coastguard Worker (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL,
1283*48a54d36SAndroid Build Coastguard Worker rr ? DNSTypeName(rr->resrec.rrtype) : "<NONE>");
1284*48a54d36SAndroid Build Coastguard Worker
1285*48a54d36SAndroid Build Coastguard Worker return(result);
1286*48a54d36SAndroid Build Coastguard Worker }
1287*48a54d36SAndroid Build Coastguard Worker
1288*48a54d36SAndroid Build Coastguard Worker // remove a resource record registered via DNSServiceRegisterRecord()
remove_record(request_state * request)1289*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus remove_record(request_state *request)
1290*48a54d36SAndroid Build Coastguard Worker {
1291*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_UnknownErr;
1292*48a54d36SAndroid Build Coastguard Worker registered_record_entry *e, **ptr = &request->u.reg_recs;
1293*48a54d36SAndroid Build Coastguard Worker
1294*48a54d36SAndroid Build Coastguard Worker while (*ptr && (*ptr)->key != request->hdr.reg_index) ptr = &(*ptr)->next;
1295*48a54d36SAndroid Build Coastguard Worker if (!*ptr) { LogMsg("%3d: DNSServiceRemoveRecord(%u) not found", request->sd, request->hdr.reg_index); return mStatus_BadReferenceErr; }
1296*48a54d36SAndroid Build Coastguard Worker e = *ptr;
1297*48a54d36SAndroid Build Coastguard Worker *ptr = e->next; // unlink
1298*48a54d36SAndroid Build Coastguard Worker
1299*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRemoveRecord(%u %s)", request->sd, e->key, RRDisplayString(&mDNSStorage, &e->rr->resrec));
1300*48a54d36SAndroid Build Coastguard Worker e->rr->RecordContext = NULL;
1301*48a54d36SAndroid Build Coastguard Worker if (e->external_advertise)
1302*48a54d36SAndroid Build Coastguard Worker {
1303*48a54d36SAndroid Build Coastguard Worker external_stop_advertising_service(&e->rr->resrec);
1304*48a54d36SAndroid Build Coastguard Worker e->external_advertise = mDNSfalse;
1305*48a54d36SAndroid Build Coastguard Worker }
1306*48a54d36SAndroid Build Coastguard Worker err = mDNS_Deregister(&mDNSStorage, e->rr); // Will free e->rr for us; we're responsible for freeing e
1307*48a54d36SAndroid Build Coastguard Worker if (err)
1308*48a54d36SAndroid Build Coastguard Worker {
1309*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: remove_record, mDNS_Deregister: %d", err);
1310*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry AuthRecord remove_record", e->rr);
1311*48a54d36SAndroid Build Coastguard Worker }
1312*48a54d36SAndroid Build Coastguard Worker
1313*48a54d36SAndroid Build Coastguard Worker freeL("registered_record_entry remove_record", e);
1314*48a54d36SAndroid Build Coastguard Worker return err;
1315*48a54d36SAndroid Build Coastguard Worker }
1316*48a54d36SAndroid Build Coastguard Worker
remove_extra(const request_state * const request,service_instance * const serv,mDNSu16 * const rrtype)1317*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus remove_extra(const request_state *const request, service_instance *const serv, mDNSu16 *const rrtype)
1318*48a54d36SAndroid Build Coastguard Worker {
1319*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_BadReferenceErr;
1320*48a54d36SAndroid Build Coastguard Worker ExtraResourceRecord *ptr;
1321*48a54d36SAndroid Build Coastguard Worker
1322*48a54d36SAndroid Build Coastguard Worker for (ptr = serv->srs.Extras; ptr; ptr = ptr->next)
1323*48a54d36SAndroid Build Coastguard Worker {
1324*48a54d36SAndroid Build Coastguard Worker if (ptr->ClientID == request->hdr.reg_index) // found match
1325*48a54d36SAndroid Build Coastguard Worker {
1326*48a54d36SAndroid Build Coastguard Worker *rrtype = ptr->r.resrec.rrtype;
1327*48a54d36SAndroid Build Coastguard Worker if (serv->external_advertise) external_stop_advertising_service(&ptr->r.resrec);
1328*48a54d36SAndroid Build Coastguard Worker err = mDNS_RemoveRecordFromService(&mDNSStorage, &serv->srs, ptr, FreeExtraRR, ptr);
1329*48a54d36SAndroid Build Coastguard Worker break;
1330*48a54d36SAndroid Build Coastguard Worker }
1331*48a54d36SAndroid Build Coastguard Worker }
1332*48a54d36SAndroid Build Coastguard Worker return err;
1333*48a54d36SAndroid Build Coastguard Worker }
1334*48a54d36SAndroid Build Coastguard Worker
handle_removerecord_request(request_state * request)1335*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_removerecord_request(request_state *request)
1336*48a54d36SAndroid Build Coastguard Worker {
1337*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_BadReferenceErr;
1338*48a54d36SAndroid Build Coastguard Worker get_flags(&request->msgptr, request->msgend); // flags unused
1339*48a54d36SAndroid Build Coastguard Worker
1340*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceRemoveRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
1341*48a54d36SAndroid Build Coastguard Worker
1342*48a54d36SAndroid Build Coastguard Worker // If this is a shared connection, check if the operation actually applies to a subordinate request_state object
1343*48a54d36SAndroid Build Coastguard Worker if (request->terminate == connection_termination) request = LocateSubordinateRequest(request);
1344*48a54d36SAndroid Build Coastguard Worker
1345*48a54d36SAndroid Build Coastguard Worker if (request->terminate == connection_termination)
1346*48a54d36SAndroid Build Coastguard Worker err = remove_record(request); // remove individually registered record
1347*48a54d36SAndroid Build Coastguard Worker else if (request->terminate != regservice_termination_callback)
1348*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceRemoveRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); }
1349*48a54d36SAndroid Build Coastguard Worker else
1350*48a54d36SAndroid Build Coastguard Worker {
1351*48a54d36SAndroid Build Coastguard Worker service_instance *i;
1352*48a54d36SAndroid Build Coastguard Worker mDNSu16 rrtype = 0;
1353*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRemoveRecord(%##s, %s)", request->sd,
1354*48a54d36SAndroid Build Coastguard Worker (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL,
1355*48a54d36SAndroid Build Coastguard Worker rrtype ? DNSTypeName(rrtype) : "<NONE>");
1356*48a54d36SAndroid Build Coastguard Worker for (i = request->u.servicereg.instances; i; i = i->next)
1357*48a54d36SAndroid Build Coastguard Worker {
1358*48a54d36SAndroid Build Coastguard Worker err = remove_extra(request, i, &rrtype);
1359*48a54d36SAndroid Build Coastguard Worker if (err && i->default_local) break;
1360*48a54d36SAndroid Build Coastguard Worker else err = mStatus_NoError; // suppress non-local default errors
1361*48a54d36SAndroid Build Coastguard Worker }
1362*48a54d36SAndroid Build Coastguard Worker }
1363*48a54d36SAndroid Build Coastguard Worker
1364*48a54d36SAndroid Build Coastguard Worker return(err);
1365*48a54d36SAndroid Build Coastguard Worker }
1366*48a54d36SAndroid Build Coastguard Worker
1367*48a54d36SAndroid Build Coastguard Worker // If there's a comma followed by another character,
1368*48a54d36SAndroid Build Coastguard Worker // FindFirstSubType overwrites the comma with a nul and returns the pointer to the next character.
1369*48a54d36SAndroid Build Coastguard Worker // Otherwise, it returns a pointer to the final nul at the end of the string
FindFirstSubType(char * p)1370*48a54d36SAndroid Build Coastguard Worker mDNSlocal char *FindFirstSubType(char *p)
1371*48a54d36SAndroid Build Coastguard Worker {
1372*48a54d36SAndroid Build Coastguard Worker while (*p)
1373*48a54d36SAndroid Build Coastguard Worker {
1374*48a54d36SAndroid Build Coastguard Worker if (p[0] == '\\' && p[1]) p += 2;
1375*48a54d36SAndroid Build Coastguard Worker else if (p[0] == ',' && p[1]) { *p++ = 0; return(p); }
1376*48a54d36SAndroid Build Coastguard Worker else p++;
1377*48a54d36SAndroid Build Coastguard Worker }
1378*48a54d36SAndroid Build Coastguard Worker return(p);
1379*48a54d36SAndroid Build Coastguard Worker }
1380*48a54d36SAndroid Build Coastguard Worker
1381*48a54d36SAndroid Build Coastguard Worker // If there's a comma followed by another character,
1382*48a54d36SAndroid Build Coastguard Worker // FindNextSubType overwrites the comma with a nul and returns the pointer to the next character.
1383*48a54d36SAndroid Build Coastguard Worker // If it finds an illegal unescaped dot in the subtype name, it returns mDNSNULL
1384*48a54d36SAndroid Build Coastguard Worker // Otherwise, it returns a pointer to the final nul at the end of the string
FindNextSubType(char * p)1385*48a54d36SAndroid Build Coastguard Worker mDNSlocal char *FindNextSubType(char *p)
1386*48a54d36SAndroid Build Coastguard Worker {
1387*48a54d36SAndroid Build Coastguard Worker while (*p)
1388*48a54d36SAndroid Build Coastguard Worker {
1389*48a54d36SAndroid Build Coastguard Worker if (p[0] == '\\' && p[1]) // If escape character
1390*48a54d36SAndroid Build Coastguard Worker p += 2; // ignore following character
1391*48a54d36SAndroid Build Coastguard Worker else if (p[0] == ',') // If we found a comma
1392*48a54d36SAndroid Build Coastguard Worker {
1393*48a54d36SAndroid Build Coastguard Worker if (p[1]) *p++ = 0;
1394*48a54d36SAndroid Build Coastguard Worker return(p);
1395*48a54d36SAndroid Build Coastguard Worker }
1396*48a54d36SAndroid Build Coastguard Worker else if (p[0] == '.')
1397*48a54d36SAndroid Build Coastguard Worker return(mDNSNULL);
1398*48a54d36SAndroid Build Coastguard Worker else p++;
1399*48a54d36SAndroid Build Coastguard Worker }
1400*48a54d36SAndroid Build Coastguard Worker return(p);
1401*48a54d36SAndroid Build Coastguard Worker }
1402*48a54d36SAndroid Build Coastguard Worker
1403*48a54d36SAndroid Build Coastguard Worker // Returns -1 if illegal subtype found
ChopSubTypes(char * regtype)1404*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSs32 ChopSubTypes(char *regtype)
1405*48a54d36SAndroid Build Coastguard Worker {
1406*48a54d36SAndroid Build Coastguard Worker mDNSs32 NumSubTypes = 0;
1407*48a54d36SAndroid Build Coastguard Worker char *stp = FindFirstSubType(regtype);
1408*48a54d36SAndroid Build Coastguard Worker while (stp && *stp) // If we found a comma...
1409*48a54d36SAndroid Build Coastguard Worker {
1410*48a54d36SAndroid Build Coastguard Worker if (*stp == ',') return(-1);
1411*48a54d36SAndroid Build Coastguard Worker NumSubTypes++;
1412*48a54d36SAndroid Build Coastguard Worker stp = FindNextSubType(stp);
1413*48a54d36SAndroid Build Coastguard Worker }
1414*48a54d36SAndroid Build Coastguard Worker if (!stp) return(-1);
1415*48a54d36SAndroid Build Coastguard Worker return(NumSubTypes);
1416*48a54d36SAndroid Build Coastguard Worker }
1417*48a54d36SAndroid Build Coastguard Worker
AllocateSubTypes(mDNSs32 NumSubTypes,char * p)1418*48a54d36SAndroid Build Coastguard Worker mDNSexport AuthRecord *AllocateSubTypes(mDNSs32 NumSubTypes, char *p)
1419*48a54d36SAndroid Build Coastguard Worker {
1420*48a54d36SAndroid Build Coastguard Worker AuthRecord *st = mDNSNULL;
1421*48a54d36SAndroid Build Coastguard Worker if (NumSubTypes)
1422*48a54d36SAndroid Build Coastguard Worker {
1423*48a54d36SAndroid Build Coastguard Worker mDNSs32 i;
1424*48a54d36SAndroid Build Coastguard Worker st = mallocL("ServiceSubTypes", NumSubTypes * sizeof(AuthRecord));
1425*48a54d36SAndroid Build Coastguard Worker if (!st) return(mDNSNULL);
1426*48a54d36SAndroid Build Coastguard Worker for (i = 0; i < NumSubTypes; i++)
1427*48a54d36SAndroid Build Coastguard Worker {
1428*48a54d36SAndroid Build Coastguard Worker mDNS_SetupResourceRecord(&st[i], mDNSNULL, mDNSInterface_Any, kDNSQType_ANY, kStandardTTL, 0, AuthRecordAny, mDNSNULL, mDNSNULL);
1429*48a54d36SAndroid Build Coastguard Worker while (*p) p++;
1430*48a54d36SAndroid Build Coastguard Worker p++;
1431*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&st[i].namestorage, p))
1432*48a54d36SAndroid Build Coastguard Worker { freeL("ServiceSubTypes", st); return(mDNSNULL); }
1433*48a54d36SAndroid Build Coastguard Worker }
1434*48a54d36SAndroid Build Coastguard Worker }
1435*48a54d36SAndroid Build Coastguard Worker return(st);
1436*48a54d36SAndroid Build Coastguard Worker }
1437*48a54d36SAndroid Build Coastguard Worker
register_service_instance(request_state * request,const domainname * domain)1438*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus register_service_instance(request_state *request, const domainname *domain)
1439*48a54d36SAndroid Build Coastguard Worker {
1440*48a54d36SAndroid Build Coastguard Worker service_instance **ptr, *instance;
1441*48a54d36SAndroid Build Coastguard Worker const int extra_size = (request->u.servicereg.txtlen > sizeof(RDataBody)) ? (request->u.servicereg.txtlen - sizeof(RDataBody)) : 0;
1442*48a54d36SAndroid Build Coastguard Worker const mDNSBool DomainIsLocal = SameDomainName(domain, &localdomain);
1443*48a54d36SAndroid Build Coastguard Worker mStatus result;
1444*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID interfaceID = request->u.servicereg.InterfaceID;
1445*48a54d36SAndroid Build Coastguard Worker mDNSu32 regFlags = 0;
1446*48a54d36SAndroid Build Coastguard Worker
1447*48a54d36SAndroid Build Coastguard Worker if (interfaceID == mDNSInterface_P2P)
1448*48a54d36SAndroid Build Coastguard Worker {
1449*48a54d36SAndroid Build Coastguard Worker interfaceID = mDNSInterface_Any;
1450*48a54d36SAndroid Build Coastguard Worker regFlags |= regFlagIncludeP2P;
1451*48a54d36SAndroid Build Coastguard Worker }
1452*48a54d36SAndroid Build Coastguard Worker else if (request->flags & kDNSServiceFlagsIncludeP2P)
1453*48a54d36SAndroid Build Coastguard Worker regFlags |= regFlagIncludeP2P;
1454*48a54d36SAndroid Build Coastguard Worker
1455*48a54d36SAndroid Build Coastguard Worker // client guarantees that record names are unique
1456*48a54d36SAndroid Build Coastguard Worker if (request->flags & kDNSServiceFlagsForce)
1457*48a54d36SAndroid Build Coastguard Worker regFlags |= regFlagKnownUnique;
1458*48a54d36SAndroid Build Coastguard Worker
1459*48a54d36SAndroid Build Coastguard Worker // If the client specified an interface, but no domain, then we honor the specified interface for the "local" (mDNS)
1460*48a54d36SAndroid Build Coastguard Worker // registration but for the wide-area registrations we don't (currently) have any concept of a wide-area unicast
1461*48a54d36SAndroid Build Coastguard Worker // registrations scoped to a specific interface, so for the automatic domains we add we must *not* specify an interface.
1462*48a54d36SAndroid Build Coastguard Worker // (Specifying an interface with an apparently wide-area domain (i.e. something other than "local")
1463*48a54d36SAndroid Build Coastguard Worker // currently forces the registration to use mDNS multicast despite the apparently wide-area domain.)
1464*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.default_domain && !DomainIsLocal) interfaceID = mDNSInterface_Any;
1465*48a54d36SAndroid Build Coastguard Worker
1466*48a54d36SAndroid Build Coastguard Worker for (ptr = &request->u.servicereg.instances; *ptr; ptr = &(*ptr)->next)
1467*48a54d36SAndroid Build Coastguard Worker {
1468*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&(*ptr)->domain, domain))
1469*48a54d36SAndroid Build Coastguard Worker {
1470*48a54d36SAndroid Build Coastguard Worker LogMsg("register_service_instance: domain %##s already registered for %#s.%##s",
1471*48a54d36SAndroid Build Coastguard Worker domain->c, &request->u.servicereg.name, &request->u.servicereg.type);
1472*48a54d36SAndroid Build Coastguard Worker return mStatus_AlreadyRegistered;
1473*48a54d36SAndroid Build Coastguard Worker }
1474*48a54d36SAndroid Build Coastguard Worker }
1475*48a54d36SAndroid Build Coastguard Worker
1476*48a54d36SAndroid Build Coastguard Worker if (mDNSStorage.KnownBugs & mDNS_KnownBug_LimitedIPv6)
1477*48a54d36SAndroid Build Coastguard Worker {
1478*48a54d36SAndroid Build Coastguard Worker // Special-case hack: On Mac OS X 10.6.x and earlier we don't advertise SMB service in AutoTunnel domains,
1479*48a54d36SAndroid Build Coastguard Worker // because AutoTunnel services have to support IPv6, and in Mac OS X 10.6.x the SMB server does not.
1480*48a54d36SAndroid Build Coastguard Worker // <rdar://problem/5482322> BTMM: Don't advertise SMB with BTMM because it doesn't support IPv6
1481*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&request->u.servicereg.type, (const domainname *) "\x4" "_smb" "\x4" "_tcp"))
1482*48a54d36SAndroid Build Coastguard Worker {
1483*48a54d36SAndroid Build Coastguard Worker DomainAuthInfo *AuthInfo = GetAuthInfoForName(&mDNSStorage, domain);
1484*48a54d36SAndroid Build Coastguard Worker if (AuthInfo && AuthInfo->AutoTunnel) return(kDNSServiceErr_Unsupported);
1485*48a54d36SAndroid Build Coastguard Worker }
1486*48a54d36SAndroid Build Coastguard Worker }
1487*48a54d36SAndroid Build Coastguard Worker
1488*48a54d36SAndroid Build Coastguard Worker instance = mallocL("service_instance", sizeof(*instance) + extra_size);
1489*48a54d36SAndroid Build Coastguard Worker if (!instance) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; }
1490*48a54d36SAndroid Build Coastguard Worker
1491*48a54d36SAndroid Build Coastguard Worker instance->next = mDNSNULL;
1492*48a54d36SAndroid Build Coastguard Worker instance->request = request;
1493*48a54d36SAndroid Build Coastguard Worker instance->subtypes = AllocateSubTypes(request->u.servicereg.num_subtypes, request->u.servicereg.type_as_string);
1494*48a54d36SAndroid Build Coastguard Worker instance->renameonmemfree = 0;
1495*48a54d36SAndroid Build Coastguard Worker instance->clientnotified = mDNSfalse;
1496*48a54d36SAndroid Build Coastguard Worker instance->default_local = (request->u.servicereg.default_domain && DomainIsLocal);
1497*48a54d36SAndroid Build Coastguard Worker instance->external_advertise = mDNSfalse;
1498*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&instance->domain, domain);
1499*48a54d36SAndroid Build Coastguard Worker
1500*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.num_subtypes && !instance->subtypes)
1501*48a54d36SAndroid Build Coastguard Worker { unlink_and_free_service_instance(instance); instance = NULL; FatalError("ERROR: malloc"); }
1502*48a54d36SAndroid Build Coastguard Worker
1503*48a54d36SAndroid Build Coastguard Worker result = mDNS_RegisterService(&mDNSStorage, &instance->srs,
1504*48a54d36SAndroid Build Coastguard Worker &request->u.servicereg.name, &request->u.servicereg.type, domain,
1505*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.host.c[0] ? &request->u.servicereg.host : NULL,
1506*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.port,
1507*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtdata, request->u.servicereg.txtlen,
1508*48a54d36SAndroid Build Coastguard Worker instance->subtypes, request->u.servicereg.num_subtypes,
1509*48a54d36SAndroid Build Coastguard Worker interfaceID, regservice_callback, instance, regFlags);
1510*48a54d36SAndroid Build Coastguard Worker
1511*48a54d36SAndroid Build Coastguard Worker if (!result)
1512*48a54d36SAndroid Build Coastguard Worker {
1513*48a54d36SAndroid Build Coastguard Worker *ptr = instance; // Append this to the end of our request->u.servicereg.instances list
1514*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegister(%##s, %u) ADDED",
1515*48a54d36SAndroid Build Coastguard Worker instance->request->sd, instance->srs.RR_SRV.resrec.name->c, mDNSVal16(request->u.servicereg.port));
1516*48a54d36SAndroid Build Coastguard Worker }
1517*48a54d36SAndroid Build Coastguard Worker else
1518*48a54d36SAndroid Build Coastguard Worker {
1519*48a54d36SAndroid Build Coastguard Worker LogMsg("register_service_instance %#s.%##s%##s error %d",
1520*48a54d36SAndroid Build Coastguard Worker &request->u.servicereg.name, &request->u.servicereg.type, domain->c, result);
1521*48a54d36SAndroid Build Coastguard Worker unlink_and_free_service_instance(instance);
1522*48a54d36SAndroid Build Coastguard Worker }
1523*48a54d36SAndroid Build Coastguard Worker
1524*48a54d36SAndroid Build Coastguard Worker return result;
1525*48a54d36SAndroid Build Coastguard Worker }
1526*48a54d36SAndroid Build Coastguard Worker
udsserver_default_reg_domain_changed(const DNameListElem * const d,const mDNSBool add)1527*48a54d36SAndroid Build Coastguard Worker mDNSlocal void udsserver_default_reg_domain_changed(const DNameListElem *const d, const mDNSBool add)
1528*48a54d36SAndroid Build Coastguard Worker {
1529*48a54d36SAndroid Build Coastguard Worker request_state *request;
1530*48a54d36SAndroid Build Coastguard Worker
1531*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
1532*48a54d36SAndroid Build Coastguard Worker machserver_automatic_registration_domain_changed(&d->name, add);
1533*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
1534*48a54d36SAndroid Build Coastguard Worker
1535*48a54d36SAndroid Build Coastguard Worker LogMsg("%s registration domain %##s", add ? "Adding" : "Removing", d->name.c);
1536*48a54d36SAndroid Build Coastguard Worker for (request = all_requests; request; request = request->next)
1537*48a54d36SAndroid Build Coastguard Worker {
1538*48a54d36SAndroid Build Coastguard Worker if (request->terminate != regservice_termination_callback) continue;
1539*48a54d36SAndroid Build Coastguard Worker if (!request->u.servicereg.default_domain) continue;
1540*48a54d36SAndroid Build Coastguard Worker if (!d->uid || SystemUID(request->uid) || request->uid == d->uid)
1541*48a54d36SAndroid Build Coastguard Worker {
1542*48a54d36SAndroid Build Coastguard Worker service_instance **ptr = &request->u.servicereg.instances;
1543*48a54d36SAndroid Build Coastguard Worker while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next;
1544*48a54d36SAndroid Build Coastguard Worker if (add)
1545*48a54d36SAndroid Build Coastguard Worker {
1546*48a54d36SAndroid Build Coastguard Worker // If we don't already have this domain in our list for this registration, add it now
1547*48a54d36SAndroid Build Coastguard Worker if (!*ptr) register_service_instance(request, &d->name);
1548*48a54d36SAndroid Build Coastguard Worker else debugf("udsserver_default_reg_domain_changed %##s already in list, not re-adding", &d->name);
1549*48a54d36SAndroid Build Coastguard Worker }
1550*48a54d36SAndroid Build Coastguard Worker else
1551*48a54d36SAndroid Build Coastguard Worker {
1552*48a54d36SAndroid Build Coastguard Worker // Normally we should not fail to find the specified instance
1553*48a54d36SAndroid Build Coastguard Worker // One case where this can happen is if a uDNS update fails for some reason,
1554*48a54d36SAndroid Build Coastguard Worker // and regservice_callback then calls unlink_and_free_service_instance and disposes of that instance.
1555*48a54d36SAndroid Build Coastguard Worker if (!*ptr)
1556*48a54d36SAndroid Build Coastguard Worker LogMsg("udsserver_default_reg_domain_changed domain %##s not found for service %#s type %s",
1557*48a54d36SAndroid Build Coastguard Worker &d->name, request->u.servicereg.name.c, request->u.servicereg.type_as_string);
1558*48a54d36SAndroid Build Coastguard Worker else
1559*48a54d36SAndroid Build Coastguard Worker {
1560*48a54d36SAndroid Build Coastguard Worker DNameListElem *p;
1561*48a54d36SAndroid Build Coastguard Worker for (p = AutoRegistrationDomains; p; p=p->next)
1562*48a54d36SAndroid Build Coastguard Worker if (!p->uid || SystemUID(request->uid) || request->uid == p->uid)
1563*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&d->name, &p->name)) break;
1564*48a54d36SAndroid Build Coastguard Worker if (p) debugf("udsserver_default_reg_domain_changed %##s still in list, not removing", &d->name);
1565*48a54d36SAndroid Build Coastguard Worker else
1566*48a54d36SAndroid Build Coastguard Worker {
1567*48a54d36SAndroid Build Coastguard Worker mStatus err;
1568*48a54d36SAndroid Build Coastguard Worker service_instance *si = *ptr;
1569*48a54d36SAndroid Build Coastguard Worker *ptr = si->next;
1570*48a54d36SAndroid Build Coastguard Worker if (si->clientnotified) SendServiceRemovalNotification(&si->srs); // Do this *before* clearing si->request backpointer
1571*48a54d36SAndroid Build Coastguard Worker // Now that we've cut this service_instance from the list, we MUST clear the si->request backpointer.
1572*48a54d36SAndroid Build Coastguard Worker // Otherwise what can happen is this: While our mDNS_DeregisterService is in the
1573*48a54d36SAndroid Build Coastguard Worker // process of completing asynchronously, the client cancels the entire operation, so
1574*48a54d36SAndroid Build Coastguard Worker // regservice_termination_callback then runs through the whole list deregistering each
1575*48a54d36SAndroid Build Coastguard Worker // instance, clearing the backpointers, and then disposing the parent request_state object.
1576*48a54d36SAndroid Build Coastguard Worker // However, because this service_instance isn't in the list any more, regservice_termination_callback
1577*48a54d36SAndroid Build Coastguard Worker // has no way to find it and clear its backpointer, and then when our mDNS_DeregisterService finally
1578*48a54d36SAndroid Build Coastguard Worker // completes later with a mStatus_MemFree message, it calls unlink_and_free_service_instance() with
1579*48a54d36SAndroid Build Coastguard Worker // a service_instance with a stale si->request backpointer pointing to memory that's already been freed.
1580*48a54d36SAndroid Build Coastguard Worker si->request = NULL;
1581*48a54d36SAndroid Build Coastguard Worker err = mDNS_DeregisterService(&mDNSStorage, &si->srs);
1582*48a54d36SAndroid Build Coastguard Worker if (err) { LogMsg("udsserver_default_reg_domain_changed err %d", err); unlink_and_free_service_instance(si); }
1583*48a54d36SAndroid Build Coastguard Worker }
1584*48a54d36SAndroid Build Coastguard Worker }
1585*48a54d36SAndroid Build Coastguard Worker }
1586*48a54d36SAndroid Build Coastguard Worker }
1587*48a54d36SAndroid Build Coastguard Worker }
1588*48a54d36SAndroid Build Coastguard Worker }
1589*48a54d36SAndroid Build Coastguard Worker
handle_regservice_request(request_state * request)1590*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_regservice_request(request_state *request)
1591*48a54d36SAndroid Build Coastguard Worker {
1592*48a54d36SAndroid Build Coastguard Worker char name[256]; // Lots of spare space for extra-long names that we'll auto-truncate down to 63 bytes
1593*48a54d36SAndroid Build Coastguard Worker char domain[MAX_ESCAPED_DOMAIN_NAME], host[MAX_ESCAPED_DOMAIN_NAME];
1594*48a54d36SAndroid Build Coastguard Worker char type_as_string[MAX_ESCAPED_DOMAIN_NAME];
1595*48a54d36SAndroid Build Coastguard Worker domainname d, srv;
1596*48a54d36SAndroid Build Coastguard Worker mStatus err;
1597*48a54d36SAndroid Build Coastguard Worker const char *msgTXTData;
1598*48a54d36SAndroid Build Coastguard Worker
1599*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
1600*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
1601*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
1602*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID)
1603*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - Couldn't find interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); }
1604*48a54d36SAndroid Build Coastguard Worker
1605*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, name, sizeof(name)) < 0 ||
1606*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, type_as_string, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
1607*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
1608*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, host, MAX_ESCAPED_DOMAIN_NAME) < 0)
1609*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); }
1610*48a54d36SAndroid Build Coastguard Worker
1611*48a54d36SAndroid Build Coastguard Worker request->flags = flags;
1612*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.InterfaceID = InterfaceID;
1613*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.instances = NULL;
1614*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtlen = 0;
1615*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtdata = NULL;
1616*48a54d36SAndroid Build Coastguard Worker mDNSPlatformStrLCopy(request->u.servicereg.type_as_string, type_as_string, sizeof(request->u.servicereg.type_as_string));
1617*48a54d36SAndroid Build Coastguard Worker
1618*48a54d36SAndroid Build Coastguard Worker if (request->msgptr + 2 > request->msgend) request->msgptr = NULL;
1619*48a54d36SAndroid Build Coastguard Worker else
1620*48a54d36SAndroid Build Coastguard Worker {
1621*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.port.b[0] = *request->msgptr++;
1622*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.port.b[1] = *request->msgptr++;
1623*48a54d36SAndroid Build Coastguard Worker }
1624*48a54d36SAndroid Build Coastguard Worker
1625*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtlen = get_uint16(&request->msgptr, request->msgend);
1626*48a54d36SAndroid Build Coastguard Worker msgTXTData = get_rdata(&request->msgptr, request->msgend, request->u.servicereg.txtlen);
1627*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr)
1628*48a54d36SAndroid Build Coastguard Worker {
1629*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: DNSServiceRegister(unreadable parameters)", request->sd);
1630*48a54d36SAndroid Build Coastguard Worker return(mStatus_BadParamErr);
1631*48a54d36SAndroid Build Coastguard Worker }
1632*48a54d36SAndroid Build Coastguard Worker
1633*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.txtlen)
1634*48a54d36SAndroid Build Coastguard Worker {
1635*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.txtdata = mallocL("service_info txtdata", request->u.servicereg.txtlen);
1636*48a54d36SAndroid Build Coastguard Worker if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_regservice_request - malloc");
1637*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(request->u.servicereg.txtdata, msgTXTData, request->u.servicereg.txtlen);
1638*48a54d36SAndroid Build Coastguard Worker }
1639*48a54d36SAndroid Build Coastguard Worker
1640*48a54d36SAndroid Build Coastguard Worker // Check for sub-types after the service type
1641*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.num_subtypes = ChopSubTypes(request->u.servicereg.type_as_string); // Note: Modifies regtype string to remove trailing subtypes
1642*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.num_subtypes < 0)
1643*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - ChopSubTypes failed %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); }
1644*48a54d36SAndroid Build Coastguard Worker
1645*48a54d36SAndroid Build Coastguard Worker // Don't try to construct "domainname t" until *after* ChopSubTypes has worked its magic
1646*48a54d36SAndroid Build Coastguard Worker if (!*request->u.servicereg.type_as_string || !MakeDomainNameFromDNSNameString(&request->u.servicereg.type, request->u.servicereg.type_as_string))
1647*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - type_as_string bad %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); }
1648*48a54d36SAndroid Build Coastguard Worker
1649*48a54d36SAndroid Build Coastguard Worker if (!name[0])
1650*48a54d36SAndroid Build Coastguard Worker {
1651*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.name = mDNSStorage.nicelabel;
1652*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.autoname = mDNStrue;
1653*48a54d36SAndroid Build Coastguard Worker }
1654*48a54d36SAndroid Build Coastguard Worker else
1655*48a54d36SAndroid Build Coastguard Worker {
1656*48a54d36SAndroid Build Coastguard Worker // If the client is allowing AutoRename, then truncate name to legal length before converting it to a DomainLabel
1657*48a54d36SAndroid Build Coastguard Worker if ((flags & kDNSServiceFlagsNoAutoRename) == 0)
1658*48a54d36SAndroid Build Coastguard Worker {
1659*48a54d36SAndroid Build Coastguard Worker int newlen = TruncateUTF8ToLength((mDNSu8*)name, mDNSPlatformStrLen(name), MAX_DOMAIN_LABEL);
1660*48a54d36SAndroid Build Coastguard Worker name[newlen] = 0;
1661*48a54d36SAndroid Build Coastguard Worker }
1662*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainLabelFromLiteralString(&request->u.servicereg.name, name))
1663*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - name bad %s", name); return(mStatus_BadParamErr); }
1664*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.autoname = mDNSfalse;
1665*48a54d36SAndroid Build Coastguard Worker }
1666*48a54d36SAndroid Build Coastguard Worker
1667*48a54d36SAndroid Build Coastguard Worker if (*domain)
1668*48a54d36SAndroid Build Coastguard Worker {
1669*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.default_domain = mDNSfalse;
1670*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&d, domain))
1671*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - domain bad %s", domain); return(mStatus_BadParamErr); }
1672*48a54d36SAndroid Build Coastguard Worker }
1673*48a54d36SAndroid Build Coastguard Worker else
1674*48a54d36SAndroid Build Coastguard Worker {
1675*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.default_domain = mDNStrue;
1676*48a54d36SAndroid Build Coastguard Worker MakeDomainNameFromDNSNameString(&d, "local.");
1677*48a54d36SAndroid Build Coastguard Worker }
1678*48a54d36SAndroid Build Coastguard Worker
1679*48a54d36SAndroid Build Coastguard Worker if (!ConstructServiceName(&srv, &request->u.servicereg.name, &request->u.servicereg.type, &d))
1680*48a54d36SAndroid Build Coastguard Worker {
1681*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: handle_regservice_request - Couldn't ConstructServiceName from, “%#s” “%##s” “%##s”",
1682*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.name.c, request->u.servicereg.type.c, d.c); return(mStatus_BadParamErr);
1683*48a54d36SAndroid Build Coastguard Worker }
1684*48a54d36SAndroid Build Coastguard Worker
1685*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&request->u.servicereg.host, host))
1686*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_regservice_request - host bad %s", host); return(mStatus_BadParamErr); }
1687*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.autorename = (flags & kDNSServiceFlagsNoAutoRename ) == 0;
1688*48a54d36SAndroid Build Coastguard Worker request->u.servicereg.allowremotequery = (flags & kDNSServiceFlagsAllowRemoteQuery) != 0;
1689*48a54d36SAndroid Build Coastguard Worker
1690*48a54d36SAndroid Build Coastguard Worker // Some clients use mDNS for lightweight copy protection, registering a pseudo-service with
1691*48a54d36SAndroid Build Coastguard Worker // a port number of zero. When two instances of the protected client are allowed to run on one
1692*48a54d36SAndroid Build Coastguard Worker // machine, we don't want to see misleading "Bogus client" messages in syslog and the console.
1693*48a54d36SAndroid Build Coastguard Worker if (!mDNSIPPortIsZero(request->u.servicereg.port))
1694*48a54d36SAndroid Build Coastguard Worker {
1695*48a54d36SAndroid Build Coastguard Worker int count = CountExistingRegistrations(&srv, request->u.servicereg.port);
1696*48a54d36SAndroid Build Coastguard Worker if (count)
1697*48a54d36SAndroid Build Coastguard Worker LogMsg("Client application registered %d identical instances of service %##s port %u.",
1698*48a54d36SAndroid Build Coastguard Worker count+1, srv.c, mDNSVal16(request->u.servicereg.port));
1699*48a54d36SAndroid Build Coastguard Worker }
1700*48a54d36SAndroid Build Coastguard Worker
1701*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceRegister(%X, %d, \"%s\", \"%s\", \"%s\", \"%s\", %u) START",
1702*48a54d36SAndroid Build Coastguard Worker request->sd, flags, interfaceIndex, name, request->u.servicereg.type_as_string, domain, host, mDNSVal16(request->u.servicereg.port));
1703*48a54d36SAndroid Build Coastguard Worker
1704*48a54d36SAndroid Build Coastguard Worker // We need to unconditionally set request->terminate, because even if we didn't successfully
1705*48a54d36SAndroid Build Coastguard Worker // start any registrations right now, subsequent configuration changes may cause successful
1706*48a54d36SAndroid Build Coastguard Worker // registrations to be added, and we'll need to cancel them before freeing this memory.
1707*48a54d36SAndroid Build Coastguard Worker // We also need to set request->terminate first, before adding additional service instances,
1708*48a54d36SAndroid Build Coastguard Worker // because the uds_validatelists uses the request->terminate function pointer to determine
1709*48a54d36SAndroid Build Coastguard Worker // what kind of request this is, and therefore what kind of list validation is required.
1710*48a54d36SAndroid Build Coastguard Worker request->terminate = regservice_termination_callback;
1711*48a54d36SAndroid Build Coastguard Worker
1712*48a54d36SAndroid Build Coastguard Worker err = register_service_instance(request, &d);
1713*48a54d36SAndroid Build Coastguard Worker
1714*48a54d36SAndroid Build Coastguard Worker #if 0
1715*48a54d36SAndroid Build Coastguard Worker err = AuthorizedDomain(request, &d, AutoRegistrationDomains) ? register_service_instance(request, &d) : mStatus_NoError;
1716*48a54d36SAndroid Build Coastguard Worker #endif
1717*48a54d36SAndroid Build Coastguard Worker if (!err)
1718*48a54d36SAndroid Build Coastguard Worker {
1719*48a54d36SAndroid Build Coastguard Worker if (request->u.servicereg.autoname) UpdateDeviceInfoRecord(&mDNSStorage);
1720*48a54d36SAndroid Build Coastguard Worker
1721*48a54d36SAndroid Build Coastguard Worker if (!*domain)
1722*48a54d36SAndroid Build Coastguard Worker {
1723*48a54d36SAndroid Build Coastguard Worker DNameListElem *ptr;
1724*48a54d36SAndroid Build Coastguard Worker // Note that we don't report errors for non-local, non-explicit domains
1725*48a54d36SAndroid Build Coastguard Worker for (ptr = AutoRegistrationDomains; ptr; ptr = ptr->next)
1726*48a54d36SAndroid Build Coastguard Worker if (!ptr->uid || SystemUID(request->uid) || request->uid == ptr->uid)
1727*48a54d36SAndroid Build Coastguard Worker register_service_instance(request, &ptr->name);
1728*48a54d36SAndroid Build Coastguard Worker }
1729*48a54d36SAndroid Build Coastguard Worker }
1730*48a54d36SAndroid Build Coastguard Worker
1731*48a54d36SAndroid Build Coastguard Worker return(err);
1732*48a54d36SAndroid Build Coastguard Worker }
1733*48a54d36SAndroid Build Coastguard Worker
1734*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
1735*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
1736*48a54d36SAndroid Build Coastguard Worker #pragma mark -
1737*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceBrowse
1738*48a54d36SAndroid Build Coastguard Worker #endif
1739*48a54d36SAndroid Build Coastguard Worker
FoundInstance(mDNS * const m,DNSQuestion * question,const ResourceRecord * const answer,QC_result AddRecord)1740*48a54d36SAndroid Build Coastguard Worker mDNSlocal void FoundInstance(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
1741*48a54d36SAndroid Build Coastguard Worker {
1742*48a54d36SAndroid Build Coastguard Worker const DNSServiceFlags flags = AddRecord ? kDNSServiceFlagsAdd : 0;
1743*48a54d36SAndroid Build Coastguard Worker request_state *req = question->QuestionContext;
1744*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
1745*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
1746*48a54d36SAndroid Build Coastguard Worker
1747*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype != kDNSType_PTR)
1748*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: FoundInstance: Should not be called with rrtype %d (not a PTR record)", req->sd, answer->rrtype); return; }
1749*48a54d36SAndroid Build Coastguard Worker
1750*48a54d36SAndroid Build Coastguard Worker if (GenerateNTDResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError) != mStatus_NoError)
1751*48a54d36SAndroid Build Coastguard Worker {
1752*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&req->u.browser.regtype, (const domainname*)"\x09_services\x07_dns-sd\x04_udp"))
1753*48a54d36SAndroid Build Coastguard Worker {
1754*48a54d36SAndroid Build Coastguard Worker // Special support to enable the DNSServiceBrowse call made by Bonjour Browser
1755*48a54d36SAndroid Build Coastguard Worker // Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse
1756*48a54d36SAndroid Build Coastguard Worker GenerateBonjourBrowserResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError);
1757*48a54d36SAndroid Build Coastguard Worker goto bonjourbrowserhack;
1758*48a54d36SAndroid Build Coastguard Worker }
1759*48a54d36SAndroid Build Coastguard Worker
1760*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: FoundInstance: %##s PTR %##s received from network is not valid DNS-SD service pointer",
1761*48a54d36SAndroid Build Coastguard Worker req->sd, answer->name->c, answer->rdata->u.name.c);
1762*48a54d36SAndroid Build Coastguard Worker return;
1763*48a54d36SAndroid Build Coastguard Worker }
1764*48a54d36SAndroid Build Coastguard Worker
1765*48a54d36SAndroid Build Coastguard Worker bonjourbrowserhack:
1766*48a54d36SAndroid Build Coastguard Worker
1767*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceBrowse(%##s, %s) RESULT %s %d: %s",
1768*48a54d36SAndroid Build Coastguard Worker req->sd, question->qname.c, DNSTypeName(question->qtype), AddRecord ? "Add" : "Rmv",
1769*48a54d36SAndroid Build Coastguard Worker mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse), RRDisplayString(m, answer));
1770*48a54d36SAndroid Build Coastguard Worker
1771*48a54d36SAndroid Build Coastguard Worker append_reply(req, rep);
1772*48a54d36SAndroid Build Coastguard Worker }
1773*48a54d36SAndroid Build Coastguard Worker
add_domain_to_browser(request_state * info,const domainname * d)1774*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus add_domain_to_browser(request_state *info, const domainname *d)
1775*48a54d36SAndroid Build Coastguard Worker {
1776*48a54d36SAndroid Build Coastguard Worker browser_t *b, *p;
1777*48a54d36SAndroid Build Coastguard Worker mStatus err;
1778*48a54d36SAndroid Build Coastguard Worker
1779*48a54d36SAndroid Build Coastguard Worker for (p = info->u.browser.browsers; p; p = p->next)
1780*48a54d36SAndroid Build Coastguard Worker {
1781*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&p->domain, d))
1782*48a54d36SAndroid Build Coastguard Worker { debugf("add_domain_to_browser %##s already in list", d->c); return mStatus_AlreadyRegistered; }
1783*48a54d36SAndroid Build Coastguard Worker }
1784*48a54d36SAndroid Build Coastguard Worker
1785*48a54d36SAndroid Build Coastguard Worker b = mallocL("browser_t", sizeof(*b));
1786*48a54d36SAndroid Build Coastguard Worker if (!b) return mStatus_NoMemoryErr;
1787*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&b->domain, d);
1788*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartBrowse(&mDNSStorage, &b->q,
1789*48a54d36SAndroid Build Coastguard Worker &info->u.browser.regtype, d, info->u.browser.interface_id, info->u.browser.ForceMCast, FoundInstance, info);
1790*48a54d36SAndroid Build Coastguard Worker if (err)
1791*48a54d36SAndroid Build Coastguard Worker {
1792*48a54d36SAndroid Build Coastguard Worker LogMsg("mDNS_StartBrowse returned %d for type %##s domain %##s", err, info->u.browser.regtype.c, d->c);
1793*48a54d36SAndroid Build Coastguard Worker freeL("browser_t/add_domain_to_browser", b);
1794*48a54d36SAndroid Build Coastguard Worker }
1795*48a54d36SAndroid Build Coastguard Worker else
1796*48a54d36SAndroid Build Coastguard Worker {
1797*48a54d36SAndroid Build Coastguard Worker b->next = info->u.browser.browsers;
1798*48a54d36SAndroid Build Coastguard Worker info->u.browser.browsers = b;
1799*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceBrowse(%##s) START", info->sd, b->q.qname.c);
1800*48a54d36SAndroid Build Coastguard Worker if (info->u.browser.interface_id == mDNSInterface_P2P || (!info->u.browser.interface_id && SameDomainName(&b->domain, &localdomain) && (info->flags & kDNSServiceFlagsIncludeP2P)))
1801*48a54d36SAndroid Build Coastguard Worker {
1802*48a54d36SAndroid Build Coastguard Worker domainname tmp;
1803*48a54d36SAndroid Build Coastguard Worker ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &b->domain);
1804*48a54d36SAndroid Build Coastguard Worker LogInfo("add_domain_to_browser: calling external_start_browsing_for_service()");
1805*48a54d36SAndroid Build Coastguard Worker external_start_browsing_for_service(&mDNSStorage, &tmp, kDNSType_PTR);
1806*48a54d36SAndroid Build Coastguard Worker }
1807*48a54d36SAndroid Build Coastguard Worker }
1808*48a54d36SAndroid Build Coastguard Worker return err;
1809*48a54d36SAndroid Build Coastguard Worker }
1810*48a54d36SAndroid Build Coastguard Worker
browse_termination_callback(request_state * info)1811*48a54d36SAndroid Build Coastguard Worker mDNSlocal void browse_termination_callback(request_state *info)
1812*48a54d36SAndroid Build Coastguard Worker {
1813*48a54d36SAndroid Build Coastguard Worker while (info->u.browser.browsers)
1814*48a54d36SAndroid Build Coastguard Worker {
1815*48a54d36SAndroid Build Coastguard Worker browser_t *ptr = info->u.browser.browsers;
1816*48a54d36SAndroid Build Coastguard Worker
1817*48a54d36SAndroid Build Coastguard Worker if (info->u.browser.interface_id == mDNSInterface_P2P || (!info->u.browser.interface_id && SameDomainName(&ptr->domain, &localdomain) && (info->flags & kDNSServiceFlagsIncludeP2P)))
1818*48a54d36SAndroid Build Coastguard Worker {
1819*48a54d36SAndroid Build Coastguard Worker domainname tmp;
1820*48a54d36SAndroid Build Coastguard Worker ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &ptr->domain);
1821*48a54d36SAndroid Build Coastguard Worker LogInfo("browse_termination_callback: calling external_stop_browsing_for_service()");
1822*48a54d36SAndroid Build Coastguard Worker external_stop_browsing_for_service(&mDNSStorage, &tmp, kDNSType_PTR);
1823*48a54d36SAndroid Build Coastguard Worker }
1824*48a54d36SAndroid Build Coastguard Worker
1825*48a54d36SAndroid Build Coastguard Worker info->u.browser.browsers = ptr->next;
1826*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceBrowse(%##s) STOP", info->sd, ptr->q.qname.c);
1827*48a54d36SAndroid Build Coastguard Worker mDNS_StopBrowse(&mDNSStorage, &ptr->q); // no need to error-check result
1828*48a54d36SAndroid Build Coastguard Worker freeL("browser_t/browse_termination_callback", ptr);
1829*48a54d36SAndroid Build Coastguard Worker }
1830*48a54d36SAndroid Build Coastguard Worker }
1831*48a54d36SAndroid Build Coastguard Worker
udsserver_automatic_browse_domain_changed(const DNameListElem * const d,const mDNSBool add)1832*48a54d36SAndroid Build Coastguard Worker mDNSlocal void udsserver_automatic_browse_domain_changed(const DNameListElem *const d, const mDNSBool add)
1833*48a54d36SAndroid Build Coastguard Worker {
1834*48a54d36SAndroid Build Coastguard Worker request_state *request;
1835*48a54d36SAndroid Build Coastguard Worker debugf("udsserver_automatic_browse_domain_changed: %s default browse domain %##s", add ? "Adding" : "Removing", d->name.c);
1836*48a54d36SAndroid Build Coastguard Worker
1837*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
1838*48a54d36SAndroid Build Coastguard Worker machserver_automatic_browse_domain_changed(&d->name, add);
1839*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
1840*48a54d36SAndroid Build Coastguard Worker
1841*48a54d36SAndroid Build Coastguard Worker for (request = all_requests; request; request = request->next)
1842*48a54d36SAndroid Build Coastguard Worker {
1843*48a54d36SAndroid Build Coastguard Worker if (request->terminate != browse_termination_callback) continue; // Not a browse operation
1844*48a54d36SAndroid Build Coastguard Worker if (!request->u.browser.default_domain) continue; // Not an auto-browse operation
1845*48a54d36SAndroid Build Coastguard Worker if (!d->uid || SystemUID(request->uid) || request->uid == d->uid)
1846*48a54d36SAndroid Build Coastguard Worker {
1847*48a54d36SAndroid Build Coastguard Worker browser_t **ptr = &request->u.browser.browsers;
1848*48a54d36SAndroid Build Coastguard Worker while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next;
1849*48a54d36SAndroid Build Coastguard Worker if (add)
1850*48a54d36SAndroid Build Coastguard Worker {
1851*48a54d36SAndroid Build Coastguard Worker // If we don't already have this domain in our list for this browse operation, add it now
1852*48a54d36SAndroid Build Coastguard Worker if (!*ptr) add_domain_to_browser(request, &d->name);
1853*48a54d36SAndroid Build Coastguard Worker else debugf("udsserver_automatic_browse_domain_changed %##s already in list, not re-adding", &d->name);
1854*48a54d36SAndroid Build Coastguard Worker }
1855*48a54d36SAndroid Build Coastguard Worker else
1856*48a54d36SAndroid Build Coastguard Worker {
1857*48a54d36SAndroid Build Coastguard Worker if (!*ptr) LogMsg("udsserver_automatic_browse_domain_changed ERROR %##s not found", &d->name);
1858*48a54d36SAndroid Build Coastguard Worker else
1859*48a54d36SAndroid Build Coastguard Worker {
1860*48a54d36SAndroid Build Coastguard Worker DNameListElem *p;
1861*48a54d36SAndroid Build Coastguard Worker for (p = AutoBrowseDomains; p; p=p->next)
1862*48a54d36SAndroid Build Coastguard Worker if (!p->uid || SystemUID(request->uid) || request->uid == p->uid)
1863*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&d->name, &p->name)) break;
1864*48a54d36SAndroid Build Coastguard Worker if (p) debugf("udsserver_automatic_browse_domain_changed %##s still in list, not removing", &d->name);
1865*48a54d36SAndroid Build Coastguard Worker else
1866*48a54d36SAndroid Build Coastguard Worker {
1867*48a54d36SAndroid Build Coastguard Worker browser_t *rem = *ptr;
1868*48a54d36SAndroid Build Coastguard Worker *ptr = (*ptr)->next;
1869*48a54d36SAndroid Build Coastguard Worker mDNS_StopQueryWithRemoves(&mDNSStorage, &rem->q);
1870*48a54d36SAndroid Build Coastguard Worker freeL("browser_t/udsserver_automatic_browse_domain_changed", rem);
1871*48a54d36SAndroid Build Coastguard Worker }
1872*48a54d36SAndroid Build Coastguard Worker }
1873*48a54d36SAndroid Build Coastguard Worker }
1874*48a54d36SAndroid Build Coastguard Worker }
1875*48a54d36SAndroid Build Coastguard Worker }
1876*48a54d36SAndroid Build Coastguard Worker }
1877*48a54d36SAndroid Build Coastguard Worker
FreeARElemCallback(mDNS * const m,AuthRecord * const rr,mStatus result)1878*48a54d36SAndroid Build Coastguard Worker mDNSlocal void FreeARElemCallback(mDNS *const m, AuthRecord *const rr, mStatus result)
1879*48a54d36SAndroid Build Coastguard Worker {
1880*48a54d36SAndroid Build Coastguard Worker (void)m; // unused
1881*48a54d36SAndroid Build Coastguard Worker if (result == mStatus_MemFree)
1882*48a54d36SAndroid Build Coastguard Worker {
1883*48a54d36SAndroid Build Coastguard Worker // On shutdown, mDNS_Close automatically deregisters all records
1884*48a54d36SAndroid Build Coastguard Worker // Since in this case no one has called DeregisterLocalOnlyDomainEnumPTR to cut the record
1885*48a54d36SAndroid Build Coastguard Worker // from the LocalDomainEnumRecords list, we do this here before we free the memory.
1886*48a54d36SAndroid Build Coastguard Worker // (This should actually no longer be necessary, now that we do the proper cleanup in
1887*48a54d36SAndroid Build Coastguard Worker // udsserver_exit. To confirm this, we'll log an error message if we do find a record that
1888*48a54d36SAndroid Build Coastguard Worker // hasn't been cut from the list yet. If these messages don't appear, we can delete this code.)
1889*48a54d36SAndroid Build Coastguard Worker ARListElem **ptr = &LocalDomainEnumRecords;
1890*48a54d36SAndroid Build Coastguard Worker while (*ptr && &(*ptr)->ar != rr) ptr = &(*ptr)->next;
1891*48a54d36SAndroid Build Coastguard Worker if (*ptr) { *ptr = (*ptr)->next; LogMsg("FreeARElemCallback: Have to cut %s", ARDisplayString(m, rr)); }
1892*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(rr->RecordContext);
1893*48a54d36SAndroid Build Coastguard Worker }
1894*48a54d36SAndroid Build Coastguard Worker }
1895*48a54d36SAndroid Build Coastguard Worker
1896*48a54d36SAndroid Build Coastguard Worker // RegisterLocalOnlyDomainEnumPTR and DeregisterLocalOnlyDomainEnumPTR largely duplicate code in
1897*48a54d36SAndroid Build Coastguard Worker // "FoundDomain" in uDNS.c for creating and destroying these special mDNSInterface_LocalOnly records.
1898*48a54d36SAndroid Build Coastguard Worker // We may want to turn the common code into a subroutine.
1899*48a54d36SAndroid Build Coastguard Worker
RegisterLocalOnlyDomainEnumPTR(mDNS * m,const domainname * d,int type)1900*48a54d36SAndroid Build Coastguard Worker mDNSlocal void RegisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type)
1901*48a54d36SAndroid Build Coastguard Worker {
1902*48a54d36SAndroid Build Coastguard Worker // allocate/register legacy and non-legacy _browse PTR record
1903*48a54d36SAndroid Build Coastguard Worker mStatus err;
1904*48a54d36SAndroid Build Coastguard Worker ARListElem *ptr = mDNSPlatformMemAllocate(sizeof(*ptr));
1905*48a54d36SAndroid Build Coastguard Worker
1906*48a54d36SAndroid Build Coastguard Worker debugf("Incrementing %s refcount for %##s",
1907*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeBrowse ) ? "browse domain " :
1908*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeRegistration ) ? "registration dom" :
1909*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c);
1910*48a54d36SAndroid Build Coastguard Worker
1911*48a54d36SAndroid Build Coastguard Worker mDNS_SetupResourceRecord(&ptr->ar, mDNSNULL, mDNSInterface_LocalOnly, kDNSType_PTR, 7200, kDNSRecordTypeShared, AuthRecordLocalOnly, FreeARElemCallback, ptr);
1912*48a54d36SAndroid Build Coastguard Worker MakeDomainNameFromDNSNameString(&ptr->ar.namestorage, mDNS_DomainTypeNames[type]);
1913*48a54d36SAndroid Build Coastguard Worker AppendDNSNameString (&ptr->ar.namestorage, "local");
1914*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&ptr->ar.resrec.rdata->u.name, d);
1915*48a54d36SAndroid Build Coastguard Worker err = mDNS_Register(m, &ptr->ar);
1916*48a54d36SAndroid Build Coastguard Worker if (err)
1917*48a54d36SAndroid Build Coastguard Worker {
1918*48a54d36SAndroid Build Coastguard Worker LogMsg("SetSCPrefsBrowseDomain: mDNS_Register returned error %d", err);
1919*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(ptr);
1920*48a54d36SAndroid Build Coastguard Worker }
1921*48a54d36SAndroid Build Coastguard Worker else
1922*48a54d36SAndroid Build Coastguard Worker {
1923*48a54d36SAndroid Build Coastguard Worker ptr->next = LocalDomainEnumRecords;
1924*48a54d36SAndroid Build Coastguard Worker LocalDomainEnumRecords = ptr;
1925*48a54d36SAndroid Build Coastguard Worker }
1926*48a54d36SAndroid Build Coastguard Worker }
1927*48a54d36SAndroid Build Coastguard Worker
DeregisterLocalOnlyDomainEnumPTR(mDNS * m,const domainname * d,int type)1928*48a54d36SAndroid Build Coastguard Worker mDNSlocal void DeregisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type)
1929*48a54d36SAndroid Build Coastguard Worker {
1930*48a54d36SAndroid Build Coastguard Worker ARListElem **ptr = &LocalDomainEnumRecords;
1931*48a54d36SAndroid Build Coastguard Worker domainname lhs; // left-hand side of PTR, for comparison
1932*48a54d36SAndroid Build Coastguard Worker
1933*48a54d36SAndroid Build Coastguard Worker debugf("Decrementing %s refcount for %##s",
1934*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeBrowse ) ? "browse domain " :
1935*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeRegistration ) ? "registration dom" :
1936*48a54d36SAndroid Build Coastguard Worker (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c);
1937*48a54d36SAndroid Build Coastguard Worker
1938*48a54d36SAndroid Build Coastguard Worker MakeDomainNameFromDNSNameString(&lhs, mDNS_DomainTypeNames[type]);
1939*48a54d36SAndroid Build Coastguard Worker AppendDNSNameString (&lhs, "local");
1940*48a54d36SAndroid Build Coastguard Worker
1941*48a54d36SAndroid Build Coastguard Worker while (*ptr)
1942*48a54d36SAndroid Build Coastguard Worker {
1943*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&(*ptr)->ar.resrec.rdata->u.name, d) && SameDomainName((*ptr)->ar.resrec.name, &lhs))
1944*48a54d36SAndroid Build Coastguard Worker {
1945*48a54d36SAndroid Build Coastguard Worker ARListElem *rem = *ptr;
1946*48a54d36SAndroid Build Coastguard Worker *ptr = (*ptr)->next;
1947*48a54d36SAndroid Build Coastguard Worker mDNS_Deregister(m, &rem->ar);
1948*48a54d36SAndroid Build Coastguard Worker return;
1949*48a54d36SAndroid Build Coastguard Worker }
1950*48a54d36SAndroid Build Coastguard Worker else ptr = &(*ptr)->next;
1951*48a54d36SAndroid Build Coastguard Worker }
1952*48a54d36SAndroid Build Coastguard Worker }
1953*48a54d36SAndroid Build Coastguard Worker
AddAutoBrowseDomain(const mDNSu32 uid,const domainname * const name)1954*48a54d36SAndroid Build Coastguard Worker mDNSlocal void AddAutoBrowseDomain(const mDNSu32 uid, const domainname *const name)
1955*48a54d36SAndroid Build Coastguard Worker {
1956*48a54d36SAndroid Build Coastguard Worker DNameListElem *new = mDNSPlatformMemAllocate(sizeof(DNameListElem));
1957*48a54d36SAndroid Build Coastguard Worker if (!new) { LogMsg("ERROR: malloc"); return; }
1958*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&new->name, name);
1959*48a54d36SAndroid Build Coastguard Worker new->uid = uid;
1960*48a54d36SAndroid Build Coastguard Worker new->next = AutoBrowseDomains;
1961*48a54d36SAndroid Build Coastguard Worker AutoBrowseDomains = new;
1962*48a54d36SAndroid Build Coastguard Worker udsserver_automatic_browse_domain_changed(new, mDNStrue);
1963*48a54d36SAndroid Build Coastguard Worker }
1964*48a54d36SAndroid Build Coastguard Worker
RmvAutoBrowseDomain(const mDNSu32 uid,const domainname * const name)1965*48a54d36SAndroid Build Coastguard Worker mDNSlocal void RmvAutoBrowseDomain(const mDNSu32 uid, const domainname *const name)
1966*48a54d36SAndroid Build Coastguard Worker {
1967*48a54d36SAndroid Build Coastguard Worker DNameListElem **p = &AutoBrowseDomains;
1968*48a54d36SAndroid Build Coastguard Worker while (*p && (!SameDomainName(&(*p)->name, name) || (*p)->uid != uid)) p = &(*p)->next;
1969*48a54d36SAndroid Build Coastguard Worker if (!*p) LogMsg("RmvAutoBrowseDomain: Got remove event for domain %##s not in list", name->c);
1970*48a54d36SAndroid Build Coastguard Worker else
1971*48a54d36SAndroid Build Coastguard Worker {
1972*48a54d36SAndroid Build Coastguard Worker DNameListElem *ptr = *p;
1973*48a54d36SAndroid Build Coastguard Worker *p = ptr->next;
1974*48a54d36SAndroid Build Coastguard Worker udsserver_automatic_browse_domain_changed(ptr, mDNSfalse);
1975*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(ptr);
1976*48a54d36SAndroid Build Coastguard Worker }
1977*48a54d36SAndroid Build Coastguard Worker }
1978*48a54d36SAndroid Build Coastguard Worker
SetPrefsBrowseDomains(mDNS * m,DNameListElem * browseDomains,mDNSBool add)1979*48a54d36SAndroid Build Coastguard Worker mDNSlocal void SetPrefsBrowseDomains(mDNS *m, DNameListElem *browseDomains, mDNSBool add)
1980*48a54d36SAndroid Build Coastguard Worker {
1981*48a54d36SAndroid Build Coastguard Worker DNameListElem *d;
1982*48a54d36SAndroid Build Coastguard Worker for (d = browseDomains; d; d = d->next)
1983*48a54d36SAndroid Build Coastguard Worker {
1984*48a54d36SAndroid Build Coastguard Worker if (add)
1985*48a54d36SAndroid Build Coastguard Worker {
1986*48a54d36SAndroid Build Coastguard Worker RegisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse);
1987*48a54d36SAndroid Build Coastguard Worker AddAutoBrowseDomain(d->uid, &d->name);
1988*48a54d36SAndroid Build Coastguard Worker }
1989*48a54d36SAndroid Build Coastguard Worker else
1990*48a54d36SAndroid Build Coastguard Worker {
1991*48a54d36SAndroid Build Coastguard Worker DeregisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse);
1992*48a54d36SAndroid Build Coastguard Worker RmvAutoBrowseDomain(d->uid, &d->name);
1993*48a54d36SAndroid Build Coastguard Worker }
1994*48a54d36SAndroid Build Coastguard Worker }
1995*48a54d36SAndroid Build Coastguard Worker }
1996*48a54d36SAndroid Build Coastguard Worker
UpdateDeviceInfoRecord(mDNS * const m)1997*48a54d36SAndroid Build Coastguard Worker mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m)
1998*48a54d36SAndroid Build Coastguard Worker {
1999*48a54d36SAndroid Build Coastguard Worker int num_autoname = 0;
2000*48a54d36SAndroid Build Coastguard Worker request_state *req;
2001*48a54d36SAndroid Build Coastguard Worker for (req = all_requests; req; req = req->next)
2002*48a54d36SAndroid Build Coastguard Worker if (req->terminate == regservice_termination_callback && req->u.servicereg.autoname)
2003*48a54d36SAndroid Build Coastguard Worker num_autoname++;
2004*48a54d36SAndroid Build Coastguard Worker
2005*48a54d36SAndroid Build Coastguard Worker // If DeviceInfo record is currently registered, see if we need to deregister it
2006*48a54d36SAndroid Build Coastguard Worker if (m->DeviceInfo.resrec.RecordType != kDNSRecordTypeUnregistered)
2007*48a54d36SAndroid Build Coastguard Worker if (num_autoname == 0 || !SameDomainLabelCS(m->DeviceInfo.resrec.name->c, m->nicelabel.c))
2008*48a54d36SAndroid Build Coastguard Worker {
2009*48a54d36SAndroid Build Coastguard Worker LogOperation("UpdateDeviceInfoRecord Deregister %##s", m->DeviceInfo.resrec.name);
2010*48a54d36SAndroid Build Coastguard Worker mDNS_Deregister(m, &m->DeviceInfo);
2011*48a54d36SAndroid Build Coastguard Worker }
2012*48a54d36SAndroid Build Coastguard Worker
2013*48a54d36SAndroid Build Coastguard Worker // If DeviceInfo record is not currently registered, see if we need to register it
2014*48a54d36SAndroid Build Coastguard Worker if (m->DeviceInfo.resrec.RecordType == kDNSRecordTypeUnregistered)
2015*48a54d36SAndroid Build Coastguard Worker if (num_autoname > 0)
2016*48a54d36SAndroid Build Coastguard Worker {
2017*48a54d36SAndroid Build Coastguard Worker mDNSu8 len = m->HIHardware.c[0] < 255 - 6 ? m->HIHardware.c[0] : 255 - 6;
2018*48a54d36SAndroid Build Coastguard Worker mDNS_SetupResourceRecord(&m->DeviceInfo, mDNSNULL, mDNSNULL, kDNSType_TXT, kStandardTTL, kDNSRecordTypeAdvisory, AuthRecordAny, mDNSNULL, mDNSNULL);
2019*48a54d36SAndroid Build Coastguard Worker ConstructServiceName(&m->DeviceInfo.namestorage, &m->nicelabel, &DeviceInfoName, &localdomain);
2020*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 1, "model=", 6);
2021*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 7, m->HIHardware.c + 1, len);
2022*48a54d36SAndroid Build Coastguard Worker m->DeviceInfo.resrec.rdata->u.data[0] = 6 + len; // "model=" plus the device string
2023*48a54d36SAndroid Build Coastguard Worker m->DeviceInfo.resrec.rdlength = 7 + len; // One extra for the length byte at the start of the string
2024*48a54d36SAndroid Build Coastguard Worker LogOperation("UpdateDeviceInfoRecord Register %##s", m->DeviceInfo.resrec.name);
2025*48a54d36SAndroid Build Coastguard Worker mDNS_Register(m, &m->DeviceInfo);
2026*48a54d36SAndroid Build Coastguard Worker }
2027*48a54d36SAndroid Build Coastguard Worker }
2028*48a54d36SAndroid Build Coastguard Worker
udsserver_handle_configchange(mDNS * const m)2029*48a54d36SAndroid Build Coastguard Worker mDNSexport void udsserver_handle_configchange(mDNS *const m)
2030*48a54d36SAndroid Build Coastguard Worker {
2031*48a54d36SAndroid Build Coastguard Worker request_state *req;
2032*48a54d36SAndroid Build Coastguard Worker service_instance *ptr;
2033*48a54d36SAndroid Build Coastguard Worker DNameListElem *RegDomains = NULL;
2034*48a54d36SAndroid Build Coastguard Worker DNameListElem *BrowseDomains = NULL;
2035*48a54d36SAndroid Build Coastguard Worker DNameListElem *p;
2036*48a54d36SAndroid Build Coastguard Worker
2037*48a54d36SAndroid Build Coastguard Worker UpdateDeviceInfoRecord(m);
2038*48a54d36SAndroid Build Coastguard Worker
2039*48a54d36SAndroid Build Coastguard Worker // For autoname services, see if the default service name has changed, necessitating an automatic update
2040*48a54d36SAndroid Build Coastguard Worker for (req = all_requests; req; req = req->next)
2041*48a54d36SAndroid Build Coastguard Worker if (req->terminate == regservice_termination_callback)
2042*48a54d36SAndroid Build Coastguard Worker if (req->u.servicereg.autoname && !SameDomainLabelCS(req->u.servicereg.name.c, m->nicelabel.c))
2043*48a54d36SAndroid Build Coastguard Worker {
2044*48a54d36SAndroid Build Coastguard Worker req->u.servicereg.name = m->nicelabel;
2045*48a54d36SAndroid Build Coastguard Worker for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next)
2046*48a54d36SAndroid Build Coastguard Worker {
2047*48a54d36SAndroid Build Coastguard Worker ptr->renameonmemfree = 1;
2048*48a54d36SAndroid Build Coastguard Worker if (ptr->clientnotified) SendServiceRemovalNotification(&ptr->srs);
2049*48a54d36SAndroid Build Coastguard Worker LogInfo("udsserver_handle_configchange: Calling deregister for Service %##s", ptr->srs.RR_PTR.resrec.name->c);
2050*48a54d36SAndroid Build Coastguard Worker if (mDNS_DeregisterService_drt(m, &ptr->srs, mDNS_Dereg_rapid))
2051*48a54d36SAndroid Build Coastguard Worker regservice_callback(m, &ptr->srs, mStatus_MemFree); // If service deregistered already, we can re-register immediately
2052*48a54d36SAndroid Build Coastguard Worker }
2053*48a54d36SAndroid Build Coastguard Worker }
2054*48a54d36SAndroid Build Coastguard Worker
2055*48a54d36SAndroid Build Coastguard Worker // Let the platform layer get the current DNS information
2056*48a54d36SAndroid Build Coastguard Worker mDNS_Lock(m);
2057*48a54d36SAndroid Build Coastguard Worker mDNSPlatformSetDNSConfig(m, mDNSfalse, mDNSfalse, mDNSNULL, &RegDomains, &BrowseDomains);
2058*48a54d36SAndroid Build Coastguard Worker mDNS_Unlock(m);
2059*48a54d36SAndroid Build Coastguard Worker
2060*48a54d36SAndroid Build Coastguard Worker // Any automatic registration domains are also implicitly automatic browsing domains
2061*48a54d36SAndroid Build Coastguard Worker if (RegDomains) SetPrefsBrowseDomains(m, RegDomains, mDNStrue); // Add the new list first
2062*48a54d36SAndroid Build Coastguard Worker if (AutoRegistrationDomains) SetPrefsBrowseDomains(m, AutoRegistrationDomains, mDNSfalse); // Then clear the old list
2063*48a54d36SAndroid Build Coastguard Worker
2064*48a54d36SAndroid Build Coastguard Worker // Add any new domains not already in our AutoRegistrationDomains list
2065*48a54d36SAndroid Build Coastguard Worker for (p=RegDomains; p; p=p->next)
2066*48a54d36SAndroid Build Coastguard Worker {
2067*48a54d36SAndroid Build Coastguard Worker DNameListElem **pp = &AutoRegistrationDomains;
2068*48a54d36SAndroid Build Coastguard Worker while (*pp && ((*pp)->uid != p->uid || !SameDomainName(&(*pp)->name, &p->name))) pp = &(*pp)->next;
2069*48a54d36SAndroid Build Coastguard Worker if (!*pp) // If not found in our existing list, this is a new default registration domain
2070*48a54d36SAndroid Build Coastguard Worker {
2071*48a54d36SAndroid Build Coastguard Worker RegisterLocalOnlyDomainEnumPTR(m, &p->name, mDNS_DomainTypeRegistration);
2072*48a54d36SAndroid Build Coastguard Worker udsserver_default_reg_domain_changed(p, mDNStrue);
2073*48a54d36SAndroid Build Coastguard Worker }
2074*48a54d36SAndroid Build Coastguard Worker else // else found same domainname in both old and new lists, so no change, just delete old copy
2075*48a54d36SAndroid Build Coastguard Worker {
2076*48a54d36SAndroid Build Coastguard Worker DNameListElem *del = *pp;
2077*48a54d36SAndroid Build Coastguard Worker *pp = (*pp)->next;
2078*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(del);
2079*48a54d36SAndroid Build Coastguard Worker }
2080*48a54d36SAndroid Build Coastguard Worker }
2081*48a54d36SAndroid Build Coastguard Worker
2082*48a54d36SAndroid Build Coastguard Worker // Delete any domains in our old AutoRegistrationDomains list that are now gone
2083*48a54d36SAndroid Build Coastguard Worker while (AutoRegistrationDomains)
2084*48a54d36SAndroid Build Coastguard Worker {
2085*48a54d36SAndroid Build Coastguard Worker DNameListElem *del = AutoRegistrationDomains;
2086*48a54d36SAndroid Build Coastguard Worker AutoRegistrationDomains = AutoRegistrationDomains->next; // Cut record from list FIRST,
2087*48a54d36SAndroid Build Coastguard Worker DeregisterLocalOnlyDomainEnumPTR(m, &del->name, mDNS_DomainTypeRegistration);
2088*48a54d36SAndroid Build Coastguard Worker udsserver_default_reg_domain_changed(del, mDNSfalse); // before calling udsserver_default_reg_domain_changed()
2089*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(del);
2090*48a54d36SAndroid Build Coastguard Worker }
2091*48a54d36SAndroid Build Coastguard Worker
2092*48a54d36SAndroid Build Coastguard Worker // Now we have our new updated automatic registration domain list
2093*48a54d36SAndroid Build Coastguard Worker AutoRegistrationDomains = RegDomains;
2094*48a54d36SAndroid Build Coastguard Worker
2095*48a54d36SAndroid Build Coastguard Worker // Add new browse domains to internal list
2096*48a54d36SAndroid Build Coastguard Worker if (BrowseDomains) SetPrefsBrowseDomains(m, BrowseDomains, mDNStrue);
2097*48a54d36SAndroid Build Coastguard Worker
2098*48a54d36SAndroid Build Coastguard Worker // Remove old browse domains from internal list
2099*48a54d36SAndroid Build Coastguard Worker if (SCPrefBrowseDomains)
2100*48a54d36SAndroid Build Coastguard Worker {
2101*48a54d36SAndroid Build Coastguard Worker SetPrefsBrowseDomains(m, SCPrefBrowseDomains, mDNSfalse);
2102*48a54d36SAndroid Build Coastguard Worker while (SCPrefBrowseDomains)
2103*48a54d36SAndroid Build Coastguard Worker {
2104*48a54d36SAndroid Build Coastguard Worker DNameListElem *fptr = SCPrefBrowseDomains;
2105*48a54d36SAndroid Build Coastguard Worker SCPrefBrowseDomains = SCPrefBrowseDomains->next;
2106*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemFree(fptr);
2107*48a54d36SAndroid Build Coastguard Worker }
2108*48a54d36SAndroid Build Coastguard Worker }
2109*48a54d36SAndroid Build Coastguard Worker
2110*48a54d36SAndroid Build Coastguard Worker // Replace the old browse domains array with the new array
2111*48a54d36SAndroid Build Coastguard Worker SCPrefBrowseDomains = BrowseDomains;
2112*48a54d36SAndroid Build Coastguard Worker }
2113*48a54d36SAndroid Build Coastguard Worker
AutomaticBrowseDomainChange(mDNS * const m,DNSQuestion * q,const ResourceRecord * const answer,QC_result AddRecord)2114*48a54d36SAndroid Build Coastguard Worker mDNSlocal void AutomaticBrowseDomainChange(mDNS *const m, DNSQuestion *q, const ResourceRecord *const answer, QC_result AddRecord)
2115*48a54d36SAndroid Build Coastguard Worker {
2116*48a54d36SAndroid Build Coastguard Worker (void)m; // unused;
2117*48a54d36SAndroid Build Coastguard Worker (void)q; // unused
2118*48a54d36SAndroid Build Coastguard Worker
2119*48a54d36SAndroid Build Coastguard Worker LogOperation("AutomaticBrowseDomainChange: %s automatic browse domain %##s",
2120*48a54d36SAndroid Build Coastguard Worker AddRecord ? "Adding" : "Removing", answer->rdata->u.name.c);
2121*48a54d36SAndroid Build Coastguard Worker
2122*48a54d36SAndroid Build Coastguard Worker if (AddRecord) AddAutoBrowseDomain(0, &answer->rdata->u.name);
2123*48a54d36SAndroid Build Coastguard Worker else RmvAutoBrowseDomain(0, &answer->rdata->u.name);
2124*48a54d36SAndroid Build Coastguard Worker }
2125*48a54d36SAndroid Build Coastguard Worker
handle_sethost_request(request_state * request)2126*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_sethost_request(request_state *request)
2127*48a54d36SAndroid Build Coastguard Worker {
2128*48a54d36SAndroid Build Coastguard Worker get_flags(&request->msgptr, request->msgend);
2129*48a54d36SAndroid Build Coastguard Worker char hostName[MAX_DOMAIN_LABEL];
2130*48a54d36SAndroid Build Coastguard Worker int len = 0;
2131*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, hostName,
2132*48a54d36SAndroid Build Coastguard Worker MAX_DOMAIN_LABEL) < 0) return (mStatus_BadParamErr);
2133*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSSetHostname(%X, %d, nonstr ) START",
2134*48a54d36SAndroid Build Coastguard Worker request->sd, request->flags);
2135*48a54d36SAndroid Build Coastguard Worker // if we start using this as a callback for notification when the
2136*48a54d36SAndroid Build Coastguard Worker // hostname changes we may need to cleanup from it
2137*48a54d36SAndroid Build Coastguard Worker // request->terminate = sethost_termination_callback;
2138*48a54d36SAndroid Build Coastguard Worker if(hostName[0] == 0) return mStatus_BadParamErr;
2139*48a54d36SAndroid Build Coastguard Worker while (len < MAX_DOMAIN_LABEL && hostName[len+1]
2140*48a54d36SAndroid Build Coastguard Worker && hostName[len+1] != '.') len++;
2141*48a54d36SAndroid Build Coastguard Worker strncpy(&(mDNSStorage.nicelabel.c[1]), hostName, len);
2142*48a54d36SAndroid Build Coastguard Worker mDNSStorage.nicelabel.c[0] = len;
2143*48a54d36SAndroid Build Coastguard Worker strncpy(&(mDNSStorage.hostlabel.c[1]), hostName, len);
2144*48a54d36SAndroid Build Coastguard Worker mDNSStorage.hostlabel.c[0] = len;
2145*48a54d36SAndroid Build Coastguard Worker mDNS_SetFQDN(&mDNSStorage);
2146*48a54d36SAndroid Build Coastguard Worker return mStatus_NoError;
2147*48a54d36SAndroid Build Coastguard Worker }
2148*48a54d36SAndroid Build Coastguard Worker
handle_browse_request(request_state * request)2149*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_browse_request(request_state *request)
2150*48a54d36SAndroid Build Coastguard Worker {
2151*48a54d36SAndroid Build Coastguard Worker char regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME];
2152*48a54d36SAndroid Build Coastguard Worker domainname typedn, d, temp;
2153*48a54d36SAndroid Build Coastguard Worker mDNSs32 NumSubTypes;
2154*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_NoError;
2155*48a54d36SAndroid Build Coastguard Worker
2156*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2157*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2158*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2159*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2160*48a54d36SAndroid Build Coastguard Worker
2161*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
2162*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) return(mStatus_BadParamErr);
2163*48a54d36SAndroid Build Coastguard Worker
2164*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceBrowse(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2165*48a54d36SAndroid Build Coastguard Worker
2166*48a54d36SAndroid Build Coastguard Worker if (domain[0] == '\0') uDNS_SetupSearchDomains(&mDNSStorage, UDNS_START_WAB_QUERY);
2167*48a54d36SAndroid Build Coastguard Worker
2168*48a54d36SAndroid Build Coastguard Worker request->flags = flags;
2169*48a54d36SAndroid Build Coastguard Worker typedn.c[0] = 0;
2170*48a54d36SAndroid Build Coastguard Worker NumSubTypes = ChopSubTypes(regtype); // Note: Modifies regtype string to remove trailing subtypes
2171*48a54d36SAndroid Build Coastguard Worker if (NumSubTypes < 0 || NumSubTypes > 1) return(mStatus_BadParamErr);
2172*48a54d36SAndroid Build Coastguard Worker if (NumSubTypes == 1 && !AppendDNSNameString(&typedn, regtype + strlen(regtype) + 1)) return(mStatus_BadParamErr);
2173*48a54d36SAndroid Build Coastguard Worker
2174*48a54d36SAndroid Build Coastguard Worker if (!regtype[0] || !AppendDNSNameString(&typedn, regtype)) return(mStatus_BadParamErr);
2175*48a54d36SAndroid Build Coastguard Worker
2176*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&temp, regtype)) return(mStatus_BadParamErr);
2177*48a54d36SAndroid Build Coastguard Worker // For over-long service types, we only allow domain "local"
2178*48a54d36SAndroid Build Coastguard Worker if (temp.c[0] > 15 && domain[0] == 0) mDNSPlatformStrLCopy(domain, "local.", sizeof(domain));
2179*48a54d36SAndroid Build Coastguard Worker
2180*48a54d36SAndroid Build Coastguard Worker // Set up browser info
2181*48a54d36SAndroid Build Coastguard Worker request->u.browser.ForceMCast = (flags & kDNSServiceFlagsForceMulticast) != 0;
2182*48a54d36SAndroid Build Coastguard Worker request->u.browser.interface_id = InterfaceID;
2183*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&request->u.browser.regtype, &typedn);
2184*48a54d36SAndroid Build Coastguard Worker request->u.browser.default_domain = !domain[0];
2185*48a54d36SAndroid Build Coastguard Worker request->u.browser.browsers = NULL;
2186*48a54d36SAndroid Build Coastguard Worker
2187*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceBrowse(%X, %d, \"%##s\", \"%s\") START",
2188*48a54d36SAndroid Build Coastguard Worker request->sd, request->flags, interfaceIndex, request->u.browser.regtype.c, domain);
2189*48a54d36SAndroid Build Coastguard Worker
2190*48a54d36SAndroid Build Coastguard Worker // We need to unconditionally set request->terminate, because even if we didn't successfully
2191*48a54d36SAndroid Build Coastguard Worker // start any browses right now, subsequent configuration changes may cause successful
2192*48a54d36SAndroid Build Coastguard Worker // browses to be added, and we'll need to cancel them before freeing this memory.
2193*48a54d36SAndroid Build Coastguard Worker request->terminate = browse_termination_callback;
2194*48a54d36SAndroid Build Coastguard Worker
2195*48a54d36SAndroid Build Coastguard Worker if (domain[0])
2196*48a54d36SAndroid Build Coastguard Worker {
2197*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&d, domain)) return(mStatus_BadParamErr);
2198*48a54d36SAndroid Build Coastguard Worker err = add_domain_to_browser(request, &d);
2199*48a54d36SAndroid Build Coastguard Worker #if 0
2200*48a54d36SAndroid Build Coastguard Worker err = AuthorizedDomain(request, &d, AutoBrowseDomains) ? add_domain_to_browser(request, &d) : mStatus_NoError;
2201*48a54d36SAndroid Build Coastguard Worker #endif
2202*48a54d36SAndroid Build Coastguard Worker }
2203*48a54d36SAndroid Build Coastguard Worker else
2204*48a54d36SAndroid Build Coastguard Worker {
2205*48a54d36SAndroid Build Coastguard Worker DNameListElem *sdom;
2206*48a54d36SAndroid Build Coastguard Worker for (sdom = AutoBrowseDomains; sdom; sdom = sdom->next)
2207*48a54d36SAndroid Build Coastguard Worker if (!sdom->uid || SystemUID(request->uid) || request->uid == sdom->uid)
2208*48a54d36SAndroid Build Coastguard Worker {
2209*48a54d36SAndroid Build Coastguard Worker err = add_domain_to_browser(request, &sdom->name);
2210*48a54d36SAndroid Build Coastguard Worker if (err)
2211*48a54d36SAndroid Build Coastguard Worker {
2212*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&sdom->name, &localdomain)) break;
2213*48a54d36SAndroid Build Coastguard Worker else err = mStatus_NoError; // suppress errors for non-local "default" domains
2214*48a54d36SAndroid Build Coastguard Worker }
2215*48a54d36SAndroid Build Coastguard Worker }
2216*48a54d36SAndroid Build Coastguard Worker }
2217*48a54d36SAndroid Build Coastguard Worker
2218*48a54d36SAndroid Build Coastguard Worker return(err);
2219*48a54d36SAndroid Build Coastguard Worker }
2220*48a54d36SAndroid Build Coastguard Worker
2221*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2222*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2223*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2224*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceResolve
2225*48a54d36SAndroid Build Coastguard Worker #endif
2226*48a54d36SAndroid Build Coastguard Worker
resolve_result_callback(mDNS * const m,DNSQuestion * question,const ResourceRecord * const answer,QC_result AddRecord)2227*48a54d36SAndroid Build Coastguard Worker mDNSlocal void resolve_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
2228*48a54d36SAndroid Build Coastguard Worker {
2229*48a54d36SAndroid Build Coastguard Worker size_t len = 0;
2230*48a54d36SAndroid Build Coastguard Worker char fullname[MAX_ESCAPED_DOMAIN_NAME], target[MAX_ESCAPED_DOMAIN_NAME];
2231*48a54d36SAndroid Build Coastguard Worker char *data;
2232*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
2233*48a54d36SAndroid Build Coastguard Worker request_state *req = question->QuestionContext;
2234*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
2235*48a54d36SAndroid Build Coastguard Worker
2236*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceResolve(%##s) %s %s", req->sd, question->qname.c, AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer));
2237*48a54d36SAndroid Build Coastguard Worker
2238*48a54d36SAndroid Build Coastguard Worker if (!AddRecord)
2239*48a54d36SAndroid Build Coastguard Worker {
2240*48a54d36SAndroid Build Coastguard Worker if (req->u.resolve.srv == answer) req->u.resolve.srv = mDNSNULL;
2241*48a54d36SAndroid Build Coastguard Worker if (req->u.resolve.txt == answer) req->u.resolve.txt = mDNSNULL;
2242*48a54d36SAndroid Build Coastguard Worker return;
2243*48a54d36SAndroid Build Coastguard Worker }
2244*48a54d36SAndroid Build Coastguard Worker
2245*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype == kDNSType_SRV) req->u.resolve.srv = answer;
2246*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype == kDNSType_TXT) req->u.resolve.txt = answer;
2247*48a54d36SAndroid Build Coastguard Worker
2248*48a54d36SAndroid Build Coastguard Worker if (!req->u.resolve.txt || !req->u.resolve.srv) return; // only deliver result to client if we have both answers
2249*48a54d36SAndroid Build Coastguard Worker
2250*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(answer->name, fullname);
2251*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(&req->u.resolve.srv->rdata->u.srv.target, target);
2252*48a54d36SAndroid Build Coastguard Worker
2253*48a54d36SAndroid Build Coastguard Worker // calculate reply length
2254*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceFlags);
2255*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32); // interface index
2256*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceErrorType);
2257*48a54d36SAndroid Build Coastguard Worker len += strlen(fullname) + 1;
2258*48a54d36SAndroid Build Coastguard Worker len += strlen(target) + 1;
2259*48a54d36SAndroid Build Coastguard Worker len += 2 * sizeof(mDNSu16); // port, txtLen
2260*48a54d36SAndroid Build Coastguard Worker len += req->u.resolve.txt->rdlength;
2261*48a54d36SAndroid Build Coastguard Worker
2262*48a54d36SAndroid Build Coastguard Worker // allocate/init reply header
2263*48a54d36SAndroid Build Coastguard Worker rep = create_reply(resolve_reply_op, len, req);
2264*48a54d36SAndroid Build Coastguard Worker rep->rhdr->flags = dnssd_htonl(0);
2265*48a54d36SAndroid Build Coastguard Worker rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse));
2266*48a54d36SAndroid Build Coastguard Worker rep->rhdr->error = dnssd_htonl(kDNSServiceErr_NoError);
2267*48a54d36SAndroid Build Coastguard Worker
2268*48a54d36SAndroid Build Coastguard Worker data = (char *)&rep->rhdr[1];
2269*48a54d36SAndroid Build Coastguard Worker
2270*48a54d36SAndroid Build Coastguard Worker // write reply data to message
2271*48a54d36SAndroid Build Coastguard Worker put_string(fullname, &data);
2272*48a54d36SAndroid Build Coastguard Worker put_string(target, &data);
2273*48a54d36SAndroid Build Coastguard Worker *data++ = req->u.resolve.srv->rdata->u.srv.port.b[0];
2274*48a54d36SAndroid Build Coastguard Worker *data++ = req->u.resolve.srv->rdata->u.srv.port.b[1];
2275*48a54d36SAndroid Build Coastguard Worker put_uint16(req->u.resolve.txt->rdlength, &data);
2276*48a54d36SAndroid Build Coastguard Worker put_rdata (req->u.resolve.txt->rdlength, req->u.resolve.txt->rdata->u.data, &data);
2277*48a54d36SAndroid Build Coastguard Worker
2278*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceResolve(%s) RESULT %s:%d", req->sd, fullname, target, mDNSVal16(req->u.resolve.srv->rdata->u.srv.port));
2279*48a54d36SAndroid Build Coastguard Worker append_reply(req, rep);
2280*48a54d36SAndroid Build Coastguard Worker }
2281*48a54d36SAndroid Build Coastguard Worker
resolve_termination_callback(request_state * request)2282*48a54d36SAndroid Build Coastguard Worker mDNSlocal void resolve_termination_callback(request_state *request)
2283*48a54d36SAndroid Build Coastguard Worker {
2284*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceResolve(%##s) STOP", request->sd, request->u.resolve.qtxt.qname.c);
2285*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qtxt);
2286*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv);
2287*48a54d36SAndroid Build Coastguard Worker if (request->u.resolve.external_advertise) external_stop_resolving_service(&request->u.resolve.qsrv.qname);
2288*48a54d36SAndroid Build Coastguard Worker }
2289*48a54d36SAndroid Build Coastguard Worker
handle_resolve_request(request_state * request)2290*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_resolve_request(request_state *request)
2291*48a54d36SAndroid Build Coastguard Worker {
2292*48a54d36SAndroid Build Coastguard Worker char name[256], regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME];
2293*48a54d36SAndroid Build Coastguard Worker domainname fqdn;
2294*48a54d36SAndroid Build Coastguard Worker mStatus err;
2295*48a54d36SAndroid Build Coastguard Worker
2296*48a54d36SAndroid Build Coastguard Worker // extract the data from the message
2297*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2298*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2299*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID;
2300*48a54d36SAndroid Build Coastguard Worker mDNSBool wasP2P = (interfaceIndex == kDNSServiceInterfaceIndexP2P);
2301*48a54d36SAndroid Build Coastguard Worker
2302*48a54d36SAndroid Build Coastguard Worker
2303*48a54d36SAndroid Build Coastguard Worker request->flags = flags;
2304*48a54d36SAndroid Build Coastguard Worker if (wasP2P) interfaceIndex = kDNSServiceInterfaceIndexAny;
2305*48a54d36SAndroid Build Coastguard Worker
2306*48a54d36SAndroid Build Coastguard Worker InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2307*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID)
2308*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_resolve_request bad interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); }
2309*48a54d36SAndroid Build Coastguard Worker
2310*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, name, 256) < 0 ||
2311*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
2312*48a54d36SAndroid Build Coastguard Worker get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0)
2313*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_resolve_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); }
2314*48a54d36SAndroid Build Coastguard Worker
2315*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceResolve(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2316*48a54d36SAndroid Build Coastguard Worker
2317*48a54d36SAndroid Build Coastguard Worker if (build_domainname_from_strings(&fqdn, name, regtype, domain) < 0)
2318*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_resolve_request bad “%s” “%s” “%s”", name, regtype, domain); return(mStatus_BadParamErr); }
2319*48a54d36SAndroid Build Coastguard Worker
2320*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(&request->u.resolve, sizeof(request->u.resolve));
2321*48a54d36SAndroid Build Coastguard Worker
2322*48a54d36SAndroid Build Coastguard Worker // format questions
2323*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.InterfaceID = InterfaceID;
2324*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.Target = zeroAddr;
2325*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&request->u.resolve.qsrv.qname, &fqdn);
2326*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.qtype = kDNSType_SRV;
2327*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.qclass = kDNSClass_IN;
2328*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2329*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.ExpectUnique = mDNStrue;
2330*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2331*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2332*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.SuppressUnusable = mDNSfalse;
2333*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.SearchListIndex = 0;
2334*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.AppendSearchDomains = 0;
2335*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.RetryWithSearchDomains = mDNSfalse;
2336*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.TimeoutQuestion = 0;
2337*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.WakeOnResolve = (flags & kDNSServiceFlagsWakeOnResolve) != 0;
2338*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.qnameOrig = mDNSNULL;
2339*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.QuestionCallback = resolve_result_callback;
2340*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qsrv.QuestionContext = request;
2341*48a54d36SAndroid Build Coastguard Worker
2342*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.InterfaceID = InterfaceID;
2343*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.Target = zeroAddr;
2344*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&request->u.resolve.qtxt.qname, &fqdn);
2345*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.qtype = kDNSType_TXT;
2346*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.qclass = kDNSClass_IN;
2347*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2348*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.ExpectUnique = mDNStrue;
2349*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
2350*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
2351*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.SuppressUnusable = mDNSfalse;
2352*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.SearchListIndex = 0;
2353*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.AppendSearchDomains = 0;
2354*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.RetryWithSearchDomains = mDNSfalse;
2355*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.TimeoutQuestion = 0;
2356*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.WakeOnResolve = 0;
2357*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.qnameOrig = mDNSNULL;
2358*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.QuestionCallback = resolve_result_callback;
2359*48a54d36SAndroid Build Coastguard Worker request->u.resolve.qtxt.QuestionContext = request;
2360*48a54d36SAndroid Build Coastguard Worker
2361*48a54d36SAndroid Build Coastguard Worker request->u.resolve.ReportTime = NonZeroTime(mDNS_TimeNow(&mDNSStorage) + 130 * mDNSPlatformOneSecond);
2362*48a54d36SAndroid Build Coastguard Worker
2363*48a54d36SAndroid Build Coastguard Worker request->u.resolve.external_advertise = mDNSfalse;
2364*48a54d36SAndroid Build Coastguard Worker
2365*48a54d36SAndroid Build Coastguard Worker #if 0
2366*48a54d36SAndroid Build Coastguard Worker if (!AuthorizedDomain(request, &fqdn, AutoBrowseDomains)) return(mStatus_NoError);
2367*48a54d36SAndroid Build Coastguard Worker #endif
2368*48a54d36SAndroid Build Coastguard Worker
2369*48a54d36SAndroid Build Coastguard Worker // ask the questions
2370*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceResolve(%##s) START", request->sd, request->u.resolve.qsrv.qname.c);
2371*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qsrv);
2372*48a54d36SAndroid Build Coastguard Worker if (!err)
2373*48a54d36SAndroid Build Coastguard Worker {
2374*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qtxt);
2375*48a54d36SAndroid Build Coastguard Worker if (err) mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv);
2376*48a54d36SAndroid Build Coastguard Worker else
2377*48a54d36SAndroid Build Coastguard Worker {
2378*48a54d36SAndroid Build Coastguard Worker request->terminate = resolve_termination_callback;
2379*48a54d36SAndroid Build Coastguard Worker // If the user explicitly passed in P2P, we don't restrict the domain in which we resolve.
2380*48a54d36SAndroid Build Coastguard Worker if (wasP2P || (!InterfaceID && IsLocalDomain(&fqdn) && (request->flags & kDNSServiceFlagsIncludeP2P)))
2381*48a54d36SAndroid Build Coastguard Worker {
2382*48a54d36SAndroid Build Coastguard Worker request->u.resolve.external_advertise = mDNStrue;
2383*48a54d36SAndroid Build Coastguard Worker LogInfo("handle_resolve_request: calling external_start_resolving_service()");
2384*48a54d36SAndroid Build Coastguard Worker external_start_resolving_service(&fqdn);
2385*48a54d36SAndroid Build Coastguard Worker }
2386*48a54d36SAndroid Build Coastguard Worker }
2387*48a54d36SAndroid Build Coastguard Worker }
2388*48a54d36SAndroid Build Coastguard Worker
2389*48a54d36SAndroid Build Coastguard Worker return(err);
2390*48a54d36SAndroid Build Coastguard Worker }
2391*48a54d36SAndroid Build Coastguard Worker
2392*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
2393*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
2394*48a54d36SAndroid Build Coastguard Worker #pragma mark -
2395*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceQueryRecord
2396*48a54d36SAndroid Build Coastguard Worker #endif
2397*48a54d36SAndroid Build Coastguard Worker
2398*48a54d36SAndroid Build Coastguard Worker // mDNS operation functions. Each operation has 3 associated functions - a request handler that parses
2399*48a54d36SAndroid Build Coastguard Worker // the client's request and makes the appropriate mDNSCore call, a result handler (passed as a callback
2400*48a54d36SAndroid Build Coastguard Worker // to the mDNSCore routine) that sends results back to the client, and a termination routine that aborts
2401*48a54d36SAndroid Build Coastguard Worker // the mDNSCore operation if the client dies or closes its socket.
2402*48a54d36SAndroid Build Coastguard Worker
2403*48a54d36SAndroid Build Coastguard Worker // Returns -1 to tell the caller that it should not try to reissue the query anymore
2404*48a54d36SAndroid Build Coastguard Worker // Returns 1 on successfully appending a search domain and the caller should reissue the new query
2405*48a54d36SAndroid Build Coastguard Worker // Returns 0 when there are no more search domains and the caller should reissue the query
AppendNewSearchDomain(mDNS * const m,DNSQuestion * question)2406*48a54d36SAndroid Build Coastguard Worker mDNSlocal int AppendNewSearchDomain(mDNS *const m, DNSQuestion *question)
2407*48a54d36SAndroid Build Coastguard Worker {
2408*48a54d36SAndroid Build Coastguard Worker domainname *sd;
2409*48a54d36SAndroid Build Coastguard Worker mStatus err;
2410*48a54d36SAndroid Build Coastguard Worker
2411*48a54d36SAndroid Build Coastguard Worker // Sanity check: The caller already checks this. We use -1 to indicate that we have searched all
2412*48a54d36SAndroid Build Coastguard Worker // the domains and should try the single label query directly on the wire.
2413*48a54d36SAndroid Build Coastguard Worker if (question->SearchListIndex == -1)
2414*48a54d36SAndroid Build Coastguard Worker {
2415*48a54d36SAndroid Build Coastguard Worker LogMsg("AppendNewSearchDomain: question %##s (%s) SearchListIndex is -1", question->qname.c, DNSTypeName(question->qtype));
2416*48a54d36SAndroid Build Coastguard Worker return -1;
2417*48a54d36SAndroid Build Coastguard Worker }
2418*48a54d36SAndroid Build Coastguard Worker
2419*48a54d36SAndroid Build Coastguard Worker if (!question->AppendSearchDomains)
2420*48a54d36SAndroid Build Coastguard Worker {
2421*48a54d36SAndroid Build Coastguard Worker LogMsg("AppendNewSearchDomain: question %##s (%s) AppendSearchDoamins is 0", question->qname.c, DNSTypeName(question->qtype));
2422*48a54d36SAndroid Build Coastguard Worker return -1;
2423*48a54d36SAndroid Build Coastguard Worker }
2424*48a54d36SAndroid Build Coastguard Worker
2425*48a54d36SAndroid Build Coastguard Worker // Save the original name, before we modify them below.
2426*48a54d36SAndroid Build Coastguard Worker if (!question->qnameOrig)
2427*48a54d36SAndroid Build Coastguard Worker {
2428*48a54d36SAndroid Build Coastguard Worker question->qnameOrig = mallocL("AppendNewSearchDomain", sizeof(domainname));
2429*48a54d36SAndroid Build Coastguard Worker if (!question->qnameOrig) { LogMsg("AppendNewSearchDomain: ERROR!! malloc failure"); return -1; }
2430*48a54d36SAndroid Build Coastguard Worker question->qnameOrig->c[0] = 0;
2431*48a54d36SAndroid Build Coastguard Worker AssignDomainName(question->qnameOrig, &question->qname);
2432*48a54d36SAndroid Build Coastguard Worker LogInfo("AppendSearchDomain: qnameOrig %##s", question->qnameOrig->c);
2433*48a54d36SAndroid Build Coastguard Worker }
2434*48a54d36SAndroid Build Coastguard Worker
2435*48a54d36SAndroid Build Coastguard Worker sd = uDNS_GetNextSearchDomain(m, question->InterfaceID, &question->SearchListIndex, !question->AppendLocalSearchDomains);
2436*48a54d36SAndroid Build Coastguard Worker // We use -1 to indicate that we have searched all the domains and should try the single label
2437*48a54d36SAndroid Build Coastguard Worker // query directly on the wire. uDNS_GetNextSearchDomain should never return a negative value
2438*48a54d36SAndroid Build Coastguard Worker if (question->SearchListIndex == -1)
2439*48a54d36SAndroid Build Coastguard Worker {
2440*48a54d36SAndroid Build Coastguard Worker LogMsg("AppendNewSearchDomain: ERROR!! uDNS_GetNextSearchDomain returned -1");
2441*48a54d36SAndroid Build Coastguard Worker return -1;
2442*48a54d36SAndroid Build Coastguard Worker }
2443*48a54d36SAndroid Build Coastguard Worker
2444*48a54d36SAndroid Build Coastguard Worker // Not a common case. Perhaps, we should try the next search domain if it exceeds ?
2445*48a54d36SAndroid Build Coastguard Worker if (sd && (DomainNameLength(question->qnameOrig) + DomainNameLength(sd)) > MAX_DOMAIN_NAME)
2446*48a54d36SAndroid Build Coastguard Worker {
2447*48a54d36SAndroid Build Coastguard Worker LogMsg("AppendNewSearchDomain: ERROR!! exceeding max domain length for %##s (%s) SearchDomain %##s length %d, Question name length %d", question->qnameOrig->c, DNSTypeName(question->qtype), sd->c, DomainNameLength(question->qnameOrig), DomainNameLength(sd));
2448*48a54d36SAndroid Build Coastguard Worker return -1;
2449*48a54d36SAndroid Build Coastguard Worker }
2450*48a54d36SAndroid Build Coastguard Worker
2451*48a54d36SAndroid Build Coastguard Worker // if there are no more search domains and we have already tried this question
2452*48a54d36SAndroid Build Coastguard Worker // without appending search domains, then we are done.
2453*48a54d36SAndroid Build Coastguard Worker if (!sd && !ApplySearchDomainsFirst(question))
2454*48a54d36SAndroid Build Coastguard Worker {
2455*48a54d36SAndroid Build Coastguard Worker LogInfo("AppnedNewSearchDomain: No more search domains for question with name %##s (%s), not trying anymore", question->qname.c, DNSTypeName(question->qtype));
2456*48a54d36SAndroid Build Coastguard Worker return -1;
2457*48a54d36SAndroid Build Coastguard Worker }
2458*48a54d36SAndroid Build Coastguard Worker
2459*48a54d36SAndroid Build Coastguard Worker // Stop the question before changing the name as negative cache entries could be pointing at this question.
2460*48a54d36SAndroid Build Coastguard Worker // Even if we don't change the question in the case of returning 0, the caller is going to restart the
2461*48a54d36SAndroid Build Coastguard Worker // question.
2462*48a54d36SAndroid Build Coastguard Worker err = mDNS_StopQuery(&mDNSStorage, question);
2463*48a54d36SAndroid Build Coastguard Worker if (err) { LogMsg("AppendNewSearchDomain: ERROR!! %##s %s mDNS_StopQuery: %d, while retrying with search domains", question->qname.c, DNSTypeName(question->qtype), (int)err); }
2464*48a54d36SAndroid Build Coastguard Worker
2465*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&question->qname, question->qnameOrig);
2466*48a54d36SAndroid Build Coastguard Worker if (sd)
2467*48a54d36SAndroid Build Coastguard Worker {
2468*48a54d36SAndroid Build Coastguard Worker AppendDomainName(&question->qname, sd);
2469*48a54d36SAndroid Build Coastguard Worker LogInfo("AppnedNewSearchDomain: Returning question with name %##s, SearchListIndex %d", question->qname.c, question->SearchListIndex);
2470*48a54d36SAndroid Build Coastguard Worker return 1;
2471*48a54d36SAndroid Build Coastguard Worker }
2472*48a54d36SAndroid Build Coastguard Worker
2473*48a54d36SAndroid Build Coastguard Worker // Try the question as single label
2474*48a54d36SAndroid Build Coastguard Worker LogInfo("AppnedNewSearchDomain: No more search domains for question with name %##s (%s), trying one last time", question->qname.c, DNSTypeName(question->qtype));
2475*48a54d36SAndroid Build Coastguard Worker return 0;
2476*48a54d36SAndroid Build Coastguard Worker }
2477*48a54d36SAndroid Build Coastguard Worker
2478*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
2479*48a54d36SAndroid Build Coastguard Worker
DomainInSearchList(domainname * domain)2480*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSBool DomainInSearchList(domainname *domain)
2481*48a54d36SAndroid Build Coastguard Worker {
2482*48a54d36SAndroid Build Coastguard Worker const SearchListElem *s;
2483*48a54d36SAndroid Build Coastguard Worker for (s=SearchList; s; s=s->next)
2484*48a54d36SAndroid Build Coastguard Worker if (SameDomainName(&s->domain, domain)) return mDNStrue;
2485*48a54d36SAndroid Build Coastguard Worker return mDNSfalse;
2486*48a54d36SAndroid Build Coastguard Worker }
2487*48a54d36SAndroid Build Coastguard Worker
2488*48a54d36SAndroid Build Coastguard Worker // Workaround for networks using Microsoft Active Directory using "local" as a private internal
2489*48a54d36SAndroid Build Coastguard Worker // top-level domain
SendAdditionalQuery(DNSQuestion * q,request_state * request,mStatus err)2490*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus SendAdditionalQuery(DNSQuestion *q, request_state *request, mStatus err)
2491*48a54d36SAndroid Build Coastguard Worker {
2492*48a54d36SAndroid Build Coastguard Worker extern domainname ActiveDirectoryPrimaryDomain;
2493*48a54d36SAndroid Build Coastguard Worker DNSQuestion **question2;
2494*48a54d36SAndroid Build Coastguard Worker #define VALID_MSAD_SRV_TRANSPORT(T) (SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_tcp") || SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_udp"))
2495*48a54d36SAndroid Build Coastguard Worker #define VALID_MSAD_SRV(Q) ((Q)->qtype == kDNSType_SRV && VALID_MSAD_SRV_TRANSPORT(SecondLabel(&(Q)->qname)))
2496*48a54d36SAndroid Build Coastguard Worker
2497*48a54d36SAndroid Build Coastguard Worker question2 = mDNSNULL;
2498*48a54d36SAndroid Build Coastguard Worker if (request->hdr.op == query_request)
2499*48a54d36SAndroid Build Coastguard Worker question2 = &request->u.queryrecord.q2;
2500*48a54d36SAndroid Build Coastguard Worker else if (request->hdr.op == addrinfo_request)
2501*48a54d36SAndroid Build Coastguard Worker {
2502*48a54d36SAndroid Build Coastguard Worker if (q->qtype == kDNSType_A)
2503*48a54d36SAndroid Build Coastguard Worker question2 = &request->u.addrinfo.q42;
2504*48a54d36SAndroid Build Coastguard Worker else if (q->qtype == kDNSType_AAAA)
2505*48a54d36SAndroid Build Coastguard Worker question2 = &request->u.addrinfo.q62;
2506*48a54d36SAndroid Build Coastguard Worker }
2507*48a54d36SAndroid Build Coastguard Worker if (!question2)
2508*48a54d36SAndroid Build Coastguard Worker {
2509*48a54d36SAndroid Build Coastguard Worker LogMsg("SendAdditionalQuery: question2 NULL for %##s (%s)", q->qname.c, DNSTypeName(q->qtype));
2510*48a54d36SAndroid Build Coastguard Worker return mStatus_BadParamErr;
2511*48a54d36SAndroid Build Coastguard Worker }
2512*48a54d36SAndroid Build Coastguard Worker
2513*48a54d36SAndroid Build Coastguard Worker // Sanity check: If we already sent an additonal query, we don't need to send one more.
2514*48a54d36SAndroid Build Coastguard Worker //
2515*48a54d36SAndroid Build Coastguard Worker // 1. When the application calls DNSServiceQueryRecord or DNSServiceGetAddrInfo with a .local name, this function
2516*48a54d36SAndroid Build Coastguard Worker // is called to see whether a unicast query should be sent or not.
2517*48a54d36SAndroid Build Coastguard Worker //
2518*48a54d36SAndroid Build Coastguard Worker // 2. As a result of appending search domains, the question may be end up with a .local suffix even though it
2519*48a54d36SAndroid Build Coastguard Worker // was not a .local name to start with. In that case, queryrecord_result_callback calls this function to
2520*48a54d36SAndroid Build Coastguard Worker // send the additional query.
2521*48a54d36SAndroid Build Coastguard Worker //
2522*48a54d36SAndroid Build Coastguard Worker // Thus, it should not be called more than once.
2523*48a54d36SAndroid Build Coastguard Worker if (*question2)
2524*48a54d36SAndroid Build Coastguard Worker {
2525*48a54d36SAndroid Build Coastguard Worker LogInfo("SendAdditionalQuery: question2 already sent for %##s (%s), no more q2", q->qname.c, DNSTypeName(q->qtype));
2526*48a54d36SAndroid Build Coastguard Worker return err;
2527*48a54d36SAndroid Build Coastguard Worker }
2528*48a54d36SAndroid Build Coastguard Worker
2529*48a54d36SAndroid Build Coastguard Worker if (!q->ForceMCast && SameDomainLabel(LastLabel(&q->qname), (const mDNSu8 *)&localdomain))
2530*48a54d36SAndroid Build Coastguard Worker if (q->qtype == kDNSType_A || q->qtype == kDNSType_AAAA || VALID_MSAD_SRV(q))
2531*48a54d36SAndroid Build Coastguard Worker {
2532*48a54d36SAndroid Build Coastguard Worker DNSQuestion *q2;
2533*48a54d36SAndroid Build Coastguard Worker int labels = CountLabels(&q->qname);
2534*48a54d36SAndroid Build Coastguard Worker q2 = mallocL("DNSQuestion", sizeof(DNSQuestion));
2535*48a54d36SAndroid Build Coastguard Worker if (!q2) FatalError("ERROR: SendAdditionalQuery malloc");
2536*48a54d36SAndroid Build Coastguard Worker *question2 = q2;
2537*48a54d36SAndroid Build Coastguard Worker *q2 = *q;
2538*48a54d36SAndroid Build Coastguard Worker q2->InterfaceID = mDNSInterface_Unicast;
2539*48a54d36SAndroid Build Coastguard Worker q2->ExpectUnique = mDNStrue;
2540*48a54d36SAndroid Build Coastguard Worker // If the query starts as a single label e.g., somehost, and we have search domains with .local,
2541*48a54d36SAndroid Build Coastguard Worker // queryrecord_result_callback calls this function when .local is appended to "somehost".
2542*48a54d36SAndroid Build Coastguard Worker // At that time, the name in "q" is pointing at somehost.local and its qnameOrig pointing at
2543*48a54d36SAndroid Build Coastguard Worker // "somehost". We need to copy that information so that when we retry with a different search
2544*48a54d36SAndroid Build Coastguard Worker // domain e.g., mycompany.local, we get "somehost.mycompany.local".
2545*48a54d36SAndroid Build Coastguard Worker if (q->qnameOrig)
2546*48a54d36SAndroid Build Coastguard Worker {
2547*48a54d36SAndroid Build Coastguard Worker (*question2)->qnameOrig = mallocL("SendAdditionalQuery", DomainNameLength(q->qnameOrig));
2548*48a54d36SAndroid Build Coastguard Worker if (!(*question2)->qnameOrig) { LogMsg("SendAdditionalQuery: ERROR!! malloc failure"); return mStatus_NoMemoryErr; }
2549*48a54d36SAndroid Build Coastguard Worker (*question2)->qnameOrig->c[0] = 0;
2550*48a54d36SAndroid Build Coastguard Worker AssignDomainName((*question2)->qnameOrig, q->qnameOrig);
2551*48a54d36SAndroid Build Coastguard Worker LogInfo("SendAdditionalQuery: qnameOrig %##s", (*question2)->qnameOrig->c);
2552*48a54d36SAndroid Build Coastguard Worker }
2553*48a54d36SAndroid Build Coastguard Worker // For names of the form "<one-or-more-labels>.bar.local." we always do a second unicast query in parallel.
2554*48a54d36SAndroid Build Coastguard Worker // For names of the form "<one-label>.local." it's less clear whether we should do a unicast query.
2555*48a54d36SAndroid Build Coastguard Worker // If the name being queried is exactly the same as the name in the DHCP "domain" option (e.g. the DHCP
2556*48a54d36SAndroid Build Coastguard Worker // "domain" is my-small-company.local, and the user types "my-small-company.local" into their web browser)
2557*48a54d36SAndroid Build Coastguard Worker // then that's a hint that it's worth doing a unicast query. Otherwise, we first check to see if the
2558*48a54d36SAndroid Build Coastguard Worker // site's DNS server claims there's an SOA record for "local", and if so, that's also a hint that queries
2559*48a54d36SAndroid Build Coastguard Worker // for names in the "local" domain will be safely answered privately before they hit the root name servers.
2560*48a54d36SAndroid Build Coastguard Worker // Note that in the "my-small-company.local" example above there will typically be an SOA record for
2561*48a54d36SAndroid Build Coastguard Worker // "my-small-company.local" but *not* for "local", which is why the "local SOA" check would fail in that case.
2562*48a54d36SAndroid Build Coastguard Worker // We need to check against both ActiveDirectoryPrimaryDomain and SearchList. If it matches against either
2563*48a54d36SAndroid Build Coastguard Worker // of those, we don't want do the SOA check for the local
2564*48a54d36SAndroid Build Coastguard Worker if (labels == 2 && !SameDomainName(&q->qname, &ActiveDirectoryPrimaryDomain) && !DomainInSearchList(&q->qname))
2565*48a54d36SAndroid Build Coastguard Worker {
2566*48a54d36SAndroid Build Coastguard Worker AssignDomainName(&q2->qname, &localdomain);
2567*48a54d36SAndroid Build Coastguard Worker q2->qtype = kDNSType_SOA;
2568*48a54d36SAndroid Build Coastguard Worker q2->LongLived = mDNSfalse;
2569*48a54d36SAndroid Build Coastguard Worker q2->ForceMCast = mDNSfalse;
2570*48a54d36SAndroid Build Coastguard Worker q2->ReturnIntermed = mDNStrue;
2571*48a54d36SAndroid Build Coastguard Worker // Don't append search domains for the .local SOA query
2572*48a54d36SAndroid Build Coastguard Worker q2->AppendSearchDomains = 0;
2573*48a54d36SAndroid Build Coastguard Worker q2->AppendLocalSearchDomains = 0;
2574*48a54d36SAndroid Build Coastguard Worker q2->RetryWithSearchDomains = mDNSfalse;
2575*48a54d36SAndroid Build Coastguard Worker q2->SearchListIndex = 0;
2576*48a54d36SAndroid Build Coastguard Worker q2->TimeoutQuestion = 0;
2577*48a54d36SAndroid Build Coastguard Worker }
2578*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast", request->sd, q2->qname.c, DNSTypeName(q2->qtype));
2579*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, q2);
2580*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q2->qname.c, DNSTypeName(q2->qtype), (int)err);
2581*48a54d36SAndroid Build Coastguard Worker }
2582*48a54d36SAndroid Build Coastguard Worker return(err);
2583*48a54d36SAndroid Build Coastguard Worker }
2584*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
2585*48a54d36SAndroid Build Coastguard Worker
2586*48a54d36SAndroid Build Coastguard Worker // This function tries to append a search domain if valid and possible. If so, returns true.
RetryQuestionWithSearchDomains(mDNS * const m,DNSQuestion * question,request_state * req)2587*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSBool RetryQuestionWithSearchDomains(mDNS *const m, DNSQuestion *question, request_state *req)
2588*48a54d36SAndroid Build Coastguard Worker {
2589*48a54d36SAndroid Build Coastguard Worker int result;
2590*48a54d36SAndroid Build Coastguard Worker // RetryWithSearchDomains tells the core to call us back so that we can retry with search domains if there is no
2591*48a54d36SAndroid Build Coastguard Worker // answer in the cache or /etc/hosts. In the first call back from the core, we clear RetryWithSearchDomains so
2592*48a54d36SAndroid Build Coastguard Worker // that we don't get called back repeatedly. If we got an answer from the cache or /etc/hosts, we don't touch
2593*48a54d36SAndroid Build Coastguard Worker // RetryWithSearchDomains which may or may not be set.
2594*48a54d36SAndroid Build Coastguard Worker //
2595*48a54d36SAndroid Build Coastguard Worker // If we get e.g., NXDOMAIN and the query is neither suppressed nor exhausted the domain search list and
2596*48a54d36SAndroid Build Coastguard Worker // is a valid question for appending search domains, retry by appending domains
2597*48a54d36SAndroid Build Coastguard Worker
2598*48a54d36SAndroid Build Coastguard Worker if (!question->SuppressQuery && question->SearchListIndex != -1 && question->AppendSearchDomains)
2599*48a54d36SAndroid Build Coastguard Worker {
2600*48a54d36SAndroid Build Coastguard Worker question->RetryWithSearchDomains = 0;
2601*48a54d36SAndroid Build Coastguard Worker result = AppendNewSearchDomain(m, question);
2602*48a54d36SAndroid Build Coastguard Worker // As long as the result is either zero or 1, we retry the question. If we exahaust the search
2603*48a54d36SAndroid Build Coastguard Worker // domains (result is zero) we try the original query (as it was before appending the search
2604*48a54d36SAndroid Build Coastguard Worker // domains) as such on the wire as a last resort if we have not tried them before. For queries
2605*48a54d36SAndroid Build Coastguard Worker // with more than one label, we have already tried them before appending search domains and
2606*48a54d36SAndroid Build Coastguard Worker // hence don't retry again
2607*48a54d36SAndroid Build Coastguard Worker if (result != -1)
2608*48a54d36SAndroid Build Coastguard Worker {
2609*48a54d36SAndroid Build Coastguard Worker mStatus err;
2610*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(m, question);
2611*48a54d36SAndroid Build Coastguard Worker if (!err)
2612*48a54d36SAndroid Build Coastguard Worker {
2613*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: RetryQuestionWithSearchDomains(%##s, %s), retrying after appending search domain", req->sd, question->qname.c, DNSTypeName(question->qtype));
2614*48a54d36SAndroid Build Coastguard Worker // If the result was zero, it meant that there are no search domains and we just retried the question
2615*48a54d36SAndroid Build Coastguard Worker // as a single label and we should not retry with search domains anymore.
2616*48a54d36SAndroid Build Coastguard Worker if (!result) question->SearchListIndex = -1;
2617*48a54d36SAndroid Build Coastguard Worker return mDNStrue;
2618*48a54d36SAndroid Build Coastguard Worker }
2619*48a54d36SAndroid Build Coastguard Worker else
2620*48a54d36SAndroid Build Coastguard Worker {
2621*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: ERROR: RetryQuestionWithSearchDomains %##s %s mDNS_StartQuery: %d, while retrying with search domains", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err);
2622*48a54d36SAndroid Build Coastguard Worker // We have already stopped the query and could not restart. Reset the appropriate pointers
2623*48a54d36SAndroid Build Coastguard Worker // so that we don't call stop again when the question terminates
2624*48a54d36SAndroid Build Coastguard Worker question->QuestionContext = mDNSNULL;
2625*48a54d36SAndroid Build Coastguard Worker }
2626*48a54d36SAndroid Build Coastguard Worker }
2627*48a54d36SAndroid Build Coastguard Worker }
2628*48a54d36SAndroid Build Coastguard Worker else
2629*48a54d36SAndroid Build Coastguard Worker {
2630*48a54d36SAndroid Build Coastguard Worker LogInfo("%3d: RetryQuestionWithSearchDomains: Not appending search domains - SuppressQuery %d, SearchListIndex %d, AppendSearchDomains %d", req->sd, question->SuppressQuery, question->SearchListIndex, question->AppendSearchDomains);
2631*48a54d36SAndroid Build Coastguard Worker }
2632*48a54d36SAndroid Build Coastguard Worker return mDNSfalse;
2633*48a54d36SAndroid Build Coastguard Worker }
2634*48a54d36SAndroid Build Coastguard Worker
queryrecord_result_callback(mDNS * const m,DNSQuestion * question,const ResourceRecord * const answer,QC_result AddRecord)2635*48a54d36SAndroid Build Coastguard Worker mDNSlocal void queryrecord_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord)
2636*48a54d36SAndroid Build Coastguard Worker {
2637*48a54d36SAndroid Build Coastguard Worker char name[MAX_ESCAPED_DOMAIN_NAME];
2638*48a54d36SAndroid Build Coastguard Worker request_state *req = question->QuestionContext;
2639*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
2640*48a54d36SAndroid Build Coastguard Worker char *data;
2641*48a54d36SAndroid Build Coastguard Worker size_t len;
2642*48a54d36SAndroid Build Coastguard Worker DNSServiceErrorType error = kDNSServiceErr_NoError;
2643*48a54d36SAndroid Build Coastguard Worker DNSQuestion *q = mDNSNULL;
2644*48a54d36SAndroid Build Coastguard Worker
2645*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
2646*48a54d36SAndroid Build Coastguard Worker {
2647*48a54d36SAndroid Build Coastguard Worker // Sanity check: QuestionContext is set to NULL after we stop the question and hence we should not
2648*48a54d36SAndroid Build Coastguard Worker // get any callbacks from the core after this.
2649*48a54d36SAndroid Build Coastguard Worker if (!req)
2650*48a54d36SAndroid Build Coastguard Worker {
2651*48a54d36SAndroid Build Coastguard Worker LogMsg("queryrecord_result_callback: ERROR!! QuestionContext NULL for %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
2652*48a54d36SAndroid Build Coastguard Worker return;
2653*48a54d36SAndroid Build Coastguard Worker }
2654*48a54d36SAndroid Build Coastguard Worker if (req->hdr.op == query_request && question == req->u.queryrecord.q2)
2655*48a54d36SAndroid Build Coastguard Worker q = &req->u.queryrecord.q;
2656*48a54d36SAndroid Build Coastguard Worker else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q42)
2657*48a54d36SAndroid Build Coastguard Worker q = &req->u.addrinfo.q4;
2658*48a54d36SAndroid Build Coastguard Worker else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q62)
2659*48a54d36SAndroid Build Coastguard Worker q = &req->u.addrinfo.q6;
2660*48a54d36SAndroid Build Coastguard Worker
2661*48a54d36SAndroid Build Coastguard Worker if (q && question->qtype != q->qtype && !SameDomainName(&question->qname, &q->qname))
2662*48a54d36SAndroid Build Coastguard Worker {
2663*48a54d36SAndroid Build Coastguard Worker mStatus err;
2664*48a54d36SAndroid Build Coastguard Worker domainname *orig = question->qnameOrig;
2665*48a54d36SAndroid Build Coastguard Worker
2666*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback: Stopping q2 local %##s", question->qname.c);
2667*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(m, question);
2668*48a54d36SAndroid Build Coastguard Worker question->QuestionContext = mDNSNULL;
2669*48a54d36SAndroid Build Coastguard Worker
2670*48a54d36SAndroid Build Coastguard Worker // We got a negative response for the SOA record indicating that .local does not exist.
2671*48a54d36SAndroid Build Coastguard Worker // But we might have other search domains (that does not end in .local) that can be
2672*48a54d36SAndroid Build Coastguard Worker // appended to this question. In that case, we want to retry the question. Otherwise,
2673*48a54d36SAndroid Build Coastguard Worker // we don't want to try this question as unicast.
2674*48a54d36SAndroid Build Coastguard Worker if (answer->RecordType == kDNSRecordTypePacketNegative && !q->AppendSearchDomains)
2675*48a54d36SAndroid Build Coastguard Worker {
2676*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback: question %##s AppendSearchDomains zero", q->qname.c);
2677*48a54d36SAndroid Build Coastguard Worker return;
2678*48a54d36SAndroid Build Coastguard Worker }
2679*48a54d36SAndroid Build Coastguard Worker
2680*48a54d36SAndroid Build Coastguard Worker // If we got a non-negative answer for our "local SOA" test query, start an additional parallel unicast query
2681*48a54d36SAndroid Build Coastguard Worker //
2682*48a54d36SAndroid Build Coastguard Worker // Note: When we copy the original question, we copy everything including the AppendSearchDomains,
2683*48a54d36SAndroid Build Coastguard Worker // RetryWithSearchDomains except for qnameOrig which can be non-NULL if the original question is
2684*48a54d36SAndroid Build Coastguard Worker // e.g., somehost and then we appended e.g., ".local" and retried that question. See comment in
2685*48a54d36SAndroid Build Coastguard Worker // SendAdditionalQuery as to how qnameOrig gets initialized.
2686*48a54d36SAndroid Build Coastguard Worker *question = *q;
2687*48a54d36SAndroid Build Coastguard Worker question->InterfaceID = mDNSInterface_Unicast;
2688*48a54d36SAndroid Build Coastguard Worker question->ExpectUnique = mDNStrue;
2689*48a54d36SAndroid Build Coastguard Worker question->qnameOrig = orig;
2690*48a54d36SAndroid Build Coastguard Worker
2691*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast, context %p", req->sd, question->qname.c, DNSTypeName(question->qtype), question->QuestionContext);
2692*48a54d36SAndroid Build Coastguard Worker
2693*48a54d36SAndroid Build Coastguard Worker // If the original question timed out, its QuestionContext would already be set to NULL and that's what we copied above.
2694*48a54d36SAndroid Build Coastguard Worker // Hence, we need to set it explicitly here.
2695*48a54d36SAndroid Build Coastguard Worker question->QuestionContext = req;
2696*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(m, question);
2697*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("%3d: ERROR: queryrecord_result_callback %##s %s mDNS_StartQuery: %d", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err);
2698*48a54d36SAndroid Build Coastguard Worker
2699*48a54d36SAndroid Build Coastguard Worker // If we got a positive response to local SOA, then try the .local question as unicast
2700*48a54d36SAndroid Build Coastguard Worker if (answer->RecordType != kDNSRecordTypePacketNegative) return;
2701*48a54d36SAndroid Build Coastguard Worker
2702*48a54d36SAndroid Build Coastguard Worker // Fall through and get the next search domain. The question is pointing at .local
2703*48a54d36SAndroid Build Coastguard Worker // and we don't want to try that. Try the next search domain. Don't try with local
2704*48a54d36SAndroid Build Coastguard Worker // search domains for the unicast question anymore.
2705*48a54d36SAndroid Build Coastguard Worker //
2706*48a54d36SAndroid Build Coastguard Worker // Note: we started the question above which will be stopped immediately (never sent on the wire)
2707*48a54d36SAndroid Build Coastguard Worker // before we pick the next search domain below. RetryQuestionWithSearchDomains assumes that the
2708*48a54d36SAndroid Build Coastguard Worker // question has already started.
2709*48a54d36SAndroid Build Coastguard Worker question->AppendLocalSearchDomains = 0;
2710*48a54d36SAndroid Build Coastguard Worker }
2711*48a54d36SAndroid Build Coastguard Worker
2712*48a54d36SAndroid Build Coastguard Worker if (q && AddRecord && (question->InterfaceID == mDNSInterface_Unicast) && !answer->rdlength)
2713*48a54d36SAndroid Build Coastguard Worker {
2714*48a54d36SAndroid Build Coastguard Worker // If we get a negative response to the unicast query that we sent above, retry after appending search domains
2715*48a54d36SAndroid Build Coastguard Worker // Note: We could have appended search domains below (where do it for regular unicast questions) instead of doing it here.
2716*48a54d36SAndroid Build Coastguard Worker // As we ignore negative unicast answers below, we would never reach the code where the search domains are appended.
2717*48a54d36SAndroid Build Coastguard Worker // To keep things simple, we handle unicast ".local" separately here.
2718*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback: Retrying .local question %##s (%s) as unicast after appending search domains", question->qname.c, DNSTypeName(question->qtype));
2719*48a54d36SAndroid Build Coastguard Worker if (RetryQuestionWithSearchDomains(m, question, req))
2720*48a54d36SAndroid Build Coastguard Worker return;
2721*48a54d36SAndroid Build Coastguard Worker if (question->AppendSearchDomains && !question->AppendLocalSearchDomains && IsLocalDomain(&question->qname))
2722*48a54d36SAndroid Build Coastguard Worker {
2723*48a54d36SAndroid Build Coastguard Worker // If "local" is the last search domain, we need to stop the question so that we don't send the "local"
2724*48a54d36SAndroid Build Coastguard Worker // question on the wire as we got a negative response for the local SOA. But, we can't stop the question
2725*48a54d36SAndroid Build Coastguard Worker // yet as we may have to timeout the question (done by the "core") for which we need to leave the question
2726*48a54d36SAndroid Build Coastguard Worker // in the list. We leave it disabled so that it does not hit the wire.
2727*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback: Disabling .local question %##s (%s)", question->qname.c, DNSTypeName(question->qtype));
2728*48a54d36SAndroid Build Coastguard Worker question->ThisQInterval = 0;
2729*48a54d36SAndroid Build Coastguard Worker }
2730*48a54d36SAndroid Build Coastguard Worker }
2731*48a54d36SAndroid Build Coastguard Worker // If we are here it means that either "question" is not "q2" OR we got a positive response for "q2" OR we have no more search
2732*48a54d36SAndroid Build Coastguard Worker // domains to append for "q2". In all cases, fall through and deliver the response
2733*48a54d36SAndroid Build Coastguard Worker }
2734*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
2735*48a54d36SAndroid Build Coastguard Worker
2736*48a54d36SAndroid Build Coastguard Worker if (answer->RecordType == kDNSRecordTypePacketNegative)
2737*48a54d36SAndroid Build Coastguard Worker {
2738*48a54d36SAndroid Build Coastguard Worker // If this question needs to be timed out and we have reached the stop time, mark
2739*48a54d36SAndroid Build Coastguard Worker // the error as timeout. It is possible that we might get a negative response from an
2740*48a54d36SAndroid Build Coastguard Worker // external DNS server at the same time when this question reaches its stop time. We
2741*48a54d36SAndroid Build Coastguard Worker // can't tell the difference as there is no indication in the callback. This should
2742*48a54d36SAndroid Build Coastguard Worker // be okay as we will be timing out this query anyway.
2743*48a54d36SAndroid Build Coastguard Worker mDNS_Lock(m);
2744*48a54d36SAndroid Build Coastguard Worker if (question->TimeoutQuestion)
2745*48a54d36SAndroid Build Coastguard Worker {
2746*48a54d36SAndroid Build Coastguard Worker if ((m->timenow - question->StopTime) >= 0)
2747*48a54d36SAndroid Build Coastguard Worker {
2748*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback:Question %##s (%s) timing out, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID);
2749*48a54d36SAndroid Build Coastguard Worker error = kDNSServiceErr_Timeout;
2750*48a54d36SAndroid Build Coastguard Worker }
2751*48a54d36SAndroid Build Coastguard Worker }
2752*48a54d36SAndroid Build Coastguard Worker mDNS_Unlock(m);
2753*48a54d36SAndroid Build Coastguard Worker // When we're doing parallel unicast and multicast queries for dot-local names (for supporting Microsoft
2754*48a54d36SAndroid Build Coastguard Worker // Active Directory sites) we need to ignore negative unicast answers. Otherwise we'll generate negative
2755*48a54d36SAndroid Build Coastguard Worker // answers for just about every single multicast name we ever look up, since the Microsoft Active Directory
2756*48a54d36SAndroid Build Coastguard Worker // server is going to assert that pretty much every single multicast name doesn't exist.
2757*48a54d36SAndroid Build Coastguard Worker //
2758*48a54d36SAndroid Build Coastguard Worker // If we are timing out this query, we need to deliver the negative answer to the application
2759*48a54d36SAndroid Build Coastguard Worker if (error != kDNSServiceErr_Timeout)
2760*48a54d36SAndroid Build Coastguard Worker {
2761*48a54d36SAndroid Build Coastguard Worker if (!answer->InterfaceID && IsLocalDomain(answer->name))
2762*48a54d36SAndroid Build Coastguard Worker {
2763*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback:Question %##s (%s) answering local with unicast", question->qname.c, DNSTypeName(question->qtype));
2764*48a54d36SAndroid Build Coastguard Worker return;
2765*48a54d36SAndroid Build Coastguard Worker }
2766*48a54d36SAndroid Build Coastguard Worker error = kDNSServiceErr_NoSuchRecord;
2767*48a54d36SAndroid Build Coastguard Worker }
2768*48a54d36SAndroid Build Coastguard Worker AddRecord = mDNStrue;
2769*48a54d36SAndroid Build Coastguard Worker }
2770*48a54d36SAndroid Build Coastguard Worker // If we get a negative answer, try appending search domains. Don't append search domains
2771*48a54d36SAndroid Build Coastguard Worker // - if we are timing out this question
2772*48a54d36SAndroid Build Coastguard Worker // - if the negative response was received as a result of a multicast query
2773*48a54d36SAndroid Build Coastguard Worker // - if this is an additional query (q2), we already appended search domains above (indicated by "!q" below)
2774*48a54d36SAndroid Build Coastguard Worker if (error != kDNSServiceErr_Timeout)
2775*48a54d36SAndroid Build Coastguard Worker {
2776*48a54d36SAndroid Build Coastguard Worker if (!q && !answer->InterfaceID && !answer->rdlength && AddRecord)
2777*48a54d36SAndroid Build Coastguard Worker {
2778*48a54d36SAndroid Build Coastguard Worker // If the original question did not end in .local, we did not send an SOA query
2779*48a54d36SAndroid Build Coastguard Worker // to figure out whether we should send an additional unicast query or not. If we just
2780*48a54d36SAndroid Build Coastguard Worker // appended .local, we need to see if we need to send an additional query. This should
2781*48a54d36SAndroid Build Coastguard Worker // normally happen just once because after we append .local, we ignore all negative
2782*48a54d36SAndroid Build Coastguard Worker // responses for .local above.
2783*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_result_callback: Retrying question %##s (%s) after appending search domains", question->qname.c, DNSTypeName(question->qtype));
2784*48a54d36SAndroid Build Coastguard Worker if (RetryQuestionWithSearchDomains(m, question, req))
2785*48a54d36SAndroid Build Coastguard Worker {
2786*48a54d36SAndroid Build Coastguard Worker // Note: We need to call SendAdditionalQuery every time after appending a search domain as .local could
2787*48a54d36SAndroid Build Coastguard Worker // be anywhere in the search domain list.
2788*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
2789*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_NoError;
2790*48a54d36SAndroid Build Coastguard Worker err = SendAdditionalQuery(question, req, err);
2791*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("queryrecord_result_callback: Sending .local SOA query failed, after appending domains");
2792*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
2793*48a54d36SAndroid Build Coastguard Worker return;
2794*48a54d36SAndroid Build Coastguard Worker }
2795*48a54d36SAndroid Build Coastguard Worker }
2796*48a54d36SAndroid Build Coastguard Worker }
2797*48a54d36SAndroid Build Coastguard Worker
2798*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(answer->name, name);
2799*48a54d36SAndroid Build Coastguard Worker
2800*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: %s(%##s, %s) %s %s", req->sd,
2801*48a54d36SAndroid Build Coastguard Worker req->hdr.op == query_request ? "DNSServiceQueryRecord" : "DNSServiceGetAddrInfo",
2802*48a54d36SAndroid Build Coastguard Worker question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer));
2803*48a54d36SAndroid Build Coastguard Worker
2804*48a54d36SAndroid Build Coastguard Worker len = sizeof(DNSServiceFlags); // calculate reply data length
2805*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32); // interface index
2806*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceErrorType);
2807*48a54d36SAndroid Build Coastguard Worker len += strlen(name) + 1;
2808*48a54d36SAndroid Build Coastguard Worker len += 3 * sizeof(mDNSu16); // type, class, rdlen
2809*48a54d36SAndroid Build Coastguard Worker len += answer->rdlength;
2810*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32); // TTL
2811*48a54d36SAndroid Build Coastguard Worker
2812*48a54d36SAndroid Build Coastguard Worker rep = create_reply(req->hdr.op == query_request ? query_reply_op : addrinfo_reply_op, len, req);
2813*48a54d36SAndroid Build Coastguard Worker
2814*48a54d36SAndroid Build Coastguard Worker rep->rhdr->flags = dnssd_htonl(AddRecord ? kDNSServiceFlagsAdd : 0);
2815*48a54d36SAndroid Build Coastguard Worker // Call mDNSPlatformInterfaceIndexfromInterfaceID, but suppressNetworkChange (last argument). Otherwise, if the
2816*48a54d36SAndroid Build Coastguard Worker // InterfaceID is not valid, then it simulates a "NetworkChanged" which in turn makes questions
2817*48a54d36SAndroid Build Coastguard Worker // to be stopped and started including *this* one. Normally the InterfaceID is valid. But when we
2818*48a54d36SAndroid Build Coastguard Worker // are using the /etc/hosts entries to answer a question, the InterfaceID may not be known to the
2819*48a54d36SAndroid Build Coastguard Worker // mDNS core . Eventually, we should remove the calls to "NetworkChanged" in
2820*48a54d36SAndroid Build Coastguard Worker // mDNSPlatformInterfaceIndexfromInterfaceID when it can't find InterfaceID as ResourceRecords
2821*48a54d36SAndroid Build Coastguard Worker // should not have existed to answer this question if the corresponding interface is not valid.
2822*48a54d36SAndroid Build Coastguard Worker rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNStrue));
2823*48a54d36SAndroid Build Coastguard Worker rep->rhdr->error = dnssd_htonl(error);
2824*48a54d36SAndroid Build Coastguard Worker
2825*48a54d36SAndroid Build Coastguard Worker data = (char *)&rep->rhdr[1];
2826*48a54d36SAndroid Build Coastguard Worker
2827*48a54d36SAndroid Build Coastguard Worker put_string(name, &data);
2828*48a54d36SAndroid Build Coastguard Worker put_uint16(answer->rrtype, &data);
2829*48a54d36SAndroid Build Coastguard Worker put_uint16(answer->rrclass, &data);
2830*48a54d36SAndroid Build Coastguard Worker put_uint16(answer->rdlength, &data);
2831*48a54d36SAndroid Build Coastguard Worker // We need to use putRData here instead of the crude put_rdata function, because the crude put_rdata
2832*48a54d36SAndroid Build Coastguard Worker // function just does a blind memory copy without regard to structures that may have holes in them.
2833*48a54d36SAndroid Build Coastguard Worker if (answer->rdlength)
2834*48a54d36SAndroid Build Coastguard Worker if (!putRData(mDNSNULL, (mDNSu8 *)data, (mDNSu8 *)rep->rhdr + len, answer))
2835*48a54d36SAndroid Build Coastguard Worker LogMsg("queryrecord_result_callback putRData failed %d", (mDNSu8 *)rep->rhdr + len - (mDNSu8 *)data);
2836*48a54d36SAndroid Build Coastguard Worker data += answer->rdlength;
2837*48a54d36SAndroid Build Coastguard Worker put_uint32(AddRecord ? answer->rroriginalttl : 0, &data);
2838*48a54d36SAndroid Build Coastguard Worker
2839*48a54d36SAndroid Build Coastguard Worker append_reply(req, rep);
2840*48a54d36SAndroid Build Coastguard Worker // Stop the question, if we just timed out
2841*48a54d36SAndroid Build Coastguard Worker if (error == kDNSServiceErr_Timeout)
2842*48a54d36SAndroid Build Coastguard Worker {
2843*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(m, question);
2844*48a54d36SAndroid Build Coastguard Worker // Reset the pointers so that we don't call stop on termination
2845*48a54d36SAndroid Build Coastguard Worker question->QuestionContext = mDNSNULL;
2846*48a54d36SAndroid Build Coastguard Worker }
2847*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
2848*48a54d36SAndroid Build Coastguard Worker #if ! NO_WCF
2849*48a54d36SAndroid Build Coastguard Worker CHECK_WCF_FUNCTION(WCFIsServerRunning)
2850*48a54d36SAndroid Build Coastguard Worker {
2851*48a54d36SAndroid Build Coastguard Worker struct xucred x;
2852*48a54d36SAndroid Build Coastguard Worker socklen_t xucredlen = sizeof(x);
2853*48a54d36SAndroid Build Coastguard Worker
2854*48a54d36SAndroid Build Coastguard Worker if (WCFIsServerRunning((WCFConnection *)m->WCF) && answer->rdlength != 0)
2855*48a54d36SAndroid Build Coastguard Worker {
2856*48a54d36SAndroid Build Coastguard Worker if (getsockopt(req->sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 &&
2857*48a54d36SAndroid Build Coastguard Worker (x.cr_version == XUCRED_VERSION))
2858*48a54d36SAndroid Build Coastguard Worker {
2859*48a54d36SAndroid Build Coastguard Worker struct sockaddr_storage addr;
2860*48a54d36SAndroid Build Coastguard Worker const RDataBody2 *const rdb = (RDataBody2 *)answer->rdata->u.data;
2861*48a54d36SAndroid Build Coastguard Worker addr.ss_len = 0;
2862*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype == kDNSType_A || answer->rrtype == kDNSType_AAAA)
2863*48a54d36SAndroid Build Coastguard Worker {
2864*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype == kDNSType_A)
2865*48a54d36SAndroid Build Coastguard Worker {
2866*48a54d36SAndroid Build Coastguard Worker struct sockaddr_in *sin = (struct sockaddr_in *)&addr;
2867*48a54d36SAndroid Build Coastguard Worker sin->sin_port = 0;
2868*48a54d36SAndroid Build Coastguard Worker if (!putRData(mDNSNULL, (mDNSu8 *)&sin->sin_addr, (mDNSu8 *)(&sin->sin_addr + sizeof(rdb->ipv4)), answer))
2869*48a54d36SAndroid Build Coastguard Worker LogMsg("queryrecord_result_callback: WCF AF_INET putRData failed");
2870*48a54d36SAndroid Build Coastguard Worker else
2871*48a54d36SAndroid Build Coastguard Worker {
2872*48a54d36SAndroid Build Coastguard Worker addr.ss_len = sizeof (struct sockaddr_in);
2873*48a54d36SAndroid Build Coastguard Worker addr.ss_family = AF_INET;
2874*48a54d36SAndroid Build Coastguard Worker }
2875*48a54d36SAndroid Build Coastguard Worker }
2876*48a54d36SAndroid Build Coastguard Worker else if (answer->rrtype == kDNSType_AAAA)
2877*48a54d36SAndroid Build Coastguard Worker {
2878*48a54d36SAndroid Build Coastguard Worker struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&addr;
2879*48a54d36SAndroid Build Coastguard Worker sin6->sin6_port = 0;
2880*48a54d36SAndroid Build Coastguard Worker if (!putRData(mDNSNULL, (mDNSu8 *)&sin6->sin6_addr, (mDNSu8 *)(&sin6->sin6_addr + sizeof(rdb->ipv6)), answer))
2881*48a54d36SAndroid Build Coastguard Worker LogMsg("queryrecord_result_callback: WCF AF_INET6 putRData failed");
2882*48a54d36SAndroid Build Coastguard Worker else
2883*48a54d36SAndroid Build Coastguard Worker {
2884*48a54d36SAndroid Build Coastguard Worker addr.ss_len = sizeof (struct sockaddr_in6);
2885*48a54d36SAndroid Build Coastguard Worker addr.ss_family = AF_INET6;
2886*48a54d36SAndroid Build Coastguard Worker }
2887*48a54d36SAndroid Build Coastguard Worker }
2888*48a54d36SAndroid Build Coastguard Worker if (addr.ss_len)
2889*48a54d36SAndroid Build Coastguard Worker {
2890*48a54d36SAndroid Build Coastguard Worker debugf("queryrecord_result_callback: Name %s, uid %u, addr length %d", name, x.cr_uid, addr.ss_len);
2891*48a54d36SAndroid Build Coastguard Worker CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr)
2892*48a54d36SAndroid Build Coastguard Worker {
2893*48a54d36SAndroid Build Coastguard Worker WCFNameResolvesToAddr(m->WCF, name, (struct sockaddr *)&addr, x.cr_uid);
2894*48a54d36SAndroid Build Coastguard Worker }
2895*48a54d36SAndroid Build Coastguard Worker }
2896*48a54d36SAndroid Build Coastguard Worker }
2897*48a54d36SAndroid Build Coastguard Worker else if (answer->rrtype == kDNSType_CNAME)
2898*48a54d36SAndroid Build Coastguard Worker {
2899*48a54d36SAndroid Build Coastguard Worker domainname cname;
2900*48a54d36SAndroid Build Coastguard Worker char cname_cstr[MAX_ESCAPED_DOMAIN_NAME];
2901*48a54d36SAndroid Build Coastguard Worker if (!putRData(mDNSNULL, cname.c, (mDNSu8 *)(cname.c + MAX_DOMAIN_NAME), answer))
2902*48a54d36SAndroid Build Coastguard Worker LogMsg("queryrecord_result_callback: WCF CNAME putRData failed");
2903*48a54d36SAndroid Build Coastguard Worker else
2904*48a54d36SAndroid Build Coastguard Worker {
2905*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(&cname, cname_cstr);
2906*48a54d36SAndroid Build Coastguard Worker CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr)
2907*48a54d36SAndroid Build Coastguard Worker {
2908*48a54d36SAndroid Build Coastguard Worker WCFNameResolvesToName(m->WCF, name, cname_cstr, x.cr_uid);
2909*48a54d36SAndroid Build Coastguard Worker }
2910*48a54d36SAndroid Build Coastguard Worker }
2911*48a54d36SAndroid Build Coastguard Worker }
2912*48a54d36SAndroid Build Coastguard Worker }
2913*48a54d36SAndroid Build Coastguard Worker else my_perror("queryrecord_result_callback: ERROR: getsockopt LOCAL_PEERCRED");
2914*48a54d36SAndroid Build Coastguard Worker }
2915*48a54d36SAndroid Build Coastguard Worker }
2916*48a54d36SAndroid Build Coastguard Worker #endif
2917*48a54d36SAndroid Build Coastguard Worker #endif
2918*48a54d36SAndroid Build Coastguard Worker }
2919*48a54d36SAndroid Build Coastguard Worker
queryrecord_termination_callback(request_state * request)2920*48a54d36SAndroid Build Coastguard Worker mDNSlocal void queryrecord_termination_callback(request_state *request)
2921*48a54d36SAndroid Build Coastguard Worker {
2922*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) STOP",
2923*48a54d36SAndroid Build Coastguard Worker request->sd, request->u.queryrecord.q.qname.c, DNSTypeName(request->u.queryrecord.q.qtype));
2924*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q.QuestionContext)
2925*48a54d36SAndroid Build Coastguard Worker {
2926*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.queryrecord.q); // no need to error check
2927*48a54d36SAndroid Build Coastguard Worker request->u.queryrecord.q.QuestionContext = mDNSNULL;
2928*48a54d36SAndroid Build Coastguard Worker }
2929*48a54d36SAndroid Build Coastguard Worker else
2930*48a54d36SAndroid Build Coastguard Worker {
2931*48a54d36SAndroid Build Coastguard Worker DNSQuestion *question = &request->u.queryrecord.q;
2932*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_termination_callback: question %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID);
2933*48a54d36SAndroid Build Coastguard Worker }
2934*48a54d36SAndroid Build Coastguard Worker
2935*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q.qnameOrig)
2936*48a54d36SAndroid Build Coastguard Worker {
2937*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination", request->u.queryrecord.q.qnameOrig);
2938*48a54d36SAndroid Build Coastguard Worker request->u.queryrecord.q.qnameOrig = mDNSNULL;
2939*48a54d36SAndroid Build Coastguard Worker }
2940*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q.InterfaceID == mDNSInterface_P2P || (!request->u.queryrecord.q.InterfaceID && SameDomainName((const domainname *)LastLabel(&request->u.queryrecord.q.qname), &localdomain) && (request->flags & kDNSServiceFlagsIncludeP2P)))
2941*48a54d36SAndroid Build Coastguard Worker {
2942*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_termination_callback: calling external_stop_browsing_for_service()");
2943*48a54d36SAndroid Build Coastguard Worker external_stop_browsing_for_service(&mDNSStorage, &request->u.queryrecord.q.qname, request->u.queryrecord.q.qtype);
2944*48a54d36SAndroid Build Coastguard Worker }
2945*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q2)
2946*48a54d36SAndroid Build Coastguard Worker {
2947*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q2->QuestionContext)
2948*48a54d36SAndroid Build Coastguard Worker {
2949*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_termination_callback: Stopping q2 %##s", request->u.queryrecord.q2->qname.c);
2950*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, request->u.queryrecord.q2);
2951*48a54d36SAndroid Build Coastguard Worker }
2952*48a54d36SAndroid Build Coastguard Worker else
2953*48a54d36SAndroid Build Coastguard Worker {
2954*48a54d36SAndroid Build Coastguard Worker DNSQuestion *question = request->u.queryrecord.q2;
2955*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_termination_callback: q2 %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID);
2956*48a54d36SAndroid Build Coastguard Worker }
2957*48a54d36SAndroid Build Coastguard Worker if (request->u.queryrecord.q2->qnameOrig)
2958*48a54d36SAndroid Build Coastguard Worker {
2959*48a54d36SAndroid Build Coastguard Worker LogInfo("queryrecord_termination_callback: freeing q2 qnameOrig %##s", request->u.queryrecord.q2->qnameOrig->c);
2960*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination q2", request->u.queryrecord.q2->qnameOrig);
2961*48a54d36SAndroid Build Coastguard Worker request->u.queryrecord.q2->qnameOrig = mDNSNULL;
2962*48a54d36SAndroid Build Coastguard Worker }
2963*48a54d36SAndroid Build Coastguard Worker freeL("queryrecord Q2", request->u.queryrecord.q2);
2964*48a54d36SAndroid Build Coastguard Worker request->u.queryrecord.q2 = mDNSNULL;
2965*48a54d36SAndroid Build Coastguard Worker }
2966*48a54d36SAndroid Build Coastguard Worker }
2967*48a54d36SAndroid Build Coastguard Worker
handle_queryrecord_request(request_state * request)2968*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_queryrecord_request(request_state *request)
2969*48a54d36SAndroid Build Coastguard Worker {
2970*48a54d36SAndroid Build Coastguard Worker DNSQuestion *const q = &request->u.queryrecord.q;
2971*48a54d36SAndroid Build Coastguard Worker char name[256];
2972*48a54d36SAndroid Build Coastguard Worker mDNSu16 rrtype, rrclass;
2973*48a54d36SAndroid Build Coastguard Worker mStatus err;
2974*48a54d36SAndroid Build Coastguard Worker
2975*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
2976*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
2977*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
2978*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
2979*48a54d36SAndroid Build Coastguard Worker
2980*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, name, 256) < 0) return(mStatus_BadParamErr);
2981*48a54d36SAndroid Build Coastguard Worker rrtype = get_uint16(&request->msgptr, request->msgend);
2982*48a54d36SAndroid Build Coastguard Worker rrclass = get_uint16(&request->msgptr, request->msgend);
2983*48a54d36SAndroid Build Coastguard Worker
2984*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr)
2985*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceQueryRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
2986*48a54d36SAndroid Build Coastguard Worker
2987*48a54d36SAndroid Build Coastguard Worker request->flags = flags;
2988*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(&request->u.queryrecord, sizeof(request->u.queryrecord));
2989*48a54d36SAndroid Build Coastguard Worker
2990*48a54d36SAndroid Build Coastguard Worker q->InterfaceID = InterfaceID;
2991*48a54d36SAndroid Build Coastguard Worker q->Target = zeroAddr;
2992*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&q->qname, name)) return(mStatus_BadParamErr);
2993*48a54d36SAndroid Build Coastguard Worker #if 0
2994*48a54d36SAndroid Build Coastguard Worker if (!AuthorizedDomain(request, &q->qname, AutoBrowseDomains)) return (mStatus_NoError);
2995*48a54d36SAndroid Build Coastguard Worker #endif
2996*48a54d36SAndroid Build Coastguard Worker q->qtype = rrtype;
2997*48a54d36SAndroid Build Coastguard Worker q->qclass = rrclass;
2998*48a54d36SAndroid Build Coastguard Worker q->LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
2999*48a54d36SAndroid Build Coastguard Worker q->ExpectUnique = mDNSfalse;
3000*48a54d36SAndroid Build Coastguard Worker q->ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
3001*48a54d36SAndroid Build Coastguard Worker q->ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
3002*48a54d36SAndroid Build Coastguard Worker q->SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0;
3003*48a54d36SAndroid Build Coastguard Worker q->TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0;
3004*48a54d36SAndroid Build Coastguard Worker q->WakeOnResolve = 0;
3005*48a54d36SAndroid Build Coastguard Worker q->QuestionCallback = queryrecord_result_callback;
3006*48a54d36SAndroid Build Coastguard Worker q->QuestionContext = request;
3007*48a54d36SAndroid Build Coastguard Worker q->SearchListIndex = 0;
3008*48a54d36SAndroid Build Coastguard Worker
3009*48a54d36SAndroid Build Coastguard Worker // Don't append search domains for fully qualified domain names including queries
3010*48a54d36SAndroid Build Coastguard Worker // such as e.g., "abc." that has only one label. We convert all names to FQDNs as internally
3011*48a54d36SAndroid Build Coastguard Worker // we only deal with FQDNs. Hence, we cannot look at qname to figure out whether we should
3012*48a54d36SAndroid Build Coastguard Worker // append search domains or not. So, we record that information in AppendSearchDomains.
3013*48a54d36SAndroid Build Coastguard Worker //
3014*48a54d36SAndroid Build Coastguard Worker // We append search domains only for queries that are a single label. If overriden using
3015*48a54d36SAndroid Build Coastguard Worker // command line argument "AlwaysAppendSearchDomains", then we do it for any query which
3016*48a54d36SAndroid Build Coastguard Worker // is not fully qualified.
3017*48a54d36SAndroid Build Coastguard Worker
3018*48a54d36SAndroid Build Coastguard Worker if ((rrtype == kDNSType_A || rrtype == kDNSType_AAAA) && name[strlen(name) - 1] != '.' &&
3019*48a54d36SAndroid Build Coastguard Worker (AlwaysAppendSearchDomains || CountLabels(&q->qname) == 1))
3020*48a54d36SAndroid Build Coastguard Worker {
3021*48a54d36SAndroid Build Coastguard Worker q->AppendSearchDomains = 1;
3022*48a54d36SAndroid Build Coastguard Worker q->AppendLocalSearchDomains = 1;
3023*48a54d36SAndroid Build Coastguard Worker }
3024*48a54d36SAndroid Build Coastguard Worker else
3025*48a54d36SAndroid Build Coastguard Worker {
3026*48a54d36SAndroid Build Coastguard Worker q->AppendSearchDomains = 0;
3027*48a54d36SAndroid Build Coastguard Worker q->AppendLocalSearchDomains = 0;
3028*48a54d36SAndroid Build Coastguard Worker }
3029*48a54d36SAndroid Build Coastguard Worker
3030*48a54d36SAndroid Build Coastguard Worker // For single label queries that are not fully qualified, look at /etc/hosts, cache and try
3031*48a54d36SAndroid Build Coastguard Worker // search domains before trying them on the wire as a single label query. RetryWithSearchDomains
3032*48a54d36SAndroid Build Coastguard Worker // tell the core to call back into the UDS layer if there is no valid response in /etc/hosts or
3033*48a54d36SAndroid Build Coastguard Worker // the cache
3034*48a54d36SAndroid Build Coastguard Worker q->RetryWithSearchDomains = ApplySearchDomainsFirst(q) ? 1 : 0;
3035*48a54d36SAndroid Build Coastguard Worker q->qnameOrig = mDNSNULL;
3036*48a54d36SAndroid Build Coastguard Worker
3037*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceQueryRecord(%X, %d, %##s, %s) START", request->sd, flags, interfaceIndex, q->qname.c, DNSTypeName(q->qtype));
3038*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, q);
3039*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q->qname.c, DNSTypeName(q->qtype), (int)err);
3040*48a54d36SAndroid Build Coastguard Worker else
3041*48a54d36SAndroid Build Coastguard Worker {
3042*48a54d36SAndroid Build Coastguard Worker request->terminate = queryrecord_termination_callback;
3043*48a54d36SAndroid Build Coastguard Worker if (q->InterfaceID == mDNSInterface_P2P || (!q->InterfaceID && SameDomainName((const domainname *)LastLabel(&q->qname), &localdomain) && (flags & kDNSServiceFlagsIncludeP2P)))
3044*48a54d36SAndroid Build Coastguard Worker {
3045*48a54d36SAndroid Build Coastguard Worker LogInfo("handle_queryrecord_request: calling external_start_browsing_for_service()");
3046*48a54d36SAndroid Build Coastguard Worker external_start_browsing_for_service(&mDNSStorage, &q->qname, q->qtype);
3047*48a54d36SAndroid Build Coastguard Worker }
3048*48a54d36SAndroid Build Coastguard Worker }
3049*48a54d36SAndroid Build Coastguard Worker
3050*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
3051*48a54d36SAndroid Build Coastguard Worker err = SendAdditionalQuery(q, request, err);
3052*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
3053*48a54d36SAndroid Build Coastguard Worker
3054*48a54d36SAndroid Build Coastguard Worker return(err);
3055*48a54d36SAndroid Build Coastguard Worker }
3056*48a54d36SAndroid Build Coastguard Worker
3057*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
3058*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
3059*48a54d36SAndroid Build Coastguard Worker #pragma mark -
3060*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceEnumerateDomains
3061*48a54d36SAndroid Build Coastguard Worker #endif
3062*48a54d36SAndroid Build Coastguard Worker
format_enumeration_reply(request_state * request,const char * domain,DNSServiceFlags flags,mDNSu32 ifi,DNSServiceErrorType err)3063*48a54d36SAndroid Build Coastguard Worker mDNSlocal reply_state *format_enumeration_reply(request_state *request,
3064*48a54d36SAndroid Build Coastguard Worker const char *domain, DNSServiceFlags flags, mDNSu32 ifi, DNSServiceErrorType err)
3065*48a54d36SAndroid Build Coastguard Worker {
3066*48a54d36SAndroid Build Coastguard Worker size_t len;
3067*48a54d36SAndroid Build Coastguard Worker reply_state *reply;
3068*48a54d36SAndroid Build Coastguard Worker char *data;
3069*48a54d36SAndroid Build Coastguard Worker
3070*48a54d36SAndroid Build Coastguard Worker len = sizeof(DNSServiceFlags);
3071*48a54d36SAndroid Build Coastguard Worker len += sizeof(mDNSu32);
3072*48a54d36SAndroid Build Coastguard Worker len += sizeof(DNSServiceErrorType);
3073*48a54d36SAndroid Build Coastguard Worker len += strlen(domain) + 1;
3074*48a54d36SAndroid Build Coastguard Worker
3075*48a54d36SAndroid Build Coastguard Worker reply = create_reply(enumeration_reply_op, len, request);
3076*48a54d36SAndroid Build Coastguard Worker reply->rhdr->flags = dnssd_htonl(flags);
3077*48a54d36SAndroid Build Coastguard Worker reply->rhdr->ifi = dnssd_htonl(ifi);
3078*48a54d36SAndroid Build Coastguard Worker reply->rhdr->error = dnssd_htonl(err);
3079*48a54d36SAndroid Build Coastguard Worker data = (char *)&reply->rhdr[1];
3080*48a54d36SAndroid Build Coastguard Worker put_string(domain, &data);
3081*48a54d36SAndroid Build Coastguard Worker return reply;
3082*48a54d36SAndroid Build Coastguard Worker }
3083*48a54d36SAndroid Build Coastguard Worker
enum_termination_callback(request_state * request)3084*48a54d36SAndroid Build Coastguard Worker mDNSlocal void enum_termination_callback(request_state *request)
3085*48a54d36SAndroid Build Coastguard Worker {
3086*48a54d36SAndroid Build Coastguard Worker mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all);
3087*48a54d36SAndroid Build Coastguard Worker mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_default);
3088*48a54d36SAndroid Build Coastguard Worker }
3089*48a54d36SAndroid Build Coastguard Worker
enum_result_callback(mDNS * const m,DNSQuestion * const question,const ResourceRecord * const answer,QC_result AddRecord)3090*48a54d36SAndroid Build Coastguard Worker mDNSlocal void enum_result_callback(mDNS *const m,
3091*48a54d36SAndroid Build Coastguard Worker DNSQuestion *const question, const ResourceRecord *const answer, QC_result AddRecord)
3092*48a54d36SAndroid Build Coastguard Worker {
3093*48a54d36SAndroid Build Coastguard Worker char domain[MAX_ESCAPED_DOMAIN_NAME];
3094*48a54d36SAndroid Build Coastguard Worker request_state *request = question->QuestionContext;
3095*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = 0;
3096*48a54d36SAndroid Build Coastguard Worker reply_state *reply;
3097*48a54d36SAndroid Build Coastguard Worker (void)m; // Unused
3098*48a54d36SAndroid Build Coastguard Worker
3099*48a54d36SAndroid Build Coastguard Worker if (answer->rrtype != kDNSType_PTR) return;
3100*48a54d36SAndroid Build Coastguard Worker
3101*48a54d36SAndroid Build Coastguard Worker #if 0
3102*48a54d36SAndroid Build Coastguard Worker if (!AuthorizedDomain(request, &answer->rdata->u.name, request->u.enumeration.flags ? AutoRegistrationDomains : AutoBrowseDomains)) return;
3103*48a54d36SAndroid Build Coastguard Worker #endif
3104*48a54d36SAndroid Build Coastguard Worker
3105*48a54d36SAndroid Build Coastguard Worker // We only return add/remove events for the browse and registration lists
3106*48a54d36SAndroid Build Coastguard Worker // For the default browse and registration answers, we only give an "ADD" event
3107*48a54d36SAndroid Build Coastguard Worker if (question == &request->u.enumeration.q_default && !AddRecord) return;
3108*48a54d36SAndroid Build Coastguard Worker
3109*48a54d36SAndroid Build Coastguard Worker if (AddRecord)
3110*48a54d36SAndroid Build Coastguard Worker {
3111*48a54d36SAndroid Build Coastguard Worker flags |= kDNSServiceFlagsAdd;
3112*48a54d36SAndroid Build Coastguard Worker if (question == &request->u.enumeration.q_default) flags |= kDNSServiceFlagsDefault;
3113*48a54d36SAndroid Build Coastguard Worker }
3114*48a54d36SAndroid Build Coastguard Worker
3115*48a54d36SAndroid Build Coastguard Worker ConvertDomainNameToCString(&answer->rdata->u.name, domain);
3116*48a54d36SAndroid Build Coastguard Worker // Note that we do NOT propagate specific interface indexes to the client - for example, a domain we learn from
3117*48a54d36SAndroid Build Coastguard Worker // a machine's system preferences may be discovered on the LocalOnly interface, but should be browsed on the
3118*48a54d36SAndroid Build Coastguard Worker // network, so we just pass kDNSServiceInterfaceIndexAny
3119*48a54d36SAndroid Build Coastguard Worker reply = format_enumeration_reply(request, domain, flags, kDNSServiceInterfaceIndexAny, kDNSServiceErr_NoError);
3120*48a54d36SAndroid Build Coastguard Worker if (!reply) { LogMsg("ERROR: enum_result_callback, format_enumeration_reply"); return; }
3121*48a54d36SAndroid Build Coastguard Worker
3122*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceEnumerateDomains(%#2s) RESULT %s: %s", request->sd, question->qname.c, AddRecord ? "Add" : "Rmv", domain);
3123*48a54d36SAndroid Build Coastguard Worker
3124*48a54d36SAndroid Build Coastguard Worker append_reply(request, reply);
3125*48a54d36SAndroid Build Coastguard Worker }
3126*48a54d36SAndroid Build Coastguard Worker
handle_enum_request(request_state * request)3127*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_enum_request(request_state *request)
3128*48a54d36SAndroid Build Coastguard Worker {
3129*48a54d36SAndroid Build Coastguard Worker mStatus err;
3130*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
3131*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags reg = flags & kDNSServiceFlagsRegistrationDomains;
3132*48a54d36SAndroid Build Coastguard Worker mDNS_DomainType t_all = reg ? mDNS_DomainTypeRegistration : mDNS_DomainTypeBrowse;
3133*48a54d36SAndroid Build Coastguard Worker mDNS_DomainType t_default = reg ? mDNS_DomainTypeRegistrationDefault : mDNS_DomainTypeBrowseDefault;
3134*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
3135*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
3136*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
3137*48a54d36SAndroid Build Coastguard Worker
3138*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr)
3139*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceEnumerateDomains(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
3140*48a54d36SAndroid Build Coastguard Worker
3141*48a54d36SAndroid Build Coastguard Worker // allocate context structures
3142*48a54d36SAndroid Build Coastguard Worker uDNS_SetupSearchDomains(&mDNSStorage, UDNS_START_WAB_QUERY);
3143*48a54d36SAndroid Build Coastguard Worker
3144*48a54d36SAndroid Build Coastguard Worker #if 0
3145*48a54d36SAndroid Build Coastguard Worker // mark which kind of enumeration we're doing so we can (de)authorize certain domains
3146*48a54d36SAndroid Build Coastguard Worker request->u.enumeration.flags = reg;
3147*48a54d36SAndroid Build Coastguard Worker #endif
3148*48a54d36SAndroid Build Coastguard Worker
3149*48a54d36SAndroid Build Coastguard Worker // enumeration requires multiple questions, so we must link all the context pointers so that
3150*48a54d36SAndroid Build Coastguard Worker // necessary context can be reached from the callbacks
3151*48a54d36SAndroid Build Coastguard Worker request->u.enumeration.q_all .QuestionContext = request;
3152*48a54d36SAndroid Build Coastguard Worker request->u.enumeration.q_default.QuestionContext = request;
3153*48a54d36SAndroid Build Coastguard Worker
3154*48a54d36SAndroid Build Coastguard Worker // if the caller hasn't specified an explicit interface, we use local-only to get the system-wide list.
3155*48a54d36SAndroid Build Coastguard Worker if (!InterfaceID) InterfaceID = mDNSInterface_LocalOnly;
3156*48a54d36SAndroid Build Coastguard Worker
3157*48a54d36SAndroid Build Coastguard Worker // make the calls
3158*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceEnumerateDomains(%X=%s)", request->sd, flags,
3159*48a54d36SAndroid Build Coastguard Worker (flags & kDNSServiceFlagsBrowseDomains ) ? "kDNSServiceFlagsBrowseDomains" :
3160*48a54d36SAndroid Build Coastguard Worker (flags & kDNSServiceFlagsRegistrationDomains) ? "kDNSServiceFlagsRegistrationDomains" : "<<Unknown>>");
3161*48a54d36SAndroid Build Coastguard Worker err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_all, t_all, NULL, InterfaceID, enum_result_callback, request);
3162*48a54d36SAndroid Build Coastguard Worker if (!err)
3163*48a54d36SAndroid Build Coastguard Worker {
3164*48a54d36SAndroid Build Coastguard Worker err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_default, t_default, NULL, InterfaceID, enum_result_callback, request);
3165*48a54d36SAndroid Build Coastguard Worker if (err) mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all);
3166*48a54d36SAndroid Build Coastguard Worker else request->terminate = enum_termination_callback;
3167*48a54d36SAndroid Build Coastguard Worker }
3168*48a54d36SAndroid Build Coastguard Worker
3169*48a54d36SAndroid Build Coastguard Worker return(err);
3170*48a54d36SAndroid Build Coastguard Worker }
3171*48a54d36SAndroid Build Coastguard Worker
3172*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
3173*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
3174*48a54d36SAndroid Build Coastguard Worker #pragma mark -
3175*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceReconfirmRecord & Misc
3176*48a54d36SAndroid Build Coastguard Worker #endif
3177*48a54d36SAndroid Build Coastguard Worker
handle_reconfirm_request(request_state * request)3178*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_reconfirm_request(request_state *request)
3179*48a54d36SAndroid Build Coastguard Worker {
3180*48a54d36SAndroid Build Coastguard Worker mStatus status = mStatus_BadParamErr;
3181*48a54d36SAndroid Build Coastguard Worker AuthRecord *rr = read_rr_from_ipc_msg(request, 0, 0);
3182*48a54d36SAndroid Build Coastguard Worker if (rr)
3183*48a54d36SAndroid Build Coastguard Worker {
3184*48a54d36SAndroid Build Coastguard Worker status = mDNS_ReconfirmByValue(&mDNSStorage, &rr->resrec);
3185*48a54d36SAndroid Build Coastguard Worker LogOperation(
3186*48a54d36SAndroid Build Coastguard Worker (status == mStatus_NoError) ?
3187*48a54d36SAndroid Build Coastguard Worker "%3d: DNSServiceReconfirmRecord(%s) interface %d initiated" :
3188*48a54d36SAndroid Build Coastguard Worker "%3d: DNSServiceReconfirmRecord(%s) interface %d failed: %d",
3189*48a54d36SAndroid Build Coastguard Worker request->sd, RRDisplayString(&mDNSStorage, &rr->resrec),
3190*48a54d36SAndroid Build Coastguard Worker mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, rr->resrec.InterfaceID, mDNSfalse), status);
3191*48a54d36SAndroid Build Coastguard Worker freeL("AuthRecord/handle_reconfirm_request", rr);
3192*48a54d36SAndroid Build Coastguard Worker }
3193*48a54d36SAndroid Build Coastguard Worker return(status);
3194*48a54d36SAndroid Build Coastguard Worker }
3195*48a54d36SAndroid Build Coastguard Worker
handle_setdomain_request(request_state * request)3196*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_setdomain_request(request_state *request)
3197*48a54d36SAndroid Build Coastguard Worker {
3198*48a54d36SAndroid Build Coastguard Worker char domainstr[MAX_ESCAPED_DOMAIN_NAME];
3199*48a54d36SAndroid Build Coastguard Worker domainname domain;
3200*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
3201*48a54d36SAndroid Build Coastguard Worker (void)flags; // Unused
3202*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, domainstr, MAX_ESCAPED_DOMAIN_NAME) < 0 ||
3203*48a54d36SAndroid Build Coastguard Worker !MakeDomainNameFromDNSNameString(&domain, domainstr))
3204*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceSetDefaultDomainForUser(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
3205*48a54d36SAndroid Build Coastguard Worker
3206*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceSetDefaultDomainForUser(%##s)", request->sd, domain.c);
3207*48a54d36SAndroid Build Coastguard Worker return(mStatus_NoError);
3208*48a54d36SAndroid Build Coastguard Worker }
3209*48a54d36SAndroid Build Coastguard Worker
3210*48a54d36SAndroid Build Coastguard Worker typedef packedstruct
3211*48a54d36SAndroid Build Coastguard Worker {
3212*48a54d36SAndroid Build Coastguard Worker mStatus err;
3213*48a54d36SAndroid Build Coastguard Worker mDNSu32 len;
3214*48a54d36SAndroid Build Coastguard Worker mDNSu32 vers;
3215*48a54d36SAndroid Build Coastguard Worker } DaemonVersionReply;
3216*48a54d36SAndroid Build Coastguard Worker
handle_getproperty_request(request_state * request)3217*48a54d36SAndroid Build Coastguard Worker mDNSlocal void handle_getproperty_request(request_state *request)
3218*48a54d36SAndroid Build Coastguard Worker {
3219*48a54d36SAndroid Build Coastguard Worker const mStatus BadParamErr = dnssd_htonl((mDNSu32)mStatus_BadParamErr);
3220*48a54d36SAndroid Build Coastguard Worker char prop[256];
3221*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, prop, sizeof(prop)) >= 0)
3222*48a54d36SAndroid Build Coastguard Worker {
3223*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceGetProperty(%s)", request->sd, prop);
3224*48a54d36SAndroid Build Coastguard Worker if (!strcmp(prop, kDNSServiceProperty_DaemonVersion))
3225*48a54d36SAndroid Build Coastguard Worker {
3226*48a54d36SAndroid Build Coastguard Worker DaemonVersionReply x = { 0, dnssd_htonl(4), dnssd_htonl(_DNS_SD_H) };
3227*48a54d36SAndroid Build Coastguard Worker send_all(request->sd, (const char *)&x, sizeof(x));
3228*48a54d36SAndroid Build Coastguard Worker return;
3229*48a54d36SAndroid Build Coastguard Worker }
3230*48a54d36SAndroid Build Coastguard Worker }
3231*48a54d36SAndroid Build Coastguard Worker
3232*48a54d36SAndroid Build Coastguard Worker // If we didn't recogize the requested property name, return BadParamErr
3233*48a54d36SAndroid Build Coastguard Worker send_all(request->sd, (const char *)&BadParamErr, sizeof(BadParamErr));
3234*48a54d36SAndroid Build Coastguard Worker }
3235*48a54d36SAndroid Build Coastguard Worker
3236*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
3237*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
3238*48a54d36SAndroid Build Coastguard Worker #pragma mark -
3239*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceNATPortMappingCreate
3240*48a54d36SAndroid Build Coastguard Worker #endif
3241*48a54d36SAndroid Build Coastguard Worker
3242*48a54d36SAndroid Build Coastguard Worker #define DNSServiceProtocol(X) ((X) == NATOp_AddrRequest ? 0 : (X) == NATOp_MapUDP ? kDNSServiceProtocol_UDP : kDNSServiceProtocol_TCP)
3243*48a54d36SAndroid Build Coastguard Worker
port_mapping_termination_callback(request_state * request)3244*48a54d36SAndroid Build Coastguard Worker mDNSlocal void port_mapping_termination_callback(request_state *request)
3245*48a54d36SAndroid Build Coastguard Worker {
3246*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) STOP", request->sd,
3247*48a54d36SAndroid Build Coastguard Worker DNSServiceProtocol(request->u.pm.NATinfo.Protocol),
3248*48a54d36SAndroid Build Coastguard Worker mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease);
3249*48a54d36SAndroid Build Coastguard Worker mDNS_StopNATOperation(&mDNSStorage, &request->u.pm.NATinfo);
3250*48a54d36SAndroid Build Coastguard Worker }
3251*48a54d36SAndroid Build Coastguard Worker
3252*48a54d36SAndroid Build Coastguard Worker // Called via function pointer when we get a NAT-PMP address request or port mapping response
port_mapping_create_request_callback(mDNS * m,NATTraversalInfo * n)3253*48a54d36SAndroid Build Coastguard Worker mDNSlocal void port_mapping_create_request_callback(mDNS *m, NATTraversalInfo *n)
3254*48a54d36SAndroid Build Coastguard Worker {
3255*48a54d36SAndroid Build Coastguard Worker request_state *request = (request_state *)n->clientContext;
3256*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
3257*48a54d36SAndroid Build Coastguard Worker int replyLen;
3258*48a54d36SAndroid Build Coastguard Worker char *data;
3259*48a54d36SAndroid Build Coastguard Worker
3260*48a54d36SAndroid Build Coastguard Worker if (!request) { LogMsg("port_mapping_create_request_callback called with unknown request_state object"); return; }
3261*48a54d36SAndroid Build Coastguard Worker
3262*48a54d36SAndroid Build Coastguard Worker // calculate reply data length
3263*48a54d36SAndroid Build Coastguard Worker replyLen = sizeof(DNSServiceFlags);
3264*48a54d36SAndroid Build Coastguard Worker replyLen += 3 * sizeof(mDNSu32); // if index + addr + ttl
3265*48a54d36SAndroid Build Coastguard Worker replyLen += sizeof(DNSServiceErrorType);
3266*48a54d36SAndroid Build Coastguard Worker replyLen += 2 * sizeof(mDNSu16); // Internal Port + External Port
3267*48a54d36SAndroid Build Coastguard Worker replyLen += sizeof(mDNSu8); // protocol
3268*48a54d36SAndroid Build Coastguard Worker
3269*48a54d36SAndroid Build Coastguard Worker rep = create_reply(port_mapping_reply_op, replyLen, request);
3270*48a54d36SAndroid Build Coastguard Worker
3271*48a54d36SAndroid Build Coastguard Worker rep->rhdr->flags = dnssd_htonl(0);
3272*48a54d36SAndroid Build Coastguard Worker rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, n->InterfaceID, mDNSfalse));
3273*48a54d36SAndroid Build Coastguard Worker rep->rhdr->error = dnssd_htonl(n->Result);
3274*48a54d36SAndroid Build Coastguard Worker
3275*48a54d36SAndroid Build Coastguard Worker data = (char *)&rep->rhdr[1];
3276*48a54d36SAndroid Build Coastguard Worker
3277*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalAddress.b[0];
3278*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalAddress.b[1];
3279*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalAddress.b[2];
3280*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalAddress.b[3];
3281*48a54d36SAndroid Build Coastguard Worker *data++ = DNSServiceProtocol(request->u.pm.NATinfo.Protocol);
3282*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.IntPort.b[0];
3283*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.IntPort.b[1];
3284*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalPort.b[0];
3285*48a54d36SAndroid Build Coastguard Worker *data++ = request->u.pm.NATinfo.ExternalPort.b[1];
3286*48a54d36SAndroid Build Coastguard Worker put_uint32(request->u.pm.NATinfo.Lifetime, &data);
3287*48a54d36SAndroid Build Coastguard Worker
3288*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) RESULT %.4a:%u TTL %u", request->sd,
3289*48a54d36SAndroid Build Coastguard Worker DNSServiceProtocol(request->u.pm.NATinfo.Protocol),
3290*48a54d36SAndroid Build Coastguard Worker mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease,
3291*48a54d36SAndroid Build Coastguard Worker &request->u.pm.NATinfo.ExternalAddress, mDNSVal16(request->u.pm.NATinfo.ExternalPort), request->u.pm.NATinfo.Lifetime);
3292*48a54d36SAndroid Build Coastguard Worker
3293*48a54d36SAndroid Build Coastguard Worker append_reply(request, rep);
3294*48a54d36SAndroid Build Coastguard Worker }
3295*48a54d36SAndroid Build Coastguard Worker
handle_port_mapping_request(request_state * request)3296*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_port_mapping_request(request_state *request)
3297*48a54d36SAndroid Build Coastguard Worker {
3298*48a54d36SAndroid Build Coastguard Worker mDNSu32 ttl = 0;
3299*48a54d36SAndroid Build Coastguard Worker mStatus err = mStatus_NoError;
3300*48a54d36SAndroid Build Coastguard Worker
3301*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
3302*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
3303*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
3304*48a54d36SAndroid Build Coastguard Worker mDNSu8 protocol = (mDNSu8)get_uint32(&request->msgptr, request->msgend);
3305*48a54d36SAndroid Build Coastguard Worker (void)flags; // Unused
3306*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr);
3307*48a54d36SAndroid Build Coastguard Worker if (request->msgptr + 8 > request->msgend) request->msgptr = NULL;
3308*48a54d36SAndroid Build Coastguard Worker else
3309*48a54d36SAndroid Build Coastguard Worker {
3310*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.IntPort.b[0] = *request->msgptr++;
3311*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.IntPort.b[1] = *request->msgptr++;
3312*48a54d36SAndroid Build Coastguard Worker request->u.pm.ReqExt.b[0] = *request->msgptr++;
3313*48a54d36SAndroid Build Coastguard Worker request->u.pm.ReqExt.b[1] = *request->msgptr++;
3314*48a54d36SAndroid Build Coastguard Worker ttl = get_uint32(&request->msgptr, request->msgend);
3315*48a54d36SAndroid Build Coastguard Worker }
3316*48a54d36SAndroid Build Coastguard Worker
3317*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr)
3318*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: DNSServiceNATPortMappingCreate(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
3319*48a54d36SAndroid Build Coastguard Worker
3320*48a54d36SAndroid Build Coastguard Worker if (protocol == 0) // If protocol == 0 (i.e. just request public address) then IntPort, ExtPort, ttl must be zero too
3321*48a54d36SAndroid Build Coastguard Worker {
3322*48a54d36SAndroid Build Coastguard Worker if (!mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort) || !mDNSIPPortIsZero(request->u.pm.ReqExt) || ttl) return(mStatus_BadParamErr);
3323*48a54d36SAndroid Build Coastguard Worker }
3324*48a54d36SAndroid Build Coastguard Worker else
3325*48a54d36SAndroid Build Coastguard Worker {
3326*48a54d36SAndroid Build Coastguard Worker if (mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort)) return(mStatus_BadParamErr);
3327*48a54d36SAndroid Build Coastguard Worker if (!(protocol & (kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP))) return(mStatus_BadParamErr);
3328*48a54d36SAndroid Build Coastguard Worker }
3329*48a54d36SAndroid Build Coastguard Worker
3330*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.Protocol = !protocol ? NATOp_AddrRequest : (protocol == kDNSServiceProtocol_UDP) ? NATOp_MapUDP : NATOp_MapTCP;
3331*48a54d36SAndroid Build Coastguard Worker // u.pm.NATinfo.IntPort = already set above
3332*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.RequestedPort = request->u.pm.ReqExt;
3333*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.NATLease = ttl;
3334*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.clientCallback = port_mapping_create_request_callback;
3335*48a54d36SAndroid Build Coastguard Worker request->u.pm.NATinfo.clientContext = request;
3336*48a54d36SAndroid Build Coastguard Worker
3337*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) START", request->sd,
3338*48a54d36SAndroid Build Coastguard Worker protocol, mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease);
3339*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartNATOperation(&mDNSStorage, &request->u.pm.NATinfo);
3340*48a54d36SAndroid Build Coastguard Worker if (err) LogMsg("ERROR: mDNS_StartNATOperation: %d", (int)err);
3341*48a54d36SAndroid Build Coastguard Worker else request->terminate = port_mapping_termination_callback;
3342*48a54d36SAndroid Build Coastguard Worker
3343*48a54d36SAndroid Build Coastguard Worker return(err);
3344*48a54d36SAndroid Build Coastguard Worker }
3345*48a54d36SAndroid Build Coastguard Worker
3346*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
3347*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
3348*48a54d36SAndroid Build Coastguard Worker #pragma mark -
3349*48a54d36SAndroid Build Coastguard Worker #pragma mark - DNSServiceGetAddrInfo
3350*48a54d36SAndroid Build Coastguard Worker #endif
3351*48a54d36SAndroid Build Coastguard Worker
addrinfo_termination_callback(request_state * request)3352*48a54d36SAndroid Build Coastguard Worker mDNSlocal void addrinfo_termination_callback(request_state *request)
3353*48a54d36SAndroid Build Coastguard Worker {
3354*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceGetAddrInfo(%##s) STOP", request->sd, request->u.addrinfo.q4.qname.c);
3355*48a54d36SAndroid Build Coastguard Worker
3356*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q4.QuestionContext)
3357*48a54d36SAndroid Build Coastguard Worker {
3358*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4);
3359*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.QuestionContext = mDNSNULL;
3360*48a54d36SAndroid Build Coastguard Worker }
3361*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q4.qnameOrig)
3362*48a54d36SAndroid Build Coastguard Worker {
3363*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination", request->u.addrinfo.q4.qnameOrig);
3364*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.qnameOrig = mDNSNULL;
3365*48a54d36SAndroid Build Coastguard Worker }
3366*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q42)
3367*48a54d36SAndroid Build Coastguard Worker {
3368*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q42->QuestionContext)
3369*48a54d36SAndroid Build Coastguard Worker {
3370*48a54d36SAndroid Build Coastguard Worker LogInfo("addrinfo_termination_callback: Stopping q42 %##s", request->u.addrinfo.q42->qname.c);
3371*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q42);
3372*48a54d36SAndroid Build Coastguard Worker }
3373*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q42->qnameOrig)
3374*48a54d36SAndroid Build Coastguard Worker {
3375*48a54d36SAndroid Build Coastguard Worker LogInfo("addrinfo_termination_callback: freeing q42 qnameOrig %##s", request->u.addrinfo.q42->qnameOrig->c);
3376*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination q42", request->u.addrinfo.q42->qnameOrig);
3377*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q42->qnameOrig = mDNSNULL;
3378*48a54d36SAndroid Build Coastguard Worker }
3379*48a54d36SAndroid Build Coastguard Worker freeL("addrinfo Q42", request->u.addrinfo.q42);
3380*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q42 = mDNSNULL;
3381*48a54d36SAndroid Build Coastguard Worker }
3382*48a54d36SAndroid Build Coastguard Worker
3383*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q6.QuestionContext)
3384*48a54d36SAndroid Build Coastguard Worker {
3385*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q6);
3386*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.QuestionContext = mDNSNULL;
3387*48a54d36SAndroid Build Coastguard Worker }
3388*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q6.qnameOrig)
3389*48a54d36SAndroid Build Coastguard Worker {
3390*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination", request->u.addrinfo.q6.qnameOrig);
3391*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.qnameOrig = mDNSNULL;
3392*48a54d36SAndroid Build Coastguard Worker }
3393*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q62)
3394*48a54d36SAndroid Build Coastguard Worker {
3395*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q62->QuestionContext)
3396*48a54d36SAndroid Build Coastguard Worker {
3397*48a54d36SAndroid Build Coastguard Worker LogInfo("addrinfo_termination_callback: Stopping q62 %##s", request->u.addrinfo.q62->qname.c);
3398*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q62);
3399*48a54d36SAndroid Build Coastguard Worker }
3400*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.q62->qnameOrig)
3401*48a54d36SAndroid Build Coastguard Worker {
3402*48a54d36SAndroid Build Coastguard Worker LogInfo("addrinfo_termination_callback: freeing q62 qnameOrig %##s", request->u.addrinfo.q62->qnameOrig->c);
3403*48a54d36SAndroid Build Coastguard Worker freeL("QueryTermination q62", request->u.addrinfo.q62->qnameOrig);
3404*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q62->qnameOrig = mDNSNULL;
3405*48a54d36SAndroid Build Coastguard Worker }
3406*48a54d36SAndroid Build Coastguard Worker freeL("addrinfo Q62", request->u.addrinfo.q62);
3407*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q62 = mDNSNULL;
3408*48a54d36SAndroid Build Coastguard Worker }
3409*48a54d36SAndroid Build Coastguard Worker }
3410*48a54d36SAndroid Build Coastguard Worker
handle_addrinfo_request(request_state * request)3411*48a54d36SAndroid Build Coastguard Worker mDNSlocal mStatus handle_addrinfo_request(request_state *request)
3412*48a54d36SAndroid Build Coastguard Worker {
3413*48a54d36SAndroid Build Coastguard Worker char hostname[256];
3414*48a54d36SAndroid Build Coastguard Worker domainname d;
3415*48a54d36SAndroid Build Coastguard Worker mStatus err = 0;
3416*48a54d36SAndroid Build Coastguard Worker
3417*48a54d36SAndroid Build Coastguard Worker DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend);
3418*48a54d36SAndroid Build Coastguard Worker mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend);
3419*48a54d36SAndroid Build Coastguard Worker
3420*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(&request->u.addrinfo, sizeof(request->u.addrinfo));
3421*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.interface_id = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex);
3422*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.flags = flags;
3423*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.protocol = get_uint32(&request->msgptr, request->msgend);
3424*48a54d36SAndroid Build Coastguard Worker
3425*48a54d36SAndroid Build Coastguard Worker if (interfaceIndex && !request->u.addrinfo.interface_id) return(mStatus_BadParamErr);
3426*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.protocol > (kDNSServiceProtocol_IPv4|kDNSServiceProtocol_IPv6)) return(mStatus_BadParamErr);
3427*48a54d36SAndroid Build Coastguard Worker
3428*48a54d36SAndroid Build Coastguard Worker if (get_string(&request->msgptr, request->msgend, hostname, 256) < 0) return(mStatus_BadParamErr);
3429*48a54d36SAndroid Build Coastguard Worker
3430*48a54d36SAndroid Build Coastguard Worker if (!request->msgptr) { LogMsg("%3d: DNSServiceGetAddrInfo(unreadable parameters)", request->sd); return(mStatus_BadParamErr); }
3431*48a54d36SAndroid Build Coastguard Worker
3432*48a54d36SAndroid Build Coastguard Worker if (!MakeDomainNameFromDNSNameString(&d, hostname))
3433*48a54d36SAndroid Build Coastguard Worker { LogMsg("ERROR: handle_addrinfo_request: bad hostname: %s", hostname); return(mStatus_BadParamErr); }
3434*48a54d36SAndroid Build Coastguard Worker
3435*48a54d36SAndroid Build Coastguard Worker #if 0
3436*48a54d36SAndroid Build Coastguard Worker if (!AuthorizedDomain(request, &d, AutoBrowseDomains)) return (mStatus_NoError);
3437*48a54d36SAndroid Build Coastguard Worker #endif
3438*48a54d36SAndroid Build Coastguard Worker
3439*48a54d36SAndroid Build Coastguard Worker if (!request->u.addrinfo.protocol)
3440*48a54d36SAndroid Build Coastguard Worker {
3441*48a54d36SAndroid Build Coastguard Worker flags |= kDNSServiceFlagsSuppressUnusable;
3442*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.protocol = (kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6);
3443*48a54d36SAndroid Build Coastguard Worker }
3444*48a54d36SAndroid Build Coastguard Worker
3445*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.InterfaceID = request->u.addrinfo.q6.InterfaceID = request->u.addrinfo.interface_id;
3446*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.Target = request->u.addrinfo.q6.Target = zeroAddr;
3447*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.qname = request->u.addrinfo.q6.qname = d;
3448*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.qclass = request->u.addrinfo.q6.qclass = kDNSServiceClass_IN;
3449*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.LongLived = request->u.addrinfo.q6.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0;
3450*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.ExpectUnique = request->u.addrinfo.q6.ExpectUnique = mDNSfalse;
3451*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.ForceMCast = request->u.addrinfo.q6.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0;
3452*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.ReturnIntermed = request->u.addrinfo.q6.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0;
3453*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.SuppressUnusable = request->u.addrinfo.q6.SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0;
3454*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.TimeoutQuestion = request->u.addrinfo.q6.TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0;
3455*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.WakeOnResolve = request->u.addrinfo.q6.WakeOnResolve = 0;
3456*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.qnameOrig = request->u.addrinfo.q6.qnameOrig = mDNSNULL;
3457*48a54d36SAndroid Build Coastguard Worker
3458*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4)
3459*48a54d36SAndroid Build Coastguard Worker {
3460*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.qtype = kDNSServiceType_A;
3461*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.SearchListIndex = 0;
3462*48a54d36SAndroid Build Coastguard Worker
3463*48a54d36SAndroid Build Coastguard Worker // We append search domains only for queries that are a single label. If overriden using
3464*48a54d36SAndroid Build Coastguard Worker // command line argument "AlwaysAppendSearchDomains", then we do it for any query which
3465*48a54d36SAndroid Build Coastguard Worker // is not fully qualified.
3466*48a54d36SAndroid Build Coastguard Worker if (hostname[strlen(hostname) - 1] != '.' && (AlwaysAppendSearchDomains || CountLabels(&d) == 1))
3467*48a54d36SAndroid Build Coastguard Worker {
3468*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.AppendSearchDomains = 1;
3469*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.AppendLocalSearchDomains = 1;
3470*48a54d36SAndroid Build Coastguard Worker }
3471*48a54d36SAndroid Build Coastguard Worker else
3472*48a54d36SAndroid Build Coastguard Worker {
3473*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.AppendSearchDomains = 0;
3474*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.AppendLocalSearchDomains = 0;
3475*48a54d36SAndroid Build Coastguard Worker }
3476*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q4) ? 1 : 0);
3477*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.QuestionCallback = queryrecord_result_callback;
3478*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.QuestionContext = request;
3479*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q4);
3480*48a54d36SAndroid Build Coastguard Worker if (err != mStatus_NoError)
3481*48a54d36SAndroid Build Coastguard Worker {
3482*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: mDNS_StartQuery: %d", (int)err);
3483*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.QuestionContext = mDNSNULL;
3484*48a54d36SAndroid Build Coastguard Worker }
3485*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
3486*48a54d36SAndroid Build Coastguard Worker err = SendAdditionalQuery(&request->u.addrinfo.q4, request, err);
3487*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
3488*48a54d36SAndroid Build Coastguard Worker }
3489*48a54d36SAndroid Build Coastguard Worker
3490*48a54d36SAndroid Build Coastguard Worker if (!err && (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6))
3491*48a54d36SAndroid Build Coastguard Worker {
3492*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.qtype = kDNSServiceType_AAAA;
3493*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.SearchListIndex = 0;
3494*48a54d36SAndroid Build Coastguard Worker if (hostname[strlen(hostname) - 1] != '.' && (AlwaysAppendSearchDomains || CountLabels(&d) == 1))
3495*48a54d36SAndroid Build Coastguard Worker {
3496*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.AppendSearchDomains = 1;
3497*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.AppendLocalSearchDomains = 1;
3498*48a54d36SAndroid Build Coastguard Worker }
3499*48a54d36SAndroid Build Coastguard Worker else
3500*48a54d36SAndroid Build Coastguard Worker {
3501*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.AppendSearchDomains = 0;
3502*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.AppendLocalSearchDomains = 0;
3503*48a54d36SAndroid Build Coastguard Worker }
3504*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q6) ? 1 : 0);
3505*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.QuestionCallback = queryrecord_result_callback;
3506*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.QuestionContext = request;
3507*48a54d36SAndroid Build Coastguard Worker err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q6);
3508*48a54d36SAndroid Build Coastguard Worker if (err != mStatus_NoError)
3509*48a54d36SAndroid Build Coastguard Worker {
3510*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: mDNS_StartQuery: %d", (int)err);
3511*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q6.QuestionContext = mDNSNULL;
3512*48a54d36SAndroid Build Coastguard Worker if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4)
3513*48a54d36SAndroid Build Coastguard Worker {
3514*48a54d36SAndroid Build Coastguard Worker // If we started a query for IPv4, we need to cancel it
3515*48a54d36SAndroid Build Coastguard Worker mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4);
3516*48a54d36SAndroid Build Coastguard Worker request->u.addrinfo.q4.QuestionContext = mDNSNULL;
3517*48a54d36SAndroid Build Coastguard Worker }
3518*48a54d36SAndroid Build Coastguard Worker }
3519*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
3520*48a54d36SAndroid Build Coastguard Worker err = SendAdditionalQuery(&request->u.addrinfo.q6, request, err);
3521*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
3522*48a54d36SAndroid Build Coastguard Worker }
3523*48a54d36SAndroid Build Coastguard Worker
3524*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: DNSServiceGetAddrInfo(%X, %d, %d, %##s) START",
3525*48a54d36SAndroid Build Coastguard Worker request->sd, flags, interfaceIndex, request->u.addrinfo.protocol, d.c);
3526*48a54d36SAndroid Build Coastguard Worker
3527*48a54d36SAndroid Build Coastguard Worker if (!err) request->terminate = addrinfo_termination_callback;
3528*48a54d36SAndroid Build Coastguard Worker
3529*48a54d36SAndroid Build Coastguard Worker return(err);
3530*48a54d36SAndroid Build Coastguard Worker }
3531*48a54d36SAndroid Build Coastguard Worker
3532*48a54d36SAndroid Build Coastguard Worker // ***************************************************************************
3533*48a54d36SAndroid Build Coastguard Worker #if COMPILER_LIKES_PRAGMA_MARK
3534*48a54d36SAndroid Build Coastguard Worker #pragma mark -
3535*48a54d36SAndroid Build Coastguard Worker #pragma mark - Main Request Handler etc.
3536*48a54d36SAndroid Build Coastguard Worker #endif
3537*48a54d36SAndroid Build Coastguard Worker
NewRequest(void)3538*48a54d36SAndroid Build Coastguard Worker mDNSlocal request_state *NewRequest(void)
3539*48a54d36SAndroid Build Coastguard Worker {
3540*48a54d36SAndroid Build Coastguard Worker request_state **p = &all_requests;
3541*48a54d36SAndroid Build Coastguard Worker while (*p) p=&(*p)->next;
3542*48a54d36SAndroid Build Coastguard Worker *p = mallocL("request_state", sizeof(request_state));
3543*48a54d36SAndroid Build Coastguard Worker if (!*p) FatalError("ERROR: malloc");
3544*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(*p, sizeof(request_state));
3545*48a54d36SAndroid Build Coastguard Worker return(*p);
3546*48a54d36SAndroid Build Coastguard Worker }
3547*48a54d36SAndroid Build Coastguard Worker
3548*48a54d36SAndroid Build Coastguard Worker // read_msg may be called any time when the transfer state (req->ts) is t_morecoming.
3549*48a54d36SAndroid Build Coastguard Worker // if there is no data on the socket, the socket will be closed and t_terminated will be returned
read_msg(request_state * req)3550*48a54d36SAndroid Build Coastguard Worker mDNSlocal void read_msg(request_state *req)
3551*48a54d36SAndroid Build Coastguard Worker {
3552*48a54d36SAndroid Build Coastguard Worker if (req->ts == t_terminated || req->ts == t_error)
3553*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: read_msg called with transfer state terminated or error", req->sd); req->ts = t_error; return; }
3554*48a54d36SAndroid Build Coastguard Worker
3555*48a54d36SAndroid Build Coastguard Worker if (req->ts == t_complete) // this must be death or something is wrong
3556*48a54d36SAndroid Build Coastguard Worker {
3557*48a54d36SAndroid Build Coastguard Worker char buf[4]; // dummy for death notification
3558*48a54d36SAndroid Build Coastguard Worker int nread = udsSupportReadFD(req->sd, buf, 4, 0, req->platform_data);
3559*48a54d36SAndroid Build Coastguard Worker if (!nread) { req->ts = t_terminated; return; }
3560*48a54d36SAndroid Build Coastguard Worker if (nread < 0) goto rerror;
3561*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: ERROR: read data from a completed request", req->sd);
3562*48a54d36SAndroid Build Coastguard Worker req->ts = t_error;
3563*48a54d36SAndroid Build Coastguard Worker return;
3564*48a54d36SAndroid Build Coastguard Worker }
3565*48a54d36SAndroid Build Coastguard Worker
3566*48a54d36SAndroid Build Coastguard Worker if (req->ts != t_morecoming)
3567*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: read_msg called with invalid transfer state (%d)", req->sd, req->ts); req->ts = t_error; return; }
3568*48a54d36SAndroid Build Coastguard Worker
3569*48a54d36SAndroid Build Coastguard Worker if (req->hdr_bytes < sizeof(ipc_msg_hdr))
3570*48a54d36SAndroid Build Coastguard Worker {
3571*48a54d36SAndroid Build Coastguard Worker mDNSu32 nleft = sizeof(ipc_msg_hdr) - req->hdr_bytes;
3572*48a54d36SAndroid Build Coastguard Worker int nread = udsSupportReadFD(req->sd, (char *)&req->hdr + req->hdr_bytes, nleft, 0, req->platform_data);
3573*48a54d36SAndroid Build Coastguard Worker if (nread == 0) { req->ts = t_terminated; return; }
3574*48a54d36SAndroid Build Coastguard Worker if (nread < 0) goto rerror;
3575*48a54d36SAndroid Build Coastguard Worker req->hdr_bytes += nread;
3576*48a54d36SAndroid Build Coastguard Worker if (req->hdr_bytes > sizeof(ipc_msg_hdr))
3577*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: read_msg - read too many header bytes", req->sd); req->ts = t_error; return; }
3578*48a54d36SAndroid Build Coastguard Worker
3579*48a54d36SAndroid Build Coastguard Worker // only read data if header is complete
3580*48a54d36SAndroid Build Coastguard Worker if (req->hdr_bytes == sizeof(ipc_msg_hdr))
3581*48a54d36SAndroid Build Coastguard Worker {
3582*48a54d36SAndroid Build Coastguard Worker ConvertHeaderBytes(&req->hdr);
3583*48a54d36SAndroid Build Coastguard Worker if (req->hdr.version != VERSION)
3584*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: client version 0x%08X daemon version 0x%08X", req->sd, req->hdr.version, VERSION); req->ts = t_error; return; }
3585*48a54d36SAndroid Build Coastguard Worker
3586*48a54d36SAndroid Build Coastguard Worker // Largest conceivable single request is a DNSServiceRegisterRecord() or DNSServiceAddRecord()
3587*48a54d36SAndroid Build Coastguard Worker // with 64kB of rdata. Adding 1009 byte for a maximal domain name, plus a safety margin
3588*48a54d36SAndroid Build Coastguard Worker // for other overhead, this means any message above 70kB is definitely bogus.
3589*48a54d36SAndroid Build Coastguard Worker if (req->hdr.datalen > 70000)
3590*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: read_msg: hdr.datalen %u (0x%X) > 70000", req->sd, req->hdr.datalen, req->hdr.datalen); req->ts = t_error; return; }
3591*48a54d36SAndroid Build Coastguard Worker req->msgbuf = mallocL("request_state msgbuf", req->hdr.datalen + MSG_PAD_BYTES);
3592*48a54d36SAndroid Build Coastguard Worker if (!req->msgbuf) { my_perror("ERROR: malloc"); req->ts = t_error; return; }
3593*48a54d36SAndroid Build Coastguard Worker req->msgptr = req->msgbuf;
3594*48a54d36SAndroid Build Coastguard Worker req->msgend = req->msgbuf + req->hdr.datalen;
3595*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(req->msgbuf, req->hdr.datalen + MSG_PAD_BYTES);
3596*48a54d36SAndroid Build Coastguard Worker }
3597*48a54d36SAndroid Build Coastguard Worker }
3598*48a54d36SAndroid Build Coastguard Worker
3599*48a54d36SAndroid Build Coastguard Worker // If our header is complete, but we're still needing more body data, then try to read it now
3600*48a54d36SAndroid Build Coastguard Worker // Note: For cancel_request req->hdr.datalen == 0, but there's no error return socket for cancel_request
3601*48a54d36SAndroid Build Coastguard Worker // Any time we need to get the error return socket we know we'll have at least one data byte
3602*48a54d36SAndroid Build Coastguard Worker // (even if only the one-byte empty C string placeholder for the old ctrl_path parameter)
3603*48a54d36SAndroid Build Coastguard Worker if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes < req->hdr.datalen)
3604*48a54d36SAndroid Build Coastguard Worker {
3605*48a54d36SAndroid Build Coastguard Worker mDNSu32 nleft = req->hdr.datalen - req->data_bytes;
3606*48a54d36SAndroid Build Coastguard Worker int nread;
3607*48a54d36SAndroid Build Coastguard Worker #if !defined(_WIN32)
3608*48a54d36SAndroid Build Coastguard Worker struct iovec vec = { req->msgbuf + req->data_bytes, nleft }; // Tell recvmsg where we want the bytes put
3609*48a54d36SAndroid Build Coastguard Worker struct msghdr msg;
3610*48a54d36SAndroid Build Coastguard Worker struct cmsghdr *cmsg;
3611*48a54d36SAndroid Build Coastguard Worker char cbuf[CMSG_SPACE(sizeof(dnssd_sock_t))];
3612*48a54d36SAndroid Build Coastguard Worker msg.msg_name = 0;
3613*48a54d36SAndroid Build Coastguard Worker msg.msg_namelen = 0;
3614*48a54d36SAndroid Build Coastguard Worker msg.msg_iov = &vec;
3615*48a54d36SAndroid Build Coastguard Worker msg.msg_iovlen = 1;
3616*48a54d36SAndroid Build Coastguard Worker msg.msg_control = cbuf;
3617*48a54d36SAndroid Build Coastguard Worker msg.msg_controllen = sizeof(cbuf);
3618*48a54d36SAndroid Build Coastguard Worker msg.msg_flags = 0;
3619*48a54d36SAndroid Build Coastguard Worker nread = recvmsg(req->sd, &msg, 0);
3620*48a54d36SAndroid Build Coastguard Worker #else
3621*48a54d36SAndroid Build Coastguard Worker nread = udsSupportReadFD(req->sd, (char *)req->msgbuf + req->data_bytes, nleft, 0, req->platform_data);
3622*48a54d36SAndroid Build Coastguard Worker #endif
3623*48a54d36SAndroid Build Coastguard Worker if (nread == 0) { req->ts = t_terminated; return; }
3624*48a54d36SAndroid Build Coastguard Worker if (nread < 0) goto rerror;
3625*48a54d36SAndroid Build Coastguard Worker req->data_bytes += nread;
3626*48a54d36SAndroid Build Coastguard Worker if (req->data_bytes > req->hdr.datalen)
3627*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: ERROR: read_msg - read too many data bytes", req->sd); req->ts = t_error; return; }
3628*48a54d36SAndroid Build Coastguard Worker #if !defined(_WIN32)
3629*48a54d36SAndroid Build Coastguard Worker cmsg = CMSG_FIRSTHDR(&msg);
3630*48a54d36SAndroid Build Coastguard Worker #if DEBUG_64BIT_SCM_RIGHTS
3631*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Expecting %d %d %d %d", req->sd, sizeof(cbuf), sizeof(cbuf), SOL_SOCKET, SCM_RIGHTS);
3632*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Got %d %d %d %d", req->sd, msg.msg_controllen, cmsg->cmsg_len, cmsg->cmsg_level, cmsg->cmsg_type);
3633*48a54d36SAndroid Build Coastguard Worker #endif // DEBUG_64BIT_SCM_RIGHTS
3634*48a54d36SAndroid Build Coastguard Worker if (msg.msg_controllen == sizeof(cbuf) &&
3635*48a54d36SAndroid Build Coastguard Worker cmsg->cmsg_len == CMSG_LEN(sizeof(dnssd_sock_t)) &&
3636*48a54d36SAndroid Build Coastguard Worker cmsg->cmsg_level == SOL_SOCKET &&
3637*48a54d36SAndroid Build Coastguard Worker cmsg->cmsg_type == SCM_RIGHTS)
3638*48a54d36SAndroid Build Coastguard Worker {
3639*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
3640*48a54d36SAndroid Build Coastguard Worker // Strictly speaking BPF_fd belongs solely in the platform support layer, but because
3641*48a54d36SAndroid Build Coastguard Worker // of privilege separation on Mac OS X we need to get BPF_fd from mDNSResponderHelper,
3642*48a54d36SAndroid Build Coastguard Worker // and it's convenient to repurpose the existing fd-passing code here for that task
3643*48a54d36SAndroid Build Coastguard Worker if (req->hdr.op == send_bpf)
3644*48a54d36SAndroid Build Coastguard Worker {
3645*48a54d36SAndroid Build Coastguard Worker dnssd_sock_t x = *(dnssd_sock_t *)CMSG_DATA(cmsg);
3646*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: Got BPF %d", req->sd, x);
3647*48a54d36SAndroid Build Coastguard Worker mDNSPlatformReceiveBPF_fd(&mDNSStorage, x);
3648*48a54d36SAndroid Build Coastguard Worker }
3649*48a54d36SAndroid Build Coastguard Worker else
3650*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
3651*48a54d36SAndroid Build Coastguard Worker req->errsd = *(dnssd_sock_t *)CMSG_DATA(cmsg);
3652*48a54d36SAndroid Build Coastguard Worker #if DEBUG_64BIT_SCM_RIGHTS
3653*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: read req->errsd %d", req->sd, req->errsd);
3654*48a54d36SAndroid Build Coastguard Worker #endif // DEBUG_64BIT_SCM_RIGHTS
3655*48a54d36SAndroid Build Coastguard Worker if (req->data_bytes < req->hdr.datalen)
3656*48a54d36SAndroid Build Coastguard Worker {
3657*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Client sent error socket %d via SCM_RIGHTS with req->data_bytes %d < req->hdr.datalen %d",
3658*48a54d36SAndroid Build Coastguard Worker req->sd, req->errsd, req->data_bytes, req->hdr.datalen);
3659*48a54d36SAndroid Build Coastguard Worker req->ts = t_error;
3660*48a54d36SAndroid Build Coastguard Worker return;
3661*48a54d36SAndroid Build Coastguard Worker }
3662*48a54d36SAndroid Build Coastguard Worker }
3663*48a54d36SAndroid Build Coastguard Worker #endif
3664*48a54d36SAndroid Build Coastguard Worker }
3665*48a54d36SAndroid Build Coastguard Worker
3666*48a54d36SAndroid Build Coastguard Worker // If our header and data are both complete, see if we need to make our separate error return socket
3667*48a54d36SAndroid Build Coastguard Worker if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes == req->hdr.datalen)
3668*48a54d36SAndroid Build Coastguard Worker {
3669*48a54d36SAndroid Build Coastguard Worker if (req->terminate && req->hdr.op != cancel_request)
3670*48a54d36SAndroid Build Coastguard Worker {
3671*48a54d36SAndroid Build Coastguard Worker dnssd_sockaddr_t cliaddr;
3672*48a54d36SAndroid Build Coastguard Worker #if defined(USE_TCP_LOOPBACK)
3673*48a54d36SAndroid Build Coastguard Worker mDNSOpaque16 port;
3674*48a54d36SAndroid Build Coastguard Worker u_long opt = 1;
3675*48a54d36SAndroid Build Coastguard Worker port.b[0] = req->msgptr[0];
3676*48a54d36SAndroid Build Coastguard Worker port.b[1] = req->msgptr[1];
3677*48a54d36SAndroid Build Coastguard Worker req->msgptr += 2;
3678*48a54d36SAndroid Build Coastguard Worker cliaddr.sin_family = AF_INET;
3679*48a54d36SAndroid Build Coastguard Worker cliaddr.sin_port = port.NotAnInteger;
3680*48a54d36SAndroid Build Coastguard Worker cliaddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR);
3681*48a54d36SAndroid Build Coastguard Worker #else
3682*48a54d36SAndroid Build Coastguard Worker char ctrl_path[MAX_CTLPATH];
3683*48a54d36SAndroid Build Coastguard Worker get_string(&req->msgptr, req->msgend, ctrl_path, MAX_CTLPATH); // path is first element in message buffer
3684*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(&cliaddr, sizeof(cliaddr));
3685*48a54d36SAndroid Build Coastguard Worker cliaddr.sun_family = AF_LOCAL;
3686*48a54d36SAndroid Build Coastguard Worker mDNSPlatformStrCopy(cliaddr.sun_path, ctrl_path);
3687*48a54d36SAndroid Build Coastguard Worker // If the error return path UDS name is empty string, that tells us
3688*48a54d36SAndroid Build Coastguard Worker // that this is a new version of the library that's going to pass us
3689*48a54d36SAndroid Build Coastguard Worker // the error return path socket via sendmsg/recvmsg
3690*48a54d36SAndroid Build Coastguard Worker if (ctrl_path[0] == 0)
3691*48a54d36SAndroid Build Coastguard Worker {
3692*48a54d36SAndroid Build Coastguard Worker if (req->errsd == req->sd)
3693*48a54d36SAndroid Build Coastguard Worker { LogMsg("%3d: read_msg: ERROR failed to get errsd via SCM_RIGHTS", req->sd); req->ts = t_error; return; }
3694*48a54d36SAndroid Build Coastguard Worker goto got_errfd;
3695*48a54d36SAndroid Build Coastguard Worker }
3696*48a54d36SAndroid Build Coastguard Worker #endif
3697*48a54d36SAndroid Build Coastguard Worker
3698*48a54d36SAndroid Build Coastguard Worker req->errsd = socket(AF_DNSSD, SOCK_STREAM, 0);
3699*48a54d36SAndroid Build Coastguard Worker if (!dnssd_SocketValid(req->errsd)) { my_perror("ERROR: socket"); req->ts = t_error; return; }
3700*48a54d36SAndroid Build Coastguard Worker
3701*48a54d36SAndroid Build Coastguard Worker if (connect(req->errsd, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0)
3702*48a54d36SAndroid Build Coastguard Worker {
3703*48a54d36SAndroid Build Coastguard Worker #if !defined(USE_TCP_LOOPBACK)
3704*48a54d36SAndroid Build Coastguard Worker struct stat sb;
3705*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: read_msg: Couldn't connect to error return path socket “%s” errno %d (%s)",
3706*48a54d36SAndroid Build Coastguard Worker req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno));
3707*48a54d36SAndroid Build Coastguard Worker if (stat(cliaddr.sun_path, &sb) < 0)
3708*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: read_msg: stat failed “%s” errno %d (%s)", req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno));
3709*48a54d36SAndroid Build Coastguard Worker else
3710*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: read_msg: file “%s” mode %o (octal) uid %d gid %d", req->sd, cliaddr.sun_path, sb.st_mode, sb.st_uid, sb.st_gid);
3711*48a54d36SAndroid Build Coastguard Worker #endif
3712*48a54d36SAndroid Build Coastguard Worker req->ts = t_error;
3713*48a54d36SAndroid Build Coastguard Worker return;
3714*48a54d36SAndroid Build Coastguard Worker }
3715*48a54d36SAndroid Build Coastguard Worker
3716*48a54d36SAndroid Build Coastguard Worker #if !defined(USE_TCP_LOOPBACK)
3717*48a54d36SAndroid Build Coastguard Worker got_errfd:
3718*48a54d36SAndroid Build Coastguard Worker #endif
3719*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: Error socket %d created %08X %08X", req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0]);
3720*48a54d36SAndroid Build Coastguard Worker #if defined(_WIN32)
3721*48a54d36SAndroid Build Coastguard Worker if (ioctlsocket(req->errsd, FIONBIO, &opt) != 0)
3722*48a54d36SAndroid Build Coastguard Worker #else
3723*48a54d36SAndroid Build Coastguard Worker if (fcntl(req->errsd, F_SETFL, fcntl(req->errsd, F_GETFL, 0) | O_NONBLOCK) != 0)
3724*48a54d36SAndroid Build Coastguard Worker #endif
3725*48a54d36SAndroid Build Coastguard Worker {
3726*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: ERROR: could not set control socket to non-blocking mode errno %d (%s)",
3727*48a54d36SAndroid Build Coastguard Worker req->sd, dnssd_errno, dnssd_strerror(dnssd_errno));
3728*48a54d36SAndroid Build Coastguard Worker req->ts = t_error;
3729*48a54d36SAndroid Build Coastguard Worker return;
3730*48a54d36SAndroid Build Coastguard Worker }
3731*48a54d36SAndroid Build Coastguard Worker }
3732*48a54d36SAndroid Build Coastguard Worker
3733*48a54d36SAndroid Build Coastguard Worker req->ts = t_complete;
3734*48a54d36SAndroid Build Coastguard Worker }
3735*48a54d36SAndroid Build Coastguard Worker
3736*48a54d36SAndroid Build Coastguard Worker return;
3737*48a54d36SAndroid Build Coastguard Worker
3738*48a54d36SAndroid Build Coastguard Worker rerror:
3739*48a54d36SAndroid Build Coastguard Worker if (dnssd_errno == dnssd_EWOULDBLOCK || dnssd_errno == dnssd_EINTR) return;
3740*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: ERROR: read_msg errno %d (%s)", req->sd, dnssd_errno, dnssd_strerror(dnssd_errno));
3741*48a54d36SAndroid Build Coastguard Worker req->ts = t_error;
3742*48a54d36SAndroid Build Coastguard Worker }
3743*48a54d36SAndroid Build Coastguard Worker
3744*48a54d36SAndroid Build Coastguard Worker #define RecordOrientedOp(X) \
3745*48a54d36SAndroid Build Coastguard Worker ((X) == reg_record_request || (X) == add_record_request || (X) == update_record_request || (X) == remove_record_request)
3746*48a54d36SAndroid Build Coastguard Worker
3747*48a54d36SAndroid Build Coastguard Worker // The lightweight operations are the ones that don't need a dedicated request_state structure allocated for them
3748*48a54d36SAndroid Build Coastguard Worker #define LightweightOp(X) (RecordOrientedOp(X) || (X) == cancel_request)
3749*48a54d36SAndroid Build Coastguard Worker
request_callback(int fd,short filter,void * info)3750*48a54d36SAndroid Build Coastguard Worker mDNSlocal void request_callback(int fd, short filter, void *info)
3751*48a54d36SAndroid Build Coastguard Worker {
3752*48a54d36SAndroid Build Coastguard Worker mStatus err = 0;
3753*48a54d36SAndroid Build Coastguard Worker request_state *req = info;
3754*48a54d36SAndroid Build Coastguard Worker mDNSs32 min_size = sizeof(DNSServiceFlags);
3755*48a54d36SAndroid Build Coastguard Worker (void)fd; // Unused
3756*48a54d36SAndroid Build Coastguard Worker (void)filter; // Unused
3757*48a54d36SAndroid Build Coastguard Worker
3758*48a54d36SAndroid Build Coastguard Worker for (;;)
3759*48a54d36SAndroid Build Coastguard Worker {
3760*48a54d36SAndroid Build Coastguard Worker read_msg(req);
3761*48a54d36SAndroid Build Coastguard Worker if (req->ts == t_morecoming) return;
3762*48a54d36SAndroid Build Coastguard Worker if (req->ts == t_terminated || req->ts == t_error) { AbortUnlinkAndFree(req); return; }
3763*48a54d36SAndroid Build Coastguard Worker if (req->ts != t_complete) { LogMsg("req->ts %d != t_complete", req->ts); AbortUnlinkAndFree(req); return; }
3764*48a54d36SAndroid Build Coastguard Worker
3765*48a54d36SAndroid Build Coastguard Worker if (req->hdr.version != VERSION)
3766*48a54d36SAndroid Build Coastguard Worker {
3767*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: client version %d incompatible with daemon version %d", req->hdr.version, VERSION);
3768*48a54d36SAndroid Build Coastguard Worker AbortUnlinkAndFree(req);
3769*48a54d36SAndroid Build Coastguard Worker return;
3770*48a54d36SAndroid Build Coastguard Worker }
3771*48a54d36SAndroid Build Coastguard Worker
3772*48a54d36SAndroid Build Coastguard Worker switch(req->hdr.op) // Interface + other data
3773*48a54d36SAndroid Build Coastguard Worker {
3774*48a54d36SAndroid Build Coastguard Worker case connection_request: min_size = 0; break;
3775*48a54d36SAndroid Build Coastguard Worker case reg_service_request: min_size += sizeof(mDNSu32) + 4 /* name, type, domain, host */ + 4 /* port, textlen */; break;
3776*48a54d36SAndroid Build Coastguard Worker case add_record_request: min_size += 4 /* type, rdlen */ + 4 /* ttl */; break;
3777*48a54d36SAndroid Build Coastguard Worker case update_record_request: min_size += 2 /* rdlen */ + 4 /* ttl */; break;
3778*48a54d36SAndroid Build Coastguard Worker case remove_record_request: break;
3779*48a54d36SAndroid Build Coastguard Worker case browse_request: min_size += sizeof(mDNSu32) + 2 /* type, domain */; break;
3780*48a54d36SAndroid Build Coastguard Worker case resolve_request: min_size += sizeof(mDNSu32) + 3 /* type, type, domain */; break;
3781*48a54d36SAndroid Build Coastguard Worker case query_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 4 /* type, class*/; break;
3782*48a54d36SAndroid Build Coastguard Worker case enumeration_request: min_size += sizeof(mDNSu32); break;
3783*48a54d36SAndroid Build Coastguard Worker case reg_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */ + 4 /* ttl */; break;
3784*48a54d36SAndroid Build Coastguard Worker case reconfirm_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */; break;
3785*48a54d36SAndroid Build Coastguard Worker case setdomain_request: min_size += 1 /* domain */; break;
3786*48a54d36SAndroid Build Coastguard Worker case getproperty_request: min_size = 2; break;
3787*48a54d36SAndroid Build Coastguard Worker case port_mapping_request: min_size += sizeof(mDNSu32) + 4 /* udp/tcp */ + 4 /* int/ext port */ + 4 /* ttl */; break;
3788*48a54d36SAndroid Build Coastguard Worker case addrinfo_request: min_size += sizeof(mDNSu32) + 4 /* v4/v6 */ + 1 /* hostname */; break;
3789*48a54d36SAndroid Build Coastguard Worker case send_bpf: // Same as cancel_request below
3790*48a54d36SAndroid Build Coastguard Worker case cancel_request: min_size = 0; break;
3791*48a54d36SAndroid Build Coastguard Worker case sethost_request: min_size = sizeof(mDNSu32) + 1 /* hostname */; break;
3792*48a54d36SAndroid Build Coastguard Worker default: LogMsg("ERROR: validate_message - unsupported req type: %d", req->hdr.op); min_size = -1; break;
3793*48a54d36SAndroid Build Coastguard Worker }
3794*48a54d36SAndroid Build Coastguard Worker
3795*48a54d36SAndroid Build Coastguard Worker if ((mDNSs32)req->data_bytes < min_size)
3796*48a54d36SAndroid Build Coastguard Worker { LogMsg("Invalid message %d bytes; min for %d is %d", req->data_bytes, req->hdr.op, min_size); AbortUnlinkAndFree(req); return; }
3797*48a54d36SAndroid Build Coastguard Worker
3798*48a54d36SAndroid Build Coastguard Worker if (LightweightOp(req->hdr.op) && !req->terminate)
3799*48a54d36SAndroid Build Coastguard Worker { LogMsg("Reg/Add/Update/Remove %d require existing connection", req->hdr.op); AbortUnlinkAndFree(req); return; }
3800*48a54d36SAndroid Build Coastguard Worker
3801*48a54d36SAndroid Build Coastguard Worker // check if client wants silent operation
3802*48a54d36SAndroid Build Coastguard Worker if (req->hdr.ipc_flags & IPC_FLAGS_NOREPLY) req->no_reply = 1;
3803*48a54d36SAndroid Build Coastguard Worker
3804*48a54d36SAndroid Build Coastguard Worker // If req->terminate is already set, this means this operation is sharing an existing connection
3805*48a54d36SAndroid Build Coastguard Worker if (req->terminate && !LightweightOp(req->hdr.op))
3806*48a54d36SAndroid Build Coastguard Worker {
3807*48a54d36SAndroid Build Coastguard Worker request_state *newreq = NewRequest();
3808*48a54d36SAndroid Build Coastguard Worker newreq->primary = req;
3809*48a54d36SAndroid Build Coastguard Worker newreq->sd = req->sd;
3810*48a54d36SAndroid Build Coastguard Worker newreq->errsd = req->errsd;
3811*48a54d36SAndroid Build Coastguard Worker newreq->uid = req->uid;
3812*48a54d36SAndroid Build Coastguard Worker newreq->hdr = req->hdr;
3813*48a54d36SAndroid Build Coastguard Worker newreq->msgbuf = req->msgbuf;
3814*48a54d36SAndroid Build Coastguard Worker newreq->msgptr = req->msgptr;
3815*48a54d36SAndroid Build Coastguard Worker newreq->msgend = req->msgend;
3816*48a54d36SAndroid Build Coastguard Worker req = newreq;
3817*48a54d36SAndroid Build Coastguard Worker }
3818*48a54d36SAndroid Build Coastguard Worker
3819*48a54d36SAndroid Build Coastguard Worker // If we're shutting down, don't allow new client requests
3820*48a54d36SAndroid Build Coastguard Worker // We do allow "cancel" and "getproperty" during shutdown
3821*48a54d36SAndroid Build Coastguard Worker if (mDNSStorage.ShutdownTime && req->hdr.op != cancel_request && req->hdr.op != getproperty_request)
3822*48a54d36SAndroid Build Coastguard Worker {
3823*48a54d36SAndroid Build Coastguard Worker err = mStatus_ServiceNotRunning;
3824*48a54d36SAndroid Build Coastguard Worker }
3825*48a54d36SAndroid Build Coastguard Worker else switch(req->hdr.op)
3826*48a54d36SAndroid Build Coastguard Worker {
3827*48a54d36SAndroid Build Coastguard Worker // These are all operations that have their own first-class request_state object
3828*48a54d36SAndroid Build Coastguard Worker case connection_request: LogOperation("%3d: DNSServiceCreateConnection START", req->sd);
3829*48a54d36SAndroid Build Coastguard Worker req->terminate = connection_termination; break;
3830*48a54d36SAndroid Build Coastguard Worker case resolve_request: err = handle_resolve_request (req); break;
3831*48a54d36SAndroid Build Coastguard Worker case query_request: err = handle_queryrecord_request (req); break;
3832*48a54d36SAndroid Build Coastguard Worker case browse_request: err = handle_browse_request (req); break;
3833*48a54d36SAndroid Build Coastguard Worker case reg_service_request: err = handle_regservice_request (req); break;
3834*48a54d36SAndroid Build Coastguard Worker case enumeration_request: err = handle_enum_request (req); break;
3835*48a54d36SAndroid Build Coastguard Worker case reconfirm_record_request: err = handle_reconfirm_request (req); break;
3836*48a54d36SAndroid Build Coastguard Worker case setdomain_request: err = handle_setdomain_request (req); break;
3837*48a54d36SAndroid Build Coastguard Worker case getproperty_request: handle_getproperty_request (req); break;
3838*48a54d36SAndroid Build Coastguard Worker case port_mapping_request: err = handle_port_mapping_request(req); break;
3839*48a54d36SAndroid Build Coastguard Worker case addrinfo_request: err = handle_addrinfo_request (req); break;
3840*48a54d36SAndroid Build Coastguard Worker case send_bpf: /* Do nothing for send_bpf */ break;
3841*48a54d36SAndroid Build Coastguard Worker
3842*48a54d36SAndroid Build Coastguard Worker // These are all operations that work with an existing request_state object
3843*48a54d36SAndroid Build Coastguard Worker case reg_record_request: err = handle_regrecord_request (req); break;
3844*48a54d36SAndroid Build Coastguard Worker case add_record_request: err = handle_add_request (req); break;
3845*48a54d36SAndroid Build Coastguard Worker case update_record_request: err = handle_update_request (req); break;
3846*48a54d36SAndroid Build Coastguard Worker case remove_record_request: err = handle_removerecord_request(req); break;
3847*48a54d36SAndroid Build Coastguard Worker case cancel_request: handle_cancel_request (req); break;
3848*48a54d36SAndroid Build Coastguard Worker case sethost_request: err = handle_sethost_request (req); break;
3849*48a54d36SAndroid Build Coastguard Worker default: LogMsg("%3d: ERROR: Unsupported UDS req: %d", req->sd, req->hdr.op);
3850*48a54d36SAndroid Build Coastguard Worker }
3851*48a54d36SAndroid Build Coastguard Worker
3852*48a54d36SAndroid Build Coastguard Worker // req->msgbuf may be NULL, e.g. for connection_request or remove_record_request
3853*48a54d36SAndroid Build Coastguard Worker if (req->msgbuf) freeL("request_state msgbuf", req->msgbuf);
3854*48a54d36SAndroid Build Coastguard Worker
3855*48a54d36SAndroid Build Coastguard Worker // There's no return data for a cancel request (DNSServiceRefDeallocate returns no result)
3856*48a54d36SAndroid Build Coastguard Worker // For a DNSServiceGetProperty call, the handler already generated the response, so no need to do it again here
3857*48a54d36SAndroid Build Coastguard Worker if (req->hdr.op != cancel_request && req->hdr.op != getproperty_request && req->hdr.op != send_bpf)
3858*48a54d36SAndroid Build Coastguard Worker {
3859*48a54d36SAndroid Build Coastguard Worker const mStatus err_netorder = dnssd_htonl(err);
3860*48a54d36SAndroid Build Coastguard Worker send_all(req->errsd, (const char *)&err_netorder, sizeof(err_netorder));
3861*48a54d36SAndroid Build Coastguard Worker if (req->errsd != req->sd)
3862*48a54d36SAndroid Build Coastguard Worker {
3863*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: Error socket %d closed %08X %08X (%d)",
3864*48a54d36SAndroid Build Coastguard Worker req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0], err);
3865*48a54d36SAndroid Build Coastguard Worker dnssd_close(req->errsd);
3866*48a54d36SAndroid Build Coastguard Worker req->errsd = req->sd;
3867*48a54d36SAndroid Build Coastguard Worker // Also need to reset the parent's errsd, if this is a subordinate operation
3868*48a54d36SAndroid Build Coastguard Worker if (req->primary) req->primary->errsd = req->primary->sd;
3869*48a54d36SAndroid Build Coastguard Worker }
3870*48a54d36SAndroid Build Coastguard Worker }
3871*48a54d36SAndroid Build Coastguard Worker
3872*48a54d36SAndroid Build Coastguard Worker // Reset ready to accept the next req on this pipe
3873*48a54d36SAndroid Build Coastguard Worker if (req->primary) req = req->primary;
3874*48a54d36SAndroid Build Coastguard Worker req->ts = t_morecoming;
3875*48a54d36SAndroid Build Coastguard Worker req->hdr_bytes = 0;
3876*48a54d36SAndroid Build Coastguard Worker req->data_bytes = 0;
3877*48a54d36SAndroid Build Coastguard Worker req->msgbuf = mDNSNULL;
3878*48a54d36SAndroid Build Coastguard Worker req->msgptr = mDNSNULL;
3879*48a54d36SAndroid Build Coastguard Worker req->msgend = 0;
3880*48a54d36SAndroid Build Coastguard Worker }
3881*48a54d36SAndroid Build Coastguard Worker }
3882*48a54d36SAndroid Build Coastguard Worker
connect_callback(int fd,short filter,void * info)3883*48a54d36SAndroid Build Coastguard Worker mDNSlocal void connect_callback(int fd, short filter, void *info)
3884*48a54d36SAndroid Build Coastguard Worker {
3885*48a54d36SAndroid Build Coastguard Worker dnssd_sockaddr_t cliaddr;
3886*48a54d36SAndroid Build Coastguard Worker dnssd_socklen_t len = (dnssd_socklen_t) sizeof(cliaddr);
3887*48a54d36SAndroid Build Coastguard Worker dnssd_sock_t sd = accept(fd, (struct sockaddr*) &cliaddr, &len);
3888*48a54d36SAndroid Build Coastguard Worker #if defined(SO_NOSIGPIPE) || defined(_WIN32)
3889*48a54d36SAndroid Build Coastguard Worker unsigned long optval = 1;
3890*48a54d36SAndroid Build Coastguard Worker #endif
3891*48a54d36SAndroid Build Coastguard Worker
3892*48a54d36SAndroid Build Coastguard Worker (void)filter; // Unused
3893*48a54d36SAndroid Build Coastguard Worker (void)info; // Unused
3894*48a54d36SAndroid Build Coastguard Worker
3895*48a54d36SAndroid Build Coastguard Worker if (!dnssd_SocketValid(sd))
3896*48a54d36SAndroid Build Coastguard Worker {
3897*48a54d36SAndroid Build Coastguard Worker if (dnssd_errno != dnssd_EWOULDBLOCK) my_perror("ERROR: accept");
3898*48a54d36SAndroid Build Coastguard Worker return;
3899*48a54d36SAndroid Build Coastguard Worker }
3900*48a54d36SAndroid Build Coastguard Worker
3901*48a54d36SAndroid Build Coastguard Worker #ifdef SO_NOSIGPIPE
3902*48a54d36SAndroid Build Coastguard Worker // Some environments (e.g. OS X) support turning off SIGPIPE for a socket
3903*48a54d36SAndroid Build Coastguard Worker if (setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &optval, sizeof(optval)) < 0)
3904*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: WARNING: setsockopt - SO_NOSIGPIPE %d (%s)", sd, dnssd_errno, dnssd_strerror(dnssd_errno));
3905*48a54d36SAndroid Build Coastguard Worker #endif
3906*48a54d36SAndroid Build Coastguard Worker
3907*48a54d36SAndroid Build Coastguard Worker #if defined(_WIN32)
3908*48a54d36SAndroid Build Coastguard Worker if (ioctlsocket(sd, FIONBIO, &optval) != 0)
3909*48a54d36SAndroid Build Coastguard Worker #else
3910*48a54d36SAndroid Build Coastguard Worker if (fcntl(sd, F_SETFL, fcntl(sd, F_GETFL, 0) | O_NONBLOCK) != 0)
3911*48a54d36SAndroid Build Coastguard Worker #endif
3912*48a54d36SAndroid Build Coastguard Worker {
3913*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: fcntl(sd, F_SETFL, O_NONBLOCK) - aborting client");
3914*48a54d36SAndroid Build Coastguard Worker dnssd_close(sd);
3915*48a54d36SAndroid Build Coastguard Worker return;
3916*48a54d36SAndroid Build Coastguard Worker }
3917*48a54d36SAndroid Build Coastguard Worker else
3918*48a54d36SAndroid Build Coastguard Worker {
3919*48a54d36SAndroid Build Coastguard Worker request_state *request = NewRequest();
3920*48a54d36SAndroid Build Coastguard Worker request->ts = t_morecoming;
3921*48a54d36SAndroid Build Coastguard Worker request->sd = sd;
3922*48a54d36SAndroid Build Coastguard Worker request->errsd = sd;
3923*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
3924*48a54d36SAndroid Build Coastguard Worker struct xucred x;
3925*48a54d36SAndroid Build Coastguard Worker socklen_t xucredlen = sizeof(x);
3926*48a54d36SAndroid Build Coastguard Worker if (getsockopt(sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 && x.cr_version == XUCRED_VERSION) request->uid = x.cr_uid;
3927*48a54d36SAndroid Build Coastguard Worker else my_perror("ERROR: getsockopt, LOCAL_PEERCRED");
3928*48a54d36SAndroid Build Coastguard Worker debugf("LOCAL_PEERCRED %d %u %u %d", xucredlen, x.cr_version, x.cr_uid, x.cr_ngroups);
3929*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
3930*48a54d36SAndroid Build Coastguard Worker LogOperation("%3d: Adding FD for uid %u", request->sd, request->uid);
3931*48a54d36SAndroid Build Coastguard Worker udsSupportAddFDToEventLoop(sd, request_callback, request, &request->platform_data);
3932*48a54d36SAndroid Build Coastguard Worker }
3933*48a54d36SAndroid Build Coastguard Worker }
3934*48a54d36SAndroid Build Coastguard Worker
uds_socket_setup(dnssd_sock_t skt)3935*48a54d36SAndroid Build Coastguard Worker mDNSlocal mDNSBool uds_socket_setup(dnssd_sock_t skt)
3936*48a54d36SAndroid Build Coastguard Worker {
3937*48a54d36SAndroid Build Coastguard Worker #if defined(SO_NP_EXTENSIONS)
3938*48a54d36SAndroid Build Coastguard Worker struct so_np_extensions sonpx;
3939*48a54d36SAndroid Build Coastguard Worker socklen_t optlen = sizeof(struct so_np_extensions);
3940*48a54d36SAndroid Build Coastguard Worker sonpx.npx_flags = SONPX_SETOPTSHUT;
3941*48a54d36SAndroid Build Coastguard Worker sonpx.npx_mask = SONPX_SETOPTSHUT;
3942*48a54d36SAndroid Build Coastguard Worker if (setsockopt(skt, SOL_SOCKET, SO_NP_EXTENSIONS, &sonpx, optlen) < 0)
3943*48a54d36SAndroid Build Coastguard Worker my_perror("WARNING: could not set sockopt - SO_NP_EXTENSIONS");
3944*48a54d36SAndroid Build Coastguard Worker #endif
3945*48a54d36SAndroid Build Coastguard Worker #if defined(_WIN32)
3946*48a54d36SAndroid Build Coastguard Worker // SEH: do we even need to do this on windows?
3947*48a54d36SAndroid Build Coastguard Worker // This socket will be given to WSAEventSelect which will automatically set it to non-blocking
3948*48a54d36SAndroid Build Coastguard Worker u_long opt = 1;
3949*48a54d36SAndroid Build Coastguard Worker if (ioctlsocket(skt, FIONBIO, &opt) != 0)
3950*48a54d36SAndroid Build Coastguard Worker #else
3951*48a54d36SAndroid Build Coastguard Worker if (fcntl(skt, F_SETFL, fcntl(skt, F_GETFL, 0) | O_NONBLOCK) != 0)
3952*48a54d36SAndroid Build Coastguard Worker #endif
3953*48a54d36SAndroid Build Coastguard Worker {
3954*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: could not set listen socket to non-blocking mode");
3955*48a54d36SAndroid Build Coastguard Worker return mDNSfalse;
3956*48a54d36SAndroid Build Coastguard Worker }
3957*48a54d36SAndroid Build Coastguard Worker
3958*48a54d36SAndroid Build Coastguard Worker if (listen(skt, LISTENQ) != 0)
3959*48a54d36SAndroid Build Coastguard Worker {
3960*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: could not listen on listen socket");
3961*48a54d36SAndroid Build Coastguard Worker return mDNSfalse;
3962*48a54d36SAndroid Build Coastguard Worker }
3963*48a54d36SAndroid Build Coastguard Worker
3964*48a54d36SAndroid Build Coastguard Worker if (mStatus_NoError != udsSupportAddFDToEventLoop(skt, connect_callback, (void *) NULL, (void **) NULL))
3965*48a54d36SAndroid Build Coastguard Worker {
3966*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: could not add listen socket to event loop");
3967*48a54d36SAndroid Build Coastguard Worker return mDNSfalse;
3968*48a54d36SAndroid Build Coastguard Worker }
3969*48a54d36SAndroid Build Coastguard Worker else LogOperation("%3d: Listening for incoming Unix Domain Socket client requests", skt);
3970*48a54d36SAndroid Build Coastguard Worker
3971*48a54d36SAndroid Build Coastguard Worker return mDNStrue;
3972*48a54d36SAndroid Build Coastguard Worker }
3973*48a54d36SAndroid Build Coastguard Worker
udsserver_init(dnssd_sock_t skts[],mDNSu32 count)3974*48a54d36SAndroid Build Coastguard Worker mDNSexport int udsserver_init(dnssd_sock_t skts[], mDNSu32 count)
3975*48a54d36SAndroid Build Coastguard Worker {
3976*48a54d36SAndroid Build Coastguard Worker dnssd_sockaddr_t laddr;
3977*48a54d36SAndroid Build Coastguard Worker int ret;
3978*48a54d36SAndroid Build Coastguard Worker mDNSu32 i = 0;
3979*48a54d36SAndroid Build Coastguard Worker
3980*48a54d36SAndroid Build Coastguard Worker LogInfo("udsserver_init");
3981*48a54d36SAndroid Build Coastguard Worker
3982*48a54d36SAndroid Build Coastguard Worker // If a particular platform wants to opt out of having a PID file, define PID_FILE to be ""
3983*48a54d36SAndroid Build Coastguard Worker if (PID_FILE[0])
3984*48a54d36SAndroid Build Coastguard Worker {
3985*48a54d36SAndroid Build Coastguard Worker FILE *fp = fopen(PID_FILE, "w");
3986*48a54d36SAndroid Build Coastguard Worker if (fp != NULL)
3987*48a54d36SAndroid Build Coastguard Worker {
3988*48a54d36SAndroid Build Coastguard Worker fprintf(fp, "%d\n", getpid());
3989*48a54d36SAndroid Build Coastguard Worker fclose(fp);
3990*48a54d36SAndroid Build Coastguard Worker }
3991*48a54d36SAndroid Build Coastguard Worker }
3992*48a54d36SAndroid Build Coastguard Worker
3993*48a54d36SAndroid Build Coastguard Worker if (skts)
3994*48a54d36SAndroid Build Coastguard Worker {
3995*48a54d36SAndroid Build Coastguard Worker for (i = 0; i < count; i++)
3996*48a54d36SAndroid Build Coastguard Worker if (dnssd_SocketValid(skts[i]) && !uds_socket_setup(skts[i]))
3997*48a54d36SAndroid Build Coastguard Worker goto error;
3998*48a54d36SAndroid Build Coastguard Worker }
3999*48a54d36SAndroid Build Coastguard Worker else
4000*48a54d36SAndroid Build Coastguard Worker {
4001*48a54d36SAndroid Build Coastguard Worker listenfd = socket(AF_DNSSD, SOCK_STREAM, 0);
4002*48a54d36SAndroid Build Coastguard Worker if (!dnssd_SocketValid(listenfd))
4003*48a54d36SAndroid Build Coastguard Worker {
4004*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: socket(AF_DNSSD, SOCK_STREAM, 0); failed");
4005*48a54d36SAndroid Build Coastguard Worker goto error;
4006*48a54d36SAndroid Build Coastguard Worker }
4007*48a54d36SAndroid Build Coastguard Worker
4008*48a54d36SAndroid Build Coastguard Worker mDNSPlatformMemZero(&laddr, sizeof(laddr));
4009*48a54d36SAndroid Build Coastguard Worker
4010*48a54d36SAndroid Build Coastguard Worker #if defined(USE_TCP_LOOPBACK)
4011*48a54d36SAndroid Build Coastguard Worker {
4012*48a54d36SAndroid Build Coastguard Worker laddr.sin_family = AF_INET;
4013*48a54d36SAndroid Build Coastguard Worker laddr.sin_port = htons(MDNS_TCP_SERVERPORT);
4014*48a54d36SAndroid Build Coastguard Worker laddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR);
4015*48a54d36SAndroid Build Coastguard Worker ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr));
4016*48a54d36SAndroid Build Coastguard Worker if (ret < 0)
4017*48a54d36SAndroid Build Coastguard Worker {
4018*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed");
4019*48a54d36SAndroid Build Coastguard Worker goto error;
4020*48a54d36SAndroid Build Coastguard Worker }
4021*48a54d36SAndroid Build Coastguard Worker }
4022*48a54d36SAndroid Build Coastguard Worker #else
4023*48a54d36SAndroid Build Coastguard Worker {
4024*48a54d36SAndroid Build Coastguard Worker mode_t mask = umask(0);
4025*48a54d36SAndroid Build Coastguard Worker unlink(MDNS_UDS_SERVERPATH); // OK if this fails
4026*48a54d36SAndroid Build Coastguard Worker laddr.sun_family = AF_LOCAL;
4027*48a54d36SAndroid Build Coastguard Worker #ifndef NOT_HAVE_SA_LEN
4028*48a54d36SAndroid Build Coastguard Worker // According to Stevens (section 3.2), there is no portable way to
4029*48a54d36SAndroid Build Coastguard Worker // determine whether sa_len is defined on a particular platform.
4030*48a54d36SAndroid Build Coastguard Worker laddr.sun_len = sizeof(struct sockaddr_un);
4031*48a54d36SAndroid Build Coastguard Worker #endif
4032*48a54d36SAndroid Build Coastguard Worker if (strlen(MDNS_UDS_SERVERPATH) >= sizeof(laddr.sun_path))
4033*48a54d36SAndroid Build Coastguard Worker {
4034*48a54d36SAndroid Build Coastguard Worker LogMsg("ERROR: MDNS_UDS_SERVERPATH must be < %d characters", (int)sizeof(laddr.sun_path));
4035*48a54d36SAndroid Build Coastguard Worker goto error;
4036*48a54d36SAndroid Build Coastguard Worker }
4037*48a54d36SAndroid Build Coastguard Worker mDNSPlatformStrCopy(laddr.sun_path, MDNS_UDS_SERVERPATH);
4038*48a54d36SAndroid Build Coastguard Worker ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr));
4039*48a54d36SAndroid Build Coastguard Worker umask(mask);
4040*48a54d36SAndroid Build Coastguard Worker if (ret < 0)
4041*48a54d36SAndroid Build Coastguard Worker {
4042*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed");
4043*48a54d36SAndroid Build Coastguard Worker goto error;
4044*48a54d36SAndroid Build Coastguard Worker }
4045*48a54d36SAndroid Build Coastguard Worker }
4046*48a54d36SAndroid Build Coastguard Worker #endif
4047*48a54d36SAndroid Build Coastguard Worker
4048*48a54d36SAndroid Build Coastguard Worker if (!uds_socket_setup(listenfd)) goto error;
4049*48a54d36SAndroid Build Coastguard Worker }
4050*48a54d36SAndroid Build Coastguard Worker
4051*48a54d36SAndroid Build Coastguard Worker #if !defined(PLATFORM_NO_RLIMIT)
4052*48a54d36SAndroid Build Coastguard Worker {
4053*48a54d36SAndroid Build Coastguard Worker // Set maximum number of open file descriptors
4054*48a54d36SAndroid Build Coastguard Worker #define MIN_OPENFILES 10240
4055*48a54d36SAndroid Build Coastguard Worker struct rlimit maxfds, newfds;
4056*48a54d36SAndroid Build Coastguard Worker
4057*48a54d36SAndroid Build Coastguard Worker // Due to bugs in OS X (<rdar://problem/2941095>, <rdar://problem/3342704>, <rdar://problem/3839173>)
4058*48a54d36SAndroid Build Coastguard Worker // you have to get and set rlimits once before getrlimit will return sensible values
4059*48a54d36SAndroid Build Coastguard Worker if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
4060*48a54d36SAndroid Build Coastguard Worker if (setrlimit(RLIMIT_NOFILE, &maxfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit");
4061*48a54d36SAndroid Build Coastguard Worker
4062*48a54d36SAndroid Build Coastguard Worker if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
4063*48a54d36SAndroid Build Coastguard Worker newfds.rlim_max = (maxfds.rlim_max > MIN_OPENFILES) ? maxfds.rlim_max : MIN_OPENFILES;
4064*48a54d36SAndroid Build Coastguard Worker newfds.rlim_cur = (maxfds.rlim_cur > MIN_OPENFILES) ? maxfds.rlim_cur : MIN_OPENFILES;
4065*48a54d36SAndroid Build Coastguard Worker if (newfds.rlim_max != maxfds.rlim_max || newfds.rlim_cur != maxfds.rlim_cur)
4066*48a54d36SAndroid Build Coastguard Worker if (setrlimit(RLIMIT_NOFILE, &newfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit");
4067*48a54d36SAndroid Build Coastguard Worker
4068*48a54d36SAndroid Build Coastguard Worker if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; }
4069*48a54d36SAndroid Build Coastguard Worker debugf("maxfds.rlim_max %d", (long)maxfds.rlim_max);
4070*48a54d36SAndroid Build Coastguard Worker debugf("maxfds.rlim_cur %d", (long)maxfds.rlim_cur);
4071*48a54d36SAndroid Build Coastguard Worker }
4072*48a54d36SAndroid Build Coastguard Worker #endif
4073*48a54d36SAndroid Build Coastguard Worker
4074*48a54d36SAndroid Build Coastguard Worker // We start a "LocalOnly" query looking for Automatic Browse Domain records.
4075*48a54d36SAndroid Build Coastguard Worker // When Domain Enumeration in uDNS.c finds an "lb" record from the network, its "FoundDomain" routine
4076*48a54d36SAndroid Build Coastguard Worker // creates a "LocalOnly" record, which results in our AutomaticBrowseDomainChange callback being invoked
4077*48a54d36SAndroid Build Coastguard Worker mDNS_GetDomains(&mDNSStorage, &mDNSStorage.AutomaticBrowseDomainQ, mDNS_DomainTypeBrowseAutomatic,
4078*48a54d36SAndroid Build Coastguard Worker mDNSNULL, mDNSInterface_LocalOnly, AutomaticBrowseDomainChange, mDNSNULL);
4079*48a54d36SAndroid Build Coastguard Worker
4080*48a54d36SAndroid Build Coastguard Worker // Add "local" as recommended registration domain ("dns-sd -E"), recommended browsing domain ("dns-sd -F"), and automatic browsing domain
4081*48a54d36SAndroid Build Coastguard Worker RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeRegistration);
4082*48a54d36SAndroid Build Coastguard Worker RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeBrowse);
4083*48a54d36SAndroid Build Coastguard Worker AddAutoBrowseDomain(0, &localdomain);
4084*48a54d36SAndroid Build Coastguard Worker
4085*48a54d36SAndroid Build Coastguard Worker udsserver_handle_configchange(&mDNSStorage);
4086*48a54d36SAndroid Build Coastguard Worker return 0;
4087*48a54d36SAndroid Build Coastguard Worker
4088*48a54d36SAndroid Build Coastguard Worker error:
4089*48a54d36SAndroid Build Coastguard Worker
4090*48a54d36SAndroid Build Coastguard Worker my_perror("ERROR: udsserver_init");
4091*48a54d36SAndroid Build Coastguard Worker return -1;
4092*48a54d36SAndroid Build Coastguard Worker }
4093*48a54d36SAndroid Build Coastguard Worker
udsserver_exit(void)4094*48a54d36SAndroid Build Coastguard Worker mDNSexport int udsserver_exit(void)
4095*48a54d36SAndroid Build Coastguard Worker {
4096*48a54d36SAndroid Build Coastguard Worker // Cancel all outstanding client requests
4097*48a54d36SAndroid Build Coastguard Worker while (all_requests) AbortUnlinkAndFree(all_requests);
4098*48a54d36SAndroid Build Coastguard Worker
4099*48a54d36SAndroid Build Coastguard Worker // Clean up any special mDNSInterface_LocalOnly records we created, both the entries for "local" we
4100*48a54d36SAndroid Build Coastguard Worker // created in udsserver_init, and others we created as a result of reading local configuration data
4101*48a54d36SAndroid Build Coastguard Worker while (LocalDomainEnumRecords)
4102*48a54d36SAndroid Build Coastguard Worker {
4103*48a54d36SAndroid Build Coastguard Worker ARListElem *rem = LocalDomainEnumRecords;
4104*48a54d36SAndroid Build Coastguard Worker LocalDomainEnumRecords = LocalDomainEnumRecords->next;
4105*48a54d36SAndroid Build Coastguard Worker mDNS_Deregister(&mDNSStorage, &rem->ar);
4106*48a54d36SAndroid Build Coastguard Worker }
4107*48a54d36SAndroid Build Coastguard Worker
4108*48a54d36SAndroid Build Coastguard Worker // If the launching environment created no listening socket,
4109*48a54d36SAndroid Build Coastguard Worker // that means we created it ourselves, so we should clean it up on exit
4110*48a54d36SAndroid Build Coastguard Worker if (dnssd_SocketValid(listenfd))
4111*48a54d36SAndroid Build Coastguard Worker {
4112*48a54d36SAndroid Build Coastguard Worker dnssd_close(listenfd);
4113*48a54d36SAndroid Build Coastguard Worker #if !defined(USE_TCP_LOOPBACK)
4114*48a54d36SAndroid Build Coastguard Worker // Currently, we're unable to remove /var/run/mdnsd because we've changed to userid "nobody"
4115*48a54d36SAndroid Build Coastguard Worker // to give up unnecessary privilege, but we need to be root to remove this Unix Domain Socket.
4116*48a54d36SAndroid Build Coastguard Worker // It would be nice if we could find a solution to this problem
4117*48a54d36SAndroid Build Coastguard Worker if (unlink(MDNS_UDS_SERVERPATH))
4118*48a54d36SAndroid Build Coastguard Worker debugf("Unable to remove %s", MDNS_UDS_SERVERPATH);
4119*48a54d36SAndroid Build Coastguard Worker #endif
4120*48a54d36SAndroid Build Coastguard Worker }
4121*48a54d36SAndroid Build Coastguard Worker
4122*48a54d36SAndroid Build Coastguard Worker if (PID_FILE[0]) unlink(PID_FILE);
4123*48a54d36SAndroid Build Coastguard Worker
4124*48a54d36SAndroid Build Coastguard Worker return 0;
4125*48a54d36SAndroid Build Coastguard Worker }
4126*48a54d36SAndroid Build Coastguard Worker
LogClientInfo(mDNS * const m,const request_state * req)4127*48a54d36SAndroid Build Coastguard Worker mDNSlocal void LogClientInfo(mDNS *const m, const request_state *req)
4128*48a54d36SAndroid Build Coastguard Worker {
4129*48a54d36SAndroid Build Coastguard Worker char prefix[16];
4130*48a54d36SAndroid Build Coastguard Worker if (req->primary) mDNS_snprintf(prefix, sizeof(prefix), " -> ");
4131*48a54d36SAndroid Build Coastguard Worker else mDNS_snprintf(prefix, sizeof(prefix), "%3d:", req->sd);
4132*48a54d36SAndroid Build Coastguard Worker
4133*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4134*48a54d36SAndroid Build Coastguard Worker
4135*48a54d36SAndroid Build Coastguard Worker if (!req->terminate)
4136*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s No operation yet on this socket", prefix);
4137*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == connection_termination)
4138*48a54d36SAndroid Build Coastguard Worker {
4139*48a54d36SAndroid Build Coastguard Worker int num_records = 0, num_ops = 0;
4140*48a54d36SAndroid Build Coastguard Worker const registered_record_entry *p;
4141*48a54d36SAndroid Build Coastguard Worker const request_state *r;
4142*48a54d36SAndroid Build Coastguard Worker for (p = req->u.reg_recs; p; p=p->next) num_records++;
4143*48a54d36SAndroid Build Coastguard Worker for (r = req->next; r; r=r->next) if (r->primary == req) num_ops++;
4144*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceCreateConnection: %d registered record%s, %d kDNSServiceFlagsShareConnection operation%s", prefix,
4145*48a54d36SAndroid Build Coastguard Worker num_records, num_records != 1 ? "s" : "",
4146*48a54d36SAndroid Build Coastguard Worker num_ops, num_ops != 1 ? "s" : "");
4147*48a54d36SAndroid Build Coastguard Worker for (p = req->u.reg_recs; p; p=p->next)
4148*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" -> DNSServiceRegisterRecord %3d %s", p->key, ARDisplayString(m, p->rr));
4149*48a54d36SAndroid Build Coastguard Worker for (r = req->next; r; r=r->next) if (r->primary == req) LogClientInfo(m, r);
4150*48a54d36SAndroid Build Coastguard Worker }
4151*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == regservice_termination_callback)
4152*48a54d36SAndroid Build Coastguard Worker {
4153*48a54d36SAndroid Build Coastguard Worker service_instance *ptr;
4154*48a54d36SAndroid Build Coastguard Worker for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next)
4155*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceRegister %##s %u/%u",
4156*48a54d36SAndroid Build Coastguard Worker (ptr == req->u.servicereg.instances) ? prefix : " ",
4157*48a54d36SAndroid Build Coastguard Worker ptr->srs.RR_SRV.resrec.name->c, mDNSVal16(req->u.servicereg.port), SRS_PORT(&ptr->srs));
4158*48a54d36SAndroid Build Coastguard Worker }
4159*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == browse_termination_callback)
4160*48a54d36SAndroid Build Coastguard Worker {
4161*48a54d36SAndroid Build Coastguard Worker browser_t *blist;
4162*48a54d36SAndroid Build Coastguard Worker for (blist = req->u.browser.browsers; blist; blist = blist->next)
4163*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceBrowse %##s", (blist == req->u.browser.browsers) ? prefix : " ", blist->q.qname.c);
4164*48a54d36SAndroid Build Coastguard Worker }
4165*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == resolve_termination_callback)
4166*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceResolve %##s", prefix, req->u.resolve.qsrv.qname.c);
4167*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == queryrecord_termination_callback)
4168*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceQueryRecord %##s (%s)", prefix, req->u.queryrecord.q.qname.c, DNSTypeName(req->u.queryrecord.q.qtype));
4169*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == enum_termination_callback)
4170*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceEnumerateDomains %##s", prefix, req->u.enumeration.q_all.qname.c);
4171*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == port_mapping_termination_callback)
4172*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceNATPortMapping %.4a %s%s Int %d Req %d Ext %d Req TTL %d Granted TTL %d",
4173*48a54d36SAndroid Build Coastguard Worker prefix,
4174*48a54d36SAndroid Build Coastguard Worker &req->u.pm.NATinfo.ExternalAddress,
4175*48a54d36SAndroid Build Coastguard Worker req->u.pm.NATinfo.Protocol & NATOp_MapTCP ? "TCP" : " ",
4176*48a54d36SAndroid Build Coastguard Worker req->u.pm.NATinfo.Protocol & NATOp_MapUDP ? "UDP" : " ",
4177*48a54d36SAndroid Build Coastguard Worker mDNSVal16(req->u.pm.NATinfo.IntPort),
4178*48a54d36SAndroid Build Coastguard Worker mDNSVal16(req->u.pm.ReqExt),
4179*48a54d36SAndroid Build Coastguard Worker mDNSVal16(req->u.pm.NATinfo.ExternalPort),
4180*48a54d36SAndroid Build Coastguard Worker req->u.pm.NATinfo.NATLease,
4181*48a54d36SAndroid Build Coastguard Worker req->u.pm.NATinfo.Lifetime);
4182*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == addrinfo_termination_callback)
4183*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s DNSServiceGetAddrInfo %s%s %##s", prefix,
4184*48a54d36SAndroid Build Coastguard Worker req->u.addrinfo.protocol & kDNSServiceProtocol_IPv4 ? "v4" : " ",
4185*48a54d36SAndroid Build Coastguard Worker req->u.addrinfo.protocol & kDNSServiceProtocol_IPv6 ? "v6" : " ",
4186*48a54d36SAndroid Build Coastguard Worker req->u.addrinfo.q4.qname.c);
4187*48a54d36SAndroid Build Coastguard Worker else
4188*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s Unrecognized operation %p", prefix, req->terminate);
4189*48a54d36SAndroid Build Coastguard Worker }
4190*48a54d36SAndroid Build Coastguard Worker
RecordTypeName(mDNSu8 rtype)4191*48a54d36SAndroid Build Coastguard Worker mDNSlocal char *RecordTypeName(mDNSu8 rtype)
4192*48a54d36SAndroid Build Coastguard Worker {
4193*48a54d36SAndroid Build Coastguard Worker switch (rtype)
4194*48a54d36SAndroid Build Coastguard Worker {
4195*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeUnregistered: return ("Unregistered ");
4196*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeDeregistering: return ("Deregistering");
4197*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeUnique: return ("Unique ");
4198*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeAdvisory: return ("Advisory ");
4199*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeShared: return ("Shared ");
4200*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeVerified: return ("Verified ");
4201*48a54d36SAndroid Build Coastguard Worker case kDNSRecordTypeKnownUnique: return ("KnownUnique ");
4202*48a54d36SAndroid Build Coastguard Worker default: return("Unknown");
4203*48a54d36SAndroid Build Coastguard Worker }
4204*48a54d36SAndroid Build Coastguard Worker }
4205*48a54d36SAndroid Build Coastguard Worker
LogEtcHosts(mDNS * const m)4206*48a54d36SAndroid Build Coastguard Worker mDNSlocal void LogEtcHosts(mDNS *const m)
4207*48a54d36SAndroid Build Coastguard Worker {
4208*48a54d36SAndroid Build Coastguard Worker mDNSBool showheader = mDNStrue;
4209*48a54d36SAndroid Build Coastguard Worker const AuthRecord *ar;
4210*48a54d36SAndroid Build Coastguard Worker mDNSu32 slot;
4211*48a54d36SAndroid Build Coastguard Worker AuthGroup *ag;
4212*48a54d36SAndroid Build Coastguard Worker int count = 0;
4213*48a54d36SAndroid Build Coastguard Worker int authslot = 0;
4214*48a54d36SAndroid Build Coastguard Worker mDNSBool truncated = 0;
4215*48a54d36SAndroid Build Coastguard Worker
4216*48a54d36SAndroid Build Coastguard Worker for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
4217*48a54d36SAndroid Build Coastguard Worker {
4218*48a54d36SAndroid Build Coastguard Worker if (m->rrauth.rrauth_hash[slot]) authslot++;
4219*48a54d36SAndroid Build Coastguard Worker for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next)
4220*48a54d36SAndroid Build Coastguard Worker for (ar = ag->members; ar; ar = ar->next)
4221*48a54d36SAndroid Build Coastguard Worker {
4222*48a54d36SAndroid Build Coastguard Worker if (ar->RecordCallback != FreeEtcHosts) continue;
4223*48a54d36SAndroid Build Coastguard Worker if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); }
4224*48a54d36SAndroid Build Coastguard Worker
4225*48a54d36SAndroid Build Coastguard Worker // Print a maximum of 50 records
4226*48a54d36SAndroid Build Coastguard Worker if (count++ >= 50) { truncated = mDNStrue; continue; }
4227*48a54d36SAndroid Build Coastguard Worker if (ar->ARType == AuthRecordLocalOnly)
4228*48a54d36SAndroid Build Coastguard Worker {
4229*48a54d36SAndroid Build Coastguard Worker if (ar->resrec.InterfaceID == mDNSInterface_LocalOnly)
4230*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar));
4231*48a54d36SAndroid Build Coastguard Worker else
4232*48a54d36SAndroid Build Coastguard Worker {
4233*48a54d36SAndroid Build Coastguard Worker mDNSu32 scopeid = (mDNSu32)(uintptr_t)ar->resrec.InterfaceID;
4234*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" %s %u %s", RecordTypeName(ar->resrec.RecordType), scopeid, ARDisplayString(m, ar));
4235*48a54d36SAndroid Build Coastguard Worker }
4236*48a54d36SAndroid Build Coastguard Worker }
4237*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4238*48a54d36SAndroid Build Coastguard Worker }
4239*48a54d36SAndroid Build Coastguard Worker }
4240*48a54d36SAndroid Build Coastguard Worker
4241*48a54d36SAndroid Build Coastguard Worker if (showheader) LogMsgNoIdent("<None>");
4242*48a54d36SAndroid Build Coastguard Worker else if (truncated) LogMsgNoIdent("<Truncated: to 50 records, Total records %d, Total Auth Groups %d, Auth Slots %d>", count, m->rrauth.rrauth_totalused, authslot);
4243*48a54d36SAndroid Build Coastguard Worker }
4244*48a54d36SAndroid Build Coastguard Worker
LogLocalOnlyAuthRecords(mDNS * const m)4245*48a54d36SAndroid Build Coastguard Worker mDNSlocal void LogLocalOnlyAuthRecords(mDNS *const m)
4246*48a54d36SAndroid Build Coastguard Worker {
4247*48a54d36SAndroid Build Coastguard Worker mDNSBool showheader = mDNStrue;
4248*48a54d36SAndroid Build Coastguard Worker const AuthRecord *ar;
4249*48a54d36SAndroid Build Coastguard Worker mDNSu32 slot;
4250*48a54d36SAndroid Build Coastguard Worker AuthGroup *ag;
4251*48a54d36SAndroid Build Coastguard Worker
4252*48a54d36SAndroid Build Coastguard Worker for (slot = 0; slot < AUTH_HASH_SLOTS; slot++)
4253*48a54d36SAndroid Build Coastguard Worker {
4254*48a54d36SAndroid Build Coastguard Worker for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next)
4255*48a54d36SAndroid Build Coastguard Worker for (ar = ag->members; ar; ar = ar->next)
4256*48a54d36SAndroid Build Coastguard Worker {
4257*48a54d36SAndroid Build Coastguard Worker if (ar->RecordCallback == FreeEtcHosts) continue;
4258*48a54d36SAndroid Build Coastguard Worker if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); }
4259*48a54d36SAndroid Build Coastguard Worker
4260*48a54d36SAndroid Build Coastguard Worker // Print a maximum of 400 records
4261*48a54d36SAndroid Build Coastguard Worker if (ar->ARType == AuthRecordLocalOnly)
4262*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar));
4263*48a54d36SAndroid Build Coastguard Worker else if (ar->ARType == AuthRecordP2P)
4264*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" %s PP %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar));
4265*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4266*48a54d36SAndroid Build Coastguard Worker }
4267*48a54d36SAndroid Build Coastguard Worker }
4268*48a54d36SAndroid Build Coastguard Worker
4269*48a54d36SAndroid Build Coastguard Worker if (showheader) LogMsgNoIdent("<None>");
4270*48a54d36SAndroid Build Coastguard Worker }
4271*48a54d36SAndroid Build Coastguard Worker
LogAuthRecords(mDNS * const m,const mDNSs32 now,AuthRecord * ResourceRecords,int * proxy)4272*48a54d36SAndroid Build Coastguard Worker mDNSlocal void LogAuthRecords(mDNS *const m, const mDNSs32 now, AuthRecord *ResourceRecords, int *proxy)
4273*48a54d36SAndroid Build Coastguard Worker {
4274*48a54d36SAndroid Build Coastguard Worker mDNSBool showheader = mDNStrue;
4275*48a54d36SAndroid Build Coastguard Worker const AuthRecord *ar;
4276*48a54d36SAndroid Build Coastguard Worker OwnerOptData owner = zeroOwner;
4277*48a54d36SAndroid Build Coastguard Worker for (ar = ResourceRecords; ar; ar=ar->next)
4278*48a54d36SAndroid Build Coastguard Worker {
4279*48a54d36SAndroid Build Coastguard Worker const char *const ifname = InterfaceNameForID(m, ar->resrec.InterfaceID);
4280*48a54d36SAndroid Build Coastguard Worker if ((ar->WakeUp.HMAC.l[0] != 0) == (proxy != mDNSNULL))
4281*48a54d36SAndroid Build Coastguard Worker {
4282*48a54d36SAndroid Build Coastguard Worker if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" Int Next Expire State"); }
4283*48a54d36SAndroid Build Coastguard Worker if (proxy) (*proxy)++;
4284*48a54d36SAndroid Build Coastguard Worker if (!mDNSPlatformMemSame(&owner, &ar->WakeUp, sizeof(owner)))
4285*48a54d36SAndroid Build Coastguard Worker {
4286*48a54d36SAndroid Build Coastguard Worker owner = ar->WakeUp;
4287*48a54d36SAndroid Build Coastguard Worker if (owner.password.l[0])
4288*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a Password %.6a seq %d", &owner.HMAC, &owner.IMAC, &owner.password, owner.seq);
4289*48a54d36SAndroid Build Coastguard Worker else if (!mDNSSameEthAddress(&owner.HMAC, &owner.IMAC))
4290*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a seq %d", &owner.HMAC, &owner.IMAC, owner.seq);
4291*48a54d36SAndroid Build Coastguard Worker else
4292*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Proxying for %.6a seq %d", &owner.HMAC, owner.seq);
4293*48a54d36SAndroid Build Coastguard Worker }
4294*48a54d36SAndroid Build Coastguard Worker if (AuthRecord_uDNS(ar))
4295*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%7d %7d %7d %7d %s",
4296*48a54d36SAndroid Build Coastguard Worker ar->ThisAPInterval / mDNSPlatformOneSecond,
4297*48a54d36SAndroid Build Coastguard Worker (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond,
4298*48a54d36SAndroid Build Coastguard Worker ar->expire ? (ar->expire - now) / mDNSPlatformOneSecond : 0,
4299*48a54d36SAndroid Build Coastguard Worker ar->state, ARDisplayString(m, ar));
4300*48a54d36SAndroid Build Coastguard Worker else if (ar->ARType == AuthRecordLocalOnly)
4301*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" LO %s", ARDisplayString(m, ar));
4302*48a54d36SAndroid Build Coastguard Worker else if (ar->ARType == AuthRecordP2P)
4303*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" PP %s", ARDisplayString(m, ar));
4304*48a54d36SAndroid Build Coastguard Worker else
4305*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%7d %7d %7d %7s %s",
4306*48a54d36SAndroid Build Coastguard Worker ar->ThisAPInterval / mDNSPlatformOneSecond,
4307*48a54d36SAndroid Build Coastguard Worker ar->AnnounceCount ? (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond : 0,
4308*48a54d36SAndroid Build Coastguard Worker ar->TimeExpire ? (ar->TimeExpire - now) / mDNSPlatformOneSecond : 0,
4309*48a54d36SAndroid Build Coastguard Worker ifname ? ifname : "ALL",
4310*48a54d36SAndroid Build Coastguard Worker ARDisplayString(m, ar));
4311*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4312*48a54d36SAndroid Build Coastguard Worker }
4313*48a54d36SAndroid Build Coastguard Worker }
4314*48a54d36SAndroid Build Coastguard Worker if (showheader) LogMsgNoIdent("<None>");
4315*48a54d36SAndroid Build Coastguard Worker }
4316*48a54d36SAndroid Build Coastguard Worker
udsserver_info(mDNS * const m)4317*48a54d36SAndroid Build Coastguard Worker mDNSexport void udsserver_info(mDNS *const m)
4318*48a54d36SAndroid Build Coastguard Worker {
4319*48a54d36SAndroid Build Coastguard Worker const mDNSs32 now = mDNS_TimeNow(m);
4320*48a54d36SAndroid Build Coastguard Worker mDNSu32 CacheUsed = 0, CacheActive = 0, slot;
4321*48a54d36SAndroid Build Coastguard Worker int ProxyA = 0, ProxyD = 0;
4322*48a54d36SAndroid Build Coastguard Worker const CacheGroup *cg;
4323*48a54d36SAndroid Build Coastguard Worker const CacheRecord *cr;
4324*48a54d36SAndroid Build Coastguard Worker const DNSQuestion *q;
4325*48a54d36SAndroid Build Coastguard Worker const DNameListElem *d;
4326*48a54d36SAndroid Build Coastguard Worker const SearchListElem *s;
4327*48a54d36SAndroid Build Coastguard Worker
4328*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Timenow 0x%08lX (%d)", (mDNSu32)now, now);
4329*48a54d36SAndroid Build Coastguard Worker
4330*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("------------ Cache -------------");
4331*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Slt Q TTL if U Type rdlen");
4332*48a54d36SAndroid Build Coastguard Worker for (slot = 0; slot < CACHE_HASH_SLOTS; slot++)
4333*48a54d36SAndroid Build Coastguard Worker for (cg = m->rrcache_hash[slot]; cg; cg=cg->next)
4334*48a54d36SAndroid Build Coastguard Worker {
4335*48a54d36SAndroid Build Coastguard Worker CacheUsed++; // Count one cache entity for the CacheGroup object
4336*48a54d36SAndroid Build Coastguard Worker for (cr = cg->members; cr; cr=cr->next)
4337*48a54d36SAndroid Build Coastguard Worker {
4338*48a54d36SAndroid Build Coastguard Worker const mDNSs32 remain = cr->resrec.rroriginalttl - (now - cr->TimeRcvd) / mDNSPlatformOneSecond;
4339*48a54d36SAndroid Build Coastguard Worker const char *ifname;
4340*48a54d36SAndroid Build Coastguard Worker mDNSInterfaceID InterfaceID = cr->resrec.InterfaceID;
4341*48a54d36SAndroid Build Coastguard Worker if (!InterfaceID && cr->resrec.rDNSServer)
4342*48a54d36SAndroid Build Coastguard Worker InterfaceID = cr->resrec.rDNSServer->interface;
4343*48a54d36SAndroid Build Coastguard Worker ifname = InterfaceNameForID(m, InterfaceID);
4344*48a54d36SAndroid Build Coastguard Worker CacheUsed++;
4345*48a54d36SAndroid Build Coastguard Worker if (cr->CRActiveQuestion) CacheActive++;
4346*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%3d %s%8ld %-7s%s %-6s%s",
4347*48a54d36SAndroid Build Coastguard Worker slot,
4348*48a54d36SAndroid Build Coastguard Worker cr->CRActiveQuestion ? "*" : " ",
4349*48a54d36SAndroid Build Coastguard Worker remain,
4350*48a54d36SAndroid Build Coastguard Worker ifname ? ifname : "-U-",
4351*48a54d36SAndroid Build Coastguard Worker (cr->resrec.RecordType == kDNSRecordTypePacketNegative) ? "-" :
4352*48a54d36SAndroid Build Coastguard Worker (cr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) ? " " : "+",
4353*48a54d36SAndroid Build Coastguard Worker DNSTypeName(cr->resrec.rrtype),
4354*48a54d36SAndroid Build Coastguard Worker CRDisplayString(m, cr));
4355*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4356*48a54d36SAndroid Build Coastguard Worker }
4357*48a54d36SAndroid Build Coastguard Worker }
4358*48a54d36SAndroid Build Coastguard Worker
4359*48a54d36SAndroid Build Coastguard Worker if (m->rrcache_totalused != CacheUsed)
4360*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Cache use mismatch: rrcache_totalused is %lu, true count %lu", m->rrcache_totalused, CacheUsed);
4361*48a54d36SAndroid Build Coastguard Worker if (m->rrcache_active != CacheActive)
4362*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Cache use mismatch: rrcache_active is %lu, true count %lu", m->rrcache_active, CacheActive);
4363*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Cache currently contains %lu entities; %lu referenced by active questions", CacheUsed, CacheActive);
4364*48a54d36SAndroid Build Coastguard Worker
4365*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--------- Auth Records ---------");
4366*48a54d36SAndroid Build Coastguard Worker LogAuthRecords(m, now, m->ResourceRecords, mDNSNULL);
4367*48a54d36SAndroid Build Coastguard Worker
4368*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--------- LocalOnly, P2P Auth Records ---------");
4369*48a54d36SAndroid Build Coastguard Worker LogLocalOnlyAuthRecords(m);
4370*48a54d36SAndroid Build Coastguard Worker
4371*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--------- /etc/hosts ---------");
4372*48a54d36SAndroid Build Coastguard Worker LogEtcHosts(m);
4373*48a54d36SAndroid Build Coastguard Worker
4374*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("------ Duplicate Records -------");
4375*48a54d36SAndroid Build Coastguard Worker LogAuthRecords(m, now, m->DuplicateRecords, mDNSNULL);
4376*48a54d36SAndroid Build Coastguard Worker
4377*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("----- Auth Records Proxied -----");
4378*48a54d36SAndroid Build Coastguard Worker LogAuthRecords(m, now, m->ResourceRecords, &ProxyA);
4379*48a54d36SAndroid Build Coastguard Worker
4380*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("-- Duplicate Records Proxied ---");
4381*48a54d36SAndroid Build Coastguard Worker LogAuthRecords(m, now, m->DuplicateRecords, &ProxyD);
4382*48a54d36SAndroid Build Coastguard Worker
4383*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("---------- Questions -----------");
4384*48a54d36SAndroid Build Coastguard Worker if (!m->Questions) LogMsgNoIdent("<None>");
4385*48a54d36SAndroid Build Coastguard Worker else
4386*48a54d36SAndroid Build Coastguard Worker {
4387*48a54d36SAndroid Build Coastguard Worker CacheUsed = 0;
4388*48a54d36SAndroid Build Coastguard Worker CacheActive = 0;
4389*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" Int Next if T NumAns VDNS Qptr DupOf SU SQ Type Name");
4390*48a54d36SAndroid Build Coastguard Worker for (q = m->Questions; q; q=q->next)
4391*48a54d36SAndroid Build Coastguard Worker {
4392*48a54d36SAndroid Build Coastguard Worker mDNSs32 i = q->ThisQInterval / mDNSPlatformOneSecond;
4393*48a54d36SAndroid Build Coastguard Worker mDNSs32 n = (NextQSendTime(q) - now) / mDNSPlatformOneSecond;
4394*48a54d36SAndroid Build Coastguard Worker char *ifname = InterfaceNameForID(m, q->InterfaceID);
4395*48a54d36SAndroid Build Coastguard Worker CacheUsed++;
4396*48a54d36SAndroid Build Coastguard Worker if (q->ThisQInterval) CacheActive++;
4397*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%6d%6d %-7s%s%s %5d 0x%x%x 0x%p 0x%p %1d %2d %-5s%##s%s",
4398*48a54d36SAndroid Build Coastguard Worker i, n,
4399*48a54d36SAndroid Build Coastguard Worker ifname ? ifname : mDNSOpaque16IsZero(q->TargetQID) ? "" : "-U-",
4400*48a54d36SAndroid Build Coastguard Worker mDNSOpaque16IsZero(q->TargetQID) ? (q->LongLived ? "l" : " ") : (q->LongLived ? "L" : "O"),
4401*48a54d36SAndroid Build Coastguard Worker PrivateQuery(q) ? "P" : " ",
4402*48a54d36SAndroid Build Coastguard Worker q->CurrentAnswers, q->validDNSServers.l[1], q->validDNSServers.l[0], q, q->DuplicateOf,
4403*48a54d36SAndroid Build Coastguard Worker q->SuppressUnusable, q->SuppressQuery, DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : "");
4404*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4405*48a54d36SAndroid Build Coastguard Worker }
4406*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%lu question%s; %lu active", CacheUsed, CacheUsed > 1 ? "s" : "", CacheActive);
4407*48a54d36SAndroid Build Coastguard Worker }
4408*48a54d36SAndroid Build Coastguard Worker
4409*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("----- Local-Only Questions -----");
4410*48a54d36SAndroid Build Coastguard Worker if (!m->LocalOnlyQuestions) LogMsgNoIdent("<None>");
4411*48a54d36SAndroid Build Coastguard Worker else for (q = m->LocalOnlyQuestions; q; q=q->next)
4412*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" %5d %-6s%##s%s",
4413*48a54d36SAndroid Build Coastguard Worker q->CurrentAnswers, DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : "");
4414*48a54d36SAndroid Build Coastguard Worker
4415*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("---- Active Client Requests ----");
4416*48a54d36SAndroid Build Coastguard Worker if (!all_requests) LogMsgNoIdent("<None>");
4417*48a54d36SAndroid Build Coastguard Worker else
4418*48a54d36SAndroid Build Coastguard Worker {
4419*48a54d36SAndroid Build Coastguard Worker const request_state *req, *r;
4420*48a54d36SAndroid Build Coastguard Worker for (req = all_requests; req; req=req->next)
4421*48a54d36SAndroid Build Coastguard Worker {
4422*48a54d36SAndroid Build Coastguard Worker if (req->primary) // If this is a subbordinate operation, check that the parent is in the list
4423*48a54d36SAndroid Build Coastguard Worker {
4424*48a54d36SAndroid Build Coastguard Worker for (r = all_requests; r && r != req; r=r->next) if (r == req->primary) goto foundparent;
4425*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%3d: Orhpan operation %p; parent %p not found in request list", req->sd);
4426*48a54d36SAndroid Build Coastguard Worker }
4427*48a54d36SAndroid Build Coastguard Worker // For non-subbordinate operations, and subbordinate operations that have lost their parent, write out their info
4428*48a54d36SAndroid Build Coastguard Worker LogClientInfo(m, req);
4429*48a54d36SAndroid Build Coastguard Worker foundparent:;
4430*48a54d36SAndroid Build Coastguard Worker }
4431*48a54d36SAndroid Build Coastguard Worker }
4432*48a54d36SAndroid Build Coastguard Worker
4433*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("-------- NAT Traversals --------");
4434*48a54d36SAndroid Build Coastguard Worker if (!m->NATTraversals) LogMsgNoIdent("<None>");
4435*48a54d36SAndroid Build Coastguard Worker else
4436*48a54d36SAndroid Build Coastguard Worker {
4437*48a54d36SAndroid Build Coastguard Worker const NATTraversalInfo *nat;
4438*48a54d36SAndroid Build Coastguard Worker for (nat = m->NATTraversals; nat; nat=nat->next)
4439*48a54d36SAndroid Build Coastguard Worker {
4440*48a54d36SAndroid Build Coastguard Worker if (nat->Protocol)
4441*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%p %s Int %5d Ext %5d Err %d Retry %5d Interval %5d Expire %5d",
4442*48a54d36SAndroid Build Coastguard Worker nat, nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP",
4443*48a54d36SAndroid Build Coastguard Worker mDNSVal16(nat->IntPort), mDNSVal16(nat->ExternalPort), nat->Result,
4444*48a54d36SAndroid Build Coastguard Worker nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0,
4445*48a54d36SAndroid Build Coastguard Worker nat->retryInterval / mDNSPlatformOneSecond,
4446*48a54d36SAndroid Build Coastguard Worker nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0);
4447*48a54d36SAndroid Build Coastguard Worker else
4448*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%p Address Request Retry %5d Interval %5d", nat,
4449*48a54d36SAndroid Build Coastguard Worker (m->retryGetAddr - now) / mDNSPlatformOneSecond,
4450*48a54d36SAndroid Build Coastguard Worker m->retryIntervalGetAddr / mDNSPlatformOneSecond);
4451*48a54d36SAndroid Build Coastguard Worker usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000);
4452*48a54d36SAndroid Build Coastguard Worker }
4453*48a54d36SAndroid Build Coastguard Worker }
4454*48a54d36SAndroid Build Coastguard Worker
4455*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--------- AuthInfoList ---------");
4456*48a54d36SAndroid Build Coastguard Worker if (!m->AuthInfoList) LogMsgNoIdent("<None>");
4457*48a54d36SAndroid Build Coastguard Worker else
4458*48a54d36SAndroid Build Coastguard Worker {
4459*48a54d36SAndroid Build Coastguard Worker const DomainAuthInfo *a;
4460*48a54d36SAndroid Build Coastguard Worker for (a = m->AuthInfoList; a; a = a->next)
4461*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%##s %##s %##s %d %s", a->domain.c, a->keyname.c, a->hostname.c, (a->port.b[0] << 8 | a->port.b[1]), a->AutoTunnel ? a->AutoTunnel : "");
4462*48a54d36SAndroid Build Coastguard Worker }
4463*48a54d36SAndroid Build Coastguard Worker
4464*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder
4465*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--------- TunnelClients --------");
4466*48a54d36SAndroid Build Coastguard Worker if (!m->TunnelClients) LogMsgNoIdent("<None>");
4467*48a54d36SAndroid Build Coastguard Worker else
4468*48a54d36SAndroid Build Coastguard Worker {
4469*48a54d36SAndroid Build Coastguard Worker const ClientTunnel *c;
4470*48a54d36SAndroid Build Coastguard Worker for (c = m->TunnelClients; c; c = c->next)
4471*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%s %##s local %.16a %.4a %.16a remote %.16a %.4a %5d %.16a interval %d",
4472*48a54d36SAndroid Build Coastguard Worker c->prefix, c->dstname.c, &c->loc_inner, &c->loc_outer, &c->loc_outer6, &c->rmt_inner, &c->rmt_outer, mDNSVal16(c->rmt_outer_port), &c->rmt_outer6, c->q.ThisQInterval);
4473*48a54d36SAndroid Build Coastguard Worker }
4474*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder
4475*48a54d36SAndroid Build Coastguard Worker
4476*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("---------- Misc State ----------");
4477*48a54d36SAndroid Build Coastguard Worker
4478*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("PrimaryMAC: %.6a", &m->PrimaryMAC);
4479*48a54d36SAndroid Build Coastguard Worker
4480*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->SleepState %d (%s) seq %d",
4481*48a54d36SAndroid Build Coastguard Worker m->SleepState,
4482*48a54d36SAndroid Build Coastguard Worker m->SleepState == SleepState_Awake ? "Awake" :
4483*48a54d36SAndroid Build Coastguard Worker m->SleepState == SleepState_Transferring ? "Transferring" :
4484*48a54d36SAndroid Build Coastguard Worker m->SleepState == SleepState_Sleeping ? "Sleeping" : "?",
4485*48a54d36SAndroid Build Coastguard Worker m->SleepSeqNum);
4486*48a54d36SAndroid Build Coastguard Worker
4487*48a54d36SAndroid Build Coastguard Worker if (!m->SPSSocket) LogMsgNoIdent("Not offering Sleep Proxy Service");
4488*48a54d36SAndroid Build Coastguard Worker else LogMsgNoIdent("Offering Sleep Proxy Service: %#s", m->SPSRecords.RR_SRV.resrec.name->c);
4489*48a54d36SAndroid Build Coastguard Worker
4490*48a54d36SAndroid Build Coastguard Worker if (m->ProxyRecords == ProxyA + ProxyD) LogMsgNoIdent("ProxyRecords: %d + %d = %d", ProxyA, ProxyD, ProxyA + ProxyD);
4491*48a54d36SAndroid Build Coastguard Worker else LogMsgNoIdent("ProxyRecords: MISMATCH %d + %d = %d ≠ %d", ProxyA, ProxyD, ProxyA + ProxyD, m->ProxyRecords);
4492*48a54d36SAndroid Build Coastguard Worker
4493*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("------ Auto Browse Domains -----");
4494*48a54d36SAndroid Build Coastguard Worker if (!AutoBrowseDomains) LogMsgNoIdent("<None>");
4495*48a54d36SAndroid Build Coastguard Worker else for (d=AutoBrowseDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c);
4496*48a54d36SAndroid Build Coastguard Worker
4497*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--- Auto Registration Domains --");
4498*48a54d36SAndroid Build Coastguard Worker if (!AutoRegistrationDomains) LogMsgNoIdent("<None>");
4499*48a54d36SAndroid Build Coastguard Worker else for (d=AutoRegistrationDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c);
4500*48a54d36SAndroid Build Coastguard Worker
4501*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("--- Search Domains --");
4502*48a54d36SAndroid Build Coastguard Worker if (!SearchList) LogMsgNoIdent("<None>");
4503*48a54d36SAndroid Build Coastguard Worker else
4504*48a54d36SAndroid Build Coastguard Worker {
4505*48a54d36SAndroid Build Coastguard Worker for (s=SearchList; s; s=s->next)
4506*48a54d36SAndroid Build Coastguard Worker {
4507*48a54d36SAndroid Build Coastguard Worker char *ifname = InterfaceNameForID(m, s->InterfaceID);
4508*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("%##s %s", s->domain.c, ifname ? ifname : "");
4509*48a54d36SAndroid Build Coastguard Worker }
4510*48a54d36SAndroid Build Coastguard Worker }
4511*48a54d36SAndroid Build Coastguard Worker
4512*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("---- Task Scheduling Timers ----");
4513*48a54d36SAndroid Build Coastguard Worker
4514*48a54d36SAndroid Build Coastguard Worker if (!m->NewQuestions)
4515*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewQuestion <NONE>");
4516*48a54d36SAndroid Build Coastguard Worker else
4517*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewQuestion DelayAnswering %d %d %##s (%s)",
4518*48a54d36SAndroid Build Coastguard Worker m->NewQuestions->DelayAnswering, m->NewQuestions->DelayAnswering-now,
4519*48a54d36SAndroid Build Coastguard Worker m->NewQuestions->qname.c, DNSTypeName(m->NewQuestions->qtype));
4520*48a54d36SAndroid Build Coastguard Worker
4521*48a54d36SAndroid Build Coastguard Worker if (!m->NewLocalOnlyQuestions)
4522*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewLocalOnlyQuestions <NONE>");
4523*48a54d36SAndroid Build Coastguard Worker else
4524*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewLocalOnlyQuestions %##s (%s)",
4525*48a54d36SAndroid Build Coastguard Worker m->NewLocalOnlyQuestions->qname.c, DNSTypeName(m->NewLocalOnlyQuestions->qtype));
4526*48a54d36SAndroid Build Coastguard Worker
4527*48a54d36SAndroid Build Coastguard Worker if (!m->NewLocalRecords)
4528*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewLocalRecords <NONE>");
4529*48a54d36SAndroid Build Coastguard Worker else
4530*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("NewLocalRecords %02X %s", m->NewLocalRecords->resrec.RecordType, ARDisplayString(m, m->NewLocalRecords));
4531*48a54d36SAndroid Build Coastguard Worker
4532*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("SPSProxyListChanged%s", m->SPSProxyListChanged ? "" : " <NONE>");
4533*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("LocalRemoveEvents%s", m->LocalRemoveEvents ? "" : " <NONE>");
4534*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->RegisterAutoTunnel6 %08X", m->RegisterAutoTunnel6);
4535*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->AutoTunnelRelayAddrIn %.16a", &m->AutoTunnelRelayAddrIn);
4536*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->AutoTunnelRelayAddrOut %.16a", &m->AutoTunnelRelayAddrOut);
4537*48a54d36SAndroid Build Coastguard Worker
4538*48a54d36SAndroid Build Coastguard Worker #define LogTimer(MSG,T) LogMsgNoIdent( MSG " %08X %11d %08X %11d", (T), (T), (T)-now, (T)-now)
4539*48a54d36SAndroid Build Coastguard Worker
4540*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent(" ABS (hex) ABS (dec) REL (hex) REL (dec)");
4541*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->timenow %08X %11d", now, now);
4542*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("m->timenow_adjust %08X %11d", m->timenow_adjust, m->timenow_adjust);
4543*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledEvent ", m->NextScheduledEvent);
4544*48a54d36SAndroid Build Coastguard Worker
4545*48a54d36SAndroid Build Coastguard Worker #ifndef UNICAST_DISABLED
4546*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextuDNSEvent ", m->NextuDNSEvent);
4547*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextSRVUpdate ", m->NextSRVUpdate);
4548*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledNATOp ", m->NextScheduledNATOp);
4549*48a54d36SAndroid Build Coastguard Worker LogTimer("m->retryGetAddr ", m->retryGetAddr);
4550*48a54d36SAndroid Build Coastguard Worker #endif
4551*48a54d36SAndroid Build Coastguard Worker
4552*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextCacheCheck ", m->NextCacheCheck);
4553*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledSPS ", m->NextScheduledSPS);
4554*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledSPRetry ", m->NextScheduledSPRetry);
4555*48a54d36SAndroid Build Coastguard Worker LogTimer("m->DelaySleep ", m->DelaySleep);
4556*48a54d36SAndroid Build Coastguard Worker
4557*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledQuery ", m->NextScheduledQuery);
4558*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledProbe ", m->NextScheduledProbe);
4559*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledResponse", m->NextScheduledResponse);
4560*48a54d36SAndroid Build Coastguard Worker
4561*48a54d36SAndroid Build Coastguard Worker LogTimer("m->SuppressSending ", m->SuppressSending);
4562*48a54d36SAndroid Build Coastguard Worker LogTimer("m->SuppressProbes ", m->SuppressProbes);
4563*48a54d36SAndroid Build Coastguard Worker LogTimer("m->ProbeFailTime ", m->ProbeFailTime);
4564*48a54d36SAndroid Build Coastguard Worker LogTimer("m->DelaySleep ", m->DelaySleep);
4565*48a54d36SAndroid Build Coastguard Worker LogTimer("m->SleepLimit ", m->SleepLimit);
4566*48a54d36SAndroid Build Coastguard Worker LogTimer("m->NextScheduledStopTime ", m->NextScheduledStopTime);
4567*48a54d36SAndroid Build Coastguard Worker }
4568*48a54d36SAndroid Build Coastguard Worker
4569*48a54d36SAndroid Build Coastguard Worker #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING
uds_validatelists(void)4570*48a54d36SAndroid Build Coastguard Worker mDNSexport void uds_validatelists(void)
4571*48a54d36SAndroid Build Coastguard Worker {
4572*48a54d36SAndroid Build Coastguard Worker const request_state *req, *p;
4573*48a54d36SAndroid Build Coastguard Worker for (req = all_requests; req; req=req->next)
4574*48a54d36SAndroid Build Coastguard Worker {
4575*48a54d36SAndroid Build Coastguard Worker if (req->next == (request_state *)~0 || (req->sd < 0 && req->sd != -2))
4576*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS request list: %p is garbage (%d)", req, req->sd);
4577*48a54d36SAndroid Build Coastguard Worker
4578*48a54d36SAndroid Build Coastguard Worker if (req->primary == req)
4579*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS request list: req->primary should not point to self %p/%d", req, req->sd);
4580*48a54d36SAndroid Build Coastguard Worker
4581*48a54d36SAndroid Build Coastguard Worker if (req->primary && req->replies)
4582*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS request list: Subordinate request %p/%d/%p should not have replies (%p)",
4583*48a54d36SAndroid Build Coastguard Worker req, req->sd, req->primary && req->replies);
4584*48a54d36SAndroid Build Coastguard Worker
4585*48a54d36SAndroid Build Coastguard Worker p = req->primary;
4586*48a54d36SAndroid Build Coastguard Worker if ((long)p & 3)
4587*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS request list: req %p primary %p is misaligned (%d)", req, p, req->sd);
4588*48a54d36SAndroid Build Coastguard Worker else if (p && (p->next == (request_state *)~0 || (p->sd < 0 && p->sd != -2)))
4589*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS request list: req %p primary %p is garbage (%d)", req, p, p->sd);
4590*48a54d36SAndroid Build Coastguard Worker
4591*48a54d36SAndroid Build Coastguard Worker reply_state *rep;
4592*48a54d36SAndroid Build Coastguard Worker for (rep = req->replies; rep; rep=rep->next)
4593*48a54d36SAndroid Build Coastguard Worker if (rep->next == (reply_state *)~0)
4594*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS req->replies: %p is garbage", rep);
4595*48a54d36SAndroid Build Coastguard Worker
4596*48a54d36SAndroid Build Coastguard Worker if (req->terminate == connection_termination)
4597*48a54d36SAndroid Build Coastguard Worker {
4598*48a54d36SAndroid Build Coastguard Worker registered_record_entry *r;
4599*48a54d36SAndroid Build Coastguard Worker for (r = req->u.reg_recs; r; r=r->next)
4600*48a54d36SAndroid Build Coastguard Worker if (r->next == (registered_record_entry *)~0)
4601*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS req->u.reg_recs: %p is garbage", r);
4602*48a54d36SAndroid Build Coastguard Worker }
4603*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == regservice_termination_callback)
4604*48a54d36SAndroid Build Coastguard Worker {
4605*48a54d36SAndroid Build Coastguard Worker service_instance *s;
4606*48a54d36SAndroid Build Coastguard Worker for (s = req->u.servicereg.instances; s; s=s->next)
4607*48a54d36SAndroid Build Coastguard Worker if (s->next == (service_instance *)~0)
4608*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS req->u.servicereg.instances: %p is garbage", s);
4609*48a54d36SAndroid Build Coastguard Worker }
4610*48a54d36SAndroid Build Coastguard Worker else if (req->terminate == browse_termination_callback)
4611*48a54d36SAndroid Build Coastguard Worker {
4612*48a54d36SAndroid Build Coastguard Worker browser_t *b;
4613*48a54d36SAndroid Build Coastguard Worker for (b = req->u.browser.browsers; b; b=b->next)
4614*48a54d36SAndroid Build Coastguard Worker if (b->next == (browser_t *)~0)
4615*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("UDS req->u.browser.browsers: %p is garbage", b);
4616*48a54d36SAndroid Build Coastguard Worker }
4617*48a54d36SAndroid Build Coastguard Worker }
4618*48a54d36SAndroid Build Coastguard Worker
4619*48a54d36SAndroid Build Coastguard Worker DNameListElem *d;
4620*48a54d36SAndroid Build Coastguard Worker for (d = SCPrefBrowseDomains; d; d=d->next)
4621*48a54d36SAndroid Build Coastguard Worker if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
4622*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("SCPrefBrowseDomains: %p is garbage (%d)", d, d->name.c[0]);
4623*48a54d36SAndroid Build Coastguard Worker
4624*48a54d36SAndroid Build Coastguard Worker ARListElem *b;
4625*48a54d36SAndroid Build Coastguard Worker for (b = LocalDomainEnumRecords; b; b=b->next)
4626*48a54d36SAndroid Build Coastguard Worker if (b->next == (ARListElem *)~0 || b->ar.resrec.name->c[0] > 63)
4627*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("LocalDomainEnumRecords: %p is garbage (%d)", b, b->ar.resrec.name->c[0]);
4628*48a54d36SAndroid Build Coastguard Worker
4629*48a54d36SAndroid Build Coastguard Worker for (d = AutoBrowseDomains; d; d=d->next)
4630*48a54d36SAndroid Build Coastguard Worker if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
4631*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("AutoBrowseDomains: %p is garbage (%d)", d, d->name.c[0]);
4632*48a54d36SAndroid Build Coastguard Worker
4633*48a54d36SAndroid Build Coastguard Worker for (d = AutoRegistrationDomains; d; d=d->next)
4634*48a54d36SAndroid Build Coastguard Worker if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63)
4635*48a54d36SAndroid Build Coastguard Worker LogMemCorruption("AutoRegistrationDomains: %p is garbage (%d)", d, d->name.c[0]);
4636*48a54d36SAndroid Build Coastguard Worker }
4637*48a54d36SAndroid Build Coastguard Worker #endif // APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING
4638*48a54d36SAndroid Build Coastguard Worker
send_msg(request_state * const req)4639*48a54d36SAndroid Build Coastguard Worker mDNSlocal int send_msg(request_state *const req)
4640*48a54d36SAndroid Build Coastguard Worker {
4641*48a54d36SAndroid Build Coastguard Worker reply_state *const rep = req->replies; // Send the first waiting reply
4642*48a54d36SAndroid Build Coastguard Worker ssize_t nwriten;
4643*48a54d36SAndroid Build Coastguard Worker if (req->no_reply) return(t_complete);
4644*48a54d36SAndroid Build Coastguard Worker
4645*48a54d36SAndroid Build Coastguard Worker ConvertHeaderBytes(rep->mhdr);
4646*48a54d36SAndroid Build Coastguard Worker nwriten = send(req->sd, (char *)&rep->mhdr + rep->nwriten, rep->totallen - rep->nwriten, 0);
4647*48a54d36SAndroid Build Coastguard Worker ConvertHeaderBytes(rep->mhdr);
4648*48a54d36SAndroid Build Coastguard Worker
4649*48a54d36SAndroid Build Coastguard Worker if (nwriten < 0)
4650*48a54d36SAndroid Build Coastguard Worker {
4651*48a54d36SAndroid Build Coastguard Worker if (dnssd_errno == dnssd_EINTR || dnssd_errno == dnssd_EWOULDBLOCK) nwriten = 0;
4652*48a54d36SAndroid Build Coastguard Worker else
4653*48a54d36SAndroid Build Coastguard Worker {
4654*48a54d36SAndroid Build Coastguard Worker #if !defined(PLATFORM_NO_EPIPE)
4655*48a54d36SAndroid Build Coastguard Worker if (dnssd_errno == EPIPE)
4656*48a54d36SAndroid Build Coastguard Worker return(req->ts = t_terminated);
4657*48a54d36SAndroid Build Coastguard Worker else
4658*48a54d36SAndroid Build Coastguard Worker #endif
4659*48a54d36SAndroid Build Coastguard Worker {
4660*48a54d36SAndroid Build Coastguard Worker LogMsg("send_msg ERROR: failed to write %d of %d bytes to fd %d errno %d (%s)",
4661*48a54d36SAndroid Build Coastguard Worker rep->totallen - rep->nwriten, rep->totallen, req->sd, dnssd_errno, dnssd_strerror(dnssd_errno));
4662*48a54d36SAndroid Build Coastguard Worker return(t_error);
4663*48a54d36SAndroid Build Coastguard Worker }
4664*48a54d36SAndroid Build Coastguard Worker }
4665*48a54d36SAndroid Build Coastguard Worker }
4666*48a54d36SAndroid Build Coastguard Worker rep->nwriten += nwriten;
4667*48a54d36SAndroid Build Coastguard Worker return (rep->nwriten == rep->totallen) ? t_complete : t_morecoming;
4668*48a54d36SAndroid Build Coastguard Worker }
4669*48a54d36SAndroid Build Coastguard Worker
udsserver_idle(mDNSs32 nextevent)4670*48a54d36SAndroid Build Coastguard Worker mDNSexport mDNSs32 udsserver_idle(mDNSs32 nextevent)
4671*48a54d36SAndroid Build Coastguard Worker {
4672*48a54d36SAndroid Build Coastguard Worker mDNSs32 now = mDNS_TimeNow(&mDNSStorage);
4673*48a54d36SAndroid Build Coastguard Worker request_state **req = &all_requests;
4674*48a54d36SAndroid Build Coastguard Worker
4675*48a54d36SAndroid Build Coastguard Worker while (*req)
4676*48a54d36SAndroid Build Coastguard Worker {
4677*48a54d36SAndroid Build Coastguard Worker request_state *const r = *req;
4678*48a54d36SAndroid Build Coastguard Worker
4679*48a54d36SAndroid Build Coastguard Worker if (r->terminate == resolve_termination_callback)
4680*48a54d36SAndroid Build Coastguard Worker if (r->u.resolve.ReportTime && now - r->u.resolve.ReportTime >= 0)
4681*48a54d36SAndroid Build Coastguard Worker {
4682*48a54d36SAndroid Build Coastguard Worker r->u.resolve.ReportTime = 0;
4683*48a54d36SAndroid Build Coastguard Worker LogMsgNoIdent("Client application bug: DNSServiceResolve(%##s) active for over two minutes. "
4684*48a54d36SAndroid Build Coastguard Worker "This places considerable burden on the network.", r->u.resolve.qsrv.qname.c);
4685*48a54d36SAndroid Build Coastguard Worker }
4686*48a54d36SAndroid Build Coastguard Worker
4687*48a54d36SAndroid Build Coastguard Worker // Note: Only primary req's have reply lists, not subordinate req's.
4688*48a54d36SAndroid Build Coastguard Worker while (r->replies) // Send queued replies
4689*48a54d36SAndroid Build Coastguard Worker {
4690*48a54d36SAndroid Build Coastguard Worker transfer_state result;
4691*48a54d36SAndroid Build Coastguard Worker if (r->replies->next) r->replies->rhdr->flags |= dnssd_htonl(kDNSServiceFlagsMoreComing);
4692*48a54d36SAndroid Build Coastguard Worker result = send_msg(r); // Returns t_morecoming if buffer full because client is not reading
4693*48a54d36SAndroid Build Coastguard Worker if (result == t_complete)
4694*48a54d36SAndroid Build Coastguard Worker {
4695*48a54d36SAndroid Build Coastguard Worker reply_state *fptr = r->replies;
4696*48a54d36SAndroid Build Coastguard Worker r->replies = r->replies->next;
4697*48a54d36SAndroid Build Coastguard Worker freeL("reply_state/udsserver_idle", fptr);
4698*48a54d36SAndroid Build Coastguard Worker r->time_blocked = 0; // reset failure counter after successful send
4699*48a54d36SAndroid Build Coastguard Worker r->unresponsiveness_reports = 0;
4700*48a54d36SAndroid Build Coastguard Worker continue;
4701*48a54d36SAndroid Build Coastguard Worker }
4702*48a54d36SAndroid Build Coastguard Worker else if (result == t_terminated || result == t_error)
4703*48a54d36SAndroid Build Coastguard Worker {
4704*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Could not write data to client because of error - aborting connection", r->sd);
4705*48a54d36SAndroid Build Coastguard Worker LogClientInfo(&mDNSStorage, r);
4706*48a54d36SAndroid Build Coastguard Worker abort_request(r);
4707*48a54d36SAndroid Build Coastguard Worker }
4708*48a54d36SAndroid Build Coastguard Worker break;
4709*48a54d36SAndroid Build Coastguard Worker }
4710*48a54d36SAndroid Build Coastguard Worker
4711*48a54d36SAndroid Build Coastguard Worker if (r->replies) // If we failed to send everything, check our time_blocked timer
4712*48a54d36SAndroid Build Coastguard Worker {
4713*48a54d36SAndroid Build Coastguard Worker if (nextevent - now > mDNSPlatformOneSecond) nextevent = now + mDNSPlatformOneSecond;
4714*48a54d36SAndroid Build Coastguard Worker
4715*48a54d36SAndroid Build Coastguard Worker if (mDNSStorage.SleepState != SleepState_Awake) r->time_blocked = 0;
4716*48a54d36SAndroid Build Coastguard Worker else if (!r->time_blocked) r->time_blocked = NonZeroTime(now);
4717*48a54d36SAndroid Build Coastguard Worker else if (now - r->time_blocked >= 10 * mDNSPlatformOneSecond * (r->unresponsiveness_reports+1))
4718*48a54d36SAndroid Build Coastguard Worker {
4719*48a54d36SAndroid Build Coastguard Worker int num = 0;
4720*48a54d36SAndroid Build Coastguard Worker struct reply_state *x = r->replies;
4721*48a54d36SAndroid Build Coastguard Worker while (x) { num++; x=x->next; }
4722*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Could not write data to client after %ld seconds, %d repl%s waiting",
4723*48a54d36SAndroid Build Coastguard Worker r->sd, (now - r->time_blocked) / mDNSPlatformOneSecond, num, num == 1 ? "y" : "ies");
4724*48a54d36SAndroid Build Coastguard Worker if (++r->unresponsiveness_reports >= 60)
4725*48a54d36SAndroid Build Coastguard Worker {
4726*48a54d36SAndroid Build Coastguard Worker LogMsg("%3d: Client unresponsive; aborting connection", r->sd);
4727*48a54d36SAndroid Build Coastguard Worker LogClientInfo(&mDNSStorage, r);
4728*48a54d36SAndroid Build Coastguard Worker abort_request(r);
4729*48a54d36SAndroid Build Coastguard Worker }
4730*48a54d36SAndroid Build Coastguard Worker }
4731*48a54d36SAndroid Build Coastguard Worker }
4732*48a54d36SAndroid Build Coastguard Worker
4733*48a54d36SAndroid Build Coastguard Worker if (!dnssd_SocketValid(r->sd)) // If this request is finished, unlink it from the list and free the memory
4734*48a54d36SAndroid Build Coastguard Worker {
4735*48a54d36SAndroid Build Coastguard Worker // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree()
4736*48a54d36SAndroid Build Coastguard Worker *req = r->next;
4737*48a54d36SAndroid Build Coastguard Worker freeL("request_state/udsserver_idle", r);
4738*48a54d36SAndroid Build Coastguard Worker }
4739*48a54d36SAndroid Build Coastguard Worker else
4740*48a54d36SAndroid Build Coastguard Worker req = &r->next;
4741*48a54d36SAndroid Build Coastguard Worker }
4742*48a54d36SAndroid Build Coastguard Worker return nextevent;
4743*48a54d36SAndroid Build Coastguard Worker }
4744*48a54d36SAndroid Build Coastguard Worker
4745*48a54d36SAndroid Build Coastguard Worker struct CompileTimeAssertionChecks_uds_daemon
4746*48a54d36SAndroid Build Coastguard Worker {
4747*48a54d36SAndroid Build Coastguard Worker // Check our structures are reasonable sizes. Including overly-large buffers, or embedding
4748*48a54d36SAndroid Build Coastguard Worker // other overly-large structures instead of having a pointer to them, can inadvertently
4749*48a54d36SAndroid Build Coastguard Worker // cause structure sizes (and therefore memory usage) to balloon unreasonably.
4750*48a54d36SAndroid Build Coastguard Worker char sizecheck_request_state [(sizeof(request_state) <= 1784) ? 1 : -1];
4751*48a54d36SAndroid Build Coastguard Worker char sizecheck_registered_record_entry[(sizeof(registered_record_entry) <= 60) ? 1 : -1];
4752*48a54d36SAndroid Build Coastguard Worker char sizecheck_service_instance [(sizeof(service_instance) <= 6552) ? 1 : -1];
4753*48a54d36SAndroid Build Coastguard Worker char sizecheck_browser_t [(sizeof(browser_t) <= 1050) ? 1 : -1];
4754*48a54d36SAndroid Build Coastguard Worker char sizecheck_reply_hdr [(sizeof(reply_hdr) <= 12) ? 1 : -1];
4755*48a54d36SAndroid Build Coastguard Worker char sizecheck_reply_state [(sizeof(reply_state) <= 64) ? 1 : -1];
4756*48a54d36SAndroid Build Coastguard Worker };
4757