]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/lib/message.c
Improve debug output scheduler+jobq+tracing
[bacula/bacula] / bacula / src / lib / message.c
index 91701359301c36968f94128e74444ea1dd95c686..08ba4dd835438bf5f0841cc074342db0d120cbec 100755 (executable)
@@ -41,7 +41,7 @@ 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 */
-char *version = VERSION " (" BDATE ")";
+const char *version = VERSION " (" BDATE ")";
 char my_name[30];                    /* daemon name is stored here */
 char *exepath = (char *)NULL;
 char *exename = (char *)NULL;
@@ -49,7 +49,6 @@ int console_msg_pending = 0;
 char con_fname[500];                 /* Console filename */
 FILE *con_fd = NULL;                 /* Console file descriptor */
 brwlock_t con_lock;                  /* Console lock structure */
-FILE *trace_fd = NULL;
 
 #ifdef HAVE_POSTGRESQL
 char catalog_db[] = "PostgreSQL";
@@ -65,9 +64,15 @@ char catalog_db[] = "Internal";
 #endif
 #endif
 
-char *host_os = HOST_OS;
-char *distname = DISTNAME;
-char *distver = DISTVER;
+const char *host_os = HOST_OS;
+const char *distname = DISTNAME;
+const char *distver = DISTVER;
+static FILE *trace_fd = NULL;
+#ifdef HAVE_WIN32
+static bool trace = true;
+#else
+static bool trace = false;
+#endif
 
 /* Forward referenced functions */
 
@@ -89,7 +94,7 @@ static MSGS *daemon_msgs;            /* global messages */
  *  argv is NULL to avoid doing the path code twice.
  */
 #define BTRACE_EXTRA 20
