Differential D9418 Diff 28200 src/avalanche/processor.cpp

Changeset View

Standalone View

src/avalanche/processor.cpp

Show First 20 Lines • Show All 130 Lines • ▼ Show 20 Lines	static bool IsWorthPolling(const CBlockIndex *pindex) {
if (::ChainstateActive().IsBlockFinalized(pindex)) {		if (::ChainstateActive().IsBlockFinalized(pindex)) {
// There is no point polling finalized block.		// There is no point polling finalized block.
return false;		return false;
}		}

return true;		return true;
}		}

struct Processor::PeerData {
Proof proof;
Delegation delegation;
};

class Processor::NotificationsHandler		class Processor::NotificationsHandler
: public interfaces::Chain::Notifications {		: public interfaces::Chain::Notifications {
Processor *m_processor;		Processor *m_processor;

public:		public:
NotificationsHandler(Processor *p) : m_processor(p) {}		NotificationsHandler(Processor *p) : m_processor(p) {}

void updatedBlockTip() override {		void updatedBlockTip() override {
LOCK(m_processor->cs_peerManager);		LOCK(m_processor->cs_peerManager);

if (m_processor->mustRegisterProof &&		if (m_processor->mustRegisterProof &&
!::ChainstateActive().IsInitialBlockDownload()) {		!::ChainstateActive().IsInitialBlockDownload()) {
m_processor->peerManager->getPeerId(m_processor->peerData->proof);		m_processor->peerManager->getPeerId(
		m_processor->peerData.proof.value());
m_processor->mustRegisterProof = false;		m_processor->mustRegisterProof = false;
}		}

m_processor->peerManager->updatedBlockTip();		m_processor->peerManager->updatedBlockTip();
}		}
};		};

Processor::Processor(interfaces::Chain &chain, CConnman *connmanIn,		Processor::Processor(interfaces::Chain &chain, CConnman *connmanIn,
NodePeerManager *nodePeerManagerIn)		NodePeerManager *nodePeerManagerIn)
: connman(connmanIn), nodePeerManager(nodePeerManagerIn),		: connman(connmanIn), nodePeerManager(nodePeerManagerIn),
queryTimeoutDuration(AVALANCHE_DEFAULT_QUERY_TIMEOUT), round(0),		queryTimeoutDuration(AVALANCHE_DEFAULT_QUERY_TIMEOUT), round(0),
peerManager(std::make_unique<PeerManager>()) {		peerManager(std::make_unique<PeerManager>()) {
if (gArgs.IsArgSet("-avasessionkey")) {		if (gArgs.IsArgSet("-avasessionkey")) {
sessionKey = DecodeSecret(gArgs.GetArg("-avasessionkey", ""));		sessionKey = DecodeSecret(gArgs.GetArg("-avasessionkey", ""));
} else {		} else {
// Pick a random key for the session.		// Pick a random key for the session.
sessionKey.MakeNewKey(true);		sessionKey.MakeNewKey(true);
}		}

if (gArgs.IsArgSet("-avaproof")) {		if (gArgs.IsArgSet("-avaproof")) {
peerData = std::make_unique<PeerData>();

{		{
// The proof.		// The proof.
CDataStream stream(ParseHex(gArgs.GetArg("-avaproof", "")),		CDataStream stream(ParseHex(gArgs.GetArg("-avaproof", "")),
SER_NETWORK, 0);		SER_NETWORK, 0);
stream >> peerData->proof;		Proof proof;
		stream >> proof;
		peerData.proof = proof;

// Schedule proof registration at the first new block after IBD.		// Schedule proof registration at the first new block after IBD.
mustRegisterProof = true;		mustRegisterProof = true;
}		}

// Generate the delegation to the session key.		// Generate the delegation to the session key.
DelegationBuilder dgb(peerData->proof);		DelegationBuilder dgb(peerData.proof.value());
if (sessionKey.GetPubKey() != peerData->proof.getMaster()) {		if (sessionKey.GetPubKey() != peerData.proof->getMaster()) {
dgb.addLevel(DecodeSecret(gArgs.GetArg("-avamasterkey", "")),		dgb.addLevel(DecodeSecret(gArgs.GetArg("-avamasterkey", "")),
sessionKey.GetPubKey());		sessionKey.GetPubKey());
}		}
peerData->delegation = dgb.build();		peerData.delegation = dgb.build();
}		}

// Make sure we get notified of chain state changes.		// Make sure we get notified of chain state changes.
chainNotificationsHandler =		chainNotificationsHandler =
chain.handleNotifications(std::make_shared<NotificationsHandler>(this));		chain.handleNotifications(std::make_shared<NotificationsHandler>(this));
}		}

