]> git.saurik.com Git - apple/mdnsresponder.git/blob - Clients/dns-sd.c
mDNSResponder-1310.40.42.tar.gz
[apple/mdnsresponder.git] / Clients / dns-sd.c
1 /* -*- Mode: C; tab-width: 4 -*-
2 *
3 * Copyright (c) 2002-2020 Apple Inc. All rights reserved.
4 *
5 * Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple 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 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 To build this tool, copy and paste the following into a command line:
41
42 OS X:
43 gcc dns-sd.c -o dns-sd
44
45 POSIX systems:
46 gcc dns-sd.c -o dns-sd -I../mDNSShared -ldns_sd
47
48 Windows:
49 cl dns-sd.c -I../mDNSShared -DNOT_HAVE_GETOPT ws2_32.lib ..\mDNSWindows\DLL\Release\dnssd.lib
50 (may require that you run a Visual Studio script such as vsvars32.bat first)
51 */
52
53 // For testing changes to dnssd_clientstub.c, uncomment this line and the code will be compiled
54 // with an embedded copy of the client stub instead of linking the system library version at runtime.
55 // This also useful to work around link errors when you're working on an older version of Mac OS X,
56 // and trying to build a newer version of the "dns-sd" command which uses new API entry points that
57 // aren't in the system's /usr/lib/libSystem.dylib.
58 //#define TEST_NEW_CLIENTSTUB 1
59
60 #include <ctype.h>
61 #include <stdio.h> // For stdout, stderr
62 #include <stdlib.h> // For exit()
63 #include <string.h> // For strlen(), strcpy()
64 #include <errno.h> // For errno, EINTR
65 #include <time.h>
66 #include <sys/types.h> // For u_char
67 #ifdef APPLE_OSX_mDNSResponder
68 #include <inttypes.h> // For PRId64
69 #endif // APPLE_OSX_mDNSResponder
70
71 #if APPLE_OSX_mDNSResponder
72 #include <xpc/xpc.h>
73 #include "xpc_clients.h"
74 #endif
75
76 #ifdef _WIN32
77 #include <winsock2.h>
78 #include <ws2tcpip.h>
79 #include <Iphlpapi.h>
80 #include <process.h>
81 typedef int pid_t;
82 #define getpid _getpid
83 #define strcasecmp _stricmp
84 #define snprintf _snprintf
85 static const char kFilePathSep = '\\';
86 #ifndef HeapEnableTerminationOnCorruption
87 # define HeapEnableTerminationOnCorruption (HEAP_INFORMATION_CLASS)1
88 #endif
89 #if !defined(IFNAMSIZ)
90 #define IFNAMSIZ 16
91 #endif
92 #define if_nametoindex if_nametoindex_win
93 #define if_indextoname if_indextoname_win
94
95 typedef PCHAR (WINAPI * if_indextoname_funcptr_t)(ULONG index, PCHAR name);
96 typedef ULONG (WINAPI * if_nametoindex_funcptr_t)(PCSTR name);
97
98 unsigned if_nametoindex_win(const char *ifname)
99 {
100 HMODULE library;
101 unsigned index = 0;
102
103 // Try and load the IP helper library dll
104 if ((library = LoadLibrary(TEXT("Iphlpapi")) ) != NULL )
105 {
106 if_nametoindex_funcptr_t if_nametoindex_funcptr;
107
108 // On Vista and above there is a Posix like implementation of if_nametoindex
109 if ((if_nametoindex_funcptr = (if_nametoindex_funcptr_t) GetProcAddress(library, "if_nametoindex")) != NULL )
110 {
111 index = if_nametoindex_funcptr(ifname);
112 }
113
114 FreeLibrary(library);
115 }
116
117 return index;
118 }
119
120 char * if_indextoname_win( unsigned ifindex, char *ifname)
121 {
122 HMODULE library;
123 char * name = NULL;
124
125 // Try and load the IP helper library dll
126 if ((library = LoadLibrary(TEXT("Iphlpapi")) ) != NULL )
127 {
128 if_indextoname_funcptr_t if_indextoname_funcptr;
129
130 // On Vista and above there is a Posix like implementation of if_indextoname
131 if ((if_indextoname_funcptr = (if_indextoname_funcptr_t) GetProcAddress(library, "if_indextoname")) != NULL )
132 {
133 name = if_indextoname_funcptr(ifindex, ifname);
134 }
135
136 FreeLibrary(library);
137 }
138
139 return name;
140 }
141
142 static size_t _sa_len(const struct sockaddr *addr)
143 {
144 if (addr->sa_family == AF_INET) return (sizeof(struct sockaddr_in));
145 else if (addr->sa_family == AF_INET6) return (sizeof(struct sockaddr_in6));
146 else return (sizeof(struct sockaddr));
147 }
148
149 # define SA_LEN(addr) (_sa_len(addr))
150
151 #else
152 #include <unistd.h> // For getopt() and optind
153 #include <netdb.h> // For getaddrinfo()
154 #include <sys/time.h> // For struct timeval
155 #include <sys/socket.h> // For AF_INET
156 #include <netinet/in.h> // For struct sockaddr_in()
157 #include <arpa/inet.h> // For inet_addr()
158 #include <net/if.h> // For if_nametoindex()
159 static const char kFilePathSep = '/';
160 // #ifndef NOT_HAVE_SA_LEN
161 // #define SA_LEN(addr) ((addr)->sa_len)
162 // #else
163 #define SA_LEN(addr) (((addr)->sa_family == AF_INET6) ? sizeof(struct sockaddr_in6) : sizeof(struct sockaddr_in))
164 // #endif
165 #endif
166
167 #if (TEST_NEW_CLIENTSTUB && !defined(__APPLE_API_PRIVATE))
168 #define __APPLE_API_PRIVATE 1
169 #endif
170
171 // DNSServiceSetDispatchQueue is not supported on 10.6 & prior
172 #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)
173 #undef _DNS_SD_LIBDISPATCH
174 #endif
175 #include "dns_sd.h"
176 #include "dns_sd_private.h"
177 #include "ClientCommon.h"
178
179
180 #if TEST_NEW_CLIENTSTUB
181 #include "../mDNSShared/dnssd_ipc.c"
182 #include "../mDNSShared/dnssd_clientlib.c"
183 #include "../mDNSShared/dnssd_clientstub.c"
184 #endif
185
186 //*************************************************************************************************************
187 // Globals
188
189 #define DS_FIXED_SIZE 4
190 typedef struct
191 {
192 unsigned short keyTag;
193 unsigned char alg;
194 unsigned char digestType;
195 unsigned char *digest;
196 } rdataDS;
197
198 #define DNSKEY_FIXED_SIZE 4
199 typedef struct
200 {
201 unsigned short flags;
202 unsigned char proto;
203 unsigned char alg;
204 unsigned char *data;
205 } rdataDNSKey;
206
207 //size of rdataRRSIG excluding signerName and signature (which are variable fields)
208 #define RRSIG_FIXED_SIZE 18
209 typedef struct
210 {
211 unsigned short typeCovered;
212 unsigned char alg;
213 unsigned char labels;
214 unsigned int origTTL;
215 unsigned int sigExpireTime;
216 unsigned int sigInceptTime;
217 unsigned short keyTag;
218 char signerName[256];
219 //unsigned char *signature
220 } rdataRRSig;
221
222 #define RR_TYPE_SIZE 16
223
224 typedef union { unsigned char b[2]; unsigned short NotAnInteger; } Opaque16;
225
226 static int operation;
227 static uint32_t opinterface = kDNSServiceInterfaceIndexAny;
228 static DNSServiceRef client = NULL;
229 static DNSServiceRef client_pa = NULL; // DNSServiceRef for RegisterProxyAddressRecord
230 static DNSServiceRef sc1, sc2, sc3; // DNSServiceRefs for kDNSServiceFlagsShareConnection testing
231
232 static int num_printed;
233 static char addtest = 0;
234 static DNSRecordRef record = NULL;
235 static char myhinfoW[14] = "\002PC\012Windows XP";
236 static char myhinfoX[ 9] = "\003Mac\004OS X";
237 static char updatetest[3] = "\002AA";
238 static char bigNULL[8192]; // 8K is maximum rdata we support
239
240 #if _DNS_SD_LIBDISPATCH
241 dispatch_queue_t main_queue;
242 dispatch_source_t timer_source;
243 #endif
244
245 // Note: the select() implementation on Windows (Winsock2) fails with any timeout much larger than this
246 #define LONG_TIME 100000000
247
248 static volatile int stopNow = 0;
249 static volatile int timeOut = LONG_TIME;
250
251 #if _DNS_SD_LIBDISPATCH
252 #define EXIT_IF_LIBDISPATCH_FATAL_ERROR(E) \
253 if (main_queue && (E) == kDNSServiceErr_ServiceNotRunning) { fprintf(stderr, "Error code %d\n", (E)); exit(0); }
254 #else
255 #define EXIT_IF_LIBDISPATCH_FATAL_ERROR(E)
256 #endif
257
258 //*************************************************************************************************************
259 // Supporting Utility Functions
260 static uint16_t GetRRClass(const char *s)
261 {
262 if (!strcasecmp(s, "IN"))
263 return kDNSServiceClass_IN;
264 else
265 return(atoi(s));
266 }
267
268 static uint16_t GetRRType(const char *s)
269 {
270 if (!strcasecmp(s, "A" )) return(kDNSServiceType_A);
271 else if (!strcasecmp(s, "NS" )) return(kDNSServiceType_NS);
272 else if (!strcasecmp(s, "MD" )) return(kDNSServiceType_MD);
273 else if (!strcasecmp(s, "MF" )) return(kDNSServiceType_MF);
274 else if (!strcasecmp(s, "CNAME" )) return(kDNSServiceType_CNAME);
275 else if (!strcasecmp(s, "SOA" )) return(kDNSServiceType_SOA);
276 else if (!strcasecmp(s, "MB" )) return(kDNSServiceType_MB);
277 else if (!strcasecmp(s, "MG" )) return(kDNSServiceType_MG);
278 else if (!strcasecmp(s, "MR" )) return(kDNSServiceType_MR);
279 else if (!strcasecmp(s, "NULL" )) return(kDNSServiceType_NULL);
280 else if (!strcasecmp(s, "WKS" )) return(kDNSServiceType_WKS);
281 else if (!strcasecmp(s, "PTR" )) return(kDNSServiceType_PTR);
282 else if (!strcasecmp(s, "HINFO" )) return(kDNSServiceType_HINFO);
283 else if (!strcasecmp(s, "MINFO" )) return(kDNSServiceType_MINFO);
284 else if (!strcasecmp(s, "MX" )) return(kDNSServiceType_MX);
285 else if (!strcasecmp(s, "TXT" )) return(kDNSServiceType_TXT);
286 else if (!strcasecmp(s, "RP" )) return(kDNSServiceType_RP);
287 else if (!strcasecmp(s, "AFSDB" )) return(kDNSServiceType_AFSDB);
288 else if (!strcasecmp(s, "X25" )) return(kDNSServiceType_X25);
289 else if (!strcasecmp(s, "ISDN" )) return(kDNSServiceType_ISDN);
290 else if (!strcasecmp(s, "RT" )) return(kDNSServiceType_RT);
291 else if (!strcasecmp(s, "NSAP" )) return(kDNSServiceType_NSAP);
292 else if (!strcasecmp(s, "NSAP_PTR")) return(kDNSServiceType_NSAP_PTR);
293 else if (!strcasecmp(s, "SIG" )) return(kDNSServiceType_SIG);
294 else if (!strcasecmp(s, "KEY" )) return(kDNSServiceType_KEY);
295 else if (!strcasecmp(s, "PX" )) return(kDNSServiceType_PX);
296 else if (!strcasecmp(s, "GPOS" )) return(kDNSServiceType_GPOS);
297 else if (!strcasecmp(s, "AAAA" )) return(kDNSServiceType_AAAA);
298 else if (!strcasecmp(s, "LOC" )) return(kDNSServiceType_LOC);
299 else if (!strcasecmp(s, "NXT" )) return(kDNSServiceType_NXT);
300 else if (!strcasecmp(s, "EID" )) return(kDNSServiceType_EID);
301 else if (!strcasecmp(s, "NIMLOC" )) return(kDNSServiceType_NIMLOC);
302 else if (!strcasecmp(s, "SRV" )) return(kDNSServiceType_SRV);
303 else if (!strcasecmp(s, "ATMA" )) return(kDNSServiceType_ATMA);
304 else if (!strcasecmp(s, "NAPTR" )) return(kDNSServiceType_NAPTR);
305 else if (!strcasecmp(s, "KX" )) return(kDNSServiceType_KX);
306 else if (!strcasecmp(s, "CERT" )) return(kDNSServiceType_CERT);
307 else if (!strcasecmp(s, "A6" )) return(kDNSServiceType_A6);
308 else if (!strcasecmp(s, "DNAME" )) return(kDNSServiceType_DNAME);
309 else if (!strcasecmp(s, "SINK" )) return(kDNSServiceType_SINK);
310 else if (!strcasecmp(s, "OPT" )) return(kDNSServiceType_OPT);
311 else if (!strcasecmp(s, "TKEY" )) return(kDNSServiceType_TKEY);
312 else if (!strcasecmp(s, "TSIG" )) return(kDNSServiceType_TSIG);
313 else if (!strcasecmp(s, "IXFR" )) return(kDNSServiceType_IXFR);
314 else if (!strcasecmp(s, "AXFR" )) return(kDNSServiceType_AXFR);
315 else if (!strcasecmp(s, "MAILB" )) return(kDNSServiceType_MAILB);
316 else if (!strcasecmp(s, "MAILA" )) return(kDNSServiceType_MAILA);
317 else if (!strcasecmp(s, "dnskey" )) return(kDNSServiceType_DNSKEY);
318 else if (!strcasecmp(s, "ds" )) return(kDNSServiceType_DS);
319 else if (!strcasecmp(s, "rrsig" )) return(kDNSServiceType_RRSIG);
320 else if (!strcasecmp(s, "nsec" )) return(kDNSServiceType_NSEC);
321 else if (!strcasecmp(s, "SVCB" )) return(kDNSServiceType_SVCB);
322 else if (!strcasecmp(s, "HTTPS" )) return(kDNSServiceType_HTTPS);
323 else if (!strcasecmp(s, "ANY" )) return(kDNSServiceType_ANY);
324 else return(atoi(s));
325 }
326
327 static char *DNSTypeName(unsigned short rr_type)
328 {
329 switch (rr_type)
330 {
331 case kDNSServiceType_A: return("Addr");
332 case kDNSServiceType_NS: return("NS");
333 case kDNSServiceType_MD: return("MD");
334 case kDNSServiceType_MF: return("MF");
335 case kDNSServiceType_CNAME: return("CNAME");
336 case kDNSServiceType_SOA: return("SOA");
337 case kDNSServiceType_MB: return("MB");
338 case kDNSServiceType_MG: return("MG");
339 case kDNSServiceType_MR: return("MR");
340 case kDNSServiceType_NULL: return("NULL");
341 case kDNSServiceType_WKS: return("WKS");
342 case kDNSServiceType_PTR: return("PTR");
343 case kDNSServiceType_HINFO: return("HINFO");
344 case kDNSServiceType_MINFO: return("MINFO");
345 case kDNSServiceType_MX: return("MX");
346 case kDNSServiceType_TXT: return("TXT");
347 case kDNSServiceType_RP: return("RP");
348 case kDNSServiceType_AFSDB: return("AFSDB");
349 case kDNSServiceType_X25: return("X25");
350 case kDNSServiceType_ISDN: return("ISDN");
351 case kDNSServiceType_RT: return("RT");
352 case kDNSServiceType_NSAP: return("NSAP");
353 case kDNSServiceType_NSAP_PTR: return("NSAP_PTR");
354 case kDNSServiceType_SIG: return("SIG");
355 case kDNSServiceType_KEY: return("KEY");
356 case kDNSServiceType_PX: return("PX");
357 case kDNSServiceType_GPOS: return("GPOS");
358 case kDNSServiceType_AAAA: return("AAAA");
359 case kDNSServiceType_LOC: return("LOC");
360 case kDNSServiceType_NXT: return("NXT");
361 case kDNSServiceType_EID: return("EID");
362 case kDNSServiceType_NIMLOC: return("NIMLOC");
363 case kDNSServiceType_SRV: return("SRV");
364 case kDNSServiceType_ATMA: return("ATMA");
365 case kDNSServiceType_NAPTR: return("NAPTR");
366 case kDNSServiceType_KX: return("KX");
367 case kDNSServiceType_CERT: return("CERT");
368 case kDNSServiceType_A6: return("A6");
369 case kDNSServiceType_DNAME: return("DNAME");
370 case kDNSServiceType_SINK: return("SINK");
371 case kDNSServiceType_OPT: return("OPT");
372 case kDNSServiceType_APL: return("APL");
373 case kDNSServiceType_DS: return("DS");
374 case kDNSServiceType_SSHFP: return("SSHFP");
375 case kDNSServiceType_IPSECKEY: return("IPSECKEY");
376 case kDNSServiceType_RRSIG: return("RRSIG");
377 case kDNSServiceType_NSEC: return("NSEC");
378 case kDNSServiceType_DNSKEY: return("DNSKEY");
379 case kDNSServiceType_DHCID: return("DHCID");
380 case kDNSServiceType_NSEC3: return("NSEC3");
381 case kDNSServiceType_NSEC3PARAM: return("NSEC3PARAM");
382 case kDNSServiceType_HIP: return("HIP");
383 case kDNSServiceType_SPF: return("SPF");
384 case kDNSServiceType_UINFO: return("UINFO");
385 case kDNSServiceType_UID: return("UID");
386 case kDNSServiceType_GID: return("GID");
387 case kDNSServiceType_UNSPEC: return("UNSPEC");
388 case kDNSServiceType_TKEY: return("TKEY");
389 case kDNSServiceType_TSIG: return("TSIG");
390 case kDNSServiceType_IXFR: return("IXFR");
391 case kDNSServiceType_AXFR: return("AXFR");
392 case kDNSServiceType_MAILB: return("MAILB");
393 case kDNSServiceType_MAILA: return("MAILA");
394 case kDNSServiceType_SVCB: return("SVCB");
395 case kDNSServiceType_HTTPS: return("HTTPS");
396 case kDNSServiceType_ANY: return("ANY");
397 default:
398 {
399 static char buffer[RR_TYPE_SIZE];
400 snprintf(buffer, sizeof(buffer), "TYPE%d", rr_type);
401 return(buffer);
402 }
403 }
404 }
405
406 static unsigned short swap16(unsigned short x)
407 {
408 unsigned char *ptr = (unsigned char *)&x;
409 return (unsigned short)((unsigned short)ptr[0] << 8 | ptr[1]);
410 }
411
412 static unsigned int swap32(unsigned int x)
413 {
414 unsigned char *ptr = (unsigned char *)&x;
415 return (unsigned int)((unsigned int)ptr[0] << 24 | (unsigned int)ptr[1] << 16 | (unsigned int)ptr[2] << 8 | ptr[3]);
416 }
417 static unsigned int keytag(unsigned char *key, unsigned int keysize)
418 {
419 unsigned long ac;
420 unsigned int i;
421
422 for (ac = 0, i = 0; i < keysize; ++i)
423 ac += (i & 1) ? key[i] : key[i] << 8;
424 ac += (ac >> 16) & 0xFFFF;
425 return ac & 0xFFFF;
426 }
427
428 // Base 64 encoding according to <https://tools.ietf.org/html/rfc4648#section-4>.
429 #define kBase64EncodingTable "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
430
431 static void base64Encode(char *buffer, size_t buflen, void *rdata, size_t rdlen)
432 {
433 const uint8_t *src = (const uint8_t *)rdata;
434 const uint8_t *const end = &src[rdlen];
435 char *dst = buffer;
436 const char *lim;
437
438 if (buflen == 0) return;
439 lim = &buffer[buflen - 1];
440 while ((src < end) && (dst < lim))
441 {
442 uint32_t i;
443 const size_t rem = (size_t)(end - src);
444
445 // Form a 24-bit input group. If less than 24 bits remain, pad with zero bits.
446 if ( rem >= 3) i = (src[0] << 16) | (src[1] << 8) | src[2]; // 24 bits are equal to 4 6-bit groups.
447 else if (rem == 2) i = (src[0] << 16) | (src[1] << 8); // 16 bits are treated as 3 6-bit groups + 1 pad
448 else i = src[0] << 16; // 8 bits are treated as 2 6-bit groups + 2 pads
449
450 // Encode each 6-bit group.
451 *dst++ = kBase64EncodingTable[(i >> 18) & 0x3F];
452 if (dst < lim) *dst++ = kBase64EncodingTable[(i >> 12) & 0x3F];
453 if (dst < lim) *dst++ = (rem >= 2) ? kBase64EncodingTable[(i >> 6) & 0x3F] : '=';
454 if (dst < lim) *dst++ = (rem >= 3) ? kBase64EncodingTable[ i & 0x3F] : '=';
455 src += (rem > 3) ? 3 : rem;
456 }
457 *dst = '\0';
458 }
459
460 static DNSServiceProtocol GetProtocol(const char *s)
461 {
462 if (!strcasecmp(s, "v4" )) return(kDNSServiceProtocol_IPv4);
463 else if (!strcasecmp(s, "v6" )) return(kDNSServiceProtocol_IPv6);
464 else if (!strcasecmp(s, "v4v6" )) return(kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6);
465 else if (!strcasecmp(s, "v6v4" )) return(kDNSServiceProtocol_IPv4 | kDNSServiceProtocol_IPv6);
466 else if (!strcasecmp(s, "udp" )) return(kDNSServiceProtocol_UDP);
467 else if (!strcasecmp(s, "tcp" )) return(kDNSServiceProtocol_TCP);
468 else if (!strcasecmp(s, "udptcp" )) return(kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP);
469 else if (!strcasecmp(s, "tcpudp" )) return(kDNSServiceProtocol_UDP | kDNSServiceProtocol_TCP);
470 else return(atoi(s));
471 }
472
473
474 //*************************************************************************************************************
475 // Sample callback functions for each of the operation types
476
477 #define printtimestamp() printtimestamp_F(stdout)
478
479 static void printtimestamp_F(FILE *outstream)
480 {
481 struct tm tm;
482 int ms;
483 static char date[16];
484 static char new_date[16];
485 #ifdef _WIN32
486 SYSTEMTIME sysTime;
487 time_t uct = time(NULL);
488 tm = *localtime(&uct);
489 GetLocalTime(&sysTime);
490 ms = sysTime.wMilliseconds;
491 #else
492 struct timeval tv;
493 gettimeofday(&tv, NULL);
494 localtime_r((time_t*)&tv.tv_sec, &tm);
495 ms = tv.tv_usec/1000;
496 #endif
497 strftime(new_date, sizeof(new_date), "%a %d %b %Y", &tm);
498 if (strncmp(date, new_date, sizeof(new_date)))
499 {
500 fprintf(outstream, "DATE: ---%s---\n", new_date); //display date only if it has changed
501 strncpy(date, new_date, sizeof(date));
502 }
503 fprintf(outstream, "%2d:%02d:%02d.%03d ", tm.tm_hour, tm.tm_min, tm.tm_sec, ms);
504 }
505
506 // formating time to RFC 4034 format
507 static void FormatTime(unsigned long te, unsigned char *buf, int bufsize)
508 {
509 struct tm tmTime;
510 #ifdef _WIN32
511 __time32_t t = (__time32_t) te;
512 _gmtime32_s(&tmTime, &t);
513 #else
514 // Time since epoch : strftime takes "tm". Convert seconds to "tm" using
515 // gmtime_r first and then use strftime
516 time_t t = (time_t)te;
517 gmtime_r(&t, &tmTime);
518 #endif
519 strftime((char *)buf, bufsize, "%Y%m%d%H%M%S", &tmTime);
520 }
521
522 static void print_usage(const char *arg0, int print_all)
523 {
524 // Print the commonly used command line options. These are listed in "the order they have been in historically".
525 fprintf(stderr, "%s -E (Enumerate recommended registration domains)\n", arg0);
526 fprintf(stderr, "%s -F (Enumerate recommended browsing domains)\n", arg0);
527 fprintf(stderr, "%s -R <Name> <Type> <Domain> <Port> [<TXT>...] (Register a service)\n", arg0);
528 fprintf(stderr, "%s -P <Name> <Type> <Domain> <Port> <Host> <IP> [<TXT>...] (Register Proxy)\n", arg0);
529 fprintf(stderr, "%s -B <Type> <Domain> (Browse for service instances)\n", arg0);
530 fprintf(stderr, "%s -Z <Type> <Domain> (Output results in Zone File format)\n", arg0);
531 fprintf(stderr, "%s -L <Name> <Type> <Domain> (Resolve (‘lookup’) a service instance)\n", arg0);
532 fprintf(stderr, "%s -Q <name> <rrtype> <rrclass> (Generic query for any record type)\n", arg0);
533 fprintf(stderr, "%s -q <name> <rrtype> <rrclass> (Generic query, using SuppressUnusable)\n", arg0);
534 fprintf(stderr, "%s -G v4/v6/v4v6 <hostname> (Get address information for hostname)\n", arg0);
535 fprintf(stderr, "%s -X udp/tcp/udptcp <IntPort> <ExtPort> <TTL> (NAT Port Mapping)\n", arg0);
536 fprintf(stderr, "%s -H (Print usage for complete command list)\n", arg0);
537 fprintf(stderr, "%s -V (Get version of currently running daemon / system service)\n", arg0);
538 #ifdef APPLE_OSX_mDNSResponder
539 fprintf(stderr, "%s -O [-compress|-stdout](Dump the state of mDNSResponder to file / STDOUT)\n", arg0);
540 #endif // APPLE_OSX_mDNSResponder
541
542 if (print_all) // Print all available options for dns-sd tool. Keep these in alphabetical order for easier maintenance.
543 {
544 fprintf(stderr, "\n");
545 fprintf(stderr, "%s -A (Test Adding/Updating/Deleting a record)\n", arg0);
546 fprintf(stderr, "%s -C <name> <rrtype> <rrclass> (Query; reconfirming each result)\n", arg0);
547 fprintf(stderr, "%s -I (Test registering and then immediately updating TXT record)\n", arg0);
548 fprintf(stderr, "%s -N (Test adding a large NULL record)\n", arg0);
549 fprintf(stderr, "%s -M (Test creating a registration with multiple TXT records)\n", arg0);
550 fprintf(stderr, "%s -S (Test multiple operations on a shared socket)\n", arg0);
551 fprintf(stderr, "%s -T (Test creating a large TXT record)\n", arg0);
552 fprintf(stderr, "%s -U (Test updating a TXT record)\n", arg0);
553 fprintf(stderr, "%s -ble (Use kDNSServiceInterfaceIndexBLE)\n", arg0);
554 fprintf(stderr, "%s -i <Interface> (Run dns-sd cmd on a specific interface (en0/en1)\n", arg0);
555 fprintf(stderr, "%s -includep2p (Set kDNSServiceFlagsIncludeP2P flag)\n", arg0);
556 fprintf(stderr, "%s -includeAWDL (Set kDNSServiceFlagsIncludeAWDL flag)\n", arg0);
557 fprintf(stderr, "%s -intermediates (Set kDNSServiceFlagsReturnIntermediates flag)\n", arg0);
558 fprintf(stderr, "%s -ku (Set kDNSServiceFlagsKnownUnique flag)\n", arg0);
559 fprintf(stderr, "%s -lo (Run dns-sd cmd using local only interface)\n", arg0);
560 fprintf(stderr, "%s -p2p (Use kDNSServiceInterfaceIndexP2P)\n", arg0);
561 fprintf(stderr, "%s -tc (Set kDNSServiceFlagsBackgroundTrafficClass flag)\n", arg0);
562 fprintf(stderr, "%s -test (Run basic API input range tests)\n", arg0);
563 fprintf(stderr, "%s -t1 (Set kDNSServiceFlagsThresholdOne flag)\n", arg0);
564 fprintf(stderr, "%s -tFinder (Set kDNSServiceFlagsThresholdFinder flag)\n", arg0);
565 fprintf(stderr, "%s -timeout (Set kDNSServiceFlagsTimeout flag)\n", arg0);
566 fprintf(stderr, "%s -unicastResponse (Set kDNSServiceFlagsUnicastResponse flag)\n", arg0);
567 fprintf(stderr, "%s -autoTrigger (Set kDNSServiceFlagsAutoTrigger flag)\n", arg0);
568 fprintf(stderr, "%s -enableDNSSEC (Enable DNSSEC validation for the '-Q' query)\n", arg0);
569 }
570 }
571
572 #define DomainMsg(X) (((X) &kDNSServiceFlagsDefault) ? "(Default)" : \
573 ((X) &kDNSServiceFlagsAdd) ? "Added" : "Removed")
574
575 #define MAX_LABELS 128
576
577 static void DNSSD_API enum_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex,
578 DNSServiceErrorType errorCode, const char *replyDomain, void *context)
579 {
580 DNSServiceFlags partialflags = flags & ~(kDNSServiceFlagsMoreComing | kDNSServiceFlagsAdd | kDNSServiceFlagsDefault);
581 int labels = 0, depth = 0, i, initial = 0;
582 char text[64];
583 const char *label[MAX_LABELS];
584
585 (void)sdref; // Unused
586 (void)ifIndex; // Unused
587 (void)context; // Unused
588 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
589
590 // 1. Print the header
591 if (num_printed++ == 0) printf("Timestamp Recommended %s domain\n", operation == 'E' ? "Registration" : "Browsing");
592 printtimestamp();
593 if (errorCode)
594 printf("Error code %d\n", errorCode);
595 else if (!*replyDomain)
596 printf("Error: No reply domain\n");
597 else
598 {
599 printf("%-10s", DomainMsg(flags));
600 printf("%-8s", (flags & kDNSServiceFlagsMoreComing) ? "(More)" : "");
601 if (partialflags) printf("Flags: %4X ", partialflags);
602 else printf(" ");
603
604 // 2. Count the labels
605 while (replyDomain && *replyDomain && labels < MAX_LABELS)
606 {
607 label[labels++] = replyDomain;
608 replyDomain = GetNextLabel(replyDomain, text);
609 }
610
611 // 3. Decide if we're going to clump the last two or three labels (e.g. "apple.com", or "nicta.com.au")
612 if (labels >= 3 && replyDomain - label[labels-1] <= 3 && label[labels-1] - label[labels-2] <= 4) initial = 3;
613 else if (labels >= 2 && replyDomain - label[labels-1] <= 4) initial = 2;
614 else initial = 1;
615 labels -= initial;
616
617 // 4. Print the initial one-, two- or three-label clump
618 for (i=0; i<initial; i++)
619 {
620 GetNextLabel(label[labels+i], text);
621 if (i>0) printf(".");
622 printf("%s", text);
623 }
624 printf("\n");
625
626 // 5. Print the remainder of the hierarchy
627 for (depth=0; depth<labels; depth++)
628 {
629 printf(" ");
630 for (i=0; i<=depth; i++) printf("- ");
631 GetNextLabel(label[labels-1-depth], text);
632 printf("> %s\n", text);
633 }
634 }
635
636 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
637 }
638
639 static int CopyLabels(char *dst, const char *lim, const char **srcp, int labels)
640 {
641 const char *src = *srcp;
642 while (*src != '.' || --labels > 0)
643 {
644 if (*src == '\\') *dst++ = *src++; // Make sure "\." doesn't confuse us
645 if (!*src || dst >= lim) return -1;
646 *dst++ = *src++;
647 if (!*src || dst >= lim) return -1;
648 }
649 *dst++ = 0;
650 *srcp = src + 1; // skip over final dot
651 return 0;
652 }
653
654 static void DNSSD_API zonedata_resolve(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
655 const char *fullname, const char *hosttarget, uint16_t opaqueport, uint16_t txtLen, const unsigned char *txt, void *context)
656 {
657 union { uint16_t s; u_char b[2]; } port = { opaqueport };
658 uint16_t PortAsNumber = ((uint16_t)port.b[0]) << 8 | port.b[1];
659
660 const char *p = fullname;
661 char n[kDNSServiceMaxDomainName];
662 char t[kDNSServiceMaxDomainName];
663
664 const unsigned char *max = txt + txtLen;
665
666 (void)sdref; // Unused
667 (void)ifIndex; // Unused
668 (void)context; // Unused
669
670 //if (!(flags & kDNSServiceFlagsAdd)) return;
671 if (errorCode) { printf("Error code %d\n", errorCode); return; }
672
673 if (CopyLabels(n, n + kDNSServiceMaxDomainName, &p, 3)) return; // Fetch name+type
674 p = fullname;
675 if (CopyLabels(t, t + kDNSServiceMaxDomainName, &p, 1)) return; // Skip first label
676 if (CopyLabels(t, t + kDNSServiceMaxDomainName, &p, 2)) return; // Fetch next two labels (service type)
677
678 if (num_printed++ == 0)
679 {
680 printf("\n");
681 printf("; To direct clients to browse a different domain, substitute that domain in place of '@'\n");
682 printf("%-47s PTR %s\n", "lb._dns-sd._udp", "@");
683 printf("\n");
684 printf("; In the list of services below, the SRV records will typically reference dot-local Multicast DNS names.\n");
685 printf("; When transferring this zone file data to your unicast DNS server, you'll need to replace those dot-local\n");
686 printf("; names with the correct fully-qualified (unicast) domain name of the target host offering the service.\n");
687 }
688
689 printf("\n");
690 printf("%-47s PTR %s\n", t, n);
691 printf("%-47s SRV 0 0 %d %s ; Replace with unicast FQDN of target host\n", n, PortAsNumber, hosttarget);
692 printf("%-47s TXT ", n);
693
694 while (txt < max)
695 {
696 const unsigned char *const end = txt + 1 + txt[0];
697 txt++; // Skip over length byte
698 printf(" \"");
699 while (txt<end)
700 {
701 if (*txt == '\\' || *txt == '\"') printf("\\");
702 printf("%c", *txt++);
703 }
704 printf("\"");
705 }
706 printf("\n");
707
708 DNSServiceRefDeallocate(sdref);
709 free(context);
710
711 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
712 }
713
714 static void DNSSD_API zonedata_browse(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
715 const char *replyName, const char *replyType, const char *replyDomain, void *context)
716 {
717 DNSServiceRef *newref;
718
719 (void)sdref; // Unused
720 (void)context; // Unused
721 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
722
723 if (!(flags & kDNSServiceFlagsAdd)) return;
724 if (errorCode) { printf("Error code %d\n", errorCode); return; }
725
726 newref = malloc(sizeof(*newref));
727 *newref = client;
728 DNSServiceResolve(newref, kDNSServiceFlagsShareConnection, ifIndex, replyName, replyType, replyDomain, zonedata_resolve, newref);
729 }
730
731 static void DNSSD_API browse_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
732 const char *replyName, const char *replyType, const char *replyDomain, void *context)
733 {
734 char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
735 (void)sdref; // Unused
736 (void)context; // Unused
737 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
738
739 if (num_printed++ == 0) printf("Timestamp A/R Flags if %-20s %-20s %s\n", "Domain", "Service Type", "Instance Name");
740 printtimestamp();
741 if (errorCode)
742 printf("Error code %d\n", errorCode);
743 else
744 printf("%s %8X %3d %-20s %-20s %s\n",
745 op, flags, ifIndex, replyDomain, replyType, replyName);
746 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
747
748 // To test selective cancellation of operations of shared sockets,
749 // cancel the current operation when we've got a multiple of five results
750 //if (operation == 'S' && num_printed % 5 == 0) DNSServiceRefDeallocate(sdref);
751 }
752
753 static void ShowTXTRecord(uint16_t txtLen, const unsigned char *txtRecord)
754 {
755 const unsigned char *ptr = txtRecord;
756 const unsigned char *max = txtRecord + txtLen;
757 while (ptr < max)
758 {
759 const unsigned char *const end = ptr + 1 + ptr[0];
760 if (end > max) { printf("<< invalid data >>"); break; }
761 if (++ptr < end) printf(" "); // As long as string is non-empty, begin with a space
762 while (ptr<end)
763 {
764 // We'd like the output to be shell-friendly, so that it can be copied and pasted unchanged into a "dns-sd -R" command.
765 // However, this is trickier than it seems. Enclosing a string in double quotes doesn't necessarily make it
766 // shell-safe, because shells still expand variables like $foo even when they appear inside quoted strings.
767 // Enclosing a string in single quotes is better, but when using single quotes even backslash escapes are ignored,
768 // meaning there's simply no way to represent a single quote (or apostrophe) inside a single-quoted string.
769 // The only remaining solution is not to surround the string with quotes at all, but instead to use backslash
770 // escapes to encode spaces and all other known shell metacharacters.
771 // (If we've missed any known shell metacharacters, please let us know.)
772 // In addition, non-printing ascii codes (0-31) are displayed as \xHH, using a two-digit hex value.
773 // Because '\' is itself a shell metacharacter (the shell escape character), it has to be escaped as "\\" to survive
774 // the round-trip to the shell and back. This means that a single '\' is represented here as EIGHT backslashes:
775 // The C compiler eats half of them, resulting in four appearing in the output.
776 // The shell parses those four as a pair of "\\" sequences, passing two backslashes to the "dns-sd -R" command.
777 // The "dns-sd -R" command interprets this single "\\" pair as an escaped literal backslash. Sigh.
778 if (strchr(" &;`'\"|*?~<>^()[]{}$", *ptr)) printf("\\");
779 if (*ptr == '\\') printf("\\\\\\\\");
780 else if (*ptr >= ' ' ) printf("%c", *ptr);
781 else printf("\\\\x%02X", *ptr);
782 ptr++;
783 }
784 }
785 }
786
787 static void DNSSD_API resolve_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
788 const char *fullname, const char *hosttarget, uint16_t opaqueport, uint16_t txtLen, const unsigned char *txtRecord, void *context)
789 {
790 union { uint16_t s; u_char b[2]; } port = { opaqueport };
791 uint16_t PortAsNumber = ((uint16_t)port.b[0]) << 8 | port.b[1];
792
793 (void)sdref; // Unused
794 (void)ifIndex; // Unused
795 (void)context; // Unused
796 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
797
798 printtimestamp();
799
800 printf("%s ", fullname);
801
802 if (errorCode == kDNSServiceErr_NoSuchRecord) printf("No Such Record");
803 else if (errorCode) printf("error code %d\n", errorCode);
804 else printf("can be reached at %s:%u (interface %d)", hosttarget, PortAsNumber, ifIndex);
805
806 if (flags) printf(" Flags: %X", flags);
807
808 // Don't show degenerate TXT records containing nothing but a single empty string
809 if (!errorCode && txtLen > 1) { printf("\n"); ShowTXTRecord(txtLen, txtRecord); }
810
811 printf("\n");
812
813 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
814 }
815
816 static void myTimerCallBack(void)
817 {
818 DNSServiceErrorType err = kDNSServiceErr_Unknown;
819
820 switch (operation)
821 {
822 case 'A':
823 {
824 switch (addtest)
825 {
826 case 0: printf("Adding Test HINFO record\n");
827 err = DNSServiceAddRecord(client, &record, 0, kDNSServiceType_HINFO, sizeof(myhinfoW), &myhinfoW[0], 0);
828 addtest = 1;
829 break;
830 case 1: printf("Updating Test HINFO record\n");
831 err = DNSServiceUpdateRecord(client, record, 0, sizeof(myhinfoX), &myhinfoX[0], 0);
832 addtest = 2;
833 break;
834 case 2: printf("Removing Test HINFO record\n");
835 err = DNSServiceRemoveRecord(client, record, 0);
836 addtest = 0;
837 break;
838 }
839 }
840 break;
841
842 case 'U':
843 {
844 if (updatetest[1] != 'Z') updatetest[1]++;
845 else updatetest[1] = 'A';
846 // The following line toggles the string length between 1 and 2 characters.
847 updatetest[0] = 3 - updatetest[0];
848 updatetest[2] = updatetest[1];
849 printtimestamp();
850 printf("Updating Test TXT record to %c\n", updatetest[1]);
851 err = DNSServiceUpdateRecord(client, NULL, 0, 1+updatetest[0], &updatetest[0], 0);
852 }
853 break;
854
855 case 'N':
856 {
857 printf("Adding big NULL record\n");
858 err = DNSServiceAddRecord(client, &record, 0, kDNSServiceType_NULL, sizeof(bigNULL), &bigNULL[0], 0);
859 if (err) printf("Failed: %d\n", err);else printf("Succeeded\n");
860 timeOut = LONG_TIME;
861 #if _DNS_SD_LIBDISPATCH
862 if (timer_source)
863 dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
864 (uint64_t)timeOut * NSEC_PER_SEC, 0);
865 #endif
866 }
867 break;
868 }
869
870 if (err != kDNSServiceErr_NoError)
871 {
872 fprintf(stderr, "DNSService add/update/remove failed %ld\n", (long int)err);
873 stopNow = 1;
874 }
875 }
876
877 static void DNSSD_API reg_reply(DNSServiceRef sdref, const DNSServiceFlags flags, DNSServiceErrorType errorCode,
878 const char *name, const char *regtype, const char *domain, void *context)
879 {
880 (void)sdref; // Unused
881 (void)flags; // Unused
882 (void)context; // Unused
883 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
884
885 printtimestamp();
886 printf("Got a reply for service %s.%s%s: ", name, regtype, domain);
887
888 if (errorCode == kDNSServiceErr_NoError)
889 {
890 if (flags & kDNSServiceFlagsAdd) printf("Name now registered and active\n");
891 else printf("Name registration removed\n");
892 if (operation == 'A' || operation == 'U' || operation == 'N')
893 {
894 timeOut = 5;
895 #if _DNS_SD_LIBDISPATCH
896 if (timer_source)
897 dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
898 (uint64_t)timeOut * NSEC_PER_SEC, 0);
899 #endif
900 }
901 }
902 else if (errorCode == kDNSServiceErr_NameConflict)
903 {
904 printf("Name in use, please choose another\n");
905 exit(-1);
906 }
907 else
908 printf("Error %d\n", errorCode);
909
910 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
911 }
912
913 // Output the wire-format domainname pointed to by rd
914 static int snprintd(char *p, int max, const unsigned char **rd)
915 {
916 const char *const buf = p;
917 const char *const end = p + max;
918 while (**rd)
919 {
920 p += snprintf(p, end-p, "%.*s.", **rd, *rd+1);
921 *rd += 1 + **rd;
922 }
923 *rd += 1; // Advance over the final zero byte
924 return(p-buf);
925 }
926
927 static void ParseDNSSECRecords(uint16_t rrtype, char *rdb, size_t rdb_size, unsigned const char *rd, uint16_t rdlen)
928 {
929 char *p = rdb;
930 switch (rrtype)
931 {
932 case kDNSServiceType_DS:
933 {
934 unsigned char *ptr;
935 int i;
936 rdataDS *rrds = (rdataDS *)rd;
937 p += snprintf(p, rdb + rdb_size - p, "%d %d %d ",
938 rrds->alg, swap16(rrds->keyTag), rrds->digestType);
939 ptr = (unsigned char *)(rd + DS_FIXED_SIZE);
940 for (i = 0; i < (rdlen - DS_FIXED_SIZE); i++)
941 p += snprintf(p, rdb + rdb_size - p, "%x", ptr[i]);
942 break;
943 }
944
945 case kDNSServiceType_DNSKEY:
946 {
947 rdataDNSKey *rrkey = (rdataDNSKey *)rd;
948 p += snprintf(p, rdb + rdb_size - p, "%d %d %d %u ", swap16(rrkey->flags), rrkey->proto,
949 rrkey->alg, (unsigned int)keytag((unsigned char *)rrkey, rdlen));
950 base64Encode(p, rdb + rdb_size - p, (unsigned char *)(rd + DNSKEY_FIXED_SIZE), rdlen - DNSKEY_FIXED_SIZE);
951 break;
952 }
953
954 case kDNSServiceType_NSEC:
955 {
956 unsigned char *next = (unsigned char *)rd;
957 int len, bitmaplen;
958 int win, wlen, type;
959 unsigned char *bmap;
960 char *l = NULL;
961
962 l = p;
963 p += snprintd(p, rdb + rdb_size - p, &rd);
964 len = p - l + 1;
965
966 bitmaplen = rdlen - len;
967 bmap = (unsigned char *)((unsigned char *)next + len);
968
969 while (bitmaplen > 0)
970 {
971 int i;
972
973 if (bitmaplen < 3)
974 {
975 printf("Case NSEC: malformed nsec, bitmaplen %d short\n", bitmaplen);
976 break;
977 }
978
979 win = *bmap++;
980 wlen = *bmap++;
981 bitmaplen -= 2;
982 if (bitmaplen < wlen || wlen < 1 || wlen > 32)
983 {
984 printf("Case NSEC: malformed nsec, bitmaplen %d wlen %d\n", bitmaplen, wlen);
985 break;
986 }
987 if (win < 0 || win >= 256)
988 {
989 printf("Case NSEC: malformed nsec, bad window win %d\n", win);
990 break;
991 }
992 type = win * 256;
993 for (i = 0; i < wlen * 8; i++)
994 {
995 if (bmap[i>>3] & (128 >> (i&7)))
996 p += snprintf(p, rdb + rdb_size - p, " %s ", DNSTypeName(type + i));
997 }
998 bmap += wlen;
999 bitmaplen -= wlen;
1000 }
1001 break;
1002 }
1003
1004 case kDNSServiceType_RRSIG:
1005 {
1006 rdataRRSig *rrsig = (rdataRRSig *)rd;
1007 unsigned char expTimeBuf[64];
1008 unsigned char inceptTimeBuf[64];
1009 unsigned long inceptClock;
1010 unsigned long expClock;
1011 const unsigned char *q = NULL;
1012 char *k = NULL;
1013 int len;
1014
1015 expClock = (unsigned long)swap32(rrsig->sigExpireTime);
1016 FormatTime(expClock, expTimeBuf, sizeof(expTimeBuf));
1017
1018 inceptClock = (unsigned long)swap32(rrsig->sigInceptTime);
1019 FormatTime(inceptClock, inceptTimeBuf, sizeof(inceptTimeBuf));
1020
1021 p += snprintf(p, rdb + rdb_size - p, " %-7s %d %d %d %s %s %7d ",
1022 DNSTypeName(swap16(rrsig->typeCovered)), rrsig->alg, rrsig->labels, swap32(rrsig->origTTL),
1023 expTimeBuf, inceptTimeBuf, swap16(rrsig->keyTag));
1024
1025 q = (const unsigned char *)&rrsig->signerName;
1026 k = p;
1027 p += snprintd(p, rdb + rdb_size - p, &q);
1028 len = p - k + 1;
1029
1030 if ((&rdb[rdb_size] - p) >= 2)
1031 {
1032 *p++ = ' ';
1033 *p = '\0';
1034 }
1035 base64Encode(p, rdb + rdb_size - p, (unsigned char *)(rd + len + RRSIG_FIXED_SIZE), rdlen - (len + RRSIG_FIXED_SIZE));
1036 break;
1037 }
1038 }
1039 return;
1040 }
1041
1042 static void DNSSD_API qr_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t ifIndex, DNSServiceErrorType errorCode,
1043 const char *fullname, uint16_t rrtype, uint16_t rrclass, uint16_t rdlen, const void *rdata, uint32_t ttl, void *context)
1044 {
1045 char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
1046 const unsigned char *rd = rdata;
1047 const unsigned char *end = (const unsigned char *) rdata + rdlen;
1048 char rdb[1000] = "0.0.0.0", *p = rdb;
1049 int unknowntype = 0;
1050 char dnssec_status[15] = "Unknown";
1051 char rr_type[RR_TYPE_SIZE];
1052 char rr_class[6];
1053 DNSServiceFlags check_flags = flags;//local flags for dnssec status checking
1054 int8_t enable_dnssec = ((check_flags & kDNSServiceFlagsEnableDNSSEC) != 0);
1055 static int8_t enabled_dnssec_before = -1;
1056
1057 if (enabled_dnssec_before == -1) {
1058 enabled_dnssec_before = enable_dnssec;
1059 }
1060
1061 (void)sdref; // Unused
1062 (void)ifIndex; // Unused
1063 (void)ttl; // Unused
1064 (void)context; // Unused
1065 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
1066
1067 if (num_printed++ == 0)
1068 {
1069 printf("Timestamp A/R Flags if %-30s%-6s%-7s%s Rdata\n", "Name", "Type", "Class", enable_dnssec ? " DNSSECResult " : "");
1070 }
1071 printtimestamp();
1072
1073 switch (rrclass)
1074 {
1075 case kDNSServiceClass_IN:
1076 strncpy(rr_class, "IN", sizeof(rr_class));
1077 break;
1078 default:
1079 snprintf(rr_class, sizeof(rr_class), "%d", rrclass);
1080 break;
1081 }
1082 strncpy(rr_type, DNSTypeName(rrtype), sizeof(rr_type));
1083
1084 if (!errorCode) //to avoid printing garbage in rdata
1085 {
1086 switch (rrtype)
1087 {
1088 case kDNSServiceType_A:
1089 snprintf(rdb, sizeof(rdb), "%d.%d.%d.%d", rd[0], rd[1], rd[2], rd[3]);
1090 break;
1091
1092 case kDNSServiceType_NS:
1093 case kDNSServiceType_CNAME:
1094 case kDNSServiceType_PTR:
1095 case kDNSServiceType_DNAME:
1096 snprintd(p, sizeof(rdb), &rd);
1097 break;
1098
1099 case kDNSServiceType_SOA:
1100 p += snprintd(p, rdb + sizeof(rdb) - p, &rd); // mname
1101 p += snprintf(p, rdb + sizeof(rdb) - p, " ");
1102 p += snprintd(p, rdb + sizeof(rdb) - p, &rd); // rname
1103 snprintf(p, rdb + sizeof(rdb) - p, " Ser %d Ref %d Ret %d Exp %d Min %d",
1104 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]));
1105 break;
1106
1107 case kDNSServiceType_AAAA:
1108 snprintf(rdb, sizeof(rdb), "%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X",
1109 rd[0x0], rd[0x1], rd[0x2], rd[0x3], rd[0x4], rd[0x5], rd[0x6], rd[0x7],
1110 rd[0x8], rd[0x9], rd[0xA], rd[0xB], rd[0xC], rd[0xD], rd[0xE], rd[0xF]);
1111 break;
1112
1113 case kDNSServiceType_SRV:
1114 p += snprintf(p, rdb + sizeof(rdb) - p, "%d %d %d ", // priority, weight, port
1115 ntohs(*(unsigned short*)rd), ntohs(*(unsigned short*)(rd+2)), ntohs(*(unsigned short*)(rd+4)));
1116 rd += 6;
1117 snprintd(p, rdb + sizeof(rdb) - p, &rd); // target host
1118 break;
1119
1120 case kDNSServiceType_DS:
1121 case kDNSServiceType_DNSKEY:
1122 case kDNSServiceType_NSEC:
1123 case kDNSServiceType_RRSIG:
1124 ParseDNSSECRecords(rrtype, rdb, sizeof(rdb), rd, rdlen);
1125 break;
1126
1127 default:
1128 snprintf(rdb, sizeof(rdb), "%d bytes%s", rdlen, rdlen ? ":" : "");
1129 unknowntype = 1;
1130 break;
1131 }
1132 }
1133
1134 if (check_flags & kDNSServiceFlagsSecure)
1135 strncpy(dnssec_status, "Secure ", sizeof(dnssec_status));
1136 else if (check_flags & kDNSServiceFlagsInsecure)
1137 strncpy(dnssec_status, "Insecure ", sizeof(dnssec_status));
1138 else if (check_flags & kDNSServiceFlagsIndeterminate)
1139 strncpy(dnssec_status, "Indeterminate ", sizeof(dnssec_status));
1140 else if (check_flags & kDNSServiceFlagsBogus)
1141 strncpy(dnssec_status, "Bogus ", sizeof(dnssec_status));
1142 else
1143 strncpy(dnssec_status, " ", sizeof(dnssec_status));
1144
1145 printf("%s%9X%3d %-30s%-7s%-6s %s%s",
1146 op, flags, ifIndex, fullname, rr_type, rr_class, enabled_dnssec_before ? dnssec_status : "", rdb);
1147
1148
1149 if (unknowntype)
1150 {
1151 while (rd < end)
1152 printf(" %02X", *rd++);
1153 }
1154 if (errorCode)
1155 {
1156 if (errorCode == kDNSServiceErr_NoSuchRecord)
1157 printf(" No Such Record");
1158 else if (errorCode == kDNSServiceErr_NoAuth)
1159 printf(" No Authorization");
1160 else if (errorCode == kDNSServiceErr_Timeout)
1161 {
1162 printf(" No Such Record\n");
1163 printf("Query Timed Out\n");
1164 exit(1);
1165 }
1166 }
1167 printf("\n");
1168
1169 if (operation == 'C')
1170 if (flags & kDNSServiceFlagsAdd)
1171 DNSServiceReconfirmRecord(flags, ifIndex, fullname, rrtype, rrclass, rdlen, rdata);
1172
1173 if (!(flags & kDNSServiceFlagsMoreComing))
1174 fflush(stdout);
1175 }
1176
1177 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)
1178 {
1179 (void)sdref; // Unused
1180 (void)flags; // Unused
1181 (void)context; // Unused
1182 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
1183
1184 if (num_printed++ == 0) printf("Timestamp if %-20s %-15s %-15s %-15s %-6s\n", "External Address", "Protocol", "Internal Port", "External Port", "TTL");
1185 printtimestamp();
1186 if (errorCode && errorCode != kDNSServiceErr_DoubleNAT) printf("Error code %d\n", errorCode);
1187 else
1188 {
1189 const unsigned char *digits = (const unsigned char *)&publicAddress;
1190 char addr[256];
1191
1192 snprintf(addr, sizeof(addr), "%d.%d.%d.%d", digits[0], digits[1], digits[2], digits[3]);
1193 printf("%-4d %-20s %-15d %-15d %-15d %-6d%s\n", ifIndex, addr, protocol, ntohs(privatePort), ntohs(publicPort), ttl, errorCode == kDNSServiceErr_DoubleNAT ? " Double NAT" : "");
1194 }
1195
1196 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
1197 }
1198
1199 static void DNSSD_API addrinfo_reply(DNSServiceRef sdref, const DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *hostname, const struct sockaddr *address, uint32_t ttl, void *context)
1200 {
1201 char *op = (flags & kDNSServiceFlagsAdd) ? "Add" : "Rmv";
1202 char addr[256] = "";
1203 char dnssec_status[15] = "Unknown";
1204 DNSServiceFlags check_flags = flags;
1205 (void) sdref;
1206 (void) context;
1207 unsigned char enable_dnssec = ((check_flags & kDNSServiceFlagsEnableDNSSEC) != 0);
1208
1209 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
1210
1211 if (num_printed++ == 0)
1212 {
1213 printf("Timestamp A/R Flags if %-38s %-44s %s%s\n", "Hostname", "Address", "TTL", enable_dnssec ? "DNSSECResult" : "");
1214 }
1215 printtimestamp();
1216
1217 if (address && address->sa_family == AF_INET)
1218 {
1219 const unsigned char *b = (const unsigned char *) &((struct sockaddr_in *)address)->sin_addr;
1220 snprintf(addr, sizeof(addr), "%d.%d.%d.%d", b[0], b[1], b[2], b[3]);
1221 }
1222 else if (address && address->sa_family == AF_INET6)
1223 {
1224 char if_name[IFNAMSIZ]; // Older Linux distributions don't define IF_NAMESIZE
1225 const struct sockaddr_in6 *s6 = (const struct sockaddr_in6 *)address;
1226 const unsigned char *b = (const unsigned char * )&s6->sin6_addr;
1227 if (!if_indextoname(s6->sin6_scope_id, if_name))
1228 snprintf(if_name, sizeof(if_name), "<%d>", s6->sin6_scope_id);
1229 snprintf(addr, sizeof(addr), "%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X:%02X%02X%%%s",
1230 b[0x0], b[0x1], b[0x2], b[0x3], b[0x4], b[0x5], b[0x6], b[0x7],
1231 b[0x8], b[0x9], b[0xA], b[0xB], b[0xC], b[0xD], b[0xE], b[0xF], if_name);
1232 }
1233
1234 if (enable_dnssec)
1235 {
1236 if (check_flags & kDNSServiceFlagsSecure)
1237 strncpy(dnssec_status, " Secure", sizeof(dnssec_status));
1238 else if (check_flags & kDNSServiceFlagsInsecure)
1239 strncpy(dnssec_status, " Insecure", sizeof(dnssec_status));
1240 else if (check_flags & kDNSServiceFlagsIndeterminate)
1241 strncpy(dnssec_status, " Indeterminate", sizeof(dnssec_status));
1242 else if (check_flags & kDNSServiceFlagsBogus)
1243 strncpy(dnssec_status, " Bogus", sizeof(dnssec_status));
1244 }
1245
1246 printf("%s%9X%3d %-38s %-44s %d%s", op, flags, interfaceIndex, hostname, addr, ttl, enable_dnssec ? dnssec_status : "");
1247
1248 if (errorCode)
1249 {
1250 if (errorCode == kDNSServiceErr_NoSuchRecord)
1251 printf(" No Such Record");
1252 else
1253 printf(" Error code %d", errorCode);
1254 }
1255 printf("\n");
1256
1257 if (!(flags & kDNSServiceFlagsMoreComing))
1258 fflush(stdout);
1259 }
1260
1261 //*************************************************************************************************************
1262 // The main test function
1263
1264 static void HandleEvents(void)
1265 #if _DNS_SD_LIBDISPATCH
1266 {
1267 main_queue = dispatch_get_main_queue();
1268 if (client) DNSServiceSetDispatchQueue(client, main_queue);
1269 if (client_pa) DNSServiceSetDispatchQueue(client_pa, main_queue);
1270 if (operation == 'A' || operation == 'U' || operation == 'N')
1271 {
1272 timer_source = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0, main_queue);
1273 if (timer_source)
1274 {
1275 // Start the timer "timeout" seconds into the future and repeat it every "timeout" seconds
1276 dispatch_source_set_timer(timer_source, dispatch_time(DISPATCH_TIME_NOW, (uint64_t)timeOut * NSEC_PER_SEC),
1277 (uint64_t)timeOut * NSEC_PER_SEC, 0);
1278 dispatch_source_set_event_handler(timer_source, ^{myTimerCallBack();});
1279 dispatch_resume(timer_source);
1280 }
1281 }
1282 dispatch_main();
1283 }
1284 #else
1285 {
1286 int dns_sd_fd = client ? DNSServiceRefSockFD(client ) : -1;
1287 int dns_sd_fd2 = client_pa ? DNSServiceRefSockFD(client_pa) : -1;
1288 int nfds = dns_sd_fd + 1;
1289 fd_set readfds;
1290 struct timeval tv;
1291 int result;
1292
1293 if (dns_sd_fd2 > dns_sd_fd) nfds = dns_sd_fd2 + 1;
1294
1295 while (!stopNow)
1296 {
1297 // 1. Set up the fd_set as usual here.
1298 // This example client has no file descriptors of its own,
1299 // but a real application would call FD_SET to add them to the set here
1300 FD_ZERO(&readfds);
1301
1302 // 2. Add the fd for our client(s) to the fd_set
1303 if (client ) FD_SET(dns_sd_fd, &readfds);
1304 if (client_pa) FD_SET(dns_sd_fd2, &readfds);
1305
1306 // 3. Set up the timeout.
1307 tv.tv_sec = timeOut;
1308 tv.tv_usec = 0;
1309
1310 result = select(nfds, &readfds, (fd_set*)NULL, (fd_set*)NULL, &tv);
1311 if (result > 0)
1312 {
1313 DNSServiceErrorType err = kDNSServiceErr_NoError;
1314 if (client && FD_ISSET(dns_sd_fd, &readfds)) err = DNSServiceProcessResult(client );
1315 else if (client_pa && FD_ISSET(dns_sd_fd2, &readfds)) err = DNSServiceProcessResult(client_pa);
1316 if (err) { printtimestamp_F(stderr); fprintf(stderr, "DNSServiceProcessResult returned %d\n", err); stopNow = 1; }
1317 }
1318 else if (result == 0)
1319 myTimerCallBack();
1320 else
1321 {
1322 printf("select() returned %d errno %d %s\n", result, errno, strerror(errno));
1323 if (errno != EINTR) stopNow = 1;
1324 }
1325 }
1326 }
1327 #endif
1328
1329 static int getfirstoption(int argc, char **argv, const char *optstr, int *pOptInd)
1330 // Return the recognized option in optstr and the option index of the next arg.
1331 #if NOT_HAVE_GETOPT
1332 {
1333 int i;
1334 for (i=1; i < argc; i++)
1335 {
1336 if (argv[i][0] == '-' && &argv[i][1] &&
1337 NULL != strchr(optstr, argv[i][1]))
1338 {
1339 *pOptInd = i + 1;
1340 return argv[i][1];
1341 }
1342 }
1343 return -1;
1344 }
1345 #else
1346 {
1347 int o = getopt(argc, (char *const *)argv, optstr);
1348 *pOptInd = optind;
1349 return o;
1350 }
1351 #endif
1352
1353 static void DNSSD_API MyRegisterRecordCallback(DNSServiceRef service, DNSRecordRef rec, const DNSServiceFlags flags,
1354 DNSServiceErrorType errorCode, void *context)
1355 {
1356 char *name = (char *)context;
1357
1358 (void)service; // Unused
1359 (void)rec; // Unused
1360 (void)flags; // Unused
1361 EXIT_IF_LIBDISPATCH_FATAL_ERROR(errorCode);
1362
1363 printtimestamp();
1364 printf("Got a reply for record %s: ", name);
1365
1366 switch (errorCode)
1367 {
1368 case kDNSServiceErr_NoError: printf("Name now registered and active\n"); break;
1369 case kDNSServiceErr_NameConflict: printf("Name in use, please choose another\n"); exit(-1);
1370 default: printf("Error %d\n", errorCode); break;
1371 }
1372 if (!(flags & kDNSServiceFlagsMoreComing)) fflush(stdout);
1373 }
1374
1375 static void getip(const char *const name, struct sockaddr_storage *result)
1376 {
1377 struct addrinfo *addrs = NULL;
1378 int err = getaddrinfo(name, NULL, NULL, &addrs);
1379 if (err) fprintf(stderr, "getaddrinfo error %d for %s", err, name);
1380 else memcpy(result, addrs->ai_addr, SA_LEN(addrs->ai_addr));
1381 if (addrs) freeaddrinfo(addrs);
1382 }
1383
1384 static DNSServiceErrorType RegisterProxyAddressRecord(DNSServiceRef sdref, const char *host, const char *ip, DNSServiceFlags flags)
1385 {
1386 // Call getip() after the call DNSServiceCreateConnection().
1387 // On the Win32 platform, WinSock must be initialized for getip() to succeed.
1388 // Any DNSService* call will initialize WinSock for us, so we make sure
1389 // DNSServiceCreateConnection() is called before getip() is.
1390 struct sockaddr_storage hostaddr;
1391 memset(&hostaddr, 0, sizeof(hostaddr));
1392 getip(ip, &hostaddr);
1393 flags |= kDNSServiceFlagsUnique;
1394 if (hostaddr.ss_family == AF_INET)
1395 return(DNSServiceRegisterRecord(sdref, &record, flags, opinterface, host,
1396 kDNSServiceType_A, kDNSServiceClass_IN, 4, &((struct sockaddr_in *)&hostaddr)->sin_addr, 240, MyRegisterRecordCallback, (void*)host));
1397 else if (hostaddr.ss_family == AF_INET6)
1398 return(DNSServiceRegisterRecord(sdref, &record, flags, opinterface, host,
1399 kDNSServiceType_AAAA, kDNSServiceClass_IN, 16, &((struct sockaddr_in6*)&hostaddr)->sin6_addr, 240, MyRegisterRecordCallback, (void*)host));
1400 else return(kDNSServiceErr_BadParam);
1401 }
1402
1403 #define HexVal(X) ( ((X) >= '0' && (X) <= '9') ? ((X) - '0' ) : \
1404 ((X) >= 'A' && (X) <= 'F') ? ((X) - 'A' + 10) : \
1405 ((X) >= 'a' && (X) <= 'f') ? ((X) - 'a' + 10) : 0)
1406
1407 #define HexPair(P) ((HexVal((P)[0]) << 4) | HexVal((P)[1]))
1408
1409 #define MAXTXTRecordSize 8900
1410 static DNSServiceErrorType RegisterService(DNSServiceRef *sdref,
1411 const char *nam, const char *typ, const char *dom, const char *host, const char *port, int argc, char **argv, DNSServiceFlags flags)
1412 {
1413 uint16_t PortAsNumber = atoi(port);
1414 Opaque16 registerPort = { { PortAsNumber >> 8, PortAsNumber & 0xFF } };
1415 unsigned char txt[MAXTXTRecordSize];
1416 txt[0] = '\0';
1417 unsigned char *ptr = txt;
1418 int i;
1419
1420 if (nam[0] == '.' && nam[1] == 0) nam = ""; // We allow '.' on the command line as a synonym for empty string
1421 if (dom[0] == '.' && dom[1] == 0) dom = ""; // We allow '.' on the command line as a synonym for empty string
1422
1423 printf("Registering Service %s.%s%s%s", nam[0] ? nam : "<<Default>>", typ, dom[0] ? "." : "", dom);
1424 if (host && *host) printf(" host %s", host);
1425 printf(" port %s", port);
1426
1427 if (argc)
1428 {
1429 for (i = 0; i < argc; i++)
1430 {
1431 const char *p = argv[i];
1432 if (ptr >= txt + sizeof(txt))
1433 return kDNSServiceErr_BadParam;
1434 *ptr = 0;
1435 while (*p && *ptr < 255)
1436 {
1437 if (ptr + 1 + *ptr >= txt + sizeof(txt))
1438 return kDNSServiceErr_BadParam;
1439 if (p[0] != '\\' || p[1] == 0) { ptr[++*ptr] = *p; p+=1; }
1440 else if (p[1] == 'x' && isxdigit(p[2]) && isxdigit(p[3])) { ptr[++*ptr] = HexPair(p+2); p+=4; }
1441 else { ptr[++*ptr] = p[1]; p+=2; }
1442 }
1443 ptr += 1 + *ptr;
1444 }
1445 printf(" TXT");
1446 ShowTXTRecord(ptr-txt, txt);
1447 }
1448 printf("\n");
1449
1450 //flags |= kDNSServiceFlagsAllowRemoteQuery;
1451 //flags |= kDNSServiceFlagsNoAutoRename;
1452
1453 return(DNSServiceRegister(sdref, flags, opinterface, nam, typ, dom, host, registerPort.NotAnInteger, (uint16_t) (ptr-txt), txt, reg_reply, NULL));
1454 }
1455
1456 #define TypeBufferSize 80
1457 static char *gettype(char *buffer, char *typ)
1458 {
1459 if (!typ || !*typ || (typ[0] == '.' && typ[1] == 0)) typ = "_http._tcp";
1460 if (!strchr(typ, '.')) { snprintf(buffer, TypeBufferSize, "%s._tcp", typ); typ = buffer; }
1461 return(typ);
1462 }
1463
1464 // Do some basic tests to verify API handles > 63 byte strings gracefully with
1465 // a returned error code.
1466
1467 #define STRING_64_BYTES "_123456789012345678901234567890123456789012345678901234567890123"
1468
1469 static int API_string_limit_test()
1470 {
1471 const char * regtype;
1472 DNSServiceRef sdRef = NULL;
1473 const char * longHost = STRING_64_BYTES ".local";
1474 const char * longDomain = "hostname." STRING_64_BYTES;
1475
1476 printf("Testing for error returns when various strings are > 63 bytes.\n");
1477
1478 printf("DNSServiceGetAddrInfo(), hostname = %s\n", longHost);
1479 if (DNSServiceGetAddrInfo(&sdRef, 0, 0, 0, longHost, addrinfo_reply, 0) == 0)
1480 {
1481 printf("DNSServiceGetAddrInfo(): expected error return\n");
1482 return 1;
1483 };
1484
1485 printf("DNSServiceGetAddrInfo(), hostname = %s\n", longDomain);
1486 if (DNSServiceGetAddrInfo(&sdRef, 0, 0, 0, longDomain, addrinfo_reply, 0) == 0)
1487 {
1488 printf("DNSServiceGetAddrInfo(): expected error return\n");
1489 return 1;
1490 };
1491
1492 printf("DNSServiceResolve(), name = %s\n", STRING_64_BYTES);
1493 if (DNSServiceResolve(&sdRef, 0, 0, STRING_64_BYTES, "_test._tcp", "local", resolve_reply, NULL) == 0)
1494 {
1495 printf("DNSServiceResolve(): expected error return\n");
1496 return 1;
1497 };
1498
1499 regtype = STRING_64_BYTES "._tcp";
1500 printf("DNSServiceResolve(), regtype = %s\n", regtype);
1501 if (DNSServiceResolve(&sdRef, 0, 0, "instanceName", regtype, "local", resolve_reply, NULL) == 0)
1502 {
1503 printf("DNSServiceResolve(): expected error return\n");
1504 return 1;
1505 };
1506
1507 printf("DNSServiceResolve(), domain = %s\n", STRING_64_BYTES);
1508 if (DNSServiceResolve(&sdRef, 0, 0, "instanceName", "_test._tcp", STRING_64_BYTES, resolve_reply, NULL) == 0)
1509 {
1510 printf("DNSServiceResolve(): expected error return\n");
1511 return 1;
1512 };
1513
1514 printf("Testing for error returns when various strings are > 63 bytes: PASSED\n");
1515 return 0;
1516 }
1517
1518 static int API_NULL_input_test()
1519 {
1520 printf("Running basic API input range tests with various pointer parameters set to NULL:\n");
1521
1522 // Test that API's handle NULL pointers by returning an error when appropriate.
1523
1524 // DNSServiceRefSockFD()
1525 if (DNSServiceRefSockFD(0) != -1)
1526 {
1527 printf("DNSServiceRefSockFD(): expected dnssd_InvalidSocket return\n");
1528 return 1;
1529 }
1530
1531 // DNSServiceProcessResult()
1532 if (DNSServiceProcessResult(0) == 0)
1533 {
1534 printf("DNSServiceProcessResult(): expected error return\n");
1535 return 1;
1536 }
1537
1538 // DNSServiceRefDeallocate(): no return value, just verify it doesn't crash
1539 DNSServiceRefDeallocate(0);
1540
1541 // DNSServiceGetProperty()
1542 {
1543 uint32_t result;
1544 uint32_t size;
1545
1546 if ( (DNSServiceGetProperty( 0, &result, &size) == 0)
1547 || (DNSServiceGetProperty(kDNSServiceProperty_DaemonVersion, 0, &size) == 0)
1548 || (DNSServiceGetProperty(kDNSServiceProperty_DaemonVersion, &result, 0) == 0)
1549 )
1550 {
1551 printf("DNSServiceGetProperty(): expected error return\n");
1552 return 1;
1553 }
1554 }
1555
1556 // DNSServiceResolve()
1557 {
1558 DNSServiceRef sdRef;
1559 DNSServiceFlags flags = 0;
1560 uint32_t interfaceIndex = 0;
1561 const char *name = "name";
1562 const char *regtype = "_test._tcp";
1563 const char *domain = "local";
1564 DNSServiceResolveReply callBack = 0;
1565 void *context = 0; // can be a NULL pointer
1566
1567 if ( (DNSServiceResolve( 0, flags, interfaceIndex, name, regtype, domain, callBack, context) == 0)
1568 || (DNSServiceResolve(&sdRef, flags, interfaceIndex, 0, regtype, domain, callBack, context) == 0)
1569 || (DNSServiceResolve(&sdRef, flags, interfaceIndex, name, 0, domain, callBack, context) == 0)
1570 || (DNSServiceResolve(&sdRef, flags, interfaceIndex, name, regtype, 0, callBack, context) == 0)
1571 || (DNSServiceResolve(&sdRef, flags, interfaceIndex, name, regtype, domain, callBack, context) == 0)
1572 )
1573 {
1574 printf("DNSServiceResolve(): expected error return\n");
1575 return 1;
1576 }
1577 }
1578
1579 // DNSServiceQueryRecord()
1580 {
1581 DNSServiceRef sdRef;
1582 DNSServiceFlags flags = 0;
1583 uint32_t interfaceIndex = 0;
1584 const char *fullname = "fullname";
1585 uint16_t rrtype = 0;
1586 uint16_t rrclass = 0;
1587 DNSServiceQueryRecordReply callBack = 0;
1588 void *context = 0; /* may be NULL */
1589
1590 if ( (DNSServiceQueryRecord( 0, flags, interfaceIndex, fullname, rrtype, rrclass, callBack, context) == 0)
1591 || (DNSServiceQueryRecord(&sdRef, flags, interfaceIndex, 0, rrtype, rrclass, callBack, context) == 0)
1592 || (DNSServiceQueryRecord(&sdRef, flags, interfaceIndex, fullname, rrtype, rrclass, 0, context) == 0)
1593 )
1594 {
1595 printf("DNSServiceQueryRecord(): expected error return\n");
1596 return 1;
1597 }
1598 }
1599
1600 // DNSServiceGetAddrInfo()
1601 {
1602 DNSServiceRef sdRef;
1603 DNSServiceFlags flags = 0;
1604 uint32_t interfaceIndex = 0;
1605 DNSServiceProtocol protocol = kDNSServiceProtocol_IPv4|kDNSServiceProtocol_IPv6;
1606 const char *hostname = "host.local";
1607 DNSServiceGetAddrInfoReply callBack = 0;
1608 void *context = 0; // may be NULL
1609
1610 if ( (DNSServiceGetAddrInfo( 0, flags, interfaceIndex, protocol, hostname, callBack, context) == 0)
1611 || (DNSServiceGetAddrInfo(&sdRef, flags, interfaceIndex, protocol, 0, callBack, context) == 0)
1612 || (DNSServiceGetAddrInfo(&sdRef, flags, interfaceIndex, protocol, hostname, 0, context) == 0)
1613 )
1614 {
1615 printf("DNSServiceGetAddrInfo(): expected error return\n");
1616 return 1;
1617 }
1618 }
1619
1620 // DNSServiceBrowse()
1621 {
1622 DNSServiceRef sdRef;
1623 DNSServiceFlags flags = 0;
1624 uint32_t interfaceIndex = 0;
1625 const char *regtype = "_test._tcp";
1626 const char *domain = 0; /* may be NULL */
1627 DNSServiceBrowseReply callBack = 0;
1628 void *context = 0; /* may be NULL */
1629
1630 if ( (DNSServiceBrowse( 0, flags, interfaceIndex, regtype, domain, callBack, context) == 0)
1631 || (DNSServiceBrowse(&sdRef, flags, interfaceIndex, 0, domain, callBack, context) == 0)
1632 || (DNSServiceBrowse(&sdRef, flags, interfaceIndex, regtype, domain, 0, context) == 0)
1633 )
1634 {
1635 printf("DNSServiceBrowse(): expected error return\n");
1636 return 1;
1637 }
1638 }
1639
1640 #if APPLE_OSX_mDNSResponder
1641 // DNSServiceSetDefaultDomainForUser()
1642 if (DNSServiceSetDefaultDomainForUser(0, 0) == 0)
1643 {
1644 printf("DNSServiceSetDefaultDomainForUser(): expected error return\n");
1645 return 1;
1646 }
1647 #endif
1648
1649 // DNSServiceRegister()
1650 {
1651 DNSServiceRef sdRef;
1652 DNSServiceFlags flags = 0;
1653 uint32_t interfaceIndex = 0;
1654 const char *name = 0; /* may be NULL */
1655 const char *regtype = "_test._tcp";
1656 const char *domain = 0; /* may be NULL */
1657 const char *host = 0; /* may be NULL */
1658 uint16_t port = 0x2211; /* In network byte order */
1659 uint16_t txtLen = 1;
1660 const void *txtRecord = "\0"; /* may be NULL */
1661 DNSServiceRegisterReply callBack = 0; /* may be NULL */
1662 void *context = 0; /* may be NULL */
1663
1664 if ( (DNSServiceRegister( 0, flags, interfaceIndex, name, regtype, domain, host, port, txtLen, txtRecord, callBack, context) == 0)
1665 || (DNSServiceRegister(&sdRef, flags, interfaceIndex, name, 0, domain, host, port, txtLen, txtRecord, callBack, context) == 0)
1666 )
1667 {
1668 printf("DNSServiceRegister(): expected error return\n");
1669 return 1;
1670 }
1671 }
1672
1673 // DNSServiceEnumerateDomains()
1674 {
1675 DNSServiceRef sdRef;
1676 DNSServiceFlags flags = 0;
1677 uint32_t interfaceIndex = 0;
1678 DNSServiceDomainEnumReply callBack = 0;
1679 void *context = 0; /* may be NULL */
1680
1681 if ( (DNSServiceEnumerateDomains( 0, flags, interfaceIndex, callBack, context) == 0)
1682 || (DNSServiceEnumerateDomains(&sdRef, flags, interfaceIndex, 0, context) == 0)
1683 )
1684 {
1685 printf("DNSServiceEnumerateDomains(): expected error return\n");
1686 return 1;
1687 }
1688 }
1689
1690 // DNSServiceCreateConnection()
1691 if (DNSServiceCreateConnection(0) == 0)
1692 {
1693 printf("DNSServiceCreateConnection(): expected error return\n");
1694 return 1;
1695 }
1696
1697 #if APPLE_OSX_mDNSResponder
1698 // DNSServiceCreateDelegateConnection()
1699 if (DNSServiceCreateDelegateConnection(0, 0, 0) == 0)
1700 {
1701 printf("DNSServiceCreateDelegateConnection(): expected error return\n");
1702 return 1;
1703 }
1704 #endif
1705
1706 // DNSServiceRegisterRecord()
1707 {
1708 DNSServiceRef sdRef;
1709 DNSRecordRef RecordRef;
1710 DNSServiceFlags flags = 0;
1711 uint32_t interfaceIndex = 0;
1712 const char *fullname = "test1._test._tcp.local";
1713 uint16_t rrtype = kDNSServiceType_TXT;
1714 uint16_t rrclass = kDNSServiceClass_IN;
1715 uint16_t rdlen = 1;
1716 const void *rdata = "\0";
1717 uint32_t ttl = 0;
1718 DNSServiceRegisterRecordReply callBack = 0;
1719 void *context = 0; /* may be NULL */
1720
1721 // Need an initialize sdRef
1722 if (DNSServiceCreateConnection(&sdRef))
1723 {
1724 printf("DNSServiceCreateConnection(): failed\n");
1725 return 1;
1726 }
1727
1728 if ( (DNSServiceRegisterRecord( 0, &RecordRef, flags, interfaceIndex, fullname, rrtype, rrclass, rdlen, rdata, ttl, callBack, context) == 0)
1729 || (DNSServiceRegisterRecord(sdRef, &RecordRef, flags, interfaceIndex, 0, rrtype, rrclass, rdlen, rdata, ttl, callBack, context) == 0)
1730 || (DNSServiceRegisterRecord(sdRef, &RecordRef, flags, interfaceIndex, fullname, rrtype, rrclass, rdlen, 0, ttl, callBack, context) == 0)
1731 || (DNSServiceRegisterRecord(sdRef, &RecordRef, flags, interfaceIndex, fullname, rrtype, rrclass, rdlen, rdata, ttl, 0, context) == 0)
1732 )
1733 {
1734 printf("DNSServiceRegisterRecord(): expected error return\n");
1735 return 1;
1736 }
1737 }
1738
1739 // DNSServiceAddRecord(), DNSServiceUpdateRecord(), and DNSServiceRemoveRecord() verify that they
1740 // get a valid DNSServiceRef returned from DNSServiceRegister()
1741 {
1742 DNSServiceErrorType err;
1743 Opaque16 registerPort = { { 0x12, 0x34 } };
1744 static const char TXT[] = "\xC" "First String";
1745 DNSServiceRef sdRef;
1746
1747 DNSRecordRef RecordRef;
1748 DNSServiceFlags flags = 0;
1749 uint16_t rrtype = kDNSServiceType_TXT;
1750 uint16_t rdlen = 1;
1751 const void *rdata = "\0";
1752 uint32_t ttl = 100;
1753
1754 err = DNSServiceRegister(&sdRef, 0, 0, "Test", "_test._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT)-1, TXT, reg_reply, NULL);
1755 if (err)
1756 {
1757 printf("DNSServiceRegister() failed with: %d\n", err);
1758 return 1;
1759 }
1760
1761 // DNSServiceAddRecord()
1762 if ( (DNSServiceAddRecord( 0, &RecordRef, flags, rrtype, rdlen, rdata, ttl) == 0)
1763 || (DNSServiceAddRecord(sdRef, 0, flags, rrtype, rdlen, rdata, ttl) == 0)
1764 || (DNSServiceAddRecord(sdRef, &RecordRef, flags, rrtype, rdlen, 0, ttl) == 0)
1765 )
1766
1767 {
1768 printf("DNSServiceAddRecord(): expected error return\n");
1769 return 1;
1770 }
1771
1772 // (rdlen == 0 && rdata == 0) should indicate a TXT with rdata containing only a 0 length byte.
1773 if (DNSServiceAddRecord(sdRef, &RecordRef, flags, rrtype, 0, 0, ttl) == kDNSServiceErr_BadParam)
1774 {
1775 printf("DNSServiceAddRecord(): with (rdlen == 0 && rdata == 0) returned kDNSServiceErr_BadParam\n");
1776 return 1;
1777 }
1778
1779 // DNSServiceUpdateRecord()
1780 // Note, RecordRef can be NULL per explanation with declaration in dns_sd.h
1781 if ( (DNSServiceUpdateRecord( 0, RecordRef, flags, rdlen, rdata, ttl) == 0)
1782 || (DNSServiceUpdateRecord(sdRef, RecordRef, flags, rdlen, 0, ttl) == 0)
1783 )
1784 {
1785 printf("DNSServiceUpdateRecord(): expected error return\n");
1786 return 1;
1787 }
1788
1789 // (rdlen == 0 && rdata == 0) should indicate a TXT with rdata containing only a 0 length byte.
1790 if (DNSServiceUpdateRecord(sdRef, RecordRef, flags, 0, 0, ttl) == kDNSServiceErr_BadParam)
1791 {
1792 printf("DNSServiceUpdateRecord(): with (rdlen == 0 && rdata == 0) returned kDNSServiceErr_BadParam\n");
1793 return 1;
1794 }
1795
1796 // DNSServiceRemoveRecord()
1797 if ( (DNSServiceRemoveRecord( 0, RecordRef, flags) == 0)
1798 || (DNSServiceRemoveRecord(sdRef, 0, flags) == 0)
1799 )
1800 {
1801 printf("DNSServiceRemoveRecord(): expected error return\n");
1802 return 1;
1803 }
1804
1805 DNSServiceRefDeallocate(sdRef);
1806 }
1807
1808 // DNSServiceReconfirmRecord()
1809 {
1810 DNSServiceFlags flags = 0;
1811 uint32_t interfaceIndex = 0;
1812 const char *fullname = "aaa._test._tcp.local";
1813 uint16_t rrtype = kDNSServiceType_TXT;
1814 uint16_t rrclass = kDNSServiceClass_IN;
1815 uint16_t rdlen = 1;
1816 const void *rdata = "\0";
1817
1818 if ( (DNSServiceReconfirmRecord(flags, interfaceIndex, 0, rrtype, rrclass, rdlen, rdata) == 0)
1819 || (DNSServiceReconfirmRecord(flags, interfaceIndex, fullname, rrtype, rrclass, rdlen, 0) == 0)
1820 )
1821 {
1822 printf("DNSServiceReconfirmRecord(): expected error return\n");
1823 return 1;
1824 }
1825 // (rdlen == 0 && rdata == 0) should indicate a TXT with rdata containing only a 0 length byte.
1826 if (DNSServiceReconfirmRecord(flags, interfaceIndex, fullname, rrtype, rrclass, 0, 0) == kDNSServiceErr_BadParam)
1827 {
1828 printf("DNSServiceReconfirmRecord(): with (rdlen == 0 && rdata == 0) returned kDNSServiceErr_BadParam\n");
1829 return 1;
1830 }
1831 }
1832
1833
1834 printf("Basic API input range tests: PASSED\n");
1835 return 0;
1836 }
1837
1838 static int API_input_range_test()
1839 {
1840
1841 if (API_string_limit_test())
1842 return 1;
1843
1844 if (API_NULL_input_test())
1845 return 1;
1846
1847 return 0;
1848 }
1849
1850 #ifdef APPLE_OSX_mDNSResponder
1851 static void handle_state_dump_request(uint8_t if_compress_state_dump, uint8_t if_dump_to_stdout);
1852 #endif // APPLE_OSX_mDNSResponder
1853 int main(int argc, char **argv)
1854 {
1855 DNSServiceErrorType err;
1856 char buffer[TypeBufferSize], *typ, *dom;
1857 int opi;
1858 DNSServiceFlags flags = 0;
1859 unsigned char enable_dnssec = 0;
1860
1861 // Extract the program name from argv[0], which by convention contains the path to this executable.
1862 // Note that this is just a voluntary convention, not enforced by the kernel --
1863 // the process calling exec() can pass bogus data in argv[0] if it chooses to.
1864 const char *a0 = strrchr(argv[0], kFilePathSep) + 1;
1865 if (a0 == (const char *)1) a0 = argv[0];
1866
1867 #if defined(_WIN32)
1868 HeapSetInformation(NULL, HeapEnableTerminationOnCorruption, NULL, 0);
1869 #endif
1870
1871 #if TEST_NEW_CLIENTSTUB
1872 printf("Using embedded copy of dnssd_clientstub instead of system library\n");
1873 if (sizeof(argv) == 8) printf("Running in 64-bit mode\n");
1874 #endif
1875
1876 // Test code for TXTRecord functions
1877 //TXTRecordRef txtRecord;
1878 //TXTRecordCreate(&txtRecord, 0, NULL);
1879 //TXTRecordSetValue(&txtRecord, "aaa", 1, "b");
1880 //printf("%d\n", TXTRecordContainsKey(TXTRecordGetLength(&txtRecord), TXTRecordGetBytesPtr(&txtRecord), "Aaa"));
1881
1882 while (argc > 1)
1883 {
1884 int entryCount;
1885
1886 // record current argc to see if we process an argument in this pass
1887 entryCount = argc;
1888
1889 if (argc > 1 && !strcmp(argv[1], "-test"))
1890 {
1891 argc--;
1892 argv++;
1893 return API_input_range_test();
1894 }
1895
1896 if (argc > 1 && !strcmp(argv[1], "-lo"))
1897 {
1898 argc--;
1899 argv++;
1900 opinterface = kDNSServiceInterfaceIndexLocalOnly;
1901 printf("Using LocalOnly\n");
1902 }
1903
1904 if (argc > 1 && (!strcasecmp(argv[1], "-p2p")))
1905 {
1906 argc--;
1907 argv++;
1908 opinterface = kDNSServiceInterfaceIndexP2P;
1909 }
1910
1911 if (argc > 1 && (!strcasecmp(argv[1], "-ble")))
1912 {
1913 argc--;
1914 argv++;
1915 opinterface = kDNSServiceInterfaceIndexBLE;
1916 }
1917
1918 if (argc > 1 && !strcasecmp(argv[1], "-allowexpired"))
1919 {
1920 argc--;
1921 argv++;
1922 flags |= kDNSServiceFlagsAllowExpiredAnswers;
1923 printf("Setting kDNSServiceFlagsAllowExpiredAnswers\n");
1924 }
1925
1926 if (argc > 1 && !strcasecmp(argv[1], "-includep2p"))
1927 {
1928 argc--;
1929 argv++;
1930 flags |= kDNSServiceFlagsIncludeP2P;
1931 printf("Setting kDNSServiceFlagsIncludeP2P\n");
1932 }
1933
1934 if (argc > 1 && !strcasecmp(argv[1], "-fmc"))
1935 {
1936 argc--;
1937 argv++;
1938 flags |= kDNSServiceFlagsForceMulticast;
1939 printf("Setting kDNSServiceFlagsForceMulticast flag for this request\n");
1940 }
1941
1942 if (argc > 1 && !strcasecmp(argv[1], "-includeAWDL"))
1943 {
1944 argc--;
1945 argv++;
1946 flags |= kDNSServiceFlagsIncludeAWDL;
1947 printf("Setting kDNSServiceFlagsIncludeAWDL\n");
1948 }
1949
1950 if (argc > 1 && !strcasecmp(argv[1], "-intermediates"))
1951 {
1952 argc--;
1953 argv++;
1954 flags |= kDNSServiceFlagsReturnIntermediates;
1955 printf("Setting kDNSServiceFlagsReturnIntermediates\n");
1956 }
1957
1958 if (argc > 1 && !strcasecmp(argv[1], "-tc"))
1959 {
1960 argc--;
1961 argv++;
1962 flags |= kDNSServiceFlagsBackgroundTrafficClass;
1963 printf("Setting kDNSServiceFlagsBackgroundTrafficClass\n");
1964 }
1965
1966 if (argc > 1 && !strcasecmp(argv[1], "-t1"))
1967 {
1968 argc--;
1969 argv++;
1970 flags |= kDNSServiceFlagsThresholdOne;
1971 printf("Setting kDNSServiceFlagsThresholdOne\n");
1972 }
1973
1974 if (argc > 1 && !strcasecmp(argv[1], "-tFinder"))
1975 {
1976 argc--;
1977 argv++;
1978 flags |= kDNSServiceFlagsThresholdFinder;
1979 printf("Setting kDNSServiceFlagsThresholdFinder\n");
1980 }
1981
1982 if (argc > 1 && !strcasecmp(argv[1], "-wo"))
1983 {
1984 argc--;
1985 argv++;
1986 flags |= kDNSServiceFlagsWakeOnlyService;
1987 printf("Setting kDNSServiceFlagsWakeOnlyService\n");
1988 }
1989
1990 if (argc > 1 && !strcasecmp(argv[1], "-ku"))
1991 {
1992 argc--;
1993 argv++;
1994 flags |= kDNSServiceFlagsKnownUnique;
1995 printf("Setting kDNSServiceFlagsKnownUnique\n");
1996 }
1997
1998 if (argc > 1 && !strcasecmp(argv[1], "-unicastResponse"))
1999 {
2000 argc--;
2001 argv++;
2002 flags |= kDNSServiceFlagsUnicastResponse;
2003 printf("Setting kDNSServiceFlagsUnicastResponse\n");
2004 }
2005
2006 if (argc > 1 && !strcasecmp(argv[1], "-timeout"))
2007 {
2008 argc--;
2009 argv++;
2010 flags |= kDNSServiceFlagsTimeout;
2011 printf("Setting kDNSServiceFlagsTimeout\n");
2012 }
2013
2014 if (argc > 1 && !strcasecmp(argv[1], "-autoTrigger"))
2015 {
2016 argc--;
2017 argv++;
2018 flags |= kDNSServiceFlagsAutoTrigger;
2019 printf("Setting kDNSServiceFlagsAutoTrigger\n");
2020 }
2021
2022 if (argc > 1 && !strcasecmp(argv[1], "-enableDNSSEC"))
2023 {
2024 argc--;
2025 argv++;
2026 enable_dnssec = 1;
2027 printf("Enable DNSSEC validation for the '-Q' query\n");
2028 }
2029
2030 if (argc > 2 && !strcmp(argv[1], "-i"))
2031 {
2032 opinterface = if_nametoindex(argv[2]);
2033 if (!opinterface) opinterface = atoi(argv[2]);
2034 if (!opinterface) { fprintf(stderr, "Unknown interface %s\n", argv[2]); goto Fail; }
2035 argc -= 2;
2036 argv += 2;
2037 }
2038
2039 // Exit loop if if we didn't match one of the multi character options.
2040 if (argc == entryCount)
2041 break;
2042 }
2043
2044 if (argc < 2) goto Fail; // Minimum command line is the command name and one argument
2045 operation = getfirstoption(argc, argv, "ABCDEFHILMNPQRSTUVZhlq"
2046 "X"
2047 "Gg"
2048 , &opi);
2049 if (operation == -1) goto Fail;
2050
2051 if (opinterface) printf("Using interface %d\n", opinterface);
2052
2053 switch (operation)
2054 {
2055 case 'E': printf("Looking for recommended registration domains:\n");
2056 err = DNSServiceEnumerateDomains(&client, kDNSServiceFlagsRegistrationDomains, opinterface, enum_reply, NULL);
2057 break;
2058
2059 case 'F': printf("Looking for recommended browsing domains:\n");
2060 err = DNSServiceEnumerateDomains(&client, kDNSServiceFlagsBrowseDomains, opinterface, enum_reply, NULL);
2061 //enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "nicta.com.au.", NULL);
2062 //enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "bonjour.nicta.com.au.", NULL);
2063 //enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "ibm.com.", NULL);
2064 //enum_reply(client, kDNSServiceFlagsAdd, 0, 0, "dns-sd.ibm.com.", NULL);
2065 break;
2066
2067 case 'B': typ = (argc < opi+1) ? "" : argv[opi+0];
2068 dom = (argc < opi+2) ? "" : argv[opi+1]; // Missing domain argument is the same as empty string i.e. use system default(s)
2069 typ = gettype(buffer, typ);
2070 if (dom[0] == '.' && dom[1] == 0) dom[0] = 0; // We allow '.' on the command line as a synonym for empty string
2071 printf("Browsing for %s%s%s\n", typ, dom[0] ? "." : "", dom);
2072 err = DNSServiceBrowse(&client, flags, opinterface, typ, dom, browse_reply, NULL);
2073 break;
2074
2075 case 'Z': typ = (argc < opi+1) ? "" : argv[opi+0];
2076 dom = (argc < opi+2) ? "" : argv[opi+1]; // Missing domain argument is the same as empty string i.e. use system default(s)
2077 typ = gettype(buffer, typ);
2078 if (dom[0] == '.' && dom[1] == 0) dom[0] = 0; // We allow '.' on the command line as a synonym for empty string
2079 printf("Browsing for %s%s%s\n", typ, dom[0] ? "." : "", dom);
2080 err = DNSServiceCreateConnection(&client);
2081 if (err) { fprintf(stderr, "DNSServiceCreateConnection returned %d\n", err); return(err); }
2082 sc1 = client;
2083 err = DNSServiceBrowse(&sc1, kDNSServiceFlagsShareConnection, opinterface, typ, dom, zonedata_browse, NULL);
2084 break;
2085
2086 case 'l':
2087 case 'L': {
2088 if (argc < opi+2) goto Fail;
2089 typ = (argc < opi+2) ? "" : argv[opi+1];
2090 dom = (argc < opi+3) ? "local" : argv[opi+2];
2091 typ = gettype(buffer, typ);
2092 if (dom[0] == '.' && dom[1] == 0) dom = "local"; // We allow '.' on the command line as a synonym for "local"
2093 printf("Lookup %s.%s.%s\n", argv[opi+0], typ, dom);
2094 if (operation == 'l') flags |= kDNSServiceFlagsWakeOnResolve;
2095 err = DNSServiceResolve(&client, flags, opinterface, argv[opi+0], typ, dom, resolve_reply, NULL);
2096 break;
2097 }
2098
2099 case 'R': if (argc < opi+4) goto Fail;
2100 typ = (argc < opi+2) ? "" : argv[opi+1];
2101 dom = (argc < opi+3) ? "" : argv[opi+2];
2102 typ = gettype(buffer, typ);
2103 if (dom[0] == '.' && dom[1] == 0) dom[0] = 0; // We allow '.' on the command line as a synonym for empty string
2104 err = RegisterService(&client, argv[opi+0], typ, dom, NULL, argv[opi+3], argc-(opi+4), argv+(opi+4), flags);
2105 break;
2106
2107
2108 case 'P': if (argc < opi+6) goto Fail;
2109 err = DNSServiceCreateConnection(&client_pa);
2110 if (err) { fprintf(stderr, "DNSServiceCreateConnection returned %d\n", err); return(err); }
2111 err = RegisterProxyAddressRecord(client_pa, argv[opi+4], argv[opi+5], flags);
2112 if (err) break;
2113 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);
2114 break;
2115
2116 case 'q':
2117 case 'Q':
2118 case 'C': {
2119 uint16_t rrtype, rrclass;
2120 flags |= kDNSServiceFlagsReturnIntermediates;
2121 if (operation == 'q')
2122 flags |= kDNSServiceFlagsSuppressUnusable;
2123 if (enable_dnssec)
2124 flags |= kDNSServiceFlagsEnableDNSSEC;
2125 if (argc < opi+1)
2126 goto Fail;
2127 rrtype = (argc <= opi+1) ? kDNSServiceType_A : GetRRType(argv[opi+1]);
2128 rrclass = (argc <= opi+2) ? kDNSServiceClass_IN : GetRRClass(argv[opi+2]);
2129 if (rrtype == kDNSServiceType_TXT || rrtype == kDNSServiceType_PTR)
2130 flags |= kDNSServiceFlagsLongLivedQuery;
2131 err = DNSServiceQueryRecord(&client, flags, opinterface, argv[opi+0], rrtype, rrclass, qr_reply, NULL);
2132 break;
2133 }
2134
2135 case 'A':
2136 case 'U':
2137 case 'N': {
2138 Opaque16 registerPort = { { 0x12, 0x34 } };
2139 static const char TXT[] = "\xC" "First String" "\xD" "Second String" "\xC" "Third String";
2140 printf("Registering Service Test._testupdate._tcp.local.\n");
2141 err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testupdate._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT)-1, TXT, reg_reply, NULL);
2142 break;
2143 }
2144
2145 case 'T': {
2146 Opaque16 registerPort = { { 0x23, 0x45 } };
2147 char TXT[1024];
2148 unsigned int i;
2149 for (i=0; i<sizeof(TXT); i++)
2150 if ((i & 0x1F) == 0) TXT[i] = 0x1F;else TXT[i] = 'A' + (i >> 5);
2151 printf("Registering Service Test._testlargetxt._tcp.local.\n");
2152 err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testlargetxt._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT), TXT, reg_reply, NULL);
2153 break;
2154 }
2155
2156 case 'M': {
2157 pid_t pid = getpid();
2158 Opaque16 registerPort = { { pid >> 8, pid & 0xFF } };
2159 static const char TXT1[] = "\xC" "First String" "\xD" "Second String" "\xC" "Third String";
2160 static const char TXT2[] = "\xD" "Fourth String" "\xC" "Fifth String" "\xC" "Sixth String";
2161 printf("Registering Service Test._testdualtxt._tcp.local.\n");
2162 err = DNSServiceRegister(&client, flags, opinterface, "Test", "_testdualtxt._tcp.", "", NULL, registerPort.NotAnInteger, sizeof(TXT1)-1, TXT1, reg_reply, NULL);
2163 if (!err) err = DNSServiceAddRecord(client, &record, flags, kDNSServiceType_TXT, sizeof(TXT2)-1, TXT2, 0);
2164 break;
2165 }
2166
2167 case 'I': {
2168 pid_t pid = getpid();
2169 Opaque16 registerPort = { { pid >> 8, pid & 0xFF } };
2170 static const char TXT[] = "\x09" "Test Data";
2171 printf("Registering Service Test._testtxt._tcp.local.\n");
2172 err = DNSServiceRegister(&client, 0, opinterface, "Test", "_testtxt._tcp.", "", NULL, registerPort.NotAnInteger, 0, NULL, reg_reply, NULL);
2173 if (!err) err = DNSServiceUpdateRecord(client, NULL, 0, sizeof(TXT)-1, TXT, 0);
2174 break;
2175 }
2176
2177 case 'X': {
2178 if (argc == opi) // If no arguments, just fetch IP address
2179 err = DNSServiceNATPortMappingCreate(&client, 0, 0, 0, 0, 0, 0, port_mapping_create_reply, NULL);
2180 else if (argc >= opi+2 && atoi(argv[opi+0]) == 0)
2181 {
2182 DNSServiceProtocol prot = GetProtocol(argv[opi+0]); // Must specify TCP or UDP
2183 uint16_t IntPortAsNumber = atoi(argv[opi+1]); // Must specify internal port
2184 uint16_t ExtPortAsNumber = (argc < opi+3) ? 0 : atoi(argv[opi+2]); // Optional desired external port
2185 uint32_t ttl = (argc < opi+4) ? 0 : atoi(argv[opi+3]); // Optional desired lease lifetime
2186 Opaque16 intp = { { IntPortAsNumber >> 8, IntPortAsNumber & 0xFF } };
2187 Opaque16 extp = { { ExtPortAsNumber >> 8, ExtPortAsNumber & 0xFF } };
2188 err = DNSServiceNATPortMappingCreate(&client, 0, 0, prot, intp.NotAnInteger, extp.NotAnInteger, ttl, port_mapping_create_reply, NULL);
2189 }
2190 else goto Fail;
2191 break;
2192 }
2193
2194 case 'G': {
2195 flags |= kDNSServiceFlagsReturnIntermediates;
2196
2197 if (argc != opi+2)
2198 goto Fail;
2199 else
2200 err = DNSServiceGetAddrInfo(&client, flags, opinterface, GetProtocol(argv[opi+0]), argv[opi+1], addrinfo_reply, NULL);
2201 break;
2202 }
2203
2204 case 'S': {
2205 Opaque16 registerPort = { { 0x23, 0x45 } }; // 9029 decimal
2206 unsigned char txtrec[16] = "\xF" "/path=test.html";
2207 DNSRecordRef rec;
2208 unsigned char nulrec[4] = "1234";
2209
2210 err = DNSServiceCreateConnection(&client);
2211 if (err) { fprintf(stderr, "DNSServiceCreateConnection failed %ld\n", (long int)err); return (-1); }
2212
2213 sc1 = client;
2214 err = DNSServiceBrowse(&sc1, kDNSServiceFlagsShareConnection, opinterface, "_http._tcp", "", browse_reply, NULL);
2215 if (err) { fprintf(stderr, "DNSServiceBrowse _http._tcp failed %ld\n", (long int)err); return (-1); }
2216
2217 sc2 = client;
2218 err = DNSServiceBrowse(&sc2, kDNSServiceFlagsShareConnection, opinterface, "_ftp._tcp", "", browse_reply, NULL);
2219 if (err) { fprintf(stderr, "DNSServiceBrowse _ftp._tcp failed %ld\n", (long int)err); return (-1); }
2220
2221 sc3 = client;
2222 err = DNSServiceRegister(&sc3, kDNSServiceFlagsShareConnection, opinterface, "kDNSServiceFlagsShareConnection",
2223 "_http._tcp", "local", NULL, registerPort.NotAnInteger, 0, NULL, reg_reply, NULL);
2224 if (err) { fprintf(stderr, "SharedConnection DNSServiceRegister failed %ld\n", (long int)err); return (-1); }
2225
2226 err = DNSServiceUpdateRecord(sc3, NULL, 0, sizeof(txtrec), txtrec, 0);
2227 if (err) { fprintf(stderr, "SharedConnection DNSServiceUpdateRecord failed %ld\n", (long int)err); return (-1); }
2228
2229 err = DNSServiceAddRecord(sc3, &rec, 0, kDNSServiceType_NULL, sizeof(nulrec), nulrec, 0);
2230 if (err) { fprintf(stderr, "SharedConnection DNSServiceAddRecord failed %ld\n", (long int)err); return (-1); }
2231
2232 err = DNSServiceRemoveRecord(sc3, rec, 0);
2233 if (err) { fprintf(stderr, "SharedConnection DNSServiceRemoveRecord failed %ld\n", (long int)err); return (-1); }
2234
2235 break;
2236 }
2237
2238 case 'V': {
2239 uint32_t v;
2240 uint32_t size = sizeof(v);
2241 err = DNSServiceGetProperty(kDNSServiceProperty_DaemonVersion, &v, &size);
2242 if (err) fprintf(stderr, "DNSServiceGetProperty failed %ld\n", (long int)err);
2243 else printf("Currently running daemon (system service) is version %d.%d.%d\n", v / 10000, v / 100 % 100, v % 100);
2244 exit(0);
2245 }
2246 #ifdef APPLE_OSX_mDNSResponder
2247 case 'O': {
2248 // check if the user specifies the flag "-compress"
2249 uint8_t if_compress_state_dump = 0;
2250 uint8_t if_dump_to_stdout = 0;
2251
2252 if (argc > opi+1) {
2253 printf("dns-sd: illegal option count\n");
2254 goto Fail;
2255 }
2256
2257 if (argc == opi+1) {
2258 const char *param = argv[opi];
2259 if (strcasecmp("-compress", param) == 0) {
2260 if_compress_state_dump = 1;
2261 } else if (strcasecmp("-stdout", param) == 0) {
2262 if_dump_to_stdout = 1;
2263 } else {
2264 printf("dns-sd: illegal option %s \n", param);
2265 goto Fail;
2266 }
2267 }
2268 handle_state_dump_request(if_compress_state_dump, if_dump_to_stdout);
2269 err = kDNSServiceErr_NoError;
2270 break;
2271 }
2272 #endif // APPLE_OSX_mDNSResponder
2273
2274 case 'H': goto Fail;
2275
2276 default: goto Fail;
2277 }
2278 #ifdef APPLE_OSX_mDNSResponder
2279 // state dump does not need to create DNSServiceRef, so we can return directly here without cleaning up.
2280 if (operation == 'O')
2281 return 0;
2282 #endif // APPLE_OSX_mDNSResponder
2283
2284 if (!client || err != kDNSServiceErr_NoError)
2285 {
2286 fprintf(stderr, "DNSService call failed %ld%s\n", (long int)err,
2287 (err == kDNSServiceErr_ServiceNotRunning) ? " (Service Not Running)" : "");
2288 return (-1);
2289 }
2290 printtimestamp();
2291 printf("...STARTING...\n");
2292 HandleEvents();
2293
2294 // Be sure to deallocate the DNSServiceRef when you're finished
2295 if (client ) DNSServiceRefDeallocate(client );
2296 if (client_pa) DNSServiceRefDeallocate(client_pa);
2297 return 0;
2298
2299 Fail:
2300 if (operation == 'H') print_usage(a0,1);
2301 else print_usage(a0,0);
2302 return 0;
2303 }
2304
2305 #ifdef APPLE_OSX_mDNSResponder
2306 /*
2307 * if_compress_state_dump and if_dump_to_stdout cannot be set at the same time.
2308 */
2309 static void handle_state_dump_request(uint8_t if_compress_state_dump, uint8_t if_dump_to_stdout)
2310 {
2311 // create xpc connection to the xpc server for log utility
2312 xpc_connection_t log_utility_connection = xpc_connection_create_mach_service(kDNSLogUtilityService, dispatch_get_main_queue(), XPC_CONNECTION_MACH_SERVICE_PRIVILEGED);
2313 xpc_connection_set_event_handler(log_utility_connection, ^(xpc_object_t event){
2314 printf("Connecting to %s, status: %s\n", kDNSLogUtilityService, xpc_dictionary_get_string(event, XPC_ERROR_KEY_DESCRIPTION));
2315 });
2316 xpc_connection_resume(log_utility_connection);
2317
2318 // set option for the state dump
2319 xpc_object_t xpc_dict = xpc_dictionary_create(NULL, NULL, 0);
2320 uint64_t dump_option;
2321 if (if_compress_state_dump) {
2322 dump_option = full_state_with_compression;
2323 }
2324 else if (if_dump_to_stdout) {
2325 // we pass the stdout directly to xpc server
2326 dump_option = full_state_to_stdout;
2327 xpc_dictionary_set_fd(xpc_dict, kDNSStateDumpFD, STDOUT_FILENO);
2328 }
2329 else {
2330 dump_option = full_state;
2331 }
2332
2333 xpc_dictionary_set_uint64(xpc_dict, kDNSStateDump, dump_option);
2334
2335 // send the request and handle the response from xpc server
2336 xpc_connection_send_message_with_reply(log_utility_connection, xpc_dict, dispatch_get_main_queue(), ^(xpc_object_t recv_msg){
2337 xpc_type_t msg_type = xpc_get_type(recv_msg);
2338
2339 if (msg_type != XPC_TYPE_DICTIONARY) {
2340 printf("Received unexpected reply from daemon, error: \"%s\"\nUnexpected reply Contents:\n%s\n",
2341 xpc_dictionary_get_string(recv_msg, XPC_ERROR_KEY_DESCRIPTION), xpc_copy_description(recv_msg));
2342 exit(1);
2343 }
2344
2345 // get the response dictionary
2346 uint32_t return_code = (uint32_t)xpc_dictionary_get_uint64(recv_msg, kDNSDaemonReply);
2347 if (return_code != kDNSMsg_NoError) {
2348 const char *error_description = xpc_dictionary_get_string(recv_msg, kDNSErrorDescription);
2349 printf("XPC service returns error, description: %s\n", error_description);
2350 exit(1);
2351 }
2352
2353 // print the state information returned from the XPC server
2354 if (dump_option != full_state_to_stdout) {
2355 const char *path = xpc_dictionary_get_string(recv_msg, kDNSDumpFilePath);
2356 printf("State Dump Is Saved to: %s\n", path);
2357 }
2358
2359 int64_t time_used = xpc_dictionary_get_int64(recv_msg, kDNSStateDumpTimeUsed);
2360 printf(" Time Used: %" PRId64 " ms\n", time_used);
2361
2362 xpc_release(xpc_dict);
2363 exit(0);
2364 });
2365
2366 dispatch_main();
2367 }
2368 #endif // APPLE_OSX_mDNSResponder
2369
2370 // Note: The C preprocessor stringify operator ('#') makes a string from its argument, without macro expansion
2371 // e.g. If "version" is #define'd to be "4", then STRINGIFY_AWE(version) will return the string "version", not "4"
2372 // To expand "version" to its value before making the string, use STRINGIFY(version) instead
2373 #define STRINGIFY_ARGUMENT_WITHOUT_EXPANSION(s) # s
2374 #define STRINGIFY(s) STRINGIFY_ARGUMENT_WITHOUT_EXPANSION(s)
2375
2376 // NOT static -- otherwise the compiler may optimize it out
2377 // The "@(#) " pattern is a special prefix the "what" command looks for
2378 const char VersionString_SCCS[] = "@(#) dns-sd " STRINGIFY(mDNSResponderVersion) " (" __DATE__ " " __TIME__ ")";
2379
2380 #if _BUILDING_XCODE_PROJECT_
2381 // If the process crashes, then this string will be magically included in the automatically-generated crash log
2382 const char *__crashreporter_info__ = VersionString_SCCS + 5;
2383 asm (".desc ___crashreporter_info__, 0x10");
2384 #endif