Line data    Source code

       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 SELF_REF_H
       8             : #define SELF_REF_H
       9             : 
      10             : #include "mozilla/Attributes.h"
      11             : 
      12             : namespace mozilla {
      13             : 
      14             : template<class T>
      15             : class SelfReference {
      16             : public:
      17           0 :   SelfReference() : mHeld(false) {}
      18           0 :   ~SelfReference()
      19             :   {
      20           0 :     NS_ASSERTION(!mHeld, "Forgot to drop the self reference?");
      21           0 :   }
      22             : 
      23           0 :   void Take(T* t)
      24             :   {
      25           0 :     if (!mHeld) {
      26           0 :       mHeld = true;
      27           0 :       t->AddRef();
      28             :     }
      29           0 :   }
      30           0 :   void Drop(T* t)
      31             :   {
      32           0 :     if (mHeld) {
      33           0 :       mHeld = false;
      34           0 :       t->Release();
      35             :     }
      36           0 :   }
      37             : 
      38             :   MOZ_IMPLICIT operator bool() const { return mHeld; }
      39             : 
      40             :   SelfReference(const SelfReference& aOther) = delete;
      41             :   void operator=(const SelfReference& aOther) = delete;
      42             : private:
      43             :   bool mHeld;
      44             : };
      45             : 
      46             : } // namespace mozilla
      47             : 
      48             : #endif // SELF_REF_H