*
*/
-#include <QAbstractEventDispatcher>
+//#include <QAbstractEventDispatcher>
#include "bat.h"
#include "console.h"
+#include "restore.h"
+#include "select.h"
+#include "run/run.h"
+
+static int tls_pem_callback(char *buf, int size, const void *userdata);
+
Console::Console(QStackedWidget *parent)
{
QFont font;
- (void)parent;
+ m_parent = parent;
+ m_closeable = false;
+ m_console = this;
setupUi(this);
m_sock = NULL;
m_at_prompt = false;
+ m_at_main_prompt = false;
m_textEdit = textEdit; /* our console screen */
m_cursor = new QTextCursor(m_textEdit->document());
- mainWin->actionConnect->setIcon(QIcon(QString::fromUtf8("images/disconnected.png")));
-
-
+ mainWin->actionConnect->setIcon(QIcon(":images/disconnected.png"));
+
+ m_timer = NULL;
+ m_contextActions.append(actionStatusDir);
+ m_contextActions.append(actionConsoleHelp);
+ m_contextActions.append(actionRequestMessages);
+ m_contextActions.append(actionConsoleReload);
+ connect(actionStatusDir, SIGNAL(triggered()), this, SLOT(status_dir()));
+ connect(actionConsoleHelp, SIGNAL(triggered()), this, SLOT(consoleHelp()));
+ connect(actionConsoleReload, SIGNAL(triggered()), this, SLOT(consoleReload()));
+ connect(actionRequestMessages, SIGNAL(triggered()), this, SLOT(messages()));
+}
- readSettings();
- /* Check for messages every 5 seconds */
-// m_timer = new QTimer(this);
-// QWidget::connect(m_timer, SIGNAL(timeout()), this, SLOT(poll_messages()));
-// m_timer->start(5000);
+Console::~Console()
+{
+}
+void Console::startTimer()
+{
+ m_timer = new QTimer(this);
+ QWidget::connect(m_timer, SIGNAL(timeout()), this, SLOT(poll_messages()));
+ m_timer->start(mainWin->m_checkMessagesInterval*1000);
}
+void Console::stopTimer()
+{
+ if (m_timer) {
+ QWidget::disconnect(m_timer, SIGNAL(timeout()), this, SLOT(poll_messages()));
+ m_timer->stop();
+ delete m_timer;
+ m_timer = NULL;
+ }
+}
+
void Console::poll_messages()
{
m_messages_pending = true;
+ if ((m_at_main_prompt) && (mainWin->m_checkMessages)){
+ write(".messages");
+ displayToPrompt();
+ }
}
/* Terminate any open socket */
void Console::terminate()
{
if (m_sock) {
+ stopTimer();
m_sock->close();
m_sock = NULL;
}
-// m_timer->stop();
}
/*
- * Connect to Director. If there are more than one, put up
- * a modal dialog so that the user chooses one.
+ * Connect to Director.
*/
-void Console::connect()
+void Console::connect_dir()
{
JCR jcr;
+ utime_t heart_beat;
+ char buf[1024];
m_textEdit = textEdit; /* our console screen */
LockRes();
/* If cons==NULL, default console will be used */
- CONRES *cons = (CONRES *)GetNextRes(R_CONSOLE, (RES *)NULL);
+ CONRES *cons = (CONRES *)GetNextRes(R_CONSOLE, NULL);
UnlockRes();
- m_sock = bnet_connect(NULL, 5, 15, _("Director daemon"), m_dir->address,
+ /* Initialize Console TLS context once */
+ if (cons && !cons->tls_ctx && (cons->tls_enable || cons->tls_require)) {
+ /* Generate passphrase prompt */
+ bsnprintf(buf, sizeof(buf), "Passphrase for Console \"%s\" TLS private key: ",
+ cons->name());
+
+ /* Initialize TLS context:
+ * Args: CA certfile, CA certdir, Certfile, Keyfile,
+ * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer
+ */
+ cons->tls_ctx = new_tls_context(cons->tls_ca_certfile,
+ cons->tls_ca_certdir, cons->tls_certfile,
+ cons->tls_keyfile, tls_pem_callback, &buf, NULL, true);
+
+ if (!cons->tls_ctx) {
+ display_textf(_("Failed to initialize TLS context for Console \"%s\".\n"),
+ m_dir->name());
+ return;
+ }
+ }
+
+ /* Initialize Director TLS context once */
+ if (!m_dir->tls_ctx && (m_dir->tls_enable || m_dir->tls_require)) {
+ /* Generate passphrase prompt */
+ bsnprintf(buf, sizeof(buf), "Passphrase for Director \"%s\" TLS private key: ",
+ m_dir->name());
+
+ /* Initialize TLS context:
+ * Args: CA certfile, CA certdir, Certfile, Keyfile,
+ * Keyfile PEM Callback, Keyfile CB Userdata, DHfile, Verify Peer */
+ m_dir->tls_ctx = new_tls_context(m_dir->tls_ca_certfile,
+ m_dir->tls_ca_certdir, m_dir->tls_certfile,
+ m_dir->tls_keyfile, tls_pem_callback, &buf, NULL, true);
+
+ if (!m_dir->tls_ctx) {
+ display_textf(_("Failed to initialize TLS context for Director \"%s\".\n"),
+ m_dir->name());
+ mainWin->set_status("Connection failed");
+ return;
+ }
+ }
+
+ if (m_dir->heartbeat_interval) {
+ heart_beat = m_dir->heartbeat_interval;
+ } else if (cons) {
+ heart_beat = cons->heartbeat_interval;
+ } else {
+ heart_beat = 0;
+ }
+
+ m_sock = bnet_connect(NULL, 5, 15, heart_beat,
+ _("Director daemon"), m_dir->address,
NULL, m_dir->DIRport, 0);
if (m_sock == NULL) {
mainWin->set_status("Connection failed");
return;
} else {
/* Update page selector to green to indicate that Console is connected */
- mainWin->actionConnect->setIcon(QIcon(QString::fromUtf8("images/connected.png")));
+ mainWin->actionConnect->setIcon(QIcon(":images/connected.png"));
QBrush greenBrush(Qt::green);
- m_consoleItem->setForeground(0, greenBrush);
+ QTreeWidgetItem *item = mainWin->getFromHash(this);
+ item->setForeground(0, greenBrush);
}
jcr.dir_bsock = m_sock;
- if (!authenticate_director(&jcr, m_dir, cons)) {
- display_text(m_sock->msg);
+ if (!authenticate_director(&jcr, m_dir, cons, buf, sizeof(buf))) {
+ display_text(buf);
return;
}
+ if (buf[0]) {
+ display_text(buf);
+ }
/* Give GUI a chance */
app->processEvents();
mainWin->set_status(_("Initializing ..."));
/* Set up input notifier */
- m_notifier = new QSocketNotifier(m_sock->fd, QSocketNotifier::Read, 0);
+ m_notifier = new QSocketNotifier(m_sock->m_fd, QSocketNotifier::Read, 0);
QObject::connect(m_notifier, SIGNAL(activated(int)), this, SLOT(read_dir(int)));
write(".api 1");
- discardToPrompt();
+ displayToPrompt();
beginNewCommand();
- job_list = get_list(".jobs");
- client_list = get_list(".clients");
- fileset_list = get_list(".filesets");
- messages_list = get_list(".messages");
- pool_list = get_list(".pools");
- storage_list = get_list(".storage");
- type_list = get_list(".types");
- level_list = get_list(".levels");
+ dir_cmd(".jobs", job_list);
+ dir_cmd(".clients", client_list);
+ dir_cmd(".filesets", fileset_list);
+ dir_cmd(".msgs", messages_list);
+ dir_cmd(".pools", pool_list);
+ dir_cmd(".storage", storage_list);
+ dir_cmd(".types", type_list);
+ dir_cmd(".levels", level_list);
mainWin->set_status(_("Connected"));
+ startTimer(); /* start message timer */
return;
}
+bool Console::dir_cmd(QString &cmd, QStringList &results)
+{
+ return dir_cmd(cmd.toUtf8().data(), results);
+}
-/*
- * Send a command to the director, and read all the resulting
- * output into a list.
+/*
+ * Send a command to the Director, and return the
+ * results in a QStringList.
*/
-QStringList Console::get_list(char *cmd)
+bool Console::dir_cmd(const char *cmd, QStringList &results)
{
- QStringList list;
int stat;
notify(false);
write(cmd);
while ((stat = read()) > 0) {
+ if (mainWin->m_displayAll) display_text(msg());
+ strip_trailing_junk(msg());
+ results << msg();
+ }
+ notify(true);
+ discardToPrompt();
+ return true; /* ***FIXME*** return any command error */
+}
+
+bool Console::sql_cmd(QString &query, QStringList &results)
+{
+ return sql_cmd(query.toUtf8().data(), results);
+}
+
+/*
+ * Send an sql query to the Director, and return the
+ * results in a QStringList.
+ */
+bool Console::sql_cmd(const char *query, QStringList &results)
+{
+ int stat;
+ POOL_MEM cmd(PM_MESSAGE);
+
+ if (!is_connectedGui()) {
+ return false;
+ }
+
+ notify(false);
+
+ pm_strcpy(cmd, ".sql query=\"");
+ pm_strcat(cmd, query);
+ pm_strcat(cmd, "\"");
+ write(cmd.c_str());
+ while ((stat = read()) > 0) {
+ if (mainWin->m_displayAll) display_text(msg());
strip_trailing_junk(msg());
- list << msg();
+ results << msg();
}
notify(true);
- return list;
+ discardToPrompt();
+ return true; /* ***FIXME*** return any command error */
}
+
/*
* Send a job name to the director, and read all the resulting
* defaults.
scmd = QString(".defaults job=\"%1\"").arg(job_defs.job_name);
write(scmd);
while ((stat = read()) > 0) {
+ if (mainWin->m_displayAll) display_text(msg());
def = strchr(msg(), '=');
if (!def) {
continue;
}
}
-#ifdef xxx
- bsnprintf(cmd, sizeof(cmd), "job=%s pool=%s client=%s storage=%s where=%s\n"
- "level=%s type=%s fileset=%s catalog=%s enabled=%d\n",
- job_defs.job_name.toUtf8().data(), job_defs.pool_name.toUtf8().data(),
- job_defs.client_name.toUtf8().data(),
- job_defs.pool_name.toUtf8().data(), job_defs.messages_name.toUtf8().data(),
- job_defs.store_name.toUtf8().data(),
- job_defs.where.toUtf8().data(), job_defs.level.toUtf8().data(),
- job_defs.type.toUtf8().data(), job_defs.fileset_name.toUtf8().data(),
- job_defs.catalog_name.toUtf8().data(), job_defs.enabled);
-#endif
-
notify(true);
return true;
{
QFont font = get_font();
- QSettings settings("bacula.org", "bat");
- /* ***FIXME*** make console name unique */
+ QSettings settings(m_dir->name(), "bat");
settings.beginGroup("Console");
settings.setValue("consoleFont", font.family());
settings.setValue("consolePointSize", font.pointSize());
{
QFont font = get_font();
- QSettings settings("bacula.org", "bat");
+ QSettings settings(m_dir->name(), "bat");
settings.beginGroup("Console");
font.setFamily(settings.value("consoleFont", "Courier").value<QString>());
font.setPointSize(settings.value("consolePointSize", 10).toInt());
return m_textEdit->font();
}
-
+/*
+ * Slot for responding to status dir button on button bar
+ */
void Console::status_dir()
{
- write_dir("status dir\n");
- displayToPrompt();
+ QString cmd("status dir");
+ consoleCommand(cmd);
+}
+
+/*
+ * Slot for responding to messages button on button bar
+ */
+void Console::messages()
+{
+ QString cmd(".messages");
+ consoleCommand(cmd);
}
/*
void Console::display_text(const QString buf)
{
- m_cursor->movePosition(QTextCursor::End);
m_cursor->insertText(buf);
+ update_cursor();
}
void Console::display_text(const char *buf)
{
- m_cursor->movePosition(QTextCursor::End);
m_cursor->insertText(buf);
+ update_cursor();
+}
+
+void Console::display_html(const QString buf)
+{
+ m_cursor->insertHtml(buf);
+ update_cursor();
}
/* Position cursor to end of screen */
write(msg);
} else {
mainWin->set_status(" Director not connected. Click on connect button.");
- mainWin->actionConnect->setIcon(QIcon(QString::fromUtf8("images/disconnected.png")));
+ mainWin->actionConnect->setIcon(QIcon(":images/disconnected.png"));
QBrush redBrush(Qt::red);
- m_consoleItem->setForeground(0, redBrush);
+ QTreeWidgetItem *item = mainWin->getFromHash(this);
+ item->setForeground(0, redBrush);
m_at_prompt = false;
+ m_at_main_prompt = false;
}
}
int Console::write(const char *msg)
{
- m_sock->msglen = strlen(msg);
- pm_strcpy(&m_sock->msg, msg);
+ if (!m_sock) {
+ return -1;
+ }
+ m_sock->msglen = pm_strcpy(m_sock->msg, msg);
m_at_prompt = false;
- if (commDebug) Pmsg1(000, "send: %s\n", msg);
+ m_at_main_prompt = false;
+ if (mainWin->m_commDebug) Pmsg1(000, "send: %s\n", msg);
return m_sock->send();
+
}
/*
- * Get to main command prompt
+ * Get to main command prompt -- i.e. abort any subcommand
*/
void Console::beginNewCommand()
{
- write(".\n");
- while (read() > 0) {
- }
- write(".\n");
- while (read() > 0) {
- }
- write(".\n");
- while (read() > 0) {
+ for (int i=0; i < 3; i++) {
+ write(".");
+ while (read() > 0) {
+ if (mainWin->m_displayAll) display_text(msg());
+ }
+ if (m_at_main_prompt) {
+ break;
+ }
}
display_text("\n");
}
void Console::displayToPrompt()
{
- int stat;
- if (commDebug) Pmsg0(000, "DisplaytoPrompt\n");
+ int stat = 0;
+ if (mainWin->m_commDebug) Pmsg0(000, "DisplaytoPrompt\n");
while (!m_at_prompt) {
if ((stat=read()) > 0) {
display_text(msg());
}
}
- if (commDebug) Pmsg1(000, "endDisplaytoPrompt=%d\n", stat);
+ if (mainWin->m_commDebug) Pmsg1(000, "endDisplaytoPrompt=%d\n", stat);
}
void Console::discardToPrompt()
{
- int stat;
- if (commDebug) Pmsg0(000, "discardToPrompt\n");
+ int stat = 0;
+ if (mainWin->m_commDebug) Pmsg0(000, "discardToPrompt\n");
while (!m_at_prompt) {
- stat = read();
+ if ((stat=read()) > 0) {
+ if (mainWin->m_displayAll) display_text(msg());
+ }
}
- if (commDebug) Pmsg1(000, "endDisplayToPrompt=%d\n", stat);
+ if (mainWin->m_commDebug) Pmsg1(000, "endDisplayToPrompt=%d\n", stat);
}
if (m_at_prompt) {
display_text("\n");
m_at_prompt = false;
+ m_at_main_prompt = false;
}
- if (commDebug) Pmsg1(000, "got: %s", m_sock->msg);
+ if (mainWin->m_commDebug) Pmsg1(000, "got: %s", m_sock->msg);
}
switch (m_sock->msglen) {
- case BNET_SERVER_READY:
- if (m_api_set && m_messages_pending) {
- write_dir(".messages");
- m_messages_pending = false;
- }
- m_at_prompt = true;
- continue;
case BNET_MSGS_PENDING:
- if (commDebug) Pmsg0(000, "MSGS PENDING\n");
- m_messages_pending = true;
+ if (mainWin->m_commDebug) Pmsg0(000, "MSGS PENDING\n");
+ write_dir(".messages");
+ displayToPrompt();
+ m_messages_pending = false;
continue;
case BNET_CMD_OK:
- if (commDebug) Pmsg0(000, "CMD OK\n");
+ if (mainWin->m_commDebug) Pmsg0(000, "CMD OK\n");
m_at_prompt = false;
+ m_at_main_prompt = false;
+ mainWin->set_status(_("Command completed ..."));
continue;
case BNET_CMD_BEGIN:
- if (commDebug) Pmsg0(000, "CMD BEGIN\n");
+ if (mainWin->m_commDebug) Pmsg0(000, "CMD BEGIN\n");
m_at_prompt = false;
+ m_at_main_prompt = false;
+ mainWin->set_status(_("Processing command ..."));
continue;
+ case BNET_MAIN_PROMPT:
+ if (mainWin->m_commDebug) Pmsg0(000, "MAIN PROMPT\n");
+ m_at_prompt = true;
+ m_at_main_prompt = true;
+ mainWin->set_status(_("At main prompt waiting for input ..."));
+ QApplication::restoreOverrideCursor();
+ break;
case BNET_PROMPT:
- if (commDebug) Pmsg0(000, "PROMPT\n");
+ if (mainWin->m_commDebug) Pmsg0(000, "PROMPT\n");
m_at_prompt = true;
+ m_at_main_prompt = false;
mainWin->set_status(_("At prompt waiting for input ..."));
- update_cursor();
QApplication::restoreOverrideCursor();
break;
case BNET_CMD_FAILED:
- if (commDebug) Pmsg0(000, "CMD FAIL\n");
- mainWin->set_status(_("Command failed. At prompt waiting for input ..."));
- update_cursor();
+ if (mainWin->m_commDebug) Pmsg0(000, "CMD FAILED\n");
+ mainWin->set_status(_("Command failed."));
QApplication::restoreOverrideCursor();
break;
/* We should not get this one */
case BNET_EOD:
- if (commDebug) Pmsg0(000, "EOD\n");
+ if (mainWin->m_commDebug) Pmsg0(000, "EOD\n");
mainWin->set_status_ready();
- update_cursor();
QApplication::restoreOverrideCursor();
if (!m_api_set) {
break;
}
continue;
case BNET_START_SELECT:
+ if (mainWin->m_commDebug) Pmsg0(000, "START SELECT\n");
new selectDialog(this);
break;
case BNET_RUN_CMD:
- new runCmdDialog(this);
+ if (mainWin->m_commDebug) Pmsg0(000, "RUN CMD\n");
+ new runCmdPage();
break;
case BNET_ERROR_MSG:
+ if (mainWin->m_commDebug) Pmsg0(000, "ERROR MSG\n");
m_sock->recv(); /* get the message */
display_text(msg());
QMessageBox::critical(this, "Error", msg(), QMessageBox::Ok);
break;
case BNET_WARNING_MSG:
+ if (mainWin->m_commDebug) Pmsg0(000, "WARNING MSG\n");
m_sock->recv(); /* get the message */
display_text(msg());
QMessageBox::critical(this, "Warning", msg(), QMessageBox::Ok);
break;
case BNET_INFO_MSG:
+ if (mainWin->m_commDebug) Pmsg0(000, "INFO MSG\n");
m_sock->recv(); /* get the message */
display_text(msg());
mainWin->set_status(msg());
break;
}
if (is_bnet_stop(m_sock)) { /* error or term request */
+ if (mainWin->m_commDebug) Pmsg0(000, "BNET STOP\n");
+ stopTimer();
m_sock->close();
m_sock = NULL;
- mainWin->actionConnect->setIcon(QIcon(QString::fromUtf8("images/disconnected.png")));
+ mainWin->actionConnect->setIcon(QIcon(":images/disconnected.png"));
QBrush redBrush(Qt::red);
- m_consoleItem->setForeground(0, redBrush);
+ QTreeWidgetItem *item = mainWin->getFromHash(this);
+ item->setForeground(0, redBrush);
m_notifier->setEnabled(false);
delete m_notifier;
m_notifier = NULL;
}
/* Called by signal when the Director has output for us */
-void Console::read_dir(int fd)
+void Console::read_dir(int /* fd */)
{
- int stat;
- (void)fd;
-
- if (commDebug) Pmsg0(000, "read_dir\n");
- while ((stat = read()) >= 0) {
+ if (mainWin->m_commDebug) Pmsg0(000, "read_dir\n");
+ while (read() >= 0) {
display_text(msg());
}
}
m_notifier->setEnabled(enable);
}
-void Console::setTreeItem(QTreeWidgetItem *item)
-{
- m_consoleItem = item;
+void Console::setDirectorTreeItem(QTreeWidgetItem *item)
+{
+ m_directorTreeItem = item;
}
void Console::setDirRes(DIRRES *dir)
{
m_dir = dir;
}
+
+/*
+ * To have the ability to get the name of the director resource.
+ */
+void Console::getDirResName(QString &name_returned)
+{
+ name_returned = m_dir->name();
+}
+
+bool Console::is_connectedGui()
+{
+ if (is_connected()) {
+ return true;
+ } else {
+ QString message("Director ");
+ message += " is currently disconnected\n Please reconnect!!";
+ QMessageBox::warning(this, "Bat",
+ tr(message.toUtf8().data()), QMessageBox::Ok );
+ return false;
+ }
+}
+
+/*
+ * A temporary function to prevent connecting to the director if the director
+ * is busy with a restore.
+ */
+bool Console::preventInUseConnect()
+{
+ if (!is_connected()) {
+ QString message("Director ");
+ message += m_dir->name();
+ message += " is currently disconnected\n Please reconnect!!";
+ QMessageBox::warning(this, "Bat",
+ tr(message.toUtf8().data()), QMessageBox::Ok );
+ return false;
+ } else if (!m_at_main_prompt){
+ QString message("Director ");
+ message += m_dir->name();
+ message += " is currently busy\n Please complete restore or other "
+ " operation !! This is a limitation that will be resolved before a beta"
+ " release. This is currently an alpha release.";
+ QMessageBox::warning(this, "Bat",
+ tr(message.toUtf8().data()), QMessageBox::Ok );
+ return false;
+ } else if (!m_at_prompt){
+ QString message("Director ");
+ message += m_dir->name();
+ message += " is currently not at a prompt\n Please try again!!";
+ QMessageBox::warning(this, "Bat",
+ tr(message.toUtf8().data()), QMessageBox::Ok );
+ return false;
+ } else {
+ return true;
+ }
+}
+
+/*
+ * Call-back for reading a passphrase for an encrypted PEM file
+ * This function uses getpass(),
+ * which uses a static buffer and is NOT thread-safe.
+ */
+static int tls_pem_callback(char *buf, int size, const void *userdata)
+{
+ (void)size;
+ (void)userdata;
+#ifdef HAVE_TLS
+ const char *prompt = (const char *)userdata;
+# if defined(HAVE_WIN32)
+ sendit(prompt);
+ if (win32_cgets(buf, size) == NULL) {
+ buf[0] = 0;
+ return 0;
+ } else {
+ return strlen(buf);
+ }
+# else
+ char *passwd;
+
+ passwd = getpass(prompt);
+ bstrncpy(buf, passwd, size);
+ return strlen(buf);
+# endif
+#else
+ buf[0] = 0;
+ return 0;
+#endif
+}
+
+/* Slot for responding to page selectors status help command */
+void Console::consoleHelp()
+{
+ QString cmd("help");
+ consoleCommand(cmd);
+}
+
+/* Slot for responding to page selectors reload bacula-dir.conf */
+void Console::consoleReload()
+{
+ QString cmd("reload");
+ consoleCommand(cmd);
+}