Refactor #703 #336

lib/common/common_helper.rb

@@ -243,3 +243,11 @@ end
 def directory_listing_enabled?(url)
   Browser.get(url.to_s).body[%r{<title>Index of}] ? true : false
 end
+
+def log
+  @log
+end
+
+def log=(log)
+  @log = log
+end

lib/common/hacks.rb

@@ -49,11 +49,9 @@ end
 
 # Override for puts to enable logging
 def puts(o = '')
-  if @log
-    if o.respond_to?(:gsub)
-      temp = o.gsub(/\e\[\d+m/, '') # remove color for logging
-      File.open(LOG_FILE, 'a+') { |f| f.puts(temp) }
-    end
+  if $log && o.respond_to?(:gsub)
+    temp = o.gsub(/\e\[\d+m/, '') # remove color for logging
+    File.open(LOG_FILE, 'a+') { |f| f.puts(temp) }
   end
   
   super(o)

wpscan.rb

@@ -11,9 +11,9 @@ def main
   begin
     wpscan_options = WpscanOptions.load_from_arguments
 
-    @log = wpscan_options.log
+    $log = wpscan_options.log
 
-    banner() # called after @log instance variable set
+    banner() # called after $log set
 
     unless wpscan_options.has_options?
       # first parameter only url?