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