]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/qt-console/console/console.cpp
kes Fix sscanf problems reported by Peter Buschman that caused
[bacula/bacula] / bacula / src / qt-console / console / console.cpp
index 60a230a57d2726c82de06fc8f5991da00a24f714..dbbc6f38df0c4ddcff3d978276a7680b22a2a67e 100644 (file)
@@ -1,7 +1,7 @@
 /*
    Bacula® - The Network Backup Solution
 
-   Copyright (C) 2000-2007 Free Software Foundation Europe e.V.
+   Copyright (C) 2007-2007 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.
    (FSFE), Fiduciary Program, Sumatrastrasse 25, 8006 Zürich,
    Switzerland, email:ftf@fsfeurope.org.
 */
 /*
  *   Version $Id$
  *
  *  Console Class
  *
- *   Kern Sibbald, January MMVI
+ *   Kern Sibbald, January MMVII
  *
  */ 
 
-#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;
-   QTreeWidgetItem *item, *topItem;
-   QTreeWidget *treeWidget = mainWin->treeWidget;
+   m_parent = parent;
+   m_closeable = false;
+   m_console = this;
 
    setupUi(this);
-   parent->addWidget(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")));
-
-   bRestore *brestore = new bRestore(parent);
-   brestore->setupUi(brestore);
-   parent->addWidget(brestore);
-
-   /* Just take the first Director */
-   LockRes();
-   m_dir = (DIRRES *)GetNextRes(R_DIRECTOR, NULL);
-   UnlockRes();
+   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()));
+}
 
-   /* ***FIXME*** Dummy setup of treeWidget */
-   treeWidget->clear();
-   treeWidget->setColumnCount(1);
-   treeWidget->setHeaderLabel("Selection");
-   topItem = new QTreeWidgetItem(treeWidget);
-   topItem->setText(0, m_dir->name());
-   topItem->setIcon(0, QIcon(QString::fromUtf8("images/server.png")));
-   item = new QTreeWidgetItem(topItem);
-   m_consoleItem = item;
-   item->setText(0, "Console");
-   item->setText(1, "0");
-   QBrush redBrush(Qt::red);
-   item->setForeground(0, redBrush);
-   item = new QTreeWidgetItem(topItem);
-   item->setText(0, "brestore");
-   item->setText(1, "1");
-   treeWidget->expandItem(topItem);
+Console::~Console()
+{
+}
 
-   readSettings();
+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;
    }
 }
 
 /*
- * 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 */
 
@@ -122,27 +140,82 @@ void Console::connect()
    
    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();
@@ -150,44 +223,87 @@ void Console::connect()
    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");
+   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;
 
-   setEnabled(false);
+   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();
    }
-   setEnabled(true);
-   list.sort();
-   return list;
+   notify(true);
+   discardToPrompt();
+   return true;              /* ***FIXME*** return any command error */
 }
 
+
 /*  
  * Send a job name to the director, and read all the resulting
  *  defaults. 
@@ -195,15 +311,15 @@ QStringList Console::get_list(char *cmd)
 bool Console::get_job_defaults(struct job_defaults &job_defs)
 {
    QString scmd;
-   char cmd[1000];
    int stat;
    char *def;
 
-   setEnabled(false);
+   notify(false);
    beginNewCommand();
    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;
@@ -259,21 +375,12 @@ bool Console::get_job_defaults(struct job_defaults &job_defs)
          continue;
       }
    }
-   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);
-
-   setEnabled(true);
+
+   notify(true);
    return true;
 
 bail_out:
-   setEnabled(true);
+   notify(true);
    return false;
 }
 
@@ -285,8 +392,7 @@ void Console::writeSettings()
 {
    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());
@@ -301,7 +407,7 @@ void Console::readSettings()
 { 
    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());
@@ -330,10 +436,22 @@ const QFont Console::get_font()
    return m_textEdit->font();
 }
 
-
+/*
+ * Slot for responding to status dir button on button bar
+ */
 void Console::status_dir()
 {
-   write_dir("status dir\n");
+   QString cmd("status dir");
+   consoleCommand(cmd);
+}
+
+/*
+ * Slot for responding to messages button on button bar
+ */
+void Console::messages()
+{
+   QString cmd(".messages");
+   consoleCommand(cmd);
 }
 
 /*
@@ -352,15 +470,21 @@ void Console::display_textf(const char *fmt, ...)
 
 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 */
@@ -386,17 +510,17 @@ char *Console::msg()
 void Console::write_dir(const char *msg)
 {
    if (m_sock) {
-      m_at_prompt = false;
       mainWin->set_status(_("Processing command ..."));
       QApplication::setOverrideCursor(Qt::WaitCursor);
-      m_sock->msglen = strlen(msg);
-      pm_strcpy(&m_sock->msg, msg);
-      m_sock->send();
+      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;
    }
 }
 
@@ -407,98 +531,312 @@ int Console::write(const QString msg)
 
 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;
