iOS - CFSocket 的使用
1、CFSocket
苹果对对底层 BSD Socket 进行轻量级的封装（纯 C）。
主要使用的 API：CFSocekt 用于建立连接，CFStream 用于读写数据。
2、基本使用
2.1 Client 客户端
-
TCP 客户端
// 包含头文件 #import <sys/socket.h> #import <netinet/in.h> #import <arpa/inet.h> @property (weak, nonatomic) IBOutlet UITextField *desIPTF; @property (weak, nonatomic) IBOutlet UITextField *desPortTF; @property (weak, nonatomic) IBOutlet UITextView *recvTextView; @property (weak, nonatomic) IBOutlet UITextField *sendTF; @property (nonatomic, assign) CFSocketRef clientSockfd; #pragma mark - 创建 TCP 连接 - (IBAction)connectTCPServer:(id)sender { [NSThread detachNewThreadSelector:@selector(connectTCPSer) toTarget:self withObject:nil]; } - (void)connectTCPSer { BOOL success; // 创建 socket self.clientSockfd = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketConnectCallBack, ServerConnectCallBack, NULL); success = (self.clientSockfd != nil); NSString *logStr = nil; if (success == NO) { logStr = @"创建 socket 失败...\n"; [self showMessage:logStr]; return; } else { logStr = @"创建 socket 成功...\n"; [self showMessage:logStr]; // 服务器地址 struct sockaddr_in ser_addr; memset(&ser_addr, 0, sizeof(ser_addr)); ser_addr.sin_len = sizeof(ser_addr); ser_addr.sin_family = AF_INET; ser_addr.sin_port = htons(_desPortTF.text.intValue); ser_addr.sin_addr.s_addr = inet_addr(_desIPTF.text.UTF8String); CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&ser_addr, sizeof((ser_addr))); // 连接 CFSocketError result = CFSocketConnectToAddress(self.clientSockfd, address, 5); CFRelease(address); success = (result == kCFSocketSuccess); } if (success == NO) { logStr = @"socket 连接失败...\n"; [self showMessage:logStr]; return; } else { logStr = [NSString stringWithFormat:@"socket 连接 %@ 成功...\n", _desIPTF.text]; [self showMessage:logStr]; char buf[2048]; do { // 接收数据 ssize_t recvLen = recv(CFSocketGetNative(self.clientSockfd), buf, sizeof(buf), 0); if (recvLen > 0) { logStr = [NSString stringWithFormat:@"recv：%@\n", [NSString stringWithFormat:@"%s", buf]]; [self showMessage:logStr]; } } while (strcmp(buf, "exit") != 0); CFRunLoopRef cfrl = CFRunLoopGetCurrent(); CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, self.clientSockfd, 0); CFRunLoopAddSource(cfrl, source, kCFRunLoopCommonModes); CFRelease(source); CFRunLoopRun(); } } // 连接成功的回调函数 void ServerConnectCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void * data, void *info) { if (data != NULL) { NSLog(@"connect\n"); } else { NSLog(@"connect success\n"); } } #pragma mark - 发送数据 - (IBAction)btnClick:(id)sender { if (_sendTF.text.length == 0) { return; } else { // 发送数据 ssize_t sendLen = send(CFSocketGetNative(self.clientSockfd), _sendTF.text.UTF8String, strlen(_sendTF.text.UTF8String), 0); if (sendLen > 0) { NSString *logStr = [NSString stringWithFormat:@"send：%@\n", _sendTF.text]; [self showMessage:logStr]; } } } // 显示信息 - (void)showMessage:(NSString *)msg { dispatch_async(dispatch_get_main_queue(), ^{ _recvTextView.text = [_recvTextView.text stringByAppendingString:msg]; }); } // 键盘回收 - (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event { [self.view endEditing:YES]; }
2.2 Server 服务端
-
TCP 服务端
// 包含头文件 #import <sys/socket.h> #import <netinet/in.h> #import <arpa/inet.h> #import <ifaddrs.h> static ViewController *selfClass = nil; @property (weak, nonatomic) IBOutlet UITextField *locIPTF; @property (weak, nonatomic) IBOutlet UITextField *locPortTF; @property (weak, nonatomic) IBOutlet UITextView *recvTextView; @property (weak, nonatomic) IBOutlet UITextField *sendTF; @property (nonatomic, assign) CFSocketRef serverSockfd; @property (nonatomic, assign) CFWriteStreamRef outputStream; - (void)viewDidLoad { [super viewDidLoad]; NSString *ip_addr = [self getIPAddress]; _locIPTF.text = ip_addr; // 函数指针指向本身 selfClass = self; } #pragma mark - 创建 TCP 监听 - (IBAction)createdTCPServer:(id)sender { [sender setTitleColor:[UIColor redColor] forState:UIControlStateNormal]; [NSThread detachNewThreadSelector:@selector(createdTCPSer) toTarget:self withObject:nil]; } - (void)createdTCPSer { BOOL success; // 创建 socket self.serverSockfd = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketAcceptCallBack, TCPServerAcceptCallBack, NULL); success = (self.serverSockfd != NULL); NSString *logStr = nil; if (success == NO) { logStr = @"创建 socket 失败...\n"; [self showMessage:logStr]; return; } else { logStr = @"创建 socket 成功...\n"; [self showMessage:logStr]; int optVal = 1; setsockopt(CFSocketGetNative(self.serverSockfd), SOL_SOCKET, SO_REUSEADDR, (void*)&optVal,sizeof(optVal)); // 本地地址 struct sockaddr_in loc_addr; memset(&loc_addr, 0, sizeof(loc_addr)); loc_addr.sin_family = AF_INET; loc_addr.sin_port = htons(_locPortTF.text.intValue); loc_addr.sin_addr.s_addr = inet_addr(_locIPTF.text.UTF8String); CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8*)&loc_addr, sizeof(loc_addr)); // 绑定 CFSocketError err = CFSocketSetAddress(self.serverSockfd, address); CFRelease(address); success = (err == kCFSocketSuccess); } if (success == NO) { logStr = @"socket 绑定失败...\n"; [self showMessage:logStr]; CFRelease(self.serverSockfd); self.serverSockfd = NULL; return; } else { logStr = @"socket 绑定成功...\n"; [self showMessage:logStr]; CFRunLoopRef cfRunloop = CFRunLoopGetCurrent(); CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, self.serverSockfd, 0); CFRunLoopAddSource(cfRunloop, source, kCFRunLoopCommonModes); CFRelease(source); CFRunLoopRun(); } } // 客户端连接成功的回调函数 void TCPServerAcceptCallBack(CFSocketRef socket ,CFSocketCallBackType type,CFDataRef address,const void *data, void *info) { // 客户端连接 if (kCFSocketAcceptCallBack == type) { // data the handle of socket CFSocketNativeHandle nativeSocketHandle = *(CFSocketNativeHandle *)data; uint8_t name[SOCK_MAXADDRLEN]; socklen_t nameLen = sizeof(name); // 获取对方 socket 信息，还有 getsocketname() 函数用于获取本程序所在 socket 信息 if (getpeername(nativeSocketHandle, (struct sockaddr *)name, &nameLen) != 0) { exit(1); } // 获取连接的客户端信息 struct sockaddr_in * addr_in = (struct sockaddr_in *)name; char *ip_addr = inet_ntoa(addr_in->sin_addr); int ip_port = addr_in->sin_port; NSString *logStr = [NSString stringWithFormat:@"已连接：%s，port：%d\n", ip_addr, ip_port]; [selfClass showMessage:logStr]; // 创建一对输入输出流用于读写数据 CFReadStreamRef iStream; CFWriteStreamRef oStream; // 创建一组可读/写的 CFStreame CFStreamCreatePairWithSocket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream); if (iStream && oStream) { // 打开输入流和输出流 CFReadStreamOpen(iStream); CFWriteStreamOpen(oStream); CFStreamClientContext streamContext = {0, NULL, NULL, NULL}; // if have data to read call the readStream function if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAvailable, readStream, &streamContext)) { exit(1); } CFReadStreamScheduleWithRunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommonModes); selfClass.outputStream = oStream; } } } // 读取数据的回调函数 void readStream(CFReadStreamRef iStream, CFStreamEventType eventType, void * clientCallBackInfo){ char buf[2048]; bzero(buf, sizeof(buf)); do { // 接收数据 CFIndex dex = CFReadStreamRead(iStream, (UInt8 *)buf, sizeof(buf)); if (dex > 0) { NSString *logStr = [NSString stringWithFormat:@"recv：%@\n", [NSString stringWithFormat:@"%s", buf]]; [selfClass showMessage:logStr]; } } while (strcmp(buf, "exit") != 0); } #pragma mark - 发送数据 - (IBAction)btnClick:(id)sender { if (_sendTF.text.length == 0) { return; } else { // 发送数据 CFIndex dex = CFWriteStreamWrite(selfClass.outputStream, (UInt8 *)_sendTF.text.UTF8String, strlen(_sendTF.text.UTF8String)+1); if (dex > 0) { NSString *logStr = [NSString stringWithFormat:@"send：%@\n", _sendTF.text]; [self showMessage:logStr]; } } } #pragma mark - 获取本地 IP 地址 - (NSString *)getIPAddress { NSString *address = @"error"; struct ifaddrs *interfaces = NULL; struct ifaddrs *temp_addr = NULL; int success = 0; // retrieve the current interfaces - returns 0 on success success = getifaddrs(&interfaces); if (success == 0) { // Loop through linked list of interfaces temp_addr = interfaces; while (temp_addr != NULL) { if (temp_addr->ifa_addr->sa_family == AF_INET) { // Check if interface is en0 which is the wifi connection on the iPhone if ([[NSString stringWithUTF8String:temp_addr->ifa_name] isEqualToString:@"en0"]) { // Get NSString from C String address = [NSString stringWithUTF8String:inet_ntoa(((struct sockaddr_in *)temp_addr->ifa_addr)->sin_addr)]; } } temp_addr = temp_addr->ifa_next; } } // Free memory freeifaddrs(interfaces); return address; } // 显示信息 - (void)showMessage:(NSString *)msg { dispatch_async(dispatch_get_main_queue(), ^{ _recvTextView.text = [_recvTextView.text stringByAppendingString:msg]; }); } // 键盘回收 - (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event { [self.view endEditing:YES]; }