r1

2022-02-03 23:45:47 -08:00
parent 42c2062cc4
commit 184ece190c
1438 changed files with 404064 additions and 0 deletions
@@ -0,0 +1,116 @@
+// Copyright (c) 2015-2019 The PIVX developers
+// Distributed under the MIT software license, see the accompanying
+// file COPYING or http://www.opensource.org/licenses/mit-license.php.
+//
+
+
+#include "lightzagrthread.h"
+#include "main.h"
+
+/****** Thread ********/
+void CLightWorker::ThreadLightZAGRSimplified() {
+    RenameThread("agrarian-light-thread");
+    isWorkerRunning = true;
+    while (true) {
+        try {
+
+            // Take a breath between requests.. TODO: Add processor usage check here
+            MilliSleep(2000);
+
+            // TODO: Future: join several similar requests into one calculation if the filter and denom match..
+            CGenWit genWit = requestsQueue.pop();
+            LogPrintf("%s pop work for %s \n\n", "agrarian-light-thread", genWit.toString());
+
+            libzerocoin::ZerocoinParams *params = Params().Zerocoin_Params(false);
+            CBlockIndex *pIndex = chainActive[genWit.getStartingHeight()];
+            if (!pIndex) {
+                // Rejects only the failed height
+                rejectWork(genWit, genWit.getStartingHeight(), NON_DETERMINED);
+            } else {
+                LogPrintf("%s calculating work for %s \n\n", "agrarian-light-thread", genWit.toString());
+                int blockHeight = pIndex->nHeight;
+                if (blockHeight >= Params().Zerocoin_Block_V2_Start()) {
+
+                    // TODO: The protocol actually doesn't care about the Accumulator..
+                    libzerocoin::Accumulator accumulator(params, genWit.getDen(), genWit.getAccWitValue());
+                    libzerocoin::PublicCoin temp(params);
+                    libzerocoin::AccumulatorWitness witness(params, accumulator, temp);
+                    string strFailReason = "";
+                    int nMintsAdded = 0;
+                    CZerocoinSpendReceipt receipt;
+
+                    list<CBigNum> ret;
+                    int heightStop;
+
+                    bool res;
+                    try {
+
+                        res = CalculateAccumulatorWitnessFor(
+                                params,
+                                blockHeight,
+                                COMP_MAX_AMOUNT,
+                                genWit.getDen(),
+                                genWit.getFilter(),
+                                accumulator,
+                                witness,
+                                nMintsAdded,
+                                strFailReason,
+                                ret,
+                                heightStop
+                        );
+
+                    } catch (NotEnoughMintsException e) {
+                        LogPrintStr(std::string("ThreadLightZAGRSimplified: ") + e.message + "\n");
+                        rejectWork(genWit, blockHeight, NOT_ENOUGH_MINTS);
+                        continue;
+                    }
+
+                    if (!res) {
+                        // TODO: Check if the GenerateAccumulatorWitnessFor can fail for node's fault or it's just because the peer sent an illegal request..
+                        rejectWork(genWit, blockHeight, NON_DETERMINED);
+                    } else {
+                        CDataStream ss(SER_NETWORK, PROTOCOL_VERSION);
+                        ss.reserve(ret.size() * 32);
+
+                        ss << genWit.getRequestNum();
+                        ss << accumulator.getValue(); // TODO: ---> this accumulator value is not necessary. The light node should get it using the other message..
+                        ss << witness.getValue();
+                        uint32_t size = ret.size();
+                        ss << size;
+                        for (CBigNum bnValue : ret) {
+                            ss << bnValue;
+                        }
+                        ss << heightStop;
+                        if (genWit.getPfrom()) {
+                            LogPrintf("%s pushing message to %s \n", "agrarian-light-thread", genWit.getPfrom()->addrName);
+                            genWit.getPfrom()->PushMessage("pubcoins", ss);
+                        } else
+                            LogPrintf("%s NOT pushing message to %s \n", "agrarian-light-thread", genWit.getPfrom()->addrName);
+                    }
+                } else {
+                    // Rejects only the failed height
+                    rejectWork(genWit, blockHeight, NON_DETERMINED);
+                }
+            }
+        } catch (std::exception& e) {
+            //std::cout << "exception in light loop, closing it. " << e.what() << std::endl;
+            PrintExceptionContinue(&e, "lightzagrthread");
+            break;
+        }
+    }
+
+
+}
+
+// TODO: Think more the peer misbehaving policy..
+void CLightWorker::rejectWork(CGenWit& wit, int blockHeight, uint32_t errorNumber) {
+    if (wit.getStartingHeight() == blockHeight){
+        LogPrintf("%s rejecting work %s , error code: %s\n", "agrarian-light-thread", wit.toString(), errorNumber);
+        CDataStream ss(SER_NETWORK, PROTOCOL_VERSION);
+        ss << wit.getRequestNum();
+        ss << errorNumber;
+        wit.getPfrom()->PushMessage("pubcoins", ss);
+    } else {
+        requestsQueue.push(wit);
+    }
+}