sploit: add ldd ability to rev module

add helper function to invoke ldd to get a list of libraries that will
be linked to a given ELF

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

2 files changed, 14 insertions, 0 deletions

diff --git a/sploit/rev/__init__.py b/sploit/rev/__init__.py
index e69de29..c489b98 100644
--- a/sploit/rev/__init__.py
+++ b/sploit/rev/__init__.py
@@ -0,0 +1,4 @@
+from . import (
+        ldd
+)
+
diff --git a/sploit/rev/ldd.py b/sploit/rev/ldd.py
new file mode 100644
index 0000000..60306f1
--- /dev/null
+++ b/sploit/rev/ldd.py
@@ -0,0 +1,10 @@
+import re
+from subprocess import run
+from collections import namedtuple as nt
+
+def get_libraries(elf):
+    out = run(['ldd',elf],capture_output=True).stdout.decode('utf-8').split('\n')[:-1]
+    out = [re.split(r'\s+',lib)[1:] for lib in out]
+    Lib = nt("Lib", "name path addr")
+    out = {l[0]:Lib(l[0],l[0] if l[0][0]=='/' else l[2] if l[1]=='=>' else None,l[-1]) for l in out}
+    return out