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 static void *device_allocation(void *arg);
  42
  43 #define CONFIG_FILE "bacula-sd.conf"  /* Default config file */
  44
  45
  46 /* Global variables exported */
  47
  48
  49 /* This is our own global resource */
  50 STORES *me;
  51
  52 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
  53 static uint32_t VolSessionId = 0;
  54 uint32_t VolSessionTime;
  55
  56 char *configfile;
  57 static int foreground = 0;
  58
  59 static workq_t dird_workq;            /* queue for processing connections */
  60
  61
  62 static void usage()
  63 {
  64    fprintf(stderr, _(
  65 "\nVersion: " VERSION " (" DATE ")\n\n"
  66 "Usage: stored [-s -f ] [-c config_file] [-d debug_level]  [config_file]\n"
  67 "        -c <file>   use <file> as configuration file\n"
  68 "        -dnn        set debug level to nn\n"
  69 "        -f          run in foreground (for debugging)\n"
  70 "        -s          no signals (for debugging)\n"
  71 "        -t          test - read config and exit\n"
  72 "        -?          print this message.\n"
  73 "\n"));
  74    exit(1);
  75 }
  76
  77 /*********************************************************************
  78  *
  79  *  Main Bacula Unix Storage Daemon
  80  *
  81  */
  82 int main (int argc, char *argv[])
  83 {
  84    int ch;
  85    int no_signals = FALSE;
  86    int test_config = FALSE;
  87    pthread_t thid;
  88
  89    init_stack_dump();
  90    my_name_is(argc, argv, "stored");
  91    init_msg(NULL, NULL);
  92    daemon_start_time = time(NULL);
  93    memset(&last_job, 0, sizeof(last_job));
  94
  95    /* Sanity checks */
  96    if (TAPE_BSIZE % DEV_BSIZE != 0 || TAPE_BSIZE / DEV_BSIZE == 0) {
  97       Emsg2(M_ABORT, 0, "Tape block size (%d) not multiple of system size (%d)\n",
  98          TAPE_BSIZE, DEV_BSIZE);
  99    }
 100    if (TAPE_BSIZE != (1 << (ffs(TAPE_BSIZE)-1))) {
 101       Emsg1(M_ABORT, 0, "Tape block size (%d) is not a power of 2\n", TAPE_BSIZE);
 102    }
 103
 104    while ((ch = getopt(argc, argv, "c:d:fst?")) != -1) {
 105       switch (ch) {
 106          case 'c':                    /* configuration file */
 107             if (configfile != NULL) {
 108                free(configfile);
 109             }
 110             configfile = bstrdup(optarg);
 111             break;
 112
 113          case 'd':                    /* debug level */
 114             debug_level = atoi(optarg);
 115             if (debug_level <= 0) {
 116                debug_level = 1;
 117             }
 118             break;
 119
 120          case 'f':                    /* run in foreground */
 121             foreground = TRUE;
 122             break;
 123
 124          case 's':                    /* no signals */
 125             no_signals = TRUE;
 126             break;
 127
 128          case 't':
 129             test_config = TRUE;
 130             break;
 131
 132          case '?':
 133          default:
 134             usage();
 135
 136       }
 137    }
 138    argc -= optind;
 139    argv += optind;
 140
 141    if (argc) {
 142       if (configfile != NULL) {
 143          free(configfile);
 144       }
 145       configfile = bstrdup(*argv);
 146       argc--;
 147       argv++;
 148    }
 149    if (argc)
 150       usage();
 151
 152    if (!no_signals) {
 153       init_signals(terminate_stored);
 154    }
 155
 156
 157    if (configfile == NULL) {
 158       configfile = bstrdup(CONFIG_FILE);
 159    }
 160
 161    parse_config(configfile);
 162    check_config();
 163
 164    if (test_config) {
 165       terminate_stored(0);
 166    }
 167
 168    if (!foreground) {
 169       daemon_start();                 /* become daemon */
 170       init_stack_dump();              /* pick up new pid */
 171    }
 172
 173    create_pid_file(me->pid_directory, "bacula-sd", me->SDport);
 174
 175    /* Ensure that Volume Session Time and Id are both
 176     * set and are both non-zero.
 177     */
 178    VolSessionTime = (long)daemon_start_time;
 179    if (VolSessionTime == 0) { /* paranoid */
 180       Emsg0(M_ABORT, 0, _("Volume Session Time is ZERO!\n"));
 181    }
 182
 183    /* Make sure on Solaris we can run concurrent, watch dog + servers + misc */
 184    set_thread_concurrency(me->max_concurrent_jobs * 2 + 4);
 185
 186    /*
 187     * Here we lock the resources then fire off the device allocation
 188     *  thread. That thread will release the resources when all the
 189     *  devices are allocated.  This allows use to start the server
 190     *  right away, but any jobs will wait until the resources are
 191     *  unlocked.
 192     */
 193    LockRes();
 194    if (pthread_create(&thid, NULL, device_allocation, NULL) != 0) {
 195       Emsg1(M_ABORT, 0, _("Unable to create thread. ERR=%s\n"), strerror(errno));
 196    }
 197
 198
 199
 200    start_watchdog();                  /* start watchdog thread */
 201
 202    /* Single server used for Director and File daemon */
 203    bnet_thread_server(me->SDaddr, me->SDport, me->max_concurrent_jobs * 2 + 1,
 204                       &dird_workq, connection_request);
 205    exit(1);                           /* to keep compiler quiet */
 206 }
 207
 208 /* Return a new Session Id */
 209 uint32_t newVolSessionId()
 210 {
 211    uint32_t Id;
 212
 213    P(mutex);
 214    VolSessionId++;
 215    Id = VolSessionId;
 216    V(mutex);
 217    return Id;
 218 }
 219
 220 /* Check Configuration file for necessary info */
 221 static void check_config()
 222 {
 223    struct stat stat_buf;
 224
 225    LockRes();
 226    me = (STORES *)GetNextRes(R_STORAGE, NULL);
 227    if (!me) {
 228       UnlockRes();
 229       Emsg1(M_ERROR_TERM, 0, _("No Storage resource defined in %s. Cannot continue.\n"),
 230          configfile);
 231    }
 232
 233    my_name_is(0, (char **)NULL, me->hdr.name);     /* Set our real name */
 234
 235    if (GetNextRes(R_STORAGE, (RES *)me) != NULL) {
 236       UnlockRes();
 237       Emsg1(M_ERROR_TERM, 0, _("Only one Storage resource permitted in %s\n"),
 238          configfile);
 239    }
 240    if (GetNextRes(R_DIRECTOR, NULL) == NULL) {
 241       UnlockRes();
 242       Emsg1(M_ERROR_TERM, 0, _("No Director resource defined in %s. Cannot continue.\n"),
 243          configfile);
 244    }
 245    if (GetNextRes(R_DEVICE, NULL) == NULL){
 246       UnlockRes();
 247       Emsg1(M_ERROR_TERM, 0, _("No Device resource defined in %s. Cannot continue.\n"),
 248            configfile);
 249    }
 250    if (!me->messages) {
 251       me->messages = (MSGS *)GetNextRes(R_MSGS, NULL);
 252       if (!me->messages) {
 253          Emsg1(M_ERROR_TERM, 0, _("No Messages resource defined in %s. Cannot continue.\n"),
 254             configfile);
 255       }
 256    }
 257    close_msg(NULL);                   /* close temp message handler */
 258    init_msg(NULL, me->messages);      /* open daemon message handler */
 259
 260    UnlockRes();
 261
 262    if (!me->working_directory) {
 263       Emsg1(M_ERROR_TERM, 0, _("No Working Directory defined in %s. Cannot continue.\n"),
 264          configfile);
 265    }
 266    if (stat(me->working_directory, &stat_buf) != 0) {
 267       Emsg1(M_ERROR_TERM, 0, _("Working Directory: %s not found. Cannot continue.\n"),
 268          me->working_directory);
 269    }
 270    if (!S_ISDIR(stat_buf.st_mode)) {
 271       Emsg1(M_ERROR_TERM, 0, _("Working Directory: %s is not a directory. Cannot continue.\n"),
 272          me->working_directory);
 273    }
 274    working_directory = me->working_directory;
 275 }
 276
 277 /*
 278  * We are started as a separate thread.  The
 279  *  resources are alread locked.
 280  */
 281 static void *device_allocation(void *arg)
 282 {
 283    int i;
 284    DEVRES *device;
 285
 286    pthread_detach(pthread_self());
 287
 288    /* LockRes() alread done */
 289    for (device=NULL,i=0;  (device=(DEVRES *)GetNextRes(R_DEVICE, (RES *)device)); i++) {
 290       Dmsg1(90, "calling init_dev %s\n", device->device_name);
 291       device->dev = init_dev(NULL, device);
 292       Dmsg1(10, "SD init done %s\n", device->device_name);
 293       if (!device->dev) {
 294          Emsg1(M_ERROR, 0, _("Could not initialize %s\n"), device->device_name);
 295       }
 296       if (device->cap_bits & CAP_ALWAYSOPEN) {
 297          Dmsg1(20, "calling open_device %s\n", device->device_name);
 298          if (!open_device(device->dev)) {
 299             Emsg1(M_ERROR, 0, _("Could not open device %s\n"), device->device_name);
 300          }
 301       }
 302       if (device->cap_bits & CAP_AUTOMOUNT && device->dev &&
 303           device->dev->state & ST_OPENED) {
 304          DEV_BLOCK *block;
 305          JCR *jcr;
 306          block = new_block(device->dev);
 307          jcr = new_jcr(sizeof(JCR), stored_free_jcr);
 308          switch (read_dev_volume_label(jcr, device->dev, block)) {
 309             case VOL_OK:
 310                break;
 311             default:
 312                Emsg1(M_WARNING, 0, _("Could not mount device %s\n"), device->device_name);
 313                break;
 314          }
 315          free_jcr(jcr);
 316          free_block(block);
 317       }
 318    }
 319    UnlockRes();
 320    return NULL;
 321 }
 322
 323
 324 /* Clean up and then exit */
 325 void terminate_stored(int sig)
 326 {
 327    static int in_here = FALSE;
 328    DEVRES *device;
 329
 330    if (in_here) {                     /* prevent loops */
 331       exit(1);
 332    }
 333    in_here = TRUE;
 334
 335    delete_pid_file(me->pid_directory, "bacula-sd", me->SDport);
 336    stop_watchdog();
 337
 338    Dmsg0(200, "In terminate_stored()\n");
 339
 340    LockRes();
 341    for (device=NULL; (device=(DEVRES *)GetNextRes(R_DEVICE, (RES *)device)); ) {
 342       if (device->dev) {
 343          term_dev(device->dev);
 344       }
 345    }
 346    UnlockRes();
 347
 348    if (configfile)
 349    free(configfile);
 350    free_config_resources();
 351
 352    if (debug_level > 10) {
 353       print_memory_pool_stats();
 354    }
 355    term_msg();
 356    close_memory_pool();
 357
 358    sm_dump(False);                    /* dump orphaned buffers */
 359    exit(1);
 360 }