11 #include <ac/unistd.h>
14 #include <sys/resource.h>
15 #include <sys/param.h>
22 #include "ldapconfig.h"
24 #include "../back-ldbm/back-ldbm.h"
26 #define EDITOR "/usr/ucb/vi"
28 static struct dbcache *openchoice(char c, int mode, int verbose, char **fname);
29 static void print_entry(FILE *fp, char c, Datum *key, char *klabel, Datum *data, char *dlabel);
30 static void free_and_close(struct dbcache *dbc, Datum key, Datum data);
31 static void edit_entry(char c, Datum *data);
32 static void get_keydata(FILE *fp, char c, Datum *key, Datum *data);
40 int ldap_syslog_level;
41 long num_entries_sent;
44 char *default_referral;
46 pthread_t listener_tid;
47 pthread_mutex_t num_sent_mutex;
48 pthread_mutex_t entry2str_mutex;
49 pthread_mutex_t active_threads_mutex;
50 pthread_mutex_t new_conn_mutex;
51 pthread_mutex_t currenttime_mutex;
52 pthread_mutex_t replog_mutex;
53 pthread_mutex_t ops_mutex;
54 pthread_mutex_t regex_mutex;
56 pthread_mutex_t crypt_mutex;
60 main( int argc, char **argv )
63 Datum savekey, key, data, last;
70 #ifdef HAVE_BERKELEY_DB2
73 ldbm_datum_init( savekey );
74 ldbm_datum_init( key );
75 ldbm_datum_init( data );
76 ldbm_datum_init( last );
79 tailorfile = SLAPD_DEFAULT_CONFIGFILE;
80 while ( (i = getopt( argc, argv, "d:f:" )) != EOF ) {
82 case 'd': /* turn on debugging */
83 ldap_debug = atoi( optarg );
86 case 'f': /* specify a tailor file */
87 tailorfile = strdup( optarg );
92 "usage: %s [-d level] [-f slapdconfigfile]\n",
100 * initialize stuff and figure out which backend we're dealing with
104 read_config( tailorfile, &be, NULL );
107 printf( "dbtest: " );
109 if ( fgets( buf, sizeof(buf), stdin ) == NULL )
113 case 'c': /* create an index */
115 if ( (dbc = openchoice( buf[1], LDBM_READER, 0,
116 &fname )) != NULL ) {
117 printf( "Already exists\n" );
118 ldbm_close( dbc->dbc_db );
121 if ( (dbc = openchoice( buf[1], LDBM_WRCREAT, 1,
122 &fname )) != NULL ) {
123 ldbm_close( dbc->dbc_db );
127 case 'l': /* lookup somethig in an index */
128 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
133 get_keydata( stdin, buf[1], &key, NULL );
134 data = ldbm_fetch( dbc->dbc_db, key );
135 print_entry( stdout, buf[1], &key, "key: ", &data,
138 free_and_close( dbc, key, data );
141 case 'L': /* get all blocks for a key from an index */
142 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
147 get_keydata( stdin, buf[1], &key, NULL );
148 if ( (idl = idl_fetch( be, dbc, key )) != NULL ) {
149 data.dptr = (char *) idl;
150 data.dsize = (idl->b_nmax + 1) * sizeof(ID);
151 print_entry( stdout, buf[1], &key, "key: ",
154 free_and_close( dbc, key, data );
157 case 't': /* traverse */
158 case 'T': /* traverse - keys only */
159 if ( (dbc = openchoice( buf[1], LDBM_READER, 1, NULL ))
161 perror( "openchoice" );
166 #ifdef HAVE_BERKELEY_DB2
167 for ( key = ldbm_firstkey( dbc->dbc_db, &cursorp );
169 key = ldbm_nextkey( dbc->dbc_db, key, cursorp ) )
171 for ( key = ldbm_firstkey( dbc->dbc_db );
173 key = ldbm_nextkey( dbc->dbc_db, key ) )
176 if ( savekey.dptr != NULL )
177 ldbm_datum_free( dbc->dbc_db, savekey );
180 data = ldbm_fetch( dbc->dbc_db, key );
182 if ( buf[0] == 't' ) {
183 print_entry( stdout, buf[1], &key,
184 "key: ", &data, "data:\n" );
186 print_entry( stdout, buf[1], &key,
187 "key: ", NULL, NULL );
190 if ( data.dptr != NULL ) {
191 ldbm_datum_free( dbc->dbc_db, data );
194 if ( savekey.dptr != NULL )
195 ldbm_datum_free( dbc->dbc_db, savekey );
197 ldbm_close( dbc->dbc_db );
200 case 'x': /* delete an entry */
201 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
206 get_keydata( stdin, buf[1], &key, NULL );
208 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
209 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
210 perror( "ldbm_delete" );
212 fprintf( stderr, "db_errno %d",
213 ldbm_errno( dbc->dbc_db ) );
218 free_and_close( dbc, key, data );
221 case 'e': /* edit an entry */
222 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
227 get_keydata( stdin, buf[1], &key, NULL );
229 data = ldbm_fetch( dbc->dbc_db, key );
230 if ( data.dptr == NULL ) {
231 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
232 perror( "ldbm_fetch" );
234 fprintf( stderr, "db_errno %d\n",
235 ldbm_errno( dbc->dbc_db ) );
237 free_and_close( dbc, key, data );
241 edit_entry( buf[1], &data );
243 if ( data.dptr == NULL ) {
244 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
245 perror( "ldbm_delete" );
247 } else if ( ldbm_store( dbc->dbc_db, key, data,
248 LDBM_REPLACE ) != 0 ) {
249 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
250 perror( "ldbm_store" );
252 fprintf( stderr, "db_errno %d\n",
253 ldbm_errno( dbc->dbc_db ) );
257 free_and_close( dbc, key, data );
260 case 'a': /* add an entry */
261 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
266 get_keydata( stdin, buf[1], &key, &data );
268 if ( ldbm_store( dbc->dbc_db, key, data, LDBM_INSERT )
270 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
271 perror( "ldbm_store" );
273 fprintf( stderr, "db_errno %d\n",
274 ldbm_errno( dbc->dbc_db ) );
278 free_and_close( dbc, key, data );
281 case 'i': /* insert an id into an index entry */
282 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
287 get_keydata( stdin, buf[1], &key, &data );
289 idl = (IDList *) data.dptr;
290 for ( id = idl_firstid( idl ); id != NOID;
291 id = idl_nextid( idl, id ) ) {
292 if ( idl_insert_key( be, dbc, key, id )
295 "idl_insert_key (%s) %ld failed\n",
301 free_and_close( dbc, key, data );
304 case 'b': /* select a backend by suffix */
305 printf( "suffix: " );
307 if ( fgets( buf, sizeof(buf), stdin ) == NULL ) {
310 buf[strlen( buf ) - 1] = '\0';
312 (void) dn_normalize( buf );
313 if ( (tbe = select_backend( buf )) == NULL ) {
314 fprintf( stderr, "unknown suffix \"%s\"\n",
321 case 'B': /* print current suffix */
323 printf( "no current backend\n" );
325 printf( "current backend has suffix \"%s\"\n",
330 case 'C': /* produce concordance of an index */
331 if ( (dbc = openchoice( 'i', LDBM_READER, 1, NULL ))
338 #ifdef HAVE_BERKELEY_DB2
339 for ( key = ldbm_firstkey( dbp, &cursorp );
341 key = ldbm_nextkey( dbp, last, cursorp ) )
343 for ( key = ldbm_firstkey( dbp ); key.dptr != NULL;
344 key = ldbm_nextkey( dbp, last ) )
347 if ( last.dptr != NULL ) {
348 ldbm_datum_free( dbp, last );
351 printf( "key(%d): (%s)\n", key.dsize,
355 free_and_close( dbc, key, last );
359 printf( "commands: l<c> => lookup index\n" );
360 printf( " L<c> => lookup index (all)\n" );
361 printf( " t<c> => traverse index\n" );
362 printf( " T<c> => traverse index keys\n" );
363 printf( " x<c> => delete from index\n" );
364 printf( " e<c> => edit index entry\n" );
365 printf( " a<c> => add index entry\n" );
366 printf( " c<c> => create index\n" );
367 printf( " i<c> => insert ids into index\n" );
368 printf( " b => change default backend\n" );
369 printf( " B => print default backend\n" );
370 printf( "where <c> is a char selecting the index:\n" );
371 printf( " c => id2children\n" );
372 printf( " d => dn2id\n" );
373 printf( " e => id2entry\n" );
374 printf( " f => arbitrary file\n" );
375 printf( " i => attribute index\n" );
384 free_and_close( struct dbcache *dbc, Datum key, Datum data )
386 ldbm_cache_really_close( be, dbc );
387 if ( key.dptr != NULL )
388 ldbm_datum_free( dbp, key );
389 if ( data.dptr != NULL )
390 ldbm_datum_free( dbp, data );
394 dnid_cmp( const void *a, const void *b )
396 return( *(const long int *)a - *(const long int *)b );
400 myrealloc( char *p, int size )
403 return( (char *) malloc( size ) );
405 return( (char *) realloc( p, size ) );
409 get_idlist( FILE *fp, Datum *data )
421 psize = 2 * sizeof(ID);
429 if ( fgets( buf, sizeof(buf), fp ) == NULL || buf[0] == '\n' )
431 if ( strncmp( buf, "nmax=", 5 ) == 0 ) {
432 nmax = atol( buf + 5 );
436 if ( psize + sizeof(ID) > pmax ) {
438 p = (IDList *) myrealloc( (char *) p, pmax );
441 if ( strncmp( buf, "nids=0", 6 ) == 0 ) {
446 p->b_ids[i++] = atol( buf );
452 printf( "%d IDs entered. Max number of ids? [%d] ", i,
454 if ( fgets( buf, sizeof(buf), fp ) != NULL &&
455 isdigit( buf[0] ) ) {
471 qsort( (void *) p->b_ids, i, sizeof(ID), dnid_cmp );
474 data->dptr = (char *) p;
475 data->dsize = (nmax + 2) * sizeof(ID);
479 get_entry( FILE *fp, Datum *data )
483 int pmax, psize, len;
488 printf( "Enter entry, <cr><cr> to end:\n" );
492 while ( fgets( buf, sizeof(buf), fp ) != NULL ) {
494 if ( psize + strlen( buf ) > pmax ) {
496 p = myrealloc( p, pmax );
504 if ( buf[0] == '\n' )
509 data->dsize = psize + 1;
513 edit_entry( char c, Datum *data )
519 WAITSTATUSTYPE status;
522 strcpy( tmpname, "/tmp/dbtestXXXXXX" );
524 if ( (fd = open( mktemp( tmpname ), O_RDWR, 0600 )) == -1 ) {
529 if ( (fd = mkstemp( tmpname )) == -1 ) {
535 fp = fdopen( fd, "w" );
536 print_entry( fp, c, NULL, NULL, data, NULL );
544 } else if ( pid == 0 ) {
547 if ( (editor = getenv( "EDITOR" )) == NULL ) {
550 execl( editor, editor, tmpname, NULL );
558 if ( waitpid( (pid_t) -1, NULL, WAIT_FLAGS ) < 0 ) {
560 if ( wait3( &status, WAIT_FLAGS, 0 ) < 0 ) {
566 if ( (fp = fopen( tmpname, "r" )) == NULL ) {
570 if ( data->dptr != NULL ) {
571 ldbm_datum_free( NULL, *data );
573 get_keydata( fp, c, NULL, data );
578 static struct dbcache *
579 openfile( char *name, int namesiz, int mode, int verbose, char c )
583 if ( name == NULL || *name == '\0' ) {
586 if ( fgets( name, namesiz, stdin ) == NULL )
588 name[strlen( name ) - 1] = '\0';
591 if ( fgets( name, namesiz, stdin ) == NULL )
593 name[strlen( name ) - 1] = '\0';
597 if ( (dbc = ldbm_cache_open( be, name, (c == 'f') ? "" : LDBM_SUFFIX,
598 LDBM_READER )) == NULL ) {
607 static struct dbcache *
608 openchoice( char c, int mode, int verbose, char **fname )
610 static char name[MAXPATHLEN];
613 case 'c': /* id2children */
614 sprintf( name, "id2children" );
616 case 'd': /* dn2id */
617 sprintf( name, "dn2id" );
619 case 'e': /* id2entry */
620 sprintf( name, "id2entry" );
622 case 'f': /* arbitrary file */
623 case 'i': /* index */
624 if ( fname != NULL && *fname != NULL ) {
625 strcpy( name, *fname );
631 printf( "specify one of [fdeci] to select file\n" );
635 if ( fname != NULL ) {
639 return( openfile( name, MAXPATHLEN, mode, verbose, c ) );
657 if ( data != NULL && data->dptr == NULL ) {
661 if ( ldbm_errno( dbp ) == 0 )
664 fprintf( stderr, "%s: db_errno %d\n", msg,
670 case 'd': /* dn2id - key is dn, data is dnid */
672 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
674 if ( data != NULL ) {
675 SAFEMEMCPY( (char *) &id, data->dptr, sizeof(ID) );
676 fprintf( fp, "%s%ld\n", dlabel ? dlabel : "", id );
680 case 'e': /* id2entry - key is dnid, data is entry */
682 SAFEMEMCPY( (char *) &id, key->dptr, sizeof(ID) );
683 fprintf( fp, "%s %lu\n", klabel, id );
685 if ( data != NULL ) {
687 fprintf( fp, "data length: %d\n", data->dsize );
690 fputs( data->dptr, fp );
695 case 'i': /* index - key is string, data is dnid[] */
697 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
699 if ( data != NULL ) {
700 idl = (IDList *) data->dptr;
703 fprintf( fp, "%s\tnmax=%ld\n\tncur=%ld\n", dlabel,
704 idl->b_nmax, idl->b_nids );
706 if ( INDIRECT_BLOCK( idl ) ) {
707 for ( i = 0; idl->b_ids[i] != NOID; i++ ) {
708 fprintf( fp, "\t%ld\n", idl->b_ids[i] );
710 } else if ( ALLIDS( idl ) ) {
711 fprintf( fp, "\tALLIDS (1..%ld)\n",
714 for ( i = 0; i < idl->b_nids; i++ ) {
715 fprintf( fp, "\t%ld\n", idl->b_ids[i] );
721 case 'f': /* arbitrary file - assume key & data are strings */
723 fprintf( fp, "%s%s\n", klabel, key->dptr );
724 if ( data != NULL ) {
725 fprintf( fp, "%s%s\n", dlabel ? dlabel : "",
731 fprintf( stderr, "specify [deci] to select a file\n" );
737 get_keydata( FILE *fp, char c, Datum *key, Datum *data )
739 static char kbuf[BUFSIZ], dbuf[BUFSIZ];
747 case 'd': /* dn2id - key is dn, data is dnid */
751 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
754 kbuf[strlen( kbuf ) - 1] = '\0';
755 key->dptr = strdup( kbuf );
756 key->dsize = strlen( kbuf ) + 1;
759 if ( data != NULL ) {
762 if ( fgets( dbuf, sizeof(dbuf), fp ) == NULL ) {
766 data->dptr = (char *) malloc( sizeof(n) );
767 memcpy( data->dptr, (char *) &n, sizeof(n) );
768 data->dsize = sizeof(n);
772 case 'e': /* id2entry - key is dnid, data is entry */
776 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
780 key->dptr = (char *) malloc( sizeof(n) );
781 memcpy( key->dptr, (char *) &n, sizeof(n) );
782 key->dsize = sizeof(n);
785 if ( data != NULL ) {
786 get_entry( fp, data );
790 case 'c': /* id2children - key is string dnid, data is dnid[] */
791 case 'i': /* index - key is string, data is dnid[] */
795 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
798 kbuf[strlen( kbuf ) - 1] = '\0';
799 key->dptr = strdup( kbuf );
800 key->dsize = strlen( kbuf ) + 1;
803 if ( data != NULL ) {
804 get_idlist( fp, data );
809 fprintf(stderr, "specify [deci] to select file type\n");