netns-vrf 0.0.3

data/bin/netns-vrf

@@ -0,0 +1,161 @@
+#!/usr/bin/env ruby
+
+require 'netns-vrf'
+
+def help
+   puts "usage: netns-vrf [options]\n\n"
+   puts "options: "
+   puts "  - create [VRF_NAME]"
+   puts "  - delete [VRF_NAME]"
+   puts "  - show [VRF_NAME:optional]"
+   puts "  - interface (push|remove) [IFACE_NAME] [VRF_NAME]"
+   puts "  - ip address (inet|inet6) (add|delete) [IP_ADDRESS] [IFACE_NAME] [VRF_NAME]" 
+   puts "  - ip route (inet|inet6) (add|delete) [DESTINATION_PREFIX] [NEXT_HOP] [IFACE_NAME] [VRF_NAME]\n\n"
+   exit 0
+end
+ 
+begin
+  if ARGV[0]
+     case ARGV[0]
+         when 'create'
+              if ARGV[1]
+	         vrf_name = ARGV[1]
+                 vrf_instance = VRFLite::NetNs.new(vrf_name)
+                 vrf_instance.build
+	         puts "+ vrf created: #{vrf_name}"
+              else
+                 help
+              end
+         
+         when 'delete'
+              if ARGV[1]
+                 vrf_name = ARGV[1]
+                 vrf_instance = VRFLite::NetNs.new(vrf_name)
+                 vrf_instance.destroy
+		 puts "+ vrf deleted: #{vrf_name}"
+              else
+                 help
+              end
+   
+         when 'show'
+              vrf_name = ARGV[1] if ARGV[1]
+              vrf_name = nil if not ARGV[1]
+              vrf_table = VRFLite::NetNs.new(vrf_name)
+              vrf_table_array = vrf_table.display
+              vrf_table_array.each do |vta|
+                puts "display vrf: #{vta}"
+                puts "--------------------------"
+                vrf_local_link = `ip netns exec #{vta} ip address show`
+                puts "#{vrf_local_link}"
+                puts "--------------------------\n\n"
+	      end
+      
+          when 'interface'
+              if ARGV[1] and ARGV[2] and ARGV[3]
+		  action_switch, if_name, vrf_name = ARGV[1..3]
+		  interface = VRFLite::Interface.new(if_name, vrf_name)
+		  if action_switch == "push"
+	             if_action = interface.add_to_vrf
+                     action_type = 0
+	          elsif action_switch == "remove"
+		     if_action = interface.del_from_vrf
+		     action_type = 1
+		  else
+                     help
+                  end
+                  if if_action.to_i == 0 and action_type == 0
+                     puts "+ interface #{if_name} correctly added to vrf: #{vrf_name}"
+                  elsif if_action.to_i == 0 and action_type == 1
+                     puts "+ interface #{if_name} correctly removed from vrf: #{vrf_name}"
+		  else 
+                     puts "+ interface #{if_name} not found in the system: #{if_action}:#{action_type}"
+                  end
+	      else
+                 help
+              end
+
+          when 'ip'
+              help if not ARGV[1]
+              case ARGV[1]
+                   when 'address'
+                       if ARGV[2] == "inet" || ARGV[2] == "inet6" && ARGV[3..6]
+                          action_switch, ip_addr, if_name, vrf_name  = ARGV[3..6]
+		          ip = VRFLite::IP.new(ip_addr, if_name, vrf_name)
+                          if action_switch == "add" && ARGV[2] == "inet"
+			     ip_action = ip.add_to_vrf
+                             action_type = 0
+                          elsif action_switch == "add" && ARGV[2] == "inet6"
+                             ip_action =  ip.add_ip6_to_vrf
+                             action_type = 0
+                          elsif action_switch == "delete" && ARGV[2] == "inet"
+                             ip_action = ip.del_from_vrf
+                             action_type = 1
+                          elsif action_switch = "delete" && ARGV[2] == "inet6"
+                             ip_action = ip.del_ip6_to_vrf
+                             action_type = 1
+			  else
+                             help
+                          end
+			  if ip_action.to_i == 0 and action_type == 0
+                             puts "+ IP address #{ip_addr} correctly added to interface #{if_name} : VRF #{vrf_name}"
+                          elsif ip_action.to_i == 0 and action_type == 1
+                             puts "+ IP address #{ip_addr} correctly deleted from interface #{if_name} : VRF #{vrf_name}"
+                          else
+                             puts "+ unable to add/delete IP address #{ip_addr} to/from interface #{if_name} : VRF #{vrf_name}"
+                          end
+                       else
+                          help
+                       end
+                    when 'route'
+                       if ARGV[2] == "inet" || ARGV[2] == "inet6" && ARGV[3..7]
+                          action_switch, ip_dst, ip_next_hop, if_name, vrf_name  = ARGV[3..7]
+                          static_route = VRFLite::Route.new(ip_dst, ip_next_hop, if_name, vrf_name)
+                          if action_switch == "add" && ARGV[2] == "inet"
+                             routing_action = static_route.add_to_vrf_inet
+			     action_type = 0
+                          elsif action_switch == "add" && ARGV[2] == "inet6"
+                             routing_action = static_route.add_to_vrf_inet6
+			     action_type = 0
+                          elsif action_switch == "delete" && ARGV[2] == "inet"
+                             routing_action = static_route.del_from_vrf_inet
+                             action_type = 1
+                          elsif action_switch == "delete" && ARGV[2] == "inet6"
+                             routing_action = static_route.del_from_vrf_inet6
+                             action_type = 1
+                          elsif action_switch == "show" && ip_dst
+			     vrf_name = ip_dst
+			     route_show = `ip netns exec #{vrf_name} ip route show`.chop if ARGV[2] == "inet"
+                             route_show = `ip netns exec #{vrf_name} ip route show`.chop if ARGV[2] == "inet6"
+		             puts "display vrf #{vrf_name} routing table:\n\n"
+			     puts route_show
+			     puts "\n\n"
+                             routing_action = "1"
+                             action_type = 1
+			  else
+                             help
+                          end
+                          if routing_action.to_i == 0 and action_type == 0
+                             puts "+ route to #{ip_dst} added - nexthop: #{ip_next_hop} - iface: #{if_name} - VRF #{vrf_name}"
+                          elsif routing_action.to_i == 0 and action_type == 1
+                             puts "+ route to #{ip_dst} correctly deleted"
+			  elsif routing_action.to_i == 1 and action_type == 1
+                          else
+                             puts "+ unable to add/delete route."
+                          end
+                       else
+                         help
+                       end
+
+              else
+                 help
+              end
+
+
+   
+     else
+        help
+     end
+  else
+     help
+  end
+end

