[Consensus] Set zc PublicSpend version v3/v4 via SPORK

random-zebra · random-zebra · commit 26045d33e918 · 2019-10-24T07:30:45.000+02:00
define SPORK_18_ZEROCOIN_PUBLICSPEND_V4. When such spork is not active, zerocoin publicspend version must be 3. Otherwise must be 4.
diff --git a/src/chainparams.cpp b/src/chainparams.cpp
@@ -132,13 +132,6 @@ bool CChainParams::HasStakeMinAgeOrDepth(const int contextHeight, const uint32_t
     return (contextHeight - utxoFromBlockHeight >= nStakeMinDepth);
 }
 
-int CChainParams::Zerocoin_PublicSpendVersion(const int nHeight) const
-{
-    if (nHeight < nPublicZCSpendsV4)
-        return 3;
-    return 4;
-}
-
 class CMainParams : public CChainParams
 {
 public:
@@ -194,7 +187,6 @@ class CMainParams : public CChainParams
 
         // Public coin spend enforcement
         nPublicZCSpends = 1880000;
-        nPublicZCSpendsV4 = 2880000;
 
         // New P2P messages signatures
         nBlockEnforceNewMessageSignatures = 2967000;
@@ -335,7 +327,6 @@ class CTestNetParams : public CMainParams
 
         // Public coin spend enforcement
         nPublicZCSpends = 1106100;
-        nPublicZCSpendsV4 = 2106100;
 
         // New P2P messages signatures
         nBlockEnforceNewMessageSignatures = 2214000;
@@ -434,7 +425,6 @@ class CRegTestParams : public CTestNetParams
 
         // Public coin spend enforcement
         nPublicZCSpends = 350;
-        nPublicZCSpendsV4 = 450;
 
         // New P2P messages signatures
         nBlockEnforceNewMessageSignatures = 1;
diff --git a/src/chainparams.h b/src/chainparams.h
@@ -144,7 +144,6 @@ class CChainParams
     int Zerocoin_Block_V2_Start() const { return nBlockZerocoinV2; }
     bool IsStakeModifierV2(const int nHeight) const { return nHeight >= nBlockStakeModifierlV2; }
     int NewSigsActive(const int nHeight) const { return nHeight >= nBlockEnforceNewMessageSignatures; }
-    int Zerocoin_PublicSpendVersion(const int nHeight) const;
 
     // fake serial attack
     int Zerocoin_Block_EndFakeSerial() const { return nFakeSerialBlockheightEnd; }
@@ -227,7 +226,6 @@ class CChainParams
     int nBlockZerocoinV2;
     int nBlockDoubleAccumulated;
     int nPublicZCSpends;
-    int nPublicZCSpendsV4;
     int nBlockStakeModifierlV2;
     int nBlockEnforceNewMessageSignatures;
 
diff --git a/src/main.cpp b/src/main.cpp
@@ -1009,8 +1009,12 @@ bool CheckPublicCoinSpendEnforced(int blockHeight, bool isPublicSpend) {
     return true;
 }
 
-bool CheckPublicCoinSpendVersion(int blockHeight, int version) {
-    return version >= Params().Zerocoin_PublicSpendVersion(blockHeight);
+int CurrentPublicCoinSpendVersion() {
+    return sporkManager.IsSporkActive(SPORK_18_ZEROCOIN_PUBLICSPEND_V4) ? 4 : 3;
+}
+
+bool CheckPublicCoinSpendVersion(int version) {
+    return version == CurrentPublicCoinSpendVersion();
 }
 
 bool ContextualCheckZerocoinSpend(const CTransaction& tx, const libzerocoin::CoinSpend* spend, CBlockIndex* pindex, const uint256& hashBlock)
@@ -1052,12 +1056,6 @@ bool ContextualCheckZerocoinSpendNoSerialCheck(const CTransaction& tx, const lib
         }
     }
 
-    // Check spend version
-    if (pindex->nHeight >= Params().Zerocoin_Block_Public_Spend_Enabled() &&
-            !CheckPublicCoinSpendVersion(pindex->nHeight, (int)spend->getVersion())) {
-        return error("%s: Invalid public spend version for tx %s", __func__, tx.GetHash().GetHex());
-    }
-
     bool fUseV1Params = spend->getCoinVersion() < libzerocoin::PrivateCoin::PUBKEY_VERSION;
 
     //Reject serial's that are not in the acceptable value range
@@ -1453,6 +1451,13 @@ bool AcceptToMemoryPool(CTxMemPool& pool, CValidationState& state, const CTransa
                     if (!ContextualCheckZerocoinSpend(tx, &publicSpend, chainActive.Tip(), 0))
                         return state.Invalid(error("%s: ContextualCheckZerocoinSpend failed for tx %s",
                                 __func__, tx.GetHash().GetHex()), REJECT_INVALID, "bad-txns-invalid-zpiv");
+
+                    // Check that the version matches the one enforced with SPORK_18
+                    if (!CheckPublicCoinSpendVersion(publicSpend.getVersion())) {
+                        return state.Invalid(error("%s : Public Zerocoin spend version %d not accepted. must be version %d.",
+                                __func__, publicSpend.getVersion(), CurrentPublicCoinSpendVersion()), REJECT_INVALID, "bad-txns-invalid-zpiv");
+                    }
+
                 } else {
                     libzerocoin::CoinSpend spend = TxInToZerocoinSpend(txIn);
                     if (!ContextualCheckZerocoinSpend(tx, &spend, chainActive.Tip(), 0))
@@ -4571,6 +4576,11 @@ bool CheckBlock(const CBlock& block, CValidationState& state, bool fCheckPOW, bo
                             return false;
                         }
                         spend = publicSpend;
+                        // check that the version matches the one enforced with SPORK_18 (don't ban if it fails)
+                        if (!IsInitialBlockDownload() && !CheckPublicCoinSpendVersion(spend.getVersion())) {
+                            return state.DoS(0, error("%s : Public Zerocoin spend version %d not accepted. must be version %d.",
+                                    __func__, spend.getVersion(), CurrentPublicCoinSpendVersion()), REJECT_INVALID, "bad-zcspend-version");
+                        }
                     } else {
                         spend = TxInToZerocoinSpend(txIn);
                     }
diff --git a/src/main.h b/src/main.h
@@ -375,7 +375,8 @@ bool isBlockBetweenFakeSerialAttackRange(int nHeight);
 
 // Public coin spend
 bool CheckPublicCoinSpendEnforced(int blockHeight, bool isPublicSpend);
-bool CheckPublicCoinSpendVersion(int blockHeight, int version);
+int CurrentPublicCoinSpendVersion();
+bool CheckPublicCoinSpendVersion(int version);
 
 /**
  * Check if transaction will be final in the next block to be created.
diff --git a/src/spork.cpp b/src/spork.cpp
@@ -25,6 +25,7 @@ std::vector<CSporkDef> sporkDefs = {
     MAKE_SPORK_DEF(SPORK_15_NEW_PROTOCOL_ENFORCEMENT_2,     4070908800ULL), // OFF
     MAKE_SPORK_DEF(SPORK_16_ZEROCOIN_MAINTENANCE_MODE,      4070908800ULL), // OFF
     MAKE_SPORK_DEF(SPORK_17_COLDSTAKING_ENFORCEMENT,        4070908800ULL), // OFF
+    MAKE_SPORK_DEF(SPORK_18_ZEROCOIN_PUBLICSPEND_V4,        4070908800ULL), // OFF
 };
 
 CSporkManager sporkManager;
diff --git a/src/sporkid.h b/src/sporkid.h
@@ -23,6 +23,7 @@ enum SporkId : int32_t {
     SPORK_15_NEW_PROTOCOL_ENFORCEMENT_2         = 10014,
     SPORK_16_ZEROCOIN_MAINTENANCE_MODE          = 10015,
     SPORK_17_COLDSTAKING_ENFORCEMENT            = 10017,
+    SPORK_18_ZEROCOIN_PUBLICSPEND_V4            = 10018,
 
     SPORK_INVALID                               = -1
 };
diff --git a/src/wallet/wallet.cpp b/src/wallet/wallet.cpp
@@ -4367,7 +4367,7 @@ bool CWallet::MintsToInputVectorPublicSpend(std::map<CBigNum, CZerocoinMint>& ma
     if (!nHeight)
         return error("%s: Unable to get chain tip height", __func__);
 
-    int spendVersion = Params().Zerocoin_PublicSpendVersion(nHeight);
+    int spendVersion = CurrentPublicCoinSpendVersion();
 
     int nLockAttempts = 0;
     while (nLockAttempts < 100) {
diff --git a/test/functional/zerocoin_valid_public_spend.py b/test/functional/zerocoin_valid_public_spend.py
