trema 0.1.3

data/ruby/trema/dsl/context.rb

@@ -0,0 +1,71 @@
+#
+# 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/dsl/configuration"
+require "trema/dsl/syntax"
+require "trema/path"
+
+
+module Trema
+  module DSL
+    class Context
+      PATH = File.join( Trema.tmp, ".context" )
+
+
+      def self.load_current
+        if FileTest.exists?( PATH )
+          Marshal.load( IO.read PATH )
+        else
+          Configuration.new
+        end
+      end
+
+
+
+      def initialize config
+        @config = config
+      end
+
+
+      #
+      # Dumps a {Configuration} object to <code>PATH</code>
+      #
+      # @example
+      #   context.dump
+      #
+      # @return [Configuration]
+      #
+      def dump
+        File.open( PATH, "w" ) do | f |
+          f.print Marshal.dump( @config )
+        end
+        @config
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/link.rb

@@ -0,0 +1,41 @@
+#
+# Network link between hosts and switches of 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.
+#
+
+
+module Trema
+  module DSL
+    class Link
+      attr_reader :peers
+
+
+      def initialize peer0, peer1
+        @peers = [ peer0, peer1 ]
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/parser.rb

@@ -0,0 +1,70 @@
+#
+# DSL parser.
+#
+# 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/context"
+require "trema/dsl/syntax"
+
+
+module Trema
+  module DSL
+    class Parser
+      def parse file_name
+        configure do | config |
+          Syntax.new( config ).instance_eval IO.read( file_name ), file_name
+        end
+      end
+
+
+      def eval &block
+        configure do | config |
+          Syntax.new( config ).instance_eval &block
+        end
+      end
+
+
+      ################################################################################
+      private
+      ################################################################################
+
+
+      def configure &block
+        config = Configuration.new
+        block.call config
+        Trema::Link.each do | each |
+          peers = each.peers
+          config.hosts[ peers[ 0 ] ].interface = each.name if config.hosts[ peers[ 0 ] ]
+          config.hosts[ peers[ 1 ] ].interface = each.name_peer if config.hosts[ peers[ 1 ] ]
+          config.switches[ peers[ 0 ] ] << each.name if config.switches[ peers[ 0 ] ]
+          config.switches[ peers[ 1 ] ] << each.name_peer if config.switches[ peers[ 1 ] ]
+        end
+        Context.new( config ).dump
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/run.rb

@@ -0,0 +1,49 @@
+#
+# The syntax definition of run { ... } stanza in Trema 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/stanza"
+
+
+module Trema
+  module DSL
+    class Run < Stanza
+      def path _path
+        @path = _path
+        if @name.nil?
+          @name = File.basename( @path )
+        end
+      end
+
+
+      def options *_options
+        @options = _options
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/runner.rb

@@ -0,0 +1,165 @@
+#
+# Runs DSL objects in right order.
+#
+# 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.
+#
+
+
+module Trema
+  module DSL
+    class Runner
+      def initialize context
+        @context = context
+      end
+
+
+      def run
+        maybe_run_trema_services
+        maybe_run_apps
+      end
+
+
+      def daemonize
+        maybe_run_trema_services
+        maybe_daemonize_apps
+      end
+
+
+      def maybe_run_switch_manager
+        switch_manager =
+          if @context.switch_manager and @context.apps.values.size > 0
+            last_app = @context.apps.values.last.name
+            if not @context.switch_manager.rule.has_key?( :port_status )
+              @context.switch_manager.rule[ :port_status ] = last_app
+            end
+            if not @context.switch_manager.rule.has_key?( :packet_in )
+              @context.switch_manager.rule[ :packet_in ] = last_app
+            end
+            if not @context.switch_manager.rule.has_key?( :state_notify )
+              @context.switch_manager.rule[ :state_notify ] = last_app
+            end
+            if not @context.switch_manager.rule.has_key?( :vendor )
+              @context.switch_manager.rule[ :vendor ] = last_app
+            end
+            @context.switch_manager
+          else
+            if @context.apps.values.size == 0
+              rule = { :port_status => "default", :packet_in => "default", :state_notify => "default", :vendor => "default" }
+            elsif @context.apps.values.size == 1
+              app_name = @context.apps.values[ 0 ].name
+              rule = { :port_status => app_name, :packet_in => app_name, :state_notify => app_name, :vendor => app_name }
+            else
+              # two or more apps without switch_manager.
+              raise "No event routing configured. Use `event' directive to specify event routing."
+            end
+            SwitchManager.new( rule, @context.port )
+          end
+        switch_manager.run!
+      end
+
+
+      ################################################################################
+      private
+      ################################################################################
+
+
+      def maybe_run_trema_services
+        maybe_run_tremashark
+        maybe_run_switch_manager
+        maybe_run_packetin_filter
+        maybe_create_links
+        maybe_run_hosts
+        maybe_run_switches
+      end
+
+
+      def maybe_run_tremashark
+        @context.tremashark.run if @context.tremashark
+      end
+
+
+      def maybe_run_packetin_filter
+        @context.packetin_filter.run! if @context.packetin_filter
+      end
+
+
+      def maybe_create_links
+        maybe_delete_links # Fool proof
+        @context.links.each do | name, link |
+          link.enable!
+        end
+      end
+
+
+      def maybe_delete_links
+        @context.links.each do | name, link |
+          link.delete!
+        end
+      end
+
+
+      def maybe_run_hosts
+        @context.hosts.each do | name, host |
+          host.run!
+        end
+      end
+
+
+      def maybe_run_switches
+        @context.switches.each do | name, switch |
+          switch.run!
+        end
+
+        @context.hosts.each do | name, host |
+          host.add_arp_entry @context.hosts.values - [ host ]
+        end
+      end
+
+
+      def maybe_run_apps
+        return if @context.apps.values.empty?
+
+        @context.apps.values[ 0..-2 ].each do | each |
+          each.daemonize!
+        end
+        trap( "SIGINT" ) do
+          print( "\nterminated\n" )
+          exit(0)
+        end
+        pid = ::Process.fork do
+          @context.apps.values.last.run!
+        end
+        ::Process.waitpid pid
+      end
+
+
+      def maybe_daemonize_apps
+        @context.apps.each do | name, app |
+          app.daemonize!
+        end
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/dsl/stanza.rb

@@ -0,0 +1,53 @@
+#
+# The stanza { ... } class of Trema 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.
+#
+
+
+module Trema
+  module DSL
+    class Stanza
+      attr_reader :name
+
+
+      def initialize name = nil
+        @name = name
+      end
+
+
+      def [] attribute
+        instance_variable_get "@#{ attribute }".intern
+      end
+
+
+      def fetch attribute
+        v = self[ attribute ]
+        raise "#{ attribute } is not set" if v.nil?
+        v
+      end
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End: