黑马程序员技术交流社区

标题: IOS socket 相关【资源共享】 [打印本页]

作者: cruzer    时间: 2015-1-22 15:44
标题: IOS socket 相关【资源共享】
本帖最后由 cruzer 于 2015-1-22 15:46 编辑

头文件:
#import <sys/socket.h>
#import <netinet/in.h>
#import <arpa/inet.h>
#import <unistd.h>1. 创建连接
  1. CFSocketContext sockContext = {0, // 结构体的版本,必须为0
  2. self,  // 一个任意指针的数据,可以用在创建时CFSocket对象相关联。这个指针被传递给所有的上下文中定义的回调。
  3. NULL, // 一个定义在上面指针中的retain的回调, 可以为NULL
  4. NULL, NULL};

  5. CFSocketRef _socket = (kCFAllocatorDefault, // 为新对象分配内存,可以为nil
  6. PF_INET, // 协议族,如果为0或者负数,则默认为PF_INET
  7. SOCK_STREAM, // 套接字类型,如果协议族为PF_INET,则它会默认为SOCK_STREAM
  8. IPPROTO_TCP, // 套接字协议,如果协议族是PF_INET且协议是0或者负数,它会默认为IPPROTO_TCP
  9. kCFSocketConnectCallBack, // 触发回调函数的socket消息类型,具体见Callback Types
  10. TCPServerConnectCallBack, // 上面情况下触发的回调函数
  11. &sockContext // 一个持有CFSocket结构信息的对象,可以为nil
  12. );

  13. if (_socket != nil) {
  14.     struct sockaddr_in addr4;   // IPV4
  15.     memset(&addr4, 0, sizeof(addr4));
  16.     addr4.sin_len = sizeof(addr4);
  17.     addr4.sin_family = AF_INET;
  18.     addr4.sin_port = htons(8888);
  19.     addr4.sin_addr.s_addr = inet_addr([strAddress UTF8String]);  // 把字符串的地址转换为机器可识别的网络地址
  20.         
  21.     // 把sockaddr_in结构体中的地址转换为Data
  22.     CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4));
  23.     CFSocketConnectToAddress(_socket, // 连接的socket
  24. address, // CFDataRef类型的包含上面socket的远程地址的对象
  25. -1  // 连接超时时间,如果为负,则不尝试连接,而是把连接放在后台进行,如果_socket消息类型为kCFSocketConnectCallBack,将会在连接成功或失败的时候在后台触发回调函数
  26. );

  27.     CFRunLoopRef cRunRef = CFRunLoopGetCurrent();    // 获取当前线程的循环
  28.     // 创建一个循环,但并没有真正加如到循环中,需要调用CFRunLoopAddSource
  29.     CFRunLoopSourceRef sourceRef = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0);
  30.     CFRunLoopAddSource(cRunRef, // 运行循环
  31.     sourceRef,  // 增加的运行循环源, 它会被retain一次
  32.     kCFRunLoopCommonModes  // 增加的运行循环源的模式
  33.     );
  34.     CFRelease(courceRef);
  35. }
复制代码
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];
复制代码
3. 接收发送数据
  1. // 读取接收的数据
  2. - (void)readStream {
  3.     char buffer[1024];
  4.     NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
  5.     while (recv(CFSocketGetNative(_socket), //与本机关联的Socket 如果已经失效返回-1:INVALID_SOCKET
  6.            buffer, sizeof(buffer), 0)) {
  7.         NSLog(@"%@", [NSString stringWithUTF8String:buffer]);
  8.     }
  9. }

  10. // 发送数据
  11. - (void)sendMessage {
  12.     NSString *stringTosend = @"你好";
  13.     char *data = [stringTosend UTF8String];
  14.     send(SFSocketGetNative(_socket), data, strlen(data) + 1, 0);
  15. }
