ospfv2 0.0.2 → 0.0.3

data/test/unit/ie/router_link_type_test.rb

@@ -0,0 +1,24 @@
+
+require "test/unit"
+require 'ie/router_link_type'
+
+
+class TestIeRouterLinkType < Test::Unit::TestCase
+  include OSPFv2
+  def test_new
+    assert RouterLinkType.new :point_to_point
+    assert_equal 'RouterLinkType: point_to_point', RouterLinkType.new(:point_to_point).to_s
+    assert_equal 'RouterLinkType: transit_network', RouterLinkType.new(:transit_network).to_s
+    assert_equal 'RouterLinkType: stub_network', RouterLinkType.new(:stub_network).to_s
+    assert_equal 'RouterLinkType: virtual_link', RouterLinkType.new(:virtual_link).to_s
+    assert_equal 1, RouterLinkType.new(:point_to_point).to_i
+    assert_equal 2, RouterLinkType.new(:transit_network).to_i
+    assert_equal 3, RouterLinkType.new(:stub_network).to_i
+    assert_equal 4, RouterLinkType.new(:virtual_link).to_i
+    assert_equal :point_to_point, RouterLinkType.new(1).to_sym
+    assert_equal :stub_network, RouterLinkType.new(3).to_sym
+    assert_equal :transit_network, RouterLinkType.new(2).to_sym
+    assert_equal :virtual_link, RouterLinkType.new(4).to_sym
+  end
+end
+

data/test/unit/ie/sequence_number_test.rb

@@ -0,0 +1,31 @@
+require "test/unit"
+
+require "ie/sequence_number"
+
+class TestIeSequenceNumber < Test::Unit::TestCase
+  include OSPFv2
+  def test1
+    sn = SequenceNumber.new()
+    sn_init = SequenceNumber.new(:init)
+    sn_max= SequenceNumber.new(:max)
+    sn_resv= SequenceNumber.new(:reserved)
+    assert_equal("0x80000001",sn.to_s)
+    assert_equal("0x80000001",sn_init.to_s)
+    assert_equal("0x7fffffff",sn_max.to_s)
+    assert_equal("0x80000000",sn_resv.to_s)
+    assert_equal(-1,sn_init <=> sn_max)
+    assert_equal(-1,sn_resv <=> sn_init)
+    assert_equal(1,sn_max <=> sn_init)
+    assert_equal(true,sn_init < sn_max)
+    assert_equal(true,sn_resv  < sn_init)
+    assert_equal(true,sn_max > sn_init)
+    assert_equal("0x80000003",(sn_init + 2).to_s)
+    assert_equal("0x80000001",(sn_init - 2).to_s)
+    assert_equal("0x80000002",SequenceNumber.to_s(0x80000002))
+    assert_equal("0x00000001",SequenceNumber.to_s(1))
+  
+    assert_equal('0x80000001', SequenceNumber.new(SequenceNumber.new).to_s)
+    
+    
+  end
+end

data/test/unit/ie/tos_metric_test.rb

@@ -0,0 +1,27 @@
+require "test/unit"
+require "ie/tos_metric"
+
+class TestIeTosMetric < Test::Unit::TestCase
+  include OSPFv2
+  def tests
+    assert TosMetric.new
+    assert_equal( '00000000', TosMetric.new.to_shex)
+    assert_equal( 'TosMetric: tos: 0 cost: 0', TosMetric.new.to_s)
+    assert_equal( '01000001', TosMetric.new(:tos => 1, :cost => 1).to_shex)
+    assert_equal( '01ffffff', TosMetric.new(:tos => 1, :cost => 0xffffff).to_shex)
+    assert_equal( '01ffffff', TosMetric.new(:tos => 1, :cost => 0xffffffff).to_shex)
+    assert_equal( 1, TosMetric.new(:tos => 1, :cost => 0xffffffff).tos)
+    assert_equal( 4294967295, TosMetric.new(:tos => 1, :cost => 0xffffffff).cost)
+    assert_equal( {:tos=>1, :cost=>1}, TosMetric.new(:tos => 1, :cost => 1).to_hash)
+    assert_equal( 'TosMetric: tos: 1 cost: 1', TosMetric.new(:tos => 1, :cost => 1).to_s)
+    assert_equal( 'TosMetric: tos: 1 cost: 1', TosMetric.new(:tos => 1, :cost => 1).to_s)
+    assert_equal( 'TosMetric: tos: 1 cost: 1', TosMetric.new([1,1]).to_s)
+    assert_equal( 'Topology (ID 1) -> Metric: 1', TosMetric.new([1,1]).to_s_junos_style)
+    assert_equal( TosMetric.new([1,1]).to_shex, TosMetric.new(TosMetric.new([1,1])).to_shex)
+    m = TosMetric.new(['01ffffff'].pack('H*'))
+    assert_equal( '01ffffff', m.to_shex)
+    assert_equal( 1, m.tos)
+    assert_equal( 1, m.id)
+    assert_equal( 0xffffff, m.cost)
+  end
+end

