preventing errors on pupygen

This commit is contained in:
n1nj4sec 2015-10-30 22:23:03 +01:00
parent f9cbf2db74
commit f9971648da
1 changed files with 5 additions and 1 deletions

View File

@ -6,6 +6,7 @@
import argparse
import sys
import os.path
import re
from pupylib.utils.network import get_local_ip
from network.conf import transports
@ -61,7 +62,10 @@ if __name__=="__main__":
myhost=myip
else:
myhost=args.host[0]
if re.match("^.*:[0-9]+$", myhost):#auto fixing errors when entering host:port
myhost, p=myhost.rsplit(':',1)
if args.port==443:
args.port=p
outpath=None
if args.type=="exe_x86":
binary=get_edit_pupyx86_exe(myhost, args.port, args.transport)