Home | History | Annotate | Line # | Download | only in back-mdb
init.c revision 1.1.1.1.14.1
      1 /*	$NetBSD: init.c,v 1.1.1.1.14.1 2017/04/21 16:52:29 bouyer Exp $	*/
      2 
      3 /* init.c - initialize mdb backend */
      4 /* $OpenLDAP$ */
      5 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
      6  *
      7  * Copyright 2000-2016 The OpenLDAP Foundation.
      8  * All rights reserved.
      9  *
     10  * Redistribution and use in source and binary forms, with or without
     11  * modification, are permitted only as authorized by the OpenLDAP
     12  * Public License.
     13  *
     14  * A copy of this license is available in the file LICENSE in the
     15  * top-level directory of the distribution or, alternatively, at
     16  * <http://www.OpenLDAP.org/license.html>.
     17  */
     18 
     19 #include <sys/cdefs.h>
     20 __RCSID("$NetBSD: init.c,v 1.1.1.1.14.1 2017/04/21 16:52:29 bouyer Exp $");
     21 
     22 #include "portable.h"
     23 
     24 #include <stdio.h>
     25 #include <ac/string.h>
     26 #include <ac/unistd.h>
     27 #include <ac/stdlib.h>
     28 #include <ac/errno.h>
     29 #include <sys/stat.h>
     30 #include "back-mdb.h"
     31 #include <lutil.h>
     32 #include <ldap_rq.h>
     33 #include "config.h"
     34 
     35 static const struct berval mdmi_databases[] = {
     36 	BER_BVC("ad2i"),
     37 	BER_BVC("dn2i"),
     38 	BER_BVC("id2e"),
     39 	BER_BVNULL
     40 };
     41 
     42 static int
     43 mdb_id_compare( const MDB_val *a, const MDB_val *b )
     44 {
     45 	return *(ID *)a->mv_data < *(ID *)b->mv_data ? -1 : *(ID *)a->mv_data > *(ID *)b->mv_data;
     46 }
     47 
     48 static int
     49 mdb_db_init( BackendDB *be, ConfigReply *cr )
     50 {
     51 	struct mdb_info	*mdb;
     52 	int rc;
     53 
     54 	Debug( LDAP_DEBUG_TRACE,
     55 		LDAP_XSTRING(mdb_db_init) ": Initializing mdb database\n",
     56 		0, 0, 0 );
     57 
     58 	/* allocate backend-database-specific stuff */
     59 	mdb = (struct mdb_info *) ch_calloc( 1, sizeof(struct mdb_info) );
     60 
     61 	/* DBEnv parameters */
     62 	mdb->mi_dbenv_home = ch_strdup( SLAPD_DEFAULT_DB_DIR );
     63 	mdb->mi_dbenv_flags = 0;
     64 	mdb->mi_dbenv_mode = SLAPD_DEFAULT_DB_MODE;
     65 
     66 	mdb->mi_search_stack_depth = DEFAULT_SEARCH_STACK_DEPTH;
     67 	mdb->mi_search_stack = NULL;
     68 
     69 	mdb->mi_mapsize = DEFAULT_MAPSIZE;
     70 	mdb->mi_rtxn_size = DEFAULT_RTXN_SIZE;
     71 
     72 	be->be_private = mdb;
     73 	be->be_cf_ocs = be->bd_info->bi_cf_ocs;
     74 
     75 #ifndef MDB_MULTIPLE_SUFFIXES
     76 	SLAP_DBFLAGS( be ) |= SLAP_DBFLAG_ONE_SUFFIX;
     77 #endif
     78 
     79 	rc = mdb_monitor_db_init( be );
     80 
     81 	return rc;
     82 }
     83 
     84 static int
     85 mdb_db_close( BackendDB *be, ConfigReply *cr );
     86 
     87 static int
     88 mdb_db_open( BackendDB *be, ConfigReply *cr )
     89 {
     90 	int rc, i;
     91 	struct mdb_info *mdb = (struct mdb_info *) be->be_private;
     92 	struct stat stat1;
     93 	uint32_t flags;
     94 	char *dbhome;
     95 	MDB_txn *txn;
     96 
     97 	if ( be->be_suffix == NULL ) {
     98 		Debug( LDAP_DEBUG_ANY,
     99 			LDAP_XSTRING(mdb_db_open) ": need suffix.\n",
    100 			1, 0, 0 );
    101 		return -1;
    102 	}
    103 
    104 	Debug( LDAP_DEBUG_ARGS,
    105 		LDAP_XSTRING(mdb_db_open) ": \"%s\"\n",
    106 		be->be_suffix[0].bv_val, 0, 0 );
    107 
    108 	/* Check existence of dbenv_home. Any error means trouble */
    109 	rc = stat( mdb->mi_dbenv_home, &stat1 );
    110 	if( rc != 0 ) {
    111 		Debug( LDAP_DEBUG_ANY,
    112 			LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    113 			"cannot access database directory \"%s\" (%d).\n",
    114 			be->be_suffix[0].bv_val, mdb->mi_dbenv_home, errno );
    115 		return -1;
    116 	}
    117 
    118 	/* mdb is always clean */
    119 	be->be_flags |= SLAP_DBFLAG_CLEAN;
    120 
    121 	rc = mdb_env_create( &mdb->mi_dbenv );
    122 	if( rc != 0 ) {
    123 		Debug( LDAP_DEBUG_ANY,
    124 			LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    125 			"mdb_env_create failed: %s (%d).\n",
    126 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    127 		goto fail;
    128 	}
    129 
    130 	if ( mdb->mi_readers ) {
    131 		rc = mdb_env_set_maxreaders( mdb->mi_dbenv, mdb->mi_readers );
    132 		if( rc != 0 ) {
    133 			Debug( LDAP_DEBUG_ANY,
    134 				LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    135 				"mdb_env_set_maxreaders failed: %s (%d).\n",
    136 				be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    137 			goto fail;
    138 		}
    139 	}
    140 
    141 	rc = mdb_env_set_mapsize( mdb->mi_dbenv, mdb->mi_mapsize );
    142 	if( rc != 0 ) {
    143 		Debug( LDAP_DEBUG_ANY,
    144 			LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    145 			"mdb_env_set_mapsize failed: %s (%d).\n",
    146 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    147 		goto fail;
    148 	}
    149 
    150 	rc = mdb_env_set_maxdbs( mdb->mi_dbenv, MDB_INDICES );
    151 	if( rc != 0 ) {
    152 		Debug( LDAP_DEBUG_ANY,
    153 			LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    154 			"mdb_env_set_maxdbs failed: %s (%d).\n",
    155 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    156 		goto fail;
    157 	}
    158 
    159 #ifdef HAVE_EBCDIC
    160 	strcpy( path, mdb->mi_dbenv_home );
    161 	__atoe( path );
    162 	dbhome = path;
    163 #else
    164 	dbhome = mdb->mi_dbenv_home;
    165 #endif
    166 
    167 	Debug( LDAP_DEBUG_TRACE,
    168 		LDAP_XSTRING(mdb_db_open) ": database \"%s\": "
    169 		"dbenv_open(%s).\n",
    170 		be->be_suffix[0].bv_val, mdb->mi_dbenv_home, 0);
    171 
    172 	flags = mdb->mi_dbenv_flags;
    173 
    174 	if ( slapMode & SLAP_TOOL_QUICK )
    175 		flags |= MDB_NOSYNC|MDB_WRITEMAP;
    176 
    177 	if ( slapMode & SLAP_TOOL_READONLY)
    178 		flags |= MDB_RDONLY;
    179 
    180 	rc = mdb_env_open( mdb->mi_dbenv, dbhome,
    181 			flags, mdb->mi_dbenv_mode );
    182 
    183 	if ( rc ) {
    184 		Debug( LDAP_DEBUG_ANY,
    185 			LDAP_XSTRING(mdb_db_open) ": database \"%s\" cannot be opened: %s (%d). "
    186 			"Restore from backup!\n",
    187 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    188 		goto fail;
    189 	}
    190 
    191 	rc = mdb_txn_begin( mdb->mi_dbenv, NULL, flags & MDB_RDONLY, &txn );
    192 	if ( rc ) {
    193 		Debug( LDAP_DEBUG_ANY,
    194 			LDAP_XSTRING(mdb_db_open) ": database \"%s\" cannot be opened: %s (%d). "
    195 			"Restore from backup!\n",
    196 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    197 		goto fail;
    198 	}
    199 
    200 	/* open (and create) main databases */
    201 	for( i = 0; mdmi_databases[i].bv_val; i++ ) {
    202 		flags = MDB_INTEGERKEY;
    203 		if( i == MDB_ID2ENTRY ) {
    204 			if ( !(slapMode & (SLAP_TOOL_READMAIN|SLAP_TOOL_READONLY) ))
    205 				flags |= MDB_CREATE;
    206 		} else {
    207 			if ( i == MDB_DN2ID )
    208 				flags |= MDB_DUPSORT;
    209 			if ( !(slapMode & SLAP_TOOL_READONLY) )
    210 				flags |= MDB_CREATE;
    211 		}
    212 
    213 		rc = mdb_dbi_open( txn,
    214 			mdmi_databases[i].bv_val,
    215 			flags,
    216 			&mdb->mi_dbis[i] );
    217 
    218 		if ( rc != 0 ) {
    219 			snprintf( cr->msg, sizeof(cr->msg), "database \"%s\": "
    220 				"mdb_dbi_open(%s/%s) failed: %s (%d).",
    221 				be->be_suffix[0].bv_val,
    222 				mdb->mi_dbenv_home, mdmi_databases[i].bv_val,
    223 				mdb_strerror(rc), rc );
    224 			Debug( LDAP_DEBUG_ANY,
    225 				LDAP_XSTRING(mdb_db_open) ": %s\n",
    226 				cr->msg, 0, 0 );
    227 			goto fail;
    228 		}
    229 
    230 		if ( i == MDB_ID2ENTRY )
    231 			mdb_set_compare( txn, mdb->mi_dbis[i], mdb_id_compare );
    232 		else if ( i == MDB_DN2ID ) {
    233 			MDB_cursor *mc;
    234 			MDB_val key, data;
    235 			ID id;
    236 			mdb_set_dupsort( txn, mdb->mi_dbis[i], mdb_dup_compare );
    237 			/* check for old dn2id format */
    238 			rc = mdb_cursor_open( txn, mdb->mi_dbis[i], &mc );
    239 			/* first record is always ID 0 */
    240 			rc = mdb_cursor_get( mc, &key, &data, MDB_FIRST );
    241 			if ( rc == 0 ) {
    242 				rc = mdb_cursor_get( mc, &key, &data, MDB_NEXT );
    243 				if ( rc == 0 ) {
    244 					int len;
    245 					unsigned char *ptr;
    246 					ptr = data.mv_data;
    247 					len = (ptr[0] & 0x7f) << 8 | ptr[1];
    248 					if (data.mv_size < 2*len + 4 + 2*sizeof(ID)) {
    249 						snprintf( cr->msg, sizeof(cr->msg),
    250 						"database \"%s\": DN index needs upgrade, "
    251 						"run \"slapindex entryDN\".",
    252 						be->be_suffix[0].bv_val );
    253 						Debug( LDAP_DEBUG_ANY,
    254 							LDAP_XSTRING(mdb_db_open) ": %s\n",
    255 							cr->msg, 0, 0 );
    256 						if ( !(slapMode & SLAP_TOOL_READMAIN ))
    257 							rc = LDAP_OTHER;
    258 						mdb->mi_flags |= MDB_NEED_UPGRADE;
    259 					}
    260 				}
    261 			}
    262 			mdb_cursor_close( mc );
    263 			if ( rc == LDAP_OTHER )
    264 				goto fail;
    265 		}
    266 	}
    267 
    268 	rc = mdb_ad_read( mdb, txn );
    269 	if ( rc ) {
    270 		mdb_txn_abort( txn );
    271 		goto fail;
    272 	}
    273 
    274 	/* slapcat doesn't need indexes. avoid a failure if
    275 	 * a configured index wasn't created yet.
    276 	 */
    277 	if ( !(slapMode & SLAP_TOOL_READONLY) ) {
    278 		rc = mdb_attr_dbs_open( be, txn, cr );
    279 		if ( rc ) {
    280 			mdb_txn_abort( txn );
    281 			goto fail;
    282 		}
    283 	}
    284 
    285 	rc = mdb_txn_commit(txn);
    286 	if ( rc != 0 ) {
    287 		Debug( LDAP_DEBUG_ANY,
    288 			LDAP_XSTRING(mdb_db_open) ": database %s: "
    289 			"txn_commit failed: %s (%d)\n",
    290 			be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    291 		goto fail;
    292 	}
    293 
    294 	/* monitor setup */
    295 	rc = mdb_monitor_db_open( be );
    296 	if ( rc != 0 ) {
    297 		goto fail;
    298 	}
    299 
    300 	mdb->mi_flags |= MDB_IS_OPEN;
    301 
    302 	return 0;
    303 
    304 fail:
    305 	mdb_db_close( be, NULL );
    306 	return rc;
    307 }
    308 
    309 static int
    310 mdb_db_close( BackendDB *be, ConfigReply *cr )
    311 {
    312 	int rc;
    313 	struct mdb_info *mdb = (struct mdb_info *) be->be_private;
    314 
    315 	/* monitor handling */
    316 	(void)mdb_monitor_db_close( be );
    317 
    318 	mdb->mi_flags &= ~MDB_IS_OPEN;
    319 
    320 	if( mdb->mi_dbenv ) {
    321 		mdb_reader_flush( mdb->mi_dbenv );
    322 	}
    323 
    324 	if ( mdb->mi_dbenv ) {
    325 		if ( mdb->mi_dbis[0] ) {
    326 			int i;
    327 
    328 			mdb_attr_dbs_close( mdb );
    329 			for ( i=0; i<MDB_NDB; i++ )
    330 				mdb_dbi_close( mdb->mi_dbenv, mdb->mi_dbis[i] );
    331 
    332 			/* force a sync, but not if we were ReadOnly,
    333 			 * and not in Quick mode.
    334 			 */
    335 			if (!(slapMode & (SLAP_TOOL_QUICK|SLAP_TOOL_READONLY))) {
    336 				rc = mdb_env_sync( mdb->mi_dbenv, 1 );
    337 				if( rc != 0 ) {
    338 					Debug( LDAP_DEBUG_ANY,
    339 						"mdb_db_close: database \"%s\": "
    340 						"mdb_env_sync failed: %s (%d).\n",
    341 						be->be_suffix[0].bv_val, mdb_strerror(rc), rc );
    342 				}
    343 			}
    344 		}
    345 
    346 		mdb_env_close( mdb->mi_dbenv );
    347 		mdb->mi_dbenv = NULL;
    348 	}
    349 
    350 	return 0;
    351 }
    352 
    353 static int
    354 mdb_db_destroy( BackendDB *be, ConfigReply *cr )
    355 {
    356 	struct mdb_info *mdb = (struct mdb_info *) be->be_private;
    357 
    358 	/* stop and remove checkpoint task */
    359 	if ( mdb->mi_txn_cp_task ) {
    360 		struct re_s *re = mdb->mi_txn_cp_task;
    361 		mdb->mi_txn_cp_task = NULL;
    362 		ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
    363 		if ( ldap_pvt_runqueue_isrunning( &slapd_rq, re ) )
    364 			ldap_pvt_runqueue_stoptask( &slapd_rq, re );
    365 		ldap_pvt_runqueue_remove( &slapd_rq, re );
    366 		ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
    367 	}
    368 
    369 	/* monitor handling */
    370 	(void)mdb_monitor_db_destroy( be );
    371 
    372 	if( mdb->mi_dbenv_home ) ch_free( mdb->mi_dbenv_home );
    373 
    374 	mdb_attr_index_destroy( mdb );
    375 
    376 	ch_free( mdb );
    377 	be->be_private = NULL;
    378 
    379 	return 0;
    380 }
    381 
    382 int
    383 mdb_back_initialize(
    384 	BackendInfo	*bi )
    385 {
    386 	int rc;
    387 
    388 	static char *controls[] = {
    389 		LDAP_CONTROL_ASSERT,
    390 		LDAP_CONTROL_MANAGEDSAIT,
    391 		LDAP_CONTROL_NOOP,
    392 		LDAP_CONTROL_PAGEDRESULTS,
    393 		LDAP_CONTROL_PRE_READ,
    394 		LDAP_CONTROL_POST_READ,
    395 		LDAP_CONTROL_SUBENTRIES,
    396 		LDAP_CONTROL_X_PERMISSIVE_MODIFY,
    397 #ifdef LDAP_X_TXN
    398 		LDAP_CONTROL_X_TXN_SPEC,
    399 #endif
    400 		NULL
    401 	};
    402 
    403 	/* initialize the underlying database system */
    404 	Debug( LDAP_DEBUG_TRACE,
    405 		LDAP_XSTRING(mdb_back_initialize) ": initialize "
    406 		MDB_UCTYPE " backend\n", 0, 0, 0 );
    407 
    408 	bi->bi_flags |=
    409 		SLAP_BFLAG_INCREMENT |
    410 		SLAP_BFLAG_SUBENTRIES |
    411 		SLAP_BFLAG_ALIASES |
    412 		SLAP_BFLAG_REFERRALS;
    413 
    414 	bi->bi_controls = controls;
    415 
    416 	{	/* version check */
    417 		int major, minor, patch, ver;
    418 		char *version = mdb_version( &major, &minor, &patch );
    419 #ifdef HAVE_EBCDIC
    420 		char v2[1024];
    421 
    422 		/* All our stdio does an ASCII to EBCDIC conversion on
    423 		 * the output. Strings from the MDB library are already
    424 		 * in EBCDIC; we have to go back and forth...
    425 		 */
    426 		strcpy( v2, version );
    427 		__etoa( v2 );
    428 		version = v2;
    429 #endif
    430 		ver = (major << 24) | (minor << 16) | patch;
    431 		if( ver != MDB_VERSION_FULL ) {
    432 			/* fail if a versions don't match */
    433 			Debug( LDAP_DEBUG_ANY,
    434 				LDAP_XSTRING(mdb_back_initialize) ": "
    435 				"MDB library version mismatch:"
    436 				" expected " MDB_VERSION_STRING ","
    437 				" got %s\n", version, 0, 0 );
    438 			return -1;
    439 		}
    440 
    441 		Debug( LDAP_DEBUG_TRACE, LDAP_XSTRING(mdb_back_initialize)
    442 			": %s\n", version, 0, 0 );
    443 	}
    444 
    445 	bi->bi_open = 0;
    446 	bi->bi_close = 0;
    447 	bi->bi_config = 0;
    448 	bi->bi_destroy = 0;
    449 
    450 	bi->bi_db_init = mdb_db_init;
    451 	bi->bi_db_config = config_generic_wrapper;
    452 	bi->bi_db_open = mdb_db_open;
    453 	bi->bi_db_close = mdb_db_close;
    454 	bi->bi_db_destroy = mdb_db_destroy;
    455 
    456 	bi->bi_op_add = mdb_add;
    457 	bi->bi_op_bind = mdb_bind;
    458 	bi->bi_op_compare = mdb_compare;
    459 	bi->bi_op_delete = mdb_delete;
    460 	bi->bi_op_modify = mdb_modify;
    461 	bi->bi_op_modrdn = mdb_modrdn;
    462 	bi->bi_op_search = mdb_search;
    463 
    464 	bi->bi_op_unbind = 0;
    465 
    466 	bi->bi_extended = mdb_extended;
    467 
    468 	bi->bi_chk_referrals = 0;
    469 	bi->bi_operational = mdb_operational;
    470 
    471 	bi->bi_has_subordinates = mdb_hasSubordinates;
    472 	bi->bi_entry_release_rw = mdb_entry_release;
    473 	bi->bi_entry_get_rw = mdb_entry_get;
    474 
    475 	/*
    476 	 * hooks for slap tools
    477 	 */
    478 	bi->bi_tool_entry_open = mdb_tool_entry_open;
    479 	bi->bi_tool_entry_close = mdb_tool_entry_close;
    480 	bi->bi_tool_entry_first = backend_tool_entry_first;
    481 	bi->bi_tool_entry_first_x = mdb_tool_entry_first_x;
    482 	bi->bi_tool_entry_next = mdb_tool_entry_next;
    483 	bi->bi_tool_entry_get = mdb_tool_entry_get;
    484 	bi->bi_tool_entry_put = mdb_tool_entry_put;
    485 	bi->bi_tool_entry_reindex = mdb_tool_entry_reindex;
    486 	bi->bi_tool_sync = 0;
    487 	bi->bi_tool_dn2id_get = mdb_tool_dn2id_get;
    488 	bi->bi_tool_entry_modify = mdb_tool_entry_modify;
    489 
    490 	bi->bi_connection_init = 0;
    491 	bi->bi_connection_destroy = 0;
    492 
    493 	rc = mdb_back_init_cf( bi );
    494 
    495 	return rc;
    496 }
    497 
    498 #if	(SLAPD_MDB == SLAPD_MOD_DYNAMIC)
    499 
    500 SLAP_BACKEND_INIT_MODULE( mdb )
    501 
    502 #endif /* SLAPD_MDB == SLAPD_MOD_DYNAMIC */
    503 
    504