98483c74d6
This is a partial step towards merging the HPHP::VM namespace up into its parent. To keep it reviewable/mergeable I'm not doing everything at once here, but most of the code I've touched seems improved. I've drawn an invisible line around the jit, Unit and its cohort (Class, Func, PreClass, etc.); we'll get back to them soon.
1902 linhas
56 KiB
C++
1902 linhas
56 KiB
C++
/*
|
|
+----------------------------------------------------------------------+
|
|
| HipHop for PHP |
|
|
+----------------------------------------------------------------------+
|
|
| Copyright (c) 2010- Facebook, Inc. (http://www.facebook.com) |
|
|
+----------------------------------------------------------------------+
|
|
| This source file is subject to version 3.01 of the PHP license, |
|
|
| that is bundled with this package in the file LICENSE, and is |
|
|
| available through the world-wide-web at the following url: |
|
|
| http://www.php.net/license/3_01.txt |
|
|
| If you did not receive a copy of the PHP license and are unable to |
|
|
| obtain it through the world-wide-web, please send a note to |
|
|
| license@php.net so we can mail you a copy immediately. |
|
|
+----------------------------------------------------------------------+
|
|
*/
|
|
|
|
#define INLINE_VARIANT_HELPER 1
|
|
|
|
#include <runtime/base/array/hphp_array.h>
|
|
#include <runtime/base/array/array_init.h>
|
|
#include <runtime/base/array/array_iterator.h>
|
|
#include <runtime/base/complex_types.h>
|
|
#include <runtime/base/runtime_option.h>
|
|
#include <runtime/base/runtime_error.h>
|
|
#include <runtime/base/variable_serializer.h>
|
|
#include <runtime/base/shared/shared_map.h>
|
|
#include <util/hash.h>
|
|
#include <util/lock.h>
|
|
#include <util/alloc.h>
|
|
#include <util/trace.h>
|
|
#include <util/util.h>
|
|
#include <runtime/base/execution_context.h>
|
|
#include <runtime/vm/member_operations.h>
|
|
#include <runtime/base/stats.h>
|
|
|
|
// If PEDANTIC is defined, extra checks are performed to ensure correct
|
|
// function even as an array approaches 2^31 elements. In practice this is
|
|
// just wasted effort though, since such an array would require on the order of
|
|
// 128 GiB of memory.
|
|
//#define PEDANTIC
|
|
|
|
namespace HPHP {
|
|
|
|
static const Trace::Module TRACEMOD = Trace::runtime;
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
/*
|
|
* Allocation of HphpArray buffers works like this: the smallest buffer
|
|
* size is allocated inline in HphpArray. Larger buffer sizes are smart
|
|
* allocated or malloc-allocated depending on whether the array itself
|
|
* was smart-allocated or not. (nonSmartCopy() is used to create static
|
|
* arrays). HphpArray::m_allocMode tracks the state as it progresses:
|
|
*
|
|
* kInline -> kSmart, or
|
|
* -> kMalloc
|
|
*
|
|
* Hashtables never shrink, so the allocMode Never goes backwards.
|
|
* If an array is pre-sized, we might skip directly to kSmart or kMalloc.
|
|
* If an array is created via nonSmartCopy(), we skip kSmart.
|
|
* Since kMalloc is only used for static arrays, and static arrays are
|
|
* never swept, we don't need any sweep method.
|
|
*
|
|
* For kInline, we use space in HphpArray defined as InlineSlots, which
|
|
* has enough room for slots and the hashtable. The next few larger array
|
|
* sizes use the inline space for just the hashtable, with slots allocated
|
|
* separately. Even larger tables allocate the hashtable and slots
|
|
* contiguously.
|
|
*/
|
|
IMPLEMENT_SMART_ALLOCATION(HphpArray);
|
|
|
|
//=============================================================================
|
|
// Static members.
|
|
|
|
HphpArray HphpArray::s_theEmptyArray(StaticEmptyArray);
|
|
|
|
//=============================================================================
|
|
// Helpers.
|
|
|
|
static inline size_t computeMaskFromNumElms(uint32_t numElms) {
|
|
assert(numElms <= 0x7fffffffU);
|
|
size_t lgSize = HphpArray::MinLgTableSize;
|
|
size_t maxElms = (size_t(3U)) << (lgSize-2);
|
|
assert(lgSize >= 2);
|
|
while (maxElms < numElms) {
|
|
++lgSize;
|
|
maxElms <<= 1;
|
|
}
|
|
assert(lgSize <= 32);
|
|
// return 2^lgSize - 1
|
|
return ((size_t(1U)) << lgSize) - 1;
|
|
static_assert(HphpArray::MinLgTableSize >= 2,
|
|
"lower limit for 0.75 load factor");
|
|
}
|
|
|
|
//=============================================================================
|
|
// Construction/destruction.
|
|
|
|
inline void HphpArray::init(uint size) {
|
|
m_size = 0;
|
|
m_tableMask = computeMaskFromNumElms(size);
|
|
size_t tableSize = computeTableSize(m_tableMask);
|
|
size_t maxElms = computeMaxElms(m_tableMask);
|
|
allocData(maxElms, tableSize);
|
|
initHash(m_hash, tableSize);
|
|
m_pos = ArrayData::invalid_index;
|
|
}
|
|
|
|
HphpArray::HphpArray(uint size) : ArrayData(kHphpArray),
|
|
m_lastE(ElmIndEmpty), m_data(nullptr), m_nextKI(0), m_hLoad(0) {
|
|
#ifdef PEDANTIC
|
|
if (size > 0x7fffffffU) {
|
|
raise_error("Cannot create an array with more than 2^31 - 1 elements");
|
|
}
|
|
#endif
|
|
init(size);
|
|
}
|
|
|
|
HphpArray::HphpArray(uint size, const TypedValue* values) :
|
|
ArrayData(kHphpArray), m_lastE(ElmIndEmpty), m_data(nullptr),
|
|
m_nextKI(0), m_hLoad(0)
|
|
{
|
|
#ifdef PEDANTIC
|
|
if (size > 0x7fffffffU) {
|
|
raise_error("Cannot create an array with more than 2^31 - 1 elements");
|
|
}
|
|
#endif
|
|
init(size);
|
|
assert(size <= m_tableMask + 1);
|
|
// append values by moving -- Caller assumes we update refcount. Values
|
|
// are in reverse order since they come from the stack, which grows down.
|
|
// This code is hand-specialized from nextInsert().
|
|
assert(m_size == 0 && m_hLoad == 0 && m_nextKI == 0);
|
|
ElmInd* hash = m_hash;
|
|
Elm* data = m_data;
|
|
for (uint i = 0; i < size; i++) {
|
|
assert(hash[i] == HphpArray::ElmIndEmpty);
|
|
const TypedValue& tv = values[size - i - 1];
|
|
data[i].data.m_data = tv.m_data;
|
|
data[i].data.m_type = tv.m_type;
|
|
data[i].setIntKey(i);
|
|
hash[i] = i;
|
|
}
|
|
m_size = size;
|
|
m_hLoad = size;
|
|
m_lastE = size - 1;
|
|
m_nextKI = size;
|
|
if (size > 0) m_pos = 0;
|
|
}
|
|
|
|
HphpArray::HphpArray(EmptyMode) : ArrayData(kHphpArray),
|
|
m_lastE(ElmIndEmpty), m_data(nullptr), m_nextKI(0), m_hLoad(0) {
|
|
init(0);
|
|
setStatic();
|
|
}
|
|
|
|
// Empty constructor for internal use by nonSmartCopy() and copyImpl()
|
|
HphpArray::HphpArray(CopyMode mode) :
|
|
ArrayData(kHphpArray, /*nonsmart*/ mode == kNonSmartCopy) {
|
|
}
|
|
|
|
HOT_FUNC_VM
|
|
HphpArray::~HphpArray() {
|
|
Elm* elms = m_data;
|
|
ssize_t lastE = (ssize_t)m_lastE;
|
|
for (ssize_t /*ElmInd*/ pos = 0; pos <= lastE; ++pos) {
|
|
Elm* e = &elms[pos];
|
|
if (e->data.m_type == KindOfTombstone) continue;
|
|
if (e->hasStrKey()) decRefStr(e->key);
|
|
tvRefcountedDecRef(&e->data);
|
|
}
|
|
if (m_allocMode == kSmart) {
|
|
smart_free(m_data);
|
|
} else if (m_allocMode == kMalloc) {
|
|
free(m_data);
|
|
}
|
|
}
|
|
|
|
ssize_t HphpArray::vsize() const {
|
|
assert(false && "vsize() called, but m_size should "
|
|
"never be -1 in HphpArray");
|
|
return m_size;
|
|
}
|
|
|
|
void HphpArray::dumpDebugInfo() const {
|
|
size_t maxElms = computeMaxElms(m_tableMask);
|
|
size_t tableSize = computeTableSize(m_tableMask);
|
|
|
|
fprintf(stderr,
|
|
"--- dumpDebugInfo(this=0x%08zx) ----------------------------\n",
|
|
uintptr_t(this));
|
|
fprintf(stderr, "m_data = %p\tm_hash = %p\n"
|
|
"m_tableMask = %u\tm_size = %d\tm_hLoad = %d\n"
|
|
"m_nextKI = %" PRId64 "\t\tm_lastE = %d\tm_pos = %zd\n",
|
|
m_data, m_hash, m_tableMask, m_size, m_hLoad,
|
|
m_nextKI, m_lastE, m_pos);
|
|
fprintf(stderr, "Elements:\n");
|
|
ssize_t lastE = m_lastE;
|
|
Elm* elms = m_data;
|
|
for (ssize_t /*ElmInd*/ i = 0; i <= lastE; ++i) {
|
|
if (elms[i].data.m_type < KindOfTombstone) {
|
|
Variant v = tvAsVariant(&elms[i].data);
|
|
VariableSerializer vs(VariableSerializer::DebugDump);
|
|
String s = vs.serialize(v, true);
|
|
if (elms[i].hasStrKey()) {
|
|
String k = Util::escapeStringForCPP(elms[i].key->data(),
|
|
elms[i].key->size());
|
|
fprintf(stderr, " [%3d] hash=0x%016x key=\"%s\" data=(%.*s)\n",
|
|
int(i), elms[i].hash(), k.data(), s.size()-1, s.data());
|
|
} else {
|
|
fprintf(stderr, " [%3d] ind=%" PRId64 " data.m_type=(%.*s)\n", int(i),
|
|
elms[i].ikey, s.size()-1, s.data());
|
|
}
|
|
} else {
|
|
fprintf(stderr, " [%3d] <tombstone>\n", int(i));
|
|
}
|
|
}
|
|
if (size_t(m_lastE+1) < maxElms) {
|
|
fprintf(stderr, " [%3d..%-3zd] <uninitialized>\n", m_lastE+1, maxElms-1);
|
|
}
|
|
fprintf(stderr, "Hash table:");
|
|
for (size_t i = 0; i < tableSize; ++i) {
|
|
if ((i % 8) == 0) {
|
|
fprintf(stderr, "\n [%3zd..%-3zd]", i, i+7);
|
|
}
|
|
switch (m_hash[i]) {
|
|
default: fprintf(stderr, "%12d", m_hash[i]); break;
|
|
case ElmIndTombstone: fprintf(stderr, "%12s", "<tombstone>"); break;
|
|
case ElmIndEmpty: fprintf(stderr, "%12s", "<empty>"); break;
|
|
}
|
|
}
|
|
fprintf(stderr, "\n");
|
|
fprintf(stderr,
|
|
"---------------------------------------------------------------\n");
|
|
}
|
|
|
|
//=============================================================================
|
|
// Iteration.
|
|
|
|
inline /*ElmInd*/ ssize_t HphpArray::prevElm(Elm* elms,
|
|
/*ElmInd*/ ssize_t ei) const {
|
|
assert(ei <= (ssize_t)(m_lastE+1));
|
|
while (ei > 0) {
|
|
--ei;
|
|
if (elms[ei].data.m_type < KindOfTombstone) {
|
|
return ei;
|
|
}
|
|
}
|
|
return (ssize_t)ElmIndEmpty;
|
|
}
|
|
|
|
ssize_t HphpArray::iter_begin() const {
|
|
return nextElm(m_data, ElmIndEmpty);
|
|
}
|
|
|
|
ssize_t HphpArray::iter_end() const {
|
|
return prevElm(m_data, (ssize_t)(m_lastE + 1));
|
|
}
|
|
|
|
ssize_t HphpArray::iter_advance(ssize_t pos) const {
|
|
ssize_t lastE = m_lastE;
|
|
assert(ArrayData::invalid_index == -1);
|
|
// Since lastE is always less than 2^32-1 and invalid_index == -1,
|
|
// we can save a check by doing an unsigned comparison instead
|
|
// of a signed comparison.
|
|
if (size_t(pos) < size_t(lastE) &&
|
|
m_data[pos + 1].data.m_type < KindOfTombstone) {
|
|
return pos + 1;
|
|
}
|
|
return iter_advance_helper(pos);
|
|
}
|
|
|
|
ssize_t HphpArray::iter_advance_helper(ssize_t pos) const {
|
|
Elm* elms = m_data;
|
|
ssize_t lastE = m_lastE;
|
|
// Since lastE is always less than 2^32-1 and invalid_index == -1,
|
|
// we can save a check by doing an unsigned comparison instead of
|
|
// a signed comparison.
|
|
while (size_t(pos) < size_t(lastE)) {
|
|
++pos;
|
|
if (elms[pos].data.m_type < KindOfTombstone) {
|
|
return pos;
|
|
}
|
|
}
|
|
return ArrayData::invalid_index;
|
|
}
|
|
|
|
ssize_t HphpArray::iter_rewind(ssize_t pos) const {
|
|
if (pos == ArrayData::invalid_index) {
|
|
return ArrayData::invalid_index;
|
|
}
|
|
return prevElm(m_data, pos);
|
|
}
|
|
|
|
Variant HphpArray::getKey(ssize_t pos) const {
|
|
assert(pos != ArrayData::invalid_index);
|
|
Elm* e = &m_data[/*(ElmInd)*/pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
if (e->hasStrKey()) {
|
|
return e->key; // String key.
|
|
}
|
|
return e->ikey; // Integer key.
|
|
}
|
|
|
|
Variant HphpArray::getValue(ssize_t pos) const {
|
|
assert(pos != ArrayData::invalid_index);
|
|
Elm* e = &m_data[/*(ElmInd)*/pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
|
|
CVarRef HphpArray::getValueRef(ssize_t pos) const {
|
|
assert(pos != ArrayData::invalid_index);
|
|
Elm* e = &m_data[/*(ElmInd)*/pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
|
|
bool HphpArray::isVectorData() const {
|
|
if (m_size == 0) {
|
|
return true;
|
|
}
|
|
Elm* elms = m_data;
|
|
int64_t i = 0;
|
|
for (ElmInd pos = 0; pos <= m_lastE; ++pos) {
|
|
Elm* e = &elms[pos];
|
|
if (e->data.m_type == KindOfTombstone) {
|
|
continue;
|
|
}
|
|
if (e->hasStrKey() || e->ikey != i) {
|
|
return false;
|
|
}
|
|
++i;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Variant HphpArray::reset() {
|
|
Elm* elms = m_data;
|
|
m_pos = ssize_t(nextElm(elms, ElmIndEmpty));
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* e = &elms[(ElmInd)m_pos];
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
m_pos = ArrayData::invalid_index;
|
|
return false;
|
|
}
|
|
|
|
Variant HphpArray::prev() {
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* elms = m_data;
|
|
m_pos = prevElm(elms, m_pos);
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* e = &elms[m_pos];
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Variant HphpArray::next() {
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* elms = m_data;
|
|
m_pos = nextElm(elms, m_pos);
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* e = &elms[m_pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Variant HphpArray::end() {
|
|
Elm* elms = m_data;
|
|
m_pos = prevElm(elms, (ssize_t)(m_lastE+1));
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* e = &elms[m_pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Variant HphpArray::key() const {
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
assert(m_pos <= (ssize_t)m_lastE);
|
|
Elm* e = &m_data[(ElmInd)m_pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
if (e->hasStrKey()) {
|
|
return e->key;
|
|
}
|
|
return e->ikey;
|
|
}
|
|
return uninit_null();
|
|
}
|
|
|
|
Variant HphpArray::value(ssize_t& pos) const {
|
|
if (pos != ArrayData::invalid_index) {
|
|
Elm* e = &m_data[pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
Variant HphpArray::current() const {
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
Elm* e = &m_data[m_pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static StaticString s_value("value");
|
|
static StaticString s_key("key");
|
|
|
|
Variant HphpArray::each() {
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
ArrayInit init(4);
|
|
Variant key = HphpArray::getKey(m_pos);
|
|
Variant value = HphpArray::getValue(m_pos);
|
|
init.set(int64_t(1), value);
|
|
init.set(s_value, value, true);
|
|
init.set(int64_t(0), key);
|
|
init.set(s_key, key, true);
|
|
m_pos = nextElm(m_data, m_pos);
|
|
return Array(init.create());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
//=============================================================================
|
|
// Lookup.
|
|
|
|
#define STRING_HASH(x) (int32_t(x) | 0x80000000)
|
|
|
|
static bool hitStringKey(const HphpArray::Elm* e, const StringData* s,
|
|
int32_t hash) {
|
|
// hitStringKey() should only be called on an Elm that is referenced by a
|
|
// hash table entry. HphpArray guarantees that when it adds a hash table
|
|
// entry that it always sets it to refer to a valid element. Likewise when
|
|
// it removes an element it always removes the corresponding hash entry.
|
|
// Therefore the assertion below must hold.
|
|
assert(e->data.m_type != HphpArray::KindOfTombstone);
|
|
|
|
if (e->hash() != hash) {
|
|
return false;
|
|
}
|
|
if (e->key == s) {
|
|
return true;
|
|
}
|
|
const char* data = e->key->data();
|
|
const char* sdata = s->data();
|
|
int slen = s->size();
|
|
return data == sdata || ((e->key->size() == slen)
|
|
&& (memcmp(data, sdata, slen) == 0));
|
|
}
|
|
|
|
static bool hitIntKey(const HphpArray::Elm* e, int64_t ki) {
|
|
// hitIntKey() should only be called on an Elm that is referenced by a
|
|
// hash table entry. HphpArray guarantees that when it adds a hash table
|
|
// entry that it always sets it to refer to a valid element. Likewise when
|
|
// it removes an element it always removes the corresponding hash entry.
|
|
// Therefore the assertion below must hold.
|
|
assert(e->data.m_type != HphpArray::KindOfTombstone);
|
|
return e->ikey == ki && e->hasIntKey();
|
|
}
|
|
|
|
// Quadratic probe is:
|
|
//
|
|
// h(k, i) = (k + c1*i + c2*(i^2)) % tableSize
|
|
//
|
|
// Use 1/2 for c1 and c2. In combination with a table size that is a power of
|
|
// 2, this guarantees a probe sequence of length tableSize that probes all
|
|
// table elements exactly once.
|
|
|
|
#define FIND_BODY(h0, hit) \
|
|
size_t tableMask = m_tableMask; \
|
|
size_t probeIndex = size_t(h0) & tableMask; \
|
|
Elm* elms = m_data; \
|
|
ssize_t /*ElmInd*/ pos = m_hash[probeIndex]; \
|
|
if ((validElmInd(pos) && hit) || pos == ssize_t(ElmIndEmpty)) { \
|
|
return pos; \
|
|
} \
|
|
/* Quadratic probe. */ \
|
|
for (size_t i = 1;; ++i) { \
|
|
assert(i <= tableMask); \
|
|
probeIndex = (probeIndex + i) & tableMask; \
|
|
assert(((size_t(h0)+((i + i*i) >> 1)) & tableMask) == probeIndex); \
|
|
pos = m_hash[probeIndex]; \
|
|
if ((validElmInd(pos) && hit) || pos == ssize_t(ElmIndEmpty)) { \
|
|
return pos; \
|
|
} \
|
|
}
|
|
|
|
NEVER_INLINE
|
|
ssize_t /*ElmInd*/ HphpArray::find(int64_t ki) const {
|
|
if (uint64_t(ki) < m_size) {
|
|
// Try to get at it without dirtying a data cache line.
|
|
Elm* e = m_data + uint64_t(ki);
|
|
if (e->data.m_type != HphpArray::KindOfTombstone && hitIntKey(e, ki)) {
|
|
Stats::inc(Stats::HA_FindIntFast);
|
|
assert([&] {
|
|
// Our results had better match the other path
|
|
FIND_BODY(ki, hitIntKey(&elms[pos], ki));
|
|
}() == ki);
|
|
return ki;
|
|
}
|
|
}
|
|
Stats::inc(Stats::HA_FindIntSlow);
|
|
FIND_BODY(ki, hitIntKey(&elms[pos], ki));
|
|
}
|
|
|
|
NEVER_INLINE
|
|
ssize_t /*ElmInd*/ HphpArray::find(const StringData* s,
|
|
strhash_t prehash) const {
|
|
int32_t h = STRING_HASH(prehash);
|
|
FIND_BODY(prehash, hitStringKey(&elms[pos], s, h));
|
|
}
|
|
#undef FIND_BODY
|
|
|
|
NEVER_INLINE
|
|
HphpArray::ElmInd* warnUnbalanced(size_t n, HphpArray::ElmInd* ei) {
|
|
raise_error("Array is too unbalanced (%lu)", n);
|
|
return ei;
|
|
}
|
|
|
|
#define FIND_FOR_INSERT_BODY(h0, hit) \
|
|
ElmInd* ret = nullptr; \
|
|
size_t tableMask = m_tableMask; \
|
|
size_t probeIndex = size_t(h0) & tableMask; \
|
|
Elm* elms = m_data; \
|
|
ElmInd* ei = &m_hash[probeIndex]; \
|
|
ssize_t /*ElmInd*/ pos = *ei; \
|
|
if ((validElmInd(pos) && hit) || pos == ssize_t(ElmIndEmpty)) { \
|
|
return ei; \
|
|
} \
|
|
if (!validElmInd(pos)) ret = ei; \
|
|
/* Quadratic probe. */ \
|
|
for (size_t i = 1;; ++i) { \
|
|
assert(i <= tableMask); \
|
|
probeIndex = (probeIndex + i) & tableMask; \
|
|
assert(((size_t(h0)+((i + i*i) >> 1)) & tableMask) == probeIndex); \
|
|
ei = &m_hash[probeIndex]; \
|
|
pos = ssize_t(*ei); \
|
|
if (validElmInd(pos)) { \
|
|
if (hit) { \
|
|
assert(m_hLoad <= computeMaxElms(tableMask)); \
|
|
return ei; \
|
|
} \
|
|
} else { \
|
|
if (!ret) ret = ei; \
|
|
if (pos == ElmIndEmpty) { \
|
|
assert(m_hLoad <= computeMaxElms(tableMask)); \
|
|
return LIKELY(i <= 100) || \
|
|
LIKELY(i <= size_t(RuntimeOption::MaxArrayChain)) ? \
|
|
ret : warnUnbalanced(i, ret);\
|
|
} \
|
|
} \
|
|
}
|
|
|
|
NEVER_INLINE
|
|
HphpArray::ElmInd* HphpArray::findForInsert(int64_t ki) const {
|
|
FIND_FOR_INSERT_BODY(ki, hitIntKey(&elms[pos], ki));
|
|
}
|
|
|
|
NEVER_INLINE
|
|
HphpArray::ElmInd* HphpArray::findForInsert(const StringData* s,
|
|
strhash_t prehash) const {
|
|
int32_t h = STRING_HASH(prehash);
|
|
FIND_FOR_INSERT_BODY(prehash, hitStringKey(&elms[pos], s, h));
|
|
}
|
|
#undef FIND_FOR_INSERT_BODY
|
|
|
|
NEVER_INLINE HphpArray::ElmInd*
|
|
HphpArray::findForNewInsertLoop(size_t tableMask, size_t h0) const {
|
|
/* Quadratic probe. */
|
|
size_t probeIndex = h0 & tableMask;
|
|
for (size_t i = 1;; ++i) {
|
|
assert(i <= tableMask);
|
|
probeIndex = (probeIndex + i) & tableMask;
|
|
assert(((h0 + ((i + i * i) >> 1)) & tableMask) == probeIndex);
|
|
ElmInd* ei = &m_hash[probeIndex];
|
|
ssize_t pos = ssize_t(*ei);
|
|
if (!validElmInd(pos)) {
|
|
return ei;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool HphpArray::exists(int64_t k) const {
|
|
return find(k) != (ssize_t)ElmIndEmpty;
|
|
}
|
|
|
|
bool HphpArray::exists(const StringData* k) const {
|
|
ssize_t /*ElmInd*/ pos = find(k, k->hash());
|
|
return pos != ssize_t(ElmIndEmpty);
|
|
}
|
|
|
|
CVarRef HphpArray::get(int64_t k, bool error /* = false */) const {
|
|
ElmInd pos = find(k);
|
|
if (pos != ElmIndEmpty) {
|
|
Elm* e = &m_data[pos];
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
return error ? getNotFound(k) : null_variant;
|
|
}
|
|
|
|
CVarRef HphpArray::get(const StringData* key, bool error /* = false */) const {
|
|
ElmInd pos = find(key, key->hash());
|
|
if (pos != ElmIndEmpty) {
|
|
Elm* e = &m_data[pos];
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
return error ? getNotFound(key) : null_variant;
|
|
}
|
|
|
|
ssize_t HphpArray::getIndex(int64_t k) const {
|
|
return ssize_t(find(k));
|
|
}
|
|
|
|
ssize_t HphpArray::getIndex(const StringData* k) const {
|
|
return ssize_t(find(k, k->hash()));
|
|
}
|
|
|
|
//=============================================================================
|
|
// Append/insert/update.
|
|
|
|
inline ALWAYS_INLINE bool HphpArray::isFull() const {
|
|
uint32_t maxElms = computeMaxElms(m_tableMask);
|
|
assert(m_lastE == ElmIndEmpty || uint32_t(m_lastE) + 1 <= maxElms);
|
|
assert(m_hLoad <= maxElms);
|
|
return uint32_t(m_lastE) + 1 == maxElms || m_hLoad == maxElms;
|
|
}
|
|
|
|
inline ALWAYS_INLINE HphpArray::Elm* HphpArray::allocElmFast(ElmInd* ei) {
|
|
assert(!validElmInd(*ei) && !isFull());
|
|
assert(m_size != 0 || m_lastE == ElmIndEmpty);
|
|
#ifdef PEDANTIC
|
|
if (m_size >= 0x7fffffffU) {
|
|
raise_error("Cannot insert into array with 2^31 - 1 elements");
|
|
return nullptr;
|
|
}
|
|
#endif
|
|
++m_size;
|
|
m_hLoad += (*ei == ElmIndEmpty);
|
|
ElmInd i = ++m_lastE;
|
|
(*ei) = i;
|
|
return &m_data[i];
|
|
}
|
|
|
|
inline ALWAYS_INLINE HphpArray::Elm* HphpArray::allocElm(ElmInd* ei) {
|
|
Elm* e = allocElmFast(ei);
|
|
if (m_pos == ArrayData::invalid_index) m_pos = ssize_t(*ei);
|
|
return e;
|
|
}
|
|
|
|
inline ALWAYS_INLINE
|
|
HphpArray::Elm* HphpArray::newElm(ElmInd* ei, size_t h0) {
|
|
if (isFull()) return newElmGrow(h0);
|
|
return allocElm(ei);
|
|
}
|
|
|
|
NEVER_INLINE
|
|
HphpArray::Elm* HphpArray::newElmGrow(size_t h0) {
|
|
resize();
|
|
return allocElm(findForNewInsert(h0));
|
|
}
|
|
|
|
inline ALWAYS_INLINE
|
|
void HphpArray::initElmInt(Elm* e, int64_t ki, CVarRef rhs, bool isRef) {
|
|
if (isRef) {
|
|
tvAsUninitializedVariant(&e->data).constructRefHelper(rhs);
|
|
} else {
|
|
tvAsUninitializedVariant(&e->data).constructValHelper(rhs);
|
|
}
|
|
e->setIntKey(ki);
|
|
}
|
|
|
|
inline ALWAYS_INLINE
|
|
void HphpArray::initElmStr(Elm* e, strhash_t h, StringData* key, CVarRef rhs,
|
|
bool isRef) {
|
|
if (isRef) {
|
|
tvAsUninitializedVariant(&e->data).constructRefHelper(rhs);
|
|
} else {
|
|
tvAsUninitializedVariant(&e->data).constructValHelper(rhs);
|
|
}
|
|
e->setStrKey(key, h);
|
|
key->incRefCount();
|
|
}
|
|
|
|
inline ALWAYS_INLINE
|
|
void HphpArray::newElmInt(ElmInd* ei, int64_t ki, CVarRef data,
|
|
bool byRef) {
|
|
initElmInt(newElm(ei, ki), ki, data, byRef);
|
|
}
|
|
|
|
inline ALWAYS_INLINE
|
|
void HphpArray::newElmStr(ElmInd* ei, strhash_t h, StringData* key,
|
|
CVarRef data, bool byRef) {
|
|
initElmStr(newElm(ei, h), h, key, data, byRef);
|
|
}
|
|
|
|
void HphpArray::allocData(size_t maxElms, size_t tableSize) {
|
|
assert(!m_data);
|
|
if (maxElms <= SmallSize) {
|
|
m_data = m_inline_data.slots;
|
|
m_hash = m_inline_data.hash;
|
|
m_allocMode = kInline;
|
|
return;
|
|
}
|
|
size_t hashSize = tableSize * sizeof(ElmInd);
|
|
size_t dataSize = maxElms * sizeof(Elm);
|
|
size_t allocSize = hashSize <= sizeof(m_inline_hash) ? dataSize :
|
|
dataSize + hashSize;
|
|
if (!m_nonsmart) {
|
|
m_data = (Elm*) smart_malloc(allocSize);
|
|
m_allocMode = kSmart;
|
|
} else {
|
|
m_data = (Elm*) Util::safe_malloc(allocSize);
|
|
m_allocMode = kMalloc;
|
|
}
|
|
m_hash = hashSize <= sizeof(m_inline_hash) ? m_inline_hash :
|
|
(ElmInd*)(uintptr_t(m_data) + dataSize);
|
|
}
|
|
|
|
void HphpArray::reallocData(size_t maxElms, size_t tableSize, uint oldMask) {
|
|
assert(m_data && oldMask > 0 && maxElms > SmallSize);
|
|
size_t hashSize = tableSize * sizeof(ElmInd);
|
|
size_t dataSize = maxElms * sizeof(Elm);
|
|
size_t allocSize = hashSize <= sizeof(m_inline_hash) ? dataSize :
|
|
dataSize + hashSize;
|
|
size_t oldDataSize = computeMaxElms(oldMask) * sizeof(Elm); // slots only.
|
|
if (!m_nonsmart) {
|
|
assert(m_allocMode == kInline || m_allocMode == kSmart);
|
|
if (m_allocMode == kInline) {
|
|
m_data = (Elm*) smart_malloc(allocSize);
|
|
memcpy(m_data, m_inline_data.slots, oldDataSize);
|
|
m_allocMode = kSmart;
|
|
} else {
|
|
m_data = (Elm*) smart_realloc(m_data, allocSize);
|
|
}
|
|
} else {
|
|
assert(m_allocMode == kInline || m_allocMode == kMalloc);
|
|
if (m_allocMode == kInline) {
|
|
m_data = (Elm*) Util::safe_malloc(allocSize);
|
|
memcpy(m_data, m_inline_data.slots, oldDataSize);
|
|
m_allocMode = kMalloc;
|
|
} else {
|
|
m_data = (Elm*) Util::safe_realloc(m_data, allocSize);
|
|
}
|
|
}
|
|
m_hash = hashSize <= sizeof(m_inline_hash) ? m_inline_hash :
|
|
(ElmInd*)(uintptr_t(m_data) + dataSize);
|
|
}
|
|
|
|
inline ALWAYS_INLINE void HphpArray::resizeIfNeeded() {
|
|
if (isFull()) resize();
|
|
}
|
|
|
|
NEVER_INLINE void HphpArray::resize() {
|
|
uint32_t maxElms = computeMaxElms(m_tableMask);
|
|
assert(m_lastE == ElmIndEmpty || uint32_t(m_lastE)+1 <= maxElms);
|
|
assert(m_hLoad <= maxElms);
|
|
// At a minimum, compaction is required. If the load factor would be >0.5
|
|
// even after compaction, grow instead, in order to avoid the possibility
|
|
// of repeated compaction if the load factor were to hover at nearly 0.75.
|
|
bool doGrow = (m_size > (maxElms >> 1));
|
|
#ifdef PEDANTIC
|
|
if (m_tableMask > 0x7fffffffU && doGrow) {
|
|
// If the hashtable is at its maximum size, we cannot grow
|
|
doGrow = false;
|
|
// Check if compaction would actually make room for at least one new
|
|
// element. If not, raise an error.
|
|
if (m_size >= 0x7fffffffU) {
|
|
raise_error("Cannot grow an array with 2^31 - 1 elements");
|
|
return;
|
|
}
|
|
}
|
|
#endif
|
|
if (doGrow) {
|
|
grow();
|
|
} else {
|
|
compact();
|
|
}
|
|
}
|
|
|
|
void HphpArray::grow() {
|
|
assert(m_tableMask <= 0x7fffffffU);
|
|
uint32_t oldMask = m_tableMask;
|
|
m_tableMask = (uint)(size_t(m_tableMask) + size_t(m_tableMask) + size_t(1));
|
|
size_t tableSize = computeTableSize(m_tableMask);
|
|
size_t maxElms = computeMaxElms(m_tableMask);
|
|
reallocData(maxElms, tableSize, oldMask);
|
|
// All the elements have been copied and their offsets from the base are
|
|
// still the same, so we just need to build the new hash table.
|
|
initHash(m_hash, tableSize);
|
|
#ifdef DEBUG
|
|
// Wait to set m_hLoad to m_size until after rebuilding is complete,
|
|
// in order to maintain invariants in findForNewInsert().
|
|
m_hLoad = 0;
|
|
#else
|
|
m_hLoad = m_size;
|
|
#endif
|
|
if (m_size > 0) {
|
|
Elm* elms = m_data;
|
|
for (ElmInd pos = 0; pos <= m_lastE; ++pos) {
|
|
Elm* e = &elms[pos];
|
|
if (e->data.m_type == KindOfTombstone) {
|
|
continue;
|
|
}
|
|
ElmInd* ei = findForNewInsert(e->hasIntKey() ? e->ikey : e->hash());
|
|
*ei = pos;
|
|
}
|
|
#ifdef DEBUG
|
|
m_hLoad = m_size;
|
|
#endif
|
|
}
|
|
}
|
|
|
|
void HphpArray::compact(bool renumber /* = false */) {
|
|
ElmKey mPos;
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
// Cache key for element associated with m_pos in order to update m_pos
|
|
// below.
|
|
assert(m_pos <= ssize_t(m_lastE));
|
|
Elm* e = &(m_data[(ElmInd)m_pos]);
|
|
mPos.hash = e->hasIntKey() ? 0 : e->hash();
|
|
mPos.key = e->key;
|
|
} else {
|
|
// Silence compiler warnings.
|
|
mPos.hash = 0;
|
|
mPos.key = nullptr;
|
|
}
|
|
TinyVector<ElmKey, 3> siKeys;
|
|
for (FullPosRange r(strongIterators()); !r.empty(); r.popFront()) {
|
|
ElmInd ei = r.front()->m_pos;
|
|
if (ei != ElmIndEmpty) {
|
|
Elm* e = &m_data[ei];
|
|
siKeys.push_back(ElmKey(e->hash(), e->key));
|
|
}
|
|
}
|
|
if (renumber) {
|
|
m_nextKI = 0;
|
|
}
|
|
Elm* elms = m_data;
|
|
size_t tableSize = computeTableSize(m_tableMask);
|
|
initHash(m_hash, tableSize);
|
|
#ifdef DEBUG
|
|
// Wait to set m_hLoad to m_size until after rebuilding is complete,
|
|
// in order to maintain invariants in findForNewInsert().
|
|
m_hLoad = 0;
|
|
#else
|
|
m_hLoad = m_size;
|
|
#endif
|
|
ElmInd frPos = 0;
|
|
for (ElmInd toPos = 0; toPos < ElmInd(m_size); ++toPos) {
|
|
Elm* frE = &elms[frPos];
|
|
while (frE->data.m_type == KindOfTombstone) {
|
|
++frPos;
|
|
assert(frPos <= m_lastE);
|
|
frE = &elms[frPos];
|
|
}
|
|
Elm* toE = &elms[toPos];
|
|
if (toE != frE) {
|
|
memcpy((void*)toE, (void*)frE, sizeof(Elm));
|
|
}
|
|
if (renumber && !toE->hasStrKey()) {
|
|
toE->ikey = m_nextKI;
|
|
++m_nextKI;
|
|
}
|
|
ElmInd* ie = findForNewInsert(toE->hasIntKey() ? toE->ikey : toE->hash());
|
|
*ie = toPos;
|
|
++frPos;
|
|
}
|
|
m_lastE = m_size - 1;
|
|
#ifdef DEBUG
|
|
m_hLoad = m_size;
|
|
#endif
|
|
if (m_pos != ArrayData::invalid_index) {
|
|
// Update m_pos, now that compaction is complete.
|
|
if (mPos.hash) {
|
|
m_pos = ssize_t(find(mPos.key, mPos.hash));
|
|
} else {
|
|
m_pos = ssize_t(find(mPos.ikey));
|
|
}
|
|
}
|
|
// Update strong iterators, now that compaction is complete.
|
|
int key = 0;
|
|
for (FullPosRange r(strongIterators()); !r.empty(); r.popFront()) {
|
|
FullPos* fp = r.front();
|
|
if (fp->m_pos != ArrayData::invalid_index) {
|
|
ElmKey &k = siKeys[key];
|
|
key++;
|
|
if (k.hash) { // string key
|
|
fp->m_pos = ssize_t(find(k.key, k.hash));
|
|
} else { // int key
|
|
fp->m_pos = ssize_t(find(k.ikey));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
static inline void elemConstruct(const TypedValue* fr, TypedValue* to) {
|
|
tvDupCell(tvToCell(fr), to);
|
|
}
|
|
|
|
bool HphpArray::nextInsert(CVarRef data) {
|
|
if (UNLIKELY(m_nextKI < 0)) {
|
|
raise_warning("Cannot add element to the array as the next element is "
|
|
"already occupied");
|
|
return false;
|
|
}
|
|
resizeIfNeeded();
|
|
int64_t ki = m_nextKI;
|
|
// The check above enforces an invariant that allows us to always
|
|
// know that m_nextKI is not present in the array, so it is safe
|
|
// to use findForNewInsert()
|
|
ElmInd* ei = findForNewInsert(ki);
|
|
assert(!validElmInd(*ei));
|
|
// Allocate and initialize a new element.
|
|
initElmInt(allocElm(ei), ki, data);
|
|
// Update next free element.
|
|
++m_nextKI;
|
|
return true;
|
|
}
|
|
|
|
ArrayData* HphpArray::nextInsertRef(CVarRef data) {
|
|
if (UNLIKELY(m_nextKI < 0)) {
|
|
raise_warning("Cannot add element to the array as the next element is "
|
|
"already occupied");
|
|
return this;
|
|
}
|
|
resizeIfNeeded();
|
|
int64_t ki = m_nextKI;
|
|
// The check above enforces an invariant that allows us to always
|
|
// know that m_nextKI is not present in the array, so it is safe
|
|
// to use findForNewInsert()
|
|
ElmInd* ei = findForNewInsert(ki);
|
|
initElmInt(allocElm(ei), ki, data, true /*byRef*/);
|
|
// Update next free element.
|
|
++m_nextKI;
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::nextInsertWithRef(CVarRef data) {
|
|
resizeIfNeeded();
|
|
int64_t ki = m_nextKI;
|
|
ElmInd* ei = findForInsert(ki);
|
|
assert(!validElmInd(*ei));
|
|
|
|
// Allocate a new element.
|
|
Elm* e = allocElm(ei);
|
|
tvWriteNull(&e->data);
|
|
tvAsVariant(&e->data).setWithRef(data);
|
|
// Set key.
|
|
e->setIntKey(ki);
|
|
// Update next free element.
|
|
++m_nextKI;
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::addLvalImpl(int64_t ki, Variant** pDest) {
|
|
assert(pDest != nullptr);
|
|
ElmInd* ei = findForInsert(ki);
|
|
if (validElmInd(*ei)) {
|
|
*pDest = &tvAsVariant(&m_data[*ei].data);
|
|
return this;
|
|
}
|
|
Elm* e = newElm(ei, ki);
|
|
tvWriteNull(&e->data);
|
|
e->setIntKey(ki);
|
|
*pDest = &(tvAsVariant(&e->data));
|
|
if (ki >= m_nextKI && m_nextKI >= 0) {
|
|
m_nextKI = ki + 1;
|
|
}
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::addLvalImpl(StringData* key, strhash_t h, Variant** pDest) {
|
|
assert(key != nullptr && pDest != nullptr);
|
|
ElmInd* ei = findForInsert(key, h);
|
|
if (validElmInd(*ei)) {
|
|
Elm* e = &m_data[*ei];
|
|
TypedValue* tv;
|
|
tv = &e->data;
|
|
*pDest = &tvAsVariant(tv);
|
|
return this;
|
|
}
|
|
Elm* e = newElm(ei, h);
|
|
// Initialize element to null and store the address of the element into
|
|
// *pDest.
|
|
tvWriteNull(&e->data);
|
|
// Set key.
|
|
e->setStrKey(key, h);
|
|
e->key->incRefCount();
|
|
*pDest = &(tvAsVariant(&e->data));
|
|
return this;
|
|
}
|
|
|
|
inline ArrayData* HphpArray::addVal(int64_t ki, CVarRef data) {
|
|
assert(!exists(ki));
|
|
resizeIfNeeded();
|
|
ElmInd* ei = findForNewInsert(ki);
|
|
Elm* e = allocElm(ei);
|
|
TypedValue* fr = (TypedValue*)(&data);
|
|
TypedValue* to = (TypedValue*)(&e->data);
|
|
elemConstruct(fr, to);
|
|
e->setIntKey(ki);
|
|
if (ki >= m_nextKI && m_nextKI >= 0) {
|
|
m_nextKI = ki + 1;
|
|
}
|
|
return this;
|
|
}
|
|
|
|
inline ArrayData* HphpArray::addVal(StringData* key, CVarRef data) {
|
|
assert(!exists(key));
|
|
resizeIfNeeded();
|
|
strhash_t h = key->hash();
|
|
ElmInd* ei = findForNewInsert(h);
|
|
Elm *e = allocElm(ei);
|
|
// Set the element
|
|
TypedValue* to = (TypedValue*)(&e->data);
|
|
TypedValue* fr = (TypedValue*)(&data);
|
|
elemConstruct(fr, to);
|
|
// Set the key after data is written
|
|
e->setStrKey(key, h);
|
|
e->key->incRefCount();
|
|
return this;
|
|
}
|
|
|
|
inline ArrayData* HphpArray::addValWithRef(int64_t ki, CVarRef data) {
|
|
resizeIfNeeded();
|
|
ElmInd* ei = findForInsert(ki);
|
|
if (!validElmInd(*ei)) {
|
|
Elm* e = allocElm(ei);
|
|
tvWriteNull(&e->data);
|
|
tvAsVariant(&e->data).setWithRef(data);
|
|
e->setIntKey(ki);
|
|
if (ki >= m_nextKI) {
|
|
m_nextKI = ki + 1;
|
|
}
|
|
}
|
|
return this;
|
|
}
|
|
|
|
inline ArrayData* HphpArray::addValWithRef(StringData* key, CVarRef data) {
|
|
resizeIfNeeded();
|
|
strhash_t h = key->hash();
|
|
ElmInd* ei = findForInsert(key, h);
|
|
if (!validElmInd(*ei)) {
|
|
Elm* e = allocElm(ei);
|
|
tvWriteNull(&e->data);
|
|
tvAsVariant(&e->data).setWithRef(data);
|
|
e->setStrKey(key, h);
|
|
e->key->incRefCount();
|
|
}
|
|
return this;
|
|
}
|
|
|
|
inline INLINE_SINGLE_CALLER
|
|
ArrayData* HphpArray::update(int64_t ki, CVarRef data) {
|
|
ElmInd* ei = findForInsert(ki);
|
|
if (validElmInd(*ei)) {
|
|
Elm* e = &m_data[*ei];
|
|
tvAsVariant(&e->data).assignValHelper(data);
|
|
return this;
|
|
}
|
|
newElmInt(ei, ki, data);
|
|
if (ki >= m_nextKI && m_nextKI >= 0) {
|
|
m_nextKI = ki + 1;
|
|
}
|
|
return this;
|
|
}
|
|
|
|
inline INLINE_SINGLE_CALLER
|
|
ArrayData* HphpArray::update(StringData* key, CVarRef data) {
|
|
strhash_t h = key->hash();
|
|
ElmInd* ei = findForInsert(key, h);
|
|
if (validElmInd(*ei)) {
|
|
Elm* e = &m_data[*ei];
|
|
tvAsVariant(&e->data).assignValHelper(data);
|
|
return this;
|
|
}
|
|
newElmStr(ei, h, key, data);
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::updateRef(int64_t ki, CVarRef data) {
|
|
ElmInd* ei = findForInsert(ki);
|
|
if (validElmInd(*ei)) {
|
|
Elm* e = &m_data[*ei];
|
|
tvAsVariant(&e->data).assignRefHelper(data);
|
|
return this;
|
|
}
|
|
newElmInt(ei, ki, data, true /*byRef*/);
|
|
if (ki >= m_nextKI && m_nextKI >= 0) {
|
|
m_nextKI = ki + 1;
|
|
}
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::updateRef(StringData* key, CVarRef data) {
|
|
strhash_t h = key->hash();
|
|
ElmInd* ei = findForInsert(key, h);
|
|
if (validElmInd(*ei)) {
|
|
Elm* e = &m_data[*ei];
|
|
tvAsVariant(&e->data).assignRefHelper(data);
|
|
return this;
|
|
}
|
|
newElmStr(ei, h, key, data, true /*byRef*/);
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::lval(int64_t k, Variant*& ret, bool copy,
|
|
bool checkExist /* = false */) {
|
|
if (!copy) return addLvalImpl(k, &ret);
|
|
if (checkExist) {
|
|
ssize_t /*ElmInd*/ pos = find(k);
|
|
if (pos != (ssize_t)ElmIndEmpty) {
|
|
Elm* e = &m_data[pos];
|
|
if (tvAsVariant(&e->data).isReferenced() ||
|
|
tvAsVariant(&e->data).isObject()) {
|
|
ret = &tvAsVariant(&e->data);
|
|
return this;
|
|
}
|
|
}
|
|
}
|
|
return copyImpl()->addLvalImpl(k, &ret);
|
|
}
|
|
|
|
ArrayData* HphpArray::lval(StringData* key, Variant*& ret, bool copy,
|
|
bool checkExist /* = false */) {
|
|
strhash_t prehash = key->hash();
|
|
if (!copy) return addLvalImpl(key, prehash, &ret);
|
|
if (checkExist) {
|
|
ssize_t /*ElmInd*/ pos = find(key, prehash);
|
|
if (pos != (ssize_t)ElmIndEmpty) {
|
|
Elm* e = &m_data[pos];
|
|
TypedValue* tv = &e->data;
|
|
if (tvAsVariant(tv).isReferenced() ||
|
|
tvAsVariant(tv).isObject()) {
|
|
ret = &tvAsVariant(tv);
|
|
return this;
|
|
}
|
|
}
|
|
}
|
|
return copyImpl()->addLvalImpl(key, prehash, &ret);
|
|
}
|
|
|
|
ArrayData *HphpArray::lvalPtr(StringData* key, Variant*& ret, bool copy,
|
|
bool create) {
|
|
strhash_t prehash = key->hash();
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
if (create) return a->addLvalImpl(key, prehash, &ret);
|
|
ssize_t /*ElmInd*/ pos = a->find(key, prehash);
|
|
if (pos != (ssize_t)ElmIndEmpty) {
|
|
Elm* e = &a->m_data[pos];
|
|
ret = &tvAsVariant(&e->data);
|
|
} else {
|
|
ret = nullptr;
|
|
}
|
|
return a;
|
|
}
|
|
|
|
ArrayData* HphpArray::lvalNew(Variant*& ret, bool copy) {
|
|
TypedValue* tv;
|
|
ArrayData* a = nvNew(tv, copy);
|
|
if (tv == nullptr) {
|
|
ret = &(Variant::lvalBlackHole());
|
|
} else {
|
|
ret = &tvAsVariant(tv);
|
|
}
|
|
return a;
|
|
}
|
|
|
|
ArrayData* HphpArray::set(int64_t k, CVarRef v, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->update(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::set(StringData* k, CVarRef v, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->update(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::setRef(int64_t k, CVarRef v, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->updateRef(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::setRef(StringData* k, CVarRef v, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->updateRef(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::add(int64_t k, CVarRef v, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->addVal(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::add(StringData* k, CVarRef v, bool copy) {
|
|
assert(!exists(k));
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->addVal(k, v);
|
|
}
|
|
|
|
ArrayData* HphpArray::addLval(int64_t k, Variant*& ret, bool copy) {
|
|
assert(!exists(k));
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->addLvalImpl(k, &ret);
|
|
}
|
|
|
|
ArrayData* HphpArray::addLval(StringData* k, Variant*& ret, bool copy) {
|
|
assert(!exists(k));
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->addLvalImpl(k, k->hash(), &ret);
|
|
}
|
|
|
|
//=============================================================================
|
|
// Delete.
|
|
|
|
ArrayData* HphpArray::erase(ElmInd* ei, bool updateNext /* = false */) {
|
|
ElmInd pos = *ei;
|
|
if (!validElmInd(pos)) {
|
|
return this;
|
|
}
|
|
|
|
Elm* elms = m_data;
|
|
|
|
ElmInd eIPrev = ElmIndTombstone;
|
|
for (FullPosRange r(strongIterators()); !r.empty(); r.popFront()) {
|
|
FullPos* fp = r.front();
|
|
if (fp->m_pos == ssize_t(pos)) {
|
|
if (eIPrev == ElmIndTombstone) {
|
|
// eIPrev will actually be used, so properly initialize it with the
|
|
// previous element before pos, or ElmIndEmpty if pos is the first
|
|
// element.
|
|
eIPrev = prevElm(elms, pos);
|
|
}
|
|
if (eIPrev == ElmIndEmpty) {
|
|
fp->setResetFlag(true);
|
|
}
|
|
fp->m_pos = ssize_t(eIPrev);
|
|
}
|
|
}
|
|
|
|
// If the internal pointer points to this element, advance it.
|
|
if (m_pos == ssize_t(pos)) {
|
|
ElmInd eINext = nextElm(elms, pos);
|
|
m_pos = ssize_t(eINext);
|
|
}
|
|
|
|
Elm* e = &elms[pos];
|
|
// Mark the value as a tombstone.
|
|
TypedValue* tv = &e->data;
|
|
DataType oldType = tv->m_type;
|
|
uint64_t oldDatum = tv->m_data.num;
|
|
tv->m_type = KindOfTombstone;
|
|
// Free the key if necessary, and clear the h and key fields in order to
|
|
// increase the chances that subsequent searches will quickly/safely fail
|
|
// when encountering tombstones, even though checking for KindOfTombstone is
|
|
// the last validation step during search.
|
|
if (e->hasStrKey()) {
|
|
decRefStr(e->key);
|
|
e->setIntKey(0);
|
|
} else {
|
|
// Match PHP 5.3.1 semantics
|
|
// Hacky: don't removed the unsigned cast, else g++ can optimize away
|
|
// the check for == 0x7fff..., since there is no signed int k
|
|
// for which k-1 == 0x7fff...
|
|
if ((uint64_t)e->ikey == (uint64_t)m_nextKI-1
|
|
&& (e->ikey == 0x7fffffffffffffffLL || updateNext)) {
|
|
--m_nextKI;
|
|
}
|
|
}
|
|
--m_size;
|
|
// If this element was last, adjust m_lastE.
|
|
if (pos == m_lastE) {
|
|
do {
|
|
--m_lastE;
|
|
} while (m_lastE >= 0 && elms[m_lastE].data.m_type == KindOfTombstone);
|
|
}
|
|
// Mark the hash entry as "deleted".
|
|
*ei = ElmIndTombstone;
|
|
assert(m_lastE == ElmIndEmpty ||
|
|
uint32_t(m_lastE)+1 <= computeMaxElms(m_tableMask));
|
|
assert(m_hLoad <= computeMaxElms(m_tableMask));
|
|
|
|
// Finally, decref the old value
|
|
tvRefcountedDecRefHelper(oldType, oldDatum);
|
|
|
|
if (m_size < (uint32_t)((m_lastE+1) >> 1)) {
|
|
// Compact in order to keep elms from being overly sparse.
|
|
compact();
|
|
}
|
|
return this;
|
|
}
|
|
|
|
ArrayData* HphpArray::remove(int64_t k, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->erase(a->findForInsert(k));
|
|
}
|
|
|
|
ArrayData* HphpArray::remove(const StringData* key, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
return a->erase(a->findForInsert(key, key->hash()));
|
|
}
|
|
|
|
ArrayData* HphpArray::copy() const {
|
|
return copyImpl();
|
|
}
|
|
|
|
ArrayData* HphpArray::copyWithStrongIterators() const {
|
|
HphpArray* copied = copyImpl();
|
|
moveStrongIterators(copied, const_cast<HphpArray*>(this));
|
|
return copied;
|
|
}
|
|
|
|
//=============================================================================
|
|
// non-variant interface
|
|
|
|
TypedValue* HphpArray::nvGetCell(int64_t k) const {
|
|
ElmInd pos = find(k);
|
|
return LIKELY(pos != ElmIndEmpty) ? tvToCell(&m_data[pos].data) :
|
|
nvGetNotFound(k);
|
|
}
|
|
|
|
TypedValue* HphpArray::nvGetCell(const StringData* k) const {
|
|
ElmInd pos = find(k, k->hash());
|
|
return LIKELY(pos != ElmIndEmpty) ? tvToCell(&m_data[pos].data) :
|
|
nvGetNotFound(k);
|
|
}
|
|
|
|
TypedValue* HphpArray::nvGet(int64_t ki) const {
|
|
ElmInd pos = find(ki);
|
|
if (LIKELY(pos != ElmIndEmpty)) {
|
|
Elm* e = &m_data[pos];
|
|
return &e->data;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
TypedValue* HphpArray::nvGet(const StringData* k) const {
|
|
ElmInd pos = find(k, k->hash());
|
|
if (LIKELY(pos != ElmIndEmpty)) {
|
|
Elm* e = &m_data[pos];
|
|
return &e->data;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
ArrayData* HphpArray::nvNew(TypedValue*& ret, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
if (UNLIKELY(!a->nextInsert(uninit_null()))) {
|
|
ret = nullptr;
|
|
return a;
|
|
}
|
|
assert(a->m_lastE != ElmIndEmpty);
|
|
ssize_t lastE = (ssize_t)a->m_lastE;
|
|
ret = &a->m_data[lastE].data;
|
|
return a;
|
|
}
|
|
|
|
TypedValue* HphpArray::nvGetValueRef(ssize_t pos) {
|
|
assert(pos != ArrayData::invalid_index);
|
|
Elm* e = &m_data[/*(ElmInd)*/pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return &e->data;
|
|
}
|
|
|
|
// nvGetKey does not touch out->_count, so can be used
|
|
// for inner or outer cells.
|
|
void HphpArray::nvGetKey(TypedValue* out, ssize_t pos) {
|
|
assert(pos != ArrayData::invalid_index);
|
|
assert(m_data[pos].data.m_type != KindOfTombstone);
|
|
Elm* e = &m_data[/*(ElmInd)*/pos];
|
|
getElmKey(e, out);
|
|
}
|
|
|
|
/*
|
|
* Insert a new element with index k in to the array,
|
|
* doing nothing and returning false if the element
|
|
* already exists.
|
|
*/
|
|
bool HphpArray::nvInsert(StringData *k, TypedValue *data) {
|
|
strhash_t h = k->hash();
|
|
ElmInd* ei = findForInsert(k, h);
|
|
if (validElmInd(*ei)) {
|
|
return false;
|
|
}
|
|
newElmStr(ei, h, k, tvAsVariant(data));
|
|
return true;
|
|
}
|
|
|
|
inline ALWAYS_INLINE HphpArray* HphpArray::copyImpl(HphpArray* target) const {
|
|
target->m_pos = m_pos;
|
|
target->m_data = nullptr;
|
|
target->m_nextKI = m_nextKI;
|
|
target->m_tableMask = m_tableMask;
|
|
target->m_size = m_size;
|
|
target->m_hLoad = m_hLoad;
|
|
target->m_lastE = m_lastE;
|
|
size_t tableSize = computeTableSize(m_tableMask);
|
|
size_t maxElms = computeMaxElms(m_tableMask);
|
|
target->allocData(maxElms, tableSize);
|
|
// Copy the hash.
|
|
memcpy(target->m_hash, m_hash, tableSize * sizeof(ElmInd));
|
|
// Copy the elements and bump up refcounts as needed.
|
|
if (m_size > 0) {
|
|
Elm* elms = m_data;
|
|
Elm* targetElms = target->m_data;
|
|
ssize_t lastE = (ssize_t)m_lastE;
|
|
for (ssize_t /*ElmInd*/ pos = 0; pos <= lastE; ++pos) {
|
|
Elm* e = &elms[pos];
|
|
Elm* te = &targetElms[pos];
|
|
if (e->data.m_type != KindOfTombstone) {
|
|
te->key = e->key;
|
|
te->data.hash() = e->data.hash();
|
|
if (te->hasStrKey()) te->key->incRefCount();
|
|
tvDupFlattenVars(&e->data, &te->data, this);
|
|
assert(te->hash() == e->hash()); // ensure not clobbered.
|
|
} else {
|
|
// Tombstone.
|
|
te->setIntKey(0);
|
|
te->data.m_type = KindOfTombstone;
|
|
}
|
|
}
|
|
// It's possible that there were indirect elements at the end that were
|
|
// converted to tombstones, so check if we should adjust target->m_lastE
|
|
while (target->m_lastE >= 0) {
|
|
Elm* te = &targetElms[target->m_lastE];
|
|
if (te->data.m_type != KindOfTombstone) {
|
|
break;
|
|
}
|
|
--(target->m_lastE);
|
|
}
|
|
// If the element density dropped below 50% due to indirect elements
|
|
// being converted into tombstones, we should do a compaction
|
|
if (target->m_size < (uint32_t)((target->m_lastE+1) >> 1)) {
|
|
target->compact();
|
|
}
|
|
}
|
|
return target;
|
|
}
|
|
|
|
NEVER_INLINE ArrayData* HphpArray::nonSmartCopy() const {
|
|
return copyImpl(new HphpArray(kNonSmartCopy));
|
|
}
|
|
|
|
NEVER_INLINE HphpArray* HphpArray::copyImpl() const {
|
|
return copyImpl(NEW(HphpArray)(kSmartCopy));
|
|
}
|
|
|
|
ArrayData* HphpArray::append(CVarRef v, bool copy) {
|
|
HphpArray *a = !copy ? this : copyImpl();
|
|
a->nextInsert(v);
|
|
return a;
|
|
}
|
|
|
|
/*
|
|
* Cold path helper for AddNewElemC delegates to the ArrayData::append
|
|
* virtual method.
|
|
*/
|
|
static NEVER_INLINE
|
|
ArrayData* genericAddNewElemC(ArrayData* a, TypedValue value) {
|
|
assert(a->getCount() <= 1);
|
|
ArrayData* UNUSED r = a->append(tvAsCVarRef(&value), false);
|
|
tvRefcountedDecRef(value);
|
|
assert(r == a);
|
|
return a;
|
|
}
|
|
|
|
/*
|
|
* The pass-by-value and move semantics of this helper are slightly different
|
|
* than other array helpers, but tuned for the opcode. See doc comment in
|
|
* hphp_array.h.
|
|
*/
|
|
ArrayData* HphpArray::AddNewElemC(ArrayData* a, TypedValue value) {
|
|
assert(a->getCount() <= 1 && value.m_type != KindOfRef);
|
|
HphpArray* h;
|
|
ElmInd* ei;
|
|
int64_t k;
|
|
if (LIKELY(a->isHphpArray()) &&
|
|
((h = (HphpArray*)a), LIKELY(h->m_pos >= 0)) &&
|
|
LIKELY(!h->isFull()) &&
|
|
((k = h->m_nextKI), LIKELY(k >= 0)) &&
|
|
((ei = &h->m_hash[k & h->m_tableMask]), LIKELY(!validElmInd(*ei)))) {
|
|
// Fast path is a streamlined copy of Variant.constructValHelper()
|
|
// with no incref+decref because we're moving (data,type) to this array.
|
|
Elm* e = h->allocElmFast(ei);
|
|
e->data.m_type = typeInitNull(value.m_type);
|
|
e->data.m_data.num = value.m_data.num;
|
|
e->setIntKey(k);
|
|
h->m_nextKI = k + 1;
|
|
return a;
|
|
}
|
|
return genericAddNewElemC(a, value);
|
|
}
|
|
|
|
ArrayData* HphpArray::appendRef(CVarRef v, bool copy) {
|
|
HphpArray *a = !copy ? this : copyImpl();
|
|
return a->nextInsertRef(v);
|
|
}
|
|
|
|
ArrayData *HphpArray::appendWithRef(CVarRef v, bool copy) {
|
|
HphpArray *a = !copy ? this : copyImpl();
|
|
return a->nextInsertWithRef(v);
|
|
}
|
|
|
|
ArrayData* HphpArray::append(const ArrayData* elems, ArrayOp op, bool copy) {
|
|
HphpArray* a = !copy ? this : copyImpl();
|
|
if (op == Plus) {
|
|
for (ArrayIter it(elems); !it.end(); it.next()) {
|
|
Variant key = it.first();
|
|
CVarRef value = it.secondRef();
|
|
if (key.isNumeric()) {
|
|
a->addValWithRef(key.toInt64(), value);
|
|
} else {
|
|
a->addValWithRef(key.getStringData(), value);
|
|
}
|
|
}
|
|
} else {
|
|
assert(op == Merge);
|
|
for (ArrayIter it(elems); !it.end(); it.next()) {
|
|
Variant key = it.first();
|
|
CVarRef value = it.secondRef();
|
|
if (key.isNumeric()) {
|
|
a->nextInsertWithRef(value);
|
|
} else {
|
|
Variant *p;
|
|
StringData *sd = key.getStringData();
|
|
a->addLvalImpl(sd, sd->hash(), &p);
|
|
p->setWithRef(value);
|
|
}
|
|
}
|
|
}
|
|
return a;
|
|
}
|
|
|
|
ArrayData* HphpArray::pop(Variant& value) {
|
|
HphpArray* a = getCount() <= 1 ? this : copyImpl();
|
|
Elm* elms = a->m_data;
|
|
ElmInd pos = a->HphpArray::iter_end();
|
|
if (validElmInd(pos)) {
|
|
Elm* e = &elms[pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
value = tvAsCVarRef(&e->data);
|
|
ElmInd* ei = e->hasStrKey()
|
|
? a->findForInsert(e->key, e->hash())
|
|
: a->findForInsert(e->ikey);
|
|
a->erase(ei, true);
|
|
} else {
|
|
value = uninit_null();
|
|
}
|
|
// To conform to PHP behavior, the pop operation resets the array's
|
|
// internal iterator.
|
|
a->m_pos = a->nextElm(elms, ElmIndEmpty);
|
|
return a;
|
|
}
|
|
|
|
ArrayData* HphpArray::dequeue(Variant& value) {
|
|
HphpArray* a = getCount() <= 1 ? this : copyImpl();
|
|
// To conform to PHP behavior, we invalidate all strong iterators when an
|
|
// element is removed from the beginning of the array.
|
|
a->freeStrongIterators();
|
|
Elm* elms = a->m_data;
|
|
ElmInd pos = a->nextElm(elms, ElmIndEmpty);
|
|
if (validElmInd(pos)) {
|
|
Elm* e = &elms[pos];
|
|
value = tvAsCVarRef(&e->data);
|
|
a->erase(e->hasStrKey() ?
|
|
a->findForInsert(e->key, e->hash()) :
|
|
a->findForInsert(e->ikey));
|
|
a->compact(true);
|
|
} else {
|
|
value = uninit_null();
|
|
}
|
|
// To conform to PHP behavior, the dequeue operation resets the array's
|
|
// internal iterator
|
|
a->m_pos = ssize_t(a->nextElm(elms, ElmIndEmpty));
|
|
return a;
|
|
}
|
|
|
|
ArrayData* HphpArray::prepend(CVarRef v, bool copy) {
|
|
HphpArray* a = getCount() <= 1 ? this : copyImpl();
|
|
// To conform to PHP behavior, we invalidate all strong iterators when an
|
|
// element is added to the beginning of the array.
|
|
a->freeStrongIterators();
|
|
|
|
Elm* elms = a->m_data;
|
|
if (a->m_lastE == 0 || elms[0].data.m_type != KindOfTombstone) {
|
|
// Make sure there is room to insert an element.
|
|
a->resizeIfNeeded();
|
|
// Reload elms, in case resizeIfNeeded() had side effects.
|
|
elms = a->m_data;
|
|
// Move the existing elements to make element 0 available.
|
|
memmove(&elms[1], &elms[0], (a->m_lastE+1) * sizeof(Elm));
|
|
++a->m_lastE;
|
|
}
|
|
// Prepend.
|
|
Elm* e = &elms[0];
|
|
|
|
TypedValue* fr = (TypedValue*)(&v);
|
|
TypedValue* to = (TypedValue*)(&e->data);
|
|
elemConstruct(fr, to);
|
|
|
|
e->setIntKey(0);
|
|
++a->m_size;
|
|
|
|
// Renumber.
|
|
a->compact(true);
|
|
// To conform to PHP behavior, the prepend operation resets the array's
|
|
// internal iterator
|
|
a->m_pos = ssize_t(a->nextElm(elms, ElmIndEmpty));
|
|
return a;
|
|
}
|
|
|
|
void HphpArray::renumber() {
|
|
compact(true);
|
|
}
|
|
|
|
void HphpArray::onSetEvalScalar() {
|
|
Elm* elms = m_data;
|
|
for (ElmInd pos = 0; pos <= m_lastE; ++pos) {
|
|
Elm* e = &elms[pos];
|
|
if (e->data.m_type != KindOfTombstone) {
|
|
StringData *key = e->key;
|
|
if (e->hasStrKey() && !key->isStatic()) {
|
|
StringData *skey = StringData::GetStaticString(key);
|
|
if (key->decRefCount() == 0) {
|
|
DELETE(StringData)(key);
|
|
}
|
|
e->key = skey;
|
|
}
|
|
tvAsVariant(&e->data).setEvalScalar();
|
|
}
|
|
}
|
|
}
|
|
|
|
bool HphpArray::validFullPos(const FullPos &fp) const {
|
|
assert(fp.getContainer() == (ArrayData*)this);
|
|
if (fp.getResetFlag()) return false;
|
|
return (fp.m_pos != ssize_t(ElmIndEmpty));
|
|
}
|
|
|
|
bool HphpArray::advanceFullPos(FullPos& fp) {
|
|
Elm* elms = m_data;
|
|
if (fp.getResetFlag()) {
|
|
fp.setResetFlag(false);
|
|
fp.m_pos = ElmIndEmpty;
|
|
} else if (fp.m_pos == ssize_t(ElmIndEmpty)) {
|
|
return false;
|
|
}
|
|
fp.m_pos = nextElm(elms, fp.m_pos);
|
|
if (fp.m_pos == ssize_t(ElmIndEmpty)) {
|
|
return false;
|
|
}
|
|
// To conform to PHP behavior, we need to set the internal
|
|
// cursor to point to the next element.
|
|
m_pos = nextElm(elms, fp.m_pos);
|
|
return true;
|
|
}
|
|
|
|
CVarRef HphpArray::currentRef() {
|
|
assert(m_pos != ArrayData::invalid_index);
|
|
Elm* e = &m_data[(ElmInd)m_pos];
|
|
assert(e->data.m_type != KindOfTombstone);
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
|
|
CVarRef HphpArray::endRef() {
|
|
assert(m_lastE != ElmIndEmpty);
|
|
ElmInd pos = m_lastE;
|
|
Elm* e = &m_data[pos];
|
|
return tvAsCVarRef(&e->data);
|
|
}
|
|
|
|
//=============================================================================
|
|
// VM runtime support functions.
|
|
namespace VM {
|
|
|
|
// Helpers for array_setm.
|
|
ArrayData* nvCheckedSet(ArrayData* a, StringData* key, TypedValue* value,
|
|
bool copy) {
|
|
int64_t i;
|
|
return UNLIKELY(key->isStrictlyInteger(i)) ? a->nvSet(i, value, copy) :
|
|
a->nvSet(key, value, copy);
|
|
}
|
|
|
|
ArrayData* nvCheckedSet(ArrayData* a, int64_t key, TypedValue* value,
|
|
bool copy) {
|
|
return a->nvSet(key, value, copy);
|
|
}
|
|
|
|
void setmDecRef(int64_t i) { /* nop */ }
|
|
void setmDecRef(StringData* sd) { decRefStr(sd); }
|
|
|
|
template<typename Key, bool DecRefValue, bool CheckInt, bool DecRefKey>
|
|
static inline
|
|
ArrayData*
|
|
array_setm(RefData* ref, ArrayData* ad, Key key, TypedValue* value) {
|
|
ArrayData* retval;
|
|
bool copy = ad->getCount() > 1;
|
|
// nvSet will decRef any old value that may have been overwritten
|
|
// if appropriate
|
|
retval = CheckInt ? nvCheckedSet(ad, key, value, copy) :
|
|
ad->nvSet(key, value, copy);
|
|
if (DecRefKey) setmDecRef(key);
|
|
|
|
// TODO Task #1970153: It would be great if there were nvSet()
|
|
// methods that didn't bump up the refcount so that we didn't
|
|
// have to decrement it here
|
|
if (DecRefValue) tvRefcountedDecRef(value);
|
|
if (!ref) return arrayRefShuffle<false>(ad, retval, nullptr);
|
|
arrayRefShuffle<true>(ad, retval, ref->tv());
|
|
return retval;
|
|
}
|
|
|
|
/**
|
|
* Unary integer keys.
|
|
* array_setm_ik1_v --
|
|
* Polymorphic value.
|
|
*
|
|
* array_setm_ik1_v0 --
|
|
* Don't count the array's reference to the polymorphic value.
|
|
*/
|
|
ArrayData* array_setm_ik1_v(RefData* ref, ArrayData* ad, int64_t key,
|
|
TypedValue* value) {
|
|
return array_setm<int64_t, false, false, false>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_ik1_v0(RefData* ref, ArrayData* ad, int64_t key,
|
|
TypedValue* value) {
|
|
return array_setm<int64_t, true, false, false>(ref, ad, key, value);
|
|
}
|
|
|
|
/**
|
|
* String keys.
|
|
*
|
|
* array_setm_sk1_v --
|
|
* $a[$keyOfTypeString] = <polymorphic value>;
|
|
*
|
|
* array_setm_sk1_v0 --
|
|
* Like above, but don't count the new reference.
|
|
*
|
|
* array_setm_s0k1_v --
|
|
* array_setm_s0k1_v0 --
|
|
* As above, but dont decRef the key
|
|
*
|
|
* array_setm_s0k1nc_v --
|
|
* array_setm_s0k1nc_v0 --
|
|
* Dont decRef the key, and skip the check for
|
|
* whether the key is really an integer.
|
|
*/
|
|
ArrayData* array_setm_sk1_v(RefData* ref, ArrayData* ad, StringData* key,
|
|
TypedValue* value) {
|
|
return array_setm<StringData*, false, true, true>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_sk1_v0(RefData* ref, ArrayData* ad, StringData* key,
|
|
TypedValue* value) {
|
|
return array_setm<StringData*, true, true, true>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_s0k1_v(RefData* ref, ArrayData* ad, StringData* key,
|
|
TypedValue* value) {
|
|
return array_setm<StringData*, false, true, false>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_s0k1_v0(RefData* ref, ArrayData* ad, StringData* key,
|
|
TypedValue* value) {
|
|
return array_setm<StringData*, true, true, false>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_s0k1nc_v(RefData* ref, ArrayData* ad, StringData* key,
|
|
TypedValue* value) {
|
|
return array_setm<StringData*, false, false, false>(ref, ad, key, value);
|
|
}
|
|
|
|
ArrayData* array_setm_s0k1nc_v0(RefData* ref, ArrayData* ad,
|
|
StringData* key, TypedValue* value) {
|
|
return array_setm<StringData*, true, false, false>(ref, ad, key, value);
|
|
}
|
|
|
|
/**
|
|
* Append.
|
|
*
|
|
* array_setm_wk1_v0 --
|
|
* $a[] = <polymorphic value>
|
|
* ... but don't count the reference to the new value.
|
|
*/
|
|
ArrayData* array_setm_wk1_v0(ArrayData* ad, TypedValue* value) {
|
|
return HphpArray::AddNewElemC(ad, *value);
|
|
}
|
|
|
|
/**
|
|
* Array runtime helpers. For code-sharing purposes, all of these handle as
|
|
* much ref-counting machinery as possible. They differ by -arity, type
|
|
* signature, and necessity of various costly checks.
|
|
*
|
|
* They return the array that was just passed in as a convenience to
|
|
* callers, which may have "lost" the array in volatile registers before
|
|
* calling.
|
|
*/
|
|
|
|
ArrayData*
|
|
array_getm_i(void* dptr, int64_t key, TypedValue* out) {
|
|
assert(dptr);
|
|
ArrayData* ad = (ArrayData*)dptr;
|
|
TRACE(2, "array_getm_ik1: (%p) <- %p[%" PRId64 "]\n", out, dptr, key);
|
|
// Ref-counting the value is the translator's responsibility. We know out
|
|
// pointed to uninitialized memory, so no need to dec it.
|
|
TypedValue* ret = ad->nvGetCell(key);
|
|
tvDup(ret, out);
|
|
return ad;
|
|
}
|
|
|
|
NEVER_INLINE
|
|
ArrayData* array_getm_s(ArrayData* ad, StringData* sd, TypedValue* out,
|
|
int flags) {
|
|
bool drKey = flags & DecRefKey;
|
|
bool checkInts = flags & CheckInts;
|
|
int64_t ikey;
|
|
TypedValue* ret = checkInts && UNLIKELY(sd->isStrictlyInteger(ikey)) ?
|
|
ad->nvGetCell(ikey) :
|
|
ad->nvGetCell(sd);
|
|
tvDup((ret), (out));
|
|
if (drKey) decRefStr(sd);
|
|
TRACE(2, "%s: (%p) <- %p[\"%s\"@sd%p]\n", __FUNCTION__,
|
|
out, ad, sd->data(), sd);
|
|
return ad;
|
|
}
|
|
|
|
// issetm's DNA.
|
|
static bool
|
|
issetMUnary(const void* dptr, StringData* sd, bool decRefKey, bool checkInt) {
|
|
const ArrayData* ad = (const ArrayData*)dptr;
|
|
bool retval;
|
|
int64_t keyAsInt;
|
|
|
|
TypedValue* c;
|
|
if (checkInt && sd->isStrictlyInteger(keyAsInt)) {
|
|
c = ad->nvGet(keyAsInt);
|
|
} else {
|
|
c = ad->nvGet(sd);
|
|
}
|
|
const Variant* cell = &tvAsVariant(c);
|
|
|
|
retval = cell && !cell->isNull();
|
|
TRACE(2, "issetMUnary: %p[\"%s\"@sd%p] -> %d\n",
|
|
dptr, sd->data(), sd, retval);
|
|
|
|
if (decRefKey) decRefStr(sd);
|
|
return retval;
|
|
}
|
|
|
|
uint64_t array_issetm_s(const void* dptr, StringData* sd)
|
|
{ return issetMUnary(dptr, sd, true /*decRefKey*/, true /*checkInt*/); }
|
|
uint64_t array_issetm_s0(const void* dptr, StringData* sd)
|
|
{ return issetMUnary(dptr, sd, false /*decRefKey*/, true /*checkInt*/); }
|
|
uint64_t array_issetm_s_fast(const void* dptr, StringData* sd)
|
|
{ return issetMUnary(dptr, sd, true /*decRefKey*/, false /*checkInt*/); }
|
|
uint64_t array_issetm_s0_fast(const void* dptr, StringData* sd)
|
|
{ return issetMUnary(dptr, sd, false /*decRefKey*/, false /*checkInt*/); }
|
|
|
|
uint64_t array_issetm_i(const void* dptr, int64_t key) {
|
|
ArrayData* ad = (ArrayData*)dptr;
|
|
TypedValue* ret = ad->nvGet(key);
|
|
// Variant.isNull unboxes ret if its KindOfRef.
|
|
return ret && !tvAsCVarRef(ret).isNull();
|
|
}
|
|
|
|
ArrayData* array_add(ArrayData* a1, ArrayData* a2) {
|
|
if (!a2->empty()) {
|
|
if (a1->empty()) {
|
|
decRefArr(a1);
|
|
return a2;
|
|
}
|
|
if (a1 != a2) {
|
|
ArrayData *escalated = a1->append(a2, ArrayData::Plus,
|
|
a1->getCount() > 1);
|
|
if (escalated != a1) {
|
|
escalated->incRefCount();
|
|
decRefArr(a2);
|
|
decRefArr(a1);
|
|
return escalated;
|
|
}
|
|
}
|
|
}
|
|
decRefArr(a2);
|
|
return a1;
|
|
}
|
|
|
|
}
|
|
|
|
//=============================================================================
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
}
|