s3:smbd: s/struct fd_event/struct tevent_fd
[Samba/gebeck_regimport.git] / source3 / smbd / perfcount.c
bloba7c268a2fef234a41f34ab662f87d2b2ecce50b1
1 /*
2 Unix SMB/Netbios implementation.
3 Perfcounter initialization and support functions
4 Copyright (C) Todd Stecher 2009
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "includes.h"
22 #include "smbd/smbd.h"
24 static struct smb_perfcount_handlers *g_smb_perfcount_handlers = NULL;
26 struct smb_perfcount_module {
27 char *name;
28 struct smb_perfcount_handlers *handlers;
29 struct smb_perfcount_module *prev, *next;
32 struct smb_perfcount_module *modules = NULL;
36 * a module is registered before it is actually loaded - keep a list
38 * @todo - currently perfcount modules are globally configured, so
39 * building the list is not strictly required.
40 * However, its a proven concept in VFS, and is here to allow a
41 * move to eventual per-service perfcount configuration.
43 * Note many pre-connection statistics are interesting
44 * (e.g. before binding to an individual share).
47 static struct smb_perfcount_module *smb_perfcount_find_module(const char *name)
49 struct smb_perfcount_module *entry = modules;
51 while (entry) {
52 if (strcmp(entry->name, name)==0)
53 return entry;
55 entry = entry->next;
58 return NULL;
60 NTSTATUS smb_register_perfcounter(int interface_version, const char *name,
61 const struct smb_perfcount_handlers *handlers)
63 struct smb_perfcount_module *entry = modules;
65 if (interface_version != SMB_PERFCOUNTER_INTERFACE_VERSION) {
66 DEBUG(0, ("Failed to register perfcount module.\n"
67 "The module was compiled against "
68 "SMB_PERFCOUNTER_INTERFACE_VERSION %d,\n"
69 "current SMB_PERFCOUNTER_INTERFACE_VERSION is %d.\n"
70 "Please recompile against the current Samba Version!\n",
71 interface_version, SMB_PERFCOUNTER_INTERFACE_VERSION));
72 return NT_STATUS_OBJECT_TYPE_MISMATCH;
75 if (!name || !name[0] || !handlers) {
76 DEBUG(0,("smb_register_perfcounter() called with NULL pointer "
77 "or empty name!\n"));
78 return NT_STATUS_INVALID_PARAMETER;
81 if (smb_perfcount_find_module(name)) {
82 DEBUG(3,("Perfcount Module %s already loaded!\n", name));
83 return NT_STATUS_OK;
86 entry = SMB_XMALLOC_P(struct smb_perfcount_module);
87 entry->name = smb_xstrdup(name);
88 entry->handlers = discard_const_p(struct smb_perfcount_handlers, handlers);
90 DLIST_ADD(modules, entry);
91 DEBUG(3, ("Successfully added perfcounter module '%s'\n", name));
92 return NT_STATUS_OK;
95 /****************************************************************************
96 initialise smb perf counters
97 ****************************************************************************/
98 static bool smb_load_perfcount_module(const char *name)
100 char *module_path = NULL;
101 char *module_name = NULL;
102 char *module_param = NULL, *p;
104 const struct smb_perfcount_module *entry;
106 DEBUG(3, ("Initialising perfcounter module [%s]\n", name));
108 if (g_smb_perfcount_handlers) {
109 DEBUG(3,("Only 1 perfcount handler may be registered in "
110 "smb.conf\n"));
111 return true;
114 module_path = smb_xstrdup(name);
116 p = strchr_m(module_path, ':');
118 if (p) {
119 *p = 0;
120 module_param = p+1;
121 trim_char(module_param, ' ', ' ');
124 trim_char(module_path, ' ', ' ');
126 module_name = smb_xstrdup(module_path);
128 if (module_name[0] == '/') {
131 * Extract the module name from the path. Just use the base
132 * name of the last path component.
135 SAFE_FREE(module_name);
136 module_name = smb_xstrdup(strrchr_m(module_path, '/')+1);
138 p = strchr_m(module_name, '.');
140 if (p != NULL) {
141 *p = '\0';
145 /* load the perfcounter module */
146 if((entry = smb_perfcount_find_module(module_name)) ||
147 (NT_STATUS_IS_OK(smb_probe_module("perfcount", module_path)) &&
148 (entry = smb_perfcount_find_module(module_name)))) {
150 DEBUG(3,("Successfully loaded perfcounter module [%s] \n", name));
151 } else {
152 DEBUG(0,("Can't find a perfcounter module [%s]\n",name));
153 goto fail;
156 g_smb_perfcount_handlers = entry->handlers;
158 SAFE_FREE(module_path);
159 SAFE_FREE(module_name);
160 return True;
162 fail:
163 SAFE_FREE(module_path);
164 SAFE_FREE(module_name);
165 return False;
168 void smb_init_perfcount_data(struct smb_perfcount_data *pcd)
171 ZERO_STRUCTP(pcd);
172 pcd->handlers = g_smb_perfcount_handlers;
175 bool smb_perfcount_init(void)
177 char *perfcount_object;
179 perfcount_object = lp_perfcount_module(talloc_tos());
181 /* don't init */
182 if (!perfcount_object || !perfcount_object[0])
183 return True;
185 if (!smb_load_perfcount_module(perfcount_object)) {
186 DEBUG(0, ("smbd_load_percount_module failed for %s\n",
187 perfcount_object));
188 return False;
192 return True;