initial osx portosx/0.0

Signed-off-by: Michael Krelin <hacker@klever.net>

1 files changed, 301 insertions, 0 deletions

diff --git a/pumpkin/PumpKIN.m b/pumpkin/PumpKIN.m
new file mode 100644
index 0000000..9a4623a
--- a/dev/null
+++ b/pumpkin/PumpKIN.m
@@ -0,0 +1,301 @@
+#include <stdarg.h>
+#include <sys/stat.h>
+
+#import "PumpKIN.h"
+#import "NumberTransformer.h"
+#import "IPTransformer.h"
+#import "XFer.h"
+#import "XFersViewDatasource.h"
+#import "ARequest.h"
+
+
+@implementation PumpKIN
+@synthesize toolbar;
+@synthesize preferencesWindow;
+@synthesize theDefaults;
+
+@synthesize window;
+@synthesize logger;
+@synthesize xfersView;
+
+-(void) updateListener {
+    if(listener) {
+	[listener release]; listener = nil;
+    }
+    if(![[theDefaults.values valueForKey:@"listen"] boolValue]) return;
+    @try {
+	listener = [[DaemonListener listenerWithDefaults] retain];
+    }
+    @catch (NSException *e) {
+	[self log:@"Error starting the server. %@: %@",e.name,e.reason];
+	NSAlert *a = [NSAlert alertWithMessageText:@"Failed to enable tftp server" defaultButton:@"OK" alternateButton:nil otherButton:nil informativeTextWithFormat:@"Failed to enable tftp server.\n%@\n\n%@",e.name,e.reason];
+	a.alertStyle = NSWarningAlertStyle;
+	enum act_type {
+	    actDont = 0,
+	    actBindToAny = NSAlertThirdButtonReturn+1,
+	    actRemoveTFTPD, actChangePort
+	};
+	id en;
+	if (e.userInfo && (en=(e.userInfo)[@"errno"])) {
+	    switch([en intValue]) {
+		case EADDRINUSE:
+		    {
+			int p = [[theDefaults.values valueForKey:@"bindPort"] intValue];
+			if(p==69) {
+			    a.informativeText = @"The OS reports that the address is already in use.\n\n"
+				"It probably means, that some other programm is listening on the TFTP port."
+				    " since Mac OS X comes with its own tftpd, it is the likeliest suspect."
+				    " If you're going to use tftp server a lot, you may prefer to use that one."
+				    " If you want to use PumpKIN, you can either use unprivileged port (but make sure"
+				    " the client is aware of that and supports it) or unload Apple tftpd using"
+				    " command 'launchctl remove com.apple.tftpd' (as root). I can try doing either for you.";
+			    [a addButtonWithTitle:@"Change port to 6969"].tag = actChangePort;
+			    [a addButtonWithTitle:@"Try to stop Apple's tftpd"].tag = actRemoveTFTPD;
+			}else if(p!=6969) {
+			    a.informativeText = @"The OS reports that the address is already in use.\n\n"
+				"It probably means, that some other program is listening on the port."
+				    " you can either try to find out who's using the port and shut it down or"
+				    " change the port. Make sure to notify your peers of the change."
+				    " I can help you with changing the port.";
+			    [a addButtonWithTitle:@"Change port to 6969"].tag = actChangePort;
+			}else {
+			    a.informativeText = @"The OS reports that the address is already in use.\n\n"
+				"It probably means that some other program is listening on the port."
+				"You should either change port to the one that is not used or find the"
+				    " offending program and shut it down. Or go on without server.";
+			}
+		    }
+		    break;
+		case EADDRNOTAVAIL:
+		    a.informativeText = @"The OS reports that the address is not available.\n\n"
+			"It probably means, that the IP address you specified is not configured on this machine.\n\n"
+			"You can either ignore the error and go on without TFTP server, fix the ip address, by entering the one"
+			    " that is configured, or bind to the special '0.0.0.0' ip address which means listen to any"
+			    " address configured. The latter you can do automatically with a single click below.";
+		    [a addButtonWithTitle:@"Listen to any address"].tag = actBindToAny;
+		    [a addButtonWithTitle:@"Try removing Apple's daemon"].tag = actRemoveTFTPD;
+		    break;
+	    }
+	};
+	[theDefaults.values setValue:@NO forKey:@"listen"];
+	switch([a runModal]) {
+	    case actBindToAny:
+		[NSTimer scheduledTimerWithTimeInterval:0 target:self selector:@selector(bindToAny) userInfo:nil repeats:NO];
+		break;
+	    case actChangePort:
+		[NSTimer scheduledTimerWithTimeInterval:0 target:self selector:@selector(bindTo6969) userInfo:nil repeats:NO];
+		break;
+	    case actRemoveTFTPD:
+		{
+		    @try {
+			char const *args[] = { 0,"-k",NULL };
+			[self runBiportal:args];
+		    }@catch(NSException *e) {
+			[self log:@"Error trying to unload com.apple.tftpd. %@: %@",e.name,e.reason];
+		    }
+		}
+		[NSTimer scheduledTimerWithTimeInterval:0 target:self selector:@selector(bindAgain) userInfo:nil repeats:NO];
+		break;
+	}
+    }
+}
+
+-(void)bindAgain {
+    [theDefaults.values setValue:@YES forKey:@"listen"];
+}
+-(void)bindTo6969 {
+    [theDefaults.values setValue:@6969 forKey:@"bindPort"];
+    [self bindAgain];
+}
+-(void)bindToAny {
+    [theDefaults.values setValue:@"0.0.0.0" forKey:@"bindAddress"];
+    [self bindAgain];
+}
+
+- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context {
+    if( object==theDefaults && (
+	[keyPath isEqualToString:@"values.listen"]
+	|| [keyPath isEqualToString:@"values.bindPort"]
+	|| [keyPath isEqualToString:@"values.bindAddress"]) ) {
+	[self updateListener];
+    }
+}
+
+- (void)applicationDidFinishLaunching:(NSNotification *)aNotification
+{
+    [theDefaults addObserver:self forKeyPath:@"values.listen" options:0 context:0];
+    [theDefaults addObserver:self forKeyPath:@"values.bindAddress" options:0 context:0];
+    [theDefaults addObserver:self forKeyPath:@"values.bindPort" options:0 context:0];
+
+    listener = nil;
+    [window.contentView setWantsLayer:true];
+    window.backgroundColor = [NSColor colorWithPatternImage:[NSImage imageNamed:@"klever-background.png"]];
+    xfersView.dataSource = (xvDatasource = [[XFersViewDatasource alloc] initWithXfers:xfers=[NSMutableArray arrayWithCapacity:4]]);
+    [self updateListener];
+    if(![[theDefaults values] valueForKey:@"tftpRoot"])
+	[self showPreferences:nil];
+}
+
+- (IBAction)showPreferences:(id)sender {
+    [preferencesWindow makeKeyAndOrderFront:preferencesWindow];
+}
+
+- (void)log:(NSString*)fmt,... {
+    va_list vl; va_start(vl, fmt);
+    NSString *s = [[[[NSString alloc] initWithFormat:fmt arguments:vl] autorelease] stringByAppendingString:@"\n"];
+    va_end(vl);
+    NSString *lf = [theDefaults.values valueForKey:@"logFile"];
+    if(lf && ![lf isEqualTo:@""]) {
+	NSFileHandle *l = [NSFileHandle fileHandleForWritingAtPath:lf];
+	if(!l) {
+	    [[NSFileManager defaultManager] createFileAtPath:lf contents:nil attributes:nil];
+	    l = [NSFileHandle fileHandleForWritingAtPath:lf];
+	}
+	if(!l) {
+	    static NSString *bl = nil;
+	    if(!(bl && [bl isEqualToString:lf])) {
+		[[logger textStorage] appendAttributedString:[[[NSAttributedString alloc] initWithString:
+							   [NSString stringWithFormat:@"Failed to open/create '%@' log file\n",lf] ] autorelease]];
+		if(bl) [bl release];
+		bl = [NSString stringWithString:lf];
+	    }
+	}else{
+	    [l seekToEndOfFile];
+	    [l writeData:[[NSString stringWithFormat:@"[%@] %@",[[NSDate date] description],s] dataUsingEncoding:NSUTF8StringEncoding]];
+	    [l closeFile];
+	}
+    }
+    [[logger textStorage] appendAttributedString:[[[NSAttributedString alloc] initWithString:
+						  s ] autorelease]];
+    [logger scrollToEndOfDocument:nil];
+}
+
+-(void)registerXfer:(id)xfer {
+    [xfers insertObject:xfer atIndex:0];
+    [self updateXfers];
+}
+-(void)unregisterXfer:(id)xfer {
+    [xfers removeObject:xfer];
+    [self updateXfers];
+}
+-(void)updateXfers {
+    [xfersView reloadData];
+}
+-(BOOL)hasPeer:(struct sockaddr_in*)sin {
+    return NSNotFound!=[xfers indexOfObjectPassingTest:^BOOL(XFer *x,NSUInteger i,BOOL *s) {
+	struct sockaddr_in *p = x.peer;
+	return p->sin_len==sin->sin_len && !memcmp(p, sin, p->sin_len);
+    }];
+}
+
+-(BOOL)hasSelectedXfer {
+    return [xfersView selectedRow]>=0;
+}
+
+-(void)tableViewSelectionDidChange:(NSNotification *)an {
+    [toolbar validateVisibleItems];
+}
+-(BOOL)validateToolbarItem:(NSToolbarItem *)theItem {
+    if([theItem.itemIdentifier isEqualToString:@"abort_xfer"])
+	return self.hasSelectedXfer;
+    return YES;
+}
+-(IBAction)abortXfer:(id)sender {
+    NSInteger r = [xfersView selectedRow];
+    NSAssert(r>=0,@"no selected row");
+    if(r<0) return;
+    XFer *x = xfers[r];
+    [self log:@"Aborting transfer of '%@' %@",x.xferFilename,x.xferPrefix];
+    [x abort];
+}
+
+- (IBAction)getFile:(id)sender {
+    [ARequest getFile];
+}
+- (IBAction)putFile:(id)sender {
+    [ARequest putFile];
+}
+
+- (IBAction)pickTFTPFolder:(id)sender {
+    NSOpenPanel *op = [NSOpenPanel openPanel];
+    op.canChooseDirectories = YES; op.canChooseFiles = NO;
+    op.canCreateDirectories = YES;
+    op.allowsMultipleSelection = NO;
+    op.prompt = @"Set TFTP root";
+    op.title = @"TFTP root";
+    op.nameFieldLabel = @"TFTP root:";
+    if([op runModal]!=NSFileHandlingPanelOKButton) return;
+    [[theDefaults values] setValue:op.URL.path forKey:@"tftpRoot"];
+}
+
+- (IBAction)pickLogFile:(id)sender {
+    NSSavePanel *op = [NSSavePanel savePanel];
+    op.canCreateDirectories = YES;
+    op.prompt = @"Set log file";
+    op.title = @"Log to";
+    op.nameFieldLabel = @"Log to";
+    if([op runModal]!=NSFileHandlingPanelOKButton) return;
+    [[theDefaults values] setValue:op.URL.path forKey:@"logFile"];
+
+}
+
+- (void)runBiportal:(char const**)args {
+    FILE *f=NULL;
+    AuthorizationRef a=nil;
+    @try {
+	NSString *bip=[[NSBundle mainBundle] pathForAuxiliaryExecutable:@"biportal"];
+	struct stat st;
+	if(stat(bip.UTF8String, &st)) [NSException raise:@"ToolFailure" format:@"Can't see my tool"];
+	if(st.st_uid || !(st.st_mode&S_ISUID)) {
+	    OSStatus r = AuthorizationCreate(NULL, kAuthorizationEmptyEnvironment, kAuthorizationFlagDefaults, &a);
+	    if(r!=errAuthorizationSuccess)
+		[NSException raise:@"AuthFailure" format:@"failed to AuthorizationCreate(): %d",r];
+	    AuthorizationItem ai = {kAuthorizationRightExecute,0,NULL,0};
+	    AuthorizationRights ar = {1,&ai};
+	    r = AuthorizationCopyRights(a, &ar, NULL, kAuthorizationFlagDefaults|kAuthorizationFlagInteractionAllowed|kAuthorizationFlagPreAuthorize|kAuthorizationFlagExtendRights, NULL);
+	    if(r!=errAuthorizationSuccess)
+		[NSException raise:@"AuthFailure" format:@"failed to AuthorizationCopyRights(): %d",r];
+	    const char *args[] = { NULL };
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wdeprecated"
+	    r = AuthorizationExecuteWithPrivileges(a,bip.UTF8String,
+						   kAuthorizationFlagDefaults, (char*const*)args, &f);
+#pragma GCC diagnostic pop
+	    if(r!=errAuthorizationSuccess)
+		[NSException raise:@"AuthFailure" format:@"failed to AuthorizationExecuteWithPrivileges(): %d",r];
+	    int e;
+	    int sr = fscanf(f, "%d", &e);
+	    fclose(f),f=NULL;
+	    if(sr!=1)
+		[NSException raise:@"ToolFailure" format:@"failed to setup tool"];
+	    if(e)
+		[NSException raise:@"ToolFailure" format:@"failed to setup tool, error code: %d",e];
+	}
+	*args = bip.UTF8String;
+	pid_t p = fork();
+	if(p<0) [NSException raise:@"ToolFailure" format:@"failed to fork"];
+	if(!p) execv(*args,(char**)args), exit(errno);
+	int r, wp;
+	while((wp=waitpid(p,&r,0))<0 && errno==EINTR);
+	if(wp!=p) [NSException raise:@"ToolFailure" format:@"failed to wait for tool"];
+	if(!WIFEXITED(r)) [NSException raise:@"ToolFailure" format:@"tool failed"];
+	if(WEXITSTATUS(r)) {
+	    [[NSException exceptionWithName:@"ToolFailure" reason:[NSString stringWithFormat:@"tool failed, error code: %d", WEXITSTATUS(r)] userInfo:@{@"errno": @WEXITSTATUS(r)}] raise];
+	}
+    }@finally {
+	if(f) fclose(f);
+	if(a) AuthorizationFree(a,kAuthorizationFlagDefaults);
+    }
+}
+
++(void)initialize {
+    [[NSUserDefaultsController sharedUserDefaultsController] setInitialValues:
+     [NSDictionary dictionaryWithContentsOfFile:
+      [[NSBundle mainBundle] pathForResource:@"pumpkin-defaults" ofType:@"plist"]
+     ]
+    ];
+    [NSValueTransformer setValueTransformer:[[[NumberTransformer alloc] init] autorelease] forName:@"PortNumberTransformer"];
+    [NSValueTransformer setValueTransformer:[[[IPTransformer alloc] init] autorelease] forName:@"IPAddressTransformer"];
+}
+
+@end