Merge branch 'master' of git://github.com/ksuther/MenuTunes

[MenuTunes.git] / NetworkController.m

diff --git a/NetworkController.m b/NetworkController.m
index d6bfca0..ac10ab6 100755 (executable)
--- a/NetworkController.m
+++ b/NetworkController.m
@@ -1,21 +1,9 @@
-/*
- *     MenuTunes
- *  NetworkController
- *    Rendezvous network controller
- *
- *  Original Author : Kent Sutherland <ksuther@ithinksw.com>
- *   Responsibility : Kent Sutherland <ksuther@ithinksw.com>
- *
- *  Copyright (c) 2003 iThink Software.
- *  All Rights Reserved
- *
- */
-

 #import "NetworkController.h"
 #import "MainController.h"
 #import "NetworkController.h"
 #import "MainController.h"

+#import "NetworkObject.h"
+#import "PreferencesController.h"

 #import <ITFoundation/ITDebug.h>
 #import <ITFoundation/ITFoundation.h>
 #import <ITFoundation/ITDebug.h>
 #import <ITFoundation/ITFoundation.h>

-#import <ITMTRemote/ITMTRemote.h>

 
 static NetworkController *sharedController;
 
 
 static NetworkController *sharedController;
 


@@ -32,6 +20,8 @@ static NetworkController *sharedController;
         sharedController = self;
         browser = [[NSNetServiceBrowser alloc] init];
         [browser setDelegate:self];
         sharedController = self;
         browser = [[NSNetServiceBrowser alloc] init];
         [browser setDelegate:self];

+        rootObject = [[NetworkObject alloc] init];
+        serverPort = [[NSSocketPort alloc] initWithTCPPort:SERVER_PORT];

     }
     return self;
 }
     }
     return self;
 }


