-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathwebserver.cpp
More file actions
435 lines (397 loc) · 11.6 KB
/
Copy pathwebserver.cpp
File metadata and controls
435 lines (397 loc) · 11.6 KB
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
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
#include "webserver.h"
WebServer::WebServer()
{
users = new http_conn[MAX_FD];
// root文件夹路径
char server_path[200];
getcwd(server_path, 200);
char root[6] = "/root"; // 资源文件的的路径
m_root = (char *)malloc(strlen(server_path) + strlen(root) + 1);
strcpy(m_root, server_path);
strcat(m_root, root); // m_root = /home/jyhlinux/share/JWebServer/root
users_timer = new client_data[MAX_FD]; // 连接资源数组（包含对应定时器）
}
WebServer::~WebServer()
{
close(m_epollfd);
close(m_listenfd);
close(m_pipefd[0]);
close(m_pipefd[1]);
delete[] users;
delete[] users_timer;
delete m_pool;
}
void WebServer::init(int port, string user, string passWord, string databaseName,
int log_write, int opt_linger, int trigmode, int sql_num,
int thread_num, int close_log, int actor_model)
{
m_port = port;
m_user = user;
m_passWord = passWord;
m_databaseName = databaseName;
m_sql_num = sql_num;
m_thread_num = thread_num;
m_log_write = log_write;
m_OPT_LINGER = opt_linger;
m_TRIGMode = trigmode;
m_close_log = close_log;
m_actormodel = actor_model;
}
void WebServer::trig_mode()
{
// LT + LT
if (0 == m_TRIGMode)
{
m_LISTENTrigmode = 0;
m_CONNTrigmode = 0;
}
// LT + ET
else if (1 == m_TRIGMode)
{
m_LISTENTrigmode = 0;
m_CONNTrigmode = 1;
}
// ET + LT
else if (2 == m_TRIGMode)
{
m_LISTENTrigmode = 1;
m_CONNTrigmode = 0;
}
// ET + ET
else if (3 == m_TRIGMode)
{
m_LISTENTrigmode = 1;
m_CONNTrigmode = 1;
}
}
void WebServer::log_write()
{
if (0 == m_close_log)
{
// async
if (1 == m_log_write)
{
// 缓冲区大小2000bytes， 每页最多800000行，阻塞队列大小800，最多800待写入
Log::get_instance()->init("./ServerLog", m_close_log, 2000, 800000, 800);
}
else
{
Log::get_instance()->init("./ServerLog", m_close_log, 2000, 800000, 0);
}
}
}
void WebServer::sql_pool()
{
// init connection pool
m_connPool = connection_pool::GetInstance();
m_connPool->init("localhost", m_user, m_passWord, m_databaseName, 3306, m_sql_num, m_close_log);
// 初始化数据库读取表
users->initmysql_result(m_connPool);
}
void WebServer::thread_pool()
{
m_pool = new threadpool<http_conn>(m_actormodel, m_connPool, m_thread_num);
}
void WebServer::eventListen()
{
m_listenfd = socket(PF_INET, SOCK_STREAM, 0);
assert(m_listenfd >= 0);
// 优雅关闭连接
if (0 == m_OPT_LINGER) // 不设置等待时间
{
struct linger tmp = {0, 1};
setsockopt(m_listenfd, SOL_SOCKET, SO_LINGER, &tmp, sizeof(tmp));
}
if (1 == m_OPT_LINGER) // 设置等待时间
{
struct linger tmp = {1, 1};
setsockopt(m_listenfd, SOL_SOCKET, SO_LINGER, &tmp, sizeof(tmp));
}
int ret = 0;
struct sockaddr_in address;
bzero(&address, sizeof(address));
address.sin_family = AF_INET;
address.sin_addr.s_addr = htonl(INADDR_ANY);
address.sin_port = htons(m_port);
int flag = 1;
setsockopt(m_listenfd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
ret = bind(m_listenfd, (struct sockaddr *)&address, sizeof(address));
assert(ret >= 0);
ret = listen(m_listenfd, 5);
assert(ret >= 0);
LOG_INFO("Listen on port:%d", m_port);
utils.init(TIMESLOT);
epoll_event events[MAX_EVENT_NUMBER];
m_epollfd = epoll_create(5);
assert(m_epollfd != -1);
utils.addfd(m_epollfd, m_listenfd, false, m_LISTENTrigmode);
http_conn::m_epollfd = m_epollfd;
ret = socketpair(PF_UNIX, SOCK_STREAM, 0, m_pipefd);
assert(ret != -1);
utils.setnonblocking(m_pipefd[1]);
// 关注m_pipefd[0]可读事件
utils.addfd(m_epollfd, m_pipefd[0], false, 0); // LT
utils.addsig(SIGPIPE, SIG_IGN);
utils.addsig(SIGALRM, utils.sig_handler, false);
utils.addsig(SIGTERM, utils.sig_handler, false);
// TIMESLOT秒后发出SIGALRM信号
alarm(TIMESLOT);
Utils::u_pipefd = m_pipefd;
Utils::u_epollfd = m_epollfd;
}
void WebServer::timer(int connfd, struct sockaddr_in client_address)
{
// http连接
users[connfd].init(connfd, client_address, m_root, m_CONNTrigmode, m_close_log, m_user, m_passWord, m_databaseName);
// 初始化client_data数据
users_timer[connfd].address = client_address;
users_timer[connfd].sockfd = connfd;
// 创建连接对应的定时器
util_timer *timer = new util_timer;
timer->user_data = &users_timer[connfd];
timer->cb_func = cb_func;
time_t cur = time(NULL);
timer->expire = cur + 3 * TIMESLOT;
users_timer[connfd].timer = timer;
utils.m_timer_lst.add_timer(timer);
}
// 某个connfd有数据传输，定时器到期时间后延3个单位
// 并对新的定时器在链表上位置调整
void WebServer::adjust_timer(util_timer *timer)
{
time_t cur = time(NULL);
timer->expire = cur + 3 * TIMESLOT;
utils.m_timer_lst.adjust_timer(timer);
LOG_INFO("%s", "adjust timer once");
}
// 处理定时器到时的任务，删除非活动连接socket在epoll上的注册事件，并关闭
// 再删除定时器
void WebServer::deal_timer(util_timer *timer, int sockfd)
{
timer->cb_func(&users_timer[sockfd]);
if (timer)
{
utils.m_timer_lst.del_timer(timer);
}
LOG_INFO("close fd %d", users_timer[sockfd].sockfd);
}
bool WebServer::dealclientdata()
{
struct sockaddr_in client_address;
socklen_t client_addresslength = sizeof(client_address);
if (0 == m_LISTENTrigmode) // LT
{
int connfd = accept(m_listenfd, (struct sockaddr *)&client_address, &client_addresslength);
if (connfd < 0)
{
LOG_ERROR("%s:errno is %d", "accept error", errno);
return false;
}
if (http_conn::m_user_count >= MAX_FD)
{
utils.show_error(connfd, "Internal server busy");
LOG_ERROR("%s", "Internal server busy");
return false;
}
timer(connfd, client_address);
}
else // ET
{
while (1) // 有缓冲区有数据到来动作才触发事件，故要循环accept
{
int connfd = accept(m_listenfd, (struct sockaddr *)&client_address, &client_addresslength);
if (connfd < 0)
{
LOG_ERROR("%s:errno is %d", "accept error", errno);
break;
}
if (http_conn::m_user_count >= MAX_FD)
{
utils.show_error(connfd, "Internal server busy");
LOG_ERROR("%s", "Internal server busy");
break;
}
timer(connfd, client_address);
}
return false; // 逻辑是否有问题???
}
return true;
}
bool WebServer::dealwithsignal(bool &timeout, bool &stop_server)
{
int ret = 0;
int sig;
char signals[1024];
ret = recv(m_pipefd[0], signals, sizeof(signals), 0);
if(ret == -1)
{
return false;
}
else if(ret == 0) // connection close
{
return false;
}
else
{
for(int i = 0; i < ret; ++i)
{
switch(signals[i])
{
case SIGALRM:
{
timeout = true;
break;
}
case SIGTERM:
{
stop_server = true;
break;
}
}
}
}
return true;
}
void WebServer::dealwithread(int sockfd)
{
// 对一个sockfd有读事件，要对其定时器延后到期
util_timer *timer = users_timer[sockfd].timer;
// reactor,交给线程池处理
if(1 == m_actormodel)
{
if(timer)
{
adjust_timer(timer);
}
// 若检测到读事件，将该事件放到请求队列,进行http解析处理
m_pool->append(users + sockfd, 0);
while(true)
{
if(1 == users[sockfd].improv)
{
if(1 == users[sockfd].timer_flag)
{
deal_timer(timer, sockfd);
users[sockfd].timer_flag = 0;
}
users[sockfd].improv = 0;
break;
}
}
}
else // proactor,直接在此处理
{
if(users[sockfd].read_once())
{
LOG_INFO("deal with the client(%s)", inet_ntoa(users[sockfd].get_address()->sin_addr));
// 若检测到读事件，将该事件放到任务队列，线程争抢，通过锁防止惊群，信号量通知任务量
m_pool->append_p(users + sockfd);
if(timer)
{
adjust_timer(timer);
}
}
else
{
deal_timer(timer, sockfd);
}
}
}
void WebServer::dealwithwrite(int sockfd)
{
util_timer *timer = users_timer[sockfd].timer;
// reactor,交给线程池处理
if(1 == m_actormodel)
{
if(timer)
{
adjust_timer(timer);
}
// 若检测到读事件，将该事件放到请求队列,进行http解析处理
m_pool->append(users + sockfd, 0);
while(true)
{
if(1 == users[sockfd].improv)
{
if(1 == users[sockfd].timer_flag)
{
deal_timer(timer, sockfd);
users[sockfd].timer_flag = 0;
}
users[sockfd].improv = 0;
break;
}
}
}
else // proactor ??
{
if(users[sockfd].write())
{
LOG_INFO("send data to the client(%s)", inet_ntoa(users[sockfd].get_address()->sin_addr));
if(timer)
{
adjust_timer(timer);
}
}
else
{
deal_timer(timer, sockfd);
}
}
}
void WebServer::eventLoop()
{
bool timeout = false;
bool stop_server = false;
while(!stop_server)
{
// block
int number = epoll_wait(m_epollfd, events, MAX_EVENT_NUMBER, -1);
if(number < 0 && errno != EINTR)
{
LOG_ERROR("%s", "epoll failure");
break;
}
for(int i = 0; i < number; ++i)
{
int sockfd = events[i].data.fd;
if(sockfd == m_listenfd)
{
bool flag = dealclientdata();
if(false == flag)
{
continue;
}
}
else if(events[i].events & (EPOLLRDHUP | EPOLLHUP | EPOLLERR))
{
// 若有异常，关闭客户连接,移除对应定时器
util_timer *timer = users_timer[sockfd].timer;
deal_timer(timer, sockfd);
}
// 处理信号
else if((sockfd == m_pipefd[0]) && (events[i].events & EPOLLIN))
{
bool flag = dealwithsignal(timeout, stop_server);
if(false == flag)
{
LOG_ERROR("%s", "dealwithsignal faiure");
}
}
else if(events[i].events & EPOLLIN)
{
dealwithread(sockfd);
}
else if(events[i].events & EPOLLOUT)
{
dealwithwrite(sockfd);
}
}
if(timeout) // ALARM信号到，检测定时器链表是否有到时定时器
{
utils.timer_handler();
LOG_INFO("%s", "timer tick");
timeout = false;
}
}
}