-void my_name_is(int argc, char *argv[], char *name)
+void my_name_is(int argc, char *argv[], const char *name)
 {
    char *l, *p, *q;
    char cpath[400], npath[400];
@@ -107,7 +112,7 @@ void my_name_is(int argc, char *argv[], char *name)
         l++;
       } else {
         l = argv[0];
-#ifdef HAVE_CYGWIN
+#if defined(HAVE_CYGWIN) || defined(HAVE_WIN32)
         /* On Windows allow c: junk */
          if (l[1] == ':') {
            l += 2;
@@ -162,12 +167,13 @@ void
 init_msg(JCR *jcr, MSGS *msg)
 {
    DEST *d, *dnew, *temp_chain = NULL;
-   int i, fd;
+   int i;
 
    if (jcr == NULL && msg == NULL) {
       init_last_jobs_list();
    }
 
+#ifndef HAVE_WIN32
    /*
     * Make sure we have fd's 0, 1, 2 open
     *  If we don't do this one of our sockets may open
@@ -175,6 +181,7 @@ init_msg(JCR *jcr, MSGS *msg)
     *  send total garbage to our socket.
     *
     */
+   int fd;
    fd = open("/dev/null", O_RDONLY, 0644);
    if (fd > 2) {
       close(fd);
@@ -184,7 +191,7 @@ init_msg(JCR *jcr, MSGS *msg)
       }
    }
 
-
+#endif
    /*
     * If msg is NULL, initialize global chain for STDOUT and syslog
     */
@@ -581,6 +588,9 @@ void dispatch_message(JCR *jcr, int type, int level, char *msg)
 
     if (type == M_ABORT || type == M_ERROR_TERM) {
        fputs(msg, stdout);        /* print this here to INSURE that it is printed */
+#if defined(HAVE_CYGWIN) || defined(HAVE_WIN32)
+       MessageBox(NULL, msg, "Bacula", MB_OK);
+#endif
     }
 
     /* Now figure out where to send the message */
@@ -728,7 +738,7 @@ void dispatch_message(JCR *jcr, int type, int level, char *msg)
  *  are not printed.
  */
 void 
-d_msg(char *file, int line, int level, char *fmt,...)
+d_msg(const char *file, int line, int level, const char *fmt,...)
 {
     char      buf[5000];
     int       len;
@@ -741,18 +751,13 @@ d_msg(char *file, int line, int level, char *fmt,...)
     }
 
     if (level <= debug_level) {
-#ifdef SEND_DMSG_TO_FILE
-       if (!trace_fd) {
-          bsnprintf(buf, sizeof(buf), "%s/bacula.trace", working_directory);
-          trace_fd = fopen(buf, "a+");
-         if (!trace_fd) {
-             Emsg2(M_ABORT, 0, _("Cannot open trace file \"%s\": ERR=%s\n"),
-                 buf, strerror(errno));
-         }
-       }
-#endif
 #ifdef FULL_LOCATION
        if (details) {
+         /* visual studio passes the whole path to the file as well
+          * which makes for very long lines
+          */
+          char *f = strrchr(file, '\\');
+         if (f) file = f + 1;
           len = bsnprintf(buf, sizeof(buf), "%s: %s:%d ", my_name, file, line);
        } else {
          len = 0;
@@ -764,15 +769,44 @@ d_msg(char *file, int line, int level, char *fmt,...)
        bvsnprintf(buf+len, sizeof(buf)-len, (char *)fmt, arg_ptr);
        va_end(arg_ptr);
 
-#ifdef SEND_DMSG_TO_FILE
-       fputs(buf, trace_fd);
-       fflush(trace_fd);
-#else
-       fputs(buf, stdout);
-#endif
+       /* 
+        * Used the "trace on" command in the console to turn on
+        *  output to the trace file.  "trace off" will close the file.
+       */
+       if (trace) {
+         if (!trace_fd) {
+             bsnprintf(buf, sizeof(buf), "%s/bacula.trace", working_directory ? working_directory : ".");
+             trace_fd = fopen(buf, "a+");
+         }
+         if (trace_fd) {
+            fputs(buf, trace_fd);
+            fflush(trace_fd);
+         }
+       } else {   /* not tracing */
+         fputs(buf, stdout);
+       }
     }
 }
 
+/*
+ * Set trace flag on/off. If argument is negative, there is no change 
+ */
+void set_trace(int trace_flag)
+{
+   if (trace_flag < 0) {
+      return;
+   } else if (trace_flag > 0) {
+      trace = true;
+   } else {
+      trace = false;
+   }
+   if (!trace && trace_fd) {
+      FILE *ltrace_fd = trace_fd;
+      trace_fd = NULL;
+      bmicrosleep(0, 100000);        /* yield to prevent seg faults */
+      fclose(ltrace_fd);
+   }
+}
 
 /*********************************************************************
  *
@@ -782,7 +816,7 @@ d_msg(char *file, int line, int level, char *fmt,...)
  *  are not printed.
  */
 void 
-p_msg(char *file, int line, int level, char *fmt,...)
+p_msg(const char *file, int line, int level, const char *fmt,...)
 {
     char      buf[5000];
     int       len;
@@ -815,7 +849,7 @@ p_msg(char *file, int line, int level, char *fmt,...)
  *  are not printed.
  */
 void 
-t_msg(char *file, int line, int level, char *fmt,...)
+t_msg(const char *file, int line, int level, const char *fmt,...)
 {
     char      buf[5000];
     int       len;
@@ -831,10 +865,6 @@ t_msg(char *file, int line, int level, char *fmt,...)
        if (!trace_fd) {
           bsnprintf(buf, sizeof(buf), "%s/bacula.trace", working_directory);
           trace_fd = fopen(buf, "a+");
-         if (!trace_fd) {
-             Emsg2(M_ABORT, 0, _("Cannot open trace file \"%s\": ERR=%s\n"),
-                 buf, strerror(errno));
-         }
        }
     
 #ifdef FULL_LOCATION
@@ -849,10 +879,11 @@ t_msg(char *file, int line, int level, char *fmt,...)
        va_start(arg_ptr, fmt);
        bvsnprintf(buf+len, sizeof(buf)-len, (char *)fmt, arg_ptr);
        va_end(arg_ptr);
-
-       fputs(buf, trace_fd);
-       fflush(trace_fd);
-    }
+       if (trace_fd != NULL) {
+          fputs(buf, trace_fd);
+          fflush(trace_fd);
+       }
+   }
 }
 
 
@@ -863,7 +894,7 @@ t_msg(char *file, int line, int level, char *fmt,...)
  *
  */
 void 
-e_msg(char *file, int line, int type, int level, char *fmt,...)
+e_msg(const char *file, int line, int type, int level, const char *fmt,...)
 {
     char     buf[5000];
     va_list   arg_ptr;
@@ -930,13 +961,13 @@ e_msg(char *file, int line, int type, int level, char *fmt,...)
  *
  */
 void 
-Jmsg(JCR *jcr, int type, int level, char *fmt,...)
+Jmsg(JCR *jcr, int type, int level, const char *fmt,...)
 {
     char     rbuf[5000];
     va_list   arg_ptr;
     int len;
     MSGS *msgs;
-    char *job;
+    const char *job;
 
     
     Dmsg1(800, "Enter Jmsg type=%d\n", type);
@@ -1021,10 +1052,38 @@ Jmsg(JCR *jcr, int type, int level, char *fmt,...)
     }
 }
 
+/*
+ * If we come here, prefix the message with the file:line-number,
+ *  then pass it on to the normal Jmsg routine.
+ */
+void j_msg(const char *file, int line, JCR *jcr, int type, int level, const char *fmt,...)
+{
+   va_list   arg_ptr;
+   int i, len, maxlen;
+   POOLMEM *pool_buf;
+
+   pool_buf = get_pool_memory(PM_EMSG);
+   i = Mmsg(&pool_buf, "%s:%d ", file, line);
+
+again:
+   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) {
+      pool_buf = realloc_pool_memory(pool_buf, maxlen + i + maxlen/2);
+      goto again;
+   }
+
+   Jmsg(jcr, type, level, "%s", pool_buf);
+   free_memory(pool_buf);
+}
+
+
 /*
  * Edit a message into a Pool memory buffer, with file:lineno
  */
