mirror of
https://github.com/dashpay/dash.git
synced 2024-12-26 12:32:48 +01:00
2b75faefee
- Implemented spork for only paying new nodes after a period of time on mainnet - protocol bump - fixed a few issues with sporks. Spork show now shows all sporks, instead of the changed ones. IsSporkActive now supports sporks set to 0 as on.
239 lines
7.7 KiB
C++
239 lines
7.7 KiB
C++
|
|
|
|
|
|
#include "sync.h"
|
|
#include "net.h"
|
|
#include "key.h"
|
|
#include "util.h"
|
|
#include "base58.h"
|
|
#include "protocol.h"
|
|
#include "spork.h"
|
|
#include "main.h"
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
using namespace std;
|
|
using namespace boost;
|
|
|
|
class CSporkMessage;
|
|
class CSporkManager;
|
|
|
|
CSporkManager sporkManager;
|
|
|
|
std::map<uint256, CSporkMessage> mapSporks;
|
|
std::map<int, CSporkMessage> mapSporksActive;
|
|
|
|
|
|
void ProcessSpork(CNode* pfrom, std::string& strCommand, CDataStream& vRecv)
|
|
{
|
|
if(fLiteMode) return; //disable all darksend/masternode related functionality
|
|
|
|
if (strCommand == "spork")
|
|
{
|
|
//LogPrintf("ProcessSpork::spork\n");
|
|
CDataStream vMsg(vRecv);
|
|
CSporkMessage spork;
|
|
vRecv >> spork;
|
|
|
|
if(chainActive.Tip() == NULL) return;
|
|
|
|
uint256 hash = spork.GetHash();
|
|
if(mapSporksActive.count(spork.nSporkID)) {
|
|
if(mapSporksActive[spork.nSporkID].nTimeSigned >= spork.nTimeSigned){
|
|
if(fDebug) LogPrintf("spork - seen %s block %d \n", hash.ToString().c_str(), chainActive.Tip()->nHeight);
|
|
return;
|
|
} else {
|
|
if(fDebug) LogPrintf("spork - got updated spork %s block %d \n", hash.ToString().c_str(), chainActive.Tip()->nHeight);
|
|
}
|
|
}
|
|
|
|
LogPrintf("spork - new %s ID %d Time %d bestHeight %d\n", hash.ToString().c_str(), spork.nSporkID, spork.nValue, chainActive.Tip()->nHeight);
|
|
|
|
if(!sporkManager.CheckSignature(spork)){
|
|
LogPrintf("spork - invalid signature\n");
|
|
Misbehaving(pfrom->GetId(), 100);
|
|
return;
|
|
}
|
|
|
|
mapSporks[hash] = spork;
|
|
mapSporksActive[spork.nSporkID] = spork;
|
|
sporkManager.Relay(spork);
|
|
|
|
//does a task if needed
|
|
ExecuteSpork(spork.nSporkID, spork.nValue);
|
|
}
|
|
if (strCommand == "getsporks")
|
|
{
|
|
std::map<int, CSporkMessage>::iterator it = mapSporksActive.begin();
|
|
|
|
while(it != mapSporksActive.end()) {
|
|
pfrom->PushMessage("spork", it->second);
|
|
it++;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// grab the spork, otherwise say it's off
|
|
bool IsSporkActive(int nSporkID)
|
|
{
|
|
int64_t r = -1;
|
|
|
|
if(mapSporksActive.count(nSporkID)){
|
|
r = mapSporksActive[nSporkID].nValue;
|
|
} else {
|
|
if(nSporkID == SPORK_2_INSTANTX) r = SPORK_2_INSTANTX_DEFAULT;
|
|
if(nSporkID == SPORK_3_INSTANTX_BLOCK_FILTERING) r = SPORK_3_INSTANTX_BLOCK_FILTERING_DEFAULT;
|
|
if(nSporkID == SPORK_5_MAX_VALUE) r = SPORK_5_MAX_VALUE_DEFAULT;
|
|
if(nSporkID == SPORK_7_MASTERNODE_SCANNING) r = SPORK_7_MASTERNODE_SCANNING_DEFAULT;
|
|
if(nSporkID == SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT) r = SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT_DEFAULT;
|
|
if(nSporkID == SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT) r = SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT_DEFAULT;
|
|
if(nSporkID == SPORK_10_MASTERNODE_PAY_NEWEST_NODES) r = SPORK_10_MASTERNODE_PAY_NEWEST_NODES_DEFAULT;
|
|
|
|
if(r == -1) LogPrintf("GetSpork::Unknown Spork %d\n", nSporkID);
|
|
}
|
|
if(r == -1) r = 4070908800; //return 2099-1-1 by default
|
|
|
|
return r < GetTime();
|
|
}
|
|
|
|
// grab the value of the spork on the network, or the default
|
|
int GetSporkValue(int nSporkID)
|
|
{
|
|
int r = 0;
|
|
|
|
if(mapSporksActive.count(nSporkID)){
|
|
r = mapSporksActive[nSporkID].nValue;
|
|
} else {
|
|
if(nSporkID == SPORK_2_INSTANTX) r = SPORK_2_INSTANTX_DEFAULT;
|
|
if(nSporkID == SPORK_3_INSTANTX_BLOCK_FILTERING) r = SPORK_3_INSTANTX_BLOCK_FILTERING_DEFAULT;
|
|
if(nSporkID == SPORK_5_MAX_VALUE) r = SPORK_5_MAX_VALUE_DEFAULT;
|
|
if(nSporkID == SPORK_7_MASTERNODE_SCANNING) r = SPORK_7_MASTERNODE_SCANNING_DEFAULT;
|
|
if(nSporkID == SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT) r = SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT_DEFAULT;
|
|
if(nSporkID == SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT) r = SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT_DEFAULT;
|
|
if(nSporkID == SPORK_10_MASTERNODE_PAY_NEWEST_NODES) r = SPORK_10_MASTERNODE_PAY_NEWEST_NODES_DEFAULT;
|
|
|
|
if(r == 0) LogPrintf("GetSpork::Unknown Spork %d\n", nSporkID);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
void ExecuteSpork(int nSporkID, int nValue)
|
|
{
|
|
}
|
|
|
|
|
|
bool CSporkManager::CheckSignature(CSporkMessage& spork)
|
|
{
|
|
//note: need to investigate why this is failing
|
|
std::string strMessage = boost::lexical_cast<std::string>(spork.nSporkID) + boost::lexical_cast<std::string>(spork.nValue) + boost::lexical_cast<std::string>(spork.nTimeSigned);
|
|
CPubKey pubkey(ParseHex(Params().SporkKey()));
|
|
|
|
std::string errorMessage = "";
|
|
if(!darkSendSigner.VerifyMessage(pubkey, spork.vchSig, strMessage, errorMessage)){
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CSporkManager::Sign(CSporkMessage& spork)
|
|
{
|
|
std::string strMessage = boost::lexical_cast<std::string>(spork.nSporkID) + boost::lexical_cast<std::string>(spork.nValue) + boost::lexical_cast<std::string>(spork.nTimeSigned);
|
|
|
|
CKey key2;
|
|
CPubKey pubkey2;
|
|
std::string errorMessage = "";
|
|
|
|
if(!darkSendSigner.SetKey(strMasterPrivKey, errorMessage, key2, pubkey2))
|
|
{
|
|
LogPrintf("CMasternodePayments::Sign - ERROR: Invalid masternodeprivkey: '%s'\n", errorMessage.c_str());
|
|
return false;
|
|
}
|
|
|
|
if(!darkSendSigner.SignMessage(strMessage, errorMessage, spork.vchSig, key2)) {
|
|
LogPrintf("CMasternodePayments::Sign - Sign message failed");
|
|
return false;
|
|
}
|
|
|
|
if(!darkSendSigner.VerifyMessage(pubkey2, spork.vchSig, strMessage, errorMessage)) {
|
|
LogPrintf("CMasternodePayments::Sign - Verify message failed");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool CSporkManager::UpdateSpork(int nSporkID, int64_t nValue)
|
|
{
|
|
|
|
CSporkMessage msg;
|
|
msg.nSporkID = nSporkID;
|
|
msg.nValue = nValue;
|
|
msg.nTimeSigned = GetTime();
|
|
|
|
if(Sign(msg)){
|
|
Relay(msg);
|
|
mapSporks[msg.GetHash()] = msg;
|
|
mapSporksActive[nSporkID] = msg;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CSporkManager::Relay(CSporkMessage& msg)
|
|
{
|
|
CInv inv(MSG_SPORK, msg.GetHash());
|
|
|
|
vector<CInv> vInv;
|
|
vInv.push_back(inv);
|
|
LOCK(cs_vNodes);
|
|
BOOST_FOREACH(CNode* pnode, vNodes){
|
|
pnode->PushMessage("inv", vInv);
|
|
}
|
|
}
|
|
|
|
bool CSporkManager::SetPrivKey(std::string strPrivKey)
|
|
{
|
|
CSporkMessage msg;
|
|
|
|
// Test signing successful, proceed
|
|
strMasterPrivKey = strPrivKey;
|
|
|
|
Sign(msg);
|
|
|
|
if(CheckSignature(msg)){
|
|
LogPrintf("CSporkManager::SetPrivKey - Successfully initialized as spork signer\n");
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
int CSporkManager::GetSporkIDByName(std::string strName)
|
|
{
|
|
if(strName == "SPORK_2_INSTANTX") return SPORK_2_INSTANTX;
|
|
if(strName == "SPORK_3_INSTANTX_BLOCK_FILTERING") return SPORK_3_INSTANTX_BLOCK_FILTERING;
|
|
if(strName == "SPORK_5_MAX_VALUE") return SPORK_5_MAX_VALUE;
|
|
if(strName == "SPORK_7_MASTERNODE_SCANNING") return SPORK_7_MASTERNODE_SCANNING;
|
|
if(strName == "SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT") return SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT;
|
|
if(strName == "SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT") return SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT;
|
|
if(strName == "SPORK_10_MASTERNODE_PAY_NEWEST_NODES") return SPORK_10_MASTERNODE_PAY_NEWEST_NODES;
|
|
|
|
return -1;
|
|
}
|
|
|
|
std::string CSporkManager::GetSporkNameByID(int id)
|
|
{
|
|
if(id == SPORK_2_INSTANTX) return "SPORK_2_INSTANTX";
|
|
if(id == SPORK_3_INSTANTX_BLOCK_FILTERING) return "SPORK_3_INSTANTX_BLOCK_FILTERING";
|
|
if(id == SPORK_5_MAX_VALUE) return "SPORK_5_MAX_VALUE";
|
|
if(id == SPORK_7_MASTERNODE_SCANNING) return "SPORK_7_MASTERNODE_SCANNING";
|
|
if(id == SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT) return "SPORK_8_MASTERNODE_PAYMENT_ENFORCEMENT";
|
|
if(id == SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT) return "SPORK_9_MASTERNODE_BUDGET_ENFORCEMENT";
|
|
if(id == SPORK_10_MASTERNODE_PAY_NEWEST_NODES) return "SPORK_10_MASTERNODE_PAY_NEWEST_NODES";
|
|
|
|
return "Unknown";
|
|
}
|