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