Rooba
/
pupy
mirror of https://github.com/n1nj4sec/pupy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pupy/pupygen.py
View File

@ -6,6 +6,7 @@
import argparse import argparse
import sys import sys
import os.path import os.path
import re
from pupylib.utils.network import get_local_ip from pupylib.utils.network import get_local_ip
from network.conf import transports from network.conf import transports
@ -61,7 +62,10 @@ if __name__=="__main__":
myhost=myip myhost=myip
else: else:
myhost=args.host[0] 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 outpath=None
if args.type=="exe_x86": if args.type=="exe_x86":
binary=get_edit_pupyx86_exe(myhost, args.port, args.transport) binary=get_edit_pupyx86_exe(myhost, args.port, args.transport)