mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-12-08 05:52:32 +00:00
549fb2406d
--HG-- extra : rebase_source : 3472eb981d0c8778be318b3d51b51edfe8c1c96a
181 lines
4.0 KiB
C++
181 lines
4.0 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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 "JSStreamWriter.h"
|
|
|
|
#include "mozilla/ArrayUtils.h" // for ArrayLength
|
|
#include "nsDataHashtable.h"
|
|
#include "nsString.h"
|
|
#include "nsTArray.h"
|
|
#include "nsUTF8Utils.h"
|
|
|
|
#if _MSC_VER
|
|
#define snprintf _snprintf
|
|
#endif
|
|
|
|
#define ARRAY (void*)1
|
|
#define OBJECT (void*)2
|
|
|
|
// Escape a UTF8 string to a stream. When an illegal encoding
|
|
// is found it will insert "INVALID" and the function will return.
|
|
static void EscapeToStream(std::ostream& stream, const char* str) {
|
|
stream << "\"";
|
|
|
|
size_t len = strlen(str);
|
|
const char* end = &str[len];
|
|
while (str < end) {
|
|
bool err;
|
|
const char* utf8CharStart = str;
|
|
uint32_t ucs4Char = UTF8CharEnumerator::NextChar(&str, end, &err);
|
|
|
|
if (err) {
|
|
// Encoding error
|
|
stream << "INVALID\"";
|
|
return;
|
|
}
|
|
|
|
// See http://www.ietf.org/rfc/rfc4627.txt?number=4627
|
|
// characters that must be escaped: quotation mark,
|
|
// reverse solidus, and the control characters
|
|
// (U+0000 through U+001F).
|
|
if (ucs4Char == '\"') {
|
|
stream << "\\\"";
|
|
} else if (ucs4Char == '\\') {
|
|
stream << "\\\\";
|
|
} else if (ucs4Char > 0xFF) {
|
|
char16_t chr[2];
|
|
ConvertUTF8toUTF16 encoder(chr);
|
|
encoder.write(utf8CharStart, uint32_t(str-utf8CharStart));
|
|
char escChar[13];
|
|
snprintf(escChar, mozilla::ArrayLength(escChar), "\\u%04X\\u%04X", chr[0], chr[1]);
|
|
stream << escChar;
|
|
} else if (ucs4Char < 0x1F || ucs4Char > 0xFF) {
|
|
char escChar[7];
|
|
snprintf(escChar, mozilla::ArrayLength(escChar), "\\u%04X", ucs4Char);
|
|
stream << escChar;
|
|
} else {
|
|
stream << char(ucs4Char);
|
|
}
|
|
}
|
|
stream << "\"";
|
|
}
|
|
|
|
JSStreamWriter::JSStreamWriter(std::ostream& aStream)
|
|
: mStream(aStream)
|
|
, mNeedsComma(false)
|
|
, mNeedsName(false)
|
|
{ }
|
|
|
|
JSStreamWriter::~JSStreamWriter()
|
|
{
|
|
MOZ_ASSERT(mStack.GetSize() == 0);
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::BeginObject()
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == ARRAY) {
|
|
mStream << ",";
|
|
}
|
|
mStream << "{";
|
|
mNeedsComma = false;
|
|
mNeedsName = true;
|
|
mStack.Push(OBJECT);
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::EndObject()
|
|
{
|
|
MOZ_ASSERT(mStack.Peek() == OBJECT);
|
|
mStream << "}";
|
|
mNeedsComma = true;
|
|
mNeedsName = false;
|
|
mStack.Pop();
|
|
if (mStack.GetSize() > 0 && mStack.Peek() == OBJECT) {
|
|
mNeedsName = true;
|
|
}
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::BeginArray()
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == ARRAY) {
|
|
mStream << ",";
|
|
}
|
|
mStream << "[";
|
|
mNeedsComma = false;
|
|
mStack.Push(ARRAY);
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::EndArray()
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
MOZ_ASSERT(mStack.Peek() == ARRAY);
|
|
mStream << "]";
|
|
mNeedsComma = true;
|
|
mStack.Pop();
|
|
if (mStack.GetSize() > 0 && mStack.Peek() == OBJECT) {
|
|
mNeedsName = true;
|
|
}
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::Name(const char *aName)
|
|
{
|
|
MOZ_ASSERT(mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == OBJECT) {
|
|
mStream << ",";
|
|
}
|
|
EscapeToStream(mStream, aName);
|
|
mStream << ":";
|
|
mNeedsName = false;
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::Value(int aValue)
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == ARRAY) {
|
|
mStream << ",";
|
|
}
|
|
mStream << aValue;
|
|
mNeedsComma = true;
|
|
if (mStack.Peek() == OBJECT) {
|
|
mNeedsName = true;
|
|
}
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::Value(double aValue)
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == ARRAY) {
|
|
mStream << ",";
|
|
}
|
|
mStream.precision(18);
|
|
mStream << aValue;
|
|
mNeedsComma = true;
|
|
if (mStack.Peek() == OBJECT) {
|
|
mNeedsName = true;
|
|
}
|
|
}
|
|
|
|
void
|
|
JSStreamWriter::Value(const char *aValue)
|
|
{
|
|
MOZ_ASSERT(!mNeedsName);
|
|
if (mNeedsComma && mStack.Peek() == ARRAY) {
|
|
mStream << ",";
|
|
}
|
|
EscapeToStream(mStream, aValue);
|
|
mNeedsComma = true;
|
|
if (mStack.Peek() == OBJECT) {
|
|
mNeedsName = true;
|
|
}
|
|
}
|