*/
/*
- Copyright (C) 2000-2003 Kern Sibbald and John Walker
+ Copyright (C) 2000-2004 Kern Sibbald and John Walker
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License as
/* Exported functions */
-int get_cmd(UAContext *ua, char *prompt)
+int get_cmd(UAContext *ua, const char *prompt)
{
BSOCK *sock = ua->UA_sock;
int stat;
if (is_bnet_stop(sock)) {
return 0; /* error or terminate */
}
- ua->cmd = check_pool_memory_size(ua->cmd, sock->msglen+1);
- bstrncpy(ua->cmd, sock->msg, sock->msglen+1);
+ pm_strcpy(ua->cmd, sock->msg);
strip_trailing_junk(ua->cmd);
if (strcmp(ua->cmd, ".messages") == 0) {
qmessagescmd(ua, ua->cmd);
}
- /* ****FIXME**** if .command, go off and do it. For now ignore it. */
- if (ua->cmd[0] == '.' && ua->cmd[1] != 0) {
- continue; /* dot command */
+ /* Lone dot => break */
+ if (ua->cmd[0] == '.' && ua->cmd[1] == 0) {
+ return 0;
}
- /* Lone dot => break or actual response */
break;
}
return 1;
}
-/*
- * Return next argument from command line. Note, this
- * routine is destructive.
+/*
+ * Get a positive integer
+ * Returns: false if failure
+ * true if success => value in ua->pint32_val
*/
-char *next_arg(char **s)
+bool get_pint(UAContext *ua, const char *prompt)
{
- char *p, *q, *n;
- int in_quote = 0;
-
- /* skip past spaces to next arg */
- for (p=*s; *p && *p == ' '; ) {
- p++;
- }
- Dmsg1(400, "Next arg=%s\n", p);
- for (n = q = p; *p ; ) {
- if (*p == '\\') {
- p++;
- if (*p) {
- *q++ = *p++;
- } else {
- *q++ = *p;
- }
- continue;
+ double dval;
+ ua->pint32_val = 0;
+ ua->int64_val = 0;
+ for (;;) {
+ ua->cmd[0] = 0;
+ if (!get_cmd(ua, prompt)) {
+ return false;
+ }
+ /* Kludge for slots blank line => 0 */
+ if (ua->cmd[0] == 0 && strncmp(prompt, "Enter slot", 10) == 0) {
+ return true;
}
- if (*p == '"') { /* start or end of quote */
- if (in_quote) {
- p++; /* skip quote */
- in_quote = 0;
- continue;
- }
- in_quote = 1;
- p++;
+ if (!is_a_number(ua->cmd)) {
+ bsendmsg(ua, "Expected a positive integer, got: %s\n", ua->cmd);
continue;
}
- if (!in_quote && *p == ' ') { /* end of field */
- p++;
- break;
+ errno = 0;
+ dval = strtod(ua->cmd, NULL);
+ if (errno != 0 || dval < 0) {
+ bsendmsg(ua, "Expected a positive integer, got: %s\n", ua->cmd);
+ continue;
}
- *q++ = *p++;
+ ua->pint32_val = (uint32_t)dval;
+ ua->int64_val = (int64_t)dval;
+ return true;
}
- *q = 0;
- *s = p;
- Dmsg2(400, "End arg=%s next=%s\n", n, p);
- return n;
-}
+}
/*
- * This routine parses the input command line.
- * It makes a copy in args, then builds an
- * argc, argv like list where
- *
- * argc = count of arguments
- * argk[i] = argument keyword (part preceding =)
- * argv[i] = argument value (part after =)
- *
- * example: arg1 arg2=abc arg3=
- *
- * argc = c
- * argk[0] = arg1
- * argv[0] = NULL
- * argk[1] = arg2
- * argv[1] = abc
- * argk[2] = arg3
- * argv[2] =
+ * Gets a yes or no response
+ * Returns: false if failure
+ * true if success => ua->pint32_val == 1 for yes
+ * ua->pint32_val == 0 for no
*/
-
-void parse_command_args(UAContext *ua)
+bool get_yesno(UAContext *ua, const char *prompt)
{
- char *p, *q, *n;
- int i, len;
-
- len = strlen(ua->cmd) + 1;
- ua->args = check_pool_memory_size(ua->args, len);
- bstrncpy(ua->args, ua->cmd, len);
- strip_trailing_junk(ua->args);
- ua->argc = 0;
- p = ua->args;
- /* Pick up all arguments */
- while (ua->argc < MAX_ARGS) {
- n = next_arg(&p);
- if (*n) {
- ua->argk[ua->argc++] = n;
- } else {
- break;
+ int len;
+
+ ua->pint32_val = 0;
+ for (;;) {
+ if (!get_cmd(ua, prompt)) {
+ return false;
}
- }
- /* Separate keyword and value */
- for (i=0; i<ua->argc; i++) {
- p = strchr(ua->argk[i], '=');
- if (p) {
- *p++ = 0; /* terminate keyword and point to value */
- /* Unquote quoted values */
- if (*p == '"') {
- for (n = q = ++p; *p && *p != '"'; ) {
- if (*p == '\\') {
- p++;
- }
- *q++ = *p++;
- }
- *q = 0; /* terminate string */
- p = n; /* point to string */
- }
- if (strlen(p) > MAX_NAME_LENGTH-1) {
- p[MAX_NAME_LENGTH-1] = 0; /* truncate to max len */
- }
+ len = strlen(ua->cmd);
+ if (len < 1 || len > 3) {
+ continue;
}
- ua->argv[i] = p; /* save ptr to value or NULL */
- }
-#ifdef xxxx
- for (i=0; i<ua->argc; i++) {
- Dmsg3(000, "Arg %d: kw=%s val=%s\n", i,
- ua->argk[i], ua->argv[i]?ua->argv[i]:"NULL");
+ if (strncasecmp(ua->cmd, _("yes"), len) == 0) {
+ ua->pint32_val = 1;
+ return true;
+ }
+ if (strncasecmp(ua->cmd, _("no"), len) == 0) {
+ return true;
+ }
+ bsendmsg(ua, _("Invalid response. You must answer yes or no.\n"));
}
-#endif
+}
+
+
+void parse_ua_args(UAContext *ua)
+{
+ parse_args(ua->cmd, &ua->args, &ua->argc, ua->argk, ua->argv, MAX_CMD_ARGS);
}