/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-

 * vim: set ts=4 sw=4 et 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 "JavaScriptShared.h"

#include "mozilla/dom/BindingUtils.h"

#include "mozilla/dom/CPOWManagerGetter.h"

#include "mozilla/dom/TabChild.h"

#include "jsfriendapi.h"

#include "xpcprivate.h"

#include "WrapperFactory.h"

#include "mozilla/Preferences.h"

using namespace js;

using namespace JS;

using namespace mozilla;

using namespace mozilla::jsipc;

IdToObjectMap::IdToObjectMap()

  : table_(SystemAllocPolicy())

{

}

bool

IdToObjectMap::init()

{

    if (table_.initialized())

        return true;

    return table_.init(32);

}

void

IdToObjectMap::trace(JSTracer *trc)

{

    for (Table::Range r(table_.all()); !r.empty(); r.popFront()) {

        DebugOnly<JSObject *> prior = r.front().value().get();

        JS_CallObjectTracer(trc, &r.front().value(), "ipc-object");

    }

}

void

IdToObjectMap::sweep()

{

    for (Table::Enum e(table_); !e.empty(); e.popFront()) {

        JS::Heap<JSObject *> *objp = &e.front().value();

        JS_UpdateWeakPointerAfterGC(objp);

        if (!*objp)

            e.removeFront();

    }

}

JSObject *

IdToObjectMap::find(ObjectId id)

{

    Table::Ptr p = table_.lookup(id);

    if (!p)

        return nullptr;

    return p->value();

}

bool

IdToObjectMap::add(ObjectId id, JSObject *obj)

{

    return table_.put(id, obj);

}

void

IdToObjectMap::remove(ObjectId id)

{

    table_.remove(id);

}

ObjectToIdMap::ObjectToIdMap()

  : table_(nullptr)

{

}

ObjectToIdMap::~ObjectToIdMap()

{

    if (table_) {

        dom::AddForDeferredFinalization<Table, nsAutoPtr>(table_);

        table_ = nullptr;

    }

}

bool

ObjectToIdMap::init()

{

    if (table_)

        return true;

    table_ = new Table(SystemAllocPolicy());

    return table_ && table_->init(32);

}

void

ObjectToIdMap::trace(JSTracer *trc)

{

    for (Table::Enum e(*table_); !e.empty(); e.popFront()) {

        JSObject *obj = e.front().key();

        JS_CallUnbarrieredObjectTracer(trc, &obj, "ipc-object");

        if (obj != e.front().key())

            e.rekeyFront(obj);

    }

}

void

ObjectToIdMap::sweep()

{

    for (Table::Enum e(*table_); !e.empty(); e.popFront()) {

        JSObject *obj = e.front().key();

        JS_UpdateWeakPointerAfterGCUnbarriered(&obj);

        if (!obj)

            e.removeFront();

        else if (obj != e.front().key())

            e.rekeyFront(obj);

    }

}

ObjectId

ObjectToIdMap::find(JSObject *obj)

{

    Table::Ptr p = table_->lookup(obj);

    if (!p)

        return ObjectId::nullId();

    return p->value();

}

bool

ObjectToIdMap::add(JSContext *cx, JSObject *obj, ObjectId id)

{

    if (!table_->put(obj, id))

        return false;

    JS_StoreObjectPostBarrierCallback(cx, keyMarkCallback, obj, table_);

    return true;

}

/*

 * This function is called during minor GCs for each key in the HashMap that has

 * been moved.

 */

/* static */ void

ObjectToIdMap::keyMarkCallback(JSTracer *trc, JSObject *key, void *data)

{

    Table *table = static_cast<Table*>(data);

    JSObject *prior = key;

    JS_CallUnbarrieredObjectTracer(trc, &key, "ObjectIdCache::table_ key");

    table->rekeyIfMoved(prior, key);

}

void

ObjectToIdMap::remove(JSObject *obj)

{

    table_->remove(obj);

}

bool JavaScriptShared::sLoggingInitialized;

bool JavaScriptShared::sLoggingEnabled;

bool JavaScriptShared::sStackLoggingEnabled;

JavaScriptShared::JavaScriptShared(JSRuntime *rt)

  : rt_(rt),

    refcount_(1),

    nextSerialNumber_(1)

{

    if (!sLoggingInitialized) {

        sLoggingInitialized = true;

        if (PR_GetEnv("MOZ_CPOW_LOG")) {

            sLoggingEnabled = true;

            sStackLoggingEnabled = true;

        } else {

            Preferences::AddBoolVarCache(&sLoggingEnabled,

                                         "dom.ipc.cpows.log.enabled", false);

            Preferences::AddBoolVarCache(&sStackLoggingEnabled,

                                         "dom.ipc.cpows.log.stack", false);

        }

    }

}

bool

JavaScriptShared::init()

