#include "db_config.h"
#include "db_int.h"
#include "dbinc/crypto.h"
#include "dbinc/db_page.h"
#include "dbinc/btree.h"
#include "dbinc/lock.h"
#include "dbinc/log.h"
#include "dbinc/mp.h"
#include "dbinc/txn.h"
static int __env_refresh __P((DB_ENV *, u_int32_t, int));
static int __file_handle_cleanup __P((DB_ENV *));
char *
db_version(majverp, minverp, patchp)
int *majverp, *minverp, *patchp;
{
if (majverp != NULL)
*majverp = DB_VERSION_MAJOR;
if (minverp != NULL)
*minverp = DB_VERSION_MINOR;
if (patchp != NULL)
*patchp = DB_VERSION_PATCH;
return ((char *)DB_VERSION_STRING);
}
int
__env_open_pp(dbenv, db_home, flags, mode)
DB_ENV *dbenv;
const char *db_home;
u_int32_t flags;
int mode;
{
int ret;
#undef OKFLAGS
#define OKFLAGS \
(DB_CREATE | DB_INIT_CDB | DB_INIT_LOCK | DB_INIT_LOG | \
DB_INIT_MPOOL | DB_INIT_REP | DB_INIT_TXN | DB_LOCKDOWN | \
DB_PRIVATE | DB_RECOVER | DB_RECOVER_FATAL | DB_REGISTER | \
DB_SYSTEM_MEM | DB_THREAD | DB_USE_ENVIRON | DB_USE_ENVIRON_ROOT)
#undef OKFLAGS_CDB
#define OKFLAGS_CDB \
(DB_CREATE | DB_INIT_CDB | DB_INIT_MPOOL | DB_LOCKDOWN | \
DB_PRIVATE | DB_SYSTEM_MEM | DB_THREAD | \
DB_USE_ENVIRON | DB_USE_ENVIRON_ROOT)
if ((ret = __db_fchk(dbenv, "DB_ENV->open", flags, OKFLAGS)) != 0)
return (ret);
if ((ret = __db_fcchk(
dbenv, "DB_ENV->open", flags, DB_INIT_CDB, ~OKFLAGS_CDB)) != 0)
return (ret);
if (LF_ISSET(DB_REGISTER)) {
if (!__os_support_db_register()) {
__db_errx(dbenv,
"Berkeley DB library does not support DB_REGISTER on this system");
return (EINVAL);
}
if ((ret = __db_fcchk(dbenv, "DB_ENV->open", flags,
DB_PRIVATE, DB_REGISTER | DB_SYSTEM_MEM)) != 0)
return (ret);
if (!LF_ISSET(DB_INIT_TXN)) {
__db_errx(
dbenv, "registration requires transaction support");
return (EINVAL);
}
}
if (LF_ISSET(DB_INIT_REP)) {
if (!__os_support_replication()) {
__db_errx(dbenv,
"Berkeley DB library does not support replication on this system");
return (EINVAL);
}
if (!LF_ISSET(DB_INIT_LOCK)) {
__db_errx(dbenv,
"replication requires locking support");
return (EINVAL);
}
if (!LF_ISSET(DB_INIT_TXN)) {
__db_errx(
dbenv, "replication requires transaction support");
return (EINVAL);
}
}
if (LF_ISSET(DB_RECOVER | DB_RECOVER_FATAL)) {
if ((ret = __db_fcchk(dbenv,
"DB_ENV->open", flags, DB_RECOVER, DB_RECOVER_FATAL)) != 0)
return (ret);
if ((ret = __db_fcchk(dbenv,
"DB_ENV->open", flags, DB_REGISTER, DB_RECOVER_FATAL)) != 0)
return (ret);
if (!LF_ISSET(DB_CREATE)) {
__db_errx(dbenv, "recovery requires the create flag");
return (EINVAL);
}
if (!LF_ISSET(DB_INIT_TXN)) {
__db_errx(
dbenv, "recovery requires transaction support");
return (EINVAL);
}
}
#ifdef HAVE_MUTEX_THREAD_ONLY
if (!LF_ISSET(DB_PRIVATE)) {
__db_errx(dbenv,
"Berkeley DB library configured to support only private environments");
return (EINVAL);
}
#endif
#ifdef HAVE_MUTEX_FCNTL
if (F_ISSET(dbenv, DB_ENV_THREAD)) {
__db_errx(dbenv,
"architecture lacks fast mutexes: applications cannot be threaded");
return (EINVAL);
}
#endif
return (__env_open(dbenv, db_home, flags, mode));
}
int
__env_open(dbenv, db_home, flags, mode)
DB_ENV *dbenv;
const char *db_home;
u_int32_t flags;
int mode;
{
DB_THREAD_INFO *ip;
REGINFO *infop;
u_int32_t init_flags, orig_flags;
int create_ok, register_recovery, rep_check, ret, t_ret;
ip = NULL;
register_recovery = rep_check = 0;
if ((ret = __env_config(dbenv, db_home, flags, mode)) != 0)
return (ret);
orig_flags = dbenv->flags;
if (LF_ISSET(DB_REGISTER)) {
if ((ret = __envreg_register(dbenv, ®ister_recovery)) != 0)
goto err;
if (register_recovery) {
if (!LF_ISSET(DB_RECOVER)) {
__db_errx(dbenv,
"The DB_RECOVER flag was not specified, and recovery is needed");
ret = DB_RUNRECOVERY;
goto err;
}
} else
LF_CLR(DB_RECOVER);
}
if (LF_ISSET(DB_RECOVER | DB_RECOVER_FATAL))
#ifdef HAVE_REPLICATION
if ((ret = __rep_reset_init(dbenv)) != 0 ||
(ret = __env_remove_env(dbenv)) != 0 ||
#else
if ((ret = __env_remove_env(dbenv)) != 0 ||
#endif
(ret = __env_refresh(dbenv, orig_flags, 0)) != 0)
goto err;
create_ok = LF_ISSET(DB_CREATE) ? 1 : 0;
if (LF_ISSET(DB_LOCKDOWN))
F_SET(dbenv, DB_ENV_LOCKDOWN);
if (LF_ISSET(DB_PRIVATE))
F_SET(dbenv, DB_ENV_PRIVATE);
if (LF_ISSET(DB_RECOVER_FATAL))
F_SET(dbenv, DB_ENV_RECOVER_FATAL);
if (LF_ISSET(DB_SYSTEM_MEM))
F_SET(dbenv, DB_ENV_SYSTEM_MEM);
if (LF_ISSET(DB_THREAD))
F_SET(dbenv, DB_ENV_THREAD);
#define DB_INITENV_CDB 0x0001
#define DB_INITENV_CDB_ALLDB 0x0002
#define DB_INITENV_LOCK 0x0004
#define DB_INITENV_LOG 0x0008
#define DB_INITENV_MPOOL 0x0010
#define DB_INITENV_REP 0x0020
#define DB_INITENV_TXN 0x0040
init_flags = 0;
if (LF_ISSET(DB_INIT_CDB))
FLD_SET(init_flags, DB_INITENV_CDB);
if (F_ISSET(dbenv, DB_ENV_CDB_ALLDB))
FLD_SET(init_flags, DB_INITENV_CDB_ALLDB);
if (LF_ISSET(DB_INIT_LOCK))
FLD_SET(init_flags, DB_INITENV_LOCK);
if (LF_ISSET(DB_INIT_LOG))
FLD_SET(init_flags, DB_INITENV_LOG);
if (LF_ISSET(DB_INIT_MPOOL))
FLD_SET(init_flags, DB_INITENV_MPOOL);
if (LF_ISSET(DB_INIT_REP))
FLD_SET(init_flags, DB_INITENV_REP);
if (LF_ISSET(DB_INIT_TXN))
FLD_SET(init_flags, DB_INITENV_TXN);
if ((ret = __env_attach(dbenv, &init_flags, create_ok, 1)) != 0)
goto err;
if (FLD_ISSET(init_flags, DB_INITENV_CDB))
LF_SET(DB_INIT_CDB);
if (FLD_ISSET(init_flags, DB_INITENV_LOCK))
LF_SET(DB_INIT_LOCK);
if (FLD_ISSET(init_flags, DB_INITENV_LOG))
LF_SET(DB_INIT_LOG);
if (FLD_ISSET(init_flags, DB_INITENV_MPOOL))
LF_SET(DB_INIT_MPOOL);
if (FLD_ISSET(init_flags, DB_INITENV_REP))
LF_SET(DB_INIT_REP);
if (FLD_ISSET(init_flags, DB_INITENV_TXN))
LF_SET(DB_INIT_TXN);
if (FLD_ISSET(init_flags, DB_INITENV_CDB_ALLDB) &&
(ret = __env_set_flags(dbenv, DB_CDB_ALLDB, 1)) != 0)
goto err;
dbenv->open_flags = flags;
if (LF_ISSET(DB_INIT_CDB)) {
LF_SET(DB_INIT_LOCK);
F_SET(dbenv, DB_ENV_CDB);
}
F_SET(dbenv, DB_ENV_OPEN_CALLED);
infop = dbenv->reginfo;
if ((ret = __env_thread_init(
dbenv, F_ISSET(infop, REGION_CREATE) ? 1 : 0)) != 0)
goto err;
ENV_ENTER(dbenv, ip);
#ifdef HAVE_MUTEX_SUPPORT
if ((ret = __mutex_open(dbenv, create_ok)) != 0)
goto err;
#endif
if ((ret = __env_ref_increment(dbenv)) != 0)
goto err;
if ((ret = __mutex_alloc(dbenv,
MTX_ENV_HANDLE, DB_MUTEX_PROCESS_ONLY, &dbenv->mtx_env)) != 0)
goto err;
if (LF_ISSET(DB_INIT_REP) && (ret = __rep_open(dbenv)) != 0)
goto err;
rep_check = IS_ENV_REPLICATED(dbenv) ? 1 : 0;
if (rep_check && (ret = __env_rep_enter(dbenv, 0)) != 0)
goto err;
if (LF_ISSET(DB_INIT_MPOOL)) {
if ((ret = __memp_open(dbenv, create_ok)) != 0)
goto err;
if (create_ok)
(void)__memp_set_config(
dbenv, DB_MEMP_SUPPRESS_WRITE, 1);
TAILQ_INIT(&dbenv->dblist);
if ((ret = __mutex_alloc(dbenv, MTX_ENV_DBLIST,
DB_MUTEX_PROCESS_ONLY, &dbenv->mtx_dblist)) != 0)
goto err;
if ((ret = __memp_register(
dbenv, DB_FTYPE_SET, __db_pgin, __db_pgout)) != 0)
goto err;
}
if (LF_ISSET(DB_INIT_MPOOL | DB_INIT_LOG | DB_INIT_TXN) &&
(ret = __crypto_region_init(dbenv)) != 0)
goto err;
if ((ret = __mutex_alloc(dbenv, MTX_TWISTER,
DB_MUTEX_PROCESS_ONLY, &dbenv->mtx_mt)) != 0)
goto err;
if (LF_ISSET(DB_INIT_LOG | DB_INIT_TXN))
if ((ret = __log_open(dbenv, create_ok)) != 0)
goto err;
if (LF_ISSET(DB_INIT_LOCK))
if ((ret = __lock_open(dbenv, create_ok)) != 0)
goto err;
if (LF_ISSET(DB_INIT_TXN)) {
if ((ret = __txn_open(dbenv, create_ok)) != 0)
goto err;
if ((ret = __env_init_rec(dbenv, DB_LOGVERSION)) != 0)
goto err;
}
if (LF_ISSET(DB_RECOVER | DB_RECOVER_FATAL) &&
(ret = __db_apprec(dbenv, NULL, NULL, 1,
LF_ISSET(DB_RECOVER | DB_RECOVER_FATAL))) != 0)
goto err;
if (TXN_ON(dbenv) &&
!F_ISSET(dbenv, DB_ENV_LOG_INMEMORY) &&
F_ISSET(infop, REGION_CREATE) &&
!LF_ISSET(DB_RECOVER | DB_RECOVER_FATAL) &&
(ret = __txn_reset(dbenv)) != 0)
goto err;
if ((ret = __env_turn_on(dbenv)) != 0)
goto err;
if (rep_check)
ret = __env_db_rep_exit(dbenv);
if (LF_ISSET(DB_INIT_MPOOL))
(void)__memp_set_config(dbenv, DB_MEMP_SUPPRESS_WRITE, 0);
err: if (ret == 0)
ENV_LEAVE(dbenv, ip);
else {
infop = dbenv->reginfo;
if (infop != NULL && F_ISSET(infop, REGION_CREATE)) {
ret = __db_panic(dbenv, ret);
(void)__env_refresh(dbenv, orig_flags, rep_check);
(void)__env_remove_env(dbenv);
(void)__env_refresh(dbenv, orig_flags, 0);
} else
(void)__env_refresh(dbenv, orig_flags, rep_check);
}
if (register_recovery) {
if (ret == 0 && (t_ret = __envreg_xunlock(dbenv)) != 0)
ret = t_ret;
if (ret != 0)
(void)__envreg_unregister(dbenv, 1);
}
return (ret);
}
int
__env_remove(dbenv, db_home, flags)
DB_ENV *dbenv;
const char *db_home;
u_int32_t flags;
{
int ret, t_ret;
#undef OKFLAGS
#define OKFLAGS \
(DB_FORCE | DB_USE_ENVIRON | DB_USE_ENVIRON_ROOT)
if ((ret = __db_fchk(dbenv, "DB_ENV->remove", flags, OKFLAGS)) != 0)
return (ret);
ENV_ILLEGAL_AFTER_OPEN(dbenv, "DB_ENV->remove");
if ((ret = __env_config(dbenv, db_home, flags, 0)) != 0)
return (ret);
if ((ret = __env_turn_off(dbenv, flags)) != 0 && !LF_ISSET(DB_FORCE))
return (ret);
ret = __env_remove_env(dbenv);
if ((t_ret = __env_close(dbenv, 0)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__env_config(dbenv, db_home, flags, mode)
DB_ENV *dbenv;
const char *db_home;
u_int32_t flags;
int mode;
{
int ret;
char *home, home_buf[DB_MAXPATHLEN];
home = (char *)db_home;
if (home == NULL && (LF_ISSET(DB_USE_ENVIRON) ||
(LF_ISSET(DB_USE_ENVIRON_ROOT) && __os_isroot()))) {
home = home_buf;
if ((ret = __os_getenv(
dbenv, "DB_HOME", &home, sizeof(home_buf))) != 0)
return (ret);
}
if (home != NULL &&
(ret = __os_strdup(dbenv, home, &dbenv->db_home)) != 0)
return (ret);
dbenv->db_mode = mode == 0 ? __db_omode("rw-rw----") : mode;
if ((ret = __env_read_db_config(dbenv)) != 0)
return (ret);
if (dbenv->db_tmp_dir == NULL && (ret = __os_tmpdir(dbenv, flags)) != 0)
return (ret);
return (0);
}
int
__env_close_pp(dbenv, flags)
DB_ENV *dbenv;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
int rep_check, ret, t_ret;
ret = 0;
if (PANIC_ISSET(dbenv)) {
(void)__file_handle_cleanup(dbenv);
if (IS_ENV_REPLICATED(dbenv))
(void)__repmgr_close(dbenv);
PANIC_CHECK(dbenv);
}
ENV_ENTER(dbenv, ip);
if (flags != 0 &&
(t_ret = __db_ferr(dbenv, "DB_ENV->close", 0)) != 0 && ret == 0)
ret = t_ret;
rep_check = IS_ENV_REPLICATED(dbenv) ? 1 : 0;
if (rep_check) {
#ifdef HAVE_REPLICATION_THREADS
if ((t_ret = __repmgr_close(dbenv)) != 0 && ret == 0)
ret = t_ret;
#endif
if ((t_ret = __env_rep_enter(dbenv, 0)) != 0 && ret == 0)
ret = t_ret;
}
if ((t_ret = __env_close(dbenv, rep_check)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
int
__env_close(dbenv, rep_check)
DB_ENV *dbenv;
int rep_check;
{
int ret, t_ret;
char **p;
ret = 0;
if (TXN_ON(dbenv) && (t_ret = __txn_preclose(dbenv)) != 0 && ret == 0)
ret = t_ret;
#ifdef HAVE_REPLICATION
if ((t_ret = __rep_env_close(dbenv)) != 0 && ret == 0)
ret = t_ret;
#endif
if ((t_ret = __env_refresh(dbenv, 0, rep_check)) != 0 && ret == 0)
ret = t_ret;
#ifdef HAVE_CRYPTO
if ((t_ret = __crypto_env_close(dbenv)) != 0 && ret == 0)
ret = t_ret;
#endif
if (dbenv->registry != NULL) {
(void)__envreg_unregister(dbenv, 0);
dbenv->registry = NULL;
}
if ((t_ret = __file_handle_cleanup(dbenv)) != 0 && ret == 0)
ret = t_ret;
if (dbenv->db_log_dir != NULL)
__os_free(dbenv, dbenv->db_log_dir);
dbenv->db_log_dir = NULL;
if (dbenv->db_tmp_dir != NULL)
__os_free(dbenv, dbenv->db_tmp_dir);
dbenv->db_tmp_dir = NULL;
if (dbenv->db_data_dir != NULL) {
for (p = dbenv->db_data_dir; *p != NULL; ++p)
__os_free(dbenv, *p);
__os_free(dbenv, dbenv->db_data_dir);
dbenv->db_data_dir = NULL;
dbenv->data_next = 0;
}
if (dbenv->db_home != NULL) {
__os_free(dbenv, dbenv->db_home);
dbenv->db_home = NULL;
}
__db_env_destroy(dbenv);
return (ret);
}
static int
__env_refresh(dbenv, orig_flags, rep_check)
DB_ENV *dbenv;
u_int32_t orig_flags;
int rep_check;
{
DB *ldbp;
DB_THREAD_INFO *ip;
int ret, t_ret;
ret = 0;
if (TXN_ON(dbenv) &&
(t_ret = __txn_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
if (LOGGING_ON(dbenv) &&
(t_ret = __log_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
if (LOCKING_ON(dbenv)) {
if (!F_ISSET(dbenv, DB_ENV_THREAD) &&
dbenv->env_lref != NULL && (t_ret = __lock_id_free(dbenv,
(DB_LOCKER *)dbenv->env_lref)) != 0 && ret == 0)
ret = t_ret;
dbenv->env_lref = NULL;
if ((t_ret = __lock_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
}
if ((t_ret = __mutex_free(dbenv, &dbenv->mtx_env)) != 0 && ret == 0)
ret = t_ret;
if (dbenv->db_ref != 0) {
__db_errx(dbenv,
"Database handles still open at environment close");
TAILQ_FOREACH(ldbp, &dbenv->dblist, dblistlinks)
__db_errx(dbenv, "Open database handle: %s%s%s",
ldbp->fname == NULL ? "unnamed" : ldbp->fname,
ldbp->dname == NULL ? "" : "/",
ldbp->dname == NULL ? "" : ldbp->dname);
if (ret == 0)
ret = EINVAL;
}
TAILQ_INIT(&dbenv->dblist);
if ((t_ret = __mutex_free(dbenv, &dbenv->mtx_dblist)) != 0 && ret == 0)
ret = t_ret;
if ((t_ret = __mutex_free(dbenv, &dbenv->mtx_mt)) != 0 && ret == 0)
ret = t_ret;
if (dbenv->mt != NULL) {
__os_free(dbenv, dbenv->mt);
dbenv->mt = NULL;
}
if (MPOOL_ON(dbenv)) {
if (F_ISSET(dbenv, DB_ENV_PRIVATE) &&
(t_ret = __memp_sync_int(dbenv, NULL, 0,
DB_SYNC_CACHE | DB_SYNC_SUPPRESS_WRITE, NULL, NULL)) != 0 &&
ret == 0)
ret = t_ret;
if ((t_ret = __memp_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
}
if (rep_check && (t_ret = __env_db_rep_exit(dbenv)) != 0 && ret == 0)
ret = t_ret;
if (REP_ON(dbenv) &&
(t_ret = __rep_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
#ifdef HAVE_CRYPTO
if ((t_ret = __crypto_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
#endif
if (dbenv->thr_hashtab != NULL &&
(t_ret = __env_set_state(dbenv, &ip, THREAD_OUT)) != 0 && ret == 0)
ret = t_ret;
if ((t_ret = __env_ref_decrement(dbenv)) != 0 && ret == 0)
ret = t_ret;
#ifdef HAVE_MUTEX_SUPPORT
if (MUTEX_ON(dbenv) &&
(t_ret = __mutex_env_refresh(dbenv)) != 0 && ret == 0)
ret = t_ret;
#endif
if (dbenv->reginfo != NULL && (t_ret = __env_detach(
dbenv, F_ISSET(dbenv, DB_ENV_PRIVATE) ? 1 : 0)) != 0 && ret == 0) {
ret = t_ret;
}
if (dbenv->mutex_iq != NULL) {
__os_free(dbenv, dbenv->mutex_iq);
dbenv->mutex_iq = NULL;
}
if (dbenv->recover_dtab != NULL) {
__os_free(dbenv, dbenv->recover_dtab);
dbenv->recover_dtab = NULL;
dbenv->recover_dtab_size = 0;
}
dbenv->flags = orig_flags;
return (ret);
}
static int
__file_handle_cleanup(dbenv)
DB_ENV *dbenv;
{
DB_FH *fhp;
if (TAILQ_FIRST(&dbenv->fdlist) == NULL)
return (0);
__db_errx(dbenv, "File handles still open at environment close");
while ((fhp = TAILQ_FIRST(&dbenv->fdlist)) != NULL) {
__db_errx(dbenv, "Open file handle: %s", fhp->name);
(void)__os_closehandle(dbenv, fhp);
}
return (EINVAL);
}
int
__env_get_open_flags(dbenv, flagsp)
DB_ENV *dbenv;
u_int32_t *flagsp;
{
ENV_ILLEGAL_BEFORE_OPEN(dbenv, "DB_ENV->get_open_flags");
*flagsp = dbenv->open_flags;
return (0);
}