Merge pull request #46 from dmdhrumilmistry/fix-web-crawlers-in-UI

fix dir and subdom crawler options in UI
dmdhrumilmistry · Apr 29, 2023 · 515432b · 515432b
2 parents af268d2 + 5b48765
commit 515432b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pyhtools/UI/functions.py b/pyhtools/UI/functions.py
@@ -167,10 +167,10 @@ async def run():
                 await attacker.webspider()
 
             elif cmd == 'webcrawldirs':
-                attacker.webcrawldirs()
+                await attacker.webcrawldirs()
 
             elif cmd == 'webcrawlsubdom':
-                attacker.webcrawlsubdom()
+                await attacker.webcrawlsubdom()
 
             elif cmd == 'weblogin':
                 attacker.brute_login()

diff --git a/pyhtools/attackers/attackers.py b/pyhtools/attackers/attackers.py
@@ -129,23 +129,23 @@ async def webspider():
     print(f'[*] Total Links Found: {len(discovered_links)}')
 
 
-def webcrawldirs():
+async def webcrawldirs():
     '''
     description: find valid directories of the website using a wordlist
     params: None
     returns: None
     '''
     domain = input('[+] DOMAIN (duckduckgo.com): ')
     wordlist_path = input('[+] WORDLIST PATH: ')
-    run(discoverer.check_dirs(domain=domain, wordlist=wordlist_path))
+    await discoverer.check_dirs(domain=domain, wordlist_path=wordlist_path)
 
 
-def webcrawlsubdom():
+async def webcrawlsubdom():
     '''
     description: find valid subdomains of the website using a wordlist
     params: None
     returns: None
     '''
     domain = input('[+] DOMAIN (duckduckgo.com) : ')
     wordlist_path = input('[+] WORDLIST PATH : ')
-    run(discoverer.check_subdomains(domain=domain, wordlist=wordlist_path))
+    await discoverer.check_subdomains(domain=domain, wordlist_path=wordlist_path)