1#include <gmock/gmock.h>
2#include <gtest/gtest.h>
47using ::testing::StrictMock;
51using tracegen::BytecodeTraceBuilder;
52using tracegen::ClassIdDerivationTraceBuilder;
53using tracegen::ContractInstanceRetrievalTraceBuilder;
54using tracegen::FieldGreaterThanTraceBuilder;
55using tracegen::Poseidon2TraceBuilder;
56using tracegen::PrecomputedTraceBuilder;
57using tracegen::RetrievedBytecodesTreeCheckTraceBuilder;
58using tracegen::TestTraceContainer;
60using simulation::ClassIdLeafValue;
61using simulation::EventEmitter;
62using simulation::FieldGreaterThan;
63using simulation::MerkleCheck;
64using simulation::MockExecutionIdManager;
65using simulation::MockGreaterThan;
66using simulation::Poseidon2;
67using simulation::Poseidon2HashEvent;
68using simulation::Poseidon2PermutationEvent;
69using simulation::Poseidon2PermutationMemoryEvent;
70using simulation::RangeCheck;
71using simulation::RetrievedBytecodesTreeCheck;
72using simulation::RetrievedBytecodesTreeCheckEvent;
80class BytecodeRetrievalConstrainingTest :
public ::testing::Test {
82 static TestTraceContainer init_trace()
85 TestTraceContainer
trace({
86 { { C::precomputed_first_row, 1 } },
92TEST_F(BytecodeRetrievalConstrainingTest, EmptyRow)
97TEST_F(BytecodeRetrievalConstrainingTest, SuccessfulRetrieval)
99 TestTraceContainer
trace = init_trace();
109 uint32_t bytecode_size = 20;
113 hash_input.reserve(1 + bytecode_fields.size());
114 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
117 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
118 .bytecode_length = bytecode_size,
119 .bytecode_fields = bytecode_fields } },
124 .nullifier_tree_root = nullifier_root,
125 .public_data_tree_root = public_data_tree_root,
130 .artifact_hash = klass.artifact_hash,
131 .private_functions_root = klass.private_functions_root,
132 .public_bytecode_commitment = bytecode_commitment } },
135 AppendOnlyTreeSnapshot snapshot_before = AppendOnlyTreeSnapshot{
140 AppendOnlyTreeSnapshot snapshot_after = AppendOnlyTreeSnapshot{
147 RetrievedBytecodesTreeCheckEvent{
148 .class_id =
instance.current_contract_class_id,
149 .prev_snapshot = snapshot_before,
150 .next_snapshot = snapshot_before,
155 RetrievedBytecodesTreeCheckEvent{
156 .class_id =
instance.current_contract_class_id,
157 .prev_snapshot = snapshot_before,
158 .next_snapshot = snapshot_after,
167 .bytecode_id = bytecode_commitment,
169 .current_class_id =
instance.current_contract_class_id,
170 .contract_class = klass,
171 .nullifier_root = nullifier_root,
172 .public_data_tree_root = public_data_tree_root,
173 .retrieved_bytecodes_snapshot_before = snapshot_before,
174 .retrieved_bytecodes_snapshot_after = snapshot_after,
175 .is_new_class =
true,
179 check_relation<bc_retrieval>(trace);
187TEST_F(BytecodeRetrievalConstrainingTest, TooManyBytecodes)
189 TestTraceContainer
trace = init_trace();
196 uint32_t bytecode_size = 20;
199 AppendOnlyTreeSnapshot snapshot_before = AppendOnlyTreeSnapshot{
201 .next_available_leaf_index =
205 AppendOnlyTreeSnapshot snapshot_after = AppendOnlyTreeSnapshot{
207 .next_available_leaf_index =
215 .current_class_id =
instance.current_contract_class_id,
216 .nullifier_root = nullifier_root,
217 .public_data_tree_root = public_data_tree_root,
218 .retrieved_bytecodes_snapshot_before = snapshot_before,
219 .retrieved_bytecodes_snapshot_after = snapshot_after,
220 .is_new_class =
true,
225 check_relation<bc_retrieval>(trace);
228TEST_F(BytecodeRetrievalConstrainingTest, NonExistentInstance)
232 TestTraceContainer
trace = init_trace();
247 { C::bc_retrieval_sel, 1 },
248 { C::bc_retrieval_instance_exists, 0 },
249 { C::bc_retrieval_current_class_id, 0 },
250 { C::bc_retrieval_artifact_hash, 0 },
251 { C::bc_retrieval_private_functions_root, 0 },
252 { C::bc_retrieval_bytecode_id, 0 },
254 { C::bc_retrieval_prev_retrieved_bytecodes_tree_size, 1 },
255 { C::bc_retrieval_next_retrieved_bytecodes_tree_size, 1 },
257 { C::bc_retrieval_error, 1 },
260 check_relation<bc_retrieval>(trace);
263 trace.
set(C::bc_retrieval_current_class_id, 1, 99);
265 (check_interaction<BytecodeTraceBuilder, lookup_bc_retrieval_contract_instance_retrieval_settings>(trace)),
266 "Failed.*LOOKUP_BC_RETRIEVAL_CONTRACT_INSTANCE_RETRIEVAL.*Could not find tuple in destination.");
268 trace.
set(C::contract_instance_retrieval_current_class_id, 1, 99);
270 ,
"INSTANCE_MEMBER_CLASS_ID_IS_ZERO_IF_DNE");
273 trace.
set(C::bc_retrieval_current_class_id, 1, 0);
274 trace.
set(C::contract_instance_retrieval_current_class_id, 1, 0);
277 trace.
set(C::bc_retrieval_bytecode_id, 1, 99);
280 trace.
set(C::bc_retrieval_bytecode_id, 1, 0);
283class BytecodeRetrievalConstrainingTestFewerMocks :
public BytecodeRetrievalConstrainingTest {
299 Poseidon2(mock_execution_id_manager, mock_gt, hash_event_emitter, perm_event_emitter, perm_mem_event_emitter);
302 FieldGreaterThan
field_gt = FieldGreaterThan(range_check, field_gt_emitter);
309 retrieved_bytecodes_tree_check_emitter);
319TEST_F(BytecodeRetrievalConstrainingTestFewerMocks, SuccessfulRetrievalFewerMocks)
321 TestTraceContainer
trace = init_trace();
327 uint32_t bytecode_size = 20;
331 hash_input.reserve(1 + bytecode_fields.size());
332 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
335 builder.process_decomposition({ { .bytecode_id = bytecode_commitment,
338 trace.
set(1, { { { C::bc_decomposition_sel_packed_read_0_, 1 } } });
339 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
340 .bytecode_length = bytecode_size,
341 .bytecode_fields = bytecode_fields } },
351 .nullifier_tree_root = nullifier_root,
352 .public_data_tree_root = public_data_tree_root,
358 .artifact_hash = klass.artifact_hash,
359 .private_functions_root = klass.private_functions_root,
360 .public_bytecode_commitment = bytecode_commitment } },
377 .bytecode_id = bytecode_commitment,
379 .current_class_id =
instance.current_contract_class_id,
380 .contract_class = klass,
381 .nullifier_root = nullifier_root,
382 .public_data_tree_root = public_data_tree_root,
383 .retrieved_bytecodes_snapshot_before = snapshot_before,
384 .retrieved_bytecodes_snapshot_after = snapshot_after,
385 .is_new_class =
true,
393 check_relation<bc_retrieval>(trace);
394 check_relation<bb::avm2::bc_decomposition<FF>>(
trace);
395 check_relation<bb::avm2::bc_hashing<FF>>(
trace);
396 check_relation<bb::avm2::contract_instance_retrieval<FF>>(
trace);
397 check_relation<bb::avm2::class_id_derivation<FF>>(
trace);
398 check_relation<bb::avm2::retrieved_bytecodes_tree_check<FF>>(
trace);
399 check_all_interactions<BytecodeTraceBuilder>(trace);
400 check_all_interactions<ClassIdDerivationTraceBuilder>(trace);
403TEST_F(BytecodeRetrievalConstrainingTestFewerMocks, SuccessfulRepeatedRetrievalFewerMocks)
405 TestTraceContainer
trace = init_trace();
411 uint32_t bytecode_size = 20;
415 hash_input.reserve(1 + bytecode_fields.size());
416 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
419 builder.process_decomposition({ { .bytecode_id = bytecode_commitment,
422 trace.
set(1, { { { C::bc_decomposition_sel_packed_read_0_, 1 } } });
423 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
424 .bytecode_length = bytecode_size,
425 .bytecode_fields = bytecode_fields } },
436 .nullifier_tree_root = nullifier_root,
437 .public_data_tree_root = public_data_tree_root,
442 .artifact_hash = klass.artifact_hash,
443 .private_functions_root = klass.private_functions_root,
444 .public_bytecode_commitment = bytecode_commitment } },
469 .bytecode_id = bytecode_commitment,
471 .current_class_id =
instance.current_contract_class_id,
472 .contract_class = klass,
473 .nullifier_root = nullifier_root,
474 .public_data_tree_root = public_data_tree_root,
475 .retrieved_bytecodes_snapshot_before = snapshot_before,
476 .retrieved_bytecodes_snapshot_after = snapshot_after,
477 .is_new_class =
true,
480 .bytecode_id = bytecode_commitment,
482 .current_class_id =
instance.current_contract_class_id,
483 .contract_class = klass,
484 .nullifier_root = nullifier_root,
485 .public_data_tree_root = public_data_tree_root,
486 .retrieved_bytecodes_snapshot_before = snapshot_after,
487 .retrieved_bytecodes_snapshot_after = snapshot_after,
488 .is_new_class =
false,
496 check_relation<bc_retrieval>(trace);
497 check_relation<bb::avm2::bc_decomposition<FF>>(
trace);
498 check_relation<bb::avm2::bc_hashing<FF>>(
trace);
499 check_relation<bb::avm2::contract_instance_retrieval<FF>>(
trace);
500 check_relation<bb::avm2::class_id_derivation<FF>>(
trace);
501 check_relation<bb::avm2::retrieved_bytecodes_tree_check<FF>>(
trace);
502 check_all_interactions<BytecodeTraceBuilder>(trace);
503 check_all_interactions<ClassIdDerivationTraceBuilder>(trace);
508 trace.
set(C::bc_retrieval_is_new_class, 1, 1);
511 trace.
set(C::retrieved_bytecodes_tree_check_leaf_not_exists, 2, 1);
512 check_all_interactions<BytecodeTraceBuilder>(trace);
513 check_relation<bc_retrieval>(trace);
#define EXPECT_THROW_WITH_MESSAGE(code, expectedMessageRegex)
std::shared_ptr< Napi::ThreadSafeFunction > instance
#define AVM_RETRIEVED_BYTECODES_TREE_INITIAL_ROOT
#define AVM_RETRIEVED_BYTECODES_TREE_INITIAL_SIZE
#define MAX_PUBLIC_CALLS_TO_UNIQUE_CONTRACT_CLASS_IDS
#define DOM_SEP__CONTRACT_CLASS_ID
StrictMock< MockGreaterThan > mock_gt
EventEmitter< Poseidon2PermutationMemoryEvent > perm_mem_event_emitter
EventEmitter< Poseidon2PermutationEvent > perm_event_emitter
EventEmitter< Poseidon2HashEvent > hash_event_emitter
Poseidon2TraceBuilder poseidon2_builder
StrictMock< MockExecutionIdManager > mock_execution_id_manager
FieldGreaterThan field_gt
ClassIdDerivationTraceBuilder class_id_builder
EventEmitter< simulation::MerkleCheckEvent > merkle_check_emitter
ContractInstanceRetrievalTraceBuilder contract_instance_retrieval_builder
RetrievedBytecodesTreeCheck retrieved_bytecodes_tree_check
EventEmitter< simulation::FieldGreaterThanEvent > field_gt_emitter
EventEmitter< simulation::RetrievedBytecodesTreeCheckEvent > retrieved_bytecodes_tree_check_emitter
EventEmitter< simulation::RangeCheckEvent > range_check_emitter
RetrievedBytecodesTreeCheckTraceBuilder retrieved_bytecodes_tree_check_builder
bool contains(const FF &class_id) override
void insert(const FF &class_id) override
AppendOnlyTreeSnapshot get_snapshot() const override
void process(const simulation::EventEmitterInterface< simulation::ClassIdDerivationEvent >::Container &events, TraceContainer &trace)
Process class id derivation events and populate the relevant columns in the trace....
void process(const simulation::EventEmitterInterface< simulation::ContractInstanceRetrievalEvent >::Container &events, TraceContainer &trace)
void process_hash(const simulation::EventEmitterInterface< simulation::Poseidon2HashEvent >::Container &hash_events, TraceContainer &trace)
Processes the hash events for the Poseidon2 hash function. It populates the columns for the poseidon2...
void process_misc(TraceContainer &trace, const uint32_t num_rows=PRECOMPUTED_TRACE_SIZE)
void process_sel_range_8(TraceContainer &trace)
void process(const simulation::EventEmitterInterface< simulation::RetrievedBytecodesTreeCheckEvent >::Container &events, TraceContainer &trace)
void set(Column col, uint32_t row, const FF &value)
static FF hash(const std::vector< FF > &input)
Hashes a vector of field elements.
PrecomputedTraceBuilder precomputed_builder
FieldGreaterThanTraceBuilder field_gt_builder
AztecAddress contract_address
void check_interaction(tracegen::TestTraceContainer &trace)
TEST_F(AvmRecursiveTests, TwoLayerAvmRecursionFailsWithWrongPIs)
void check_relation(const tracegen::TestTraceContainer &trace, Ts... subrelation)
IndexedLeaf< ClassIdLeafValue > RetrievedBytecodesTreeLeafPreimage
RetrievedBytecodesTree build_retrieved_bytecodes_tree()
std::vector< FF > encode_bytecode(std::span< const uint8_t > bytecode)
FF compute_public_bytecode_first_field(size_t bytecode_size)
ContractClass random_contract_class(size_t bytecode_size)
ContractInstance random_contract_instance()
TestTraceContainer empty_trace()
lookup_settings< lookup_bc_retrieval_retrieved_bytecodes_insertion_settings_ > lookup_bc_retrieval_retrieved_bytecodes_insertion_settings
lookup_settings< lookup_bc_retrieval_contract_instance_retrieval_settings_ > lookup_bc_retrieval_contract_instance_retrieval_settings
lookup_settings< lookup_bc_retrieval_class_id_derivation_settings_ > lookup_bc_retrieval_class_id_derivation_settings
lookup_settings< lookup_bc_retrieval_is_new_class_check_settings_ > lookup_bc_retrieval_is_new_class_check_settings
constexpr decltype(auto) get(::tuplet::tuple< T... > &&t) noexcept
constexpr field invert() const noexcept
static field random_element(numeric::RNG *engine=nullptr) noexcept