//

 //  MYBonjourQuery.m

 //  MYNetwork

 //

 //  Created by Jens Alfke on 4/24/09.

 //  Copyright 2009 Jens Alfke. All rights reserved.

 //

 #import "MYBonjourQuery.h"

 #import "MYBonjourService.h"

 #import "Test.h"

 #import "Logging.h"

 #import "ExceptionUtils.h"

 #import <dns_sd.h>

 static NSString* kRecordTypeNames[] = {

     @"0",

     @"A", //         = 1,      /* Host address. */

     @"NS", //        = 2,      /* Authoritative server. */

     @"MD", //        = 3,      /* Mail destination. */

     @"MF", //        = 4,      /* Mail forwarder. */

     @"CNAME", //     = 5,      /* Canonical name. */

     @"SOA", //       = 6,      /* Start of authority zone. */

     @"MB", //        = 7,      /* Mailbox domain name. */

     @"MG", //        = 8,      /* Mail group member. */

     @"MR", //        = 9,      /* Mail rename name. */

     @"NULL", //      = 10,     /* Null resource record. */

     @"WKS", //       = 11,     /* Well known service. */

     @"PTR", //       = 12,     /* Domain name pointer. */

     @"HINFO", //     = 13,     /* Host information. */

     @"MINFO", //     = 14,     /* Mailbox information. */

     @"MX", //        = 15,     /* Mail routing information. */

     @"TXT" //       = 16,     /* One or more text strings (NOT "zero or more..."). */

     // this isn't a complete list; it just includes the most common ones.

     // For the full list, see the "kDNSServiceType_..." constants in <dns_sd.h>.

 };

 @interface MYBonjourQuery ()

 @property (copy) NSData *recordData;

 @end

 @implementation MYBonjourQuery

 - (id) initWithBonjourService: (MYBonjourService*)service recordType: (uint16_t)recordType;

 {

     self = [super init];

     if (self) {

         _bonjourService = service;

         _recordType = recordType;

     }

     return self;

 }

 - (void) dealloc

 {

     [_recordData release];

     [super dealloc];

 }

 - (NSString*) description

 {

     NSString *typeName;

     if (_recordType <= 16)

         typeName = kRecordTypeNames[_recordType];

     else

         typeName = $sprintf(@"%u", _recordType);

     return $sprintf(@"%@[%@ /%@]", self.class, _bonjourService.name, typeName);

 }

 @synthesize recordData=_recordData;

 - (void) priv_gotRecordBytes: (const void *)rdata

                       length: (uint16_t)rdlen

                         type: (uint16_t)rrtype

                          ttl: (uint32_t)ttl

                        flags: (DNSServiceFlags)flags

 {

     NSData *data = [NSData dataWithBytes: rdata length: rdlen];

     if (!$equal(data,_recordData)) {

         if (data.length <= 16)

             LogTo(Bonjour,@"%@ = %@", self, data);

         else

             LogTo(Bonjour,@"%@ = %@...", self, [data subdataWithRange: NSMakeRange(0,16)]);

         self.recordData = data;

     }

     [_bonjourService queryDidUpdate: self];

 }

 static void queryCallback( DNSServiceRef                       DNSServiceRef,

                            DNSServiceFlags                     flags,

                            uint32_t                            interfaceIndex,

                            DNSServiceErrorType                 errorCode,

                            const char                          *fullname,

                            uint16_t                            rrtype,

                            uint16_t                            rrclass,

                            uint16_t                            rdlen,

                            const void                          *rdata,

                            uint32_t                            ttl,

                            void                                *context)

 {

     MYBonjourQuery *query = context;

     @try{

         //LogTo(Bonjour, @"queryCallback for %@ (err=%i)", context,errorCode);

         if (!errorCode)

             [query priv_gotRecordBytes: rdata

                                 length: rdlen

                                   type: rrtype

                                    ttl: ttl

                                  flags: flags];

     }catchAndReport(@"MYBonjourResolver query callback");

     [query gotResponse: errorCode];

 }

 - (DNSServiceErrorType) createServiceRef: (DNSServiceRef*)sdRefPtr {

     const char *fullName = _bonjourService.fullName.UTF8String;

     if (fullName)

         return DNSServiceQueryRecord(sdRefPtr,

                                      kDNSServiceFlagsShareConnection, 

                                      _bonjourService.interfaceIndex, 

                                      fullName,

                                      _recordType, kDNSServiceClass_IN, 

                                      &queryCallback, self);

     else

         return kDNSServiceErr_NoSuchName;

 }

 @end

 /*

  Copyright (c) 2009, Jens Alfke <jens@mooseyard.com>. All rights reserved.

  Redistribution and use in source and binary forms, with or without modification, are permitted

  provided that the following conditions are met:

  * Redistributions of source code must retain the above copyright notice, this list of conditions

  and the following disclaimer.

  * Redistributions in binary form must reproduce the above copyright notice, this list of conditions

  and the following disclaimer in the documentation and/or other materials provided with the

  distribution.

  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR

  IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 

  FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRI-

  BUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 

   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 

  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF 

  THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

  */