]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/stored/bscan.c
Update utilities to accept Device name
[bacula/bacula] / bacula / src / stored / bscan.c
1 /*
2  *
3  *  Program to scan a Bacula Volume and compare it with
4  *    the catalog and optionally synchronize the catalog
5  *    with the tape.
6  *
7  *   Kern E. Sibbald, December 2001
8  *
9  *
10  *   Version $Id$
11  */
12 /*
13    Copyright (C) 2001-2004 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 #include "findlib/find.h"
35 #include "cats/cats.h"
36
37 /* Forward referenced functions */
38 static void do_scan(void);
39 static int record_cb(JCR *jcr, DEVICE *dev, DEV_BLOCK *block, DEV_RECORD *rec);
40 static int  create_file_attributes_record(B_DB *db, JCR *mjcr, 
41                                char *fname, char *lname, int type,
42                                char *ap, DEV_RECORD *rec);
43 static int  create_media_record(B_DB *db, MEDIA_DBR *mr, VOLUME_LABEL *vl);
44 static int  update_media_record(B_DB *db, MEDIA_DBR *mr);
45 static int  create_pool_record(B_DB *db, POOL_DBR *pr);
46 static JCR *create_job_record(B_DB *db, JOB_DBR *mr, SESSION_LABEL *label, DEV_RECORD *rec);
47 static int  update_job_record(B_DB *db, JOB_DBR *mr, SESSION_LABEL *elabel, 
48                               DEV_RECORD *rec);
49 static int  create_client_record(B_DB *db, CLIENT_DBR *cr);
50 static int  create_fileset_record(B_DB *db, FILESET_DBR *fsr);
51 static int  create_jobmedia_record(B_DB *db, JCR *jcr);
52 static JCR *create_jcr(JOB_DBR *jr, DEV_RECORD *rec, uint32_t JobId);
53 static int update_SIG_record(B_DB *db, char *SIGbuf, DEV_RECORD *rec, int type);
54
55
56 /* Global variables */
57 STORES *me;
58 #if defined(HAVE_CYGWIN) || defined(HAVE_WIN32)
59 int win32_client = 1;
60 #else
61 int win32_client = 0;
62 #endif
63
64
65 /* Local variables */
66 static DEVICE *dev = NULL;
67 static B_DB *db;
68 static JCR *bjcr;                     /* jcr for bscan */
69 static BSR *bsr = NULL;
70 static MEDIA_DBR mr;
71 static POOL_DBR pr;
72 static JOB_DBR jr;
73 static CLIENT_DBR cr;
74 static FILESET_DBR fsr;
75 static ATTR_DBR ar;
76 static FILE_DBR fr;
77 static SESSION_LABEL label;
78 static SESSION_LABEL elabel;
79 static ATTR *attr;
80
81 static time_t lasttime = 0;
82
83 static char *db_name = "bacula";
84 static char *db_user = "bacula";
85 static char *db_password = "";
86 static char *db_host = NULL;
87 static char *wd = NULL;
88 static int update_db = 0;
89 static int update_vol_info = 0;
90 static int list_records = 0;
91 static int ignored_msgs = 0;
92
93 static int num_jobs = 0;
94 static int num_pools = 0;
95 static int num_media = 0;
96 static int num_files = 0;
97
98 #define CONFIG_FILE "bacula-sd.conf"
99 char *configfile;
100 bool forge_on = false;
101
102
103 static void usage()
104 {
105    fprintf(stderr, _(
106 "\nVersion: " VERSION " (" BDATE ")\n\n"
107 "Usage: bscan [ options ] <bacula-archive>\n"
108 "       -b bootstrap      specify a bootstrap file\n"
109 "       -c <file>         specify configuration file\n"
110 "       -d <nn>           set debug level to nn\n"
111 "       -m                update media info in database\n"
112 "       -n <name>         specify the database name (default bacula)\n"
113 "       -u <user>         specify database user name (default bacula)\n"
114 "       -P <password      specify database password (default none)\n"
115 "       -h <host>         specify database host (default NULL)\n"
116 "       -p                proceed inspite of I/O errors\n"
117 "       -r                list records\n"
118 "       -s                synchronize or store in database\n"
119 "       -v                verbose\n"
120 "       -V <Volumes>      specify Volume names (separated by |)\n"
121 "       -w <dir>          specify working directory (default from conf file)\n"
122 "       -?                print this message\n\n"));
123    exit(1);
124 }
125
126 int main (int argc, char *argv[])
127 {
128    int ch;
129    struct stat stat_buf;
130    char *VolumeName = NULL;
131
132    my_name_is(argc, argv, "bscan");
133    init_msg(NULL, NULL);
134
135
136    while ((ch = getopt(argc, argv, "b:c:d:h:mn:pP:rsu:vV:w:?")) != -1) {
137       switch (ch) {
138       case 'b':
139          bsr = parse_bsr(NULL, optarg);
140          break;
141
142       case 'c':                    /* specify config file */
143          if (configfile != NULL) {
144             free(configfile);
145          }
146          configfile = bstrdup(optarg);
147          break;
148
149       case 'd':                    /* debug level */
150          debug_level = atoi(optarg);
151          if (debug_level <= 0)
152             debug_level = 1; 
153          break;
154
155       case 'h':
156          db_host = optarg;
157          break;
158
159       case 'm':
160          update_vol_info = 1;
161          break;
162
163       case 'n':
164          db_name = optarg;
165          break;
166
167       case 'u':
168          db_user = optarg;
169          break;
170
171       case 'P':
172          db_password = optarg;
173          break;
174
175       case 'p':
176          forge_on = true;
177          break;
178
179       case 'r':
180          list_records = 1;
181          break;
182
183       case 's':
184          update_db = 1;
185          break;
186
187       case 'v':
188          verbose++;
189          break;
190
191       case 'V':                    /* Volume name */
192          VolumeName = optarg;
193          break;
194
195       case 'w':
196          wd = optarg;
197          break;
198
199       case '?':
200       default:
201          usage();
202
203       }  
204    }
205    argc -= optind;
206    argv += optind;
207
208    if (argc != 1) {
209       Pmsg0(0, _("Wrong number of arguments: \n"));
210       usage();
211    }
212
213    if (configfile == NULL) {
214       configfile = bstrdup(CONFIG_FILE);
215    }
216
217    parse_config(configfile);
218    LockRes();
219    me = (STORES *)GetNextRes(R_STORAGE, NULL);
220    if (!me) {
221       UnlockRes();
222       Emsg1(M_ERROR_TERM, 0, _("No Storage resource defined in %s. Cannot continue.\n"), 
223          configfile);
224    }
225    UnlockRes();
226    /* Check if -w option given, otherwise use resource for working directory */
227    if (wd) { 
228       working_directory = wd;
229    } else if (!me->working_directory) {
230       Emsg1(M_ERROR_TERM, 0, _("No Working Directory defined in %s. Cannot continue.\n"),
231          configfile);
232    } else {
233       working_directory = me->working_directory;
234    }
235
236    /* Check that working directory is good */
237    if (stat(working_directory, &stat_buf) != 0) {
238       Emsg1(M_ERROR_TERM, 0, _("Working Directory: %s not found. Cannot continue.\n"),
239          working_directory);
240    }
241    if (!S_ISDIR(stat_buf.st_mode)) {
242       Emsg1(M_ERROR_TERM, 0, _("Working Directory: %s is not a directory. Cannot continue.\n"),
243          working_directory);
244    }
245
246    bjcr = setup_jcr("bscan", argv[0], bsr, VolumeName);
247    dev = setup_to_access_device(bjcr, 1);   /* read device */
248    if (!dev) { 
249       exit(1);
250    }
251
252    if ((db=db_init_database(NULL, db_name, db_user, db_password, db_host, 0, NULL)) == NULL) {
253       Emsg0(M_ERROR_TERM, 0, _("Could not init Bacula database\n"));
254    }
255    if (!db_open_database(NULL, db)) {
256       Emsg0(M_ERROR_TERM, 0, db_strerror(db));
257    }
258    Dmsg0(200, "Database opened\n");
259    if (verbose) {
260       Pmsg2(000, _("Using Database: %s, User: %s\n"), db_name, db_user);
261    }
262
263    do_scan();
264    printf("Records added to catalog:\n%7d Media\n%7d Pool\n%7d Job\n%7d File\n",
265       num_media, num_pools, num_jobs, num_files);
266
267    free_jcr(bjcr);
268    return 0;
269 }
270   
271 /*  
272  * We are at the end of reading a tape. Now, we simulate handling
273  *   the end of writing a tape by wiffling through the attached
274  *   jcrs creating jobmedia records.
275  */
276 static int bscan_mount_next_read_volume(JCR *jcr, DEVICE *dev, DEV_BLOCK *block)
277 {
278    Dmsg1(100, "Walk attached jcrs. Volume=%s\n", dev->VolCatInfo.VolCatName);
279    for (JCR *mjcr=NULL; (mjcr=next_attached_jcr(dev, mjcr)); ) {
280       DCR *dcr = mjcr->dcr;
281       if (verbose) {
282          Pmsg1(000, _("Create JobMedia for Job %s\n"), mjcr->Job);
283       }
284       if (dev->state & ST_TAPE) {
285          dcr->EndBlock = dev->EndBlock;
286          dcr->EndFile = dev->EndFile;
287       } else {
288          dcr->EndBlock = (uint32_t)dev->file_addr;
289          dcr->EndFile = (uint32_t)(dev->file_addr >> 32);
290       }
291       if (!create_jobmedia_record(db, mjcr)) {
292          Pmsg2(000, _("Could not create JobMedia record for Volume=%s Job=%s\n"),
293             dev->VolCatInfo.VolCatName, mjcr->Job);
294       }
295    }  
296    /* Now let common read routine get up next tape. Note,
297     * we call mount_next... with bscan's jcr because that is where we
298     * have the Volume list, but we get attached.
299     */
300    int stat = mount_next_read_volume(jcr, dev, block);
301    /* we must once more detach ourselves (attached by mount_next ...) */
302    detach_jcr_from_device(dev, jcr); /* detach bscan jcr */
303    return stat;
304 }
305
306 static void do_scan()             
307 {
308    attr = new_attr();
309
310    memset(&ar, 0, sizeof(ar));
311    memset(&pr, 0, sizeof(pr));
312    memset(&jr, 0, sizeof(jr));
313    memset(&cr, 0, sizeof(cr));
314    memset(&fsr, 0, sizeof(fsr));
315    memset(&fr, 0, sizeof(fr));
316
317    /* Detach bscan's jcr as we are not a real Job on the tape */
318    detach_jcr_from_device(dev, bjcr);
319
320    read_records(bjcr, dev, record_cb, bscan_mount_next_read_volume);
321    release_device(bjcr);
322
323    free_attr(attr);
324    term_dev(dev);
325 }
326
327 static int record_cb(JCR *bjcr, DEVICE *dev, DEV_BLOCK *block, DEV_RECORD *rec)
328 {
329    JCR *mjcr;
330    DCR *dcr;
331    char ec1[30];
332
333    if (rec->data_len > 0) {
334       mr.VolBytes += rec->data_len + WRITE_RECHDR_LENGTH; /* Accumulate Volume bytes */
335    }
336    if (list_records) {
337       Pmsg5(000, _("Record: SessId=%u SessTim=%u FileIndex=%d Stream=%d len=%u\n"),
338             rec->VolSessionId, rec->VolSessionTime, rec->FileIndex, 
339             rec->Stream, rec->data_len);
340    }
341    /* 
342     * Check for Start or End of Session Record 
343     *
344     */
345    if (rec->FileIndex < 0) {
346       int save_update_db = update_db;
347
348       if (verbose > 1) {
349          dump_label_record(dev, rec, 1);
350       }
351       switch (rec->FileIndex) {
352       case PRE_LABEL:
353          Pmsg0(000, _("Volume is prelabeled. This tape cannot be scanned.\n"));
354          return 1;
355          break;
356
357       case VOL_LABEL:
358          unser_volume_label(dev, rec);
359          /* Check Pool info */
360          bstrncpy(pr.Name, dev->VolHdr.PoolName, sizeof(pr.Name));
361          bstrncpy(pr.PoolType, dev->VolHdr.PoolType, sizeof(pr.PoolType));
362          num_pools++;
363          if (db_get_pool_record(bjcr, db, &pr)) {
364             if (verbose) {
365                Pmsg1(000, _("Pool record for %s found in DB.\n"), pr.Name);
366             }
367          } else {
368             if (!update_db) {
369                Pmsg1(000, _("VOL_LABEL: Pool record not found for Pool: %s\n"),
370                   pr.Name);
371             }
372             create_pool_record(db, &pr);
373          }
374          if (strcmp(pr.PoolType, dev->VolHdr.PoolType) != 0) {
375             Pmsg2(000, _("VOL_LABEL: PoolType mismatch. DB=%s Vol=%s\n"),
376                pr.PoolType, dev->VolHdr.PoolType);
377             return 1;
378          } else if (verbose) {
379             Pmsg1(000, _("Pool type \"%s\" is OK.\n"), pr.PoolType);
380          }
381
382          /* Check Media Info */
383          memset(&mr, 0, sizeof(mr));
384          bstrncpy(mr.VolumeName, dev->VolHdr.VolName, sizeof(mr.VolumeName));
385          mr.PoolId = pr.PoolId;
386          num_media++;
387          if (db_get_media_record(bjcr, db, &mr)) {
388             if (verbose) {
389                Pmsg1(000, _("Media record for %s found in DB.\n"), mr.VolumeName);
390             }
391             /* Clear out some volume statistics that will be updated */
392             mr.VolJobs = mr.VolFiles = mr.VolBlocks = 0;
393             mr.VolBytes = rec->data_len + 20;
394          } else {
395             if (!update_db) {
396                Pmsg1(000, _("VOL_LABEL: Media record not found for Volume: %s\n"),
397                   mr.VolumeName);
398             }
399             bstrncpy(mr.MediaType, dev->VolHdr.MediaType, sizeof(mr.MediaType));
400             create_media_record(db, &mr, &dev->VolHdr);
401          }
402          if (strcmp(mr.MediaType, dev->VolHdr.MediaType) != 0) {
403             Pmsg2(000, _("VOL_LABEL: MediaType mismatch. DB=%s Vol=%s\n"),
404                mr.MediaType, dev->VolHdr.MediaType);
405             return 1;
406          } else if (verbose) {
407             Pmsg1(000, _("Media type \"%s\" is OK.\n"), mr.MediaType);
408          }
409          /* Reset some JCR variables */
410          for (mjcr=NULL; (mjcr=next_attached_jcr(dev, mjcr)); ) {
411             dcr = mjcr->dcr;
412             dcr->VolFirstIndex = dcr->FileIndex = 0;
413             dcr->StartBlock = dcr->EndBlock = 0;
414             dcr->StartFile = dcr->EndFile = 0;
415          }
416
417          Pmsg1(000, _("VOL_LABEL: OK for Volume: %s\n"), mr.VolumeName);
418          break;
419
420       case SOS_LABEL:
421          mr.VolJobs++;
422          num_jobs++;
423          if (ignored_msgs > 0) {
424             Pmsg1(000, _("%d \"errors\" ignored before first Start of Session record.\n"), 
425                   ignored_msgs);
426             ignored_msgs = 0;
427          }
428          unser_session_label(&label, rec);
429          memset(&jr, 0, sizeof(jr));
430          jr.JobId = label.JobId;
431          if (db_get_job_record(bjcr, db, &jr)) {
432             /* Job record already exists in DB */
433             update_db = 0;  /* don't change db in create_job_record */
434             if (verbose) {
435                Pmsg1(000, _("SOS_LABEL: Found Job record for JobId: %d\n"), jr.JobId);
436             }
437          } else {
438             /* Must create a Job record in DB */
439             if (!update_db) {
440                Pmsg1(000, _("SOS_LABEL: Job record not found for JobId: %d\n"),
441                   jr.JobId);
442             }
443          }
444          /* Create Client record if not already there */
445             bstrncpy(cr.Name, label.ClientName, sizeof(cr.Name));
446             create_client_record(db, &cr);
447             jr.ClientId = cr.ClientId;
448
449          /* process label, if Job record exists don't update db */
450          mjcr = create_job_record(db, &jr, &label, rec);
451          dcr = mjcr->dcr;
452          update_db = save_update_db;
453
454          jr.PoolId = pr.PoolId;
455          /* Set start positions into JCR */
456          if (dev->state & ST_TAPE) {
457             /*
458              * Note, we have already advanced past current block,
459              *  so the correct number is block_num - 1 
460              */ 
461             dcr->StartBlock = dev->block_num - 1;
462             dcr->StartFile = dev->file;
463          } else {
464             dcr->StartBlock = (uint32_t)dev->file_addr;
465             dcr->StartFile = (uint32_t)(dev->file_addr >> 32);
466          }
467          mjcr->start_time = jr.StartTime;
468          mjcr->JobLevel = jr.Level;
469
470          mjcr->client_name = get_pool_memory(PM_FNAME);
471          pm_strcpy(&mjcr->client_name, label.ClientName);
472          mjcr->pool_type = get_pool_memory(PM_FNAME);
473          pm_strcpy(&mjcr->pool_type, label.PoolType);
474          mjcr->fileset_name = get_pool_memory(PM_FNAME);
475          pm_strcpy(&mjcr->fileset_name, label.FileSetName);
476          mjcr->pool_name = get_pool_memory(PM_FNAME);
477          pm_strcpy(&mjcr->pool_name, label.PoolName);
478
479          if (rec->VolSessionId != jr.VolSessionId) {
480             Pmsg3(000, _("SOS_LABEL: VolSessId mismatch for JobId=%u. DB=%d Vol=%d\n"),
481                jr.JobId,
482                jr.VolSessionId, rec->VolSessionId);
483             return 1;
484          }
485          if (rec->VolSessionTime != jr.VolSessionTime) {
486             Pmsg3(000, _("SOS_LABEL: VolSessTime mismatch for JobId=%u. DB=%d Vol=%d\n"),
487                jr.JobId,
488                jr.VolSessionTime, rec->VolSessionTime);
489             return 1;
490          }
491          if (jr.PoolId != pr.PoolId) {
492             Pmsg3(000, _("SOS_LABEL: PoolId mismatch for JobId=%u. DB=%d Vol=%d\n"),
493                jr.JobId,
494                jr.PoolId, pr.PoolId);
495             return 1;
496          }
497          break;
498
499       case EOS_LABEL:
500          unser_session_label(&elabel, rec);
501
502          /* Create FileSet record */
503          bstrncpy(fsr.FileSet, label.FileSetName, sizeof(fsr.FileSet));
504          bstrncpy(fsr.MD5, label.FileSetMD5, sizeof(fsr.MD5));
505          create_fileset_record(db, &fsr);
506          jr.FileSetId = fsr.FileSetId;
507
508          mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
509          if (!mjcr) {
510             Pmsg2(000, _("Could not find SessId=%d SessTime=%d for EOS record.\n"),
511                   rec->VolSessionId, rec->VolSessionTime);
512             break;
513          }
514
515          /* Do the final update to the Job record */
516          update_job_record(db, &jr, &elabel, rec);
517
518          mjcr->end_time = jr.EndTime;
519          mjcr->JobStatus = JS_Terminated;
520
521          /* Create JobMedia record */
522          create_jobmedia_record(db, mjcr);
523          detach_jcr_from_device(dev, mjcr);
524          free_jcr(mjcr);
525
526          break;
527
528       case EOM_LABEL:
529          break;
530
531       case EOT_LABEL:              /* end of all tapes */
532          /* 
533           * Wiffle through all jobs still open and close
534           *   them.
535           */
536          if (update_db) {
537             mjcr=next_attached_jcr(dev, NULL);
538             for ( ; mjcr; ) {
539                JCR *njcr; 
540                jr.JobId = mjcr->JobId;
541                jr.JobStatus = JS_ErrorTerminated;
542                jr.JobFiles = mjcr->JobFiles;
543                jr.JobBytes = mjcr->JobBytes;
544                jr.VolSessionId = mjcr->VolSessionId;
545                jr.VolSessionTime = mjcr->VolSessionTime;
546                jr.JobTDate = (utime_t)mjcr->start_time;
547                jr.ClientId = mjcr->ClientId;
548                if (!db_update_job_end_record(bjcr, db, &jr)) {
549                   Pmsg1(0, _("Could not update job record. ERR=%s\n"), db_strerror(db));
550                }
551                njcr = mjcr->next_dev;
552                free_jcr(mjcr);
553                mjcr = njcr;
554             }
555          }
556          mr.VolFiles = rec->File;
557          mr.VolBlocks = rec->Block;
558          mr.VolBytes += mr.VolBlocks * WRITE_BLKHDR_LENGTH; /* approx. */
559          mr.VolMounts++;
560          update_media_record(db, &mr);
561          Pmsg3(0, _("End of all Volumes. VolFiles=%u VolBlocks=%u VolBytes=%s\n"), mr.VolFiles,
562                     mr.VolBlocks, edit_uint64_with_commas(mr.VolBytes, ec1));
563          break;
564       default:
565          break;
566       } /* end switch */
567       return 1;
568    }
569
570    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
571    if (!mjcr) {
572       if (mr.VolJobs > 0) {
573          Pmsg2(000, _("Could not find Job for SessId=%d SessTime=%d record.\n"),
574                       rec->VolSessionId, rec->VolSessionTime);
575       } else {
576          ignored_msgs++;
577       }
578       return 1;
579    }
580    dcr = mjcr->dcr;
581    if (dcr->VolFirstIndex == 0) {
582       dcr->VolFirstIndex = block->FirstIndex;
583    }
584
585    /* File Attributes stream */
586    switch (rec->Stream) {
587    case STREAM_UNIX_ATTRIBUTES:   
588    case STREAM_UNIX_ATTRIBUTES_EX:  
589
590       if (!unpack_attributes_record(bjcr, rec->Stream, rec->data, attr)) {
591          Emsg0(M_ERROR_TERM, 0, _("Cannot continue.\n"));
592       }
593
594       if (attr->file_index != rec->FileIndex) {
595          Emsg2(M_ERROR_TERM, 0, _("Record header file index %ld not equal record index %ld\n"),
596             rec->FileIndex, attr->file_index);
597       }
598        
599       if (verbose > 1) {
600          decode_stat(attr->attr, &attr->statp, &attr->LinkFI);
601          build_attr_output_fnames(bjcr, attr);
602          print_ls_output(bjcr, attr);
603       }
604       fr.JobId = mjcr->JobId;
605       fr.FileId = 0;
606       num_files++;
607       if (db_get_file_attributes_record(bjcr, db, attr->fname, NULL, &fr)) {
608          if (verbose > 1) {
609             Pmsg1(000, _("File record already exists for: %s\n"), attr->fname);
610          }
611       } else {
612          create_file_attributes_record(db, mjcr, attr->fname, attr->lname, 
613             attr->type, attr->attr, rec);
614       }
615       free_jcr(mjcr);
616       break;
617
618    /* Data stream */
619    case STREAM_WIN32_DATA:
620    case STREAM_FILE_DATA:
621    case STREAM_SPARSE_DATA:
622       mjcr->JobBytes += rec->data_len;
623       if (rec->Stream == STREAM_SPARSE_DATA) {
624          mjcr->JobBytes -= sizeof(uint64_t);
625       }
626          
627       free_jcr(mjcr);                 /* done using JCR */
628       break;
629
630    case STREAM_GZIP_DATA:
631       mjcr->JobBytes += rec->data_len; /* No correct, we should expand it */
632       free_jcr(mjcr);                 /* done using JCR */
633       break;
634
635    case STREAM_SPARSE_GZIP_DATA:
636       mjcr->JobBytes += rec->data_len - sizeof(uint64_t); /* No correct, we should expand it */
637       free_jcr(mjcr);                 /* done using JCR */
638       break;
639
640    /* Win32 GZIP stream */
641    case STREAM_WIN32_GZIP_DATA:
642       mjcr->JobBytes += rec->data_len;
643       free_jcr(mjcr);                 /* done using JCR */
644       break;
645
646    case STREAM_MD5_SIGNATURE:
647       char MD5buf[50];
648       bin_to_base64(MD5buf, (char *)rec->data, 16); /* encode 16 bytes */
649       if (verbose > 1) {
650          Pmsg1(000, _("Got MD5 record: %s\n"), MD5buf);
651       }
652       update_SIG_record(db, MD5buf, rec, MD5_SIG);
653       break;
654
655    case STREAM_SHA1_SIGNATURE:
656       char SIGbuf[50];
657       bin_to_base64(SIGbuf, (char *)rec->data, 20); /* encode 20 bytes */
658       if (verbose > 1) {
659          Pmsg1(000, _("Got SHA1 record: %s\n"), SIGbuf);
660       }
661       update_SIG_record(db, SIGbuf, rec, SHA1_SIG);
662       break;
663
664
665    case STREAM_PROGRAM_NAMES:
666       if (verbose) {
667          Pmsg1(000, _("Got Prog Names Stream: %s\n"), rec->data);
668       }
669       break;
670
671    case STREAM_PROGRAM_DATA:
672       if (verbose > 1) {
673          Pmsg0(000, _("Got Prog Data Stream record.\n"));
674       }
675       break;
676    default:
677       Pmsg2(0, _("Unknown stream type!!! stream=%d data=%s\n"), rec->Stream, rec->data);
678       break;
679    }
680    return 1;
681 }
682
683 /*
684  * Free the Job Control Record if no one is still using it.
685  *  Called from main free_jcr() routine in src/lib/jcr.c so
686  *  that we can do our Director specific cleanup of the jcr.
687  */
688 static void bscan_free_jcr(JCR *jcr)
689 {
690    Dmsg0(200, "Start dird free_jcr\n");
691
692    if (jcr->file_bsock) {
693       Dmsg0(200, "Close File bsock\n");
694       bnet_close(jcr->file_bsock);
695    }
696    if (jcr->store_bsock) {
697       Dmsg0(200, "Close Store bsock\n");
698       bnet_close(jcr->store_bsock);
699    }
700    if (jcr->RestoreBootstrap) {
701       free(jcr->RestoreBootstrap);
702    }
703    if (jcr->dcr) {
704       free_dcr(jcr->dcr);
705       jcr->dcr = NULL;
706    }
707    Dmsg0(200, "End dird free_jcr\n");
708 }
709
710 /*
711  * We got a File Attributes record on the tape.  Now, lookup the Job
712  *   record, and then create the attributes record.
713  */
714 static int create_file_attributes_record(B_DB *db, JCR *mjcr,
715                                char *fname, char *lname, int type,
716                                char *ap, DEV_RECORD *rec)
717 {
718    DCR *dcr = mjcr->dcr;
719    ar.fname = fname;
720    ar.link = lname;
721    ar.ClientId = mjcr->ClientId;
722    ar.JobId = mjcr->JobId;
723    ar.Stream = rec->Stream;
724    ar.FileIndex = rec->FileIndex;
725    ar.attr = ap;
726    if (dcr->VolFirstIndex == 0) {
727       dcr->VolFirstIndex = rec->FileIndex;
728    }
729    dcr->FileIndex = rec->FileIndex;
730    mjcr->JobFiles++;
731
732    if (!update_db) {
733       return 1;
734    }
735
736    if (!db_create_file_attributes_record(bjcr, db, &ar)) {
737       Pmsg1(0, _("Could not create File Attributes record. ERR=%s\n"), db_strerror(db));
738       return 0;
739    }
740    mjcr->FileId = ar.FileId;
741
742    if (verbose > 1) {
743       Pmsg1(000, _("Created File record: %s\n"), fname);   
744    }
745    return 1;
746 }
747
748 /*
749  * For each Volume we see, we create a Medium record
750  */
751 static int create_media_record(B_DB *db, MEDIA_DBR *mr, VOLUME_LABEL *vl)
752 {
753    struct date_time dt;
754    struct tm tm;
755
756    /* We mark Vols as Archive to keep them from being re-written */
757    bstrncpy(mr->VolStatus, "Archive", sizeof(mr->VolStatus));
758    mr->VolRetention = 365 * 3600 * 24; /* 1 year */
759    if (vl->VerNum >= 11) {
760       mr->FirstWritten = btime_to_utime(vl->write_btime);
761       mr->LabelDate    = btime_to_utime(vl->label_btime);
762    } else {
763       /* DEPRECATED DO NOT USE */
764       dt.julian_day_number = vl->write_date;
765       dt.julian_day_fraction = vl->write_time;
766       tm_decode(&dt, &tm);
767       mr->FirstWritten = mktime(&tm);
768       dt.julian_day_number = vl->label_date;
769       dt.julian_day_fraction = vl->label_time;
770       tm_decode(&dt, &tm);
771       mr->LabelDate = mktime(&tm);
772    }
773    lasttime = mr->LabelDate;
774
775    if (!update_db) {
776       return 1;
777    }
778
779    if (!db_create_media_record(bjcr, db, mr)) {
780       Pmsg1(0, _("Could not create media record. ERR=%s\n"), db_strerror(db));
781       return 0;
782    }
783    if (!db_update_media_record(bjcr, db, mr)) {
784       Pmsg1(0, _("Could not update media record. ERR=%s\n"), db_strerror(db));
785       return 0;
786    }
787    if (verbose) {
788       Pmsg1(000, _("Created Media record for Volume: %s\n"), mr->VolumeName);
789    }
790    return 1;
791
792 }
793
794 /*
795  * Called at end of media to update it
796  */
797 static int update_media_record(B_DB *db, MEDIA_DBR *mr)
798 {
799    if (!update_db && !update_vol_info) {
800       return 1;
801    }
802
803    mr->LastWritten = lasttime;
804    if (!db_update_media_record(bjcr, db, mr)) {
805       Pmsg1(0, _("Could not update media record. ERR=%s\n"), db_strerror(db));
806       return 0;
807    }
808    if (verbose) {
809       Pmsg1(000, _("Updated Media record at end of Volume: %s\n"), mr->VolumeName);
810    }
811    return 1;
812
813 }
814
815
816 static int create_pool_record(B_DB *db, POOL_DBR *pr)
817 {
818    pr->NumVols++;
819    pr->UseCatalog = 1;
820    pr->VolRetention = 355 * 3600 * 24; /* 1 year */
821
822    if (!update_db) {
823       return 1;
824    }
825    if (!db_create_pool_record(bjcr, db, pr)) {
826       Pmsg1(0, _("Could not create pool record. ERR=%s\n"), db_strerror(db));
827       return 0;
828    }
829    if (verbose) {
830       Pmsg1(000, _("Created Pool record for Pool: %s\n"), pr->Name);
831    }
832    return 1;
833
834 }
835
836
837 /*
838  * Called from SOS to create a client for the current Job 
839  */
840 static int create_client_record(B_DB *db, CLIENT_DBR *cr)
841 {
842    if (!update_db) {
843       return 1;
844    }
845    if (!db_create_client_record(bjcr, db, cr)) {
846       Pmsg1(0, _("Could not create Client record. ERR=%s\n"), db_strerror(db));
847       return 0;
848    }
849    if (verbose) {
850       Pmsg1(000, _("Created Client record for Client: %s\n"), cr->Name);
851    }
852    return 1;
853 }
854
855 static int create_fileset_record(B_DB *db, FILESET_DBR *fsr)
856 {
857    if (!update_db) {
858       return 1;
859    }
860    fsr->FileSetId = 0;
861    if (fsr->MD5[0] == 0) {
862       fsr->MD5[0] = ' ';              /* Equivalent to nothing */
863       fsr->MD5[1] = 0;
864    }
865    if (db_get_fileset_record(bjcr, db, fsr)) {
866       if (verbose) {
867          Pmsg1(000, _("Fileset \"%s\" already exists.\n"), fsr->FileSet);
868       }
869    } else {
870       if (!db_create_fileset_record(bjcr, db, fsr)) {
871          Pmsg2(0, _("Could not create FileSet record \"%s\". ERR=%s\n"), 
872             fsr->FileSet, db_strerror(db));
873          return 0;
874       }
875       if (verbose) {
876          Pmsg1(000, _("Created FileSet record \"%s\"\n"), fsr->FileSet);
877       }
878    }
879    return 1;
880 }
881
882 /*
883  * Simulate the two calls on the database to create
884  *  the Job record and to update it when the Job actually
885  *  begins running.
886  */
887 static JCR *create_job_record(B_DB *db, JOB_DBR *jr, SESSION_LABEL *label, 
888                              DEV_RECORD *rec)
889 {
890    JCR *mjcr;
891    struct date_time dt;
892    struct tm tm;
893
894    jr->JobId = label->JobId;
895    jr->Type = label->JobType;
896    jr->Level = label->JobLevel;
897    jr->JobStatus = JS_Created;
898    bstrncpy(jr->Name, label->JobName, sizeof(jr->Name));
899    bstrncpy(jr->Job, label->Job, sizeof(jr->Job));
900    if (label->VerNum >= 11) {
901       jr->SchedTime = btime_to_unix(label->write_btime);
902    } else {
903       dt.julian_day_number = label->write_date;
904       dt.julian_day_fraction = label->write_time;
905       tm_decode(&dt, &tm);
906       jr->SchedTime = mktime(&tm);
907    }
908
909    jr->StartTime = jr->SchedTime;
910    jr->JobTDate = (utime_t)jr->SchedTime;
911    jr->VolSessionId = rec->VolSessionId;
912    jr->VolSessionTime = rec->VolSessionTime;
913
914    /* Now create a JCR as if starting the Job */
915    mjcr = create_jcr(jr, rec, label->JobId);
916
917    if (!update_db) {
918       return mjcr;
919    }
920
921    /* This creates the bare essentials */
922    if (!db_create_job_record(bjcr, db, jr)) {
923       Pmsg1(0, _("Could not create JobId record. ERR=%s\n"), db_strerror(db));
924       return mjcr;
925    }
926
927    /* This adds the client, StartTime, JobTDate, ... */
928    if (!db_update_job_start_record(bjcr, db, jr)) {
929       Pmsg1(0, _("Could not update job start record. ERR=%s\n"), db_strerror(db));
930       return mjcr;
931    }
932    Pmsg2(000, _("Created new JobId=%u record for original JobId=%u\n"), jr->JobId, 
933          label->JobId);
934    mjcr->JobId = jr->JobId;           /* set new JobId */
935    return mjcr;
936 }
937
938 /* 
939  * Simulate the database call that updates the Job 
940  *  at Job termination time.
941  */
942 static int update_job_record(B_DB *db, JOB_DBR *jr, SESSION_LABEL *elabel,
943                               DEV_RECORD *rec)
944 {
945    struct date_time dt;
946    struct tm tm;
947    JCR *mjcr;
948
949    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
950    if (!mjcr) {
951       Pmsg2(000, _("Could not find SessId=%d SessTime=%d for EOS record.\n"),
952                    rec->VolSessionId, rec->VolSessionTime);
953       return 0;
954    }
955    if (elabel->VerNum >= 11) {
956       jr->EndTime = btime_to_unix(elabel->write_btime);
957    } else {
958       dt.julian_day_number = elabel->write_date;
959       dt.julian_day_fraction = elabel->write_time;
960       tm_decode(&dt, &tm);
961       jr->EndTime = mktime(&tm);
962    }
963    lasttime = jr->EndTime;
964    mjcr->end_time = jr->EndTime;
965
966    jr->JobId = mjcr->JobId;
967    jr->JobStatus = elabel->JobStatus;
968    mjcr->JobStatus = elabel->JobStatus;
969    jr->JobFiles = elabel->JobFiles;
970    jr->JobBytes = elabel->JobBytes;
971    jr->VolSessionId = rec->VolSessionId;
972    jr->VolSessionTime = rec->VolSessionTime;
973    jr->JobTDate = (utime_t)mjcr->start_time;
974    jr->ClientId = mjcr->ClientId;
975
976    if (!update_db) {
977       free_jcr(mjcr);
978       return 1;
979    }
980    
981    if (!db_update_job_end_record(bjcr, db, jr)) {
982       Pmsg2(0, _("Could not update JobId=%u record. ERR=%s\n"), jr->JobId,  db_strerror(db));
983       free_jcr(mjcr);
984       return 0;
985    }
986    if (verbose) {
987       Pmsg1(000, _("Updated Job termination record for new JobId=%u\n"), jr->JobId);
988    }
989    if (verbose > 1) {
990       char *term_msg;
991       static char term_code[70];
992       char sdt[50], edt[50];
993       char ec1[30], ec2[30], ec3[30];
994
995       switch (mjcr->JobStatus) {
996       case JS_Terminated:
997          term_msg = _("Backup OK");
998          break;
999       case JS_FatalError:
1000       case JS_ErrorTerminated:
1001          term_msg = _("*** Backup Error ***");
1002          break;
1003       case JS_Canceled:
1004          term_msg = _("Backup Canceled");
1005          break;
1006       default:
1007          term_msg = term_code;
1008          sprintf(term_code, _("Job Termination code: %d"), mjcr->JobStatus);
1009          break;
1010       }
1011       bstrftime(sdt, sizeof(sdt), mjcr->start_time);
1012       bstrftime(edt, sizeof(edt), mjcr->end_time);
1013       Pmsg14(000,  _("%s\n\
1014 JobId:                  %d\n\
1015 Job:                    %s\n\
1016 FileSet:                %s\n\
1017 Backup Level:           %s\n\
1018 Client:                 %s\n\
1019 Start time:             %s\n\
1020 End time:               %s\n\
1021 Files Written:          %s\n\
1022 Bytes Written:          %s\n\
1023 Volume Session Id:      %d\n\
1024 Volume Session Time:    %d\n\
1025 Last Volume Bytes:      %s\n\
1026 Termination:            %s\n\n"),
1027         edt,
1028         mjcr->JobId,
1029         mjcr->Job,
1030         mjcr->fileset_name,
1031         job_level_to_str(mjcr->JobLevel),
1032         mjcr->client_name,
1033         sdt,
1034         edt,
1035         edit_uint64_with_commas(mjcr->JobFiles, ec1),
1036         edit_uint64_with_commas(mjcr->JobBytes, ec2),
1037         mjcr->VolSessionId,
1038         mjcr->VolSessionTime,
1039         edit_uint64_with_commas(mr.VolBytes, ec3),
1040         term_msg);
1041    }
1042    free_jcr(mjcr);
1043    return 1;
1044 }
1045
1046 static int create_jobmedia_record(B_DB *db, JCR *mjcr)
1047 {
1048    JOBMEDIA_DBR jmr;
1049    DCR *dcr = mjcr->dcr;
1050
1051    if (dev->state & ST_TAPE) {
1052       dcr->EndBlock = dev->EndBlock;
1053       dcr->EndFile  = dev->EndFile;
1054    } else {
1055       dcr->EndBlock = (uint32_t)dev->file_addr;
1056       dcr->EndFile = (uint32_t)(dev->file_addr >> 32);
1057    }
1058
1059    memset(&jmr, 0, sizeof(jmr));
1060    jmr.JobId = mjcr->JobId;
1061    jmr.MediaId = mr.MediaId;
1062    jmr.FirstIndex = dcr->VolFirstIndex;
1063    jmr.LastIndex = dcr->FileIndex;
1064    jmr.StartFile = dcr->StartFile;
1065    jmr.EndFile = dcr->EndFile;
1066    jmr.StartBlock = dcr->StartBlock;
1067    jmr.EndBlock = dcr->EndBlock;
1068
1069
1070    if (!update_db) {
1071       return 1;
1072    }
1073
1074    if (!db_create_jobmedia_record(bjcr, db, &jmr)) {
1075       Pmsg1(0, _("Could not create JobMedia record. ERR=%s\n"), db_strerror(db));
1076       return 0;
1077    }
1078    if (verbose) {
1079       Pmsg2(000, _("Created JobMedia record JobId %d, MediaId %d\n"), 
1080                 jmr.JobId, jmr.MediaId);
1081    }
1082    return 1;
1083 }
1084
1085 /* 
1086  * Simulate the database call that updates the MD5/SHA1 record
1087  */
1088 static int update_SIG_record(B_DB *db, char *SIGbuf, DEV_RECORD *rec, int type)
1089 {
1090    JCR *mjcr;
1091
1092    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
1093    if (!mjcr) {
1094       if (mr.VolJobs > 0) {
1095          Pmsg2(000, _("Could not find SessId=%d SessTime=%d for MD5/SHA1 record.\n"),
1096                       rec->VolSessionId, rec->VolSessionTime);
1097       } else {
1098          ignored_msgs++;
1099       }
1100       return 0;
1101    }
1102
1103    if (!update_db) {
1104       free_jcr(mjcr);
1105       return 1;
1106    }
1107    
1108    if (!db_add_SIG_to_file_record(bjcr, db, mjcr->FileId, SIGbuf, type)) {
1109       Pmsg1(0, _("Could not add MD5/SHA1 to File record. ERR=%s\n"), db_strerror(db));
1110       free_jcr(mjcr);
1111       return 0;
1112    }
1113    if (verbose > 1) {
1114       Pmsg0(000, _("Updated MD5/SHA1 record\n"));
1115    }
1116    free_jcr(mjcr);
1117    return 1;
1118 }
1119
1120
1121 /* 
1122  * Create a JCR as if we are really starting the job
1123  */
1124 static JCR *create_jcr(JOB_DBR *jr, DEV_RECORD *rec, uint32_t JobId)
1125 {
1126    JCR *jobjcr;
1127    /*
1128     * Transfer as much as possible to the Job JCR. Most important is
1129     *   the JobId and the ClientId.
1130     */
1131    jobjcr = new_jcr(sizeof(JCR), bscan_free_jcr);
1132    jobjcr->JobType = jr->Type;
1133    jobjcr->JobLevel = jr->Level;
1134    jobjcr->JobStatus = jr->JobStatus;
1135    bstrncpy(jobjcr->Job, jr->Job, sizeof(jobjcr->Job));
1136    jobjcr->JobId = JobId;      /* this is JobId on tape */
1137    jobjcr->sched_time = jr->SchedTime;
1138    jobjcr->start_time = jr->StartTime;
1139    jobjcr->VolSessionId = rec->VolSessionId;
1140    jobjcr->VolSessionTime = rec->VolSessionTime;
1141    jobjcr->ClientId = jr->ClientId;
1142    attach_jcr_to_device(dev, jobjcr);
1143    new_dcr(jobjcr, dev);
1144    return jobjcr;
1145 }
1146
1147 /* Dummies to replace askdir.c */
1148 int     dir_get_volume_info(JCR *jcr, enum get_vol_info_rw  writing) { return 1;}
1149 int     dir_find_next_appendable_volume(JCR *jcr) { return 1;}
1150 int     dir_update_volume_info(JCR *jcr, DEVICE *dev, int relabel) { return 1; }
1151 int     dir_create_jobmedia_record(JCR *jcr) { return 1; }
1152 int     dir_ask_sysop_to_create_appendable_volume(JCR *jcr, DEVICE *dev) { return 1; }
1153 int     dir_update_file_attributes(JCR *jcr, DEV_RECORD *rec) { return 1;}
1154 int     dir_send_job_status(JCR *jcr) {return 1;}
1155
1156
1157 int dir_ask_sysop_to_mount_volume(JCR *jcr, DEVICE *dev)
1158 {
1159    fprintf(stderr, _("Mount Volume \"%s\" on device \"%s\" and press return when ready: "),
1160       jcr->VolumeName, dev_name(dev));
1161    getchar();   
1162    return 1;
1163 }