diff --git a/lib/common/common_helper.rb b/lib/common/common_helper.rb index 00d781c4..ff718ffa 100644 --- a/lib/common/common_helper.rb +++ b/lib/common/common_helper.rb @@ -78,6 +78,13 @@ else REVISION = nil end +def version + if REVISION + return "v#{WPSCAN_VERSION}r#{REVISION}" + end + return "v#{WPSCAN_VERSION}" +end + # our 1337 banner def banner puts '_______________________________________________________________' @@ -90,9 +97,9 @@ def banner puts puts ' WordPress Security Scanner by the WPScan Team ' if REVISION - puts " Version v#{WPSCAN_VERSION}r#{REVISION}" + puts " Version #{version}" else - puts " Version v#{WPSCAN_VERSION}" + puts " Version #{version}" end puts ' Sponsored by the RandomStorm Open Source Initiative' puts ' @_WPScan_, @ethicalhack3r, @erwan_lr, @gbrindisi, @_FireFart_' diff --git a/lib/wpscan/wpscan_options.rb b/lib/wpscan/wpscan_options.rb index 67e17243..16046e48 100644 --- a/lib/wpscan/wpscan_options.rb +++ b/lib/wpscan/wpscan_options.rb @@ -29,7 +29,8 @@ class WpscanOptions :config_file, :exclude_content_based, :basic_auth, - :debug_output + :debug_output, + :version ] attr_accessor *ACCESSOR_OPTIONS @@ -237,7 +238,8 @@ class WpscanOptions ['--config-file', '-c', GetoptLong::REQUIRED_ARGUMENT], ['--exclude-content-based', GetoptLong::REQUIRED_ARGUMENT], ['--basic-auth', GetoptLong::REQUIRED_ARGUMENT], - ['--debug-output', GetoptLong::NO_ARGUMENT] + ['--debug-output', GetoptLong::NO_ARGUMENT], + ['--version', GetoptLong::NO_ARGUMENT] ) end diff --git a/wpscan.rb b/wpscan.rb index 3943eef2..a1f18ead 100755 --- a/wpscan.rb +++ b/wpscan.rb @@ -24,6 +24,11 @@ def main exit(0) end + if wpscan_options.version + puts "Current version is #{version}" + exit(0) + end + # Check for updates if wpscan_options.update if !@updater.nil?