]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/stored/bscan.c
Add stats to bscan
[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 [-d debug_level] <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->StartBlock = (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             dcr->StartBlock = dev->block_num;
458             dcr->StartFile = dev->file;
459          } else {
460             dcr->StartBlock = (uint32_t)dev->file_addr;
461             dcr->StartFile = (uint32_t)(dev->file_addr >> 32);
462          }
463          mjcr->start_time = jr.StartTime;
464          mjcr->JobLevel = jr.Level;
465
466          mjcr->client_name = get_pool_memory(PM_FNAME);
467          pm_strcpy(&mjcr->client_name, label.ClientName);
468          mjcr->pool_type = get_pool_memory(PM_FNAME);
469          pm_strcpy(&mjcr->pool_type, label.PoolType);
470          mjcr->fileset_name = get_pool_memory(PM_FNAME);
471          pm_strcpy(&mjcr->fileset_name, label.FileSetName);
472          mjcr->pool_name = get_pool_memory(PM_FNAME);
473          pm_strcpy(&mjcr->pool_name, label.PoolName);
474
475          if (rec->VolSessionId != jr.VolSessionId) {
476             Pmsg3(000, _("SOS_LABEL: VolSessId mismatch for JobId=%u. DB=%d Vol=%d\n"),
477                jr.JobId,
478                jr.VolSessionId, rec->VolSessionId);
479             return 1;
480          }
481          if (rec->VolSessionTime != jr.VolSessionTime) {
482             Pmsg3(000, _("SOS_LABEL: VolSessTime mismatch for JobId=%u. DB=%d Vol=%d\n"),
483                jr.JobId,
484                jr.VolSessionTime, rec->VolSessionTime);
485             return 1;
486          }
487          if (jr.PoolId != pr.PoolId) {
488             Pmsg3(000, _("SOS_LABEL: PoolId mismatch for JobId=%u. DB=%d Vol=%d\n"),
489                jr.JobId,
490                jr.PoolId, pr.PoolId);
491             return 1;
492          }
493          break;
494
495       case EOS_LABEL:
496          unser_session_label(&elabel, rec);
497
498          /* Create FileSet record */
499          bstrncpy(fsr.FileSet, label.FileSetName, sizeof(fsr.FileSet));
500          bstrncpy(fsr.MD5, label.FileSetMD5, sizeof(fsr.MD5));
501          create_fileset_record(db, &fsr);
502          jr.FileSetId = fsr.FileSetId;
503
504          mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
505          if (!mjcr) {
506             Pmsg2(000, _("Could not find SessId=%d SessTime=%d for EOS record.\n"),
507                   rec->VolSessionId, rec->VolSessionTime);
508             break;
509          }
510
511          /* Do the final update to the Job record */
512          update_job_record(db, &jr, &elabel, rec);
513
514          mjcr->end_time = jr.EndTime;
515          mjcr->JobStatus = JS_Terminated;
516
517          /* Create JobMedia record */
518          create_jobmedia_record(db, mjcr);
519          detach_jcr_from_device(dev, mjcr);
520          free_jcr(mjcr);
521
522          break;
523
524       case EOM_LABEL:
525          break;
526
527       case EOT_LABEL:              /* end of all tapes */
528          /* 
529           * Wiffle through all jobs still open and close
530           *   them.
531           */
532          if (update_db) {
533             mjcr=next_attached_jcr(dev, NULL);
534             for ( ; mjcr; ) {
535                JCR *njcr; 
536                jr.JobId = mjcr->JobId;
537                jr.JobStatus = JS_ErrorTerminated;
538                jr.JobFiles = mjcr->JobFiles;
539                jr.JobBytes = mjcr->JobBytes;
540                jr.VolSessionId = mjcr->VolSessionId;
541                jr.VolSessionTime = mjcr->VolSessionTime;
542                jr.JobTDate = (utime_t)mjcr->start_time;
543                jr.ClientId = mjcr->ClientId;
544                if (!db_update_job_end_record(bjcr, db, &jr)) {
545                   Pmsg1(0, _("Could not update job record. ERR=%s\n"), db_strerror(db));
546                }
547                njcr = mjcr->next_dev;
548                free_jcr(mjcr);
549                mjcr = njcr;
550             }
551          }
552          mr.VolFiles = rec->File;
553          mr.VolBlocks = rec->Block;
554          mr.VolBytes += mr.VolBlocks * WRITE_BLKHDR_LENGTH; /* approx. */
555          mr.VolMounts++;
556          update_media_record(db, &mr);
557          Pmsg3(0, _("End of all Volumes. VolFiles=%u VolBlocks=%u VolBytes=%s\n"), mr.VolFiles,
558                     mr.VolBlocks, edit_uint64_with_commas(mr.VolBytes, ec1));
559          break;
560       default:
561          break;
562       } /* end switch */
563       return 1;
564    }
565
566    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
567    if (!mjcr) {
568       if (mr.VolJobs > 0) {
569          Pmsg2(000, _("Could not find Job for SessId=%d SessTime=%d record.\n"),
570                       rec->VolSessionId, rec->VolSessionTime);
571       } else {
572          ignored_msgs++;
573       }
574       return 1;
575    }
576    dcr = mjcr->dcr;
577    if (dcr->VolFirstIndex == 0) {
578       dcr->VolFirstIndex = block->FirstIndex;
579    }
580
581    /* File Attributes stream */
582    switch (rec->Stream) {
583    case STREAM_UNIX_ATTRIBUTES:   
584    case STREAM_UNIX_ATTRIBUTES_EX:  
585
586       if (!unpack_attributes_record(bjcr, rec->Stream, rec->data, attr)) {
587          Emsg0(M_ERROR_TERM, 0, _("Cannot continue.\n"));
588       }
589
590       if (attr->file_index != rec->FileIndex) {
591          Emsg2(M_ERROR_TERM, 0, _("Record header file index %ld not equal record index %ld\n"),
592             rec->FileIndex, attr->file_index);
593       }
594        
595       if (verbose > 1) {
596          decode_stat(attr->attr, &attr->statp, &attr->LinkFI);
597          build_attr_output_fnames(bjcr, attr);
598          print_ls_output(bjcr, attr);
599       }
600       fr.JobId = mjcr->JobId;
601       fr.FileId = 0;
602       num_files++;
603       if (db_get_file_attributes_record(bjcr, db, attr->fname, NULL, &fr)) {
604          if (verbose > 1) {
605             Pmsg1(000, _("File record already exists for: %s\n"), attr->fname);
606          }
607       } else {
608          create_file_attributes_record(db, mjcr, attr->fname, attr->lname, 
609             attr->type, attr->attr, rec);
610       }
611       free_jcr(mjcr);
612       break;
613
614    /* Data stream */
615    case STREAM_WIN32_DATA:
616    case STREAM_FILE_DATA:
617    case STREAM_SPARSE_DATA:
618       mjcr->JobBytes += rec->data_len;
619       if (rec->Stream == STREAM_SPARSE_DATA) {
620          mjcr->JobBytes -= sizeof(uint64_t);
621       }
622          
623       free_jcr(mjcr);                 /* done using JCR */
624       break;
625
626    case STREAM_GZIP_DATA:
627       mjcr->JobBytes += rec->data_len; /* No correct, we should expand it */
628       free_jcr(mjcr);                 /* done using JCR */
629       break;
630
631    case STREAM_SPARSE_GZIP_DATA:
632       mjcr->JobBytes += rec->data_len - sizeof(uint64_t); /* No correct, we should expand it */
633       free_jcr(mjcr);                 /* done using JCR */
634       break;
635
636    /* Win32 GZIP stream */
637    case STREAM_WIN32_GZIP_DATA:
638       mjcr->JobBytes += rec->data_len;
639       free_jcr(mjcr);                 /* done using JCR */
640       break;
641
642    case STREAM_MD5_SIGNATURE:
643       char MD5buf[50];
644       bin_to_base64(MD5buf, (char *)rec->data, 16); /* encode 16 bytes */
645       if (verbose > 1) {
646          Pmsg1(000, _("Got MD5 record: %s\n"), MD5buf);
647       }
648       update_SIG_record(db, MD5buf, rec, MD5_SIG);
649       break;
650
651    case STREAM_SHA1_SIGNATURE:
652       char SIGbuf[50];
653       bin_to_base64(SIGbuf, (char *)rec->data, 20); /* encode 20 bytes */
654       if (verbose > 1) {
655          Pmsg1(000, _("Got SHA1 record: %s\n"), SIGbuf);
656       }
657       update_SIG_record(db, SIGbuf, rec, SHA1_SIG);
658       break;
659
660
661    case STREAM_PROGRAM_NAMES:
662       if (verbose) {
663          Pmsg1(000, _("Got Prog Names Stream: %s\n"), rec->data);
664       }
665       break;
666
667    case STREAM_PROGRAM_DATA:
668       if (verbose > 1) {
669          Pmsg0(000, _("Got Prog Data Stream record.\n"));
670       }
671       break;
672    default:
673       Pmsg2(0, _("Unknown stream type!!! stream=%d data=%s\n"), rec->Stream, rec->data);
674       break;
675    }
676    return 1;
677 }
678
679 /*
680  * Free the Job Control Record if no one is still using it.
681  *  Called from main free_jcr() routine in src/lib/jcr.c so
682  *  that we can do our Director specific cleanup of the jcr.
683  */
684 static void bscan_free_jcr(JCR *jcr)
685 {
686    Dmsg0(200, "Start dird free_jcr\n");
687
688    if (jcr->file_bsock) {
689       Dmsg0(200, "Close File bsock\n");
690       bnet_close(jcr->file_bsock);
691    }
692    if (jcr->store_bsock) {
693       Dmsg0(200, "Close Store bsock\n");
694       bnet_close(jcr->store_bsock);
695    }
696    if (jcr->RestoreBootstrap) {
697       free(jcr->RestoreBootstrap);
698    }
699    if (jcr->dcr) {
700       free_dcr(jcr->dcr);
701       jcr->dcr = NULL;
702    }
703    Dmsg0(200, "End dird free_jcr\n");
704 }
705
706 /*
707  * We got a File Attributes record on the tape.  Now, lookup the Job
708  *   record, and then create the attributes record.
709  */
710 static int create_file_attributes_record(B_DB *db, JCR *mjcr,
711                                char *fname, char *lname, int type,
712                                char *ap, DEV_RECORD *rec)
713 {
714    DCR *dcr = mjcr->dcr;
715    ar.fname = fname;
716    ar.link = lname;
717    ar.ClientId = mjcr->ClientId;
718    ar.JobId = mjcr->JobId;
719    ar.Stream = rec->Stream;
720    ar.FileIndex = rec->FileIndex;
721    ar.attr = ap;
722    if (dcr->VolFirstIndex == 0) {
723       dcr->VolFirstIndex = rec->FileIndex;
724    }
725    dcr->FileIndex = rec->FileIndex;
726    mjcr->JobFiles++;
727
728    if (!update_db) {
729       return 1;
730    }
731
732    if (!db_create_file_attributes_record(bjcr, db, &ar)) {
733       Pmsg1(0, _("Could not create File Attributes record. ERR=%s\n"), db_strerror(db));
734       return 0;
735    }
736    mjcr->FileId = ar.FileId;
737
738    if (verbose > 1) {
739       Pmsg1(000, _("Created File record: %s\n"), fname);   
740    }
741    return 1;
742 }
743
744 /*
745  * For each Volume we see, we create a Medium record
746  */
747 static int create_media_record(B_DB *db, MEDIA_DBR *mr, VOLUME_LABEL *vl)
748 {
749    struct date_time dt;
750    struct tm tm;
751
752    /* We mark Vols as Archive to keep them from being re-written */
753    bstrncpy(mr->VolStatus, "Archive", sizeof(mr->VolStatus));
754    mr->VolRetention = 365 * 3600 * 24; /* 1 year */
755    if (vl->VerNum >= 11) {
756       mr->FirstWritten = btime_to_utime(vl->write_btime);
757       mr->LabelDate    = btime_to_utime(vl->label_btime);
758    } else {
759       /* DEPRECATED DO NOT USE */
760       dt.julian_day_number = vl->write_date;
761       dt.julian_day_fraction = vl->write_time;
762       tm_decode(&dt, &tm);
763       mr->FirstWritten = mktime(&tm);
764       dt.julian_day_number = vl->label_date;
765       dt.julian_day_fraction = vl->label_time;
766       tm_decode(&dt, &tm);
767       mr->LabelDate = mktime(&tm);
768    }
769    lasttime = mr->LabelDate;
770
771    if (!update_db) {
772       return 1;
773    }
774
775    if (!db_create_media_record(bjcr, db, mr)) {
776       Pmsg1(0, _("Could not create media record. ERR=%s\n"), db_strerror(db));
777       return 0;
778    }
779    if (!db_update_media_record(bjcr, db, mr)) {
780       Pmsg1(0, _("Could not update media record. ERR=%s\n"), db_strerror(db));
781       return 0;
782    }
783    if (verbose) {
784       Pmsg1(000, _("Created Media record for Volume: %s\n"), mr->VolumeName);
785    }
786    return 1;
787
788 }
789
790 /*
791  * Called at end of media to update it
792  */
793 static int update_media_record(B_DB *db, MEDIA_DBR *mr)
794 {
795    if (!update_db && !update_vol_info) {
796       return 1;
797    }
798
799    mr->LastWritten = lasttime;
800    if (!db_update_media_record(bjcr, db, mr)) {
801       Pmsg1(0, _("Could not update media record. ERR=%s\n"), db_strerror(db));
802       return 0;
803    }
804    if (verbose) {
805       Pmsg1(000, _("Updated Media record at end of Volume: %s\n"), mr->VolumeName);
806    }
807    return 1;
808
809 }
810
811
812 static int create_pool_record(B_DB *db, POOL_DBR *pr)
813 {
814    pr->NumVols++;
815    pr->UseCatalog = 1;
816    pr->VolRetention = 355 * 3600 * 24; /* 1 year */
817
818    if (!update_db) {
819       return 1;
820    }
821    if (!db_create_pool_record(bjcr, db, pr)) {
822       Pmsg1(0, _("Could not create pool record. ERR=%s\n"), db_strerror(db));
823       return 0;
824    }
825    if (verbose) {
826       Pmsg1(000, _("Created Pool record for Pool: %s\n"), pr->Name);
827    }
828    return 1;
829
830 }
831
832
833 /*
834  * Called from SOS to create a client for the current Job 
835  */
836 static int create_client_record(B_DB *db, CLIENT_DBR *cr)
837 {
838    if (!update_db) {
839       return 1;
840    }
841    if (!db_create_client_record(bjcr, db, cr)) {
842       Pmsg1(0, _("Could not create Client record. ERR=%s\n"), db_strerror(db));
843       return 0;
844    }
845    if (verbose) {
846       Pmsg1(000, _("Created Client record for Client: %s\n"), cr->Name);
847    }
848    return 1;
849 }
850
851 static int create_fileset_record(B_DB *db, FILESET_DBR *fsr)
852 {
853    if (!update_db) {
854       return 1;
855    }
856    fsr->FileSetId = 0;
857    if (fsr->MD5[0] == 0) {
858       fsr->MD5[0] = ' ';              /* Equivalent to nothing */
859       fsr->MD5[1] = 0;
860    }
861    if (db_get_fileset_record(bjcr, db, fsr)) {
862       if (verbose) {
863          Pmsg1(000, _("Fileset \"%s\" already exists.\n"), fsr->FileSet);
864       }
865    } else {
866       if (!db_create_fileset_record(bjcr, db, fsr)) {
867          Pmsg2(0, _("Could not create FileSet record \"%s\". ERR=%s\n"), 
868             fsr->FileSet, db_strerror(db));
869          return 0;
870       }
871       if (verbose) {
872          Pmsg1(000, _("Created FileSet record \"%s\"\n"), fsr->FileSet);
873       }
874    }
875    return 1;
876 }
877
878 /*
879  * Simulate the two calls on the database to create
880  *  the Job record and to update it when the Job actually
881  *  begins running.
882  */
883 static JCR *create_job_record(B_DB *db, JOB_DBR *jr, SESSION_LABEL *label, 
884                              DEV_RECORD *rec)
885 {
886    JCR *mjcr;
887    struct date_time dt;
888    struct tm tm;
889
890    jr->JobId = label->JobId;
891    jr->Type = label->JobType;
892    jr->Level = label->JobLevel;
893    jr->JobStatus = JS_Created;
894    bstrncpy(jr->Name, label->JobName, sizeof(jr->Name));
895    bstrncpy(jr->Job, label->Job, sizeof(jr->Job));
896    if (label->VerNum >= 11) {
897       jr->SchedTime = btime_to_unix(label->write_btime);
898    } else {
899       dt.julian_day_number = label->write_date;
900       dt.julian_day_fraction = label->write_time;
901       tm_decode(&dt, &tm);
902       jr->SchedTime = mktime(&tm);
903    }
904
905    jr->StartTime = jr->SchedTime;
906    jr->JobTDate = (utime_t)jr->SchedTime;
907    jr->VolSessionId = rec->VolSessionId;
908    jr->VolSessionTime = rec->VolSessionTime;
909
910    /* Now create a JCR as if starting the Job */
911    mjcr = create_jcr(jr, rec, label->JobId);
912
913    if (!update_db) {
914       return mjcr;
915    }
916
917    /* This creates the bare essentials */
918    if (!db_create_job_record(bjcr, db, jr)) {
919       Pmsg1(0, _("Could not create JobId record. ERR=%s\n"), db_strerror(db));
920       return mjcr;
921    }
922
923    /* This adds the client, StartTime, JobTDate, ... */
924    if (!db_update_job_start_record(bjcr, db, jr)) {
925       Pmsg1(0, _("Could not update job start record. ERR=%s\n"), db_strerror(db));
926       return mjcr;
927    }
928    Pmsg2(000, _("Created new JobId=%u record for original JobId=%u\n"), jr->JobId, 
929          label->JobId);
930    mjcr->JobId = jr->JobId;           /* set new JobId */
931    return mjcr;
932 }
933
934 /* 
935  * Simulate the database call that updates the Job 
936  *  at Job termination time.
937  */
938 static int update_job_record(B_DB *db, JOB_DBR *jr, SESSION_LABEL *elabel,
939                               DEV_RECORD *rec)
940 {
941    struct date_time dt;
942    struct tm tm;
943    JCR *mjcr;
944
945    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
946    if (!mjcr) {
947       Pmsg2(000, _("Could not find SessId=%d SessTime=%d for EOS record.\n"),
948                    rec->VolSessionId, rec->VolSessionTime);
949       return 0;
950    }
951    if (elabel->VerNum >= 11) {
952       jr->EndTime = btime_to_unix(elabel->write_btime);
953    } else {
954       dt.julian_day_number = elabel->write_date;
955       dt.julian_day_fraction = elabel->write_time;
956       tm_decode(&dt, &tm);
957       jr->EndTime = mktime(&tm);
958    }
959    lasttime = jr->EndTime;
960    mjcr->end_time = jr->EndTime;
961
962    jr->JobId = mjcr->JobId;
963    jr->JobStatus = elabel->JobStatus;
964    mjcr->JobStatus = elabel->JobStatus;
965    jr->JobFiles = elabel->JobFiles;
966    jr->JobBytes = elabel->JobBytes;
967    jr->VolSessionId = rec->VolSessionId;
968    jr->VolSessionTime = rec->VolSessionTime;
969    jr->JobTDate = (utime_t)mjcr->start_time;
970    jr->ClientId = mjcr->ClientId;
971
972    if (!update_db) {
973       free_jcr(mjcr);
974       return 1;
975    }
976    
977    if (!db_update_job_end_record(bjcr, db, jr)) {
978       Pmsg2(0, _("Could not update JobId=%u record. ERR=%s\n"), jr->JobId,  db_strerror(db));
979       free_jcr(mjcr);
980       return 0;
981    }
982    if (verbose) {
983       Pmsg1(000, _("Updated Job termination record for new JobId=%u\n"), jr->JobId);
984    }
985    if (verbose > 1) {
986       char *term_msg;
987       static char term_code[70];
988       char sdt[50], edt[50];
989       char ec1[30], ec2[30], ec3[30];
990
991       switch (mjcr->JobStatus) {
992       case JS_Terminated:
993          term_msg = _("Backup OK");
994          break;
995       case JS_FatalError:
996       case JS_ErrorTerminated:
997          term_msg = _("*** Backup Error ***");
998          break;
999       case JS_Canceled:
1000          term_msg = _("Backup Canceled");
1001          break;
1002       default:
1003          term_msg = term_code;
1004          sprintf(term_code, _("Job Termination code: %d"), mjcr->JobStatus);
1005          break;
1006       }
1007       bstrftime(sdt, sizeof(sdt), mjcr->start_time);
1008       bstrftime(edt, sizeof(edt), mjcr->end_time);
1009       Pmsg14(000,  _("%s\n\
1010 JobId:                  %d\n\
1011 Job:                    %s\n\
1012 FileSet:                %s\n\
1013 Backup Level:           %s\n\
1014 Client:                 %s\n\
1015 Start time:             %s\n\
1016 End time:               %s\n\
1017 Files Written:          %s\n\
1018 Bytes Written:          %s\n\
1019 Volume Session Id:      %d\n\
1020 Volume Session Time:    %d\n\
1021 Last Volume Bytes:      %s\n\
1022 Termination:            %s\n\n"),
1023         edt,
1024         mjcr->JobId,
1025         mjcr->Job,
1026         mjcr->fileset_name,
1027         job_level_to_str(mjcr->JobLevel),
1028         mjcr->client_name,
1029         sdt,
1030         edt,
1031         edit_uint64_with_commas(mjcr->JobFiles, ec1),
1032         edit_uint64_with_commas(mjcr->JobBytes, ec2),
1033         mjcr->VolSessionId,
1034         mjcr->VolSessionTime,
1035         edit_uint64_with_commas(mr.VolBytes, ec3),
1036         term_msg);
1037    }
1038    free_jcr(mjcr);
1039    return 1;
1040 }
1041
1042 static int create_jobmedia_record(B_DB *db, JCR *mjcr)
1043 {
1044    JOBMEDIA_DBR jmr;
1045    DCR *dcr = mjcr->dcr;
1046
1047    if (dev->state & ST_TAPE) {
1048       dcr->EndBlock = dev->EndBlock;
1049       dcr->EndFile  = dev->EndFile;
1050    } else {
1051       dcr->EndBlock = (uint32_t)dev->file_addr;
1052       dcr->EndFile = (uint32_t)(dev->file_addr >> 32);
1053    }
1054
1055    memset(&jmr, 0, sizeof(jmr));
1056    jmr.JobId = mjcr->JobId;
1057    jmr.MediaId = mr.MediaId;
1058    jmr.FirstIndex = dcr->VolFirstIndex;
1059    jmr.LastIndex = dcr->FileIndex;
1060    jmr.StartFile = dcr->StartFile;
1061    jmr.EndFile = dcr->EndFile;
1062    jmr.StartBlock = dcr->StartBlock;
1063    jmr.EndBlock = dcr->EndBlock;
1064
1065
1066    if (!update_db) {
1067       return 1;
1068    }
1069
1070    if (!db_create_jobmedia_record(bjcr, db, &jmr)) {
1071       Pmsg1(0, _("Could not create JobMedia record. ERR=%s\n"), db_strerror(db));
1072       return 0;
1073    }
1074    if (verbose) {
1075       Pmsg2(000, _("Created JobMedia record JobId %d, MediaId %d\n"), 
1076                 jmr.JobId, jmr.MediaId);
1077    }
1078    return 1;
1079 }
1080
1081 /* 
1082  * Simulate the database call that updates the MD5/SHA1 record
1083  */
1084 static int update_SIG_record(B_DB *db, char *SIGbuf, DEV_RECORD *rec, int type)
1085 {
1086    JCR *mjcr;
1087
1088    mjcr = get_jcr_by_session(rec->VolSessionId, rec->VolSessionTime);
1089    if (!mjcr) {
1090       if (mr.VolJobs > 0) {
1091          Pmsg2(000, _("Could not find SessId=%d SessTime=%d for MD5/SHA1 record.\n"),
1092                       rec->VolSessionId, rec->VolSessionTime);
1093       } else {
1094          ignored_msgs++;
1095       }
1096       return 0;
1097    }
1098
1099    if (!update_db) {
1100       free_jcr(mjcr);
1101       return 1;
1102    }
1103    
1104    if (!db_add_SIG_to_file_record(bjcr, db, mjcr->FileId, SIGbuf, type)) {
1105       Pmsg1(0, _("Could not add MD5/SHA1 to File record. ERR=%s\n"), db_strerror(db));
1106       free_jcr(mjcr);
1107       return 0;
1108    }
1109    if (verbose > 1) {
1110       Pmsg0(000, _("Updated MD5/SHA1 record\n"));
1111    }
1112    free_jcr(mjcr);
1113    return 1;
1114 }
1115
1116
1117 /* 
1118  * Create a JCR as if we are really starting the job
1119  */
1120 static JCR *create_jcr(JOB_DBR *jr, DEV_RECORD *rec, uint32_t JobId)
1121 {
1122    JCR *jobjcr;
1123    /*
1124     * Transfer as much as possible to the Job JCR. Most important is
1125     *   the JobId and the ClientId.
1126     */
1127    jobjcr = new_jcr(sizeof(JCR), bscan_free_jcr);
1128    jobjcr->JobType = jr->Type;
1129    jobjcr->JobLevel = jr->Level;
1130    jobjcr->JobStatus = jr->JobStatus;
1131    bstrncpy(jobjcr->Job, jr->Job, sizeof(jobjcr->Job));
1132    jobjcr->JobId = JobId;      /* this is JobId on tape */
1133    jobjcr->sched_time = jr->SchedTime;
1134    jobjcr->start_time = jr->StartTime;
1135    jobjcr->VolSessionId = rec->VolSessionId;
1136    jobjcr->VolSessionTime = rec->VolSessionTime;
1137    jobjcr->ClientId = jr->ClientId;
1138    attach_jcr_to_device(dev, jobjcr);
1139    new_dcr(jobjcr, dev);
1140    return jobjcr;
1141 }
1142
1143 /* Dummies to replace askdir.c */
1144 int     dir_get_volume_info(JCR *jcr, enum get_vol_info_rw  writing) { return 1;}
1145 int     dir_find_next_appendable_volume(JCR *jcr) { return 1;}
1146 int     dir_update_volume_info(JCR *jcr, DEVICE *dev, int relabel) { return 1; }
1147 int     dir_create_jobmedia_record(JCR *jcr) { return 1; }
1148 int     dir_ask_sysop_to_create_appendable_volume(JCR *jcr, DEVICE *dev) { return 1; }
1149 int     dir_update_file_attributes(JCR *jcr, DEV_RECORD *rec) { return 1;}
1150 int     dir_send_job_status(JCR *jcr) {return 1;}
1151
1152
1153 int dir_ask_sysop_to_mount_volume(JCR *jcr, DEVICE *dev)
1154 {
1155    fprintf(stderr, _("Mount Volume \"%s\" on device \"%s\" and press return when ready: "),
1156       jcr->VolumeName, dev_name(dev));
1157    getchar();   
1158    return 1;
1159 }