openldap/servers/slapd/tools/ldif2common.c

/*
 * Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
 * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
 */
/* ldif2common.c - common definitions for the ldif2* tools */

#include "portable.h"

#include <stdio.h>

#include <ac/stdlib.h>
#include <ac/ctype.h>
#include <ac/string.h>
#include <ac/socket.h>
#include <ac/unistd.h>

#include "ldif2common.h"


char	*progname;
char	*tailorfile = SLAPD_DEFAULT_CONFIGFILE;
char	*inputfile	= NULL;
char	*sbindir    = LDAP_SBINDIR;             /* used by ldif2ldbm */
int 	cmdkids     = 1;                        /* used by ldif2ldbm */
int 	dbnum;


static void
usage( int tool )
{
	fprintf( stderr, "usage: %s %s\n\t%s%s\n",
			 progname, "-i inputfile [-d debuglevel] [-f configfile]",
			 "[-n databasenumber]",
			 ((tool == LDIF2LDBM)  ? " [-j #jobs] [-s sbindir]" :
			  (tool == LDIF2INDEX) ? " attr" :
			  "") );
	exit( EXIT_FAILURE );
}


/*
 * slap_ldif_init - initialize ldif utility, handle program options.
 * args: tool - which ldif2* program is running.
 *       argc, argv - from main.
 *       dbtype - "ldbm"/"bdb2".
 *       options - to getopt.
 */

void
slap_ldif_init( int argc, char **argv, int tool, const char *dbtype, int mode )
{
	char *options = (tool == LDIF2LDBM ? "e:s:j:d:f:i:n:" : "d:f:i:n:");
	int rc, i;

	progname = strrchr ( argv[0], '/' );
	progname = ch_strdup( progname ? progname + 1 : argv[0] );

	inputfile = NULL;
	tailorfile = SLAPD_DEFAULT_CONFIGFILE;
	dbnum = -1;
	while ( (i = getopt( argc, argv, options )) != EOF ) {
		switch ( i ) {
		case 'd':	/* turn on debugging */
			ldap_debug = atoi( optarg );
			break;

		case 's':	/* alternate sbindir (index cmd location) */
		case 'e':	/* accept -e for backwards compatibility */
			/* only used by ldif2ldbm and ldif2ldbm-bdb2 */
			sbindir = strdup( optarg );
			break;

		case 'f':	/* specify a tailor file */
			tailorfile = strdup( optarg );
			break;

		case 'i':	/* input file */
			inputfile = strdup( optarg );
			break;

		case 'j':	/* number of parallel index procs */
			/* only in ldif2ldbm and ldif2ldbm-bdb2 */
			cmdkids = atoi( optarg );
			break;

		case 'n':	/* which config file db to index */
			dbnum = atoi( optarg ) - 1;
			break;

		default:
			usage( tool );
			break;
		}
	}
	if ( inputfile == NULL || (argc != optind + (tool == LDIF2INDEX)) )
		usage( tool );

	if ( freopen( inputfile, "r", stdin ) == NULL ) {
		perror( inputfile );
		exit( EXIT_FAILURE );
	}

	/*
	 * initialize stuff and figure out which backend we're dealing with
	 */

	rc = slap_init( mode, progname );
	if (rc != 0 ) {
		fprintf( stderr, "%s: slap_init failed!\n", progname );
		exit( EXIT_FAILURE );
	}

	read_config( tailorfile );

	if ( dbnum == -1 ) {
		for ( dbnum = 0; dbnum < nbackends; dbnum++ ) {
			if ( strcasecmp( backends[dbnum].be_type, dbtype )
			    == 0 ) {
				break;
			}
		}
		if ( dbnum == nbackends ) {
			fprintf( stderr, "No %s database found in config file\n", dbtype );
			exit( EXIT_FAILURE );
		}
	} else if ( dbnum < 0 || dbnum > (nbackends-1) ) {
		fprintf( stderr, "Database number selected via -n is out of range\n" );
		fprintf( stderr, "Must be in the range 1 to %d (number of databases in the config file)\n", nbackends );
		exit( EXIT_FAILURE );
	} else if ( strcasecmp( backends[dbnum].be_type, dbtype ) != 0 ) {
		fprintf( stderr, "Database number %d selected via -n is not an %s database\n", dbnum, dbtype );
		exit( EXIT_FAILURE );
	}
}


#define IDBUFSIZ 100			/* enough to hold <decimal ID>\n\0 */

/*
 * slap_read_ldif - read an ldif record.  Return 1 for success, 0 for EOF.
 */
int
slap_read_ldif(
	int         *lno,		/* ptr to line number counter              */
	char        **bufp,     /* ptr to malloced output buffer           */
	int         *buflenp,   /* ptr to length of *bufp                  */
	ID          *idp,       /* ptr to ID number to be read/incremented */
	int         idout )     /* flag to begin *buf with ID number       */
{
	char        linebuf[IDBUFSIZ + BUFSIZ], *line;
	ber_len_t   lcur = 0, idlen, len, linesize;
	int         last_ch = '\n', found_entry = 0, stop;

	line     = linebuf + IDBUFSIZ;
	linesize = sizeof( linebuf ) - IDBUFSIZ;

	for ( stop = feof( stdin );  !stop;  last_ch = line[len-1] ) {
		if ( fgets( line, linesize, stdin ) == NULL ) {
			stop = 1;
			/* Add \n in case the file does not end with newline */
			line = "\n";
		}
		len = strlen( line );

		if ( last_ch == '\n' ) {
			(*lno)++;

			if ( line[0] == '\n' ) {
				if ( !found_entry )
					continue;
				break;
			}

			if ( !found_entry ) {
				/* Found a new entry */
				found_entry = 1;

				if ( isdigit( (unsigned char) line[0] ) ) {
					*idp = atol( line );
					if ( !idout )
						continue;
				} else {
					++(*idp);
					if ( idout ) {
						sprintf( linebuf, "%ld\n", (long) *idp );
						idlen = strlen( linebuf );
						line -= idlen;
						linesize += idlen;
						len += idlen;
						SAFEMEMCPY( line, linebuf, idlen );
					}
				}
			}			
		}

		if ( *buflenp - lcur <= len )
			*bufp = ch_realloc( *bufp, *buflenp += len + BUFSIZ );
		strcpy( *bufp + lcur, line );
		lcur += len;
	}

	return( found_entry );
}