trema 0.1.3

data/ruby/trema/barrier-request.h

@@ -0,0 +1,44 @@
+/*
+ * Ruby wrapper class for OFPT_BARRIER_REQUEST message.
+ *
+ * Author: Nick Karanatsios <nickkaranatsios@gmail.com>
+ *
+ * Copyright (C) 2008-2012 NEC Corporation
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License, version 2, as
+ * published by the Free Software Foundation.
+ *
+ * This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+
+#ifndef BARRIER_REQUEST_H
+#define BARRIER_REQUEST_H
+
+
+#include "ruby.h"
+
+
+extern VALUE cBarrierRequest;
+
+
+void Init_barrier_request( void );
+
+
+#endif // BARRIER_REQUEST_H
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/ruby/trema/cli.rb

@@ -0,0 +1,269 @@
+#
+# The controller class of phost cli.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require "trema/executables"
+
+
+module Trema
+  class Stats
+    class Packet
+    end
+
+    
+    attr_reader :ip_dst
+    attr_reader :tp_dst
+    attr_reader :ip_src
+    attr_reader :tp_src
+    attr_reader :n_pkts
+    attr_reader :n_octets
+
+    
+    def initialize ip_dst, tp_dst, ip_src, tp_src, n_pkts, n_octets
+      @ip_dst = ip_dst
+      @tp_dst = tp_dst.to_i
+      @ip_src = ip_src
+      @tp_src = tp_src.to_i
+      @n_pkts = n_pkts.to_i
+      @n_octets = n_octets.to_i
+    end
+
+
+    def packets
+      list = []
+      @n_pkts.times do
+        list << Packet.new
+      end
+      list
+    end
+  end
+end
+
+
+module Trema
+  class Cli
+    def initialize host
+      @host = host
+    end
+
+
+    def send_packets dest, options = {}
+      if options[ :duration ] and options[ :n_pkts ]
+        raise "--duration and --n_pkts are exclusive."
+      end
+
+      sh( "#{ Executables.cli } -i #{ @host.interface } send_packets " +
+          "--ip_src #{ @host.ip } --ip_dst #{ dest.ip } " + 
+          send_packets_options( options ) )
+    end
+
+
+    def show_tx_stats
+      puts stats( :tx )
+    end
+
+
+    def show_rx_stats
+      puts stats( :rx )
+    end
+    
+
+    def tx_stats
+      stat = stats( :tx ).split( "\n" )[ 1 ]
+      if stat
+        Trema::Stats.new *stat.split( "," )        
+      else
+        nil
+      end
+    end
+
+
+    def rx_stats
+      stat = stats( :rx ).split( "\n" )[ 1 ]
+      if stat
+        Trema::Stats.new *stat.split( "," )        
+      else
+        Trema::Stats.new nil, nil, nil, nil, 0, 0
+      end
+    end
+
+
+    def reset_stats
+      sh "sudo #{ Executables.cli } -i #{ @host.interface } reset_stats --tx"
+      sh "sudo #{ Executables.cli } -i #{ @host.interface } reset_stats --rx"
+    end
+
+
+    def add_arp_entry other
+      sh "sudo #{ Executables.cli } -i #{ @host.interface } add_arp_entry --ip_addr #{ other.ip } --mac_addr #{ other.mac }"
+    end
+    
+
+    def set_ip_and_mac_address
+      sh "sudo #{ Executables.cli } -i #{ @host.interface } set_host_addr --ip_addr #{ @host.ip } --ip_mask #{ @host.netmask } --mac_addr #{ @host.mac }"
+    end
+
+
+    def enable_promisc
+      sh "sudo #{ Executables.cli } -i #{ @host.interface } enable_promisc"
+    end
+    
+
+    ################################################################################
+    private
+    ################################################################################
+
+
+    def send_packets_options options
+      [
+       tp_src( options[ :tp_src ] || default_tp_src ),
+       tp_dst( options[ :tp_dst ] || default_tp_dst ),
+       pps( options[ :pps ] || default_pps ),
+       options[ :n_pkts ] ? nil : duration( options[ :duration ] || default_duration ),
+       length( options[ :length ] || default_length ),
+       n_pkts( options[ :n_pkts ] ),
+       inc_ip_src( options[ :inc_ip_src ] ),
+       inc_ip_dst( options[ :inc_ip_dst ] ),
+       inc_tp_src( options[ :inc_tp_src ] ),
+       inc_tp_dst( options[ :inc_tp_dst ] ),
+       inc_payload( options[ :inc_payload ] ),
+      ].compact.join( " " )
+    end
+
+
+    def tp_src value
+      "--tp_src #{ value }"
+    end
+
+
+    def default_tp_src
+      1
+    end
+
+
+    def tp_dst value
+      "--tp_dst #{ value }"
+    end
+
+
+    def default_tp_dst
+      1
+    end
+
+
+    def pps value
+      "--pps #{ value }"
+    end
+
+
+    def default_pps
+      1
+    end
+
+
+    def duration value
+      "--duration #{ value }"
+    end
+
+
+    def default_duration
+      1
+    end
+
+
+    def length value
+      "--length #{ value }"
+    end
+
+
+    def default_length
+      22
+    end
+
+
+    def inc_ip_src value
+      return nil if value.nil?
+      if value == true
+        "--inc_ip_src"
+      else
+        "--inc_ip_src=#{ value }"
+      end
+    end
+
+
+    def inc_ip_dst value
+      return nil if value.nil?
+      if value == true
+        "--inc_ip_dst"
+      else
+        "--inc_ip_dst=#{ value }"
+      end
+    end
+
+
+    def inc_tp_src value
+      return nil if value.nil?
+      if value == true
+        "--inc_tp_src"
+      else
+        "--inc_tp_src=#{ value }"
+      end
+    end
+
+
+    def inc_tp_dst value
+      return nil if value.nil?
+      if value == true
+        "--inc_tp_dst"
+      else
+        "--inc_tp_dst=#{ value }"
+      end
+    end
+
+
+    def inc_payload value
+      return nil if value.nil?
+      if value == true
+        "--inc_payload"
+      else
+        "--inc_payload=#{ value }"
+      end
+    end
+
+
+    def n_pkts value
+      return nil if value.nil?
+      "--n_pkts=#{ value }"
+    end
+
+
+    def stats type
+      `sudo #{ Executables.cli } -i #{ @host.interface } show_stats --#{ type }`
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command.rb

@@ -0,0 +1,40 @@
+#
+# Trema sub-commands.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require "trema/command/dump_flows"
+require "trema/command/kill"
+require "trema/command/killall"
+require "trema/command/reset_stats"
+require "trema/command/ruby"
+require "trema/command/run"
+require "trema/command/send_packets"
+require "trema/command/shell"
+require "trema/command/show_stats"
+require "trema/command/usage"
+require "trema/command/version"
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/dump_flows.rb

@@ -0,0 +1,62 @@
+#
+# trema dump_flows command.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require "optparse"
+require "trema/dsl"
+require "trema/ofctl"
+require "trema/util"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def dump_flows
+      sanity_check
+
+      switch = Trema::DSL::Context.load_current.switches[ ARGV[ 0 ] ]
+
+      options = OptionParser.new
+      options.banner = "Usage: #{ $PROGRAM_NAME } dump_flows SWITCH [OPTIONS ...]"
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      puts Trema::Ofctl.new.dump_flows( switch )
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/kill.rb

@@ -0,0 +1,71 @@
+#
+# trema kill command.
+#
+# Author: Yasuhito Takamiya <yasuhito@gmail.com>
+#
+# Copyright (C) 2008-2012 NEC Corporation
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2, as
+# published by the Free Software Foundation.
+#
+# This program 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 this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+
+
+require "optparse"
+require "trema/dsl"
+require "trema/util"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def kill
+      options = OptionParser.new
+      options.banner = "Usage: #{ $PROGRAM_NAME } kill NAME [OPTIONS ...]"
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      context = Trema::DSL::Context.load_current
+
+      # [FIXME] Trema apps does not appear in context.apps. why?
+      pid_file = File.join( Trema.pid_directory, "#{ ARGV[ 0 ] }.pid" )
+      if FileTest.exist?( pid_file )
+        Trema::Process.read( pid_file ).kill!
+      end
+
+      host = context.hosts[ ARGV[ 0 ] ]
+      host.shutdown! if host
+
+      switch = context.switches[ ARGV[ 0 ] ]
+      switch.shutdown! if switch
+
+      # [TODO] kill a link by its name. Needs a good naming convension for link.
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End: