git.sur5r.net Git - openldap/blob - servers/slapd/tools/ldif2id2entry-bdb2.c

   1 #include "portable.h"
   2
   3 #include <stdio.h>
   4
   5 #include <ac/stdlib.h>
   6
   7 #include <ac/ctype.h>
   8 #include <ac/string.h>
   9 #include <ac/socket.h>
  10 #include <ac/unistd.h>
  11
  12 #include "ldap_defaults.h"
  13 #include "../slap.h"
  14 #include "../back-bdb2/back-bdb2.h"
  15
  16 #define MAXARGS                 100
  17
  18 static char     *tailorfile;
  19 static char     *inputfile;
  20
  21 static void
  22 usage( char *name )
  23 {
  24         fprintf( stderr, "usage: %s -i inputfile [-d debuglevel] [-f configfile] [-n databasenumber]\n", name );
  25         exit( 1 );
  26 }
  27
  28 int
  29 main( int argc, char **argv )
  30 {
  31         int             i, stop;
  32         char            *buf;
  33         char            line[BUFSIZ], idbuf[BUFSIZ];
  34         int             lmax, lcur;
  35         int             dbnum;
  36         ID              id;
  37         struct dbcache  *db;
  38         Backend         *be = NULL;
  39         struct ldbminfo *li;
  40         struct berval   bv;
  41         struct berval   *vals[2];
  42         FILE            *fp;
  43
  44         tailorfile = SLAPD_DEFAULT_CONFIGFILE;
  45         dbnum = -1;
  46         while ( (i = getopt( argc, argv, "d:f:i:n:" )) != EOF ) {
  47                 switch ( i ) {
  48                 case 'd':       /* turn on debugging */
  49                         ldap_debug = atoi( optarg );
  50                         break;
  51
  52                 case 'f':       /* specify a tailor file */
  53                         tailorfile = strdup( optarg );
  54                         break;
  55
  56                 case 'i':       /* input file */
  57                         inputfile = strdup( optarg );
  58                         break;
  59
  60                 case 'n':       /* which config file db to index */
  61                         dbnum = atoi( optarg ) - 1;
  62                         break;
  63
  64                 default:
  65                         usage( argv[0] );
  66                         break;
  67                 }
  68         }
  69         if ( inputfile == NULL ) {
  70                 usage( argv[0] );
  71         } else {
  72                 if ( freopen( inputfile, "r", stdin ) == NULL ) {
  73                         perror( inputfile );
  74                         exit( 1 );
  75                 }
  76         }
  77
  78         /*
  79          * initialize stuff and figure out which backend we're dealing with
  80          */
  81
  82         slap_init(SLAP_TOOLID_MODE, "ldif2id2entry");
  83         read_config( tailorfile );
  84
  85         if ( dbnum == -1 ) {
  86                 for ( dbnum = 0; dbnum < nbackends; dbnum++ ) {
  87                         if ( strcasecmp( backends[dbnum].be_type, "bdb2" )
  88                             == 0 ) {
  89                                 break;
  90                         }
  91                 }
  92                 if ( dbnum == nbackends ) {
  93                         fprintf( stderr, "No bdb2 database found in config file\n" );
  94                         exit( 1 );
  95                 }
  96         } else if ( dbnum < 0 || dbnum > (nbackends-1) ) {
  97                 fprintf( stderr, "Database number selected via -n is out of range\n" );
  98                 fprintf( stderr, "Must be in the range 1 to %d (number of databases in the config file)\n", nbackends );
  99                 exit( 1 );
 100         } else if ( strcasecmp( backends[dbnum].be_type, "bdb2" ) != 0 ) {
 101                 fprintf( stderr, "Database number %d selected via -n is not an bdb2 database\n", dbnum );
 102                 exit( 1 );
 103         }
 104
 105         slap_startup(dbnum);
 106
 107         be = &backends[dbnum];
 108
 109         /* disable write sync'ing */
 110         li = (struct ldbminfo *) be->be_private;
 111         li->li_dbcachewsync = 0;
 112
 113         if ( (db = bdb2i_cache_open( be, "id2entry", BDB2_SUFFIX, LDBM_NEWDB ))
 114             == NULL ) {
 115                 perror( "id2entry file" );
 116                 exit( 1 );
 117         }
 118
 119         id = 0;
 120         stop = 0;
 121         buf = NULL;
 122         lcur = lmax = 0;
 123         vals[0] = &bv;
 124         vals[1] = NULL;
 125         while ( ! stop ) {
 126                 Datum           key, data;
 127
 128                 ldbm_datum_init( key );
 129                 ldbm_datum_init( data );
 130
 131                 if ( fgets( line, sizeof(line), stdin ) != NULL ) {
 132                         int     len, idlen;
 133
 134                         len = strlen( line );
 135                         if ( buf == NULL || *buf == '\0' ) {
 136                                 if (!isdigit((unsigned char) line[0])) {
 137                                         sprintf( idbuf, "%ld\n", id + 1 );
 138                                         idlen = strlen( idbuf );
 139                                 } else {
 140                                         id = atol(line) - 1;
 141                                         idlen = 0;
 142                                 }
 143                         } else {
 144                                 idlen = 0;
 145                         }
 146
 147                         while ( lcur + len + idlen + 1 > lmax ) {
 148                                 lmax += BUFSIZ;
 149                                 buf = (char *) ch_realloc( buf, lmax );
 150                         }
 151
 152                         if ( idlen > 0 ) {
 153                                 strcpy( buf + lcur, idbuf );
 154                                 lcur += idlen;
 155                         }
 156                         strcpy( buf + lcur, line );
 157                         lcur += len;
 158                 } else {
 159                         stop = 1;
 160                 }
 161                 if ( line[0] == '\n' || stop && buf && *buf ) {
 162                         if ( *buf != '\n' ) {
 163                                 int len;
 164
 165                                 id++;
 166                                 key.dptr = (char *) &id;
 167                                 key.dsize = sizeof(ID);
 168                                 data.dptr = buf;
 169                                 len = strlen(buf);
 170                                 if (buf[len - 1] == '\n')
 171                                         buf[--len] = '\0';
 172                                 data.dsize = len + 1;
 173                                 if ( ldbm_store( db->dbc_db, key, data,
 174                                     LDBM_INSERT ) != 0 ) {
 175                                         fputs("id2entry ldbm_store failed\n",
 176                                               stderr);
 177                                         exit( 1 );
 178                                 }
 179                         }
 180                         *buf = '\0';
 181                         lcur = 0;
 182                         line[0] = '\0';
 183                 }
 184         }
 185
 186         id++;
 187         bdb2i_put_nextid( be, id );
 188
 189 #ifdef SLAP_CLEANUP
 190         bdb2i_cache_close( be, db );
 191 #endif
 192
 193         slap_shutdown(dbnum);
 194
 195         slap_destroy();
 196
 197         exit( 0 );
 198 }