]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/dird/ua_purge.c
Fix conio.h problem on Solaris
[bacula/bacula] / bacula / src / dird / ua_purge.c
index b2c4a502151f05ea1f0937872653a6e95c068a31..87d43b3d76d5d282f039e572167689c98e4619f8 100644 (file)
@@ -8,11 +8,11 @@
  *
  *     Kern Sibbald, February MMII
  *
- *     $Id$
+ *   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"
 
 /* Forward referenced functions */
-int purge_files_from_client(UAContext *ua, CLIENT *client);
-int purge_jobs_from_client(UAContext *ua, CLIENT *client);
-void purge_files_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr );
-void purge_jobs_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr);
-void purge_files_from_job(UAContext *ua, JOB_DBR *jr);
-static int mark_media_purged(UAContext *ua, MEDIA_DBR *mr);
+static int purge_files_from_client(UAContext *ua, CLIENT *client);
+static int purge_jobs_from_client(UAContext *ua, CLIENT *client);
 
+void purge_files_from_volume(UAContext *ua, MEDIA_DBR *mr );
+int purge_jobs_from_volume(UAContext *ua, MEDIA_DBR *mr);
+void purge_files_from_job(UAContext *ua, JOB_DBR *jr);
+int mark_media_purged(UAContext *ua, MEDIA_DBR *mr);
 
 #define MAX_DEL_LIST_LEN 1000000
 
 
-static char *select_jobsfiles_from_client =
+static const char *select_jobsfiles_from_client =
    "SELECT JobId FROM Job "
    "WHERE ClientId=%d "
    "AND PurgedFiles=0";
 
-static char *select_jobs_from_client =
+static const char *select_jobs_from_client =
    "SELECT JobId, PurgedFiles FROM Job "
    "WHERE ClientId=%d";
 
@@ -133,8 +132,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;
 }
 
@@ -150,7 +149,7 @@ 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;
 }
 
@@ -162,35 +161,37 @@ static int file_delete_handler(void *ctx, int num_fields, char **row)
  *
  *  N.B. Not all above is implemented yet.
  */
