| -rw-r--r-- | pumpkin/PumpKIN.m | 301 |
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 |