mdbx: extract internal check_key().

This commit is contained in:
Леонид Юрьев (Leonid Yuriev)
2025-09-04 20:13:30 +03:00
parent 3615b87b4b
commit 5d12764a8f
2 changed files with 40 additions and 32 deletions

View File

@@ -236,6 +236,38 @@ MDBX_NOTHROW_PURE_FUNCTION static inline pgno_t largechunk_npages(const MDBX_env
return bytes2pgno(env, PAGEHDRSZ - 1 + bytes) + 1; return bytes2pgno(env, PAGEHDRSZ - 1 + bytes) + 1;
} }
typedef struct alignkey {
MDBX_val key;
uint64_t intbuf;
} alignkey_t;
static inline int check_key(const MDBX_cursor *mc, const MDBX_val *key, alignkey_t *alignkey) {
/* coverity[logical_vs_bitwise] */
if (unlikely(key->iov_len < mc->clc->k.lmin ||
(key->iov_len > mc->clc->k.lmax &&
(mc->clc->k.lmin == mc->clc->k.lmax || MDBX_DEBUG || MDBX_FORCE_ASSERTIONS)))) {
cASSERT(mc, !"Invalid key-size");
return MDBX_BAD_VALSIZE;
}
alignkey->key = *key;
if (mc->tree->flags & MDBX_INTEGERKEY) {
if (alignkey->key.iov_len == 8) {
if (unlikely(7 & (uintptr_t)alignkey->key.iov_base))
/* copy instead of return error to avoid break compatibility */
alignkey->key.iov_base = bcopy_8(&alignkey->intbuf, alignkey->key.iov_base);
} else if (alignkey->key.iov_len == 4) {
if (unlikely(3 & (uintptr_t)alignkey->key.iov_base))
/* copy instead of return error to avoid break compatibility */
alignkey->key.iov_base = bcopy_4(&alignkey->intbuf, alignkey->key.iov_base);
} else {
cASSERT(mc, !"key-size is invalid for MDBX_INTEGERKEY");
return MDBX_BAD_VALSIZE;
}
}
return MDBX_SUCCESS;
}
MDBX_NOTHROW_PURE_FUNCTION static inline MDBX_val get_key(const node_t *node) { MDBX_NOTHROW_PURE_FUNCTION static inline MDBX_val get_key(const node_t *node) {
MDBX_val key; MDBX_val key;
key.iov_len = node_ks(node); key.iov_len = node_ks(node);

View File

@@ -1723,34 +1723,10 @@ fail:
__hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cursor_op op) { __hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cursor_op op) {
DKBUF_DEBUG; DKBUF_DEBUG;
csr_t ret; alignkey_t aligned;
ret.exact = false; csr_t ret = {.exact = false, .err = check_key(mc, key, &aligned)};
/* coverity[logical_vs_bitwise] */ if (unlikely(ret.err != MDBX_SUCCESS))
if (unlikely(key->iov_len < mc->clc->k.lmin ||
(key->iov_len > mc->clc->k.lmax &&
(mc->clc->k.lmin == mc->clc->k.lmax || MDBX_DEBUG || MDBX_FORCE_ASSERTIONS)))) {
cASSERT(mc, !"Invalid key-size");
ret.err = MDBX_BAD_VALSIZE;
return ret; return ret;
}
MDBX_val aligned_key = *key;
uint64_t aligned_key_buf;
if (mc->tree->flags & MDBX_INTEGERKEY) {
if (aligned_key.iov_len == 8) {
if (unlikely(7 & (uintptr_t)aligned_key.iov_base))
/* copy instead of return error to avoid break compatibility */
aligned_key.iov_base = bcopy_8(&aligned_key_buf, aligned_key.iov_base);
} else if (aligned_key.iov_len == 4) {
if (unlikely(3 & (uintptr_t)aligned_key.iov_base))
/* copy instead of return error to avoid break compatibility */
aligned_key.iov_base = bcopy_4(&aligned_key_buf, aligned_key.iov_base);
} else {
cASSERT(mc, !"key-size is invalid for MDBX_INTEGERKEY");
ret.err = MDBX_BAD_VALSIZE;
return ret;
}
}
page_t *mp; page_t *mp;
node_t *node = nullptr; node_t *node = nullptr;
@@ -1778,7 +1754,7 @@ __hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cur
nodekey = get_key(node); nodekey = get_key(node);
inner_gone(mc); inner_gone(mc);
} }
int cmp = mc->clc->k.cmp(&aligned_key, &nodekey); int cmp = mc->clc->k.cmp(&aligned.key, &nodekey);
if (unlikely(cmp == 0)) { if (unlikely(cmp == 0)) {
/* Probably happens rarely, but first node on the page was the one we wanted. */ /* Probably happens rarely, but first node on the page was the one we wanted. */
mc->ki[mc->top] = 0; mc->ki[mc->top] = 0;
@@ -1796,7 +1772,7 @@ __hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cur
node = page_node(mp, nkeys - 1); node = page_node(mp, nkeys - 1);
nodekey = get_key(node); nodekey = get_key(node);
} }
cmp = mc->clc->k.cmp(&aligned_key, &nodekey); cmp = mc->clc->k.cmp(&aligned.key, &nodekey);
if (cmp == 0) { if (cmp == 0) {
/* last node was the one we wanted */ /* last node was the one we wanted */
mc->ki[mc->top] = (indx_t)(nkeys - 1); mc->ki[mc->top] = (indx_t)(nkeys - 1);
@@ -1817,7 +1793,7 @@ __hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cur
node = page_node(mp, mc->ki[mc->top]); node = page_node(mp, mc->ki[mc->top]);
nodekey = get_key(node); nodekey = get_key(node);
} }
cmp = mc->clc->k.cmp(&aligned_key, &nodekey); cmp = mc->clc->k.cmp(&aligned.key, &nodekey);
if (cmp == 0) { if (cmp == 0) {
/* current node was the one we wanted */ /* current node was the one we wanted */
ret.exact = true; ret.exact = true;
@@ -1867,7 +1843,7 @@ __hot csr_t cursor_seek(MDBX_cursor *mc, MDBX_val *key, MDBX_val *data, MDBX_cur
cASSERT(mc, !inner_pointed(mc)); cASSERT(mc, !inner_pointed(mc));
continue_other_pages: continue_other_pages:
ret.err = tree_search(mc, &aligned_key, 0); ret.err = tree_search(mc, &aligned.key, 0);
if (unlikely(ret.err != MDBX_SUCCESS)) if (unlikely(ret.err != MDBX_SUCCESS))
return ret; return ret;
@@ -1878,7 +1854,7 @@ continue_other_pages:
search_node: search_node:
cASSERT(mc, is_pointed(mc) && !inner_pointed(mc)); cASSERT(mc, is_pointed(mc) && !inner_pointed(mc));
struct node_search_result nsr = node_search(mc, &aligned_key); struct node_search_result nsr = node_search(mc, &aligned.key);
node = nsr.node; node = nsr.node;
ret.exact = nsr.exact; ret.exact = nsr.exact;
if (!ret.exact) { if (!ret.exact) {