neobytes-project/neobytes
neobytes-project/neobytes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Acquire cs_vNodes before changing refrence counts

Browse Source
This commit is contained in:
Patrick Strateman 2015-08-25 16:30:02 -07:00
parent 69ee1aab00
commit fed30940ef
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/net.cpp
View File

@ -777,8 +777,15 @@ static list<CNode*> vNodesDisconnected;
class CNodeRef { class CNodeRef {
public: public:
CNodeRef(CNode *pnode) : _pnode(pnode) {_pnode->AddRef();} CNodeRef(CNode *pnode) : _pnode(pnode) {
~CNodeRef() {_pnode->Release();} LOCK(cs_vNodes);
_pnode->AddRef();
}
~CNodeRef() {
LOCK(cs_vNodes);
_pnode->Release();
}
CNode& operator *() const {return *_pnode;}; CNode& operator *() const {return *_pnode;};
CNode* operator ->() const {return _pnode;}; CNode* operator ->() const {return _pnode;};
@ -786,6 +793,8 @@ public:
CNodeRef& operator =(const CNodeRef& other) CNodeRef& operator =(const CNodeRef& other)
{ {
if (this != &other) { if (this != &other) {
LOCK(cs_vNodes);
_pnode->Release(); _pnode->Release();
_pnode = other._pnode; _pnode = other._pnode;
_pnode->AddRef(); _pnode->AddRef();
@ -796,6 +805,7 @@ public:
CNodeRef(const CNodeRef& other): CNodeRef(const CNodeRef& other):
_pnode(other._pnode) _pnode(other._pnode)
{ {
LOCK(cs_vNodes);
_pnode->AddRef(); _pnode->AddRef();
} }
private: private: