git.sur5r.net Git - openldap/blob - servers/slapd/back-bdb/tools.c

   1 /* tools.c - tools for slap tools */
   2 /* $OpenLDAP$ */
   3 /*
   4  * Copyright 1998-2002 The OpenLDAP Foundation, All Rights Reserved.
   5  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
   6  */
   7
   8 #include "portable.h"
   9
  10 #include <stdio.h>
  11 #include <ac/string.h>
  12
  13 #include "back-bdb.h"
  14
  15 static DBC *cursor = NULL;
  16 static DBT key, data;
  17
  18 int bdb_tool_entry_open(
  19         BackendDB *be, int mode )
  20 {
  21         /* initialize key and data thangs */
  22         DBTzero( &key );
  23         DBTzero( &data );
  24         key.flags = DB_DBT_REALLOC;
  25         data.flags = DB_DBT_REALLOC;
  26
  27         return 0;
  28 }
  29
  30 int bdb_tool_entry_close(
  31         BackendDB *be )
  32 {
  33         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
  34
  35         assert( be != NULL );
  36
  37         if( key.data ) {
  38                 ch_free( key.data );
  39                 key.data = NULL;
  40         }
  41         if( data.data ) {
  42                 ch_free( data.data );
  43                 data.data = NULL;
  44         }
  45
  46         if( cursor ) {
  47                 cursor->c_close( cursor );
  48                 cursor = NULL;
  49         }
  50
  51         return 0;
  52 }
  53
  54 ID bdb_tool_entry_next(
  55         BackendDB *be )
  56 {
  57         int rc;
  58         ID id;
  59         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
  60
  61         assert( be != NULL );
  62         assert( slapMode & SLAP_TOOL_MODE );
  63         assert( bdb != NULL );
  64
  65         if (cursor == NULL) {
  66                 rc = bdb->bi_id2entry->bdi_db->cursor(
  67                         bdb->bi_id2entry->bdi_db, NULL, &cursor,
  68                         bdb->bi_db_opflags );
  69                 if( rc != 0 ) {
  70                         return NOID;
  71                 }
  72         }
  73
  74         rc = cursor->c_get( cursor, &key, &data, DB_NEXT );
  75
  76         if( rc != 0 ) {
  77                 return NOID;
  78         }
  79
  80         if( data.data == NULL ) {
  81                 return NOID;
  82         }
  83
  84         AC_MEMCPY( &id, key.data, key.size );
  85         return id;
  86 }
  87
  88 Entry* bdb_tool_entry_get( BackendDB *be, ID id )
  89 {
  90         int rc;
  91         Entry *e;
  92         struct berval bv;
  93
  94         assert( be != NULL );
  95         assert( slapMode & SLAP_TOOL_MODE );
  96         assert( data.data != NULL );
  97
  98         DBT2bv( &data, &bv );
  99
 100         rc = entry_decode( &bv, &e );
 101
 102         if( rc == LDAP_SUCCESS ) {
 103                 e->e_id = id;
 104         }
 105
 106 #ifdef BDB_HIER
 107         bdb_fix_dn(be, id, e);
 108 #endif
 109
 110         return e;
 111 }
 112
 113 ID bdb_tool_entry_put(
 114         BackendDB *be,
 115         Entry *e,
 116         struct berval *text )
 117 {
 118         int rc;
 119         struct bdb_info *bdb = (struct bdb_info *) be->be_private;
 120         DB_TXN *tid = NULL;
 121         struct berval pdn;
 122
 123         assert( be != NULL );
 124         assert( slapMode & SLAP_TOOL_MODE );
 125
 126         assert( text );
 127         assert( text->bv_val );
 128         assert( text->bv_val[0] == '\0' );
 129
 130         Debug( LDAP_DEBUG_TRACE, "=> bdb_tool_entry_put( %ld, \"%s\" )\n",
 131                 (long) e->e_id, e->e_dn, 0 );
 132
 133         if( bdb->bi_txn ) {
 134                 rc = txn_begin( bdb->bi_dbenv, NULL, &tid,
 135                         bdb->bi_db_opflags );
 136                 if( rc != 0 ) {
 137                         snprintf( text->bv_val, text->bv_len,
 138                                         "txn_begin failed: %s (%d)",
 139                                         db_strerror(rc), rc );
 140                         Debug( LDAP_DEBUG_ANY,
 141                                 "=> bdb_tool_entry_put: %s\n",
 142                                  text->bv_val, 0, 0 );
 143                         return NOID;
 144                 }
 145         }
 146
 147         rc = bdb_next_id( be, tid, &e->e_id );
 148         if( rc != 0 ) {
 149                 snprintf( text->bv_val, text->bv_len,
 150                                 "next_id failed: %s (%d)",
 151                                 db_strerror(rc), rc );
 152                 Debug( LDAP_DEBUG_ANY,
 153                         "=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
 154                 goto done;
 155         }
 156
 157         /* add dn2id indices */
 158         if ( be_issuffix( be, e->e_nname.bv_val ) ) {
 159                 pdn.bv_len = 0;
 160                 pdn.bv_val = "";
 161         } else {
 162                 rc = dnParent( e->e_nname.bv_val, &pdn.bv_val );
 163                 if ( rc != LDAP_SUCCESS ) {
 164                         Debug( LDAP_DEBUG_ANY, "=> bdb_tool_entry_put: "
 165                                 "dnParent(\"%s\") failed\n",
 166                                 e->e_nname.bv_val, 0, 0 );
 167
 168                         goto done;
 169                 }
 170                 pdn.bv_len = e->e_nname.bv_len - (pdn.bv_val - e->e_nname.bv_val);
 171         }
 172         rc = bdb_dn2id_add( be, tid, &pdn, e );
 173         if( rc != 0 ) {
 174                 snprintf( text->bv_val, text->bv_len,
 175                                 "dn2id_add failed: %s (%d)",
 176                                 db_strerror(rc), rc );
 177                 Debug( LDAP_DEBUG_ANY,
 178                         "=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
 179                 goto done;
 180         }
 181
 182         /* id2entry index */
 183         rc = bdb_id2entry_add( be, tid, e );
 184         if( rc != 0 ) {
 185                 snprintf( text->bv_val, text->bv_len,
 186                                 "id2entry_add failed: %s (%d)",
 187                                 db_strerror(rc), rc );
 188                 Debug( LDAP_DEBUG_ANY,
 189                         "=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
 190                 goto done;
 191         }
 192
 193         rc = bdb_index_entry_add( be, tid, e, e->e_attrs );
 194         if( rc != 0 ) {
 195                 snprintf( text->bv_val, text->bv_len,
 196                                 "index_entry_add failed: %s (%d)",
 197                                 db_strerror(rc), rc );
 198                 Debug( LDAP_DEBUG_ANY,
 199                         "=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
 200                 goto done;
 201         }
 202
 203 done:
 204         if( bdb->bi_txn ) {
 205                 if( rc == 0 ) {
 206                         rc = txn_commit( tid, 0 );
 207                         if( rc != 0 ) {
 208                                 snprintf( text->bv_val, text->bv_len,
 209                                                 "txn_commit failed: %s (%d)",
 210                                                 db_strerror(rc), rc );
 211                                 Debug( LDAP_DEBUG_ANY,
 212                                         "=> bdb_tool_entry_put: %s\n",
 213                                         text->bv_val, 0, 0 );
 214                                 e->e_id = NOID;
 215                         }
 216
 217                 } else {
 218                         txn_abort( tid );
 219                         snprintf( text->bv_val, text->bv_len,
 220                                         "txn_aborted! %s (%d)",
 221                                         db_strerror(rc), rc );
 222                         Debug( LDAP_DEBUG_ANY,
 223                                 "=> bdb_tool_entry_put: %s\n",
 224                                 text->bv_val, 0, 0 );
 225                         e->e_id = NOID;
 226                 }
 227         }
 228
 229         return e->e_id;
 230 }
 231
 232 int bdb_tool_entry_reindex(
 233         BackendDB *be,
 234         ID id )
 235 {
 236         struct bdb_info *bi = (struct bdb_info *) be->be_private;
 237         int rc;
 238         Entry *e;
 239         DB_TXN *tid = NULL;
 240         struct berval pdn;
 241
 242         Debug( LDAP_DEBUG_ARGS, "=> bdb_tool_entry_reindex( %ld )\n",
 243                 (long) id, 0, 0 );
 244
 245         e = bdb_tool_entry_get( be, id );
 246
 247         if( e == NULL ) {
 248                 Debug( LDAP_DEBUG_ANY,
 249                         "bdb_tool_entry_reindex:: could not locate id=%ld\n",
 250                         (long) id, 0, 0 );
 251                 return -1;
 252         }
 253
 254         if( bi->bi_txn ) {
 255                 rc = txn_begin( bi->bi_dbenv, NULL, &tid, bi->bi_db_opflags );
 256                 if( rc != 0 ) {
 257                         Debug( LDAP_DEBUG_ANY,
 258                                 "=> bdb_tool_entry_reindex: txn_begin failed: %s (%d)\n",
 259                                 db_strerror(rc), rc, 0 );
 260                         goto done;
 261                 }
 262         }
 263
 264         /*
 265          * just (re)add them for now
 266          * assume that some other routine (not yet implemented)
 267          * will zap index databases
 268          *
 269          */
 270
 271         Debug( LDAP_DEBUG_TRACE, "=> bdb_tool_entry_reindex( %ld, \"%s\" )\n",
 272                 (long) id, e->e_dn, 0 );
 273
 274         /* add dn2id indices */
 275         if ( be_issuffix( be, e->e_nname.bv_val ) ) {
 276                 pdn.bv_len = 0;
 277                 pdn.bv_val = "";
 278         } else {
 279                 rc = dnParent( e->e_nname.bv_val, &pdn.bv_val );
 280                 if ( rc != LDAP_SUCCESS ) {
 281                         Debug( LDAP_DEBUG_ANY, "=> bdb_tool_entry_reindex: "
 282                                         "dnParent(\"%s\") failed\n",
 283                                         e->e_nname.bv_val, 0, 0 );
 284                         goto done;
 285                 }
 286                 pdn.bv_len = e->e_nname.bv_len - (pdn.bv_val - e->e_nname.bv_val);
 287         }
 288         rc = bdb_dn2id_add( be, tid, &pdn, e );
 289         if( rc != 0 && rc != DB_KEYEXIST ) {
 290                 Debug( LDAP_DEBUG_ANY,
 291                         "=> bdb_tool_entry_reindex: dn2id_add failed: %s (%d)\n",
 292                         db_strerror(rc), rc, 0 );
 293                 goto done;
 294         }
 295
 296         rc = bdb_index_entry_add( be, tid, e, e->e_attrs );
 297
 298 done:
 299         if( bi->bi_txn ) {
 300                 if( rc == 0 ) {
 301                         rc = txn_commit( tid, 0 );
 302                         if( rc != 0 ) {
 303                                 Debug( LDAP_DEBUG_ANY,
 304                                         "=> bdb_tool_entry_reindex: txn_commit failed: %s (%d)\n",
 305                                         db_strerror(rc), rc, 0 );
 306                                 e->e_id = NOID;
 307                         }
 308
 309                 } else {
 310                         txn_abort( tid );
 311                         Debug( LDAP_DEBUG_ANY,
 312                                 "=> bdb_tool_entry_reindex: txn_aborted! %s (%d)\n",
 313                                 db_strerror(rc), rc, 0 );
 314                         e->e_id = NOID;
 315                 }
 316         }
 317
 318         return rc;
 319 }