From 50e1fa485853eb3fce95ac6927bd305f63032d05 Mon Sep 17 00:00:00 2001
From: YairVaknin-starkware <yair.v@starkware.co>
Date: Sun, 24 Nov 2024 17:20:02 +0200
Subject: [PATCH] Changes_for_using_hint_extention_feature

---
 CHANGELOG.md                                     |  2 ++
 .../builtin_hint_processor_definition.rs         |  3 +++
 .../cairo_1_hint_processor/hint_processor.rs     |  4 ++++
 .../hint_processor/hint_processor_definition.rs  |  1 +
 vm/src/types/program.rs                          | 16 ++++++++--------
 5 files changed, 18 insertions(+), 8 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 2e843ee992..03373a39be 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -3,6 +3,8 @@
 #### Upcoming Changes
 
 #### [2.0.0-rc1] - 2024-11-20
+* chore: [#1855](https://github.com/lambdaclass/cairo-vm/pull/1880):
+  * Refactor vm crate to make it possible to use hint extension feature for nested programs with hints.
 
 * feat: add `EvalCircuit` and `TestLessThanOrEqualAddress` hints [#1843](https://github.com/lambdaclass/cairo-vm/pull/1843)
 
diff --git a/vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs b/vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs
index fcc6ad0dcc..bc1519d5f3 100644
--- a/vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs
+++ b/vm/src/hint_processor/builtin_hint_processor/builtin_hint_processor_definition.rs
@@ -180,6 +180,9 @@ impl BuiltinHintProcessor {
 }
 
 impl HintProcessorLogic for BuiltinHintProcessor {
+    fn as_any_mut(&mut self) -> &mut dyn Any {
+        self
+    }
     fn execute_hint(
         &mut self,
         vm: &mut VirtualMachine,
diff --git a/vm/src/hint_processor/cairo_1_hint_processor/hint_processor.rs b/vm/src/hint_processor/cairo_1_hint_processor/hint_processor.rs
index 9b6e4b8651..ae3f16e91d 100644
--- a/vm/src/hint_processor/cairo_1_hint_processor/hint_processor.rs
+++ b/vm/src/hint_processor/cairo_1_hint_processor/hint_processor.rs
@@ -1286,6 +1286,10 @@ impl HintProcessorLogic for Cairo1HintProcessor {
         }
         Ok(())
     }
+
+    fn as_any_mut(&mut self) -> &mut dyn Any {
+        self
+    }
 }
 
 impl ResourceTracker for Cairo1HintProcessor {
diff --git a/vm/src/hint_processor/hint_processor_definition.rs b/vm/src/hint_processor/hint_processor_definition.rs
index 496e8b1428..05b1db2cdc 100644
--- a/vm/src/hint_processor/hint_processor_definition.rs
+++ b/vm/src/hint_processor/hint_processor_definition.rs
@@ -51,6 +51,7 @@ pub trait HintProcessorLogic {
         }))
     }
 
+    fn as_any_mut(&mut self) -> &mut dyn Any;
     #[cfg(feature = "extensive_hints")]
     // Executes the hint which's data is provided by a dynamic structure previously created by compile_hint
     // Also returns a map of hints to be loaded after the current hint is executed
diff --git a/vm/src/types/program.rs b/vm/src/types/program.rs
index 477f996577..1c97580ee0 100644
--- a/vm/src/types/program.rs
+++ b/vm/src/types/program.rs
@@ -60,9 +60,9 @@ use arbitrary::{Arbitrary, Unstructured};
 // failures.
 // Fields in `Program` (other than `SharedProgramData` itself) are used by the main logic.
 #[derive(Clone, Default, Debug, PartialEq, Eq)]
-pub(crate) struct SharedProgramData {
+pub struct SharedProgramData {
     pub(crate) data: Vec<MaybeRelocatable>,
-    pub(crate) hints_collection: HintsCollection,
+    pub hints_collection: HintsCollection,
     pub(crate) main: Option<usize>,
     //start and end labels will only be used in proof-mode
     pub(crate) start: Option<usize>,
@@ -70,7 +70,7 @@ pub(crate) struct SharedProgramData {
     pub(crate) error_message_attributes: Vec<Attribute>,
     pub(crate) instruction_locations: Option<HashMap<usize, InstructionLocation>>,
     pub(crate) identifiers: HashMap<String, Identifier>,
-    pub(crate) reference_manager: Vec<HintReference>,
+    pub reference_manager: Vec<HintReference>,
 }
 
 #[cfg(feature = "test_utils")]
@@ -107,13 +107,13 @@ impl<'a> Arbitrary<'a> for SharedProgramData {
 }
 
 #[derive(Clone, Default, Debug, PartialEq, Eq)]
-pub(crate) struct HintsCollection {
-    hints: Vec<HintParams>,
+pub struct HintsCollection {
+    pub hints: Vec<HintParams>,
     /// This maps a PC to the range of hints in `hints` that correspond to it.
     #[cfg(not(feature = "extensive_hints"))]
     pub(crate) hints_ranges: Vec<HintRange>,
     #[cfg(feature = "extensive_hints")]
-    pub(crate) hints_ranges: HashMap<Relocatable, HintRange>,
+    pub hints_ranges: HashMap<Relocatable, HintRange>,
 }
 
 impl HintsCollection {
@@ -200,8 +200,8 @@ pub type HintRange = (usize, NonZeroUsize);
 #[cfg_attr(feature = "test_utils", derive(Arbitrary))]
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub struct Program {
-    pub(crate) shared_program_data: Arc<SharedProgramData>,
-    pub(crate) constants: HashMap<String, Felt252>,
+    pub shared_program_data: Arc<SharedProgramData>,
+    pub constants: HashMap<String, Felt252>,
     pub(crate) builtins: Vec<BuiltinName>,
 }