/* 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/. */

#ifndef ProfileBufferEntrySerializationGeckoExtensions_h
#define ProfileBufferEntrySerializationGeckoExtensions_h

#include "mozilla/ProfileBufferEntrySerialization.h"

#include "js/AllocPolicy.h"
#include "js/Utility.h"
#include "nsString.h"

namespace mozilla {

// ----------------------------------------------------------------------------
// ns[C]String

// nsString or nsCString contents are serialized as the number of bytes (encoded
// as ULEB128) and all the characters in the string. The terminal '\0' is
// omitted.
// Make sure you write and read with the same character type!
//
// Usage: `nsCString s = ...; aEW.WriteObject(s);`
template <typename CHAR>
struct ProfileBufferEntryWriter::Serializer<nsTString<CHAR>> {
  static Length Bytes(const nsTString<CHAR>& aS) {
    const auto length = aS.Length();
    return ProfileBufferEntryWriter::ULEB128Size(length) +
           static_cast<Length>(length * sizeof(CHAR));
  }

  static void Write(ProfileBufferEntryWriter& aEW, const nsTString<CHAR>& aS) {
    const auto length = aS.Length();
    aEW.WriteULEB128(length);
    // Copy the bytes from the string's buffer.
    aEW.WriteBytes(aS.Data(), length * sizeof(CHAR));
  }
};

template <typename CHAR>
struct ProfileBufferEntryReader::Deserializer<nsTString<CHAR>> {
  static void ReadInto(ProfileBufferEntryReader& aER, nsTString<CHAR>& aS) {
    aS = Read(aER);
  }

  static nsTString<CHAR> Read(ProfileBufferEntryReader& aER) {
    const Length length = aER.ReadULEB128<Length>();
    nsTString<CHAR> s;
    // BulkWrite is the most efficient way to copy bytes into the target string.
    auto writerOrErr = s.BulkWrite(length, 0, true);
    MOZ_RELEASE_ASSERT(!writerOrErr.isErr());

    auto writer = writerOrErr.unwrap();

    aER.ReadBytes(writer.Elements(), length * sizeof(CHAR));
    writer.Finish(length, true);
    return s;
  }
};

// ----------------------------------------------------------------------------
// nsAuto[C]String

// nsAuto[C]String contents are serialized as the number of bytes (encoded as
// ULEB128) and all the characters in the string. The terminal '\0' is omitted.
// Make sure you write and read with the same character type!
//
// Usage: `nsAutoCString s = ...; aEW.WriteObject(s);`
template <typename CHAR, size_t N>
struct ProfileBufferEntryWriter::Serializer<nsTAutoStringN<CHAR, N>> {
  static Length Bytes(const nsTAutoStringN<CHAR, N>& aS) {
    const auto length = aS.Length();
    return ProfileBufferEntryWriter::ULEB128Size(length) +
           static_cast<Length>(length * sizeof(CHAR));
  }

  static void Write(ProfileBufferEntryWriter& aEW,
                    const nsTAutoStringN<CHAR, N>& aS) {
    const auto length = aS.Length();
    aEW.WriteULEB128(length);
    // Copy the bytes from the string's buffer.
    aEW.WriteBytes(aS.BeginReading(), length * sizeof(CHAR));
  }
};

template <typename CHAR, size_t N>
struct ProfileBufferEntryReader::Deserializer<nsTAutoStringN<CHAR, N>> {
  static void ReadInto(ProfileBufferEntryReader& aER,
                       nsTAutoStringN<CHAR, N>& aS) {
    aS = Read(aER);
  }

  static nsTAutoStringN<CHAR, N> Read(ProfileBufferEntryReader& aER) {
    const auto length = aER.ReadULEB128<Length>();
    nsTAutoStringN<CHAR, N> s;
    // BulkWrite is the most efficient way to copy bytes into the target string.
    auto writerOrErr = s.BulkWrite(length, 0, true);
    MOZ_RELEASE_ASSERT(!writerOrErr.isErr());

    auto writer = writerOrErr.unwrap();
    aER.ReadBytes(writer.Elements(), length * sizeof(CHAR));
    writer.Finish(length, true);
    return s;
  }
};

// ----------------------------------------------------------------------------
// JS::UniqueChars

// JS::UniqueChars contents are serialized as the number of bytes (encoded as
// ULEB128) and all the characters in the string. The terminal '\0' is omitted.
// Note: A nullptr pointer will be serialized like an empty string, so when
// deserializing it will result in an allocated buffer only containing a
// single null terminator.
//
// Usage: `JS::UniqueChars s = ...; aEW.WriteObject(s);`
template <>
struct ProfileBufferEntryWriter::Serializer<JS::UniqueChars> {
  static Length Bytes(const JS::UniqueChars& aS) {
    if (!aS) {
      return ProfileBufferEntryWriter::ULEB128Size<Length>(0);
    }
    const auto len = static_cast<Length>(strlen(aS.get()));
    return ProfileBufferEntryWriter::ULEB128Size(len) + len;
  }

  static void Write(ProfileBufferEntryWriter& aEW, const JS::UniqueChars& aS) {
    if (!aS) {
      aEW.WriteULEB128<Length>(0);
      return;
    }
    const auto len = static_cast<Length>(strlen(aS.get()));
    aEW.WriteULEB128(len);
    aEW.WriteBytes(aS.get(), len);
  }
};

template <>
struct ProfileBufferEntryReader::Deserializer<JS::UniqueChars> {
  static void ReadInto(ProfileBufferEntryReader& aER, JS::UniqueChars& aS) {
    aS = Read(aER);
  }

  static JS::UniqueChars Read(ProfileBufferEntryReader& aER) {
    const auto len = aER.ReadULEB128<Length>();
    // Use the same allocation policy as JS_smprintf.
    char* buffer =
        static_cast<char*>(js::SystemAllocPolicy{}.pod_malloc<char>(len + 1));
    aER.ReadBytes(buffer, len);
    buffer[len] = '\0';
    return JS::UniqueChars(buffer);
  }
};

}  // namespace mozilla

#endif  // ProfileBufferEntrySerializationGeckoExtensions_h