-int m_msg(char *file, int line, POOLMEM **pool_buf, char *fmt, ...)
+int m_msg(const char *file, int line, POOLMEM **pool_buf, const char *fmt, ...)
 {
    va_list   arg_ptr;
    int i, len, maxlen;
@@ -1047,7 +1106,7 @@ again:
  * Edit a message into a Pool Memory buffer NO file:lineno
  *  Returns: string length of what was edited.
  */
-int Mmsg(POOLMEM **pool_buf, char *fmt, ...)
+int Mmsg(POOLMEM **pool_buf, const char *fmt, ...)
 {
    va_list   arg_ptr;
    int len, maxlen;
@@ -1064,12 +1123,66 @@ again:
    return len;
 }
 
+static pthread_mutex_t msg_queue_mutex = PTHREAD_MUTEX_INITIALIZER;
+
+/*
+ * We queue messages rather than print them directly. This
+ *  is generally used in low level routines (msg handler, bnet)
+ *  to prevent recursion.
+ */
+void Qmsg(JCR *jcr, int type, int level, const char *fmt,...)
+{
+   va_list   arg_ptr;
+   int len, maxlen;
+   POOLMEM *pool_buf;
+   MQUEUE_ITEM *item;
+
+   if (jcr->dequeuing) {             /* do not allow recursion */
+      return;
+   }
+   pool_buf = get_pool_memory(PM_EMSG);
+
+again:
+   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) {
+      pool_buf = realloc_pool_memory(pool_buf, maxlen + maxlen/2);
+      goto again;
+   }
+   P(msg_queue_mutex);
+   item = (MQUEUE_ITEM *)malloc(sizeof(MQUEUE_ITEM) + strlen(pool_buf) + 1);
+   item->type = type;
+   item->level = level;
+   strcpy(item->msg, pool_buf);  
+   jcr->msg_queue->append(item);
+   V(msg_queue_mutex);
+   free_memory(pool_buf);
+}
+
+/*
+ * Dequeue messages 
+ */
+void dequeue_messages(JCR *jcr)
+{
+   MQUEUE_ITEM *item;
+   P(msg_queue_mutex);
+   jcr->dequeuing = true;
+   foreach_dlist(item, jcr->msg_queue) {
+      Jmsg(jcr, item->type, item->level, "%s", item->msg);
+   }
+   jcr->msg_queue->destroy();
+   jcr->dequeuing = false;
+   V(msg_queue_mutex);
+}
+
 
 /*
  * If we come here, prefix the message with the file:line-number,
- *  then pass it on to the normal Jmsg routine.
+ *  then pass it on to the normal Qmsg routine.
  */
-void j_msg(char *file, int line, JCR *jcr, int type, int level, char *fmt,...)
+void q_msg(const char *file, int line, JCR *jcr, int type, int level, const char *fmt,...)
 {
    va_list   arg_ptr;
    int i, len, maxlen;
@@ -1088,6 +1201,6 @@ again:
       goto again;
    }
 
-   Jmsg(jcr, type, level, "%s", pool_buf);
+   Qmsg(jcr, type, level, "%s", pool_buf);
    free_memory(pool_buf);
 }