复制代码
服务器端:
  1. CFSockteRef _socket;
  2. CFWriteStreamRef outputStream = NULL;

  3. int setupSocket() {
  4.     _socket = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketAcceptCallBack, TCPServerAcceptCallBack, NULL);
  5.     if (NULL == _socket) {
  6.         NSLog(@"Cannot create socket!");
  7.         return 0;
  8.     }
  9.    
  10.     int optval = 1;
  11.     setsockopt(CFSocketGetNative(_socket), SOL_SOCKET, SO_REUSEADDR, // 允许重用本地地址和端口
  12. (void *)&optval, sizeof(optval));
  13.    
  14.     struct sockaddr_in addr4;
  15.     memset(&addr4, 0, sizeof(addr4));
  16.     addr4.sin_len = sizeof(addr4);
  17.     addr4.sin_family = AF_INET;
  18.     addr4.sin_port = htons(port);
  19.     addr4.sin_addr.s_addr = htonl(INADDR_ANY);
  20.     CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4));
  21.    
  22.     if (kCFSocketSuccess != CFSocketSetAddress(_socket, address)) {
  23.         NSLog(@"Bind to address failed!");
  24.         if (_socket)
  25.              CFRelease(_socket);
  26.         _socket = NULL;
  27.         return 0;
  28.     }
  29.         
  30.     CFRunLoopRef cfRunLoop = CFRunLoopGetCurrent();
  31.     CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0);
  32.     CFRunLoopAddSource(cfRunLoop, source, kCFRunLoopCommonModes);
  33.     CFRelease(source);
  34.    
  35.     return 1;
  36. }

  37. // socket回调函数,同客户端
  38. void TCPServerAcceptCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) {
  39.     if (kCFSocketAcceptCallBack == type) {
  40.         // 本地套接字句柄
  41.         CFSocketNativeHandle nativeSocketHandle = *(CFSocketNativeHandle *)data;
  42.         uint8_t name[SOCK_MAXADDRLEN];      
  43.         socklen_t nameLen = sizeof(name);
  44.         if (0 != getpeername(nativeSocketHandle, (struct sockaddr *)name, &nameLen)) {
  45.             NSLog(@"error");
  46.             exit(1);
  47.         }
  48.         NSLog(@"%@ connected.", inet_ntoa( ((struct sockaddr_in *)name)->sin_addr )):

  49.         CFReadStreamRef iStream;
  50.         CFWriteStreamRef oStream;
  51.         // 创建一个可读写的socket连接
  52.         CFStreamCreatePairWithSocket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream);  
  53.         if (iStream && oStream) {
  54.             CFStreamClientContext streamContext = {0, NULL, NULL, NULL};
  55.             if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAvaiable,
  56.                                        readStream, // 回调函数,当有可读的数据时调用
  57.                                        &streamContext)){
  58.                 exit(1);
  59.             }

  60.             if (!CFReadStreamSetClient(iStream, kCFStreamEventCanAcceptBytes, writeStream, &streamContext)){
  61.                 exit(1);
  62.             }

  63.             CFReadStreamScheduleWithRunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);
  64.             CFWriteStreamScheduleWithRunLoop(wStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);
  65.             CFReadStreamOpen(iStream);
  66.             CFWriteStreamOpen(wStream);
  67.         } else {
  68.              close(nativeSocketHandle);  
  69.         }
  70.     }
  71. }

  72. // 读取数据
  73. void readStream(CFReadStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {
  74.     UInt8 buff[255];
  75.     CFReadStreamRead(stream, buff, 255);
  76.     printf("received: %s", buff);
  77. }

  78. void writeStream (CFWriteStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {
  79.     outputStream = stream;
  80. }

  81. main {
  82.     char *str = "nihao";
  83.    
  84.     if (outputStream != NULL) {
  85.         CFWriteStreamWrite(outputStream, str, strlen(line) + 1);
  86.     } else {
  87.         NSLog(@"Cannot send data!");
  88.     }
  89. }

  90. // 开辟一个线程线程函数中
  91. void runLoopInThread() {
  92.     int res = setupSocket();
  93.     if (!res) {
  94.         exit(1);
  95.     }
  96.     CFRunLoopRun();    // 运行当前线程的CFRunLoop对象
  97. }
复制代码



原帖地址:http://www.cocoachina.com/bbs/read.php?tid=149651








欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/) 黑马程序员IT技术论坛 X3.2