Fixes Memory Leak in JIT (#139)

* Fixes memory leak in JIT by introducing a memory management container struct.
solana-labs · Feb 10, 2021 · 575506c · 575506c
1 parent fcde4bd
commit 575506c
Show file tree

Hide file tree

Showing 2 changed files with 136 additions and 108 deletions.
diff --git a/src/elf.rs b/src/elf.rs
@@ -12,7 +12,7 @@ extern crate scroll;
 use crate::{
     ebpf,
     error::{EbpfError, UserDefinedError},
-    jit::{compile, JitProgram},
+    jit::JitProgram,
     vm::{Config, Executable, InstructionMeter, SyscallRegistry},
 };
 use byteorder::{ByteOrder, LittleEndian};
@@ -253,7 +253,7 @@ impl<E: UserDefinedError, I: InstructionMeter> Executable<E, I> for EBpfElf<E, I
 
     /// JIT compile the executable
     fn jit_compile(&mut self) -> Result<(), EbpfError<E>> {
-        self.compiled_program = Some(compile::<E, I>(self)?);
+        self.compiled_program = Some(JitProgram::<E, I>::new(self)?);
         Ok(())
     }