StringConcatenate.h [plain text]
#pragma once
#include <string.h>
#include <wtf/CheckedArithmetic.h>
#include <wtf/text/AtomicString.h>
#include <wtf/text/StringView.h>
#ifndef WTF_STRINGTYPEADAPTER_COPIED_WTF_STRING
#define WTF_STRINGTYPEADAPTER_COPIED_WTF_STRING() ((void)0)
#endif
namespace WTF {
template<typename StringType, typename>
class StringTypeAdapter;
template<>
class StringTypeAdapter<char, void> {
public:
StringTypeAdapter(char character)
: m_character(character)
{
}
unsigned length() { return 1; }
bool is8Bit() { return true; }
void writeTo(LChar* destination) const
{
*destination = m_character;
}
void writeTo(UChar* destination) const
{
*destination = m_character;
}
String toString() const { return String(&m_character, 1); }
private:
char m_character;
};
template<>
class StringTypeAdapter<UChar, void> {
public:
StringTypeAdapter(UChar character)
: m_character(character)
{
}
unsigned length() const { return 1; }
bool is8Bit() const { return m_character <= 0xff; }
void writeTo(LChar* destination) const
{
ASSERT(is8Bit());
*destination = static_cast<LChar>(m_character);
}
void writeTo(UChar* destination) const
{
*destination = m_character;
}
String toString() const { return String(&m_character, 1); }
private:
UChar m_character;
};
template<>
class StringTypeAdapter<const LChar*, void> {
public:
StringTypeAdapter(const LChar* characters)
: m_characters(characters)
{
size_t length = strlen(reinterpret_cast<const char*>(characters));
RELEASE_ASSERT(length <= String::MaxLength);
m_length = static_cast<unsigned>(length);
}
unsigned length() const { return m_length; }
bool is8Bit() const { return true; }
void writeTo(LChar* destination) const
{
StringView(m_characters, m_length).getCharactersWithUpconvert(destination);
}
void writeTo(UChar* destination) const
{
StringView(m_characters, m_length).getCharactersWithUpconvert(destination);
}
String toString() const { return String(m_characters, m_length); }
private:
const LChar* m_characters;
unsigned m_length;
};
template<>
class StringTypeAdapter<const UChar*, void> {
public:
StringTypeAdapter(const UChar* characters)
: m_characters(characters)
{
size_t length = 0;
while (m_characters[length])
++length;
RELEASE_ASSERT(length <= String::MaxLength);
m_length = static_cast<unsigned>(length);
}
unsigned length() const { return m_length; }
bool is8Bit() const { return false; }
NO_RETURN_DUE_TO_CRASH void writeTo(LChar*) const
{
CRASH(); }
void writeTo(UChar* destination) const
{
memcpy(destination, m_characters, m_length * sizeof(UChar));
}
String toString() const { return String(m_characters, m_length); }
private:
const UChar* m_characters;
unsigned m_length;
};
template<>
class StringTypeAdapter<const char*, void> : public StringTypeAdapter<const LChar*, void> {
public:
StringTypeAdapter(const char* characters)
: StringTypeAdapter<const LChar*, void>(reinterpret_cast<const LChar*>(characters))
{
}
};
template<>
class StringTypeAdapter<char*, void> : public StringTypeAdapter<const char*, void> {
public:
StringTypeAdapter(const char* characters)
: StringTypeAdapter<const char*, void>(characters)
{
}
};
template<>
class StringTypeAdapter<ASCIILiteral, void> : public StringTypeAdapter<const char*, void> {
public:
StringTypeAdapter(ASCIILiteral characters)
: StringTypeAdapter<const char*, void>(characters)
{
}
};
template<>
class StringTypeAdapter<Vector<char>, void> {
public:
StringTypeAdapter(const Vector<char>& vector)
: m_vector(vector)
{
}
size_t length() const { return m_vector.size(); }
bool is8Bit() const { return true; }
void writeTo(LChar* destination) const
{
StringView(reinterpret_cast<const LChar*>(m_vector.data()), m_vector.size()).getCharactersWithUpconvert(destination);
}
void writeTo(UChar* destination) const
{
StringView(reinterpret_cast<const LChar*>(m_vector.data()), m_vector.size()).getCharactersWithUpconvert(destination);
}
String toString() const { return String(m_vector.data(), m_vector.size()); }
private:
const Vector<char>& m_vector;
};
template<>
class StringTypeAdapter<String, void> {
public:
StringTypeAdapter(const String& string)
: m_string(string)
{
}
unsigned length() const { return m_string.length(); }
bool is8Bit() const { return m_string.isNull() || m_string.is8Bit(); }
void writeTo(LChar* destination) const
{
StringView(m_string).getCharactersWithUpconvert(destination);
WTF_STRINGTYPEADAPTER_COPIED_WTF_STRING();
}
void writeTo(UChar* destination) const
{
StringView(m_string).getCharactersWithUpconvert(destination);
WTF_STRINGTYPEADAPTER_COPIED_WTF_STRING();
}
String toString() const { return m_string; }
private:
const String& m_string;
};
template<>
class StringTypeAdapter<AtomicString, void> : public StringTypeAdapter<String, void> {
public:
StringTypeAdapter(const AtomicString& string)
: StringTypeAdapter<String, void>(string.string())
{
}
};
template<typename Adapter>
inline bool are8Bit(Adapter adapter)
{
return adapter.is8Bit();
}
template<typename Adapter, typename... Adapters>
inline bool are8Bit(Adapter adapter, Adapters ...adapters)
{
return adapter.is8Bit() && are8Bit(adapters...);
}
template<typename ResultType, typename Adapter>
inline void makeStringAccumulator(ResultType* result, Adapter adapter)
{
adapter.writeTo(result);
}
template<typename ResultType, typename Adapter, typename... Adapters>
inline void makeStringAccumulator(ResultType* result, Adapter adapter, Adapters ...adapters)
{
adapter.writeTo(result);
makeStringAccumulator(result + adapter.length(), adapters...);
}
template<typename StringTypeAdapter, typename... StringTypeAdapters>
String tryMakeStringFromAdapters(StringTypeAdapter adapter, StringTypeAdapters ...adapters)
{
static_assert(String::MaxLength == std::numeric_limits<int32_t>::max(), "");
auto sum = checkedSum<int32_t>(adapter.length(), adapters.length()...);
if (sum.hasOverflowed())
return String();
unsigned length = sum.unsafeGet();
ASSERT(length <= String::MaxLength);
if (are8Bit(adapter, adapters...)) {
LChar* buffer;
RefPtr<StringImpl> resultImpl = StringImpl::tryCreateUninitialized(length, buffer);
if (!resultImpl)
return String();
makeStringAccumulator(buffer, adapter, adapters...);
return WTFMove(resultImpl);
}
UChar* buffer;
RefPtr<StringImpl> resultImpl = StringImpl::tryCreateUninitialized(length, buffer);
if (!resultImpl)
return String();
makeStringAccumulator(buffer, adapter, adapters...);
return WTFMove(resultImpl);
}
template<typename... StringTypes>
String tryMakeString(StringTypes ...strings)
{
return tryMakeStringFromAdapters(StringTypeAdapter<StringTypes>(strings)...);
}
template<typename StringType>
String makeString(StringType string)
{
return String(string);
}
template<typename... StringTypes>
String makeString(StringTypes... strings)
{
String result = tryMakeString(strings...);
if (!result)
CRASH();
return result;
}
}
using WTF::makeString;
using WTF::tryMakeString;
#include <wtf/text/StringOperators.h>