]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/ua_restore.c
- Put Dmsg() on inside if() to avoid calling subroutine.
[bacula/bacula] / bacula / src / dird / ua_restore.c
index a62b4ec9612dde4d852197529ced8a93e5ff9387..2615ad2d2e53f259bae76fe279c2aa6d0672c361 100644 (file)
  *
  *   Version $Id$
  */
-
 /*
-   Copyright (C) 2002-2004 Kern Sibbald and John Walker
+   Copyright (C) 2002-2005 Kern Sibbald
 
    This program is free software; you can redistribute it and/or
-   modify it under the terms of the GNU General Public License as
-   published by the Free Software Foundation; either version 2 of
-   the License, or (at your option) any later version.
+   modify it under the terms of the GNU General Public License
+   version 2 as ammended with additional clauses defined in the
+   file LICENSE in the main source directory.
 
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
-   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
-   General Public License for more details.
-
-   You should have received a copy of the GNU General Public
-   License along with this program; if not, write to the Free
-   Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
-   MA 02111-1307, USA.
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 
+   the file LICENSE for additional details.
 
  */
 
+
 #include "bacula.h"
 #include "dird.h"
 
@@ -48,6 +43,7 @@ extern char *uar_inc,          *uar_list_temp,   *uar_sel_jobid_temp;
 extern char *uar_sel_all_temp1,  *uar_sel_fileset, *uar_mediatype;
 extern char *uar_jobid_fileindex, *uar_dif,       *uar_sel_all_temp;
 extern char *uar_count_files,    *uar_jobids_fileindex;
+extern char *uar_jobid_fileindex_from_dir;
 
 
 struct NAME_LIST {
@@ -65,7 +61,7 @@ struct RESTORE_CTX {
    uint32_t TotalFiles;
    uint32_t JobId;
    char ClientName[MAX_NAME_LENGTH];
-   char last_jobid[10];
+   char last_jobid[20];
    POOLMEM *JobIds;                  /* User entered string of JobIds */
    STORE  *store;
    JOB *restore_job;
@@ -98,14 +94,16 @@ static void print_name_list(UAContext *ua, NAME_LIST *name_list);
 static int unique_name_list_handler(void *ctx, int num_fields, char **row);
 static void free_name_list(NAME_LIST *name_list);
 static void get_storage_from_mediatype(UAContext *ua, NAME_LIST *name_list, RESTORE_CTX *rx);
-static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date);
+static bool select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date);
 static bool build_directory_tree(UAContext *ua, RESTORE_CTX *rx);
 static void free_rx(RESTORE_CTX *rx);
 static void split_path_and_filename(RESTORE_CTX *rx, char *fname);
 static int jobid_fileindex_handler(void *ctx, int num_fields, char **row);
-static int insert_file_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *file,
+static bool insert_file_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *file,
+                                        char *date);
+static bool insert_dir_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *dir,
                                        char *date);
-static void insert_one_file(UAContext *ua, RESTORE_CTX *rx, char *date);
+static void insert_one_file_or_dir(UAContext *ua, RESTORE_CTX *rx, char *date, bool dir);
 static int get_client_name(UAContext *ua, RESTORE_CTX *rx);
 static int get_date(UAContext *ua, char *date, int date_len);
 static int count_handler(void *ctx, int num_fields, char **row);
@@ -149,8 +147,8 @@ int restore_cmd(UAContext *ua, const char *cmd)
    UnlockRes();
    if (!rx.restore_jobs) {
       bsendmsg(ua, _(
-        "No Restore Job Resource found. You must create at least\n"
-        "one before running this command.\n"));
+         "No Restore Job Resource found in bacula-dir.conf.\n"
+         "You must create at least one before running this command.\n"));
       goto bail_out;
    }
 
@@ -161,28 +159,29 @@ int restore_cmd(UAContext *ua, const char *cmd)
     *  add_findex()
     */
    switch (user_select_jobids_or_files(ua, &rx)) {
-   case 0:
+   case 0:                           /* error */
       goto bail_out;
-   case 1:                           /* select by jobid */
+   case 1:                           /* selected by jobid */
       if (!build_directory_tree(ua, &rx)) {
-        bsendmsg(ua, _("Restore not done.\n"));
+         bsendmsg(ua, _("Restore not done.\n"));
         goto bail_out;
       }
       break;
-   case 2:                           /* select by filename, no tree needed */
+   case 2:                           /* selected by filename, no tree needed */
       break;
    }
 
    if (rx.bsr->JobId) {
       if (!complete_bsr(ua, rx.bsr)) {  /* find Vol, SessId, SessTime from JobIds */
-        bsendmsg(ua, _("Unable to construct a valid BSR. Cannot continue.\n"));
+         bsendmsg(ua, _("Unable to construct a valid BSR. Cannot continue.\n"));
         goto bail_out;
       }
-      if (!write_bsr_file(ua, rx.bsr)) {
+      if (!(rx.selected_files = write_bsr_file(ua, rx.bsr))) {
+         bsendmsg(ua, _("No files selected to be restored.\n"));
         goto bail_out;
       }
       bsendmsg(ua, _("\n%u file%s selected to be restored.\n\n"), rx.selected_files,
-        rx.selected_files==1?"":"s");
+         rx.selected_files==1?"":"s");
    } else {
       bsendmsg(ua, _("No files selected to be restored.\n"));
       goto bail_out;
@@ -204,19 +203,22 @@ int restore_cmd(UAContext *ua, const char *cmd)
    }
 
    /* Build run command */
