客户端:
导入头文件:
#import <sys/socket.h>
#import <netinet/in.h>
#import <arpa/inet.h>
#import <unistd.h>
1. 创建连接
CFSocketContext sockContext = {0, // 结构体的版本,必须为0
self,
// 一个任意指针的数据,可以用在创建时CFSocket对象相关联。这个指针被传递给所有的上下文中定义的回调。
NULL, // 一个定义在上面指针中的retain的回调, 可以为NULL
NULL, NULL};
CFSocketRef _socket = (kCFAllocatorDefault, // 为新对象分配内存,可以为nil
PF_INET, // 协议族,如果为0或者负数,则默认为PF_INET
SOCK_STREAM, // 套接字类型,如果协议族为PF_INET,则它会默认为SOCK_STREAM
IPPROTO_TCP, // 套接字协议,如果协议族是PF_INET且协议是0或者负数,它会默认为IPPROTO_TCP
kCFSocketConnectCallBack, // 触发回调函数的socket消息类型,具体见Callback Types
TCPServerConnectCallBack, // 上面情况下触发的回调函数
&sockContext // 一个持有CFSocket结构信息的对象,可以为nil
);
if (_socket != nil) {
struct sockaddr_in addr4; // IPV4
memset(&addr4, 0, sizeof(addr4));
addr4.sin_len = sizeof(addr4);
addr4.sin_family = AF_INET;
addr4.sin_port = htons(8888);
addr4.sin_addr.s_addr = inet_addr([strAddress UTF8String]); // 把字符串的地址转换为机器可识别的网络地址
// 把sockaddr_in结构体中的地址转换为Data
CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4));
CFSocketConnectToAddress(_socket, // 连接的socket
address, // CFDataRef类型的包含上面socket的远程地址的对象
-1 // 连接超时时间,如果为负,则不尝试连接,而是把连接放在后台进行,如果_socket消息类型为kCFSocketConnectCallBack,将会在连接成功或失败的时候在后台触发回调函数
);
CFRunLoopRef cRunRef = CFRunLoopGetCurrent(); // 获取当前线程的循环
// 创建一个循环,但并没有真正加如到循环中,需要调用CFRunLoopAddSource
CFRunLoopSourceRef sourceRef = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0);
CFRunLoopAddSource(cRunRef, // 运行循环
sourceRef, // 增加的运行循环源, 它会被retain一次
kCFRunLoopCommonModes // 增加的运行循环源的模式
);
CFRelease(courceRef);
}
2. 设置回调函数
1 // socket回调函数的格式: 2 static void TCPServerConnectCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) { 3 if (data != NULL) { 4 // 当socket为kCFSocketConnectCallBack时,失败时回调失败会返回一个错误代码指针,其他情况返回NULL 5 NSLog(@"连接失败"); 6 return; 7 } 8 TCPClient *client = (TCPClient *)info; 9 // 读取接收的数据 10 [info performSlectorInBackground:@selector(readStream) withObject:nil]; 11 } 12 3. 接收发送数据 13 // 读取接收的数据 14 - (void)readStream { 15 char buffer[1024]; 16 NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init]; 17 while (recv(CFSocketGetNative(_socket), //与本机关联的Socket 如果已经失效返回-1:INVALID_SOCKET 18 buffer, sizeof(buffer), 0)) { 19 NSLog(@"%@", [NSString stringWithUTF8String:buffer]); 20 } 21 } 22 // 发送数据 23 - (void)sendMessage { 24 NSString *stringTosend = @"你好"; 25 char *data = [stringTosend UTF8String]; 26 send(SFSocketGetNative(_socket), data, strlen(data) + 1, 0); 27 } 28 服务器端: 29 CFSockteRef _socket; 30 CFWriteStreamRef outputStream = NULL; 31 int setupSocket() { 32 _socket = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketAcceptCallBack, TCPServerAcceptCallBack, NULL); 33 if (NULL == _socket) { 34 NSLog(@"Cannot create socket!"); 35 return 0; 36 } 37 38 int optval = 1; 39 setsockopt(CFSocketGetNative(_socket), SOL_SOCKET, SO_REUSEADDR, // 允许重用本地地址和端口 40 (void *)&optval, sizeof(optval)); 41 42 struct sockaddr_in addr4; 43 memset(&addr4, 0, sizeof(addr4)); 44 addr4.sin_len = sizeof(addr4); 45 addr4.sin_family = AF_INET; 46 addr4.sin_port = htons(port); 47 addr4.sin_addr.s_addr = htonl(INADDR_ANY); 48 CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4)); 49 50 if (kCFSocketSuccess != CFSocketSetAddress(_socket, address)) { 51 NSLog(@"Bind to address failed!"); 52 if (_socket) 53 CFRelease(_socket); 54 _socket = NULL; 55 return 0; 56 } 57 58 CFRunLoopRef cfRunLoop = CFRunLoopGetCurrent(); 59 CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0); 60 CFRunLoopAddSource(cfRunLoop, source, kCFRunLoopCommonModes); 61 CFRelease(source); 62 63 return 1; 64 } 65 // socket回调函数,同客户端 66 void TCPServerAcceptCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) { 67 if (kCFSocketAcceptCallBack == type) { 68 // 本地套接字句柄 69 CFSocketNativeHandle nativeSocketHandle = *(CFSocketNativeHandle *)data; 70 uint8_t name[SOCK_MAXADDRLEN]; 71 socklen_t nameLen = sizeof(name); 72 if (0 != getpeername(nativeSocketHandle, (struct sockaddr *)name, &nameLen)) { 73 NSLog(@"error"); 74 exit(1); 75 } 76 NSLog(@"%@ connected.", inet_ntoa( ((struct sockaddr_in *)name)->sin_addr )): 77 CFReadStreamRef iStream; 78 CFWriteStreamRef oStream; 79 // 创建一个可读写的socket连接 80 CFStreamCreatePairWithSocket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream); 81 if (iStream && oStream) { 82 CFStreamClientContext streamContext = {0, NULL, NULL, NULL}; 83 if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAvaiable, 84 readStream, // 回调函数,当有可读的数据时调用 85 &streamContext)){ 86 exit(1); 87 } 88 if (!CFReadStreamSetClient(iStream, kCFStreamEventCanAcceptBytes, writeStream, &streamContext)){ 89 exit(1); 90 } 91 CFReadStreamScheduleWithRunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes); 92 CFWriteStreamScheduleWithRunLoop(wStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes); 93 CFReadStreamOpen(iStream); 94 CFWriteStreamOpen(wStream); 95 } else { 96 close(nativeSocketHandle); 97 } 98 } 99 } 100 // 读取数据 101 void readStream(CFReadStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) { 102 UInt8 buff[255]; 103 CFReadStreamRead(stream, buff, 255); 104 printf("received: %s", buff); 105 } 106 void writeStream (CFWriteStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) { 107 outputStream = stream; 108 } 109 main { 110 char *str = "nihao"; 111 112 if (outputStream != NULL) { 113 CFWriteStreamWrite(outputStream, str, strlen(line) + 1); 114 } else { 115 NSLog(@"Cannot send data!"); 116 } 117 } 118 // 开辟一个线程线程函数中 119 void runLoopInThread() { 120 int res = setupSocket(); 121 if (!res) { 122 exit(1); 123 } 124 CFRunLoopRun(); // 运行当前线程的CFRunLoop对象 125 }