-int purgecmd(UAContext *ua, char *cmd)
+int purgecmd(UAContext *ua, const char *cmd)
 {
+   int i;
    CLIENT *client;
    MEDIA_DBR mr;
-   POOL_DBR pr;
    JOB_DBR  jr;
-   static char *keywords[] = {
+   static const char *keywords[] = {
       N_("files"),
       N_("jobs"),
+      N_("volume"),
       NULL};
 
-   static char *files_keywords[] = {
+   static const char *files_keywords[] = {
       N_("Job"),
       N_("JobId"),
       N_("Client"),
       N_("Volume"),
       NULL};
 
-   static char *jobs_keywords[] = {
+   static const char *jobs_keywords[] = {
       N_("Client"),
       N_("Volume"),
       NULL};
       
    bsendmsg(ua, _(
-      "This command is DANGEROUS!!!\n"
+      "\nThis command is can be DANGEROUS!!!\n\n"
       "It purges (deletes) all Files from a Job,\n"
       "JobId, Client or Volume; or it purges (deletes)\n"
-      "all Jobs from a Client or Volume. Normally you\n" 
-      "should use the PRUNE command instead.\n"));
+      "all Jobs from a Client or Volume without regard\n"
+      "for retention periods. Normally you should use the\n" 
+      "PRUNE command, which respects retention periods.\n"));
 
    if (!open_db(ua)) {
       return 1;
@@ -200,18 +201,20 @@ int purgecmd(UAContext *ua, char *cmd)
    case 0:
       switch(find_arg_keyword(ua, files_keywords)) {
       case 0:                        /* Job */
-      case 1:
+      case 1:                        /* JobId */
         if (get_job_dbr(ua, &jr)) {
            purge_files_from_job(ua, &jr);
         }
         return 1;
       case 2:                        /* client */
-        client = select_client_resource(ua);
-        purge_files_from_client(ua, client);
+        client = get_client_resource(ua);
+        if (client) {
+           purge_files_from_client(ua, client);
+        }
         return 1;
-      case 3:
-        if (select_pool_and_media_dbr(ua, &pr, &mr)) {
-           purge_files_from_volume(ua, &pr, &mr);
+      case 3:                        /* Volume */
+        if (select_media_dbr(ua, &mr)) {
+           purge_files_from_volume(ua, &mr);
         }
         return 1;
       }
@@ -219,46 +222,61 @@ int purgecmd(UAContext *ua, char *cmd)
    case 1:
       switch(find_arg_keyword(ua, jobs_keywords)) {
       case 0:                        /* client */
-        client = select_client_resource(ua);
-        purge_jobs_from_client(ua, client);
+        client = get_client_resource(ua);
+        if (client) {
+           purge_jobs_from_client(ua, client);
+        }
         return 1;
-      case 1:
-        if (select_pool_and_media_dbr(ua, &pr, &mr)) {
-           purge_jobs_from_volume(ua, &pr, &mr);
+      case 1:                        /* Volume */
+        if (select_media_dbr(ua, &mr)) {
+           purge_jobs_from_volume(ua, &mr);
         }
         return 1;
       }
+   /* Volume */
+   case 2:
+      while ((i=find_arg(ua, _("volume"))) >= 0) {
+        if (select_media_dbr(ua, &mr)) {
+           purge_jobs_from_volume(ua, &mr);
+        }
+        *ua->argk[i] = 0;            /* zap keyword already seen */
+         bsendmsg(ua, "\n");
+      }
+      return 1;
    default:
       break;
    }
    switch (do_keyword_prompt(ua, _("Choose item to purge"), keywords)) {
-   case 0:
-      client = select_client_resource(ua);
-      if (!client) {
-        return 1;
+   case 0:                           /* files */
+      client = get_client_resource(ua);
+      if (client) {
+        purge_files_from_client(ua, client);
       }
-      purge_files_from_client(ua, client);
       break;
-   case 1:
-      client = select_client_resource(ua);
-      if (!client) {
-        return 1;
+   case 1:                           /* jobs */
+      client = get_client_resource(ua);
+      if (client) {
+        purge_jobs_from_client(ua, client);
+      }
+      break;
+   case 2:                           /* Volume */
+      if (select_media_dbr(ua, &mr)) {
+        purge_jobs_from_volume(ua, &mr);
       }
-      purge_jobs_from_client(ua, client);
       break;
    }
    return 1;
 }
 
 /*
- * Prune File records from the database. For any Job which
+ * Purge File records from the database. For any Job which
  * is older than the retention period, we unconditionally delete
  * all File records for that Job.  This is simple enough that no
  * 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.
  */
-int purge_files_from_client(UAContext *ua, CLIENT *client)
+static int purge_files_from_client(UAContext *ua, CLIENT *client)
 {
    struct s_file_del_ctx del;
    char *query = (char *)get_pool_memory(PM_MESSAGE);
@@ -268,12 +286,12 @@ int purge_files_from_client(UAContext *ua, CLIENT *client)
    memset(&cr, 0, sizeof(cr));
    memset(&del, 0, sizeof(del));
 
-   strcpy(cr.Name, client->hdr.name);
-   if (!db_create_client_record(ua->db, &cr)) {
+   bstrncpy(cr.Name, client->hdr.name, sizeof(cr.Name));
+   if (!db_create_client_record(ua->jcr, ua->db, &cr)) {
       return 0;
    }
-
-   Mmsg(&query, select_jobsfiles_from_client, cr.ClientId);
+   bsendmsg(ua, _("Begin purging files for Client \"%s\"\n"), cr.Name);
+   Mmsg(query, select_jobsfiles_from_client, cr.ClientId);
 
    Dmsg1(050, "select sql=%s\n", query);
  
@@ -302,7 +320,7 @@ int purge_files_from_client(UAContext *ua, CLIENT *client)
 
    for (i=0; i < del.num_ids; i++) {
       Dmsg1(050, "Delete JobId=%d\n", del.JobId[i]);
-      Mmsg(&query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       /* 
        * Now mark Job as having files purged. This is necessary to
@@ -310,11 +328,11 @@ int purge_files_from_client(UAContext *ua, CLIENT *client)
        * we don't do this, the number of JobId's in our in memory list
        * will grow very large.
        */
-      Mmsg(&query, "UPDATE Job Set PurgedFiles=1 WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "UPDATE Job Set PurgedFiles=1 WHERE JobId=%d", 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 purged from %s catalog.\n"), del.num_ids,
+   bsendmsg(ua, _("%d Files for client \"%s\" purged from %s catalog.\n"), del.num_ids,
       client->hdr.name, client->catalog->hdr.name);
    
 bail_out:
@@ -328,14 +346,14 @@ bail_out:
 
 
 /*
- * Purging Jobs is a bit more complicated than purging Files
- * because we delete Job records only if there is a more current
- * backup of the FileSet. Otherwise, we keep the Job record.
- * 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.
+ * Purge Job records from the database. For any Job which
+ * is older than the retention period, we unconditionally delete
+ * it and all File records for that Job.  This is simple enough that no
+ * temporary tables are needed. We simply make an in memory list of
+ * the JobIds meeting the prune conditions, then delete the Job,
+ * Files, and JobMedia records in that list.
  */
-int purge_jobs_from_client(UAContext *ua, CLIENT *client)
+static int purge_jobs_from_client(UAContext *ua, CLIENT *client)
 {
    struct s_job_del_ctx del;
    char *query = (char *)get_pool_memory(PM_MESSAGE);
@@ -345,12 +363,13 @@ int purge_jobs_from_client(UAContext *ua, CLIENT *client)
    memset(&cr, 0, sizeof(cr));
    memset(&del, 0, sizeof(del));
 
-   strcpy(cr.Name, client->hdr.name);
-   if (!db_create_client_record(ua->db, &cr)) {
+   bstrncpy(cr.Name, client->hdr.name, sizeof(cr.Name));
+   if (!db_create_client_record(ua->jcr, ua->db, &cr)) {
       return 0;
    }
 
-   Mmsg(&query, select_jobs_from_client, cr.ClientId);
+   bsendmsg(ua, _("Begin purging jobs from Client \"%s\"\n"), cr.Name);
+   Mmsg(query, select_jobs_from_client, cr.ClientId);
 
    Dmsg1(050, "select sql=%s\n", query);
  
@@ -386,16 +405,16 @@ int purge_jobs_from_client(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]);
+         Mmsg(query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
         db_sql_query(ua->db, query, NULL, (void *)NULL);
          Dmsg1(050, "Del sql=%s\n", query);
       }
 
-      Mmsg(&query, "DELETE FROM Job WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM Job WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       Dmsg1(050, "Del sql=%s\n", query);
 
-      Mmsg(&query, "DELETE FROM JobMedia WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM JobMedia WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       Dmsg1(050, "Del sql=%s\n", query);
    }
@@ -417,30 +436,46 @@ void purge_files_from_job(UAContext *ua, JOB_DBR *jr)
 {
    char *query = (char *)get_pool_memory(PM_MESSAGE);
    
-   Mmsg(&query, "DELETE FROM File WHERE JobId=%d", jr->JobId);
+   Mmsg(query, "DELETE FROM File WHERE JobId=%u", jr->JobId);
    db_sql_query(ua->db, query, NULL, (void *)NULL);
 
-   Mmsg(&query, "UPDATE Job Set PurgedFiles=1 WHERE JobId=%d", jr->JobId);
+   Mmsg(query, "UPDATE Job Set PurgedFiles=1 WHERE JobId=%u", jr->JobId);
    db_sql_query(ua->db, query, NULL, (void *)NULL);
 
    free_pool_memory(query);
 }
 
-void purge_files_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr ) 
+void purge_files_from_volume(UAContext *ua, MEDIA_DBR *mr ) 
 {} /* ***FIXME*** implement */
 
-void purge_jobs_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr) 
+/*
+ * Returns: 1 if Volume purged
+ *         0 if Volume not purged
+ */
+int purge_jobs_from_volume(UAContext *ua, MEDIA_DBR *mr) 
 {
    char *query = (char *)get_pool_memory(PM_MESSAGE);
    struct s_count_ctx cnt;
    struct s_file_del_ctx del;
-   int i;
+   int i, stat = 0;
    JOB_DBR jr;
 
+   stat = strcmp(mr->VolStatus, "Append") == 0 || 
+          strcmp(mr->VolStatus, "Full")   == 0 ||
+          strcmp(mr->VolStatus, "Used")   == 0 || 
+          strcmp(mr->VolStatus, "Error")  == 0; 
+   if (!stat) {
+      bsendmsg(ua, "\n");
+      bsendmsg(ua, _("Volume \"%s\" has VolStatus \"%s\" and cannot be purged.\n"
+                     "The VolStatus must be: Append, Full, Used, or Error to be purged.\n"),
+                    mr->VolumeName, mr->VolStatus);
+      goto bail_out;
+   }
+   
    memset(&jr, 0, sizeof(jr));
    memset(&del, 0, sizeof(del));
    cnt.count = 0;
-   Mmsg(&query, "SELECT count(*) FROM JobMedia WHERE MediaId=%d", mr->MediaId);
+   Mmsg(query, "SELECT count(*) FROM JobMedia WHERE MediaId=%d", 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");
@@ -448,9 +483,10 @@ void purge_jobs_from_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",
+      bsendmsg(ua, "There are no Jobs associated with Volume \"%s\". Marking it purged.\n",
         mr->VolumeName);
       if (!mark_media_purged(ua, mr)) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
         goto bail_out;
       }
       goto bail_out;
@@ -462,22 +498,35 @@ void purge_jobs_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr)
       del.max_ids = MAX_DEL_LIST_LEN; 
    }
 
-   del.JobId = (JobId_t *)malloc(sizeof(JobId_t) * del.max_ids);
+   /*
+    * Check if he wants to purge a single jobid 
+    */
+   i = find_arg_with_value(ua, "jobid");
+   if (i >= 0) {
+      del.JobId = (JobId_t *)malloc(sizeof(JobId_t));
+      del.num_ids = 1;
+      del.JobId[0] = str_to_int64(ua->argv[i]);
+   } else {
+      /* 
+       * Purge ALL JobIds 
+       */
+      del.JobId = (JobId_t *)malloc(sizeof(JobId_t) * del.max_ids);
 
-   Mmsg(&query, "SELECT JobId FROM JobMedia WHERE MediaId=%d", mr->MediaId);
-   if (!db_sql_query(ua->db, query, file_delete_handler, (void *)&del)) {
-      bsendmsg(ua, "%s", db_strerror(ua->db));
-      Dmsg0(050, "Count failed\n");
-      goto bail_out;
+      Mmsg(query, "SELECT JobId FROM JobMedia WHERE MediaId=%d", mr->MediaId);
+      if (!db_sql_query(ua->db, query, file_delete_handler, (void *)&del)) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+         Dmsg0(050, "Count failed\n");
+        goto bail_out;
+      }
    }
 
    for (i=0; i < del.num_ids; i++) {
       Dmsg1(050, "Delete JobId=%d\n", del.JobId[i]);
-      Mmsg(&query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM File WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
-      Mmsg(&query, "DELETE FROM Job WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM Job WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
-      Mmsg(&query, "DELETE FROM JobMedia WHERE JobId=%d", del.JobId[i]);
+      Mmsg(query, "DELETE FROM JobMedia WHERE JobId=%d", del.JobId[i]);
       db_sql_query(ua->db, query, NULL, (void *)NULL);
       Dmsg1(050, "Del sql=%s\n", query);
       del.num_del++;
@@ -485,27 +534,49 @@ void purge_jobs_from_volume(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr)
    if (del.JobId) {
       free(del.JobId);
    }
-   bsendmsg(ua, _("%d Files for Volume %s purged from catalog.\n"), del.num_del,
-      mr->VolumeName);
+   bsendmsg(ua, _("%d File%s on Volume \"%s\" purged from catalog.\n"), del.num_del,
+      del.num_del==1?"":"s", mr->VolumeName);
 
    /* If purged, mark it so */
-   if (del.num_ids == del.num_del) {
-      mark_media_purged(ua, mr);
+   cnt.count = 0;
+   Mmsg(query, "SELECT count(*) FROM JobMedia WHERE MediaId=%d", 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");
+      goto bail_out;
+   }
+      
+   if (cnt.count == 0) {
+      bsendmsg(ua, "There are no more Jobs associated with Volume \"%s\". Marking it purged.\n",
+        mr->VolumeName);
+      if (!(stat = mark_media_purged(ua, mr))) {
+         bsendmsg(ua, "%s", db_strerror(ua->db));
+        goto bail_out;
+      }
    }
 
 bail_out:   
    free_pool_memory(query);
+   return stat;
 }
 
-static int mark_media_purged(UAContext *ua, MEDIA_DBR *mr)
+/*
+ * IF volume status is Append, Full, Used, or Error, mark it Purged
+ *   Purged volumes can then be recycled (if enabled).
+ */
+int mark_media_purged(UAContext *ua, MEDIA_DBR *mr)
 {
    if (strcmp(mr->VolStatus, "Append") == 0 || 
-       strcmp(mr->VolStatus, "Full")   == 0) {
-      strcpy(mr->VolStatus, "Purged");
-      if (!db_update_media_record(ua->db, mr)) {
-         bsendmsg(ua, "%s", db_strerror(ua->db));
+       strcmp(mr->VolStatus, "Full")   == 0 ||
+       strcmp(mr->VolStatus, "Used")   == 0 || 
+       strcmp(mr->VolStatus, "Error")  == 0) {
+      bstrncpy(mr->VolStatus, "Purged", sizeof(mr->VolStatus));
+      if (!db_update_media_record(ua->jcr, ua->db, mr)) {
         return 0;
       }
+      return 1;
+   } else {
+      bsendmsg(ua, _("Cannot purge Volume with VolStatus=%s\n"), mr->VolStatus);
    }
-   return 1;
+   return strcpy(mr->VolStatus, "Purged") == 0;
 }