data/lib/netns-vrf/interface.rb

@@ -0,0 +1,33 @@
+module VRFLite
+  class Interface
+    def initialize(if_name, vrf_name)
+       @if_name = if_name
+       @vrf_name = vrf_name
+    end
+    def add_to_vrf
+        dev_file = File.open("/proc/net/dev", "r")
+        dev_file.each do |dev|
+          if dev =~ /#{@if_name}:/ # check if the interface is in default vrf before pushing it to vrf #
+             system("ip link set #{@if_name} netns #{@vrf_name}")
+             system("ip netns exec #{@vrf_name} ip link set #{@if_name} up")
+             @lock = 0
+	     return @lock.to_i 
+          else
+             @lock = 1 # device is not in default netns #
+          end
+        end
+        return @lock.to_i
+    end
+    def del_from_vrf
+      dev_location = `ip netns exec #{@vrf_name} cat /proc/net/dev | grep ':' | cut -d ':' -f 1 | grep #{@if_name} | head -n 1`.chop # got to find a way to lookup the dev file of the netns #
+      if dev_location != ''
+         system("ip netns exec #{@vrf_name} ip link set #{@if_name} down")
+         system("ip netns exec #{@vrf_name} ip link del #{@if_name}")
+         return 0
+      else
+         return 1 # device is not in vrf #
+      end
+    end
+  end
+end
+

data/lib/netns-vrf/ip.rb

