Bluebox fix. (#64)

* Initial commit. * Fix compile error.
Chia-Network · Mar 2, 2021 · 1a200b5 · 1a200b5
1 parent 5825ec0
commit 1a200b5
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 13 deletions.
diff --git a/src/1weso_test.cpp b/src/1weso_test.cpp
@@ -3,6 +3,7 @@
 #include "verifier.h"
 
 #include <atomic>
+#include <cassert>
 
 int segments = 7;
 int thread_count = 3;
@@ -46,10 +47,10 @@ int main(int argc, char const* argv[]) try
     fast_algorithm = false;
 
     uint64_t iter = iter_multiplier;
-    OneWesolowskiCallback weso(D, iter);
+    OneWesolowskiCallback weso(D, f, iter);
     FastStorage* fast_storage = nullptr;
     std::thread vdf_worker(repeated_square, f, D, L, &weso, fast_storage, std::ref(stopped));
-    Proof const proof = ProveOneWesolowski(iter, D, &weso, stopped);
+    Proof const proof = ProveOneWesolowski(iter, D, f, &weso, stopped);
     stopped = true;
     vdf_worker.join();
 
@@ -59,7 +60,7 @@ int main(int argc, char const* argv[]) try
     form proof_form = DeserializeForm(D, proof.proof.data(), proof.proof.size());
     VerifyWesolowskiProof(D, x_init, y, proof_form, iter, is_valid);
     std::cout << "Verify result: " << is_valid << "\n";
-    return is_valid ? 0 : 1;
+    assert(is_valid);
 }
 catch (std::exception const& e) {
     std::cerr << "Exception " << e.what() << '\n';

diff --git a/src/callback.h b/src/callback.h
@@ -72,7 +72,7 @@ class WesolowskiCallback :public INUDUPLListener {
 
 class OneWesolowskiCallback: public WesolowskiCallback {
   public:
-    OneWesolowskiCallback(integer& D, uint64_t wanted_iter) : WesolowskiCallback(D) {
+    OneWesolowskiCallback(integer& D, form& f, uint64_t wanted_iter) : WesolowskiCallback(D) {
         uint32_t k, l;
         this->wanted_iter = wanted_iter;
         if (wanted_iter >= (1 << 16)) {
@@ -84,7 +84,6 @@ class OneWesolowskiCallback: public WesolowskiCallback {
         kl = k * l;
         uint64_t space_needed = wanted_iter / (k * l) + 100;
         forms.reset(new form[space_needed]);
-        form f = form::generator(D);
         forms[0] = f;
     }
 

diff --git a/src/vdf.h b/src/vdf.h
@@ -239,13 +239,12 @@ void repeated_square(form f, const integer& D, const integer& L,
     #endif
 }
 
-Proof ProveOneWesolowski(uint64_t iters, integer& D, OneWesolowskiCallback* weso,
+Proof ProveOneWesolowski(uint64_t iters, integer& D, form f, OneWesolowskiCallback* weso,
     std::atomic<bool>& stopped)
 {
     while (weso->iterations < iters) {
         this_thread::sleep_for(1s);
     }
-    form f = form::generator(D);
     Segment sg(
         /*start=*/0,
         /*length=*/iters,

diff --git a/src/vdf_client.cpp b/src/vdf_client.cpp
@@ -60,8 +60,8 @@ void CreateAndWriteProof(ProverManager& pm, uint64_t iteration, std::atomic<bool
     WriteProof(iteration, result, sock);
 }
 
-void CreateAndWriteProofOneWeso(uint64_t iters, integer& D, OneWesolowskiCallback* weso, std::atomic<bool>& stop_signal, tcp::socket& sock) {
-    Proof result = ProveOneWesolowski(iters, D, weso, stop_signal);
+void CreateAndWriteProofOneWeso(uint64_t iters, integer& D, form f, OneWesolowskiCallback* weso, std::atomic<bool>& stop_signal, tcp::socket& sock) {
+    Proof result = ProveOneWesolowski(iters, D, f, weso, stop_signal);
     if (stop_signal) {
         PrintInfo("Got stop signal before completing the proof!");
         return ;
@@ -198,9 +198,8 @@ void SessionOneWeso(tcp::socket& sock) {
     try {
         integer D(disc);
         integer L=root(-D, 4);
-        form f=form::generator(D);
         PrintInfo("Discriminant = " + to_string(D.impl));
-
+        form f = DeserializeForm(D, (uint8_t *)initial_form_s, sizeof(initial_form_s));
         // Tell client that I'm ready to get the challenges.
         boost::asio::write(sock, boost::asio::buffer("OK", 2));
 
@@ -210,10 +209,10 @@ void SessionOneWeso(tcp::socket& sock) {
             return;
         }
         std::atomic<bool> stopped(false);
-        WesolowskiCallback* weso = new OneWesolowskiCallback(D, iter);
+        WesolowskiCallback* weso = new OneWesolowskiCallback(D, f, iter);
         FastStorage* fast_storage = NULL;
         std::thread vdf_worker(repeated_square, f, std::ref(D), std::ref(L), weso, fast_storage, std::ref(stopped));
-        std::thread th_prover(CreateAndWriteProofOneWeso, iter, std::ref(D), (OneWesolowskiCallback*)weso, std::ref(stopped), std::ref(sock));
+        std::thread th_prover(CreateAndWriteProofOneWeso, iter, std::ref(D), f, (OneWesolowskiCallback*)weso, std::ref(stopped), std::ref(sock));
         iter = ReadIteration(sock);
         while (iter != 0) {
             std::cout << "Warning: did not receive stop signal\n";