From ed986e4ed5c16053d45a3bc4e738cd87946bce57 Mon Sep 17 00:00:00 2001
From: soh_cah_toa <soh_cah_toa@cybercrud.net>
Date: Wed, 23 Apr 2014 12:07:59 -0400
Subject: [PATCH] Implemented #remove_rule method.

---
 extensions/dns/dns.rb | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/extensions/dns/dns.rb b/extensions/dns/dns.rb
index c562dc603..2cf90e66d 100644
--- a/extensions/dns/dns.rb
+++ b/extensions/dns/dns.rb
@@ -76,6 +76,18 @@ module BeEF
           end
         end
 
+        # Removes the given DNS rule.
+        #
+        # @param id [String] rule identifier
+        #
+        # @return [Boolean] true if rule was removed, otherwise false
+        def remove_rule(id)
+          @lock.synchronize do
+            rule = BeEF::Core::Models::Dns::Rule.get(id)
+            rule.nil? ? false : rule.destroy
+          end
+        end
+
         # Entry point for processing incoming DNS requests. Attempts to find a matching rule and
         # sends back its associated response.
         #