main.cpp
2.38 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
/*
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());
logger->logf(LOG_NOTICE, "Starting FlashMQ");
mainApp->start();
}
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;
}