Skip to content
Snippets Groups Projects
Commit 19746d53 authored by Marek Vavrusa's avatar Marek Vavrusa
Browse files

Command line parameters, daemonization, fixed log.

refs #277,#273
parent 37a977c0
No related branches found
No related tags found
No related merge requests found
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "common.h"
#include "server.h"
......@@ -14,50 +15,122 @@ static cute_server *s_server = NULL;
void interrupt_handle(int s)
{
// Omit other signals
if (s != SIGINT || s_server == NULL) {
if (s_server == NULL) {
return;
}
// Reload configuration
if (s == SIGHUP) {
log_info("TODO: reload configuration...\n");
/// \todo Reload configuration?
}
// Stop server
if (s_stopping == 0) {
s_stopping = 1;
cute_stop(s_server);
} else {
log_error("\nOK! OK! Exiting immediately.\n");
exit(1);
if (s == SIGINT || s == SIGTERM) {
if (s_stopping == 0) {
s_stopping = 1;
cute_stop(s_server);
} else {
log_error("\nOK! OK! Exiting immediately.\n");
exit(1);
}
}
}
void help(int argc, char **argv)
{
printf("Usage: %s [parameters] <filename1> [<filename2> ...]\n",
argv[0]);
printf("Parameters:\n"
" -d\tRun server as a daemon.\n"
" -v\tVerbose mode - additional runtime information.\n"
" -V\tPrint version of the server.\n"
" -h\tPrint help and usage.\n");
}
int main(int argc, char **argv)
{
// Open log
log_open(LOG_UPTO(LOG_ERR), LOG_MASK(LOG_ERR) | LOG_MASK(LOG_WARNING));
// Parse command line arguments
int c = 0;
int verbose = 0;
int daemonize = 0;
while ((c = getopt (argc, argv, "dvVh")) != -1) {
switch (c)
{
case 'd':
daemonize = 1;
break;
case 'v':
verbose = 1;
break;
case 'V':
printf("%s, version %d.%d.%d\n", PROJECT_NAME,
PROJECT_VER & 0xFF0000,
PROJECT_VER & 0x00FF00,
PROJECT_VER & 0x0000FF);
return 1;
case 'h':
case '?':
default:
help(argc, argv);
return 1;
}
}
if (argc < 2) {
print_msg(LOG_ERR, "Usage: %s <filename1> [<filename2> ...].\n",
argv[0]);
return -1;
// Check if there's at least one remaining non-option
if (argc - optind < 1) {
help(argc, argv);
return 1;
}
int res = 0;
// Now check if we want to daemonize
if (daemonize) {
if (daemon(1, 0) != 0) {
log_open(0, LOG_MASK(LOG_ERR));
log_error("Daemonization failed, shutting down...\n");
log_close();
return 1;
}
}
// Open log
int log_mask = LOG_MASK(LOG_ERR) | LOG_MASK(LOG_WARNING);
int print_mask = LOG_MASK(LOG_ERR);
if (verbose) {
print_mask |= LOG_MASK(LOG_NOTICE);
print_mask |= LOG_MASK(LOG_WARNING);
print_mask |= LOG_MASK(LOG_INFO);
log_mask = print_mask;
}
log_open(print_mask, log_mask);
// Create server instance
s_server = cute_create();
// Run server
if ((res = cute_start(s_server, argv + 1, argc - 1)) == 0) {
int res = 0;
if ((res = cute_start(s_server, argv + optind, argc - optind)) == 0) {
// Register service and signal handler
struct sigaction sa;
sa.sa_handler = interrupt_handle;
sigemptyset(&sa.sa_mask);
sa.sa_flags = 0;
sigaction(SIGINT, &sa, NULL);
sigaction(SIGINT, &sa, NULL);
sigaction(SIGTERM, &sa, NULL);
sigaction(SIGHUP, &sa, NULL);
sigaction(SIGALRM, &sa, NULL); // Interrupt
// Change directory if daemonized
log_info("Server started.\n");
if (daemonize) {
chdir("/");
}
if ((res = cute_wait(s_server)) != 0) {
log_error("There was an error while waiting for server"
" to finish.");
" to finish.\n");
}
} else {
log_error("There was an error while starting the server, "
......@@ -66,6 +139,7 @@ int main(int argc, char **argv)
// Stop server and close log
cute_destroy(&s_server);
log_info("Shutting down...\n");
log_close();
return res;
......
......@@ -13,7 +13,7 @@ int log_open(int print_mask, int log_mask)
setlogmask(log_mask);
/// \todo May change to LOG_DAEMON.
openlog(PROJECT_NAME, LOG_CONS | LOG_PID | LOG_NDELAY, LOG_LOCAL1);
openlog(PROJECT_NAME, LOG_CONS | LOG_PID, LOG_LOCAL1);
_LOG_MASK = print_mask;
_LOG_OPEN = 1;
return 0;
......@@ -43,7 +43,7 @@ int print_msg(int level, const char *msg, ...)
// Check mask
int ret = 0;
if (level & _LOG_MASK) {
if (LOG_MASK(level) & _LOG_MASK) {
va_start(ap, msg);
ret = vfprintf(stream, msg, ap);
va_end(ap);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment