Update modules.py

EntySec · May 18, 2024 · 241b557 · 241b557
1 parent 2aa6c35
commit 241b557
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/hatsploit/lib/modules.py b/hatsploit/lib/modules.py
@@ -343,14 +343,19 @@ def run_module(module: Module) -> None:
         """ Run module.
 
         :param Module module: module object
+        :raises RuntimeError: with trailing error message
         :return None: None
         """
 
-        if hasattr(module, "check"):
-            if module.check():
-                module.run()
-        else:
+        if not hasattr(module, "check"):
             module.run()
+            return
+
+        if not module.check():
+            raise RuntimeWarning(
+                f"{module.details['Category'].title()} module failed! (not vulnerable)")
+
+        module.run()
 
     def entry_to_module(self, module: Module) -> None:
         """ Prepare to entry the module.