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-2005 Kern Sibbald
  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.
  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
  36 /* Exported subroutines */
  37
  38 extern "C" void reload_config(int sig);
  39
  40
  41 /* Imported subroutines */
  42 JCR *wait_for_next_job(char *runjob);
  43 void term_scheduler();
  44 void term_ua_server();
  45 void start_UA_server(dlist *addrs);
  46 void init_job_server(int max_workers);
  47 void term_job_server();
  48 void store_jobtype(LEX *lc, RES_ITEM *item, int index, int pass);
  49 void store_level(LEX *lc, RES_ITEM *item, int index, int pass);
  50 void store_replace(LEX *lc, RES_ITEM *item, int index, int pass);
  51 void init_device_resources();
  52
  53 static char *configfile = NULL;
  54 static char *runjob = NULL;
  55 static int background = 1;
  56 static void init_reload(void);
  57
  58 /* Globals Exported */
  59 DIRRES *director;                     /* Director resource */
  60 int FDConnectTimeout;
  61 int SDConnectTimeout;
  62
  63 /* Globals Imported */
  64 extern int r_first, r_last;           /* first and last resources */
  65 extern RES_TABLE resources[];
  66 extern RES **res_head;
  67 extern RES_ITEM job_items[];
  68 extern URES res_all;
  69
  70
  71 #define CONFIG_FILE "./bacula-dir.conf" /* default configuration file */
  72
  73 static void usage()
  74 {
  75    fprintf(stderr, _(
  76 "Copyright (C) 2000-2005 Kern Sibbald.\n"
  77 "\nVersion: " VERSION " (" BDATE ")\n\n"
  78 "Usage: dird [-f -s] [-c config_file] [-d debug_level] [config_file]\n"
  79 "       -c <file>   set configuration file to file\n"
  80 "       -dnn        set debug level to nn\n"
  81 "       -f          run in foreground (for debugging)\n"
  82 "       -g          groupid\n"
  83 "       -r <job>    run <job> now\n"
  84 "       -s          no signals\n"
  85 "       -t          test - read configuration and exit\n"
  86 "       -u          userid\n"
  87 "       -v          verbose user messages\n"
  88 "       -?          print this message.\n"
  89 "\n"));
  90
  91    exit(1);
  92 }
  93
  94
  95 /*********************************************************************
  96  *
  97  *         Main Bacula Server program
  98  *
  99  */
 100 int main (int argc, char *argv[])
 101 {
 102    int ch;
 103    JCR *jcr;
 104    int no_signals = FALSE;
 105    int test_config = FALSE;
 106    char *uid = NULL;
 107    char *gid = NULL;
 108
 109    init_stack_dump();
 110    my_name_is(argc, argv, "bacula-dir");
 111    textdomain("bacula");
 112    init_msg(NULL, NULL);              /* initialize message handler */
 113    init_reload();
 114    daemon_start_time = time(NULL);
 115
 116    while ((ch = getopt(argc, argv, "c:d:fg:r:stu:v?")) != -1) {
 117       switch (ch) {
 118       case 'c':                    /* specify config file */
 119          if (configfile != NULL) {
 120             free(configfile);
 121          }
 122          configfile = bstrdup(optarg);
 123          break;
 124
 125       case 'd':                    /* set debug level */
 126          debug_level = atoi(optarg);
 127          if (debug_level <= 0) {
 128             debug_level = 1;
 129          }
 130          Dmsg1(0, "Debug level = %d\n", debug_level);
 131          break;
 132
 133       case 'f':                    /* run in foreground */
 134          background = FALSE;
 135          break;
 136
 137       case 'g':                    /* set group id */
 138          gid = optarg;
 139          break;
 140
 141       case 'r':                    /* run job */
 142          if (runjob != NULL) {
 143             free(runjob);
 144          }
 145          if (optarg) {
 146             runjob = bstrdup(optarg);
 147          }
 148          break;
 149
 150       case 's':                    /* turn off signals */
 151          no_signals = TRUE;
 152          break;
 153
 154       case 't':                    /* test config */
 155          test_config = TRUE;
 156          break;
 157
 158       case 'u':                    /* set uid */
 159          uid = optarg;
 160          break;
 161
 162       case 'v':                    /* verbose */
 163          verbose++;
 164          break;
 165
 166       case '?':
 167       default:
 168          usage();
 169
 170       }
 171    }
 172    argc -= optind;
 173    argv += optind;
 174
 175    if (!no_signals) {
 176       init_signals(terminate_dird);
 177    }
 178
 179    if (argc) {
 180       if (configfile != NULL) {
 181          free(configfile);
 182       }
 183       configfile = bstrdup(*argv);
 184       argc--;
 185       argv++;
 186    }
 187    if (argc) {
 188       usage();
 189    }
 190
 191    if (configfile == NULL) {
 192       configfile = bstrdup(CONFIG_FILE);
 193    }
 194
 195    parse_config(configfile);
 196
 197    if (init_tls() != 0) {
 198       Jmsg((JCR *)NULL, M_ERROR_TERM, 0, _("TLS library initialization failed.\n"));
 199    }
 200
 201    if (!check_resources()) {
 202       Jmsg((JCR *)NULL, M_ERROR_TERM, 0, _("Please correct configuration file: %s\n"), configfile);
 203    }
 204
 205    if (test_config) {
 206       terminate_dird(0);
 207    }
 208
 209    my_name_is(0, NULL, director->hdr.name);    /* set user defined name */
 210
 211    FDConnectTimeout = (int)director->FDConnectTimeout;
 212    SDConnectTimeout = (int)director->SDConnectTimeout;
 213
 214    if (background) {
 215       daemon_start();
 216       init_stack_dump();              /* grab new pid */
 217    }
 218
 219    /* Create pid must come after we are a daemon -- so we have our final pid */
 220    create_pid_file(director->pid_directory, "bacula-dir", get_first_port_host_order(director->DIRaddrs));
 221    read_state_file(director->working_directory, "bacula-dir", get_first_port_host_order(director->DIRaddrs));
 222
 223    drop(uid, gid);                    /* reduce priveleges if requested */
 224
 225    signal(SIGHUP, reload_config);
 226
 227    init_console_msg(working_directory);
 228
 229    init_python_interpreter(director->hdr.name, director->scripts_directory,
 230        "DirStartUp");
 231
 232    set_thread_concurrency(director->MaxConcurrentJobs * 2 +
 233       4 /* UA */ + 4 /* sched+watchdog+jobsvr+misc */);
 234
 235    Dmsg0(200, "Start UA server\n");
 236    start_UA_server(director->DIRaddrs);
 237
 238    start_watchdog();                  /* start network watchdog thread */
 239
 240    init_jcr_subsystem();              /* start JCR watchdogs etc. */
 241
 242    init_job_server(director->MaxConcurrentJobs);
 243
 244 // init_device_resources();
 245
 246    Dmsg0(200, "wait for next job\n");
 247    /* Main loop -- call scheduler to get next job to run */
 248    while ((jcr = wait_for_next_job(runjob))) {
 249       run_job(jcr);                   /* run job */
 250       free_jcr(jcr);                  /* release jcr */
 251       if (runjob) {                   /* command line, run a single job? */
 252          break;                       /* yes, terminate */
 253       }
 254    }
 255
 256    terminate_dird(0);
 257 }
 258
 259 /* Cleanup and then exit */
 260 static void terminate_dird(int sig)
 261 {
 262    static bool already_here = false;
 263
 264    if (already_here) {                /* avoid recursive temination problems */
 265       exit(1);
 266    }
 267    already_here = true;
 268    generate_daemon_event(NULL, "Exit");
 269    write_state_file(director->working_directory, "bacula-dir", get_first_port_host_order(director->DIRaddrs));
 270    delete_pid_file(director->pid_directory, "bacula-dir", get_first_port_host_order(director->DIRaddrs));
 271 // signal(SIGCHLD, SIG_IGN);          /* don't worry about children now */
 272    term_scheduler();
 273    term_job_server();
 274    if (runjob) {
 275       free(runjob);
 276    }
 277    if (configfile != NULL) {
 278       free(configfile);
 279    }
 280    if (debug_level > 5) {
 281       print_memory_pool_stats();
 282    }
 283    free_config_resources();
 284    term_ua_server();
 285    term_msg();                        /* terminate message handler */
 286    stop_watchdog();
 287    cleanup_tls();
 288    close_memory_pool();               /* release free memory in pool */
 289    sm_dump(false);
 290    exit(sig);
 291 }
 292
 293 struct RELOAD_TABLE {
 294    int job_count;
 295    RES **res_table;
 296 };
 297
 298 static const int max_reloads = 10;
 299 static RELOAD_TABLE reload_table[max_reloads];
 300
 301 static void init_reload(void)
 302 {
 303    for (int i=0; i < max_reloads; i++) {
 304       reload_table[i].job_count = 0;
 305       reload_table[i].res_table = NULL;
 306    }
 307 }
 308
 309 static void free_saved_resources(int table)
 310 {
 311    int num = r_last - r_first + 1;
 312    RES **res_tab = reload_table[table].res_table;
 313    if (!res_tab) {
 314       Dmsg1(100, "res_tab for table %d already released.\n", table);
 315       return;
 316    }
 317    Dmsg1(100, "Freeing resources for table %d\n", table);
 318    for (int j=0; j<num; j++) {
 319       free_resource(res_tab[j], r_first + j);
 320    }
 321    free(res_tab);
 322    reload_table[table].job_count = 0;
 323    reload_table[table].res_table = NULL;
 324 }
 325
 326 /*
 327  * Called here at the end of every job that was
 328  * hooked decrementing the active job_count. When
 329  * it goes to zero, no one is using the associated
 330  * resource table, so free it.
 331  */
 332 static void reload_job_end_cb(JCR *jcr, void *ctx)
 333 {
 334    int reload_id = (int)((long int)ctx);
 335    Dmsg3(100, "reload job_end JobId=%d table=%d cnt=%d\n", jcr->JobId,
 336       reload_id, reload_table[reload_id].job_count);
 337    lock_jcr_chain();
 338    LockRes();
 339    if (--reload_table[reload_id].job_count <= 0) {
 340       free_saved_resources(reload_id);
 341    }
 342    UnlockRes();
 343    unlock_jcr_chain();
 344 }
 345
 346 static int find_free_reload_table_entry()
 347 {
 348    int table = -1;
 349    for (int i=0; i < max_reloads; i++) {
 350       if (reload_table[i].res_table == NULL) {
 351          table = i;
 352          break;
 353       }
 354    }
 355    return table;
 356 }
 357
 358 /*
 359  * If we get here, we have received a SIGHUP, which means to
 360  *    reread our configuration file.
 361  *
 362  * The algorithm used is as follows: we count how many jobs are
 363  *   running and mark the running jobs to make a callback on
 364  *   exiting. The old config is saved with the reload table
 365  *   id in a reload table. The new config file is read. Now, as
 366  *   each job exits, it calls back to the reload_job_end_cb(), which
 367  *   decrements the count of open jobs for the given reload table.
 368  *   When the count goes to zero, we release those resources.
 369  *   This allows us to have pointers into the resource table (from
 370  *   jobs), and once they exit and all the pointers are released, we
 371  *   release the old table. Note, if no new jobs are running since the
 372  *   last reload, then the old resources will be immediately release.
 373  *   A console is considered a job because it may have pointers to
 374  *   resources, but a SYSTEM job is not since it *should* not have any
 375  *   permanent pointers to jobs.
 376  */
 377 extern "C"
 378 void reload_config(int sig)
 379 {
 380    static bool already_here = false;
 381    sigset_t set;
 382    JCR *jcr;
 383    int njobs = 0;                     /* number of running jobs */
 384    int table, rtable;
 385    bool ok;
 386
 387    if (already_here) {
 388       abort();                        /* Oops, recursion -> die */
 389    }
 390    already_here = true;
 391    sigemptyset(&set);
 392    sigaddset(&set, SIGHUP);
 393    sigprocmask(SIG_BLOCK, &set, NULL);
 394
 395    lock_jcr_chain();
 396    LockRes();
 397
 398    table = find_free_reload_table_entry();
 399    if (table < 0) {
 400       Jmsg(NULL, M_ERROR, 0, _("Too many open reload requests. Request ignored.\n"));
 401       goto bail_out;
 402    }
 403
 404    Dmsg1(100, "Reload_config njobs=%d\n", njobs);
 405    reload_table[table].res_table = save_config_resources();
 406    Dmsg1(100, "Saved old config in table %d\n", table);
 407
 408    ok = parse_config(configfile, 0);  /* no exit on error */
 409
 410    Dmsg0(100, "Reloaded config file\n");
 411    if (!ok || !check_resources()) {
 412       rtable = find_free_reload_table_entry();    /* save new, bad table */
 413       if (rtable < 0) {
 414          Jmsg(NULL, M_ERROR, 0, _("Please correct configuration file: %s\n"), configfile);
 415          Jmsg(NULL, M_ERROR_TERM, 0, _("Out of reload table entries. Giving up.\n"));
 416       } else {
 417          Jmsg(NULL, M_ERROR, 0, _("Please correct configuration file: %s\n"), configfile);
 418          Jmsg(NULL, M_ERROR, 0, _("Resetting previous configuration.\n"));
 419       }
 420       reload_table[rtable].res_table = save_config_resources();
 421       /* Now restore old resoure values */
 422       int num = r_last - r_first + 1;
 423       RES **res_tab = reload_table[table].res_table;
 424       for (int i=0; i<num; i++) {
 425          res_head[i] = res_tab[i];
 426       }
 427       table = rtable;                 /* release new, bad, saved table below */
 428    } else {
 429       /*
 430        * Hook all active jobs so that they release this table
 431        */
 432       foreach_jcr(jcr) {
 433          if (jcr->JobType != JT_SYSTEM) {
 434             reload_table[table].job_count++;
 435             job_end_push(jcr, reload_job_end_cb, (void *)((long int)table));
 436             njobs++;
 437          }
 438          free_jcr(jcr);
 439       }
 440    }
 441
 442    /* Reset globals */
 443    set_working_directory(director->working_directory);
 444    FDConnectTimeout = director->FDConnectTimeout;
 445    SDConnectTimeout = director->SDConnectTimeout;
 446    Dmsg0(0, "Director's configuration file reread.\n");
 447
 448    /* Now release saved resources, if no jobs using the resources */
 449    if (njobs == 0) {
 450       free_saved_resources(table);
 451    }
 452
 453 bail_out:
 454    UnlockRes();
 455    unlock_jcr_chain();
 456    sigprocmask(SIG_UNBLOCK, &set, NULL);
 457    signal(SIGHUP, reload_config);
 458    already_here = false;
 459 }
 460
 461 /*
 462  * Make a quick check to see that we have all the
 463  * resources needed.
 464  *
 465  *  **** FIXME **** this routine could be a lot more
 466  *   intelligent and comprehensive.
 467  */
 468 static int check_resources()
 469 {
 470    bool OK = true;
 471    JOB *job;
 472
 473    LockRes();
 474
 475    job = (JOB *)GetNextRes(R_JOB, NULL);
 476    director = (DIRRES *)GetNextRes(R_DIRECTOR, NULL);
 477    if (!director) {
 478       Jmsg(NULL, M_FATAL, 0, _("No Director resource defined in %s\n"
 479 "Without that I don't know who I am :-(\n"), configfile);
 480       OK = false;
 481    } else {
 482       set_working_directory(director->working_directory);
 483       if (!director->messages) {       /* If message resource not specified */
 484          director->messages = (MSGS *)GetNextRes(R_MSGS, NULL);
 485          if (!director->messages) {
 486             Jmsg(NULL, M_FATAL, 0, _("No Messages resource defined in %s\n"), configfile);
 487             OK = false;
 488          }
 489       }
 490       if (GetNextRes(R_DIRECTOR, (RES *)director) != NULL) {
 491          Jmsg(NULL, M_FATAL, 0, _("Only one Director resource permitted in %s\n"),
 492             configfile);
 493          OK = false;
 494       }
 495 #ifdef HAVE_TLS
 496       /* tls_require implies tls_enable */
 497       if (director->tls_require) {
 498          director->tls_enable = true;
 499       }
 500
 501       if (!director->tls_certfile && director->tls_enable) {
 502          Jmsg(NULL, M_FATAL, 0, _("\"TLS Certificate\" file not defined for Director \"%s\" in %s.\n"),
 503             director->hdr.name, configfile);
 504          OK = false;
 505       }
 506
 507       if (!director->tls_keyfile && director->tls_enable) {
 508          Jmsg(NULL, M_FATAL, 0, _("\"TLS Key\" file not defined for Director \"%s\" in %s.\n"),
 509             director->hdr.name, configfile);
 510          OK = false;
 511       }
 512
 513       if ((!director->tls_ca_certfile && !director->tls_ca_certdir) && director->tls_enable && director->tls_verify_peer) {
 514          Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\" or \"TLS CA"
 515               " Certificate Dir\" are defined for Director \"%s\" in %s."
 516               " At least one CA certificate store is required"
 517               " when using \"TLS Verify Peer\".\n"),
 518               director->hdr.name, configfile);
 519          OK = false;
 520       }
 521
 522       /* If everything is well, attempt to initialize our per-resource TLS context */
 523       if (OK && (director->tls_enable || director->tls_require)) {
 524          /* Initialize TLS context:
 525           * Args: CA certfile, CA certdir, Certfile, Keyfile,
 526           * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
 527          director->tls_ctx = new_tls_context(director->tls_ca_certfile,
 528             director->tls_ca_certdir, director->tls_certfile,
 529             director->tls_keyfile, NULL, NULL, director->tls_dhfile,
 530             director->tls_verify_peer);
 531
 532          if (!director->tls_ctx) {
 533             Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for Director \"%s\" in %s.\n"),
 534                  director->hdr.name, configfile);
 535             OK = false;
 536          }
 537       }
 538 #endif /* HAVE_TLS */
 539    }
 540
 541    if (!job) {
 542       Jmsg(NULL, M_FATAL, 0, _("No Job records defined in %s\n"), configfile);
 543       OK = false;
 544    }
 545    foreach_res(job, R_JOB) {
 546       int i;
 547
 548       if (job->jobdefs) {
 549          /* Handle Storage alists specifically */
 550          JOB *jobdefs = job->jobdefs;
 551          if (jobdefs->storage && !job->storage) {
 552             STORE *st;
 553             job->storage = New(alist(10, not_owned_by_alist));
 554             foreach_alist(st, jobdefs->storage) {
 555                job->storage->append(st);
 556             }
 557          }
 558
 559          /* Transfer default items from JobDefs Resource */
 560          for (i=0; job_items[i].name; i++) {
 561             char **def_svalue, **svalue;  /* string value */
 562             int *def_ivalue, *ivalue;     /* integer value */
 563             int64_t *def_lvalue, *lvalue; /* 64 bit values */
 564             uint32_t offset;
 565
 566             Dmsg4(1400, "Job \"%s\", field \"%s\" bit=%d def=%d\n",
 567                 job->hdr.name, job_items[i].name,
 568                 bit_is_set(i, job->hdr.item_present),
 569                 bit_is_set(i, job->jobdefs->hdr.item_present));
 570
 571             if (!bit_is_set(i, job->hdr.item_present) &&
 572                  bit_is_set(i, job->jobdefs->hdr.item_present)) {
 573                Dmsg2(400, "Job \"%s\", field \"%s\": getting default.\n",
 574                  job->hdr.name, job_items[i].name);
 575                offset = (char *)(job_items[i].value) - (char *)&res_all;
 576                /*
 577                 * Handle strings and directory strings
 578                 */
 579                if (job_items[i].handler == store_str ||
 580                    job_items[i].handler == store_dir) {
 581                   def_svalue = (char **)((char *)(job->jobdefs) + offset);
 582                   Dmsg5(400, "Job \"%s\", field \"%s\" def_svalue=%s item %d offset=%u\n",
 583                        job->hdr.name, job_items[i].name, *def_svalue, i, offset);
 584                   svalue = (char **)((char *)job + offset);
 585                   if (*svalue) {
 586                      Pmsg1(000, "Hey something is wrong. p=0x%lu\n", *svalue);
 587                   }
 588                   *svalue = bstrdup(*def_svalue);
 589                   set_bit(i, job->hdr.item_present);
 590                /*
 591                 * Handle resources
 592                 */
 593                } else if (job_items[i].handler == store_res) {
 594                   def_svalue = (char **)((char *)(job->jobdefs) + offset);
 595                   Dmsg4(400, "Job \"%s\", field \"%s\" item %d offset=%u\n",
 596                        job->hdr.name, job_items[i].name, i, offset);
 597                   svalue = (char **)((char *)job + offset);
 598                   if (*svalue) {
 599                      Pmsg1(000, "Hey something is wrong. p=0x%lu\n", *svalue);
 600                   }
 601                   *svalue = *def_svalue;
 602                   set_bit(i, job->hdr.item_present);
 603                /*
 604                 * Handle alist resources
 605                 */
 606                } else if (job_items[i].handler == store_alist_res) {
 607                   if (bit_is_set(i, job->jobdefs->hdr.item_present)) {
 608                      set_bit(i, job->hdr.item_present);
 609                   }
 610                /*
 611                 * Handle integer fields
 612                 *    Note, our store_yesno does not handle bitmaped fields
 613                 */
 614                } else if (job_items[i].handler == store_yesno   ||
 615                           job_items[i].handler == store_pint    ||
 616                           job_items[i].handler == store_jobtype ||
 617                           job_items[i].handler == store_level   ||
 618                           job_items[i].handler == store_pint    ||
 619                           job_items[i].handler == store_replace) {
 620                   def_ivalue = (int *)((char *)(job->jobdefs) + offset);
 621                   Dmsg5(400, "Job \"%s\", field \"%s\" def_ivalue=%d item %d offset=%u\n",
 622                        job->hdr.name, job_items[i].name, *def_ivalue, i, offset);
 623                   ivalue = (int *)((char *)job + offset);
 624                   *ivalue = *def_ivalue;
 625                   set_bit(i, job->hdr.item_present);
 626                /*
 627                 * Handle 64 bit integer fields
 628                 */
 629                } else if (job_items[i].handler == store_time   ||
 630                           job_items[i].handler == store_size   ||
 631                           job_items[i].handler == store_int64) {
 632                   def_lvalue = (int64_t *)((char *)(job->jobdefs) + offset);
 633                   Dmsg5(400, "Job \"%s\", field \"%s\" def_lvalue=%" lld " item %d offset=%u\n",
 634                        job->hdr.name, job_items[i].name, *def_lvalue, i, offset);
 635                   lvalue = (int64_t *)((char *)job + offset);
 636                   *lvalue = *def_lvalue;
 637                   set_bit(i, job->hdr.item_present);
 638                }
 639             }
 640          }
 641       }
 642       /*
 643        * Ensure that all required items are present
 644        */
 645       for (i=0; job_items[i].name; i++) {
 646          if (job_items[i].flags & ITEM_REQUIRED) {
 647                if (!bit_is_set(i, job->hdr.item_present)) {
 648                   Jmsg(NULL, M_FATAL, 0, "\"%s\" directive in Job \"%s\" resource is required, but not found.\n",
 649                     job_items[i].name, job->hdr.name);
 650                   OK = false;
 651                 }
 652          }
 653          /* If this triggers, take a look at lib/parse_conf.h */
 654          if (i >= MAX_RES_ITEMS) {
 655             Emsg0(M_ERROR_TERM, 0, "Too many items in Job resource\n");
 656          }
 657       }
 658    } /* End loop over Job res */
 659
 660    /* Loop over databases */
 661    CAT *catalog;
 662    foreach_res(catalog, R_CATALOG) {
 663       B_DB *db;
 664       /*
 665        * Make sure we can open catalog, otherwise print a warning
 666        * message because the server is probably not running.
 667        */
 668       db = db_init_database(NULL, catalog->db_name, catalog->db_user,
 669                          catalog->db_password, catalog->db_address,
 670                          catalog->db_port, catalog->db_socket,
 671                          catalog->mult_db_connections);
 672       if (!db || !db_open_database(NULL, db)) {
 673          Jmsg(NULL, M_FATAL, 0, _("Could not open database \"%s\".\n"),
 674               catalog->db_name);
 675          if (db) {
 676             Jmsg(NULL, M_FATAL, 0, _("%s"), db_strerror(db));
 677          }
 678          OK = false;
 679          continue;
 680       }
 681
 682       /* Loop over all pools, defining/updating them in each database */
 683       POOL *pool;
 684       foreach_res(pool, R_POOL) {
 685          create_pool(NULL, db, pool, POOL_OP_UPDATE);  /* update request */
 686       }
 687
 688       STORE *store;
 689       foreach_res(store, R_STORAGE) {
 690          STORAGE_DBR sr;
 691          MEDIATYPE_DBR mr;
 692          if (store->media_type) {
 693             bstrncpy(mr.MediaType, store->media_type, sizeof(mr.MediaType));
 694             mr.ReadOnly = 0;
 695             db_create_mediatype_record(NULL, db, &mr);
 696          } else {
 697             mr.MediaTypeId = 0;
 698          }
 699          bstrncpy(sr.Name, store->name(), sizeof(sr.Name));
 700          sr.AutoChanger = store->autochanger;
 701          db_create_storage_record(NULL, db, &sr);
 702          store->StorageId = sr.StorageId;   /* set storage Id */
 703          if (!sr.created) {                 /* if not created, update it */
 704             db_update_storage_record(NULL, db, &sr);
 705          }
 706
 707 #ifdef HAVE_TLS
 708          /* tls_require implies tls_enable */
 709          if (store->tls_require) {
 710             store->tls_enable = true;
 711          }
 712
 713          if ((!store->tls_ca_certfile && !store->tls_ca_certdir) && store->tls_enable) {
 714             Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\""
 715                  " or \"TLS CA Certificate Dir\" are defined for Storage \"%s\" in %s.\n"),
 716                  store->hdr.name, configfile);
 717             OK = false;
 718          }
 719
 720          /* If everything is well, attempt to initialize our per-resource TLS context */
 721          if (OK && (store->tls_enable || store->tls_require)) {
 722            /* Initialize TLS context:
 723             * Args: CA certfile, CA certdir, Certfile, Keyfile,
 724             * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
 725             store->tls_ctx = new_tls_context(store->tls_ca_certfile,
 726                store->tls_ca_certdir, store->tls_certfile,
 727                store->tls_keyfile, NULL, NULL, NULL, true);
 728
 729             if (!store->tls_ctx) {
 730                Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for Storage \"%s\" in %s.\n"),
 731                     store->hdr.name, configfile);
 732                OK = false;
 733             }
 734          }
 735 #endif /* HAVE_TLS */
 736       }
 737
 738       /* Loop over all counters, defining them in each database */
 739       /* Set default value in all counters */
 740       COUNTER *counter;
 741       foreach_res(counter, R_COUNTER) {
 742          /* Write to catalog? */
 743          if (!counter->created && counter->Catalog == catalog) {
 744             COUNTER_DBR cr;
 745             bstrncpy(cr.Counter, counter->hdr.name, sizeof(cr.Counter));
 746             cr.MinValue = counter->MinValue;
 747             cr.MaxValue = counter->MaxValue;
 748             cr.CurrentValue = counter->MinValue;
 749             if (counter->WrapCounter) {
 750                bstrncpy(cr.WrapCounter, counter->WrapCounter->hdr.name, sizeof(cr.WrapCounter));
 751             } else {
 752                cr.WrapCounter[0] = 0;  /* empty string */
 753             }
 754             if (db_create_counter_record(NULL, db, &cr)) {
 755                counter->CurrentValue = cr.CurrentValue;
 756                counter->created = true;
 757                Dmsg2(100, "Create counter %s val=%d\n", counter->hdr.name, counter->CurrentValue);
 758             }
 759          }
 760          if (!counter->created) {
 761             counter->CurrentValue = counter->MinValue;  /* default value */
 762          }
 763       }
 764       db_close_database(NULL, db);
 765    }
 766
 767 #ifdef HAVE_TLS
 768    /* Loop over Consoles */
 769    CONRES *cons;
 770    foreach_res(cons, R_CONSOLE) {
 771       /* tls_require implies tls_enable */
 772       if (cons->tls_require) {
 773          cons->tls_enable = true;
 774       }
 775
 776       if (!cons->tls_certfile && cons->tls_enable) {
 777          Jmsg(NULL, M_FATAL, 0, _("\"TLS Certificate\" file not defined for Console \"%s\" in %s.\n"),
 778             cons->hdr.name, configfile);
 779          OK = false;
 780       }
 781
 782       if (!cons->tls_keyfile && cons->tls_enable) {
 783          Jmsg(NULL, M_FATAL, 0, _("\"TLS Key\" file not defined for Console \"%s\" in %s.\n"),
 784             cons->hdr.name, configfile);
 785          OK = false;
 786       }
 787
 788       if ((!cons->tls_ca_certfile && !cons->tls_ca_certdir) && cons->tls_enable && cons->tls_verify_peer) {
 789          Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\" or \"TLS CA"
 790             " Certificate Dir\" are defined for Console \"%s\" in %s."
 791             " At least one CA certificate store is required"
 792             " when using \"TLS Verify Peer\".\n"),
 793             cons->hdr.name, configfile);
 794          OK = false;
 795       }
 796       /* If everything is well, attempt to initialize our per-resource TLS context */
 797       if (OK && (cons->tls_enable || cons->tls_require)) {
 798          /* Initialize TLS context:
 799           * Args: CA certfile, CA certdir, Certfile, Keyfile,
 800           * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
 801          cons->tls_ctx = new_tls_context(cons->tls_ca_certfile,
 802             cons->tls_ca_certdir, cons->tls_certfile,
 803             cons->tls_keyfile, NULL, NULL, cons->tls_dhfile, cons->tls_verify_peer);
 804
 805          if (!cons->tls_ctx) {
 806             Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for File daemon \"%s\" in %s.\n"),
 807                cons->hdr.name, configfile);
 808             OK = false;
 809          }
 810       }
 811
 812    }
 813 #endif /* HAVE_TLS */
 814
 815 #ifdef HAVE_TLS
 816    /* Loop over Clients */
 817    CLIENT *client;
 818    foreach_res(client, R_CLIENT) {
 819       /* tls_require implies tls_enable */
 820       if (client->tls_require) {
 821          client->tls_enable = true;
 822       }
 823
 824       if ((!client->tls_ca_certfile && !client->tls_ca_certdir) && client->tls_enable) {
 825          Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\""
 826             " or \"TLS CA Certificate Dir\" are defined for File daemon \"%s\" in %s.\n"),
 827             client->hdr.name, configfile);
 828          OK = false;
 829       }
 830
 831       /* If everything is well, attempt to initialize our per-resource TLS context */
 832       if (OK && (client->tls_enable || client->tls_require)) {
 833          /* Initialize TLS context:
 834           * Args: CA certfile, CA certdir, Certfile, Keyfile,
 835           * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
 836          client->tls_ctx = new_tls_context(client->tls_ca_certfile,
 837             client->tls_ca_certdir, client->tls_certfile,
 838             client->tls_keyfile, NULL, NULL, NULL,
 839             true);
 840
 841          if (!client->tls_ctx) {
 842             Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for File daemon \"%s\" in %s.\n"),
 843                client->hdr.name, configfile);
 844             OK = false;
 845          }
 846       }
 847    }
 848 #endif /* HAVE_TLS */
 849
 850    UnlockRes();
 851    if (OK) {
 852       close_msg(NULL);                /* close temp message handler */
 853       init_msg(NULL, director->messages); /* open daemon message handler */
 854    }
 855    return OK;
 856 }