JSStringJoiner.cpp [plain text]
#include "config.h"
#include "JSStringJoiner.h"
#include "JSCInlines.h"
namespace JSC {
template<typename CharacterType>
static inline void appendStringToData(CharacterType*& data, StringView string)
{
string.getCharactersWithUpconvert(data);
data += string.length();
}
template<typename CharacterType>
static inline String joinStrings(const Vector<StringViewWithUnderlyingString>& strings, StringView separator, unsigned joinedLength)
{
ASSERT(joinedLength);
CharacterType* data;
String result = StringImpl::tryCreateUninitialized(joinedLength, data);
if (result.isNull())
return result;
appendStringToData(data, strings[0].view);
unsigned size = strings.size();
switch (separator.length()) {
case 0:
for (unsigned i = 1; i < size; ++i)
appendStringToData(data, strings[i].view);
break;
case 1: {
CharacterType separatorCharacter = separator[0];
for (unsigned i = 1; i < size; ++i) {
*data++ = separatorCharacter;
appendStringToData(data, strings[i].view);
}
break;
}
default:
for (unsigned i = 1; i < size; ++i) {
appendStringToData(data, separator);
appendStringToData(data, strings[i].view);
}
}
ASSERT(data == result.characters<CharacterType>() + joinedLength);
return result;
}
inline unsigned JSStringJoiner::joinedLength(ExecState& state) const
{
unsigned numberOfStrings = m_strings.size();
if (!numberOfStrings)
return 0;
Checked<unsigned, RecordOverflow> separatorLength = m_separator.length();
Checked<unsigned, RecordOverflow> totalSeparatorsLength = separatorLength * (numberOfStrings - 1);
Checked<unsigned, RecordOverflow> totalLength = totalSeparatorsLength + m_accumulatedStringsLength;
unsigned result;
if (totalLength.safeGet(result) == CheckedState::DidOverflow) {
throwOutOfMemoryError(&state);
return 0;
}
return result;
}
JSValue JSStringJoiner::join(ExecState& state)
{
ASSERT(m_strings.size() <= m_strings.capacity());
unsigned length = joinedLength(state);
if (state.hadException())
return jsUndefined();
if (!length)
return jsEmptyString(&state);
String result;
if (m_isAll8Bit)
result = joinStrings<LChar>(m_strings, m_separator, length);
else
result = joinStrings<UChar>(m_strings, m_separator, length);
if (result.isNull())
return throwOutOfMemoryError(&state);
return jsString(&state, WTF::move(result));
}
}