added FullScan wrapper to loop in dangling execution paths

This commit is contained in:
epi052
2019-10-18 14:07:55 -05:00
parent 5e8d6668f5
commit a49e4f9619

28
recon/wrappers.py Normal file
View File

@@ -0,0 +1,28 @@
import luigi
from luigi.util import inherits
from recon.nmap import Searchsploit
from recon.web.aquatone import AquatoneScan
@inherits(Searchsploit, AquatoneScan)
class FullScan(luigi.WrapperTask):
""" Wraps multiple scan types in order to run tasks on the same hierarchical level at the same time. """
def requires(self):
""" FullScan is a wrapper, as such it requires any Tasks that it wraps. """
args = {
"rate": self.rate,
"target_file": self.target_file,
"top_ports": self.top_ports,
"interface": self.interface,
"ports": self.ports,
"exempt_list": self.exempt_list,
"threads": self.threads,
"scan_timeout": self.scan_timeout,
}
yield AquatoneScan(**args)
del args["scan_timeout"]
yield Searchsploit(**args)