From 9140fca812063be69842a6c64030e32c65c9dff1 Mon Sep 17 00:00:00 2001 From: Rusty Russell Date: Wed, 14 Sep 2011 07:56:13 +0930 Subject: tdb2: unify tdb1_chainlock et al. into tdb_chainlock Switch on the TDB_VERSION1 flag. Signed-off-by: Rusty Russell (Imported from CCAN commit ceb2f3eacbad917ee990afbc3dd800bfb0607887) --- lib/tdb2/hash.c | 19 +++++++++++++++++ lib/tdb2/lock.c | 18 ++++++++++------ lib/tdb2/private.h | 7 +++++++ lib/tdb2/tdb1.h | 15 -------------- lib/tdb2/tdb1_check.c | 6 +++--- lib/tdb2/tdb1_lock.c | 26 +----------------------- lib/tdb2/tdb1_open.c | 6 +++--- lib/tdb2/tdb1_private.h | 2 -- lib/tdb2/tdb1_summary.c | 4 ++-- lib/tdb2/tdb1_tdb.c | 11 +++------- lib/tdb2/test/run-tdb1-no-lock-during-traverse.c | 14 ++++++------- 11 files changed, 57 insertions(+), 71 deletions(-) (limited to 'lib') diff --git a/lib/tdb2/hash.c b/lib/tdb2/hash.c index 56c5086e74..5eee41a88d 100644 --- a/lib/tdb2/hash.c +++ b/lib/tdb2/hash.c @@ -852,6 +852,11 @@ static enum TDB_ERROR chainlock(struct tdb_context *tdb, const TDB_DATA *key, contention - it cannot guarantee how many records will be locked */ enum TDB_ERROR tdb_chainlock(struct tdb_context *tdb, TDB_DATA key) { + if (tdb->flags & TDB_VERSION1) { + if (tdb1_chainlock(tdb, key) == -1) + return tdb->last_error; + return TDB_SUCCESS; + } return tdb->last_error = chainlock(tdb, &key, F_WRLCK, TDB_LOCK_WAIT, "tdb_chainlock"); } @@ -862,6 +867,11 @@ void tdb_chainunlock(struct tdb_context *tdb, TDB_DATA key) tdb_off_t lockstart, locksize; unsigned int group, gbits; + if (tdb->flags & TDB_VERSION1) { + tdb1_chainunlock(tdb, key); + return; + } + gbits = TDB_TOPLEVEL_HASH_BITS - TDB_HASH_GROUP_BITS; group = bits_from(h, 64 - gbits, gbits); @@ -873,6 +883,11 @@ void tdb_chainunlock(struct tdb_context *tdb, TDB_DATA key) enum TDB_ERROR tdb_chainlock_read(struct tdb_context *tdb, TDB_DATA key) { + if (tdb->flags & TDB_VERSION1) { + if (tdb1_chainlock_read(tdb, key) == -1) + return tdb->last_error; + return TDB_SUCCESS; + } return tdb->last_error = chainlock(tdb, &key, F_RDLCK, TDB_LOCK_WAIT, "tdb_chainlock_read"); } @@ -883,6 +898,10 @@ void tdb_chainunlock_read(struct tdb_context *tdb, TDB_DATA key) tdb_off_t lockstart, locksize; unsigned int group, gbits; + if (tdb->flags & TDB_VERSION1) { + tdb1_chainunlock_read(tdb, key); + return; + } gbits = TDB_TOPLEVEL_HASH_BITS - TDB_HASH_GROUP_BITS; group = bits_from(h, 64 - gbits, gbits); diff --git a/lib/tdb2/lock.c b/lib/tdb2/lock.c index bd896a35e0..7d4311c8e5 100644 --- a/lib/tdb2/lock.c +++ b/lib/tdb2/lock.c @@ -534,6 +534,12 @@ enum TDB_ERROR tdb_allrecord_lock(struct tdb_context *tdb, int ltype, enum TDB_ERROR ecode; tdb_bool_err berr; + if (tdb->flags & TDB_VERSION1) { + if (tdb1_allrecord_lock(tdb, ltype, flags, upgradable) == -1) + return tdb->last_error; + return TDB_SUCCESS; + } + if (tdb->flags & TDB_NOLOCK) return TDB_SUCCESS; @@ -648,6 +654,11 @@ void tdb_unlock_expand(struct tdb_context *tdb, int ltype) /* unlock entire db */ void tdb_allrecord_unlock(struct tdb_context *tdb, int ltype) { + if (tdb->flags & TDB_VERSION1) { + tdb1_allrecord_unlock(tdb, ltype); + return; + } + if (tdb->flags & TDB_NOLOCK) return; @@ -862,12 +873,7 @@ void tdb_lock_cleanup(struct tdb_context *tdb) while (tdb->file->allrecord_lock.count && tdb->file->allrecord_lock.owner == tdb) { - if (tdb->flags & TDB_VERSION1) - tdb1_allrecord_unlock(tdb, - tdb->file->allrecord_lock.ltype); - else - tdb_allrecord_unlock(tdb, - tdb->file->allrecord_lock.ltype); + tdb_allrecord_unlock(tdb, tdb->file->allrecord_lock.ltype); } for (i=0; ifile->num_lockrecs; i++) { diff --git a/lib/tdb2/private.h b/lib/tdb2/private.h index e981372351..4132ac2949 100644 --- a/lib/tdb2/private.h +++ b/lib/tdb2/private.h @@ -652,8 +652,15 @@ enum TDB_ERROR tdb1_open(struct tdb_context *tdb); enum TDB_ERROR tdb1_probe_length(struct tdb_context *tdb); /* tdb1_lock.c: */ +int tdb1_allrecord_lock(struct tdb_context *tdb, int ltype, + enum tdb_lock_flags flags, bool upgradable); int tdb1_allrecord_unlock(struct tdb_context *tdb, int ltype); +int tdb1_chainlock(struct tdb_context *tdb, TDB_DATA key); +int tdb1_chainunlock(struct tdb_context *tdb, TDB_DATA key); +int tdb1_chainlock_read(struct tdb_context *tdb, TDB_DATA key); +int tdb1_chainunlock_read(struct tdb_context *tdb, TDB_DATA key); + /* tdb1_transaction.c: */ int tdb1_transaction_recover(struct tdb_context *tdb); int tdb1_transaction_cancel(struct tdb_context *tdb); diff --git a/lib/tdb2/tdb1.h b/lib/tdb2/tdb1.h index 4b0e454d1b..75cf39d7e1 100644 --- a/lib/tdb2/tdb1.h +++ b/lib/tdb2/tdb1.h @@ -47,14 +47,6 @@ TDB_DATA tdb1_firstkey(struct tdb_context *tdb); TDB_DATA tdb1_nextkey(struct tdb_context *tdb, TDB_DATA key); -int tdb1_lockall(struct tdb_context *tdb); - -int tdb1_unlockall(struct tdb_context *tdb); - -int tdb1_lockall_read(struct tdb_context *tdb); - -int tdb1_unlockall_read(struct tdb_context *tdb); - int tdb1_transaction_start(struct tdb_context *tdb); int tdb1_transaction_prepare_commit(struct tdb_context *tdb); @@ -73,13 +65,6 @@ int tdb1_check(struct tdb_context *tdb, /* @} ******************************************************************/ -/* Low level locking functions: use with care */ -int tdb1_chainlock(struct tdb_context *tdb, TDB_DATA key); -int tdb1_chainunlock(struct tdb_context *tdb, TDB_DATA key); -int tdb1_chainlock_read(struct tdb_context *tdb, TDB_DATA key); -int tdb1_chainunlock_read(struct tdb_context *tdb, TDB_DATA key); - - /* wipe and repack */ int tdb1_wipe_all(struct tdb_context *tdb); int tdb1_repack(struct tdb_context *tdb); diff --git a/lib/tdb2/tdb1_check.c b/lib/tdb2/tdb1_check.c index 50e71d9783..c3c3c60f43 100644 --- a/lib/tdb2/tdb1_check.c +++ b/lib/tdb2/tdb1_check.c @@ -338,7 +338,7 @@ int tdb1_check(struct tdb_context *tdb, if (tdb->file->allrecord_lock.count != 0) { locked = false; } else { - if (tdb1_lockall_read(tdb) == -1) + if (tdb_lockall_read(tdb) != TDB_SUCCESS) return -1; locked = true; } @@ -455,7 +455,7 @@ int tdb1_check(struct tdb_context *tdb, free(hashes); if (locked) { - tdb1_unlockall_read(tdb); + tdb_unlockall_read(tdb); } return 0; @@ -463,7 +463,7 @@ free: free(hashes); unlock: if (locked) { - tdb1_unlockall_read(tdb); + tdb_unlockall_read(tdb); } return -1; } diff --git a/lib/tdb2/tdb1_lock.c b/lib/tdb2/tdb1_lock.c index 7cc17d7a7e..f21a22cacd 100644 --- a/lib/tdb2/tdb1_lock.c +++ b/lib/tdb2/tdb1_lock.c @@ -306,7 +306,7 @@ int tdb1_allrecord_lock(struct tdb_context *tdb, int ltype, } /* FIXME: Temporary cast. */ - tdb->file->allrecord_lock.owner = (void *)(struct tdb1_context *)tdb; + tdb->file->allrecord_lock.owner = (void *)(struct tdb_context *)tdb; tdb->file->allrecord_lock.count = 1; /* If it's upgradable, it's actually exclusive so we can treat * it as a write lock. */ @@ -364,30 +364,6 @@ int tdb1_allrecord_unlock(struct tdb_context *tdb, int ltype) return 0; } -/* lock entire database with write lock */ -int tdb1_lockall(struct tdb_context *tdb) -{ - return tdb1_allrecord_lock(tdb, F_WRLCK, TDB_LOCK_WAIT, false); -} - -/* unlock entire database with write lock */ -int tdb1_unlockall(struct tdb_context *tdb) -{ - return tdb1_allrecord_unlock(tdb, F_WRLCK); -} - -/* lock entire database with read lock */ -int tdb1_lockall_read(struct tdb_context *tdb) -{ - return tdb1_allrecord_lock(tdb, F_RDLCK, TDB_LOCK_WAIT, false); -} - -/* unlock entire database with read lock */ -int tdb1_unlockall_read(struct tdb_context *tdb) -{ - return tdb1_allrecord_unlock(tdb, F_RDLCK); -} - /* lock/unlock one hash chain. This is meant to be used to reduce contention - it cannot guarantee how many records will be locked */ int tdb1_chainlock(struct tdb_context *tdb, TDB_DATA key) diff --git a/lib/tdb2/tdb1_open.c b/lib/tdb2/tdb1_open.c index 6e0c25db05..df46450c90 100644 --- a/lib/tdb2/tdb1_open.c +++ b/lib/tdb2/tdb1_open.c @@ -42,7 +42,7 @@ void tdb1_header_hash(struct tdb_context *tdb, *magic1_hash = 1; } -static void tdb1_context_init(struct tdb_context *tdb) +static void tdb_context_init(struct tdb_context *tdb) { assert(tdb->flags & TDB_VERSION1); @@ -71,7 +71,7 @@ enum TDB_ERROR tdb1_new_database(struct tdb_context *tdb, int hash_size = TDB1_DEFAULT_HASH_SIZE; enum TDB_ERROR ret = TDB_ERR_IO; - tdb1_context_init(tdb); + tdb_context_init(tdb); /* Default TDB2 hash becomes default TDB1 hash. */ if (tdb->hash_fn == tdb_jenkins_hash) @@ -171,7 +171,7 @@ enum TDB_ERROR tdb1_open(struct tdb_context *tdb) tdb->flags |= TDB_VERSION1; - tdb1_context_init(tdb); + tdb_context_init(tdb); /* Default TDB2 hash becomes default TDB1 hash. */ if (tdb->hash_fn == tdb_jenkins_hash) { diff --git a/lib/tdb2/tdb1_private.h b/lib/tdb2/tdb1_private.h index 2e292ee0bc..13c51df9f3 100644 --- a/lib/tdb2/tdb1_private.h +++ b/lib/tdb2/tdb1_private.h @@ -142,8 +142,6 @@ int tdb1_recovery_area(struct tdb_context *tdb, const struct tdb1_methods *methods, tdb1_off_t *recovery_offset, struct tdb1_record *rec); -int tdb1_allrecord_lock(struct tdb_context *tdb, int ltype, - enum tdb_lock_flags flags, bool upgradable); int tdb1_allrecord_upgrade(struct tdb_context *tdb); int tdb1_write_lock_record(struct tdb_context *tdb, tdb1_off_t off); int tdb1_write_unlock_record(struct tdb_context *tdb, tdb1_off_t off); diff --git a/lib/tdb2/tdb1_summary.c b/lib/tdb2/tdb1_summary.c index d7b498df89..b74b8f4474 100644 --- a/lib/tdb2/tdb1_summary.c +++ b/lib/tdb2/tdb1_summary.c @@ -98,7 +98,7 @@ char *tdb1_summary(struct tdb_context *tdb) if (tdb->file->allrecord_lock.count != 0) { locked = false; } else { - if (tdb1_lockall_read(tdb) == -1) + if (tdb_lockall_read(tdb) != TDB_SUCCESS) return NULL; locked = true; } @@ -196,7 +196,7 @@ char *tdb1_summary(struct tdb_context *tdb) unlock: if (locked) { - tdb1_unlockall_read(tdb); + tdb_unlockall_read(tdb); } return ret; } diff --git a/lib/tdb2/tdb1_tdb.c b/lib/tdb2/tdb1_tdb.c index d7f518dce4..0a4f8b5503 100644 --- a/lib/tdb2/tdb1_tdb.c +++ b/lib/tdb2/tdb1_tdb.c @@ -732,7 +732,7 @@ int tdb1_wipe_all(struct tdb_context *tdb) tdb1_off_t recovery_head; tdb1_len_t recovery_size = 0; - if (tdb1_lockall(tdb) != 0) { + if (tdb_lockall(tdb) != TDB_SUCCESS) { return -1; } @@ -801,16 +801,11 @@ int tdb1_wipe_all(struct tdb_context *tdb) } } - if (tdb1_unlockall(tdb) != 0) { - tdb_logerr(tdb, tdb->last_error, TDB_LOG_ERROR, - "tdb1_wipe_all: failed to unlock"); - goto failed; - } - + tdb_unlockall(tdb); return 0; failed: - tdb1_unlockall(tdb); + tdb_unlockall(tdb); return -1; } diff --git a/lib/tdb2/test/run-tdb1-no-lock-during-traverse.c b/lib/tdb2/test/run-tdb1-no-lock-during-traverse.c index 71404c2d75..b2b7a781db 100644 --- a/lib/tdb2/test/run-tdb1-no-lock-during-traverse.c +++ b/lib/tdb2/test/run-tdb1-no-lock-during-traverse.c @@ -74,7 +74,7 @@ int main(int argc, char *argv[]) hsize.base.next = &tap_log_attr; hsize.tdb1_hashsize.hsize = 1024; - plan_tests(43); + plan_tests(40); tdb = tdb_open("run-no-lock-during-traverse.tdb1", TDB_VERSION1, O_CREAT|O_TRUNC|O_RDWR, 0600, &hsize); @@ -82,28 +82,28 @@ int main(int argc, char *argv[]) ok1(tdb); ok1(prepare_entries(tdb)); ok1(locking_errors1 == 0); - ok1(tdb1_lockall(tdb) == 0); + ok1(tdb_lockall(tdb) == 0); ok1(locking_errors1 == 0); ok1(tdb_traverse(tdb, delete_other, &errors) >= 0); ok1(errors == 0); ok1(locking_errors1 == 0); - ok1(tdb1_unlockall(tdb) == 0); + tdb_unlockall(tdb); ok1(prepare_entries(tdb)); ok1(locking_errors1 == 0); - ok1(tdb1_lockall(tdb) == 0); + ok1(tdb_lockall(tdb) == 0); ok1(locking_errors1 == 0); ok1(tdb_traverse(tdb, delete_self, NULL) == NUM_ENTRIES); ok1(locking_errors1 == 0); - ok1(tdb1_unlockall(tdb) == 0); + tdb_unlockall(tdb); ok1(prepare_entries(tdb)); ok1(locking_errors1 == 0); - ok1(tdb1_lockall(tdb) == 0); + ok1(tdb_lockall(tdb) == 0); ok1(locking_errors1 == 0); delete_entries(tdb); ok1(locking_errors1 == 0); - ok1(tdb1_unlockall(tdb) == 0); + tdb_unlockall(tdb); ok1(tdb_close(tdb) == 0); -- cgit