-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
114 lines (108 loc) · 2.97 KB
/
Copy pathmain.cpp
File metadata and controls
114 lines (108 loc) · 2.97 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
#include "unistd.h"
#include "signal.h"
#include "Server.h"
#include "Log.h"
#include <iostream>
#include <fstream>
#include <cassert>
using namespace std;
Server server;
//用于接收HUP信号，终止服务器运行的信号处理函数
void closeServer(int sig){
server.free();
}
int main(int argc, char* argv[]){
if (argc != 2){
cerr << "please input the arg start/stop/status to continue" << endl;
return -1;
}
//验证是否有超级用户权限
if (getuid() != 0){
cerr << "please use sudo to run the serve" << endl;
return -1;
}
//服务器运行
if (string(argv[1]) == "start"){
//检查服务器是否正在运行
ifstream ipidf("/var/run/httpserver.pid");
if (ipidf.is_open()){
int pid;
ipidf >> pid;
ipidf.close();
kill(pid, SIGHUP);
if (remove("/var/run/httpserver.pid") < 0){
cerr << "httpserver is already running" << endl;
return -1;
}
else
cout << "restart the server" << endl;
}
//初始化
try
{
server.loadConfig("./config.cfg");
daemon(1, 1);
server.init();
ofstream pidf("/var/run/httpserver.pid");
pidf << getpid() << endl;
pidf.close();
}
catch(const std::exception& e)
{
std::cerr << e.what() << '\n';
return -1;
}
//相关系统设置
system("ulimit -Sn 16394");
struct sigaction sa;
memset(&sa, '\0', sizeof(sa));
sa.sa_handler = closeServer;
sigfillset(&sa.sa_mask);
if (sigaction(SIGHUP, &sa, NULL) == -1){
Log::log("signal fault", ERROR);
return -1;
}
Log::log("start the server", DEBUG);
try
{
server.start();
}
catch(const std::exception& e)
{
Log::log(e.what(), ERROR);
return -1;
}
return 0;
}
//服务器终止
if (string(argv[1]) == "stop"){
ifstream pidf("/var/run/httpserver.pid");
if (!pidf.is_open()){
cerr << "httpserver is not running" << endl;
return -1;
}
int pid;
pidf >> pid;
pidf.close();
kill(pid, SIGHUP);
if (remove("/var/run/httpserver.pid") < 0){
cerr << "remove file fault" << endl;
return -1;
}
cout << "server in pid " << pid << " has finished" << endl;
return 0;
}
//查询服务器状态
if (string(argv[1]) == "status"){
ifstream pidf("/var/run/httpserver.pid");
if (!pidf.is_open()){
cout << "httpserver is not running" << endl;
return -1;
}
int pid;
pidf >> pid;
pidf.close();
cout << "httpserver is running with pid " << pid << endl;
return 0;
}
}