1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
|
#import "XFer.h"
#import "TFTPPacket.h"
#import "StringsAttached.h"
static void cbXfer(CFSocketRef sockie,CFSocketCallBackType cbt,CFDataRef cba,
const void *cbd,void *i) {
[(XFer*)i callbackWithType:cbt addr:cba data:cbd];
}
@implementation XFer
@synthesize initialPacket;
@synthesize xferFilename;
@synthesize localFile;
@synthesize xferPrefix;
- (id) init {
if(!(self = [super init])) return self;
blockSize = 512;
sockie = NULL;
theFile = nil;
acked = 0;
xferSize = 0; xferBlocks = 0;
xferType = nil; xferFilename = nil;
state = xferStateNone;
pumpkin = NSApplication.sharedApplication.delegate;
queue = [[NSMutableArray alloc]initWithCapacity:4];
localFile = nil;
retryTimeout = 3;
giveupTimeout = [[[[NSUserDefaultsController sharedUserDefaultsController] values] valueForKey:@"giveUpTimeout"] intValue];
lastPacket = nil; retryTimer = nil;
giveupTimer = nil;
initialPacket = nil;
return self;
}
- (id) initWithPeer:(struct sockaddr_in *)sin andPacket:(TFTPPacket*)p {
if(!(self=[self init])) return self;
memmove(&peer,sin,sizeof(peer));
initialPacket = [p retain];
return self;
}
- (struct sockaddr_in*)peer { return &peer; }
- (BOOL) makeLocalFileName:(NSString *)xf {
NSString *fn = [xf stringByReplacingOccurrencesOfString:@"\\" withString:@"/"];
if([fn hasPrefix:@"../"] || [fn hasSuffix:@"/.."] || [fn rangeOfString:@"/../"].location!=NSNotFound) {
[self queuePacket:[TFTPPacket packetErrorWithCode:tftpErrAccessViolation andMessage:@"bad path"]];
return NO;
}
localFile = [[[pumpkin.theDefaults.values valueForKey:@"tftpRoot"] stringByAppendingPathComponent:fn] retain];
return YES;
}
- (void) retryTimeout {
[self queuePacket:lastPacket]; [lastPacket release]; lastPacket = nil;
}
- (void) giveUp {
[pumpkin log:@"Connection timeout for '%@'",xferFilename];
[self abort];
}
- (void) renewHope {
if(giveupTimer) {
[giveupTimer invalidate]; [giveupTimer release];
}
giveupTimer = [[NSTimer scheduledTimerWithTimeInterval:giveupTimeout target:self selector:@selector(giveUp) userInfo:nil repeats:NO] retain];
}
- (void) callbackWithType:(CFSocketCallBackType)t addr:(CFDataRef)a data:(const void *)d {
if(!giveupTimer) [self renewHope];
if(retryTimer) {
[retryTimer release]; [retryTimer invalidate]; retryTimer = nil;
}
switch (t) {
case kCFSocketWriteCallBack:
if(queue.count) {
TFTPPacket *p = queue[0];
CFSocketError r = CFSocketSendData(sockie, (CFDataRef)[NSData dataWithBytesNoCopy:&peer length:sizeof(peer) freeWhenDone:NO], (CFDataRef)[NSData dataWithData:p.data], 0);
if(r!=kCFSocketSuccess)
[pumpkin log:@"Failed to send data, error %d",errno];
if(!(p.op==tftpOpDATA || p.op==tftpOpERROR)) {
if(lastPacket) [lastPacket release];
lastPacket = [p retain];
if(retryTimer) {
[retryTimer invalidate]; [retryTimer release];
}
retryTimer = [[NSTimer scheduledTimerWithTimeInterval:retryTimeout target:self selector:@selector(retryTimeout) userInfo:nil repeats:NO] retain];
}else{
[lastPacket release]; lastPacket = nil;
}
[queue removeObjectAtIndex:0];
if([queue count] || state==xferStateShutdown)
CFSocketEnableCallBacks(sockie, kCFSocketWriteCallBack);
}else if(state==xferStateShutdown) {
[pumpkin log:@"%@ Transfer of '%@' finished.",xferPrefix,xferFilename];
[self disappear];
}
break;
case kCFSocketDataCallBack:
[self renewHope];
[self eatTFTPPacket:[TFTPPacket packetWithData:(NSData*)d] from:(struct sockaddr_in*)CFDataGetBytePtr(a)];
break;
default:
NSLog(@"unhandled %lu callback",t);
break;
}
}
- (BOOL) createSocket {
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|kCFSocketWriteCallBack|kCFSocketDataCallBack,
cbXfer, &ctx);
if(!sockie) return NO;
struct sockaddr_in a; memset(&a, 0, sizeof(a));
a.sin_family = AF_INET;
if(CFSocketSetAddress(sockie, (CFDataRef)[NSData dataWithBytesNoCopy:&a length:sizeof(a) freeWhenDone:NO])
!=kCFSocketSuccess) {
[pumpkin log:@"failed to set socket address"];
return NO;
}
runloopSource = CFSocketCreateRunLoopSource(kCFAllocatorDefault, sockie, 0);
CFRunLoopAddSource(CFRunLoopGetCurrent(), runloopSource, kCFRunLoopDefaultMode);
return YES;
}
- (void) queuePacket:(TFTPPacket*)p {
[queue addObject:p];
CFSocketEnableCallBacks(sockie, kCFSocketWriteCallBack|kCFSocketReadCallBack);
if(p.op==tftpOpERROR) state = xferStateShutdown;
}
- (void) goOnWithVerdict:(int)verdict {
NSAssert(false,@"unimplemented goOnWithVerdict");
}
- (void) eatTFTPPacket:(TFTPPacket*)p from:(struct sockaddr_in*)sin {
NSAssert(false,@"unimplemented eatTFTPPacket");
}
-(void) abort {
[self queuePacket:[TFTPPacket packetErrorWithCode:tftpErrUndefined andMessage:@"transfer cancelled"]];
}
- (id) cellValueForColumn:(NSString*)ci {
if([ci isEqualToString:@"fileName"]) {
return [NSString stringWithFormat:@"%@ %@",xferPrefix,xferFilename];
}else if([ci isEqualToString:@"xferType"]) {
return xferType;
}else if([ci isEqualToString:@"peerAddress"]) {
switch (state) {
case xferStateConnecting: return [NSString stringWithHostAddress:&peer];
default: return [NSString stringWithSocketAddress:&peer];
}
}else if([ci isEqualToString:@"ackBytes"]) {
return [NSString stringWithFormat:@"%u",acked*blockSize];
}else if([ci isEqualToString:@"xferSize"]) {
return xferSize?[NSString stringWithFormat:@"%llu",xferSize]:nil;
}
return nil;
}
- (void) updateView {
[pumpkin updateXfers];
}
- (void) appear {
[pumpkin registerXfer:self];
}
- (void) disappear {
if(retryTimer) {
[retryTimer invalidate]; [retryTimer release]; retryTimer = nil;
}
if(giveupTimer) {
[giveupTimer invalidate]; [giveupTimer release]; retryTimer = nil;
}
[pumpkin unregisterXfer:self];
}
- (BOOL) isPeer:(struct sockaddr_in*)sin {
return sin->sin_len==peer.sin_len && !memcmp(sin,&peer,sin->sin_len);
}
-(void)dealloc {
if(runloopSource) {
CFRunLoopSourceInvalidate(runloopSource);
CFRelease(runloopSource);
}
if(sockie) {
CFSocketInvalidate(sockie);
CFRelease(sockie);
}
[queue release];
if(theFile) [theFile release];
if(xferFilename) [xferFilename release];
if(xferType) [xferType release];
if(lastPacket) [lastPacket release];
if(initialPacket) [initialPacket release];
if(localFile) [localFile release];
[super dealloc];
}
@end
|
