ospfv2 0.0.2 → 0.0.3

data/lib/lsa/tlv/maximum_bandwidth.rb

@@ -0,0 +1,55 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+#FIXME: make bw an IE
+
+require_relative 'tlv'
+
+module OSPFv2
+
+  class MaximumBandwidth_Tlv
+    include SubTlv
+    include Common
+
+    attr_reader :tlv_type, :max_bw
+
+    def initialize(arg={})
+      @tlv_type = 6
+      @max_bw = 0
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 4, @max_bw/8.0].pack('nng')
+    end
+
+    def __parse(s)
+      @tlv_type, _, max_bw = s.unpack('nng')
+      @max_bw = (max_bw * 8).to_int
+    end
+
+    def to_s
+      "Maximum bandwidth : #{max_bw.to_i}"
+    end
+
+    def to_s_junos_style(ident=0)
+      "  "*ident + "RtrAddr (1), length #{@length}: #{router_id.to_ip}"
+    end
+
+  end
+end
+

data/lib/lsa/tlv/maximum_reservable_bandwidth.rb

@@ -0,0 +1,52 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'tlv'
+
+module OSPFv2
+
+  class MaximumReservableBandwidth_Tlv
+    include SubTlv
+    include Common
+
+    attr_reader :tlv_type, :max_resv_bw
+
+    def initialize(arg={})
+      @tlv_type, = 7
+      @max_resv_bw = 0
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 4, @max_resv_bw/8.0].pack('nng')
+    end
+
+    def __parse(s)
+      @tlv_type, _, max_resv_bw = s.unpack('nng')
+      @max_resv_bw = (max_resv_bw*8).to_int
+    end
+
+    def to_s
+      "Maximum reservable bandwidth : #{max_resv_bw}"
+    end
+
+    def to_s_junos_style(ident=0)
+      "  "*ident + "RtrAddr (1), length #{@length}: #{router_id.to_ip}"
+    end
+
+  end
+end

data/lib/lsa/tlv/remote_interface.rb

@@ -0,0 +1,58 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'tlv'
+
+#TODO interface_adrress base tlv : DRY
+
+module OSPFv2
+
+  class RemoteInterfaceIpAddress_Tlv
+    include SubTlv
+    include Common
+
+    IpAddress = Class.new(Id) unless const_defined?(:IpAddress)
+    attr_reader :tlv_type, :ip_address
+
+    attr_writer_delegate :ip_address
+
+    def initialize(arg={})
+      @tlv_type = 4
+      @ip_address = IpAddress.new
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 4, @ip_address.encode].pack('nna*')
+    end
+
+    def __parse(s)
+      @tlv_type, _, ip_address = s.unpack('nna*')
+      @ip_address = IpAddress.new_ntoh(ip_address)
+    end
+
+
+    def to_s
+      "Neighbor Address : #{ip_address.to_ip}"
+    end
+
+    def to_s_junos_style(ident=0)
+      "  "*ident + "RtrAddr (1), length #{@length}: #{router_id.to_ip}"
+    end
+
+  end
+end

data/lib/lsa/tlv/router_address.rb

@@ -0,0 +1,57 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'tlv'
+
+module OSPFv2
+
+  class RouterAddress_Tlv
+    include SubTlv
+    include Common
+
+    IpAddress = Class.new(Id) unless const_defined?(:IpAddress)
+    attr_reader :tlv_type, :ip_address
+
+    attr_writer_delegate :ip_address
+
+    def initialize(arg={})
+      @tlv_type = 1
+      @ip_address = IpAddress.new
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 4, @ip_address.encode].pack('nna*')
+    end
+
+    def __parse(s)
+      @tlv_type, _, ip_address = s.unpack('nna*')
+      @ip_address = IpAddress.new_ntoh(ip_address)
+    end
+
+
+    def to_s
+      "RouterID TLV: #{ip_address.to_ip}"
+    end
+
+    def to_s_junos_style(ident=0)
+      "  "*ident + "RtrAddr (1), length #{@length}: #{ip_address.to_ip}"
+    end
+
+  end
+end
+

data/lib/lsa/tlv/tlv.rb

@@ -0,0 +1,32 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative '../../infra/ospf_common'
+require_relative '../../ie/id'
+
+module OSPFv2
+  module Tlv
+    module Common
+      def stlv_len(n)
+        (((n+3)/4)*4)+4
+      end
+      def tlv_len
+        @length
+      end
+      def to_hash
+        {:tlv_type=> tlv_type}
+      end
+    end
+  end
+  module SubTlv
+    include Tlv
+  end
+end
+

data/lib/lsa/tlv/tlv_factory.rb

