bbapi_chonk.hpp

Go to the documentation of this file.

#pragma once
#include "barretenberg/chonk/chonk.hpp"
#include "barretenberg/common/named_union.hpp"
#include "barretenberg/serialize/msgpack.hpp"
 
#ifndef __wasm__
#define BBAPI_CHONK_BATCH_VERIFIER_SUPPORTED
#endif
 
#ifdef BBAPI_CHONK_BATCH_VERIFIER_SUPPORTED
#include "barretenberg/chonk/batch_verifier_types.hpp"
#include "barretenberg/chonk/chonk_batch_verifier.hpp"
#include "barretenberg/chonk/chonk_proof.hpp"
#include <atomic>
#include <mutex>
#endif
 
#include <string>
#include <vector>
 
namespace bb::bbapi {
 
#ifdef BBAPI_CHONK_BATCH_VERIFIER_SUPPORTED
class ChonkBatchVerifierService {
  public:
    ChonkBatchVerifierService() = default;
    ~ChonkBatchVerifierService();
 
    ChonkBatchVerifierService(const ChonkBatchVerifierService&) = delete;
    ChonkBatchVerifierService& operator=(const ChonkBatchVerifierService&) = delete;
 
    void start(std::vector<std::shared_ptr<MegaZKFlavor::VKAndHash>> vks,
               uint32_t num_cores,
               uint32_t batch_size,
               const std::string& fifo_path);
    void enqueue(VerifyRequest request);
    void fail_request(uint64_t request_id, std::string error_message);
    void stop();
    bool is_running() const { return running_.load(); }
 
  private:
    bool write_result(VerifyResult result);
    bool ensure_fifo_open();
    void close_fifo_locked();
    bool fail_fifo_locked(const std::string& message);
 
    ChonkBatchVerifier verifier_;
 
    std::mutex fifo_mutex_;
    std::string fifo_path_;
    int fifo_fd_ = -1;
    std::atomic_bool running_ = false;
    std::atomic_bool fifo_failed_ = false;
};
#endif // BBAPI_CHONK_BATCH_VERIFIER_SUPPORTED
 
} // namespace bb::bbapi