Add support for accessible scopes

Co-authored-by: Mathieu <60658558+enitrat@users.noreply.github.com>

Author: JulianGCalderon

fuzzer/Cargo.lock

@@ -51,11 +51,12 @@ fn run() {
     }
     let mut vm = VirtualMachine::new(false, false);
     let mut hint_executor = BuiltinHintProcessor::new_empty();
-    let (ap_tracking_data, reference_ids, references, mut exec_scopes) = (
+    let (ap_tracking_data, reference_ids, references, mut exec_scopes, accessible_scopes) = (
         ApTracking::default(),
         HashMap::new(),
         Vec::new(),
         ExecutionScopes::new(),
+        Vec::new(),
     );
     let missing_hints: HashSet<_> = whitelists
         .into_iter()
@@ -69,6 +70,7 @@ fn run() {
                     &reference_ids,
                     &references,
                     Default::default(),
+                    &accessible_scopes,
                 )
                 .expect("this implementation is infallible");
             matches!(

hint_accountant/src/main.rs

@@ -133,6 +133,7 @@ pub struct HintProcessorData {
     pub ap_tracking: ApTracking,
     pub ids_data: HashMap<String, HintReference>,
     pub constants: Rc<HashMap<String, Felt252>>,
+    pub accessible_scopes: Vec<String>,
 }
 
 impl HintProcessorData {
@@ -142,6 +143,7 @@ impl HintProcessorData {
             ap_tracking: ApTracking::default(),
             ids_data,
             constants: Default::default(),
+            accessible_scopes: Default::default(),
         }
     }
 }

vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs

@@ -1266,6 +1266,7 @@ impl HintProcessorLogic for Cairo1HintProcessor {
         _references: &[HintReference],
         // Identifiers stored in the hint's program.
         _constants: Rc<HashMap<String, Felt252>>,
+        _accessible_scopes: &[String],
     ) -> Result<Box<dyn Any>, VirtualMachineError> {
         let data = hint_code.parse().ok().and_then(|x: usize| self.hints.get(&x).cloned())
         .ok_or_else(|| VirtualMachineError::CompileHintFail(

vm/src/hint_processor/cairo_1_hint_processor/hint_processor.rs

@@ -43,12 +43,15 @@ pub trait HintProcessorLogic {
         references: &[HintReference],
         // Identifiers stored in the hint's program.
         constants: Rc<HashMap<String, Felt252>>,
+        // List of accessible scopes in the hint
+        accessible_scopes: &[String],
     ) -> Result<Box<dyn Any>, VirtualMachineError> {
         Ok(any_box!(HintProcessorData {
             code: hint_code.to_string(),
             ap_tracking: ap_tracking_data.clone(),
             ids_data: get_ids_data(reference_ids, references)?,
             constants,
+            accessible_scopes: accessible_scopes.to_vec(),
         }))
     }
 

vm/src/hint_processor/hint_processor_definition.rs

@@ -290,13 +290,18 @@ pub fn vm_load_program(
     let hint_ap_tracking_data = ApTracking::default();
     let reference_ids = HashMap::default();
     let references = vec![];
+    let accessible_scopes = vec![
+        String::from("__main__"),
+        String::from("__main__.get_number"),
+    ];
     // Compile the hint
     let compiled_hint = hint_processor.compile_hint(
         hint_code,
         &hint_ap_tracking_data,
         &reference_ids,
         &references,
         Default::default(),
+        &accessible_scopes,
     )?;
     // Create the hint extension
     // As the hint from the compiled constract has offset 0, the hint pc will be equal to the loaded contract's program base:

vm/src/tests/run_deprecated_contract_class_simplified.rs

@@ -658,6 +658,7 @@ impl CairoRunner {
                         &hint.flow_tracking_data.reference_ids,
                         references,
                         constants.clone(),
+                        &hint.accessible_scopes,
                     )
                     .map_err(|_| VirtualMachineError::CompileHintFail(hint.code.clone().into()))
             })