diff --git a/setup.py b/setup.py index 642a1e0..ef2f2e8 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,7 @@ author='xcapri', author_email='N/A', url='https://github.com/xcapri/subdosec', - version='0.7', + version='0.8', package_data={"subdosec_": ["config/*"]}, include_package_data=True, packages=find_packages(), diff --git a/subdosec b/subdosec index 169ee45..f97ca31 100644 --- a/subdosec +++ b/subdosec @@ -194,10 +194,14 @@ def analyze_target(target, mode, apikey, output_scan, host_scan, host_scan_prod, web_data = next(item.get('website_data') for item in match_response if item.get('isMatched')) fingerprint_id = next(item.get('service').get('fid') for item in match_response if item.get('isMatched')) - action = save_local if o else vuln_site - print(f" [VULN] {'[SAVED]' if o else f'{output_scan}{service}'}") - asyncio.run(action(web_data, fingerprint_id, apikey if not o else o, host_scan_prod, mode)) - + msg = " [VULN] [SAVED]" if o else f" [VULN] {output_scan}{service}" + print(msg) + + if o: + asyncio.run(save_local(web_data, service, fingerprint_id, o)) + else: + asyncio.run(vuln_site(web_data, fingerprint_id, apikey, host_scan_prod, mode)) + elif not vuln_only: print(f" [UNDETECT]") asyncio.run(undetect_site(match_response[0], apikey, host_scan_prod, mode)) diff --git a/subdosec_/main.py b/subdosec_/main.py index 5a1b263..29b3b89 100644 --- a/subdosec_/main.py +++ b/subdosec_/main.py @@ -194,9 +194,13 @@ def analyze_target(target, mode, apikey, output_scan, host_scan, host_scan_prod, web_data = next(item.get('website_data') for item in match_response if item.get('isMatched')) fingerprint_id = next(item.get('service').get('fid') for item in match_response if item.get('isMatched')) - action = save_local if o else vuln_site - print(f" [VULN] {'[SAVED]' if o else f'{output_scan}{service}'}") - asyncio.run(action(web_data, fingerprint_id, apikey if not o else o, host_scan_prod, mode)) + msg = " [VULN] [SAVED]" if o else f" [VULN] {output_scan}{service}" + print(msg) + + if o: + asyncio.run(save_local(web_data, service, fingerprint_id, o)) + else: + asyncio.run(vuln_site(web_data, fingerprint_id, apikey, host_scan_prod, mode)) elif not vuln_only: print(f" [UNDETECT]")