Objective-C TCP 通訊實例

TCP通訊實例

Communicator.h

#import <Foundation/Foundation.h>

@interface Communicator : NSObject <NSStreamDelegate> {
@public

NSString *host;
int port;
}

- (void)setup;
- (void)open;
- (void)close;
- (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)event;
- (void)readIn:(NSString *)s;
- (void)writeOut:(NSString *)s;

@end

Communicator.m

#import "Communicator.h"

CFReadStreamRef readStream;
CFWriteStreamRef writeStream;

NSInputStream *inputStream;
NSOutputStream *outputStream;

@implementation Communicator

- (void)setup {
NSURL *url = [NSURL URLWithString:host];

NSLog(@"Setting up connection to %@ : %i", [url absoluteString], port);

CFStreamCreatePairWithSocketToHost(kCFAllocatorDefault, (CFStringRef)[url host], port, &readStream, &writeStream);

if(!CFWriteStreamOpen(writeStream)) {
NSLog(@"Error, writeStream not open");

return;
}
[self open];

NSLog(@"Status of outputStream: %i", [outputStream streamStatus]);

return;
}

- (void)open {
NSLog(@"Opening streams.");

inputStream = (NSInputStream *)readStream;
outputStream = (NSOutputStream *)writeStream;

[inputStream retain];
[outputStream retain];

[inputStream setDelegate:self];
[outputStream setDelegate:self];

[inputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
[outputStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];

[inputStream open];
[outputStream open];
}

- (void)close {
NSLog(@"Closing streams.");

[inputStream close];
[outputStream close];

[inputStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
[outputStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];

[inputStream setDelegate:nil];
[outputStream setDelegate:nil];

[inputStream release];
[outputStream release];

inputStream = nil;
outputStream = nil;
}

- (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)event {
NSLog(@"Stream triggered.");

switch(event) {
case NSStreamEventHasSpaceAvailable: {
if(stream == outputStream) {
NSLog(@"outputStream is ready.");
}
break;
}
case NSStreamEventHasBytesAvailable: {
if(stream == inputStream) {
NSLog(@"inputStream is ready.");

uint8_t buf[1024];
unsigned int len = 0;

len = [inputStream read:buf maxLength:1024];

if(len > 0) {
NSMutableData* data=[[NSMutableData alloc] initWithLength:0];

[data appendBytes: (const void *)buf length:len];

NSString *s = [[NSString alloc] initWithData:data encoding:NSASCIIStringEncoding];

[self readIn:s];

[data release];
}
}
break;
}
default: {
NSLog(@"Stream is sending an Event: %i", event);

break;
}
}
}

- (void)readIn:(NSString *)s {
NSLog(@"Reading in the following:");
NSLog(@"%@", s);
}

- (void)writeOut:(NSString *)s {
uint8_t *buf = (uint8_t *)[s UTF8String];

[outputStream write:buf maxLength:strlen((char *)buf)];

NSLog(@"Writing out the following:");
NSLog(@"%@", s);
}

@end