3 * Bacula Director -- Tape labeling commands
5 * Kern Sibbald, April MMIII
11 Copyright (C) 2000-2004 Kern Sibbald and John Walker
13 This program is free software; you can redistribute it and/or
14 modify it under the terms of the GNU General Public License as
15 published by the Free Software Foundation; either version 2 of
16 the License, or (at your option) any later version.
18 This program is distributed in the hope that it will be useful,
19 but WITHOUT ANY WARRANTY; without even the implied warranty of
20 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
21 General Public License for more details.
23 You should have received a copy of the GNU General Public
24 License along with this program; if not, write to the Free
25 Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
33 /* Slot list definition */
34 typedef struct s_vol_list {
35 struct s_vol_list *next;
41 /* Forward referenced functions */
42 static int do_label(UAContext *ua, const char *cmd, int relabel);
43 static void label_from_barcodes(UAContext *ua);
44 static bool send_label_request(UAContext *ua, MEDIA_DBR *mr, MEDIA_DBR *omr,
45 POOL_DBR *pr, int relabel, bool media_record_exits);
46 static vol_list_t *get_vol_list_from_SD(UAContext *ua, bool scan);
47 static void free_vol_list(vol_list_t *vol_list);
48 static bool is_cleaning_tape(UAContext *ua, MEDIA_DBR *mr, POOL_DBR *pr);
49 static BSOCK *open_sd_bsock(UAContext *ua);
50 static void close_sd_bsock(UAContext *ua);
51 static char *get_volume_name_from_SD(UAContext *ua, int Slot);
57 * label storage=xxx volume=vvv
59 int label_cmd(UAContext *ua, const char *cmd)
61 return do_label(ua, cmd, 0); /* standard label */
64 int relabel_cmd(UAContext *ua, const char *cmd)
66 return do_label(ua, cmd, 1); /* relabel tape */
69 static int const max_slots = 5000;
71 static bool get_user_slot_list(UAContext *ua, char *slot_list, int num_slots)
76 for (int i=0; i<num_slots; i++) {
79 i = find_arg_with_value(ua, "slots");
81 /* scan slot list in ua->argv[i] */
85 strip_trailing_junk(ua->argv[i]);
86 for (p=ua->argv[i]; p && *p; p=e) {
93 h = strchr(p, '-'); /* range? */
95 msg = _("Negative numbers not permitted\n");
100 if (!is_an_integer(h)) {
101 msg = _("Range end is not integer.\n");
105 if (!is_an_integer(p)) {
106 msg = _("Range start is not an integer.\n");
112 msg = _("Range end not bigger than start.\n");
117 if (!is_an_integer(p)) {
118 msg = _("Input value is not an integer.\n");
123 if (beg <= 0 || end <= 0) {
124 msg = _("Values must be be greater than zero.\n");
127 if (end >= num_slots) {
128 msg = _("Slot too large.\n");
131 for (i=beg; i<=end; i++) {
132 slot_list[i] = 1; /* Turn on specified range */
136 /* Turn everything on */
137 for (i=0; i<num_slots; i++) {
142 printf("Slots turned on:\n");
143 for (i=1; i<num_slots; i++) {
156 * Update Slots corresponding to Volumes in autochanger
158 int update_slots(UAContext *ua)
161 vol_list_t *vl, *vol_list = NULL;
169 store = get_storage_resource(ua, 1);
173 ua->jcr->store = store;
175 scan = find_arg(ua, _("scan")) >= 0;
177 slot_list = (char *)malloc(max_slots);
178 if (!get_user_slot_list(ua, slot_list, max_slots)) {
183 vol_list = get_vol_list_from_SD(ua, scan);
186 bsendmsg(ua, _("No Volumes found to label, or no barcodes.\n"));
190 /* Walk through the list updating the media records */
191 for (vl=vol_list; vl; vl=vl->next) {
192 if (vl->Slot >= max_slots) {
193 bsendmsg(ua, _("Slot %d larger than max %d ignored.\n"));
196 /* Check if user wants us to look at this slot */
197 if (!slot_list[vl->Slot]) {
198 Dmsg1(100, "Skipping slot=%d\n", vl->Slot);
201 /* If scanning, we read the label rather than the barcode */
207 vl->VolName = get_volume_name_from_SD(ua, vl->Slot);
208 Dmsg2(100, "Got Vol=%s from SD for Slot=%d\n", vl->VolName, vl->Slot);
210 slot_list[vl->Slot] = 0; /* clear Slot */
212 Dmsg1(100, "No VolName for Slot=%d setting InChanger to zero.\n", vl->Slot);
213 memset(&mr, 0, sizeof(mr));
216 /* Set InChanger to zero for this Slot */
218 db_make_inchanger_unique(ua->jcr, ua->db, &mr);
220 bsendmsg(ua, _("No VolName for Slot=%d set InChanger to zero.\n"), vl->Slot);
223 memset(&mr, 0, sizeof(mr));
224 bstrncpy(mr.VolumeName, vl->VolName, sizeof(mr.VolumeName));
226 if (db_get_media_record(ua->jcr, ua->db, &mr)) {
227 if (mr.Slot != vl->Slot || !mr.InChanger) {
230 if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
231 bsendmsg(ua, "%s", db_strerror(ua->db));
234 "Catalog record for Volume \"%s\" updated to reference slot %d.\n"),
235 mr.VolumeName, mr.Slot);
238 bsendmsg(ua, _("Catalog record for Volume \"%s\" is up to date.\n"),
244 bsendmsg(ua, _("Record for Volume \"%s\" not found in catalog.\n"),
249 memset(&mr, 0, sizeof(mr));
252 for (int i=1; i<max_slots; i++) {
255 /* Set InChanger to zero for this Slot */
256 db_make_inchanger_unique(ua->jcr, ua->db, &mr);
263 free_vol_list(vol_list);
272 * Common routine for both label and relabel
274 static int do_label(UAContext *ua, const char *cmd, int relabel)
278 char dev_name[MAX_NAME_LENGTH];
281 bool print_reminder = true;
284 bool media_record_exists = false;
285 static const char *barcode_keyword[] = {
291 memset(&pr, 0, sizeof(pr));
295 store = get_storage_resource(ua, 1);
299 ua->jcr->store = store;
301 if (!relabel && find_arg_keyword(ua, barcode_keyword) >= 0) {
302 label_from_barcodes(ua);
306 /* If relabel get name of Volume to relabel */
308 /* Check for oldvolume=name */
309 i = find_arg_with_value(ua, "oldvolume");
311 memset(&omr, 0, sizeof(omr));
312 bstrncpy(omr.VolumeName, ua->argv[i], sizeof(omr.VolumeName));
313 if (db_get_media_record(ua->jcr, ua->db, &omr)) {
316 bsendmsg(ua, "%s", db_strerror(ua->db));
318 /* No keyword or Vol not found, ask user to select */
319 if (!select_media_dbr(ua, &omr)) {
323 /* Require Volume to be Purged or Recycled */
325 if (strcmp(omr.VolStatus, "Purged") != 0 && strcmp(omr.VolStatus, "Recycle") != 0) {
326 bsendmsg(ua, _("Volume \"%s\" has VolStatus %s. It must be Purged or Recycled before relabeling.\n"),
327 omr.VolumeName, omr.VolStatus);
332 /* Check for volume=NewVolume */
333 i = find_arg_with_value(ua, "volume");
335 pm_strcpy(&ua->cmd, ua->argv[i]);
339 /* Get a new Volume name */
341 media_record_exists = false;
342 if (!get_cmd(ua, _("Enter new Volume name: "))) {
346 if (!is_volume_name_legal(ua, ua->cmd)) {
350 memset(&mr, 0, sizeof(mr));
351 bstrncpy(mr.VolumeName, ua->cmd, sizeof(mr.VolumeName));
352 /* If VolBytes are zero the Volume is not labeled */
353 if (db_get_media_record(ua->jcr, ua->db, &mr)) {
354 if (mr.VolBytes != 0) {
355 bsendmsg(ua, _("Media record for new Volume \"%s\" already exists.\n"),
359 media_record_exists = true;
364 /* If autochanger, request slot */
365 if (store->autochanger) {
366 i = find_arg_with_value(ua, "slot");
368 mr.Slot = atoi(ua->argv[i]);
369 } else if (!get_pint(ua, _("Enter slot (0 for none): "))) {
372 mr.Slot = ua->pint32_val;
374 mr.InChanger = 1; /* assumed if we are labeling it */
377 bstrncpy(mr.MediaType, store->media_type, sizeof(mr.MediaType));
379 /* Must select Pool if not already done */
380 if (pr.PoolId == 0) {
381 memset(&pr, 0, sizeof(pr));
382 if (!select_pool_dbr(ua, &pr)) {
388 ok = send_label_request(ua, &mr, &omr, &pr, relabel, media_record_exists);
391 sd = ua->jcr->store_bsock;
393 /* Delete the old media record */
394 if (!db_delete_media_record(ua->jcr, ua->db, &omr)) {
395 bsendmsg(ua, _("Delete of Volume \"%s\" failed. ERR=%s"),
396 omr.VolumeName, db_strerror(ua->db));
398 bsendmsg(ua, _("Old volume \"%s\" deleted from catalog.\n"),
400 /* Update the number of Volumes in the pool */
402 if (!db_update_pool_record(ua->jcr, ua->db, &pr)) {
403 bsendmsg(ua, "%s", db_strerror(ua->db));
408 bstrncpy(dev_name, store->dev_name, sizeof(dev_name));
409 bsendmsg(ua, _("Requesting to mount %s ...\n"), dev_name);
410 bash_spaces(dev_name);
411 bnet_fsend(sd, "mount %s", dev_name);
412 unbash_spaces(dev_name);
413 while (bnet_recv(sd) >= 0) {
414 bsendmsg(ua, "%s", sd->msg);
416 * 3001 OK mount. Device=xxx or
417 * 3001 Mounted Volume vvvv
418 * 3906 is cannot mount non-tape
419 * So for those, no need to print a reminder
421 if (strncmp(sd->msg, "3001 ", 5) == 0 ||
422 strncmp(sd->msg, "3906 ", 5) == 0) {
423 print_reminder = false;
428 if (print_reminder) {
429 bsendmsg(ua, _("Do not forget to mount the drive!!!\n"));
437 * Request SD to send us the slot:barcodes, then wiffle
438 * through them all labeling them.
440 static void label_from_barcodes(UAContext *ua)
442 STORE *store = ua->jcr->store;
445 vol_list_t *vl, *vol_list = NULL;
446 bool media_record_exists;
449 slot_list = (char *)malloc(max_slots);
450 if (!get_user_slot_list(ua, slot_list, max_slots)) {
454 vol_list = get_vol_list_from_SD(ua, false /*no scan*/);
457 bsendmsg(ua, _("No Volumes found to label, or no barcodes.\n"));
461 /* Display list of Volumes and ask if he really wants to proceed */
462 bsendmsg(ua, _("The following Volumes will be labeled:\n"
464 "==============\n"));
465 for (vl=vol_list; vl; vl=vl->next) {
466 if (!vl->VolName || !slot_list[vl->Slot]) {
469 bsendmsg(ua, "%4d %s\n", vl->Slot, vl->VolName);
471 if (!get_cmd(ua, _("Do you want to continue? (y/n): ")) ||
472 (ua->cmd[0] != 'y' && ua->cmd[0] != 'Y')) {
476 memset(&pr, 0, sizeof(pr));
477 if (!select_pool_dbr(ua, &pr)) {
480 memset(&omr, 0, sizeof(omr));
482 /* Fire off the label requests */
483 for (vl=vol_list; vl; vl=vl->next) {
484 if (!vl->VolName || !slot_list[vl->Slot]) {
487 memset(&mr, 0, sizeof(mr));
488 bstrncpy(mr.VolumeName, vl->VolName, sizeof(mr.VolumeName));
489 media_record_exists = false;
490 if (db_get_media_record(ua->jcr, ua->db, &mr)) {
491 if (mr.VolBytes != 0) {
492 bsendmsg(ua, _("Media record for Slot %d Volume \"%s\" already exists.\n"),
493 vl->Slot, mr.VolumeName);
496 if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
497 bsendmsg(ua, "Error setting InChanger: ERR=%s", db_strerror(ua->db));
502 media_record_exists = true;
506 * Deal with creating cleaning tape here. Normal tapes created in
507 * send_label_request() below
509 if (is_cleaning_tape(ua, &mr, &pr)) {
510 if (media_record_exists) { /* we update it */
512 if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
513 bsendmsg(ua, "%s", db_strerror(ua->db));
515 } else { /* create the media record */
516 set_pool_dbr_defaults_in_media_dbr(&mr, &pr);
517 if (db_create_media_record(ua->jcr, ua->db, &mr)) {
518 bsendmsg(ua, _("Catalog record for cleaning tape \"%s\" successfully created.\n"),
520 pr.NumVols++; /* this is a bit suspect */
521 if (!db_update_pool_record(ua->jcr, ua->db, &pr)) {
522 bsendmsg(ua, "%s", db_strerror(ua->db));
525 bsendmsg(ua, "Catalog error on cleaning tape: %s", db_strerror(ua->db));
528 continue; /* done, go handle next volume */
530 bstrncpy(mr.MediaType, store->media_type, sizeof(mr.MediaType));
533 send_label_request(ua, &mr, &omr, &pr, 0, media_record_exists);
539 free_vol_list(vol_list);
546 * Check if the Volume name has legal characters
547 * If ua is non-NULL send the message
549 bool is_volume_name_legal(UAContext *ua, const char *name)
553 const char *accept = ":.-_";
555 /* Restrict the characters permitted in the Volume name */
556 for (p=name; *p; p++) {
557 if (B_ISALPHA(*p) || B_ISDIGIT(*p) || strchr(accept, (int)(*p))) {
561 bsendmsg(ua, _("Illegal character \"%c\" in a volume name.\n"), *p);
566 if (len >= MAX_NAME_LENGTH) {
568 bsendmsg(ua, _("Volume name too long.\n"));
574 bsendmsg(ua, _("Volume name must be at least one character long.\n"));
582 * NOTE! This routine opens the SD socket but leaves it open
584 static bool send_label_request(UAContext *ua, MEDIA_DBR *mr, MEDIA_DBR *omr,
585 POOL_DBR *pr, int relabel, bool media_record_exists)
588 char dev_name[MAX_NAME_LENGTH];
591 if (!(sd=open_sd_bsock(ua))) {
594 bstrncpy(dev_name, ua->jcr->store->dev_name, sizeof(dev_name));
595 bash_spaces(dev_name);
596 bash_spaces(mr->VolumeName);
597 bash_spaces(mr->MediaType);
598 bash_spaces(pr->Name);
600 bash_spaces(omr->VolumeName);
601 bnet_fsend(sd, "relabel %s OldName=%s NewName=%s PoolName=%s MediaType=%s Slot=%d",
602 dev_name, omr->VolumeName, mr->VolumeName, pr->Name, mr->MediaType, mr->Slot);
603 bsendmsg(ua, _("Sending relabel command from \"%s\" to \"%s\" ...\n"),
604 omr->VolumeName, mr->VolumeName);
606 bnet_fsend(sd, "label %s VolumeName=%s PoolName=%s MediaType=%s Slot=%d",
607 dev_name, mr->VolumeName, pr->Name, mr->MediaType, mr->Slot);
608 bsendmsg(ua, _("Sending label command for Volume \"%s\" Slot %d ...\n"),
609 mr->VolumeName, mr->Slot);
610 Dmsg5(200, "label %s VolumeName=%s PoolName=%s MediaType=%s Slot=%d\n",
611 dev_name, mr->VolumeName, pr->Name, mr->MediaType, mr->Slot);
614 while (bnet_recv(sd) >= 0) {
615 bsendmsg(ua, "%s", sd->msg);
616 if (strncmp(sd->msg, "3000 OK label.", 14) == 0) {
620 unbash_spaces(mr->VolumeName);
621 unbash_spaces(mr->MediaType);
622 unbash_spaces(pr->Name);
623 mr->LabelDate = time(NULL);
625 if (media_record_exists) { /* we update it */
628 if (!db_update_media_record(ua->jcr, ua->db, mr)) {
629 bsendmsg(ua, "%s", db_strerror(ua->db));
632 } else { /* create the media record */
633 set_pool_dbr_defaults_in_media_dbr(mr, pr);
634 mr->VolBytes = 1; /* flag indicating Volume labeled */
636 if (db_create_media_record(ua->jcr, ua->db, mr)) {
637 bsendmsg(ua, _("Catalog record for Volume \"%s\", Slot %d successfully created.\n"),
638 mr->VolumeName, mr->Slot);
639 /* Update number of volumes in pool */
641 if (!db_update_pool_record(ua->jcr, ua->db, pr)) {
642 bsendmsg(ua, "%s", db_strerror(ua->db));
645 bsendmsg(ua, "%s", db_strerror(ua->db));
650 bsendmsg(ua, _("Label command failed for Volume %s.\n"), mr->VolumeName);
655 static BSOCK *open_sd_bsock(UAContext *ua)
657 STORE *store = ua->jcr->store;
659 if (!ua->jcr->store_bsock) {
660 bsendmsg(ua, _("Connecting to Storage daemon %s at %s:%d ...\n"),
661 store->hdr.name, store->address, store->SDport);
662 if (!connect_to_storage_daemon(ua->jcr, 10, SDConnectTimeout, 1)) {
663 bsendmsg(ua, _("Failed to connect to Storage daemon.\n"));
667 return ua->jcr->store_bsock;
670 static void close_sd_bsock(UAContext *ua)
672 if (ua->jcr->store_bsock) {
673 bnet_sig(ua->jcr->store_bsock, BNET_TERMINATE);
674 bnet_close(ua->jcr->store_bsock);
675 ua->jcr->store_bsock = NULL;
679 static char *get_volume_name_from_SD(UAContext *ua, int Slot)
681 STORE *store = ua->jcr->store;
683 char dev_name[MAX_NAME_LENGTH];
684 char *VolName = NULL;
687 if (!(sd=open_sd_bsock(ua))) {
688 bsendmsg(ua, _("Could not open SD socket.\n"));
691 bstrncpy(dev_name, store->dev_name, sizeof(dev_name));
692 bash_spaces(dev_name);
693 /* Ask for autochanger list of volumes */
694 bnet_fsend(sd, _("readlabel %s Slot=%d\n"), dev_name, Slot);
695 Dmsg1(100, "Sent: %s", sd->msg);
697 /* Get Volume name in this Slot */
698 while (bnet_recv(sd) >= 0) {
699 bsendmsg(ua, "%s", sd->msg);
700 Dmsg1(100, "Got: %s", sd->msg);
701 if (strncmp(sd->msg, "3001 Volume=", 12) == 0) {
702 VolName = (char *)malloc(sd->msglen);
703 if (sscanf(sd->msg, "3001 Volume=%s Slot=%d", VolName, &rtn_slot) == 2) {
711 Dmsg1(200, "get_vol_name=%s\n", NPRT(VolName));
716 * We get the slot list from the Storage daemon.
717 * If scan is set, we return all slots found,
718 * otherwise, we return only slots with valid barcodes (Volume names)
720 static vol_list_t *get_vol_list_from_SD(UAContext *ua, bool scan)
722 STORE *store = ua->jcr->store;
723 char dev_name[MAX_NAME_LENGTH];
726 vol_list_t *vol_list = NULL;
729 if (!(sd=open_sd_bsock(ua))) {
733 bstrncpy(dev_name, store->dev_name, sizeof(dev_name));
734 bash_spaces(dev_name);
735 /* Ask for autochanger list of volumes */
736 bnet_fsend(sd, _("autochanger list %s \n"), dev_name);
738 /* Read and organize list of Volumes */
739 while (bnet_recv(sd) >= 0) {
742 strip_trailing_junk(sd->msg);
744 /* Check for returned SD messages */
745 if (sd->msg[0] == '3' && B_ISDIGIT(sd->msg[1]) &&
746 B_ISDIGIT(sd->msg[2]) && B_ISDIGIT(sd->msg[3]) &&
748 bsendmsg(ua, "%s\n", sd->msg); /* pass them on to user */
752 /* Validate Slot: if scanning, otherwise Slot:Barcode */
753 p = strchr(sd->msg, ':');
755 /* Scanning -- require only valid slot */
756 Slot = atoi(sd->msg);
760 bsendmsg(ua, _("Invalid Slot number: %s\n"), sd->msg);
765 if (p && strlen(p) > 1) {
767 if (!is_an_integer(sd->msg) || (Slot=atoi(sd->msg)) <= 0) {
770 bsendmsg(ua, _("Invalid Slot number: %s\n"), sd->msg);
776 if (!is_volume_name_legal(ua, p)) {
779 bsendmsg(ua, _("Invalid Volume name: %s\n"), sd->msg);
784 /* Add Slot and VolumeName to list */
785 vl = (vol_list_t *)malloc(sizeof(vol_list_t));
789 p++; /* skip separator */
791 vl->VolName = bstrdup(p);
795 Dmsg2(100, "Add slot=%d Vol=%s to SD list.\n", vl->Slot, NPRT(vl->VolName));
800 /* Add new entry to end of list */
801 for (vol_list_t *tvl=vol_list; tvl; tvl=tvl->next) {
814 static void free_vol_list(vol_list_t *vol_list)
819 for (vl=vol_list; vl; ) {
832 * Check if this is a cleaning tape by comparing the Volume name
833 * with the Cleaning Prefix. If they match, this is a cleaning
836 static bool is_cleaning_tape(UAContext *ua, MEDIA_DBR *mr, POOL_DBR *pr)
838 /* Find Pool resource */
839 ua->jcr->pool = (POOL *)GetResWithName(R_POOL, pr->Name);
840 if (!ua->jcr->pool) {
841 bsendmsg(ua, _("Pool \"%s\" resource not found!\n"), pr->Name);
844 if (ua->jcr->pool->cleaning_prefix == NULL) {
847 Dmsg4(100, "CLNprefix=%s: Vol=%s: len=%d strncmp=%d\n",
848 ua->jcr->pool->cleaning_prefix, mr->VolumeName,
849 strlen(ua->jcr->pool->cleaning_prefix),
850 strncmp(mr->VolumeName, ua->jcr->pool->cleaning_prefix,
851 strlen(ua->jcr->pool->cleaning_prefix)));
852 return strncmp(mr->VolumeName, ua->jcr->pool->cleaning_prefix,
853 strlen(ua->jcr->pool->cleaning_prefix)) == 0;