Merge pull request #1008 from gavinandresen/addrworkaround

Workaround hangs when upgrading old addr.dat files
This commit is contained in:
Gavin Andresen 2012-03-29 12:10:08 -07:00
commit dac1888c18

View file

@ -722,13 +722,14 @@ bool CAddrDB::WriteAddrman(const CAddrMan& addrman)
bool CAddrDB::LoadAddresses() bool CAddrDB::LoadAddresses()
{ {
bool fAddrMan = false;
if (Read(string("addrman"), addrman)) if (Read(string("addrman"), addrman))
{ {
printf("Loaded %i addresses\n", addrman.size()); printf("Loaded %i addresses\n", addrman.size());
fAddrMan = true; return true;
} }
// Read pre-0.6 addr records
vector<CAddress> vAddr; vector<CAddress> vAddr;
vector<vector<unsigned char> > vDelete; vector<vector<unsigned char> > vDelete;
@ -752,32 +753,20 @@ bool CAddrDB::LoadAddresses()
string strType; string strType;
ssKey >> strType; ssKey >> strType;
if (strType == "addr") if (strType == "addr")
{
if (fAddrMan)
{
vector<unsigned char> vchKey;
ssKey >> vchKey;
vDelete.push_back(vchKey);
}
else
{ {
CAddress addr; CAddress addr;
ssValue >> addr; ssValue >> addr;
vAddr.push_back(addr); vAddr.push_back(addr);
} }
}
} }
pcursor->close(); 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")); addrman.Add(vAddr, CNetAddr("0.0.0.0"));
printf("Loaded %i addresses\n", addrman.size()); printf("Loaded %i addresses\n", addrman.size());
}
// Note: old records left; we ran into hangs-on-startup
// bugs for some users who (we think) were running after
// an unclean shutdown.
return true; return true;
} }