ospfv2 0.0.1

data/lib/infra/parse_options.rb

@@ -0,0 +1,135 @@
+#--
+# Copyright 2010 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+# OSPFv2 is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# OSPFv2 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with OSPFv2.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'optparse'
+require 'ostruct'
+
+require 'ie/id'
+
+class OptParse
+  def self.parse(args)
+
+    options = OpenStruct.new
+    options.ipaddr = '192.168.1.123'
+    options.network = '192.168.1.0'
+    options.netmask = '255.255.255.0'
+    options.base_link_addr = '1.3.0.0/30'
+    options.base_router_id = 0
+    options.hello_int = 10
+    options.router_id = 1
+    options.neighbor_id = 2
+    options.area_id = 0
+    options.log_fname = $stdout
+    options.grid = [2,2]
+    options.num_sum = 10
+    options.num_ext = 10
+
+    options.parse  = Proc.new do |filename|
+      y_conf = YAML::load_file(filename)
+      y_conf.keys.each do |k|
+        case k.downcase
+        when "key1"
+        when "key2"
+        end
+      end
+      options
+    end
+
+    to_ip = lambda { |id| [id].pack('N').unpack('C*').join('.') }
+    to_id = lambda { |x| OSPFv2::Id.to_i(x) }
+    dead_int = lambda { @dead_int || (options.hello_int * 4)}
+
+    option_help = "blabla ...."
+    hlp_address =    "IP Address of the OSPF Interface."
+    hlp_base_link_addr = "base p2p links addres [#{options.base_link_addr}]"
+    hlp_base_router_id   = "base router-id [#{to_ip.call(options.base_router_id)}]"
+    hlp_neighbor_id= "Neighbor Id. [#{to_ip.call(options.neighbor_id)}]"
+    hlp_router_id =  "Router Id.   [#{to_ip.call(options.router_id)}]"
+    hlp_area_id =    "Area Id.     [#{to_ip.call(options.area_id)}]"
+    hlp_grid =       "Area Grid.   [#{options.grid.join('x')}]"
+    hlp_hello_int =  "Hello Int.   [#{options.hello_int}]"
+    hlp_dead_int =   "Dead Int.    [#{dead_int.call}]"
+    hlp_sum =        "\#Summary.   [#{options.num_sum}]"
+    hlp_ext =        "\#External.  [#{options.num_ext}]"
+
+    optparse = OptionParser.new do |opts|
+
+      opts.banner = "Usage: #{$0} [options]"
+
+      opts.on( "--base-p2p-addr [PREFIX]", hlp_base_link_addr) { |x| 
+        options.base_link_addr = x
+      }
+      opts.on("-i", "--address [PREFIX]", hlp_address) { |x| 
+        options.ipaddr = x
+        options.ipaddr = x.split('/')[0]
+        _addr = IPAddr.new x
+        options.network = _addr.to_s
+        options.netmask = _addr.netmask
+      }
+      opts.on("--base-router-id [ID]", hlp_base_router_id) { |id| 
+        options.base_router_id = id.to_i
+      }
+      opts.on("-r", "--router-id [ID]", hlp_router_id) { |id| 
+        options.router_id = OSPFv2::Id.to_i(id) 
+      }
+      opts.on("-n", "--neighbor-id [ID]", hlp_neighbor_id) { |id| 
+        options.neighbor_id = OSPFv2::Id.to_i(id) 
+      }
+      opts.on("-a", "--area-id [ID]", hlp_area_id) { |id| 
+        options.area_id = to_id.call(id) 
+      }
+      opts.on("-g", "--grid [colxrow]", hlp_grid) { |grid| 
+        options.grid = grid.split('x').collect { |x| x.to_i }
+      }
+      opts.on( "--hello-interval [INT]", hlp_hello_int) { |int| 
+        options.hello_int = int.to_i
+      }
+      opts.on("--dead-interval [INT]", hlp_dead_int) { |int| 
+        @dead_int = int.to_i
+      }
+      opts.on("-g", "--grid [colxrow]", hlp_grid) { |grid| 
+        options.grid = grid.split('x').collect { |x| x.to_i }
+      }
+      opts.on('-S', "--number-of-summary [INT]", "Number of Summary LSA.") { |x| 
+        options.num_sum = x.to_i
+      }
+      opts.on('-E',"--number-of-external [INT]", "Number AsExternal") { |x| 
+        options.num_ext = x.to_i
+      }
+      opts.on( '-f', "--log-fname [FILENAME]", "To redirect logs to a file.") { |fname|
+        options.log_fname = fname 
+      }
+      opts.on_tail("-h", "--help", "Show this message") { puts "\n  #{opts}\n" ; exit }
+      opts.on_tail("-?", "--help") { puts "\n  #{opts}\n" ; exit }
+      opts.on
+    end
+
+    optparse.parse!(args)
+    options.dead_int = @dead_int || (options.hello_int * 4)
+
+    options
+
+  end
+
+end
+
+load "../../../test/ospfv2/infra/#{ File.basename($0.gsub(/.rb/,'_test.rb'))}" if __FILE__ == $0

data/lib/infra/timer.rb

