mirror of
https://git.openldap.org/openldap/openldap.git
synced 2024-12-21 03:10:25 +08:00
204 lines
3.7 KiB
C
204 lines
3.7 KiB
C
|
/*
|
||
|
* Copyright 1998-1999 The OpenLDAP Foundation, All Rights Reserved.
|
||
|
* COPYING RESTRICTIONS APPLY, see COPYRIGHT file
|
||
|
*/
|
||
|
/* slapcommon.c - common routine for the slap 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 "slapcommon.h"
|
||
|
#include "lutil.h"
|
||
|
|
||
|
|
||
|
char *progname = NULL;
|
||
|
char *conffile = SLAPD_DEFAULT_CONFIGFILE;
|
||
|
int truncatemode = 0;
|
||
|
int verbose = 0;
|
||
|
|
||
|
char *ldiffile = NULL;
|
||
|
FILE *ldiffp = NULL;
|
||
|
|
||
|
Backend *be = NULL;
|
||
|
|
||
|
static void
|
||
|
usage( int tool )
|
||
|
{
|
||
|
char *options = NULL;
|
||
|
fprintf( stderr,
|
||
|
"usage: %s [-v] [-d debuglevel] [-f configfile]\n"
|
||
|
"\t[-n databasenumber | -b suffix]", progname );
|
||
|
|
||
|
switch( tool ) {
|
||
|
case SLAPADD:
|
||
|
options = "\t[-l ldiffile]\n";
|
||
|
break;
|
||
|
|
||
|
case SLAPCAT:
|
||
|
options = "\t[-l ldiffile]\n";
|
||
|
break;
|
||
|
|
||
|
case SLAPINDEX:
|
||
|
options = "\tattributetype\n";
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
if( options != NULL ) {
|
||
|
fputs( options, stderr );
|
||
|
}
|
||
|
exit( EXIT_FAILURE );
|
||
|
}
|
||
|
|
||
|
|
||
|
/*
|
||
|
* slap_tool_init - initialize slap utility, handle program options.
|
||
|
* arguments:
|
||
|
* name program name
|
||
|
* tool tool code
|
||
|
* argc, argv command line arguments
|
||
|
*/
|
||
|
|
||
|
void
|
||
|
slap_tool_init(
|
||
|
const char* name,
|
||
|
int tool,
|
||
|
int argc, char **argv )
|
||
|
{
|
||
|
char *options;
|
||
|
char *base = NULL;
|
||
|
int rc, i, dbnum;
|
||
|
int mode = SLAP_TOOL_MODE;
|
||
|
|
||
|
progname = lutil_progname( name, argc, argv );
|
||
|
|
||
|
switch( tool ) {
|
||
|
case SLAPADD:
|
||
|
options = "b:d:f:l:n:tv";
|
||
|
break;
|
||
|
|
||
|
case SLAPINDEX:
|
||
|
options = "b:d:f:n:v";
|
||
|
break;
|
||
|
|
||
|
case SLAPCAT:
|
||
|
options = "b:d:f:l:n:v";
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
fprintf( stderr, "%s: unknown tool mode (%d)\n", tool );
|
||
|
exit( EXIT_FAILURE );
|
||
|
}
|
||
|
|
||
|
ldiffile = NULL;
|
||
|
conffile = SLAPD_DEFAULT_CONFIGFILE;
|
||
|
dbnum = -1;
|
||
|
while ( (i = getopt( argc, argv, options )) != EOF ) {
|
||
|
switch ( i ) {
|
||
|
case 'b':
|
||
|
base = strdup( optarg );
|
||
|
|
||
|
case 'd': /* turn on debugging */
|
||
|
ldap_debug = atoi( optarg );
|
||
|
break;
|
||
|
|
||
|
case 'f': /* specify a conf file */
|
||
|
conffile = strdup( optarg );
|
||
|
break;
|
||
|
|
||
|
case 'l': /* LDIF file */
|
||
|
ldiffile = strdup( optarg );
|
||
|
break;
|
||
|
|
||
|
case 'n': /* which config file db to index */
|
||
|
dbnum = atoi( optarg ) - 1;
|
||
|
break;
|
||
|
|
||
|
case 't': /* turn on truncate */
|
||
|
truncatemode++;
|
||
|
mode |= SLAP_TRUNCATE_MODE;
|
||
|
break;
|
||
|
|
||
|
case 'v': /* turn on verbose */
|
||
|
verbose++;
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
usage( tool );
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if ( ( argc != optind + (tool == SLAPINDEX ? 1 : 0) )
|
||
|
|| (dbnum >= 0 && base != NULL ) )
|
||
|
{
|
||
|
usage( tool );
|
||
|
}
|
||
|
|
||
|
if ( ldiffile == NULL ) {
|
||
|
ldiffp = tool == SLAPCAT ? stdout : stdin;
|
||
|
|
||
|
} else if( (ldiffp = fopen( ldiffile, tool == SLAPCAT ? "w" : "r" ))
|
||
|
== NULL )
|
||
|
{
|
||
|
perror( ldiffile );
|
||
|
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( conffile );
|
||
|
|
||
|
if ( !nbackends ) {
|
||
|
fprintf( stderr, "No databases found in config file\n" );
|
||
|
exit( EXIT_FAILURE );
|
||
|
}
|
||
|
|
||
|
if( base != NULL ) {
|
||
|
char *tbase = ch_strdup( base );
|
||
|
|
||
|
if( dn_normalize_case( tbase ) == NULL ) {
|
||
|
fprintf( stderr, "%s: slap_init invalid suffix (\"%s\")\n",
|
||
|
progname, base );
|
||
|
exit( EXIT_FAILURE );
|
||
|
}
|
||
|
|
||
|
be = select_backend( tbase );
|
||
|
free( tbase );
|
||
|
|
||
|
if( be == NULL ) {
|
||
|
fprintf( stderr, "%s: slap_init no backend for \"%s\"\n",
|
||
|
progname, base );
|
||
|
exit( EXIT_FAILURE );
|
||
|
}
|
||
|
|
||
|
} else if ( dbnum == -1 ) {
|
||
|
be = &backends[dbnum=0];
|
||
|
|
||
|
} else if ( dbnum < 0 || dbnum > (nbackends-1) ) {
|
||
|
fprintf( stderr,
|
||
|
"Database number selected via -n is out of range\n"
|
||
|
"Must be in the range 1 to %d (number of databases in the config file)\n",
|
||
|
nbackends );
|
||
|
exit( EXIT_FAILURE );
|
||
|
|
||
|
} else {
|
||
|
be = &backends[dbnum];
|
||
|
}
|
||
|
}
|