Merge branch 'deladdr' of https://github.com/sipa/bitcoin
This commit is contained in:
commit
bf5b80a8ae
2 changed files with 12 additions and 20 deletions
28
src/db.cpp
28
src/db.cpp
|
@ -714,22 +714,12 @@ bool CTxDB::LoadBlockIndex()
|
|||
// CAddrDB
|
||||
//
|
||||
|
||||
bool CAddrDB::WriteAddress(const CAddress& addr)
|
||||
{
|
||||
return Write(make_pair(string("addr"), addr.GetKey()), addr);
|
||||
}
|
||||
|
||||
bool CAddrDB::WriteAddrman(const CAddrMan& addrman)
|
||||
{
|
||||
return Write(string("addrman"), addrman);
|
||||
}
|
||||
|
||||
bool CAddrDB::EraseAddress(const CAddress& addr)
|
||||
{
|
||||
return Erase(make_pair(string("addr"), addr.GetKey()));
|
||||
}
|
||||
|
||||
bool CAddrDB::LoadAddresses(bool &fUpdate)
|
||||
bool CAddrDB::LoadAddresses()
|
||||
{
|
||||
bool fAddrMan = false;
|
||||
if (Read(string("addrman"), addrman))
|
||||
|
@ -739,6 +729,7 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
|
|||
}
|
||||
|
||||
vector<CAddress> vAddr;
|
||||
vector<vector<unsigned char> > vDelete;
|
||||
|
||||
// Get cursor
|
||||
Dbc* pcursor = GetCursor();
|
||||
|
@ -762,7 +753,11 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
|
|||
if (strType == "addr")
|
||||
{
|
||||
if (fAddrMan)
|
||||
fUpdate = true;
|
||||
{
|
||||
vector<unsigned char> vchKey;
|
||||
ssKey >> vchKey;
|
||||
vDelete.push_back(vchKey);
|
||||
}
|
||||
else
|
||||
{
|
||||
CAddress addr;
|
||||
|
@ -774,6 +769,9 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
|
|||
}
|
||||
pcursor->close();
|
||||
|
||||
BOOST_FOREACH(const vector<unsigned char> &vchKey, vDelete)
|
||||
Erase(make_pair(string("addr"), vchKey));
|
||||
|
||||
if (!fAddrMan)
|
||||
{
|
||||
addrman.Add(vAddr, CNetAddr("0.0.0.0"));
|
||||
|
@ -785,11 +783,7 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
|
|||
|
||||
bool LoadAddresses()
|
||||
{
|
||||
bool fUpdate = false;
|
||||
bool fRet = CAddrDB("cr+").LoadAddresses(fUpdate);
|
||||
if (fUpdate)
|
||||
CDB::Rewrite("addr.dat", "\004addr");
|
||||
return fRet;
|
||||
return CAddrDB("cr+").LoadAddresses();
|
||||
}
|
||||
|
||||
|
||||
|
|
4
src/db.h
4
src/db.h
|
@ -306,10 +306,8 @@ private:
|
|||
CAddrDB(const CAddrDB&);
|
||||
void operator=(const CAddrDB&);
|
||||
public:
|
||||
bool WriteAddress(const CAddress& addr);
|
||||
bool EraseAddress(const CAddress& addr);
|
||||
bool WriteAddrman(const CAddrMan& addr);
|
||||
bool LoadAddresses(bool &fUpdate);
|
||||
bool LoadAddresses();
|
||||
};
|
||||
|
||||
bool LoadAddresses();
|
||||
|
|
Loading…
Reference in a new issue