malectricasoftware
/
balsamic
mirror of https://github.com/malectricasoftware/balsamic.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update __main__.py

main
witchdocsec 2024-05-24 23:22:05 +01:00 committed by GitHub
parent 9aa81f21a4
commit cf6b67652b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
balsamic/__main__.py
View File

@ -46,12 +46,8 @@ if __name__ == "__main__":
balsamic.updatecmd(args.command)
# 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)
balsamic.webreq(args.method, args.url, args.payload, args.parameter, args.cookie, custom_header=args.headers)
elif args.attack == "socksend":
balsamic.socksend(args.rhost, args.rport, args.payload, args.encode, args.steps,args.ipv6)
elif args.attack == "socklisten":