Initial import from (slightly modified) CocoaMySQL v0.7b4

1 files changed, 225 insertions, 0 deletions

diff --git a/SSHTunnel_old.m b/SSHTunnel_old.m
new file mode 100755
index 00000000..57a458eb
--- /dev/null
+++ b/SSHTunnel_old.m
@@ -0,0 +1,225 @@
+/*
+
+SSHTunnel.m
+
+Original code by tynsoe.org, Copyright 2002
+Modified by Lorenz Textor for use with CocoaMySQL
+
+*/
+
+#import "SSHTunnel.h"
+#include <unistd.h>
+
+// start diff lorenz textor
+/*
+#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"
+*/
+#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
+
+// initialization
+- (id)init
+{
+    self = [super init];
+	
+    // Make this class the root one for AppleEvent calls
+//    [[ NSScriptExecutionContext sharedScriptExecutionContext] setTopLevelObject: self ];
+
+	return self;
+}
+
+// Getting tunnels informations
+- (BOOL)isRunning
+/* returns YES if tunnel is running */
+{
+    return [ task isRunning ];
+}
+
+- (NSString*)status
+{
+    if (status)
+		return status;
+    return S_IDLE;
+}
+
+// starting & stopping the tunnel
+- (void)startTunnel
+/* starts tunnel with saved arguments */
+{
+	[self startTunnelWithArguments:tunnelArguments];
+}
+
+- (void)startTunnelWithArguments:(NSDictionary *)args
+/* starts the tunnel */
+{
+	NSMutableArray *arguments = [[ NSMutableArray alloc] init ];
+
+	if (tunnelArguments )
+		[tunnelArguments release];
+	tunnelArguments = [args retain];
+
+	// stop tunnel if already running
+	if ( [self isRunning] )
+//		[self stopTunnel];
+		return;
+
+    shouldStop = NO;
+
+// get arguments
+	[ arguments addObject: @"-N" ];
+	[ arguments addObject: @"-v" ];
+
+//	[ arguments addObject: @"-p" ];
+//	[ arguments addObject: @"-p" ];
+//	[ arguments addObject: @"22" ];
+
+//	[ arguments addObject: @"-c"];
+//	[ arguments addObject: @"3des"];
+
+	[ arguments addObject: [ NSString stringWithFormat: @"%@@%@", [args objectForKey:@"connUser"], [args objectForKey:@"connHost"] ]];
+
+	[ arguments addObject: @"-L" ];
+	[ arguments addObject: [NSString stringWithFormat:@"%@/%@/%@", [args objectForKey:@"localPort"], [args objectForKey:@"host"], [args objectForKey:@"remotePort"]] ];
+
+    [ NSThread detachNewThreadSelector:@selector(launchTunnel:)
+			      toTarget: self
+			    withObject: arguments ];
+
+	[ arguments release ];
+}
+
+- (void)stopTunnel
+/* stops the tunnel */
+{
+    if (! [ self isRunning ])
+		return;
+    shouldStop=YES;
+    [ self setValue: nil forKey: @"status" ];
+    [ task terminate ];
+    [[ NSNotificationCenter defaultCenter]  postNotificationName:@"STMStatusChanged" object:self ];
+}
+
+- (void)launchTunnel:(NSArray*)arguments
+/* launches the tunnel in a separate thread */
+{
+    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" ];
+    
+    [ task setLaunchPath: @"/usr/bin/ssh" ];
+    [ task setArguments: arguments ];
+
+// really necessary???
+	[ environment removeObjectForKey: @"SSH_AGENT_PID" ];
+	[ environment removeObjectForKey: @"SSH_AUTH_SOCK" ];
+	[ environment setObject: pathToAuthentifier forKey: @"SSH_ASKPASS" ];
+	[ environment setObject:@":0" forKey:@"DISPLAY" ];
+    
+    [ environment setObject: @"CocoaMySQL Tunnel" 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,@"CocoaMySQL Tunnel");
+    [ self setValue: S_CONNECTING forKey: @"status" ];
+    [ task launch ];
+    [[ NSNotificationCenter defaultCenter]  postNotificationName:@"STMStatusChanged" object:self ];
+    [ task waitUntilExit ];
+    sleep(1);
+    [ self setValue: S_IDLE forKey: @"status" ];
+    NSLog(T_STOP,@"CocoaMySQL Tunnel");
+    [[ 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)
+			{
+				[ 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] ;
+}
+
+// deallocation
+- (void) dealloc
+{
+	[self stopTunnel];
+	
+	[task release];
+    [stdErrPipe release];
+    [status release];
+	[tunnelArguments release];
+
+    [super dealloc];
+}
+
+@end