source3/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 #include "librpc/gen_ndr/messaging.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(struct messaging_context *msg,
  39                                  void *private_data,
  40                                  uint32_t msg_type,
  41                                  struct server_id server_id,
  42                                  DATA_BLOB *data)
  43 {
  44 #ifdef ENABLE_DMALLOC
  45         our_dm_mark = dmalloc_mark();
  46         DEBUG(2,("Got MSG_REQ_DMALLOC_MARK: mark set\n"));
  47 #else
  48         DEBUG(2,("Got MSG_REQ_DMALLOC_MARK but dmalloc not in this process\n"));
  49 #endif
  50 }
  51
  52
  53
  54 static void msg_req_dmalloc_log_changed(struct messaging_context *msg,
  55                                         void *private_data,
  56                                         uint32_t msg_type,
  57                                         struct server_id server_id,
  58                                         DATA_BLOB *data)
  59 {
  60 #ifdef ENABLE_DMALLOC
  61         dmalloc_log_changed(our_dm_mark, True, True, True);
  62         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED: done\n"));
  63 #else
  64         DEBUG(2,("Got MSG_REQ_DMALLOC_LOG_CHANGED but dmalloc not in this process\n"));
  65 #endif
  66 }
  67
  68
  69 /**
  70  * Register handler for MSG_REQ_POOL_USAGE
  71  **/
  72 void register_dmalloc_msgs(struct messaging_context *msg_ctx)
  73 {
  74         messaging_register(msg_ctx, NULL, MSG_REQ_DMALLOC_MARK,
  75                            msg_req_dmalloc_mark);
  76         messaging_register(msg_ctx, NULL, MSG_REQ_DMALLOC_LOG_CHANGED,
  77                            msg_req_dmalloc_log_changed);
  78         DEBUG(2, ("Registered MSG_REQ_DMALLOC_MARK and LOG_CHANGED\n"));
  79 }