X-Git-Url: https://git.sur5r.net/?a=blobdiff_plain;f=bacula%2Fsrc%2Fdird%2Fua_select.c;h=aebdaa0b8be6ec3b4b0d66ce38bc8cff86f1635c;hb=6ecbf169a434395848de8f658ab5361093f50991;hp=e043fdfe3b39b597018f5929cb167b5490a5adb4;hpb=b88931200834b9ced689971786f3055f2f1a387c;p=bacula%2Fbacula diff --git a/bacula/src/dird/ua_select.c b/bacula/src/dird/ua_select.c index e043fdfe3b..aebdaa0b8b 100644 --- a/bacula/src/dird/ua_select.c +++ b/bacula/src/dird/ua_select.c @@ -1,3 +1,30 @@ +/* + Bacula® - The Network Backup Solution + + Copyright (C) 2001-2009 Free Software Foundation Europe e.V. + + The main author of Bacula is Kern Sibbald, with contributions from + many others, a complete list can be found in the file AUTHORS. + This program is Free Software; you can redistribute it and/or + modify it under the terms of version two of the GNU General Public + License as published by the Free Software Foundation and included + in the file LICENSE. + + This program is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + 02110-1301, USA. + + Bacula® is a registered trademark of Kern Sibbald. + The licensor of Bacula is the Free Software Foundation Europe + (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich, + Switzerland, email:ftf@fsfeurope.org. +*/ /* * * Bacula Director -- User Agent Prompt and Selection code @@ -6,20 +33,6 @@ * * Version $Id$ */ -/* - Copyright (C) 2001-2005 Kern Sibbald - - This program is free software; you can redistribute it and/or - modify it under the terms of the GNU General Public License - version 2 as amended with additional clauses defined in the - file LICENSE in the main source directory. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - the file LICENSE for additional details. - - */ #include "bacula.h" #include "dird.h" @@ -34,10 +47,16 @@ extern struct s_jl joblevels[]; int confirm_retention(UAContext *ua, utime_t *ret, const char *msg) { char ed1[100]; + int val; + + int yes_in_arg = find_arg(ua, NT_("yes")); for ( ;; ) { - bsendmsg(ua, _("The current %s retention period is: %s\n"), + ua->info_msg(_("The current %s retention period is: %s\n"), msg, edit_utime(*ret, ed1, sizeof(ed1))); + if (yes_in_arg != -1) { + return 1; + } if (!get_cmd(ua, _("Continue? (yes/mod/no): "))) { return 0; } @@ -46,16 +65,13 @@ int confirm_retention(UAContext *ua, utime_t *ret, const char *msg) return 0; } if (!duration_to_utime(ua->cmd, ret)) { - bsendmsg(ua, _("Invalid period.\n")); + ua->error_msg(_("Invalid period.\n")); continue; } continue; } - if (strcasecmp(ua->cmd, _("yes")) == 0) { - return 1; - } - if (strcasecmp(ua->cmd, _("no")) == 0) { - return 0; + if (is_yesno(ua->cmd, &val)) { + return val; /* is 1 for yes, 0 for no */ } } return 1; @@ -71,7 +87,7 @@ int find_arg_keyword(UAContext *ua, const char **list) { for (int i=1; iargc; i++) { for(int j=0; list[j]; j++) { - if (strcasecmp(_(list[j]), ua->argk[i]) == 0) { + if (strcasecmp(list[j], ua->argk[i]) == 0) { return j; } } @@ -144,12 +160,14 @@ STORE *select_storage_resource(UAContext *ua) start_prompt(ua, _("The defined Storage resources are:\n")); LockRes(); foreach_res(store, R_STORAGE) { - if (acl_access_ok(ua, Storage_ACL, store->hdr.name)) { - add_prompt(ua, store->hdr.name); + if (acl_access_ok(ua, Storage_ACL, store->name())) { + add_prompt(ua, store->name()); } } UnlockRes(); - do_prompt(ua, _("Storage"), _("Select Storage resource"), name, sizeof(name)); + if (do_prompt(ua, _("Storage"), _("Select Storage resource"), name, sizeof(name)) < 0) { + return NULL; + } store = (STORE *)GetResWithName(R_STORAGE, name); return store; } @@ -165,12 +183,14 @@ FILESET *select_fileset_resource(UAContext *ua) start_prompt(ua, _("The defined FileSet resources are:\n")); LockRes(); foreach_res(fs, R_FILESET) { - if (acl_access_ok(ua, FileSet_ACL, fs->hdr.name)) { - add_prompt(ua, fs->hdr.name); + if (acl_access_ok(ua, FileSet_ACL, fs->name())) { + add_prompt(ua, fs->name()); } } UnlockRes(); - do_prompt(ua, _("FileSet"), _("Select FileSet resource"), name, sizeof(name)); + if (do_prompt(ua, _("FileSet"), _("Select FileSet resource"), name, sizeof(name)) < 0) { + return NULL; + } fs = (FILESET *)GetResWithName(R_FILESET, name); return fs; } @@ -186,29 +206,71 @@ CAT *get_catalog_resource(UAContext *ua) int i; for (i=1; iargc; i++) { - if (strcasecmp(ua->argk[i], _("catalog")) == 0 && ua->argv[i]) { + if (strcasecmp(ua->argk[i], NT_("catalog")) == 0 && ua->argv[i]) { if (acl_access_ok(ua, Catalog_ACL, ua->argv[i])) { catalog = (CAT *)GetResWithName(R_CATALOG, ua->argv[i]); break; } } } + if (ua->gui && !catalog) { + LockRes(); + catalog = (CAT *)GetNextRes(R_CATALOG, NULL); + UnlockRes(); + if (!catalog) { + ua->error_msg(_("Could not find a Catalog resource\n")); + return NULL; + } else if (!acl_access_ok(ua, Catalog_ACL, catalog->name())) { + ua->error_msg(_("You must specify a \"use \" command before continuing.\n")); + return NULL; + } + return catalog; + } if (!catalog) { start_prompt(ua, _("The defined Catalog resources are:\n")); LockRes(); foreach_res(catalog, R_CATALOG) { - if (acl_access_ok(ua, Catalog_ACL, catalog->hdr.name)) { - add_prompt(ua, catalog->hdr.name); + if (acl_access_ok(ua, Catalog_ACL, catalog->name())) { + add_prompt(ua, catalog->name()); } } UnlockRes(); - do_prompt(ua, _("Catalog"), _("Select Catalog resource"), name, sizeof(name)); + if (do_prompt(ua, _("Catalog"), _("Select Catalog resource"), name, sizeof(name)) < 0) { + return NULL; + } catalog = (CAT *)GetResWithName(R_CATALOG, name); } return catalog; } +/* + * Select a job to enable or disable + */ +JOB *select_enable_disable_job_resource(UAContext *ua, bool enable) +{ + char name[MAX_NAME_LENGTH]; + JOB *job; + + LockRes(); + start_prompt(ua, _("The defined Job resources are:\n")); + foreach_res(job, R_JOB) { + if (!acl_access_ok(ua, Job_ACL, job->name())) { + continue; + } + if (job->enabled == enable) { /* Already enabled/disabled? */ + continue; /* yes, skip */ + } + add_prompt(ua, job->name()); + } + UnlockRes(); + if (do_prompt(ua, _("Job"), _("Select Job resource"), name, sizeof(name)) < 0) { + return NULL; + } + job = (JOB *)GetResWithName(R_JOB, name); + return job; +} + /* * Select a Job resource from prompt list */ @@ -220,12 +282,14 @@ JOB *select_job_resource(UAContext *ua) start_prompt(ua, _("The defined Job resources are:\n")); LockRes(); foreach_res(job, R_JOB) { - if (acl_access_ok(ua, Job_ACL, job->hdr.name)) { - add_prompt(ua, job->hdr.name); + if (acl_access_ok(ua, Job_ACL, job->name())) { + add_prompt(ua, job->name()); } } UnlockRes(); - do_prompt(ua, _("Job"), _("Select Job resource"), name, sizeof(name)); + if (do_prompt(ua, _("Job"), _("Select Job resource"), name, sizeof(name)) < 0) { + return NULL; + } job = (JOB *)GetResWithName(R_JOB, name); return job; } @@ -241,12 +305,14 @@ JOB *select_restore_job_resource(UAContext *ua) start_prompt(ua, _("The defined Restore Job resources are:\n")); LockRes(); foreach_res(job, R_JOB) { - if (job->JobType == JT_RESTORE && acl_access_ok(ua, Job_ACL, job->hdr.name)) { - add_prompt(ua, job->hdr.name); + if (job->JobType == JT_RESTORE && acl_access_ok(ua, Job_ACL, job->name())) { + add_prompt(ua, job->name()); } } UnlockRes(); - do_prompt(ua, _("Job"), _("Select Restore Job"), name, sizeof(name)); + if (do_prompt(ua, _("Job"), _("Select Restore Job"), name, sizeof(name)) < 0) { + return NULL; + } job = (JOB *)GetResWithName(R_JOB, name); return job; } @@ -264,12 +330,14 @@ CLIENT *select_client_resource(UAContext *ua) start_prompt(ua, _("The defined Client resources are:\n")); LockRes(); foreach_res(client, R_CLIENT) { - if (acl_access_ok(ua, Client_ACL, client->hdr.name)) { - add_prompt(ua, client->hdr.name); + if (acl_access_ok(ua, Client_ACL, client->name())) { + add_prompt(ua, client->name()); } } UnlockRes(); - do_prompt(ua, _("Client"), _("Select Client (File daemon) resource"), name, sizeof(name)); + if (do_prompt(ua, _("Client"), _("Select Client (File daemon) resource"), name, sizeof(name)) < 0) { + return NULL; + } client = (CLIENT *)GetResWithName(R_CLIENT, name); return client; } @@ -285,8 +353,8 @@ CLIENT *get_client_resource(UAContext *ua) int i; for (i=1; iargc; i++) { - if ((strcasecmp(ua->argk[i], N_("client")) == 0 || - strcasecmp(ua->argk[i], N_("fd")) == 0) && ua->argv[i]) { + if ((strcasecmp(ua->argk[i], NT_("client")) == 0 || + strcasecmp(ua->argk[i], NT_("fd")) == 0) && ua->argv[i]) { if (!acl_access_ok(ua, Client_ACL, ua->argv[i])) { break; } @@ -294,7 +362,7 @@ CLIENT *get_client_resource(UAContext *ua) if (client) { return client; } - bsendmsg(ua, _("Error: Client resource %s does not exist.\n"), ua->argv[i]); + ua->error_msg(_("Error: Client resource %s does not exist.\n"), ua->argv[i]); break; } } @@ -311,7 +379,7 @@ CLIENT *get_client_resource(UAContext *ua) * returns: 0 on error * 1 on success and fills in CLIENT_DBR */ -int get_client_dbr(UAContext *ua, CLIENT_DBR *cr) +bool get_client_dbr(UAContext *ua, CLIENT_DBR *cr) { int i; @@ -319,17 +387,17 @@ int get_client_dbr(UAContext *ua, CLIENT_DBR *cr) if (db_get_client_record(ua->jcr, ua->db, cr)) { return 1; } - bsendmsg(ua, _("Could not find Client %s: ERR=%s"), cr->Name, db_strerror(ua->db)); + ua->error_msg(_("Could not find Client %s: ERR=%s"), cr->Name, db_strerror(ua->db)); } for (i=1; iargc; i++) { - if ((strcasecmp(ua->argk[i], _("client")) == 0 || - strcasecmp(ua->argk[i], _("fd")) == 0) && ua->argv[i]) { + if ((strcasecmp(ua->argk[i], NT_("client")) == 0 || + strcasecmp(ua->argk[i], NT_("fd")) == 0) && ua->argv[i]) { if (!acl_access_ok(ua, Client_ACL, ua->argv[i])) { break; } bstrncpy(cr->Name, ua->argv[i], sizeof(cr->Name)); if (!db_get_client_record(ua->jcr, ua->db, cr)) { - bsendmsg(ua, _("Could not find Client \"%s\": ERR=%s"), ua->argv[i], + ua->error_msg(_("Could not find Client \"%s\": ERR=%s"), ua->argv[i], db_strerror(ua->db)); cr->ClientId = 0; break; @@ -348,7 +416,7 @@ int get_client_dbr(UAContext *ua, CLIENT_DBR *cr) * Returns 1 on success * 0 on failure */ -int select_client_dbr(UAContext *ua, CLIENT_DBR *cr) +bool select_client_dbr(UAContext *ua, CLIENT_DBR *cr) { CLIENT_DBR ocr; char name[MAX_NAME_LENGTH]; @@ -358,11 +426,11 @@ int select_client_dbr(UAContext *ua, CLIENT_DBR *cr) cr->ClientId = 0; if (!db_get_client_ids(ua->jcr, ua->db, &num_clients, &ids)) { - bsendmsg(ua, _("Error obtaining client ids. ERR=%s\n"), db_strerror(ua->db)); + ua->error_msg(_("Error obtaining client ids. ERR=%s\n"), db_strerror(ua->db)); return 0; } if (num_clients <= 0) { - bsendmsg(ua, _("No clients defined. You must run a job before using this command.\n")); + ua->error_msg(_("No clients defined. You must run a job before using this command.\n")); return 0; } @@ -383,18 +451,18 @@ int select_client_dbr(UAContext *ua, CLIENT_DBR *cr) bstrncpy(ocr.Name, name, sizeof(ocr.Name)); if (!db_get_client_record(ua->jcr, ua->db, &ocr)) { - bsendmsg(ua, _("Could not find Client \"%s\": ERR=%s"), name, db_strerror(ua->db)); + ua->error_msg(_("Could not find Client \"%s\": ERR=%s"), name, db_strerror(ua->db)); return 0; } memcpy(cr, &ocr, sizeof(ocr)); return 1; } - - /* Scan what the user has entered looking for: * - * pool= + * argk= + * + * where argk can be : pool, recyclepool, scratchpool, nextpool etc.. * * if error or not found, put up a list of pool DBRs * to choose from. @@ -402,25 +470,26 @@ int select_client_dbr(UAContext *ua, CLIENT_DBR *cr) * returns: false on error * true on success and fills in POOL_DBR */ -bool get_pool_dbr(UAContext *ua, POOL_DBR *pr) +bool get_pool_dbr(UAContext *ua, POOL_DBR *pr, const char *argk) { if (pr->Name[0]) { /* If name already supplied */ if (db_get_pool_record(ua->jcr, ua->db, pr) && acl_access_ok(ua, Pool_ACL, pr->Name)) { return true; } - bsendmsg(ua, _("Could not find Pool \"%s\": ERR=%s"), pr->Name, db_strerror(ua->db)); + ua->error_msg(_("Could not find Pool \"%s\": ERR=%s"), pr->Name, db_strerror(ua->db)); } - if (!select_pool_dbr(ua, pr)) { /* try once more */ + if (!select_pool_dbr(ua, pr, argk)) { /* try once more */ return false; } return true; } /* - * Select a Pool record from the catalog + * Select a Pool record from catalog + * argk can be pool, recyclepool, scratchpool etc.. */ -bool select_pool_dbr(UAContext *ua, POOL_DBR *pr) +bool select_pool_dbr(UAContext *ua, POOL_DBR *pr, const char *argk) { POOL_DBR opr; char name[MAX_NAME_LENGTH]; @@ -428,11 +497,11 @@ bool select_pool_dbr(UAContext *ua, POOL_DBR *pr) uint32_t *ids; for (i=1; iargc; i++) { - if (strcasecmp(ua->argk[i], N_("pool")) == 0 && ua->argv[i] && + if (strcasecmp(ua->argk[i], argk) == 0 && ua->argv[i] && acl_access_ok(ua, Pool_ACL, ua->argv[i])) { bstrncpy(pr->Name, ua->argv[i], sizeof(pr->Name)); if (!db_get_pool_record(ua->jcr, ua->db, pr)) { - bsendmsg(ua, _("Could not find Pool \"%s\": ERR=%s"), ua->argv[i], + ua->error_msg(_("Could not find Pool \"%s\": ERR=%s"), ua->argv[i], db_strerror(ua->db)); pr->PoolId = 0; break; @@ -443,15 +512,18 @@ bool select_pool_dbr(UAContext *ua, POOL_DBR *pr) pr->PoolId = 0; if (!db_get_pool_ids(ua->jcr, ua->db, &num_pools, &ids)) { - bsendmsg(ua, _("Error obtaining pool ids. ERR=%s\n"), db_strerror(ua->db)); + ua->error_msg(_("Error obtaining pool ids. ERR=%s\n"), db_strerror(ua->db)); return 0; } if (num_pools <= 0) { - bsendmsg(ua, _("No pools defined. Use the \"create\" command to create one.\n")); + ua->error_msg(_("No pools defined. Use the \"create\" command to create one.\n")); return false; } start_prompt(ua, _("Defined Pools:\n")); + if (bstrcmp(argk, NT_("recyclepool"))) { + add_prompt(ua, _("*None*")); + } for (i=0; i < num_pools; i++) { opr.PoolId = ids[i]; if (!db_get_pool_record(ua->jcr, ua->db, &opr) || @@ -464,13 +536,21 @@ bool select_pool_dbr(UAContext *ua, POOL_DBR *pr) if (do_prompt(ua, _("Pool"), _("Select the Pool"), name, sizeof(name)) < 0) { return false; } - memset(&opr, 0, sizeof(opr)); - bstrncpy(opr.Name, name, sizeof(opr.Name)); - if (!db_get_pool_record(ua->jcr, ua->db, &opr)) { - bsendmsg(ua, _("Could not find Pool \"%s\": ERR=%s"), name, db_strerror(ua->db)); - return false; + memset(&opr, 0, sizeof(opr)); + /* *None* is only returned when selecting a recyclepool, and in that case + * the calling code is only interested in opr.Name, so then we can leave + * pr as all zero. + */ + if (!bstrcmp(name, _("*None*"))) { + bstrncpy(opr.Name, name, sizeof(opr.Name)); + + if (!db_get_pool_record(ua->jcr, ua->db, &opr)) { + ua->error_msg(_("Could not find Pool \"%s\": ERR=%s"), name, db_strerror(ua->db)); + return false; + } } + memcpy(pr, &opr, sizeof(opr)); return true; } @@ -487,11 +567,11 @@ int select_pool_and_media_dbr(UAContext *ua, POOL_DBR *pr, MEDIA_DBR *mr) memset(pr, 0, sizeof(POOL_DBR)); pr->PoolId = mr->PoolId; if (!db_get_pool_record(ua->jcr, ua->db, pr)) { - bsendmsg(ua, "%s", db_strerror(ua->db)); + ua->error_msg("%s", db_strerror(ua->db)); return 0; } if (!acl_access_ok(ua, Pool_ACL, pr->Name)) { - bsendmsg(ua, _("No access to Pool \"%s\"\n"), pr->Name); + ua->error_msg(_("No access to Pool \"%s\"\n"), pr->Name); return 0; } return 1; @@ -517,18 +597,18 @@ int select_media_dbr(UAContext *ua, MEDIA_DBR *mr) } mr->PoolId = pr.PoolId; db_list_media_records(ua->jcr, ua->db, mr, prtit, ua, HORZ_LIST); - if (!get_cmd(ua, _("Enter MediaId or Volume name: "))) { + if (!get_cmd(ua, _("Enter *MediaId or Volume name: "))) { return 0; } - if (is_a_number(ua->cmd)) { - mr->MediaId = str_to_int64(ua->cmd); + if (ua->cmd[0] == '*' && is_a_number(ua->cmd+1)) { + mr->MediaId = str_to_int64(ua->cmd+1); } else { bstrncpy(mr->VolumeName, ua->cmd, sizeof(mr->VolumeName)); } } if (!db_get_media_record(ua->jcr, ua->db, mr)) { - bsendmsg(ua, "%s", db_strerror(ua->db)); + ua->error_msg("%s", db_strerror(ua->db)); return 0; } return 1; @@ -546,12 +626,14 @@ POOL *select_pool_resource(UAContext *ua) start_prompt(ua, _("The defined Pool resources are:\n")); LockRes(); foreach_res(pool, R_POOL) { - if (acl_access_ok(ua, Pool_ACL, pool->hdr.name)) { - add_prompt(ua, pool->hdr.name); + if (acl_access_ok(ua, Pool_ACL, pool->name())) { + add_prompt(ua, pool->name()); } } UnlockRes(); - do_prompt(ua, _("Pool"), _("Select Pool resource"), name, sizeof(name)); + if (do_prompt(ua, _("Pool"), _("Select Pool resource"), name, sizeof(name)) < 0) { + return NULL; + } pool = (POOL *)GetResWithName(R_POOL, name); return pool; } @@ -573,7 +655,7 @@ POOL *get_pool_resource(UAContext *ua) if (pool) { return pool; } - bsendmsg(ua, _("Error: Pool resource \"%s\" does not exist.\n"), ua->argv[i]); + ua->error_msg(_("Error: Pool resource \"%s\" does not exist.\n"), ua->argv[i]); } return select_pool_resource(ua); } @@ -589,7 +671,7 @@ int select_job_dbr(UAContext *ua, JOB_DBR *jr) } jr->JobId = ua->int64_val; if (!db_get_job_record(ua->jcr, ua->db, jr)) { - bsendmsg(ua, "%s", db_strerror(ua->db)); + ua->error_msg("%s", db_strerror(ua->db)); return 0; } return jr->JobId; @@ -612,16 +694,17 @@ int get_job_dbr(UAContext *ua, JOB_DBR *jr) int i; for (i=1; iargc; i++) { - if (strcasecmp(ua->argk[i], N_("job")) == 0 && ua->argv[i]) { + if (strcasecmp(ua->argk[i], NT_("ujobid")) == 0 && ua->argv[i]) { jr->JobId = 0; bstrncpy(jr->Job, ua->argv[i], sizeof(jr->Job)); - } else if (strcasecmp(ua->argk[i], N_("jobid")) == 0 && ua->argv[i]) { + } else if (strcasecmp(ua->argk[i], NT_("jobid")) == 0 && ua->argv[i]) { jr->JobId = str_to_int64(ua->argv[i]); + jr->Job[0] = 0; } else { continue; } if (!db_get_job_record(ua->jcr, ua->db, jr)) { - bsendmsg(ua, _("Could not find Job \"%s\": ERR=%s"), ua->argv[i], + ua->error_msg(_("Could not find Job \"%s\": ERR=%s"), ua->argv[i], db_strerror(ua->db)); jr->JobId = 0; break; @@ -629,6 +712,17 @@ int get_job_dbr(UAContext *ua, JOB_DBR *jr) return jr->JobId; } + jr->JobId = 0; + jr->Job[0] = 0; + + for (i=1; iargc; i++) { + if ((strcasecmp(ua->argk[i], NT_("jobname")) == 0 || + strcasecmp(ua->argk[i], NT_("job")) == 0) && ua->argv[i]) { + jr->JobId = 0; + bstrncpy(jr->Name, ua->argv[i], sizeof(jr->Name)); + break; + } + } if (!select_job_dbr(ua, jr)) { /* try once more */ return 0; } @@ -673,45 +767,59 @@ void add_prompt(UAContext *ua, const char *prompt) * Returns: -1 on error * index base 0 on success, and choice * is copied to prompt if not NULL + * prompt is set to the chosen prompt item string */ -int do_prompt(UAContext *ua, const char *automsg, const char *msg, char *prompt, int max_prompt) +int do_prompt(UAContext *ua, const char *automsg, const char *msg, + char *prompt, int max_prompt) { int i, item; char pmsg[MAXSTRING]; + BSOCK *user = ua->UA_sock; + if (prompt) { + *prompt = 0; + } if (ua->num_prompts == 2) { item = 1; if (prompt) { bstrncpy(prompt, ua->prompt[1], max_prompt); } - bsendmsg(ua, _("Automatically selected %s: %s\n"), automsg, ua->prompt[1]); + ua->send_msg(_("Automatically selected %s: %s\n"), automsg, ua->prompt[1]); goto done; } /* If running non-interactive, bail out */ if (ua->batch) { - bsendmsg(ua, _("Cannot select %s in batch mode.\n"), automsg); + /* First print the choices he wanted to make */ + ua->send_msg(ua->prompt[0]); + for (i=1; i < ua->num_prompts; i++) { + ua->send_msg("%6d: %s\n", i, ua->prompt[i]); + } + /* Now print error message */ + ua->send_msg(_("Your request has multiple choices for \"%s\". Selection is not possible in batch mode.\n"), automsg); item = -1; goto done; } - bsendmsg(ua, ua->prompt[0]); + if (ua->api) user->signal(BNET_START_SELECT); + ua->send_msg(ua->prompt[0]); for (i=1; i < ua->num_prompts; i++) { - bsendmsg(ua, "%6d: %s\n", i, ua->prompt[i]); - } - - if (prompt) { - *prompt = 0; + if (ua->api) { + ua->send_msg("%s", ua->prompt[i]); + } else { + ua->send_msg("%6d: %s\n", i, ua->prompt[i]); + } } + if (ua->api) user->signal(BNET_END_SELECT); for ( ;; ) { /* First item is the prompt string, not the items */ if (ua->num_prompts == 1) { - bsendmsg(ua, _("Selection is empty!\n")); - item = 0; /* list is empty ! */ + ua->error_msg(_("Selection list for \"%s\" is empty!\n"), automsg); + item = -1; /* list is empty ! */ break; } if (ua->num_prompts == 2) { item = 1; - bsendmsg(ua, _("Item 1 selected automatically.\n")); + ua->send_msg(_("Automatically selected: %s\n"), ua->prompt[1]); if (prompt) { bstrncpy(prompt, ua->prompt[1], max_prompt); } @@ -720,14 +828,15 @@ int do_prompt(UAContext *ua, const char *automsg, const char *msg, char *prompt, sprintf(pmsg, "%s (1-%d): ", msg, ua->num_prompts-1); } /* Either a . or an @ will get you out of the loop */ + if (ua->api) user->signal(BNET_SELECT_INPUT); if (!get_pint(ua, pmsg)) { item = -1; /* error */ - bsendmsg(ua, _("Selection aborted, nothing done.\n")); + ua->info_msg(_("Selection aborted, nothing done.\n")); break; } item = ua->pint32_val; if (item < 1 || item >= ua->num_prompts) { - bsendmsg(ua, _("Please enter a number between 1 and %d\n"), ua->num_prompts-1); + ua->warning_msg(_("Please enter a number between 1 and %d\n"), ua->num_prompts-1); continue; } if (prompt) { @@ -741,7 +850,7 @@ done: free(ua->prompt[i]); } ua->num_prompts = 0; - return item - 1; + return item>0 ? item-1 : item; } @@ -762,7 +871,7 @@ STORE *get_storage_resource(UAContext *ua, bool use_default) STORE *store = NULL; int jobid; JCR *jcr; - int i, drive; + int i; char ed1[50]; for (i=1; iargc; i++) { @@ -770,12 +879,13 @@ STORE *get_storage_resource(UAContext *ua, bool use_default) /* Ignore slots, scan and barcode(s) keywords */ if (strcasecmp("scan", ua->argk[i]) == 0 || strcasecmp("barcode", ua->argk[i]) == 0 || + strcasecmp("barcodes", ua->argk[i]) == 0 || strcasecmp("slots", ua->argk[i]) == 0) { continue; } /* Default argument is storage */ if (store_name) { - bsendmsg(ua, _("Storage name given twice.\n")); + ua->error_msg(_("Storage name given twice.\n")); return NULL; } store_name = ua->argk[i]; @@ -784,77 +894,124 @@ STORE *get_storage_resource(UAContext *ua, bool use_default) break; } } else { - if (strcasecmp(ua->argk[i], N_("storage")) == 0 || - strcasecmp(ua->argk[i], N_("sd")) == 0) { + if (strcasecmp(ua->argk[i], NT_("storage")) == 0 || + strcasecmp(ua->argk[i], NT_("sd")) == 0) { store_name = ua->argv[i]; break; - } else if (strcasecmp(ua->argk[i], N_("jobid")) == 0) { + } else if (strcasecmp(ua->argk[i], NT_("jobid")) == 0) { jobid = str_to_int64(ua->argv[i]); if (jobid <= 0) { - bsendmsg(ua, _("Expecting jobid=nn command, got: %s\n"), ua->argk[i]); + ua->error_msg(_("Expecting jobid=nn command, got: %s\n"), ua->argk[i]); return NULL; } if (!(jcr=get_jcr_by_id(jobid))) { - bsendmsg(ua, _("JobId %s is not running.\n"), edit_int64(jobid, ed1)); + ua->error_msg(_("JobId %s is not running.\n"), edit_int64(jobid, ed1)); return NULL; } - store = jcr->store; + store = jcr->wstore; free_jcr(jcr); break; - } else if (strcasecmp(ua->argk[i], N_("job")) == 0) { + } else if (strcasecmp(ua->argk[i], NT_("job")) == 0 || + strcasecmp(ua->argk[i], NT_("jobname")) == 0) { if (!ua->argv[i]) { - bsendmsg(ua, _("Expecting job=xxx, got: %s.\n"), ua->argk[i]); + ua->error_msg(_("Expecting job=xxx, got: %s.\n"), ua->argk[i]); return NULL; } if (!(jcr=get_jcr_by_partial_name(ua->argv[i]))) { - bsendmsg(ua, _("Job \"%s\" is not running.\n"), ua->argv[i]); + ua->error_msg(_("Job \"%s\" is not running.\n"), ua->argv[i]); return NULL; } - store = jcr->store; + store = jcr->wstore; + free_jcr(jcr); + break; + } else if (strcasecmp(ua->argk[i], NT_("ujobid")) == 0) { + if (!ua->argv[i]) { + ua->error_msg(_("Expecting ujobid=xxx, got: %s.\n"), ua->argk[i]); + return NULL; + } + if (!(jcr=get_jcr_by_full_name(ua->argv[i]))) { + ua->error_msg(_("Job \"%s\" is not running.\n"), ua->argv[i]); + return NULL; + } + store = jcr->wstore; free_jcr(jcr); break; } } } - if (store && !acl_access_ok(ua, Storage_ACL, store->hdr.name)) { + if (store && !acl_access_ok(ua, Storage_ACL, store->name())) { store = NULL; } - if (!store && store_name) { + if (!store && store_name && store_name[0] != 0) { store = (STORE *)GetResWithName(R_STORAGE, store_name); if (!store) { - bsendmsg(ua, _("Storage resource \"%s\": not found\n"), store_name); + ua->error_msg(_("Storage resource \"%s\": not found\n"), store_name); } } - if (store && !acl_access_ok(ua, Storage_ACL, store->hdr.name)) { + if (store && !acl_access_ok(ua, Storage_ACL, store->name())) { store = NULL; } /* No keywords found, so present a selection list */ if (!store) { store = select_storage_resource(ua); } + return store; +} + +/* Get drive that we are working with for this storage */ +int get_storage_drive(UAContext *ua, STORE *store) +{ + int i, drive = -1; /* Get drive for autochanger if possible */ - drive = -2; /* dummy */ - if (store && store->autochanger) { - i = find_arg_with_value(ua, "drive"); - if (i >=0) { - drive = atoi(ua->argv[i]); + i = find_arg_with_value(ua, "drive"); + if (i >=0) { + drive = atoi(ua->argv[i]); + } else if (store && store->autochanger) { + /* If our structure is not set ask SD for # drives */ + if (store->drives == 0) { + store->drives = get_num_drives_from_SD(ua); + } + /* If only one drive, default = 0 */ + if (store->drives == 1) { + drive = 0; } else { + /* Ask user to enter drive number */ ua->cmd[0] = 0; if (!get_cmd(ua, _("Enter autochanger drive[0]: "))) { drive = -1; /* None */ } else { drive = atoi(ua->cmd); } + } + } + return drive; +} + +/* Get slot that we are working with for this storage */ +int get_storage_slot(UAContext *ua, STORE *store) +{ + int i, slot = -1; + /* Get slot for autochanger if possible */ + i = find_arg_with_value(ua, "slot"); + if (i >=0) { + slot = atoi(ua->argv[i]); + } else if (store && store->autochanger) { + /* Ask user to enter slot number */ + ua->cmd[0] = 0; + if (!get_cmd(ua, _("Enter autochanger slot: "))) { + slot = -1; /* None */ + } else { + slot = atoi(ua->cmd); } } - ua->int32_val = drive; - return store; + return slot; } + /* * Scan looking for mediatype= * @@ -889,7 +1046,7 @@ bool get_level_from_name(JCR *jcr, const char *level_name) bool found = false; for (int i=0; joblevels[i].level_name; i++) { if (strcasecmp(level_name, joblevels[i].level_name) == 0) { - jcr->JobLevel = joblevels[i].level; + jcr->set_JobLevel(joblevels[i].level); found = true; break; }