trema 0.1.3

data/ruby/trema/ip.rb

@@ -0,0 +1,101 @@
+#
+# 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.
+#
+
+
+module Trema
+  #
+  # A wrapper class to IPAddr
+  #
+  class IP
+    require "ipaddr"
+
+
+    #
+    # @return [IPAddr] value object instance of proxied IPAddr.
+    #
+    attr_reader :value
+
+
+    #
+    # Creates a {IP} instance object as a proxy to IPAddr class.
+    #
+    # @overload initialize(addr)
+    #
+    # @param [String, Number] addr
+    #   an IPv4 address specified either as a String or Number.
+    #
+    # @param [Number] prefixlen
+    #   masking IPv4 address with given prefixlen.
+    #
+    # @raise [ArgumentError] invalid address if supplied argument is invalid
+    #   IPv4 address.
+    #
+    # @return [IP] self
+    #   a proxy to IPAddr.
+    #
+    def initialize addr, prefixlen = 32
+      if !addr.kind_of? String
+        @value = IPAddr.new( addr, Socket::AF_INET )
+      else
+        @value = IPAddr.new( addr )
+      end
+      if prefixlen < 32
+        @value = @value.mask( prefixlen )
+      end
+    end
+
+
+    #
+    # @return [ Number ] prefixlen of IPv4 address.
+    #
+    def prefixlen
+      range = @value.to_range
+      mask = range.first.to_i ^ range.last.to_i
+      masklen = 0
+      while mask != 0 do
+        mask = mask >> 1
+        masklen += 1
+      end
+      return 32 - masklen
+    end
+
+
+    #
+    # @return [String] the IPv4 address in its text representation.
+    #
+    def to_s
+      @value.to_s
+    end
+
+
+    #
+    # @return [Number] the IPv4 address in its numeric representation.
+    #
+    def to_i
+      @value.to_i
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/link.rb

@@ -0,0 +1,176 @@
+#
+# 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 "ifconfig"
+require "trema/network-component"
+
+
+module Trema
+  #
+  # Network link between hosts and switches.
+  #
+  class Link < NetworkComponent
+    #
+    # Returns the name of link interface
+    #
+    # @example
+    #   link.name => "trema3-0"
+    #
+    # @return [String]
+    #
+    attr_reader :name
+
+
+    #
+    # Returns the name of link peer interface
+    #
+    # @example
+    #   link.name => "trema3-1"
+    #
+    # @return [String]
+    #
+    attr_reader :name_peer
+
+
+    #
+    # Returns the configuration names of link peers
+    #
+    # @example
+    #   link.peers => [ "host 0", "switch 1" ]
+    #
+    # @return [Array]
+    #
+    attr_reader :peers
+
+
+    #
+    # Creates a new Trema link from {DSL::Link}
+    #
+    # @example
+    #   link = Trema::Link.new( stanza )
+    #
+    # @return [Link]
+    #
+    def initialize stanza
+      @link_id = Link.instances.size
+      @stanza = stanza
+      if real_eth?
+        @name = real_eth
+        @name_peer = nil
+        @peers = @stanza.peers - [ real_eth ]
+      else
+        @name = "trema#{ @link_id }-0"
+        @name_peer = "trema#{ @link_id }-1"
+        @peers = @stanza.peers
+      end
+      Link.add self
+    end
+
+
+    #
+    # Adds a virtual link
+    #
+    # @example
+    #   link.add!
+    #
+    # @return [undefined]
+    #
+    def add!
+      return if real_eth?
+      sh "sudo ip link add name #{ @name } type veth peer name #{ @name_peer }"
+      sh "sudo sysctl -w net.ipv6.conf.#{ @name }.disable_ipv6=1 >/dev/null 2>&1"
+      sh "sudo sysctl -w net.ipv6.conf.#{ @name_peer }.disable_ipv6=1 >/dev/null 2>&1"
+    end
+
+
+    #
+    # Ups the peer interfaces of a virtual link
+    #
+    # @example
+    #   link.up!
+    #
+    # @return [undefined]
+    #
+    def up!
+      return if real_eth?
+      sh "sudo /sbin/ifconfig #{ @name } up"
+      sh "sudo /sbin/ifconfig #{ @name_peer } up"
+    end
+
+
+    #
+    # Creates and enables a virtual link
+    #
+    # @example
+    #   link.enable!
+    #
+    # @return [undefined]
+    #
+    def enable!
+      add!
+      up!
+    end
+
+
+    #
+    # Deletes a virtual link
+    #
+    # @example
+    #   link.delete!
+    #
+    # @return [undefined]
+    #
+    def delete!
+      return if real_eth?
+      # FIXME: do not rescue nil
+      sh "sudo ip link delete #{ @name } 2>/dev/null" rescue nil
+    end
+
+
+    ############################################################################
+    private
+    ############################################################################
+
+
+    def real_eth
+      interfaces = IfconfigWrapper.new.parse.interfaces
+      @stanza.peers.each do | each |
+        return each if interfaces.include?( each )
+      end
+      raise
+    end
+
+
+    def real_eth?
+      interfaces = IfconfigWrapper.new.parse.interfaces
+      @stanza.peers.each do | each |
+        return true if interfaces.include?( each )
+      end
+      false
+    end
+  end
+end
+
+
+### Local variables:
+### mode: Ruby
+### coding: utf-8-unix
+### indent-tabs-mode: nil
+### End:

