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
 155         assert( slapMode & SLAP_TOOL_MODE );
 156         assert( id2entry != NULL );
 157
 158         if( next_id_get( be ) == NOID ) {
 159                 return NOID;
 160         }
 161
 162         e->e_id = li->li_nextid++;
 163
 164         Debug( LDAP_DEBUG_TRACE, "=> ldbm_tool_entry_put( %ld, \"%s\" )\n",
 165                 e->e_id, e->e_dn, 0 );
 166
 167         rc = index_entry_add( be, e, e->e_attrs );
 168
 169         if( rc != 0 ) {
 170                 return NOID;
 171         }
 172
 173         rc = dn2id_add( be, e->e_ndn, e->e_id );
 174
 175         if( rc != 0 ) {
 176                 return NOID;
 177         }
 178
 179         ldbm_datum_init( key );
 180         ldbm_datum_init( data );
 181
 182         key.dptr = (char *) &e->e_id;
 183         key.dsize = sizeof(ID);
 184
 185         data.dptr = entry2str( e, &len );
 186         data.dsize = len + 1;
 187
 188         /* store it */
 189         rc = ldbm_cache_store( id2entry, key, data, LDBM_REPLACE );
 190
 191         if( rc != 0 ) {
 192                 (void) dn2id_delete( be, e->e_ndn, e->e_id );
 193                 return NOID;
 194         }
 195
 196         return e->e_id;
 197 }
 198
 199 int ldbm_tool_entry_reindex(
 200         BackendDB *be,
 201         ID id )
 202 {
 203         int rc;
 204         Entry *e;
 205
 206         Debug( LDAP_DEBUG_ARGS, "=> ldbm_tool_entry_reindex( %ld )\n",
 207                 (long) id, 0, 0 );
 208
 209         e = ldbm_tool_entry_get( be, id );
 210
 211         if( e == NULL ) {
 212                 Debug( LDAP_DEBUG_ANY,
 213                         "ldbm_tool_entry_reindex:: could not locate id=%ld\n",
 214                         (long) id, 0, 0 );
 215                 return -1;
 216         }
 217
 218         /*
 219          * just (re)add them for now
 220          * assume that some other routine (not yet implemented)
 221          * will zap index databases
 222          *
 223          */
 224
 225         Debug( LDAP_DEBUG_TRACE, "=> ldbm_tool_entry_reindex( %ld, \"%s\" )\n",
 226                 id, e->e_dn, 0 );
 227
 228         rc = index_entry_add( be, e, e->e_attrs );
 229
 230         entry_free( e );
 231
 232         return rc;
 233 }
 234
 235 int ldbm_tool_sync( BackendDB *be )
 236 {
 237         struct ldbminfo *li = (struct ldbminfo *) be->be_private;
 238
 239         assert( slapMode & SLAP_TOOL_MODE );
 240
 241         if ( li->li_nextid != NOID ) {
 242                 next_id_write( be, li->li_nextid );
 243         }
 244
 245         return 0;
 246 }