Fix ndrdump to use a common setup_logging() API
[ira/wip.git] / source3 / lib / debug.c
index 0836dbe7900cb642a4a7ac3f9a39b2b7fcdfda6a..e7dcfb4fdf3f6b2e887224a7055f00612f5016b1 100644 (file)
@@ -7,7 +7,7 @@
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
 
    This program is distributed in the hope that it will be useful,
@@ -16,8 +16,7 @@
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
@@ -30,7 +29,8 @@
  *                    for a terminating null byte.
  */
 
-#define FORMAT_BUFR_MAX ( sizeof( format_bufr ) - 1 )
+#define FORMAT_BUFR_SIZE 1024
+#define FORMAT_BUFR_MAX (FORMAT_BUFR_SIZE - 1)
 
 /* -------------------------------------------------------------------------- **
  * This module implements Samba's debugging utility.
  */
 
 XFILE   *dbf        = NULL;
-pstring debugf     = "";
-BOOL    debug_warn_unknown_class = True;
-BOOL    debug_auto_add_unknown_class = True;
-BOOL    AllowDebugChange = True;
+static char *debugf = NULL;
+bool    debug_warn_unknown_class = True;
+bool    debug_auto_add_unknown_class = True;
+bool    AllowDebugChange = True;
+
+/*
+   used to check if the user specified a
+   logfile on the command line
+*/
+bool    override_logfile;
+
+static TALLOC_CTX *tmp_debug_ctx;
 
 /*
  * This is to allow assignment to DEBUGLEVEL before the debug
- * system has been initialised.
+ * system has been initialized.
  */
 static int debug_all_class_hack = 1;
-static BOOL debug_all_class_isset_hack = True;
+static bool debug_all_class_isset_hack = True;
 
 static int debug_num_classes = 0;
 int     *DEBUGLEVEL_CLASS = &debug_all_class_hack;
-BOOL    *DEBUGLEVEL_CLASS_ISSET = &debug_all_class_isset_hack;
+bool    *DEBUGLEVEL_CLASS_ISSET = &debug_all_class_isset_hack;
 
 /* DEBUGLEVEL is #defined to *debug_level */
 int     DEBUGLEVEL = &debug_all_class_hack;
@@ -126,14 +134,14 @@ int     DEBUGLEVEL = &debug_all_class_hack;
  *                    are unable to open a new log file for some reason.
  */
 
-static BOOL    stdout_logging = False;
+static bool    stdout_logging = False;
 static int     debug_count    = 0;
 #ifdef WITH_SYSLOG
 static int     syslog_level   = 0;
 #endif
-static pstring format_bufr    = { '\0' };
+static char *format_bufr = NULL;
 static size_t     format_pos     = 0;
-static BOOL    log_overflow   = False;
+static bool    log_overflow   = False;
 
 /*
  * Define all the debug class selection names here. Names *MUST NOT* contain 
@@ -154,6 +162,13 @@ static const char *default_classname_table[] = {
        "auth",              /* DBGC_AUTH         */
        "winbind",           /* DBGC_WINBIND      */
        "vfs",               /* DBGC_VFS          */
+       "idmap",             /* DBGC_IDMAP        */
+       "quota",             /* DBGC_QUOTA        */
+       "acls",              /* DBGC_ACLS         */
+       "locking",           /* DBGC_LOCKING      */
+       "msdfs",             /* DBGC_MSDFS        */
+       "dmapi",             /* DBGC_DMAPI        */
+       "registry",          /* DBGC_REGISTRY     */
        NULL
 };
 
@@ -164,6 +179,41 @@ static char **classname_table = NULL;
  * Functions...
  */
 
+/***************************************************************************
+ Free memory pointed to by global pointers.
+****************************************************************************/
+
+static bool initialized;
+
+void gfree_debugsyms(void)
+{
+       int i;
+
+       if ( classname_table ) {
+               for ( i = 0; i < debug_num_classes; i++ ) {
+                       SAFE_FREE( classname_table[i] );
+               }
+               SAFE_FREE( classname_table );
+       }
+
+       if ( DEBUGLEVEL_CLASS != &debug_all_class_hack ) {
+               SAFE_FREE( DEBUGLEVEL_CLASS );
+               DEBUGLEVEL_CLASS = &debug_all_class_hack;
+       }
+
+       if ( DEBUGLEVEL_CLASS_ISSET != &debug_all_class_isset_hack ) {
+               SAFE_FREE( DEBUGLEVEL_CLASS_ISSET );
+               DEBUGLEVEL_CLASS_ISSET = &debug_all_class_isset_hack;
+       }
+
+       SAFE_FREE(format_bufr);
+
+       debug_num_classes = 0;
+
+       debug_level = DEBUGLEVEL_CLASS;
+
+       initialized = false;
+}
 
 /****************************************************************************
 utility lists registered debug class names's
@@ -177,14 +227,16 @@ static char *debug_list_class_names_and_levels(void)
        char **list;
        char *buf = NULL;
        char *b;
-       BOOL err = False;
+       bool err = False;
 
-       if (DEBUGLEVEL_CLASS == &debug_all_class_hack)
+       if (DEBUGLEVEL_CLASS == &debug_all_class_hack) {
                return NULL;
+       }
 
-       list = calloc(debug_num_classes + 1, sizeof(char *));
-       if (!list)
+       list = SMB_CALLOC_ARRAY(char *, debug_num_classes + 1);
+       if (!list) {
                return NULL;
+       }
 
        /* prepare strings */
        for (i = 0, dim = 0; i < debug_num_classes; i++) {
@@ -199,8 +251,8 @@ static char *debug_list_class_names_and_levels(void)
                dim += l;
        }
 
