1 /* -*- Mode: C; tab-width: 4 -*-
2  *
3  * Copyright (c) 2002-2008 Apple Inc. All rights reserved.
4  *
5  * Disclaimer: IMPORTANT:  This Apple software is supplied to you by Apple Computer, Inc.
6  * ("Apple") in consideration of your agreement to the following terms, and your
7  * use, installation, modification or redistribution of this Apple software
8  * constitutes acceptance of these terms.  If you do not agree with these terms,
9  * please do not use, install, modify or redistribute this Apple software.
10  *
11  * In consideration of your agreement to abide by the following terms, and subject
12  * to these terms, Apple grants you a personal, non-exclusive license, under Apple's
13  * copyrights in this original Apple software (the "Apple Software"), to use,
14  * reproduce, modify and redistribute the Apple Software, with or without
15  * modifications, in source and/or binary forms; provided that if you redistribute
16  * the Apple Software in its entirety and without modifications, you must retain
17  * this notice and the following text and disclaimers in all such redistributions of
18  * the Apple Software.  Neither the name, trademarks, service marks or logos of
19  * Apple Computer, Inc. may be used to endorse or promote products derived from the
20  * Apple Software without specific prior written permission from Apple.  Except as
21  * expressly stated in this notice, no other rights or licenses, express or implied,
22  * are granted by Apple herein, including but not limited to any patent rights that
23  * may be infringed by your derivative works or by other works in which the Apple
24  * Software may be incorporated.
25  *
26  * The Apple Software is provided by Apple on an "AS IS" basis.  APPLE MAKES NO
27  * WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED
28  * WARRANTIES OF NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29  * PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS USE AND OPERATION ALONE OR IN
30  * COMBINATION WITH YOUR PRODUCTS.
31  *
32  * IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR
33  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
34  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
35  * ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION
36  * OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER UNDER THEORY OF CONTRACT, TORT
37  * (INCLUDING NEGLIGENCE), STRICT LIABILITY OR OTHERWISE, EVEN IF APPLE HAS BEEN
38  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
39  *
40  * Formatting notes:
41  * This code follows the "Whitesmiths style" C indentation rules. Plenty of discussion
42  * on C indentation can be found on the web, such as <http://www.kafejo.com/komp/1tbs.htm>,
43  * but for the sake of brevity here I will say just this: Curly braces are not syntactially
44  * part of an "if" statement; they are the beginning and ending markers of a compound statement;
45  * therefore common sense dictates that if they are part of a compound statement then they
46  * should be indented to the same level as everything else in that compound statement.
47  * Indenting curly braces at the same level as the "if" implies that curly braces are
48  * part of the "if", which is false. (This is as misleading as people who write "char* x,y;"
49  * thinking that variables x and y are both of type "char*" -- and anyone who doesn't
50  * understand why variable y is not of type "char*" just proves the point that poor code
51  * layout leads people to unfortunate misunderstandings about how the C language really works.)
52 
53 To build this tool, copy and paste the following into a command line:
54 
55 OS X:
56 gcc dns-sd.c -o dns-sd
57 
58 POSIX systems:
59 gcc dns-sd.c -o dns-sd -I../mDNSShared -ldns_sd
60 
61 Windows:
62 cl dns-sd.c -I../mDNSShared -DNOT_HAVE_GETOPT ws2_32.lib ..\mDNSWindows\DLL\Release\dnssd.lib
63 (may require that you run a Visual Studio script such as vsvars32.bat first)
64 */
65 
66 // For testing changes to dnssd_clientstub.c, uncomment this line and the code will be compiled
67 // with an embedded copy of the client stub instead of linking the system library version at runtime.
68 // This also useful to work around link errors when you're working on an older version of Mac OS X,
69 // and trying to build a newer version of the "dns-sd" command which uses new API entry points that
70 // aren't in the system's /usr/lib/libSystem.dylib.
71 //#define TEST_NEW_CLIENTSTUB 1
72 
73 // When building mDNSResponder for Mac OS X 10.4 and earlier, /usr/lib/libSystem.dylib is built using its own private
74 // copy of dnssd_clientstub.c, which is old and doesn't have all the entry points defined in the latest version, so
75 // when we're building dns-sd.c on Mac OS X 10.4 or earlier, we automatically set TEST_NEW_CLIENTSTUB so that we'll
76 // embed a copy of the latest dnssd_clientstub.c instead of trying to link to the incomplete version in libSystem.dylib
77 #if defined(__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__) && __ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ <= 1040
78 #define TEST_NEW_CLIENTSTUB 1
79 #endif
80 
81 #include <ctype.h>
82 #include <stdio.h>			// For stdout, stderr
83 #include <stdlib.h>			// For exit()
84 #include <string.h>			// For strlen(), strcpy()
85 #include <errno.h>			// For errno, EINTR
86 #include <time.h>
87 #include <sys/types.h>		// For u_char
88 
89 #ifdef _WIN32
90 	#include <winsock2.h>
91 	#include <ws2tcpip.h>
92 	#include <Iphlpapi.h>
93 	#include <process.h>
94 	typedef int        pid_t;
95 	#define getpid     _getpid
96 	#define strcasecmp _stricmp
97 	#define snprintf   _snprintf
98 	static const char kFilePathSep = '\\';
99 	#ifndef HeapEnableTerminationOnCorruption
100 	#     define HeapEnableTerminationOnCorruption (HEAP_INFORMATION_CLASS)1
101 	#endif
102 	#if !defined(IFNAMSIZ)
103 	 #define IFNAMSIZ 16
104     #endif
105 	#define if_nametoindex if_nametoindex_win
106 	#define if_indextoname if_indextoname_win
107 
108 	typedef PCHAR (WINAPI * if_indextoname_funcptr_t)(ULONG index, PCHAR name);
109 	typedef ULONG (WINAPI * if_nametoindex_funcptr_t)(PCSTR name);
110 
if_nametoindex_win(const char * ifname)111 	unsigned if_nametoindex_win(const char *ifname)
112 		{
113 		HMODULE library;
114 		unsigned index = 0;
115 
116 		// Try and load the IP helper library dll
117 		if ((library = LoadLibrary(TEXT("Iphlpapi")) ) != NULL )
118 			{
119 			if_nametoindex_funcptr_t if_nametoindex_funcptr;
120 
121 			// On Vista and above there is a Posix like implementation of if_nametoindex
122 			if ((if_nametoindex_funcptr = (if_nametoindex_funcptr_t) GetProcAddress(library, "if_nametoindex")) != NULL )
123 				{
124 				index = if_nametoindex_funcptr(ifname);
125 				}
126 
127 			FreeLibrary(library);
128 			}
129 
130 		return index;
131 		}
132 
if_indextoname_win(unsigned ifindex,char * ifname)133 	char * if_indextoname_win( unsigned ifindex, char *ifname)
134 		{
135 		HMODULE library;
136 		char * name = NULL;
137 
138 		// Try and load the IP helper library dll
139 		if ((library = LoadLibrary(TEXT("Iphlpapi")) ) != NULL )
140 			{
141 			if_indextoname_funcptr_t if_indextoname_funcptr;
142 
143 			// On Vista and above there is a Posix like implementation of if_indextoname
144 			if ((if_indextoname_funcptr = (if_indextoname_funcptr_t) GetProcAddress(library, "if_indextoname")) != NULL )
145 				{
146 				name = if_indextoname_funcptr(ifindex, ifname);
147 				}
148 
149 			FreeLibrary(library);
150 			}
151 
152 		return name;
153 		}
154 
_sa_len(const struct sockaddr * addr)155 	static size_t _sa_len(const struct sockaddr *addr)
156 		{
157 		if (addr->sa_family == AF_INET) return (sizeof(struct sockaddr_in));
158 		else if (addr->sa_family == AF_INET6) return (sizeof(struct sockaddr_in6));
159 		else return (sizeof(struct sockaddr));
160 		}
161 
162 #   define SA_LEN(addr) (_sa_len(addr))
163 
164 #else
165 	#include <unistd.h>			// For getopt() and optind
166 	#include <netdb.h>			// For getaddrinfo()
167 	#include <sys/time.h>		// For struct timeval
168 	#include <sys/socket.h>		// For AF_INET
169 	#include <netinet/in.h>		// For struct sockaddr_in()
170 	#include <arpa/inet.h>		// For inet_addr()
171 	#include <net/if.h>			// For if_nametoindex()
172 	static const char kFilePathSep = '/';
173 // #ifndef NOT_HAVE_SA_LEN
174 // 	#define SA_LEN(addr) ((addr)->sa_len)
175 // #else
176     #define SA_LEN(addr) (((addr)->sa_family == AF_INET6)? sizeof(struct sockaddr_in6) : sizeof(struct sockaddr_in))
177 // #endif
178 #endif
179 
180 #if (TEST_NEW_CLIENTSTUB && !defined(__APPLE_API_PRIVATE))
181 #define __APPLE_API_PRIVATE 1
182 #endif
183 
184 // DNSServiceSetDispatchQueue is not supported on 10.6 & prior
185 #if ! TEST_NEW_CLIENTSTUB && defined(__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__) && (__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ - (__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ % 10) <= 1060)
186 #undef _DNS_SD_LIBDISPATCH
187 #endif
188 #include "dns_sd.h"
189 #include "ClientCommon.h"
190 
191 #if TEST_NEW_CLIENTSTUB
192 #include "../mDNSShared/dnssd_ipc.c"
193 #include "../mDNSShared/dnssd_clientlib.c"
194 #include "../mDNSShared/dnssd_clientstub.c"
195 #endif
196 
197 // The "+0" is to cope with the case where _DNS_SD_H is defined but empty (e.g. on Mac OS X 10.4 and earlier)
198 #if _DNS_SD_H+0 >= 116
199 #define HAS_NAT_PMP_API 1
200 #define HAS_ADDRINFO_API 1
201 #else
202 #define kDNSServiceFlagsReturnIntermediates 0
203 #endif
204 
205 //*************************************************************************************************************
206 // Globals
207 
208 typedef union { unsigned char b[2]; unsigned short NotAnInteger; } Opaque16;
209 
210 static int operation;
211 static uint32_t opinterface = kDNSServiceInterfaceIndexAny;
212 static DNSServiceRef client    = NULL;
213 static DNSServiceRef client_pa = NULL;	// DNSServiceRef for RegisterProxyAddressRecord
214 static DNSServiceRef sc1, sc2, sc3;		// DNSServiceRefs for kDNSServiceFlagsShareConnection testing
215 
216 static int num_printed;
217 static char addtest = 0;
218 static DNSRecordRef record = NULL;
219 static char myhinfoW[14] = "\002PC\012Windows XP";
220 static char myhinfoX[ 9] = "\003Mac\004OS X";
221 static char updatetest[3] = "\002AA";
222 static char bigNULL[8192];	// 8K is maximum rdata we support
223 
224 #if _DNS_SD_LIBDISPATCH
225 dispatch_queue_t main_queue;
226 dispatch_source_t timer_source;
227 #endif
228 
229 // Note: the select() implementation on Windows (Winsock2) fails with any timeout much larger than this
230 #define LONG_TIME 100000000
231 
232 static volatile int stopNow = 0;
233 static volatile int timeOut = LONG_TIME;
234 
235 #if _DNS_SD_LIBDISPATCH
236 #define EXIT_IF_LIBDISPATCH_FATAL_ERROR(E) \
237 	if (main_queue && (E) == kDNSServiceErr_ServiceNotRunning) { fprintf(stderr, "Error code %d\n", (E)); exit(0); }
238 #else
239 #define EXIT_IF_LIBDISPATCH_FATAL_ERROR(E)
240 #endif
241 
242 //*************************************************************************************************************
243 // Supporting Utility Functions
244 
GetRRType(const char * s)245 static uint16_t GetRRType(const char *s)
246 	{
247 	if      (!strcasecmp(s, "A"       )) return(kDNSServiceType_A);
248 	else if (!strcasecmp(s, "NS"      )) return(kDNSServiceType_NS);
249 	else if (!strcasecmp(s, "MD"      )) return(kDNSServiceType_MD);
250 	else if (!strcasecmp(s, "MF"      )) return(kDNSServiceType_MF);
251 	else if (!strcasecmp(s, "CNAME"   )) return(kDNSServiceType_CNAME);
252 	else if (!strcasecmp(s, "SOA"     )) return(kDNSServiceType_SOA);
253 	else if (!strcasecmp(s, "MB"      )) return(kDNSServiceType_MB);
254 	else if (!strcasecmp(s, "MG"      )) return(kDNSServiceType_MG);
255 	else if (!strcasecmp(s, "MR"      )) return(kDNSServiceType_MR);
256 	else if (!strcasecmp(s, "NULL"    )) return(kDNSServiceType_NULL);
257 	else if (!strcasecmp(s, "WKS"     )) return(kDNSServiceType_WKS);
258 	else if (!strcasecmp(s, "PTR"     )) return(kDNSServiceType_PTR);
259 	else if (!strcasecmp(s, "HINFO"   )) return(kDNSServiceType_HINFO);
260 	else if (!strcasecmp(s, "MINFO"   )) return(kDNSServiceType_MINFO);
261 	else if (!strcasecmp(s, "MX"      )) return(kDNSServiceType_MX);
262 	else if (!strcasecmp(s, "TXT"     )) return(kDNSServiceType_TXT);
263 	else if (!strcasecmp(s, "RP"      )) return(kDNSServiceType_RP);
264 	else if (!strcasecmp(s, "AFSDB"   )) return(kDNSServiceType_AFSDB);
265 	else if (!strcasecmp(s, "X25"     )) return(kDNSServiceType_X25);
266 	else if (!strcasecmp(s, "ISDN"    )) return(kDNSServiceType_ISDN);
267 	else if (!strcasecmp(s, "RT"      )) return(kDNSServiceType_RT);
268 	else if (!strcasecmp(s, "NSAP"    )) return(kDNSServiceType_NSAP);
269 	else if (!strcasecmp(s, "NSAP_PTR")) return(kDNSServiceType_NSAP_PTR);
270 	else if (!strcasecmp(s, "SIG"     )) return(kDNSServiceType_SIG);
271 	else if (!strcasecmp(s, "KEY"     )) return(kDNSServiceType_KEY);
272 	else if (!strcasecmp(s, "PX"      )) return(kDNSServiceType_PX);
273 	else if (!strcasecmp(s, "GPOS"    )) return(kDNSServiceType_GPOS);
274 	else if (!strcasecmp(s, "AAAA"    )) return(kDNSServiceType_AAAA);
275 	else if (!strcasecmp(s, "LOC"     )) return(kDNSServiceType_LOC);
276 	else if (!strcasecmp(s, "NXT"     )) return(kDNSServiceType_NXT);
277 	else if (!strcasecmp(s, "EID"     )) return(kDNSServiceType_EID);
278 	else if (!strcasecmp(s, "NIMLOC"  )) return(kDNSServiceType_NIMLOC);
279 	else if (!strcasecmp(s, "SRV"     )) return(kDNSServiceType_SRV);
280 	else if (!strcasecmp(s, "ATMA"    )) return(kDNSServiceType_ATMA);
281 	else if (!strcasecmp(s, "NAPTR"   )) return(kDNSServiceType_NAPTR);
282 	else if (!strcasecmp(s, "KX"      )) return(kDNSServiceType_KX);
283 	else if (!strcasecmp(s, "CERT"    )) return(kDNSServiceType_CERT);
284 	else if (!strcasecmp(s, "A6"      )) return(kDNSServiceType_A6);
285 	else if (!strcasecmp(s, "DNAME"   )) return(kDNSServiceType_DNAME);
286 	else if (!strcasecmp(s, "SINK"    )) return(kDNSServiceType_SINK);
287 	else if (!strcasecmp(s, "OPT"     )) return(kDNSServiceType_OPT);
288 	else if (!strcasecmp(s, "TKEY"    )) return(kDNSServiceType_TKEY);
289 	else if (!strcasecmp(s, "TSIG"    )) return(kDNSServiceType_TSIG);
290 	else if (!strcasecmp(s, "IXFR"    )) return(kDNSServiceType_IXFR);
291 	else if (!strcasecmp(s, "AXFR"    )) return(kDNSServiceType_AXFR);
292 	else if (!strcasecmp(s, "MAILB"   )) return(kDNSServiceType_MAILB);
293 	else if (!strcasecmp(s, "MAILA"   )) return(kDNSServiceType_MAILA);
294 	else if (!strcasecmp(s, "ANY"     )) return(kDNSServiceType_ANY);
295 	else                                 return(atoi(s));
296 	}
297 
298 #if HAS_NAT_PMP_API | HAS_ADDRINFO_API
GetProtocol(const char * s)299 static DNSServiceProtocol GetProtocol(const char *s)
300 	{
301 	if      (!strcasecmp(s, "v4"      )) return(kDNSServiceProtocol_IPv4);
302 	else if (!strcasecmp(s, "v6"      )) return(kDNSServiceProtocol_IPv6);
303 	else if (!strcasecmp(s, "v4v6"    )) return(kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6);
304 	else if (!strcasecmp(s, "v6v4"    )) return(kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6);
305 	else if (!strcasecmp(s, "udp"     )) return(kDNSServiceProtocol_UDP);
306 	else if (!strcasecmp(s, "tcp"     )) return(kDNSServiceProtocol_TCP);
307 	else if (!strcasecmp(s, "udptcp"  )) return(kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP);
308 	else if (!strcasecmp(s, "tcpudp"  )) return(kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP);
309 	else                                 return(atoi(s));
310 	}
311 #endif
312 
313 //*************************************************************************************************************
314 // Sample callback functions for each of the operation types
315 
printtimestamp(void)316 static void printtimestamp(void)
317 	{
318 	struct tm tm;
319 	int ms;
320 #ifdef _WIN32
321 	SYSTEMTIME sysTime;
322 	time_t uct = time(NULL);
323 	tm = *localtime(&uct);
324 	GetLocalTime(&sysTime);
325 	ms = sysTime.wMilliseconds;
326 #else
327 	struct timeval tv;
328 	gettimeofday(&tv, NULL);
329 	localtime_r((time_t*)&tv.tv_sec, &tm);
330 	ms = tv.tv_usec/1000;
331 #endif
332 	printf("%2d:%02d:%02d.%03d  ", tm.tm_hour, tm.tm_min, tm.tm_sec, ms);
333 	}
334 
335 #define DomainMsg(X) (((X) & kDNSServiceFlagsDefault) ? "(Default)" : \
336                       ((X) & kDNSServiceFlagsAdd)     ? "Added"     : "Removed")
337 
338 #define MAX_LABELS 128
339 
enum_reply(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * replyDomain,void * context)340 static void DNSSD_API enum_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex,
341 	DNSServiceErrorType errorCode, const char *replyDomain, void *context)
342 	{
343 	DNSServiceFlags partialflags = flags & ~(kDNSServiceFlagsMoreComing | kDNSServiceFlagsAdd | kDNSServiceFlagsDefault);
344 	int labels = 0, depth = 0, i, initial = 0;
345 	char text[64];
346 	const char *label[MAX_LABELS];
347 
348 	(void)sdref;        // Unused
349 	(void)ifIndex;      // Unused
350 	(void)context;      // Unused
351 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
352 
353 	// 1. Print the header
354 	if (num_printed++ == 0) printf("Timestamp     Recommended %s domain\n", operation == 'E' ? "Registration" : "Browsing");
355 	printtimestamp();
356 	if (errorCode)
357 		printf("Error code %d\n", errorCode);
358 	else if (!*replyDomain)
359 		printf("Error: No reply domain\n");
360 	else
361 		{
362 		printf("%-10s", DomainMsg(flags));
363 		printf("%-8s", (flags & kDNSServiceFlagsMoreComing) ? "(More)" : "");
364 		if (partialflags) printf("Flags: %4X  ", partialflags);
365 		else printf("             ");
366 
367 		// 2. Count the labels
368 		while (replyDomain && *replyDomain && labels < MAX_LABELS)
369 			{
370 			label[labels++] = replyDomain;
371 			replyDomain = GetNextLabel(replyDomain, text);
372 			}
373 
374 		// 3. Decide if we're going to clump the last two or three labels (e.g. "apple.com", or "nicta.com.au")
375 		if      (labels >= 3 && replyDomain - label[labels-1] <= 3 && label[labels-1] - label[labels-2] <= 4) initial = 3;
376 		else if (labels >= 2 && replyDomain - label[labels-1] <= 4) initial = 2;
377 		else initial = 1;
378 		labels -= initial;
379 
380 		// 4. Print the initial one-, two- or three-label clump
381 		for (i=0; i<initial; i++)
382 			{
383 			GetNextLabel(label[labels+i], text);
384 			if (i>0) printf(".");
385 			printf("%s", text);
386 			}
387 		printf("\n");
388 
389 		// 5. Print the remainder of the hierarchy
390 		for (depth=0; depth<labels; depth++)
391 			{
392 			printf("                                             ");
393 			for (i=0; i<=depth; i++) printf("- ");
394 			GetNextLabel(label[labels-1-depth], text);
395 			printf("> %s\n", text);
396 			}
397 		}
398 
399 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
400 	}
401 
CopyLabels(char * dst,const char * lim,const char ** srcp,int labels)402 static int CopyLabels(char *dst, const char *lim, const char **srcp, int labels)
403 	{
404 	const char *src = *srcp;
405 	while (*src != '.' || --labels > 0)
406 		{
407 		if (*src == '\\') *dst++ = *src++;	// Make sure "\." doesn't confuse us
408 		if (!*src || dst >= lim) return -1;
409 		*dst++ = *src++;
410 		if (!*src || dst >= lim) return -1;
411 		}
412 	*dst++ = 0;
413 	*srcp = src + 1;	// skip over final dot
414 	return 0;
415 	}
416 
zonedata_resolve(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * fullname,const char * hosttarget,uint16_t opaqueport,uint16_t txtLen,const unsigned char * txt,void * context)417 static void DNSSD_API zonedata_resolve(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
418 	const char *fullname, const char *hosttarget, uint16_t opaqueport, uint16_t txtLen, const unsigned char *txt, void *context)
419 	{
420 	union { uint16_t s; u_char b[2]; } port = { opaqueport };
421 	uint16_t PortAsNumber = ((uint16_t)port.b[0]) << 8 | port.b[1];
422 
423 	const char *p = fullname;
424 	char n[kDNSServiceMaxDomainName];
425 	char t[kDNSServiceMaxDomainName];
426 
427 	const unsigned char *max = txt + txtLen;
428 
429 	(void)sdref;        // Unused
430 	(void)ifIndex;      // Unused
431 	(void)context;      // Unused
432 
433 	//if (!(flags & kDNSServiceFlagsAdd)) return;
434 	if (errorCode) { printf("Error code %d\n", errorCode); return; }
435 
436 	if (CopyLabels(n, n + kDNSServiceMaxDomainName, &p, 3)) return;		// Fetch name+type
437 	p = fullname;
438 	if (CopyLabels(t, t + kDNSServiceMaxDomainName, &p, 1)) return;		// Skip first label
439 	if (CopyLabels(t, t + kDNSServiceMaxDomainName, &p, 2)) return;		// Fetch next two labels (service type)
440 
441 	if (num_printed++ == 0)
442 		{
443 		printf("\n");
444 		printf("; To direct clients to browse a different domain, substitute that domain in place of '@'\n");
445 		printf("%-47s PTR     %s\n", "lb._dns-sd._udp", "@");
446 		printf("\n");
447 		printf("; In the list of services below, the SRV records will typically reference dot-local Multicast DNS names.\n");
448 		printf("; When transferring this zone file data to your unicast DNS server, you'll need to replace those dot-local\n");
449 		printf("; names with the correct fully-qualified (unicast) domain name of the target host offering the service.\n");
450 		}
451 
452 	printf("\n");
453 	printf("%-47s PTR     %s\n", t, n);
454 	printf("%-47s SRV     0 0 %d %s ; Replace with unicast FQDN of target host\n", n, PortAsNumber, hosttarget);
455 	printf("%-47s TXT    ", n);
456 
457 	while (txt < max)
458 		{
459 		const unsigned char *const end = txt + 1 + txt[0];
460 		txt++;		// Skip over length byte
461 		printf(" \"");
462 		while (txt<end)
463 			{
464 			if (*txt == '\\' || *txt == '\"') printf("\\");
465 			printf("%c", *txt++);
466 			}
467 		printf("\"");
468 		}
469 	printf("\n");
470 
471 	DNSServiceRefDeallocate(sdref);
472 	free(context);
473 
474 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
475 	}
476 
zonedata_browse(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * replyName,const char * replyType,const char * replyDomain,void * context)477 static void DNSSD_API zonedata_browse(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
478 	const char *replyName, const char *replyType, const char *replyDomain, void *context)
479 	{
480 	DNSServiceRef *newref;
481 
482 	(void)sdref;        // Unused
483 	(void)context;      // Unused
484 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
485 
486 	if (!(flags & kDNSServiceFlagsAdd)) return;
487 	if (errorCode) { printf("Error code %d\n", errorCode); return; }
488 
489 	newref = malloc(sizeof(*newref));
490 	*newref = client;
491 	DNSServiceResolve(newref, kDNSServiceFlagsShareConnection, ifIndex, replyName, replyType, replyDomain, zonedata_resolve, newref);
492 	}
493 
browse_reply(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * replyName,const char * replyType,const char * replyDomain,void * context)494 static void DNSSD_API browse_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
495 	const char *replyName, const char *replyType, const char *replyDomain, void *context)
496 	{
497 	char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
498 	(void)sdref;        // Unused
499 	(void)context;      // Unused
500 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
501 
502 	if (num_printed++ == 0) printf("Timestamp     A/R Flags if %-25s %-25s %s\n", "Domain", "Service Type", "Instance Name");
503 	printtimestamp();
504 	if (errorCode) printf("Error code %d\n", errorCode);
505 	else printf("%s%6X%3d %-25s %-25s %s\n", op, flags, ifIndex, replyDomain, replyType, replyName);
506 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
507 
508 	// To test selective cancellation of operations of shared sockets,
509 	// cancel the current operation when we've got a multiple of five results
510 	//if (operation == 'S' && num_printed % 5 == 0) DNSServiceRefDeallocate(sdref);
511 	}
512 
ShowTXTRecord(uint16_t txtLen,const unsigned char * txtRecord)513 static void ShowTXTRecord(uint16_t txtLen, const unsigned char *txtRecord)
514 	{
515 	const unsigned char *ptr = txtRecord;
516 	const unsigned char *max = txtRecord + txtLen;
517 	while (ptr < max)
518 		{
519 		const unsigned char *const end = ptr + 1 + ptr[0];
520 		if (end > max) { printf("<< invalid data >>"); break; }
521 		if (++ptr < end) printf(" ");   // As long as string is non-empty, begin with a space
522 		while (ptr<end)
523 			{
524 			// We'd like the output to be shell-friendly, so that it can be copied and pasted unchanged into a "dns-sd -R" command.
525 			// However, this is trickier than it seems. Enclosing a string in double quotes doesn't necessarily make it
526 			// shell-safe, because shells still expand variables like $foo even when they appear inside quoted strings.
527 			// Enclosing a string in single quotes is better, but when using single quotes even backslash escapes are ignored,
528 			// meaning there's simply no way to represent a single quote (or apostrophe) inside a single-quoted string.
529 			// The only remaining solution is not to surround the string with quotes at all, but instead to use backslash
530 			// escapes to encode spaces and all other known shell metacharacters.
531 			// (If we've missed any known shell metacharacters, please let us know.)
532 			// In addition, non-printing ascii codes (0-31) are displayed as \xHH, using a two-digit hex value.
533 			// Because '\' is itself a shell metacharacter (the shell escape character), it has to be escaped as "\\" to survive
534 			// the round-trip to the shell and back. This means that a single '\' is represented here as EIGHT backslashes:
535 			// The C compiler eats half of them, resulting in four appearing in the output.
536 			// The shell parses those four as a pair of "\\" sequences, passing two backslashes to the "dns-sd -R" command.
537 			// The "dns-sd -R" command interprets this single "\\" pair as an escaped literal backslash. Sigh.
538 			if (strchr(" &;`'\"|*?~<>^()[]{}$", *ptr)) printf("\\");
539 			if      (*ptr == '\\') printf("\\\\\\\\");
540 			else if (*ptr >= ' ' ) printf("%c",        *ptr);
541 			else                   printf("\\\\x%02X", *ptr);
542 			ptr++;
543 			}
544 		}
545 	}
546 
resolve_reply(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * fullname,const char * hosttarget,uint16_t opaqueport,uint16_t txtLen,const unsigned char * txtRecord,void * context)547 static void DNSSD_API resolve_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
548 	const char *fullname, const char *hosttarget, uint16_t opaqueport, uint16_t txtLen, const unsigned char *txtRecord, void *context)
549 	{
550 	union { uint16_t s; u_char b[2]; } port = { opaqueport };
551 	uint16_t PortAsNumber = ((uint16_t)port.b[0]) << 8 | port.b[1];
552 
553 	(void)sdref;        // Unused
554 	(void)ifIndex;      // Unused
555 	(void)context;      // Unused
556 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
557 
558 	if (errorCode)
559 		printf("Error code %d\n", errorCode);
560 	else
561 		{
562 		printtimestamp();
563 		printf("%s can be reached at %s:%u (interface %d)", fullname, hosttarget, PortAsNumber, ifIndex);
564 		if (flags) printf(" Flags: %X", flags);
565 		// Don't show degenerate TXT records containing nothing but a single empty string
566 		if (txtLen > 1) { printf("\n"); ShowTXTRecord(txtLen, txtRecord); }
567 		printf("\n");
568 		}
569 
570 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
571 	}
572 
myTimerCallBack(void)573 static void myTimerCallBack(void)
574 	{
575 	DNSServiceErrorType err = kDNSServiceErr_Unknown;
576 
577 	switch (operation)
578 		{
579 		case 'A':
580 			{
581 			switch (addtest)
582 				{
583 				case 0: printf("Adding Test HINFO record\n");
584 						err = DNSServiceAddRecord(client, &record, 0, kDNSServiceType_HINFO, sizeof(myhinfoW), &myhinfoW[0], 0);
585 						addtest = 1;
586 						break;
587 				case 1: printf("Updating Test HINFO record\n");
588 						err = DNSServiceUpdateRecord(client, record, 0, sizeof(myhinfoX), &myhinfoX[0], 0);
589 						addtest = 2;
590 						break;
591 				case 2: printf("Removing Test HINFO record\n");
592 						err = DNSServiceRemoveRecord(client, record, 0);
593 						addtest = 0;
594 						break;
595 				}
596 			}
597 			break;
598 
599 		case 'U':
600 			{
601 			if (updatetest[1] != 'Z') updatetest[1]++;
602 			else                      updatetest[1] = 'A';
603 			updatetest[0] = 3 - updatetest[0];
604 			updatetest[2] = updatetest[1];
605 			printtimestamp();
606 			printf("Updating Test TXT record to %c\n", updatetest[1]);
607 			err = DNSServiceUpdateRecord(client, NULL, 0, 1+updatetest[0], &updatetest[0], 0);
608 			}
609 			break;
610 
611 		case 'N':
612 			{
613 			printf("Adding big NULL record\n");
614 			err = DNSServiceAddRecord(client, &record, 0, kDNSServiceType_NULL, sizeof(bigNULL), &bigNULL[0], 0);
615 			if (err) printf("Failed: %d\n", err); else printf("Succeeded\n");
616 			timeOut = LONG_TIME;
617 #if _DNS_SD_LIBDISPATCH
618 			if (timer_source)
619 				dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
620 					(uint64_t)timeOut * NSEC_PER_SEC, 0);
621 #endif
622 			}
623 			break;
624 		}
625 
626 	if (err != kDNSServiceErr_NoError)
627 		{
628 		fprintf(stderr, "DNSService add/update/remove failed %ld\n", (long int)err);
629 		stopNow = 1;
630 		}
631 	}
632 
reg_reply(DNSServiceRef sdref,const DNSServiceFlags flags,DNSServiceErrorType errorCode,const char * name,const char * regtype,const char * domain,void * context)633 static void DNSSD_API reg_reply(DNSServiceRef sdref, const DNSServiceFlags flags, DNSServiceErrorType errorCode,
634 	const char *name, const char *regtype, const char *domain, void *context)
635 	{
636 	(void)sdref;    // Unused
637 	(void)flags;    // Unused
638 	(void)context;  // Unused
639 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
640 
641 	printtimestamp();
642 	printf("Got a reply for service %s.%s%s: ", name, regtype, domain);
643 
644 	if (errorCode == kDNSServiceErr_NoError)
645 		{
646 		if (flags & kDNSServiceFlagsAdd) printf("Name now registered and active\n");
647 		else printf("Name registration removed\n");
648 		if (operation == 'A' || operation == 'U' || operation == 'N')
649 			{
650 			timeOut = 5;
651 #if _DNS_SD_LIBDISPATCH
652 			if (timer_source)
653 				dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
654 					(uint64_t)timeOut * NSEC_PER_SEC, 0);
655 #endif
656 			}
657 		}
658 	else if (errorCode == kDNSServiceErr_NameConflict)
659 		{
660 		printf("Name in use, please choose another\n");
661 		exit(-1);
662 		}
663 	else
664 		printf("Error %d\n", errorCode);
665 
666 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
667 	}
668 
669 // Output the wire-format domainname pointed to by rd
snprintd(char * p,int max,const unsigned char ** rd)670 static int snprintd(char *p, int max, const unsigned char **rd)
671 	{
672 	const char *const buf = p;
673 	const char *const end = p + max;
674 	while (**rd) { p += snprintf(p, end-p, "%.*s.", **rd, *rd+1); *rd += 1 + **rd; }
675 	*rd += 1;	// Advance over the final zero byte
676 	return(p-buf);
677 	}
678 
qr_reply(DNSServiceRef sdref,const DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,const char * fullname,uint16_t rrtype,uint16_t rrclass,uint16_t rdlen,const void * rdata,uint32_t ttl,void * context)679 static void DNSSD_API qr_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
680 	const char *fullname, uint16_t rrtype, uint16_t rrclass, uint16_t rdlen, const void *rdata, uint32_t ttl, void *context)
681 	{
682 	char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
683 	const unsigned char *rd  = rdata;
684 	const unsigned char *end = (const unsigned char *) rdata + rdlen;
685 	char rdb[1000] = "", *p = rdb;
686 	int unknowntype = 0;
687 
688 	(void)sdref;    // Unused
689 	(void)flags;    // Unused
690 	(void)ifIndex;  // Unused
691 	(void)ttl;      // Unused
692 	(void)context;  // Unused
693 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
694 
695 	if (num_printed++ == 0) printf("Timestamp     A/R Flags if %-30s%4s%4s Rdata\n", "Name", "T", "C");
696 	printtimestamp();
697 
698 	if (!errorCode)
699 		{
700 		switch (rrtype)
701 			{
702 			case kDNSServiceType_A:
703 				snprintf(rdb, sizeof(rdb), "%d.%d.%d.%d", rd[0], rd[1], rd[2], rd[3]);
704 				break;
705 
706 			case kDNSServiceType_NS:
707 			case kDNSServiceType_CNAME:
708 			case kDNSServiceType_PTR:
709 			case kDNSServiceType_DNAME:
710 				p += snprintd(p, sizeof(rdb), &rd);
711 				break;
712 
713 			case kDNSServiceType_SOA:
714 				p += snprintd(p, rdb + sizeof(rdb) - p, &rd);		// mname
715 				p += snprintf(p, rdb + sizeof(rdb) - p, " ");
716 				p += snprintd(p, rdb + sizeof(rdb) - p, &rd);		// rname
717 				p += snprintf(p, rdb + sizeof(rdb) - p, " Ser %d Ref %d Ret %d Exp %d Min %d",
718 					ntohl(((uint32_t*)rd)[0]), ntohl(((uint32_t*)rd)[1]), ntohl(((uint32_t*)rd)[2]), ntohl(((uint32_t*)rd)[3]), ntohl(((uint32_t*)rd)[4]));
719 				break;
720 
721 			case kDNSServiceType_AAAA:
722 				snprintf(rdb, sizeof(rdb), "%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X",
723 					rd[0x0], rd[0x1], rd[0x2], rd[0x3], rd[0x4], rd[0x5], rd[0x6], rd[0x7],
724 					rd[0x8], rd[0x9], rd[0xA], rd[0xB], rd[0xC], rd[0xD], rd[0xE], rd[0xF]);
725 				break;
726 
727 			case kDNSServiceType_SRV:
728 				p += snprintf(p, rdb + sizeof(rdb) - p, "%d %d %d ",	// priority, weight, port
729 					ntohs(*(unsigned short*)rd), ntohs(*(unsigned short*)(rd+2)), ntohs(*(unsigned short*)(rd+4)));
730 				rd += 6;
731 				p += snprintd(p, rdb + sizeof(rdb) - p, &rd);			// target host
732 				break;
733 
734 			default : snprintf(rdb, sizeof(rdb), "%d bytes%s", rdlen, rdlen ? ":" : ""); unknowntype = 1; break;
735 			}
736 		}
737 
738 	printf("%s%6X%3d %-30s%4d%4d %s", op, flags, ifIndex, fullname, rrtype, rrclass, rdb);
739 	if (unknowntype) while (rd < end) printf(" %02X", *rd++);
740 	if (errorCode)
741 		{
742 		if (errorCode == kDNSServiceErr_NoSuchRecord) printf("No Such Record");
743 		else if (errorCode == kDNSServiceErr_Timeout)
744 			{
745 			printf("No Such Record\n");
746 			printf("Query Timed Out\n");
747 			exit(1);
748 			}
749 		}
750 	printf("\n");
751 
752 	if (operation == 'C')
753 		if (flags & kDNSServiceFlagsAdd)
754 			DNSServiceReconfirmRecord(flags, ifIndex, fullname, rrtype, rrclass, rdlen, rdata);
755 
756 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
757 	}
758 
759 #if HAS_NAT_PMP_API
port_mapping_create_reply(DNSServiceRef sdref,DNSServiceFlags flags,uint32_t ifIndex,DNSServiceErrorType errorCode,uint32_t publicAddress,uint32_t protocol,uint16_t privatePort,uint16_t publicPort,uint32_t ttl,void * context)760 static void DNSSD_API port_mapping_create_reply(DNSServiceRef sdref, DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode, uint32_t publicAddress, uint32_t protocol, uint16_t privatePort, uint16_t publicPort, uint32_t ttl, void *context)
761 	{
762 	(void)sdref;       // Unused
763 	(void)flags;       // Unused
764 	(void)context;     // Unused
765 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
766 
767 	if (num_printed++ == 0) printf("Timestamp     if   %-20s %-15s %-15s %-15s %-6s\n", "External Address", "Protocol", "Internal Port", "External Port", "TTL");
768 	printtimestamp();
769 	if (errorCode && errorCode != kDNSServiceErr_DoubleNAT) printf("Error code %d\n", errorCode);
770 	else
771 		{
772 		const unsigned char *digits = (const unsigned char *)&publicAddress;
773 		char                 addr[256];
774 
775 		snprintf(addr, sizeof(addr), "%d.%d.%d.%d", digits[0], digits[1], digits[2], digits[3]);
776 		printf("%-4d %-20s %-15d %-15d %-15d %-6d%s\n", ifIndex, addr, protocol, ntohs(privatePort), ntohs(publicPort), ttl, errorCode == kDNSServiceErr_DoubleNAT ? " Double NAT" : "");
777 		}
778 
779 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
780 	}
781 #endif
782 
783 #if HAS_ADDRINFO_API
addrinfo_reply(DNSServiceRef sdref,DNSServiceFlags flags,uint32_t interfaceIndex,DNSServiceErrorType errorCode,const char * hostname,const struct sockaddr * address,uint32_t ttl,void * context)784 static void DNSSD_API addrinfo_reply(DNSServiceRef sdref, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *address, uint32_t ttl, void *context)
785 	{
786 	char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
787 	char addr[256] = "";
788 	(void) sdref;
789 	(void) context;
790 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
791 
792 	if (num_printed++ == 0) printf("Timestamp     A/R Flags if %-25s %-44s %s\n", "Hostname", "Address", "TTL");
793 	printtimestamp();
794 
795 	if (address && address->sa_family == AF_INET)
796 		{
797 		const unsigned char *b = (const unsigned char *) &((struct sockaddr_in *)address)->sin_addr;
798 		snprintf(addr, sizeof(addr), "%d.%d.%d.%d", b[0], b[1], b[2], b[3]);
799 		}
800 	else if (address && address->sa_family == AF_INET6)
801 		{
802 		char if_name[IFNAMSIZ];		// Older Linux distributions don't define IF_NAMESIZE
803 		const struct sockaddr_in6 *s6 = (const struct sockaddr_in6 *)address;
804 		const unsigned char       *b  = (const unsigned char *      )&s6->sin6_addr;
805 		if (!if_indextoname(s6->sin6_scope_id, if_name))
806 			snprintf(if_name, sizeof(if_name), "<%d>", s6->sin6_scope_id);
807 		snprintf(addr, sizeof(addr), "%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X%%%s",
808 				b[0x0], b[0x1], b[0x2], b[0x3], b[0x4], b[0x5], b[0x6], b[0x7],
809 				b[0x8], b[0x9], b[0xA], b[0xB], b[0xC], b[0xD], b[0xE], b[0xF], if_name);
810 		}
811 
812 	printf("%s%6X%3d %-25s %-44s %d", op, flags, interfaceIndex, hostname, addr, ttl);
813 	if (errorCode)
814 		{
815 		if (errorCode == kDNSServiceErr_NoSuchRecord) printf("   No Such Record");
816 		else                                          printf("   Error code %d", errorCode);
817 		}
818 	printf("\n");
819 
820 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
821 	}
822 #endif
823 
824 //*************************************************************************************************************
825 // The main test function
826 
HandleEvents(void)827 static void HandleEvents(void)
828 #if _DNS_SD_LIBDISPATCH
829 	{
830 	main_queue = dispatch_get_main_queue();
831 	if (client)  DNSServiceSetDispatchQueue(client, main_queue);
832 	if (client_pa)  DNSServiceSetDispatchQueue(client_pa, main_queue);
833 	if (operation == 'A' || operation == 'U' || operation == 'N')
834 		{
835 		timer_source = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, main_queue);
836 		if (timer_source)
837 			{
838 			// Start the timer "timeout" seconds into the future and repeat it every "timeout" seconds
839 			dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
840 				(uint64_t)timeOut * NSEC_PER_SEC, 0);
841 			dispatch_source_set_event_handler(timer_source, ^{myTimerCallBack();});
842 			dispatch_resume(timer_source);
843 			}
844 		}
845 	dispatch_main();
846 	}
847 #else
848 	{
849 	int dns_sd_fd  = client    ? DNSServiceRefSockFD(client   ) : -1;
850 	int dns_sd_fd2 = client_pa ? DNSServiceRefSockFD(client_pa) : -1;
851 	int nfds = dns_sd_fd + 1;
852 	fd_set readfds;
853 	struct timeval tv;
854 	int result;
855 
856 	if (dns_sd_fd2 > dns_sd_fd) nfds = dns_sd_fd2 + 1;
857 
858 	while (!stopNow)
859 		{
860 		// 1. Set up the fd_set as usual here.
861 		// This example client has no file descriptors of its own,
862 		// but a real application would call FD_SET to add them to the set here
863 		FD_ZERO(&readfds);
864 
865 		// 2. Add the fd for our client(s) to the fd_set
866 		if (client   ) FD_SET(dns_sd_fd , &readfds);
867 		if (client_pa) FD_SET(dns_sd_fd2, &readfds);
868 
869 		// 3. Set up the timeout.
870 		tv.tv_sec  = timeOut;
871 		tv.tv_usec = 0;
872 
873 		result = select(nfds, &readfds, (fd_set*)NULL, (fd_set*)NULL, &tv);
874 		if (result > 0)
875 			{
876 			DNSServiceErrorType err = kDNSServiceErr_NoError;
877 			if      (client    && FD_ISSET(dns_sd_fd , &readfds)) err = DNSServiceProcessResult(client   );
878 			else if (client_pa && FD_ISSET(dns_sd_fd2, &readfds)) err = DNSServiceProcessResult(client_pa);
879 			if (err) { fprintf(stderr, "DNSServiceProcessResult returned %d\n", err); stopNow = 1; }
880 			}
881 		else if (result == 0)
882 			myTimerCallBack();
883 		else
884 			{
885 			printf("select() returned %d errno %d %s\n", result, errno, strerror(errno));
886 			if (errno != EINTR) stopNow = 1;
887 			}
888 		}
889 	}
890 #endif
891 
getfirstoption(int argc,char ** argv,const char * optstr,int * pOptInd)892 static int getfirstoption(int argc, char **argv, const char *optstr, int *pOptInd)
893 // Return the recognized option in optstr and the option index of the next arg.
894 #if NOT_HAVE_GETOPT
895 	{
896 	int i;
897 	for (i=1; i < argc; i++)
898 		{
899 		if (argv[i][0] == '-' && &argv[i][1] &&
900 			 NULL != strchr(optstr, argv[i][1]))
901 			{
902 			*pOptInd = i + 1;
903 			return argv[i][1];
904 			}
905 		}
906 	return -1;
907 	}
908 #else
909 	{
910 	int o = getopt(argc, (char *const *)argv, optstr);
911 	*pOptInd = optind;
912 	return o;
913 	}
914 #endif
915 
MyRegisterRecordCallback(DNSServiceRef service,DNSRecordRef rec,const DNSServiceFlags flags,DNSServiceErrorType errorCode,void * context)916 static void DNSSD_API MyRegisterRecordCallback(DNSServiceRef service, DNSRecordRef rec, const DNSServiceFlags flags,
917     DNSServiceErrorType errorCode, void *context)
918 	{
919 	char *name = (char *)context;
920 
921 	(void)service;	// Unused
922 	(void)rec;		// Unused
923 	(void)flags;	// Unused
924 	EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
925 
926 	printtimestamp();
927 	printf("Got a reply for record %s: ", name);
928 
929 	switch (errorCode)
930 		{
931 		case kDNSServiceErr_NoError:      printf("Name now registered and active\n"); break;
932 		case kDNSServiceErr_NameConflict: printf("Name in use, please choose another\n"); exit(-1);
933 		default:                          printf("Error %d\n", errorCode); break;
934 		}
935 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
936 	// DNSServiceRemoveRecord(service, rec, 0); to test record removal
937 
938 #if 0	// To test updating of individual records registered via DNSServiceRegisterRecord
939 	if (!errorCode)
940 		{
941 		int x = 0x11111111;
942 		printf("Updating\n");
943 		DNSServiceUpdateRecord(service, rec, 0, sizeof(x), &x, 0);
944 		}
945 #endif
946 
947 	if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
948 	}
949 
getip(const char * const name,struct sockaddr_storage * result)950 static void getip(const char *const name, struct sockaddr_storage *result)
951 	{
952 	struct addrinfo *addrs = NULL;
953 	int err = getaddrinfo(name, NULL, NULL, &addrs);
954 	if (err) fprintf(stderr, "getaddrinfo error %d for %s", err, name);
955 	else memcpy(result, addrs->ai_addr, SA_LEN(addrs->ai_addr));
956 	if (addrs) freeaddrinfo(addrs);
957 	}
958 
RegisterProxyAddressRecord(DNSServiceRef sdref,const char * host,const char * ip,DNSServiceFlags flags)959 static DNSServiceErrorType RegisterProxyAddressRecord(DNSServiceRef sdref, const char *host, const char *ip, DNSServiceFlags flags)
960 	{
961 	// Call getip() after the call DNSServiceCreateConnection().
962 	// On the Win32 platform, WinSock must be initialized for getip() to succeed.
963 	// Any DNSService* call will initialize WinSock for us, so we make sure
964 	// DNSServiceCreateConnection() is called before getip() is.
965 	struct sockaddr_storage hostaddr;
966 	getip(ip, &hostaddr);
967 	flags |= kDNSServiceFlagsUnique;
968 	if (hostaddr.ss_family == AF_INET)
969 		return(DNSServiceRegisterRecord(sdref, &record, flags, opinterface, host,
970 			kDNSServiceType_A,    kDNSServiceClass_IN,  4, &((struct sockaddr_in *)&hostaddr)->sin_addr,  240, MyRegisterRecordCallback, (void*)host));
971 	else if (hostaddr.ss_family == AF_INET6)
972 		return(DNSServiceRegisterRecord(sdref, &record, flags, opinterface, host,
973 			kDNSServiceType_AAAA, kDNSServiceClass_IN, 16, &((struct sockaddr_in6*)&hostaddr)->sin6_addr, 240, MyRegisterRecordCallback, (void*)host));
974 	else return(kDNSServiceErr_BadParam);
975 	}
976 
977 #define HexVal(X) ( ((X) >= '0' && (X) <= '9') ? ((X) - '0'     ) :  \
978 					((X) >= 'A' && (X) <= 'F') ? ((X) - 'A' + 10) :  \
979 					((X) >= 'a' && (X) <= 'f') ? ((X) - 'a' + 10) : 0)
980 
981 #define HexPair(P) ((HexVal((P)[0]) << 4) | HexVal((P)[1]))
982 
RegisterService(DNSServiceRef * sdref,const char * nam,const char * typ,const char * dom,const char * host,const char * port,int argc,char ** argv,DNSServiceFlags flags)983 static DNSServiceErrorType RegisterService(DNSServiceRef *sdref,
984 	const char *nam, const char *typ, const char *dom, const char *host, const char *port, int argc, char **argv, DNSServiceFlags flags)
985 	{
986 	uint16_t PortAsNumber = atoi(port);
987 	Opaque16 registerPort = { { PortAsNumber >> 8, PortAsNumber & 0xFF } };
988 	unsigned char txt[2048] = "";
989 	unsigned char *ptr = txt;
990 	int i;
991 
992 	if (nam[0] == '.' && nam[1] == 0) nam = "";   // We allow '.' on the command line as a synonym for empty string
993 	if (dom[0] == '.' && dom[1] == 0) dom = "";   // We allow '.' on the command line as a synonym for empty string
994 
995 	printf("Registering Service %s.%s%s%s", nam[0] ? nam : "<<Default>>", typ, dom[0] ? "." : "", dom);
996 	if (host && *host) printf(" host %s", host);
997 	printf(" port %s", port);
998 
999 	if (argc)
1000 		{
1001 		for (i = 0; i < argc; i++)
1002 			{
1003 			const char *p = argv[i];
1004 			*ptr = 0;
1005 			while (*p && *ptr < 255 && ptr + 1 + *ptr < txt+sizeof(txt))
1006 				{
1007 				if      (p[0] != '\\' || p[1] == 0)                       { ptr[++*ptr] = *p;           p+=1; }
1008 				else if (p[1] == 'x' && isxdigit(p[2]) && isxdigit(p[3])) { ptr[++*ptr] = HexPair(p+2); p+=4; }
1009 				else                                                      { ptr[++*ptr] = p[1];         p+=2; }
1010 				}
1011 			ptr += 1 + *ptr;
1012 			}
1013 		printf(" TXT");
1014 		ShowTXTRecord(ptr-txt, txt);
1015 		}
1016 	printf("\n");
1017 
1018 	//flags |= kDNSServiceFlagsAllowRemoteQuery;
1019 	//flags |= kDNSServiceFlagsNoAutoRename;
1020 
1021 	return(DNSServiceRegister(sdref, flags, opinterface, nam, typ, dom, host, registerPort.NotAnInteger, (uint16_t) (ptr-txt), txt, reg_reply, NULL));
1022 	}
1023 
1024 #define TypeBufferSize 80
gettype(char * buffer,char * typ)1025 static char *gettype(char *buffer, char *typ)
1026 	{
1027 	if (!typ || !*typ || (typ[0] == '.' && typ[1] == 0)) typ = "_http._tcp";
1028 	if (!strchr(typ, '.')) { snprintf(buffer, TypeBufferSize, "%s._tcp", typ); typ = buffer; }
1029 	return(typ);
1030 	}
1031 
main(int argc,char ** argv)1032 int main(int argc, char **argv)
1033 	{
1034 	DNSServiceErrorType err;
1035 	char buffer[TypeBufferSize], *typ, *dom;
1036 	int opi;
1037 	DNSServiceFlags flags = 0;
1038 
1039 	// Extract the program name from argv[0], which by convention contains the path to this executable.
1040 	// Note that this is just a voluntary convention, not enforced by the kernel --
1041 	// the process calling exec() can pass bogus data in argv[0] if it chooses to.
1042 	const char *a0 = strrchr(argv[0], kFilePathSep) + 1;
1043 	if (a0 == (const char *)1) a0 = argv[0];
1044 
1045 #if defined(_WIN32)
1046 	HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
1047 #endif
1048 
1049 #if TEST_NEW_CLIENTSTUB
1050 	printf("Using embedded copy of dnssd_clientstub instead of system library\n");
1051 	if (sizeof(argv) == 8) printf("Running in 64-bit mode\n");
1052 #endif
1053 
1054 	// Test code for TXTRecord functions
1055 	//TXTRecordRef txtRecord;
1056 	//TXTRecordCreate(&txtRecord, 0, NULL);
1057 	//TXTRecordSetValue(&txtRecord, "aaa", 1, "b");
1058 	//printf("%d\n", TXTRecordContainsKey(TXTRecordGetLength(&txtRecord), TXTRecordGetBytesPtr(&txtRecord), "Aaa"));
1059 
1060 	if (argc > 1 && !strcmp(argv[1], "-lo"))
1061 		{
1062 		argc--;
1063 		argv++;
1064 		opinterface = kDNSServiceInterfaceIndexLocalOnly;
1065 		printf("Using LocalOnly\n");
1066 		}
1067 
1068 	if (argc > 1 && (!strcmp(argv[1], "-p2p") || !strcmp(argv[1], "-P2P")))
1069 		{
1070 		argc--;
1071 		argv++;
1072 		opinterface = kDNSServiceInterfaceIndexP2P;
1073 		printf("Using P2P\n");
1074 		}
1075 
1076 	if (argc > 1 && !strcasecmp(argv[1], "-includep2p"))
1077 		{
1078 		argc--;
1079 		argv++;
1080 		flags |= kDNSServiceFlagsIncludeP2P;
1081 		printf("Including P2P\n");
1082 		}
1083 
1084 	if (argc > 2 && !strcmp(argv[1], "-i"))
1085 		{
1086 		opinterface = if_nametoindex(argv[2]);
1087 		if (!opinterface) opinterface = atoi(argv[2]);
1088 		if (!opinterface) { fprintf(stderr, "Unknown interface %s\n", argv[2]); goto Fail; }
1089 		argc -= 2;
1090 		argv += 2;
1091 		}
1092 
1093 	if (argc < 2) goto Fail;        // Minimum command line is the command name and one argument
1094 	operation = getfirstoption(argc, argv, "EFBZLlRPQqtCAUNTMISV"
1095 								#if HAS_NAT_PMP_API
1096 									"X"
1097 								#endif
1098 								#if HAS_ADDRINFO_API
1099 									"G"
1100 								#endif
1101 								, &opi);
1102 	if (operation == -1) goto Fail;
1103 
1104 	if (opinterface) printf("Using interface %d\n", opinterface);
1105 
1106 	switch (operation)
1107 		{
1108 		case 'E':	printf("Looking for recommended registration domains:\n");
1109 					err = DNSServiceEnumerateDomains(&client, kDNSServiceFlagsRegistrationDomains, opinterface, enum_reply, NULL);
1110 					break;
1111 
1112 		case 'F':	printf("Looking for recommended browsing domains:\n");
1113 					err = DNSServiceEnumerateDomains(&client, kDNSServiceFlagsBrowseDomains, opinterface, enum_reply, NULL);
1114 					//enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "nicta.com.au.", NULL);
1115 					//enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "bonjour.nicta.com.au.", NULL);
1116 					//enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "ibm.com.", NULL);
1117 					//enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "dns-sd.ibm.com.", NULL);
1118 					break;
1119 
1120 		case 'B':	typ = (argc < opi+1) ? "" : argv[opi+0];
1121 					dom = (argc < opi+2) ? "" : argv[opi+1];  // Missing domain argument is the same as empty string i.e. use system default(s)
1122 					typ = gettype(buffer, typ);
1123 					if (dom[0] == '.' && dom[1] == 0) dom[0] = 0;   // We allow '.' on the command line as a synonym for empty string
1124 					printf("Browsing for %s%s%s\n", typ, dom[0] ? "." : "", dom);
1125 					err = DNSServiceBrowse(&client, flags, opinterface, typ, dom, browse_reply, NULL);
1126 					break;
1127 
1128 		case 'Z':	typ = (argc < opi+1) ? "" : argv[opi+0];
1129 					dom = (argc < opi+2) ? "" : argv[opi+1];  // Missing domain argument is the same as empty string i.e. use system default(s)
1130 					typ = gettype(buffer, typ);
1131 					if (dom[0] == '.' && dom[1] == 0) dom[0] = 0;   // We allow '.' on the command line as a synonym for empty string
1132 					printf("Browsing for %s%s%s\n", typ, dom[0] ? "." : "", dom);
1133 					err = DNSServiceCreateConnection(&client);
1134 					sc1 = client;
1135 					err = DNSServiceBrowse(&sc1, kDNSServiceFlagsShareConnection, opinterface, typ, dom, zonedata_browse, NULL);
1136 					break;
1137 
1138 		case 'l':
1139 		case 'L':	{
1140 					DNSServiceFlags rflags = 0;
1141 					if (argc < opi+2) goto Fail;
1142  					typ = (argc < opi+2) ? ""      : argv[opi+1];
1143  					dom = (argc < opi+3) ? "local" : argv[opi+2];
1144  					typ = gettype(buffer, typ);
1145  					if (dom[0] == '.' && dom[1] == 0) dom = "local";   // We allow '.' on the command line as a synonym for "local"
1146  					printf("Lookup %s.%s.%s\n", argv[opi+0], typ, dom);
1147 					if (operation == 'l') rflags |= kDNSServiceFlagsWakeOnResolve;
1148 					err = DNSServiceResolve(&client, rflags, opinterface, argv[opi+0], typ, dom, resolve_reply, NULL);
1149  					break;
1150 					}
1151 
1152 		case 'R':	if (argc < opi+4) goto Fail;
1153 					typ = (argc < opi+2) ? "" : argv[opi+1];
1154 					dom = (argc < opi+3) ? "" : argv[opi+2];
1155 					typ = gettype(buffer, typ);
1156 					if (dom[0] == '.' && dom[1] == 0) dom[0] = 0;   // We allow '.' on the command line as a synonym for empty string
1157 					err = RegisterService(&client, argv[opi+0], typ, dom, NULL, argv[opi+3], argc-(opi+4), argv+(opi+4), flags);
1158 					break;
1159 
1160 		case 'P':	if (argc < opi+6) goto Fail;
1161 					err = DNSServiceCreateConnection(&client_pa);
1162 					if (err) { fprintf(stderr, "DNSServiceCreateConnection returned %d\n", err); return(err); }
1163 					err = RegisterProxyAddressRecord(client_pa, argv[opi+4], argv[opi+5], flags);
1164 					if (err) break;
1165 					err = RegisterService(&client, argv[opi+0], gettype(buffer, argv[opi+1]), argv[opi+2], argv[opi+4], argv[opi+3], argc-(opi+6), argv+(opi+6), flags);
1166 					break;
1167 
1168 		case 't':
1169 		case 'q':
1170 		case 'Q':
1171 		case 'C':	{
1172 					uint16_t rrtype, rrclass;
1173 					flags |= kDNSServiceFlagsReturnIntermediates;
1174 					if (operation == 'q') flags |= kDNSServiceFlagsSuppressUnusable;
1175 					if (operation == 't') flags |= (kDNSServiceFlagsSuppressUnusable | kDNSServiceFlagsTimeout);
1176 					if (argc < opi+1) goto Fail;
1177 					rrtype = (argc <= opi+1) ? kDNSServiceType_A  : GetRRType(argv[opi+1]);
1178 					rrclass = (argc <= opi+2) ? kDNSServiceClass_IN : atoi(argv[opi+2]);
1179 					if (rrtype == kDNSServiceType_TXT || rrtype == kDNSServiceType_PTR) flags |= kDNSServiceFlagsLongLivedQuery;
1180 					err = DNSServiceQueryRecord(&client, flags, opinterface, argv[opi+0], rrtype, rrclass, qr_reply, NULL);
1181 					break;
1182 					}
1183 
1184 		case 'A':
1185 		case 'U':
1186 		case 'N':	{
1187 					Opaque16 registerPort = { { 0x12, 0x34 } };
1188 					static const char TXT[] = "\xC" "First String" "\xD" "Second String" "\xC" "Third String";
1189 					printf("Registering Service Test._testupdate._tcp.local.\n");
1190 					err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testupdate._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT)-1, TXT, reg_reply, NULL);
1191 					break;
1192 					}
1193 
1194 		case 'T':	{
1195 					Opaque16 registerPort = { { 0x23, 0x45 } };
1196 					char TXT[1024];
1197 					unsigned int i;
1198 					for (i=0; i<sizeof(TXT); i++)
1199 						if ((i & 0x1F) == 0) TXT[i] = 0x1F; else TXT[i] = 'A' + (i >> 5);
1200 					printf("Registering Service Test._testlargetxt._tcp.local.\n");
1201 					err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testlargetxt._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT), TXT, reg_reply, NULL);
1202 					break;
1203 					}
1204 
1205 		case 'M':	{
1206 					pid_t pid = getpid();
1207 					Opaque16 registerPort = { { pid >> 8, pid & 0xFF } };
1208 					static const char TXT1[] = "\xC" "First String"  "\xD" "Second String" "\xC" "Third String";
1209 					static const char TXT2[] = "\xD" "Fourth String" "\xC" "Fifth String"  "\xC" "Sixth String";
1210 					printf("Registering Service Test._testdualtxt._tcp.local.\n");
1211 					err = DNSServiceRegister(&client, flags, opinterface, "Test", "_testdualtxt._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT1)-1, TXT1, reg_reply, NULL);
1212 					if (!err) err = DNSServiceAddRecord(client, &record, flags, kDNSServiceType_TXT, sizeof(TXT2)-1, TXT2, 0);
1213 					break;
1214 					}
1215 
1216 		case 'I':	{
1217 					pid_t pid = getpid();
1218 					Opaque16 registerPort = { { pid >> 8, pid & 0xFF } };
1219 					static const char TXT[] = "\x09" "Test Data";
1220 					printf("Registering Service Test._testtxt._tcp.local.\n");
1221 					err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testtxt._tcp.", "", NULL, registerPort.NotAnInteger, 0, NULL, reg_reply, NULL);
1222 					if (!err) err = DNSServiceUpdateRecord(client, NULL, 0, sizeof(TXT)-1, TXT, 0);
1223 					break;
1224 					}
1225 
1226 #if HAS_NAT_PMP_API
1227 		case 'X':   {
1228 					if (argc == opi)	// If no arguments, just fetch IP address
1229 						err = DNSServiceNATPortMappingCreate(&client, 0, 0, 0, 0, 0, 0, port_mapping_create_reply, NULL);
1230 					else if (argc >= opi+2 && atoi(argv[opi+0]) == 0)
1231 						{
1232 						DNSServiceProtocol prot  = GetProtocol(argv[opi+0]);						// Must specify TCP or UDP
1233 						uint16_t IntPortAsNumber = atoi(argv[opi+1]);							// Must specify internal port
1234 						uint16_t ExtPortAsNumber = (argc < opi+3) ? 0 : atoi(argv[opi+2]);	// Optional desired external port
1235 						uint32_t ttl             = (argc < opi+4) ? 0 : atoi(argv[opi+3]);	// Optional desired lease lifetime
1236 						Opaque16 intp = { { IntPortAsNumber >> 8, IntPortAsNumber & 0xFF } };
1237 						Opaque16 extp = { { ExtPortAsNumber >> 8, ExtPortAsNumber & 0xFF } };
1238 						err = DNSServiceNATPortMappingCreate(&client, 0, 0, prot, intp.NotAnInteger, extp.NotAnInteger, ttl, port_mapping_create_reply, NULL);
1239 						}
1240 					else goto Fail;
1241 					break;
1242 		            }
1243 #endif
1244 
1245 #if HAS_ADDRINFO_API
1246 		case 'G':   {
1247 					if (argc != opi+2) goto Fail;
1248 					else err = DNSServiceGetAddrInfo(&client, kDNSServiceFlagsReturnIntermediates, opinterface, GetProtocol(argv[opi+0]), argv[opi+1], addrinfo_reply, NULL);
1249 					break;
1250 		            }
1251 #endif
1252 
1253 		case 'S':	{
1254 					Opaque16 registerPort = { { 0x23, 0x45 } };		// 9029 decimal
1255 					unsigned char txtrec[16] = "\xF" "/path=test.html";
1256 					DNSRecordRef rec;
1257 					unsigned char nulrec[4] = "1234";
1258 
1259 					err = DNSServiceCreateConnection(&client);
1260 					if (err) { fprintf(stderr, "DNSServiceCreateConnection failed %ld\n", (long int)err); return (-1); }
1261 
1262 					sc1 = client;
1263 					err = DNSServiceBrowse(&sc1, kDNSServiceFlagsShareConnection, opinterface, "_http._tcp", "", browse_reply, NULL);
1264 					if (err) { fprintf(stderr, "DNSServiceBrowse _http._tcp failed %ld\n", (long int)err); return (-1); }
1265 
1266 					sc2 = client;
1267 					err = DNSServiceBrowse(&sc2, kDNSServiceFlagsShareConnection, opinterface, "_ftp._tcp", "", browse_reply, NULL);
1268 					if (err) { fprintf(stderr, "DNSServiceBrowse _ftp._tcp failed %ld\n", (long int)err); return (-1); }
1269 
1270 					sc3 = client;
1271 					err = DNSServiceRegister(&sc3, kDNSServiceFlagsShareConnection, opinterface, "kDNSServiceFlagsShareConnection",
1272 						"_http._tcp", "local", NULL, registerPort.NotAnInteger, 0, NULL, reg_reply, NULL);
1273 					if (err) { fprintf(stderr, "SharedConnection DNSServiceRegister failed %ld\n", (long int)err); return (-1); }
1274 
1275 					err = DNSServiceUpdateRecord(sc3, NULL, 0, sizeof(txtrec), txtrec, 0);
1276 					if (err) { fprintf(stderr, "SharedConnection DNSServiceUpdateRecord failed %ld\n", (long int)err); return (-1); }
1277 
1278 					err = DNSServiceAddRecord(sc3, &rec, 0, kDNSServiceType_NULL, sizeof(nulrec), nulrec, 0);
1279 					if (err) { fprintf(stderr, "SharedConnection DNSServiceAddRecord failed %ld\n", (long int)err); return (-1); }
1280 
1281 					err = DNSServiceRemoveRecord(sc3, rec, 0);
1282 					if (err) { fprintf(stderr, "SharedConnection DNSServiceRemoveRecord failed %ld\n", (long int)err); return (-1); }
1283 
1284 					break;
1285 					}
1286 
1287 		case 'V':   {
1288 					uint32_t v;
1289 					uint32_t size = sizeof(v);
1290 					err = DNSServiceGetProperty(kDNSServiceProperty_DaemonVersion, &v, &size);
1291 					if (err) fprintf(stderr, "DNSServiceGetProperty failed %ld\n", (long int)err);
1292 					else printf("Currently running daemon (system service) is version %d.%d\n", v / 10000, v / 100 % 100);
1293 					exit(0);
1294     				}
1295 
1296 		default: goto Fail;
1297 		}
1298 
1299 	if (!client || err != kDNSServiceErr_NoError) { fprintf(stderr, "DNSService call failed %ld\n", (long int)err); return (-1); }
1300 	HandleEvents();
1301 
1302 	// Be sure to deallocate the DNSServiceRef when you're finished
1303 	if (client   ) DNSServiceRefDeallocate(client   );
1304 	if (client_pa) DNSServiceRefDeallocate(client_pa);
1305 	return 0;
1306 
1307 Fail:
1308 	fprintf(stderr, "%s -E                  (Enumerate recommended registration domains)\n", a0);
1309 	fprintf(stderr, "%s -F                      (Enumerate recommended browsing domains)\n", a0);
1310 	fprintf(stderr, "%s -B        <Type> <Domain>        (Browse for services instances)\n", a0);
1311 	fprintf(stderr, "%s -L <Name> <Type> <Domain>           (Look up a service instance)\n", a0);
1312 	fprintf(stderr, "%s -R <Name> <Type> <Domain> <Port> [<TXT>...] (Register a service)\n", a0);
1313 	fprintf(stderr, "%s -P <Name> <Type> <Domain> <Port> <Host> <IP> [<TXT>...]  (Proxy)\n", a0);
1314 	fprintf(stderr, "%s -Z        <Type> <Domain>   (Output results in Zone File format)\n", a0);
1315 	fprintf(stderr, "%s -Q <FQDN> <rrtype> <rrclass> (Generic query for any record type)\n", a0);
1316 	fprintf(stderr, "%s -C <FQDN> <rrtype> <rrclass>   (Query; reconfirming each result)\n", a0);
1317 #if HAS_NAT_PMP_API
1318 	fprintf(stderr, "%s -X udp/tcp/udptcp <IntPort> <ExtPort> <TTL>   (NAT Port Mapping)\n", a0);
1319 #endif
1320 #if HAS_ADDRINFO_API
1321 	fprintf(stderr, "%s -G v4/v6/v4v6 <Hostname>  (Get address information for hostname)\n", a0);
1322 #endif
1323 	fprintf(stderr, "%s -V    (Get version of currently running daemon / system service)\n", a0);
1324 
1325 	fprintf(stderr, "%s -A                      (Test Adding/Updating/Deleting a record)\n", a0);
1326 	fprintf(stderr, "%s -U                                  (Test updating a TXT record)\n", a0);
1327 	fprintf(stderr, "%s -N                             (Test adding a large NULL record)\n", a0);
1328 	fprintf(stderr, "%s -T                            (Test creating a large TXT record)\n", a0);
1329 	fprintf(stderr, "%s -M      (Test creating a registration with multiple TXT records)\n", a0);
1330 	fprintf(stderr, "%s -I   (Test registering and then immediately updating TXT record)\n", a0);
1331 	fprintf(stderr, "%s -S                 (Test multiple operations on a shared socket)\n", a0);
1332 	return 0;
1333 	}
1334 
1335 // Note: The C preprocessor stringify operator ('#') makes a string from its argument, without macro expansion
1336 // e.g. If "version" is #define'd to be "4", then STRINGIFY_AWE(version) will return the string "version", not "4"
1337 // To expand "version" to its value before making the string, use STRINGIFY(version) instead
1338 #define STRINGIFY_ARGUMENT_WITHOUT_EXPANSION(s) #s
1339 #define STRINGIFY(s) STRINGIFY_ARGUMENT_WITHOUT_EXPANSION(s)
1340 
1341 #ifndef __ANDROID__
1342 // NOT static -- otherwise the compiler may optimize it out
1343 // The "@(#) " pattern is a special prefix the "what" command looks for
1344 const char VersionString_SCCS[] = "@(#) dns-sd " STRINGIFY(mDNSResponderVersion) " (" __DATE__ " " __TIME__ ")";
1345 #endif
1346 
1347 #if _BUILDING_XCODE_PROJECT_
1348 // If the process crashes, then this string will be magically included in the automatically-generated crash log
1349 const char *__crashreporter_info__ = VersionString_SCCS + 5;
1350 asm(".desc ___crashreporter_info__, 0x10");
1351 #endif
1352