Agent: Use `ExploiterWrapper` for loading the Hadoop exploiter

This commit is contained in:
Shreya Malviya 2022-02-24 13:29:24 +05:30
parent eb9adc08c2
commit 4d6869fbf6
1 changed files with 3 additions and 1 deletions

View File

@ -223,7 +223,9 @@ class InfectionMonkey:
exploit_wrapper.wrap(SSHExploiter),
PluginType.EXPLOITER,
)
puppet.load_plugin("HadoopExploiter", HadoopExploiter(), PluginType.EXPLOITER)
puppet.load_plugin(
"HadoopExploiter", exploit_wrapper.wrap(HadoopExploiter), PluginType.EXPLOITER
)
puppet.load_plugin("ransomware", RansomwarePayload(), PluginType.PAYLOAD)