-       /* create single string list */
-       b = buf = malloc(dim);
+       /* create single string list - add space for newline */
+       b = buf = (char *)SMB_MALLOC(dim+1);
        if (!buf) {
                err = True;
                goto done;
@@ -210,17 +262,17 @@ static char *debug_list_class_names_and_levels(void)
                strncpy(b, list[i], l);
                b = b + l;
        }
-       b[-1] = '\0';
+       b[-1] = '\n'; /* replace last space with newline */
+       b[0] = '\0';  /* null terminate string */
 
 done:
        /* free strings list */
-       for (i = 0; i < debug_num_classes; i++)
-               if (list[i]) free(list[i]);
-       free(list);
+       for (i = 0; i < debug_num_classes; i++) {
+               SAFE_FREE(list[i]);
+       }
+       SAFE_FREE(list);
 
        if (err) {
-               if (buf)
-                       free(buf);
                return NULL;
        } else {
                return buf;
@@ -228,8 +280,9 @@ done:
 }
 
 /****************************************************************************
-utility access to debug class names's
+ Utility access to debug class names's.
 ****************************************************************************/
+
 const char *debug_classname_from_index(int ndx)
 {
        if (ndx < 0 || ndx >= debug_num_classes)
@@ -239,8 +292,9 @@ const char *debug_classname_from_index(int ndx)
 }
 
 /****************************************************************************
-utility to translate names to debug class index's (internal version)
+ Utility to translate names to debug class index's (internal version).
 ****************************************************************************/
+
 static int debug_lookup_classname_int(const char* classname)
 {
        int i;
@@ -255,8 +309,9 @@ static int debug_lookup_classname_int(const char* classname)
 }
 
 /****************************************************************************
-Add a new debug class to the system
+ Add a new debug class to the system.
 ****************************************************************************/
