Ethan Clay
910 lines
28 KiB
C++
910 lines
28 KiB
C++
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
// Copyright (c) 2009-2012 The Bitcoin developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
#ifndef TRIANGLES_WALLET_H
|
|
#define TRIANGLES_WALLET_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <atomic>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include "main.h"
|
|
#include "key.h"
|
|
#include "keystore.h"
|
|
#include "script.h"
|
|
#include "ui_interface.h"
|
|
#include "util.h"
|
|
#include "walletdb.h"
|
|
#include "smessage.h"
|
|
|
|
extern bool fWalletUnlockStakingOnly;
|
|
extern bool fConfChange;
|
|
class CAccountingEntry;
|
|
class CWalletTx;
|
|
class CReserveKey;
|
|
class COutput;
|
|
class CCoinControl;
|
|
|
|
//typedef std::map<CKeyID, CStealthKeyMetadata> StealthKeyMetaMap;
|
|
using mapValue_t = std::map<std::string, std::string>;
|
|
|
|
/** (client) version numbers for particular wallet features */
|
|
enum class WalletFeature : int
|
|
{
|
|
Base = 10500,
|
|
WalletCrypt = 40000,
|
|
ComprPubKey = 60000,
|
|
Latest = 60000
|
|
};
|
|
|
|
/** A key pool entry */
|
|
class CKeyPool
|
|
{
|
|
public:
|
|
int64_t nTime;
|
|
CPubKey vchPubKey;
|
|
|
|
CKeyPool()
|
|
{
|
|
nTime = GetTime();
|
|
}
|
|
|
|
CKeyPool(const CPubKey& vchPubKeyIn)
|
|
{
|
|
nTime = GetTime();
|
|
vchPubKey = vchPubKeyIn;
|
|
}
|
|
|
|
IMPLEMENT_SERIALIZE
|
|
(
|
|
if (!(nType & SER_GETHASH))
|
|
READWRITE(nVersion);
|
|
READWRITE(nTime);
|
|
READWRITE(vchPubKey);
|
|
)
|
|
};
|
|
|
|
/** A CWallet is an extension of a keystore, which also maintains a set of transactions and balances,
|
|
* and provides the ability to create new transactions.
|
|
*/
|
|
class CWallet : public CCryptoKeyStore
|
|
{
|
|
private:
|
|
bool SelectCoinsSimple(int64_t nTargetValue, unsigned int nSpendTime, int nMinConf, std::set<std::pair<const CWalletTx*,unsigned int> >& setCoinsRet, int64_t& nValueRet) const;
|
|
bool SelectCoins(int64_t nTargetValue, unsigned int nSpendTime, std::set<std::pair<const CWalletTx*,unsigned int> >& setCoinsRet, int64_t& nValueRet, const CCoinControl *coinControl=nullptr) const;
|
|
|
|
// the current wallet version: clients below this version are not able to load the wallet
|
|
int nWalletVersion;
|
|
|
|
// the maximum wallet format version: memory-only variable that specifies to what version this wallet may be upgraded
|
|
int nWalletMaxVersion;
|
|
|
|
public:
|
|
mutable CCriticalSection cs_wallet;
|
|
|
|
bool fFileBacked;
|
|
std::string strWalletFile;
|
|
|
|
std::set<int64_t> setKeyPool;
|
|
std::map<CKeyID, CKeyMetadata> mapKeyMetadata;
|
|
|
|
|
|
typedef std::map<unsigned int, CMasterKey> MasterKeyMap;
|
|
MasterKeyMap mapMasterKeys;
|
|
unsigned int nMasterKeyMaxID;
|
|
|
|
CWallet()
|
|
{
|
|
nWalletVersion = static_cast<int>(WalletFeature::Base);
|
|
nWalletMaxVersion = static_cast<int>(WalletFeature::Base);
|
|
fFileBacked = false;
|
|
nMasterKeyMaxID = 0;
|
|
fHDEnabled = false;
|
|
nHDChainIndex = 0;
|
|
nTimeFirstKey = 0;
|
|
nOrderPosNext = 0;
|
|
nCachedStakeWeight = 0;
|
|
nCachedStakeWeightTime = 0;
|
|
}
|
|
CWallet(std::string strWalletFileIn) : CWallet()
|
|
{
|
|
strWalletFile = strWalletFileIn;
|
|
fFileBacked = true;
|
|
}
|
|
|
|
std::map<uint256, CWalletTx> mapWallet;
|
|
int64_t nOrderPosNext;
|
|
std::map<uint256, int> mapRequestCount;
|
|
|
|
std::map<CTxDestination, std::string> mapAddressBook;
|
|
|
|
CPubKey vchDefaultKey;
|
|
int64_t nTimeFirstKey;
|
|
|
|
// ---- HD (BIP39/BIP32) wallet state ----
|
|
bool fHDEnabled; // an HD seed has been set
|
|
int64_t nHDChainIndex; // next external index (m/44'/2222'/0'/0/n)
|
|
std::string hdMnemonic; // in-memory phrase (present when unlocked/unencrypted)
|
|
std::vector<unsigned char> vchCryptedHDMnemonic; // encrypted phrase (loaded, decrypted on unlock)
|
|
uint256 hdMnemonicIV; // IV for the encrypted phrase
|
|
std::string hdPassphrase; // BIP39 "25th word"; empty = none. Same lifecycle as hdMnemonic.
|
|
std::vector<unsigned char> vchCryptedHDPassphrase; // encrypted passphrase (loaded, decrypted on unlock)
|
|
uint256 hdPassphraseIV; // IV for the encrypted passphrase
|
|
|
|
// check whether we are allowed to upgrade (or already support) to the named feature
|
|
bool CanSupportFeature(WalletFeature wf) { return nWalletMaxVersion >= static_cast<int>(wf); }
|
|
|
|
void AvailableCoinsMinConf(std::vector<COutput>& vCoins, int nConf) const;
|
|
void AvailableCoins(std::vector<COutput>& vCoins, bool fOnlyConfirmed=true, const CCoinControl *coinControl=nullptr) const;
|
|
bool SelectCoinsMinConf(int64_t nTargetValue, unsigned int nSpendTime, int nConfMine, int nConfTheirs, std::vector<COutput> vCoins, std::set<std::pair<const CWalletTx*,unsigned int> >& setCoinsRet, int64_t& nValueRet) const;
|
|
// keystore implementation
|
|
// Generate a new key
|
|
CPubKey GenerateNewKey();
|
|
|
|
// ---- HD wallet (BIP39/BIP32) ----
|
|
bool IsHDEnabled() const { return fHDEnabled; }
|
|
bool SetHDSeed(const std::string& mnemonicIn, const std::string& passphrase, bool fGenerate, std::string& mnemonicOut, std::string& strError);
|
|
bool GetHDMnemonic(std::string& mnemonicOut) const;
|
|
bool DeriveHDKey(int64_t index, CKey& keyOut) const;
|
|
bool LoadHDMnemonic(const std::string& m) { hdMnemonic = m; fHDEnabled = true; return true; }
|
|
bool LoadCryptedHDMnemonic(const uint256& iv, const std::vector<unsigned char>& cipher) { hdMnemonicIV = iv; vchCryptedHDMnemonic = cipher; fHDEnabled = true; return true; }
|
|
bool LoadHDPassphrase(const std::string& p) { hdPassphrase = p; return true; }
|
|
bool LoadCryptedHDPassphrase(const uint256& iv, const std::vector<unsigned char>& cipher) { hdPassphraseIV = iv; vchCryptedHDPassphrase = cipher; return true; }
|
|
// Adds a key to the store, and saves it to disk.
|
|
bool AddKey(const CKey& key);
|
|
// Adds a key to the store, without saving it to disk (used by LoadWallet)
|
|
bool LoadKey(const CKey& key) { return CCryptoKeyStore::AddKey(key); }
|
|
// Load metadata (used by LoadWallet)
|
|
bool LoadKeyMetadata(const CPubKey &pubkey, const CKeyMetadata &metadata);
|
|
|
|
bool LoadMinVersion(int nVersion) { nWalletVersion = nVersion; nWalletMaxVersion = std::max(nWalletMaxVersion, nVersion); return true; }
|
|
|
|
// Adds an encrypted key to the store, and saves it to disk.
|
|
bool AddCryptedKey(const CPubKey &vchPubKey, const std::vector<unsigned char> &vchCryptedSecret);
|
|
// Adds an encrypted key to the store, without saving it to disk (used by LoadWallet)
|
|
bool LoadCryptedKey(const CPubKey &vchPubKey, const std::vector<unsigned char> &vchCryptedSecret);
|
|
bool AddCScript(const CScript& redeemScript);
|
|
bool LoadCScript(const CScript& redeemScript) { return CCryptoKeyStore::AddCScript(redeemScript); }
|
|
|
|
bool Lock();
|
|
bool Unlock(const SecureString& strWalletPassphrase);
|
|
bool ChangeWalletPassphrase(const SecureString& strOldWalletPassphrase, const SecureString& strNewWalletPassphrase);
|
|
bool EncryptWallet(const SecureString& strWalletPassphrase);
|
|
|
|
void GetKeyBirthTimes(std::map<CKeyID, int64_t> &mapKeyBirth) const;
|
|
|
|
|
|
/** Increment the next transaction order id
|
|
@return next transaction order id
|
|
*/
|
|
int64_t IncOrderPosNext(CWalletDB *pwalletdb = nullptr);
|
|
|
|
typedef std::pair<CWalletTx*, CAccountingEntry*> TxPair;
|
|
typedef std::multimap<int64_t, TxPair > TxItems;
|
|
|
|
/** Get the wallet's activity log
|
|
@return multimap of ordered transactions and accounting entries
|
|
@warning Returned pointers are *only* valid within the scope of passed acentries
|
|
*/
|
|
TxItems OrderedTxItems(std::list<CAccountingEntry>& acentries, std::string strAccount = "");
|
|
|
|
void MarkDirty();
|
|
bool AddToWallet(const CWalletTx& wtxIn);
|
|
bool AddToWalletIfInvolvingMe(const CTransaction& tx, const CBlock* pblock, bool fUpdate = false, bool fFindBlock = false);
|
|
bool EraseFromWallet(uint256 hash);
|
|
bool AbandonTransaction(const uint256& hashTx);
|
|
void WalletUpdateSpent(const CTransaction& prevout, bool fBlock = false);
|
|
int ScanForWalletTransactions(CBlockIndex* pindexStart, bool fUpdate = false);
|
|
bool ScanForWalletTransactionsFromIndex(CBlockIndex* pindexStart, bool fUpdate, int* pnFound = nullptr);
|
|
int ScanForWalletTransaction(const uint256& hashTx);
|
|
void ReacceptWalletTransactions();
|
|
void ResendWalletTransactions(bool fForce = false);
|
|
int64_t GetBalance() const;
|
|
int64_t GetUnconfirmedBalance() const;
|
|
int64_t GetImmatureBalance() const;
|
|
int64_t GetStake() const;
|
|
int64_t GetNewMint() const;
|
|
// Get all balances in a single lock acquisition + single pass (avoids 4x lock + 4x iteration)
|
|
bool GetAllBalances(int64_t& nBalance, int64_t& nStake, int64_t& nUnconfirmed, int64_t& nImmature) const;
|
|
bool CreateTransaction(const std::vector<std::pair<CScript, int64_t> >& vecSend, CWalletTx& wtxNew, CReserveKey& reservekey, int64_t& nFeeRet, const CCoinControl *coinControl=nullptr);
|
|
bool CreateTransaction(CScript scriptPubKey, int64_t nValue, std::string& sNarr, CWalletTx& wtxNew, CReserveKey& reservekey, int64_t& nFeeRet, const CCoinControl *coinControl=nullptr);
|
|
bool CommitTransaction(CWalletTx& wtxNew, CReserveKey& reservekey);
|
|
|
|
bool GetStakeWeight(const CKeyStore& keystore, uint64_t& nMinWeight, uint64_t& nMaxWeight, uint64_t& nWeight);
|
|
bool CreateCoinStake(const CKeyStore& keystore, unsigned int nBits, int64_t nSearchInterval, int64_t nFees, CTransaction& txNew, CKey& key);
|
|
|
|
// Cached staking info - updated by the staking thread, read by the UI thread.
|
|
std::atomic<uint64_t> nCachedStakeWeight;
|
|
std::atomic<int64_t> nCachedStakeWeightTime; // GetTime() when last updated
|
|
|
|
std::string SendMoney(CScript scriptPubKey, int64_t nValue, std::string& sNarr, CWalletTx& wtxNew, bool fAskFee=false);
|
|
std::string SendMoneyToDestination(const CTxDestination& address, int64_t nValue, std::string& sNarr, CWalletTx& wtxNew, bool fAskFee=false);
|
|
|
|
|
|
bool NewKeyPool();
|
|
bool TopUpKeyPool(unsigned int nSize = 0);
|
|
int64_t AddReserveKey(const CKeyPool& keypool);
|
|
void ReserveKeyFromKeyPool(int64_t& nIndex, CKeyPool& keypool);
|
|
void KeepKey(int64_t nIndex);
|
|
void ReturnKey(int64_t nIndex);
|
|
bool GetKeyFromPool(CPubKey &key, bool fAllowReuse=true);
|
|
int64_t GetOldestKeyPoolTime();
|
|
void GetAllReserveKeys(std::set<CKeyID>& setAddress) const;
|
|
|
|
std::set< std::set<CTxDestination> > GetAddressGroupings();
|
|
std::map<CTxDestination, int64_t> GetAddressBalances();
|
|
|
|
[[nodiscard]] bool IsMine(const CTxIn& txin) const;
|
|
int64_t GetDebit(const CTxIn& txin) const;
|
|
[[nodiscard]] bool IsMine(const CTxOut& txout) const
|
|
{
|
|
return ::IsMine(*this, txout.scriptPubKey);
|
|
}
|
|
int64_t GetCredit(const CTxOut& txout) const
|
|
{
|
|
if (!MoneyRange(txout.nValue))
|
|
throw std::runtime_error("CWallet::GetCredit() : value out of range");
|
|
return (IsMine(txout) ? txout.nValue : 0);
|
|
}
|
|
bool IsChange(const CTxOut& txout) const;
|
|
int64_t GetChange(const CTxOut& txout) const
|
|
{
|
|
if (!MoneyRange(txout.nValue))
|
|
throw std::runtime_error("CWallet::GetChange() : value out of range");
|
|
return (IsChange(txout) ? txout.nValue : 0);
|
|
}
|
|
[[nodiscard]] bool IsMine(const CTransaction& tx) const
|
|
{
|
|
for (const CTxOut& txout : tx.vout)
|
|
if (IsMine(txout) && txout.nValue >= nMinimumInputValue)
|
|
return true;
|
|
return false;
|
|
}
|
|
bool IsFromMe(const CTransaction& tx) const
|
|
{
|
|
return (GetDebit(tx) > 0);
|
|
}
|
|
int64_t GetDebit(const CTransaction& tx) const
|
|
{
|
|
int64_t nDebit = 0;
|
|
for (const CTxIn& txin : tx.vin)
|
|
{
|
|
nDebit += GetDebit(txin);
|
|
if (!MoneyRange(nDebit))
|
|
throw std::runtime_error("CWallet::GetDebit() : value out of range");
|
|
}
|
|
return nDebit;
|
|
}
|
|
int64_t GetCredit(const CTransaction& tx) const
|
|
{
|
|
int64_t nCredit = 0;
|
|
for (const CTxOut& txout : tx.vout)
|
|
{
|
|
nCredit += GetCredit(txout);
|
|
if (!MoneyRange(nCredit))
|
|
throw std::runtime_error("CWallet::GetCredit() : value out of range");
|
|
}
|
|
return nCredit;
|
|
}
|
|
int64_t GetChange(const CTransaction& tx) const
|
|
{
|
|
int64_t nChange = 0;
|
|
for (const CTxOut& txout : tx.vout)
|
|
{
|
|
nChange += GetChange(txout);
|
|
if (!MoneyRange(nChange))
|
|
throw std::runtime_error("CWallet::GetChange() : value out of range");
|
|
}
|
|
return nChange;
|
|
}
|
|
void SetBestChain(const CBlockLocator& loc);
|
|
|
|
DBErrors LoadWallet(bool& fFirstRunRet);
|
|
|
|
bool SetAddressBookName(const CTxDestination& address, const std::string& strName);
|
|
|
|
bool DelAddressBookName(const CTxDestination& address);
|
|
|
|
void UpdatedTransaction(const uint256 &hashTx);
|
|
|
|
void PrintWallet(const CBlock& block);
|
|
|
|
void Inventory(const uint256 &hash)
|
|
{
|
|
{
|
|
LOCK(cs_wallet);
|
|
std::map<uint256, int>::iterator mi = mapRequestCount.find(hash);
|
|
if (mi != mapRequestCount.end())
|
|
(*mi).second++;
|
|
}
|
|
}
|
|
|
|
unsigned int GetKeyPoolSize()
|
|
{
|
|
return setKeyPool.size();
|
|
}
|
|
|
|
bool GetTransaction(const uint256 &hashTx, CWalletTx& wtx);
|
|
|
|
bool SetDefaultKey(const CPubKey &vchPubKey);
|
|
|
|
// signify that a particular wallet feature is now used. this may change nWalletVersion and nWalletMaxVersion if those are lower
|
|
bool SetMinVersion(WalletFeature, CWalletDB* pwalletdbIn = nullptr, bool fExplicit = false);
|
|
|
|
// change which version we're allowed to upgrade to (note that this does not immediately imply upgrading to that format)
|
|
bool SetMaxVersion(int nVersion);
|
|
|
|
// get the current wallet format (the oldest client version guaranteed to understand this wallet)
|
|
int GetVersion() { return nWalletVersion; }
|
|
|
|
void FixSpentCoins(int& nMismatchSpent, int64_t& nBalanceInQuestion, bool fCheckOnly = false);
|
|
void DisableTransaction(const CTransaction &tx);
|
|
|
|
/** Address book entry changed.
|
|
* @note called with lock cs_wallet held.
|
|
*/
|
|
CSignal<void(CWallet*, const CTxDestination&, const std::string&, bool, ChangeType)> NotifyAddressBookChanged;
|
|
|
|
/** Wallet transaction added, removed or updated.
|
|
* @note called with lock cs_wallet held.
|
|
*/
|
|
CSignal<void(CWallet*, const uint256&, ChangeType)> NotifyTransactionChanged;
|
|
};
|
|
|
|
/** A key allocated from the key pool. */
|
|
class CReserveKey
|
|
{
|
|
protected:
|
|
CWallet* pwallet;
|
|
int64_t nIndex;
|
|
CPubKey vchPubKey;
|
|
public:
|
|
CReserveKey(CWallet* pwalletIn)
|
|
{
|
|
nIndex = -1;
|
|
pwallet = pwalletIn;
|
|
}
|
|
|
|
~CReserveKey()
|
|
{
|
|
if (!fShutdown)
|
|
ReturnKey();
|
|
}
|
|
|
|
void ReturnKey();
|
|
CPubKey GetReservedKey();
|
|
void KeepKey();
|
|
};
|
|
|
|
|
|
|
|
static void ReadOrderPos(int64_t& nOrderPos, mapValue_t& mapValue)
|
|
{
|
|
if (!mapValue.count("n"))
|
|
{
|
|
nOrderPos = -1; // TODO: calculate elsewhere
|
|
return;
|
|
}
|
|
nOrderPos = atoi64(mapValue["n"].c_str());
|
|
}
|
|
|
|
|
|
static void WriteOrderPos(const int64_t& nOrderPos, mapValue_t& mapValue)
|
|
{
|
|
if (nOrderPos == -1)
|
|
return;
|
|
mapValue["n"] = i64tostr(nOrderPos);
|
|
}
|
|
|
|
|
|
/** A transaction with a bunch of additional info that only the owner cares about.
|
|
* It includes any unrecorded transactions needed to link it back to the block chain.
|
|
*/
|
|
class CWalletTx : public CMerkleTx
|
|
{
|
|
private:
|
|
const CWallet* pwallet;
|
|
|
|
public:
|
|
std::vector<CMerkleTx> vtxPrev;
|
|
mapValue_t mapValue;
|
|
std::vector<std::pair<std::string, std::string> > vOrderForm;
|
|
unsigned int fTimeReceivedIsTxTime;
|
|
unsigned int nTimeReceived; // time received by this node
|
|
unsigned int nTimeSmart;
|
|
char fFromMe;
|
|
std::string strFromAccount;
|
|
std::vector<char> vfSpent; // which outputs are already spent
|
|
int64_t nOrderPos; // position in ordered transaction list
|
|
|
|
// memory only
|
|
mutable bool fDebitCached;
|
|
mutable bool fCreditCached;
|
|
mutable bool fAvailableCreditCached;
|
|
mutable bool fChangeCached;
|
|
mutable int64_t nDebitCached;
|
|
mutable int64_t nCreditCached;
|
|
mutable int64_t nAvailableCreditCached;
|
|
mutable int64_t nChangeCached;
|
|
|
|
CWalletTx()
|
|
{
|
|
Init(nullptr);
|
|
}
|
|
|
|
CWalletTx(const CWallet* pwalletIn)
|
|
{
|
|
Init(pwalletIn);
|
|
}
|
|
|
|
CWalletTx(const CWallet* pwalletIn, const CMerkleTx& txIn) : CMerkleTx(txIn)
|
|
{
|
|
Init(pwalletIn);
|
|
}
|
|
|
|
CWalletTx(const CWallet* pwalletIn, const CTransaction& txIn) : CMerkleTx(txIn)
|
|
{
|
|
Init(pwalletIn);
|
|
}
|
|
|
|
void Init(const CWallet* pwalletIn)
|
|
{
|
|
pwallet = pwalletIn;
|
|
vtxPrev.clear();
|
|
mapValue.clear();
|
|
vOrderForm.clear();
|
|
fTimeReceivedIsTxTime = false;
|
|
nTimeReceived = 0;
|
|
nTimeSmart = 0;
|
|
fFromMe = false;
|
|
strFromAccount.clear();
|
|
vfSpent.clear();
|
|
fDebitCached = false;
|
|
fCreditCached = false;
|
|
fAvailableCreditCached = false;
|
|
fChangeCached = false;
|
|
nDebitCached = 0;
|
|
nCreditCached = 0;
|
|
nAvailableCreditCached = 0;
|
|
nChangeCached = 0;
|
|
nOrderPos = -1;
|
|
}
|
|
|
|
IMPLEMENT_SERIALIZE
|
|
(
|
|
CWalletTx* pthis = const_cast<CWalletTx*>(this);
|
|
if (fRead)
|
|
pthis->Init(nullptr);
|
|
char fSpent = false;
|
|
|
|
if (!fRead)
|
|
{
|
|
pthis->mapValue["fromaccount"] = pthis->strFromAccount;
|
|
|
|
std::string str;
|
|
for (char f : vfSpent)
|
|
{
|
|
str += (f ? '1' : '0');
|
|
if (f)
|
|
fSpent = true;
|
|
}
|
|
pthis->mapValue["spent"] = str;
|
|
|
|
WriteOrderPos(pthis->nOrderPos, pthis->mapValue);
|
|
|
|
if (nTimeSmart)
|
|
pthis->mapValue["timesmart"] = strprintf("%u", nTimeSmart);
|
|
}
|
|
|
|
nSerSize += SerReadWrite(s, *(CMerkleTx*)this, nType, nVersion,ser_action);
|
|
READWRITE(vtxPrev);
|
|
READWRITE(mapValue);
|
|
READWRITE(vOrderForm);
|
|
READWRITE(fTimeReceivedIsTxTime);
|
|
READWRITE(nTimeReceived);
|
|
READWRITE(fFromMe);
|
|
READWRITE(fSpent);
|
|
|
|
if (fRead)
|
|
{
|
|
pthis->strFromAccount = pthis->mapValue["fromaccount"];
|
|
|
|
if (mapValue.count("spent"))
|
|
for (char c : pthis->mapValue["spent"])
|
|
pthis->vfSpent.push_back(c != '0');
|
|
else
|
|
pthis->vfSpent.assign(vout.size(), fSpent);
|
|
|
|
ReadOrderPos(pthis->nOrderPos, pthis->mapValue);
|
|
|
|
pthis->nTimeSmart = mapValue.count("timesmart") ? (unsigned int)atoi64(pthis->mapValue["timesmart"]) : 0;
|
|
}
|
|
|
|
pthis->mapValue.erase("fromaccount");
|
|
pthis->mapValue.erase("version");
|
|
pthis->mapValue.erase("spent");
|
|
pthis->mapValue.erase("n");
|
|
pthis->mapValue.erase("timesmart");
|
|
)
|
|
|
|
// marks certain txout's as spent
|
|
// returns true if any update took place
|
|
bool UpdateSpent(const std::vector<char>& vfNewSpent)
|
|
{
|
|
bool fReturn = false;
|
|
for (unsigned int i = 0; i < vfNewSpent.size(); i++)
|
|
{
|
|
if (i == vfSpent.size())
|
|
break;
|
|
|
|
if (vfNewSpent[i] && !vfSpent[i])
|
|
{
|
|
vfSpent[i] = true;
|
|
fReturn = true;
|
|
fAvailableCreditCached = false;
|
|
}
|
|
}
|
|
return fReturn;
|
|
}
|
|
|
|
// make sure balances are recalculated
|
|
void MarkDirty()
|
|
{
|
|
fCreditCached = false;
|
|
fAvailableCreditCached = false;
|
|
fDebitCached = false;
|
|
fChangeCached = false;
|
|
}
|
|
|
|
void BindWallet(CWallet *pwalletIn)
|
|
{
|
|
pwallet = pwalletIn;
|
|
MarkDirty();
|
|
}
|
|
|
|
void MarkSpent(unsigned int nOut)
|
|
{
|
|
if (nOut >= vout.size())
|
|
throw std::runtime_error("CWalletTx::MarkSpent() : nOut out of range");
|
|
vfSpent.resize(vout.size());
|
|
if (!vfSpent[nOut])
|
|
{
|
|
vfSpent[nOut] = true;
|
|
fAvailableCreditCached = false;
|
|
}
|
|
}
|
|
|
|
void MarkUnspent(unsigned int nOut)
|
|
{
|
|
if (nOut >= vout.size())
|
|
throw std::runtime_error("CWalletTx::MarkUnspent() : nOut out of range");
|
|
vfSpent.resize(vout.size());
|
|
if (vfSpent[nOut])
|
|
{
|
|
vfSpent[nOut] = false;
|
|
fAvailableCreditCached = false;
|
|
}
|
|
}
|
|
|
|
bool IsSpent(unsigned int nOut) const
|
|
{
|
|
if (nOut >= vout.size())
|
|
throw std::runtime_error("CWalletTx::IsSpent() : nOut out of range");
|
|
if (nOut >= vfSpent.size())
|
|
return false;
|
|
return (!!vfSpent[nOut]);
|
|
}
|
|
|
|
int64_t GetDebit() const
|
|
{
|
|
if (vin.empty())
|
|
return 0;
|
|
LOCK(pwallet->cs_wallet);
|
|
if (fDebitCached)
|
|
return nDebitCached;
|
|
nDebitCached = pwallet->GetDebit(*this);
|
|
fDebitCached = true;
|
|
return nDebitCached;
|
|
}
|
|
|
|
int64_t GetCredit(bool fUseCache=true) const
|
|
{
|
|
if ((IsCoinBase() || IsCoinStake()) && GetBlocksToMaturity() > 0)
|
|
return 0;
|
|
|
|
LOCK(pwallet->cs_wallet);
|
|
if (fUseCache && fCreditCached)
|
|
return nCreditCached;
|
|
nCreditCached = pwallet->GetCredit(*this);
|
|
fCreditCached = true;
|
|
return nCreditCached;
|
|
}
|
|
|
|
int64_t GetAvailableCredit(bool fUseCache=true) const
|
|
{
|
|
if ((IsCoinBase() || IsCoinStake()) && GetBlocksToMaturity() > 0)
|
|
return 0;
|
|
|
|
LOCK(pwallet->cs_wallet);
|
|
if (fUseCache && fAvailableCreditCached)
|
|
return nAvailableCreditCached;
|
|
|
|
int64_t nCredit = 0;
|
|
for (unsigned int i = 0; i < vout.size(); i++)
|
|
{
|
|
if (!IsSpent(i))
|
|
{
|
|
const CTxOut &txout = vout[i];
|
|
nCredit += pwallet->GetCredit(txout);
|
|
if (!MoneyRange(nCredit))
|
|
throw std::runtime_error("CWalletTx::GetAvailableCredit() : value out of range");
|
|
}
|
|
}
|
|
|
|
nAvailableCreditCached = nCredit;
|
|
fAvailableCreditCached = true;
|
|
return nCredit;
|
|
}
|
|
|
|
|
|
int64_t GetChange() const
|
|
{
|
|
LOCK(pwallet->cs_wallet);
|
|
if (fChangeCached)
|
|
return nChangeCached;
|
|
nChangeCached = pwallet->GetChange(*this);
|
|
fChangeCached = true;
|
|
return nChangeCached;
|
|
}
|
|
|
|
void GetAmounts(std::list<std::pair<CTxDestination, int64_t> >& listReceived,
|
|
std::list<std::pair<CTxDestination, int64_t> >& listSent, int64_t& nFee, std::string& strSentAccount) const;
|
|
|
|
void GetAccountAmounts(const std::string& strAccount, int64_t& nReceived,
|
|
int64_t& nSent, int64_t& nFee) const;
|
|
|
|
bool IsFromMe() const
|
|
{
|
|
return (GetDebit() > 0);
|
|
}
|
|
|
|
bool IsTrusted() const
|
|
{
|
|
// Quick answer in most cases
|
|
if (!IsFinal())
|
|
return false;
|
|
int nDepth = GetDepthInMainChain();
|
|
//traingles fix: show as available balance after 4 confirmations (instead of after 1 confirmation)
|
|
if (nDepth >= 4)
|
|
return true;
|
|
if (nDepth < 0)
|
|
return false;
|
|
if (fConfChange || !IsFromMe()) // using wtx's cached debit
|
|
return false;
|
|
|
|
// If no confirmations but it's from us, we can still
|
|
// consider it confirmed if all dependencies are confirmed
|
|
std::map<uint256, const CMerkleTx*> mapPrev;
|
|
std::vector<const CMerkleTx*> vWorkQueue;
|
|
vWorkQueue.reserve(vtxPrev.size()+1);
|
|
vWorkQueue.push_back(this);
|
|
for (unsigned int i = 0; i < vWorkQueue.size(); i++)
|
|
{
|
|
const CMerkleTx* ptx = vWorkQueue[i];
|
|
|
|
if (!ptx->IsFinal())
|
|
return false;
|
|
int nPDepth = ptx->GetDepthInMainChain();
|
|
if (nPDepth >= 1)
|
|
continue;
|
|
if (nPDepth < 0)
|
|
return false;
|
|
if (!pwallet->IsFromMe(*ptx))
|
|
return false;
|
|
|
|
if (mapPrev.empty())
|
|
{
|
|
for (const CMerkleTx& tx : vtxPrev)
|
|
mapPrev[tx.GetHash()] = &tx;
|
|
}
|
|
|
|
for (const CTxIn& txin : ptx->vin)
|
|
{
|
|
if (!mapPrev.count(txin.prevout.hash))
|
|
return false;
|
|
vWorkQueue.push_back(mapPrev[txin.prevout.hash]);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool WriteToDisk();
|
|
|
|
int64_t GetTxTime() const;
|
|
int GetRequestCount() const;
|
|
|
|
void AddSupportingTransactions(CTxDBBase& txdb);
|
|
|
|
bool AcceptWalletTransaction(CTxDBBase& txdb, bool fCheckInputs=true);
|
|
bool AcceptWalletTransaction();
|
|
|
|
void RelayWalletTransaction(CTxDBBase& txdb);
|
|
void RelayWalletTransaction();
|
|
};
|
|
|
|
|
|
|
|
|
|
class COutput
|
|
{
|
|
public:
|
|
const CWalletTx *tx;
|
|
int i;
|
|
int nDepth;
|
|
|
|
COutput(const CWalletTx *txIn, int iIn, int nDepthIn)
|
|
{
|
|
tx = txIn; i = iIn; nDepth = nDepthIn;
|
|
}
|
|
|
|
std::string ToString() const
|
|
{
|
|
return strprintf("COutput(%s, %d, %d) [%s]", tx->GetHash().ToString().substr(0,10).c_str(), i, nDepth, FormatMoney(tx->vout[i].nValue).c_str());
|
|
}
|
|
|
|
void print() const
|
|
{
|
|
printf("%s\n", ToString().c_str());
|
|
}
|
|
};
|
|
|
|
|
|
|
|
|
|
/** Private key that includes an expiration date in case it never gets used. */
|
|
class CWalletKey
|
|
{
|
|
public:
|
|
CPrivKey vchPrivKey;
|
|
int64_t nTimeCreated;
|
|
int64_t nTimeExpires;
|
|
std::string strComment;
|
|
//// todo: add something to note what created it (user, getnewaddress, change)
|
|
//// maybe should have a map<string, string> property map
|
|
|
|
CWalletKey(int64_t nExpires=0)
|
|
{
|
|
nTimeCreated = (nExpires ? GetTime() : 0);
|
|
nTimeExpires = nExpires;
|
|
}
|
|
|
|
IMPLEMENT_SERIALIZE
|
|
(
|
|
if (!(nType & SER_GETHASH))
|
|
READWRITE(nVersion);
|
|
READWRITE(vchPrivKey);
|
|
READWRITE(nTimeCreated);
|
|
READWRITE(nTimeExpires);
|
|
READWRITE(strComment);
|
|
)
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** Account information.
|
|
* Stored in wallet with key "acc"+string account name.
|
|
*/
|
|
class CAccount
|
|
{
|
|
public:
|
|
CPubKey vchPubKey;
|
|
|
|
CAccount()
|
|
{
|
|
SetNull();
|
|
}
|
|
|
|
void SetNull()
|
|
{
|
|
vchPubKey = CPubKey();
|
|
}
|
|
|
|
IMPLEMENT_SERIALIZE
|
|
(
|
|
if (!(nType & SER_GETHASH))
|
|
READWRITE(nVersion);
|
|
READWRITE(vchPubKey);
|
|
)
|
|
};
|
|
|
|
|
|
|
|
/** Internal transfers.
|
|
* Database key is acentry<account><counter>.
|
|
*/
|
|
class CAccountingEntry
|
|
{
|
|
public:
|
|
std::string strAccount;
|
|
int64_t nCreditDebit;
|
|
int64_t nTime;
|
|
std::string strOtherAccount;
|
|
std::string strComment;
|
|
mapValue_t mapValue;
|
|
int64_t nOrderPos; // position in ordered transaction list
|
|
uint64_t nEntryNo;
|
|
|
|
CAccountingEntry()
|
|
{
|
|
SetNull();
|
|
}
|
|
|
|
void SetNull()
|
|
{
|
|
nCreditDebit = 0;
|
|
nTime = 0;
|
|
strAccount.clear();
|
|
strOtherAccount.clear();
|
|
strComment.clear();
|
|
nOrderPos = -1;
|
|
}
|
|
|
|
IMPLEMENT_SERIALIZE
|
|
(
|
|
CAccountingEntry& me = *const_cast<CAccountingEntry*>(this);
|
|
if (!(nType & SER_GETHASH))
|
|
READWRITE(nVersion);
|
|
// Note: strAccount is serialized as part of the key, not here.
|
|
READWRITE(nCreditDebit);
|
|
READWRITE(nTime);
|
|
READWRITE(strOtherAccount);
|
|
|
|
if (!fRead)
|
|
{
|
|
WriteOrderPos(nOrderPos, me.mapValue);
|
|
|
|
if (!(mapValue.empty() && _ssExtra.empty()))
|
|
{
|
|
CDataStream ss(nType, nVersion);
|
|
ss.insert(ss.begin(), '\0');
|
|
ss << mapValue;
|
|
ss.insert(ss.end(), _ssExtra.begin(), _ssExtra.end());
|
|
me.strComment.append(ss.str());
|
|
}
|
|
}
|
|
|
|
READWRITE(strComment);
|
|
|
|
size_t nSepPos = strComment.find("\0", 0, 1);
|
|
if (fRead)
|
|
{
|
|
me.mapValue.clear();
|
|
if (std::string::npos != nSepPos)
|
|
{
|
|
CDataStream ss(std::vector<char>(strComment.begin() + nSepPos + 1, strComment.end()), nType, nVersion);
|
|
ss >> me.mapValue;
|
|
me._ssExtra = std::vector<char>(ss.begin(), ss.end());
|
|
}
|
|
ReadOrderPos(me.nOrderPos, me.mapValue);
|
|
}
|
|
if (std::string::npos != nSepPos)
|
|
me.strComment.erase(nSepPos);
|
|
|
|
me.mapValue.erase("n");
|
|
)
|
|
|
|
private:
|
|
std::vector<char> _ssExtra;
|
|
};
|
|
|
|
bool GetWalletFile(CWallet* pwallet, std::string &strWalletFileOut);
|
|
|
|
#endif
|