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

   1 /*
   2  * Read code for Storage daemon
   3  *
   4  *     Kern Sibbald, November MM
   5  *
   6  *   Version $Id$
   7  */
   8 /*
   9    Copyright (C) 2000-2003 Kern Sibbald and John Walker
  10
  11    This program is free software; you can redistribute it and/or
  12    modify it under the terms of the GNU General Public License as
  13    published by the Free Software Foundation; either version 2 of
  14    the License, or (at your option) any later version.
  15
  16    This program is distributed in the hope that it will be useful,
  17    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19    General Public License for more details.
  20
  21    You should have received a copy of the GNU General Public
  22    License along with this program; if not, write to the Free
  23    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  24    MA 02111-1307, USA.
  25
  26  */
  27
  28 #include "bacula.h"
  29 #include "stored.h"
  30
  31 /* Forward referenced subroutines */
  32 static void get_session_record(DEVICE *dev, DEV_RECORD *rec, SESSION_LABEL *sessrec);
  33
  34 /* Variables used by Child process */
  35 /* Global statistics */
  36 /* Note, these probably should be in shared memory so that
  37  * they are truly global for all processes
  38  */
  39 extern struct s_shm *shm;             /* shared memory structure */
  40 extern int  FiledDataChan;            /* File daemon data channel (port) */
  41
  42
  43 /* Responses sent to the File daemon */
  44 static char OK_data[]    = "3000 OK data\n";
  45 static char rec_header[] = "rechdr %ld %ld %ld %ld %ld";
  46
  47
  48 /*
  49  *  Read Data and send to File Daemon
  50  *   Returns: 0 on failure
  51  *            1 on success
  52  */
  53 int do_read_data(JCR *jcr)
  54 {
  55    BSOCK *ds;
  56    BSOCK *fd_sock = jcr->file_bsock;
  57    int ok = TRUE;
  58    int done = FALSE;
  59    DEVICE *dev;
  60    DEV_RECORD *rec;
  61    DEV_BLOCK *block;
  62    POOLMEM *hdr;
  63    SESSION_LABEL sessrec;              /* session record */
  64
  65    Dmsg0(20, "Start read data.\n");
  66
  67    dev = jcr->device->dev;
  68    memset(&sessrec, 0, sizeof(sessrec));
  69
  70    /* Tell File daemon we will send data */
  71    bnet_fsend(fd_sock, OK_data);
  72    Dmsg1(10, "bstored>filed: %s\n", fd_sock->msg);
  73
  74    ds = fd_sock;
  75
  76    if (!bnet_set_buffer_size(ds, MAX_NETWORK_BUFFER_SIZE, BNET_SETBUF_READ)) {
  77       return 0;
  78    }
  79
  80
  81    Dmsg1(20, "Begin read device=%s\n", dev_name(dev));
  82
  83    block = new_block(dev);
  84
  85    create_vol_list(jcr);
  86    if (jcr->NumVolumes == 0) {
  87       Jmsg(jcr, M_FATAL, 0, _("No Volume names found for restore.\n"));
  88       free_block(block);
  89       free_vol_list(jcr);
  90       return 0;
  91    }
  92
  93    Dmsg2(200, "Found %d volumes names to restore. First=%s\n", jcr->NumVolumes,
  94       jcr->VolList->VolumeName);
  95
  96    pm_strcpy(&jcr->VolumeName, jcr->VolList->VolumeName);
  97
  98    /*
  99     * Ready device for reading, and read records
 100     */
 101    if (!acquire_device_for_read(jcr, dev, block)) {
 102       free_block(block);
 103       free_vol_list(jcr);
 104       return 0;
 105    }
 106
 107    rec = new_record();
 108    free_pool_memory(rec->data);
 109    rec->data = ds->msg;                /* use socket message buffer */
 110    hdr = get_pool_memory(PM_MESSAGE);
 111
 112    /*
 113     *   Read records, apply BSR filtering, and return any that are
 114     *    matched.
 115     */
 116    for ( ; ok && !done; ) {
 117       if (job_canceled(jcr)) {
 118          ok = FALSE;
 119          break;
 120       }
 121       /* Read Record */
 122       Dmsg1(500, "Main read_record. rem=%d\n", rec->remainder);
 123
 124       if (block_is_empty(block) && !read_block_from_device(jcr, dev, block, CHECK_BLOCK_NUMBERS)) {
 125          Dmsg1(500, "Main read record failed. rem=%d\n", rec->remainder);
 126          if (dev->state & ST_EOT) {
 127             DEV_RECORD *record;
 128             if (!mount_next_read_volume(jcr, dev, block)) {
 129                break;
 130             }
 131             /* Read and discard Volume Label */
 132             record = new_record();
 133             read_block_from_device(jcr, dev, block, NO_BLOCK_NUMBER_CHECK);
 134             read_record_from_block(block, record);
 135             get_session_record(dev, record, &sessrec);
 136             free_record(record);
 137          }
 138          if (dev->state & ST_EOF) {
 139             Dmsg0(90, "Got End of File. Trying again ...\n");
 140             continue;                 /* End of File */
 141          }
 142          if (dev->state & ST_SHORT) {
 143             continue;
 144          }
 145          /* I/O error -- or non-standard end of tape */
 146          ok = FALSE;
 147          break;
 148       }
 149
 150       for (rec->state=0; !is_block_empty(rec); ) {
 151
 152          if (!read_record_from_block(block, rec)) {
 153             break;
 154          }
 155          /*
 156           * At this point, we have at least a record header.
 157           *  Now decide if we want this record or not, but remember
 158           *  before accessing the record, we may need to read again to
 159           *  get all the data.
 160           */
 161
 162          if (rec->FileIndex == EOM_LABEL) { /* end of tape? */
 163             Dmsg0(40, "Get EOM LABEL\n");
 164             rec->remainder = 0;
 165             break;                         /* yes, get out */
 166          }
 167
 168          /* Some sort of label? */
 169          if (rec->FileIndex < 0) {
 170             get_session_record(dev, rec, &sessrec);
 171             continue;
 172          } /* end if label record */
 173
 174
 175          /* Match BSR against current record */
 176          if (jcr->bsr) {
 177             int stat = match_bsr(jcr->bsr, rec, &dev->VolHdr, &sessrec);
 178             if (stat == -1) {         /* no more possible matches */
 179                done = TRUE;
 180                break;
 181             } else if (stat == 0) {   /* no match */
 182                Dmsg0(50, "BSR rejected record\n");
 183                rec->remainder = 0;
 184                continue;
 185             }
 186          } else {
 187             /* Old way, deprecated */
 188             if (rec->VolSessionId != jcr->read_VolSessionId ||
 189                 rec->VolSessionTime != jcr->read_VolSessionTime) {
 190                Dmsg0(50, "Ignore record ids not equal\n");
 191                rec->remainder = 0;
 192                continue;                    /* ignore */
 193             }
 194          }
 195
 196          if (is_partial_record(rec)) {
 197             break;                    /* Go read full record */
 198          }
 199
 200          /* Generate Header parameters and send to File daemon
 201           * Note, we build header in hdr buffer to avoid wiping
 202           * out the data record
 203           */
 204          ds->msg = hdr;
 205          Dmsg5(400, "Send to FD: SessId=%u SessTim=%u FI=%d Strm=%d, len=%d\n",
 206             rec->VolSessionId, rec->VolSessionTime, rec->FileIndex, rec->Stream,
 207             rec->data_len);
 208          if (!bnet_fsend(ds, rec_header, rec->VolSessionId, rec->VolSessionTime,
 209                 rec->FileIndex, rec->Stream, rec->data_len)) {
 210             Dmsg1(30, ">filed: Error Hdr=%s\n", ds->msg);
 211             hdr = ds->msg;
 212             ds->msg = rec->data;
 213             ok = FALSE;
 214             break;
 215          } else {
 216             Dmsg1(30, ">filed: Hdr=%s\n", ds->msg);
 217          }
 218
 219          hdr = ds->msg;               /* restore hdr buffer */
 220          ds->msg = rec->data;         /* restore data record address */
 221
 222          /* Send data record to File daemon */
 223          ds->msglen = rec->data_len;
 224          Dmsg1(40, ">filed: send %d bytes data.\n", ds->msglen);
 225          if (!bnet_send(ds)) {
 226             Pmsg1(000, "Error sending to FD. ERR=%s\n", bnet_strerror(ds));
 227             Dmsg1(100, "Hdr=%s\n", hdr);
 228             Dmsg1(100, "data=%s\n", ds->msg);
 229             Jmsg1(jcr, M_FATAL, 0, _("Error sending to File daemon. ERR=%s\n"),
 230                bnet_strerror(ds));
 231             ok = FALSE;
 232             break;
 233          }
 234       }
 235    }
 236    /* Send end of data to FD */
 237    bnet_sig(ds, BNET_EOD);
 238
 239    if (!release_device(jcr, dev)) {
 240       ok = FALSE;
 241    }
 242    free_pool_memory(hdr);
 243    free_block(block);
 244    rec->data = NULL;                  /* previously released */
 245    free_record(rec);
 246    free_vol_list(jcr);
 247    Dmsg0(30, "Done reading.\n");
 248    return ok ? 1 : 0;
 249 }
 250
 251 static void get_session_record(DEVICE *dev, DEV_RECORD *rec, SESSION_LABEL *sessrec)
 252 {
 253    char *rtype;
 254    memset(sessrec, 0, sizeof(sessrec));
 255    switch (rec->FileIndex) {
 256       case PRE_LABEL:
 257          rtype = "Fresh Volume Label";
 258          break;
 259       case VOL_LABEL:
 260          rtype = "Volume Label";
 261          unser_volume_label(dev, rec);
 262          break;
 263       case SOS_LABEL:
 264          rtype = "Begin Session";
 265          unser_session_label(sessrec, rec);
 266          break;
 267       case EOS_LABEL:
 268          rtype = "End Session";
 269          break;
 270       case EOM_LABEL:
 271          rtype = "End of Media";
 272          break;
 273       default:
 274          rtype = "Unknown";
 275          break;
 276    }
 277    Dmsg5(10, "%s Record: VolSessionId=%d VolSessionTime=%d JobId=%d DataLen=%d\n",
 278          rtype, rec->VolSessionId, rec->VolSessionTime, rec->Stream, rec->data_len);
 279 }