fuzzer_context.hpp

Go to the documentation of this file.

 
 
#pragma once
 
#include <memory>
#include <vector>
 
#include "barretenberg/avm_fuzzer/common/interfaces/dbs.hpp"
#include "barretenberg/vm2/common/field.hpp"
 
namespace bb::avm2::fuzzer {
 
class FuzzerContext {
  public:
    FuzzerContext()
        : contract_db_(std::make_unique<FuzzerContractDB>())
    {}
 
    // ---- Mutable API (for top-level fuzzers) ----
 
    FF register_contract_from_bytecode(const std::vector<uint8_t>& bytecode);
 
    void add_contract_address(FF address) { contract_addresses_.push_back(address); }
 
    void reset();
 
    // ---- Const API (for instruction generation, passed as const&) ----
 
    FF get_contract_address(size_t index) const
    {
        if (contract_addresses_.empty()) {
            return FF::zero();
        }
        return contract_addresses_[index % contract_addresses_.size()];
    }
 
    bool has_contracts() const { return !contract_addresses_.empty(); }
 
    size_t contract_count() const { return contract_addresses_.size(); }
 
    FuzzerContractDB& get_contract_db() const { return *contract_db_; }
 
    std::optional<std::pair<FF, uint64_t>> get_existing_note_hash(size_t index) const;
 
    void set_existing_note_hashes(std::span<const std::pair<FF, uint64_t>> note_hashes);
 
    void set_existing_contract_addresses(std::span<const FF> contract_addresses);
 
  private:
    std::vector<FF> contract_addresses_;
    std::unique_ptr<FuzzerContractDB> contract_db_;
    std::vector<std::pair<FF, uint64_t>> existing_note_hashes_;
};
 
} // namespace bb::avm2::fuzzer