11 #include <sys/param.h>
13 #include "ldapconfig.h"
15 #include "../back-ldbm/back-ldbm.h"
18 #define INDEXCMD "ldif2index"
19 #define ID2ENTRYCMD "ldif2id2entry"
20 #define ID2CHILDRENCMD "ldif2id2children"
23 static void fork_child( char *prog, char *args[] );
24 static void wait4kids( int nkidval );
26 static char *indexcmd;
27 static char *tailorfile;
28 static char *inputfile;
29 static int maxkids = 1;
35 fprintf( stderr, "usage: %s -i inputfile [-d debuglevel] [-f configfile] [-j #jobs] [-n databasenumber] [-s sbindir]\n", name );
40 main( int argc, char **argv )
43 char *linep, *buf, *sbindir;
45 char buf2[20], buf3[20];
56 struct berval *vals[2];
57 Avlnode *avltypes = NULL;
59 sbindir = DEFAULT_SBINDIR;
60 tailorfile = SLAPD_DEFAULT_CONFIGFILE;
62 while ( (i = getopt( argc, argv, "d:e:s:f:i:j:n:" )) != EOF ) {
64 case 'd': /* turn on debugging */
65 ldap_debug = atoi( optarg );
68 case 's': /* alternate sbindir (index cmd location) */
69 case 'e': /* accept -e for backwards compatibility */
70 sbindir = strdup( optarg );
73 case 'f': /* specify a tailor file */
74 tailorfile = strdup( optarg );
77 case 'i': /* input file */
78 inputfile = strdup( optarg );
81 case 'j': /* number of parallel index procs */
82 maxkids = atoi( optarg );
85 case 'n': /* which config file db to index */
86 dbnum = atoi( optarg ) - 1;
94 if ( inputfile == NULL ) {
97 if ( freopen( inputfile, "r", stdin ) == NULL ) {
104 * initialize stuff and figure out which backend we're dealing with
107 rc = slap_init(SLAP_TOOL_MODE, "ldif2ldbm");
109 fprintf( stderr, "ldif2ldbm: slap_init failed!\n");
113 read_config( tailorfile );
116 for ( dbnum = 0; dbnum < nbackends; dbnum++ ) {
117 if ( strcasecmp( backends[dbnum].be_type, "ldbm" )
122 if ( dbnum == nbackends ) {
123 fprintf( stderr, "No ldbm database found in config file\n" );
126 } else if ( dbnum < 0 || dbnum > (nbackends-1) ) {
127 fprintf( stderr, "Database number selected via -n is out of range\n" );
128 fprintf( stderr, "Must be in the range 1 to %d (number of databases in the config file)\n", nbackends );
130 } else if ( strcasecmp( backends[dbnum].be_type, "ldbm" ) != 0 ) {
131 fprintf( stderr, "Database number %d selected via -n is not an ldbm database\n", dbnum );
137 be = &backends[dbnum];
139 /* disable write sync'ing */
140 li = (struct ldbminfo *) be->be_private;
141 li->li_dbcachewsync = 0;
144 * generate the id2entry index
148 sprintf( cmd, "%s/%s", sbindir, ID2ENTRYCMD );
151 args[i++] = inputfile;
153 args[i++] = tailorfile;
155 sprintf( buf2, "%d", dbnum+1 );
158 sprintf( buf3, "%d", ldap_debug );
163 fork_child( cmd, args );
166 * generate the dn2id and id2children indexes
170 sprintf( cmd, "%s/%s", sbindir, ID2CHILDRENCMD );
173 args[i++] = inputfile;
175 args[i++] = tailorfile;
177 sprintf( buf2, "%d", dbnum+1 );
180 sprintf( buf3, "%d", ldap_debug );
185 fork_child( cmd, args );
188 * generate the attribute indexes
192 sprintf( cmd, "%s/%s", sbindir, INDEXCMD );
195 args[i++] = inputfile;
197 args[i++] = tailorfile;
199 sprintf( buf2, "%d", dbnum+1 );
202 sprintf( buf3, "%d", ldap_debug );
206 args[i++] = NULL; /* will hold the attribute name */
217 char *type, *val, *s;
218 int vlen, indexmask, syntaxmask;
221 ldbm_datum_init( key );
222 ldbm_datum_init( data );
224 if ( fgets( line, sizeof(line), stdin ) != NULL ) {
228 len = strlen( line );
229 while ( lcur + len + 1 > lmax ) {
231 buf = (char *) ch_realloc( buf, lmax );
233 strcpy( buf + lcur, line );
238 if ( line[0] == '\n' || stop && buf && *buf ) {
242 while ( (linep = ldif_getline( &s )) != NULL ) {
244 if ( ldif_parse_line( linep, &type, &val, &vlen )
246 Debug( LDAP_DEBUG_PARSE,
247 "bad line %d in entry ending at line %d ignored\n",
248 elineno, lineno, 0 );
252 if ( !isascii( *type ) || isdigit( *type ) )
255 type = strdup( type );
256 if ( avl_insert( &avltypes, type, strcasecmp,
257 avl_dup_error ) != 0 ) {
260 attr_masks( be->be_private, type,
261 &indexmask, &syntaxmask );
264 fork_child( cmd, args );
273 slap_shutdown(dbnum);
283 fork_child( char *prog, char *args[] )
287 wait4kids( maxkids );
289 switch ( pid = fork() ) {
291 execvp( prog, args );
292 fprintf( stderr, "%s: ", prog );
297 case -1: /* trouble */
298 fprintf( stderr, "Could not fork to run %s\n", prog );
302 default: /* parent */
309 wait4kids( int nkidval )
314 while ( nkids >= nkidval ) {
316 p = (unsigned char *) &status;
317 if ( p[sizeof(int) - 1] == 0177 ) {
319 "stopping: child stopped with signal %d\n",
320 p[sizeof(int) - 2] );
321 } else if ( p[sizeof(int) - 1] != 0 ) {
323 "stopping: child terminated with signal %d\n",
324 p[sizeof(int) - 1] );
325 exit( p[sizeof(int) - 1] );
326 } else if ( p[sizeof(int) - 2] != 0 ) {
328 "stopping: child exited with status %d\n",
329 p[sizeof(int) - 2] );
330 exit( p[sizeof(int) - 2] );