forked from mirrors/gecko-dev
		
	
		
			
				
	
	
		
			501 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			501 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* -*- 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 "jsfriendapi.h"
 | 
						|
#include "xpcprivate.h"
 | 
						|
 | 
						|
using namespace js;
 | 
						|
using namespace JS;
 | 
						|
using namespace mozilla;
 | 
						|
using namespace mozilla::jsipc;
 | 
						|
 | 
						|
ObjectStore::ObjectStore()
 | 
						|
  : table_(SystemAllocPolicy())
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
ObjectStore::init()
 | 
						|
{
 | 
						|
    return table_.init(32);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
ObjectStore::trace(JSTracer *trc)
 | 
						|
{
 | 
						|
    for (ObjectTable::Range r(table_.all()); !r.empty(); r.popFront()) {
 | 
						|
        DebugOnly<JSObject *> prior = r.front().value().get();
 | 
						|
        JS_CallHeapObjectTracer(trc, &r.front().value(), "ipc-object");
 | 
						|
        MOZ_ASSERT(r.front().value() == prior);
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
JSObject *
 | 
						|
ObjectStore::find(ObjectId id)
 | 
						|
{
 | 
						|
    ObjectTable::Ptr p = table_.lookup(id);
 | 
						|
    if (!p)
 | 
						|
        return nullptr;
 | 
						|
    return p->value();
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
ObjectStore::add(ObjectId id, JSObject *obj)
 | 
						|
{
 | 
						|
    return table_.put(id, obj);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
ObjectStore::remove(ObjectId id)
 | 
						|
{
 | 
						|
    table_.remove(id);
 | 
						|
}
 | 
						|
 | 
						|
ObjectIdCache::ObjectIdCache()
 | 
						|
  : table_(nullptr)
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
ObjectIdCache::~ObjectIdCache()
 | 
						|
{
 | 
						|
    if (table_) {
 | 
						|
        dom::AddForDeferredFinalization<ObjectIdTable, nsAutoPtr>(table_);
 | 
						|
        table_ = nullptr;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
ObjectIdCache::init()
 | 
						|
{
 | 
						|
    MOZ_ASSERT(!table_);
 | 
						|
    table_ = new ObjectIdTable(SystemAllocPolicy());
 | 
						|
    return table_ && table_->init(32);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
ObjectIdCache::trace(JSTracer *trc)
 | 
						|
{
 | 
						|
    for (ObjectIdTable::Range r(table_->all()); !r.empty(); r.popFront()) {
 | 
						|
        JSObject *obj = r.front().key();
 | 
						|
        JS_CallObjectTracer(trc, &obj, "ipc-id");
 | 
						|
        MOZ_ASSERT(obj == r.front().key());
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
ObjectId
 | 
						|
ObjectIdCache::find(JSObject *obj)
 | 
						|
{
 | 
						|
    ObjectIdTable::Ptr p = table_->lookup(obj);
 | 
						|
    if (!p)
 | 
						|
        return 0;
 | 
						|
    return p->value();
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
ObjectIdCache::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
 | 
						|
ObjectIdCache::keyMarkCallback(JSTracer *trc, JSObject *key, void *data) {
 | 
						|
    ObjectIdTable* table = static_cast<ObjectIdTable*>(data);
 | 
						|
    JSObject *prior = key;
 | 
						|
    JS_CallObjectTracer(trc, &key, "ObjectIdCache::table_ key");
 | 
						|
    table->rekeyIfMoved(prior, key);
 | 
						|
}
 | 
						|
 | 
						|
void
 | 
						|
ObjectIdCache::remove(JSObject *obj)
 | 
						|
{
 | 
						|
    table_->remove(obj);
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
JavaScriptShared::init()
 | 
						|
{
 | 
						|
    if (!objects_.init())
 | 
						|
        return false;
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
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;
 | 
						|
 | 
						|
    const jschar *chars = JS_GetStringCharsZ(cx, str);
 | 
						|
    if (!chars)
 | 
						|
        return false;
 | 
						|
 | 
						|
    *to = chars;
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
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 = void_t();
 | 
						|
        return true;
 | 
						|
 | 
						|
      case JSTYPE_NULL:
 | 
						|
      {
 | 
						|
        *to = uint64_t(0);
 | 
						|
        return true;
 | 
						|
      }
 | 
						|
 | 
						|
      case JSTYPE_OBJECT:
 | 
						|
      case JSTYPE_FUNCTION:
 | 
						|
      {
 | 
						|
        RootedObject obj(cx, from.toObjectOrNull());
 | 
						|
        if (!obj) {
 | 
						|
            MOZ_ASSERT(from == JSVAL_NULL);
 | 
						|
            *to = uint64_t(0);
 | 
						|
            return true;
 | 
						|
        }
 | 
						|
 | 
						|
        if (xpc_JSObjectIsID(cx, obj)) {
 | 
						|
            JSIID iid;
 | 
						|
            const nsID *id = xpc_JSObjectToID(cx, obj);
 | 
						|
            ConvertID(*id, &iid);
 | 
						|
            *to = iid;
 | 
						|
            return true;
 | 
						|
        }
 | 
						|
 | 
						|
        ObjectId id;
 | 
						|
        if (!makeId(cx, obj, &id))
 | 
						|
            return false;
 | 
						|
        *to = uint64_t(id);
 | 
						|
        return true;
 | 
						|
      }
 | 
						|
 | 
						|
      case JSTYPE_STRING:
 | 
						|
      {
 | 
						|
        nsDependentJSString dep;
 | 
						|
        if (!dep.init(cx, from))
 | 
						|
            return false;
 | 
						|
        *to = dep;
 | 
						|
        return true;
 | 
						|
      }
 | 
						|
 | 
						|
      case JSTYPE_NUMBER:
 | 
						|
        if (JSVAL_IS_INT(from))
 | 
						|
            *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::toValue(JSContext *cx, const JSVariant &from, MutableHandleValue to)
 | 
						|
{
 | 
						|
    switch (from.type()) {
 | 
						|
        case JSVariant::Tvoid_t:
 | 
						|
          to.set(UndefinedValue());
 | 
						|
          return true;
 | 
						|
 | 
						|
        case JSVariant::Tuint64_t:
 | 
						|
        {
 | 
						|
          ObjectId id = from.get_uint64_t();
 | 
						|
          if (id) {
 | 
						|
              JSObject *obj = unwrap(cx, id);
 | 
						|
              if (!obj)
 | 
						|
                  return false;
 | 
						|
              to.set(ObjectValue(*obj));
 | 
						|
          } else {
 | 
						|
              to.set(JSVAL_NULL);
 | 
						|
          }
 | 
						|
          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:
 | 
						|
          return false;
 | 
						|
    }
 | 
						|
}
 | 
						|
 | 
						|
/* static */ void
 | 
						|
JavaScriptShared::ConvertID(const nsID &from, JSIID *to)
 | 
						|
{
 | 
						|
    to->m0() = from.m0;
 | 
						|
    to->m1() = from.m1;
 | 
						|
    to->m2() = from.m2;
 | 
						|
    to->m3_0() = from.m3[0];
 | 
						|
    to->m3_1() = from.m3[1];
 | 
						|
    to->m3_2() = from.m3[2];
 | 
						|
    to->m3_3() = from.m3[3];
 | 
						|
    to->m3_4() = from.m3[4];
 | 
						|
    to->m3_5() = from.m3[5];
 | 
						|
    to->m3_6() = from.m3[6];
 | 
						|
    to->m3_7() = from.m3[7];
 | 
						|
}
 | 
						|
 | 
						|
/* static */ void
 | 
						|
JavaScriptShared::ConvertID(const JSIID &from, nsID *to)
 | 
						|
{
 | 
						|
    to->m0 = from.m0();
 | 
						|
    to->m1 = from.m1();
 | 
						|
    to->m2 = from.m2();
 | 
						|
    to->m3[0] = from.m3_0();
 | 
						|
    to->m3[1] = from.m3_1();
 | 
						|
    to->m3[2] = from.m3_2();
 | 
						|
    to->m3[3] = from.m3_3();
 | 
						|
    to->m3[4] = from.m3_4();
 | 
						|
    to->m3[5] = from.m3_5();
 | 
						|
    to->m3[6] = from.m3_6();
 | 
						|
    to->m3[7] = from.m3_7();
 | 
						|
}
 | 
						|
 | 
						|
static const uint32_t DefaultPropertyOp = 1;
 | 
						|
static const uint32_t GetterOnlyPropertyStub = 2;
 | 
						|
static const uint32_t UnknownPropertyOp = 3;
 | 
						|
 | 
						|
bool
 | 
						|
JavaScriptShared::fromDescriptor(JSContext *cx, Handle<JSPropertyDescriptor> desc,
 | 
						|
                                 PPropertyDescriptor *out)
 | 
						|
{
 | 
						|
    out->attrs() = desc.attributes();
 | 
						|
    if (!toVariant(cx, desc.value(), &out->value()))
 | 
						|
        return false;
 | 
						|
 | 
						|
    if (!makeId(cx, desc.object(), &out->objId()))
 | 
						|
        return false;
 | 
						|
 | 
						|
    if (!desc.getter()) {
 | 
						|
        out->getter() = 0;
 | 
						|
    } else if (desc.hasGetterObject()) {
 | 
						|
        JSObject *getter = desc.getterObject();
 | 
						|
        if (!makeId(cx, getter, &out->getter()))
 | 
						|
            return false;
 | 
						|
    } else {
 | 
						|
        if (desc.getter() == JS_PropertyStub)
 | 
						|
            out->getter() = DefaultPropertyOp;
 | 
						|
        else
 | 
						|
            out->getter() = UnknownPropertyOp;
 | 
						|
    }
 | 
						|
 | 
						|
    if (!desc.setter()) {
 | 
						|
        out->setter() = 0;
 | 
						|
    } else if (desc.hasSetterObject()) {
 | 
						|
        JSObject *setter = desc.setterObject();
 | 
						|
        if (!makeId(cx, setter, &out->setter()))
 | 
						|
            return false;
 | 
						|
    } else {
 | 
						|
        if (desc.setter() == JS_StrictPropertyStub)
 | 
						|
            out->setter() = DefaultPropertyOp;
 | 
						|
        else if (desc.setter() == js_GetterOnlyPropertyStub)
 | 
						|
            out->setter() = GetterOnlyPropertyStub;
 | 
						|
        else
 | 
						|
            out->setter() = UnknownPropertyOp;
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
UnknownPropertyStub(JSContext *cx, HandleObject obj, HandleId id, MutableHandleValue vp)
 | 
						|
{
 | 
						|
    JS_ReportError(cx, "getter could not be wrapped via CPOWs");
 | 
						|
    return false;
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
UnknownStrictPropertyStub(JSContext *cx, HandleObject obj, HandleId id, bool strict, MutableHandleValue vp)
 | 
						|
{
 | 
						|
    JS_ReportError(cx, "setter could not be wrapped via CPOWs");
 | 
						|
    return false;
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
JavaScriptShared::toDescriptor(JSContext *cx, const PPropertyDescriptor &in,
 | 
						|
                               MutableHandle<JSPropertyDescriptor> out)
 | 
						|
{
 | 
						|
    out.setAttributes(in.attrs());
 | 
						|
    if (!toValue(cx, in.value(), out.value()))
 | 
						|
        return false;
 | 
						|
    Rooted<JSObject*> obj(cx);
 | 
						|
    if (!unwrap(cx, in.objId(), &obj))
 | 
						|
        return false;
 | 
						|
    out.object().set(obj);
 | 
						|
 | 
						|
    if (!in.getter()) {
 | 
						|
        out.setGetter(nullptr);
 | 
						|
    } else if (in.attrs() & JSPROP_GETTER) {
 | 
						|
        Rooted<JSObject*> getter(cx);
 | 
						|
        if (!unwrap(cx, in.getter(), &getter))
 | 
						|
            return false;
 | 
						|
        out.setGetter(JS_DATA_TO_FUNC_PTR(JSPropertyOp, getter.get()));
 | 
						|
    } else {
 | 
						|
        if (in.getter() == DefaultPropertyOp)
 | 
						|
            out.setGetter(JS_PropertyStub);
 | 
						|
        else
 | 
						|
            out.setGetter(UnknownPropertyStub);
 | 
						|
    }
 | 
						|
 | 
						|
    if (!in.setter()) {
 | 
						|
        out.setSetter(nullptr);
 | 
						|
    } else if (in.attrs() & JSPROP_SETTER) {
 | 
						|
        Rooted<JSObject*> setter(cx);
 | 
						|
        if (!unwrap(cx, in.setter(), &setter))
 | 
						|
            return false;
 | 
						|
        out.setSetter(JS_DATA_TO_FUNC_PTR(JSStrictPropertyOp, setter.get()));
 | 
						|
    } else {
 | 
						|
        if (in.setter() == DefaultPropertyOp)
 | 
						|
            out.setSetter(JS_StrictPropertyStub);
 | 
						|
        else if (in.setter() == GetterOnlyPropertyStub)
 | 
						|
            out.setSetter(js_GetterOnlyPropertyStub);
 | 
						|
        else
 | 
						|
            out.setSetter(UnknownStrictPropertyStub);
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
CpowIdHolder::ToObject(JSContext *cx, JS::MutableHandleObject objp)
 | 
						|
{
 | 
						|
    return js_->Unwrap(cx, cpows_, objp);
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
JavaScriptShared::Unwrap(JSContext *cx, const InfallibleTArray<CpowEntry> &aCpows,
 | 
						|
                         JS::MutableHandleObject objp)
 | 
						|
{
 | 
						|
    objp.set(nullptr);
 | 
						|
 | 
						|
    if (!aCpows.Length())
 | 
						|
        return true;
 | 
						|
 | 
						|
    RootedObject obj(cx, JS_NewObject(cx, nullptr, JS::NullPtr(), JS::NullPtr()));
 | 
						|
    if (!obj)
 | 
						|
        return false;
 | 
						|
 | 
						|
    RootedValue v(cx);
 | 
						|
    RootedString str(cx);
 | 
						|
    for (size_t i = 0; i < aCpows.Length(); i++) {
 | 
						|
        const nsString &name = aCpows[i].name();
 | 
						|
 | 
						|
        if (!toValue(cx, aCpows[i].value(), &v))
 | 
						|
            return false;
 | 
						|
 | 
						|
        if (!JS_DefineUCProperty(cx,
 | 
						|
                                 obj,
 | 
						|
                                 name.BeginReading(),
 | 
						|
                                 name.Length(),
 | 
						|
                                 v,
 | 
						|
                                 nullptr,
 | 
						|
                                 nullptr,
 | 
						|
                                 JSPROP_ENUMERATE))
 | 
						|
        {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    objp.set(obj);
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 | 
						|
bool
 | 
						|
JavaScriptShared::Wrap(JSContext *cx, HandleObject aObj, InfallibleTArray<CpowEntry> *outCpows)
 | 
						|
{
 | 
						|
    if (!aObj)
 | 
						|
        return true;
 | 
						|
 | 
						|
    AutoIdArray ids(cx, JS_Enumerate(cx, aObj));
 | 
						|
    if (!ids)
 | 
						|
        return false;
 | 
						|
 | 
						|
    RootedId id(cx);
 | 
						|
    RootedValue v(cx);
 | 
						|
    for (size_t i = 0; i < ids.length(); i++) {
 | 
						|
        id = ids[i];
 | 
						|
 | 
						|
        nsString str;
 | 
						|
        if (!convertIdToGeckoString(cx, id, &str))
 | 
						|
            return false;
 | 
						|
 | 
						|
        if (!JS_GetPropertyById(cx, aObj, id, &v))
 | 
						|
            return false;
 | 
						|
 | 
						|
        JSVariant var;
 | 
						|
        if (!toVariant(cx, v, &var))
 | 
						|
            return false;
 | 
						|
 | 
						|
        outCpows->AppendElement(CpowEntry(str, var));
 | 
						|
    }
 | 
						|
 | 
						|
    return true;
 | 
						|
}
 | 
						|
 |