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 2 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, write to the Free Software
  17    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18 */
  19
  20 #include "includes.h"
  21
  22 /**
  23  * @file dmallocmsg.c
  24  *
  25  * Glue code to cause dmalloc info to come out when we receive a prod
  26  * over samba messaging.
  27  **/
  28
  29 #ifdef ENABLE_DMALLOC
  30 static unsigned long our_dm_mark = 0;
  31 #endif /* ENABLE_DMALLOC */
  32
  33
  34 /**
  35  * Respond to a POOL_USAGE message by sending back string form of memory
  36  * usage stats.
  37  **/
  38 static void msg_req_dmalloc_mark(int UNUSED(msg_type),
  39                                  struct process_id UNUSED(src_pid),
  40                                  void *UNUSED(buf), size_t UNUSED(len),
  41                                  void *private_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(int UNUSED(msg_type),
  54                                         struct process_id UNUSED(src_pid),
  55                                         void *UNUSED(buf), size_t UNUSED(len),
  56                                         void *private_data)
  57 {
  58 #ifdef ENABLE_DMALLOC
  59         dmalloc_log_changed(our_dm_mark, True, True, True);
  60         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED: done\n"));
  61 #else
  62         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED but dmalloc not in this process\n"));
  63 #endif
  64 }
  65
  66
  67 /**
  68  * Register handler for MSG_REQ_POOL_USAGE
  69  **/
  70 void register_dmalloc_msgs(void)
  71 {
  72         message_register(MSG_REQ_DMALLOC_MARK, msg_req_dmalloc_mark, NULL);
  73         message_register(MSG_REQ_DMALLOC_LOG_CHANGED,
  74                          msg_req_dmalloc_log_changed, NULL);
  75         DEBUG(2, ("Registered MSG_REQ_DMALLOC_MARK and LOG_CHANGED\n"));
  76 }