]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/lib/message.c
Fix conio.h problem on Solaris
[bacula/bacula] / bacula / src / lib / message.c
index f9742687492cf20d35103af4ba4f98775ca61f3f..8519389e9f2c1560bc9f60cbf7dc4a0f1e09c004 100755 (executable)
 #include "bacula.h"
 #include "jcr.h"
 
+#if !defined(HAVE_CONSOLE)
 #if defined(HAVE_CYGWIN) || defined(HAVE_WIN32)
 #include <windows.h>
 #endif
+#endif
 
 #define FULL_LOCATION 1              /* set for file:line in Debug messages */
 
@@ -41,7 +43,7 @@
  *  This is where we define "Globals" because all the
  *    daemons include this file.
  */
-char *working_directory = NULL;       /* working directory path stored here */
+const char *working_directory = NULL;      /* working directory path stored here */
 int verbose = 0;                     /* increase User messages */
 int debug_level = 0;                 /* debug level */
 time_t daemon_start_time = 0;        /* Daemon start time */
@@ -87,6 +89,9 @@ static bool trace = false;
 
 static MSGS *daemon_msgs;             /* global messages */
 
+/* Define if e_msg must exit when M_ERROR_TERM is received */
+static int exit_on_error = 1;
+
 /* 
  * Set daemon name. Also, find canonical execution
  *  path.  Note, exepath has spare room for tacking on
@@ -203,7 +208,9 @@ init_msg(JCR *jcr, MSGS *msg)
       daemon_msgs = (MSGS *)malloc(sizeof(MSGS));
       memset(daemon_msgs, 0, sizeof(MSGS));
       for (i=1; i<=M_MAX; i++) {
+#ifndef WIN32
         add_msg_dest(daemon_msgs, MD_STDOUT, i, NULL, NULL);
+#endif
         add_msg_dest(daemon_msgs, MD_SYSLOG, i, NULL, NULL);
       }
       Dmsg1(050, "Create daemon global message resource 0x%x\n", daemon_msgs);
@@ -251,7 +258,7 @@ init_msg(JCR *jcr, MSGS *msg)
 /* Initialize so that the console (User Agent) can
  * receive messages -- stored in a file.
  */
