]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/stored/askdir.c
Use rentrant mysql lib, eliminate race in sql_list, Win32 streams, misc see kes-1.31
[bacula/bacula] / bacula / src / stored / askdir.c
1 /*
2  *  Subroutines to handle Catalog reqests sent to the Director
3  *   Reqests/commands from the Director are handled in dircmd.c
4  *
5  *   Kern Sibbald, December 2000
6  *
7  *   Version $Id$
8  */
9 /*
10    Copyright (C) 2000-2003 Kern Sibbald and John Walker
11
12    This program is free software; you can redistribute it and/or
13    modify it under the terms of the GNU General Public License as
14    published by the Free Software Foundation; either version 2 of
15    the License, or (at your option) any later version.
16
17    This program is distributed in the hope that it will be useful,
18    but WITHOUT ANY WARRANTY; without even the implied warranty of
19    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
20    General Public License for more details.
21
22    You should have received a copy of the GNU General Public
23    License along with this program; if not, write to the Free
24    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
25    MA 02111-1307, USA.
26
27  */
28
29 #include "bacula.h"                   /* pull in global headers */
30 #include "stored.h"                   /* pull in Storage Deamon headers */
31
32 /* Requests sent to the Director */
33 static char Find_media[]   = "CatReq Job=%s FindMedia=%d\n";
34 static char Get_Vol_Info[] = "CatReq Job=%s GetVolInfo VolName=%s write=%d\n";
35 static char Update_media[] = "CatReq Job=%s UpdateMedia VolName=%s\
36  VolJobs=%u VolFiles=%u VolBlocks=%u VolBytes=%s VolMounts=%u\
37  VolErrors=%u VolWrites=%u MaxVolBytes=%s EndTime=%d VolStatus=%s\
38  Slot=%d relabel=%d\n";
39
40 static char Create_job_media[] = "CatReq Job=%s CreateJobMedia \
41  FirstIndex=%u LastIndex=%u StartFile=%u EndFile=%u \
42  StartBlock=%u EndBlock=%u\n";
43 static char FileAttributes[] = "UpdCat Job=%s FileAttributes ";
44 static char Job_status[]     = "3012 Job %s jobstatus %d\n";
45
46
47 /* Responses received from the Director */
48 static char OK_media[] = "1000 OK VolName=%127s VolJobs=%u VolFiles=%u\
49  VolBlocks=%u VolBytes=%" lld " VolMounts=%u VolErrors=%u VolWrites=%u\
50  MaxVolBytes=%" lld " VolCapacityBytes=%" lld " VolStatus=%20s\
51  Slot=%d MaxVolJobs=%u MaxVolFiles=%u\n";
52 static char OK_update[] = "1000 OK UpdateMedia\n";
53
54 /* Forward referenced functions */
55 static int wait_for_sysop(JCR *jcr, DEVICE *dev, int wait_sec);
56
57 /*
58  * Send current JobStatus to Director
59  */
60 int dir_send_job_status(JCR *jcr)
61 {
62    return bnet_fsend(jcr->dir_bsock, Job_status, jcr->Job, jcr->JobStatus);
63 }
64
65 /*
66  * Common routine for:
67  *   dir_get_volume_info()
68  * and
69  *   dir_find_next_appendable_volume()
70  */
71 static int do_request_volume_info(JCR *jcr)
72 {
73     BSOCK *dir = jcr->dir_bsock;
74     VOLUME_CAT_INFO vol;
75
76     jcr->VolumeName[0] = 0;           /* No volume */
77     if (bnet_recv(dir) <= 0) {
78        Dmsg0(200, "getvolname error bnet_recv\n");
79        Mmsg(&jcr->errmsg, _("Network error on bnet_recv in req_vol_info.\n"));
80        return 0;
81     }
82     if (sscanf(dir->msg, OK_media, vol.VolCatName, 
83                &vol.VolCatJobs, &vol.VolCatFiles,
84                &vol.VolCatBlocks, &vol.VolCatBytes,
85                &vol.VolCatMounts, &vol.VolCatErrors,
86                &vol.VolCatWrites, &vol.VolCatMaxBytes,
87                &vol.VolCatCapacityBytes, vol.VolCatStatus,
88                &vol.Slot, &vol.VolCatMaxJobs, &vol.VolCatMaxFiles) != 14) {
89
90        Dmsg1(200, "Bad response from Dir: %s\n", dir->msg);
91        Mmsg(&jcr->errmsg, _("Error scanning Dir response: %s\n"), dir->msg);
92        return 0;
93     }
94     unbash_spaces(vol.VolCatName);
95     pm_strcpy(&jcr->VolumeName, vol.VolCatName); /* set desired VolumeName */
96     memcpy(&jcr->VolCatInfo, &vol, sizeof(jcr->VolCatInfo));
97     
98     Dmsg2(200, "do_reqest_vol_info got slot=%d Volume=%s\n", 
99           vol.Slot, vol.VolCatName);
100     return 1;
101 }
102
103
104 /*
105  * Get Volume info for a specific volume from the Director's Database
106  *
107  * Returns: 1 on success   (not Director guarantees that Pool and MediaType
108  *                          are correct and VolStatus==Append or
109  *                          VolStatus==Recycle)
110  *          0 on failure
111  *
112  *          Volume information returned in jcr
113  */
114 int dir_get_volume_info(JCR *jcr, int writing)
115 {
116     BSOCK *dir = jcr->dir_bsock;
117
118     strcpy(jcr->VolCatInfo.VolCatName, jcr->VolumeName);
119     Dmsg1(200, "dir_get_volume_info=%s\n", jcr->VolCatInfo.VolCatName);
120     bash_spaces(jcr->VolCatInfo.VolCatName);
121     bnet_fsend(dir, Get_Vol_Info, jcr->Job, jcr->VolCatInfo.VolCatName, writing);
122     return do_request_volume_info(jcr);
123 }
124
125
126
127 /*
128  * Get info on the next appendable volume in the Director's database
129  * Returns: 1 on success
130  *          0 on failure
131  *
132  *          Volume information returned in jcr
133  *
134  */
135 int dir_find_next_appendable_volume(JCR *jcr)
136 {
137     BSOCK *dir = jcr->dir_bsock;
138
139     Dmsg0(200, "dir_find_next_appendable_volume\n");
140     bnet_fsend(dir, Find_media, jcr->Job, 1);
141     return do_request_volume_info(jcr);
142 }
143
144     
145 /*
146  * After writing a Volume, send the updated statistics
147  * back to the director.
148  */
149 int dir_update_volume_info(JCR *jcr, VOLUME_CAT_INFO *vol, int relabel)
150 {
151    BSOCK *dir = jcr->dir_bsock;
152    time_t EndTime = time(NULL);
153    char ed1[50], ed2[50];
154
155    if (vol->VolCatName[0] == 0) {
156       Jmsg0(jcr, M_ERROR, 0, _("NULL Volume name. This shouldn't happen!!!\n"));
157       return 0;
158    }
159    bash_spaces(vol->VolCatName);
160    bnet_fsend(dir, Update_media, jcr->Job, 
161       vol->VolCatName, vol->VolCatJobs, vol->VolCatFiles,
162       vol->VolCatBlocks, edit_uint64(vol->VolCatBytes, ed1),
163       vol->VolCatMounts, vol->VolCatErrors,
164       vol->VolCatWrites, edit_uint64(vol->VolCatMaxBytes, ed2), 
165       EndTime, vol->VolCatStatus, vol->Slot, relabel);
166    Dmsg1(120, "update_volume_data(): %s", dir->msg);
167    unbash_spaces(vol->VolCatName);
168    if (bnet_recv(dir) <= 0) {
169       Dmsg0(190, "updateVolCatInfo error bnet_recv\n");
170       Jmsg(jcr, M_ERROR, 0, _("Error updating Volume info Vol=\"%s\": ERR=%s\n"), 
171            vol->VolCatName, bnet_strerror(dir));
172       return 0;
173    }
174    Dmsg1(120, "Updatevol: %s", dir->msg);
175    if (strcmp(dir->msg, OK_update) != 0) {
176       Dmsg1(130, "Bad response from Dir: %s\n", dir->msg);
177       Jmsg(jcr, M_ERROR, 0, _("Error updating Volume info Vol=\"%s\": %s\n"), 
178            vol->VolCatName, dir->msg);
179       return 0;
180    }
181    return 1;
182 }
183
184 /*
185  * After writing a Volume, create the JobMedia record.
186  */
187 int dir_create_jobmedia_record(JCR *jcr)
188 {
189    BSOCK *dir = jcr->dir_bsock;
190
191    bnet_fsend(dir, Create_job_media, jcr->Job, 
192       jcr->VolFirstFile, jcr->JobFiles,
193       jcr->StartFile, jcr->EndFile,
194       jcr->StartBlock, jcr->EndBlock);
195    Dmsg1(100, "create_jobmedia(): %s", dir->msg);
196    if (bnet_recv(dir) <= 0) {
197       Dmsg0(190, "create_jobmedia error bnet_recv\n");
198       Jmsg(jcr, M_ERROR, 0, _("Error creating JobMedia record: ERR=%s\n"), 
199            bnet_strerror(dir));
200       return 0;
201    }
202    Dmsg1(120, "Create_jobmedia: %s", dir->msg);
203    if (strcmp(dir->msg, OK_update) != 0) {
204       Dmsg1(130, "Bad response from Dir: %s\n", dir->msg);
205       Jmsg(jcr, M_ERROR, 0, _("Error creating JobMedia record: %s\n"), dir->msg);
206       return 0;
207    }
208    return 1;
209 }
210
211
212 /* 
213  * Update File Attribute data
214  */
215 int dir_update_file_attributes(JCR *jcr, DEV_RECORD *rec)
216 {
217    BSOCK *dir = jcr->dir_bsock;
218    ser_declare;
219
220    dir->msglen = sprintf(dir->msg, FileAttributes, jcr->Job);
221    dir->msg = check_pool_memory_size(dir->msg, dir->msglen + 
222                 sizeof(DEV_RECORD) + rec->data_len);
223    ser_begin(dir->msg + dir->msglen, 0);
224    ser_uint32(rec->VolSessionId);
225    ser_uint32(rec->VolSessionTime);
226    ser_int32(rec->FileIndex);
227    ser_int32(rec->Stream);
228    ser_uint32(rec->data_len);
229    ser_bytes(rec->data, rec->data_len);
230    dir->msglen = ser_length(dir->msg);
231    return bnet_send(dir);
232 }
233
234
235 /*
236  *   
237  *   Entered with device blocked.
238  *   Leaves with device blocked.
239  *
240  *   Returns: 1 on success (operator issues a mount command)
241  *            0 on failure
242  *              Note, must create dev->errmsg on error return.
243  *
244  *    On success, jcr->VolumeName and jcr->VolCatInfo contain
245  *      information on suggested volume, but this may not be the
246  *      same as what is actually mounted.
247  *
248  *    When we return with success, the correct tape may or may not
249  *      actually be mounted. The calling routine must read it and
250  *      verify the label.
251  */
252 int dir_ask_sysop_to_mount_next_volume(JCR *jcr, DEVICE *dev)
253 {
254    int stat = 0, jstat;
255    /* ******FIXME******* put these on config variable */
256    int min_wait = 60 * 60;
257    int max_wait = 24 * 60 * 60;
258    int max_num_wait = 9;              /* 5 waits =~ 1 day, then 1 day at a time */
259
260    int wait_sec;
261    int num_wait = 0;
262
263    Dmsg0(130, "enter dir_ask_sysop_to_mount_next_volume\n");
264    ASSERT(dev->dev_blocked);
265    wait_sec = min_wait;
266    for ( ;; ) {
267       if (job_canceled(jcr)) {
268          Mmsg(&dev->errmsg, _("Job %s canceled while waiting for mount on Storage Device \"%s\".\n"), 
269               jcr->Job, jcr->dev_name);
270          Jmsg(jcr, M_FATAL, 0, "%s", dev->errmsg);
271          return 0;
272       }
273       if (dir_find_next_appendable_volume(jcr)) {    /* get suggested volume */
274          jstat = JS_WaitMount;
275          /*
276           * If we have a valid volume name and we are not
277           * removable media, return now, otherwise wait
278           * for the operator to mount the media.
279           */
280          if (jcr->VolumeName[0] && !dev_cap(dev, CAP_REM) && dev_cap(dev, CAP_LABEL)) {
281             Dmsg0(190, "Return 1 from mount without wait.\n");
282             return 1;
283          }
284          Jmsg(jcr, M_MOUNT, 0, _(
285 "Please mount Volume \"%s\" on Storage Device \"%s\" for Job %s\n"
286 "Use \"mount\" command to release Job.\n"),
287               jcr->VolumeName, jcr->dev_name, jcr->Job);
288          Dmsg3(190, "Mount %s on %s for Job %s\n",
289                 jcr->VolumeName, jcr->dev_name, jcr->Job);
290       } else {
291          jstat = JS_WaitMedia;
292          Jmsg(jcr, M_MOUNT, 0, _(
293 "Job %s waiting. Cannot find any appendable volumes.\n\
294 Please use the \"label\"  command to create a new Volume for:\n\
295     Storage:      %s\n\
296     Media type:   %s\n\
297     Pool:         %s\n"),
298               jcr->Job, 
299               jcr->dev_name, 
300               jcr->media_type,
301               jcr->pool_name);
302       }
303
304       jcr->JobStatus = jstat;
305       dir_send_job_status(jcr);
306
307       stat = wait_for_sysop(jcr, dev, wait_sec);
308
309       if (stat == ETIMEDOUT) {
310          wait_sec *= 2;               /* double wait time */
311          if (wait_sec > max_wait) {   /* but not longer than maxtime */
312             wait_sec = max_wait;
313          }
314          num_wait++;
315          if (num_wait >= max_num_wait) {
316             Mmsg(&dev->errmsg, _("Gave up waiting to mount Storage Device \"%s\" for Job %s\n"), 
317                  jcr->dev_name, jcr->Job);
318             Jmsg(jcr, M_FATAL, 0, "%s", dev->errmsg);
319             Dmsg1(190, "Gave up waiting on device %s\n", dev->dev_name);
320             return 0;                 /* exceeded maximum waits */
321          }
322          continue;
323       }
324       if (stat == EINVAL) {
325          Mmsg2(&dev->errmsg, _("pthread error in mount_next_volume stat=%d ERR=%s\n"),
326                stat, strerror(stat));
327          Jmsg(jcr, M_FATAL, 0, "%s", dev->errmsg);
328          return 0;
329       }
330       if (stat != 0) {
331          Jmsg(jcr, M_WARNING, 0, _("pthread error in mount_next_volume stat=%d ERR=%s\n"), stat,
332             strerror(stat));
333       }
334       Dmsg1(190, "Someone woke me for device %s\n", dev->dev_name);
335
336       /* Restart wait counters */
337       wait_sec = min_wait;
338       num_wait = 0;
339       /* If no VolumeName, and cannot get one, try again */
340       if (jcr->VolumeName[0] == 0 && 
341           !dir_find_next_appendable_volume(jcr)) {
342          Jmsg(jcr, M_MOUNT, 0, _(
343 "Someone woke me up, but I cannot find any appendable\n\
344 volumes for Job=%s.\n"), jcr->Job);
345          continue;
346       }       
347       break;
348    }
349    set_jcr_job_status(jcr, JS_Running);
350    dir_send_job_status(jcr);
351    Dmsg0(130, "leave dir_ask_sysop_to_mount_next_volume\n");
352    return 1;
353 }
354
355 /*
356  *   
357  *   Entered with device blocked and jcr->VolumeName is desired
358  *      volume.
359  *   Leaves with device blocked.
360  *
361  *   Returns: 1 on success (operator issues a mount command)
362  *            0 on failure
363  *              Note, must create dev->errmsg on error return.
364  *
365  */
366 int dir_ask_sysop_to_mount_volume(JCR *jcr, DEVICE *dev)
367 {
368    int stat = 0;
369    /* ******FIXME******* put these on config variable */
370    int min_wait = 60 * 60;
371    int max_wait = 24 * 60 * 60;
372    int max_num_wait = 9;              /* 5 waits =~ 1 day, then 1 day at a time */
373    int wait_sec;
374    int num_wait = 0;
375    char *msg;
376
377    Dmsg0(130, "enter dir_ask_sysop_to_mount_next_volume\n");
378    if (!jcr->VolumeName[0]) {
379       Mmsg0(&dev->errmsg, _("Cannot request another volume: no volume name given.\n"));
380       return 0;
381    }
382    ASSERT(dev->dev_blocked);
383    wait_sec = min_wait;
384    for ( ;; ) {
385       if (job_canceled(jcr)) {
386          Mmsg(&dev->errmsg, _("Job %s canceled while waiting for mount on Storage Device \"%s\".\n"), 
387               jcr->Job, jcr->dev_name);
388          return 0;
389       }
390       msg = _("Please mount");
391       Jmsg(jcr, M_MOUNT, 0, _("%s Volume \"%s\" on Storage Device \"%s\" for Job %s\n"),
392            msg, jcr->VolumeName, jcr->dev_name, jcr->Job);
393       Dmsg3(190, "Mount %s on %s for Job %s\n",
394             jcr->VolumeName, jcr->dev_name, jcr->Job);
395
396       jcr->JobStatus = JS_WaitMount;
397       dir_send_job_status(jcr);
398
399       stat = wait_for_sysop(jcr, dev, wait_sec); /* wait on device */
400
401       if (stat == ETIMEDOUT) {
402          wait_sec *= 2;               /* double wait time */
403          if (wait_sec > max_wait) {   /* but not longer than maxtime */
404             wait_sec = max_wait;
405          }
406          num_wait++;
407          if (num_wait >= max_num_wait) {
408             Mmsg(&dev->errmsg, _("Gave up waiting to mount Storage Device \"%s\" for Job %s\n"), 
409                  jcr->dev_name, jcr->Job);
410             Jmsg(jcr, M_FATAL, 0, "%s", dev->errmsg);
411             Dmsg1(190, "Gave up waiting on device %s\n", dev->dev_name);
412             return 0;                 /* exceeded maximum waits */
413          }
414          continue;
415       }
416       if (stat == EINVAL) {
417          Mmsg2(&dev->errmsg, _("pthread error in mount_volume stat=%d ERR=%s\n"),
418                stat, strerror(stat));
419          Jmsg(jcr, M_FATAL, 0, "%s", dev->errmsg);
420          return 0;
421       }
422       if (stat != 0) {
423          Jmsg(jcr, M_ERROR, 0, _("pthread error in mount_next_volume stat=%d ERR=%s\n"), stat,
424             strerror(stat));
425       }
426       Dmsg1(190, "Someone woke me for device %s\n", dev->dev_name);
427
428       /* Restart wait counters */
429       wait_sec = min_wait;
430       num_wait = 0;
431       break;
432    }
433    set_jcr_job_status(jcr, JS_Running);
434    dir_send_job_status(jcr);
435    Dmsg0(130, "leave dir_ask_sysop_to_mount_next_volume\n");
436    return 1;
437 }
438
439 /*
440  * Wait for SysOp to mount a tape
441  */
442 static int wait_for_sysop(JCR *jcr, DEVICE *dev, int wait_sec)
443 {
444    struct timeval tv;
445    struct timezone tz;
446    struct timespec timeout;
447    int dev_blocked;
448    time_t start = time(NULL);
449    time_t last_heartbeat = 0;
450    int stat = 0;
451    int add_wait;
452    
453    /*
454     * Wait requested time (wait_sec).  However, we also wake up every
455     *    HB_TIME seconds and send a heartbeat to the FD and the Director
456     *    to keep stateful firewalls from closing them down while waiting
457     *    for the operator.
458     */
459    add_wait = wait_sec;
460    if (me->heartbeat_interval && add_wait > me->heartbeat_interval) {
461       add_wait = me->heartbeat_interval;
462    }
463    gettimeofday(&tv, &tz);
464    timeout.tv_nsec = tv.tv_usec * 1000;
465    timeout.tv_sec = tv.tv_sec + add_wait;
466
467    P(dev->mutex);
468    dev_blocked = dev->dev_blocked;
469    dev->dev_blocked = BST_WAITING_FOR_SYSOP; /* indicate waiting for mount */
470
471    for ( ; !job_canceled(jcr); ) {
472       time_t now;
473
474       Dmsg3(100, "I'm going to sleep on device %s. HB=%d wait=%d\n", dev->dev_name,
475          (int)me->heartbeat_interval, wait_sec);
476       stat = pthread_cond_timedwait(&dev->wait_next_vol, &dev->mutex, &timeout);
477       Dmsg1(100, "Wokeup from sleep on device stat=%d\n", stat);
478
479       now = time(NULL);
480
481       /* Note, this always triggers the first time. We want that. */
482       if (me->heartbeat_interval) {
483          if (now - last_heartbeat >= me->heartbeat_interval) {
484             /* send heartbeats */
485             if (jcr->file_bsock) {
486                bnet_sig(jcr->file_bsock, BNET_HEARTBEAT);
487                Dmsg0(100, "Send heartbeat to FD.\n");
488             }
489             if (jcr->dir_bsock) {
490                bnet_sig(jcr->dir_bsock, BNET_HEARTBEAT);
491             }
492             last_heartbeat = now;
493          }
494       }
495
496       if (dev->dev_blocked == BST_MOUNT) {   /* mount request ? */
497          stat = 0;
498          break;
499       }
500
501       if (stat != ETIMEDOUT) {     /* we blocked the device */
502          break;                    /* on error return */
503       }
504       if (now - start >= wait_sec) {  /* on exceeding wait time return */
505          Dmsg0(100, "Exceed wait time.\n");
506          break;
507       }
508       add_wait = wait_sec - (now - start);
509       if (me->heartbeat_interval && add_wait > me->heartbeat_interval) {
510          add_wait = me->heartbeat_interval;
511       }
512       gettimeofday(&tv, &tz);
513       timeout.tv_nsec = tv.tv_usec * 1000;
514       timeout.tv_sec = tv.tv_sec + add_wait; /* additional wait */
515       Dmsg1(100, "Additional wait %d sec.\n", add_wait);
516    }
517
518    dev->dev_blocked = dev_blocked;
519    V(dev->mutex);
520    return stat;
521 }