vagrant-autodns 0.0.1

data/lib/vagrant-autodns/commands/add.rb

@@ -0,0 +1,37 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Add < Vagrant.plugin("2", :command)
+
+      def execute
+        require_relative '../autodnsdb'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns add <hostname> <ip> [vagrant_id]"
+        end
+        argv = parse_options(opts) || return
+        #vagrant_id is an option parameter
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length < 2 || argv.length > 3)
+
+        hostname, ip, vagrant_id = argv
+
+        if VagrantAutoDNS.autodnsdb.update_record(hostname, ip, vagrant_id)
+          @env.ui.info("Entry #{ip} added for host #{hostname}")
+          0
+        else
+          @env.ui.error("Entry #{ip} not added for host #{hostname}")
+          1
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-autodns/commands/clear.rb

@@ -0,0 +1,34 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Clear < Vagrant.plugin("2", :command)
+
+      def execute
+        require_relative '../autodnsdb'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns clear"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        if VagrantAutoDNS.autodnsdb.delete_all_records
+          @env.ui.info("DNS cleared")
+          0
+        else
+          @env.ui.error("Something went wrong")
+          1
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/delete.rb

@@ -0,0 +1,34 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Delete < Vagrant.plugin("2", :command)
+      def execute
+        require_relative '../autodnsdb'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns delete <hostname>"
+        end
+        argv = parse_options(opts) || return
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 1)
+
+        db = VagrantAutoDNS.autodnsdb
+        hostname = argv[0]
+        if db.delete_record(hostname)
+          @env.ui.info("Host #{hostname} deleted")
+          0
+        else
+          @env.ui.error("Host #{hostname} not deleted")
+          1
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/list.rb

@@ -0,0 +1,36 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class List < Vagrant.plugin("2", :command)
+
+      def execute
+        require_relative '../autodnsdb'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns list"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        records = VagrantAutoDNS.autodnsdb.list_all_records
+        if records.empty?
+          @env.ui.error("No records found")
+        else
+          records.each do |record|
+            @env.ui.success("Host #{record["hostname"]} has IP #{record["ip"]}")
+          end
+        end
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/reload_all.rb

@@ -0,0 +1,28 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Reload < Vagrant.plugin("2", :command)
+      def execute
+        require_relative '../daemon'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns [re]start"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        #TODO: Fixme
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/restart.rb

@@ -0,0 +1,28 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Restart < Vagrant.plugin("2", :command)
+      def execute
+      	require_relative '../daemon'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns [re]start"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        VagrantAutoDNS::Daemon.restart
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/status.rb

@@ -0,0 +1,28 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Status < Vagrant.plugin("2", :command)
+      def execute
+        require_relative '../daemon'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns [re]start"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        @env.ui.success("Daemon status is #{VagrantAutoDNS::Daemon.daemon_status}")
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/commands/stop.rb

@@ -0,0 +1,28 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+module VagrantAutoDNS
+  class Command
+    class Stop < Vagrant.plugin("2", :command)
+      def execute
+        require_relative '../daemon'
+
+        opts = OptionParser.new do |optp|
+          optp.banner = "Usage: vagrant autodns [re]start"
+        end
+        argv = parse_options(opts) || return
+        #Takes no arguments
+        raise Vagrant::Errors::CLIInvalidUsage, :help => opts.help.chomp if (argv.length != 0)
+
+        VagrantAutoDNS::Daemon.stop
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-autodns/config.rb

