int
main(int argc, char **argv)
{
int pid, ret;
struct sigaction new_action, old_action;
GOptionContext *ctx;
GError *parse_err = NULL;
daemon_pid_file_ident = daemon_log_ident = daemon_ident_from_argv0(argv[0]);
daemon_pid_file_proc = conf_pid_file_proc;
if (conf_init() < 0)
return EXIT_FAILURE;
ctx = g_option_context_new("");
g_option_context_add_main_entries(ctx, options, PACKAGE);
g_option_context_set_summary(ctx, PACKAGE"-"VERSION GITHEAD" - mpd cron daemon");
if (!g_option_context_parse(ctx, &argc, &argv, &parse_err)) {
g_printerr("option parsing failed: %s\n", parse_err->message);
g_option_context_free(ctx);
g_error_free(parse_err);
return EXIT_FAILURE;
}
g_option_context_free(ctx);
if (optv) {
about();
cleanup();
return EXIT_SUCCESS;
}
#ifdef DAEMON_SET_VERBOSITY_AVAILABLE
if (conf.no_daemon)
daemon_set_verbosity(LOG_DEBUG);
#endif /* DAEMON_SET_VERBOSITY_AVAILABLE */
/* Version to environment variable */
g_setenv("MPDCRON_PACKAGE", PACKAGE, 1);
g_setenv("MPDCRON_VERSION", VERSION, 1);
g_setenv("MPDCRON_GITHEAD", GITHEAD, 1);
/* Command line options to environment variables */
if (conf.no_daemon)
g_unsetenv("MCOPT_DAEMONIZE");
else
g_setenv("MCOPT_DAEMONIZE", "1", 1);
/* Important! Parse configuration file before killing the daemon
* because the configuration file has a pidfile and killwait option.
*/
cfd = g_key_file_new();
if (keyfile_load(&cfd) < 0) {
cleanup();
return EXIT_FAILURE;
}
if (optk) {
if (daemon_pid_file_kill_wait(SIGINT, conf.killwait) < 0) {
g_warning("Failed to kill daemon: %s", strerror(errno));
cleanup();
return EXIT_FAILURE;
}
daemon_pid_file_remove();
cleanup();
return EXIT_SUCCESS;
}
/* Logging */
g_log_set_default_handler(log_handler, GINT_TO_POINTER(conf.no_daemon ? 5 : conf.loglevel));
/* Signal handling */
new_action.sa_handler = sig_cleanup;
sigemptyset(&new_action.sa_mask);
new_action.sa_flags = 0;
#define HANDLE_SIGNAL(sig) \
do { \
sigaction((sig), NULL, &old_action); \
if (old_action.sa_handler != SIG_IGN) \
sigaction((sig), &new_action, NULL); \
} while (0)
HANDLE_SIGNAL(SIGABRT);
HANDLE_SIGNAL(SIGSEGV);
HANDLE_SIGNAL(SIGINT);
HANDLE_SIGNAL(SIGTERM);
#undef HANDLE_SIGNAL
if (conf.no_daemon) {
/* Create the main loop */
loop = g_main_loop_new(NULL, FALSE);
#ifdef HAVE_GMODULE
/* Load modules which may add initial events */
keyfile_load_modules(&cfd);
#endif /* HAVE_GMODULE */
g_key_file_free(cfd);
cfd = NULL;
/* Add default initial events */
loop_connect();
/* Run the main loop */
g_main_loop_run(loop);
cleanup();
return EXIT_SUCCESS;
}
/* Daemonize */
if ((pid = daemon_pid_file_is_running()) > 0) {
g_critical("Daemon already running on PID %u", pid);
return EXIT_FAILURE;
}
daemon_retval_init();
pid = daemon_fork();
if (pid < 0) {
g_critical("Failed to fork: %s", strerror(errno));
daemon_retval_done();
return EXIT_FAILURE;
}
else if (pid != 0) { /* Parent */
cleanup();
if ((ret = daemon_retval_wait(2)) < 0) {
g_critical("Could not receive return value from daemon process: %s",
strerror(errno));
return 255;
}
if (ret != 0)
g_critical("Daemon returned %i as return value", ret);
else
g_critical("Daemon returned %i as return value", ret);
return ret;
}
else { /* Daemon */
if (daemon_close_all(-1) < 0) {
g_critical("Failed to close all file descriptors: %s",
strerror(errno));
daemon_retval_send(1);
return EXIT_FAILURE;
}
if (daemon_pid_file_create() < 0) {
g_critical("Failed to create PID file: %s",
strerror(errno));
daemon_retval_send(2);
return EXIT_FAILURE;
}
/* Send OK to parent process */
daemon_retval_send(0);
/* Create the main loop */
loop = g_main_loop_new(NULL, FALSE);
#ifdef HAVE_GMODULE
/* Load modules which may add initial events */
keyfile_load_modules(&cfd);
#endif /* HAVE_GMODULE */
g_key_file_free(cfd);
cfd = NULL;
/* Add default initial events */
loop_connect();
/* Run the main loop */
g_main_loop_run(loop);
cleanup();
return EXIT_SUCCESS;
}
return EXIT_SUCCESS;
}
int main(int argc, char *argv[])
{
int index;
int c;
std::string ptp_dev("tun0");
std::string app_name("tunnel");
std::string endpoint("dtn:none");
unsigned int lifetime = 60;
bool daemonize = false;
bool stop_daemon = false;
std::string pidfile;
bool throughput = false;
#ifdef HAVE_LIBDAEMON
while ((c = getopt (argc, argv, "td:s:l:hDkp:")) != -1)
#else
while ((c = getopt (argc, argv, "td:s:l:h")) != -1)
#endif
switch (c)
{
#ifdef HAVE_LIBDAEMON
case 'D':
daemonize = true;
break;
case 'k':
daemonize = true;
stop_daemon = true;
break;
case 'p':
pidfile = optarg;
break;
#endif
case 'd':
ptp_dev = optarg;
break;
case 't':
throughput = true;
break;
case 's':
app_name = optarg;
break;
case 'l':
lifetime = atoi(optarg);
break;
default:
print_help(argv[0]);
return 1;
}
int optindex = 0;
for (index = optind; index < argc; ++index)
{
switch (optindex)
{
case 0:
endpoint = std::string(argv[index]);
break;
}
optindex++;
}
// print help if not enough parameters are set
if (!stop_daemon && (optindex < 1)) { print_help(argv[0]); exit(0); }
// catch process signals
ibrcommon::SignalHandler sighandler(term);
sighandler.handle(SIGINT);
sighandler.handle(SIGTERM);
sighandler.handle(SIGQUIT);
//initialize sighandler after possible exit call
sighandler.initialize();
// logging options
//const unsigned char logopts = ibrcommon::Logger::LOG_DATETIME | ibrcommon::Logger::LOG_LEVEL;
const unsigned char logopts = 0;
// error filter
const unsigned char logerr = ibrcommon::Logger::LOGGER_ERR | ibrcommon::Logger::LOGGER_CRIT;
// logging filter, everything but debug, err and crit
const unsigned char logstd = ibrcommon::Logger::LOGGER_ALL ^ (ibrcommon::Logger::LOGGER_DEBUG | logerr);
// syslog filter, everything but DEBUG and NOTICE
const unsigned char logsys = ibrcommon::Logger::LOGGER_ALL ^ (ibrcommon::Logger::LOGGER_DEBUG | ibrcommon::Logger::LOGGER_NOTICE);
#ifdef HAVE_LIBDAEMON
if (daemonize) {
// enable syslog logging
ibrcommon::Logger::enableSyslog(argv[0], LOG_PID, LOG_DAEMON, logsys);
} else
#endif
{
// add logging to the cout
ibrcommon::Logger::addStream(std::cout, logstd, logopts);
// add logging to the cerr
ibrcommon::Logger::addStream(std::cerr, logerr, logopts);
}
#ifdef HAVE_LIBDAEMON
if (daemonize)
{
#ifdef HAVE_DAEMON_RESET_SIGS
/* Reset signal handlers */
if (daemon_reset_sigs(-1) < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Failed to reset all signal handlers: " << strerror(errno) << IBRCOMMON_LOGGER_ENDL;
return 1;
}
/* Unblock signals */
if (daemon_unblock_sigs(-1) < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Failed to unblock all signals: " << strerror(errno) << IBRCOMMON_LOGGER_ENDL;
return 1;
}
#endif
pid_t pid;
/* Set identification string for the daemon for both syslog and PID file */
daemon_pid_file_ident = daemon_log_ident = daemon_ident_from_argv0(argv[0]);
/* set the pid file path */
if (pidfile.length() > 0) {
__daemon_pidfile__ = new char[pidfile.length() + 1];
::strcpy(__daemon_pidfile__, pidfile.c_str());
daemon_pid_file_proc = __daemon_pid_file_proc__;
}
/* Check if we are called with -k parameter */
if (stop_daemon)
{
int ret;
/* Kill daemon with SIGTERM */
/* Check if the new function daemon_pid_file_kill_wait() is available, if it is, use it. */
if ((ret = daemon_pid_file_kill_wait(SIGTERM, 5)) < 0)
IBRCOMMON_LOGGER_TAG("Core", warning) << "Failed to kill daemon: " << strerror(errno) << IBRCOMMON_LOGGER_ENDL;
return ret < 0 ? 1 : 0;
}
/* Check that the daemon is not rung twice a the same time */
if ((pid = daemon_pid_file_is_running()) >= 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Daemon already running on PID file " << pid << IBRCOMMON_LOGGER_ENDL;
return 1;
}
/* Prepare for return value passing from the initialization procedure of the daemon process */
if (daemon_retval_init() < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Failed to create pipe." << IBRCOMMON_LOGGER_ENDL;
return 1;
}
/* Do the fork */
if ((pid = daemon_fork()) < 0) {
/* Exit on error */
daemon_retval_done();
return 1;
} else if (pid) { /* The parent */
int ret;
/* Wait for 20 seconds for the return value passed from the daemon process */
if ((ret = daemon_retval_wait(20)) < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Could not recieve return value from daemon process: " << strerror(errno) << IBRCOMMON_LOGGER_ENDL;
return 255;
}
return ret;
} else { /* The daemon */
/* Close FDs */
if (daemon_close_all(-1) < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Failed to close all file descriptors: " << strerror(errno) << IBRCOMMON_LOGGER_ENDL;
/* Send the error condition to the parent process */
daemon_retval_send(1);
/* Do a cleanup */
daemon_retval_send(255);
daemon_signal_done();
daemon_pid_file_remove();
return -1;
}
/* Create the PID file */
if (daemon_pid_file_create() < 0) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Could not create PID file ( " << strerror(errno) << ")." << IBRCOMMON_LOGGER_ENDL;
daemon_retval_send(2);
/* Do a cleanup */
daemon_retval_send(255);
daemon_signal_done();
daemon_pid_file_remove();
return -1;
}
/* Send OK to parent process */
daemon_retval_send(0);
}
}
#endif
IBRCOMMON_LOGGER_TAG("Core", info) << "IBR-DTN IP <-> Bundle Tunnel" << IBRCOMMON_LOGGER_ENDL;
// create a connection to the dtn daemon
ibrcommon::vaddress addr("localhost", 4550);
ibrcommon::socketstream conn(new ibrcommon::tcpsocket(addr));
try {
// set-up tun2bundle gateway
TUN2BundleGateway gateway(app_name, conn, ptp_dev);
_gateway = &gateway;
IBRCOMMON_LOGGER_TAG("Core", info) << "Local: " << app_name << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", info) << "Peer: " << endpoint << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", info) << "Device: " << gateway.getDeviceName() << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", notice) << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", notice) << "Now you need to set-up the ip tunnel. You can use commands like this:" << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", notice) << "# sudo ip link set " << gateway.getDeviceName() << " up mtu 65535" << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", notice) << "# sudo ip addr add 10.0.0.1/24 dev " << gateway.getDeviceName() << IBRCOMMON_LOGGER_ENDL;
IBRCOMMON_LOGGER_TAG("Core", notice) << IBRCOMMON_LOGGER_ENDL;
timer_t timerid;
struct sigevent sev;
if (!daemonize && throughput) {
// enable throughput timer
signal(SIGRTMIN, timer_display_throughput);
sev.sigev_notify = SIGEV_SIGNAL;
sev.sigev_signo = SIGRTMIN;
sev.sigev_value.sival_ptr = &timerid;
// create a timer
timer_create(CLOCK_MONOTONIC, &sev, &timerid);
// arm the timer
struct itimerspec its;
size_t freq_nanosecs = 200000000;
its.it_value.tv_sec = freq_nanosecs / 1000000000;;
its.it_value.tv_nsec = freq_nanosecs % 1000000000;
its.it_interval.tv_sec = its.it_value.tv_sec;
its.it_interval.tv_nsec = its.it_value.tv_nsec;
if (timer_settime(timerid, 0, &its, NULL) == -1) {
IBRCOMMON_LOGGER_TAG("Core", error) << "Timer set failed." << IBRCOMMON_LOGGER_ENDL;
}
}
// destination
dtn::data::EID eid(endpoint);
while (m_running)
{
gateway.process(eid, lifetime);
}
gateway.shutdown();
} catch (const ibrcommon::Exception &ex) {
if (m_running) {
IBRCOMMON_LOGGER_TAG("Core", error) << ex.what() << IBRCOMMON_LOGGER_ENDL;
return -1;
}
}
#ifdef HAVE_LIBDAEMON
if (daemonize) {
/* Do a cleanup */
IBRCOMMON_LOGGER_TAG("Core", info) << "Stopped " << app_name << IBRCOMMON_LOGGER_ENDL;
daemon_retval_send(255);
daemon_signal_done();
daemon_pid_file_remove();
} else
#endif
{
std::cout << std::endl;
}
return 0;
}
int main(int argc, char *argv[]) {
int r = 255;
int wrote_pid_file = 0;
avahi_set_log_function(log_function);
init_rand_seed();
avahi_server_config_init(&config.server_config);
config.command = DAEMON_RUN;
config.daemonize = 0;
config.config_file = NULL;
#ifdef HAVE_DBUS
config.enable_dbus = 1;
config.fail_on_missing_dbus = 1;
#endif
config.drop_root = 1;
config.set_rlimits = 1;
#ifdef ENABLE_CHROOT
config.use_chroot = 1;
#endif
config.modify_proc_title = 1;
config.disable_user_service_publishing = 0;
config.publish_dns_servers = NULL;
config.publish_resolv_conf = 0;
config.use_syslog = 0;
config.debug = 0;
config.rlimit_as_set = 0;
config.rlimit_core_set = 0;
config.rlimit_data_set = 0;
config.rlimit_fsize_set = 0;
config.rlimit_nofile_set = 0;
config.rlimit_stack_set = 0;
#ifdef RLIMIT_NPROC
config.rlimit_nproc_set = 0;
#endif
if ((argv0 = strrchr(argv[0], '/')))
argv0 = avahi_strdup(argv0 + 1);
else
argv0 = avahi_strdup(argv[0]);
daemon_pid_file_ident = (const char *) argv0;
daemon_log_ident = (char*) argv0;
daemon_pid_file_proc = pid_file_proc;
if (parse_command_line(&config, argc, argv) < 0)
goto finish;
if (config.modify_proc_title)
avahi_init_proc_title(argc, argv);
#ifdef ENABLE_CHROOT
config.use_chroot = config.use_chroot && config.drop_root;
#endif
if (config.command == DAEMON_HELP) {
help(stdout);
r = 0;
} else if (config.command == DAEMON_VERSION) {
printf("%s "PACKAGE_VERSION"\n", argv0);
r = 0;
} else if (config.command == DAEMON_KILL) {
if (daemon_pid_file_kill_wait(SIGTERM, 5) < 0) {
avahi_log_warn("Failed to kill daemon: %s", strerror(errno));
goto finish;
}
r = 0;
} else if (config.command == DAEMON_RELOAD) {
if (daemon_pid_file_kill(SIGHUP) < 0) {
avahi_log_warn("Failed to kill daemon: %s", strerror(errno));
goto finish;
}
r = 0;
} else if (config.command == DAEMON_CHECK)
r = (daemon_pid_file_is_running() >= 0) ? 0 : 1;
else if (config.command == DAEMON_RUN) {
pid_t pid;
if (getuid() != 0 && config.drop_root) {
avahi_log_error("This program is intended to be run as root.");
goto finish;
}
if ((pid = daemon_pid_file_is_running()) >= 0) {
avahi_log_error("Daemon already running on PID %u", pid);
goto finish;
}
if (load_config_file(&config) < 0)
goto finish;
if (config.daemonize) {
daemon_retval_init();
if ((pid = daemon_fork()) < 0)
goto finish;
else if (pid != 0) {
int ret;
/** Parent **/
if ((ret = daemon_retval_wait(20)) < 0) {
avahi_log_error("Could not receive return value from daemon process.");
goto finish;
}
r = ret;
goto finish;
}
/* Child */
}
if (config.use_syslog || config.daemonize)
daemon_log_use = DAEMON_LOG_SYSLOG;
if (daemon_close_all(-1) < 0) {
avahi_log_error("Failed to close remaining file descriptors: %s", strerror(errno));
goto finish;
}
if (make_runtime_dir() < 0)
goto finish;
if (config.drop_root) {
#ifdef ENABLE_CHROOT
if (config.use_chroot)
if (avahi_caps_reduce() < 0)
goto finish;
#endif
if (drop_root() < 0)
goto finish;
#ifdef ENABLE_CHROOT
if (config.use_chroot)
if (avahi_caps_reduce2() < 0)
goto finish;
#endif
}
if (daemon_pid_file_create() < 0) {
avahi_log_error("Failed to create PID file: %s", strerror(errno));
if (config.daemonize)
daemon_retval_send(1);
goto finish;
} else
wrote_pid_file = 1;
if (config.set_rlimits)
enforce_rlimits();
chdir("/");
#ifdef ENABLE_CHROOT
if (config.drop_root && config.use_chroot)
if (avahi_chroot_helper_start(argv0) < 0) {
avahi_log_error("failed to start chroot() helper daemon.");
goto finish;
}
#endif
avahi_log_info("%s "PACKAGE_VERSION" starting up.", argv0);
avahi_set_proc_title(argv0, "%s: starting up", argv0);
if (run_server(&config) == 0)
r = 0;
}
finish:
if (config.daemonize)
daemon_retval_done();
avahi_server_config_free(&config.server_config);
avahi_free(config.config_file);
avahi_strfreev(config.publish_dns_servers);
avahi_strfreev(resolv_conf_name_servers);
avahi_strfreev(resolv_conf_search_domains);
if (wrote_pid_file) {
#ifdef ENABLE_CHROOT
avahi_chroot_helper_unlink(pid_file_proc());
#else
daemon_pid_file_remove();
#endif
}
#ifdef ENABLE_CHROOT
avahi_chroot_helper_shutdown();
#endif
avahi_free(argv0);
return r;
}
/**
* Read the config file, determine if we should daemonize or not,
* and set up libevent dispatches.
*/
int main(int argc, char *argv[]) {
int cmdline_debug_level = 0;
int foreground = 0;
int option;
pid_t pid;
int kill=0;
int ret;
/* set some sane config defaults */
memset((void*)&config, 0, sizeof(gopher_conf_t));
config.port = 70;
config.base_dir = ".";
config.socket_backlog = 5;
config.config_file = DEFAULT_CONFIGFILE;
while((option = getopt(argc, argv, "d:c:fp:s:k")) != -1) {
switch(option) {
case 'd':
cmdline_debug_level = atoi(optarg);
if(cmdline_debug_level < 1 || cmdline_debug_level > 5) {
ERROR("Debug level must be 1-5 (default %d)", DEFAULT_DEBUGLEVEL);
usage_quit(argv[0]);
}
break;
case 'c':
config.config_file = optarg;
break;
case 'f':
foreground = 1;
break;
case 'p':
config.port = atoi(optarg);
break;
case 's':
config.base_dir = optarg;
case 'k':
kill = 1;
break;
default:
usage_quit(argv[0]);
}
}
debug_level(cmdline_debug_level ? cmdline_debug_level : DEFAULT_DEBUGLEVEL);
/* kill before config, so we don't have to specify a config file */
daemon_pid_file_ident = daemon_ident_from_argv0(argv[0]);
/* kill? */
if(kill) {
if((ret = daemon_pid_file_kill_wait(SIGTERM, 5)) < 0) {
ERROR("Failed to kill daemon: %s", strerror(errno));
}
exit(ret < 0 ? EXIT_FAILURE : EXIT_SUCCESS);
}
if((pid = daemon_pid_file_is_running()) >= 0) {
ERROR("Daemon already running as pid %u", pid);
exit(EXIT_FAILURE);
}
/* TODO: read the config file and verify sufficient config */
debug_level(cmdline_debug_level);
/* daemonize, or check for background daemon */
if(!foreground) {
debug_output(DBG_OUTPUT_SYSLOG, "evgopherd");
if(daemon_retval_init() < 0) {
ERROR("Could not set up daemon pipe");
exit(EXIT_FAILURE);
}
if((pid = daemon_fork()) < 0) {
daemon_retval_done();
ERROR("Forking error: %s", strerror(errno));
exit(EXIT_FAILURE);
} else if (pid) { /* parent */
if((ret = daemon_retval_wait(5)) < 0) {
ERROR("Could not receive startup retval from daemon process: %s", strerror(errno));
exit(EXIT_FAILURE);
}
if(ret > 0)
ERROR("Daemon declined to start. Error: %d\n", ret);
exit(ret == 0 ? EXIT_SUCCESS :EXIT_FAILURE);
}
}
if(daemon_signal_init(SIGINT, SIGTERM, SIGQUIT, SIGHUP, SIGPIPE, 0) < 0) {
ERROR("Could not set up signal handlers: %s", strerror(errno));
goto finish;
}
if(!foreground) {
if(daemon_pid_file_create() < 0) {
ERROR("Could not create pidfile: %s", strerror(errno));
if(!foreground)
daemon_retval_send(2);
goto finish;
}
/* should close handles here */
daemon_retval_send(0); /* started up to the point that we can rely on syslog */
}
WARN("Daemon started");
/* need to fork off a watchdog process */
while(!g_quitflag) {
/* fork and wait */
pid_t pid = fork();
if(pid == -1) {
ERROR("Error forking child process. Aborting");
g_quitflag = 1;
} else if(pid == 0) { /* child */
do_child_process();
} else { /* parent */
int status;
waitpid(pid, &status, 0);
if(WIFEXITED(status) && WEXITSTATUS(status)) {
/* exited with error */
ERROR("Error initializing child process. Aborting");
g_quitflag=1;
} else if(!WIFEXITED(status)) {
ERROR("Child process (%d) crashed. Restarting.", pid);
} else {
/* graceful exit... we've obviously terminated */
g_quitflag = 1;
}
}
}
WARN("Daemon exiting gracefully");
finish:
daemon_signal_done();
daemon_pid_file_remove();
return EXIT_SUCCESS;
}
