X-Git-Url: http://git.ithinksw.org/ITFoundation.git/blobdiff_plain/4883c610168ebeb51a54ae8cce12d24638034b08..f7db9c0e898822c50164af5abda849c16410e035:/ITInetSocket.m diff --git a/ITInetSocket.m b/ITInetSocket.m index 4e40628..c51c05a 100755 --- a/ITInetSocket.m +++ b/ITInetSocket.m @@ -10,7 +10,7 @@ #import "ITServiceBrowserDelegate.h" #import #import - +#import @interface ITInetSocket(Debugging) -(NSString*)dumpv6Addrinfo:(struct addrinfo *)_ai; @@ -18,10 +18,13 @@ @interface ITInetSocket(Private) -(void)doConnSetupWithHost:(NSString*)host namedPort:(NSString*)namedPort; +-(void)realDoConnection; +-(void)spinoffReadLoop; +-(void)socketReadLoop:(id)data; @end @implementation ITInetSocket -+(void)startAutoconnectingToService:(NSString*)type delegate:(id)d ++(void)startAutoconnectingToService:(NSString*)type delegate:(id )d { NSNetServiceBrowser *browse = [[NSNetServiceBrowser alloc] init]; ITServiceBrowserDelegate *bd = [[ITServiceBrowserDelegate alloc] initWithDelegate:d]; @@ -30,7 +33,7 @@ [browse searchForServicesOfType:[NSString stringWithFormat:@"._%@._tcp",type] inDomain:nil]; } --(id)initWithFD:(int)fd delegate:(id)d +-(id)initWithFD:(int)fd delegate:(id )d { if (self = [super init]) { @@ -42,21 +45,13 @@ readPipe = [[ITByteStream alloc] init]; ai = nil; sarr = nil; + bufs = 512; + actionflag = dieflag = 0; } return self; } --(void) dealloc -{ - shutdown(sockfd,2); - [delegate release]; - [writePipe release]; - [readPipe release]; - if (!sarr) freeaddrinfo(ai); - [sarr release]; -} - --(id)initWithDelegate:(id)d +-(id)initWithDelegate:(id )d { if (self = [super init]) { @@ -68,10 +63,23 @@ readPipe = [[ITByteStream alloc] init]; ai = nil; sarr = nil; + bufs = 512; + actionflag = dieflag = 0; } return self; } + +-(void) dealloc +{ + shutdown(sockfd,2); + [delegate release]; + [writePipe release]; + [readPipe release]; + if (!sarr) freeaddrinfo(ai); + [sarr release]; +} + -(void) connectToHost:(NSString*)host onPort:(short)thePort { if (state == ITInetSocketDisconnected) @@ -91,57 +99,82 @@ -(void) connectWithSockaddrArray:(NSArray*)arr { - NSEnumerator *e = [arr objectEnumerator]; - NSData *d; - struct addrinfo *a; - ai = malloc(sizeof(struct addrinfo)); - a = ai; - while (d = [e nextObject]) + if (state == ITInetSocketDisconnected) + { + NSEnumerator *e = [arr objectEnumerator]; + NSData *d; + struct addrinfo *a; + ai = malloc(sizeof(struct addrinfo)); + a = ai; + while (d = [e nextObject]) { - struct sockaddr *s = (struct sockaddr*)[d bytes]; - a->ai_family = s->sa_family; - a->ai_addr = s; - a->ai_next = malloc(sizeof(struct addrinfo)); - a = a->ai_next; + struct sockaddr *s = (struct sockaddr*)[d bytes]; + a->ai_family = s->sa_family; + a->ai_addr = s; + a->ai_next = malloc(sizeof(struct addrinfo)); + a = a->ai_next; } + } +} + +-(void)disconnect +{ + dieflag = 1; + do {} while (dieflag == 1); +} + +-(void)retryConnection +{ + ai_cur = ai_cur->ai_next?ai_cur->ai_next:ai_cur; + [self disconnect]; + [self realDoConnection]; } -(ITInetSocketState)state { return state; } +-(id )delegate +{ + return delegate; +} + +-(unsigned short)bufferSize +{ + return bufs; +} + +-(void)setBufferSize:(unsigned short)_bufs +{ + bufs = _bufs; +} @end @implementation ITInetSocket(Debugging) -(NSString*)dumpv6Addrinfo:(struct addrinfo *)_ai { const char *cfmt = - "\{\n" + "{\n" "Flags = %x\n" "Family = %x\n" "Socktype = %x\n" "Protocol = %x\n" "Canonname = %s\n" "Sockaddr = \n" - "{\n" + "\t{\n" "\tLength = %x\n" "\tFamily = %x\n" "\tPort = %d\n" "\tFlowinfo = %x\n" "\tAddr = {%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx}\n" "\tScope = %x\n" - "}\n" - "Next = "; + "\t}\n" + "Next = %@\n" + "}\n"; NSString *nsfmt = [NSString stringWithCString:cfmt]; - NSMutableString *buf = [[[NSMutableString alloc] init] autorelease]; + struct sockaddr_in6 *sa = (struct sockaddr_in6 *)_ai->ai_addr; + 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"]; - do - { - struct sockaddr_in6 *sa = (struct sockaddr_in6 *)_ai->ai_addr; - [buf appendFormat: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]; - } - while (_ai = _ai->ai_next); - [buf appendString:@"nil\n}"]; return buf; } @end @@ -154,18 +187,88 @@ const char *portNam = [namedPort cString], *hostCStr = [host cString]; hints.ai_flags = 0; - hints.ai_family = PF_INET6; + hints.ai_family = PF_UNSPEC; hints.ai_socktype = SOCK_STREAM; hints.ai_protocol = IPPROTO_TCP; + hints.ai_addrlen = 0; hints.ai_canonname = NULL; hints.ai_addr = NULL; hints.ai_next = NULL; err = getaddrinfo(hostCStr,portNam,&hints,&ai); - if (err == EAI_NODATA) //it's a dotted-decimal IPv4 string, so we use v6compat stuff now + + NSLog(@"%s, h %@ p %@",gai_strerror(err),host,namedPort); + NSLog(ai?[self dumpv6Addrinfo:ai]:@""); + ai_cur = ai; + [self realDoConnection]; +} + +-(void)realDoConnection +{ + sockfd = socket(ai_cur->ai_addr->sa_family,SOCK_STREAM,IPPROTO_TCP); + [self spinoffReadLoop]; +} + +-(void)spinoffReadLoop +{ + NSPort *p1 = [NSPort port], *p2 = [NSPort port]; + NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:p1 sendPort:p2]; + NSArray *par = [NSArray arrayWithObjects:p2,p1,nil]; + [dcon setRootObject:delegate]; + [NSThread detachNewThreadSelector:@selector(socketReadLoop:) toTarget:self withObject:par]; //spawn read thread +} + +-(void)socketReadLoop:(id)data +{ + NSAutoreleasePool *ap = [[NSAutoreleasePool alloc] init]; + NSConnection *dcon = [[NSConnection alloc] initWithReceivePort:[data objectAtIndex:0] sendPort:[data objectAtIndex:1]]; + NSProxy *dp = [dcon rootProxy]; + char *buf = malloc(bufs); + unsigned long readLen = 0; + signed int err; + NSLog(@"EYE MAEK CONNECT"); + err = connect(sockfd,ai_cur->ai_addr,ai_cur->ai_addrlen); + if (err == -1) { - err = getaddrinfo([[NSString stringWithFormat:@"ffff::%s",hostCStr] cString],portNam,&hints,&ai); + perror("CAwh"); + [(id)dp errorOccured:ITInetCouldNotConnect during:ITInetSocketConnecting onSocket:self]; + goto dieaction; + } + [(id)dp finishedConnecting:self]; +lstart: + + while (!actionflag && ![writePipe availableDataLength] && !dieflag) + { + NSData *d; + readLen = recv(sockfd,buf,bufs,0); + if (readLen) { + d = [NSData alloc]; + [d initWithBytesNoCopy:buf length:readLen freeWhenDone:NO]; + [readPipe writeData:d]; + [d release]; + [(id)dp dataReceived:self]; + } + } + + actionflag = 0; + + if (dieflag) + { +dieaction: + perror("Awh"); + free(buf); + shutdown(sockfd,2); + [dcon release]; + [ap release]; + dieflag = 0; + return; + } + + { + NSData *d = [writePipe readAllData]; + write(sockfd,[d bytes],[d length]); + goto lstart; } - NSLog([self dumpv6Addrinfo:ai]); + goto dieaction; } @end \ No newline at end of file