socket的简单实用

标签：des   style   color   os   ar   strong   for   文件   数据   

客户端： 

导入头文件： 
#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. 设置回调函数 
// 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]; 
｝ 
 
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 = 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);