vagrant-rubydns 0.0.6 → 0.1.0

data/README.md

@@ -1,9 +1,10 @@
 # Vagrant RubyDNS Plugin
 
+Simple DNS that's visible on both the guest and the host.
+
 Spins up a small RubyDNS server that you can point your VMs at, automatically
 registers/deregisters IP addresseses of guests as they come up and down.
 
-**Goal**: simple DNS that's visible on both the guest and the host
 
 ## Installation
 
@@ -13,10 +14,6 @@ Install under Vagrant (1.1 or later):
 
 ## Usage
 
-Spin up the DNS server
-
-    $ vagrant rubydns
-
 Enable the plugin in your `Vagrantfile`:
 
     config.rubydns.enable

data/examples/Vagrantfile

@@ -16,7 +16,6 @@ Vagrant.configure("2") do |config|
   config.vm.box = "precise64"
 
   config.rubydns.enable
-  config.vm.provision :rubydns
 
   config.vm.network :private_network, ip: '172.16.32.111'
 

data/lib/vagrant-rubydns.rb

@@ -4,9 +4,19 @@ rescue LoadError
   raise 'The Vagrant RubyDNS plugin must be run within Vagrant.'
 end
 
+module VagrantRubydns
+  def self.working_dir
+    @working_dir ||= Pathname(File.expand_path('~/.vagrant_rubydns')).tap(&:mkpath)
+  end
+
+  def self.working_dir=(working_dir)
+    @working_dir = Pathname(working_dir).tap(&:mkpath)
+  end
+end
+
 require 'vagrant-rubydns/plugin'
 require 'vagrant-rubydns/store'
+require 'vagrant-rubydns/dependent_vms'
+require 'vagrant-rubydns/server'
 require 'vagrant-rubydns/util'
 require 'vagrant-rubydns/version'
-
-module VagrantRubydns; end

data/lib/vagrant-rubydns/action/redirect_dns.rb

