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
  35 /* Responses sent to the File daemon */
  36 static char OK_data[]    = "3000 OK data\n";
  37 static char rec_header[] = "rechdr %ld %ld %ld %ld %ld";
  38
  39 /*
  40  *  Read Data and send to File Daemon
  41  *   Returns: 0 on failure
  42  *            1 on success
  43  */
  44 int do_read_data(JCR *jcr)
  45 {
  46    BSOCK *ds;
  47    BSOCK *fd_sock = jcr->file_bsock;
  48    int ok = TRUE;
  49    int done = FALSE;
  50    DEVICE *dev;
  51    DEV_RECORD *rec;
  52    DEV_BLOCK *block;
  53    POOLMEM *hdr;
  54    SESSION_LABEL sessrec;              /* session record */
  55    uint32_t BlockNumber = 0;
  56
  57    Dmsg0(20, "Start read data.\n");
  58
  59    dev = jcr->device->dev;
  60    memset(&sessrec, 0, sizeof(sessrec));
  61
  62    /* Tell File daemon we will send data */
  63    bnet_fsend(fd_sock, OK_data);
  64    Dmsg1(10, "bstored>filed: %s\n", fd_sock->msg);
  65
  66    ds = fd_sock;
  67
  68    if (!bnet_set_buffer_size(ds, MAX_NETWORK_BUFFER_SIZE, BNET_SETBUF_READ)) {
  69       return 0;
  70    }
  71
  72
  73    Dmsg1(20, "Begin read device=%s\n", dev_name(dev));
  74
  75    block = new_block(dev);
  76
  77    create_vol_list(jcr);
  78    if (jcr->NumVolumes == 0) {
  79       Jmsg(jcr, M_FATAL, 0, _("No Volume names found for restore.\n"));
  80       free_block(block);
  81       free_vol_list(jcr);
  82       return 0;
  83    }
  84
  85    Dmsg2(200, "Found %d volumes names to restore. First=%s\n", jcr->NumVolumes,
  86       jcr->VolList->VolumeName);
  87
  88    pm_strcpy(&jcr->VolumeName, jcr->VolList->VolumeName);
  89
  90    /*
  91     * Ready device for reading, and read records
  92     */
  93    if (!acquire_device_for_read(jcr, dev, block)) {
  94       free_block(block);
  95       free_vol_list(jcr);
  96       return 0;
  97    }
  98
  99    rec = new_record();
 100    free_pool_memory(rec->data);
 101    rec->data = ds->msg;                /* use socket message buffer */
 102    hdr = get_pool_memory(PM_MESSAGE);
 103
 104    /*
 105     *   Read records, apply BSR filtering, and return any that are
 106     *    matched.
 107     */
 108    for ( ; ok && !done; ) {
 109       if (job_canceled(jcr)) {
 110          ok = FALSE;
 111          break;
 112       }
 113       /* Read Record */
 114       Dmsg1(500, "Main read_record. rem=%d\n", rec->remainder);
 115
 116       if (block_is_empty(block) && !read_block_from_device(jcr, dev, block, CHECK_BLOCK_NUMBERS)) {
 117          Dmsg1(500, "Main read record failed. rem=%d\n", rec->remainder);
 118          if (dev->state & ST_EOT) {
 119             DEV_RECORD *record;
 120             if (!mount_next_read_volume(jcr, dev, block)) {
 121                break;
 122             }
 123             /* Read and discard Volume Label */
 124             record = new_record();
 125             read_block_from_device(jcr, dev, block, NO_BLOCK_NUMBER_CHECK);
 126             read_record_from_block(block, record);
 127             get_session_record(dev, record, &sessrec);
 128             free_record(record);
 129             /* Now, continue to read the records on this volume */
 130
 131          } else if (dev->state & ST_EOF) {
 132             Dmsg0(90, "Got End of File. Trying again ...\n");
 133             continue;                 /* End of File */
 134          } else if (dev->state & ST_SHORT) {
 135             continue;
 136          } else {
 137             /* I/O error -- or non-standard end of tape */
 138             Dmsg0(000, "I/O error, bailing out.\n");
 139             ok = FALSE;
 140             break;
 141          }
 142       }
 143
 144       if (!block_is_empty(block) && !match_bsr_block(jcr->bsr, block)) {
 145          Dmsg5(150, "reject Blk=%u blen=%u bVer=%d SessId=%u SessTim=%u\n",
 146             block->BlockNumber, block->block_len, block->BlockVer,
 147             block->VolSessionId, block->VolSessionTime);
 148          empty_block(block);          /* force read next block */
 149          continue;
 150       }
 151
 152       for (rec->state=0; !is_block_empty(rec); ) {
 153
 154          if (!read_record_from_block(block, rec)) {
 155             break;
 156          }
 157          /*
 158           * At this point, we have at least a record header.
 159           *  Now decide if we want this record or not, but remember
 160           *  before accessing the record, we may need to read again to
 161           *  get all the data.
 162           */
 163
 164          if (rec->FileIndex == EOM_LABEL) { /* end of tape? */
 165             Dmsg0(40, "Get EOM LABEL\n");
 166             rec->remainder = 0;
 167             break;                         /* yes, get out */
 168          }
 169
 170          /* Some sort of label? */
 171          if (rec->FileIndex < 0) {
 172             get_session_record(dev, rec, &sessrec);
 173             continue;
 174          } /* end if label record */
 175
 176
 177          /* Match BSR against current record */
 178          if (jcr->bsr) {
 179             int stat = match_bsr(jcr->bsr, rec, &dev->VolHdr, &sessrec);
 180             if (stat == -1) {         /* no more possible matches */
 181                done = TRUE;
 182                break;
 183             } else if (stat == 0) {   /* no match */
 184                Dmsg0(50, "BSR rejected record\n");
 185                rec->remainder = 0;
 186                continue;
 187             }
 188          } else {
 189             /* Old way, deprecated */
 190             if (rec->VolSessionId != jcr->read_VolSessionId ||
 191                 rec->VolSessionTime != jcr->read_VolSessionTime) {
 192                Dmsg0(50, "Ignore record ids not equal\n");
 193                rec->remainder = 0;
 194                continue;                    /* ignore */
 195             }
 196          }
 197
 198          if (block->block_read) {
 199             if (BlockNumber != block->BlockNumber) {
 200                Jmsg(jcr, M_ERROR, 0, _("Invalid block number. Expected %u, got %u\n"),
 201                     BlockNumber, block->BlockNumber);
 202             }
 203             BlockNumber = block->BlockNumber + 1;
 204             block->block_read = false;
 205          }
 206
 207          if (is_partial_record(rec)) {
 208             break;                    /* Go read full record */
 209          }
 210
 211          /*
 212           * We "finally" have a full record here. Now
 213           *   generate Header parameters and send to File daemon
 214           * Note, we build header in hdr buffer to avoid wiping
 215           *   out the data record
 216           */
 217          ds->msg = hdr;
 218          Dmsg5(400, "Send to FD: SessId=%u SessTim=%u FI=%d Strm=%d, len=%d\n",
 219             rec->VolSessionId, rec->VolSessionTime, rec->FileIndex, rec->Stream,
 220             rec->data_len);
 221          if (!bnet_fsend(ds, rec_header, rec->VolSessionId, rec->VolSessionTime,
 222                 rec->FileIndex, rec->Stream, rec->data_len)) {
 223             Dmsg1(30, ">filed: Error Hdr=%s\n", ds->msg);
 224             hdr = ds->msg;
 225             ds->msg = rec->data;
 226             Jmsg1(jcr, M_FATAL, 0, _("Error sending to File daemon. ERR=%s\n"),
 227                bnet_strerror(ds));
 228             ok = FALSE;
 229             break;
 230          } else {
 231             Dmsg1(30, ">filed: Hdr=%s\n", ds->msg);
 232          }
 233
 234          hdr = ds->msg;               /* restore hdr buffer */
 235          ds->msg = rec->data;         /* restore data record address */
 236
 237          /* Send data record to File daemon */
 238          ds->msglen = rec->data_len;
 239          Dmsg1(40, ">filed: send %d bytes data.\n", ds->msglen);
 240          if (!bnet_send(ds)) {
 241             Pmsg1(000, "Error sending to FD. ERR=%s\n", bnet_strerror(ds));
 242             Dmsg1(100, "Hdr=%s\n", hdr);
 243             Dmsg1(100, "data=%s\n", ds->msg);
 244             Jmsg1(jcr, M_FATAL, 0, _("Error sending to File daemon. ERR=%s\n"),
 245                bnet_strerror(ds));
 246             ok = FALSE;
 247             break;
 248          }
 249       } /* end for loop reading records */
 250    } /* end for loop reading blocks */
 251    /* Send end of data to FD */
 252    bnet_sig(ds, BNET_EOD);
 253
 254    if (!release_device(jcr, dev)) {
 255       ok = FALSE;
 256    }
 257    free_pool_memory(hdr);
 258    free_block(block);
 259    rec->data = NULL;                  /* previously released */
 260    free_record(rec);
 261    free_vol_list(jcr);
 262    Dmsg0(30, "Done reading.\n");
 263    return ok ? 1 : 0;
 264 }
 265
 266 static void get_session_record(DEVICE *dev, DEV_RECORD *rec, SESSION_LABEL *sessrec)
 267 {
 268    char *rtype;
 269    memset(sessrec, 0, sizeof(sessrec));
 270    switch (rec->FileIndex) {
 271    case PRE_LABEL:
 272       rtype = "Fresh Volume Label";
 273       break;
 274    case VOL_LABEL:
 275       rtype = "Volume Label";
 276       unser_volume_label(dev, rec);
 277       break;
 278    case SOS_LABEL:
 279       rtype = "Begin Session";
 280       unser_session_label(sessrec, rec);
 281       break;
 282    case EOS_LABEL:
 283       rtype = "End Session";
 284       break;
 285    case EOM_LABEL:
 286       rtype = "End of Media";
 287       break;
 288    default:
 289       rtype = "Unknown";
 290       break;
 291    }
 292    Dmsg5(10, "%s Record: VolSessionId=%d VolSessionTime=%d JobId=%d DataLen=%d\n",
 293          rtype, rec->VolSessionId, rec->VolSessionTime, rec->Stream, rec->data_len);
 294 }