//  PortMapperTest.m

 //  MYNetwork

 //

 //  Created by Jens Alfke on 1/4/08.

 //  Copyright 2008 Jens Alfke. All rights reserved.

 //

 #import "MYPortMapper.h"

 #import "IPAddress.h"

 #import "Test.h"

 #import "Logging.h"

 #if DEBUG

 /** A trivial class that just demonstrates how to create a MYPortMapper and listen for changes. */

 @interface MYPortMapperTest : NSObject

 {

     MYPortMapper *_mapper;

 }

 @end

 @implementation MYPortMapperTest

 - (id) init

 {

     self = [super init];

     if( self ) {

         // Create MYPortMapper. This example doesn't open a real socket; just pretend that there's

         // already a TCP socket listening on port 8080. To experiment, you could turn on Web

         // Sharing, then change this port number to 80 -- that will make your computer's local

         // Apache web server reachable from the outside world while this program is running.

         _mapper = [[MYPortMapper alloc] initWithLocalPort: 80];

         // Optionally, request a public port number.

         // The NAT is free to ignore this and return a random number instead.

         _mapper.desiredPublicPort = 22222;

         // Now open the mapping (asynchronously):

         if( [_mapper start] ) {

             Log(@"Opening port mapping...");

             // Now listen for notifications to find out when the mapping opens, fails, or changes:

             [[NSNotificationCenter defaultCenter] addObserver: self 

                                                      selector: @selector(portMappingChanged:) 

                                                          name: MYPortMapperChangedNotification 

                                                        object: _mapper];

         } else {

             // MYPortMapper failed -- this is unlikely, but be graceful:

             Log(@"!! Error: MYPortMapper wouldn't start: %i",_mapper.error);

             [self release];

             return nil;

         }

     }

     return self;

 }

 - (void) portMappingChanged: (NSNotification*)n

 {

     // This is where we get notified that the mapping was created, or that no mapping exists,

     // or that mapping failed.

     if( _mapper.error )

         Log(@"!! MYPortMapper error %i", _mapper.error);

     else {

         NSString *message = @"";

         if( !_mapper.isMapped )

             message = @" (no address translation!)";

         Log(@"** Public address:port is %@%@", _mapper.publicAddress, message);

         Log(@"    local address:port is %@", _mapper.localAddress);

     }

 }

 - (void) dealloc

 {

     [_mapper stop];

     [_mapper release];

     [super dealloc];

 }

 @end

 TestCase(MYPortMapper) {

     EnableLogTo(DNS,YES);

     EnableLogTo(PortMapper,YES);

     // Here's how to simply obtain your local and public address(es):

     IPAddress *addr = [IPAddress localAddress];

     Log(@"** Local address is %@%@ ...getting public addr...", 

         addr, (addr.isPrivate ?@" (private)" :@""));

     addr = [MYPortMapper findPublicAddress];

     Log(@"** Public address is %@", addr);

     // Start up the test class to create a mapping:

     MYPortMapperTest *test = [[MYPortMapperTest alloc] init];

     // Now let the runloop run forever...

     Log(@"Running the runloop forever...");

     [[NSRunLoop currentRunLoop] run];

     [test release];

 }

 #endif DEBUG