8fb937d56f1ee3774d03af2643f9d81ece4394dd
[tprouty/samba.git] / source / lib / tallocmsg.c
1 /* 
2    samba -- Unix SMB/CIFS implementation.
3    Copyright (C) 2001, 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, 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 tallocmsg.c
24  *
25  * Glue code between talloc profiling and the Samba messaging system.
26  **/
27
28 struct msg_pool_usage_state {
29         TALLOC_CTX *mem_ctx;
30         ssize_t len;
31         size_t buflen;
32         char *s;
33 };
34
35 static void msg_pool_usage_helper(const void *ptr, int depth, int max_depth, int is_ref, void *_s)
36 {
37         const char *name = talloc_get_name(ptr);
38         struct msg_pool_usage_state *state = (struct msg_pool_usage_state *)_s;
39
40         if (is_ref) {
41                 sprintf_append(state->mem_ctx, &state->s, &state->len, &state->buflen,
42                                "%*sreference to: %s\n", depth*4, "", name);
43                 return;
44         }
45
46         if (depth == 0) {
47                 sprintf_append(state->mem_ctx, &state->s, &state->len, &state->buflen,
48                                "%stalloc report on '%s' (total %6lu bytes in %3lu blocks)\n", 
49                                (max_depth < 0 ? "full " :""), name,
50                                (unsigned long)talloc_total_size(ptr),
51                                (unsigned long)talloc_total_blocks(ptr));
52                 return;
53         }
54
55         sprintf_append(state->mem_ctx, &state->s, &state->len, &state->buflen,
56                        "%*s%-30s contains %6lu bytes in %3lu blocks (ref %d)\n", 
57                        depth*4, "",
58                        name,
59                        (unsigned long)talloc_total_size(ptr),
60                        (unsigned long)talloc_total_blocks(ptr),
61                        talloc_reference_count(ptr));
62 }
63
64 /**
65  * Respond to a POOL_USAGE message by sending back string form of memory
66  * usage stats.
67  **/
68 static void msg_pool_usage(struct messaging_context *msg_ctx,
69                            void *private_data, 
70                            uint32_t msg_type, 
71                            struct server_id src,
72                            DATA_BLOB *data)
73 {
74         struct msg_pool_usage_state state;
75
76         SMB_ASSERT(msg_type == MSG_REQ_POOL_USAGE);
77         
78         DEBUG(2,("Got POOL_USAGE\n"));
79
80         state.mem_ctx = talloc_init("msg_pool_usage");
81         if (!state.mem_ctx) {
82                 return;
83         }
84         state.len       = 0;
85         state.buflen    = 512;
86         state.s         = NULL;
87
88         talloc_report_depth_cb(NULL, 0, -1, msg_pool_usage_helper, &state);
89
90         if (!state.s) {
91                 talloc_destroy(state.mem_ctx);
92                 return;
93         }
94         
95         messaging_send_buf(msg_ctx, src, MSG_POOL_USAGE,
96                            (uint8 *)state.s, strlen(state.s)+1);
97
98         talloc_destroy(state.mem_ctx);
99 }
100
101 /**
102  * Register handler for MSG_REQ_POOL_USAGE
103  **/
104 void register_msg_pool_usage(struct messaging_context *msg_ctx)
105 {
106         messaging_register(msg_ctx, NULL, MSG_REQ_POOL_USAGE, msg_pool_usage);
107         DEBUG(2, ("Registered MSG_REQ_POOL_USAGE\n"));
108 }