blob: 2654f4614e75b75e2cd906514076dad0f8501966 [file] [log] [blame]
/*
* Copyright (C) 2015 Apple Inc. All Rights Reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include "config.h"
#include "VariableEnvironment.h"
#include <wtf/CommaPrinter.h>
#include <wtf/HexNumber.h>
#include <wtf/text/UniquedStringImpl.h>
namespace JSC {
void VariableEnvironmentEntry::dump(PrintStream& out) const
{
out.print(hex(m_bits));
}
VariableEnvironment& VariableEnvironment::operator=(const VariableEnvironment& other)
{
VariableEnvironment env(other);
swap(env);
return *this;
}
void VariableEnvironment::markVariableAsCapturedIfDefined(const RefPtr<UniquedStringImpl>& identifier)
{
auto findResult = m_map.find(identifier);
if (findResult != m_map.end())
findResult->value.setIsCaptured();
}
void VariableEnvironment::markVariableAsCaptured(const RefPtr<UniquedStringImpl>& identifier)
{
auto findResult = m_map.find(identifier);
RELEASE_ASSERT(findResult != m_map.end());
findResult->value.setIsCaptured();
}
void VariableEnvironment::markAllVariablesAsCaptured()
{
if (m_isEverythingCaptured)
return;
m_isEverythingCaptured = true; // For fast queries.
// We must mark every entry as captured for when we iterate through m_map and entry.isCaptured() is called.
for (auto& value : m_map.values())
value.setIsCaptured();
}
bool VariableEnvironment::hasCapturedVariables() const
{
if (m_isEverythingCaptured)
return size() > 0;
for (auto& value : m_map.values()) {
if (value.isCaptured())
return true;
}
return false;
}
bool VariableEnvironment::captures(UniquedStringImpl* identifier) const
{
if (m_isEverythingCaptured)
return true;
auto findResult = m_map.find(identifier);
if (findResult == m_map.end())
return false;
return findResult->value.isCaptured();
}
void VariableEnvironment::swap(VariableEnvironment& other)
{
m_map.swap(other.m_map);
m_isEverythingCaptured = other.m_isEverythingCaptured;
m_rareData.swap(other.m_rareData);
}
void VariableEnvironment::markVariableAsImported(const RefPtr<UniquedStringImpl>& identifier)
{
auto findResult = m_map.find(identifier);
RELEASE_ASSERT(findResult != m_map.end());
findResult->value.setIsImported();
}
void VariableEnvironment::markVariableAsExported(const RefPtr<UniquedStringImpl>& identifier)
{
auto findResult = m_map.find(identifier);
RELEASE_ASSERT(findResult != m_map.end());
findResult->value.setIsExported();
}
VariableEnvironment::Map::AddResult VariableEnvironment::declarePrivateField(const RefPtr<UniquedStringImpl>& identifier)
{
getOrAddPrivateName(identifier.get());
auto entry = VariableEnvironmentEntry();
entry.setIsPrivateField();
entry.setIsConst();
entry.setIsCaptured();
return m_map.add(identifier, entry);
}
VariableEnvironment::PrivateDeclarationResult VariableEnvironment::declarePrivateAccessor(const RefPtr<UniquedStringImpl>& identifier, PrivateNameEntry accessorTraits)
{
if (!m_rareData)
m_rareData = WTF::makeUnique<VariableEnvironment::RareData>();
auto findResult = m_rareData->m_privateNames.find(identifier);
if (findResult == m_rareData->m_privateNames.end()) {
PrivateNameEntry meta(accessorTraits.bits());
auto entry = VariableEnvironmentEntry();
if (accessorTraits.isSetter())
entry.setIsPrivateSetter();
else {
ASSERT(accessorTraits.isGetter());
entry.setIsPrivateGetter();
}
entry.setIsConst();
entry.setIsCaptured();
m_map.add(identifier, entry);
m_rareData->m_privateNames.add(identifier, meta);
return PrivateDeclarationResult::Success;
}
PrivateNameEntry currentEntry = findResult->value;
if ((accessorTraits.isSetter() && !currentEntry.isGetter())
|| (accessorTraits.isGetter() && !currentEntry.isSetter()))
return PrivateDeclarationResult::DuplicatedName;
if (accessorTraits.isStatic() != currentEntry.isStatic())
return PrivateDeclarationResult::InvalidStaticNonStatic;
PrivateNameEntry meta(currentEntry.bits() | accessorTraits.bits());
m_rareData->m_privateNames.set(identifier, meta);
auto entryIterator = m_map.find(identifier);
ASSERT(entryIterator != m_map.end());
if (accessorTraits.isSetter())
entryIterator->value.setIsPrivateSetter();
else {
ASSERT(accessorTraits.isGetter());
entryIterator->value.setIsPrivateGetter();
}
return PrivateDeclarationResult::Success;
}
VariableEnvironment::PrivateDeclarationResult VariableEnvironment::declarePrivateSetter(const RefPtr<UniquedStringImpl>& identifier, PrivateNameEntry::Traits modifierTraits)
{
return declarePrivateAccessor(identifier, PrivateNameEntry(PrivateNameEntry::Traits::IsSetter | modifierTraits));
}
VariableEnvironment::PrivateDeclarationResult VariableEnvironment::declarePrivateGetter(const RefPtr<UniquedStringImpl>& identifier, PrivateNameEntry::Traits modifierTraits)
{
return declarePrivateAccessor(identifier, PrivateNameEntry(PrivateNameEntry::Traits::IsGetter | modifierTraits));
}
bool VariableEnvironment::declarePrivateMethod(const RefPtr<UniquedStringImpl>& identifier, PrivateNameEntry::Traits addionalTraits)
{
if (!m_rareData)
m_rareData = makeUnique<VariableEnvironment::RareData>();
auto findResult = m_rareData->m_privateNames.find(identifier);
if (findResult == m_rareData->m_privateNames.end()) {
PrivateNameEntry meta(PrivateNameEntry::Traits::IsMethod | addionalTraits);
auto entry = VariableEnvironmentEntry();
entry.setIsPrivateMethod();
entry.setIsConst();
entry.setIsCaptured();
m_map.add(identifier, entry);
auto addResult = m_rareData->m_privateNames.add(identifier, meta);
return addResult.isNewEntry;
}
return false; // Error: declaring a duplicate private name.
}
void VariableEnvironment::dump(PrintStream& out) const
{
CommaPrinter comma(", ");
for (auto& pair : m_map)
out.print(comma, pair.key, " => ", pair.value);
}
void CompactTDZEnvironment::sortCompact(Compact& compact)
{
std::sort(compact.begin(), compact.end(), [] (auto& a, auto& b) {
return a.get() < b.get();
});
}
CompactTDZEnvironment::CompactTDZEnvironment(const TDZEnvironment& env)
{
m_hash = 0; // Note: XOR is commutative so order doesn't matter here.
Compact variables = WTF::map(env, [this](auto& key) -> PackedRefPtr<UniquedStringImpl> {
m_hash ^= key->hash();
return key.get();
});
sortCompact(variables);
m_variables = WTFMove(variables);
}
bool CompactTDZEnvironment::operator==(const CompactTDZEnvironment& other) const
{
if (this == &other)
return true;
if (m_hash != other.m_hash)
return false;
auto equal = [&] (const Compact& compact, const Inflated& inflated) {
if (compact.size() != inflated.size())
return false;
for (auto& ident : compact) {
if (!inflated.contains(ident))
return false;
}
return true;
};
bool result;
WTF::switchOn(m_variables,
[&] (const Compact& compact) {
WTF::switchOn(other.m_variables,
[&] (const Compact& otherCompact) {
result = compact == otherCompact;
},
[&] (const Inflated& otherInflated) {
result = equal(compact, otherInflated);
});
},
[&] (const Inflated& inflated) {
WTF::switchOn(other.m_variables,
[&] (const Compact& otherCompact) {
result = equal(otherCompact, inflated);
},
[&] (const Inflated& otherInflated) {
result = inflated == otherInflated;
});
});
return result;
}
TDZEnvironment& CompactTDZEnvironment::toTDZEnvironmentSlow() const
{
Inflated inflated;
{
auto& compact = std::get<Compact>(m_variables);
for (size_t i = 0; i < compact.size(); ++i) {
auto addResult = inflated.add(compact[i]);
ASSERT_UNUSED(addResult, addResult.isNewEntry);
}
}
m_variables = Variables(WTFMove(inflated));
return const_cast<Inflated&>(std::get<Inflated>(m_variables));
}
CompactTDZEnvironmentMap::Handle CompactTDZEnvironmentMap::get(const TDZEnvironment& env)
{
auto* environment = new CompactTDZEnvironment(env);
bool isNewEntry;
auto handle = get(environment, isNewEntry);
if (!isNewEntry)
delete environment;
return handle;
}
CompactTDZEnvironmentMap::Handle CompactTDZEnvironmentMap::get(CompactTDZEnvironment* environment, bool& isNewEntry)
{
CompactTDZEnvironmentKey key { *environment };
auto addResult = m_map.add(key, 1);
isNewEntry = addResult.isNewEntry;
if (addResult.isNewEntry)
return CompactTDZEnvironmentMap::Handle(*environment, *this);
++addResult.iterator->value;
return CompactTDZEnvironmentMap::Handle(addResult.iterator->key.environment(), *this);
}
CompactTDZEnvironmentMap::Handle::~Handle()
{
if (!m_map) {
ASSERT(!m_environment);
// This happens if we were moved into a different handle.
return;
}
RELEASE_ASSERT(m_environment);
auto iter = m_map->m_map.find(CompactTDZEnvironmentKey { *m_environment });
RELEASE_ASSERT(iter != m_map->m_map.end());
--iter->value;
if (!iter->value) {
ASSERT(m_environment == &iter->key.environment());
m_map->m_map.remove(iter);
delete m_environment;
}
}
CompactTDZEnvironmentMap::Handle::Handle(const CompactTDZEnvironmentMap::Handle& other)
: m_environment(other.m_environment)
, m_map(other.m_map)
{
if (m_map) {
auto iter = m_map->m_map.find(CompactTDZEnvironmentKey { *m_environment });
RELEASE_ASSERT(iter != m_map->m_map.end());
++iter->value;
}
}
CompactTDZEnvironmentMap::Handle::Handle(CompactTDZEnvironment& environment, CompactTDZEnvironmentMap& map)
: m_environment(&environment)
, m_map(&map)
{
}
} // namespace JSC