faildns 0.0.1

data/bin/failnamed

@@ -0,0 +1,203 @@
+#! /usr/bin/env ruby
+# Copyleft meh. [http://meh.doesntexist.org | meh@paranoici.org]
+#
+# This file is part of faildns.
+#
+# faildns is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# faildns is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with faildns. If not, see <http://www.gnu.org/licenses/>.
+
+require 'faildns/server'
+require 'faildns/client'
+require 'optimus'
+require 'rexml/document'
+
+opt = Optimus.new {|o|
+  o.set(
+    :type => :numeric,
+
+    :long  => 'port',
+    :short => 'p',
+
+    :default => 53
+  )
+
+  o.set(
+    :type => :string,
+
+    :long  => 'host',
+    :short => 'h',
+
+    :default => '0.0.0.0'
+  )
+
+  o.set(
+    :type => :string,
+
+    :long  => 'configuration',
+    :short => 'c'
+  )
+}
+
+$config = REXML::Document.new(File.new(opt.params[:c]))
+
+$server = DNS::Server.new {|s|
+    s.options[:host] = ($config.elements.each('//bind/host') {}.first.text rescue nil) || opt.params[:host]
+    s.options[:port] = ($config.elements.each('//bind/port') {}.first.text rescue nil) || opt.params[:port]
+}
+
+$client = DNS::Client.new {|c|
+  $config.elements.each('//servers/server') {|server|
+    c.servers.push server.text
+  }
+}
+
+$options = {}
+
+class Domain
+  attr_accessor :type, :matches, :location
+
+  def initialize
+    if block_given?
+      yield self
+    end
+
+    self.normalize
+  end
+
+  def match (domain)
+    domain.to_s.match(@regexp)
+  end
+
+  def normalize
+    case @type
+      when 'regexp'
+        @regexp = Regexp.new(@matches)
+
+      else
+        @regexp = Regexp.new('^' + Regexp.escape(@matches).gsub(/\\\*/, '.*').gsub(/\\\?/, '.') + '$')
+    end
+  end
+end
+
+$options[:domains] = []
+
+$config.elements.each('//domains/domain') {|domain|
+  $options[:domains].push Domain.new {|d|
+    d.type     = domain.attributes['type']
+    d.matches  = domain.attributes['matches']
+    d.location = domain.text.strip
+  }
+}
+    
+module Commands
+  module IN
+    def self.A (message)
+      question = message.questions.first
+
+      ip = $options[:domains].find {|domain|
+        domain.match(question.name)
+      }.location rescue nil
+
+      if !ip
+        ip = $client.resolve(question.name)
+      else
+        # if it's not an IP but a domain, do magic
+        if !(IPAddr.new(ip) rescue false)
+          tmp = $options[:domains].find {|domain|
+            domain.match(ip)
+          }
+          
+          if !tmp
+            ip = $client.resolve(ip)
+          else
+            ip = tmp.location
+          end
+        end
+      end
+
+      if !ip
+        return DNS::Message.new(
+          DNS::Header.new {|h|
+            h.id = message.header.id
+    
+            h.type  = :RESPONSE
+            h.class = :QUERY
+    
+            h.status = :NXDOMAIN
+    
+            h.questions = 1
+          },
+    
+          [question]
+        )
+      end
+    
+      return DNS::Message.new(
+        DNS::Header.new {|h|
+          h.id = message.header.id
+    
+          h.type  = :RESPONSE
+          h.class = :QUERY
+    
+          h.status = :NOERROR
+    
+          h.questions = 1
+          h.answers   = 1
+        },
+    
+        [question],
+    
+        [question].map {|question|
+          DNS::ResourceRecord.new {|r|
+            r.name  = question.name
+            r.type  = :A
+            r.class = :IN
+    
+            r.ttl = 3600
+    
+            r.length = DNS::ResourceRecord::IN::A.length
+            r.data   = DNS::ResourceRecord::IN::A.new(ip)
+          }
+        }
+      )
+    end
+  end
+end
+
+$server.register :input, lambda {|socket, message|
+  question = message.questions.first
+  method   = Commands.const_get(question.class.to_sym).method(question.type.to_sym) rescue nil
+
+  if !method
+    socket.send DNS::Message.new(
+      DNS::Header.new {|h|
+        h.id = message.header.id
+
+        h.type  = :RESPONSE
+        h.class = :QUERY
+
+        h.status = :NOTIMP
+
+        h.questions = 1
+      },
+
+      [question]
+    )
+
+    return
+  end
+
+  socket.send method.call(message)
+}
+
+$server.start

data/bin/failresolve

