2011-03-19 08:56:14 +08:00
|
|
|
#define JEMALLOC_RTREE_C_
|
2010-09-06 01:35:13 +08:00
|
|
|
#include "jemalloc/internal/jemalloc_internal.h"
|
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
static unsigned
|
|
|
|
hmin(unsigned ha, unsigned hb)
|
2010-09-06 01:35:13 +08:00
|
|
|
{
|
2015-01-31 14:54:08 +08:00
|
|
|
|
|
|
|
return (ha < hb ? ha : hb);
|
|
|
|
}
|
|
|
|
|
2016-03-28 18:06:35 +08:00
|
|
|
/*
|
|
|
|
* Only the most significant bits of keys passed to rtree_{read,write}() are
|
|
|
|
* used.
|
|
|
|
*/
|
2015-01-31 14:54:08 +08:00
|
|
|
bool
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_new(rtree_t *rtree, unsigned bits)
|
2015-01-31 14:54:08 +08:00
|
|
|
{
|
|
|
|
unsigned bits_in_leaf, height, i;
|
2010-09-06 01:35:13 +08:00
|
|
|
|
2016-03-23 08:54:35 +08:00
|
|
|
assert(RTREE_HEIGHT_MAX == ((ZU(1) << (LG_SIZEOF_PTR+3)) /
|
|
|
|
RTREE_BITS_PER_LEVEL));
|
2014-01-03 08:08:28 +08:00
|
|
|
assert(bits > 0 && bits <= (sizeof(uintptr_t) << 3));
|
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
bits_in_leaf = (bits % RTREE_BITS_PER_LEVEL) == 0 ? RTREE_BITS_PER_LEVEL
|
|
|
|
: (bits % RTREE_BITS_PER_LEVEL);
|
2014-01-03 09:36:38 +08:00
|
|
|
if (bits > bits_in_leaf) {
|
2015-01-31 14:54:08 +08:00
|
|
|
height = 1 + (bits - bits_in_leaf) / RTREE_BITS_PER_LEVEL;
|
|
|
|
if ((height-1) * RTREE_BITS_PER_LEVEL + bits_in_leaf != bits)
|
2014-01-03 09:36:38 +08:00
|
|
|
height++;
|
2015-01-31 14:54:08 +08:00
|
|
|
} else
|
2014-01-03 09:36:38 +08:00
|
|
|
height = 1;
|
2015-01-31 14:54:08 +08:00
|
|
|
assert((height-1) * RTREE_BITS_PER_LEVEL + bits_in_leaf == bits);
|
|
|
|
|
|
|
|
rtree->height = height;
|
|
|
|
|
|
|
|
/* Root level. */
|
|
|
|
rtree->levels[0].subtree = NULL;
|
|
|
|
rtree->levels[0].bits = (height > 1) ? RTREE_BITS_PER_LEVEL :
|
|
|
|
bits_in_leaf;
|
|
|
|
rtree->levels[0].cumbits = rtree->levels[0].bits;
|
|
|
|
/* Interior levels. */
|
|
|
|
for (i = 1; i < height-1; i++) {
|
|
|
|
rtree->levels[i].subtree = NULL;
|
|
|
|
rtree->levels[i].bits = RTREE_BITS_PER_LEVEL;
|
|
|
|
rtree->levels[i].cumbits = rtree->levels[i-1].cumbits +
|
|
|
|
RTREE_BITS_PER_LEVEL;
|
2014-01-03 09:36:38 +08:00
|
|
|
}
|
2015-01-31 14:54:08 +08:00
|
|
|
/* Leaf level. */
|
2014-01-03 09:36:38 +08:00
|
|
|
if (height > 1) {
|
2015-01-31 14:54:08 +08:00
|
|
|
rtree->levels[height-1].subtree = NULL;
|
|
|
|
rtree->levels[height-1].bits = bits_in_leaf;
|
|
|
|
rtree->levels[height-1].cumbits = bits;
|
|
|
|
}
|
2010-09-06 01:35:13 +08:00
|
|
|
|
2016-06-03 09:43:10 +08:00
|
|
|
/* Compute lookup table to be used by rtree_[ctx_]start_level(). */
|
2015-01-31 14:54:08 +08:00
|
|
|
for (i = 0; i < RTREE_HEIGHT_MAX; i++) {
|
|
|
|
rtree->start_level[i] = hmin(RTREE_HEIGHT_MAX - 1 - i, height -
|
|
|
|
1);
|
2010-09-06 01:35:13 +08:00
|
|
|
}
|
2016-06-03 09:43:10 +08:00
|
|
|
rtree->start_level[RTREE_HEIGHT_MAX] = 0;
|
2010-09-06 01:35:13 +08:00
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
return (false);
|
2010-09-06 01:35:13 +08:00
|
|
|
}
|
2012-10-10 05:46:22 +08:00
|
|
|
|
2016-04-16 15:36:11 +08:00
|
|
|
#ifdef JEMALLOC_JET
|
|
|
|
#undef rtree_node_alloc
|
|
|
|
#define rtree_node_alloc JEMALLOC_N(rtree_node_alloc_impl)
|
|
|
|
#endif
|
|
|
|
static rtree_elm_t *
|
|
|
|
rtree_node_alloc(tsdn_t *tsdn, rtree_t *rtree, size_t nelms)
|
|
|
|
{
|
|
|
|
|
|
|
|
return ((rtree_elm_t *)base_alloc(tsdn, nelms * sizeof(rtree_elm_t)));
|
|
|
|
}
|
|
|
|
#ifdef JEMALLOC_JET
|
|
|
|
#undef rtree_node_alloc
|
|
|
|
#define rtree_node_alloc JEMALLOC_N(rtree_node_alloc)
|
|
|
|
rtree_node_alloc_t *rtree_node_alloc = JEMALLOC_N(rtree_node_alloc_impl);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef JEMALLOC_JET
|
|
|
|
#undef rtree_node_dalloc
|
|
|
|
#define rtree_node_dalloc JEMALLOC_N(rtree_node_dalloc_impl)
|
|
|
|
#endif
|
|
|
|
UNUSED static void
|
|
|
|
rtree_node_dalloc(tsdn_t *tsdn, rtree_t *rtree, rtree_elm_t *node)
|
|
|
|
{
|
|
|
|
|
|
|
|
/* Nodes are never deleted during normal operation. */
|
|
|
|
not_reached();
|
|
|
|
}
|
|
|
|
#ifdef JEMALLOC_JET
|
|
|
|
#undef rtree_node_dalloc
|
|
|
|
#define rtree_node_dalloc JEMALLOC_N(rtree_node_dalloc)
|
|
|
|
rtree_node_dalloc_t *rtree_node_dalloc = JEMALLOC_N(rtree_node_dalloc_impl);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef JEMALLOC_JET
|
2014-01-03 08:08:28 +08:00
|
|
|
static void
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_delete_subtree(tsdn_t *tsdn, rtree_t *rtree, rtree_elm_t *node,
|
|
|
|
unsigned level)
|
2014-01-03 08:08:28 +08:00
|
|
|
{
|
|
|
|
|
2015-03-12 14:14:50 +08:00
|
|
|
if (level + 1 < rtree->height) {
|
2014-01-03 08:08:28 +08:00
|
|
|
size_t nchildren, i;
|
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
nchildren = ZU(1) << rtree->levels[level].bits;
|
2014-01-03 08:08:28 +08:00
|
|
|
for (i = 0; i < nchildren; i++) {
|
2016-03-28 18:06:35 +08:00
|
|
|
rtree_elm_t *child = node[i].child;
|
2016-04-16 15:36:11 +08:00
|
|
|
if (child != NULL) {
|
|
|
|
rtree_delete_subtree(tsdn, rtree, child, level +
|
|
|
|
1);
|
|
|
|
}
|
2014-01-03 08:08:28 +08:00
|
|
|
}
|
|
|
|
}
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_node_dalloc(tsdn, rtree, node);
|
2014-01-03 08:08:28 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_delete(tsdn_t *tsdn, rtree_t *rtree)
|
2014-01-03 08:08:28 +08:00
|
|
|
{
|
2015-01-31 14:54:08 +08:00
|
|
|
unsigned i;
|
2014-01-03 08:08:28 +08:00
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
for (i = 0; i < rtree->height; i++) {
|
2016-03-28 18:06:35 +08:00
|
|
|
rtree_elm_t *subtree = rtree->levels[i].subtree;
|
2015-01-31 14:54:08 +08:00
|
|
|
if (subtree != NULL)
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_delete_subtree(tsdn, rtree, subtree, i);
|
2015-01-31 14:54:08 +08:00
|
|
|
}
|
2014-01-03 08:08:28 +08:00
|
|
|
}
|
2016-04-16 15:36:11 +08:00
|
|
|
#endif
|
2014-01-03 08:08:28 +08:00
|
|
|
|
2016-03-28 18:06:35 +08:00
|
|
|
static rtree_elm_t *
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_node_init(tsdn_t *tsdn, rtree_t *rtree, unsigned level,
|
|
|
|
rtree_elm_t **elmp)
|
2012-10-10 05:46:22 +08:00
|
|
|
{
|
2016-03-28 18:06:35 +08:00
|
|
|
rtree_elm_t *node;
|
2015-01-31 14:54:08 +08:00
|
|
|
|
|
|
|
if (atomic_cas_p((void **)elmp, NULL, RTREE_NODE_INITIALIZING)) {
|
2016-10-14 05:47:50 +08:00
|
|
|
spin_t spinner;
|
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
/*
|
|
|
|
* Another thread is already in the process of initializing.
|
|
|
|
* Spin-wait until initialization is complete.
|
|
|
|
*/
|
2016-10-14 05:47:50 +08:00
|
|
|
spin_init(&spinner);
|
2015-01-31 14:54:08 +08:00
|
|
|
do {
|
2016-10-14 05:47:50 +08:00
|
|
|
spin_adaptive(&spinner);
|
2015-01-31 14:54:08 +08:00
|
|
|
node = atomic_read_p((void **)elmp);
|
|
|
|
} while (node == RTREE_NODE_INITIALIZING);
|
|
|
|
} else {
|
2016-04-16 15:36:11 +08:00
|
|
|
node = rtree_node_alloc(tsdn, rtree, ZU(1) <<
|
|
|
|
rtree->levels[level].bits);
|
2015-01-31 14:54:08 +08:00
|
|
|
if (node == NULL)
|
|
|
|
return (NULL);
|
|
|
|
atomic_write_p((void **)elmp, node);
|
|
|
|
}
|
2012-10-10 05:46:22 +08:00
|
|
|
|
2015-01-31 14:54:08 +08:00
|
|
|
return (node);
|
2012-10-10 05:46:22 +08:00
|
|
|
}
|
|
|
|
|
2016-03-28 18:06:35 +08:00
|
|
|
rtree_elm_t *
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_subtree_read_hard(tsdn_t *tsdn, rtree_t *rtree, unsigned level)
|
2012-10-10 05:46:22 +08:00
|
|
|
{
|
|
|
|
|
2016-04-16 15:36:11 +08:00
|
|
|
return (rtree_node_init(tsdn, rtree, level,
|
|
|
|
&rtree->levels[level].subtree));
|
2012-10-10 05:46:22 +08:00
|
|
|
}
|
|
|
|
|
2016-03-28 18:06:35 +08:00
|
|
|
rtree_elm_t *
|
2016-04-16 15:36:11 +08:00
|
|
|
rtree_child_read_hard(tsdn_t *tsdn, rtree_t *rtree, rtree_elm_t *elm,
|
|
|
|
unsigned level)
|
2012-10-10 05:46:22 +08:00
|
|
|
{
|
|
|
|
|
2016-10-28 15:16:55 +08:00
|
|
|
return (rtree_node_init(tsdn, rtree, level+1, &elm->child));
|
2012-10-10 05:46:22 +08:00
|
|
|
}
|
2016-04-18 03:55:10 +08:00
|
|
|
|
|
|
|
static int
|
|
|
|
rtree_elm_witness_comp(const witness_t *a, void *oa, const witness_t *b,
|
|
|
|
void *ob)
|
|
|
|
{
|
|
|
|
uintptr_t ka = (uintptr_t)oa;
|
|
|
|
uintptr_t kb = (uintptr_t)ob;
|
|
|
|
|
|
|
|
assert(ka != 0);
|
|
|
|
assert(kb != 0);
|
|
|
|
|
|
|
|
return ((ka > kb) - (ka < kb));
|
|
|
|
}
|
|
|
|
|
|
|
|
static witness_t *
|
|
|
|
rtree_elm_witness_alloc(tsd_t *tsd, uintptr_t key, const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
witness_t *witness;
|
|
|
|
size_t i;
|
|
|
|
rtree_elm_witness_tsd_t *witnesses = tsd_rtree_elm_witnessesp_get(tsd);
|
|
|
|
|
|
|
|
/* Iterate over entire array to detect double allocation attempts. */
|
|
|
|
witness = NULL;
|
|
|
|
for (i = 0; i < sizeof(rtree_elm_witness_tsd_t) / sizeof(witness_t);
|
|
|
|
i++) {
|
|
|
|
rtree_elm_witness_t *rew = &witnesses->witnesses[i];
|
|
|
|
|
|
|
|
assert(rew->elm != elm);
|
|
|
|
if (rew->elm == NULL && witness == NULL) {
|
|
|
|
rew->elm = elm;
|
|
|
|
witness = &rew->witness;
|
|
|
|
witness_init(witness, "rtree_elm",
|
|
|
|
WITNESS_RANK_RTREE_ELM, rtree_elm_witness_comp,
|
|
|
|
(void *)key);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
assert(witness != NULL);
|
|
|
|
return (witness);
|
|
|
|
}
|
|
|
|
|
|
|
|
static witness_t *
|
|
|
|
rtree_elm_witness_find(tsd_t *tsd, const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
size_t i;
|
|
|
|
rtree_elm_witness_tsd_t *witnesses = tsd_rtree_elm_witnessesp_get(tsd);
|
|
|
|
|
|
|
|
for (i = 0; i < sizeof(rtree_elm_witness_tsd_t) / sizeof(witness_t);
|
|
|
|
i++) {
|
|
|
|
rtree_elm_witness_t *rew = &witnesses->witnesses[i];
|
|
|
|
|
|
|
|
if (rew->elm == elm)
|
|
|
|
return (&rew->witness);
|
|
|
|
}
|
|
|
|
not_reached();
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
rtree_elm_witness_dalloc(tsd_t *tsd, witness_t *witness, const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
size_t i;
|
|
|
|
rtree_elm_witness_tsd_t *witnesses = tsd_rtree_elm_witnessesp_get(tsd);
|
|
|
|
|
|
|
|
for (i = 0; i < sizeof(rtree_elm_witness_tsd_t) / sizeof(witness_t);
|
|
|
|
i++) {
|
|
|
|
rtree_elm_witness_t *rew = &witnesses->witnesses[i];
|
|
|
|
|
|
|
|
if (rew->elm == elm) {
|
|
|
|
rew->elm = NULL;
|
|
|
|
witness_init(&rew->witness, "rtree_elm",
|
|
|
|
WITNESS_RANK_RTREE_ELM, rtree_elm_witness_comp,
|
|
|
|
NULL);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
not_reached();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rtree_elm_witness_acquire(tsdn_t *tsdn, const rtree_t *rtree, uintptr_t key,
|
|
|
|
const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
witness_t *witness;
|
|
|
|
|
|
|
|
if (tsdn_null(tsdn))
|
|
|
|
return;
|
|
|
|
|
|
|
|
witness = rtree_elm_witness_alloc(tsdn_tsd(tsdn), key, elm);
|
|
|
|
witness_lock(tsdn, witness);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rtree_elm_witness_access(tsdn_t *tsdn, const rtree_t *rtree,
|
|
|
|
const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
witness_t *witness;
|
|
|
|
|
|
|
|
if (tsdn_null(tsdn))
|
|
|
|
return;
|
|
|
|
|
|
|
|
witness = rtree_elm_witness_find(tsdn_tsd(tsdn), elm);
|
|
|
|
witness_assert_owner(tsdn, witness);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
rtree_elm_witness_release(tsdn_t *tsdn, const rtree_t *rtree,
|
|
|
|
const rtree_elm_t *elm)
|
|
|
|
{
|
|
|
|
witness_t *witness;
|
|
|
|
|
|
|
|
if (tsdn_null(tsdn))
|
|
|
|
return;
|
|
|
|
|
|
|
|
witness = rtree_elm_witness_find(tsdn_tsd(tsdn), elm);
|
|
|
|
witness_unlock(tsdn, witness);
|
|
|
|
rtree_elm_witness_dalloc(tsdn_tsd(tsdn), witness, elm);
|
|
|
|
}
|