本帖最后由 cruzer 于 2015-1-22 15:46 编辑
头文件:
#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);
- }
-
- 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对象
- }
复制代码
原帖地址:http://www.cocoachina.com/bbs/read.php?tid=149651
|
|