+
 int debug_add_class(const char *classname)
 {
        int ndx;
@@ -274,45 +329,39 @@ int debug_add_class(const char *classname)
        ndx = debug_num_classes;
 
        new_ptr = DEBUGLEVEL_CLASS;
-       if (DEBUGLEVEL_CLASS == &debug_all_class_hack)
-       {
+       if (DEBUGLEVEL_CLASS == &debug_all_class_hack) {
                /* Initial loading... */
                new_ptr = NULL;
        }
-       new_ptr = Realloc(new_ptr,
-                         sizeof(int) * (debug_num_classes + 1));
+       new_ptr = SMB_REALLOC_ARRAY(new_ptr, int, debug_num_classes + 1);
        if (!new_ptr)
                return -1;
-       DEBUGLEVEL_CLASS = new_ptr;
+       DEBUGLEVEL_CLASS = (int *)new_ptr;
        DEBUGLEVEL_CLASS[ndx] = 0;
 
        /* debug_level is the pointer used for the DEBUGLEVEL-thingy */
-       if (ndx==0)
-       {
+       if (ndx==0) {
                /* Transfer the initial level from debug_all_class_hack */
                DEBUGLEVEL_CLASS[ndx] = DEBUGLEVEL;
        }
        debug_level = DEBUGLEVEL_CLASS;
 
        new_ptr = DEBUGLEVEL_CLASS_ISSET;
-       if (new_ptr == &debug_all_class_isset_hack)
-       {
+       if (new_ptr == &debug_all_class_isset_hack) {
                new_ptr = NULL;
        }
-       new_ptr = Realloc(new_ptr,
-                         sizeof(BOOL) * (debug_num_classes + 1));
+       new_ptr = SMB_REALLOC_ARRAY(new_ptr, bool, debug_num_classes + 1);
        if (!new_ptr)
                return -1;
-       DEBUGLEVEL_CLASS_ISSET = new_ptr;
+       DEBUGLEVEL_CLASS_ISSET = (bool *)new_ptr;
        DEBUGLEVEL_CLASS_ISSET[ndx] = False;
 
-       new_ptr = Realloc(classname_table,
-                         sizeof(char *) * (debug_num_classes + 1));
+       new_ptr = SMB_REALLOC_ARRAY(classname_table, char *, debug_num_classes + 1);
        if (!new_ptr)
                return -1;
-       classname_table = new_ptr;
+       classname_table = (char **)new_ptr;
 
-       classname_table[ndx] = strdup(classname);
+       classname_table[ndx] = SMB_STRDUP(classname);
        if (! classname_table[ndx])
                return -1;
        
@@ -322,42 +371,41 @@ int debug_add_class(const char *classname)
 }
 
 /****************************************************************************
-utility to translate names to debug class index's (public version)
+ Utility to translate names to debug class index's (public version).
 ****************************************************************************/
+
 int debug_lookup_classname(const char *classname)
 {
        int ndx;
        
-       if (!classname || !*classname) return -1;
+       if (!classname || !*classname)
+               return -1;
 
        ndx = debug_lookup_classname_int(classname);
 
        if (ndx != -1)
                return ndx;
 
-       if (debug_warn_unknown_class)
-       {
+       if (debug_warn_unknown_class) {
                DEBUG(0, ("debug_lookup_classname(%s): Unknown class\n",
                          classname));
        }
-       if (debug_auto_add_unknown_class)
-       {
+       if (debug_auto_add_unknown_class) {
                return debug_add_class(classname);
        }
        return -1;
 }
 
-
 /****************************************************************************
-dump the current registered debug levels
+ Dump the current registered debug levels.
 ****************************************************************************/
+
 static void debug_dump_status(int level)
 {
        int q;
 
        DEBUG(level, ("INFO: Current debug levels:\n"));
-       for (q = 0; q < debug_num_classes; q++)
-       {
+       for (q = 0; q < debug_num_classes; q++) {
                DEBUGADD(level, ("  %s: %s/%d\n",
                                 classname_table[q],
                                 (DEBUGLEVEL_CLASS_ISSET[q]
@@ -367,10 +415,11 @@ static void debug_dump_status(int level)
 }
 
 /****************************************************************************
-parse the debug levels from smbcontrol. Example debug level parameter:
 printdrivers:7
+ parse the debug levels from smbcontrol. Example debug level parameter:
+ printdrivers:7
 ****************************************************************************/
-static BOOL debug_parse_params(char **params)
+
+static bool debug_parse_params(char **params)
 {
        int   i, ndx;
        char *class_name;
@@ -386,14 +435,15 @@ static BOOL debug_parse_params(char **params)
                DEBUGLEVEL_CLASS[DBGC_ALL] = atoi(params[0]);
                DEBUGLEVEL_CLASS_ISSET[DBGC_ALL] = True;
                i = 1; /* start processing at the next params */
-       }
-       else
+       } else {
                i = 0; /* DBGC_ALL not specified OR class name was included */
+       }
 
        /* Fill in new debug class levels */
        for (; i < debug_num_classes && params[i]; i++) {
-               if ((class_name=strtok(params[i],":")) &&
-                       (class_level=strtok(NULL, "\0")) &&
+               char *saveptr;
+               if ((class_name = strtok_r(params[i],":", &saveptr)) &&
+                       (class_level = strtok_r(NULL, "\0", &saveptr)) &&
             ((ndx = debug_lookup_classname(class_name)) != -1)) {
                                DEBUGLEVEL_CLASS[ndx] = atoi(class_level);
                                DEBUGLEVEL_CLASS_ISSET[ndx] = True;
@@ -407,11 +457,12 @@ static BOOL debug_parse_params(char **params)
 }
 
 /****************************************************************************
-parse the debug levels from smb.conf. Example debug level string:
+ Parse the debug levels from smb.conf. Example debug level string:
   3 tdb:5 printdrivers:7
-Note: the 1st param has no "name:" preceeding it.
+ Note: the 1st param has no "name:" preceeding it.
 ****************************************************************************/
-BOOL debug_parse_levels(const char *params_str)
+
+bool debug_parse_levels(const char *params_str)
 {
        char **params;
 
@@ -421,110 +472,123 @@ BOOL debug_parse_levels(const char *params_str)
        if (AllowDebugChange == False)
                return True;
 
-       params = str_list_make(params_str, NULL);
+       params = str_list_make_v3(talloc_tos(), params_str, NULL);
 
-       if (debug_parse_params(params))
-       {
+       if (debug_parse_params(params)) {
                debug_dump_status(5);
-               str_list_free(&params);
+               TALLOC_FREE(params);
                return True;
        } else {
-               str_list_free(&params);
+               TALLOC_FREE(params);
                return False;
        }
 }
 
 /****************************************************************************
-receive a "set debug level" message
+ Receive a "set debug level" message.
 ****************************************************************************/
-static void debug_message(int msg_type, pid_t src, void *buf, size_t len)
+
+void debug_message(struct messaging_context *msg_ctx,
+                         void *private_data, 
+                         uint32_t msg_type, 
+                         struct server_id src,
+                         DATA_BLOB *data)
 {
-       const char *params_str = buf;
+       const char *params_str = (const char *)data->data;
 
        /* Check, it's a proper string! */
-       if (params_str[len-1] != '\0')
-       {
+       if (params_str[(data->length)-1] != '\0') {
                DEBUG(1, ("Invalid debug message from pid %u to pid %u\n",
-                         (unsigned int)src, (unsigned int)getpid()));
+                         (unsigned int)procid_to_pid(&src),
+                         (unsigned int)getpid()));
                return;
        }
 
        DEBUG(3, ("INFO: Remote set of debug to `%s'  (pid %u from pid %u)\n",
-                 params_str, (unsigned int)getpid(), (unsigned int)src));
+                 params_str, (unsigned int)getpid(),
+                 (unsigned int)procid_to_pid(&src)));
 
        debug_parse_levels(params_str);
 }
 
-
-/****************************************************************************
-send a "set debug level" message
-****************************************************************************/
-void debug_message_send(pid_t pid, const char *params_str)
-{
-       if (!params_str)
-               return;
-       message_send_pid(pid, MSG_DEBUG, params_str, strlen(params_str) + 1,
-                        False);
-}
-
-
 /****************************************************************************
  Return current debug level.
 ****************************************************************************/
 
-static void debuglevel_message(int msg_type, pid_t src, void *buf, size_t len)
+static void debuglevel_message(struct messaging_context *msg_ctx,
+                              void *private_data, 
+                              uint32_t msg_type, 
+                              struct server_id src,
+                              DATA_BLOB *data)
 {
-       char *debug_level_classes;
-       DEBUG(1,("INFO: Received REQ_DEBUGLEVEL message from PID %u\n",(unsigned int)src));
+       char *message = debug_list_class_names_and_levels();
 
-       if ((debug_level_classes = debug_list_class_names_and_levels())) {
-       /*{ debug_level_classes = "test:1000";*/
-               message_send_pid(src, MSG_DEBUGLEVEL, debug_level_classes, strlen(debug_level_classes) + 1, True);
-               SAFE_FREE(debug_level_classes);
-       } else {
-               DEBUG(0, ("debuglevel_message: error retrieving class levels!\n"));
+       if (!message) {
+               DEBUG(0,("debuglevel_message - debug_list_class_names_and_levels returned NULL\n"));
+               return;
        }
+
+       DEBUG(1,("INFO: Received REQ_DEBUGLEVEL message from PID %s\n",
+                procid_str_static(&src)));
+       messaging_send_buf(msg_ctx, src, MSG_DEBUGLEVEL,
+                          (uint8 *)message, strlen(message) + 1);
+
+       SAFE_FREE(message);
 }
 
 /****************************************************************************
 Init debugging (one time stuff)
 ****************************************************************************/
+
 void debug_init(void)
 {
-       static BOOL initialised = False;
        const char **p;
 
-       if (initialised)
+       if (initialized)
                return;
-       
-       initialised = True;
 
-       message_register(MSG_DEBUG, debug_message);
-       message_register(MSG_REQ_DEBUGLEVEL, debuglevel_message);
+       initialized = true;
 
-       for(p = default_classname_table; *p; p++)
-       {
+       for(p = default_classname_table; *p; p++) {
                debug_add_class(*p);
        }
+       format_bufr = (char *)SMB_MALLOC(FORMAT_BUFR_SIZE);
+       if (!format_bufr) {
+               smb_panic("debug_init: unable to create buffer");
+       }
 }
 
+void debug_register_msgs(struct messaging_context *msg_ctx)
+{
+       messaging_register(msg_ctx, NULL, MSG_DEBUG, debug_message);
+       messaging_register(msg_ctx, NULL, MSG_REQ_DEBUGLEVEL,
+                          debuglevel_message);
+}
 
-/* ************************************************************************** **
* get ready for syslog stuff
- * ************************************************************************** **
- */
-void setup_logging(const char *pname, BOOL interactive)
+/***************************************************************************
Get ready for syslog stuff
+**************************************************************************/
+
+void setup_logging(const char *pname, bool interactive)
 {
        debug_init();
 
        /* reset to allow multiple setup calls, going from interactive to
           non-interactive */
        stdout_logging = False;
+       if (dbf) {
+               x_fflush(dbf);
+                if (dbf != x_stdout) {
+                        (void) x_fclose(dbf);
+                }
+       }
+
        dbf = NULL;
 
        if (interactive) {
                stdout_logging = True;
                dbf = x_stdout;
+               x_setbuf( x_stdout, NULL );
        }
 #ifdef WITH_SYSLOG
        else {
@@ -539,41 +603,68 @@ void setup_logging(const char *pname, BOOL interactive)
 #endif
        }
 #endif
-} /* setup_logging */
-
-/* ************************************************************************** **
- * reopen the log files
- * note that we now do this unconditionally
- * We attempt to open the new debug fp before closing the old. This means
- * if we run out of fd's we just keep using the old fd rather than aborting.
- * Fix from dgibson@linuxcare.com.
- * ************************************************************************** **
- */
+}
+
+/**
+   Just run logging to stdout for this program 
+*/
+_PUBLIC_ void setup_logging_stdout(void)
+{
+       setup_logging(NULL, True);
+}
+
 
-BOOL reopen_logs( void )
+/***************************************************************************
+ Set the logfile name.
+**************************************************************************/
+
+void debug_set_logfile(const char *name)
 {
-       pstring fname;
+       SAFE_FREE(debugf);
+       debugf = SMB_STRDUP(name);
+}
+
+/**************************************************************************
+ reopen the log files
+ note that we now do this unconditionally
+ We attempt to open the new debug fp before closing the old. This means
+ if we run out of fd's we just keep using the old fd rather than aborting.
+ Fix from dgibson@linuxcare.com.
+**************************************************************************/
+
+bool reopen_logs( void )
+{
+       char *fname = NULL;
        mode_t oldumask;
        XFILE *new_dbf = NULL;
        XFILE *old_dbf = NULL;
-       BOOL ret = True;
+       bool ret = True;
 
        if (stdout_logging)
                return True;
 
        oldumask = umask( 022 );
-  
-       pstrcpy(fname, debugf );
+
+       fname = debugf;
+       if (!fname) {
+               return false;
+       }
+       debugf = NULL;
 
        if (lp_loaded()) {
                char *logfname;
 
                logfname = lp_logfile();
-               if (*logfname)
-                       pstrcpy(fname, logfname);
+               if (*logfname) {
+                       SAFE_FREE(fname);
+                       fname = SMB_STRDUP(logfname);
+                       if (!fname) {
+                               return false;
+                       }
+               }
        }
 
-       pstrcpy( debugf, fname );
+       debugf = fname;
        new_dbf = x_fopen( debugf, O_WRONLY|O_APPEND|O_CREAT, 0644);
 
        if (!new_dbf) {
@@ -598,8 +689,8 @@ BOOL reopen_logs( void )
        force_check_log_size();
        (void)umask(oldumask);
 
-       /* Take over stderr to catch ouput into logs */
-       if (dbf && sys_dup2(x_fileno(dbf), 2) == -1) {
+       /* Take over stderr to catch output into logs */
+       if (dbf && dup2(x_fileno(dbf), 2) == -1) {
                close_low_fds(True); /* Close stderr too, if dup2 can't point it
                                        at the logfile */
        }
@@ -607,24 +698,24 @@ BOOL reopen_logs( void )
        return ret;
 }
 
-/* ************************************************************************** **
Force a check of the log size.
- * ************************************************************************** **
- */
+/**************************************************************************
+ Force a check of the log size.
+ ***************************************************************************/
+
 void force_check_log_size( void )
 {
-  debug_count = 100;
+       debug_count = 100;
 }
 
 /***************************************************************************
  Check to see if there is any need to check if the logfile has grown too big.
 **************************************************************************/
 
-BOOL need_to_check_log_size( void )
+bool need_to_check_log_size( void )
 {
        int maxlog;
 
-       if( debug_count++ < 100 )
+       if( debug_count < 100 )
                return( False );
 
        maxlog = lp_max_log_size() * 1024;
@@ -635,10 +726,9 @@ BOOL need_to_check_log_size( void )
        return( True );
 }
 
-/* ************************************************************************** **
- * Check to see if the log has grown to be too big.
- * ************************************************************************** **
- */
+/**************************************************************************
+ Check to see if the log has grown to be too big.
+ **************************************************************************/
 
 void check_log_size( void )
 {
@@ -658,18 +748,21 @@ void check_log_size( void )
 
        maxlog = lp_max_log_size() * 1024;
 
-       if( sys_fstat( x_fileno( dbf ), &st ) == 0 && st.st_size > maxlog ) {
+       if( sys_fstat( x_fileno( dbf ), &st ) == 0 && st.st_ex_size > maxlog ) {
                (void)reopen_logs();
                if( dbf && get_file_size( debugf ) > maxlog ) {
-                       pstring name;
+                       char *name = NULL;
+
+                       if (asprintf(&name, "%s.old", debugf ) < 0) {
+                               return;
+                       }
+                       (void)rename(debugf, name);
 
-                       slprintf( name, sizeof(name)-1, "%s.old", debugf );
-                       (void)rename( debugf, name );
-      
                        if (!reopen_logs()) {
                                /* We failed to reopen a log - continue using the old name. */
                                (void)rename(name, debugf);
                        }
+                       SAFE_FREE(name);
                }
        }
 
@@ -697,291 +790,328 @@ void check_log_size( void )
                }
        }
        debug_count = 0;
-} /* check_log_size */
+}
+
+/*************************************************************************
+ Write an debug message on the debugfile.
+ This is called by dbghdr() and format_debug_text().
+************************************************************************/
 
-/* ************************************************************************** **
- * Write an debug message on the debugfile.
- * This is called by dbghdr() and format_debug_text().
- * ************************************************************************** **
- */
  int Debug1( const char *format_str, ... )
 {
-  va_list ap;  
-  int old_errno = errno;
-
-  if( stdout_logging )
-    {
-    va_start( ap, format_str );
-    if(dbf)
-      (void)x_vfprintf( dbf, format_str, ap );
-    va_end( ap );
-    errno = old_errno;
-    return( 0 );
-    }
-  
+       va_list ap;
+       int old_errno = errno;
+
+       debug_count++;
+
+       if( stdout_logging ) {
+               va_start( ap, format_str );
+               if(dbf)
+                       (void)x_vfprintf( dbf, format_str, ap );
+               va_end( ap );
+               errno = old_errno;
+               goto done;
+       }
+
+       /* prevent recursion by checking if reopen_logs() has temporaily
+          set the debugf string to NULL */
+       if( debugf == NULL)
+               goto done;
+
 #ifdef WITH_SYSLOG
-  if( !lp_syslog_only() )
+       if( !lp_syslog_only() )
 #endif
-    {
-    if( !dbf )
-      {
-      mode_t oldumask = umask( 022 );
-
-      dbf = x_fopen( debugf, O_WRONLY|O_APPEND|O_CREAT, 0644 );
-      (void)umask( oldumask );
-      if( dbf )
-        {
-        x_setbuf( dbf, NULL );
-        }
-      else
-        {
-        errno = old_errno;
-        return(0);
-        }
-      }
-    }
+       {
+               if( !dbf ) {
+                       mode_t oldumask = umask( 022 );
+
+                       dbf = x_fopen( debugf, O_WRONLY|O_APPEND|O_CREAT, 0644 );
+                       (void)umask( oldumask );
+                       if( dbf ) {
+                               x_setbuf( dbf, NULL );
+                       } else {
+                               errno = old_errno;
+                               goto done;
+                       }
+               }
+       }
 
 #ifdef WITH_SYSLOG
-  if( syslog_level < lp_syslog() )
-    {
-    /* map debug levels to syslog() priorities
-     * note that not all DEBUG(0, ...) calls are
-     * necessarily errors
-     */
-    static int priority_map[] = { 
-      LOG_ERR,     /* 0 */
-      LOG_WARNING, /* 1 */
-      LOG_NOTICE,  /* 2 */
-      LOG_INFO,    /* 3 */
-      };
-    int     priority;
-    pstring msgbuf;
-
-    if( syslog_level >= ( sizeof(priority_map) / sizeof(priority_map[0]) )
-     || syslog_level < 0)
-      priority = LOG_DEBUG;
-    else
-      priority = priority_map[syslog_level];
-      
-    va_start( ap, format_str );
-    vslprintf( msgbuf, sizeof(msgbuf)-1, format_str, ap );
-    va_end( ap );
-      
-    msgbuf[255] = '\0';
-    syslog( priority, "%s", msgbuf );
-    }
+       if( syslog_level < lp_syslog() ) {
+               /* map debug levels to syslog() priorities
+                * note that not all DEBUG(0, ...) calls are
+                * necessarily errors */
+               static const int priority_map[4] = {
+                       LOG_ERR,     /* 0 */
+                       LOG_WARNING, /* 1 */
+                       LOG_NOTICE,  /* 2 */
+                       LOG_INFO,    /* 3 */
+               };
+               int     priority;
+               char *msgbuf = NULL;
+               int ret;
+
+               if( syslog_level >= ARRAY_SIZE(priority_map) || syslog_level < 0)
+                       priority = LOG_DEBUG;
+               else
+                       priority = priority_map[syslog_level];
+
+               va_start(ap, format_str);
+               ret = vasprintf(&msgbuf, format_str, ap);
+               va_end(ap);
+
+               if (ret != -1) {
+                       syslog(priority, "%s", msgbuf);
+               }
+               SAFE_FREE(msgbuf);
+       }
 #endif
-  
-  check_log_size();
+
+       check_log_size();
 
 #ifdef WITH_SYSLOG
-  if( !lp_syslog_only() )
+       if( !lp_syslog_only() )
 #endif
-    {
-    va_start( ap, format_str );
-    if(dbf)
-      (void)x_vfprintf( dbf, format_str, ap );
-    va_end( ap );
-    if(dbf)
-      (void)x_fflush( dbf );
-    }
+       {
+               va_start( ap, format_str );
+               if(dbf)
+                       (void)x_vfprintf( dbf, format_str, ap );
+               va_end( ap );
+               if(dbf)
+                       (void)x_fflush( dbf );
+       }
 
-  errno = old_errno;
+ done:
+       TALLOC_FREE(tmp_debug_ctx);
 
-  return( 0 );
-  } /* Debug1 */
+       errno = old_errno;
 
+       return( 0 );
+}
+
+
+/**************************************************************************
+ Print the buffer content via Debug1(), then reset the buffer.
+ Input:  none
+ Output: none
+****************************************************************************/
 
-/* ************************************************************************** **
- * Print the buffer content via Debug1(), then reset the buffer.
- *
- *  Input:  none
- *  Output: none
- *
- * ************************************************************************** **
- */
 static void bufr_print( void )
-  {
-  format_bufr[format_pos] = '\0';
-  (void)Debug1( "%s", format_bufr );
-  format_pos = 0;
-  } /* bufr_print */
-
-/* ************************************************************************** **
- * Format the debug message text.
- *
- *  Input:  msg - Text to be added to the "current" debug message text.
- *
- *  Output: none.
- *
- *  Notes:  The purpose of this is two-fold.  First, each call to syslog()
- *          (used by Debug1(), see above) generates a new line of syslog
- *          output.  This is fixed by storing the partial lines until the
- *          newline character is encountered.  Second, printing the debug
- *          message lines when a newline is encountered allows us to add
- *          spaces, thus indenting the body of the message and making it
- *          more readable.
- *
- * ************************************************************************** **
- */
-static void format_debug_text( char *msg )
-  {
-  size_t i;
-  BOOL timestamp = (!stdout_logging && (lp_timestamp_logs() || 
-                                       !(lp_loaded())));
-
-  for( i = 0; msg[i]; i++ )
-    {
-    /* Indent two spaces at each new line. */
-    if(timestamp && 0 == format_pos)
-      {
-      format_bufr[0] = format_bufr[1] = ' ';
-      format_pos = 2;
-      }
-
-    /* If there's room, copy the character to the format buffer. */
-    if( format_pos < FORMAT_BUFR_MAX )
-      format_bufr[format_pos++] = msg[i];
-
-    /* If a newline is encountered, print & restart. */
-    if( '\n' == msg[i] )
-      bufr_print();
-
-    /* If the buffer is full dump it out, reset it, and put out a line
-     * continuation indicator.
-     */
-    if( format_pos >= FORMAT_BUFR_MAX )
-      {
-      bufr_print();
-      (void)Debug1( " +>\n" );
-      }
-    }
-
-  /* Just to be safe... */
-  format_bufr[format_pos] = '\0';
-  } /* format_debug_text */
-
-/* ************************************************************************** **
- * Flush debug output, including the format buffer content.
- *
- *  Input:  none
- *  Output: none
- *
- * ************************************************************************** **
- */
+{
+       format_bufr[format_pos] = '\0';
+       (void)Debug1( "%s", format_bufr );
+       format_pos = 0;
+}
+
+/***************************************************************************
+ Format the debug message text.
+
+ Input:  msg - Text to be added to the "current" debug message text.
+
+ Output: none.
+
+ Notes:  The purpose of this is two-fold.  First, each call to syslog()
+         (used by Debug1(), see above) generates a new line of syslog
+         output.  This is fixed by storing the partial lines until the
+         newline character is encountered.  Second, printing the debug
+         message lines when a newline is encountered allows us to add
+         spaces, thus indenting the body of the message and making it
+         more readable.
+**************************************************************************/
+
+static void format_debug_text( const char *msg )
+{
+       size_t i;
+       bool timestamp = (!stdout_logging && (lp_timestamp_logs() || !(lp_loaded())));
+
+       if (!format_bufr) {
+               debug_init();
+       }
+
+       for( i = 0; msg[i]; i++ ) {
+               /* Indent two spaces at each new line. */
+               if(timestamp && 0 == format_pos) {
+                       format_bufr[0] = format_bufr[1] = ' ';
+                       format_pos = 2;
+               }
+
+               /* If there's room, copy the character to the format buffer. */
+               if( format_pos < FORMAT_BUFR_MAX )
+                       format_bufr[format_pos++] = msg[i];
+
+               /* If a newline is encountered, print & restart. */
+               if( '\n' == msg[i] )
+                       bufr_print();
+
+               /* If the buffer is full dump it out, reset it, and put out a line
+                * continuation indicator.
+                */
+               if( format_pos >= FORMAT_BUFR_MAX ) {
+                       bufr_print();
+                       (void)Debug1( " +>\n" );
+               }
+       }
+
+       /* Just to be safe... */
+       format_bufr[format_pos] = '\0';
+}
+
+/***************************************************************************
+ Flush debug output, including the format buffer content.
+
+ Input:  none
+ Output: none
+***************************************************************************/
+
 void dbgflush( void )
-  {
-  bufr_print();
-  if(dbf)
-    (void)x_fflush( dbf );
-  } /* dbgflush */
-
-/* ************************************************************************** **
- * Print a Debug Header.
- *
- *  Input:  level - Debug level of the message (not the system-wide debug
- *                  level. )
- *          file  - Pointer to a string containing the name of the file
- *                  from which this function was called, or an empty string
- *                  if the __FILE__ macro is not implemented.
- *          func  - Pointer to a string containing the name of the function
- *                  from which this function was called, or an empty string
- *                  if the __FUNCTION__ macro is not implemented.
- *          line  - line number of the call to dbghdr, assuming __LINE__
- *                  works.
- *
- *  Output: Always True.  This makes it easy to fudge a call to dbghdr()
- *          in a macro, since the function can be called as part of a test.
- *          Eg: ( (level <= DEBUGLEVEL) && (dbghdr(level,"",line)) )
- *
- *  Notes:  This function takes care of setting syslog_level.
- *
- * ************************************************************************** **
- */
+{
+       bufr_print();
+       if(dbf)
+               (void)x_fflush( dbf );
+}
 
-BOOL dbghdr( int level, const char *file, const char *func, int line )
+/***************************************************************************
+ Print a Debug Header.
+
+ Input:  level - Debug level of the message (not the system-wide debug
+                  level. )
+         cls   - Debuglevel class of the calling module.
+          file  - Pointer to a string containing the name of the file
+                  from which this function was called, or an empty string
+                  if the __FILE__ macro is not implemented.
+          func  - Pointer to a string containing the name of the function
+                  from which this function was called, or an empty string
+                  if the __FUNCTION__ macro is not implemented.
+         line  - line number of the call to dbghdr, assuming __LINE__
+                 works.
+
+  Output: Always True.  This makes it easy to fudge a call to dbghdr()
+          in a macro, since the function can be called as part of a test.
+          Eg: ( (level <= DEBUGLEVEL) && (dbghdr(level,"",line)) )
+
+  Notes:  This function takes care of setting syslog_level.
+
+****************************************************************************/
+
+bool dbghdrclass(int level, int cls, const char *location, const char *func)
 {
-  /* Ensure we don't lose any real errno value. */
-  int old_errno = errno;
-
-  if( format_pos ) {
-    /* This is a fudge.  If there is stuff sitting in the format_bufr, then
-     * the *right* thing to do is to call
-     *   format_debug_text( "\n" );
-     * to write the remainder, and then proceed with the new header.
-     * Unfortunately, there are several places in the code at which
-     * the DEBUG() macro is used to build partial lines.  That in mind,
-     * we'll work under the assumption that an incomplete line indicates
-     * that a new header is *not* desired.
-     */
-    return( True );
-  }
+       /* Ensure we don't lose any real errno value. */
+       int old_errno = errno;
+
+       if( format_pos ) {
+               /* This is a fudge.  If there is stuff sitting in the format_bufr, then
+                * the *right* thing to do is to call
+                *   format_debug_text( "\n" );
+                * to write the remainder, and then proceed with the new header.
+                * Unfortunately, there are several places in the code at which
+                * the DEBUG() macro is used to build partial lines.  That in mind,
+                * we'll work under the assumption that an incomplete line indicates
+                * that a new header is *not* desired.
+                */
+               return( True );
+       }
 
 #ifdef WITH_SYSLOG
-  /* Set syslog_level. */
-  syslog_level = level;
+       /* Set syslog_level. */
+       syslog_level = level;
 #endif
 
-  /* Don't print a header if we're logging to stdout. */
-  if( stdout_logging )
-    return( True );
+       /* Don't print a header if we're logging to stdout. */
+       if( stdout_logging )
+               return( True );
 
-  /* Print the header if timestamps are turned on.  If parameters are
-   * not yet loaded, then default to timestamps on.
-   */
-  if( lp_timestamp_logs() || !(lp_loaded()) ) {
-    char header_str[200];
+       /* Print the header if timestamps are turned on.  If parameters are
+        * not yet loaded, then default to timestamps on.
+        */
+       if( lp_timestamp_logs() || lp_debug_prefix_timestamp() || !(lp_loaded()) ) {
+               char header_str[200];
 
-       header_str[0] = '\0';
+               header_str[0] = '\0';
 
-       if( lp_debug_pid())
-         slprintf(header_str,sizeof(header_str)-1,", pid=%u",(unsigned int)sys_getpid());
+               if( lp_debug_pid())
+                       slprintf(header_str,sizeof(header_str)-1,", pid=%u",(unsigned int)sys_getpid());
 
-       if( lp_debug_uid()) {
-      size_t hs_len = strlen(header_str);
-         slprintf(header_str + hs_len,
-               sizeof(header_str) - 1 - hs_len,
-                          ", effective(%u, %u), real(%u, %u)",
-               (unsigned int)geteuid(), (unsigned int)getegid(),
-                          (unsigned int)getuid(), (unsigned int)getgid()); 
-       }
+               if( lp_debug_uid()) {
+                       size_t hs_len = strlen(header_str);
+                       slprintf(header_str + hs_len,
+                       sizeof(header_str) - 1 - hs_len,
+                               ", effective(%u, %u), real(%u, %u)",
+                               (unsigned int)geteuid(), (unsigned int)getegid(),
+                               (unsigned int)getuid(), (unsigned int)getgid()); 
+               }
+
+               if (lp_debug_class() && (cls != DBGC_ALL)) {
+                       size_t hs_len = strlen(header_str);
+                       slprintf(header_str + hs_len,
+                                sizeof(header_str) -1 - hs_len,
+                                ", class=%s",
+                                default_classname_table[cls]);
+               }
   
-    /* Print it all out at once to prevent split syslog output. */
-    (void)Debug1( "[%s, %d%s] %s:%s(%d)\n",
-                  timestring(lp_debug_hires_timestamp()), level,
-                                 header_str, file, func, line );
-  }
-
-  errno = old_errno;
-  return( True );
+               /* Print it all out at once to prevent split syslog output. */
+               if( lp_debug_prefix_timestamp() ) {
+                   (void)Debug1( "[%s, %2d%s] ",
+                       current_timestring(debug_ctx(),
+                                          lp_debug_hires_timestamp()),
+                       level, header_str);
+               } else {
+                   (void)Debug1( "[%s, %2d%s] %s(%s)\n",
+                       current_timestring(debug_ctx(),
+                                          lp_debug_hires_timestamp()),
+                       level, header_str, location, func );
+               }
+       }
+
+       errno = old_errno;
+       return( True );
 }
 
-/* ************************************************************************** **
- * Add text to the body of the "current" debug message via the format buffer.
- *
- *  Input:  format_str  - Format string, as used in printf(), et. al.
- *          ...         - Variable argument list.
- *
- *  ..or..  va_alist    - Old style variable parameter list starting point.
- *
- *  Output: Always True.  See dbghdr() for more info, though this is not
- *          likely to be used in the same way.
- *
- * ************************************************************************** **
- */
- BOOL dbgtext( const char *format_str, ... )
-  {
-  va_list ap;
-  pstring msgbuf;
+bool dbghdr(int level, const char *location, const char *func)
+{
+       /* For compatibility with Samba 4, which doesn't have debug classes */
+       return dbghdrclass(level, 0, location, func);
+}
 
-  va_start( ap, format_str ); 
-  vslprintf( msgbuf, sizeof(msgbuf)-1, format_str, ap );
-  va_end( ap );
+/***************************************************************************
+ Add text to the body of the "current" debug message via the format buffer.
+
+  Input:  format_str  - Format string, as used in printf(), et. al.
+          ...         - Variable argument list.
 
-  format_debug_text( msgbuf );
+  ..or..  va_alist    - Old style variable parameter list starting point.
 
-  return( True );
-  } /* dbgtext */
+  Output: Always True.  See dbghdr() for more info, though this is not
+          likely to be used in the same way.
 
+***************************************************************************/
 
-/* ************************************************************************** */
+ bool dbgtext( const char *format_str, ... )
+{
+       va_list ap;
+       char *msgbuf = NULL;
+       bool ret = true;
+       int res;
+
+       va_start(ap, format_str);
+       res = vasprintf(&msgbuf, format_str, ap);
+       va_end(ap);
+
+       if (res != -1) {
+               format_debug_text(msgbuf);
+       } else {
+               ret = false;
+       }
+       SAFE_FREE(msgbuf);
+       return ret;
+}
+
+/*
+ * Get us a temporary talloc context usable just for DEBUG arguments
+ */
+TALLOC_CTX *debug_ctx(void)
+{
+        if (tmp_debug_ctx == NULL) {
+                tmp_debug_ctx = talloc_named_const(NULL, 0, "debug_ctx");
+        }
+        return tmp_debug_ctx;
+}