Fix getting filenames in analyzers (#140)

TheHive-Project · Dec 13, 2017 · 1c4d3ec · 1c4d3ec
1 parent f6d836c
commit 1c4d3ec
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/analyzers/File_Info/fileinfo_analyzer.py b/analyzers/File_Info/fileinfo_analyzer.py
@@ -14,7 +14,7 @@ class FileAnalyzer(Analyzer):
     def __init__(self):
         Analyzer.__init__(self)
 
-        self.filename = self.getParam('attachment.name', 'noname.ext')
+        self.filename = self.getParam('filename', 'noname.ext')
         self.filepath = self.getParam('file', None, 'File is missing')
 
 

diff --git a/analyzers/MsgParser/parse.py b/analyzers/MsgParser/parse.py
@@ -13,7 +13,7 @@ class MsgParserAnalyzer(Analyzer):
     def __init__(self):
         Analyzer.__init__(self)
 
-        self.filename = self.getParam('attachment.name', 'noname.ext')
+        self.filename = self.getParam('filename', 'noname.ext')
         self.filepath = self.getParam('file', None, 'File is missing')
 
     def summary(self, raw):

diff --git a/analyzers/VirusTotal/virustotal.py b/analyzers/VirusTotal/virustotal.py
@@ -138,7 +138,7 @@ def run(self):
 
         if self.service == 'scan':
             if self.data_type == 'file':
-                filename = self.getParam('attachment.name', 'noname.ext')
+                filename = self.getParam('filename', 'noname.ext')
                 filepath = self.getParam('file', None, 'File is missing')
                 self.read_scan_response(self.vt.scan_file(
                     (filename, open(filepath, 'rb'))), self.wait_file_report)