+   POOLMEM *fname = get_pool_memory(PM_MESSAGE);
+   make_unique_restore_filename(ua, &fname);
    if (rx.where) {
       Mmsg(ua->cmd,
-         "run job=\"%s\" client=\"%s\" storage=\"%s\" bootstrap=\"%s/restore.bsr\""
-         " where=\"%s\" files=%d catalog=\"%s\"",
-         job->hdr.name, rx.ClientName, rx.store?rx.store->hdr.name:"",
-         working_directory, rx.where, rx.selected_files, ua->catalog->hdr.name);
+          "run job=\"%s\" client=\"%s\" storage=\"%s\" bootstrap=\"%s\""
+          " where=\"%s\" files=%d catalog=\"%s\"",
+          job->hdr.name, rx.ClientName, rx.store?rx.store->hdr.name:"",
+         fname, rx.where, rx.selected_files, ua->catalog->hdr.name);
    } else {
       Mmsg(ua->cmd,
-         "run job=\"%s\" client=\"%s\" storage=\"%s\" bootstrap=\"%s/restore.bsr\""
-         " files=%d catalog=\"%s\"",
-         job->hdr.name, rx.ClientName, rx.store?rx.store->hdr.name:"",
-         working_directory, rx.selected_files, ua->catalog->hdr.name);
+          "run job=\"%s\" client=\"%s\" storage=\"%s\" bootstrap=\"%s\""
+          " files=%d catalog=\"%s\"",
+          job->hdr.name, rx.ClientName, rx.store?rx.store->hdr.name:"",
+         fname, rx.selected_files, ua->catalog->hdr.name);
    }
+   free_pool_memory(fname);
    if (find_arg(ua, _("yes")) > 0) {
       pm_strcat(ua->cmd, " yes");    /* pass it on to the run command */
    }
@@ -279,6 +281,10 @@ static int get_client_name(UAContext *ua, RESTORE_CTX *rx)
  * The first step in the restore process is for the user to
  *  select a list of JobIds from which he will subsequently
  *  select which files are to be restored.
+ *
+ *  Returns:  2  if filename list made
+ *           1  if jobid list made
+ *           0  on error
  */
 static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
 {
@@ -298,23 +304,31 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
       "Select backup for a client before a specified time",
       "Enter a list of files to restore",
       "Enter a list of files to restore before a specified time",
+      "Find the JobIds of the most recent backup for a client",
+      "Find the JobIds for a backup for a client before a specified time",
+      "Enter a list of directories to restore for given JobIds",
       "Cancel",
       NULL };
 
    const char *kw[] = {
+       /* These keywords are handled in a for loop */
       "jobid",     /* 0 */
       "current",   /* 1 */
       "before",    /* 2 */
       "file",      /* 3 */
-      "select",    /* 4 */
-      "pool",      /* 5 */
-      "all",       /* 6 */
-      "client",    /* 7 */
-      "storage",   /* 8 */
-      "fileset",   /* 9 */
-      "where",     /* 10 */
-      "yes",       /* 11 */
-      "done",      /* 12 */
+      "directory", /* 4 */
+      "select",    /* 5 */
+      "pool",      /* 6 */
+      "all",       /* 7 */
+
+      /* The keyword below are handled by individual arg lookups */
+      "client",    /* 8 */
+      "storage",   /* 9 */
+      "fileset",   /* 10 */
+      "where",     /* 11 */
+      "yes",       /* 12 */
+      "bootstrap", /* 13 */
+      "done",      /* 14 */
       NULL
    };
 
@@ -329,14 +343,14 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         }
       }
       if (!found_kw) {
-        bsendmsg(ua, _("Unknown keyword: %s\n"), ua->argk[i]);
+         bsendmsg(ua, _("Unknown keyword: %s\n"), ua->argk[i]);
         return 0;
       }
       /* Found keyword in kw[] list, process it */
       switch (j) {
       case 0:                           /* jobid */
         if (*rx->JobIds != 0) {
-           pm_strcat(rx->JobIds, ",");
+            pm_strcat(rx->JobIds, ",");
         }
         pm_strcat(rx->JobIds, ua->argv[i]);
         done = true;
@@ -347,13 +361,14 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         break;
       case 2:                           /* before */
         if (str_to_utime(ua->argv[i]) == 0) {
-           bsendmsg(ua, _("Improper date format: %s\n"), ua->argv[i]);
+            bsendmsg(ua, _("Improper date format: %s\n"), ua->argv[i]);
            return 0;
         }
         bstrncpy(date, ua->argv[i], sizeof(date));
         have_date = true;
         break;
       case 3:                           /* file */
+      case 4:                           /* dir */
         if (!have_date) {
            bstrutime(date, sizeof(date), time(NULL));
         }
@@ -361,14 +376,14 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
            return 0;
         }
         pm_strcpy(ua->cmd, ua->argv[i]);
