diff options
Diffstat (limited to 'sploit/payload/payload_entry.py')
| -rw-r--r-- | sploit/payload/payload_entry.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/sploit/payload/payload_entry.py b/sploit/payload/payload_entry.py index 4dca83d..295a91f 100644 --- a/sploit/payload/payload_entry.py +++ b/sploit/payload/payload_entry.py @@ -91,13 +91,14 @@ class padrel(padlen): class padalign(padlen): """Generate padding to reach next aligned address.""" - def __init__(self, size=None, data=None): + def __init__(self, size=None, data=None, reference=0): self.size = size self.data = data + self.reference = reference def payload_len(self, payload): size = self.size or arch.alignment - return -self.base % size + return (self.reference - self.base) % size class placeholder(padlen): """Generate fixed length of magic bytes, one word length by default.""" |