@@ -0,0 +1,110 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+require_relative 'daemon'
+
+module VagrantAutoDNS
+  class Config < Vagrant.plugin('2', :config)
+
+    # @return [Boolean]
+    attr_accessor :enabled
+
+    # @return [String, Array] (default System resolver)
+    attr_accessor :resolver
+
+    # @return [String] Valid IPV4 address (default 127.0.0.1)
+    attr_accessor :listen_ip
+
+    # @return [Fixnum] Port number (default 15353)
+    attr_accessor :listen_port
+
+    # @return [String] IPv6 address
+    attr_accessor :listen_ipv6
+
+    # @return [Boolean] Enable ipv6
+    attr_accessor :enable_ipv6
+
+    # @return [String] Directory to store data and state
+    attr_accessor :working_directory
+
+    # @return [String] DB File name
+    attr_accessor :db_file
+
+    DAEMON_DEFAULT_OPTS = {
+      :working_directory => '.vagrant/autodns',
+      :listen_ip => '127.0.0.1',
+      :listen_port => 15353,
+      :resolver => :system, #Deamon defaults to system
+      :db_file => 'autodns.db',
+      :enable_ipv6 => true,
+    }
+
+    CLIENT_DEFAULT_OPTS = {
+      :aliases => []
+    }
+
+    DEFAULT_VALUES = {
+      :enabled => false
+    }.merge(DAEMON_DEFAULT_OPTS).merge(CLIENT_DEFAULT_OPTS)
+
+    def initialize
+      super
+      DEFAULT_VALUES.keys.each do |default_key|
+        instance_key = instance_var(default_key)
+        instance_variable_set(instance_key, UNSET_VALUE)
+      end
+    end
+
+    def alias(*aliases)
+      @aliases.concat(aliases)
+    end
+
+    def enable
+      @enabled = true
+    end
+
+    def disable
+      @enabled = false
+    end
+
+    def enabled?
+      #Coerce to boolean
+      !!@enabled
+    end
+
+    def finalize!
+      #special handling for working_directory
+      expand_working_directory
+
+      DEFAULT_VALUES.each do |default_key, default_value|
+        instance_key = instance_var(default_key)
+        if instance_variable_get(instance_key) == UNSET_VALUE
+          instance_variable_set(instance_key, default_value)
+        end
+        next unless DAEMON_DEFAULT_OPTS.has_key?(default_key)
+        Daemon.send(:instance_variable_set, instance_key, instance_variable_get(instance_key))
+      end
+    end
+
+    private
+
+    def expand_working_directory
+      case @working_directory
+      when UNSET_VALUE
+        @working_directory = File.expand_path DAEMON_DEFAULT_OPTS[:working_directory]
+      else
+        @working_directory = File.expand_path working_directory
+      end
+    end
+
+    def instance_var(key)
+      '@' + key.to_s
+    end
+  end
+end

data/lib/vagrant-autodns/daemon.rb

