1 files changed, 111 insertions, 0 deletions
diff --git a/pumpkin/DaemonListener.m b/pumpkin/DaemonListener.m
new file mode 100644
index 0000000..ab14296
--- a/dev/null
+++ b/pumpkin/DaemonListener.m
@@ -0,0 +1,111 @@
+#import "DaemonListener.h"
+#import "TFTPPacket.h"
+#import "SendXFer.h"
+#import "ReceiveXFer.h"
+#import "StringsAttached.h"
+
+#include <sys/socket.h>
+#include <arpa/inet.h>
+#include <sys/stat.h>
+
+static void cbListener(CFSocketRef sockie,CFSocketCallBackType cbt,CFDataRef cba,
+					   const void *cbd,void *i) {
+    [(DaemonListener*)i callbackWithType:cbt addr:cba data:cbd];
+}
+
+@implementation DaemonListener
+
+-(void)callbackWithType:(CFSocketCallBackType)t addr:(CFDataRef)a data:(const void *)d {
+    switch(t) {
+	case kCFSocketDataCallBack:
+	{
+	    struct sockaddr_in *sin = (struct sockaddr_in*)CFDataGetBytePtr(a);
+	    if([pumpkin hasPeer:sin]) {
+		[pumpkin log:@"I'm already processing the request from %@",[NSString stringWithSocketAddress:sin]];
+		return;
+	    }
+	    TFTPPacket *p = [TFTPPacket packetWithData:(NSData*)d];
+	    switch([p op]) {
+		case tftpOpRRQ: [[[SendXFer alloc] initWithPeer:sin andPacket:p] autorelease]; break;
+		case tftpOpWRQ: [[[ReceiveXFer alloc] initWithPeer:sin andPacket:p] autorelease]; break;
+		default:
+		    [pumpkin log:@"Invalid OP %d received from %@",p.op,[NSString stringWithSocketAddress:sin]];
+		    break;
+	    }
+	}
+	    break;
+	default:
+	    NSLog(@"unhandled callback: %lu",t);
+	    break;
+    }
+}
+
+
+-(DaemonListener*)initWithAddress:(struct sockaddr_in*)sin {
+    if(!(self=[super init])) return self;
+    
+    pumpkin = NSApplication.sharedApplication.delegate;
+
+    @try {
+	CFSocketContext ctx;
+	ctx.version = 0;
+	ctx.info = self;
+	ctx.retain = 0; ctx.release = 0;
+	ctx.copyDescription = 0;
+	sockie = CFSocketCreate(kCFAllocatorDefault,PF_INET,SOCK_DGRAM,IPPROTO_UDP,
+				kCFSocketReadCallBack|kCFSocketDataCallBack,
+				cbListener,&ctx);
+	if(ntohs(sin->sin_port)>1024) {
+	    NSData *nsd = [NSData dataWithBytes:sin length:sizeof(*sin)];
+	    if(CFSocketSetAddress(sockie, (CFDataRef)nsd))
+		[[NSException exceptionWithName:@"BindFailure"
+					 reason:[NSString stringWithFormat:@"Binding failed, error code: %d", errno]
+				       userInfo:@{@"errno": @errno}
+		  ] raise];
+	}else{
+	    const char *args[] = {
+		0,
+		[[NSString stringWithFormat:@"%d", CFSocketGetNative(sockie)] UTF8String],
+		[[NSString stringWithHostAddress:sin] UTF8String],
+		[[NSString stringWithPortNumber:sin] UTF8String],
+		NULL
+	    };
+	    [pumpkin runBiportal:args];
+	}
+    }@catch(NSException *e) {
+	if(sockie) {
+	    CFSocketInvalidate(sockie);
+	    CFRelease(sockie);
+	}
+	@throw;
+    }
+
+    runloopSource = CFSocketCreateRunLoopSource(kCFAllocatorDefault, sockie, 0);
+    CFRunLoopAddSource(CFRunLoopGetCurrent(),runloopSource, kCFRunLoopDefaultMode);
+    return self;
+}
+
+-(void)dealloc {
+    if(runloopSource) {
+	CFRunLoopRemoveSource(CFRunLoopGetCurrent(), runloopSource, kCFRunLoopDefaultMode);
+	CFRelease(runloopSource);
+    }
+    if(sockie) {
+	CFSocketInvalidate(sockie);
+	CFRelease(sockie);
+    }
+    [super dealloc];
+}
+
++(DaemonListener*) listenerWithDefaults {
+    struct sockaddr_in sin;
+    memset(&sin,0,sizeof(sin));
+    sin.sin_len=sizeof(sin);
+    sin.sin_family=AF_INET;
+    id d = [[NSUserDefaultsController sharedUserDefaultsController] values];
+    sin.sin_port=htons([[d valueForKey:@"bindPort"] intValue]);
+    sin.sin_addr.s_addr=inet_addr([[d valueForKey:@"bindAddress"] UTF8String]);
+    return [[[DaemonListener alloc] initWithAddress:&sin] autorelease];
+}
+
+@end