@@ -0,0 +1,37 @@
+#! /usr/bin/env ruby
+#--
+# Copyleft meh. [http://meh.doesntexist.org | meh@paranoici.org]
+#
+# This file is part of faildns.
+#
+# faildns is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# faildns is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with faildns. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'faildns/client'
+require 'optimus'
+
+opt = Optimus.new {|o|
+  o.set(
+    :type => :array,
+
+    :long  => 'servers',
+    :short => 's'
+  )
+}
+
+client = DNS::Client.new(:servers => opt.params[:s])
+
+opt.args.each {|arg|
+  puts "#{arg}: #{client.resolve arg}"
+}

data/lib/faildns.rb

@@ -0,0 +1,20 @@
+#--
+# Copyleft meh. [http://meh.doesntexist.org | meh@paranoici.org]
+#
+# This file is part of faildns.
+#
+# faildns is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# faildns is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with faildns. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'faildns/message'

data/lib/faildns/class.rb

@@ -0,0 +1,96 @@
+#--
+# Copyleft meh. [http://meh.doesntexist.org | meh@paranoici.org]
+#
+# This file is part of faildns.
+#
+# faildns is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# faildns is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with faildns. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module DNS
+
+#--
+# CLASS fields appear in resource records.  The following CLASS mnemonics
+# and values are defined:
+# 
+# IN              1 the Internet
+# 
+# CS              2 the CSNET class (Obsolete - used only for examples in
+#                 some obsolete RFCs)
+# 
+# CH              3 the CHAOS class
+# 
+# HS              4 Hesiod [Dyer 87]
+#++
+
+class Class
+  Values = {
+    1 => :IN,
+    2 => :CS,
+    3 => :CH,
+    4 => :HS
+  }
+
+  def self.parse (string)
+    string.force_encoding 'BINARY'
+
+    result = self.new(string.unpack('n').first)
+    string[0, self.length] = ''
+
+    return result
+  end
+
+  def self.length (string=nil)
+    2
+  end
+
+  attr_reader :value
+
+  def initialize (value)
+    if value.is_a? Symbol
+      @value = Values.find {|key, val| val == value}.first rescue nil
+    elsif value.is_a? Integer
+      @value = value
+    else
+      @value = value.value rescue nil
+    end
+
+    if !self.to_sym
+      raise ArgumentError.new('The passed value is not a suitable class.')
+    end
+  end
+
+  def pack
+    [@value].pack('n')
+  end
+
+  def == (what)
+    if what.is_a? Symbol
+      self.to_sym == what
+    elsif value.is_a? Integer
+      @value == what
+    else
+      @value == what.value rescue false
+    end
+  end
+
+  def to_sym
+    Values[@value]
+  end
+
+  def to_s
+    Values[@value].to_s
+  end
+end
+
+end

data/lib/faildns/client.rb

@@ -0,0 +1,114 @@
+#--
+# Copyleft meh. [http://meh.doesntexist.org | meh@paranoici.org]
+#
+# This file is part of faildns.
+#
+# faildns is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published
+# by the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# faildns is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with faildns. If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'timeout'
+require 'socket'
+require 'faildns/message'
+
+module DNS
+
+class Client
+  attr_reader :options, :servers
+
+  def initialize (options={})
+    @options = options
+
+    @servers = @options[:servers] || []
+
+    if block_given?
+      yield self
+    end
+  end
+
+  def resolve (domain, timeout=10, tries=3)
+    result = nil
+    socket = UDPSocket.new
+
+    id = (rand * 100000).to_i % 65536
+
+    1.upto(tries) {
+      @servers.each {|server|
+        socket.connect(server.to_s, 53)
+  
+        socket.print Message.new(
+          Header.new {|h|
+            h.id = id
+  
+            h.type  = :QUERY
+            h.class = :QUERY
+  
+            h.recursive!
+  
+            h.questions = 1
+          },
+  
+          [Question.new {|q|
+            q.name = domain
+  
+            q.class = :IN
+            q.type  = :A
+          }]
+        ).pack
+
+        if (tmp = Timeout.timeout(timeout) { socket.recvfrom(512) } rescue nil)
+          DNS.debug tmp, { :level => 9 }
+
+          tmp = Message.parse(tmp[0])
+
+          if tmp.header.status == :NXDOMAIN
+            result = false
+            break
+          end
+
+          if tmp.header.status == :NOERROR && tmp.header.id == id
+            result = tmp.answers.find {|answer| answer.type == :A}.data.to_s rescue nil
+            break
+          end
+        end
+      }
+
+      if !result.nil?
+        break
+      end
+    }
+
+    return result
+  end
+
+  def query (message, timeout=10)
+    result = []
+    socket = UDPSocket.new
+
+    @servers.each {|server|
+      socket.connect(server.to_s, 53)
+
+      socket.print message.pack
+
+      if (tmp = Timeout.timeout(timeout) { socket.recvfrom(512) } rescue nil)
+        DNS.debug tmp, { :level => 9 }
+
+        result.push Message.parse(tmp[0])
+      end
+    }
+
+    return result
+  end
+end
+
+end