Maintenance: automate header guards 2/3 (#1655)

[thirdparty/squid.git] / src / base / RefCount.h

diff --git a/src/base/RefCount.h b/src/base/RefCount.h
index ec2a98c9d7dfb958a6a121b8e15caed0cfc7486f..15bca428580548b6d1293e67453684e78f158adf 100644 (file)
--- a/src/base/RefCount.h
+++ b/src/base/RefCount.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 1996-2015 The Squid Software Foundation and contributors
+ * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
  *
  * Squid software is distributed under GPLv2+ license and includes
  * contributions from numerous individuals and organizations.
@@ -8,13 +8,14 @@
 
 /* DEBUG: section --    Refcount allocator */
 
-#ifndef SQUID_REFCOUNT_H_
-#define SQUID_REFCOUNT_H_
+#ifndef SQUID_SRC_BASE_REFCOUNT_H
+#define SQUID_SRC_BASE_REFCOUNT_H
 
 // reference counting requires the Lock API on base classes
 #include "base/Lock.h"
 
 #include <iostream>
+#include <utility>
 
 /**
  * Template for Reference Counting pointers.
@@ -27,7 +28,13 @@ class RefCount
 {
 
 public:
-    RefCount () : p_ (NULL) {}
+    /// creates a new C object using given C constructor arguments (if any)
+    /// \returns a refcounting pointer to the created object
+    template<typename... Args>
+    inline static auto Make(Args&&... args) {
+        return RefCount<C>(new C(std::forward<Args>(args)...));
+    }
+    RefCount () : p_ (nullptr) {}
 
     RefCount (C const *p) : p_(p) { reference (*this); }
 
@@ -39,11 +46,15 @@ public:
         reference (p);
     }
 
-#if __cplusplus >= 201103L
     RefCount (RefCount &&p) : p_(std::move(p.p_)) {
-        p.p_=NULL;
+        p.p_=nullptr;
+    }
+
+    /// Base::Pointer = Derived::Pointer
+    template <class Other>
+    RefCount(const RefCount<Other> &p): p_(p.getRaw()) {
+        reference(*this);
     }
-#endif
 
     RefCount& operator = (const RefCount& p) {
         // DO NOT CHANGE THE ORDER HERE!!!
@@ -54,25 +65,28 @@ public:
         return *this;
     }
 
-#if __cplusplus >= 201103L
     RefCount& operator = (RefCount&& p) {
         if (this != &p) {
             dereference(p.p_);
-            p.p_ = NULL;
+            p.p_ = nullptr;
         }
         return *this;
     }
-#endif
+
+    RefCount &operator =(std::nullptr_t) { dereference(); return *this; }
+
+    explicit operator bool() const { return p_; }
 
     bool operator !() const { return !p_; }
 
     C * operator-> () const {return const_cast<C *>(p_); }
 
-    C & operator * () const {return *const_cast<C *>(p_); }
-
-    C const * getRaw() const {return p_; }
+    C & operator * () const {
+        assert(p_);
+        return *const_cast<C *>(p_);
+    }
 
-    C * getRaw() {return const_cast<C *>(p_); }
+    C * getRaw() const { return const_cast<C *>(p_); }
 
     bool operator == (const RefCount& p) const {
         return p.p_ == p_;
@@ -82,8 +96,20 @@ public:
         return p.p_ != p_;
     }
 
+    template <class Other>
+    bool operator ==(const Other * const p) const
+    {
+        return p == p_;
+    }
+
+    template <class Other>
+    bool operator !=(const Other * const p) const
+    {
+        return p != p_;
+    }
+
 private:
-    void dereference(C const *newP = NULL) {
+    void dereference(C const *newP = nullptr) {
         /* Setting p_ first is important:
         * we may be freed ourselves as a result of
         * delete p_;
@@ -107,11 +133,11 @@ private:
 template <class C>
 inline std::ostream &operator <<(std::ostream &os, const RefCount<C> &p)
 {
-    if (p != NULL)
+    if (p != nullptr)
         return os << p.getRaw() << '*' << p->LockCount();
     else
         return os << "NULL";
 }
 
-#endif /* SQUID_REFCOUNT_H_ */
+#endif /* SQUID_SRC_BASE_REFCOUNT_H */