From e920c68b5e82e9cb7b044102d0835ff609ad52da Mon Sep 17 00:00:00 2001
From: IntegralTeam V <shengguangxiao@gmail.com>
Date: Wed, 11 Sep 2019 16:25:47 +0700
Subject: [PATCH] add proofOfStake

---
 src/chain.h              | 18 +++++++++++++++++-
 src/miner.cpp            |  3 +++
 src/pos.cpp              |  6 ------
 src/pos.h                |  4 ----
 src/primitives/block.cpp |  2 ++
 src/primitives/block.h   | 18 ++++++++++++++----
 src/txdb.cpp             |  4 ++++
 7 files changed, 40 insertions(+), 15 deletions(-)

diff --git a/src/chain.h b/src/chain.h
index 4cc4847d3e1..26c303c38d8 100644
--- a/src/chain.h
+++ b/src/chain.h
@@ -209,6 +209,10 @@ class CBlockIndex
         nSequenceId = 0;
         nTimeMax = 0;
 
+        // PoS
+        proofOfStakeBody = boost::optional<CBlockHeader::PoS>{};
+        stakeModifier = uint256{};
+
         nVersion       = 0;
         hashMerkleRoot = uint256();
         nTime          = 0;
@@ -230,6 +234,8 @@ class CBlockIndex
         nTime          = block.nTime;
         nBits          = block.nBits;
         nNonce         = block.nNonce;
+        stakeModifier  = block.stakeModifier;
+        proofOfStakeBody = block.proofOfStakeBody;
     }
 
     FlatFilePos GetBlockPos() const {
@@ -260,6 +266,8 @@ class CBlockIndex
         block.nTime          = nTime;
         block.nBits          = nBits;
         block.nNonce         = nNonce;
+        block.stakeModifier   = stakeModifier;
+        block.proofOfStakeBody = proofOfStakeBody;
         return block;
     }
 
@@ -305,8 +313,9 @@ class CBlockIndex
 
     std::string ToString() const
     {
-        return strprintf("CBlockIndex(pprev=%p, nHeight=%d, merkle=%s, hashBlock=%s)",
+        return strprintf("CBlockIndex(pprev=%p, nHeight=%d, stakeModifier=(%s), merkle=%s, hashBlock=%s)",
             pprev, nHeight,
+            stakeModifier.ToString(),
             hashMerkleRoot.ToString(),
             GetBlockHash().ToString());
     }
@@ -386,6 +395,11 @@ class CDiskBlockIndex : public CBlockIndex
         if (nStatus & BLOCK_HAVE_UNDO)
             READWRITE(VARINT(nUndoPos));
 
+        //PoS serialization
+        CBlockHeader::PoS loc_proofOfStake = proofOfStakeBody ? *proofOfStakeBody : CBlockHeader::PoS{};
+        READWRITE(loc_proofOfStake);
+        proofOfStakeBody = loc_proofOfStake;
+
         // block header
         READWRITE(this->nVersion);
         READWRITE(hashPrev);
@@ -404,6 +418,8 @@ class CDiskBlockIndex : public CBlockIndex
         block.nTime           = nTime;
         block.nBits           = nBits;
         block.nNonce          = nNonce;
+        block.stakeModifier    = stakeModifier;
+        block.proofOfStakeBody = proofOfStakeBody;
         return block.GetHash();
     }
 
diff --git a/src/miner.cpp b/src/miner.cpp
index 015645c9c66..63ed457ca2d 100644
--- a/src/miner.cpp
+++ b/src/miner.cpp
@@ -16,6 +16,7 @@
 #include <policy/feerate.h>
 #include <policy/policy.h>
 #include <pow.h>
+#include <pos_kernel.h>
 #include <primitives/transaction.h>
 #include <script/standard.h>
 #include <timedata.h>
@@ -162,6 +163,8 @@ std::unique_ptr<CBlockTemplate> BlockAssembler::CreateNewBlock(const CScript& sc
     UpdateTime(pblock, chainparams.GetConsensus(), pindexPrev);
     pblock->nBits          = GetNextWorkRequired(pindexPrev, pblock, chainparams.GetConsensus());
     pblock->nNonce         = 0;
+    pblock->stakeModifier = uint256{}; // SS
+
     pblocktemplate->vTxSigOpsCost[0] = WITNESS_SCALE_FACTOR * GetLegacySigOpCount(*pblock->vtx[0]);
 
     CValidationState state;
diff --git a/src/pos.cpp b/src/pos.cpp
index 802bda8a564..ccbe7bc5201 100644
--- a/src/pos.cpp
+++ b/src/pos.cpp
@@ -125,10 +125,4 @@ namespace pos {
         }
         return CheckHeaderSignature(block);
     }