-        insert_one_file(ua, rx, date);
+        insert_one_file_or_dir(ua, rx, date, j==4);
         if (rx->name_list.num_ids) {
            /* Check MediaType and select storage that corresponds */
            get_storage_from_mediatype(ua, &rx->name_list, rx);
            done = true;
         }
         break;
-      case 4:                           /* select */
+      case 5:                           /* select */
         if (!have_date) {
            bstrutime(date, sizeof(date), time(NULL));
         }
@@ -377,19 +392,19 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         }
         done = true;
         break;
-      case 5:                           /* pool specified */
+      case 6:                           /* pool specified */
         rx->pool = (POOL *)GetResWithName(R_POOL, ua->argv[i]);
         if (!rx->pool) {
-           bsendmsg(ua, _("Error: Pool resource \"%s\" does not exist.\n"), ua->argv[i]);
+            bsendmsg(ua, _("Error: Pool resource \"%s\" does not exist.\n"), ua->argv[i]);
            return 0;
         }
         if (!acl_access_ok(ua, Pool_ACL, ua->argv[i])) {
            rx->pool = NULL;
-           bsendmsg(ua, _("Error: Pool resource \"%s\" access not allowed.\n"), ua->argv[i]);
+            bsendmsg(ua, _("Error: Pool resource \"%s\" access not allowed.\n"), ua->argv[i]);
            return 0;
         }
         break;
-      case 6:                        /* all specified */
+      case 7:                        /* all specified */
         rx->all = true;
         break;
       /*
@@ -405,9 +420,9 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
 
    if (!done) {
       bsendmsg(ua, _("\nFirst you select one or more JobIds that contain files\n"
-                 "to be restored. You will be presented several methods\n"
-                 "of specifying the JobIds. Then you will be allowed to\n"
-                 "select which files from those JobIds are to be restored.\n\n"));
+                  "to be restored. You will be presented several methods\n"
+                  "of specifying the JobIds. Then you will be allowed to\n"
+                  "select which files from those JobIds are to be restored.\n\n"));
    }
 
    /* If choice not already made above, prompt */
@@ -432,13 +447,16 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         done = false;
         break;
       case 1:                        /* list where a file is saved */
