screenshot plugin creates 2 files (fixes #494)

auooo · Oct 12, 2016 · a589195 · a589195
1 parent f7cf258
commit a589195
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/plugins/scripts/screenshot.py b/plugins/scripts/screenshot.py
@@ -4,6 +4,7 @@
 import subprocess
 import tempfile
 import pipes
+import os
 
 @moduleman_plugin
 class screenshot(BasePlugin):
@@ -16,7 +17,10 @@ def validate(self, fuzzresult):
 	return fuzzresult.code not in [404]
 
     def process(self, fuzzresult):
-	(fd, filename) = tempfile.mkstemp()
+        temp_name = next(tempfile._get_candidate_names())
+        defult_tmp_dir = tempfile._get_default_tempdir()
 
-	subprocess.call(['cutycapt', '--url=%s' % pipes.quote(fuzzresult.url), '--out=%s.png' % filename])
-	self.add_result("Screnshot taken, output at %s.png" % filename)
+        filename = os.path.join(defult_tmp_dir, temp_name + ".png")
+
+	subprocess.call(['cutycapt', '--url=%s' % pipes.quote(fuzzresult.url), '--out=%s' % filename])
+	self.add_result("Screnshot taken, output at %s" % filename)