initial osx portosx/0.0

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

1 files changed, 156 insertions, 0 deletions

diff --git a/pumpkin/SendXFer.m b/pumpkin/SendXFer.m
new file mode 100644
index 0000000..9a1d85b
--- a/dev/null
+++ b/pumpkin/SendXFer.m
@@ -0,0 +1,156 @@
+#import "SendXFer.h"
+#import "StringsAttached.h"
+#import "ConfirmRequest.h"
+
+@implementation SendXFer
+
+-(SendXFer*)initWithLocalFile:(NSString *)lf peerAddress:(const struct sockaddr_in *)pa remoteFile:(NSString *)rf xferType:(NSString *)xt blockSize:(uint16_t)bs andTimeout:(int)to {
+    if(!(self = [super init])) return self;
+    xferPrefix = @"⬆";
+    retryTimeout = to;
+    localFile = lf;
+    memmove(&peer,pa,sizeof(peer));
+    if(!(theFile = [[NSFileHandle fileHandleForReadingAtPath:localFile] retain])) {
+	[pumpkin log:@"Failed to open '%@', transfer aborted.",localFile];
+	return self;
+    }
+    
+    long xb = ((xferSize=[theFile seekToEndOfFile])/blockSize)+1;
+    if(xb > UINT16_MAX) {
+	[pumpkin log:@"file is too big (%lld bytes) and will take %ld blocks to be sent with block size of %d bytes",xferSize,xb,blockSize];
+	return self;
+    }
+    xferBlocks = xb;
+    
+    [self createSocket];
+    NSMutableDictionary *o = [NSMutableDictionary dictionaryWithCapacity:4];
+    [o setValue:[NSString stringWithFormat:@"%u",bs] forKey:@"blksize"];
+    [o setValue:[NSString stringWithFormat:@"%llu",xferSize] forKey:@"tsize"];
+    [o setValue:[NSString stringWithFormat:@"%d",(int)retryTimeout] forKey:@"timeout"];
+    state = xferStateConnecting;
+    [self queuePacket:[TFTPPacket packetWRQWithFile:xferFilename=rf xferType:xferType=xt andOptions:o]];
+    [self appear];
+    return self;
+}
+
+-(SendXFer*)initWithPeer:(struct sockaddr_in *)sin andPacket:(TFTPPacket*)p {
+    if(!(self = [super initWithPeer:sin andPacket:p])) return self;
+    xferPrefix = @"⬆";
+    xferFilename = [p.rqFilename retain]; xferType = [p.rqType retain];
+    [pumpkin log:@"'%@' of type '%@' is requested from %@",
+		xferFilename, xferType, [NSString stringWithSocketAddress:&peer] ];
+
+    [self createSocket];
+    [self appear];
+
+    if(![self makeLocalFileName:xferFilename])
+	return self;
+
+    switch([[pumpkin.theDefaults.values valueForKey:@"rrqBehavior"] intValue]) {
+	case onRRQDeny: [self goOnWithVerdict:verdictDeny]; break;
+	case onRRQGive: [self goOnWithVerdict:verdictAllow]; break;
+	default: 
+	    [ConfirmRequest confirmationWithXfer:self];
+	    break;
+    }
+    return self;
+}
+-(void)goOnWithVerdict:(int)verdict {
+    if(verdict!=verdictAllow) {
+	[self queuePacket:[TFTPPacket packetErrorWithCode:tftpErrAccessViolation andMessage:@"Access denied"]];
+	return;
+    }
+    if(!(theFile = [[NSFileHandle fileHandleForReadingAtPath:localFile] retain])) {
+	[self queuePacket:[TFTPPacket packetErrorWithErrno:errno andFallback:@"couldn't open file"]];
+	return;
+    }
+    xferSize = [theFile seekToEndOfFile];
+    NSMutableDictionary *o = [NSMutableDictionary dictionaryWithCapacity:4];
+    [[initialPacket rqOptions] enumerateKeysAndObjectsUsingBlock:^(NSString* k, NSString* v, BOOL *stop) {
+	if([k isEqualToString:@"blksize"]) {
+	    [o setValue:[NSString stringWithFormat:@"%u",blockSize=v.intValue] forKey:@"blksize"];
+	}else if([k isEqualToString:@"tsize"]) {
+	    [o setValue:[NSString stringWithFormat:@"%lld",xferSize] forKey:@"tsize"];
+	}else if([k isEqualToString:@"timeout"]) {
+	    [o setValue:[NSString stringWithFormat:@"%d",v.intValue] forKey:@"timeout"];
+	    retryTimeout = v.intValue;
+	}else
+	    [pumpkin log:@"Unknown option '%@' with value '%@'. Ignoring.",k,v];
+    }];
+    long xb = (xferSize/blockSize)+1;
+    if(xb > UINT16_MAX) {
+	[self queuePacket:[TFTPPacket packetErrorWithCode:tftpErrUndefined andMessage:[NSString stringWithFormat:@"file is too big (%lld bytes) and will take %ld blocks to be sent with block size of %d bytes",xferSize,xb,blockSize]]];
+	return;
+    }
+    xferBlocks = xb;
+    state = xferStateXfer;
+    if(o.count) {
+	[self queuePacket:[TFTPPacket packetOACKWithOptions:o]];
+    }else{
+	[self xfer];
+    }
+}
+
+- (void) xfer {
+    NSAssert(theFile,@"no file!");
+    [theFile seekToFileOffset:acked*blockSize];
+    [self queuePacket:[TFTPPacket packetDataWithBlock:acked+1 andData:[theFile readDataOfLength:blockSize]]];
+}
+
+- (void) eatTFTPPacket:(TFTPPacket*)p from:(struct sockaddr_in*)sin{
+    if(state==xferStateConnecting) {
+	peer.sin_port = sin->sin_port;
+	[self updateView];
+    }else if(![self isPeer:sin]) {
+	[pumpkin log:@"Packet from unexpected source (%@) recevied",[NSString stringWithSocketAddress:sin]];
+	return;
+    }
+    switch(p.op) {
+	case tftpOpACK:
+	    if(state==xferStateShutdown || ( (acked=p.block)==xferBlocks && (state=xferStateShutdown) ) ) {
+		CFSocketEnableCallBacks(sockie, kCFSocketWriteCallBack);
+		return;
+	    }
+	    [self updateView];
+	    [self xfer];
+	    break;
+	case tftpOpERROR:
+	    [pumpkin log:@"Error %u:%@",p.rqCode, p.rqMessage];
+	    [self updateView];
+	    [self disappear];
+	    return;
+	case tftpOpOACK:
+	    if(acked) {
+		[pumpkin log:@"It's a bit too late to acknowledge options, ignoring OACK packet"];
+		break;
+	    }
+	{
+	    __block BOOL a=NO;
+	    [p.rqOptions enumerateKeysAndObjectsUsingBlock:^(NSString *k,NSString *v,BOOL *s) {
+		if([k isEqualToString:@"blksize"])
+		    blockSize = v.intValue;
+		else if([k isEqualToString:@"tsize"]) {
+		}else if([k isEqualToString:@"timeout"])
+		    retryTimeout = v.intValue;
+		else{
+		    [pumpkin log:@"Totally unknown option '%@' with value '%@' acknowledged by peer",k,v];
+		    a=YES;
+		}
+	    }];
+	    if(a) {
+		[self abort];
+		break;
+	    }
+	    state = xferStateXfer;
+	    [self updateView];
+	    [self xfer];
+	}
+	    break;
+	default:
+	    [pumpkin log:@"Totaly unexpected opcode %d received",p.op];
+	    break;
+    }
+}
+
+
+@end