]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/stored/stored.c
Make a few tls ifdef tweaks
[bacula/bacula] / bacula / src / stored / stored.c
index 21d4fff4bb9c53f805e2e5914afc07f485f3f184..49ff47b2df87a3eaa2c27a10170aaf61a87e1f92 100644 (file)
@@ -4,13 +4,13 @@
  * It accepts a number of simple commands from the File daemon
  * and acts on them. When a request to append data is made,
  * it opens a data channel and accepts data from the
- * File daemon. 
+ * File daemon.
  *
  *   Version $Id$
- * 
+ *
  */
 /*
-   Copyright (C) 2000-2004 Kern Sibbald and John Walker
+   Copyright (C) 2000-2005 Kern Sibbald
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
 
 /* Forward referenced functions */
 void terminate_stored(int sig);
-static void check_config();
+static int check_resources();
 
-extern "C" void *device_allocation(void *arg);
+extern "C" void *device_initialization(void *arg);
 
 #define CONFIG_FILE "bacula-sd.conf"  /* Default config file */
 
-
 /* Global variables exported */
 char OK_msg[]   = "3000 OK\n";
 char TERM_msg[] = "3999 Terminate\n";
-STORES *me = NULL;                   /* our Global resource */
-bool forge_on = false;               /* proceed inspite of I/O errors */
+STORES *me = NULL;                    /* our Global resource */
+bool forge_on = false;                /* proceed inspite of I/O errors */
+pthread_mutex_t device_release_mutex = PTHREAD_MUTEX_INITIALIZER;
+pthread_cond_t wait_device_release = PTHREAD_COND_INITIALIZER;
+
 
 static uint32_t VolSessionId = 0;
 uint32_t VolSessionTime;
@@ -57,13 +59,13 @@ char *configfile;
 /* Global static variables */
 static int foreground = 0;
 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
