Fix for missing API key

target · Mar 29, 2024 · c4873a1 · c4873a1
1 parent 773c11e
commit c4873a1
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/app/blueprints/strelka.py b/app/blueprints/strelka.py
@@ -179,6 +179,9 @@ def submit_file(
                         file_hash=submitted_hash,
                     )
 
+                # Assume file is a BytesIO object and set a filename attribute
+                file.filename = submitted_hash
+
             except Exception as e:
                 return (
                     jsonify(
@@ -189,9 +192,16 @@ def submit_file(
                     ),
                     400,
                 )
-
-            # Assume file is a BytesIO object and set a filename attribute
-            file.filename = submitted_hash
+        else:
+            return (
+                jsonify(
+                    {
+                        "error": "VirusTotal request was not successful.",
+                        "details": "No VirusTotal API key has been loaded. Check the README for details.",
+                    }
+                ),
+                400,
+            )
 
     # Submit the file to the Strelka analysis engine.
     if file:

diff --git a/app/services/virustotal.py b/app/services/virustotal.py
@@ -43,7 +43,7 @@ def create_vt_zip_and_download(api_key, file_hash, password, max_attempts=3):
 
     Args:
         api_key (str): The API key for VirusTotal.
-        file_hash (str): The hash of the file to download.
+        file_hash (list): The hash of the file to download.
         password (str): The password to set for the ZIP file.
         max_attempts (int): The maximum number of attempts to check for ZIP readiness.