openldap/servers/slapd/back-bdb/tools.c

445 lines
9.1 KiB
C
Raw Normal View History

/* tools.c - tools for slap tools */
/* $OpenLDAP$ */
/*
2003-01-04 04:20:47 +08:00
* Copyright 1998-2003 The OpenLDAP Foundation, All Rights Reserved.
* COPYING RESTRICTIONS APPLY, see COPYRIGHT file
*/
#include "portable.h"
#include <stdio.h>
#include <ac/string.h>
#include "back-bdb.h"
static DBC *cursor = NULL;
static DBT key, data;
typedef struct dn_id {
ID id;
struct berval dn;
} dn_id;
#define HOLE_SIZE 4096
dn_id hbuf[HOLE_SIZE], *holes = hbuf;
unsigned nhmax = HOLE_SIZE;
unsigned nholes;
int bdb_tool_entry_open(
BackendDB *be, int mode )
{
/* initialize key and data thangs */
DBTzero( &key );
DBTzero( &data );
key.flags = DB_DBT_REALLOC;
data.flags = DB_DBT_REALLOC;
return 0;
}
int bdb_tool_entry_close(
BackendDB *be )
{
assert( be != NULL );
if( key.data ) {
ch_free( key.data );
key.data = NULL;
}
if( data.data ) {
ch_free( data.data );
data.data = NULL;
}
if( cursor ) {
cursor->c_close( cursor );
cursor = NULL;
}
if( nholes ) {
unsigned i;
fprintf( stderr, "Error, entries missing!\n");
for (i=0; i<nholes; i++) {
fprintf(stderr, " entry %d: %s\n",
holes[i].id, holes[i].dn.bv_val, 0);
}
return -1;
}
return 0;
}
ID bdb_tool_entry_next(
BackendDB *be )
{
int rc;
ID id;
struct bdb_info *bdb = (struct bdb_info *) be->be_private;
assert( be != NULL );
assert( slapMode & SLAP_TOOL_MODE );
assert( bdb != NULL );
if (cursor == NULL) {
rc = bdb->bi_id2entry->bdi_db->cursor(
bdb->bi_id2entry->bdi_db, NULL, &cursor,
bdb->bi_db_opflags );
if( rc != 0 ) {
return NOID;
}
}
rc = cursor->c_get( cursor, &key, &data, DB_NEXT );
if( rc != 0 ) {
return NOID;
}
if( data.data == NULL ) {
return NOID;
}
AC_MEMCPY( &id, key.data, key.size );
return id;
}
Entry* bdb_tool_entry_get( BackendDB *be, ID id )
{
int rc;
Entry *e;
struct berval bv;
assert( be != NULL );
assert( slapMode & SLAP_TOOL_MODE );
assert( data.data != NULL );
DBT2bv( &data, &bv );
rc = entry_decode( &bv, &e );
if( rc == LDAP_SUCCESS ) {
e->e_id = id;
}
#ifdef BDB_HIER
bdb_fix_dn(be, id, e);
#endif
return e;
}
int bdb_tool_next_id(
BackendDB *be,
DB_TXN *tid,
Entry *e,
struct berval *text,
int hole )
{
struct bdb_info *bdb = (struct bdb_info *) be->be_private;
struct berval dn = e->e_nname;
struct berval pdn;
int rc;
rc = bdb_dn2id( be, tid, &dn, &e->e_id, 0 );
if ( rc == DB_NOTFOUND ) {
if ( be_issuffix( be, &dn ) ) {
pdn = slap_empty_bv;
} else {
dnParent( &dn, &pdn );
e->e_nname = pdn;
rc = bdb_tool_next_id( be, tid, e, text, 1 );
if ( rc ) {
return rc;
}
}
rc = bdb_next_id( be, tid, &e->e_id );
if ( rc ) {
snprintf( text->bv_val, text->bv_len,
"next_id failed: %s (%d)",
db_strerror(rc), rc );
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
#else
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
#endif
return rc;
}
e->e_nname = dn;
rc = bdb_dn2id_add( be, tid, &pdn, e );
if ( rc ) {
snprintf( text->bv_val, text->bv_len,
"dn2id_add failed: %s (%d)",
db_strerror(rc), rc );
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
#else
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
#endif
} else if ( hole ) {
if ( nholes == nhmax - 1 ) {
if ( holes == hbuf ) {
holes = malloc( nhmax * sizeof(ID) * 2 );
AC_MEMCPY( holes, hbuf, sizeof(hbuf) );
} else {
holes = realloc( holes, nhmax * sizeof(ID) * 2 );
}
nhmax *= 2;
}
ber_dupbv( &holes[nholes].dn, &dn );
holes[nholes++].id = e->e_id;
}
} else if ( !hole ) {
unsigned i;
for ( i=0; i<nholes; i++) {
if ( holes[i].id == e->e_id ) {
int j;
free(holes[i].dn.bv_val);
for (j=i;j<nholes;j++) holes[j] = holes[j+1];
holes[j].id = 0;
nholes--;
break;
} else if ( holes[i].id > e->e_id ) {
break;
}
}
}
return rc;
}
ID bdb_tool_entry_put(
BackendDB *be,
Entry *e,
struct berval *text )
{
int rc;
struct bdb_info *bdb = (struct bdb_info *) be->be_private;
DB_TXN *tid = NULL;
2001-12-27 09:38:15 +08:00
struct berval pdn;
assert( be != NULL );
assert( slapMode & SLAP_TOOL_MODE );
assert( text );
assert( text->bv_val );
assert( text->bv_val[0] == '\0' ); /* overconservative? */
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ARGS, "=> bdb_tool_entry_put( %ld, \"%s\" )\n",
(long) e->e_id, e->e_dn, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_TRACE, "=> bdb_tool_entry_put( %ld, \"%s\" )\n",
(long) e->e_id, e->e_dn, 0 );
2002-03-27 06:40:31 +08:00
#endif
rc = TXN_BEGIN( bdb->bi_dbenv, NULL, &tid,
2002-01-29 06:57:50 +08:00
bdb->bi_db_opflags );
if( rc != 0 ) {
snprintf( text->bv_val, text->bv_len,
"txn_begin failed: %s (%d)",
db_strerror(rc), rc );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR, "=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
2002-01-29 06:57:50 +08:00
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n",
text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
2002-01-29 06:57:50 +08:00
return NOID;
}
2000-09-22 14:46:32 +08:00
/* add dn2id indices */
rc = bdb_tool_next_id( be, tid, e, text, 0 );
if( rc != 0 ) {
goto done;
}
2000-09-22 14:46:32 +08:00
/* id2entry index */
rc = bdb_id2entry_add( be, tid, e );
if( rc != 0 ) {
snprintf( text->bv_val, text->bv_len,
"id2entry_add failed: %s (%d)",
db_strerror(rc), rc );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
goto done;
}
rc = bdb_index_entry_add( be, tid, e, e->e_attrs );
if( rc != 0 ) {
snprintf( text->bv_val, text->bv_len,
"index_entry_add failed: %s (%d)",
db_strerror(rc), rc );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
2001-10-04 07:49:38 +08:00
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
goto done;
}
done:
2002-01-29 06:57:50 +08:00
if( rc == 0 ) {
rc = TXN_COMMIT( tid, 0 );
2002-01-29 06:57:50 +08:00
if( rc != 0 ) {
snprintf( text->bv_val, text->bv_len,
2002-01-29 06:57:50 +08:00
"txn_commit failed: %s (%d)",
db_strerror(rc), rc );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n",
text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
e->e_id = NOID;
}
2002-01-29 06:57:50 +08:00
} else {
TXN_ABORT( tid );
2002-01-29 06:57:50 +08:00
snprintf( text->bv_val, text->bv_len,
"txn_aborted! %s (%d)",
db_strerror(rc), rc );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_put: %s\n", text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
2002-01-29 06:57:50 +08:00
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_put: %s\n",
text->bv_val, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
2002-01-29 06:57:50 +08:00
e->e_id = NOID;
}
return e->e_id;
}
int bdb_tool_entry_reindex(
BackendDB *be,
ID id )
{
2001-10-04 07:28:20 +08:00
struct bdb_info *bi = (struct bdb_info *) be->be_private;
int rc;
Entry *e;
2001-10-04 07:28:20 +08:00
DB_TXN *tid = NULL;
2001-12-27 09:38:15 +08:00
struct berval pdn;
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ARGS,
"=> bdb_tool_entry_reindex( %ld )\n", (long) id, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ARGS, "=> bdb_tool_entry_reindex( %ld )\n",
(long) id, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
2001-10-04 07:28:20 +08:00
e = bdb_tool_entry_get( be, id );
if( e == NULL ) {
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, DETAIL1,
2002-03-27 06:40:31 +08:00
"bdb_tool_entry_reindex:: could not locate id=%ld\n",
(long) id, 0, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ANY,
"bdb_tool_entry_reindex:: could not locate id=%ld\n",
(long) id, 0, 0 );
2002-03-27 06:40:31 +08:00
#endif
return -1;
}
rc = TXN_BEGIN( bi->bi_dbenv, NULL, &tid, bi->bi_db_opflags );
2002-01-29 06:57:50 +08:00
if( rc != 0 ) {
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
2002-03-27 06:40:31 +08:00
"=> bdb_tool_entry_reindex: txn_begin failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#else
2002-01-29 06:57:50 +08:00
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_reindex: txn_begin failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#endif
2002-01-29 06:57:50 +08:00
goto done;
}
/*
* just (re)add them for now
* assume that some other routine (not yet implemented)
* will zap index databases
*
*/
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
"=> bdb_tool_entry_reindex( %ld, \"%s\" )\n", (long) id, e->e_dn, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_TRACE, "=> bdb_tool_entry_reindex( %ld, \"%s\" )\n",
(long) id, e->e_dn, 0 );
2002-03-27 06:40:31 +08:00
#endif
/* add dn2id indices */
if ( be_issuffix( be, &e->e_nname ) ) {
pdn = slap_empty_bv;
2002-01-18 03:58:19 +08:00
} else {
dnParent( &e->e_nname, &pdn );
2002-01-18 03:58:19 +08:00
}
2001-12-27 09:38:15 +08:00
rc = bdb_dn2id_add( be, tid, &pdn, e );
if( rc != 0 && rc != DB_KEYEXIST ) {
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
2002-03-27 06:40:31 +08:00
"=> bdb_tool_entry_reindex: dn2id_add failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_reindex: dn2id_add failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#endif
goto done;
}
2001-10-04 07:28:20 +08:00
rc = bdb_index_entry_add( be, tid, e, e->e_attrs );
done:
2002-01-29 06:57:50 +08:00
if( rc == 0 ) {
rc = TXN_COMMIT( tid, 0 );
2002-01-29 06:57:50 +08:00
if( rc != 0 ) {
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, ERR,
2002-03-27 06:40:31 +08:00
"=> bdb_tool_entry_reindex: txn_commit failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#else
Debug( LDAP_DEBUG_ANY,
2002-01-29 06:57:50 +08:00
"=> bdb_tool_entry_reindex: txn_commit failed: %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#endif
e->e_id = NOID;
}
2002-01-29 06:57:50 +08:00
} else {
TXN_ABORT( tid );
2002-03-27 06:40:31 +08:00
#ifdef NEW_LOGGING
LDAP_LOG ( TOOLS, DETAIL1,
2002-03-27 06:40:31 +08:00
"=> bdb_tool_entry_reindex: txn_aborted! %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#else
2002-01-29 06:57:50 +08:00
Debug( LDAP_DEBUG_ANY,
"=> bdb_tool_entry_reindex: txn_aborted! %s (%d)\n",
db_strerror(rc), rc, 0 );
2002-03-27 06:40:31 +08:00
#endif
2002-01-29 06:57:50 +08:00
e->e_id = NOID;
}
return rc;
}