data/test/unit/infra/ospf_common_test.rb

@@ -0,0 +1,42 @@
+require "test/unit"
+
+require "infra/ospf_common"
+
+class TestCommon < Test::Unit::TestCase
+  def test_class_string_to__
+    assert_equal 'camel_case', "CamelCase".to_underscore
+    assert_equal 'router_id', "RouterId".to_underscore
+    assert_equal 'area_id', "AreaId".to_underscore
+    assert_equal 'ospf_bdr', "OspfBdr".to_underscore
+    assert_equal 'ospf_dr', "OspfDr".to_underscore
+    assert_equal 'as_external7', "AsExternal7".to_underscore
+  end
+  def test_class_string_to_camel
+    assert_equal 'CamelCase', 'camel_case'.to_camel
+    assert_equal 'RouterId', 'router_id'.to_camel
+    assert_equal 'AreaId', 'area_id'.to_camel
+  end
+  def test_class_symbol_to_klass
+    assert_equal :AnInstanceVar, :an_instance_var.to_klass
+    assert_equal :AreaId, :area_id.to_klass
+    assert_equal :RouterId, :router_id.to_klass
+  end
+  def test_attr_writer_delegate
+  end
+  def test_ipaddr
+    assert_equal IPAddr, IPAddr.to_arr('10.0.0.1/24')[0].class
+    assert_equal '10.0.0.1', IPAddr.to_arr('10.0.0.1/24')[1]
+    assert_equal 24, IPAddr.to_arr('10.0.0.1/24')[2]
+    assert_equal '10.0.0.0', IPAddr.to_arr('10.0.0.1/24')[3]
+    assert_equal '255.255.255.0', IPAddr.to_arr('10.0.0.1/24')[4]
+    assert_equal '10.0.0.8/30', IPAddr.new('10.0.0.0/30') ^ 2
+    assert_equal '10.0.0.12/30', IPAddr.new('10.0.0.0/30') ^ 3
+    assert_equal '10.0.0.1/30', IPAddr.new('10.0.0.0/30') + 1
+    assert_equal '10.0.0.2/30', IPAddr.new('10.0.0.0/30') + 2
+    assert_raise(NoMethodError)  { IPAddr.new('10.0.0.0/30') -1 }
+    
+    assert_equal '10.0.0.0', IPAddr.new('10.0.0.1/24').to_s
+    assert_equal '10.0.0.0/24', IPAddr.new('10.0.0.1/24').to_s_net
+    
+  end
+end

data/test/unit/infra/parse_options_test.rb

