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

   1 /*
   2  * Second generation Storage daemon.
   3  *
   4  * It accepts a number of simple commands from the File daemon
   5  * and acts on them. When a request to append data is made,
   6  * it opens a data channel and accepts data from the
   7  * File daemon.
   8  *
   9  *   Version $Id$
  10  *
  11  */
  12 /*
  13    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
  14
  15    This program is free software; you can redistribute it and/or
  16    modify it under the terms of the GNU General Public License as
  17    published by the Free Software Foundation; either version 2 of
  18    the License, or (at your option) any later version.
  19
  20    This program is distributed in the hope that it will be useful,
  21    but WITHOUT ANY WARRANTY; without even the implied warranty of
  22    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  23    General Public License for more details.
  24
  25    You should have received a copy of the GNU General Public
  26    License along with this program; if not, write to the Free
  27    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  28    MA 02111-1307, USA.
  29
  30  */
  31
  32 #include "bacula.h"
  33 #include "stored.h"
  34
  35 /* Imported functions */
  36
  37
  38 /* Forward referenced functions */
  39 void terminate_stored(int sig);
  40 static void check_config();
  41
  42 #define CONFIG_FILE "bacula-sd.conf"  /* Default config file */
  43
  44
  45 /* Global variables exported */
  46
  47
  48 struct s_shm *shm;                    /* memory shared with children */
  49 BSHM bshm;                            /* shared memory control packet */
  50
  51
  52 /* This is our own global resource */
  53 STORES *me;
  54
  55 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
  56 static uint32_t VolSessionId = 0;
  57 uint32_t VolSessionTime;
  58
  59 char *configfile;
  60 static int foreground = 0;
  61
  62 static workq_t dird_workq;            /* queue for processing connections */
  63
  64
  65 static void usage()
  66 {
  67    fprintf(stderr, _(
  68 "\nVersion: " VERSION " (" DATE ")\n\n"
  69 "Usage: stored [-s -f ] [-c config_file] [-d debug_level]  [config_file]\n"
  70 "        -c <file>   use <file> as configuration file\n"
  71 "        -dnn        set debug level to nn\n"
  72 "        -f          run in foreground (for debugging)\n"
  73 "        -s          no signals (for debugging)\n"
  74 "        -t          test - read config and exit\n"
  75 "        -?          print this message.\n"
  76 "\n"));
  77    exit(1);
  78 }
  79
  80 /*********************************************************************
  81  *
  82  *  Main Bacula Unix Storage Daemon
  83  *
  84  */
  85 int main (int argc, char *argv[])
  86 {
  87    int ch, i;
  88    int no_signals = FALSE;
  89    int test_config = FALSE;
  90    DEVRES *device;
  91
  92    init_stack_dump();
  93    my_name_is(argc, argv, "stored");
  94    init_msg(NULL, NULL);
  95    daemon_start_time = time(NULL);
  96    memset(&last_job, 0, sizeof(last_job));
  97
  98    /* Sanity checks */
  99    if (TAPE_BSIZE % DEV_BSIZE != 0 || TAPE_BSIZE / DEV_BSIZE == 0) {
 100       Emsg2(M_ABORT, 0, "Tape block size (%d) not multiple of system size (%d)\n",
 101          TAPE_BSIZE, DEV_BSIZE);
 102    }
 103    if (TAPE_BSIZE != (1 << (ffs(TAPE_BSIZE)-1))) {
 104       Emsg1(M_ABORT, 0, "Tape block size (%d) is not a power of 2\n", TAPE_BSIZE);
 105    }
 106
 107    while ((ch = getopt(argc, argv, "c:d:fst?")) != -1) {
 108       switch (ch) {
 109          case 'c':                    /* configuration file */
 110             if (configfile != NULL) {
 111                free(configfile);
 112             }
 113             configfile = bstrdup(optarg);
 114             break;
 115
 116          case 'd':                    /* debug level */
 117             debug_level = atoi(optarg);
 118             if (debug_level <= 0) {
 119                debug_level = 1;
 120             }
 121             break;
 122
 123          case 'f':                    /* run in foreground */
 124             foreground = TRUE;
 125             break;
 126
 127          case 's':                    /* no signals */
 128             no_signals = TRUE;
 129             break;
 130
 131          case 't':
 132             test_config = TRUE;
 133             break;
 134
 135          case '?':
 136          default:
 137             usage();
 138
 139       }
 140    }
 141    argc -= optind;
 142    argv += optind;
 143
 144    if (argc) {
 145       if (configfile != NULL) {
 146          free(configfile);
 147       }
 148       configfile = bstrdup(*argv);
 149       argc--;
 150       argv++;
 151    }
 152    if (argc)
 153       usage();
 154
 155    if (!no_signals) {
 156       init_signals(terminate_stored);
 157    }
 158
 159
 160    if (configfile == NULL) {
 161       configfile = bstrdup(CONFIG_FILE);
 162    }
 163
 164    parse_config(configfile);
 165    check_config();
 166
 167    bshm.size = 0;
 168    if (test_config) {
 169       terminate_stored(0);
 170    }
 171
 172    if (!foreground) {
 173       daemon_start();                 /* become daemon */
 174       init_stack_dump();              /* pick up new pid */
 175    }
 176
 177    create_pid_file(me->pid_directory, "bacula-sd", me->SDport);
 178
 179    /*  ****FIXME**** clean this up */
 180    /* Create and attach to shared memory. This is a
 181     * hold over from the days of child processes.
 182     * Note, in reality all memory is shared. This
 183     * is just a global buffer for the device packets.
 184     */
 185    shm = (s_shm *) malloc(sizeof(struct s_shm));
 186    /* Zero shared memory */
 187    memset(shm, 0, sizeof(struct s_shm));
 188
 189    /* Ensure that Volume Session Time and Id are both
 190     * set and are both non-zero.
 191     */
 192    VolSessionTime = (long)daemon_start_time;
 193    if (VolSessionTime == 0) { /* paranoid */
 194       Emsg0(M_ABORT, 0, _("Volume Session Time is ZERO!\n"));
 195    }
 196
 197    LockRes();
 198    for (device=NULL,i=0;  (device=(DEVRES *)GetNextRes(R_DEVICE, (RES *)device)); i++) {
 199       if (i >= MAX_DEVICES) {
 200          UnlockRes();
 201          Emsg1(M_ABORT, 0, _("Too many Device Resources. Max=%d\n"), MAX_DEVICES);
 202       }
 203       Dmsg1(90, "calling init_dev %s\n", device->device_name);
 204       device->dev = init_dev(&shm->dev[i], device);
 205       Dmsg1(10, "SD init done %s\n", device->device_name);
 206       if (!device->dev) {
 207          Emsg1(M_ERROR, 0, _("Could not initialize %s\n"), device->device_name);
 208       }
 209       if (device->cap_bits & CAP_ALWAYSOPEN) {
 210          Dmsg1(20, "calling open_device %s\n", device->device_name);
 211          if (!open_device(device->dev)) {
 212             Emsg1(M_ERROR, 0, _("Could not open device %s\n"), device->device_name);
 213          }
 214       }
 215       if (device->cap_bits & CAP_AUTOMOUNT && device->dev &&
 216           device->dev->state & ST_OPENED) {
 217          DEV_BLOCK *block;
 218          JCR *jcr;
 219          block = new_block(device->dev);
 220          jcr = new_jcr(sizeof(JCR), stored_free_jcr);
 221          switch (read_dev_volume_label(jcr, device->dev, block)) {
 222             case VOL_OK:
 223                break;
 224             default:
 225                Emsg1(M_WARNING, 0, _("Could not mount device %s\n"), device->device_name);
 226                break;
 227          }
 228          free_jcr(jcr);
 229          free_block(block);
 230       }
 231    }
 232    UnlockRes();
 233    device = NULL;
 234
 235    set_thread_concurrency(me->max_concurrent_jobs * 2 +
 236       4 /* watch dog + servers + misc */);
 237
 238    start_watchdog();                  /* start watchdog thread */
 239
 240    /* Single server used for Director and File daemon */
 241    bnet_thread_server(me->SDaddr, me->SDport, me->max_concurrent_jobs * 2 + 1,
 242                       &dird_workq, connection_request);
 243    exit(1);                           /* to keep compiler quiet */
 244 }
 245
 246 /* Return a new Session Id */
 247 uint32_t newVolSessionId()
 248 {
 249    uint32_t Id;
 250
 251    P(mutex);
 252    VolSessionId++;
 253    Id = VolSessionId;
 254    V(mutex);
 255    return Id;
 256 }
 257
 258 /* Check Configuration file for necessary info */
 259 static void check_config()
 260 {
 261    struct stat stat_buf;
 262
 263    LockRes();
 264    me = (STORES *)GetNextRes(R_STORAGE, NULL);
 265    if (!me) {
 266       UnlockRes();
 267       Emsg1(M_ABORT, 0, _("No Storage resource defined in %s. Cannot continue.\n"),
 268          configfile);
 269    }
 270
 271    my_name_is(0, (char **)NULL, me->hdr.name);     /* Set our real name */
 272
 273    if (GetNextRes(R_STORAGE, (RES *)me) != NULL) {
 274       UnlockRes();
 275       Emsg1(M_ABORT, 0, _("Only one Storage resource permitted in %s\n"),
 276          configfile);
 277    }
 278    if (GetNextRes(R_DIRECTOR, NULL) == NULL) {
 279       UnlockRes();
 280       Emsg1(M_ABORT, 0, _("No Director resource defined in %s. Cannot continue.\n"),
 281          configfile);
 282    }
 283    if (GetNextRes(R_DEVICE, NULL) == NULL){
 284       UnlockRes();
 285       Emsg1(M_ABORT, 0, _("No Device resource defined in %s. Cannot continue.\n"),
 286            configfile);
 287    }
 288    if (!me->messages) {
 289       me->messages = (MSGS *)GetNextRes(R_MSGS, NULL);
 290       if (!me->messages) {
 291          Emsg1(M_ABORT, 0, _("No Messages resource defined in %s. Cannot continue.\n"),
 292             configfile);
 293       }
 294    }
 295    close_msg(NULL);                   /* close temp message handler */
 296    init_msg(NULL, me->messages);      /* open daemon message handler */
 297
 298    UnlockRes();
 299
 300    if (!me->working_directory) {
 301       Emsg1(M_ABORT, 0, _("No Working Directory defined in %s. Cannot continue.\n"),
 302          configfile);
 303    }
 304    if (stat(me->working_directory, &stat_buf) != 0) {
 305       Emsg1(M_ABORT, 0, _("Working Directory: %s not found. Cannot continue.\n"),
 306          me->working_directory);
 307    }
 308    if (!S_ISDIR(stat_buf.st_mode)) {
 309       Emsg1(M_ABORT, 0, _("Working Directory: %s is not a directory. Cannot continue.\n"),
 310          me->working_directory);
 311    }
 312    working_directory = me->working_directory;
 313 }
 314
 315 /* Clean up and then exit */
 316 void terminate_stored(int sig)
 317 {
 318    static int in_here = FALSE;
 319    DEVRES *device;
 320
 321    if (in_here) {                     /* prevent loops */
 322       exit(1);
 323    }
 324    in_here = TRUE;
 325
 326    delete_pid_file(me->pid_directory, "bacula-sd", me->SDport);
 327    stop_watchdog();
 328
 329    Dmsg0(200, "In terminate_stored()\n");
 330
 331    LockRes();
 332    for (device=NULL; (device=(DEVRES *)GetNextRes(R_DEVICE, (RES *)device)); ) {
 333       if (device->dev) {
 334          term_dev(device->dev);
 335       }
 336    }
 337    UnlockRes();
 338
 339    if (configfile)
 340    free(configfile);
 341    free_config_resources();
 342
 343    if (debug_level > 10) {
 344       print_memory_pool_stats();
 345    }
 346    term_msg();
 347    close_memory_pool();
 348
 349    if (shm) {
 350       free(shm);
 351    }
 352
 353    sm_dump(False);                    /* dump orphaned buffers */
 354    exit(1);
 355 }