Fix for an Issue #329

stamparm · stamparm · commit cb15fcc8afaf · 2013-01-02T22:17:06.000+01:00
diff --git a/lib/takeover/metasploit.py b/lib/takeover/metasploit.py
@@ -513,13 +513,10 @@ def createMsfShellcode(self, exitfunc, format, extra, encode):
         pollProcess(process)
         payloadStderr = process.communicate()[1]
 
-        if Backend.isOs(OS.WINDOWS) or extra == "BufferRegister=EAX":
-            payloadSize = re.search("size ([\d]+)", payloadStderr, re.I)
-        else:
-            payloadSize = re.search("Length\:\s([\d]+)", payloadStderr, re.I)
+        match = re.search("(Total size:|Length:|succeeded with size) ([\d]+)", payloadStderr)
 
-        if payloadSize:
-            payloadSize = int(payloadSize.group(1))
+        if match:
+            payloadSize = int(match.group(2))
 
             if extra == "BufferRegister=EAX":
                 payloadSize = payloadSize / 2