-        if (!get_cmd(ua, _("Enter Filename (no path):"))) {
+        if (!get_client_name(ua, rx)) {
+           return 0;
+        }
+         if (!get_cmd(ua, _("Enter Filename (no path):"))) {
            return 0;
         }
         len = strlen(ua->cmd);
         fname = (char *)malloc(len * 2 + 1);
         db_escape_string(fname, ua->cmd, len);
-        Mmsg(rx->query, uar_file, fname);
+        Mmsg(rx->query, uar_file, rx->ClientName, fname);
         free(fname);
         gui_save = ua->jcr->gui;
         ua->jcr->gui = true;
@@ -447,13 +465,13 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         done = false;
         break;
       case 2:                        /* enter a list of JobIds */
-        if (!get_cmd(ua, _("Enter JobId(s), comma separated, to restore: "))) {
+         if (!get_cmd(ua, _("Enter JobId(s), comma separated, to restore: "))) {
            return 0;
         }
         pm_strcpy(rx->JobIds, ua->cmd);
         break;
       case 3:                        /* Enter an SQL list command */
-        if (!get_cmd(ua, _("Enter SQL list command: "))) {
+         if (!get_cmd(ua, _("Enter SQL list command: "))) {
            return 0;
         }
         gui_save = ua->jcr->gui;
@@ -481,18 +499,18 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         if (!get_client_name(ua, rx)) {
            return 0;
         }
-        bsendmsg(ua, _("Enter file names with paths, or < to enter a filename\n"
-                       "containg a list of file names with paths, and terminate\n"
-                       "them with a blank line.\n"));
+         bsendmsg(ua, _("Enter file names with paths, or < to enter a filename\n"
+                        "containg a list of file names with paths, and terminate\n"
+                        "them with a blank line.\n"));
         for ( ;; ) {
-           if (!get_cmd(ua, _("Enter full filename: "))) {
+            if (!get_cmd(ua, _("Enter full filename: "))) {
               return 0;
            }
            len = strlen(ua->cmd);
            if (len == 0) {
               break;
            }
-           insert_one_file(ua, rx, date);
+           insert_one_file_or_dir(ua, rx, date, false);
         }
         /* Check MediaType and select storage that corresponds */
         if (rx->name_list.num_ids) {
@@ -506,18 +524,18 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         if (!get_client_name(ua, rx)) {
            return 0;
         }
-        bsendmsg(ua, _("Enter file names with paths, or < to enter a filename\n"
-                       "containg a list of file names with paths, and terminate\n"
-                       "them with a blank line.\n"));
+         bsendmsg(ua, _("Enter file names with paths, or < to enter a filename\n"
+                        "containg a list of file names with paths, and terminate\n"
+                        "them with a blank line.\n"));
         for ( ;; ) {
-           if (!get_cmd(ua, _("Enter full filename: "))) {
+            if (!get_cmd(ua, _("Enter full filename: "))) {
               return 0;
            }
            len = strlen(ua->cmd);
            if (len == 0) {
               break;
            }
-           insert_one_file(ua, rx, date);
+           insert_one_file_or_dir(ua, rx, date, false);
         }
         /* Check MediaType and select storage that corresponds */
         if (rx->name_list.num_ids) {
@@ -525,8 +543,64 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
         }
         return 2;
 
+      case 8:                        /* Find JobIds for current backup */
+        bstrutime(date, sizeof(date), time(NULL));
+        if (!select_backups_before_date(ua, rx, date)) {
+           return 0;
+        }
+        done = false;
+        break;
+
+      case 9:                        /* Find JobIds for give date */
+        if (!get_date(ua, date, sizeof(date))) {
+           return 0;
+        }
+        if (!select_backups_before_date(ua, rx, date)) {
+           return 0;
+        }
+        done = false;
+        break;
+
+      case 10:                       /* Enter directories */
+        if (*rx->JobIds != 0) {
+            bsendmsg(ua, _("You have already seleted the following JobIds: %s\n"),
+              rx->JobIds);
+         } else if (get_cmd(ua, _("Enter JobId(s), comma separated, to restore: "))) {
+           if (*rx->JobIds != 0 && *ua->cmd) {
+               pm_strcat(rx->JobIds, ",");
+           }
+           pm_strcat(rx->JobIds, ua->cmd);
+        }
+         if (*rx->JobIds == 0 || *rx->JobIds == '.') {
+           return 0;                 /* nothing entered, return */
+        }
+        bstrutime(date, sizeof(date), time(NULL));
+        if (!get_client_name(ua, rx)) {
+           return 0;
+        }
+         bsendmsg(ua, _("Enter directory names with a trailing /, or < to enter a filename\n"
+                        "containg a list of directories and terminate\n"
+                        "them with a blank line.\n"));
+        for ( ;; ) {
+            if (!get_cmd(ua, _("Enter directory name: "))) {
+              return 0;
+           }
+           len = strlen(ua->cmd);
+           if (len == 0) {
+              break;
+           }
+            if (ua->cmd[len-1] != '/') {
+               strcat(ua->cmd, "/");
+           }
+           insert_one_file_or_dir(ua, rx, date, true);
+        }
+        /* Check MediaType and select storage that corresponds */
+        if (rx->name_list.num_ids) {
+           get_storage_from_mediatype(ua, &rx->name_list, rx);
+        }
+        return 2;
 
-      case 8:                        /* Cancel or quit */
+      case 11:                       /* Cancel or quit */
         return 0;
       }
    }
@@ -544,7 +618,7 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
    for (p=rx->JobIds; ; ) {
       int stat = get_next_jobid_from_list(&p, &JobId);
       if (stat < 0) {
-        bsendmsg(ua, _("Invalid JobId in list.\n"));
+         bsendmsg(ua, _("Invalid JobId in list.\n"));
         return 0;
       }
       if (stat == 0) {
@@ -555,12 +629,13 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
       }
       jr.JobId = JobId;
       if (!db_get_job_record(ua->jcr, ua->db, &jr)) {
-        bsendmsg(ua, _("Unable to get Job record for JobId=%u: ERR=%s\n"),
-           JobId, db_strerror(ua->db));
+        char ed1[50];
+         bsendmsg(ua, _("Unable to get Job record for JobId=%s: ERR=%s\n"),
+           edit_int64(JobId, ed1), db_strerror(ua->db));
         return 0;
       }
       if (!acl_access_ok(ua, Job_ACL, jr.Name)) {
-        bsendmsg(ua, _("No authorization. Job \"%s\" not selected.\n"),
+         bsendmsg(ua, _("No authorization. Job \"%s\" not selected.\n"),
            jr.Name);
         continue;
       }
@@ -575,7 +650,7 @@ static int user_select_jobids_or_files(UAContext *ua, RESTORE_CTX *rx)
 static int get_date(UAContext *ua, char *date, int date_len)
 {
    bsendmsg(ua, _("The restored files will the most current backup\n"
-                 "BEFORE the date you specify below.\n\n"));
+                  "BEFORE the date you specify below.\n\n"));
    for ( ;; ) {
       if (!get_cmd(ua, _("Enter date as YYYY-MM-DD HH:MM:SS :"))) {
         return 0;
@@ -592,7 +667,7 @@ static int get_date(UAContext *ua, char *date, int date_len)
 /*
  * Insert a single file, or read a list of files from a file
  */
-static void insert_one_file(UAContext *ua, RESTORE_CTX *rx, char *date)
+static void insert_one_file_or_dir(UAContext *ua, RESTORE_CTX *rx, char *date, bool dir)
 {
    FILE *ffd;
    char file[5000];
@@ -603,20 +678,31 @@ static void insert_one_file(UAContext *ua, RESTORE_CTX *rx, char *date)
    case '<':
       p++;
       if ((ffd = fopen(p, "r")) == NULL) {
-        bsendmsg(ua, _("Cannot open file %s: ERR=%s\n"),
-           p, strerror(errno));
+        berrno be;
+         bsendmsg(ua, _("Cannot open file %s: ERR=%s\n"),
+           p, be.strerror());
         break;
       }
       while (fgets(file, sizeof(file), ffd)) {
         line++;
-        if (!insert_file_into_findex_list(ua, rx, file, date)) {
-           bsendmsg(ua, _("Error occurred on line %d of %s\n"), line, p);
+        if (dir) {
+           if (!insert_dir_into_findex_list(ua, rx, file, date)) {
+               bsendmsg(ua, _("Error occurred on line %d of %s\n"), line, p);
+           }
+        } else {
+           if (!insert_file_into_findex_list(ua, rx, file, date)) {
+               bsendmsg(ua, _("Error occurred on line %d of %s\n"), line, p);
+           }
         }
       }
       fclose(ffd);
       break;
    default:
-      insert_file_into_findex_list(ua, rx, ua->cmd, date);
+      if (dir) {
+        insert_dir_into_findex_list(ua, rx, ua->cmd, date);
+      } else {
+        insert_file_into_findex_list(ua, rx, ua->cmd, date);
+      }
       break;
    }
 }
@@ -626,13 +712,16 @@ static void insert_one_file(UAContext *ua, RESTORE_CTX *rx, char *date)
  *   lookup the most recent backup in the catalog to get the JobId
  *   and FileIndex, then insert them into the findex list.
  */
-static int insert_file_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *file,
+static bool insert_file_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *file,
                                        char *date)
 {
+   char ed1[50];
+
    strip_trailing_junk(file);
    split_path_and_filename(rx, file);
    if (*rx->JobIds == 0) {
-      Mmsg(rx->query, uar_jobid_fileindex, date, rx->path, rx->fname, rx->ClientName);
+      Mmsg(rx->query, uar_jobid_fileindex, date, rx->path, rx->fname, 
+          rx->ClientName);
    } else {
       Mmsg(rx->query, uar_jobids_fileindex, rx->JobIds, date,
           rx->path, rx->fname, rx->ClientName);
@@ -645,20 +734,58 @@ static int insert_file_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *fi
    }
    if (!rx->found) {
       bsendmsg(ua, _("No database record found for: %s\n"), file);
-      return 0;
+      return true;
    }
-   rx->selected_files++;
    /*
     * Find the MediaTypes for this JobId and add to the name_list
     */
-   Mmsg(rx->query, uar_mediatype, rx->JobId);
+   Mmsg(rx->query, uar_mediatype, edit_int64(rx->JobId, ed1));
    if (!db_sql_query(ua->db, rx->query, unique_name_list_handler, (void *)&rx->name_list)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
-      return 0;
+      return false;
    }
-   return 1;
+   return true;
 }
 
+/*
+ * For a given path lookup the most recent backup in the catalog
+ * to get the JobId and FileIndexes of all files in that directory.
+ */
+static bool insert_dir_into_findex_list(UAContext *ua, RESTORE_CTX *rx, char *dir,
+                                       char *date)
+{
+   char ed1[50];
+
+   strip_trailing_junk(dir);
+   if (*rx->JobIds == 0) {
+      bsendmsg(ua, _("No JobId specified cannot continue.\n"));
+      return false;
+   } else {
+      Mmsg(rx->query, uar_jobid_fileindex_from_dir, rx->JobIds, 
+          dir, rx->ClientName);
+   }
+   rx->found = false;
+   /* Find and insert jobid and File Index */
+   if (!db_sql_query(ua->db, rx->query, jobid_fileindex_handler, (void *)rx)) {
+      bsendmsg(ua, _("Query failed: %s. ERR=%s\n"),
+        rx->query, db_strerror(ua->db));
+   }
+   if (!rx->found) {
+      bsendmsg(ua, _("No database record found for: %s\n"), dir);
+      return true;
+   }
+   /*
+    * Find the MediaTypes for this JobId and add to the name_list
+    */
+   Mmsg(rx->query, uar_mediatype, edit_int64(rx->JobId, ed1));
+   if (!db_sql_query(ua->db, rx->query, unique_name_list_handler, (void *)&rx->name_list)) {
+      bsendmsg(ua, "%s", db_strerror(ua->db));
+      return false;
+   }
+   return true;
+}
+
+
 static void split_path_and_filename(RESTORE_CTX *rx, char *name)
 {
    char *p, *f;
@@ -714,6 +841,7 @@ static bool build_directory_tree(UAContext *ua, RESTORE_CTX *rx)
    JobId_t JobId, last_JobId;
    char *p;
    bool OK = true;
+   char ed1[50];
 
    memset(&tree, 0, sizeof(TREE_CTX));
    /*
@@ -732,9 +860,9 @@ static bool build_directory_tree(UAContext *ua, RESTORE_CTX *rx)
    tree.FileEstimate = 0;
    if (get_next_jobid_from_list(&p, &JobId) > 0) {
       /* Use first JobId as estimate of the number of files to restore */
-      Mmsg(rx->query, uar_count_files, JobId);
+      Mmsg(rx->query, uar_count_files, edit_int64(JobId, ed1));
       if (!db_sql_query(ua->db, rx->query, count_handler, (void *)rx)) {
-        bsendmsg(ua, "%s\n", db_strerror(ua->db));
+         bsendmsg(ua, "%s\n", db_strerror(ua->db));
       }
       if (rx->found) {
         /* Add about 25% more than this job for over estimate */
@@ -743,53 +871,72 @@ static bool build_directory_tree(UAContext *ua, RESTORE_CTX *rx)
       }
    }
    for (p=rx->JobIds; get_next_jobid_from_list(&p, &JobId) > 0; ) {
+      char ed1[50];
 
       if (JobId == last_JobId) {
         continue;                    /* eliminate duplicate JobIds */
       }
       last_JobId = JobId;
-      bsendmsg(ua, _("\nBuilding directory tree for JobId %u ...  "), JobId);
+      bsendmsg(ua, _("\nBuilding directory tree for JobId %s ...  "), 
+        edit_int64(JobId, ed1));
       items++;
       /*
        * Find files for this JobId and insert them in the tree
        */
-      Mmsg(rx->query, uar_sel_files, JobId);
+      Mmsg(rx->query, uar_sel_files, edit_int64(JobId, ed1));
       if (!db_sql_query(ua->db, rx->query, insert_tree_handler, (void *)&tree)) {
-        bsendmsg(ua, "%s", db_strerror(ua->db));
+         bsendmsg(ua, "%s", db_strerror(ua->db));
       }
       /*
        * Find the MediaTypes for this JobId and add to the name_list
        */
-      Mmsg(rx->query, uar_mediatype, JobId);
+      Mmsg(rx->query, uar_mediatype, edit_int64(JobId, ed1));
       if (!db_sql_query(ua->db, rx->query, unique_name_list_handler, (void *)&rx->name_list)) {
-        bsendmsg(ua, "%s", db_strerror(ua->db));
+         bsendmsg(ua, "%s", db_strerror(ua->db));
       }
    }
-   char ec1[50];
-   bsendmsg(ua, "\n%d Job%s, %s files inserted into the tree%s.\n",
-      items, items==1?"":"s", edit_uint64_with_commas(tree.FileCount, ec1),
-      tree.all?" and marked for extraction":"");
+   if (tree.FileCount == 0) {
+      bsendmsg(ua, "\nThere were no files inserted into the tree, so file selection\n"
+         "is not possible.Most likely your retention policy pruned the files\n");
+      if (!get_yesno(ua, _("\nDo you want to restore all the files? (yes|no): "))) {
+        OK = false;
+      } else {
+        last_JobId = 0;
+        for (p=rx->JobIds; get_next_jobid_from_list(&p, &JobId) > 0; ) {
+            if (JobId == last_JobId) {
+               continue;                    /* eliminate duplicate JobIds */
+            }
+            add_findex_all(rx->bsr, JobId);
+         }
+         OK = true;
+      }
+   } else {
+      char ec1[50];
+      bsendmsg(ua, "\n%d Job%s, %s files inserted into the tree%s.\n",
+         items, items==1?"":"s", edit_uint64_with_commas(tree.FileCount, ec1),
+         tree.all?" and marked for extraction":"");
 
-   /* Check MediaType and select storage that corresponds */
-   get_storage_from_mediatype(ua, &rx->name_list, rx);
+      /* Check MediaType and select storage that corresponds */
+      get_storage_from_mediatype(ua, &rx->name_list, rx);
 
-   if (find_arg(ua, _("done")) < 0) {
-      /* Let the user interact in selecting which files to restore */
-      OK = user_select_files_from_tree(&tree);
-   }
+      if (find_arg(ua, _("done")) < 0) {
+        /* Let the user interact in selecting which files to restore */
+        OK = user_select_files_from_tree(&tree);
+      }
 
-   /*
-    * Walk down through the tree finding all files marked to be
-    *  extracted making a bootstrap file.
-    */
-   if (OK) {
-      for (TREE_NODE *node=first_tree_node(tree.root); node; node=next_tree_node(node)) {
-        Dmsg2(400, "FI=%d node=0x%x\n", node->FileIndex, node);
-        if (node->extract || node->extract_dir) {
-           Dmsg2(400, "type=%d FI=%d\n", node->type, node->FileIndex);
-           add_findex(rx->bsr, node->JobId, node->FileIndex);
-           if (node->extract && node->type != TN_NEWDIR) {
-              rx->selected_files++;  /* count only saved files */
+      /*
+       * Walk down through the tree finding all files marked to be
+       *  extracted making a bootstrap file.
+       */
+      if (OK) {
+        for (TREE_NODE *node=first_tree_node(tree.root); node; node=next_tree_node(node)) {
+            Dmsg2(400, "FI=%d node=0x%x\n", node->FileIndex, node);
+           if (node->extract || node->extract_dir) {
+               Dmsg2(400, "type=%d FI=%d\n", node->type, node->FileIndex);
+              add_findex(rx->bsr, node->JobId, node->FileIndex);
+              if (node->extract && node->type != TN_NEWDIR) {
+                 rx->selected_files++;  /* count only saved files */
+              }
            }
         }
       }
@@ -804,13 +951,13 @@ static bool build_directory_tree(UAContext *ua, RESTORE_CTX *rx)
  * This routine is used to get the current backup or a backup
  *   before the specified date.
  */
-static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date)
+static bool select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date)
 {
-   int stat = 0;
+   bool ok = false;
    FILESET_DBR fsr;
    CLIENT_DBR cr;
    char fileset_name[MAX_NAME_LENGTH];
-   char ed1[50];
+   char ed1[50], ed2[50];
    char pool_select[MAX_NAME_LENGTH];
    int i;
 
@@ -841,16 +988,17 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
    if (i >= 0) {
       bstrncpy(fsr.FileSet, ua->argv[i], sizeof(fsr.FileSet));
       if (!db_get_fileset_record(ua->jcr, ua->db, &fsr)) {
-        bsendmsg(ua, _("Error getting FileSet \"%s\": ERR=%s\n"), fsr.FileSet,
+         bsendmsg(ua, _("Error getting FileSet \"%s\": ERR=%s\n"), fsr.FileSet,
            db_strerror(ua->db));
         i = -1;
       }
    }
    if (i < 0) {                      /* fileset not found */
-      Mmsg(rx->query, uar_sel_fileset, cr.ClientId, cr.ClientId);
+      edit_int64(cr.ClientId, ed1);
+      Mmsg(rx->query, uar_sel_fileset, ed1, ed1);
       start_prompt(ua, _("The defined FileSet resources are:\n"));
       if (!db_sql_query(ua->db, rx->query, fileset_handler, (void *)ua)) {
-        bsendmsg(ua, "%s\n", db_strerror(ua->db));
+         bsendmsg(ua, "%s\n", db_strerror(ua->db));
       }
       if (do_prompt(ua, _("FileSet"), _("Select FileSet resource"),
                 fileset_name, sizeof(fileset_name)) < 0) {
@@ -859,9 +1007,9 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
 
       bstrncpy(fsr.FileSet, fileset_name, sizeof(fsr.FileSet));
       if (!db_get_fileset_record(ua->jcr, ua->db, &fsr)) {
-        bsendmsg(ua, _("Error getting FileSet record: %s\n"), db_strerror(ua->db));
-        bsendmsg(ua, _("This probably means you modified the FileSet.\n"
-                    "Continuing anyway.\n"));
+         bsendmsg(ua, _("Error getting FileSet record: %s\n"), db_strerror(ua->db));
+         bsendmsg(ua, _("This probably means you modified the FileSet.\n"
+                     "Continuing anyway.\n"));
       }
    }
 
@@ -872,14 +1020,16 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
       memset(&pr, 0, sizeof(pr));
       bstrncpy(pr.Name, rx->pool->hdr.name, sizeof(pr.Name));
       if (db_get_pool_record(ua->jcr, ua->db, &pr)) {
-        bsnprintf(pool_select, sizeof(pool_select), "AND Media.PoolId=%u ", pr.PoolId);
+         bsnprintf(pool_select, sizeof(pool_select), "AND Media.PoolId=%s ", 
+           edit_int64(pr.PoolId, ed1));
       } else {
-        bsendmsg(ua, _("Pool \"%s\" not found, using any pool.\n"), pr.Name);
+         bsendmsg(ua, _("Pool \"%s\" not found, using any pool.\n"), pr.Name);
       }
    }
 
    /* Find JobId of last Full backup for this client, fileset */
-   Mmsg(rx->query, uar_last_full, cr.ClientId, cr.ClientId, date, fsr.FileSet,
+   edit_int64(cr.ClientId, ed1);
+   Mmsg(rx->query, uar_last_full, ed1, ed1, date, fsr.FileSet,
         pool_select);
    if (!db_sql_query(ua->db, rx->query, NULL, NULL)) {
       bsendmsg(ua, "%s\n", db_strerror(ua->db));
@@ -905,7 +1055,7 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
 
    /* Now find most recent Differental Job after Full save, if any */
    Mmsg(rx->query, uar_dif, edit_uint64(rx->JobTDate, ed1), date,
-       cr.ClientId, fsr.FileSet, pool_select);
+       edit_int64(cr.ClientId, ed2), fsr.FileSet, pool_select);
    if (!db_sql_query(ua->db, rx->query, NULL, NULL)) {
       bsendmsg(ua, "%s\n", db_strerror(ua->db));
    }
@@ -921,7 +1071,7 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
 
    /* Now find all Incremental Jobs after Full/dif save */
    Mmsg(rx->query, uar_inc, edit_uint64(rx->JobTDate, ed1), date,
-       cr.ClientId, fsr.FileSet, pool_select);
+       edit_int64(cr.ClientId, ed2), fsr.FileSet, pool_select);
    if (!db_sql_query(ua->db, rx->query, NULL, NULL)) {
       bsendmsg(ua, "%s\n", db_strerror(ua->db));
    }
@@ -936,16 +1086,15 @@ static int select_backups_before_date(UAContext *ua, RESTORE_CTX *rx, char *date
    if (rx->JobIds[0] != 0) {
       /* Display a list of Jobs selected for this restore */
       db_list_sql_query(ua->jcr, ua->db, uar_list_temp, prtit, ua, 1, HORZ_LIST);
+      ok = true;
    } else {
       bsendmsg(ua, _("No jobs found.\n"));
    }
 
-   stat = 1;
-
 bail_out:
    db_sql_query(ua->db, uar_del_temp, NULL, NULL);
    db_sql_query(ua->db, uar_del_temp1, NULL, NULL);
-   return stat;
+   return ok;
 }
 
 
@@ -957,38 +1106,44 @@ static int get_next_jobid_from_list(char **p, uint32_t *JobId)
 
    jobid[0] = 0;
    for (int i=0; i<(int)sizeof(jobid); i++) {
-      if (*q == ',' || *q == 0) {
+      if (*q == 0) {
+        break;
+      } else if (*q == ',') {
         q++;
         break;
       }
       jobid[i] = *q++;
       jobid[i+1] = 0;
    }
-   if (jobid[0] == 0 || !is_a_number(jobid)) {
+   if (jobid[0] == 0) {
       return 0;
+   } else if (!is_a_number(jobid)) {
+      return -1;                     /* error */
    }
    *p = q;
-   *JobId = strtoul(jobid, NULL, 10);
+   *JobId = str_to_int64(jobid);
    return 1;
 }
 
 static int count_handler(void *ctx, int num_fields, char **row)
 {
    RESTORE_CTX *rx = (RESTORE_CTX *)ctx;
-   rx->JobId = atoi(row[0]);
+   rx->JobId = str_to_int64(row[0]);
    rx->found = true;
    return 0;
 }
 
 /*
  * Callback handler to get JobId and FileIndex for files
+ *   can insert more than one depending on the caller.
  */
 static int jobid_fileindex_handler(void *ctx, int num_fields, char **row)
 {
    RESTORE_CTX *rx = (RESTORE_CTX *)ctx;
-   rx->JobId = atoi(row[0]);
-   add_findex(rx->bsr, rx->JobId, atoi(row[1]));
+   rx->JobId = str_to_int64(row[0]);
+   add_findex(rx->bsr, rx->JobId, str_to_int64(row[1]));
    rx->found = true;
+   rx->selected_files++;
    return 0;
 }
 
@@ -1100,7 +1255,7 @@ static void get_storage_from_mediatype(UAContext *ua, NAME_LIST *name_list, REST
 
    if (name_list->num_ids > 1) {
       bsendmsg(ua, _("Warning, the JobIds that you selected refer to more than one MediaType.\n"
-        "Restore is not possible. The MediaTypes used are:\n"));
+         "Restore is not possible. The MediaTypes used are:\n"));
       print_name_list(ua, name_list);
       rx->store = select_storage_resource(ua);
       return;
@@ -1139,7 +1294,7 @@ static void get_storage_from_mediatype(UAContext *ua, NAME_LIST *name_list, REST
         }
       }
       if (store && (store != rx->store)) {
-        bsendmsg(ua, _("Warning default storage overridden by %s on command line.\n"),
+         bsendmsg(ua, _("Warning default storage overridden by %s on command line.\n"),
            store->hdr.name);
         rx->store = store;
       }
@@ -1151,8 +1306,8 @@ static void get_storage_from_mediatype(UAContext *ua, NAME_LIST *name_list, REST
 
    if (!rx->store) {
       bsendmsg(ua, _("\nWarning. Unable to find Storage resource for\n"
-        "MediaType \"%s\", needed by the Jobs you selected.\n"
-        "You will be allowed to select a Storage device later.\n"),
+         "MediaType \"%s\", needed by the Jobs you selected.\n"
+         "You will be allowed to select a Storage device later.\n"),
         name_list->name[0]);
    }
 }