Add RelativeMemLeak alias for MemLeak(..., relative=True)

Gallopsled · Dec 19, 2016 · ccd9253 · ccd9253
1 parent 8d3cf83
commit ccd9253
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pwn/toplevel.py b/pwn/toplevel.py
@@ -32,7 +32,7 @@
 from pwnlib.flag import *
 from pwnlib.fmtstr import FmtStr, fmtstr_payload
 from pwnlib.log import getLogger
-from pwnlib.memleak import MemLeak
+from pwnlib.memleak import MemLeak, RelativeMemLeak
 from pwnlib.regsort import *
 from pwnlib.replacements import *
 from pwnlib.rop import ROP

diff --git a/pwnlib/memleak.py b/pwnlib/memleak.py
@@ -625,3 +625,8 @@ def string_wrapper(address, *a, **kw):
     p32 = setd
     p16 = setw
     p8 = setb
+
+class RelativeMemLeak(MemLeak):
+    def __init__(self, *a, **kw):
+        kw.setdefault('relative', True)
+        return super(RelativeMemLeak, self).__init__(*a, **kw)