]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/ua_cmds.c
Misc see kes-1.31 13May03
[bacula/bacula] / bacula / src / dird / ua_cmds.c
index 0c479ccd7c5be46b09e8b9102c92549ce5e86c0b..30d2476c79b63b2d66283b8d777b9f02cabefbb3 100644 (file)
@@ -76,6 +76,7 @@ static int delete_volume(UAContext *ua);
 static int delete_pool(UAContext *ua);
 static int mountcmd(UAContext *ua, char *cmd);
 static int updatecmd(UAContext *ua, char *cmd);
+static int waitcmd(UAContext *ua, char *cmd);
 
 int quitcmd(UAContext *ua, char *cmd);
 
@@ -111,6 +112,7 @@ static struct cmdstruct commands[] = {
  { N_("query"),      querycmd,     _("query catalog")},
  { N_("time"),       timecmd,      _("print current time")},
  { N_("exit"),       quitcmd,      _("exit = quit")},
+ { N_("wait"),       waitcmd,      _("wait until no jobs are running")},
             };
 #define comsize (sizeof(commands)/sizeof(struct cmdstruct))
 
@@ -132,14 +134,15 @@ int do_a_command(UAContext *ua, char *cmd)
    }
 
    len = strlen(ua->argk[0]);
-   for (i=0; i<comsize; i++)      /* search for command */
+   for (i=0; i<comsize; i++) {    /* search for command */
       if (strncasecmp(ua->argk[0],  _(commands[i].key), len) == 0) {
         stat = (*commands[i].func)(ua, cmd);   /* go execute command */
         found = 1;
         break;
       }
+   }
    if (!found) {
-      strcat(ua->UA_sock->msg, _(": is an illegal command\n"));
+      pm_strcat(&ua->UA_sock->msg, _(": is an illegal command\n"));
       ua->UA_sock->msglen = strlen(ua->UA_sock->msg);
       bnet_send(ua->UA_sock);
    }
