tomFlowee
defd3950f0
The CAmount name is not helpful as its just an int64_t and not a class, like the name implies. There were a handful of places where it was passed in as const-ref, as a good example of this actually creating sub-par code.
285 lines
10 KiB
C++
285 lines
10 KiB
C++
/*
|
|
* This file is part of the Flowee project
|
|
* Copyright (C) 2011-2015 The Bitcoin Core developers
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef FLOWEE_QT_WALLETMODEL_H
|
|
#define FLOWEE_QT_WALLETMODEL_H
|
|
|
|
#include "paymentrequestplus.h"
|
|
#include "walletmodeltransaction.h"
|
|
|
|
#include "support/allocators/secure.h"
|
|
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
#include <QObject>
|
|
|
|
class AddressTableModel;
|
|
class OptionsModel;
|
|
class PlatformStyle;
|
|
class RecentRequestsTableModel;
|
|
class TransactionTableModel;
|
|
class WalletModelTransaction;
|
|
|
|
class CCoinControl;
|
|
class CKeyID;
|
|
class COutPoint;
|
|
class COutput;
|
|
class CPubKey;
|
|
class CWallet;
|
|
class uint256;
|
|
|
|
QT_BEGIN_NAMESPACE
|
|
class QTimer;
|
|
QT_END_NAMESPACE
|
|
|
|
class SendCoinsRecipient
|
|
{
|
|
public:
|
|
explicit SendCoinsRecipient() : amount(0), fSubtractFeeFromAmount(false), nVersion(SendCoinsRecipient::CURRENT_VERSION) { }
|
|
explicit SendCoinsRecipient(const QString &addr, const QString &label, const int64_t& amount, const QString &message):
|
|
address(addr), label(label), amount(amount), message(message), fSubtractFeeFromAmount(false), nVersion(SendCoinsRecipient::CURRENT_VERSION) {}
|
|
|
|
// If from an unauthenticated payment request, this is used for storing
|
|
// the addresses, e.g. address-A<br />address-B<br />address-C.
|
|
// Info: As we don't need to process addresses in here when using
|
|
// payment requests, we can abuse it for displaying an address list.
|
|
// Todo: This is a hack, should be replaced with a cleaner solution!
|
|
QString address;
|
|
QString label;
|
|
int64_t amount;
|
|
// If from a payment request, this is used for storing the memo
|
|
QString message;
|
|
|
|
// If from a payment request, paymentRequest.IsInitialized() will be true
|
|
PaymentRequestPlus paymentRequest;
|
|
// Empty if no authentication or invalid signature/cert/etc.
|
|
QString authenticatedMerchant;
|
|
|
|
bool fSubtractFeeFromAmount; // memory only
|
|
|
|
static const int CURRENT_VERSION = 1;
|
|
int nVersion;
|
|
|
|
ADD_SERIALIZE_METHODS
|
|
|
|
template <typename Stream, typename Operation>
|
|
inline void SerializationOp(Stream& s, Operation ser_action, int nType, int nVersion) {
|
|
std::string sAddress = address.toStdString();
|
|
std::string sLabel = label.toStdString();
|
|
std::string sMessage = message.toStdString();
|
|
std::string sPaymentRequest;
|
|
if (!ser_action.ForRead() && paymentRequest.IsInitialized())
|
|
paymentRequest.SerializeToString(&sPaymentRequest);
|
|
std::string sAuthenticatedMerchant = authenticatedMerchant.toStdString();
|
|
|
|
READWRITE(this->nVersion);
|
|
nVersion = this->nVersion;
|
|
READWRITE(sAddress);
|
|
READWRITE(sLabel);
|
|
READWRITE(amount);
|
|
READWRITE(sMessage);
|
|
READWRITE(sPaymentRequest);
|
|
READWRITE(sAuthenticatedMerchant);
|
|
|
|
if (ser_action.ForRead())
|
|
{
|
|
address = QString::fromStdString(sAddress);
|
|
label = QString::fromStdString(sLabel);
|
|
message = QString::fromStdString(sMessage);
|
|
if (!sPaymentRequest.empty())
|
|
paymentRequest.parse(QByteArray::fromRawData(sPaymentRequest.data(), sPaymentRequest.size()));
|
|
authenticatedMerchant = QString::fromStdString(sAuthenticatedMerchant);
|
|
}
|
|
}
|
|
};
|
|
|
|
/** Interface to Bitcoin wallet from Qt view code. */
|
|
class WalletModel : public QObject
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
explicit WalletModel(const PlatformStyle *platformStyle, CWallet *wallet, OptionsModel *optionsModel, QObject *parent = 0);
|
|
~WalletModel();
|
|
|
|
enum StatusCode // Returned by sendCoins
|
|
{
|
|
OK,
|
|
InvalidAmount,
|
|
InvalidAddress,
|
|
AmountExceedsBalance,
|
|
AmountWithFeeExceedsBalance,
|
|
DuplicateAddress,
|
|
TransactionCreationFailed, // Error returned when wallet is still locked
|
|
TransactionCommitFailed,
|
|
AbsurdFee,
|
|
PaymentRequestExpired
|
|
};
|
|
|
|
enum EncryptionStatus
|
|
{
|
|
Unencrypted, // !wallet->IsCrypted()
|
|
Locked, // wallet->IsCrypted() && wallet->IsLocked()
|
|
Unlocked // wallet->IsCrypted() && !wallet->IsLocked()
|
|
};
|
|
|
|
OptionsModel *getOptionsModel();
|
|
AddressTableModel *getAddressTableModel();
|
|
TransactionTableModel *getTransactionTableModel();
|
|
RecentRequestsTableModel *getRecentRequestsTableModel();
|
|
|
|
int64_t getBalance(const CCoinControl *coinControl = NULL) const;
|
|
int64_t getUnconfirmedBalance() const;
|
|
int64_t getImmatureBalance() const;
|
|
bool haveWatchOnly() const;
|
|
int64_t getWatchBalance() const;
|
|
int64_t getWatchUnconfirmedBalance() const;
|
|
int64_t getWatchImmatureBalance() const;
|
|
EncryptionStatus getEncryptionStatus() const;
|
|
|
|
// Check address for validity
|
|
bool validateAddress(const QString &address);
|
|
|
|
// Return status record for SendCoins, contains error id + information
|
|
struct SendCoinsReturn
|
|
{
|
|
SendCoinsReturn(StatusCode status = OK):
|
|
status(status) {}
|
|
StatusCode status;
|
|
};
|
|
|
|
// prepare transaction for getting txfee before sending coins
|
|
SendCoinsReturn prepareTransaction(WalletModelTransaction &transaction, const CCoinControl *coinControl = NULL);
|
|
|
|
// Send coins to a list of recipients
|
|
SendCoinsReturn sendCoins(WalletModelTransaction &transaction);
|
|
|
|
// Wallet encryption
|
|
bool setWalletEncrypted(bool encrypted, const SecureString &passphrase);
|
|
// Passphrase only needed when unlocking
|
|
bool setWalletLocked(bool locked, const SecureString &passPhrase=SecureString());
|
|
bool changePassphrase(const SecureString &oldPass, const SecureString &newPass);
|
|
// Wallet backup
|
|
bool backupWallet(const QString &filename);
|
|
|
|
// RAI object for unlocking wallet, returned by requestUnlock()
|
|
class UnlockContext
|
|
{
|
|
public:
|
|
UnlockContext(WalletModel *wallet, bool valid, bool relock);
|
|
~UnlockContext();
|
|
|
|
bool isValid() const { return valid; }
|
|
|
|
// Copy operator and constructor transfer the context
|
|
UnlockContext(const UnlockContext& obj) { CopyFrom(obj); }
|
|
UnlockContext& operator=(const UnlockContext& rhs) { CopyFrom(rhs); return *this; }
|
|
private:
|
|
WalletModel *wallet;
|
|
bool valid;
|
|
mutable bool relock; // mutable, as it can be set to false by copying
|
|
|
|
void CopyFrom(const UnlockContext& rhs);
|
|
};
|
|
|
|
UnlockContext requestUnlock();
|
|
|
|
bool getPubKey(const CKeyID &address, CPubKey& vchPubKeyOut) const;
|
|
bool havePrivKey(const CKeyID &address) const;
|
|
void getOutputs(const std::vector<COutPoint>& vOutpoints, std::vector<COutput>& vOutputs);
|
|
bool isSpent(const COutPoint& outpoint) const;
|
|
void listCoins(std::map<QString, std::vector<COutput> >& mapCoins) const;
|
|
|
|
bool isLockedCoin(uint256 hash, unsigned int n) const;
|
|
void lockCoin(COutPoint& output);
|
|
void unlockCoin(COutPoint& output);
|
|
void listLockedCoins(std::vector<COutPoint>& vOutpts);
|
|
|
|
void loadReceiveRequests(std::vector<std::string>& vReceiveRequests);
|
|
bool saveReceiveRequest(const std::string &sAddress, const int64_t nId, const std::string &sRequest);
|
|
|
|
private:
|
|
CWallet *wallet;
|
|
bool fHaveWatchOnly;
|
|
bool fForceCheckBalanceChanged;
|
|
|
|
// Wallet has an options model for wallet-specific options
|
|
// (transaction fee, for example)
|
|
OptionsModel *optionsModel;
|
|
|
|
AddressTableModel *addressTableModel;
|
|
TransactionTableModel *transactionTableModel;
|
|
RecentRequestsTableModel *recentRequestsTableModel;
|
|
|
|
// Cache some values to be able to detect changes
|
|
int64_t cachedBalance;
|
|
int64_t cachedUnconfirmedBalance;
|
|
int64_t cachedImmatureBalance;
|
|
int64_t cachedWatchOnlyBalance;
|
|
int64_t cachedWatchUnconfBalance;
|
|
int64_t cachedWatchImmatureBalance;
|
|
EncryptionStatus cachedEncryptionStatus;
|
|
int cachedNumBlocks;
|
|
|
|
QTimer *pollTimer;
|
|
|
|
void subscribeToCoreSignals();
|
|
void unsubscribeFromCoreSignals();
|
|
void checkBalanceChanged();
|
|
|
|
Q_SIGNALS:
|
|
// Signal that balance in wallet changed
|
|
void balanceChanged(const int64_t& balance, const int64_t& unconfirmedBalance, const int64_t& immatureBalance,
|
|
const int64_t& watchOnlyBalance, const int64_t& watchUnconfBalance, const int64_t& watchImmatureBalance);
|
|
|
|
// Encryption status of wallet changed
|
|
void encryptionStatusChanged(int status);
|
|
|
|
// Signal emitted when wallet needs to be unlocked
|
|
// It is valid behaviour for listeners to keep the wallet locked after this signal;
|
|
// this means that the unlocking failed or was cancelled.
|
|
void requireUnlock();
|
|
|
|
// Fired when a message should be reported to the user
|
|
void message(const QString &title, const QString &message, unsigned int style);
|
|
|
|
// Coins sent: from wallet, to recipient, in (serialized) transaction:
|
|
void coinsSent(CWallet* wallet, SendCoinsRecipient recipient, QByteArray transaction);
|
|
|
|
// Show progress dialog e.g. for rescan
|
|
void showProgress(const QString &title, int nProgress);
|
|
|
|
// Watch-only address added
|
|
void notifyWatchonlyChanged(bool fHaveWatchonly);
|
|
|
|
public Q_SLOTS:
|
|
/* Wallet status might have changed */
|
|
void updateStatus();
|
|
/* New transaction, or transaction changed status */
|
|
void updateTransaction();
|
|
/* New, updated or removed address book entry */
|
|
void updateAddressBook(const QString &address, const QString &label, bool isMine, const QString &purpose, int status);
|
|
/* Watch-only added */
|
|
void updateWatchOnlyFlag(bool fHaveWatchonly);
|
|
/* Current, immature or unconfirmed balance might have changed - emit 'balanceChanged' if so */
|
|
void pollBalanceChanged();
|
|
};
|
|
|
|
#endif
|