mirror of
https://github.com/samba-team/samba.git
synced 2024-12-23 17:34:34 +03:00
1b8c07ac7d
Signed-off-by: Volker Lendecke <vl@samba.org> Reviewed-by: Ira Cooper <ira@samba.org>
189 lines
4.7 KiB
C
189 lines
4.7 KiB
C
#include "../common/tdb_private.h"
|
|
#include "../common/io.c"
|
|
#include "../common/tdb.c"
|
|
#include "../common/lock.c"
|
|
#include "../common/freelist.c"
|
|
#include "../common/traverse.c"
|
|
#include "../common/transaction.c"
|
|
#include "../common/error.c"
|
|
#include "../common/open.c"
|
|
#include "../common/check.c"
|
|
#include "../common/hash.c"
|
|
#include "../common/mutex.c"
|
|
#include "tap-interface.h"
|
|
#include <stdlib.h>
|
|
|
|
static unsigned int tdb_dumb_hash(TDB_DATA *key)
|
|
{
|
|
return key->dsize;
|
|
}
|
|
|
|
static void log_fn(struct tdb_context *tdb, enum tdb_debug_level level, const char *fmt, ...)
|
|
{
|
|
unsigned int *count = tdb_get_logging_private(tdb);
|
|
if (strstr(fmt, "hash"))
|
|
(*count)++;
|
|
}
|
|
|
|
static unsigned int hdr_rwlocks(const char *fname)
|
|
{
|
|
struct tdb_header hdr;
|
|
ssize_t nread;
|
|
|
|
int fd = open(fname, O_RDONLY);
|
|
if (fd == -1)
|
|
return -1;
|
|
|
|
nread = read(fd, &hdr, sizeof(hdr));
|
|
close(fd);
|
|
if (nread != sizeof(hdr)) {
|
|
return -1;
|
|
}
|
|
return hdr.rwlocks;
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
struct tdb_context *tdb;
|
|
unsigned int log_count, flags;
|
|
TDB_DATA d, r;
|
|
struct tdb_logging_context log_ctx = { log_fn, &log_count };
|
|
|
|
plan_tests(38 * 2);
|
|
|
|
for (flags = 0; flags <= TDB_CONVERT; flags += TDB_CONVERT) {
|
|
unsigned int rwmagic = TDB_HASH_RWLOCK_MAGIC;
|
|
|
|
if (flags & TDB_CONVERT)
|
|
tdb_convert(&rwmagic, sizeof(rwmagic));
|
|
|
|
/* Create an old-style hash. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, flags,
|
|
O_CREAT|O_RDWR|O_TRUNC, 0600, &log_ctx,
|
|
NULL);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
d.dptr = discard_const_p(uint8_t, "Hello");
|
|
d.dsize = 5;
|
|
ok1(tdb_store(tdb, d, d, TDB_INSERT) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* Should not have marked rwlocks field. */
|
|
ok1(hdr_rwlocks("run-incompatible.tdb") == 0);
|
|
|
|
/* We can still open any old-style with incompat flag. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0,
|
|
TDB_INCOMPATIBLE_HASH,
|
|
O_RDWR, 0600, &log_ctx, NULL);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
r = tdb_fetch(tdb, d);
|
|
ok1(r.dsize == 5);
|
|
free(r.dptr);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("test/jenkins-le-hash.tdb", 0, 0, O_RDONLY,
|
|
0, &log_ctx, tdb_jenkins_hash);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("test/jenkins-be-hash.tdb", 0, 0, O_RDONLY,
|
|
0, &log_ctx, tdb_jenkins_hash);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* OK, now create with incompatible flag, default hash. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0,
|
|
flags|TDB_INCOMPATIBLE_HASH,
|
|
O_CREAT|O_RDWR|O_TRUNC, 0600, &log_ctx,
|
|
NULL);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
d.dptr = discard_const_p(uint8_t, "Hello");
|
|
d.dsize = 5;
|
|
ok1(tdb_store(tdb, d, d, TDB_INSERT) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* Should have marked rwlocks field. */
|
|
ok1(hdr_rwlocks("run-incompatible.tdb") == rwmagic);
|
|
|
|
/* Cannot open with old hash. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, 0,
|
|
O_RDWR, 0600, &log_ctx, tdb_old_hash);
|
|
ok1(!tdb);
|
|
ok1(log_count == 1);
|
|
|
|
/* Can open with jenkins hash. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, 0,
|
|
O_RDWR, 0600, &log_ctx, tdb_jenkins_hash);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
r = tdb_fetch(tdb, d);
|
|
ok1(r.dsize == 5);
|
|
free(r.dptr);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* Can open by letting it figure it out itself. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, 0,
|
|
O_RDWR, 0600, &log_ctx, NULL);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
r = tdb_fetch(tdb, d);
|
|
ok1(r.dsize == 5);
|
|
free(r.dptr);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* We can also use incompatible hash with other hashes. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0,
|
|
flags|TDB_INCOMPATIBLE_HASH,
|
|
O_CREAT|O_RDWR|O_TRUNC, 0600, &log_ctx,
|
|
tdb_dumb_hash);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
d.dptr = discard_const_p(uint8_t, "Hello");
|
|
d.dsize = 5;
|
|
ok1(tdb_store(tdb, d, d, TDB_INSERT) == 0);
|
|
tdb_close(tdb);
|
|
|
|
/* Should have marked rwlocks field. */
|
|
ok1(hdr_rwlocks("run-incompatible.tdb") == rwmagic);
|
|
|
|
/* It should not open if we don't specify. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, 0, O_RDWR, 0,
|
|
&log_ctx, NULL);
|
|
ok1(!tdb);
|
|
ok1(log_count == 1);
|
|
|
|
/* Should reopen with correct hash. */
|
|
log_count = 0;
|
|
tdb = tdb_open_ex("run-incompatible.tdb", 0, 0, O_RDWR, 0,
|
|
&log_ctx, tdb_dumb_hash);
|
|
ok1(tdb);
|
|
ok1(log_count == 0);
|
|
r = tdb_fetch(tdb, d);
|
|
ok1(r.dsize == 5);
|
|
free(r.dptr);
|
|
ok1(tdb_check(tdb, NULL, NULL) == 0);
|
|
tdb_close(tdb);
|
|
}
|
|
|
|
return exit_status();
|
|
}
|