aboutsummaryrefslogtreecommitdiffstats
path: root/SSHTunnel.m
diff options
context:
space:
mode:
authorabhibeckert <abhi@abhibeckert.com>2008-04-04 11:37:36 +0000
committerabhibeckert <abhi@abhibeckert.com>2008-04-04 11:37:36 +0000
commit0f586da34dac2c764a12055158e4929cf69340a1 (patch)
treebca3868c9715a10c08017a0a7b6d5148f8bd3556 /SSHTunnel.m
parent5fb284b7c12887b8eea6c3a02fa83abf50793992 (diff)
downloadsequelpro-0f586da34dac2c764a12055158e4929cf69340a1.tar.gz
sequelpro-0f586da34dac2c764a12055158e4929cf69340a1.tar.bz2
sequelpro-0f586da34dac2c764a12055158e4929cf69340a1.zip
Initial import from (slightly modified) CocoaMySQL v0.7b4
Diffstat (limited to 'SSHTunnel.m')
-rwxr-xr-xSSHTunnel.m531
1 files changed, 531 insertions, 0 deletions
diff --git a/SSHTunnel.m b/SSHTunnel.m
new file mode 100755
index 00000000..a2d8e1f7
--- /dev/null
+++ b/SSHTunnel.m
@@ -0,0 +1,531 @@
+//
+// SSHTunnel.m
+// SSH Tunnel Manager 2
+//
+// Created by Yann Bizeul on Wed Nov 19 2003.
+// Copyright (c) 2003 __MyCompanyName__. All rights reserved.
+//
+
+#import "SSHTunnel.h"
+#include <unistd.h>
+
+// start diff lorenz textor
+/*
+#define T_START NSLocalizedString(@"T_START",@"")
+#define T_STOP NSLocalizedString(@"T_STOP",@"")
+#define S_IDLE NSLocalizedString(@"S_IDLE",@"")
+#define S_CONNECTING NSLocalizedString(@"S_CONNECTING",@"")
+#define S_CONNECTED NSLocalizedString(@"S_CONNECTED",@"")
+#define S_AUTH NSLocalizedString(@"S_AUTH",@"")
+#define S_PORT NSLocalizedString(@"S_PORT",@"")
+*/
+#define T_START @"START: %@"
+#define T_STOP @"STOP: %@"
+#define S_IDLE @"Idle"
+#define S_CONNECTING @"Connecting..."
+#define S_CONNECTED @"Connected"
+#define S_AUTH @"Authenticated"
+#define S_PORT "Port %@ forwarded"
+// end diff lorenz textor
+
+@implementation SSHTunnel
+
+#pragma mark -
+#pragma mark Initialization
+-(id)init
+{
+ return [ self initWithName:@"New Tunnel"];
+}
+-(id)initWithName:(NSString*)aName
+{
+ NSDictionary *dictionary = [ NSDictionary dictionaryWithObjectsAndKeys:
+ [ NSNumber numberWithBool: NO ],@"compression",
+ [ NSNumber numberWithBool: YES ],@"connAuth",
+ @"", @"connHost",
+ aName, @"connName",
+ @"", @"connPort",
+ [ NSNumber numberWithBool: NO ],@"connRemote",
+ @"", @"connUser",
+ @"3des", @"encryption",
+ [ NSNumber numberWithBool: NO ],@"socks4",
+ [ NSNumber numberWithInt: 1080 ], @"socks4p",
+ [ NSArray array ], @"tunnelsLocal",
+ [ NSArray array ], @"tunnelsRemote",
+ [ NSNumber numberWithBool: NO ],@"v1", nil
+ ];
+ return [ self initWithDictionary: dictionary ];
+}
+-(id)initWithDictionary:(NSDictionary*)aDictionary
+{
+ NSEnumerator *e;
+ NSString *key;
+
+ self = [ super init ];
+ e = [[ aDictionary allKeys ] objectEnumerator ];
+ while (key = [ e nextObject ])
+ {
+ [ self setValue: [ aDictionary objectForKey: key ] forKey: key ];
+ }
+ code = 0;
+ if ([[ self valueForKey: @"autoConnect" ] boolValue ])
+ [ self startTunnel ];
+ return self;
+}
++(id)tunnelWithName:(NSString*)aName
+{
+ return [[ SSHTunnel alloc ] initWithName: aName ];
+}
++(SSHTunnel*)tunnelFromDictionary:(NSDictionary*)aDictionary
+{
+ return [[ SSHTunnel alloc ] initWithDictionary: aDictionary ];
+}
++(NSArray*)tunnelsFromArray:(NSArray*)anArray
+{
+ NSMutableArray *newArray;
+ SSHTunnel *currentTunnel;
+ NSEnumerator *e;
+ NSDictionary *currentTunnelDictionary;
+
+ newArray = [ NSMutableArray array ];
+ e = [ anArray objectEnumerator ];
+ while (currentTunnelDictionary = [ e nextObject ])
+ {
+ currentTunnel = [ SSHTunnel tunnelFromDictionary: currentTunnelDictionary ];
+ [ newArray addObject: currentTunnel ];
+ }
+ return [[ newArray copy ] autorelease ];
+}
+
+#pragma mark -
+#pragma mark Adding and removing port redir.
+-(void)addLocalTunnel:(NSDictionary*)aDictionary;
+{
+ NSMutableArray *tempTunnelsLocal = [ NSMutableArray arrayWithArray: tunnelsLocal ];
+ [ tempTunnelsLocal addObject: aDictionary ];
+ [ tunnelsLocal release ];
+ tunnelsLocal = [ tempTunnelsLocal copy ];
+}
+- (void)removeLocal:(int)index
+{
+ NSMutableArray *tempLocalTunnels = [ tunnelsLocal mutableCopy ];
+ [ tempLocalTunnels removeObjectAtIndex: index ];
+ [ tunnelsLocal release ];
+ tunnelsLocal = [ tempLocalTunnels copy ];
+ [ tempLocalTunnels release ];
+}
+-(void)addRemoteTunnel:(NSDictionary*)aDictionary;
+{
+ NSMutableArray *tempTunnelsRemote = [ NSMutableArray arrayWithArray: tunnelsRemote ];
+ [ tempTunnelsRemote addObject: aDictionary ];
+ [ tunnelsRemote release ];
+ tunnelsRemote = [ tempTunnelsRemote copy ];
+}
+- (void)removeRemote:(int)index
+{
+ NSMutableArray *tempRemoteTunnels = [ tunnelsRemote mutableCopy ];
+ [ tempRemoteTunnels removeObjectAtIndex: index ];
+ [ tunnelsRemote release ];
+ tunnelsRemote = [ tempRemoteTunnels copy ];
+ [ tempRemoteTunnels release ];
+}
+- (void)setLocalValue:(NSString*)aValue ofTunnel:(int)index forKey:(NSString*)key
+{
+ NSMutableArray *tempLocalTunnel;
+ NSMutableDictionary *tempCurrentTunnel;
+
+ tempLocalTunnel = [tunnelsLocal mutableCopy];
+ tempCurrentTunnel = [[ tempLocalTunnel objectAtIndex: index ] mutableCopy ];
+
+ [ tempCurrentTunnel setObject: aValue forKey: key ];
+ [ tempLocalTunnel replaceObjectAtIndex:index withObject:[tempCurrentTunnel copy ]];
+ [ tempCurrentTunnel release ];
+ [ tunnelsLocal release ];
+ tunnelsLocal = [ tempLocalTunnel copy ];
+}
+- (void)setRemoteValue:(NSString*)aValue ofTunnel:(int)index forKey:(NSString*)key
+{
+ NSMutableArray *tempRemoteTunnel;
+ NSMutableDictionary *tempCurrentTunnel;
+
+ tempRemoteTunnel = [tunnelsRemote mutableCopy];
+ tempCurrentTunnel = [[ tempRemoteTunnel objectAtIndex: index ] mutableCopy ];
+
+ [ tempCurrentTunnel setObject: aValue forKey: key ];
+ [ tempRemoteTunnel replaceObjectAtIndex:index withObject:[tempCurrentTunnel copy ]];
+ [ tempCurrentTunnel release ];
+ [ tunnelsRemote release ];
+ tunnelsRemote = [ tempRemoteTunnel copy ];
+}
+
+#pragma mark -
+#pragma mark Execution related
+- (void)startTunnel
+{
+// NSDictionary *t;
+// NSEnumerator *e;
+// BOOL asRoot = NO;
+
+ if ([ self isRunning ])
+ return;
+
+ shouldStop = NO;
+ /*
+ [ arguments addObject: @"-N" ];
+ [ arguments addObject: @"-v" ];
+ [ arguments addObject: @"-p" ];
+ if ([ connPort length ])
+ [ arguments addObject: connPort];
+ else
+ [ arguments addObject: @"22" ];
+
+ if (connRemote)
+ [ arguments addObject: @"-g" ];
+ if (compression)
+ [ arguments addObject: @"-C" ];
+ if (v1)
+ [ arguments addObject: @"-1" ];
+
+ [ arguments addObject: @"-c"];
+ if (encryption)
+ [ arguments addObject: encryption];
+ else
+ [ arguments addObject: @"3des"];
+
+ if (socks4 && socks4p != nil)
+ {
+ [ arguments addObject: @"-D" ];
+ [ arguments addObject: [ socks4p stringValue ]];
+ }
+ [ arguments addObject: [ NSString stringWithFormat: @"%@@%@",
+ connUser, connHost ]
+ ];
+
+ NSString *hostPort;
+ e = [ tunnelsLocal objectEnumerator ];
+ while (t = [ e nextObject ])
+ {
+ [ arguments addObject: @"-L" ];
+ if ([[ t objectForKey:@"hostport"] isEqualTo: @"" ])
+ hostPort = [ t objectForKey:@"port" ];
+ else
+ hostPort = [ t objectForKey:@"hostport" ];
+ [ arguments addObject: [NSString stringWithFormat:@"%@/%@/%@",
+ [ t objectForKey:@"port"],
+ [ t objectForKey:@"host"],
+ hostPort
+ ] ];
+ if ([[ t objectForKey:@"port"] intValue] < 1024)
+ asRoot=YES;
+ }
+
+ e = [ tunnelsRemote objectEnumerator ];
+ while (t = [ e nextObject ])
+ {
+ [ arguments addObject: @"-R" ];
+ if ([[ t objectForKey:@"hostport"] isEqualTo: @"" ])
+ hostPort = [ t objectForKey:@"port" ];
+ else
+ hostPort = [ t objectForKey:@"hostport" ];
+ [ arguments addObject: [NSString stringWithFormat:@"%@/%@/%@",
+ [ t objectForKey:@"port"],
+ [ t objectForKey:@"host"],
+ hostPort
+ ]];
+ }
+ args = [ NSMutableDictionary dictionary ];
+ [ args setObject: arguments forKey:@"arguments" ];
+ [ args setObject: [ NSNumber numberWithBool: connAuth ] forKey: @"handleAuth" ];
+ [ args setObject: connName forKey:@"tunnelName" ];
+ [ args setObject: [ NSNumber numberWithBool: asRoot ] forKey: @"asRoot" ];
+
+
+ [ NSThread detachNewThreadSelector:@selector(launchTunnel:)
+ toTarget: self
+ withObject: args ];
+ */
+ [ NSThread detachNewThreadSelector:@selector(launchTunnel:)
+ toTarget: self
+ withObject: nil ];
+// [ arguments release ];
+
+}
+- (void)stopTunnel
+{
+ if (! [ self isRunning ])
+ return;
+ shouldStop=YES;
+ [ self setValue: nil forKey: @"status" ];
+ [ task terminate ];
+ code = 0;
+ [[ NSNotificationCenter defaultCenter] postNotificationName:@"STMStatusChanged" object:self ];
+}
+
+- (void)toggleTunnel
+{
+ if ([ self isRunning ])
+ [ self stopTunnel ];
+ else
+ [ self startTunnel ];
+}
+
+- (void)launchTunnel:(id)foo;
+{
+ NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
+
+ if (task)
+ [ task release ];
+ task = [[ NSTask alloc ] init ];
+ NSMutableDictionary *environment = [ NSMutableDictionary dictionaryWithDictionary: [[ NSProcessInfo processInfo ] environment ]];
+ NSString *pathToAuthentifier = [[ NSBundle mainBundle ] pathForResource: @"askForPass" ofType: @"sh" ];
+ if (socks4)
+ [ task setLaunchPath: [[ NSBundle mainBundle ] pathForResource: @"ssh" ofType: @"" ]];
+ else
+ [ task setLaunchPath: @"/usr/bin/ssh" ];
+ [ task setArguments: [ self arguments ]];
+ if (connAuth)
+ {
+ [ environment removeObjectForKey: @"SSH_AGENT_PID" ];
+ [ environment removeObjectForKey: @"SSH_AUTH_SOCK" ];
+ [ environment setObject: pathToAuthentifier forKey: @"SSH_ASKPASS" ];
+ [ environment setObject:@":0" forKey:@"DISPLAY" ];
+ }
+ [ environment setObject: connName forKey: @"TUNNEL_NAME" ];
+ [ task setEnvironment: environment ];
+
+ stdErrPipe = [[ NSPipe alloc ] init ];
+ [ task setStandardError: stdErrPipe ];
+
+ [[ NSNotificationCenter defaultCenter] addObserver:self
+ selector:@selector(stdErr:)
+ name: @"NSFileHandleDataAvailableNotification"
+ object:[ stdErrPipe fileHandleForReading]];
+
+ [[ stdErrPipe fileHandleForReading] waitForDataInBackgroundAndNotify ];
+
+ NSLog(T_START,connName);
+ [ self setValue: S_CONNECTING forKey: @"status" ];
+ code = 1;
+ [ task launch ];
+ [[ NSNotificationCenter defaultCenter] postNotificationName:@"STMStatusChanged" object:self ];
+ [ task waitUntilExit ];
+ sleep(1);
+ code = 0;
+ [ self setValue: S_IDLE forKey: @"status" ];
+ NSLog(T_STOP,connName);
+ [[ NSNotificationCenter defaultCenter] removeObserver:self
+ name: @"NSFileHandleDataAvailableNotification"
+ object:[ stdErrPipe fileHandleForReading]];
+ [ task release ];
+ task = nil;
+ [ stdErrPipe release ];
+ stdErrPipe = nil;
+ [[ NSNotificationCenter defaultCenter] postNotificationName:@"STMStatusChanged" object:self ];
+ if (! shouldStop)
+ [ self startTunnel ];
+ [ pool release ];
+}
+
+- (void)stdErr:(NSNotification*)aNotification
+{
+ NSData *data = [[ aNotification object ] availableData ];
+ NSString *log = [[ NSString alloc ] initWithData: data encoding: NSASCIIStringEncoding ];
+ BOOL wait = YES;
+ if ([ log length ])
+ {
+ //NSLog(log);
+ NSArray *lines = [ log componentsSeparatedByString:@"\n" ];
+ NSEnumerator *e = [ lines objectEnumerator ];
+ NSString *line;
+ while (line = [ e nextObject ])
+ {
+ if ([ line rangeOfString:@"Entering interactive session." ].location != NSNotFound)
+ {
+ code = 2;
+ [ self setValue: S_CONNECTED forKey: @"status"];
+ }
+ if ([ line rangeOfString:@"Authentication succeeded" ].location != NSNotFound)
+ [ self setValue: S_AUTH forKey: @"status"];
+ if ([ line rangeOfString:@"Connections to local port" ].location != NSNotFound)
+ {
+ NSScanner *s;
+ NSString *port;
+ s = [ NSScanner scannerWithString:log];
+ [ s scanUpToString: @"Connections to local port " intoString: nil ];
+ [ s scanString: @"Connections to local port " intoString: nil ];
+ [ s scanUpToString: @"forwarded" intoString:&port];
+ [ self setValue: [ NSString stringWithFormat: @"Port %@ forwarded", port ] forKey: @"status"];
+ }
+ if ([ line rangeOfString:@"closed by remote host." ].location != NSNotFound)
+ {
+ [ task terminate];
+ [ self setValue: @"Connection closed" forKey: @"status"];
+ wait = NO;
+ }
+ [[ NSNotificationCenter defaultCenter] postNotificationName:@"STMStatusChanged" object:self ];
+ }
+ if (wait)
+ [[ stdErrPipe fileHandleForReading ] waitForDataInBackgroundAndNotify ];
+ }
+ [ log release] ;
+}
+
+- (BOOL)isRunning
+{
+ if ([ task isRunning ])
+ return YES;
+ return NO;
+}
+
+#pragma mark -
+#pragma mark Getting tunnel informations
+- (NSString *)status
+{
+ if (status)
+ return status;
+ return S_IDLE;
+}
+- (NSArray*)arguments
+{
+ NSMutableArray *arguments;
+ NSEnumerator *e;
+ NSDictionary *t;
+ BOOL asRoot;
+
+ arguments = [ NSMutableArray array ];
+ [ arguments addObject: @"-N" ];
+ [ arguments addObject: @"-v" ];
+ [ arguments addObject: @"-p" ];
+ if ([ connPort length ])
+ [ arguments addObject: connPort];
+ else
+ [ arguments addObject: @"22" ];
+
+ if (connRemote)
+ [ arguments addObject: @"-g" ];
+ if (compression)
+ [ arguments addObject: @"-C" ];
+ if (v1)
+ [ arguments addObject: @"-1" ];
+
+ [ arguments addObject: @"-c"];
+ if (encryption)
+ [ arguments addObject: encryption];
+ else
+ [ arguments addObject: @"3des"];
+
+ if (socks4 && socks4p != nil)
+ {
+ [ arguments addObject: @"-D" ];
+ [ arguments addObject: [ socks4p stringValue ]];
+ }
+ [ arguments addObject: [ NSString stringWithFormat: @"%@@%@",
+ connUser, connHost ]
+ ];
+
+ NSString *hostPort;
+ e = [ tunnelsLocal objectEnumerator ];
+ while (t = [ e nextObject ])
+ {
+ [ arguments addObject: @"-L" ];
+ if ([[ t objectForKey:@"hostport"] isEqualTo: @"" ])
+ hostPort = [ t objectForKey:@"port" ];
+ else
+ hostPort = [ t objectForKey:@"hostport" ];
+ [ arguments addObject: [NSString stringWithFormat:@"%@/%@/%@",
+ [ t objectForKey:@"port"],
+ [ t objectForKey:@"host"],
+ hostPort
+ ] ];
+ if ([[ t objectForKey:@"port"] intValue] < 1024)
+ asRoot=YES;
+ }
+
+ e = [ tunnelsRemote objectEnumerator ];
+ while (t = [ e nextObject ])
+ {
+ [ arguments addObject: @"-R" ];
+ if ([[ t objectForKey:@"hostport"] isEqualTo: @"" ])
+ hostPort = [ t objectForKey:@"port" ];
+ else
+ hostPort = [ t objectForKey:@"hostport" ];
+ [ arguments addObject: [NSString stringWithFormat:@"%@/%@/%@",
+ [ t objectForKey:@"port"],
+ [ t objectForKey:@"host"],
+ hostPort
+ ]];
+ }
+
+ return [[ arguments copy ] autorelease ];
+}
+
+- (NSDictionary*)dictionary
+{
+ return [ NSDictionary dictionaryWithObjectsAndKeys:
+ [ NSNumber numberWithBool: compression ],@"compression",
+ [ NSNumber numberWithBool: connAuth ],@"connAuth",
+ [ NSNumber numberWithBool: autoConnect ],@"autoConnect",
+ connHost, @"connHost",
+ connName, @"connName",
+ connPort, @"connPort",
+ [ NSNumber numberWithBool: connRemote ],@"connRemote",
+ connUser, @"connUser",
+ encryption, @"encryption",
+ [ NSNumber numberWithBool: socks4 ],@"socks4",
+ socks4p, @"socks4p",
+ tunnelsLocal, @"tunnelsLocal",
+ tunnelsRemote, @"tunnelsRemote",
+ [ NSNumber numberWithBool: v1 ],@"v1", nil
+ ];
+}
+
+
+#pragma mark -
+#pragma mark Key/Value coding
+- (NSImage*)icon
+{
+ switch (code)
+ {
+ case 0:
+ return [ NSImage imageNamed: @"offState" ];
+ break;
+ case 1:
+ return [ NSImage imageNamed: @"middleState" ];
+ break;
+ case 2:
+ return [ NSImage imageNamed: @"onState" ];
+ break;
+ }
+ return [ NSImage imageNamed: @"offState" ];
+}
+- (void)setValue:(id)value forUndefinedKey:(NSString *)key
+{
+ NSLog(@"key %@ undefined",key);
+}
+- (id)valueForUndefinedKey:(NSString *)key
+{
+ return nil;
+}
+
+#pragma mark -
+#pragma mark Misc.
+-(void)dealloc
+{
+ [ self stopTunnel ];
+ [ tunnelsLocal release ];
+ [ tunnelsRemote release ];
+
+ [ task release ];
+ [ stdErrPipe release ];
+ [ connName release ];
+ [ status release ];
+ [ connPort release ];
+ [ encryption release ];
+ [ socks4p release ];
+ [ connUser release ];
+ [ connHost release ];
+
+ // start diff lorenz textor
+ [super dealloc];
+ // end diff lorenz textor
+}
+@end