@@ -0,0 +1,35 @@
+module VagrantRubydns
+  module Action
+    class RedirectDns
+      def initialize(app, env)
+        @app = app
+      end
+
+      def call(env)
+        @machine = env[:machine]
+
+        @machine.ui.info "setting up machine's DNS to point to our server"
+
+        redirect_dns('10.0.2.3', 53, '10.0.2.2', 10053)
+      end
+
+      def redirect_dns(original_server, original_port, target_server, target_port)
+        %w[tcp udp].each do |protocol|
+          rule = "OUTPUT -t nat -d #{original_server} -p #{protocol} --dport #{original_port} -j DNAT --to-destination #{target_server}:#{target_port}"
+          command = %Q(iptables -C #{rule} 2> /dev/null || iptables -A #{rule})
+          _run_command(command)
+        end
+      end
+
+      def _run_command(command)
+        @machine.communicate.sudo(command) do |data, type|
+          if [:stderr, :stdout].include?(type)
+            color = (type == :stdout) ? :green : :red
+            @machine.env.ui.info(data.chomp, :color => color, :prefix => false)
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/vagrant-rubydns/action/setup.rb

@@ -7,22 +7,34 @@ module VagrantRubydns
 
       def call(env)
         if env[:global_config].rubydns.enabled?
+          DependentVMs.add(env[:machine])
+          start_server_if_necessary(env)
           setup_machine_dns(env)
           setup_static_dns(env)
+          env[:machine].config.vm.provision :rubydns
         end
         @app.call(env)
       end
 
+      def start_server_if_necessary(env)
+        if Server.running?
+          env[:ui].info "[rubydns] dns server already running"
+        else
+          env[:ui].info "[rubydns] starting dns server"
+          Server.start
+        end
+      end
+
       def setup_machine_dns(env)
         hostname, ip_address = Util.host_and_ip(env[:machine])
         env[:ui].info "[rubydns] adding machine entry: #{hostname} => #{ip_address}"
-        Store.set(hostname, ip_address)
+        Store.hosts.set(hostname, ip_address)
       end
 
       def setup_static_dns(env)
         env[:global_config].rubydns.hosts.each do |hostname, ip_address|
           env[:ui].info "[rubydns] adding static entry: #{hostname} => #{ip_address}"
-          Store.set hostname, ip_address
+          Store.hosts.set hostname, ip_address
         end
       end
     end

data/lib/vagrant-rubydns/action/teardown.rb

@@ -9,20 +9,36 @@ module VagrantRubydns
         if env[:global_config].rubydns.enabled?
           teardown_static_dns(env)
           teardown_machine_dns(env)
+
+          DependentVMs.remove(env[:machine])
+          stop_server_if_necessary(env)
         end
         @app.call(env)
       end
 
+      def stop_server_if_necessary(env)
+        if Server.running?
+          if DependentVMs.none?
+            env[:ui].info "[rubydns] no dependent vms left, stopping dns server"
+            Server.stop
+          else
+            env[:ui].info "[rubydns] there are dependent vms left, leaving dns server"
+          end
+        else
+          env[:ui].info "[rubydns] dns server already stopped"
+        end
+      end
+
       def teardown_machine_dns(env)
         hostname = Util.hostname(env[:machine])
         env[:ui].info "[rubydns] removing machine entry: #{hostname}"
-        Store.delete(hostname)
+        Store.hosts.delete(hostname)
       end
 
       def teardown_static_dns(env)
         env[:global_config].rubydns.hosts.each do |hostname, _|
           env[:ui].info "[rubydns] removing static entry: #{hostname}"
-          Store.delete hostname
+          Store.hosts.delete hostname
         end
       end
     end

data/lib/vagrant-rubydns/command.rb

@@ -1,9 +1,42 @@
 module VagrantRubydns
   class Command < Vagrant.plugin('2', :command)
+    DAEMON_COMMANDS = %w(start stop restart status)
+    DEPENDENT_VM_COMMADNS = %w(dependentvms)
+
     def execute
-      require_relative 'server'
-      Server.run
-      0
+      ARGV.shift # flush rubydns from ARGV, RExec wants to use it for daemon commands
+
+      command = ARGV.first
+      if DAEMON_COMMANDS.include?(command)
+        Server.daemonize
+      elsif command == 'dependentvms'
+        if DependentVMs.any?
+          @env.ui.info(DependentVMs.list.map { |dvm| " - #{dvm}" }.join("\n"))
+        else
+          @env.ui.info("No dependent VMs")
+        end
+      else
+        boom("'#{command}' is not a command")
+      end
+
+      0 # happy exit code
+    end
+
+    def boom(msg)
+      raise Vagrant::Errors::CLIInvalidOptions, :help => usage(msg)
+    end
+
+    def usage(msg); <<-EOS.gsub(/^      /, '')
+      ERROR: #{msg}
+
+      vagrant rubydns <command>
+
+      commands:
+        {start|stop|restart|status}
+          control the rubydns server daemon
+        dependentvms
+          list vms currently dependent on the rubydns server
+      EOS
     end
   end
 end

data/lib/vagrant-rubydns/dependent_vms.rb

@@ -0,0 +1,40 @@
+require 'fileutils'
+
+#
+# Keep track of dependent VMs.
+#
+# Poor man's race condition defense - touch and rm files in a directory and count them.
+#
+module VagrantRubydns
+  class DependentVMs
+    extend Enumerable
+
+    def self.each(&block)
+      (dir.directory? ? dir.children : []).each(&block)
+    end
+
+    def self.add(machine)
+      FileUtils.touch(file_for(machine))
+    end
+
+    def self.remove(machine)
+      file_for(machine).tap { |f| f.delete if f.exist? }
+    end
+
+    def self.list
+      self.map { |path| path.basename.to_s }
+    end
+
+    def self.clear!
+      dir.rmtree
+    end
+
+    def self.file_for(machine)
+      dir.join(Util.hostname(machine))
+    end
+
+    def self.dir
+      VagrantRubydns.working_dir.join('dependent_vms').tap(&:mkpath)
+    end
+  end
+end

data/lib/vagrant-rubydns/plugin.rb

@@ -12,14 +12,11 @@ module VagrantRubydns
       Config
     end
 
-    provisioner 'rubydns' do
-      require_relative 'provisioner'
-      Provisioner
-    end
-
     action_hook 'rubydns_setup', :machine_action_up do |hook|
       require_relative 'action/setup'
+      require_relative 'action/redirect_dns'
       hook.before(VagrantPlugins::ProviderVirtualBox::Action::Boot, Action::Setup)
+      hook.after(VagrantPlugins::ProviderVirtualBox::Action::Boot, Action::RedirectDns)
     end
 
     action_hook 'rubydns_teardown', :machine_action_halt do |hook|

data/lib/vagrant-rubydns/server.rb

@@ -1,7 +1,7 @@
 require 'rubydns'
 
 module VagrantRubydns
-  class Server
+  class Server < RExec::Daemon::Base
 
     INTERFACES = [
       [:udp, "0.0.0.0", 10053],
@@ -14,13 +14,24 @@ module VagrantRubydns
       @upstream ||= RubyDNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
     end
 
+    def self.logfile
+      VagrantRubydns.working_dir.join('server.log')
+    end
+
+    # For RExec
+    @@base_directory = VagrantRubydns.working_dir
+
+    def self.running?
+      RExec::Daemon::ProcessFile.status(self) == :running
+    end
+
     def self.run
       server = self
       RubyDNS::run_server(:listen => INTERFACES) do
-        logger.level = Logger::INFO
+        self.logger.level = Logger::INFO
         
         match(/.*/, IN::A) do |transaction|
-          ip = Store.get(transaction.name)
+          ip = Store.hosts.get(transaction.name)
           if ip
             transaction.respond!(ip)
           else

data/lib/vagrant-rubydns/store.rb

@@ -1,35 +1,47 @@
 module VagrantRubydns
   class Store
-    STORE_FILE = Pathname('.vagrant_dns.json')
+    def self.hosts
+      @hosts ||= new(VagrantRubydns.working_dir.join('hosts.json'))
+    end
+
+    attr_accessor :backing_file
+
+    def initialize(backing_file)
+      @backing_file = Pathname(backing_file)
+    end
 
-    def self.set(key, value)
+    def set(key, value)
       write(current_config.merge(key => value))
     end
 
-    def self.delete(key)
+    def delete(key)
       write(current_config.reject { |k, _| k == key })
     end
 
-    def self.get(key)
+    def get(key)
       current_config[key]
     end
 
-    def self.clear!
+    def clear!
       write({})
     end
 
     protected
 
-    def self.current_config
-      if STORE_FILE.exist?
-        JSON.parse(File.read(STORE_FILE))
+    def current_config
+      if backing_file.exist?
+        begin
+          JSON.parse(File.read(backing_file))
+        rescue JSON::ParserError
+          {}
+        end
       else
         {}
       end
     end
 
-    def self.write(config)
-      File.open(STORE_FILE, "w") do |f|
+    def write(config)
+      File.open(backing_file, "w") do |f|
         f.write(JSON.pretty_generate(config))
       end
     end

data/lib/vagrant-rubydns/version.rb

@@ -1,5 +1,5 @@
 module Vagrant
   module Rubydns
-    VERSION = "0.0.6"
+    VERSION = "0.1.0"
   end
 end

data/test/support/clear_dependent_vms.rb

@@ -0,0 +1,10 @@
+module ClearDependentVms
+  def setup
+    super
+    VagrantRubydns::DependentVMs.clear!
+  end
+end
+
+class MiniTest::Spec
+  include ClearDependentVms
+end

data/test/support/disable_server_daemon.rb

@@ -0,0 +1,37 @@
+class VagrantRubydns::Server < RExec::Daemon::Base
+  def self.start
+    @start_count += 1
+  end
+
+  def self.stop
+    @stop_count += 1
+  end
+
+  def self.running?
+    @start_count > 0
+  end
+
+  def self.clear!
+    @start_count = 0
+    @stop_count = 0
+  end
+
+  def self.start_count
+    @start_count
+  end
+
+  def self.stop_count
+    @stop_count
+  end
+end
+
+module FakeServerHooks
+  def setup
+    super
+    VagrantRubydns::Server.clear!
+  end
+end
+
+class MiniTest::Spec
+  include FakeServerHooks
+end

data/test/support/fake_working_dir.rb

@@ -0,0 +1,16 @@
+module FakeWorkingDirHooks
+  def setup
+    super
+    VagrantRubydns.working_dir = '/tmp/vagrant_rubydns_test_working_dir'
+  end
+
+  def teardown
+    super
+    VagrantRubydns.working_dir.rmtree if VagrantRubydns.working_dir.directory?
+  end
+end
+
+class MiniTest::Spec
+  include FakeWorkingDirHooks
+end
+

data/test/test_helper.rb

@@ -3,13 +3,8 @@ $:.push(File.expand_path('../../lib', __FILE__))
 require 'bundler/setup'
 require 'minitest/spec'
 
-require 'support/fake_ui'
-
 require 'vagrant-rubydns'
 
-# must be called before minitest/autorun to ensure proper at_exit ordering
-MiniTest::Unit.after_tests { VagrantRubydns::Store.clear! }
-
 require 'minitest/autorun'
 
 def fake_environment(extras={})
@@ -43,3 +38,10 @@ def fake_environment_with_machine(hostname, ip)
 
   { machine: machine, ui: FakeUI, global_config: env.config_global }
 end
+
+# order is important on these
+require 'support/fake_working_dir'
+require 'support/clear_dependent_vms'
+
+require 'support/fake_ui'
+require 'support/disable_server_daemon'

data/test/vagrant-rubydns/action/setup_test.rb

@@ -6,39 +6,64 @@ module VagrantRubydns
     describe Setup do
       it "calls the next app in the chain" do
         env = fake_environment(called: false)
-
         app = lambda { |e| e[:called] = true }
-
         setup = Setup.new(app, nil)
+
         setup.call(env)
 
         env[:called].must_equal true
       end
 
       it "stores the machine's hostname => ip address" do
-        Store.clear!
-
         app = Proc.new {}
         setup = Setup.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        setup.call(env)
 
+        Store.hosts.get('somehost.vagrant.dev').must_equal '1.2.3.4'
+      end
+
+      it "records the booting host as a dependent VM" do
+        app = Proc.new {}
+        setup = Setup.new(app, nil)
         env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
         setup.call(env)
 
-        Store.get('somehost.vagrant.dev').must_equal '1.2.3.4'
+        DependentVMs.list.must_equal %w[somehost.vagrant.dev]
       end
 
-      it "does nothing if it is not enabled via config" do
-        Store.clear!
+      it "starts the rubydns server if it's not already started" do
+        app = Proc.new {}
+        setup = Setup.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        setup.call(env)
+
+        Server.running?.must_equal true
+      end
 
+      it "does not attempt to start the server if it's already up" do
         app = Proc.new {}
         setup = Setup.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        Server.start
+        setup.call(env)
 
+        Server.start_count.must_equal 1
+      end
+
+      it "does nothing if it is not enabled via config" do
+        app = Proc.new {}
+        setup = Setup.new(app, nil)
         env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
-        env[:global_config].rubydns.disable
 
+        env[:global_config].rubydns.disable
         setup.call(env)
 
-        Store.get('somehost.vagrant.dev').must_equal nil
+        Store.hosts.get('somehost.vagrant.dev').must_equal nil
       end
     end
   end

data/test/vagrant-rubydns/action/teardown_test.rb

@@ -6,31 +6,75 @@ module VagrantRubydns
     describe Teardown do
       it "calls the next app in the chain" do
         env = fake_environment(called: false)
-
         app = lambda { |e| e[:called] = true }
-
         teardown = Teardown.new(app, nil)
+
         teardown.call(env)
 
         env[:called].must_equal true
       end
 
       it "clears the machine's hostname => ip address" do
-        Store.set('somehost.vagrant.dev', '1.2.3.4')
+        app = Proc.new {}
+        teardown = Teardown.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        Store.hosts.set('somehost.vagrant.dev', '1.2.3.4')
+        teardown.call(env)
+
+        Store.hosts.get('somehost.vagrant.dev').must_equal nil
+      end
 
+      it "removes the machine as a dependent VM" do
         app = Proc.new {}
         teardown = Teardown.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
 
+        DependentVMs.add(env[:machine])
+        teardown.call(env)
+
+        DependentVMs.list.must_equal []
+      end
+
+      it "stops the rubydns server when there are no dependent machines left" do
+        app = Proc.new {}
+        teardown = Teardown.new(app, nil)
         env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        Server.start
+        teardown.call(env)
+
+        Server.stop_count.must_equal 1
+      end
+
+      it "leaves the rubydns server when other dependent vms exist" do
+        app = Proc.new {}
+        teardown = Teardown.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
+        other_env = fake_environment_with_machine('otherhost.vagrant.dev', '1.2.3.4')
+        DependentVMs.add(other_env[:machine])
+
+        Server.start
+        teardown.call(env)
+
+        Server.stop_count.must_equal 0
+      end
+
+      it "leaves the server alone if it's not running" do
+        app = Proc.new {}
+        teardown = Teardown.new(app, nil)
+        env = fake_environment_with_machine('somehost.vagrant.dev', '1.2.3.4')
+
         teardown.call(env)
 
-        Store.get('somehost.vagrant.dev').must_equal nil
+        Server.stop_count.must_equal 0
       end
 
       it "does nothing when rubydns is disabled" do
         # somewhat unrealistic since this entry shouldn't be there if it was
         # disabled in the first place, but oh well
-        Store.set('somehost.vagrant.dev', '1.2.3.4')
+        Store.hosts.set('somehost.vagrant.dev', '1.2.3.4')
 
         app = Proc.new {}
         teardown = Teardown.new(app, nil)
@@ -40,7 +84,7 @@ module VagrantRubydns
 
         teardown.call(env)
 
-        Store.get('somehost.vagrant.dev').must_equal '1.2.3.4'
+        Store.hosts.get('somehost.vagrant.dev').must_equal '1.2.3.4'
       end
     end
   end

data/test/vagrant-rubydns/dependent_vms_test.rb

@@ -0,0 +1,33 @@
+require 'test_helper'
+
+module VagrantRubydns
+  describe DependentVMs do
+    describe "any?" do
+      it "reports false when nothing has happened" do
+        DependentVMs.any?.must_equal false
+      end
+
+      it "reports true once a machine has been added" do
+        env = fake_environment_with_machine('recordme.example.dev', '1.2.3.4')
+        DependentVMs.add(env[:machine])
+        DependentVMs.any?.must_equal true
+      end
+
+      it "reports false if a machine has been added then removed" do
+        env = fake_environment_with_machine('recordme.example.dev', '1.2.3.4')
+        DependentVMs.add(env[:machine])
+        DependentVMs.remove(env[:machine])
+        DependentVMs.any?.must_equal false
+      end
+
+      it "reports true if not all machines have been removed" do
+        first_env = fake_environment_with_machine('recordme.example.dev', '1.2.3.4')
+        second_env = fake_environment_with_machine('alsome.example.dev', '2.3.4.5')
+        DependentVMs.add(first_env[:machine])
+        DependentVMs.add(second_env[:machine])
+        DependentVMs.remove(first_env[:machine])
+        DependentVMs.any?.must_equal true
+      end
+    end
+  end
+end

data/test/vagrant-rubydns/store_test.rb

@@ -2,47 +2,53 @@ require 'test_helper'
 
 module VagrantRubydns
   describe Store do
-    before { Store.clear! }
-    
+    before {
+      @store = Store.new(Tempfile.new(%w[rubydns_test_store .json]))
+    }
+
+    after {
+      @store.backing_file.unlink
+    }
+
     describe "set" do
       it "sets the key to the value and makes it available for getting" do
-        Store.set('foo', 'bar')
+        @store.set('foo', 'bar')
 
-        Store.get('foo').must_equal 'bar'
+        @store.get('foo').must_equal 'bar'
       end
 
       it "allows updating keys that already exist" do
-        Store.set('foo', 'bar')
-        Store.set('foo', 'qux')
+        @store.set('foo', 'bar')
+        @store.set('foo', 'qux')
 
-        Store.get('foo').must_equal 'qux'
+        @store.get('foo').must_equal 'qux'
       end
     end
 
     describe "get" do
       it "returns nil for unset values" do
-        Store.get('notakey').must_equal nil
+        @store.get('notakey').must_equal nil
       end
 
       it "returns the latest set value (no caching)" do
-        Store.set('foo', 'first')
-        Store.get('foo').must_equal 'first'
-        Store.set('foo', 'second')
-        Store.get('foo').must_equal 'second'
-        Store.delete('foo')
-        Store.get('foo').must_equal nil
+        @store.set('foo', 'first')
+        @store.get('foo').must_equal 'first'
+        @store.set('foo', 'second')
+        @store.get('foo').must_equal 'second'
+        @store.delete('foo')
+        @store.get('foo').must_equal nil
       end
     end
 
     describe "delete" do
       it "removes the key from the store" do
-        Store.set('now', 'you see me')
+        @store.set('now', 'you see me')
 
-        Store.get('now').must_equal 'you see me'
+        @store.get('now').must_equal 'you see me'
 
-        Store.delete('now')
+        @store.delete('now')
 
-        Store.get('now').must_equal nil # you don't!
+        @store.get('now').must_equal nil # you don't!
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vagrant-rubydns
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-21 00:00:00.000000000 Z
+date: 2013-05-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rubydns
@@ -73,20 +73,25 @@ files:
 - Rakefile
 - examples/Vagrantfile
 - lib/vagrant-rubydns.rb
+- lib/vagrant-rubydns/action/redirect_dns.rb
 - lib/vagrant-rubydns/action/setup.rb
 - lib/vagrant-rubydns/action/teardown.rb
 - lib/vagrant-rubydns/command.rb
 - lib/vagrant-rubydns/config.rb
+- lib/vagrant-rubydns/dependent_vms.rb
 - lib/vagrant-rubydns/plugin.rb
-- lib/vagrant-rubydns/provisioner.rb
 - lib/vagrant-rubydns/server.rb
 - lib/vagrant-rubydns/store.rb
 - lib/vagrant-rubydns/util.rb
 - lib/vagrant-rubydns/version.rb
+- test/support/clear_dependent_vms.rb
+- test/support/disable_server_daemon.rb
 - test/support/fake_ui.rb
+- test/support/fake_working_dir.rb
 - test/test_helper.rb
 - test/vagrant-rubydns/action/setup_test.rb
 - test/vagrant-rubydns/action/teardown_test.rb
+- test/vagrant-rubydns/dependent_vms_test.rb
 - test/vagrant-rubydns/store_test.rb
 - vagrant-rubydns.gemspec
 homepage: ''
@@ -115,9 +120,13 @@ signing_key:
 specification_version: 3
 summary: a simple dns server for vagrant guests
 test_files:
+- test/support/clear_dependent_vms.rb
+- test/support/disable_server_daemon.rb
 - test/support/fake_ui.rb
+- test/support/fake_working_dir.rb
 - test/test_helper.rb
 - test/vagrant-rubydns/action/setup_test.rb
 - test/vagrant-rubydns/action/teardown_test.rb
+- test/vagrant-rubydns/dependent_vms_test.rb
 - test/vagrant-rubydns/store_test.rb
 has_rdoc: 

data/lib/vagrant-rubydns/provisioner.rb

@@ -1,34 +0,0 @@
-module VagrantRubydns
-  class Provisioner < Vagrant.plugin('2', :provisioner)
-    def initialize(machine, config)
-      super
-    end
-
-    def configure(root_config)
-    end
-
-    def provision
-      @machine.env.ui.info "setting up machine's DNS to point to our server"
-
-      redirect_dns_to_unpriviledged_port_tcp = 'OUTPUT -t nat -d 10.0.2.2 -p tcp --dport 53 -j DNAT --to-destination 10.0.2.2:10053'
-      command = %Q(iptables -C #{redirect_dns_to_unpriviledged_port_tcp} 2> /dev/null || iptables -A #{redirect_dns_to_unpriviledged_port_tcp})
-      _run_command(command)
-
-      redirect_dns_to_unpriviledged_port_udp = 'OUTPUT -t nat -d 10.0.2.2 -p udp --dport 53 -j DNAT --to-destination 10.0.2.2:10053'
-      command = %Q(iptables -C #{redirect_dns_to_unpriviledged_port_udp} 2> /dev/null || iptables -A #{redirect_dns_to_unpriviledged_port_udp})
-      _run_command(command)
-
-      command = %q(sed -i'' -e 's/10.0.2.3/10.0.2.2/' /etc/resolv.conf)
-      _run_command(command)
-    end
-
-    def _run_command(command)
-      @machine.communicate.sudo(command) do |data, type|
-        if [:stderr, :stdout].include?(type)
-          color = (type == :stdout) ? :green : :red
-          @machine.env.ui.info(data.chomp, :color => color, :prefix => false)
-        end
-      end
-    end
-  end
-end