From e6b74d51860c79e6943090da7d879295b2fc31ab Mon Sep 17 00:00:00 2001 From: Brendan Coles Date: Sun, 27 Apr 2014 00:14:25 +1000 Subject: [PATCH] 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. --- extensions/dns/model.rb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions/dns/model.rb b/extensions/dns/model.rb index bd2610a4f..d12c2fb15 100644 --- a/extensions/dns/model.rb +++ b/extensions/dns/model.rb @@ -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