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

   1 /* tools.c - tools for slap tools */
   2 /* $OpenLDAP$ */
   3 /*
   4  * Copyright 1998-2000 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
  12 #include <ac/string.h>
  13 #include <ac/socket.h>
  14
  15 #include "slap.h"
  16 #include "back-ldbm.h"
  17
  18 static LDBMCursor *cursorp = NULL;
  19 static DBCache *id2entry = NULL;
  20
  21 int ldbm_tool_entry_open(
  22         BackendDB *be, int mode )
  23 {
  24         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
  25         int flags;
  26
  27         assert( slapMode & SLAP_TOOL_MODE );
  28         assert( id2entry == NULL );
  29
  30         switch( mode ) {
  31         case 1:
  32                 flags = LDBM_WRCREAT;
  33                 break;
  34         case 2:
  35 #ifdef TRUNCATE_MODE
  36                 flags = LDBM_NEWDB;
  37 #else
  38                 flags = LDBM_WRCREAT;
  39 #endif
  40                 break;
  41         default:
  42                 flags = LDBM_READER;
  43         }
  44
  45         li->li_dbwritesync = 0;
  46
  47         if ( (id2entry = ldbm_cache_open( be, "id2entry", LDBM_SUFFIX, flags ))
  48             == NULL ) {
  49                 Debug( LDAP_DEBUG_ANY, "Could not open/create id2entry" LDBM_SUFFIX "\n",
  50                     0, 0, 0 );
  51                 return( -1 );
  52         }
  53
  54         return 0;
  55 }
  56
  57 int ldbm_tool_entry_close(
  58         BackendDB *be )
  59 {
  60         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
  61
  62         assert( slapMode & SLAP_TOOL_MODE );
  63         assert( id2entry != NULL );
  64
  65         ldbm_cache_close( be, id2entry );
  66         li->li_dbwritesync = 1;
  67         id2entry = NULL;
  68
  69         return 0;
  70 }
  71
  72 ID ldbm_tool_entry_first(
  73         BackendDB *be )
  74 {
  75         Datum key;
  76         ID id;
  77
  78         assert( slapMode & SLAP_TOOL_MODE );
  79         assert( id2entry != NULL );
  80
  81         key = ldbm_firstkey( id2entry->dbc_db, &cursorp );
  82
  83         if( key.dptr == NULL ) {
  84                 return NOID;
  85         }
  86
  87         AC_MEMCPY( &id, key.dptr, key.dsize );
  88
  89         ldbm_datum_free( id2entry->dbc_db, key );
  90
  91         return id;
  92 }
  93
  94 ID ldbm_tool_entry_next(
  95         BackendDB *be )
  96 {
  97         Datum key;
  98         ID id;
  99
 100         assert( slapMode & SLAP_TOOL_MODE );
 101         assert( id2entry != NULL );
 102
 103         /* allow for NEXTID */
 104         ldbm_datum_init( key );
 105
 106         key = ldbm_nextkey( id2entry->dbc_db, key, cursorp );
 107
 108         if( key.dptr == NULL ) {
 109                 return NOID;
 110         }
 111
 112         AC_MEMCPY( &id, key.dptr, key.dsize );
 113
 114         ldbm_datum_free( id2entry->dbc_db, key );
 115
 116         return id;
 117 }
 118
 119 Entry* ldbm_tool_entry_get( BackendDB *be, ID id )
 120 {
 121         Entry *e;
 122         Datum key, data;
 123         assert( slapMode & SLAP_TOOL_MODE );
 124         assert( id2entry != NULL );
 125
 126         ldbm_datum_init( key );
 127
 128         key.dptr = (char *) &id;
 129         key.dsize = sizeof(ID);
 130
 131         data = ldbm_cache_fetch( id2entry, key );
 132
 133         if ( data.dptr == NULL ) {
 134                 return NULL;
 135         }
 136
 137         e = str2entry( data.dptr );
 138         ldbm_datum_free( id2entry->dbc_db, data );
 139
 140         if( e != NULL ) {
 141                 e->e_id = id;
 142         }
 143
 144         return e;
 145 }
 146
 147 ID ldbm_tool_entry_put(
 148         BackendDB *be,
 149         Entry *e )
 150 {
 151         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
 152         Datum key, data;
 153         int rc, len;
 154         ID id;
 155
 156         assert( slapMode & SLAP_TOOL_MODE );
 157         assert( id2entry != NULL );
 158
 159         if( next_id_get( be ) == NOID ) {
 160                 return NOID;
 161         }
 162
 163         e->e_id = li->li_nextid++;
 164
 165         Debug( LDAP_DEBUG_TRACE, "=> ldbm_tool_entry_put( %ld, \"%s\" )\n",
 166                 e->e_id, e->e_dn, 0 );
 167
 168         id = dn2id( be, e->e_ndn );
 169         if( id != NOID ) {
 170                 Debug( LDAP_DEBUG_TRACE,
 171                         "<= ldbm_tool_entry_put: \"%s\" already exists (id=%ld)\n",
 172                         e->e_ndn, id, 0 );
 173                 return NOID;
 174         }
 175
 176         rc = index_entry_add( be, e, e->e_attrs );
 177         if( rc != 0 ) {
 178                 return NOID;
 179         }
 180
 181         rc = dn2id_add( be, e->e_ndn, e->e_id );
 182         if( rc != 0 ) {
 183                 return NOID;
 184         }
 185
 186         ldbm_datum_init( key );
 187         ldbm_datum_init( data );
 188
 189         key.dptr = (char *) &e->e_id;
 190         key.dsize = sizeof(ID);
 191
 192         data.dptr = entry2str( e, &len );
 193         data.dsize = len + 1;
 194
 195         /* store it */
 196         rc = ldbm_cache_store( id2entry, key, data, LDBM_REPLACE );
 197
 198         if( rc != 0 ) {
 199                 (void) dn2id_delete( be, e->e_ndn, e->e_id );
 200                 return NOID;
 201         }
 202
 203         return e->e_id;
 204 }
 205
 206 int ldbm_tool_entry_reindex(
 207         BackendDB *be,
 208         ID id )
 209 {
 210         int rc;
 211         Entry *e;
 212
 213         Debug( LDAP_DEBUG_ARGS, "=> ldbm_tool_entry_reindex( %ld )\n",
 214                 (long) id, 0, 0 );
 215
 216         e = ldbm_tool_entry_get( be, id );
 217
 218         if( e == NULL ) {
 219                 Debug( LDAP_DEBUG_ANY,
 220                         "ldbm_tool_entry_reindex:: could not locate id=%ld\n",
 221                         (long) id, 0, 0 );
 222                 return -1;
 223         }
 224
 225         /*
 226          * just (re)add them for now
 227          * assume that some other routine (not yet implemented)
 228          * will zap index databases
 229          *
 230          */
 231
 232         Debug( LDAP_DEBUG_TRACE, "=> ldbm_tool_entry_reindex( %ld, \"%s\" )\n",
 233                 id, e->e_dn, 0 );
 234
 235         rc = index_entry_add( be, e, e->e_attrs );
 236
 237         entry_free( e );
 238
 239         return rc;
 240 }
 241
 242 int ldbm_tool_sync( BackendDB *be )
 243 {
 244         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
 245
 246         assert( slapMode & SLAP_TOOL_MODE );
 247
 248         if ( li->li_nextid != NOID ) {
 249                 next_id_write( be, li->li_nextid );
 250         }
 251
 252         return 0;
 253 }