[Sapling] Load zk-snarks params.

furszy · furszy · commit 06b69667ba57 · 2020-08-02T23:03:37.000-03:00
diff --git a/src/init.cpp b/src/init.cpp
@@ -52,6 +52,10 @@
 #include "validationinterface.h"
 #include "zpivchain.h"
 
+// Sapling
+#include "sapling/util.h"
+#include <librustzcash.h>
+
 #ifdef ENABLE_WALLET
 #include "wallet/db.h"
 #include "wallet/wallet.h"
@@ -752,6 +756,30 @@ bool InitSanityCheck(void)
     return true;
 }
 
+static void LoadSaplingParams()
+{
+    struct timeval tv_start, tv_end;
+    float elapsed;
+    gettimeofday(&tv_start, 0);
+
+    try {
+        initZKSNARKS();
+    } catch (std::runtime_error &e) {
+        uiInterface.ThreadSafeMessageBox(strprintf(
+                _("Cannot find the Sapling parameters in the following directory:\n"
+                  "%s\n"
+                  "Please run 'sapling-fetch-params' or './util/fetch-params.sh' and then restart."),
+                ZC_GetParamsDir()),
+                                         "", CClientUIInterface::MSG_ERROR);
+        StartShutdown();
+        return;
+    }
+
+    gettimeofday(&tv_end, 0);
+    elapsed = float(tv_end.tv_sec-tv_start.tv_sec) + (tv_end.tv_usec-tv_start.tv_usec)/float(1000000);
+    LogPrintf("Loaded Sapling parameters in %fs seconds.\n", elapsed);
+}
+
 bool AppInitServers()
 {
     RPCServer::OnStarted(&OnRPCStarted);
@@ -1177,6 +1205,9 @@ bool AppInit2()
     CScheduler::Function serviceLoop = boost::bind(&CScheduler::serviceQueue, &scheduler);
     threadGroup.create_thread(boost::bind(&TraceThread<CScheduler::Function>, "scheduler", serviceLoop));
 
+    // Initialize Sapling circuit parameters
+    LoadSaplingParams();
+
     /* Start the RPC server already.  It will be started in "warmup" mode
      * and not really process calls already (but it will signify connections
      * that the server is there and will be ready later).  Warmup mode will
diff --git a/src/wallet/test/wallet_test_fixture.cpp b/src/wallet/test/wallet_test_fixture.cpp
@@ -6,10 +6,13 @@
 #include "wallet/test/wallet_test_fixture.h"
 
 #include "rpc/server.h"
+#include "sapling/util.h"
 #include "wallet/db.h"
 #include "wallet/wallet.h"
 #include "wallet/rpcwallet.h"
 
+#include <librustzcash.h>
+
 void clean()
 {
     delete pwalletMain;
@@ -21,6 +24,8 @@ void clean()
 
 WalletTestingSetup::WalletTestingSetup(): TestingSetup()
 {
+    initZKSNARKS(); // init zk-snarks lib
+
     clean(); // todo: research why we have an initialized bitdb here.
     bitdb.MakeMock();
     walletRegisterRPCCommands();
