summaryrefslogtreecommitdiffabout
path: root/pumpkin/PumpKIN.m
authorMichael Krelin <hacker@klever.net>2012-12-08 21:19:17 (UTC)
committer Michael Krelin <hacker@klever.net>2012-12-11 21:59:29 (UTC)
commit8808689fe340bec6e90ab13dd502292b0579cf1f (patch) (side-by-side diff)
tree45b7c863151341f687b74e40bffcbd7ab5468783 /pumpkin/PumpKIN.m
parent6e7e413ca364d79673e523c09767c18e7cff1bec (diff)
downloadpumpkin-osx/0.0.zip
pumpkin-osx/0.0.tar.gz
pumpkin-osx/0.0.tar.bz2
initial osx portosx/0.0
Signed-off-by: Michael Krelin <hacker@klever.net>
Diffstat (limited to 'pumpkin/PumpKIN.m') (more/less context) (show whitespace changes)
-rw-r--r--pumpkin/PumpKIN.m301
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