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 nsXPTCUtils_h__
8 #define nsXPTCUtils_h__
9 
10 #include "xptcall.h"
11 #include "mozilla/MemoryReporting.h"
12 
13 /**
14  * A helper class that initializes an xptcall helper at construction
15  * and releases it at destruction.
16  */
17 class nsAutoXPTCStub : protected nsIXPTCProxy
18 {
19 public:
20   nsISomeInterface* mXPTCStub;
21 
22 protected:
nsAutoXPTCStub()23   nsAutoXPTCStub() : mXPTCStub(nullptr) {}
24 
25   nsresult
InitStub(const nsIID & aIID)26   InitStub(const nsIID& aIID)
27   {
28     return NS_GetXPTCallStub(aIID, this, &mXPTCStub);
29   }
30 
~nsAutoXPTCStub()31   ~nsAutoXPTCStub()
32   {
33     if (mXPTCStub) {
34       NS_DestroyXPTCallStub(mXPTCStub);
35     }
36   }
37 
38   size_t
SizeOfExcludingThis(mozilla::MallocSizeOf aMallocSizeOf)39   SizeOfExcludingThis(mozilla::MallocSizeOf aMallocSizeOf) const
40   {
41     return mXPTCStub ? NS_SizeOfIncludingThisXPTCallStub(mXPTCStub, aMallocSizeOf) : 0;
42   }
43 };
44 
45 #endif // nsXPTCUtils_h__
46