faildns 0.0.1

data/lib/faildns/header/opcode.rb

@@ -0,0 +1,95 @@
+#--
+# 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 Header
+
+#--
+# OPCODE          A four bit field that specifies kind of query in this
+#                 message.  This value is set by the originator of a query
+#                 and copied into the response.  The values are:
+# 
+#                 0               a standard query (QUERY)
+# 
+#                 1               an inverse query (IQUERY)
+# 
+#                 2               a server status request (STATUS)
+# 
+#                 3-15            reserved for future use
+#++
+
+class Opcode
+  Values = {
+    0  => :QUERY,
+    1  => :IQUERY,
+    2  => :STATUS,
+    3  => :RESERVED3,
+    4  => :NOTIFY,
+    5  => :UPDATE,
+    6  => :RESERVED6,
+    7  => :RESERVED7,
+    8  => :RESERVED8,
+    9  => :RESERVED9,
+    10 => :RESERVED10,
+    11 => :RESERVED11,
+    12 => :RESERVED12,
+    13 => :RESERVED13,
+    14 => :RESERVED14,
+    15 => :RESERVED15
+  }
+
+  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 == (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
+
+end

data/lib/faildns/header/status.rb

@@ -0,0 +1,123 @@
+#--
+# 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 Header
+
+#--
+# RCODE           Response code - this 4 bit field is set as part of
+#                 responses.  The values have the following
+#                 interpretation:
+# 
+#                 0               No error condition
+# 
+#                 1               Format error - The name server was
+#                                 unable to interpret the query.
+# 
+#                 2               Server failure - The name server was
+#                                 unable to process this query due to a
+#                                 problem with the name server.
+# 
+#                 3               Name Error - Meaningful only for
+#                                 responses from an authoritative name
+#                                 server, this code signifies that the
+#                                 domain name referenced in the query does
+#                                 not exist.
+# 
+#                 4               Not Implemented - The name server does
+#                                 not support the requested kind of query.
+# 
+#                 5               Refused - The name server refuses to
+#                                 perform the specified operation for
+#                                 policy reasons.  For example, a name
+#                                 server may not wish to provide the
+#                                 information to the particular requester,
+#                                 or a name server may not wish to perform
+#                                 a particular operation (e.g., zone
+#                                 transfer) for particular data.
+# 
+#                 6-15            Reserved for future use.
+#++
+
+class Status
+  Values = {
+    0  => :NOERROR,
+    1  => :FORMERR,
+    2  => :SERVFAIL,
+    3  => :NXDOMAIN,
+    4  => :NOTIMP,
+    5  => :REFUSED,
+    6  => :YXDOMAIN,
+    7  => :YXRRSET,
+    8  => :NXRRSET,
+    9  => :NOTAUTH,
+    10 => :NOTZONE,
+    11 => :RESERVED11,
+    12 => :RESERVED12,
+    13 => :RESERVED13,
+    14 => :RESERVED14,
+    15 => :RESERVED15,
+    16 => :BADSIG,
+    17 => :BADKEY,
+    18 => :BADTIME,
+    19 => :BADMODE,
+    20 => :BADNAME,
+    21 => :BADALG,
+    22 => :BADTRUNC
+  }
+
+  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 == (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
+
+end

data/lib/faildns/header/type.rb

@@ -0,0 +1,72 @@
+#--
+# 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 Header
+
+#--
+# QR              A one bit field that specifies whether this message is a
+#                 query (0), or a response (1).
+#++
+
+class Type
+  Values = {
+    0 => :QUERY,
+    1 => :RESPONSE
+  }
+
+  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 == (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
+
+end

data/lib/faildns/ip.rb

@@ -0,0 +1,57 @@
+#--
+# 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 'ipaddr'
+
+module DNS
+
+class IP
+  def self.parse (string)
+    return IP.new(IPAddr.new_ntoh(string))
+  end
+
+  attr_reader :version, :ip
+
+  def initialize (what)
+    if what.is_a?(String)
+      @ip = IPAddr.new(what.to_s)
+    elsif what.is_a?(IP)
+      @ip = what.ip
+    elsif what.is_a?(IPAddr)
+      @ip = what
+    else
+      DNS.debug what.inspect
+      raise ArgumentError.new 'wat is dis i dont even'
+    end
+
+    @version = (@ip.ipv4?) ? 4 : 6
+  end
+
+  def pack
+    @ip.hton
+  end
+
+  def to_s
+    @ip.to_s
+  end
+
+  alias to_str to_s
+end
+
+end

data/lib/faildns/message.rb

@@ -0,0 +1,100 @@
+#--
+# 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/common'
+require 'faildns/header'
+require 'faildns/question'
+require 'faildns/resourcerecord'
+
+module DNS
+
+class Message
+  def self.parse (string)
+    string.force_encoding 'BINARY'
+    original = string.clone
+
+    header = Header.parse(string);
+
+    questions = []
+    1.upto(header.questions) {
+      questions << Question.parse(string, original);
+    }
+
+    answers = []
+    1.upto(header.answers) {
+      answers << ResourceRecord.parse(string, original);
+    }
+
+    authorities = []
+    1.upto(header.authorities) {
+      authorities << ResourceRecord.parse(string, original);
+    }
+
+    additionals = []
+    1.upto(header.additionals) {
+      additionals << ResourceRecord.parse(string, original);
+    }
+
+    Message.new(header, questions, answers, authorities, additionals)
+  end
+
+  def self.length (string)
+    string = string.clone
+  end
+
+  attr_reader :header, :questions, :answers, :authorities, :additionals
+
+  def initialize (*args)
+    if args.length > 1
+      @header, @questions, @answers, @authorities, @additionals = *(args.concat([[], [], [], []]))
+    else
+      raise ArgumentError.new('You have to pass at least 1 parameter.')
+    end
+
+    if block_given?
+      yield self
+    end
+  end
+
+  def pack
+    result = ''
+
+    result += @header.pack
+
+    @questions.each {|question|
+      result += question.pack
+    }
+
+    @answers.each {|answer|
+      result += answer.pack
+    }
+
+    @authorities.each {|authority|
+      result += answer.pack
+    }
+
+    @additionals.each {|additional|
+      result += additional.pack
+    }
+
+    return result
+  end
+end
+
+end