1#include <gmock/gmock.h>
2#include <gtest/gtest.h>
48using ::testing::StrictMock;
52using tracegen::BytecodeTraceBuilder;
53using tracegen::ClassIdDerivationTraceBuilder;
54using tracegen::ContractInstanceRetrievalTraceBuilder;
55using tracegen::FieldGreaterThanTraceBuilder;
56using tracegen::IndexedTreeCheckTraceBuilder;
57using tracegen::Poseidon2TraceBuilder;
58using tracegen::PrecomputedTraceBuilder;
59using tracegen::TestTraceContainer;
61using simulation::EventEmitter;
62using simulation::FieldGreaterThan;
63using simulation::IndexedTreeCheck;
65using simulation::IndexedTreeLeafData;
66using simulation::IndexedTreeReadWriteEvent;
67using simulation::MerkleCheck;
68using simulation::MockExecutionIdManager;
69using simulation::MockGreaterThan;
70using simulation::Poseidon2;
71using simulation::Poseidon2HashEvent;
72using simulation::Poseidon2PermutationEvent;
73using simulation::Poseidon2PermutationMemoryEvent;
74using simulation::RangeCheck;
75using simulation::RetrievedBytecodesTreeCheck;
82class BytecodeRetrievalConstrainingTest :
public ::testing::Test {
84 static TestTraceContainer init_trace()
87 TestTraceContainer
trace({
88 { { C::precomputed_first_row, 1 } },
94TEST_F(BytecodeRetrievalConstrainingTest, EmptyRow)
99TEST_F(BytecodeRetrievalConstrainingTest, SuccessfulRetrieval)
101 TestTraceContainer
trace = init_trace();
111 uint32_t bytecode_size = 20;
115 hash_input.reserve(1 + bytecode_fields.size());
116 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
119 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
120 .bytecode_length = bytecode_size,
121 .bytecode_fields = bytecode_fields } },
126 .nullifier_tree_root = nullifier_root,
127 .public_data_tree_root = public_data_tree_root,
132 .artifact_hash = klass.artifact_hash,
133 .private_functions_root = klass.private_functions_root,
134 .public_bytecode_commitment = bytecode_commitment } },
137 AppendOnlyTreeSnapshot snapshot_before = AppendOnlyTreeSnapshot{
142 AppendOnlyTreeSnapshot snapshot_after = AppendOnlyTreeSnapshot{
149 IndexedTreeReadWriteEvent{
150 .value =
instance.current_contract_class_id,
151 .prev_snapshot = snapshot_before,
152 .next_snapshot = snapshot_before,
154 .low_leaf_data = IndexedTreeLeafData{ .value = 0, .next_value = 0, .next_index = 0 },
159 IndexedTreeReadWriteEvent{
160 .value =
instance.current_contract_class_id,
161 .prev_snapshot = snapshot_before,
162 .next_snapshot = snapshot_after,
164 .low_leaf_data = IndexedTreeLeafData{ .value = 0, .next_value = 0, .next_index = 0 },
173 .bytecode_id = bytecode_commitment,
175 .current_class_id =
instance.current_contract_class_id,
176 .contract_class = klass,
177 .nullifier_root = nullifier_root,
178 .public_data_tree_root = public_data_tree_root,
179 .retrieved_bytecodes_snapshot_before = snapshot_before,
180 .retrieved_bytecodes_snapshot_after = snapshot_after,
181 .is_new_class =
true,
185 check_relation<bc_retrieval>(trace);
193TEST_F(BytecodeRetrievalConstrainingTest, TooManyBytecodes)
195 TestTraceContainer
trace = init_trace();
202 uint32_t bytecode_size = 20;
205 AppendOnlyTreeSnapshot snapshot_before = AppendOnlyTreeSnapshot{
207 .next_available_leaf_index =
211 AppendOnlyTreeSnapshot snapshot_after = AppendOnlyTreeSnapshot{
213 .next_available_leaf_index =
221 .current_class_id =
instance.current_contract_class_id,
222 .nullifier_root = nullifier_root,
223 .public_data_tree_root = public_data_tree_root,
224 .retrieved_bytecodes_snapshot_before = snapshot_before,
225 .retrieved_bytecodes_snapshot_after = snapshot_after,
226 .is_new_class =
true,
231 check_relation<bc_retrieval>(trace);
234TEST_F(BytecodeRetrievalConstrainingTest, NonExistentInstance)
238 TestTraceContainer
trace = init_trace();
243 .address = contract_address,
253 { C::bc_retrieval_sel, 1 },
254 { C::bc_retrieval_instance_exists, 0 },
255 { C::bc_retrieval_current_class_id, 0 },
256 { C::bc_retrieval_artifact_hash, 0 },
257 { C::bc_retrieval_private_functions_root, 0 },
258 { C::bc_retrieval_bytecode_id, 0 },
259 { C::bc_retrieval_address, contract_address },
260 { C::bc_retrieval_prev_retrieved_bytecodes_tree_size, 1 },
261 { C::bc_retrieval_next_retrieved_bytecodes_tree_size, 1 },
264 { C::bc_retrieval_error, 1 },
267 check_relation<bc_retrieval>(trace);
270 trace.
set(C::bc_retrieval_current_class_id, 1, 99);
272 (check_interaction<BytecodeTraceBuilder, lookup_bc_retrieval_contract_instance_retrieval_settings>(trace)),
273 "Failed.*LOOKUP_BC_RETRIEVAL_CONTRACT_INSTANCE_RETRIEVAL.*Could not find tuple in destination.");
275 trace.
set(C::contract_instance_retrieval_current_class_id, 1, 99);
277 ,
"INSTANCE_MEMBER_CLASS_ID_IS_ZERO_IF_DNE");
280 trace.
set(C::bc_retrieval_current_class_id, 1, 0);
281 trace.
set(C::contract_instance_retrieval_current_class_id, 1, 0);
284 trace.
set(C::bc_retrieval_bytecode_id, 1, 99);
287 trace.
set(C::bc_retrieval_bytecode_id, 1, 0);
290class BytecodeRetrievalConstrainingTestFewerMocks :
public BytecodeRetrievalConstrainingTest {
306 Poseidon2(mock_execution_id_manager, mock_gt, hash_event_emitter, perm_event_emitter, perm_mem_event_emitter);
309 FieldGreaterThan
field_gt = FieldGreaterThan(range_check, field_gt_emitter);
311 IndexedTreeCheck(
poseidon2, merkle_check, field_gt, indexed_tree_check_event_emitter);
324TEST_F(BytecodeRetrievalConstrainingTestFewerMocks, SuccessfulRetrievalFewerMocks)
326 TestTraceContainer
trace = init_trace();
332 uint32_t bytecode_size = 20;
336 hash_input.reserve(1 + bytecode_fields.size());
337 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
340 builder.process_decomposition({ { .bytecode_id = bytecode_commitment,
343 trace.
set(1, { { { C::bc_decomposition_sel_packed_read_0_, 1 } } });
344 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
345 .bytecode_length = bytecode_size,
346 .bytecode_fields = bytecode_fields } },
351 instance.current_contract_class_id = class_id;
356 .nullifier_tree_root = nullifier_root,
357 .public_data_tree_root = public_data_tree_root,
363 .artifact_hash = klass.artifact_hash,
364 .private_functions_root = klass.private_functions_root,
365 .public_bytecode_commitment = bytecode_commitment } },
382 .bytecode_id = bytecode_commitment,
384 .current_class_id =
instance.current_contract_class_id,
385 .contract_class = klass,
386 .nullifier_root = nullifier_root,
387 .public_data_tree_root = public_data_tree_root,
388 .retrieved_bytecodes_snapshot_before = snapshot_before,
389 .retrieved_bytecodes_snapshot_after = snapshot_after,
390 .is_new_class =
true,
398 check_relation<bc_retrieval>(trace);
399 check_relation<bb::avm2::bc_decomposition<FF>>(
trace);
400 check_relation<bb::avm2::bc_hashing<FF>>(
trace);
401 check_relation<bb::avm2::contract_instance_retrieval<FF>>(
trace);
402 check_relation<bb::avm2::class_id_derivation<FF>>(
trace);
403 check_relation<bb::avm2::indexed_tree_check<FF>>(
trace);
404 check_all_interactions<BytecodeTraceBuilder>(trace);
405 check_all_interactions<ClassIdDerivationTraceBuilder>(trace);
408TEST_F(BytecodeRetrievalConstrainingTestFewerMocks, SuccessfulRepeatedRetrievalFewerMocks)
410 TestTraceContainer
trace = init_trace();
416 uint32_t bytecode_size = 20;
420 hash_input.reserve(1 + bytecode_fields.size());
421 hash_input.insert(hash_input.end(), bytecode_fields.begin(), bytecode_fields.end());
424 builder.process_decomposition({ { .bytecode_id = bytecode_commitment,
427 trace.
set(1, { { { C::bc_decomposition_sel_packed_read_0_, 1 } } });
428 builder.process_hashing({ { .bytecode_id = bytecode_commitment,
429 .bytecode_length = bytecode_size,
430 .bytecode_fields = bytecode_fields } },
436 instance.current_contract_class_id = class_id;
441 .nullifier_tree_root = nullifier_root,
442 .public_data_tree_root = public_data_tree_root,
447 .artifact_hash = klass.artifact_hash,
448 .private_functions_root = klass.private_functions_root,
449 .public_bytecode_commitment = bytecode_commitment } },
474 .bytecode_id = bytecode_commitment,
476 .current_class_id =
instance.current_contract_class_id,
477 .contract_class = klass,
478 .nullifier_root = nullifier_root,
479 .public_data_tree_root = public_data_tree_root,
480 .retrieved_bytecodes_snapshot_before = snapshot_before,
481 .retrieved_bytecodes_snapshot_after = snapshot_after,
482 .is_new_class =
true,
485 .bytecode_id = bytecode_commitment,
487 .current_class_id =
instance.current_contract_class_id,
488 .contract_class = klass,
489 .nullifier_root = nullifier_root,
490 .public_data_tree_root = public_data_tree_root,
491 .retrieved_bytecodes_snapshot_before = snapshot_after,
492 .retrieved_bytecodes_snapshot_after = snapshot_after,
493 .is_new_class =
false,
501 check_relation<bc_retrieval>(trace);
502 check_relation<bb::avm2::bc_decomposition<FF>>(
trace);
503 check_relation<bb::avm2::bc_hashing<FF>>(
trace);
504 check_relation<bb::avm2::contract_instance_retrieval<FF>>(
trace);
505 check_relation<bb::avm2::class_id_derivation<FF>>(
trace);
506 check_relation<bb::avm2::indexed_tree_check<FF>>(
trace);
507 check_all_interactions<BytecodeTraceBuilder>(trace);
508 check_all_interactions<ClassIdDerivationTraceBuilder>(trace);
513 trace.
set(C::bc_retrieval_is_new_class, 1, 1);
516 trace.
set(C::indexed_tree_check_not_exists, 2, 1);
517 trace.
set(C::indexed_tree_check_exists, 2, 0);
518 check_all_interactions<BytecodeTraceBuilder>(trace);
519 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 AVM_RETRIEVED_BYTECODES_TREE_HEIGHT
#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
IndexedTreeCheckTraceBuilder indexed_tree_check_builder
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
IndexedTreeCheck indexed_tree_check
EventEmitter< simulation::RangeCheckEvent > range_check_emitter
EventEmitter< simulation::IndexedTreeCheckEvent > indexed_tree_check_event_emitter
bool contains(const FF &class_id) override
Checks whether a contract class ID exists in the retrieved bytecodes tree.
void insert(const FF &class_id) override
Inserts a contract class ID into the retrieved bytecodes tree.
AppendOnlyTreeSnapshot get_snapshot() const override
Returns the current tree snapshot.
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(const simulation::EventEmitterInterface< simulation::IndexedTreeCheckEvent >::Container &events, TraceContainer &trace)
Process generic indexed tree check events and populate the relevant columns in the 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 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
void check_interaction(tracegen::TestTraceContainer &trace)
TEST_F(AvmRecursiveTests, TwoLayerAvmRecursionFailsWithWrongPIs)
void check_relation(const tracegen::TestTraceContainer &trace, Ts... subrelation)
RetrievedBytecodesTree build_retrieved_bytecodes_tree()
std::vector< FF > encode_bytecode(std::span< const uint8_t > bytecode)
std::variant< IndexedTreeReadWriteEvent, CheckPointEventType > IndexedTreeCheckEvent
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