lib-dhcp 0.1.0

data/lib/lib/dhcp/messages/lease_unassigned.rb

@@ -0,0 +1,64 @@
+#Author: Piotr Wojcieszonek
+#e-mail: piotr@wojcieszonek.pl
+# Copyright 25.03.2016 by Piotr Wojcieszonek
+
+
+require 'lib/dhcp'
+
+module Lib
+  module DHCP
+    class Message
+      class LeaseUnassigned < Message
+        def initialize(htype: 1, hlen: 6, hops: 0, xid: nil, secs: 0, flags: 0, ciaddr: 0, yiaddr: 0, siaddr: 0, giaddr: 0, chaddr: , sname: '.', file: '.', options: nil)
+          super(
+              :op => BOOTREPLY,
+              :htype => htype,
+              :hlen => hlen,
+              :hops => hops,
+              :xid => xid,
+              :secs => secs,
+              :flags => flags,
+              :ciaddr => ciaddr,
+              :yiaddr => yiaddr,
+              :siaddr => siaddr,
+              :giaddr => giaddr,
+              :chaddr => chaddr,
+              :sname => sname,
+              :file => file
+          )
+          if options.is_a? Array or options.is_a? Lib::DHCP::Options
+            options.each { |option| self.options.add option unless option.oid.to_i == Option::MESSAGE_TYPE }
+          elsif options.is_a? Lib::DHCP::Option
+            self.options.add options
+          elsif !options.nil?
+            raise TypeError, "Can't convert #{options.class.name} to Lib::DHCP::Option"
+          end
+          self.options.add Lib::DHCP::Option53.new(LEASE_UNASSIGNED)
+        end
+
+        def pack
+          # noinspection RubyResolve
+          self.option53 = LEASE_UNASSIGNED
+          self.op = BOOTREPLY
+          sanity_check
+          super
+        end
+
+        def self.unpack(packet)
+          res = super(packet)
+          raise TypeError, "No implicit conversion of #{res.class.name} into #{self.class.name}" unless res.is_a? self.class
+          res
+        end
+
+        protected :op=
+        # noinspection RubyResolve
+        protected :option53=
+
+        protected
+        def sanity_check
+          #TODO LeaseUnassigned Sanity Check
+        end
+      end
+    end
+  end
+end

data/lib/lib/dhcp/messages/lease_unknown.rb

@@ -0,0 +1,64 @@
+#Author: Piotr Wojcieszonek
+#e-mail: piotr@wojcieszonek.pl
+# Copyright 25.03.2016 by Piotr Wojcieszonek
+
+
+require 'lib/dhcp'
+
+module Lib
+  module DHCP
+    class Message
+      class LeaseUnknown < Message
+        def initialize(htype: 1, hlen: 6, hops: 0, xid: nil, secs: 0, flags: 0, ciaddr: 0, yiaddr: 0, siaddr: 0, giaddr: 0, chaddr: , sname: '.', file: '.', options: nil)
+          super(
+              :op => BOOTREPLY,
+              :htype => htype,
+              :hlen => hlen,
+              :hops => hops,
+              :xid => xid,
+              :secs => secs,
+              :flags => flags,
+              :ciaddr => ciaddr,
+              :yiaddr => yiaddr,
+              :siaddr => siaddr,
+              :giaddr => giaddr,
+              :chaddr => chaddr,
+              :sname => sname,
+              :file => file
+          )
+          if options.is_a? Array or options.is_a? Lib::DHCP::Options
+            options.each { |option| self.options.add option unless option.oid.to_i == Option::MESSAGE_TYPE }
+          elsif options.is_a? Lib::DHCP::Option
+            self.options.add options
+          elsif !options.nil?
+            raise TypeError, "Can't convert #{options.class.name} to Lib::DHCP::Option"
+          end
+          self.options.add Lib::DHCP::Option53.new(LEASE_UNKNOWN)
+        end
+
+        def pack
+          # noinspection RubyResolve
+          self.option53 = LEASE_UNKNOWN
+          self.op = BOOTREPLY
+          sanity_check
+          super
+        end
+
+        def self.unpack(packet)
+          res = super(packet)
+          raise TypeError, "No implicit conversion of #{res.class.name} into #{self.class.name}" unless res.is_a? self.class
+          res
+        end
+
+        protected :op=
+        # noinspection RubyResolve
+        protected :option53=
+
+        protected
+        def sanity_check
+          #TODO LeaseUnknown Sanity Check
+        end
+      end
+    end
+  end
+end

data/lib/lib/dhcp/messages/message.rb

