garfield/wpscan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use constants for data files

Browse Source
This commit is contained in:
Christian Mehlmauer
2013-01-19 21:33:07 +01:00
parent 5d6ce70863
commit 14be8c700f
9 changed files with 24 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/wpstools/plugins/checker/checker_plugin.rb
View File

@@ -39,14 +39,14 @@ class CheckerPlugin < Plugin
end
def check_vuln_ref_urls
vuln_ref_files = ["plugin_vulns.xml", "theme_vulns.xml", "wp_vulns.xml"]
vuln_ref_files = [ PLUGINS_VULNS_FILE , THEMES_VULNS_FILE, WP_VULNS_FILE ]
error_codes = [404, 500, 403]
not_found_regexp = %r{No Results Found|error 404|ID Invalid or Not Found}i
puts "[+] Checking vulnerabilities reference urls"
vuln_ref_files.each do |vuln_ref_file|
xml = Nokogiri::XML(File.open(DATA_DIR + '/' + vuln_ref_file)) do |config|
xml = Nokogiri::XML(File.open(vuln_ref_file)) do |config|
config.noblanks
end

8
lib/wpstools/plugins/list_generator/generate_list.rb
View File

@@ -48,18 +48,18 @@ class GenerateList
when "plugin"
case type
when :full
@file_name = DATA_DIR + "/plugins_full.txt"
@file_name = PLUGINS_FULL_FILE
when :popular
@file_name = DATA_DIR + "/plugins.txt"
@file_name = PLUGINS_FILE
else
raise "Unknown type"
end
when "theme"
case type
when :full
@file_name = DATA_DIR + "/themes_full.txt"
@file_name = THEMES_FULL_FILE
when :popular
@file_name = DATA_DIR + "/themes.txt"
@file_name = THEMES_FILE
else
raise "Unknown type"
end