1 files changed, 151 insertions, 0 deletions
diff --git a/js/xpconnect/src/XPCWrappedNativeProto.cpp b/js/xpconnect/src/XPCWrappedNativeProto.cpp
new file mode 100644
index 0000000000..4b492bfa9e
--- /dev/null
+++ b/js/xpconnect/src/XPCWrappedNativeProto.cpp
@@ -0,0 +1,151 @@
+/* -*- 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/. */
+
+/* Shared proto object for XPCWrappedNative. */
+
+#include "xpcprivate.h"
+#include "js/Object.h"  // JS::SetReservedSlot
+#include "pratom.h"
+#include "XPCMaps.h"
+
+using namespace mozilla;
+
+#ifdef DEBUG
+int32_t XPCWrappedNativeProto::gDEBUG_LiveProtoCount = 0;
+#endif
+
+XPCWrappedNativeProto::XPCWrappedNativeProto(XPCWrappedNativeScope* Scope,
+                                             nsIClassInfo* ClassInfo,
+                                             RefPtr<XPCNativeSet>&& Set)
+    : mScope(Scope),
+      mJSProtoObject(nullptr),
+      mClassInfo(ClassInfo),
+      mSet(std::move(Set)) {
+  // This native object lives as long as its associated JSObject - killed
+  // by finalization of the JSObject (or explicitly if Init fails).
+
+  MOZ_COUNT_CTOR(XPCWrappedNativeProto);
+  MOZ_ASSERT(mScope);
+
+#ifdef DEBUG
+  gDEBUG_LiveProtoCount++;
+#endif
+}
+
+XPCWrappedNativeProto::~XPCWrappedNativeProto() {
+  MOZ_ASSERT(!mJSProtoObject, "JSProtoObject still alive");
+
+  MOZ_COUNT_DTOR(XPCWrappedNativeProto);
+
+#ifdef DEBUG
+  gDEBUG_LiveProtoCount--;
+#endif
+
+  // Note that our weak ref to mScope is not to be trusted at this point.
+
+  XPCNativeSet::ClearCacheEntryForClassInfo(mClassInfo);
+
+  DeferredFinalize(mClassInfo.forget().take());
+}
+
+bool XPCWrappedNativeProto::Init(JSContext* cx, nsIXPCScriptable* scriptable) {
+  mScriptable = scriptable;
+
+  JS::RootedObject proto(cx, JS::GetRealmObjectPrototype(cx));
+  mJSProtoObject = JS_NewObjectWithGivenProto(cx, &XPC_WN_Proto_JSClass, proto);
+
+  bool success = !!mJSProtoObject;
+  if (success) {
+    JS::SetReservedSlot(mJSProtoObject, ProtoSlot, JS::PrivateValue(this));
+  }
+
+  return success;
+}
+
+void XPCWrappedNativeProto::JSProtoObjectFinalized(JS::GCContext* gcx,
+                                                   JSObject* obj) {
+  MOZ_ASSERT(obj == mJSProtoObject, "huh?");
+
+#ifdef DEBUG
+  // Check that this object has already been swept from the map.
+  ClassInfo2WrappedNativeProtoMap* map = GetScope()->GetWrappedNativeProtoMap();
+  MOZ_ASSERT(map->Find(mClassInfo) != this);
+#endif
+
+  MOZ_ALWAYS_TRUE(GetRuntime()->GetDyingWrappedNativeProtos().append(this));
+  mJSProtoObject = nullptr;
+}
+
+void XPCWrappedNativeProto::JSProtoObjectMoved(JSObject* obj,
+                                               const JSObject* old) {
+  // Update without triggering barriers.
+  MOZ_ASSERT(mJSProtoObject == old);
+  mJSProtoObject.unbarrieredSet(obj);
+}
+
+void XPCWrappedNativeProto::SystemIsBeingShutDown() {
+  // Note that the instance might receive this call multiple times
+  // as we walk to here from various places.
+
+  if (mJSProtoObject) {
+    // short circuit future finalization
+    JS::SetReservedSlot(mJSProtoObject, ProtoSlot, JS::UndefinedValue());
+    mJSProtoObject = nullptr;
+  }
+}
+
+// static
+XPCWrappedNativeProto* XPCWrappedNativeProto::GetNewOrUsed(
+    JSContext* cx, XPCWrappedNativeScope* scope, nsIClassInfo* classInfo,
+    nsIXPCScriptable* scriptable) {
+  MOZ_ASSERT(scope, "bad param");
+  MOZ_ASSERT(classInfo, "bad param");
+
+  AutoMarkingWrappedNativeProtoPtr proto(cx);
+  ClassInfo2WrappedNativeProtoMap* map = nullptr;
+
+  map = scope->GetWrappedNativeProtoMap();
+  proto = map->Find(classInfo);
+  if (proto) {
+    return proto;
+  }
+
+  RefPtr<XPCNativeSet> set = XPCNativeSet::GetNewOrUsed(cx, classInfo);
+  if (!set) {
+    return nullptr;
+  }
+
+  proto = new XPCWrappedNativeProto(scope, classInfo, std::move(set));
+
+  if (!proto->Init(cx, scriptable)) {
+    delete proto.get();
+    return nullptr;
+  }
+
+  map->Add(classInfo, proto);
+
+  return proto;
+}
+
+void XPCWrappedNativeProto::DebugDump(int16_t depth) {
+#ifdef DEBUG
+  depth--;
+  XPC_LOG_ALWAYS(("XPCWrappedNativeProto @ %p", this));
+  XPC_LOG_INDENT();
+  XPC_LOG_ALWAYS(("gDEBUG_LiveProtoCount is %d", gDEBUG_LiveProtoCount));
+  XPC_LOG_ALWAYS(("mScope @ %p", mScope));
+  XPC_LOG_ALWAYS(("mJSProtoObject @ %p", mJSProtoObject.get()));
+  XPC_LOG_ALWAYS(("mSet @ %p", mSet.get()));
+  XPC_LOG_ALWAYS(("mScriptable @ %p", mScriptable.get()));
+  if (depth && mScriptable) {
+    XPC_LOG_INDENT();
+    XPC_LOG_ALWAYS(("mFlags of %x", mScriptable->GetScriptableFlags()));
+    XPC_LOG_ALWAYS(("mJSClass @ %p", mScriptable->GetJSClass()));
+    XPC_LOG_OUTDENT();
+  }
+  XPC_LOG_OUTDENT();
+#endif
+}