--- /dev/null
+/*
+ * Copyright (c) 2003 Apple Computer, Inc. All rights reserved.
+ *
+ * @APPLE_LICENSE_HEADER_START@
+ *
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ *
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ *
+ * @APPLE_LICENSE_HEADER_END@
+
+ Change History (most recent first):
+
+$Log: SampleUDSClient.c,v $
+Revision 1.7 2003/08/18 18:50:15 cheshire
+Can now give "-lo" as first parameter, to test "local only" mode
+
+Revision 1.6 2003/08/12 19:56:25 cheshire
+Update to APSL 2.0
+
+ */
+
+#include <dns_sd.h>
+#include <unistd.h>
+#include <DNSServiceDiscovery/DNSServiceDiscovery.h> // include Mach API to ensure no conflicts exist
+#include <CoreFoundation/CoreFoundation.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#define BIND_8_COMPAT 1
+#include <nameser.h>
+// T_SRV is not defined in older versions of nameser.h
+#ifndef T_SRV
+#define T_SRV 33
+#endif
+
+// constants
+#define MAX_DOMAIN_LABEL 63
+#define MAX_DOMAIN_NAME 255
+#define MAX_CSTRING 2044
+
+
+// data structure defs
+typedef union { unsigned char b[2]; unsigned short NotAnInteger; } Opaque16;
+
+typedef struct { u_char c[ 64]; } domainlabel;
+typedef struct { u_char c[256]; } domainname;
+
+
+typedef struct
+ {
+ uint16_t priority;
+ uint16_t weight;
+ uint16_t port;
+ domainname target;
+ } srv_rdata;
+
+
+// private function prototypes
+static void sighdlr(int signo);
+static char *ConvertDomainNameToCString_withescape(const domainname *const name, char *ptr, char esc);
+static char *ConvertDomainLabelToCString_withescape(const domainlabel *const label, char *ptr, char esc);
+//static void MyCallbackWrapper(CFSocketRef sr, CFSocketCallBackType t, CFDataRef dr, const void *i, void *context);
+static void print_rdata(int type, int len, const u_char *rdata);
+static void query_cb(const DNSServiceRef DNSServiceRef, const DNSServiceFlags flags, const u_int32_t interfaceIndex, const DNSServiceErrorType errorCode, const char *name, const u_int16_t rrtype, const u_int16_t rrclass, const u_int16_t rdlen, const void *rdata, const u_int32_t ttl, void *context);
+static void resolve_cb(const DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *fullname, const char *hosttarget, uint16_t port, uint16_t txtLen, const char *txtRecord, void *context);
+static void my_enum_cb( DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *replyDomain, void *context);
+static void my_regecordcb(DNSServiceRef sdRef, DNSRecordRef RecordRef, DNSServiceFlags flags, DNSServiceErrorType errorCode, void *context);
+static void browse_cb(DNSServiceRef sdr, DNSServiceFlags flags, uint32_t ifi, DNSServiceErrorType err, const char *serviceName, const char *regtype, const char *domain, void *context);
+
+
+// globals
+static DNSServiceRef sdr = NULL;
+static uint32_t InterfaceIndex = 0;
+
+static void regservice_cb(DNSServiceRef sdRef, DNSServiceFlags flags, DNSServiceErrorType errorCode, const char *name, const char *regtype, const char *domain, void *context)
+ {
+ #pragma unused (sdRef, flags, errorCode, context)
+ printf("regservice_cb %s %s %s\n", name, regtype, domain);
+ }
+
+int main (int argc, char * argv[]) {
+ int err, t, i;
+ char *name, *type, *domain;
+ DNSServiceFlags flags;
+ DNSRecordRef recordrefs[10];
+ char host[256];
+ int ipaddr = 12345; // random IP address
+
+ char full[1024];
+
+ // First parameter "-lo" means "local only"
+ if (!strcmp(argv[1], "-lo")) { InterfaceIndex = -1; argv++; argc--; }
+
+ if (signal(SIGINT, sighdlr) == SIG_ERR) fprintf(stderr, "ERROR - can't catch interupt!\n");
+ if (argc < 2) exit(1);
+
+ if (!strcmp(argv[1], "-regrecord"))
+ {
+ err = DNSServiceCreateConnection(&sdr);
+ if (err)
+ {
+ printf("DNSServiceCreateConnection returned %d\n", err);
+ exit(1);
+ }
+ printf("registering 10 address records...\n");
+ for (i = 0; i < 10; i++)
+ {
+ sprintf(host, "testhost-%d.local.", i);
+ ipaddr++;
+ err = DNSServiceRegisterRecord(sdr, &recordrefs[i], kDNSServiceFlagsUnique, InterfaceIndex,
+ host, 1, 1, 4, &ipaddr, 60, my_regecordcb, NULL);
+ if (err)
+ {
+ printf("DNSServiceRegisterRecord returned error %d\n", err);
+ exit(1);
+ }
+ }
+ printf("processing results...\n");
+ for (i = 0; i < 10; i++) DNSServiceProcessResult(sdr);
+ printf("deregistering half of the records\n");
+ for (i = 0; i < 10; i++)
+ {
+ if (i % 2)
+ {
+ err = DNSServiceRemoveRecord(sdr, recordrefs[i], 0);
+ if (err)
+ {
+ printf("DNSServiceRemoveRecord returned error %d\n" ,err);
+ exit(1);
+ }
+ }
+ }
+ printf("sleeping 10...\n");
+ sleep(10);
+ printf("deregistering all remaining records\n");;
+ DNSServiceRefDeallocate(sdr);
+ printf("done. sleeping 10..\n");
+ sleep(10);
+ exit(1);
+ }
+
+ if (!strcmp(argv[1], "-browse"))
+ {
+ if (argc < 3) exit(1);
+ err = DNSServiceBrowse(&sdr, 0, InterfaceIndex, argv[2], NULL /*"local."*/, browse_cb, NULL);
+ if (err)
+ {
+ printf("DNSServiceBrowse returned error %d\n", err);
+ exit(1);
+ }
+ while(1) DNSServiceProcessResult(sdr);
+ }
+
+ if (!strcmp(argv[1], "-enum"))
+ {
+ if (!strcmp(argv[2], "browse")) flags = kDNSServiceFlagsBrowseDomains;
+ else if (!strcmp(argv[2], "register")) flags = kDNSServiceFlagsRegistrationDomains;
+ else exit(1);
+
+ err = DNSServiceEnumerateDomains(&sdr, flags, InterfaceIndex, my_enum_cb, NULL);
+ if (err)
+ {
+ printf("EnumerateDomains returned error %d\n", err);
+ exit(1);
+ }
+ while(1) DNSServiceProcessResult(sdr);
+ }
+ if (!strcmp(argv[1], "-query"))
+ {
+ t = atol(argv[5]);
+ err = DNSServiceConstructFullName(full, argv[2], argv[3], argv[4]);
+ if (err) exit(1);
+ printf("resolving fullname %s type %d\n", full, t);
+ err = DNSServiceQueryRecord(&sdr, 0, 0, full, t, 1, query_cb, NULL);
+ while (1) DNSServiceProcessResult(sdr);
+ }
+
+ if (!strcmp(argv[1], "-regservice"))
+ {
+ char *regtype = "_http._tcp";
+ char txtstring[] = "\x0DMy Txt Record";
+ if (argc > 2) name = argv[2];
+ else name = NULL;
+ if (argc > 3) regtype = argv[3];
+ uint16_t PortAsNumber = 123;
+ if (argc > 4) PortAsNumber = atoi(argv[4]);
+ Opaque16 registerPort = { { PortAsNumber >> 8, PortAsNumber & 0xFF } };
+ err = DNSServiceRegister(&sdr, 0, InterfaceIndex, name, regtype, "local.", NULL, registerPort.NotAnInteger, sizeof(txtstring)-1, txtstring, regservice_cb, NULL);
+ if (err)
+ {
+ printf("DNSServiceRegister returned error %d\n", err);
+ exit(1);
+ }
+ while (1) DNSServiceProcessResult(sdr);
+ }
+ if (!strcmp(argv[1], "-resolve"))
+ {
+ name = argv[2];
+ type = argv[3];
+ domain = argv[4];
+ err = DNSServiceResolve(&sdr, 0, InterfaceIndex, name, type, domain, resolve_cb, NULL);
+ if (err)
+ {
+ printf("DNSServiceResolve returned error %d\n", err);
+ exit(1);
+ }
+ while(1) DNSServiceProcessResult(sdr);
+ }
+ exit(1);
+ }
+
+
+
+// callbacks
+
+// wrapper to make callbacks fit CFRunLoop callback signature
+/*
+static void MyCallbackWrapper(CFSocketRef sr, CFSocketCallBackType t, CFDataRef dr, const void *i, void *context)
+ {
+ (void)sr;
+ (void)t;
+ (void)dr;
+ (void)i;
+
+ DNSServiceRef *sdr = context;
+ DNSServiceDiscoveryProcessResult(*sdr);
+ }
+*/
+
+static void browse_cb(DNSServiceRef sdr, DNSServiceFlags flags, uint32_t ifi, DNSServiceErrorType err, const char *serviceName, const char *regtype, const char *domain, void *context)
+ {
+ #pragma unused(sdr, ifi, context)
+
+ if (err)
+ {
+ printf("Callback: error %d\n", err);
+ return;
+ }
+ printf("BrowseCB: %s %s %s %s (%s)\n", serviceName, regtype, domain, (flags & kDNSServiceFlagsMoreComing ? "(more coming)" : ""), flags & kDNSServiceFlagsAdd ? "(ADD)" : "(REMOVE)");
+
+ }
+
+static void my_enum_cb( DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *replyDomain, void *context)
+ {
+ #pragma unused(sdRef, context)
+ char *type;
+ if (flags == kDNSServiceFlagsAdd) type = "add";
+ else if (flags == kDNSServiceFlagsRemove) type = "remove";
+ else if (flags == (kDNSServiceFlagsAdd | kDNSServiceFlagsDefault)) type = "add default";
+ else type = "unknown";
+
+
+ if (errorCode) printf("EnumerateDomainsCB: error code %d\n", errorCode);
+ else printf("%s domain %s on interface %d\n", type, replyDomain, interfaceIndex);
+ }
+
+static void query_cb(const DNSServiceRef DNSServiceRef, const DNSServiceFlags flags, const u_int32_t interfaceIndex, const DNSServiceErrorType errorCode, const char *name, const u_int16_t rrtype, const u_int16_t rrclass, const u_int16_t rdlen, const void *rdata, const u_int32_t ttl, void *context)
+ {
+ (void)DNSServiceRef;
+ (void)flags;
+ (void)interfaceIndex;
+ (void)rrclass;
+ (void)ttl;
+ (void)context;
+
+ if (errorCode)
+ {
+ printf("query callback: error==%d\n", errorCode);
+ return;
+ }
+ printf("query callback - name = %s, rdata=\n", name);
+ print_rdata(rrtype, rdlen, rdata);
+ }
+
+static void resolve_cb(const DNSServiceRef sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, DNSServiceErrorType errorCode, const char *fullname, const char *hosttarget, uint16_t port, uint16_t txtLen, const char *txtRecord, void *context)
+ {
+ int i;
+
+ #pragma unused(sdRef, flags, interfaceIndex, errorCode, context, txtRecord)
+ printf("Resolved %s to %s:%d (%d bytes txt data)\n", fullname, hosttarget, port, txtLen);
+ printf("TXT Data:\n");
+ for (i = 0; i < txtLen; i++)
+ if (txtRecord[i] >= ' ') printf("%c", txtRecord[i]);
+ }
+
+
+
+static void my_regecordcb(DNSServiceRef sdRef, DNSRecordRef RecordRef, DNSServiceFlags flags, DNSServiceErrorType errorCode, void *context)
+ {
+ #pragma unused (sdRef, RecordRef, flags, context)
+ if (errorCode) printf("regrecord CB received error %d\n", errorCode);
+ else printf("regrecord callback - no errors\n");
+ }
+
+
+// resource record data interpretation routines
+static char *ConvertDomainLabelToCString_withescape(const domainlabel *const label, char *ptr, char esc)
+ {
+ const u_char * src = label->c; // Domain label we're reading
+ const u_char len = *src++; // Read length of this (non-null) label
+ const u_char *const end = src + len; // Work out where the label ends
+ if (len > MAX_DOMAIN_LABEL) return(NULL); // If illegal label, abort
+ while (src < end) // While we have characters in the label
+ {
+ u_char c = *src++;
+ if (esc)
+ {
+ if (c == '.') // If character is a dot,
+ *ptr++ = esc; // Output escape character
+ else if (c <= ' ') // If non-printing ascii,
+ { // Output decimal escape sequence
+ *ptr++ = esc;
+ *ptr++ = (char) ('0' + (c / 100) );
+ *ptr++ = (char) ('0' + (c / 10) % 10);
+ c = (u_char)('0' + (c ) % 10);
+ }
+ }
+ *ptr++ = (char)c; // Copy the character
+ }
+ *ptr = 0; // Null-terminate the string
+ return(ptr); // and return
+ }
+
+static char *ConvertDomainNameToCString_withescape(const domainname *const name, char *ptr, char esc)
+ {
+ const u_char *src = name->c; // Domain name we're reading
+ const u_char *const max = name->c + MAX_DOMAIN_NAME; // Maximum that's valid
+
+ if (*src == 0) *ptr++ = '.'; // Special case: For root, just write a dot
+
+ while (*src) // While more characters in the domain name
+ {
+ if (src + 1 + *src >= max) return(NULL);
+ ptr = ConvertDomainLabelToCString_withescape((const domainlabel *)src, ptr, esc);
+ if (!ptr) return(NULL);
+ src += 1 + *src;
+ *ptr++ = '.'; // Write the dot after the label
+ }
+
+ *ptr++ = 0; // Null-terminate the string
+ return(ptr); // and return
+ }
+
+// print arbitrary rdata in a readable manned
+static void print_rdata(int type, int len, const u_char *rdata)
+ {
+ int i;
+ srv_rdata *srv;
+ char targetstr[MAX_CSTRING];
+ struct in_addr in;
+
+ switch (type)
+ {
+ case T_TXT:
+ // print all the alphanumeric and punctuation characters
+ for (i = 0; i < len; i++)
+ if (rdata[i] >= 32 && rdata[i] <= 127) printf("%c", rdata[i]);
+ printf("\n");
+ return;
+ case T_SRV:
+ srv = (srv_rdata *)rdata;
+ ConvertDomainNameToCString_withescape(&srv->target, targetstr, 0);
+ printf("pri=%d, w=%d, port=%d, target=%s\n", srv->priority, srv->weight, srv->port, targetstr);
+ return;
+ case T_A:
+ assert(len == 4);
+ memcpy(&in, rdata, sizeof(in));
+ printf("%s\n", inet_ntoa(in));
+ return;
+ case T_PTR:
+ ConvertDomainNameToCString_withescape((domainname *)rdata, targetstr, 0);
+ printf("%s\n", targetstr);
+ return;
+ default:
+ printf("ERROR: I dont know how to print RData of type %d\n", type);
+ return;
+ }
+ }
+
+
+
+
+// signal handlers, setup/teardown, etc.
+static void sighdlr(int signo)
+ {
+ assert(signo == SIGINT);
+ fprintf(stderr, "Received sigint - deallocating serviceref and exiting\n");
+ if (sdr)
+ DNSServiceRefDeallocate(sdr);
+ exit(1);
+ }
+
+
+
+
+
+
+
projects / apple / mdnsresponder.git / blobdiff