sploit: cache ELF loads

With recursive ELF loads, there is the possibility of loading in a heavy
ELF (like libc) multiple times.  Hiding instantiation of the class
behind a factory method and caching instances should eliminate this
problem.

Signed-off-by: dusoleil <howcansocksbereal@gmail.com>

1 files changed, 11 insertions, 1 deletions

diff --git a/sploit/rev/elf.py b/sploit/rev/elf.py
index a748f10..bdced0a 100644
--- a/sploit/rev/elf.py
+++ b/sploit/rev/elf.py
@@ -1,6 +1,16 @@
 from sploit.rev import ldd, r2
 
-class ELF:
+__ELF_CACHE__ = {}
+
+def ELF(path):
+    if path in __ELF_CACHE__:
+        return __ELF_CACHE__[path]
+    else:
+        elf = __ELF__(path)
+        __ELF_CACHE__[path] = elf
+        return elf
+
+class __ELF__:
     def __init__(self, path):
         self.path = path
         self.sym = r2.get_elf_symbols(self.path)