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