2 * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
3 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
10 #include <ac/stdlib.h>
11 #include <ac/socket.h>
12 #include <ac/string.h>
15 #include <ac/unistd.h>
18 #ifdef HAVE_SYS_RESOURCE_H
19 #include <sys/resource.h>
21 #ifdef HAVE_SYS_PARAM_H
22 #include <sys/param.h>
35 #include "../back-ldbm/back-ldbm.h"
37 static DBCache *openchoice(char c, int mode, int verbose, char **fname);
38 static void print_entry(FILE *fp, char c, Datum *key, char *klabel, Datum *data, char *dlabel);
39 static void free_and_close(DBCache *dbc, Datum key, Datum data);
40 static void edit_entry(char c, Datum *data);
41 static void get_keydata(FILE *fp, char c, Datum *key, Datum *data);
45 static Backend *be = NULL;
48 main( int argc, char **argv )
51 Datum savekey, key, data, last;
62 "ldbmtest not updated to support new index formats!\n" );
66 ldbm_datum_init( savekey );
67 ldbm_datum_init( key );
68 ldbm_datum_init( data );
69 ldbm_datum_init( last );
71 tailorfile = SLAPD_DEFAULT_CONFIGFILE;
72 while ( (i = getopt( argc, argv, "d:f:" )) != EOF ) {
74 case 'd': /* turn on debugging */
75 ldap_debug = atoi( optarg );
78 case 'f': /* specify a tailor file */
79 tailorfile = strdup( optarg );
84 "usage: %s [-d level] [-f slapdconfigfile]\n",
91 * initialize stuff and figure out which backend we're dealing with
94 slap_init(SLAP_TOOL_MODE, "ldbmtest");
95 read_config( tailorfile );
101 if ( fgets( buf, sizeof(buf), stdin ) == NULL )
105 case 'c': /* create an index */
107 if ( (dbc = openchoice( buf[1], LDBM_READER, 0,
108 &fname )) != NULL ) {
109 printf( "Already exists\n" );
110 ldbm_close( dbc->dbc_db );
113 if ( (dbc = openchoice( buf[1], LDBM_WRCREAT, 1,
114 &fname )) != NULL ) {
115 ldbm_close( dbc->dbc_db );
119 case 'l': /* lookup something in an index */
120 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
125 get_keydata( stdin, buf[1], &key, NULL );
126 data = ldbm_fetch( dbc->dbc_db, key );
127 print_entry( stdout, buf[1], &key, "key: ", &data,
130 free_and_close( dbc, key, data );
133 case 'L': /* get all blocks for a key from an index */
134 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
139 get_keydata( stdin, buf[1], &key, NULL );
140 if ( (idl = idl_fetch( be, dbc, key )) != NULL ) {
141 data.dptr = (char *) idl;
142 data.dsize = (ID_BLOCK_NMAX(idl) + 1) * sizeof(ID);
143 print_entry( stdout, buf[1], &key, "key: ",
146 free_and_close( dbc, key, data );
149 case 't': /* traverse */
150 case 'T': /* traverse - keys only */
151 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
153 perror( "openchoice" );
158 for ( key = ldbm_firstkey( dbc->dbc_db, &cursorp );
160 key = ldbm_nextkey( dbc->dbc_db, key, cursorp ) )
162 if ( savekey.dptr != NULL )
163 ldbm_datum_free( dbc->dbc_db, savekey );
166 data = ldbm_fetch( dbc->dbc_db, key );
168 if ( buf[0] == 't' ) {
169 print_entry( stdout, buf[1], &key,
170 "key: ", &data, "data:\n" );
172 print_entry( stdout, buf[1], &key,
173 "key: ", NULL, NULL );
176 if ( data.dptr != NULL ) {
177 ldbm_datum_free( dbc->dbc_db, data );
180 if ( savekey.dptr != NULL )
181 ldbm_datum_free( dbc->dbc_db, savekey );
183 ldbm_close( dbc->dbc_db );
186 case 'x': /* delete an entry */
187 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
192 get_keydata( stdin, buf[1], &key, NULL );
194 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
195 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
196 perror( "ldbm_delete" );
198 fprintf( stderr, "db_errno=%d",
199 ldbm_errno( dbc->dbc_db ) );
204 free_and_close( dbc, key, data );
208 case 'e': /* edit an entry */
209 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
214 get_keydata( stdin, buf[1], &key, NULL );
216 data = ldbm_fetch( dbc->dbc_db, key );
217 if ( data.dptr == NULL ) {
218 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
219 perror( "ldbm_fetch" );
221 fprintf( stderr, "db_errno=%d\n",
222 ldbm_errno( dbc->dbc_db ) );
224 free_and_close( dbc, key, data );
228 edit_entry( buf[1], &data );
230 if ( data.dptr == NULL ) {
231 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
232 perror( "ldbm_delete" );
234 } else if ( ldbm_store( dbc->dbc_db, key, data,
235 LDBM_REPLACE ) != 0 ) {
236 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
237 perror( "ldbm_store" );
239 fprintf( stderr, "db_errno=%d\n",
240 ldbm_errno( dbc->dbc_db ) );
244 free_and_close( dbc, key, data );
248 case 'a': /* add an entry */
249 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
254 get_keydata( stdin, buf[1], &key, &data );
256 if ( ldbm_store( dbc->dbc_db, key, data, LDBM_INSERT )
258 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
259 perror( "ldbm_store" );
261 fprintf( stderr, "db_errno=%d\n",
262 ldbm_errno( dbc->dbc_db ) );
266 free_and_close( dbc, key, data );
269 case 'i': /* insert an id into an index entry */
270 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
275 get_keydata( stdin, buf[1], &key, &data );
277 idl = (ID_BLOCK *) data.dptr;
278 for ( id = idl_firstid( idl, &cursor ); id != NOID;
279 id = idl_nextid( idl, &cursor ) ) {
280 if ( idl_insert_key( be, dbc, key, id )
283 "idl_insert_key (%s) %ld failed\n",
289 free_and_close( dbc, key, data );
292 case 'b': /* select a backend by suffix */
293 printf( "suffix: " );
295 if ( fgets( buf, sizeof(buf), stdin ) == NULL ) {
296 exit( EXIT_SUCCESS );
298 buf[strlen( buf ) - 1] = '\0';
300 (void) dn_normalize_case( buf );
301 if ( (tbe = select_backend( buf )) == NULL ) {
302 fprintf( stderr, "unknown suffix \"%s\"\n",
309 case 'B': /* print current suffix */
311 printf( "no current backend\n" );
313 printf( "current backend has suffix \"%s\"\n",
318 case 'C': /* produce concordance of an index */
319 if ( (dbc = openchoice( 'i', LDBM_READER, 1, NULL ))
326 for ( key = ldbm_firstkey( dbp, &cursorp );
328 key = ldbm_nextkey( dbp, last, cursorp ) )
330 if ( last.dptr != NULL ) {
331 ldbm_datum_free( dbp, last );
334 printf( "key(%d): (%s)\n", key.dsize,
338 free_and_close( dbc, key, last );
342 printf( "commands: l<c> => lookup index\n" );
343 printf( " L<c> => lookup index (all)\n" );
344 printf( " t<c> => traverse index\n" );
345 printf( " T<c> => traverse index keys\n" );
346 printf( " x<c> => delete from index\n" );
347 printf( " e<c> => edit index entry\n" );
348 printf( " a<c> => add index entry\n" );
349 printf( " c<c> => create index\n" );
350 printf( " i<c> => insert ids into index\n" );
351 printf( " b => change default backend\n" );
352 printf( " B => print default backend\n" );
353 printf( "where <c> is a char selecting the index:\n" );
354 printf( " d => dn2id\n" );
355 printf( " e => id2entry\n" );
356 printf( " f => arbitrary file\n" );
357 printf( " i => attribute index\n" );
362 slap_shutdown( NULL );
365 return( EXIT_SUCCESS );
369 free_and_close( DBCache *dbc, Datum key, Datum data )
371 ldbm_cache_really_close( be, dbc );
372 if ( key.dptr != NULL )
373 ldbm_datum_free( dbp, key );
374 if ( data.dptr != NULL )
375 ldbm_datum_free( dbp, data );
379 dnid_cmp( const void *a, const void *b )
381 return( *(const long int *)a - *(const long int *)b );
385 myrealloc( char *p, int size )
388 return( (char *) malloc( size ) );
390 return( (char *) realloc( p, size ) );
394 get_idlist( FILE *fp, Datum *data )
399 unsigned int psize, pmax;
400 unsigned int nmax, nids;
406 psize = 2 * sizeof(ID);
414 if ( fgets( buf, sizeof(buf), fp ) == NULL || buf[0] == '\n' )
416 if ( strncmp( buf, "nmax=", 5 ) == 0 ) {
417 nmax = atol( buf + 5 );
421 if ( psize + sizeof(ID) > pmax ) {
423 p = (ID_BLOCK *) myrealloc( (char *) p, pmax );
426 if ( strncmp( buf, "nids=0", 6 ) == 0 ) {
431 ID_BLOCK_ID(p,i++) = atol( buf );
437 printf( "%d IDs entered. Max number of ids? [%d] ", i,
439 if ( fgets( buf, sizeof(buf), fp ) != NULL &&
440 isdigit( (unsigned char) buf[0] ) ) {
448 ID_BLOCK_NMAX(p) = nmax;
450 ID_BLOCK_NIDS(p) = 0;
451 ID_BLOCK_ID(p,i) = NOID;
453 ID_BLOCK_NIDS(p) = i;
456 qsort( (void *) &ID_BLOCK_ID(p, 0), i, sizeof(ID), dnid_cmp );
459 data->dptr = (char *) p;
460 data->dsize = (nmax + ID_BLOCK_IDS_OFFSET) * sizeof(ID);
464 get_entry( FILE *fp, Datum *data )
468 unsigned int pmax, psize, len;
473 printf( "Enter entry, <cr><cr> to end:\n" );
477 while ( fgets( buf, sizeof(buf), fp ) != NULL ) {
479 if ( psize + strlen( buf ) > pmax ) {
481 p = myrealloc( p, pmax );
489 if ( buf[0] == '\n' )
494 data->dsize = psize + 1;
499 edit_entry( char c, Datum *data )
505 WAITSTATUSTYPE status;
508 strcpy( tmpname, "/tmp/dbtestXXXXXX" );
510 if ( (fd = open( mktemp( tmpname ), O_RDWR, 0600 )) == -1 ) {
515 if ( (fd = mkstemp( tmpname )) == -1 ) {
521 fp = fdopen( fd, "w" );
522 print_entry( fp, c, NULL, NULL, data, NULL );
530 } else if ( pid == 0 ) {
533 if ( (editor = getenv( "EDITOR" )) == NULL ) {
534 editor = LDAP_EDITOR;
536 execl( editor, editor, tmpname, NULL );
538 exit( EXIT_FAILURE );
544 if ( waitpid( (pid_t) -1, NULL, WAIT_FLAGS ) < 0 )
546 if ( wait3( (pid_t) -1, &status, WAIT_FLAGS, 0 ) < 0 )
553 if ( (fp = fopen( tmpname, "r" )) == NULL ) {
557 if ( data->dptr != NULL ) {
558 ldbm_datum_free( NULL, *data );
560 get_keydata( fp, c, NULL, data );
567 openfile( char *name, int namesiz, int mode, int verbose, char c )
571 if ( name == NULL || *name == '\0' ) {
574 if ( fgets( name, namesiz, stdin ) == NULL )
575 exit( EXIT_SUCCESS );
576 name[strlen( name ) - 1] = '\0';
579 if ( fgets( name, namesiz, stdin ) == NULL )
580 exit( EXIT_SUCCESS );
581 name[strlen( name ) - 1] = '\0';
585 if ( (dbc = ldbm_cache_open( be, name, (c == 'f') ? "" : LDBM_SUFFIX,
586 LDBM_READER )) == NULL ) {
596 openchoice( char c, int mode, int verbose, char **fname )
598 static char name[MAXPATHLEN];
601 case 'd': /* dn2id */
602 sprintf( name, "dn2id" );
604 case 'e': /* id2entry */
605 sprintf( name, "id2entry" );
607 case 'f': /* arbitrary file */
608 case 'i': /* index */
609 if ( fname != NULL && *fname != NULL ) {
610 strcpy( name, *fname );
616 printf( "specify one of [fdei] to select file\n" );
620 if ( fname != NULL ) {
624 return( openfile( name, MAXPATHLEN, mode, verbose, c ) );
642 if ( data != NULL && data->dptr == NULL ) {
646 if ( ldbm_errno( dbp ) == 0 )
649 fprintf( stderr, "%s: db_errno=%d\n", msg,
655 case 'd': /* dn2id - key is dn, data is dnid */
657 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
659 if ( data != NULL ) {
660 SAFEMEMCPY( (char *) &id, data->dptr, sizeof(ID) );
661 fprintf( fp, "%s%ld\n", dlabel ? dlabel : "", id );
665 case 'e': /* id2entry - key is dnid, data is entry */
667 SAFEMEMCPY( (char *) &id, key->dptr, sizeof(ID) );
668 fprintf( fp, "%s %ld\n", klabel, id );
670 if ( data != NULL ) {
672 fprintf( fp, "data length: %d\n", data->dsize );
675 fputs( data->dptr, fp );
679 case 'i': /* index - key is string, data is dnid[] */
681 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
683 if ( data != NULL ) {
684 idl = (ID_BLOCK *) data->dptr;
687 fprintf( fp, "%s\tnmax=%ld\n\tncur=%ld\n", dlabel,
688 ID_BLOCK_NMAX(idl), ID_BLOCK_NIDS(idl) );
690 if ( ID_BLOCK_INDIRECT( idl ) ) {
691 for ( i = 0; !ID_BLOCK_NOID(idl, i); i++ ) {
692 fprintf( fp, "\t%ld\n", ID_BLOCK_ID(idl, i) );
694 } else if ( ID_BLOCK_ALLIDS( idl ) ) {
695 fprintf( fp, "\tALLIDS\n" );
697 for ( i = 0; i < ID_BLOCK_NIDS(idl); i++ ) {
698 fprintf( fp, "\t%ld\n", ID_BLOCK_ID(idl,i) );
704 case 'f': /* arbitrary file - assume key & data are strings */
706 fprintf( fp, "%s%s\n", klabel, key->dptr );
707 if ( data != NULL ) {
708 fprintf( fp, "%s%s\n", dlabel ? dlabel : "",
714 fprintf( stderr, "specify [dei] to select a file\n" );
720 get_keydata( FILE *fp, char c, Datum *key, Datum *data )
722 static char kbuf[BUFSIZ], dbuf[BUFSIZ];
730 case 'd': /* dn2id - key is dn, data is dnid */
734 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
735 exit( EXIT_SUCCESS );
737 kbuf[strlen( kbuf ) - 1] = '\0';
738 key->dptr = strdup( kbuf );
739 key->dsize = strlen( kbuf ) + 1;
742 if ( data != NULL ) {
745 if ( fgets( dbuf, sizeof(dbuf), fp ) == NULL ) {
746 exit( EXIT_SUCCESS );
749 data->dptr = (char *) malloc( sizeof(n) );
750 memcpy( data->dptr, (char *) &n, sizeof(n) );
751 data->dsize = sizeof(n);
755 case 'e': /* id2entry - key is dnid, data is entry */
759 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
760 exit( EXIT_SUCCESS );
763 key->dptr = (char *) malloc( sizeof(n) );
764 memcpy( key->dptr, (char *) &n, sizeof(n) );
765 key->dsize = sizeof(n);
768 if ( data != NULL ) {
769 get_entry( fp, data );
773 case 'i': /* index - key is string, data is dnid[] */
777 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
778 exit( EXIT_SUCCESS );
780 kbuf[strlen( kbuf ) - 1] = '\0';
781 key->dptr = strdup( kbuf );
782 key->dsize = strlen( kbuf ) + 1;
785 if ( data != NULL ) {
786 get_idlist( fp, data );
791 fprintf(stderr, "specify [dei] to select file type\n");