diff --git a/src/net.cpp b/src/net.cpp index 4f4c7b81c..cb5a24f0a 100644 --- a/src/net.cpp +++ b/src/net.cpp @@ -782,6 +782,22 @@ public: 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; };