-void init_console_msg(char *wd)
+void init_console_msg(const char *wd)
 {
    int fd;
 
@@ -347,7 +354,7 @@ void rem_msg_dest(MSGS *msg, int dest_code, int msg_type, char *where)
 /*
  * Create a unique filename for the mail command
  */
-static void make_unique_mail_filename(JCR *jcr, POOLMEM **name, DEST *d)
+static void make_unique_mail_filename(JCR *jcr, POOLMEM *&name, DEST *d)
 {
    if (jcr) {
       Mmsg(name, "%s/%s.mail.%s.%d", working_directory, my_name,
@@ -362,21 +369,34 @@ static void make_unique_mail_filename(JCR *jcr, POOLMEM **name, DEST *d)
 /*
  * Open a mail pipe
  */
-static BPIPE *open_mail_pipe(JCR *jcr, POOLMEM **cmd, DEST *d)
+static BPIPE *open_mail_pipe(JCR *jcr, POOLMEM *&cmd, DEST *d)
 {
    BPIPE *bpipe;
-
-   if (d->mail_cmd && jcr) {
-      *cmd = edit_job_codes(jcr, *cmd, d->mail_cmd, d->where);
+   int use_bsmtp = (d->mail_cmd && jcr);
+       
+   if (use_bsmtp) {
+      cmd = edit_job_codes(jcr, cmd, d->mail_cmd, d->where);
    } else {
+#if 1
+      Mmsg(cmd, "/usr/lib/sendmail -F Bacula %s", d->where);
+#else
       Mmsg(cmd, "mail -s \"Bacula Message\" %s", d->where);
+#endif
    }
    fflush(stdout);
 
-   if (!(bpipe = open_bpipe(*cmd, 120, "rw"))) {
+   if (!(bpipe = open_bpipe(cmd, 120, "rw"))) {
+      berrno be;
       Jmsg(jcr, M_ERROR, 0, "open mail pipe %s failed: ERR=%s\n", 
-        *cmd, strerror(errno));
-   } 
+        cmd, be.strerror());
+   }
+
+#if 1
+   if (!use_bsmtp) {
+       fprintf(bpipe->wfd, "Subject: Bacula Message\r\n\r\n");
+   }
+#endif
+   
    return bpipe;
 }
 
@@ -425,7 +445,7 @@ void close_msg(JCR *jcr)
               goto rem_temp_file;
            }
            
-           if (!(bpipe=open_mail_pipe(jcr, &cmd, d))) {
+           if (!(bpipe=open_mail_pipe(jcr, cmd, d))) {
                Pmsg0(000, "open mail pipe failed.\n");
               goto rem_temp_file;
            }
@@ -454,10 +474,12 @@ void close_msg(JCR *jcr)
 
            stat = close_bpipe(bpipe);
            if (stat != 0 && msgs != daemon_msgs) {
+              berrno be;
+              be.set_errno(stat);
                Dmsg1(150, "Calling emsg. CMD=%s\n", cmd);
-               Jmsg3(jcr, M_ERROR, 0, _("Mail program terminated in error. stat=%d\n"
+               Jmsg2(jcr, M_ERROR, 0, _("Mail program terminated in error.\n"
                                         "CMD=%s\n"
-                                        "ERR=%s\n"), stat, cmd, strerror(stat));
+                                        "ERR=%s\n"), cmd, be.strerror());
            }
            free_memory(line);
 rem_temp_file:
@@ -558,10 +580,17 @@ void dispatch_message(JCR *jcr, int type, int level, char *msg)
     Dmsg2(800, "Enter dispatch_msg type=%d msg=%s\n", type, msg);
 
     if (type == M_ABORT || type == M_ERROR_TERM) {
+#ifndef HAVE_WIN32
        fputs(msg, stdout);        /* print this here to INSURE that it is printed */
        fflush(stdout);
+#endif
+#if !defined(HAVE_CONSOLE)
 #if defined(HAVE_CYGWIN) || defined(HAVE_WIN32)
-       MessageBox(NULL, msg, "Bacula", MB_OK);
+       /* If we don't exit on error, error messages are parsed by UA */
+       if (exit_on_error) {
+          MessageBox(NULL, msg, "Bacula", MB_OK);
+       }
+#endif
 #endif
     }
 
@@ -613,15 +642,17 @@ void dispatch_message(JCR *jcr, int type, int level, char *msg)
             case MD_OPERATOR:
                 Dmsg1(800, "OPERATOR for collowing msg: %s\n", msg);
                mcmd = get_pool_memory(PM_MESSAGE);
-               if ((bpipe=open_mail_pipe(jcr, &mcmd, d))) {
+               if ((bpipe=open_mail_pipe(jcr, mcmd, d))) {
                   int stat;
                   fputs(msg, bpipe->wfd);
                   /* Messages to the operator go one at a time */
                   stat = close_bpipe(bpipe);
                   if (stat != 0) {
+                     berrno be;
+                     be.set_errno(stat);
                       Jmsg2(jcr, M_ERROR, 0, _("Operator mail program terminated in error.\n"
                             "CMD=%s\n"
-                            "ERR=%s\n"), mcmd, strerror(stat));
+                            "ERR=%s\n"), mcmd, be.strerror());
                   }
                }
                free_pool_memory(mcmd);
@@ -631,7 +662,7 @@ void dispatch_message(JCR *jcr, int type, int level, char *msg)
                 Dmsg1(800, "MAIL for following msg: %s", msg);
                if (!d->fd) {
                   POOLMEM *name = get_pool_memory(PM_MESSAGE);
-                  make_unique_mail_filename(jcr, &mp_chr(name), d);
+                  make_unique_mail_filename(jcr, name, d);
                    d->fd = fopen(mp_chr(name), "w+");
                   if (!d->fd) {
                      d->fd = stdout;
@@ -728,7 +759,7 @@ d_msg(const char *file, int line, int level, const char *fmt,...)
          /* visual studio passes the whole path to the file as well
           * which makes for very long lines
           */
-          char *f = strrchr(file, '\\');
+          const char *f = strrchr(file, '\\');
          if (f) file = f + 1;
           len = bsnprintf(buf, sizeof(buf), "%s: %s:%d ", my_name, file, line);
        } else {
@@ -922,7 +953,7 @@ e_msg(const char *file, int line, int type, int level, const char *fmt,...)
        char *p = 0;
        p[0] = 0;                     /* generate segmentation violation */
     }
-    if (type == M_ERROR_TERM) {
+    if ((type == M_ERROR_TERM) && exit_on_error) {
        exit(1);
     }
 }
@@ -1019,7 +1050,7 @@ Jmsg(JCR *jcr, int type, int level, const char *fmt,...)
        char *p = 0;
        p[0] = 0;                     /* generate segmentation violation */
     }
-    if (type == M_ERROR_TERM) {
+    if ((type == M_ERROR_TERM) && exit_on_error) {
        exit(1);
     }
 }
@@ -1035,14 +1066,14 @@ void j_msg(const char *file, int line, JCR *jcr, int type, int level, const char
    POOLMEM *pool_buf;
 
    pool_buf = get_pool_memory(PM_EMSG);
-   i = Mmsg(&pool_buf, "%s:%d ", file, line);
+   i = Mmsg(pool_buf, "%s:%d ", file, line);
 
    for (;;) {
       maxlen = sizeof_pool_memory(pool_buf) - i - 1; 
       va_start(arg_ptr, fmt);
       len = bvsnprintf(pool_buf+i, maxlen, fmt, arg_ptr);
       va_end(arg_ptr);
-      if (len < 0 || len >= maxlen) {
+      if (len < 0 || len >= (maxlen-5)) {
         pool_buf = realloc_pool_memory(pool_buf, maxlen + i + maxlen/2);
         continue;
       }
@@ -1056,7 +1087,7 @@ void j_msg(const char *file, int line, JCR *jcr, int type, int level, const char
 
 /*
  * Edit a message into a Pool memory buffer, with file:lineno
- */
+ */                                                 
 int m_msg(const char *file, int line, POOLMEM **pool_buf, const char *fmt, ...)
 {
    va_list   arg_ptr;
@@ -1069,7 +1100,7 @@ int m_msg(const char *file, int line, POOLMEM **pool_buf, const char *fmt, ...)
       va_start(arg_ptr, fmt);
       len = bvsnprintf(*pool_buf+i, maxlen, fmt, arg_ptr);
       va_end(arg_ptr);
-      if (len < 0 || len >= maxlen) {
+      if (len < 0 || len >= (maxlen-5)) {
         *pool_buf = realloc_pool_memory(*pool_buf, maxlen + i + maxlen/2);
         continue;
       }
@@ -1078,6 +1109,28 @@ int m_msg(const char *file, int line, POOLMEM **pool_buf, const char *fmt, ...)
    return len;
 }
 
+int m_msg(const char *file, int line, POOLMEM *&pool_buf, const char *fmt, ...)
+{
+   va_list   arg_ptr;
+   int i, len, maxlen;
+
+   i = sprintf(pool_buf, "%s:%d ", file, line);
+
+   for (;;) {
+      maxlen = sizeof_pool_memory(pool_buf) - i - 1; 
+      va_start(arg_ptr, fmt);
+      len = bvsnprintf(pool_buf+i, maxlen, fmt, arg_ptr);
+      va_end(arg_ptr);
+      if (len < 0 || len >= (maxlen-5)) {
+        pool_buf = realloc_pool_memory(pool_buf, maxlen + i + maxlen/2);
+        continue;
+      }
+      break;
+   }
+   return len;
+}
+
+
 /*
  * Edit a message into a Pool Memory buffer NO file:lineno
  *  Returns: string length of what was edited.
@@ -1092,7 +1145,7 @@ int Mmsg(POOLMEM **pool_buf, const char *fmt, ...)
       va_start(arg_ptr, fmt);
       len = bvsnprintf(*pool_buf, maxlen, fmt, arg_ptr);
       va_end(arg_ptr);
-      if (len < 0 || len >= maxlen) {
+      if (len < 0 || len >= (maxlen-5)) {
         *pool_buf = realloc_pool_memory(*pool_buf, maxlen + maxlen/2);
         continue;
       }
@@ -1101,6 +1154,26 @@ int Mmsg(POOLMEM **pool_buf, const char *fmt, ...)
    return len;
 }
 
+int Mmsg(POOLMEM *&pool_buf, const char *fmt, ...)
+{
+   va_list   arg_ptr;
+   int len, maxlen;
+
+   for (;;) {
+      maxlen = sizeof_pool_memory(pool_buf) - 1; 
+      va_start(arg_ptr, fmt);
+      len = bvsnprintf(pool_buf, maxlen, fmt, arg_ptr);
+      va_end(arg_ptr);
+      if (len < 0 || len >= (maxlen-5)) {
+        pool_buf = realloc_pool_memory(pool_buf, maxlen + maxlen/2);
+        continue;
+      }
+      break;
+   }
+   return len;
+}
+
+
 static pthread_mutex_t msg_queue_mutex = PTHREAD_MUTEX_INITIALIZER;
 
 /*
@@ -1124,7 +1197,7 @@ void Qmsg(JCR *jcr, int type, int level, const char *fmt,...)
       va_start(arg_ptr, fmt);
       len = bvsnprintf(pool_buf, maxlen, fmt, arg_ptr);
       va_end(arg_ptr);
-      if (len < 0 || len >= maxlen) {
+      if (len < 0 || len >= (maxlen-5)) {
         pool_buf = realloc_pool_memory(pool_buf, maxlen + maxlen/2);
         continue;
       }
@@ -1143,6 +1216,7 @@ void Qmsg(JCR *jcr, int type, int level, const char *fmt,...)
    } else {
       /* Queue message for later sending */
       jcr->msg_queue->append(item);
+//    Dmsg1(000, "queue item=%lu\n", (long unsigned)item);
    }
    V(msg_queue_mutex);
    free_memory(pool_buf);
@@ -1157,12 +1231,13 @@ void dequeue_messages(JCR *jcr)
    P(msg_queue_mutex);
    jcr->dequeuing = true;
    foreach_dlist(item, jcr->msg_queue) {
+//    Dmsg1(000, "dequeue item=%lu\n", (long unsigned)item);
       Jmsg(jcr, item->type, item->level, "%s", item->msg);
    }
    jcr->msg_queue->destroy();
    jcr->dequeuing = false;
    V(msg_queue_mutex);
-}
+}                                                
 
 
 /*
@@ -1176,14 +1251,14 @@ void q_msg(const char *file, int line, JCR *jcr, int type, int level, const char
    POOLMEM *pool_buf;
 
    pool_buf = get_pool_memory(PM_EMSG);
-   i = Mmsg(&pool_buf, "%s:%d ", file, line);
+   i = Mmsg(pool_buf, "%s:%d ", file, line);
 
    for (;;) {
       maxlen = sizeof_pool_memory(pool_buf) - i - 1; 
       va_start(arg_ptr, fmt);
       len = bvsnprintf(pool_buf+i, maxlen, fmt, arg_ptr);
       va_end(arg_ptr);
-      if (len < 0 || len >= maxlen) {
+      if (len < 0 || len >= (maxlen-5)) {
         pool_buf = realloc_pool_memory(pool_buf, maxlen + i + maxlen/2);
         continue;
       }
@@ -1193,3 +1268,10 @@ void q_msg(const char *file, int line, JCR *jcr, int type, int level, const char
    Qmsg(jcr, type, level, "%s", pool_buf);
    free_memory(pool_buf);
 }
+
+/* 
+ * Define if e_msg must exit when M_ERROR_TERM is received
+ */
+void set_exit_on_error(int value) {
+   exit_on_error = value;
+}