IOS上的socket通信 CFsocket

客户端：

导入头文件：
#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 = CFSocketCreateRunLoopSou rce(kCFAllocatorDefault, _socket, 0);
       CFRunLoopAddSource(cRunRef, // 运行循环
       sourceRef,   // 增加的运行循环源, 它会被retain一次
      kCFRunLoopCommonModes   // 增加的运行循环源的模式
       );
       CFRelease(courceRef);
}

2. 设置回调函数
// 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 performSlectorInBackgrou nd:@selector(readStream) withObject:nil];
｝

3. 接收发送数据
// 读取接收的数据
- (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);
}


服务器端：

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 = CFSocketCreateRunLoopSou rce(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连接
            CFStreamCreatePairWithSo cket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream); 
            if (iStream && oStream) {
                    CFStreamClientContext streamContext = {0, NULL, NULL, NULL};
                    if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAv aiable,
                                                                  readStream, // 回调函数，当有可读的数据时调用
                                                                  &streamContext)){
                            exit(1);
                    }

                    if (!CFReadStreamSetClient(iStream, kCFStreamEventCanAcceptB ytes, writeStream, &streamContext)){
                            exit(1);
                    }

                    CFReadStreamScheduleWith RunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);
                    CFWriteStreamScheduleWit hRunLoop(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对象
}