IOS系列——socket通信

来源：互联网发布：钱夫人淘宝店编辑：程序博客网时间：2024/06/04 06:20

1、需要导入的头文件

#import <sys/socket.h>#import <netinet/in.h>#import <arpa/inet.h>#import <unistd.h>

2、创建连接

CFSocketContext sockContext = {0, // 结构体的版本，必须为0self,  // 一个任意指针的数据，可以用在创建时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);}

3、设置回调函数

// socket回调函数的格式：static void TCPServerConnectCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) ｛     if (data != NULL) {         // 当socket为kCFSocketConnectCallBack时，失败时回调失败会返回一个错误代码指针，其他情况返回NULL         NSLog(@"连接失败");         return;     }     TCPClient *client = (TCPClient *)info;     // 读取接收的数据    [info performSlectorInBackground:@selector(readStream) withObject:nil]; ｝

4、接收发送数据

// 读取接收的数据 - (void)readStream {     char buffer[1024];     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];     while (recv(CFSocketGetNative(_socket), //与本机关联的Socket 如果已经失效返回－1:INVALID_SOCKET            buffer, sizeof(buffer), 0)) {         NSLog(@"%@", [NSString stringWithUTF8String:buffer]);     } } // 发送数据 - (void)sendMessage {     NSString *stringTosend = @"你好";     char *data = [stringTosend UTF8String];     send(SFSocketGetNative(_socket), data, strlen(data) + 1, 0); }

5、服务器端

CFSockteRef _socket; CFWriteStreamRef outputStream = NULL; int setupSocket() {     _socket = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketAcceptCallBack, TCPServerAcceptCallBack, NULL);     if (NULL == _socket) {         NSLog(@"Cannot create socket!");         return 0;     }         int optval = 1;     setsockopt(CFSocketGetNative(_socket), SOL_SOCKET, SO_REUSEADDR, // 允许重用本地地址和端口 (void *)&optval, sizeof(optval));         struct sockaddr_in addr4;     memset(&addr4, 0, sizeof(addr4));     addr4.sin_len = sizeof(addr4);     addr4.sin_family = AF_INET;     addr4.sin_port = htons(port);     addr4.sin_addr.s_addr = htonl(INADDR_ANY);     CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4));         if (kCFSocketSuccess != CFSocketSetAddress(_socket, address)) {         NSLog(@"Bind to address failed!");         if (_socket)              CFRelease(_socket);         _socket = NULL;         return 0;     }             CFRunLoopRef cfRunLoop = CFRunLoopGetCurrent();     CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0);     CFRunLoopAddSource(cfRunLoop, source, kCFRunLoopCommonModes);     CFRelease(source);         return 1; } // socket回调函数，同客户端 void TCPServerAcceptCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) {     if (kCFSocketAcceptCallBack == type) {         // 本地套接字句柄         CFSocketNativeHandle nativeSocketHandle = *(CFSocketNativeHandle *)data;         uint8_t name[SOCK_MAXADDRLEN];               socklen_t nameLen = sizeof(name);         if (0 != getpeername(nativeSocketHandle, (struct sockaddr *)name, &nameLen)) {             NSLog(@"error");             exit(1);         }         NSLog(@"%@ connected.", inet_ntoa( ((struct sockaddr_in *)name)->sin_addr )):         CFReadStreamRef iStream;         CFWriteStreamRef oStream;         // 创建一个可读写的socket连接         CFStreamCreatePairWithSocket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream);          if (iStream && oStream) {             CFStreamClientContext streamContext = {0, NULL, NULL, NULL};             if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAvaiable,                                        readStream, // 回调函数，当有可读的数据时调用                                        &streamContext)){                 exit(1);             }             if (!CFReadStreamSetClient(iStream, kCFStreamEventCanAcceptBytes, writeStream, &streamContext)){                 exit(1);             }             CFReadStreamScheduleWithRunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);             CFWriteStreamScheduleWithRunLoop(wStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);             CFReadStreamOpen(iStream);             CFWriteStreamOpen(wStream);         } else {              close(nativeSocketHandle);          }     } } // 读取数据 void readStream(CFReadStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {     UInt8 buff[255];     CFReadStreamRead(stream, buff, 255);     printf("received: %s", buff); } void writeStream (CFWriteStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {     outputStream = stream; } main {     char *str = "nihao";         if (outputStream != NULL) {         CFWriteStreamWrite(outputStream, str, strlen(line) + 1);     } else {         NSLog(@"Cannot send data!");     } } // 开辟一个线程线程函数中 void runLoopInThread() {     int res = setupSocket();     if (!res) {         exit(1);     }     CFRunLoopRun();    // 运行当前线程的CFRunLoop对象 }