#include "bacula.h"
#include "console_conf.h"
#include "jcr.h"
-
-/* Imported functions */
-int authenticate_director(JCR *jcr, DIRRES *director);
-
+#ifdef HAVE_CONIO
+#include "conio.h"
+#else
+#define con_init(x)
+#define con_term()
+#define con_set_zed_keys();
+#define trapctlc()
+#define clrbrk()
+#define usrbrk() 0
+#endif
+
/* Exported variables */
-
#ifdef HAVE_CYGWIN
int rl_catch_signals;
#else
extern int rl_catch_signals;
#endif
+/* Imported functions */
+int authenticate_director(JCR *jcr, DIRRES *director, CONRES *cons);
+
+
+
/* Forward referenced functions */
static void terminate_console(int sig);
int get_cmd(FILE *input, char *prompt, BSOCK *sock, int sec);
static int do_outputcmd(FILE *input, BSOCK *UA_sock);
-static void sendit(char *fmt, ...);
+void senditf(char *fmt, ...);
+void sendit(char *buf);
/* Static variables */
static char *configfile = NULL;
static BSOCK *UA_sock = NULL;
static DIRRES *dir;
static FILE *output = stdout;
-int tee = 0; /* output to output and stdout */
-static int stop = FALSE;
+static bool tee = false; /* output to output and stdout */
+static bool stop = false;
static int argc;
static POOLMEM *args;
static char *argk[MAX_CMD_ARGS];
static char *argv[MAX_CMD_ARGS];
+
/* Command prototypes */
static int versioncmd(FILE *input, BSOCK *UA_sock);
static int inputcmd(FILE *input, BSOCK *UA_sock);
static int sleepcmd(FILE *input, BSOCK *UA_sock);
-#define CONFIG_FILE "./console.conf" /* default configuration file */
+#define CONFIG_FILE "./bconsole.conf" /* default configuration file */
static void usage()
{
fprintf(stderr, _(
"\nVersion: " VERSION " (" BDATE ") %s %s %s\n\n"
-"Usage: console [-s] [-c config_file] [-d debug_level] [config_file]\n"
+"Usage: bconsole [-s] [-c config_file] [-d debug_level] [config_file]\n"
" -c <file> set configuration file to file\n"
" -dnn set debug level to nn\n"
" -s no signals\n"
" -t test - read configuration and exit\n"
" -? print this message.\n"
"\n"), HOST_OS, DISTNAME, DISTVER);
-
- exit(1);
}
-
-void got_stop(int sig)
+void got_sigstop(int sig)
{
- stop = TRUE;
+ stop = true;
}
-void got_continue(int sig)
+void got_sigcontinue(int sig)
{
- stop = FALSE;
+ stop = false;
}
-void got_tout(int sig)
+void got_sigtout(int sig)
{
// printf("Got tout\n");
}
-void got_tin(int sig)
+void got_sigtin(int sig)
{
// printf("Got tin\n");
}
+static int zed_keyscmd(FILE *input, BSOCK *UA_sock)
+{
+ con_set_zed_keys();
+ return 1;
+}
+
+/*
+ * These are the @command
+ */
struct cmdstruct { char *key; int (*func)(FILE *input, BSOCK *UA_sock); char *help; };
static struct cmdstruct commands[] = {
{ N_("input"), inputcmd, _("input from file")},
{ N_("time"), timecmd, _("print current time")},
{ N_("version"), versioncmd, _("print Console's version")},
{ N_("exit"), quitcmd, _("exit = quit")},
+ { N_("zed_keyst"), zed_keyscmd, _("zed_keys = use zed keys instead of bash keys")},
};
#define comsize (sizeof(commands)/sizeof(struct cmdstruct))
if (!found) {
pm_strcat(&UA_sock->msg, _(": is an illegal command\n"));
UA_sock->msglen = strlen(UA_sock->msg);
- fputs(UA_sock->msg, output);
- fflush(output);
+ sendit(UA_sock->msg);
}
return stat;
}
static void read_and_process_input(FILE *input, BSOCK *UA_sock)
{
char *prompt = "*";
- int at_prompt = FALSE;
+ bool at_prompt = false;
int tty_input = isatty(fileno(input));
int stat;
prompt = "";
} else {
prompt = "*";
- at_prompt = TRUE;
+ at_prompt = true;
}
if (tty_input) {
stat = get_cmd(input, prompt, UA_sock, 30);
+ if (usrbrk() == 1) {
+ clrbrk();
+ }
+ if (usrbrk()) {
+ break;
+ }
} else {
+ /* Reading input from a file */
int len = sizeof_pool_memory(UA_sock->msg) - 1;
+ if (usrbrk()) {
+ break;
+ }
if (fgets(UA_sock->msg, len, input) == NULL) {
stat = -1;
} else {
- sendit("%s", UA_sock->msg); /* echo to terminal */
+ sendit(UA_sock->msg); /* echo to terminal */
strip_trailing_junk(UA_sock->msg);
UA_sock->msglen = strlen(UA_sock->msg);
stat = 1;
}
}
if (stat < 0) {
- break; /* error */
+ break; /* error or interrupt */
} else if (stat == 0) { /* timeout */
- bnet_fsend(UA_sock, ".messages");
+ if (strcmp(prompt, "*") == 0) {
+ bnet_fsend(UA_sock, ".messages");
+ } else {
+ continue;
+ }
} else {
at_prompt = FALSE;
/* @ => internal command for us */
if (!stop) {
sendit("\n");
}
- at_prompt = FALSE;
+ at_prompt = false;
}
- if (!stop) {
- sendit("%s", UA_sock->msg);
+ /* Suppress output if running in background or user hit ctl-c */
+ if (!stop && !usrbrk()) {
+ sendit(UA_sock->msg);
}
}
+ if (usrbrk() > 1) {
+ break;
+ } else {
+ clrbrk();
+ }
if (!stop) {
fflush(stdout);
}
break; /* error or term */
} else if (stat == BNET_SIGNAL) {
if (UA_sock->msglen == BNET_PROMPT) {
- at_prompt = TRUE;
+ at_prompt = true;
}
Dmsg1(100, "Got poll %s\n", bnet_sig_to_ascii(UA_sock));
}
int main(int argc, char *argv[])
{
int ch, i, ndir, item;
- int no_signals = FALSE;
- int test_config = FALSE;
+ bool no_signals = false;
+ bool test_config = false;
JCR jcr;
init_stack_dump();
- my_name_is(argc, argv, "console");
+ my_name_is(argc, argv, "bconsole");
textdomain("bacula-console");
init_msg(NULL, NULL);
working_directory = "/tmp";
args = get_pool_memory(PM_FNAME);
+ con_init(stdin);
while ((ch = getopt(argc, argv, "bc:d:r:st?")) != -1) {
switch (ch) {
- case 'c': /* configuration file */
- if (configfile != NULL) {
- free(configfile);
- }
- configfile = bstrdup(optarg);
- break;
-
- case 'd':
- debug_level = atoi(optarg);
- if (debug_level <= 0) {
- debug_level = 1;
- }
- break;
+ case 'c': /* configuration file */
+ if (configfile != NULL) {
+ free(configfile);
+ }
+ configfile = bstrdup(optarg);
+ break;
- case 's': /* turn off signals */
- no_signals = TRUE;
- break;
+ case 'd':
+ debug_level = atoi(optarg);
+ if (debug_level <= 0) {
+ debug_level = 1;
+ }
+ break;
- case 't':
- test_config = TRUE;
- break;
+ case 's': /* turn off signals */
+ no_signals = true;
+ break;
- case '?':
- default:
- usage();
+ case 't':
+ test_config = true;
+ break;
+ case '?':
+ default:
+ usage();
+ con_term();
+ exit(1);
}
}
argc -= optind;
if (!no_signals) {
init_signals(terminate_console);
}
+
+ /* Override Bacula default signals */
signal(SIGCHLD, SIG_IGN);
- signal(SIGTSTP, got_stop);
- signal(SIGCONT, got_continue);
- signal(SIGTTIN, got_tin);
- signal(SIGTTOU, got_tout);
+ signal(SIGTSTP, got_sigstop);
+ signal(SIGCONT, got_sigcontinue);
+ signal(SIGTTIN, got_sigtin);
+ signal(SIGTTOU, got_sigtout);
+ trapctlc();
if (argc) {
usage();
+ con_term();
+ exit(1);
}
if (configfile == NULL) {
LockRes();
ndir = 0;
- for (dir=NULL; (dir = (DIRRES *)GetNextRes(R_DIRECTOR, (RES *)dir)); ) {
+ foreach_res(dir, R_DIRECTOR) {
ndir++;
}
UnlockRes();
if (ndir == 0) {
+ con_term();
Emsg1(M_ERROR_TERM, 0, _("No Director resource defined in %s\n\
Without that I don't how to speak to the Director :-(\n"), configfile);
}
memset(&jcr, 0, sizeof(jcr));
+
if (ndir > 1) {
- UA_sock = init_bsock(NULL, 0, "", "", 0);
+ struct sockaddr_in client_addr;
+ memset(&client_addr, 0, sizeof(client_addr));
+ UA_sock = init_bsock(NULL, 0, "", "", 0, &client_addr);
try_again:
sendit(_("Available Directors:\n"));
LockRes();
ndir = 0;
- for (dir = NULL; (dir = (DIRRES *)GetNextRes(R_DIRECTOR, (RES *)dir)); ) {
- fprintf(output, _("%d %s at %s:%d\n"), 1+ndir++, dir->hdr.name, dir->address,
+ foreach_res(dir, R_DIRECTOR) {
+ senditf( _("%d %s at %s:%d\n"), 1+ndir++, dir->hdr.name, dir->address,
dir->DIRport);
}
UnlockRes();
}
item = atoi(UA_sock->msg);
if (item < 0 || item > ndir) {
- sendit(_("You must enter a number between 1 and %d\n"), ndir);
+ senditf(_("You must enter a number between 1 and %d\n"), ndir);
goto try_again;
}
LockRes();
UnlockRes();
}
-
- sendit(_("Connecting to Director %s:%d\n"), dir->address,dir->DIRport);
+ senditf(_("Connecting to Director %s:%d\n"), dir->address,dir->DIRport);
UA_sock = bnet_connect(NULL, 5, 15, "Director daemon", dir->address,
NULL, dir->DIRport, 0);
if (UA_sock == NULL) {
return 1;
}
jcr.dir_bsock = UA_sock;
- if (!authenticate_director(&jcr, dir)) {
+
+ LockRes();
+ CONRES *cons = (CONRES *)GetNextRes(R_CONSOLE, (RES *)NULL);
+ UnlockRes();
+ /* If cons==NULL, default console will be used */
+ if (!authenticate_director(&jcr, dir, cons)) {
fprintf(stderr, "ERR=%s", UA_sock->msg);
terminate_console(0);
return 1;
Dmsg0(40, "Opened connection with Director daemon\n");
+ sendit(_("Enter a period to cancel a command.\n"));
+
+ /* Run commands in ~/.bconsolerc if any */
+ char *env = getenv("HOME");
+ if (env) {
+ FILE *fd;
+ pm_strcpy(&UA_sock->msg, env);
+ pm_strcat(&UA_sock->msg, "/.bconsolerc");
+ fd = fopen(UA_sock->msg, "r");
+ if (fd) {
+ read_and_process_input(fd, UA_sock);
+ fclose(fd);
+ }
+ }
+
read_and_process_input(stdin, UA_sock);
if (UA_sock) {
/* Cleanup and then exit */
static void terminate_console(int sig)
{
- static int already_here = FALSE;
+ static bool already_here = false;
if (already_here) { /* avoid recursive temination problems */
exit(1);
}
- already_here = TRUE;
+ already_here = true;
free_pool_memory(args);
+ con_term();
if (sig != 0) {
exit(1);
}
if (!line) {
exit(1);
}
- strcpy(sock->msg, line);
- strip_trailing_junk(sock->msg);
- sock->msglen = strlen(sock->msg);
+ strip_trailing_junk(line);
+ sock->msglen = pm_strcpy(&sock->msg, line);
if (sock->msglen) {
add_history(sock->msg);
}
int len;
if (!stop) {
if (output == stdout || tee) {
- fputs(prompt, stdout);
- fflush(stdout);
+ sendit(prompt);
}
}
again:
sleep(1);
goto again;
}
+#ifdef HAVE_CONIO
+ if (isatty(fileno(input))) {
+ input_line(sock->msg, len);
+ break;
+ }
+#endif
if (fgets(sock->msg, len, input) == NULL) {
return -1;
}
static int versioncmd(FILE *input, BSOCK *UA_sock)
{
- sendit("Version: " VERSION " (" BDATE ") %s %s %s\n",
+ senditf("Version: " VERSION " (" BDATE ") %s %s %s\n",
HOST_OS, DISTNAME, DISTVER);
return 1;
}
FILE *fd;
if (argc > 2) {
- sendit(_("Too many arguments.\n"));
- return 0;
+ sendit(_("Too many arguments on input command.\n"));
+ return 1;
}
if (argc == 1) {
- sendit(_("First argument must be a filename.\n"));
- return 0;
+ sendit(_("First argument to input command must be a filename.\n"));
+ return 1;
}
fd = fopen(argk[1], "r");
if (!fd) {
- sendit(_("Cannot open file. ERR=%s\n"), strerror(errno));
- return 0;
+ senditf(_("Cannot open file %s for input. ERR=%s\n"),
+ argk[1], strerror(errno));
+ return 1;
}
read_and_process_input(fd, UA_sock);
fclose(fd);
return 1;
}
+/* Send output to both termina and specified file */
static int teecmd(FILE *input, BSOCK *UA_sock)
{
- tee = 1;
+ tee = true;
return do_outputcmd(input, UA_sock);
}
+/* Send output to specified "file" */
static int outputcmd(FILE *input, BSOCK *UA_sock)
{
- tee = 0;
+ tee = false;
return do_outputcmd(input, UA_sock);
}
char *mode = "a+";
if (argc > 3) {
- sendit(_("Too many arguments.\n"));
+ sendit(_("Too many arguments on output/tee command.\n"));
return 1;
}
if (argc == 1) {
if (output != stdout) {
fclose(output);
output = stdout;
- tee = 0;
+ tee = false;
}
return 1;
}
}
fd = fopen(argk[1], mode);
if (!fd) {
- sendit(_("Cannot open file. ERR=%s\n"), strerror(errno));
+ senditf(_("Cannot open file %s for output. ERR=%s\n"),
+ argk[1], strerror(errno));
return 1;
}
output = fd;
struct tm tm;
localtime_r(&ttime, &tm);
strftime(sdt, sizeof(sdt), "%d-%b-%Y %H:%M:%S", &tm);
- sendit(sdt);
sendit("\n");
return 1;
}
-static void sendit(char *fmt,...)
+/*
+ * Send a line to the output file and or the terminal
+ */
+void senditf(char *fmt,...)
{
char buf[3000];
va_list arg_ptr;
va_start(arg_ptr, fmt);
bvsnprintf(buf, sizeof(buf), (char *)fmt, arg_ptr);
va_end(arg_ptr);
+ sendit(buf);
+}
+
+void sendit(char *buf)
+{
+#ifdef xHAVE_CONIO
+ if (output == stdout || tee) {
+ char *p, *q;
+ /*
+ * Here, we convert every \n into \r\n because the
+ * terminal is in raw mode when we are using
+ * conio.
+ */
+ for (p=q=buf; (p=strchr(q, '\n')); ) {
+ if (p-q > 0) {
+ t_sendl(q, p-q);
+ }
+ t_sendl("\r\n", 2);
+ q = ++p; /* point after \n */
+ }
+ if (*q) {
+ t_send(q);
+ }
+ }
+ if (output != stdout) {
+ fputs(buf, output);
+ }
+#else
fputs(buf, output);
if (tee) {
fputs(buf, stdout);
}
fflush(stdout);
+#endif
}