]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/ua_update.c
- Add Rudolf Cejka's new rc-chio-changer.
[bacula/bacula] / bacula / src / dird / ua_update.c
index bf219264b814d414d4df531401924aa0cac2aebb..8737ecc61ada5ec564a2330c1e4be2c69211d8bc 100644 (file)
@@ -8,7 +8,7 @@
  *   Version $Id$
  */
 /*
-   Copyright (C) 2000-2005 Kern Sibbald
+   Copyright (C) 2000-2006 Kern Sibbald
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License
@@ -29,8 +29,8 @@
 extern char *list_pool;               /* in sql_cmds.c */
 
 /* Imported functions */
-void set_pooldbr_from_poolres(POOL_DBR *pr, POOL *pool, e_pool_op op);
-int update_slots(UAContext *ua);
+void update_slots(UAContext *ua);
+
 
 
 /* Forward referenced functions */
@@ -98,16 +98,16 @@ int update_cmd(UAContext *ua, const char *cmd)
 
 static void update_volstatus(UAContext *ua, const char *val, MEDIA_DBR *mr)
 {
-   POOLMEM *query = get_pool_memory(PM_MESSAGE);
+   POOL_MEM query(PM_MESSAGE);
    const char *kw[] = {
-      "Append",
-      "Archive",
-      "Disabled",
-      "Full",
-      "Used",
-      "Cleaning",
-      "Recycle",
-      "Read-Only",
+      N_("Append"),
+      N_("Archive"),
+      N_("Disabled"),
+      N_("Full"),
+      N_("Used"),
+      N_("Cleaning"),
+      N_("Recycle"),
+      N_("Read-Only"),
       NULL};
    bool found = false;
    int i;
@@ -125,128 +125,169 @@ static void update_volstatus(UAContext *ua, const char *val, MEDIA_DBR *mr)
       bstrncpy(mr->VolStatus, kw[i], sizeof(mr->VolStatus));
       Mmsg(query, "UPDATE Media SET VolStatus='%s' WHERE MediaId=%s",
          mr->VolStatus, edit_int64(mr->MediaId,ed1));
-      if (!db_sql_query(ua->db, query, NULL, NULL)) {
+      if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
          bsendmsg(ua, "%s", db_strerror(ua->db));
       } else {
          bsendmsg(ua, _("New Volume status is: %s\n"), mr->VolStatus);
       }
    }
-   free_pool_memory(query);
 }
 
 static void update_volretention(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
    char ed1[150], ed2[50];
-   POOLMEM *query;
+   POOL_MEM query(PM_MESSAGE);
    if (!duration_to_utime(val, &mr->VolRetention)) {
       bsendmsg(ua, _("Invalid retention period specified: %s\n"), val);
       return;
    }
-   query = get_pool_memory(PM_MESSAGE);
    Mmsg(query, "UPDATE Media SET VolRetention=%s WHERE MediaId=%s",
       edit_uint64(mr->VolRetention, ed1), edit_int64(mr->MediaId,ed2));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New retention period is: %s\n"),
          edit_utime(mr->VolRetention, ed1, sizeof(ed1)));
    }
-   free_pool_memory(query);
 }
 
 static void update_voluseduration(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
    char ed1[150], ed2[50];
-   POOLMEM *query;
+   POOL_MEM query(PM_MESSAGE);
 
    if (!duration_to_utime(val, &mr->VolUseDuration)) {
       bsendmsg(ua, _("Invalid use duration specified: %s\n"), val);
       return;
    }
-   query = get_pool_memory(PM_MESSAGE);
    Mmsg(query, "UPDATE Media SET VolUseDuration=%s WHERE MediaId=%s",
       edit_uint64(mr->VolUseDuration, ed1), edit_int64(mr->MediaId,ed2));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New use duration is: %s\n"),
          edit_utime(mr->VolUseDuration, ed1, sizeof(ed1)));
    }
-   free_pool_memory(query);
 }
 
 static void update_volmaxjobs(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
-   POOLMEM *query = get_pool_memory(PM_MESSAGE);
+   POOL_MEM query(PM_MESSAGE);
    char ed1[50];
    Mmsg(query, "UPDATE Media SET MaxVolJobs=%s WHERE MediaId=%s",
       val, edit_int64(mr->MediaId,ed1));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New max jobs is: %s\n"), val);
    }
