Merge branch 'main' into issue-OWASP-BLT#1338

JisanAR03 · Oct 28, 2023 · b800702 · b800702
2 parents b3adda8 + 22ddc4a
commit b800702
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/website/templates/report.html b/website/templates/report.html
@@ -358,13 +358,17 @@ <h2 class="text-2xl font-semibold leading-7 text-gray-900">
 
     function previewFile(file_name) {
         event.preventDefault();
+        if (!screenshots || !(screenshots instanceof HTMLInputElement) || !screenshots.files || !screenshots.files.length) return;
         Array.from(screenshots.files).map(file => {
-            if (file.name == file_name) {
-                let src = URL.createObjectURL(file);
-                $("#image-preview").attr("src", src);
+            if (file.name === file_name && file.type.startsWith('image/')) {
+                let src = URL.createObjectURL(file);                
+                // Set the src attribute for the image preview
+                $("#image-preview").attr("src", src).on('load', function() {
+                    URL.revokeObjectURL(src);  // Revoke the blob URL after it's used
+                });
                 $("#image-preview-wrapper").css("display", "flex");
             }
-        })
+        });
     }
 
     $("#image-preview-wrapper").on("click", () => {