#include <config.h>
#endif
-#define VERSION 0
-#define REVISION 1
-#define RELEASE 0
-
-#define PIDFILE "/var/run/bpdyndnsd.pid"
-#define OBJECT_FILE "/home/bjoern/intranator/bpdyndnsd/objects.ser"
+#define PIDFILE "/var/run/bpdyndnsd/bpdyndnsd.pid"
#include <iostream>
#include <fstream>
#include <sys/types.h>
#include <signal.h>
-#include "updater.h"
-
-#include "config.cpp"
-#include "dhs.cpp"
-#include "logger.cpp"
-#include "ods.cpp"
-#include "service.cpp"
-#include "serviceholder.cpp"
-#include "updater.cpp"
-
+#include "updater.hpp"
using namespace std;
Updater::Ptr updater;
-bool is_online = true;
+volatile bool caught_sig_hup = false;
+volatile bool caught_sig_usr1 = false;
+volatile bool caught_sig_usr2 = false;
+volatile bool caught_sig_rtmin = false;
+volatile bool caught_sig_rtmax = false;
+volatile bool caught_sig_term = false;
/**
* Checks if a bpdyndnsd process is already running.
/**
- * Signal SIGTERM caught, releasing resources and exit.
- * @param param Parameter from the signal interface.
+ * Parent shutdown function
+ * @return 0 if all is fine, -1 otherwise
*/
-void terminate(int param)
+int shutdown_parent(bool remove_pid, int ret_val)
{
- updater->get_logger()->print_caught_sigterm();
+ // starting shutdown_parent
+ updater->get_logger()->print_starting_shutdown_parent();
- // unfortunately we can't call serialize_services in any destructor
- // cause some singleton are already destroyed !?!
- if ( updater->get_config()->serialize_services() != 0 )
- {
- updater.reset();
- exit(-1);
- }
+ // remove pidfile if requested
+ if(remove_pid)
+ unlink(PIDFILE);
+
+ // shutdown parent complete
+ updater->get_logger()->print_shutdown_parent_succeeded();
+
+ // release shared pointer
updater.reset();
- exit(0);
+
+ exit(ret_val);
}
+/**
+ * Shutdown function
+ * @return 0 if all is fine, -1 otherwise
+ */
+int shutdown()
+{
+ int ret_val = 0;
+
+ // starting shutdown
+ updater->get_logger()->print_starting_shutdown();
+
+ // serialize actual service objects
+ if ( updater->get_service_holder()->serialize_services() != 0 )
+ ret_val = -1;
+
+ // unlink pidfile
+ unlink(PIDFILE);
+
+ // shutdown complete
+ updater->get_logger()->print_shutdown_succeeded();
+
+ // release shared pointer
+ updater.reset();
+
+ return ret_val;
+}
+
/**
- * Signal SIGUSR1 caught, switching to offline mode.
+ * Signal SIGTERM caught (exit program)
* @param param Parameter from the signal interface.
*/
-void switch_to_offline(int param)
+void sigterm_func(int param)
{
- updater->get_logger()->print_caught_siguser1();
- is_online = false;
-}
+ caught_sig_term = true;
+} /*lint !e715 */
/**
- * Signal SIGHUP caught, reloading config and switching to online mode.
+ * Signal SIGUSR1 caught (switch to offline mode)
* @param param Parameter from the signal interface.
*/
-void reload_config(int param)
+void sigusr1_func(int param)
{
- updater->get_logger()->print_caught_sighup();
- if ( updater->reload_config() != 0 )
- exit(-1);
- is_online = true;
-}
+ caught_sig_usr1 = true;
+} /*lint !e715 */
+
+
+/**
+ * Signal SIGHUP caught (reload config and reset log level)
+ * @param param Parameter from the signal interface.
+ */
+void sighup_func(int param)
+{
+ caught_sig_hup = true;
+} /*lint !e715 */
+
+
+/**
+ * Signal SIGUSR2 caught (switch to online mode)
+ * @param param Parameter from the signal interface.
+ */
+void sigusr2_func(int param)
+{
+ caught_sig_usr2 = true;
+} /*lint !e715 */
+
+
+/**
+ * Signal SIGRTMIN caught (switch to online mode with webcheck enabled)
+ * @param param Parameter from the signal interface.
+ */
+void sigrtmin_func(int param)
+{
+ caught_sig_rtmin = true;
+} /*lint !e715 */
+
+
+/**
+ * Signal SIGRTMAX caught (increase log level)
+ * @param param Parameter from the signal interface.
+ */
+void sigrtmax_func(int param)
+{
+ caught_sig_rtmax = true;
+} /*lint !e715 */
/**
* Initialize the signals we handle.
+ * @return 0 if all is fine, -1 on error.
*/
int init_signals()
{
sighandler_t ret_val;
- ret_val = signal(SIGTERM,terminate);
+ ret_val = signal(SIGTERM, sigterm_func);
if ( ret_val == SIG_ERR )
{
updater->get_logger()->print_error_setting_signal("SIGTERM");
return -1;
}
- ret_val = signal(SIGUSR1,switch_to_offline);
+ ret_val = signal(SIGUSR1, sigusr1_func);
if ( ret_val == SIG_ERR )
{
updater->get_logger()->print_error_setting_signal("SIGUSR1");
return -1;
}
- ret_val = signal(SIGHUP,reload_config);
+ ret_val = signal(SIGHUP, sighup_func);
if ( ret_val == SIG_ERR )
{
updater->get_logger()->print_error_setting_signal("SIGHUP");
return -1;
}
+ ret_val = signal(SIGUSR2, sigusr2_func);
+ if ( ret_val == SIG_ERR )
+ {
+ updater->get_logger()->print_error_setting_signal("SIGUSR2");
+ return -1;
+ }
+ ret_val = signal(SIGRTMIN, sigrtmin_func);
+ if ( ret_val == SIG_ERR )
+ {
+ updater->get_logger()->print_error_setting_signal("SIGRTMIN");
+ return -1;
+ }
+ ret_val = signal(SIGRTMAX, sigrtmax_func);
+ if ( ret_val == SIG_ERR )
+ {
+ updater->get_logger()->print_error_setting_signal("SIGRTMAX");
+ return -1;
+ }
return 0;
}
if ( write_pidfile(pid) != 0 )
{
if ( kill(pid,SIGTERM) != 0 )
+ {
updater->get_logger()->print_error_kill_child(pid);
+ shutdown_parent(false,-1); /*lint !e534 */
+ }
else
+ {
updater->get_logger()->print_child_killed(pid);
- exit(-1);
+ shutdown_parent(true,-1); /*lint !e534 */
+ }
}
updater->get_logger()->print_runnig_as_daemon(pid);
- exit(0);
+ shutdown_parent(false,0); /*lint !e534 */
}
// child starts here
}
+ else
+ {
+ if ( write_pidfile(getpid()) != 0 )
+ {
+ shutdown(); /*lint !e534 */
+ exit(-1);
+ }
+ }
+ return 0;
+}
+
+/**
+ * Calculate how long the dialup mode burst period will last (dialup mode only)
+ * @return Timestamp until dialup period lasts or zero if dialup mode is disabled
+*/
+time_t calc_dialup_mode_burst_period()
+{
+ if (updater->get_config()->get_dialup_mode() )
+ return time(NULL) + updater->get_config()->get_dialup_burst_period_seconds();
+
return 0;
}
int main(int argc, char *argv[])
{
// initialize Updater
- Updater::Ptr _updater(new Updater);
- updater = _updater;
- _updater.reset();
-
- // load the cmd options
- if ( updater->init_config_from_cmd(argc,argv) != 0 )
- return -1;
+ updater = Updater::Ptr(new Updater);
- // load the config and service files
- if ( updater->init_config_from_files() != 0 )
+ // load config and initialize helper classes
+ if ( updater->load_config(argc,argv) != 0 )
return -1;
// open pidfile and check for running process
if ( init_daemon_mode(updater->get_config()->get_daemon_mode()) != 0 )
return -1;
+ // Should we start in offline mode?
+ bool old_online_state = false;
+ bool is_online = !updater->get_config()->get_start_offline();
+ bool webcheck_enabled = updater->get_config()->get_webcheck_enabled();
+ bool exit_now = false;
+
+ // One shot run if daemon mode is disabled
+ if (updater->get_config()->get_daemon_mode() != 1)
+ exit_now = true;
+
+ // Tell the world we are running
+ updater->get_logger()->print_started();
+
+ int original_log_level = updater->get_logger()->get_loglevel();
+
+ // Dialup mode: Sleep a specified number of seconds between network traffic
+ time_t dialup_mode_burst_mode_until = calc_dialup_mode_burst_period();
+ time_t dialup_mode_sleep_until = 0;
+
// service processing starts here
do
{
+ // Signal processing
+ if (caught_sig_usr1)
+ {
+ caught_sig_usr1 = false;
+ updater->get_logger()->print_caught_siguser1();
+
+ // Go offline
+ is_online = false;
+ } else if (caught_sig_usr2)
+ {
+ caught_sig_usr2 = false;
+ updater->get_logger()->print_caught_siguser2();
+
+ // Go online
+ is_online = true;
+ webcheck_enabled = false;
+ dialup_mode_burst_mode_until = calc_dialup_mode_burst_period();
+ } else if (caught_sig_rtmin)
+ {
+ caught_sig_rtmin = false;
+ updater->get_logger()->print_caught_sigrtmin();
+
+ // Go online - with webcheck
+ is_online = true;
+ webcheck_enabled = true;
+ dialup_mode_burst_mode_until = calc_dialup_mode_burst_period();
+ } else if (caught_sig_rtmax)
+ {
+ caught_sig_rtmax = false;
+
+ // Increase log level
+ int new_loglevel = updater->get_logger()->get_loglevel() + 1;
+ updater->get_logger()->print_caught_sigrtmax(new_loglevel);
+ updater->get_logger()->set_loglevel(new_loglevel);
+ } else if (caught_sig_term)
+ {
+ caught_sig_term = false;
+ updater->get_logger()->print_caught_sigterm();
+
+ exit_now = true;
+ } else if (caught_sig_hup)
+ {
+ caught_sig_hup = false;
+ updater->get_logger()->print_caught_sighup();
+
+ if ( updater->reload_config() != 0 )
+ {
+ updater->get_logger()->print_conf_reload_failed_exit();
+ exit(-1);
+ }
+
+ // Reset log level to startup log level in case it got elevated by SIGRTMAX
+ updater->get_logger()->set_loglevel(original_log_level);
+
+ // Reset dialup mode burst period on config reload
+ dialup_mode_burst_mode_until = calc_dialup_mode_burst_period();
+ }
+
+ // State handling
if ( is_online == true )
{
- // TODO: Get the actual IP of this host (through DNS or WEBCHECK_IP). Perhaps new Class which can handle both.
+ // Check if webcheck_enabled differs due to caught singnal then set it in config correspondingly
+ if ( updater->get_config()->get_webcheck_enabled() != webcheck_enabled ) /*lint !e731 */
+ updater->get_config()->set_webcheck_enabled(webcheck_enabled);
- // update all configured services
- updater->update_services();
+ if (updater->get_config()->get_dialup_mode() == false ||
+ (time(NULL) > dialup_mode_sleep_until || time(NULL) < dialup_mode_burst_mode_until) )
+ {
+ // update all configured services
+ updater->update_services(is_online != old_online_state);
+
+ // Refresh sleep state (dialup mode only)
+ dialup_mode_sleep_until = time(NULL) + updater->get_config()->get_dialup_sleep_seconds();
+ } else
+ updater->get_logger()->print_sleep_dialup_mode(dialup_mode_sleep_until);
}
else
{
updater->get_logger()->print_offline_mode();
}
- // Snore, snore... don't hug the cpu if we are in daemon_mode.
- if ( updater->get_config()->get_daemon_mode() == 1 )
- sleep(5);
-
- }while ( updater->get_config()->get_daemon_mode() == 1 );
+ // Keep old online state so we can determine
+ // if we switched from offline to online
+ old_online_state = is_online;
+ // Snore, snore... don't hog the cpu if we are in daemon_mode.
+ if ( !exit_now )
+ sleep(10); /*lint !e534 */
+ } while ( !exit_now );
// Serialize services to save their actual state.
- if ( updater->get_config()->serialize_services() != 0 )
+ if ( shutdown() != 0 )
return -1;
return 0;
git://developer.intra2net.com / bpdyndnsd / blobdiff