Source code
Revision control
Copy as Markdown
Other Tools
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
* vim: set ts=8 sts=2 et sw=2 tw=80:
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
#include "util/Text.h"
#include "mozilla/Assertions.h"
#include "mozilla/Maybe.h"
#include "mozilla/PodOperations.h"
#include "mozilla/Utf8.h"
#include <stddef.h>
#include <stdint.h>
#include "frontend/FrontendContext.h" // frontend::FrontendContext
#include "gc/GC.h"
#include "js/GCAPI.h"
#include "js/Printer.h"
#include "js/Utility.h" // JS::FreePolicy
#include "util/Unicode.h"
#include "vm/JSContext.h"
#include "vm/StringType.h"
using namespace JS;
using namespace js;
using mozilla::DecodeOneUtf8CodePoint;
using mozilla::IsAscii;
using mozilla::Maybe;
using mozilla::PodCopy;
using mozilla::Utf8Unit;
template <typename CharT>
const CharT* js_strchr_limit(const CharT* s, char16_t c, const CharT* limit) {
while (s < limit) {
if (*s == c) {
return s;
}
s++;
}
return nullptr;
}
template const Latin1Char* js_strchr_limit(const Latin1Char* s, char16_t c,
const Latin1Char* limit);
template const char16_t* js_strchr_limit(const char16_t* s, char16_t c,
const char16_t* limit);
template <typename AllocT, typename CharT>
static UniquePtr<CharT[], JS::FreePolicy> DuplicateStringToArenaImpl(
arena_id_t destArenaId, AllocT* alloc, const CharT* s, size_t n) {
auto ret = alloc->template make_pod_arena_array<CharT>(destArenaId, n + 1);
if (!ret) {
return nullptr;
}
PodCopy(ret.get(), s, n);
ret[n] = '\0';
return ret;
}
UniqueChars js::DuplicateStringToArena(arena_id_t destArenaId, JSContext* cx,
const char* s, size_t n) {
return DuplicateStringToArenaImpl(destArenaId, cx, s, n);
}
static UniqueChars DuplicateStringToArena(arena_id_t destArenaId,
FrontendContext* fc, const char* s,
size_t n) {
return DuplicateStringToArenaImpl(destArenaId, fc->getAllocator(), s, n);
}
UniqueChars js::DuplicateStringToArena(arena_id_t destArenaId, JSContext* cx,
const char* s) {
return DuplicateStringToArena(destArenaId, cx, s, strlen(s));
}
static UniqueChars DuplicateStringToArena(arena_id_t destArenaId,
FrontendContext* fc, const char* s) {
return DuplicateStringToArena(destArenaId, fc, s, strlen(s));
}
UniqueLatin1Chars js::DuplicateStringToArena(arena_id_t destArenaId,
JSContext* cx,
const JS::Latin1Char* s,
size_t n) {
return DuplicateStringToArenaImpl(destArenaId, cx, s, n);
}
UniqueTwoByteChars js::DuplicateStringToArena(arena_id_t destArenaId,
JSContext* cx, const char16_t* s,
size_t n) {
return DuplicateStringToArenaImpl(destArenaId, cx, s, n);
}
static UniqueTwoByteChars DuplicateStringToArena(arena_id_t destArenaId,
FrontendContext* fc,
const char16_t* s, size_t n) {
return DuplicateStringToArenaImpl(destArenaId, fc->getAllocator(), s, n);
}
UniqueTwoByteChars js::DuplicateStringToArena(arena_id_t destArenaId,
JSContext* cx,
const char16_t* s) {
return DuplicateStringToArena(destArenaId, cx, s, js_strlen(s));
}
static UniqueTwoByteChars DuplicateStringToArena(arena_id_t destArenaId,
FrontendContext* fc,
const char16_t* s) {
return DuplicateStringToArena(destArenaId, fc, s, js_strlen(s));
}
UniqueChars js::DuplicateStringToArena(arena_id_t destArenaId, const char* s) {
return DuplicateStringToArena(destArenaId, s, strlen(s));
}
UniqueChars js::DuplicateStringToArena(arena_id_t destArenaId, const char* s,
size_t n) {
UniqueChars ret(js_pod_arena_malloc<char>(destArenaId, n + 1));
if (!ret) {
return nullptr;
}
PodCopy(ret.get(), s, n);
ret[n] = '\0';
return ret;
}
UniqueLatin1Chars js::DuplicateStringToArena(arena_id_t destArenaId,
const JS::Latin1Char* s,
size_t n) {
UniqueLatin1Chars ret(
js_pod_arena_malloc<JS::Latin1Char>(destArenaId, n + 1));
if (!ret) {
return nullptr;
}
PodCopy(ret.get(), s, n);
ret[n] = '\0';
return ret;
}
UniqueTwoByteChars js::DuplicateStringToArena(arena_id_t destArenaId,
const char16_t* s) {
return DuplicateStringToArena(destArenaId, s, js_strlen(s));
}
UniqueTwoByteChars js::DuplicateStringToArena(arena_id_t destArenaId,
const char16_t* s, size_t n) {
UniqueTwoByteChars ret(js_pod_arena_malloc<char16_t>(destArenaId, n + 1));
if (!ret) {
return nullptr;
}
PodCopy(ret.get(), s, n);
ret[n] = '\0';
return ret;
}
UniqueChars js::DuplicateString(JSContext* cx, const char* s, size_t n) {
return DuplicateStringToArena(js::MallocArena, cx, s, n);
}
UniqueChars js::DuplicateString(JSContext* cx, const char* s) {
return DuplicateStringToArena(js::MallocArena, cx, s);
}
UniqueChars js::DuplicateString(FrontendContext* fc, const char* s) {
return ::DuplicateStringToArena(js::MallocArena, fc, s);
}
UniqueLatin1Chars js::DuplicateString(JSContext* cx, const JS::Latin1Char* s,
size_t n) {
return DuplicateStringToArena(js::MallocArena, cx, s, n);
}
UniqueTwoByteChars js::DuplicateString(JSContext* cx, const char16_t* s) {
return DuplicateStringToArena(js::MallocArena, cx, s);
}
UniqueTwoByteChars js::DuplicateString(FrontendContext* fc, const char16_t* s) {
return ::DuplicateStringToArena(js::MallocArena, fc, s);
}
UniqueTwoByteChars js::DuplicateString(JSContext* cx, const char16_t* s,
size_t n) {
return DuplicateStringToArena(js::MallocArena, cx, s, n);
}
UniqueChars js::DuplicateString(const char* s) {
return DuplicateStringToArena(js::MallocArena, s);
}
UniqueChars js::DuplicateString(const char* s, size_t n) {
return DuplicateStringToArena(js::MallocArena, s, n);
}
UniqueLatin1Chars js::DuplicateString(const JS::Latin1Char* s, size_t n) {
return DuplicateStringToArena(js::MallocArena, s, n);
}
UniqueTwoByteChars js::DuplicateString(const char16_t* s) {
return DuplicateStringToArena(js::MallocArena, s);
}
UniqueTwoByteChars js::DuplicateString(const char16_t* s, size_t n) {
return DuplicateStringToArena(js::MallocArena, s, n);
}
char16_t* js::InflateString(JSContext* cx, const char* bytes, size_t length) {
char16_t* chars = cx->pod_malloc<char16_t>(length + 1);
if (!chars) {
return nullptr;
}
CopyAndInflateChars(chars, bytes, length);
chars[length] = '\0';
return chars;
}
/*
* Convert one UCS-4 char and write it into a UTF-8 buffer, which must be at
* least 4 bytes long. Return the number of UTF-8 bytes of data written.
*/
uint32_t js::OneUcs4ToUtf8Char(uint8_t* utf8Buffer, char32_t ucs4Char) {
MOZ_ASSERT(ucs4Char <= unicode::NonBMPMax);
if (ucs4Char < 0x80) {
utf8Buffer[0] = uint8_t(ucs4Char);
return 1;
}
uint32_t a = ucs4Char >> 11;
uint32_t utf8Length = 2;
while (a) {
a >>= 5;
utf8Length++;
}
MOZ_ASSERT(utf8Length <= 4);
uint32_t i = utf8Length;
while (--i) {
utf8Buffer[i] = uint8_t((ucs4Char & 0x3F) | 0x80);
ucs4Char >>= 6;
}
utf8Buffer[0] = uint8_t(0x100 - (1 << (8 - utf8Length)) + ucs4Char);
return utf8Length;
}
size_t js::PutEscapedStringImpl(char* buffer, size_t bufferSize,
GenericPrinter* out, JSLinearString* str,
uint32_t quote) {
size_t len = str->length();
AutoCheckCannotGC nogc;
return str->hasLatin1Chars()
? PutEscapedStringImpl(buffer, bufferSize, out,
str->latin1Chars(nogc), len, quote)
: PutEscapedStringImpl(buffer, bufferSize, out,
str->twoByteChars(nogc), len, quote);
}
template <typename CharT>
size_t js::PutEscapedStringImpl(char* buffer, size_t bufferSize,
GenericPrinter* out, const CharT* chars,
size_t length, uint32_t quote) {
enum {
STOP,
FIRST_QUOTE,
LAST_QUOTE,
CHARS,
ESCAPE_START,
ESCAPE_MORE
} state;
MOZ_ASSERT(quote == 0 || quote == '\'' || quote == '"');
MOZ_ASSERT_IF(!buffer, bufferSize == 0);
MOZ_ASSERT_IF(out, !buffer);
if (bufferSize == 0) {
buffer = nullptr;
} else {
bufferSize--;
}
const CharT* charsEnd = chars + length;
size_t n = 0;
state = FIRST_QUOTE;
unsigned shift = 0;
unsigned hex = 0;
unsigned u = 0;
char c = 0; /* to quell GCC warnings */
for (;;) {
switch (state) {
case STOP:
goto stop;
case FIRST_QUOTE:
state = CHARS;
goto do_quote;
case LAST_QUOTE:
state = STOP;
do_quote:
if (quote == 0) {
continue;
}
c = (char)quote;
break;
case CHARS:
if (chars == charsEnd) {
state = LAST_QUOTE;
continue;
}
u = *chars++;
if (u < ' ') {
if (u != 0) {
const char* escape = strchr(js_EscapeMap, (int)u);
if (escape) {
u = escape[1];
goto do_escape;
}
}
goto do_hex_escape;
}
if (u < 127) {
if (u == quote || u == '\\') {
goto do_escape;
}
c = (char)u;
} else if (u < 0x100) {
goto do_hex_escape;
} else {
shift = 16;
hex = u;
u = 'u';
goto do_escape;
}
break;
do_hex_escape:
shift = 8;
hex = u;
u = 'x';
do_escape:
c = '\\';
state = ESCAPE_START;
break;
case ESCAPE_START:
MOZ_ASSERT(' ' <= u && u < 127);
c = (char)u;
state = ESCAPE_MORE;
break;
case ESCAPE_MORE:
if (shift == 0) {
state = CHARS;
continue;
}
shift -= 4;
u = 0xF & (hex >> shift);
c = (char)(u + (u < 10 ? '0' : 'A' - 10));
break;
}
if (buffer) {
MOZ_ASSERT(n <= bufferSize);
if (n != bufferSize) {
buffer[n] = c;
} else {
buffer[n] = '\0';
buffer = nullptr;
}
} else if (out) {
out->put(&c, 1);
}
n++;
}
stop:
if (buffer) {
buffer[n] = '\0';
}
return n;
}
bool js::ContainsFlag(const char* str, const char* flag) {
size_t flaglen = strlen(flag);
const char* index = strstr(str, flag);
while (index) {
if ((index == str || index[-1] == ',') &&
(index[flaglen] == 0 || index[flaglen] == ',')) {
return true;
}
index = strstr(index + flaglen, flag);
}
return false;
}
template size_t js::PutEscapedStringImpl(char* buffer, size_t bufferSize,
GenericPrinter* out,
const Latin1Char* chars, size_t length,
uint32_t quote);
template size_t js::PutEscapedStringImpl(char* buffer, size_t bufferSize,
GenericPrinter* out, const char* chars,
size_t length, uint32_t quote);
template size_t js::PutEscapedStringImpl(char* buffer, size_t bufferSize,
GenericPrinter* out,
const char16_t* chars, size_t length,
uint32_t quote);
template size_t js::PutEscapedString(char* buffer, size_t bufferSize,
const Latin1Char* chars, size_t length,
uint32_t quote);
template size_t js::PutEscapedString(char* buffer, size_t bufferSize,
const char16_t* chars, size_t length,
uint32_t quote);
size_t js::unicode::CountUTF16CodeUnits(const Utf8Unit* begin,
const Utf8Unit* end) {
MOZ_ASSERT(begin <= end);
size_t count = 0;
const Utf8Unit* ptr = begin;
while (ptr < end) {
count++;
Utf8Unit lead = *ptr++;
if (IsAscii(lead)) {
continue;
}
Maybe<char32_t> cp = DecodeOneUtf8CodePoint(lead, &ptr, end);
MOZ_ASSERT(cp.isSome());
if (*cp > unicode::UTF16Max) {
// This uses surrogate pair.
count++;
}
}
MOZ_ASSERT(ptr == end, "bad code unit count in line?");
return count;
}