@@ -0,0 +1,172 @@
+#© 2014 LinkedIn Corp. All rights reserved.
+#Licensed under the Apache License, Version 2.0 (the "License"); you may not
+#use this file except in compliance with the License. You may obtain a copy of
+#the License at  http://www.apache.org/licenses/LICENSE-2.0
+
+#Unless required by applicable law or agreed to in writing, software
+#distributed under the License is distributed on an "AS IS" BASIS,
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+
+require 'rexec'
+require 'rexec/daemon'
+require 'rubydns'
+require 'rubydns/resolver'
+require 'rubydns/system'
+require 'rainbow/ext/string'
+require_relative 'autodnsdb'
+
+module VagrantAutoDNS
+  class Daemon < RExec::Daemon::Base
+
+    @@base_directory = nil
+
+    ProcessFile = RExec::Daemon::ProcessFile
+    A_RECORD = Resolv::DNS::Resource::IN::A
+    AAAA_RECORD = Resolv::DNS::Resource::IN::AAAA
+
+    class << self
+      attr_accessor :working_directory, :resolver, :db_file
+      attr_accessor :listen_ip, :listen_port, :enable_ipv6
+    end
+
+    def self.run
+      upstream = upstream_resolver
+      db_lib = AutoDNSDB
+      db_absolute_path = autodnsdb_path
+      allow_ipv6 = ipv6_enabled?
+      RubyDNS::run_server(:listen => listen_interface) do
+        db = db_lib.new(db_absolute_path)
+        match(//, A_RECORD) do |transaction|
+          domain = transaction.name.to_s.encode('US-ASCII')
+          record = db.find_record(domain)
+          if record
+            transaction.respond!(record['ip'])
+          else
+            transaction.passthrough!(upstream)
+          end
+        end
+
+        match(//, AAAA_RECORD) do |transaction|
+          if allow_ipv6
+            transaction.passthrough!(upstream)
+          else
+            transaction.fail!(:NXDomain)
+          end
+        end
+
+        # Default DNS handler
+        otherwise do |transaction|
+          transaction.passthrough!(upstream)
+        end
+      end
+    end
+
+    def self.restart
+      stop unless stopped?
+      start
+    end
+
+    def self.ensure_running
+      restart unless running?
+    end
+
+    def self.daemon_status
+      ProcessFile.status(self)
+    end
+
+    def self.pid
+      ProcessFile.recall(self)
+    end
+
+    def self.running?
+      daemon_status == :running
+    end
+
+    def self.stopped?
+      daemon_status == :stopped
+    end
+
+    def self.ipv6_enabled?
+      @enable_ipv6.nil? || @enable_ipv6
+    end
+
+    def self.autodnsdb
+      return @autodnsdb if @autodnsdb.is_a?(AutoDNSDB)
+      @autodnsdb = AutoDNSDB.new(autodnsdb_path)
+    end
+
+    def self.autodnsdb_path
+      @autodnsdb_path ||= File.expand_path(File.join(working_directory, db_file))
+    end
+
+    def self.working_directory=(dir)
+      @working_directory = File.expand_path(dir)
+    end
+
+    private
+
+    # RubyDNS upstream resolver
+    def self.upstream_resolver
+      #Bail out if already set
+      return @resolver if @resolver.is_a?(RubyDNS::Resolver)
+      #Case statment must set new_resolver to a valid interface
+      case @resolver
+      when String
+        # IP or IP/PORT combo
+        new_resolver = to_interface(@resolver)
+      when Array
+        #If 2 strings are provided assume they are IPs
+        if @resolver.length == 2 && @resolver.all?{|v| v.is_a? String}
+          new_resolver = @resolver.map{|ip| to_interface(ip)}.flatten(1)
+        #Else expect array formated in interface form
+        else
+          new_resolver = @resolver
+        end
+      when Symbol
+        #Use the predefined custom_resolvers
+        new_resolver = custom_resolver(@resolver)
+      when NilClass
+        #Default to system_resolver
+        new_resolver = system_resolver
+      end
+      @resolver = RubyDNS::Resolver.new(new_resolver)
+    end
+
+    def self.listen_interface
+      to_interface(listen_ip, listen_port)
+    end
+
+    def self.custom_resolver(resolver = :random)
+      return system_resolver if resolver == :system
+      #TODO: move to config file somewhere
+      hosted_resolvers = {
+        :level3 => ['209.244.0.3', '209.244.0.4'],
+        :google => ['8.8.8.8', '8.8.4.4'],
+        :securly => ['184.169.143.224', '184.169.161.155'],
+        :comodo => ['8.26.56.26', '8.20.247.20'],
+        :opendns => ['208.67.222.222', '208.67.220.220'],
+        :norton => ['198.153.192.40', '198.153.194.40'],
+        :opennic => ['216.87.84.211', '23.90.4.6']
+      }
+      resolver = hosted_resolvers.keys.sample if resolver == :random
+      resolver_ip = hosted_resolvers[resolver]
+      resolver_ip.map{|ip| to_interface(ip)}.flatten(1)
+    end
+
+    def self.system_resolver
+      RubyDNS::System::nameservers
+    end
+
+    IPV4_6_PORT_REGEX = 
+      /^(?:[0-9.]+|(?:\[[0-9a-fA-F:]+\]))(:[0-9]+)?$/
+
+    def self.to_interface(ip, port = 53)
+      #Try to extract IP and PORT from IP
+      ip_r, port_r = ip.match(IPV4_6_PORT_REGEX).to_a
+      #If regex found port, set it accordingly
+      # remove preceding semi-colon
+      port = port_r[1..-1].to_i if port_r
+      [[:udp, ip_r, port], [:tcp, ip_r, port]]
+    end
+  end
+end