5 // Created by Alexander Strange on Tue Feb 11 2003.
6 // Copyright (c) 2003 __MyCompanyName__. All rights reserved.
9 #import "ITInetSocket.h"
10 #import "ITServiceBrowserDelegate.h"
11 #import <sys/socket.h>
15 @interface ITInetSocket(Debugging)
16 -(NSString*)dumpv6Addrinfo:(struct addrinfo *)_ai;
19 @interface ITInetSocket(Private)
20 -(void)doConnSetupWithHost:(NSString*)host namedPort:(NSString*)namedPort;
21 -(void)realDoConnection;
22 -(void)spinoffReadLoop;
23 -(void)socketReadLoop:(id)data;
26 @implementation ITInetSocket
27 +(void)startAutoconnectingToService:(NSString*)type delegate:(id <ITInetSocketDelegate,NSObject>)d
29 NSNetServiceBrowser *browse = [[NSNetServiceBrowser alloc] init];
30 ITServiceBrowserDelegate *bd = [[ITServiceBrowserDelegate alloc] initWithDelegate:d];
32 [browse setDelegate:bd];
33 [browse searchForServicesOfType:[NSString stringWithFormat:@"_%@._tcp.",type] inDomain:@""];
36 -(id)initWithFD:(int)fd delegate:(id <ITInetSocketDelegate,NSObject>)d
38 if (self = [super init])
40 state = ITInetSocketListening;
42 delegate = [d retain];
44 writePipe = [[ITByteStream alloc] initWithDelegate:self];
45 readPipe = [[ITByteStream alloc] initWithDelegate:d];
49 actionflag = dieflag = 0;
52 [self spinoffReadLoop];
56 -(id)initWithDelegate:(id <ITInetSocketDelegate,NSObject>)d
58 if (self = [super init])
60 state = ITInetSocketDisconnected;
62 delegate = [d retain];
64 writePipe = [[ITByteStream alloc] initWithDelegate:self];
65 readPipe = [[ITByteStream alloc] initWithDelegate:d];
69 actionflag = dieflag = 0;
82 if (!sarr) freeaddrinfo(ai);
86 -(void) connectToHost:(NSString*)host onPort:(short)thePort
88 if (state == ITInetSocketDisconnected)
90 NSString *nport = [[NSNumber numberWithShort:thePort] stringValue];
91 [self doConnSetupWithHost:host namedPort:nport];
95 -(void) connectToHost:(NSString*)host onNamedPort:(NSString*)_port
97 if (state == ITInetSocketDisconnected)
99 [self doConnSetupWithHost:host namedPort:_port];
103 -(void) connectWithSockaddrArray:(NSArray*)arr
105 if (state == ITInetSocketDisconnected)
107 NSEnumerator *e = [arr objectEnumerator];
109 struct addrinfo *a,*oa;
110 ai = malloc(sizeof(struct addrinfo));
113 bzero(a,sizeof(struct addrinfo));
114 while (d = [e nextObject])
116 struct sockaddr *s = (struct sockaddr*)[d bytes];
117 bzero(a,sizeof(struct addrinfo));
118 a->ai_family = s->sa_family;
120 a->ai_next = malloc(sizeof(struct addrinfo));
126 NSLog(@"Sockaddr connecting....");
127 [self dumpv6Addrinfo:ai];
128 [self realDoConnection];
134 NSLog(@"Got a disconnect");
136 do {} while (dieflag == 1);
139 -(void)retryConnection
141 ai_cur = ai_cur->ai_next?ai_cur->ai_next:ai_cur;
143 [self realDoConnection];
146 -(ITInetSocketState)state
150 -(id <ITInetSocketDelegate>)delegate
155 -(unsigned short)bufferSize
160 -(void)setBufferSize:(unsigned short)_bufs
165 -(void)newDataAdded:(ITByteStream*)sender
169 -(ITByteStream*)readPipe {return readPipe;}
170 -(ITByteStream*)writePipe {return writePipe;}
173 @implementation ITInetSocket(Debugging)
174 -(NSString*)dumpv6Addrinfo:(struct addrinfo *)_ai
189 "\tAddr = {%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx}\n"
194 NSString *nsfmt = [NSString stringWithCString:cfmt];
195 struct sockaddr_in6 *sa = (struct sockaddr_in6 *)_ai->ai_addr;
196 NSString *buf = [[NSMutableString alloc] initWithFormat:nsfmt,_ai->ai_flags,_ai->ai_family,_ai->ai_socktype,_ai->ai_protocol,_ai->ai_canonname?_ai->ai_canonname:"",sa->sin6_len,sa->sin6_family,sa->sin6_port,sa->sin6_flowinfo,sa->sin6_addr.__u6_addr.__u6_addr16[0],sa->sin6_addr.__u6_addr.__u6_addr16[1],sa->sin6_addr.__u6_addr.__u6_addr16[2],sa->sin6_addr.__u6_addr.__u6_addr16[3],sa->sin6_addr.__u6_addr.__u6_addr16[4],sa->sin6_addr.__u6_addr.__u6_addr16[5],sa->sin6_addr.__u6_addr.__u6_addr16[6],sa->sin6_addr.__u6_addr.__u6_addr16[7],sa->sin6_scope_id,_ai->ai_next?[self dumpv6Addrinfo:_ai->ai_next]:@"nil"];
202 @implementation ITInetSocket(Private)
203 -(void)doConnSetupWithHost:(NSString*)host namedPort:(NSString*)namedPort
205 struct addrinfo hints;
207 const char *portNam = [namedPort cString], *hostCStr = [host cString];
208 state = ITInetSocketConnecting;
210 hints.ai_family = PF_UNSPEC;
211 hints.ai_socktype = SOCK_STREAM;
212 hints.ai_protocol = IPPROTO_TCP;
213 hints.ai_addrlen = 0;
214 hints.ai_canonname = NULL;
215 hints.ai_addr = NULL;
216 hints.ai_next = NULL;
218 err = getaddrinfo(hostCStr,portNam,&hints,&ai);
220 NSLog(@"%s, h %@ p %@",gai_strerror(err),host,namedPort);
221 NSLog(ai?[self dumpv6Addrinfo:ai]:@"");
223 [self realDoConnection];
226 -(void)realDoConnection
228 sockfd = socket(ai_cur->ai_family,SOCK_STREAM,IPPROTO_TCP);
229 [self spinoffReadLoop];
232 -(void)spinoffReadLoop
234 NSPort *p1 = [NSPort port], *p2 = [NSPort port];
235 NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:p1 sendPort:p2];
236 NSArray *par = [NSArray arrayWithObjects:p2,p1,nil];
237 [dcon setRootObject:delegate];
238 [NSThread detachNewThreadSelector:@selector(socketReadLoop:) toTarget:self withObject:par]; //spawn read thread
241 -(void)socketReadLoop:(id)data
243 NSAutoreleasePool *ap = [[NSAutoreleasePool alloc] init];
244 NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:[data objectAtIndex:0] sendPort:[data objectAtIndex:1]];
245 NSProxy *dp = [[dcon rootProxy] retain];
246 char *buf = malloc(bufs);
247 unsigned long readLen = 0,wpl = 0;
249 [readPipe setDelegate:(id <ITInetSocketDelegate>)dp];
251 NSLog(@"Connecting");
252 err = connect(sockfd,ai_cur->ai_addr,ai_cur->ai_addrlen);
257 [(id)dp errorOccured:ITInetCouldNotConnect during:ITInetSocketConnecting onSocket:self];
261 NSLog(@"Sending finishedConnecting");
262 [(id)dp finishedConnecting:self];
264 state = ITInetSocketListening;
265 while (!actionflag && !dieflag && !(wpl = CFDataGetLength((CFDataRef)writePipe->data)))
267 readLen = recv(sockfd,buf,bufs,0);
268 state = ITInetSocketReading;
269 if (readLen == -1) {[(id)dp errorOccured:ITInetConnectionDropped during:ITInetSocketReading onSocket:self];goto dieaction;}
272 NSLog(@"Doing writeData to readPipe");
273 [readPipe writeBytes:buf len:readLen];
275 ap = [[NSAutoreleasePool alloc] init];
278 state = ITInetSocketListening;
284 state = ITInetSocketDisconnected;
296 const char *d = CFDataGetBytePtr((CFDataRef)writePipe->data);
297 state = ITInetSocketWriting;
299 [writePipe lockStream];
300 wpl = send(sockfd,d,wpl,0);
301 [writePipe shortenData:wpl];
302 [writePipe unlockStream];
304 ap = [[NSAutoreleasePool alloc] init];