developer.intra2net.com Git - bpdyndnsd/blob - src/main.cpp

   1 /** @file
   2  * @brief The main function.
   3  *
   4  *
   5  *
   6  * @copyright Intra2net AG
   7  * @license GPLv2
   8 */
   9
  10 #ifdef HAVE_CONFIG_H
  11 #include <config.h>
  12 #endif
  13
  14 #define PIDFILE "/var/run/bpdyndnsd/bpdyndnsd.pid"
  15
  16 #include <iostream>
  17 #include <fstream>
  18 #include <list>
  19 #include <string>
  20 #include <boost/foreach.hpp>
  21 #include <sys/types.h>
  22 #include <signal.h>
  23
  24 #include "updater.hpp"
  25
  26 using namespace std;
  27
  28 Updater::Ptr updater;
  29 volatile bool exit_now = false;
  30 volatile bool caught_sig_hup = false;
  31 volatile bool caught_sig_usr1 = false;
  32 volatile bool caught_sig_usr2 = false;
  33 volatile bool caught_sig_rtmin = false;
  34 volatile bool caught_sig_term = false;
  35
  36 /**
  37  * Checks if a bpdyndnsd process is already running.
  38  * @param updater Shared Pointer to updater, needed for logging.
  39  * @return 0 if process not running already or pid of already running process.
  40  */
  41 int check_for_running_process()
  42 {
  43     ifstream pidfile(PIDFILE);
  44     if ( pidfile.is_open() )
  45     {
  46         int pid;
  47         pidfile >> pid;
  48         if ( pid > 0 )
  49             updater->get_logger()->print_pid_found(pid);
  50         else
  51             return 0;
  52
  53         // check if process still running ret_val==-1 -> not runnig, ret_val==0 -> running
  54         if ( kill(pid,0) == 0)
  55         {
  56             updater->get_logger()->print_process_already_running(pid);
  57             pidfile.close();
  58             return pid;
  59         }
  60     }
  61     pidfile.close();
  62     return 0;
  63 }
  64
  65
  66 /**
  67  * Writes the pid into the pidfile.
  68  * @param pid The process's pid.
  69  */
  70 int write_pidfile(int pid)
  71 {
  72     ofstream pidfile(PIDFILE);
  73     if ( pidfile.is_open() )
  74     {
  75         pidfile << pid << endl;
  76     }
  77     else
  78     {
  79         updater->get_logger()->print_error_opening_rw(PIDFILE);
  80         return -1;
  81     }
  82     pidfile.close();
  83     return 0;
  84 }
  85
  86
  87 /**
  88  * Parent shutdown function
  89  * @return 0 if all is fine, -1 otherwise
  90  */
  91 int shutdown_parent(bool remove_pid, int ret_val)
  92 {
  93     // starting shutdown_parent
  94     updater->get_logger()->print_starting_shutdown_parent();
  95
  96     // remove pidfile if requested
  97     if(remove_pid)
  98         unlink(PIDFILE);
  99
 100     // shutdown parent complete
 101     updater->get_logger()->print_shutdown_parent_succeeded();
 102
 103     // release shared pointer
 104     updater.reset();
 105
 106     exit(ret_val);
 107 }
 108
 109
 110 /**
 111  * Shutdown function
 112  * @return 0 if all is fine, -1 otherwise
 113  */
 114 int shutdown()
 115 {
 116     int ret_val = 0;
 117
 118     // starting shutdown
 119     updater->get_logger()->print_starting_shutdown();
 120
 121     // serialize actual service objects
 122     if ( updater->get_service_holder()->serialize_services() != 0 )
 123         ret_val = -1;
 124
 125     // unlink pidfile
 126     unlink(PIDFILE);
 127
 128     // shutdown complete
 129     updater->get_logger()->print_shutdown_succeeded();
 130
 131     // release shared pointer
 132     updater.reset();
 133
 134     return ret_val;
 135 }
 136
 137
 138 /**
 139  * Signal SIGTERM caught (exit program)
 140  * @param param Parameter from the signal interface.
 141  */
 142 void sigterm_func(int param)
 143 {
 144     caught_sig_term = true;
 145 } /*lint !e715 */
 146
 147
 148 /**
 149  * Signal SIGUSR1 caught (switch to offline mode)
 150  * @param param Parameter from the signal interface.
 151  */
 152 void sigusr1_func(int param)
 153 {
 154     caught_sig_usr1 = true;
 155 } /*lint !e715 */
 156
 157
 158 /**
 159  * Signal SIGHUP caught (reload config)
 160  * @param param Parameter from the signal interface.
 161  */
 162 void sighup_func(int param)
 163 {
 164     caught_sig_hup = true;
 165 } /*lint !e715 */
 166
 167
 168 /**
 169  * Signal SIGUSR2 caught (switch to online mode)
 170  * @param param Parameter from the signal interface.
 171  */
 172 void sigusr2_func(int param)
 173 {
 174     caught_sig_usr2 = true;
 175 } /*lint !e715 */
 176
 177
 178 /**
 179  * Signal SIGRTMIN caught (switch to online mode with webcheck enabled)
 180  * @param param Parameter from the signal interface.
 181  */
 182 void sigrtmin_func(int param)
 183 {
 184     caught_sig_rtmin = true;
 185 } /*lint !e715 */
 186
 187
 188 /**
 189  * Initialize the signals we handle.
 190  * @return 0 if all is fine, -1 on error.
 191  */
 192 int init_signals()
 193 {
 194     sighandler_t ret_val;
 195     ret_val = signal(SIGTERM, sigterm_func);
 196     if ( ret_val == SIG_ERR )
 197     {
 198         updater->get_logger()->print_error_setting_signal("SIGTERM");
 199         return -1;
 200     }
 201     ret_val = signal(SIGUSR1, sigusr1_func);
 202     if ( ret_val == SIG_ERR )
 203     {
 204         updater->get_logger()->print_error_setting_signal("SIGUSR1");
 205         return -1;
 206     }
 207     ret_val = signal(SIGHUP, sighup_func);
 208     if ( ret_val == SIG_ERR )
 209     {
 210         updater->get_logger()->print_error_setting_signal("SIGHUP");
 211         return -1;
 212     }
 213     ret_val = signal(SIGUSR2, sigusr2_func);
 214     if ( ret_val == SIG_ERR )
 215     {
 216         updater->get_logger()->print_error_setting_signal("SIGUSR2");
 217         return -1;
 218     }
 219     ret_val = signal(SIGRTMIN, sigrtmin_func);
 220     if ( ret_val == SIG_ERR )
 221     {
 222         updater->get_logger()->print_error_setting_signal("SIGRTMIN");
 223         return -1;
 224     }
 225
 226     return 0;
 227 }
 228
 229
 230 /**
 231  * Try to run in daemon mode if enabled in config.
 232  * @param daemon_mode True if process should detach to init (run as daemon), false if not detach to init.
 233  * @return 0 if all is fine, -1 on error.
 234  */
 235 int init_daemon_mode(bool daemon_mode)
 236 {
 237     updater->get_logger()->print_daemon_mode(daemon_mode);
 238     if ( daemon_mode == true )
 239     {
 240         int pid = fork();
 241         if ( pid < 0 )
 242         {
 243             // error fork
 244             updater->get_logger()->print_error_fork();
 245             return -1;
 246         }
 247         else if ( pid > 0 )
 248         {
 249             // parent continues here
 250             if ( write_pidfile(pid) != 0 )
 251             {
 252                 if ( kill(pid,SIGTERM) != 0 )
 253                 {
 254                     updater->get_logger()->print_error_kill_child(pid);
 255                     shutdown_parent(false,-1); /*lint !e534 */
 256                 }
 257                 else
 258                 {
 259                     updater->get_logger()->print_child_killed(pid);
 260                     shutdown_parent(true,-1); /*lint !e534 */
 261                 }
 262             }
 263             updater->get_logger()->print_runnig_as_daemon(pid);
 264             shutdown_parent(false,0); /*lint !e534 */
 265         }
 266         // child starts here
 267     }
 268     else
 269     {
 270         if ( write_pidfile(getpid()) != 0 )
 271         {
 272             shutdown(); /*lint !e534 */
 273             exit(-1);
 274         }
 275     }
 276     return 0;
 277 }
 278
 279 /**
 280  * @brief The main part.
 281  * @param argc Number of arguments
 282  * @param argv Command line arguments
 283  * @return 0 if all is fine.
 284  */
 285 int main(int argc, char *argv[])
 286 {
 287     // initialize Updater
 288     updater = Updater::Ptr(new Updater);
 289
 290     // load config and initialize helper classes
 291     if ( updater->load_config(argc,argv) != 0 )
 292         return -1;
 293
 294     // open pidfile and check for running process
 295     if ( check_for_running_process() != 0)
 296         return -1;
 297
 298     // init signal handling
 299     if ( init_signals() != 0)
 300         return -1;
 301
 302     // init daemon_mode if enabled
 303     if ( init_daemon_mode(updater->get_config()->get_daemon_mode()) != 0 )
 304         return -1;
 305
 306     // Should we start in offline mode?
 307     bool is_online = !updater->get_config()->get_start_offline();
 308     bool webcheck_enabled = updater->get_config()->get_webcheck_enabled();
 309
 310     // One shot run if daemon mode is disabled
 311     if (updater->get_config()->get_daemon_mode() != 1)
 312         exit_now = true;
 313
 314     // service processing starts here
 315     do
 316     {
 317         // Signal processing
 318         if (caught_sig_usr1)
 319         {
 320             caught_sig_usr1 = false;
 321             updater->get_logger()->print_caught_siguser1();
 322
 323             // Go offline
 324             is_online = false;
 325         } else if (caught_sig_usr2)
 326         {
 327             caught_sig_usr2 = false;
 328             updater->get_logger()->print_caught_siguser2();
 329
 330             // Go online
 331             is_online = true;
 332             webcheck_enabled = false;
 333         } else if (caught_sig_rtmin)
 334         {
 335             caught_sig_rtmin = false;
 336             updater->get_logger()->print_caught_sigrtmin();
 337
 338             // Go online - with webcheck
 339             is_online = true;
 340             webcheck_enabled = true;
 341         } else if (caught_sig_term)
 342         {
 343             caught_sig_term = false;
 344             updater->get_logger()->print_caught_sigterm();
 345
 346             exit_now = true;
 347         } else if (caught_sig_hup)
 348         {
 349             caught_sig_hup = false;
 350             updater->get_logger()->print_caught_sighup();
 351
 352             if ( updater->reload_config() != 0 )
 353             {
 354                 updater->get_logger()->print_conf_reload_failed_exit();
 355                 exit(-1);
 356             }
 357         }
 358
 359         // State handling
 360         if ( is_online == true )
 361         {
 362             // Check if webcheck_enabled differs due to caught singnal then set it in config correspondingly
 363             if ( updater->get_config()->get_webcheck_enabled() != webcheck_enabled )    /*lint !e731 */
 364                 updater->get_config()->set_webcheck_enabled(webcheck_enabled);
 365
 366             // update all configured services
 367             updater->update_services();
 368         }
 369         else
 370         {
 371             // We are in offline mode, do nothing, expect printing "offline mode".
 372             updater->get_logger()->print_offline_mode();
 373         }
 374
 375         // Snore, snore... don't hog the cpu if we are in daemon_mode.
 376         if ( !exit_now )
 377             sleep(10); /*lint !e534 */
 378     } while ( !exit_now );
 379
 380     // Serialize services to save their actual state.
 381     if ( shutdown() != 0 )
 382         return -1;
 383
 384     return 0;
 385 }