Caching is controlled via a command line option

leonluffy · Dec 21, 2016 · f6dfdf1 · f6dfdf1
1 parent 6b2bc35
commit f6dfdf1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/server/controller/controller_commands.rb b/server/controller/controller_commands.rb
@@ -196,7 +196,7 @@ def _register_commands()
         TunnelDrivers.start({
           :controller => self,
           :driver     => DriverDNS,
-          :args       => [dns[:host], dns[:port], dns[:domains]]
+          :args       => [dns[:host], dns[:port], dns[:domains], opts[:cache]]
         })
       end
     )

diff --git a/server/dnscat2.rb b/server/dnscat2.rb
@@ -88,6 +88,9 @@
 
   opt :firehose,       "If set, all output goes to stdout instead of being put in windows.",
     :type => :boolean, :default => false
+
+  opt :cache,       "If set, caching is enabled on the server.",
+    :type => :boolean, :default => true
 end
 
 SWindow.set_firehose(opts[:firehose])
@@ -205,7 +208,7 @@
 TunnelDrivers.start({
   :controller => controller,
   :driver     => DriverDNS,
-  :args       => [dns_settings[:host], dns_settings[:port], dns_settings[:domains]],
+  :args       => [dns_settings[:host], dns_settings[:port], dns_settings[:domains], opts[:cache]],
 })
 
 # Wait for the input window to finish its thing

diff --git a/server/tunnel_drivers/driver_dns.rb b/server/tunnel_drivers/driver_dns.rb
@@ -242,13 +242,13 @@ def DriverDNS.do_encoding(question, domains, response)
     return response
   end
 
-  def initialize(parent_window, host, port, domains)
+  def initialize(parent_window, host, port, domains, cache)
     if(domains.nil?)
       domains = []
     end
 
     # Do this as early as we can, so we can fail early
-    @dnser = DNSer.new(host, port, true)
+    @dnser = DNSer.new(host, port, cache)
 
     @id = 'dns%d' % (@@id += 1)
     @window = SWindow.new(parent_window, false, {