2016-06-09 13:54:46 +08:00
|
|
|
|
/*
|
2017-01-01 19:56:00 +08:00
|
|
|
|
* Copyright 2015-2017 Leonid Yuriev <leo@yuriev.ru>.
|
|
|
|
|
* Copyright 2015,2016 Peter-Service R&D LLC.
|
2016-09-27 18:42:15 +08:00
|
|
|
|
* All rights reserved.
|
2016-06-09 18:16:28 +08:00
|
|
|
|
*
|
2016-09-27 18:42:15 +08:00
|
|
|
|
* Redistribution and use in source and binary forms, with or without
|
|
|
|
|
* modification, are permitted only as authorized by the OpenLDAP
|
|
|
|
|
* Public License.
|
2016-06-09 18:16:28 +08:00
|
|
|
|
*
|
2016-09-27 18:42:15 +08:00
|
|
|
|
* A copy of this license is available in the file LICENSE in the
|
|
|
|
|
* top-level directory of the distribution or, alternatively, at
|
|
|
|
|
* <http://www.OpenLDAP.org/license.html>.
|
2016-06-09 18:16:28 +08:00
|
|
|
|
*/
|
2016-06-09 13:54:46 +08:00
|
|
|
|
|
|
|
|
|
#include "mdbx.h"
|
|
|
|
|
|
|
|
|
|
int mdb_runtime_flags = MDBX_DBG_PRINT
|
|
|
|
|
#if MDB_DEBUG
|
|
|
|
|
| MDBX_DBG_ASSERT
|
|
|
|
|
#endif
|
|
|
|
|
#if MDB_DEBUG > 1
|
|
|
|
|
| MDBX_DBG_TRACE
|
|
|
|
|
#endif
|
|
|
|
|
#if MDB_DEBUG > 2
|
|
|
|
|
| MDBX_DBG_AUDIT
|
|
|
|
|
#endif
|
|
|
|
|
#if MDB_DEBUG > 3
|
|
|
|
|
| MDBX_DBG_EXTRA
|
|
|
|
|
#endif
|
|
|
|
|
;
|
|
|
|
|
|
|
|
|
|
static MDBX_debug_func *mdb_debug_logger;
|
|
|
|
|
|
|
|
|
|
int mdbx_setup_debug(int flags, MDBX_debug_func* logger, long edge_txn);
|
|
|
|
|
|
|
|
|
|
#include "mdb.c"
|
|
|
|
|
|
|
|
|
|
int __cold
|
|
|
|
|
mdbx_setup_debug(int flags, MDBX_debug_func* logger, long edge_txn) {
|
|
|
|
|
unsigned ret = mdb_runtime_flags;
|
|
|
|
|
if (flags != (int) MDBX_DBG_DNT)
|
|
|
|
|
mdb_runtime_flags = flags;
|
|
|
|
|
if (logger != (MDBX_debug_func*) MDBX_DBG_DNT)
|
|
|
|
|
mdb_debug_logger = logger;
|
2016-08-23 19:56:06 +08:00
|
|
|
|
if (edge_txn != (long) MDBX_DBG_DNT) {
|
2016-06-09 13:54:46 +08:00
|
|
|
|
#if MDB_DEBUG
|
|
|
|
|
mdb_debug_edge = edge_txn;
|
|
|
|
|
#endif
|
2016-08-23 19:56:06 +08:00
|
|
|
|
}
|
2016-06-09 13:54:46 +08:00
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static txnid_t __cold
|
|
|
|
|
mdbx_oomkick(MDB_env *env, txnid_t oldest)
|
|
|
|
|
{
|
|
|
|
|
int retry;
|
|
|
|
|
txnid_t snap;
|
|
|
|
|
mdb_debug("DB size maxed out");
|
|
|
|
|
|
|
|
|
|
for(retry = 0; ; ++retry) {
|
|
|
|
|
int reader;
|
|
|
|
|
|
|
|
|
|
if (mdb_reader_check(env, NULL))
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
snap = mdb_find_oldest(env, &reader);
|
|
|
|
|
if (oldest < snap || reader < 0) {
|
|
|
|
|
if (retry && env->me_oom_func) {
|
|
|
|
|
/* LY: notify end of oom-loop */
|
|
|
|
|
env->me_oom_func(env, 0, 0, oldest, snap - oldest, -retry);
|
|
|
|
|
}
|
|
|
|
|
return snap;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
MDB_reader *r;
|
|
|
|
|
pthread_t tid;
|
|
|
|
|
pid_t pid;
|
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
|
|
if (!env->me_oom_func)
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
r = &env->me_txns->mti_readers[ reader ];
|
|
|
|
|
pid = r->mr_pid;
|
|
|
|
|
tid = r->mr_tid;
|
|
|
|
|
if (r->mr_txnid != oldest || pid <= 0)
|
|
|
|
|
continue;
|
|
|
|
|
|
|
|
|
|
rc = env->me_oom_func(env, pid, (void*) tid, oldest,
|
|
|
|
|
mdb_meta_head_w(env)->mm_txnid - oldest, retry);
|
|
|
|
|
if (rc < 0)
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
if (rc) {
|
|
|
|
|
r->mr_txnid = ~(txnid_t)0;
|
|
|
|
|
if (rc > 1) {
|
|
|
|
|
r->mr_tid = 0;
|
|
|
|
|
r->mr_pid = 0;
|
|
|
|
|
mdbx_coherent_barrier();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (retry && env->me_oom_func) {
|
|
|
|
|
/* LY: notify end of oom-loop */
|
|
|
|
|
env->me_oom_func(env, 0, 0, oldest, 0, -retry);
|
|
|
|
|
}
|
|
|
|
|
return mdb_find_oldest(env, NULL);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int __cold
|
|
|
|
|
mdbx_env_set_syncbytes(MDB_env *env, size_t bytes)
|
|
|
|
|
{
|
|
|
|
|
if (unlikely(!env))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if(unlikely(env->me_signature != MDBX_ME_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
env->me_sync_threshold = bytes;
|
2017-01-31 01:13:56 +08:00
|
|
|
|
return env->me_map ? mdb_env_sync(env, 0) : MDB_SUCCESS;
|
2016-06-09 13:54:46 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void __cold
|
|
|
|
|
mdbx_env_set_oomfunc(MDB_env *env, MDBX_oom_func *oomfunc)
|
|
|
|
|
{
|
|
|
|
|
if (likely(env && env->me_signature == MDBX_ME_SIGNATURE))
|
|
|
|
|
env->me_oom_func = oomfunc;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
MDBX_oom_func* __cold
|
|
|
|
|
mdbx_env_get_oomfunc(MDB_env *env)
|
|
|
|
|
{
|
|
|
|
|
return likely(env && env->me_signature == MDBX_ME_SIGNATURE)
|
|
|
|
|
? env->me_oom_func : NULL;
|
|
|
|
|
}
|
|
|
|
|
|
2016-11-22 00:32:12 +08:00
|
|
|
|
ATTRIBUTE_NO_SANITIZE_THREAD /* LY: avoid tsan-trap by me_txn, mm_last_pg and mt_next_pgno */
|
|
|
|
|
int mdbx_txn_straggler(MDB_txn *txn, int *percent)
|
|
|
|
|
{
|
|
|
|
|
MDB_env *env;
|
|
|
|
|
MDB_meta *meta;
|
|
|
|
|
txnid_t lag;
|
|
|
|
|
|
|
|
|
|
if(unlikely(!txn))
|
|
|
|
|
return -EINVAL;
|
|
|
|
|
|
|
|
|
|
if(unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
if (unlikely(! txn->mt_u.reader))
|
|
|
|
|
return -1;
|
|
|
|
|
|
|
|
|
|
env = txn->mt_env;
|
|
|
|
|
meta = mdb_meta_head_r(env);
|
|
|
|
|
if (percent) {
|
|
|
|
|
size_t maxpg = env->me_maxpg;
|
|
|
|
|
size_t last = meta->mm_last_pg + 1;
|
|
|
|
|
if (env->me_txn)
|
|
|
|
|
last = env->me_txn0->mt_next_pgno;
|
|
|
|
|
*percent = (last * 100ull + maxpg / 2) / maxpg;
|
|
|
|
|
}
|
|
|
|
|
lag = meta->mm_txnid - txn->mt_u.reader->mr_txnid;
|
|
|
|
|
return (0 > (long) lag) ? ~0u >> 1: lag;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
typedef struct mdb_walk_ctx {
|
2016-06-09 13:54:46 +08:00
|
|
|
|
MDB_txn *mw_txn;
|
|
|
|
|
void *mw_user;
|
|
|
|
|
MDBX_pgvisitor_func *mw_visitor;
|
2016-11-22 00:32:12 +08:00
|
|
|
|
} mdb_walk_ctx_t;
|
2016-06-09 13:54:46 +08:00
|
|
|
|
|
|
|
|
|
/** Depth-first tree traversal. */
|
|
|
|
|
static int __cold
|
|
|
|
|
mdb_env_walk(mdb_walk_ctx_t *ctx, const char* dbi, pgno_t pg, int flags, int deep)
|
|
|
|
|
{
|
|
|
|
|
MDB_page *mp;
|
|
|
|
|
int rc, i, nkeys;
|
|
|
|
|
unsigned header_size, unused_size, payload_size, align_bytes;
|
|
|
|
|
const char* type;
|
|
|
|
|
|
|
|
|
|
if (pg == P_INVALID)
|
2016-11-21 23:38:26 +08:00
|
|
|
|
return MDB_SUCCESS; /* empty db */
|
2016-06-09 13:54:46 +08:00
|
|
|
|
|
2015-08-24 02:33:02 +08:00
|
|
|
|
MDB_cursor mc;
|
|
|
|
|
memset(&mc, 0, sizeof(mc));
|
|
|
|
|
mc.mc_snum = 1;
|
|
|
|
|
mc.mc_txn = ctx->mw_txn;
|
|
|
|
|
|
|
|
|
|
rc = mdb_page_get(&mc, pg, &mp, NULL);
|
2016-06-09 13:54:46 +08:00
|
|
|
|
if (rc)
|
|
|
|
|
return rc;
|
|
|
|
|
if (pg != mp->mp_p.p_pgno)
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
|
|
|
|
|
nkeys = NUMKEYS(mp);
|
|
|
|
|
header_size = IS_LEAF2(mp) ? PAGEHDRSZ : PAGEBASE + mp->mp_lower;
|
|
|
|
|
unused_size = SIZELEFT(mp);
|
|
|
|
|
payload_size = 0;
|
|
|
|
|
|
|
|
|
|
/* LY: Don't use mask here, e.g bitwise (P_BRANCH|P_LEAF|P_LEAF2|P_META|P_OVERFLOW|P_SUBP).
|
|
|
|
|
* Pages should not me marked dirty/loose or otherwise. */
|
|
|
|
|
switch (mp->mp_flags) {
|
|
|
|
|
case P_BRANCH:
|
|
|
|
|
type = "branch";
|
|
|
|
|
if (nkeys < 1)
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
break;
|
|
|
|
|
case P_LEAF:
|
|
|
|
|
type = "leaf";
|
|
|
|
|
break;
|
|
|
|
|
case P_LEAF|P_SUBP:
|
|
|
|
|
type = "dupsort-subleaf";
|
|
|
|
|
break;
|
|
|
|
|
case P_LEAF|P_LEAF2:
|
|
|
|
|
type = "dupfixed-leaf";
|
|
|
|
|
break;
|
|
|
|
|
case P_LEAF|P_LEAF2|P_SUBP:
|
|
|
|
|
type = "dupsort-dupfixed-subleaf";
|
|
|
|
|
break;
|
|
|
|
|
case P_META:
|
|
|
|
|
case P_OVERFLOW:
|
|
|
|
|
default:
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for (align_bytes = i = 0; i < nkeys;
|
2015-08-24 02:33:02 +08:00
|
|
|
|
align_bytes += ((payload_size + align_bytes) & 1), i++) {
|
2016-06-09 13:54:46 +08:00
|
|
|
|
MDB_node *node;
|
|
|
|
|
|
|
|
|
|
if (IS_LEAF2(mp)) {
|
|
|
|
|
/* LEAF2 pages have no mp_ptrs[] or node headers */
|
2016-08-03 04:17:54 +08:00
|
|
|
|
payload_size += mp->mp_leaf2_ksize;
|
2016-06-09 13:54:46 +08:00
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
node = NODEPTR(mp, i);
|
|
|
|
|
payload_size += NODESIZE + node->mn_ksize;
|
|
|
|
|
|
|
|
|
|
if (IS_BRANCH(mp)) {
|
|
|
|
|
rc = mdb_env_walk(ctx, dbi, NODEPGNO(node), flags, deep);
|
|
|
|
|
if (rc)
|
|
|
|
|
return rc;
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
assert(IS_LEAF(mp));
|
|
|
|
|
if (node->mn_flags & F_BIGDATA) {
|
|
|
|
|
MDB_page *omp;
|
|
|
|
|
pgno_t *opg;
|
|
|
|
|
size_t over_header, over_payload, over_unused;
|
|
|
|
|
|
|
|
|
|
payload_size += sizeof(pgno_t);
|
|
|
|
|
opg = NODEDATA(node);
|
2015-08-24 02:33:02 +08:00
|
|
|
|
rc = mdb_page_get(&mc, *opg, &omp, NULL);
|
2016-06-09 13:54:46 +08:00
|
|
|
|
if (rc)
|
|
|
|
|
return rc;
|
|
|
|
|
if (*opg != omp->mp_p.p_pgno)
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
/* LY: Don't use mask here, e.g bitwise (P_BRANCH|P_LEAF|P_LEAF2|P_META|P_OVERFLOW|P_SUBP).
|
|
|
|
|
* Pages should not me marked dirty/loose or otherwise. */
|
|
|
|
|
if (P_OVERFLOW != omp->mp_flags)
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
|
|
|
|
|
over_header = PAGEHDRSZ;
|
|
|
|
|
over_payload = NODEDSZ(node);
|
|
|
|
|
over_unused = omp->mp_pages * ctx->mw_txn->mt_env->me_psize
|
|
|
|
|
- over_payload - over_header;
|
|
|
|
|
|
|
|
|
|
rc = ctx->mw_visitor(*opg, omp->mp_pages, ctx->mw_user, dbi,
|
|
|
|
|
"overflow-data", 1, over_payload, over_header, over_unused);
|
|
|
|
|
if (rc)
|
|
|
|
|
return rc;
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
payload_size += NODEDSZ(node);
|
|
|
|
|
if (node->mn_flags & F_SUBDATA) {
|
|
|
|
|
MDB_db *db = NODEDATA(node);
|
|
|
|
|
char* name = NULL;
|
|
|
|
|
|
|
|
|
|
if (NODEDSZ(node) < 1)
|
|
|
|
|
return MDB_CORRUPTED;
|
|
|
|
|
if (! (node->mn_flags & F_DUPDATA)) {
|
|
|
|
|
name = NODEKEY(node);
|
|
|
|
|
int namelen = (char*) db - name;
|
|
|
|
|
name = memcpy(alloca(namelen + 1), name, namelen);
|
|
|
|
|
name[namelen] = 0;
|
|
|
|
|
}
|
|
|
|
|
rc = mdb_env_walk(ctx, (name && name[0]) ? name : dbi,
|
|
|
|
|
db->md_root, node->mn_flags & F_DUPDATA, deep + 1);
|
|
|
|
|
if (rc)
|
|
|
|
|
return rc;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return ctx->mw_visitor(mp->mp_p.p_pgno, 1, ctx->mw_user, dbi, type,
|
|
|
|
|
nkeys, payload_size, header_size, unused_size + align_bytes);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
int __cold
|
|
|
|
|
mdbx_env_pgwalk(MDB_txn *txn, MDBX_pgvisitor_func* visitor, void* user)
|
|
|
|
|
{
|
|
|
|
|
mdb_walk_ctx_t ctx;
|
|
|
|
|
int rc;
|
|
|
|
|
|
|
|
|
|
if (unlikely(!txn))
|
|
|
|
|
return MDB_BAD_TXN;
|
|
|
|
|
if (unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
ctx.mw_txn = txn;
|
|
|
|
|
ctx.mw_user = user;
|
|
|
|
|
ctx.mw_visitor = visitor;
|
|
|
|
|
|
|
|
|
|
rc = visitor(0, 2, user, "lmdb", "meta", 2, sizeof(MDB_meta)*2, PAGEHDRSZ*2,
|
|
|
|
|
(txn->mt_env->me_psize - sizeof(MDB_meta) - PAGEHDRSZ) *2);
|
2016-11-21 23:38:26 +08:00
|
|
|
|
if (! rc)
|
2016-06-09 13:54:46 +08:00
|
|
|
|
rc = mdb_env_walk(&ctx, "free", txn->mt_dbs[FREE_DBI].md_root, 0, 0);
|
2016-11-21 23:38:26 +08:00
|
|
|
|
if (! rc)
|
2016-06-09 13:54:46 +08:00
|
|
|
|
rc = mdb_env_walk(&ctx, "main", txn->mt_dbs[MAIN_DBI].md_root, 0, 0);
|
|
|
|
|
if (! rc)
|
|
|
|
|
rc = visitor(P_INVALID, 0, user, NULL, NULL, 0, 0, 0, 0);
|
|
|
|
|
return rc;
|
|
|
|
|
}
|
2016-11-22 01:50:39 +08:00
|
|
|
|
|
|
|
|
|
int mdbx_canary_put(MDB_txn *txn, const mdbx_canary* canary)
|
|
|
|
|
{
|
|
|
|
|
if (unlikely(!txn))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
if (unlikely(F_ISSET(txn->mt_flags, MDB_TXN_RDONLY)))
|
|
|
|
|
return EACCES;
|
|
|
|
|
|
|
|
|
|
if (likely(canary)) {
|
|
|
|
|
txn->mt_canary.x = canary->x;
|
|
|
|
|
txn->mt_canary.y = canary->y;
|
|
|
|
|
txn->mt_canary.z = canary->z;
|
|
|
|
|
}
|
|
|
|
|
txn->mt_canary.v = txn->mt_txnid;
|
|
|
|
|
|
|
|
|
|
return MDB_SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
size_t mdbx_canary_get(MDB_txn *txn, mdbx_canary* canary)
|
|
|
|
|
{
|
|
|
|
|
if(unlikely(!txn || txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
|
|
if (likely(canary))
|
|
|
|
|
*canary = txn->mt_canary;
|
|
|
|
|
|
|
|
|
|
return txn->mt_txnid;
|
|
|
|
|
}
|
2016-12-07 01:32:09 +08:00
|
|
|
|
|
|
|
|
|
int mdbx_cursor_eof(MDB_cursor *mc)
|
|
|
|
|
{
|
|
|
|
|
if (unlikely(mc == NULL))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
2017-01-25 01:17:20 +08:00
|
|
|
|
if (unlikely(mc->mc_signature != MDBX_MC_SIGNATURE_LIVE))
|
2016-12-07 01:32:09 +08:00
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
2017-01-13 03:36:52 +08:00
|
|
|
|
if ((mc->mc_flags & C_INITIALIZED) == 0)
|
2017-01-31 01:13:56 +08:00
|
|
|
|
return MDBX_RESULT_TRUE;
|
2017-01-13 03:36:52 +08:00
|
|
|
|
|
|
|
|
|
if (mc->mc_snum == 0)
|
2017-01-31 01:13:56 +08:00
|
|
|
|
return MDBX_RESULT_TRUE;
|
2017-01-13 03:36:52 +08:00
|
|
|
|
|
|
|
|
|
if ((mc->mc_flags & C_EOF)
|
|
|
|
|
&& mc->mc_ki[mc->mc_top] >= NUMKEYS(mc->mc_pg[mc->mc_top]))
|
2017-01-31 01:13:56 +08:00
|
|
|
|
return MDBX_RESULT_TRUE;
|
2017-01-13 03:36:52 +08:00
|
|
|
|
|
2017-01-31 01:13:56 +08:00
|
|
|
|
return MDBX_RESULT_FALSE;
|
2016-12-07 01:32:09 +08:00
|
|
|
|
}
|
2016-12-22 01:32:27 +08:00
|
|
|
|
|
2016-12-23 01:54:06 +08:00
|
|
|
|
static int mdbx_is_samedata(const MDB_val* a, const MDB_val* b) {
|
2016-12-22 01:32:27 +08:00
|
|
|
|
return a->iov_len == b->iov_len
|
|
|
|
|
&& memcmp(a->iov_base, b->iov_base, a->iov_len) == 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Позволяет обновить или удалить существующую запись с получением
|
|
|
|
|
* в old_data предыдущего значения данных. При этом если new_data равен
|
|
|
|
|
* нулю, то выполняется удаление, иначе обновление/вставка.
|
|
|
|
|
*
|
|
|
|
|
* Текущее значение может находиться в уже измененной (грязной) странице.
|
|
|
|
|
* В этом случае страница будет перезаписана при обновлении, а само старое
|
|
|
|
|
* значение утрачено. Поэтому исходно в old_data должен быть передан
|
|
|
|
|
* дополнительный буфер для копирования старого значения.
|
|
|
|
|
* Если переданный буфер слишком мал, то функция вернет -1, установив
|
|
|
|
|
* old_data->iov_len в соответствующее значение.
|
|
|
|
|
*
|
|
|
|
|
* Для не-уникальных ключей также возможен второй сценарий использования,
|
|
|
|
|
* когда посредством old_data из записей с одинаковым ключом для
|
|
|
|
|
* удаления/обновления выбирается конкретная. Для выбора этого сценария
|
|
|
|
|
* во flags следует одновременно указать MDB_CURRENT и MDB_NOOVERWRITE.
|
|
|
|
|
*
|
|
|
|
|
* Функция может быть замещена соответствующими операциями с курсорами
|
|
|
|
|
* после двух доработок (TODO):
|
|
|
|
|
* - внешняя аллокация курсоров, в том числе на стеке (без malloc).
|
|
|
|
|
* - получения статуса страницы по адресу (знать о P_DIRTY).
|
|
|
|
|
*/
|
|
|
|
|
int mdbx_replace(MDB_txn *txn, MDB_dbi dbi,
|
|
|
|
|
MDB_val *key, MDB_val *new_data, MDB_val *old_data, unsigned flags)
|
|
|
|
|
{
|
|
|
|
|
MDB_cursor mc;
|
|
|
|
|
MDB_xcursor mx;
|
|
|
|
|
|
2017-01-27 04:12:31 +08:00
|
|
|
|
if (unlikely(!key || !old_data || !txn || old_data == new_data))
|
2016-12-22 01:32:27 +08:00
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
if (unlikely(old_data->iov_base == NULL && old_data->iov_len))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(new_data == NULL && !(flags & MDB_CURRENT)))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(!TXN_DBI_EXIST(txn, dbi, DB_USRVALID)))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(flags & ~(MDB_NOOVERWRITE|MDB_NODUPDATA|MDB_RESERVE|MDB_APPEND|MDB_APPENDDUP|MDB_CURRENT)))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_flags & (MDB_TXN_RDONLY|MDB_TXN_BLOCKED)))
|
|
|
|
|
return (txn->mt_flags & MDB_TXN_RDONLY) ? EACCES : MDB_BAD_TXN;
|
|
|
|
|
|
|
|
|
|
mdb_cursor_init(&mc, txn, dbi, &mx);
|
|
|
|
|
mc.mc_next = txn->mt_cursors[dbi];
|
|
|
|
|
txn->mt_cursors[dbi] = &mc;
|
|
|
|
|
|
|
|
|
|
int rc;
|
|
|
|
|
MDB_val present_key = *key;
|
|
|
|
|
if (F_ISSET(flags, MDB_CURRENT | MDB_NOOVERWRITE)
|
|
|
|
|
&& (txn->mt_dbs[dbi].md_flags & MDB_DUPSORT)) {
|
|
|
|
|
/* в old_data значение для выбора конкретного дубликата */
|
|
|
|
|
rc = mdbx_cursor_get(&mc, &present_key, old_data, MDB_GET_BOTH);
|
|
|
|
|
if (rc != MDB_SUCCESS)
|
|
|
|
|
goto bailout;
|
|
|
|
|
/* если данные совпадают, то ничего делать не надо */
|
|
|
|
|
if (new_data && mdbx_is_samedata(old_data, new_data))
|
|
|
|
|
goto bailout;
|
|
|
|
|
} else {
|
2017-01-27 08:18:51 +08:00
|
|
|
|
/* в old_data буфер для сохранения предыдущего значения */
|
2017-01-27 04:12:31 +08:00
|
|
|
|
if (unlikely(new_data && old_data->iov_base == new_data->iov_base))
|
|
|
|
|
return EINVAL;
|
2016-12-22 01:32:27 +08:00
|
|
|
|
MDB_val present_data;
|
|
|
|
|
rc = mdbx_cursor_get(&mc, &present_key, &present_data, MDB_SET_KEY);
|
|
|
|
|
if (unlikely(rc != MDB_SUCCESS)) {
|
|
|
|
|
old_data->iov_base = NULL;
|
|
|
|
|
old_data->iov_len = rc;
|
|
|
|
|
if (rc != MDB_NOTFOUND || (flags & MDB_CURRENT))
|
|
|
|
|
goto bailout;
|
|
|
|
|
} else if (flags & MDB_NOOVERWRITE) {
|
|
|
|
|
rc = MDB_KEYEXIST;
|
|
|
|
|
*old_data = present_data;
|
|
|
|
|
goto bailout;
|
|
|
|
|
} else {
|
|
|
|
|
MDB_page *page = mc.mc_pg[mc.mc_top];
|
|
|
|
|
if (txn->mt_dbs[dbi].md_flags & MDB_DUPSORT) {
|
|
|
|
|
if (flags & MDB_CURRENT) {
|
|
|
|
|
/* для не-уникальных ключей позволяем update/delete только если ключ один */
|
|
|
|
|
MDB_node *leaf = NODEPTR(page, mc.mc_ki[mc.mc_top]);
|
|
|
|
|
if (F_ISSET(leaf->mn_flags, F_DUPDATA)) {
|
|
|
|
|
mdb_tassert(txn, XCURSOR_INITED(&mc) && mc.mc_xcursor->mx_db.md_entries > 1);
|
2017-01-24 20:38:20 +08:00
|
|
|
|
rc = MDBX_EMULTIVAL;
|
2016-12-22 01:32:27 +08:00
|
|
|
|
goto bailout;
|
|
|
|
|
}
|
|
|
|
|
/* если данные совпадают, то ничего делать не надо */
|
2016-12-23 01:54:06 +08:00
|
|
|
|
if (new_data && mdbx_is_samedata(&present_data, new_data)) {
|
|
|
|
|
*old_data = *new_data;
|
2016-12-22 01:32:27 +08:00
|
|
|
|
goto bailout;
|
2016-12-23 01:54:06 +08:00
|
|
|
|
}
|
2016-12-22 01:32:27 +08:00
|
|
|
|
} else if ((flags & MDB_NODUPDATA) && mdbx_is_samedata(&present_data, new_data)) {
|
|
|
|
|
/* если данные совпадают и установлен MDB_NODUPDATA */
|
|
|
|
|
rc = MDB_KEYEXIST;
|
|
|
|
|
goto bailout;
|
|
|
|
|
}
|
|
|
|
|
} else {
|
2016-12-23 01:54:06 +08:00
|
|
|
|
/* если данные совпадают, то ничего делать не надо */
|
|
|
|
|
if (new_data && mdbx_is_samedata(&present_data, new_data)) {
|
|
|
|
|
*old_data = *new_data;
|
|
|
|
|
goto bailout;
|
|
|
|
|
}
|
2016-12-22 01:32:27 +08:00
|
|
|
|
flags |= MDB_CURRENT;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (page->mp_flags & P_DIRTY) {
|
|
|
|
|
if (unlikely(old_data->iov_len < present_data.iov_len)) {
|
|
|
|
|
old_data->iov_base = NULL;
|
|
|
|
|
old_data->iov_len = present_data.iov_len;
|
|
|
|
|
rc = -1;
|
|
|
|
|
goto bailout;
|
|
|
|
|
}
|
|
|
|
|
memcpy(old_data->iov_base, present_data.iov_base, present_data.iov_len);
|
|
|
|
|
old_data->iov_len = present_data.iov_len;
|
|
|
|
|
} else {
|
|
|
|
|
*old_data = present_data;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (likely(new_data))
|
|
|
|
|
rc = mdbx_cursor_put(&mc, key, new_data, flags);
|
|
|
|
|
else
|
|
|
|
|
rc = mdbx_cursor_del(&mc, 0);
|
|
|
|
|
|
|
|
|
|
bailout:
|
|
|
|
|
txn->mt_cursors[dbi] = mc.mc_next;
|
|
|
|
|
return rc;
|
|
|
|
|
}
|
2017-01-19 22:11:30 +08:00
|
|
|
|
|
|
|
|
|
int
|
|
|
|
|
mdbx_get_ex(MDB_txn *txn, MDB_dbi dbi,
|
|
|
|
|
MDB_val *key, MDB_val *data, int* values_count)
|
|
|
|
|
{
|
|
|
|
|
DKBUF;
|
|
|
|
|
mdb_debug("===> get db %u key [%s]", dbi, DKEY(key));
|
|
|
|
|
|
|
|
|
|
if (unlikely(!key || !data || !txn))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
if (unlikely(!TXN_DBI_EXIST(txn, dbi, DB_USRVALID)))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_flags & MDB_TXN_BLOCKED))
|
|
|
|
|
return MDB_BAD_TXN;
|
|
|
|
|
|
|
|
|
|
MDB_cursor mc;
|
|
|
|
|
MDB_xcursor mx;
|
|
|
|
|
mdb_cursor_init(&mc, txn, dbi, &mx);
|
|
|
|
|
|
|
|
|
|
int exact = 0;
|
2017-01-27 08:18:51 +08:00
|
|
|
|
int rc = mdb_cursor_set(&mc, key, data, MDB_SET_KEY, &exact);
|
2017-01-19 22:11:30 +08:00
|
|
|
|
if (unlikely(rc != MDB_SUCCESS)) {
|
|
|
|
|
if (rc == MDB_NOTFOUND && values_count)
|
|
|
|
|
*values_count = 0;
|
|
|
|
|
return rc;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (values_count) {
|
|
|
|
|
if (mc.mc_xcursor == NULL)
|
|
|
|
|
*values_count = 1;
|
|
|
|
|
else {
|
|
|
|
|
MDB_page *mp = mc.mc_pg[mc.mc_top];
|
|
|
|
|
if (IS_LEAF2(mp))
|
|
|
|
|
*values_count = 1;
|
|
|
|
|
else {
|
|
|
|
|
MDB_node *leaf = NODEPTR(mp, mc.mc_ki[mc.mc_top]);
|
|
|
|
|
if (!F_ISSET(leaf->mn_flags, F_DUPDATA))
|
|
|
|
|
*values_count = 1;
|
|
|
|
|
else {
|
|
|
|
|
mdb_tassert(txn, mc.mc_xcursor == &mx);
|
2017-01-27 08:18:51 +08:00
|
|
|
|
mdb_tassert(txn, mx.mx_cursor.mc_flags & C_INITIALIZED);
|
2017-01-19 22:11:30 +08:00
|
|
|
|
*values_count = mx.mx_db.md_entries;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2017-01-27 08:18:51 +08:00
|
|
|
|
}
|
|
|
|
|
return MDB_SUCCESS;
|
2017-01-19 22:11:30 +08:00
|
|
|
|
}
|
2017-01-31 01:16:12 +08:00
|
|
|
|
|
|
|
|
|
/* Функция сообщает находится ли указанный адрес в "грязной" странице у
|
|
|
|
|
* заданной пишущей транзакции. В конечном счете это позволяет избавиться от
|
|
|
|
|
* лишнего копирования данных из НЕ-грязных страниц.
|
|
|
|
|
*
|
|
|
|
|
* "Грязные" страницы - это те, которые уже были изменены в ходе пишущей
|
|
|
|
|
* транзакции. Соответственно, какие-либо дальнейшие изменения могут привести
|
|
|
|
|
* к перезаписи таких страниц. Поэтому все функции, выполняющие изменения, в
|
|
|
|
|
* качестве аргументов НЕ должны получать указатели на данные в таких
|
|
|
|
|
* страницах. В свою очередь "НЕ грязные" страницы перед модификацией будут
|
|
|
|
|
* скопированы.
|
|
|
|
|
*
|
|
|
|
|
* Другими словами, данные из "грязных" страниц должны быть либо скопированы
|
|
|
|
|
* перед передачей в качестве аргументов для дальнейших модификаций, либо
|
|
|
|
|
* отвергнуты на стадии проверки корректности аргументов.
|
|
|
|
|
*
|
|
|
|
|
* Таким образом, функция позволяет как избавится от лишнего копирования,
|
|
|
|
|
* так и выполнить более полную проверку аргументов.
|
|
|
|
|
*
|
|
|
|
|
* ВАЖНО: Передаваемый указатель должен указывать на начало данных. Только
|
|
|
|
|
* так гарантируется что актуальный заголовок страницы будет физически
|
|
|
|
|
* расположен в той-же странице памяти, в том числе для многостраничных
|
|
|
|
|
* P_OVERFLOW страниц с длинными данными. */
|
|
|
|
|
int mdbx_is_dirty(const MDB_txn *txn, const void* ptr)
|
|
|
|
|
{
|
|
|
|
|
if (unlikely(!txn))
|
|
|
|
|
return EINVAL;
|
|
|
|
|
|
|
|
|
|
if(unlikely(txn->mt_signature != MDBX_MT_SIGNATURE))
|
|
|
|
|
return MDB_VERSION_MISMATCH;
|
|
|
|
|
|
|
|
|
|
if (unlikely(txn->mt_flags & MDB_TXN_RDONLY))
|
|
|
|
|
return MDB_BAD_TXN;
|
|
|
|
|
|
|
|
|
|
const MDB_env *env = txn->mt_env;
|
|
|
|
|
const uintptr_t mask = ~(uintptr_t) (env->me_psize - 1);
|
|
|
|
|
const MDB_page *page = (const MDB_page *) ((uintptr_t) ptr & mask);
|
|
|
|
|
|
|
|
|
|
/* LY: Тут не всё хорошо с абсолютной достоверностью результата,
|
|
|
|
|
* так как флажок P_DIRTY в LMDB может означать не совсем то,
|
|
|
|
|
* что было исходно задумано, детали см в логике кода mdb_page_touch().
|
|
|
|
|
*
|
|
|
|
|
* Более того, в режиме БЕЗ WRITEMAP грязные страницы выделяются через
|
|
|
|
|
* malloc(), т.е. находятся вне mmap-диаппазона.
|
|
|
|
|
*
|
|
|
|
|
* Тем не менее, однозначно страница "не грязная" если:
|
|
|
|
|
* - адрес находится внутри mmap-диаппазона и в заголовке страницы
|
|
|
|
|
* нет флажка P_DIRTY, то однозначно страница "не грязная".
|
|
|
|
|
* - адрес вне mmap-диаппазона и его нет среди списка "грязных" страниц.
|
|
|
|
|
*/
|
|
|
|
|
if (env->me_map < (char*) page) {
|
|
|
|
|
const size_t used_size = env->me_psize * txn->mt_next_pgno;
|
|
|
|
|
if (env->me_map + used_size > (char*) page) {
|
|
|
|
|
/* страница внутри диапазона */
|
|
|
|
|
if (page->mp_flags & P_DIRTY)
|
|
|
|
|
return MDBX_RESULT_TRUE;
|
|
|
|
|
return MDBX_RESULT_FALSE;
|
|
|
|
|
}
|
|
|
|
|
/* Гипотетически здесь возможна ситуация, когда указатель адресует что-то
|
|
|
|
|
* в пределах mmap, но за границей распределенных страниц. Это тяжелая
|
|
|
|
|
* ошибка, которой не возможно добиться без каких-то мега-нарушений.
|
|
|
|
|
* Поэтому не проверяем этот случай кроме как assert-ом, ибо бестолку. */
|
|
|
|
|
mdb_tassert(txn, env->me_map + env->me_mapsize > (char*) page);
|
|
|
|
|
}
|
|
|
|
|
/* Страница вне mmap-диаппазона */
|
|
|
|
|
|
|
|
|
|
if (env->me_flags & MDB_WRITEMAP)
|
|
|
|
|
/* Если MDB_WRITEMAP, то результат уже ясен. */
|
|
|
|
|
return MDBX_RESULT_FALSE;
|
|
|
|
|
|
|
|
|
|
/* Смотрим список грязных страниц у заданной транзакции. */
|
|
|
|
|
MDB_ID2 *list = txn->mt_u.dirty_list;
|
|
|
|
|
if (list) {
|
|
|
|
|
unsigned i, n = list[0].mid;
|
|
|
|
|
for (i = 1; i <= n; i++) {
|
|
|
|
|
const MDB_page *dirty = list[i].mptr;
|
|
|
|
|
if (dirty == page)
|
|
|
|
|
return MDBX_RESULT_TRUE;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* При вложенных транзакциях, страница может быть в dirty-списке
|
|
|
|
|
* родительской транзакции, но в этом случае она будет скопирована перед
|
|
|
|
|
* изменением в текущей транзакции, т.е. относительно заданной транзакции
|
|
|
|
|
* проверяемый адрес "не грязный". */
|
|
|
|
|
return MDBX_RESULT_FALSE;
|
|
|
|
|
}
|
2017-01-31 07:47:54 +08:00
|
|
|
|
|
|
|
|
|
int mdbx_dbi_open_ex(MDB_txn *txn, const char *name, unsigned flags,
|
|
|
|
|
MDB_dbi *pdbi, MDB_cmp_func *keycmp, MDB_cmp_func *datacmp)
|
|
|
|
|
{
|
|
|
|
|
int rc = mdbx_dbi_open(txn, name, flags, pdbi);
|
|
|
|
|
if (likely(rc == MDB_SUCCESS)) {
|
|
|
|
|
MDB_dbi dbi = *pdbi;
|
|
|
|
|
unsigned flags = txn->mt_dbs[dbi].md_flags;
|
|
|
|
|
txn->mt_dbxs[dbi].md_cmp = keycmp ? keycmp : mdbx_default_keycmp(flags);
|
|
|
|
|
txn->mt_dbxs[dbi].md_dcmp = datacmp ? datacmp : mdbx_default_datacmp(flags);
|
|
|
|
|
}
|
|
|
|
|
return rc;
|
|
|
|
|
}
|