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

Adds the HttpError exception - Fixes #792

Browse Source
This commit is contained in:
erwanlr
2015-04-03 16:22:28 +01:00
parent 5b94714ca7
commit 413ee7a6d3
2 changed files with 29 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/common/db_updater.rb
View File

@@ -36,8 +36,7 @@ class DbUpdater
url = "#{remote_file_url(filename)}.sha512"
res = Browser.get(url, request_params)
additional_text = res.code == 0 ? res.return_message : "return code #{res.code}"
fail "Unable to get #{url} (#{additional_text})" unless res.code == 200
fail DownloadError, res unless res.code == 200
res.body
end
@@ -73,8 +72,7 @@ class DbUpdater
file_url = remote_file_url(filename)
res = Browser.get(file_url, request_params)
additional_text = res.code == 0 ? res.return_message : "return code #{res.code}"
fail "Error while downloading #{file_url} (#{additional_text})" unless res.code == 200
fail DownloadError, res unless res.code == 200
File.open(file_path, 'wb') { |f| f.write(res.body) }
local_file_checksum(filename)