trema 0.1.3

data/ruby/trema/controller.h

@@ -0,0 +1,44 @@
+/*
+ * Base class for OpenFlow controllers.
+ *
+ * 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.
+ */
+
+
+#ifndef CONTROLLER_H
+#define CONTROLLER_H
+
+
+#include "ruby.h"
+
+
+extern VALUE cController;
+
+
+void Init_controller( void );
+
+
+#endif // CONTROLLER_H
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/ruby/trema/controller.rb

@@ -0,0 +1,81 @@
+#
+# 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/app"
+require "trema/monkey-patch/integer"
+require "trema/monkey-patch/string"
+require "trema/timers"
+
+
+module Trema
+  #
+  # The base class of Trema controller.
+  #
+  class Controller < App
+    include Timers
+
+
+    #
+    # Callback invoked whenever a subclass of this class is created.
+    # This adds the created object to the DB of controllers.
+    #
+    def self.inherited subclass
+      subclass.new
+    end
+
+
+    def initialize
+      App.add self
+    end
+
+
+    #
+    # Run as a daemon.
+    #
+    def daemonize!
+      fork do
+        ::Process.setsid
+        fork do
+          STDIN.close
+          STDOUT.reopen "/dev/null", "a"
+          STDERR.reopen "/dev/null", "a"
+          self.run!
+        end
+      end
+    end
+
+
+    #
+    # Name of the controller.
+    #
+    # @return [String]
+    #
+    def name
+      self.class.to_s.split( "::" ).last
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/daemon.rb

@@ -0,0 +1,167 @@
+#
+# 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 "fileutils"
+require "trema/monkey-patch/string"
+require "trema/process"
+
+
+module Trema
+  module Daemon
+    module ClassMethods
+      def singleton_daemon
+        class_variable_set :@@singleton_daemon, true
+      end
+
+
+      def log_file &block
+        class_variable_set :@@log_file, block
+      end
+
+
+      def command &block
+        class_variable_set :@@command, block
+      end
+
+
+      def wait_until_up
+        class_variable_set :@@wait_until_up, true
+      end
+
+
+      def daemon_id method_id
+        class_variable_set :@@daemon_id, method_id
+      end
+    end
+
+
+    def self.included base
+      base.class_eval do
+        class_variable_set :@@singleton_daemon, false
+        class_variable_set :@@log_file, nil
+        class_variable_set :@@command, nil
+        class_variable_set :@@wait_until_up, false
+        class_variable_set :@@daemon_id, nil
+      end
+      base.extend ClassMethods
+    end
+
+
+    def run!
+      raise "'#{ name }' is already running!" if running?
+      FileUtils.rm_f log_file if log_file
+      command_block = self.class.class_eval do
+        class_variable_get( :@@command )
+      end
+      if command_block
+        sh command_block.call( self )
+      else
+        sh self.__send__( :command )
+      end
+      wait_until_up
+    end
+
+
+    #
+    # Kills running daemon process
+    #
+    # @example
+    #   daemon.shutdown!
+    #
+    # @return [undefined]
+    #
+    def shutdown!
+      Trema::Process.read( pid_file, name ).kill!
+    end
+
+
+    #
+    # Restarts running daemon process
+    #
+    # @example
+    #   daemon.restart!
+    #
+    # @return [undefined]
+    #
+    def restart!
+      return if not running?
+      shutdown!
+      sleep 1
+      run!
+    end
+
+
+    def pid_file
+      prefix = self.class.name.demodulize.underscore
+      if self.class.class_eval { class_variable_get :@@singleton_daemon }
+        File.join Trema.pid_directory, "#{ prefix }.pid"
+      else
+        File.join Trema.pid_directory, "#{ prefix }.#{ daemon_id }.pid"
+      end
+    end
+
+
+    def running?
+      FileTest.exists? pid_file
+    end
+
+
+    ############################################################################
+    private
+    ############################################################################
+
+
+    def log_file
+      log_file_block = self.class.class_eval do
+        class_variable_get( :@@log_file )
+      end
+      return nil if log_file_block.nil?
+      name = log_file_block.call( self )
+      File.join Trema.log_directory, name
+    end
+
+
+    def daemon_id
+      m = self.class.class_eval do
+        class_variable_get( :@@daemon_id ) || :name
+      end
+      self.__send__ m
+    end
+
+
+    def wait_until_up
+      wait = self.class.class_eval do
+        class_variable_get( :@@wait_until_up )
+      end
+      return if not wait
+      loop do
+        sleep 0.1
+        break if FileTest.exists?( pid_file )
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl.rb

