其实写这个socket一开始我是拒绝的。
因为大家学c 语言和linux基础时肯定都有接触，客户端和服务端的通信也都了解过，加上现在很多开放的第三方库都不需要我们来操作底层的通信。
但是！还是想写。底层的东西最好了解下。
效果
由于5m的上传限制gif可能看不清 我再截两张图吧
服务器
客户端a
客户端b
模型图
分析
由上图可以了解到服务器和客户端需要做哪些工作
服务器
抽象一点分为:
1.创建线程等待接收客户端的连接
2.接收并解析客户端发来的消息
3.给客户端发送消息
具体一点:
1.创建socket. 绑定端口.开始监听.
2.创建线程.等待接收客户端连接.
3.接收客户端发来的消息
4.解析消息内容
a.设置用户名
b.发送消息给指定客户端
客户端
抽象一点分为:
1.连接服务器
2.给服务器发送消息
3.接收服务器消息
4.解析消息内容
具体一点:
1.创建socket.绑定端口.连接服务器
2.发送消息
a.设置用户名
b.给指定用户发消息:按服务器格式拼接字符串
3.接收消息
a.普通消息
b.用户列表:保存至用户列表
ui方面
服务器:其实不用什么ui放个控件展示下日志就是了
客户端:比较简单,一个俗套聊天室的界面,直接storyboard里拖拖控件设置约束了
demo而已别太当真
代码部分
服务器
要使用scoket需要引用这三个头文件
|
1
2
3
|
#include <netinet/in.h>#include <sys/socket.h>#include <arpa/inet.h> |
只有一个model,用来绑定用户名和socket
|
1
2
3
4
|
@interface clientmodel : nsobject@property(nonatomic,assign)int clientsocket;@property(nonatomic,copy)nsstring *clientname;@end |
只有一个文件全给你
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
|
#import "viewcontroller.h"#include <netinet/in.h>#include <sys/socket.h>#include <arpa/inet.h>#import "clientmodel.h"static int const kmaxconnectcount = 5;@interface viewcontroller()@property (weak) iboutlet nstextfield *textfield;//@property (nonatomic,assign)int client_socket; //客户端socket@property (unsafe_unretained) iboutlet nstextview *textview; @property (nonatomic,strong)nsmutablearray *clientarray; @property (nonatomic,strong)nsmutablearray *clientnamearray;@end@implementation viewcontroller- (nsmutablearray *)clientarray { if (!_clientarray) { _clientarray = [nsmutablearray array]; } return _clientarray;}- (nsmutablearray *)clientnamearray { if (!_clientnamearray) { _clientnamearray = [nsmutablearray array]; } return _clientnamearray;}- (void)viewdidload { [super viewdidload]; // do any additional setup after loading the view. //创建socket int server_socket = socket(af_inet, sock_stream, 0); if (server_socket == -1) { nslog(@"创建失败"); [self showlogswithstring:@"socket创建失败"]; }else{ //绑定地址和端口 struct sockaddr_in server_addr; server_addr.sin_len = sizeof(struct sockaddr_in); server_addr.sin_family = af_inet; server_addr.sin_port = htons(1234); server_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); bzero(&(server_addr.sin_zero), 8); int bind_result = bind(server_socket, (struct sockaddr*)&server_addr, sizeof(server_addr)); if (bind_result == -1) { nslog(@"绑定端口失败"); [self showlogswithstring:@"绑定端口失败"]; }else{ if (listen(server_socket, kmaxconnectcount)==-1) { nslog(@"监听失败"); [self showlogswithstring:@"监听失败"]; }else{ for (int i = 0; i < kmaxconnectcount; i++) { //接受客户端的链接 [self acceptclientwithserversocket:server_socket]; } } } }}- (void)setrepresentedobject:(id)representedobject { [super setrepresentedobject:representedobject]; // update the view, if already loaded.}//创建线程接受客户端-(void)acceptclientwithserversocket:(int)server_socket{ struct sockaddr_in client_address; socklen_t address_len; dispatch_queue_t queue = dispatch_get_global_queue(dispatch_queue_priority_default, 0); dispatch_async(queue, ^{ //创建新的socket while (1) { int client_socket = accept(server_socket, (struct sockaddr*)&client_address,&address_len ); if (client_socket == -1) { [self showlogswithstring:@"接受客户端链接失败"]; nslog(@"接受客户端链接失败"); }else{ nsstring *acceptinfo = [nsstring stringwithformat:@"客户端 in,socket:%d",client_socket]; [self showlogswithstring:acceptinfo]; //接受客户端数据 [self recvfromclinetwithsocket:client_socket]; } } });}//接受客户端数据- (void)recvfromclinetwithsocket:(int)client_socket{ while (1) { //接受客户端传来的数据 char buf[1024] = {0}; long ireturn = recv(client_socket, buf, 1024, 0); if (ireturn>0) { nslog(@"客户端来消息了"); nsstring *str = [nsstring stringwithcstring:buf encoding:nsutf8stringencoding]; [self showlogswithstring:[nsstring stringwithformat:@"客户端来消息了:%@",str]]; [self checkrecvstr:str andclientsocket:client_socket]; }else if (ireturn == -1){ nslog(@"读取消息失败"); [self showlogswithstring:@"读取消息失败"]; break; }else if (ireturn == 0){ nslog(@"客户端走了"); [self showlogswithstring:[nsstring stringwithformat:@"客户端 out socket:%d",client_socket]]; nsmutablearray *array = [nsmutablearray arraywitharray:self.clientarray]; for (clientmodel *model in array) { if (model.clientsocket == client_socket) { [self.clientnamearray removeobject:model.clientname]; [self.clientarray removeobject:model]; } } close(client_socket); break; } }} //检查接受到的字符串- (void)checkrecvstr:(nsstring*)str andclientsocket:(int)socket{ if ([str hasprefix:@"name:"]) { nsstring *name = [str substringfromindex:5]; clientmodel *model = [[clientmodel alloc] init]; model.clientsocket = socket; model.clientname = name; if (self.clientarray.count > 0) { int flag = 999; //用户名不能相同 int i = 0; for (clientmodel *client in self.clientarray) { //改名 if (client.clientsocket == socket) { nsstring *oldname = self.clientnamearray[i]; self.clientnamearray[i] = name; self.clientarray[i] = model; for (clientmodel *oldclient in self.clientarray) { [self sendmsg:[nsstring stringwithformat:@"%@ 改名 %@",oldname,name] toclient:oldclient.clientsocket]; [self showlogswithstring:[nsstring stringwithformat:@"%@ 改名 %@",oldname,name]]; nsstring *list = [self.clientnamearray componentsjoinedbystring:@","]; //向客户端推送当前在线列表 [self sendmsg:[nsstring stringwithformat:@"list:%@",list] toclient:oldclient.clientsocket]; } flag = 2; }else{ if ([client.clientname isequaltostring:model.clientname]) { //用户名已存在 flag = 1; break; } } i++; } if (flag != 1 & flag != 2) { [self.clientarray addobject:model]; [self.clientnamearray addobject:model.clientname]; //向客户端推送当前在线列表 for (clientmodel *client in self.clientarray) { [self sendmsg:[nsstring stringwithformat:@"%@,上线了",name] toclient:client.clientsocket]; nsstring *list = [self.clientnamearray componentsjoinedbystring:@","]; //向客户端推送当前在线列表 [self sendmsg:[nsstring stringwithformat:@"list:%@",list] toclient:client.clientsocket]; } //给当前客户端发送一条欢迎信息 nsstring *msg = [nsstring stringwithformat:@"welcome %@ !",name]; [self sendmsg:msg toclient:socket]; [self showlogswithstring:msg]; }else if (flag == 1){ [self sendmsg:@"注册用户名失败，用户名已经存在，请重新设置用户名" toclient:socket]; [self showlogswithstring:[nsstring stringwithformat:@"socket %d 注册用户名失败，设置的用户名已经存在",socket]]; for (clientmodel *model in self.clientarray) { [name isequaltostring:model.clientname]; } } }else{ [self.clientarray addobject:model]; [self.clientnamearray addobject:model.clientname]; //向客户端推送当前在线列表 //给当前客户端发送一条欢迎信息 nsstring *msg = [nsstring stringwithformat:@"welcome %@ !",name]; [self sendmsg:msg toclient:socket]; [self showlogswithstring:msg]; nsstring *list = [self.clientnamearray componentsjoinedbystring:@","]; //向客户端推送当前在线列表 [self sendmsg:[nsstring stringwithformat:@"list:%@",list] toclient:socket]; } } //给某人发消息 else if ([str hasprefix:@"to:"]){ nsrange namerange = [str rangeofstring:@"*"]; nsstring *name = [str substringwithrange:nsmakerange(3, namerange.location-3)]; nsstring *content = [str substringfromindex:namerange.location+1]; nsstring *fromclientname; //找出发送者 for (clientmodel *model in self.clientarray) { if (socket == model.clientsocket) { fromclientname = model.clientname; break; } } //给目标发送信息 for (clientmodel *model in self.clientarray) { if ([name isequaltostring:model.clientname]) { nsstring *msg = [nsstring stringwithformat:@"%@ to you\n%@",fromclientname,content]; [self sendmsg:msg toclient:model.clientsocket]; [self showlogswithstring:[nsstring stringwithformat:@"%@ 发送给 %@ 内容是：%@",fromclientname,name,content]]; break; } } }}//给客户端发送信息- (void)sendmsg:(nsstring*)msg toclient:(int)socket{ char *buf[1024] = {0}; const char *p1 = (char*)buf; p1 = [msg cstringusingencoding:nsutf8stringencoding]; send(socket, p1, 1024, 0);} //在界面上显示日志- (void)showlogswithstring:(nsstring*)str { dispatch_async(dispatch_get_main_queue(), ^{ nsstring *newstr = [nsstring stringwithformat:@"\n%@",str]; self.textview.string = [self.textview.string stringbyappendingstring:newstr]; });}@end |
客户端
由于客户端设计的就比较简单,所以代码量也很少,全给你.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
#import "viewcontroller.h"#include <netinet/in.h>#include <sys/socket.h>#include <arpa/inet.h>@interface viewcontroller ()<uitableviewdelegate,uitableviewdatasource>//服务器socket@property (nonatomic,assign)int server_socket;//ui@property (weak, nonatomic) iboutlet uitextfield *usernamefield;@property (weak, nonatomic) iboutlet uitextview *chatview;@property (weak, nonatomic) iboutlet uitextfield *msgfield;@property (weak, nonatomic) iboutlet uilabel *toname;@property (weak, nonatomic) iboutlet uiview *onlineuserview;@property (nonatomic,strong)uitableview *onlinetable;//user列表@property (nonatomic,strong)nsmutablearray *userarray;@end@implementation viewcontroller- (nsmutablearray *)userarray { if (!_userarray) { _userarray = [nsmutablearray array]; } return _userarray;}- (void)viewdidload { [super viewdidload]; [self.usernamefield becomefirstresponder]; self.usernamefield.text = @""; self.msgfield.text = @""; //添加table用户列表 self.onlinetable = [[uitableview alloc] initwithframe:self.onlineuserview.frame style:uitableviewstyleplain]; self.onlinetable.delegate = self; self.onlinetable.datasource = self; [self.view addsubview:self.onlinetable]; int server_socket = socket(af_inet, sock_stream, 0); if (server_socket == -1) { nslog(@"创建失败"); }else{ //绑定地址和端口 struct sockaddr_in server_addr; server_addr.sin_len = sizeof(struct sockaddr_in); server_addr.sin_family = af_inet; server_addr.sin_port = htons(1234); server_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); bzero(&(server_addr.sin_zero), 8); //接受客户端的链接 dispatch_queue_t queue = dispatch_get_global_queue(dispatch_queue_priority_default, 0); dispatch_async(queue, ^{ //创建新的socket int aresult = connect(server_socket, (struct sockaddr*)&server_addr, sizeof(struct sockaddr_in)); if (aresult == -1) { nslog(@"链接失败"); }else{ self.server_socket = server_socket; [self acceptfromserver]; } }); }}//从服务端接受消息- (void)acceptfromserver{ while (1) { //接受服务器传来的数据 char buf[1024]; long ireturn = recv(self.server_socket, buf, 1024, 0); if (ireturn>0) { nsstring *str = [nsstring stringwithcstring:buf encoding:nsutf8stringencoding]; //筛选前缀 if ([str hasprefix:@"list:"]) { nsstring *arraystr = [str substringfromindex:5]; nsarray *list = [arraystr componentsseparatedbystring:@","]; self.userarray = [nsmutablearray arraywitharray:list]; dispatch_async(dispatch_get_main_queue(), ^{ [self.onlinetable reloaddata]; }); nslog(@"当前在线用户列表：%@",arraystr); }else{ //回到主线程 界面上显示内容 [self showlogswithstring:str]; } }else if (ireturn == -1){ nslog(@"接受失败-1"); break; } }}//在界面上显示日志- (void)showlogswithstring:(nsstring*)str { dispatch_async(dispatch_get_main_queue(), ^{ nsstring *newstr = [nsstring stringwithformat:@"\n%@",str]; self.chatview.text = [self.chatview.text stringbyappendingstring:newstr]; });}- (void)didreceivememorywarning { [super didreceivememorywarning]; // dispose of any resources that can be recreated.}//设置用户名- (ibaction)clicksetusername:(id)sender { nsstring *msg = [nsstring stringwithformat:@"name:%@",self.usernamefield.text] ; [self sendmsg:msg];// [self showlogswithstring:msg]; [self.msgfield becomefirstresponder];}//发送信息- (ibaction)clicksendmsg:(id)sender { if ([self.msgfield.text isequaltostring:@""] || ![self.userarray containsobject:self.usernamefield.text] || [self.toname.text isequaltostring:self.usernamefield.text]) { [self showlogswithstring:@"请设置用户名、检查发送对象、消息不能为空"]; return; } nsstring *msg = [nsstring stringwithformat:@"to:%@*%@",self.toname.text,self.msgfield.text]; [self sendmsg:msg]; nsstring *displaymsg = [nsstring stringwithformat:@"to:%@\n%@",self.toname.text,self.msgfield.text]; [self showlogswithstring:displaymsg]; self.msgfield.text = @"";}//给客户端发送信息- (void)sendmsg:(nsstring*)msg { char *buf[1024] = {0}; const char *p1 = (char*)buf; p1 = [msg cstringusingencoding:nsutf8stringencoding]; send(self.server_socket, p1, 1024, 0);}#pragma mark - tableviewdelegate & datasource- (nsinteger)tableview:(uitableview *)tableview numberofrowsinsection:(nsinteger)section{ return self.userarray.count;}- (uitableviewcell *)tableview:(uitableview *)tableview cellforrowatindexpath:(nsindexpath *)indexpath{ static nsstring *cellid = @"onlinetableviewcellid"; uitableviewcell *cell = [tableview dequeuereusablecellwithidentifier:cellid]; if (cell == nil) { cell = [[uitableviewcell alloc] initwithstyle:uitableviewcellstyledefault reuseidentifier:cellid]; }else{ nslog(@"cell重用了"); } cell.textlabel.text = self.userarray[indexpath.row]; return cell;}//点击cell- (void)tableview:(uitableview *)tableview didselectrowatindexpath:(nsindexpath *)indexpath{ self.toname.text = self.userarray[indexpath.row]; [self.msgfield becomefirstresponder];}@end |
demo地址：imsocketdemo.rar
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：http://www.jianshu.com/p/9d4a353e6682#
