Fixed ios multiple sockets streams

This commit is contained in:
kitolog 2017-10-26 18:48:36 +03:00
parent 5f5b0264a1
commit 44dfa14566
3 changed files with 100 additions and 91 deletions

View File

@ -19,6 +19,11 @@
@interface SocketAdapter : NSObject <NSStreamDelegate> { @interface SocketAdapter : NSObject <NSStreamDelegate> {
@public @public
CFReadStreamRef readStream1;
CFWriteStreamRef writeStream1;
NSInputStream *inputStream1;
NSOutputStream *outputStream1;
} }
- (void)open:(NSString *)host port:(NSNumber*)port; - (void)open:(NSString *)host port:(NSNumber*)port;

View File

@ -35,29 +35,33 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
- (void)open:(NSString *)host port:(NSNumber*)port { - (void)open:(NSString *)host port:(NSNumber*)port {
CFReadStreamRef readStream2;
CFWriteStreamRef writeStream2;
NSLog(@"Setting up connection to %@ : %@", host, [port stringValue]); NSLog(@"Setting up connection to %@ : %@", host, [port stringValue]);
if (![self isIp:host]) { if (![self isIp:host]) {
host = [self resolveIp:host]; host = [self resolveIp:host];
} }
CFStreamCreatePairWithSocketToHost(kCFAllocatorDefault, (__bridge CFStringRef)host, [port intValue], &readStream, &writeStream); CFStreamCreatePairWithSocketToHost(kCFAllocatorDefault, (__bridge CFStringRef)host, [port intValue], &readStream2, &writeStream2);
CFReadStreamSetProperty(readStream, kCFStreamPropertyShouldCloseNativeSocket, kCFBooleanTrue); CFReadStreamSetProperty(readStream2, kCFStreamPropertyShouldCloseNativeSocket, kCFBooleanTrue);
CFWriteStreamSetProperty(writeStream, kCFStreamPropertyShouldCloseNativeSocket, kCFBooleanTrue); CFWriteStreamSetProperty(writeStream2, kCFStreamPropertyShouldCloseNativeSocket, kCFBooleanTrue);
if(!CFWriteStreamOpen(writeStream) || !CFReadStreamOpen(readStream)) { if(!CFWriteStreamOpen(writeStream2) || !CFReadStreamOpen(readStream2)) {
NSLog(@"Error, streams not open"); NSLog(@"Error, streams not open");
@throw [NSException exceptionWithName:@"SocketException" reason:@"Cannot open streams." userInfo:nil]; @throw [NSException exceptionWithName:@"SocketException" reason:@"Cannot open streams." userInfo:nil];
} }
inputStream = (__bridge NSInputStream *)readStream; inputStream1 = (__bridge NSInputStream *)readStream2;
[inputStream setDelegate:self]; [inputStream1 setDelegate:self];
[inputStream open]; [inputStream1 open];
outputStream = (__bridge NSOutputStream *)writeStream; outputStream1 = (__bridge NSOutputStream *)writeStream2;
[outputStream open]; [outputStream1 open];
[self performSelectorOnMainThread:@selector(runReadLoop) withObject:nil waitUntilDone:NO]; [self performSelectorOnMainThread:@selector(runReadLoop) withObject:nil waitUntilDone:NO];
} }
@ -96,7 +100,7 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
} }
- (void)runReadLoop { - (void)runReadLoop {
[inputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode]; [inputStream1 scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
} }
- (void)shutdownWrite { - (void)shutdownWrite {
@ -104,20 +108,20 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
[self closeOutputStream]; [self closeOutputStream];
int socket = [self socknumForStream: inputStream]; int socket = [self socknumForStream: inputStream1];
shutdown(socket, 1); shutdown(socket, 1);
} }
- (void)closeInputStream { - (void)closeInputStream {
[inputStream close]; [inputStream1 close];
[inputStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode]; [inputStream1 removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
[inputStream setDelegate:nil]; [inputStream1 setDelegate:nil];
inputStream = nil; inputStream1 = nil;
} }
- (void)closeOutputStream { - (void)closeOutputStream {
[outputStream close]; [outputStream1 close];
outputStream = nil; outputStream1 = nil;
} }
-(int) socknumForStream: (NSStream *)stream -(int) socknumForStream: (NSStream *)stream
@ -142,9 +146,9 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
break; break;
} }
case NSStreamEventHasBytesAvailable: { case NSStreamEventHasBytesAvailable: {
if(stream == inputStream) { if(stream == inputStream1) {
uint8_t buf[65535]; uint8_t buf[65535];
long len = [inputStream read:buf maxLength:65535]; long len = [inputStream1 read:buf maxLength:65535];
if(len > 0) { if(len > 0) {
NSMutableArray *dataArray = [[NSMutableArray alloc] init]; NSMutableArray *dataArray = [[NSMutableArray alloc] init];
@ -159,7 +163,7 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
} }
case NSStreamEventEndEncountered: { case NSStreamEventEndEncountered: {
if(stream == inputStream) { if(stream == inputStream1) {
[self closeInputStream]; [self closeInputStream];
self.closeEventHandler(FALSE); self.closeEventHandler(FALSE);
@ -203,9 +207,9 @@ int const WRITE_BUFFER_SIZE = 10 * 1024;
unsigned char byte = (unsigned char)[[dataArray objectAtIndex:(offset + i)] integerValue]; unsigned char byte = (unsigned char)[[dataArray objectAtIndex:(offset + i)] integerValue];
buf[i] = byte; buf[i] = byte;
} }
NSInteger bytesWritten = [outputStream write:buf maxLength:length]; NSInteger bytesWritten = [outputStream1 write:buf maxLength:length];
if (bytesWritten == -1) { if (bytesWritten == -1) {
@throw [NSException exceptionWithName:@"SocketException" reason:[outputStream.streamError localizedDescription] userInfo:nil]; @throw [NSException exceptionWithName:@"SocketException" reason:[outputStream1.streamError localizedDescription] userInfo:nil];
} }
if (bytesWritten != length) { if (bytesWritten != length) {
[self writeSubarray:dataArray offset:(offset + bytesWritten) length:(length - bytesWritten)]; [self writeSubarray:dataArray offset:(offset + bytesWritten) length:(length - bytesWritten)];