data/ruby/trema/list-switches-reply.c

@@ -0,0 +1,46 @@
+/*
+ * Author: Kazushi SUGYO
+ *
+ * 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.
+ */
+
+
+#include "trema.h"
+#include "ruby.h"
+
+
+void
+handle_list_switches_reply( const list_element *switches, void *controller ) {
+  if ( rb_respond_to( ( VALUE ) controller, rb_intern( "list_switches_reply" ) ) == Qfalse ) { 
+   return;
+  }
+  VALUE dpids = rb_ary_new();
+
+  const list_element *element = NULL;
+  for ( element = switches; element != NULL; element = element->next ) {
+    rb_ary_push( dpids, ULL2NUM( *( uint64_t * ) element->data ) );
+  }
+
+  rb_funcall( ( VALUE ) controller, rb_intern( "list_switches_reply" ), 1, dpids );
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/ruby/trema/list-switches-reply.h

@@ -0,0 +1,40 @@
+/*
+ * Author: Kazushi SUGYO
+ *
+ * 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 LIST_SWITCHES_REPLY_H
+#define LIST_SWITCHES_REPLY_H
+
+
+#include "ruby.h"
+#include "trema.h"
+
+
+void handle_list_switches_reply( const list_element *switches, void *user_data );
+
+
+#endif // LIST_SWITCHES_REPLY_H
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */

data/ruby/trema/logger.c

@@ -0,0 +1,162 @@
+/*
+ * 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.
+ */
+
+
+#include "checks.h"
+#include "log.h"
+#include "logger.h"
+
+
+extern VALUE mTrema;
+VALUE mLogger;
+
+
+static VALUE
+do_log( void ( *log_function )( const char *format, ... ), int argc, VALUE *argv ) {
+  VALUE message = rb_f_sprintf( argc, argv );
+  log_function( STR2CSTR( message ) );
+  return message;
+}
+
+
+/*
+ * @overload critical(format ...)
+ *   Outputs a message representing that "the system is completely
+ *   unusable" to log file.
+ *
+ *   @example
+ *     critical "Trema blue screen. Memory dump = %s", memory
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_critical( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( critical, argc, argv ) );
+}
+
+
+/*
+ * @overload error(format ...)
+ *   Outputs a message representing that "something went wrong" to log
+ *   file.
+ *
+ *   @example
+ *     error "Failed to accept %s", app_socket
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_error( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( error, argc, argv ) );
+}
+
+
+/*
+ * @overload warn(format ...)
+ *   Outputs a message representing that "something in the system was
+ *   not as expected" to log file.
+ *
+ *   @example
+ *     warn "%s: trema is already initialized", app_name
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_warn( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( warn, argc, argv ) );
+}
+
+
+/*
+ * @overload notice(format ...)
+ *   Outputs a message representing that "normal but significant
+ *   condition occurred" to log file.
+ *
+ *   @example
+ *     notice "The switch %s disconnected its secure channel connection", datapath_id
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_notice( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( notice, argc, argv ) );
+}
+
+
+/*
+ * @overload info(format ...)
+ *   Outputs an informational massage to log file.
+ *
+ *   @example
+ *     info "Hello world from %s!", datapath_id
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_info( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( info, argc, argv ) );
+}
+
+
+/*
+ * @overload debug(format ...)
+ *   Outputs a debug-level massage to log file.
+ *
+ *   @example
+ *     debug "Setting a packet_in handler: %s", method
+ *
+ *   @return [String] the string resulting from applying format to any
+ *     additional arguments.
+ */
+static VALUE
+logger_debug( int argc, VALUE *argv, VALUE self ) {
+  UNUSED( self );
+  return( do_log( debug, argc, argv ) );
+}
+
+
+void
+Init_logger() {
+  mLogger = rb_define_module_under( mTrema, "Logger" );
+
+  rb_define_method( mLogger, "critical", logger_critical, -1 );
+  rb_define_method( mLogger, "error", logger_error, -1 );
+  rb_define_method( mLogger, "warn", logger_warn, -1 );
+  rb_define_method( mLogger, "notice", logger_notice, -1 );
+  rb_define_method( mLogger, "info", logger_info, -1 );
+  rb_define_method( mLogger, "debug", logger_debug, -1 );
+}
+
+
+/*
+ * Local variables:
+ * c-basic-offset: 2
+ * indent-tabs-mode: nil
+ * End:
+ */