From 6d2a7710844c4c1bddc8897ed3ba129612d0da19 Mon Sep 17 00:00:00 2001
From: soh_cah_toa <soh_cah_toa@cybercrud.net>
Date: Wed, 15 May 2013 22:29:42 -0400
Subject: [PATCH] Changed model name to BeEF::Core::Models::DNS::Rule. This is
 more descriptive and follows the singular name convention.

---
 extensions/dns/model.rb        | 6 ++++--
 extensions/dns/ruby/rubydns.rb | 8 ++++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/extensions/dns/model.rb b/extensions/dns/model.rb
index 41068c4da..2a03f2fc8 100644
--- a/extensions/dns/model.rb
+++ b/extensions/dns/model.rb
@@ -6,12 +6,13 @@
 module BeEF
 module Core
 module Models
+module DNS
 
-  class DNS
+  class Rule
 
     include DataMapper::Resource
 
-    storage_names[:default] = 'extensions_dns'
+    storage_names[:default] = 'extension_dns_rules'
 
     property :id, Serial
     property :pattern, Object
@@ -22,3 +23,4 @@ module Models
 end
 end
 end
+end
diff --git a/extensions/dns/ruby/rubydns.rb b/extensions/dns/ruby/rubydns.rb
index 899aeb17d..3dbc837e8 100644
--- a/extensions/dns/ruby/rubydns.rb
+++ b/extensions/dns/ruby/rubydns.rb
@@ -48,12 +48,12 @@ module RubyDNS
     def match(id, *pattern, block)
       catch :match do
         # Check if rule is already present
-        BeEF::Core::Models::DNS.each { |rule| throw :match if rule.id == id }
+        BeEF::Core::Models::DNS::Rule.each { |rule| throw :match if rule.id == id }
 
         @rules << Rule.new(id, pattern, block)
 
         # Add new rule to database
-        BeEF::Core::Models::DNS.create(
+        BeEF::Core::Models::DNS::Rule.create(
           :id => id,
           :pattern => pattern,
           :block => block.to_source
@@ -66,7 +66,7 @@ module RubyDNS
       @rules.delete_if { |rule| rule.id == id }
 
       begin
-        BeEF::Core::Models::DNS.get!(id).destroy
+        BeEF::Core::Models::DNS::Rule.get!(id).destroy
       rescue DataMapper::ObjectNotFoundError => e
         @logger.error(e.message)
       end
@@ -74,7 +74,7 @@ module RubyDNS
 
     # New method that loads all rules from the database at server startup
     def load_rules
-      BeEF::Core::Models::DNS.each do |rule|
+      BeEF::Core::Models::DNS::Rule.each do |rule|
         id = rule.id
         pattern = rule.pattern
         block = eval rule.block