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

Check for user-agents.txt before using it

Browse Source
This commit is contained in:
g0tmi1k
2018-05-09 13:15:12 +01:00
parent 2c40913a64
commit 435fb34233
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lib/common/common_helper.rb
View File

@@ -255,6 +255,11 @@ end
# @return [ String ] A random user-agent from data/user-agents.txt # @return [ String ] A random user-agent from data/user-agents.txt
def get_random_user_agent def get_random_user_agent
user_agents = [] user_agents = []
unless File.exist?(USER_AGENTS_FILE)
raise('[ERROR] Missing user-agent data. Please re-run with --update.')
end
f = File.open(USER_AGENTS_FILE, 'r') f = File.open(USER_AGENTS_FILE, 'r')
f.each_line do |line| f.each_line do |line|
# ignore comments # ignore comments