CNodeRef copy constructor and assignment operator

maflcko · Aug 25, 2015 · 69ee1aa · 69ee1aa
1 parent dc81dd0
commit 69ee1aa
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/net.cpp b/src/net.cpp
@@ -782,6 +782,22 @@ class CNodeRef {
 
     CNode& operator *() const {return *_pnode;};
     CNode* operator ->() const {return _pnode;};
+
+    CNodeRef& operator =(const CNodeRef& other)
+    {
+        if (this != &other) {
+            _pnode->Release();
+            _pnode = other._pnode;
+            _pnode->AddRef();
+        }
+        return *this;
+    }
+
+    CNodeRef(const CNodeRef& other):
+        _pnode(other._pnode)
+    {
+        _pnode->AddRef();
+    }
 private:
     CNode *_pnode;
 };