@@ -0,0 +1,285 @@
+#Author: Piotr Wojcieszonek
+#e-mail: piotr@wojcieszonek.pl
+# Copyright 24.03.2016 by Piotr Wojcieszonek
+
+
+require 'lib/dhcp'
+
+module Lib
+  module DHCP
+    class Message < Packet
+
+      BOOTREQUEST       = 1
+      BOOTREPLY         = 2
+
+      DISCOVER          = 1
+      OFFER             = 2
+      REQUEST           = 3
+      DECLINE           = 4
+      ACKNOWLEDGE       = 5
+      NOT_ACKNOWLEDGE   = 6
+      RELEASE           = 7
+      INFORM            = 8
+      LEASE_QUERY       = 10
+      LEASE_UNASSIGNED  = 11
+      LEASE_UNKNOWN     = 12
+      LEASE_ACTIVE      = 13
+
+      NAME = {
+          1   => 'DHCP Discover',
+          2   => 'DHCP Offer',
+          3   => 'DHCP Request',
+          4   => 'DHCP Decline',
+          5   => 'DHCP ACK',
+          6   => 'DHCP NAK',
+          7   => 'DHCP Release',
+          8   => 'DHCP Inform',
+          9   => '',
+          10  => 'DHCP LeaseQuery',
+          11  => 'DHCP LeaseUnassigned',
+          12  => 'DHCP LeaseUnknown',
+          13  => 'DHCP LeaseActive'
+      }
+
+      def name
+        if option53.nil?
+          nil
+        else
+          NAME[option53.to_i]
+        end
+      end
+
+      def self.unpack(packet)
+        packet =  super(packet)
+        # noinspection RubyResolve
+        res = case packet.option53.value.to_i
+                when DISCOVER
+                  Lib::DHCP::Message::Discover.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when OFFER
+                  Lib::DHCP::Message::Offer.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when REQUEST
+                  Lib::DHCP::Message::Request.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+
+                when DECLINE
+                  Lib::DHCP::Message::Decline.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when ACKNOWLEDGE
+                  Lib::DHCP::Message::ACK.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when NOT_ACKNOWLEDGE
+                  Lib::DHCP::Message::NACK.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when RELEASE
+                  Lib::DHCP::Message::Release.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when INFORM
+                  Lib::DHCP::Message::Inform.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  =>  packet.options
+                  )
+                when LEASE_QUERY
+                  Lib::DHCP::Message::LeaseQuery.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when LEASE_UNASSIGNED
+                  Lib::DHCP::Message::LeaseUnassigned.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when LEASE_UNKNOWN
+                  Lib::DHCP::Message::LeaseUnknown.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                when LEASE_ACTIVE
+                  Lib::DHCP::Message::LeaseActive.new(
+                      :htype    => packet.htype,
+                      :hlen     => packet.hlen,
+                      :hops     => packet.hops,
+                      :xid      => packet.xid,
+                      :secs     => packet.secs,
+                      :flags    => packet.flags,
+                      :ciaddr   => packet.ciaddr,
+                      :yiaddr   => packet.yiaddr,
+                      :siaddr   => packet.siaddr,
+                      :giaddr   => packet.giaddr,
+                      :chaddr   => packet.chaddr,
+                      :sname    => packet.sname,
+                      :file     => packet.file,
+                      :options  => packet.options
+                  )
+                else
+                  raise ArgumentError, "Unknown DHCP MESSAGE TYPE #{packet.option53.to_i}"
+              end
+        res.send(:sanity_check)
+        res
+      end
+
+      def pack
+        sanity_check
+        super
+      end
+
+      # protected :op=
+      # # noinspection RubyResolve
+      # protected :option53=
+
+
+      protected
+
+      def sanity_check
+        raise Lib::DHCP::SanityCheck, "Can't pack DHCP Message without Option 53 - Message Type set" if self.option53.nil?
+      end
+    end
+  end
+end

data/lib/lib/dhcp/messages/nack.rb

@@ -0,0 +1,64 @@
+#Author: Piotr Wojcieszonek
+#e-mail: piotr@wojcieszonek.pl
+# Copyright 25.03.2016 by Piotr Wojcieszonek
+
+
+require 'lib/dhcp'
+
+module Lib
+  module DHCP
+    class Message
+      class NACK < Message
+        def initialize(htype: 1, hlen: 6, hops: 0, xid: nil, secs: 0, flags: 0, ciaddr: 0, yiaddr: 0, siaddr: 0, giaddr: 0, chaddr: , sname: '.', file: '.', options: nil)
+          super(
+              :op => BOOTREPLY,
+              :htype => htype,
+              :hlen => hlen,
+              :hops => hops,
+              :xid => xid,
+              :secs => secs,
+              :flags => flags,
+              :ciaddr => ciaddr,
+              :yiaddr => yiaddr,
+              :siaddr => siaddr,
+              :giaddr => giaddr,
+              :chaddr => chaddr,
+              :sname => sname,
+              :file => file
+          )
+          if options.is_a? Array or options.is_a? Lib::DHCP::Options
+            options.each { |option| self.options.add option unless option.oid.to_i == Option::MESSAGE_TYPE }
+          elsif options.is_a? Lib::DHCP::Option
+            self.options.add options
+          elsif !options.nil?
+            raise TypeError, "Can't convert #{options.class.name} to Lib::DHCP::Option"
+          end
+          self.options.add Lib::DHCP::Option53.new(NOT_ACKNOWLEDGE)
+        end
+
+        def pack
+          # noinspection RubyResolve
+          self.option53 = NOT_ACKNOWLEDGE
+          self.op = BOOTREPLY
+          sanity_check
+          super
+        end
+
+        def self.unpack(packet)
+          res = super(packet)
+          raise TypeError, "No implicit conversion of #{res.class.name} into #{self.class.name}" unless res.is_a? self.class
+          res
+        end
+
+        protected :op=
+        # noinspection RubyResolve
+        protected :option53=
+
+        protected
+        def sanity_check
+          #TODO NACK Sanity Check
+        end
+      end
+    end
+  end
+end