X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=bacula%2Fsrc%2Fdird%2Fua_dotcmds.c;h=8d41289b8afb6608b4598b6d3be842642701d0f7;hb=b8224aab234012c2d127b84eceb160e99dd4a14d;hp=a236ff3c2e17253b36468a333df4590b428472e5;hpb=245788effd044ce5f8fb380cf11066ecd61b6f2e;p=bacula%2Fbacula

diff --git a/bacula/src/dird/ua_dotcmds.c b/bacula/src/dird/ua_dotcmds.c
index a236ff3c2e..8d41289b8a 100644
--- a/bacula/src/dird/ua_dotcmds.c
+++ b/bacula/src/dird/ua_dotcmds.c
@@ -68,6 +68,7 @@ static bool levelscmd(UAContext *ua, const char *cmd);
 static bool getmsgscmd(UAContext *ua, const char *cmd);
 
 static bool api_cmd(UAContext *ua, const char *cmd);
+static bool sql_cmd(UAContext *ua, const char *cmd);
 static bool dot_quit_cmd(UAContext *ua, const char *cmd);
 static bool dot_help_cmd(UAContext *ua, const char *cmd);
 
@@ -87,6 +88,7 @@ static struct cmdstruct commands[] = {
  { NT_(".msgs"),       msgscmd,        NULL},
  { NT_(".pools"),      poolscmd,       NULL},
  { NT_(".quit"),       dot_quit_cmd,   NULL},
+ { NT_(".sql"),        sql_cmd,        NULL},
  { NT_(".status"),     dot_status_cmd, NULL},
  { NT_(".storage"),    storagecmd,     NULL},
  { NT_(".types"),      typescmd,       NULL} 
@@ -102,22 +104,29 @@ int do_a_dot_command(UAContext *ua, const char *cmd)
    int len;
    bool ok = false;
    bool found = false;
-   BSOCK *sock = ua->UA_sock;
+   BSOCK *user = ua->UA_sock;
 
-   Dmsg1(1400, "Dot command: %s\n", ua->UA_sock->msg);
+   Dmsg1(1400, "Dot command: %s\n", user->msg);
    if (ua->argc == 0) {
       return 1;
    }
 
    len = strlen(ua->argk[0]);
    if (len == 1) {
+      if (ua->api) user->signal(BNET_CMD_BEGIN);
+      if (ua->api) user->signal(BNET_CMD_OK);
       return 1;                       /* no op */
    }
    for (i=0; i<comsize; i++) {     /* search for command */
       if (strncasecmp(ua->argk[0],  _(commands[i].key), len) == 0) {
          bool gui = ua->gui;
+         /* Check if command permitted, but "quit" is always OK */
+         if (strcmp(ua->argk[0], NT_(".quit")) != 0 &&
+             !acl_access_ok(ua, Command_ACL, ua->argk[0], len)) {
+            break;
+         }
          ua->gui = true;
-         if (ua->api) sock->signal(BNET_CMD_BEGIN);
+         if (ua->api) user->signal(BNET_CMD_BEGIN);
          ok = (*commands[i].func)(ua, cmd);   /* go execute command */
          ua->gui = gui;
          found = true;
@@ -125,12 +134,11 @@ int do_a_dot_command(UAContext *ua, const char *cmd)
       }
    }
    if (!found) {
-      pm_strcat(sock->msg, _(": is an invalid command\n"));
-      sock->msglen = strlen(sock->msg);
-      sock->send();
-      if (ua->api) sock->signal(BNET_INVALID_CMD);
+      pm_strcat(user->msg, _(": is an invalid command.\n"));
+      user->msglen = strlen(user->msg);
+      user->send();
    }
-   if (ua->api) sock->signal(ok?BNET_CMD_OK:BNET_CMD_FAILED);
+   if (ua->api) user->signal(ok?BNET_CMD_OK:BNET_CMD_FAILED);
    return 1;
 }
 
@@ -154,27 +162,169 @@ static bool getmsgscmd(UAContext *ua, const char *cmd)
    return 1;
 }
 
