r14030: Fix resource leak in error codepath. Coverity CID #64.
[Samba/nascimento.git] / source3 / tdb / tdb.c
blob4303976087d1dd9e59ac7f2ef7338d1dbfd78e05
1 /*
2 Unix SMB/CIFS implementation.
4 trivial database library
6 Copyright (C) Andrew Tridgell 1999-2004
7 Copyright (C) Paul `Rusty' Russell 2000
8 Copyright (C) Jeremy Allison 2000-2003
10 ** NOTE! The following LGPL license applies to the tdb
11 ** library. This does NOT imply that all of Samba is released
12 ** under the LGPL
14 This library is free software; you can redistribute it and/or
15 modify it under the terms of the GNU Lesser General Public
16 License as published by the Free Software Foundation; either
17 version 2 of the License, or (at your option) any later version.
19 This library is distributed in the hope that it will be useful,
20 but WITHOUT ANY WARRANTY; without even the implied warranty of
21 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 Lesser General Public License for more details.
24 You should have received a copy of the GNU Lesser General Public
25 License along with this library; if not, write to the Free Software
26 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
30 /* NOTE: If you use tdbs under valgrind, and in particular if you run
31 * tdbtorture, you may get spurious "uninitialized value" warnings. I
32 * think this is because valgrind doesn't understand that the mmap'd
33 * area may be written to by other processes. Memory can, from the
34 * point of view of the grinded process, spontaneously become
35 * initialized.
37 * I can think of a few solutions. [mbp 20030311]
39 * 1 - Write suppressions for Valgrind so that it doesn't complain
40 * about this. Probably the most reasonable but people need to
41 * remember to use them.
43 * 2 - Use IO not mmap when running under valgrind. Not so nice.
45 * 3 - Use the special valgrind macros to mark memory as valid at the
46 * right time. Probably too hard -- the process just doesn't know.
47 */
49 #ifdef STANDALONE
50 #if HAVE_CONFIG_H
51 #include <config.h>
52 #endif
54 #include <stdlib.h>
55 #include <stdio.h>
56 #include <fcntl.h>
57 #include <unistd.h>
58 #include <string.h>
59 #include <fcntl.h>
60 #include <errno.h>
61 #include <sys/mman.h>
62 #include <sys/stat.h>
63 #include <signal.h>
64 #include "tdb.h"
65 #include "spinlock.h"
66 #else
67 #include "includes.h"
69 #if defined(PARANOID_MALLOC_CHECKER)
70 #ifdef malloc
71 #undef malloc
72 #endif
74 #ifdef realloc
75 #undef realloc
76 #endif
78 #ifdef calloc
79 #undef calloc
80 #endif
82 #ifdef strdup
83 #undef strdup
84 #endif
86 #ifdef strndup
87 #undef strndup
88 #endif
90 #endif
92 #endif
94 #define TDB_MAGIC_FOOD "TDB file\n"
95 #define TDB_VERSION (0x26011967 + 6)
96 #define TDB_MAGIC (0x26011999U)
97 #define TDB_FREE_MAGIC (~TDB_MAGIC)
98 #define TDB_DEAD_MAGIC (0xFEE1DEAD)
99 #define TDB_ALIGNMENT 4
100 #define MIN_REC_SIZE (2*sizeof(struct list_struct) + TDB_ALIGNMENT)
101 #define DEFAULT_HASH_SIZE 131
102 #define TDB_PAGE_SIZE 0x2000
103 #define FREELIST_TOP (sizeof(struct tdb_header))
104 #define TDB_ALIGN(x,a) (((x) + (a)-1) & ~((a)-1))
105 #define TDB_BYTEREV(x) (((((x)&0xff)<<24)|((x)&0xFF00)<<8)|(((x)>>8)&0xFF00)|((x)>>24))
106 #define TDB_DEAD(r) ((r)->magic == TDB_DEAD_MAGIC)
107 #define TDB_BAD_MAGIC(r) ((r)->magic != TDB_MAGIC && !TDB_DEAD(r))
108 #define TDB_HASH_TOP(hash) (FREELIST_TOP + (BUCKET(hash)+1)*sizeof(tdb_off))
109 #define TDB_DATA_START(hash_size) (TDB_HASH_TOP(hash_size-1) + TDB_SPINLOCK_SIZE(hash_size))
112 /* NB assumes there is a local variable called "tdb" that is the
113 * current context, also takes doubly-parenthesized print-style
114 * argument. */
115 #define TDB_LOG(x) (tdb->log_fn?((tdb->log_fn x),0) : 0)
117 /* lock offsets */
118 #define GLOBAL_LOCK 0
119 #define ACTIVE_LOCK 4
121 #ifndef MAP_FILE
122 #define MAP_FILE 0
123 #endif
125 #ifndef MAP_FAILED
126 #define MAP_FAILED ((void *)-1)
127 #endif
129 /* free memory if the pointer is valid and zero the pointer */
130 #ifndef SAFE_FREE
131 #define SAFE_FREE(x) do { if ((x) != NULL) {free((x)); (x)=NULL;} } while(0)
132 #endif
134 #define BUCKET(hash) ((hash) % tdb->header.hash_size)
135 TDB_DATA tdb_null;
137 /* all contexts, to ensure no double-opens (fcntl locks don't nest!) */
138 static TDB_CONTEXT *tdbs = NULL;
140 static int tdb_munmap(TDB_CONTEXT *tdb)
142 if (tdb->flags & TDB_INTERNAL)
143 return 0;
145 #ifdef HAVE_MMAP
146 if (tdb->map_ptr) {
147 int ret = munmap(tdb->map_ptr, tdb->map_size);
148 if (ret != 0)
149 return ret;
151 #endif
152 tdb->map_ptr = NULL;
153 return 0;
156 static void tdb_mmap(TDB_CONTEXT *tdb)
158 if (tdb->flags & TDB_INTERNAL)
159 return;
161 #ifdef HAVE_MMAP
162 if (!(tdb->flags & TDB_NOMMAP)) {
163 tdb->map_ptr = mmap(NULL, tdb->map_size,
164 PROT_READ|(tdb->read_only? 0:PROT_WRITE),
165 MAP_SHARED|MAP_FILE, tdb->fd, 0);
168 * NB. When mmap fails it returns MAP_FAILED *NOT* NULL !!!!
171 if (tdb->map_ptr == MAP_FAILED) {
172 tdb->map_ptr = NULL;
173 TDB_LOG((tdb, 2, "tdb_mmap failed for size %d (%s)\n",
174 tdb->map_size, strerror(errno)));
176 } else {
177 tdb->map_ptr = NULL;
179 #else
180 tdb->map_ptr = NULL;
181 #endif
184 /* Endian conversion: we only ever deal with 4 byte quantities */
185 static void *convert(void *buf, u32 size)
187 u32 i, *p = buf;
188 for (i = 0; i < size / 4; i++)
189 p[i] = TDB_BYTEREV(p[i]);
190 return buf;
192 #define DOCONV() (tdb->flags & TDB_CONVERT)
193 #define CONVERT(x) (DOCONV() ? convert(&x, sizeof(x)) : &x)
195 /* the body of the database is made of one list_struct for the free space
196 plus a separate data list for each hash value */
197 struct list_struct {
198 tdb_off next; /* offset of the next record in the list */
199 tdb_len rec_len; /* total byte length of record */
200 tdb_len key_len; /* byte length of key */
201 tdb_len data_len; /* byte length of data */
202 u32 full_hash; /* the full 32 bit hash of the key */
203 u32 magic; /* try to catch errors */
204 /* the following union is implied:
205 union {
206 char record[rec_len];
207 struct {
208 char key[key_len];
209 char data[data_len];
211 u32 totalsize; (tailer)
216 /***************************************************************
217 Allow a caller to set a "alarm" flag that tdb can check to abort
218 a blocking lock on SIGALRM.
219 ***************************************************************/
221 static sig_atomic_t *palarm_fired;
223 void tdb_set_lock_alarm(sig_atomic_t *palarm)
225 palarm_fired = palarm;
228 /* a byte range locking function - return 0 on success
229 this functions locks/unlocks 1 byte at the specified offset.
231 On error, errno is also set so that errors are passed back properly
232 through tdb_open(). */
233 static int tdb_brlock(TDB_CONTEXT *tdb, tdb_off offset,
234 int rw_type, int lck_type, int probe)
236 struct flock fl;
237 int ret;
239 if (tdb->flags & TDB_NOLOCK)
240 return 0;
241 if ((rw_type == F_WRLCK) && (tdb->read_only)) {
242 errno = EACCES;
243 return -1;
246 fl.l_type = rw_type;
247 fl.l_whence = SEEK_SET;
248 fl.l_start = offset;
249 fl.l_len = 1;
250 fl.l_pid = 0;
252 do {
253 ret = fcntl(tdb->fd,lck_type,&fl);
254 if (ret == -1 && errno == EINTR && palarm_fired && *palarm_fired)
255 break;
256 } while (ret == -1 && errno == EINTR);
258 if (ret == -1) {
259 if (!probe && lck_type != F_SETLK) {
260 /* Ensure error code is set for log fun to examine. */
261 if (errno == EINTR && palarm_fired && *palarm_fired)
262 tdb->ecode = TDB_ERR_LOCK_TIMEOUT;
263 else
264 tdb->ecode = TDB_ERR_LOCK;
265 TDB_LOG((tdb, 5,"tdb_brlock failed (fd=%d) at offset %d rw_type=%d lck_type=%d\n",
266 tdb->fd, offset, rw_type, lck_type));
268 /* Was it an alarm timeout ? */
269 if (errno == EINTR && palarm_fired && *palarm_fired) {
270 TDB_LOG((tdb, 5, "tdb_brlock timed out (fd=%d) at offset %d rw_type=%d lck_type=%d\n",
271 tdb->fd, offset, rw_type, lck_type));
272 return TDB_ERRCODE(TDB_ERR_LOCK_TIMEOUT, -1);
274 /* Otherwise - generic lock error. errno set by fcntl.
275 * EAGAIN is an expected return from non-blocking
276 * locks. */
277 if (errno != EAGAIN) {
278 TDB_LOG((tdb, 5, "tdb_brlock failed (fd=%d) at offset %d rw_type=%d lck_type=%d: %s\n",
279 tdb->fd, offset, rw_type, lck_type,
280 strerror(errno)));
282 return TDB_ERRCODE(TDB_ERR_LOCK, -1);
284 return 0;
287 /* lock a list in the database. list -1 is the alloc list */
288 static int tdb_lock(TDB_CONTEXT *tdb, int list, int ltype)
290 if (list < -1 || list >= (int)tdb->header.hash_size) {
291 TDB_LOG((tdb, 0,"tdb_lock: invalid list %d for ltype=%d\n",
292 list, ltype));
293 return -1;
295 if (tdb->flags & TDB_NOLOCK)
296 return 0;
298 /* Since fcntl locks don't nest, we do a lock for the first one,
299 and simply bump the count for future ones */
300 if (tdb->locked[list+1].count == 0) {
301 if (!tdb->read_only && tdb->header.rwlocks) {
302 if (tdb_spinlock(tdb, list, ltype)) {
303 TDB_LOG((tdb, 0, "tdb_lock spinlock failed on list %d ltype=%d\n",
304 list, ltype));
305 return -1;
307 } else if (tdb_brlock(tdb,FREELIST_TOP+4*list,ltype,F_SETLKW, 0)) {
308 TDB_LOG((tdb, 0,"tdb_lock failed on list %d ltype=%d (%s)\n",
309 list, ltype, strerror(errno)));
310 return -1;
312 tdb->locked[list+1].ltype = ltype;
314 tdb->locked[list+1].count++;
315 return 0;
318 /* unlock the database: returns void because it's too late for errors. */
319 /* changed to return int it may be interesting to know there
320 has been an error --simo */
321 static int tdb_unlock(TDB_CONTEXT *tdb, int list, int ltype)
323 int ret = -1;
325 if (tdb->flags & TDB_NOLOCK)
326 return 0;
328 /* Sanity checks */
329 if (list < -1 || list >= (int)tdb->header.hash_size) {
330 TDB_LOG((tdb, 0, "tdb_unlock: list %d invalid (%d)\n", list, tdb->header.hash_size));
331 return ret;
334 if (tdb->locked[list+1].count==0) {
335 TDB_LOG((tdb, 0, "tdb_unlock: count is 0\n"));
336 return ret;
339 if (tdb->locked[list+1].count == 1) {
340 /* Down to last nested lock: unlock underneath */
341 if (!tdb->read_only && tdb->header.rwlocks) {
342 ret = tdb_spinunlock(tdb, list, ltype);
343 } else {
344 ret = tdb_brlock(tdb, FREELIST_TOP+4*list, F_UNLCK, F_SETLKW, 0);
346 } else {
347 ret = 0;
349 tdb->locked[list+1].count--;
351 if (ret)
352 TDB_LOG((tdb, 0,"tdb_unlock: An error occurred unlocking!\n"));
353 return ret;
356 /* check for an out of bounds access - if it is out of bounds then
357 see if the database has been expanded by someone else and expand
358 if necessary
359 note that "len" is the minimum length needed for the db
361 static int tdb_oob(TDB_CONTEXT *tdb, tdb_off len, int probe)
363 struct stat st;
364 if (len <= tdb->map_size)
365 return 0;
366 if (tdb->flags & TDB_INTERNAL) {
367 if (!probe) {
368 /* Ensure ecode is set for log fn. */
369 tdb->ecode = TDB_ERR_IO;
370 TDB_LOG((tdb, 0,"tdb_oob len %d beyond internal malloc size %d\n",
371 (int)len, (int)tdb->map_size));
373 return TDB_ERRCODE(TDB_ERR_IO, -1);
376 if (fstat(tdb->fd, &st) == -1)
377 return TDB_ERRCODE(TDB_ERR_IO, -1);
379 if (st.st_size < (size_t)len) {
380 if (!probe) {
381 /* Ensure ecode is set for log fn. */
382 tdb->ecode = TDB_ERR_IO;
383 TDB_LOG((tdb, 0,"tdb_oob len %d beyond eof at %d\n",
384 (int)len, (int)st.st_size));
386 return TDB_ERRCODE(TDB_ERR_IO, -1);
389 /* Unmap, update size, remap */
390 if (tdb_munmap(tdb) == -1)
391 return TDB_ERRCODE(TDB_ERR_IO, -1);
392 tdb->map_size = st.st_size;
393 tdb_mmap(tdb);
394 return 0;
397 /* write a lump of data at a specified offset */
398 static int tdb_write(TDB_CONTEXT *tdb, tdb_off off, void *buf, tdb_len len)
400 if (tdb_oob(tdb, off + len, 0) != 0)
401 return -1;
403 if (tdb->map_ptr)
404 memcpy(off + (char *)tdb->map_ptr, buf, len);
405 #ifdef HAVE_PWRITE
406 else if (pwrite(tdb->fd, buf, len, off) != (ssize_t)len) {
407 #else
408 else if (lseek(tdb->fd, off, SEEK_SET) != off
409 || write(tdb->fd, buf, len) != (ssize_t)len) {
410 #endif
411 /* Ensure ecode is set for log fn. */
412 tdb->ecode = TDB_ERR_IO;
413 TDB_LOG((tdb, 0,"tdb_write failed at %d len=%d (%s)\n",
414 off, len, strerror(errno)));
415 return TDB_ERRCODE(TDB_ERR_IO, -1);
417 return 0;
420 /* read a lump of data at a specified offset, maybe convert */
421 static int tdb_read(TDB_CONTEXT *tdb,tdb_off off,void *buf,tdb_len len,int cv)
423 if (tdb_oob(tdb, off + len, 0) != 0)
424 return -1;
426 if (tdb->map_ptr)
427 memcpy(buf, off + (char *)tdb->map_ptr, len);
428 #ifdef HAVE_PREAD
429 else if (pread(tdb->fd, buf, len, off) != (ssize_t)len) {
430 #else
431 else if (lseek(tdb->fd, off, SEEK_SET) != off
432 || read(tdb->fd, buf, len) != (ssize_t)len) {
433 #endif
434 /* Ensure ecode is set for log fn. */
435 tdb->ecode = TDB_ERR_IO;
436 TDB_LOG((tdb, 0,"tdb_read failed at %d len=%d (%s)\n",
437 off, len, strerror(errno)));
438 return TDB_ERRCODE(TDB_ERR_IO, -1);
440 if (cv)
441 convert(buf, len);
442 return 0;
445 /* read a lump of data, allocating the space for it */
446 static char *tdb_alloc_read(TDB_CONTEXT *tdb, tdb_off offset, tdb_len len)
448 char *buf;
450 if (!(buf = malloc(len))) {
451 /* Ensure ecode is set for log fn. */
452 tdb->ecode = TDB_ERR_OOM;
453 TDB_LOG((tdb, 0,"tdb_alloc_read malloc failed len=%d (%s)\n",
454 len, strerror(errno)));
455 return TDB_ERRCODE(TDB_ERR_OOM, buf);
457 if (tdb_read(tdb, offset, buf, len, 0) == -1) {
458 SAFE_FREE(buf);
459 return NULL;
461 return buf;
464 /* read/write a tdb_off */
465 static int ofs_read(TDB_CONTEXT *tdb, tdb_off offset, tdb_off *d)
467 return tdb_read(tdb, offset, (char*)d, sizeof(*d), DOCONV());
469 static int ofs_write(TDB_CONTEXT *tdb, tdb_off offset, tdb_off *d)
471 tdb_off off = *d;
472 return tdb_write(tdb, offset, CONVERT(off), sizeof(*d));
475 /* read/write a record */
476 static int rec_read(TDB_CONTEXT *tdb, tdb_off offset, struct list_struct *rec)
478 if (tdb_read(tdb, offset, rec, sizeof(*rec),DOCONV()) == -1)
479 return -1;
480 if (TDB_BAD_MAGIC(rec)) {
481 /* Ensure ecode is set for log fn. */
482 tdb->ecode = TDB_ERR_CORRUPT;
483 TDB_LOG((tdb, 0,"rec_read bad magic 0x%x at offset=%d\n", rec->magic, offset));
484 return TDB_ERRCODE(TDB_ERR_CORRUPT, -1);
486 return tdb_oob(tdb, rec->next+sizeof(*rec), 0);
488 static int rec_write(TDB_CONTEXT *tdb, tdb_off offset, struct list_struct *rec)
490 struct list_struct r = *rec;
491 return tdb_write(tdb, offset, CONVERT(r), sizeof(r));
494 /* read a freelist record and check for simple errors */
495 static int rec_free_read(TDB_CONTEXT *tdb, tdb_off off, struct list_struct *rec)
497 if (tdb_read(tdb, off, rec, sizeof(*rec),DOCONV()) == -1)
498 return -1;
500 if (rec->magic == TDB_MAGIC) {
501 /* this happens when a app is showdown while deleting a record - we should
502 not completely fail when this happens */
503 TDB_LOG((tdb, 0,"rec_free_read non-free magic 0x%x at offset=%d - fixing\n",
504 rec->magic, off));
505 rec->magic = TDB_FREE_MAGIC;
506 if (tdb_write(tdb, off, rec, sizeof(*rec)) == -1)
507 return -1;
510 if (rec->magic != TDB_FREE_MAGIC) {
511 /* Ensure ecode is set for log fn. */
512 tdb->ecode = TDB_ERR_CORRUPT;
513 TDB_LOG((tdb, 0,"rec_free_read bad magic 0x%x at offset=%d\n",
514 rec->magic, off));
515 return TDB_ERRCODE(TDB_ERR_CORRUPT, -1);
517 if (tdb_oob(tdb, rec->next+sizeof(*rec), 0) != 0)
518 return -1;
519 return 0;
522 /* update a record tailer (must hold allocation lock) */
523 static int update_tailer(TDB_CONTEXT *tdb, tdb_off offset,
524 const struct list_struct *rec)
526 tdb_off totalsize;
528 /* Offset of tailer from record header */
529 totalsize = sizeof(*rec) + rec->rec_len;
530 return ofs_write(tdb, offset + totalsize - sizeof(tdb_off),
531 &totalsize);
534 static tdb_off tdb_dump_record(TDB_CONTEXT *tdb, tdb_off offset)
536 struct list_struct rec;
537 tdb_off tailer_ofs, tailer;
539 if (tdb_read(tdb, offset, (char *)&rec, sizeof(rec), DOCONV()) == -1) {
540 printf("ERROR: failed to read record at %u\n", offset);
541 return 0;
544 printf(" rec: offset=%u next=%d rec_len=%d key_len=%d data_len=%d full_hash=0x%x magic=0x%x\n",
545 offset, rec.next, rec.rec_len, rec.key_len, rec.data_len, rec.full_hash, rec.magic);
547 tailer_ofs = offset + sizeof(rec) + rec.rec_len - sizeof(tdb_off);
548 if (ofs_read(tdb, tailer_ofs, &tailer) == -1) {
549 printf("ERROR: failed to read tailer at %u\n", tailer_ofs);
550 return rec.next;
553 if (tailer != rec.rec_len + sizeof(rec)) {
554 printf("ERROR: tailer does not match record! tailer=%u totalsize=%u\n",
555 (unsigned)tailer, (unsigned)(rec.rec_len + sizeof(rec)));
557 return rec.next;
560 static int tdb_dump_chain(TDB_CONTEXT *tdb, int i)
562 tdb_off rec_ptr, top;
563 int hash_length = 0;
565 top = TDB_HASH_TOP(i);
567 if (tdb_lock(tdb, i, F_WRLCK) != 0)
568 return -1;
570 if (ofs_read(tdb, top, &rec_ptr) == -1)
571 return tdb_unlock(tdb, i, F_WRLCK);
573 if (rec_ptr)
574 printf("hash=%d\n", i);
576 while (rec_ptr) {
577 rec_ptr = tdb_dump_record(tdb, rec_ptr);
578 hash_length += 1;
581 printf("chain %d length %d\n", i, hash_length);
583 return tdb_unlock(tdb, i, F_WRLCK);
586 void tdb_dump_all(TDB_CONTEXT *tdb)
588 int i;
589 for (i=0;i<tdb->header.hash_size;i++) {
590 tdb_dump_chain(tdb, i);
592 tdb_printfreelist(tdb);
595 int tdb_printfreelist(TDB_CONTEXT *tdb)
597 int ret;
598 long total_free = 0;
599 tdb_off offset, rec_ptr;
600 struct list_struct rec;
602 if ((ret = tdb_lock(tdb, -1, F_WRLCK)) != 0)
603 return ret;
605 offset = FREELIST_TOP;
607 /* read in the freelist top */
608 if (ofs_read(tdb, offset, &rec_ptr) == -1) {
609 tdb_unlock(tdb, -1, F_WRLCK);
610 return 0;
613 printf("freelist top=[0x%08x]\n", rec_ptr );
614 while (rec_ptr) {
615 if (tdb_read(tdb, rec_ptr, (char *)&rec, sizeof(rec), DOCONV()) == -1) {
616 tdb_unlock(tdb, -1, F_WRLCK);
617 return -1;
620 if (rec.magic != TDB_FREE_MAGIC) {
621 printf("bad magic 0x%08x in free list\n", rec.magic);
622 tdb_unlock(tdb, -1, F_WRLCK);
623 return -1;
626 printf("entry offset=[0x%08x], rec.rec_len = [0x%08x (%d)]\n", rec.next, rec.rec_len, rec.rec_len );
627 total_free += rec.rec_len;
629 /* move to the next record */
630 rec_ptr = rec.next;
632 printf("total rec_len = [0x%08x (%d)]\n", (int)total_free,
633 (int)total_free);
635 return tdb_unlock(tdb, -1, F_WRLCK);
638 /* Remove an element from the freelist. Must have alloc lock. */
639 static int remove_from_freelist(TDB_CONTEXT *tdb, tdb_off off, tdb_off next)
641 tdb_off last_ptr, i;
643 /* read in the freelist top */
644 last_ptr = FREELIST_TOP;
645 while (ofs_read(tdb, last_ptr, &i) != -1 && i != 0) {
646 if (i == off) {
647 /* We've found it! */
648 return ofs_write(tdb, last_ptr, &next);
650 /* Follow chain (next offset is at start of record) */
651 last_ptr = i;
653 TDB_LOG((tdb, 0,"remove_from_freelist: not on list at off=%d\n", off));
654 return TDB_ERRCODE(TDB_ERR_CORRUPT, -1);
657 /* Add an element into the freelist. Merge adjacent records if
658 neccessary. */
659 static int tdb_free(TDB_CONTEXT *tdb, tdb_off offset, struct list_struct *rec)
661 tdb_off right, left;
663 /* Allocation and tailer lock */
664 if (tdb_lock(tdb, -1, F_WRLCK) != 0)
665 return -1;
667 /* set an initial tailer, so if we fail we don't leave a bogus record */
668 if (update_tailer(tdb, offset, rec) != 0) {
669 TDB_LOG((tdb, 0, "tdb_free: upfate_tailer failed!\n"));
670 goto fail;
673 /* Look right first (I'm an Australian, dammit) */
674 right = offset + sizeof(*rec) + rec->rec_len;
675 if (right + sizeof(*rec) <= tdb->map_size) {
676 struct list_struct r;
678 if (tdb_read(tdb, right, &r, sizeof(r), DOCONV()) == -1) {
679 TDB_LOG((tdb, 0, "tdb_free: right read failed at %u\n", right));
680 goto left;
683 /* If it's free, expand to include it. */
684 if (r.magic == TDB_FREE_MAGIC) {
685 if (remove_from_freelist(tdb, right, r.next) == -1) {
686 TDB_LOG((tdb, 0, "tdb_free: right free failed at %u\n", right));
687 goto left;
689 rec->rec_len += sizeof(r) + r.rec_len;
693 left:
694 /* Look left */
695 left = offset - sizeof(tdb_off);
696 if (left > TDB_DATA_START(tdb->header.hash_size)) {
697 struct list_struct l;
698 tdb_off leftsize;
700 /* Read in tailer and jump back to header */
701 if (ofs_read(tdb, left, &leftsize) == -1) {
702 TDB_LOG((tdb, 0, "tdb_free: left offset read failed at %u\n", left));
703 goto update;
705 left = offset - leftsize;
707 /* Now read in record */
708 if (tdb_read(tdb, left, &l, sizeof(l), DOCONV()) == -1) {
709 TDB_LOG((tdb, 0, "tdb_free: left read failed at %u (%u)\n", left, leftsize));
710 goto update;
713 /* If it's free, expand to include it. */
714 if (l.magic == TDB_FREE_MAGIC) {
715 if (remove_from_freelist(tdb, left, l.next) == -1) {
716 TDB_LOG((tdb, 0, "tdb_free: left free failed at %u\n", left));
717 goto update;
718 } else {
719 offset = left;
720 rec->rec_len += leftsize;
725 update:
726 if (update_tailer(tdb, offset, rec) == -1) {
727 TDB_LOG((tdb, 0, "tdb_free: update_tailer failed at %u\n", offset));
728 goto fail;
731 /* Now, prepend to free list */
732 rec->magic = TDB_FREE_MAGIC;
734 if (ofs_read(tdb, FREELIST_TOP, &rec->next) == -1 ||
735 rec_write(tdb, offset, rec) == -1 ||
736 ofs_write(tdb, FREELIST_TOP, &offset) == -1) {
737 TDB_LOG((tdb, 0, "tdb_free record write failed at offset=%d\n", offset));
738 goto fail;
741 /* And we're done. */
742 tdb_unlock(tdb, -1, F_WRLCK);
743 return 0;
745 fail:
746 tdb_unlock(tdb, -1, F_WRLCK);
747 return -1;
751 /* expand a file. we prefer to use ftruncate, as that is what posix
752 says to use for mmap expansion */
753 static int expand_file(TDB_CONTEXT *tdb, tdb_off size, tdb_off addition)
755 char buf[1024];
756 #if HAVE_FTRUNCATE_EXTEND
757 if (ftruncate(tdb->fd, size+addition) != 0) {
758 TDB_LOG((tdb, 0, "expand_file ftruncate to %d failed (%s)\n",
759 size+addition, strerror(errno)));
760 return -1;
762 #else
763 char b = 0;
765 #ifdef HAVE_PWRITE
766 if (pwrite(tdb->fd, &b, 1, (size+addition) - 1) != 1) {
767 #else
768 if (lseek(tdb->fd, (size+addition) - 1, SEEK_SET) != (size+addition) - 1 ||
769 write(tdb->fd, &b, 1) != 1) {
770 #endif
771 TDB_LOG((tdb, 0, "expand_file to %d failed (%s)\n",
772 size+addition, strerror(errno)));
773 return -1;
775 #endif
777 /* now fill the file with something. This ensures that the file isn't sparse, which would be
778 very bad if we ran out of disk. This must be done with write, not via mmap */
779 memset(buf, 0x42, sizeof(buf));
780 while (addition) {
781 int n = addition>sizeof(buf)?sizeof(buf):addition;
782 #ifdef HAVE_PWRITE
783 int ret = pwrite(tdb->fd, buf, n, size);
784 #else
785 int ret;
786 if (lseek(tdb->fd, size, SEEK_SET) != size)
787 return -1;
788 ret = write(tdb->fd, buf, n);
789 #endif
790 if (ret != n) {
791 TDB_LOG((tdb, 0, "expand_file write of %d failed (%s)\n",
792 n, strerror(errno)));
793 return -1;
795 addition -= n;
796 size += n;
798 return 0;
802 /* expand the database at least size bytes by expanding the underlying
803 file and doing the mmap again if necessary */
804 static int tdb_expand(TDB_CONTEXT *tdb, tdb_off size)
806 struct list_struct rec;
807 tdb_off offset;
809 if (tdb_lock(tdb, -1, F_WRLCK) == -1) {
810 TDB_LOG((tdb, 0, "lock failed in tdb_expand\n"));
811 return -1;
814 /* must know about any previous expansions by another process */
815 tdb_oob(tdb, tdb->map_size + 1, 1);
817 /* always make room for at least 10 more records, and round
818 the database up to a multiple of TDB_PAGE_SIZE */
819 size = TDB_ALIGN(tdb->map_size + size*10, TDB_PAGE_SIZE) - tdb->map_size;
821 if (!(tdb->flags & TDB_INTERNAL))
822 tdb_munmap(tdb);
825 * We must ensure the file is unmapped before doing this
826 * to ensure consistency with systems like OpenBSD where
827 * writes and mmaps are not consistent.
830 /* expand the file itself */
831 if (!(tdb->flags & TDB_INTERNAL)) {
832 if (expand_file(tdb, tdb->map_size, size) != 0)
833 goto fail;
836 tdb->map_size += size;
838 if (tdb->flags & TDB_INTERNAL) {
839 char *new_map_ptr = realloc(tdb->map_ptr, tdb->map_size);
840 if (!new_map_ptr) {
841 tdb->map_size -= size;
842 goto fail;
844 tdb->map_ptr = new_map_ptr;
845 } else {
847 * We must ensure the file is remapped before adding the space
848 * to ensure consistency with systems like OpenBSD where
849 * writes and mmaps are not consistent.
852 /* We're ok if the mmap fails as we'll fallback to read/write */
853 tdb_mmap(tdb);
856 /* form a new freelist record */
857 memset(&rec,'\0',sizeof(rec));
858 rec.rec_len = size - sizeof(rec);
860 /* link it into the free list */
861 offset = tdb->map_size - size;
862 if (tdb_free(tdb, offset, &rec) == -1)
863 goto fail;
865 tdb_unlock(tdb, -1, F_WRLCK);
866 return 0;
867 fail:
868 tdb_unlock(tdb, -1, F_WRLCK);
869 return -1;
872 /* allocate some space from the free list. The offset returned points
873 to a unconnected list_struct within the database with room for at
874 least length bytes of total data
876 0 is returned if the space could not be allocated
878 static tdb_off tdb_allocate(TDB_CONTEXT *tdb, tdb_len length,
879 struct list_struct *rec)
881 tdb_off rec_ptr, last_ptr, newrec_ptr;
882 struct list_struct newrec;
884 memset(&newrec, '\0', sizeof(newrec));
886 if (tdb_lock(tdb, -1, F_WRLCK) == -1)
887 return 0;
889 /* Extra bytes required for tailer */
890 length += sizeof(tdb_off);
892 again:
893 last_ptr = FREELIST_TOP;
895 /* read in the freelist top */
896 if (ofs_read(tdb, FREELIST_TOP, &rec_ptr) == -1)
897 goto fail;
899 /* keep looking until we find a freelist record big enough */
900 while (rec_ptr) {
901 if (rec_free_read(tdb, rec_ptr, rec) == -1)
902 goto fail;
904 if (rec->rec_len >= length) {
905 /* found it - now possibly split it up */
906 if (rec->rec_len > length + MIN_REC_SIZE) {
907 /* Length of left piece */
908 length = TDB_ALIGN(length, TDB_ALIGNMENT);
910 /* Right piece to go on free list */
911 newrec.rec_len = rec->rec_len
912 - (sizeof(*rec) + length);
913 newrec_ptr = rec_ptr + sizeof(*rec) + length;
915 /* And left record is shortened */
916 rec->rec_len = length;
917 } else
918 newrec_ptr = 0;
920 /* Remove allocated record from the free list */
921 if (ofs_write(tdb, last_ptr, &rec->next) == -1)
922 goto fail;
924 /* Update header: do this before we drop alloc
925 lock, otherwise tdb_free() might try to
926 merge with us, thinking we're free.
927 (Thanks Jeremy Allison). */
928 rec->magic = TDB_MAGIC;
929 if (rec_write(tdb, rec_ptr, rec) == -1)
930 goto fail;
932 /* Did we create new block? */
933 if (newrec_ptr) {
934 /* Update allocated record tailer (we
935 shortened it). */
936 if (update_tailer(tdb, rec_ptr, rec) == -1)
937 goto fail;
939 /* Free new record */
940 if (tdb_free(tdb, newrec_ptr, &newrec) == -1)
941 goto fail;
944 /* all done - return the new record offset */
945 tdb_unlock(tdb, -1, F_WRLCK);
946 return rec_ptr;
948 /* move to the next record */
949 last_ptr = rec_ptr;
950 rec_ptr = rec->next;
952 /* we didn't find enough space. See if we can expand the
953 database and if we can then try again */
954 if (tdb_expand(tdb, length + sizeof(*rec)) == 0)
955 goto again;
956 fail:
957 tdb_unlock(tdb, -1, F_WRLCK);
958 return 0;
961 /* initialise a new database with a specified hash size */
962 static int tdb_new_database(TDB_CONTEXT *tdb, int hash_size)
964 struct tdb_header *newdb;
965 int size, ret = -1;
967 /* We make it up in memory, then write it out if not internal */
968 size = sizeof(struct tdb_header) + (hash_size+1)*sizeof(tdb_off);
969 if (!(newdb = calloc(size, 1)))
970 return TDB_ERRCODE(TDB_ERR_OOM, -1);
972 /* Fill in the header */
973 newdb->version = TDB_VERSION;
974 newdb->hash_size = hash_size;
975 #ifdef USE_SPINLOCKS
976 newdb->rwlocks = size;
977 #endif
978 if (tdb->flags & TDB_INTERNAL) {
979 tdb->map_size = size;
980 tdb->map_ptr = (char *)newdb;
981 memcpy(&tdb->header, newdb, sizeof(tdb->header));
982 /* Convert the `ondisk' version if asked. */
983 CONVERT(*newdb);
984 return 0;
986 if (lseek(tdb->fd, 0, SEEK_SET) == -1)
987 goto fail;
989 if (ftruncate(tdb->fd, 0) == -1)
990 goto fail;
992 /* This creates an endian-converted header, as if read from disk */
993 CONVERT(*newdb);
994 memcpy(&tdb->header, newdb, sizeof(tdb->header));
995 /* Don't endian-convert the magic food! */
996 memcpy(newdb->magic_food, TDB_MAGIC_FOOD, strlen(TDB_MAGIC_FOOD)+1);
997 if (write(tdb->fd, newdb, size) != size)
998 ret = -1;
999 else
1000 ret = tdb_create_rwlocks(tdb->fd, hash_size);
1002 fail:
1003 SAFE_FREE(newdb);
1004 return ret;
1007 /* Returns 0 on fail. On success, return offset of record, and fills
1008 in rec */
1009 static tdb_off tdb_find(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash,
1010 struct list_struct *r)
1012 tdb_off rec_ptr;
1014 /* read in the hash top */
1015 if (ofs_read(tdb, TDB_HASH_TOP(hash), &rec_ptr) == -1)
1016 return 0;
1018 /* keep looking until we find the right record */
1019 while (rec_ptr) {
1020 if (rec_read(tdb, rec_ptr, r) == -1)
1021 return 0;
1023 if (!TDB_DEAD(r) && hash==r->full_hash && key.dsize==r->key_len) {
1024 char *k;
1025 /* a very likely hit - read the key */
1026 k = tdb_alloc_read(tdb, rec_ptr + sizeof(*r),
1027 r->key_len);
1028 if (!k)
1029 return 0;
1031 if (memcmp(key.dptr, k, key.dsize) == 0) {
1032 SAFE_FREE(k);
1033 return rec_ptr;
1035 SAFE_FREE(k);
1037 rec_ptr = r->next;
1039 return TDB_ERRCODE(TDB_ERR_NOEXIST, 0);
1042 /* As tdb_find, but if you succeed, keep the lock */
1043 static tdb_off tdb_find_lock_hash(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash, int locktype,
1044 struct list_struct *rec)
1046 u32 rec_ptr;
1048 if (tdb_lock(tdb, BUCKET(hash), locktype) == -1)
1049 return 0;
1050 if (!(rec_ptr = tdb_find(tdb, key, hash, rec)))
1051 tdb_unlock(tdb, BUCKET(hash), locktype);
1052 return rec_ptr;
1055 enum TDB_ERROR tdb_error(TDB_CONTEXT *tdb)
1057 return tdb->ecode;
1060 static struct tdb_errname {
1061 enum TDB_ERROR ecode; const char *estring;
1062 } emap[] = { {TDB_SUCCESS, "Success"},
1063 {TDB_ERR_CORRUPT, "Corrupt database"},
1064 {TDB_ERR_IO, "IO Error"},
1065 {TDB_ERR_LOCK, "Locking error"},
1066 {TDB_ERR_OOM, "Out of memory"},
1067 {TDB_ERR_EXISTS, "Record exists"},
1068 {TDB_ERR_NOLOCK, "Lock exists on other keys"},
1069 {TDB_ERR_NOEXIST, "Record does not exist"} };
1071 /* Error string for the last tdb error */
1072 const char *tdb_errorstr(TDB_CONTEXT *tdb)
1074 u32 i;
1075 for (i = 0; i < sizeof(emap) / sizeof(struct tdb_errname); i++)
1076 if (tdb->ecode == emap[i].ecode)
1077 return emap[i].estring;
1078 return "Invalid error code";
1081 /* update an entry in place - this only works if the new data size
1082 is <= the old data size and the key exists.
1083 on failure return -1.
1086 static int tdb_update_hash(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash, TDB_DATA dbuf)
1088 struct list_struct rec;
1089 tdb_off rec_ptr;
1091 /* find entry */
1092 if (!(rec_ptr = tdb_find(tdb, key, hash, &rec)))
1093 return -1;
1095 /* must be long enough key, data and tailer */
1096 if (rec.rec_len < key.dsize + dbuf.dsize + sizeof(tdb_off)) {
1097 tdb->ecode = TDB_SUCCESS; /* Not really an error */
1098 return -1;
1101 if (tdb_write(tdb, rec_ptr + sizeof(rec) + rec.key_len,
1102 dbuf.dptr, dbuf.dsize) == -1)
1103 return -1;
1105 if (dbuf.dsize != rec.data_len) {
1106 /* update size */
1107 rec.data_len = dbuf.dsize;
1108 return rec_write(tdb, rec_ptr, &rec);
1111 return 0;
1114 /* find an entry in the database given a key */
1115 /* If an entry doesn't exist tdb_err will be set to
1116 * TDB_ERR_NOEXIST. If a key has no data attached
1117 * tdb_err will not be set. Both will return a
1118 * zero pptr and zero dsize.
1121 TDB_DATA tdb_fetch(TDB_CONTEXT *tdb, TDB_DATA key)
1123 tdb_off rec_ptr;
1124 struct list_struct rec;
1125 TDB_DATA ret;
1126 u32 hash;
1128 /* find which hash bucket it is in */
1129 hash = tdb->hash_fn(&key);
1130 if (!(rec_ptr = tdb_find_lock_hash(tdb,key,hash,F_RDLCK,&rec)))
1131 return tdb_null;
1133 if (rec.data_len)
1134 ret.dptr = tdb_alloc_read(tdb, rec_ptr + sizeof(rec) + rec.key_len,
1135 rec.data_len);
1136 else
1137 ret.dptr = NULL;
1138 ret.dsize = rec.data_len;
1139 tdb_unlock(tdb, BUCKET(rec.full_hash), F_RDLCK);
1140 return ret;
1143 /* check if an entry in the database exists
1145 note that 1 is returned if the key is found and 0 is returned if not found
1146 this doesn't match the conventions in the rest of this module, but is
1147 compatible with gdbm
1149 static int tdb_exists_hash(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash)
1151 struct list_struct rec;
1153 if (tdb_find_lock_hash(tdb, key, hash, F_RDLCK, &rec) == 0)
1154 return 0;
1155 tdb_unlock(tdb, BUCKET(rec.full_hash), F_RDLCK);
1156 return 1;
1159 int tdb_exists(TDB_CONTEXT *tdb, TDB_DATA key)
1161 u32 hash = tdb->hash_fn(&key);
1162 return tdb_exists_hash(tdb, key, hash);
1165 /* record lock stops delete underneath */
1166 static int lock_record(TDB_CONTEXT *tdb, tdb_off off)
1168 return off ? tdb_brlock(tdb, off, F_RDLCK, F_SETLKW, 0) : 0;
1171 Write locks override our own fcntl readlocks, so check it here.
1172 Note this is meant to be F_SETLK, *not* F_SETLKW, as it's not
1173 an error to fail to get the lock here.
1176 static int write_lock_record(TDB_CONTEXT *tdb, tdb_off off)
1178 struct tdb_traverse_lock *i;
1179 for (i = &tdb->travlocks; i; i = i->next)
1180 if (i->off == off)
1181 return -1;
1182 return tdb_brlock(tdb, off, F_WRLCK, F_SETLK, 1);
1186 Note this is meant to be F_SETLK, *not* F_SETLKW, as it's not
1187 an error to fail to get the lock here.
1190 static int write_unlock_record(TDB_CONTEXT *tdb, tdb_off off)
1192 return tdb_brlock(tdb, off, F_UNLCK, F_SETLK, 0);
1194 /* fcntl locks don't stack: avoid unlocking someone else's */
1195 static int unlock_record(TDB_CONTEXT *tdb, tdb_off off)
1197 struct tdb_traverse_lock *i;
1198 u32 count = 0;
1200 if (off == 0)
1201 return 0;
1202 for (i = &tdb->travlocks; i; i = i->next)
1203 if (i->off == off)
1204 count++;
1205 return (count == 1 ? tdb_brlock(tdb, off, F_UNLCK, F_SETLKW, 0) : 0);
1208 /* actually delete an entry in the database given the offset */
1209 static int do_delete(TDB_CONTEXT *tdb, tdb_off rec_ptr, struct list_struct*rec)
1211 tdb_off last_ptr, i;
1212 struct list_struct lastrec;
1214 if (tdb->read_only) return -1;
1216 if (write_lock_record(tdb, rec_ptr) == -1) {
1217 /* Someone traversing here: mark it as dead */
1218 rec->magic = TDB_DEAD_MAGIC;
1219 return rec_write(tdb, rec_ptr, rec);
1221 if (write_unlock_record(tdb, rec_ptr) != 0)
1222 return -1;
1224 /* find previous record in hash chain */
1225 if (ofs_read(tdb, TDB_HASH_TOP(rec->full_hash), &i) == -1)
1226 return -1;
1227 for (last_ptr = 0; i != rec_ptr; last_ptr = i, i = lastrec.next)
1228 if (rec_read(tdb, i, &lastrec) == -1)
1229 return -1;
1231 /* unlink it: next ptr is at start of record. */
1232 if (last_ptr == 0)
1233 last_ptr = TDB_HASH_TOP(rec->full_hash);
1234 if (ofs_write(tdb, last_ptr, &rec->next) == -1)
1235 return -1;
1237 /* recover the space */
1238 if (tdb_free(tdb, rec_ptr, rec) == -1)
1239 return -1;
1240 return 0;
1243 /* Uses traverse lock: 0 = finish, -1 = error, other = record offset */
1244 static int tdb_next_lock(TDB_CONTEXT *tdb, struct tdb_traverse_lock *tlock,
1245 struct list_struct *rec)
1247 int want_next = (tlock->off != 0);
1249 /* Lock each chain from the start one. */
1250 for (; tlock->hash < tdb->header.hash_size; tlock->hash++) {
1252 /* this is an optimisation for the common case where
1253 the hash chain is empty, which is particularly
1254 common for the use of tdb with ldb, where large
1255 hashes are used. In that case we spend most of our
1256 time in tdb_brlock(), locking empty hash chains.
1258 To avoid this, we do an unlocked pre-check to see
1259 if the hash chain is empty before starting to look
1260 inside it. If it is empty then we can avoid that
1261 hash chain. If it isn't empty then we can't believe
1262 the value we get back, as we read it without a
1263 lock, so instead we get the lock and re-fetch the
1264 value below.
1266 Notice that not doing this optimisation on the
1267 first hash chain is critical. We must guarantee
1268 that we have done at least one fcntl lock at the
1269 start of a search to guarantee that memory is
1270 coherent on SMP systems. If records are added by
1271 others during the search then thats OK, and we
1272 could possibly miss those with this trick, but we
1273 could miss them anyway without this trick, so the
1274 semantics don't change.
1276 With a non-indexed ldb search this trick gains us a
1277 factor of around 80 in speed on a linux 2.6.x
1278 system (testing using ldbtest).
1280 if (!tlock->off && tlock->hash != 0) {
1281 u32 off;
1282 if (tdb->map_ptr) {
1283 for (;tlock->hash < tdb->header.hash_size;tlock->hash++) {
1284 if (0 != *(u32 *)(TDB_HASH_TOP(tlock->hash) + (unsigned char *)tdb->map_ptr)) {
1285 break;
1288 if (tlock->hash == tdb->header.hash_size) {
1289 continue;
1291 } else {
1292 if (ofs_read(tdb, TDB_HASH_TOP(tlock->hash), &off) == 0 &&
1293 off == 0) {
1294 continue;
1299 if (tdb_lock(tdb, tlock->hash, F_WRLCK) == -1)
1300 return -1;
1302 /* No previous record? Start at top of chain. */
1303 if (!tlock->off) {
1304 if (ofs_read(tdb, TDB_HASH_TOP(tlock->hash),
1305 &tlock->off) == -1)
1306 goto fail;
1307 } else {
1308 /* Otherwise unlock the previous record. */
1309 if (unlock_record(tdb, tlock->off) != 0)
1310 goto fail;
1313 if (want_next) {
1314 /* We have offset of old record: grab next */
1315 if (rec_read(tdb, tlock->off, rec) == -1)
1316 goto fail;
1317 tlock->off = rec->next;
1320 /* Iterate through chain */
1321 while( tlock->off) {
1322 tdb_off current;
1323 if (rec_read(tdb, tlock->off, rec) == -1)
1324 goto fail;
1326 /* Detect infinite loops. From "Shlomi Yaakobovich" <Shlomi@exanet.com>. */
1327 if (tlock->off == rec->next) {
1328 TDB_LOG((tdb, 0, "tdb_next_lock: loop detected.\n"));
1329 goto fail;
1332 if (!TDB_DEAD(rec)) {
1333 /* Woohoo: we found one! */
1334 if (lock_record(tdb, tlock->off) != 0)
1335 goto fail;
1336 return tlock->off;
1339 /* Try to clean dead ones from old traverses */
1340 current = tlock->off;
1341 tlock->off = rec->next;
1342 if (!tdb->read_only &&
1343 do_delete(tdb, current, rec) != 0)
1344 goto fail;
1346 tdb_unlock(tdb, tlock->hash, F_WRLCK);
1347 want_next = 0;
1349 /* We finished iteration without finding anything */
1350 return TDB_ERRCODE(TDB_SUCCESS, 0);
1352 fail:
1353 tlock->off = 0;
1354 if (tdb_unlock(tdb, tlock->hash, F_WRLCK) != 0)
1355 TDB_LOG((tdb, 0, "tdb_next_lock: On error unlock failed!\n"));
1356 return -1;
1359 /* traverse the entire database - calling fn(tdb, key, data) on each element.
1360 return -1 on error or the record count traversed
1361 if fn is NULL then it is not called
1362 a non-zero return value from fn() indicates that the traversal should stop
1364 int tdb_traverse(TDB_CONTEXT *tdb, tdb_traverse_func fn, void *private_val)
1366 TDB_DATA key, dbuf;
1367 struct list_struct rec;
1368 struct tdb_traverse_lock tl = { NULL, 0, 0 };
1369 int ret, count = 0;
1371 /* This was in the initializaton, above, but the IRIX compiler
1372 * did not like it. crh
1374 tl.next = tdb->travlocks.next;
1376 /* fcntl locks don't stack: beware traverse inside traverse */
1377 tdb->travlocks.next = &tl;
1379 /* tdb_next_lock places locks on the record returned, and its chain */
1380 while ((ret = tdb_next_lock(tdb, &tl, &rec)) > 0) {
1381 count++;
1382 /* now read the full record */
1383 key.dptr = tdb_alloc_read(tdb, tl.off + sizeof(rec),
1384 rec.key_len + rec.data_len);
1385 if (!key.dptr) {
1386 ret = -1;
1387 if (tdb_unlock(tdb, tl.hash, F_WRLCK) != 0)
1388 goto out;
1389 if (unlock_record(tdb, tl.off) != 0)
1390 TDB_LOG((tdb, 0, "tdb_traverse: key.dptr == NULL and unlock_record failed!\n"));
1391 goto out;
1393 key.dsize = rec.key_len;
1394 dbuf.dptr = key.dptr + rec.key_len;
1395 dbuf.dsize = rec.data_len;
1397 /* Drop chain lock, call out */
1398 if (tdb_unlock(tdb, tl.hash, F_WRLCK) != 0) {
1399 ret = -1;
1400 SAFE_FREE(key.dptr);
1401 goto out;
1403 if (fn && fn(tdb, key, dbuf, private_val)) {
1404 /* They want us to terminate traversal */
1405 ret = count;
1406 if (unlock_record(tdb, tl.off) != 0) {
1407 TDB_LOG((tdb, 0, "tdb_traverse: unlock_record failed!\n"));;
1408 ret = -1;
1410 SAFE_FREE(key.dptr);
1411 goto out;
1413 SAFE_FREE(key.dptr);
1415 out:
1416 tdb->travlocks.next = tl.next;
1417 if (ret < 0)
1418 return -1;
1419 else
1420 return count;
1423 /* find the first entry in the database and return its key */
1424 TDB_DATA tdb_firstkey(TDB_CONTEXT *tdb)
1426 TDB_DATA key;
1427 struct list_struct rec;
1429 /* release any old lock */
1430 if (unlock_record(tdb, tdb->travlocks.off) != 0)
1431 return tdb_null;
1432 tdb->travlocks.off = tdb->travlocks.hash = 0;
1434 if (tdb_next_lock(tdb, &tdb->travlocks, &rec) <= 0)
1435 return tdb_null;
1436 /* now read the key */
1437 key.dsize = rec.key_len;
1438 key.dptr =tdb_alloc_read(tdb,tdb->travlocks.off+sizeof(rec),key.dsize);
1439 if (tdb_unlock(tdb, BUCKET(tdb->travlocks.hash), F_WRLCK) != 0)
1440 TDB_LOG((tdb, 0, "tdb_firstkey: error occurred while tdb_unlocking!\n"));
1441 return key;
1444 /* find the next entry in the database, returning its key */
1445 TDB_DATA tdb_nextkey(TDB_CONTEXT *tdb, TDB_DATA oldkey)
1447 u32 oldhash;
1448 TDB_DATA key = tdb_null;
1449 struct list_struct rec;
1450 char *k = NULL;
1452 /* Is locked key the old key? If so, traverse will be reliable. */
1453 if (tdb->travlocks.off) {
1454 if (tdb_lock(tdb,tdb->travlocks.hash,F_WRLCK))
1455 return tdb_null;
1456 if (rec_read(tdb, tdb->travlocks.off, &rec) == -1
1457 || !(k = tdb_alloc_read(tdb,tdb->travlocks.off+sizeof(rec),
1458 rec.key_len))
1459 || memcmp(k, oldkey.dptr, oldkey.dsize) != 0) {
1460 /* No, it wasn't: unlock it and start from scratch */
1461 if (unlock_record(tdb, tdb->travlocks.off) != 0) {
1462 SAFE_FREE(k);
1463 return tdb_null;
1465 if (tdb_unlock(tdb, tdb->travlocks.hash, F_WRLCK) != 0) {
1466 SAFE_FREE(k);
1467 return tdb_null;
1469 tdb->travlocks.off = 0;
1472 SAFE_FREE(k);
1475 if (!tdb->travlocks.off) {
1476 /* No previous element: do normal find, and lock record */
1477 tdb->travlocks.off = tdb_find_lock_hash(tdb, oldkey, tdb->hash_fn(&oldkey), F_WRLCK, &rec);
1478 if (!tdb->travlocks.off)
1479 return tdb_null;
1480 tdb->travlocks.hash = BUCKET(rec.full_hash);
1481 if (lock_record(tdb, tdb->travlocks.off) != 0) {
1482 TDB_LOG((tdb, 0, "tdb_nextkey: lock_record failed (%s)!\n", strerror(errno)));
1483 return tdb_null;
1486 oldhash = tdb->travlocks.hash;
1488 /* Grab next record: locks chain and returned record,
1489 unlocks old record */
1490 if (tdb_next_lock(tdb, &tdb->travlocks, &rec) > 0) {
1491 key.dsize = rec.key_len;
1492 key.dptr = tdb_alloc_read(tdb, tdb->travlocks.off+sizeof(rec),
1493 key.dsize);
1494 /* Unlock the chain of this new record */
1495 if (tdb_unlock(tdb, tdb->travlocks.hash, F_WRLCK) != 0)
1496 TDB_LOG((tdb, 0, "tdb_nextkey: WARNING tdb_unlock failed!\n"));
1498 /* Unlock the chain of old record */
1499 if (tdb_unlock(tdb, BUCKET(oldhash), F_WRLCK) != 0)
1500 TDB_LOG((tdb, 0, "tdb_nextkey: WARNING tdb_unlock failed!\n"));
1501 return key;
1504 /* delete an entry in the database given a key */
1505 static int tdb_delete_hash(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash)
1507 tdb_off rec_ptr;
1508 struct list_struct rec;
1509 int ret;
1511 if (!(rec_ptr = tdb_find_lock_hash(tdb, key, hash, F_WRLCK, &rec)))
1512 return -1;
1513 ret = do_delete(tdb, rec_ptr, &rec);
1514 if (tdb_unlock(tdb, BUCKET(rec.full_hash), F_WRLCK) != 0)
1515 TDB_LOG((tdb, 0, "tdb_delete: WARNING tdb_unlock failed!\n"));
1516 return ret;
1519 int tdb_delete(TDB_CONTEXT *tdb, TDB_DATA key)
1521 u32 hash = tdb->hash_fn(&key);
1522 return tdb_delete_hash(tdb, key, hash);
1525 /* store an element in the database, replacing any existing element
1526 with the same key
1528 return 0 on success, -1 on failure
1530 int tdb_store(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA dbuf, int flag)
1532 struct list_struct rec;
1533 u32 hash;
1534 tdb_off rec_ptr;
1535 char *p = NULL;
1536 int ret = 0;
1538 /* find which hash bucket it is in */
1539 hash = tdb->hash_fn(&key);
1540 if (tdb_lock(tdb, BUCKET(hash), F_WRLCK) == -1)
1541 return -1;
1543 /* check for it existing, on insert. */
1544 if (flag == TDB_INSERT) {
1545 if (tdb_exists_hash(tdb, key, hash)) {
1546 tdb->ecode = TDB_ERR_EXISTS;
1547 goto fail;
1549 } else {
1550 /* first try in-place update, on modify or replace. */
1551 if (tdb_update_hash(tdb, key, hash, dbuf) == 0)
1552 goto out;
1553 if (tdb->ecode == TDB_ERR_NOEXIST &&
1554 flag == TDB_MODIFY) {
1555 /* if the record doesn't exist and we are in TDB_MODIFY mode then
1556 we should fail the store */
1557 goto fail;
1560 /* reset the error code potentially set by the tdb_update() */
1561 tdb->ecode = TDB_SUCCESS;
1563 /* delete any existing record - if it doesn't exist we don't
1564 care. Doing this first reduces fragmentation, and avoids
1565 coalescing with `allocated' block before it's updated. */
1566 if (flag != TDB_INSERT)
1567 tdb_delete_hash(tdb, key, hash);
1569 /* Copy key+value *before* allocating free space in case malloc
1570 fails and we are left with a dead spot in the tdb. */
1572 if (!(p = (char *)malloc(key.dsize + dbuf.dsize))) {
1573 tdb->ecode = TDB_ERR_OOM;
1574 goto fail;
1577 memcpy(p, key.dptr, key.dsize);
1578 if (dbuf.dsize)
1579 memcpy(p+key.dsize, dbuf.dptr, dbuf.dsize);
1581 /* we have to allocate some space */
1582 if (!(rec_ptr = tdb_allocate(tdb, key.dsize + dbuf.dsize, &rec)))
1583 goto fail;
1585 /* Read hash top into next ptr */
1586 if (ofs_read(tdb, TDB_HASH_TOP(hash), &rec.next) == -1)
1587 goto fail;
1589 rec.key_len = key.dsize;
1590 rec.data_len = dbuf.dsize;
1591 rec.full_hash = hash;
1592 rec.magic = TDB_MAGIC;
1594 /* write out and point the top of the hash chain at it */
1595 if (rec_write(tdb, rec_ptr, &rec) == -1
1596 || tdb_write(tdb, rec_ptr+sizeof(rec), p, key.dsize+dbuf.dsize)==-1
1597 || ofs_write(tdb, TDB_HASH_TOP(hash), &rec_ptr) == -1) {
1598 /* Need to tdb_unallocate() here */
1599 goto fail;
1601 out:
1602 SAFE_FREE(p);
1603 tdb_unlock(tdb, BUCKET(hash), F_WRLCK);
1604 return ret;
1605 fail:
1606 ret = -1;
1607 goto out;
1610 /* Attempt to append data to an entry in place - this only works if the new data size
1611 is <= the old data size and the key exists.
1612 on failure return -1. Record must be locked before calling.
1614 static int tdb_append_inplace(TDB_CONTEXT *tdb, TDB_DATA key, u32 hash, TDB_DATA new_dbuf)
1616 struct list_struct rec;
1617 tdb_off rec_ptr;
1619 /* find entry */
1620 if (!(rec_ptr = tdb_find(tdb, key, hash, &rec)))
1621 return -1;
1623 /* Append of 0 is always ok. */
1624 if (new_dbuf.dsize == 0)
1625 return 0;
1627 /* must be long enough for key, old data + new data and tailer */
1628 if (rec.rec_len < key.dsize + rec.data_len + new_dbuf.dsize + sizeof(tdb_off)) {
1629 /* No room. */
1630 tdb->ecode = TDB_SUCCESS; /* Not really an error */
1631 return -1;
1634 if (tdb_write(tdb, rec_ptr + sizeof(rec) + rec.key_len + rec.data_len,
1635 new_dbuf.dptr, new_dbuf.dsize) == -1)
1636 return -1;
1638 /* update size */
1639 rec.data_len += new_dbuf.dsize;
1640 return rec_write(tdb, rec_ptr, &rec);
1643 /* Append to an entry. Create if not exist. */
1645 int tdb_append(TDB_CONTEXT *tdb, TDB_DATA key, TDB_DATA new_dbuf)
1647 struct list_struct rec;
1648 u32 hash;
1649 tdb_off rec_ptr;
1650 char *p = NULL;
1651 int ret = 0;
1652 size_t new_data_size = 0;
1654 /* find which hash bucket it is in */
1655 hash = tdb->hash_fn(&key);
1656 if (tdb_lock(tdb, BUCKET(hash), F_WRLCK) == -1)
1657 return -1;
1659 /* first try in-place. */
1660 if (tdb_append_inplace(tdb, key, hash, new_dbuf) == 0)
1661 goto out;
1663 /* reset the error code potentially set by the tdb_append_inplace() */
1664 tdb->ecode = TDB_SUCCESS;
1666 /* find entry */
1667 if (!(rec_ptr = tdb_find(tdb, key, hash, &rec))) {
1668 if (tdb->ecode != TDB_ERR_NOEXIST)
1669 goto fail;
1671 /* Not found - create. */
1673 ret = tdb_store(tdb, key, new_dbuf, TDB_INSERT);
1674 goto out;
1677 new_data_size = rec.data_len + new_dbuf.dsize;
1679 /* Copy key+old_value+value *before* allocating free space in case malloc
1680 fails and we are left with a dead spot in the tdb. */
1682 if (!(p = (char *)malloc(key.dsize + new_data_size))) {
1683 tdb->ecode = TDB_ERR_OOM;
1684 goto fail;
1687 /* Copy the key in place. */
1688 memcpy(p, key.dptr, key.dsize);
1690 /* Now read the old data into place. */
1691 if (rec.data_len &&
1692 tdb_read(tdb, rec_ptr + sizeof(rec) + rec.key_len, p + key.dsize, rec.data_len, 0) == -1)
1693 goto fail;
1695 /* Finally append the new data. */
1696 if (new_dbuf.dsize)
1697 memcpy(p+key.dsize+rec.data_len, new_dbuf.dptr, new_dbuf.dsize);
1699 /* delete any existing record - if it doesn't exist we don't
1700 care. Doing this first reduces fragmentation, and avoids
1701 coalescing with `allocated' block before it's updated. */
1703 tdb_delete_hash(tdb, key, hash);
1705 if (!(rec_ptr = tdb_allocate(tdb, key.dsize + new_data_size, &rec)))
1706 goto fail;
1708 /* Read hash top into next ptr */
1709 if (ofs_read(tdb, TDB_HASH_TOP(hash), &rec.next) == -1)
1710 goto fail;
1712 rec.key_len = key.dsize;
1713 rec.data_len = new_data_size;
1714 rec.full_hash = hash;
1715 rec.magic = TDB_MAGIC;
1717 /* write out and point the top of the hash chain at it */
1718 if (rec_write(tdb, rec_ptr, &rec) == -1
1719 || tdb_write(tdb, rec_ptr+sizeof(rec), p, key.dsize+new_data_size)==-1
1720 || ofs_write(tdb, TDB_HASH_TOP(hash), &rec_ptr) == -1) {
1721 /* Need to tdb_unallocate() here */
1722 goto fail;
1725 out:
1726 SAFE_FREE(p);
1727 tdb_unlock(tdb, BUCKET(hash), F_WRLCK);
1728 return ret;
1730 fail:
1731 ret = -1;
1732 goto out;
1735 static int tdb_already_open(dev_t device,
1736 ino_t ino)
1738 TDB_CONTEXT *i;
1740 for (i = tdbs; i; i = i->next) {
1741 if (i->device == device && i->inode == ino) {
1742 return 1;
1746 return 0;
1749 /* This is based on the hash algorithm from gdbm */
1750 static u32 default_tdb_hash(TDB_DATA *key)
1752 u32 value; /* Used to compute the hash value. */
1753 u32 i; /* Used to cycle through random values. */
1755 /* Set the initial value from the key size. */
1756 for (value = 0x238F13AF * key->dsize, i=0; i < key->dsize; i++)
1757 value = (value + (key->dptr[i] << (i*5 % 24)));
1759 return (1103515243 * value + 12345);
1762 /* open the database, creating it if necessary
1764 The open_flags and mode are passed straight to the open call on the
1765 database file. A flags value of O_WRONLY is invalid. The hash size
1766 is advisory, use zero for a default value.
1768 Return is NULL on error, in which case errno is also set. Don't
1769 try to call tdb_error or tdb_errname, just do strerror(errno).
1771 @param name may be NULL for internal databases. */
1772 TDB_CONTEXT *tdb_open(const char *name, int hash_size, int tdb_flags,
1773 int open_flags, mode_t mode)
1775 return tdb_open_ex(name, hash_size, tdb_flags, open_flags, mode, NULL, NULL);
1779 TDB_CONTEXT *tdb_open_ex(const char *name, int hash_size, int tdb_flags,
1780 int open_flags, mode_t mode,
1781 tdb_log_func log_fn,
1782 tdb_hash_func hash_fn)
1784 TDB_CONTEXT *tdb;
1785 struct stat st;
1786 int rev = 0, locked = 0;
1787 unsigned char *vp;
1788 u32 vertest;
1790 if (!(tdb = calloc(1, sizeof *tdb))) {
1791 /* Can't log this */
1792 errno = ENOMEM;
1793 goto fail;
1795 tdb->fd = -1;
1796 tdb->name = NULL;
1797 tdb->map_ptr = NULL;
1798 tdb->flags = tdb_flags;
1799 tdb->open_flags = open_flags;
1800 tdb->log_fn = log_fn;
1801 tdb->hash_fn = hash_fn ? hash_fn : default_tdb_hash;
1803 if ((open_flags & O_ACCMODE) == O_WRONLY) {
1804 TDB_LOG((tdb, 0, "tdb_open_ex: can't open tdb %s write-only\n",
1805 name));
1806 errno = EINVAL;
1807 goto fail;
1810 if (hash_size == 0)
1811 hash_size = DEFAULT_HASH_SIZE;
1812 if ((open_flags & O_ACCMODE) == O_RDONLY) {
1813 tdb->read_only = 1;
1814 /* read only databases don't do locking or clear if first */
1815 tdb->flags |= TDB_NOLOCK;
1816 tdb->flags &= ~TDB_CLEAR_IF_FIRST;
1819 /* internal databases don't mmap or lock, and start off cleared */
1820 if (tdb->flags & TDB_INTERNAL) {
1821 tdb->flags |= (TDB_NOLOCK | TDB_NOMMAP);
1822 tdb->flags &= ~TDB_CLEAR_IF_FIRST;
1823 if (tdb_new_database(tdb, hash_size) != 0) {
1824 TDB_LOG((tdb, 0, "tdb_open_ex: tdb_new_database failed!"));
1825 goto fail;
1827 goto internal;
1830 if ((tdb->fd = open(name, open_flags, mode)) == -1) {
1831 TDB_LOG((tdb, 5, "tdb_open_ex: could not open file %s: %s\n",
1832 name, strerror(errno)));
1833 goto fail; /* errno set by open(2) */
1836 /* ensure there is only one process initialising at once */
1837 if (tdb_brlock(tdb, GLOBAL_LOCK, F_WRLCK, F_SETLKW, 0) == -1) {
1838 TDB_LOG((tdb, 0, "tdb_open_ex: failed to get global lock on %s: %s\n",
1839 name, strerror(errno)));
1840 goto fail; /* errno set by tdb_brlock */
1843 /* we need to zero database if we are the only one with it open */
1844 if ((tdb_flags & TDB_CLEAR_IF_FIRST) &&
1845 (locked = (tdb_brlock(tdb, ACTIVE_LOCK, F_WRLCK, F_SETLK, 0) == 0))) {
1846 open_flags |= O_CREAT;
1847 if (ftruncate(tdb->fd, 0) == -1) {
1848 TDB_LOG((tdb, 0, "tdb_open_ex: "
1849 "failed to truncate %s: %s\n",
1850 name, strerror(errno)));
1851 goto fail; /* errno set by ftruncate */
1855 if (read(tdb->fd, &tdb->header, sizeof(tdb->header)) != sizeof(tdb->header)
1856 || strcmp(tdb->header.magic_food, TDB_MAGIC_FOOD) != 0
1857 || (tdb->header.version != TDB_VERSION
1858 && !(rev = (tdb->header.version==TDB_BYTEREV(TDB_VERSION))))) {
1859 /* its not a valid database - possibly initialise it */
1860 if (!(open_flags & O_CREAT) || tdb_new_database(tdb, hash_size) == -1) {
1861 errno = EIO; /* ie bad format or something */
1862 goto fail;
1864 rev = (tdb->flags & TDB_CONVERT);
1866 vp = (unsigned char *)&tdb->header.version;
1867 vertest = (((u32)vp[0]) << 24) | (((u32)vp[1]) << 16) |
1868 (((u32)vp[2]) << 8) | (u32)vp[3];
1869 tdb->flags |= (vertest==TDB_VERSION) ? TDB_BIGENDIAN : 0;
1870 if (!rev)
1871 tdb->flags &= ~TDB_CONVERT;
1872 else {
1873 tdb->flags |= TDB_CONVERT;
1874 convert(&tdb->header, sizeof(tdb->header));
1876 if (fstat(tdb->fd, &st) == -1)
1877 goto fail;
1879 /* Is it already in the open list? If so, fail. */
1880 if (tdb_already_open(st.st_dev, st.st_ino)) {
1881 TDB_LOG((tdb, 2, "tdb_open_ex: "
1882 "%s (%d,%d) is already open in this process\n",
1883 name, (int)st.st_dev, (int)st.st_ino));
1884 errno = EBUSY;
1885 goto fail;
1888 if (!(tdb->name = (char *)strdup(name))) {
1889 errno = ENOMEM;
1890 goto fail;
1893 tdb->map_size = st.st_size;
1894 tdb->device = st.st_dev;
1895 tdb->inode = st.st_ino;
1896 tdb->locked = calloc(tdb->header.hash_size+1, sizeof(tdb->locked[0]));
1897 if (!tdb->locked) {
1898 TDB_LOG((tdb, 2, "tdb_open_ex: "
1899 "failed to allocate lock structure for %s\n",
1900 name));
1901 errno = ENOMEM;
1902 goto fail;
1904 tdb_mmap(tdb);
1905 if (locked) {
1906 if (!tdb->read_only)
1907 if (tdb_clear_spinlocks(tdb) != 0) {
1908 TDB_LOG((tdb, 0, "tdb_open_ex: "
1909 "failed to clear spinlock\n"));
1910 goto fail;
1912 if (tdb_brlock(tdb, ACTIVE_LOCK, F_UNLCK, F_SETLK, 0) == -1) {
1913 TDB_LOG((tdb, 0, "tdb_open_ex: "
1914 "failed to take ACTIVE_LOCK on %s: %s\n",
1915 name, strerror(errno)));
1916 goto fail;
1921 /* We always need to do this if the CLEAR_IF_FIRST flag is set, even if
1922 we didn't get the initial exclusive lock as we need to let all other
1923 users know we're using it. */
1925 if (tdb_flags & TDB_CLEAR_IF_FIRST) {
1926 /* leave this lock in place to indicate it's in use */
1927 if (tdb_brlock(tdb, ACTIVE_LOCK, F_RDLCK, F_SETLKW, 0) == -1)
1928 goto fail;
1932 internal:
1933 /* Internal (memory-only) databases skip all the code above to
1934 * do with disk files, and resume here by releasing their
1935 * global lock and hooking into the active list. */
1936 if (tdb_brlock(tdb, GLOBAL_LOCK, F_UNLCK, F_SETLKW, 0) == -1)
1937 goto fail;
1938 tdb->next = tdbs;
1939 tdbs = tdb;
1940 return tdb;
1942 fail:
1943 { int save_errno = errno;
1945 if (!tdb)
1946 return NULL;
1948 if (tdb->map_ptr) {
1949 if (tdb->flags & TDB_INTERNAL)
1950 SAFE_FREE(tdb->map_ptr);
1951 else
1952 tdb_munmap(tdb);
1954 SAFE_FREE(tdb->name);
1955 if (tdb->fd != -1)
1956 if (close(tdb->fd) != 0)
1957 TDB_LOG((tdb, 5, "tdb_open_ex: failed to close tdb->fd on error!\n"));
1958 SAFE_FREE(tdb->locked);
1959 SAFE_FREE(tdb);
1960 errno = save_errno;
1961 return NULL;
1966 * Close a database.
1968 * @returns -1 for error; 0 for success.
1970 int tdb_close(TDB_CONTEXT *tdb)
1972 TDB_CONTEXT **i;
1973 int ret = 0;
1975 if (tdb->map_ptr) {
1976 if (tdb->flags & TDB_INTERNAL)
1977 SAFE_FREE(tdb->map_ptr);
1978 else
1979 tdb_munmap(tdb);
1981 SAFE_FREE(tdb->name);
1982 if (tdb->fd != -1)
1983 ret = close(tdb->fd);
1984 SAFE_FREE(tdb->locked);
1986 /* Remove from contexts list */
1987 for (i = &tdbs; *i; i = &(*i)->next) {
1988 if (*i == tdb) {
1989 *i = tdb->next;
1990 break;
1994 memset(tdb, 0, sizeof(*tdb));
1995 SAFE_FREE(tdb);
1997 return ret;
2000 /* lock/unlock entire database */
2001 int tdb_lockall(TDB_CONTEXT *tdb)
2003 u32 i;
2005 /* There are no locks on read-only dbs */
2006 if (tdb->read_only)
2007 return TDB_ERRCODE(TDB_ERR_LOCK, -1);
2008 for (i = 0; i < tdb->header.hash_size; i++)
2009 if (tdb_lock(tdb, i, F_WRLCK))
2010 break;
2012 /* If error, release locks we have... */
2013 if (i < tdb->header.hash_size) {
2014 u32 j;
2016 for ( j = 0; j < i; j++)
2017 tdb_unlock(tdb, j, F_WRLCK);
2018 return TDB_ERRCODE(TDB_ERR_NOLOCK, -1);
2021 return 0;
2023 void tdb_unlockall(TDB_CONTEXT *tdb)
2025 u32 i;
2026 for (i=0; i < tdb->header.hash_size; i++)
2027 tdb_unlock(tdb, i, F_WRLCK);
2030 /* lock/unlock one hash chain. This is meant to be used to reduce
2031 contention - it cannot guarantee how many records will be locked */
2032 int tdb_chainlock(TDB_CONTEXT *tdb, TDB_DATA key)
2034 return tdb_lock(tdb, BUCKET(tdb->hash_fn(&key)), F_WRLCK);
2037 int tdb_chainunlock(TDB_CONTEXT *tdb, TDB_DATA key)
2039 return tdb_unlock(tdb, BUCKET(tdb->hash_fn(&key)), F_WRLCK);
2042 int tdb_chainlock_read(TDB_CONTEXT *tdb, TDB_DATA key)
2044 return tdb_lock(tdb, BUCKET(tdb->hash_fn(&key)), F_RDLCK);
2047 int tdb_chainunlock_read(TDB_CONTEXT *tdb, TDB_DATA key)
2049 return tdb_unlock(tdb, BUCKET(tdb->hash_fn(&key)), F_RDLCK);
2053 /* register a loging function */
2054 void tdb_logging_function(TDB_CONTEXT *tdb, void (*fn)(TDB_CONTEXT *, int , const char *, ...))
2056 tdb->log_fn = fn;
2059 /* reopen a tdb - this can be used after a fork to ensure that we have an independent
2060 seek pointer from our parent and to re-establish locks */
2061 int tdb_reopen(TDB_CONTEXT *tdb)
2063 struct stat st;
2065 if (tdb->flags & TDB_INTERNAL)
2066 return 0; /* Nothing to do. */
2067 if (tdb_munmap(tdb) != 0) {
2068 TDB_LOG((tdb, 0, "tdb_reopen: munmap failed (%s)\n", strerror(errno)));
2069 goto fail;
2071 if (close(tdb->fd) != 0)
2072 TDB_LOG((tdb, 0, "tdb_reopen: WARNING closing tdb->fd failed!\n"));
2073 tdb->fd = open(tdb->name, tdb->open_flags & ~(O_CREAT|O_TRUNC), 0);
2074 if (tdb->fd == -1) {
2075 TDB_LOG((tdb, 0, "tdb_reopen: open failed (%s)\n", strerror(errno)));
2076 goto fail;
2078 if ((tdb->flags & TDB_CLEAR_IF_FIRST) &&
2079 (tdb_brlock(tdb, ACTIVE_LOCK, F_RDLCK, F_SETLKW, 0) == -1)) {
2080 TDB_LOG((tdb, 0, "tdb_reopen: failed to obtain active lock\n"));
2081 goto fail;
2083 if (fstat(tdb->fd, &st) != 0) {
2084 TDB_LOG((tdb, 0, "tdb_reopen: fstat failed (%s)\n", strerror(errno)));
2085 goto fail;
2087 if (st.st_ino != tdb->inode || st.st_dev != tdb->device) {
2088 TDB_LOG((tdb, 0, "tdb_reopen: file dev/inode has changed!\n"));
2089 goto fail;
2091 tdb_mmap(tdb);
2093 return 0;
2095 fail:
2096 tdb_close(tdb);
2097 return -1;
2100 /* reopen all tdb's */
2101 int tdb_reopen_all(void)
2103 TDB_CONTEXT *tdb;
2105 for (tdb=tdbs; tdb; tdb = tdb->next) {
2106 if (tdb_reopen(tdb) != 0)
2107 return -1;
2110 return 0;