//

//  MYAddressLookup.m

//  MYNetwork

//

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

//  Copyright 2009 Jens Alfke. All rights reserved.

//

#import "MYAddressLookup.h"

#import "IPAddress.h"

#import "ExceptionUtils.h"

#import "Test.h"

#import "Logging.h"

#import <dns_sd.h>

@implementation MYAddressLookup

- (id) initWithHostname: (NSString*)hostname

{

    self = [super init];

    if (self != nil) {

        if (!hostname) {

            [self release];

            return nil;

        }

        _hostname = [hostname copy];

        _addresses = [[NSMutableArray alloc] init];

    }

    return self;

}

- (void) dealloc

{

    [_hostname release];

    [_addresses release];

    [super dealloc];

}

- (NSString*) description

{

    return $sprintf(@"%@[%@]", self.class,_hostname);

}

@synthesize port=_port, interfaceIndex=_interfaceIndex, addresses=_addresses;

- (NSTimeInterval) timeToLive {

    return MAX(0.0, _expires - CFAbsoluteTimeGetCurrent());

}

- (void) priv_resolvedAddress: (const struct sockaddr*)sockaddr

                          ttl: (uint32_t)ttl

                        flags: (DNSServiceFlags)flags

{

    HostAddress *address = [[HostAddress alloc] initWithHostname: _hostname 

                                                        sockaddr: sockaddr

                                                            port: _port];

    if (address) {

        if (flags & kDNSServiceFlagsAdd)

            [_addresses addObject: address];

        else

            [_addresses removeObject: address];

        [address release];

    }

    _expires = CFAbsoluteTimeGetCurrent() + ttl;

    if (!(flags & kDNSServiceFlagsMoreComing))

        LogTo(DNS,@"Got addresses of %@: %@ [TTL = %u]", self, _addresses, ttl);

}

static void lookupCallback(DNSServiceRef                    sdRef,

                           DNSServiceFlags                  flags,

                           uint32_t                         interfaceIndex,

                           DNSServiceErrorType              errorCode,

                           const char                       *hostname,

                           const struct sockaddr            *address,

                           uint32_t                         ttl,

                           void                             *context)

{

    MYAddressLookup *lookup = context;

    @try{

        //LogTo(Bonjour, @"lookupCallback for %s (err=%i)", hostname,errorCode);

        if (errorCode)

            [lookup setError: errorCode];

        else

            [lookup priv_resolvedAddress: address ttl: ttl flags: flags];

    }catchAndReport(@"MYDNSLookup query callback");

}

- (DNSServiceRef) createServiceRef {

    [_addresses removeAllObjects];

    DNSServiceRef serviceRef = NULL;

    self.error = DNSServiceGetAddrInfo(&serviceRef, 0,

                                       _interfaceIndex, 0,

                                       _hostname.UTF8String,

                                       &lookupCallback, self);

    return serviceRef;

}

@end

TestCase(MYDNSLookup) {

    EnableLogTo(Bonjour,YES);

    EnableLogTo(DNS,YES);

    [NSRunLoop currentRunLoop]; // create runloop

    MYAddressLookup *lookup = [[MYAddressLookup alloc] initWithHostname: @"www.apple.com" port: 80];

    [lookup start];

    [[NSRunLoop currentRunLoop] runUntilDate: [NSDate dateWithTimeIntervalSinceNow: 10]];

    [lookup release];

}