faildns 0.0.1

data/lib/faildns/server/dispatcher/connectiondispatcher.rb

@@ -0,0 +1,93 @@
+#--
+# 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 'thread'
+require 'socket'
+
+require 'faildns/server/dispatcher/socket'
+require 'faildns/message'
+
+module DNS
+
+class Server
+
+class Dispatcher
+
+class ConnectionDispatcher
+  attr_reader :server, :dispatcher, :listening
+
+  def initialize (dispatcher)
+    @server     = dispatcher.server
+    @dispatcher = dispatcher
+
+    @host = @server.options[:host] || '0.0.0.0'
+    @port = @server.options[:port] || 53
+
+    @sockets = []
+  end
+
+  def start
+    @listening = {
+      :TCP => TCPServer.new(@host, @port),
+      :UDP => UDPSocket.new
+    }
+
+    @listening[:UDP].bind(@host, @port)
+  end
+
+  def read
+    reading, = IO::select @sockets.concat([@listening[:UDP], @listening[:TCP]])
+
+    if !reading
+      return
+    end
+
+    reading.each {|socket|
+      if socket.is_a? TCPServer
+        @sockets.push @listening[:TCP].accept_nonblock
+      elsif socket.is_a? TCPSocket
+        self.handle socket.recv_nonblock(65535), socket
+        @sockets.delete(socket)
+      else
+        begin
+          self.handle *@listening[:UDP].recvfrom_nonblock(512)
+        rescue Errno::EAGAIN
+        end
+      end
+    }
+  end
+
+  def handle (string, socket)
+    DNS.debug "#{socket.inspect}: #{string.inspect}", { :level => 9 }
+
+    Thread.new(string) {|input|
+      begin
+        @dispatcher.dispatch :input, Socket.new(@dispatcher, socket), Message.parse(string)
+      rescue Exception => e
+        DNS.debug e
+      end
+    }
+  end
+end
+
+end
+
+end
+
+end

data/lib/faildns/server/dispatcher/event.rb

@@ -0,0 +1,47 @@
+#--
+# 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 Server
+
+class Dispatcher
+
+class Event
+    class Callback
+        attr_reader :method
+        attr_accessor :priority
+
+        def initialize (method, priority=0)
+            @method   = method
+            @priority = priority
+        end
+
+        def call (*args)
+            return @method.call(*args)
+        end
+    end
+end
+
+end
+
+end
+
+end

data/lib/faildns/server/dispatcher/eventdispatcher.rb

@@ -0,0 +1,73 @@
+#--
+# 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/dispatcher/event'
+
+module DNS
+
+class Server
+
+class Dispatcher
+
+class EventDispatcher
+  attr_reader :server, :dispatcher
+
+  def initialize (dispatcher)
+    @server     = dispatcher.server
+    @dispatcher = dispatcher
+
+    @chains = {
+      :input  => [],
+      :output => [],
+
+      :custom => {}
+    }
+  end
+
+  def dispatch (chain, socket, message)
+    @chains[chain].each {|callback|
+      callback.call(socket, message)
+    }
+  end
+
+  def fire (event, *args)
+    (@chain[:custom][event] || []).each {|callback|
+      begin
+        if callback.call(*args) == false
+          return false
+        end
+      rescue Exception => e
+      end
+    }
+  end
+
+  def observe (event, callback, priority=0)
+    @chains[:custom][event].push((callback.is_a?(Event::Callback)) ? callback : Event::Callback.new(callback, priority))
+  end
+
+  def register (chain, callback, priority=0)
+    @chains[chain].push((callback.is_a?(Event::Callback)) ? callback : Event::Callback.new(callback, priority))
+  end
+end
+
+end
+
+end
+
+end

data/lib/faildns/server/dispatcher/socket.rb

