From e56494d4860a57af74854cb31cc93b239f386349 Mon Sep 17 00:00:00 2001
From: soh_cah_toa <soh_cah_toa@cybercrud.net>
Date: Wed, 5 Jun 2013 16:30:24 -0400
Subject: [PATCH] Renamed /rules GET route to /ruleset for the sake of
 consistency. Also added new "count" key to result that lists the number of
 rules.

---
 extensions/dns/rest/dns.rb | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/extensions/dns/rest/dns.rb b/extensions/dns/rest/dns.rb
index 5fde2d05f..d603df58e 100644
--- a/extensions/dns/rest/dns.rb
+++ b/extensions/dns/rest/dns.rb
@@ -25,10 +25,14 @@ module Dns
     end
 
     # Returns the entire current DNS ruleset
-    get '/rules' do
+    get '/ruleset' do
       begin
+        ruleset = BeEF::Extension::Dns::Server.instance.get_ruleset
+        count = ruleset.length
+
         result = {}
-        result[:rules] = BeEF::Extension::Dns::Server.instance.get_ruleset
+        result[:count] = count
+        result[:ruleset] = ruleset
         result.to_json
       rescue StandardError => e
         print_error "Internal error while retrieving DNS ruleset (#{e.message})"