@@ -0,0 +1,34 @@
+#
+# Trema network DSL.
+#
+# 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/dsl/configuration"
+require "trema/dsl/parser"
+require "trema/dsl/runner"
+require "trema/dsl/syntax"
+require "trema/path"
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/configuration.rb

@@ -0,0 +1,153 @@
+#
+# 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/app"
+require "trema/host"
+require "trema/link"
+require "trema/packetin-filter"
+require "trema/switch"
+require "trema/switch-manager"
+
+
+module Trema
+  module DSL
+    #
+    # The current configuration of Trema.
+    #
+    class Configuration
+      #
+      # set/get the port number for switch manager to listen to
+      #
+      # @example
+      #   config.port = 5432
+      #
+      # @return [Number]
+      #
+      attr_accessor :port
+
+      #
+      # use tremashark?
+      #
+      # @example
+      #   config.tremashark = true
+      #
+      # @return [Boolean]
+      #
+      attr_accessor :tremashark
+
+      #
+      # the hash of {App}
+      #
+      # @example
+      #   p config.apps
+      #   #=> {"trema tetris"=>#<Trema::App:0xb73c9328>, ...}
+      #
+      # @return [Hash]
+      #
+      attr_reader :apps
+
+      #
+      # the hash of {Host}
+      #
+      # @example
+      #   p config.hosts
+      #   #=> {"host #0"=>#<Trema::Host:0xb73c9328>, ...}
+      #
+      # @return [Hash]
+      #
+      attr_reader :hosts
+
+      #
+      # the hash of {Link}
+      #
+      # @example
+      #   p config.links
+      #   #=> {"link #0"=>#<Trema::Link:0xb73c9328>, ...}
+      #
+      # @return [Hash]
+      #
+      attr_reader :links
+
+      #
+      # the hash of {Switch}
+      #
+      # @example
+      #   p config.switches
+      #   #=> {"switch #0"=>#<Trema::Switch:0xb73c9328>, ...}
+      #
+      # @return [Hash]
+      #
+      attr_reader :switches
+
+
+      #
+      # Creates a new Trema configuration
+      #
+      # @example
+      #   config = Trema::DSL::Configuration.new
+      #
+      # @return [Configuration]
+      #
+      def initialize
+        @port = 6633
+        @tremashark = false
+        @apps = Trema::App.clear
+        @hosts = Trema::Host.clear
+        @links = Trema::Link.clear
+        @packetin_filter = Trema::PacketinFilter.clear
+        @switch_manager = Trema::SwitchManager.clear
+        @switches = Trema::Switch.clear
+      end
+
+
+      #
+      # Returns {PacketinFilter} configuration
+      #
+      # @example
+      #   config.packetin_filter => #<Trema::PacketinFilter:0xb73c9328>
+      #
+      # @return [PacketinFilter]
+      #
+      def packetin_filter
+        @packetin_filter.values.last
+      end
+
+
+      #
+      # Returns {SwitchManager} configuration
+      #
+      # @example
+      #   config.switch_manager => #<Trema::SwitchManager:0xb73c9328>
+      #
+      # @return [SwitchManager]
+      #
+      def switch_manager
+        @switch_manager.values.last
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: