Browse Source

Merge pull request #2566 from sipa/nodef

Allow the default key to be unavailable
0.8
Gavin Andresen 12 years ago
parent
commit
61a845dcb6
  1. 10
      src/init.cpp
  2. 5
      src/main.cpp
  3. 39
      src/wallet.cpp
  4. 2
      src/wallet.h

10
src/init.cpp

@ -966,11 +966,11 @@ bool AppInit2(boost::thread_group& threadGroup)
RandAddSeedPerfmon(); RandAddSeedPerfmon();
CPubKey newDefaultKey; CPubKey newDefaultKey;
if (!pwalletMain->GetKeyFromPool(newDefaultKey, false)) if (pwalletMain->GetKeyFromPool(newDefaultKey, false)) {
strErrors << _("Cannot initialize keypool") << "\n"; pwalletMain->SetDefaultKey(newDefaultKey);
pwalletMain->SetDefaultKey(newDefaultKey); if (!pwalletMain->SetAddressBookName(pwalletMain->vchDefaultKey.GetID(), ""))
if (!pwalletMain->SetAddressBookName(pwalletMain->vchDefaultKey.GetID(), "")) strErrors << _("Cannot write default address") << "\n";
strErrors << _("Cannot write default address") << "\n"; }
} }
printf("%s", strErrors.str().c_str()); printf("%s", strErrors.str().c_str());

5
src/main.cpp

@ -4158,7 +4158,10 @@ CBlockTemplate* CreateNewBlock(CReserveKey& reservekey)
txNew.vin.resize(1); txNew.vin.resize(1);
txNew.vin[0].prevout.SetNull(); txNew.vin[0].prevout.SetNull();
txNew.vout.resize(1); txNew.vout.resize(1);
txNew.vout[0].scriptPubKey << reservekey.GetReservedKey() << OP_CHECKSIG; CPubKey pubkey;
if (!reservekey.GetReservedKey(pubkey))
return NULL;
txNew.vout[0].scriptPubKey << pubkey << OP_CHECKSIG;
// Add our coinbase tx as first transaction // Add our coinbase tx as first transaction
pblock->vtx.push_back(txNew); pblock->vtx.push_back(txNew);

39
src/wallet.cpp

@ -462,17 +462,19 @@ bool CWallet::AddToWallet(const CWalletTx& wtxIn)
return false; return false;
#ifndef QT_GUI #ifndef QT_GUI
// If default receiving address gets used, replace it with a new one // If default receiving address gets used, replace it with a new one
CScript scriptDefaultKey; if (vchDefaultKey.IsValid()) {
scriptDefaultKey.SetDestination(vchDefaultKey.GetID()); CScript scriptDefaultKey;
BOOST_FOREACH(const CTxOut& txout, wtx.vout) scriptDefaultKey.SetDestination(vchDefaultKey.GetID());
{ BOOST_FOREACH(const CTxOut& txout, wtx.vout)
if (txout.scriptPubKey == scriptDefaultKey)
{ {
CPubKey newDefaultKey; if (txout.scriptPubKey == scriptDefaultKey)
if (GetKeyFromPool(newDefaultKey, false))
{ {
SetDefaultKey(newDefaultKey); CPubKey newDefaultKey;
SetAddressBookName(vchDefaultKey.GetID(), ""); if (GetKeyFromPool(newDefaultKey, false))
{
SetDefaultKey(newDefaultKey);
SetAddressBookName(vchDefaultKey.GetID(), "");
}
} }
} }
} }
@ -1204,8 +1206,8 @@ bool CWallet::CreateTransaction(const vector<pair<CScript, int64> >& vecSend, CW
// post-backup change. // post-backup change.
// Reserve a new key pair from key pool // Reserve a new key pair from key pool
CPubKey vchPubKey = reservekey.GetReservedKey(); CPubKey vchPubKey;
// assert(mapKeys.count(vchPubKey)); assert(reservekey.GetReservedKey(vchPubKey)); // should never fail, as we just unlocked
// Fill a vout to ourself // Fill a vout to ourself
// TODO: pass in scriptChange instead of reservekey so // TODO: pass in scriptChange instead of reservekey so
@ -1742,7 +1744,7 @@ set< set<CTxDestination> > CWallet::GetAddressGroupings()
return ret; return ret;
} }
CPubKey CReserveKey::GetReservedKey() bool CReserveKey::GetReservedKey(CPubKey& pubkey)
{ {
if (nIndex == -1) if (nIndex == -1)
{ {
@ -1750,14 +1752,17 @@ CPubKey CReserveKey::GetReservedKey()
pwallet->ReserveKeyFromKeyPool(nIndex, keypool); pwallet->ReserveKeyFromKeyPool(nIndex, keypool);
if (nIndex != -1) if (nIndex != -1)
vchPubKey = keypool.vchPubKey; vchPubKey = keypool.vchPubKey;
else else {
{ if (pwallet->vchDefaultKey.IsValid()) {
printf("CReserveKey::GetReservedKey(): Warning: Using default key instead of a new key, top up your keypool!"); printf("CReserveKey::GetReservedKey(): Warning: Using default key instead of a new key, top up your keypool!");
vchPubKey = pwallet->vchDefaultKey; vchPubKey = pwallet->vchDefaultKey;
} else
return false;
} }
} }
assert(vchPubKey.IsValid()); assert(vchPubKey.IsValid());
return vchPubKey; pubkey = vchPubKey;
return true;
} }
void CReserveKey::KeepKey() void CReserveKey::KeepKey()

2
src/wallet.h

@ -331,7 +331,7 @@ public:
} }
void ReturnKey(); void ReturnKey();
CPubKey GetReservedKey(); bool GetReservedKey(CPubKey &pubkey);
void KeepKey(); void KeepKey();
}; };

Loading…
Cancel
Save