-static workq_t dird_workq;           /* queue for processing connections */
+static workq_t dird_workq;            /* queue for processing connections */
 
 
 static void usage()
 {
    fprintf(stderr, _(
-"Copyright (C) 2000-2004 Kern Sibbald and John Walker.\n"
+"Copyright (C) 2000-2005 Kern Sibbald.\n"
 "\nVersion: " VERSION " (" BDATE ")\n\n"
 "Usage: stored [options] [-c config_file] [config_file]\n"
 "        -c <file>   use <file> as configuration file\n"
@@ -80,14 +82,14 @@ static void usage()
    exit(1);
 }
 
-/********************************************************************* 
+/*********************************************************************
  *
  *  Main Bacula Unix Storage Daemon
  *
  */
 int main (int argc, char *argv[])
 {
-   int ch;   
+   int ch;
    int no_signals = FALSE;
    int test_config = FALSE;
    pthread_t thid;
@@ -101,9 +103,9 @@ int main (int argc, char *argv[])
    daemon_start_time = time(NULL);
 
    /* Sanity checks */
-   if (TAPE_BSIZE % DEV_BSIZE != 0 || TAPE_BSIZE / DEV_BSIZE == 0) {
+   if (TAPE_BSIZE % B_DEV_BSIZE != 0 || TAPE_BSIZE / B_DEV_BSIZE == 0) {
       Emsg2(M_ABORT, 0, "Tape block size (%d) not multiple of system size (%d)\n",
-        TAPE_BSIZE, DEV_BSIZE);
+         TAPE_BSIZE, B_DEV_BSIZE);
    }
    if (TAPE_BSIZE != (1 << (ffs(TAPE_BSIZE)-1))) {
       Emsg1(M_ABORT, 0, "Tape block size (%d) is not a power of 2\n", TAPE_BSIZE);
@@ -112,62 +114,62 @@ int main (int argc, char *argv[])
    while ((ch = getopt(argc, argv, "c:d:fg:pstu:v?")) != -1) {
       switch (ch) {
       case 'c':                    /* configuration file */
-        if (configfile != NULL) {
-           free(configfile);
-        }
-        configfile = bstrdup(optarg);
-        break;
+         if (configfile != NULL) {
+            free(configfile);
+         }
+         configfile = bstrdup(optarg);
+         break;
 
       case 'd':                    /* debug level */
-        debug_level = atoi(optarg);
-        if (debug_level <= 0) {
-           debug_level = 1; 
-        }
-        break;
+         debug_level = atoi(optarg);
+         if (debug_level <= 0) {
+            debug_level = 1;
+         }
+         break;
 
       case 'f':                    /* run in foreground */
-        foreground = TRUE;
-        break;
+         foreground = TRUE;
+         break;
 
       case 'g':                    /* set group id */
-        gid = optarg;
-        break;
+         gid = optarg;
+         break;
 
       case 'p':                    /* proceed in spite of I/O errors */
-        forge_on = true;
-        break;
+         forge_on = true;
+         break;
 
       case 's':                    /* no signals */
-        no_signals = TRUE;
-        break;
+         no_signals = TRUE;
+         break;
 
       case 't':
-        test_config = TRUE;
-        break;
+         test_config = TRUE;
+         break;
 
       case 'u':                    /* set uid */
-        uid = optarg;
-        break;
+         uid = optarg;
+         break;
 
       case 'v':                    /* verbose */
-        verbose++;
-        break;
+         verbose++;
+         break;
 
       case '?':
       default:
-        usage();
-        break;
-      }  
+         usage();
+         break;
+      }
    }
    argc -= optind;
    argv += optind;
 
    if (argc) {
       if (configfile != NULL) {
-        free(configfile);
+         free(configfile);
       }
       configfile = bstrdup(*argv);
-      argc--; 
+      argc--;
       argv++;
    }
    if (argc)
@@ -177,25 +179,33 @@ int main (int argc, char *argv[])
       init_signals(terminate_stored);
    }
 
-
    if (configfile == NULL) {
       configfile = bstrdup(CONFIG_FILE);
    }
 
    parse_config(configfile);
-   check_config();
+
+   if (init_tls() != 0) {
+      Jmsg((JCR *)NULL, M_ERROR_TERM, 0, _("TLS library initialization failed.\n"));
+   }
+
+   if (!check_resources()) {
+      Jmsg((JCR *)NULL, M_ERROR_TERM, 0, _("Please correct configuration file: %s\n"), configfile);
+   }
 
    if (test_config) {
       terminate_stored(0);
    }
 
+   my_name_is(0, (char **)NULL, me->hdr.name);     /* Set our real name */
+
    if (!foreground) {
-      daemon_start();                /* become daemon */
-      init_stack_dump();             /* pick up new pid */
+      daemon_start();                 /* become daemon */
+      init_stack_dump();              /* pick up new pid */
    }
 
-   create_pid_file(me->pid_directory, "bacula-sd", me->SDport);
-   read_state_file(me->working_directory, "bacula-sd", me->SDport);
+   create_pid_file(me->pid_directory, "bacula-sd", get_first_port_host_order(me->sdaddrs));
+   read_state_file(me->working_directory, "bacula-sd", get_first_port_host_order(me->sdaddrs));
 
    drop(uid, gid);
 
@@ -204,33 +214,35 @@ int main (int argc, char *argv[])
     */
    VolSessionTime = (long)daemon_start_time;
    if (VolSessionTime == 0) { /* paranoid */
-      Emsg0(M_ABORT, 0, _("Volume Session Time is ZERO!\n"));
+      Jmsg0(NULL, M_ABORT, 0, _("Volume Session Time is ZERO!\n"));
    }
 
+   init_python_interpreter(me->hdr.name, me->scripts_directory, "SDStartUp");
+
    /* Make sure on Solaris we can run concurrent, watch dog + servers + misc */
    set_thread_concurrency(me->max_concurrent_jobs * 2 + 4);
 
     /*
      * Start the device allocation thread
      */
-   if (pthread_create(&thid, NULL, device_allocation, NULL) != 0) {
+   if (pthread_create(&thid, NULL, device_initialization, NULL) != 0) {
       Emsg1(M_ABORT, 0, _("Unable to create thread. ERR=%s\n"), strerror(errno));
    }
 
-   start_watchdog();                 /* start watchdog thread */
+   start_watchdog();                  /* start watchdog thread */
 
-   init_jcr_subsystem();             /* start JCR watchdogs etc. */
+   init_jcr_subsystem();              /* start JCR watchdogs etc. */
 
-   /* 
+   /*
     * Sleep a bit to give device thread a chance to lock the resource
     * chain before we start the server.
     */
    bmicrosleep(1, 0);
-                                
+
    /* Single server used for Director and File daemon */
-   bnet_thread_server(me->SDaddr, me->SDport, me->max_concurrent_jobs * 2 + 1,
-                     &dird_workq, handle_connection_request);
-   exit(1);                          /* to keep compiler quiet */
+   bnet_thread_server(me->sdaddrs, me->max_concurrent_jobs * 2 + 1,
+                      &dird_workq, handle_connection_request);
+   exit(1);                           /* to keep compiler quiet */
 }
 
 /* Return a new Session Id */
@@ -246,59 +258,191 @@ uint32_t newVolSessionId()
 }
 
 /* Check Configuration file for necessary info */
-static void check_config()
+static int check_resources()
 {
+   bool OK = true;
+   AUTOCHANGER *changer;
+
    LockRes();
+
    me = (STORES *)GetNextRes(R_STORAGE, NULL);
    if (!me) {
-      UnlockRes();
-      Emsg1(M_ERROR_TERM, 0, _("No Storage resource defined in %s. Cannot continue.\n"),
-        configfile);
+      Jmsg1(NULL, M_ERROR, 0, _("No Storage resource defined in %s. Cannot continue.\n"),
+         configfile);
+      OK = false;
    }
 
-   my_name_is(0, (char **)NULL, me->hdr.name);    /* Set our real name */
-
    if (GetNextRes(R_STORAGE, (RES *)me) != NULL) {
-      UnlockRes();
-      Emsg1(M_ERROR_TERM, 0, _("Only one Storage resource permitted in %s\n"), 
-        configfile);
+      Jmsg1(NULL, M_ERROR, 0, _("Only one Storage resource permitted in %s\n"),
+         configfile);
+      OK = false;
    }
    if (GetNextRes(R_DIRECTOR, NULL) == NULL) {
-      UnlockRes();
-      Emsg1(M_ERROR_TERM, 0, _("No Director resource defined in %s. Cannot continue.\n"),
-        configfile);
+      Jmsg1(NULL, M_ERROR, 0, _("No Director resource defined in %s. Cannot continue.\n"),
+         configfile);
+      OK = false;
    }
    if (GetNextRes(R_DEVICE, NULL) == NULL){
-      UnlockRes();
-      Emsg1(M_ERROR_TERM, 0, _("No Device resource defined in %s. Cannot continue.\n"),
-          configfile);
+      Jmsg1(NULL, M_ERROR, 0, _("No Device resource defined in %s. Cannot continue.\n"),
+           configfile);
+      OK = false;
    }
+
    if (!me->messages) {
       me->messages = (MSGS *)GetNextRes(R_MSGS, NULL);
       if (!me->messages) {
-         Emsg1(M_ERROR_TERM, 0, _("No Messages resource defined in %s. Cannot continue.\n"),
-           configfile);
+         Jmsg1(NULL, M_ERROR, 0, _("No Messages resource defined in %s. Cannot continue.\n"),
+            configfile);
+         OK = false;
       }
    }
-   close_msg(NULL);                  /* close temp message handler */
-   init_msg(NULL, me->messages);      /* open daemon message handler */
-
-   UnlockRes();
 
    if (!me->working_directory) {
-      Emsg1(M_ERROR_TERM, 0, _("No Working Directory defined in %s. Cannot continue.\n"),
-        configfile);
+      Jmsg1(NULL, M_ERROR, 0, _("No Working Directory defined in %s. Cannot continue.\n"),
+         configfile);
+      OK = false;
+   }
+
+#ifdef HAVE_TLS
+   DIRRES *director;
+   STORES *store;
+   foreach_res(store, R_STORAGE) { 
+      /* tls_require implies tls_enable */
+      if (store->tls_require) {
+         store->tls_enable = true;
+      }
+
+      if (!store->tls_certfile && store->tls_enable) {
+         Jmsg(NULL, M_FATAL, 0, _("\"TLS Certificate\" file not defined for Storage \"%s\" in %s.\n"),
+              store->hdr.name, configfile);
+         OK = false;
+      }
+
+      if (!store->tls_keyfile && store->tls_enable) {
+         Jmsg(NULL, M_FATAL, 0, _("\"TLS Key\" file not defined for Storage \"%s\" in %s.\n"),
+              store->hdr.name, configfile);
+         OK = false;
+      }
+
+      if ((!store->tls_ca_certfile && !store->tls_ca_certdir) && store->tls_enable && store->tls_verify_peer) {
+         Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\""
+              " or \"TLS CA Certificate Dir\" are defined for Storage \"%s\" in %s."
+              " At least one CA certificate store is required"
+              " when using \"TLS Verify Peer\".\n"),
+              store->hdr.name, configfile);
+         OK = false;
+      }
+
+      /* If everything is well, attempt to initialize our per-resource TLS context */
+      if (OK && (store->tls_enable || store->tls_require)) {
+         /* Initialize TLS context:
+          * Args: CA certfile, CA certdir, Certfile, Keyfile,
+          * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
+         store->tls_ctx = new_tls_context(store->tls_ca_certfile,
+            store->tls_ca_certdir, store->tls_certfile,
+            store->tls_keyfile, NULL, NULL, store->tls_dhfile,
+            store->tls_verify_peer);
+
+         if (!store->tls_ctx) { 
+            Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for Storage \"%s\" in %s.\n"),
+                 store->hdr.name, configfile);
+            OK = false;
+         }
+      }
+   }
+
+   foreach_res(director, R_DIRECTOR) { 
+      /* tls_require implies tls_enable */
+      if (director->tls_require) {
+         director->tls_enable = true;
+      }
+
+      if (!director->tls_certfile && director->tls_enable) {
+         Jmsg(NULL, M_FATAL, 0, _("\"TLS Certificate\" file not defined for Director \"%s\" in %s.\n"),
+              director->hdr.name, configfile);
+         OK = false;
+      }
+
+      if (!director->tls_keyfile && director->tls_enable) {
+         Jmsg(NULL, M_FATAL, 0, _("\"TLS Key\" file not defined for Director \"%s\" in %s.\n"),
+              director->hdr.name, configfile);
+         OK = false;
+      }
+
+      if ((!director->tls_ca_certfile && !director->tls_ca_certdir) && director->tls_enable && director->tls_verify_peer) {
+         Jmsg(NULL, M_FATAL, 0, _("Neither \"TLS CA Certificate\""
+              " or \"TLS CA Certificate Dir\" are defined for Director \"%s\" in %s."
+              " At least one CA certificate store is required"
+              " when using \"TLS Verify Peer\".\n"),
+              director->hdr.name, configfile);
+         OK = false;
+      }
+
+      /* If everything is well, attempt to initialize our per-resource TLS context */
+      if (OK && (director->tls_enable || director->tls_require)) {
+         /* Initialize TLS context:
+          * Args: CA certfile, CA certdir, Certfile, Keyfile,
+          * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
+         director->tls_ctx = new_tls_context(director->tls_ca_certfile,
+            director->tls_ca_certdir, director->tls_certfile,
+            director->tls_keyfile, NULL, NULL, director->tls_dhfile,
+            director->tls_verify_peer);
+
+         if (!director->tls_ctx) { 
+            Jmsg(NULL, M_FATAL, 0, _("Failed to initialize TLS context for Director \"%s\" in %s.\n"),
+                 director->hdr.name, configfile);
+            OK = false;
+         }
+      }
+   }
+#endif /* HAVE_TLS */
+
+   /* Ensure that the media_type for each device is the same */
+   foreach_res(changer, R_AUTOCHANGER) {
+      DEVRES *device;
+      char *media_type = NULL;
+      foreach_alist(device, changer->device) {
+         if (media_type == NULL) {
+            media_type = device->media_type;
+            continue;
+         }     
+         if (strcmp(media_type, device->media_type) != 0) {
+            Jmsg(NULL, M_ERROR, 0, 
+               _("Media Type not the same for all devices in changer %s. Cannot continue.\n"),
+               changer->hdr.name);
+            OK = false;
+            continue;
+         }
+         /*
+          * If the device does not have a changer name or changer command
+          * defined, used the one from the Autochanger resource 
+          */
+         if (!device->changer_name) {
+            device->changer_name = bstrdup(changer->changer_name);
+         }
+         if (!device->changer_command) {
+            device->changer_command = bstrdup(changer->changer_command);
+         }
+      }
    }
    
-   set_working_directory(me->working_directory);
+   UnlockRes();
+
+   if (OK) {
+      close_msg(NULL);                   /* close temp message handler */
+      init_msg(NULL, me->messages);      /* open daemon message handler */
+      set_working_directory(me->working_directory);
+   }
+
+   return OK;
 }
 
 /*
- * We are started as a separate thread.  The
- *  resources are alread locked.
+ * Here we attempt to init and open each device. This is done
+ *  once at startup in a separate thread.
  */
 extern "C"
-void *device_allocation(void *arg)
+void *device_initialization(void *arg)
 {
    DEVRES *device;
 
@@ -307,42 +451,42 @@ void *device_allocation(void *arg)
 
    foreach_res(device, R_DEVICE) {
       Dmsg1(90, "calling init_dev %s\n", device->device_name);
-      device->dev = init_dev(NULL, device);
+      device->dev = init_dev(NULL, NULL, device);
       Dmsg1(10, "SD init done %s\n", device->device_name);
       if (!device->dev) {
-         Emsg1(M_ERROR, 0, _("Could not initialize %s\n"), device->device_name);
-        continue;
+         Jmsg1(NULL, M_ERROR, 0, _("Could not initialize %s\n"), device->device_name);
+         continue;
       }
 
       if (device->cap_bits & CAP_ALWAYSOPEN) {
          Dmsg1(20, "calling first_open_device %s\n", device->device_name);
-        if (!first_open_device(device->dev)) {
-            Emsg1(M_ERROR, 0, _("Could not open device %s\n"), device->device_name);
-        }
+         if (!first_open_device(device->dev)) {
+            Jmsg1(NULL, M_ERROR, 0, _("Could not open device %s\n"), device->device_name);
+         }
       }
-      if (device->cap_bits & CAP_AUTOMOUNT && device->dev && 
-         device->dev->state & ST_OPENED) {
-        JCR *jcr;
-        DCR *dcr;
-        jcr = new_jcr(sizeof(JCR), stored_free_jcr);
-        jcr->JobType = JT_SYSTEM;
-        /* Initialize FD start condition variable */
-        int errstat = pthread_cond_init(&jcr->job_start_wait, NULL);
-        if (errstat != 0) {
+      if (device->cap_bits & CAP_AUTOMOUNT && device->dev &&
+          device->dev->is_open()) {
+         JCR *jcr;
+         DCR *dcr;
+         jcr = new_jcr(sizeof(JCR), stored_free_jcr);
+         jcr->JobType = JT_SYSTEM;
+         /* Initialize FD start condition variable */
+         int errstat = pthread_cond_init(&jcr->job_start_wait, NULL);
+         if (errstat != 0) {
             Jmsg1(jcr, M_ABORT, 0, _("Unable to init job cond variable: ERR=%s\n"), strerror(errstat));
-        }
-        jcr->device = device;
-        dcr = new_dcr(jcr, device->dev);
-        switch (read_dev_volume_label(dcr, dcr->block)) {
-           case VOL_OK:
-              break;
-           default:
-               Emsg1(M_WARNING, 0, _("Could not mount device %s\n"), device->device_name);
-              break;
-        }
-        free_jcr(jcr);
+         }
+         dcr = new_dcr(jcr, device->dev);
+         switch (read_dev_volume_label(dcr)) {
+         case VOL_OK:
+            memcpy(&dcr->dev->VolCatInfo, &dcr->VolCatInfo, sizeof(dcr->dev->VolCatInfo));
+            break;
+         default:
+            Jmsg1(NULL, M_WARNING, 0, _("Could not mount device %s\n"), device->device_name);
+            break;
+         }
+         free_jcr(jcr);
       }
-   } 
+   }
    UnlockRes();
    return NULL;
 }
@@ -351,16 +495,16 @@ void *device_allocation(void *arg)
 /* Clean up and then exit */
 void terminate_stored(int sig)
 {
-   static int in_here = FALSE;
+   static bool in_here = false;
    DEVRES *device;
    JCR *jcr;
 
-   if (in_here) {                    /* prevent loops */
+   if (in_here) {                     /* prevent loops */
       exit(1);
    }
-   in_here = TRUE;
+   in_here = true;
 
-   if (sig == SIGTERM) {             /* normal shutdown request? */
+   if (sig == SIGTERM) {              /* normal shutdown request? */
       /*
        * This is a normal shutdown request. We wiffle through
        *   all open jobs canceling them and trying to wake
@@ -369,38 +513,40 @@ void terminate_stored(int sig)
        */
       lock_jcr_chain();
       foreach_jcr(jcr) {
-        BSOCK *fd;
-        free_locked_jcr(jcr);
-        if (jcr->JobId == 0) {
-           continue;                 /* ignore console */
-        }
-        set_jcr_job_status(jcr, JS_Canceled);
-        fd = jcr->file_bsock;  
-        if (fd) {
-           fd->timed_out = TRUE;
-            Dmsg1(100, "killing JobId=%d\n", jcr->JobId);
-           pthread_kill(jcr->my_thread_id, TIMEOUT_SIGNAL);
-           if (jcr->device && jcr->device->dev && jcr->device->dev->dev_blocked) {
-              pthread_cond_signal(&jcr->device->dev->wait_next_vol);
-           }
-           bmicrosleep(0, 50000);
-         }
+         BSOCK *fd;
+         free_locked_jcr(jcr);
+         if (jcr->JobId == 0) {
+            continue;                 /* ignore console */
+         }
+         set_jcr_job_status(jcr, JS_Canceled);
+         fd = jcr->file_bsock;
+         if (fd) {
+            fd->timed_out = true;
+            Dmsg1(100, "term_stored killing JobId=%d\n", jcr->JobId);
+            pthread_kill(jcr->my_thread_id, TIMEOUT_SIGNAL);
+            /* ***FIXME*** wiffle through all dcrs */
+            if (jcr->dcr && jcr->dcr->dev && jcr->dcr->dev->dev_blocked) {
+               pthread_cond_broadcast(&jcr->dcr->dev->wait_next_vol);
+               pthread_cond_broadcast(&wait_device_release);
+            }
+            bmicrosleep(0, 50000);
+          }
       }
       unlock_jcr_chain();
-      bmicrosleep(0, 500000);        /* give them 1/2 sec to clean up */
+      bmicrosleep(0, 500000);         /* give them 1/2 sec to clean up */
    }
 
-   write_state_file(me->working_directory, "bacula-sd", me->SDport);
-   delete_pid_file(me->pid_directory, "bacula-sd", me->SDport);
+   write_state_file(me->working_directory, "bacula-sd", get_first_port_host_order(me->sdaddrs));
+   delete_pid_file(me->pid_directory, "bacula-sd", get_first_port_host_order(me->sdaddrs));
 
    Dmsg1(200, "In terminate_stored() sig=%d\n", sig);
 
    LockRes();
    foreach_res(device, R_DEVICE) {
       if (device->dev) {
-        term_dev(device->dev);
+         term_dev(device->dev);
       }
-   } 
+   }
    UnlockRes();
 
    if (configfile)
@@ -412,8 +558,9 @@ void terminate_stored(int sig)
    }
    term_msg();
    stop_watchdog();
+   cleanup_tls();
    close_memory_pool();
 
-   sm_dump(false);                   /* dump orphaned buffers */
+   sm_dump(false);                    /* dump orphaned buffers */
    exit(sig);
 }