2 Unix SMB/CIFS implementation.
6 Copyright (C) Andrew Tridgell 1992-2006
7 Copyright (C) Volker Lendecke 2007-2011
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include "libcli/util/ntstatus.h"
26 #include "lib/util/memory.h"
27 #include "lib/util/byteorder.h"
28 #include "system/filesys.h"
29 #include "../lib/tdb/include/tdb.h"
30 #include "../lib/util/util_tdb.h"
32 /* these are little tdb utility functions that are meant to make
33 dealing with a tdb database a little less cumbersome in Samba */
35 /***************************************************************
36 Make a TDB_DATA and keep the const warning in one place
37 ****************************************************************/
39 TDB_DATA
make_tdb_data(const uint8_t *dptr
, size_t dsize
)
42 ret
.dptr
= discard_const_p(uint8_t, dptr
);
47 bool tdb_data_equal(TDB_DATA t1
, TDB_DATA t2
)
49 if (t1
.dsize
!= t2
.dsize
) {
52 return (memcmp(t1
.dptr
, t2
.dptr
, t1
.dsize
) == 0);
55 bool tdb_data_is_empty(TDB_DATA d
)
57 return (d
.dsize
== 0) || (d
.dptr
== NULL
);
60 TDB_DATA
string_tdb_data(const char *string
)
62 return make_tdb_data((const uint8_t *)string
, string
? strlen(string
) : 0 );
65 TDB_DATA
string_term_tdb_data(const char *string
)
67 return make_tdb_data((const uint8_t *)string
, string
? strlen(string
) + 1 : 0);
70 TDB_DATA
tdb_data_talloc_copy(TALLOC_CTX
* mem_ctx
, TDB_DATA data
) {
72 .dptr
= (uint8_t *)talloc_size(mem_ctx
, data
.dsize
+1),
75 if (ret
.dptr
== NULL
) {
78 memcpy(ret
.dptr
, data
.dptr
, data
.dsize
);
79 ret
.dptr
[ret
.dsize
] = '\0';
85 /****************************************************************************
86 Lock a chain by string. Return non-zero if lock failed.
87 ****************************************************************************/
89 int tdb_lock_bystring(struct tdb_context
*tdb
, const char *keyval
)
91 TDB_DATA key
= string_term_tdb_data(keyval
);
93 return tdb_chainlock(tdb
, key
);
96 /****************************************************************************
97 Unlock a chain by string.
98 ****************************************************************************/
100 void tdb_unlock_bystring(struct tdb_context
*tdb
, const char *keyval
)
102 TDB_DATA key
= string_term_tdb_data(keyval
);
104 tdb_chainunlock(tdb
, key
);
107 /****************************************************************************
108 Read lock a chain by string. Return non-zero if lock failed.
109 ****************************************************************************/
111 int tdb_read_lock_bystring(struct tdb_context
*tdb
, const char *keyval
)
113 TDB_DATA key
= string_term_tdb_data(keyval
);
115 return tdb_chainlock_read(tdb
, key
);
118 /****************************************************************************
119 Read unlock a chain by string.
120 ****************************************************************************/
122 void tdb_read_unlock_bystring(struct tdb_context
*tdb
, const char *keyval
)
124 TDB_DATA key
= string_term_tdb_data(keyval
);
126 tdb_chainunlock_read(tdb
, key
);
130 /****************************************************************************
131 Fetch a int32_t value by a arbitrary blob key, return -1 if not found.
132 Output is int32_t in native byte order.
133 ****************************************************************************/
135 static int fetch_int32_parser(TDB_DATA key
, TDB_DATA data
, void *private_data
)
137 if (data
.dsize
== sizeof(int32_t)) {
138 *((int32_t *)private_data
) = PULL_LE_I32(data
.dptr
, 0);
143 static int32_t tdb_fetch_int32_byblob(struct tdb_context
*tdb
, TDB_DATA key
)
146 tdb_parse_record(tdb
, key
, fetch_int32_parser
, &v
);
150 /****************************************************************************
151 Fetch a int32_t value by string key, return -1 if not found.
152 Output is int32_t in native byte order.
153 ****************************************************************************/
155 int32_t tdb_fetch_int32(struct tdb_context
*tdb
, const char *keystr
)
157 return tdb_fetch_int32_byblob(tdb
, string_term_tdb_data(keystr
));
160 /****************************************************************************
161 Store a int32_t value by an arbitrary blob key, return 0 on success, -ve on failure.
162 Input is int32_t in native byte order. Output in tdb is in little-endian.
163 ****************************************************************************/
165 static int tdb_store_int32_byblob(struct tdb_context
*tdb
, TDB_DATA key
,
172 data
.dptr
= (unsigned char *)&v_store
;
173 data
.dsize
= sizeof(int32_t);
175 return tdb_store(tdb
, key
, data
, TDB_REPLACE
);
178 /****************************************************************************
179 Store a int32_t value by string key, return 0 on success, -ve on failure.
180 Input is int32_t in native byte order. Output in tdb is in little-endian.
181 ****************************************************************************/
183 int tdb_store_int32(struct tdb_context
*tdb
, const char *keystr
, int32_t v
)
185 return tdb_store_int32_byblob(tdb
, string_term_tdb_data(keystr
), v
);
188 /****************************************************************************
189 Fetch a uint32_t value by a arbitrary blob key, return false if not found.
190 Output is uint32_t in native byte order.
191 ****************************************************************************/
193 static int fetch_uint32_parser(TDB_DATA key
, TDB_DATA data
, void *private_data
)
195 if (data
.dsize
!= sizeof(uint32_t)) {
198 *((uint32_t *)private_data
) = PULL_LE_U32(data
.dptr
, 0);
202 static bool tdb_fetch_uint32_byblob(struct tdb_context
*tdb
, TDB_DATA key
,
205 int ret
= tdb_parse_record(tdb
, key
, fetch_uint32_parser
, value
);
214 /****************************************************************************
215 Fetch a uint32_t value by string key, return false if not found.
216 Output is uint32_t in native byte order.
217 ****************************************************************************/
219 bool tdb_fetch_uint32(struct tdb_context
*tdb
, const char *keystr
, uint32_t *value
)
221 return tdb_fetch_uint32_byblob(tdb
, string_term_tdb_data(keystr
), value
);
224 /****************************************************************************
225 Store a uint32_t value by an arbitrary blob key, return true on success, false on failure.
226 Input is uint32_t in native byte order. Output in tdb is in little-endian.
227 ****************************************************************************/
229 static bool tdb_store_uint32_byblob(struct tdb_context
*tdb
, TDB_DATA key
,
236 SIVAL(&v_store
, 0, value
);
237 data
.dptr
= (unsigned char *)&v_store
;
238 data
.dsize
= sizeof(uint32_t);
240 if (tdb_store(tdb
, key
, data
, TDB_REPLACE
) != 0)
246 /****************************************************************************
247 Store a uint32_t value by string key, return true on success, false on failure.
248 Input is uint32_t in native byte order. Output in tdb is in little-endian.
249 ****************************************************************************/
251 bool tdb_store_uint32(struct tdb_context
*tdb
, const char *keystr
, uint32_t value
)
253 return tdb_store_uint32_byblob(tdb
, string_term_tdb_data(keystr
), value
);
255 /****************************************************************************
256 Store a buffer by a null terminated string key. Return 0 on success, -ve
258 ****************************************************************************/
260 int tdb_store_bystring(struct tdb_context
*tdb
, const char *keystr
, TDB_DATA data
, int flags
)
262 TDB_DATA key
= string_term_tdb_data(keystr
);
264 return tdb_store(tdb
, key
, data
, flags
);
267 /****************************************************************************
268 Fetch a buffer using a null terminated string key. Don't forget to call
269 free() on the result dptr.
270 ****************************************************************************/
272 TDB_DATA
tdb_fetch_bystring(struct tdb_context
*tdb
, const char *keystr
)
274 TDB_DATA key
= string_term_tdb_data(keystr
);
276 return tdb_fetch(tdb
, key
);
279 /****************************************************************************
280 Delete an entry using a null terminated string key.
281 ****************************************************************************/
283 int tdb_delete_bystring(struct tdb_context
*tdb
, const char *keystr
)
285 TDB_DATA key
= string_term_tdb_data(keystr
);
287 return tdb_delete(tdb
, key
);
290 /****************************************************************************
291 Atomic integer change. Returns old value. To create, set initial value in *oldval.
292 ****************************************************************************/
294 int32_t tdb_change_int32_atomic(struct tdb_context
*tdb
, const char *keystr
, int32_t *oldval
, int32_t change_val
)
299 if (tdb_lock_bystring(tdb
, keystr
) != 0)
302 if ((val
= tdb_fetch_int32(tdb
, keystr
)) == -1) {
303 /* The lookup failed */
304 if (tdb_error(tdb
) != TDB_ERR_NOEXIST
) {
305 /* but not because it didn't exist */
309 /* Start with 'old' value */
313 /* It worked, set return value (oldval) to tdb data */
317 /* Increment value for storage and return next time */
320 if (tdb_store_int32(tdb
, keystr
, val
) != 0)
327 tdb_unlock_bystring(tdb
, keystr
);
331 /****************************************************************************
332 Atomic unsigned integer change. Returns old value. To create, set initial value in *oldval.
333 ****************************************************************************/
335 bool tdb_change_uint32_atomic(struct tdb_context
*tdb
, const char *keystr
, uint32_t *oldval
, uint32_t change_val
)
340 if (tdb_lock_bystring(tdb
, keystr
) != 0)
343 if (!tdb_fetch_uint32(tdb
, keystr
, &val
)) {
345 if (tdb_error(tdb
) != TDB_ERR_NOEXIST
) {
346 /* and not because it didn't exist */
350 /* Start with 'old' value */
354 /* it worked, set return value (oldval) to tdb data */
359 /* get a new value to store */
362 if (!tdb_store_uint32(tdb
, keystr
, val
))
369 tdb_unlock_bystring(tdb
, keystr
);
373 /****************************************************************************
374 Return an NTSTATUS from a TDB_ERROR
375 ****************************************************************************/
377 NTSTATUS
map_nt_error_from_tdb(enum TDB_ERROR err
)
383 result
= NT_STATUS_OK
;
385 case TDB_ERR_CORRUPT
:
386 result
= NT_STATUS_INTERNAL_DB_CORRUPTION
;
389 result
= NT_STATUS_UNEXPECTED_IO_ERROR
;
392 result
= NT_STATUS_NO_MEMORY
;
395 result
= NT_STATUS_OBJECT_NAME_COLLISION
;
400 * TDB_ERR_LOCK is very broad, we could for example
401 * distinguish between fcntl locks and invalid lock
402 * sequences. So NT_STATUS_FILE_LOCK_CONFLICT is a
405 result
= NT_STATUS_FILE_LOCK_CONFLICT
;
409 case TDB_ERR_LOCK_TIMEOUT
:
411 * These two ones in the enum are not actually used
413 result
= NT_STATUS_FILE_LOCK_CONFLICT
;
415 case TDB_ERR_NOEXIST
:
416 result
= NT_STATUS_NOT_FOUND
;
419 result
= NT_STATUS_INVALID_PARAMETER
;
422 result
= NT_STATUS_ACCESS_DENIED
;
424 case TDB_ERR_NESTING
:
425 result
= NT_STATUS_INTERNAL_ERROR
;
428 result
= NT_STATUS_INTERNAL_ERROR
;
434 int map_unix_error_from_tdb(enum TDB_ERROR err
)
442 case TDB_ERR_CORRUPT
:
457 * TDB_ERR_LOCK is very broad, we could for example
458 * distinguish between fcntl locks and invalid lock
459 * sequences. EWOULDBLOCK is wrong, but there is no real
460 * generic lock error code in errno.h
462 result
= EWOULDBLOCK
;
466 case TDB_ERR_LOCK_TIMEOUT
:
468 * These two ones in the enum are not actually used
472 case TDB_ERR_NOEXIST
:
481 case TDB_ERR_NESTING
:
483 * Well, this db is already busy...
491 struct tdb_fetch_talloc_state
{
496 static int tdb_fetch_talloc_parser(TDB_DATA key
, TDB_DATA data
,
499 struct tdb_fetch_talloc_state
*state
= private_data
;
500 state
->buf
= talloc_memdup(state
->mem_ctx
, data
.dptr
, data
.dsize
);
504 int tdb_fetch_talloc(struct tdb_context
*tdb
, TDB_DATA key
,
505 TALLOC_CTX
*mem_ctx
, uint8_t **buf
)
507 struct tdb_fetch_talloc_state state
= { .mem_ctx
= mem_ctx
};
510 ret
= tdb_parse_record(tdb
, key
, tdb_fetch_talloc_parser
, &state
);
512 enum TDB_ERROR err
= tdb_error(tdb
);
513 return map_unix_error_from_tdb(err
);
516 if (state
.buf
== NULL
) {