Fixed issues with propagation of governance objects (#1489)

* process governance objects in CheckMasternodeOrphanObjects as usual

* code refactoring: SetRateChecksHelper class added

* fixed race condition issues with propagation of governance objects

* change GetCollateralConfirmations signature

* code refactoring

* reduced minimum number of collateral confirmations required for relaying proposals

* bug fixes and improvements
This commit is contained in:
Ilya Savinov 2017-07-05 03:31:50 +03:00 committed by UdjinM6
parent d787fe4ab6
commit 109c5fd1d8
5 changed files with 217 additions and 83 deletions

View File

@ -435,13 +435,15 @@ void CGovernanceObject::UpdateLocalValidity()
bool CGovernanceObject::IsValidLocally(std::string& strError, bool fCheckCollateral) bool CGovernanceObject::IsValidLocally(std::string& strError, bool fCheckCollateral)
{ {
bool fMissingMasternode = false; bool fMissingMasternode = false;
bool fMissingConfirmations = false;
return IsValidLocally(strError, fMissingMasternode, fCheckCollateral); return IsValidLocally(strError, fMissingMasternode, fMissingConfirmations, fCheckCollateral);
} }
bool CGovernanceObject::IsValidLocally(std::string& strError, bool& fMissingMasternode, bool fCheckCollateral) bool CGovernanceObject::IsValidLocally(std::string& strError, bool& fMissingMasternode, bool& fMissingConfirmations, bool fCheckCollateral)
{ {
fMissingMasternode = false; fMissingMasternode = false;
fMissingConfirmations = false;
if(fUnparsable) { if(fUnparsable) {
strError = "Object data unparseable"; strError = "Object data unparseable";
@ -481,11 +483,8 @@ bool CGovernanceObject::IsValidLocally(std::string& strError, bool& fMissingMast
return true; return true;
} }
if(!IsCollateralValid(strError)) { if (!IsCollateralValid(strError, fMissingConfirmations))
// strError set in IsCollateralValid
if(strError == "") strError = "Collateral is invalid";
return false; return false;
}
} }
/* /*
@ -515,9 +514,10 @@ CAmount CGovernanceObject::GetMinCollateralFee()
} }
} }
bool CGovernanceObject::IsCollateralValid(std::string& strError) bool CGovernanceObject::IsCollateralValid(std::string& strError, bool& fMissingConfirmations)
{ {
strError = ""; strError = "";
fMissingConfirmations = false;
CAmount nMinFee = GetMinCollateralFee(); CAmount nMinFee = GetMinCollateralFee();
uint256 nExpectedHash = GetHash(); uint256 nExpectedHash = GetHash();
@ -543,7 +543,7 @@ bool CGovernanceObject::IsCollateralValid(std::string& strError)
CScript findScript; CScript findScript;
findScript << OP_RETURN << ToByteVector(nExpectedHash); findScript << OP_RETURN << ToByteVector(nExpectedHash);
DBG( cout << "IsCollateralValid txCollateral.vout.size() = " << txCollateral.vout.size() << endl; ); DBG( cout << "IsCollateralValid: txCollateral.vout.size() = " << txCollateral.vout.size() << endl; );
DBG( cout << "IsCollateralValid: findScript = " << ScriptToAsmStr( findScript, false ) << endl; ); DBG( cout << "IsCollateralValid: findScript = " << ScriptToAsmStr( findScript, false ) << endl; );
@ -591,14 +591,20 @@ bool CGovernanceObject::IsCollateralValid(std::string& strError)
} }
} }
if(nConfirmationsIn >= GOVERNANCE_FEE_CONFIRMATIONS) { if(nConfirmationsIn < GOVERNANCE_FEE_CONFIRMATIONS) {
strError = "valid"; strError = strprintf("Collateral requires at least %d confirmations to be relayed throughout the network (it has only %d)", GOVERNANCE_FEE_CONFIRMATIONS, nConfirmationsIn);
} else { if (nConfirmationsIn >= GOVERNANCE_MIN_RELAY_FEE_CONFIRMATIONS) {
strError = strprintf("Collateral requires at least %d confirmations - %d confirmations", GOVERNANCE_FEE_CONFIRMATIONS, nConfirmationsIn); fMissingConfirmations = true;
LogPrintf ("CGovernanceObject::IsCollateralValid -- %s - %d confirmations\n", strError, nConfirmationsIn); strError += ", pre-accepted -- waiting for required confirmations";
} else {
strError += ", rejected -- try again later";
}
LogPrintf ("CGovernanceObject::IsCollateralValid -- %s\n", strError);
return false; return false;
} }
strError = "valid";
return true; return true;
} }

View File

@ -37,6 +37,7 @@ static const int GOVERNANCE_OBJECT_WATCHDOG = 3;
static const CAmount GOVERNANCE_PROPOSAL_FEE_TX = (5.0*COIN); static const CAmount GOVERNANCE_PROPOSAL_FEE_TX = (5.0*COIN);
static const int64_t GOVERNANCE_FEE_CONFIRMATIONS = 6; static const int64_t GOVERNANCE_FEE_CONFIRMATIONS = 6;
static const int64_t GOVERNANCE_MIN_RELAY_FEE_CONFIRMATIONS = 1;
static const int64_t GOVERNANCE_UPDATE_MIN = 60*60; static const int64_t GOVERNANCE_UPDATE_MIN = 60*60;
static const int64_t GOVERNANCE_DELETION_DELAY = 10*60; static const int64_t GOVERNANCE_DELETION_DELAY = 10*60;
static const int64_t GOVERNANCE_ORPHAN_EXPIRATION_TIME = 10*60; static const int64_t GOVERNANCE_ORPHAN_EXPIRATION_TIME = 10*60;
@ -263,10 +264,10 @@ public:
bool IsValidLocally(std::string& strError, bool fCheckCollateral); bool IsValidLocally(std::string& strError, bool fCheckCollateral);
bool IsValidLocally(std::string& strError, bool& fMissingMasternode, bool fCheckCollateral); bool IsValidLocally(std::string& strError, bool& fMissingMasternode, bool& fMissingConfirmations, bool fCheckCollateral);
/// Check the collateral transaction for the budget proposal/finalized budget /// Check the collateral transaction for the budget proposal/finalized budget
bool IsCollateralValid(std::string& strError); bool IsCollateralValid(std::string& strError, bool &fMissingConfirmations);
void UpdateLocalValidity(); void UpdateLocalValidity();

View File

@ -19,6 +19,8 @@ CGovernanceManager governance;
int nSubmittedFinalBudget; int nSubmittedFinalBudget;
const std::string CGovernanceManager::SERIALIZATION_VERSION_STRING = "CGovernanceManager-Version-11"; const std::string CGovernanceManager::SERIALIZATION_VERSION_STRING = "CGovernanceManager-Version-11";
const int CGovernanceManager::MAX_TIME_FUTURE_DEVIATION = 60*60;
const int CGovernanceManager::RELIABLE_PROPAGATION_TIME = 60;
CGovernanceManager::CGovernanceManager() CGovernanceManager::CGovernanceManager()
: pCurrentBlockIndex(NULL), : pCurrentBlockIndex(NULL),
@ -42,7 +44,7 @@ CGovernanceManager::CGovernanceManager()
// Accessors for thread-safe access to maps // Accessors for thread-safe access to maps
bool CGovernanceManager::HaveObjectForHash(uint256 nHash) { bool CGovernanceManager::HaveObjectForHash(uint256 nHash) {
LOCK(cs); LOCK(cs);
return (mapObjects.count(nHash) == 1); return (mapObjects.count(nHash) == 1 || mapPostponedObjects.count(nHash) == 1);
} }
bool CGovernanceManager::SerializeObjectForHash(uint256 nHash, CDataStream& ss) bool CGovernanceManager::SerializeObjectForHash(uint256 nHash, CDataStream& ss)
@ -50,7 +52,9 @@ bool CGovernanceManager::SerializeObjectForHash(uint256 nHash, CDataStream& ss)
LOCK(cs); LOCK(cs);
object_m_it it = mapObjects.find(nHash); object_m_it it = mapObjects.find(nHash);
if (it == mapObjects.end()) { if (it == mapObjects.end()) {
return false; it = mapPostponedObjects.find(nHash);
if (it == mapPostponedObjects.end())
return false;
} }
ss << it->second; ss << it->second;
return true; return true;
@ -195,51 +199,32 @@ void CGovernanceManager::ProcessMessage(CNode* pfrom, std::string& strCommand, C
// CHECK OBJECT AGAINST LOCAL BLOCKCHAIN // CHECK OBJECT AGAINST LOCAL BLOCKCHAIN
bool fMasternodeMissing = false; bool fMasternodeMissing = false;
bool fIsValid = govobj.IsValidLocally(strError, fMasternodeMissing, true); bool fMissingConfirmations = false;
bool fIsValid = govobj.IsValidLocally(strError, fMasternodeMissing, fMissingConfirmations, true);
if(fMasternodeMissing) { if(fRateCheckBypassed && (fIsValid || fMasternodeMissing)) {
mapMasternodeOrphanObjects.insert(std::make_pair(nHash, object_time_pair_t(govobj, GetAdjustedTime() + GOVERNANCE_ORPHAN_EXPIRATION_TIME))); if(!MasternodeRateCheck(govobj, UPDATE_FAIL_ONLY)) {
LogPrintf("MNGOVERNANCEOBJECT -- Missing masternode for: %s, strError = %s\n", strHash, strError);
// fIsValid must also be false here so we will return early in the next if block
}
if(!fIsValid) {
mapSeenGovernanceObjects.insert(std::make_pair(nHash, SEEN_OBJECT_ERROR_INVALID));
LogPrintf("MNGOVERNANCEOBJECT -- Governance object is invalid - %s\n", strError);
return;
}
if(fRateCheckBypassed) {
if(!MasternodeRateCheck(govobj, UPDATE_FAIL_ONLY, true, fRateCheckBypassed)) {
LogPrintf("MNGOVERNANCEOBJECT -- masternode rate check failed (after signature verification) - %s - (current block height %d) \n", strHash, nCachedBlockHeight); LogPrintf("MNGOVERNANCEOBJECT -- masternode rate check failed (after signature verification) - %s - (current block height %d) \n", strHash, nCachedBlockHeight);
return; return;
} }
} }
// UPDATE CACHED VARIABLES FOR THIS OBJECT AND ADD IT TO OUR MANANGED DATA if(!fIsValid) {
if(fMasternodeMissing) {
mapMasternodeOrphanObjects.insert(std::make_pair(nHash, object_time_pair_t(govobj, GetAdjustedTime() + GOVERNANCE_ORPHAN_EXPIRATION_TIME)));
LogPrintf("MNGOVERNANCEOBJECT -- Missing masternode for: %s, strError = %s\n", strHash, strError);
} else if(fMissingConfirmations) {
AddPostponedObject(govobj);
LogPrintf("MNGOVERNANCEOBJECT -- Not enough fee confirmations for: %s, strError = %s\n", strHash, strError);
} else {
LogPrintf("MNGOVERNANCEOBJECT -- Governance object is invalid - %s\n", strError);
}
govobj.UpdateSentinelVariables(); //this sets local vars in object mapSeenGovernanceObjects.insert(std::make_pair(nHash, SEEN_OBJECT_ERROR_INVALID));
return;
bool fAddToSeen = true;
if(AddGovernanceObject(govobj, fAddToSeen, pfrom))
{
LogPrintf("MNGOVERNANCEOBJECT -- %s new\n", strHash);
govobj.Relay();
} }
if(fAddToSeen) { AddGovernanceObject(govobj, pfrom);
// UPDATE THAT WE'VE SEEN THIS OBJECT
mapSeenGovernanceObjects.insert(std::make_pair(nHash, SEEN_OBJECT_IS_VALID));
// Update the rate buffer
MasternodeRateCheck(govobj, UPDATE_TRUE, true, fRateCheckBypassed);
}
masternodeSync.AddedGovernanceItem();
// WE MIGHT HAVE PENDING/ORPHAN VOTES FOR THIS OBJECT
CGovernanceException exception;
CheckOrphanVotes(govobj, exception);
} }
// A NEW GOVERNANCE OBJECT VOTE HAS ARRIVED // A NEW GOVERNANCE OBJECT VOTE HAS ARRIVED
@ -290,7 +275,8 @@ void CGovernanceManager::CheckOrphanVotes(CGovernanceObject& govobj, CGovernance
std::vector<vote_time_pair_t> vecVotePairs; std::vector<vote_time_pair_t> vecVotePairs;
mapOrphanVotes.GetAll(nHash, vecVotePairs); mapOrphanVotes.GetAll(nHash, vecVotePairs);
fRateChecksEnabled = false; CRateChecksGuard guard(false, *this);
int64_t nNow = GetAdjustedTime(); int64_t nNow = GetAdjustedTime();
for(size_t i = 0; i < vecVotePairs.size(); ++i) { for(size_t i = 0; i < vecVotePairs.size(); ++i) {
bool fRemove = false; bool fRemove = false;
@ -308,7 +294,39 @@ void CGovernanceManager::CheckOrphanVotes(CGovernanceObject& govobj, CGovernance
mapOrphanVotes.Erase(nHash, pairVote); mapOrphanVotes.Erase(nHash, pairVote);
} }
} }
fRateChecksEnabled = true; }
void CGovernanceManager::AddGovernanceObject(CGovernanceObject& govobj, CNode* pfrom)
{
uint256 nHash = govobj.GetHash();
std::string strHash = nHash.ToString();
// UPDATE CACHED VARIABLES FOR THIS OBJECT AND ADD IT TO OUR MANANGED DATA
govobj.UpdateSentinelVariables(); //this sets local vars in object
bool fAddToSeen = true;
if(AddGovernanceObject(govobj, fAddToSeen, pfrom))
{
LogPrintf("AddGovernanceObject -- %s new, received form %s\n", strHash, pfrom? pfrom->addrName : "NULL");
govobj.Relay();
}
// PROCESS OBJECT EXACTLY THE SAME WAY AS USUAL
if(fAddToSeen) {
// UPDATE THAT WE'VE SEEN THIS OBJECT
mapSeenGovernanceObjects.insert(std::make_pair(nHash, SEEN_OBJECT_IS_VALID));
// Update the rate buffer
MasternodeRateCheck(govobj, UPDATE_TRUE);
}
masternodeSync.AddedGovernanceItem();
// WE MIGHT HAVE PENDING/ORPHAN VOTES FOR THIS OBJECT
CGovernanceException exception;
CheckOrphanVotes(govobj, exception);
} }
bool CGovernanceManager::AddGovernanceObject(CGovernanceObject& govobj, bool& fAddToSeen, CNode* pfrom) bool CGovernanceManager::AddGovernanceObject(CGovernanceObject& govobj, bool& fAddToSeen, CNode* pfrom)
@ -471,7 +489,7 @@ void CGovernanceManager::UpdateCachesAndClean()
if(!pCurrentBlockIndex) return; if(!pCurrentBlockIndex) return;
fRateChecksEnabled = false; CRateChecksGuard guard(false, *this);
LogPrint("gobject", "CGovernanceManager::UpdateCachesAndClean -- After pCurrentBlockIndex (not NULL)\n"); LogPrint("gobject", "CGovernanceManager::UpdateCachesAndClean -- After pCurrentBlockIndex (not NULL)\n");
@ -541,7 +559,6 @@ void CGovernanceManager::UpdateCachesAndClean()
} }
} }
fRateChecksEnabled = true;
LogPrintf("CGovernanceManager::UpdateCachesAndClean -- %s\n", ToString()); LogPrintf("CGovernanceManager::UpdateCachesAndClean -- %s\n", ToString());
} }
@ -689,8 +706,7 @@ bool CGovernanceManager::ConfirmInventoryRequest(const CInv& inv)
switch(inv.type) { switch(inv.type) {
case MSG_GOVERNANCE_OBJECT: case MSG_GOVERNANCE_OBJECT:
{ {
object_m_it it = mapObjects.find(inv.hash); if(mapObjects.count(inv.hash) == 1 || mapPostponedObjects.count(inv.hash) == 1) {
if(it != mapObjects.end()) {
LogPrint("gobject", "CGovernanceManager::ConfirmInventoryRequest already have governance object, returning false\n"); LogPrint("gobject", "CGovernanceManager::ConfirmInventoryRequest already have governance object, returning false\n");
return false; return false;
} }
@ -819,7 +835,7 @@ void CGovernanceManager::Sync(CNode* pfrom, const uint256& nProp, const CBloomFi
bool CGovernanceManager::MasternodeRateCheck(const CGovernanceObject& govobj, update_mode_enum_t eUpdateLast) bool CGovernanceManager::MasternodeRateCheck(const CGovernanceObject& govobj, update_mode_enum_t eUpdateLast)
{ {
bool fRateCheckBypassed = false; bool fRateCheckBypassed;
return MasternodeRateCheck(govobj, eUpdateLast, true, fRateCheckBypassed); return MasternodeRateCheck(govobj, eUpdateLast, true, fRateCheckBypassed);
} }
@ -880,10 +896,14 @@ bool CGovernanceManager::MasternodeRateCheck(const CGovernanceObject& govobj, up
return false; return false;
} }
if(nTimestamp > nNow + 60*60) { bool fAdditionalRelay = false;
if(nTimestamp > nNow + MAX_TIME_FUTURE_DEVIATION) {
LogPrintf("CGovernanceManager::MasternodeRateCheck -- object %s rejected due to too new (future) timestamp, masternode vin = %s, timestamp = %d, current time = %d\n", LogPrintf("CGovernanceManager::MasternodeRateCheck -- object %s rejected due to too new (future) timestamp, masternode vin = %s, timestamp = %d, current time = %d\n",
strHash, vin.prevout.ToStringShort(), nTimestamp, nNow); strHash, vin.prevout.ToStringShort(), nTimestamp, nNow);
return false; return false;
} else if (nTimestamp > nNow + MAX_TIME_FUTURE_DEVIATION - RELIABLE_PROPAGATION_TIME) {
// schedule additional relay for the object
fAdditionalRelay = true;
} }
double dMaxRate = 1.1 / nSuperblockCycleSeconds; double dMaxRate = 1.1 / nSuperblockCycleSeconds;
@ -916,6 +936,9 @@ bool CGovernanceManager::MasternodeRateCheck(const CGovernanceObject& govobj, up
bool fRateOK = ( dRate < dMaxRate ); bool fRateOK = ( dRate < dMaxRate );
if (eUpdateLast == UPDATE_TRUE && fAdditionalRelay)
setAdditionalRelayObjects.insert(govobj.GetHash());
switch(eUpdateLast) { switch(eUpdateLast) {
case UPDATE_TRUE: case UPDATE_TRUE:
pBuffer->AddTimestamp(nTimestamp); pBuffer->AddTimestamp(nTimestamp);
@ -994,18 +1017,19 @@ bool CGovernanceManager::ProcessVote(CNode* pfrom, const CGovernanceVote& vote,
void CGovernanceManager::CheckMasternodeOrphanVotes() void CGovernanceManager::CheckMasternodeOrphanVotes()
{ {
LOCK2(cs_main, cs); LOCK2(cs_main, cs);
fRateChecksEnabled = false;
CRateChecksGuard guard(false, *this);
for(object_m_it it = mapObjects.begin(); it != mapObjects.end(); ++it) { for(object_m_it it = mapObjects.begin(); it != mapObjects.end(); ++it) {
it->second.CheckOrphanVotes(); it->second.CheckOrphanVotes();
} }
fRateChecksEnabled = true;
} }
void CGovernanceManager::CheckMasternodeOrphanObjects() void CGovernanceManager::CheckMasternodeOrphanObjects()
{ {
LOCK2(cs_main, cs); LOCK2(cs_main, cs);
int64_t nNow = GetAdjustedTime(); int64_t nNow = GetAdjustedTime();
fRateChecksEnabled = false; CRateChecksGuard guard(false, *this);
object_time_m_it it = mapMasternodeOrphanObjects.begin(); object_time_m_it it = mapMasternodeOrphanObjects.begin();
while(it != mapMasternodeOrphanObjects.end()) { while(it != mapMasternodeOrphanObjects.end()) {
object_time_pair_t& pair = it->second; object_time_pair_t& pair = it->second;
@ -1018,7 +1042,8 @@ void CGovernanceManager::CheckMasternodeOrphanObjects()
string strError; string strError;
bool fMasternodeMissing = false; bool fMasternodeMissing = false;
bool fIsValid = govobj.IsValidLocally(strError, fMasternodeMissing, true); bool fConfirmationsMissing = false;
bool fIsValid = govobj.IsValidLocally(strError, fMasternodeMissing, fConfirmationsMissing, true);
if(!fIsValid) { if(!fIsValid) {
if(!fMasternodeMissing) { if(!fMasternodeMissing) {
mapMasternodeOrphanObjects.erase(it++); mapMasternodeOrphanObjects.erase(it++);
@ -1029,17 +1054,76 @@ void CGovernanceManager::CheckMasternodeOrphanObjects()
continue; continue;
} }
bool fAddToSeen = true; AddGovernanceObject(govobj);
if(AddGovernanceObject(govobj, fAddToSeen)) { mapMasternodeOrphanObjects.erase(it++);
LogPrintf("CGovernanceManager::CheckMasternodeOrphanObjects -- %s new\n", govobj.GetHash().ToString()); }
govobj.Relay(); }
mapMasternodeOrphanObjects.erase(it++);
} void CGovernanceManager::CheckPostponedObjects()
else { {
++it; LOCK2(cs_main, cs);
}
// Check postponed proposals
for(object_m_it it = mapPostponedObjects.begin(); it != mapPostponedObjects.end();) {
const uint256& nHash = it->first;
CGovernanceObject& govobj = it->second;
assert(govobj.GetObjectType() != GOVERNANCE_OBJECT_WATCHDOG &&
govobj.GetObjectType() != GOVERNANCE_OBJECT_TRIGGER);
std::string strError;
bool fMissingConfirmations;
if (govobj.IsCollateralValid(strError, fMissingConfirmations))
{
if(govobj.IsValidLocally(strError, false))
AddGovernanceObject(govobj);
else
LogPrintf("CGovernanceManager::CheckPostponedObjects -- %s invalid\n", nHash.ToString());
} else if(fMissingConfirmations) {
// wait for more confirmations
++it;
continue;
}
// remove processed or invalid object from the queue
mapPostponedObjects.erase(it++);
}
// Perform additional relays for triggers/watchdogs
int64_t nNow = GetTime();
int64_t nSuperblockCycleSeconds = Params().GetConsensus().nSuperblockCycle * Params().GetConsensus().nPowTargetSpacing;
for(hash_s_it it = setAdditionalRelayObjects.begin(); it != setAdditionalRelayObjects.end();) {
object_m_it itObject = mapObjects.find(*it);
if(itObject != mapObjects.end()) {
CGovernanceObject& govobj = itObject->second;
int64_t nTimestamp = govobj.GetCreationTime();
bool fValid = (nTimestamp <= nNow + MAX_TIME_FUTURE_DEVIATION) && (nTimestamp >= nNow - 2 * nSuperblockCycleSeconds);
bool fReady = (nTimestamp <= nNow + MAX_TIME_FUTURE_DEVIATION - RELIABLE_PROPAGATION_TIME);
if(fValid) {
if(fReady) {
LogPrintf("CGovernanceManager::CheckPostponedObjects -- additional relay: hash = %s\n", govobj.GetHash().ToString());
govobj.Relay();
} else {
it++;
continue;
}
}
} else {
LogPrintf("CGovernanceManager::CheckPostponedObjects -- additional relay of unknown object: %s\n", it->ToString());
}
setAdditionalRelayObjects.erase(it++);
} }
fRateChecksEnabled = true;
} }
void CGovernanceManager::RequestGovernanceObject(CNode* pfrom, const uint256& nHash, bool fUseFilter) void CGovernanceManager::RequestGovernanceObject(CNode* pfrom, const uint256& nHash, bool fUseFilter)
@ -1324,6 +1408,8 @@ void CGovernanceManager::UpdatedBlockTip(const CBlockIndex *pindex)
nCachedBlockHeight = pCurrentBlockIndex->nHeight; nCachedBlockHeight = pCurrentBlockIndex->nHeight;
LogPrint("gobject", "CGovernanceManager::UpdatedBlockTip pCurrentBlockIndex->nHeight: %d\n", pCurrentBlockIndex->nHeight); LogPrint("gobject", "CGovernanceManager::UpdatedBlockTip pCurrentBlockIndex->nHeight: %d\n", pCurrentBlockIndex->nHeight);
} }
CheckPostponedObjects();
} }
void CGovernanceManager::RequestOrphanObjects() void CGovernanceManager::RequestOrphanObjects()

View File

@ -226,6 +226,9 @@ private:
static const std::string SERIALIZATION_VERSION_STRING; static const std::string SERIALIZATION_VERSION_STRING;
static const int MAX_TIME_FUTURE_DEVIATION;
static const int RELIABLE_PROPAGATION_TIME;
// Keep track of current block index // Keep track of current block index
const CBlockIndex *pCurrentBlockIndex; const CBlockIndex *pCurrentBlockIndex;
@ -239,6 +242,9 @@ private:
object_time_m_t mapMasternodeOrphanObjects; object_time_m_t mapMasternodeOrphanObjects;
object_m_t mapPostponedObjects;
hash_s_t setAdditionalRelayObjects;
hash_time_m_t mapWatchdogObjects; hash_time_m_t mapWatchdogObjects;
uint256 nHashWatchdogCurrent; uint256 nHashWatchdogCurrent;
@ -259,6 +265,26 @@ private:
bool fRateChecksEnabled; bool fRateChecksEnabled;
class CRateChecksGuard
{
CGovernanceManager& govman;
bool fRateChecksPrev;
public:
CRateChecksGuard(bool value, CGovernanceManager& gm) : govman(gm)
{
ENTER_CRITICAL_SECTION(govman.cs)
fRateChecksPrev = govman.fRateChecksEnabled;
govman.fRateChecksEnabled = value;
}
~CRateChecksGuard()
{
govman.fRateChecksEnabled = fRateChecksPrev;
LEAVE_CRITICAL_SECTION(govman.cs)
}
};
public: public:
// critical section to protect the inner data structures // critical section to protect the inner data structures
mutable CCriticalSection cs; mutable CCriticalSection cs;
@ -294,6 +320,7 @@ public:
std::vector<CGovernanceObject*> GetAllNewerThan(int64_t nMoreThanTime); std::vector<CGovernanceObject*> GetAllNewerThan(int64_t nMoreThanTime);
bool IsBudgetPaymentBlock(int nBlockHeight); bool IsBudgetPaymentBlock(int nBlockHeight);
void AddGovernanceObject(CGovernanceObject& govobj, CNode* pfrom = NULL);
bool AddGovernanceObject(CGovernanceObject& govobj, bool& fAddToSeen, CNode* pfrom = NULL); bool AddGovernanceObject(CGovernanceObject& govobj, bool& fAddToSeen, CNode* pfrom = NULL);
std::string GetRequiredPaymentsString(int nBlockHeight); std::string GetRequiredPaymentsString(int nBlockHeight);
@ -364,6 +391,12 @@ public:
bool SerializeVoteForHash(uint256 nHash, CDataStream& ss); bool SerializeVoteForHash(uint256 nHash, CDataStream& ss);
void AddPostponedObject(const CGovernanceObject& govobj)
{
LOCK(cs);
mapPostponedObjects.insert(std::make_pair(govobj.GetHash(), govobj));
}
void AddSeenGovernanceObject(uint256 nHash, int status); void AddSeenGovernanceObject(uint256 nHash, int status);
void AddSeenVote(uint256 nHash, int status); void AddSeenVote(uint256 nHash, int status);
@ -384,6 +417,8 @@ public:
void CheckMasternodeOrphanObjects(); void CheckMasternodeOrphanObjects();
void CheckPostponedObjects();
bool AreRateChecksEnabled() const { bool AreRateChecksEnabled() const {
LOCK(cs); LOCK(cs);
return fRateChecksEnabled; return fRateChecksEnabled;

View File

@ -254,7 +254,9 @@ UniValue gobject(const UniValue& params, bool fHelp)
std::string strHash = govobj.GetHash().ToString(); std::string strHash = govobj.GetHash().ToString();
std::string strError = ""; std::string strError = "";
if(!govobj.IsValidLocally(strError, true)) { bool fMissingMasternode;
bool fMissingConfirmations;
if(!govobj.IsValidLocally(strError, fMissingMasternode, fMissingConfirmations, true) && !fMissingConfirmations) {
LogPrintf("gobject(submit) -- Object submission rejected because object is not valid - hash = %s, strError = %s\n", strHash, strError); LogPrintf("gobject(submit) -- Object submission rejected because object is not valid - hash = %s, strError = %s\n", strHash, strError);
throw JSONRPCError(RPC_INTERNAL_ERROR, "Governance object is not valid - " + strHash + " - " + strError); throw JSONRPCError(RPC_INTERNAL_ERROR, "Governance object is not valid - " + strHash + " - " + strError);
} }
@ -270,11 +272,15 @@ UniValue gobject(const UniValue& params, bool fHelp)
LogPrintf("gobject(submit) -- Object submission rejected because of rate check failure (buffer updated) - hash = %s\n", strHash); LogPrintf("gobject(submit) -- Object submission rejected because of rate check failure (buffer updated) - hash = %s\n", strHash);
throw JSONRPCError(RPC_INVALID_PARAMETER, "Object creation rate limit exceeded"); throw JSONRPCError(RPC_INVALID_PARAMETER, "Object creation rate limit exceeded");
} }
governance.AddSeenGovernanceObject(govobj.GetHash(), SEEN_OBJECT_IS_VALID);
govobj.Relay();
LogPrintf("gobject(submit) -- Adding locally created governance object - %s\n", strHash); LogPrintf("gobject(submit) -- Adding locally created governance object - %s\n", strHash);
bool fAddToSeen = true;
governance.AddGovernanceObject(govobj, fAddToSeen); if(fMissingConfirmations) {
governance.AddPostponedObject(govobj);
govobj.Relay();
} else {
governance.AddGovernanceObject(govobj);
}
return govobj.GetHash().ToString(); return govobj.GetHash().ToString();
} }