@@ -0,0 +1,104 @@
+#--
+# Copyright 2010 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+# OSPFv2 is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# OSPFv2 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with OSPFv2.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'observer'
+require 'thread'
+
+class Timer
+  include Observable
+
+  class << self
+    def start interval, observer=nil, &block
+      timer = if block
+        new interval, observer, &block
+      else
+        new interval, observer
+      end
+      timer.start
+    end
+  end
+
+  def initialize interval, observer=nil, &block
+    @interval = interval
+    @code = block if block
+    add_observer observer if observer
+  end
+
+  def start _interval=@interval, &block
+    stop if running?
+    changed and notify_observers(:ev_timer_start, id, Time.now.strftime("%M:%S"))
+    _code = block || @code
+    @_timer_thread_ = Thread.new(_interval, _code) do |interval, code|
+      sleep(interval)
+      code.call if code
+      changed and notify_observers(:ev_timer_fire, id, Time.now.strftime("%M:%S"))
+    end
+    self
+  end
+
+  def cancel
+    return unless @_timer_thread_
+    [:exit, :join].each { |x| @_timer_thread_.send x }
+    changed and notify_observers(:ev_timer_cancel, id, Time.now.strftime("%M:%S"))
+    yield if block_given?
+  end
+  alias :stop :cancel
+
+  def running?
+    @_timer_thread_ and @_timer_thread_.alive?
+  end
+
+  def reset &block
+    cancel
+    start &block
+  end
+
+  private
+
+  def id
+    self.class.to_s.to_sym
+  end
+
+end
+
+class PeriodicTimer < Timer
+  def start _interval=@interval, &block
+    @continue = true
+    changed and notify_observers(:ev_timer_start, id, Time.now.strftime("%M:%S"))
+    _code = block || @code
+    @_timer_thread_ = Thread.new(_interval, _code) do |interval, code|
+      loop do
+        sleep(interval)
+        code.call if code
+        changed and notify_observers(:ev_timer_fire, id, Time.now.strftime("%M:%S"))
+        break unless @continue
+      end
+      changed and notify_observers(:ev_timer_stop, id, Time.now.strftime("%M:%S"))
+    end
+    self
+  end
+  def last_shot
+    @continue = false
+  end
+end
+
+load "../../../test/ospfv2/infra/#{ File.basename($0.gsub(/.rb/,'_test.rb'))}" if __FILE__ == $0

data/lib/infra/to_s.rb

@@ -0,0 +1,38 @@
+#--
+# Copyright 2010 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+# OSPFv2 is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# OSPFv2 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with OSPFv2.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module OSPFv2
+  module TO_S
+    
+    def to_s(*args)
+      return to_s_default(*args) unless defined?($style)
+      case $style
+      when :junos ; to_s_junos(*args)
+      when :junos_verbose ; to_s_junos_verbose(*args)
+      else
+        to_s_default(*args)
+      end
+    end
+    
+  end
+
+end

data/lib/ls_db/advertised_routers.rb

@@ -0,0 +1,78 @@
+#--
+# Copyright 2010 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+# OSPFv2 is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# OSPFv2 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with OSPFv2.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'set'
+require 'ie/id'
+
+module OSPFv2::LSDB
+  class AdvertisedRouters
+    AdvertisedRouter = Class.new(OSPFv2::Id)
+    attr_reader :routers
+    def initialize
+      @set = Set.new
+    end
+    def +(id)
+      @set << router_id(id)
+    end
+    def routers
+      @set.collect.sort
+    end
+    alias :ids :routers
+    def has?(id)
+      routers.include?(router_id(id))
+    end
+    def -(id)
+      @set.delete router_id(id)
+    end
+    private
+    def router_id(id)
+      AdvertisedRouter.new(id).to_i
+    end
+  end
+end
+
+if __FILE__ == $0
+
+  require "test/unit"
+
+  # require "ls_db/advertised_routers"
+
+  class TestLsDbAdvertisedRouters < Test::Unit::TestCase
+    include OSPFv2::LSDB
+    def tests
+      assert AdvertisedRouters.new
+      routers = AdvertisedRouters.new
+      routers + 1
+      routers + '0.0.0.1'
+      routers + 2
+      routers + OSPFv2::Id.new(3)
+      assert_equal [1,2,3], routers.routers
+      routers -1 
+      assert_equal [2,3], routers.routers
+      routers -3
+      assert_equal [2], routers.routers
+      routers - '0.0.0.2'
+      assert_equal [], routers.routers
+    end
+  end
+
+end

data/lib/ls_db/common.rb

@@ -0,0 +1,31 @@
+#--
+# Copyright 2010 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of OSPFv2.
+# 
+# OSPFv2 is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# OSPFv2 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with OSPFv2.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+module OSPFv2
+  module LSDB
+    ROUTER_ID_BASE=0x01000000
+    EXTERNAL_BASE_ADDRESS='50.0.0.0/24'
+    SUMMARY_BASE_ADDRESS='30.0.0.0/24'
+    NETWORK_BASE_ADDRESS='20.0.0.0/24'
+    LINK_BASE_ADDRESS='13.0.0.0/30'
+  end
+end