#include "db_config.h"
#include "db_int.h"
#include "dbinc/db_page.h"
#include "dbinc/btree.h"
#include "dbinc/hash.h"
#ifndef HAVE_QUEUE
#include "dbinc/qam.h"
#endif
#include "dbinc/lock.h"
#include "dbinc/log.h"
#include "dbinc/mp.h"
#include "dbinc/txn.h"
static int __db_associate_arg __P((DB *, DB *,
int (*)(DB *, const DBT *, const DBT *, DBT *), u_int32_t));
static int __dbc_del_arg __P((DBC *, u_int32_t));
static int __dbc_get_arg __P((DBC *, DBT *, DBT *, u_int32_t));
static int __dbc_pget_arg __P((DBC *, DBT *, u_int32_t));
static int __dbc_put_arg __P((DBC *, DBT *, DBT *, u_int32_t));
static int __db_curinval __P((const ENV *));
static int __db_cursor_arg __P((DB *, u_int32_t));
static int __db_del_arg __P((DB *, DBT *, u_int32_t));
static int __db_get_arg __P((const DB *, DBT *, DBT *, u_int32_t));
static int __db_join_arg __P((DB *, DBC **, u_int32_t));
static int __db_open_arg __P((DB *,
DB_TXN *, const char *, const char *, DBTYPE, u_int32_t));
static int __db_pget_arg __P((DB *, DBT *, u_int32_t));
static int __db_put_arg __P((DB *, DBT *, DBT *, u_int32_t));
static int __dbt_ferr __P((const DB *, const char *, const DBT *, int));
static int __db_associate_foreign_arg __P((DB *, DB *,
int (*)(DB *, const DBT *, DBT *, const DBT *, int *),
u_int32_t));
int
__db_associate_pp(dbp, txn, sdbp, callback, flags)
DB *dbp, *sdbp;
DB_TXN *txn;
int (*callback) __P((DB *, const DBT *, const DBT *, DBT *));
u_int32_t flags;
{
DBC *sdbc;
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret, txn_local;
env = dbp->env;
txn_local = 0;
STRIP_AUTO_COMMIT(flags);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if (TAILQ_FIRST(&sdbp->active_queue) != NULL ||
TAILQ_FIRST(&sdbp->join_queue) != NULL) {
__db_errx(env,
"Databases may not become secondary indices while cursors are open");
ret = EINVAL;
goto err;
}
if ((ret = __db_associate_arg(dbp, sdbp, callback, flags)) != 0)
goto err;
if (IS_DB_AUTO_COMMIT(dbp, txn)) {
if ((ret = __txn_begin(env, ip, NULL, &txn, 0)) != 0)
goto err;
txn_local = 1;
}
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 0)) != 0)
goto err;
while ((sdbc = TAILQ_FIRST(&sdbp->free_queue)) != NULL)
if ((ret = __dbc_destroy(sdbc)) != 0)
goto err;
ret = __db_associate(dbp, ip, txn, sdbp, callback, flags);
err: if (txn_local &&
(t_ret = __db_txn_auto_resolve(env, txn, 0, ret)) && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
static int
__db_associate_arg(dbp, sdbp, callback, flags)
DB *dbp, *sdbp;
int (*callback) __P((DB *, const DBT *, const DBT *, DBT *));
u_int32_t flags;
{
ENV *env;
int ret;
env = dbp->env;
if (F_ISSET(sdbp, DB_AM_SECONDARY)) {
__db_errx(env,
"Secondary index handles may not be re-associated");
return (EINVAL);
}
if (F_ISSET(dbp, DB_AM_SECONDARY)) {
__db_errx(env,
"Secondary indices may not be used as primary databases");
return (EINVAL);
}
if (F_ISSET(dbp, DB_AM_DUP)) {
__db_errx(env,
"Primary databases may not be configured with duplicates");
return (EINVAL);
}
if (F_ISSET(dbp, DB_AM_RENUMBER)) {
__db_errx(env,
"Renumbering recno databases may not be used as primary databases");
return (EINVAL);
}
if (dbp->env != sdbp->env &&
(!F_ISSET(dbp->env, ENV_DBLOCAL) ||
!F_ISSET(sdbp->env, ENV_DBLOCAL))) {
__db_errx(env,
"The primary and secondary must be opened in the same environment");
return (EINVAL);
}
if ((DB_IS_THREADED(dbp) && !DB_IS_THREADED(sdbp)) ||
(!DB_IS_THREADED(dbp) && DB_IS_THREADED(sdbp))) {
__db_errx(env,
"The DB_THREAD setting must be the same for primary and secondary");
return (EINVAL);
}
if (callback == NULL &&
(!F_ISSET(dbp, DB_AM_RDONLY) || !F_ISSET(sdbp, DB_AM_RDONLY))) {
__db_errx(env,
"Callback function may be NULL only when database handles are read-only");
return (EINVAL);
}
if ((ret = __db_fchk(env, "DB->associate", flags, DB_CREATE |
DB_IMMUTABLE_KEY)) != 0)
return (ret);
return (0);
}
int
__db_close_pp(dbp, flags)
DB *dbp;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
ret = 0;
if (flags != 0 && flags != DB_NOSYNC)
ret = __db_ferr(env, "DB->close", 0);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (t_ret = __db_rep_enter(dbp, 0, 0, 0)) != 0) {
handle_check = 0;
if (ret == 0)
ret = t_ret;
}
if ((t_ret = __db_close(dbp, NULL, flags)) != 0 && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
int
__db_cursor_pp(dbp, txn, dbcp, flags)
DB *dbp;
DB_TXN *txn;
DBC **dbcp;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
REGENV *renv;
int rep_blocked, ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->cursor");
ENV_ENTER(env, ip);
rep_blocked = 0;
if (txn == NULL && IS_ENV_REPLICATED(env)) {
if ((ret = __op_rep_enter(env)) != 0)
goto err;
rep_blocked = 1;
renv = env->reginfo->primary;
if (dbp->timestamp != renv->rep_timestamp) {
__db_errx(env, "%s %s",
"replication recovery unrolled committed transactions;",
"open DB and DBcursor handles must be closed");
ret = DB_REP_HANDLE_DEAD;
goto err;
}
}
if ((ret = __db_cursor_arg(dbp, flags)) != 0)
goto err;
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 1)) != 0)
goto err;
ret = __db_cursor(dbp, ip, txn, dbcp, flags);
err:
if (ret != 0 && rep_blocked)
(void)__op_rep_exit(env);
ENV_LEAVE(env, ip);
return (ret);
}
int
__db_cursor(dbp, ip, txn, dbcp, flags)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
DBC **dbcp;
u_int32_t flags;
{
DBC *dbc;
ENV *env;
db_lockmode_t mode;
u_int32_t op;
int ret;
env = dbp->env;
if (MULTIVERSION(dbp) && txn == NULL && (LF_ISSET(DB_TXN_SNAPSHOT) ||
F_ISSET(env->dbenv, DB_ENV_TXN_SNAPSHOT))) {
if ((ret =
__txn_begin(env, ip, NULL, &txn, DB_TXN_SNAPSHOT)) != 0)
return (ret);
F_SET(txn, TXN_PRIVATE);
}
if ((ret = __db_cursor_int(dbp, ip,
txn, dbp->type, PGNO_INVALID, 0, NULL, &dbc)) != 0)
return (ret);
if (CDB_LOCKING(env)) {
op = LF_ISSET(DB_OPFLAGS_MASK);
mode = (op == DB_WRITELOCK) ? DB_LOCK_WRITE :
((op == DB_WRITECURSOR || txn != NULL) ? DB_LOCK_IWRITE :
DB_LOCK_READ);
if ((ret = __lock_get(env, dbc->locker, 0,
&dbc->lock_dbt, mode, &dbc->mylock)) != 0)
goto err;
if (op == DB_WRITECURSOR)
F_SET(dbc, DBC_WRITECURSOR);
if (op == DB_WRITELOCK)
F_SET(dbc, DBC_WRITER);
}
if (LF_ISSET(DB_READ_UNCOMMITTED) ||
(txn != NULL && F_ISSET(txn, TXN_READ_UNCOMMITTED)))
F_SET(dbc, DBC_READ_UNCOMMITTED);
if (LF_ISSET(DB_READ_COMMITTED) ||
(txn != NULL && F_ISSET(txn, TXN_READ_COMMITTED)))
F_SET(dbc, DBC_READ_COMMITTED);
*dbcp = dbc;
return (0);
err: (void)__dbc_close(dbc);
return (ret);
}
static int
__db_cursor_arg(dbp, flags)
DB *dbp;
u_int32_t flags;
{
ENV *env;
env = dbp->env;
if (LF_ISSET(DB_READ_COMMITTED | DB_READ_UNCOMMITTED)) {
if (!LOCKING_ON(env))
return (__db_fnl(env, "DB->cursor"));
}
LF_CLR(DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_TXN_SNAPSHOT);
switch (flags) {
case 0:
break;
case DB_WRITECURSOR:
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DB->cursor"));
if (!CDB_LOCKING(env))
return (__db_ferr(env, "DB->cursor", 0));
break;
case DB_WRITELOCK:
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DB->cursor"));
break;
default:
return (__db_ferr(env, "DB->cursor", 0));
}
return (0);
}
int
__db_del_pp(dbp, txn, key, flags)
DB *dbp;
DB_TXN *txn;
DBT *key;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret, txn_local;
env = dbp->env;
txn_local = 0;
STRIP_AUTO_COMMIT(flags);
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->del");
#ifdef CONFIG_TEST
if (IS_REP_MASTER(env))
DB_TEST_WAIT(env, env->test_check);
#endif
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if ((ret = __db_del_arg(dbp, key, flags)) != 0)
goto err;
if (IS_DB_AUTO_COMMIT(dbp, txn)) {
if ((ret = __txn_begin(env, ip, NULL, &txn, 0)) != 0)
goto err;
txn_local = 1;
}
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 0)) != 0)
goto err;
ret = __db_del(dbp, ip, txn, key, flags);
err: if (txn_local &&
(t_ret = __db_txn_auto_resolve(env, txn, 0, ret)) && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
__dbt_userfree(env, key, NULL, NULL);
return (ret);
}
static int
__db_del_arg(dbp, key, flags)
DB *dbp;
DBT *key;
u_int32_t flags;
{
ENV *env;
int ret;
env = dbp->env;
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DB->del"));
switch (flags) {
case 0:
if ((ret = __dbt_usercopy(env, key)) != 0)
return (ret);
break;
default:
return (__db_ferr(env, "DB->del", 0));
}
return (0);
}
int
__db_exists(dbp, txn, key, flags)
DB *dbp;
DB_TXN *txn;
DBT *key;
u_int32_t flags;
{
DBT data;
int ret;
if ((ret = __db_fchk(dbp->env, "DB->exists", flags,
DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_RMW)) != 0)
return (ret);
memset(&data, 0, sizeof(data));
data.dlen = 0;
data.flags = DB_DBT_PARTIAL | DB_DBT_USERMEM;
return (dbp->get(dbp, txn, key, &data, flags));
}
int
__db_fd_pp(dbp, fdp)
DB *dbp;
int *fdp;
{
DB_FH *fhp;
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->fd");
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (ret = __db_rep_enter(dbp, 1, 0, 0)) != 0)
goto err;
if ((ret = __mp_xxx_fh(dbp->mpf, &fhp)) == 0) {
if (fhp == NULL) {
*fdp = -1;
__db_errx(env,
"Database does not have a valid file handle");
ret = ENOENT;
} else
*fdp = fhp->fd;
}
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
err: ENV_LEAVE(env, ip);
return (ret);
}
int
__db_get_pp(dbp, txn, key, data, flags)
DB *dbp;
DB_TXN *txn;
DBT *key, *data;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
u_int32_t mode;
int handle_check, ignore_lease, ret, t_ret, txn_local;
env = dbp->env;
mode = 0;
txn_local = 0;
STRIP_AUTO_COMMIT(flags);
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->get");
ignore_lease = LF_ISSET(DB_IGNORE_LEASE) ? 1 : 0;
LF_CLR(DB_IGNORE_LEASE);
if ((ret = __db_get_arg(dbp, key, data, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if (LF_ISSET(DB_READ_UNCOMMITTED))
mode = DB_READ_UNCOMMITTED;
else if ((flags & DB_OPFLAGS_MASK) == DB_CONSUME ||
(flags & DB_OPFLAGS_MASK) == DB_CONSUME_WAIT) {
mode = DB_WRITELOCK;
if (IS_DB_AUTO_COMMIT(dbp, txn)) {
if ((ret = __txn_begin(env, ip, NULL, &txn, 0)) != 0)
goto err;
txn_local = 1;
}
}
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID,
mode == DB_WRITELOCK || LF_ISSET(DB_RMW) ? 0 : 1)) != 0)
goto err;
ret = __db_get(dbp, ip, txn, key, data, flags);
if (ret == 0 &&
IS_REP_MASTER(env) && IS_USING_LEASES(env) && !ignore_lease)
ret = __rep_lease_check(env, 1);
err: if (txn_local &&
(t_ret = __db_txn_auto_resolve(env, txn, 0, ret)) && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
__dbt_userfree(env, key, NULL, data);
return (ret);
}
int
__db_get(dbp, ip, txn, key, data, flags)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
DBT *key, *data;
u_int32_t flags;
{
DBC *dbc;
u_int32_t mode;
int ret, t_ret;
mode = 0;
if (LF_ISSET(DB_READ_UNCOMMITTED)) {
mode = DB_READ_UNCOMMITTED;
LF_CLR(DB_READ_UNCOMMITTED);
} else if (LF_ISSET(DB_READ_COMMITTED)) {
mode = DB_READ_COMMITTED;
LF_CLR(DB_READ_COMMITTED);
} else if ((flags & DB_OPFLAGS_MASK) == DB_CONSUME ||
(flags & DB_OPFLAGS_MASK) == DB_CONSUME_WAIT)
mode = DB_WRITELOCK;
if ((ret = __db_cursor(dbp, ip, txn, &dbc, mode)) != 0)
return (ret);
DEBUG_LREAD(dbc, txn, "DB->get", key, NULL, flags);
F_SET(dbc, DBC_TRANSIENT);
SET_RET_MEM(dbc, dbp);
if (LF_ISSET(~(DB_RMW | DB_MULTIPLE)) == 0)
LF_SET(DB_SET);
ret = __dbc_get(dbc, key, data, flags);
if (dbc != NULL && (t_ret = __dbc_close(dbc)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
static int
__db_get_arg(dbp, key, data, flags)
const DB *dbp;
DBT *key, *data;
u_int32_t flags;
{
ENV *env;
int dirty, multi, ret;
env = dbp->env;
dirty = 0;
if (LF_ISSET(DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_RMW)) {
if (!LOCKING_ON(env))
return (__db_fnl(env, "DB->get"));
if ((ret = __db_fcchk(env, "DB->get",
flags, DB_READ_UNCOMMITTED, DB_READ_COMMITTED)) != 0)
return (ret);
if (LF_ISSET(DB_READ_COMMITTED | DB_READ_UNCOMMITTED))
dirty = 1;
LF_CLR(DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_RMW);
}
multi = 0;
if (LF_ISSET(DB_MULTIPLE | DB_MULTIPLE_KEY)) {
if (LF_ISSET(DB_MULTIPLE_KEY))
goto multi_err;
multi = LF_ISSET(DB_MULTIPLE) ? 1 : 0;
LF_CLR(DB_MULTIPLE);
}
switch (flags) {
case DB_GET_BOTH:
if ((ret = __dbt_usercopy(env, data)) != 0)
return (ret);
case 0:
if ((ret = __dbt_usercopy(env, key)) != 0) {
__dbt_userfree(env, key, NULL, data);
return (ret);
}
break;
case DB_SET_RECNO:
if (!F_ISSET(dbp, DB_AM_RECNUM))
goto err;
if ((ret = __dbt_usercopy(env, key)) != 0)
return (ret);
break;
case DB_CONSUME:
case DB_CONSUME_WAIT:
if (dirty) {
__db_errx(env,
"%s is not supported with DB_CONSUME or DB_CONSUME_WAIT",
LF_ISSET(DB_READ_UNCOMMITTED) ?
"DB_READ_UNCOMMITTED" : "DB_READ_COMMITTED");
return (EINVAL);
}
if (multi)
multi_err: return (__db_ferr(env, "DB->get", 1));
if (dbp->type == DB_QUEUE)
break;
default:
err: return (__db_ferr(env, "DB->get", 0));
}
if ((ret =
__dbt_ferr(dbp, "key", key, DB_RETURNS_A_KEY(dbp, flags))) != 0)
return (ret);
if ((ret = __dbt_ferr(dbp, "data", data, 1)) != 0)
return (ret);
if (multi) {
if (!F_ISSET(data, DB_DBT_USERMEM)) {
__db_errx(env,
"DB_MULTIPLE requires DB_DBT_USERMEM be set");
return (EINVAL);
}
if (F_ISSET(key, DB_DBT_PARTIAL) ||
F_ISSET(data, DB_DBT_PARTIAL)) {
__db_errx(env,
"DB_MULTIPLE does not support DB_DBT_PARTIAL");
return (EINVAL);
}
if (data->ulen < 1024 ||
data->ulen < dbp->pgsize || data->ulen % 1024 != 0) {
__db_errx(env, "%s%s",
"DB_MULTIPLE buffers must be ",
"aligned, at least page size and multiples of 1KB");
return (EINVAL);
}
}
return (0);
}
int
__db_join_pp(primary, curslist, dbcp, flags)
DB *primary;
DBC **curslist, **dbcp;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = primary->env;
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (ret =
__db_rep_enter(primary, 1, 0, curslist[0]->txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if ((ret = __db_join_arg(primary, curslist, flags)) == 0)
ret = __db_join(primary, curslist, dbcp, flags);
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
err: ENV_LEAVE(env, ip);
return (ret);
}
static int
__db_join_arg(primary, curslist, flags)
DB *primary;
DBC **curslist;
u_int32_t flags;
{
DB_TXN *txn;
ENV *env;
int i;
env = primary->env;
switch (flags) {
case 0:
case DB_JOIN_NOSORT:
break;
default:
return (__db_ferr(env, "DB->join", 0));
}
if (curslist == NULL || curslist[0] == NULL) {
__db_errx(env,
"At least one secondary cursor must be specified to DB->join");
return (EINVAL);
}
txn = curslist[0]->txn;
for (i = 1; curslist[i] != NULL; i++)
if (curslist[i]->txn != txn) {
__db_errx(env,
"All secondary cursors must share the same transaction");
return (EINVAL);
}
return (0);
}
int
__db_key_range_pp(dbp, txn, key, kr, flags)
DB *dbp;
DB_TXN *txn;
DBT *key;
DB_KEY_RANGE *kr;
u_int32_t flags;
{
DBC *dbc;
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->key_range");
if (flags != 0)
return (__db_ferr(env, "DB->key_range", 0));
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 1)) != 0)
goto err;
switch (dbp->type) {
case DB_BTREE:
#ifndef HAVE_BREW
if ((ret = __dbt_usercopy(env, key)) != 0)
goto err;
if ((ret = __db_cursor(dbp, ip, txn, &dbc, 0)) != 0)
break;
DEBUG_LWRITE(dbc, NULL, "bam_key_range", NULL, NULL, 0);
ret = __bam_key_range(dbc, key, kr, flags);
if ((t_ret = __dbc_close(dbc)) != 0 && ret == 0)
ret = t_ret;
__dbt_userfree(env, key, NULL, NULL);
break;
#else
COMPQUIET(dbc, NULL);
COMPQUIET(key, NULL);
COMPQUIET(kr, NULL);
#endif
case DB_HASH:
case DB_QUEUE:
case DB_RECNO:
ret = __dbh_am_chk(dbp, DB_OK_BTREE);
break;
case DB_UNKNOWN:
default:
ret = __db_unknown_type(env, "DB->key_range", dbp->type);
break;
}
err:
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
int
__db_open_pp(dbp, txn, fname, dname, type, flags, mode)
DB *dbp;
DB_TXN *txn;
const char *fname, *dname;
DBTYPE type;
u_int32_t flags;
int mode;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, nosync, remove_me, ret, t_ret, txn_local;
env = dbp->env;
nosync = 1;
handle_check = remove_me = txn_local = 0;
ENV_ENTER(env, ip);
if ((fname != NULL &&
(ret = __os_strdup(env, fname, &dbp->fname)) != 0))
goto err;
if ((dname != NULL &&
(ret = __os_strdup(env, dname, &dbp->dname)) != 0))
goto err;
dbp->open_flags = flags;
dbp->orig_flags = dbp->flags;
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if (IS_ENV_AUTO_COMMIT(env, txn, flags)) {
if ((ret = __db_txn_auto_init(env, ip, &txn)) != 0)
goto err;
txn_local = 1;
} else if (txn != NULL && !TXN_ON(env) &&
(!CDB_LOCKING(env) || !F_ISSET(txn, TXN_CDSGROUP))) {
ret = __db_not_txn_env(env);
goto err;
}
LF_CLR(DB_AUTO_COMMIT);
if ((ret = __db_open_arg(dbp, txn, fname, dname, type, flags)) == 0)
if ((ret = __db_open(dbp, ip, txn, fname, dname, type,
flags, mode, PGNO_BASE_MD)) != 0)
goto txnerr;
if (dname == NULL && !IS_RECOVERING(env) && !LF_ISSET(DB_RDONLY) &&
!LF_ISSET(DB_RDWRMASTER) && F_ISSET(dbp, DB_AM_SUBDB)) {
__db_errx(env,
"files containing multiple databases may only be opened read-only");
ret = EINVAL;
goto txnerr;
}
if (F_ISSET(dbp, DB_AM_CREATED | DB_AM_CREATED_MSTR))
nosync = 0;
F_CLR(dbp, DB_AM_DISCARD | DB_AM_CREATED | DB_AM_CREATED_MSTR);
txnerr: if (ret != 0 && !IS_REAL_TXN(txn)) {
remove_me = (F_ISSET(dbp, DB_AM_CREATED) &&
(fname != NULL || dname != NULL)) ? 1 : 0;
if (F_ISSET(dbp, DB_AM_CREATED_MSTR) ||
(dname == NULL && remove_me))
(void)__db_remove_int(dbp,
ip, txn, fname, NULL, DB_FORCE);
else if (remove_me)
(void)__db_remove_int(dbp,
ip, txn, fname, dname, DB_FORCE);
}
if (txn_local && (t_ret =
__db_txn_auto_resolve(env, txn, nosync, ret)) && ret == 0)
ret = t_ret;
err:
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
static int
__db_open_arg(dbp, txn, fname, dname, type, flags)
DB *dbp;
DB_TXN *txn;
const char *fname, *dname;
DBTYPE type;
u_int32_t flags;
{
ENV *env;
u_int32_t ok_flags;
int ret;
env = dbp->env;
#undef OKFLAGS
#define OKFLAGS \
(DB_AUTO_COMMIT | DB_CREATE | DB_EXCL | DB_FCNTL_LOCKING | \
DB_MULTIVERSION | DB_NOMMAP | DB_NO_AUTO_COMMIT | DB_RDONLY | \
DB_RDWRMASTER | DB_READ_UNCOMMITTED | DB_THREAD | DB_TRUNCATE)
if ((ret = __db_fchk(env, "DB->open", flags, OKFLAGS)) != 0)
return (ret);
if (LF_ISSET(DB_EXCL) && !LF_ISSET(DB_CREATE))
return (__db_ferr(env, "DB->open", 1));
if (LF_ISSET(DB_RDONLY) && LF_ISSET(DB_CREATE))
return (__db_ferr(env, "DB->open", 1));
#ifdef HAVE_VXWORKS
if (LF_ISSET(DB_TRUNCATE)) {
__db_errx(env, "DB_TRUNCATE not supported on VxWorks");
return (DB_OPNOTSUP);
}
#endif
switch (type) {
case DB_UNKNOWN:
if (LF_ISSET(DB_CREATE|DB_TRUNCATE)) {
__db_errx(env,
"DB_UNKNOWN type specified with DB_CREATE or DB_TRUNCATE");
return (EINVAL);
}
ok_flags = 0;
break;
case DB_BTREE:
ok_flags = DB_OK_BTREE;
break;
case DB_HASH:
#ifndef HAVE_HASH
return (__db_no_hash_am(env));
#endif
ok_flags = DB_OK_HASH;
break;
case DB_QUEUE:
#ifndef HAVE_QUEUE
return (__db_no_queue_am(env));
#endif
ok_flags = DB_OK_QUEUE;
break;
case DB_RECNO:
ok_flags = DB_OK_RECNO;
break;
default:
__db_errx(env, "unknown type: %lu", (u_long)type);
return (EINVAL);
}
if (ok_flags)
DB_ILLEGAL_METHOD(dbp, ok_flags);
if (!F_ISSET(env, ENV_DBLOCAL | ENV_OPEN_CALLED)) {
__db_errx(env, "database environment not yet opened");
return (EINVAL);
}
if (!F_ISSET(env, ENV_DBLOCAL) && !MPOOL_ON(env)) {
__db_errx(env, "environment did not include a memory pool");
return (EINVAL);
}
if (LF_ISSET(DB_THREAD) && !F_ISSET(env, ENV_DBLOCAL | ENV_THREAD)) {
__db_errx(env, "environment not created using DB_THREAD");
return (EINVAL);
}
if (LF_ISSET(DB_MULTIVERSION) && !IS_REAL_TXN(txn)) {
__db_errx(env,
"DB_MULTIVERSION illegal without a transaction specified");
return (EINVAL);
}
if (LF_ISSET(DB_MULTIVERSION) && type == DB_QUEUE) {
__db_errx(env,
"DB_MULTIVERSION illegal with queue databases");
return (EINVAL);
}
if (LF_ISSET(DB_TRUNCATE) && (LOCKING_ON(env) || txn != NULL)) {
__db_errx(env,
"DB_TRUNCATE illegal with %s specified",
LOCKING_ON(env) ? "locking" : "transactions");
return (EINVAL);
}
if (dname != NULL) {
if (type == DB_QUEUE && fname != NULL) {
__db_errx(
env, "Queue databases must be one-per-file");
return (EINVAL);
}
if (fname == NULL)
F_CLR(dbp, DB_AM_CHKSUM | DB_AM_ENCRYPT);
}
return (0);
}
int
__db_pget_pp(dbp, txn, skey, pkey, data, flags)
DB *dbp;
DB_TXN *txn;
DBT *skey, *pkey, *data;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ignore_lease, ret, t_ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->pget");
ignore_lease = LF_ISSET(DB_IGNORE_LEASE) ? 1 : 0;
LF_CLR(DB_IGNORE_LEASE);
if ((ret = __db_pget_arg(dbp, pkey, flags)) != 0 ||
(ret = __db_get_arg(dbp, skey, data, flags)) != 0) {
__dbt_userfree(env, skey, pkey, data);
return (ret);
}
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
ret = __db_pget(dbp, ip, txn, skey, pkey, data, flags);
if (ret == 0 &&
IS_REP_MASTER(env) && IS_USING_LEASES(env) && !ignore_lease)
ret = __rep_lease_check(env, 1);
err:
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
__dbt_userfree(env, skey, pkey, data);
return (ret);
}
int
__db_pget(dbp, ip, txn, skey, pkey, data, flags)
DB *dbp;
DB_THREAD_INFO *ip;
DB_TXN *txn;
DBT *skey, *pkey, *data;
u_int32_t flags;
{
DBC *dbc;
u_int32_t mode;
int ret, t_ret;
if (LF_ISSET(DB_READ_UNCOMMITTED)) {
mode = DB_READ_UNCOMMITTED;
LF_CLR(DB_READ_UNCOMMITTED);
} else if (LF_ISSET(DB_READ_COMMITTED)) {
mode = DB_READ_COMMITTED;
LF_CLR(DB_READ_COMMITTED);
} else
mode = 0;
if ((ret = __db_cursor(dbp, ip, txn, &dbc, mode)) != 0)
return (ret);
SET_RET_MEM(dbc, dbp);
DEBUG_LREAD(dbc, txn, "__db_pget", skey, NULL, flags);
if (pkey == NULL)
dbc->rkey = &dbc->my_rkey;
if (flags == 0 || flags == DB_RMW)
flags |= DB_SET;
ret = __dbc_pget(dbc, skey, pkey, data, flags);
if ((t_ret = __dbc_close(dbc)) != 0 && ret == 0)
ret = t_ret;
return (ret);
}
static int
__db_pget_arg(dbp, pkey, flags)
DB *dbp;
DBT *pkey;
u_int32_t flags;
{
ENV *env;
int ret;
env = dbp->env;
if (!F_ISSET(dbp, DB_AM_SECONDARY)) {
__db_errx(env,
"DB->pget may only be used on secondary indices");
return (EINVAL);
}
if (LF_ISSET(DB_MULTIPLE | DB_MULTIPLE_KEY)) {
__db_errx(env,
"DB_MULTIPLE and DB_MULTIPLE_KEY may not be used on secondary indices");
return (EINVAL);
}
LF_CLR(DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_RMW);
switch (flags) {
case DB_CONSUME:
case DB_CONSUME_WAIT:
return (__db_ferr(env, "DB->pget", 0));
default:
break;
}
if (pkey != NULL &&
(ret = __dbt_ferr(dbp, "primary key", pkey, 1)) != 0)
return (ret);
if (flags == DB_GET_BOTH) {
if (pkey == NULL) {
__db_errx(env,
"DB_GET_BOTH on a secondary index requires a primary key");
return (EINVAL);
}
if ((ret = __dbt_usercopy(env, pkey)) != 0)
return (ret);
}
return (0);
}
int
__db_put_pp(dbp, txn, key, data, flags)
DB *dbp;
DB_TXN *txn;
DBT *key, *data;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, txn_local, t_ret;
env = dbp->env;
txn_local = 0;
STRIP_AUTO_COMMIT(flags);
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->put");
if ((ret = __db_put_arg(dbp, key, data, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, txn != NULL)) != 0) {
handle_check = 0;
goto err;
}
if (IS_DB_AUTO_COMMIT(dbp, txn)) {
if ((ret = __txn_begin(env, ip, NULL, &txn, 0)) != 0)
goto err;
txn_local = 1;
}
if ((ret = __db_check_txn(dbp, txn, DB_LOCK_INVALIDID, 0)) != 0)
goto err;
ret = __db_put(dbp, ip, txn, key, data, flags);
err: if (txn_local &&
(t_ret = __db_txn_auto_resolve(env, txn, 0, ret)) && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
__dbt_userfree(env, key, NULL, data);
return (ret);
}
static int
__db_put_arg(dbp, key, data, flags)
DB *dbp;
DBT *key, *data;
u_int32_t flags;
{
ENV *env;
int ret, returnkey;
env = dbp->env;
returnkey = 0;
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DB->put"));
if (F_ISSET(dbp, DB_AM_SECONDARY)) {
__db_errx(env, "DB->put forbidden on secondary indices");
return (EINVAL);
}
switch (flags) {
case 0:
case DB_NOOVERWRITE:
break;
case DB_APPEND:
if (dbp->type != DB_RECNO && dbp->type != DB_QUEUE)
goto err;
returnkey = 1;
break;
case DB_NODUPDATA:
if (F_ISSET(dbp, DB_AM_DUPSORT))
break;
default:
err: return (__db_ferr(env, "DB->put", 0));
}
if (((returnkey && key != NULL) || !returnkey) &&
(ret = __dbt_ferr(dbp, "key", key, returnkey)) != 0)
return (ret);
if ((ret = __dbt_ferr(dbp, "data", data, 0)) != 0)
return (ret);
if ((returnkey &&
key != NULL && F_ISSET(key, DB_DBT_PARTIAL) && key->dlen != 0) ||
(!returnkey && F_ISSET(key, DB_DBT_PARTIAL)))
return (__db_ferr(env, "key DBT", 0));
if (F_ISSET(data, DB_DBT_PARTIAL) &&
(F_ISSET(dbp, DB_AM_DUP) || F_ISSET(key, DB_DBT_DUPOK))) {
__db_errx(env,
"a partial put in the presence of duplicates requires a cursor operation");
return (EINVAL);
}
if ((flags != DB_APPEND && (ret = __dbt_usercopy(env, key)) != 0) ||
(ret = __dbt_usercopy(env, data)) != 0)
return (ret);
return (0);
}
int
__db_compact_pp(dbp, txn, start, stop, c_data, flags, end)
DB *dbp;
DB_TXN *txn;
DBT *start, *stop;
DB_COMPACT *c_data;
u_int32_t flags;
DBT *end;
{
DB_COMPACT *dp, l_data;
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->compact");
if ((ret = __db_fchk(
env, "DB->compact", flags, DB_FREELIST_ONLY | DB_FREE_SPACE)) != 0)
return (ret);
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DB->compact"));
if (start != NULL && (ret = __dbt_usercopy(env, start)) != 0)
return (ret);
if (stop != NULL && (ret = __dbt_usercopy(env, stop)) != 0)
return (ret);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (ret = __db_rep_enter(dbp, 1, 0, 0)) != 0) {
handle_check = 0;
goto err;
}
if (c_data == NULL) {
dp = &l_data;
memset(dp, 0, sizeof(*dp));
} else
dp = c_data;
switch (dbp->type) {
case DB_HASH:
if (!LF_ISSET(DB_FREELIST_ONLY))
goto err;
case DB_BTREE:
case DB_RECNO:
ret = __bam_compact(dbp, ip, txn, start, stop, dp, flags, end);
break;
default:
err: ret = __dbh_am_chk(dbp, DB_OK_BTREE);
break;
}
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
__dbt_userfree(env, start, stop, NULL);
return (ret);
}
int
__db_associate_foreign_pp(fdbp, dbp, callback, flags)
DB *dbp, *fdbp;
int (*callback) __P((DB *, const DBT *, DBT *, const DBT *, int *));
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
PANIC_CHECK(env);
STRIP_AUTO_COMMIT(flags);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check &&
(ret = __db_rep_enter(dbp, 1, 0, 0)) != 0) {
handle_check = 0;
goto err;
}
if ((ret = __db_associate_foreign_arg(fdbp, dbp, callback, flags)) != 0)
goto err;
ret = __db_associate_foreign(fdbp, dbp, callback, flags);
err:
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
static int
__db_associate_foreign_arg(fdbp, dbp, callback, flags)
DB *dbp, *fdbp;
int (*callback) __P((DB *, const DBT *, DBT *, const DBT *, int *));
u_int32_t flags;
{
ENV *env;
env = fdbp->env;
if (F_ISSET(fdbp, DB_AM_SECONDARY)) {
__db_errx(env,
"Secondary indices may not be used as foreign databases");
return (EINVAL);
}
if (F_ISSET(fdbp, DB_AM_DUP)) {
__db_errx(env,
"Foreign databases may not be configured with duplicates");
return (EINVAL);
}
if (F_ISSET(fdbp, DB_AM_RENUMBER)) {
__db_errx(env,
"Renumbering recno databases may not be used as foreign databases");
return (EINVAL);
}
if (!F_ISSET(dbp, DB_AM_SECONDARY)) {
__db_errx(env,
"The associating database must be a secondary index.");
return (EINVAL);
}
if (LF_ISSET(DB_FOREIGN_NULLIFY) && callback == NULL) {
__db_errx(env,
"When specifying a delete action of nullify, a callback function needs to be configured");
return (EINVAL);
} else if (!LF_ISSET(DB_FOREIGN_NULLIFY) && callback != NULL) {
__db_errx(env,
"When not specifying a delete action of nullify, a callback function cannot be configured");
return (EINVAL);
}
return (0);
}
int
__db_sync_pp(dbp, flags)
DB *dbp;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
DB_ILLEGAL_BEFORE_OPEN(dbp, "DB->sync");
if (flags != 0)
return (__db_ferr(env, "DB->sync", 0));
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (ret = __db_rep_enter(dbp, 1, 0, 0)) != 0) {
handle_check = 0;
goto err;
}
ret = __db_sync(dbp);
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
err: ENV_LEAVE(env, ip);
return (ret);
}
int
__dbc_close_pp(dbc)
DBC *dbc;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
dbp = dbc->dbp;
env = dbp->env;
if (!F_ISSET(dbc, DBC_ACTIVE)) {
__db_errx(env, "Closing already-closed cursor");
return (EINVAL);
}
ENV_ENTER(env, ip);
handle_check = dbc->txn == NULL && IS_ENV_REPLICATED(env);
ret = __dbc_close(dbc);
if (handle_check &&
(t_ret = __op_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
int
__dbc_count_pp(dbc, recnop, flags)
DBC *dbc;
db_recno_t *recnop;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ret;
dbp = dbc->dbp;
env = dbp->env;
if (flags != 0)
return (__db_ferr(env, "DBcursor->count", 0));
if (!IS_INITIALIZED(dbc))
return (__db_curinval(env));
ENV_ENTER(env, ip);
ret = __dbc_count(dbc, recnop);
ENV_LEAVE(env, ip);
return (ret);
}
int
__dbc_del_pp(dbc, flags)
DBC *dbc;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ret;
dbp = dbc->dbp;
env = dbp->env;
if ((ret = __dbc_del_arg(dbc, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
if ((ret = __db_check_txn(dbp, dbc->txn, dbc->locker, 0)) != 0)
goto err;
DEBUG_LWRITE(dbc, dbc->txn, "DBcursor->del", NULL, NULL, flags);
ret = __dbc_del(dbc, flags);
err: ENV_LEAVE(env, ip);
return (ret);
}
static int
__dbc_del_arg(dbc, flags)
DBC *dbc;
u_int32_t flags;
{
DB *dbp;
ENV *env;
dbp = dbc->dbp;
env = dbp->env;
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DBcursor->del"));
switch (flags) {
case 0:
break;
case DB_UPDATE_SECONDARY:
DB_ASSERT(env, F_ISSET(dbp, DB_AM_SECONDARY));
break;
default:
return (__db_ferr(env, "DBcursor->del", 0));
}
if (!IS_INITIALIZED(dbc))
return (__db_curinval(env));
return (0);
}
int
__dbc_dup_pp(dbc, dbcp, flags)
DBC *dbc, **dbcp;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ret;
dbp = dbc->dbp;
env = dbp->env;
if (flags != 0 && flags != DB_POSITION)
return (__db_ferr(env, "DBcursor->dup", 0));
ENV_ENTER(env, ip);
ret = __dbc_dup(dbc, dbcp, flags);
ENV_LEAVE(env, ip);
return (ret);
}
int
__dbc_get_pp(dbc, key, data, flags)
DBC *dbc;
DBT *key, *data;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ignore_lease, ret;
dbp = dbc->dbp;
env = dbp->env;
ignore_lease = LF_ISSET(DB_IGNORE_LEASE) ? 1 : 0;
LF_CLR(DB_IGNORE_LEASE);
if ((ret = __dbc_get_arg(dbc, key, data, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
DEBUG_LREAD(dbc, dbc->txn, "DBcursor->get",
flags == DB_SET || flags == DB_SET_RANGE ? key : NULL, NULL, flags);
ret = __dbc_get(dbc, key, data, flags);
if (ret == 0 &&
IS_REP_MASTER(env) && IS_USING_LEASES(env) && !ignore_lease)
ret = __rep_lease_check(env, 1);
ENV_LEAVE(env, ip);
__dbt_userfree(env, key, NULL, data);
return (ret);
}
static int
__dbc_get_arg(dbc, key, data, flags)
DBC *dbc;
DBT *key, *data;
u_int32_t flags;
{
DB *dbp;
ENV *env;
int dirty, multi, ret;
dbp = dbc->dbp;
env = dbp->env;
dirty = 0;
if (LF_ISSET(DB_READ_UNCOMMITTED | DB_RMW)) {
if (!LOCKING_ON(env))
return (__db_fnl(env, "DBcursor->get"));
if (LF_ISSET(DB_READ_UNCOMMITTED))
dirty = 1;
LF_CLR(DB_READ_UNCOMMITTED | DB_RMW);
}
multi = 0;
if (LF_ISSET(DB_MULTIPLE | DB_MULTIPLE_KEY)) {
multi = 1;
if (LF_ISSET(DB_MULTIPLE) && LF_ISSET(DB_MULTIPLE_KEY))
goto multi_err;
LF_CLR(DB_MULTIPLE | DB_MULTIPLE_KEY);
}
switch (flags) {
case DB_CONSUME:
case DB_CONSUME_WAIT:
if (dirty) {
__db_errx(env,
"DB_READ_UNCOMMITTED is not supported with DB_CONSUME or DB_CONSUME_WAIT");
return (EINVAL);
}
if (dbp->type != DB_QUEUE)
goto err;
break;
case DB_CURRENT:
case DB_FIRST:
case DB_NEXT:
case DB_NEXT_DUP:
case DB_NEXT_NODUP:
break;
case DB_LAST:
case DB_PREV:
case DB_PREV_DUP:
case DB_PREV_NODUP:
if (multi)
multi_err: return (__db_ferr(env, "DBcursor->get", 1));
break;
case DB_GET_BOTHC:
if (dbp->type == DB_QUEUE)
goto err;
case DB_GET_BOTH:
case DB_GET_BOTH_RANGE:
if ((ret = __dbt_usercopy(env, data)) != 0)
goto err;
case DB_SET:
case DB_SET_RANGE:
if ((ret = __dbt_usercopy(env, key)) != 0)
goto err;
break;
case DB_GET_RECNO:
if (!F_ISSET(dbp, DB_AM_RECNUM) &&
(!F_ISSET(dbp, DB_AM_SECONDARY) ||
!F_ISSET(dbp->s_primary, DB_AM_RECNUM)))
goto err;
break;
case DB_SET_RECNO:
if (!F_ISSET(dbp, DB_AM_RECNUM))
goto err;
if ((ret = __dbt_usercopy(env, key)) != 0)
goto err;
break;
default:
err: __dbt_userfree(env, key, NULL, data);
return (__db_ferr(env, "DBcursor->get", 0));
}
if ((ret = __dbt_ferr(dbp, "key", key, 0)) != 0)
return (ret);
if ((ret = __dbt_ferr(dbp, "data", data, 0)) != 0)
return (ret);
if (multi) {
if (!F_ISSET(data, DB_DBT_USERMEM)) {
__db_errx(env,
"DB_MULTIPLE/DB_MULTIPLE_KEY require DB_DBT_USERMEM be set");
return (EINVAL);
}
if (F_ISSET(key, DB_DBT_PARTIAL) ||
F_ISSET(data, DB_DBT_PARTIAL)) {
__db_errx(env,
"DB_MULTIPLE/DB_MULTIPLE_KEY do not support DB_DBT_PARTIAL");
return (EINVAL);
}
if (data->ulen < 1024 ||
data->ulen < dbp->pgsize || data->ulen % 1024 != 0) {
__db_errx(env, "%s%s",
"DB_MULTIPLE/DB_MULTIPLE_KEY buffers must be ",
"aligned, at least page size and multiples of 1KB");
return (EINVAL);
}
}
if (!IS_INITIALIZED(dbc) && (flags == DB_CURRENT ||
flags == DB_GET_RECNO ||
flags == DB_NEXT_DUP || flags == DB_PREV_DUP))
return (__db_curinval(env));
if (LF_ISSET(DB_RMW) &&
(ret = __db_check_txn(dbp, dbc->txn, dbc->locker, 0)) != 0)
return (ret);
return (0);
}
int
__db_secondary_close_pp(dbp, flags)
DB *dbp;
u_int32_t flags;
{
DB_THREAD_INFO *ip;
ENV *env;
int handle_check, ret, t_ret;
env = dbp->env;
ret = 0;
if (flags != 0 && flags != DB_NOSYNC)
ret = __db_ferr(env, "DB->close", 0);
ENV_ENTER(env, ip);
handle_check = IS_ENV_REPLICATED(env);
if (handle_check && (t_ret = __db_rep_enter(dbp, 0, 0, 0)) != 0) {
handle_check = 0;
if (ret == 0)
ret = t_ret;
}
if ((t_ret = __db_secondary_close(dbp, flags)) != 0 && ret == 0)
ret = t_ret;
if (handle_check && (t_ret = __env_db_rep_exit(env)) != 0 && ret == 0)
ret = t_ret;
ENV_LEAVE(env, ip);
return (ret);
}
int
__dbc_pget_pp(dbc, skey, pkey, data, flags)
DBC *dbc;
DBT *skey, *pkey, *data;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ignore_lease, ret;
dbp = dbc->dbp;
env = dbp->env;
ignore_lease = LF_ISSET(DB_IGNORE_LEASE) ? 1 : 0;
LF_CLR(DB_IGNORE_LEASE);
if ((ret = __dbc_pget_arg(dbc, pkey, flags)) != 0 ||
(ret = __dbc_get_arg(dbc, skey, data, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
ret = __dbc_pget(dbc, skey, pkey, data, flags);
if (ret == 0 &&
IS_REP_MASTER(env) && IS_USING_LEASES(env) && !ignore_lease)
ret = __rep_lease_check(env, 1);
ENV_LEAVE(env, ip);
__dbt_userfree(env, skey, pkey, data);
return (ret);
}
static int
__dbc_pget_arg(dbc, pkey, flags)
DBC *dbc;
DBT *pkey;
u_int32_t flags;
{
DB *dbp;
ENV *env;
int ret;
dbp = dbc->dbp;
env = dbp->env;
if (!F_ISSET(dbp, DB_AM_SECONDARY)) {
__db_errx(env,
"DBcursor->pget may only be used on secondary indices");
return (EINVAL);
}
if (LF_ISSET(DB_MULTIPLE | DB_MULTIPLE_KEY)) {
__db_errx(env,
"DB_MULTIPLE and DB_MULTIPLE_KEY may not be used on secondary indices");
return (EINVAL);
}
switch (LF_ISSET(DB_OPFLAGS_MASK)) {
case DB_CONSUME:
case DB_CONSUME_WAIT:
return (__db_ferr(env, "DBcursor->pget", 0));
case DB_GET_BOTH:
case DB_GET_BOTH_RANGE:
if (pkey == NULL) {
__db_errx(env,
"%s requires both a secondary and a primary key",
LF_ISSET(DB_GET_BOTH) ?
"DB_GET_BOTH" : "DB_GET_BOTH_RANGE");
return (EINVAL);
}
if ((ret = __dbt_usercopy(env, pkey)) != 0)
return (ret);
break;
default:
break;
}
if (pkey != NULL &&
(ret = __dbt_ferr(dbp, "primary key", pkey, 0)) != 0)
return (ret);
if (pkey == NULL && (flags & DB_OPFLAGS_MASK) == DB_GET_BOTH) {
__db_errx(env,
"DB_GET_BOTH on a secondary index requires a primary key");
return (EINVAL);
}
return (0);
}
int
__dbc_put_pp(dbc, key, data, flags)
DBC *dbc;
DBT *key, *data;
u_int32_t flags;
{
DB *dbp;
DB_THREAD_INFO *ip;
ENV *env;
int ret;
dbp = dbc->dbp;
env = dbp->env;
if ((ret = __dbc_put_arg(dbc, key, data, flags)) != 0)
return (ret);
ENV_ENTER(env, ip);
if ((ret = __db_check_txn(dbp, dbc->txn, dbc->locker, 0)) != 0)
goto err;
DEBUG_LWRITE(dbc, dbc->txn, "DBcursor->put",
flags == DB_KEYFIRST || flags == DB_KEYLAST ||
flags == DB_NODUPDATA || flags == DB_UPDATE_SECONDARY ?
key : NULL, data, flags);
ret = __dbc_put(dbc, key, data, flags);
err: ENV_LEAVE(env, ip);
__dbt_userfree(env, key, NULL, data);
return (ret);
}
static int
__dbc_put_arg(dbc, key, data, flags)
DBC *dbc;
DBT *key, *data;
u_int32_t flags;
{
DB *dbp;
ENV *env;
int key_flags, ret;
dbp = dbc->dbp;
env = dbp->env;
key_flags = 0;
if (DB_IS_READONLY(dbp))
return (__db_rdonly(env, "DBcursor->put"));
if (F_ISSET(dbp, DB_AM_SECONDARY)) {
if (flags == DB_UPDATE_SECONDARY)
flags = DB_KEYLAST;
else {
__db_errx(env,
"DBcursor->put forbidden on secondary indices");
return (EINVAL);
}
}
if ((ret = __dbt_usercopy(env, data)) != 0)
return (ret);
switch (flags) {
case DB_AFTER:
case DB_BEFORE:
switch (dbp->type) {
case DB_BTREE:
case DB_HASH:
if (!F_ISSET(dbp, DB_AM_DUP))
goto err;
if (dbp->dup_compare != NULL)
goto err;
break;
case DB_QUEUE:
goto err;
case DB_RECNO:
if (!F_ISSET(dbp, DB_AM_RENUMBER))
goto err;
key_flags = key == NULL ? 0 : 1;
break;
case DB_UNKNOWN:
default:
goto err;
}
break;
case DB_CURRENT:
break;
case DB_NODUPDATA:
if (!F_ISSET(dbp, DB_AM_DUPSORT))
goto err;
case DB_KEYFIRST:
case DB_KEYLAST:
key_flags = 1;
if ((ret = __dbt_usercopy(env, key)) != 0)
return (ret);
break;
default:
err: return (__db_ferr(env, "DBcursor->put", 0));
}
if (key_flags && (ret = __dbt_ferr(dbp, "key", key, 0)) != 0)
return (ret);
if ((ret = __dbt_ferr(dbp, "data", data, 0)) != 0)
return (ret);
if (key_flags && F_ISSET(key, DB_DBT_PARTIAL) && key->dlen != 0)
return (__db_ferr(env, "key DBT", 0));
if (!IS_INITIALIZED(dbc) && flags != DB_KEYFIRST &&
flags != DB_KEYLAST && flags != DB_NODUPDATA)
return (__db_curinval(env));
return (0);
}
static int
__dbt_ferr(dbp, name, dbt, check_thread)
const DB *dbp;
const char *name;
const DBT *dbt;
int check_thread;
{
ENV *env;
int ret;
env = dbp->env;
if ((ret = __db_fchk(env, name, dbt->flags, DB_DBT_APPMALLOC |
DB_DBT_MALLOC | DB_DBT_DUPOK | DB_DBT_REALLOC |
DB_DBT_USERCOPY | DB_DBT_USERMEM | DB_DBT_PARTIAL)) != 0)
return (ret);
switch (F_ISSET(dbt, DB_DBT_MALLOC | DB_DBT_REALLOC |
DB_DBT_USERCOPY | DB_DBT_USERMEM)) {
case 0:
case DB_DBT_MALLOC:
case DB_DBT_REALLOC:
case DB_DBT_USERCOPY:
case DB_DBT_USERMEM:
break;
default:
return (__db_ferr(env, name, 1));
}
if (check_thread && DB_IS_THREADED(dbp) &&
!F_ISSET(dbt, DB_DBT_MALLOC | DB_DBT_REALLOC |
DB_DBT_USERCOPY | DB_DBT_USERMEM)) {
__db_errx(env,
"DB_THREAD mandates memory allocation flag on %s DBT",
name);
return (EINVAL);
}
return (0);
}
static int
__db_curinval(env)
const ENV *env;
{
__db_errx(env,
"Cursor position must be set before performing this operation");
return (EINVAL);
}
int
__db_txn_auto_init(env, ip, txnidp)
ENV *env;
DB_THREAD_INFO *ip;
DB_TXN **txnidp;
{
if (*txnidp != NULL) {
__db_errx(env,
"DB_AUTO_COMMIT may not be specified along with a transaction handle");
return (EINVAL);
}
if (!TXN_ON(env)) {
__db_errx(env,
"DB_AUTO_COMMIT may not be specified in non-transactional environment");
return (EINVAL);
}
return (__txn_begin(env, ip, NULL, txnidp, 0));
}
int
__db_txn_auto_resolve(env, txn, nosync, ret)
ENV *env;
DB_TXN *txn;
int nosync, ret;
{
int t_ret;
if (ret == 0)
return (__txn_commit(txn, nosync ? DB_TXN_NOSYNC : 0));
if ((t_ret = __txn_abort(txn)) != 0)
return (__env_panic(env, t_ret));
return (ret);
}