source/lib/dmallocmsg.c

   1 /*
   2    samba -- Unix SMB/CIFS implementation.
   3    Copyright (C) 2002 by Martin Pool
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 3 of the License, or
   8    (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 #include "includes.h"
  20
  21 /**
  22  * @file dmallocmsg.c
  23  *
  24  * Glue code to cause dmalloc info to come out when we receive a prod
  25  * over samba messaging.
  26  **/
  27
  28 #ifdef ENABLE_DMALLOC
  29 static unsigned long our_dm_mark = 0;
  30 #endif /* ENABLE_DMALLOC */
  31
  32
  33 /**
  34  * Respond to a POOL_USAGE message by sending back string form of memory
  35  * usage stats.
  36  **/
  37 static void msg_req_dmalloc_mark(struct messaging_context *msg,
  38                                  void *private_data,
  39                                  uint32_t msg_type,
  40                                  struct server_id server_id,
  41                                  DATA_BLOB *data)
  42 {
  43 #ifdef ENABLE_DMALLOC
  44         our_dm_mark = dmalloc_mark();
  45         DEBUG(2,("Got MSG_REQ_DMALLOC_MARK: mark set\n"));
  46 #else
  47         DEBUG(2,("Got MSG_REQ_DMALLOC_MARK but dmalloc not in this process\n"));
  48 #endif
  49 }
  50
  51
  52
  53 static void msg_req_dmalloc_log_changed(struct messaging_context *msg,
  54                                         void *private_data,
  55                                         uint32_t msg_type,
  56                                         struct server_id server_id,
  57                                         DATA_BLOB *data)
  58 {
  59 #ifdef ENABLE_DMALLOC
  60         dmalloc_log_changed(our_dm_mark, True, True, True);
  61         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED: done\n"));
  62 #else
  63         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED but dmalloc not in this process\n"));
  64 #endif
  65 }
  66
  67
  68 /**
  69  * Register handler for MSG_REQ_POOL_USAGE
  70  **/
  71 void register_dmalloc_msgs(struct messaging_context *msg_ctx)
  72 {
  73         messaging_register(msg_ctx, NULL, MSG_REQ_DMALLOC_MARK,
  74                            msg_req_dmalloc_mark);
  75         messaging_register(msg_ctx, NULL, MSG_REQ_DMALLOC_LOG_CHANGED,
  76                            msg_req_dmalloc_log_changed);
  77         DEBUG(2, ("Registered MSG_REQ_DMALLOC_MARK and LOG_CHANGED\n"));
  78 }