147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt/* -*- Mode: C; tab-width: 4 -*- 247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * 347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * Copyright (c) 2003-2006 Apple Computer, Inc. All rights reserved. 447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * 547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * Licensed under the Apache License, Version 2.0 (the "License"); 647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * you may not use this file except in compliance with the License. 747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * You may obtain a copy of the License at 847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * 947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * http://www.apache.org/licenses/LICENSE-2.0 1047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * 1147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * Unless required by applicable law or agreed to in writing, software 1247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * distributed under the License is distributed on an "AS IS" BASIS, 1347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * See the License for the specific language governing permissions and 1547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt * limitations under the License. 1647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt */ 1747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 1847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(_WIN32) 1947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <process.h> 2047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define usleep(X) Sleep(((X)+999)/1000) 2147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 2247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <fcntl.h> 2347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <errno.h> 2447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <sys/ioctl.h> 2547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <sys/types.h> 2647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <sys/time.h> 2747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <sys/resource.h> 2847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 2947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 3047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <stdlib.h> 3147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <stdio.h> 3247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 3347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include "mDNSEmbeddedAPI.h" 3447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include "DNSCommon.h" 3547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include "uDNS.h" 3647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include "uds_daemon.h" 3747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 380b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon#ifdef __ANDROID__ 390b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon#include "cutils/sockets.h" 400b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon#endif 410b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon 4247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Normally we append search domains only for queries with a single label that are not 4347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// fully qualified. This can be overridden to apply search domains for queries (that are 4447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// not fully qualified) with any number of labels e.g., moon, moon.cs, moon.cs.be, etc. 4547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSBool AlwaysAppendSearchDomains = mDNSfalse; 4647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 4747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Apple-specific functionality, not required for other platforms 4847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 4947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <sys/ucred.h> 5047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#ifndef PID_FILE 5147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define PID_FILE "" 5247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 5347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 5447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 5547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 5647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#include <WebFilterDNS/WebFilterDNS.h> 5747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 5847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if ! NO_WCF 5947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 6047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltint WCFIsServerRunning(WCFConnection *conn) __attribute__((weak_import)); 6147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltint WCFNameResolvesToAddr(WCFConnection *conn, char* domainName, struct sockaddr* address, uid_t userid) __attribute__((weak_import)); 6247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltint WCFNameResolvesToName(WCFConnection *conn, char* fromName, char* toName, uid_t userid) __attribute__((weak_import)); 6347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 6447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Do we really need to define a macro for "if"? 6547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define CHECK_WCF_FUNCTION(X) if (X) 6647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // ! NO_WCF 6747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 6847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 6947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define NO_WCF 1 7047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 7147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 7247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// User IDs 0-500 are system-wide processes, not actual users in the usual sense 7347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// User IDs for real user accounts start at 501 and count up from there 7447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define SystemUID(X) ((X) <= 500) 7547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 7647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 7747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 7847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 7947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - Types and Data Structures 8047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 8147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 8247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef enum 8347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 8447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt t_uninitialized, 8547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt t_morecoming, 8647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt t_complete, 8747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt t_error, 8847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt t_terminated 8947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } transfer_state; 9047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 9147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct request_state request_state; 9247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 9347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef void (*req_termination_fn)(request_state *request); 9447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 9547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct registered_record_entry 9647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 9747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct registered_record_entry *next; 9847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 key; 9947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt client_context_t regrec_client_context; 10047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request; 10147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool external_advertise; 10247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID origInterfaceID; 10347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr; // Pointer to variable-sized AuthRecord (Why a pointer? Why not just embed it here?) 10447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } registered_record_entry; 10547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 10647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// A single registered service: ServiceRecordSet + bookkeeping 10747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Note that we duplicate some fields from parent service_info object 10847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// to facilitate cleanup, when instances and parent may be deallocated at different times. 10947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct service_instance 11047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 11147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct service_instance *next; 11247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request; 11347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *subtypes; 11447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool renameonmemfree; // Set on config change when we deregister original name 11547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool clientnotified; // Has client been notified of successful registration yet? 11647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool default_local; // is this the "local." from an empty-string registration? 11747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool external_advertise; // is this is being advertised externally? 11847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname domain; 11947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ServiceRecordSet srs; // note -- variable-sized object -- must be last field in struct 12047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } service_instance; 12147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 12247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// for multi-domain default browsing 12347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct browser_t 12447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 12547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct browser_t *next; 12647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname domain; 12747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q; 12847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } browser_t; 12947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 13047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstruct request_state 13147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 13247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *next; 13347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *primary; // If this operation is on a shared socket, pointer to primary 13447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // request_state for the original DNSServiceCreateConnection() operation 13547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sock_t sd; 13647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sock_t errsd; 13747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 uid; 13847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt void * platform_data; 13947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 14047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: On a shared connection these fields in the primary structure, including hdr, are re-used 14147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // for each new request. This is because, until we've read the ipc_msg_hdr to find out what the 14247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // operation is, we don't know if we're going to need to allocate a new request_state or not. 14347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt transfer_state ts; 14447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 hdr_bytes; // bytes of header already read 14547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ipc_msg_hdr hdr; 14647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 data_bytes; // bytes of message data already read 14747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *msgbuf; // pointer to data storage to pass to free() 14847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *msgptr; // pointer to data to be read from (may be modified) 14947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *msgend; // pointer to byte after last byte of message 15047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 15147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // reply, termination, error, and client context info 15247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int no_reply; // don't send asynchronous replies to client 15347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 time_blocked; // record time of a blocked client 15447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int unresponsiveness_reports; 15547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct reply_state *replies; // corresponding (active) reply list 15647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req_termination_fn terminate; 15747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags; 15847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 15947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt union 16047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 16147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *reg_recs; // list of registrations for a connection-oriented request 16247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 16347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 16447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID interface_id; 16547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool default_domain; 16647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool ForceMCast; 16747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname regtype; 16847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *browsers; 16947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } browser; 17047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 17147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 17247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID; 17347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 txtlen; 17447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt void *txtdata; 17547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSIPPort port; 17647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainlabel name; 17747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char type_as_string[MAX_ESCAPED_DOMAIN_NAME]; 17847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname type; 17947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool default_domain; 18047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname host; 18147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool autoname; // Set if this name is tied to the Computer Name 18247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool autorename; // Set if this client wants us to automatically rename on conflict 18347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool allowremotequery; // Respond to unicast queries from outside the local link? 18447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int num_subtypes; 18547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *instances; 18647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } servicereg; 18747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 18847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 18947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID interface_id; 19047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 flags; 19147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 protocol; 19247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q4; 19347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *q42; 19447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q6; 19547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *q62; 19647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } addrinfo; 19747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 19847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 19947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSIPPort ReqExt; // External port we originally requested, for logging purposes 20047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt NATTraversalInfo NATinfo; 20147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } pm; 20247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 20347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 20447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 20547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags; 20647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 20747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q_all; 20847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q_default; 20947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } enumeration; 21047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 21147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 21247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion q; 21347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *q2; 21447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } queryrecord; 21547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct 21647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 21747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion qtxt; 21847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion qsrv; 21947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const ResourceRecord *txt; 22047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const ResourceRecord *srv; 22147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 ReportTime; 22247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool external_advertise; 22347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } resolve; 22447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } u; 22547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt }; 22647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 22747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// struct physically sits between ipc message header and call-specific fields in the message buffer 22847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct 22947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 23047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags; // Note: This field is in NETWORK byte order 23147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ifi; // Note: This field is in NETWORK byte order 23247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceErrorType error; // Note: This field is in NETWORK byte order 23347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } reply_hdr; 23447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 23547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef struct reply_state 23647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 23747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct reply_state *next; // If there are multiple unsent replies 23847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 totallen; 23947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 nwriten; 24047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ipc_msg_hdr mhdr[1]; 24147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_hdr rhdr[1]; 24247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } reply_state; 24347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 24447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 24547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 24647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 24747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - Globals 24847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 24947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 25047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// globals 25147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport mDNS mDNSStorage; 25247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport const char ProgramName[] = "mDNSResponder"; 25347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 25447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstatic dnssd_sock_t listenfd = dnssd_InvalidSocket; 25547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstatic request_state *all_requests = NULL; 25647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 25747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Note asymmetry here between registration and browsing. 25847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// For service registrations we only automatically register in domains that explicitly appear in local configuration data 25947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// (so AutoRegistrationDomains could equally well be called SCPrefRegDomains) 26047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// For service browsing we also learn automatic browsing domains from the network, so for that case we have: 26147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// 1. SCPrefBrowseDomains (local configuration data) 26247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// 2. LocalDomainEnumRecords (locally-generated local-only PTR records -- equivalent to slElem->AuthRecs in uDNS.c) 26347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// 3. AutoBrowseDomains, which is populated by tracking add/rmv events in AutomaticBrowseDomainChange, the callback function for our mDNS_GetDomains call. 26447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// By creating and removing our own LocalDomainEnumRecords, we trigger AutomaticBrowseDomainChange callbacks just like domains learned from the network would. 26547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 26647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport DNameListElem *AutoRegistrationDomains; // Domains where we automatically register for empty-string registrations 26747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 26847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstatic DNameListElem *SCPrefBrowseDomains; // List of automatic browsing domains read from SCPreferences for "empty string" browsing 26947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstatic ARListElem *LocalDomainEnumRecords; // List of locally-generated PTR records to augment those we learn from the network 27047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport DNameListElem *AutoBrowseDomains; // List created from those local-only PTR records plus records we get from the network 27147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 27247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define MSG_PAD_BYTES 5 // pad message buffer (read from client) with n zero'd bytes to guarantee 27347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // n get_string() calls w/o buffer overrun 27447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// initialization, setup/teardown functions 27547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 27647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// If a platform specifies its own PID file name, we use that 27747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#ifndef PID_FILE 27847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define PID_FILE "/var/run/mDNSResponder.pid" 27947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 28047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 28147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 28247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 28347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 28447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - General Utility Functions 28547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 28647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 28747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void FatalError(char *errmsg) 28847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 28947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%s: %s", errmsg, dnssd_strerror(dnssd_errno)); 2904eea2fde995314ff68ed9ef0a3f0cc745695e544Stephen Hines *(volatile long*)0 = 0; // On OS X abort() doesn't generate a crash log, but writing to zero does 29147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort(); // On platforms where writing to zero doesn't generate an exception, abort instead 29247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 29347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 29447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mDNSu32 dnssd_htonl(mDNSu32 l) 29547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 29647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ret; 29747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data = (char*) &ret; 29847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint32(l, &data); 29947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return ret; 30047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 30147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 30247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// hack to search-replace perror's to LogMsg's 30347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void my_perror(char *errmsg) 30447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 30547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%s: %d (%s)", errmsg, dnssd_errno, dnssd_strerror(dnssd_errno)); 30647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 30747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 30847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void abort_request(request_state *req) 30947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 31047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate == (req_termination_fn)~0) 31147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("abort_request: ERROR: Attempt to abort operation %p with req->terminate %p", req, req->terminate); return; } 31247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 31347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // First stop whatever mDNSCore operation we were doing 31447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If this is actually a shared connection operation, then its req->terminate function will scan 31547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the all_requests list and terminate any subbordinate operations sharing this file descriptor 31647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate) req->terminate(req); 31747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 31847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!dnssd_SocketValid(req->sd)) 31947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("abort_request: ERROR: Attempt to abort operation %p with invalid fd %d", req, req->sd); return; } 32047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 32147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Now, if this request_state is not subordinate to some other primary, close file descriptor and discard replies 32247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!req->primary) 32347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 32447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->errsd != req->sd) LogOperation("%3d: Removing FD and closing errsd %d", req->sd, req->errsd); 32547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else LogOperation("%3d: Removing FD", req->sd); 32647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsSupportRemoveFDFromEventLoop(req->sd, req->platform_data); // Note: This also closes file descriptor req->sd for us 32747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->errsd != req->sd) { dnssd_close(req->errsd); req->errsd = req->sd; } 32847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 32947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (req->replies) // free pending replies 33047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 33147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *ptr = req->replies; 33247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->replies = req->replies->next; 33347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("reply_state (abort)", ptr); 33447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 33547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 33647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 33747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Set req->sd to something invalid, so that udsserver_idle knows to unlink and free this structure 33847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 33947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't use dnssd_InvalidSocket (-1) because that's the sentinel value MACOSX_MDNS_MALLOC_DEBUGGING uses 34047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // for detecting when the memory for an object is inadvertently freed while the object is still on some list 34147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd = req->errsd = -2; 34247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 34347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd = req->errsd = dnssd_InvalidSocket; 34447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 34547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We also set req->terminate to a bogus value so we know if abort_request() gets called again for this request 34647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->terminate = (req_termination_fn)~0; 34747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 34847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 34947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void AbortUnlinkAndFree(request_state *req) 35047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 35147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state **p = &all_requests; 35247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort_request(req); 35347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p && *p != req) p=&(*p)->next; 35447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*p) { *p = req->next; freeL("request_state/AbortUnlinkAndFree", req); } 35547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else LogMsg("AbortUnlinkAndFree: ERROR: Attempt to abort operation %p not in list", req); 35647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 35747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 35847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal reply_state *create_reply(const reply_op_t op, const size_t datalen, request_state *const request) 35947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 36047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *reply; 36147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 36247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((unsigned)datalen < sizeof(reply_hdr)) 36347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 36447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: create_reply - data length less than length of required fields"); 36547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return NULL; 36647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 36747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 36847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply = mallocL("reply_state", sizeof(reply_state) + datalen - sizeof(reply_hdr)); 36947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!reply) FatalError("ERROR: malloc"); 37047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 37147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->next = mDNSNULL; 37247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->totallen = (mDNSu32)datalen + sizeof(ipc_msg_hdr); 37347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->nwriten = 0; 37447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 37547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->version = VERSION; 37647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->datalen = (mDNSu32)datalen; 37747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->ipc_flags = 0; 37847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->op = op; 37947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->client_context = request->hdr.client_context; 38047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->reg_index = 0; 38147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 38247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return reply; 38347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 38447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 38547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Append a reply to the list in a request object 38647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// If our request is sharing a connection, then we append our reply_state onto the primary's list 38747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void append_reply(request_state *req, reply_state *rep) 38847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 38947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *r = req->primary ? req->primary : req; 39047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state **ptr = &r->replies; 39147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr) ptr = &(*ptr)->next; 39247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = rep; 39347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->next = NULL; 39447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 39547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 39647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Generates a response message giving name, type, domain, plus interface index, 39747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// suitable for a browse result or service registration result. 39847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// On successful completion rep is set to point to a malloc'd reply_state struct 39947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus GenerateNTDResponse(const domainname *const servicename, const mDNSInterfaceID id, 40047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err) 40147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 40247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainlabel name; 40347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname type, dom; 40447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *rep = NULL; 40547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!DeconstructServiceName(servicename, &name, &type, &dom)) 40647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return kDNSServiceErr_Invalid; 40747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 40847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 40947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char namestr[MAX_DOMAIN_LABEL+1]; 41047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char typestr[MAX_ESCAPED_DOMAIN_NAME]; 41147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char domstr [MAX_ESCAPED_DOMAIN_NAME]; 41247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int len; 41347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 41447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 41547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainLabelToCString_unescaped(&name, namestr); 41647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(&type, typestr); 41747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(&dom, domstr); 41847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 41947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Calculate reply data length 42047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len = sizeof(DNSServiceFlags); 42147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); // if index 42247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceErrorType); 42347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(namestr) + 1); 42447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(typestr) + 1); 42547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(domstr) + 1); 42647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 42747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Build reply header 42847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *rep = create_reply(op, len, request); 42947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->flags = dnssd_htonl(flags); 43047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse)); 43147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->error = dnssd_htonl(err); 43247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 43347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Build reply body 43447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&(*rep)->rhdr[1]; 43547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(namestr, &data); 43647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(typestr, &data); 43747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(domstr, &data); 43847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 43947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mStatus_NoError; 44047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 44147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 44247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 44347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Special support to enable the DNSServiceBrowse call made by Bonjour Browser 44447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse 44547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void GenerateBonjourBrowserResponse(const domainname *const servicename, const mDNSInterfaceID id, 44647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *const request, reply_state **const rep, reply_op_t op, DNSServiceFlags flags, mStatus err) 44747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 44847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char namestr[MAX_DOMAIN_LABEL+1]; 44947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char typestr[MAX_ESCAPED_DOMAIN_NAME]; 45047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt static const char domstr[] = "."; 45147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int len; 45247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 45347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 45447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *rep = NULL; 45547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 45647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 1. Put first label in namestr 45747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainLabelToCString_unescaped((const domainlabel *)servicename, namestr); 45847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 45947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 2. Put second label and "local" into typestr 46047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_snprintf(typestr, sizeof(typestr), "%#s.local.", SecondLabel(servicename)); 46147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 46247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Calculate reply data length 46347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len = sizeof(DNSServiceFlags); 46447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); // if index 46547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceErrorType); 46647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(namestr) + 1); 46747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(typestr) + 1); 46847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += (int) (strlen(domstr) + 1); 46947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 47047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Build reply header 47147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *rep = create_reply(op, len, request); 47247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->flags = dnssd_htonl(flags); 47347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, id, mDNSfalse)); 47447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*rep)->rhdr->error = dnssd_htonl(err); 47547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 47647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Build reply body 47747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&(*rep)->rhdr[1]; 47847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(namestr, &data); 47947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(typestr, &data); 48047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(domstr, &data); 48147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 48247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 48347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Returns a resource record (allocated w/ malloc) containing the data found in an IPC message 48447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Data must be in the following format: flags, interfaceIndex, name, rrtype, rrclass, rdlen, rdata, (optional) ttl 48547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// (ttl only extracted/set if ttl argument is non-zero). Returns NULL for a bad-parameter error 48647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal AuthRecord *read_rr_from_ipc_msg(request_state *request, int GetTTL, int validate_flags) 48747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 48847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 48947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 49047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char name[256]; 49147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int str_err = get_string(&request->msgptr, request->msgend, name, sizeof(name)); 49247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 type = get_uint16(&request->msgptr, request->msgend); 49347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 class = get_uint16(&request->msgptr, request->msgend); 49447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 49547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 49647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ttl = GetTTL ? get_uint32(&request->msgptr, request->msgend) : 0; 49747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int storage_size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 49847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr; 49947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID; 50047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecType artype; 50147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 50247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->flags = flags; 50347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 50447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (str_err) { LogMsg("ERROR: read_rr_from_ipc_msg - get_string"); return NULL; } 50547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 50647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("Error reading Resource Record from client"); return NULL; } 50747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 50847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (validate_flags && 50947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt !((flags & kDNSServiceFlagsShared) == kDNSServiceFlagsShared) && 51047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt !((flags & kDNSServiceFlagsUnique) == kDNSServiceFlagsUnique)) 51147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 51247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: Bad resource record flags (must be kDNSServiceFlagsShared or kDNSServiceFlagsUnique)"); 51347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return NULL; 51447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 51547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 51647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr = mallocL("AuthRecord/read_rr_from_ipc_msg", sizeof(AuthRecord) - sizeof(RDataBody) + storage_size); 51747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!rr) FatalError("ERROR: malloc"); 51847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 51947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 52047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (InterfaceID == mDNSInterface_LocalOnly) 52147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt artype = AuthRecordLocalOnly; 52247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (InterfaceID == mDNSInterface_P2P) 52347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt artype = AuthRecordP2P; 52447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if ((InterfaceID == mDNSInterface_Any) && (flags & kDNSServiceFlagsIncludeP2P)) 52547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt artype = AuthRecordAnyIncludeP2P; 52647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 52747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt artype = AuthRecordAny; 52847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 52947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_SetupResourceRecord(rr, mDNSNULL, InterfaceID, type, 0, 53047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (mDNSu8) ((flags & kDNSServiceFlagsShared) ? kDNSRecordTypeShared : kDNSRecordTypeUnique), artype, mDNSNULL, mDNSNULL); 53147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 53247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&rr->namestorage, name)) 53347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 53447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: bad name: %s", name); 53547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("AuthRecord/read_rr_from_ipc_msg", rr); 53647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return NULL; 53747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 53847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 53947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (flags & kDNSServiceFlagsAllowRemoteQuery) rr->AllowRemoteQuery = mDNStrue; 54047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rrclass = class; 54147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rdlength = rdlen; 54247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rdata->MaxRDLength = rdlen; 54347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(rr->resrec.rdata->u.data, rdata, rdlen); 54447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GetTTL) rr->resrec.rroriginalttl = ttl; 54547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.namehash = DomainNameHashValue(rr->resrec.name); 54647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SetNewRData(&rr->resrec, mDNSNULL, 0); // Sets rr->rdatahash for us 54747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return rr; 54847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 54947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 55047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal int build_domainname_from_strings(domainname *srv, char *name, char *regtype, char *domain) 55147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 55247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainlabel n; 55347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname d, t; 55447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 55547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainLabelFromLiteralString(&n, name)) return -1; 55647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&t, regtype)) return -1; 55747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&d, domain)) return -1; 55847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!ConstructServiceName(srv, &n, &t, &d)) return -1; 55947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return 0; 56047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 56147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 56247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void send_all(dnssd_sock_t s, const char *ptr, int len) 56347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 56447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int n = send(s, ptr, len, 0); 56547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // On a freshly-created Unix Domain Socket, the kernel should *never* fail to buffer a small write for us 56647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // (four bytes for a typical error code return, 12 bytes for DNSServiceGetProperty(DaemonVersion)). 56747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If it does fail, we don't attempt to handle this failure, but we do log it so we know something is wrong. 56847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (n < len) 56947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: send_all(%d) wrote %d of %d errno %d (%s)", 57047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt s, n, len, dnssd_errno, dnssd_strerror(dnssd_errno)); 57147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 57247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 57347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 57447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mDNSBool AuthorizedDomain(const request_state * const request, const domainname * const d, const DNameListElem * const doms) 57547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt{ 57647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const DNameListElem *delem = mDNSNULL; 57747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int bestDelta = -1; // the delta of the best match, lower is better 57847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int dLabels = 0; 57947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool allow = mDNSfalse; 58047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 58147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SystemUID(request->uid)) return mDNStrue; 58247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 58347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dLabels = CountLabels(d); 58447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (delem = doms; delem; delem = delem->next) 58547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 58647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (delem->uid) 58747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 58847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int delemLabels = CountLabels(&delem->name); 58947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int delta = dLabels - delemLabels; 59047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((bestDelta == -1 || delta <= bestDelta) && SameDomainName(&delem->name, SkipLeadingLabels(d, delta))) 59147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 59247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt bestDelta = delta; 59347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt allow = (allow || (delem->uid == request->uid)); 59447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 59547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 59647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 59747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 59847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return bestDelta == -1 ? mDNStrue : allow; 59947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt} 60047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 60147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 60247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 60347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 60447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 60547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - external helpers 60647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 60747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 60847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void external_start_advertising_helper(service_instance *const instance) 60947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 61047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *st = instance->subtypes; 61147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *e; 61247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int i; 61347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 61447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSIPPortIsZero(instance->request->u.servicereg.port)) 61547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 61647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("external_start_advertising_helper: Not registering service with port number zero"); 61747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 61847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 61947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 62047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 62147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Update packet filter if p2p interface already exists, otherwise, 62247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // if will be updated when we get the KEV_DL_IF_ATTACHED event for 62347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the interface. Called here since we don't call external_start_advertising_service() 62447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // with the SRV record when advertising a service. 62547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInitPacketFilter(); 62647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 62747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 62847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->external_advertise) LogMsg("external_start_advertising_helper: external_advertise already set!"); 62947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 63047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++) 63147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&st[i].resrec); 63247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 63347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&instance->srs.RR_PTR.resrec); 63447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&instance->srs.RR_TXT.resrec); 63547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 63647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (e = instance->srs.Extras; e; e = e->next) 63747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&e->r.resrec); 63847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 63947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->external_advertise = mDNStrue; 64047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 64147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 64247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void external_stop_advertising_helper(service_instance *const instance) 64347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 64447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *st = instance->subtypes; 64547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *e; 64647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int i; 64747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 64847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!instance->external_advertise) return; 64947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 65047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("external_stop_advertising_helper: calling external_stop_advertising_service"); 65147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 65247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for ( i = 0; i < instance->request->u.servicereg.num_subtypes; i++) 65347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&st[i].resrec); 65447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 65547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&instance->srs.RR_PTR.resrec); 65647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&instance->srs.RR_TXT.resrec); 65747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 65847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (e = instance->srs.Extras; e; e = e->next) 65947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&e->r.resrec); 66047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 66147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->external_advertise = mDNSfalse; 66247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 66347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 66447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 66547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 66647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 66747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceRegister 66847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 66947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 67047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport void FreeExtraRR(mDNS *const m, AuthRecord *const rr, mStatus result) 67147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 67247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *extra = (ExtraResourceRecord *)rr->RecordContext; 67347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 67447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 67547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result != mStatus_MemFree) { LogMsg("Error: FreeExtraRR invoked with unexpected error %d", result); return; } 67647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 67747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo(" FreeExtraRR %s", RRDisplayString(m, &rr->resrec)); 67847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 67947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rdata != &rr->rdatastorage) 68047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("Extra RData", rr->resrec.rdata); 68147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("ExtraResourceRecord/FreeExtraRR", extra); 68247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 68347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 68447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void unlink_and_free_service_instance(service_instance *srv) 68547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 68647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *e = srv->srs.Extras, *tmp; 68747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 68847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_helper(srv); 68947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 69047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // clear pointers from parent struct 69147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (srv->request) 69247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 69347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance **p = &srv->request->u.servicereg.instances; 69447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p) 69547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 69647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*p == srv) { *p = (*p)->next; break; } 69747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt p = &(*p)->next; 69847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 69947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 70047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 70147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (e) 70247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 70347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt e->r.RecordContext = e; 70447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt tmp = e; 70547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt e = e->next; 70647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt FreeExtraRR(&mDNSStorage, &tmp->r, mStatus_MemFree); 70747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 70847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 70947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (srv->srs.RR_TXT.resrec.rdata != &srv->srs.RR_TXT.rdatastorage) 71047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("TXT RData", srv->srs.RR_TXT.resrec.rdata); 71147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 71247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (srv->subtypes) { freeL("ServiceSubTypes", srv->subtypes); srv->subtypes = NULL; } 71347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("service_instance", srv); 71447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 71547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 71647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Count how many other service records we have locally with the same name, but different rdata. 71747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// For auto-named services, we can have at most one per machine -- if we allowed two auto-named services of 71847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// the same type on the same machine, we'd get into an infinite autoimmune-response loop of continuous renaming. 71947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport int CountPeerRegistrations(mDNS *const m, ServiceRecordSet *const srs) 72047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 72147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int count = 0; 72247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ResourceRecord *r = &srs->RR_SRV.resrec; 72347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr; 72447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 72547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (rr = m->ResourceRecords; rr; rr=rr->next) 72647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rrtype == kDNSType_SRV && SameDomainName(rr->resrec.name, r->name) && !IdenticalSameNameRecord(&rr->resrec, r)) 72747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt count++; 72847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 72947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt verbosedebugf("%d peer registrations for %##s", count, r->name->c); 73047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(count); 73147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 73247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 73347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport int CountExistingRegistrations(domainname *srv, mDNSIPPort port) 73447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 73547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int count = 0; 73647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr; 73747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (rr = mDNSStorage.ResourceRecords; rr; rr=rr->next) 73847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rrtype == kDNSType_SRV && 73947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSSameIPPort(rr->resrec.rdata->u.srv.port, port) && 74047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SameDomainName(rr->resrec.name, srv)) 74147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt count++; 74247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(count); 74347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 74447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 74547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void SendServiceRemovalNotification(ServiceRecordSet *const srs) 74647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 74747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 74847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *instance = srs->ServiceContext; 74947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, 0, mStatus_NoError) != mStatus_NoError) 75047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: SendServiceRemovalNotification: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 75147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else { append_reply(instance->request, rep); instance->clientnotified = mDNSfalse; } 75247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 75347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 75447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// service registration callback performs three duties - frees memory for deregistered services, 75547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// handles name conflicts, and delivers completed registration information to the client 75647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void regservice_callback(mDNS *const m, ServiceRecordSet *const srs, mStatus result) 75747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 75847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 75947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool SuppressError = mDNSfalse; 76047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *instance; 76147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 76247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 76347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 76447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!srs) { LogMsg("regservice_callback: srs is NULL %d", result); return; } 76547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 76647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance = srs->ServiceContext; 76747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!instance) { LogMsg("regservice_callback: srs->ServiceContext is NULL %d", result); return; } 76847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 76947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // don't send errors up to client for wide-area, empty-string registrations 77047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request && 77147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->request->u.servicereg.default_domain && 77247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt !instance->default_local) 77347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SuppressError = mDNStrue; 77447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 77547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNS_LoggingEnabled) 77647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 77747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *const fmt = 77847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (result == mStatus_NoError) ? "%s DNSServiceRegister(%##s, %u) REGISTERED" : 77947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (result == mStatus_MemFree) ? "%s DNSServiceRegister(%##s, %u) DEREGISTERED" : 78047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (result == mStatus_NameConflict) ? "%s DNSServiceRegister(%##s, %u) NAME CONFLICT" : 78147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt "%s DNSServiceRegister(%##s, %u) %s %d"; 78247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char prefix[16] = "---:"; 78347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request) mDNS_snprintf(prefix, sizeof(prefix), "%3d:", instance->request->sd); 78447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation(fmt, prefix, srs->RR_SRV.resrec.name->c, mDNSVal16(srs->RR_SRV.resrec.rdata->u.srv.port), 78547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SuppressError ? "suppressed error" : "CALLBACK", result); 78647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 78747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 78847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!instance->request && result != mStatus_MemFree) { LogMsg("regservice_callback: instance->request is NULL %d", result); return; } 78947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 79047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result == mStatus_NoError) 79147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 79247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request->u.servicereg.allowremotequery) 79347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 79447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *e; 79547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt srs->RR_ADV.AllowRemoteQuery = mDNStrue; 79647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt srs->RR_PTR.AllowRemoteQuery = mDNStrue; 79747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt srs->RR_SRV.AllowRemoteQuery = mDNStrue; 79847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt srs->RR_TXT.AllowRemoteQuery = mDNStrue; 79947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (e = instance->srs.Extras; e; e = e->next) e->r.AllowRemoteQuery = mDNStrue; 80047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 80147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 80247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 80347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 80447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 80547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 80647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request->u.servicereg.InterfaceID == mDNSInterface_P2P || (!instance->request->u.servicereg.InterfaceID && SameDomainName(&instance->domain, &localdomain) && (instance->request->flags & kDNSServiceFlagsIncludeP2P))) 80747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 80847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("regservice_callback: calling external_start_advertising_helper()"); 80947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_helper(instance); 81047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 81147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0) 81247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RecordUpdatedNiceLabel(m, 0); // Successfully got new name, tell user immediately 81347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 81447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (result == mStatus_MemFree) 81547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 81647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request && instance->renameonmemfree) 81747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 81847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_helper(instance); 81947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->renameonmemfree = 0; 82047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_RenameAndReregisterService(m, srs, &instance->request->u.servicereg.name); 82147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("ERROR: regservice_callback - RenameAndReregisterService returned %d", err); 82247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // error should never happen - safest to log and continue 82347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 82447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 82547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt unlink_and_free_service_instance(instance); 82647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 82747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (result == mStatus_NameConflict) 82847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 82947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request->u.servicereg.autorename) 83047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 83147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_helper(instance); 83247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->request->u.servicereg.autoname && CountPeerRegistrations(m, srs) == 0) 83347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 83447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // On conflict for an autoname service, rename and reregister *all* autoname services 83547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt IncrementLabelSuffix(&m->nicelabel, mDNStrue); 83647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_ConfigChanged(m); // Will call back into udsserver_handle_configchange() 83747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 83847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else // On conflict for a non-autoname service, rename and reregister just that one service 83947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 84047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->clientnotified) SendServiceRemovalNotification(srs); 84147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_RenameAndReregisterService(m, srs, mDNSNULL); 84247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 84347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 84447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 84547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 84647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!SuppressError) 84747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 84847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 84947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 85047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 85147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 85247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt unlink_and_free_service_instance(instance); 85347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 85447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 85547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else // Not mStatus_NoError, mStatus_MemFree, or mStatus_NameConflict 85647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 85747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!SuppressError) 85847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 85947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GenerateNTDResponse(srs->RR_SRV.resrec.name, srs->RR_SRV.resrec.InterfaceID, instance->request, &rep, reg_service_reply_op, kDNSServiceFlagsAdd, result) != mStatus_NoError) 86047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: regservice_callback: %##s is not valid DNS-SD SRV name", instance->request->sd, srs->RR_SRV.resrec.name->c); 86147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else { append_reply(instance->request, rep); instance->clientnotified = mDNStrue; } 86247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 86347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 86447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 86547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 86647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void regrecord_callback(mDNS *const m, AuthRecord *rr, mStatus result) 86747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 86847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 86947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!rr->RecordContext) // parent struct already freed by termination callback 87047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 87147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result == mStatus_NoError) 87247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("Error: regrecord_callback: successful registration of orphaned record %s", ARDisplayString(m, rr)); 87347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 87447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 87547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result != mStatus_MemFree) LogMsg("regrecord_callback: error %d received after parent termination", result); 87647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 87747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We come here when the record is being deregistered either from DNSServiceRemoveRecord or connection_termination. 87847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the record has been updated, we need to free the rdata. Everytime we call mDNS_Update, it calls update_callback 87947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // with the old rdata (so that we can free it) and stores the new rdata in "rr->resrec.rdata". This means, we need 88047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // to free the latest rdata for which the update_callback was never called with. 88147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rdata != &rr->rdatastorage) freeL("RData/regrecord_callback", rr->resrec.rdata); 88247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("AuthRecord/regrecord_callback", rr); 88347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 88447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 88547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 88647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 88747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *re = rr->RecordContext; 88847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request = re->request; 88947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 89047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNS_LoggingEnabled) 89147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 89247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *fmt = (result == mStatus_NoError) ? "%3d: DNSServiceRegisterRecord(%u %s) REGISTERED" : 89347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (result == mStatus_MemFree) ? "%3d: DNSServiceRegisterRecord(%u %s) DEREGISTERED" : 89447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (result == mStatus_NameConflict) ? "%3d: DNSServiceRegisterRecord(%u %s) NAME CONFLICT" : 89547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt "%3d: DNSServiceRegisterRecord(%u %s) %d"; 89647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation(fmt, request->sd, re->key, RRDisplayString(m, &rr->resrec), result); 89747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 89847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 89947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result != mStatus_MemFree) 90047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 90147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int len = sizeof(DNSServiceFlags) + sizeof(mDNSu32) + sizeof(DNSServiceErrorType); 90247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *reply = create_reply(reg_record_reply_op, len, request); 90347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->mhdr->client_context = re->regrec_client_context; 90447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->flags = dnssd_htonl(0); 90547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, rr->resrec.InterfaceID, mDNSfalse)); 90647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->error = dnssd_htonl(result); 90747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(request, reply); 90847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 90947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 91047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result) 91147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 91247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // unlink from list, free memory 91347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry **ptr = &request->u.reg_recs; 91447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr && (*ptr) != re) ptr = &(*ptr)->next; 91547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) { LogMsg("regrecord_callback - record not in list!"); return; } 91647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = (*ptr)->next; 91747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry AuthRecord regrecord_callback", re->rr); 91847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry regrecord_callback", re); 91947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 92047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 92147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 92247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (re->external_advertise) LogMsg("regrecord_callback: external_advertise already set!"); 92347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 92447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (re->origInterfaceID == mDNSInterface_P2P || (!re->origInterfaceID && IsLocalDomain(&rr->namestorage) && (request->flags & kDNSServiceFlagsIncludeP2P))) 92547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 92647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("regrecord_callback: calling external_start_advertising_service"); 92747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&rr->resrec); 92847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->external_advertise = mDNStrue; 92947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 93047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 93147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 93247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 93347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 93447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void connection_termination(request_state *request) 93547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 93647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // When terminating a shared connection, we need to scan the all_requests list 93747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // and terminate any subbordinate operations sharing this file descriptor 93847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state **req = &all_requests; 93947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 94047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceCreateConnection STOP", request->sd); 94147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 94247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*req) 94347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 94447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((*req)->primary == request) 94547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 94647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 94747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *tmp = *req; 94847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (tmp->primary == tmp) LogMsg("connection_termination ERROR (*req)->primary == *req for %p %d", tmp, tmp->sd); 94947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (tmp->replies) LogMsg("connection_termination ERROR How can subordinate req %p %d have replies queued?", tmp, tmp->sd); 95047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort_request(tmp); 95147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *req = tmp->next; 95247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("request_state/connection_termination", tmp); 95347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 95447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 95547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req = &(*req)->next; 95647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 95747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 95847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (request->u.reg_recs) 95947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 96047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *ptr = request->u.reg_recs; 96147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegisterRecord(%u %s) STOP", request->sd, ptr->key, RRDisplayString(&mDNSStorage, &ptr->rr->resrec)); 96247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.reg_recs = request->u.reg_recs->next; 96347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ptr->rr->RecordContext = NULL; 96447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ptr->external_advertise) 96547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 96647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ptr->external_advertise = mDNSfalse; 96747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&ptr->rr->resrec); 96847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 96947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Deregister(&mDNSStorage, ptr->rr); // Will free ptr->rr for us 97047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry/connection_termination", ptr); 97147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 97247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 97347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 97447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void handle_cancel_request(request_state *request) 97547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 97647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state **req = &all_requests; 97747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: Cancel %08X %08X", request->sd, request->hdr.client_context.u32[1], request->hdr.client_context.u32[0]); 97847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*req) 97947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 98047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((*req)->primary == request && 98147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*req)->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] && 98247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*req)->hdr.client_context.u32[1] == request->hdr.client_context.u32[1]) 98347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 98447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 98547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *tmp = *req; 98647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort_request(tmp); 98747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *req = tmp->next; 98847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("request_state/handle_cancel_request", tmp); 98947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 99047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 99147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req = &(*req)->next; 99247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 99347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 99447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 99547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_regrecord_request(request_state *request) 99647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 99747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_BadParamErr; 99847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr = read_rr_from_ipc_msg(request, 1, 1); 99947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr) 100047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 100147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *re; 100247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't allow non-local domains to be regsitered as LocalOnly. Allowing this would permit 100347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // clients to register records such as www.bigbank.com A w.x.y.z to redirect Safari. 100447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.InterfaceID == mDNSInterface_LocalOnly && !IsLocalDomain(rr->resrec.name) && 100547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rrclass == kDNSClass_IN && (rr->resrec.rrtype == kDNSType_A || rr->resrec.rrtype == kDNSType_AAAA || 100647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rrtype == kDNSType_CNAME)) 100747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 100847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("AuthRecord/handle_regrecord_request", rr); 100947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return (mStatus_BadParamErr); 101047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 101147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // allocate registration entry, link into list 101247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re = mallocL("registered_record_entry", sizeof(registered_record_entry)); 101347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!re) FatalError("ERROR: malloc"); 101447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->key = request->hdr.reg_index; 101547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->rr = rr; 101647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->regrec_client_context = request->hdr.client_context; 101747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->request = request; 101847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->external_advertise = mDNSfalse; 101947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->RecordContext = re; 102047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->RecordCallback = regrecord_callback; 102147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 102247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->origInterfaceID = rr->resrec.InterfaceID; 102347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.InterfaceID == mDNSInterface_P2P) rr->resrec.InterfaceID = mDNSInterface_Any; 102447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 102547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AuthorizedDomain(request, rr->resrec.name, AutoRegistrationDomains)) return (mStatus_NoError); 102647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 102747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rroriginalttl == 0) 102847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr->resrec.rroriginalttl = DefaultTTLforRRType(rr->resrec.rrtype); 102947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 103047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegisterRecord(%u %s) START", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec)); 103147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_Register(&mDNSStorage, rr); 103247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) 103347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 103447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegisterRecord(%u %s) ERROR (%d)", request->sd, re->key, RRDisplayString(&mDNSStorage, &rr->resrec), err); 103547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry", re); 103647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry/AuthRecord", rr); 103747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 103847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 103947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 104047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt re->next = request->u.reg_recs; 104147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.reg_recs = re; 104247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 104347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 104447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 104547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 104647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 104747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void UpdateDeviceInfoRecord(mDNS *const m); 104847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 104947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void regservice_termination_callback(request_state *request) 105047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 105147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request) { LogMsg("regservice_termination_callback context is NULL"); return; } 105247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (request->u.servicereg.instances) 105347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 105447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *p = request->u.servicereg.instances; 105547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.instances = request->u.servicereg.instances->next; 105647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // only safe to free memory if registration is not valid, i.e. deregister fails (which invalidates p) 105747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegister(%##s, %u) STOP", 105847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, p->srs.RR_SRV.resrec.name->c, mDNSVal16(p->srs.RR_SRV.resrec.rdata->u.srv.port)); 105947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 106047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_helper(p); 106147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 106247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Clear backpointer *before* calling mDNS_DeregisterService/unlink_and_free_service_instance 106347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We don't need unlink_and_free_service_instance to cut its element from the list, because we're already advancing 106447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // request->u.servicereg.instances as we work our way through the list, implicitly cutting one element at a time 106547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We can't clear p->request *after* the calling mDNS_DeregisterService/unlink_and_free_service_instance 106647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // because by then we might have already freed p 106747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt p->request = NULL; 106847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNS_DeregisterService(&mDNSStorage, &p->srs)) unlink_and_free_service_instance(p); 106947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't touch service_instance *p after this -- it's likely to have been freed already 107047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 107147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.txtdata) 107247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; } 107347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.autoname) 107447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 107547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Clear autoname before calling UpdateDeviceInfoRecord() so it doesn't mistakenly include this in its count of active autoname registrations 107647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.autoname = mDNSfalse; 107747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt UpdateDeviceInfoRecord(&mDNSStorage); 107847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 107947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 108047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 108147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal request_state *LocateSubordinateRequest(request_state *request) 108247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 108347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req; 108447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (req = all_requests; req; req = req->next) 108547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary == request && 108647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->hdr.client_context.u32[0] == request->hdr.client_context.u32[0] && 108747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->hdr.client_context.u32[1] == request->hdr.client_context.u32[1]) return(req); 108847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(request); 108947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 109047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 109147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus add_record_to_service(request_state *request, service_instance *instance, mDNSu16 rrtype, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl) 109247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 109347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ServiceRecordSet *srs = &instance->srs; 109447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus result; 109547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int size = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 109647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *extra = mallocL("ExtraResourceRecord", sizeof(*extra) - sizeof(RDataBody) + size); 109747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!extra) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; } 109847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 109947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(extra, sizeof(ExtraResourceRecord)); // OK if oversized rdata not zero'd 110047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt extra->r.resrec.rrtype = rrtype; 110147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt extra->r.rdatastorage.MaxRDLength = (mDNSu16) size; 110247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt extra->r.resrec.rdlength = rdlen; 110347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(&extra->r.rdatastorage.u.data, rdata, rdlen); 110447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 110547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = mDNS_AddRecordToService(&mDNSStorage, srs, extra, &extra->r.rdatastorage, ttl, 110647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (request->flags & kDNSServiceFlagsIncludeP2P) ? 1: 0); 110747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result) { freeL("ExtraResourceRecord/add_record_to_service", extra); return result; } 110847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 110947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt extra->ClientID = request->hdr.reg_index; 111047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (instance->external_advertise && (instance->request->u.servicereg.InterfaceID == mDNSInterface_P2P || (!instance->request->u.servicereg.InterfaceID && SameDomainName(&instance->domain, &localdomain) && (instance->request->flags & kDNSServiceFlagsIncludeP2P)))) 111147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 111247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("add_record_to_service: calling external_start_advertising_service"); 111347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&extra->r.resrec); 111447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 111547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return result; 111647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 111747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 111847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_add_request(request_state *request) 111947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 112047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *i; 112147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus result = mStatus_UnknownErr; 112247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend); 112347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rrtype = get_uint16(&request->msgptr, request->msgend); 112447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 112547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 112647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend); 112747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!ttl) ttl = DefaultTTLforRRType(rrtype); 112847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)flags; // Unused 112947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 113047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceAddRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 113147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 113247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 113347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 113447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 113547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate != regservice_termination_callback) 113647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceAddRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 113747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 113847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For a service registered with zero port, don't allow adding records. This mostly happens due to a bug 113947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // in the application. See radar://9165807. 114047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSIPPortIsZero(request->u.servicereg.port)) 114147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceAddRecord: adding record to a service registered with zero port", request->sd); return(mStatus_BadParamErr); } 114247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 114347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceAddRecord(%X, %##s, %s, %d)", request->sd, flags, 114447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, DNSTypeName(rrtype), rdlen); 114547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 114647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (i = request->u.servicereg.instances; i; i = i->next) 114747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 114847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = add_record_to_service(request, i, rrtype, rdlen, rdata, ttl); 114947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result && i->default_local) break; 115047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else result = mStatus_NoError; // suppress non-local default errors 115147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 115247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 115347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(result); 115447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 115547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 115647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void update_callback(mDNS *const m, AuthRecord *const rr, RData *oldrd, mDNSu16 oldrdlen) 115747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 115847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool external_advertise = (rr->UpdateContext) ? *((mDNSBool *)rr->UpdateContext) : mDNSfalse; 115947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 116047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 116147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // There are three cases. 116247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 116347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 1. We have updated the primary TXT record of the service 116447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 2. We have updated the TXT record that was added to the service using DNSServiceAddRecord 116547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 3. We have updated the TXT record that was registered using DNSServiceRegisterRecord 116647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 116747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // external_advertise is set if we have advertised at least once during the initial addition 116847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // of the record in all of the three cases above. We should have checked for InterfaceID/LocalDomain 116947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // checks during the first time and hence we don't do any checks here 117047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (external_advertise) 117147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 117247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ResourceRecord ext = rr->resrec; 117347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ext.rdlength == oldrdlen && mDNSPlatformMemSame(&ext.rdata->u, &oldrd->u, oldrdlen)) goto exit; 117447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SetNewRData(&ext, oldrd, oldrdlen); 117547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&ext); 117647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("update_callback: calling external_start_advertising_service"); 117747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_advertising_service(&rr->resrec); 117847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 117947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltexit: 118047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (oldrd != &rr->rdatastorage) freeL("RData/update_callback", oldrd); 118147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 118247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 118347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus update_record(AuthRecord *rr, mDNSu16 rdlen, const char *rdata, mDNSu32 ttl, const mDNSBool *const external_advertise) 118447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 118547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus result; 118647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const int rdsize = rdlen > sizeof(RDataBody) ? rdlen : sizeof(RDataBody); 118747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RData *newrd = mallocL("RData/update_record", sizeof(RData) - sizeof(RDataBody) + rdsize); 118847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!newrd) FatalError("ERROR: malloc"); 118947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newrd->MaxRDLength = (mDNSu16) rdsize; 119047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(&newrd->u, rdata, rdlen); 119147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 119247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // BIND named (name daemon) doesn't allow TXT records with zero-length rdata. This is strictly speaking correct, 119347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // since RFC 1035 specifies a TXT record as "One or more <character-string>s", not "Zero or more <character-string>s". 119447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Since some legacy apps try to create zero-length TXT records, we'll silently correct it here. 119547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr->resrec.rrtype == kDNSType_TXT && rdlen == 0) { rdlen = 1; newrd->u.txt.c[0] = 0; } 119647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 119747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (external_advertise) rr->UpdateContext = (void *)external_advertise; 119847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 119947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = mDNS_Update(&mDNSStorage, rr, ttl, rdlen, newrd, update_callback); 120047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result) { LogMsg("update_record: Error %d for %s", (int)result, ARDisplayString(&mDNSStorage, rr)); freeL("RData/update_record", newrd); } 120147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return result; 120247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 120347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 120447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_update_request(request_state *request) 120547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 120647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const ipc_msg_hdr *const hdr = &request->hdr; 120747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus result = mStatus_BadReferenceErr; 120847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *i; 120947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr = NULL; 121047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 121147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // get the message data 121247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags (&request->msgptr, request->msgend); // flags unused 121347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rdlen = get_uint16(&request->msgptr, request->msgend); 121447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *rdata = get_rdata (&request->msgptr, request->msgend, rdlen); 121547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ttl = get_uint32(&request->msgptr, request->msgend); 121647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)flags; // Unused 121747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 121847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceUpdateRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 121947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 122047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 122147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 122247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 122347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == connection_termination) 122447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 122547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // update an individually registered record 122647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *reptr; 122747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (reptr = request->u.reg_recs; reptr; reptr = reptr->next) 122847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 122947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (reptr->key == hdr->reg_index) 123047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 123147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = update_record(reptr->rr, rdlen, rdata, ttl, &reptr->external_advertise); 123247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s)", 123347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, reptr->rr->resrec.name->c, reptr->rr ? DNSTypeName(reptr->rr->resrec.rrtype) : "<NONE>"); 123447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto end; 123547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 123647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 123747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = mStatus_BadReferenceErr; 123847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto end; 123947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 124047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 124147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate != regservice_termination_callback) 124247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceUpdateRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 124347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 124447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For a service registered with zero port, only SRV record is initialized. Don't allow any updates. 124547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSIPPortIsZero(request->u.servicereg.port)) 124647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceUpdateRecord: updating the record of a service registered with zero port", request->sd); return(mStatus_BadParamErr); } 124747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 124847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // update the saved off TXT data for the service 124947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (hdr->reg_index == TXT_RECORD_INDEX) 125047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 125147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.txtdata) 125247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { freeL("service_info txtdata", request->u.servicereg.txtdata); request->u.servicereg.txtdata = NULL; } 125347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rdlen > 0) 125447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 125547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtdata = mallocL("service_info txtdata", rdlen); 125647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_update_request - malloc"); 125747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(request->u.servicereg.txtdata, rdata, rdlen); 125847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 125947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtlen = rdlen; 126047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 126147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 126247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // update a record from a service record set 126347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (i = request->u.servicereg.instances; i; i = i->next) 126447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 126547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (hdr->reg_index == TXT_RECORD_INDEX) rr = &i->srs.RR_TXT; 126647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 126747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 126847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *e; 126947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (e = i->srs.Extras; e; e = e->next) 127047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (e->ClientID == hdr->reg_index) { rr = &e->r; break; } 127147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 127247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 127347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!rr) { result = mStatus_BadReferenceErr; goto end; } 127447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = update_record(rr, rdlen, rdata, ttl, &i->external_advertise); 127547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result && i->default_local) goto end; 127647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else result = mStatus_NoError; // suppress non-local default errors 127747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 127847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 127947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltend: 128047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == regservice_termination_callback) 128147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceUpdateRecord(%##s, %s)", request->sd, 128247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, 128347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rr ? DNSTypeName(rr->resrec.rrtype) : "<NONE>"); 128447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 128547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(result); 128647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 128747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 128847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// remove a resource record registered via DNSServiceRegisterRecord() 128947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus remove_record(request_state *request) 129047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 129147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_UnknownErr; 129247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *e, **ptr = &request->u.reg_recs; 129347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 129447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr && (*ptr)->key != request->hdr.reg_index) ptr = &(*ptr)->next; 129547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) { LogMsg("%3d: DNSServiceRemoveRecord(%u) not found", request->sd, request->hdr.reg_index); return mStatus_BadReferenceErr; } 129647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt e = *ptr; 129747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = e->next; // unlink 129847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 129947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRemoveRecord(%u %s)", request->sd, e->key, RRDisplayString(&mDNSStorage, &e->rr->resrec)); 130047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt e->rr->RecordContext = NULL; 130147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (e->external_advertise) 130247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 130347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_advertising_service(&e->rr->resrec); 130447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt e->external_advertise = mDNSfalse; 130547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 130647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_Deregister(&mDNSStorage, e->rr); // Will free e->rr for us; we're responsible for freeing e 130747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) 130847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 130947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: remove_record, mDNS_Deregister: %d", err); 131047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry AuthRecord remove_record", e->rr); 131147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 131247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 131347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("registered_record_entry remove_record", e); 131447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return err; 131547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 131647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 131747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus remove_extra(const request_state *const request, service_instance *const serv, mDNSu16 *const rrtype) 131847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 131947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_BadReferenceErr; 132047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ExtraResourceRecord *ptr; 132147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 132247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ptr = serv->srs.Extras; ptr; ptr = ptr->next) 132347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 132447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ptr->ClientID == request->hdr.reg_index) // found match 132547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 132647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *rrtype = ptr->r.resrec.rrtype; 132747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (serv->external_advertise) external_stop_advertising_service(&ptr->r.resrec); 132847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_RemoveRecordFromService(&mDNSStorage, &serv->srs, ptr, FreeExtraRR, ptr); 132947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt break; 133047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 133147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 133247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return err; 133347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 133447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 133547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_removerecord_request(request_state *request) 133647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 133747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_BadReferenceErr; 133847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_flags(&request->msgptr, request->msgend); // flags unused 133947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 134047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceRemoveRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 134147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 134247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If this is a shared connection, check if the operation actually applies to a subordinate request_state object 134347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == connection_termination) request = LocateSubordinateRequest(request); 134447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 134547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate == connection_termination) 134647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = remove_record(request); // remove individually registered record 134747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (request->terminate != regservice_termination_callback) 134847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceRemoveRecord(not a registered service ref)", request->sd); return(mStatus_BadParamErr); } 134947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 135047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 135147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *i; 135247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rrtype = 0; 135347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRemoveRecord(%##s, %s)", request->sd, 135447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (request->u.servicereg.instances) ? request->u.servicereg.instances->srs.RR_SRV.resrec.name->c : NULL, 135547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rrtype ? DNSTypeName(rrtype) : "<NONE>"); 135647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (i = request->u.servicereg.instances; i; i = i->next) 135747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 135847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = remove_extra(request, i, &rrtype); 135947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err && i->default_local) break; 136047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else err = mStatus_NoError; // suppress non-local default errors 136147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 136247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 136347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 136447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 136547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 136647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 136747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// If there's a comma followed by another character, 136847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// FindFirstSubType overwrites the comma with a nul and returns the pointer to the next character. 136947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Otherwise, it returns a pointer to the final nul at the end of the string 137047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal char *FindFirstSubType(char *p) 137147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 137247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p) 137347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 137447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (p[0] == '\\' && p[1]) p += 2; 137547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (p[0] == ',' && p[1]) { *p++ = 0; return(p); } 137647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else p++; 137747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 137847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(p); 137947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 138047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 138147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// If there's a comma followed by another character, 138247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// FindNextSubType overwrites the comma with a nul and returns the pointer to the next character. 138347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// If it finds an illegal unescaped dot in the subtype name, it returns mDNSNULL 138447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Otherwise, it returns a pointer to the final nul at the end of the string 138547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal char *FindNextSubType(char *p) 138647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 138747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p) 138847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 138947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (p[0] == '\\' && p[1]) // If escape character 139047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt p += 2; // ignore following character 139147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (p[0] == ',') // If we found a comma 139247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 139347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (p[1]) *p++ = 0; 139447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(p); 139547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 139647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (p[0] == '.') 139747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(mDNSNULL); 139847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else p++; 139947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 140047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(p); 140147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 140247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 140347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Returns -1 if illegal subtype found 140447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport mDNSs32 ChopSubTypes(char *regtype) 140547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 140647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 NumSubTypes = 0; 140747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *stp = FindFirstSubType(regtype); 140847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (stp && *stp) // If we found a comma... 140947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 141047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*stp == ',') return(-1); 141147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt NumSubTypes++; 141247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt stp = FindNextSubType(stp); 141347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 141447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!stp) return(-1); 141547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(NumSubTypes); 141647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 141747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 141847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport AuthRecord *AllocateSubTypes(mDNSs32 NumSubTypes, char *p) 141947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 142047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *st = mDNSNULL; 142147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (NumSubTypes) 142247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 142347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 i; 142447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt st = mallocL("ServiceSubTypes", NumSubTypes * sizeof(AuthRecord)); 142547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!st) return(mDNSNULL); 142647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (i = 0; i < NumSubTypes; i++) 142747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 142847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_SetupResourceRecord(&st[i], mDNSNULL, mDNSInterface_Any, kDNSQType_ANY, kStandardTTL, 0, AuthRecordAny, mDNSNULL, mDNSNULL); 142947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p) p++; 143047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt p++; 143147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&st[i].namestorage, p)) 143247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { freeL("ServiceSubTypes", st); return(mDNSNULL); } 143347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 143447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 143547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(st); 143647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 143747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 143847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus register_service_instance(request_state *request, const domainname *domain) 143947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 144047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance **ptr, *instance; 144147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const int extra_size = (request->u.servicereg.txtlen > sizeof(RDataBody)) ? (request->u.servicereg.txtlen - sizeof(RDataBody)) : 0; 144247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const mDNSBool DomainIsLocal = SameDomainName(domain, &localdomain); 144347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus result; 144447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID interfaceID = request->u.servicereg.InterfaceID; 144547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 regFlags = 0; 144647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 144747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceID == mDNSInterface_P2P) 144847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 144947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt interfaceID = mDNSInterface_Any; 145047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt regFlags |= regFlagIncludeP2P; 145147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 145247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (request->flags & kDNSServiceFlagsIncludeP2P) 145347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt regFlags |= regFlagIncludeP2P; 145447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 145547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // client guarantees that record names are unique 145647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->flags & kDNSServiceFlagsForce) 145747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt regFlags |= regFlagKnownUnique; 145847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 145947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the client specified an interface, but no domain, then we honor the specified interface for the "local" (mDNS) 146047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // registration but for the wide-area registrations we don't (currently) have any concept of a wide-area unicast 146147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // registrations scoped to a specific interface, so for the automatic domains we add we must *not* specify an interface. 146247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // (Specifying an interface with an apparently wide-area domain (i.e. something other than "local") 146347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // currently forces the registration to use mDNS multicast despite the apparently wide-area domain.) 146447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.default_domain && !DomainIsLocal) interfaceID = mDNSInterface_Any; 146547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 146647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ptr = &request->u.servicereg.instances; *ptr; ptr = &(*ptr)->next) 146747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 146847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&(*ptr)->domain, domain)) 146947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 147047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("register_service_instance: domain %##s already registered for %#s.%##s", 147147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domain->c, &request->u.servicereg.name, &request->u.servicereg.type); 147247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mStatus_AlreadyRegistered; 147347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 147447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 147547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 147647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSStorage.KnownBugs & mDNS_KnownBug_LimitedIPv6) 147747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 147847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Special-case hack: On Mac OS X 10.6.x and earlier we don't advertise SMB service in AutoTunnel domains, 147947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // because AutoTunnel services have to support IPv6, and in Mac OS X 10.6.x the SMB server does not. 148047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // <rdar://problem/5482322> BTMM: Don't advertise SMB with BTMM because it doesn't support IPv6 148147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&request->u.servicereg.type, (const domainname *) "\x4" "_smb" "\x4" "_tcp")) 148247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 148347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DomainAuthInfo *AuthInfo = GetAuthInfoForName(&mDNSStorage, domain); 148447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (AuthInfo && AuthInfo->AutoTunnel) return(kDNSServiceErr_Unsupported); 148547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 148647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 148747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 148847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance = mallocL("service_instance", sizeof(*instance) + extra_size); 148947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!instance) { my_perror("ERROR: malloc"); return mStatus_NoMemoryErr; } 149047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 149147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->next = mDNSNULL; 149247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->request = request; 149347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->subtypes = AllocateSubTypes(request->u.servicereg.num_subtypes, request->u.servicereg.type_as_string); 149447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->renameonmemfree = 0; 149547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->clientnotified = mDNSfalse; 149647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->default_local = (request->u.servicereg.default_domain && DomainIsLocal); 149747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->external_advertise = mDNSfalse; 149847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&instance->domain, domain); 149947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 150047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.num_subtypes && !instance->subtypes) 150147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { unlink_and_free_service_instance(instance); instance = NULL; FatalError("ERROR: malloc"); } 150247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 150347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = mDNS_RegisterService(&mDNSStorage, &instance->srs, 150447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &request->u.servicereg.name, &request->u.servicereg.type, domain, 150547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.host.c[0] ? &request->u.servicereg.host : NULL, 150647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.port, 150747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtdata, request->u.servicereg.txtlen, 150847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->subtypes, request->u.servicereg.num_subtypes, 150947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt interfaceID, regservice_callback, instance, regFlags); 151047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 151147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!result) 151247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 151347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = instance; // Append this to the end of our request->u.servicereg.instances list 151447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegister(%##s, %u) ADDED", 151547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt instance->request->sd, instance->srs.RR_SRV.resrec.name->c, mDNSVal16(request->u.servicereg.port)); 151647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 151747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 151847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 151947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("register_service_instance %#s.%##s%##s error %d", 152047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &request->u.servicereg.name, &request->u.servicereg.type, domain->c, result); 152147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt unlink_and_free_service_instance(instance); 152247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 152347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 152447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return result; 152547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 152647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 152747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void udsserver_default_reg_domain_changed(const DNameListElem *const d, const mDNSBool add) 152847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 152947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request; 153047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 153147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 153247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt machserver_automatic_registration_domain_changed(&d->name, add); 153347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 153447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 153547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%s registration domain %##s", add ? "Adding" : "Removing", d->name.c); 153647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (request = all_requests; request; request = request->next) 153747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 153847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate != regservice_termination_callback) continue; 153947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->u.servicereg.default_domain) continue; 154047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!d->uid || SystemUID(request->uid) || request->uid == d->uid) 154147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 154247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance **ptr = &request->u.servicereg.instances; 154347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next; 154447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (add) 154547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 154647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we don't already have this domain in our list for this registration, add it now 154747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) register_service_instance(request, &d->name); 154847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else debugf("udsserver_default_reg_domain_changed %##s already in list, not re-adding", &d->name); 154947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 155047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 155147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 155247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Normally we should not fail to find the specified instance 155347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // One case where this can happen is if a uDNS update fails for some reason, 155447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // and regservice_callback then calls unlink_and_free_service_instance and disposes of that instance. 155547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) 155647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("udsserver_default_reg_domain_changed domain %##s not found for service %#s type %s", 155747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &d->name, request->u.servicereg.name.c, request->u.servicereg.type_as_string); 155847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 155947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 156047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *p; 156147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p = AutoRegistrationDomains; p; p=p->next) 156247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!p->uid || SystemUID(request->uid) || request->uid == p->uid) 156347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&d->name, &p->name)) break; 156447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (p) debugf("udsserver_default_reg_domain_changed %##s still in list, not removing", &d->name); 156547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 156647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 156747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 156847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *si = *ptr; 156947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = si->next; 157047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (si->clientnotified) SendServiceRemovalNotification(&si->srs); // Do this *before* clearing si->request backpointer 157147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Now that we've cut this service_instance from the list, we MUST clear the si->request backpointer. 157247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Otherwise what can happen is this: While our mDNS_DeregisterService is in the 157347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // process of completing asynchronously, the client cancels the entire operation, so 157447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // regservice_termination_callback then runs through the whole list deregistering each 157547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // instance, clearing the backpointers, and then disposing the parent request_state object. 157647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // However, because this service_instance isn't in the list any more, regservice_termination_callback 157747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // has no way to find it and clear its backpointer, and then when our mDNS_DeregisterService finally 157847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // completes later with a mStatus_MemFree message, it calls unlink_and_free_service_instance() with 157947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // a service_instance with a stale si->request backpointer pointing to memory that's already been freed. 158047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt si->request = NULL; 158147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_DeregisterService(&mDNSStorage, &si->srs); 158247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) { LogMsg("udsserver_default_reg_domain_changed err %d", err); unlink_and_free_service_instance(si); } 158347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 158947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 159047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_regservice_request(request_state *request) 159147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 159247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char name[256]; // Lots of spare space for extra-long names that we'll auto-truncate down to 63 bytes 159347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char domain[MAX_ESCAPED_DOMAIN_NAME], host[MAX_ESCAPED_DOMAIN_NAME]; 159447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char type_as_string[MAX_ESCAPED_DOMAIN_NAME]; 159547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname d, srv; 159647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 15975e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai const char *msgTXTData; 159847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 159947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 160047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 160147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 160247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) 160347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - Couldn't find interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); } 160447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 160547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, name, sizeof(name)) < 0 || 160647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, type_as_string, MAX_ESCAPED_DOMAIN_NAME) < 0 || 160747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0 || 160847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, host, MAX_ESCAPED_DOMAIN_NAME) < 0) 160947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); } 161047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 161147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->flags = flags; 161247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.InterfaceID = InterfaceID; 161347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.instances = NULL; 161447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtlen = 0; 161547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtdata = NULL; 16165e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai mDNSPlatformStrLCopy(request->u.servicereg.type_as_string, type_as_string, sizeof(request->u.servicereg.type_as_string)); 161747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 161847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->msgptr + 2 > request->msgend) request->msgptr = NULL; 161947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 162047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 162147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.port.b[0] = *request->msgptr++; 162247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.port.b[1] = *request->msgptr++; 162347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 162447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 162547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtlen = get_uint16(&request->msgptr, request->msgend); 16265e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai msgTXTData = get_rdata(&request->msgptr, request->msgend, request->u.servicereg.txtlen); 16275e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai if (!request->msgptr) 16285e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai { 16295e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai LogMsg("%3d: DNSServiceRegister(unreadable parameters)", request->sd); 16305e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai return(mStatus_BadParamErr); 16315e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai } 16325e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai 163347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.txtlen) 163447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 163547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.txtdata = mallocL("service_info txtdata", request->u.servicereg.txtlen); 163647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->u.servicereg.txtdata) FatalError("ERROR: handle_regservice_request - malloc"); 16375e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai mDNSPlatformMemCopy(request->u.servicereg.txtdata, msgTXTData, request->u.servicereg.txtlen); 163847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 163947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 164047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Check for sub-types after the service type 164147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.num_subtypes = ChopSubTypes(request->u.servicereg.type_as_string); // Note: Modifies regtype string to remove trailing subtypes 164247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.num_subtypes < 0) 164347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - ChopSubTypes failed %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); } 164447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 164547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't try to construct "domainname t" until *after* ChopSubTypes has worked its magic 164647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*request->u.servicereg.type_as_string || !MakeDomainNameFromDNSNameString(&request->u.servicereg.type, request->u.servicereg.type_as_string)) 164747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - type_as_string bad %s", request->u.servicereg.type_as_string); return(mStatus_BadParamErr); } 164847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 164947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!name[0]) 165047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 165147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.name = mDNSStorage.nicelabel; 165247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.autoname = mDNStrue; 165347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 165447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 165547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 165647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the client is allowing AutoRename, then truncate name to legal length before converting it to a DomainLabel 165747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((flags & kDNSServiceFlagsNoAutoRename) == 0) 165847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 165947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int newlen = TruncateUTF8ToLength((mDNSu8*)name, mDNSPlatformStrLen(name), MAX_DOMAIN_LABEL); 166047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt name[newlen] = 0; 166147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 166247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainLabelFromLiteralString(&request->u.servicereg.name, name)) 166347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - name bad %s", name); return(mStatus_BadParamErr); } 166447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.autoname = mDNSfalse; 166547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 166647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 166747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*domain) 166847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 166947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.default_domain = mDNSfalse; 167047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&d, domain)) 167147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - domain bad %s", domain); return(mStatus_BadParamErr); } 167247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 167347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 167447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 167547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.default_domain = mDNStrue; 167647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt MakeDomainNameFromDNSNameString(&d, "local."); 167747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 167847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 167947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!ConstructServiceName(&srv, &request->u.servicereg.name, &request->u.servicereg.type, &d)) 168047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 168147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: handle_regservice_request - Couldn't ConstructServiceName from, “%#s” “%##s” “%##s”", 168247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.name.c, request->u.servicereg.type.c, d.c); return(mStatus_BadParamErr); 168347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 168447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 168547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&request->u.servicereg.host, host)) 168647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_regservice_request - host bad %s", host); return(mStatus_BadParamErr); } 168747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.autorename = (flags & kDNSServiceFlagsNoAutoRename ) == 0; 168847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.servicereg.allowremotequery = (flags & kDNSServiceFlagsAllowRemoteQuery) != 0; 168947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 169047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Some clients use mDNS for lightweight copy protection, registering a pseudo-service with 169147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // a port number of zero. When two instances of the protected client are allowed to run on one 169247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // machine, we don't want to see misleading "Bogus client" messages in syslog and the console. 169347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!mDNSIPPortIsZero(request->u.servicereg.port)) 169447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 169547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int count = CountExistingRegistrations(&srv, request->u.servicereg.port); 169647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (count) 169747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("Client application registered %d identical instances of service %##s port %u.", 169847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt count+1, srv.c, mDNSVal16(request->u.servicereg.port)); 169947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 170047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 170147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceRegister(%X, %d, \"%s\", \"%s\", \"%s\", \"%s\", %u) START", 170247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, flags, interfaceIndex, name, request->u.servicereg.type_as_string, domain, host, mDNSVal16(request->u.servicereg.port)); 170347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 170447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We need to unconditionally set request->terminate, because even if we didn't successfully 170547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // start any registrations right now, subsequent configuration changes may cause successful 170647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // registrations to be added, and we'll need to cancel them before freeing this memory. 170747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We also need to set request->terminate first, before adding additional service instances, 170847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // because the uds_validatelists uses the request->terminate function pointer to determine 170947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // what kind of request this is, and therefore what kind of list validation is required. 171047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->terminate = regservice_termination_callback; 171147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 171247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = register_service_instance(request, &d); 171347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 171447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 171547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = AuthorizedDomain(request, &d, AutoRegistrationDomains) ? register_service_instance(request, &d) : mStatus_NoError; 171647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 171747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err) 171847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 171947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.servicereg.autoname) UpdateDeviceInfoRecord(&mDNSStorage); 172047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 172147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*domain) 172247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 172347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *ptr; 172447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note that we don't report errors for non-local, non-explicit domains 172547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ptr = AutoRegistrationDomains; ptr; ptr = ptr->next) 172647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!ptr->uid || SystemUID(request->uid) || request->uid == ptr->uid) 172747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt register_service_instance(request, &ptr->name); 172847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 172947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 173047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 173147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 173247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 173347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 173447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 173547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 173647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 173747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceBrowse 173847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 173947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 174047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void FoundInstance(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 174147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 174247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const DNSServiceFlags flags = AddRecord ? kDNSServiceFlagsAdd : 0; 174347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req = question->QuestionContext; 174447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 174547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 174647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 174747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype != kDNSType_PTR) 174847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: FoundInstance: Should not be called with rrtype %d (not a PTR record)", req->sd, answer->rrtype); return; } 174947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 175047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (GenerateNTDResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError) != mStatus_NoError) 175147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 175247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&req->u.browser.regtype, (const domainname*)"\x09_services\x07_dns-sd\x04_udp")) 175347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 175447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Special support to enable the DNSServiceBrowse call made by Bonjour Browser 175547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Remove after Bonjour Browser is updated to use DNSServiceQueryRecord instead of DNSServiceBrowse 175647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt GenerateBonjourBrowserResponse(&answer->rdata->u.name, answer->InterfaceID, req, &rep, browse_reply_op, flags, mStatus_NoError); 175747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto bonjourbrowserhack; 175847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 175947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 176047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: FoundInstance: %##s PTR %##s received from network is not valid DNS-SD service pointer", 176147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, answer->name->c, answer->rdata->u.name.c); 176247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 176347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 176447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 176547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltbonjourbrowserhack: 176647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 176747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceBrowse(%##s, %s) RESULT %s %d: %s", 176847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, question->qname.c, DNSTypeName(question->qtype), AddRecord ? "Add" : "Rmv", 176947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse), RRDisplayString(m, answer)); 177047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 177147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(req, rep); 177247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 177347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 177447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus add_domain_to_browser(request_state *info, const domainname *d) 177547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 177647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *b, *p; 177747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 177847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 177947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p = info->u.browser.browsers; p; p = p->next) 178047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 178147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&p->domain, d)) 178247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { debugf("add_domain_to_browser %##s already in list", d->c); return mStatus_AlreadyRegistered; } 178347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 178447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 178547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt b = mallocL("browser_t", sizeof(*b)); 178647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!b) return mStatus_NoMemoryErr; 178747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&b->domain, d); 178847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartBrowse(&mDNSStorage, &b->q, 178947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &info->u.browser.regtype, d, info->u.browser.interface_id, info->u.browser.ForceMCast, FoundInstance, info); 179047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) 179147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 179247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("mDNS_StartBrowse returned %d for type %##s domain %##s", err, info->u.browser.regtype.c, d->c); 179347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("browser_t/add_domain_to_browser", b); 179447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 179547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 179647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 179747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt b->next = info->u.browser.browsers; 179847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt info->u.browser.browsers = b; 179947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceBrowse(%##s) START", info->sd, b->q.qname.c); 180047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (info->u.browser.interface_id == mDNSInterface_P2P || (!info->u.browser.interface_id && SameDomainName(&b->domain, &localdomain) && (info->flags & kDNSServiceFlagsIncludeP2P))) 180147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 180247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname tmp; 180347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &b->domain); 180447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("add_domain_to_browser: calling external_start_browsing_for_service()"); 180547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_browsing_for_service(&mDNSStorage, &tmp, kDNSType_PTR); 180647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 180747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 180847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return err; 180947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 181047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 181147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void browse_termination_callback(request_state *info) 181247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 181347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (info->u.browser.browsers) 181447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 181547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *ptr = info->u.browser.browsers; 181647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 181747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (info->u.browser.interface_id == mDNSInterface_P2P || (!info->u.browser.interface_id && SameDomainName(&ptr->domain, &localdomain) && (info->flags & kDNSServiceFlagsIncludeP2P))) 181847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 181947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname tmp; 182047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConstructServiceName(&tmp, NULL, &info->u.browser.regtype, &ptr->domain); 182147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("browse_termination_callback: calling external_stop_browsing_for_service()"); 182247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_browsing_for_service(&mDNSStorage, &tmp, kDNSType_PTR); 182347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 182447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 182547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt info->u.browser.browsers = ptr->next; 182647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceBrowse(%##s) STOP", info->sd, ptr->q.qname.c); 182747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopBrowse(&mDNSStorage, &ptr->q); // no need to error-check result 182847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("browser_t/browse_termination_callback", ptr); 182947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 183047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 183147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 183247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void udsserver_automatic_browse_domain_changed(const DNameListElem *const d, const mDNSBool add) 183347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 183447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request; 183547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("udsserver_automatic_browse_domain_changed: %s default browse domain %##s", add ? "Adding" : "Removing", d->name.c); 183647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 183747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 183847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt machserver_automatic_browse_domain_changed(&d->name, add); 183947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 184047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 184147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (request = all_requests; request; request = request->next) 184247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 184347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->terminate != browse_termination_callback) continue; // Not a browse operation 184447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->u.browser.default_domain) continue; // Not an auto-browse operation 184547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!d->uid || SystemUID(request->uid) || request->uid == d->uid) 184647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 184747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t **ptr = &request->u.browser.browsers; 184847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr && !SameDomainName(&(*ptr)->domain, &d->name)) ptr = &(*ptr)->next; 184947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (add) 185047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 185147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we don't already have this domain in our list for this browse operation, add it now 185247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) add_domain_to_browser(request, &d->name); 185347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else debugf("udsserver_automatic_browse_domain_changed %##s already in list, not re-adding", &d->name); 185447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 185547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 185647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 185747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*ptr) LogMsg("udsserver_automatic_browse_domain_changed ERROR %##s not found", &d->name); 185847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 185947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 186047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *p; 186147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p = AutoBrowseDomains; p; p=p->next) 186247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!p->uid || SystemUID(request->uid) || request->uid == p->uid) 186347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&d->name, &p->name)) break; 186447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (p) debugf("udsserver_automatic_browse_domain_changed %##s still in list, not removing", &d->name); 186547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 186647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 186747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *rem = *ptr; 186847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = (*ptr)->next; 186947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQueryWithRemoves(&mDNSStorage, &rem->q); 187047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("browser_t/udsserver_automatic_browse_domain_changed", rem); 187147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 187747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 187847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void FreeARElemCallback(mDNS *const m, AuthRecord *const rr, mStatus result) 187947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 188047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // unused 188147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result == mStatus_MemFree) 188247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 188347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // On shutdown, mDNS_Close automatically deregisters all records 188447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Since in this case no one has called DeregisterLocalOnlyDomainEnumPTR to cut the record 188547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // from the LocalDomainEnumRecords list, we do this here before we free the memory. 188647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // (This should actually no longer be necessary, now that we do the proper cleanup in 188747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // udsserver_exit. To confirm this, we'll log an error message if we do find a record that 188847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // hasn't been cut from the list yet. If these messages don't appear, we can delete this code.) 188947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem **ptr = &LocalDomainEnumRecords; 189047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr && &(*ptr)->ar != rr) ptr = &(*ptr)->next; 189147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*ptr) { *ptr = (*ptr)->next; LogMsg("FreeARElemCallback: Have to cut %s", ARDisplayString(m, rr)); } 189247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(rr->RecordContext); 189347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 189447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 189547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 189647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// RegisterLocalOnlyDomainEnumPTR and DeregisterLocalOnlyDomainEnumPTR largely duplicate code in 189747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// "FoundDomain" in uDNS.c for creating and destroying these special mDNSInterface_LocalOnly records. 189847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// We may want to turn the common code into a subroutine. 189947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 190047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void RegisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type) 190147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 190247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // allocate/register legacy and non-legacy _browse PTR record 190347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 190447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem *ptr = mDNSPlatformMemAllocate(sizeof(*ptr)); 190547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 190647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("Incrementing %s refcount for %##s", 190747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeBrowse ) ? "browse domain " : 190847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeRegistration ) ? "registration dom" : 190947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c); 191047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 191147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_SetupResourceRecord(&ptr->ar, mDNSNULL, mDNSInterface_LocalOnly, kDNSType_PTR, 7200, kDNSRecordTypeShared, AuthRecordLocalOnly, FreeARElemCallback, ptr); 191247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt MakeDomainNameFromDNSNameString(&ptr->ar.namestorage, mDNS_DomainTypeNames[type]); 191347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AppendDNSNameString (&ptr->ar.namestorage, "local"); 191447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&ptr->ar.resrec.rdata->u.name, d); 191547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_Register(m, &ptr->ar); 191647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) 191747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 191847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("SetSCPrefsBrowseDomain: mDNS_Register returned error %d", err); 191947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(ptr); 192047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 192147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 192247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 192347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ptr->next = LocalDomainEnumRecords; 192447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LocalDomainEnumRecords = ptr; 192547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 192647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 192747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 192847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void DeregisterLocalOnlyDomainEnumPTR(mDNS *m, const domainname *d, int type) 192947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 193047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem **ptr = &LocalDomainEnumRecords; 193147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname lhs; // left-hand side of PTR, for comparison 193247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 193347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("Decrementing %s refcount for %##s", 193447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeBrowse ) ? "browse domain " : 193547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeRegistration ) ? "registration dom" : 193647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (type == mDNS_DomainTypeBrowseAutomatic) ? "automatic browse" : "?", d->c); 193747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 193847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt MakeDomainNameFromDNSNameString(&lhs, mDNS_DomainTypeNames[type]); 193947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AppendDNSNameString (&lhs, "local"); 194047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 194147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*ptr) 194247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 194347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&(*ptr)->ar.resrec.rdata->u.name, d) && SameDomainName((*ptr)->ar.resrec.name, &lhs)) 194447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 194547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem *rem = *ptr; 194647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *ptr = (*ptr)->next; 194747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Deregister(m, &rem->ar); 194847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 194947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 195047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else ptr = &(*ptr)->next; 195147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 195247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 195347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 195447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void AddAutoBrowseDomain(const mDNSu32 uid, const domainname *const name) 195547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 195647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *new = mDNSPlatformMemAllocate(sizeof(DNameListElem)); 195747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!new) { LogMsg("ERROR: malloc"); return; } 195847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&new->name, name); 195947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt new->uid = uid; 196047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt new->next = AutoBrowseDomains; 196147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AutoBrowseDomains = new; 196247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsserver_automatic_browse_domain_changed(new, mDNStrue); 196347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 196447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 196547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void RmvAutoBrowseDomain(const mDNSu32 uid, const domainname *const name) 196647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 196747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem **p = &AutoBrowseDomains; 196847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p && (!SameDomainName(&(*p)->name, name) || (*p)->uid != uid)) p = &(*p)->next; 196947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*p) LogMsg("RmvAutoBrowseDomain: Got remove event for domain %##s not in list", name->c); 197047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 197147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 197247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *ptr = *p; 197347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *p = ptr->next; 197447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsserver_automatic_browse_domain_changed(ptr, mDNSfalse); 197547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(ptr); 197647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 197747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 197847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 197947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void SetPrefsBrowseDomains(mDNS *m, DNameListElem *browseDomains, mDNSBool add) 198047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 198147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *d; 198247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (d = browseDomains; d; d = d->next) 198347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 198447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (add) 198547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 198647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RegisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse); 198747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AddAutoBrowseDomain(d->uid, &d->name); 198847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 198947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 199047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 199147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DeregisterLocalOnlyDomainEnumPTR(m, &d->name, mDNS_DomainTypeBrowse); 199247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RmvAutoBrowseDomain(d->uid, &d->name); 199347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 199447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 199547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 199647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 199747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void UpdateDeviceInfoRecord(mDNS *const m) 199847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 199947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int num_autoname = 0; 200047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req; 200147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (req = all_requests; req; req = req->next) 200247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate == regservice_termination_callback && req->u.servicereg.autoname) 200347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt num_autoname++; 200447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 200547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If DeviceInfo record is currently registered, see if we need to deregister it 200647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->DeviceInfo.resrec.RecordType != kDNSRecordTypeUnregistered) 200747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (num_autoname == 0 || !SameDomainLabelCS(m->DeviceInfo.resrec.name->c, m->nicelabel.c)) 200847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 200947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("UpdateDeviceInfoRecord Deregister %##s", m->DeviceInfo.resrec.name); 201047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Deregister(m, &m->DeviceInfo); 201147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 201247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 201347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If DeviceInfo record is not currently registered, see if we need to register it 201447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->DeviceInfo.resrec.RecordType == kDNSRecordTypeUnregistered) 201547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (num_autoname > 0) 201647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 201747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu8 len = m->HIHardware.c[0] < 255 - 6 ? m->HIHardware.c[0] : 255 - 6; 201847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_SetupResourceRecord(&m->DeviceInfo, mDNSNULL, mDNSNULL, kDNSType_TXT, kStandardTTL, kDNSRecordTypeAdvisory, AuthRecordAny, mDNSNULL, mDNSNULL); 201947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConstructServiceName(&m->DeviceInfo.namestorage, &m->nicelabel, &DeviceInfoName, &localdomain); 202047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 1, "model=", 6); 202147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemCopy(m->DeviceInfo.resrec.rdata->u.data + 7, m->HIHardware.c + 1, len); 202247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->DeviceInfo.resrec.rdata->u.data[0] = 6 + len; // "model=" plus the device string 202347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->DeviceInfo.resrec.rdlength = 7 + len; // One extra for the length byte at the start of the string 202447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("UpdateDeviceInfoRecord Register %##s", m->DeviceInfo.resrec.name); 202547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Register(m, &m->DeviceInfo); 202647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 202747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 202847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 202947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport void udsserver_handle_configchange(mDNS *const m) 203047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 203147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req; 203247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *ptr; 203347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *RegDomains = NULL; 203447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *BrowseDomains = NULL; 203547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *p; 203647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 203747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt UpdateDeviceInfoRecord(m); 203847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 203947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For autoname services, see if the default service name has changed, necessitating an automatic update 204047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (req = all_requests; req; req = req->next) 204147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate == regservice_termination_callback) 204247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->u.servicereg.autoname && !SameDomainLabelCS(req->u.servicereg.name.c, m->nicelabel.c)) 204347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 204447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.servicereg.name = m->nicelabel; 204547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 204647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 204747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ptr->renameonmemfree = 1; 204847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ptr->clientnotified) SendServiceRemovalNotification(&ptr->srs); 204947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("udsserver_handle_configchange: Calling deregister for Service %##s", ptr->srs.RR_PTR.resrec.name->c); 205047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNS_DeregisterService_drt(m, &ptr->srs, mDNS_Dereg_rapid)) 205147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt regservice_callback(m, &ptr->srs, mStatus_MemFree); // If service deregistered already, we can re-register immediately 205247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 205347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 205447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 205547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Let the platform layer get the current DNS information 205647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Lock(m); 205747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformSetDNSConfig(m, mDNSfalse, mDNSfalse, mDNSNULL, &RegDomains, &BrowseDomains); 205847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Unlock(m); 205947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 206047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Any automatic registration domains are also implicitly automatic browsing domains 206147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (RegDomains) SetPrefsBrowseDomains(m, RegDomains, mDNStrue); // Add the new list first 206247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (AutoRegistrationDomains) SetPrefsBrowseDomains(m, AutoRegistrationDomains, mDNSfalse); // Then clear the old list 206347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 206447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Add any new domains not already in our AutoRegistrationDomains list 206547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p=RegDomains; p; p=p->next) 206647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 206747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem **pp = &AutoRegistrationDomains; 206847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*pp && ((*pp)->uid != p->uid || !SameDomainName(&(*pp)->name, &p->name))) pp = &(*pp)->next; 206947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*pp) // If not found in our existing list, this is a new default registration domain 207047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 207147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RegisterLocalOnlyDomainEnumPTR(m, &p->name, mDNS_DomainTypeRegistration); 207247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsserver_default_reg_domain_changed(p, mDNStrue); 207347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 207447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else // else found same domainname in both old and new lists, so no change, just delete old copy 207547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 207647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *del = *pp; 207747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *pp = (*pp)->next; 207847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(del); 207947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 208047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 208147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 208247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Delete any domains in our old AutoRegistrationDomains list that are now gone 208347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (AutoRegistrationDomains) 208447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 208547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *del = AutoRegistrationDomains; 208647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AutoRegistrationDomains = AutoRegistrationDomains->next; // Cut record from list FIRST, 208747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DeregisterLocalOnlyDomainEnumPTR(m, &del->name, mDNS_DomainTypeRegistration); 208847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsserver_default_reg_domain_changed(del, mDNSfalse); // before calling udsserver_default_reg_domain_changed() 208947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(del); 209047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 209147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 209247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Now we have our new updated automatic registration domain list 209347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AutoRegistrationDomains = RegDomains; 209447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 209547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Add new browse domains to internal list 209647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (BrowseDomains) SetPrefsBrowseDomains(m, BrowseDomains, mDNStrue); 209747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 209847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Remove old browse domains from internal list 209947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SCPrefBrowseDomains) 210047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 210147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SetPrefsBrowseDomains(m, SCPrefBrowseDomains, mDNSfalse); 210247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (SCPrefBrowseDomains) 210347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 210447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *fptr = SCPrefBrowseDomains; 210547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SCPrefBrowseDomains = SCPrefBrowseDomains->next; 210647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemFree(fptr); 210747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 210847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 210947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 211047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Replace the old browse domains array with the new array 211147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt SCPrefBrowseDomains = BrowseDomains; 211247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 211347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 211447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void AutomaticBrowseDomainChange(mDNS *const m, DNSQuestion *q, const ResourceRecord *const answer, QC_result AddRecord) 211547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 211647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // unused; 211747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)q; // unused 211847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 211947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("AutomaticBrowseDomainChange: %s automatic browse domain %##s", 212047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AddRecord ? "Adding" : "Removing", answer->rdata->u.name.c); 212147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 212247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (AddRecord) AddAutoBrowseDomain(0, &answer->rdata->u.name); 212347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else RmvAutoBrowseDomain(0, &answer->rdata->u.name); 212447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 212547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 2126afc86e3216439a9437ea54905853275111eac4e2Robert GreenwaltmDNSlocal mStatus handle_sethost_request(request_state *request) 21270b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon{ 2128afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt get_flags(&request->msgptr, request->msgend); 2129afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt char hostName[MAX_DOMAIN_LABEL]; 2130afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt int len = 0; 21310b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon if (get_string(&request->msgptr, request->msgend, hostName, 21320b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon MAX_DOMAIN_LABEL) < 0) return (mStatus_BadParamErr); 2133afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt LogOperation("%3d: DNSSetHostname(%X, %d, nonstr ) START", 21340b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon request->sd, request->flags); 21350b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon // if we start using this as a callback for notification when the 21360b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon // hostname changes we may need to cleanup from it 21370b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon // request->terminate = sethost_termination_callback; 2138afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt if(hostName[0] == 0) return mStatus_BadParamErr; 21390b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon while (len < MAX_DOMAIN_LABEL && hostName[len+1] 21400b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon && hostName[len+1] != '.') len++; 2141afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt strncpy(&(mDNSStorage.nicelabel.c[1]), hostName, len); 2142afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt mDNSStorage.nicelabel.c[0] = len; 2143afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt strncpy(&(mDNSStorage.hostlabel.c[1]), hostName, len); 2144afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt mDNSStorage.hostlabel.c[0] = len; 2145afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt mDNS_SetFQDN(&mDNSStorage); 2146afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt return mStatus_NoError; 21470b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon} 2148afc86e3216439a9437ea54905853275111eac4e2Robert Greenwalt 214947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_browse_request(request_state *request) 215047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 215147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME]; 215247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname typedn, d, temp; 215347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 NumSubTypes; 215447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_NoError; 215547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 215647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 215747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 215847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 215947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 216047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 216147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 || 216247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) return(mStatus_BadParamErr); 216347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 216447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceBrowse(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 216547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 216647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (domain[0] == '\0') uDNS_SetupSearchDomains(&mDNSStorage, UDNS_START_WAB_QUERY); 216747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 216847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->flags = flags; 216947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt typedn.c[0] = 0; 217047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt NumSubTypes = ChopSubTypes(regtype); // Note: Modifies regtype string to remove trailing subtypes 217147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (NumSubTypes < 0 || NumSubTypes > 1) return(mStatus_BadParamErr); 217247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (NumSubTypes == 1 && !AppendDNSNameString(&typedn, regtype + strlen(regtype) + 1)) return(mStatus_BadParamErr); 217347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 217447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!regtype[0] || !AppendDNSNameString(&typedn, regtype)) return(mStatus_BadParamErr); 217547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 217647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&temp, regtype)) return(mStatus_BadParamErr); 217747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For over-long service types, we only allow domain "local" 21785e33e384dc5d19f8aa63ecd4a157e7acbaed579fPierre Imai if (temp.c[0] > 15 && domain[0] == 0) mDNSPlatformStrLCopy(domain, "local.", sizeof(domain)); 217947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 218047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Set up browser info 218147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.browser.ForceMCast = (flags & kDNSServiceFlagsForceMulticast) != 0; 218247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.browser.interface_id = InterfaceID; 218347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&request->u.browser.regtype, &typedn); 218447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.browser.default_domain = !domain[0]; 218547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.browser.browsers = NULL; 218647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 218747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceBrowse(%X, %d, \"%##s\", \"%s\") START", 218847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, request->flags, interfaceIndex, request->u.browser.regtype.c, domain); 218947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 219047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We need to unconditionally set request->terminate, because even if we didn't successfully 219147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // start any browses right now, subsequent configuration changes may cause successful 219247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // browses to be added, and we'll need to cancel them before freeing this memory. 219347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->terminate = browse_termination_callback; 219447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 219547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (domain[0]) 219647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 219747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&d, domain)) return(mStatus_BadParamErr); 219847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = add_domain_to_browser(request, &d); 219947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 220047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = AuthorizedDomain(request, &d, AutoBrowseDomains) ? add_domain_to_browser(request, &d) : mStatus_NoError; 220147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 220247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 220347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 220447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 220547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *sdom; 220647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (sdom = AutoBrowseDomains; sdom; sdom = sdom->next) 220747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!sdom->uid || SystemUID(request->uid) || request->uid == sdom->uid) 220847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 220947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = add_domain_to_browser(request, &sdom->name); 221047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) 221147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 221247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&sdom->name, &localdomain)) break; 221347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else err = mStatus_NoError; // suppress errors for non-local "default" domains 221447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 221547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 221647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 221747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 221847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 221947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 222047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 222147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 222247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 222347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 222447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceResolve 222547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 222647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 222747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void resolve_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 222847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 222947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt size_t len = 0; 223047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char fullname[MAX_ESCAPED_DOMAIN_NAME], target[MAX_ESCAPED_DOMAIN_NAME]; 223147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 223247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 223347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req = question->QuestionContext; 223447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 223547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 223647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceResolve(%##s) %s %s", req->sd, question->qname.c, AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer)); 223747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 223847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AddRecord) 223947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 224047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->u.resolve.srv == answer) req->u.resolve.srv = mDNSNULL; 224147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->u.resolve.txt == answer) req->u.resolve.txt = mDNSNULL; 224247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 224347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 224447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 224547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype == kDNSType_SRV) req->u.resolve.srv = answer; 224647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype == kDNSType_TXT) req->u.resolve.txt = answer; 224747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 224847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!req->u.resolve.txt || !req->u.resolve.srv) return; // only deliver result to client if we have both answers 224947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 225047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(answer->name, fullname); 225147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(&req->u.resolve.srv->rdata->u.srv.target, target); 225247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 225347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // calculate reply length 225447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceFlags); 225547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); // interface index 225647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceErrorType); 225747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += strlen(fullname) + 1; 225847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += strlen(target) + 1; 225947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += 2 * sizeof(mDNSu16); // port, txtLen 226047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += req->u.resolve.txt->rdlength; 226147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 226247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // allocate/init reply header 226347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep = create_reply(resolve_reply_op, len, req); 226447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->flags = dnssd_htonl(0); 226547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNSfalse)); 226647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->error = dnssd_htonl(kDNSServiceErr_NoError); 226747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 226847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&rep->rhdr[1]; 226947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 227047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // write reply data to message 227147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(fullname, &data); 227247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(target, &data); 227347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = req->u.resolve.srv->rdata->u.srv.port.b[0]; 227447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = req->u.resolve.srv->rdata->u.srv.port.b[1]; 227547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint16(req->u.resolve.txt->rdlength, &data); 227647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_rdata (req->u.resolve.txt->rdlength, req->u.resolve.txt->rdata->u.data, &data); 227747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 227847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceResolve(%s) RESULT %s:%d", req->sd, fullname, target, mDNSVal16(req->u.resolve.srv->rdata->u.srv.port)); 227947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(req, rep); 228047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 228147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 228247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void resolve_termination_callback(request_state *request) 228347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 228447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceResolve(%##s) STOP", request->sd, request->u.resolve.qtxt.qname.c); 228547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qtxt); 228647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv); 228747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.resolve.external_advertise) external_stop_resolving_service(&request->u.resolve.qsrv.qname); 228847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 228947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 229047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_resolve_request(request_state *request) 229147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 229247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char name[256], regtype[MAX_ESCAPED_DOMAIN_NAME], domain[MAX_ESCAPED_DOMAIN_NAME]; 229347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname fqdn; 229447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 229547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 229647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // extract the data from the message 229747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 229847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 229947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID; 230047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool wasP2P = (interfaceIndex == kDNSServiceInterfaceIndexP2P); 230147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 230247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 230347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->flags = flags; 230447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (wasP2P) interfaceIndex = kDNSServiceInterfaceIndexAny; 230547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 230647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 230747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) 230847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_resolve_request bad interfaceIndex %d", interfaceIndex); return(mStatus_BadParamErr); } 230947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 231047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, name, 256) < 0 || 231147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, regtype, MAX_ESCAPED_DOMAIN_NAME) < 0 || 231247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&request->msgptr, request->msgend, domain, MAX_ESCAPED_DOMAIN_NAME) < 0) 231347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_resolve_request - Couldn't read name/regtype/domain"); return(mStatus_BadParamErr); } 231447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 231547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceResolve(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 231647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 231747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (build_domainname_from_strings(&fqdn, name, regtype, domain) < 0) 231847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_resolve_request bad “%s” “%s” “%s”", name, regtype, domain); return(mStatus_BadParamErr); } 231947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 232047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(&request->u.resolve, sizeof(request->u.resolve)); 232147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 232247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // format questions 232347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.InterfaceID = InterfaceID; 232447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.Target = zeroAddr; 232547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&request->u.resolve.qsrv.qname, &fqdn); 232647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.qtype = kDNSType_SRV; 232747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.qclass = kDNSClass_IN; 232847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 232947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.ExpectUnique = mDNStrue; 233047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 233147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 233247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.SuppressUnusable = mDNSfalse; 233347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.SearchListIndex = 0; 233447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.AppendSearchDomains = 0; 233547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.RetryWithSearchDomains = mDNSfalse; 233647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.TimeoutQuestion = 0; 233747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.WakeOnResolve = (flags & kDNSServiceFlagsWakeOnResolve) != 0; 233847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.qnameOrig = mDNSNULL; 233947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.QuestionCallback = resolve_result_callback; 234047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qsrv.QuestionContext = request; 234147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 234247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.InterfaceID = InterfaceID; 234347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.Target = zeroAddr; 234447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&request->u.resolve.qtxt.qname, &fqdn); 234547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.qtype = kDNSType_TXT; 234647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.qclass = kDNSClass_IN; 234747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 234847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.ExpectUnique = mDNStrue; 234947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 235047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 235147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.SuppressUnusable = mDNSfalse; 235247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.SearchListIndex = 0; 235347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.AppendSearchDomains = 0; 235447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.RetryWithSearchDomains = mDNSfalse; 235547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.TimeoutQuestion = 0; 235647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.WakeOnResolve = 0; 235747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.qnameOrig = mDNSNULL; 235847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.QuestionCallback = resolve_result_callback; 235947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.qtxt.QuestionContext = request; 236047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 236147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.ReportTime = NonZeroTime(mDNS_TimeNow(&mDNSStorage) + 130 * mDNSPlatformOneSecond); 236247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 236347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.external_advertise = mDNSfalse; 236447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 236547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 236647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AuthorizedDomain(request, &fqdn, AutoBrowseDomains)) return(mStatus_NoError); 236747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 236847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 236947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // ask the questions 237047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceResolve(%##s) START", request->sd, request->u.resolve.qsrv.qname.c); 237147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qsrv); 237247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err) 237347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 237447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, &request->u.resolve.qtxt); 237547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) mDNS_StopQuery(&mDNSStorage, &request->u.resolve.qsrv); 237647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 237747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 237847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->terminate = resolve_termination_callback; 237947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the user explicitly passed in P2P, we don't restrict the domain in which we resolve. 238047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (wasP2P || (!InterfaceID && IsLocalDomain(&fqdn) && (request->flags & kDNSServiceFlagsIncludeP2P))) 238147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 238247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.resolve.external_advertise = mDNStrue; 238347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("handle_resolve_request: calling external_start_resolving_service()"); 238447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_resolving_service(&fqdn); 238547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 238647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 238747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 238847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 238947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 239047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 239147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 239247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 239347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 239447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 239547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceQueryRecord 239647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 239747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 239847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// mDNS operation functions. Each operation has 3 associated functions - a request handler that parses 239947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// the client's request and makes the appropriate mDNSCore call, a result handler (passed as a callback 240047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// to the mDNSCore routine) that sends results back to the client, and a termination routine that aborts 240147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// the mDNSCore operation if the client dies or closes its socket. 240247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 240347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Returns -1 to tell the caller that it should not try to reissue the query anymore 240447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Returns 1 on successfully appending a search domain and the caller should reissue the new query 240547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Returns 0 when there are no more search domains and the caller should reissue the query 240647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal int AppendNewSearchDomain(mDNS *const m, DNSQuestion *question) 240747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 240847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname *sd; 240947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 241047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 241147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Sanity check: The caller already checks this. We use -1 to indicate that we have searched all 241247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the domains and should try the single label query directly on the wire. 241347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question->SearchListIndex == -1) 241447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 241547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("AppendNewSearchDomain: question %##s (%s) SearchListIndex is -1", question->qname.c, DNSTypeName(question->qtype)); 241647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 241747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 241847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 241947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!question->AppendSearchDomains) 242047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 242147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("AppendNewSearchDomain: question %##s (%s) AppendSearchDoamins is 0", question->qname.c, DNSTypeName(question->qtype)); 242247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 242347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 242447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 242547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Save the original name, before we modify them below. 242647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!question->qnameOrig) 242747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 242847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->qnameOrig = mallocL("AppendNewSearchDomain", sizeof(domainname)); 242947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!question->qnameOrig) { LogMsg("AppendNewSearchDomain: ERROR!! malloc failure"); return -1; } 243047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->qnameOrig->c[0] = 0; 243147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(question->qnameOrig, &question->qname); 243247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("AppendSearchDomain: qnameOrig %##s", question->qnameOrig->c); 243347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 243447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 243547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt sd = uDNS_GetNextSearchDomain(m, question->InterfaceID, &question->SearchListIndex, !question->AppendLocalSearchDomains); 243647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We use -1 to indicate that we have searched all the domains and should try the single label 243747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // query directly on the wire. uDNS_GetNextSearchDomain should never return a negative value 243847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question->SearchListIndex == -1) 243947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 244047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("AppendNewSearchDomain: ERROR!! uDNS_GetNextSearchDomain returned -1"); 244147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 244247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 244347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 244447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Not a common case. Perhaps, we should try the next search domain if it exceeds ? 244547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (sd && (DomainNameLength(question->qnameOrig) + DomainNameLength(sd)) > MAX_DOMAIN_NAME) 244647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 244747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 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)); 244847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 244947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 245047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 245147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // if there are no more search domains and we have already tried this question 245247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // without appending search domains, then we are done. 245347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!sd && !ApplySearchDomainsFirst(question)) 245447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 245547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("AppnedNewSearchDomain: No more search domains for question with name %##s (%s), not trying anymore", question->qname.c, DNSTypeName(question->qtype)); 245647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 245747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 245847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 245947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Stop the question before changing the name as negative cache entries could be pointing at this question. 246047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Even if we don't change the question in the case of returning 0, the caller is going to restart the 246147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // question. 246247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StopQuery(&mDNSStorage, question); 246347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) { LogMsg("AppendNewSearchDomain: ERROR!! %##s %s mDNS_StopQuery: %d, while retrying with search domains", question->qname.c, DNSTypeName(question->qtype), (int)err); } 246447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 246547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&question->qname, question->qnameOrig); 246647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (sd) 246747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 246847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AppendDomainName(&question->qname, sd); 246947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("AppnedNewSearchDomain: Returning question with name %##s, SearchListIndex %d", question->qname.c, question->SearchListIndex); 247047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return 1; 247147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 247247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 247347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Try the question as single label 247447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("AppnedNewSearchDomain: No more search domains for question with name %##s (%s), trying one last time", question->qname.c, DNSTypeName(question->qtype)); 247547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return 0; 247647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 247747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 247847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 247947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 248047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mDNSBool DomainInSearchList(domainname *domain) 248147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 248247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const SearchListElem *s; 248347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (s=SearchList; s; s=s->next) 248447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (SameDomainName(&s->domain, domain)) return mDNStrue; 248547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNSfalse; 248647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 248747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 248847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Workaround for networks using Microsoft Active Directory using "local" as a private internal 248947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// top-level domain 249047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus SendAdditionalQuery(DNSQuestion *q, request_state *request, mStatus err) 249147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 249247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt extern domainname ActiveDirectoryPrimaryDomain; 249347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion **question2; 249447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #define VALID_MSAD_SRV_TRANSPORT(T) (SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_tcp") || SameDomainLabel((T)->c, (const mDNSu8 *)"\x4_udp")) 249547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #define VALID_MSAD_SRV(Q) ((Q)->qtype == kDNSType_SRV && VALID_MSAD_SRV_TRANSPORT(SecondLabel(&(Q)->qname))) 249647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 249747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question2 = mDNSNULL; 249847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->hdr.op == query_request) 249947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question2 = &request->u.queryrecord.q2; 250047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (request->hdr.op == addrinfo_request) 250147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 250247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q->qtype == kDNSType_A) 250347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question2 = &request->u.addrinfo.q42; 250447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (q->qtype == kDNSType_AAAA) 250547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question2 = &request->u.addrinfo.q62; 250647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 250747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!question2) 250847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 250947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("SendAdditionalQuery: question2 NULL for %##s (%s)", q->qname.c, DNSTypeName(q->qtype)); 251047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mStatus_BadParamErr; 251147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 251247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 251347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Sanity check: If we already sent an additonal query, we don't need to send one more. 251447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 251547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 1. When the application calls DNSServiceQueryRecord or DNSServiceGetAddrInfo with a .local name, this function 251647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // is called to see whether a unicast query should be sent or not. 251747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 251847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 2. As a result of appending search domains, the question may be end up with a .local suffix even though it 251947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // was not a .local name to start with. In that case, queryrecord_result_callback calls this function to 252047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // send the additional query. 252147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 252247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Thus, it should not be called more than once. 252347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (*question2) 252447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 252547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("SendAdditionalQuery: question2 already sent for %##s (%s), no more q2", q->qname.c, DNSTypeName(q->qtype)); 252647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return err; 252747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 252847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 252947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!q->ForceMCast && SameDomainLabel(LastLabel(&q->qname), (const mDNSu8 *)&localdomain)) 253047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q->qtype == kDNSType_A || q->qtype == kDNSType_AAAA || VALID_MSAD_SRV(q)) 253147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 253247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *q2; 253347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int labels = CountLabels(&q->qname); 253447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2 = mallocL("DNSQuestion", sizeof(DNSQuestion)); 253547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!q2) FatalError("ERROR: SendAdditionalQuery malloc"); 253647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *question2 = q2; 253747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *q2 = *q; 253847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->InterfaceID = mDNSInterface_Unicast; 253947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->ExpectUnique = mDNStrue; 254047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the query starts as a single label e.g., somehost, and we have search domains with .local, 254147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // queryrecord_result_callback calls this function when .local is appended to "somehost". 254247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // At that time, the name in "q" is pointing at somehost.local and its qnameOrig pointing at 254347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // "somehost". We need to copy that information so that when we retry with a different search 254447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // domain e.g., mycompany.local, we get "somehost.mycompany.local". 254547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q->qnameOrig) 254647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 254747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*question2)->qnameOrig = mallocL("SendAdditionalQuery", DomainNameLength(q->qnameOrig)); 254847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!(*question2)->qnameOrig) { LogMsg("SendAdditionalQuery: ERROR!! malloc failure"); return mStatus_NoMemoryErr; } 254947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (*question2)->qnameOrig->c[0] = 0; 255047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName((*question2)->qnameOrig, q->qnameOrig); 255147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("SendAdditionalQuery: qnameOrig %##s", (*question2)->qnameOrig->c); 255247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 255347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For names of the form "<one-or-more-labels>.bar.local." we always do a second unicast query in parallel. 255447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For names of the form "<one-label>.local." it's less clear whether we should do a unicast query. 255547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the name being queried is exactly the same as the name in the DHCP "domain" option (e.g. the DHCP 255647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // "domain" is my-small-company.local, and the user types "my-small-company.local" into their web browser) 255747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // then that's a hint that it's worth doing a unicast query. Otherwise, we first check to see if the 255847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // site's DNS server claims there's an SOA record for "local", and if so, that's also a hint that queries 255947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // for names in the "local" domain will be safely answered privately before they hit the root name servers. 256047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note that in the "my-small-company.local" example above there will typically be an SOA record for 256147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // "my-small-company.local" but *not* for "local", which is why the "local SOA" check would fail in that case. 256247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We need to check against both ActiveDirectoryPrimaryDomain and SearchList. If it matches against either 256347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // of those, we don't want do the SOA check for the local 256447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (labels == 2 && !SameDomainName(&q->qname, &ActiveDirectoryPrimaryDomain) && !DomainInSearchList(&q->qname)) 256547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 256647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AssignDomainName(&q2->qname, &localdomain); 256747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->qtype = kDNSType_SOA; 256847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->LongLived = mDNSfalse; 256947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->ForceMCast = mDNSfalse; 257047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->ReturnIntermed = mDNStrue; 257147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't append search domains for the .local SOA query 257247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->AppendSearchDomains = 0; 257347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->AppendLocalSearchDomains = 0; 257447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->RetryWithSearchDomains = mDNSfalse; 257547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->SearchListIndex = 0; 257647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q2->TimeoutQuestion = 0; 257747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 257847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast", request->sd, q2->qname.c, DNSTypeName(q2->qtype)); 257947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, q2); 258047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q2->qname.c, DNSTypeName(q2->qtype), (int)err); 258147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 258247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 258347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 258447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 258547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 258647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// This function tries to append a search domain if valid and possible. If so, returns true. 258747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mDNSBool RetryQuestionWithSearchDomains(mDNS *const m, DNSQuestion *question, request_state *req) 258847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 258947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int result; 259047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // RetryWithSearchDomains tells the core to call us back so that we can retry with search domains if there is no 259147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // answer in the cache or /etc/hosts. In the first call back from the core, we clear RetryWithSearchDomains so 259247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // that we don't get called back repeatedly. If we got an answer from the cache or /etc/hosts, we don't touch 259347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // RetryWithSearchDomains which may or may not be set. 259447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 259547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we get e.g., NXDOMAIN and the query is neither suppressed nor exhausted the domain search list and 259647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // is a valid question for appending search domains, retry by appending domains 259747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 259847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!question->SuppressQuery && question->SearchListIndex != -1 && question->AppendSearchDomains) 259947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 260047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->RetryWithSearchDomains = 0; 260147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = AppendNewSearchDomain(m, question); 260247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // As long as the result is either zero or 1, we retry the question. If we exahaust the search 260347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // domains (result is zero) we try the original query (as it was before appending the search 260447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // domains) as such on the wire as a last resort if we have not tried them before. For queries 260547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // with more than one label, we have already tried them before appending search domains and 260647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // hence don't retry again 260747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result != -1) 260847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 260947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 261047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(m, question); 261147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err) 261247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 261347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: RetryQuestionWithSearchDomains(%##s, %s), retrying after appending search domain", req->sd, question->qname.c, DNSTypeName(question->qtype)); 261447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the result was zero, it meant that there are no search domains and we just retried the question 261547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // as a single label and we should not retry with search domains anymore. 261647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!result) question->SearchListIndex = -1; 261747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNStrue; 261847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 261947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 262047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 262147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: ERROR: RetryQuestionWithSearchDomains %##s %s mDNS_StartQuery: %d, while retrying with search domains", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err); 262247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We have already stopped the query and could not restart. Reset the appropriate pointers 262347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // so that we don't call stop again when the question terminates 262447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->QuestionContext = mDNSNULL; 262547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 262647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 262747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 262847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 262947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 263047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("%3d: RetryQuestionWithSearchDomains: Not appending search domains - SuppressQuery %d, SearchListIndex %d, AppendSearchDomains %d", req->sd, question->SuppressQuery, question->SearchListIndex, question->AppendSearchDomains); 263147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 263247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNSfalse; 263347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 263447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 263547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void queryrecord_result_callback(mDNS *const m, DNSQuestion *question, const ResourceRecord *const answer, QC_result AddRecord) 263647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 263747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char name[MAX_ESCAPED_DOMAIN_NAME]; 263847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req = question->QuestionContext; 263947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 264047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 264147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt size_t len; 264247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceErrorType error = kDNSServiceErr_NoError; 264347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *q = mDNSNULL; 264447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 264547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 264647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 264747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Sanity check: QuestionContext is set to NULL after we stop the question and hence we should not 264847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // get any callbacks from the core after this. 264947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!req) 265047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 265147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("queryrecord_result_callback: ERROR!! QuestionContext NULL for %##s (%s)", question->qname.c, DNSTypeName(question->qtype)); 265247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 265347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 265447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.op == query_request && question == req->u.queryrecord.q2) 265547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q = &req->u.queryrecord.q; 265647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q42) 265747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q = &req->u.addrinfo.q4; 265847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->hdr.op == addrinfo_request && question == req->u.addrinfo.q62) 265947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q = &req->u.addrinfo.q6; 266047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 266147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q && question->qtype != q->qtype && !SameDomainName(&question->qname, &q->qname)) 266247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 266347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 266447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname *orig = question->qnameOrig; 266547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 266647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback: Stopping q2 local %##s", question->qname.c); 266747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(m, question); 266847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->QuestionContext = mDNSNULL; 266947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 267047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We got a negative response for the SOA record indicating that .local does not exist. 267147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // But we might have other search domains (that does not end in .local) that can be 267247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // appended to this question. In that case, we want to retry the question. Otherwise, 267347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // we don't want to try this question as unicast. 267447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->RecordType == kDNSRecordTypePacketNegative && !q->AppendSearchDomains) 267547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 267647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback: question %##s AppendSearchDomains zero", q->qname.c); 267747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 267847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 267947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 268047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we got a non-negative answer for our "local SOA" test query, start an additional parallel unicast query 268147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 268247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: When we copy the original question, we copy everything including the AppendSearchDomains, 268347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // RetryWithSearchDomains except for qnameOrig which can be non-NULL if the original question is 268447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // e.g., somehost and then we appended e.g., ".local" and retried that question. See comment in 268547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // SendAdditionalQuery as to how qnameOrig gets initialized. 268647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *question = *q; 268747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->InterfaceID = mDNSInterface_Unicast; 268847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->ExpectUnique = mDNStrue; 268947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->qnameOrig = orig; 269047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 269147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) unicast, context %p", req->sd, question->qname.c, DNSTypeName(question->qtype), question->QuestionContext); 269247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 269347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the original question timed out, its QuestionContext would already be set to NULL and that's what we copied above. 269447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Hence, we need to set it explicitly here. 269547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->QuestionContext = req; 269647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(m, question); 269747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("%3d: ERROR: queryrecord_result_callback %##s %s mDNS_StartQuery: %d", req->sd, question->qname.c, DNSTypeName(question->qtype), (int)err); 269847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 269947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we got a positive response to local SOA, then try the .local question as unicast 270047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->RecordType != kDNSRecordTypePacketNegative) return; 270147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 270247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Fall through and get the next search domain. The question is pointing at .local 270347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // and we don't want to try that. Try the next search domain. Don't try with local 270447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // search domains for the unicast question anymore. 270547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 270647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: we started the question above which will be stopped immediately (never sent on the wire) 270747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // before we pick the next search domain below. RetryQuestionWithSearchDomains assumes that the 270847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // question has already started. 270947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->AppendLocalSearchDomains = 0; 271047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 271147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 271247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q && AddRecord && (question->InterfaceID == mDNSInterface_Unicast) && !answer->rdlength) 271347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 271447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we get a negative response to the unicast query that we sent above, retry after appending search domains 271547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: We could have appended search domains below (where do it for regular unicast questions) instead of doing it here. 271647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // As we ignore negative unicast answers below, we would never reach the code where the search domains are appended. 271747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // To keep things simple, we handle unicast ".local" separately here. 271847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback: Retrying .local question %##s (%s) as unicast after appending search domains", question->qname.c, DNSTypeName(question->qtype)); 271947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (RetryQuestionWithSearchDomains(m, question, req)) 272047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 272147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question->AppendSearchDomains && !question->AppendLocalSearchDomains && IsLocalDomain(&question->qname)) 272247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 272347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If "local" is the last search domain, we need to stop the question so that we don't send the "local" 272447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // question on the wire as we got a negative response for the local SOA. But, we can't stop the question 272547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // yet as we may have to timeout the question (done by the "core") for which we need to leave the question 272647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // in the list. We leave it disabled so that it does not hit the wire. 272747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback: Disabling .local question %##s (%s)", question->qname.c, DNSTypeName(question->qtype)); 272847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->ThisQInterval = 0; 272947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 273047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 273147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 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 273247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // domains to append for "q2". In all cases, fall through and deliver the response 273347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 273447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 273547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 273647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->RecordType == kDNSRecordTypePacketNegative) 273747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 273847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If this question needs to be timed out and we have reached the stop time, mark 273947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the error as timeout. It is possible that we might get a negative response from an 274047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // external DNS server at the same time when this question reaches its stop time. We 274147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // can't tell the difference as there is no indication in the callback. This should 274247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // be okay as we will be timing out this query anyway. 274347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Lock(m); 274447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question->TimeoutQuestion) 274547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 274647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((m->timenow - question->StopTime) >= 0) 274747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 274847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback:Question %##s (%s) timing out, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 274947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt error = kDNSServiceErr_Timeout; 275047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 275147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 275247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Unlock(m); 275347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // When we're doing parallel unicast and multicast queries for dot-local names (for supporting Microsoft 275447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Active Directory sites) we need to ignore negative unicast answers. Otherwise we'll generate negative 275547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // answers for just about every single multicast name we ever look up, since the Microsoft Active Directory 275647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // server is going to assert that pretty much every single multicast name doesn't exist. 275747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 275847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we are timing out this query, we need to deliver the negative answer to the application 275947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (error != kDNSServiceErr_Timeout) 276047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 276147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!answer->InterfaceID && IsLocalDomain(answer->name)) 276247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 276347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback:Question %##s (%s) answering local with unicast", question->qname.c, DNSTypeName(question->qtype)); 276447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 276547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 276647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt error = kDNSServiceErr_NoSuchRecord; 276747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 276847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AddRecord = mDNStrue; 276947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 277047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we get a negative answer, try appending search domains. Don't append search domains 277147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // - if we are timing out this question 277247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // - if the negative response was received as a result of a multicast query 277347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // - if this is an additional query (q2), we already appended search domains above (indicated by "!q" below) 277447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (error != kDNSServiceErr_Timeout) 277547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 277647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!q && !answer->InterfaceID && !answer->rdlength && AddRecord) 277747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 277847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the original question did not end in .local, we did not send an SOA query 277947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // to figure out whether we should send an additional unicast query or not. If we just 278047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // appended .local, we need to see if we need to send an additional query. This should 278147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // normally happen just once because after we append .local, we ignore all negative 278247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // responses for .local above. 278347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_result_callback: Retrying question %##s (%s) after appending search domains", question->qname.c, DNSTypeName(question->qtype)); 278447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (RetryQuestionWithSearchDomains(m, question, req)) 278547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 278647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: We need to call SendAdditionalQuery every time after appending a search domain as .local could 278747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // be anywhere in the search domain list. 278847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 278947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_NoError; 279047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = SendAdditionalQuery(question, req, err); 279147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("queryrecord_result_callback: Sending .local SOA query failed, after appending domains"); 279247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 279347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 279447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 279547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 279647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 279747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 279847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(answer->name, name); 279947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 280047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: %s(%##s, %s) %s %s", req->sd, 280147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->hdr.op == query_request ? "DNSServiceQueryRecord" : "DNSServiceGetAddrInfo", 280247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->qname.c, DNSTypeName(question->qtype), AddRecord ? "ADD" : "RMV", RRDisplayString(m, answer)); 280347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 280447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len = sizeof(DNSServiceFlags); // calculate reply data length 280547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); // interface index 280647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceErrorType); 280747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += strlen(name) + 1; 280847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += 3 * sizeof(mDNSu16); // type, class, rdlen 280947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += answer->rdlength; 281047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); // TTL 281147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 281247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep = create_reply(req->hdr.op == query_request ? query_reply_op : addrinfo_reply_op, len, req); 281347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 281447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->flags = dnssd_htonl(AddRecord ? kDNSServiceFlagsAdd : 0); 281547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Call mDNSPlatformInterfaceIndexfromInterfaceID, but suppressNetworkChange (last argument). Otherwise, if the 281647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // InterfaceID is not valid, then it simulates a "NetworkChanged" which in turn makes questions 281747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // to be stopped and started including *this* one. Normally the InterfaceID is valid. But when we 281847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // are using the /etc/hosts entries to answer a question, the InterfaceID may not be known to the 281947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // mDNS core . Eventually, we should remove the calls to "NetworkChanged" in 282047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // mDNSPlatformInterfaceIndexfromInterfaceID when it can't find InterfaceID as ResourceRecords 282147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // should not have existed to answer this question if the corresponding interface is not valid. 282247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, answer->InterfaceID, mDNStrue)); 282347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->error = dnssd_htonl(error); 282447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 282547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&rep->rhdr[1]; 282647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 282747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(name, &data); 282847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint16(answer->rrtype, &data); 282947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint16(answer->rrclass, &data); 283047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint16(answer->rdlength, &data); 283147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We need to use putRData here instead of the crude put_rdata function, because the crude put_rdata 283247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // function just does a blind memory copy without regard to structures that may have holes in them. 283347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rdlength) 283447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!putRData(mDNSNULL, (mDNSu8 *)data, (mDNSu8 *)rep->rhdr + len, answer)) 283547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("queryrecord_result_callback putRData failed %d", (mDNSu8 *)rep->rhdr + len - (mDNSu8 *)data); 283647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data += answer->rdlength; 283747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint32(AddRecord ? answer->rroriginalttl : 0, &data); 283847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 283947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(req, rep); 284047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Stop the question, if we just timed out 284147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (error == kDNSServiceErr_Timeout) 284247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 284347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(m, question); 284447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Reset the pointers so that we don't call stop on termination 284547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt question->QuestionContext = mDNSNULL; 284647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 284747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 284847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if ! NO_WCF 284947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CHECK_WCF_FUNCTION(WCFIsServerRunning) 285047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 285147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct xucred x; 285247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt socklen_t xucredlen = sizeof(x); 285347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 285447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (WCFIsServerRunning((WCFConnection *)m->WCF) && answer->rdlength != 0) 285547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 285647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (getsockopt(req->sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 && 285747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (x.cr_version == XUCRED_VERSION)) 285847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 285947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct sockaddr_storage addr; 286047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const RDataBody2 *const rdb = (RDataBody2 *)answer->rdata->u.data; 286147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt addr.ss_len = 0; 286247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype == kDNSType_A || answer->rrtype == kDNSType_AAAA) 286347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 286447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype == kDNSType_A) 286547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 286647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct sockaddr_in *sin = (struct sockaddr_in *)&addr; 286747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt sin->sin_port = 0; 286847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!putRData(mDNSNULL, (mDNSu8 *)&sin->sin_addr, (mDNSu8 *)(&sin->sin_addr + sizeof(rdb->ipv4)), answer)) 286947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("queryrecord_result_callback: WCF AF_INET putRData failed"); 287047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 287147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 287247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt addr.ss_len = sizeof (struct sockaddr_in); 287347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt addr.ss_family = AF_INET; 287447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 287547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 287647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (answer->rrtype == kDNSType_AAAA) 287747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 287847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&addr; 287947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt sin6->sin6_port = 0; 288047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!putRData(mDNSNULL, (mDNSu8 *)&sin6->sin6_addr, (mDNSu8 *)(&sin6->sin6_addr + sizeof(rdb->ipv6)), answer)) 288147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("queryrecord_result_callback: WCF AF_INET6 putRData failed"); 288247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 288347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 288447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt addr.ss_len = sizeof (struct sockaddr_in6); 288547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt addr.ss_family = AF_INET6; 288647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 288747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 288847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (addr.ss_len) 288947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 289047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("queryrecord_result_callback: Name %s, uid %u, addr length %d", name, x.cr_uid, addr.ss_len); 289147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr) 289247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 289347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt WCFNameResolvesToAddr(m->WCF, name, (struct sockaddr *)&addr, x.cr_uid); 289447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 289547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 289647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 289747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (answer->rrtype == kDNSType_CNAME) 289847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 289947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname cname; 290047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char cname_cstr[MAX_ESCAPED_DOMAIN_NAME]; 290147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!putRData(mDNSNULL, cname.c, (mDNSu8 *)(cname.c + MAX_DOMAIN_NAME), answer)) 290247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("queryrecord_result_callback: WCF CNAME putRData failed"); 290347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 290447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 290547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(&cname, cname_cstr); 290647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CHECK_WCF_FUNCTION((WCFConnection *)WCFNameResolvesToAddr) 290747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 290847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt WCFNameResolvesToName(m->WCF, name, cname_cstr, x.cr_uid); 290947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else my_perror("queryrecord_result_callback: ERROR: getsockopt LOCAL_PEERCRED"); 291447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 291747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 291847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 291947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 292047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void queryrecord_termination_callback(request_state *request) 292147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 292247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceQueryRecord(%##s, %s) STOP", 292347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, request->u.queryrecord.q.qname.c, DNSTypeName(request->u.queryrecord.q.qtype)); 292447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.queryrecord.q.QuestionContext) 292547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 292647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.queryrecord.q); // no need to error check 292747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.queryrecord.q.QuestionContext = mDNSNULL; 292847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 292947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 293047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 293147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *question = &request->u.queryrecord.q; 293247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_termination_callback: question %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 293347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 293447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 293547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.queryrecord.q.qnameOrig) 293647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 293747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination", request->u.queryrecord.q.qnameOrig); 293847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.queryrecord.q.qnameOrig = mDNSNULL; 293947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 294047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 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))) 294147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 294247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_termination_callback: calling external_stop_browsing_for_service()"); 294347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_stop_browsing_for_service(&mDNSStorage, &request->u.queryrecord.q.qname, request->u.queryrecord.q.qtype); 294447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 294547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.queryrecord.q2) 294647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 294747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.queryrecord.q2->QuestionContext) 294847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 294947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_termination_callback: Stopping q2 %##s", request->u.queryrecord.q2->qname.c); 295047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, request->u.queryrecord.q2); 295147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 295247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 295347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 295447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *question = request->u.queryrecord.q2; 295547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_termination_callback: q2 %##s (%s) already stopped, InterfaceID %p", question->qname.c, DNSTypeName(question->qtype), question->InterfaceID); 295647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 295747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.queryrecord.q2->qnameOrig) 295847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 295947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("queryrecord_termination_callback: freeing q2 qnameOrig %##s", request->u.queryrecord.q2->qnameOrig->c); 296047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination q2", request->u.queryrecord.q2->qnameOrig); 296147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.queryrecord.q2->qnameOrig = mDNSNULL; 296247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 296347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("queryrecord Q2", request->u.queryrecord.q2); 296447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.queryrecord.q2 = mDNSNULL; 296547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 296647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 296747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 296847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_queryrecord_request(request_state *request) 296947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 297047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *const q = &request->u.queryrecord.q; 297147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char name[256]; 297247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu16 rrtype, rrclass; 297347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 297447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 297547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 297647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 297747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 297847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 297947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 298047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, name, 256) < 0) return(mStatus_BadParamErr); 298147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rrtype = get_uint16(&request->msgptr, request->msgend); 298247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rrclass = get_uint16(&request->msgptr, request->msgend); 298347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 298447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) 298547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceQueryRecord(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 298647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 298747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->flags = flags; 298847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(&request->u.queryrecord, sizeof(request->u.queryrecord)); 298947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 299047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->InterfaceID = InterfaceID; 299147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->Target = zeroAddr; 299247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&q->qname, name)) return(mStatus_BadParamErr); 299347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 299447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AuthorizedDomain(request, &q->qname, AutoBrowseDomains)) return (mStatus_NoError); 299547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 299647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->qtype = rrtype; 299747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->qclass = rrclass; 299847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 299947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->ExpectUnique = mDNSfalse; 300047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 300147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 300247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0; 300347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0; 300447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->WakeOnResolve = 0; 300547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->QuestionCallback = queryrecord_result_callback; 300647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->QuestionContext = request; 300747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->SearchListIndex = 0; 300847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 300947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Don't append search domains for fully qualified domain names including queries 301047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // such as e.g., "abc." that has only one label. We convert all names to FQDNs as internally 301147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // we only deal with FQDNs. Hence, we cannot look at qname to figure out whether we should 301247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // append search domains or not. So, we record that information in AppendSearchDomains. 301347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // 301447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We append search domains only for queries that are a single label. If overriden using 301547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // command line argument "AlwaysAppendSearchDomains", then we do it for any query which 301647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // is not fully qualified. 301747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 301847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((rrtype == kDNSType_A || rrtype == kDNSType_AAAA) && name[strlen(name) - 1] != '.' && 301947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (AlwaysAppendSearchDomains || CountLabels(&q->qname) == 1)) 302047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 302147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->AppendSearchDomains = 1; 302247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->AppendLocalSearchDomains = 1; 302347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 302447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 302547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 302647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->AppendSearchDomains = 0; 302747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->AppendLocalSearchDomains = 0; 302847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 302947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 303047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For single label queries that are not fully qualified, look at /etc/hosts, cache and try 303147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // search domains before trying them on the wire as a single label query. RetryWithSearchDomains 303247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // tell the core to call back into the UDS layer if there is no valid response in /etc/hosts or 303347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the cache 303447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->RetryWithSearchDomains = ApplySearchDomainsFirst(q) ? 1 : 0; 303547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->qnameOrig = mDNSNULL; 303647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 303747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceQueryRecord(%X, %d, %##s, %s) START", request->sd, flags, interfaceIndex, q->qname.c, DNSTypeName(q->qtype)); 303847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, q); 303947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("%3d: ERROR: DNSServiceQueryRecord %##s %s mDNS_StartQuery: %d", request->sd, q->qname.c, DNSTypeName(q->qtype), (int)err); 304047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 304147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 304247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->terminate = queryrecord_termination_callback; 304347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q->InterfaceID == mDNSInterface_P2P || (!q->InterfaceID && SameDomainName((const domainname *)LastLabel(&q->qname), &localdomain) && (flags & kDNSServiceFlagsIncludeP2P))) 304447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 304547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("handle_queryrecord_request: calling external_start_browsing_for_service()"); 304647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt external_start_browsing_for_service(&mDNSStorage, &q->qname, q->qtype); 304747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 304847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 304947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 305047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 305147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = SendAdditionalQuery(q, request, err); 305247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 305347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 305447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 305547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 305647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 305747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 305847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 305947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 306047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceEnumerateDomains 306147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 306247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 306347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal reply_state *format_enumeration_reply(request_state *request, 306447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *domain, DNSServiceFlags flags, mDNSu32 ifi, DNSServiceErrorType err) 306547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 306647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt size_t len; 306747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *reply; 306847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 306947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 307047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len = sizeof(DNSServiceFlags); 307147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(mDNSu32); 307247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += sizeof(DNSServiceErrorType); 307347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt len += strlen(domain) + 1; 307447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 307547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply = create_reply(enumeration_reply_op, len, request); 307647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->flags = dnssd_htonl(flags); 307747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->ifi = dnssd_htonl(ifi); 307847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply->rhdr->error = dnssd_htonl(err); 307947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&reply->rhdr[1]; 308047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_string(domain, &data); 308147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return reply; 308247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 308347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 308447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void enum_termination_callback(request_state *request) 308547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 308647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all); 308747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_default); 308847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 308947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 309047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void enum_result_callback(mDNS *const m, 309147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSQuestion *const question, const ResourceRecord *const answer, QC_result AddRecord) 309247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 309347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char domain[MAX_ESCAPED_DOMAIN_NAME]; 309447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request = question->QuestionContext; 309547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = 0; 309647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *reply; 309747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)m; // Unused 309847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 309947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (answer->rrtype != kDNSType_PTR) return; 310047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 310147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 310247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AuthorizedDomain(request, &answer->rdata->u.name, request->u.enumeration.flags ? AutoRegistrationDomains : AutoBrowseDomains)) return; 310347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 310447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 310547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We only return add/remove events for the browse and registration lists 310647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For the default browse and registration answers, we only give an "ADD" event 310747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question == &request->u.enumeration.q_default && !AddRecord) return; 310847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 310947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (AddRecord) 311047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 311147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt flags |= kDNSServiceFlagsAdd; 311247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (question == &request->u.enumeration.q_default) flags |= kDNSServiceFlagsDefault; 311347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 311447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 311547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertDomainNameToCString(&answer->rdata->u.name, domain); 311647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note that we do NOT propagate specific interface indexes to the client - for example, a domain we learn from 311747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // a machine's system preferences may be discovered on the LocalOnly interface, but should be browsed on the 311847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // network, so we just pass kDNSServiceInterfaceIndexAny 311947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply = format_enumeration_reply(request, domain, flags, kDNSServiceInterfaceIndexAny, kDNSServiceErr_NoError); 312047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!reply) { LogMsg("ERROR: enum_result_callback, format_enumeration_reply"); return; } 312147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 312247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceEnumerateDomains(%#2s) RESULT %s: %s", request->sd, question->qname.c, AddRecord ? "Add" : "Rmv", domain); 312347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 312447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(request, reply); 312547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 312647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 312747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_enum_request(request_state *request) 312847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 312947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 313047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 313147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags reg = flags & kDNSServiceFlagsRegistrationDomains; 313247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_DomainType t_all = reg ? mDNS_DomainTypeRegistration : mDNS_DomainTypeBrowse; 313347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_DomainType t_default = reg ? mDNS_DomainTypeRegistrationDefault : mDNS_DomainTypeBrowseDefault; 313447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 313547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 313647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 313747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 313847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) 313947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceEnumerateDomains(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 314047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 314147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // allocate context structures 314247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt uDNS_SetupSearchDomains(&mDNSStorage, UDNS_START_WAB_QUERY); 314347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 314447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 314547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // mark which kind of enumeration we're doing so we can (de)authorize certain domains 314647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.enumeration.flags = reg; 314747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 314847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 314947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // enumeration requires multiple questions, so we must link all the context pointers so that 315047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // necessary context can be reached from the callbacks 315147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.enumeration.q_all .QuestionContext = request; 315247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.enumeration.q_default.QuestionContext = request; 315347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 315447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // if the caller hasn't specified an explicit interface, we use local-only to get the system-wide list. 315547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!InterfaceID) InterfaceID = mDNSInterface_LocalOnly; 315647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 315747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // make the calls 315847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceEnumerateDomains(%X=%s)", request->sd, flags, 315947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (flags & kDNSServiceFlagsBrowseDomains ) ? "kDNSServiceFlagsBrowseDomains" : 316047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (flags & kDNSServiceFlagsRegistrationDomains) ? "kDNSServiceFlagsRegistrationDomains" : "<<Unknown>>"); 316147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_all, t_all, NULL, InterfaceID, enum_result_callback, request); 316247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err) 316347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 316447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_GetDomains(&mDNSStorage, &request->u.enumeration.q_default, t_default, NULL, InterfaceID, enum_result_callback, request); 316547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) mDNS_StopGetDomains(&mDNSStorage, &request->u.enumeration.q_all); 316647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else request->terminate = enum_termination_callback; 316747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 316847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 316947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 317047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 317147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 317247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 317347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 317447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 317547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceReconfirmRecord & Misc 317647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 317747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 317847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_reconfirm_request(request_state *request) 317947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 318047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus status = mStatus_BadParamErr; 318147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthRecord *rr = read_rr_from_ipc_msg(request, 0, 0); 318247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rr) 318347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 318447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt status = mDNS_ReconfirmByValue(&mDNSStorage, &rr->resrec); 318547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation( 318647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (status == mStatus_NoError) ? 318747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt "%3d: DNSServiceReconfirmRecord(%s) interface %d initiated" : 318847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt "%3d: DNSServiceReconfirmRecord(%s) interface %d failed: %d", 318947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, RRDisplayString(&mDNSStorage, &rr->resrec), 319047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformInterfaceIndexfromInterfaceID(&mDNSStorage, rr->resrec.InterfaceID, mDNSfalse), status); 319147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("AuthRecord/handle_reconfirm_request", rr); 319247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 319347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(status); 319447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 319547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 319647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_setdomain_request(request_state *request) 319747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 319847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char domainstr[MAX_ESCAPED_DOMAIN_NAME]; 319947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname domain; 320047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 320147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)flags; // Unused 320247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, domainstr, MAX_ESCAPED_DOMAIN_NAME) < 0 || 320347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt !MakeDomainNameFromDNSNameString(&domain, domainstr)) 320447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceSetDefaultDomainForUser(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 320547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 320647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceSetDefaultDomainForUser(%##s)", request->sd, domain.c); 320747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(mStatus_NoError); 320847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 320947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 321047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalttypedef packedstruct 321147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 321247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err; 321347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 len; 321447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 vers; 321547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } DaemonVersionReply; 321647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 321747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void handle_getproperty_request(request_state *request) 321847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 321947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const mStatus BadParamErr = dnssd_htonl((mDNSu32)mStatus_BadParamErr); 322047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char prop[256]; 322147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, prop, sizeof(prop)) >= 0) 322247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 322347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceGetProperty(%s)", request->sd, prop); 322447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!strcmp(prop, kDNSServiceProperty_DaemonVersion)) 322547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 322647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DaemonVersionReply x = { 0, dnssd_htonl(4), dnssd_htonl(_DNS_SD_H) }; 322747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt send_all(request->sd, (const char *)&x, sizeof(x)); 322847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 322947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 323047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 323147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 323247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we didn't recogize the requested property name, return BadParamErr 323347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt send_all(request->sd, (const char *)&BadParamErr, sizeof(BadParamErr)); 323447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 323547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 323647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 323747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 323847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 323947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceNATPortMappingCreate 324047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 324147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 324247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define DNSServiceProtocol(X) ((X) == NATOp_AddrRequest ? 0 : (X) == NATOp_MapUDP ? kDNSServiceProtocol_UDP : kDNSServiceProtocol_TCP) 324347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 324447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void port_mapping_termination_callback(request_state *request) 324547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 324647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) STOP", request->sd, 324747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceProtocol(request->u.pm.NATinfo.Protocol), 324847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease); 324947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopNATOperation(&mDNSStorage, &request->u.pm.NATinfo); 325047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 325147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 325247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// Called via function pointer when we get a NAT-PMP address request or port mapping response 325347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void port_mapping_create_request_callback(mDNS *m, NATTraversalInfo *n) 325447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 325547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request = (request_state *)n->clientContext; 325647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 325747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int replyLen; 325847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *data; 325947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 326047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request) { LogMsg("port_mapping_create_request_callback called with unknown request_state object"); return; } 326147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 326247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // calculate reply data length 326347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt replyLen = sizeof(DNSServiceFlags); 326447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt replyLen += 3 * sizeof(mDNSu32); // if index + addr + ttl 326547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt replyLen += sizeof(DNSServiceErrorType); 326647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt replyLen += 2 * sizeof(mDNSu16); // Internal Port + External Port 326747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt replyLen += sizeof(mDNSu8); // protocol 326847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 326947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep = create_reply(port_mapping_reply_op, replyLen, request); 327047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 327147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->flags = dnssd_htonl(0); 327247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->ifi = dnssd_htonl(mDNSPlatformInterfaceIndexfromInterfaceID(m, n->InterfaceID, mDNSfalse)); 327347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->rhdr->error = dnssd_htonl(n->Result); 327447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 327547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt data = (char *)&rep->rhdr[1]; 327647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 327747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalAddress.b[0]; 327847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalAddress.b[1]; 327947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalAddress.b[2]; 328047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalAddress.b[3]; 328147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = DNSServiceProtocol(request->u.pm.NATinfo.Protocol); 328247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.IntPort.b[0]; 328347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.IntPort.b[1]; 328447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalPort.b[0]; 328547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *data++ = request->u.pm.NATinfo.ExternalPort.b[1]; 328647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt put_uint32(request->u.pm.NATinfo.Lifetime, &data); 328747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 328847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) RESULT %.4a:%u TTL %u", request->sd, 328947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceProtocol(request->u.pm.NATinfo.Protocol), 329047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease, 329147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &request->u.pm.NATinfo.ExternalAddress, mDNSVal16(request->u.pm.NATinfo.ExternalPort), request->u.pm.NATinfo.Lifetime); 329247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 329347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt append_reply(request, rep); 329447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 329547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 329647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_port_mapping_request(request_state *request) 329747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 329847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 ttl = 0; 329947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = mStatus_NoError; 330047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 330147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 330247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 330347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 330447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu8 protocol = (mDNSu8)get_uint32(&request->msgptr, request->msgend); 330547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)flags; // Unused 330647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !InterfaceID) return(mStatus_BadParamErr); 330747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->msgptr + 8 > request->msgend) request->msgptr = NULL; 330847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 330947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 331047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.IntPort.b[0] = *request->msgptr++; 331147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.IntPort.b[1] = *request->msgptr++; 331247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.ReqExt.b[0] = *request->msgptr++; 331347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.ReqExt.b[1] = *request->msgptr++; 331447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ttl = get_uint32(&request->msgptr, request->msgend); 331547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 331647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 331747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) 331847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: DNSServiceNATPortMappingCreate(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 331947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 332047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (protocol == 0) // If protocol == 0 (i.e. just request public address) then IntPort, ExtPort, ttl must be zero too 332147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 332247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort) || !mDNSIPPortIsZero(request->u.pm.ReqExt) || ttl) return(mStatus_BadParamErr); 332347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 332447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 332547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 332647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSIPPortIsZero(request->u.pm.NATinfo.IntPort)) return(mStatus_BadParamErr); 332747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!(protocol & (kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP))) return(mStatus_BadParamErr); 332847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 332947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 333047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.Protocol = !protocol ? NATOp_AddrRequest : (protocol == kDNSServiceProtocol_UDP) ? NATOp_MapUDP : NATOp_MapTCP; 333147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // u.pm.NATinfo.IntPort = already set above 333247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.RequestedPort = request->u.pm.ReqExt; 333347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.NATLease = ttl; 333447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.clientCallback = port_mapping_create_request_callback; 333547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.pm.NATinfo.clientContext = request; 333647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 333747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceNATPortMappingCreate(%X, %u, %u, %d) START", request->sd, 333847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt protocol, mDNSVal16(request->u.pm.NATinfo.IntPort), mDNSVal16(request->u.pm.ReqExt), request->u.pm.NATinfo.NATLease); 333947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartNATOperation(&mDNSStorage, &request->u.pm.NATinfo); 334047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err) LogMsg("ERROR: mDNS_StartNATOperation: %d", (int)err); 334147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else request->terminate = port_mapping_termination_callback; 334247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 334347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 334447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 334547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 334647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 334747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 334847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 334947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - DNSServiceGetAddrInfo 335047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 335147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 335247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void addrinfo_termination_callback(request_state *request) 335347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 335447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceGetAddrInfo(%##s) STOP", request->sd, request->u.addrinfo.q4.qname.c); 335547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 335647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q4.QuestionContext) 335747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 335847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4); 335947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.QuestionContext = mDNSNULL; 336047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 336147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q4.qnameOrig) 336247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 336347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination", request->u.addrinfo.q4.qnameOrig); 336447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.qnameOrig = mDNSNULL; 336547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 336647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q42) 336747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 336847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q42->QuestionContext) 336947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 337047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("addrinfo_termination_callback: Stopping q42 %##s", request->u.addrinfo.q42->qname.c); 337147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q42); 337247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 337347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q42->qnameOrig) 337447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 337547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("addrinfo_termination_callback: freeing q42 qnameOrig %##s", request->u.addrinfo.q42->qnameOrig->c); 337647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination q42", request->u.addrinfo.q42->qnameOrig); 337747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q42->qnameOrig = mDNSNULL; 337847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 337947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("addrinfo Q42", request->u.addrinfo.q42); 338047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q42 = mDNSNULL; 338147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 338247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 338347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q6.QuestionContext) 338447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 338547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q6); 338647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.QuestionContext = mDNSNULL; 338747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 338847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q6.qnameOrig) 338947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 339047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination", request->u.addrinfo.q6.qnameOrig); 339147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.qnameOrig = mDNSNULL; 339247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 339347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q62) 339447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 339547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q62->QuestionContext) 339647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 339747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("addrinfo_termination_callback: Stopping q62 %##s", request->u.addrinfo.q62->qname.c); 339847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, request->u.addrinfo.q62); 339947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 340047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.q62->qnameOrig) 340147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 340247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("addrinfo_termination_callback: freeing q62 qnameOrig %##s", request->u.addrinfo.q62->qnameOrig->c); 340347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("QueryTermination q62", request->u.addrinfo.q62->qnameOrig); 340447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q62->qnameOrig = mDNSNULL; 340547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 340647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("addrinfo Q62", request->u.addrinfo.q62); 340747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q62 = mDNSNULL; 340847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 340947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 341047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 341147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mStatus handle_addrinfo_request(request_state *request) 341247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 341347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char hostname[256]; 341447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt domainname d; 341547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = 0; 341647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 341747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSServiceFlags flags = get_flags(&request->msgptr, request->msgend); 341847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 interfaceIndex = get_uint32(&request->msgptr, request->msgend); 341947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 342047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(&request->u.addrinfo, sizeof(request->u.addrinfo)); 342147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.interface_id = mDNSPlatformInterfaceIDfromInterfaceIndex(&mDNSStorage, interfaceIndex); 342247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.flags = flags; 342347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.protocol = get_uint32(&request->msgptr, request->msgend); 342447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 342547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (interfaceIndex && !request->u.addrinfo.interface_id) return(mStatus_BadParamErr); 342647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.protocol > (kDNSServiceProtocol_IPv4|kDNSServiceProtocol_IPv6)) return(mStatus_BadParamErr); 342747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 342847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (get_string(&request->msgptr, request->msgend, hostname, 256) < 0) return(mStatus_BadParamErr); 342947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 343047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->msgptr) { LogMsg("%3d: DNSServiceGetAddrInfo(unreadable parameters)", request->sd); return(mStatus_BadParamErr); } 343147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 343247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!MakeDomainNameFromDNSNameString(&d, hostname)) 343347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("ERROR: handle_addrinfo_request: bad hostname: %s", hostname); return(mStatus_BadParamErr); } 343447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 343547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if 0 343647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AuthorizedDomain(request, &d, AutoBrowseDomains)) return (mStatus_NoError); 343747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 343847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 343947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!request->u.addrinfo.protocol) 344047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 344147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt flags |= kDNSServiceFlagsSuppressUnusable; 344247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.protocol = (kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6); 344347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 344447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 344547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.InterfaceID = request->u.addrinfo.q6.InterfaceID = request->u.addrinfo.interface_id; 344647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.Target = request->u.addrinfo.q6.Target = zeroAddr; 344747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.qname = request->u.addrinfo.q6.qname = d; 344847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.qclass = request->u.addrinfo.q6.qclass = kDNSServiceClass_IN; 344947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.LongLived = request->u.addrinfo.q6.LongLived = (flags & kDNSServiceFlagsLongLivedQuery ) != 0; 345047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.ExpectUnique = request->u.addrinfo.q6.ExpectUnique = mDNSfalse; 345147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.ForceMCast = request->u.addrinfo.q6.ForceMCast = (flags & kDNSServiceFlagsForceMulticast ) != 0; 345247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.ReturnIntermed = request->u.addrinfo.q6.ReturnIntermed = (flags & kDNSServiceFlagsReturnIntermediates) != 0; 345347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.SuppressUnusable = request->u.addrinfo.q6.SuppressUnusable = (flags & kDNSServiceFlagsSuppressUnusable ) != 0; 345447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.TimeoutQuestion = request->u.addrinfo.q6.TimeoutQuestion = (flags & kDNSServiceFlagsTimeout ) != 0; 345547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.WakeOnResolve = request->u.addrinfo.q6.WakeOnResolve = 0; 345647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.qnameOrig = request->u.addrinfo.q6.qnameOrig = mDNSNULL; 345747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 345847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4) 345947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 346047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.qtype = kDNSServiceType_A; 346147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.SearchListIndex = 0; 346247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 346347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We append search domains only for queries that are a single label. If overriden using 346447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // command line argument "AlwaysAppendSearchDomains", then we do it for any query which 346547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // is not fully qualified. 346647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (hostname[strlen(hostname) - 1] != '.' && (AlwaysAppendSearchDomains || CountLabels(&d) == 1)) 346747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 346847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.AppendSearchDomains = 1; 346947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.AppendLocalSearchDomains = 1; 347047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 347147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 347247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 347347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.AppendSearchDomains = 0; 347447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.AppendLocalSearchDomains = 0; 347547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 347647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q4) ? 1 : 0); 347747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.QuestionCallback = queryrecord_result_callback; 347847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.QuestionContext = request; 347947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q4); 348047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err != mStatus_NoError) 348147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 348247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: mDNS_StartQuery: %d", (int)err); 348347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.QuestionContext = mDNSNULL; 348447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 348547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #if APPLE_OSX_mDNSResponder 348647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = SendAdditionalQuery(&request->u.addrinfo.q4, request, err); 348747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #endif // APPLE_OSX_mDNSResponder 348847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 348947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 349047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err && (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv6)) 349147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 349247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.qtype = kDNSServiceType_AAAA; 349347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.SearchListIndex = 0; 349447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (hostname[strlen(hostname) - 1] != '.' && (AlwaysAppendSearchDomains || CountLabels(&d) == 1)) 349547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 349647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.AppendSearchDomains = 1; 349747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.AppendLocalSearchDomains = 1; 349847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 349947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 350047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 350147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.AppendSearchDomains = 0; 350247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.AppendLocalSearchDomains = 0; 350347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 350447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.RetryWithSearchDomains = (ApplySearchDomainsFirst(&request->u.addrinfo.q6) ? 1 : 0); 350547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.QuestionCallback = queryrecord_result_callback; 350647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.QuestionContext = request; 350747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mDNS_StartQuery(&mDNSStorage, &request->u.addrinfo.q6); 350847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (err != mStatus_NoError) 350947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 351047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: mDNS_StartQuery: %d", (int)err); 351147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q6.QuestionContext = mDNSNULL; 351247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (request->u.addrinfo.protocol & kDNSServiceProtocol_IPv4) 351347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 351447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we started a query for IPv4, we need to cancel it 351547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_StopQuery(&mDNSStorage, &request->u.addrinfo.q4); 351647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->u.addrinfo.q4.QuestionContext = mDNSNULL; 351747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 351847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 351947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #if APPLE_OSX_mDNSResponder 352047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = SendAdditionalQuery(&request->u.addrinfo.q6, request, err); 352147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #endif // APPLE_OSX_mDNSResponder 352247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 352347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 352447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: DNSServiceGetAddrInfo(%X, %d, %d, %##s) START", 352547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd, flags, interfaceIndex, request->u.addrinfo.protocol, d.c); 352647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 352747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!err) request->terminate = addrinfo_termination_callback; 352847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 352947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(err); 353047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 353147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 353247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// *************************************************************************** 353347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if COMPILER_LIKES_PRAGMA_MARK 353447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - 353547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#pragma mark - Main Request Handler etc. 353647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 353747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 353847e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal request_state *NewRequest(void) 353947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 354047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state **p = &all_requests; 354147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*p) p=&(*p)->next; 354247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *p = mallocL("request_state", sizeof(request_state)); 354347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!*p) FatalError("ERROR: malloc"); 354447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(*p, sizeof(request_state)); 354547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(*p); 354647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 354747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 354847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// read_msg may be called any time when the transfer state (req->ts) is t_morecoming. 354947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// if there is no data on the socket, the socket will be closed and t_terminated will be returned 355047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void read_msg(request_state *req) 355147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 355247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts == t_terminated || req->ts == t_error) 355347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: read_msg called with transfer state terminated or error", req->sd); req->ts = t_error; return; } 355447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 355547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts == t_complete) // this must be death or something is wrong 355647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 355747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char buf[4]; // dummy for death notification 355847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int nread = udsSupportReadFD(req->sd, buf, 4, 0, req->platform_data); 355947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!nread) { req->ts = t_terminated; return; } 356047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nread < 0) goto rerror; 356147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: ERROR: read data from a completed request", req->sd); 356247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_error; 356347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 356447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 356547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 356647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts != t_morecoming) 356747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: read_msg called with invalid transfer state (%d)", req->sd, req->ts); req->ts = t_error; return; } 356847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 356947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr_bytes < sizeof(ipc_msg_hdr)) 357047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 357147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 nleft = sizeof(ipc_msg_hdr) - req->hdr_bytes; 357247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int nread = udsSupportReadFD(req->sd, (char *)&req->hdr + req->hdr_bytes, nleft, 0, req->platform_data); 357347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nread == 0) { req->ts = t_terminated; return; } 357447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nread < 0) goto rerror; 357547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->hdr_bytes += nread; 357647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr_bytes > sizeof(ipc_msg_hdr)) 357747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: read_msg - read too many header bytes", req->sd); req->ts = t_error; return; } 357847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 357947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // only read data if header is complete 358047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr_bytes == sizeof(ipc_msg_hdr)) 358147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 358247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertHeaderBytes(&req->hdr); 358347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.version != VERSION) 358447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: client version 0x%08X daemon version 0x%08X", req->sd, req->hdr.version, VERSION); req->ts = t_error; return; } 358547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 358647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Largest conceivable single request is a DNSServiceRegisterRecord() or DNSServiceAddRecord() 358747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // with 64kB of rdata. Adding 1009 byte for a maximal domain name, plus a safety margin 358847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // for other overhead, this means any message above 70kB is definitely bogus. 358947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.datalen > 70000) 359047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: read_msg: hdr.datalen %u (0x%X) > 70000", req->sd, req->hdr.datalen, req->hdr.datalen); req->ts = t_error; return; } 359147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgbuf = mallocL("request_state msgbuf", req->hdr.datalen + MSG_PAD_BYTES); 359247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!req->msgbuf) { my_perror("ERROR: malloc"); req->ts = t_error; return; } 359347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgptr = req->msgbuf; 359447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgend = req->msgbuf + req->hdr.datalen; 359547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(req->msgbuf, req->hdr.datalen + MSG_PAD_BYTES); 359647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 359747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 359847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 359947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If our header is complete, but we're still needing more body data, then try to read it now 360047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: For cancel_request req->hdr.datalen == 0, but there's no error return socket for cancel_request 360147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Any time we need to get the error return socket we know we'll have at least one data byte 360247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // (even if only the one-byte empty C string placeholder for the old ctrl_path parameter) 360347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes < req->hdr.datalen) 360447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 360547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 nleft = req->hdr.datalen - req->data_bytes; 360647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int nread; 360747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(_WIN32) 360847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct iovec vec = { req->msgbuf + req->data_bytes, nleft }; // Tell recvmsg where we want the bytes put 360947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct msghdr msg; 361047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct cmsghdr *cmsg; 361147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char cbuf[CMSG_SPACE(sizeof(dnssd_sock_t))]; 361247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_name = 0; 361347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_namelen = 0; 361447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_iov = &vec; 361547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_iovlen = 1; 361647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_control = cbuf; 361747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_controllen = sizeof(cbuf); 361847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt msg.msg_flags = 0; 361947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nread = recvmsg(req->sd, &msg, 0); 362047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 362147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nread = udsSupportReadFD(req->sd, (char *)req->msgbuf + req->data_bytes, nleft, 0, req->platform_data); 362247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 362347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nread == 0) { req->ts = t_terminated; return; } 362447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nread < 0) goto rerror; 362547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->data_bytes += nread; 362647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->data_bytes > req->hdr.datalen) 362747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: ERROR: read_msg - read too many data bytes", req->sd); req->ts = t_error; return; } 362847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(_WIN32) 362947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cmsg = CMSG_FIRSTHDR(&msg); 363047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if DEBUG_64BIT_SCM_RIGHTS 363147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Expecting %d %d %d %d", req->sd, sizeof(cbuf), sizeof(cbuf), SOL_SOCKET, SCM_RIGHTS); 363247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Got %d %d %d %d", req->sd, msg.msg_controllen, cmsg->cmsg_len, cmsg->cmsg_level, cmsg->cmsg_type); 363347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // DEBUG_64BIT_SCM_RIGHTS 363447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (msg.msg_controllen == sizeof(cbuf) && 363547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cmsg->cmsg_len == CMSG_LEN(sizeof(dnssd_sock_t)) && 363647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cmsg->cmsg_level == SOL_SOCKET && 363747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cmsg->cmsg_type == SCM_RIGHTS) 363847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 363947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 364047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Strictly speaking BPF_fd belongs solely in the platform support layer, but because 364147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // of privilege separation on Mac OS X we need to get BPF_fd from mDNSResponderHelper, 364247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // and it's convenient to repurpose the existing fd-passing code here for that task 364347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.op == send_bpf) 364447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 364547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sock_t x = *(dnssd_sock_t *)CMSG_DATA(cmsg); 364647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: Got BPF %d", req->sd, x); 364747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformReceiveBPF_fd(&mDNSStorage, x); 364847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 364947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 365047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 365147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->errsd = *(dnssd_sock_t *)CMSG_DATA(cmsg); 365247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if DEBUG_64BIT_SCM_RIGHTS 365347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: read req->errsd %d", req->sd, req->errsd); 365447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // DEBUG_64BIT_SCM_RIGHTS 365547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->data_bytes < req->hdr.datalen) 365647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 365747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Client sent error socket %d via SCM_RIGHTS with req->data_bytes %d < req->hdr.datalen %d", 365847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, req->errsd, req->data_bytes, req->hdr.datalen); 365947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_error; 366047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 366147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 366247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 366347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 366447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 366547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 366647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If our header and data are both complete, see if we need to make our separate error return socket 366747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr_bytes == sizeof(ipc_msg_hdr) && req->data_bytes == req->hdr.datalen) 366847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 366947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate && req->hdr.op != cancel_request) 367047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 367147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sockaddr_t cliaddr; 367247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(USE_TCP_LOOPBACK) 367347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSOpaque16 port; 367447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt u_long opt = 1; 367547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt port.b[0] = req->msgptr[0]; 367647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt port.b[1] = req->msgptr[1]; 367747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgptr += 2; 367847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cliaddr.sin_family = AF_INET; 367947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cliaddr.sin_port = port.NotAnInteger; 368047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cliaddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR); 368147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 368247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char ctrl_path[MAX_CTLPATH]; 368347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt get_string(&req->msgptr, req->msgend, ctrl_path, MAX_CTLPATH); // path is first element in message buffer 368447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(&cliaddr, sizeof(cliaddr)); 368547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cliaddr.sun_family = AF_LOCAL; 368647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformStrCopy(cliaddr.sun_path, ctrl_path); 368747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the error return path UDS name is empty string, that tells us 368847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // that this is a new version of the library that's going to pass us 368947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // the error return path socket via sendmsg/recvmsg 369047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ctrl_path[0] == 0) 369147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 369247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->errsd == req->sd) 369347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("%3d: read_msg: ERROR failed to get errsd via SCM_RIGHTS", req->sd); req->ts = t_error; return; } 369447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto got_errfd; 369547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 369647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 369747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 369847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->errsd = socket(AF_DNSSD, SOCK_STREAM, 0); 369947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!dnssd_SocketValid(req->errsd)) { my_perror("ERROR: socket"); req->ts = t_error; return; } 370047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 370147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (connect(req->errsd, (struct sockaddr *)&cliaddr, sizeof(cliaddr)) < 0) 370247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 370347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(USE_TCP_LOOPBACK) 370447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct stat sb; 370547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: read_msg: Couldn't connect to error return path socket “%s” errno %d (%s)", 370647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno)); 370747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (stat(cliaddr.sun_path, &sb) < 0) 370847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: read_msg: stat failed “%s” errno %d (%s)", req->sd, cliaddr.sun_path, dnssd_errno, dnssd_strerror(dnssd_errno)); 370947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 371047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 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); 371147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 371247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_error; 371347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 371447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 371547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 371647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(USE_TCP_LOOPBACK) 371747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltgot_errfd: 371847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 371947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: Error socket %d created %08X %08X", req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0]); 372047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(_WIN32) 372147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ioctlsocket(req->errsd, FIONBIO, &opt) != 0) 372247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 372347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (fcntl(req->errsd, F_SETFL, fcntl(req->errsd, F_GETFL, 0) | O_NONBLOCK) != 0) 372447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 372547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 372647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: ERROR: could not set control socket to non-blocking mode errno %d (%s)", 372747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 372847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_error; 372947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 373047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 373147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 373247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 373347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_complete; 373447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 373547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 373647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 373747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 373847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltrerror: 373947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_errno == dnssd_EWOULDBLOCK || dnssd_errno == dnssd_EINTR) return; 374047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: ERROR: read_msg errno %d (%s)", req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 374147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_error; 374247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 374347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 374447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define RecordOrientedOp(X) \ 374547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ((X) == reg_record_request || (X) == add_record_request || (X) == update_record_request || (X) == remove_record_request) 374647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 374747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt// The lightweight operations are the ones that don't need a dedicated request_state structure allocated for them 374847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define LightweightOp(X) (RecordOrientedOp(X) || (X) == cancel_request) 374947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 375047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void request_callback(int fd, short filter, void *info) 375147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 375247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mStatus err = 0; 375347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *req = info; 375447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 min_size = sizeof(DNSServiceFlags); 375547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)fd; // Unused 375647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)filter; // Unused 375747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 375847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (;;) 375947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 376047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt read_msg(req); 376147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts == t_morecoming) return; 376247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts == t_terminated || req->ts == t_error) { AbortUnlinkAndFree(req); return; } 376347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->ts != t_complete) { LogMsg("req->ts %d != t_complete", req->ts); AbortUnlinkAndFree(req); return; } 376447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 376547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.version != VERSION) 376647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 376747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: client version %d incompatible with daemon version %d", req->hdr.version, VERSION); 376847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AbortUnlinkAndFree(req); 376947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 377047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 377147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 377247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt switch(req->hdr.op) // Interface + other data 377347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 377447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case connection_request: min_size = 0; break; 377547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reg_service_request: min_size += sizeof(mDNSu32) + 4 /* name, type, domain, host */ + 4 /* port, textlen */; break; 377647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case add_record_request: min_size += 4 /* type, rdlen */ + 4 /* ttl */; break; 377747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case update_record_request: min_size += 2 /* rdlen */ + 4 /* ttl */; break; 377847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case remove_record_request: break; 377947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case browse_request: min_size += sizeof(mDNSu32) + 2 /* type, domain */; break; 378047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case resolve_request: min_size += sizeof(mDNSu32) + 3 /* type, type, domain */; break; 378147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case query_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 4 /* type, class*/; break; 378247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case enumeration_request: min_size += sizeof(mDNSu32); break; 378347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reg_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */ + 4 /* ttl */; break; 378447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reconfirm_record_request: min_size += sizeof(mDNSu32) + 1 /* name */ + 6 /* type, class, rdlen */; break; 378547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case setdomain_request: min_size += 1 /* domain */; break; 378647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case getproperty_request: min_size = 2; break; 378747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case port_mapping_request: min_size += sizeof(mDNSu32) + 4 /* udp/tcp */ + 4 /* int/ext port */ + 4 /* ttl */; break; 378847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case addrinfo_request: min_size += sizeof(mDNSu32) + 4 /* v4/v6 */ + 1 /* hostname */; break; 378947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case send_bpf: // Same as cancel_request below 37900b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon case cancel_request: min_size = 0; break; 37910b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon case sethost_request: min_size = sizeof(mDNSu32) + 1 /* hostname */; break; 379247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt default: LogMsg("ERROR: validate_message - unsupported req type: %d", req->hdr.op); min_size = -1; break; 379347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 379447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 379547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((mDNSs32)req->data_bytes < min_size) 379647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("Invalid message %d bytes; min for %d is %d", req->data_bytes, req->hdr.op, min_size); AbortUnlinkAndFree(req); return; } 379747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 379847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (LightweightOp(req->hdr.op) && !req->terminate) 379947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { LogMsg("Reg/Add/Update/Remove %d require existing connection", req->hdr.op); AbortUnlinkAndFree(req); return; } 380047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 380147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // check if client wants silent operation 380247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.ipc_flags & IPC_FLAGS_NOREPLY) req->no_reply = 1; 380347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 380447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If req->terminate is already set, this means this operation is sharing an existing connection 380547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate && !LightweightOp(req->hdr.op)) 380647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 380747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *newreq = NewRequest(); 380847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->primary = req; 380947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->sd = req->sd; 381047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->errsd = req->errsd; 381147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->uid = req->uid; 381247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->hdr = req->hdr; 381347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->msgbuf = req->msgbuf; 381447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->msgptr = req->msgptr; 381547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newreq->msgend = req->msgend; 381647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req = newreq; 381747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 381847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 381947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If we're shutting down, don't allow new client requests 382047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We do allow "cancel" and "getproperty" during shutdown 382147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSStorage.ShutdownTime && req->hdr.op != cancel_request && req->hdr.op != getproperty_request) 382247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 382347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt err = mStatus_ServiceNotRunning; 382447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 382547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else switch(req->hdr.op) 382647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 382747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // These are all operations that have their own first-class request_state object 382847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case connection_request: LogOperation("%3d: DNSServiceCreateConnection START", req->sd); 382947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->terminate = connection_termination; break; 383047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case resolve_request: err = handle_resolve_request (req); break; 383147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case query_request: err = handle_queryrecord_request (req); break; 383247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case browse_request: err = handle_browse_request (req); break; 383347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reg_service_request: err = handle_regservice_request (req); break; 383447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case enumeration_request: err = handle_enum_request (req); break; 383547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reconfirm_record_request: err = handle_reconfirm_request (req); break; 383647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case setdomain_request: err = handle_setdomain_request (req); break; 383747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case getproperty_request: handle_getproperty_request (req); break; 383847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case port_mapping_request: err = handle_port_mapping_request(req); break; 383947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case addrinfo_request: err = handle_addrinfo_request (req); break; 384047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case send_bpf: /* Do nothing for send_bpf */ break; 384147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 384247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // These are all operations that work with an existing request_state object 384347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case reg_record_request: err = handle_regrecord_request (req); break; 384447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case add_record_request: err = handle_add_request (req); break; 384547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case update_record_request: err = handle_update_request (req); break; 384647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case remove_record_request: err = handle_removerecord_request(req); break; 384747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case cancel_request: handle_cancel_request (req); break; 38480b6b4e32c36ce143bfd095119b4939e4cf826021Joshua Melcon case sethost_request: err = handle_sethost_request (req); break; 384947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt default: LogMsg("%3d: ERROR: Unsupported UDS req: %d", req->sd, req->hdr.op); 385047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 385147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 385247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // req->msgbuf may be NULL, e.g. for connection_request or remove_record_request 385347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->msgbuf) freeL("request_state msgbuf", req->msgbuf); 385447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 385547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // There's no return data for a cancel request (DNSServiceRefDeallocate returns no result) 385647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For a DNSServiceGetProperty call, the handler already generated the response, so no need to do it again here 385747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->hdr.op != cancel_request && req->hdr.op != getproperty_request && req->hdr.op != send_bpf) 385847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 385947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const mStatus err_netorder = dnssd_htonl(err); 386047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt send_all(req->errsd, (const char *)&err_netorder, sizeof(err_netorder)); 386147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->errsd != req->sd) 386247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 386347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: Error socket %d closed %08X %08X (%d)", 386447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->sd, req->errsd, req->hdr.client_context.u32[1], req->hdr.client_context.u32[0], err); 386547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_close(req->errsd); 386647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->errsd = req->sd; 386747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Also need to reset the parent's errsd, if this is a subordinate operation 386847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary) req->primary->errsd = req->primary->sd; 386947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 387047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 387147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 387247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Reset ready to accept the next req on this pipe 387347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary) req = req->primary; 387447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->ts = t_morecoming; 387547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->hdr_bytes = 0; 387647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->data_bytes = 0; 387747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgbuf = mDNSNULL; 387847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgptr = mDNSNULL; 387947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->msgend = 0; 388047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 388147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 388247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 388347e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void connect_callback(int fd, short filter, void *info) 388447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 388547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sockaddr_t cliaddr; 388647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_socklen_t len = (dnssd_socklen_t) sizeof(cliaddr); 388747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sock_t sd = accept(fd, (struct sockaddr*) &cliaddr, &len); 388847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(SO_NOSIGPIPE) || defined(_WIN32) 388947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt unsigned long optval = 1; 389047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 389147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 389247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)filter; // Unused 389347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (void)info; // Unused 389447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 389547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!dnssd_SocketValid(sd)) 389647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 389747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_errno != dnssd_EWOULDBLOCK) my_perror("ERROR: accept"); 389847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 389947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 390047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 390147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#ifdef SO_NOSIGPIPE 390247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Some environments (e.g. OS X) support turning off SIGPIPE for a socket 390347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (setsockopt(sd, SOL_SOCKET, SO_NOSIGPIPE, &optval, sizeof(optval)) < 0) 390447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: WARNING: setsockopt - SO_NOSIGPIPE %d (%s)", sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 390547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 390647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 390747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(_WIN32) 390847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ioctlsocket(sd, FIONBIO, &optval) != 0) 390947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 391047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (fcntl(sd, F_SETFL, fcntl(sd, F_GETFL, 0) | O_NONBLOCK) != 0) 391147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 391247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 391347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: fcntl(sd, F_SETFL, O_NONBLOCK) - aborting client"); 391447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_close(sd); 391547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return; 391647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 391747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 391847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 391947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *request = NewRequest(); 392047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->ts = t_morecoming; 392147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->sd = sd; 392247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request->errsd = sd; 392347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder 392447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct xucred x; 392547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt socklen_t xucredlen = sizeof(x); 392647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (getsockopt(sd, 0, LOCAL_PEERCRED, &x, &xucredlen) >= 0 && x.cr_version == XUCRED_VERSION) request->uid = x.cr_uid; 392747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else my_perror("ERROR: getsockopt, LOCAL_PEERCRED"); 392847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("LOCAL_PEERCRED %d %u %u %d", xucredlen, x.cr_version, x.cr_uid, x.cr_ngroups); 392947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder 393047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogOperation("%3d: Adding FD for uid %u", request->sd, request->uid); 393147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsSupportAddFDToEventLoop(sd, request_callback, request, &request->platform_data); 393247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 393347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 393447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 393547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal mDNSBool uds_socket_setup(dnssd_sock_t skt) 393647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 393747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(SO_NP_EXTENSIONS) 393847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct so_np_extensions sonpx; 393947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt socklen_t optlen = sizeof(struct so_np_extensions); 394047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt sonpx.npx_flags = SONPX_SETOPTSHUT; 394147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt sonpx.npx_mask = SONPX_SETOPTSHUT; 394247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (setsockopt(skt, SOL_SOCKET, SO_NP_EXTENSIONS, &sonpx, optlen) < 0) 394347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("WARNING: could not set sockopt - SO_NP_EXTENSIONS"); 394447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 394547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if defined(_WIN32) 394647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // SEH: do we even need to do this on windows? 394747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // This socket will be given to WSAEventSelect which will automatically set it to non-blocking 394847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt u_long opt = 1; 394947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ioctlsocket(skt, FIONBIO, &opt) != 0) 395047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#else 395147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (fcntl(skt, F_SETFL, fcntl(skt, F_GETFL, 0) | O_NONBLOCK) != 0) 395247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 395347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 395447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: could not set listen socket to non-blocking mode"); 395547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNSfalse; 395647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 395747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 395847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (listen(skt, LISTENQ) != 0) 395947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 396047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: could not listen on listen socket"); 396147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNSfalse; 396247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 396347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 396447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mStatus_NoError != udsSupportAddFDToEventLoop(skt, connect_callback, (void *) NULL, (void **) NULL)) 396547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 396647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: could not add listen socket to event loop"); 396747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNSfalse; 396847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 396947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else LogOperation("%3d: Listening for incoming Unix Domain Socket client requests", skt); 397047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 397147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return mDNStrue; 397247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 397347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 397447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport int udsserver_init(dnssd_sock_t skts[], mDNSu32 count) 397547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 397647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_sockaddr_t laddr; 397747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int ret; 397847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 i = 0; 397947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 398047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogInfo("udsserver_init"); 398147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 398247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If a particular platform wants to opt out of having a PID file, define PID_FILE to be "" 398347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (PID_FILE[0]) 398447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 398547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt FILE *fp = fopen(PID_FILE, "w"); 398647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (fp != NULL) 398747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 398847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt fprintf(fp, "%d\n", getpid()); 398947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt fclose(fp); 399047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 399147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 399247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 399347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (skts) 399447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 399547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (i = 0; i < count; i++) 399647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_SocketValid(skts[i]) && !uds_socket_setup(skts[i])) 399747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto error; 399847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 399947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 400047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 400147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt listenfd = socket(AF_DNSSD, SOCK_STREAM, 0); 400247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!dnssd_SocketValid(listenfd)) 400347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 400447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: socket(AF_DNSSD, SOCK_STREAM, 0); failed"); 400547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto error; 400647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 400747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 400847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformMemZero(&laddr, sizeof(laddr)); 400947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 401047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #if defined(USE_TCP_LOOPBACK) 401147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 401247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt laddr.sin_family = AF_INET; 401347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt laddr.sin_port = htons(MDNS_TCP_SERVERPORT); 401447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt laddr.sin_addr.s_addr = inet_addr(MDNS_TCP_SERVERADDR); 401547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); 401647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ret < 0) 401747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 401847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed"); 401947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto error; 402047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 402147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 402247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #else 402347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 402447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mode_t mask = umask(0); 402547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt unlink(MDNS_UDS_SERVERPATH); // OK if this fails 402647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt laddr.sun_family = AF_LOCAL; 402747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #ifndef NOT_HAVE_SA_LEN 402847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // According to Stevens (section 3.2), there is no portable way to 402947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // determine whether sa_len is defined on a particular platform. 403047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt laddr.sun_len = sizeof(struct sockaddr_un); 403147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #endif 403247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (strlen(MDNS_UDS_SERVERPATH) >= sizeof(laddr.sun_path)) 403347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 403447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("ERROR: MDNS_UDS_SERVERPATH must be < %d characters", (int)sizeof(laddr.sun_path)); 403547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto error; 403647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 403747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSPlatformStrCopy(laddr.sun_path, MDNS_UDS_SERVERPATH); 403847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ret = bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); 403947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt umask(mask); 404047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ret < 0) 404147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 404247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: bind(listenfd, (struct sockaddr *) &laddr, sizeof(laddr)); failed"); 404347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt goto error; 404447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 404547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 404647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #endif 404747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 404847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!uds_socket_setup(listenfd)) goto error; 404947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 405047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 405147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(PLATFORM_NO_RLIMIT) 405247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 405347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Set maximum number of open file descriptors 405447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #define MIN_OPENFILES 10240 405547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct rlimit maxfds, newfds; 405647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 405747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Due to bugs in OS X (<rdar://problem/2941095>, <rdar://problem/3342704>, <rdar://problem/3839173>) 405847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // you have to get and set rlimits once before getrlimit will return sensible values 405947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 406047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (setrlimit(RLIMIT_NOFILE, &maxfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit"); 406147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 406247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 406347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newfds.rlim_max = (maxfds.rlim_max > MIN_OPENFILES) ? maxfds.rlim_max : MIN_OPENFILES; 406447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt newfds.rlim_cur = (maxfds.rlim_cur > MIN_OPENFILES) ? maxfds.rlim_cur : MIN_OPENFILES; 406547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (newfds.rlim_max != maxfds.rlim_max || newfds.rlim_cur != maxfds.rlim_cur) 406647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (setrlimit(RLIMIT_NOFILE, &newfds) < 0) my_perror("ERROR: Unable to set maximum file descriptor limit"); 406747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 406847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (getrlimit(RLIMIT_NOFILE, &maxfds) < 0) { my_perror("ERROR: Unable to get file descriptor limit"); return 0; } 406947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("maxfds.rlim_max %d", (long)maxfds.rlim_max); 407047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("maxfds.rlim_cur %d", (long)maxfds.rlim_cur); 407147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 407247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 407347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 407447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // We start a "LocalOnly" query looking for Automatic Browse Domain records. 407547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // When Domain Enumeration in uDNS.c finds an "lb" record from the network, its "FoundDomain" routine 407647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // creates a "LocalOnly" record, which results in our AutomaticBrowseDomainChange callback being invoked 407747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_GetDomains(&mDNSStorage, &mDNSStorage.AutomaticBrowseDomainQ, mDNS_DomainTypeBrowseAutomatic, 407847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSNULL, mDNSInterface_LocalOnly, AutomaticBrowseDomainChange, mDNSNULL); 407947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 408047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Add "local" as recommended registration domain ("dns-sd -E"), recommended browsing domain ("dns-sd -F"), and automatic browsing domain 408147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeRegistration); 408247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt RegisterLocalOnlyDomainEnumPTR(&mDNSStorage, &localdomain, mDNS_DomainTypeBrowse); 408347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AddAutoBrowseDomain(0, &localdomain); 408447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 408547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt udsserver_handle_configchange(&mDNSStorage); 408647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return 0; 408747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 408847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalterror: 408947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 409047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt my_perror("ERROR: udsserver_init"); 409147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return -1; 409247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 409347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 409447e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport int udsserver_exit(void) 409547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 409647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Cancel all outstanding client requests 409747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (all_requests) AbortUnlinkAndFree(all_requests); 409847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 409947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Clean up any special mDNSInterface_LocalOnly records we created, both the entries for "local" we 410047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // created in udsserver_init, and others we created as a result of reading local configuration data 410147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (LocalDomainEnumRecords) 410247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 410347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem *rem = LocalDomainEnumRecords; 410447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LocalDomainEnumRecords = LocalDomainEnumRecords->next; 410547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNS_Deregister(&mDNSStorage, &rem->ar); 410647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 410747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 410847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // If the launching environment created no listening socket, 410947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // that means we created it ourselves, so we should clean it up on exit 411047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_SocketValid(listenfd)) 411147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 411247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt dnssd_close(listenfd); 411347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(USE_TCP_LOOPBACK) 411447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Currently, we're unable to remove /var/run/mdnsd because we've changed to userid "nobody" 411547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // to give up unnecessary privilege, but we need to be root to remove this Unix Domain Socket. 411647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // It would be nice if we could find a solution to this problem 411747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (unlink(MDNS_UDS_SERVERPATH)) 411847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt debugf("Unable to remove %s", MDNS_UDS_SERVERPATH); 411947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 412047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 412147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 412247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (PID_FILE[0]) unlink(PID_FILE); 412347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 412447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return 0; 412547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 412647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 412747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void LogClientInfo(mDNS *const m, const request_state *req) 412847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 412947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char prefix[16]; 413047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary) mDNS_snprintf(prefix, sizeof(prefix), " -> "); 413147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else mDNS_snprintf(prefix, sizeof(prefix), "%3d:", req->sd); 413247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 413347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 413447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 413547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!req->terminate) 413647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s No operation yet on this socket", prefix); 413747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == connection_termination) 413847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 413947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int num_records = 0, num_ops = 0; 414047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const registered_record_entry *p; 414147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const request_state *r; 414247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p = req->u.reg_recs; p; p=p->next) num_records++; 414347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (r = req->next; r; r=r->next) if (r->primary == req) num_ops++; 414447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceCreateConnection: %d registered record%s, %d kDNSServiceFlagsShareConnection operation%s", prefix, 414547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt num_records, num_records != 1 ? "s" : "", 414647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt num_ops, num_ops != 1 ? "s" : ""); 414747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (p = req->u.reg_recs; p; p=p->next) 414847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" -> DNSServiceRegisterRecord %3d %s", p->key, ARDisplayString(m, p->rr)); 414947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (r = req->next; r; r=r->next) if (r->primary == req) LogClientInfo(m, r); 415047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 415147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == regservice_termination_callback) 415247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 415347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *ptr; 415447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ptr = req->u.servicereg.instances; ptr; ptr = ptr->next) 415547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceRegister %##s %u/%u", 415647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (ptr == req->u.servicereg.instances) ? prefix : " ", 415747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ptr->srs.RR_SRV.resrec.name->c, mDNSVal16(req->u.servicereg.port), SRS_PORT(&ptr->srs)); 415847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 415947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == browse_termination_callback) 416047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 416147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *blist; 416247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (blist = req->u.browser.browsers; blist; blist = blist->next) 416347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceBrowse %##s", (blist == req->u.browser.browsers) ? prefix : " ", blist->q.qname.c); 416447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 416547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == resolve_termination_callback) 416647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceResolve %##s", prefix, req->u.resolve.qsrv.qname.c); 416747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == queryrecord_termination_callback) 416847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceQueryRecord %##s (%s)", prefix, req->u.queryrecord.q.qname.c, DNSTypeName(req->u.queryrecord.q.qtype)); 416947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == enum_termination_callback) 417047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceEnumerateDomains %##s", prefix, req->u.enumeration.q_all.qname.c); 417147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == port_mapping_termination_callback) 417247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceNATPortMapping %.4a %s%s Int %d Req %d Ext %d Req TTL %d Granted TTL %d", 417347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt prefix, 417447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt &req->u.pm.NATinfo.ExternalAddress, 417547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.pm.NATinfo.Protocol & NATOp_MapTCP ? "TCP" : " ", 417647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.pm.NATinfo.Protocol & NATOp_MapUDP ? "UDP" : " ", 417747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(req->u.pm.NATinfo.IntPort), 417847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(req->u.pm.ReqExt), 417947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(req->u.pm.NATinfo.ExternalPort), 418047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.pm.NATinfo.NATLease, 418147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.pm.NATinfo.Lifetime); 418247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == addrinfo_termination_callback) 418347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s DNSServiceGetAddrInfo %s%s %##s", prefix, 418447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.addrinfo.protocol & kDNSServiceProtocol_IPv4 ? "v4" : " ", 418547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.addrinfo.protocol & kDNSServiceProtocol_IPv6 ? "v6" : " ", 418647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req->u.addrinfo.q4.qname.c); 418747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 418847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s Unrecognized operation %p", prefix, req->terminate); 418947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 419047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 419147e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal char *RecordTypeName(mDNSu8 rtype) 419247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 419347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt switch (rtype) 419447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 419547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeUnregistered: return ("Unregistered "); 419647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeDeregistering: return ("Deregistering"); 419747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeUnique: return ("Unique "); 419847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeAdvisory: return ("Advisory "); 419947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeShared: return ("Shared "); 420047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeVerified: return ("Verified "); 420147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt case kDNSRecordTypeKnownUnique: return ("KnownUnique "); 420247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt default: return("Unknown"); 420347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 420447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 420547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 420647e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void LogEtcHosts(mDNS *const m) 420747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 420847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool showheader = mDNStrue; 420947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const AuthRecord *ar; 421047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 slot; 421147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthGroup *ag; 421247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int count = 0; 421347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int authslot = 0; 421447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool truncated = 0; 421547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 421647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (slot = 0; slot < AUTH_HASH_SLOTS; slot++) 421747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 421847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->rrauth.rrauth_hash[slot]) authslot++; 421947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next) 422047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ar = ag->members; ar; ar = ar->next) 422147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 422247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ar->RecordCallback != FreeEtcHosts) continue; 422347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); } 422447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 422547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Print a maximum of 50 records 422647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (count++ >= 50) { truncated = mDNStrue; continue; } 422747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ar->ARType == AuthRecordLocalOnly) 422847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 422947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ar->resrec.InterfaceID == mDNSInterface_LocalOnly) 423047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 423147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 423247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 423347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 scopeid = (mDNSu32)(uintptr_t)ar->resrec.InterfaceID; 423447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" %s %u %s", RecordTypeName(ar->resrec.RecordType), scopeid, ARDisplayString(m, ar)); 423547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 423647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 423747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 423847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 423947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 424047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 424147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) LogMsgNoIdent("<None>"); 424247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (truncated) LogMsgNoIdent("<Truncated: to 50 records, Total records %d, Total Auth Groups %d, Auth Slots %d>", count, m->rrauth.rrauth_totalused, authslot); 424347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 424447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 424547e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void LogLocalOnlyAuthRecords(mDNS *const m) 424647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 424747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool showheader = mDNStrue; 424847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const AuthRecord *ar; 424947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 slot; 425047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt AuthGroup *ag; 425147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 425247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (slot = 0; slot < AUTH_HASH_SLOTS; slot++) 425347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 425447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ag = m->rrauth.rrauth_hash[slot]; ag; ag = ag->next) 425547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ar = ag->members; ar; ar = ar->next) 425647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 425747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ar->RecordCallback == FreeEtcHosts) continue; 425847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" State Interface"); } 425947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 426047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Print a maximum of 400 records 426147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (ar->ARType == AuthRecordLocalOnly) 426247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" %s LO %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 426347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (ar->ARType == AuthRecordP2P) 426447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" %s PP %s", RecordTypeName(ar->resrec.RecordType), ARDisplayString(m, ar)); 426547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 426647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 426747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 426847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 426947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) LogMsgNoIdent("<None>"); 427047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 427147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 427247e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal void LogAuthRecords(mDNS *const m, const mDNSs32 now, AuthRecord *ResourceRecords, int *proxy) 427347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 427447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSBool showheader = mDNStrue; 427547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const AuthRecord *ar; 427647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt OwnerOptData owner = zeroOwner; 427747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (ar = ResourceRecords; ar; ar=ar->next) 427847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 427947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *const ifname = InterfaceNameForID(m, ar->resrec.InterfaceID); 428047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((ar->WakeUp.HMAC.l[0] != 0) == (proxy != mDNSNULL)) 428147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 428247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) { showheader = mDNSfalse; LogMsgNoIdent(" Int Next Expire State"); } 428347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (proxy) (*proxy)++; 428447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!mDNSPlatformMemSame(&owner, &ar->WakeUp, sizeof(owner))) 428547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 428647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt owner = ar->WakeUp; 428747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (owner.password.l[0]) 428847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a Password %.6a seq %d", &owner.HMAC, &owner.IMAC, &owner.password, owner.seq); 428947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (!mDNSSameEthAddress(&owner.HMAC, &owner.IMAC)) 429047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Proxying for H-MAC %.6a I-MAC %.6a seq %d", &owner.HMAC, &owner.IMAC, owner.seq); 429147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 429247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Proxying for %.6a seq %d", &owner.HMAC, owner.seq); 429347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 429447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (AuthRecord_uDNS(ar)) 429547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%7d %7d %7d %7d %s", 429647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->ThisAPInterval / mDNSPlatformOneSecond, 429747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond, 429847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->expire ? (ar->expire - now) / mDNSPlatformOneSecond : 0, 429947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->state, ARDisplayString(m, ar)); 430047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (ar->ARType == AuthRecordLocalOnly) 430147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" LO %s", ARDisplayString(m, ar)); 430247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (ar->ARType == AuthRecordP2P) 430347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" PP %s", ARDisplayString(m, ar)); 430447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 430547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%7d %7d %7d %7s %s", 430647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->ThisAPInterval / mDNSPlatformOneSecond, 430747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->AnnounceCount ? (ar->LastAPTime + ar->ThisAPInterval - now) / mDNSPlatformOneSecond : 0, 430847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ar->TimeExpire ? (ar->TimeExpire - now) / mDNSPlatformOneSecond : 0, 430947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ifname ? ifname : "ALL", 431047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARDisplayString(m, ar)); 431147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 431247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 431347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 431447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (showheader) LogMsgNoIdent("<None>"); 431547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 431647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 431747e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport void udsserver_info(mDNS *const m) 431847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 431947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const mDNSs32 now = mDNS_TimeNow(m); 432047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSu32 CacheUsed = 0, CacheActive = 0, slot; 432147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int ProxyA = 0, ProxyD = 0; 432247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const CacheGroup *cg; 432347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const CacheRecord *cr; 432447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const DNSQuestion *q; 432547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const DNameListElem *d; 432647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const SearchListElem *s; 432747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 432847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Timenow 0x%08lX (%d)", (mDNSu32)now, now); 432947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 433047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("------------ Cache -------------"); 433147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Slt Q TTL if U Type rdlen"); 433247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (slot = 0; slot < CACHE_HASH_SLOTS; slot++) 433347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (cg = m->rrcache_hash[slot]; cg; cg=cg->next) 433447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 433547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CacheUsed++; // Count one cache entity for the CacheGroup object 433647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (cr = cg->members; cr; cr=cr->next) 433747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 433847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const mDNSs32 remain = cr->resrec.rroriginalttl - (now - cr->TimeRcvd) / mDNSPlatformOneSecond; 433947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const char *ifname; 434047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSInterfaceID InterfaceID = cr->resrec.InterfaceID; 434147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!InterfaceID && cr->resrec.rDNSServer) 434247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt InterfaceID = cr->resrec.rDNSServer->interface; 434347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ifname = InterfaceNameForID(m, InterfaceID); 434447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CacheUsed++; 434547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (cr->CRActiveQuestion) CacheActive++; 434647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%3d %s%8ld %-7s%s %-6s%s", 434747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt slot, 434847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt cr->CRActiveQuestion ? "*" : " ", 434947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt remain, 435047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ifname ? ifname : "-U-", 435147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (cr->resrec.RecordType == kDNSRecordTypePacketNegative) ? "-" : 435247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (cr->resrec.RecordType & kDNSRecordTypePacketUniqueMask) ? " " : "+", 435347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNSTypeName(cr->resrec.rrtype), 435447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CRDisplayString(m, cr)); 435547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 435647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 435747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 435847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 435947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->rrcache_totalused != CacheUsed) 436047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Cache use mismatch: rrcache_totalused is %lu, true count %lu", m->rrcache_totalused, CacheUsed); 436147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->rrcache_active != CacheActive) 436247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Cache use mismatch: rrcache_active is %lu, true count %lu", m->rrcache_active, CacheActive); 436347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Cache currently contains %lu entities; %lu referenced by active questions", CacheUsed, CacheActive); 436447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 436547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--------- Auth Records ---------"); 436647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogAuthRecords(m, now, m->ResourceRecords, mDNSNULL); 436747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 436847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--------- LocalOnly, P2P Auth Records ---------"); 436947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogLocalOnlyAuthRecords(m); 437047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 437147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--------- /etc/hosts ---------"); 437247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogEtcHosts(m); 437347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 437447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("------ Duplicate Records -------"); 437547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogAuthRecords(m, now, m->DuplicateRecords, mDNSNULL); 437647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 437747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("----- Auth Records Proxied -----"); 437847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogAuthRecords(m, now, m->ResourceRecords, &ProxyA); 437947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 438047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("-- Duplicate Records Proxied ---"); 438147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogAuthRecords(m, now, m->DuplicateRecords, &ProxyD); 438247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 438347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("---------- Questions -----------"); 438447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->Questions) LogMsgNoIdent("<None>"); 438547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 438647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 438747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CacheUsed = 0; 438847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CacheActive = 0; 438947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" Int Next if T NumAns VDNS Qptr DupOf SU SQ Type Name"); 439047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (q = m->Questions; q; q=q->next) 439147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 439247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 i = q->ThisQInterval / mDNSPlatformOneSecond; 439347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 n = (NextQSendTime(q) - now) / mDNSPlatformOneSecond; 439447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *ifname = InterfaceNameForID(m, q->InterfaceID); 439547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt CacheUsed++; 439647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (q->ThisQInterval) CacheActive++; 439747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%6d%6d %-7s%s%s %5d 0x%x%x 0x%p 0x%p %1d %2d %-5s%##s%s", 439847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt i, n, 439947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ifname ? ifname : mDNSOpaque16IsZero(q->TargetQID) ? "" : "-U-", 440047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSOpaque16IsZero(q->TargetQID) ? (q->LongLived ? "l" : " ") : (q->LongLived ? "L" : "O"), 440147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt PrivateQuery(q) ? "P" : " ", 440247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->CurrentAnswers, q->validDNSServers.l[1], q->validDNSServers.l[0], q, q->DuplicateOf, 440347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->SuppressUnusable, q->SuppressQuery, DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : ""); 440447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 440547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 440647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%lu question%s; %lu active", CacheUsed, CacheUsed > 1 ? "s" : "", CacheActive); 440747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 440847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 440947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("----- Local-Only Questions -----"); 441047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->LocalOnlyQuestions) LogMsgNoIdent("<None>"); 441147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else for (q = m->LocalOnlyQuestions; q; q=q->next) 441247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" %5d %-6s%##s%s", 441347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt q->CurrentAnswers, DNSTypeName(q->qtype), q->qname.c, q->DuplicateOf ? " (dup)" : ""); 441447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 441547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("---- Active Client Requests ----"); 441647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!all_requests) LogMsgNoIdent("<None>"); 441747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 441847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 441947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const request_state *req, *r; 442047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (req = all_requests; req; req=req->next) 442147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 442247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary) // If this is a subbordinate operation, check that the parent is in the list 442347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 442447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (r = all_requests; r && r != req; r=r->next) if (r == req->primary) goto foundparent; 442547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%3d: Orhpan operation %p; parent %p not found in request list", req->sd); 442647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 442747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // For non-subbordinate operations, and subbordinate operations that have lost their parent, write out their info 442847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogClientInfo(m, req); 442947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt foundparent:; 443047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 443147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 443247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 443347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("-------- NAT Traversals --------"); 443447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->NATTraversals) LogMsgNoIdent("<None>"); 443547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 443647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 443747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const NATTraversalInfo *nat; 443847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (nat = m->NATTraversals; nat; nat=nat->next) 443947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 444047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nat->Protocol) 444147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%p %s Int %5d Ext %5d Err %d Retry %5d Interval %5d Expire %5d", 444247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nat, nat->Protocol == NATOp_MapTCP ? "TCP" : "UDP", 444347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSVal16(nat->IntPort), mDNSVal16(nat->ExternalPort), nat->Result, 444447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nat->retryPortMap ? (nat->retryPortMap - now) / mDNSPlatformOneSecond : 0, 444547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nat->retryInterval / mDNSPlatformOneSecond, 444647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nat->ExpiryTime ? (nat->ExpiryTime - now) / mDNSPlatformOneSecond : 0); 444747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 444847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%p Address Request Retry %5d Interval %5d", nat, 444947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt (m->retryGetAddr - now) / mDNSPlatformOneSecond, 445047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->retryIntervalGetAddr / mDNSPlatformOneSecond); 445147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt usleep((m->KnownBugs & mDNS_KnownBug_LossySyslog) ? 3333 : 1000); 445247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 445347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 445447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 445547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--------- AuthInfoList ---------"); 445647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->AuthInfoList) LogMsgNoIdent("<None>"); 445747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 445847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 445947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const DomainAuthInfo *a; 446047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (a = m->AuthInfoList; a; a = a->next) 446147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 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 : ""); 446247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 446347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 446447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #if APPLE_OSX_mDNSResponder 446547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--------- TunnelClients --------"); 446647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->TunnelClients) LogMsgNoIdent("<None>"); 446747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 446847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 446947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const ClientTunnel *c; 447047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (c = m->TunnelClients; c; c = c->next) 447147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%s %##s local %.16a %.4a %.16a remote %.16a %.4a %5d %.16a interval %d", 447247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 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); 447347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 447447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt #endif // APPLE_OSX_mDNSResponder 447547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 447647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("---------- Misc State ----------"); 447747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 447847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("PrimaryMAC: %.6a", &m->PrimaryMAC); 447947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 448047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->SleepState %d (%s) seq %d", 448147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->SleepState, 448247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->SleepState == SleepState_Awake ? "Awake" : 448347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->SleepState == SleepState_Transferring ? "Transferring" : 448447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->SleepState == SleepState_Sleeping ? "Sleeping" : "?", 448547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->SleepSeqNum); 448647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 448747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->SPSSocket) LogMsgNoIdent("Not offering Sleep Proxy Service"); 448847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else LogMsgNoIdent("Offering Sleep Proxy Service: %#s", m->SPSRecords.RR_SRV.resrec.name->c); 448947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 449047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (m->ProxyRecords == ProxyA + ProxyD) LogMsgNoIdent("ProxyRecords: %d + %d = %d", ProxyA, ProxyD, ProxyA + ProxyD); 449147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else LogMsgNoIdent("ProxyRecords: MISMATCH %d + %d = %d ≠ %d", ProxyA, ProxyD, ProxyA + ProxyD, m->ProxyRecords); 449247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 449347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("------ Auto Browse Domains -----"); 449447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AutoBrowseDomains) LogMsgNoIdent("<None>"); 449547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else for (d=AutoBrowseDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c); 449647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 449747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--- Auto Registration Domains --"); 449847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!AutoRegistrationDomains) LogMsgNoIdent("<None>"); 449947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else for (d=AutoRegistrationDomains; d; d=d->next) LogMsgNoIdent("%##s", d->name.c); 450047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 450147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("--- Search Domains --"); 450247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!SearchList) LogMsgNoIdent("<None>"); 450347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 450447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 450547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (s=SearchList; s; s=s->next) 450647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 450747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char *ifname = InterfaceNameForID(m, s->InterfaceID); 450847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("%##s %s", s->domain.c, ifname ? ifname : ""); 450947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 451047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 451147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 451247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("---- Task Scheduling Timers ----"); 451347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 451447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->NewQuestions) 451547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewQuestion <NONE>"); 451647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 451747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewQuestion DelayAnswering %d %d %##s (%s)", 451847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->NewQuestions->DelayAnswering, m->NewQuestions->DelayAnswering-now, 451947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->NewQuestions->qname.c, DNSTypeName(m->NewQuestions->qtype)); 452047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 452147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->NewLocalOnlyQuestions) 452247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewLocalOnlyQuestions <NONE>"); 452347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 452447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewLocalOnlyQuestions %##s (%s)", 452547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt m->NewLocalOnlyQuestions->qname.c, DNSTypeName(m->NewLocalOnlyQuestions->qtype)); 452647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 452747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!m->NewLocalRecords) 452847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewLocalRecords <NONE>"); 452947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 453047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("NewLocalRecords %02X %s", m->NewLocalRecords->resrec.RecordType, ARDisplayString(m, m->NewLocalRecords)); 453147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 453247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("SPSProxyListChanged%s", m->SPSProxyListChanged ? "" : " <NONE>"); 453347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("LocalRemoveEvents%s", m->LocalRemoveEvents ? "" : " <NONE>"); 453447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->RegisterAutoTunnel6 %08X", m->RegisterAutoTunnel6); 453547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->AutoTunnelRelayAddrIn %.16a", &m->AutoTunnelRelayAddrIn); 453647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->AutoTunnelRelayAddrOut %.16a", &m->AutoTunnelRelayAddrOut); 453747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 453847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#define LogTimer(MSG,T) LogMsgNoIdent( MSG " %08X %11d %08X %11d", (T), (T), (T)-now, (T)-now) 453947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 454047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent(" ABS (hex) ABS (dec) REL (hex) REL (dec)"); 454147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->timenow %08X %11d", now, now); 454247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("m->timenow_adjust %08X %11d", m->timenow_adjust, m->timenow_adjust); 454347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledEvent ", m->NextScheduledEvent); 454447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 454547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#ifndef UNICAST_DISABLED 454647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextuDNSEvent ", m->NextuDNSEvent); 454747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextSRVUpdate ", m->NextSRVUpdate); 454847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledNATOp ", m->NextScheduledNATOp); 454947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->retryGetAddr ", m->retryGetAddr); 455047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 455147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 455247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextCacheCheck ", m->NextCacheCheck); 455347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledSPS ", m->NextScheduledSPS); 455447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledSPRetry ", m->NextScheduledSPRetry); 455547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->DelaySleep ", m->DelaySleep); 455647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 455747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledQuery ", m->NextScheduledQuery); 455847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledProbe ", m->NextScheduledProbe); 455947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledResponse", m->NextScheduledResponse); 456047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 456147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->SuppressSending ", m->SuppressSending); 456247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->SuppressProbes ", m->SuppressProbes); 456347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->ProbeFailTime ", m->ProbeFailTime); 456447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->DelaySleep ", m->DelaySleep); 456547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->SleepLimit ", m->SleepLimit); 456647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogTimer("m->NextScheduledStopTime ", m->NextScheduledStopTime); 456747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 456847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 456947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 457047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport void uds_validatelists(void) 457147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 457247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt const request_state *req, *p; 457347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (req = all_requests; req; req=req->next) 457447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 457547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->next == (request_state *)~0 || (req->sd < 0 && req->sd != -2)) 457647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS request list: %p is garbage (%d)", req, req->sd); 457747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 457847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary == req) 457947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS request list: req->primary should not point to self %p/%d", req, req->sd); 458047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 458147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->primary && req->replies) 458247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS request list: Subordinate request %p/%d/%p should not have replies (%p)", 458347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req, req->sd, req->primary && req->replies); 458447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 458547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt p = req->primary; 458647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if ((long)p & 3) 458747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS request list: req %p primary %p is misaligned (%d)", req, p, req->sd); 458847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (p && (p->next == (request_state *)~0 || (p->sd < 0 && p->sd != -2))) 458947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS request list: req %p primary %p is garbage (%d)", req, p, p->sd); 459047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 459147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *rep; 459247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (rep = req->replies; rep; rep=rep->next) 459347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (rep->next == (reply_state *)~0) 459447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS req->replies: %p is garbage", rep); 459547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 459647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->terminate == connection_termination) 459747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 459847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt registered_record_entry *r; 459947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (r = req->u.reg_recs; r; r=r->next) 460047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (r->next == (registered_record_entry *)~0) 460147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS req->u.reg_recs: %p is garbage", r); 460247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 460347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == regservice_termination_callback) 460447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 460547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt service_instance *s; 460647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (s = req->u.servicereg.instances; s; s=s->next) 460747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (s->next == (service_instance *)~0) 460847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS req->u.servicereg.instances: %p is garbage", s); 460947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 461047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (req->terminate == browse_termination_callback) 461147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 461247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt browser_t *b; 461347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (b = req->u.browser.browsers; b; b=b->next) 461447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (b->next == (browser_t *)~0) 461547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("UDS req->u.browser.browsers: %p is garbage", b); 461647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 461747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 461847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 461947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt DNameListElem *d; 462047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (d = SCPrefBrowseDomains; d; d=d->next) 462147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 462247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("SCPrefBrowseDomains: %p is garbage (%d)", d, d->name.c[0]); 462347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 462447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ARListElem *b; 462547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (b = LocalDomainEnumRecords; b; b=b->next) 462647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (b->next == (ARListElem *)~0 || b->ar.resrec.name->c[0] > 63) 462747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("LocalDomainEnumRecords: %p is garbage (%d)", b, b->ar.resrec.name->c[0]); 462847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 462947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (d = AutoBrowseDomains; d; d=d->next) 463047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 463147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("AutoBrowseDomains: %p is garbage (%d)", d, d->name.c[0]); 463247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 463347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt for (d = AutoRegistrationDomains; d; d=d->next) 463447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (d->next == (DNameListElem *)~0 || d->name.c[0] > 63) 463547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMemCorruption("AutoRegistrationDomains: %p is garbage (%d)", d, d->name.c[0]); 463647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 463747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif // APPLE_OSX_mDNSResponder && MACOSX_MDNS_MALLOC_DEBUGGING 463847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 463947e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSlocal int send_msg(request_state *const req) 464047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 464147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *const rep = req->replies; // Send the first waiting reply 464247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ssize_t nwriten; 464347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (req->no_reply) return(t_complete); 464447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 464547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertHeaderBytes(rep->mhdr); 464647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt nwriten = send(req->sd, (char *)&rep->mhdr + rep->nwriten, rep->totallen - rep->nwriten, 0); 464747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt ConvertHeaderBytes(rep->mhdr); 464847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 464947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nwriten < 0) 465047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 465147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_errno == dnssd_EINTR || dnssd_errno == dnssd_EWOULDBLOCK) nwriten = 0; 465247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 465347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 465447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#if !defined(PLATFORM_NO_EPIPE) 465547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (dnssd_errno == EPIPE) 465647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(req->ts = t_terminated); 465747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 465847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt#endif 465947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 466047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("send_msg ERROR: failed to write %d of %d bytes to fd %d errno %d (%s)", 466147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->totallen - rep->nwriten, rep->totallen, req->sd, dnssd_errno, dnssd_strerror(dnssd_errno)); 466247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return(t_error); 466347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 466447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 466547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 466647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt rep->nwriten += nwriten; 466747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return (rep->nwriten == rep->totallen) ? t_complete : t_morecoming; 466847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 466947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 467047e4cebad7397422144bb03a21f3f7682c062c4aRobert GreenwaltmDNSexport mDNSs32 udsserver_idle(mDNSs32 nextevent) 467147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 467247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt mDNSs32 now = mDNS_TimeNow(&mDNSStorage); 467347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state **req = &all_requests; 467447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 467547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (*req) 467647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 467747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt request_state *const r = *req; 467847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 467947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (r->terminate == resolve_termination_callback) 468047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (r->u.resolve.ReportTime && now - r->u.resolve.ReportTime >= 0) 468147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 468247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt r->u.resolve.ReportTime = 0; 468347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsgNoIdent("Client application bug: DNSServiceResolve(%##s) active for over two minutes. " 468447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt "This places considerable burden on the network.", r->u.resolve.qsrv.qname.c); 468547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 468647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 468747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Note: Only primary req's have reply lists, not subordinate req's. 468847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (r->replies) // Send queued replies 468947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 469047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt transfer_state result; 469147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (r->replies->next) r->replies->rhdr->flags |= dnssd_htonl(kDNSServiceFlagsMoreComing); 469247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt result = send_msg(r); // Returns t_morecoming if buffer full because client is not reading 469347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (result == t_complete) 469447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 469547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt reply_state *fptr = r->replies; 469647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt r->replies = r->replies->next; 469747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("reply_state/udsserver_idle", fptr); 469847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt r->time_blocked = 0; // reset failure counter after successful send 469947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt r->unresponsiveness_reports = 0; 470047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt continue; 470147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 470247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (result == t_terminated || result == t_error) 470347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 470447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Could not write data to client because of error - aborting connection", r->sd); 470547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogClientInfo(&mDNSStorage, r); 470647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort_request(r); 470747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 470847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt break; 470947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 471047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 471147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (r->replies) // If we failed to send everything, check our time_blocked timer 471247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 471347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (nextevent - now > mDNSPlatformOneSecond) nextevent = now + mDNSPlatformOneSecond; 471447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 471547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (mDNSStorage.SleepState != SleepState_Awake) r->time_blocked = 0; 471647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (!r->time_blocked) r->time_blocked = NonZeroTime(now); 471747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else if (now - r->time_blocked >= 10 * mDNSPlatformOneSecond * (r->unresponsiveness_reports+1)) 471847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 471947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt int num = 0; 472047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt struct reply_state *x = r->replies; 472147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt while (x) { num++; x=x->next; } 472247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Could not write data to client after %ld seconds, %d repl%s waiting", 472347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt r->sd, (now - r->time_blocked) / mDNSPlatformOneSecond, num, num == 1 ? "y" : "ies"); 472447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (++r->unresponsiveness_reports >= 60) 472547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 472647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogMsg("%3d: Client unresponsive; aborting connection", r->sd); 472747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt LogClientInfo(&mDNSStorage, r); 472847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt abort_request(r); 472947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 473047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 473147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 473247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 473347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt if (!dnssd_SocketValid(r->sd)) // If this request is finished, unlink it from the list and free the memory 473447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 473547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Since we're already doing a list traversal, we unlink the request directly instead of using AbortUnlinkAndFree() 473647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt *req = r->next; 473747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt freeL("request_state/udsserver_idle", r); 473847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 473947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt else 474047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt req = &r->next; 474147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 474247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt return nextevent; 474347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt } 474447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt 474547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwaltstruct CompileTimeAssertionChecks_uds_daemon 474647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt { 474747e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // Check our structures are reasonable sizes. Including overly-large buffers, or embedding 474847e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // other overly-large structures instead of having a pointer to them, can inadvertently 474947e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt // cause structure sizes (and therefore memory usage) to balloon unreasonably. 475047e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_request_state [(sizeof(request_state) <= 1784) ? 1 : -1]; 475147e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_registered_record_entry[(sizeof(registered_record_entry) <= 60) ? 1 : -1]; 475247e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_service_instance [(sizeof(service_instance) <= 6552) ? 1 : -1]; 475347e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_browser_t [(sizeof(browser_t) <= 1050) ? 1 : -1]; 475447e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_reply_hdr [(sizeof(reply_hdr) <= 12) ? 1 : -1]; 475547e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt char sizecheck_reply_state [(sizeof(reply_state) <= 64) ? 1 : -1]; 475647e4cebad7397422144bb03a21f3f7682c062c4aRobert Greenwalt }; 4757