Only load files once

Author: wchen-r7

modules/exploits/unix/http/pfsense_clickjacking.rb

@@ -52,29 +52,41 @@ def initialize(info = {})
     )
   end
 
+  def js_file
+    @js ||= lambda {
+      path = File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'cookieconsent.min.js')
+      return File.read(path)
+    }.call
+  end
+
+  def css_file
+    @css ||= lambda {
+      path = File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'cookieconsent.min.css')
+      return File.read(path)
+    }.call
+  end
+
+  def background_file
+    @background ||= lambda {
+      path = File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'background.jpg')
+      return File.read(path)
+    }.call
+  end
+
   def on_request_uri(cli, request)
     print_status("GET #{request.uri} #{request.headers['User-Agent']}")
 
     resp = create_response(200, "OK")
     if request.uri =~ /\.js$/
-      path = ::File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'cookieconsent.min.js')
-      fd = ::File.open(path, "rb")
-      resp.body = fd.read(fd.stat.size)
-      fd.close
+      resp.body = js_file
       resp['Content-Type'] = 'text/javascript'
 
     elsif request.uri =~ /\.css$/
-      path = ::File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'cookieconsent.min.css')
-      fd = ::File.open(path, "rb")
-      resp.body = fd.read(fd.stat.size)
-      fd.close
+      resp.body = css_file
       resp['Content-Type'] = 'text/css'
 
     elsif request.uri =~ /\.jpg$/
-      path = ::File.join(Msf::Config.data_directory, 'exploits', 'pfsense_clickjacking', 'background.jpg')
-      fd = ::File.open(path, "rb")
-      resp.body = fd.read(fd.stat.size)
-      fd.close
+      resp.body = background_file
       resp['Content-Type'] = 'image/jpg'
 
     else