]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/ua_prune.c
Fix conio.h problem on Solaris
[bacula/bacula] / bacula / src / dird / ua_prune.c
index 389ac441df82d07939e98ce4d5772a82db705734..d72fe1d370854552dd1eed55afe3880e703be570 100644 (file)
@@ -4,10 +4,12 @@
  *     Applies retention periods
  *
  *     Kern Sibbald, February MMII
+ *
+ *   Version $Id$
  */
 
 /*
-   Copyright (C) 2002 Kern Sibbald and John Walker
+   Copyright (C) 2002-2004 Kern Sibbald and John Walker
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
 
 #include "bacula.h"
 #include "dird.h"
-#include "ua.h"
+
+/* Imported functions */
+int mark_media_purged(UAContext *ua, MEDIA_DBR *mr);
 
 /* Forward referenced functions */
-int prune_files(UAContext *ua, CLIENT *client);
-int prune_jobs(UAContext *ua, CLIENT *client);
-int prune_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr);
 
 
 #define MAX_DEL_LIST_LEN 1000000
 
-/*
- * Select JobIds for File deletion.
- */
-static char *select_job =
-   "SELECT JobId from Job "    
-   "WHERE StartDay < %s "
-   "AND ClientId=%d "
-   "AND PurgedFiles=0";
-
-/*
- * List of SQL commands terminated by NULL for deleting
- *  temporary tables and indicies 
- */
-static char *drop_deltabs[] = {
-   "DROP TABLE DelCandidates",
-   "DROP INDEX DelInx1",
-   NULL};
-
-/*
- * List of SQL commands to create temp table and indicies
- */
-static char *create_deltabs[] = {
-   "CREATE TABLE DelCandidates ("
-      "JobId INTEGER UNSIGNED NOT NULL, "
-      "PurgedFiles TINYINT, "
-      "FileSetId INTEGER UNSIGNED)",
-   "CREATE INDEX DelInx1 ON DelCandidates (JobId)",
-   NULL};
-
-
-/*
- * Fill candidates table with all Files subject to being deleted
- */
-static char *insert_delcand = 
-   "INSERT INTO DelCandidates "
-   "SELECT JobId, PurgedFiles, FileSetId FROM Job "
-   "WHERE StartDay < %s " 
-   "AND ClientId=%d";
+/* Imported variables */
+extern char *select_job;
+extern char *drop_deltabs[];
+extern char *create_deltabs[];
+extern char *insert_delcand;
+extern char *select_backup_del;
+extern char *select_verify_del;
+extern char *select_restore_del;
+extern char *select_admin_del;
+extern char *cnt_File;
+extern char *del_File;
+extern char *upd_Purged;
+extern char *cnt_DelCand;
+extern char *del_Job;
+extern char *del_JobMedia;
+extern char *cnt_JobMedia;
+extern char *sel_JobMedia;
 
-/*
- * Select files from the DelCandidates table that have a
- * more recent backup -- i.e. are not the only backup.
- * This is the list of files to delete.
- */
-static char *select_del =
-   "SELECT DelCandidates.JobId "
-   "FROM Job,DelCandidates "
-   "WHERE Job.StartDay >= %s "
-   "AND Job.ClientId=%d "
-   "AND Job.Level='F' "
-   "AND Job.JobStatus='T' "
-   "AND Job.FileSetId=DelCandidates.FileSetId";
 
 /* In memory list of JobIds */
 struct s_file_del_ctx {
@@ -131,7 +98,7 @@ static int count_handler(void *ctx, int num_fields, char **row)
 
 
 /*
- * Called here to count entries to be deleted 
+ * Called here to count the number of Jobs to be pruned
  */
 static int file_count_handler(void *ctx, int num_fields, char **row)
 {
@@ -161,8 +128,8 @@ static int job_delete_handler(void *ctx, int num_fields, char **row)
       del->JobId = (JobId_t *)brealloc(del->JobId, sizeof(JobId_t) * del->max_ids);
       del->PurgedFiles = (char *)brealloc(del->PurgedFiles, del->max_ids);
    }
-   del->JobId[del->num_ids] = (JobId_t)strtod(row[0], NULL);
-   del->PurgedFiles[del->num_ids++] = (char)atoi(row[0]);
+   del->JobId[del->num_ids] = (JobId_t)str_to_int64(row[0]);
+   del->PurgedFiles[del->num_ids++] = (char)str_to_int64(row[0]);
    return 0;
 }
 
