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:nil];
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] init];
45 readPipe = [[ITByteStream alloc] init];
49 actionflag = dieflag = 0;
54 -(id)initWithDelegate:(id <ITInetSocketDelegate,NSObject>)d
56 if (self = [super init])
58 state = ITInetSocketDisconnected;
60 delegate = [d retain];
62 writePipe = [[ITByteStream alloc] init];
63 readPipe = [[ITByteStream alloc] init];
67 actionflag = dieflag = 0;
79 if (!sarr) freeaddrinfo(ai);
83 -(void) connectToHost:(NSString*)host onPort:(short)thePort
85 if (state == ITInetSocketDisconnected)
87 NSString *nport = [[NSNumber numberWithShort:thePort] stringValue];
88 [self doConnSetupWithHost:host namedPort:nport];
92 -(void) connectToHost:(NSString*)host onNamedPort:(NSString*)_port
94 if (state == ITInetSocketDisconnected)
96 [self doConnSetupWithHost:host namedPort:_port];
100 -(void) connectWithSockaddrArray:(NSArray*)arr
102 if (state == ITInetSocketDisconnected)
104 NSEnumerator *e = [arr objectEnumerator];
107 ai = malloc(sizeof(struct addrinfo));
109 while (d = [e nextObject])
111 struct sockaddr *s = (struct sockaddr*)[d bytes];
112 a->ai_family = s->sa_family;
114 a->ai_next = malloc(sizeof(struct addrinfo));
123 do {} while (dieflag == 1);
126 -(void)retryConnection
128 ai_cur = ai_cur->ai_next?ai_cur->ai_next:ai_cur;
130 [self realDoConnection];
133 -(ITInetSocketState)state
137 -(id <ITInetSocketDelegate>)delegate
142 -(unsigned short)bufferSize
147 -(void)setBufferSize:(unsigned short)_bufs
153 @implementation ITInetSocket(Debugging)
154 -(NSString*)dumpv6Addrinfo:(struct addrinfo *)_ai
169 "\tAddr = {%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx}\n"
174 NSString *nsfmt = [NSString stringWithCString:cfmt];
175 struct sockaddr_in6 *sa = (struct sockaddr_in6 *)_ai->ai_addr;
176 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"];
182 @implementation ITInetSocket(Private)
183 -(void)doConnSetupWithHost:(NSString*)host namedPort:(NSString*)namedPort
185 struct addrinfo hints;
187 const char *portNam = [namedPort cString], *hostCStr = [host cString];
190 hints.ai_family = PF_UNSPEC;
191 hints.ai_socktype = SOCK_STREAM;
192 hints.ai_protocol = IPPROTO_TCP;
193 hints.ai_addrlen = 0;
194 hints.ai_canonname = NULL;
195 hints.ai_addr = NULL;
196 hints.ai_next = NULL;
198 err = getaddrinfo(hostCStr,portNam,&hints,&ai);
200 NSLog(@"%s, h %@ p %@",gai_strerror(err),host,namedPort);
201 NSLog(ai?[self dumpv6Addrinfo:ai]:@"");
203 [self realDoConnection];
206 -(void)realDoConnection
208 sockfd = socket(ai_cur->ai_addr->sa_family,SOCK_STREAM,IPPROTO_TCP);
209 [self spinoffReadLoop];
212 -(void)spinoffReadLoop
214 NSPort *p1 = [NSPort port], *p2 = [NSPort port];
215 NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:p1 sendPort:p2];
216 NSArray *par = [NSArray arrayWithObjects:p2,p1,nil];
217 [dcon setRootObject:delegate];
218 [NSThread detachNewThreadSelector:@selector(socketReadLoop:) toTarget:self withObject:par]; //spawn read thread
221 -(void)socketReadLoop:(id)data
223 NSAutoreleasePool *ap = [[NSAutoreleasePool alloc] init];
224 NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:[data objectAtIndex:0] sendPort:[data objectAtIndex:1]];
225 NSProxy *dp = [dcon rootProxy];
226 char *buf = malloc(bufs);
227 unsigned long readLen = 0;
229 NSLog(@"EYE MAEK CONNECT");
230 err = connect(sockfd,ai_cur->ai_addr,ai_cur->ai_addrlen);
234 [(id)dp errorOccured:ITInetCouldNotConnect during:ITInetSocketConnecting onSocket:self];
237 [(id)dp finishedConnecting:self];
240 while (!actionflag && ![writePipe availableDataLength])
243 readLen = recv(sockfd,buf,bufs,0);
246 [d initWithBytesNoCopy:buf length:readLen];
247 [readPipe writeData:d];
249 [(id)dp dataReceived:self];
268 NSData *d = [writePipe readAllData];
269 write(sockfd,[d bytes],[d length]);