diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /js/src/util/Text.cpp | |
parent | Initial commit. (diff) | |
download | firefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz firefox-26a029d407be480d791972afb5975cf62c9360a6.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'js/src/util/Text.cpp')
-rw-r--r-- | js/src/util/Text.cpp | 441 |
1 files changed, 441 insertions, 0 deletions
diff --git a/js/src/util/Text.cpp b/js/src/util/Text.cpp new file mode 100644 index 0000000000..d2ac6dfaa4 --- /dev/null +++ b/js/src/util/Text.cpp @@ -0,0 +1,441 @@ +/* -*- 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 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#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; +} |