]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/dird/backup.c
094c2b9a5043de5ec49ee4850581ce62c3bb9835
[bacula/bacula] / bacula / src / dird / backup.c
1 /*
2  *
3  *   Bacula Director -- backup.c -- responsible for doing backup jobs
4  *
5  *     Kern Sibbald, March MM
6  *
7  *    This routine is called as a thread. It may not yet be totally
8  *      thread reentrant!!!
9  *
10  *  Basic tasks done here:
11  *     Open DB and create records for this job.
12  *     Open Message Channel with Storage daemon to tell him a job will be starting.
13  *     Open connection with File daemon and pass him commands
14  *       to do the backup.
15  *     When the File daemon finishes the job, update the DB.
16  *
17  *   Version $Id$
18  */
19
20 /*
21    Copyright (C) 2000, 2001, 2002 Kern Sibbald and John Walker
22
23    This program is free software; you can redistribute it and/or
24    modify it under the terms of the GNU General Public License as
25    published by the Free Software Foundation; either version 2 of
26    the License, or (at your option) any later version.
27
28    This program is distributed in the hope that it will be useful,
29    but WITHOUT ANY WARRANTY; without even the implied warranty of
30    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
31    General Public License for more details.
32
33    You should have received a copy of the GNU General Public
34    License along with this program; if not, write to the Free
35    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
36    MA 02111-1307, USA.
37
38  */
39
40 #include "bacula.h"
41 #include "dird.h"
42 #include "ua.h"
43
44 /* Commands sent to File daemon */
45 static char backupcmd[] = "backup\n";
46 static char storaddr[]  = "storage address=%s port=%d\n";
47 static char levelcmd[]  = "level = %s%s\n";
48
49 /* Responses received from File daemon */
50 static char OKbackup[]  = "2000 OK backup\n";
51 static char OKstore[]   = "2000 OK storage\n";
52 static char OKlevel[]   = "2000 OK level\n";
53 static char EndBackup[] = "2801 End Backup Job TermCode=%d JobFiles=%u ReadBytes=%" lld " JobBytes=%" lld "\n";
54
55
56 /* Forward referenced functions */
57 static void backup_cleanup(JCR *jcr, int TermCode, char *since);
58 static int wait_for_job_termination(JCR *jcr);               
59
60 /* External functions */
61
62 /* 
63  * Do a backup of the specified FileSet
64  *    
65  *  Returns:  0 on failure
66  *            1 on success
67  */
68 int do_backup(JCR *jcr) 
69 {
70    char since[MAXSTRING];
71    int stat;
72    BSOCK   *fd;
73    POOL_DBR pr;
74    FILESET_DBR fsr;
75
76    since[0] = 0;
77
78    if (!get_or_create_client_record(jcr)) {
79       Jmsg(jcr, M_ERROR, 0, _("Could not get/create Client record. ERR=%s\n"), 
80          db_strerror(jcr->db));
81       goto bail_out;
82    }
83
84    /*
85     * Get or Create FileSet record
86     */
87    memset(&fsr, 0, sizeof(fsr));
88    strcpy(fsr.FileSet, jcr->fileset->hdr.name);
89    if (jcr->fileset->have_MD5) {
90       struct MD5Context md5c;
91       unsigned char signature[16];
92       memcpy(&md5c, &jcr->fileset->md5c, sizeof(md5c));
93       MD5Final(signature, &md5c);
94       bin_to_base64(fsr.MD5, (char *)signature, 16); /* encode 16 bytes */
95       strcpy(jcr->fileset->MD5, fsr.MD5);
96    } else {
97       Jmsg(jcr, M_WARNING, 0, _("FileSet MD5 signature not found.\n"));
98    }
99    if (!db_create_fileset_record(jcr->db, &fsr)) {
100       Jmsg(jcr, M_ERROR, 0, _("Could not create FileSet record. ERR=%s\n"), 
101          db_strerror(jcr->db));
102       goto bail_out;
103    }   
104    jcr->jr.FileSetId = fsr.FileSetId;
105    Dmsg2(119, "Created FileSet %s record %d\n", jcr->fileset->hdr.name, 
106       jcr->jr.FileSetId);
107
108    /* Look up the last
109     * FULL backup job to get the time/date for a 
110     * differential or incremental save.
111     */
112    jcr->stime = get_pool_memory(PM_MESSAGE);
113    jcr->stime[0] = 0;
114    since[0] = 0;
115    switch (jcr->JobLevel) {
116       case L_DIFFERENTIAL:
117       case L_INCREMENTAL:
118          /* Look up start time of last job */
119          jcr->jr.JobId = 0;
120          if (!db_find_job_start_time(jcr->db, &jcr->jr, jcr->stime)) {
121             Jmsg(jcr, M_INFO, 0, _("Last FULL backup time not found. Doing FULL backup.\n"));
122             jcr->JobLevel = jcr->jr.Level = L_FULL;
123          } else {
124             strcpy(since, ", since=");
125             strcat(since, jcr->stime);
126          }
127          Dmsg1(115, "Last start time = %s\n", jcr->stime);
128          break;
129    }
130
131    jcr->jr.JobId = jcr->JobId;
132    jcr->jr.StartTime = jcr->start_time;
133    if (!db_update_job_start_record(jcr->db, &jcr->jr)) {
134       Jmsg(jcr, M_ERROR, 0, "%s", db_strerror(jcr->db));
135       goto bail_out;
136    }
137
138    jcr->fname = (char *) get_pool_memory(PM_FNAME);
139
140    /* Print Job Start message */
141    Jmsg(jcr, M_INFO, 0, _("Start Backup JobId %d, Job=%s\n"),
142         jcr->JobId, jcr->Job);
143
144    /* 
145     * Get the Pool record  
146     */
147    memset(&pr, 0, sizeof(pr));
148    strcpy(pr.Name, jcr->pool->hdr.name);
149    while (!db_get_pool_record(jcr->db, &pr)) { /* get by Name */
150       /* Try to create the pool */
151       if (create_pool(jcr->db, jcr->pool) < 0) {
152          Jmsg(jcr, M_FATAL, 0, _("Pool %s not in database. %s"), pr.Name, 
153             db_strerror(jcr->db));
154          goto bail_out;
155       } else {
156          Jmsg(jcr, M_INFO, 0, _("Pool %s created in database.\n"), pr.Name);
157       }
158    }
159    jcr->PoolId = pr.PoolId;               /****FIXME**** this can go away */
160    jcr->jr.PoolId = pr.PoolId;
161
162    /*
163     * Open a message channel connection with the Storage
164     * daemon. This is to let him know that our client
165     * will be contacting him for a backup  session.
166     *
167     */
168    Dmsg0(110, "Open connection with storage daemon\n");
169    jcr->JobStatus = JS_Blocked;
170    /*
171     * Start conversation with Storage daemon  
172     */
173    if (!connect_to_storage_daemon(jcr, 10, SDConnectTimeout, 1)) {
174       goto bail_out;
175    }
176    /*
177     * Now start a job with the Storage daemon
178     */
179    if (!start_storage_daemon_job(jcr)) {
180       goto bail_out;
181    }
182    /*
183     * Now start a Storage daemon message thread
184     */
185    if (!start_storage_daemon_message_thread(jcr)) {
186       goto bail_out;
187    }
188    Dmsg0(150, "Storage daemon connection OK\n");
189
190    jcr->JobStatus = JS_Blocked;
191    if (!connect_to_file_daemon(jcr, 10, FDConnectTimeout, 1)) {
192       goto bail_out;
193    }
194
195    jcr->JobStatus = JS_Running;
196    fd = jcr->file_bsock;
197
198    if (!send_include_list(jcr)) {
199       goto bail_out;
200    }
201
202    if (!send_exclude_list(jcr)) {
203       goto bail_out;
204    }
205
206    /* 
207     * send Storage daemon address to the File daemon
208     */
209    if (jcr->store->SDDport == 0) {
210       jcr->store->SDDport = jcr->store->SDport;
211    }
212    bnet_fsend(fd, storaddr, jcr->store->address, jcr->store->SDDport);
213    if (!response(fd, OKstore, "Storage")) {
214       goto bail_out;
215    }
216
217    /* 
218     * Send Level command to File daemon
219     */
220    switch (jcr->JobLevel) {
221       case L_FULL:
222          bnet_fsend(fd, levelcmd, "full", " ");
223          break;
224       case L_DIFFERENTIAL:
225       case L_INCREMENTAL:
226          bnet_fsend(fd, levelcmd, "since ", jcr->stime);
227          free_pool_memory(jcr->stime);
228          jcr->stime = NULL;
229          break;
230       case L_SINCE:
231       default:
232          Jmsg2(jcr, M_FATAL, 0, _("Unimplemented backup level %d %c\n"), 
233             jcr->JobLevel, jcr->JobLevel);
234          goto bail_out;
235    }
236    Dmsg1(120, ">filed: %s", fd->msg);
237    if (!response(fd, OKlevel, "Level")) {
238       goto bail_out;
239    }
240
241    /* Send backup command */
242    bnet_fsend(fd, backupcmd);
243    if (!response(fd, OKbackup, "backup")) {
244       goto bail_out;
245    }
246
247    /* Pickup Job termination data */        
248    stat = wait_for_job_termination(jcr);
249    backup_cleanup(jcr, stat, since);
250    return 1;
251
252 bail_out:
253    if (jcr->stime) {
254       free_pool_memory(jcr->stime);
255       jcr->stime = NULL;
256    }
257    backup_cleanup(jcr, JS_ErrorTerminated, since);
258    return 0;
259
260 }
261
262 /*
263  * Here we wait for the File daemon to signal termination,
264  *   then we wait for the Storage daemon.  When both
265  *   are done, we return the job status.
266  */
267 static int wait_for_job_termination(JCR *jcr)
268 {
269    int32_t n = 0;
270    BSOCK *fd = jcr->file_bsock;
271    int fd_ok = FALSE;
272
273    jcr->JobStatus = JS_WaitFD;
274    /* Wait for Client to terminate */
275    while ((n = bget_msg(fd, 0)) > 0 && !job_cancelled(jcr)) {
276       if (sscanf(fd->msg, EndBackup, &jcr->JobStatus, &jcr->JobFiles,
277           &jcr->ReadBytes, &jcr->JobBytes) == 4) {
278          fd_ok = TRUE;
279       }
280    }
281    bnet_sig(fd, BNET_TERMINATE);      /* tell Client we are terminating */
282    if (n < 0) {
283       Jmsg(jcr, M_FATAL, 0, _("<filed: network error during BACKUP command. ERR=%s\n"),
284           bnet_strerror(fd));
285    }
286
287    wait_for_storage_daemon_termination(jcr);
288
289    /* Return the first error status we find FD or SD */
290    if (fd_ok && jcr->JobStatus != JS_Terminated) {
291       return jcr->JobStatus;
292    }
293    if (!fd_ok || n < 0) {                                     
294       return JS_ErrorTerminated;
295    }
296    return jcr->SDJobStatus;
297 }
298
299 /*
300  * Release resources allocated during backup.
301  */
302 static void backup_cleanup(JCR *jcr, int TermCode, char *since)
303 {
304    char sdt[50], edt[50];
305    char ec1[30], ec2[30], ec3[30], compress[50];
306    char term_code[100];
307    char *term_msg;
308    int msg_type;
309    MEDIA_DBR mr;
310    double kbps, compression;
311    btime_t RunTime;
312
313    Dmsg0(100, "Enter backup_cleanup()\n");
314    memset(&mr, 0, sizeof(mr));
315    jcr->JobStatus = TermCode;
316
317    update_job_end_record(jcr);        /* update database */
318    
319    if (!db_get_job_record(jcr->db, &jcr->jr)) {
320       Jmsg(jcr, M_WARNING, 0, _("Error getting job record for stats: %s"), 
321          db_strerror(jcr->db));
322    }
323
324    strcpy(mr.VolumeName, jcr->VolumeName);
325    if (!db_get_media_record(jcr->db, &mr)) {
326       Jmsg(jcr, M_WARNING, 0, _("Error getting Media record for stats: %s"), 
327          db_strerror(jcr->db));
328    }
329
330    msg_type = M_INFO;                 /* by default INFO message */
331    switch (TermCode) {
332       case JS_Terminated:
333          term_msg = _("Backup OK");
334          break;
335       case JS_FatalError:
336       case JS_ErrorTerminated:
337          term_msg = _("*** Backup Error ***"); 
338          msg_type = M_ERROR;          /* Generate error message */
339          if (jcr->store_bsock) {
340             bnet_sig(jcr->store_bsock, BNET_TERMINATE);
341             pthread_cancel(jcr->SD_msg_chan);
342          }
343          break;
344       case JS_Cancelled:
345          term_msg = _("Backup Cancelled");
346          if (jcr->store_bsock) {
347             bnet_sig(jcr->store_bsock, BNET_TERMINATE);
348             pthread_cancel(jcr->SD_msg_chan);
349          }
350          break;
351       default:
352          term_msg = term_code;
353          sprintf(term_code, _("Inappropriate term code: %c\n"), TermCode);
354          break;
355    }
356    bstrftime(sdt, sizeof(sdt), jcr->jr.StartTime);
357    bstrftime(edt, sizeof(edt), jcr->jr.EndTime);
358    RunTime = jcr->jr.EndTime - jcr->jr.StartTime;
359    if (RunTime <= 0) {
360       kbps = 0;
361    } else {
362       kbps = (double)jcr->jr.JobBytes / (1000 * RunTime);
363    }
364    if (!db_get_job_volume_names(jcr->db, jcr->jr.JobId, &jcr->VolumeName)) {
365       /*
366        * Note, if the job has erred, most likely it did not write any
367        *  tape, so suppress this "error" message since in that case
368        *  it is normal.  Or look at it the other way, only for a
369        *  normal exit should we complain about this error.
370        */
371       if (TermCode == JS_Terminated) {                                
372          Jmsg(jcr, M_ERROR, 0, "%s", db_strerror(jcr->db));
373       }
374       jcr->VolumeName[0] = 0;         /* none */
375    }
376
377    if (jcr->ReadBytes == 0) {
378       strcpy(compress, "None");
379    } else {
380       compression = (double)100 - 100.0 * ((double)jcr->JobBytes / (double)jcr->ReadBytes);
381       if (compression < 0.5) {
382          strcpy(compress, "None");
383       } else {
384          sprintf(compress, "%.1f %%", (float)compression);
385       }
386    }
387
388    Jmsg(jcr, msg_type, 0, _("%s\n\
389 JobId:                  %d\n\
390 Job:                    %s\n\
391 FileSet:                %s\n\
392 Backup Level:           %s%s\n\
393 Client:                 %s\n\
394 Start time:             %s\n\
395 End time:               %s\n\
396 Files Written:          %s\n\
397 Bytes Written:          %s\n\
398 Rate:                   %.1f KB/s\n\
399 Software Compression:   %s\n\
400 Volume names(s):        %s\n\
401 Volume Session Id:      %d\n\
402 Volume Session Time:    %d\n\
403 Last Volume Bytes:      %s\n\
404 Termination:            %s\n\n"),
405         edt,
406         jcr->jr.JobId,
407         jcr->jr.Job,
408         jcr->fileset->hdr.name,
409         level_to_str(jcr->JobLevel), since,
410         jcr->client->hdr.name,
411         sdt,
412         edt,
413         edit_uint64_with_commas(jcr->jr.JobFiles, ec1),
414         edit_uint64_with_commas(jcr->jr.JobBytes, ec2),
415         (float)kbps,
416         compress,
417         jcr->VolumeName,
418         jcr->VolSessionId,
419         jcr->VolSessionTime,
420         edit_uint64_with_commas(mr.VolBytes, ec3),
421         term_msg);
422
423
424    /* Now update the bootstrap file if any */
425    if (jcr->job->WriteBootstrap) {
426       FILE *fd;
427       int got_pipe = 0;
428       char *fname = jcr->job->WriteBootstrap;
429
430       if (*fname == '|') {
431          fname++;
432          got_pipe = 1;
433          fd = popen(fname, "w");
434       } else {
435          fd = fopen(fname, jcr->JobLevel==L_FULL?"w+":"a+");
436       }
437       if (!fd) {
438          Jmsg(jcr, M_ERROR, 0, _("Could not open WriteBootstrap file:\n"
439               "%s: ERR=%s\n"), fname, strerror(errno));
440          return;
441       }
442       fprintf(fd, "Volume=\"%s\"\n", jcr->VolumeName);
443       fprintf(fd, "VolSessionId=%u\n", jcr->VolSessionId);
444       fprintf(fd, "VolSessionTime=%u\n", jcr->VolSessionTime);
445       if (got_pipe) {
446          pclose(fd);
447       } else {
448          fclose(fd);
449       }
450    }
451    Dmsg0(100, "Leave backup_cleanup()\n");
452 }