Override load_elf and load_binary (#437)

nervosnetwork · Jun 18, 2024 · 2b738a4 · 2b738a4
1 parent c51c230
commit 2b738a4
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/machine/mod.rs b/src/machine/mod.rs
@@ -353,6 +353,19 @@ impl<R: Register, M: Memory<REG = R>> SupportMachine for DefaultCoreMachine<R, M
         self.running = running;
     }
 
+    fn load_binary(
+        &mut self,
+        program: &Bytes,
+        metadata: &ProgramMetadata,
+        update_pc: bool,
+    ) -> Result<u64, Error> {
+        #[cfg(feature = "pprof")]
+        {
+            self.code = program.clone();
+        }
+        self.load_binary_inner(program, metadata, update_pc)
+    }
+
     fn load_elf(&mut self, program: &Bytes, update_pc: bool) -> Result<u64, Error> {
         #[cfg(feature = "pprof")]
         {
@@ -488,6 +501,19 @@ impl<Inner: SupportMachine> SupportMachine for DefaultMachine<Inner> {
         self.inner.set_running(running);
     }
 
+    fn load_binary(
+        &mut self,
+        program: &Bytes,
+        metadata: &ProgramMetadata,
+        update_pc: bool,
+    ) -> Result<u64, Error> {
+        self.inner.load_binary(program, metadata, update_pc)
+    }
+
+    fn load_elf(&mut self, program: &Bytes, update_pc: bool) -> Result<u64, Error> {
+        self.inner.load_elf(program, update_pc)
+    }
+
     #[cfg(feature = "pprof")]
     fn code(&self) -> &Bytes {
         self.inner.code()