11 #include <ac/unistd.h>
14 #ifdef HAVE_SYS_RESOURCE_H
15 #include <sys/resource.h>
17 #ifdef HAVE_SYS_PARAM_H
18 #include <sys/param.h>
30 #include "ldapconfig.h"
32 #include "../back-ldbm/back-ldbm.h"
34 #define EDITOR "/usr/ucb/vi"
36 static struct dbcache *openchoice(char c, int mode, int verbose, char **fname);
37 static void print_entry(FILE *fp, char c, Datum *key, char *klabel, Datum *data, char *dlabel);
38 static void free_and_close(struct dbcache *dbc, Datum key, Datum data);
39 static void edit_entry(char c, Datum *data);
40 static void get_keydata(FILE *fp, char c, Datum *key, Datum *data);
42 static struct dbcache *dbc;
44 static Backend *be = NULL;
47 main( int argc, char **argv )
50 Datum savekey, key, data, last;
58 #ifdef HAVE_BERKELEY_DB2
62 ldbm_datum_init( savekey );
63 ldbm_datum_init( key );
64 ldbm_datum_init( data );
65 ldbm_datum_init( last );
67 tailorfile = SLAPD_DEFAULT_CONFIGFILE;
68 while ( (i = getopt( argc, argv, "d:f:" )) != EOF ) {
70 case 'd': /* turn on debugging */
71 ldap_debug = atoi( optarg );
74 case 'f': /* specify a tailor file */
75 tailorfile = strdup( optarg );
80 "usage: %s [-d level] [-f slapdconfigfile]\n",
88 * initialize stuff and figure out which backend we're dealing with
91 slap_init(SLAP_TOOL_MODE, "ldbmtest");
92 read_config( tailorfile );
98 if ( fgets( buf, sizeof(buf), stdin ) == NULL )
102 case 'c': /* create an index */
104 if ( (dbc = openchoice( buf[1], LDBM_READER, 0,
105 &fname )) != NULL ) {
106 printf( "Already exists\n" );
107 ldbm_close( dbc->dbc_db );
110 if ( (dbc = openchoice( buf[1], LDBM_WRCREAT, 1,
111 &fname )) != NULL ) {
112 ldbm_close( dbc->dbc_db );
116 case 'l': /* lookup somethig in an index */
117 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
122 get_keydata( stdin, buf[1], &key, NULL );
123 data = ldbm_fetch( dbc->dbc_db, key );
124 print_entry( stdout, buf[1], &key, "key: ", &data,
127 free_and_close( dbc, key, data );
130 case 'L': /* get all blocks for a key from an index */
131 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
136 get_keydata( stdin, buf[1], &key, NULL );
137 if ( (idl = idl_fetch( be, dbc, key )) != NULL ) {
138 data.dptr = (char *) idl;
139 data.dsize = (ID_BLOCK_NMAX(idl) + 1) * sizeof(ID);
140 print_entry( stdout, buf[1], &key, "key: ",
143 free_and_close( dbc, key, data );
146 case 't': /* traverse */
147 case 'T': /* traverse - keys only */
148 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
150 perror( "openchoice" );
155 #ifdef HAVE_BERKELEY_DB2
156 for ( key = ldbm_firstkey( dbc->dbc_db, &cursorp );
158 key = ldbm_nextkey( dbc->dbc_db, key, cursorp ) )
160 for ( key = ldbm_firstkey( dbc->dbc_db );
162 key = ldbm_nextkey( dbc->dbc_db, key ) )
165 if ( savekey.dptr != NULL )
166 ldbm_datum_free( dbc->dbc_db, savekey );
169 data = ldbm_fetch( dbc->dbc_db, key );
171 if ( buf[0] == 't' ) {
172 print_entry( stdout, buf[1], &key,
173 "key: ", &data, "data:\n" );
175 print_entry( stdout, buf[1], &key,
176 "key: ", NULL, NULL );
179 if ( data.dptr != NULL ) {
180 ldbm_datum_free( dbc->dbc_db, data );
183 if ( savekey.dptr != NULL )
184 ldbm_datum_free( dbc->dbc_db, savekey );
186 ldbm_close( dbc->dbc_db );
189 case 'x': /* delete an entry */
190 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
195 get_keydata( stdin, buf[1], &key, NULL );
197 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
198 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
199 perror( "ldbm_delete" );
201 fprintf( stderr, "db_errno %d",
202 ldbm_errno( dbc->dbc_db ) );
207 free_and_close( dbc, key, data );
210 case 'e': /* edit an entry */
211 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
216 get_keydata( stdin, buf[1], &key, NULL );
218 data = ldbm_fetch( dbc->dbc_db, key );
219 if ( data.dptr == NULL ) {
220 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
221 perror( "ldbm_fetch" );
223 fprintf( stderr, "db_errno %d\n",
224 ldbm_errno( dbc->dbc_db ) );
226 free_and_close( dbc, key, data );
230 edit_entry( buf[1], &data );
232 if ( data.dptr == NULL ) {
233 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
234 perror( "ldbm_delete" );
236 } else if ( ldbm_store( dbc->dbc_db, key, data,
237 LDBM_REPLACE ) != 0 ) {
238 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
239 perror( "ldbm_store" );
241 fprintf( stderr, "db_errno %d\n",
242 ldbm_errno( dbc->dbc_db ) );
246 free_and_close( dbc, key, data );
249 case 'a': /* add an entry */
250 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
255 get_keydata( stdin, buf[1], &key, &data );
257 if ( ldbm_store( dbc->dbc_db, key, data, LDBM_INSERT )
259 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
260 perror( "ldbm_store" );
262 fprintf( stderr, "db_errno %d\n",
263 ldbm_errno( dbc->dbc_db ) );
267 free_and_close( dbc, key, data );
270 case 'i': /* insert an id into an index entry */
271 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
276 get_keydata( stdin, buf[1], &key, &data );
278 idl = (ID_BLOCK *) data.dptr;
279 for ( id = idl_firstid( idl ); id != NOID;
280 id = idl_nextid( idl, id ) ) {
281 if ( idl_insert_key( be, dbc, key, id )
284 "idl_insert_key (%s) %ld failed\n",
290 free_and_close( dbc, key, data );
293 case 'b': /* select a backend by suffix */
294 printf( "suffix: " );
296 if ( fgets( buf, sizeof(buf), stdin ) == NULL ) {
299 buf[strlen( buf ) - 1] = '\0';
301 (void) dn_normalize_case( buf );
302 if ( (tbe = select_backend( buf )) == NULL ) {
303 fprintf( stderr, "unknown suffix \"%s\"\n",
310 case 'B': /* print current suffix */
312 printf( "no current backend\n" );
314 printf( "current backend has suffix \"%s\"\n",
319 case 'C': /* produce concordance of an index */
320 if ( (dbc = openchoice( 'i', LDBM_READER, 1, NULL ))
327 #ifdef HAVE_BERKELEY_DB2
328 for ( key = ldbm_firstkey( dbp, &cursorp );
330 key = ldbm_nextkey( dbp, last, cursorp ) )
332 for ( key = ldbm_firstkey( dbp ); key.dptr != NULL;
333 key = ldbm_nextkey( dbp, last ) )
336 if ( last.dptr != NULL ) {
337 ldbm_datum_free( dbp, last );
340 printf( "key(%d): (%s)\n", key.dsize,
344 free_and_close( dbc, key, last );
348 printf( "commands: l<c> => lookup index\n" );
349 printf( " L<c> => lookup index (all)\n" );
350 printf( " t<c> => traverse index\n" );
351 printf( " T<c> => traverse index keys\n" );
352 printf( " x<c> => delete from index\n" );
353 printf( " e<c> => edit index entry\n" );
354 printf( " a<c> => add index entry\n" );
355 printf( " c<c> => create index\n" );
356 printf( " i<c> => insert ids into index\n" );
357 printf( " b => change default backend\n" );
358 printf( " B => print default backend\n" );
359 printf( "where <c> is a char selecting the index:\n" );
360 printf( " c => id2children\n" );
361 printf( " d => dn2id\n" );
362 printf( " e => id2entry\n" );
363 printf( " f => arbitrary file\n" );
364 printf( " i => attribute index\n" );
376 free_and_close( struct dbcache *dbc, Datum key, Datum data )
378 ldbm_cache_really_close( be, dbc );
379 if ( key.dptr != NULL )
380 ldbm_datum_free( dbp, key );
381 if ( data.dptr != NULL )
382 ldbm_datum_free( dbp, data );
386 dnid_cmp( const void *a, const void *b )
388 return( *(const long int *)a - *(const long int *)b );
392 myrealloc( char *p, int size )
395 return( (char *) malloc( size ) );
397 return( (char *) realloc( p, size ) );
401 get_idlist( FILE *fp, Datum *data )
406 unsigned int psize, pmax;
407 unsigned int nmax, nids;
413 psize = 2 * sizeof(ID);
421 if ( fgets( buf, sizeof(buf), fp ) == NULL || buf[0] == '\n' )
423 if ( strncmp( buf, "nmax=", 5 ) == 0 ) {
424 nmax = atol( buf + 5 );
428 if ( psize + sizeof(ID) > pmax ) {
430 p = (ID_BLOCK *) myrealloc( (char *) p, pmax );
433 if ( strncmp( buf, "nids=0", 6 ) == 0 ) {
438 ID_BLOCK_ID(p,i++) = atol( buf );
444 printf( "%d IDs entered. Max number of ids? [%d] ", i,
446 if ( fgets( buf, sizeof(buf), fp ) != NULL &&
447 isdigit( (unsigned char) buf[0] ) ) {
455 ID_BLOCK_NMAX(p) = nmax;
457 ID_BLOCK_NIDS(p) = 0;
458 ID_BLOCK_ID(p,i) = NOID;
460 ID_BLOCK_NIDS(p) = i;
463 qsort( (void *) &ID_BLOCK_ID(p, 0), i, sizeof(ID), dnid_cmp );
466 data->dptr = (char *) p;
467 data->dsize = (nmax + ID_BLOCK_IDS_OFFSET) * sizeof(ID);
471 get_entry( FILE *fp, Datum *data )
475 unsigned int pmax, psize, len;
480 printf( "Enter entry, <cr><cr> to end:\n" );
484 while ( fgets( buf, sizeof(buf), fp ) != NULL ) {
486 if ( psize + strlen( buf ) > pmax ) {
488 p = myrealloc( p, pmax );
496 if ( buf[0] == '\n' )
501 data->dsize = psize + 1;
505 edit_entry( char c, Datum *data )
511 WAITSTATUSTYPE status;
514 strcpy( tmpname, "/tmp/dbtestXXXXXX" );
516 if ( (fd = open( mktemp( tmpname ), O_RDWR, 0600 )) == -1 ) {
521 if ( (fd = mkstemp( tmpname )) == -1 ) {
527 fp = fdopen( fd, "w" );
528 print_entry( fp, c, NULL, NULL, data, NULL );
536 } else if ( pid == 0 ) {
539 if ( (editor = getenv( "EDITOR" )) == NULL ) {
542 execl( editor, editor, tmpname, NULL );
550 if ( waitpid( (pid_t) -1, NULL, WAIT_FLAGS ) < 0 ) {
552 if ( wait4( (pid_t) -1, &status, WAIT_FLAGS, 0 ) < 0 ) {
558 if ( (fp = fopen( tmpname, "r" )) == NULL ) {
562 if ( data->dptr != NULL ) {
563 ldbm_datum_free( NULL, *data );
565 get_keydata( fp, c, NULL, data );
570 static struct dbcache *
571 openfile( char *name, int namesiz, int mode, int verbose, char c )
575 if ( name == NULL || *name == '\0' ) {
578 if ( fgets( name, namesiz, stdin ) == NULL )
580 name[strlen( name ) - 1] = '\0';
583 if ( fgets( name, namesiz, stdin ) == NULL )
585 name[strlen( name ) - 1] = '\0';
589 if ( (dbc = ldbm_cache_open( be, name, (c == 'f') ? "" : LDBM_SUFFIX,
590 LDBM_READER )) == NULL ) {
599 static struct dbcache *
600 openchoice( char c, int mode, int verbose, char **fname )
602 static char name[MAXPATHLEN];
605 case 'c': /* id2children */
606 sprintf( name, "id2children" );
608 case 'd': /* dn2id */
609 sprintf( name, "dn2id" );
611 case 'e': /* id2entry */
612 sprintf( name, "id2entry" );
614 case 'f': /* arbitrary file */
615 case 'i': /* index */
616 if ( fname != NULL && *fname != NULL ) {
617 strcpy( name, *fname );
623 printf( "specify one of [fdeci] to select file\n" );
627 if ( fname != NULL ) {
631 return( openfile( name, MAXPATHLEN, mode, verbose, c ) );
649 if ( data != NULL && data->dptr == NULL ) {
653 if ( ldbm_errno( dbp ) == 0 )
656 fprintf( stderr, "%s: db_errno %d\n", msg,
662 case 'd': /* dn2id - key is dn, data is dnid */
664 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
666 if ( data != NULL ) {
667 SAFEMEMCPY( (char *) &id, data->dptr, sizeof(ID) );
668 fprintf( fp, "%s%ld\n", dlabel ? dlabel : "", id );
672 case 'e': /* id2entry - key is dnid, data is entry */
674 SAFEMEMCPY( (char *) &id, key->dptr, sizeof(ID) );
675 fprintf( fp, "%s %ld\n", klabel, id );
677 if ( data != NULL ) {
679 fprintf( fp, "data length: %d\n", data->dsize );
682 fputs( data->dptr, fp );
687 case 'i': /* index - key is string, data is dnid[] */
689 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
691 if ( data != NULL ) {
692 idl = (ID_BLOCK *) data->dptr;
695 fprintf( fp, "%s\tnmax=%ld\n\tncur=%ld\n", dlabel,
696 ID_BLOCK_NMAX(idl), ID_BLOCK_NIDS(idl) );
698 if ( ID_BLOCK_INDIRECT( idl ) ) {
699 for ( i = 0; !ID_BLOCK_NOID(idl, i); i++ ) {
700 fprintf( fp, "\t%ld\n", ID_BLOCK_ID(idl, i) );
702 } else if ( ID_BLOCK_ALLIDS( idl ) ) {
703 fprintf( fp, "\tALLIDS (1..%ld)\n",
704 ID_BLOCK_NIDS(idl) - 1 );
706 for ( i = 0; i < ID_BLOCK_NIDS(idl); i++ ) {
707 fprintf( fp, "\t%ld\n", ID_BLOCK_ID(idl,i) );
713 case 'f': /* arbitrary file - assume key & data are strings */
715 fprintf( fp, "%s%s\n", klabel, key->dptr );
716 if ( data != NULL ) {
717 fprintf( fp, "%s%s\n", dlabel ? dlabel : "",
723 fprintf( stderr, "specify [deci] to select a file\n" );
729 get_keydata( FILE *fp, char c, Datum *key, Datum *data )
731 static char kbuf[BUFSIZ], dbuf[BUFSIZ];
739 case 'd': /* dn2id - key is dn, data is dnid */
743 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
746 kbuf[strlen( kbuf ) - 1] = '\0';
747 key->dptr = strdup( kbuf );
748 key->dsize = strlen( kbuf ) + 1;
751 if ( data != NULL ) {
754 if ( fgets( dbuf, sizeof(dbuf), fp ) == NULL ) {
758 data->dptr = (char *) malloc( sizeof(n) );
759 memcpy( data->dptr, (char *) &n, sizeof(n) );
760 data->dsize = sizeof(n);
764 case 'e': /* id2entry - key is dnid, data is entry */
768 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
772 key->dptr = (char *) malloc( sizeof(n) );
773 memcpy( key->dptr, (char *) &n, sizeof(n) );
774 key->dsize = sizeof(n);
777 if ( data != NULL ) {
778 get_entry( fp, data );
782 case 'c': /* id2children - key is string dnid, data is dnid[] */
783 case 'i': /* index - key is string, data is dnid[] */
787 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
790 kbuf[strlen( kbuf ) - 1] = '\0';
791 key->dptr = strdup( kbuf );
792 key->dsize = strlen( kbuf ) + 1;
795 if ( data != NULL ) {
796 get_idlist( fp, data );
801 fprintf(stderr, "specify [deci] to select file type\n");