Added locks on the setpwalletRegistered functions in main.cpp and added an UnregisterAllWallets function.
This commit is contained in:
parent
79f6925fac
commit
e5cefb9be1
1 changed files with 14 additions and 0 deletions
14
src/main.cpp
14
src/main.cpp
|
@ -99,9 +99,16 @@ void UnregisterWallet(CWallet* pwalletIn)
|
|||
}
|
||||
}
|
||||
|
||||
void UnregisterAllWallets()
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
setpwalletRegistered.clear();
|
||||
}
|
||||
|
||||
// get the wallet transaction with the given hash (if it exists)
|
||||
bool static GetTransaction(const uint256& hashTx, CWalletTx& wtx)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
if (pwallet->GetTransaction(hashTx,wtx))
|
||||
return true;
|
||||
|
@ -111,6 +118,7 @@ bool static GetTransaction(const uint256& hashTx, CWalletTx& wtx)
|
|||
// erases transaction with the given hash from all wallets
|
||||
void static EraseFromWallets(uint256 hash)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->EraseFromWallet(hash);
|
||||
}
|
||||
|
@ -118,6 +126,7 @@ void static EraseFromWallets(uint256 hash)
|
|||
// make sure all wallets know about the given transaction, in the given block
|
||||
void SyncWithWallets(const uint256 &hash, const CTransaction& tx, const CBlock* pblock, bool fUpdate)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->AddToWalletIfInvolvingMe(hash, tx, pblock, fUpdate);
|
||||
}
|
||||
|
@ -125,6 +134,7 @@ void SyncWithWallets(const uint256 &hash, const CTransaction& tx, const CBlock*
|
|||
// notify wallets about a new best chain
|
||||
void static SetBestChain(const CBlockLocator& loc)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->SetBestChain(loc);
|
||||
}
|
||||
|
@ -132,6 +142,7 @@ void static SetBestChain(const CBlockLocator& loc)
|
|||
// notify wallets about an updated transaction
|
||||
void static UpdatedTransaction(const uint256& hashTx)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->UpdatedTransaction(hashTx);
|
||||
}
|
||||
|
@ -139,6 +150,7 @@ void static UpdatedTransaction(const uint256& hashTx)
|
|||
// dump all wallets
|
||||
void static PrintWallets(const CBlock& block)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->PrintWallet(block);
|
||||
}
|
||||
|
@ -146,6 +158,7 @@ void static PrintWallets(const CBlock& block)
|
|||
// notify wallets about an incoming inventory (for request counts)
|
||||
void static Inventory(const uint256& hash)
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->Inventory(hash);
|
||||
}
|
||||
|
@ -153,6 +166,7 @@ void static Inventory(const uint256& hash)
|
|||
// ask wallets to resend their transactions
|
||||
void static ResendWalletTransactions()
|
||||
{
|
||||
LOCK(cs_setpwalletRegistered);
|
||||
BOOST_FOREACH(CWallet* pwallet, setpwalletRegistered)
|
||||
pwallet->ResendWalletTransactions();
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue