//

 //  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)

 {

     @try{

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

         if (errorCode)

             [(MYBonjourQuery*)context setError: errorCode];

         else

             [(MYBonjourQuery*)context priv_gotRecordBytes: rdata

                                                       length: rdlen

                                                         type: rrtype

                                                          ttl: ttl

                                                        flags: flags];

     }catchAndReport(@"MYBonjourResolver query callback");

 }

 - (DNSServiceRef) createServiceRef {

     DNSServiceRef serviceRef = NULL;

     const char *fullName = _bonjourService.fullName.UTF8String;

     if (fullName)

         self.error = DNSServiceQueryRecord(&serviceRef, 0, 

                                            _bonjourService.interfaceIndex, 

                                            fullName,

                                            _recordType, kDNSServiceClass_IN, 

                                            &queryCallback, self);

     return serviceRef;

 }

 @end