-   free_pool_memory(query);
 }
 
 static void update_volmaxfiles(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
-   POOLMEM *query = get_pool_memory(PM_MESSAGE);
+   POOL_MEM query(PM_MESSAGE);
    char ed1[50];
    Mmsg(query, "UPDATE Media SET MaxVolFiles=%s WHERE MediaId=%s",
       val, edit_int64(mr->MediaId, ed1));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New max files is: %s\n"), val);
    }
-   free_pool_memory(query);
 }
 
 static void update_volmaxbytes(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
    uint64_t maxbytes;
    char ed1[50], ed2[50];
-   POOLMEM *query;
+   POOL_MEM query(PM_MESSAGE);
 
    if (!size_to_uint64(val, strlen(val), &maxbytes)) {
       bsendmsg(ua, _("Invalid max. bytes specification: %s\n"), val);
       return;
    }
-   query = get_pool_memory(PM_MESSAGE);
    Mmsg(query, "UPDATE Media SET MaxVolBytes=%s WHERE MediaId=%s",
       edit_uint64(maxbytes, ed1), edit_int64(mr->MediaId, ed2));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New Max bytes is: %s\n"), edit_uint64(maxbytes, ed1));
    }
-   free_pool_memory(query);
 }
 
 static void update_volrecycle(UAContext *ua, char *val, MEDIA_DBR *mr)
 {
    int recycle;
    char ed1[50];
-   POOLMEM *query;
+   POOL_MEM query(PM_MESSAGE);
    if (strcasecmp(val, _("yes")) == 0) {
       recycle = 1;
    } else if (strcasecmp(val, _("no")) == 0) {
       recycle = 0;
    } else {
-      bsendmsg(ua, _("Invalid value. It must by yes or no.\n"));
+      bsendmsg(ua, _("Invalid value. It must be yes or no.\n"));
       return;
    }
-   query = get_pool_memory(PM_MESSAGE);
    Mmsg(query, "UPDATE Media SET Recycle=%d WHERE MediaId=%s",
       recycle, edit_int64(mr->MediaId, ed1));
-   if (!db_sql_query(ua->db, query, NULL, NULL)) {
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
    } else {
       bsendmsg(ua, _("New Recycle flag is: %s\n"),
          mr->Recycle==1?_("yes"):_("no"));
    }
-   free_pool_memory(query);
+}
+
+static void update_volinchanger(UAContext *ua, char *val, MEDIA_DBR *mr)
+{
+   int InChanger;
+   char ed1[50];
+
+   POOL_MEM query(PM_MESSAGE);
+   if (strcasecmp(val, _("yes")) == 0) {
+      InChanger = 1;
+   } else if (strcasecmp(val, _("no")) == 0) {
+      InChanger = 0;
+   } else {
+      bsendmsg(ua, _("Invalid value. It must be yes or no.\n"));
+      return;
+   }
+   Mmsg(query, "UPDATE Media SET InChanger=%d WHERE MediaId=%s",
+      InChanger, edit_int64(mr->MediaId, ed1));
+   if (!db_sql_query(ua->db, query.c_str(), NULL, NULL)) {
+      bsendmsg(ua, "%s", db_strerror(ua->db));
+   } else {
+      bsendmsg(ua, _("New InChanger flag is: %s\n"),
+         mr->InChanger==1?_("yes"):_("no"));
+   }
+}
+
+
+static void update_volslot(UAContext *ua, char *val, MEDIA_DBR *mr)
+{
+   POOL_DBR pr;
+
+   memset(&pr, 0, sizeof(POOL_DBR));
+   pr.PoolId = mr->PoolId;
+   if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
+      bsendmsg(ua, "%s", db_strerror(ua->db));
+      return;
+   }
+   mr->Slot = atoi(val);
+   if (pr.MaxVols > 0 && mr->Slot > (int)pr.MaxVols) {
+      bsendmsg(ua, _("Invalid slot, it must be between 0 and MaxVols=%d\n"),
+         pr.MaxVols);
+      return;
+   }
+   /*
+    * Make sure to use db_update... rather than doing this directly,
+    *   so that any Slot is handled correctly.
+    */
+   if (!db_update_media_record(ua->jcr, ua->db, mr)) {
+      bsendmsg(ua, _("Error updating media record Slot: ERR=%s"), db_strerror(ua->db));
+   } else {
+      bsendmsg(ua, _("New Slot is: %d\n"), mr->Slot);
+   }
 }
 
 /* Modify the Pool in which this Volume is located */