@@ -199,15 +202,10 @@ static int addcmd(UAContext *ua, char *cmd)
    while (pr.MaxVols > 0 && pr.NumVols >= pr.MaxVols) {
       bsendmsg(ua, _("Pool already has maximum volumes = %d\n"), pr.MaxVols);
       for (;;) {
-         if (!get_cmd(ua, _("Enter new maximum (zero for unlimited): "))) {
+         if (!get_pint(ua, _("Enter new maximum (zero for unlimited): "))) {
            return 1;
         }
-        pr.MaxVols = atoi(ua->cmd);
-        if (pr.MaxVols < 0) {
-            bsendmsg(ua, _("Max vols must be zero or greater.\n"));
-           continue;
-        }
-        break;
+        pr.MaxVols = ua->pint32_val;
       }
    }
 
@@ -226,10 +224,10 @@ static int addcmd(UAContext *ua, char *cmd)
    for (;;) {
       char buf[100]; 
       sprintf(buf, _("Enter number of Volumes to create. 0=>fixed name. Max=%d: "), max);
-      if (!get_cmd(ua, buf)) {
+      if (!get_pint(ua, buf)) {
         return 1;
       }
-      num = atoi(ua->cmd);
+      num = ua->pint32_val;
       if (num < 0 || num > max) {
          bsendmsg(ua, _("The number must be between 0 and %d\n"), max);
         continue;
@@ -265,10 +263,10 @@ getVolName:
       strcat(name, "%04d");
 
       for (;;) {
-         if (!get_cmd(ua, _("Enter the starting number: "))) {
+         if (!get_pint(ua, _("Enter the starting number: "))) {
            return 1;
         }
-        startnum = atoi(ua->cmd);
+        startnum = ua->pint32_val;
         if (startnum < 1) {
             bsendmsg(ua, _("Start number must be greater than zero.\n"));
            continue;
@@ -281,10 +279,10 @@ getVolName:
    }
 
    if (store && store->autochanger) {
-      if (!get_cmd(ua, _("Enter slot (0 for none): "))) {
+      if (!get_pint(ua, _("Enter slot (0 for none): "))) {
         return 1;
       }
-      slot = atoi(ua->cmd);
+      slot = ua->pint32_val;
    }
           
    set_pool_dbr_defaults_in_media_dbr(&mr, &pr);
@@ -407,10 +405,7 @@ static int cancelcmd(UAContext *ua, char *cmd)
         return 1;
       }
       if (njobs == 1) {
-         if (!get_cmd(ua, _("Confirm cancel (yes/no): "))) {
-           return 1;
-        }
-         if (strcasecmp(ua->cmd, _("yes")) != 0) {
+         if (!get_yesno(ua, _("Confirm cancel (yes/no): ")) || ua->pint32_val == 0) {
            return 1;
         }
       }
@@ -423,6 +418,11 @@ static int cancelcmd(UAContext *ua, char *cmd)
      
    switch (jcr->JobStatus) {
    case JS_Created:
+   case JS_WaitJobRes:
+   case JS_WaitClientRes:
+   case JS_WaitStoreRes:
+   case JS_WaitMaxJobs:
+   case JS_WaitStartTime:
       set_jcr_job_status(jcr, JS_Canceled);
       bsendmsg(ua, _("JobId %d, Job %s marked to be canceled.\n"),
              jcr->JobId, jcr->Job);
@@ -434,40 +434,44 @@ static int cancelcmd(UAContext *ua, char *cmd)
         
    default:
       set_jcr_job_status(jcr, JS_Canceled);
+
       /* Cancel File daemon */
-      ua->jcr->client = jcr->client;
-      if (!connect_to_file_daemon(ua->jcr, 10, FDConnectTimeout, 1)) {
-         bsendmsg(ua, _("Failed to connect to File daemon.\n"));
-        free_jcr(jcr);
-        return 1;
-      }
-      Dmsg0(200, "Connected to file daemon\n");
-      fd = ua->jcr->file_bsock;
-      bnet_fsend(fd, "cancel Job=%s\n", jcr->Job);
-      while (bnet_recv(fd) >= 0) {
-         bsendmsg(ua, "%s", fd->msg);
+      if (jcr->file_bsock) {
+        ua->jcr->client = jcr->client;
+        if (!connect_to_file_daemon(ua->jcr, 10, FDConnectTimeout, 1)) {
+            bsendmsg(ua, _("Failed to connect to File daemon.\n"));
+           free_jcr(jcr);
+           return 1;
+        }
+         Dmsg0(200, "Connected to file daemon\n");
+        fd = ua->jcr->file_bsock;
+         bnet_fsend(fd, "cancel Job=%s\n", jcr->Job);
+        while (bnet_recv(fd) >= 0) {
+            bsendmsg(ua, "%s", fd->msg);
+        }
+        bnet_sig(fd, BNET_TERMINATE);
+        bnet_close(fd);
+        ua->jcr->file_bsock = NULL;
       }
-      bnet_sig(fd, BNET_TERMINATE);
-      bnet_close(fd);
-      ua->jcr->file_bsock = NULL;
 
       /* Cancel Storage daemon */
-      ua->jcr->store = jcr->store;
-      if (!connect_to_storage_daemon(ua->jcr, 10, SDConnectTimeout, 1)) {
-         bsendmsg(ua, _("Failed to connect to Storage daemon.\n"));
-        free_jcr(jcr);
-        return 1;
-      }
-      Dmsg0(200, "Connected to storage daemon\n");
-      sd = ua->jcr->store_bsock;
-      bnet_fsend(sd, "cancel Job=%s\n", jcr->Job);
-      while (bnet_recv(sd) >= 0) {
-         bsendmsg(ua, "%s", sd->msg);
+      if (jcr->store_bsock) {
+        ua->jcr->store = jcr->store;
+        if (!connect_to_storage_daemon(ua->jcr, 10, SDConnectTimeout, 1)) {
+            bsendmsg(ua, _("Failed to connect to Storage daemon.\n"));
+           free_jcr(jcr);
+           return 1;
+        }
+         Dmsg0(200, "Connected to storage daemon\n");
+        sd = ua->jcr->store_bsock;
+         bnet_fsend(sd, "cancel Job=%s\n", jcr->Job);
+        while (bnet_recv(sd) >= 0) {
+            bsendmsg(ua, "%s", sd->msg);
+        }
+        bnet_sig(sd, BNET_TERMINATE);
+        bnet_close(sd);
+        ua->jcr->store_bsock = NULL;
       }
-      bnet_sig(sd, BNET_TERMINATE);
-      bnet_close(sd);
-      ua->jcr->store_bsock = NULL;
-
    }
    free_jcr(jcr);
 
@@ -649,22 +653,13 @@ static int updatecmd(UAContext *ua, char *cmd)
  */             
 static int update_volume(UAContext *ua)
 {
-   POOL_DBR pr;
    MEDIA_DBR mr;
    POOLMEM *query;
    char ed1[30];
 
-   if (!select_pool_and_media_dbr(ua, &pr, &mr)) {
-      return 0;
-   }
 
    for (int done=0; !done; ) {
-      if (!db_get_media_record(ua->jcr, ua->db, &mr)) {
-        if (mr.MediaId != 0) {
-            bsendmsg(ua, _("Volume record for MediaId %d not found.\n"), mr.MediaId);
-        } else {
-            bsendmsg(ua, _("Volume record for %s not found.\n"), mr.VolumeName);
-        }
+      if (!select_media_dbr(ua, &mr)) {
         return 0;
       }
       bsendmsg(ua, _("Updating Volume \"%s\"\n"), mr.VolumeName);
@@ -689,6 +684,7 @@ static int update_volume(UAContext *ua)
          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");
         }
@@ -754,14 +750,10 @@ static int update_volume(UAContext *ua)
       case 3:                        /* Max Jobs */
         int32_t maxjobs;
          bsendmsg(ua, _("Current max jobs is: %u\n"), mr.MaxVolJobs);
-         if (!get_cmd(ua, _("Enter new Maximum Jobs: "))) {
+         if (!get_pint(ua, _("Enter new Maximum Jobs: "))) {
            return 0;
         }
-        maxjobs = atoi(ua->cmd);
-        if (maxjobs < 0) {
-            bsendmsg(ua, _("Invalid number, it must be 0 or greater\n"));
-           break;
-        } 
+        maxjobs = ua->pint32_val;
         query = get_pool_memory(PM_MESSAGE);
          Mmsg(&query, "UPDATE Media SET MaxVolJobs=%u WHERE MediaId=%u",
            maxjobs, mr.MediaId);
@@ -776,14 +768,10 @@ static int update_volume(UAContext *ua)
       case 4:                        /* Max Files */
         int32_t maxfiles;
          bsendmsg(ua, _("Current max files is: %u\n"), mr.MaxVolFiles);
-         if (!get_cmd(ua, _("Enter new Maximum Files: "))) {
+         if (!get_pint(ua, _("Enter new Maximum Files: "))) {
            return 0;
         }
-        maxfiles = atoi(ua->cmd);
-        if (maxfiles < 0) {
-            bsendmsg(ua, _("Invalid number, it must be 0 or greater\n"));
-           break;
-        } 
+        maxfiles = ua->pint32_val;
         query = get_pool_memory(PM_MESSAGE);
          Mmsg(&query, "UPDATE Media SET MaxVolFiles=%u WHERE MediaId=%u",
            maxfiles, mr.MediaId);
@@ -821,17 +809,10 @@ static int update_volume(UAContext *ua)
         int recycle;
          bsendmsg(ua, _("Current recycle flag is: %s\n"),
             mr.Recycle==1?_("yes"):_("no"));
-         if (!get_cmd(ua, _("Enter new Recycle status: "))) {
+         if (!get_yesno(ua, _("Enter new Recycle status: "))) {
            return 0;
         }
-         if (strcasecmp(ua->cmd, _("yes")) == 0) {
-           recycle = 1;
-         } else if (strcasecmp(ua->cmd, _("no")) == 0) {
-           recycle = 0;
-        } else {
-            bsendmsg(ua, _("Invalid recycle status specified.\n"));
-           break;
-        }
+        recycle = ua->pint32_val;
         query = get_pool_memory(PM_MESSAGE);
          Mmsg(&query, "UPDATE Media SET Recycle=%d WHERE MediaId=%u",
            recycle, mr.MediaId);
@@ -846,15 +827,20 @@ static int update_volume(UAContext *ua)
 
       case 7:                        /* Slot */
         int slot;
+        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 0;
+        }
          bsendmsg(ua, _("Current Slot is: %d\n"), mr.Slot);
-         if (!get_cmd(ua, _("Enter new Slot: "))) {
+         if (!get_pint(ua, _("Enter new Slot: "))) {
            return 0;
         }
-        slot = atoi(ua->cmd);
-        if (slot < 0) {
-            bsendmsg(ua, _("Invalid slot, it must be 0 or greater\n"));
-           break;
-        } else if (pr.MaxVols > 0 && slot >(int)pr.MaxVols) {
+        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;
@@ -875,18 +861,13 @@ static int update_volume(UAContext *ua)
          bsendmsg(ua, _("Warning changing Volume Files can result\n"
                         "in loss of data on your Volume\n\n"));
          bsendmsg(ua, _("Current Volume Files is: %u\n"), mr.VolFiles);
-         if (!get_cmd(ua, _("Enter new number of Files for Volume: "))) {
+         if (!get_pint(ua, _("Enter new number of Files for Volume: "))) {
            return 0;
         }
-        VolFiles = atoi(ua->cmd);
-        if (VolFiles < 0) {
-            bsendmsg(ua, _("Invalid number, it must be 0 or greater\n"));
-           break;
-        } 
+        VolFiles = ua->pint32_val;
         if (VolFiles != (int)(mr.VolFiles + 1)) {
             bsendmsg(ua, _("Normally, you should only increase Volume Files by one!\n"));
-            if (!get_cmd(ua, _("Continue? (yes/no): ")) || 
-                 strcasecmp(ua->cmd, "yes") != 0) {
+            if (!get_yesno(ua, _("Continue? (yes/no): ")) || ua->pint32_val == 0) {
               break;
            }
         }
@@ -1089,21 +1070,15 @@ static int setdebugcmd(UAContext *ua, char *cmd)
    Dmsg1(120, "setdebug:%s:\n", cmd);
 
    level = -1;
-   for (i=1; i<ua->argc; i++) {
-      if (strcasecmp(ua->argk[i], _("level")) == 0 && ua->argv[i]) {
-        level = atoi(ua->argv[i]);
-        break;
-      }
+   i = find_arg_with_value(ua, _("level"));
+   if (i >= 0) {
+      level = atoi(ua->argv[i]);
    }
    if (level < 0) {
-      if (!get_cmd(ua, _("Enter new debug level: "))) {
+      if (!get_pint(ua, _("Enter new debug level: "))) {
         return 1;
       }
-      level = atoi(ua->cmd);
-   }
-   if (level < 0) {
-      bsendmsg(ua, _("level cannot be negative.\n"));
-      return 1;
+      level = ua->pint32_val;
    }
 
    /* General debug? */
@@ -1250,20 +1225,19 @@ static int deletecmd(UAContext *ua, char *cmd)
  */
 static int delete_volume(UAContext *ua)
 {
-   POOL_DBR pr;
    MEDIA_DBR mr;
 
-   if (!select_pool_and_media_dbr(ua, &pr, &mr)) {
+   if (!select_media_dbr(ua, &mr)) {
       return 1;
    }
    bsendmsg(ua, _("\nThis command will delete volume %s\n"
       "and all Jobs saved on that volume from the Catalog\n"),
       mr.VolumeName);
 
-   if (!get_cmd(ua, _("Are you sure you want to delete this Volume? (yes/no): "))) {
+   if (!get_yesno(ua, _("Are you sure you want to delete this Volume? (yes/no): "))) {
       return 1;
    }
-   if (strcasecmp(ua->cmd, _("yes")) == 0) {
+   if (ua->pint32_val) {
       db_delete_media_record(ua->jcr, ua->db, &mr);
    }
    return 1;
@@ -1281,10 +1255,10 @@ static int delete_pool(UAContext *ua)
    if (!get_pool_dbr(ua, &pr)) {
       return 1;
    }
-   if (!get_cmd(ua, _("Are you sure you want to delete this Pool? (yes/no): "))) {
+   if (!get_yesno(ua, _("Are you sure you want to delete this Pool? (yes/no): "))) {
       return 1;
    }
-   if (strcasecmp(ua->cmd, _("yes")) == 0) {
+   if (ua->pint32_val) {
       db_delete_pool_record(ua->jcr, ua->db, &pr);
    }
    return 1;
@@ -1385,6 +1359,32 @@ int quitcmd(UAContext *ua, char *cmd)
    return 1;
 }
 
+/*
+ * Wait until no job is running 
+ */
+int waitcmd(UAContext *ua, char *cmd) 
+{
+   bmicrosleep(0, 200000);           /* let job actually start */
+   for (int running=1; running; ) {
+      running = 0;
+      lock_jcr_chain();
+      for (JCR *jcr=NULL; (jcr=get_next_jcr(jcr)); ) {
+        if (jcr->JobId != 0) {
+           running = 1;
+           free_locked_jcr(jcr);
+           break;
+        }
+        free_locked_jcr(jcr);
+      }
+      unlock_jcr_chain();
+      if (running) {
+        bmicrosleep(1, 0);
+      }
+   }
+   return 1;
+}
+
+
 static int helpcmd(UAContext *ua, char *cmd)
 {
    unsigned int i;