Use erase instead of rewrite to remove old addr entries

This commit is contained in:
Pieter Wuille 2012-03-26 20:17:35 +02:00
parent 01a196e08d
commit 53cb1a49e7
2 changed files with 12 additions and 20 deletions

View file

@ -714,22 +714,12 @@ bool CTxDB::LoadBlockIndex()
// CAddrDB // CAddrDB
// //
bool CAddrDB::WriteAddress(const CAddress& addr)
{
return Write(make_pair(string("addr"), addr.GetKey()), addr);
}
bool CAddrDB::WriteAddrman(const CAddrMan& addrman) bool CAddrDB::WriteAddrman(const CAddrMan& addrman)
{ {
return Write(string("addrman"), addrman); return Write(string("addrman"), addrman);
} }
bool CAddrDB::EraseAddress(const CAddress& addr) bool CAddrDB::LoadAddresses()
{
return Erase(make_pair(string("addr"), addr.GetKey()));
}
bool CAddrDB::LoadAddresses(bool &fUpdate)
{ {
bool fAddrMan = false; bool fAddrMan = false;
if (Read(string("addrman"), addrman)) if (Read(string("addrman"), addrman))
@ -739,6 +729,7 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
} }
vector<CAddress> vAddr; vector<CAddress> vAddr;
vector<vector<unsigned char> > vDelete;
// Get cursor // Get cursor
Dbc* pcursor = GetCursor(); Dbc* pcursor = GetCursor();
@ -762,7 +753,11 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
if (strType == "addr") if (strType == "addr")
{ {
if (fAddrMan) if (fAddrMan)
fUpdate = true; {
vector<unsigned char> vchKey;
ssKey >> vchKey;
vDelete.push_back(vchKey);
}
else else
{ {
CAddress addr; CAddress addr;
@ -774,6 +769,9 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
} }
pcursor->close(); pcursor->close();
BOOST_FOREACH(const vector<unsigned char> &vchKey, vDelete)
Erase(make_pair(string("addr"), vchKey));
if (!fAddrMan) if (!fAddrMan)
{ {
addrman.Add(vAddr, CNetAddr("0.0.0.0")); addrman.Add(vAddr, CNetAddr("0.0.0.0"));
@ -785,11 +783,7 @@ bool CAddrDB::LoadAddresses(bool &fUpdate)
bool LoadAddresses() bool LoadAddresses()
{ {
bool fUpdate = false; return CAddrDB("cr+").LoadAddresses();
bool fRet = CAddrDB("cr+").LoadAddresses(fUpdate);
if (fUpdate)
CDB::Rewrite("addr.dat", "\004addr");
return fRet;
} }

View file

@ -306,10 +306,8 @@ private:
CAddrDB(const CAddrDB&); CAddrDB(const CAddrDB&);
void operator=(const CAddrDB&); void operator=(const CAddrDB&);
public: public:
bool WriteAddress(const CAddress& addr);
bool EraseAddress(const CAddress& addr);
bool WriteAddrman(const CAddrMan& addr); bool WriteAddrman(const CAddrMan& addr);
bool LoadAddresses(bool &fUpdate); bool LoadAddresses();
}; };
bool LoadAddresses(); bool LoadAddresses();