@@ -289,7 +330,7 @@ static void update_vol_pool(UAContext *ua, char *val, MEDIA_DBR *mr, POOL_DBR *o
 /*
  * Refresh the Volume information from the Pool record
  */
-static void update_volfrompool(UAContext *ua, MEDIA_DBR *mr)
+static void update_vol_from_pool(UAContext *ua, MEDIA_DBR *mr)
 {
    POOL_DBR pr;
 
@@ -345,24 +386,29 @@ static int update_volume(UAContext *ua)
    POOLMEM *query;
    char ed1[130];
    bool done = false;
+   int i;
    const char *kw[] = {
-      N_("VolStatus"),                /* 0 */
-      N_("VolRetention"),             /* 1 */
-      N_("VolUse"),                   /* 2 */
-      N_("MaxVolJobs"),               /* 3 */
-      N_("MaxVolFiles"),              /* 4 */
-      N_("MaxVolBytes"),              /* 5 */
-      N_("Recycle"),                  /* 6 */
-      N_("Pool"),                     /* 7 */
-      N_("FromPool"),                 /* 8 */
-      N_("AllFromPool"),              /* 9 */
+      _("VolStatus"),                /* 0 */
+      _("VolRetention"),             /* 1 */
+      _("VolUse"),                   /* 2 */
+      _("MaxVolJobs"),               /* 3 */
+      _("MaxVolFiles"),              /* 4 */
+      _("MaxVolBytes"),              /* 5 */
+      _("Recycle"),                  /* 6 */
+      _("InChanger"),                /* 7 */
+      _("Slot"),                     /* 8 */
+      _("Pool"),                     /* 9 */
+      _("FromPool"),                 /* 10 */
+      _("AllFromPool"),              /* 11 !!! see below !!! */
       NULL };
 
-   for (int i=0; kw[i]; i++) {
+#define AllFromPool 11               /* keep this updated with above */
+
+   for (i=0; kw[i]; i++) {
       int j;
       POOL_DBR pr;
       if ((j=find_arg_with_value(ua, kw[i])) > 0) {
-         if (i != 9 && !select_media_dbr(ua, &mr)) {
+         if (i != AllFromPool && !select_media_dbr(ua, &mr)) {
             return 0;
          }
          switch (i) {
@@ -388,6 +434,12 @@ static int update_volume(UAContext *ua)
             update_volrecycle(ua, ua->argv[j], &mr);
             break;
          case 7:
+            update_volinchanger(ua, ua->argv[j], &mr);
+            break;
+         case 8:
+            update_volslot(ua, ua->argv[j], &mr);
+            break;
+         case 9:
             memset(&pr, 0, sizeof(POOL_DBR));
             pr.PoolId = mr.PoolId;
             if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
@@ -396,10 +448,10 @@ static int update_volume(UAContext *ua)
             }
             update_vol_pool(ua, ua->argv[j], &mr, &pr);
             break;
-         case 8:
-            update_volfrompool(ua, &mr);
+         case 10:
+            update_vol_from_pool(ua, &mr);
             return 1;
-         case 9:
+         case 11:
             update_all_vols_from_pool(ua);
             return 1;
          }
@@ -408,9 +460,6 @@ static int update_volume(UAContext *ua)
    }
 
    for ( ; !done; ) {
-      if (!select_media_dbr(ua, &mr)) {
-         return 0;
-      }
       bsendmsg(ua, _("Updating Volume \"%s\"\n"), mr.VolumeName);
       start_prompt(ua, _("Parameters to modify:\n"));
       add_prompt(ua, _("Volume Status"));
@@ -427,21 +476,28 @@ static int update_volume(UAContext *ua)
       add_prompt(ua, _("Volume from Pool"));
       add_prompt(ua, _("All Volumes from Pool"));
       add_prompt(ua, _("Done"));
-      switch (do_prompt(ua, "", _("Select parameter to modify"), NULL, 0)) {
+      i = do_prompt(ua, "", _("Select parameter to modify"), NULL, 0);  
+      /* For All Volumes from Pool we don't need a Volume record */
+      if (i != 12) {
+         if (!select_media_dbr(ua, &mr)) {  /* Get Volume record */
+            return 0;
+         }
+      }
+      switch (i) {
       case 0:                         /* Volume Status */
          /* Modify Volume Status */
          bsendmsg(ua, _("Current Volume status is: %s\n"), mr.VolStatus);
          start_prompt(ua, _("Possible Values are:\n"));
-         add_prompt(ua, "Append");      /* Better not translate these as */
-         add_prompt(ua, "Archive");     /* They are known in the database code */
-         add_prompt(ua, "Disabled");
-         add_prompt(ua, "Full");
-         add_prompt(ua, "Used");
-         add_prompt(ua, "Cleaning");
-         if (strcmp(mr.VolStatus, "Purged") == 0) {
-            add_prompt(ua, "Recycle");
+         add_prompt(ua, N_("Append")); 
+         add_prompt(ua, N_("Archive"));
+         add_prompt(ua, N_("Disabled"));
+         add_prompt(ua, N_("Full"));
+         add_prompt(ua, N_("Used"));
+         add_prompt(ua, N_("Cleaning"));
+         if (strcmp(mr.VolStatus, N_("Purged")) == 0) {
+            add_prompt(ua, N_("Recycle"));
          }
-         add_prompt(ua, "Read-Only");
+         add_prompt(ua, N_("Read-Only"));
          if (do_prompt(ua, "", _("Choose new Volume Status"), ua->cmd, sizeof(mr.VolStatus)) < 0) {
             return 1;
          }
@@ -500,36 +556,13 @@ static int update_volume(UAContext *ua)
          break;
 
       case 7:                         /* Slot */
-         int Slot;
-
-         memset(&pr, 0, sizeof(POOL_DBR));
-         pr.PoolId = mr.PoolId;
-         if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
-            bsendmsg(ua, "%s", db_strerror(ua->db));
-            return 0;
-         }
          bsendmsg(ua, _("Current Slot is: %d\n"), mr.Slot);
          if (!get_pint(ua, _("Enter new Slot: "))) {
             return 0;
          }
-         Slot = ua->pint32_val;
-         if (pr.MaxVols > 0 && Slot > (int)pr.MaxVols) {
-            bsendmsg(ua, _("Invalid slot, it must be between 0 and %d\n"),
-               pr.MaxVols);
-            break;
-         }
-         mr.Slot = Slot;
-         /*
-          * Make sure to use db_update... rather than doing this directly,
-          *   so that any Slot is handled correctly.
-          */
-         if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
-            bsendmsg(ua, _("Error updating media record Slot: ERR=%s"), db_strerror(ua->db));
-         } else {
-            bsendmsg(ua, _("New Slot is: %d\n"), mr.Slot);
-         }
+         update_volslot(ua, ua->cmd, &mr);
          break;
-
+         
       case 8:                         /* InChanger */
          bsendmsg(ua, _("Current InChanger flag is: %d\n"), mr.InChanger);
          if (!get_yesno(ua, _("Set InChanger flag? yes/no: "))) {
@@ -589,13 +622,13 @@ static int update_volume(UAContext *ua)
          return 1;
 
       case 11:
-         update_volfrompool(ua, &mr);
+         update_vol_from_pool(ua, &mr);
          return 1;
       case 12:
          update_all_vols_from_pool(ua);
          return 1;
       default:                        /* Done or error */
-         bsendmsg(ua, "Selection done.\n");
+         bsendmsg(ua, _("Selection terminated.\n"));
          return 1;
       }
    }