Make filename easy to override

rapid7 · Sep 10, 2012 · 221eb88 · 221eb88
1 parent f5a0f74
commit 221eb88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/msf/core/exploit/fileformat.rb b/lib/msf/core/exploit/fileformat.rb
@@ -26,8 +26,12 @@ def initialize(info = {})
 		)
 	end
 
+	def file_format_filename
+		datastore['FILENAME']
+	end
+
 	def file_create(data)
-		fname = datastore['FILENAME']
+		fname = file_format_filename
 		ltype = "exploit.fileformat.#{self.shortname}"
 		full_path = store_local(ltype, nil, data, fname)
 		print_good "#{fname} stored at #{full_path}"