@@ -0,0 +1,93 @@
+#--
+# 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 Server
+
+class Dispatcher
+
+class ConnectionDispatcher
+
+class Socket
+  attr_reader :ip, :port, :type
+
+  def initialize (dispatcher, what)
+    @dispatcher = dispatcher
+
+    if what.is_a? TCPSocket
+      @type = :TCP
+      @ip   = what.peeraddr[3]
+      @port = what.addr[1]
+
+      @socket = what
+    else
+      @type = :UDP
+      @ip   = what[3]
+      @port = what[1]
+
+      @socket = dispatcher.connection.listening[:UDP]
+    end
+  end
+
+  def send (message, close=true)
+    @dispatcher.dispatch :output, self, message
+
+    if @type == :UDP && message.pack.length > 512
+      [message.additionals, message.authorities, message.answers, message.questions].each {|rr|
+        while (tmp = message.pack).length > 512 && r.pop; end
+
+        if tmp.length <= 512
+          break
+        end
+      }
+
+      message.header.questions   = message.questions.length
+      message.header.answers     = message.answers.length
+      message.header.authorities = message.authorities.length
+      message.header.additionals = message.additionals.length
+
+      message.header.truncated!
+    end
+
+    if @socket.is_a? TCPSocket
+      @socket.send_nonblock(message.pack)
+
+      if close
+        @socket.close
+      end
+    else
+      @socket.send(message.pack, 0, ::Socket.pack_sockaddr_in(@port, @ip))
+    end
+  end
+
+  def close
+    if @socket.is_a? TCPSocket
+      @socket.close
+    end
+  end
+end
+
+end
+
+end
+
+end
+
+end

data/lib/faildns/type.rb

@@ -0,0 +1,186 @@
+#--
+# 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
+
+#--
+# TYPE fields are used in resource records.  Note that these types are a
+# subset of QTYPEs.
+# 
+# TYPE            value and meaning
+# 
+# A               1 a host address
+# 
+# NS              2 an authoritative name server
+# 
+# MD              3 a mail destination (Obsolete - use MX)
+# 
+# MF              4 a mail forwarder (Obsolete - use MX)
+# 
+# CNAME           5 the canonical name for an alias
+# 
+# SOA             6 marks the start of a zone of authority
+# 
+# MB              7 a mailbox domain name (EXPERIMENTAL)
+# 
+# MG              8 a mail group member (EXPERIMENTAL)
+# 
+# MR              9 a mail rename domain name (EXPERIMENTAL)
+# 
+# NULL            10 a null RR (EXPERIMENTAL)
+# 
+# WKS             11 a well known service description
+# 
+# PTR             12 a domain name pointer
+# 
+# HINFO           13 host information
+# 
+# MINFO           14 mailbox or mail list information
+# 
+# MX              15 mail exchange
+# 
+# TXT             16 text strings
+#++
+
+class Type
+  Values = {
+    1  => :A,
+    2  => :NS,
+    3  => :MD,
+    4  => :MF,
+    5  => :CNAME,
+    6  => :SOA,
+    7  => :MB,
+    8  => :MG,
+    9  => :MR,
+    10 => :NULL,
+    11 => :WKS,
+    12 => :PTR,
+    13 => :HINFO,
+    14 => :MINFO,
+    15 => :MX,
+    16 => :TXT,
+    17 => :RP,
+    18 => :AFSDB,
+    19 => :X25,
+    20 => :ISDN,
+    21 => :RT,
+    22 => :NSAP,
+    23 => :NSAP_PTR,
+    24 => :SIG,
+    25 => :KEY,
+    26 => :PX,
+    27 => :GPOS,
+    28 => :AAAA,
+    29 => :LOC,
+    30 => :NXT,
+    31 => :EID,
+    32 => :NIMLOC,
+    33 => :SRV,
+    34 => :ATMA,
+    35 => :NAPTR,
+    36 => :KX,
+    37 => :CERT,
+    38 => :A6,
+    39 => :DNAME,
+    40 => :SINK,
+    41 => :OPT,
+    42 => :APL,
+    43 => :DS,
+    44 => :SSHFP,
+    45 => :IPSECKEY,
+    46 => :RRSIG,
+    47 => :NSEC,
+    48 => :DNSKEY,
+    49 => :DHCID,
+    50 => :NSEC3,
+    51 => :NSEC3PARAM,
+
+    55 => :HIP,
+    56 => :NINFO,
+    57 => :RKEY,
+    58 => :TALINK,
+
+    99  => :SPF,
+    100 => :UINFO,
+    101 => :UID,
+    102 => :GID,
+    103 => :UNSPEC,
+
+    249 => :TKEY,
+    250 => :TSIG,
+    251 => :IXFR,
+
+    32768 => :TA,
+    32769 => :DLV
+  }
+
+  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 type.')
+    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