/*
ldb database library
Copyright (C) Andrew Tridgell 2004
** NOTE! The following LGPL license applies to the ldb
** library. This does NOT imply that all of Samba is released
** under the LGPL
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 3 of the License, or (at your option) any later version.
This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, see .
*/
/*
* Name: ldb
*
* Component: ldbtest
*
* Description: utility to test ldb
*
* Author: Andrew Tridgell
*/
#ifdef _SAMBA_BUILD_
#include "includes.h"
#else
#include "ldb_includes.h"
#endif
#include "ldb.h"
#include "tools/cmdline.h"
static struct timespec tp1,tp2;
static struct ldb_cmdline *options;
static void _start_timer(void)
{
clock_gettime_mono(&tp1);
}
static double _end_timer(void)
{
clock_gettime_mono(&tp2);
return((tp2.tv_sec - tp1.tv_sec) +
(tp2.tv_nsec - tp1.tv_nsec)*1.0e-9);
}
static void add_records(struct ldb_context *ldb,
struct ldb_dn *basedn,
unsigned int count)
{
struct ldb_message msg;
unsigned int i;
#if 0
if (ldb_lock(ldb, "transaction") != 0) {
printf("transaction lock failed\n");
exit(1);
}
#endif
for (i=0;icount != 1)) {
printf("Failed to find %s - %s\n", expr, ldb_errstring(ldb));
exit(1);
}
if (uid >= nrecords && res->count > 0) {
printf("Found %s !? - %d\n", expr, ret);
exit(1);
}
printf("Testing uid %d/%d - %d \r", i, uid, res->count);
fflush(stdout);
talloc_free(res);
talloc_free(expr);
}
printf("\n");
}
static void start_test(struct ldb_context *ldb, unsigned int nrecords,
unsigned int nsearches)
{
struct ldb_dn *basedn;
basedn = ldb_dn_new(ldb, ldb, options->basedn);
if ( ! ldb_dn_validate(basedn)) {
printf("Invalid base DN\n");
exit(1);
}
printf("Adding %d records\n", nrecords);
add_records(ldb, basedn, nrecords);
printf("Starting search on uid\n");
_start_timer();
search_uid(ldb, basedn, nrecords, nsearches);
printf("uid search took %.2f seconds\n", _end_timer());
printf("Modifying records\n");
modify_records(ldb, basedn, nrecords);
printf("Deleting records\n");
delete_records(ldb, basedn, nrecords);
}
/*
2) Store an @indexlist record
3) Store a record that contains fields that should be index according
to @index
4) disconnection from database
5) connect to same database
6) search for record added in step 3 using a search key that should
be indexed
*/
static void start_test_index(struct ldb_context **ldb)
{
struct ldb_message *msg;
struct ldb_result *res = NULL;
struct ldb_dn *indexlist;
struct ldb_dn *basedn;
int ret;
int flags = 0;
const char *specials;
specials = getenv("LDB_SPECIALS");
if (specials && atoi(specials) == 0) {
printf("LDB_SPECIALS disabled - skipping index test\n");
return;
}
if (options->nosync) {
flags |= LDB_FLG_NOSYNC;
}
printf("Starting index test\n");
indexlist = ldb_dn_new(*ldb, *ldb, "@INDEXLIST");
ldb_delete(*ldb, indexlist);
msg = ldb_msg_new(NULL);
msg->dn = indexlist;
ldb_msg_add_string(msg, "@IDXATTR", strdup("uid"));
if (ldb_add(*ldb, msg) != 0) {
printf("Add of %s failed - %s\n", ldb_dn_get_linearized(msg->dn), ldb_errstring(*ldb));
exit(1);
}
basedn = ldb_dn_new(*ldb, *ldb, options->basedn);
memset(msg, 0, sizeof(*msg));
msg->dn = ldb_dn_copy(msg, basedn);
ldb_dn_add_child_fmt(msg->dn, "cn=test");
ldb_msg_add_string(msg, "cn", strdup("test"));
ldb_msg_add_string(msg, "sn", strdup("test"));
ldb_msg_add_string(msg, "uid", strdup("test"));
ldb_msg_add_string(msg, "objectClass", strdup("OpenLDAPperson"));
if (ldb_add(*ldb, msg) != 0) {
printf("Add of %s failed - %s\n", ldb_dn_get_linearized(msg->dn), ldb_errstring(*ldb));
exit(1);
}
if (talloc_free(*ldb) != 0) {
printf("failed to free/close ldb database");
exit(1);
}
(*ldb) = ldb_init(options, NULL);
ret = ldb_connect(*ldb, options->url, flags, NULL);
if (ret != 0) {
printf("failed to connect to %s\n", options->url);
exit(1);
}
basedn = ldb_dn_new(*ldb, *ldb, options->basedn);
msg->dn = basedn;
ldb_dn_add_child_fmt(msg->dn, "cn=test");
ret = ldb_search(*ldb, *ldb, &res, basedn, LDB_SCOPE_SUBTREE, NULL, "uid=test");
if (ret != LDB_SUCCESS) {
printf("Search with (uid=test) filter failed!\n");
exit(1);
}
if(res->count != 1) {
printf("Should have found 1 record - found %d\n", res->count);
exit(1);
}
indexlist = ldb_dn_new(*ldb, *ldb, "@INDEXLIST");
if (ldb_delete(*ldb, msg->dn) != 0 ||
ldb_delete(*ldb, indexlist) != 0) {
printf("cleanup failed - %s\n", ldb_errstring(*ldb));
exit(1);
}
printf("Finished index test\n");
}
static void usage(void)
{
printf("Usage: ldbtest \n");
printf("Options:\n");
printf(" -H ldb_url choose the database (or $LDB_URL)\n");
printf(" --num-records nrecords database size to use\n");
printf(" --num-searches nsearches number of searches to do\n");
printf("\n");
printf("tests ldb API\n\n");
exit(1);
}
int main(int argc, const char **argv)
{
TALLOC_CTX *mem_ctx = talloc_new(NULL);
struct ldb_context *ldb;
ldb = ldb_init(mem_ctx, NULL);
options = ldb_cmdline_process(ldb, argc, argv, usage);
talloc_steal(mem_ctx, options);
if (options->basedn == NULL) {
options->basedn = "ou=Ldb Test,ou=People,o=University of Michigan,c=TEST";
}
srandom(1);
printf("Testing with num-records=%d and num-searches=%d\n",
options->num_records, options->num_searches);
start_test(ldb,
(unsigned int) options->num_records,
(unsigned int) options->num_searches);
start_test_index(&ldb);
talloc_free(mem_ctx);
return 0;
}