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