summaryrefslogtreecommitdiffstats
path: root/ipc/mscom/FastMarshaler.h
blob: e1f3e888018906ee3d1fc337e5ecf744e1a77bae (plain)
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
/* -*- 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_FastMarshaler_h
#define mozilla_mscom_FastMarshaler_h

#include "mozilla/Atomics.h"
#include "mozilla/mscom/Aggregation.h"
#include "mozilla/RefPtr.h"

#include <objidl.h>

namespace mozilla {
namespace mscom {

/**
 * COM ping functionality is enabled by default and is designed to free strong
 * references held by defunct client processes. However, this incurs a
 * significant performance penalty in a11y code due to large numbers of remote
 * objects being created and destroyed within a short period of time. Thus, we
 * turn off pings to improve performance.
 * ACHTUNG! When COM pings are disabled, Release calls from remote clients are
 * never sent to the server! If you use this marshaler, you *must* explicitly
 * disconnect clients using CoDisconnectObject when the object is no longer
 * relevant. Otherwise, references to the object will never be released, causing
 * a leak.
 */
class FastMarshaler final : public IMarshal {
 public:
  static HRESULT Create(IUnknown* aOuter, IUnknown** aOutMarshalerUnk);

  // IMarshal
  STDMETHODIMP GetUnmarshalClass(REFIID riid, void* pv, DWORD dwDestContext,
                                 void* pvDestContext, DWORD mshlflags,
                                 CLSID* pCid) override;
  STDMETHODIMP GetMarshalSizeMax(REFIID riid, void* pv, DWORD dwDestContext,
                                 void* pvDestContext, DWORD mshlflags,
                                 DWORD* pSize) override;
  STDMETHODIMP MarshalInterface(IStream* pStm, REFIID riid, void* pv,
                                DWORD dwDestContext, void* pvDestContext,
                                DWORD mshlflags) override;
  STDMETHODIMP UnmarshalInterface(IStream* pStm, REFIID riid,
                                  void** ppv) override;
  STDMETHODIMP ReleaseMarshalData(IStream* pStm) override;
  STDMETHODIMP DisconnectObject(DWORD dwReserved) override;

 private:
  FastMarshaler(IUnknown* aOuter, HRESULT* aResult);
  ~FastMarshaler() = default;

  static DWORD GetMarshalFlags(DWORD aDestContext, DWORD aMshlFlags);

  Atomic<ULONG> mRefCnt;
  IUnknown* mOuter;
  RefPtr<IUnknown> mStdMarshalUnk;
  IMarshal* mStdMarshalWeak;
  DECLARE_AGGREGATABLE(FastMarshaler);
};

}  // namespace mscom
}  // namespace mozilla

#endif  // mozilla_mscom_FastMarshaler_h