Update __main__.py
witchdocsec
GitHub
parent
74fdd20c72
commit
b384ac0870
|
|
@ -1,55 +1,56 @@
|
|||
from balsamic import balsamic
|
||||
import argparse
|
||||
if __name__ == "__main__":
|
||||
#import argparse
|
||||
import argparse
|
||||
# Importing argparse
|
||||
import argparse
|
||||
|
||||
#defines parser function
|
||||
def parser():
|
||||
|
||||
#create base argument parser
|
||||
parser = argparse.ArgumentParser(description="balsamic args")
|
||||
subparse=parser.add_subparsers(dest="attack")
|
||||
# Define the parser function
|
||||
def parser():
|
||||
|
||||
# Create the base argument parser
|
||||
parser = argparse.ArgumentParser(description="balsamic args")
|
||||
subparse = parser.add_subparsers(dest="attack")
|
||||
|
||||
#create subparser for webreq attack
|
||||
webreqparser=subparse.add_parser("webreq")
|
||||
webreqparser.add_argument("-s","--schema",required=True, choices=["http","https"])
|
||||
webreqparser.add_argument("-m","--method")
|
||||
webreqparser.add_argument("-rh","--rhost",required=True)
|
||||
webreqparser.add_argument("-rp","--rport",required=True)
|
||||
webreqparser.add_argument("-p","--parameter")
|
||||
webreqparser.add_argument("-co","--cookie")
|
||||
webreqparser.add_argument("-P","--payload",required=True)
|
||||
webreqparser.add_argument("-c","--command")
|
||||
# Create subparsers for different attacks
|
||||
webreqparser = subparse.add_parser("webreq")
|
||||
webreqparser.add_argument("-m", "--method")
|
||||
webreqparser.add_argument("-u", "--url", required=True)
|
||||
webreqparser.add_argument("-p", "--parameter")
|
||||
webreqparser.add_argument("-co", "--cookie")
|
||||
webreqparser.add_argument("-P", "--payload", required=True)
|
||||
webreqparser.add_argument("-c", "--command")
|
||||
webreqparser.add_argument("-H", "--headers") # New argument for custom headers
|
||||
|
||||
#create subparser for socksend attack
|
||||
socksendparser=subparse.add_parser("socksend")
|
||||
socksendparser.add_argument("-rh","--rhost",required=True)
|
||||
socksendparser.add_argument("-rp","--rport",required=True)
|
||||
socksendparser.add_argument("-P","--payload",required=True)
|
||||
socksendparser.add_argument("-c","--command")
|
||||
socksendparser.add_argument("-s","--steps",default="0")
|
||||
socksendparser.add_argument("-e","--encode",action="store_true")
|
||||
socksendparser = subparse.add_parser("socksend")
|
||||
socksendparser.add_argument("-rh", "--rhost", required=True)
|
||||
socksendparser.add_argument("-rp", "--rport", required=True)
|
||||
socksendparser.add_argument("-P", "--payload", required=True)
|
||||
socksendparser.add_argument("-c", "--command")
|
||||
socksendparser.add_argument("-s", "--steps", default="0")
|
||||
socksendparser.add_argument("-e", "--encode", action="store_true")
|
||||
|
||||
#create subparser for socklisten attack
|
||||
socklistenparser=subparse.add_parser("socklisten")
|
||||
socklistenparser.add_argument("-lp","--lport",required=True)
|
||||
socklistenparser.add_argument("-P","--payload",required=True)
|
||||
socklistenparser.add_argument("-c","--command")
|
||||
socklistenparser.add_argument("-s","--steps",default="0")
|
||||
socklistenparser.add_argument("-e","--encode",action="store_true")
|
||||
|
||||
#return parsed arguments
|
||||
args = parser.parse_args()
|
||||
return args
|
||||
socklistenparser = subparse.add_parser("socklisten")
|
||||
socklistenparser.add_argument("-lp", "--lport", required=True)
|
||||
socklistenparser.add_argument("-P", "--payload", required=True)
|
||||
socklistenparser.add_argument("-c", "--command")
|
||||
socklistenparser.add_argument("-s", "--steps", default="0")
|
||||
socklistenparser.add_argument("-e", "--encode", action="store_true")
|
||||
|
||||
# Return parsed arguments
|
||||
args = parser.parse_args()
|
||||
return args
|
||||
|
||||
args=parser()
|
||||
if args.command:
|
||||
balsamic.updatecmd(args.command)
|
||||
args = parser()
|
||||
if args.command:
|
||||
balsamic.updatecmd(args.command)
|
||||
|
||||
if args.attack == "webreq":
|
||||
balsamic.webreq(args.schema,args.method,args.rhost,args.rport,args.payload,args.parameter,args.cookie)
|
||||
elif args.attack == "socksend":
|
||||
balsamic.socksend(args.rhost,args.rport,args.payload,args.encode,args.steps)
|
||||
elif args.attack == "socklisten":
|
||||
balsamic.socklisten(args.lport,args.payload,args.encode,args.steps)
|
||||
# Extract custom headers from the arguments
|
||||
try:
|
||||
custom_headers = dict([header.split(":") for header in args.headers.split(";")]) if args.headers else None
|
||||
except AttributeError:
|
||||
pass
|
||||
if args.attack == "webreq":
|
||||
balsamic.webreq(args.method, args.url, args.payload, args.parameter, args.cookie, custom_headers=custom_headers)
|
||||
elif args.attack == "socksend":
|
||||
balsamic.socksend(args.rhost, args.rport, args.payload, args.encode, args.steps)
|
||||
elif args.attack == "socklisten":
|
||||
balsamic.socklisten(args.lport, args.payload, args.encode, args.steps)
|
||||
|
|
|
|||
Loading…
Reference in New Issue