thehub/hub/api/TransactionMonitorService.h

/*
 * This file is part of the Flowee project
 * Copyright (C) 2018-2019 Tom Zander <tom@flowee.org>
 *
 * 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 TRANSACTIONMONITORSERVICE_H
#define TRANSACTIONMONITORSERVICE_H

#include <primitives/PublicKey.h>

#include <validationinterface.h>
#include <NetworkService.h>
#include <NetworkConnection.h>
#include <primitives/Tx.h>
#include <script/standard.h>

#include <set>
#include <memory>

class CTxMemPool;

class TransactionMonitorService : public ValidationInterface, public NetworkService
{
public:
    TransactionMonitorService();
    ~TransactionMonitorService() override;

    // the hub pushed a transaction into its mempool
    void syncTx(const Tx &tx) override;
    void syncAllTransactionsInBlock(const Block &block, CBlockIndex *index) override;
    void doubleSpendFound(const Tx &first, const Tx &duplicate) override;
    void doubleSpendFound(const Tx &txInMempool, const DoubleSpendProof &proof) override;

    void onIncomingMessage(Remote *con, const Message &message, const EndPoint &ep) override;

    inline void setMempool(CTxMemPool *mempool) {
        m_mempool = mempool;
    }

protected:
    class RemoteWithHashes : public Remote {
    public:
        std::set<uint256> hashes;
    };

    // NetworkService interface
    Remote *createRemote() override {
        return new RemoteWithHashes();
    }

private:
    void updateBools();
    /// Callback for just subscribed addresses to find if there is a hit in the mempool.
    void findTxInMempool(int connectionId, const uint256 &hash);

    std::mutex m_poolMutex;
    std::shared_ptr<Streaming::BufferPool> m_pool;

    // true if any remote added a watch
    bool m_findByHash = false;

    CTxMemPool *m_mempool = nullptr;
};

#endif