main.cpp
2.92 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
/*
This file is part of FlashMQ (https://www.flashmq.org)
Copyright (C) 2021 Wiebe Cazemier
FlashMQ is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, version 3.
FlashMQ is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public
License along with FlashMQ. If not, see <https://www.gnu.org/licenses/>.
*/
#include <iostream>
#include <signal.h>
#include <memory>
#include <string.h>
#include <openssl/ssl.h>
#include "mainapp.h"
MainApp *mainApp = nullptr;
static void signal_handler(int signal)
{
if (signal == SIGPIPE)
{
return;
}
if (signal == SIGHUP)
{
mainApp->queueConfigReload();
}
else if (signal == SIGTERM || signal == SIGINT)
{
mainApp->quit();
}
else
{
Logger *logger = Logger::getInstance();
logger->logf(LOG_INFO, "Received unhandled signal %d", signal);
}
}
int register_signal_handers()
{
struct sigaction sa;
memset(&sa, 0, sizeof (struct sigaction));
sa.sa_handler = &signal_handler;
sigemptyset(&sa.sa_mask);
sa.sa_flags = SA_RESTART;
if (sigaction(SIGHUP, &sa, nullptr) != 0 || sigaction(SIGTERM, &sa, nullptr) != 0 || sigaction(SIGINT, &sa, nullptr) != 0)
{
Logger *logger = Logger::getInstance();
logger->logf(LOG_ERR, "Error registering signal handlers");
return -1;
}
sigset_t set;
sigemptyset(&set);
sigaddset(&set,SIGPIPE);
int r;
if ((r = sigprocmask(SIG_BLOCK, &set, NULL) != 0))
{
return r;
}
return 0;
}
int main(int argc, char *argv[])
{
try
{
MainApp::initMainApp(argc, argv);
Logger *logger = Logger::getInstance();
mainApp = MainApp::getMainApp();
check<std::runtime_error>(register_signal_handers());
std::string sse = "without SSE support";
#ifdef __SSE4_2__
sse = "with SSE4.2 support";
#endif
#ifdef NDEBUG
logger->logf(LOG_NOTICE, "Starting FlashMQ version %s, release build %s.", FLASHMQ_VERSION, sse.c_str());
#else
logger->logf(LOG_NOTICE, "Starting FlashMQ version %s, debug build %s.", FLASHMQ_VERSION, sse.c_str());
#endif
mainApp->start();
logger->quit();
}
catch (ConfigFileException &ex)
{
// Not using the logger here, because we may have had all sorts of init errors while setting it up.
std::cerr << ex.what() << std::endl;
return 99;
}
catch (std::exception &ex)
{
// Not using the logger here, because we may have had all sorts of init errors while setting it up.
std::cerr << ex.what() << std::endl;
return 1;
}
return 0;
}