Browse Source

* add handlers of SIGTERM & SIGHUP

master
Alex 'AdUser' Z 9 years ago
parent
commit
d51695494a
  1. 2
      src/common.h
  2. 28
      src/main.c

2
src/common.h

@ -17,6 +17,8 @@
#include <time.h> #include <time.h>
#include <unistd.h> #include <unistd.h>
#define UNUSED(x) (void)(x)
#define FREE(x) free(x), x = NULL #define FREE(x) free(x), x = NULL
#endif /* F2B_COMMON_H_ */ #endif /* F2B_COMMON_H_ */

28
src/main.c

@ -6,6 +6,29 @@
#include "backend.h" #include "backend.h"
#include <getopt.h> #include <getopt.h>
#include <signal.h>
bool run = true;
bool rcfg = false;
void sa_term(int signum) {
UNUSED(signum);
f2b_log_msg(log_info, "got SIGTERM, exiting");
run = false;
}
void sa_hup(int signum) {
UNUSED(signum);
f2b_log_msg(log_info, "got SIGHUP, reloading config");
rcfg = true;
}
#define SA_REGISTER(signum, handler) \
memset(&act, 0x0, sizeof(act)); \
act.sa_handler = handler; \
if (sigaction(SIGTERM, &act, NULL) != 0) { \
f2b_log_msg(log_error, "can't register handler for " #signum); \
return EXIT_FAILURE; \
}
void usage(int exitcode) { void usage(int exitcode) {
fprintf(stderr, "Usage: f2b -c <config>\n"); fprintf(stderr, "Usage: f2b -c <config>\n");
@ -13,13 +36,13 @@ void usage(int exitcode) {
} }
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
struct sigaction act;
f2b_config_t *config = NULL; f2b_config_t *config = NULL;
f2b_config_section_t *section = NULL; f2b_config_section_t *section = NULL;
f2b_jail_t *jails = NULL; f2b_jail_t *jails = NULL;
f2b_jail_t *jail = NULL; f2b_jail_t *jail = NULL;
char *config_file = NULL; char *config_file = NULL;
char opt = '\0'; char opt = '\0';
bool run = true;
while ((opt = getopt(argc, argv, "c:h")) != -1) { while ((opt = getopt(argc, argv, "c:h")) != -1) {
switch (opt) { switch (opt) {
@ -35,6 +58,9 @@ int main(int argc, char *argv[]) {
} }
} }
SA_REGISTER(SIGTERM, &sa_term);
SA_REGISTER(SIGHUP, &sa_hup);
if (!config_file) if (!config_file)
usage(EXIT_FAILURE); usage(EXIT_FAILURE);
if ((config = f2b_config_load(config_file)) == NULL) { if ((config = f2b_config_load(config_file)) == NULL) {

Loading…
Cancel
Save