diff options
author | Rusty Russell <rusty@rustcorp.com.au> | 2011-09-14 08:03:13 +0930 |
---|---|---|
committer | Rusty Russell <rusty@rustcorp.com.au> | 2011-09-14 08:03:13 +0930 |
commit | a15c1cf175d7f29a50d7b2c1acb0f67faab1b06f (patch) | |
tree | 63ed869da6d1608a97b1b6ae73984e48b5157e80 /lib/tdb2/tdb.c | |
parent | a347a4802695c41437e2966404d1e2fe2dee78b4 (diff) | |
download | samba-a15c1cf175d7f29a50d7b2c1acb0f67faab1b06f.tar.gz samba-a15c1cf175d7f29a50d7b2c1acb0f67faab1b06f.tar.bz2 samba-a15c1cf175d7f29a50d7b2c1acb0f67faab1b06f.zip |
tdb2: tdb_repack
Move the tdb1_repack() code into the core, make it general, rename to
tdb_repack().
It's generic code: copy database into temporary, wipe it, copy back.
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
(Imported from CCAN commit e487983a4099b6f760056ff7182f2ff543e6da71)
Diffstat (limited to 'lib/tdb2/tdb.c')
-rw-r--r-- | lib/tdb2/tdb.c | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/lib/tdb2/tdb.c b/lib/tdb2/tdb.c index d4036d99be..b3f74aa0b3 100644 --- a/lib/tdb2/tdb.c +++ b/lib/tdb2/tdb.c @@ -591,3 +591,64 @@ int tdb_fd(const struct tdb_context *tdb) { return tdb->file->fd; } + +struct traverse_state { + enum TDB_ERROR error; + struct tdb_context *dest_db; +}; + +/* + traverse function for repacking + */ +static int repack_traverse(struct tdb_context *tdb, TDB_DATA key, TDB_DATA data, + struct traverse_state *state) +{ + state->error = tdb_store(state->dest_db, key, data, TDB_INSERT); + if (state->error != TDB_SUCCESS) { + return -1; + } + return 0; +} + +enum TDB_ERROR tdb_repack(struct tdb_context *tdb) +{ + struct tdb_context *tmp_db; + struct traverse_state state; + + state.error = tdb_transaction_start(tdb); + if (state.error != TDB_SUCCESS) { + return state.error; + } + + tmp_db = tdb_open("tmpdb", TDB_INTERNAL, O_RDWR|O_CREAT, 0, NULL); + if (tmp_db == NULL) { + state.error = tdb_logerr(tdb, TDB_ERR_OOM, TDB_LOG_ERROR, + __location__ + " Failed to create tmp_db"); + tdb_transaction_cancel(tdb); + return tdb->last_error = state.error; + } + + state.dest_db = tmp_db; + if (tdb_traverse(tdb, repack_traverse, &state) < 0) { + goto fail; + } + + state.error = tdb_wipe_all(tdb); + if (state.error != TDB_SUCCESS) { + goto fail; + } + + state.dest_db = tdb; + if (tdb_traverse(tmp_db, repack_traverse, &state) < 0) { + goto fail; + } + + tdb_close(tmp_db); + return tdb_transaction_commit(tdb); + +fail: + tdb_transaction_cancel(tdb); + tdb_close(tmp_db); + return state.error; +} |