From 0f586da34dac2c764a12055158e4929cf69340a1 Mon Sep 17 00:00:00 2001 From: abhibeckert Date: Fri, 4 Apr 2008 11:37:36 +0000 Subject: Initial import from (slightly modified) CocoaMySQL v0.7b4 --- SSHTunnel.m | 531 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 531 insertions(+) create mode 100755 SSHTunnel.m (limited to 'SSHTunnel.m') 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 + +// 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 -- cgit v1.2.3