trema 0.1.3

data/ruby/trema/command/killall.rb

@@ -0,0 +1,56 @@
+#
+# trema killall 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/util"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def killall
+      options = OptionParser.new
+      options.banner = "Usage: #{ $PROGRAM_NAME } killall [OPTIONS ...]"
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      cleanup_current_session
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/reset_stats.rb

@@ -0,0 +1,61 @@
+#
+# trema reset_stats 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/cli"
+require "trema/dsl"
+require "trema/util"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def reset_stats
+      sanity_check
+
+      options = OptionParser.new
+      options.banner = "Usage: #{ $PROGRAM_NAME } reset_stats [OPTIONS ...]"
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      host = Trema::DSL::Context.load_current.hosts[ ARGV[ 0 ] ]
+      Trema::Cli.new( host ).reset_stats
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/ruby.rb

@@ -0,0 +1,55 @@
+#
+# trema ruby 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"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def ruby
+      options = OptionParser.new
+      options.banner = "Usage: #{ $PROGRAM_NAME } ruby [OPTIONS ...]"
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      sh "x-www-browser http://rubydoc.info/github/trema/trema/master/frames"
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/run.rb

@@ -0,0 +1,120 @@
+#
+# trema run 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 run
+      sanity_check
+
+      options = OptionParser.new
+      options.banner = "Usage: #{ $0 } run [OPTIONS ...]"
+
+      options.on( "-c", "--conf FILE" ) do | v |
+        @config_file = v
+      end
+      options.on( "-d", "--daemonize" ) do
+        $run_as_daemon = true
+      end
+
+      options.separator ""
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      cleanup_current_session
+
+      if $run_as_daemon
+        Trema::DSL::Runner.new( load_config ).daemonize
+      else
+        begin
+          Trema::DSL::Runner.new( load_config ).run
+        ensure
+          cleanup_current_session
+        end
+      end
+    end
+
+
+    ############################################################################
+    private
+    ############################################################################
+
+
+    def load_config
+      config = nil
+      dsl_parser = Trema::DSL::Parser.new
+
+      if @config_file
+        config = dsl_parser.parse( @config_file )
+      elsif FileTest.exists?( "./trema.conf" )
+        config = dsl_parser.parse( "./trema.conf" )
+      else
+        config = Trema::DSL::Configuration.new
+      end
+
+      if ARGV[ 0 ]
+        controller_file = ARGV[ 0 ].split.first
+        if c_controller?
+          stanza = Trema::DSL::Run.new
+          stanza.path controller_file
+          stanza.options ARGV[ 0 ].split[ 1..-1 ]
+          Trema::App.new( stanza )
+        else
+          # Ruby controller
+          require "trema"
+          ARGV.replace ARGV[ 0 ].split
+          $LOAD_PATH << File.dirname( controller_file )
+          Trema.module_eval IO.read( controller_file )
+        end
+      end
+
+      config
+    end
+
+
+    def c_controller?
+      /ELF/=~ `file #{ ARGV[ 0 ].split.first }`
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/send_packets.rb

@@ -0,0 +1,130 @@
+#
+# trema send_packets 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/cli"
+require "trema/dsl"
+require "trema/util"
+
+
+module Trema
+  module Command
+    include Trema::Util
+
+
+    def send_packets
+      sanity_check
+
+      source = nil
+      dest = nil
+      cli_options = {}
+
+      dsl_parser = Trema::DSL::Parser.new
+
+      options = OptionParser.new
+      options.banner = "Usage: #{ $0 } send_packets --source HOSTNAME --dest HOSTNAME [OPTIONS ...]"
+
+      options.on( "-s", "--source HOSTNAME" ) do | v |
+        source = Trema::DSL::Context.load_current.hosts[ v ]
+        raise "Unknown host: #{ v }" if source.nil?
+      end
+      options.on( "--inc_ip_src [NUMBER]" ) do | v |
+        if v
+          cli_options[ :inc_ip_src ] = v
+        else
+          cli_options[ :inc_ip_src ] = true
+        end
+      end
+      options.on( "-d", "--dest HOSTNAME" ) do | v |
+        dest = Trema::DSL::Context.load_current.hosts[ v ]
+        raise "Unknown host: #{ v }" if dest.nil?
+      end
+      options.on( "--inc_ip_dst [NUMBER]" ) do | v |
+        if v
+          cli_options[ :inc_ip_dst ] = v
+        else
+          cli_options[ :inc_ip_dst ] = true
+        end
+      end
+      options.on( "--tp_src NUMBER" ) do | v |
+        cli_options[ :tp_src ] = v
+      end
+      options.on( "--inc_tp_src [NUMBER]" ) do | v |
+        if v
+          cli_options[ :inc_tp_src ] = v
+        else
+          cli_options[ :inc_tp_src ] = true
+        end
+      end
+      options.on( "--tp_dst NUMBER" ) do | v |
+        cli_options[ :tp_dst ] = v
+      end
+      options.on( "--inc_tp_dst [NUMBER]" ) do | v |
+        if v
+          cli_options[ :inc_tp_dst ] = v
+        else
+          cli_options[ :inc_tp_dst ] = true
+        end
+      end
+      options.on( "--pps NUMBER" ) do | v |
+        cli_options[ :pps ] = v
+      end
+      options.on( "--n_pkts NUMBER" ) do | v |
+        cli_options[ :n_pkts ] = v
+      end
+      options.on( "--duration NUMBER" ) do | v |
+        cli_options[ :duration ] = v
+      end
+      options.on( "--length NUMBER" ) do | v |
+        cli_options[ :length ] = v
+      end
+      options.on( "--inc_payload [NUMBER]" ) do | v |
+        if v
+          cli_options[ :inc_payload ] = v
+        else
+          cli_options[ :inc_payload ] = true
+        end
+      end
+
+      options.separator ""
+
+      options.on( "-h", "--help" ) do
+        puts options.to_s
+        exit 0
+      end
+      options.on( "-v", "--verbose" ) do
+        $verbose = true
+      end
+
+      options.parse! ARGV
+
+      Trema::Cli.new( source ).send_packets( dest, cli_options )
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/command/shell.rb

@@ -0,0 +1,61 @@
+#
+# trema shell 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 "irb"
+require "trema/util"
+
+
+include Trema::Util
+
+
+module Trema
+  module Command
+    def shell
+      begin
+        undef :kill
+
+        require "tempfile"
+        require "trema"
+        require "trema/shell"
+        f = Tempfile.open( "irbrc" )
+        f.print <<EOF
+include Trema::Shell
+ENV[ "TREMA_HOME" ] = Trema.home
+$config = Trema::DSL::Configuration.new
+$context = Trema::DSL::Context.new( $config )
+EOF
+        f.close
+        load f.path
+        IRB.start
+      ensure
+        cleanup $config
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End: