4 Copyright (C) Andrew Tridgell 2004
6 ** NOTE! The following LGPL license applies to the ldb
7 ** library. This does NOT imply that all of Samba is released
10 This library is free software; you can redistribute it and/or
11 modify it under the terms of the GNU Lesser General Public
12 License as published by the Free Software Foundation; either
13 version 3 of the License, or (at your option) any later version.
15 This library is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 Lesser General Public License for more details.
20 You should have received a copy of the GNU Lesser General Public
21 License along with this library; if not, write to the Free Software
22 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
30 * Description: utility to test ldb
32 * Author: Andrew Tridgell
36 #include "ldb/include/includes.h"
37 #include "ldb/tools/cmdline.h"
39 static struct timeval tp1,tp2;
40 static struct ldb_cmdline *options;
42 static void _start_timer(void)
44 gettimeofday(&tp1,NULL);
47 static double _end_timer(void)
49 gettimeofday(&tp2,NULL);
50 return((tp2.tv_sec - tp1.tv_sec) +
51 (tp2.tv_usec - tp1.tv_usec)*1.0e-6);
54 static void add_records(struct ldb_context *ldb,
55 const struct ldb_dn *basedn,
58 struct ldb_message msg;
62 if (ldb_lock(ldb, "transaction") != 0) {
63 printf("transaction lock failed\n");
67 for (i=0;i<count;i++) {
68 struct ldb_message_element el[6];
69 struct ldb_val vals[6][1];
71 TALLOC_CTX *tmp_ctx = talloc_new(ldb);
73 name = talloc_asprintf(tmp_ctx, "Test%d", i);
75 msg.dn = ldb_dn_build_child(tmp_ctx, "cn", name, basedn);
80 el[0].name = talloc_strdup(tmp_ctx, "cn");
82 el[0].values = vals[0];
83 vals[0][0].data = (uint8_t *)name;
84 vals[0][0].length = strlen(name);
89 el[1].values = vals[1];
90 vals[1][0].data = (uint8_t *)talloc_asprintf(tmp_ctx, "The title of %s", name);
91 vals[1][0].length = strlen((char *)vals[1][0].data);
94 el[2].name = talloc_strdup(tmp_ctx, "uid");
96 el[2].values = vals[2];
97 vals[2][0].data = (uint8_t *)ldb_casefold(ldb, tmp_ctx, name);
98 vals[2][0].length = strlen((char *)vals[2][0].data);
101 el[3].name = talloc_strdup(tmp_ctx, "mail");
102 el[3].num_values = 1;
103 el[3].values = vals[3];
104 vals[3][0].data = (uint8_t *)talloc_asprintf(tmp_ctx, "%s@example.com", name);
105 vals[3][0].length = strlen((char *)vals[3][0].data);
108 el[4].name = talloc_strdup(tmp_ctx, "objectClass");
109 el[4].num_values = 1;
110 el[4].values = vals[4];
111 vals[4][0].data = (uint8_t *)talloc_strdup(tmp_ctx, "OpenLDAPperson");
112 vals[4][0].length = strlen((char *)vals[4][0].data);
115 el[5].name = talloc_strdup(tmp_ctx, "sn");
116 el[5].num_values = 1;
117 el[5].values = vals[5];
118 vals[5][0].data = (uint8_t *)name;
119 vals[5][0].length = strlen((char *)vals[5][0].data);
121 ldb_delete(ldb, msg.dn);
123 if (ldb_add(ldb, &msg) != 0) {
124 printf("Add of %s failed - %s\n", name, ldb_errstring(ldb));
128 printf("adding uid %s\r", name);
131 talloc_free(tmp_ctx);
134 if (ldb_unlock(ldb, "transaction") != 0) {
135 printf("transaction unlock failed\n");
142 static void modify_records(struct ldb_context *ldb,
143 const struct ldb_dn *basedn,
146 struct ldb_message msg;
149 for (i=0;i<count;i++) {
150 struct ldb_message_element el[3];
151 struct ldb_val vals[3];
153 TALLOC_CTX *tmp_ctx = talloc_new(ldb);
155 name = talloc_asprintf(tmp_ctx, "Test%d", i);
156 msg.dn = ldb_dn_build_child(tmp_ctx, "cn", name, basedn);
158 msg.num_elements = 3;
161 el[0].flags = LDB_FLAG_MOD_DELETE;
162 el[0].name = talloc_strdup(tmp_ctx, "mail");
163 el[0].num_values = 0;
165 el[1].flags = LDB_FLAG_MOD_ADD;
166 el[1].name = talloc_strdup(tmp_ctx, "mail");
167 el[1].num_values = 1;
168 el[1].values = &vals[1];
169 vals[1].data = (uint8_t *)talloc_asprintf(tmp_ctx, "%s@other.example.com", name);
170 vals[1].length = strlen((char *)vals[1].data);
172 el[2].flags = LDB_FLAG_MOD_REPLACE;
173 el[2].name = talloc_strdup(tmp_ctx, "mail");
174 el[2].num_values = 1;
175 el[2].values = &vals[2];
176 vals[2].data = (uint8_t *)talloc_asprintf(tmp_ctx, "%s@other2.example.com", name);
177 vals[2].length = strlen((char *)vals[2].data);
179 if (ldb_modify(ldb, &msg) != 0) {
180 printf("Modify of %s failed - %s\n", name, ldb_errstring(ldb));
184 printf("Modifying uid %s\r", name);
187 talloc_free(tmp_ctx);
194 static void delete_records(struct ldb_context *ldb,
195 const struct ldb_dn *basedn,
200 for (i=0;i<count;i++) {
202 char *name = talloc_asprintf(ldb, "Test%d", i);
203 dn = ldb_dn_build_child(name, "cn", name, basedn);
205 printf("Deleting uid Test%d\r", i);
208 if (ldb_delete(ldb, dn) != 0) {
209 printf("Delete of %s failed - %s\n", ldb_dn_linearize(ldb, dn), ldb_errstring(ldb));
218 static void search_uid(struct ldb_context *ldb, struct ldb_dn *basedn, int nrecords, int nsearches)
222 for (i=0;i<nsearches;i++) {
223 int uid = (i * 700 + 17) % (nrecords * 2);
225 struct ldb_result *res = NULL;
228 expr = talloc_asprintf(ldb, "(uid=TEST%d)", uid);
229 ret = ldb_search(ldb, basedn, LDB_SCOPE_SUBTREE, expr, NULL, &res);
231 if (ret != LDB_SUCCESS || (uid < nrecords && res->count != 1)) {
232 printf("Failed to find %s - %s\n", expr, ldb_errstring(ldb));
236 if (uid >= nrecords && res->count > 0) {
237 printf("Found %s !? - %d\n", expr, ret);
241 printf("testing uid %d/%d - %d \r", i, uid, res->count);
251 static void start_test(struct ldb_context *ldb, int nrecords, int nsearches)
253 struct ldb_dn *basedn;
255 basedn = ldb_dn_explode(ldb, options->basedn);
257 printf("Adding %d records\n", nrecords);
258 add_records(ldb, basedn, nrecords);
260 printf("Starting search on uid\n");
262 search_uid(ldb, basedn, nrecords, nsearches);
263 printf("uid search took %.2f seconds\n", _end_timer());
265 printf("Modifying records\n");
266 modify_records(ldb, basedn, nrecords);
268 printf("Deleting records\n");
269 delete_records(ldb, basedn, nrecords);
274 2) Store an @indexlist record
276 3) Store a record that contains fields that should be index according
279 4) disconnection from database
281 5) connect to same database
283 6) search for record added in step 3 using a search key that should
286 static void start_test_index(struct ldb_context **ldb)
288 struct ldb_message *msg;
289 struct ldb_result *res = NULL;
290 struct ldb_dn *indexlist;
291 struct ldb_dn *basedn;
294 const char *specials;
296 specials = getenv("LDB_SPECIALS");
297 if (specials && atoi(specials) == 0) {
298 printf("LDB_SPECIALS disabled - skipping index test\n");
302 if (options->nosync) {
303 flags |= LDB_FLG_NOSYNC;
306 printf("Starting index test\n");
308 indexlist = ldb_dn_explode(NULL, "@INDEXLIST");
310 ldb_delete(*ldb, indexlist);
312 msg = ldb_msg_new(NULL);
315 ldb_msg_add_string(msg, "@IDXATTR", strdup("uid"));
317 if (ldb_add(*ldb, msg) != 0) {
318 printf("Add of %s failed - %s\n", ldb_dn_linearize(*ldb, msg->dn), ldb_errstring(*ldb));
322 basedn = ldb_dn_explode(NULL, options->basedn);
324 memset(msg, 0, sizeof(*msg));
325 msg->dn = ldb_dn_build_child(msg, "cn", "test", basedn);
326 ldb_msg_add_string(msg, "cn", strdup("test"));
327 ldb_msg_add_string(msg, "sn", strdup("test"));
328 ldb_msg_add_string(msg, "uid", strdup("test"));
329 ldb_msg_add_string(msg, "objectClass", strdup("OpenLDAPperson"));
331 if (ldb_add(*ldb, msg) != 0) {
332 printf("Add of %s failed - %s\n", ldb_dn_linearize(*ldb, msg->dn), ldb_errstring(*ldb));
336 if (talloc_free(*ldb) != 0) {
337 printf("failed to free/close ldb database");
341 (*ldb) = ldb_init(options);
343 ret = ldb_connect(*ldb, options->url, flags, NULL);
345 printf("failed to connect to %s\n", options->url);
349 ret = ldb_search(*ldb, basedn, LDB_SCOPE_SUBTREE, "uid=test", NULL, &res);
350 if (ret != LDB_SUCCESS) {
351 printf("Search with (uid=test) filter failed!\n");
354 if(res->count != 1) {
355 printf("Should have found 1 record - found %d\n", res->count);
359 if (ldb_delete(*ldb, msg->dn) != 0 ||
360 ldb_delete(*ldb, indexlist) != 0) {
361 printf("cleanup failed - %s\n", ldb_errstring(*ldb));
365 printf("Finished index test\n");
369 static void usage(void)
371 printf("Usage: ldbtest <options>\n");
372 printf("Options:\n");
373 printf(" -H ldb_url choose the database (or $LDB_URL)\n");
374 printf(" --num-records nrecords database size to use\n");
375 printf(" --num-searches nsearches number of searches to do\n");
377 printf("tests ldb API\n\n");
381 int main(int argc, const char **argv)
383 TALLOC_CTX *mem_ctx = talloc_new(NULL);
384 struct ldb_context *ldb;
388 ldb = ldb_init(mem_ctx);
390 options = ldb_cmdline_process(ldb, argc, argv, usage);
392 talloc_steal(mem_ctx, options);
394 if (options->basedn == NULL) {
395 options->basedn = "ou=Ldb Test,ou=People,o=University of Michigan,c=TEST";
400 printf("Testing with num-records=%d and num-searches=%d\n",
401 options->num_records, options->num_searches);
403 start_test(ldb, options->num_records, options->num_searches);
405 start_test_index(&ldb);
407 talloc_free(mem_ctx);