| 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 | }
|
|---|