]> git.sur5r.net Git - openldap/blob - servers/slapd/tools/slapcommon.c
Added bdb_attribute and bdb_group ACL support routines
[openldap] / servers / slapd / tools / slapcommon.c
1 /* $OpenLDAP$ */
2 /*
3  * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
4  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
5  */
6 /* slapcommon.c - common routine for the slap tools */
7
8 #include "portable.h"
9
10 #include <stdio.h>
11
12 #include <ac/stdlib.h>
13 #include <ac/ctype.h>
14 #include <ac/string.h>
15 #include <ac/socket.h>
16 #include <ac/unistd.h>
17
18 #include "slapcommon.h"
19 #include "lutil.h"
20
21
22 char    *progname       = NULL;
23 char    *conffile       = SLAPD_DEFAULT_CONFIGFILE;
24 int             truncatemode = 0;
25 int             verbose         = 0;
26 int             continuemode = 0;
27
28 char    *ldiffile       = NULL;
29 FILE    *ldiffp         = NULL;
30
31 #ifdef CSRIMALLOC
32         char *leakfilename;
33         FILE *leakfile;
34 #endif
35
36 Backend *be = NULL;
37
38 static void
39 usage( int tool )
40 {
41         char *options = NULL;
42         fprintf( stderr,
43                 "usage: %s [-v] [-c] [-d debuglevel] [-f configfile]\n"
44                         "\t[-n databasenumber | -b suffix]", progname );
45
46         switch( tool ) {
47         case SLAPADD:
48                 options = "\t[-l ldiffile]\n";
49                 break;
50
51         case SLAPCAT:
52                 options = "\t[-l ldiffile]\n";
53                 break;
54
55         case SLAPINDEX:
56                 options = "\n";
57                 break;
58         }
59
60         if( options != NULL ) {
61                 fputs( options, stderr );
62         }
63         exit( EXIT_FAILURE );
64 }
65
66
67 /*
68  * slap_tool_init - initialize slap utility, handle program options.
69  * arguments:
70  *      name            program name
71  *      tool            tool code
72  *      argc, argv      command line arguments
73  */
74
75 void
76 slap_tool_init(
77         const char* name,
78         int tool,
79         int argc, char **argv )
80 {
81         char *options;
82         char *base = NULL;
83         int rc, i, dbnum;
84         int mode = SLAP_TOOL_MODE;
85
86         progname = lutil_progname( name, argc, argv );
87
88 #ifdef CSRIMALLOC
89         leakfilename = malloc( strlen( progname ) + sizeof(".leak") );
90         sprintf( leakfilename, "%s.leak", progname );
91         if( ( leakfile = fopen( leakfilename, "w" )) == NULL ) {
92                 leakfile = stderr;
93         }
94         free( leakfilename );
95 #endif
96
97         switch( tool ) {
98         case SLAPADD:
99                 options = "b:cd:f:l:n:tv";
100                 break;
101
102         case SLAPINDEX:
103                 options = "b:cd:f:n:v";
104                 break;
105
106         case SLAPCAT:
107                 options = "b:cd:f:l:n:v";
108                 break;
109
110         default:
111                 fprintf( stderr, "%s: unknown tool mode (%d)\n",
112                          progname, tool );
113                 exit( EXIT_FAILURE );
114         }
115
116         ldiffile = NULL;
117         conffile = SLAPD_DEFAULT_CONFIGFILE;
118         dbnum = -1;
119         while ( (i = getopt( argc, argv, options )) != EOF ) {
120                 switch ( i ) {
121                 case 'b':
122                         base = strdup( optarg );
123
124                 case 'c':       /* enable continue mode */
125                         continuemode++;
126                         break;
127
128                 case 'd':       /* turn on debugging */
129                         ldap_debug += atoi( optarg );
130                         break;
131
132                 case 'f':       /* specify a conf file */
133                         conffile = strdup( optarg );
134                         break;
135
136                 case 'l':       /* LDIF file */
137                         ldiffile = strdup( optarg );
138                         break;
139
140                 case 'n':       /* which config file db to index */
141                         dbnum = atoi( optarg ) - 1;
142                         break;
143
144                 case 't':       /* turn on truncate */
145                         truncatemode++;
146                         mode |= SLAP_TRUNCATE_MODE;
147                         break;
148
149                 case 'v':       /* turn on verbose */
150                         verbose++;
151                         break;
152
153                 default:
154                         usage( tool );
155                         break;
156                 }
157         }
158
159         if ( ( argc != optind ) || (dbnum >= 0 && base != NULL ) ) {
160                 usage( tool );
161         }
162
163         if ( ldiffile == NULL ) {
164                 ldiffp = tool == SLAPCAT ? stdout : stdin;
165
166         } else if( (ldiffp = fopen( ldiffile, tool == SLAPCAT ? "w" : "r" ))
167                 == NULL )
168         {
169                 perror( ldiffile );
170                 exit( EXIT_FAILURE );
171         }
172
173         /*
174          * initialize stuff and figure out which backend we're dealing with
175          */
176
177         rc = slap_init( mode, progname );
178
179         if (rc != 0 ) {
180                 fprintf( stderr, "%s: slap_init failed!\n", progname );
181                 exit( EXIT_FAILURE );
182         }
183
184         rc = schema_init();
185
186         if (rc != 0 ) {
187                 fprintf( stderr, "%s: slap_schema_init failed!\n", progname );
188                 exit( EXIT_FAILURE );
189         }
190
191         read_config( conffile );
192
193         if ( !nbackends ) {
194                 fprintf( stderr, "No databases found in config file\n" );
195                 exit( EXIT_FAILURE );
196         }
197
198         rc = schema_prep();
199
200         if (rc != 0 ) {
201                 fprintf( stderr, "%s: slap_schema_prep failed!\n", progname );
202                 exit( EXIT_FAILURE );
203         }
204
205         if( base != NULL ) {
206                 char *tbase = ch_strdup( base );
207
208                 if( dn_normalize( tbase ) == NULL ) {
209                         fprintf( stderr, "%s: slap_init invalid suffix (\"%s\")\n",
210                                 progname, base );
211                         exit( EXIT_FAILURE );
212                 }
213
214                 be = select_backend( tbase, 0 );
215                 free( tbase );
216
217                 if( be == NULL ) {
218                         fprintf( stderr, "%s: slap_init no backend for \"%s\"\n",
219                                 progname, base );
220                         exit( EXIT_FAILURE );
221                 }
222
223         } else if ( dbnum == -1 ) {
224                 be = &backends[dbnum=0];
225
226         } else if ( dbnum < 0 || dbnum > (nbackends-1) ) {
227                 fprintf( stderr,
228                         "Database number selected via -n is out of range\n"
229                         "Must be in the range 1 to %d (number of databases in the config file)\n",
230                         nbackends );
231                 exit( EXIT_FAILURE );
232
233         } else {
234                 be = &backends[dbnum];
235         }
236
237 #ifdef CSRIMALLOC
238         mal_leaktrace(1);
239 #endif
240
241         slap_startup( be );
242 }
243
244 void slap_tool_destroy( void )
245 {
246         slap_shutdown( be );
247         slap_destroy();
248
249 #ifdef CSRIMALLOC
250         mal_dumpleaktrace( leakfile );
251 #endif
252 }