1 /* -*- Mode: C; tab-width: 4 -*- 2 * 3 * Copyright (c) 2003-2015 Apple Inc. All rights reserved. 4 * 5 * Licensed under the Apache License, Version 2.0 (the "License"); 6 * you may not use this file except in compliance with the License. 7 * You may obtain a copy of the License at 8 * 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 */ 17 18 #if defined(_WIN32) 19 #include <process.h> 20 #define usleep(X) Sleep(((X)+999)/1000) 21 #else 22 #include <fcntl.h> 23 #include <errno.h> 24 #include <sys/ioctl.h> 25 #include <sys/types.h> 26 #include <sys/time.h> 27 #include <sys/resource.h> 28 #endif 29 30 #include <stdlib.h> 31 #include <stdio.h> 32 33 #include "mDNSEmbeddedAPI.h" 34 #include "DNSCommon.h" 35 #include "uDNS.h" 36 #include "uds_daemon.h" 37 #include "dns_sd_internal.h" 38 39 // Normally we append search domains only for queries with a single label that are not 40 // fully qualified. This can be overridden to apply search domains for queries (that are 41 // not fully qualified) with any number of labels e.g., moon, moon.cs, moon.cs.be, etc. 42 mDNSBool AlwaysAppendSearchDomains = mDNSfalse; 43 44 // Apple-specific functionality, not required for other platforms 45 #if APPLE_OSX_mDNSResponder 46 #include <sys/ucred.h> 47 #ifndef PID_FILE 48 #define PID_FILE "" 49 #endif 50 #endif 51 52 #ifdef LOCAL_PEEREPID 53 #include <sys/un.h> // for LOCAL_PEEREPID 54 #include <sys/socket.h> // for getsockopt 55 #include <sys/proc_info.h> // for struct proc_bsdshortinfo 56 #include <libproc.h> // for proc_pidinfo() 57 #endif //LOCAL_PEEREPID 58 59 #ifdef UNIT_TEST 60 #include "unittest.h" 61 #endif 62 63 #if APPLE_OSX_mDNSResponder 64 #include <WebFilterDNS/WebFilterDNS.h> 65 #include "BLE.h" 66 67 #if !NO_WCF 68 69 int WCFIsServerRunning(WCFConnection *conn) __attribute__((weak_import)); 70 int WCFNameResolvesToAddr(WCFConnection *conn, char* domainName, struct sockaddr* address, uid_t userid) __attribute__((weak_import)); 71 int WCFNameResolvesToName(WCFConnection *conn, char* fromName, char* toName, uid_t userid) __attribute__((weak_import)); 72 73 // Do we really need to define a macro for "if"? 74 #define CHECK_WCF_FUNCTION(X) if (X) 75 #endif // ! NO_WCF 76 77 #else 78 #define NO_WCF 1 79 #endif // APPLE_OSX_mDNSResponder 80 81 // User IDs 0-500 are system-wide processes, not actual users in the usual sense 82 // User IDs for real user accounts start at 501 and count up from there 83 #define SystemUID(X) ((X) <= 500) 84 85 #define MAX_ANONYMOUS_DATA 256 86 87 // *************************************************************************** 88 #if COMPILER_LIKES_PRAGMA_MARK 89 #pragma mark - 90 #pragma mark - Globals 91 #endif 92 93 // globals 94 mDNSexport mDNS mDNSStorage; 95 mDNSexport const char ProgramName[] = "mDNSResponder"; 96 97 #if defined(USE_TCP_LOOPBACK) 98 static char* boundPath = NULL; 99 #else 100 static char* boundPath = MDNS_UDS_SERVERPATH; 101 #endif 102 #if DEBUG 103 #define MDNS_UDS_SERVERPATH_DEBUG "/var/tmp/mDNSResponder" 104 #endif 105 static dnssd_sock_t listenfd = dnssd_InvalidSocket; 106 static request_state *all_requests = NULL; 107 #ifdef LOCAL_PEEREPID 108 struct proc_bsdshortinfo proc; 109 #endif //LOCAL_PEEREPID 110 mDNSlocal void set_peer_pid(request_state *request); 111 mDNSlocal void LogMcastClientInfo(request_state *req); 112 mDNSlocal void GetMcastClients(request_state *req); 113 static mDNSu32 mcount; // tracks the current active mcast operations for McastLogging 114 static mDNSu32 i_mcount; // sets mcount when McastLogging is enabled(PROF signal is sent) 115 static mDNSu32 n_mrecords; // tracks the current active mcast records for McastLogging 116 static mDNSu32 n_mquests; // tracks the current active mcast questions for McastLogging 117 118 119 #if TARGET_OS_EMBEDDED 120 mDNSu32 curr_num_regservices = 0; 121 mDNSu32 max_num_regservices = 0; 122 #endif 123 124 125 // Note asymmetry here between registration and browsing. 126 // For service registrations we only automatically register in domains that explicitly appear in local configuration data 127 // (so AutoRegistrationDomains could equally well be called SCPrefRegDomains) 128 // For service browsing we also learn automatic browsing domains from the network, so for that case we have: 129 // 1. SCPrefBrowseDomains (local configuration data) 130 // 2. LocalDomainEnumRecords (locally-generated local-only PTR records -- equivalent to slElem->AuthRecs in uDNS.c) 131 // 3. AutoBrowseDomains, which is populated by tracking add/rmv events in AutomaticBrowseDomainChange, the callback function for our mDNS_GetDomains call. 132 // By creating and removing our own LocalDomainEnumRecords, we trigger AutomaticBrowseDomainChange callbacks just like domains learned from the network would. 133 134 mDNSexport DNameListElem *AutoRegistrationDomains; // Domains where we automatically register for empty-string registrations 135 136 static DNameListElem *SCPrefBrowseDomains; // List of automatic browsing domains read from SCPreferences for "empty string" browsing 137 static ARListElem *LocalDomainEnumRecords; // List of locally-generated PTR records to augment those we learn from the network 138 mDNSexport DNameListElem *AutoBrowseDomains; // List created from those local-only PTR records plus records we get from the network 139 140 #define MSG_PAD_BYTES 5 // pad message buffer (read from client) with n zero'd bytes to guarantee 141 // n get_string() calls w/o buffer overrun 142 // initialization, setup/teardown functions 143 144 // If a platform specifies its own PID file name, we use that 145 #ifndef PID_FILE 146 #define PID_FILE "/var/run/mDNSResponder.pid" 147 #endif 148 149 mDNSlocal char *AnonDataToString(const mDNSu8 *ad, int adlen, char *adstr, int adstrlen); 150 151 // *************************************************************************** 152 #if COMPILER_LIKES_PRAGMA_MARK 153 #pragma mark - 154 #pragma mark - General Utility Functions 155 #endif 156 157 mDNSlocal void FatalError(char *errmsg) 158 { 159 LogMsg("%s: %s", errmsg, dnssd_strerror(dnssd_errno)); 160 abort(); 161 } 162 163 mDNSlocal mDNSu32 dnssd_htonl(mDNSu32 l) 164 { 165 mDNSu32 ret; 166 char *data = (char*) &ret; 167 put_uint32(l, &data); 168 return ret; 169 } 170 171 // hack to search-replace perror's to LogMsg's 172 mDNSlocal void my_perror(char *errmsg) 173 { 174 LogMsg("%s: %d (%s)", errmsg, dnssd_errno, dnssd_strerror(dnssd_errno)); 175 } 176 177 // Throttled version of my_perror: Logs once every 250 msgs 178 mDNSlocal void my_throttled_perror(char *err_msg) 179 { 180 static int uds_throttle_count = 0; 181 if ((uds_throttle_count++ % 250) == 0) 182 my_perror(err_msg); 183 } 184 185 // LogMcastQuestion/LogMcastQ should be called after the DNSQuestion struct is initialized(especially for q->TargetQID) 186 // Hence all calls are made after mDNS_StartQuery()/mDNS_StopQuery()/mDNS_StopBrowse() is called. 187 mDNSlocal void LogMcastQuestion(const DNSQuestion *const q, request_state *req, q_state status) 188 { 189 if (mDNSOpaque16IsZero(q->TargetQID)) // Check for Mcast Query 190 { 191 mDNSBool mflag = mDNSfalse; 192 if (status == q_start) 193 { 194 if (++mcount == 1) 195 mflag = mDNStrue; 196 } 197 else 198 { 199 mcount--; 200 } 201 LogMcast("%s: %##s (%s) (%s) Client(%d)[%s]", status ? "+Question" : "-Question", q->qname.c, DNSTypeName(q->qtype), 202 q->InterfaceID == mDNSInterface_LocalOnly ? "lo" : 203 q->InterfaceID == mDNSInterface_P2P ? "p2p" : 204 q->InterfaceID == mDNSInterface_BLE ? "BLE" : 205 q->InterfaceID == mDNSInterface_Any ? "any" : InterfaceNameForID(&mDNSStorage, q->InterfaceID), 206 req->process_id, req->pid_name); 207 LogMcastStateInfo(mflag, mDNSfalse, mDNSfalse); 208 } 209 return; 210 } 211 212 // LogMcastService/LogMcastS should be called after the AuthRecord struct is initialized 213 // Hence all calls are made after mDNS_Register()/ just before mDNS_Deregister() 214 mDNSlocal void LogMcastService(const AuthRecord *const ar, request_state *req, reg_state status) 215 { 216 if (!AuthRecord_uDNS(ar)) // Check for Mcast Service 217 { 218 mDNSBool mflag = mDNSfalse; 219 if (status == reg_start) 220 { 221 if (++mcount == 1) 222 mflag = mDNStrue; 223 } 224 else 225 { 226 mcount--; 227 } 228 LogMcast("%s: %##s (%s) (%s) Client(%d)[%s]", status ? "+Service" : "-Service", ar->resrec.name->c, DNSTypeName(ar->resrec.rrtype), 229 ar->resrec.InterfaceID == mDNSInterface_LocalOnly ? "lo" : 230 ar->resrec.InterfaceID == mDNSInterface_P2P ? "p2p" : 231 ar->resrec.InterfaceID == mDNSInterface_BLE ? "BLE" : 232 ar->resrec.InterfaceID == mDNSInterface_Any ? "all" : InterfaceNameForID(&mDNSStorage, ar->resrec.InterfaceID), 233 req->process_id, req->pid_name); 234 LogMcastStateInfo(mflag, mDNSfalse, mDNSfalse); 235 } 236 return; 237 } 238 239 // For complete Mcast State Log, pass mDNStrue to mstatelog in LogMcastStateInfo() 240 mDNSexport void LogMcastStateInfo(mDNSBool mflag, mDNSBool start, mDNSBool mstatelog) 241 { 242 mDNS *const m = &mDNSStorage; 243 if (!mstatelog) 244 { 245 if (!all_requests) 246 { 247 LogMcastNoIdent("<None>"); 248 } 249 else 250 { 251 request_state *req, *r; 252 for (req = all_requests; req; req=req->next) 253 { 254 if (req->primary) // If this is a subbordinate operation, check that the parent is in the list 255 { 256 for (r = all_requests; r && r != req; r=r->next) 257 if (r == req->primary) 258 goto foundpar; 259 } 260 // For non-subbordinate operations, and subbordinate operations that have lost their parent, write out their info 261 GetMcastClients(req); 262 foundpar:; 263 } 264 LogMcastNoIdent("--- MCAST RECORDS COUNT[%d] MCAST QUESTIONS COUNT[%d] ---", n_mrecords, n_mquests); 265 n_mrecords = n_mquests = 0; // Reset the values 266 } 267 } 268 else 269 { 270 static mDNSu32 i_mpktnum; 271 i_mcount = 0; 272 if (start) 273 mcount = 0; 274 // mcount is initialized to 0 when the PROF signal is sent since mcount could have 275 // wrong value if MulticastLogging is disabled and then re-enabled 276 LogMcastNoIdent("--- START MCAST STATE LOG ---"); 277 if (!all_requests) 278 { 279 mcount = 0; 280 LogMcastNoIdent("<None>"); 281 } 282 else 283 { 284 request_state *req, *r; 285 for (req = all_requests; req; req=req->next) 286 { 287 if (req->primary) // If this is a subbordinate operation, check that the parent is in the list 288 { 289 for (r = all_requests; r && r != req; r=r->next) 290 if (r == req->primary) 291 goto foundparent; 292 LogMcastNoIdent("%3d: Orphan operation; parent not found in request list", req->sd); 293 } 294 // For non-subbordinate operations, and subbordinate operations that have lost their parent, write out their info 295 LogMcastClientInfo(req); 296 foundparent:; 297 } 298 if(!mcount) // To initially set mcount 299 mcount = i_mcount; 300 } 301 if (mcount == 0) 302 { 303 i_mpktnum = m->MPktNum; 304 LogMcastNoIdent("--- MCOUNT[%d]: IMPKTNUM[%d] ---", mcount, i_mpktnum); 305 } 306 if (mflag) 307 LogMcastNoIdent("--- MCOUNT[%d]: CMPKTNUM[%d] - IMPKTNUM[%d] = [%d]PKTS ---", mcount, m->MPktNum, i_mpktnum, (m->MPktNum - i_mpktnum)); 308 LogMcastNoIdent("--- END MCAST STATE LOG ---"); 309 } 310 } 311 312 mDNSlocal void abort_request(request_state *req) 313 { 314 if (req->terminate == (req_termination_fn) ~0) 315 { LogMsg("abort_request: ERROR: Attempt to abort operation %p with req->terminate %p", req, req->terminate); return; } 316 317 // First stop whatever mDNSCore operation we were doing 318 // If this is actually a shared connection operation, then its req->terminate function will scan 319 // the all_requests list and terminate any subbordinate operations sharing this file descriptor 320 if (req->terminate) req->terminate(req); 321 322 if (!dnssd_SocketValid(req->sd)) 323 { LogMsg("abort_request: ERROR: Attempt to abort operation %p with invalid fd %d", req, req->sd); return; } 324 325 // Now, if this request_state is not subordinate to some other primary, close file descriptor and discard replies 326 if (!req->primary) 327 { 328 if (req->errsd != req->sd) LogOperation("%3d: Removing FD and closing errsd %d", req->sd, req->errsd); 329 else LogOperation("%3d: Removing FD", req->sd); 330 udsSupportRemoveFDFromEventLoop(req->sd, req->platform_data); // Note: This also closes file descriptor req->sd for us 331 if (req->errsd != req->sd) { dnssd_close(req->errsd); req->errsd = req->sd; } 332 333 while (req->replies) // free pending replies 334 { 335 reply_state *ptr = req->replies; 336 req->replies = req->replies->next; 337 freeL("reply_state (abort)", ptr); 338 } 339 } 340 341 // Set req->sd to something invalid, so that udsserver_idle knows to unlink and free this structure 342 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 343 // Don't use dnssd_InvalidSocket (-1) because that's the sentinel value MACOSX_MDNS_MALLOC_DEBUGGING uses 344 // for detecting when the memory for an object is inadvertently freed while the object is still on some list 345 req->sd = req->errsd = -2; 346 #else 347 req->sd = req->errsd = dnssd_InvalidSocket; 348 #endif 349 // We also set req->terminate to a bogus value so we know if abort_request() gets called again for this request 350 req->terminate = (req_termination_fn) ~0; 351 } 352 353 #if DEBUG 354 mDNSexport void SetDebugBoundPath(void) 355 { 356 #if !defined(USE_TCP_LOOPBACK) 357 boundPath = MDNS_UDS_SERVERPATH_DEBUG; 358 #endif 359 } 360 361 mDNSexport int IsDebugSocketInUse(void) 362 { 363 #if !defined(USE_TCP_LOOPBACK) 364 return !strcmp(boundPath, MDNS_UDS_SERVERPATH_DEBUG); 365 #else 366 return mDNSfalse; 367 #endif 368 } 369 #endif 370 371 mDNSlocal void AbortUnlinkAndFree(request_state *req) 372 { 373 request_state **p = &all_requests; 374 abort_request(req); 375 while (*p && *p != req) p=&(*p)->next; 376 if (*p) { *p = req->next; freeL("request_state/AbortUnlinkAndFree", req); } 377 else LogMsg("AbortUnlinkAndFree: ERROR: Attempt to abort operation %p not in list", req); 378 } 379 380 mDNSlocal reply_state *create_reply(const reply_op_t op, const size_t datalen, request_state *const request) 381 { 382 reply_state *reply; 383 384 if ((unsigned)datalen < sizeof(reply_hdr)) 385 { 386 LogMsg("ERROR: create_reply - data length less than length of required fields"); 387 return NULL; 388 } 389 390 reply = mallocL("reply_state", sizeof(reply_state) + datalen - sizeof(reply_hdr)); 391 if (!reply) FatalError("ERROR: malloc"); 392 393 reply->next = mDNSNULL; 394 reply->totallen = (mDNSu32)datalen + sizeof(ipc_msg_hdr); 395 reply->nwriten = 0; 396 397 reply->mhdr->version = VERSION; 398 reply->mhdr->datalen = (mDNSu32)datalen; 399 reply->mhdr->ipc_flags = 0; 400 reply->mhdr->op = op; 401 reply->mhdr->client_context = request->hdr.client_context; 402 reply->mhdr->reg_index = 0; 403 404 return reply; 405 } 406 407 // Append a reply to the list in a request object 408 // If our request is sharing a connection, then we append our reply_state onto the primary's list 409 // If the request does not want asynchronous replies, then the reply is freed instead of being appended to any list. 410 mDNSlocal void append_reply(request_state *req, reply_state *rep) 411 { 412 request_state *r; 413 reply_state **ptr; 414 415 if (req->no_reply) 416 { 417 freeL("reply_state/append_reply", rep); 418 return; 419 } 420 421 r = req->primary ? req->primary : req; 422 ptr = &r->replies; 423 while (*ptr) ptr = &(*ptr)->next; 424 *ptr = rep; 425 rep->next = NULL; 426 } 427 428 // Generates a response message giving name, type, domain, plus interface index, 429 // suitable for a browse result or service registration result. 430 // On successful completion rep is set to point to a malloc'd reply_state struct 431 mDNSlocal mStatus GenerateNTDResponse(const domainname *const servicename, const mDNSInterfaceID id, 432 request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err) 433 { 434 domainlabel name; 435 domainname type, dom; 436 *rep = NULL; 437 if (!DeconstructServiceName(servicename, &name, &type, &dom)) 438 return kDNSServiceErr_Invalid; 439 else 440 { 441 char namestr[MAX_DOMAIN_LABEL+1]; 442 char typestr[MAX_ESCAPED_DOMAIN_NAME]; 443 char domstr [MAX_ESCAPED_DOMAIN_NAME]; 444 int len; 445 char *data; 446 447 ConvertDomainLabelToCString_unescaped(&name, namestr); 448 ConvertDomainNameToCString(&type, typestr); 449 ConvertDomainNameToCString(&dom, domstr); 450 451 // Calculate reply data length 452 len = sizeof(DNSServiceFlags); 453 len += sizeof(mDNSu32); // if index 454 len += sizeof(DNSServiceErrorType); 455 len += (int) (strlen(namestr) + 1); 456 len += (int) (strlen(typestr) + 1); 457 len += (int) (strlen(domstr) + 1); 458 459 // Build reply header 460 *rep = create_reply(op, len, request); 461 (*rep)->rhdr->flags = dnssd_htonl(flags); 462 (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse)); 463 (*rep)->rhdr->error = dnssd_htonl(err); 464 465 // Build reply body 466 data = (char *)&(*rep)->rhdr[1]; 467 put_string(namestr, &data); 468 put_string(typestr, &data); 469 put_string(domstr, &data); 470 471 return mStatus_NoError; 472 } 473 } 474 475 mDNSlocal void GenerateBrowseReply(const domainname *const servicename, const mDNSInterfaceID id, 476 request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err) 477 { 478 char namestr[MAX_DOMAIN_LABEL+1]; 479 char typestr[MAX_ESCAPED_DOMAIN_NAME]; 480 static const char domstr[] = "."; 481 int len; 482 char *data; 483 484 *rep = NULL; 485 486 // 1. Put first label in namestr 487 ConvertDomainLabelToCString_unescaped((const domainlabel *)servicename, namestr); 488 489 // 2. Put second label and "local" into typestr 490 mDNS_snprintf(typestr, sizeof(typestr), "%#s.local.", SecondLabel(servicename)); 491 492 // Calculate reply data length 493 len = sizeof(DNSServiceFlags); 494 len += sizeof(mDNSu32); // if index 495 len += sizeof(DNSServiceErrorType); 496 len += (int) (strlen(namestr) + 1); 497 len += (int) (strlen(typestr) + 1); 498 len += (int) (strlen(domstr) + 1); 499 500 // Build reply header 501 *rep = create_reply(op, len, request); 502 (*rep)->rhdr->flags = dnssd_htonl(flags); 503 (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse)); 504 (*rep)->rhdr->error = dnssd_htonl(err); 505 506 // Build reply body 507 data = (char *)&(*rep)->rhdr[1]; 508 put_string(namestr, &data); 509 put_string(typestr, &data); 510 put_string(domstr, &data); 511 } 512 513 // Returns a resource record (allocated w/ malloc) containing the data found in an IPC message 514 // Data must be in the following format: flags, interfaceIndex, name, rrtype, rrclass, rdlen, rdata, (optional) ttl 515 // (ttl only extracted/set if ttl argument is non-zero). Returns NULL for a bad-parameter error 516 mDNSlocal AuthRecord *read_rr_from_ipc_msg(request_state *request, int GetTTL, int validate_flags) 517 { 518 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 519 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 520 char name[256]; 521 int str_err = get_string(&request->msgptr, request->msgend, name, sizeof(name)); 522 mDNSu16 type = get_uint16(&request->msgptr, request->msgend); 523 mDNSu16 class = get_uint16(&request->msgptr, request->msgend); 524 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 525 const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 526 mDNSu32 ttl = GetTTL ? get_uint32(&request->msgptr, request->msgend) : 0; 527 size_t storage_size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 528 AuthRecord *rr; 529 mDNSInterfaceID InterfaceID; 530 AuthRecType artype; 531 532 request->flags = flags; 533 request->interfaceIndex = interfaceIndex; 534 535 if (str_err) { LogMsg("ERROR: read_rr_from_ipc_msg - get_string"); return NULL; } 536 537 if (!request->msgptr) { LogMsg("Error reading Resource Record from client"); return NULL; } 538 539 if (validate_flags && 540 !((flags & kDNSServiceFlagsShared) == kDNSServiceFlagsShared) && 541 !((flags & kDNSServiceFlagsUnique) == kDNSServiceFlagsUnique)) 542 { 543 LogMsg("ERROR: Bad resource record flags (must be kDNSServiceFlagsShared or kDNSServiceFlagsUnique)"); 544 return NULL; 545 } 546 547 rr = mallocL("AuthRecord/read_rr_from_ipc_msg", sizeof(AuthRecord) - sizeof(RDataBody) + storage_size); 548 if (!rr) FatalError("ERROR: malloc"); 549 550 InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 551 if (InterfaceID == mDNSInterface_LocalOnly) 552 artype = AuthRecordLocalOnly; 553 else if (InterfaceID == mDNSInterface_P2P || InterfaceID == mDNSInterface_BLE) 554 artype = AuthRecordP2P; 555 else if ((InterfaceID == mDNSInterface_Any) && (flags & kDNSServiceFlagsIncludeP2P) 556 && (flags & kDNSServiceFlagsIncludeAWDL)) 557 artype = AuthRecordAnyIncludeAWDLandP2P; 558 else if ((InterfaceID == mDNSInterface_Any) && (flags & kDNSServiceFlagsIncludeP2P)) 559 artype = AuthRecordAnyIncludeP2P; 560 else if ((InterfaceID == mDNSInterface_Any) && (flags & kDNSServiceFlagsIncludeAWDL)) 561 artype = AuthRecordAnyIncludeAWDL; 562 else 563 artype = AuthRecordAny; 564 565 mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, type, 0, 566 (mDNSu8) ((flags & kDNSServiceFlagsShared) ? kDNSRecordTypeShared : kDNSRecordTypeUnique), artype, mDNSNULL, mDNSNULL); 567 568 if (!MakeDomainNameFromDNSNameString(&rr->namestorage, name)) 569 { 570 LogMsg("ERROR: bad name: %s", name); 571 freeL("AuthRecord/read_rr_from_ipc_msg", rr); 572 return NULL; 573 } 574 575 if (flags & kDNSServiceFlagsAllowRemoteQuery) rr->AllowRemoteQuery = mDNStrue; 576 rr->resrec.rrclass = class; 577 rr->resrec.rdlength = rdlen; 578 rr->resrec.rdata->MaxRDLength = rdlen; 579 mDNSPlatformMemCopy(rr->resrec.rdata->u.data, rdata, rdlen); 580 if (GetTTL) rr->resrec.rroriginalttl = ttl; 581 rr->resrec.namehash = DomainNameHashValue(rr->resrec.name); 582 SetNewRData(&rr->resrec, mDNSNULL, 0); // Sets rr->rdatahash for us 583 return rr; 584 } 585 586 mDNSlocal int build_domainname_from_strings(domainname *srv, char *name, char *regtype, char *domain) 587 { 588 domainlabel n; 589 domainname d, t; 590 591 if (!MakeDomainLabelFromLiteralString(&n, name)) return -1; 592 if (!MakeDomainNameFromDNSNameString(&t, regtype)) return -1; 593 if (!MakeDomainNameFromDNSNameString(&d, domain)) return -1; 594 if (!ConstructServiceName(srv, &n, &t, &d)) return -1; 595 return 0; 596 } 597 598 mDNSlocal void send_all(dnssd_sock_t s, const char *ptr, int len) 599 { 600 int n = send(s, ptr, len, 0); 601 // On a freshly-created Unix Domain Socket, the kernel should *never* fail to buffer a small write for us 602 // (four bytes for a typical error code return, 12 bytes for DNSServiceGetProperty(DaemonVersion)). 603 // If it does fail, we don't attempt to handle this failure, but we do log it so we know something is wrong. 604 if (n < len) 605 LogMsg("ERROR: send_all(%d) wrote %d of %d errno %d (%s)", 606 s, n, len, dnssd_errno, dnssd_strerror(dnssd_errno)); 607 } 608 609 #if 0 610 mDNSlocal mDNSBool AuthorizedDomain(const request_state * const request, const domainname * const d, const DNameListElem * const doms) 611 { 612 const DNameListElem *delem = mDNSNULL; 613 int bestDelta = -1; // the delta of the best match, lower is better 614 int dLabels = 0; 615 mDNSBool allow = mDNSfalse; 616 617 if (SystemUID(request->uid)) return mDNStrue; 618 619 dLabels = CountLabels(d); 620 for (delem = doms; delem; delem = delem->next) 621 { 622 if (delem->uid) 623 { 624 int delemLabels = CountLabels(&delem->name); 625 int delta = dLabels - delemLabels; 626 if ((bestDelta == -1 || delta <= bestDelta) && SameDomainName(&delem->name, SkipLeadingLabels(d, delta))) 627 { 628 bestDelta = delta; 629 allow = (allow || (delem->uid == request->uid)); 630 } 631 } 632 } 633 634 return bestDelta == -1 ? mDNStrue : allow; 635 } 636 #endif 637 638 // *************************************************************************** 639 #if COMPILER_LIKES_PRAGMA_MARK 640 #pragma mark - 641 #pragma mark - external helpers 642 #endif 643 644 mDNSexport mDNSBool callExternalHelpers(mDNSInterfaceID InterfaceID, const domainname *const domain, DNSServiceFlags flags) 645 { 646 #if APPLE_OSX_mDNSResponder 647 648 // Only call D2D layer routines if request applies to a D2D interface and the domain is "local". 649 if ( (((InterfaceID == mDNSInterface_Any) && (flags & (kDNSServiceFlagsIncludeP2P | kDNSServiceFlagsIncludeAWDL | kDNSServiceFlagsAutoTrigger))) 650 || mDNSPlatformInterfaceIsD2D(InterfaceID) || (InterfaceID == mDNSInterface_BLE)) 651 && IsLocalDomain(domain)) 652 { 653 return mDNStrue; 654 } 655 else 656 return mDNSfalse; 657 658 #else 659 (void) InterfaceID; 660 (void) domain; 661 (void) flags; 662 663 return mDNSfalse; 664 #endif // APPLE_OSX_mDNSResponder 665 } 666 667 mDNSlocal void external_start_advertising_helper(service_instance *const instance) 668 { 669 AuthRecord *st = instance->subtypes; 670 ExtraResourceRecord *e; 671 int i; 672 673 if (mDNSIPPortIsZero(instance->request->u.servicereg.port)) 674 { 675 LogInfo("external_start_advertising_helper: Not registering service with port number zero"); 676 return; 677 } 678 679 if (instance->external_advertise) LogMsg("external_start_advertising_helper: external_advertise already set!"); 680 681 for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++) 682 external_start_advertising_service(&st[i].resrec, instance->request->flags); 683 684 external_start_advertising_service(&instance->srs.RR_PTR.resrec, instance->request->flags); 685 external_start_advertising_service(&instance->srs.RR_SRV.resrec, instance->request->flags); 686 687 external_start_advertising_service(&instance->srs.RR_TXT.resrec, instance->request->flags); 688 689 for (e = instance->srs.Extras; e; e = e->next) 690 external_start_advertising_service(&e->r.resrec, instance->request->flags); 691 692 instance->external_advertise = mDNStrue; 693 } 694 695 mDNSlocal void external_stop_advertising_helper(service_instance *const instance) 696 { 697 AuthRecord *st = instance->subtypes; 698 ExtraResourceRecord *e; 699 int i; 700 701 if (!instance->external_advertise) return; 702 703 LogInfo("external_stop_advertising_helper: calling external_stop_advertising_service"); 704 705 for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++) 706 external_stop_advertising_service(&st[i].resrec, instance->request->flags); 707 708 external_stop_advertising_service(&instance->srs.RR_PTR.resrec, instance->request->flags); 709 external_stop_advertising_service(&instance->srs.RR_SRV.resrec, instance->request->flags); 710 external_stop_advertising_service(&instance->srs.RR_TXT.resrec, instance->request->flags); 711 712 for (e = instance->srs.Extras; e; e = e->next) 713 external_stop_advertising_service(&e->r.resrec, instance->request->flags); 714 715 instance->external_advertise = mDNSfalse; 716 } 717 718 // *************************************************************************** 719 #if COMPILER_LIKES_PRAGMA_MARK 720 #pragma mark - 721 #pragma mark - DNSServiceRegister 722 #endif 723 724 mDNSexport void FreeExtraRR(mDNS *const m, AuthRecord *const rr, mStatus result) 725 { 726 ExtraResourceRecord *extra = (ExtraResourceRecord *)rr->RecordContext; 727 (void)m; // Unused 728 729 if (result != mStatus_MemFree) { LogMsg("Error: FreeExtraRR invoked with unexpected error %d", result); return; } 730 731 LogInfo(" FreeExtraRR %s", RRDisplayString(m, &rr->resrec)); 732 733 if (rr->resrec.rdata != &rr->rdatastorage) 734 freeL("Extra RData", rr->resrec.rdata); 735 freeL("ExtraResourceRecord/FreeExtraRR", extra); 736 } 737 738 mDNSlocal void unlink_and_free_service_instance(service_instance *srv) 739 { 740 ExtraResourceRecord *e = srv->srs.Extras, *tmp; 741 742 external_stop_advertising_helper(srv); 743 744 // clear pointers from parent struct 745 if (srv->request) 746 { 747 service_instance **p = &srv->request->u.servicereg.instances; 748 while (*p) 749 { 750 if (*p == srv) { *p = (*p)->next; break; } 751 p = &(*p)->next; 752 } 753 } 754 755 while (e) 756 { 757 e->r.RecordContext = e; 758 tmp = e; 759 e = e->next; 760 FreeExtraRR(&mDNSStorage, &tmp->r, mStatus_MemFree); 761 } 762 763 if (srv->srs.RR_TXT.resrec.rdata != &srv->srs.RR_TXT.rdatastorage) 764 freeL("TXT RData", srv->srs.RR_TXT.resrec.rdata); 765 766 if (srv->subtypes) 767 { 768 freeL("ServiceSubTypes", srv->subtypes); 769 srv->subtypes = NULL; 770 } 771 if (srv->srs.AnonData) 772 { 773 freeL("Anonymous", (void *)srv->srs.AnonData); 774 srv->srs.AnonData = NULL; 775 } 776 freeL("service_instance", srv); 777 } 778 779 // Count how many other service records we have locally with the same name, but different rdata. 780 // For auto-named services, we can have at most one per machine -- if we allowed two auto-named services of 781 // the same type on the same machine, we'd get into an infinite autoimmune-response loop of continuous renaming. 782 mDNSexport int CountPeerRegistrations(ServiceRecordSet *const srs) 783 { 784 int count = 0; 785 ResourceRecord *r = &srs->RR_SRV.resrec; 786 AuthRecord *rr; 787 788 for (rr = mDNSStorage.ResourceRecords; rr; rr=rr->next) 789 if (rr->resrec.rrtype == kDNSType_SRV && SameDomainName(rr->resrec.name, r->name) && !IdenticalSameNameRecord(&rr->resrec, r)) 790 count++; 791 792 verbosedebugf("%d peer registrations for %##s", count, r->name->c); 793 return(count); 794 } 795 796 mDNSexport int CountExistingRegistrations(domainname *srv, mDNSIPPort port) 797 { 798 int count = 0; 799 AuthRecord *rr; 800 for (rr = mDNSStorage.ResourceRecords; rr; rr=rr->next) 801 if (rr->resrec.rrtype == kDNSType_SRV && 802 mDNSSameIPPort(rr->resrec.rdata->u.srv.port, port) && 803 SameDomainName(rr->resrec.name, srv)) 804 count++; 805 return(count); 806 } 807 808 mDNSlocal void SendServiceRemovalNotification(ServiceRecordSet *const srs) 809 { 810 reply_state *rep; 811 service_instance *instance = srs->ServiceContext; 812 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, 0, mStatus_NoError) != mStatus_NoError) 813 LogMsg("%3d: SendServiceRemovalNotification: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 814 else { append_reply(instance->request, rep); instance->clientnotified = mDNSfalse; } 815 } 816 817 // service registration callback performs three duties - frees memory for deregistered services, 818 // handles name conflicts, and delivers completed registration information to the client 819 mDNSlocal void regservice_callback(mDNS *const m, ServiceRecordSet *const srs, mStatus result) 820 { 821 mStatus err; 822 mDNSBool SuppressError = mDNSfalse; 823 service_instance *instance; 824 reply_state *rep; 825 (void)m; // Unused 826 827 if (!srs) { LogMsg("regservice_callback: srs is NULL %d", result); return; } 828 829 instance = srs->ServiceContext; 830 if (!instance) { LogMsg("regservice_callback: srs->ServiceContext is NULL %d", result); return; } 831 832 // don't send errors up to client for wide-area, empty-string registrations 833 if (instance->request && 834 instance->request->u.servicereg.default_domain && 835 !instance->default_local) 836 SuppressError = mDNStrue; 837 838 if (mDNS_LoggingEnabled) 839 { 840 const char *const fmt = 841 (result == mStatus_NoError) ? "%s DNSServiceRegister(%##s, %u) REGISTERED" : 842 (result == mStatus_MemFree) ? "%s DNSServiceRegister(%##s, %u) DEREGISTERED" : 843 (result == mStatus_NameConflict) ? "%s DNSServiceRegister(%##s, %u) NAME CONFLICT" : 844 "%s DNSServiceRegister(%##s, %u) %s %d"; 845 char prefix[16] = "---:"; 846 if (instance->request) mDNS_snprintf(prefix, sizeof(prefix), "%3d:", instance->request->sd); 847 LogOperation(fmt, prefix, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port), 848 SuppressError ? "suppressed error" : "CALLBACK", result); 849 } 850 851 if (!instance->request && result != mStatus_MemFree) { LogMsg("regservice_callback: instance->request is NULL %d", result); return; } 852 853 if (result == mStatus_NoError) 854 { 855 if (instance->request->u.servicereg.allowremotequery) 856 { 857 ExtraResourceRecord *e; 858 srs->RR_ADV.AllowRemoteQuery = mDNStrue; 859 srs->RR_PTR.AllowRemoteQuery = mDNStrue; 860 srs->RR_SRV.AllowRemoteQuery = mDNStrue; 861 srs->RR_TXT.AllowRemoteQuery = mDNStrue; 862 for (e = instance->srs.Extras; e; e = e->next) e->r.AllowRemoteQuery = mDNStrue; 863 } 864 865 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 866 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 867 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 868 869 if (callExternalHelpers(instance->request->u.servicereg.InterfaceID, &instance->domain, instance->request->flags)) 870 { 871 LogInfo("regservice_callback: calling external_start_advertising_helper()"); 872 external_start_advertising_helper(instance); 873 } 874 if (instance->request->u.servicereg.autoname && CountPeerRegistrations(srs) == 0) 875 RecordUpdatedNiceLabel(0); // Successfully got new name, tell user immediately 876 } 877 else if (result == mStatus_MemFree) 878 { 879 #if TARGET_OS_EMBEDDED 880 curr_num_regservices--; 881 #endif 882 if (instance->request && instance->renameonmemfree) 883 { 884 external_stop_advertising_helper(instance); 885 instance->renameonmemfree = 0; 886 err = mDNS_RenameAndReregisterService(m, srs, &instance->request->u.servicereg.name); 887 if (err) LogMsg("ERROR: regservice_callback - RenameAndReregisterService returned %d", err); 888 // error should never happen - safest to log and continue 889 } 890 else 891 unlink_and_free_service_instance(instance); 892 } 893 else if (result == mStatus_NameConflict) 894 { 895 if (instance->request->u.servicereg.autorename) 896 { 897 external_stop_advertising_helper(instance); 898 if (instance->request->u.servicereg.autoname && CountPeerRegistrations(srs) == 0) 899 { 900 // On conflict for an autoname service, rename and reregister *all* autoname services 901 IncrementLabelSuffix(&m->nicelabel, mDNStrue); 902 mDNS_ConfigChanged(m); // Will call back into udsserver_handle_configchange() 903 } 904 else // On conflict for a non-autoname service, rename and reregister just that one service 905 { 906 if (instance->clientnotified) SendServiceRemovalNotification(srs); 907 mDNS_RenameAndReregisterService(m, srs, mDNSNULL); 908 } 909 } 910 else 911 { 912 if (!SuppressError) 913 { 914 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 915 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 916 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 917 } 918 unlink_and_free_service_instance(instance); 919 } 920 } 921 else // Not mStatus_NoError, mStatus_MemFree, or mStatus_NameConflict 922 { 923 if (!SuppressError) 924 { 925 if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 926 LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 927 else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 928 } 929 } 930 } 931 932 mDNSlocal void regrecord_callback(mDNS *const m, AuthRecord *rr, mStatus result) 933 { 934 (void)m; // Unused 935 if (!rr->RecordContext) // parent struct already freed by termination callback 936 { 937 if (result == mStatus_NoError) 938 LogMsg("Error: regrecord_callback: successful registration of orphaned record %s", ARDisplayString(m, rr)); 939 else 940 { 941 if (result != mStatus_MemFree) LogMsg("regrecord_callback: error %d received after parent termination", result); 942 943 // We come here when the record is being deregistered either from DNSServiceRemoveRecord or connection_termination. 944 // If the record has been updated, we need to free the rdata. Every time we call mDNS_Update, it calls update_callback 945 // with the old rdata (so that we can free it) and stores the new rdata in "rr->resrec.rdata". This means, we need 946 // to free the latest rdata for which the update_callback was never called with. 947 if (rr->resrec.rdata != &rr->rdatastorage) freeL("RData/regrecord_callback", rr->resrec.rdata); 948 freeL("AuthRecord/regrecord_callback", rr); 949 } 950 } 951 else 952 { 953 registered_record_entry *re = rr->RecordContext; 954 request_state *request = re->request; 955 956 if (mDNS_LoggingEnabled) 957 { 958 char *fmt = (result == mStatus_NoError) ? "%3d: DNSServiceRegisterRecord(%u %s) REGISTERED" : 959 (result == mStatus_MemFree) ? "%3d: DNSServiceRegisterRecord(%u %s) DEREGISTERED" : 960 (result == mStatus_NameConflict) ? "%3d: DNSServiceRegisterRecord(%u %s) NAME CONFLICT" : 961 "%3d: DNSServiceRegisterRecord(%u %s) %d"; 962 LogOperation(fmt, request->sd, re->key, RRDisplayString(m, &rr->resrec), result); 963 } 964 965 if (result != mStatus_MemFree) 966 { 967 int len = sizeof(DNSServiceFlags) + sizeof(mDNSu32) + sizeof(DNSServiceErrorType); 968 reply_state *reply = create_reply(reg_record_reply_op, len, request); 969 reply->mhdr->client_context = re->regrec_client_context; 970 reply->rhdr->flags = dnssd_htonl(0); 971 reply->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, rr->resrec.InterfaceID, mDNSfalse)); 972 reply->rhdr->error = dnssd_htonl(result); 973 append_reply(request, reply); 974 } 975 976 if (result) 977 { 978 // If this is a callback to a keepalive record, do not free it. 979 if (result == mStatus_BadStateErr) 980 { 981 LogInfo("regrecord_callback: Callback with error code mStatus_BadStateErr - not freeing the record."); 982 } 983 else 984 { 985 // unlink from list, free memory 986 registered_record_entry **ptr = &request->u.reg_recs; 987 while (*ptr && (*ptr) != re) ptr = &(*ptr)->next; 988 if (!*ptr) { LogMsg("regrecord_callback - record not in list!"); return; } 989 *ptr = (*ptr)->next; 990 freeL("registered_record_entry AuthRecord regrecord_callback", re->rr); 991 freeL("registered_record_entry regrecord_callback", re); 992 } 993 } 994 else 995 { 996 if (re->external_advertise) LogMsg("regrecord_callback: external_advertise already set!"); 997 998 if (callExternalHelpers(re->origInterfaceID, &rr->namestorage, request->flags)) 999 { 1000 LogInfo("regrecord_callback: calling external_start_advertising_service"); 1001 external_start_advertising_service(&rr->resrec, request->flags); 1002 re->external_advertise = mDNStrue; 1003 } 1004 } 1005 } 1006 } 1007 1008 // set_peer_pid() is called after mem is allocated for each new request in NewRequest() 1009 // This accounts for 2 places (connect_callback, request_callback) 1010 mDNSlocal void set_peer_pid(request_state *request) 1011 { 1012 pid_t p = (pid_t) -1; 1013 socklen_t len = sizeof(p); 1014 request->pid_name[0] = '\0'; 1015 request->process_id = -1; 1016 #ifdef LOCAL_PEEREPID 1017 if (request->sd < 0) 1018 return; 1019 // to extract the effective pid value 1020 if (getsockopt(request->sd, SOL_LOCAL, LOCAL_PEEREPID, &p, &len) != 0) 1021 return; 1022 // to extract the process name from the pid value 1023 if (proc_pidinfo(p, PROC_PIDT_SHORTBSDINFO, 1, &proc, PROC_PIDT_SHORTBSDINFO_SIZE) == 0) 1024 return; 1025 mDNSPlatformStrLCopy(request->pid_name, proc.pbsi_comm, sizeof(request->pid_name)); 1026 request->process_id = p; 1027 debugf("set_peer_pid: Client PEEREPID is %d %s", p, request->pid_name); 1028 #else // !LOCAL_PEEREPID 1029 len = 0; 1030 LogInfo("set_peer_pid: Not Supported on this version of OS"); 1031 if (request->sd < 0) 1032 return; 1033 #endif // LOCAL_PEEREPID 1034 } 1035 1036 mDNSlocal void connection_termination(request_state *request) 1037 { 1038 // When terminating a shared connection, we need to scan the all_requests list 1039 // and terminate any subbordinate operations sharing this file descriptor 1040 request_state **req = &all_requests; 1041 1042 LogOperation("%3d: DNSServiceCreateConnection STOP PID[%d](%s)", request->sd, request->process_id, request->pid_name); 1043 1044 while (*req) 1045 { 1046 if ((*req)->primary == request) 1047 { 1048 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 1049 request_state *tmp = *req; 1050 if (tmp->primary == tmp) LogMsg("connection_termination ERROR (*req)->primary == *req for %p %d", tmp, tmp->sd); 1051 if (tmp->replies) LogMsg("connection_termination ERROR How can subordinate req %p %d have replies queued?", tmp, tmp->sd); 1052 abort_request(tmp); 1053 *req = tmp->next; 1054 freeL("request_state/connection_termination", tmp); 1055 } 1056 else 1057 req = &(*req)->next; 1058 } 1059 1060 while (request->u.reg_recs) 1061 { 1062 registered_record_entry *ptr = request->u.reg_recs; 1063 LogOperation("%3d: DNSServiceRegisterRecord(%u %s) STOP PID[%d](%s)", request->sd, ptr->key, RRDisplayString(&mDNSStorage, &ptr->rr->resrec), request->process_id, request->pid_name); 1064 request->u.reg_recs = request->u.reg_recs->next; 1065 ptr->rr->RecordContext = NULL; 1066 if (ptr->external_advertise) 1067 { 1068 ptr->external_advertise = mDNSfalse; 1069 external_stop_advertising_service(&ptr->rr->resrec, request->flags); 1070 } 1071 LogMcastS(ptr->rr, request, reg_stop); 1072 mDNS_Deregister(&mDNSStorage, ptr->rr); // Will free ptr->rr for us 1073 freeL("registered_record_entry/connection_termination", ptr); 1074 } 1075 } 1076 1077 mDNSlocal void handle_cancel_request(request_state *request) 1078 { 1079 request_state **req = &all_requests; 1080 LogOperation("%3d: Cancel %08X %08X", request->sd, request->hdr.client_context.u32[1], request->hdr.client_context.u32[0]); 1081 while (*req) 1082 { 1083 if ((*req)->primary == request && 1084 (*req)->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] && 1085 (*req)->hdr.client_context.u32[1] == request->hdr.client_context.u32[1]) 1086 { 1087 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 1088 request_state *tmp = *req; 1089 abort_request(tmp); 1090 *req = tmp->next; 1091 freeL("request_state/handle_cancel_request", tmp); 1092 } 1093 else 1094 req = &(*req)->next; 1095 } 1096 } 1097 1098 mDNSlocal mStatus handle_regrecord_request(request_state *request) 1099 { 1100 mStatus err = mStatus_BadParamErr; 1101 AuthRecord *rr; 1102 1103 if (request->terminate != connection_termination) 1104 { LogMsg("%3d: DNSServiceRegisterRecord(not a shared connection ref)", request->sd); return(err); } 1105 1106 rr = read_rr_from_ipc_msg(request, 1, 1); 1107 if (rr) 1108 { 1109 registered_record_entry *re; 1110 // Don't allow non-local domains to be regsitered as LocalOnly. Allowing this would permit 1111 // clients to register records such as www.bigbank.com A w.x.y.z to redirect Safari. 1112 if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly && !IsLocalDomain(rr->resrec.name) && 1113 rr->resrec.rrclass == kDNSClass_IN && (rr->resrec.rrtype == kDNSType_A || rr->resrec.rrtype == kDNSType_AAAA || 1114 rr->resrec.rrtype == kDNSType_CNAME)) 1115 { 1116 freeL("AuthRecord/handle_regrecord_request", rr); 1117 return (mStatus_BadParamErr); 1118 } 1119 // allocate registration entry, link into list 1120 re = mallocL("registered_record_entry", sizeof(registered_record_entry)); 1121 if (!re) 1122 FatalError("ERROR: malloc"); 1123 re->key = request->hdr.reg_index; 1124 re->rr = rr; 1125 re->regrec_client_context = request->hdr.client_context; 1126 re->request = request; 1127 re->external_advertise = mDNSfalse; 1128 rr->RecordContext = re; 1129 rr->RecordCallback = regrecord_callback; 1130 1131 re->origInterfaceID = rr->resrec.InterfaceID; 1132 if (rr->resrec.InterfaceID == mDNSInterface_P2P) 1133 rr->resrec.InterfaceID = mDNSInterface_Any; 1134 #if 0 1135 if (!AuthorizedDomain(request, rr->resrec.name, AutoRegistrationDomains)) return (mStatus_NoError); 1136 #endif 1137 if (rr->resrec.rroriginalttl == 0) 1138 rr->resrec.rroriginalttl = DefaultTTLforRRType(rr->resrec.rrtype); 1139 1140 LogOperation("%3d: DNSServiceRegisterRecord(%u %s) START PID[%d](%s)", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec), 1141 request->process_id, request->pid_name); 1142 1143 err = mDNS_Register(&mDNSStorage, rr); 1144 if (err) 1145 { 1146 LogOperation("%3d: DNSServiceRegisterRecord(%u %s) ERROR (%d)", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec), err); 1147 freeL("registered_record_entry", re); 1148 freeL("registered_record_entry/AuthRecord", rr); 1149 } 1150 else 1151 { 1152 LogMcastS(rr, request, reg_start); 1153 re->next = request->u.reg_recs; 1154 request->u.reg_recs = re; 1155 } 1156 } 1157 return(err); 1158 } 1159 1160 mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m); 1161 1162 mDNSlocal void regservice_termination_callback(request_state *request) 1163 { 1164 if (!request) 1165 { 1166 LogMsg("regservice_termination_callback context is NULL"); 1167 return; 1168 } 1169 while (request->u.servicereg.instances) 1170 { 1171 service_instance *p = request->u.servicereg.instances; 1172 request->u.servicereg.instances = request->u.servicereg.instances->next; 1173 // only safe to free memory if registration is not valid, i.e. deregister fails (which invalidates p) 1174 LogOperation("%3d: DNSServiceRegister(%##s, %u) STOP PID[%d](%s)", request->sd, p->srs.RR_SRV.resrec.name->c, 1175 mDNSVal16(p->srs.RR_SRV.resrec.rdata->u.srv.port), request->process_id, request->pid_name); 1176 1177 external_stop_advertising_helper(p); 1178 1179 // Clear backpointer *before* calling mDNS_DeregisterService/unlink_and_free_service_instance 1180 // We don't need unlink_and_free_service_instance to cut its element from the list, because we're already advancing 1181 // request->u.servicereg.instances as we work our way through the list, implicitly cutting one element at a time 1182 // We can't clear p->request *after* the calling mDNS_DeregisterService/unlink_and_free_service_instance 1183 // because by then we might have already freed p 1184 p->request = NULL; 1185 LogMcastS(&p->srs.RR_SRV, request, reg_stop); 1186 if (mDNS_DeregisterService(&mDNSStorage, &p->srs)) 1187 { 1188 unlink_and_free_service_instance(p); 1189 // Don't touch service_instance *p after this -- it's likely to have been freed already 1190 } 1191 } 1192 if (request->u.servicereg.txtdata) 1193 { 1194 freeL("service_info txtdata", request->u.servicereg.txtdata); 1195 request->u.servicereg.txtdata = NULL; 1196 } 1197 if (request->u.servicereg.autoname) 1198 { 1199 // Clear autoname before calling UpdateDeviceInfoRecord() so it doesn't mistakenly include this in its count of active autoname registrations 1200 request->u.servicereg.autoname = mDNSfalse; 1201 UpdateDeviceInfoRecord(&mDNSStorage); 1202 } 1203 } 1204 1205 mDNSlocal request_state *LocateSubordinateRequest(request_state *request) 1206 { 1207 request_state *req; 1208 for (req = all_requests; req; req = req->next) 1209 if (req->primary == request && 1210 req->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] && 1211 req->hdr.client_context.u32[1] == request->hdr.client_context.u32[1]) return(req); 1212 return(request); 1213 } 1214 1215 mDNSlocal mStatus add_record_to_service(request_state *request, service_instance *instance, mDNSu16 rrtype, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl) 1216 { 1217 ServiceRecordSet *srs = &instance->srs; 1218 mStatus result; 1219 size_t size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 1220 ExtraResourceRecord *extra = mallocL("ExtraResourceRecord", sizeof(*extra) - sizeof(RDataBody) + size); 1221 if (!extra) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; } 1222 1223 mDNSPlatformMemZero(extra, sizeof(ExtraResourceRecord)); // OK if oversized rdata not zero'd 1224 extra->r.resrec.rrtype = rrtype; 1225 extra->r.rdatastorage.MaxRDLength = (mDNSu16) size; 1226 extra->r.resrec.rdlength = rdlen; 1227 mDNSPlatformMemCopy(&extra->r.rdatastorage.u.data, rdata, rdlen); 1228 // use InterfaceID value from DNSServiceRegister() call that created the original service 1229 extra->r.resrec.InterfaceID = request->u.servicereg.InterfaceID; 1230 1231 result = mDNS_AddRecordToService(&mDNSStorage, srs, extra, &extra->r.rdatastorage, ttl, request->flags); 1232 if (result) 1233 { 1234 freeL("ExtraResourceRecord/add_record_to_service", extra); 1235 return result; 1236 } 1237 LogMcastS(&srs->RR_PTR, request, reg_start); 1238 1239 extra->ClientID = request->hdr.reg_index; 1240 if ( instance->external_advertise 1241 && callExternalHelpers(request->u.servicereg.InterfaceID, &instance->domain, request->flags)) 1242 { 1243 LogInfo("add_record_to_service: calling external_start_advertising_service"); 1244 external_start_advertising_service(&extra->r.resrec, request->flags); 1245 } 1246 return result; 1247 } 1248 1249 mDNSlocal mStatus handle_add_request(request_state *request) 1250 { 1251 service_instance *i; 1252 mStatus result = mStatus_UnknownErr; 1253 DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend); 1254 mDNSu16 rrtype = get_uint16(&request->msgptr, request->msgend); 1255 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 1256 const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 1257 mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend); 1258 if (!ttl) ttl = DefaultTTLforRRType(rrtype); 1259 (void)flags; // Unused 1260 1261 if (!request->msgptr) { LogMsg("%3d: DNSServiceAddRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 1262 1263 // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 1264 if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 1265 1266 if (request->terminate != regservice_termination_callback) 1267 { LogMsg("%3d: DNSServiceAddRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 1268 1269 // For a service registered with zero port, don't allow adding records. This mostly happens due to a bug 1270 // in the application. See radar://9165807. 1271 if (mDNSIPPortIsZero(request->u.servicereg.port)) 1272 { LogMsg("%3d: DNSServiceAddRecord: adding record to a service registered with zero port", request->sd); return(mStatus_BadParamErr); } 1273 1274 LogOperation("%3d: DNSServiceAddRecord(%X, %##s, %s, %d) PID[%d](%s)", request->sd, flags, 1275 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, DNSTypeName(rrtype), rdlen, 1276 request->process_id, request->pid_name); 1277 1278 for (i = request->u.servicereg.instances; i; i = i->next) 1279 { 1280 result = add_record_to_service(request, i, rrtype, rdlen, rdata, ttl); 1281 if (result && i->default_local) break; 1282 else result = mStatus_NoError; // suppress non-local default errors 1283 } 1284 1285 return(result); 1286 } 1287 1288 mDNSlocal void update_callback(mDNS *const m, AuthRecord *const rr, RData *oldrd, mDNSu16 oldrdlen) 1289 { 1290 mDNSBool external_advertise = (rr->UpdateContext) ? *((mDNSBool *)rr->UpdateContext) : mDNSfalse; 1291 (void)m; // Unused 1292 1293 // There are three cases. 1294 // 1295 // 1. We have updated the primary TXT record of the service 1296 // 2. We have updated the TXT record that was added to the service using DNSServiceAddRecord 1297 // 3. We have updated the TXT record that was registered using DNSServiceRegisterRecord 1298 // 1299 // external_advertise is set if we have advertised at least once during the initial addition 1300 // of the record in all of the three cases above. We should have checked for InterfaceID/LocalDomain 1301 // checks during the first time and hence we don't do any checks here 1302 if (external_advertise) 1303 { 1304 ResourceRecord ext = rr->resrec; 1305 DNSServiceFlags flags = deriveD2DFlagsFromAuthRecType(rr->ARType); 1306 1307 if (ext.rdlength == oldrdlen && mDNSPlatformMemSame(&ext.rdata->u, &oldrd->u, oldrdlen)) goto exit; 1308 SetNewRData(&ext, oldrd, oldrdlen); 1309 external_stop_advertising_service(&ext, flags); 1310 LogInfo("update_callback: calling external_start_advertising_service"); 1311 external_start_advertising_service(&rr->resrec, flags); 1312 } 1313 exit: 1314 if (oldrd != &rr->rdatastorage) freeL("RData/update_callback", oldrd); 1315 } 1316 1317 mDNSlocal mStatus update_record(AuthRecord *rr, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl, const mDNSBool *const external_advertise) 1318 { 1319 mStatus result; 1320 const size_t rdsize = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 1321 RData *newrd = mallocL("RData/update_record", sizeof(RData) - sizeof(RDataBody) + rdsize); 1322 if (!newrd) FatalError("ERROR: malloc"); 1323 newrd->MaxRDLength = (mDNSu16) rdsize; 1324 mDNSPlatformMemCopy(&newrd->u, rdata, rdlen); 1325 1326 // BIND named (name daemon) doesn't allow TXT records with zero-length rdata. This is strictly speaking correct, 1327 // since RFC 1035 specifies a TXT record as "One or more <character-string>s", not "Zero or more <character-string>s". 1328 // Since some legacy apps try to create zero-length TXT records, we'll silently correct it here. 1329 if (rr->resrec.rrtype == kDNSType_TXT && rdlen == 0) { rdlen = 1; newrd->u.txt.c[0] = 0; } 1330 1331 if (external_advertise) rr->UpdateContext = (void *)external_advertise; 1332 1333 result = mDNS_Update(&mDNSStorage, rr, ttl, rdlen, newrd, update_callback); 1334 if (result) { LogMsg("update_record: Error %d for %s", (int)result, ARDisplayString(&mDNSStorage, rr)); freeL("RData/update_record", newrd); } 1335 return result; 1336 } 1337 1338 mDNSlocal mStatus handle_update_request(request_state *request) 1339 { 1340 const ipc_msg_hdr *const hdr = &request->hdr; 1341 mStatus result = mStatus_BadReferenceErr; 1342 service_instance *i; 1343 AuthRecord *rr = NULL; 1344 1345 // get the message data 1346 DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend); // flags unused 1347 mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 1348 const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 1349 mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend); 1350 (void)flags; // Unused 1351 1352 if (!request->msgptr) { LogMsg("%3d: DNSServiceUpdateRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 1353 1354 // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 1355 if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 1356 1357 if (request->terminate == connection_termination) 1358 { 1359 // update an individually registered record 1360 registered_record_entry *reptr; 1361 for (reptr = request->u.reg_recs; reptr; reptr = reptr->next) 1362 { 1363 if (reptr->key == hdr->reg_index) 1364 { 1365 result = update_record(reptr->rr, rdlen, rdata, ttl, &reptr->external_advertise); 1366 LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s) PID[%d](%s)", 1367 request->sd, reptr->rr->resrec.name->c, reptr->rr ? DNSTypeName(reptr->rr->resrec.rrtype) : "<NONE>", 1368 request->process_id, request->pid_name); 1369 goto end; 1370 } 1371 } 1372 result = mStatus_BadReferenceErr; 1373 goto end; 1374 } 1375 1376 if (request->terminate != regservice_termination_callback) 1377 { LogMsg("%3d: DNSServiceUpdateRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 1378 1379 // For a service registered with zero port, only SRV record is initialized. Don't allow any updates. 1380 if (mDNSIPPortIsZero(request->u.servicereg.port)) 1381 { LogMsg("%3d: DNSServiceUpdateRecord: updating the record of a service registered with zero port", request->sd); return(mStatus_BadParamErr); } 1382 1383 // update the saved off TXT data for the service 1384 if (hdr->reg_index == TXT_RECORD_INDEX) 1385 { 1386 if (request->u.servicereg.txtdata) 1387 { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; } 1388 if (rdlen > 0) 1389 { 1390 request->u.servicereg.txtdata = mallocL("service_info txtdata", rdlen); 1391 if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_update_request - malloc"); 1392 mDNSPlatformMemCopy(request->u.servicereg.txtdata, rdata, rdlen); 1393 } 1394 request->u.servicereg.txtlen = rdlen; 1395 } 1396 1397 // update a record from a service record set 1398 for (i = request->u.servicereg.instances; i; i = i->next) 1399 { 1400 if (hdr->reg_index == TXT_RECORD_INDEX) rr = &i->srs.RR_TXT; 1401 else 1402 { 1403 ExtraResourceRecord *e; 1404 for (e = i->srs.Extras; e; e = e->next) 1405 if (e->ClientID == hdr->reg_index) { rr = &e->r; break; } 1406 } 1407 1408 if (!rr) { result = mStatus_BadReferenceErr; goto end; } 1409 result = update_record(rr, rdlen, rdata, ttl, &i->external_advertise); 1410 if (result && i->default_local) goto end; 1411 else result = mStatus_NoError; // suppress non-local default errors 1412 } 1413 1414 end: 1415 if (request->terminate == regservice_termination_callback) 1416 LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s) PID[%d](%s)", request->sd, 1417 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, 1418 rr ? DNSTypeName(rr->resrec.rrtype) : "<NONE>", 1419 request->process_id, request->pid_name); 1420 1421 return(result); 1422 } 1423 1424 // remove a resource record registered via DNSServiceRegisterRecord() 1425 mDNSlocal mStatus remove_record(request_state *request) 1426 { 1427 mStatus err = mStatus_UnknownErr; 1428 registered_record_entry *e, **ptr = &request->u.reg_recs; 1429 1430 while (*ptr && (*ptr)->key != request->hdr.reg_index) ptr = &(*ptr)->next; 1431 if (!*ptr) { LogMsg("%3d: DNSServiceRemoveRecord(%u) not found", request->sd, request->hdr.reg_index); return mStatus_BadReferenceErr; } 1432 e = *ptr; 1433 *ptr = e->next; // unlink 1434 1435 LogOperation("%3d: DNSServiceRemoveRecord(%u %s) PID[%d](%s)", 1436 request->sd, e->key, RRDisplayString(&mDNSStorage, &e->rr->resrec), request->process_id, request->pid_name); 1437 e->rr->RecordContext = NULL; 1438 if (e->external_advertise) 1439 { 1440 external_stop_advertising_service(&e->rr->resrec, request->flags); 1441 e->external_advertise = mDNSfalse; 1442 } 1443 LogMcastS(e->rr, request, reg_stop); 1444 err = mDNS_Deregister(&mDNSStorage, e->rr); // Will free e->rr for us; we're responsible for freeing e 1445 if (err) 1446 { 1447 LogMsg("ERROR: remove_record, mDNS_Deregister: %d", err); 1448 freeL("registered_record_entry AuthRecord remove_record", e->rr); 1449 } 1450 freeL("registered_record_entry remove_record", e); 1451 return err; 1452 } 1453 1454 mDNSlocal mStatus remove_extra(const request_state *const request, service_instance *const serv, mDNSu16 *const rrtype) 1455 { 1456 mStatus err = mStatus_BadReferenceErr; 1457 ExtraResourceRecord *ptr; 1458 1459 for (ptr = serv->srs.Extras; ptr; ptr = ptr->next) 1460 { 1461 if (ptr->ClientID == request->hdr.reg_index) // found match 1462 { 1463 *rrtype = ptr->r.resrec.rrtype; 1464 if (serv->external_advertise) external_stop_advertising_service(&ptr->r.resrec, request->flags); 1465 err = mDNS_RemoveRecordFromService(&mDNSStorage, &serv->srs, ptr, FreeExtraRR, ptr); 1466 break; 1467 } 1468 } 1469 return err; 1470 } 1471 1472 mDNSlocal mStatus handle_removerecord_request(request_state *request) 1473 { 1474 mStatus err = mStatus_BadReferenceErr; 1475 get_flags(&request->msgptr, request->msgend); // flags unused 1476 1477 if (!request->msgptr) { LogMsg("%3d: DNSServiceRemoveRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 1478 1479 // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 1480 if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 1481 1482 if (request->terminate == connection_termination) 1483 err = remove_record(request); // remove individually registered record 1484 else if (request->terminate != regservice_termination_callback) 1485 { LogMsg("%3d: DNSServiceRemoveRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 1486 else 1487 { 1488 service_instance *i; 1489 mDNSu16 rrtype = 0; 1490 LogOperation("%3d: DNSServiceRemoveRecord(%##s, %s) PID[%d](%s)", request->sd, 1491 (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, 1492 rrtype ? DNSTypeName(rrtype) : "<NONE>", request->process_id, request->pid_name); 1493 for (i = request->u.servicereg.instances; i; i = i->next) 1494 { 1495 err = remove_extra(request, i, &rrtype); 1496 if (err && i->default_local) break; 1497 else err = mStatus_NoError; // suppress non-local default errors 1498 } 1499 } 1500 1501 return(err); 1502 } 1503 1504 // If there's a comma followed by another character, 1505 // FindFirstSubType overwrites the comma with a nul and returns the pointer to the next character. 1506 // Otherwise, it returns a pointer to the final nul at the end of the string 1507 mDNSlocal char *FindFirstSubType(char *p, char **AnonData) 1508 { 1509 while (*p) 1510 { 1511 if (p[0] == '\\' && p[1]) 1512 { 1513 p += 2; 1514 } 1515 else if (p[0] == ',' && p[1]) 1516 { 1517 *p++ = 0; 1518 return(p); 1519 } 1520 else if (p[0] == ':' && p[1]) 1521 { 1522 *p++ = 0; 1523 *AnonData = p; 1524 } 1525 else 1526 { 1527 p++; 1528 } 1529 } 1530 return(p); 1531 } 1532 1533 // If there's a comma followed by another character, 1534 // FindNextSubType overwrites the comma with a nul and returns the pointer to the next character. 1535 // If it finds an illegal unescaped dot in the subtype name, it returns mDNSNULL 1536 // Otherwise, it returns a pointer to the final nul at the end of the string 1537 mDNSlocal char *FindNextSubType(char *p) 1538 { 1539 while (*p) 1540 { 1541 if (p[0] == '\\' && p[1]) // If escape character 1542 p += 2; // ignore following character 1543 else if (p[0] == ',') // If we found a comma 1544 { 1545 if (p[1]) *p++ = 0; 1546 return(p); 1547 } 1548 else if (p[0] == '.') 1549 return(mDNSNULL); 1550 else p++; 1551 } 1552 return(p); 1553 } 1554 1555 // Returns -1 if illegal subtype found 1556 mDNSexport mDNSs32 ChopSubTypes(char *regtype, char **AnonData) 1557 { 1558 mDNSs32 NumSubTypes = 0; 1559 char *stp = FindFirstSubType(regtype, AnonData); 1560 while (stp && *stp) // If we found a comma... 1561 { 1562 if (*stp == ',') return(-1); 1563 NumSubTypes++; 1564 stp = FindNextSubType(stp); 1565 } 1566 if (!stp) return(-1); 1567 return(NumSubTypes); 1568 } 1569 1570 mDNSexport AuthRecord *AllocateSubTypes(mDNSs32 NumSubTypes, char *p, char **AnonData) 1571 { 1572 AuthRecord *st = mDNSNULL; 1573 // 1574 // "p" is pointing at the regtype e.g., _http._tcp followed by ":<AnonData>" indicated 1575 // by AnonData being non-NULL which is in turn follwed by ",<SubTypes>" indicated by 1576 // NumSubTypes being non-zero. We need to skip the initial regtype to get to the actual 1577 // data that we want. When we come here, ChopSubTypes has null terminated like this e.g., 1578 // 1579 // _http._tcp<NULL><AnonData><NULL><SubType1><NULL><SubType2><NULL> etc. 1580 // 1581 // 1. If we have Anonymous data and subtypes, skip the regtype (e.g., "_http._tcp") 1582 // to get the AnonData and then skip the AnonData to get to the SubType. 1583 // 1584 // 2. If we have only SubTypes, skip the regtype to get to the SubType data. 1585 // 1586 // 3. If we have only AnonData, skip the regtype to get to the AnonData. 1587 // 1588 // 4. If we don't have AnonData or NumStypes, it is a noop. 1589 // 1590 if (AnonData) 1591 { 1592 int len; 1593 1594 // Skip the regtype 1595 while (*p) p++; 1596 p++; 1597 1598 len = strlen(p) + 1; 1599 *AnonData = mallocL("Anonymous", len); 1600 if (!(*AnonData)) 1601 { 1602 return (mDNSNULL); 1603 } 1604 mDNSPlatformMemCopy(*AnonData, p, len); 1605 } 1606 if (NumSubTypes) 1607 { 1608 mDNSs32 i; 1609 st = mallocL("ServiceSubTypes", NumSubTypes * sizeof(AuthRecord)); 1610 if (!st) return(mDNSNULL); 1611 for (i = 0; i < NumSubTypes; i++) 1612 { 1613 mDNS_SetupResourceRecord(&st[i], mDNSNULL, mDNSInterface_Any, kDNSQType_ANY, kStandardTTL, 0, AuthRecordAny, mDNSNULL, mDNSNULL); 1614 // First time through we skip the regtype or AnonData. Subsequently, the 1615 // previous subtype. 1616 while (*p) p++; 1617 p++; 1618 if (!MakeDomainNameFromDNSNameString(&st[i].namestorage, p)) 1619 { 1620 freeL("ServiceSubTypes", st); 1621 if (AnonData && *AnonData) 1622 freeL("AnonymousData", *AnonData); 1623 return(mDNSNULL); 1624 } 1625 } 1626 } 1627 // If NumSubTypes is zero and AnonData is non-NULL, we still return NULL but AnonData has been 1628 // initialized. The caller knows how to handle this. 1629 return(st); 1630 } 1631 1632 mDNSlocal mStatus register_service_instance(request_state *request, const domainname *domain) 1633 { 1634 service_instance **ptr, *instance; 1635 size_t extra_size = (request->u.servicereg.txtlen > sizeof(RDataBody)) ? (request->u.servicereg.txtlen - sizeof(RDataBody)) : 0; 1636 const mDNSBool DomainIsLocal = SameDomainName(domain, &localdomain); 1637 mStatus result; 1638 mDNSInterfaceID interfaceID = request->u.servicereg.InterfaceID; 1639 1640 // If the client specified an interface, but no domain, then we honor the specified interface for the "local" (mDNS) 1641 // registration but for the wide-area registrations we don't (currently) have any concept of a wide-area unicast 1642 // registrations scoped to a specific interface, so for the automatic domains we add we must *not* specify an interface. 1643 // (Specifying an interface with an apparently wide-area domain (i.e. something other than "local") 1644 // currently forces the registration to use mDNS multicast despite the apparently wide-area domain.) 1645 if (request->u.servicereg.default_domain && !DomainIsLocal) interfaceID = mDNSInterface_Any; 1646 1647 for (ptr = &request->u.servicereg.instances; *ptr; ptr = &(*ptr)->next) 1648 { 1649 if (SameDomainName(&(*ptr)->domain, domain)) 1650 { 1651 LogMsg("register_service_instance: domain %##s already registered for %#s.%##s", 1652 domain->c, &request->u.servicereg.name, &request->u.servicereg.type); 1653 return mStatus_AlreadyRegistered; 1654 } 1655 } 1656 1657 instance = mallocL("service_instance", sizeof(*instance) + extra_size); 1658 if (!instance) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; } 1659 1660 instance->next = mDNSNULL; 1661 instance->request = request; 1662 instance->renameonmemfree = 0; 1663 instance->clientnotified = mDNSfalse; 1664 instance->default_local = (request->u.servicereg.default_domain && DomainIsLocal); 1665 instance->external_advertise = mDNSfalse; 1666 AssignDomainName(&instance->domain, domain); 1667 1668 instance->srs.AnonData = mDNSNULL; 1669 if (!request->u.servicereg.AnonData) 1670 { 1671 instance->subtypes = AllocateSubTypes(request->u.servicereg.num_subtypes, request->u.servicereg.type_as_string, mDNSNULL); 1672 } 1673 else 1674 { 1675 char *AnonData = mDNSNULL; 1676 instance->subtypes = AllocateSubTypes(request->u.servicereg.num_subtypes, request->u.servicereg.type_as_string, &AnonData); 1677 if (AnonData) 1678 instance->srs.AnonData = (const mDNSu8 *)AnonData; 1679 } 1680 1681 if (request->u.servicereg.num_subtypes && !instance->subtypes) 1682 { 1683 unlink_and_free_service_instance(instance); 1684 instance = NULL; 1685 FatalError("ERROR: malloc"); 1686 } 1687 1688 result = mDNS_RegisterService(&mDNSStorage, &instance->srs, 1689 &request->u.servicereg.name, &request->u.servicereg.type, domain, 1690 request->u.servicereg.host.c[0] ? &request->u.servicereg.host : NULL, 1691 request->u.servicereg.port, 1692 request->u.servicereg.txtdata, request->u.servicereg.txtlen, 1693 instance->subtypes, request->u.servicereg.num_subtypes, 1694 interfaceID, regservice_callback, instance, request->flags); 1695 1696 if (!result) 1697 { 1698 *ptr = instance; // Append this to the end of our request->u.servicereg.instances list 1699 LogOperation("%3d: DNSServiceRegister(%##s, %u) ADDED", instance->request->sd, 1700 instance->srs.RR_SRV.resrec.name->c, mDNSVal16(request->u.servicereg.port)); 1701 LogMcastS(&instance->srs.RR_SRV, request, reg_start); 1702 } 1703 else 1704 { 1705 LogMsg("register_service_instance %#s.%##s%##s error %d", 1706 &request->u.servicereg.name, &request->u.servicereg.type, domain->c, result); 1707 unlink_and_free_service_instance(instance); 1708 } 1709 1710 return result; 1711 } 1712 1713 mDNSlocal void udsserver_default_reg_domain_changed(const DNameListElem *const d, const mDNSBool add) 1714 { 1715 request_state *request; 1716 1717 LogMsg("%s registration domain %##s", add ? "Adding" : "Removing", d->name.c); 1718 for (request = all_requests; request; request = request->next) 1719 { 1720 if (request->terminate != regservice_termination_callback) continue; 1721 if (!request->u.servicereg.default_domain) continue; 1722 if (!d->uid || SystemUID(request->uid) || request->uid == d->uid) 1723 { 1724 service_instance **ptr = &request->u.servicereg.instances; 1725 while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next; 1726 if (add) 1727 { 1728 // If we don't already have this domain in our list for this registration, add it now 1729 if (!*ptr) register_service_instance(request, &d->name); 1730 else debugf("udsserver_default_reg_domain_changed %##s already in list, not re-adding", &d->name); 1731 } 1732 else 1733 { 1734 // Normally we should not fail to find the specified instance 1735 // One case where this can happen is if a uDNS update fails for some reason, 1736 // and regservice_callback then calls unlink_and_free_service_instance and disposes of that instance. 1737 if (!*ptr) 1738 LogMsg("udsserver_default_reg_domain_changed domain %##s not found for service %#s type %s", 1739 &d->name, request->u.servicereg.name.c, request->u.servicereg.type_as_string); 1740 else 1741 { 1742 DNameListElem *p; 1743 for (p = AutoRegistrationDomains; p; p=p->next) 1744 if (!p->uid || SystemUID(request->uid) || request->uid == p->uid) 1745 if (SameDomainName(&d->name, &p->name)) break; 1746 if (p) debugf("udsserver_default_reg_domain_changed %##s still in list, not removing", &d->name); 1747 else 1748 { 1749 mStatus err; 1750 service_instance *si = *ptr; 1751 *ptr = si->next; 1752 if (si->clientnotified) SendServiceRemovalNotification(&si->srs); // Do this *before* clearing si->request backpointer 1753 // Now that we've cut this service_instance from the list, we MUST clear the si->request backpointer. 1754 // Otherwise what can happen is this: While our mDNS_DeregisterService is in the 1755 // process of completing asynchronously, the client cancels the entire operation, so 1756 // regservice_termination_callback then runs through the whole list deregistering each 1757 // instance, clearing the backpointers, and then disposing the parent request_state object. 1758 // However, because this service_instance isn't in the list any more, regservice_termination_callback 1759 // has no way to find it and clear its backpointer, and then when our mDNS_DeregisterService finally 1760 // completes later with a mStatus_MemFree message, it calls unlink_and_free_service_instance() with 1761 // a service_instance with a stale si->request backpointer pointing to memory that's already been freed. 1762 si->request = NULL; 1763 err = mDNS_DeregisterService(&mDNSStorage, &si->srs); 1764 if (err) { LogMsg("udsserver_default_reg_domain_changed err %d", err); unlink_and_free_service_instance(si); } 1765 } 1766 } 1767 } 1768 } 1769 } 1770 } 1771 1772 // Don't allow normal and anonymous registration to coexist. 1773 mDNSlocal mDNSBool CheckForMixedRegistrations(domainname *regtype, domainname *domain, mDNSBool AnonData) 1774 { 1775 request_state *request; 1776 1777 // We only care about local domains where the anonymous extension is 1778 // implemented. 1779 if (!SameDomainName(domain, (const domainname *) "\x5" "local")) 1780 { 1781 return mDNStrue; 1782 } 1783 1784 for (request = all_requests; request; request = request->next) 1785 { 1786 service_instance *ptr; 1787 1788 if (request->terminate != regservice_termination_callback) continue; 1789 for (ptr = request->u.servicereg.instances; ptr ; ptr = ptr->next) 1790 { 1791 if (!SameDomainName(&ptr->domain, (const domainname *)"\x5" "local") || 1792 !SameDomainName(&request->u.servicereg.type, regtype)) 1793 { 1794 continue; 1795 } 1796 1797 // If we are about to register a anonymous registraion, we dont't want to 1798 // allow the regular ones and vice versa. 1799 if (AnonData) 1800 { 1801 if (!ptr->srs.AnonData) 1802 { 1803 LogMsg("CheckForMixedRegistrations: Normal registration already exists for %##s", regtype->c); 1804 return mDNSfalse; 1805 } 1806 } 1807 else 1808 { 1809 // Allow multiple regular registrations 1810 if (ptr->srs.AnonData) 1811 { 1812 LogMsg("CheckForMixedRegistrations: Anonymous registration already exists for %##s", regtype->c); 1813 return mDNSfalse; 1814 } 1815 } 1816 } 1817 } 1818 return mDNStrue; 1819 } 1820 1821 // Returns true if the interfaceIndex value matches one of the pre-defined 1822 // special values listed in the switch statement below. 1823 mDNSlocal mDNSBool PreDefinedInterfaceIndex(mDNSu32 interfaceIndex) 1824 { 1825 switch(interfaceIndex) 1826 { 1827 case kDNSServiceInterfaceIndexAny: 1828 case kDNSServiceInterfaceIndexLocalOnly: 1829 case kDNSServiceInterfaceIndexUnicast: 1830 case kDNSServiceInterfaceIndexP2P: 1831 case kDNSServiceInterfaceIndexBLE: 1832 return mDNStrue; 1833 default: 1834 return mDNSfalse; 1835 } 1836 } 1837 1838 mDNSlocal mStatus handle_regservice_request(request_state *request) 1839 { 1840 char name[256]; // Lots of spare space for extra-long names that we'll auto-truncate down to 63 bytes 1841 char domain[MAX_ESCAPED_DOMAIN_NAME], host[MAX_ESCAPED_DOMAIN_NAME]; 1842 char type_as_string[MAX_ESCAPED_DOMAIN_NAME]; 1843 domainname d, srv; 1844 mStatus err; 1845 char *AnonData = mDNSNULL; 1846 const char *msgTXTData; 1847 1848 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 1849 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 1850 mDNSInterfaceID InterfaceID; 1851 1852 // Map kDNSServiceInterfaceIndexP2P to kDNSServiceInterfaceIndexAny with the 1853 // kDNSServiceFlagsIncludeP2P flag set. 1854 if (interfaceIndex == kDNSServiceInterfaceIndexP2P) 1855 { 1856 LogOperation("handle_regservice_request: mapping kDNSServiceInterfaceIndexP2P to kDNSServiceInterfaceIndexAny + kDNSServiceFlagsIncludeP2P"); 1857 flags |= kDNSServiceFlagsIncludeP2P; 1858 interfaceIndex = kDNSServiceInterfaceIndexAny; 1859 } 1860 1861 InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 1862 1863 // The registration is scoped to a specific interface index, but the 1864 // interface is not currently in our list. 1865 if (interfaceIndex && !InterfaceID) 1866 { 1867 // If it's one of the specially defined inteface index values, just return an error. 1868 if (PreDefinedInterfaceIndex(interfaceIndex)) 1869 { 1870 LogInfo("handle_regservice_request: bad interfaceIndex %d", interfaceIndex); 1871 return(mStatus_BadParamErr); 1872 } 1873 1874 // Otherwise, use the specified interface index value and the registration will 1875 // be applied to that interface when it comes up. 1876 InterfaceID = (mDNSInterfaceID)(uintptr_t)interfaceIndex; 1877 LogInfo("handle_regservice_request: registration pending for interface index %d", interfaceIndex); 1878 } 1879 1880 if (get_string(&request->msgptr, request->msgend, name, sizeof(name)) < 0 || 1881 get_string(&request->msgptr, request->msgend, type_as_string, MAX_ESCAPED_DOMAIN_NAME) < 0 || 1882 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0 || 1883 get_string(&request->msgptr, request->msgend, host, MAX_ESCAPED_DOMAIN_NAME) < 0) 1884 { LogMsg("ERROR: handle_regservice_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); } 1885 1886 request->flags = flags; 1887 request->interfaceIndex = interfaceIndex; 1888 request->u.servicereg.InterfaceID = InterfaceID; 1889 request->u.servicereg.instances = NULL; 1890 request->u.servicereg.txtlen = 0; 1891 request->u.servicereg.txtdata = NULL; 1892 mDNSPlatformStrLCopy(request->u.servicereg.type_as_string, type_as_string, sizeof(request->u.servicereg.type_as_string)); 1893 1894 if (request->msgptr + 2 > request->msgend) request->msgptr = NULL; 1895 else 1896 { 1897 request->u.servicereg.port.b[0] = *request->msgptr++; 1898 request->u.servicereg.port.b[1] = *request->msgptr++; 1899 } 1900 1901 request->u.servicereg.txtlen = get_uint16(&request->msgptr, request->msgend); 1902 msgTXTData = get_rdata(&request->msgptr, request->msgend, request->u.servicereg.txtlen); 1903 1904 if (!request->msgptr) { LogMsg("%3d: DNSServiceRegister(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 1905 1906 if (request->u.servicereg.txtlen) 1907 { 1908 request->u.servicereg.txtdata = mallocL("service_info txtdata", request->u.servicereg.txtlen); 1909 if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_regservice_request - malloc"); 1910 mDNSPlatformMemCopy(request->u.servicereg.txtdata, msgTXTData, request->u.servicereg.txtlen); 1911 } 1912 1913 // Check for sub-types after the service type 1914 request->u.servicereg.num_subtypes = ChopSubTypes(request->u.servicereg.type_as_string, &AnonData); // Note: Modifies regtype string to remove trailing subtypes 1915 if (request->u.servicereg.num_subtypes < 0) 1916 { 1917 LogMsg("ERROR: handle_regservice_request - ChopSubTypes failed %s", request->u.servicereg.type_as_string); 1918 goto bad_param; 1919 } 1920 if (AnonData) 1921 { 1922 int AnonDataLen = strlen(AnonData); 1923 if (AnonDataLen > MAX_ANONYMOUS_DATA) 1924 { 1925 LogMsg("ERROR: handle_regservice_request: AnonDataLen %d", AnonDataLen); 1926 goto bad_param; 1927 } 1928 request->u.servicereg.AnonData = mDNStrue; 1929 } 1930 else 1931 { 1932 request->u.servicereg.AnonData = mDNSfalse; 1933 } 1934 1935 // Don't try to construct "domainname t" until *after* ChopSubTypes has worked its magic 1936 if (!*request->u.servicereg.type_as_string || !MakeDomainNameFromDNSNameString(&request->u.servicereg.type, request->u.servicereg.type_as_string)) 1937 { LogMsg("ERROR: handle_regservice_request - type_as_string bad %s", request->u.servicereg.type_as_string); goto bad_param; } 1938 1939 if (!name[0]) 1940 { 1941 request->u.servicereg.name = mDNSStorage.nicelabel; 1942 request->u.servicereg.autoname = mDNStrue; 1943 } 1944 else 1945 { 1946 // If the client is allowing AutoRename, then truncate name to legal length before converting it to a DomainLabel 1947 if ((flags & kDNSServiceFlagsNoAutoRename) == 0) 1948 { 1949 int newlen = TruncateUTF8ToLength((mDNSu8*)name, mDNSPlatformStrLen(name), MAX_DOMAIN_LABEL); 1950 name[newlen] = 0; 1951 } 1952 if (!MakeDomainLabelFromLiteralString(&request->u.servicereg.name, name)) 1953 { LogMsg("ERROR: handle_regservice_request - name bad %s", name); goto bad_param; } 1954 request->u.servicereg.autoname = mDNSfalse; 1955 } 1956 1957 if (*domain) 1958 { 1959 request->u.servicereg.default_domain = mDNSfalse; 1960 if (!MakeDomainNameFromDNSNameString(&d, domain)) 1961 { LogMsg("ERROR: handle_regservice_request - domain bad %s", domain); goto bad_param; } 1962 } 1963 else 1964 { 1965 request->u.servicereg.default_domain = mDNStrue; 1966 MakeDomainNameFromDNSNameString(&d, "local."); 1967 } 1968 1969 // We don't allow the anonymous and the regular ones to coexist 1970 if (!CheckForMixedRegistrations(&request->u.servicereg.type, &d, request->u.servicereg.AnonData)) { goto bad_param; } 1971 1972 if (!ConstructServiceName(&srv, &request->u.servicereg.name, &request->u.servicereg.type, &d)) 1973 { 1974 LogMsg("ERROR: handle_regservice_request - Couldn't ConstructServiceName from, “%#s” “%##s” “%##s”", 1975 request->u.servicereg.name.c, request->u.servicereg.type.c, d.c); goto bad_param; 1976 } 1977 1978 if (!MakeDomainNameFromDNSNameString(&request->u.servicereg.host, host)) 1979 { LogMsg("ERROR: handle_regservice_request - host bad %s", host); goto bad_param; } 1980 request->u.servicereg.autorename = (flags & kDNSServiceFlagsNoAutoRename ) == 0; 1981 request->u.servicereg.allowremotequery = (flags & kDNSServiceFlagsAllowRemoteQuery) != 0; 1982 1983 // Some clients use mDNS for lightweight copy protection, registering a pseudo-service with 1984 // a port number of zero. When two instances of the protected client are allowed to run on one 1985 // machine, we don't want to see misleading "Bogus client" messages in syslog and the console. 1986 if (!mDNSIPPortIsZero(request->u.servicereg.port)) 1987 { 1988 int count = CountExistingRegistrations(&srv, request->u.servicereg.port); 1989 if (count) 1990 LogMsg("Client application[%d](%s) registered %d identical instances of service %##s port %u.", request->process_id, 1991 request->pid_name, count+1, srv.c, mDNSVal16(request->u.servicereg.port)); 1992 } 1993 1994 #if APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 1995 // Determine if this request should be promoted to use BLE triggered feature. 1996 if (shouldUseBLE(InterfaceID, 0, &request->u.servicereg.type, &d)) 1997 { 1998 request->flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 1999 LogInfo("handle_regservice_request: registration promoted to use kDNSServiceFlagsAutoTrigger"); 2000 } 2001 #endif // APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 2002 2003 LogOperation("%3d: DNSServiceRegister(%X, %d, \"%s\", \"%s\", \"%s\", \"%s\", %u) START PID[%d](%s)", 2004 request->sd, request->flags, interfaceIndex, name, request->u.servicereg.type_as_string, domain, host, 2005 mDNSVal16(request->u.servicereg.port), request->process_id, request->pid_name); 2006 2007 // We need to unconditionally set request->terminate, because even if we didn't successfully 2008 // start any registrations right now, subsequent configuration changes may cause successful 2009 // registrations to be added, and we'll need to cancel them before freeing this memory. 2010 // We also need to set request->terminate first, before adding additional service instances, 2011 // because the uds_validatelists uses the request->terminate function pointer to determine 2012 // what kind of request this is, and therefore what kind of list validation is required. 2013 request->terminate = regservice_termination_callback; 2014 2015 err = register_service_instance(request, &d); 2016 2017 #if TARGET_OS_EMBEDDED 2018 ++curr_num_regservices; 2019 if (curr_num_regservices > max_num_regservices) 2020 max_num_regservices = curr_num_regservices; 2021 #endif 2022 2023 #if 0 2024 err = AuthorizedDomain(request, &d, AutoRegistrationDomains) ? register_service_instance(request, &d) : mStatus_NoError; 2025 #endif 2026 if (!err) 2027 { 2028 if (request->u.servicereg.autoname) UpdateDeviceInfoRecord(&mDNSStorage); 2029 2030 if (!*domain) 2031 { 2032 DNameListElem *ptr; 2033 // Note that we don't report errors for non-local, non-explicit domains 2034 for (ptr = AutoRegistrationDomains; ptr; ptr = ptr->next) 2035 if (!ptr->uid || SystemUID(request->uid) || request->uid == ptr->uid) 2036 register_service_instance(request, &ptr->name); 2037 } 2038 } 2039 2040 return(err); 2041 2042 bad_param: 2043 freeL("handle_regservice_request (txtdata)", request->u.servicereg.txtdata); 2044 request->u.servicereg.txtdata = NULL; 2045 return mStatus_BadParamErr; 2046 } 2047 2048 // *************************************************************************** 2049 #if COMPILER_LIKES_PRAGMA_MARK 2050 #pragma mark - 2051 #pragma mark - DNSServiceBrowse 2052 #endif 2053 2054 mDNSlocal void FoundInstance(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 2055 { 2056 DNSServiceFlags flags = AddRecord ? kDNSServiceFlagsAdd : 0; 2057 request_state *req = question->QuestionContext; 2058 reply_state *rep; 2059 (void)m; // Unused 2060 2061 if (answer->rrtype != kDNSType_PTR) 2062 { LogMsg("%3d: FoundInstance: Should not be called with rrtype %d (not a PTR record)", req->sd, answer->rrtype); return; } 2063 2064 if (mDNSOpaque16IsZero(question->TargetQID) && (question->BrowseThreshold > 0) && (question->CurrentAnswers >= question->BrowseThreshold)) 2065 { 2066 flags |= kDNSServiceFlagsThresholdReached; 2067 } 2068 2069 // if returning a negative answer, then use question's name in reply 2070 if (answer->RecordType == kDNSRecordTypePacketNegative) 2071 { 2072 GenerateBrowseReply(&question->qname, answer->InterfaceID, req, &rep, browse_reply_op, flags, kDNSServiceErr_NoSuchRecord); 2073 goto validReply; 2074 } 2075 2076 if (GenerateNTDResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError) != mStatus_NoError) 2077 { 2078 if (SameDomainName(&req->u.browser.regtype, (const domainname*)"\x09_services\x07_dns-sd\x04_udp")) 2079 { 2080 // Special support to enable the DNSServiceBrowse call made by Bonjour Browser 2081 // Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse 2082 GenerateBrowseReply(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError); 2083 goto validReply; 2084 } 2085 2086 LogMsg("%3d: FoundInstance: %##s PTR %##s received from network is not valid DNS-SD service pointer", 2087 req->sd, answer->name->c, answer->rdata->u.name.c); 2088 return; 2089 } 2090 2091 validReply: 2092 2093 LogOperation("%3d: DNSServiceBrowse(%##s, %s) RESULT %s interface %d: %s", 2094 req->sd, question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", 2095 mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse), RRDisplayString(m, answer)); 2096 2097 append_reply(req, rep); 2098 } 2099 2100 mDNSlocal void SetQuestionPolicy(DNSQuestion *q, request_state *req) 2101 { 2102 q->euid = req->uid; 2103 // The policy is either based on pid or UUID. Pass a zero pid 2104 // to the "core" if the UUID is valid. If we always pass the pid, 2105 // then the "core" needs to determine whether the uuid is valid 2106 // by examining all the 16 bytes at the time of the policy 2107 // check and also when setting the delegate socket option. Also, it 2108 // requires that we zero out the uuid wherever the question is 2109 // initialized to make sure that it is not interpreted as valid. 2110 // To prevent these intrusive changes, just pass a zero pid to indicate 2111 // that pid is not valid when uuid is valid. In future if we need the 2112 // pid in the question, we will reevaluate this strategy. 2113 if (req->validUUID) 2114 { 2115 mDNSPlatformMemCopy(q->uuid, req->uuid, UUID_SIZE); 2116 q->pid = 0; 2117 } 2118 else 2119 { 2120 q->pid = req->process_id; 2121 } 2122 2123 //debugf("SetQuestionPolicy: q->euid[%d] q->pid[%d] uuid is valid : %s", q->euid, q->pid, req->validUUID ? "true" : "false"); 2124 } 2125 2126 mDNSlocal mStatus add_domain_to_browser(request_state *info, const domainname *d) 2127 { 2128 browser_t *b, *p; 2129 mStatus err; 2130 2131 for (p = info->u.browser.browsers; p; p = p->next) 2132 { 2133 if (SameDomainName(&p->domain, d)) 2134 { debugf("add_domain_to_browser %##s already in list", d->c); return mStatus_AlreadyRegistered; } 2135 } 2136 2137 b = mallocL("browser_t", sizeof(*b)); 2138 if (!b) return mStatus_NoMemoryErr; 2139 mDNSPlatformMemZero(b, sizeof(*b)); 2140 AssignDomainName(&b->domain, d); 2141 SetQuestionPolicy(&b->q, info); 2142 err = mDNS_StartBrowse(&mDNSStorage, &b->q, &info->u.browser.regtype, d, info->u.browser.AnonData, info->u.browser.interface_id, info->flags, 2143 info->u.browser.ForceMCast, (info->flags & kDNSServiceFlagsBackgroundTrafficClass) != 0, FoundInstance, info); 2144 if (err) 2145 { 2146 LogMsg("mDNS_StartBrowse returned %d for type %##s domain %##s", err, info->u.browser.regtype.c, d->c); 2147 freeL("browser_t/add_domain_to_browser", b); 2148 } 2149 else 2150 { 2151 b->next = info->u.browser.browsers; 2152 info->u.browser.browsers = b; 2153 2154 #if APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 2155 // Determine if this request should be promoted to use BLE triggered discovery. 2156 if (shouldUseBLE(info->u.browser.interface_id, 0, &info->u.browser.regtype, (domainname *) d)) 2157 { 2158 info->flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 2159 b->q.flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 2160 LogInfo("add_domain_to_browser: request promoted to use kDNSServiceFlagsAutoTrigger"); 2161 } 2162 #endif // APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 2163 2164 LogMcastQ(&b->q, info, q_start); 2165 if (callExternalHelpers(info->u.browser.interface_id, &b->domain, info->flags)) 2166 { 2167 domainname tmp; 2168 ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &b->domain); 2169 LogInfo("add_domain_to_browser: calling external_start_browsing_for_service()"); 2170 external_start_browsing_for_service(info->u.browser.interface_id, &tmp, kDNSType_PTR, info->flags); 2171 } 2172 } 2173 return err; 2174 } 2175 2176 mDNSlocal void browse_termination_callback(request_state *info) 2177 { 2178 if (info->u.browser.default_domain) 2179 { 2180 // Stop the domain enumeration queries to discover the WAB legacy browse domains 2181 LogInfo("%3d: DNSServiceBrowse Cancel WAB PID[%d](%s)", info->sd, info->process_id, info->pid_name); 2182 uDNS_StopWABQueries(&mDNSStorage, UDNS_WAB_LBROWSE_QUERY); 2183 } 2184 if (info->u.browser.AnonData) 2185 freeL("Anonymous", (void *)info->u.browser.AnonData); 2186 while (info->u.browser.browsers) 2187 { 2188 browser_t *ptr = info->u.browser.browsers; 2189 2190 if (callExternalHelpers(ptr->q.InterfaceID, &ptr->domain, ptr->q.flags)) 2191 { 2192 domainname tmp; 2193 ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &ptr->domain); 2194 LogInfo("browse_termination_callback: calling external_stop_browsing_for_service()"); 2195 external_stop_browsing_for_service(ptr->q.InterfaceID, &tmp, kDNSType_PTR, ptr->q.flags); 2196 } 2197 2198 LogOperation("%3d: DNSServiceBrowse(%X, %d, \"%##s\") STOP PID[%d](%s)", 2199 info->sd, info->flags, info->interfaceIndex, ptr->q.qname.c, info->process_id, info->pid_name); 2200 2201 info->u.browser.browsers = ptr->next; 2202 mDNS_StopBrowse(&mDNSStorage, &ptr->q); // no need to error-check result 2203 LogMcastQ(&ptr->q, info, q_stop); 2204 freeL("browser_t/browse_termination_callback", ptr); 2205 } 2206 } 2207 2208 mDNSlocal void udsserver_automatic_browse_domain_changed(const DNameListElem *const d, const mDNSBool add) 2209 { 2210 request_state *request; 2211 debugf("udsserver_automatic_browse_domain_changed: %s default browse domain %##s", add ? "Adding" : "Removing", d->name.c); 2212 2213 for (request = all_requests; request; request = request->next) 2214 { 2215 if (request->terminate != browse_termination_callback) continue; // Not a browse operation 2216 if (!request->u.browser.default_domain) continue; // Not an auto-browse operation 2217 if (!d->uid || SystemUID(request->uid) || request->uid == d->uid) 2218 { 2219 browser_t **ptr = &request->u.browser.browsers; 2220 while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next; 2221 if (add) 2222 { 2223 // If we don't already have this domain in our list for this browse operation, add it now 2224 if (!*ptr) add_domain_to_browser(request, &d->name); 2225 else debugf("udsserver_automatic_browse_domain_changed %##s already in list, not re-adding", &d->name); 2226 } 2227 else 2228 { 2229 if (!*ptr) LogMsg("udsserver_automatic_browse_domain_changed ERROR %##s not found", &d->name); 2230 else 2231 { 2232 DNameListElem *p; 2233 for (p = AutoBrowseDomains; p; p=p->next) 2234 if (!p->uid || SystemUID(request->uid) || request->uid == p->uid) 2235 if (SameDomainName(&d->name, &p->name)) break; 2236 if (p) debugf("udsserver_automatic_browse_domain_changed %##s still in list, not removing", &d->name); 2237 else 2238 { 2239 browser_t *rem = *ptr; 2240 *ptr = (*ptr)->next; 2241 mDNS_StopQueryWithRemoves(&mDNSStorage, &rem->q); 2242 freeL("browser_t/udsserver_automatic_browse_domain_changed", rem); 2243 } 2244 } 2245 } 2246 } 2247 } 2248 } 2249 2250 mDNSlocal void FreeARElemCallback(mDNS *const m, AuthRecord *const rr, mStatus result) 2251 { 2252 (void)m; // unused 2253 if (result == mStatus_MemFree) 2254 { 2255 // On shutdown, mDNS_Close automatically deregisters all records 2256 // Since in this case no one has called DeregisterLocalOnlyDomainEnumPTR to cut the record 2257 // from the LocalDomainEnumRecords list, we do this here before we free the memory. 2258 // (This should actually no longer be necessary, now that we do the proper cleanup in 2259 // udsserver_exit. To confirm this, we'll log an error message if we do find a record that 2260 // hasn't been cut from the list yet. If these messages don't appear, we can delete this code.) 2261 ARListElem **ptr = &LocalDomainEnumRecords; 2262 while (*ptr && &(*ptr)->ar != rr) ptr = &(*ptr)->next; 2263 if (*ptr) { *ptr = (*ptr)->next; LogMsg("FreeARElemCallback: Have to cut %s", ARDisplayString(m, rr)); } 2264 mDNSPlatformMemFree(rr->RecordContext); 2265 } 2266 } 2267 2268 // RegisterLocalOnlyDomainEnumPTR and DeregisterLocalOnlyDomainEnumPTR largely duplicate code in 2269 // "FoundDomain" in uDNS.c for creating and destroying these special mDNSInterface_LocalOnly records. 2270 // We may want to turn the common code into a subroutine. 2271 2272 mDNSlocal void RegisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type) 2273 { 2274 // allocate/register legacy and non-legacy _browse PTR record 2275 mStatus err; 2276 ARListElem *ptr = mDNSPlatformMemAllocate(sizeof(*ptr)); 2277 2278 debugf("Incrementing %s refcount for %##s", 2279 (type == mDNS_DomainTypeBrowse ) ? "browse domain " : 2280 (type == mDNS_DomainTypeRegistration ) ? "registration dom" : 2281 (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c); 2282 2283 mDNS_SetupResourceRecord(&ptr->ar, mDNSNULL, mDNSInterface_LocalOnly, kDNSType_PTR, 7200, kDNSRecordTypeShared, AuthRecordLocalOnly, FreeARElemCallback, ptr); 2284 MakeDomainNameFromDNSNameString(&ptr->ar.namestorage, mDNS_DomainTypeNames[type]); 2285 AppendDNSNameString (&ptr->ar.namestorage, "local"); 2286 AssignDomainName(&ptr->ar.resrec.rdata->u.name, d); 2287 err = mDNS_Register(m, &ptr->ar); 2288 if (err) 2289 { 2290 LogMsg("SetSCPrefsBrowseDomain: mDNS_Register returned error %d", err); 2291 mDNSPlatformMemFree(ptr); 2292 } 2293 else 2294 { 2295 ptr->next = LocalDomainEnumRecords; 2296 LocalDomainEnumRecords = ptr; 2297 } 2298 } 2299 2300 mDNSlocal void DeregisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type) 2301 { 2302 ARListElem **ptr = &LocalDomainEnumRecords; 2303 domainname lhs; // left-hand side of PTR, for comparison 2304 2305 debugf("Decrementing %s refcount for %##s", 2306 (type == mDNS_DomainTypeBrowse ) ? "browse domain " : 2307 (type == mDNS_DomainTypeRegistration ) ? "registration dom" : 2308 (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c); 2309 2310 MakeDomainNameFromDNSNameString(&lhs, mDNS_DomainTypeNames[type]); 2311 AppendDNSNameString (&lhs, "local"); 2312 2313 while (*ptr) 2314 { 2315 if (SameDomainName(&(*ptr)->ar.resrec.rdata->u.name, d) && SameDomainName((*ptr)->ar.resrec.name, &lhs)) 2316 { 2317 ARListElem *rem = *ptr; 2318 *ptr = (*ptr)->next; 2319 mDNS_Deregister(m, &rem->ar); 2320 return; 2321 } 2322 else ptr = &(*ptr)->next; 2323 } 2324 } 2325 2326 mDNSlocal void AddAutoBrowseDomain(const mDNSu32 uid, const domainname *const name) 2327 { 2328 DNameListElem *new = mDNSPlatformMemAllocate(sizeof(DNameListElem)); 2329 if (!new) { LogMsg("ERROR: malloc"); return; } 2330 AssignDomainName(&new->name, name); 2331 new->uid = uid; 2332 new->next = AutoBrowseDomains; 2333 AutoBrowseDomains = new; 2334 udsserver_automatic_browse_domain_changed(new, mDNStrue); 2335 } 2336 2337 mDNSlocal void RmvAutoBrowseDomain(const mDNSu32 uid, const domainname *const name) 2338 { 2339 DNameListElem **p = &AutoBrowseDomains; 2340 while (*p && (!SameDomainName(&(*p)->name, name) || (*p)->uid != uid)) p = &(*p)->next; 2341 if (!*p) LogMsg("RmvAutoBrowseDomain: Got remove event for domain %##s not in list", name->c); 2342 else 2343 { 2344 DNameListElem *ptr = *p; 2345 *p = ptr->next; 2346 udsserver_automatic_browse_domain_changed(ptr, mDNSfalse); 2347 mDNSPlatformMemFree(ptr); 2348 } 2349 } 2350 2351 mDNSlocal void SetPrefsBrowseDomains(mDNS *m, DNameListElem *browseDomains, mDNSBool add) 2352 { 2353 DNameListElem *d; 2354 for (d = browseDomains; d; d = d->next) 2355 { 2356 if (add) 2357 { 2358 RegisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse); 2359 AddAutoBrowseDomain(d->uid, &d->name); 2360 } 2361 else 2362 { 2363 DeregisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse); 2364 RmvAutoBrowseDomain(d->uid, &d->name); 2365 } 2366 } 2367 } 2368 2369 #if APPLE_OSX_mDNSResponder 2370 2371 mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m) 2372 { 2373 int num_autoname = 0; 2374 request_state *req; 2375 2376 // Don't need to register the device info record for kDNSServiceInterfaceIndexLocalOnly registrations. 2377 for (req = all_requests; req; req = req->next) 2378 { 2379 if (req->terminate == regservice_termination_callback && req->u.servicereg.autoname && req->interfaceIndex != kDNSServiceInterfaceIndexLocalOnly) 2380 num_autoname++; 2381 } 2382 2383 // If DeviceInfo record is currently registered, see if we need to deregister it 2384 if (m->DeviceInfo.resrec.RecordType != kDNSRecordTypeUnregistered) 2385 if (num_autoname == 0 || !SameDomainLabelCS(m->DeviceInfo.resrec.name->c, m->nicelabel.c)) 2386 { 2387 LogOperation("UpdateDeviceInfoRecord Deregister %##s", m->DeviceInfo.resrec.name); 2388 mDNS_Deregister(m, &m->DeviceInfo); 2389 } 2390 2391 // If DeviceInfo record is not currently registered, see if we need to register it 2392 if (m->DeviceInfo.resrec.RecordType == kDNSRecordTypeUnregistered) 2393 if (num_autoname > 0) 2394 { 2395 mDNS_SetupResourceRecord(&m->DeviceInfo, mDNSNULL, mDNSNULL, kDNSType_TXT, kStandardTTL, kDNSRecordTypeAdvisory, AuthRecordAny, mDNSNULL, mDNSNULL); 2396 ConstructServiceName(&m->DeviceInfo.namestorage, &m->nicelabel, &DeviceInfoName, &localdomain); 2397 m->DeviceInfo.resrec.rdlength = initializeDeviceInfoTXT(m, m->DeviceInfo.resrec.rdata->u.data); 2398 LogOperation("UpdateDeviceInfoRecord Register %##s", m->DeviceInfo.resrec.name); 2399 mDNS_Register(m, &m->DeviceInfo); 2400 } 2401 } 2402 #else // APPLE_OSX_mDNSResponder 2403 mDNSlocal void UpdateDeviceInfoRecord(mDNS *const m) 2404 { 2405 (void)m; // unused 2406 } 2407 #endif // APPLE_OSX_mDNSResponder 2408 2409 mDNSexport void udsserver_handle_configchange(mDNS *const m) 2410 { 2411 request_state *req; 2412 service_instance *ptr; 2413 DNameListElem *RegDomains = NULL; 2414 DNameListElem *BrowseDomains = NULL; 2415 DNameListElem *p; 2416 2417 UpdateDeviceInfoRecord(m); 2418 2419 // For autoname services, see if the default service name has changed, necessitating an automatic update 2420 for (req = all_requests; req; req = req->next) 2421 if (req->terminate == regservice_termination_callback) 2422 if (req->u.servicereg.autoname && !SameDomainLabelCS(req->u.servicereg.name.c, m->nicelabel.c)) 2423 { 2424 req->u.servicereg.name = m->nicelabel; 2425 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 2426 { 2427 ptr->renameonmemfree = 1; 2428 if (ptr->clientnotified) SendServiceRemovalNotification(&ptr->srs); 2429 LogInfo("udsserver_handle_configchange: Calling deregister for Service %##s", ptr->srs.RR_PTR.resrec.name->c); 2430 if (mDNS_DeregisterService_drt(m, &ptr->srs, mDNS_Dereg_rapid)) 2431 regservice_callback(m, &ptr->srs, mStatus_MemFree); // If service deregistered already, we can re-register immediately 2432 } 2433 } 2434 2435 // Let the platform layer get the current DNS information 2436 mDNS_Lock(m); 2437 mDNSPlatformSetDNSConfig(mDNSfalse, mDNSfalse, mDNSNULL, &RegDomains, &BrowseDomains, mDNSfalse); 2438 mDNS_Unlock(m); 2439 2440 // Any automatic registration domains are also implicitly automatic browsing domains 2441 if (RegDomains) SetPrefsBrowseDomains(m, RegDomains, mDNStrue); // Add the new list first 2442 if (AutoRegistrationDomains) SetPrefsBrowseDomains(m, AutoRegistrationDomains, mDNSfalse); // Then clear the old list 2443 2444 // Add any new domains not already in our AutoRegistrationDomains list 2445 for (p=RegDomains; p; p=p->next) 2446 { 2447 DNameListElem **pp = &AutoRegistrationDomains; 2448 while (*pp && ((*pp)->uid != p->uid || !SameDomainName(&(*pp)->name, &p->name))) pp = &(*pp)->next; 2449 if (!*pp) // If not found in our existing list, this is a new default registration domain 2450 { 2451 RegisterLocalOnlyDomainEnumPTR(m, &p->name, mDNS_DomainTypeRegistration); 2452 udsserver_default_reg_domain_changed(p, mDNStrue); 2453 } 2454 else // else found same domainname in both old and new lists, so no change, just delete old copy 2455 { 2456 DNameListElem *del = *pp; 2457 *pp = (*pp)->next; 2458 mDNSPlatformMemFree(del); 2459 } 2460 } 2461 2462 // Delete any domains in our old AutoRegistrationDomains list that are now gone 2463 while (AutoRegistrationDomains) 2464 { 2465 DNameListElem *del = AutoRegistrationDomains; 2466 AutoRegistrationDomains = AutoRegistrationDomains->next; // Cut record from list FIRST, 2467 DeregisterLocalOnlyDomainEnumPTR(m, &del->name, mDNS_DomainTypeRegistration); 2468 udsserver_default_reg_domain_changed(del, mDNSfalse); // before calling udsserver_default_reg_domain_changed() 2469 mDNSPlatformMemFree(del); 2470 } 2471 2472 // Now we have our new updated automatic registration domain list 2473 AutoRegistrationDomains = RegDomains; 2474 2475 // Add new browse domains to internal list 2476 if (BrowseDomains) SetPrefsBrowseDomains(m, BrowseDomains, mDNStrue); 2477 2478 // Remove old browse domains from internal list 2479 if (SCPrefBrowseDomains) 2480 { 2481 SetPrefsBrowseDomains(m, SCPrefBrowseDomains, mDNSfalse); 2482 while (SCPrefBrowseDomains) 2483 { 2484 DNameListElem *fptr = SCPrefBrowseDomains; 2485 SCPrefBrowseDomains = SCPrefBrowseDomains->next; 2486 mDNSPlatformMemFree(fptr); 2487 } 2488 } 2489 2490 // Replace the old browse domains array with the new array 2491 SCPrefBrowseDomains = BrowseDomains; 2492 } 2493 2494 mDNSlocal void AutomaticBrowseDomainChange(mDNS *const m, DNSQuestion *q, const ResourceRecord *const answer, QC_result AddRecord) 2495 { 2496 (void)m; // unused; 2497 (void)q; // unused 2498 2499 LogOperation("AutomaticBrowseDomainChange: %s automatic browse domain %##s", 2500 AddRecord ? "Adding" : "Removing", answer->rdata->u.name.c); 2501 2502 if (AddRecord) AddAutoBrowseDomain(0, &answer->rdata->u.name); 2503 else RmvAutoBrowseDomain(0, &answer->rdata->u.name); 2504 } 2505 2506 mDNSlocal mStatus handle_browse_request(request_state *request) 2507 { 2508 char regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME]; 2509 domainname typedn, d, temp; 2510 mDNSs32 NumSubTypes; 2511 char *AnonData = mDNSNULL; 2512 mStatus err = mStatus_NoError; 2513 int AnonDataLen; 2514 2515 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 2516 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 2517 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 2518 2519 // The browse is scoped to a specific interface index, but the 2520 // interface is not currently in our list. 2521 if (interfaceIndex && !InterfaceID) 2522 { 2523 // If it's one of the specially defined inteface index values, just return an error. 2524 if (PreDefinedInterfaceIndex(interfaceIndex)) 2525 { 2526 LogInfo("handle_browse_request: bad interfaceIndex %d", interfaceIndex); 2527 return(mStatus_BadParamErr); 2528 } 2529 2530 // Otherwise, use the specified interface index value and the browse will 2531 // be applied to that interface when it comes up. 2532 InterfaceID = (mDNSInterfaceID)(uintptr_t)interfaceIndex; 2533 LogInfo("handle_browse_request: browse pending for interface index %d", interfaceIndex); 2534 } 2535 2536 if (get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 || 2537 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) return(mStatus_BadParamErr); 2538 2539 if (!request->msgptr) { LogMsg("%3d: DNSServiceBrowse(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 2540 2541 request->flags = flags; 2542 request->interfaceIndex = interfaceIndex; 2543 typedn.c[0] = 0; 2544 NumSubTypes = ChopSubTypes(regtype, &AnonData); // Note: Modifies regtype string to remove trailing subtypes 2545 if (NumSubTypes < 0 || NumSubTypes > 1) 2546 return(mStatus_BadParamErr); 2547 AnonDataLen = 0; 2548 if (AnonData) 2549 { 2550 AnonDataLen = strlen(AnonData); 2551 if (AnonDataLen > MAX_ANONYMOUS_DATA) 2552 { 2553 LogMsg("handle_browse_request: AnonDataLen %d", AnonDataLen); 2554 return(mStatus_BadParamErr); 2555 } 2556 // Account for the null byte 2557 AnonDataLen += 1; 2558 } 2559 if (NumSubTypes == 1) 2560 { 2561 if (!AppendDNSNameString(&typedn, regtype + strlen(regtype) + 1 + AnonDataLen)) 2562 return(mStatus_BadParamErr); 2563 } 2564 2565 if (!regtype[0] || !AppendDNSNameString(&typedn, regtype)) return(mStatus_BadParamErr); 2566 2567 if (!MakeDomainNameFromDNSNameString(&temp, regtype)) return(mStatus_BadParamErr); 2568 // For over-long service types, we only allow domain "local" 2569 if (temp.c[0] > 15 && domain[0] == 0) mDNSPlatformStrLCopy(domain, "local.", sizeof(domain)); 2570 2571 // Set up browser info 2572 request->u.browser.ForceMCast = (flags & kDNSServiceFlagsForceMulticast) != 0; 2573 request->u.browser.interface_id = InterfaceID; 2574 AssignDomainName(&request->u.browser.regtype, &typedn); 2575 request->u.browser.default_domain = !domain[0]; 2576 request->u.browser.browsers = NULL; 2577 2578 LogOperation("%3d: DNSServiceBrowse(%X, %d, \"%##s\", \"%s\") START PID[%d](%s)", 2579 request->sd, request->flags, interfaceIndex, request->u.browser.regtype.c, domain, request->process_id, request->pid_name); 2580 2581 if (request->u.browser.default_domain) 2582 { 2583 // Start the domain enumeration queries to discover the WAB browse domains 2584 LogInfo("%3d: DNSServiceBrowse Start WAB PID[%d](%s)", request->sd, request->process_id, request->pid_name); 2585 uDNS_StartWABQueries(&mDNSStorage, UDNS_WAB_LBROWSE_QUERY); 2586 } 2587 request->u.browser.AnonData = mDNSNULL; 2588 if (AnonData) 2589 { 2590 int len = strlen(AnonData) + 1; 2591 request->u.browser.AnonData = mallocL("Anonymous", len); 2592 if (!request->u.browser.AnonData) 2593 return mStatus_NoMemoryErr; 2594 else 2595 mDNSPlatformMemCopy((void *)request->u.browser.AnonData, AnonData, len); 2596 } 2597 // We need to unconditionally set request->terminate, because even if we didn't successfully 2598 // start any browses right now, subsequent configuration changes may cause successful 2599 // browses to be added, and we'll need to cancel them before freeing this memory. 2600 request->terminate = browse_termination_callback; 2601 2602 if (domain[0]) 2603 { 2604 if (!MakeDomainNameFromDNSNameString(&d, domain)) return(mStatus_BadParamErr); 2605 err = add_domain_to_browser(request, &d); 2606 } 2607 else 2608 { 2609 DNameListElem *sdom; 2610 for (sdom = AutoBrowseDomains; sdom; sdom = sdom->next) 2611 if (!sdom->uid || SystemUID(request->uid) || request->uid == sdom->uid) 2612 { 2613 err = add_domain_to_browser(request, &sdom->name); 2614 if (err) 2615 { 2616 if (SameDomainName(&sdom->name, &localdomain)) break; 2617 else err = mStatus_NoError; // suppress errors for non-local "default" domains 2618 } 2619 } 2620 } 2621 2622 return(err); 2623 } 2624 2625 // *************************************************************************** 2626 #if COMPILER_LIKES_PRAGMA_MARK 2627 #pragma mark - 2628 #pragma mark - DNSServiceResolve 2629 #endif 2630 2631 mDNSlocal void resolve_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 2632 { 2633 size_t len = 0; 2634 char fullname[MAX_ESCAPED_DOMAIN_NAME], target[MAX_ESCAPED_DOMAIN_NAME] = "0"; 2635 char *data; 2636 reply_state *rep; 2637 request_state *req = question->QuestionContext; 2638 const DNSServiceErrorType error = 2639 (answer->RecordType == kDNSRecordTypePacketNegative) ? kDNSServiceErr_NoSuchRecord : kDNSServiceErr_NoError; 2640 (void)m; // Unused 2641 2642 LogOperation("%3d: DNSServiceResolve(%##s) %s interface %d: %s", 2643 req->sd, question->qname.c, AddRecord ? "ADD" : "RMV", 2644 mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse), RRDisplayString(m, answer)); 2645 2646 if (!AddRecord) 2647 { 2648 if (req->u.resolve.srv == answer) req->u.resolve.srv = mDNSNULL; 2649 if (req->u.resolve.txt == answer) req->u.resolve.txt = mDNSNULL; 2650 return; 2651 } 2652 2653 if (answer->rrtype == kDNSType_SRV) req->u.resolve.srv = answer; 2654 if (answer->rrtype == kDNSType_TXT) req->u.resolve.txt = answer; 2655 2656 if (!req->u.resolve.txt || !req->u.resolve.srv) return; // only deliver result to client if we have both answers 2657 2658 ConvertDomainNameToCString(answer->name, fullname); 2659 2660 if (answer->RecordType != kDNSRecordTypePacketNegative) 2661 ConvertDomainNameToCString(&req->u.resolve.srv->rdata->u.srv.target, target); 2662 2663 // calculate reply length 2664 len += sizeof(DNSServiceFlags); 2665 len += sizeof(mDNSu32); // interface index 2666 len += sizeof(DNSServiceErrorType); 2667 len += strlen(fullname) + 1; 2668 len += strlen(target) + 1; 2669 len += 2 * sizeof(mDNSu16); // port, txtLen 2670 len += req->u.resolve.txt->rdlength; 2671 2672 // allocate/init reply header 2673 rep = create_reply(resolve_reply_op, len, req); 2674 rep->rhdr->flags = dnssd_htonl(0); 2675 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse)); 2676 rep->rhdr->error = dnssd_htonl(error); 2677 2678 data = (char *)&rep->rhdr[1]; 2679 2680 // write reply data to message 2681 put_string(fullname, &data); 2682 put_string(target, &data); 2683 *data++ = req->u.resolve.srv->rdata->u.srv.port.b[0]; 2684 *data++ = req->u.resolve.srv->rdata->u.srv.port.b[1]; 2685 put_uint16(req->u.resolve.txt->rdlength, &data); 2686 put_rdata (req->u.resolve.txt->rdlength, req->u.resolve.txt->rdata->u.data, &data); 2687 2688 LogOperation("%3d: DNSServiceResolve(%s) RESULT %s:%d", req->sd, fullname, target, mDNSVal16(req->u.resolve.srv->rdata->u.srv.port)); 2689 append_reply(req, rep); 2690 } 2691 2692 mDNSlocal void resolve_termination_callback(request_state *request) 2693 { 2694 LogOperation("%3d: DNSServiceResolve(%X, %d, \"%##s\") STOP PID[%d](%s)", 2695 request->sd, request->flags, request->interfaceIndex, request->u.resolve.qtxt.qname.c, request->process_id, request->pid_name); 2696 mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qtxt); 2697 mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv); 2698 LogMcastQ(&request->u.resolve.qsrv, request, q_stop); 2699 if (request->u.resolve.external_advertise) 2700 external_stop_resolving_service(request->u.resolve.qsrv.InterfaceID, &request->u.resolve.qsrv.qname, request->flags); 2701 } 2702 2703 mDNSlocal mStatus handle_resolve_request(request_state *request) 2704 { 2705 char name[256], regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME]; 2706 domainname fqdn; 2707 mStatus err; 2708 2709 // extract the data from the message 2710 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 2711 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 2712 mDNSInterfaceID InterfaceID; 2713 2714 // Map kDNSServiceInterfaceIndexP2P to kDNSServiceInterfaceIndexAny with the kDNSServiceFlagsIncludeP2P 2715 // flag set so that the resolve will run over P2P interfaces that are not yet created. 2716 if (interfaceIndex == kDNSServiceInterfaceIndexP2P) 2717 { 2718 LogOperation("handle_resolve_request: mapping kDNSServiceInterfaceIndexP2P to kDNSServiceInterfaceIndexAny + kDNSServiceFlagsIncludeP2P"); 2719 flags |= kDNSServiceFlagsIncludeP2P; 2720 interfaceIndex = kDNSServiceInterfaceIndexAny; 2721 } 2722 2723 InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 2724 2725 // The operation is scoped to a specific interface index, but the 2726 // interface is not currently in our list. 2727 if (interfaceIndex && !InterfaceID) 2728 { 2729 // If it's one of the specially defined inteface index values, just return an error. 2730 if (PreDefinedInterfaceIndex(interfaceIndex)) 2731 { 2732 LogInfo("handle_resolve_request: bad interfaceIndex %d", interfaceIndex); 2733 return(mStatus_BadParamErr); 2734 } 2735 2736 // Otherwise, use the specified interface index value and the operation will 2737 // be applied to that interface when it comes up. 2738 InterfaceID = (mDNSInterfaceID)(uintptr_t)interfaceIndex; 2739 LogInfo("handle_resolve_request: resolve pending for interface index %d", interfaceIndex); 2740 } 2741 2742 if (get_string(&request->msgptr, request->msgend, name, 256) < 0 || 2743 get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 || 2744 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) 2745 { LogMsg("ERROR: handle_resolve_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); } 2746 2747 if (!request->msgptr) { LogMsg("%3d: DNSServiceResolve(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 2748 2749 if (build_domainname_from_strings(&fqdn, name, regtype, domain) < 0) 2750 { LogMsg("ERROR: handle_resolve_request bad “%s” “%s” “%s”", name, regtype, domain); return(mStatus_BadParamErr); } 2751 2752 mDNSPlatformMemZero(&request->u.resolve, sizeof(request->u.resolve)); 2753 2754 #if APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 2755 // Determine if this request should be promoted to use BLE triggered discovery. 2756 if (shouldUseBLE(InterfaceID, 0, (domainname *)SkipLeadingLabels(&fqdn, 1), &fqdn)) 2757 { 2758 flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 2759 LogInfo("handle_resolve_request: request promoted to use kDNSServiceFlagsAutoTrigger"); 2760 } 2761 #endif // APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 2762 2763 request->flags = flags; 2764 request->interfaceIndex = interfaceIndex; 2765 2766 // format questions 2767 request->u.resolve.qsrv.InterfaceID = InterfaceID; 2768 request->u.resolve.qsrv.flags = flags; 2769 request->u.resolve.qsrv.Target = zeroAddr; 2770 AssignDomainName(&request->u.resolve.qsrv.qname, &fqdn); 2771 request->u.resolve.qsrv.qtype = kDNSType_SRV; 2772 request->u.resolve.qsrv.qclass = kDNSClass_IN; 2773 request->u.resolve.qsrv.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 2774 request->u.resolve.qsrv.ExpectUnique = mDNStrue; 2775 request->u.resolve.qsrv.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 2776 request->u.resolve.qsrv.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 2777 request->u.resolve.qsrv.SuppressUnusable = mDNSfalse; 2778 request->u.resolve.qsrv.SearchListIndex = 0; 2779 request->u.resolve.qsrv.AppendSearchDomains = 0; 2780 request->u.resolve.qsrv.RetryWithSearchDomains = mDNSfalse; 2781 request->u.resolve.qsrv.TimeoutQuestion = 0; 2782 request->u.resolve.qsrv.WakeOnResolve = (flags & kDNSServiceFlagsWakeOnResolve) != 0; 2783 request->u.resolve.qsrv.UseBackgroundTrafficClass = (flags & kDNSServiceFlagsBackgroundTrafficClass) != 0; 2784 request->u.resolve.qsrv.ValidationRequired = 0; 2785 request->u.resolve.qsrv.ValidatingResponse = 0; 2786 request->u.resolve.qsrv.ProxyQuestion = 0; 2787 request->u.resolve.qsrv.qnameOrig = mDNSNULL; 2788 request->u.resolve.qsrv.AnonInfo = mDNSNULL; 2789 request->u.resolve.qsrv.pid = request->process_id; 2790 request->u.resolve.qsrv.euid = request->uid; 2791 request->u.resolve.qsrv.QuestionCallback = resolve_result_callback; 2792 request->u.resolve.qsrv.QuestionContext = request; 2793 2794 request->u.resolve.qtxt.InterfaceID = InterfaceID; 2795 request->u.resolve.qtxt.flags = flags; 2796 request->u.resolve.qtxt.Target = zeroAddr; 2797 AssignDomainName(&request->u.resolve.qtxt.qname, &fqdn); 2798 request->u.resolve.qtxt.qtype = kDNSType_TXT; 2799 request->u.resolve.qtxt.qclass = kDNSClass_IN; 2800 request->u.resolve.qtxt.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 2801 request->u.resolve.qtxt.ExpectUnique = mDNStrue; 2802 request->u.resolve.qtxt.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 2803 request->u.resolve.qtxt.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 2804 request->u.resolve.qtxt.SuppressUnusable = mDNSfalse; 2805 request->u.resolve.qtxt.SearchListIndex = 0; 2806 request->u.resolve.qtxt.AppendSearchDomains = 0; 2807 request->u.resolve.qtxt.RetryWithSearchDomains = mDNSfalse; 2808 request->u.resolve.qtxt.TimeoutQuestion = 0; 2809 request->u.resolve.qtxt.WakeOnResolve = 0; 2810 request->u.resolve.qtxt.UseBackgroundTrafficClass = (flags & kDNSServiceFlagsBackgroundTrafficClass) != 0; 2811 request->u.resolve.qtxt.ValidationRequired = 0; 2812 request->u.resolve.qtxt.ValidatingResponse = 0; 2813 request->u.resolve.qtxt.ProxyQuestion = 0; 2814 request->u.resolve.qtxt.qnameOrig = mDNSNULL; 2815 request->u.resolve.qtxt.AnonInfo = mDNSNULL; 2816 request->u.resolve.qtxt.pid = request->process_id; 2817 request->u.resolve.qtxt.euid = request->uid; 2818 request->u.resolve.qtxt.QuestionCallback = resolve_result_callback; 2819 request->u.resolve.qtxt.QuestionContext = request; 2820 2821 request->u.resolve.ReportTime = NonZeroTime(mDNS_TimeNow(&mDNSStorage) + 130 * mDNSPlatformOneSecond); 2822 2823 request->u.resolve.external_advertise = mDNSfalse; 2824 2825 #if 0 2826 if (!AuthorizedDomain(request, &fqdn, AutoBrowseDomains)) return(mStatus_NoError); 2827 #endif 2828 2829 // ask the questions 2830 LogOperation("%3d: DNSServiceResolve(%X, %d, \"%##s\") START PID[%d](%s)", request->sd, flags, interfaceIndex, 2831 request->u.resolve.qsrv.qname.c, request->process_id, request->pid_name); 2832 2833 err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qsrv); 2834 2835 if (!err) 2836 { 2837 err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qtxt); 2838 if (err) 2839 { 2840 mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv); 2841 } 2842 else 2843 { 2844 request->terminate = resolve_termination_callback; 2845 LogMcastQ(&request->u.resolve.qsrv, request, q_start); 2846 if (callExternalHelpers(InterfaceID, &fqdn, flags)) 2847 { 2848 request->u.resolve.external_advertise = mDNStrue; 2849 LogInfo("handle_resolve_request: calling external_start_resolving_service()"); 2850 external_start_resolving_service(InterfaceID, &fqdn, flags); 2851 } 2852 } 2853 } 2854 2855 return(err); 2856 } 2857 2858 // *************************************************************************** 2859 #if COMPILER_LIKES_PRAGMA_MARK 2860 #pragma mark - 2861 #pragma mark - DNSServiceQueryRecord 2862 #endif 2863 2864 // mDNS operation functions. Each operation has 3 associated functions - a request handler that parses 2865 // the client's request and makes the appropriate mDNSCore call, a result handler (passed as a callback 2866 // to the mDNSCore routine) that sends results back to the client, and a termination routine that aborts 2867 // the mDNSCore operation if the client dies or closes its socket. 2868 2869 // Returns -1 to tell the caller that it should not try to reissue the query anymore 2870 // Returns 1 on successfully appending a search domain and the caller should reissue the new query 2871 // Returns 0 when there are no more search domains and the caller should reissue the query 2872 mDNSlocal int AppendNewSearchDomain(DNSQuestion *question) 2873 { 2874 domainname *sd; 2875 mStatus err; 2876 2877 // Sanity check: The caller already checks this. We use -1 to indicate that we have searched all 2878 // the domains and should try the single label query directly on the wire. 2879 if (question->SearchListIndex == -1) 2880 { 2881 LogMsg("AppendNewSearchDomain: question %##s (%s) SearchListIndex is -1", question->qname.c, DNSTypeName(question->qtype)); 2882 return -1; 2883 } 2884 2885 if (!question->AppendSearchDomains) 2886 { 2887 LogMsg("AppendNewSearchDomain: question %##s (%s) AppendSearchDoamins is 0", question->qname.c, DNSTypeName(question->qtype)); 2888 return -1; 2889 } 2890 2891 // Save the original name, before we modify them below. 2892 if (!question->qnameOrig) 2893 { 2894 question->qnameOrig = mallocL("AppendNewSearchDomain", sizeof(domainname)); 2895 if (!question->qnameOrig) { LogMsg("AppendNewSearchDomain: ERROR!! malloc failure"); return -1; } 2896 question->qnameOrig->c[0] = 0; 2897 AssignDomainName(question->qnameOrig, &question->qname); 2898 LogInfo("AppendSearchDomain: qnameOrig %##s", question->qnameOrig->c); 2899 } 2900 2901 sd = uDNS_GetNextSearchDomain(question->InterfaceID, &question->SearchListIndex, !question->AppendLocalSearchDomains); 2902 // We use -1 to indicate that we have searched all the domains and should try the single label 2903 // query directly on the wire. uDNS_GetNextSearchDomain should never return a negative value 2904 if (question->SearchListIndex == -1) 2905 { 2906 LogMsg("AppendNewSearchDomain: ERROR!! uDNS_GetNextSearchDomain returned -1"); 2907 return -1; 2908 } 2909 2910 // Not a common case. Perhaps, we should try the next search domain if it exceeds ? 2911 if (sd && (DomainNameLength(question->qnameOrig) + DomainNameLength(sd)) > MAX_DOMAIN_NAME) 2912 { 2913 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)); 2914 return -1; 2915 } 2916 2917 // if there are no more search domains and we have already tried this question 2918 // without appending search domains, then we are done. 2919 if (!sd && !ApplySearchDomainsFirst(question)) 2920 { 2921 LogInfo("AppendNewSearchDomain: No more search domains for question with name %##s (%s), not trying anymore", question->qname.c, DNSTypeName(question->qtype)); 2922 return -1; 2923 } 2924 2925 // Stop the question before changing the name as negative cache entries could be pointing at this question. 2926 // Even if we don't change the question in the case of returning 0, the caller is going to restart the 2927 // question. 2928 err = mDNS_StopQuery(&mDNSStorage, question); 2929 if (err) { LogMsg("AppendNewSearchDomain: ERROR!! %##s %s mDNS_StopQuery: %d, while retrying with search domains", question->qname.c, DNSTypeName(question->qtype), (int)err); } 2930 2931 AssignDomainName(&question->qname, question->qnameOrig); 2932 if (sd) 2933 { 2934 AppendDomainName(&question->qname, sd); 2935 LogInfo("AppnedNewSearchDomain: Returning question with name %##s, SearchListIndex %d", question->qname.c, question->SearchListIndex); 2936 return 1; 2937 } 2938 2939 // Try the question as single label 2940 LogInfo("AppnedNewSearchDomain: No more search domains for question with name %##s (%s), trying one last time", question->qname.c, DNSTypeName(question->qtype)); 2941 return 0; 2942 } 2943 2944 #if APPLE_OSX_mDNSResponder 2945 2946 mDNSlocal mDNSBool DomainInSearchList(const domainname *domain, mDNSBool excludeLocal) 2947 { 2948 const SearchListElem *s; 2949 int qcount, scount; 2950 2951 qcount = CountLabels(domain); 2952 for (s=SearchList; s; s=s->next) 2953 { 2954 if (excludeLocal && SameDomainName(&s->domain, &localdomain)) 2955 continue; 2956 scount = CountLabels(&s->domain); 2957 if (qcount >= scount) 2958 { 2959 // Note: When qcount == scount, we do a complete match of the domain 2960 // which is expected by the callers. 2961 const domainname *d = SkipLeadingLabels(domain, (qcount - scount)); 2962 if (SameDomainName(&s->domain, d)) 2963 { 2964 return mDNStrue; 2965 } 2966 } 2967 } 2968 return mDNSfalse; 2969 } 2970 2971 // The caller already checks that this is a dotlocal question. 2972 mDNSlocal mDNSBool ShouldDeliverNegativeResponse(DNSQuestion *question) 2973 { 2974 mDNSu16 qtype; 2975 2976 // If the question matches the search domain exactly or the search domain is a 2977 // subdomain of the question, it is most likely a valid unicast domain and hence 2978 // don't suppress negative responses. 2979 // 2980 // If the user has configured ".local" as a search domain, we don't want 2981 // to deliver a negative response for names ending in ".local" as that would 2982 // prevent bonjour discovery. Passing mDNStrue for the last argument excludes 2983 // ".local" search domains. 2984 if (DomainInSearchList(&question->qname, mDNStrue)) 2985 { 2986 LogOperation("ShouldDeliverNegativeResponse: Question %##s (%s) in SearchList", question->qname.c, DNSTypeName(question->qtype)); 2987 return mDNStrue; 2988 } 2989 2990 // Deliver negative response for A/AAAA if there was a positive response for AAAA/A respectively. 2991 if (question->qtype != kDNSType_A && question->qtype != kDNSType_AAAA) 2992 { 2993 LogOperation("ShouldDeliverNegativeResponse: Question %##s (%s) not answering local question with negative unicast response", 2994 question->qname.c, DNSTypeName(question->qtype)); 2995 return mDNSfalse; 2996 } 2997 qtype = (question->qtype == kDNSType_A ? kDNSType_AAAA : kDNSType_A); 2998 if (!mDNS_CheckForCacheRecord(&mDNSStorage, question, qtype)) 2999 { 3000 LogOperation("ShouldDeliverNegativeResponse:Question %##s (%s) not answering local question with negative unicast response" 3001 " (can't find positive record)", question->qname.c, DNSTypeName(question->qtype)); 3002 return mDNSfalse; 3003 } 3004 LogOperation("ShouldDeliverNegativeResponse:Question %##s (%s) answering local with negative unicast response (found positive record)", 3005 question->qname.c, DNSTypeName(question->qtype)); 3006 return mDNStrue; 3007 } 3008 3009 // Workaround for networks using Microsoft Active Directory using "local" as a private internal 3010 // top-level domain 3011 mDNSlocal mStatus SendAdditionalQuery(DNSQuestion *q, request_state *request, mStatus err) 3012 { 3013 #ifndef UNICAST_DISABLED 3014 extern domainname ActiveDirectoryPrimaryDomain; 3015 DNSQuestion **question2; 3016 #define VALID_MSAD_SRV_TRANSPORT(T) (SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_tcp") || SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_udp")) 3017 #define VALID_MSAD_SRV(Q) ((Q)->qtype == kDNSType_SRV && VALID_MSAD_SRV_TRANSPORT(SecondLabel(&(Q)->qname))) 3018 3019 question2 = mDNSNULL; 3020 if (request->hdr.op == query_request) 3021 question2 = &request->u.queryrecord.q2; 3022 else if (request->hdr.op == addrinfo_request) 3023 { 3024 if (q->qtype == kDNSType_A) 3025 question2 = &request->u.addrinfo.q42; 3026 else if (q->qtype == kDNSType_AAAA) 3027 question2 = &request->u.addrinfo.q62; 3028 } 3029 if (!question2) 3030 { 3031 LogMsg("SendAdditionalQuery: question2 NULL for %##s (%s)", q->qname.c, DNSTypeName(q->qtype)); 3032 return mStatus_BadParamErr; 3033 } 3034 3035 // Sanity check: If we already sent an additonal query, we don't need to send one more. 3036 // 3037 // 1. When the application calls DNSServiceQueryRecord or DNSServiceGetAddrInfo with a .local name, this function 3038 // is called to see whether a unicast query should be sent or not. 3039 // 3040 // 2. As a result of appending search domains, the question may be end up with a .local suffix even though it 3041 // was not a .local name to start with. In that case, queryrecord_result_callback calls this function to 3042 // send the additional query. 3043 // 3044 // Thus, it should not be called more than once. 3045 if (*question2) 3046 { 3047 LogInfo("SendAdditionalQuery: question2 already sent for %##s (%s), no more q2", q->qname.c, DNSTypeName(q->qtype)); 3048 return err; 3049 } 3050 3051 if (!q->ForceMCast && SameDomainLabel(LastLabel(&q->qname), (const mDNSu8 *)&localdomain)) 3052 if (q->qtype == kDNSType_A || q->qtype == kDNSType_AAAA || VALID_MSAD_SRV(q)) 3053 { 3054 DNSQuestion *q2; 3055 int labels = CountLabels(&q->qname); 3056 q2 = mallocL("DNSQuestion", sizeof(DNSQuestion)); 3057 if (!q2) FatalError("ERROR: SendAdditionalQuery malloc"); 3058 *question2 = q2; 3059 *q2 = *q; 3060 q2->InterfaceID = mDNSInterface_Unicast; 3061 q2->ExpectUnique = mDNStrue; 3062 // Always set the QuestionContext to indicate that this question should be stopped 3063 // before freeing. Don't rely on "q". 3064 q2->QuestionContext = request; 3065 // If the query starts as a single label e.g., somehost, and we have search domains with .local, 3066 // queryrecord_result_callback calls this function when .local is appended to "somehost". 3067 // At that time, the name in "q" is pointing at somehost.local and its qnameOrig pointing at 3068 // "somehost". We need to copy that information so that when we retry with a different search 3069 // domain e.g., mycompany.local, we get "somehost.mycompany.local". 3070 if (q->qnameOrig) 3071 { 3072 (*question2)->qnameOrig = mallocL("SendAdditionalQuery", DomainNameLength(q->qnameOrig)); 3073 if (!(*question2)->qnameOrig) { LogMsg("SendAdditionalQuery: ERROR!! malloc failure"); return mStatus_NoMemoryErr; } 3074 (*question2)->qnameOrig->c[0] = 0; 3075 AssignDomainName((*question2)->qnameOrig, q->qnameOrig); 3076 LogInfo("SendAdditionalQuery: qnameOrig %##s", (*question2)->qnameOrig->c); 3077 } 3078 // For names of the form "<one-or-more-labels>.bar.local." we always do a second unicast query in parallel. 3079 // For names of the form "<one-label>.local." it's less clear whether we should do a unicast query. 3080 // If the name being queried is exactly the same as the name in the DHCP "domain" option (e.g. the DHCP 3081 // "domain" is my-small-company.local, and the user types "my-small-company.local" into their web browser) 3082 // then that's a hint that it's worth doing a unicast query. Otherwise, we first check to see if the 3083 // site's DNS server claims there's an SOA record for "local", and if so, that's also a hint that queries 3084 // for names in the "local" domain will be safely answered privately before they hit the root name servers. 3085 // Note that in the "my-small-company.local" example above there will typically be an SOA record for 3086 // "my-small-company.local" but *not* for "local", which is why the "local SOA" check would fail in that case. 3087 // We need to check against both ActiveDirectoryPrimaryDomain and SearchList. If it matches against either 3088 // of those, we don't want do the SOA check for the local 3089 if (labels == 2 && !SameDomainName(&q->qname, &ActiveDirectoryPrimaryDomain) && !DomainInSearchList(&q->qname, mDNSfalse)) 3090 { 3091 AssignDomainName(&q2->qname, &localdomain); 3092 q2->qtype = kDNSType_SOA; 3093 q2->LongLived = mDNSfalse; 3094 q2->ForceMCast = mDNSfalse; 3095 q2->ReturnIntermed = mDNStrue; 3096 // Don't append search domains for the .local SOA query 3097 q2->AppendSearchDomains = 0; 3098 q2->AppendLocalSearchDomains = 0; 3099 q2->RetryWithSearchDomains = mDNSfalse; 3100 q2->SearchListIndex = 0; 3101 q2->TimeoutQuestion = 0; 3102 q2->AnonInfo = mDNSNULL; 3103 q2->pid = request->process_id; 3104 q2->euid = request->uid; 3105 } 3106 LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast", request->sd, q2->qname.c, DNSTypeName(q2->qtype)); 3107 err = mDNS_StartQuery(&mDNSStorage, q2); 3108 if (err) LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q2->qname.c, DNSTypeName(q2->qtype), (int)err); 3109 } 3110 return(err); 3111 #else // !UNICAST_DISABLED 3112 (void) q; 3113 (void) request; 3114 (void) err; 3115 3116 return mStatus_NoError; 3117 #endif // !UNICAST_DISABLED 3118 } 3119 #endif // APPLE_OSX_mDNSResponder 3120 3121 // This function tries to append a search domain if valid and possible. If so, returns true. 3122 mDNSlocal mDNSBool RetryQuestionWithSearchDomains(DNSQuestion *question, request_state *req, QC_result AddRecord) 3123 { 3124 int result; 3125 // RetryWithSearchDomains tells the core to call us back so that we can retry with search domains if there is no 3126 // answer in the cache or /etc/hosts. In the first call back from the core, we clear RetryWithSearchDomains so 3127 // that we don't get called back repeatedly. If we got an answer from the cache or /etc/hosts, we don't touch 3128 // RetryWithSearchDomains which may or may not be set. 3129 // 3130 // If we get e.g., NXDOMAIN and the query is neither suppressed nor exhausted the domain search list and 3131 // is a valid question for appending search domains, retry by appending domains 3132 3133 if ((AddRecord != QC_suppressed) && question->SearchListIndex != -1 && question->AppendSearchDomains) 3134 { 3135 question->RetryWithSearchDomains = 0; 3136 result = AppendNewSearchDomain(question); 3137 // As long as the result is either zero or 1, we retry the question. If we exahaust the search 3138 // domains (result is zero) we try the original query (as it was before appending the search 3139 // domains) as such on the wire as a last resort if we have not tried them before. For queries 3140 // with more than one label, we have already tried them before appending search domains and 3141 // hence don't retry again 3142 if (result != -1) 3143 { 3144 mStatus err; 3145 err = mDNS_StartQuery(&mDNSStorage, question); 3146 if (!err) 3147 { 3148 LogOperation("%3d: RetryQuestionWithSearchDomains(%##s, %s), retrying after appending search domain", req->sd, question->qname.c, DNSTypeName(question->qtype)); 3149 // If the result was zero, it meant that there are no search domains and we just retried the question 3150 // as a single label and we should not retry with search domains anymore. 3151 if (!result) question->SearchListIndex = -1; 3152 return mDNStrue; 3153 } 3154 else 3155 { 3156 LogMsg("%3d: ERROR: RetryQuestionWithSearchDomains %##s %s mDNS_StartQuery: %d, while retrying with search domains", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err); 3157 // We have already stopped the query and could not restart. Reset the appropriate pointers 3158 // so that we don't call stop again when the question terminates 3159 question->QuestionContext = mDNSNULL; 3160 } 3161 } 3162 } 3163 else 3164 { 3165 LogInfo("%3d: RetryQuestionWithSearchDomains: Not appending search domains - SuppressQuery %d, SearchListIndex %d, AppendSearchDomains %d", req->sd, AddRecord, question->SearchListIndex, question->AppendSearchDomains); 3166 } 3167 return mDNSfalse; 3168 } 3169 3170 mDNSlocal void queryrecord_result_reply(mDNS *const m, request_state *req, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord, 3171 DNSServiceErrorType error) 3172 { 3173 char name[MAX_ESCAPED_DOMAIN_NAME]; 3174 size_t len; 3175 DNSServiceFlags flags = 0; 3176 reply_state *rep; 3177 char *data; 3178 3179 ConvertDomainNameToCString(answer->name, name); 3180 3181 LogOperation("%3d: %s(%##s, %s) RESULT %s interface %d: %s", req->sd, 3182 req->hdr.op == query_request ? "DNSServiceQueryRecord" : "DNSServiceGetAddrInfo", 3183 question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", 3184 mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse), RRDisplayString(m, answer)); 3185 3186 len = sizeof(DNSServiceFlags); // calculate reply data length 3187 len += sizeof(mDNSu32); // interface index 3188 len += sizeof(DNSServiceErrorType); 3189 len += strlen(name) + 1; 3190 len += 3 * sizeof(mDNSu16); // type, class, rdlen 3191 len += answer->rdlength; 3192 len += sizeof(mDNSu32); // TTL 3193 3194 rep = create_reply(req->hdr.op == query_request ? query_reply_op : addrinfo_reply_op, len, req); 3195 3196 if (AddRecord) 3197 flags |= kDNSServiceFlagsAdd; 3198 if (question->ValidationStatus != 0) 3199 { 3200 error = kDNSServiceErr_NoError; 3201 if (question->ValidationRequired && question->ValidationState == DNSSECValDone) 3202 { 3203 switch (question->ValidationStatus) //Set the dnssec flags to be passed on to the Apps here 3204 { 3205 case DNSSEC_Secure: 3206 flags |= kDNSServiceFlagsSecure; 3207 break; 3208 case DNSSEC_Insecure: 3209 flags |= kDNSServiceFlagsInsecure; 3210 break; 3211 case DNSSEC_Indeterminate: 3212 flags |= kDNSServiceFlagsIndeterminate; 3213 break; 3214 case DNSSEC_Bogus: 3215 flags |= kDNSServiceFlagsBogus; 3216 break; 3217 default: 3218 LogMsg("queryrecord_result_reply unknown status %d for %##s", question->ValidationStatus, question->qname.c); 3219 } 3220 } 3221 } 3222 3223 rep->rhdr->flags = dnssd_htonl(flags); 3224 // Call mDNSPlatformInterfaceIndexfromInterfaceID, but suppressNetworkChange (last argument). Otherwise, if the 3225 // InterfaceID is not valid, then it simulates a "NetworkChanged" which in turn makes questions 3226 // to be stopped and started including *this* one. Normally the InterfaceID is valid. But when we 3227 // are using the /etc/hosts entries to answer a question, the InterfaceID may not be known to the 3228 // mDNS core . Eventually, we should remove the calls to "NetworkChanged" in 3229 // mDNSPlatformInterfaceIndexfromInterfaceID when it can't find InterfaceID as ResourceRecords 3230 // should not have existed to answer this question if the corresponding interface is not valid. 3231 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNStrue)); 3232 rep->rhdr->error = dnssd_htonl(error); 3233 3234 data = (char *)&rep->rhdr[1]; 3235 3236 put_string(name, &data); 3237 put_uint16(answer->rrtype, &data); 3238 put_uint16(answer->rrclass, &data); 3239 put_uint16(answer->rdlength, &data); 3240 // We need to use putRData here instead of the crude put_rdata function, because the crude put_rdata 3241 // function just does a blind memory copy without regard to structures that may have holes in them. 3242 if (answer->rdlength) 3243 if (!putRData(mDNSNULL, (mDNSu8 *)data, (mDNSu8 *)rep->rhdr + len, answer)) 3244 LogMsg("queryrecord_result_reply putRData failed %d", (mDNSu8 *)rep->rhdr + len - (mDNSu8 *)data); 3245 data += answer->rdlength; 3246 put_uint32(AddRecord ? answer->rroriginalttl : 0, &data); 3247 3248 append_reply(req, rep); 3249 // Stop the question, if we just timed out 3250 if (error == kDNSServiceErr_Timeout) 3251 { 3252 mDNS_StopQuery(m, question); 3253 // Reset the pointers so that we don't call stop on termination 3254 question->QuestionContext = mDNSNULL; 3255 } 3256 else if ((AddRecord == QC_add) && req->hdr.op == addrinfo_request) 3257 { 3258 // Note: We count all answers including LocalOnly e.g., /etc/hosts. If we 3259 // exclude that, v4ans/v6ans will be zero and we would wrongly think that 3260 // we did not answer questions and setup the status to deliver triggers. 3261 if (question->qtype == kDNSType_A) 3262 req->u.addrinfo.v4ans = 1; 3263 if (question->qtype == kDNSType_AAAA) 3264 req->u.addrinfo.v6ans = 1; 3265 } 3266 else if ((AddRecord == QC_add) && req->hdr.op == query_request) 3267 { 3268 if (question->qtype == kDNSType_A || question->qtype == kDNSType_AAAA) 3269 req->u.queryrecord.ans = 1; 3270 } 3271 3272 #if APPLE_OSX_mDNSResponder 3273 #if !NO_WCF 3274 CHECK_WCF_FUNCTION(WCFIsServerRunning) 3275 { 3276 struct xucred x; 3277 socklen_t xucredlen = sizeof(x); 3278 3279 if (WCFIsServerRunning((WCFConnection *)m->WCF) && answer->rdlength != 0) 3280 { 3281 if (getsockopt(req->sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 && 3282 (x.cr_version == XUCRED_VERSION)) 3283 { 3284 struct sockaddr_storage addr; 3285 addr.ss_len = 0; 3286 if (answer->rrtype == kDNSType_A || answer->rrtype == kDNSType_AAAA) 3287 { 3288 if (answer->rrtype == kDNSType_A) 3289 { 3290 struct sockaddr_in *const sin = (struct sockaddr_in *)&addr; 3291 sin->sin_port = 0; 3292 // Instead of this stupid call to putRData it would be much simpler to just assign the value in the sensible way, like this: 3293 // sin->sin_addr.s_addr = answer->rdata->u.ipv4.NotAnInteger; 3294 if (!putRData(mDNSNULL, (mDNSu8 *)&sin->sin_addr, (mDNSu8 *)(&sin->sin_addr + sizeof(mDNSv4Addr)), answer)) 3295 LogMsg("queryrecord_result_reply: WCF AF_INET putRData failed"); 3296 else 3297 { 3298 addr.ss_len = sizeof (struct sockaddr_in); 3299 addr.ss_family = AF_INET; 3300 } 3301 } 3302 else if (answer->rrtype == kDNSType_AAAA) 3303 { 3304 struct sockaddr_in6 *const sin6 = (struct sockaddr_in6 *)&addr; 3305 sin6->sin6_port = 0; 3306 // Instead of this stupid call to putRData it would be much simpler to just assign the value in the sensible way, like this: 3307 // sin6->sin6_addr.__u6_addr.__u6_addr32[0] = answer->rdata->u.ipv6.l[0]; 3308 // sin6->sin6_addr.__u6_addr.__u6_addr32[1] = answer->rdata->u.ipv6.l[1]; 3309 // sin6->sin6_addr.__u6_addr.__u6_addr32[2] = answer->rdata->u.ipv6.l[2]; 3310 // sin6->sin6_addr.__u6_addr.__u6_addr32[3] = answer->rdata->u.ipv6.l[3]; 3311 if (!putRData(mDNSNULL, (mDNSu8 *)&sin6->sin6_addr, (mDNSu8 *)(&sin6->sin6_addr + sizeof(mDNSv6Addr)), answer)) 3312 LogMsg("queryrecord_result_reply: WCF AF_INET6 putRData failed"); 3313 else 3314 { 3315 addr.ss_len = sizeof (struct sockaddr_in6); 3316 addr.ss_family = AF_INET6; 3317 } 3318 } 3319 if (addr.ss_len) 3320 { 3321 debugf("queryrecord_result_reply: Name %s, uid %u, addr length %d", name, x.cr_uid, addr.ss_len); 3322 CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr) 3323 { 3324 WCFNameResolvesToAddr(m->WCF, name, (struct sockaddr *)&addr, x.cr_uid); 3325 } 3326 } 3327 } 3328 else if (answer->rrtype == kDNSType_CNAME) 3329 { 3330 domainname cname; 3331 char cname_cstr[MAX_ESCAPED_DOMAIN_NAME]; 3332 if (!putRData(mDNSNULL, cname.c, (mDNSu8 *)(cname.c + MAX_DOMAIN_NAME), answer)) 3333 LogMsg("queryrecord_result_reply: WCF CNAME putRData failed"); 3334 else 3335 { 3336 ConvertDomainNameToCString(&cname, cname_cstr); 3337 CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr) 3338 { 3339 WCFNameResolvesToName(m->WCF, name, cname_cstr, x.cr_uid); 3340 } 3341 } 3342 } 3343 } 3344 else my_perror("queryrecord_result_reply: ERROR: getsockopt LOCAL_PEERCRED"); 3345 } 3346 } 3347 #endif 3348 #endif 3349 } 3350 3351 mDNSlocal void queryrecord_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 3352 { 3353 request_state *req = question->QuestionContext; 3354 DNSServiceErrorType error = kDNSServiceErr_NoError; 3355 DNSQuestion *q = mDNSNULL; 3356 3357 #if APPLE_OSX_mDNSResponder 3358 { 3359 // Sanity check: QuestionContext is set to NULL after we stop the question and hence we should not 3360 // get any callbacks from the core after this. 3361 if (!req) 3362 { 3363 LogMsg("queryrecord_result_callback: ERROR!! QuestionContext NULL for %##s (%s)", question->qname.c, DNSTypeName(question->qtype)); 3364 return; 3365 } 3366 if (req->hdr.op == query_request && question == req->u.queryrecord.q2) 3367 q = &req->u.queryrecord.q; 3368 else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q42) 3369 q = &req->u.addrinfo.q4; 3370 else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q62) 3371 q = &req->u.addrinfo.q6; 3372 3373 if (q && question->qtype != q->qtype && !SameDomainName(&question->qname, &q->qname)) 3374 { 3375 mStatus err; 3376 domainname *orig = question->qnameOrig; 3377 3378 LogInfo("queryrecord_result_callback: Stopping q2 local %##s", question->qname.c); 3379 mDNS_StopQuery(m, question); 3380 question->QuestionContext = mDNSNULL; 3381 3382 // We got a negative response for the SOA record indicating that .local does not exist. 3383 // But we might have other search domains (that does not end in .local) that can be 3384 // appended to this question. In that case, we want to retry the question. Otherwise, 3385 // we don't want to try this question as unicast. 3386 if (answer->RecordType == kDNSRecordTypePacketNegative && !q->AppendSearchDomains) 3387 { 3388 LogInfo("queryrecord_result_callback: question %##s AppendSearchDomains zero", q->qname.c); 3389 return; 3390 } 3391 3392 // If we got a non-negative answer for our "local SOA" test query, start an additional parallel unicast query 3393 // 3394 // Note: When we copy the original question, we copy everything including the AppendSearchDomains, 3395 // RetryWithSearchDomains except for qnameOrig which can be non-NULL if the original question is 3396 // e.g., somehost and then we appended e.g., ".local" and retried that question. See comment in 3397 // SendAdditionalQuery as to how qnameOrig gets initialized. 3398 *question = *q; 3399 question->InterfaceID = mDNSInterface_Unicast; 3400 question->ExpectUnique = mDNStrue; 3401 question->qnameOrig = orig; 3402 3403 LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast, context %p", req->sd, question->qname.c, DNSTypeName(question->qtype), question->QuestionContext); 3404 3405 // If the original question timed out, its QuestionContext would already be set to NULL and that's what we copied above. 3406 // Hence, we need to set it explicitly here. 3407 question->QuestionContext = req; 3408 err = mDNS_StartQuery(m, question); 3409 if (err) LogMsg("%3d: ERROR: queryrecord_result_callback %##s %s mDNS_StartQuery: %d", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err); 3410 3411 // If we got a positive response to local SOA, then try the .local question as unicast 3412 if (answer->RecordType != kDNSRecordTypePacketNegative) return; 3413 3414 // Fall through and get the next search domain. The question is pointing at .local 3415 // and we don't want to try that. Try the next search domain. Don't try with local 3416 // search domains for the unicast question anymore. 3417 // 3418 // Note: we started the question above which will be stopped immediately (never sent on the wire) 3419 // before we pick the next search domain below. RetryQuestionWithSearchDomains assumes that the 3420 // question has already started. 3421 question->AppendLocalSearchDomains = 0; 3422 } 3423 3424 if (q && AddRecord && AddRecord != QC_dnssec && (question->InterfaceID == mDNSInterface_Unicast) && !answer->rdlength) 3425 { 3426 // If we get a negative response to the unicast query that we sent above, retry after appending search domains 3427 // Note: We could have appended search domains below (where do it for regular unicast questions) instead of doing it here. 3428 // As we ignore negative unicast answers below, we would never reach the code where the search domains are appended. 3429 // To keep things simple, we handle unicast ".local" separately here. 3430 LogInfo("queryrecord_result_callback: Retrying .local question %##s (%s) as unicast after appending search domains", question->qname.c, DNSTypeName(question->qtype)); 3431 if (RetryQuestionWithSearchDomains(question, req, AddRecord)) 3432 return; 3433 if (question->AppendSearchDomains && !question->AppendLocalSearchDomains && IsLocalDomain(&question->qname)) 3434 { 3435 // If "local" is the last search domain, we need to stop the question so that we don't send the "local" 3436 // question on the wire as we got a negative response for the local SOA. But, we can't stop the question 3437 // yet as we may have to timeout the question (done by the "core") for which we need to leave the question 3438 // in the list. We leave it disabled so that it does not hit the wire. 3439 LogInfo("queryrecord_result_callback: Disabling .local question %##s (%s)", question->qname.c, DNSTypeName(question->qtype)); 3440 question->ThisQInterval = 0; 3441 } 3442 } 3443 // 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 3444 // domains to append for "q2". In all cases, fall through and deliver the response 3445 } 3446 #endif // APPLE_OSX_mDNSResponder 3447 3448 // If a query is being suppressed for some reason, we don't have to do any other 3449 // processing. 3450 // 3451 // Note: We don't check for "SuppressQuery" and instead use QC_suppressed because 3452 // the "core" needs to temporarily turn off SuppressQuery to answer this query. 3453 if (AddRecord == QC_suppressed) 3454 { 3455 LogInfo("queryrecord_result_callback: Suppressed question %##s (%s)", question->qname.c, DNSTypeName(question->qtype)); 3456 queryrecord_result_reply(m, req, question, answer, AddRecord, kDNSServiceErr_NoSuchRecord); 3457 return; 3458 } 3459 3460 if (answer->RecordType == kDNSRecordTypePacketNegative) 3461 { 3462 // If this question needs to be timed out and we have reached the stop time, mark 3463 // the error as timeout. It is possible that we might get a negative response from an 3464 // external DNS server at the same time when this question reaches its stop time. We 3465 // can't tell the difference as there is no indication in the callback. This should 3466 // be okay as we will be timing out this query anyway. 3467 mDNS_Lock(m); 3468 if (question->TimeoutQuestion) 3469 { 3470 if ((m->timenow - question->StopTime) >= 0) 3471 { 3472 LogInfo("queryrecord_result_callback:Question %##s (%s) timing out, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 3473 error = kDNSServiceErr_Timeout; 3474 } 3475 } 3476 mDNS_Unlock(m); 3477 // When we're doing parallel unicast and multicast queries for dot-local names (for supporting Microsoft 3478 // Active Directory sites) we need to ignore negative unicast answers. Otherwise we'll generate negative 3479 // answers for just about every single multicast name we ever look up, since the Microsoft Active Directory 3480 // server is going to assert that pretty much every single multicast name doesn't exist. 3481 // 3482 // If we are timing out this query, we need to deliver the negative answer to the application 3483 if (error != kDNSServiceErr_Timeout) 3484 { 3485 if (!answer->InterfaceID && IsLocalDomain(answer->name)) 3486 { 3487 // Sanity check: "q" will be set only if "question" is the .local unicast query. 3488 if (!q) 3489 { 3490 LogMsg("queryrecord_result_callback: ERROR!! answering multicast question %s with unicast cache record", 3491 RRDisplayString(m, answer)); 3492 return; 3493 } 3494 #if APPLE_OSX_mDNSResponder 3495 if (!ShouldDeliverNegativeResponse(question)) 3496 { 3497 return; 3498 } 3499 #endif // APPLE_OSX_mDNSResponder 3500 LogInfo("queryrecord_result_callback:Question %##s (%s) answering local with negative unicast response", question->qname.c, 3501 DNSTypeName(question->qtype)); 3502 } 3503 error = kDNSServiceErr_NoSuchRecord; 3504 } 3505 } 3506 // If we get a negative answer, try appending search domains. Don't append search domains 3507 // - if we are timing out this question 3508 // - if the negative response was received as a result of a multicast query 3509 // - if this is an additional query (q2), we already appended search domains above (indicated by "!q" below) 3510 // - if this response is forced e.g., dnssec validation result 3511 if (error != kDNSServiceErr_Timeout) 3512 { 3513 if (!q && !answer->InterfaceID && !answer->rdlength && AddRecord && AddRecord != QC_dnssec) 3514 { 3515 // If the original question did not end in .local, we did not send an SOA query 3516 // to figure out whether we should send an additional unicast query or not. If we just 3517 // appended .local, we need to see if we need to send an additional query. This should 3518 // normally happen just once because after we append .local, we ignore all negative 3519 // responses for .local above. 3520 LogInfo("queryrecord_result_callback: Retrying question %##s (%s) after appending search domains", question->qname.c, DNSTypeName(question->qtype)); 3521 if (RetryQuestionWithSearchDomains(question, req, AddRecord)) 3522 { 3523 // Note: We need to call SendAdditionalQuery every time after appending a search domain as .local could 3524 // be anywhere in the search domain list. 3525 #if APPLE_OSX_mDNSResponder 3526 mStatus err = mStatus_NoError; 3527 err = SendAdditionalQuery(question, req, err); 3528 if (err) LogMsg("queryrecord_result_callback: Sending .local SOA query failed, after appending domains"); 3529 #endif // APPLE_OSX_mDNSResponder 3530 return; 3531 } 3532 } 3533 } 3534 queryrecord_result_reply(m, req, question, answer, AddRecord, error); 3535 } 3536 3537 mDNSlocal void queryrecord_termination_callback(request_state *request) 3538 { 3539 LogOperation("%3d: DNSServiceQueryRecord(%X, %d, %##s, %s) STOP PID[%d](%s)", 3540 request->sd, request->flags, request->interfaceIndex, request->u.queryrecord.q.qname.c, DNSTypeName(request->u.queryrecord.q.qtype), request->process_id, request->pid_name); 3541 if (request->u.queryrecord.q.QuestionContext) 3542 { 3543 mDNS_StopQuery(&mDNSStorage, &request->u.queryrecord.q); // no need to error check 3544 LogMcastQ(&request->u.queryrecord.q, request, q_stop); 3545 request->u.queryrecord.q.QuestionContext = mDNSNULL; 3546 } 3547 else 3548 { 3549 DNSQuestion *question = &request->u.queryrecord.q; 3550 LogInfo("queryrecord_termination_callback: question %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 3551 } 3552 3553 if (request->u.queryrecord.q.qnameOrig) 3554 { 3555 freeL("QueryTermination", request->u.queryrecord.q.qnameOrig); 3556 request->u.queryrecord.q.qnameOrig = mDNSNULL; 3557 } 3558 3559 if (callExternalHelpers(request->u.queryrecord.q.InterfaceID, &request->u.queryrecord.q.qname, request->u.queryrecord.q.flags)) 3560 { 3561 LogInfo("queryrecord_termination_callback: calling external_stop_browsing_for_service()"); 3562 external_stop_browsing_for_service(request->u.queryrecord.q.InterfaceID, &request->u.queryrecord.q.qname, request->u.queryrecord.q.qtype, request->u.queryrecord.q.flags); 3563 } 3564 if (request->u.queryrecord.q2) 3565 { 3566 if (request->u.queryrecord.q2->QuestionContext) 3567 { 3568 LogInfo("queryrecord_termination_callback: Stopping q2 %##s", request->u.queryrecord.q2->qname.c); 3569 mDNS_StopQuery(&mDNSStorage, request->u.queryrecord.q2); 3570 LogMcastQ(request->u.queryrecord.q2, request, q_stop); 3571 } 3572 else 3573 { 3574 DNSQuestion *question = request->u.queryrecord.q2; 3575 LogInfo("queryrecord_termination_callback: q2 %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 3576 } 3577 if (request->u.queryrecord.q2->qnameOrig) 3578 { 3579 LogInfo("queryrecord_termination_callback: freeing q2 qnameOrig %##s", request->u.queryrecord.q2->qnameOrig->c); 3580 freeL("QueryTermination q2", request->u.queryrecord.q2->qnameOrig); 3581 request->u.queryrecord.q2->qnameOrig = mDNSNULL; 3582 } 3583 freeL("queryrecord Q2", request->u.queryrecord.q2); 3584 request->u.queryrecord.q2 = mDNSNULL; 3585 } 3586 #if APPLE_OSX_mDNSResponder 3587 { 3588 if (request->u.queryrecord.ans) 3589 { 3590 DNSQuestion *v4q, *v6q; 3591 // If we are receiving poisitive answers, provide the hint to the 3592 // upper layer. 3593 v4q = v6q = mDNSNULL; 3594 if (request->u.queryrecord.q.qtype == kDNSType_A) 3595 v4q = &request->u.queryrecord.q; 3596 else if (request->u.queryrecord.q.qtype == kDNSType_AAAA) 3597 v6q = &request->u.queryrecord.q; 3598 mDNSPlatformTriggerDNSRetry(v4q, v6q); 3599 } 3600 } 3601 #endif // APPLE_OSX_mDNSResponder 3602 } 3603 3604 mDNSlocal mStatus handle_queryrecord_request(request_state *request) 3605 { 3606 DNSQuestion *const q = &request->u.queryrecord.q; 3607 char name[256]; 3608 size_t nameLen; 3609 mDNSu16 rrtype, rrclass; 3610 mStatus err; 3611 3612 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 3613 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 3614 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 3615 3616 // The request is scoped to a specific interface index, but the 3617 // interface is not currently in our list. 3618 if (interfaceIndex && !InterfaceID) 3619 { 3620 if (interfaceIndex > 1) 3621 LogMsg("handle_queryrecord_request: interfaceIndex %d is currently inactive requested by client[%d][%s]", 3622 interfaceIndex, request->process_id, request->pid_name); 3623 // If it's one of the specially defined inteface index values, just return an error. 3624 // Also, caller should return an error immediately if lo0 (index 1) is not configured 3625 // into the current active interfaces. See background in Radar 21967160. 3626 if (PreDefinedInterfaceIndex(interfaceIndex) || interfaceIndex == 1) 3627 { 3628 LogInfo("handle_queryrecord_request: bad interfaceIndex %d", interfaceIndex); 3629 return(mStatus_BadParamErr); 3630 } 3631 3632 // Otherwise, use the specified interface index value and the request will 3633 // be applied to that interface when it comes up. 3634 InterfaceID = (mDNSInterfaceID)(uintptr_t)interfaceIndex; 3635 LogInfo("handle_queryrecord_request: query pending for interface index %d", interfaceIndex); 3636 } 3637 3638 if (get_string(&request->msgptr, request->msgend, name, 256) < 0) return(mStatus_BadParamErr); 3639 rrtype = get_uint16(&request->msgptr, request->msgend); 3640 rrclass = get_uint16(&request->msgptr, request->msgend); 3641 3642 if (!request->msgptr) 3643 { LogMsg("%3d: DNSServiceQueryRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 3644 3645 request->flags = flags; 3646 request->interfaceIndex = interfaceIndex; 3647 mDNSPlatformMemZero(&request->u.queryrecord, sizeof(request->u.queryrecord)); 3648 3649 q->InterfaceID = InterfaceID; 3650 q->flags = flags; 3651 q->Target = zeroAddr; 3652 if (!MakeDomainNameFromDNSNameString(&q->qname, name)) return(mStatus_BadParamErr); 3653 #if 0 3654 if (!AuthorizedDomain(request, &q->qname, AutoBrowseDomains)) return (mStatus_NoError); 3655 #endif 3656 q->qtype = rrtype; 3657 q->qclass = rrclass; 3658 q->LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 3659 q->ExpectUnique = mDNSfalse; 3660 q->ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 3661 q->ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 3662 q->SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0; 3663 q->TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0; 3664 q->WakeOnResolve = 0; 3665 q->UseBackgroundTrafficClass = (flags & kDNSServiceFlagsBackgroundTrafficClass) != 0; 3666 if ((flags & kDNSServiceFlagsValidate) != 0) 3667 q->ValidationRequired = DNSSEC_VALIDATION_SECURE; 3668 else if ((flags & kDNSServiceFlagsValidateOptional) != 0) 3669 q->ValidationRequired = DNSSEC_VALIDATION_SECURE_OPTIONAL; 3670 q->ValidatingResponse = 0; 3671 q->ProxyQuestion = 0; 3672 q->AnonInfo = mDNSNULL; 3673 q->QuestionCallback = queryrecord_result_callback; 3674 q->QuestionContext = request; 3675 q->SearchListIndex = 0; 3676 q->StopTime = 0; 3677 3678 q->DNSSECAuthInfo = mDNSNULL; 3679 q->DAIFreeCallback = mDNSNULL; 3680 3681 //Turn off dnssec validation for local domains and Question Types: RRSIG/ANY(ANY Type is not supported yet) 3682 if ((IsLocalDomain(&q->qname)) || (q->qtype == kDNSServiceType_RRSIG) || (q->qtype == kDNSServiceType_ANY)) 3683 q->ValidationRequired = 0; 3684 3685 // Don't append search domains for fully qualified domain names including queries 3686 // such as e.g., "abc." that has only one label. We convert all names to FQDNs as internally 3687 // we only deal with FQDNs. Hence, we cannot look at qname to figure out whether we should 3688 // append search domains or not. So, we record that information in AppendSearchDomains. 3689 // 3690 // We append search domains only for queries that are a single label. If overriden using command line 3691 // argument "AlwaysAppendSearchDomains", then we do it for any query which is not fully qualified. 3692 // For DNSSEC questions, append search domains only if kDNSServiceFlagsValidateOptional is set. 3693 3694 nameLen = strlen(name); 3695 if ((!(q->ValidationRequired == DNSSEC_VALIDATION_SECURE)) && (!(q->ValidationRequired == DNSSEC_VALIDATION_INSECURE)) 3696 && (rrtype == kDNSType_A || rrtype == kDNSType_AAAA) && ((nameLen == 0) || (name[nameLen - 1] != '.')) && 3697 (AlwaysAppendSearchDomains || CountLabels(&q->qname) == 1)) 3698 { 3699 q->AppendSearchDomains = 1; 3700 q->AppendLocalSearchDomains = 1; 3701 } 3702 else 3703 { 3704 q->AppendSearchDomains = 0; 3705 q->AppendLocalSearchDomains = 0; 3706 } 3707 3708 // For single label queries that are not fully qualified, look at /etc/hosts, cache and try 3709 // search domains before trying them on the wire as a single label query. RetryWithSearchDomains 3710 // tell the core to call back into the UDS layer if there is no valid response in /etc/hosts or 3711 // the cache 3712 q->RetryWithSearchDomains = ApplySearchDomainsFirst(q) ? 1 : 0; 3713 q->qnameOrig = mDNSNULL; 3714 SetQuestionPolicy(q, request); 3715 3716 #if APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 3717 // Determine if this request should be promoted to use BLE triggered discovery. 3718 if (shouldUseBLE(InterfaceID, rrtype, (domainname *)SkipLeadingLabels(&q->qname, 1), &q->qname)) 3719 { 3720 q->flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 3721 request->flags |= (kDNSServiceFlagsAutoTrigger | kDNSServiceFlagsIncludeAWDL); 3722 LogInfo("handle_queryrecord_request: request promoted to use kDNSServiceFlagsAutoTrigger"); 3723 } 3724 #endif // APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 3725 3726 LogOperation("%3d: DNSServiceQueryRecord(%X, %d, %##s, %s) START PID[%d](%s)", 3727 request->sd, request->flags, interfaceIndex, q->qname.c, DNSTypeName(q->qtype), request->process_id, request->pid_name); 3728 err = mDNS_StartQuery(&mDNSStorage, q); 3729 3730 if (err) 3731 { 3732 LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q->qname.c, DNSTypeName(q->qtype), (int)err); 3733 } 3734 else 3735 { 3736 request->terminate = queryrecord_termination_callback; 3737 LogMcastQ(q, request, q_start); 3738 if (callExternalHelpers(q->InterfaceID, &q->qname, q->flags)) 3739 { 3740 LogInfo("handle_queryrecord_request: calling external_start_browsing_for_service()"); 3741 external_start_browsing_for_service(q->InterfaceID, &q->qname, q->qtype, q->flags); 3742 } 3743 } 3744 3745 #if APPLE_OSX_mDNSResponder 3746 err = SendAdditionalQuery(q, request, err); 3747 #endif // APPLE_OSX_mDNSResponder 3748 3749 return(err); 3750 } 3751 3752 // *************************************************************************** 3753 #if COMPILER_LIKES_PRAGMA_MARK 3754 #pragma mark - 3755 #pragma mark - DNSServiceEnumerateDomains 3756 #endif 3757 3758 mDNSlocal reply_state *format_enumeration_reply(request_state *request, 3759 const char *domain, DNSServiceFlags flags, mDNSu32 ifi, DNSServiceErrorType err) 3760 { 3761 size_t len; 3762 reply_state *reply; 3763 char *data; 3764 3765 len = sizeof(DNSServiceFlags); 3766 len += sizeof(mDNSu32); 3767 len += sizeof(DNSServiceErrorType); 3768 len += strlen(domain) + 1; 3769 3770 reply = create_reply(enumeration_reply_op, len, request); 3771 reply->rhdr->flags = dnssd_htonl(flags); 3772 reply->rhdr->ifi = dnssd_htonl(ifi); 3773 reply->rhdr->error = dnssd_htonl(err); 3774 data = (char *)&reply->rhdr[1]; 3775 put_string(domain, &data); 3776 return reply; 3777 } 3778 3779 mDNSlocal void enum_termination_callback(request_state *request) 3780 { 3781 // Stop the domain enumeration queries to discover the WAB Browse/Registration domains 3782 if (request->u.enumeration.flags & kDNSServiceFlagsRegistrationDomains) 3783 { 3784 LogInfo("%3d: DNSServiceEnumeration Cancel WAB Registration PID[%d](%s)", request->sd, request->process_id, request->pid_name); 3785 uDNS_StopWABQueries(&mDNSStorage, UDNS_WAB_REG_QUERY); 3786 } 3787 else 3788 { 3789 LogInfo("%3d: DNSServiceEnumeration Cancel WAB Browse PID[%d](%s)", request->sd, request->process_id, request->pid_name); 3790 uDNS_StopWABQueries(&mDNSStorage, UDNS_WAB_BROWSE_QUERY | UDNS_WAB_LBROWSE_QUERY); 3791 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_autoall); 3792 } 3793 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all); 3794 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_default); 3795 } 3796 3797 mDNSlocal void enum_result_callback(mDNS *const m, 3798 DNSQuestion *const question, const ResourceRecord *const answer, QC_result AddRecord) 3799 { 3800 char domain[MAX_ESCAPED_DOMAIN_NAME]; 3801 request_state *request = question->QuestionContext; 3802 DNSServiceFlags flags = 0; 3803 reply_state *reply; 3804 (void)m; // Unused 3805 3806 if (answer->rrtype != kDNSType_PTR) return; 3807 3808 #if 0 3809 if (!AuthorizedDomain(request, &answer->rdata->u.name, request->u.enumeration.flags ? AutoRegistrationDomains : AutoBrowseDomains)) return; 3810 #endif 3811 3812 // We only return add/remove events for the browse and registration lists 3813 // For the default browse and registration answers, we only give an "ADD" event 3814 if (question == &request->u.enumeration.q_default && !AddRecord) return; 3815 3816 if (AddRecord) 3817 { 3818 flags |= kDNSServiceFlagsAdd; 3819 if (question == &request->u.enumeration.q_default) flags |= kDNSServiceFlagsDefault; 3820 } 3821 3822 ConvertDomainNameToCString(&answer->rdata->u.name, domain); 3823 // Note that we do NOT propagate specific interface indexes to the client - for example, a domain we learn from 3824 // a machine's system preferences may be discovered on the LocalOnly interface, but should be browsed on the 3825 // network, so we just pass kDNSServiceInterfaceIndexAny 3826 reply = format_enumeration_reply(request, domain, flags, kDNSServiceInterfaceIndexAny, kDNSServiceErr_NoError); 3827 if (!reply) { LogMsg("ERROR: enum_result_callback, format_enumeration_reply"); return; } 3828 3829 LogOperation("%3d: DNSServiceEnumerateDomains(%#2s) RESULT %s: %s", request->sd, question->qname.c, AddRecord ? "ADD" : "RMV", domain); 3830 3831 append_reply(request, reply); 3832 } 3833 3834 mDNSlocal mStatus handle_enum_request(request_state *request) 3835 { 3836 mStatus err; 3837 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 3838 DNSServiceFlags reg = flags & kDNSServiceFlagsRegistrationDomains; 3839 mDNS_DomainType t_all = reg ? mDNS_DomainTypeRegistration : mDNS_DomainTypeBrowse; 3840 mDNS_DomainType t_default = reg ? mDNS_DomainTypeRegistrationDefault : mDNS_DomainTypeBrowseDefault; 3841 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 3842 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 3843 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 3844 3845 if (!request->msgptr) 3846 { LogMsg("%3d: DNSServiceEnumerateDomains(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 3847 3848 request->flags = flags; 3849 request->interfaceIndex = interfaceIndex; 3850 3851 // mark which kind of enumeration we're doing so that we know what domain enumeration queries to stop 3852 request->u.enumeration.flags = reg; 3853 3854 // enumeration requires multiple questions, so we must link all the context pointers so that 3855 // necessary context can be reached from the callbacks 3856 request->u.enumeration.q_all.QuestionContext = request; 3857 request->u.enumeration.q_default.QuestionContext = request; 3858 if (!reg) request->u.enumeration.q_autoall.QuestionContext = request; 3859 3860 // if the caller hasn't specified an explicit interface, we use local-only to get the system-wide list. 3861 if (!InterfaceID) InterfaceID = mDNSInterface_LocalOnly; 3862 3863 // make the calls 3864 LogOperation("%3d: DNSServiceEnumerateDomains(%X=%s)", request->sd, flags, 3865 (flags & kDNSServiceFlagsBrowseDomains ) ? "kDNSServiceFlagsBrowseDomains" : 3866 (flags & kDNSServiceFlagsRegistrationDomains) ? "kDNSServiceFlagsRegistrationDomains" : "<<Unknown>>"); 3867 err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_all, t_all, NULL, InterfaceID, enum_result_callback, request); 3868 if (!err) 3869 { 3870 err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_default, t_default, NULL, InterfaceID, enum_result_callback, request); 3871 if (err) mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all); 3872 else if (!reg) 3873 { 3874 err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_autoall, mDNS_DomainTypeBrowseAutomatic, NULL, InterfaceID, enum_result_callback, request); 3875 if (err) 3876 { 3877 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all); 3878 mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_default); 3879 } 3880 } 3881 if (!err) request->terminate = enum_termination_callback; 3882 } 3883 if (!err) 3884 { 3885 // Start the domain enumeration queries to discover the WAB Browse/Registration domains 3886 if (reg) 3887 { 3888 LogInfo("%3d: DNSServiceEnumerateDomains Start WAB Registration PID[%d](%s)", request->sd, request->process_id, request->pid_name); 3889 uDNS_StartWABQueries(&mDNSStorage, UDNS_WAB_REG_QUERY); 3890 } 3891 else 3892 { 3893 LogInfo("%3d: DNSServiceEnumerateDomains Start WAB Browse PID[%d](%s)", request->sd, request->process_id, request->pid_name); 3894 uDNS_StartWABQueries(&mDNSStorage, UDNS_WAB_BROWSE_QUERY | UDNS_WAB_LBROWSE_QUERY); 3895 } 3896 } 3897 3898 return(err); 3899 } 3900 3901 // *************************************************************************** 3902 #if COMPILER_LIKES_PRAGMA_MARK 3903 #pragma mark - 3904 #pragma mark - DNSServiceReconfirmRecord & Misc 3905 #endif 3906 3907 mDNSlocal mStatus handle_reconfirm_request(request_state *request) 3908 { 3909 mStatus status = mStatus_BadParamErr; 3910 AuthRecord *rr = read_rr_from_ipc_msg(request, 0, 0); 3911 if (rr) 3912 { 3913 status = mDNS_ReconfirmByValue(&mDNSStorage, &rr->resrec); 3914 LogOperation( 3915 (status == mStatus_NoError) ? 3916 "%3d: DNSServiceReconfirmRecord(%s) interface %d initiated PID[%d](%s)" : 3917 "%3d: DNSServiceReconfirmRecord(%s) interface %d failed PID[%d](%s) status %d", 3918 request->sd, RRDisplayString(&mDNSStorage, &rr->resrec), 3919 mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, rr->resrec.InterfaceID, mDNSfalse), 3920 request->process_id, request->pid_name, status); 3921 freeL("AuthRecord/handle_reconfirm_request", rr); 3922 } 3923 return(status); 3924 } 3925 3926 #if APPLE_OSX_mDNSResponder 3927 3928 mDNSlocal mStatus handle_release_request(request_state *request) 3929 { 3930 mStatus err = 0; 3931 char name[256], regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME]; 3932 domainname instance; 3933 3934 // extract the data from the message 3935 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 3936 3937 if (get_string(&request->msgptr, request->msgend, name, 256) < 0 || 3938 get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 || 3939 get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) 3940 { 3941 LogMsg("ERROR: handle_release_request - Couldn't read name/regtype/domain"); 3942 return(mStatus_BadParamErr); 3943 } 3944 3945 if (!request->msgptr) 3946 { 3947 LogMsg("%3d: PeerConnectionRelease(unreadable parameters)", request->sd); 3948 return(mStatus_BadParamErr); 3949 } 3950 3951 if (build_domainname_from_strings(&instance, name, regtype, domain) < 0) 3952 { 3953 LogMsg("ERROR: handle_release_request bad “%s” “%s” “%s”", name, regtype, domain); 3954 return(mStatus_BadParamErr); 3955 } 3956 3957 LogOperation("%3d: PeerConnectionRelease(%X %##s) START PID[%d](%s)", 3958 request->sd, flags, instance.c, request->process_id, request->pid_name); 3959 3960 external_connection_release(&instance); 3961 return(err); 3962 } 3963 3964 #else // APPLE_OSX_mDNSResponder 3965 3966 mDNSlocal mStatus handle_release_request(request_state *request) 3967 { 3968 (void) request; 3969 return mStatus_UnsupportedErr; 3970 } 3971 3972 #endif // APPLE_OSX_mDNSResponder 3973 3974 mDNSlocal mStatus handle_setdomain_request(request_state *request) 3975 { 3976 char domainstr[MAX_ESCAPED_DOMAIN_NAME]; 3977 domainname domain; 3978 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 3979 (void)flags; // Unused 3980 if (get_string(&request->msgptr, request->msgend, domainstr, MAX_ESCAPED_DOMAIN_NAME) < 0 || 3981 !MakeDomainNameFromDNSNameString(&domain, domainstr)) 3982 { LogMsg("%3d: DNSServiceSetDefaultDomainForUser(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 3983 3984 LogOperation("%3d: DNSServiceSetDefaultDomainForUser(%##s)", request->sd, domain.c); 3985 return(mStatus_NoError); 3986 } 3987 3988 typedef packedstruct 3989 { 3990 mStatus err; 3991 mDNSu32 len; 3992 mDNSu32 vers; 3993 } DaemonVersionReply; 3994 3995 mDNSlocal void handle_getproperty_request(request_state *request) 3996 { 3997 const mStatus BadParamErr = dnssd_htonl((mDNSu32)mStatus_BadParamErr); 3998 char prop[256]; 3999 if (get_string(&request->msgptr, request->msgend, prop, sizeof(prop)) >= 0) 4000 { 4001 LogOperation("%3d: DNSServiceGetProperty(%s)", request->sd, prop); 4002 if (!strcmp(prop, kDNSServiceProperty_DaemonVersion)) 4003 { 4004 DaemonVersionReply x = { 0, dnssd_htonl(4), dnssd_htonl(_DNS_SD_H) }; 4005 send_all(request->sd, (const char *)&x, sizeof(x)); 4006 return; 4007 } 4008 } 4009 4010 // If we didn't recogize the requested property name, return BadParamErr 4011 send_all(request->sd, (const char *)&BadParamErr, sizeof(BadParamErr)); 4012 } 4013 4014 #ifdef APPLE_OSX_mDNSResponder 4015 // The caller can specify either the pid or the uuid. If the pid is not specified, 4016 // update the effective uuid. Don't overwrite the pid which is used for debugging 4017 // purposes and initialized when the socket is opened. 4018 mDNSlocal void handle_connection_delegate_request(request_state *request) 4019 { 4020 mDNSs32 pid; 4021 socklen_t len; 4022 4023 LogOperation("%3d: DNSServiceCreateDelegateConnection START PID[%d](%s)", 4024 request->sd, request->process_id, request->pid_name); 4025 request->terminate = connection_termination; 4026 4027 len = 0; 4028 pid = get_uint32(&request->msgptr, request->msgend); 4029 #ifdef LOCAL_PEEREPID 4030 if (pid) 4031 { 4032 len = sizeof(pid); 4033 if (getsockopt(request->sd, SOL_LOCAL, LOCAL_PEEREPID, &request->process_id, &len) != 0) 4034 { 4035 LogMsg("handle_connection_delegate_request: getsockopt for LOCAL_PEEREPID failed errno:%d / %s", errno, strerror(errno)); 4036 return; 4037 } 4038 // to extract the process name from the pid value 4039 if (proc_pidinfo(request->process_id, PROC_PIDT_SHORTBSDINFO, 1, &proc, PROC_PIDT_SHORTBSDINFO_SIZE) == 0) 4040 return; 4041 mDNSPlatformStrLCopy(request->pid_name, proc.pbsi_comm, sizeof(request->pid_name)); 4042 debugf("handle_connection_delegate_request: process id %d, name %s", request->process_id, request->pid_name); 4043 } 4044 #endif 4045 #ifdef LOCAL_PEEREUUID 4046 if (!pid) 4047 { 4048 len = UUID_SIZE; 4049 if (getsockopt(request->sd, SOL_LOCAL, LOCAL_PEEREUUID, request->uuid, &len) != 0) 4050 { 4051 LogMsg("handle_connection_delegate_request: getsockopt for LOCAL_PEEREUUID failed errno:%d / %s", errno, strerror(errno)); 4052 return; 4053 } 4054 request->validUUID = mDNStrue; 4055 } 4056 #endif 4057 } 4058 #else 4059 mDNSlocal void handle_connection_delegate_request(request_state *request) 4060 { 4061 (void) request; 4062 } 4063 #endif 4064 4065 typedef packedstruct 4066 { 4067 mStatus err; 4068 mDNSs32 pid; 4069 } PIDInfo; 4070 4071 mDNSlocal void handle_getpid_request(request_state *request) 4072 { 4073 const request_state *req; 4074 mDNSs32 pid = -1; 4075 mDNSu16 srcport = get_uint16(&request->msgptr, request->msgend); 4076 const DNSQuestion *q = NULL; 4077 PIDInfo pi; 4078 4079 LogMsg("%3d: DNSServiceGetPID START", request->sd); 4080 4081 for (req = all_requests; req; req=req->next) 4082 { 4083 if (req->hdr.op == query_request) 4084 q = &req->u.queryrecord.q; 4085 else if (req->hdr.op == addrinfo_request) 4086 q = &req->u.addrinfo.q4; 4087 else if (req->hdr.op == addrinfo_request) 4088 q = &req->u.addrinfo.q6; 4089 4090 if (q && q->LocalSocket != NULL) 4091 { 4092 mDNSu16 port = mDNSPlatformGetUDPPort(q->LocalSocket); 4093 if (port == srcport) 4094 { 4095 pid = req->process_id; 4096 LogMsg("DNSServiceGetPID: srcport %d, pid %d [%s] question %##s", htons(srcport), pid, req->pid_name, q->qname.c); 4097 break; 4098 } 4099 } 4100 } 4101 // If we cannot find in the client requests, look to see if this was 4102 // started by mDNSResponder. 4103 if (pid == -1) 4104 { 4105 for (q = mDNSStorage.Questions; q; q = q->next) 4106 { 4107 if (q && q->LocalSocket != NULL) 4108 { 4109 mDNSu16 port = mDNSPlatformGetUDPPort(q->LocalSocket); 4110 if (port == srcport) 4111 { 4112 #if APPLE_OSX_mDNSResponder 4113 pid = getpid(); 4114 #endif // APPLE_OSX_mDNSResponder 4115 LogMsg("DNSServiceGetPID: srcport %d, pid %d [%s], question %##s", htons(srcport), pid, "_mDNSResponder", q->qname.c); 4116 break; 4117 } 4118 } 4119 } 4120 } 4121 4122 pi.err = 0; 4123 pi.pid = pid; 4124 send_all(request->sd, (const char *)&pi, sizeof(PIDInfo)); 4125 LogMsg("%3d: DNSServiceGetPID STOP", request->sd); 4126 } 4127 4128 // *************************************************************************** 4129 #if COMPILER_LIKES_PRAGMA_MARK 4130 #pragma mark - 4131 #pragma mark - DNSServiceNATPortMappingCreate 4132 #endif 4133 4134 #define DNSServiceProtocol(X) ((X) == NATOp_AddrRequest ? 0 : (X) == NATOp_MapUDP ? kDNSServiceProtocol_UDP : kDNSServiceProtocol_TCP) 4135 4136 mDNSlocal void port_mapping_termination_callback(request_state *request) 4137 { 4138 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) STOP PID[%d](%s)", request->sd, 4139 DNSServiceProtocol(request->u.pm.NATinfo.Protocol), 4140 mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease, 4141 request->process_id, request->pid_name); 4142 mDNS_StopNATOperation(&mDNSStorage, &request->u.pm.NATinfo); 4143 } 4144 4145 // Called via function pointer when we get a NAT Traversal (address request or port mapping) response 4146 mDNSlocal void port_mapping_create_request_callback(mDNS *m, NATTraversalInfo *n) 4147 { 4148 request_state *request = (request_state *)n->clientContext; 4149 reply_state *rep; 4150 int replyLen; 4151 char *data; 4152 4153 if (!request) { LogMsg("port_mapping_create_request_callback called with unknown request_state object"); return; } 4154 4155 // calculate reply data length 4156 replyLen = sizeof(DNSServiceFlags); 4157 replyLen += 3 * sizeof(mDNSu32); // if index + addr + ttl 4158 replyLen += sizeof(DNSServiceErrorType); 4159 replyLen += 2 * sizeof(mDNSu16); // Internal Port + External Port 4160 replyLen += sizeof(mDNSu8); // protocol 4161 4162 rep = create_reply(port_mapping_reply_op, replyLen, request); 4163 4164 rep->rhdr->flags = dnssd_htonl(0); 4165 rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, n->InterfaceID, mDNSfalse)); 4166 rep->rhdr->error = dnssd_htonl(n->Result); 4167 4168 data = (char *)&rep->rhdr[1]; 4169 4170 *data++ = request->u.pm.NATinfo.ExternalAddress.b[0]; 4171 *data++ = request->u.pm.NATinfo.ExternalAddress.b[1]; 4172 *data++ = request->u.pm.NATinfo.ExternalAddress.b[2]; 4173 *data++ = request->u.pm.NATinfo.ExternalAddress.b[3]; 4174 *data++ = DNSServiceProtocol(request->u.pm.NATinfo.Protocol); 4175 *data++ = request->u.pm.NATinfo.IntPort.b[0]; 4176 *data++ = request->u.pm.NATinfo.IntPort.b[1]; 4177 *data++ = request->u.pm.NATinfo.ExternalPort.b[0]; 4178 *data++ = request->u.pm.NATinfo.ExternalPort.b[1]; 4179 put_uint32(request->u.pm.NATinfo.Lifetime, &data); 4180 4181 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) RESULT %.4a:%u TTL %u", request->sd, 4182 DNSServiceProtocol(request->u.pm.NATinfo.Protocol), 4183 mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease, 4184 &request->u.pm.NATinfo.ExternalAddress, mDNSVal16(request->u.pm.NATinfo.ExternalPort), request->u.pm.NATinfo.Lifetime); 4185 4186 append_reply(request, rep); 4187 } 4188 4189 mDNSlocal mStatus handle_port_mapping_request(request_state *request) 4190 { 4191 mDNSu32 ttl = 0; 4192 mStatus err = mStatus_NoError; 4193 4194 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 4195 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 4196 mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 4197 mDNSu8 protocol = (mDNSu8)get_uint32(&request->msgptr, request->msgend); 4198 (void)flags; // Unused 4199 if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 4200 if (request->msgptr + 8 > request->msgend) request->msgptr = NULL; 4201 else 4202 { 4203 request->u.pm.NATinfo.IntPort.b[0] = *request->msgptr++; 4204 request->u.pm.NATinfo.IntPort.b[1] = *request->msgptr++; 4205 request->u.pm.ReqExt.b[0] = *request->msgptr++; 4206 request->u.pm.ReqExt.b[1] = *request->msgptr++; 4207 ttl = get_uint32(&request->msgptr, request->msgend); 4208 } 4209 4210 if (!request->msgptr) 4211 { LogMsg("%3d: DNSServiceNATPortMappingCreate(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 4212 4213 if (protocol == 0) // If protocol == 0 (i.e. just request public address) then IntPort, ExtPort, ttl must be zero too 4214 { 4215 if (!mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort) || !mDNSIPPortIsZero(request->u.pm.ReqExt) || ttl) return(mStatus_BadParamErr); 4216 } 4217 else 4218 { 4219 if (mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort)) return(mStatus_BadParamErr); 4220 if (!(protocol & (kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP))) return(mStatus_BadParamErr); 4221 } 4222 4223 request->flags = flags; 4224 request->interfaceIndex = interfaceIndex; 4225 request->u.pm.NATinfo.Protocol = !protocol ? NATOp_AddrRequest : (protocol == kDNSServiceProtocol_UDP) ? NATOp_MapUDP : NATOp_MapTCP; 4226 // u.pm.NATinfo.IntPort = already set above 4227 request->u.pm.NATinfo.RequestedPort = request->u.pm.ReqExt; 4228 request->u.pm.NATinfo.NATLease = ttl; 4229 request->u.pm.NATinfo.clientCallback = port_mapping_create_request_callback; 4230 request->u.pm.NATinfo.clientContext = request; 4231 4232 LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) START PID[%d](%s)", request->sd, 4233 protocol, mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease, 4234 request->process_id, request->pid_name); 4235 err = mDNS_StartNATOperation(&mDNSStorage, &request->u.pm.NATinfo); 4236 if (err) LogMsg("ERROR: mDNS_StartNATOperation: %d", (int)err); 4237 else request->terminate = port_mapping_termination_callback; 4238 4239 return(err); 4240 } 4241 4242 // *************************************************************************** 4243 #if COMPILER_LIKES_PRAGMA_MARK 4244 #pragma mark - 4245 #pragma mark - DNSServiceGetAddrInfo 4246 #endif 4247 4248 mDNSlocal void addrinfo_termination_callback(request_state *request) 4249 { 4250 LogOperation("%3d: DNSServiceGetAddrInfo(%##s) STOP PID[%d](%s)", request->sd, request->u.addrinfo.q4.qname.c, 4251 request->process_id, request->pid_name); 4252 4253 if (request->u.addrinfo.q4.QuestionContext) 4254 { 4255 mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4); 4256 LogMcastQ(&request->u.addrinfo.q4, request, q_stop); 4257 request->u.addrinfo.q4.QuestionContext = mDNSNULL; 4258 4259 if (callExternalHelpers(request->u.addrinfo.interface_id, &request->u.addrinfo.q4.qname, request->flags)) 4260 { 4261 LogInfo("addrinfo_termination_callback: calling external_stop_browsing_for_service() for kDNSServiceType_A record"); 4262 external_stop_browsing_for_service(request->u.addrinfo.interface_id, &request->u.addrinfo.q4.qname, kDNSServiceType_A, request->flags); 4263 } 4264 } 4265 if (request->u.addrinfo.q4.qnameOrig) 4266 { 4267 freeL("QueryTermination", request->u.addrinfo.q4.qnameOrig); 4268 request->u.addrinfo.q4.qnameOrig = mDNSNULL; 4269 } 4270 if (request->u.addrinfo.q42) 4271 { 4272 if (request->u.addrinfo.q42->QuestionContext) 4273 { 4274 LogInfo("addrinfo_termination_callback: Stopping q42 %##s", request->u.addrinfo.q42->qname.c); 4275 mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q42); 4276 LogMcastQ(request->u.addrinfo.q42, request, q_stop); 4277 } 4278 if (request->u.addrinfo.q42->qnameOrig) 4279 { 4280 LogInfo("addrinfo_termination_callback: freeing q42 qnameOrig %##s", request->u.addrinfo.q42->qnameOrig->c); 4281 freeL("QueryTermination q42", request->u.addrinfo.q42->qnameOrig); 4282 request->u.addrinfo.q42->qnameOrig = mDNSNULL; 4283 } 4284 freeL("addrinfo Q42", request->u.addrinfo.q42); 4285 request->u.addrinfo.q42 = mDNSNULL; 4286 } 4287 4288 if (request->u.addrinfo.q6.QuestionContext) 4289 { 4290 mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q6); 4291 LogMcastQ(&request->u.addrinfo.q6, request, q_stop); 4292 request->u.addrinfo.q6.QuestionContext = mDNSNULL; 4293 4294 if (callExternalHelpers(request->u.addrinfo.interface_id, &request->u.addrinfo.q6.qname, request->flags)) 4295 { 4296 LogInfo("addrinfo_termination_callback: calling external_stop_browsing_for_service() for kDNSServiceType_AAAA record"); 4297 external_stop_browsing_for_service(request->u.addrinfo.interface_id, &request->u.addrinfo.q6.qname, kDNSServiceType_AAAA, request->flags); 4298 } 4299 } 4300 if (request->u.addrinfo.q6.qnameOrig) 4301 { 4302 freeL("QueryTermination", request->u.addrinfo.q6.qnameOrig); 4303 request->u.addrinfo.q6.qnameOrig = mDNSNULL; 4304 } 4305 if (request->u.addrinfo.q62) 4306 { 4307 if (request->u.addrinfo.q62->QuestionContext) 4308 { 4309 LogInfo("addrinfo_termination_callback: Stopping q62 %##s", request->u.addrinfo.q62->qname.c); 4310 mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q62); 4311 LogMcastQ(request->u.addrinfo.q62, request, q_stop); 4312 } 4313 if (request->u.addrinfo.q62->qnameOrig) 4314 { 4315 LogInfo("addrinfo_termination_callback: freeing q62 qnameOrig %##s", request->u.addrinfo.q62->qnameOrig->c); 4316 freeL("QueryTermination q62", request->u.addrinfo.q62->qnameOrig); 4317 request->u.addrinfo.q62->qnameOrig = mDNSNULL; 4318 } 4319 freeL("addrinfo Q62", request->u.addrinfo.q62); 4320 request->u.addrinfo.q62 = mDNSNULL; 4321 } 4322 #if APPLE_OSX_mDNSResponder 4323 { 4324 DNSQuestion *v4q, *v6q; 4325 v4q = v6q = mDNSNULL; 4326 if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4) 4327 { 4328 // If we are not delivering answers, we may be timing out prematurely. 4329 // Note down the current state so that we know to retry when we see a 4330 // valid response again. 4331 if (request->u.addrinfo.q4.TimeoutQuestion && !request->u.addrinfo.v4ans) 4332 { 4333 mDNSPlatformUpdateDNSStatus(&request->u.addrinfo.q4); 4334 } 4335 // If we have a v4 answer and if we timed out prematurely before, provide 4336 // a trigger to the upper layer so that it can retry questions if needed. 4337 if (request->u.addrinfo.v4ans) 4338 v4q = &request->u.addrinfo.q4; 4339 } 4340 if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6) 4341 { 4342 if (request->u.addrinfo.q6.TimeoutQuestion && !request->u.addrinfo.v6ans) 4343 { 4344 mDNSPlatformUpdateDNSStatus(&request->u.addrinfo.q6); 4345 } 4346 if (request->u.addrinfo.v6ans) 4347 v6q = &request->u.addrinfo.q6; 4348 } 4349 mDNSPlatformTriggerDNSRetry(v4q, v6q); 4350 } 4351 #endif // APPLE_OSX_mDNSResponder 4352 } 4353 4354 mDNSlocal mStatus handle_addrinfo_request(request_state *request) 4355 { 4356 char hostname[256]; 4357 size_t hostnameLen; 4358 domainname d; 4359 mStatus err = 0; 4360 mDNSs32 serviceIndex = -1; // default unscoped value for ServiceID is -1 4361 mDNSInterfaceID InterfaceID; 4362 4363 DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 4364 4365 mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 4366 4367 if (flags & kDNSServiceFlagsServiceIndex) 4368 { 4369 // NOTE: kDNSServiceFlagsServiceIndex flag can only be set for DNSServiceGetAddrInfo() 4370 LogInfo("DNSServiceGetAddrInfo: kDNSServiceFlagsServiceIndex is SET by the client"); 4371 // if kDNSServiceFlagsServiceIndex is SET, 4372 // interpret the interfaceID as the serviceId and set the interfaceID to 0. 4373 serviceIndex = interfaceIndex; 4374 interfaceIndex = 0; 4375 } 4376 4377 mDNSPlatformMemZero(&request->u.addrinfo, sizeof(request->u.addrinfo)); 4378 4379 InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 4380 4381 // The request is scoped to a specific interface index, but the 4382 // interface is not currently in our list. 4383 if (interfaceIndex && !InterfaceID) 4384 { 4385 if (interfaceIndex > 1) 4386 LogMsg("handle_addrinfo_request: interfaceIndex %d is currently inactive requested by client[%d][%s]", 4387 interfaceIndex, request->process_id, request->pid_name); 4388 // If it's one of the specially defined inteface index values, just return an error. 4389 if (PreDefinedInterfaceIndex(interfaceIndex)) 4390 { 4391 LogInfo("handle_addrinfo_request: bad interfaceIndex %d", interfaceIndex); 4392 return(mStatus_BadParamErr); 4393 } 4394 4395 // Otherwise, use the specified interface index value and the request will 4396 // be applied to that interface when it comes up. 4397 InterfaceID = (mDNSInterfaceID)(uintptr_t)interfaceIndex; 4398 LogInfo("handle_addrinfo_request: query pending for interface index %d", interfaceIndex); 4399 } 4400 4401 request->flags = flags; 4402 request->interfaceIndex = interfaceIndex; 4403 request->u.addrinfo.interface_id = InterfaceID; 4404 request->u.addrinfo.flags = flags; 4405 request->u.addrinfo.protocol = get_uint32(&request->msgptr, request->msgend); 4406 4407 if (request->u.addrinfo.protocol > (kDNSServiceProtocol_IPv4|kDNSServiceProtocol_IPv6)) return(mStatus_BadParamErr); 4408 4409 if (get_string(&request->msgptr, request->msgend, hostname, 256) < 0) return(mStatus_BadParamErr); 4410 4411 if (!request->msgptr) { LogMsg("%3d: DNSServiceGetAddrInfo(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 4412 4413 if (!MakeDomainNameFromDNSNameString(&d, hostname)) 4414 { LogMsg("ERROR: handle_addrinfo_request: bad hostname: %s", hostname); return(mStatus_BadParamErr); } 4415 4416 #if 0 4417 if (!AuthorizedDomain(request, &d, AutoBrowseDomains)) return (mStatus_NoError); 4418 #endif 4419 4420 if (!request->u.addrinfo.protocol) 4421 { 4422 flags |= kDNSServiceFlagsSuppressUnusable; 4423 request->u.addrinfo.protocol = (kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6); 4424 } 4425 4426 request->u.addrinfo.q4.InterfaceID = request->u.addrinfo.q6.InterfaceID = request->u.addrinfo.interface_id; 4427 request->u.addrinfo.q4.ServiceID = request->u.addrinfo.q6.ServiceID = serviceIndex; 4428 request->u.addrinfo.q4.flags = request->u.addrinfo.q6.flags = flags; 4429 request->u.addrinfo.q4.Target = request->u.addrinfo.q6.Target = zeroAddr; 4430 request->u.addrinfo.q4.qname = request->u.addrinfo.q6.qname = d; 4431 request->u.addrinfo.q4.qclass = request->u.addrinfo.q6.qclass = kDNSServiceClass_IN; 4432 request->u.addrinfo.q4.LongLived = request->u.addrinfo.q6.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 4433 request->u.addrinfo.q4.ExpectUnique = request->u.addrinfo.q6.ExpectUnique = mDNSfalse; 4434 request->u.addrinfo.q4.ForceMCast = request->u.addrinfo.q6.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 4435 request->u.addrinfo.q4.ReturnIntermed = request->u.addrinfo.q6.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 4436 request->u.addrinfo.q4.SuppressUnusable = request->u.addrinfo.q6.SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0; 4437 request->u.addrinfo.q4.TimeoutQuestion = request->u.addrinfo.q6.TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0; 4438 request->u.addrinfo.q4.WakeOnResolve = request->u.addrinfo.q6.WakeOnResolve = 0; 4439 request->u.addrinfo.q4.UseBackgroundTrafficClass = request->u.addrinfo.q6.UseBackgroundTrafficClass = (flags & kDNSServiceFlagsBackgroundTrafficClass) != 0; 4440 if ((flags & kDNSServiceFlagsValidate) != 0) 4441 request->u.addrinfo.q4.ValidationRequired = request->u.addrinfo.q6.ValidationRequired = DNSSEC_VALIDATION_SECURE; 4442 else if ((flags & kDNSServiceFlagsValidateOptional) != 0) 4443 request->u.addrinfo.q4.ValidationRequired = request->u.addrinfo.q6.ValidationRequired = DNSSEC_VALIDATION_SECURE_OPTIONAL; 4444 request->u.addrinfo.q4.ValidatingResponse = request->u.addrinfo.q6.ValidatingResponse = 0; 4445 request->u.addrinfo.q4.ProxyQuestion = request->u.addrinfo.q6.ProxyQuestion = 0; 4446 request->u.addrinfo.q4.qnameOrig = request->u.addrinfo.q6.qnameOrig = mDNSNULL; 4447 request->u.addrinfo.q4.AnonInfo = request->u.addrinfo.q6.AnonInfo = mDNSNULL; 4448 4449 SetQuestionPolicy(&request->u.addrinfo.q4, request); 4450 SetQuestionPolicy(&request->u.addrinfo.q6, request); 4451 4452 request->u.addrinfo.q4.StopTime = request->u.addrinfo.q6.StopTime = 0; 4453 4454 request->u.addrinfo.q4.DNSSECAuthInfo = request->u.addrinfo.q6.DNSSECAuthInfo = mDNSNULL; 4455 request->u.addrinfo.q4.DAIFreeCallback = request->u.addrinfo.q6.DAIFreeCallback = mDNSNULL; 4456 4457 //Turn off dnssec validation for local domains 4458 if (IsLocalDomain(&d)) 4459 request->u.addrinfo.q4.ValidationRequired = request->u.addrinfo.q6.ValidationRequired = 0; 4460 4461 hostnameLen = strlen(hostname); 4462 4463 LogOperation("%3d: DNSServiceGetAddrInfo(%X, %d, %d, %##s) START PID[%d](%s)", 4464 request->sd, flags, interfaceIndex, request->u.addrinfo.protocol, d.c, request->process_id, request->pid_name); 4465 4466 if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6) 4467 { 4468 request->u.addrinfo.q6.qtype = kDNSServiceType_AAAA; 4469 request->u.addrinfo.q6.SearchListIndex = 0; 4470 // For DNSSEC questions, append search domains only if kDNSServiceFlagsValidateOptional is set 4471 if ((!(request->u.addrinfo.q6.ValidationRequired == DNSSEC_VALIDATION_SECURE)) && (!(request->u.addrinfo.q6.ValidationRequired == DNSSEC_VALIDATION_INSECURE)) 4472 && ((hostnameLen == 0) || (hostname[hostnameLen - 1] != '.')) && (AlwaysAppendSearchDomains || CountLabels(&d) == 1)) 4473 { 4474 request->u.addrinfo.q6.AppendSearchDomains = 1; 4475 request->u.addrinfo.q6.AppendLocalSearchDomains = 1; 4476 } 4477 else 4478 { 4479 request->u.addrinfo.q6.AppendSearchDomains = 0; 4480 request->u.addrinfo.q6.AppendLocalSearchDomains = 0; 4481 } 4482 request->u.addrinfo.q6.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q6) ? 1 : 0); 4483 request->u.addrinfo.q6.QuestionCallback = queryrecord_result_callback; 4484 request->u.addrinfo.q6.QuestionContext = request; 4485 err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q6); 4486 if (err != mStatus_NoError) 4487 { 4488 LogMsg("ERROR: mDNS_StartQuery: %d", (int)err); 4489 request->u.addrinfo.q6.QuestionContext = mDNSNULL; 4490 } 4491 #if APPLE_OSX_mDNSResponder 4492 err = SendAdditionalQuery(&request->u.addrinfo.q6, request, err); 4493 #endif // APPLE_OSX_mDNSResponder 4494 if (!err) 4495 { 4496 request->terminate = addrinfo_termination_callback; 4497 LogMcastQ(&request->u.addrinfo.q6, request, q_start); 4498 if (callExternalHelpers(InterfaceID, &d, flags)) 4499 { 4500 LogInfo("handle_addrinfo_request: calling external_start_browsing_for_service() for kDNSServiceType_AAAA record"); 4501 external_start_browsing_for_service(InterfaceID, &d, kDNSServiceType_AAAA, flags); 4502 } 4503 } 4504 } 4505 4506 if (!err && (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4)) 4507 { 4508 request->u.addrinfo.q4.qtype = kDNSServiceType_A; 4509 request->u.addrinfo.q4.SearchListIndex = 0; 4510 4511 // We append search domains only for queries that are a single label. If overriden using cmd line arg 4512 // "AlwaysAppendSearchDomains", then we do it for any query which is not fully qualified. 4513 // For DNSSEC questions, append search domains only if kDNSServiceFlagsValidateOptional is set. 4514 4515 if ((!(request->u.addrinfo.q4.ValidationRequired == DNSSEC_VALIDATION_SECURE)) && (!(request->u.addrinfo.q4.ValidationRequired == DNSSEC_VALIDATION_INSECURE)) 4516 && ((hostnameLen == 0) || (hostname[hostnameLen - 1] != '.')) && (AlwaysAppendSearchDomains || CountLabels(&d) == 1)) 4517 { 4518 request->u.addrinfo.q4.AppendSearchDomains = 1; 4519 request->u.addrinfo.q4.AppendLocalSearchDomains = 1; 4520 } 4521 else 4522 { 4523 request->u.addrinfo.q4.AppendSearchDomains = 0; 4524 request->u.addrinfo.q4.AppendLocalSearchDomains = 0; 4525 } 4526 request->u.addrinfo.q4.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q4) ? 1 : 0); 4527 request->u.addrinfo.q4.QuestionCallback = queryrecord_result_callback; 4528 request->u.addrinfo.q4.QuestionContext = request; 4529 err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q4); 4530 if (err != mStatus_NoError) 4531 { 4532 LogMsg("ERROR: mDNS_StartQuery: %d", (int)err); 4533 request->u.addrinfo.q4.QuestionContext = mDNSNULL; 4534 if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6) 4535 { 4536 // If we started a query for IPv6, we need to cancel it 4537 mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q6); 4538 request->u.addrinfo.q6.QuestionContext = mDNSNULL; 4539 4540 if (callExternalHelpers(InterfaceID, &d, flags)) 4541 { 4542 LogInfo("addrinfo_termination_callback: calling external_stop_browsing_for_service() for kDNSServiceType_AAAA record"); 4543 external_stop_browsing_for_service(InterfaceID, &d, kDNSServiceType_AAAA, flags); 4544 } 4545 } 4546 } 4547 #if APPLE_OSX_mDNSResponder 4548 err = SendAdditionalQuery(&request->u.addrinfo.q4, request, err); 4549 #endif // APPLE_OSX_mDNSResponder 4550 if (!err) 4551 { 4552 request->terminate = addrinfo_termination_callback; 4553 LogMcastQ(&request->u.addrinfo.q4, request, q_start); 4554 if (callExternalHelpers(InterfaceID, &d, flags)) 4555 { 4556 LogInfo("handle_addrinfo_request: calling external_start_browsing_for_service() for kDNSServiceType_A record"); 4557 external_start_browsing_for_service(InterfaceID, &d, kDNSServiceType_A, flags); 4558 } 4559 } 4560 } 4561 4562 return(err); 4563 } 4564 4565 // *************************************************************************** 4566 #if COMPILER_LIKES_PRAGMA_MARK 4567 #pragma mark - 4568 #pragma mark - Main Request Handler etc. 4569 #endif 4570 4571 mDNSlocal request_state *NewRequest(void) 4572 { 4573 request_state **p = &all_requests; 4574 while (*p) 4575 p=&(*p)->next; 4576 *p = mallocL("request_state", sizeof(request_state)); 4577 if (!*p) 4578 FatalError("ERROR: malloc"); 4579 mDNSPlatformMemZero(*p, sizeof(request_state)); 4580 return(*p); 4581 } 4582 4583 // read_msg may be called any time when the transfer state (req->ts) is t_morecoming. 4584 // if there is no data on the socket, the socket will be closed and t_terminated will be returned 4585 mDNSlocal void read_msg(request_state *req) 4586 { 4587 if (req->ts == t_terminated || req->ts == t_error) 4588 { LogMsg("%3d: ERROR: read_msg called with transfer state terminated or error", req->sd); req->ts = t_error; return; } 4589 4590 if (req->ts == t_complete) // this must be death or something is wrong 4591 { 4592 char buf[4]; // dummy for death notification 4593 int nread = udsSupportReadFD(req->sd, buf, 4, 0, req->platform_data); 4594 if (!nread) { req->ts = t_terminated; return; } 4595 if (nread < 0) goto rerror; 4596 LogMsg("%3d: ERROR: read data from a completed request", req->sd); 4597 req->ts = t_error; 4598 return; 4599 } 4600 4601 if (req->ts != t_morecoming) 4602 { LogMsg("%3d: ERROR: read_msg called with invalid transfer state (%d)", req->sd, req->ts); req->ts = t_error; return; } 4603 4604 if (req->hdr_bytes < sizeof(ipc_msg_hdr)) 4605 { 4606 mDNSu32 nleft = sizeof(ipc_msg_hdr) - req->hdr_bytes; 4607 int nread = udsSupportReadFD(req->sd, (char *)&req->hdr + req->hdr_bytes, nleft, 0, req->platform_data); 4608 if (nread == 0) { req->ts = t_terminated; return; } 4609 if (nread < 0) goto rerror; 4610 req->hdr_bytes += nread; 4611 if (req->hdr_bytes > sizeof(ipc_msg_hdr)) 4612 { LogMsg("%3d: ERROR: read_msg - read too many header bytes", req->sd); req->ts = t_error; return; } 4613 4614 // only read data if header is complete 4615 if (req->hdr_bytes == sizeof(ipc_msg_hdr)) 4616 { 4617 ConvertHeaderBytes(&req->hdr); 4618 if (req->hdr.version != VERSION) 4619 { LogMsg("%3d: ERROR: client version 0x%08X daemon version 0x%08X", req->sd, req->hdr.version, VERSION); req->ts = t_error; return; } 4620 4621 // Largest conceivable single request is a DNSServiceRegisterRecord() or DNSServiceAddRecord() 4622 // with 64kB of rdata. Adding 1009 byte for a maximal domain name, plus a safety margin 4623 // for other overhead, this means any message above 70kB is definitely bogus. 4624 if (req->hdr.datalen > 70000) 4625 { LogMsg("%3d: ERROR: read_msg: hdr.datalen %u (0x%X) > 70000", req->sd, req->hdr.datalen, req->hdr.datalen); req->ts = t_error; return; } 4626 req->msgbuf = mallocL("request_state msgbuf", req->hdr.datalen + MSG_PAD_BYTES); 4627 if (!req->msgbuf) { my_perror("ERROR: malloc"); req->ts = t_error; return; } 4628 req->msgptr = req->msgbuf; 4629 req->msgend = req->msgbuf + req->hdr.datalen; 4630 mDNSPlatformMemZero(req->msgbuf, req->hdr.datalen + MSG_PAD_BYTES); 4631 } 4632 } 4633 4634 // If our header is complete, but we're still needing more body data, then try to read it now 4635 // Note: For cancel_request req->hdr.datalen == 0, but there's no error return socket for cancel_request 4636 // Any time we need to get the error return socket we know we'll have at least one data byte 4637 // (even if only the one-byte empty C string placeholder for the old ctrl_path parameter) 4638 if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes < req->hdr.datalen) 4639 { 4640 mDNSu32 nleft = req->hdr.datalen - req->data_bytes; 4641 int nread; 4642 #if !defined(_WIN32) 4643 struct iovec vec = { req->msgbuf + req->data_bytes, nleft }; // Tell recvmsg where we want the bytes put 4644 struct msghdr msg; 4645 struct cmsghdr *cmsg; 4646 char cbuf[CMSG_SPACE(4 * sizeof(dnssd_sock_t))]; 4647 msg.msg_name = 0; 4648 msg.msg_namelen = 0; 4649 msg.msg_iov = &vec; 4650 msg.msg_iovlen = 1; 4651 msg.msg_control = cbuf; 4652 msg.msg_controllen = sizeof(cbuf); 4653 msg.msg_flags = 0; 4654 nread = recvmsg(req->sd, &msg, 0); 4655 #else 4656 nread = udsSupportReadFD(req->sd, (char *)req->msgbuf + req->data_bytes, nleft, 0, req->platform_data); 4657 #endif 4658 if (nread == 0) { req->ts = t_terminated; return; } 4659 if (nread < 0) goto rerror; 4660 req->data_bytes += nread; 4661 if (req->data_bytes > req->hdr.datalen) 4662 { LogMsg("%3d: ERROR: read_msg - read too many data bytes", req->sd); req->ts = t_error; return; } 4663 #if !defined(_WIN32) 4664 cmsg = CMSG_FIRSTHDR(&msg); 4665 #if DEBUG_64BIT_SCM_RIGHTS 4666 LogMsg("%3d: Expecting %d %d %d %d", req->sd, sizeof(cbuf), sizeof(cbuf), SOL_SOCKET, SCM_RIGHTS); 4667 LogMsg("%3d: Got %d %d %d %d", req->sd, msg.msg_controllen, cmsg ? cmsg->cmsg_len : -1, cmsg ? cmsg->cmsg_level : -1, cmsg ? cmsg->cmsg_type : -1); 4668 #endif // DEBUG_64BIT_SCM_RIGHTS 4669 if (cmsg && cmsg->cmsg_level == SOL_SOCKET && cmsg->cmsg_type == SCM_RIGHTS) 4670 { 4671 #if APPLE_OSX_mDNSResponder 4672 // Strictly speaking BPF_fd belongs solely in the platform support layer, but because 4673 // of privilege separation on Mac OS X we need to get BPF_fd from mDNSResponderHelper, 4674 // and it's convenient to repurpose the existing fd-passing code here for that task 4675 if (req->hdr.op == send_bpf) 4676 { 4677 dnssd_sock_t x = *(dnssd_sock_t *)CMSG_DATA(cmsg); 4678 LogOperation("%3d: Got len %d, BPF %d", req->sd, cmsg->cmsg_len, x); 4679 mDNSPlatformReceiveBPF_fd(x); 4680 } 4681 else 4682 #endif // APPLE_OSX_mDNSResponder 4683 req->errsd = *(dnssd_sock_t *)CMSG_DATA(cmsg); 4684 #if DEBUG_64BIT_SCM_RIGHTS 4685 LogMsg("%3d: read req->errsd %d", req->sd, req->errsd); 4686 #endif // DEBUG_64BIT_SCM_RIGHTS 4687 if (req->data_bytes < req->hdr.datalen) 4688 { 4689 LogMsg("%3d: Client(PID [%d](%s)) sent result code socket %d via SCM_RIGHTS with req->data_bytes %d < req->hdr.datalen %d", 4690 req->sd, req->process_id, req->pid_name, req->errsd, req->data_bytes, req->hdr.datalen); 4691 req->ts = t_error; 4692 return; 4693 } 4694 } 4695 #endif 4696 } 4697 4698 // If our header and data are both complete, see if we need to make our separate error return socket 4699 if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes == req->hdr.datalen) 4700 { 4701 if (req->terminate && req->hdr.op != cancel_request) 4702 { 4703 dnssd_sockaddr_t cliaddr; 4704 #if defined(USE_TCP_LOOPBACK) 4705 mDNSOpaque16 port; 4706 u_long opt = 1; 4707 port.b[0] = req->msgptr[0]; 4708 port.b[1] = req->msgptr[1]; 4709 req->msgptr += 2; 4710 cliaddr.sin_family = AF_INET; 4711 cliaddr.sin_port = port.NotAnInteger; 4712 cliaddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR); 4713 #else 4714 char ctrl_path[MAX_CTLPATH]; 4715 get_string(&req->msgptr, req->msgend, ctrl_path, MAX_CTLPATH); // path is first element in message buffer 4716 mDNSPlatformMemZero(&cliaddr, sizeof(cliaddr)); 4717 cliaddr.sun_family = AF_LOCAL; 4718 mDNSPlatformStrLCopy(cliaddr.sun_path, ctrl_path, sizeof(cliaddr.sun_path)); 4719 // If the error return path UDS name is empty string, that tells us 4720 // that this is a new version of the library that's going to pass us 4721 // the error return path socket via sendmsg/recvmsg 4722 if (ctrl_path[0] == 0) 4723 { 4724 if (req->errsd == req->sd) 4725 { LogMsg("%3d: read_msg: ERROR failed to get errsd via SCM_RIGHTS", req->sd); req->ts = t_error; return; } 4726 goto got_errfd; 4727 } 4728 #endif 4729 4730 req->errsd = socket(AF_DNSSD, SOCK_STREAM, 0); 4731 if (!dnssd_SocketValid(req->errsd)) 4732 { 4733 my_throttled_perror("ERROR: socket"); 4734 req->ts = t_error; 4735 return; 4736 } 4737 4738 if (connect(req->errsd, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0) 4739 { 4740 #if !defined(USE_TCP_LOOPBACK) 4741 struct stat sb; 4742 LogMsg("%3d: read_msg: Couldn't connect to error return path socket “%s” errno %d (%s)", 4743 req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno)); 4744 if (stat(cliaddr.sun_path, &sb) < 0) 4745 LogMsg("%3d: read_msg: stat failed “%s” errno %d (%s)", req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno)); 4746 else 4747 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); 4748 #endif 4749 req->ts = t_error; 4750 return; 4751 } 4752 4753 #if !defined(USE_TCP_LOOPBACK) 4754 got_errfd: 4755 #endif 4756 LogOperation("%3d: Result code socket %d created %08X %08X", req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0]); 4757 #if defined(_WIN32) 4758 if (ioctlsocket(req->errsd, FIONBIO, &opt) != 0) 4759 #else 4760 if (fcntl(req->errsd, F_SETFL, fcntl(req->errsd, F_GETFL, 0) | O_NONBLOCK) != 0) 4761 #endif 4762 { 4763 LogMsg("%3d: ERROR: could not set control socket to non-blocking mode errno %d (%s)", 4764 req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 4765 req->ts = t_error; 4766 return; 4767 } 4768 } 4769 4770 req->ts = t_complete; 4771 } 4772 4773 return; 4774 4775 rerror: 4776 if (dnssd_errno == dnssd_EWOULDBLOCK || dnssd_errno == dnssd_EINTR) return; 4777 LogMsg("%3d: ERROR: read_msg errno %d (%s)", req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 4778 req->ts = t_error; 4779 } 4780 4781 mDNSlocal mStatus handle_client_request(request_state *req) 4782 { 4783 mStatus err = mStatus_NoError; 4784 switch(req->hdr.op) 4785 { 4786 // These are all operations that have their own first-class request_state object 4787 case connection_request: 4788 LogOperation("%3d: DNSServiceCreateConnection START PID[%d](%s)", 4789 req->sd, req->process_id, req->pid_name); 4790 req->terminate = connection_termination; 4791 break; 4792 case connection_delegate_request: 4793 LogOperation("%3d: DNSServiceCreateDelegateConnection START PID[%d](%s)", 4794 req->sd, req->process_id, req->pid_name); 4795 req->terminate = connection_termination; 4796 handle_connection_delegate_request(req); 4797 break; 4798 case resolve_request: err = handle_resolve_request (req); break; 4799 case query_request: err = handle_queryrecord_request (req); break; 4800 case browse_request: err = handle_browse_request (req); break; 4801 case reg_service_request: err = handle_regservice_request (req); break; 4802 case enumeration_request: err = handle_enum_request (req); break; 4803 case reconfirm_record_request: err = handle_reconfirm_request (req); break; 4804 case setdomain_request: err = handle_setdomain_request (req); break; 4805 case getproperty_request: handle_getproperty_request (req); break; 4806 case getpid_request: handle_getpid_request (req); break; 4807 case port_mapping_request: err = handle_port_mapping_request(req); break; 4808 case addrinfo_request: err = handle_addrinfo_request (req); break; 4809 case send_bpf: /* Do nothing for send_bpf */ break; 4810 4811 // These are all operations that work with an existing request_state object 4812 case reg_record_request: err = handle_regrecord_request (req); break; 4813 case add_record_request: err = handle_add_request (req); break; 4814 case update_record_request: err = handle_update_request (req); break; 4815 case remove_record_request: err = handle_removerecord_request(req); break; 4816 case cancel_request: handle_cancel_request (req); break; 4817 case release_request: err = handle_release_request (req); break; 4818 default: LogMsg("request_callback: %3d:ERROR: Unsupported UDS req:%d PID[%d][%s]", 4819 req->sd, req->hdr.op, req->process_id, req->pid_name); 4820 err = mStatus_BadParamErr; 4821 break; 4822 } 4823 4824 return err; 4825 } 4826 4827 #define RecordOrientedOp(X) \ 4828 ((X) == reg_record_request || (X) == add_record_request || (X) == update_record_request || (X) == remove_record_request) 4829 4830 // The lightweight operations are the ones that don't need a dedicated request_state structure allocated for them 4831 #define LightweightOp(X) (RecordOrientedOp(X) || (X) == cancel_request) 4832 4833 mDNSlocal void request_callback(int fd, short filter, void *info) 4834 { 4835 mStatus err = 0; 4836 request_state *req = info; 4837 mDNSs32 min_size = sizeof(DNSServiceFlags); 4838 (void)fd; // Unused 4839 (void)filter; // Unused 4840 4841 for (;;) 4842 { 4843 read_msg(req); 4844 if (req->ts == t_morecoming) 4845 return; 4846 if (req->ts == t_terminated || req->ts == t_error) 4847 { 4848 AbortUnlinkAndFree(req); 4849 return; 4850 } 4851 if (req->ts != t_complete) 4852 { 4853 LogMsg("request_callback: req->ts %d != t_complete PID[%d][%s]", req->ts, req->process_id, req->pid_name); 4854 AbortUnlinkAndFree(req); 4855 return; 4856 } 4857 4858 switch(req->hdr.op) // Interface + other data 4859 { 4860 case connection_request: min_size = 0; break; 4861 case connection_delegate_request: min_size = 4; /* pid */ break; 4862 case reg_service_request: min_size += sizeof(mDNSu32) + 4 /* name, type, domain, host */ + 4 /* port, textlen */; break; 4863 case add_record_request: min_size += 4 /* type, rdlen */ + 4 /* ttl */; break; 4864 case update_record_request: min_size += 2 /* rdlen */ + 4 /* ttl */; break; 4865 case remove_record_request: break; 4866 case browse_request: min_size += sizeof(mDNSu32) + 2 /* type, domain */; break; 4867 case resolve_request: min_size += sizeof(mDNSu32) + 3 /* type, type, domain */; break; 4868 case query_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 4 /* type, class*/; break; 4869 case enumeration_request: min_size += sizeof(mDNSu32); break; 4870 case reg_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */ + 4 /* ttl */; break; 4871 case reconfirm_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */; break; 4872 case setdomain_request: min_size += 1 /* domain */; break; 4873 case getproperty_request: min_size = 2; break; 4874 case getpid_request: min_size = 2; break; 4875 case port_mapping_request: min_size += sizeof(mDNSu32) + 4 /* udp/tcp */ + 4 /* int/ext port */ + 4 /* ttl */; break; 4876 case addrinfo_request: min_size += sizeof(mDNSu32) + 4 /* v4/v6 */ + 1 /* hostname */; break; 4877 case send_bpf: // Same as cancel_request below 4878 case cancel_request: min_size = 0; break; 4879 case release_request: min_size += sizeof(mDNSu32) + 3 /* type, type, domain */; break; 4880 default: LogMsg("request_callback: ERROR: validate_message - unsupported req type: %d PID[%d][%s]", 4881 req->hdr.op, req->process_id, req->pid_name); 4882 min_size = -1; break; 4883 } 4884 4885 if ((mDNSs32)req->data_bytes < min_size) 4886 { 4887 LogMsg("request_callback: Invalid message %d bytes; min for %d is %d PID[%d][%s]", 4888 req->data_bytes, req->hdr.op, min_size, req->process_id, req->pid_name); 4889 AbortUnlinkAndFree(req); 4890 return; 4891 } 4892 if (LightweightOp(req->hdr.op) && !req->terminate) 4893 { 4894 LogMsg("request_callback: Reg/Add/Update/Remove %d require existing connection PID[%d][%s]", 4895 req->hdr.op, req->process_id, req->pid_name); 4896 AbortUnlinkAndFree(req); 4897 return; 4898 } 4899 4900 // If req->terminate is already set, this means this operation is sharing an existing connection 4901 if (req->terminate && !LightweightOp(req->hdr.op)) 4902 { 4903 request_state *newreq = NewRequest(); 4904 newreq->primary = req; 4905 newreq->sd = req->sd; 4906 newreq->errsd = req->errsd; 4907 newreq->uid = req->uid; 4908 newreq->hdr = req->hdr; 4909 newreq->msgbuf = req->msgbuf; 4910 newreq->msgptr = req->msgptr; 4911 newreq->msgend = req->msgend; 4912 // if the parent request is a delegate connection, copy the 4913 // relevant bits 4914 if (req->validUUID) 4915 { 4916 newreq->validUUID = mDNStrue; 4917 mDNSPlatformMemCopy(newreq->uuid, req->uuid, UUID_SIZE); 4918 } 4919 else 4920 { 4921 if (req->process_id) 4922 { 4923 newreq->process_id = req->process_id; 4924 mDNSPlatformStrLCopy(newreq->pid_name, req->pid_name, (mDNSu32)sizeof(newreq->pid_name)); 4925 } 4926 else 4927 { 4928 set_peer_pid(newreq); 4929 } 4930 } 4931 req = newreq; 4932 } 4933 4934 // Check if the request wants no asynchronous replies. 4935 if (req->hdr.ipc_flags & IPC_FLAGS_NOREPLY) req->no_reply = 1; 4936 4937 // If we're shutting down, don't allow new client requests 4938 // We do allow "cancel" and "getproperty" during shutdown 4939 if (mDNSStorage.ShutdownTime && req->hdr.op != cancel_request && req->hdr.op != getproperty_request) 4940 err = mStatus_ServiceNotRunning; 4941 else 4942 err = handle_client_request(req); 4943 4944 // req->msgbuf may be NULL, e.g. for connection_request or remove_record_request 4945 if (req->msgbuf) freeL("request_state msgbuf", req->msgbuf); 4946 4947 // There's no return data for a cancel request (DNSServiceRefDeallocate returns no result) 4948 // For a DNSServiceGetProperty call, the handler already generated the response, so no need to do it again here 4949 if (req->hdr.op != cancel_request && req->hdr.op != getproperty_request && req->hdr.op != send_bpf && req->hdr.op != getpid_request) 4950 { 4951 const mStatus err_netorder = dnssd_htonl(err); 4952 send_all(req->errsd, (const char *)&err_netorder, sizeof(err_netorder)); 4953 if (req->errsd != req->sd) 4954 { 4955 LogOperation("%3d: Result code socket %d closed %08X %08X (%d)", 4956 req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0], err); 4957 dnssd_close(req->errsd); 4958 req->errsd = req->sd; 4959 // Also need to reset the parent's errsd, if this is a subordinate operation 4960 if (req->primary) req->primary->errsd = req->primary->sd; 4961 } 4962 } 4963 4964 // Reset ready to accept the next req on this pipe 4965 if (req->primary) req = req->primary; 4966 req->ts = t_morecoming; 4967 req->hdr_bytes = 0; 4968 req->data_bytes = 0; 4969 req->msgbuf = mDNSNULL; 4970 req->msgptr = mDNSNULL; 4971 req->msgend = 0; 4972 } 4973 } 4974 4975 mDNSlocal void connect_callback(int fd, short filter, void *info) 4976 { 4977 dnssd_sockaddr_t cliaddr; 4978 dnssd_socklen_t len = (dnssd_socklen_t) sizeof(cliaddr); 4979 dnssd_sock_t sd = accept(fd, (struct sockaddr*) &cliaddr, &len); 4980 #if defined(SO_NOSIGPIPE) || defined(_WIN32) 4981 unsigned int optval = 1; 4982 #endif 4983 4984 (void)filter; // Unused 4985 (void)info; // Unused 4986 4987 if (!dnssd_SocketValid(sd)) 4988 { 4989 if (dnssd_errno != dnssd_EWOULDBLOCK) 4990 my_throttled_perror("ERROR: accept"); 4991 return; 4992 } 4993 4994 #ifdef SO_NOSIGPIPE 4995 // Some environments (e.g. OS X) support turning off SIGPIPE for a socket 4996 if (setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &optval, sizeof(optval)) < 0) 4997 LogMsg("%3d: WARNING: setsockopt - SO_NOSIGPIPE %d (%s)", sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 4998 #endif 4999 5000 #if defined(_WIN32) 5001 if (ioctlsocket(sd, FIONBIO, &optval) != 0) 5002 #else 5003 if (fcntl(sd, F_SETFL, fcntl(sd, F_GETFL, 0) | O_NONBLOCK) != 0) 5004 #endif 5005 { 5006 my_perror("ERROR: fcntl(sd, F_SETFL, O_NONBLOCK) - aborting client"); 5007 dnssd_close(sd); 5008 return; 5009 } 5010 else 5011 { 5012 request_state *request = NewRequest(); 5013 request->ts = t_morecoming; 5014 request->sd = sd; 5015 request->errsd = sd; 5016 set_peer_pid(request); 5017 #if APPLE_OSX_mDNSResponder 5018 struct xucred x; 5019 socklen_t xucredlen = sizeof(x); 5020 if (getsockopt(sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 && x.cr_version == XUCRED_VERSION) 5021 request->uid = x.cr_uid; // save the effective userid of the client 5022 else 5023 my_perror("ERROR: getsockopt, LOCAL_PEERCRED"); 5024 5025 debugf("LOCAL_PEERCRED %d %u %u %d", xucredlen, x.cr_version, x.cr_uid, x.cr_ngroups); 5026 #endif // APPLE_OSX_mDNSResponder 5027 LogOperation("%3d: connect_callback: Adding FD for uid %u", request->sd, request->uid); 5028 udsSupportAddFDToEventLoop(sd, request_callback, request, &request->platform_data); 5029 } 5030 } 5031 5032 mDNSlocal mDNSBool uds_socket_setup(dnssd_sock_t skt) 5033 { 5034 #if defined(SO_NP_EXTENSIONS) 5035 struct so_np_extensions sonpx; 5036 socklen_t optlen = sizeof(struct so_np_extensions); 5037 sonpx.npx_flags = SONPX_SETOPTSHUT; 5038 sonpx.npx_mask = SONPX_SETOPTSHUT; 5039 if (setsockopt(skt, SOL_SOCKET, SO_NP_EXTENSIONS, &sonpx, optlen) < 0) 5040 my_perror("WARNING: could not set sockopt - SO_NP_EXTENSIONS"); 5041 #endif 5042 #if defined(_WIN32) 5043 // SEH: do we even need to do this on windows? 5044 // This socket will be given to WSAEventSelect which will automatically set it to non-blocking 5045 u_long opt = 1; 5046 if (ioctlsocket(skt, FIONBIO, &opt) != 0) 5047 #else 5048 if (fcntl(skt, F_SETFL, fcntl(skt, F_GETFL, 0) | O_NONBLOCK) != 0) 5049 #endif 5050 { 5051 my_perror("ERROR: could not set listen socket to non-blocking mode"); 5052 return mDNSfalse; 5053 } 5054 5055 if (listen(skt, LISTENQ) != 0) 5056 { 5057 my_perror("ERROR: could not listen on listen socket"); 5058 return mDNSfalse; 5059 } 5060 5061 if (mStatus_NoError != udsSupportAddFDToEventLoop(skt, connect_callback, (void *) NULL, (void **) NULL)) 5062 { 5063 my_perror("ERROR: could not add listen socket to event loop"); 5064 return mDNSfalse; 5065 } 5066 else 5067 { 5068 LogOperation("%3d: Listening for incoming Unix Domain Socket client requests", skt); 5069 mDNSStorage.uds_listener_skt = skt; 5070 } 5071 return mDNStrue; 5072 } 5073 5074 mDNSexport int udsserver_init(dnssd_sock_t skts[], mDNSu32 count) 5075 { 5076 dnssd_sockaddr_t laddr; 5077 int ret; 5078 mDNSu32 i = 0; 5079 5080 LogInfo("udsserver_init: %d %d", _DNS_SD_H, mDNSStorage.mDNS_plat); 5081 5082 // If a particular platform wants to opt out of having a PID file, define PID_FILE to be "" 5083 if (PID_FILE[0]) 5084 { 5085 FILE *fp = fopen(PID_FILE, "w"); 5086 if (fp != NULL) 5087 { 5088 fprintf(fp, "%d\n", getpid()); 5089 fclose(fp); 5090 } 5091 } 5092 5093 if (skts) 5094 { 5095 for (i = 0; i < count; i++) 5096 if (dnssd_SocketValid(skts[i]) && !uds_socket_setup(skts[i])) 5097 goto error; 5098 } 5099 else 5100 { 5101 listenfd = socket(AF_DNSSD, SOCK_STREAM, 0); 5102 if (!dnssd_SocketValid(listenfd)) 5103 { 5104 my_perror("ERROR: socket(AF_DNSSD, SOCK_STREAM, 0); failed"); 5105 goto error; 5106 } 5107 5108 mDNSPlatformMemZero(&laddr, sizeof(laddr)); 5109 5110 #if defined(USE_TCP_LOOPBACK) 5111 { 5112 laddr.sin_family = AF_INET; 5113 laddr.sin_port = htons(MDNS_TCP_SERVERPORT); 5114 laddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR); 5115 ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); 5116 if (ret < 0) 5117 { 5118 my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed"); 5119 goto error; 5120 } 5121 } 5122 #else 5123 { 5124 mode_t mask = umask(0); 5125 unlink(boundPath); // OK if this fails 5126 laddr.sun_family = AF_LOCAL; 5127 #ifndef NOT_HAVE_SA_LEN 5128 // According to Stevens (section 3.2), there is no portable way to 5129 // determine whether sa_len is defined on a particular platform. 5130 laddr.sun_len = sizeof(struct sockaddr_un); 5131 #endif 5132 if (strlen(boundPath) >= sizeof(laddr.sun_path)) 5133 { 5134 LogMsg("ERROR: MDNS_UDS_SERVERPATH must be < %d characters", (int)sizeof(laddr.sun_path)); 5135 goto error; 5136 } 5137 mDNSPlatformStrLCopy(laddr.sun_path, boundPath, sizeof(laddr.sun_path)); 5138 ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); 5139 umask(mask); 5140 if (ret < 0) 5141 { 5142 LogMsg("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed `%s': %s", boundPath, strerror(errno)); 5143 goto error; 5144 } 5145 } 5146 #endif 5147 5148 if (!uds_socket_setup(listenfd)) goto error; 5149 } 5150 5151 #if !defined(PLATFORM_NO_RLIMIT) 5152 { 5153 // Set maximum number of open file descriptors 5154 #define MIN_OPENFILES 10240 5155 struct rlimit maxfds, newfds; 5156 5157 // Due to bugs in OS X (<rdar://problem/2941095>, <rdar://problem/3342704>, <rdar://problem/3839173>) 5158 // you have to get and set rlimits once before getrlimit will return sensible values 5159 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 5160 if (setrlimit(RLIMIT_NOFILE, &maxfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit"); 5161 5162 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 5163 newfds.rlim_max = (maxfds.rlim_max > MIN_OPENFILES) ? maxfds.rlim_max : MIN_OPENFILES; 5164 newfds.rlim_cur = (maxfds.rlim_cur > MIN_OPENFILES) ? maxfds.rlim_cur : MIN_OPENFILES; 5165 if (newfds.rlim_max != maxfds.rlim_max || newfds.rlim_cur != maxfds.rlim_cur) 5166 if (setrlimit(RLIMIT_NOFILE, &newfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit"); 5167 5168 if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 5169 debugf("maxfds.rlim_max %d", (long)maxfds.rlim_max); 5170 debugf("maxfds.rlim_cur %d", (long)maxfds.rlim_cur); 5171 } 5172 #endif 5173 5174 // We start a "LocalOnly" query looking for Automatic Browse Domain records. 5175 // When Domain Enumeration in uDNS.c finds an "lb" record from the network, its "FoundDomain" routine 5176 // creates a "LocalOnly" record, which results in our AutomaticBrowseDomainChange callback being invoked 5177 mDNS_GetDomains(&mDNSStorage, &mDNSStorage.AutomaticBrowseDomainQ, mDNS_DomainTypeBrowseAutomatic, 5178 mDNSNULL, mDNSInterface_LocalOnly, AutomaticBrowseDomainChange, mDNSNULL); 5179 5180 // Add "local" as recommended registration domain ("dns-sd -E"), recommended browsing domain ("dns-sd -F"), and automatic browsing domain 5181 RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeRegistration); 5182 RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeBrowse); 5183 AddAutoBrowseDomain(0, &localdomain); 5184 5185 udsserver_handle_configchange(&mDNSStorage); 5186 return 0; 5187 5188 error: 5189 5190 my_perror("ERROR: udsserver_init"); 5191 return -1; 5192 } 5193 5194 mDNSexport int udsserver_exit(void) 5195 { 5196 // Cancel all outstanding client requests 5197 while (all_requests) AbortUnlinkAndFree(all_requests); 5198 5199 // Clean up any special mDNSInterface_LocalOnly records we created, both the entries for "local" we 5200 // created in udsserver_init, and others we created as a result of reading local configuration data 5201 while (LocalDomainEnumRecords) 5202 { 5203 ARListElem *rem = LocalDomainEnumRecords; 5204 LocalDomainEnumRecords = LocalDomainEnumRecords->next; 5205 mDNS_Deregister(&mDNSStorage, &rem->ar); 5206 } 5207 5208 // If the launching environment created no listening socket, 5209 // that means we created it ourselves, so we should clean it up on exit 5210 if (dnssd_SocketValid(listenfd)) 5211 { 5212 dnssd_close(listenfd); 5213 #if !defined(USE_TCP_LOOPBACK) 5214 // Currently, we're unable to remove /var/run/mdnsd because we've changed to userid "nobody" 5215 // to give up unnecessary privilege, but we need to be root to remove this Unix Domain Socket. 5216 // It would be nice if we could find a solution to this problem 5217 if (unlink(boundPath)) 5218 debugf("Unable to remove %s", MDNS_UDS_SERVERPATH); 5219 #endif 5220 } 5221 5222 if (PID_FILE[0]) unlink(PID_FILE); 5223 5224 return 0; 5225 } 5226 5227 mDNSlocal void LogClientInfo(request_state *req) 5228 { 5229 char prefix[16]; 5230 if (req->primary) 5231 mDNS_snprintf(prefix, sizeof(prefix), " -> "); 5232 else 5233 mDNS_snprintf(prefix, sizeof(prefix), "%3d:", req->sd); 5234 5235 if (!req->terminate) 5236 LogMsgNoIdent("%s No operation yet on this socket", prefix); 5237 else if (req->terminate == connection_termination) 5238 { 5239 int num_records = 0, num_ops = 0; 5240 const registered_record_entry *p; 5241 request_state *r; 5242 for (p = req->u.reg_recs; p; p=p->next) num_records++; 5243 for (r = req->next; r; r=r->next) if (r->primary == req) num_ops++; 5244 LogMsgNoIdent("%s DNSServiceCreateConnection: %d registered record%s, %d kDNSServiceFlagsShareConnection operation%s PID[%d](%s)", 5245 prefix, num_records, num_records != 1 ? "s" : "", num_ops, num_ops != 1 ? "s" : "", 5246 req->process_id, req->pid_name); 5247 for (p = req->u.reg_recs; p; p=p->next) 5248 LogMsgNoIdent(" -> DNSServiceRegisterRecord 0x%08X %2d %3d %s PID[%d](%s)", 5249 req->flags, req->interfaceIndex, p->key, ARDisplayString(&mDNSStorage, p->rr), req->process_id, req->pid_name); 5250 for (r = req->next; r; r=r->next) if (r->primary == req) LogClientInfo(r); 5251 } 5252 else if (req->terminate == regservice_termination_callback) 5253 { 5254 service_instance *ptr; 5255 char anonstr[256]; 5256 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 5257 LogMsgNoIdent("%s DNSServiceRegister 0x%08X %2d %##s%s %u/%u PID[%d](%s)", 5258 (ptr == req->u.servicereg.instances) ? prefix : " ", req->flags, req->interfaceIndex, ptr->srs.RR_SRV.resrec.name->c, 5259 AnonDataToString(ptr->srs.AnonData, 0, anonstr, sizeof(anonstr)), mDNSVal16(req->u.servicereg.port), 5260 SRS_PORT(&ptr->srs), req->process_id, req->pid_name); 5261 } 5262 else if (req->terminate == browse_termination_callback) 5263 { 5264 browser_t *blist; 5265 char anonstr[256]; 5266 for (blist = req->u.browser.browsers; blist; blist = blist->next) 5267 LogMsgNoIdent("%s DNSServiceBrowse 0x%08X %2d %##s%s PID[%d](%s)", 5268 (blist == req->u.browser.browsers) ? prefix : " ", req->flags, req->interfaceIndex, blist->q.qname.c, 5269 AnonDataToString(req->u.browser.AnonData, 0, anonstr, sizeof(anonstr)), req->process_id, req->pid_name); 5270 } 5271 else if (req->terminate == resolve_termination_callback) 5272 LogMsgNoIdent("%s DNSServiceResolve 0x%08X %2d %##s PID[%d](%s)", 5273 prefix, req->flags, req->interfaceIndex, req->u.resolve.qsrv.qname.c, req->process_id, req->pid_name); 5274 else if (req->terminate == queryrecord_termination_callback) 5275 LogMsgNoIdent("%s DNSServiceQueryRecord 0x%08X %2d %##s (%s) PID[%d](%s)", 5276 prefix, req->flags, req->interfaceIndex, req->u.queryrecord.q.qname.c, DNSTypeName(req->u.queryrecord.q.qtype), req->process_id, req->pid_name); 5277 else if (req->terminate == enum_termination_callback) 5278 LogMsgNoIdent("%s DNSServiceEnumerateDomains 0x%08X %2d %##s PID[%d](%s)", 5279 prefix, req->flags, req->interfaceIndex, req->u.enumeration.q_all.qname.c, req->process_id, req->pid_name); 5280 else if (req->terminate == port_mapping_termination_callback) 5281 LogMsgNoIdent("%s DNSServiceNATPortMapping 0x%08X %2d %s%s Int %5d Req %5d Ext %.4a:%5d Req TTL %5d Granted TTL %5d PID[%d](%s)", 5282 prefix, 5283 req->flags, 5284 req->interfaceIndex, 5285 req->u.pm.NATinfo.Protocol & NATOp_MapTCP ? "TCP" : " ", 5286 req->u.pm.NATinfo.Protocol & NATOp_MapUDP ? "UDP" : " ", 5287 mDNSVal16(req->u.pm.NATinfo.IntPort), 5288 mDNSVal16(req->u.pm.ReqExt), 5289 &req->u.pm.NATinfo.ExternalAddress, 5290 mDNSVal16(req->u.pm.NATinfo.ExternalPort), 5291 req->u.pm.NATinfo.NATLease, 5292 req->u.pm.NATinfo.Lifetime, 5293 req->process_id, req->pid_name); 5294 else if (req->terminate == addrinfo_termination_callback) 5295 LogMsgNoIdent("%s DNSServiceGetAddrInfo 0x%08X %2d %s%s %##s PID[%d](%s)", 5296 prefix, req->flags, req->interfaceIndex, 5297 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv4 ? "v4" : " ", 5298 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv6 ? "v6" : " ", 5299 req->u.addrinfo.q4.qname.c, req->process_id, req->pid_name); 5300 else 5301 LogMsgNoIdent("%s Unrecognized operation %p", prefix, req->terminate); 5302 } 5303 5304 mDNSlocal void GetMcastClients(request_state *req) 5305 { 5306 if (req->terminate == connection_termination) 5307 { 5308 int num_records = 0, num_ops = 0; 5309 const registered_record_entry *p; 5310 request_state *r; 5311 for (p = req->u.reg_recs; p; p=p->next) 5312 num_records++; 5313 for (r = req->next; r; r=r->next) 5314 if (r->primary == req) 5315 num_ops++; 5316 for (p = req->u.reg_recs; p; p=p->next) 5317 { 5318 if (!AuthRecord_uDNS(p->rr)) 5319 n_mrecords++; 5320 } 5321 for (r = req->next; r; r=r->next) 5322 if (r->primary == req) 5323 GetMcastClients(r); 5324 } 5325 else if (req->terminate == regservice_termination_callback) 5326 { 5327 service_instance *ptr; 5328 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 5329 { 5330 if (!AuthRecord_uDNS(&ptr->srs.RR_SRV)) 5331 n_mrecords++; 5332 } 5333 } 5334 else if (req->terminate == browse_termination_callback) 5335 { 5336 browser_t *blist; 5337 for (blist = req->u.browser.browsers; blist; blist = blist->next) 5338 { 5339 if (mDNSOpaque16IsZero(blist->q.TargetQID)) 5340 n_mquests++; 5341 } 5342 } 5343 else if (req->terminate == resolve_termination_callback) 5344 { 5345 if ((mDNSOpaque16IsZero(req->u.resolve.qsrv.TargetQID)) && (req->u.resolve.qsrv.ThisQInterval > 0)) 5346 n_mquests++; 5347 } 5348 else if (req->terminate == queryrecord_termination_callback) 5349 { 5350 if ((mDNSOpaque16IsZero(req->u.queryrecord.q.TargetQID)) && (req->u.queryrecord.q.ThisQInterval > 0)) 5351 n_mquests++; 5352 } 5353 else if (req->terminate == addrinfo_termination_callback) 5354 { 5355 if ((mDNSOpaque16IsZero(req->u.addrinfo.q4.TargetQID)) && (req->u.addrinfo.q4.ThisQInterval > 0)) 5356 n_mquests++; 5357 } 5358 else 5359 { 5360 return; 5361 } 5362 } 5363 5364 5365 mDNSlocal void LogMcastClientInfo(request_state *req) 5366 { 5367 if (!req->terminate) 5368 LogMcastNoIdent("No operation yet on this socket"); 5369 else if (req->terminate == connection_termination) 5370 { 5371 int num_records = 0, num_ops = 0; 5372 const registered_record_entry *p; 5373 request_state *r; 5374 for (p = req->u.reg_recs; p; p=p->next) 5375 num_records++; 5376 for (r = req->next; r; r=r->next) 5377 if (r->primary == req) 5378 num_ops++; 5379 for (p = req->u.reg_recs; p; p=p->next) 5380 { 5381 if (!AuthRecord_uDNS(p->rr)) 5382 LogMcastNoIdent("R: -> DNSServiceRegisterRecord: %##s %s PID[%d](%s)", p->rr->resrec.name->c, 5383 DNSTypeName(p->rr->resrec.rrtype), req->process_id, req->pid_name, i_mcount++); 5384 } 5385 for (r = req->next; r; r=r->next) 5386 if (r->primary == req) 5387 LogMcastClientInfo(r); 5388 } 5389 else if (req->terminate == regservice_termination_callback) 5390 { 5391 service_instance *ptr; 5392 for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 5393 { 5394 if (!AuthRecord_uDNS(&ptr->srs.RR_SRV)) 5395 LogMcastNoIdent("R: DNSServiceRegister: %##s %u/%u PID[%d](%s)", ptr->srs.RR_SRV.resrec.name->c, mDNSVal16(req->u.servicereg.port), 5396 SRS_PORT(&ptr->srs), req->process_id, req->pid_name, i_mcount++); 5397 } 5398 } 5399 else if (req->terminate == browse_termination_callback) 5400 { 5401 browser_t *blist; 5402 for (blist = req->u.browser.browsers; blist; blist = blist->next) 5403 { 5404 if (mDNSOpaque16IsZero(blist->q.TargetQID)) 5405 LogMcastNoIdent("Q: DNSServiceBrowse %##s %s PID[%d](%s)", blist->q.qname.c, DNSTypeName(blist->q.qtype), 5406 req->process_id, req->pid_name, i_mcount++); 5407 } 5408 } 5409 else if (req->terminate == resolve_termination_callback) 5410 { 5411 if ((mDNSOpaque16IsZero(req->u.resolve.qsrv.TargetQID)) && (req->u.resolve.qsrv.ThisQInterval > 0)) 5412 LogMcastNoIdent("Q: DNSServiceResolve %##s %s PID[%d](%s)", req->u.resolve.qsrv.qname.c, DNSTypeName(req->u.resolve.qsrv.qtype), 5413 req->process_id, req->pid_name, i_mcount++); 5414 } 5415 else if (req->terminate == queryrecord_termination_callback) 5416 { 5417 if ((mDNSOpaque16IsZero(req->u.queryrecord.q.TargetQID)) && (req->u.queryrecord.q.ThisQInterval > 0)) 5418 LogMcastNoIdent("Q: DNSServiceQueryRecord %##s %s PID[%d](%s)", req->u.queryrecord.q.qname.c, DNSTypeName(req->u.queryrecord.q.qtype), 5419 req->process_id, req->pid_name, i_mcount++); 5420 } 5421 else if (req->terminate == addrinfo_termination_callback) 5422 { 5423 if ((mDNSOpaque16IsZero(req->u.addrinfo.q4.TargetQID)) && (req->u.addrinfo.q4.ThisQInterval > 0)) 5424 LogMcastNoIdent("Q: DNSServiceGetAddrInfo %s%s %##s PID[%d](%s)", 5425 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv4 ? "v4" : " ", 5426 req->u.addrinfo.protocol & kDNSServiceProtocol_IPv6 ? "v6" : " ", 5427 req->u.addrinfo.q4.qname.c, req->process_id, req->pid_name, i_mcount++); 5428 } 5429 else 5430 { 5431 return; 5432 } 5433 5434 } 5435 5436 mDNSlocal char *RecordTypeName(mDNSu8 rtype) 5437 { 5438 switch (rtype) 5439 { 5440 case kDNSRecordTypeUnregistered: return ("Unregistered "); 5441 case kDNSRecordTypeDeregistering: return ("Deregistering"); 5442 case kDNSRecordTypeUnique: return ("Unique "); 5443 case kDNSRecordTypeAdvisory: return ("Advisory "); 5444 case kDNSRecordTypeShared: return ("Shared "); 5445 case kDNSRecordTypeVerified: return ("Verified "); 5446 case kDNSRecordTypeKnownUnique: return ("KnownUnique "); 5447 default: return("Unknown"); 5448 } 5449 } 5450 5451 mDNSlocal int LogEtcHosts(mDNS *const m) 5452 { 5453 mDNSBool showheader = mDNStrue; 5454 const AuthRecord *ar; 5455 mDNSu32 slot; 5456 AuthGroup *ag; 5457 int count = 0; 5458 int authslot = 0; 5459 mDNSBool truncated = 0; 5460 5461 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++) 5462 { 5463 if (m->rrauth.rrauth_hash[slot]) authslot++; 5464 for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next) 5465 for (ar = ag->members; ar; ar = ar->next) 5466 { 5467 if (ar->RecordCallback != FreeEtcHosts) continue; 5468 if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); } 5469 5470 // Print a maximum of 50 records 5471 if (count++ >= 50) { truncated = mDNStrue; continue; } 5472 if (ar->ARType == AuthRecordLocalOnly) 5473 { 5474 if (ar->resrec.InterfaceID == mDNSInterface_LocalOnly) 5475 LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 5476 else 5477 { 5478 mDNSu32 scopeid = (mDNSu32)(uintptr_t)ar->resrec.InterfaceID; 5479 LogMsgNoIdent(" %s %u %s", RecordTypeName(ar->resrec.RecordType), scopeid, ARDisplayString(m, ar)); 5480 } 5481 } 5482 } 5483 } 5484 5485 if (showheader) LogMsgNoIdent("<None>"); 5486 else if (truncated) LogMsgNoIdent("<Truncated: to 50 records, Total records %d, Total Auth Groups %d, Auth Slots %d>", count, m->rrauth.rrauth_totalused, authslot); 5487 return count; 5488 } 5489 5490 mDNSlocal void LogLocalOnlyAuthRecords(mDNS *const m) 5491 { 5492 mDNSBool showheader = mDNStrue; 5493 const AuthRecord *ar; 5494 mDNSu32 slot; 5495 AuthGroup *ag; 5496 5497 for (slot = 0; slot < AUTH_HASH_SLOTS; slot++) 5498 { 5499 for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next) 5500 for (ar = ag->members; ar; ar = ar->next) 5501 { 5502 if (ar->RecordCallback == FreeEtcHosts) continue; 5503 if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); } 5504 5505 // Print a maximum of 400 records 5506 if (ar->ARType == AuthRecordLocalOnly) 5507 LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 5508 else if (ar->ARType == AuthRecordP2P) 5509 { 5510 if (ar->resrec.InterfaceID == mDNSInterface_BLE) 5511 LogMsgNoIdent(" %s BLE %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 5512 else 5513 LogMsgNoIdent(" %s PP %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 5514 } 5515 } 5516 } 5517 5518 if (showheader) LogMsgNoIdent("<None>"); 5519 } 5520 5521 mDNSlocal char *AnonInfoToString(AnonymousInfo *ai, char *anonstr, int anstrlen) 5522 { 5523 anonstr[0] = 0; 5524 if (ai && ai->AnonData) 5525 { 5526 return (AnonDataToString(ai->AnonData, ai->AnonDataLen, anonstr, anstrlen)); 5527 } 5528 return anonstr; 5529 } 5530 5531 mDNSlocal void LogOneAuthRecord(const AuthRecord *ar, mDNSs32 now, const char *const ifname) 5532 { 5533 char anstr[256]; 5534 if (AuthRecord_uDNS(ar)) 5535 { 5536 LogMsgNoIdent("%7d %7d %7d %-7s %4d %s %s", 5537 ar->ThisAPInterval / mDNSPlatformOneSecond, 5538 (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond, 5539 ar->expire ? (ar->expire - now) / mDNSPlatformOneSecond : 0, 5540 "-U-", 5541 ar->state, 5542 ar->AllowRemoteQuery ? "☠" : " ", 5543 ARDisplayString(&mDNSStorage, ar)); 5544 } 5545 else 5546 { 5547 LogMsgNoIdent("%7d %7d %7d %-7s 0x%02X %s %s%s", 5548 ar->ThisAPInterval / mDNSPlatformOneSecond, 5549 ar->AnnounceCount ? (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond : 0, 5550 ar->TimeExpire ? (ar->TimeExpire - now) / mDNSPlatformOneSecond : 0, 5551 ifname ? ifname : "ALL", 5552 ar->resrec.RecordType, 5553 ar->AllowRemoteQuery ? "☠" : " ", 5554 ARDisplayString(&mDNSStorage, ar), AnonInfoToString(ar->resrec.AnonInfo, anstr, sizeof(anstr))); 5555 } 5556 } 5557 5558 mDNSlocal void LogAuthRecords(const mDNSs32 now, AuthRecord *ResourceRecords, int *proxy) 5559 { 5560 mDNSBool showheader = mDNStrue; 5561 const AuthRecord *ar; 5562 OwnerOptData owner = zeroOwner; 5563 for (ar = ResourceRecords; ar; ar=ar->next) 5564 { 5565 const char *const ifname = InterfaceNameForID(&mDNSStorage, ar->resrec.InterfaceID); 5566 if ((ar->WakeUp.HMAC.l[0] != 0) == (proxy != mDNSNULL)) 5567 { 5568 if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" Int Next Expire if State"); } 5569 if (proxy) (*proxy)++; 5570 if (!mDNSPlatformMemSame(&owner, &ar->WakeUp, sizeof(owner))) 5571 { 5572 owner = ar->WakeUp; 5573 if (owner.password.l[0]) 5574 LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a Password %.6a seq %d", &owner.HMAC, &owner.IMAC, &owner.password, owner.seq); 5575 else if (!mDNSSameEthAddress(&owner.HMAC, &owner.IMAC)) 5576 LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a seq %d", &owner.HMAC, &owner.IMAC, owner.seq); 5577 else 5578 LogMsgNoIdent("Proxying for %.6a seq %d", &owner.HMAC, owner.seq); 5579 } 5580 if (AuthRecord_uDNS(ar)) 5581 { 5582 LogOneAuthRecord(ar, now, ifname); 5583 } 5584 else if (ar->ARType == AuthRecordLocalOnly) 5585 { 5586 LogMsgNoIdent(" LO %s", ARDisplayString(&mDNSStorage, ar)); 5587 } 5588 else if (ar->ARType == AuthRecordP2P) 5589 { 5590 if (ar->resrec.InterfaceID == mDNSInterface_BLE) 5591 LogMsgNoIdent(" BLE %s", ARDisplayString(&mDNSStorage, ar)); 5592 else 5593 LogMsgNoIdent(" PP %s", ARDisplayString(&mDNSStorage, ar)); 5594 } 5595 else 5596 { 5597 LogOneAuthRecord(ar, now, ifname); 5598 if (ar->resrec.AnonInfo) 5599 { 5600 ResourceRecord *nsec3 = ar->resrec.AnonInfo->nsec3RR; 5601 // We just print the values from the AuthRecord to keep it nicely aligned though 5602 // all we want here is the nsec3 information. 5603 LogMsgNoIdent("%7d %7d %7d %7s %s", 5604 ar->ThisAPInterval / mDNSPlatformOneSecond, 5605 ar->AnnounceCount ? (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond : 0, 5606 ar->TimeExpire ? (ar->TimeExpire - now) / mDNSPlatformOneSecond : 0, 5607 ifname ? ifname : "ALL", 5608 RRDisplayString(&mDNSStorage, nsec3)); 5609 } 5610 } 5611 } 5612 } 5613 if (showheader) LogMsgNoIdent("<None>"); 5614 } 5615 5616 mDNSlocal void PrintOneCacheRecord(const CacheRecord *cr, mDNSu32 slot, const mDNSu32 remain, const char *ifname, mDNSu32 *CacheUsed) 5617 { 5618 LogMsgNoIdent("%3d %s%8d %-7s%s %-6s%s", 5619 slot, 5620 cr->CRActiveQuestion ? "*" : " ", 5621 remain, 5622 ifname ? ifname : "-U-", 5623 (cr->resrec.RecordType == kDNSRecordTypePacketNegative) ? "-" : 5624 (cr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) ? " " : "+", 5625 DNSTypeName(cr->resrec.rrtype), 5626 CRDisplayString(&mDNSStorage, cr)); 5627 (*CacheUsed)++; 5628 } 5629 5630 mDNSlocal void PrintCachedRecords(const CacheRecord *cr, mDNSu32 slot, const mDNSu32 remain, const char *ifname, mDNSu32 *CacheUsed) 5631 { 5632 CacheRecord *nsec; 5633 CacheRecord *soa; 5634 nsec = cr->nsec; 5635 5636 // The records that are cached under the main cache record like nsec, soa don't have 5637 // their own lifetime. If the main cache record expires, they also expire. 5638 while (nsec) 5639 { 5640 PrintOneCacheRecord(nsec, slot, remain, ifname, CacheUsed); 5641 nsec = nsec->next; 5642 } 5643 soa = cr->soa; 5644 if (soa) 5645 { 5646 PrintOneCacheRecord(soa, slot, remain, ifname, CacheUsed); 5647 } 5648 if (cr->resrec.AnonInfo) 5649 { 5650 ResourceRecord *nsec3 = cr->resrec.AnonInfo->nsec3RR; 5651 // Even though it is a resource record, we print the sameway 5652 // as a cache record so that it aligns properly. 5653 if (nsec3) 5654 { 5655 LogMsgNoIdent("%3d %s%8d %-7s%s %-6s%s", 5656 slot, 5657 " ", 5658 remain, 5659 ifname ? ifname : "-U-", 5660 (nsec3->RecordType == kDNSRecordTypePacketNegative) ? "-" : 5661 (nsec3->RecordType & kDNSRecordTypePacketUniqueMask) ? " " : "+", 5662 DNSTypeName(nsec3->rrtype), 5663 RRDisplayString(&mDNSStorage, nsec3)); 5664 } 5665 } 5666 } 5667 5668 mDNSlocal char *AnonDataToString(const mDNSu8 *ad, int adlen, char *adstr, int adstrlen) 5669 { 5670 adstr[0] = 0; 5671 if (ad) 5672 { 5673 int len; 5674 char *orig = adstr; 5675 5676 // If the caller is lazy to compute the length, we do it for them. 5677 if (!adlen) 5678 len = strlen((const char *)ad); 5679 else 5680 len = adlen; 5681 5682 // Print the anondata within brackets. Hence, we need space for two 5683 // brackets and a NULL byte. 5684 if (len > (adstrlen - 3)) 5685 len = adstrlen - 3; 5686 5687 *adstr++ = '('; 5688 mDNSPlatformMemCopy(adstr, ad, len); 5689 adstr[len] = ')'; 5690 adstr[len+1] = 0; 5691 return orig; 5692 } 5693 return adstr; 5694 } 5695 5696 mDNSexport void LogMDNSStatistics(mDNS *const m) 5697 { 5698 LogMsgNoIdent("--- MDNS Statistics ---"); 5699 5700 LogMsgNoIdent("Name Conflicts %u", m->mDNSStats.NameConflicts); 5701 LogMsgNoIdent("KnownUnique Name Conflicts %u", m->mDNSStats.KnownUniqueNameConflicts); 5702 LogMsgNoIdent("Duplicate Query Suppressions %u", m->mDNSStats.DupQuerySuppressions); 5703 LogMsgNoIdent("KA Suppressions %u", m->mDNSStats.KnownAnswerSuppressions); 5704 LogMsgNoIdent("KA Multiple Packets %u", m->mDNSStats.KnownAnswerMultiplePkts); 5705 LogMsgNoIdent("Poof Cache Deletions %u", m->mDNSStats.PoofCacheDeletions); 5706 LogMsgNoIdent("--------------------------------"); 5707 5708 LogMsgNoIdent("Multicast packets Sent %u", m->MulticastPacketsSent); 5709 LogMsgNoIdent("Multicast packets Received %u", m->MPktNum); 5710 LogMsgNoIdent("Remote Subnet packets %u", m->RemoteSubnet); 5711 LogMsgNoIdent("QU questions received %u", m->mDNSStats.UnicastBitInQueries); 5712 LogMsgNoIdent("Normal multicast questions %u", m->mDNSStats.NormalQueries); 5713 LogMsgNoIdent("Answers for questions %u", m->mDNSStats.MatchingAnswersForQueries); 5714 LogMsgNoIdent("Unicast responses %u", m->mDNSStats.UnicastResponses); 5715 LogMsgNoIdent("Multicast responses %u", m->mDNSStats.MulticastResponses); 5716 LogMsgNoIdent("Unicast response Demotions %u", m->mDNSStats.UnicastDemotedToMulticast); 5717 LogMsgNoIdent("--------------------------------"); 5718 5719 LogMsgNoIdent("Sleeps %u", m->mDNSStats.Sleeps); 5720 LogMsgNoIdent("Wakeups %u", m->mDNSStats.Wakes); 5721 LogMsgNoIdent("Interface UP events %u", m->mDNSStats.InterfaceUp); 5722 LogMsgNoIdent("Interface UP Flap events %u", m->mDNSStats.InterfaceUpFlap); 5723 LogMsgNoIdent("Interface Down events %u", m->mDNSStats.InterfaceDown); 5724 LogMsgNoIdent("Interface DownFlap events %u", m->mDNSStats.InterfaceDownFlap); 5725 LogMsgNoIdent("Cache refresh queries %u", m->mDNSStats.CacheRefreshQueries); 5726 LogMsgNoIdent("Cache refreshed %u", m->mDNSStats.CacheRefreshed); 5727 LogMsgNoIdent("Wakeup on Resolves %u", m->mDNSStats.WakeOnResolves); 5728 } 5729 5730 mDNSexport void udsserver_info() 5731 { 5732 mDNS *const m = &mDNSStorage; 5733 const mDNSs32 now = mDNS_TimeNow(m); 5734 mDNSu32 CacheUsed = 0, CacheActive = 0, slot; 5735 int ProxyA = 0, ProxyD = 0; 5736 mDNSu32 groupCount = 0; 5737 mDNSu32 mcastRecordCount = 0; 5738 mDNSu32 ucastRecordCount = 0; 5739 const CacheGroup *cg; 5740 const CacheRecord *cr; 5741 const DNSQuestion *q; 5742 const DNameListElem *d; 5743 const SearchListElem *s; 5744 5745 LogMsgNoIdent("Timenow 0x%08lX (%d)", (mDNSu32)now, now); 5746 5747 LogMsgNoIdent("------------ Cache -------------"); 5748 LogMsgNoIdent("Slt Q TTL if U Type rdlen"); 5749 for (slot = 0; slot < CACHE_HASH_SLOTS; slot++) 5750 { 5751 for (cg = m->rrcache_hash[slot]; cg; cg=cg->next) 5752 { 5753 groupCount++; // Count one cache entity for the CacheGroup object 5754 for (cr = cg->members; cr; cr=cr->next) 5755 { 5756 const mDNSs32 remain = cr->resrec.rroriginalttl - (now - cr->TimeRcvd) / mDNSPlatformOneSecond; 5757 const char *ifname; 5758 mDNSInterfaceID InterfaceID = cr->resrec.InterfaceID; 5759 mDNSu32 *const countPtr = InterfaceID ? &mcastRecordCount : &ucastRecordCount; 5760 if (!InterfaceID && cr->resrec.rDNSServer && cr->resrec.rDNSServer->scoped) 5761 InterfaceID = cr->resrec.rDNSServer->interface; 5762 ifname = InterfaceNameForID(m, InterfaceID); 5763 if (cr->CRActiveQuestion) CacheActive++; 5764 PrintOneCacheRecord(cr, slot, remain, ifname, countPtr); 5765 PrintCachedRecords(cr, slot, remain, ifname, countPtr); 5766 } 5767 } 5768 } 5769 5770 CacheUsed = groupCount + mcastRecordCount + ucastRecordCount; 5771 if (m->rrcache_totalused != CacheUsed) 5772 LogMsgNoIdent("Cache use mismatch: rrcache_totalused is %lu, true count %lu", m->rrcache_totalused, CacheUsed); 5773 if (m->rrcache_active != CacheActive) 5774 LogMsgNoIdent("Cache use mismatch: rrcache_active is %lu, true count %lu", m->rrcache_active, CacheActive); 5775 LogMsgNoIdent("Cache size %u entities; %u in use (%u group, %u multicast, %u unicast); %u referenced by active questions", 5776 m->rrcache_size, CacheUsed, groupCount, mcastRecordCount, ucastRecordCount, CacheActive); 5777 5778 LogMsgNoIdent("--------- Auth Records ---------"); 5779 LogAuthRecords(now, m->ResourceRecords, mDNSNULL); 5780 5781 LogMsgNoIdent("--------- LocalOnly, P2P Auth Records ---------"); 5782 LogLocalOnlyAuthRecords(m); 5783 5784 LogMsgNoIdent("--------- /etc/hosts ---------"); 5785 LogEtcHosts(m); 5786 5787 LogMsgNoIdent("------ Duplicate Records -------"); 5788 LogAuthRecords(now, m->DuplicateRecords, mDNSNULL); 5789 5790 LogMsgNoIdent("----- Auth Records Proxied -----"); 5791 LogAuthRecords(now, m->ResourceRecords, &ProxyA); 5792 5793 LogMsgNoIdent("-- Duplicate Records Proxied ---"); 5794 LogAuthRecords(now, m->DuplicateRecords, &ProxyD); 5795 5796 LogMsgNoIdent("---------- Questions -----------"); 5797 if (!m->Questions) LogMsgNoIdent("<None>"); 5798 else 5799 { 5800 char anonstr[256]; 5801 CacheUsed = 0; 5802 CacheActive = 0; 5803 LogMsgNoIdent(" Int Next if T NumAns VDNS Qptr DupOf SU SQ Type Name"); 5804 for (q = m->Questions; q; q=q->next) 5805 { 5806 mDNSs32 i = q->ThisQInterval / mDNSPlatformOneSecond; 5807 mDNSs32 n = (NextQSendTime(q) - now) / mDNSPlatformOneSecond; 5808 char *ifname = InterfaceNameForID(m, q->InterfaceID); 5809 CacheUsed++; 5810 if (q->ThisQInterval) CacheActive++; 5811 LogMsgNoIdent("%6d%6d %-7s%s%s %5d 0x%x%x%x%x 0x%p 0x%p %1d %2d %-5s%##s%s%s", 5812 i, n, 5813 ifname ? ifname : mDNSOpaque16IsZero(q->TargetQID) ? "" : "-U-", 5814 mDNSOpaque16IsZero(q->TargetQID) ? (q->LongLived ? "l" : " ") : (q->LongLived ? "L" : "O"), 5815 PrivateQuery(q) ? "P" : q->ValidationRequired ? "V" : q->ValidatingResponse ? "R" : " ", 5816 q->CurrentAnswers, q->validDNSServers.l[3], q->validDNSServers.l[2], q->validDNSServers.l[1], 5817 q->validDNSServers.l[0], q, q->DuplicateOf, 5818 q->SuppressUnusable, q->SuppressQuery, DNSTypeName(q->qtype), q->qname.c, 5819 AnonInfoToString(q->AnonInfo, anonstr, sizeof(anonstr)), 5820 q->DuplicateOf ? " (dup)" : ""); 5821 } 5822 LogMsgNoIdent("%lu question%s; %lu active", CacheUsed, CacheUsed > 1 ? "s" : "", CacheActive); 5823 } 5824 5825 LogMsgNoIdent("----- LocalOnly, P2P Questions -----"); 5826 if (!m->LocalOnlyQuestions) LogMsgNoIdent("<None>"); 5827 else for (q = m->LocalOnlyQuestions; q; q=q->next) 5828 LogMsgNoIdent(" %3s %5d %-6s%##s%s", 5829 q->InterfaceID == mDNSInterface_LocalOnly ? "LO ": q->InterfaceID == mDNSInterface_BLE ? "BLE": "P2P", 5830 q->CurrentAnswers, DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : ""); 5831 5832 LogMsgNoIdent("---- Active UDS Client Requests ----"); 5833 if (!all_requests) LogMsgNoIdent("<None>"); 5834 else 5835 { 5836 request_state *req, *r; 5837 for (req = all_requests; req; req=req->next) 5838 { 5839 if (req->primary) // If this is a subbordinate operation, check that the parent is in the list 5840 { 5841 for (r = all_requests; r && r != req; r=r->next) if (r == req->primary) goto foundparent; 5842 LogMsgNoIdent("%3d: Orhpan operation %p; parent %p not found in request list", req->sd); 5843 } 5844 // For non-subbordinate operations, and subbordinate operations that have lost their parent, write out their info 5845 LogClientInfo(req); 5846 foundparent:; 5847 } 5848 } 5849 5850 LogMsgNoIdent("-------- NAT Traversals --------"); 5851 LogMsgNoIdent("ExtAddress %.4a Retry %d Interval %d", 5852 &m->ExtAddress, 5853 m->retryGetAddr ? (m->retryGetAddr - now) / mDNSPlatformOneSecond : 0, 5854 m->retryIntervalGetAddr / mDNSPlatformOneSecond); 5855 if (m->NATTraversals) 5856 { 5857 const NATTraversalInfo *nat; 5858 for (nat = m->NATTraversals; nat; nat=nat->next) 5859 { 5860 LogMsgNoIdent("%p %s Int %5d %s Err %d Retry %5d Interval %5d Expire %5d Req %.4a:%d Ext %.4a:%d", 5861 nat, 5862 nat->Protocol ? (nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP") : "ADD", 5863 mDNSVal16(nat->IntPort), 5864 (nat->lastSuccessfulProtocol == NATTProtocolNone ? "None " : 5865 nat->lastSuccessfulProtocol == NATTProtocolNATPMP ? "NAT-PMP " : 5866 nat->lastSuccessfulProtocol == NATTProtocolUPNPIGD ? "UPnP/IGD" : 5867 nat->lastSuccessfulProtocol == NATTProtocolPCP ? "PCP " : 5868 /* else */ "Unknown " ), 5869 nat->Result, 5870 nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0, 5871 nat->retryInterval / mDNSPlatformOneSecond, 5872 nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0, 5873 &nat->NewAddress, mDNSVal16(nat->RequestedPort), 5874 &nat->ExternalAddress, mDNSVal16(nat->ExternalPort)); 5875 } 5876 } 5877 5878 LogMsgNoIdent("--------- AuthInfoList ---------"); 5879 if (!m->AuthInfoList) LogMsgNoIdent("<None>"); 5880 else 5881 { 5882 const DomainAuthInfo *a; 5883 for (a = m->AuthInfoList; a; a = a->next) 5884 { 5885 LogMsgNoIdent("%##s %##s %##s %d %d %.16a%s", 5886 a->domain.c, a->keyname.c, 5887 a->hostname.c, (a->port.b[0] << 8 | a->port.b[1]), 5888 (a->deltime ? (a->deltime - now) : 0), 5889 &a->AutoTunnelInnerAddress, a->AutoTunnel ? " AutoTunnel" : ""); 5890 } 5891 } 5892 5893 #if APPLE_OSX_mDNSResponder 5894 LogMsgNoIdent("--------- TunnelClients --------"); 5895 if (!m->TunnelClients) LogMsgNoIdent("<None>"); 5896 else 5897 { 5898 const ClientTunnel *c; 5899 for (c = m->TunnelClients; c; c = c->next) 5900 LogMsgNoIdent("%##s local %.16a %.4a %.16a remote %.16a %.4a %5d %.16a interval %d", 5901 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); 5902 } 5903 #endif // APPLE_OSX_mDNSResponder 5904 5905 LogMsgNoIdent("---------- Misc State ----------"); 5906 5907 LogMsgNoIdent("PrimaryMAC: %.6a", &m->PrimaryMAC); 5908 5909 LogMsgNoIdent("m->SleepState %d (%s) seq %d", 5910 m->SleepState, 5911 m->SleepState == SleepState_Awake ? "Awake" : 5912 m->SleepState == SleepState_Transferring ? "Transferring" : 5913 m->SleepState == SleepState_Sleeping ? "Sleeping" : "?", 5914 m->SleepSeqNum); 5915 5916 if (!m->SPSSocket) LogMsgNoIdent("Not offering Sleep Proxy Service"); 5917 #ifndef SPC_DISABLED 5918 else LogMsgNoIdent("Offering Sleep Proxy Service: %#s", m->SPSRecords.RR_SRV.resrec.name->c); 5919 #endif 5920 if (m->ProxyRecords == ProxyA + ProxyD) LogMsgNoIdent("ProxyRecords: %d + %d = %d", ProxyA, ProxyD, ProxyA + ProxyD); 5921 else LogMsgNoIdent("ProxyRecords: MISMATCH %d + %d = %d ≠ %d", ProxyA, ProxyD, ProxyA + ProxyD, m->ProxyRecords); 5922 5923 LogMsgNoIdent("------ Auto Browse Domains -----"); 5924 if (!AutoBrowseDomains) LogMsgNoIdent("<None>"); 5925 else for (d=AutoBrowseDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c); 5926 5927 LogMsgNoIdent("--- Auto Registration Domains --"); 5928 if (!AutoRegistrationDomains) LogMsgNoIdent("<None>"); 5929 else for (d=AutoRegistrationDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c); 5930 5931 LogMsgNoIdent("--- Search Domains --"); 5932 if (!SearchList) LogMsgNoIdent("<None>"); 5933 else 5934 { 5935 for (s=SearchList; s; s=s->next) 5936 { 5937 char *ifname = InterfaceNameForID(m, s->InterfaceID); 5938 LogMsgNoIdent("%##s %s", s->domain.c, ifname ? ifname : ""); 5939 } 5940 } 5941 LogInfo("--- Trust Anchors ---"); 5942 if (!m->TrustAnchors) 5943 { 5944 LogInfo("<None>"); 5945 } 5946 else 5947 { 5948 TrustAnchor *ta; 5949 mDNSu8 fromTimeBuf[64]; 5950 mDNSu8 untilTimeBuf[64]; 5951 5952 for (ta=m->TrustAnchors; ta; ta=ta->next) 5953 { 5954 mDNSPlatformFormatTime((unsigned long)ta->validFrom, fromTimeBuf, sizeof(fromTimeBuf)); 5955 mDNSPlatformFormatTime((unsigned long)ta->validUntil, untilTimeBuf, sizeof(untilTimeBuf)); 5956 LogInfo("%##s %d %d %d %d %s %s", ta->zone.c, ta->rds.keyTag, 5957 ta->rds.alg, ta->rds.digestType, ta->digestLen, fromTimeBuf, untilTimeBuf); 5958 } 5959 } 5960 5961 LogInfo("--- DNSSEC Statistics ---"); 5962 5963 LogMsgNoIdent("Unicast Cache size %u", m->rrcache_totalused_unicast); 5964 LogInfo("DNSSEC Cache size %u", m->DNSSECStats.TotalMemUsed); 5965 if (m->rrcache_totalused_unicast) 5966 LogInfo("DNSSEC usage percentage %u", ((unsigned long)(m->DNSSECStats.TotalMemUsed * 100))/m->rrcache_totalused_unicast); 5967 LogInfo("DNSSEC Extra Packets (0 to 2) %u", m->DNSSECStats.ExtraPackets0); 5968 LogInfo("DNSSEC Extra Packets (3 to 6) %u", m->DNSSECStats.ExtraPackets3); 5969 LogInfo("DNSSEC Extra Packets (7 to 9) %u", m->DNSSECStats.ExtraPackets7); 5970 LogInfo("DNSSEC Extra Packets ( >= 10) %u", m->DNSSECStats.ExtraPackets10); 5971 5972 LogInfo("DNSSEC Latency (0 to 4ms) %u", m->DNSSECStats.Latency0); 5973 LogInfo("DNSSEC Latency (4 to 9ms) %u", m->DNSSECStats.Latency5); 5974 LogInfo("DNSSEC Latency (10 to 19ms) %u", m->DNSSECStats.Latency10); 5975 LogInfo("DNSSEC Latency (20 to 49ms) %u", m->DNSSECStats.Latency20); 5976 LogInfo("DNSSEC Latency (50 to 99ms) %u", m->DNSSECStats.Latency50); 5977 LogInfo("DNSSEC Latency ( >=100ms) %u", m->DNSSECStats.Latency100); 5978 5979 LogInfo("DNSSEC Secure Status %u", m->DNSSECStats.SecureStatus); 5980 LogInfo("DNSSEC Insecure Status %u", m->DNSSECStats.InsecureStatus); 5981 LogInfo("DNSSEC Indeterminate Status %u", m->DNSSECStats.IndeterminateStatus); 5982 LogInfo("DNSSEC Bogus Status %u", m->DNSSECStats.BogusStatus); 5983 LogInfo("DNSSEC NoResponse Status %u", m->DNSSECStats.NoResponseStatus); 5984 LogInfo("DNSSEC Probes sent %u", m->DNSSECStats.NumProbesSent); 5985 LogInfo("DNSSEC Msg Size (<=1024) %u", m->DNSSECStats.MsgSize0); 5986 LogInfo("DNSSEC Msg Size (<=2048) %u", m->DNSSECStats.MsgSize1); 5987 LogInfo("DNSSEC Msg Size (> 2048) %u", m->DNSSECStats.MsgSize2); 5988 5989 LogMDNSStatistics(m); 5990 5991 LogMsgNoIdent("---- Task Scheduling Timers ----"); 5992 5993 #if BONJOUR_ON_DEMAND 5994 LogMsgNoIdent("BonjourEnabled %d", m->BonjourEnabled); 5995 #endif // BONJOUR_ON_DEMAND 5996 5997 #if APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 5998 LogMsgNoIdent("EnableBLEBasedDiscovery %d", EnableBLEBasedDiscovery); 5999 LogMsgNoIdent("DefaultToBLETriggered %d", DefaultToBLETriggered); 6000 #endif // APPLE_OSX_mDNSResponder && ENABLE_BLE_TRIGGERED_BONJOUR 6001 6002 if (!m->NewQuestions) 6003 LogMsgNoIdent("NewQuestion <NONE>"); 6004 else 6005 LogMsgNoIdent("NewQuestion DelayAnswering %d %d %##s (%s)", 6006 m->NewQuestions->DelayAnswering, m->NewQuestions->DelayAnswering-now, 6007 m->NewQuestions->qname.c, DNSTypeName(m->NewQuestions->qtype)); 6008 6009 if (!m->NewLocalOnlyQuestions) 6010 LogMsgNoIdent("NewLocalOnlyQuestions <NONE>"); 6011 else 6012 LogMsgNoIdent("NewLocalOnlyQuestions %##s (%s)", 6013 m->NewLocalOnlyQuestions->qname.c, DNSTypeName(m->NewLocalOnlyQuestions->qtype)); 6014 6015 if (!m->NewLocalRecords) 6016 LogMsgNoIdent("NewLocalRecords <NONE>"); 6017 else 6018 LogMsgNoIdent("NewLocalRecords %02X %s", m->NewLocalRecords->resrec.RecordType, ARDisplayString(m, m->NewLocalRecords)); 6019 6020 LogMsgNoIdent("SPSProxyListChanged%s", m->SPSProxyListChanged ? "" : " <NONE>"); 6021 LogMsgNoIdent("LocalRemoveEvents%s", m->LocalRemoveEvents ? "" : " <NONE>"); 6022 LogMsgNoIdent("m->AutoTunnelRelayAddr %.16a", &m->AutoTunnelRelayAddr); 6023 LogMsgNoIdent("m->WABBrowseQueriesCount %d", m->WABBrowseQueriesCount); 6024 LogMsgNoIdent("m->WABLBrowseQueriesCount %d", m->WABLBrowseQueriesCount); 6025 LogMsgNoIdent("m->WABRegQueriesCount %d", m->WABRegQueriesCount); 6026 LogMsgNoIdent("m->AutoTargetServices %d", m->AutoTargetServices); 6027 6028 LogMsgNoIdent(" ABS (hex) ABS (dec) REL (hex) REL (dec)"); 6029 LogMsgNoIdent("m->timenow %08X %11d", now, now); 6030 LogMsgNoIdent("m->timenow_adjust %08X %11d", m->timenow_adjust, m->timenow_adjust); 6031 LogTimer("m->NextScheduledEvent ", m->NextScheduledEvent); 6032 6033 #ifndef UNICAST_DISABLED 6034 LogTimer("m->NextuDNSEvent ", m->NextuDNSEvent); 6035 LogTimer("m->NextSRVUpdate ", m->NextSRVUpdate); 6036 LogTimer("m->NextScheduledNATOp ", m->NextScheduledNATOp); 6037 LogTimer("m->retryGetAddr ", m->retryGetAddr); 6038 #endif 6039 6040 LogTimer("m->NextCacheCheck ", m->NextCacheCheck); 6041 LogTimer("m->NextScheduledSPS ", m->NextScheduledSPS); 6042 LogTimer("m->NextScheduledKA ", m->NextScheduledKA); 6043 6044 #if BONJOUR_ON_DEMAND 6045 LogTimer("m->NextBonjourDisableTime ", m->NextBonjourDisableTime); 6046 #endif // BONJOUR_ON_DEMAND 6047 6048 LogTimer("m->NextScheduledSPRetry ", m->NextScheduledSPRetry); 6049 LogTimer("m->DelaySleep ", m->DelaySleep); 6050 6051 LogTimer("m->NextScheduledQuery ", m->NextScheduledQuery); 6052 LogTimer("m->NextScheduledProbe ", m->NextScheduledProbe); 6053 LogTimer("m->NextScheduledResponse", m->NextScheduledResponse); 6054 6055 LogTimer("m->SuppressSending ", m->SuppressSending); 6056 LogTimer("m->SuppressProbes ", m->SuppressProbes); 6057 LogTimer("m->ProbeFailTime ", m->ProbeFailTime); 6058 LogTimer("m->DelaySleep ", m->DelaySleep); 6059 LogTimer("m->SleepLimit ", m->SleepLimit); 6060 LogTimer("m->NextScheduledStopTime ", m->NextScheduledStopTime); 6061 } 6062 6063 #if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 6064 mDNSexport void uds_validatelists(void) 6065 { 6066 const request_state *req, *p; 6067 for (req = all_requests; req; req=req->next) 6068 { 6069 if (req->next == (request_state *)~0 || (req->sd < 0 && req->sd != -2)) 6070 LogMemCorruption("UDS request list: %p is garbage (%d)", req, req->sd); 6071 6072 if (req->primary == req) 6073 LogMemCorruption("UDS request list: req->primary should not point to self %p/%d", req, req->sd); 6074 6075 if (req->primary && req->replies) 6076 LogMemCorruption("UDS request list: Subordinate request %p/%d/%p should not have replies (%p)", 6077 req, req->sd, req->primary && req->replies); 6078 6079 p = req->primary; 6080 if ((long)p & 3) 6081 LogMemCorruption("UDS request list: req %p primary %p is misaligned (%d)", req, p, req->sd); 6082 else if (p && (p->next == (request_state *)~0 || (p->sd < 0 && p->sd != -2))) 6083 LogMemCorruption("UDS request list: req %p primary %p is garbage (%d)", req, p, p->sd); 6084 6085 reply_state *rep; 6086 for (rep = req->replies; rep; rep=rep->next) 6087 if (rep->next == (reply_state *)~0) 6088 LogMemCorruption("UDS req->replies: %p is garbage", rep); 6089 6090 if (req->terminate == connection_termination) 6091 { 6092 registered_record_entry *r; 6093 for (r = req->u.reg_recs; r; r=r->next) 6094 if (r->next == (registered_record_entry *)~0) 6095 LogMemCorruption("UDS req->u.reg_recs: %p is garbage", r); 6096 } 6097 else if (req->terminate == regservice_termination_callback) 6098 { 6099 service_instance *s; 6100 for (s = req->u.servicereg.instances; s; s=s->next) 6101 if (s->next == (service_instance *)~0) 6102 LogMemCorruption("UDS req->u.servicereg.instances: %p is garbage", s); 6103 } 6104 else if (req->terminate == browse_termination_callback) 6105 { 6106 browser_t *b; 6107 for (b = req->u.browser.browsers; b; b=b->next) 6108 if (b->next == (browser_t *)~0) 6109 LogMemCorruption("UDS req->u.browser.browsers: %p is garbage", b); 6110 } 6111 } 6112 6113 DNameListElem *d; 6114 for (d = SCPrefBrowseDomains; d; d=d->next) 6115 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 6116 LogMemCorruption("SCPrefBrowseDomains: %p is garbage (%d)", d, d->name.c[0]); 6117 6118 ARListElem *b; 6119 for (b = LocalDomainEnumRecords; b; b=b->next) 6120 if (b->next == (ARListElem *)~0 || b->ar.resrec.name->c[0] > 63) 6121 LogMemCorruption("LocalDomainEnumRecords: %p is garbage (%d)", b, b->ar.resrec.name->c[0]); 6122 6123 for (d = AutoBrowseDomains; d; d=d->next) 6124 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 6125 LogMemCorruption("AutoBrowseDomains: %p is garbage (%d)", d, d->name.c[0]); 6126 6127 for (d = AutoRegistrationDomains; d; d=d->next) 6128 if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 6129 LogMemCorruption("AutoRegistrationDomains: %p is garbage (%d)", d, d->name.c[0]); 6130 } 6131 #endif // APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 6132 6133 mDNSlocal int send_msg(request_state *const req) 6134 { 6135 reply_state *const rep = req->replies; // Send the first waiting reply 6136 ssize_t nwriten; 6137 6138 ConvertHeaderBytes(rep->mhdr); 6139 nwriten = send(req->sd, (char *)&rep->mhdr + rep->nwriten, rep->totallen - rep->nwriten, 0); 6140 ConvertHeaderBytes(rep->mhdr); 6141 6142 if (nwriten < 0) 6143 { 6144 if (dnssd_errno == dnssd_EINTR || dnssd_errno == dnssd_EWOULDBLOCK) nwriten = 0; 6145 else 6146 { 6147 #if !defined(PLATFORM_NO_EPIPE) 6148 if (dnssd_errno == EPIPE) 6149 return(req->ts = t_terminated); 6150 else 6151 #endif 6152 { 6153 LogMsg("send_msg ERROR: failed to write %d of %d bytes to fd %d errno %d (%s)", 6154 rep->totallen - rep->nwriten, rep->totallen, req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 6155 return(t_error); 6156 } 6157 } 6158 } 6159 rep->nwriten += nwriten; 6160 return (rep->nwriten == rep->totallen) ? t_complete : t_morecoming; 6161 } 6162 6163 mDNSexport mDNSs32 udsserver_idle(mDNSs32 nextevent) 6164 { 6165 mDNSs32 now = mDNS_TimeNow(&mDNSStorage); 6166 request_state **req = &all_requests; 6167 6168 while (*req) 6169 { 6170 request_state *const r = *req; 6171 6172 if (r->terminate == resolve_termination_callback) 6173 if (r->u.resolve.ReportTime && now - r->u.resolve.ReportTime >= 0) 6174 { 6175 r->u.resolve.ReportTime = 0; 6176 // if client received results and resolve still active 6177 if (r->u.resolve.txt && r->u.resolve.srv) 6178 LogMsgNoIdent("Client application PID[%d](%s) has received results for DNSServiceResolve(%##s) yet remains active over two minutes.", r->process_id, r->pid_name, r->u.resolve.qsrv.qname.c); 6179 } 6180 6181 // Note: Only primary req's have reply lists, not subordinate req's. 6182 while (r->replies) // Send queued replies 6183 { 6184 transfer_state result; 6185 if (r->replies->next) 6186 r->replies->rhdr->flags |= dnssd_htonl(kDNSServiceFlagsMoreComing); 6187 result = send_msg(r); // Returns t_morecoming if buffer full because client is not reading 6188 if (result == t_complete) 6189 { 6190 reply_state *fptr = r->replies; 6191 r->replies = r->replies->next; 6192 freeL("reply_state/udsserver_idle", fptr); 6193 r->time_blocked = 0; // reset failure counter after successful send 6194 r->unresponsiveness_reports = 0; 6195 continue; 6196 } 6197 else if (result == t_terminated) 6198 { 6199 LogInfo("%3d: Could not write data to client PID[%d](%s) because connection is terminated by the client", r->sd, r->process_id, r->pid_name); 6200 abort_request(r); 6201 } 6202 else if (result == t_error) 6203 { 6204 LogMsg("%3d: Could not write data to client PID[%d](%s) because of error - aborting connection", r->sd, r->process_id, r->pid_name); 6205 LogClientInfo(r); 6206 abort_request(r); 6207 } 6208 break; 6209 } 6210 6211 if (r->replies) // If we failed to send everything, check our time_blocked timer 6212 { 6213 if (nextevent - now > mDNSPlatformOneSecond) 6214 nextevent = now + mDNSPlatformOneSecond; 6215 6216 if (mDNSStorage.SleepState != SleepState_Awake) 6217 r->time_blocked = 0; 6218 else if (!r->time_blocked) 6219 r->time_blocked = NonZeroTime(now); 6220 else if (now - r->time_blocked >= 10 * mDNSPlatformOneSecond * (r->unresponsiveness_reports+1)) 6221 { 6222 int num = 0; 6223 struct reply_state *x = r->replies; 6224 while (x) 6225 { 6226 num++; 6227 x=x->next; 6228 } 6229 LogMsg("%3d: Could not write data to client PID[%d](%s) after %ld seconds, %d repl%s waiting", 6230 r->sd, r->process_id, r->pid_name, (now - r->time_blocked) / mDNSPlatformOneSecond, num, num == 1 ? "y" : "ies"); 6231 if (++r->unresponsiveness_reports >= 60) 6232 { 6233 LogMsg("%3d: Client PID[%d](%s) unresponsive; aborting connection", r->sd, r->process_id, r->pid_name); 6234 LogClientInfo(r); 6235 abort_request(r); 6236 } 6237 } 6238 } 6239 6240 if (!dnssd_SocketValid(r->sd)) // If this request is finished, unlink it from the list and free the memory 6241 { 6242 // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 6243 *req = r->next; 6244 freeL("request_state/udsserver_idle", r); 6245 } 6246 else 6247 req = &r->next; 6248 } 6249 return nextevent; 6250 } 6251 6252 struct CompileTimeAssertionChecks_uds_daemon 6253 { 6254 // Check our structures are reasonable sizes. Including overly-large buffers, or embedding 6255 // other overly-large structures instead of having a pointer to them, can inadvertently 6256 // cause structure sizes (and therefore memory usage) to balloon unreasonably. 6257 char sizecheck_request_state [(sizeof(request_state) <= 2954) ? 1 : -1]; 6258 char sizecheck_registered_record_entry[(sizeof(registered_record_entry) <= 60) ? 1 : -1]; 6259 char sizecheck_service_instance [(sizeof(service_instance) <= 6552) ? 1 : -1]; 6260 char sizecheck_browser_t [(sizeof(browser_t) <= 1202) ? 1 : -1]; 6261 char sizecheck_reply_hdr [(sizeof(reply_hdr) <= 12) ? 1 : -1]; 6262 char sizecheck_reply_state [(sizeof(reply_state) <= 64) ? 1 : -1]; 6263 }; 6264 6265 #ifdef UNIT_TEST 6266 #include "../unittests/uds_daemon_ut.c" 6267 #endif // UNIT_TEST 6268