+#ifdef DEVELOPER
+static void do_storage_die(UAContext *ua, STORE *store)
+{
+   BSOCK *sd;
+   JCR *jcr = ua->jcr;
+   USTORE lstore;
+   
+   lstore.store = store;
+   pm_strcpy(lstore.store_source, _("unknown source"));
+   set_wstorage(jcr, &lstore);
+   /* Try connecting for up to 15 seconds */
+   ua->send_msg(_("Connecting to Storage daemon %s at %s:%d\n"),
+      store->name(), store->address, store->SDport);
+   if (!connect_to_storage_daemon(jcr, 1, 15, 0)) {
+      ua->error_msg(_("Failed to connect to Storage daemon.\n"));
+      return;
+   }
+   Dmsg0(120, _("Connected to storage daemon\n"));
+   sd = jcr->store_bsock;
+   sd->fsend(".die");
+   if (sd->recv() >= 0) {
+      ua->send_msg("%s", sd->msg);
+   }
+   sd->signal(BNET_TERMINATE);
+   sd->close();
+   jcr->store_bsock = NULL;
+   return;
+}
+
+static void do_client_die(UAContext *ua, CLIENT *client)
+{
+   BSOCK *fd;
+
+   /* Connect to File daemon */
+
+   ua->jcr->client = client;
+   /* Try to connect for 15 seconds */
+   ua->send_msg(_("Connecting to Client %s at %s:%d\n"),
+      client->name(), client->address, client->FDport);
+   if (!connect_to_file_daemon(ua->jcr, 1, 15, 0)) {
+      ua->error_msg(_("Failed to connect to Client.\n"));
+      return;
+   }
+   Dmsg0(120, "Connected to file daemon\n");
+   fd = ua->jcr->file_bsock;
+   fd->fsend(".die");
+   if (fd->recv() >= 0) {
+      ua->send_msg("%s", fd->msg);
+   }
+   fd->signal(BNET_TERMINATE);
+   fd->close();
+   ua->jcr->file_bsock = NULL;
+   return;
+}
+
 /*
  * Create segmentation fault
  */
 static bool diecmd(UAContext *ua, const char *cmd)
 {
+   STORE *store;
+   CLIENT *client;
+   int i;
    JCR *jcr = NULL;
    int a;
 
-   bsendmsg(ua, _("The Director will segment fault.\n"));
-   a = jcr->JobId; /* ref NULL pointer */
-   jcr->JobId = 1000; /* another ref NULL pointer */
+   Dmsg1(120, "diecmd:%s:\n", cmd);
+
+   /* General debug? */
+   for (i=1; i<ua->argc; i++) {
+      if (strcasecmp(ua->argk[i], "dir") == 0 ||
+          strcasecmp(ua->argk[i], "director") == 0) {
+         ua->send_msg(_("The Director will segment fault.\n"));
+         a = jcr->JobId; /* ref NULL pointer */
+         jcr->JobId = 1000; /* another ref NULL pointer */
+         return 1;
+      }
+      if (strcasecmp(ua->argk[i], "client") == 0 ||
+          strcasecmp(ua->argk[i], "fd") == 0) {
+         client = NULL;
+         if (ua->argv[i]) {
+            client = (CLIENT *)GetResWithName(R_CLIENT, ua->argv[i]);
+            if (client) {
+               do_client_die(ua, client);
+               return 1;
+            }
+         }
+         client = select_client_resource(ua);
+         if (client) {
+            do_client_die(ua, client);
+            return 1;
+         }
+      }
+
+      if (strcasecmp(ua->argk[i], NT_("store")) == 0 ||
+          strcasecmp(ua->argk[i], NT_("storage")) == 0 ||
+          strcasecmp(ua->argk[i], NT_("sd")) == 0) {
+         store = NULL;
+         if (ua->argv[i]) {
+            store = (STORE *)GetResWithName(R_STORAGE, ua->argv[i]);
+            if (store) {
+               do_storage_die(ua, store);
+               return 1;
+            }
+         }
+         store = get_storage_resource(ua, false/*no default*/);
+         if (store) {
+            do_storage_die(ua, store);
+            return 1;
+         }
+      }
+   }
+   /*
+    * We didn't find an appropriate keyword above, so
+    * prompt the user.
+    */
+   start_prompt(ua, _("Available daemons are: \n"));
+   add_prompt(ua, _("Director"));
+   add_prompt(ua, _("Storage"));
+   add_prompt(ua, _("Client"));
+   switch(do_prompt(ua, "", _("Select daemon type to make die"), NULL, 0)) {
+   case 0:                         /* Director */
+      ua->send_msg(_("The Director will segment fault.\n"));
+      a = jcr->JobId; /* ref NULL pointer */
+      jcr->JobId = 1000; /* another ref NULL pointer */
+      break;
+   case 1:
+      store = get_storage_resource(ua, false/*no default*/);
+      if (store) {
+         do_storage_die(ua, store);
+      }
+      break;
+   case 2:
+      client = select_client_resource(ua);
+      if (client) {
+         do_client_die(ua, client);
+      }
+      break;
+   default:
+      break;
+   }
    return true;
 }
 
