lib/util/debug_s3.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Bartlett 2011
   5    Copyright (C) Andrew Tridgell 1992-2002
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/util/server_id.h"
  23 #include "librpc/gen_ndr/messaging.h"
  24 #include "messages.h"
  25 #include "lib/util/memory.h"
  26
  27 /* This is the Samba3-specific implementation of reopen_logs(), which
  28  * calls out to the s3 loadparm code, and means that we don't depend
  29  * on loadparm directly. */
  30
  31 bool reopen_logs(void)
  32 {
  33         if (lp_loaded()) {
  34                 struct debug_settings settings;
  35                 debug_set_logfile(lp_logfile(talloc_tos()));
  36
  37                 ZERO_STRUCT(settings);
  38                 settings.max_log_size = lp_max_log_size();
  39                 settings.timestamp_logs = lp_timestamp_logs();
  40                 settings.debug_prefix_timestamp = lp_debug_prefix_timestamp();
  41                 settings.debug_hires_timestamp = lp_debug_hires_timestamp();
  42                 settings.debug_pid = lp_debug_pid();
  43                 settings.debug_uid = lp_debug_uid();
  44                 settings.debug_class = lp_debug_class();
  45                 debug_set_settings(&settings, lp_logging(talloc_tos()),
  46                                    lp_syslog(), lp_syslog_only());
  47         }
  48         return reopen_logs_internal();
  49 }
  50
  51 /****************************************************************************
  52  Receive a "set debug level" message.
  53 ****************************************************************************/
  54
  55 void debug_message(struct messaging_context *msg_ctx,
  56                           void *private_data,
  57                           uint32_t msg_type,
  58                           struct server_id src,
  59                           DATA_BLOB *data)
  60 {
  61         const char *params_str = (const char *)data->data;
  62
  63         /* Check, it's a proper string! */
  64         if (params_str[(data->length)-1] != '\0') {
  65                 DEBUG(1, ("Invalid debug message from pid %u to pid %u\n",
  66                           (unsigned int)procid_to_pid(&src),
  67                           (unsigned int)getpid()));
  68                 return;
  69         }
  70
  71         DEBUG(3, ("INFO: Remote set of debug to `%s'  (pid %u from pid %u)\n",
  72                   params_str, (unsigned int)getpid(),
  73                   (unsigned int)procid_to_pid(&src)));
  74
  75         debug_parse_levels(params_str);
  76 }
  77
  78 /****************************************************************************
  79  Return current debug level.
  80 ****************************************************************************/
  81
  82 static void debuglevel_message(struct messaging_context *msg_ctx,
  83                                void *private_data,
  84                                uint32_t msg_type,
  85                                struct server_id src,
  86                                DATA_BLOB *data)
  87 {
  88         char *message = debug_list_class_names_and_levels();
  89         struct server_id_buf tmp;
  90
  91         if (!message) {
  92                 DEBUG(0,("debuglevel_message - debug_list_class_names_and_levels returned NULL\n"));
  93                 return;
  94         }
  95
  96         DEBUG(1, ("INFO: Received REQ_DEBUGLEVEL message from PID %s\n",
  97                   server_id_str_buf(src, &tmp)));
  98         messaging_send_buf(msg_ctx, src, MSG_DEBUGLEVEL,
  99                            (uint8_t *)message, strlen(message) + 1);
 100
 101         TALLOC_FREE(message);
 102 }
 103
 104 static void debug_ringbuf_log(struct messaging_context *msg_ctx,
 105                               void *private_data,
 106                               uint32_t msg_type,
 107                               struct server_id src,
 108                               DATA_BLOB *data)
 109 {
 110         char *log = debug_get_ringbuf();
 111         size_t logsize = debug_get_ringbuf_size();
 112
 113         if (log == NULL) {
 114                 log = discard_const_p(char, "*disabled*\n");
 115                 logsize = strlen(log) + 1;
 116         }
 117
 118         messaging_send_buf(msg_ctx, src, MSG_RINGBUF_LOG, (uint8_t *)log,
 119                            logsize);
 120 }
 121
 122 void debug_register_msgs(struct messaging_context *msg_ctx)
 123 {
 124         messaging_register(msg_ctx, NULL, MSG_DEBUG, debug_message);
 125         messaging_register(msg_ctx, NULL, MSG_REQ_DEBUGLEVEL,
 126                            debuglevel_message);
 127         messaging_register(msg_ctx, NULL, MSG_REQ_RINGBUF_LOG,
 128                            debug_ringbuf_log);
 129 }