ipc/mscom/ActivationContext.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef mozilla_mscom_ActivationContext_h
   8 #define mozilla_mscom_ActivationContext_h
   9
  10 #include <utility>
  11
  12 #include "mozilla/Attributes.h"
  13 #include "mozilla/mscom/ActCtxResource.h"
  14
  15 #if defined(MOZILLA_INTERNAL_API)
  16 #  include "mozilla/ResultVariant.h"
  17 #  include "nsString.h"
  18 #endif  // defined(MOZILLA_INTERNAL_API)
  19
  20 #include <windows.h>
  21
  22 namespace mozilla {
  23 namespace mscom {
  24
  25 class ActivationContext final {
  26  public:
  27   // This is the default resource ID that the Windows dynamic linker searches
  28   // for when seeking a manifest while loading a DLL.
  29   static constexpr WORD kDllManifestDefaultResourceId = 2;
  30
  31   ActivationContext() : mActCtx(INVALID_HANDLE_VALUE) {}
  32
  33   explicit ActivationContext(ActCtxResource aResource);
  34   explicit ActivationContext(HMODULE aLoadFromModule,
  35                              WORD aResourceId = kDllManifestDefaultResourceId);
  36
  37   ActivationContext(ActivationContext&& aOther);
  38   ActivationContext& operator=(ActivationContext&& aOther);
  39
  40   ActivationContext(const ActivationContext& aOther);
  41   ActivationContext& operator=(const ActivationContext& aOther);
  42
  43   ~ActivationContext();
  44
  45   explicit operator bool() const { return mActCtx != INVALID_HANDLE_VALUE; }
  46
  47 #if defined(MOZILLA_INTERNAL_API)
  48   static Result<uintptr_t, HRESULT> GetCurrent();
  49   static HRESULT GetCurrentManifestPath(nsAString& aOutManifestPath);
  50 #endif  // defined(MOZILLA_INTERNAL_API)
  51
  52  private:
  53   void Init(ACTCTXW& aActCtx);
  54   void AddRef();
  55   void Release();
  56
  57  private:
  58   HANDLE mActCtx;
  59
  60   friend class ActivationContextRegion;
  61 };
  62
  63 class MOZ_NON_TEMPORARY_CLASS ActivationContextRegion final {
  64  public:
  65   template <typename... Args>
  66   explicit ActivationContextRegion(Args&&... aArgs)
  67       : mActCtx(std::forward<Args>(aArgs)...), mActCookie(0) {
  68     Activate();
  69   }
  70
  71   ActivationContextRegion();
  72
  73   explicit ActivationContextRegion(const ActivationContext& aActCtx);
  74   ActivationContextRegion& operator=(const ActivationContext& aActCtx);
  75
  76   explicit ActivationContextRegion(ActivationContext&& aActCtx);
  77   ActivationContextRegion& operator=(ActivationContext&& aActCtx);
  78
  79   ActivationContextRegion(ActivationContextRegion&& aRgn);
  80   ActivationContextRegion& operator=(ActivationContextRegion&& aRgn);
  81
  82   ~ActivationContextRegion();
  83
  84   explicit operator bool() const { return !!mActCookie; }
  85
  86   ActivationContextRegion(const ActivationContextRegion&) = delete;
  87   ActivationContextRegion& operator=(const ActivationContextRegion&) = delete;
  88
  89   bool Deactivate();
  90
  91  private:
  92   void Activate();
  93
  94   ActivationContext mActCtx;
  95   ULONG_PTR mActCookie;
  96 };
  97
  98 }  // namespace mscom
  99 }  // namespace mozilla
 100
 101 #endif  // mozilla_mscom_ActivationContext_h