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

add --debug option to generate payloads with the debug template

This commit is contained in:
n1nj4sec 2016-11-18 18:36:11 +01:00
parent 33650ccbb2
commit 9e2a86509d
1 changed files with 34 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
pupy/pupygen.py
View File

@ -22,28 +22,44 @@ import os
ROOT=os.path.abspath(os.path.join(os.path.dirname(__file__)))
def get_edit_pupyx86_dll(conf):
def get_edit_pupyx86_dll(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86d.dll"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86.dll"), conf)
def get_edit_pupyx64_dll(conf):
def get_edit_pupyx64_dll(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64d.dll"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64.dll"), conf)
def get_edit_pupyx86_exe(conf):
def get_edit_pupyx86_exe(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86d.exe"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86.exe"), conf)
def get_edit_pupyx64_exe(conf):
def get_edit_pupyx64_exe(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64d.exe"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64.exe"), conf)
def get_edit_pupyx86_lin(conf):
def get_edit_pupyx86_lin(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86d.lin"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86.lin"), conf)
def get_edit_pupyx64_lin(conf):
def get_edit_pupyx64_lin(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64d.lin"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64.lin"), conf)
def get_edit_pupyx86_so(conf):
def get_edit_pupyx86_so(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86d.so"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx86.so"), conf)
def get_edit_pupyx64_so(conf):
def get_edit_pupyx64_so(conf, debug=False):
if debug:
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64d.so"), conf)
return get_edit_binary(os.path.join(ROOT, "payload_templates","pupyx64.so"), conf)
def get_edit_binary(path, conf):
@ -302,6 +318,7 @@ if __name__=="__main__":
parser.add_argument('--ps1-oneliner-listen-port', default=8080, type=int, help="Port used by ps1_oneliner listener (default: %(default)s)")
parser.add_argument('--randomize-hash', action='store_true', help="add a random string in the exe to make it's hash unknown")
parser.add_argument('--debug-scriptlets', action='store_true', help="don't catch scriptlets exceptions on the client for debug purposes")
parser.add_argument('--debug', action='store_true', help="build with the debug template (the payload open a console)")
parser.add_argument('--workdir', help='Set Workdir (Default = current workdir)')
parser.add_argument('launcher', choices=[x for x in launchers.iterkeys()], default='auto_proxy', help="Choose a launcher. Launchers make payloads behave differently at startup.")
parser.add_argument('launcher_args', nargs=argparse.REMAINDER, help="launcher options")
@ -341,53 +358,53 @@ if __name__=="__main__":
conf['offline_script']=script_code
outpath=args.output
if args.format=="exe_x86":
binary=get_edit_pupyx86_exe(conf)
binary=get_edit_pupyx86_exe(conf, debug=args.debug)
if not outpath:
outpath="pupyx86.exe"
with open(outpath, 'wb') as w:
w.write(binary)
elif args.format=="lin_x86":
binary=get_edit_pupyx86_lin(conf)
binary=get_edit_pupyx86_lin(conf, debug=args.debug)
if not outpath:
outpath="pupyx86.lin"
with open(outpath, 'wb') as w:
w.write(binary)
os.chmod(outpath, 0711)
elif args.format=="so_x86":
binary=get_edit_pupyx86_lin(conf)
binary=get_edit_pupyx86_lin(conf, debug=args.debug)
if not outpath:
outpath="pupyx86.so"
with open(outpath, 'wb') as w:
w.write(binary)
os.chmod(outpath, 0711)
elif args.format=="lin_x64":
binary=get_edit_pupyx64_lin(conf)
binary=get_edit_pupyx64_lin(conf, debug=args.debug)
if not outpath:
outpath="pupyx64.lin"
with open(outpath, 'wb') as w:
w.write(binary)
os.chmod(outpath, 0711)
elif args.format=="so_x64":
binary=get_edit_pupyx64_lin(conf)
binary=get_edit_pupyx64_lin(conf, debug=args.debug)
if not outpath:
outpath="pupyx64.so"
with open(outpath, 'wb') as w:
w.write(binary)
os.chmod(outpath, 0711)
elif args.format=="exe_x64":
binary=get_edit_pupyx64_exe(conf)
binary=get_edit_pupyx64_exe(conf, debug=args.debug)
if not outpath:
outpath="pupyx64.exe"
with open(outpath, 'wb') as w:
w.write(binary)
elif args.format=="dll_x64":
binary=get_edit_pupyx64_dll(conf)
binary=get_edit_pupyx64_dll(conf, debug=args.debug)
if not outpath:
outpath="pupyx64.dll"
with open(outpath, 'wb') as w:
w.write(binary)
elif args.format=="dll_x86":
binary=get_edit_pupyx86_dll(conf)
binary=get_edit_pupyx86_dll(conf, debug=args.debug)
if not outpath:
outpath="pupyx86.dll"
with open(outpath, 'wb') as w:
@ -458,3 +475,4 @@ if __name__=="__main__":
print("LAUNCHER = %s"%repr(args.launcher))
print("LAUNCHER_ARGS = %s"%repr(args.launcher_args))
print("SCRIPTLETS = %s"%args.scriptlet)
print("DEBUG = %s"%args.debug)