1 files changed, 98 insertions, 0 deletions
diff --git a/ipc/mscom/oop/Module.h b/ipc/mscom/oop/Module.h
new file mode 100644
index 0000000000..1fa4b31c49
--- /dev/null
+++ b/ipc/mscom/oop/Module.h
@@ -0,0 +1,98 @@
+/* -*- 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 mozilla_mscom_Module_h
+#define mozilla_mscom_Module_h
+
+#if defined(MOZILLA_INTERNAL_API)
+#  error This code is NOT for internal Gecko use!
+#endif  // defined(MOZILLA_INTERNAL_API)
+
+#include <objbase.h>
+
+/* WARNING! The code in this file may be loaded into the address spaces of other
+   processes! It MUST NOT link against xul.dll or other Gecko binaries! Only
+   inline code may be included! */
+
+namespace mozilla {
+namespace mscom {
+
+class Module {
+ public:
+  static HRESULT CanUnload() { return sRefCount == 0 ? S_OK : S_FALSE; }
+
+  static void Lock() { ++sRefCount; }
+  static void Unlock() { --sRefCount; }
+
+  enum class ThreadingModel {
+    DedicatedUiThreadOnly,
+    MultiThreadedApartmentOnly,
+    DedicatedUiThreadXorMultiThreadedApartment,
+    AllThreadsAllApartments,
+  };
+
+  enum class ClassType {
+    InprocServer,
+    InprocHandler,
+  };
+
+  /**
+   * In the Register functions, the aMsixContainer parameter specifies whether
+   * this registration is being performed inside an MSIX container. If true,
+   * the CLSID is registered in HKCU and a registry transaction is not used, as
+   * registry transactions don't work in an MSIX container. If false (the
+   * default), the CLSID is registered in HKLM and a registry transaction is
+   * used so that any failures roll back the entire operation. Specifying aAppId
+   * is only valid when aMsixContainer is false.
+   */
+  static HRESULT Register(REFCLSID aClsid, const ThreadingModel aThreadingModel,
+                          const ClassType aClassType = ClassType::InprocServer,
+                          const GUID* const aAppId = nullptr,
+                          const bool aMsixContainer = false) {
+    const CLSID* clsidArray[] = {&aClsid};
+    return Register(clsidArray, aThreadingModel, aClassType, aAppId,
+                    aMsixContainer);
+  }
+
+  template <size_t N>
+  static HRESULT Register(const CLSID* (&aClsids)[N],
+                          const ThreadingModel aThreadingModel,
+                          const ClassType aClassType = ClassType::InprocServer,
+                          const GUID* const aAppId = nullptr,
+                          const bool aMsixContainer = false) {
+    return Register(aClsids, N, aThreadingModel, aClassType, aAppId,
+                    aMsixContainer);
+  }
+
+  static HRESULT Deregister(REFCLSID aClsid,
+                            const GUID* const aAppId = nullptr) {
+    const CLSID* clsidArray[] = {&aClsid};
+    return Deregister(clsidArray, aAppId);
+  }
+
+  template <size_t N>
+  static HRESULT Deregister(const CLSID* (&aClsids)[N],
+                            const GUID* const aAppId = nullptr) {
+    return Deregister(aClsids, N, aAppId);
+  }
+
+ private:
+  static HRESULT Register(const CLSID* const* aClsids, const size_t aNumClsids,
+                          const ThreadingModel aThreadingModel,
+                          const ClassType aClassType, const GUID* const aAppId,
+                          const bool aMsixContainer);
+
+  static HRESULT Deregister(const CLSID* const* aClsids,
+                            const size_t aNumClsids, const GUID* const aAppId);
+
+ private:
+  static ULONG sRefCount;
+};
+
+}  // namespace mscom
+}  // namespace mozilla
+
+#endif  // mozilla_mscom_Module_h