-
-    uint32_t GetNextTargetRequired(const CBlockIndex* pindexLast, const CBlockHeader* pblock,
-                                   const Consensus::Params& params) {
-        return ::GetNextWorkRequired(pindexLast, pblock, params);
-    }
-
 }
diff --git a/src/pos.h b/src/pos.h
index bc8c10cb077..4fb3dbbeaa7 100644
--- a/src/pos.h
+++ b/src/pos.h
@@ -23,9 +23,5 @@ namespace pos {
     bool CheckProofOfStake(const CBlockIndex* pindexPrev, const CBlock& block, CCoinsViewCache& view,
                            const Consensus::Params& params);
 
-/// @return nBits
-    uint32_t GetNextTargetRequired(const CBlockIndex* pindexLast, const CBlockHeader* pblock,
-                                   const Consensus::Params& params);
-
 }
 
diff --git a/src/primitives/block.cpp b/src/primitives/block.cpp
index 0e80f125541..dcc292b555d 100644
--- a/src/primitives/block.cpp
+++ b/src/primitives/block.cpp
@@ -7,6 +7,7 @@
 
 #include <hash.h>
 #include <tinyformat.h>
+#include <util/strencodings.h>
 #include <crypto/common.h>
 
 uint256 CBlockHeader::GetHash() const
@@ -28,6 +29,7 @@ std::string CBlock::ToString() const
         hashPrevBlock.ToString(),
         hashMerkleRoot.ToString(),
         nTime, nBits, nNonce,
+        proofOfStakeBody ? HexStr(proofOfStakeBody->sig.begin(), proofOfStakeBody->sig.end()) : "Empty",
         vtx.size());
     for (const auto& tx : vtx) {
         s << "  " << tx->ToString() << "\n";
diff --git a/src/primitives/block.h b/src/primitives/block.h
index 8023b5a0faa..aaba64d7a4b 100644
--- a/src/primitives/block.h
+++ b/src/primitives/block.h
@@ -40,10 +40,10 @@ class CBlockHeader
             READWRITE(coinstakePrevout);
             READWRITE(pubKeyHash);
             READWRITE(coinstakeAmount);
-            bool hashToSignAction = s.GetType() & SER_GETSIGNHASH;
-            if (!hashToSignAction) {
-                READWRITE(sig);
-            }
+//            bool hashToSignAction = s.GetType() & SER_GETSIGNHASH;
+//            if (!hashToSignAction) {
+//                READWRITE(sig);
+//            }
         }
     };
 
@@ -73,6 +73,12 @@ class CBlockHeader
         READWRITE(nTime);
         READWRITE(nBits);
         READWRITE(nNonce);
+        READWRITE(stakeModifier);
+
+        //PoS serialization
+        PoS loc_proofOfStake = proofOfStakeBody ? *proofOfStakeBody : PoS{};
+        READWRITE(loc_proofOfStake);
+        proofOfStakeBody = loc_proofOfStake;
     }
 
     void SetNull()
@@ -83,6 +89,8 @@ class CBlockHeader
         nTime = 0;
         nBits = 0;
         nNonce = 0;
+        stakeModifier.SetNull();
+        proofOfStakeBody = boost::optional<PoS>{};
     }
 
     bool IsNull() const
@@ -160,6 +168,8 @@ class CBlock : public CBlockHeader
         block.nTime          = nTime;
         block.nBits          = nBits;
         block.nNonce         = nNonce;
+        block.stakeModifier  = stakeModifier;
+        block.proofOfStakeBody = proofOfStakeBody;
         return block;
     }
 
diff --git a/src/txdb.cpp b/src/txdb.cpp
index 18be07e6dbb..be4adfa52de 100644
--- a/src/txdb.cpp
+++ b/src/txdb.cpp
@@ -274,6 +274,10 @@ bool CBlockTreeDB::LoadBlockIndexGuts(const Consensus::Params& consensusParams,
                 pindexNew->nStatus        = diskindex.nStatus;
                 pindexNew->nTx            = diskindex.nTx;
 
+                //PoS
+                pindexNew->stakeModifier = diskindex.stakeModifier;
+                pindexNew->proofOfStakeBody = diskindex.proofOfStakeBody;
+
                 if (!CheckProofOfWork(pindexNew->GetBlockHash(), pindexNew->nBits, consensusParams))
                     return error("%s: CheckProofOfWork failed: %s", __func__, pindexNew->ToString());