2 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 2004
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 the stupidity of the unix fcntl locking design forces us to never
24 allow a database file to be opened twice in the same process. These
25 wrappers provide convenient access to a tdb or ldb, taking advantage
26 of talloc destructors to ensure that only a single open is done
30 #include "lib/events/events.h"
31 #include "lib/ldb/include/ldb.h"
32 #include "lib/ldb/include/ldb_errors.h"
33 #include "lib/ldb-samba/ldif_handlers.h"
35 #include "dsdb/samdb/samdb.h"
36 #include "param/param.h"
39 this is used to catch debug messages from ldb
41 static void ldb_wrap_debug(void *context
, enum ldb_debug_level level
,
42 const char *fmt
, va_list ap
) PRINTF_ATTRIBUTE(3,0);
44 static void ldb_wrap_debug(void *context
, enum ldb_debug_level level
,
45 const char *fmt
, va_list ap
)
56 case LDB_DEBUG_WARNING
:
64 vasprintf(&s
, fmt
, ap
);
66 DEBUG(samba_level
, ("ldb: %s\n", s
));
70 /* check for memory leaks on the ldb context */
71 static int ldb_wrap_destructor(struct ldb_context
*ldb
)
73 size_t *startup_blocks
= (size_t *)ldb_get_opaque(ldb
, "startup_blocks");
75 talloc_total_blocks(ldb
) > *startup_blocks
+ 400) {
76 DEBUG(0,("WARNING: probable memory leak in ldb %s - %lu blocks (startup %lu) %lu bytes\n",
77 (char *)ldb_get_opaque(ldb
, "wrap_url"),
78 (unsigned long)talloc_total_blocks(ldb
),
79 (unsigned long)*startup_blocks
,
80 (unsigned long)talloc_total_size(ldb
)));
82 talloc_report_full(ldb
, stdout
);
84 smb_panic("probable memory leak in ldb");
91 wrapped connection to a ldb database
92 to close just talloc_free() the returned ldb_context
94 TODO: We need an error_string parameter
96 struct ldb_context
*ldb_wrap_connect(TALLOC_CTX
*mem_ctx
,
97 struct event_context
*ev
,
98 struct loadparm_context
*lp_ctx
,
100 struct auth_session_info
*session_info
,
101 struct cli_credentials
*credentials
,
103 const char *options
[])
105 struct ldb_context
*ldb
;
107 char *real_url
= NULL
;
108 size_t *startup_blocks
;
110 ldb
= ldb_init(mem_ctx
);
115 ldb_set_modules_dir(ldb
,
116 talloc_asprintf(ldb
, "%s/ldb", lp_modulesdir(lp_ctx
)));
122 if (ldb_set_opaque(ldb
, "EventContext", ev
)) {
127 if (ldb_set_opaque(ldb
, "sessionInfo", session_info
)) {
132 if (ldb_set_opaque(ldb
, "credentials", credentials
)) {
137 if (ldb_set_opaque(ldb
, "loadparm", lp_ctx
)) {
142 if (lp_ctx
!= NULL
&& strcmp(lp_sam_url(lp_ctx
), url
) == 0) {
143 dsdb_set_global_schema(ldb
);
146 ret
= ldb_register_samba_handlers(ldb
);
152 ldb_set_debug(ldb
, ldb_wrap_debug
, NULL
);
154 ldb_set_utf8_fns(ldb
, NULL
, wrap_casefold
);
156 real_url
= private_path(ldb
, lp_ctx
, url
);
157 if (real_url
== NULL
) {
162 /* allow admins to force non-sync ldb for all databases */
163 if (lp_parm_bool(lp_ctx
, NULL
, "ldb", "nosync", false)) {
164 flags
|= LDB_FLG_NOSYNC
;
167 /* we usually want Samba databases to be private. If we later
168 find we need one public, we will need to add a parameter to
169 ldb_wrap_connect() */
170 ldb_set_create_perms(ldb
, 0600);
172 ret
= ldb_connect(ldb
, real_url
, flags
, options
);
173 if (ret
!= LDB_SUCCESS
) {
178 /* setup for leak detection */
179 ldb_set_opaque(ldb
, "wrap_url", real_url
);
180 startup_blocks
= talloc(ldb
, size_t);
181 *startup_blocks
= talloc_total_blocks(ldb
);
182 ldb_set_opaque(ldb
, "startup_blocks", startup_blocks
);
184 talloc_set_destructor(ldb
, ldb_wrap_destructor
);