1 files changed, 89 insertions, 0 deletions
diff --git a/dom/base/NodeUbiReporting.h b/dom/base/NodeUbiReporting.h
new file mode 100644
index 0000000000..be1c4e19c9
--- /dev/null
+++ b/dom/base/NodeUbiReporting.h
@@ -0,0 +1,89 @@
+/* -*- 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/. */
+
+#ifndef dom_NodeUbiReporting_h
+#define dom_NodeUbiReporting_h
+
+#include "Attr.h"
+#include "Document.h"
+#include "nsIContent.h"
+#include "nsIContentInlines.h"
+#include "nsINode.h"
+#include "js/UbiNode.h"
+
+/*
+ * This file defines specializations of JS::ubi::Concrete for DOM nodes
+ * so that the JS memory tools, which operate on the UbiNode graph, can
+ * define subclasses of JS::ubi::Base that represent DOM nodes and
+ * yield the outgoing edges in a DOM node graph for reporting.
+ */
+
+namespace JS::ubi {
+
+// The DOM node base class.
+// This is an abstract class and therefore does not require a concreteTypeName.
+template <>
+class Concrete<nsINode> : public Base {
+ protected:
+  explicit Concrete(nsINode* ptr) : Base(ptr) {}
+
+ public:
+  static void construct(void* storage, nsINode* ptr);
+  Size size(mozilla::MallocSizeOf mallocSizeOf) const override;
+  js::UniquePtr<EdgeRange> edges(JSContext* cx, bool wantNames) const override;
+
+  nsINode& get() const { return *static_cast<nsINode*>(ptr); }
+  CoarseType coarseType() const final { return CoarseType::DOMNode; }
+  const char16_t* descriptiveTypeName() const override;
+};
+
+template <>
+class Concrete<nsIContent> : public Concrete<nsINode> {
+ protected:
+  explicit Concrete(nsIContent* ptr) : Concrete<nsINode>(ptr) {}
+
+ public:
+  static void construct(void* storage, nsIContent* ptr) {
+    new (storage) Concrete(ptr);
+  }
+  const char16_t* typeName() const override { return concreteTypeName; };
+  static const char16_t concreteTypeName[];
+};
+
+template <>
+class Concrete<mozilla::dom::Document> : public Concrete<nsINode> {
+ protected:
+  explicit Concrete(mozilla::dom::Document* ptr) : Concrete<nsINode>(ptr) {}
+
+ public:
+  static void construct(void* storage, mozilla::dom::Document* ptr) {
+    new (storage) Concrete(ptr);
+  }
+  Size size(mozilla::MallocSizeOf mallocSizeOf) const override;
+
+  mozilla::dom::Document& getDoc() const {
+    return *static_cast<mozilla::dom::Document*>(ptr);
+  }
+  const char16_t* typeName() const override { return concreteTypeName; };
+  static const char16_t concreteTypeName[];
+};
+
+template <>
+class Concrete<mozilla::dom::Attr> : public Concrete<nsINode> {
+ protected:
+  explicit Concrete(mozilla::dom::Attr* ptr) : Concrete<nsINode>(ptr) {}
+
+ public:
+  static void construct(void* storage, mozilla::dom::Attr* ptr) {
+    new (storage) Concrete(ptr);
+  }
+  const char16_t* typeName() const override { return concreteTypeName; };
+  static const char16_t concreteTypeName[];
+};
+
+}  // namespace JS::ubi
+
+#endif