]> git.sur5r.net Git - bacula/bacula/blob - bacula/src/stored/acquire.c
Don't allocated a DCR twice
[bacula/bacula] / bacula / src / stored / acquire.c
1 /*
2  *  Routines to acquire and release a device for read/write
3  *
4  *   Kern Sibbald, August MMII
5  *                            
6  *   Version $Id$
7  */
8 /*
9    Copyright (C) 2002-2003 Kern Sibbald and John Walker
10
11    This program is free software; you can redistribute it and/or
12    modify it under the terms of the GNU General Public License as
13    published by the Free Software Foundation; either version 2 of
14    the License, or (at your option) any later version.
15
16    This program is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19    General Public License for more details.
20
21    You should have received a copy of the GNU General Public
22    License along with this program; if not, write to the Free
23    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
24    MA 02111-1307, USA.
25
26  */
27
28 #include "bacula.h"                   /* pull in global headers */
29 #include "stored.h"                   /* pull in Storage Deamon headers */
30
31 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
32
33 DCR *new_dcr(JCR *jcr, DEVICE *dev)
34 {
35    if (jcr && jcr->dcr) {
36       return jcr->dcr;
37    }
38    DCR *dcr = (DCR *)malloc(sizeof(DCR));
39    memset(dcr, 0, sizeof(DCR));
40    if (jcr) {
41       jcr->dcr = dcr;
42    }
43    dcr->jcr = jcr;
44    dcr->dev = dev;
45    dcr->block = new_block(dev);
46    dcr->record = new_record();
47    dcr->spool_fd = -1;
48    dcr->max_spool_size = dev->device->max_spool_size;
49    return dcr;
50 }
51
52 void free_dcr(DCR *dcr)
53 {
54    if (dcr->block) {
55       free_block(dcr->block);
56    }
57    if (dcr->record) {
58       free_record(dcr->record);
59    }
60    if (dcr->jcr) {
61       dcr->jcr->dcr = NULL;
62    }
63    free(dcr);
64 }
65
66
67 /********************************************************************* 
68  * Acquire device for reading.  We permit (for the moment)
69  *  only one reader.  We read the Volume label from the block and
70  *  leave the block pointers just after the label.
71  *
72  *  Returns: 0 if failed for any reason
73  *           1 if successful
74  */
75 DCR *acquire_device_for_read(JCR *jcr)
76 {
77    bool vol_ok = false;
78    bool tape_previously_mounted;
79    bool tape_initially_mounted;
80    VOL_LIST *vol;
81    int autochanger = 0;
82    int i;
83    DCR *dcr = jcr->dcr;
84    DEVICE *dev;
85    
86    /* Called for each volume */
87    if (!dcr) {
88       dcr = new_dcr(jcr, jcr->device->dev);
89    }
90    dev = dcr->dev;
91    if (device_is_unmounted(dev)) {
92       Jmsg(jcr, M_WARNING, 0, _("device %s is BLOCKED due to user unmount.\n"),
93          dev_name(dev));
94    }
95    lock_device(dev);
96    block_device(dev, BST_DOING_ACQUIRE);
97    unlock_device(dev);
98
99    init_dev_wait_timers(dev);
100    if (dev_state(dev, ST_READ) || dev->num_writers > 0) {
101       Jmsg2(jcr, M_FATAL, 0, _("Device %s is busy. Job %d canceled.\n"), 
102             dev_name(dev), jcr->JobId);
103       goto get_out;
104    }
105
106    tape_previously_mounted = dev_state(dev, ST_READ) || 
107                              dev_state(dev, ST_APPEND) ||
108                              dev_state(dev, ST_LABEL);
109    tape_initially_mounted = tape_previously_mounted;
110
111    /* Find next Volume, if any */
112    vol = jcr->VolList;
113    if (!vol) {
114       Jmsg(jcr, M_FATAL, 0, _("No volumes specified. Job %d canceled.\n"), jcr->JobId);
115       goto get_out;
116    }
117    jcr->CurVolume++;
118    for (i=1; i<jcr->CurVolume; i++) {
119       vol = vol->next;
120    }
121    pm_strcpy(&jcr->VolumeName, vol->VolumeName);
122    bstrncpy(dcr->VolumeName, vol->VolumeName, sizeof(dcr->VolumeName));
123
124    for (i=0; i<5; i++) {
125       if (job_canceled(jcr)) {
126          Mmsg1(&dev->errmsg, _("Job %d canceled.\n"), jcr->JobId);
127          goto get_out;                /* error return */
128       }
129       /*
130        * This code ensures that the device is ready for
131        * reading. If it is a file, it opens it.
132        * If it is a tape, it checks the volume name 
133        */
134       for ( ; !(dev->state & ST_OPENED); ) {
135          Dmsg1(120, "bstored: open vol=%s\n", jcr->VolumeName);
136          if (open_dev(dev, dcr->VolumeName, OPEN_READ_ONLY) < 0) {
137             Jmsg(jcr, M_FATAL, 0, _("Open device %s volume %s failed, ERR=%s\n"), 
138                 dev_name(dev), dcr->VolumeName, strerror_dev(dev));
139             goto get_out;
140          }
141          Dmsg1(129, "open_dev %s OK\n", dev_name(dev));
142       }
143       /****FIXME***** do not reread label if ioctl() says we are
144        *  correctly possitioned.  Possibly have way user can turn
145        *  this optimization (to be implemented) off.
146        */
147       dcr->dev->state &= ~ST_LABEL;           /* force reread of label */
148       Dmsg0(200, "calling read-vol-label\n");
149       switch (read_dev_volume_label(jcr, dev, dcr->block)) {
150       case VOL_OK:
151          vol_ok = true;
152          break;                    /* got it */
153       case VOL_IO_ERROR:
154          /*
155           * Send error message generated by read_dev_volume_label()
156           *  only we really had a tape mounted. This supresses superfluous
157           *  error messages when nothing is mounted.
158           */
159          if (tape_previously_mounted) {
160             Jmsg(jcr, M_WARNING, 0, "%s", jcr->errmsg);                         
161          }
162          goto default_path;
163       case VOL_NAME_ERROR:
164          if (tape_initially_mounted) {
165             tape_initially_mounted = false;
166             goto default_path;
167          }
168          /* Fall through */
169       default:
170          Jmsg(jcr, M_WARNING, 0, "%s", jcr->errmsg);
171 default_path:
172          tape_previously_mounted = true;
173          Dmsg0(200, "dir_get_volume_info\n");
174          if (!dir_get_volume_info(jcr, GET_VOL_INFO_FOR_READ)) { 
175             Jmsg1(jcr, M_WARNING, 0, "%s", jcr->errmsg);
176          }
177          /* Call autochanger only once unless ask_sysop called */
178          if (!autochanger) {
179             int stat;
180             Dmsg2(200, "calling autoload Vol=%s Slot=%d\n",
181                jcr->VolumeName, jcr->VolCatInfo.Slot);                         
182             stat = autoload_device(jcr, dev, 0, NULL);
183             if (stat > 0) {
184                autochanger = 1;
185                continue;
186             }
187          }
188          /* Mount a specific volume and no other */
189          Dmsg0(200, "calling dir_ask_sysop\n");
190          if (!dir_ask_sysop_to_mount_volume(jcr, dev)) {
191             goto get_out;             /* error return */
192          }
193          autochanger = 0;             /* permit using autochanger again */
194          continue;                    /* try reading again */
195       } /* end switch */
196       break;
197    } /* end for loop */
198    if (!vol_ok) {
199       Jmsg1(jcr, M_FATAL, 0, _("Too many errors trying to mount device \"%s\".\n"),
200             dev_name(dev));
201       goto get_out;
202    }
203
204    dev->state &= ~ST_APPEND;          /* clear any previous append mode */
205    dev->state |= ST_READ;             /* set read mode */
206    attach_jcr_to_device(dev, jcr);    /* attach jcr to device */
207    Jmsg(jcr, M_INFO, 0, _("Ready to read from volume \"%s\" on device %s.\n"),
208       jcr->VolumeName, dev_name(dev));
209
210 get_out:
211    P(dev->mutex); 
212    unblock_device(dev);
213    V(dev->mutex);
214    if (!vol_ok) {
215       free_dcr(dcr);
216       dcr = NULL;
217    }
218    return dcr;
219 }
220
221 /*
222  * Acquire device for writing. We permit multiple writers.
223  *  If this is the first one, we read the label.
224  *
225  *  Returns: NULL if failed for any reason
226  *           dev if successful (may change if new dev opened)
227  *  This routine must be single threaded because we may create
228  *   multiple devices (for files), thus we have our own mutex 
229  *   on top of the device mutex.
230  */
231 DCR *acquire_device_for_append(JCR *jcr)
232 {
233    int release = 0;
234    bool recycle = false;
235    bool do_mount = false;
236    DCR *dcr;
237    DEVICE *dev = jcr->device->dev;
238
239    dcr = new_dcr(jcr, dev);
240    if (device_is_unmounted(dev)) {
241       Jmsg(jcr, M_WARNING, 0, _("device %s is BLOCKED due to user unmount.\n"),
242          dev_name(dev));
243    }
244    lock_device(dev);
245    block_device(dev, BST_DOING_ACQUIRE);
246    unlock_device(dev);
247    P(mutex);                         /* lock all devices */
248    Dmsg1(190, "acquire_append device is %s\n", dev_is_tape(dev)?"tape":"disk");
249              
250
251    if (dev_state(dev, ST_APPEND)) {
252       /* 
253        * Device already in append mode   
254        *
255        * Check if we have the right Volume mounted   
256        *   OK if current volume info OK
257        *   OK if next volume matches current volume
258        *   otherwise mount desired volume obtained from
259        *    dir_find_next_appendable_volume
260        */
261       pm_strcpy(&jcr->VolumeName, dev->VolHdr.VolName);
262       bstrncpy(dcr->VolumeName, dev->VolHdr.VolName, sizeof(dcr->VolumeName));
263       if (!dir_get_volume_info(jcr, GET_VOL_INFO_FOR_WRITE) &&
264           !(dir_find_next_appendable_volume(jcr) &&
265             strcmp(dev->VolHdr.VolName, jcr->VolumeName) == 0)) { /* wrong tape mounted */
266          if (dev->num_writers != 0) {
267             DEVICE *d = ((DEVRES *)dev->device)->dev;
268             uint32_t open_vols = 0;
269             for ( ; d; d=d->next) {
270                open_vols++;
271             }
272             if (dev_state(dev, ST_FILE) && dev->max_open_vols > open_vols) {
273                d = init_dev(NULL, (DEVRES *)dev->device); /* init new device */
274                d->prev = dev;                   /* chain in new device */
275                d->next = dev->next;
276                dev->next = d;
277                /* Release old device */
278                P(dev->mutex); 
279                unblock_device(dev);
280                V(dev->mutex);
281                /* Make new device current device and lock it */
282                dev = d;
283                lock_device(dev);
284                block_device(dev, BST_DOING_ACQUIRE);
285                unlock_device(dev);
286             } else {
287                Jmsg(jcr, M_FATAL, 0, _("Device %s is busy writing on another Volume.\n"), dev_name(dev));
288                goto get_out;
289             }
290          }
291          /* Wrong tape mounted, release it, then fall through to get correct one */
292          release = 1;
293          do_mount = true;
294       } else {
295          /*       
296           * At this point, the correct tape is already mounted, so 
297           *   we do not need to do mount_next_write_volume(), unless
298           *   we need to recycle the tape.
299           */
300           recycle = strcmp(jcr->VolCatInfo.VolCatStatus, "Recycle") == 0;
301           if (recycle && dev->num_writers != 0) {
302              Jmsg(jcr, M_FATAL, 0, _("Cannot recycle volume \"%s\""
303                   " because it is in use by another job.\n"));
304              goto get_out;
305           }
306        }
307    } else { 
308       /* Not already in append mode, so mount the device */
309       if (dev_state(dev, ST_READ)) {
310          Jmsg(jcr, M_FATAL, 0, _("Device %s is busy reading.\n"), dev_name(dev));
311          goto get_out;
312       } 
313       ASSERT(dev->num_writers == 0);
314       do_mount = true;
315    }
316
317    if (do_mount || recycle) {
318       if (!mount_next_write_volume(jcr, dev, dcr->block, release)) {
319          if (!job_canceled(jcr)) {
320             /* Reduce "noise" -- don't print if job canceled */
321             Jmsg(jcr, M_FATAL, 0, _("Could not ready device \"%s\" for append.\n"),
322                dev_name(dev));
323          }
324          goto get_out;
325       }
326    }
327
328    dev->num_writers++;
329    if (jcr->NumVolumes == 0) {
330       jcr->NumVolumes = 1;
331    }
332    attach_jcr_to_device(dev, jcr);    /* attach jcr to device */
333    goto ok_out;
334
335 /*
336  * If we jump here, it is an error return because
337  *  rtn_dev will still be NULL
338  */
339 get_out:
340    free_dcr(dcr);
341    dcr = NULL;
342 ok_out:
343    P(dev->mutex); 
344    unblock_device(dev);
345    V(dev->mutex);
346    V(mutex);                          /* unlock other threads */
347    return dcr;
348 }
349
350 /*
351  * This job is done, so release the device. From a Unix standpoint,
352  *  the device remains open.
353  *
354  */
355 int release_device(JCR *jcr)
356 {
357    DEVICE *dev = jcr->dcr->dev;   
358    lock_device(dev);
359    Dmsg1(100, "release_device device is %s\n", dev_is_tape(dev)?"tape":"disk");
360    if (dev_state(dev, ST_READ)) {
361       dev->state &= ~ST_READ;         /* clear read bit */
362       if (!dev_is_tape(dev) || !dev_cap(dev, CAP_ALWAYSOPEN)) {
363          offline_or_rewind_dev(dev);
364          close_dev(dev);
365       }
366       /******FIXME**** send read volume usage statistics to director */
367
368    } else if (dev->num_writers > 0) {
369       dev->num_writers--;
370       Dmsg1(100, "There are %d writers in release_device\n", dev->num_writers);
371       if (dev_state(dev, ST_LABEL)) {
372          Dmsg0(100, "dir_create_jobmedia_record. Release\n");
373          if (!dir_create_jobmedia_record(jcr)) {
374             Jmsg(jcr, M_ERROR, 0, _("Could not create JobMedia record for Volume=\"%s\" Job=%s\n"),
375             jcr->VolCatInfo.VolCatName, jcr->Job);
376          }
377          /* If no more writers, write an EOF */
378          if (!dev->num_writers && dev_can_write(dev)) {
379             weof_dev(dev, 1);
380          }
381          dev->VolCatInfo.VolCatFiles = dev->file;   /* set number of files */
382          dev->VolCatInfo.VolCatJobs++;              /* increment number of jobs */
383          /* Note! do volume update before close, which zaps VolCatInfo */
384          Dmsg0(100, "dir_update_vol_info. Release0\n");
385          dir_update_volume_info(jcr, dev, 0); /* send Volume info to Director */
386       }
387
388       if (!dev->num_writers && (!dev_is_tape(dev) || !dev_cap(dev, CAP_ALWAYSOPEN))) {
389          offline_or_rewind_dev(dev);
390          close_dev(dev);
391       }
392    } else {
393       Jmsg2(jcr, M_ERROR, 0, _("BAD ERROR: release_device %s, Volume \"%s\" not in use.\n"), 
394             dev_name(dev), NPRT(jcr->VolumeName));
395    }
396    detach_jcr_from_device(dev, jcr);
397    if (dev->prev && !dev_state(dev, ST_READ) && !dev->num_writers) {
398       P(mutex);
399       unlock_device(dev);
400       dev->prev->next = dev->next;    /* dechain */
401       term_dev(dev);
402       V(mutex);
403    } else {
404       unlock_device(dev);
405    }
406    free_dcr(jcr->dcr);
407    jcr->dcr = NULL;
408    return 1;
409 }