Processor::~Processor() {		Processor::~Processor() {
▲ Show 20 Lines • Show All 193 Lines • ▼ Show 20 Lines
}		}

CPubKey Processor::getSessionPubKey() const {		CPubKey Processor::getSessionPubKey() const {
return sessionKey.GetPubKey();		return sessionKey.GetPubKey();
}		}

uint256 Processor::buildLocalSighash(CNode *pfrom) const {		uint256 Processor::buildLocalSighash(CNode *pfrom) const {
CHashWriter hasher(SER_GETHASH, 0);		CHashWriter hasher(SER_GETHASH, 0);
hasher << peerData->delegation.getId();		hasher << peerData.delegation->getId();
hasher << pfrom->GetLocalNonce();		hasher << pfrom->GetLocalNonce();
hasher << pfrom->nRemoteHostNonce;		hasher << pfrom->nRemoteHostNonce;
hasher << pfrom->GetLocalExtraEntropy();		hasher << pfrom->GetLocalExtraEntropy();
hasher << pfrom->nRemoteExtraEntropy;		hasher << pfrom->nRemoteExtraEntropy;
return hasher.GetHash();		return hasher.GetHash();
}		}

uint256 Processor::buildRemoteSighash(CNode *pfrom) const {		uint256 Processor::buildRemoteSighash(CNode *pfrom) const {
CHashWriter hasher(SER_GETHASH, 0);		CHashWriter hasher(SER_GETHASH, 0);
hasher << pfrom->m_avalanche_state->delegation.getId();		hasher << pfrom->m_avalanche_state->delegation.getId();
hasher << pfrom->nRemoteHostNonce;		hasher << pfrom->nRemoteHostNonce;
hasher << pfrom->GetLocalNonce();		hasher << pfrom->GetLocalNonce();
hasher << pfrom->nRemoteExtraEntropy;		hasher << pfrom->nRemoteExtraEntropy;
hasher << pfrom->GetLocalExtraEntropy();		hasher << pfrom->GetLocalExtraEntropy();
return hasher.GetHash();		return hasher.GetHash();
}		}

bool Processor::sendHello(CNode *pfrom) const {		bool Processor::sendHello(CNode *pfrom) const {
if (!peerData) {		if (!peerData.delegation) {
// We do not have a delegation to advertise.		// We do not have a delegation to advertise.
return false;		return false;
}		}

// Now let's sign!		// Now let's sign!
SchnorrSig sig;		SchnorrSig sig;

{		{
const uint256 hash = buildLocalSighash(pfrom);		const uint256 hash = buildLocalSighash(pfrom);

if (!sessionKey.SignSchnorr(hash, sig)) {		if (!sessionKey.SignSchnorr(hash, sig)) {
return false;		return false;
}		}
}		}

connman->PushMessage(pfrom, CNetMsgMaker(pfrom->GetCommonVersion())		connman->PushMessage(pfrom,
		CNetMsgMaker(pfrom->GetCommonVersion())
.Make(NetMsgType::AVAHELLO,		.Make(NetMsgType::AVAHELLO,
Hello(peerData->delegation, sig)));		Hello(peerData.delegation.value(), sig)));

return true;		return true;
}		}

const Proof Processor::getProof() const {		const std::optional<Proof> &Processor::getLocalProof() const {
if (!peerData) {		return peerData.proof;
throw std::runtime_error("proof not set");
}
return peerData->proof;
}		}

bool Processor::startEventLoop(CScheduler &scheduler) {		bool Processor::startEventLoop(CScheduler &scheduler) {
return eventLoop.startEventLoop(		return eventLoop.startEventLoop(
scheduler, [this]() { this->runEventLoop(); }, AVALANCHE_TIME_STEP);		scheduler, [this]() { this->runEventLoop(); }, AVALANCHE_TIME_STEP);
}		}

bool Processor::stopEventLoop() {		bool Processor::stopEventLoop() {
▲ Show 20 Lines • Show All 170 Lines • Show Last 20 Lines