forked from mirrors/gecko-dev
87fbdfd39e
We add new DLL registration code. This is a rather generic function that permits the following: * Registering multiple `CLSID`s for the same DLL; * Registering an optional `AppID`. Registering an `AppID` allows us to use a `DllSurrogate` to host the DLL out-of-process using Windows' built-in `dllhost.exe`. I'll be using this feature in a future bug. * Supporting all available threading modelsl; * Capable of registering either inproc servers or inproc handlers; * Using the transaction-based registry API so that we can cleanly rollback during registration if any part(s) of it fail. Differential Revision: https://phabricator.services.mozilla.com/D95606
84 lines
2.7 KiB
C++
84 lines
2.7 KiB
C++
/* -*- 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,
|
|
};
|
|
|
|
static HRESULT Register(REFCLSID aClsid, const ThreadingModel aThreadingModel,
|
|
const ClassType aClassType = ClassType::InprocServer,
|
|
const GUID* const aAppId = nullptr) {
|
|
const CLSID* clsidArray[] = {&aClsid};
|
|
return Register(clsidArray, aThreadingModel, aClassType, aAppId);
|
|
}
|
|
|
|
template <size_t N>
|
|
static HRESULT Register(const CLSID* (&aClsids)[N],
|
|
const ThreadingModel aThreadingModel,
|
|
const ClassType aClassType = ClassType::InprocServer,
|
|
const GUID* const aAppId = nullptr) {
|
|
return Register(aClsids, N, aThreadingModel, aClassType, aAppId);
|
|
}
|
|
|
|
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);
|
|
|
|
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
|