+   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()
 {
-// displayToPrompt();
-   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()
 { 
-   while (read() > 0) {
-      display_text(msg());
+   int stat = 0;
+   if (mainWin->m_commDebug) Pmsg0(000, "DisplaytoPrompt\n");
+   while (!m_at_prompt) {
+      if ((stat=read()) > 0) {
+         display_text(msg());
+      }
    }
+   if (mainWin->m_commDebug) Pmsg1(000, "endDisplaytoPrompt=%d\n", stat);
+}
 
+void Console::discardToPrompt()
+{ 
+   int stat = 0;
+   if (mainWin->m_commDebug) Pmsg0(000, "discardToPrompt\n");
+   while (!m_at_prompt) {
+      if ((stat=read()) > 0) {
+         if (mainWin->m_displayAll) display_text(msg());
+      }
+   }
+   if (mainWin->m_commDebug) Pmsg1(000, "endDisplayToPrompt=%d\n", stat);
 }
 
+
 /* 
- * Blocking read from director */
+ * Blocking read from director
+ */
 int Console::read()
 {
-   int stat;
-   if (m_sock) {
+   int stat = 0;
+   while (m_sock) {
       for (;;) {
          stat = bnet_wait_data_intr(m_sock, 1);
          if (stat > 0) {
             break;
          } 
          app->processEvents();
-         if (stat < 0) {
-            return BNET_ERROR;
+         if (m_api_set && m_messages_pending) {
+            write_dir(".messages");
+            m_messages_pending = false;
+         }
+      }
+      stat = m_sock->recv();
+      if (stat >= 0) {
+         if (m_at_prompt) {
+            display_text("\n");
+            m_at_prompt = false;
+            m_at_main_prompt = false;
          }
+         if (mainWin->m_commDebug) Pmsg1(000, "got: %s", m_sock->msg);
+      }
+      switch (m_sock->msglen) {
+      case BNET_MSGS_PENDING:
+         if (mainWin->m_commDebug) Pmsg0(000, "MSGS PENDING\n");
+         write_dir(".messages");
+         displayToPrompt();
+         m_messages_pending = false;
+         continue;
+      case BNET_CMD_OK:
+         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 (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 (mainWin->m_commDebug) Pmsg0(000, "PROMPT\n");
+         m_at_prompt = true;
+         m_at_main_prompt = false;
+         mainWin->set_status(_("At prompt waiting for input ..."));
+         QApplication::restoreOverrideCursor();
+         break;
+      case BNET_CMD_FAILED:
+         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 (mainWin->m_commDebug) Pmsg0(000, "EOD\n");
+         mainWin->set_status_ready();
+         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:
+         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;
       }
-      return m_sock->recv();
+      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(":images/disconnected.png"));
+         QBrush redBrush(Qt::red);
+         QTreeWidgetItem *item = mainWin->getFromHash(this);
+         item->setForeground(0, redBrush);
+         m_notifier->setEnabled(false);
+         delete m_notifier;
+         m_notifier = NULL;
+         mainWin->set_status(_("Director disconnected."));
+         QApplication::restoreOverrideCursor();
+         stat = BNET_HARDEOF;
+      }
+      break;
    } 
-   return BNET_HARDEOF;
+   return stat;
 }
 
 /* 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 (!m_sock) {
-      return;
-   }
-   stat = read();
-   if (stat >= 0) {
-      if (m_at_prompt) {
-         display_text("\n");
-         m_at_prompt = false;
-      }
+   if (mainWin->m_commDebug) Pmsg0(000, "read_dir\n");
+   while (read() >= 0) {
       display_text(msg());
-      return;
    }
-   if (is_bnet_stop(m_sock)) {         /* error or term request */
-      m_sock->close();
-      m_sock = NULL;
-      mainWin->actionConnect->setIcon(QIcon(QString::fromUtf8("images/disconnected.png")));
-      QBrush redBrush(Qt::red);
-      m_consoleItem->setForeground(0, redBrush);
-      m_notifier->setEnabled(false);
-      delete m_notifier;
-      m_notifier = NULL;
-      mainWin->set_status(_("Director disconnected."));
-      QApplication::restoreOverrideCursor();
-      return;
+}
+
+/*
+ * When the notifier is enabled, read_dir() will automatically be
+ * called by the Qt event loop when ever there is any output 
+ * from the Directory, and read_dir() will then display it on
+ * the console.
+ *
+ * When we are in a bat dialog, we want to control *all* output
+ * from the Directory, so we set notify to off.
+ *    m_console->notifiy(false);
+ */
+void Console::notify(bool enable) 
+{ 
+   m_notifier->setEnabled(enable);   
+}
+
+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;
    }
-   /* Must be a signal -- either do something or ignore it */
-   if (m_sock->msglen == BNET_PROMPT) {
-      m_at_prompt = true;
-      mainWin->set_status(_("At prompt waiting for input ..."));
-      update_cursor();
+}
+
+/*
+ * 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;
    }
-   if (m_sock->msglen == BNET_EOD) {
-      mainWin->set_status_ready();
-      update_cursor();
+}
+
+/*
+ * 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);
    }
-   return;
+# 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);
 }