IOS上的socket通信
客户端:
导入头文件:
#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. 设置回调函数
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];
11 }
12 3. 接收发送数据
13 // 读取接收的数据
14 - (void)readStream {
15 char buffer[1024];
16 NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
17 while (recv(CFSocketGetNative(_socket), //与本机关联的Socket 如果已经失效返回-1:INVALID_SOCKET
18 buffer, sizeof(buffer), 0)) {
19 NSLog(@"%@", [NSString stringWithUTF8String:buffer]);
20 }
21 }
22 // 发送数据
23 - (void)sendMessage {
24 NSString *stringTosend = @"你好";
25 char *data = [stringTosend UTF8String];
26 send(SFSocketGetNative(_socket), data, strlen(data) + 1, 0);
27 }
28 服务器端:
29 CFSockteRef _socket;
30 CFWriteStreamRef outputStream = NULL;
31 int setupSocket() {
32 _socket = CFSocketCreate(kCFAllocatorDefault, PF_INET, SOCK_STREAM, IPPROTO_TCP, kCFSocketAcceptCallBack, TCPServerAcceptCallBack, NULL);
33 if (NULL == _socket) {
34 NSLog(@"Cannot create socket!");
35 return 0;
36 }
37
38 int optval = 1;
39 setsockopt(CFSocketGetNative(_socket), SOL_SOCKET, SO_REUSEADDR, // 允许重用本地地址和端口
40 (void *)&optval, sizeof(optval));
41
42 struct sockaddr_in addr4;
43 memset(&addr4, 0, sizeof(addr4));
44 addr4.sin_len = sizeof(addr4);
45 addr4.sin_family = AF_INET;
46 addr4.sin_port = htons(port);
47 addr4.sin_addr.s_addr = htonl(INADDR_ANY);
48 CFDataRef address = CFDataCreate(kCFAllocatorDefault, (UInt8 *)&addr4, sizeof(addr4));
49
50 if (kCFSocketSuccess != CFSocketSetAddress(_socket, address)) {
51 NSLog(@"Bind to address failed!");
52 if (_socket)
53 CFRelease(_socket);
54 _socket = NULL;
55 return 0;
56 }
57
58 CFRunLoopRef cfRunLoop = CFRunLoopGetCurrent();
59 CFRunLoopSourceRef source = CFSocketCreateRunLoopSource(kCFAllocatorDefault, _socket, 0);
60 CFRunLoopAddSource(cfRunLoop, source, kCFRunLoopCommonModes);
61 CFRelease(source);
62
63 return 1;
64 }
65 // socket回调函数,同客户端
66 void TCPServerAcceptCallBack(CFSocketRef socket, CFSocketCallBackType type, CFDataRef address, const void *data, void *info) {
67 if (kCFSocketAcceptCallBack == type) {
68 // 本地套接字句柄
69 CFSocketNativeHandle nativeSocketHandle = *(CFSocketNativeHandle *)data;
70 uint8_t name[SOCK_MAXADDRLEN];
71 socklen_t nameLen = sizeof(name);
72 if (0 != getpeername(nativeSocketHandle, (struct sockaddr *)name, &nameLen)) {
73 NSLog(@"error");
74 exit(1);
75 }
76 NSLog(@"%@ connected.", inet_ntoa( ((struct sockaddr_in *)name)->sin_addr )):
77 CFReadStreamRef iStream;
78 CFWriteStreamRef oStream;
79 // 创建一个可读写的socket连接
80 CFStreamCreatePairWithSocket(kCFAllocatorDefault, nativeSocketHandle, &iStream, &oStream);
81 if (iStream && oStream) {
82 CFStreamClientContext streamContext = {0, NULL, NULL, NULL};
83 if (!CFReadStreamSetClient(iStream, kCFStreamEventHasBytesAvaiable,
84 readStream, // 回调函数,当有可读的数据时调用
85 &streamContext)){
86 exit(1);
87 }
88 if (!CFReadStreamSetClient(iStream, kCFStreamEventCanAcceptBytes, writeStream, &streamContext)){
89 exit(1);
90 }
91 CFReadStreamScheduleWithRunLoop(iStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);
92 CFWriteStreamScheduleWithRunLoop(wStream, CFRunLoopGetCurrent(), kCFRunLoopCommomModes);
93 CFReadStreamOpen(iStream);
94 CFWriteStreamOpen(wStream);
95 } else {
96 close(nativeSocketHandle);
97 }
98 }
99 }
100 // 读取数据
101 void readStream(CFReadStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {
102 UInt8 buff[255];
103 CFReadStreamRead(stream, buff, 255);
104 printf("received: %s", buff);
105 }
106 void writeStream (CFWriteStreamRef stream, CFStreamEventType eventType, void *clientCallBackInfo) {
107 outputStream = stream;
108 }
109 main {
110 char *str = "nihao";
111
112 if (outputStream != NULL) {
113 CFWriteStreamWrite(outputStream, str, strlen(line) + 1);
114 } else {
115 NSLog(@"Cannot send data!");
116 }
117 }
118 // 开辟一个线程线程函数中
119 void runLoopInThread() {
120 int res = setupSocket();
121 if (!res) {
122 exit(1);
123 }
124 CFRunLoopRun(); // 运行当前线程的CFRunLoop对象
125 }