bgp4r 0.0.3

data/bgp/io.rb

@@ -0,0 +1,116 @@
+#--
+# Copyright 2008, 2009 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of BGP4R.
+# 
+# BGP4R 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.
+# 
+# BGP4R 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 BGP4R.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+
+require 'observer'
+require 'thread'
+require 'bgp/common'
+require 'bgp/message'
+
+module BGP
+  module IO
+
+    class Output < Queue
+      include Observable
+      def initialize(session, holdtime, *obs)
+        @session, @holdtime = session, holdtime
+        obs.each { |o| add_observer(o) }
+        @continue = true
+        super()
+      end
+      attr_reader :thread
+      attr_writer :holdtime
+      def start
+        @thread = Thread.new(@session, @holdtime) do |s, h|
+          Thread.current['name']='BGP IO Ouput'
+          Log.debug "#{self} #{Thread.current} started" 
+          begin
+            while @continue
+              obj = deq
+              break unless @continue
+              s.write obj.respond_to?(:encode) ? obj.encode : obj
+            end
+          rescue IOError, Errno::ECONNRESET, Errno::EBADF => e
+            puts "ECONNRESET"
+            changed and notify_observers(:ev_conn_reset, e)
+          ensure
+            Log.debug "Exiting #{Thread.current['name']}"
+          end
+        end
+      end
+      def exit
+        @continue = false
+        enq('you\'re done dude!')
+      end
+    end
+
+    class Input
+      include Observable
+      def initialize(session, holdtime, *obs)
+        @session, @holdtime = session, holdtime
+        obs.each { |o| add_observer(o) }
+      end
+      attr_reader :thread
+      attr_writer :holdtime
+      def recv_msg(s)
+        loop do
+          if @buf.size<18 or @buf.size <@len
+            recv = s.recv(4100) 
+            @buf += recv unless recv.nil?
+            break unless @continue
+          end
+          if @buf.size>18
+            @len, @type=@buf[16,3].unpack('nC')
+            return @buf.slice!(0,@len) if @len<=@buf.size
+          end
+        end
+      end
+      
+      def start
+        @thread = Thread.new(@session, @holdtime) do |s, h|
+          Thread.current['name']='BGP IO Input'
+          Log.debug "#{self} #{Thread.current} started"
+          @buf = ''
+          @continue = true
+          @len=0
+          begin
+            while @continue
+              begin
+                Timeout::timeout(h) do |h| 
+                  message, type = recv_msg(s)
+                  break unless @continue
+                  changed and notify_observers(:ev_msg, @type, message)
+                end
+              rescue Timeout::Error => e
+                changed and notify_observers(:ev_holdtime_expire, e)
+                @continue = false
+              end
+            end
+          rescue IOError, Errno::ECONNRESET, Errno::EBADF => e
+            changed and notify_observers(:ev_conn_reset, e)
+          end
+          Log.debug "Exiting #{Thread.current['name']}"
+        end 
+      end
+    end
+  end
+end

data/bgp/label.rb

@@ -0,0 +1,94 @@
+#--
+# Copyright 2008, 2009 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of BGP4R.
+# 
+# BGP4R 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.
+# 
+# BGP4R 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 BGP4R.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'bgp/common'
+
+module BGP
+  class Label
+    attr_reader :label
+    def initialize(*args)
+      @label, @exp = [0]*3
+      if args.size==1 and args[0].is_a?(Hash)
+        @label=args[0][:label] if args[0][:label]
+        @exp=args[0][:exp] if args[0][:exp]
+      elsif args.size==1 and args[0].is_a?(String) and args[0].is_packed?
+        parse(args[0])
+      else args[0].is_a?(Fixnum)
+        @label, @exp = args + [0]
+      end
+    end
+    def encode(bottom=1)
+      n = (@label <<4) | (@exp & 0x7)<<1 |(bottom &0x1)
+      o1 = (n & 0xff0000) >> 16
+      o2 = (n & 0x00ff00) >> 8
+      o3 = (n & 0x0000ff)
+      [o1,o2,o3].pack('CCC')
+    end
+    def parse(s)
+      octets = s.slice!(0,3).unpack('CCC')
+      n = (octets[0] << 16) + (octets[1] << 8) + octets[2]
+      @exp = (n >> 1) & 0x7
+      @label = (n >> 4)
+    end
+    def to_hash
+      {:label=>@label, :exp=>@exp}
+    end
+  end
+  class Label_stack
+    def initialize(*args)
+      @label_stack=[]
+      if args.size==1 and args[0].is_a?(String) and args[0].is_packed?
+        parse(args[0])
+      else
+        args.each { |arg| @label_stack << (arg.is_a?(Label) ? arg : Label.new(arg)) }
+      end
+    end
+    def size
+      @label_stack.size
+    end
+    def encode
+      enc = @label_stack[0..-2].collect { |l| l.encode(0) }
+      enc << @label_stack[-1].encode
+      enc.join
+    end
+    def parse(s)
+      while s.size>0
+        bottom = s[2,1].unpack('C')[0]
+        @label_stack << label = Label.new(s)
+        break if bottom & 1 > 0
+      end
+    end
+    def to_s
+      if @label_stack.empty?
+        "Label stack:(empty)"
+      else
+        "Label Stack=#{@label_stack.collect{ |l| l.label }.join(',')} (bottom)"
+      end
+    end
+    def bit_length
+      @label_stack.compact.size*24
+    end
+  end
+end
+
+load "../test/#{ File.basename($0.gsub(/.rb/,'_test.rb'))}" if __FILE__ == $0
+

data/bgp/local_pref.rb

@@ -0,0 +1,75 @@
+#--
+# Copyright 2008, 2009 Jean-Michel Esnault.
+# All rights reserved.
+# See LICENSE.txt for permissions.
+#
+#
+# This file is part of BGP4R.
+# 
+# BGP4R 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.
+# 
+# BGP4R 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 BGP4R.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+
+require 'bgp/attribute'
+
+module BGP
+
+  class Local_pref < Attr
+
+    def initialize(arg=100)
+      @flags, @type = WELL_KNOWN_MANDATORY, LOCAL_PREF
+      if arg.is_a?(String) and arg.is_packed?
+        parse(arg)
+      elsif arg.is_a?(self.class)
+        parse(arg.encode)
+      elsif arg.is_a?(Fixnum) or arg.is_a?(Bignum)
+        self.local_pref = arg
+      elsif arg.is_a?(Hash) and arg[:local_pref]
+        self.local_pref = arg[:local_pref]
+      else
+        raise ArgumentError, "invalid argument, #{arg.inspect}"
+      end
+    end
+
+    def local_pref=(val)
+      raise ArgumentError, "invalid argument" unless val.is_a?(Integer)
+      @local_pref=val
+    end
+
+    def to_i
+      @local_pref
+    end
+
+    def encode
+      super([to_i].pack('N'))
+    end
+
+    def local_pref
+      format("(0x%4.4x) %d", to_i, to_i)
+    end
+
+    def to_s(method=:default)
+      super(local_pref, method)
+    end
+
+    private
+
+    def parse(s)
+      @flags, @type, len, @local_pref = super(s,'N')
+    end
+
+  end
+
+end
+load "../test/#{ File.basename($0.gsub(/.rb/,'_test.rb'))}" if __FILE__ == $0