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

   1 /*
   2  *
   3  *   Bacula Director -- msgchan.c -- handles the message channel
   4  *    to the Storage daemon and the File daemon.
   5  *
   6  *     Kern Sibbald, August MM
   7  *
   8  *    This routine runs as a thread and must be thread reentrant.
   9  *
  10  *  Basic tasks done here:
  11  *    Open a message channel with the Storage daemon
  12  *      to authenticate ourself and to pass the JobId.
  13  *    Create a thread to interact with the Storage daemon
  14  *      who returns a job status and requests Catalog services, etc.
  15  *
  16  *   Version $Id$
  17  */
  18 /*
  19    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
  20
  21    This program is free software; you can redistribute it and/or
  22    modify it under the terms of the GNU General Public License as
  23    published by the Free Software Foundation; either version 2 of
  24    the License, or (at your option) any later version.
  25
  26    This program is distributed in the hope that it will be useful,
  27    but WITHOUT ANY WARRANTY; without even the implied warranty of
  28    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  29    General Public License for more details.
  30
  31    You should have received a copy of the GNU General Public
  32    License along with this program; if not, write to the Free
  33    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  34    MA 02111-1307, USA.
  35
  36  */
  37
  38 #include "bacula.h"
  39 #include "dird.h"
  40
  41 /* Commands sent to Storage daemon */
  42 static char jobcmd[]     = "JobId=%d job=%s job_name=%s client_name=%s \
  43 type=%d level=%d FileSet=%s NoAttr=%d SpoolAttr=%d FileSetMD5=%s\n";
  44 static char use_device[] = "use device=%s media_type=%s pool_name=%s pool_type=%s\n";
  45
  46 /* Response from Storage daemon */
  47 static char OKjob[]      = "3000 OK Job SDid=%d SDtime=%d Authorization=%100s\n";
  48 static char OK_device[]  = "3000 OK use device\n";
  49
  50 /* Storage Daemon requests */
  51 static char Job_start[]  = "3010 Job %127s start\n";
  52 static char Job_end[]    =
  53    "3099 Job %127s end JobStatus=%d JobFiles=%d JobBytes=%" lld "\n";
  54 static char Job_status[] = "3012 Job %127s jobstatus %d\n";
  55
  56 /* Forward referenced functions */
  57 static void *msg_thread(void *arg);
  58
  59 /*
  60  * Establish a message channel connection with the Storage daemon
  61  * and perform authentication.
  62  */
  63 int connect_to_storage_daemon(JCR *jcr, int retry_interval,
  64                               int max_retry_time, int verbose)
  65 {
  66    BSOCK *sd;
  67
  68    /*
  69     *  Open message channel with the Storage daemon
  70     */
  71    Dmsg2(200, "bnet_connect to Storage daemon %s:%d\n", jcr->store->address,
  72       jcr->store->SDport);
  73    sd = bnet_connect(jcr, retry_interval, max_retry_time,
  74           _("Storage daemon"), jcr->store->address,
  75           NULL, jcr->store->SDport, verbose);
  76    if (sd == NULL) {
  77       return 0;
  78    }
  79    sd->res = (RES *)jcr->store;        /* save pointer to other end */
  80    jcr->store_bsock = sd;
  81
  82    if (!authenticate_storage_daemon(jcr)) {
  83       return 0;
  84    }
  85    return 1;
  86 }
  87
  88 /*
  89  * Start a job with the Storage daemon
  90  */
  91 int start_storage_daemon_job(JCR *jcr)
  92 {
  93    int status;
  94    STORE *storage;
  95    BSOCK *sd;
  96    char auth_key[100];
  97    char *device_name, *pool_name, *pool_type, *media_type;
  98    int device_name_len, pool_name_len, pool_type_len, media_type_len;
  99
 100    storage = jcr->store;
 101    sd = jcr->store_bsock;
 102    /*
 103     * Now send JobId and permissions, and get back the authorization key.
 104     */
 105    bash_spaces(jcr->job->hdr.name);
 106    bash_spaces(jcr->client->hdr.name);
 107    bash_spaces(jcr->fileset->hdr.name);
 108    bnet_fsend(sd, jobcmd, jcr->JobId, jcr->Job, jcr->job->hdr.name,
 109               jcr->client->hdr.name, jcr->JobType, jcr->JobLevel,
 110               jcr->fileset->hdr.name, !jcr->pool->catalog_files,
 111               jcr->job->SpoolAttributes, jcr->fileset->MD5);
 112    unbash_spaces(jcr->job->hdr.name);
 113    unbash_spaces(jcr->client->hdr.name);
 114    unbash_spaces(jcr->fileset->hdr.name);
 115    if (bnet_recv(sd) > 0) {
 116        Dmsg1(110, "<stored: %s", sd->msg);
 117        if (sscanf(sd->msg, OKjob, &jcr->VolSessionId,
 118                   &jcr->VolSessionTime, &auth_key) != 3) {
 119           Jmsg(jcr, M_FATAL, 0, _("Storage daemon rejected Job command: %s\n"), sd->msg);
 120           return 0;
 121        } else {
 122           jcr->sd_auth_key = bstrdup(auth_key);
 123           Dmsg1(150, "sd_auth_key=%s\n", jcr->sd_auth_key);
 124        }
 125    } else {
 126       Jmsg(jcr, M_FATAL, 0, _("<stored: bad response to Job command: %s\n"),
 127          bnet_strerror(sd));
 128       return 0;
 129    }
 130
 131    /*
 132     * Send use device = xxx media = yyy pool = zzz
 133     */
 134    device_name_len = strlen(storage->dev_name) + 1;
 135    media_type_len = strlen(storage->media_type) + 1;
 136    pool_type_len = strlen(jcr->pool->pool_type) + 1;
 137    pool_name_len = strlen(jcr->pool->hdr.name) + 1;
 138    device_name = (char *) get_memory(device_name_len);
 139    pool_name = (char *) get_memory(pool_name_len);
 140    pool_type = (char *) get_memory(pool_type_len);
 141    media_type = (char *) get_memory(media_type_len);
 142    memcpy(device_name, storage->dev_name, device_name_len);
 143    memcpy(media_type, storage->media_type, media_type_len);
 144    memcpy(pool_type, jcr->pool->pool_type, pool_type_len);
 145    memcpy(pool_name, jcr->pool->hdr.name, pool_name_len);
 146    bash_spaces(device_name);
 147    bash_spaces(media_type);
 148    bash_spaces(pool_type);
 149    bash_spaces(pool_name);
 150    sd->msg = check_pool_memory_size(sd->msg, sizeof(device_name) +
 151       device_name_len + media_type_len + pool_type_len + pool_name_len);
 152    bnet_fsend(sd, use_device, device_name, media_type, pool_name, pool_type);
 153    Dmsg1(110, ">stored: %s", sd->msg);
 154    status = response(sd, OK_device, "Use Device");
 155
 156    free_memory(device_name);
 157    free_memory(media_type);
 158    free_memory(pool_name);
 159    free_memory(pool_type);
 160
 161    return status;
 162 }
 163
 164 /*
 165  * Start a thread to handle Storage daemon messages and
 166  *  Catalog requests.
 167  */
 168 int start_storage_daemon_message_thread(JCR *jcr)
 169 {
 170    int status;
 171    pthread_t thid;
 172
 173    P(jcr->mutex);
 174    jcr->use_count++;                  /* mark in use by msg thread */
 175    V(jcr->mutex);
 176    if ((status=pthread_create(&thid, NULL, msg_thread, (void *)jcr)) != 0) {
 177       Jmsg1(jcr, M_ABORT, 0, _("Cannot create message thread: %s\n"), strerror(status));
 178    }
 179    jcr->SD_msg_chan = thid;
 180    return 1;
 181 }
 182
 183 static void msg_thread_cleanup(void *arg)
 184 {
 185    JCR *jcr = (JCR *)arg;
 186    Dmsg0(200, "End msg_thread\n");
 187    db_end_transaction(jcr->db);       /* terminate any open transaction */
 188    P(jcr->mutex);
 189    jcr->msg_thread_done = TRUE;
 190    pthread_cond_broadcast(&jcr->term_wait); /* wakeup any waiting threads */
 191    V(jcr->mutex);
 192    free_jcr(jcr);                     /* release jcr */
 193 }
 194
 195 /*
 196  * Handle the message channel (i.e. requests from the
 197  *  Storage daemon).
 198  * Note, we are running in a separate thread.
 199  */
 200 static void *msg_thread(void *arg)
 201 {
 202    JCR *jcr = (JCR *)arg;
 203    BSOCK *sd;
 204    int JobStatus;
 205    char Job[MAX_NAME_LENGTH];
 206    uint32_t JobFiles;
 207    uint64_t JobBytes;
 208    int stat;
 209
 210    pthread_cleanup_push(msg_thread_cleanup, arg);
 211    Dmsg0(200, "msg_thread\n");
 212    sd = jcr->store_bsock;
 213    pthread_detach(pthread_self());
 214
 215    /* Read the Storage daemon's output.
 216     */
 217    Dmsg0(200, "Start msg_thread loop\n");
 218    while ((stat=bget_msg(sd, 0)) > 0) {
 219       Dmsg1(200, "<stored: %s", sd->msg);
 220       if (sscanf(sd->msg, Job_start, &Job) == 1) {
 221          continue;
 222       }
 223       if (sscanf(sd->msg, Job_end, &Job, &JobStatus, &JobFiles,
 224                  &JobBytes) == 4) {
 225          jcr->SDJobStatus = JobStatus; /* termination status */
 226          jcr->JobFiles = JobFiles;
 227          jcr->JobBytes = JobBytes;
 228          break;
 229       }
 230       if (sscanf(sd->msg, Job_status, &Job, &JobStatus) == 2) {
 231          jcr->SDJobStatus = JobStatus; /* current status */
 232          continue;
 233       }
 234    }
 235    if (stat < 0) {
 236       jcr->SDJobStatus = JS_ErrorTerminated;
 237    }
 238    pthread_cleanup_pop(1);
 239    return NULL;
 240 }
 241
 242 void wait_for_storage_daemon_termination(JCR *jcr)
 243 {
 244    /* Now wait for Storage daemon to terminate our message thread */
 245    P(jcr->mutex);
 246    jcr->JobStatus = JS_WaitSD;
 247    while (!jcr->msg_thread_done && !job_cancelled(jcr)) {
 248       struct timeval tv;
 249       struct timezone tz;
 250       struct timespec timeout;
 251
 252       gettimeofday(&tv, &tz);
 253       timeout.tv_nsec = 0;
 254       timeout.tv_sec = tv.tv_sec + 10; /* wait 10 seconds */
 255       Dmsg0(300, "I'm waiting for message thread termination.\n");
 256       pthread_cond_timedwait(&jcr->term_wait, &jcr->mutex, &timeout);
 257    }
 258    V(jcr->mutex);
 259    jcr->JobStatus = jcr->SDJobStatus;
 260 }