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,12 +49,10 @@ end
 
 # Override for puts to enable logging
 def puts(o = '')
-  if @log
+  if $log && o.respond_to?(:gsub)
-    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
-  end
   
   super(o)
 end

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?