Add a check for Custom404

wchen-r7 · wchen-r7 · commit acf02647fb98 · 2015-01-27T20:18:10.000-06:00
diff --git a/lib/msf/core/exploit/remote/browser_exploit_server.rb b/lib/msf/core/exploit/remote/browser_exploit_server.rb
@@ -95,6 +95,14 @@ def initialize(info={})
       ], Exploit::Remote::BrowserExploitServer)
     end
 
+    def setup
+      custom_404 = get_custom_404_url
+      if !custom_404.blank? && custom_404 !~ /^http/i
+        raise Msf::OptionValidateError.new(['Custom404 (must begin with http or https)'])
+      end
+      super
+    end
+
     #
     # Returns the custom 404 URL set by the user
     #

Original file line number	Diff line number	Diff line change
`@@ -95,6 +95,14 @@ def initialize(info={})`
`95`	`95`	`], Exploit::Remote::BrowserExploitServer)`
`96`	`96`	`end`
`97`	`97`
	`98`	`+ def setup`
	`99`	`+ custom_404 = get_custom_404_url`
	`100`	`+ if !custom_404.blank? && custom_404 !~ /^http/i`
	`101`	`+ raise Msf::OptionValidateError.new(['Custom404 (must begin with http or https)'])`
	`102`	`+ end`
	`103`	`+ super`
	`104`	`+ end`
	`105`	`+`
`98`	`106`	`#`
`99`	`107`	`# Returns the custom 404 URL set by the user`
`100`	`108`	`#`