xpcom/glue/nsCOMPtr.cpp

   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 #include "nsCOMPtr.h"
   8
   9 nsresult
  10 nsQueryInterface::operator()(const nsIID& aIID, void** aAnswer) const
  11 {
  12   nsresult status;
  13   if (mRawPtr) {
  14     status = mRawPtr->QueryInterface(aIID, aAnswer);
  15 #ifdef NSCAP_FEATURE_TEST_NONNULL_QUERY_SUCCEEDS
  16     NS_ASSERTION(NS_SUCCEEDED(status),
  17                  "interface not found---were you expecting that?");
  18 #endif
  19   } else {
  20     status = NS_ERROR_NULL_POINTER;
  21   }
  22
  23   return status;
  24 }
  25
  26 nsresult
  27 nsQueryInterfaceWithError::operator()(const nsIID& aIID, void** aAnswer) const
  28 {
  29   nsresult status;
  30   if (mRawPtr) {
  31     status = mRawPtr->QueryInterface(aIID, aAnswer);
  32 #ifdef NSCAP_FEATURE_TEST_NONNULL_QUERY_SUCCEEDS
  33     NS_ASSERTION(NS_SUCCEEDED(status),
  34                  "interface not found---were you expecting that?");
  35 #endif
  36   } else {
  37     status = NS_ERROR_NULL_POINTER;
  38   }
  39
  40   if (mErrorPtr) {
  41     *mErrorPtr = status;
  42   }
  43   return status;
  44 }
  45
  46 void
  47 nsCOMPtr_base::assign_with_AddRef(nsISupports* aRawPtr)
  48 {
  49   if (aRawPtr) {
  50     NSCAP_ADDREF(this, aRawPtr);
  51   }
  52   assign_assuming_AddRef(aRawPtr);
  53 }
  54
  55 void
  56 nsCOMPtr_base::assign_from_qi(const nsQueryInterface aQI, const nsIID& aIID)
  57 {
  58   void* newRawPtr;
  59   if (NS_FAILED(aQI(aIID, &newRawPtr))) {
  60     newRawPtr = 0;
  61   }
  62   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
  63 }
  64
  65 void
  66 nsCOMPtr_base::assign_from_qi_with_error(const nsQueryInterfaceWithError& aQI,
  67                                          const nsIID& aIID)
  68 {
  69   void* newRawPtr;
  70   if (NS_FAILED(aQI(aIID, &newRawPtr))) {
  71     newRawPtr = 0;
  72   }
  73   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
  74 }
  75
  76 void
  77 nsCOMPtr_base::assign_from_gs_cid(const nsGetServiceByCID aGS,
  78                                   const nsIID& aIID)
  79 {
  80   void* newRawPtr;
  81   if (NS_FAILED(aGS(aIID, &newRawPtr))) {
  82     newRawPtr = 0;
  83   }
  84   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
  85 }
  86
  87 void
  88 nsCOMPtr_base::assign_from_gs_cid_with_error(
  89     const nsGetServiceByCIDWithError& aGS, const nsIID& aIID)
  90 {
  91   void* newRawPtr;
  92   if (NS_FAILED(aGS(aIID, &newRawPtr))) {
  93     newRawPtr = 0;
  94   }
  95   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
  96 }
  97
  98 void
  99 nsCOMPtr_base::assign_from_gs_contractid(const nsGetServiceByContractID aGS,
 100                                          const nsIID& aIID)
 101 {
 102   void* newRawPtr;
 103   if (NS_FAILED(aGS(aIID, &newRawPtr))) {
 104     newRawPtr = 0;
 105   }
 106   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
 107 }
 108
 109 void
 110 nsCOMPtr_base::assign_from_gs_contractid_with_error(
 111     const nsGetServiceByContractIDWithError& aGS, const nsIID& aIID)
 112 {
 113   void* newRawPtr;
 114   if (NS_FAILED(aGS(aIID, &newRawPtr))) {
 115     newRawPtr = 0;
 116   }
 117   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
 118 }
 119
 120 void
 121 nsCOMPtr_base::assign_from_helper(const nsCOMPtr_helper& aHelper,
 122                                   const nsIID& aIID)
 123 {
 124   void* newRawPtr;
 125   if (NS_FAILED(aHelper(aIID, &newRawPtr))) {
 126     newRawPtr = 0;
 127   }
 128   assign_assuming_AddRef(static_cast<nsISupports*>(newRawPtr));
 129 }
 130
 131 void**
 132 nsCOMPtr_base::begin_assignment()
 133 {
 134   assign_assuming_AddRef(0);
 135   return reinterpret_cast<void**>(&mRawPtr);
 136 }