@@ -40,9 +30,12 @@ static NetworkController *sharedController;
 {
     [self disconnect];
     if (serverOn) {
 {
     [self disconnect];
     if (serverOn) {

-        [serverConnection invalidate];

         [serverConnection release];
     }
         [serverConnection release];
     }

+    [serverPass release];
+    [clientPass release];
+    [serverPort release];
+    [rootObject release];

     [clientProxy release];
     [remoteServices release];
     [browser release];
     [clientProxy release];
     [remoteServices release];
     [browser release];


@@ -67,16 +60,21 @@ static NetworkController *sharedController;
 {
     if (!serverOn && status) {
         NSString *name = [[NSUserDefaults standardUserDefaults] stringForKey:@"sharedPlayerName"];
 {
     if (!serverOn && status) {
         NSString *name = [[NSUserDefaults standardUserDefaults] stringForKey:@"sharedPlayerName"];

+        unsigned char buffer;
+        NSData *fullPass;

         //Turn on
         NS_DURING
         //Turn on
         NS_DURING

-            serverPort = [[NSSocketPort alloc] initWithTCPPort:SERVER_PORT];

             serverConnection = [[NSConnection alloc] initWithReceivePort:serverPort
                                                      sendPort:serverPort];
             serverConnection = [[NSConnection alloc] initWithReceivePort:serverPort
                                                      sendPort:serverPort];

-            [serverConnection setRootObject:[[MainController sharedController] currentRemote]];
+            [serverConnection setRootObject:rootObject];
+            [rootObject makeValid];

             [serverConnection registerName:@"ITMTPlayerHost"];
             [serverConnection registerName:@"ITMTPlayerHost"];

-            [serverConnection setDelegate:self];

         NS_HANDLER
         NS_HANDLER

+            [serverConnection setRootObject:nil];
+            [serverConnection release];
+            [serverPort release];

             ITDebugLog(@"Error starting server!");
             ITDebugLog(@"Error starting server!");

+            return;

         NS_ENDHANDLER
         ITDebugLog(@"Started server.");
         if (!name) {
         NS_ENDHANDLER
         ITDebugLog(@"Started server.");
         if (!name) {


@@ -86,50 +84,163 @@ static NetworkController *sharedController;
                                         type:@"_mttp._tcp."
                                         name:name
                                         port:SERVER_PORT];
                                         type:@"_mttp._tcp."
                                         name:name
                                         port:SERVER_PORT];

+        fullPass = [[NSUserDefaults standardUserDefaults] dataForKey:@"sharedPlayerPassword"];
+        if ([fullPass length]) {
+            [fullPass getBytes:&buffer range:NSMakeRange(6, 4)];
+            [serverPass release];
+            serverPass = [[NSData alloc] initWithBytes:&buffer length:strlen(&buffer)];
+        } else {
+            serverPass = nil;
+        }

         [service publish];
         serverOn = YES;
         [service publish];
         serverOn = YES;

+        ITDebugLog(@"Server service published.");

     } else if (serverOn && !status && [serverConnection isValid]) {
         //Turn off
         [service stop];
     } else if (serverOn && !status && [serverConnection isValid]) {
         //Turn off
         [service stop];

+        [service release];
+        [rootObject invalidate];

         [serverConnection registerName:nil];
         [serverConnection registerName:nil];

-        [serverPort invalidate];

         [serverConnection invalidate];
         [serverConnection invalidate];

+        //[serverConnection setRootObject:nil];
+        //[[serverConnection sendPort] autorelease];

         [serverConnection release];
         ITDebugLog(@"Stopped server.");
         serverOn = NO;
     }
 }
 
         [serverConnection release];
         ITDebugLog(@"Stopped server.");
         serverOn = NO;
     }
 }
 

-- (BOOL)connectToHost:(NSString *)host
+- (int)connectToHost:(NSString *)host

 {
 {

+    NSData *fullPass = [[NSUserDefaults standardUserDefaults] dataForKey:@"connectPassword"];
+    unsigned char buffer;

     ITDebugLog(@"Connecting to host: %@", host);
     ITDebugLog(@"Connecting to host: %@", host);

+    [remoteHost release];
+    remoteHost = [host copy];
+    if (fullPass) {
+        [fullPass getBytes:&buffer range:NSMakeRange(6, 4)];
+        [clientPass release];
+        clientPass = [[NSData alloc] initWithBytes:&buffer length:strlen(&buffer)];
+    } else {
+        clientPass = nil;
+    }

     NS_DURING
         clientPort = [[NSSocketPort alloc] initRemoteWithTCPPort:SERVER_PORT
                                            host:host];
         clientConnection = [[NSConnection connectionWithReceivePort:nil sendPort:clientPort] retain];
     NS_DURING
         clientPort = [[NSSocketPort alloc] initRemoteWithTCPPort:SERVER_PORT
                                            host:host];
         clientConnection = [[NSConnection connectionWithReceivePort:nil sendPort:clientPort] retain];

+        [clientConnection setReplyTimeout:10];

         clientProxy = [[clientConnection rootProxy] retain];
         clientProxy = [[clientConnection rootProxy] retain];

+        connectedToServer = YES;

     NS_HANDLER
     NS_HANDLER

+        [clientConnection release];
+        [clientPort release];

         ITDebugLog(@"Connection to host failed: %@", host);
         ITDebugLog(@"Connection to host failed: %@", host);

-        return NO;
+        return 0;

     NS_ENDHANDLER
     NS_ENDHANDLER

-    [clientConnection setReplyTimeout:5];
+    
+    if (!clientProxy) {
+        ITDebugLog(@"Null proxy! Couldn't connect!");
+        [self disconnect];
+        return 0;
+    }
+    
+    if ([clientProxy requiresPassword]) {
+        ITDebugLog(@"Server requires password.");
+        //Check to see if a password is set in defaults
+        if ([[NSUserDefaults standardUserDefaults] dataForKey:@"connectPassword"] == nil) {
+            ITDebugLog(@"Asking for password.");
+            if (![[PreferencesController sharedPrefs] showPasswordPanel]) {
+                ITDebugLog(@"Giving up connection attempt.");
+                [self disconnect];
+                return -1;
+            }
+        }
+        
+        //Send the password
+        ITDebugLog(@"Sending password.");
+        while (![clientProxy sendPassword:[[NSUserDefaults standardUserDefaults] dataForKey:@"connectPassword"]]) {
+            ITDebugLog(@"Invalid password!");
+            if (![[PreferencesController sharedPrefs] showInvalidPasswordPanel]) {
+                ITDebugLog(@"Giving up connection attempt.");
+                [self disconnect];
+                return -1;
+            }
+        }
+    }
+    

     ITDebugLog(@"Connected to host: %@", host);
     [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(disconnect) name:NSConnectionDidDieNotification object:clientConnection];
     ITDebugLog(@"Connected to host: %@", host);
     [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(disconnect) name:NSConnectionDidDieNotification object:clientConnection];

-    connectedToServer = YES;
-    return YES;
+    return 1;

 }
 
 - (BOOL)disconnect
 {
     ITDebugLog(@"Disconnecting from host.");
     connectedToServer = NO;
 }
 
 - (BOOL)disconnect
 {
     ITDebugLog(@"Disconnecting from host.");
     connectedToServer = NO;

+    [remoteHost release];
+    remoteHost = nil;

     [[NSNotificationCenter defaultCenter] removeObserver:self];
     [clientProxy release];
     [[NSNotificationCenter defaultCenter] removeObserver:self];
     [clientProxy release];

-    [clientConnection invalidate];

     [clientConnection release];
     return YES;
 }
 
     [clientConnection release];
     return YES;
 }
 

+- (BOOL)checkForServerAtHost:(NSString *)host
+{
+    NSData *fullPass = [[NSUserDefaults standardUserDefaults] dataForKey:@"connectPassword"];
+    unsigned char buffer;
+    NSConnection *testConnection = nil;
+    NSSocketPort *testPort = nil;
+    NetworkObject *tempProxy;
+    BOOL valid;
+    ITDebugLog(@"Checking for shared remote at %@.", host);
+    if (fullPass) {
+        [fullPass getBytes:&buffer range:NSMakeRange(6, 4)];
+        [clientPass release];
+        clientPass = [[NSData alloc] initWithBytes:&buffer length:strlen(&buffer)];
+    } else {
+        clientPass = nil;
+    }
+    
+    NS_DURING
+        testPort = [[NSSocketPort alloc] initRemoteWithTCPPort:SERVER_PORT
+                                         host:host];
+        testConnection = [[NSConnection connectionWithReceivePort:nil sendPort:testPort] retain];
+        [testConnection setReplyTimeout:5];
+        [testConnection setRequestTimeout:5];
+        tempProxy = (NetworkObject *)[testConnection rootProxy];
+        [tempProxy serverName];
+        valid = [tempProxy isValid];
+    NS_HANDLER
+        ITDebugLog(@"Connection to host failed: %@", host);
+        [testConnection release];
+        [testPort release];
+        return NO;
+    NS_ENDHANDLER
+    
+    if (!tempProxy) {
+        ITDebugLog(@"Null proxy! Couldn't connect!");
+        [testConnection release];
+        [testPort release];
+        return NO;
+    }
+    [testConnection release];
+    [testPort release];
+    return valid;
+}
+
+- (void)resetServerName
+{
+    if ([self isServerOn]) {
+        [service stop];
+        [service release];
+        service = [[NSNetService alloc] initWithDomain:@""
+                                        type:@"_mttp._tcp."
+                                        name:[[NSUserDefaults standardUserDefaults] stringForKey:@"sharedPlayerName"]
+                                        port:SERVER_PORT];
+    }
+}
+

 - (BOOL)isServerOn
 {
     return serverOn;
 - (BOOL)isServerOn
 {
     return serverOn;


@@ -145,9 +256,14 @@ static NetworkController *sharedController;
     return connectedToServer;
 }
 
     return connectedToServer;
 }
 

-- (ITMTRemote *)sharedRemote
+- (NSString *)remoteHost

 {
 {

-    return (ITMTRemote *)clientProxy;
+    return remoteHost;
+}
+
+- (NetworkObject *)networkObject
+{
+    return clientProxy;

 }
 
 - (NSArray *)remoteServices
 }
 
 - (NSArray *)remoteServices


@@ -160,7 +276,12 @@ static NetworkController *sharedController;
     ITDebugLog(@"Found service named %@.", [aNetService name]);
     [remoteServices addObject:aNetService];
     [aNetService setDelegate:self];
     ITDebugLog(@"Found service named %@.", [aNetService name]);
     [remoteServices addObject:aNetService];
     [aNetService setDelegate:self];

-    [aNetService resolve];
+       //Figure out if it responds to the 10.4 method
+       if ([aNetService respondsToSelector:@selector(resolveWithTimeout:)]) {
+               (void)[aNetService resolveWithTimeout:5.0];
+       } else {
+               [aNetService resolve];
+       }

     if (!moreComing) {
         [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTFoundNetService" object:nil];
     }
     if (!moreComing) {
         [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTFoundNetService" object:nil];
     }


@@ -178,8 +299,11 @@ static NetworkController *sharedController;
 - (void)netServiceDidResolveAddress:(NSNetService *)sender
 {
     ITDebugLog(@"Resolved service named %@.", [sender name]);
 - (void)netServiceDidResolveAddress:(NSNetService *)sender
 {
     ITDebugLog(@"Resolved service named %@.", [sender name]);

-    NSLog(@"Resolved service named %@.", [sender name]);

     [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTFoundNetService" object:nil];
     [[NSNotificationCenter defaultCenter] postNotificationName:@"ITMTFoundNetService" object:nil];

+    if ([[NSUserDefaults standardUserDefaults] boolForKey:@"useSharedPlayer"] && !connectedToServer) {
+        [[MainController sharedController] checkForRemoteServerAndConnectImmediately:NO];
+    }
+    [sender stop];

 }
 
 - (void)netServiceWillResolve:(NSNetService *)sender
 }
 
 - (void)netServiceWillResolve:(NSNetService *)sender