Skip to content

Commit 8057936

Browse files
starknet_os: allow full OS output with variable full_output flag
1 parent b8b425b commit 8057936

File tree

1 file changed

+10
-3
lines changed
  • crates/starknet_os/src/hints/hint_implementation/aggregator

1 file changed

+10
-3
lines changed

crates/starknet_os/src/hints/hint_implementation/aggregator/utils.rs

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,7 @@ fn write_full_os_output(
237237
vm: &mut VirtualMachine,
238238
address: Relocatable,
239239
state_diff_writer: &mut FullStateDiffWriter,
240+
full_output_flag: bool,
240241
) -> VmUtilsResult<Relocatable> {
241242
let FullOsOutput { common_os_output, state_diff } = output;
242243
let messages_to_l1_start = vm.add_temporary_segment();
@@ -269,7 +270,7 @@ fn write_full_os_output(
269270
common_os_output.os_program_hash.into(),
270271
common_os_output.starknet_os_config_hash.into(),
271272
Felt::ZERO.into(), // use_kzg_da field (False in the aggregator input).
272-
Felt::ONE.into(), // full_output field (True in the aggregator input).
273+
Felt::from(full_output_flag).into(),
273274
])?;
274275

275276
let squashed_os_state_update = vm.gen_arg(&vec![
@@ -296,11 +297,17 @@ impl LoadIntoVmMemory for FullOsOutputs {
296297
vm: &mut VirtualMachine,
297298
address: Relocatable,
298299
) -> VmUtilsResult<Relocatable> {
300+
let full_output = true;
299301
let mut os_output_ptr = address;
300302
let mut contract_changes_writer = FullStateDiffWriter::new(vm);
301303
for output in &self.0 {
302-
os_output_ptr =
303-
write_full_os_output(output, vm, os_output_ptr, &mut contract_changes_writer)?;
304+
os_output_ptr = write_full_os_output(
305+
output,
306+
vm,
307+
os_output_ptr,
308+
&mut contract_changes_writer,
309+
full_output,
310+
)?;
304311
}
305312
Ok(os_output_ptr)
306313
}

0 commit comments

Comments
 (0)