+#else
+
+/*
+ * Dummy routine for non-development version
+ */
+static bool diecmd(UAContext *ua, const char *cmd)
+{
+   return true;
+}
+
+#endif
+
 static bool jobscmd(UAContext *ua, const char *cmd)
 {
-   JOB *job = NULL;
+   JOB *job;
    LockRes();
-   while ( (job = (JOB *)GetNextRes(R_JOB, (RES *)job)) ) {
+   foreach_res(job, R_JOB) {
       if (acl_access_ok(ua, Job_ACL, job->name())) {
-         bsendmsg(ua, "%s\n", job->name());
+         ua->send_msg("%s\n", job->name());
       }
    }
    UnlockRes();
@@ -183,11 +333,11 @@ static bool jobscmd(UAContext *ua, const char *cmd)
 
 static bool filesetscmd(UAContext *ua, const char *cmd)
 {
-   FILESET *fs = NULL;
+   FILESET *fs;
    LockRes();
-   while ( (fs = (FILESET *)GetNextRes(R_FILESET, (RES *)fs)) ) {
+   foreach_res(fs, R_FILESET) {
       if (acl_access_ok(ua, FileSet_ACL, fs->name())) {
-         bsendmsg(ua, "%s\n", fs->name());
+         ua->send_msg("%s\n", fs->name());
       }
    }
    UnlockRes();
@@ -196,11 +346,11 @@ static bool filesetscmd(UAContext *ua, const char *cmd)
 
 static bool clientscmd(UAContext *ua, const char *cmd)
 {
-   CLIENT *client = NULL;
+   CLIENT *client;       
    LockRes();
-   while ( (client = (CLIENT *)GetNextRes(R_CLIENT, (RES *)client)) ) {
+   foreach_res(client, R_CLIENT) {
       if (acl_access_ok(ua, Client_ACL, client->name())) {
-         bsendmsg(ua, "%s\n", client->name());
+         ua->send_msg("%s\n", client->name());
       }
    }
    UnlockRes();
@@ -211,8 +361,8 @@ static bool msgscmd(UAContext *ua, const char *cmd)
 {
    MSGS *msgs = NULL;
    LockRes();
-   while ( (msgs = (MSGS *)GetNextRes(R_MSGS, (RES *)msgs)) ) {
-      bsendmsg(ua, "%s\n", msgs->name());
+   foreach_res(msgs, R_MSGS) {
+      ua->send_msg("%s\n", msgs->name());
    }
    UnlockRes();
    return true;
@@ -220,11 +370,11 @@ static bool msgscmd(UAContext *ua, const char *cmd)
 
 static bool poolscmd(UAContext *ua, const char *cmd)
 {
-   POOL *pool = NULL;
+   POOL *pool;       
    LockRes();
-   while ( (pool = (POOL *)GetNextRes(R_POOL, (RES *)pool)) ) {
+   foreach_res(pool, R_POOL) {
       if (acl_access_ok(ua, Pool_ACL, pool->name())) {
-         bsendmsg(ua, "%s\n", pool->name());
+         ua->send_msg("%s\n", pool->name());
       }
    }
    UnlockRes();
@@ -233,11 +383,11 @@ static bool poolscmd(UAContext *ua, const char *cmd)
 
 static bool storagecmd(UAContext *ua, const char *cmd)
 {
-   STORE *store = NULL;
+   STORE *store;
    LockRes();
-   while ( (store = (STORE *)GetNextRes(R_STORAGE, (RES *)store)) ) {
+   foreach_res(store, R_STORAGE) {
       if (acl_access_ok(ua, Storage_ACL, store->name())) {
-         bsendmsg(ua, "%s\n", store->name());
+         ua->send_msg("%s\n", store->name());
       }
    }
    UnlockRes();
@@ -247,21 +397,14 @@ static bool storagecmd(UAContext *ua, const char *cmd)
 
 static bool typescmd(UAContext *ua, const char *cmd)
 {
-   bsendmsg(ua, "Backup\n");
-   bsendmsg(ua, "Restore\n");
-   bsendmsg(ua, "Admin\n");
-   bsendmsg(ua, "Verify\n");
-   bsendmsg(ua, "Migrate\n");
+   ua->send_msg("Backup\n");
+   ua->send_msg("Restore\n");
+   ua->send_msg("Admin\n");
+   ua->send_msg("Verify\n");
+   ua->send_msg("Migrate\n");
    return true;
 }
 
-static int client_backups_handler(void *ctx, int num_field, char **row)
-{
-   UAContext *ua = (UAContext *)ctx;
-   bsendmsg(ua, "| %s | %s | %s | %s | %s | %s | %s | %s |\n",
-      row[0], row[1], row[2], row[3], row[4], row[5], row[6], row[7], row[8]);
-   return 0;
-}
 
 /*
  * If this command is called, it tells the director that we
@@ -273,46 +416,96 @@ static int client_backups_handler(void *ctx, int num_field, char **row)
  */
 static bool api_cmd(UAContext *ua, const char *cmd)
 {
-   /* Eventually we will probably have several levels or
-    *  capabilities enabled by this.
-    */
-   ua->api = 1;
+   if (ua->argc == 2) {
+      ua->api = atoi(ua->argk[1]);
+   } else {
+      ua->api = 1;
+   }
    return true;
 }
 
+static int client_backups_handler(void *ctx, int num_field, char **row)
+{
+   UAContext *ua = (UAContext *)ctx;
+   ua->send_msg("| %s | %s | %s | %s | %s | %s | %s | %s |\n",
+      row[0], row[1], row[2], row[3], row[4], row[5], row[6], row[7], row[8]);
+   return 0;
+}
+
 /*
  * Return the backups for this client 
+ *
+ *  .backups client=xxx fileset=yyy
+ *
  */
 static bool backupscmd(UAContext *ua, const char *cmd)
 {
    if (!open_client_db(ua)) {
       return true;
    }
-   if (ua->argc != 3 || strcmp(ua->argk[1], "client") != 0 || strcmp(ua->argk[2], "fileset") != 0) {
+   if (ua->argc != 3 || strcmp(ua->argk[1], "client") != 0 || 
+       strcmp(ua->argk[2], "fileset") != 0) {
       return true;
    }
    if (!acl_access_ok(ua, Client_ACL, ua->argv[1]) ||
        !acl_access_ok(ua, FileSet_ACL, ua->argv[2])) {
+      ua->error_msg(_("Access to specified Client or FileSet not allowed.\n"));
       return true;
    }
    Mmsg(ua->cmd, client_backups, ua->argv[1], ua->argv[2]);
    if (!db_sql_query(ua->db, ua->cmd, client_backups_handler, (void *)ua)) {
-      bsendmsg(ua, _("Query failed: %s. ERR=%s\n"), ua->cmd, db_strerror(ua->db));
+      ua->error_msg(_("Query failed: %s. ERR=%s\n"), ua->cmd, db_strerror(ua->db));
       return true;
    }
    return true;
 }
 
+static int sql_handler(void *ctx, int num_field, char **row)
+{
+   UAContext *ua = (UAContext *)ctx;
+   POOL_MEM rows(PM_MESSAGE);
+
+   for (int i=0; num_field--; i++) {
+      if (i == 0) {
+         pm_strcpy(rows, row[0]);
+      } else {
+         pm_strcat(rows, row[i]);
+      }
+      pm_strcat(rows, "\t");
+   }
+   ua->send_msg(rows.c_str());
+   return 0;
+}
+
+static bool sql_cmd(UAContext *ua, const char *cmd)
+{
+   int index;
+   if (!open_client_db(ua)) {
+      return true;
+   }
+   index = find_arg_with_value(ua, "query");
+   if (index < 0) {
+      ua->error_msg(_("query keyword not found.\n"));
+      return true;
+   }
+   if (!db_sql_query(ua->db, ua->argv[index], sql_handler, (void *)ua)) {
+      Dmsg1(100, "Query failed: ERR=%s\n", db_strerror(ua->db));
+      ua->error_msg(_("Query failed: %s. ERR=%s\n"), ua->cmd, db_strerror(ua->db));
+      return true;
+   }
+   return true;
+}
+      
 
 
 static bool levelscmd(UAContext *ua, const char *cmd)
 {
-   bsendmsg(ua, "Incremental\n");
-   bsendmsg(ua, "Full\n");
-   bsendmsg(ua, "Differential\n");
-   bsendmsg(ua, "Catalog\n");
-   bsendmsg(ua, "InitCatalog\n");
-   bsendmsg(ua, "VolumeToCatalog\n");
+   ua->send_msg("Incremental\n");
+   ua->send_msg("Full\n");
+   ua->send_msg("Differential\n");
+   ua->send_msg("Catalog\n");
+   ua->send_msg("InitCatalog\n");
+   ua->send_msg("VolumeToCatalog\n");
    return true;
 }
 
@@ -325,6 +518,7 @@ static bool defaultscmd(UAContext *ua, const char *cmd)
    CLIENT *client;
    STORE *storage;
    POOL *pool;
+   char ed1[50];
 
    if (ua->argc != 2 || !ua->argv[1]) {
       return true;
@@ -338,18 +532,18 @@ static bool defaultscmd(UAContext *ua, const char *cmd)
       job = (JOB *)GetResWithName(R_JOB, ua->argv[1]);
       if (job) {
          USTORE store;
-         bsendmsg(ua, "job=%s", job->name());
-         bsendmsg(ua, "pool=%s", job->pool->name());
-         bsendmsg(ua, "messages=%s", job->messages->name());
-         bsendmsg(ua, "client=%s", job->client->name());
+         ua->send_msg("job=%s", job->name());
+         ua->send_msg("pool=%s", job->pool->name());
+         ua->send_msg("messages=%s", job->messages->name());
+         ua->send_msg("client=%s", job->client->name());
          get_job_storage(&store, job, NULL);
-         bsendmsg(ua, "storage=%s", store.store->name());
-         bsendmsg(ua, "where=%s", job->RestoreWhere?job->RestoreWhere:"");
-         bsendmsg(ua, "level=%s", level_to_str(job->JobLevel));
-         bsendmsg(ua, "type=%s", job_type_to_str(job->JobType));
-         bsendmsg(ua, "fileset=%s", job->fileset->name());
-         bsendmsg(ua, "enabled=%d", job->enabled);
-         bsendmsg(ua, "catalog=%s", job->client->catalog->name());
+         ua->send_msg("storage=%s", store.store->name());
+         ua->send_msg("where=%s", job->RestoreWhere?job->RestoreWhere:"");
+         ua->send_msg("level=%s", level_to_str(job->JobLevel));
+         ua->send_msg("type=%s", job_type_to_str(job->JobType));
+         ua->send_msg("fileset=%s", job->fileset->name());
+         ua->send_msg("enabled=%d", job->enabled);
+         ua->send_msg("catalog=%s", job->client->catalog->name());
       }
    } 
    /* Client defaults */
@@ -359,13 +553,13 @@ static bool defaultscmd(UAContext *ua, const char *cmd)
       }
       client = (CLIENT *)GetResWithName(R_CLIENT, ua->argv[1]);
       if (client) {
-         bsendmsg(ua, "client=%s", client->name());
-         bsendmsg(ua, "address=%s", client->address);
-         bsendmsg(ua, "fdport=%d", client->FDport);
-         bsendmsg(ua, "file_retention=%d", client->FileRetention);
-         bsendmsg(ua, "job_retention=%d", client->JobRetention);
-         bsendmsg(ua, "autoprune=%d", client->AutoPrune);
-         bsendmsg(ua, "catalog=%s", client->catalog->name());
+         ua->send_msg("client=%s", client->name());
+         ua->send_msg("address=%s", client->address);
+         ua->send_msg("fdport=%d", client->FDport);
+         ua->send_msg("file_retention=%s", edit_uint64(client->FileRetention, ed1));
+         ua->send_msg("job_retention=%s", edit_uint64(client->JobRetention, ed1));
+         ua->send_msg("autoprune=%d", client->AutoPrune);
+         ua->send_msg("catalog=%s", client->catalog->name());
       }
    }
    /* Storage defaults */
@@ -376,16 +570,16 @@ static bool defaultscmd(UAContext *ua, const char *cmd)
       storage = (STORE *)GetResWithName(R_STORAGE, ua->argv[1]);
       DEVICE *device;
       if (storage) {
-         bsendmsg(ua, "storage=%s", storage->name());
-         bsendmsg(ua, "address=%s", storage->address);
-         bsendmsg(ua, "enabled=%d", storage->enabled);
-         bsendmsg(ua, "media_type=%s", storage->media_type);
-         bsendmsg(ua, "sdport=%d", storage->SDport);
+         ua->send_msg("storage=%s", storage->name());
+         ua->send_msg("address=%s", storage->address);
+         ua->send_msg("enabled=%d", storage->enabled);
+         ua->send_msg("media_type=%s", storage->media_type);
+         ua->send_msg("sdport=%d", storage->SDport);
          device = (DEVICE *)storage->device->first();
-         bsendmsg(ua, "device=%s", device->name());
+         ua->send_msg("device=%s", device->name());
          if (storage->device->size() > 1) {
             while ((device = (DEVICE *)storage->device->next())) {
-               bsendmsg(ua, ",%s", device->name());
+               ua->send_msg(",%s", device->name());
             }
          }
       }
@@ -397,21 +591,21 @@ static bool defaultscmd(UAContext *ua, const char *cmd)
       }
       pool = (POOL *)GetResWithName(R_POOL, ua->argv[1]);
       if (pool) {
-         bsendmsg(ua, "pool=%s", pool->name());
-         bsendmsg(ua, "pool_type=%s", pool->pool_type);
-         bsendmsg(ua, "label_format=%s", pool->label_format?pool->label_format:"");
-         bsendmsg(ua, "use_volume_once=%d", pool->use_volume_once);
-         bsendmsg(ua, "purge_oldest_volume=%d", pool->purge_oldest_volume);
-         bsendmsg(ua, "recycle_oldest_volume=%d", pool->recycle_oldest_volume);
-         bsendmsg(ua, "recycle_current_volume=%d", pool->recycle_current_volume);
-         bsendmsg(ua, "max_volumes=%d", pool->max_volumes);
-         bsendmsg(ua, "vol_retention=%d", pool->VolRetention);
-         bsendmsg(ua, "vol_use_duration=%d", pool->VolUseDuration);
-         bsendmsg(ua, "max_vol_jobs=%d", pool->MaxVolJobs);
-         bsendmsg(ua, "max_vol_files=%d", pool->MaxVolFiles);
-         bsendmsg(ua, "max_vol_bytes=%d", pool->MaxVolBytes);
-         bsendmsg(ua, "auto_prune=%d", pool->AutoPrune);
-         bsendmsg(ua, "recycle=%d", pool->Recycle);
+         ua->send_msg("pool=%s", pool->name());
+         ua->send_msg("pool_type=%s", pool->pool_type);
+         ua->send_msg("label_format=%s", pool->label_format?pool->label_format:"");
+         ua->send_msg("use_volume_once=%d", pool->use_volume_once);
+         ua->send_msg("purge_oldest_volume=%d", pool->purge_oldest_volume);
+         ua->send_msg("recycle_oldest_volume=%d", pool->recycle_oldest_volume);
+         ua->send_msg("recycle_current_volume=%d", pool->recycle_current_volume);
+         ua->send_msg("max_volumes=%d", pool->max_volumes);
+         ua->send_msg("vol_retention=%s", edit_uint64(pool->VolRetention, ed1));
+         ua->send_msg("vol_use_duration=%s", edit_uint64(pool->VolUseDuration, ed1));
+         ua->send_msg("max_vol_jobs=%d", pool->MaxVolJobs);
+         ua->send_msg("max_vol_files=%d", pool->MaxVolFiles);
+         ua->send_msg("max_vol_bytes=%s", edit_uint64(pool->MaxVolBytes, ed1));
+         ua->send_msg("auto_prune=%d", pool->AutoPrune);
+         ua->send_msg("recycle=%d", pool->Recycle);
       }
    }
    return true;