DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Mercurial (375f9429ba87)

VCS Links

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94
/* -*- 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_ApartmentRegion_h
#define mozilla_mscom_ApartmentRegion_h

#include "mozilla/Assertions.h"
#include "mozilla/Attributes.h"

#include <objbase.h>

namespace mozilla {
namespace mscom {

class MOZ_NON_TEMPORARY_CLASS ApartmentRegion {
 public:
  /**
   * This constructor is to be used when we want to instantiate the object but
   * we do not yet know which type of apartment we want. Call Init() to
   * complete initialization.
   */
  ApartmentRegion() : mInitResult(CO_E_NOTINITIALIZED) {}

  explicit ApartmentRegion(COINIT aAptType)
      : mInitResult(::CoInitializeEx(nullptr, aAptType)) {
    // If this fires then we're probably mixing apartments on the same thread
    MOZ_ASSERT(IsValid());
  }

  ~ApartmentRegion() {
    if (IsValid()) {
      ::CoUninitialize();
    }
  }

  explicit operator bool() const { return IsValid(); }

  bool IsValidOutermost() const { return mInitResult == S_OK; }

  bool IsValid() const { return SUCCEEDED(mInitResult); }

  bool Init(COINIT aAptType) {
    MOZ_ASSERT(mInitResult == CO_E_NOTINITIALIZED);
    mInitResult = ::CoInitializeEx(nullptr, aAptType);
    MOZ_ASSERT(IsValid());
    return IsValid();
  }

  HRESULT
  GetHResult() const { return mInitResult; }

 private:
  ApartmentRegion(const ApartmentRegion&) = delete;
  ApartmentRegion& operator=(const ApartmentRegion&) = delete;
  ApartmentRegion(ApartmentRegion&&) = delete;
  ApartmentRegion& operator=(ApartmentRegion&&) = delete;

  HRESULT mInitResult;
};

template <COINIT T>
class MOZ_NON_TEMPORARY_CLASS ApartmentRegionT {
 public:
  ApartmentRegionT() : mAptRgn(T) {}

  ~ApartmentRegionT() = default;

  explicit operator bool() const { return mAptRgn.IsValid(); }

  bool IsValidOutermost() const { return mAptRgn.IsValidOutermost(); }

  bool IsValid() const { return mAptRgn.IsValid(); }

  HRESULT GetHResult() const { return mAptRgn.GetHResult(); }

 private:
  ApartmentRegionT(const ApartmentRegionT&) = delete;
  ApartmentRegionT& operator=(const ApartmentRegionT&) = delete;
  ApartmentRegionT(ApartmentRegionT&&) = delete;
  ApartmentRegionT& operator=(ApartmentRegionT&&) = delete;

  ApartmentRegion mAptRgn;
};

typedef ApartmentRegionT<COINIT_APARTMENTTHREADED> STARegion;
typedef ApartmentRegionT<COINIT_MULTITHREADED> MTARegion;

}  // namespace mscom
}  // namespace mozilla

#endif  // mozilla_mscom_ApartmentRegion_h