Update model.rb - Throw 'UnknownDnsResourceError'

Throw `UnknownDnsResourceError` instead of `InvalidDnsResourceError`

Prevents `[20:30:55][!] Internal error while adding DNS rule (uninitialized constant BeEF::Core::Models::Dns::Rule::InvalidDnsResourceError)` for invalid user supplied DNS response types.

'BeEF::Core::Models::Dns::Rule::InvalidDnsResourceError' does not exist, and it's unlikely we'll need to differentiate between invalid and unknown resource types.
This commit is contained in:
Brendan Coles
2014-04-27 00:14:25 +10:00
parent 3b3d7fe95e
commit e6b74d5186

View File

@@ -25,7 +25,7 @@ module BeEF
begin
validate_pattern(rule.pattern)
rule.callback = format_callback(rule.resource, rule.response)
rescue InvalidDnsPatternError, InvalidDnsResourceError, InvalidDnsResponseError => e
rescue InvalidDnsPatternError, UnknownDnsResourceError, InvalidDnsResponseError => e
print_error e.message
throw :halt
end