Browse Source

Acquire cs_vNodes before changing refrence counts

0.13
Patrick Strateman 9 years ago
parent
commit
fed30940ef
  1. 14
      src/net.cpp

14
src/net.cpp

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

Loading…
Cancel
Save