{

    if (!objects_.init())

        return false;

    if (!cpows_.init())

        return false;

    if (!unwaivedObjectIds_.init())

        return false;

    if (!waivedObjectIds_.init())

        return false;

    return true;

}

void

JavaScriptShared::decref()

{

    refcount_--;

    if (!refcount_)

        delete this;

}

void

JavaScriptShared::incref()

{

    refcount_++;

}

bool

JavaScriptShared::convertIdToGeckoString(JSContext *cx, JS::HandleId id, nsString *to)

{

    RootedValue idval(cx);

    if (!JS_IdToValue(cx, id, &idval))

        return false;

    RootedString str(cx, ToString(cx, idval));

    if (!str)

        return false;

    return AssignJSString(cx, *to, str);

}

bool

JavaScriptShared::convertGeckoStringToId(JSContext *cx, const nsString &from, JS::MutableHandleId to)

{

    RootedString str(cx, JS_NewUCStringCopyN(cx, from.BeginReading(), from.Length()));

    if (!str)

        return false;

    return JS_StringToId(cx, str, to);

}

bool

JavaScriptShared::toVariant(JSContext *cx, JS::HandleValue from, JSVariant *to)

{

    switch (JS_TypeOfValue(cx, from)) {

      case JSTYPE_VOID:

        *to = UndefinedVariant();

        return true;

      case JSTYPE_OBJECT:

      case JSTYPE_FUNCTION:

      {

        RootedObject obj(cx, from.toObjectOrNull());

        if (!obj) {

            MOZ_ASSERT(from == JSVAL_NULL);

            *to = NullVariant();

            return true;

        }

        if (xpc_JSObjectIsID(cx, obj)) {

            JSIID iid;

            const nsID *id = xpc_JSObjectToID(cx, obj);

            ConvertID(*id, &iid);

            *to = iid;

            return true;

        }

        ObjectVariant objVar;

        if (!toObjectVariant(cx, obj, &objVar))

            return false;

        *to = objVar;

        return true;

      }

      case JSTYPE_SYMBOL:

      {

        RootedSymbol sym(cx, from.toSymbol());

        SymbolVariant symVar;

        if (!toSymbolVariant(cx, sym, &symVar))

            return false;

        *to = symVar;

        return true;

      }

      case JSTYPE_STRING:

      {

        nsAutoJSString autoStr;

        if (!autoStr.init(cx, from))

            return false;

        *to = autoStr;

        return true;

      }

      case JSTYPE_NUMBER:

        if (from.isInt32())

            *to = double(from.toInt32());

        else

            *to = from.toDouble();

        return true;

      case JSTYPE_BOOLEAN:

        *to = from.toBoolean();

        return true;

      default:

        MOZ_ASSERT(false);

        return false;

    }

}

bool

JavaScriptShared::fromVariant(JSContext *cx, const JSVariant &from, MutableHandleValue to)

{

    switch (from.type()) {

        case JSVariant::TUndefinedVariant:

          to.set(UndefinedValue());

          return true;

        case JSVariant::TNullVariant:

          to.set(NullValue());

          return true;

        case JSVariant::TObjectVariant:

        {

          JSObject *obj = fromObjectVariant(cx, from.get_ObjectVariant());

          if (!obj)

              return false;

          to.set(ObjectValue(*obj));

          return true;

        }

        case JSVariant::TSymbolVariant:

        {

          Symbol *sym = fromSymbolVariant(cx, from.get_SymbolVariant());

          if (!sym)

              return false;

          to.setSymbol(sym);

          return true;

        }

        case JSVariant::Tdouble:

          to.set(JS_NumberValue(from.get_double()));

          return true;

        case JSVariant::Tbool:

          to.set(BOOLEAN_TO_JSVAL(from.get_bool()));

          return true;

        case JSVariant::TnsString:

        {

          const nsString &old = from.get_nsString();

          JSString *str = JS_NewUCStringCopyN(cx, old.BeginReading(), old.Length());

          if (!str)

              return false;

          to.set(StringValue(str));

          return true;

        }

        case JSVariant::TJSIID:

        {

          nsID iid;

          const JSIID &id = from.get_JSIID();

          ConvertID(id, &iid);

          JSCompartment *compartment = GetContextCompartment(cx);

          RootedObject global(cx, JS_GetGlobalForCompartmentOrNull(cx, compartment));

          JSObject *obj = xpc_NewIDObject(cx, global, iid);

          if (!obj)

              return false;

          to.set(ObjectValue(*obj));

          return true;

        }

        default:

          MOZ_CRASH("NYI");

          return false;

    }

}

bool

JavaScriptShared::toJSIDVariant(JSContext *cx, HandleId from, JSIDVariant *to)

{

    if (JSID_IS_STRING(from)) {

        nsAutoJSString autoStr;