Refactored HostExploiter to have pre_exploit and post_exploit methods

monkey/infection_monkey/exploit/__init__.py

@@ -49,8 +49,14 @@ class HostExploiter(object):
                                        'lm_hash': lm_hash, 'ntlm_hash': ntlm_hash, 'ssh_key': ssh_key})
 
     def exploit_host(self):
-        self.set_start_time()
+        self.pre_exploit()
         self._exploit_host()
+        self.post_exploit()
+
+    def pre_exploit(self):
+        self.set_start_time()
+
+    def post_exploit(self):
         self.set_finish_time()
 
     @abstractmethod