1 /* ldbm.c - ldap dbm compatibility routines */
3 /* Patched for Berkeley DB version 2.0; /KSp; 98/02/23
5 * - DB version 2.6.4b ; 1998/12/28, /KSp
6 * - DB_DBT_MALLOC ; 1998/03/22, /KSp
7 * - basic implementation; 1998/02/23, /KSp
16 #include <ac/stdlib.h>
17 #include <ac/string.h>
21 #include "ldap_pvt_thread.h"
25 ldbm_datum_free( LDBM ldbm, Datum data )
29 memset( &data, 0, sizeof( Datum ));
36 ldbm_datum_dup( LDBM ldbm, Datum data )
40 ldbm_datum_init( dup );
42 if ( data.dsize == 0 ) {
48 dup.dsize = data.dsize;
49 if ( (dup.dptr = (char *) malloc( data.dsize )) != NULL )
50 memcpy( dup.dptr, data.dptr, data.dsize );
55 static int ldbm_initialized = 0;
57 #ifndef HAVE_BERKELEY_DB2
58 /* Everything but DB2 is non-reentrant */
60 static ldap_pvt_thread_mutex_t ldbm_big_mutex;
61 #define LDBM_LOCK (ldap_pvt_thread_mutex_lock(&ldbm_big_mutex))
62 #define LDBM_UNLOCK (ldap_pvt_thread_mutex_unlock(&ldbm_big_mutex))
64 int ldbm_initialize( void )
66 if(ldbm_initialized++) return 1;
68 ldap_pvt_thread_mutex_init( &ldbm_big_mutex );
73 int ldbm_shutdown( void )
75 if( !ldbm_initialized ) return 1;
77 ldap_pvt_thread_mutex_destroy( &ldbm_big_mutex );
85 ldbm_malloc( size_t size )
87 return( calloc( 1, size ));
91 #include <ac/syslog.h>
95 ldbm_db_errcall( const char *prefix, char *message )
98 syslog( LOG_INFO, "ldbm_db_errcall(): %s %s", prefix, message );
102 /* a dbEnv for BERKELEYv2 */
103 static DB_ENV ldbm_Env_internal;
104 DB_ENV *ldbm_Env = NULL;
106 /* Berkeley DB 2.x is reentrant */
107 #define LDBM_LOCK ((void)0)
108 #define LDBM_UNLOCK ((void)0)
110 int ldbm_initialize( void )
115 if(ldbm_initialized++) return 1;
117 memset( &ldbm_Env_internal, 0, sizeof( DB_ENV ));
118 ldbm_Env = &ldbm_Env_internal;
120 ldbm_Env->db_errcall = ldbm_db_errcall;
121 ldbm_Env->db_errpfx = "==>";
123 envFlags = DB_CREATE | DB_THREAD;
125 if ( ( err = db_appinit( NULL, NULL, ldbm_Env, envFlags )) ) {
129 sprintf( error, "%ld\n", (long) err );
131 sprintf( error, "%s\n", strerror( err ));
136 "ldbm_initialize(): FATAL error in db_appinit() : %s\n",
145 int ldbm_shutdown( void )
147 if( !ldbm_initialized ) return 1;
149 db_appexit( ldbm_Env );
156 #if defined( LDBM_USE_DBHASH ) || defined( LDBM_USE_DBBTREE )
158 /*****************************************************************
160 * use berkeley db hash or btree package *
162 *****************************************************************/
165 ldbm_open( char *name, int rw, int mode, int dbcachesize )
169 #ifdef HAVE_BERKELEY_DB2
172 memset( &dbinfo, 0, sizeof( dbinfo ));
174 #if defined( DB_VERSION_MAJOR ) && defined( DB_VERSION_MINOR ) && \
175 DB_VERSION_MAJOR == 2 && DB_VERSION_MINOR == 4
177 * BerkeleyDB 2.4 do not allow db_cachesize
178 * to be specified if an DB_ENV is.
181 /* set db_cachesize of MPOOL is NOT being used. */
182 if (( ldbm_Env == NULL ) || ( ldbm_Env->mp_info == NULL ))
183 dbinfo.db_cachesize = dbcachesize;
186 dbinfo.db_pagesize = DEFAULT_DB_PAGE_SIZE;
187 dbinfo.db_malloc = ldbm_malloc;
190 (void) db_open( name, DB_TYPE, rw, mode, ldbm_Env, &dbinfo, &ret );
198 if ( DB_TYPE == DB_HASH ) {
199 memset( (char *) &hinfo, '\0', sizeof(hinfo) );
200 hinfo.cachesize = dbcachesize;
202 } else if ( DB_TYPE == DB_BTREE ) {
203 memset( (char *) &binfo, '\0', sizeof(binfo) );
204 binfo.cachesize = dbcachesize;
211 ret = dbopen( name, rw, mode, DB_TYPE, info );
220 ldbm_close( LDBM ldbm )
223 #ifdef HAVE_BERKELEY_DB2
224 (*ldbm->close)( ldbm, 0 );
226 (*ldbm->close)( ldbm );
232 ldbm_sync( LDBM ldbm )
235 (*ldbm->sync)( ldbm, 0 );
240 ldbm_fetch( LDBM ldbm, Datum key )
247 #ifdef HAVE_BERKELEY_DB2
248 ldbm_datum_init( data );
250 data.flags = DB_DBT_MALLOC;
252 if ( (rc = (*ldbm->get)( ldbm, NULL, &key, &data, 0 )) != 0 ) {
253 ldbm_datum_free( ldbm, data );
255 if ( (rc = (*ldbm->get)( ldbm, &key, &data, 0 )) == 0 ) {
256 /* Berkeley DB 1.85 don't malloc the data for us */
257 /* duplicate it for to ensure reentrancy */
258 data = ldbm_datum_dup( ldbm, data );
271 ldbm_store( LDBM ldbm, Datum key, Datum data, int flags )
277 #ifdef HAVE_BERKELEY_DB2
278 rc = (*ldbm->put)( ldbm, NULL, &key, &data, flags & ~LDBM_SYNC );
281 rc = (*ldbm->put)( ldbm, &key, &data, flags & ~LDBM_SYNC );
284 if ( flags & LDBM_SYNC )
285 (*ldbm->sync)( ldbm, 0 );
293 ldbm_delete( LDBM ldbm, Datum key )
299 #ifdef HAVE_BERKELEY_DB2
300 rc = (*ldbm->del)( ldbm, NULL, &key, 0 );
303 rc = (*ldbm->del)( ldbm, &key, 0 );
305 (*ldbm->sync)( ldbm, 0 );
313 #ifdef HAVE_BERKELEY_DB2
314 ldbm_firstkey( LDBM ldbm, DBC **dbch )
316 ldbm_firstkey( LDBM ldbm )
321 #ifdef HAVE_BERKELEY_DB2
324 ldbm_datum_init( key );
325 ldbm_datum_init( data );
327 key.flags = data.flags = DB_DBT_MALLOC;
331 /* acquire a cursor for the DB */
333 # if defined( DB_VERSION_MAJOR ) && defined( DB_VERSION_MINOR ) && \
334 DB_VERSION_MAJOR == 2 && DB_VERSION_MINOR < 6
336 if ( (*ldbm->cursor)( ldbm, NULL, &dbci ))
339 if ( (*ldbm->cursor)( ldbm, NULL, &dbci, 0 ))
346 if ( (*dbci->c_get)( dbci, &key, &data, DB_NEXT ) == 0 ) {
347 ldbm_datum_free( ldbm, data );
355 if ( (rc = (*ldbm->seq)( ldbm, &key, &data, R_FIRST )) == 0 ) {
356 key = ldbm_datum_dup( ldbm, key );
364 #ifdef HAVE_BERKELEY_DB2
374 #ifdef HAVE_BERKELEY_DB2
375 ldbm_nextkey( LDBM ldbm, Datum key, DBC *dbcp )
377 ldbm_nextkey( LDBM ldbm, Datum key )
382 #ifdef HAVE_BERKELEY_DB2
383 ldbm_datum_init( data );
385 ldbm_datum_free( ldbm, key );
386 key.flags = data.flags = DB_DBT_MALLOC;
390 if ( (*dbcp->c_get)( dbcp, &key, &data, DB_NEXT ) == 0 ) {
391 ldbm_datum_free( ldbm, data );
399 if ( (rc = (*ldbm->seq)( ldbm, &key, &data, R_NEXT )) == 0 ) {
400 key = ldbm_datum_dup( ldbm, key );
414 ldbm_errno( LDBM ldbm )
419 #elif defined( HAVE_GDBM )
421 #ifdef HAVE_ST_BLKSIZE
422 #include <sys/stat.h>
425 /*****************************************************************
429 *****************************************************************/
432 ldbm_open( char *name, int rw, int mode, int dbcachesize )
435 #ifdef HAVE_ST_BLKSIZE
441 if ( (db = gdbm_open( name, 0, rw | GDBM_FAST, mode, 0 )) == NULL ) {
446 #ifdef HAVE_ST_BLKSIZE
447 if ( dbcachesize > 0 && stat( name, &st ) == 0 ) {
448 dbcachesize = (dbcachesize / st.st_blksize);
449 gdbm_setopt( db, GDBM_CACHESIZE, &dbcachesize, sizeof(int) );
452 dbcachesize = (dbcachesize / 4096);
453 gdbm_setopt( db, GDBM_CACHESIZE, &dbcachesize, sizeof(int) );
462 ldbm_close( LDBM ldbm )
470 ldbm_sync( LDBM ldbm )
478 ldbm_fetch( LDBM ldbm, Datum key )
483 d = gdbm_fetch( ldbm, key );
490 ldbm_store( LDBM ldbm, Datum key, Datum data, int flags )
495 rc = gdbm_store( ldbm, key, data, flags & ~LDBM_SYNC );
496 if ( flags & LDBM_SYNC )
504 ldbm_delete( LDBM ldbm, Datum key )
509 rc = gdbm_delete( ldbm, key );
517 ldbm_firstkey( LDBM ldbm )
522 d = gdbm_firstkey( ldbm );
529 ldbm_nextkey( LDBM ldbm, Datum key )
534 d = gdbm_nextkey( ldbm, key );
541 ldbm_errno( LDBM ldbm )
554 /* MMAPED DBM HASHING DATABASE */
556 #include <ac/alloca.h>
557 #include <ac/string.h>
559 /* #define MDBM_DEBUG */
566 /* #define MDBM_CHAIN */
573 #define mdbm_store mdbm_chain_store
574 #define mdbm_fetch mdbm_chain_fetch
575 #define mdbm_delete mdbm_chain_delete
576 #define mdbm_first mdbm_chain_first
577 #define mdbm_next mdbm_chain_next
581 #define MDBM_PG_SZ (4*1024)
583 /*****************************************************************
587 *****************************************************************/
590 ldbm_open( char *name, int rw, int mode, int dbcachesize )
596 "==>(mdbm)ldbm_open(name=%s,rw=%x,mode=%x,cachesize=%d)\n",
597 name ? name : "NULL", rw, mode, dbcachesize );
601 LDBM_LOCK; /* We need locking here, this is the only non-thread
602 * safe function we have.
605 if ( (db = mdbm_open( name, rw, mode, MDBM_PG_SZ )) == NULL ) {
609 fprintf( stdout, "<==(mdbm)ldbm_open(db=NULL)\n" );
617 (void)mdbm_set_chain(db);
623 fprintf( stdout, "<==(mdbm)ldbm_open(db=%p)\n", db );
629 }/* LDBM ldbm_open() */
635 ldbm_close( LDBM ldbm )
638 /* Open and close are not reentrant so we need to use locks here */
642 "==>(mdbm)ldbm_close(db=%p)\n", ldbm );
651 fprintf( stdout, "<==(mdbm)ldbm_close()\n" );
655 }/* void ldbm_close() */
661 ldbm_sync( LDBM ldbm )
664 /* XXX: Not sure if this is re-entrant need to check code, if so
665 * you can leave LOCKS out.
672 }/* void ldbm_sync() */
675 #define MAX_MDBM_RETRY 5
678 ldbm_fetch( LDBM ldbm, Datum key )
684 /* This hack is needed because MDBM does not take keys
685 * which begin with NULL when working in the chaining
692 k.key.dsize = key.dsize + 1;
693 k.key.dptr = alloca(k.key.dsize);
695 memcpy( (void *)(k.key.dptr + 1), key.dptr, key.dsize );
705 d = mdbm_fetch( ldbm, k );
709 if ( k.val.dptr != NULL ) {
715 if ( (k.val.dptr = malloc( d.dsize )) != NULL ) {
717 k.val.dsize = d.dsize;
718 d = mdbm_fetch( ldbm, k );
727 }/* if ( d.dsize > 0 ) */
729 } while ((d.dsize > k.val.dsize) && (++retry < MAX_MDBM_RETRY));
735 }/* Datum ldbm_fetch() */
741 ldbm_store( LDBM ldbm, Datum key, Datum data, int flags )
744 Datum int_key; /* Internal key */
748 "==>(mdbm)ldbm_store(db=%p, key(dptr=%p,sz=%d), data(dptr=%p,sz=%d), flags=%x)\n",
749 ldbm, key.dptr, key.dsize, data.dptr, data.dsize, flags );
756 int_key.dsize = key.dsize + 1;
757 int_key.dptr = alloca( int_key.dsize );
758 *(int_key.dptr) = 'l'; /* Must not be NULL !*/
759 memcpy( (void *)(int_key.dptr + 1), key.dptr, key.dsize );
764 rc = mdbm_store( ldbm, int_key, data, flags );
765 if ( flags & LDBM_SYNC ) {
772 fprintf( stdout, "<==(mdbm)ldbm_store(rc=%d)\n", rc );
778 }/* int ldbm_store() */
784 ldbm_delete( LDBM ldbm, Datum key )
792 int_key.dsize = key.dsize + 1;
793 int_key.dptr = alloca(int_key.dsize);
794 *(int_key.dptr) = 'l';
795 memcpy( (void *)(int_key.dptr + 1), key.dptr, key.dsize );
800 rc = mdbm_delete( ldbm, int_key );
806 }/* int ldbm_delete() */
812 ldbm_get_next( LDBM ldbm, kvpair (*fptr)(MDBM *, kvpair) )
818 size_t sz = MDBM_PAGE_SIZE(ldbm);
827 in.key.dsize = sz; /* Assume first key in one pg */
828 in.key.dptr = alloca(sz);
830 in.val.dptr = NULL; /* Don't need data just key */
836 out = fptr( ldbm, in );
838 if (out.key.dsize > 0) {
840 ret.dsize = out.key.dsize - delta;
841 if ((ret.dptr = (char *)malloc(ret.dsize)) == NULL) {
848 memcpy(ret.dptr, (void *)(out.key.dptr + delta),
859 }/* static Datum ldbm_get_next() */
865 ldbm_firstkey( LDBM ldbm )
868 return ldbm_get_next( ldbm, mdbm_first );
870 }/* Datum ldbm_firstkey() */
876 ldbm_nextkey( LDBM ldbm, Datum key )
880 * don't know if this will affect the LDAP server opertaion
881 * but mdbm cannot take and input key.
884 return ldbm_get_next( ldbm, mdbm_next );
886 }/* Datum ldbm_nextkey() */
889 ldbm_errno( LDBM ldbm )
891 /* XXX: best we can do with current mdbm interface */
894 }/* int ldbm_errno() */
899 #elif defined( HAVE_NDBM )
901 /*****************************************************************
903 * if no gdbm or mdbm, fall back to using ndbm, the standard unix thing *
905 *****************************************************************/
909 ldbm_open( char *name, int rw, int mode, int dbcachesize )
914 ldbm = dbm_open( name, rw, mode );
921 ldbm_close( LDBM ldbm )
930 ldbm_sync( LDBM ldbm )
936 ldbm_fetch( LDBM ldbm, Datum key )
941 d = ldbm_datum_dup( ldbm, dbm_fetch( ldbm, key ) );
948 ldbm_store( LDBM ldbm, Datum key, Datum data, int flags )
953 rc = dbm_store( ldbm, key, data, flags );
960 ldbm_delete( LDBM ldbm, Datum key )
965 rc = dbm_delete( ldbm, key );
972 ldbm_firstkey( LDBM ldbm )
977 d = dbm_firstkey( ldbm );
984 ldbm_nextkey( LDBM ldbm, Datum key )
989 d = dbm_nextkey( ldbm );
996 ldbm_errno( LDBM ldbm )
1001 err = dbm_error( ldbm );