#include "config.h"
#include "UString.h"
#include "JSGlobalObjectFunctions.h"
#include "Collector.h"
#include "dtoa.h"
#include "Identifier.h"
#include "Operations.h"
#include <ctype.h>
#include <limits.h>
#include <limits>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <wtf/ASCIICType.h>
#include <wtf/Assertions.h>
#include <wtf/MathExtras.h>
#include <wtf/StringExtras.h>
#include <wtf/Vector.h>
#include <wtf/unicode/UTF8.h>
#include <wtf/StringExtras.h>
#if HAVE(STRINGS_H)
#include <strings.h>
#endif
using namespace WTF;
using namespace WTF::Unicode;
using namespace std;
namespace JSC {
extern const double NaN;
extern const double Inf;
UString* UString::s_nullUString;
void initializeUString()
{
UStringImpl::empty();
UString::s_nullUString = new UString;
}
UString::UString(const char* c)
: m_rep(Rep::create(c))
{
}
UString::UString(const char* c, unsigned length)
: m_rep(Rep::create(c, length))
{
}
UString::UString(const UChar* c, unsigned length)
: m_rep(Rep::create(c, length))
{
}
UString UString::from(int i)
{
UChar buf[1 + sizeof(i) * 3];
UChar* end = buf + sizeof(buf) / sizeof(UChar);
UChar* p = end;
if (i == 0)
*--p = '0';
else if (i == INT_MIN) {
char minBuf[1 + sizeof(i) * 3];
snprintf(minBuf, sizeof(minBuf), "%d", INT_MIN);
return UString(minBuf);
} else {
bool negative = false;
if (i < 0) {
negative = true;
i = -i;
}
while (i) {
*--p = static_cast<unsigned short>((i % 10) + '0');
i /= 10;
}
if (negative)
*--p = '-';
}
return UString(p, static_cast<unsigned>(end - p));
}
UString UString::from(long long i)
{
UChar buf[1 + sizeof(i) * 3];
UChar* end = buf + sizeof(buf) / sizeof(UChar);
UChar* p = end;
if (i == 0)
*--p = '0';
else if (i == std::numeric_limits<long long>::min()) {
char minBuf[1 + sizeof(i) * 3];
snprintf(minBuf, sizeof(minBuf), "%lld", std::numeric_limits<long long>::min());
return UString(minBuf);
} else {
bool negative = false;
if (i < 0) {
negative = true;
i = -i;
}
while (i) {
*--p = static_cast<unsigned short>((i % 10) + '0');
i /= 10;
}
if (negative)
*--p = '-';
}
return UString(p, static_cast<unsigned>(end - p));
}
UString UString::from(unsigned u)
{
UChar buf[sizeof(u) * 3];
UChar* end = buf + sizeof(buf) / sizeof(UChar);
UChar* p = end;
if (u == 0)
*--p = '0';
else {
while (u) {
*--p = static_cast<unsigned short>((u % 10) + '0');
u /= 10;
}
}
return UString(p, static_cast<unsigned>(end - p));
}
UString UString::from(long l)
{
UChar buf[1 + sizeof(l) * 3];
UChar* end = buf + sizeof(buf) / sizeof(UChar);
UChar* p = end;
if (l == 0)
*--p = '0';
else if (l == LONG_MIN) {
char minBuf[1 + sizeof(l) * 3];
snprintf(minBuf, sizeof(minBuf), "%ld", LONG_MIN);
return UString(minBuf);
} else {
bool negative = false;
if (l < 0) {
negative = true;
l = -l;
}
while (l) {
*--p = static_cast<unsigned short>((l % 10) + '0');
l /= 10;
}
if (negative)
*--p = '-';
}
return UString(p, end - p);
}
UString UString::from(double d)
{
DtoaBuffer buffer;
unsigned length;
doubleToStringInJavaScriptFormat(d, buffer, &length);
return UString(buffer, length);
}
char* UString::ascii() const
{
static char* asciiBuffer = 0;
unsigned length = size();
unsigned neededSize = length + 1;
delete[] asciiBuffer;
asciiBuffer = new char[neededSize];
const UChar* p = data();
char* q = asciiBuffer;
const UChar* limit = p + length;
while (p != limit) {
*q = static_cast<char>(p[0]);
++p;
++q;
}
*q = '\0';
return asciiBuffer;
}
bool UString::is8Bit() const
{
const UChar* u = data();
const UChar* limit = u + size();
while (u < limit) {
if (u[0] > 0xFF)
return false;
++u;
}
return true;
}
UChar UString::operator[](unsigned pos) const
{
if (pos >= size())
return '\0';
return data()[pos];
}
double UString::toDouble(bool tolerateTrailingJunk, bool tolerateEmptyString) const
{
if (size() == 1) {
UChar c = data()[0];
if (isASCIIDigit(c))
return c - '0';
if (isASCIISpace(c) && tolerateEmptyString)
return 0;
return NaN;
}
CString s = UTF8String();
if (s.isNull())
return NaN;
const char* c = s.data();
while (isASCIISpace(*c))
c++;
if (*c == '\0')
return tolerateEmptyString ? 0.0 : NaN;
double d;
if (*c == '0' && (*(c + 1) == 'x' || *(c + 1) == 'X')) {
const char* firstDigitPosition = c + 2;
c++;
d = 0.0;
while (*(++c)) {
if (*c >= '0' && *c <= '9')
d = d * 16.0 + *c - '0';
else if ((*c >= 'A' && *c <= 'F') || (*c >= 'a' && *c <= 'f'))
d = d * 16.0 + (*c & 0xdf) - 'A' + 10.0;
else
break;
}
if (d >= mantissaOverflowLowerBound)
d = parseIntOverflow(firstDigitPosition, c - firstDigitPosition, 16);
} else {
char* end;
d = WTF::strtod(c, &end);
if ((d != 0.0 || end != c) && d != Inf && d != -Inf) {
c = end;
} else {
double sign = 1.0;
if (*c == '+')
c++;
else if (*c == '-') {
sign = -1.0;
c++;
}
if (c[0] == 'I' && c[1] == 'n' && c[2] == 'f' && c[3] == 'i' && c[4] == 'n' && c[5] == 'i' && c[6] == 't' && c[7] == 'y') {
d = sign * Inf;
c += 8;
} else if ((d == Inf || d == -Inf) && *c != 'I' && *c != 'i')
c = end;
else
return NaN;
}
}
while (isASCIISpace(*c))
c++;
if (!tolerateTrailingJunk && *c != '\0')
d = NaN;
return d;
}
double UString::toDouble(bool tolerateTrailingJunk) const
{
return toDouble(tolerateTrailingJunk, true);
}
double UString::toDouble() const
{
return toDouble(false, true);
}
uint32_t UString::toUInt32(bool* ok) const
{
double d = toDouble();
bool b = true;
if (d != static_cast<uint32_t>(d)) {
b = false;
d = 0;
}
if (ok)
*ok = b;
return static_cast<uint32_t>(d);
}
uint32_t UString::toUInt32(bool* ok, bool tolerateEmptyString) const
{
double d = toDouble(false, tolerateEmptyString);
bool b = true;
if (d != static_cast<uint32_t>(d)) {
b = false;
d = 0;
}
if (ok)
*ok = b;
return static_cast<uint32_t>(d);
}
uint32_t UString::toStrictUInt32(bool* ok) const
{
if (ok)
*ok = false;
unsigned len = m_rep->length();
if (len == 0)
return 0;
const UChar* p = m_rep->characters();
unsigned short c = p[0];
if (c == '0') {
if (len == 1 && ok)
*ok = true;
return 0;
}
uint32_t i = 0;
while (1) {
if (c < '0' || c > '9')
return 0;
const unsigned d = c - '0';
if (i > 0xFFFFFFFFU / 10)
return 0;
i *= 10;
const unsigned max = 0xFFFFFFFFU - d;
if (i > max)
return 0;
i += d;
if (--len == 0) {
if (ok)
*ok = true;
return i;
}
c = *(++p);
}
}
unsigned UString::find(const UString& f, unsigned pos) const
{
unsigned fsz = f.size();
if (fsz == 1) {
UChar ch = f[0];
const UChar* end = data() + size();
for (const UChar* c = data() + pos; c < end; c++) {
if (*c == ch)
return static_cast<unsigned>(c - data());
}
return NotFound;
}
unsigned sz = size();
if (sz < fsz)
return NotFound;
if (fsz == 0)
return pos;
const UChar* end = data() + sz - fsz;
unsigned fsizeminusone = (fsz - 1) * sizeof(UChar);
const UChar* fdata = f.data();
unsigned short fchar = fdata[0];
++fdata;
for (const UChar* c = data() + pos; c <= end; c++) {
if (c[0] == fchar && !memcmp(c + 1, fdata, fsizeminusone))
return static_cast<unsigned>(c - data());
}
return NotFound;
}
unsigned UString::find(UChar ch, unsigned pos) const
{
const UChar* end = data() + size();
for (const UChar* c = data() + pos; c < end; c++) {
if (*c == ch)
return static_cast<unsigned>(c - data());
}
return NotFound;
}
unsigned UString::rfind(const UString& f, unsigned pos) const
{
unsigned sz = size();
unsigned fsz = f.size();
if (sz < fsz)
return NotFound;
if (pos > sz - fsz)
pos = sz - fsz;
if (fsz == 0)
return pos;
unsigned fsizeminusone = (fsz - 1) * sizeof(UChar);
const UChar* fdata = f.data();
for (const UChar* c = data() + pos; c >= data(); c--) {
if (*c == *fdata && !memcmp(c + 1, fdata + 1, fsizeminusone))
return static_cast<unsigned>(c - data());
}
return NotFound;
}
unsigned UString::rfind(UChar ch, unsigned pos) const
{
if (isEmpty())
return NotFound;
if (pos + 1 >= size())
pos = size() - 1;
for (const UChar* c = data() + pos; c >= data(); c--) {
if (*c == ch)
return static_cast<unsigned>(c - data());
}
return NotFound;
}
UString UString::substr(unsigned pos, unsigned len) const
{
unsigned s = size();
if (pos >= s)
pos = s;
unsigned limit = s - pos;
if (len > limit)
len = limit;
if (pos == 0 && len == s)
return *this;
return UString(Rep::create(m_rep, pos, len));
}
bool operator==(const UString& s1, const char *s2)
{
if (s2 == 0)
return s1.isEmpty();
const UChar* u = s1.data();
const UChar* uend = u + s1.size();
while (u != uend && *s2) {
if (u[0] != (unsigned char)*s2)
return false;
s2++;
u++;
}
return u == uend && *s2 == 0;
}
bool operator<(const UString& s1, const UString& s2)
{
const unsigned l1 = s1.size();
const unsigned l2 = s2.size();
const unsigned lmin = l1 < l2 ? l1 : l2;
const UChar* c1 = s1.data();
const UChar* c2 = s2.data();
unsigned l = 0;
while (l < lmin && *c1 == *c2) {
c1++;
c2++;
l++;
}
if (l < lmin)
return (c1[0] < c2[0]);
return (l1 < l2);
}
bool operator>(const UString& s1, const UString& s2)
{
const unsigned l1 = s1.size();
const unsigned l2 = s2.size();
const unsigned lmin = l1 < l2 ? l1 : l2;
const UChar* c1 = s1.data();
const UChar* c2 = s2.data();
unsigned l = 0;
while (l < lmin && *c1 == *c2) {
c1++;
c2++;
l++;
}
if (l < lmin)
return (c1[0] > c2[0]);
return (l1 > l2);
}
int compare(const UString& s1, const UString& s2)
{
const unsigned l1 = s1.size();
const unsigned l2 = s2.size();
const unsigned lmin = l1 < l2 ? l1 : l2;
const UChar* c1 = s1.data();
const UChar* c2 = s2.data();
unsigned l = 0;
while (l < lmin && *c1 == *c2) {
c1++;
c2++;
l++;
}
if (l < lmin)
return (c1[0] > c2[0]) ? 1 : -1;
if (l1 == l2)
return 0;
return (l1 > l2) ? 1 : -1;
}
CString UString::UTF8String(bool strict) const
{
const unsigned length = size();
Vector<char, 1024> buffer(length * 3);
char* p = buffer.data();
const UChar* d = reinterpret_cast<const UChar*>(&data()[0]);
ConversionResult result = convertUTF16ToUTF8(&d, d + length, &p, p + buffer.size(), strict);
if (result != conversionOK)
return CString();
return CString(buffer.data(), p - buffer.data());
}
}