Merge pull request #101 from login-securite/3.1.12-patch1

Fix #100

Author: Hackndo

.pre-commit-config.yaml

@@ -0,0 +1,8 @@
+repos:
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v3.2.0
+    hooks:
+    -   id: trailing-whitespace
+    -   id: end-of-file-fixer
+    -   id: check-yaml
+    -   id: check-added-large-files

lsassy/dumpmethod/__init__.py

@@ -37,15 +37,15 @@ def __init__(self, name, file=None, content=None):
         self.uploaded = False
         self.content = content
         self.share_mode = False
-        
+
 
     def get_remote_path(self):
         return self.remote_path + self.file
 
     def init(self, options):
         if self.content is not None:
             return True
-        
+
         self.path = options.get("{}_path".format(self.name), self.path)
 
         if not self.path:
@@ -121,7 +121,7 @@ def __init__(self, session, timeout, time_between_commands, *args, **kwargs):
         self._executor_copied = False
         self._timeout = timeout
         self._time_between_commands = time_between_commands
-        
+
 
     def get_exec_method(self, exec_method, no_powershell=False):
         try:
@@ -226,6 +226,10 @@ def dump(self, dump_path=None, dump_name=None, no_powershell=False, copy=False,
             if not self.custom_dump_name_support:
                 lsassy_logger.warning("A custom dump name was provided, but dump method {} doesn't support custom dump name".format(self.__module__))
                 lsassy_logger.warning("Dump file will be {}".format(self.dump_name))
+            elif not self.custom_dump_ext_support:
+                lsassy_logger.warning("A custom dump name was provided, but dump method {} doesn't support custom extension".format(self.__module__))
+                lsassy_logger.warning("Dump file will be {}.{}".format(dump_name, self.dump_ext))
+                self.dump_name = f"{dump_name}.{self.dump_ext}"
             else:
                 self.dump_name = dump_name
         elif self.dump_name == "":
@@ -303,7 +307,7 @@ def dump(self, dump_path=None, dump_name=None, no_powershell=False, copy=False,
                 continue
             lsassy_logger.info("Lsass dumped in C:{}{} ({} Bytes)".format(self.dump_path, self.dump_name, self._file_handle.size()))
             return self._file_handle
-            
+
         lsassy_logger.error("All execution methods have failed")
         self.clean()
         return None