Merge pull request #8 from mpgn/patch-1

Fix password error and error on the migration to python3
Wh1t3Fox · Jan 25, 2020 · 371d5bf · 371d5bf
2 parents 61cbff5 + 94ae6b6
commit 371d5bf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/polenum.py b/polenum.py
@@ -25,7 +25,7 @@ def d2b(a):
     tbin = []
     while a:
         tbin.append(a % 2)
-        a /= 2
+        a //= 2
 
     t2bin = tbin[::-1]
     if len(t2bin) != 8:
@@ -266,6 +266,9 @@ def main():
     if args.enum4linux:
         enum4linux_regex = re.compile('(?:([^@:]*)(?::([^@]*))?@)?(.*)')
         user, passw, target = enum4linux_regex.match(args.enum4linux).groups()
+        if '@' in target:
+            passw = passw + '@' + target.rpartition('@')[0]
+            target = target.rpartition('@')[2]
 
     if args.protocols:
         dumper = SAMRDump(args.protocols, user, passw)