@@ -0,0 +1,58 @@
+
+require "test/unit"
+
+require "infra/parse_options"
+
+class TestOptParse < Test::Unit::TestCase
+
+  def test_parse_grid
+    assert_equal [2,2], parse.grid
+    assert_equal [3,4], parse('--grid 3x4').grid
+    assert_equal [3,4], parse('-g 3x4').grid
+  end
+  def test_parse_area_id
+    assert_equal 0, parse.area_id
+    assert_equal 1, parse('--area-id 1').area_id
+    assert_equal 2, parse('--area-id 0.0.0.2').area_id
+    assert_equal 2, parse('-a 0.0.0.2').area_id
+  end
+  def test_parse_router_id
+    assert_equal 1, parse.router_id
+    assert_equal 2, parse('--router-id 2').router_id
+    assert_equal 2, parse('-r 0.0.0.2').router_id
+  end
+  def test_parse_ipaddr
+    assert_equal '192.168.1.123' ,parse.ipaddr
+    assert_equal '1.1.1.1' ,parse('--address 1.1.1.1/23').ipaddr
+    assert_equal '255.255.254.0' ,parse('--address 1.1.1.1/23').netmask
+  end
+  def test_hello_int
+    assert_equal 10, parse.hello_int
+    assert_equal 60, parse('--hello-int 60').hello_int
+    assert_equal 40, parse.dead_int
+    assert_equal 20, parse('--hello-int 20').hello_int
+    assert_equal 80, parse('--hello-int 20').dead_int
+    assert_equal 200, parse('--hello-int 20 --dead-int 200').dead_int
+    assert_equal 200, parse('--dead-int 200').dead_int
+    assert_equal 10, parse('--dead-int 200').hello_int
+  end
+  def test_num_of
+    assert_equal 10, parse.num_sum
+    assert_equal 10, parse.num_ext
+    assert_equal 20, parse('--number-of-summary 20').num_sum
+    assert_equal 20, parse('--number-of-external 20').num_ext
+  end
+
+  private
+  
+  def set(a, v)
+    a.clear
+    v.each { |x| a << x }
+  end
+  
+  def parse(s='')
+    set ARGV, s.split
+    OptParse.parse(ARGV)
+  end
+
+end

data/test/unit/infra/timer_test.rb

