6 #include <sys/socket.h>
8 #include <sys/resource.h>
10 #include <sys/param.h>
14 #include "ldapconfig.h"
16 #include "../back-ldbm/back-ldbm.h"
18 #define EDITOR "/usr/ucb/vi"
20 extern IDList *idl_fetch();
21 extern Backend *select_backend();
22 extern struct dbcache *ldbm_cache_open();
24 static struct dbcache *openchoice();
25 static void print_entry();
26 static void free_and_close();
27 static void edit_entry();
28 static void get_keydata();
36 int ldap_syslog_level;
37 int global_schemacheck;
41 char *default_referral;
42 struct objclass *global_oc;
44 pthread_t listener_tid;
45 pthread_mutex_t num_sent_mutex;
46 pthread_mutex_t entry2str_mutex;
47 pthread_mutex_t active_threads_mutex;
48 pthread_mutex_t new_conn_mutex;
49 pthread_mutex_t currenttime_mutex;
50 pthread_mutex_t replog_mutex;
51 pthread_mutex_t ops_mutex;
52 pthread_mutex_t regex_mutex;
59 Datum savekey, key, data, 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",
92 * initialize stuff and figure out which backend we're dealing with
96 read_config( tailorfile, &be, NULL );
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 somethig 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 = (idl->b_nmax + 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" );
159 for ( key = ldbm_firstkey( dbc->dbc_db, &cursorp );
161 key = ldbm_nextkey( dbc->dbc_db, key, cursorp ) )
163 for ( key = ldbm_firstkey( dbc->dbc_db );
165 key = ldbm_nextkey( dbc->dbc_db, key ) )
168 if ( savekey.dptr != NULL )
169 ldbm_datum_free( dbc->dbc_db, savekey );
172 data = ldbm_fetch( dbc->dbc_db, key );
174 if ( buf[0] == 't' ) {
175 print_entry( stdout, buf[1], &key,
176 "key: ", &data, "data:\n" );
178 print_entry( stdout, buf[1], &key,
179 "key: ", NULL, NULL );
182 ldbm_datum_free( dbc->dbc_db, data );
184 if ( savekey.dptr != NULL )
185 ldbm_datum_free( dbc->dbc_db, savekey );
187 ldbm_close( dbc->dbc_db );
190 case 'x': /* delete an entry */
191 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
196 get_keydata( stdin, buf[1], &key, NULL );
198 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
199 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
200 perror( "ldbm_delete" );
202 fprintf( stderr, "db_errno %d",
203 ldbm_errno( dbc->dbc_db ) );
208 free_and_close( dbc, key, data );
211 case 'e': /* edit an entry */
212 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
217 get_keydata( stdin, buf[1], &key, NULL );
219 data = ldbm_fetch( dbc->dbc_db, key );
220 if ( data.dptr == NULL ) {
221 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
222 perror( "ldbm_fetch" );
224 fprintf( stderr, "db_errno %d\n",
225 ldbm_errno( dbc->dbc_db ) );
227 free_and_close( dbc, key, data );
231 edit_entry( buf[1], &data );
233 if ( data.dptr == NULL ) {
234 if ( ldbm_delete( dbc->dbc_db, key ) != 0 ) {
235 perror( "ldbm_delete" );
237 } else if ( ldbm_store( dbc->dbc_db, key, data,
238 LDBM_REPLACE ) != 0 ) {
239 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
240 perror( "ldbm_store" );
242 fprintf( stderr, "db_errno %d\n",
243 ldbm_errno( dbc->dbc_db ) );
247 free_and_close( dbc, key, data );
250 case 'a': /* add an entry */
251 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
256 get_keydata( stdin, buf[1], &key, &data );
258 if ( ldbm_store( dbc->dbc_db, key, data, LDBM_INSERT )
260 if ( ldbm_errno( dbc->dbc_db ) == 0 ) {
261 perror( "ldbm_store" );
263 fprintf( stderr, "db_errno %d\n",
264 ldbm_errno( dbc->dbc_db ) );
268 free_and_close( dbc, key, data );
271 case 'i': /* insert an id into an index entry */
272 if ( (dbc = openchoice( buf[1], LDBM_WRITER, 1, NULL ))
277 get_keydata( stdin, buf[1], &key, &data );
279 idl = (IDList *) data.dptr;
280 for ( id = idl_firstid( idl ); id != NOID;
281 id = idl_nextid( idl, id ) ) {
282 if ( idl_insert_key( be, dbc, key, id )
285 "idl_insert_key (%s) %d failed\n",
291 free_and_close( dbc, key, data );
294 case 'b': /* select a backend by suffix */
295 printf( "suffix: " );
297 if ( fgets( buf, sizeof(buf), stdin ) == NULL ) {
300 buf[strlen( buf ) - 1] = '\0';
302 (void) dn_normalize( buf );
303 if ( (tbe = select_backend( buf )) == NULL ) {
304 fprintf( stderr, "unknown suffix \"%s\"\n",
311 case 'B': /* print current suffix */
313 printf( "no current backend\n" );
315 printf( "current backend has suffix \"%s\"\n",
320 case 'C': /* produce concordance of an index */
321 if ( (dbc = openchoice( 'i', LDBM_READER, 1, NULL ))
329 for ( key = ldbm_firstkey( dbp, &cursorp );
331 key = ldbm_nextkey( dbp, last, cursorp ) )
333 for ( key = ldbm_firstkey( dbp ); key.dptr != NULL;
334 key = ldbm_nextkey( dbp, last ) )
337 if ( last.dptr != NULL ) {
338 ldbm_datum_free( dbp, last );
341 printf( "key(%d): (%s)\n", key.dsize,
345 free_and_close( dbc, key, last );
349 printf( "commands: l<c> => lookup index\n" );
350 printf( " L<c> => lookup index (all)\n" );
351 printf( " t<c> => traverse index\n" );
352 printf( " T<c> => traverse index keys\n" );
353 printf( " x<c> => delete from index\n" );
354 printf( " e<c> => edit index entry\n" );
355 printf( " a<c> => add index entry\n" );
356 printf( " c<c> => create index\n" );
357 printf( " i<c> => insert ids into index\n" );
358 printf( " b => change default backend\n" );
359 printf( " B => print default backend\n" );
360 printf( "where <c> is a char selecting the index:\n" );
361 printf( " c => id2children\n" );
362 printf( " d => dn2id\n" );
363 printf( " e => id2entry\n" );
364 printf( " f => arbitrary file\n" );
365 printf( " i => attribute index\n" );
374 free_and_close( dbc, key, data )
379 ldbm_cache_really_close( be, dbc );
380 if ( key.dptr != NULL )
381 ldbm_datum_free( dbp, key );
382 if ( data.dptr != NULL )
383 ldbm_datum_free( dbp, data );
400 return( (char *) malloc( size ) );
402 return( (char *) realloc( p, size ) );
406 get_idlist( fp, data )
420 psize = 2 * sizeof(ID);
428 if ( fgets( buf, sizeof(buf), fp ) == NULL || buf[0] == '\n' )
430 if ( strncmp( buf, "nmax=", 5 ) == 0 ) {
431 nmax = atol( buf + 5 );
435 if ( psize + sizeof(ID) > pmax ) {
437 p = (IDList *) myrealloc( (char *) p, pmax );
440 if ( strncmp( buf, "nids=0", 6 ) == 0 ) {
445 p->b_ids[i++] = atol( buf );
451 printf( "%d IDs entered. Max number of ids? [%d] ", i,
453 if ( fgets( buf, sizeof(buf), fp ) != NULL &&
454 isdigit( buf[0] ) ) {
470 qsort( (void *) p->b_ids, i, sizeof(ID), (void *) dnid_cmp );
473 data->dptr = (char *) p;
474 data->dsize = (nmax + 2) * sizeof(ID);
478 get_entry( fp, data )
484 int pmax, psize, len;
489 printf( "Enter entry, <cr><cr> to end:\n" );
493 while ( fgets( buf, sizeof(buf), fp ) != NULL ) {
495 if ( psize + strlen( buf ) > pmax ) {
497 p = myrealloc( p, pmax );
505 if ( buf[0] == '\n' )
510 data->dsize = psize + 1;
514 edit_entry( c, data )
521 WAITSTATUSTYPE status;
523 strcpy( tmpname, "/tmp/dbtestXXXXXX" );
525 if ( (fd = open( mktemp( tmpname ), O_RDWR, 0600 )) == -1 ) {
530 if ( (fd = mkstemp( tmpname )) == -1 ) {
536 fp = fdopen( fd, "w" );
537 print_entry( fp, c, NULL, NULL, data, NULL );
545 } else if ( pid == 0 ) {
548 if ( (editor = getenv( "EDITOR" )) == NULL ) {
551 execl( editor, editor, tmpname, NULL );
559 if ( waitpid( (pid_t) -1, 0, WAIT_FLAGS ) < 0 ) {
561 if ( wait3( &status, WAIT_FLAGS, 0 ) < 0 ) {
567 if ( (fp = fopen( tmpname, "r" )) == NULL ) {
571 ldbm_datum_free( NULL, *data );
572 get_keydata( fp, c, NULL, data );
577 static struct dbcache *
578 openfile( name, namesiz, mode, verbose, c )
587 if ( name == NULL || *name == '\0' ) {
590 if ( fgets( name, namesiz, stdin ) == NULL )
592 name[strlen( name ) - 1] = '\0';
595 if ( fgets( name, namesiz, stdin ) == NULL )
597 name[strlen( name ) - 1] = '\0';
601 if ( (dbc = ldbm_cache_open( be, name, (c == 'f') ? "" : LDBM_SUFFIX,
602 LDBM_READER )) == NULL ) {
611 static struct dbcache *
612 openchoice( c, mode, verbose, fname )
618 static char name[MAXPATHLEN];
621 case 'c': /* id2children */
622 sprintf( name, "id2children" );
624 case 'd': /* dn2id */
625 sprintf( name, "dn2id" );
627 case 'e': /* id2entry */
628 sprintf( name, "id2entry" );
630 case 'f': /* arbitrary file */
631 case 'i': /* index */
632 if ( fname != NULL && *fname != NULL ) {
633 strcpy( name, *fname );
639 printf( "specify one of [fdeci] to select file\n" );
643 if ( fname != NULL ) {
647 return( openfile( name, MAXPATHLEN, mode, verbose, c ) );
651 print_entry( fp, c, key, klabel, data, dlabel )
664 if ( data != NULL && data->dptr == NULL ) {
668 if ( ldbm_errno( dbp ) == 0 )
671 fprintf( stderr, "%s: db_errno %d\n", msg,
677 case 'd': /* dn2id - key is dn, data is dnid */
679 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
681 if ( data != NULL ) {
682 SAFEMEMCPY( (char *) &id, data->dptr, sizeof(ID) );
683 fprintf( fp, "%s%d\n", dlabel ? dlabel : "", id );
687 case 'e': /* id2entry - key is dnid, data is entry */
689 SAFEMEMCPY( (char *) &id, key->dptr, sizeof(ID) );
690 fprintf( fp, "%s %d\n", klabel, id );
692 if ( data != NULL ) {
694 fprintf( fp, "data length: %d\n", data->dsize );
697 fputs( data->dptr, fp );
702 case 'i': /* index - key is string, data is dnid[] */
704 fprintf( fp, "%s%s (len %d)\n", klabel, key->dptr,
706 if ( data != NULL ) {
707 idl = (IDList *) data->dptr;
710 fprintf( fp, "%s\tnmax=%d\n\tncur=%d\n", dlabel,
711 idl->b_nmax, idl->b_nids );
713 if ( INDIRECT_BLOCK( idl ) ) {
714 for ( i = 0; idl->b_ids[i] != NOID; i++ ) {
715 fprintf( fp, "\t%d\n", idl->b_ids[i] );
717 } else if ( ALLIDS( idl ) ) {
718 fprintf( fp, "\tALLIDS (1..%d)\n",
721 for ( i = 0; i < idl->b_nids; i++ ) {
722 fprintf( fp, "\t%d\n", idl->b_ids[i] );
728 case 'f': /* arbitrary file - assume key & data are strings */
730 fprintf( fp, "%s%s\n", klabel, key->dptr );
731 if ( data != NULL ) {
732 fprintf( fp, "%s%s\n", dlabel ? dlabel : "",
738 fprintf( stderr, "specify [deci] to select a file\n" );
744 get_keydata( fp, c, key, data )
750 static char kbuf[BUFSIZ], dbuf[BUFSIZ];
758 case 'd': /* dn2id - key is dn, data is dnid */
762 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
765 kbuf[strlen( kbuf ) - 1] = '\0';
766 key->dptr = strdup( kbuf );
767 key->dsize = strlen( kbuf ) + 1;
770 if ( data != NULL ) {
773 if ( fgets( dbuf, sizeof(dbuf), fp ) == NULL ) {
777 data->dptr = (char *) malloc( sizeof(n) );
778 memcpy( data->dptr, (char *) &n, sizeof(n) );
779 data->dsize = sizeof(n);
783 case 'e': /* id2entry - key is dnid, data is entry */
787 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
791 key->dptr = (char *) malloc( sizeof(n) );
792 memcpy( key->dptr, (char *) &n, sizeof(n) );
793 key->dsize = sizeof(n);
796 if ( data != NULL ) {
797 get_entry( fp, data );
801 case 'c': /* id2children - key is string dnid, data is dnid[] */
802 case 'i': /* index - key is string, data is dnid[] */
806 if ( fgets( kbuf, sizeof(kbuf), fp ) == NULL ) {
809 kbuf[strlen( kbuf ) - 1] = '\0';
810 key->dptr = strdup( kbuf );
811 key->dsize = strlen( kbuf ) + 1;
814 if ( data != NULL ) {
815 get_idlist( fp, data );
820 fprintf(stderr, "specify [deci] to select file type\n");