secator 0.10.1a4__tar.gz → 0.10.1a5__tar.gz

{secator-0.10.1a4 → secator-0.10.1a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: secator
-Version: 0.10.1a4
+Version: 0.10.1a5
 Summary: The pentester's swiss knife.
 Project-URL: Homepage, https://github.com/freelabz/secator
 Project-URL: Issues, https://github.com/freelabz/secator/issues

{secator-0.10.1a4 → secator-0.10.1a5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = 'hatchling.build'
 
 [project]
 name = 'secator'
-version = "0.10.1a4"
+version = "0.10.1a5"
 authors = [{ name = 'FreeLabz', email = 'sales@freelabz.com' }]
 readme = 'README.md'
 description = "The pentester's swiss knife."

{secator-0.10.1a4 → secator-0.10.1a5}/secator/celery.py

@@ -260,8 +260,6 @@ def mark_runner_complete(results, runner):
 
 	# Run final processing
 	runner.results = results
-	runner.done = True
-	runner.progress = 100
 	if not runner.no_process:
 		runner.mark_duplicates()
 		runner.results = runner.filter_results()

{secator-0.10.1a4 → secator-0.10.1a5}/secator/report.py

@@ -38,7 +38,7 @@ class Report:
 		exporters (list): List of exporter classes.
 	"""
 	def __init__(self, runner, title=None, exporters=[]):
-		self.title = title or f'{runner.__class__.__name__.lower()}_{runner.config.name}'
+		self.title = title or f'{runner.config.type}_{runner.config.name}'
 		self.runner = runner
 		self.timestamp = get_file_timestamp()
 		self.exporters = exporters

{secator-0.10.1a4 → secator-0.10.1a5}/secator/runners/_base.py

@@ -686,8 +686,6 @@ class Runner:
 
 	def log_results(self):
 		"""Log runner results."""
-		if self.no_poll:
-			return
 		self.started = True
 		self.done = True
 		self.progress = 100

secator-0.10.1a5/secator/scans/__init__.py

@@ -0,0 +1,29 @@
+from secator.cli import ALL_SCANS
+from secator.runners import Scan
+
+
+class DynamicScan(Scan):
+	def __init__(self, config):
+		self.config = config
+
+	def __call__(self, targets, **kwargs):
+		hooks = kwargs.pop('hooks', {})
+		results = kwargs.pop('results', [])
+		context = kwargs.pop('context', {})
+		super().__init__(
+			config=self.config,
+			inputs=targets,
+			results=results,
+			hooks=hooks,
+			context=context,
+			run_opts=kwargs)
+		return self
+
+
+DYNAMIC_SCANS = {}
+for scan in ALL_SCANS:
+	instance = DynamicScan(scan)
+	DYNAMIC_SCANS[scan.name] = instance
+
+globals().update(DYNAMIC_SCANS)
+__all__ = list(DYNAMIC_SCANS)

secator-0.10.1a5/secator/workflows/__init__.py

@@ -0,0 +1,29 @@
+from secator.cli import ALL_WORKFLOWS
+from secator.runners import Workflow
+
+
+class DynamicWorkflow(Workflow):
+	def __init__(self, config):
+		self.config = config
+
+	def __call__(self, targets, **kwargs):
+		hooks = kwargs.pop('hooks', {})
+		results = kwargs.pop('results', [])
+		context = kwargs.pop('context', {})
+		super().__init__(
+			config=self.config,
+			inputs=targets,
+			results=results,
+			hooks=hooks,
+			context=context,
+			run_opts=kwargs)
+		return self
+
+
+DYNAMIC_WORKFLOWS = {}
+for workflow in ALL_WORKFLOWS:
+	instance = DynamicWorkflow(workflow)
+	DYNAMIC_WORKFLOWS[workflow.name] = instance
+
+globals().update(DYNAMIC_WORKFLOWS)
+__all__ = list(DYNAMIC_WORKFLOWS)

{secator-0.10.1a4 → secator-0.10.1a5}/tests/integration/test_celery.py

@@ -10,7 +10,7 @@ from celery import chain, chord
 
 from secator.celery import app, forward_results
 from secator.config import CONFIG
-from secator.utils_test import TEST_TASKS, load_fixture
+from secator.utils_test import TEST_TASKS, TEST_WORKFLOWS,load_fixture
 from secator.runners import Command
 from secator.output_types import Url
 from tests.integration.inputs import INPUTS_SCANS
@@ -24,7 +24,8 @@ OPTS = {
 	'ffuf.wordlist': load_fixture('wordlist', INTEGRATION_DIR, only_path=True),
 }
 URL_TARGETS = INPUTS_SCANS['url']
-URL_TARGETS_RESULTS_COUNT = [12, 1]
+URL_RESULTS_COUNT = [14, 1]
+TAG_RESULTS_COUNT = []
 HOST_TARGETS = INPUTS_SCANS['host']
 
 
@@ -192,4 +193,15 @@ class TestCelery(unittest.TestCase):
 		targets = [r.name for r in results if r._type == 'target']
 		urls = [r.url for r in results if r._type == 'url']
 		self.assertEqual(len(targets), len(URL_TARGETS))
-		self.assertEqual(len(urls), sum(URL_TARGETS_RESULTS_COUNT))
+		self.assertEqual(len(urls), sum(URL_RESULTS_COUNT))
+
+	def test_url_vuln_workflow(self):
+		from secator.workflows import url_vuln
+		workflow = url_vuln([t + '?id=1' for t in URL_TARGETS])
+		workflow = workflow.build_celery_workflow()
+		result = workflow.apply()
+		results = result.get()
+		targets = [r.name for r in results if r._type == 'target']
+		tags = [r.name for r in results if r._type == 'tag']
+		self.assertEqual(len(targets), 16)
+		self.assertEqual(len(tags), 6)

secator-0.10.1a4/secator/scans/__init__.py

@@ -1,28 +0,0 @@
-from secator.cli import ALL_SCANS
-
-
-def generate_class(config):
-	from secator.runners import Scan
-
-	class scan(Scan):
-		def __init__(self, inputs=[], **run_opts):
-			hooks = run_opts.pop('hooks', {})
-			results = run_opts.pop('results', [])
-			context = run_opts.pop('context', {})
-			super().__init__(
-				config=config,
-				inputs=inputs,
-				results=results,
-				run_opts=run_opts,
-				hooks=hooks,
-				context=context)
-	return scan, config.name
-
-
-DYNAMIC_SCANS = {}
-for scan in ALL_SCANS:
-	cls, name = generate_class(scan)
-	DYNAMIC_SCANS[name] = cls
-
-globals().update(DYNAMIC_SCANS)
-__all__ = list(DYNAMIC_SCANS)

secator-0.10.1a4/secator/workflows/__init__.py

@@ -1,28 +0,0 @@
-from secator.cli import ALL_WORKFLOWS
-
-
-def generate_class(config):
-	from secator.runners import Workflow
-
-	class workflow(Workflow):
-		def __init__(self, inputs=[], **run_opts):
-			hooks = run_opts.pop('hooks', {})
-			results = run_opts.pop('results', [])
-			context = run_opts.pop('context', {})
-			super().__init__(
-				config=config,
-				inputs=inputs,
-				results=results,
-				run_opts=run_opts,
-				hooks=hooks,
-				context=context)
-	return workflow, config.name
-
-
-DYNAMIC_WORKFLOWS = {}
-for workflow in ALL_WORKFLOWS:
-	cls, name = generate_class(workflow)
-	DYNAMIC_WORKFLOWS[name] = cls
-
-globals().update(DYNAMIC_WORKFLOWS)
-__all__ = list(DYNAMIC_WORKFLOWS)