@@ -0,0 +1,72 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'router_address'
+require_relative 'link_type'
+
+module OSPFv2::Tlv
+  def self.factory(arg={})
+    type=0
+    case arg
+    when Hash
+      return nil unless arg.has_key?(:tlv_type)
+      type = arg[:tlv_type]
+    when String
+      type = arg.unpack('n')[0]
+      return nil unless type
+    else
+      raise
+    end
+    case type
+    when 1 ; OSPFv2::RouterAddress_Tlv.new(arg.dup)
+    when 2 ; OSPFv2::Link_Tlv.new(arg.dup)
+    else
+      raise
+    end
+  end
+end
+
+require_relative 'link'
+require_relative 'link_type'
+require_relative 'link_id'
+require_relative 'local_interface'
+require_relative 'remote_interface'
+require_relative 'maximum_bandwidth'
+require_relative 'maximum_reservable_bandwidth'
+require_relative 'unreserved_bandwidth'
+require_relative 'traffic_engineering_metric'
+require_relative 'color.rb'
+
+module OSPFv2::SubTlv
+  def self.factory(arg={})
+    if arg.is_a?(Hash)
+      return nil if arg[:tlv_type].nil?
+      type = arg[:tlv_type]
+    elsif arg.is_a?(String)
+      type = arg.unpack('n')[0]
+    else
+      return
+    end
+    case type
+    when 1 ; OSPFv2::LinkType_Tlv.new(arg)
+    when 2 ; OSPFv2::LinkId_Tlv.new(arg)
+    when 3 ; OSPFv2::LocalInterfaceIpAddress_Tlv.new(arg)
+    when 4 ; OSPFv2::RemoteInterfaceIpAddress_Tlv.new(arg)
+    when 5 ; OSPFv2::TrafficEngineeringMetric_Tlv.new(arg)
+    when 6 ; OSPFv2::MaximumBandwidth_Tlv.new(arg)
+    when 7 ; OSPFv2::MaximumReservableBandwidth_Tlv.new(arg)
+    when 8 ; OSPFv2::UnreservedBandwidth_Tlv.new(arg)
+    when 9 ; OSPFv2::Color_Tlv.new(arg)
+    else
+      raise
+    end
+  end
+end
+

data/lib/lsa/tlv/traffic_engineering_metric.rb

@@ -0,0 +1,52 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'tlv'
+
+module OSPFv2
+
+  class TrafficEngineeringMetric_Tlv
+    include SubTlv
+    include Common
+
+    attr_reader :tlv_type, :te_metric
+
+    def initialize(arg={})
+      @tlv_type,  = 5
+      @te_metric = 0
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 4, @te_metric].pack('nnN')
+    end
+
+    def __parse(s)
+      @tlv_type, _, @te_metric = s.unpack('nnN')
+    end
+
+
+    def to_s
+      "Admin Metric : #{te_metric}"
+    end
+
+    def to_s_junos_style(ident=0)
+      "  "*ident + "RtrAddr (1), length #{@length}: #{router_id.to_ip}"
+    end
+
+  end
+end

data/lib/lsa/tlv/unreserved_bandwidth.rb

@@ -0,0 +1,55 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+#
+#++
+
+require_relative 'tlv'
+
+module OSPFv2
+
+  class UnreservedBandwidth_Tlv
+    include SubTlv
+    include Common
+
+    attr_reader :tlv_type, :unreserved_bw
+
+    def initialize(arg={})
+      @tlv_type = 8
+      @unreserved_bw = [0]*8
+
+      if arg.is_a?(Hash) then
+        set(arg.dup)
+      elsif arg.is_a?(String)
+        __parse(arg)
+      else
+        raise ArgumentError, "Invalid argument", caller
+      end
+    end
+
+    def encode
+      [@tlv_type, 32].pack('nn') +
+      unreserved_bw.collect { |bw|   bw / 8.0 }.pack('g*')
+    end
+
+    def __parse(s)
+      @tlv_type, _, *unreserved_bw = s.unpack('nng*')
+      @unreserved_bw = unreserved_bw.collect {|bw| (bw*8).to_i }
+    end
+
+    def to_hash
+      h=super
+      h[:unreserved_bw] = unreserved_bw
+      h
+    end
+    def to_s
+      "Unreserved bandwidth : " + unreserved_bw.collect { |bw| "%s" % bw }.join(", ")
+    end
+
+  end
+end
+

data/lib/lsa/traffic_engineering.rb

@@ -0,0 +1,95 @@
+#--
+# Copyright 2011 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+#++
+
+require_relative 'opaque'
+require 'ie/opaque_id'
+require 'ie/opaque_type'
+require_relative 'tlv/tlv_factory'
+
+module OSPFv2
+  class TrafficEngineering < Lsa
+    include Tlv
+    #FIXME: move this under OpaqueId ?????
+    @_opaque_id = 0
+    
+    class << self
+      attr_accessor :_opaque_id
+      def opaque_id
+        @_opaque_id +=1
+      end
+      def reset_opaque_id
+        @_opaque_id=0
+      end
+      def new_hash(h)
+        r = new(h.dup)
+        r
+      end
+    end
+    
+    attr_accessor :top_lvl_tlv, :opaque_id
+
+    def initialize(_arg={})
+      arg = _arg.dup
+      @ls_type = LsType.new(:area)
+      case arg
+      when Hash
+        super()
+          set arg
+      when String
+        parse arg
+      end
+    end
+
+    def set(h)
+      return if h.empty?
+      if h.has_key?(:top_lvl_tlv)
+        tlv = h.delete(:top_lvl_tlv)
+        case tlv
+        when Hash
+          @top_lvl_tlv = OSPFv2::Tlv.factory(tlv)
+        when Tlv
+          @top_lvl_tlv = tlv
+        else
+          raise
+        end
+      end
+      super
+    end
+
+    def encode
+      if top_lvl_tlv
+        super top_lvl_tlv.encode
+      else
+        super
+      end
+    end
+
+    def to_s_ios
+      super + @opaque_id.to_i.to_s
+    end
+
+    def to_s_ios_verbose
+      to_s_verbose # for now
+    end
+
+    def to_s_verbose
+      s = []
+      s << super
+      s << top_lvl_tlv.to_s(:ident=>5)
+      s.join("\n   ")
+    end
+
+    def parse(s)
+      @top_lvl_tlv = Tlv.factory(super(s))
+    end
+    
+  end
+  
+end