git.sur5r.net Git - bacula/bacula/blob - bacula/src/dird/dird.c

   1 /*
   2  *
   3  *   Bacula Director daemon -- this is the main program
   4  *
   5  *     Kern Sibbald, March MM
   6  *
   7  *   Version $Id$
   8  */
   9 /*
  10    Copyright (C) 2000-2003 Kern Sibbald and John Walker
  11
  12    This program is free software; you can redistribute it and/or
  13    modify it under the terms of the GNU General Public License as
  14    published by the Free Software Foundation; either version 2 of
  15    the License, or (at your option) any later version.
  16
  17    This program is distributed in the hope that it will be useful,
  18    but WITHOUT ANY WARRANTY; without even the implied warranty of
  19    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  20    General Public License for more details.
  21
  22    You should have received a copy of the GNU General Public
  23    License along with this program; if not, write to the Free
  24    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  25    MA 02111-1307, USA.
  26
  27  */
  28
  29 #include "bacula.h"
  30 #include "dird.h"
  31
  32 /* Forward referenced subroutines */
  33 static void terminate_dird(int sig);
  34 static int check_resources();
  35 static void reload_config(int sig);
  36
  37 /* Exported subroutines */
  38
  39
  40 /* Imported subroutines */
  41 extern JCR *wait_for_next_job(char *runjob);
  42 extern void term_scheduler();
  43 extern void term_ua_server();
  44 extern int do_backup(JCR *jcr);
  45 extern void backup_cleanup(void);
  46 extern void start_UA_server(char *addr, int port);
  47 extern void run_job(JCR *jcr);
  48 extern void init_job_server(int max_workers);
  49
  50 static char *configfile = NULL;
  51 static char *runjob = NULL;
  52 static int background = 1;
  53
  54 /* Globals Exported */
  55 DIRRES *director;                     /* Director resource */
  56 int FDConnectTimeout;
  57 int SDConnectTimeout;
  58
  59 #define CONFIG_FILE "./bacula-dir.conf" /* default configuration file */
  60
  61 static void usage()
  62 {
  63    fprintf(stderr, _(
  64 "\nVersion: " VERSION " (" BDATE ")\n\n"
  65 "Usage: dird [-f -s] [-c config_file] [-d debug_level] [config_file]\n"
  66 "       -c <file>   set configuration file to file\n"
  67 "       -dnn        set debug level to nn\n"
  68 "       -f          run in foreground (for debugging)\n"
  69 "       -g          groupid\n"
  70 "       -r <job>    run <job> now\n"
  71 "       -s          no signals\n"
  72 "       -t          test - read configuration and exit\n"
  73 "       -u          userid\n"
  74 "       -v          verbose user messages\n"
  75 "       -?          print this message.\n"
  76 "\n"));
  77
  78    exit(1);
  79 }
  80
  81
  82 /*********************************************************************
  83  *
  84  *         Main Bacula Server program
  85  *
  86  */
  87 int main (int argc, char *argv[])
  88 {
  89    int ch;
  90    JCR *jcr;
  91    int no_signals = FALSE;
  92    int test_config = FALSE;
  93    char *uid = NULL;
  94    char *gid = NULL;
  95
  96    init_stack_dump();
  97    my_name_is(argc, argv, "bacula-dir");
  98    textdomain("bacula-dir");
  99    init_msg(NULL, NULL);              /* initialize message handler */
 100    daemon_start_time = time(NULL);
 101    memset(&last_job, 0, sizeof(last_job));
 102
 103    while ((ch = getopt(argc, argv, "c:d:fg:r:stu:v?")) != -1) {
 104       switch (ch) {
 105          case 'c':                    /* specify config file */
 106             if (configfile != NULL) {
 107                free(configfile);
 108             }
 109             configfile = bstrdup(optarg);
 110             break;
 111
 112          case 'd':                    /* set debug level */
 113             debug_level = atoi(optarg);
 114             if (debug_level <= 0) {
 115                debug_level = 1;
 116             }
 117             Dmsg1(0, "Debug level = %d\n", debug_level);
 118             break;
 119
 120          case 'f':                    /* run in foreground */
 121             background = FALSE;
 122             break;
 123
 124          case 'g':                    /* set group id */
 125             gid = optarg;
 126             break;
 127
 128          case 'r':                    /* run job */
 129             if (runjob != NULL) {
 130                free(runjob);
 131             }
 132             if (optarg) {
 133                runjob = bstrdup(optarg);
 134             }
 135             break;
 136
 137          case 's':                    /* turn off signals */
 138             no_signals = TRUE;
 139             break;
 140
 141          case 't':                    /* test config */
 142             test_config = TRUE;
 143             break;
 144
 145          case 'u':                    /* set uid */
 146             uid = optarg;
 147             break;
 148
 149          case 'v':                    /* verbose */
 150             verbose++;
 151             break;
 152
 153          case '?':
 154          default:
 155             usage();
 156
 157       }
 158    }
 159    argc -= optind;
 160    argv += optind;
 161
 162    if (!no_signals) {
 163       init_signals(terminate_dird);
 164    }
 165    signal(SIGCHLD, SIG_IGN);
 166
 167    if (argc) {
 168       if (configfile != NULL) {
 169          free(configfile);
 170       }
 171       configfile = bstrdup(*argv);
 172       argc--;
 173       argv++;
 174    }
 175    if (argc) {
 176       usage();
 177    }
 178
 179    if (configfile == NULL) {
 180       configfile = bstrdup(CONFIG_FILE);
 181    }
 182
 183    parse_config(configfile);
 184
 185    if (!check_resources()) {
 186       Jmsg(NULL, M_ERROR_TERM, 0, "Please correct configuration file: %s\n", configfile);
 187    }
 188
 189    if (test_config) {
 190       terminate_dird(0);
 191    }
 192
 193    my_name_is(0, NULL, director->hdr.name);    /* set user defined name */
 194
 195    FDConnectTimeout = (int)director->FDConnectTimeout;
 196    SDConnectTimeout = (int)director->SDConnectTimeout;
 197
 198    if (background) {
 199       daemon_start();
 200       init_stack_dump();              /* grab new pid */
 201    }
 202
 203    drop(uid, gid);                    /* reduce priveleges if requested */
 204
 205    /* Create pid must come after we are a daemon -- so we have our final pid */
 206    create_pid_file(director->pid_directory, "bacula-dir", director->DIRport);
 207
 208 /* signal(SIGHUP, reload_config); */
 209
 210    init_console_msg(working_directory);
 211
 212    set_thread_concurrency(director->MaxConcurrentJobs * 2 +
 213       4 /* UA */ + 4 /* sched+watchdog+jobsvr+misc */);
 214
 215    Dmsg0(200, "Start UA server\n");
 216    start_UA_server(director->DIRaddr, director->DIRport);
 217
 218    start_watchdog();                  /* start network watchdog thread */
 219
 220    init_job_server(director->MaxConcurrentJobs);
 221
 222    Dmsg0(200, "wait for next job\n");
 223    /* Main loop -- call scheduler to get next job to run */
 224    while ((jcr = wait_for_next_job(runjob))) {
 225       run_job(jcr);                   /* run job */
 226       if (runjob) {                   /* command line, run a single job? */
 227          break;                       /* yes, terminate */
 228       }
 229    }
 230
 231    terminate_dird(0);
 232 }
 233
 234 /* Cleanup and then exit */
 235 static void terminate_dird(int sig)
 236 {
 237    static int already_here = FALSE;
 238
 239    if (already_here) {                /* avoid recursive temination problems */
 240       exit(1);
 241    }
 242    already_here = TRUE;
 243    delete_pid_file(director->pid_directory, "bacula-dir",
 244                    director->DIRport);
 245    stop_watchdog();
 246    signal(SIGCHLD, SIG_IGN);          /* don't worry about children now */
 247    term_scheduler();
 248    if (runjob) {
 249       free(runjob);
 250    }
 251    if (configfile != NULL) {
 252       free(configfile);
 253    }
 254    if (debug_level > 5) {
 255       print_memory_pool_stats();
 256    }
 257    free_config_resources();
 258    term_ua_server();
 259    term_msg();                        /* terminate message handler */
 260    close_memory_pool();               /* release free memory in pool */
 261    sm_dump(False);
 262    exit(sig != 0);
 263 }
 264
 265 /*
 266  * If we get here, we have received a SIGHUP, which means to
 267  * reread our configuration file.
 268  *
 269  *  ***FIXME***  Check that there are no jobs running before
 270  *               doing this.
 271  */
 272 static void reload_config(int sig)
 273 {
 274    static int already_here = FALSE;
 275    sigset_t set;
 276
 277    if (already_here) {
 278       abort();                        /* Oops, recursion -> die */
 279    }
 280    already_here = TRUE;
 281    sigfillset(&set);
 282    sigprocmask(SIG_BLOCK, &set, NULL);
 283
 284    free_config_resources();
 285
 286    parse_config(configfile);
 287
 288    Dmsg0(200, "check_resources()\n");
 289    if (!check_resources()) {
 290       Jmsg(NULL, M_ERROR_TERM, 0, _("Please correct configuration file: %s\n"), configfile);
 291    }
 292
 293    /* Reset globals */
 294    working_directory = director->working_directory;
 295    FDConnectTimeout = director->FDConnectTimeout;
 296    SDConnectTimeout = director->SDConnectTimeout;
 297
 298    sigprocmask(SIG_UNBLOCK, &set, NULL);
 299    signal(SIGHUP, reload_config);
 300    already_here = FALSE;
 301    Dmsg0(0, "Director's configuration file reread.\n");
 302 }
 303
 304 /*
 305  * Make a quick check to see that we have all the
 306  * resources needed.
 307  *
 308  *  **** FIXME **** this routine could be a lot more
 309  *   intelligent and comprehensive.
 310  */
 311 static int check_resources()
 312 {
 313    int OK = TRUE;
 314    JOB *job;
 315
 316    LockRes();
 317
 318    job  = (JOB *)GetNextRes(R_JOB, NULL);
 319    director = (DIRRES *)GetNextRes(R_DIRECTOR, NULL);
 320    if (!director) {
 321       Jmsg(NULL, M_FATAL, 0, _("No Director resource defined in %s\n\
 322 Without that I don't know who I am :-(\n"), configfile);
 323       OK = FALSE;
 324    } else {
 325       if (!director->working_directory) {
 326          Jmsg(NULL, M_FATAL, 0, _("No working directory specified. Cannot continue.\n"));
 327          OK = FALSE;
 328       }
 329       working_directory = director->working_directory;
 330       if (!director->messages) {       /* If message resource not specified */
 331          director->messages = (MSGS *)GetNextRes(R_MSGS, NULL);
 332          if (!director->messages) {
 333             Jmsg(NULL, M_FATAL, 0, _("No Messages resource defined in %s\n"), configfile);
 334             OK = FALSE;
 335          }
 336       }
 337       if (GetNextRes(R_DIRECTOR, (RES *)director) != NULL) {
 338          Jmsg(NULL, M_FATAL, 0, _("Only one Director resource permitted in %s\n"),
 339             configfile);
 340          OK = FALSE;
 341       }
 342    }
 343
 344    if (!job) {
 345       Jmsg(NULL, M_FATAL, 0, _("No Job records defined in %s\n"), configfile);
 346       OK = FALSE;
 347    }
 348    for (job=NULL; (job = (JOB *)GetNextRes(R_JOB, (RES *)job)); ) {
 349       if (!job->client) {
 350          Jmsg(NULL, M_FATAL, 0, _("No Client record defined for job %s\n"), job->hdr.name);
 351          OK = FALSE;
 352       }
 353       if (!job->fileset) {
 354          Jmsg(NULL, M_FATAL, 0, _("No FileSet record defined for job %s\n"), job->hdr.name);
 355          OK = FALSE;
 356       }
 357       if (!job->storage && job->JobType != JT_VERIFY) {
 358          Jmsg(NULL, M_FATAL, 0, _("No Storage resource defined for job %s\n"), job->hdr.name);
 359          OK = FALSE;
 360       }
 361       if (!job->pool) {
 362          Jmsg(NULL, M_FATAL, 0, _("No Pool resource defined for job %s\n"), job->hdr.name);
 363          OK = FALSE;
 364       }
 365       if (job->client && job->client->catalog) {
 366          CAT *catalog = job->client->catalog;
 367          B_DB *db;
 368
 369          /*
 370           * Make sure we can open catalog, otherwise print a warning
 371           * message because the server is probably not running.
 372           */
 373          db = db_init_database(NULL, catalog->db_name, catalog->db_user,
 374                             catalog->db_password, catalog->db_address,
 375                             catalog->db_port, catalog->db_socket);
 376          if (!db_open_database(NULL, db)) {
 377             Jmsg(NULL, M_FATAL,  0, "%s", db_strerror(db));
 378          } else {
 379             /* If a pool is defined for this job, create the pool DB
 380              *  record if it is not already created.
 381              */
 382             if (job->pool) {
 383                create_pool(NULL, db, job->pool, 0);  /* update request */
 384             }
 385             db_close_database(NULL, db);
 386          }
 387
 388       } else {
 389          if (job->client) {
 390             Jmsg(NULL, M_FATAL, 0, _("No Catalog resource defined for client %s\n"),
 391                job->client->hdr.name);
 392             OK = FALSE;
 393          }
 394       }
 395    }
 396
 397    UnlockRes();
 398    if (OK) {
 399       close_msg(NULL);                /* close temp message handler */
 400       init_msg(NULL, director->messages); /* open daemon message handler */
 401    }
 402    return OK;
 403 }