@@ -0,0 +1,94 @@
+require "test/unit"
+
+require "infra/timer"
+
+class TestInfraTimer < Test::Unit::TestCase
+
+  Timer1 = Class.new(Timer)
+  Timer2 = Class.new(Timer)
+  Timer3 = Class.new(Timer)
+
+  def update(*args)
+    @evQ.enq args
+  end
+
+  def setup
+    @evQ = Queue.new
+  end
+
+  def _tests
+    assert Timer.new(10)
+    t0 = Time.now
+    # + notification
+    t1 = Timer1.start(1,self)
+    assert_equal [:ev_timer_start, :"TestTimer::Timer1"], @evQ.deq
+    assert_equal [:ev_timer_expire, :"TestTimer::Timer1"], @evQ.deq
+    # - notification
+    # + block
+    t2 = Timer2.start(1) { @evQ.enq ['DONE', (Time.now-t0).to_int]}
+    assert_equal ['DONE',2], @evQ.deq
+    # + notificaion
+    # + block
+    t3 = Timer3.start(1,self) {  @evQ.enq ['DONE', (Time.now-t0).to_int]}
+    assert_equal [:ev_timer_start, :"TestTimer::Timer3"], @evQ.deq
+    assert_equal ['DONE',3], @evQ.deq
+    assert_equal [:ev_timer_expire, :"TestTimer::Timer3"], @evQ.deq
+  end
+  def test_running?
+    t1 = Timer1.new(10)
+    assert ! t1.running?
+    t1.start 1
+    assert t1.running?
+  end
+  def _test_cancel
+    t1 = Timer1.new(10)
+    t1.start 1
+    assert t1.running?
+    t1.cancel
+    assert ! t1.running?
+    t0 = Time.now
+    t1 = Timer.start(3)
+    sleep 2
+    t1.reset { @evQ.enq ['DONE', (Time.now-t0).to_int]}
+    assert_equal ['DONE', 2+3], @evQ.deq
+  end
+end
+
+class TestPeriodicTimer < Test::Unit::TestCase
+
+  PeriodicTimer1 = Class.new(PeriodicTimer)
+
+  def update(*args)
+    @evQ.enq args
+  end
+
+  def setup
+    @evQ = Queue.new
+  end
+
+  def tests
+    t0 = Time.now
+    p = PeriodicTimer1.start(1,self) {  @evQ.enq ['DONE', (Time.now-t0).to_int]}
+    assert_equal [:ev_timer_start, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+    assert_equal ['DONE',1], @evQ.deq
+    assert_equal [:ev_timer_fire, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+    assert_equal ['DONE',2], @evQ.deq
+    assert_equal [:ev_timer_fire, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+    assert_equal ['DONE',3], @evQ.deq
+    assert_equal [:ev_timer_fire, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+  end
+
+  def test_stop
+    t0 = Time.now
+    p = PeriodicTimer1.start(1,self) {  @evQ.enq ['DONE', (Time.now-t0).to_int]}
+    assert_equal [:ev_timer_start, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+    assert_equal ['DONE',1], @evQ.deq
+    assert_equal [:ev_timer_fire, :"TestPeriodicTimer::PeriodicTimer1"], @evQ.deq[0,2]
+    assert p.running?
+    p.stop
+    sleep(1.2)
+    assert ! p.running?
+  end
+
+end
+

data/test/unit/ls_db/all

@@ -0,0 +1,10 @@
+require 'fileutils'
+dir = File.dirname(File.expand_path(__FILE__ ))
+
+FileUtils.chdir(dir) do
+  Dir.glob("**/*rb") do |filename| 
+    next if __FILE__ == filename
+    require File.join(dir,filename) unless filename == File.basename(__FILE__)
+  end
+end
+

data/test/unit/ls_db/link_state_database_build_test.rb

@@ -0,0 +1,140 @@
+require "test/unit"
+require "ls_db/link_state_database"
+require 'packet/ospf_packet'
+
+class TestLsDbLinkStateBuildDatabase < Test::Unit::TestCase
+  include OSPFv2
+  include OSPFv2::LSDB
+
+def _test_add_link_to_stub
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_link_to_stub_network :router_id=> 1, :link_id => '99.99.1.0', :link_data => '255.255.255.0'
+  assert_equal '99.99.1.0', ls_db[1,1][0].link_id.to_ip
+  assert_equal '255.255.255.0', ls_db[1,1][0].link_data.to_ip
+  assert_equal 0, ls_db[1,1][0].metric.to_i
+  ls_db.add_link_to_stub_network :router_id=> 1, :network => '99.99.1.1/24', :metric=>10
+  assert_equal '99.99.1.0', ls_db[1,1][0].link_id.to_ip
+  assert_equal '255.255.255.0', ls_db[1,1][0].link_data.to_ip  
+  assert_equal 10, ls_db[1,1][0].metric.to_i
+end
+
+def test_add_adjacency
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_adjacency(1, 2, '192.168.0.1/24', 2)
+  
+  assert_equal 2, ls_db[:router,1].links.size
+  ls_db.add_adjacency(1, 3, '192.168.1.1/24', 2)
+  assert_equal 4, ls_db[:router,1].links.size
+  ls_db.add_adjacency(2, 3, '192.168.2.1/24', 2)
+  assert_equal 2, ls_db[:router,2].links.size
+  assert_equal 2, ls_db.size
+  assert_equal Router, ls_db[:router, 1].class
+  assert_equal RouterLink::PointToPoint, ls_db[:router,1][0].class
+  
+  assert_equal 4, ls_db[:router,1].links.size    
+  lsa = ls_db.remove_adjacency(1, 2, '192.168.0.1/24')
+  assert_equal 2, ls_db[:router,1].links.size    
+  ls_db.remove_adjacency(1, 3, '192.168.1.1/24')
+  assert_equal 0, ls_db[:router,1].links.size
+  
+  assert_equal 2, ls_db[:router,2].links.size
+  lsa = ls_db.remove_adjacency(2, 3, '192.168.2.1/24')
+  assert_equal 0, ls_db[:router,2].links.size
+  
+  assert ls_db.advertised_routers.has?('0.0.0.1')
+  assert ls_db.advertised_routers.has?('0.0.0.2')
+  assert ! ls_db.advertised_routers.has?('0.0.0.3')
+  assert ls_db.advertised_routers.has?(1)
+  assert ls_db.advertised_routers.has?(2)
+  assert ! ls_db.advertised_routers.has?(3)
+  assert ls_db.proxied?('0.0.0.1')
+  assert ls_db.proxied?(2)
+  assert ! ls_db.proxied?('0.0.0.3')
+end
+
+def test_add_adjacency_from_hash
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_adjacency :router_id =>1, :neighbor_router_id => 4, :prefix => '192.168.4.1/24', :metric=>2
+
+  assert_equal '0.0.0.1', ls_db[1,1].ls_id.to_ip
+  assert_equal 2, ls_db[1,1][0].metric.to_i    
+
+  assert_equal :point_to_point, ls_db[1,1][0].router_link_type.to_sym
+  assert_equal 'LinkId: 0.0.0.4', ls_db[1,1][0].link_id.to_s
+  assert_equal 'LinkData: 192.168.4.1', ls_db[1,1][0].link_data.to_s
+
+  assert_equal :stub_network, ls_db[1,1][1].router_link_type.to_sym
+  assert_equal 'LinkId: 192.168.4.0', ls_db[1,1][1].link_id.to_s
+  assert_equal 'LinkData: 255.255.255.0', ls_db[1,1][1].link_data.to_s
+  #puts ls_db.to_s
+end
+
+def test_1
+  
+  # 
+  # point-to-point:
+  # 
+  #                88.0.0.0/24
+  #    routerA <---- p2p ------> routerB
+  #           .1              .2
+  # 
+  # jme@olive> show ospf database logical-router routerA detail 
+  # 
+  #     OSPF link state database, Area 0.0.0.0
+  #  Type       ID               Adv Rtr           Seq      Age  Opt  Cksum  Len 
+  # Router  *88.0.0.1         88.0.0.1         0x80000003   126  0x22 0xa5bc  48
+  #   bits 0x0, link count 2
+  #   id 88.0.0.2, data 88.0.0.1, Type PointToPoint (1)
+  #     Topology count: 0, Default metric: 1
+  #   id 88.0.0.0, data 255.255.255.0, Type Stub (3)
+  #     Topology count: 0, Default metric: 1
+  # Router   88.0.0.2         88.0.0.2         0x80000002   127  0x22 0x9bc5  48
+  #   bits 0x0, link count 2
+  #   id 88.0.0.1, data 88.0.0.2, Type PointToPoint (1)
+  #     Topology count: 0, Default metric: 1
+  #   id 88.0.0.0, data 255.255.255.0, Type Stub (3)
+  #     Topology count: 0, Default metric: 1
+  # 
+  # 
+  # routerA:  set router-options router-id 0.0.0.10
+  # 
+  #     OSPF link state database, Area 0.0.0.0
+  #  Type       ID               Adv Rtr           Seq      Age  Opt  Cksum  Len 
+  # Router  *10.0.0.0         10.0.0.0         0x80000008  1294  0x22 0x6695  48
+  #   bits 0x0, link count 2
+  #   id 88.0.0.2, data 88.0.0.1, Type PointToPoint (1)
+  #     Topology count: 0, Default metric: 1
+  #   id 88.0.0.0, data 255.255.255.0, Type Stub (3)
+  #     Topology count: 0, Default metric: 1
+  # Router   88.0.0.2         88.0.0.2         0x8000000b  1291  0x22 0x5b4c  48
+  #   bits 0x0, link count 2
+  #   id 10.0.0.0, data 88.0.0.2, Type PointToPoint (1)
+  #     Topology count: 0, Default metric: 1
+  #   id 88.0.0.0, data 255.255.255.0, Type Stub (3)
+  #     Topology count: 0, Default metric: 1
+  # 
+  # 
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_p2p_adjacency :router_id =>'88.0.0.1', :neighbor_router_id => '88.0.0.2', :prefix => '88.0.0.1/24', :metric=>1
+  ls_db.add_p2p_adjacency :router_id =>'88.0.0.2', :neighbor_router_id => '88.0.0.1', :prefix => '88.0.0.2/24', :metric=>1
+
+  # if router_id is not specified use address part of prefix
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_p2p_adjacency :neighbor_router_id => '88.0.0.2', :prefix => '88.0.0.1/24', :metric=>1
+  ls_db.add_p2p_adjacency :neighbor_router_id => '88.0.0.1', :prefix => '88.0.0.2/24', :metric=>1
+
+  # routerA with router-id 10
+  ls_db = LinkStateDatabase.new(:area_id=> 0)
+  ls_db.add_p2p_adjacency :router_id => '10.0.0.0', :neighbor_router_id => '88.0.0.2', :prefix => '88.0.0.1/24', :metric=>1
+  ls_db.add_p2p_adjacency :router_id =>'88.0.0.2', :neighbor_router_id => '10.0.0.0', :prefix => '88.0.0.2/24', :metric=>1
+
+  ls_db.add_loopback :router_id=> '10.0.0.0', :address=>'99.99.1.1'
+  
+  ls_db.add_p2p_adjacency  :neighbor_router_id => '88.0.0.2', :prefix => '192.168.1.200/24', :metric=>1
+  
+  
+    
+end
+
+
+end

data/test/unit/ls_db/link_state_database_links_tests.rb

@@ -0,0 +1,74 @@
+require 'ls_db/link_state_database_links'
+require "test/unit"
+
+# require "ls_db/link_add"
+
+class TestLsDbLinks < Test::Unit::TestCase
+  include OSPFv2::LSDB
+  def setup
+    @ls_db = LinkStateDatabase.new(:area_id=> 0)
+    Link.reset
+  end
+  def test_new_link
+    ls_db = LinkStateDatabase.new :area_id=>0
+    ls_db.new_link :router_id=> 1, :neighbor_id=>2
+    ls_db.new_link :router_id=> 1, :neighbor_id=>3
+    ls_db.new_link :router_id=> 1, :neighbor_id=>4
+    ls_db.new_link :router_id=> 1, :neighbor_id=>5
+
+    Link.all.values.each { |lnk| ls_db.link lnk, :down  }
+    Link.all.values.each { |lnk| ls_db.link lnk, :up  }
+
+    rlsa.delete(:point_to_point,'0.0.0.2')
+    assert_equal 7, rlsa.links.size      
+    puts ls_db
+    rlsa.delete(:point_to_point,'0.0.0.3')
+    rlsa.delete(:point_to_point,'0.0.0.4')
+    rlsa.delete(:point_to_point,'0.0.0.5')
+    assert_equal 4, rlsa.links.size
+    assert_equal 5, ls_db.size
+    ls_db.link Link[1], :down
+  end
+
+  def test_new_link_local_only
+    @ls_db = LinkStateDatabase.new :area_id=>0
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>2, :direction => :local_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>3, :direction => :local_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>4, :direction => :local_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>5, :direction => :local_only
+    assert_equal 1,  @ls_db.size
+    assert_equal 1,  @ls_db.all_router[0].advertising_router.to_i
+  end
+  def test_new_link_remote_only
+    @ls_db = LinkStateDatabase.new :area_id=>0
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>2, :direction => :remote_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>3, :direction => :remote_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>4, :direction => :remote_only
+    @ls_db.new_link :router_id=> 1, :neighbor_id=>5, :direction => :remote_only
+    assert_equal 4,  @ls_db.size
+  end
+
+  def test_create
+    ls_db = LinkStateDatabase.create :columns=> 2, :rows=> 2, 
+    :base_prefix => '169.0.0.0/24', 
+    :base_router_id=> 0x80000000
+    assert_equal 4, ls_db.size
+    assert  ls_db[1,'128.1.0.1'].has_link?(3,'169.0.0.0')
+    assert  ls_db[1,'128.1.0.2'].has_link?(3,'169.0.2.0')
+  end
+  
+  def test_create_empty_database
+    ls_db = LinkStateDatabase.create :columns=> 0, :rows=> 0
+    assert_equal 0, ls_db.size
+    
+    
+    
+    
+    
+  end
+
+
+
+
+
+end