git.sur5r.net Git - bacula/bacula/blob - bacula/src/dird/ua_cmds.c

   1 /*
   2  *
   3  *   Bacula Director -- User Agent Commands
   4  *
   5  *     Kern Sibbald, September MM
   6  *
   7  *   Version $Id$
   8  */
   9
  10 /*
  11    Copyright (C) 2000-2004 Kern Sibbald and John Walker
  12
  13    This program is free software; you can redistribute it and/or
  14    modify it under the terms of the GNU General Public License as
  15    published by the Free Software Foundation; either version 2 of
  16    the License, or (at your option) any later version.
  17
  18    This program is distributed in the hope that it will be useful,
  19    but WITHOUT ANY WARRANTY; without even the implied warranty of
  20    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  21    General Public License for more details.
  22
  23    You should have received a copy of the GNU General Public
  24    License along with this program; if not, write to the Free
  25    Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  26    MA 02111-1307, USA.
  27
  28  */
  29
  30 #include "bacula.h"
  31 #include "dird.h"
  32
  33 /* Imported subroutines */
  34
  35 /* Imported variables */
  36 extern int r_first;
  37 extern int r_last;
  38 extern struct s_res resources[];
  39 extern char my_name[];
  40 extern jobq_t job_queue;              /* job queue */
  41
  42
  43 extern char *list_pool;
  44
  45 /* Imported functions */
  46 extern int status_cmd(UAContext *ua, const char *cmd);
  47 extern int list_cmd(UAContext *ua, const char *cmd);
  48 extern int llist_cmd(UAContext *ua, const char *cmd);
  49 extern int show_cmd(UAContext *ua, const char *cmd);
  50 extern int messagescmd(UAContext *ua, const char *cmd);
  51 extern int autodisplay_cmd(UAContext *ua, const char *cmd);
  52 extern int gui_cmd(UAContext *ua, const char *cmd);
  53 extern int sqlquerycmd(UAContext *ua, const char *cmd);
  54 extern int querycmd(UAContext *ua, const char *cmd);
  55 extern int run_cmd(UAContext *ua, const char *cmd);
  56 extern int retentioncmd(UAContext *ua, const char *cmd);
  57 extern int prunecmd(UAContext *ua, const char *cmd);
  58 extern int purgecmd(UAContext *ua, const char *cmd);
  59 extern int restore_cmd(UAContext *ua, const char *cmd);
  60 extern int label_cmd(UAContext *ua, const char *cmd);
  61 extern int relabel_cmd(UAContext *ua, const char *cmd);
  62 extern int update_slots(UAContext *ua);  /* ua_label.c */
  63
  64 /* Forward referenced functions */
  65 static int add_cmd(UAContext *ua, const char *cmd);
  66 static int create_cmd(UAContext *ua, const char *cmd);
  67 static int cancel_cmd(UAContext *ua, const char *cmd);
  68 static int setdebug_cmd(UAContext *ua, const char *cmd);
  69 static int trace_cmd(UAContext *ua, const char *cmd);
  70 static int var_cmd(UAContext *ua, const char *cmd);
  71 static int estimate_cmd(UAContext *ua, const char *cmd);
  72 static int help_cmd(UAContext *ua, const char *cmd);
  73 static int delete_cmd(UAContext *ua, const char *cmd);
  74 static int use_cmd(UAContext *ua, const char *cmd);
  75 static int unmount_cmd(UAContext *ua, const char *cmd);
  76 static int version_cmd(UAContext *ua, const char *cmd);
  77 static int automount_cmd(UAContext *ua, const char *cmd);
  78 static int time_cmd(UAContext *ua, const char *cmd);
  79 static int reload_cmd(UAContext *ua, const char *cmd);
  80 static int update_volume(UAContext *ua);
  81 static int update_pool(UAContext *ua);
  82 static int delete_volume(UAContext *ua);
  83 static int delete_pool(UAContext *ua);
  84 static int delete_job(UAContext *ua);
  85 static int mount_cmd(UAContext *ua, const char *cmd);
  86 static int release_cmd(UAContext *ua, const char *cmd);
  87 static int update_cmd(UAContext *ua, const char *cmd);
  88 static int wait_cmd(UAContext *ua, const char *cmd);
  89 static int setip_cmd(UAContext *ua, const char *cmd);
  90
  91 int quit_cmd(UAContext *ua, const char *cmd);
  92
  93
  94 struct cmdstruct { const char *key; int (*func)(UAContext *ua, const char *cmd); const char *help; };
  95 static struct cmdstruct commands[] = {
  96  { N_("add"),        add_cmd,         _("add media to a pool")},
  97  { N_("autodisplay"), autodisplay_cmd, _("autodisplay [on/off] -- console messages")},
  98  { N_("automount"),   automount_cmd,  _("automount [on/off] -- after label")},
  99  { N_("cancel"),     cancel_cmd,    _("cancel job=nnn -- cancel a job")},
 100  { N_("create"),     create_cmd,    _("create DB Pool from resource")},
 101  { N_("delete"),     delete_cmd,    _("delete [pool=<pool-name> | media volume=<volume-name>]")},
 102  { N_("estimate"),   estimate_cmd,  _("performs FileSet estimate, listing gives full listing")},
 103  { N_("exit"),       quit_cmd,      _("exit = quit")},
 104  { N_("gui"),        gui_cmd,       _("gui [on/off] -- non-interactive gui mode")},
 105  { N_("help"),       help_cmd,      _("print this command")},
 106  { N_("list"),       list_cmd,      _("list [pools | jobs | jobtotals | media <pool> | files jobid=<nn>]; from catalog")},
 107  { N_("label"),      label_cmd,     _("label a tape")},
 108  { N_("llist"),      llist_cmd,     _("full or long list like list command")},
 109  { N_("messages"),   messagescmd,   _("messages")},
 110  { N_("mount"),      mount_cmd,     _("mount <storage-name>")},
 111  { N_("prune"),      prunecmd,      _("prune expired records from catalog")},
 112  { N_("purge"),      purgecmd,      _("purge records from catalog")},
 113  { N_("quit"),       quit_cmd,      _("quit")},
 114  { N_("query"),      querycmd,      _("query catalog")},
 115  { N_("restore"),    restore_cmd,   _("restore files")},
 116  { N_("relabel"),    relabel_cmd,   _("relabel a tape")},
 117  { N_("release"),    release_cmd,   _("release <storage-name>")},
 118  { N_("reload"),     reload_cmd,    _("reload conf file")},
 119  { N_("run"),        run_cmd,       _("run <job-name>")},
 120  { N_("status"),     status_cmd,    _("status [storage | client]=<name>")},
 121  { N_("setdebug"),   setdebug_cmd,  _("sets debug level")},
 122  { N_("setip"),      setip_cmd,     _("sets new client address -- if authorized")},
 123  { N_("show"),       show_cmd,      _("show (resource records) [jobs | pools | ... | all]")},
 124  { N_("sqlquery"),   sqlquerycmd,   _("use SQL to query catalog")},
 125  { N_("time"),       time_cmd,      _("print current time")},
 126  { N_("trace"),      trace_cmd,     _("turn on/off trace to file")},
 127  { N_("unmount"),    unmount_cmd,   _("unmount <storage-name>")},
 128  { N_("update"),     update_cmd,    _("update Volume or Pool")},
 129  { N_("use"),        use_cmd,       _("use catalog xxx")},
 130  { N_("var"),        var_cmd,       _("does variable expansion")},
 131  { N_("version"),    version_cmd,   _("print Director version")},
 132  { N_("wait"),       wait_cmd,      _("wait until no jobs are running")},
 133              };
 134 #define comsize (sizeof(commands)/sizeof(struct cmdstruct))
 135
 136 /*
 137  * Execute a command from the UA
 138  */
 139 int do_a_command(UAContext *ua, const char *cmd)
 140 {
 141    unsigned int i;
 142    int len, stat;
 143    bool found = false;
 144
 145    stat = 1;
 146
 147    Dmsg1(200, "Command: %s\n", ua->UA_sock->msg);
 148    if (ua->argc == 0) {
 149       return 1;
 150    }
 151
 152    len = strlen(ua->argk[0]);
 153    for (i=0; i<comsize; i++) {     /* search for command */
 154       if (strncasecmp(ua->argk[0],  _(commands[i].key), len) == 0) {
 155          if (!acl_access_ok(ua, Command_ACL, ua->argk[0], len)) {
 156             break;
 157          }
 158          stat = (*commands[i].func)(ua, cmd);   /* go execute command */
 159          found = true;
 160          break;
 161       }
 162    }
 163    if (!found) {
 164       bnet_fsend(ua->UA_sock, _("%s: is an illegal command.\n"), ua->argk[0]);
 165    }
 166    return stat;
 167 }
 168
 169 /*
 170  * This is a common routine used to stuff the Pool DB record defaults
 171  *   into the Media DB record just before creating a media (Volume)
 172  *   record.
 173  */
 174 void set_pool_dbr_defaults_in_media_dbr(MEDIA_DBR *mr, POOL_DBR *pr)
 175 {
 176    mr->PoolId = pr->PoolId;
 177    bstrncpy(mr->VolStatus, "Append", sizeof(mr->VolStatus));
 178    mr->Recycle = pr->Recycle;
 179    mr->VolRetention = pr->VolRetention;
 180    mr->VolUseDuration = pr->VolUseDuration;
 181    mr->MaxVolJobs = pr->MaxVolJobs;
 182    mr->MaxVolFiles = pr->MaxVolFiles;
 183    mr->MaxVolBytes = pr->MaxVolBytes;
 184 }
 185
 186
 187 /*
 188  *  Add Volumes to an existing Pool
 189  */
 190 static int add_cmd(UAContext *ua, const char *cmd)
 191 {
 192    POOL_DBR pr;
 193    MEDIA_DBR mr;
 194    int num, i, max, startnum;
 195    int first_id = 0;
 196    char name[MAX_NAME_LENGTH];
 197    STORE *store;
 198    int Slot = 0, InChanger = 0;
 199
 200    bsendmsg(ua, _(
 201 "You probably don't want to be using this command since it\n"
 202 "creates database records without labeling the Volumes.\n"
 203 "You probably want to use the \"label\" command.\n\n"));
 204
 205    if (!open_db(ua)) {
 206       return 1;
 207    }
 208
 209    memset(&pr, 0, sizeof(pr));
 210    memset(&mr, 0, sizeof(mr));
 211
 212    if (!get_pool_dbr(ua, &pr)) {
 213       return 1;
 214    }
 215
 216    Dmsg4(120, "id=%d Num=%d Max=%d type=%s\n", pr.PoolId, pr.NumVols,
 217       pr.MaxVols, pr.PoolType);
 218
 219    while (pr.MaxVols > 0 && pr.NumVols >= pr.MaxVols) {
 220       bsendmsg(ua, _("Pool already has maximum volumes = %d\n"), pr.MaxVols);
 221       for (;;) {
 222          if (!get_pint(ua, _("Enter new maximum (zero for unlimited): "))) {
 223             return 1;
 224          }
 225          pr.MaxVols = ua->pint32_val;
 226       }
 227    }
 228
 229    /* Get media type */
 230    if ((store = get_storage_resource(ua, 0)) != NULL) {
 231       bstrncpy(mr.MediaType, store->media_type, sizeof(mr.MediaType));
 232    } else if (!get_media_type(ua, mr.MediaType, sizeof(mr.MediaType))) {
 233       return 1;
 234    }
 235
 236    if (pr.MaxVols == 0) {
 237       max = 1000;
 238    } else {
 239       max = pr.MaxVols - pr.NumVols;
 240    }
 241    for (;;) {
 242       char buf[100];
 243       bsnprintf(buf, sizeof(buf), _("Enter number of Volumes to create. 0=>fixed name. Max=%d: "), max);
 244       if (!get_pint(ua, buf)) {
 245          return 1;
 246       }
 247       num = ua->pint32_val;
 248       if (num < 0 || num > max) {
 249          bsendmsg(ua, _("The number must be between 0 and %d\n"), max);
 250          continue;
 251       }
 252       break;
 253    }
 254 getVolName:
 255    if (num == 0) {
 256       if (!get_cmd(ua, _("Enter Volume name: "))) {
 257          return 1;
 258       }
 259    } else {
 260       if (!get_cmd(ua, _("Enter base volume name: "))) {
 261          return 1;
 262       }
 263    }
 264    /* Don't allow | in Volume name because it is the volume separator character */
 265    if (!is_volume_name_legal(ua, ua->cmd)) {
 266       goto getVolName;
 267    }
 268    if (strlen(ua->cmd) >= MAX_NAME_LENGTH-10) {
 269       bsendmsg(ua, _("Volume name too long.\n"));
 270       goto getVolName;
 271    }
 272    if (strlen(ua->cmd) == 0) {
 273       bsendmsg(ua, _("Volume name must be at least one character long.\n"));
 274       goto getVolName;
 275    }
 276
 277    bstrncpy(name, ua->cmd, sizeof(name));
 278    if (num > 0) {
 279       strcat(name, "%04d");
 280
 281       for (;;) {
 282          if (!get_pint(ua, _("Enter the starting number: "))) {
 283             return 1;
 284          }
 285          startnum = ua->pint32_val;
 286          if (startnum < 1) {
 287             bsendmsg(ua, _("Start number must be greater than zero.\n"));
 288             continue;
 289          }
 290          break;
 291       }
 292    } else {
 293       startnum = 1;
 294       num = 1;
 295    }
 296
 297    if (store && store->autochanger) {
 298       if (!get_pint(ua, _("Enter slot (0 for none): "))) {
 299          return 1;
 300       }
 301       Slot = ua->pint32_val;
 302       if (!get_yesno(ua, _("InChanger? yes/no: "))) {
 303          return 1;
 304       }
 305       InChanger = ua->pint32_val;
 306    }
 307
 308    set_pool_dbr_defaults_in_media_dbr(&mr, &pr);
 309    for (i=startnum; i < num+startnum; i++) {
 310       bsnprintf(mr.VolumeName, sizeof(mr.VolumeName), name, i);
 311       mr.Slot = Slot++;
 312       mr.InChanger = InChanger;
 313       Dmsg1(200, "Create Volume %s\n", mr.VolumeName);
 314       if (!db_create_media_record(ua->jcr, ua->db, &mr)) {
 315          bsendmsg(ua, "%s", db_strerror(ua->db));
 316          return 1;
 317       }
 318       if (i == startnum) {
 319          first_id = mr.PoolId;
 320       }
 321    }
 322    pr.NumVols += num;
 323    Dmsg0(200, "Update pool record.\n");
 324    if (db_update_pool_record(ua->jcr, ua->db, &pr) != 1) {
 325       bsendmsg(ua, "%s", db_strerror(ua->db));
 326       return 1;
 327    }
 328    bsendmsg(ua, _("%d Volumes created in pool %s\n"), num, pr.Name);
 329
 330    return 1;
 331 }
 332
 333 /*
 334  * Turn auto mount on/off
 335  *
 336  *  automount on
 337  *  automount off
 338  */
 339 int automount_cmd(UAContext *ua, const char *cmd)
 340 {
 341    char *onoff;
 342
 343    if (ua->argc != 2) {
 344       if (!get_cmd(ua, _("Turn on or off? "))) {
 345             return 1;
 346       }
 347       onoff = ua->cmd;
 348    } else {
 349       onoff = ua->argk[1];
 350    }
 351
 352    ua->automount = (strcasecmp(onoff, _("off")) == 0) ? 0 : 1;
 353    return 1;
 354 }
 355
 356
 357 /*
 358  * Cancel a job
 359  */
 360 static int cancel_cmd(UAContext *ua, const char *cmd)
 361 {
 362    int i, ret;
 363    int njobs = 0;
 364    JCR *jcr = NULL;
 365    char JobName[MAX_NAME_LENGTH];
 366
 367    if (!open_db(ua)) {
 368       return 1;
 369    }
 370
 371    for (i=1; i<ua->argc; i++) {
 372       if (strcasecmp(ua->argk[i], _("jobid")) == 0) {
 373          uint32_t JobId;
 374          if (!ua->argv[i]) {
 375             break;
 376          }
 377          JobId = str_to_int64(ua->argv[i]);
 378          if (!(jcr=get_jcr_by_id(JobId))) {
 379             bsendmsg(ua, _("JobId %d is not running.\n"),  JobId);
 380             return 1;
 381          }
 382          break;
 383       } else if (strcasecmp(ua->argk[i], _("job")) == 0) {
 384          if (!ua->argv[i]) {
 385             break;
 386          }
 387          if (!(jcr=get_jcr_by_partial_name(ua->argv[i]))) {
 388             bsendmsg(ua, _("Job %s is not running.\n"), ua->argv[i]);
 389             return 1;
 390          }
 391          break;
 392       }
 393    }
 394    /* If we still do not have a jcr,
 395     *   throw up a list and ask the user to select one.
 396     */
 397    if (!jcr) {
 398       /* Count Jobs running */
 399       lock_jcr_chain();
 400       foreach_jcr(jcr) {
 401          if (jcr->JobId == 0) {      /* this is us */
 402             free_locked_jcr(jcr);
 403             continue;
 404          }
 405          free_locked_jcr(jcr);
 406          njobs++;
 407       }
 408       unlock_jcr_chain();
 409
 410       if (njobs == 0) {
 411          bsendmsg(ua, _("No Jobs running.\n"));
 412          return 1;
 413       }
 414       start_prompt(ua, _("Select Job:\n"));
 415       lock_jcr_chain();
 416       foreach_jcr(jcr) {
 417          if (jcr->JobId == 0) {      /* this is us */
 418             free_locked_jcr(jcr);
 419             continue;
 420          }
 421          add_prompt(ua, jcr->Job);
 422          free_locked_jcr(jcr);
 423       }
 424       unlock_jcr_chain();
 425
 426       if (do_prompt(ua, _("Job"),  _("Choose Job to cancel"), JobName, sizeof(JobName)) < 0) {
 427          return 1;
 428       }
 429       if (njobs == 1) {
 430          if (!get_yesno(ua, _("Confirm cancel (yes/no): ")) || ua->pint32_val == 0) {
 431             return 1;
 432          }
 433       }
 434       /* NOTE! This increments the ref_count */
 435       jcr = get_jcr_by_full_name(JobName);
 436       if (!jcr) {
 437          bsendmsg(ua, _("Job %s not found.\n"), JobName);
 438          return 1;
 439       }
 440    }
 441
 442    ret = cancel_job(ua, jcr);
 443    free_jcr(jcr);
 444
 445    return ret;
 446 }
 447
 448 /*
 449  * This is a common routine to create or update a
 450  *   Pool DB base record from a Pool Resource. We handle
 451  *   the setting of MaxVols and NumVols slightly differently
 452  *   depending on if we are creating the Pool or we are
 453  *   simply bringing it into agreement with the resource (updage).
 454  */
 455 static void set_pooldbr_from_poolres(POOL_DBR *pr, POOL *pool, e_pool_op op)
 456 {
 457    strcpy(pr->PoolType, pool->pool_type);
 458    if (op == POOL_OP_CREATE) {
 459       pr->MaxVols = pool->max_volumes;
 460       pr->NumVols = 0;
 461    } else {          /* update pool */
 462       if (pr->MaxVols != pool->max_volumes) {
 463          pr->MaxVols = pool->max_volumes;
 464       }
 465       if (pr->MaxVols != 0 && pr->MaxVols < pr->NumVols) {
 466          pr->MaxVols = pr->NumVols;
 467       }
 468    }
 469    pr->UseOnce = pool->use_volume_once;
 470    pr->UseCatalog = pool->use_catalog;
 471    pr->AcceptAnyVolume = pool->accept_any_volume;
 472    pr->Recycle = pool->Recycle;
 473    pr->VolRetention = pool->VolRetention;
 474    pr->VolUseDuration = pool->VolUseDuration;
 475    pr->MaxVolJobs = pool->MaxVolJobs;
 476    pr->MaxVolFiles = pool->MaxVolFiles;
 477    pr->MaxVolBytes = pool->MaxVolBytes;
 478    pr->AutoPrune = pool->AutoPrune;
 479    pr->Recycle = pool->Recycle;
 480    if (pool->label_format) {
 481       strcpy(pr->LabelFormat, pool->label_format);
 482    } else {
 483       strcpy(pr->LabelFormat, "*");    /* none */
 484    }
 485 }
 486
 487
 488 /*
 489  * Create a pool record from a given Pool resource
 490  *   Also called from backup.c
 491  * Returns: -1  on error
 492  *           0  record already exists
 493  *           1  record created
 494  */
 495
 496 int create_pool(JCR *jcr, B_DB *db, POOL *pool, e_pool_op op)
 497 {
 498    POOL_DBR  pr;
 499
 500    memset(&pr, 0, sizeof(POOL_DBR));
 501
 502    strcpy(pr.Name, pool->hdr.name);
 503
 504    if (db_get_pool_record(jcr, db, &pr)) {
 505       /* Pool Exists */
 506       if (op == POOL_OP_UPDATE) {  /* update request */
 507          set_pooldbr_from_poolres(&pr, pool, op);
 508          db_update_pool_record(jcr, db, &pr);
 509       }
 510       return 0;                       /* exists */
 511    }
 512
 513    set_pooldbr_from_poolres(&pr, pool, op);
 514
 515    if (!db_create_pool_record(jcr, db, &pr)) {
 516       return -1;                      /* error */
 517    }
 518    return 1;
 519 }
 520
 521
 522
 523 /*
 524  * Create a Pool Record in the database.
 525  *  It is always created from the Resource record.
 526  */
 527 static int create_cmd(UAContext *ua, const char *cmd)
 528 {
 529    POOL *pool;
 530
 531    if (!open_db(ua)) {
 532       return 1;
 533    }
 534
 535    pool = get_pool_resource(ua);
 536    if (!pool) {
 537       return 1;
 538    }
 539
 540    switch (create_pool(ua->jcr, ua->db, pool, POOL_OP_CREATE)) {
 541    case 0:
 542       bsendmsg(ua, _("Error: Pool %s already exists.\n"
 543                "Use update to change it.\n"), pool->hdr.name);
 544       break;
 545
 546    case -1:
 547       bsendmsg(ua, "%s", db_strerror(ua->db));
 548       break;
 549
 550    default:
 551      break;
 552    }
 553    bsendmsg(ua, _("Pool %s created.\n"), pool->hdr.name);
 554    return 1;
 555 }
 556
 557
 558 /*
 559  * Set a new address in a Client resource. We do this only
 560  *  if the Console name is the same as the Client name
 561  *  and the Console can access the client.
 562  */
 563 static int setip_cmd(UAContext *ua, const char *cmd)
 564 {
 565    CLIENT *client;
 566    char mybuf[1024];
 567    char *buf = mybuf;
 568    if (!ua->cons || !acl_access_ok(ua, Client_ACL, ua->cons->hdr.name)) {
 569       bsendmsg(ua, _("Illegal command from this console.\n"));
 570       return 1;
 571    }
 572    LockRes();
 573    client = (CLIENT *)GetResWithName(R_CLIENT, ua->cons->hdr.name);
 574
 575    if (!client) {
 576       bsendmsg(ua, _("Client \"%s\" not found.\n"), ua->cons->hdr.name);
 577       goto get_out;
 578    }
 579    if (client->address) {
 580       free(client->address);
 581    }
 582 #ifdef HAVE_INET_NTOP
 583    inet_ntop(ua->UA_sock->client_addr.sa_family,
 584                 &(ua->UA_sock->client_addr), buf,
 585                 ua->UA_sock->client_addr.sa_family == AF_INET ?
 586                 sizeof(sockaddr_in) : sizeof(sockaddr_in6));
 587 #else
 588    buf = inet_ntoa(((struct sockaddr_in *)&(ua->UA_sock->client_addr))->sin_addr);
 589 #endif
 590
 591    client->address = bstrdup(buf);
 592    bsendmsg(ua, _("Client \"%s\" address set to %s\n"),
 593             client->hdr.name, client->address);
 594 get_out:
 595    UnlockRes();
 596    return 1;
 597 }
 598
 599
 600 /*
 601  * Update a Pool Record in the database.
 602  *  It is always updated from the Resource record.
 603  *
 604  *    update pool=<pool-name>
 605  *         updates pool from Pool resource
 606  *    update media pool=<pool-name> volume=<volume-name>
 607  *         changes pool info for volume
 608  *    update slots [scan=...]
 609  *         updates autochanger slots
 610  */
 611 static int update_cmd(UAContext *ua, const char *cmd)
 612 {
 613    static const char *kw[] = {
 614       N_("media"),  /* 0 */
 615       N_("volume"), /* 1 */
 616       N_("pool"),   /* 2 */
 617       N_("slots"),  /* 3 */
 618       NULL};
 619
 620    if (!open_db(ua)) {
 621       return 1;
 622    }
 623
 624    switch (find_arg_keyword(ua, kw)) {
 625    case 0:
 626    case 1:
 627       update_volume(ua);
 628       return 1;
 629    case 2:
 630       update_pool(ua);
 631       return 1;
 632    case 3:
 633       update_slots(ua);
 634       return 1;
 635    default:
 636       break;
 637    }
 638
 639    start_prompt(ua, _("Update choice:\n"));
 640    add_prompt(ua, _("Volume parameters"));
 641    add_prompt(ua, _("Pool from resource"));
 642    add_prompt(ua, _("Slots from autochanger"));
 643    switch (do_prompt(ua, _("item"), _("Choose catalog item to update"), NULL, 0)) {
 644    case 0:
 645       update_volume(ua);
 646       break;
 647    case 1:
 648       update_pool(ua);
 649       break;
 650    case 2:
 651       update_slots(ua);
 652       break;
 653    default:
 654       break;
 655    }
 656    return 1;
 657 }
 658
 659 static void update_volstatus(UAContext *ua, const char *val, MEDIA_DBR *mr)
 660 {
 661    POOLMEM *query = get_pool_memory(PM_MESSAGE);
 662    const char *kw[] = {
 663       "Append",
 664       "Archive",
 665       "Disabled",
 666       "Full",
 667       "Used",
 668       "Cleaning",
 669       "Recycle",
 670       "Read-Only",
 671       NULL};
 672    bool found = false;
 673    int i;
 674
 675    for (i=0; kw[i]; i++) {
 676       if (strcasecmp(val, kw[i]) == 0) {
 677          found = true;
 678          break;
 679       }
 680    }
 681    if (!found) {
 682       bsendmsg(ua, _("Invalid VolStatus specified: %s\n"), val);
 683    } else {
 684       bstrncpy(mr->VolStatus, kw[i], sizeof(mr->VolStatus));
 685       Mmsg(&query, "UPDATE Media SET VolStatus='%s' WHERE MediaId=%u",
 686          mr->VolStatus, mr->MediaId);
 687       if (!db_sql_query(ua->db, query, NULL, NULL)) {
 688          bsendmsg(ua, "%s", db_strerror(ua->db));
 689       } else {
 690          bsendmsg(ua, _("New Volume status is: %s\n"), mr->VolStatus);
 691       }
 692    }
 693    free_pool_memory(query);
 694 }
 695
 696 static void update_volretention(UAContext *ua, char *val, MEDIA_DBR *mr)
 697 {
 698    char ed1[150];
 699    POOLMEM *query;
 700    if (!duration_to_utime(val, &mr->VolRetention)) {
 701       bsendmsg(ua, _("Invalid retention period specified: %s\n"), val);
 702       return;
 703    }
 704    query = get_pool_memory(PM_MESSAGE);
 705    Mmsg(&query, "UPDATE Media SET VolRetention=%s WHERE MediaId=%u",
 706       edit_uint64(mr->VolRetention, ed1), mr->MediaId);
 707    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 708       bsendmsg(ua, "%s", db_strerror(ua->db));
 709    } else {
 710       bsendmsg(ua, _("New retention period is: %s\n"),
 711          edit_utime(mr->VolRetention, ed1, sizeof(ed1)));
 712    }
 713    free_pool_memory(query);
 714 }
 715
 716 static void update_voluseduration(UAContext *ua, char *val, MEDIA_DBR *mr)
 717 {
 718    char ed1[150];
 719    POOLMEM *query;
 720
 721    if (!duration_to_utime(val, &mr->VolUseDuration)) {
 722       bsendmsg(ua, _("Invalid use duration specified: %s\n"), val);
 723       return;
 724    }
 725    query = get_pool_memory(PM_MESSAGE);
 726    Mmsg(&query, "UPDATE Media SET VolUseDuration=%s WHERE MediaId=%u",
 727       edit_uint64(mr->VolUseDuration, ed1), mr->MediaId);
 728    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 729       bsendmsg(ua, "%s", db_strerror(ua->db));
 730    } else {
 731       bsendmsg(ua, _("New use duration is: %s\n"),
 732          edit_utime(mr->VolUseDuration, ed1, sizeof(ed1)));
 733    }
 734    free_pool_memory(query);
 735 }
 736
 737 static void update_volmaxjobs(UAContext *ua, char *val, MEDIA_DBR *mr)
 738 {
 739    POOLMEM *query = get_pool_memory(PM_MESSAGE);
 740    Mmsg(&query, "UPDATE Media SET MaxVolJobs=%s WHERE MediaId=%u",
 741       val, mr->MediaId);
 742    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 743       bsendmsg(ua, "%s", db_strerror(ua->db));
 744    } else {
 745       bsendmsg(ua, _("New max jobs is: %s\n"), val);
 746    }
 747    free_pool_memory(query);
 748 }
 749
 750 static void update_volmaxfiles(UAContext *ua, char *val, MEDIA_DBR *mr)
 751 {
 752    POOLMEM *query = get_pool_memory(PM_MESSAGE);
 753    Mmsg(&query, "UPDATE Media SET MaxVolFiles=%s WHERE MediaId=%u",
 754       val, mr->MediaId);
 755    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 756       bsendmsg(ua, "%s", db_strerror(ua->db));
 757    } else {
 758       bsendmsg(ua, _("New max files is: %s\n"), val);
 759    }
 760    free_pool_memory(query);
 761 }
 762
 763 static void update_volmaxbytes(UAContext *ua, char *val, MEDIA_DBR *mr)
 764 {
 765    uint64_t maxbytes;
 766    char ed1[50];
 767    POOLMEM *query;
 768
 769    if (!size_to_uint64(val, strlen(val), &maxbytes)) {
 770       bsendmsg(ua, _("Invalid max. bytes specification: %s\n"), val);
 771       return;
 772    }
 773    query = get_pool_memory(PM_MESSAGE);
 774    Mmsg(&query, "UPDATE Media SET MaxVolBytes=%s WHERE MediaId=%u",
 775       edit_uint64(maxbytes, ed1), mr->MediaId);
 776    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 777       bsendmsg(ua, "%s", db_strerror(ua->db));
 778    } else {
 779       bsendmsg(ua, _("New Max bytes is: %s\n"), edit_uint64(maxbytes, ed1));
 780    }
 781    free_pool_memory(query);
 782 }
 783
 784 static void update_volrecycle(UAContext *ua, char *val, MEDIA_DBR *mr)
 785 {
 786    int recycle;
 787    POOLMEM *query;
 788    if (strcasecmp(val, _("yes")) == 0) {
 789       recycle = 1;
 790    } else if (strcasecmp(val, _("no")) == 0) {
 791       recycle = 0;
 792    } else {
 793       bsendmsg(ua, _("Invalid value. It must by yes or no.\n"));
 794       return;
 795    }
 796    query = get_pool_memory(PM_MESSAGE);
 797    Mmsg(&query, "UPDATE Media SET Recycle=%d WHERE MediaId=%u",
 798       recycle, mr->MediaId);
 799    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 800       bsendmsg(ua, "%s", db_strerror(ua->db));
 801    } else {
 802       bsendmsg(ua, _("New Recycle flag is: %s\n"),
 803          mr->Recycle==1?_("yes"):_("no"));
 804    }
 805    free_pool_memory(query);
 806 }
 807
 808 /* Modify the Pool in which this Volume is located */
 809 static void update_vol_pool(UAContext *ua, char *val, MEDIA_DBR *mr, POOL_DBR *opr)
 810 {
 811    POOL_DBR pr;
 812    POOLMEM *query;
 813
 814    memset(&pr, 0, sizeof(pr));
 815    bstrncpy(pr.Name, val, sizeof(pr.Name));
 816    if (!get_pool_dbr(ua, &pr)) {
 817       return;
 818    }
 819    mr->PoolId = pr.PoolId;            /* set new PoolId */
 820    /*
 821     */
 822    query = get_pool_memory(PM_MESSAGE);
 823    db_lock(ua->db);
 824    Mmsg(&query, "UPDATE Media SET PoolId=%d WHERE MediaId=%u",
 825       mr->PoolId, mr->MediaId);
 826    if (!db_sql_query(ua->db, query, NULL, NULL)) {
 827       bsendmsg(ua, "%s", db_strerror(ua->db));
 828    } else {
 829       bsendmsg(ua, _("New Pool is: %s\n"), pr.Name);
 830       opr->NumVols--;
 831       if (!db_update_pool_record(ua->jcr, ua->db, opr)) {
 832          bsendmsg(ua, "%s", db_strerror(ua->db));
 833       }
 834       pr.NumVols++;
 835       if (!db_update_pool_record(ua->jcr, ua->db, &pr)) {
 836          bsendmsg(ua, "%s", db_strerror(ua->db));
 837       }
 838       db_make_inchanger_unique(ua->jcr, ua->db, mr);
 839    }
 840    db_unlock(ua->db);
 841    free_pool_memory(query);
 842 }
 843
 844 /*
 845  * Refresh the Volume information from the Pool record
 846  */
 847 static void update_volfrompool(UAContext *ua, MEDIA_DBR *mr)
 848 {
 849    POOL_DBR pr;
 850
 851    memset(&pr, 0, sizeof(pr));
 852    pr.PoolId = mr->PoolId;
 853    if (!get_pool_dbr(ua, &pr)) {
 854       return;
 855    }
 856    set_pool_dbr_defaults_in_media_dbr(mr, &pr);
 857    if (!db_update_media_defaults(ua->jcr, ua->db, mr)) {
 858       bsendmsg(ua, _("Error updating Volume record: ERR=%s"), db_strerror(ua->db));
 859    } else {
 860       bsendmsg(ua, _("Volume defaults updated from Pool record.\n"));
 861    }
 862 }
 863
 864 /*
 865  * Refresh the Volume information from the Pool record
 866  *   for all Volumes
 867  */
 868 static void update_all_vols_from_pool(UAContext *ua)
 869 {
 870    POOL_DBR pr;
 871    MEDIA_DBR mr;
 872
 873    memset(&pr, 0, sizeof(pr));
 874    if (!get_pool_dbr(ua, &pr)) {
 875       return;
 876    }
 877    memset(&mr, 0, sizeof(mr));
 878    set_pool_dbr_defaults_in_media_dbr(&mr, &pr);
 879    mr.PoolId = pr.PoolId;
 880    if (!db_update_media_defaults(ua->jcr, ua->db, &mr)) {
 881       bsendmsg(ua, _("Error updating Volume records: ERR=%s"), db_strerror(ua->db));
 882    } else {
 883       bsendmsg(ua, _("All Volume defaults updated from Pool record.\n"));
 884    }
 885 }
 886
 887
 888 /*
 889  * Update a media record -- allows you to change the
 890  *  Volume status. E.g. if you want Bacula to stop
 891  *  writing on the volume, set it to anything other
 892  *  than Append.
 893  */
 894 static int update_volume(UAContext *ua)
 895 {
 896    MEDIA_DBR mr;
 897    POOL_DBR pr;
 898    POOLMEM *query;
 899    char ed1[130];
 900    bool done = false;
 901    const char *kw[] = {
 902       N_("VolStatus"),                /* 0 */
 903       N_("VolRetention"),             /* 1 */
 904       N_("VolUse"),                   /* 2 */
 905       N_("MaxVolJobs"),               /* 3 */
 906       N_("MaxVolFiles"),              /* 4 */
 907       N_("MaxVolBytes"),              /* 5 */
 908       N_("Recycle"),                  /* 6 */
 909       N_("Pool"),                     /* 7 */
 910       N_("FromPool"),                 /* 8 */
 911       N_("AllFromPool"),              /* 9 */
 912       NULL };
 913
 914    for (int i=0; kw[i]; i++) {
 915       int j;
 916       POOL_DBR pr;
 917       if ((j=find_arg_with_value(ua, kw[i])) > 0) {
 918          if (i != 9 && !select_media_dbr(ua, &mr)) {
 919             return 0;
 920          }
 921          switch (i) {
 922          case 0:
 923             update_volstatus(ua, ua->argv[j], &mr);
 924             break;
 925          case 1:
 926             update_volretention(ua, ua->argv[j], &mr);
 927             break;
 928          case 2:
 929             update_voluseduration(ua, ua->argv[j], &mr);
 930             break;
 931          case 3:
 932             update_volmaxjobs(ua, ua->argv[j], &mr);
 933             break;
 934          case 4:
 935             update_volmaxfiles(ua, ua->argv[j], &mr);
 936             break;
 937          case 5:
 938             update_volmaxbytes(ua, ua->argv[j], &mr);
 939             break;
 940          case 6:
 941             update_volrecycle(ua, ua->argv[j], &mr);
 942             break;
 943          case 7:
 944             memset(&pr, 0, sizeof(POOL_DBR));
 945             pr.PoolId = mr.PoolId;
 946             if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
 947                bsendmsg(ua, "%s", db_strerror(ua->db));
 948                break;
 949             }
 950             update_vol_pool(ua, ua->argv[j], &mr, &pr);
 951             break;
 952          case 8:
 953             update_volfrompool(ua, &mr);
 954             break;
 955          case 9:
 956             update_all_vols_from_pool(ua);
 957          }
 958          done = true;
 959       }
 960    }
 961
 962    for ( ; !done; ) {
 963       if (!select_media_dbr(ua, &mr)) {
 964          return 0;
 965       }
 966       bsendmsg(ua, _("Updating Volume \"%s\"\n"), mr.VolumeName);
 967       start_prompt(ua, _("Parameters to modify:\n"));
 968       add_prompt(ua, _("Volume Status"));
 969       add_prompt(ua, _("Volume Retention Period"));
 970       add_prompt(ua, _("Volume Use Duration"));
 971       add_prompt(ua, _("Maximum Volume Jobs"));
 972       add_prompt(ua, _("Maximum Volume Files"));
 973       add_prompt(ua, _("Maximum Volume Bytes"));
 974       add_prompt(ua, _("Recycle Flag"));
 975       add_prompt(ua, _("Slot"));
 976       add_prompt(ua, _("InChanger Flag"));
 977       add_prompt(ua, _("Volume Files"));
 978       add_prompt(ua, _("Pool"));
 979       add_prompt(ua, _("Volume from Pool"));
 980       add_prompt(ua, _("All Volumes from Pool"));
 981       add_prompt(ua, _("Done"));
 982       switch (do_prompt(ua, "", _("Select parameter to modify"), NULL, 0)) {
 983       case 0:                         /* Volume Status */
 984          /* Modify Volume Status */
 985          bsendmsg(ua, _("Current Volume status is: %s\n"), mr.VolStatus);
 986          start_prompt(ua, _("Possible Values are:\n"));
 987          add_prompt(ua, "Append");      /* Better not translate these as */
 988          add_prompt(ua, "Archive");     /* They are known in the database code */
 989          add_prompt(ua, "Disabled");
 990          add_prompt(ua, "Full");
 991          add_prompt(ua, "Used");
 992          add_prompt(ua, "Cleaning");
 993          if (strcmp(mr.VolStatus, "Purged") == 0) {
 994             add_prompt(ua, "Recycle");
 995          }
 996          add_prompt(ua, "Read-Only");
 997          if (do_prompt(ua, "", _("Choose new Volume Status"), ua->cmd, sizeof(mr.VolStatus)) < 0) {
 998             return 1;
 999          }
1000          update_volstatus(ua, ua->cmd, &mr);
1001          break;
1002       case 1:                         /* Retention */
1003          bsendmsg(ua, _("Current retention period is: %s\n"),
1004             edit_utime(mr.VolRetention, ed1, sizeof(ed1)));
1005          if (!get_cmd(ua, _("Enter Volume Retention period: "))) {
1006             return 0;
1007          }
1008          update_volretention(ua, ua->cmd, &mr);
1009          break;
1010
1011       case 2:                         /* Use Duration */
1012          bsendmsg(ua, _("Current use duration is: %s\n"),
1013             edit_utime(mr.VolUseDuration, ed1, sizeof(ed1)));
1014          if (!get_cmd(ua, _("Enter Volume Use Duration: "))) {
1015             return 0;
1016          }
1017          update_voluseduration(ua, ua->cmd, &mr);
1018          break;
1019
1020       case 3:                         /* Max Jobs */
1021          bsendmsg(ua, _("Current max jobs is: %u\n"), mr.MaxVolJobs);
1022          if (!get_pint(ua, _("Enter new Maximum Jobs: "))) {
1023             return 0;
1024          }
1025          update_volmaxjobs(ua, ua->cmd, &mr);
1026          break;
1027
1028       case 4:                         /* Max Files */
1029          bsendmsg(ua, _("Current max files is: %u\n"), mr.MaxVolFiles);
1030          if (!get_pint(ua, _("Enter new Maximum Files: "))) {
1031             return 0;
1032          }
1033          update_volmaxfiles(ua, ua->cmd, &mr);
1034          break;
1035
1036       case 5:                         /* Max Bytes */
1037          bsendmsg(ua, _("Current value is: %s\n"), edit_uint64(mr.MaxVolBytes, ed1));
1038          if (!get_cmd(ua, _("Enter new Maximum Bytes: "))) {
1039             return 0;
1040          }
1041          update_volmaxbytes(ua, ua->cmd, &mr);
1042          break;
1043
1044
1045       case 6:                         /* Recycle */
1046          bsendmsg(ua, _("Current recycle flag is: %s\n"),
1047             mr.Recycle==1?_("yes"):_("no"));
1048          if (!get_yesno(ua, _("Enter new Recycle status: "))) {
1049             return 0;
1050          }
1051          update_volrecycle(ua, ua->cmd, &mr);
1052          break;
1053
1054       case 7:                         /* Slot */
1055          int Slot;
1056
1057          memset(&pr, 0, sizeof(POOL_DBR));
1058          pr.PoolId = mr.PoolId;
1059          if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
1060             bsendmsg(ua, "%s", db_strerror(ua->db));
1061             return 0;
1062          }
1063          bsendmsg(ua, _("Current Slot is: %d\n"), mr.Slot);
1064          if (!get_pint(ua, _("Enter new Slot: "))) {
1065             return 0;
1066          }
1067          Slot = ua->pint32_val;
1068          if (pr.MaxVols > 0 && Slot > (int)pr.MaxVols) {
1069             bsendmsg(ua, _("Invalid slot, it must be between 0 and %d\n"),
1070                pr.MaxVols);
1071             break;
1072          }
1073          mr.Slot = Slot;
1074          /*
1075           * Make sure to use db_update... rather than doing this directly,
1076           *   so that any Slot is handled correctly.
1077           */
1078          if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
1079             bsendmsg(ua, _("Error updating media record Slot: ERR=%s"), db_strerror(ua->db));
1080          } else {
1081             bsendmsg(ua, _("New Slot is: %d\n"), mr.Slot);
1082          }
1083          break;
1084
1085       case 8:                         /* InChanger */
1086          bsendmsg(ua, _("Current InChanger flag is: %d\n"), mr.InChanger);
1087          if (!get_yesno(ua, _("Set InChanger flag? yes/no: "))) {
1088             return 0;
1089          }
1090          mr.InChanger = ua->pint32_val;
1091          /*
1092           * Make sure to use db_update... rather than doing this directly,
1093           *   so that any Slot is handled correctly.
1094           */
1095          if (!db_update_media_record(ua->jcr, ua->db, &mr)) {
1096             bsendmsg(ua, _("Error updating media record Slot: ERR=%s"), db_strerror(ua->db));
1097          } else {
1098             bsendmsg(ua, _("New InChanger flag is: %d\n"), mr.InChanger);
1099          }
1100          break;
1101
1102
1103       case 9:                         /* Volume Files */
1104          int32_t VolFiles;
1105          bsendmsg(ua, _("Warning changing Volume Files can result\n"
1106                         "in loss of data on your Volume\n\n"));
1107          bsendmsg(ua, _("Current Volume Files is: %u\n"), mr.VolFiles);
1108          if (!get_pint(ua, _("Enter new number of Files for Volume: "))) {
1109             return 0;
1110          }
1111          VolFiles = ua->pint32_val;
1112          if (VolFiles != (int)(mr.VolFiles + 1)) {
1113             bsendmsg(ua, _("Normally, you should only increase Volume Files by one!\n"));
1114             if (!get_yesno(ua, _("Continue? (yes/no): ")) || ua->pint32_val == 0) {
1115                break;
1116             }
1117          }
1118          query = get_pool_memory(PM_MESSAGE);
1119          Mmsg(&query, "UPDATE Media SET VolFiles=%u WHERE MediaId=%u",
1120             VolFiles, mr.MediaId);
1121          if (!db_sql_query(ua->db, query, NULL, NULL)) {
1122             bsendmsg(ua, "%s", db_strerror(ua->db));
1123          } else {
1124             bsendmsg(ua, _("New Volume Files is: %u\n"), VolFiles);
1125          }
1126          free_pool_memory(query);
1127          break;
1128
1129       case 10:                        /* Volume's Pool */
1130          memset(&pr, 0, sizeof(POOL_DBR));
1131          pr.PoolId = mr.PoolId;
1132          if (!db_get_pool_record(ua->jcr, ua->db, &pr)) {
1133             bsendmsg(ua, "%s", db_strerror(ua->db));
1134             return 0;
1135          }
1136          bsendmsg(ua, _("Current Pool is: %s\n"), pr.Name);
1137          if (!get_cmd(ua, _("Enter new Pool name: "))) {
1138             return 0;
1139          }
1140          update_vol_pool(ua, ua->cmd, &mr, &pr);
1141          return 1;
1142
1143       case 11:
1144          update_volfrompool(ua, &mr);
1145          break;
1146       case 12:
1147          update_all_vols_from_pool(ua);
1148          break;
1149       default:                        /* Done or error */
1150          bsendmsg(ua, "Selection done.\n");
1151          return 1;
1152       }
1153    }
1154    return 1;
1155 }
1156
1157 /*
1158  * Update pool record -- pull info from current POOL resource
1159  */
1160 static int update_pool(UAContext *ua)
1161 {
1162    POOL_DBR  pr;
1163    int id;
1164    POOL *pool;
1165    POOLMEM *query;
1166
1167    pool = get_pool_resource(ua);
1168    if (!pool) {
1169       return 0;
1170    }
1171
1172    memset(&pr, 0, sizeof(pr));
1173    strcpy(pr.Name, pool->hdr.name);
1174    if (!get_pool_dbr(ua, &pr)) {
1175       return 0;
1176    }
1177
1178    set_pooldbr_from_poolres(&pr, pool, POOL_OP_UPDATE); /* update */
1179
1180    id = db_update_pool_record(ua->jcr, ua->db, &pr);
1181    if (id <= 0) {
1182       bsendmsg(ua, _("db_update_pool_record returned %d. ERR=%s\n"),
1183          id, db_strerror(ua->db));
1184    }
1185    query = get_pool_memory(PM_MESSAGE);
1186    Mmsg(&query, list_pool, pr.PoolId);
1187    db_list_sql_query(ua->jcr, ua->db, query, prtit, ua, 1, HORZ_LIST);
1188    free_pool_memory(query);
1189    bsendmsg(ua, _("Pool DB record updated from resource.\n"));
1190    return 1;
1191 }
1192
1193
1194 static void do_storage_setdebug(UAContext *ua, STORE *store, int level, int trace_flag)
1195 {
1196    BSOCK *sd;
1197
1198    ua->jcr->store = store;
1199    /* Try connecting for up to 15 seconds */
1200    bsendmsg(ua, _("Connecting to Storage daemon %s at %s:%d\n"),
1201       store->hdr.name, store->address, store->SDport);
1202    if (!connect_to_storage_daemon(ua->jcr, 1, 15, 0)) {
1203       bsendmsg(ua, _("Failed to connect to Storage daemon.\n"));
1204       return;
1205    }
1206    Dmsg0(120, _("Connected to storage daemon\n"));
1207    sd = ua->jcr->store_bsock;
1208    bnet_fsend(sd, "setdebug=%d trace=%d\n", level, trace_flag);
1209    if (bnet_recv(sd) >= 0) {
1210       bsendmsg(ua, "%s", sd->msg);
1211    }
1212    bnet_sig(sd, BNET_TERMINATE);
1213    bnet_close(sd);
1214    ua->jcr->store_bsock = NULL;
1215    return;
1216 }
1217
1218 static void do_client_setdebug(UAContext *ua, CLIENT *client, int level, int trace_flag)
1219 {
1220    BSOCK *fd;
1221
1222    /* Connect to File daemon */
1223
1224    ua->jcr->client = client;
1225    /* Try to connect for 15 seconds */
1226    bsendmsg(ua, _("Connecting to Client %s at %s:%d\n"),
1227       client->hdr.name, client->address, client->FDport);
1228    if (!connect_to_file_daemon(ua->jcr, 1, 15, 0)) {
1229       bsendmsg(ua, _("Failed to connect to Client.\n"));
1230       return;
1231    }
1232    Dmsg0(120, "Connected to file daemon\n");
1233    fd = ua->jcr->file_bsock;
1234    bnet_fsend(fd, "setdebug=%d trace=%d\n", level, trace_flag);
1235    if (bnet_recv(fd) >= 0) {
1236       bsendmsg(ua, "%s", fd->msg);
1237    }
1238    bnet_sig(fd, BNET_TERMINATE);
1239    bnet_close(fd);
1240    ua->jcr->file_bsock = NULL;
1241    return;
1242 }
1243
1244
1245 static void do_all_setdebug(UAContext *ua, int level, int trace_flag)
1246 {
1247    STORE *store, **unique_store;
1248    CLIENT *client, **unique_client;
1249    int i, j, found;
1250
1251    /* Director */
1252    debug_level = level;
1253
1254    /* Count Storage items */
1255    LockRes();
1256    store = NULL;
1257    for (i=0; (store = (STORE *)GetNextRes(R_STORAGE, (RES *)store)); i++)
1258       { }
1259    unique_store = (STORE **) malloc(i * sizeof(STORE));
1260    /* Find Unique Storage address/port */
1261    store = (STORE *)GetNextRes(R_STORAGE, NULL);
1262    i = 0;
1263    unique_store[i++] = store;
1264    while ((store = (STORE *)GetNextRes(R_STORAGE, (RES *)store))) {
1265       found = 0;
1266       for (j=0; j<i; j++) {
1267          if (strcmp(unique_store[j]->address, store->address) == 0 &&
1268              unique_store[j]->SDport == store->SDport) {
1269             found = 1;
1270             break;
1271          }
1272       }
1273       if (!found) {
1274          unique_store[i++] = store;
1275          Dmsg2(140, "Stuffing: %s:%d\n", store->address, store->SDport);
1276       }
1277    }
1278    UnlockRes();
1279
1280    /* Call each unique Storage daemon */
1281    for (j=0; j<i; j++) {
1282       do_storage_setdebug(ua, unique_store[j], level, trace_flag);
1283    }
1284    free(unique_store);
1285
1286    /* Count Client items */
1287    LockRes();
1288    client = NULL;
1289    for (i=0; (client = (CLIENT *)GetNextRes(R_CLIENT, (RES *)client)); i++)
1290       { }
1291    unique_client = (CLIENT **) malloc(i * sizeof(CLIENT));
1292    /* Find Unique Client address/port */
1293    client = (CLIENT *)GetNextRes(R_CLIENT, NULL);
1294    i = 0;
1295    unique_client[i++] = client;
1296    while ((client = (CLIENT *)GetNextRes(R_CLIENT, (RES *)client))) {
1297       found = 0;
1298       for (j=0; j<i; j++) {
1299          if (strcmp(unique_client[j]->address, client->address) == 0 &&
1300              unique_client[j]->FDport == client->FDport) {
1301             found = 1;
1302             break;
1303          }
1304       }
1305       if (!found) {
1306          unique_client[i++] = client;
1307          Dmsg2(140, "Stuffing: %s:%d\n", client->address, client->FDport);
1308       }
1309    }
1310    UnlockRes();
1311
1312    /* Call each unique File daemon */
1313    for (j=0; j<i; j++) {
1314       do_client_setdebug(ua, unique_client[j], level, trace_flag);
1315    }
1316    free(unique_client);
1317 }
1318
1319 /*
1320  * setdebug level=nn all trace=1/0
1321  */
1322 static int setdebug_cmd(UAContext *ua, const char *cmd)
1323 {
1324    STORE *store;
1325    CLIENT *client;
1326    int level;
1327    int trace_flag = -1;
1328    int i;
1329
1330    if (!open_db(ua)) {
1331       return 1;
1332    }
1333    Dmsg1(120, "setdebug:%s:\n", cmd);
1334
1335    level = -1;
1336    i = find_arg_with_value(ua, _("level"));
1337    if (i >= 0) {
1338       level = atoi(ua->argv[i]);
1339    }
1340    if (level < 0) {
1341       if (!get_pint(ua, _("Enter new debug level: "))) {
1342          return 1;
1343       }
1344       level = ua->pint32_val;
1345    }
1346
1347    /* Look for trace flag. -1 => not change */
1348    i = find_arg_with_value(ua, _("trace"));
1349    if (i >= 0) {
1350       trace_flag = atoi(ua->argv[i]);
1351       if (trace_flag > 0) {
1352          trace_flag = 1;
1353       }
1354    }
1355
1356    /* General debug? */
1357    for (i=1; i<ua->argc; i++) {
1358       if (strcasecmp(ua->argk[i], _("all")) == 0) {
1359          do_all_setdebug(ua, level, trace_flag);
1360          return 1;
1361       }
1362       if (strcasecmp(ua->argk[i], _("dir")) == 0 ||
1363           strcasecmp(ua->argk[i], _("director")) == 0) {
1364          debug_level = level;
1365          set_trace(trace_flag);
1366          return 1;
1367       }
1368       if (strcasecmp(ua->argk[i], _("client")) == 0 ||
1369           strcasecmp(ua->argk[i], _("fd")) == 0) {
1370          client = NULL;
1371          if (ua->argv[i]) {
1372             client = (CLIENT *)GetResWithName(R_CLIENT, ua->argv[i]);
1373             if (client) {
1374                do_client_setdebug(ua, client, level, trace_flag);
1375                return 1;
1376             }
1377          }
1378          client = select_client_resource(ua);
1379          if (client) {
1380             do_client_setdebug(ua, client, level, trace_flag);
1381             return 1;
1382          }
1383       }
1384
1385       if (strcasecmp(ua->argk[i], _("store")) == 0 ||
1386           strcasecmp(ua->argk[i], _("storage")) == 0 ||
1387           strcasecmp(ua->argk[i], _("sd")) == 0) {
1388          store = NULL;
1389          if (ua->argv[i]) {
1390             store = (STORE *)GetResWithName(R_STORAGE, ua->argv[i]);
1391             if (store) {
1392                do_storage_setdebug(ua, store, level, trace_flag);
1393                return 1;
1394             }
1395          }
1396          store = get_storage_resource(ua, 0);
1397          if (store) {
1398             do_storage_setdebug(ua, store, level, trace_flag);
1399             return 1;
1400          }
1401       }
1402    }
1403    /*
1404     * We didn't find an appropriate keyword above, so
1405     * prompt the user.
1406     */
1407    start_prompt(ua, _("Available daemons are: \n"));
1408    add_prompt(ua, _("Director"));
1409    add_prompt(ua, _("Storage"));
1410    add_prompt(ua, _("Client"));
1411    add_prompt(ua, _("All"));
1412    switch(do_prompt(ua, "", _("Select daemon type to set debug level"), NULL, 0)) {
1413    case 0:                         /* Director */
1414       debug_level = level;
1415       set_trace(trace_flag);
1416       break;
1417    case 1:
1418       store = get_storage_resource(ua, 0);
1419       if (store) {
1420          do_storage_setdebug(ua, store, level, trace_flag);
1421       }
1422       break;
1423    case 2:
1424       client = select_client_resource(ua);
1425       if (client) {
1426          do_client_setdebug(ua, client, level, trace_flag);
1427       }
1428       break;
1429    case 3:
1430       do_all_setdebug(ua, level, trace_flag);
1431       break;
1432    default:
1433       break;
1434    }
1435    return 1;
1436 }
1437
1438 /*
1439  * Turn debug tracing to file on/off
1440  */
1441 static int trace_cmd(UAContext *ua, const char *cmd)
1442 {
1443    char *onoff;
1444
1445    if (ua->argc != 2) {
1446       if (!get_cmd(ua, _("Turn on or off? "))) {
1447             return 1;
1448       }
1449       onoff = ua->cmd;
1450    } else {
1451       onoff = ua->argk[1];
1452    }
1453
1454    set_trace((strcasecmp(onoff, _("off")) == 0) ? false : true);
1455    return 1;
1456
1457 }
1458
1459 static int var_cmd(UAContext *ua, const char *cmd)
1460 {
1461    POOLMEM *val = get_pool_memory(PM_FNAME);
1462    char *var;
1463
1464    if (!open_db(ua)) {
1465       return 1;
1466    }
1467    for (var=ua->cmd; *var != ' '; ) {    /* skip command */
1468       var++;
1469    }
1470    while (*var == ' ') {                 /* skip spaces */
1471       var++;
1472    }
1473    Dmsg1(100, "Var=%s:\n", var);
1474    variable_expansion(ua->jcr, var, &val);
1475    bsendmsg(ua, "%s\n", val);
1476    free_pool_memory(val);
1477    return 1;
1478 }
1479
1480 static int estimate_cmd(UAContext *ua, const char *cmd)
1481 {
1482    JOB *job = NULL;
1483    CLIENT *client = NULL;
1484    FILESET *fileset = NULL;
1485    FILESET_DBR fsr;
1486    int listing = 0;
1487    char since[MAXSTRING];
1488    JCR *jcr = ua->jcr;
1489
1490    jcr->JobLevel = L_FULL;
1491    for (int i=1; i<ua->argc; i++) {
1492       if (strcasecmp(ua->argk[i], _("client")) == 0 ||
1493           strcasecmp(ua->argk[i], _("fd")) == 0) {
1494          if (ua->argv[i]) {
1495             client = (CLIENT *)GetResWithName(R_CLIENT, ua->argv[i]);
1496             continue;
1497          }
1498       }
1499       if (strcasecmp(ua->argk[i], _("job")) == 0) {
1500          if (ua->argv[i]) {
1501             job = (JOB *)GetResWithName(R_JOB, ua->argv[i]);
1502             continue;
1503          }
1504       }
1505       if (strcasecmp(ua->argk[i], _("fileset")) == 0) {
1506          if (ua->argv[i]) {
1507             fileset = (FILESET *)GetResWithName(R_FILESET, ua->argv[i]);
1508             continue;
1509          }
1510       }
1511       if (strcasecmp(ua->argk[i], _("listing")) == 0) {
1512          listing = 1;
1513          continue;
1514       }
1515       if (strcasecmp(ua->argk[i], _("level")) == 0) {
1516          if (!get_level_from_name(ua->jcr, ua->argv[i])) {
1517             bsendmsg(ua, _("Level %s not valid.\n"), ua->argv[i]);
1518          }
1519          continue;
1520       }
1521    }
1522    if (!job && !(client && fileset)) {
1523       if (!(job = select_job_resource(ua))) {
1524          return 1;
1525       }
1526    }
1527    if (!job) {
1528       job = (JOB *)GetResWithName(R_JOB, ua->argk[1]);
1529       if (!job) {
1530          bsendmsg(ua, _("No job specified.\n"));
1531          return 1;
1532       }
1533    }
1534    if (!client) {
1535       client = job->client;
1536    }
1537    if (!fileset) {
1538       fileset = job->fileset;
1539    }
1540    jcr->client = client;
1541    jcr->fileset = fileset;
1542    close_db(ua);
1543    ua->catalog = client->catalog;
1544
1545    if (!open_db(ua)) {
1546       return 1;
1547    }
1548
1549    jcr->job = job;
1550    jcr->JobType = JT_BACKUP;
1551    init_jcr_job_record(jcr);
1552
1553    if (!get_or_create_client_record(jcr)) {
1554       return 1;
1555    }
1556    if (!get_or_create_fileset_record(jcr, &fsr)) {
1557       return 1;
1558    }
1559
1560    get_level_since_time(ua->jcr, since, sizeof(since));
1561
1562    bsendmsg(ua, _("Connecting to Client %s at %s:%d\n"),
1563       job->client->hdr.name, job->client->address, job->client->FDport);
1564    if (!connect_to_file_daemon(jcr, 1, 15, 0)) {
1565       bsendmsg(ua, _("Failed to connect to Client.\n"));
1566       return 1;
1567    }
1568
1569    if (!send_include_list(jcr)) {
1570       bsendmsg(ua, _("Error sending include list.\n"));
1571       goto bail_out;
1572    }
1573
1574    if (!send_exclude_list(jcr)) {
1575       bsendmsg(ua, _("Error sending exclude list.\n"));
1576       goto bail_out;
1577    }
1578
1579    if (!send_level_command(jcr)) {
1580       goto bail_out;
1581    }
1582
1583    bnet_fsend(jcr->file_bsock, "estimate listing=%d\n", listing);
1584    while (bnet_recv(jcr->file_bsock) >= 0) {
1585       bsendmsg(ua, "%s", jcr->file_bsock->msg);
1586    }
1587
1588 bail_out:
1589    if (jcr->file_bsock) {
1590       bnet_sig(jcr->file_bsock, BNET_TERMINATE);
1591       bnet_close(jcr->file_bsock);
1592       jcr->file_bsock = NULL;
1593    }
1594    return 1;
1595 }
1596
1597
1598 /*
1599  * print time
1600  */
1601 static int time_cmd(UAContext *ua, const char *cmd)
1602 {
1603    char sdt[50];
1604    time_t ttime = time(NULL);
1605    struct tm tm;
1606    localtime_r(&ttime, &tm);
1607    strftime(sdt, sizeof(sdt), "%d-%b-%Y %H:%M:%S", &tm);
1608    bsendmsg(ua, "%s\n", sdt);
1609    return 1;
1610 }
1611
1612 /*
1613  * reload the conf file
1614  */
1615 extern "C" void reload_config(int sig);
1616
1617 static int reload_cmd(UAContext *ua, const char *cmd)
1618 {
1619    reload_config(1);
1620    return 1;
1621 }
1622
1623
1624
1625 /*
1626  * Delete Pool records (should purge Media with it).
1627  *
1628  *  delete pool=<pool-name>
1629  *  delete volume pool=<pool-name> volume=<name>
1630  *  delete job jobid=xxx
1631  */
1632 static int delete_cmd(UAContext *ua, const char *cmd)
1633 {
1634    static const char *keywords[] = {
1635       N_("volume"),
1636       N_("pool"),
1637       N_("job"),
1638       N_("jobid"),
1639       NULL};
1640
1641    if (!open_db(ua)) {
1642       return 1;
1643    }
1644
1645    switch (find_arg_keyword(ua, keywords)) {
1646    case 0:
1647       delete_volume(ua);
1648       return 1;
1649    case 1:
1650       delete_pool(ua);
1651       return 1;
1652    case 2:
1653    case 3:
1654       int i;
1655       while ((i=find_arg(ua, _("jobid"))) > 0) {
1656          delete_job(ua);
1657          *ua->argk[i] = 0;         /* zap keyword already visited */
1658       }
1659       return 1;
1660    default:
1661       break;
1662    }
1663
1664    bsendmsg(ua, _(
1665 "In general it is not a good idea to delete either a\n"
1666 "Pool or a Volume since they may contain data.\n\n"));
1667
1668    switch (do_keyword_prompt(ua, _("Choose catalog item to delete"), keywords)) {
1669    case 0:
1670       delete_volume(ua);
1671       break;
1672    case 1:
1673       delete_pool(ua);
1674       break;
1675    case 2:
1676       delete_job(ua);
1677       return 1;
1678    default:
1679       bsendmsg(ua, _("Nothing done.\n"));
1680       break;
1681    }
1682    return 1;
1683 }
1684
1685 static int delete_job(UAContext *ua)
1686 {
1687    POOLMEM *query = get_pool_memory(PM_MESSAGE);
1688    JobId_t JobId;
1689
1690    int i = find_arg_with_value(ua, _("jobid"));
1691    if (i >= 0) {
1692       JobId = str_to_int64(ua->argv[i]);
1693    } else if (!get_pint(ua, _("Enter JobId to delete: "))) {
1694       return 0;
1695    } else {
1696       JobId = ua->pint32_val;
1697    }
1698    Mmsg(&query, "DELETE FROM Job WHERE JobId=%u", JobId);
1699    db_sql_query(ua->db, query, NULL, (void *)NULL);
1700    Mmsg(&query, "DELETE FROM File WHERE JobId=%u", JobId);
1701    db_sql_query(ua->db, query, NULL, (void *)NULL);
1702    Mmsg(&query, "DELETE FROM JobMedia WHERE JobId=%u", JobId);
1703    db_sql_query(ua->db, query, NULL, (void *)NULL);
1704    free_pool_memory(query);
1705    bsendmsg(ua, _("Job %u and associated records deleted from the catalog.\n"), JobId);
1706    return 1;
1707 }
1708
1709 /*
1710  * Delete media records from database -- dangerous
1711  */
1712 static int delete_volume(UAContext *ua)
1713 {
1714    MEDIA_DBR mr;
1715
1716    if (!select_media_dbr(ua, &mr)) {
1717       return 1;
1718    }
1719    bsendmsg(ua, _("\nThis command will delete volume %s\n"
1720       "and all Jobs saved on that volume from the Catalog\n"),
1721       mr.VolumeName);
1722
1723    if (!get_yesno(ua, _("Are you sure you want to delete this Volume? (yes/no): "))) {
1724       return 1;
1725    }
1726    if (ua->pint32_val) {
1727       db_delete_media_record(ua->jcr, ua->db, &mr);
1728    }
1729    return 1;
1730 }
1731
1732 /*
1733  * Delete a pool record from the database -- dangerous
1734  */
1735 static int delete_pool(UAContext *ua)
1736 {
1737    POOL_DBR  pr;
1738
1739    memset(&pr, 0, sizeof(pr));
1740
1741    if (!get_pool_dbr(ua, &pr)) {
1742       return 1;
1743    }
1744    if (!get_yesno(ua, _("Are you sure you want to delete this Pool? (yes/no): "))) {
1745       return 1;
1746    }
1747    if (ua->pint32_val) {
1748       db_delete_pool_record(ua->jcr, ua->db, &pr);
1749    }
1750    return 1;
1751 }
1752
1753
1754 static void do_mount_cmd(UAContext *ua, const char *command)
1755 {
1756    STORE *store;
1757    BSOCK *sd;
1758    char dev_name[MAX_NAME_LENGTH];
1759
1760
1761    if (!open_db(ua)) {
1762       return;
1763    }
1764    Dmsg2(120, "%s: %s\n", command, ua->UA_sock->msg);
1765
1766    store = get_storage_resource(ua, 1);
1767    if (!store) {
1768       return;
1769    }
1770
1771    Dmsg2(120, "Found storage, MediaType=%s DevName=%s\n",
1772       store->media_type, store->dev_name);
1773
1774    ua->jcr->store = store;
1775    if (!connect_to_storage_daemon(ua->jcr, 10, SDConnectTimeout, 1)) {
1776       bsendmsg(ua, _("Failed to connect to Storage daemon.\n"));
1777       return;
1778    }
1779    sd = ua->jcr->store_bsock;
1780    strcpy(dev_name, store->dev_name);
1781    bash_spaces(dev_name);
1782    bnet_fsend(sd, "%s %s", command, dev_name);
1783    while (bnet_recv(sd) >= 0) {
1784       bsendmsg(ua, "%s", sd->msg);
1785    }
1786    bnet_sig(sd, BNET_TERMINATE);
1787    bnet_close(sd);
1788    ua->jcr->store_bsock = NULL;
1789 }
1790
1791 /*
1792  * mount [storage | device] <name>
1793  */
1794 static int mount_cmd(UAContext *ua, const char *cmd)
1795 {
1796    do_mount_cmd(ua, "mount");          /* mount */
1797    return 1;
1798 }
1799
1800
1801 /*
1802  * unmount [storage | device] <name>
1803  */
1804 static int unmount_cmd(UAContext *ua, const char *cmd)
1805 {
1806    do_mount_cmd(ua, "unmount");          /* unmount */
1807    return 1;
1808 }
1809
1810
1811 /*
1812  * release [storage | device] <name>
1813  */
1814 static int release_cmd(UAContext *ua, const char *cmd)
1815 {
1816    do_mount_cmd(ua, "release");          /* release */
1817    return 1;
1818 }
1819
1820
1821 /*
1822  * Switch databases
1823  *   use catalog=<name>
1824  */
1825 static int use_cmd(UAContext *ua, const char *cmd)
1826 {
1827    CAT *oldcatalog, *catalog;
1828
1829
1830    close_db(ua);                      /* close any previously open db */
1831    oldcatalog = ua->catalog;
1832
1833    if (!(catalog = get_catalog_resource(ua))) {
1834       ua->catalog = oldcatalog;
1835    } else {
1836       ua->catalog = catalog;
1837    }
1838    if (open_db(ua)) {
1839       bsendmsg(ua, _("Using Catalog name=%s DB=%s\n"),
1840          ua->catalog->hdr.name, ua->catalog->db_name);
1841    }
1842    return 1;
1843 }
1844
1845 int quit_cmd(UAContext *ua, const char *cmd)
1846 {
1847    ua->quit = TRUE;
1848    return 1;
1849 }
1850
1851 /*
1852  * Wait until no job is running
1853  */
1854 int wait_cmd(UAContext *ua, const char *cmd)
1855 {
1856    JCR *jcr;
1857    bmicrosleep(0, 200000);            /* let job actually start */
1858    for (bool running=true; running; ) {
1859       running = false;
1860       lock_jcr_chain();
1861       foreach_jcr(jcr) {
1862          if (jcr->JobId != 0) {
1863             running = true;
1864             free_locked_jcr(jcr);
1865             break;
1866          }
1867          free_locked_jcr(jcr);
1868       }
1869       unlock_jcr_chain();
1870       if (running) {
1871          bmicrosleep(1, 0);
1872       }
1873    }
1874    return 1;
1875 }
1876
1877
1878 static int help_cmd(UAContext *ua, const char *cmd)
1879 {
1880    unsigned int i;
1881
1882    bsendmsg(ua, _("  Command    Description\n  =======    ===========\n"));
1883    for (i=0; i<comsize; i++) {
1884       bsendmsg(ua, _("  %-10s %s\n"), _(commands[i].key), _(commands[i].help));
1885    }
1886    bsendmsg(ua, _("\nWhen at a prompt, entering a period cancels the command.\n\n"));
1887    return 1;
1888 }
1889
1890 static int version_cmd(UAContext *ua, const char *cmd)
1891 {
1892    bsendmsg(ua, "%s Version: " VERSION " (" BDATE ")\n", my_name);
1893    return 1;
1894 }
1895
1896
1897 /* A bit brain damaged in that if the user has not done
1898  * a "use catalog xxx" command, we simply find the first
1899  * catalog resource and open it.
1900  */
1901 int open_db(UAContext *ua)
1902 {
1903    if (ua->db) {
1904       return 1;
1905    }
1906    if (!ua->catalog) {
1907       LockRes();
1908       ua->catalog = (CAT *)GetNextRes(R_CATALOG, NULL);
1909       UnlockRes();
1910       if (!ua->catalog) {
1911          bsendmsg(ua, _("Could not find a Catalog resource\n"));
1912          return 0;
1913       } else {
1914          bsendmsg(ua, _("Using default Catalog name=%s DB=%s\n"),
1915             ua->catalog->hdr.name, ua->catalog->db_name);
1916       }
1917    }
1918
1919    ua->jcr->catalog = ua->catalog;
1920
1921    Dmsg0(150, "Open database\n");
1922    ua->db = db_init_database(ua->jcr, ua->catalog->db_name, ua->catalog->db_user,
1923                              ua->catalog->db_password, ua->catalog->db_address,
1924                              ua->catalog->db_port, ua->catalog->db_socket);
1925    if (!ua->db || !db_open_database(ua->jcr, ua->db)) {
1926       bsendmsg(ua, _("Could not open database \"%s\".\n"),
1927                  ua->catalog->db_name);
1928       if (ua->db) {
1929          bsendmsg(ua, "%s", db_strerror(ua->db));
1930       }
1931       close_db(ua);
1932       return 0;
1933    }
1934    ua->jcr->db = ua->db;
1935    Dmsg1(150, "DB %s opened\n", ua->catalog->db_name);
1936    return 1;
1937 }
1938
1939 void close_db(UAContext *ua)
1940 {
1941    if (ua->db) {
1942       db_close_database(ua->jcr, ua->db);
1943       ua->db = NULL;
1944       if (ua->jcr) {
1945          ua->jcr->db = NULL;
1946       }
1947    }
1948 }