mirror of
https://github.com/XRPLF/rippled.git
synced 2025-12-06 17:27:55 +00:00
138 lines
3.7 KiB
C++
138 lines
3.7 KiB
C++
#ifndef __LEDGERACQUIRE__
|
|
#define __LEDGERACQUIRE__
|
|
|
|
#include <vector>
|
|
#include <map>
|
|
#include <list>
|
|
|
|
#include <boost/enable_shared_from_this.hpp>
|
|
#include <boost/function.hpp>
|
|
#include <boost/asio.hpp>
|
|
#include <boost/thread/mutex.hpp>
|
|
#include <boost/weak_ptr.hpp>
|
|
|
|
#include "Ledger.h"
|
|
#include "Peer.h"
|
|
#include "TaggedCache.h"
|
|
#include "../obj/src/newcoin.pb.h"
|
|
|
|
class PeerSet
|
|
{
|
|
protected:
|
|
uint256 mHash;
|
|
int mTimerInterval;
|
|
bool mComplete, mFailed;
|
|
|
|
boost::recursive_mutex mLock;
|
|
boost::asio::deadline_timer mTimer;
|
|
std::vector< boost::weak_ptr<Peer> > mPeers;
|
|
|
|
PeerSet(const uint256& hash, int interval);
|
|
virtual ~PeerSet() { ; }
|
|
|
|
void sendRequest(const newcoin::TMGetLedger& message);
|
|
void sendRequest(const newcoin::TMGetLedger& message, Peer::pointer peer);
|
|
|
|
public:
|
|
const uint256& getHash() const { return mHash; }
|
|
bool isComplete() const { return mComplete; }
|
|
bool isFailed() const { return mFailed; }
|
|
|
|
void peerHas(Peer::pointer);
|
|
void badPeer(Peer::pointer);
|
|
void resetTimer();
|
|
|
|
protected:
|
|
virtual void newPeer(Peer::pointer) = 0;
|
|
virtual void onTimer(void) = 0;
|
|
virtual boost::weak_ptr<PeerSet> pmDowncast() = 0;
|
|
|
|
void setComplete() { mComplete = true; }
|
|
void setFailed() { mFailed = true; }
|
|
|
|
private:
|
|
static void TimerEntry(boost::weak_ptr<PeerSet>, const boost::system::error_code& result);
|
|
};
|
|
|
|
typedef TaggedCache< uint256, std::vector<unsigned char> > NodeCache;
|
|
typedef std::vector<unsigned char> VUC;
|
|
|
|
class THSyncFilter : public SHAMapSyncFilter
|
|
{
|
|
protected:
|
|
NodeCache* mCache; // holds nodes we see during the consensus process
|
|
|
|
public:
|
|
THSyncFilter(NodeCache* cache) : mCache(cache) { ; }
|
|
virtual void gotNode(const SHAMapNode& id, const uint256& nodeHash,
|
|
const std::vector<unsigned char>& nodeData, bool)
|
|
{
|
|
boost::shared_ptr<VUC> ptr = boost::make_shared<VUC>(nodeData);
|
|
mCache->canonicalize(nodeHash, ptr);
|
|
}
|
|
virtual bool haveNode(const SHAMapNode& id, const uint256& nodeHash, std::vector<unsigned char>& nodeData)
|
|
{
|
|
boost::shared_ptr<VUC> entry = mCache->fetch(nodeHash);
|
|
if (!entry) return false;
|
|
nodeData = *entry;
|
|
return true;
|
|
}
|
|
};
|
|
|
|
class LedgerAcquire : public PeerSet, public boost::enable_shared_from_this<LedgerAcquire>
|
|
{ // A ledger we are trying to acquire
|
|
public:
|
|
typedef boost::shared_ptr<LedgerAcquire> pointer;
|
|
|
|
protected:
|
|
Ledger::pointer mLedger;
|
|
THSyncFilter mFilter;
|
|
bool mHaveBase, mHaveState, mHaveTransactions;
|
|
|
|
std::vector< boost::function<void (LedgerAcquire::pointer)> > mOnComplete;
|
|
|
|
void done();
|
|
void onTimer() { trigger(Peer::pointer()); }
|
|
|
|
void newPeer(Peer::pointer peer) { trigger(peer); }
|
|
void trigger(Peer::pointer);
|
|
|
|
boost::weak_ptr<PeerSet> pmDowncast();
|
|
|
|
public:
|
|
LedgerAcquire(const uint256& hash);
|
|
|
|
bool isBase() const { return mHaveBase; }
|
|
bool isAcctStComplete() const { return mHaveState; }
|
|
bool isTransComplete() const { return mHaveTransactions; }
|
|
Ledger::pointer getLedger() { return mLedger; }
|
|
|
|
void addOnComplete(boost::function<void (LedgerAcquire::pointer)>);
|
|
|
|
bool takeBase(const std::string& data, Peer::pointer);
|
|
bool takeTxNode(const std::list<SHAMapNode>& IDs, const std::list<std::vector<unsigned char> >& data,
|
|
Peer::pointer);
|
|
bool takeAsNode(const std::list<SHAMapNode>& IDs, const std::list<std::vector<unsigned char> >& data,
|
|
Peer::pointer);
|
|
};
|
|
|
|
class LedgerAcquireMaster
|
|
{
|
|
protected:
|
|
boost::mutex mLock;
|
|
std::map<uint256, LedgerAcquire::pointer> mLedgers;
|
|
|
|
public:
|
|
LedgerAcquireMaster() { ; }
|
|
|
|
LedgerAcquire::pointer findCreate(const uint256& hash);
|
|
LedgerAcquire::pointer find(const uint256& hash);
|
|
bool hasLedger(const uint256& ledgerHash);
|
|
void dropLedger(const uint256& ledgerHash);
|
|
bool gotLedgerData(newcoin::TMLedgerData& packet, Peer::pointer);
|
|
};
|
|
|
|
#endif
|
|
|
|
// vim:ts=4
|