@@ -178,67 +145,70 @@ static int file_delete_handler(void *ctx, int num_fields, char **row)
       del->JobId = (JobId_t *)brealloc(del->JobId, sizeof(JobId_t) *
         del->max_ids);
    }
-   del->JobId[del->num_ids++] = (JobId_t)strtod(row[0], NULL);
+   del->JobId[del->num_ids++] = (JobId_t)str_to_int64(row[0]);
    return 0;
 }
 
 /*
  *   Prune records from database
+ *
+ *    prune files (from) client=xxx
+ *    prune jobs (from) client=xxx
+ *    prune volume=xxx 
  */
-int prunecmd(UAContext *ua, char *cmd)
+int prunecmd(UAContext *ua, const char *cmd)
 {
    CLIENT *client;
    POOL_DBR pr;
    MEDIA_DBR mr;
+   int kw;
 
-   static char *keywords[] = {
-      N_("files"),
-      N_("jobs"),
-      N_("volume"),
+   static const char *keywords[] = {
+      N_("Files"),
+      N_("Jobs"),
+      N_("Volume"),
       NULL};
+
    if (!open_db(ua)) {
-      return 1;
+      return 01;
    }
-   switch (find_arg_keyword(ua, keywords)) {
-   case 0:
-      client = select_client_resource(ua);
+
+   /* First search args */
+   kw = find_arg_keyword(ua, keywords);  
+   if (kw < 0 || kw > 2) {
+      /* no args, so ask user */
+      kw = do_keyword_prompt(ua, _("Choose item to prune"), keywords);  
+   }      
+    
+   switch (kw) {
+   case 0:  /* prune files */
+      client = get_client_resource(ua);
+      if (!client || !confirm_retention(ua, &client->FileRetention, "File")) {
+        return 0;
+      }
       prune_files(ua, client);
       return 1;
-   case 1:
-      client = select_client_resource(ua);
-      prune_jobs(ua, client);
+   case 1:  /* prune jobs */
+      client = get_client_resource(ua);
+      if (!client || !confirm_retention(ua, &client->JobRetention, "Job")) {
+        return 0;
+      }
+      /* ****FIXME**** allow user to select JobType */
+      prune_jobs(ua, client, JT_BACKUP);
       return 1;
-   case 2:
+   case 2:  /* prune volume */
       if (!select_pool_and_media_dbr(ua, &pr, &mr)) {
-        return 1;
+        return 0;
+      }
+      if (!confirm_retention(ua, &mr.VolRetention, "Volume")) {
+        return 0;
       }
-      prune_volume(ua, &pr, &mr);
+      prune_volume(ua, &mr);
       return 1;
    default:
       break;
    }
-   switch (do_keyword_prompt(ua, _("Choose item to prune"), keywords)) {
-   case 0:
-      client = select_client_resource(ua);
-      if (!client) {
-        return 1;
-      }
-      prune_files(ua, client);
-      break;
-   case 1:
-      client = select_client_resource(ua);
-      if (!client) {
-        return 1;
-      }
-      prune_jobs(ua, client);
-      break;
-   case 2:
-      if (!select_pool_and_media_dbr(ua, &pr, &mr)) {
-        return 1;
-      }
-      prune_volume(ua, &pr, &mr);
-      return 1;
-   }
+
    return 1;
 }
 
@@ -249,53 +219,46 @@ int prunecmd(UAContext *ua, char *cmd)
  * temporary tables are needed. We simply make an in memory list of
  * the JobIds meeting the prune conditions, then delete all File records
  * pointing to each of those JobIds.
+ *
+ * This routine assumes you want the pruning to be done. All checking
+ *  must be done before calling this routine.
  */
 int prune_files(UAContext *ua, CLIENT *client)
 {
    struct s_file_del_ctx del;
-   char *query = (char *)get_pool_memory(PM_MESSAGE);
+   POOLMEM *query = get_pool_memory(PM_MESSAGE);
    int i;
-   struct tm tm;
-   uint64_t today, period;
-   time_t now;
+   utime_t now, period;
    CLIENT_DBR cr;
-   char ed1[50];
+   char ed1[50], ed2[50];
 
+   db_lock(ua->db);
    memset(&cr, 0, sizeof(cr));
-   strcpy(cr.Name, client->hdr.name);
-   if (!db_create_client_record(ua->db, &cr)) {
+   memset(&del, 0, sizeof(del));
+   bstrncpy(cr.Name, client->hdr.name, sizeof(cr.Name));
+   if (!db_create_client_record(ua->jcr, ua->db, &cr)) {
+      db_unlock(ua->db);
       return 0;
    }
 
    period = client->FileRetention;
-   now = time(NULL);
-   localtime_r(&now, &tm);
-   today = (uint64_t)(date_encode(tm.tm_year+1900, tm.tm_mon+1, tm.tm_mday) -
-       date_encode(2000, 1, 1));
+   now = (utime_t)time(NULL);
        
-   del.JobId = NULL;
-   del.num_ids = 0;
-   del.tot_ids = 0;
-   del.num_del = 0;
-   del.max_ids = 0;
-
-   Dmsg3(100, "Today=%d period=%d period=%d\n", (uint32_t)today, (uint32_t)period,          
-      (uint32_t)(period/(3600*24)));
-
-   Mmsg(&query, select_job, 
-       edit_uint64(today - period/(3600*24), ed1), cr.ClientId);
-
+   /* Select Jobs -- for counting */
+   Mmsg(query, select_job, edit_uint64(now - period, ed1), cr.ClientId);
    Dmsg1(050, "select sql=%s\n", query);
    if (!db_sql_query(ua->db, query, file_count_handler, (void *)&del)) {
-      bsendmsg(ua, "%s", db_strerror(ua->db));
+      if (ua->verbose) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+      }
       Dmsg0(050, "Count failed\n");
       goto bail_out;
    }
       
    if (del.tot_ids == 0) {
-      bsendmsg(ua, _("No Files found for client %s to prune from %s catalog.\n"),
-        client->hdr.name, client->catalog->hdr.name);
+      if (ua->verbose) {
+         bsendmsg(ua, _("No Files found to prune.\n"));
+      }
       goto bail_out;
    }
 
@@ -308,26 +271,35 @@ int prune_files(UAContext *ua, CLIENT *client)
 
    del.JobId = (JobId_t *)malloc(sizeof(JobId_t) * del.max_ids);
 
+   /* Now process same set but making a delete list */
    db_sql_query(ua->db, query, file_delete_handler, (void *)&del);
 
    for (i=0; i < del.num_ids; i++) {
+      struct s_count_ctx cnt;
       Dmsg1(050, "Delete JobId=%d\n", del.JobId[i]);
-      Mmsg(&query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, cnt_File, del.JobId[i]);
+      cnt.count = 0;
+      db_sql_query(ua->db, query, count_handler, (void *)&cnt);
+      del.tot_ids += cnt.count;
+      Mmsg(query, del_File, del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       /* 
        * Now mark Job as having files purged. This is necessary to
        * avoid having too many Jobs to process in future prunings. If
        * we don't do this, the number of JobId's in our in memory list
-       * will grow very large.
+       * could grow very large.
        */
-      Mmsg(&query, "UPDATE Job Set PurgedFiles=1 WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, upd_Purged, del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       Dmsg1(050, "Del sql=%s\n", query);
    }
-   bsendmsg(ua, _("%d Files for client %s pruned from %s catalog.\n"), del.num_ids,
-      client->hdr.name, client->catalog->hdr.name);
+   edit_uint64_with_commas(del.tot_ids, ed1);
+   edit_uint64_with_commas(del.num_ids, ed2);
+   bsendmsg(ua, _("Pruned %s Files from %s Jobs for client %s from catalog.\n"), 
+      ed1, ed2, client->hdr.name);
    
 bail_out:
+   db_unlock(ua->db);
    if (del.JobId) {
       free(del.JobId);
    }
@@ -367,40 +339,32 @@ static int create_temp_tables(UAContext *ua)
  * In other words, we never delete the only Job record that
  * contains a current backup of a FileSet. This prevents the
  * Volume from being recycled and destroying a current backup.
+ *
+ * For Verify Jobs, we do not delete the last InitCatalog.
+ *
+ * For Restore Jobs there are no restrictions.
  */
-int prune_jobs(UAContext *ua, CLIENT *client)
+int prune_jobs(UAContext *ua, CLIENT *client, int JobType)
 {
    struct s_job_del_ctx del;
    struct s_count_ctx cnt;
-   char *query = (char *)get_pool_memory(PM_MESSAGE);
+   POOLMEM *query = (char *)get_pool_memory(PM_MESSAGE);
    int i;
-   struct tm tm;
-   uint64_t today, period;
-   time_t now;
+   utime_t now, period;
    CLIENT_DBR cr;
    char ed1[50];
 
+   db_lock(ua->db);
    memset(&cr, 0, sizeof(cr));
-   strcpy(cr.Name, client->hdr.name);
-   if (!db_create_client_record(ua->db, &cr)) {
+   memset(&del, 0, sizeof(del));
+   bstrncpy(cr.Name, client->hdr.name, sizeof(cr.Name));
+   if (!db_create_client_record(ua->jcr, ua->db, &cr)) {
+      db_unlock(ua->db);
       return 0;
    }
 
    period = client->JobRetention;
-   now = time(NULL);
-   localtime_r(&now, &tm);
-   today = (uint64_t)(date_encode(tm.tm_year+1900, tm.tm_mon+1, tm.tm_mday) -
-       date_encode(2000, 1, 1));
-       
-
-   del.JobId = NULL;
-   del.num_ids = 0;
-   del.tot_ids = 0;
-   del.num_del = 0;
-   del.max_ids = 0;
-
-   Dmsg3(050, "Today=%d period=%d period=%d\n", (uint32_t)today, (uint32_t)period,          
-      (uint32_t)(period/(3600*24)));
+   now = (utime_t)time(NULL);
 
    /* Drop any previous temporary tables still there */
    drop_temp_tables(ua);
@@ -414,19 +378,20 @@ int prune_jobs(UAContext *ua, CLIENT *client)
     * Select all files that are older than the JobRetention period
     *  and stuff them into the "DeletionCandidates" table.
     */
-   edit_uint64(today - period/(3600*24), ed1);
-   Mmsg(&query, insert_delcand, ed1, cr.ClientId);
-
+   edit_uint64(now - period, ed1);
+   Mmsg(query, insert_delcand, (char)JobType, ed1, cr.ClientId);
    if (!db_sql_query(ua->db, query, NULL, (void *)NULL)) {
-      bsendmsg(ua, "%s", db_strerror(ua->db));
+      if (ua->verbose) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+      }
       Dmsg0(050, "insert delcand failed\n");
       goto bail_out;
    }
 
-   strcpy(query, "SELECT count(*) FROM DelCandidates");
-   
+   /* Count Files to be deleted */
+   pm_strcpy(query, cnt_DelCand);
    Dmsg1(100, "select sql=%s\n", query);
+   cnt.count = 0;
    if (!db_sql_query(ua->db, query, count_handler, (void *)&cnt)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
       Dmsg0(050, "Count failed\n");
@@ -434,8 +399,9 @@ int prune_jobs(UAContext *ua, CLIENT *client)
    }
       
    if (cnt.count == 0) {
-      bsendmsg(ua, _("No Jobs for client %s found to prune from %s catalog.\n"),
-        client->hdr.name, client->catalog->hdr.name);
+      if (ua->verbose) {
+         bsendmsg(ua, _("No Jobs found to prune.\n"));
+      }
       goto bail_out;
    }
 
@@ -447,8 +413,23 @@ int prune_jobs(UAContext *ua, CLIENT *client)
    del.JobId = (JobId_t *)malloc(sizeof(JobId_t) * del.max_ids);
    del.PurgedFiles = (char *)malloc(del.max_ids);
 
-   Mmsg(&query, select_del, ed1, cr.ClientId);
-   db_sql_query(ua->db, query, job_delete_handler, (void *)&del);
+   switch (JobType) {
+   case JT_BACKUP:
+      Mmsg(query, select_backup_del, ed1, ed1, cr.ClientId);
+      break;
+   case JT_RESTORE:
+      Mmsg(query, select_restore_del, ed1, ed1, cr.ClientId);
+      break;
+   case JT_VERIFY:
+      Mmsg(query, select_verify_del, ed1, ed1, cr.ClientId);
+      break;
+   case JT_ADMIN:
+      Mmsg(query, select_admin_del, ed1, ed1, cr.ClientId);
+      break;
+   }
+   if (!db_sql_query(ua->db, query, job_delete_handler, (void *)&del)) {
+      bsendmsg(ua, "%s", db_strerror(ua->db));
+   }
 
    /* 
     * OK, now we have the list of JobId's to be pruned, first check
@@ -458,24 +439,31 @@ int prune_jobs(UAContext *ua, CLIENT *client)
    for (i=0; i < del.num_ids; i++) {
       Dmsg1(050, "Delete JobId=%d\n", del.JobId[i]);
       if (!del.PurgedFiles[i]) {
-         Mmsg(&query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
-        db_sql_query(ua->db, query, NULL, (void *)NULL);
+        Mmsg(query, del_File, del.JobId[i]);
+        if (!db_sql_query(ua->db, query, NULL, (void *)NULL)) {
+            bsendmsg(ua, "%s", db_strerror(ua->db));
+        }
          Dmsg1(050, "Del sql=%s\n", query);
       }
 
-      Mmsg(&query, "DELETE FROM Job WHERE JobId=%d", del.JobId[i]);
-      db_sql_query(ua->db, query, NULL, (void *)NULL);
+      Mmsg(query, del_Job, del.JobId[i]);
+      if (!db_sql_query(ua->db, query, NULL, (void *)NULL)) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+      }
       Dmsg1(050, "Del sql=%s\n", query);
 
-      Mmsg(&query, "DELETE FROM JobMedia WHERE JobId=%d", del.JobId[i]);
-      db_sql_query(ua->db, query, NULL, (void *)NULL);
+      Mmsg(query, del_JobMedia, del.JobId[i]);
+      if (!db_sql_query(ua->db, query, NULL, (void *)NULL)) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+      }
       Dmsg1(050, "Del sql=%s\n", query);
    }
-   bsendmsg(ua, _("%d Jobs for client %s pruned from %s catalog.\n"), del.num_ids,
-      client->hdr.name, client->catalog->hdr.name);
+   bsendmsg(ua, _("Pruned %d %s for client %s from catalog.\n"), del.num_ids,
+      del.num_ids==1?_("Job"):_("Jobs"), client->hdr.name);
    
 bail_out:
    drop_temp_tables(ua);
+   db_unlock(ua->db);
    if (del.JobId) {
       free(del.JobId);
    }
@@ -487,14 +475,27 @@ bail_out:
 }
 
 /*
- * Prune volumes
+ * Prune a given Volume
  */
-int prune_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr)
+int prune_volume(UAContext *ua, MEDIA_DBR *mr)
 {
-   char *query = (char *)get_pool_memory(PM_MESSAGE);
+   POOLMEM *query = (char *)get_pool_memory(PM_MESSAGE);
    struct s_count_ctx cnt;
+   struct s_file_del_ctx del;
+   int i, stat = 0;
+   JOB_DBR jr;
+   utime_t now, period;
 
-   Mmsg(&query, "SELECT count(*) FROM JobMedia WHERE MediaId=%d", mr->MediaId);
+   db_lock(ua->db);
+   memset(&jr, 0, sizeof(jr));
+   memset(&del, 0, sizeof(del));
+
+   /*
+    * Find out how many Jobs remain on this Volume by 
+    *  counting the JobMedia records.
+    */
+   cnt.count = 0;
+   Mmsg(query, cnt_JobMedia, mr->MediaId);
    if (!db_sql_query(ua->db, query, count_handler, (void *)&cnt)) {
       bsendmsg(ua, "%s", db_strerror(ua->db));
       Dmsg0(050, "Count failed\n");
@@ -502,13 +503,76 @@ int prune_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr)
    }
       
    if (cnt.count == 0) {
-      bsendmsg(ua, "There are no Jobs associated with Volume %s. It is purged.\n",
-        mr->VolumeName);
+      if (strcmp(mr->VolStatus, "Purged") != 0 && verbose) {
+         bsendmsg(ua, "There are no Jobs associated with Volume \"%s\". Marking it purged.\n",
+           mr->VolumeName);
+      }
+      stat = mark_media_purged(ua, mr);
+      goto bail_out;
+   }
+
+   if (cnt.count < MAX_DEL_LIST_LEN) {
+      del.max_ids = cnt.count + 1;
    } else {
-      bsendmsg(ua, "There are still %d Jobs on Volume %s. It is not purged.\n",
-        cnt.count, mr->VolumeName);
+      del.max_ids = MAX_DEL_LIST_LEN; 
+   }
+
+   /* 
+    * Now get a list of JobIds for Jobs written to this Volume
+    *  Could optimize here by adding JobTDate > (now - period).
+    */
+   del.JobId = (JobId_t *)malloc(sizeof(JobId_t) * del.max_ids);
+   Mmsg(query, sel_JobMedia, mr->MediaId);
+   if (!db_sql_query(ua->db, query, file_delete_handler, (void *)&del)) {
+      if (ua->verbose) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+      }
+      Dmsg0(050, "Count failed\n");
+      goto bail_out;
+   }
+
+   /* Use Volume Retention to prune Jobs and their Files */
+   period = mr->VolRetention;
+   now = (utime_t)time(NULL);
+
+   Dmsg3(200, "Now=%d period=%d now-period=%d\n", (int)now, (int)period,
+      (int)(now-period));
+
+   for (i=0; i < del.num_ids; i++) {
+      jr.JobId = del.JobId[i];
+      if (!db_get_job_record(ua->jcr, ua->db, &jr)) {
+        continue;
+      }
+      Dmsg2(200, "Looking at %s JobTdate=%d\n", jr.Job, (int)jr.JobTDate);
+      if (jr.JobTDate >= (now - period)) {
+        continue;
+      }
+      Dmsg2(200, "Delete JobId=%d Job=%s\n", del.JobId[i], jr.Job);
+      Mmsg(query, del_File, del.JobId[i]);
+      db_sql_query(ua->db, query, NULL, (void *)NULL);
+      Mmsg(query, del_Job, del.JobId[i]);
+      db_sql_query(ua->db, query, NULL, (void *)NULL);
+      Mmsg(query, del_JobMedia, del.JobId[i]);
+      db_sql_query(ua->db, query, NULL, (void *)NULL);
+      Dmsg1(050, "Del sql=%s\n", query);
+      del.num_del++;
+   }
+   if (del.JobId) {
+      free(del.JobId);
+   }
+   if (ua->verbose && del.num_del != 0) {
+      bsendmsg(ua, _("Pruned %d %s on Volume \"%s\" from catalog.\n"), del.num_del,
+         del.num_del == 1 ? "Job" : "Jobs", mr->VolumeName);
+   }
+
+   /* If purged, mark it so */
+   if (del.num_ids == del.num_del) {
+      Dmsg0(200, "Volume is purged.\n");
+      stat = mark_media_purged(ua, mr);
    }
-bail_out:   
+
+bail_out:
+   db_unlock(ua->db);
    free_pool_memory(query);
-   return 1;
+   return stat;
 }