309 lines
14 KiB
C++
Executable file
309 lines
14 KiB
C++
Executable file
// Copyright (c) 2011-2016 The Cryptonote developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#pragma once
|
|
|
|
#include "IWallet.h"
|
|
|
|
#include <queue>
|
|
#include <unordered_map>
|
|
|
|
#include "IFusionManager.h"
|
|
#include "WalletIndices.h"
|
|
|
|
#include <System/Dispatcher.h>
|
|
#include <System/Event.h>
|
|
#include "Transfers/TransfersSynchronizer.h"
|
|
#include "Transfers/BlockchainSynchronizer.h"
|
|
|
|
namespace CryptoNote {
|
|
|
|
class WalletGreen : public IWallet,
|
|
ITransfersObserver,
|
|
IBlockchainSynchronizerObserver,
|
|
ITransfersSynchronizerObserver,
|
|
IFusionManager {
|
|
public:
|
|
WalletGreen(System::Dispatcher& dispatcher, const Currency& currency, INode& node, uint32_t transactionSoftLockTime = 1);
|
|
virtual ~WalletGreen();
|
|
|
|
virtual void initialize(const std::string& password) override;
|
|
virtual void initializeWithViewKey(const Crypto::SecretKey& viewSecretKey, const std::string& password) override;
|
|
virtual void load(std::istream& source, const std::string& password) override;
|
|
virtual void shutdown() override;
|
|
|
|
virtual void changePassword(const std::string& oldPassword, const std::string& newPassword) override;
|
|
virtual void save(std::ostream& destination, bool saveDetails = true, bool saveCache = true) override;
|
|
|
|
virtual size_t getAddressCount() const override;
|
|
virtual std::string getAddress(size_t index) const override;
|
|
virtual KeyPair getAddressSpendKey(size_t index) const override;
|
|
virtual KeyPair getAddressSpendKey(const std::string& address) const override;
|
|
virtual KeyPair getViewKey() const override;
|
|
virtual std::string createAddress() override;
|
|
virtual std::string createAddress(const Crypto::SecretKey& spendSecretKey) override;
|
|
virtual std::string createAddress(const Crypto::PublicKey& spendPublicKey) override;
|
|
virtual void deleteAddress(const std::string& address) override;
|
|
|
|
virtual uint64_t getActualBalance() const override;
|
|
virtual uint64_t getActualBalance(const std::string& address) const override;
|
|
virtual uint64_t getPendingBalance() const override;
|
|
virtual uint64_t getPendingBalance(const std::string& address) const override;
|
|
|
|
virtual size_t getTransactionCount() const override;
|
|
virtual WalletTransaction getTransaction(size_t transactionIndex) const override;
|
|
virtual size_t getTransactionTransferCount(size_t transactionIndex) const override;
|
|
virtual WalletTransfer getTransactionTransfer(size_t transactionIndex, size_t transferIndex) const override;
|
|
|
|
virtual WalletTransactionWithTransfers getTransaction(const Crypto::Hash& transactionHash) const override;
|
|
virtual std::vector<TransactionsInBlockInfo> getTransactions(const Crypto::Hash& blockHash, size_t count) const override;
|
|
virtual std::vector<TransactionsInBlockInfo> getTransactions(uint32_t blockIndex, size_t count) const override;
|
|
virtual std::vector<Crypto::Hash> getBlockHashes(uint32_t blockIndex, size_t count) const override;
|
|
virtual uint32_t getBlockCount() const override;
|
|
virtual std::vector<WalletTransactionWithTransfers> getUnconfirmedTransactions() const override;
|
|
virtual std::vector<size_t> getDelayedTransactionIds() const override;
|
|
|
|
virtual size_t transfer(const TransactionParameters& sendingTransaction) override;
|
|
|
|
virtual size_t makeTransaction(const TransactionParameters& sendingTransaction) override;
|
|
virtual void commitTransaction(size_t) override;
|
|
virtual void rollbackUncommitedTransaction(size_t) override;
|
|
|
|
virtual void start() override;
|
|
virtual void stop() override;
|
|
virtual WalletEvent getEvent() override;
|
|
|
|
virtual size_t createFusionTransaction(uint64_t threshold, uint64_t mixin) override;
|
|
virtual bool isFusionTransaction(size_t transactionId) const override;
|
|
virtual IFusionManager::EstimateResult estimate(uint64_t threshold) const override;
|
|
|
|
protected:
|
|
void throwIfNotInitialized() const;
|
|
void throwIfStopped() const;
|
|
void throwIfTrackingMode() const;
|
|
void doShutdown();
|
|
void clearCaches();
|
|
void initWithKeys(const Crypto::PublicKey& viewPublicKey, const Crypto::SecretKey& viewSecretKey, const std::string& password);
|
|
std::string doCreateAddress(const Crypto::PublicKey& spendPublicKey, const Crypto::SecretKey& spendSecretKey, uint64_t creationTimestamp);
|
|
|
|
struct InputInfo {
|
|
TransactionTypes::InputKeyInfo keyInfo;
|
|
WalletRecord* walletRecord = nullptr;
|
|
KeyPair ephKeys;
|
|
};
|
|
|
|
struct OutputToTransfer {
|
|
TransactionOutputInformation out;
|
|
WalletRecord* wallet;
|
|
};
|
|
|
|
struct ReceiverAmounts {
|
|
CryptoNote::AccountPublicAddress receiver;
|
|
std::vector<uint64_t> amounts;
|
|
};
|
|
|
|
struct WalletOuts {
|
|
WalletRecord* wallet;
|
|
std::vector<TransactionOutputInformation> outs;
|
|
};
|
|
|
|
typedef std::pair<WalletTransfers::const_iterator, WalletTransfers::const_iterator> TransfersRange;
|
|
|
|
struct AddressAmounts {
|
|
int64_t input = 0;
|
|
int64_t output = 0;
|
|
};
|
|
|
|
struct ContainerAmounts {
|
|
ITransfersContainer* container;
|
|
AddressAmounts amounts;
|
|
};
|
|
|
|
typedef std::unordered_map<std::string, AddressAmounts> TransfersMap;
|
|
|
|
virtual void onError(ITransfersSubscription* object, uint32_t height, std::error_code ec) override;
|
|
|
|
virtual void onTransactionUpdated(ITransfersSubscription* object, const Crypto::Hash& transactionHash) override;
|
|
virtual void onTransactionUpdated(const Crypto::PublicKey& viewPublicKey, const Crypto::Hash& transactionHash,
|
|
const std::vector<ITransfersContainer*>& containers) override;
|
|
void transactionUpdated(const TransactionInformation& transactionInfo, const std::vector<ContainerAmounts>& containerAmountsList);
|
|
|
|
virtual void onTransactionDeleted(ITransfersSubscription* object, const Crypto::Hash& transactionHash) override;
|
|
void transactionDeleted(ITransfersSubscription* object, const Crypto::Hash& transactionHash);
|
|
|
|
virtual void synchronizationProgressUpdated(uint32_t processedBlockCount, uint32_t totalBlockCount) override;
|
|
virtual void synchronizationCompleted(std::error_code result) override;
|
|
|
|
void onSynchronizationProgressUpdated(uint32_t processedBlockCount, uint32_t totalBlockCount);
|
|
void onSynchronizationCompleted();
|
|
|
|
virtual void onBlocksAdded(const Crypto::PublicKey& viewPublicKey, const std::vector<Crypto::Hash>& blockHashes) override;
|
|
void blocksAdded(const std::vector<Crypto::Hash>& blockHashes);
|
|
|
|
virtual void onBlockchainDetach(const Crypto::PublicKey& viewPublicKey, uint32_t blockIndex) override;
|
|
void blocksRollback(uint32_t blockIndex);
|
|
|
|
virtual void onTransactionDeleteBegin(const Crypto::PublicKey& viewPublicKey, Crypto::Hash transactionHash) override;
|
|
void transactionDeleteBegin(Crypto::Hash transactionHash);
|
|
|
|
virtual void onTransactionDeleteEnd(const Crypto::PublicKey& viewPublicKey, Crypto::Hash transactionHash) override;
|
|
void transactionDeleteEnd(Crypto::Hash transactionHash);
|
|
|
|
std::vector<WalletOuts> pickWalletsWithMoney() const;
|
|
WalletOuts pickWallet(const std::string& address);
|
|
std::vector<WalletOuts> pickWallets(const std::vector<std::string>& addresses);
|
|
|
|
void updateBalance(CryptoNote::ITransfersContainer* container);
|
|
void unlockBalances(uint32_t height);
|
|
|
|
const WalletRecord& getWalletRecord(const Crypto::PublicKey& key) const;
|
|
const WalletRecord& getWalletRecord(const std::string& address) const;
|
|
const WalletRecord& getWalletRecord(CryptoNote::ITransfersContainer* container) const;
|
|
|
|
CryptoNote::AccountPublicAddress parseAddress(const std::string& address) const;
|
|
std::string addWallet(const Crypto::PublicKey& spendPublicKey, const Crypto::SecretKey& spendSecretKey, uint64_t creationTimestamp);
|
|
AccountKeys makeAccountKeys(const WalletRecord& wallet) const;
|
|
size_t getTransactionId(const Crypto::Hash& transactionHash) const;
|
|
void pushEvent(const WalletEvent& event);
|
|
bool isFusionTransaction(const WalletTransaction& walletTx) const;
|
|
|
|
struct PreparedTransaction {
|
|
std::unique_ptr<ITransaction> transaction;
|
|
std::vector<WalletTransfer> destinations;
|
|
uint64_t neededMoney;
|
|
uint64_t changeAmount;
|
|
};
|
|
|
|
void prepareTransaction(std::vector<WalletOuts>&& wallets,
|
|
const std::vector<WalletOrder>& orders,
|
|
uint64_t fee,
|
|
uint64_t mixIn,
|
|
const std::string& extra,
|
|
uint64_t unlockTimestamp,
|
|
const DonationSettings& donation,
|
|
const CryptoNote::AccountPublicAddress& changeDestinationAddress,
|
|
PreparedTransaction& preparedTransaction);
|
|
|
|
void validateTransactionParameters(const TransactionParameters& transactionParameters);
|
|
size_t doTransfer(const TransactionParameters& transactionParameters);
|
|
|
|
void requestMixinOuts(const std::vector<OutputToTransfer>& selectedTransfers,
|
|
uint64_t mixIn,
|
|
std::vector<CryptoNote::COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount>& mixinResult);
|
|
|
|
void prepareInputs(const std::vector<OutputToTransfer>& selectedTransfers,
|
|
std::vector<CryptoNote::COMMAND_RPC_GET_RANDOM_OUTPUTS_FOR_AMOUNTS::outs_for_amount>& mixinResult,
|
|
uint64_t mixIn,
|
|
std::vector<InputInfo>& keysInfo);
|
|
|
|
uint64_t selectTransfers(uint64_t needeMoney,
|
|
bool dust,
|
|
uint64_t dustThreshold,
|
|
std::vector<WalletOuts>&& wallets,
|
|
std::vector<OutputToTransfer>& selectedTransfers);
|
|
|
|
std::vector<ReceiverAmounts> splitDestinations(const std::vector<WalletTransfer>& destinations,
|
|
uint64_t dustThreshold, const Currency& currency);
|
|
ReceiverAmounts splitAmount(uint64_t amount, const AccountPublicAddress& destination, uint64_t dustThreshold);
|
|
|
|
std::unique_ptr<CryptoNote::ITransaction> makeTransaction(const std::vector<ReceiverAmounts>& decomposedOutputs,
|
|
std::vector<InputInfo>& keysInfo, const std::string& extra, uint64_t unlockTimestamp);
|
|
|
|
void sendTransaction(const CryptoNote::Transaction& cryptoNoteTransaction);
|
|
size_t validateSaveAndSendTransaction(const ITransactionReader& transaction, const std::vector<WalletTransfer>& destinations, bool isFusion, bool send);
|
|
|
|
size_t insertBlockchainTransaction(const TransactionInformation& info, int64_t txBalance);
|
|
size_t insertOutgoingTransactionAndPushEvent(const Crypto::Hash& transactionHash, uint64_t fee, const BinaryArray& extra, uint64_t unlockTimestamp);
|
|
void updateTransactionStateAndPushEvent(size_t transactionId, WalletTransactionState state);
|
|
bool updateWalletTransactionInfo(size_t transactionId, const CryptoNote::TransactionInformation& info, int64_t totalAmount);
|
|
bool updateTransactionTransfers(size_t transactionId, const std::vector<ContainerAmounts>& containerAmountsList,
|
|
int64_t allInputsAmount, int64_t allOutputsAmount);
|
|
TransfersMap getKnownTransfersMap(size_t transactionId, size_t firstTransferIdx) const;
|
|
bool updateAddressTransfers(size_t transactionId, size_t firstTransferIdx, const std::string& address, int64_t knownAmount, int64_t targetAmount);
|
|
bool updateUnknownTransfers(size_t transactionId, size_t firstTransferIdx, const std::unordered_set<std::string>& myAddresses,
|
|
int64_t knownAmount, int64_t myAmount, int64_t totalAmount, bool isOutput);
|
|
void appendTransfer(size_t transactionId, size_t firstTransferIdx, const std::string& address, int64_t amount);
|
|
bool adjustTransfer(size_t transactionId, size_t firstTransferIdx, const std::string& address, int64_t amount);
|
|
bool eraseTransfers(size_t transactionId, size_t firstTransferIdx, std::function<bool(bool, const std::string&)>&& predicate);
|
|
bool eraseTransfersByAddress(size_t transactionId, size_t firstTransferIdx, const std::string& address, bool eraseOutputTransfers);
|
|
bool eraseForeignTransfers(size_t transactionId, size_t firstTransferIdx, const std::unordered_set<std::string>& knownAddresses, bool eraseOutputTransfers);
|
|
void pushBackOutgoingTransfers(size_t txId, const std::vector<WalletTransfer>& destinations);
|
|
void insertUnlockTransactionJob(const Crypto::Hash& transactionHash, uint32_t blockHeight, CryptoNote::ITransfersContainer* container);
|
|
void deleteUnlockTransactionJob(const Crypto::Hash& transactionHash);
|
|
void startBlockchainSynchronizer();
|
|
void stopBlockchainSynchronizer();
|
|
void addUnconfirmedTransaction(const ITransactionReader& transaction);
|
|
void removeUnconfirmedTransaction(const Crypto::Hash& transactionHash);
|
|
|
|
void unsafeLoad(std::istream& source, const std::string& password);
|
|
void unsafeSave(std::ostream& destination, bool saveDetails, bool saveCache);
|
|
|
|
std::vector<OutputToTransfer> pickRandomFusionInputs(uint64_t threshold, size_t minInputCount, size_t maxInputCount);
|
|
ReceiverAmounts decomposeFusionOutputs(uint64_t inputsAmount);
|
|
|
|
enum class WalletState {
|
|
INITIALIZED,
|
|
NOT_INITIALIZED
|
|
};
|
|
|
|
enum class WalletTrackingMode {
|
|
TRACKING,
|
|
NOT_TRACKING,
|
|
NO_ADDRESSES
|
|
};
|
|
|
|
WalletTrackingMode getTrackingMode() const;
|
|
|
|
TransfersRange getTransactionTransfersRange(size_t transactionIndex) const;
|
|
std::vector<TransactionsInBlockInfo> getTransactionsInBlocks(uint32_t blockIndex, size_t count) const;
|
|
Crypto::Hash getBlockHashByIndex(uint32_t blockIndex) const;
|
|
|
|
std::vector<WalletTransfer> getTransactionTransfers(const WalletTransaction& transaction) const;
|
|
void filterOutTransactions(WalletTransactions& transactions, WalletTransfers& transfers, std::function<bool (const WalletTransaction&)>&& pred) const;
|
|
void getViewKeyKnownBlocks(const Crypto::PublicKey& viewPublicKey);
|
|
CryptoNote::AccountPublicAddress getChangeDestination(const std::string& changeDestinationAddress, const std::vector<std::string>& sourceAddresses) const;
|
|
bool isMyAddress(const std::string& address) const;
|
|
|
|
void deleteContainerFromUnlockTransactionJobs(const ITransfersContainer* container);
|
|
std::vector<size_t> deleteTransfersForAddress(const std::string& address, std::vector<size_t>& deletedTransactions);
|
|
void deleteFromUncommitedTransactions(const std::vector<size_t>& deletedTransactions);
|
|
|
|
System::Dispatcher& m_dispatcher;
|
|
const Currency& m_currency;
|
|
INode& m_node;
|
|
bool m_stopped;
|
|
|
|
WalletsContainer m_walletsContainer;
|
|
UnlockTransactionJobs m_unlockTransactionsJob;
|
|
WalletTransactions m_transactions;
|
|
WalletTransfers m_transfers; //sorted
|
|
mutable std::unordered_map<size_t, bool> m_fusionTxsCache; // txIndex -> isFusion
|
|
UncommitedTransactions m_uncommitedTransactions;
|
|
|
|
bool m_blockchainSynchronizerStarted;
|
|
BlockchainSynchronizer m_blockchainSynchronizer;
|
|
TransfersSyncronizer m_synchronizer;
|
|
|
|
System::Event m_eventOccurred;
|
|
std::queue<WalletEvent> m_events;
|
|
mutable System::Event m_readyEvent;
|
|
|
|
WalletState m_state;
|
|
|
|
std::string m_password;
|
|
|
|
Crypto::PublicKey m_viewPublicKey;
|
|
Crypto::SecretKey m_viewSecretKey;
|
|
|
|
uint64_t m_actualBalance;
|
|
uint64_t m_pendingBalance;
|
|
|
|
uint64_t m_upperTransactionSizeLimit;
|
|
uint32_t m_transactionSoftLockTime;
|
|
|
|
BlockHashesContainer m_blockchain;
|
|
};
|
|
|
|
} //namespace CryptoNote
|