4f3acad9ead05a5c7e8d1ac38cdaaa790978064a
[sfrench/samba-autobuild/.git] / source / lib / ldb / ldb_tdb / ldb_tdb_wrap.c
1 /* 
2    ldb database library
3
4    Copyright (C) Andrew Tridgell  2005
5
6      ** NOTE! The following LGPL license applies to the ldb
7      ** library. This does NOT imply that all of Samba is released
8      ** under the LGPL
9    
10    This library is free software; you can redistribute it and/or
11    modify it under the terms of the GNU Lesser General Public
12    License as published by the Free Software Foundation; either
13    version 2 of the License, or (at your option) any later version.
14
15    This library is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18    Lesser General Public License for more details.
19
20    You should have received a copy of the GNU Lesser General Public
21    License along with this library; if not, write to the Free Software
22    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
23 */
24
25 #include "includes.h"
26 #include "ldb/include/includes.h"
27
28 #include "ldb/ldb_tdb/ldb_tdb.h"
29
30 /*
31   the purpose of this code is to work around the braindead posix locking
32   rules, to allow us to have a ldb open more than once while allowing 
33   locking to work
34 */
35
36 struct ltdb_wrap {
37         struct ltdb_wrap *next, *prev;
38         struct tdb_context *tdb;
39         dev_t device;
40         ino_t inode;
41 };
42
43 static struct ltdb_wrap *tdb_list;
44
45 /* destroy the last connection to a tdb */
46 static int ltdb_wrap_destructor(struct ltdb_wrap *w)
47 {
48         tdb_close(w->tdb);
49         if (w->next) {
50                 w->next->prev = w->prev;
51         }
52         if (w->prev) {
53                 w->prev->next = w->next;
54         }
55         if (w == tdb_list) {
56                 tdb_list = w->next;
57         }
58         return 0;
59 }                                
60
61 #if defined(_SAMBA_BUILD_) && (_SAMBA_BUILD_ <= 3)
62 static void ltdb_log_fn(struct tdb_context *tdb, enum tdb_debug_level level, const char *fmt, ...) PRINTF_ATTRIBUTE(3, 4);
63 static void ltdb_log_fn(struct tdb_context *tdb, enum tdb_debug_level level, const char *fmt, ...)
64 {
65         /* until we merge the tdb debug changes into samba3, we don't know 
66            how serious the error is, and we can't go via the ldb loggin code */
67         va_list ap;
68         const char *name = tdb_name(tdb);
69         char *message; 
70         va_start(ap, fmt);
71         message = talloc_vasprintf(NULL, fmt, ap);
72         va_end(ap);
73         DEBUG(3, ("ltdb: tdb(%s): %s", name, message));
74         talloc_free(message);
75 }
76 #else
77 static void ltdb_log_fn(struct tdb_context *tdb, enum tdb_debug_level level, const char *fmt, ...) PRINTF_ATTRIBUTE(3, 4);
78 static void ltdb_log_fn(struct tdb_context *tdb, enum tdb_debug_level level, const char *fmt, ...)
79 {
80         va_list ap;
81         const char *name = tdb_name(tdb);
82         struct ldb_context *ldb = talloc_get_type(tdb_get_logging_private(tdb), struct ldb_context);
83         enum ldb_debug_level ldb_level;
84         char *message; 
85         va_start(ap, fmt);
86         message = talloc_vasprintf(ldb, fmt, ap);
87         va_end(ap);
88
89         switch (level) {
90         case TDB_DEBUG_FATAL:
91                 ldb_level = LDB_DEBUG_FATAL;
92                 break;
93         case TDB_DEBUG_ERROR:
94                 ldb_level = LDB_DEBUG_ERROR;
95                 break;
96         case TDB_DEBUG_WARNING:
97                 ldb_level = LDB_DEBUG_WARNING;
98                 break;
99         case TDB_DEBUG_TRACE:
100                 ldb_level = LDB_DEBUG_TRACE;
101                 break;
102         default:
103                 ldb_level = LDB_DEBUG_FATAL;
104         }
105
106         ldb_debug(ldb, ldb_level, "ltdb: tdb(%s): %s", name, message);
107         talloc_free(message);
108 }
109 #endif
110
111 /*
112   wrapped connection to a tdb database. The caller should _not_ free
113   this as it is not a talloc structure (as tdb does not use talloc
114   yet). It will auto-close when the caller frees the mem_ctx that is
115   passed to this call
116  */
117 struct tdb_context *ltdb_wrap_open(TALLOC_CTX *mem_ctx,
118                                    const char *path, int hash_size, 
119                                    int tdb_flags,
120                                    int open_flags, mode_t mode, 
121                                    struct ldb_context *ldb)
122 {
123         struct ltdb_wrap *w;
124         struct stat st;
125 #if defined(_SAMBA_BUILD_) && (_SAMBA_BUILD_ <= 3)
126         tdb_log_func log_ctx_p = ltdb_log_fn;
127 #else
128         struct tdb_logging_context log_ctx;
129         const struct tdb_logging_context *log_ctx_p = &log_ctx;
130         log_ctx.log_fn = ltdb_log_fn;
131         log_ctx.log_private = ldb;
132 #endif
133
134         if (stat(path, &st) == 0) {
135                 for (w=tdb_list;w;w=w->next) {
136                         if (st.st_dev == w->device && st.st_ino == w->inode) {
137                                 if (!talloc_reference(mem_ctx, w)) {
138                                         return NULL;
139                                 }
140                                 return w->tdb;
141                         }
142                 }
143         }
144
145         w = talloc(mem_ctx, struct ltdb_wrap);
146         if (w == NULL) {
147                 return NULL;
148         }
149
150         w->tdb = tdb_open_ex(path, hash_size, tdb_flags, open_flags, mode,
151                              (const struct tdb_logging_context *)log_ctx_p,
152                              NULL);
153         if (w->tdb == NULL) {
154                 talloc_free(w);
155                 return NULL;
156         }
157
158         if (fstat(tdb_fd(w->tdb), &st) != 0) {
159                 tdb_close(w->tdb);
160                 talloc_free(w);
161                 return NULL;
162         }
163
164         w->device = st.st_dev;
165         w->inode  = st.st_ino;
166
167         talloc_set_destructor(w, ltdb_wrap_destructor);
168
169         w->next = tdb_list;
170         w->prev = NULL;
171         if (tdb_list) {
172                 tdb_list->prev = w;
173         }
174         tdb_list = w;
175         
176         return w->tdb;
177 }
178