@@ -0,0 +1,25 @@
+module VRFLite
+  class IP
+    def initialize(ip_addr, if_name, vrf_name)
+        @if_name = if_name
+        @vrf_name = vrf_name
+        @ip_addr = ip_addr
+    end
+    def add_to_vrf
+      system("ip netns exec #{@vrf_name} ip address add #{@ip_addr} dev #{@if_name}")
+      return 0
+    end
+    def del_from_vrf
+      system("ip netns exec #{@vrf_name} ip address del #{@ip_addr} dev #{@if_name}")
+      return 0
+    end
+    def add_ip6_to_vrf
+      system("ip netns exec #{@vrf_name} ip -6 address add #{@ip_addr} dev #{@if_name}")
+      return 0
+    end
+    def del_ip6_from_vrf
+      system("ip netns exec #{@vrf_name} ip -6 address del #{@ip_addr} dev #{@if_name}")
+      return 0
+    end
+  end
+end

data/lib/netns-vrf/netns.rb

@@ -0,0 +1,35 @@
+module VRFLite
+  class NetNs # provides: build / destroy / display on NetNS #
+    def initialize(vrf_name)
+       @vrf_name = vrf_name
+    end
+
+    def build
+       system("ip netns add #{@vrf_name}")
+       system("ip netns exec #{@vrf_name} ip link set lo up") # initialize stack loopback #
+       return 0
+    end
+
+    def destroy
+       system("ip netns delete #{@vrf_name}")
+       return 0
+    end
+
+    def display
+       vrf_config = []
+       if @vrf_name == nil || @vrf_name == "all"
+          net_ns = Dir["/var/run/netns/*"]
+       else
+          net_ns = Dir["/var/run/netns/#{@vrf_name}"]
+       end
+       net_ns.each do |ns|
+           vrf_path = net_ns.to_s.split '/'
+	   vrf_id = vrf_path[4].chop.chop
+           vrf_config << vrf_id
+       end
+       return vrf_config if @vrf_name != nil || @vrf_name == "all"
+       return vrf_config[0..-1]
+    end
+  end
+end             
+          

data/lib/netns-vrf/route.rb

@@ -0,0 +1,30 @@
+module VRFLite
+  class Route
+    def initialize(vrf_rt_dst, vrf_gw, if_name, vrf_name)
+      @if_name = if_name
+      @vrf_name = vrf_name
+      @vrf_gw = vrf_gw
+      @vrf_rt_dst = vrf_rt_dst
+    end
+    def add_to_vrf_inet
+      system("ip netns exec #{@vrf_name} ip route add #{@vrf_rt_dst} via #{@vrf_gw} dev #{@if_name}")
+      return 0
+    end
+    def del_from_vrf_inet
+      system("ip netns exec #{@vrf_name} ip route del #{@vrf_rt_dst} via #{@vrf_gw} dev #{@if_name}")
+      return 0
+    end
+
+    def add_to_vrf_inet6
+      system("ip netns exec #{@vrf_name} ip -6 route add #{@vrf_rt_dst} via #{@vrf_gw} dev #{@if_name}")
+      return 0
+    end
+
+    def del_from_vrf_inet6
+      system("ip netns exec #{@vrf_name} ip -6 route del #{@vrf_rt_dst} via #{@vrf_gw} dev #{@if_name}")
+      return 0
+    end
+
+  end
+
+end

data/lib/netns-vrf.rb

@@ -0,0 +1,7 @@
+module VRFLite
+  Directory = File.expand_path File.join File.dirname(__FILE__), '../' 
+  require 'netns-vrf/ip.rb'
+  require 'netns-vrf/route.rb'
+  require 'netns-vrf/netns.rb'
+  require 'netns-vrf/interface.rb'
+end

metadata

@@ -0,0 +1,52 @@
+--- !ruby/object:Gem::Specification
+name: netns-vrf
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Samer Abdel-Hafez
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-05-12 00:00:00.000000000 Z
+dependencies: []
+description: linux vrf-lite command line wannabe
+email:
+- sam@arahant.net
+executables:
+- netns-vrf
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/netns-vrf.rb
+- lib/netns-vrf/ip.rb
+- lib/netns-vrf/route.rb
+- lib/netns-vrf/interface.rb
+- lib/netns-vrf/netns.rb
+- bin/netns-vrf
+homepage: http://github.com/nopedial/netns-vrf
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: netns-vrf
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: quick and dirty linux netns wrapper
+test_files: []